From aa9fe148021425e0e87cd122ff5e6b742a15e8b7 Mon Sep 17 00:00:00 2001 From: LaraStuStu Date: Sat, 28 Mar 2020 14:02:22 +0800 Subject: [PATCH] Create qt.py --- DataAnnotation/labelme/labelme/utils/qt.py | 88 ++++++++++++++++++++++ 1 file changed, 88 insertions(+) create mode 100644 DataAnnotation/labelme/labelme/utils/qt.py diff --git a/DataAnnotation/labelme/labelme/utils/qt.py b/DataAnnotation/labelme/labelme/utils/qt.py new file mode 100644 index 0000000..a7f7344 --- /dev/null +++ b/DataAnnotation/labelme/labelme/utils/qt.py @@ -0,0 +1,88 @@ +from math import sqrt +import os.path as osp + +import numpy as np + +from qtpy import QtCore +from qtpy import QtGui +from qtpy import QtWidgets + + +here = osp.dirname(osp.abspath(__file__)) + + +def newIcon(icon): + icons_dir = osp.join(here, '../icons') + return QtGui.QIcon(osp.join(':/', icons_dir, '%s.png' % icon)) + + +def newButton(text, icon=None, slot=None): + b = QtWidgets.QPushButton(text) + if icon is not None: + b.setIcon(newIcon(icon)) + if slot is not None: + b.clicked.connect(slot) + return b + + +def newAction(parent, text, slot=None, shortcut=None, icon=None, + tip=None, checkable=False, enabled=True): + """Create a new action and assign callbacks, shortcuts, etc.""" + a = QtWidgets.QAction(text, parent) + if icon is not None: + a.setIconText(text.replace(' ', '\n')) + a.setIcon(newIcon(icon)) + if shortcut is not None: + if isinstance(shortcut, (list, tuple)): + a.setShortcuts(shortcut) + else: + a.setShortcut(shortcut) + if tip is not None: + a.setToolTip(tip) + a.setStatusTip(tip) + if slot is not None: + a.triggered.connect(slot) + if checkable: + a.setCheckable(True) + a.setEnabled(enabled) + return a + + +def addActions(widget, actions): + for action in actions: + if action is None: + widget.addSeparator() + elif isinstance(action, QtWidgets.QMenu): + widget.addMenu(action) + else: + widget.addAction(action) + + +def labelValidator(): + return QtGui.QRegExpValidator(QtCore.QRegExp(r'^[^ \t].+'), None) + + +class struct(object): + def __init__(self, **kwargs): + self.__dict__.update(kwargs) + + +def distance(p): + return sqrt(p.x() * p.x() + p.y() * p.y()) + + +def distancetoline(point, line): + p1, p2 = line + p1 = np.array([p1.x(), p1.y()]) + p2 = np.array([p2.x(), p2.y()]) + p3 = np.array([point.x(), point.y()]) + if np.dot((p3 - p1), (p2 - p1)) < 0: + return np.linalg.norm(p3 - p1) + if np.dot((p3 - p2), (p1 - p2)) < 0: + return np.linalg.norm(p3 - p2) + return np.linalg.norm(np.cross(p2 - p1, p1 - p3)) / np.linalg.norm(p2 - p1) + + +def fmtShortcut(text): + mod, key = text.split('+', 1) + return '%s+%s' % (mod, key) -- GitLab