serialDimmerWidget.py

import math
from PyQt5.QtCore import Qt
from PyQt5.QtGui import QColor
from PyQt5.QtWidgets import QApplication, QHBoxLayout, QLabel, QPushButton, QVBoxLayout, QWidget, QColorDialog,\
                            QSlider, QStyleOptionSlider, QStyle

from serialWidgets import SerialMonitor
from serialhelper import SerialHelper
import sys
from myutils import *
from RGBDimmer import *

DIMMER_WIDGET_STYLE = """
    QSlider::handle:horizontal {
                    background: qlineargradient(x1:0, y1:0, x2:1, y2:1, stop:0 #b4b4b4, stop:1 #8f8f8f);
                    border: 1px solid #5c5c5c;
                    width: 10px;
                    margin: -2px 0; /* handle is placed by default on the contents rect of the groove. Expand outside the groove */
                    border-radius: 3px;
                    }
     QSlider::handle:vertical {
                    background: qlineargradient(x1:0, y1:0, x2:1, y2:1, stop:0 #b4b4b4, stop:1 #8f8f8f);
                    border: 1px solid #5c5c5c;
                    height: 10px;
                    margin: -2px 0; /* handle is placed by default on the contents rect of the groove. Expand outside the groove */
                    border-radius: 3px;
                    }
    QSlider::groove:horizontal {
                    border: 1px solid #999999;
                    /* height: 8px; */                         
                    margin: 2px 0;
                    }
    QSlider::groove:vertical {
                    border: 1px solid #999999;
                    /* height: 8px; */ 
                    margin: 2px 0;
                        }
                        """

class ColorPickerWidget(QColorDialog):
    def __init__(self, parent=None):
        super(ColorPickerWidget,self).__init__(parent)
        self.setOptions(self.options() | QColorDialog.DontUseNativeDialog)

        for children in self.findChildren(QWidget):
            classname = children.metaObject().className()
            if classname not in ("QColorPicker", "QColorLuminancePicker"):
                children.hide()



class ClickSlider(QSlider):
    def __init__(self, *args, **kwargs):
        super(ClickSlider, self).__init__(*args, **kwargs)
    def mousePressEvent(self, event):
        super(ClickSlider, self).mousePressEvent(event)
        if event.button() == Qt.LeftButton:
            val = self.pixelPosToRangeValue(event.pos())
            self.setValue(val)

    def pixelPosToRangeValue(self, pos):
        opt = QStyleOptionSlider()
        self.initStyleOption(opt)
        gr = self.style().subControlRect(QStyle.CC_Slider, opt, QStyle.SC_SliderGroove, self)
        sr = self.style().subControlRect(QStyle.CC_Slider, opt, QStyle.SC_SliderHandle, self)

        if self.orientation() == Qt.Horizontal:
            sliderLength = sr.width()
            sliderMin = gr.x()
            sliderMax = gr.right() - sliderLength + 1
        else:
            sliderLength = sr.height()
            sliderMin = gr.y()
            sliderMax = gr.bottom() - sliderLength + 1
        pr = pos - sr.center() + sr.topLeft()
        p = pr.x() if self.orientation() == Qt.Horizontal else pr.y()
        return QStyle.sliderValueFromPosition(self.minimum(), self.maximum(), p - sliderMin,
                                               sliderMax - sliderMin, opt.upsideDown)

class TemperatureSlider(ClickSlider):
    STYLE = """         
               TemperatureSlider::groove:horizontal {
                    background-color: qlineargradient(x1: 0, y1: 0, x2: 1, y2: 0,
                                    stop: 0.000 #ff3800
                                    stop: 0.045 #ff6d00
                                    stop: 0.091 #ff8a12
                                    stop: 0.136 #ffa148
                                    stop: 0.182 #ffb46b
                                    stop: 0.227 #ffc489
                                    stop: 0.273 #ffd1a3
                                    stop: 0.318 #ffdbba
                                    stop: 0.364 #ffe4ce
                                    stop: 0.409 #ffece0
                                    stop: 0.455 #fff3ef
                                    stop: 0.500 #fff9fd
                                    stop: 0.545 #f5f3ff
                                    stop: 0.591 #ebeeff
                                    stop: 0.636 #e3e9ff
                                    stop: 0.682 #dce5ff
                                    stop: 0.727 #d6e1ff
                                    stop: 0.773 #d0deff
                                    stop: 0.818 #cfdaff
                                    stop: 0.864 #ccd8ff
                                    stop: 0.909 #c8d5ff
                                    stop: 0.955 #c5d3ff
                                    stop: 1.000 #c3d1ff);
                }
                TemperatureSlider::groove:vertical {
                    background-color: qlineargradient(x1: 0, y1: 0, x2: 0, y2: 1,
                                    stop: 0.000 #ff3800
                                    stop: 0.045 #ff6d00
                                    stop: 0.091 #ff8a12
                                    stop: 0.136 #ffa148
                                    stop: 0.182 #ffb46b
                                    stop: 0.227 #ffc489
                                    stop: 0.273 #ffd1a3
                                    stop: 0.318 #ffdbba
                                    stop: 0.364 #ffe4ce
                                    stop: 0.409 #ffece0
                                    stop: 0.455 #fff3ef
                                    stop: 0.500 #fff9fd
                                    stop: 0.545 #f5f3ff
                                    stop: 0.591 #ebeeff
                                    stop: 0.636 #e3e9ff
                                    stop: 0.682 #dce5ff
                                    stop: 0.727 #d6e1ff
                                    stop: 0.773 #d0deff
                                    stop: 0.818 #cfdaff
                                    stop: 0.864 #ccd8ff
                                    stop: 0.909 #c8d5ff
                                    stop: 0.955 #c5d3ff
                                    stop: 1.000 #c3d1ff);
                }
            """
    def __init__(self, *args, **kwargs):
        super(TemperatureSlider, self).__init__(*args, **kwargs)        

        self.setRange(RGBDimmerController.MIN_TEMPERATURE,RGBDimmerController.MAX_TEMPERATURE)
        self.setMouseTracking(True)
        self.setSingleStep(100)   
        self.setStyleSheet(self.STYLE)

class HueSlider(ClickSlider):
#  HueSlider{
#                     background-color: qlineargradient(x1: 0, y1: 0, x2: 1, y2: 0,
#                                     stop: 0 red, 
#                                     stop: 0.17 #ff0,
#                                     stop: 0.33 lime,
#                                     stop: 0.5 cyan,
#                                     stop: 0.66 blue
#                                     stop: 0.83 #f0f
#                                     stop: 1 red);
#                 }
    STYLE = """         
               HueSlider::groove:horizontal {
                    background-color: qlineargradient(x1: 0, y1: 0, x2: 1, y2: 0,
                                    stop: 0 red, 
                                    stop: 0.17 #ff0,
                                    stop: 0.33 lime,
                                    stop: 0.5 cyan,
                                    stop: 0.66 blue
                                    stop: 0.83 #f0f
                                    stop: 1 red);
                }
                HueSlider::groove:vertical {       
                    background-color: qlineargradient(x1: 0, y1: 0, x2: 0, y2: 1,
                                    stop: 0 red, 
                                    stop: 0.17 #ff0,
                                    stop: 0.33 lime,
                                    stop: 0.5 cyan,
                                    stop: 0.66 blue
                                    stop: 0.83 #f0f
                                    stop: 1 red);
                }
            """
    def __init__(self, *args, **kwargs):
        super(HueSlider, self).__init__(*args, **kwargs)        

        self.setRange(RGBDimmerController.MIN_HUE,RGBDimmerController.MAX_HUE)
        self.setMouseTracking(True)
        self.setSingleStep(5)   
        self.setStyleSheet(self.STYLE)
        self.setObjectName("hue")
        for children in self.findChildren(super):
            classname = children.metaObject().className()
            print(classname)

class BrightnessSlider(ClickSlider):
    STYLE_TEMPLATE = """BrightnessSlider::groove:horizontal {{
                            background-color: qlineargradient(x1: 0, y1: 0, x2: 1, y2: 0,
                                stop: 0 black,                                
                                stop: 1 "{0}");
                        }}
                        BrightnessSlider::groove:vertical {{
                            background-color: qlineargradient(x1: 0, y1: 0, x2: 0, y2: 1,
                                stop: 0 black,                                
                                stop: 1 "{0}");
                        }}
                        """
    
    def __init__(self, *args, **kwargs):
        super(BrightnessSlider, self).__init__(*args, **kwargs)        

        self.setRange(RGBDimmerController.MIN_BRIGHTNESS,RGBDimmerController.MAX_BRIGHTNESS)
        self.setMouseTracking(True)
        self.setSingleStep(5)  
        self.set_color("white")
     

    def set_color(self,hexrgb):
     
        self.setStyleSheet(self.STYLE_TEMPLATE.format(hexrgb))

class RGBDimmerWidget(QWidget):
    def __init__(self, rgbDimmer = 0):
        super(RGBDimmerWidget,self).__init__()

      
        # self.serial_monitor.serial = RGBDimmerSerial()

        if rgbDimmer == 0:
            self.dimmer = RGBDimmerController()
        else :
            self.dimmer = rgbDimmer

        self.on_btn = QPushButton("On",self)
        self.on_btn.clicked.connect(self.on_button)

        # self.color_btn = QPushButton("Set Color",self)
        # self.color_btn.clicked.connect(self.set_color)


        self.color_dialog = ColorPickerWidget()
      
        self.color_dialog.currentColorChanged.connect(self.colorSelected)
      
        
        # self.color_dialog.colorSelected.connect(lambda: print("Color Changed")) 
       

        self.temperature_slider = TemperatureSlider(Qt.Horizontal, self)
        self.temperature_slider.valueChanged[int].connect(self.set_temperature)

        self.hue_slider = HueSlider(Qt.Horizontal, self)
        self.hue_slider.valueChanged[int].connect(self.set_hue)

        self.brighness_slider = BrightnessSlider(Qt.Horizontal, self)
        self.brighness_slider.valueChanged[int].connect(self.set_brightness)

        self.sliders_lo = QVBoxLayout()
        self.sliders_lo.addWidget(self.temperature_slider)
        self.sliders_lo.addWidget(self.hue_slider)
        self.sliders_lo.addWidget(self.brighness_slider)

        left_column_lo = QVBoxLayout(self)

        left_column_lo.addWidget(self.on_btn)
        # left_column_lo.addWidget(self.color_btn)
        left_column_lo.addLayout(self.sliders_lo)

        # left_column_lo.addWidget(self.temperature_slider)
        # left_column_lo.addWidget(self.hue_slider)
        # left_column_lo.addWidget(self.brighness_slider)

        left_column_lo.addWidget(self.color_dialog)

        c = QColor(255,0,0)
        self.color_dialog.setCurrentColor(c)
        self.colorSelected(c)

    def on_button(self):
        self.dimmer.sendCommand(3)
    
    def set_temperature(self, value):
        b = self.color_dialog.currentColor().value()
        k = Kelvin2RGB(value,math.ceil(mapValue(b,0.0,255.0,Kelvin2RGB.MIN_BRIGHTNESS,Kelvin2RGB.MAX_BRIGHTNESS)))
        self.color_dialog.blockSignals(True)
        self.color_dialog.setCurrentColor(QColor(k.red,k.green,k.blue))
        self.color_dialog.blockSignals(False)
        self.dimmer.setTemperature(value)
        self.brighness_slider.set_color(rgb_to_hex((k.red,k.green,k.blue),True))
  
    def set_hue(self, value):
        b = self.color_dialog.currentColor().value()
        c = QColor.fromHsv(value,255,b)
        self.brighness_slider.set_color(rgb_to_hex((c.red(),c.green(),c.blue()),True))
        self.color_dialog.blockSignals(True)
        self.color_dialog.setCurrentColor(c)
        self.color_dialog.blockSignals(False)
        self.dimmer.setHue(value)

    def set_brightness(self, value):
      
        if value > 0:
            c = self.color_dialog.currentColor()
            c.setHsv(c.hue(),c.saturation(),value)
            self.color_dialog.blockSignals(True)
            self.color_dialog.setCurrentColor(c)
            self.color_dialog.blockSignals(False)
        self.dimmer.setBrightness(value)

    # def set_color(self):
    #     # color_dialog = QColorDialog()
    #     self.color_dialog.show()
    #     # self.color_dialog.currentColorChanged.connect(self.colorSelected)
    #     # color = self.color_dialog.getColor()
    #     # color = QColorDialog.getColor()

    #     # if color.isValid():
    #     #     self.dimmer.setRGB(color.red(),color.green(),color.blue())

    def colorSelected(self,color):
        try:
            if color.isValid():
               
                self.dimmer.setRGB(color.red(),color.green(),color.blue())
                self.brighness_slider.blockSignals(True)
                self.hue_slider.blockSignals(True)

                self.brighness_slider.setValue(color.value())
                self.brighness_slider.set_color(rgb_to_hex((color.red(),color.green(),color.blue()),True))
                if color.hue() > -1 :
                    self.hue_slider.setValue(color.hue())

                self.brighness_slider.blockSignals(False)
                self.hue_slider.blockSignals(False)
        except:
            print("Error selecting color")





class DimmerSerialWidget(QWidget):

    def __init__(self, rgbDimmer = 0, serial = 0):
        super(DimmerSerialWidget,self).__init__()

        self.dimmerWidget = RGBDimmerWidget(rgbDimmer)
        self.dimmer = self.dimmerWidget.dimmer

        self.serial_monitor = SerialMonitor(True,serial)
        self.serial = self.serial_monitor.serial


        self.dimmerWidget.dimmer.senders.add(self.serial.sendStr)

        h_lo = QHBoxLayout(self)

        h_lo.addWidget(self.dimmerWidget)
        h_lo.addWidget(self.serial_monitor)
        self.setStyleSheet(DIMMER_WIDGET_STYLE)

    def __del__(self):
        self.dimmer.senders.remove(self.serial.sendStr)


if __name__ == '__main__':
    app = QApplication(sys.argv)
    w = DimmerSerialWidget()
    w.show()    
    sys.exit(app.exec_())