base implementation of ColorEntity view

openpype/tools/settings/settings/categories.py

@@ -21,6 +21,7 @@ from openpype.settings.entities import (
     TextEntity,
     PathInput,
     RawJsonEntity,
+    ColorEntity,
 
     DefaultsNotDefined,
     StudioDefaultsNotDefined,
@@ -44,7 +45,7 @@ from .item_widgets import (
     PathWidget,
     PathInputWidget
 )
-
+from .color_widget import ColorWidget
 from avalon.vendor import qtawesome
 
 
@@ -113,6 +114,9 @@ class SettingsCategoryWidget(QtWidgets.QWidget):
         elif isinstance(entity, RawJsonEntity):
             return RawJsonWidget(*args)
 
+        elif isinstance(entity, ColorEntity):
+            return ColorWidget(*args)
+
         elif isinstance(entity, BaseEnumEntity):
             return EnumeratorWidget(*args)
 

openpype/tools/settings/settings/color_widget.py

@@ -0,0 +1,174 @@
+from Qt import QtWidgets, QtCore, QtGui
+
+from .item_widgets import InputWidget
+
+from openpype.widgets.color_widgets import ColorPickerWidget
+
+
+class ColorWidget(InputWidget):
+    def _add_inputs_to_layout(self):
+        self.input_field = ColorViewer(self.content_widget)
+
+        self.setFocusProxy(self.input_field)
+
+        self.content_layout.addWidget(self.input_field, 1)
+
+        self.input_field.clicked.connect(self._on_click)
+
+        self._dialog = None
+
+    def _on_click(self):
+        if self._dialog:
+            self._dialog.open()
+            return
+
+        dialog = ColorDialog(self.input_field.color(), self)
+        self._dialog = dialog
+
+        dialog.open()
+        dialog.finished.connect(self._on_dialog_finish)
+
+    def _on_dialog_finish(self, *_args):
+        if not self._dialog:
+            return
+
+        color = self._dialog.result()
+        if color is not None:
+            self.input_field.set_color(color)
+            self._on_value_change()
+
+        self._dialog.deleteLater()
+        self._dialog = None
+
+    def _on_entity_change(self):
+        if self.entity.value != self.input_value():
+            self.set_entity_value()
+
+    def set_entity_value(self):
+        self.input_field.set_color(*self.entity.value)
+
+    def input_value(self):
+        color = self.input_field.color()
+        return [color.red(), color.green(), color.blue(), color.alpha()]
+
+    def _on_value_change(self):
+        if self.ignore_input_changes:
+            return
+
+        self.entity.set(self.input_value())
+
+
+class ColorViewer(QtWidgets.QWidget):
+    clicked = QtCore.Signal()
+
+    def __init__(self, parent=None):
+        super(ColorViewer, self).__init__(parent)
+
+        self.setMinimumSize(10, 10)
+
+        self.actual_pen = QtGui.QPen()
+        self.actual_color = QtGui.QColor()
+        self._checkerboard = None
+
+    def mouseReleaseEvent(self, event):
+        if event.button() == QtCore.Qt.LeftButton:
+            self.clicked.emit()
+        super(ColorViewer, self).mouseReleaseEvent(event)
+
+    def checkerboard(self):
+        if not self._checkerboard:
+            checkboard_piece_size = 10
+            color_1 = QtGui.QColor(188, 188, 188)
+            color_2 = QtGui.QColor(90, 90, 90)
+
+            pix = QtGui.QPixmap(
+                checkboard_piece_size * 2,
+                checkboard_piece_size * 2
+            )
+            pix_painter = QtGui.QPainter(pix)
+
+            rect = QtCore.QRect(
+                0, 0, checkboard_piece_size, checkboard_piece_size
+            )
+            pix_painter.fillRect(rect, color_1)
+            rect.moveTo(checkboard_piece_size, checkboard_piece_size)
+            pix_painter.fillRect(rect, color_1)
+            rect.moveTo(checkboard_piece_size, 0)
+            pix_painter.fillRect(rect, color_2)
+            rect.moveTo(0, checkboard_piece_size)
+            pix_painter.fillRect(rect, color_2)
+            pix_painter.end()
+            self._checkerboard = pix
+
+        return self._checkerboard
+
+    def color(self):
+        return self.actual_color
+
+    def set_color(self, *args):
+        # Create copy of entered color
+        self.actual_color = QtGui.QColor(*args)
+        # Repaint
+        self.update()
+
+    def set_alpha(self, alpha):
+        # Change alpha of current color
+        self.actual_color.setAlpha(alpha)
+        # Repaint
+        self.update()
+
+    def paintEvent(self, event):
+        rect = event.rect()
+
+        # Paint everything to pixmap as it has transparency
+        pix = QtGui.QPixmap(rect.width(), rect.height())
+        pix_painter = QtGui.QPainter(pix)
+        pix_painter.drawTiledPixmap(rect, self.checkerboard())
+        pix_painter.fillRect(rect, self.actual_color)
+        pix_painter.end()
+
+        painter = QtGui.QPainter(self)
+        painter.drawPixmap(rect, pix)
+        painter.end()
+
+
+class ColorDialog(QtWidgets.QDialog):
+    def __init__(self, color=None, parent=None):
+        super(ColorDialog, self).__init__(parent)
+
+        self.setWindowTitle("Color picker dialog")
+
+        picker_widget = ColorPickerWidget(color, self)
+
+        footer_widget = QtWidgets.QWidget(self)
+        footer_layout = QtWidgets.QHBoxLayout(footer_widget)
+
+        ok_btn = QtWidgets.QPushButton("Ok", footer_widget)
+        cancel_btn = QtWidgets.QPushButton("Cancel", footer_widget)
+
+        footer_layout.addWidget(ok_btn)
+        footer_layout.addWidget(cancel_btn)
+        footer_layout.addWidget(QtWidgets.QWidget(self), 1)
+
+        layout = QtWidgets.QVBoxLayout(self)
+
+        layout.addWidget(picker_widget, 1)
+        layout.addWidget(footer_widget, 0)
+
+        ok_btn.clicked.connect(self.on_ok_clicked)
+        cancel_btn.clicked.connect(self.on_cancel_clicked)
+
+        self.picker_widget = picker_widget
+
+        self._result = None
+
+    def on_ok_clicked(self):
+        self._result = self.picker_widget.color()
+        self.close()
+
+    def on_cancel_clicked(self):
+        self._result = None
+        self.close()
+
+    def result(self):
+        return self._result