Update

2023/scripts/animation_tools/dwpicker/quick.py

@@ -0,0 +1,120 @@
+from .pyside import QtWidgets, QtGui, QtCore
+from maya import cmds
+
+from .colorwheel import ColorDialog
+from .optionvar import (
+    save_optionvar, DEFAULT_LABEL, DEFAULT_HEIGHT, DEFAULT_WIDTH,
+    DEFAULT_TEXT_COLOR, DEFAULT_BG_COLOR)
+
+
+class QuickOptions(QtWidgets.QWidget):
+    def __init__(self, parent=None):
+        super(QuickOptions, self).__init__(parent=parent)
+        self.bg_color = ColorButton()
+        self.bg_color.colorChanged.connect(self.save_ui_states)
+        self.text_color = ColorButton()
+        self.text_color.colorChanged.connect(self.save_ui_states)
+        validator = QtGui.QIntValidator()
+        self.width = QtWidgets.QLineEdit()
+        self.width.returnPressed.connect(self.save_ui_states)
+        self.width.setValidator(validator)
+        self.width.setFixedWidth(50)
+        self.height = QtWidgets.QLineEdit()
+        self.height.returnPressed.connect(self.save_ui_states)
+        self.height.setValidator(validator)
+        self.height.setFixedWidth(50)
+        self.label = QtWidgets.QLineEdit()
+        self.label.returnPressed.connect(self.save_ui_states)
+
+        self.layout = QtWidgets.QHBoxLayout(self)
+        self.layout.setSpacing(0)
+        self.layout.addWidget(QtWidgets.QLabel('Bg-color: '))
+        self.layout.addWidget(self.bg_color)
+        self.layout.addSpacing(12)
+        self.layout.addWidget(QtWidgets.QLabel('Text-color: '))
+        self.layout.addWidget(self.text_color)
+        self.layout.addSpacing(12)
+        self.layout.addWidget(QtWidgets.QLabel('Size: '))
+        self.layout.addWidget(self.width)
+        self.layout.addWidget(self.height)
+        self.layout.addSpacing(12)
+        self.layout.addWidget(QtWidgets.QLabel('Label: '))
+        self.layout.addWidget(self.label)
+
+        self.load_ui_states()
+
+    def save_ui_states(self, *_):
+        values = self.values
+        save_optionvar(DEFAULT_BG_COLOR, values['bgcolor.normal'])
+        save_optionvar(DEFAULT_TEXT_COLOR, values['text.color'])
+        save_optionvar(DEFAULT_WIDTH, values['shape.width'])
+        save_optionvar(DEFAULT_HEIGHT, values['shape.height'])
+        save_optionvar(DEFAULT_LABEL, values['text.content'])
+
+    def load_ui_states(self):
+        self.values = {
+            'bgcolor.normal': cmds.optionVar(query=DEFAULT_BG_COLOR),
+            'text.color': cmds.optionVar(query=DEFAULT_TEXT_COLOR),
+            'shape.width': cmds.optionVar(query=DEFAULT_WIDTH),
+            'shape.height': cmds.optionVar(query=DEFAULT_HEIGHT),
+            'text.content': cmds.optionVar(query=DEFAULT_LABEL)}
+
+    @property
+    def values(self):
+        return {
+            'bgcolor.normal': self.bg_color.name,
+            'text.color': self.text_color.name,
+            'shape.width': int(self.width.text()) if self.width.text() else 10,
+            'shape.height': int(self.height.text()) if self.height.text() else 10,
+            'text.content': self.label.text()}
+
+    @values.setter
+    def values(self, values):
+        self.bg_color.name = values['bgcolor.normal']
+        self.text_color.name = values['text.color']
+        self.width.setText(str(values['shape.width']))
+        self.height.setText(str(values['shape.height']))
+        self.label.setText(str(values['text.content']))
+
+
+class ColorButton(QtWidgets.QAbstractButton):
+    colorChanged = QtCore.Signal()
+
+    def __init__(self, parent=None):
+        super(ColorButton, self).__init__(parent=parent)
+        self.setFixedSize(20, 20)
+        self.color = QtGui.QColor(QtCore.Qt.black)
+        self.released.connect(self.pick_color)
+
+    def pick_color(self):
+        dialog = ColorDialog(self.name)
+        if not dialog.exec_():
+            return
+        self.name = dialog.colorname()
+        self.colorChanged.emit()
+        self.update()
+
+    @property
+    def name(self):
+        return self.color.name()
+
+    @name.setter
+    def name(self, value):
+        self.color.setNamedColor(value)
+
+    def paintEvent(self, _):
+        try:
+            painter = QtGui.QPainter()
+            painter.begin(self)
+            painter.setBrush(QtGui.QBrush(self.color))
+            if self.rect().contains(QtGui.QCursor.pos()):
+                color = QtCore.Qt.transparent
+            else:
+                color = QtCore.Qt.gray
+            painter.setPen(QtGui.QPen(color))
+            painter.drawRect(self.rect())
+        except BaseException:
+            pass  # avoid crash
+            # TODO: log the error
+        finally:
+            painter.end()