50 lines
1.5 KiB
Python
50 lines
1.5 KiB
Python
from pyqtgraph.Qt import QtGui, QtCore
|
|
import pyqtgraph as pg
|
|
__all__ = ['PathButton']
|
|
|
|
|
|
class PathButton(QtGui.QPushButton):
|
|
"""Simple PushButton extension which paints a QPainterPath on its face"""
|
|
def __init__(self, parent=None, path=None, pen='default', brush=None, size=(30,30)):
|
|
QtGui.QPushButton.__init__(self, parent)
|
|
self.path = None
|
|
if pen == 'default':
|
|
pen = 'k'
|
|
self.setPen(pen)
|
|
self.setBrush(brush)
|
|
if path is not None:
|
|
self.setPath(path)
|
|
if size is not None:
|
|
self.setFixedWidth(size[0])
|
|
self.setFixedHeight(size[1])
|
|
|
|
|
|
def setBrush(self, brush):
|
|
self.brush = pg.mkBrush(brush)
|
|
|
|
def setPen(self, pen):
|
|
self.pen = pg.mkPen(pen)
|
|
|
|
def setPath(self, path):
|
|
self.path = path
|
|
self.update()
|
|
|
|
def paintEvent(self, ev):
|
|
QtGui.QPushButton.paintEvent(self, ev)
|
|
margin = 7
|
|
geom = QtCore.QRectF(0, 0, self.width(), self.height()).adjusted(margin, margin, -margin, -margin)
|
|
rect = self.path.boundingRect()
|
|
scale = min(geom.width() / float(rect.width()), geom.height() / float(rect.height()))
|
|
|
|
p = QtGui.QPainter(self)
|
|
p.setRenderHint(p.Antialiasing)
|
|
p.translate(geom.center())
|
|
p.scale(scale, scale)
|
|
p.translate(-rect.center())
|
|
p.setPen(self.pen)
|
|
p.setBrush(self.brush)
|
|
p.drawPath(self.path)
|
|
p.end()
|
|
|
|
|
|
|