PyQt5教程(十)——自定義控件

PyQt5包含種類豐富的控件。但能滿足所有需求的控件庫是不存在的。通常控件庫只提供了像按鈕、文本控件、滑塊等最常用的控件。但如果需要某種特殊的控件，我們只能自己動手來實現。 自定義控件需要使用工具庫提供的繪圖工具，可能有兩種方式：在已有的控件上進行拓展或從頭開始創建自定義控件。

Burning widget(燒錄控件)

這個控件可能會在Nero，K3B或其他CD/DVD燒錄軟件中見到。

import sys
from PyQt5.QtWidgets import (QWidget, QSlider, QApplication, QHBoxLayout, QVBoxLayout)
from PyQt5.QtCore import QObject, Qt, pyqtSignal
from PyQt5.QtGui import QPainter, QFont, QColor, QPen


class Communicate(QObject):
    updateBW = pyqtSignal(int)


class BurningWidget(QWidget):
    def __init__(self):
        super().__init__()
        self.initUI()

    def initUI(self):
        self.setMinimumSize(1, 30)
        self.value = 75
        self.num = [75, 150, 225, 300, 375, 450, 525, 600, 675]

    def setValue(self, value):
        self.value = value

    def paintEvent(self, e):
        qp = QPainter()
        qp.begin(self)
        self.drawWidget(qp)
        qp.end()

    def drawWidget(self, qp):
        font = QFont("Serif", 7, QFont.Light)
        qp.setFont(font)

        size = self.size()
        w = size.width()
        h = size.height()

        step = int(round(w / 10.0))

        till = int(((w / 750.0) * self.value))
        full = int(((w / 750.0) * 700))

        if self.value >=700:
            qp.setPen(QColor(255, 255, 255))
            qp.setBrush(QColor(255, 255, 184))
            qp.drawRect(0, 0, full, h)
            qp.setPen(QColor(255, 175, 175))
            qp.setBrush(QColor(255, 175, 175))
            qp.drawRect(full, 0, till-full, h)
        else:
            qp.setPen(QColor(255, 255, 255))
            qp.setBrush(QColor(255, 255, 184))
            qp.drawRect(0, 0, till, h)

        pen = QPen(QColor(20, 20, 20), 1, Qt.SolidLine)

        qp.setPen(pen)
        qp.setBrush(Qt.NoBrush)
        qp.drawRect(0, 0, w-1, h-1)

        j = 0

        for i in range(step, 10*step, step):
            qp.drawLine(i, 0, i, 5)
            metrics = qp.fontMetrics()
            fw = metrics.width(str(self.num[j]))
            qp.drawText(i-fw/2, h/2, str(self.num[j]))
            j = j + 1


class Example(QWidget):
    def __init__(self):
        super().__init__()
        self.initUI()

    def initUI(self):
        sld = QSlider(Qt.Horizontal, self)
        sld.setFocusPolicy(Qt.NoFocus)
        sld.setRange(1, 750)
        sld.setValue(75)
        sld.setGeometry(30, 40, 150, 30)

        self.c = Communicate()
        self.wid = BurningWidget()
        self.c.updateBW[int].connect(self.wid.setValue)

        sld.valueChanged[int].connect(self.changeValue)
        hbox = QHBoxLayout()
        hbox.addWidget(self.wid)
        vbox = QVBoxLayout()
        vbox.addStretch(1)
        vbox.addLayout(hbox)
        self.setLayout(vbox)

        self.setGeometry(300, 300, 390, 210)
        self.setWindowTitle("Burning widget")
        self.show()

    def changeValue(self, value):
        self.c.updateBW.emit(value)
        self.wid.repaint()


if __name__ == "__main__":
    app = QApplication(sys.argv)
    ex = Example()
    sys.exit(app.exec_())

在示例中我們使用了滑塊與一個自定義控件。自定義控件受滑塊控制。控件顯示了媒體介質的容量和剩餘空間。該控件的最小值爲1,最大值爲750。在值超過700時顏色變爲紅色。這通常意味着超刻(即實際寫入光盤的容量超過刻錄盤片官方標稱容量的一種操作)。

BurningWidget控件通過QHBoxLayout與QVBoxLayout置於窗體的底部。

class BurningWidget(QWidget):

    def __init__(self):      
        super().__init__()

BurningWidget控件拓展自QWidget控件。

self.setMinimumSize(1, 30)

我們修改了它的最小尺寸(高)。默認的最小尺寸對我們來說有點小。

font = QFont('Serif', 7, QFont.Light)
qp.setFont(font)

我們使用了一種更小的字體，它比較適於我們的需求。

size = self.size()
w = size.width()
h = size.height()

step = int(round(w / 10.0))


till = int(((w / 750.0) * self.value))
full = int(((w / 750.0) * 700))

控件採用了動態繪製技術。窗體越大，控件也隨之變大；反之亦然。這也是我們需要計算自定義控件的載體控件(即窗體)尺寸的原因。till參數定義了需要繪製的總尺寸，它根據slider控件計算得出，是整體區域的比例值。full參數定義了紅色區域的繪製起點。注意在繪製時爲取得較大精度而使用的浮點數運算。

實際的繪製分三個步驟。黃色或紅黃矩形的繪製，然後是刻度線的繪製，最後是刻度值的繪製。

metrics = qp.fontMetrics()
fw = metrics.width(str(self.num[j]))
qp.drawText(i-fw/2, h/2, str(self.num[j]))

我們採用fondMetris繪製刻度值。爲了使刻度值居中於刻度線，我們需要計算刻度值文本的寬度。

def changeValue(self, value):

    self.c.updateBW.emit(value)        
    self.wid.repaint()

當滑塊發生移動時，changeValue()方法會被調用。在方法內我們觸發了一個自定義的updateBW信號，其參數是當前滾動條的值。該值被用於計算Burning widget的容量值。然後對控件進行重繪。

---

在本教程中我們創建了一個自定義控件。