def initUI(self):
# Create some widgets (these won't appear immediately):
self.leftWidget = QWidget()
self.pointsCombobox = QComboBox()
self.pointsCombobox.setFocusPolicy(Qt.NoFocus)
self.pointsCombobox.currentIndexChanged.connect(self.datasetChanged)
self.leftWidget.setLayout(QFormLayout())
self.leftWidget.layout().addRow("Point sets", self.pointsCombobox)
self.addSlider(self.leftWidget.layout(),
text="Point Size",
max=15,
callback=self.update_point_size)
cDialog = QColorDialog(self)
cDialog.currentColorChanged.connect(self.update_color)
cDialog.setWindowFlags(Qt.Widget)
cDialog.setOptions(QColorDialog.DontUseNativeDialog
| QColorDialog.NoButtons)
cDialog.setFocusPolicy(Qt.NoFocus)
self.leftWidget.layout().addRow(cDialog)
self.leftWidget.setFixedWidth(600)
self.mywidget = Window(data_filepath=None)
# Put the widgets in a layout (now they start to appear):
layout = QGridLayout(self)
layout.setContentsMargins(0, 0, 0, 0)
layout.addWidget(self.mywidget, 0, 1)
layout.addWidget(self.leftWidget, 0, 0)
self.widget.setLayout(layout)
def showColorDialog(self):
# leave fullscreen mode while the color selector is displayed
# to prevent the color selector falling behind the fullscreen
# window and the app being unclosable
oldstate = self.windowState()
self.setWindowState(oldstate & ~Qt.WindowFullScreen)
tmpcolor = self.color
def setColor(color):
nonlocal tmpcolor
self.setColor(color)
tmpcolor = None
color_dlg = QColorDialog()
color_dlg.setCurrentColor(self.color)
color_dlg.currentColorChanged.connect(self.setColor)
color_dlg.colorSelected.connect(setColor)
flags = self.windowFlags()
color_dlg.setWindowFlags(flags | Qt.WindowStaysOnTopHint)
color_dlg.exec_()
if tmpcolor is not None:
self.setColor(tmpcolor)
self.setWindowState(oldstate)
def on_pressed(self):
"""Button pressed event handler
Shows color dialog and sets the chosen color.
"""
dlg = QColorDialog(self.parent())
dlg.setCurrentColor(self.color)
dlg.setWindowTitle(self.title)
dlg.setWindowFlags(Qt.Tool | Qt.FramelessWindowHint)
dlg.setWindowModality(Qt.ApplicationModal)
dlg.setOptions(QColorDialog.DontUseNativeDialog)
p = self.mapFromGlobal(QCursor.pos())
p.setX(p.x() + (self.rect().width() / 2))
p.setY(p.y() + (self.rect().height() / 2))
dlg.move(self.mapToGlobal(p))
if dlg.exec_():
self.color = dlg.currentColor()
self.colorChanged.emit()
class ColorPicker(QToolButton):
colorReset = pyqtSignal()
def __init__(self,
action: QAction = None,
color_group=None,
default_color=Qt.blue,
*args,
**kwargs):
super().__init__(*args, **kwargs)
self.last_color_action = action if action is not None else QAction(
self)
self._color_group = color_group
dialog_action = QWidgetAction(self)
self._dialog = QColorDialog(self)
self.btReset = None
button_box = self._dialog.findChild(QDialogButtonBox, "")
if button_box is not None:
self.btReset = button_box.addButton(QDialogButtonBox.Reset)
self._dialog.setWindowFlags(Qt.Widget)
self._dialog.setOptions(self._dialog.options()
| QColorDialog.DontUseNativeDialog)
dialog_action.setDefaultWidget(self._dialog)
# Hide pick screen color button
button = self._dialog.findChild(QAbstractButton)
if button:
button.hide()
# Load custom colors
settings = QSettings()
settings.beginGroup("Colors")
custom_colors = settings.value("Custom", [])
for i, color in enumerate(custom_colors):
self._dialog.setCustomColor(i, color)
current_color = QColor(default_color) if color_group is None\
else settings.value(f"{color_group}/Current", QColor(default_color), type=QColor)
if current_color.isValid():
self._dialog.setCurrentColor(current_color)
self.on_color_selected(current_color)
settings.endGroup()
menu = QMenu()
menu.addAction(dialog_action)
self.setMenu(menu)
self.setPopupMode(QToolButton.MenuButtonPopup)
self.setDefaultAction(self.last_color_action)
# Connect events
self.colorSelected = self._dialog.colorSelected
if self.btReset is not None:
self.btReset.clicked.connect(self.colorReset.emit)
self.currentColorChanged = self._dialog.currentColorChanged
menu.aboutToShow.connect(self._dialog.show)
self._dialog.rejected.connect(menu.hide)
self._dialog.colorSelected.connect(menu.hide)
self._dialog.colorSelected.connect(self.on_color_selected)
self.last_color_action.triggered.connect(self.on_use_last_color)
def setIcon(self, icon: QIcon):
super().setIcon(icon)
self.update_icon()
def showEvent(self, *args, **kwargs):
self.update_icon()
super().showEvent(*args, **kwargs)
# noinspection PyUnusedLocal
def on_color_selected(self, color):
self.update_icon()
self.save_settings()
def save_settings(self):
settings = QSettings()
settings.beginGroup("Colors")
custom_colors = [
self._dialog.customColor(i)
for i in range(self._dialog.customCount())
]
settings.setValue('Custom', custom_colors)
if self._color_group is not None:
settings.setValue(f"{self._color_group}/Current",
self._dialog.currentColor())
def update_icon(self):
"""Show selected color on top-left edge of icon"""
current_icon = self.icon()
icon = QIcon()
for size in current_icon.availableSizes():
pixmap = current_icon.pixmap(size)
painter = QPainter(pixmap)
painter.setPen(Qt.NoPen)
painter.setBrush(QBrush(self._dialog.currentColor()))
painter.drawEllipse(0, 0, int(size.width() / 4),
int(size.height() / 4))
painter.end()
icon.addPixmap(pixmap)
super().setIcon(icon)
def on_use_last_color(self):
self.colorSelected.emit(self._dialog.currentColor())
