mirror of
https://github.com/PaddlePaddle/PaddleOCR.git
synced 2025-08-11 10:07:17 +00:00
48 lines
1.4 KiB
Python
48 lines
1.4 KiB
Python
|
# -*- encoding: utf-8 -*-
|
||
|
|
|
||
|
|
from PyQt5.QtCore import Qt
|
||
|
|
from PyQt5 import QtWidgets
|
||
|
|
|
||
|
|
|
||
|
|
class EscapableQListWidget(QtWidgets.QListWidget):
|
||
|
|
def keyPressEvent(self, event):
|
||
|
|
super(EscapableQListWidget, self).keyPressEvent(event)
|
||
|
|
if event.key() == Qt.Key_Escape:
|
||
|
|
self.clearSelection()
|
||
|
|
|
||
|
|
|
||
|
|
class UniqueLabelQListWidget(EscapableQListWidget):
|
||
|
|
def mousePressEvent(self, event):
|
||
|
|
super(UniqueLabelQListWidget, self).mousePressEvent(event)
|
||
|
|
if not self.indexAt(event.pos()).isValid():
|
||
|
|
self.clearSelection()
|
||
|
|
|
||
|
|
def findItemsByLabel(self, label):
|
||
|
|
items = []
|
||
|
|
for row in range(self.count()):
|
||
|
|
item = self.item(row)
|
||
|
|
if item.data(Qt.UserRole) == label:
|
||
|
|
items.append(item)
|
||
|
|
return items
|
||
|
|
|
||
|
|
def createItemFromLabel(self, label):
|
||
|
|
item = QtWidgets.QListWidgetItem()
|
||
|
|
item.setData(Qt.UserRole, label)
|
||
|
|
return item
|
||
|
|
|
||
|
|
def setItemLabel(self, item, label, color=None):
|
||
|
|
qlabel = QtWidgets.QLabel()
|
||
|
|
if color is None:
|
||
|
|
qlabel.setText("{}".format(label))
|
||
|
|
else:
|
||
|
|
qlabel.setText(
|
||
|
|
'{} <font color="#{:02x}{:02x}{:02x}">●</font>'.format(
|
||
|
|
label, *color
|
||
|
|
)
|
||
|
|
)
|
||
|
|
qlabel.setAlignment(Qt.AlignBottom)
|
||
|
|
|
||
|
|
item.setSizeHint(qlabel.sizeHint())
|
||
|
|
|
||
|
|
self.setItemWidget(item, qlabel)
|