class IconColorEditor(QDialog):
"""An editor to let the user select an icon and a color for an object_class.
"""
def __init__(self, parent):
"""Init class."""
super().__init__(parent) # , Qt.Popup)
icon_size = QSize(32, 32)
self.icon_mngr = IconListManager(icon_size)
self.setWindowTitle("Select icon and color")
self.icon_widget = QWidget(self)
self.icon_list = QListView(self.icon_widget)
self.icon_list.setViewMode(QListView.IconMode)
self.icon_list.setIconSize(icon_size)
self.icon_list.setResizeMode(QListView.Adjust)
self.icon_list.setItemDelegate(_IconPainterDelegate(self))
self.icon_list.setMovement(QListView.Static)
self.icon_list.setMinimumHeight(400)
icon_widget_layout = QVBoxLayout(self.icon_widget)
icon_widget_layout.addWidget(QLabel("Font Awesome icons"))
self.line_edit = QLineEdit()
self.line_edit.setPlaceholderText("Search icons for...")
icon_widget_layout.addWidget(self.line_edit)
icon_widget_layout.addWidget(self.icon_list)
self.color_dialog = QColorDialog(self)
self.color_dialog.setWindowFlags(Qt.Widget)
self.color_dialog.setOption(QColorDialog.NoButtons, True)
self.color_dialog.setOption(QColorDialog.DontUseNativeDialog, True)
self.button_box = QDialogButtonBox(self)
self.button_box.setStandardButtons(QDialogButtonBox.Cancel
| QDialogButtonBox.Ok)
top_widget = QWidget(self)
top_layout = QHBoxLayout(top_widget)
top_layout.addWidget(self.icon_widget)
top_layout.addWidget(self.color_dialog)
layout = QVBoxLayout(self)
layout.addWidget(top_widget)
layout.addWidget(self.button_box)
self.proxy_model = QSortFilterProxyModel(self)
self.proxy_model.setSourceModel(self.icon_mngr.model)
self.proxy_model.filterAcceptsRow = self._proxy_model_filter_accepts_row
self.icon_list.setModel(self.proxy_model)
self.setAttribute(Qt.WA_DeleteOnClose)
self.connect_signals()
def _proxy_model_filter_accepts_row(self, source_row, source_parent):
"""Overridden method to filter icons according to search terms.
"""
text = self.line_edit.text()
if not text:
return QSortFilterProxyModel.filterAcceptsRow(
self.proxy_model, source_row, source_parent)
searchterms = self.icon_mngr.model.index(
source_row, 0, source_parent).data(Qt.UserRole + 1)
return any([text in term for term in searchterms])
def connect_signals(self):
"""Connect signals to slots."""
self.line_edit.textEdited.connect(self.proxy_model.invalidateFilter)
self.button_box.accepted.connect(self.accept)
self.button_box.rejected.connect(self.reject)
def set_data(self, data):
icon_code, color_code = interpret_icon_id(data)
self.icon_mngr.init_model()
for i in range(self.proxy_model.rowCount()):
index = self.proxy_model.index(i, 0)
if index.data(Qt.UserRole) == icon_code:
self.icon_list.setCurrentIndex(index)
break
self.color_dialog.setCurrentColor(QColor(color_code))
def data(self):
icon_code = self.icon_list.currentIndex().data(Qt.UserRole)
color_code = self.color_dialog.currentColor().rgb()
return make_icon_id(icon_code, color_code)
class LayersList(QWidget):
'''
LayerList class which acts as collapsable list.
'''
def __init__(self, name, layers, filter, expand=True):
super().__init__()
self.setWindowModality(QtCore.Qt.WindowModal)
self.currently_expanded = True
self.main_layout = QVBoxLayout()
self.main_layout.setMargin(0)
self.main_layout.setSpacing(0)
self.main_layout.setContentsMargins(0, 0, 0, 0)
self.expand_button = QPushButton(name)
self.expand_button.setToolTip(f"List of {name} Layers")
self.expand_button.setIcon(
QIcon(os.path.join(PATH, 'LayersList_Down.png')))
self.layer_list = QListView()
self.layer_list.setDragEnabled(True)
self.layer_list.setEditTriggers(QAbstractItemView.NoEditTriggers)
self.layer_list.setWrapping(False)
self.layer_list.setViewMode(self.layer_list.ListMode)
self.container_model = QStandardItemModel()
self.model = QSortFilterProxyModel()
self.model.setSourceModel(self.container_model)
self.model.setFilterCaseSensitivity(QtCore.Qt.CaseInsensitive)
#self.model.cas
filter.textChanged.connect(self.filter_model)
for l in layers:
self.container_model.appendRow(
QStandardItem(
QIcon(os.path.join(PATH, 'LayersList_Layer_Icon.png')), l))
self.layer_list.setModel(self.model)
self.main_layout.addWidget(self.expand_button, 0, Qt.AlignTop)
self.main_layout.addWidget(self.layer_list, 0, Qt.AlignTop)
self.expand_button.clicked.connect(self.expand)
self.setLayout(self.main_layout)
self.resized_size = len(layers) * (self.layer_list.sizeHintForRow(0) +
self.layer_list.frameWidth())
self.layer_list.setMaximumHeight(self.resized_size)
self.layer_list.setMinimumHeight(self.resized_size)
self.setMinimumWidth(self.layer_list.frameWidth())
self.set_styling()
if not expand:
self.expand()
@QtCore.Slot()
def expand(self):
if self.currently_expanded:
self.layer_list.setMinimumHeight(0)
self.currently_expanded = False
self.expand_button.setIcon(
QIcon(os.path.join(PATH, 'LayersList_Up2.png')))
self.layer_list.setMaximumHeight(0)
else:
self.layer_list.setMinimumHeight(self.resized_size)
self.currently_expanded = True
self.expand_button.setIcon(
QIcon(os.path.join(PATH, 'LayersList_Down.png')))
self.layer_list.setMaximumHeight(self.resized_size)
def set_styling(self):
self.setStyleSheet('''
background-color:white;
''')
self.expand_button.setStyleSheet('''
background-color:#d6d2d2;
text-align:left;
''')
@QtCore.Slot()
def filter_model(self, text):
self.show()
self.model.setFilterRegExp(QRegExp(text, QtCore.Qt.CaseInsensitive))
if not self.currently_expanded:
self.expand()
if self.model.rowCount() == 0:
self.hide()
