labelDialog.py 4.3 KB

123456789101112131415161718192021222324252627282930313233343536373839404142434445464748495051525354555657585960616263646566676869707172737475767778798081828384858687888990919293949596979899100101102103104105106107108109110111112113114115116117118119120121122123124125126127128129
  1. #
  2. # Copyright (C) 2011 Michael Pitidis, Hussein Abdulwahid.
  3. #
  4. # This file is part of Labelme.
  5. #
  6. # Labelme is free software: you can redistribute it and/or modify
  7. # it under the terms of the GNU General Public License as published by
  8. # the Free Software Foundation, either version 3 of the License, or
  9. # (at your option) any later version.
  10. #
  11. # Labelme is distributed in the hope that it will be useful,
  12. # but WITHOUT ANY WARRANTY; without even the implied warranty of
  13. # MERCHANTABILITY or FITNESS FOR A PARTICULAR PURPOSE. See the
  14. # GNU General Public License for more details.
  15. #
  16. # You should have received a copy of the GNU General Public License
  17. # along with Labelme. If not, see <http://www.gnu.org/licenses/>.
  18. #
  19. try:
  20. from PyQt5 import QtCore
  21. from PyQt5 import QtGui
  22. from PyQt5 import QtWidgets
  23. PYQT5 = True
  24. except ImportError:
  25. from PyQt4 import QtCore
  26. from PyQt4 import QtGui
  27. from PyQt4 import QtGui as QtWidgets
  28. PYQT5 = False
  29. from .lib import labelValidator
  30. from .lib import newIcon
  31. # TODO(unknown):
  32. # - Calculate optimal position so as not to go out of screen area.
  33. class LabelQLineEdit(QtWidgets.QLineEdit):
  34. def setListWidget(self, list_widget):
  35. self.list_widget = list_widget
  36. def keyPressEvent(self, e):
  37. if e.key() in [QtCore.Qt.Key_Up, QtCore.Qt.Key_Down]:
  38. self.list_widget.keyPressEvent(e)
  39. else:
  40. super(LabelQLineEdit, self).keyPressEvent(e)
  41. class LabelDialog(QtWidgets.QDialog):
  42. def __init__(self, text="Enter object label", parent=None, labels=None,
  43. sort_labels=True):
  44. super(LabelDialog, self).__init__(parent)
  45. self.edit = LabelQLineEdit()
  46. self.edit.setPlaceholderText(text)
  47. self.edit.setValidator(labelValidator())
  48. self.edit.editingFinished.connect(self.postProcess)
  49. layout = QtWidgets.QVBoxLayout()
  50. layout.addWidget(self.edit)
  51. # buttons
  52. self.buttonBox = bb = QtWidgets.QDialogButtonBox(
  53. QtWidgets.QDialogButtonBox.Ok | QtWidgets.QDialogButtonBox.Cancel,
  54. QtCore.Qt.Horizontal,
  55. self,
  56. )
  57. bb.button(bb.Ok).setIcon(newIcon('done'))
  58. bb.button(bb.Cancel).setIcon(newIcon('undo'))
  59. bb.accepted.connect(self.validate)
  60. bb.rejected.connect(self.reject)
  61. layout.addWidget(bb)
  62. # label_list
  63. self.labelList = QtWidgets.QListWidget()
  64. self._sort_labels = sort_labels
  65. if labels:
  66. self.labelList.addItems(labels)
  67. if self._sort_labels:
  68. self.labelList.sortItems()
  69. else:
  70. self.labelList.setDragDropMode(
  71. QtGui.QAbstractItemView.InternalMove)
  72. self.labelList.currentItemChanged.connect(self.labelSelected)
  73. self.edit.setListWidget(self.labelList)
  74. layout.addWidget(self.labelList)
  75. self.setLayout(layout)
  76. # completion
  77. completer = QtGui.QCompleter()
  78. completer.setCompletionMode(QtGui.QCompleter.InlineCompletion)
  79. completer.setModel(self.labelList.model())
  80. self.edit.setCompleter(completer)
  81. def addLabelHistory(self, label):
  82. if self.labelList.findItems(label, QtCore.Qt.MatchExactly):
  83. return
  84. self.labelList.addItem(label)
  85. if self._sort_labels:
  86. self.labelList.sortItems()
  87. def labelSelected(self, item):
  88. self.edit.setText(item.text())
  89. def validate(self):
  90. if PYQT5:
  91. if self.edit.text().strip():
  92. self.accept()
  93. else:
  94. if self.edit.text().trimmed():
  95. self.accept()
  96. def postProcess(self):
  97. if PYQT5:
  98. self.edit.setText(self.edit.text().strip())
  99. else:
  100. self.edit.setText(self.edit.text().trimmed())
  101. def popUp(self, text=None, move=True):
  102. # if text is None, the previous label in self.edit is kept
  103. if text is not None:
  104. self.edit.setText(text)
  105. self.edit.setSelection(0, len(text))
  106. items = self.labelList.findItems(text, QtCore.Qt.MatchFixedString)
  107. if items:
  108. assert len(items) == 1
  109. self.labelList.setCurrentItem(items[0])
  110. self.edit.setFocus(QtCore.Qt.PopupFocusReason)
  111. if move:
  112. self.move(QtGui.QCursor.pos())
  113. return self.edit.text() if self.exec_() else None