Added PathButton -- simple button displaying a QPainterPath
This commit is contained in:
parent
c79e794e53
commit
29be68d7fe
50
widgets/PathButton.py
Normal file
50
widgets/PathButton.py
Normal file
@ -0,0 +1,50 @@
|
||||
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()
|
||||
|
||||
|
||||
|
Loading…
Reference in New Issue
Block a user