PyQt5自定义组件之飞机水平仪
PyQt5自定义组件之飞机水平仪
效果如下图:
from PyQt5.QtCore import *
from PyQt5.QtGui import *
from PyQt5.QtWidgets import *
class GaugePlane(QWidget):
def __init__(self, parent=None):
super(GaugePlane, self).__init__(parent)
self.initUI()
def initUI(self):
self.borderOutColorStart = Qt.black
self.borderOutColorEnd = QColor(23, 149, 166)
self.borderInColorStart = QColor(50, 50, 50)
self.borderInColorEnd = QColor(220, 220, 220)
self.bgColor = QColor(41, 127, 184)
self.planeColor = QColor("#d68070")
self.glassColor = QColor(250, 250, 250)
self.scaleColor = QColor(255, 255, 255)
self.lineColor = QColor(255, 255, 255)
self.textColor = QColor(255, 255, 255)
self.pointerColor = QColor(255, 107, 107)
self.handleColor = QColor("grey")
self.degValue = 0 # 倾斜角
self.rollValue = 0 # 俯仰角
self.resize(400, 400)
self.time = QTimer()
self.time.start(1000)
self.time.timeout.connect(self.tsetDegValue)
def paintEvent(self, QPaintEvent):
width = self.width()
height = self.height()
side = min(width, height)
# 绘制准备工作, 启用反锯齿, 平移坐标轴中心, 等比例缩放
painter = QPainter(self)
painter.setRenderHints(QPainter.Antialiasing | QPainter.TextAntialiasing)
painter.translate(width / 2, height / 2)
painter.scale(side / 200.0, side / 200.0)
# 绘制外边框
self.drawBorderOut(painter)
# 绘制内边框
self.drawBorderIn(painter)
# 绘制背景
self.drawBg(painter)
# 绘制姿势仪背景
self.drawPlane(painter)
# 绘制玻璃遮罩层
self.drawGlass(painter)
# 绘制刻度尺
self.drawScale(painter)
# 绘制线条
self.drawLine(painter)
# 绘制指针
self.drawPointer(painter)
# 绘制手柄
self.drawHandle(painter)
def tsetDegValue(self):
import random
dev = random.randint(-30, 30)
roll = random.randint(-30, 30)
self.setRollValue(roll)
self.setDegValue(dev)
def drawBorderOut(self, painter):
radius = 99
painter.save()
painter.setPen(Qt.NoPen)
borderGradient = QLinearGradient(0, -radius, 0, radius)
borderGradient.setColorAt(0, self.borderOutColorStart)
borderGradient.setColorAt(1, self.borderOutColorEnd)
painter.setBrush(borderGradient)
painter.drawEllipse(-radius, -radius, radius * 2, radius * 2)
painter.restore()
def drawBorderIn(self, painter):
radius = 93
painter.save()
painter.setPen(Qt.NoPen)
borderGradient = QLinearGradient(0, -radius, 0, radius)
borderGradient.setColorAt(0, self.borderInColorStart)
borderGradient.setColorAt(1, self.borderInColorEnd)
painter.setBrush(borderGradient)
painter.drawEllipse(-radius, -radius, radius * 2, radius * 2)
painter.restore()
def drawBg(self, painter):
radius = 90
painter.save()
painter.setPen(Qt.NoPen)
borderGradient = QLinearGradient(0, -radius, 0, radius)
self.bgColor.setAlpha(255)
borderGradient.setColorAt(0, self.bgColor)
self.bgColor.setAlpha(150)
borderGradient.setColorAt(1, self.bgColor)
painter.setBrush(borderGradient)
painter.drawEllipse(-radius, -radius, radius * 2, radius * 2)
painter.restore()
def drawPlane(self, painter):
radius = 90
painter.save()
painter.rotate(self.degValue)
painter.setPen(Qt.NoPen)
painter.setBrush(self.planeColor)
rect = QRect(-radius, -radius, radius * 2, radius * 2)
if not painter:
painter.drawPie(rect, 0, -16 * 180)
else:
offset = -(self.rollValue * radius / 60)
startAngle = 180 + offset
endAngle = offset
span = endAngle + startAngle
painter.drawChord(rect, -16 * startAngle, 16 * span)
painter.restore()
def drawGlass(self, painter):
radius = 80
painter.save()
painter.setPen(Qt.NoPen)
# 饼圆1为整个圆形区域
pieRect1 = QRect(-radius, -radius, radius * 2, radius * 2)
# 饼圆2位饼圆1区域的中间部分的四分之一
pieRect2 = QRect(-radius, -radius / 4, radius * 2, radius / 4 * 2)
# 颜色线性渐变, 颜色透明度产生立体感
linearGradient = QLinearGradient(pieRect1.topLeft(), pieRect1.bottomRight())
self.glassColor.setAlpha(30)
linearGradient.setColorAt(0.1, self.glassColor)
self.glassColor.setAlpha(100)
linearGradient.setColorAt(0.5, self.glassColor)
# 绘制两个饼圆
painter.setBrush(linearGradient)
painter.drawPie(pieRect1, 0, 16 * 180)
painter.drawPie(pieRect2, 0, -16 * 180)
painter.restore()
def drawScale(self, painter):
radius = 88
painter.save()
# 设置画笔颜色
pen = QPen()
pen.setColor(self.scaleColor)
pen.setWidthF(4)
pen.setCapStyle(Qt.RoundCap)
painter.setPen(pen)
# 线条长度
len = 8
# 左侧平行线
painter.rotate(90)
painter.drawLine(0, radius - len, 0, radius)
# 逐个绘制大刻度
for i in range(6):
painter.rotate(30)
painter.drawLine(0, radius - len, 0, radius)
# 回到小刻度,重新设置画笔宽度
pen.setWidthF(2)
painter.setPen(pen)
painter.rotate(-120)
# 逐个绘制小刻度
for i in range(5):
painter.rotate(10)
painter.drawLine(0, radius - len, 0, radius)
painter.restore()
def drawLine(self, painter):
radius = 95
painter.save()
width = self.width()
height = self.height()
side = min(width, height)
# 重置坐标系
painter.resetTransform()
painter.translate(width / 2, height / 2 + self.rollValue)
painter.scale(side / 200.0, side / 200.0)
painter.rotate(self.degValue)
# 设置画笔颜色
pen = QPen()
pen.setColor(self.lineColor)
pen.setWidthF(2)
pen.setCapStyle(Qt.RoundCap)
painter.setPen(pen)
# 设置画笔字体
font = QFont()
font.setPixelSize(10)
font.setBold(True)
painter.setFont(font)
# 依次绘制水平距离线条及值
i = -30
while i <= 30:
if i == 0:
# 中心点不用绘制
i += 10
continue
value = qAbs(i)
# 绘制横向水平线条
pt1 = QPointF(-radius * value / 100, radius / 70 * i)
pt2 = QPointF(-pt1.x(), pt1.y())
painter.drawLine(pt1, pt2)
# 根据字体大小计算绘制文字的区域
# strValue = QString.number(value)
strValue = str(value).replace(".0", "")
fm = painter.fontMetrics()
textRect = QRectF(QPointF(QPoint(0, 0)), QSizeF(fm.size(Qt.AlignCenter, strValue)))
textRect.moveCenter(pt1 - QPointF(radius / 10, 0))
painter.drawText(textRect, Qt.AlignCenter, strValue)
textRect.moveCenter(pt2 + QPointF(radius / 10, 0))
painter.drawText(textRect, Qt.AlignCenter, strValue)
i += 10
painter.restore()
def drawPointer(self, painter):
radius = 70
painter.save()
painter.rotate(self.degValue + 90)
painter.setPen(Qt.NoPen)
# 指针长度
len = 10
# 三角形坐标
pts = QPolygon()
pts.setPoints(-radius, 0, -radius + len, -len / 2, -radius + len, len / 2)
painter.setBrush(self.pointerColor)
painter.drawConvexPolygon(pts)
painter.restore()
def drawHandle(self, painter):
painter.save()
pen = QPen()
pen.setColor(self.handleColor)
pen.setWidthF(4)
pen.setCapStyle(Qt.RoundCap)
pen.setJoinStyle(Qt.MiterJoin)
painter.setPen(pen)
painter.setBrush(self.handleColor)
# 绘制弧形
arcRadius = 15
rect = QRect(-arcRadius, -arcRadius, arcRadius * 2, arcRadius * 2)
painter.drawArc(rect, 0, -16 * 180)
# 绘制中心圆
centerRadius = 2
rect = QRect(-centerRadius, -centerRadius, centerRadius * 2, centerRadius * 2)
painter.drawEllipse(rect)
# 绘制左侧右侧横向水平线条
len = 30
pt1 = QPoint(-len, 0)
pt2 = QPoint(-arcRadius, 0)
painter.drawLine(pt1, pt2)
pt1 = QPoint(len, 0)
pt2 = QPoint(arcRadius, 0)
painter.drawLine(pt1, pt2)
# 绘制纵向垂直线条
pt1 = QPoint(0, len / 2)
pt2 = QPoint(0, 80)
painter.drawLine(pt1, pt2)
# 绘制底部梯形
p1 = QPoint(-30, 82)
p2 = QPoint(-20, 60)
p3 = QPoint(20, 60)
p4 = QPoint(30, 82)
pts = QPolygon([p1, p2, p3, p4])
# pts << p1 << p2 << p3 << p4
# pts.setPoints()
painter.drawConvexPolygon(pts)
# 绘制底部填充圆弧
radius = 32
rect = QRect(-radius, 77, radius * 2, 13)
painter.setPen(Qt.NoPen)
painter.drawPie(rect, 0, -16 * 180)
painter.restore()
def getBorderOutColorStart(self):
return self.borderOutColorStart
def getBorderOutColorEnd(self):
return self.borderOutColorEnd
def getBorderInColorStart(self):
return self.borderInColorStart
def getBorderInColorEnd(self):
return self.borderInColorEnd
def getBgColor(self):
return self.bgColor
def getPlaneColor(self):
return self.planeColor
def getGlassColor(self):
return self.glassColor
def getScaleColor(self):
return self.scaleColor
def getLineColor(self):
return self.lineColor
def getTextColor(self):
return self.textColor
def getPointerColor(self):
return self.pointerColor
def getHandleColor(self):
return self.handleColor
def getDegValue(self):
return self.degValue
def getRollValue(self):
return self.rollValue
def sizeHint(self):
return QSize(200, 200)
def minimumSizeHint(self):
return QSize(50, 50)
def setBorderOutColorStart(self, borderOutColorStart):
if (self.borderOutColorStart != borderOutColorStart):
self.borderOutColorStart = borderOutColorStart
self.update()
def setBorderOutColorEnd(self, borderOutColorEnd):
if (self.borderOutColorEnd != borderOutColorEnd):
self.borderOutColorEnd = borderOutColorEnd
self.update()
def setBorderInColorStart(self, borderInColorStart):
if (self.borderInColorStart != borderInColorStart):
self.borderInColorStart = borderInColorStart
self.update()
def setBorderInColorEnd(self, borderInColorEnd):
if (self.borderInColorEnd != borderInColorEnd):
self.borderInColorEnd = borderInColorEnd
self.update()
def setBgColor(self, bgColor):
if (self.bgColor != bgColor):
self.bgColor = bgColor
self.update()
def setPlaneColor(self, planeColor):
if (self.planeColor != planeColor):
self.planeColor = planeColor
self.update()
def setGlassColor(self, glassColor):
if (self.glassColor != glassColor):
self.glassColor = glassColor
self.update()
def setScaleColor(self, scaleColor):
if (self.scaleColor != scaleColor):
self.scaleColor = scaleColor
self.update()
def setLineColor(self, lineColor):
if (self.lineColor != lineColor):
self.lineColor = lineColor
self.update()
def setTextColor(self, textColor):
if (self.textColor != textColor):
self.textColor = textColor
self.update()
def setPointerColor(self, pointerColor):
if (self.pointerColor != pointerColor):
self.pointerColor = pointerColor
self.update()
def setHandleColor(self, handleColor):
if (self.handleColor != handleColor):
self.handleColor = handleColor
self.update()
def setDegValue(self, degValue):
if (self.degValue != degValue and degValue >= -180 and degValue <= 180):
self.degValue = degValue
self.update()
def setRollValue(self, rollValue):
if (self.rollValue != rollValue and rollValue >= -100 and rollValue <= 100):
self.rollValue = rollValue
self.update()
import sys
if __name__ == '__main__':
app = QApplication(sys.argv)
mainWindow = GaugePlane()
mainWindow.show()
sys.exit(app.exec_())
