pyqt4 自定义圆形指示灯控件

# -*- coding: utf-8 -*-#

#-------------------------------------------------------------------------------
# Name:         自定义圆形指示灯控件
#-------------------------------------------------------------------------------

import sys
from PyQt4.QtGui import *
from PyQt4.QtCore import *

allAttributes =   [  'colorOnBegin', 'colorOnEnd', 'colorOffBegin', 'colorOffEnd', 'colorBorderIn', 'colorBorderOut',
                     'radiusBorderOut', 'radiusBorderIn', 'radiusCircle']
allDefaultVal =   [ QColor(0, 240, 0), QColor(0, 160, 0), QColor(0, 68, 0), QColor(0, 28, 0), QColor(140, 140, 140), QColor(100, 100, 100),
                    500, 450, 400]
allLabelNames =   [ u'灯亮圆心颜色：', u'灯亮边缘颜色：', u'灯灭圆心颜色：', u'灯灭边缘颜色：', u'边框内测颜色：', u'边框外侧颜色：',
                    u'边框外侧半径：', u'边框内侧半径：', u'中间圆灯半径：']


class MyLed(QAbstractButton):
    def __init__(self, parent=None):
        super(MyLed, self).__init__(parent)
        self.initUI()

    def initUI(self):
        self.setMinimumSize(24, 24)
        self.setCheckable(True)
        self.scaledSize = 1000.0    #为方便计算，将窗口短边值映射为1000
        self.setLedDefaultOption()

    def setLedDefaultOption(self):
        for attr, val in zip(allAttributes, allDefaultVal):
            setattr(self, attr, val)
        self.update()

    def setLedOption(self, opt='colorOnBegin', val=QColor(0,240,0)):
        if hasattr(self, opt):
            setattr(self, opt, val)
            self.update()

    def resizeEvent(self, evt):
        self.update()

    def paintEvent(self, evt):
        painter = QPainter(self)
        painter.setRenderHint(QPainter.Antialiasing, True)
        painter.setPen(QPen(Qt.black, 1))

        realSize = min(self.width(), self.height())                         #窗口的短边
        painter.translate(self.width()/2.0, self.height()/2.0)              #原点平移到窗口中心
        painter.scale(realSize/self.scaledSize, realSize/self.scaledSize)   #缩放，窗口的短边值映射为self.scaledSize
        gradient = QRadialGradient(QPointF(0, 0), self.scaledSize/2.0, QPointF(0, 0))   #辐射渐变

        #画边框外圈和内圈
        for color, radius in [(self.colorBorderOut, self.radiusBorderOut),  #边框外圈
                               (self.colorBorderIn, self.radiusBorderIn)]:   #边框内圈
            gradient.setColorAt(1, color)
            painter.setBrush(QBrush(gradient))
            painter.drawEllipse(QPointF(0, 0), radius, radius)

        # 画内圆
        gradient.setColorAt(0, self.colorOnBegin if self.isChecked() else self.colorOffBegin)
        gradient.setColorAt(1, self.colorOnEnd if self.isChecked() else self.colorOffEnd)
        painter.setBrush(QBrush(gradient))
        painter.drawEllipse(QPointF(0, 0), self.radiusCircle, self.radiusCircle)

class MyColorBox(QFrame):
    sigColorChanged = pyqtSignal(QColor)
    def __init__(self, parent=None, height=20, color=QColor(0,240,0)):
        super(MyColorBox, self).__init__(parent)
        self.setFixedHeight(height)
        self.setAutoFillBackground(True)
        self.setPalette(QPalette(color))
        self.setFrameStyle(QFrame.Panel | QFrame.Sunken)

    def mousePressEvent(self, *args, **kwargs):
        color = QColorDialog.getColor(initial=self.palette().color(QPalette.Window))
        if color.isValid():
            self.setPalette(QPalette(color))
            self.sigColorChanged.emit(color)

    def setColor(self, color):
        self.setPalette(QPalette(color))

class MyRadiusCtrl(QSpinBox):
    def __init__(self, parent=None, initVal=500):
        super(MyRadiusCtrl, self).__init__(parent)
        self.setRange(1, 500)
        self.setValue(initVal)

class ConfigWnd(QFrame):
    def __init__(self, parent=None):
        super(ConfigWnd, self).__init__(parent)
        self.initUI()

    def initUI(self):
        self.setFrameStyle(QFrame.Box|QFrame.Sunken)

        mainLayout = QVBoxLayout(self)
        mainLayout.addWidget(self.createColorParaGroupBox(), 0)
        mainLayout.addSpacing(20)
        mainLayout.addWidget(self.createRadiusParaGroupBox(), 0)
        mainLayout.addStretch()
        mainLayout.addSpacing(20)
        self.restoreDefaultBtn = QPushButton(u'恢复默认设置')
        mainLayout.addWidget(self.restoreDefaultBtn, 0)
        mainLayout.addSpacing(10)
        self.animateBtn = QPushButton(u'开始动画')
        self.animateBtn.setCheckable(True)
        mainLayout.addWidget(self.animateBtn, 0)

    def createColorParaGroupBox(self):
        colorParaGroupBox = QGroupBox(u"颜色参数设置", self)
        layout = QGridLayout(colorParaGroupBox)
        layout.setSpacing(10)

        self.allColorBoxCtrls = []
        for name, color, row in zip(allLabelNames[:6], allDefaultVal[:6], range(6)):
            layout.addWidget(QLabel(name), row, 0)
            colorBox = MyColorBox(color=color)
            layout.addWidget(colorBox, row, 1)
            self.allColorBoxCtrls.append(colorBox)

        layout.setColumnStretch(0, 0)
        layout.setColumnStretch(1, 1)
        return colorParaGroupBox

    def createRadiusParaGroupBox(self):
        radiusParaGroupBox = QGroupBox(u"半径设置（1～500）", self)
        layout = QGridLayout(radiusParaGroupBox)
        layout.setSpacing(10)

        self.allRadiusCtrls = []
        for name, radius, row in zip(allLabelNames[6:], allDefaultVal[6:], range(3)):
            layout.addWidget(QLabel(name), row, 0)
            radiusCtrl = MyRadiusCtrl(initVal=radius)
            layout.addWidget(radiusCtrl, row, 1)
            self.allRadiusCtrls.append(radiusCtrl)

        layout.setColumnStretch(0, 0)
        layout.setColumnStretch(1, 1)
        return radiusParaGroupBox

class MainWindow(QMainWindow):
    def __init__(self):
        super(MainWindow, self).__init__()
        self.initUI()
        self.initSlotFunc()
        self.cnt = 0
        self.show()

    def initUI(self):
        self.resize(580, 350)
        self.setWindowTitle(u'自定义圆形指示灯控件')

        mainSplitter = self.createSplitter(style=Qt.Horizontal, parent=self, width=4)

        self.configWnd = ConfigWnd(mainSplitter)

        rightSplitter = self.createSplitter(style=Qt.Vertical, parent=mainSplitter, width=4)

        rightTopWnd = self.createSubWnd(rightSplitter)
        rightTopLayout = QVBoxLayout(rightTopWnd)
        rightTopLayout.setContentsMargins(60, 60, 60, 60)
        self.ledSingle = MyLed()
        self.ledSingle.setSizePolicy(QSizePolicy.Expanding, QSizePolicy.Expanding)
        rightTopLayout.addWidget(self.ledSingle)

        rightBottomWnd = self.createSubWnd(rightSplitter)
        rightBottomLayout = QHBoxLayout(rightBottomWnd)
        rightBottomLayout.setContentsMargins(10, 10, 10, 10)
        self.ledGroup = []
        for i in range(8):
            led = MyLed()
            led.setSizePolicy(QSizePolicy.Expanding, QSizePolicy.Expanding)
            self.ledGroup.append(led)
            rightBottomLayout.addWidget(led)

        self.setSplitterStrechFactor(rightSplitter, 1, 0)
        self.setSplitterStrechFactor(mainSplitter, 0, 1)
        self.setCentralWidget(mainSplitter)

    def createSplitter(self, style=Qt.Horizontal, parent=None, width=3):
        splitter = QSplitter(style, parent)
        splitter.setHandleWidth(width)
        return splitter

    def setSplitterStrechFactor(self, splitter=None, factor1=1, factor2=1): #设置分割条两部分的比例
        splitter.setStretchFactor(0, factor1)
        splitter.setStretchFactor(1, factor2)

    def createSubWnd(self, parent=None):
        wnd = QFrame(parent)
        wnd.setFrameStyle(QFrame.Box | QFrame.Sunken)
        return wnd

    def initSlotFunc(self):
        self.configWnd.restoreDefaultBtn.clicked.connect(self.slotRestoreDefault)
        map(lambda x: x.sigColorChanged.connect(self.slotattributeChanged), self.configWnd.allColorBoxCtrls)   #设定每个颜色控件的槽函数
        map(lambda x: x.valueChanged.connect(self.slotattributeChanged), self.configWnd.allRadiusCtrls)        #设定每个半径控件的槽函数
        self.configWnd.animateBtn.clicked.connect(self.slotAnimation)
        self.timer = QTimer()
        self.timer.timeout.connect(self.slotTimeout) #动画定时器

    def slotattributeChanged(self, val):
        allCtrls = self.configWnd.allColorBoxCtrls + self.configWnd.allRadiusCtrls
        idx = allCtrls.index(self.sender())
        self.ledSingle.setLedOption(allAttributes[idx], val)

    def slotRestoreDefault(self):
        for colorBox, val in zip(self.configWnd.allColorBoxCtrls, allDefaultVal[:6]):
            colorBox.setColor(val)

        for radiusCtrl, val in zip(self.configWnd.allRadiusCtrls, allDefaultVal[6:]):
            radiusCtrl.setValue(val)

        self.ledSingle.setLedDefaultOption()

    def slotAnimation(self):
        if self.configWnd.animateBtn.isChecked():
            self.cnt = 0
            self.configWnd.animateBtn.setText(u'停止动画')
            self.timer.start(300)
        else:
            self.configWnd.animateBtn.setText(u'开始动画')
            self.timer.stop()

    def slotTimeout(self):
        self.cnt = self.cnt % 256
        ledBits = QString('%1').arg(self.cnt, 8, 2, fillChar=QChar('0'))    #将数值转换为二进制字符串
        for ledBit, led in zip(ledBits, self.ledGroup):
            led.setChecked(ledBit=='1')
        self.cnt += 1

def main():
    app = QApplication(sys.argv)
    mainWnd = MainWindow()
    sys.exit(app.exec_())

if __name__ == '__main__':
    main()