【PyQt4 实例5】利用QPainter绘制各种图形

#-*- coding:utf-8 -*-
import sys

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

QTextCodec.setCodecForTr(QTextCodec.codecForName("utf-8"))

class StockDialog(QWidget):
    def __init__(self, parent=None):
        super(StockDialog, self).__init__(parent)
        self.setWindowTitle(self.tr("利用QPainter绘制各种图形"))
        
        mainSplitter = QSplitter(Qt.Horizontal)
        mainSplitter.setOpaqueResize(True)
                  
        frame = QFrame(mainSplitter)
        mainLayout = QGridLayout(frame)
        mainLayout.setMargin(10)
        mainLayout.setSpacing(6)

        label1=QLabel(self.tr("形状："))        
        label2=QLabel(self.tr("画笔线宽："))
        label3=QLabel(self.tr("画笔颜色："))
        label4=QLabel(self.tr("画笔风格："))
        label5=QLabel(self.tr("画笔顶端："))
        label6=QLabel(self.tr("画笔连接点："))
        label7=QLabel(self.tr("画刷风格："))
        label8=QLabel(self.tr("画刷颜色："))
    
        self.shapeComboBox = QComboBox()
        self.shapeComboBox.addItem(self.tr("Line"),"Line")
        self.shapeComboBox.addItem(self.tr("Rectangle"),"Rectangle")
        self.widthSpinBox = QSpinBox()
        self.widthSpinBox.setRange(0,20)
        self.penColorFrame = QFrame()
        self.penColorFrame.setAutoFillBackground(True)
        self.penColorFrame.setPalette(QPalette(Qt.blue))
        self.penColorPushButton = QPushButton(self.tr("更改"))
        self.penStyleComboBox = QComboBox()
        self.penStyleComboBox.addItem(self.tr("Solid"),Qt.SolidLine)
        self.penCapComboBox = QComboBox()
        self.penCapComboBox.addItem(self.tr("Flat"),Qt.FlatCap)  
        self.penJoinComboBox = QComboBox()
        self.penJoinComboBox.addItem(self.tr("Miter"),Qt.MiterJoin)
        self.brushStyleComboBox = QComboBox()
        self.brushStyleComboBox.addItem(self.tr("Linear Gradient"),Qt.LinearGradientPattern)
        self.brushColorFrame = QFrame()
        self.brushColorFrame.setAutoFillBackground(True)
        self.brushColorFrame.setPalette(QPalette(Qt.green))
        self.brushColorPushButton = QPushButton(self.tr("更改"))
                                                              
        labelCol=0
        contentCol=1
        
        #建立布局
        mainLayout.addWidget(label1,1,labelCol)
        mainLayout.addWidget(self.shapeComboBox,1,contentCol)
        mainLayout.addWidget(label2,2,labelCol)
        mainLayout.addWidget(self.widthSpinBox,2,contentCol)
        mainLayout.addWidget(label3,4,labelCol)
        mainLayout.addWidget(self.penColorFrame,4,contentCol)
        mainLayout.addWidget(self.penColorPushButton,4,3)
        mainLayout.addWidget(label4,6,labelCol)
        mainLayout.addWidget(self.penStyleComboBox,6,contentCol)
        mainLayout.addWidget(label5,8,labelCol)
        mainLayout.addWidget(self.penCapComboBox,8,contentCol)
        mainLayout.addWidget(label6,10,labelCol)
        mainLayout.addWidget(self.penJoinComboBox,10,contentCol)
        mainLayout.addWidget(label7,12,labelCol)
        mainLayout.addWidget(self.brushStyleComboBox,12,contentCol)
        mainLayout.addWidget(label8,14,labelCol)
        mainLayout.addWidget(self.brushColorFrame,14,contentCol)
        mainLayout.addWidget(self.brushColorPushButton,14,3)
        mainSplitter1 = QSplitter(Qt.Horizontal)
        mainSplitter1.setOpaqueResize(True)
        
        stack1 = QStackedWidget()
        stack1.setFrameStyle(QFrame.Panel|QFrame.Raised)
        self.area = PaintArea()
        stack1.addWidget(self.area)        
        frame1 = QFrame(mainSplitter1)
        mainLayout1 = QVBoxLayout(frame1)
        mainLayout1.setMargin(10)
        mainLayout1.setSpacing(6)
        mainLayout1.addWidget(stack1)

        layout = QGridLayout(self)
        layout.addWidget(mainSplitter1,0,0)
        layout.addWidget(mainSplitter,0,1)
        self.setLayout(layout)
        
        #信号和槽函数
        self.connect(self.shapeComboBox,SIGNAL("activated(int)"),self.slotShape)
        self.connect(self.widthSpinBox,SIGNAL("valueChanged(int)"),self.slotPenWidth)
        self.connect(self.penColorPushButton,SIGNAL("clicked()"),self.slotPenColor)
        self.connect(self.penStyleComboBox,SIGNAL("activated(int)"),self.slotPenStyle)
        self.connect(self.penCapComboBox,SIGNAL("activated(int)"),self.slotPenCap)
        self.connect(self.penJoinComboBox,SIGNAL("activated(int)"),self.slotPenJoin)
        self.connect(self.brushStyleComboBox,SIGNAL("activated(int)"),self.slotBrush)
        self.connect(self.brushColorPushButton,SIGNAL("clicked()"),self.slotBrushColor)
        
        self.slotShape(self.shapeComboBox.currentIndex())
        self.slotPenWidth(self.widthSpinBox.value())
        self.slotBrush(self.brushStyleComboBox.currentIndex())        
        
    def slotShape(self,value):
        shape =  self.area.Shape[value]
        self.area.setShape(shape)
    
    def slotPenWidth(self,value):
        color = self.penColorFrame.palette().color(QPalette.Window)
        style = Qt.PenStyle((self.penStyleComboBox.itemData(self.penStyleComboBox.currentIndex(),Qt.UserRole).toInt())[0])
        cap = Qt.PenCapStyle((self.penCapComboBox.itemData(self.penCapComboBox.currentIndex(),Qt.UserRole).toInt())[0])
        join = Qt.PenJoinStyle((self.penJoinComboBox.itemData(self.penJoinComboBox.currentIndex(),Qt.UserRole).toInt())[0])
        self.area.setPen(QPen(color,value,style,cap,join))
    
    def slotPenStyle(self,value):
        self.slotPenWidth(value)
    
    def slotPenCap(self,value):
        self.slotPenWidth(value)
    
    def slotPenJoin(self,value):
        self.slotPenWidth(value)
    
    def slotPenColor(self):
        color = QColorDialog.getColor(Qt.blue)
        self.penColorFrame.setPalette(QPalette(color))
        self.area.setPen(QPen(color))
        
    def slotBrushColor(self):
        color = QColorDialog.getColor(Qt.blue)
        self.brushColorFrame.setPalette(QPalette(color))
        self.slotBrush(self.brushStyleComboBox.currentIndex())
    
    def slotBrush(self,value):
        color = self.brushColorFrame.palette().color(QPalette.Window)
        style = Qt.BrushStyle((self.brushStyleComboBox.itemData(value,Qt.UserRole).toInt())[0])
        
        if(style == Qt.LinearGradientPattern):
            linearGradient = QLinearGradient(0,0,400,400)
            linearGradient.setColorAt(0.0,Qt.white)
            linearGradient.setColorAt(0.2,color)
            linearGradient.setColorAt(1.0,Qt.black)
            self.area.setBrush(linearGradient)
        elif(style == Qt.ConicalGradientPattern):
            conicalGradient = QConicalGradient(200,200,30)
            conicalGradient.setColorAt(0.0,Qt.white)
            conicalGradient.setColorAt(0.2,color)
            conicalGradient.setColorAt(1.0,Qt.black)
            self.area.setBrush(conicalGradient)
        elif(style == Qt.TexturePattern):
            self.area.setBrush(QBrush(QPixmap("image/cheese.jpg")))
        else:
            self.area.setBrush(QBrush(color,style))
        
    
class PaintArea(QWidget):
    def __init__(self):
        super(PaintArea,self).__init__()
        self.Shape = ["Line","Rectangle"]
        self.setPalette(QPalette(Qt.white))
        self.setAutoFillBackground(True)
        self.setMinimumSize(400,400)
        self.pen = QPen()
        self.brush = QBrush()        
    
    def setShape(self,s):
        self.shape = s
        self.update()
        
    def setPen(self,p):
        self.pen = p
        self.update()
    
    def setBrush(self,b):
        self.brush = b
        self.update()
    
    def paintEvent(self,QPaintEvent):
        p = QPainter(self)
        p.setPen(self.pen)
        p.setBrush(self.brush)
        
        rect = QRect(50,100,300,200) 
        points = [QPoint(150,100),QPoint(300,150),QPoint(350,250),QPoint(100,300)]
        startAngle = 30 * 16
        spanAngle = 120 * 16
        
        path = QPainterPath();
        path.addRect(150,150,100,100)
        path.moveTo(100,100)
        path.cubicTo(300,100,200,200,300,300)
        path.cubicTo(100,300,200,200,100,100)
        
        if self.shape == "Line":
            p.drawLine(rect.topLeft(),rect.bottomRight())
        elif self.shape == "Rectangle":
            p.drawRect(rect)
        elif self.shape == "RoundRect":
            p.drawRoundRect(rect)
        elif self.shape == "Ellipse":
            p.drawEllipse(rect)
        elif self.shape == "Polygon":
            p.drawPolygon(points,4)
        elif self.shape == "Polyline":
            p.drawPolyline(points,4)
        elif self.shape == "Points":
            p.drawPoints(points,4)
        elif self.shape == "Arc":
            p.drawArc(rect,startAngle,spanAngle)
        elif self.shape == "Path":
            p.drawPath(path)
        elif self.shape == "Text":
            p.drawText(rect,Qt.AlignCenter,tr("Hello Qt!"))
        elif self.shape == "Pixmap":
            p.drawPixmap(150,150,QPixmap("image/butterfly.png"))
        
if __name__=='__main__':
    
    app = QApplication(sys.argv)
    form = StockDialog()
    form.show()
    app.exec_()