return self._action.isEnabled()
-class QPieMenu(QtGui.QWidget):
+class PieFiling(object):
INNER_RADIUS_DEFAULT = 24
OUTER_RADIUS_DEFAULT = 64
- ICON_SIZE_DEFAULT = 32
-
- activated = QtCore.pyqtSignal(int)
- highlighted = QtCore.pyqtSignal(int)
- canceled = QtCore.pyqtSignal()
- aboutToShow = QtCore.pyqtSignal()
- aboutToHide = QtCore.pyqtSignal()
SELECTION_CENTER = -1
SELECTION_NONE = -2
NULL_CENTER = QtGui.QAction(None)
- def __init__(self, parent = None):
- QtGui.QWidget.__init__(self, parent)
+ def __init__(self, centerPos):
self._innerRadius = self.INNER_RADIUS_DEFAULT
self._outerRadius = self.OUTER_RADIUS_DEFAULT
self._children = []
self._center = self.NULL_CENTER
- self._selectionIndex = self.SELECTION_NONE
- self._mouseButtonPressed = False
- self._mousePosition = ()
-
- canvasSize = self._outerRadius * 2 + 1
- self._canvas = QtGui.QPixmap(canvasSize, canvasSize)
- self._mask = None
-
- def popup(self, pos):
- index = self.indexAt(self.mapFromGlobal(QtGui.QCursor.pos()))
- self._mousePosition = pos
- self.show()
+ self._centerPos = centerPos
def insertItem(self, item, index = -1):
self._children.insert(index, item)
- self.update()
def removeItemAt(self, index):
item = self._children.pop(index)
- self.update()
def set_center(self, item):
+ if item is None:
+ item = self.NULL_CENTER
self._center = item
+ def center(self):
+ return self._center
+
def clear(self):
del self._children[:]
- self.update()
def itemAt(self, index):
return self._children[index]
def setInnerRadius(self, radius):
self._innerRadius = radius
+ def setCenterPosition(self, centerPos):
+ self._centerPos = centerPos
+
def outerRadius(self):
return self._outerRadius
def setOuterRadius(self, radius):
self._outerRadius = radius
- self._canvas = self._canvas.scaled(self.sizeHint())
- def sizeHint(self):
- diameter = self._outerRadius * 2 + 1
- return QtCore.QSize(diameter, diameter)
+ def __iter__(self):
+ return iter(self._children)
- def mousePressEvent(self, mouseEvent):
- lastSelection = self._selectionIndex
+ def __len__(self):
+ return len(self._children)
- lastMousePos = mouseEvent.pos()
- self._update_selection(lastMousePos)
- self._mouseButtonPressed = True
- self._mousePosition = lastMousePos
+ def __getitem__(self, index):
+ return self._children[index]
- if lastSelection != self._selectionIndex:
- self.highlighted.emit(self._selectionIndex)
- self.update()
+ def _index_to_angle(self, index, isShifted):
+ index = index % len(self._children)
- def mouseMoveEvent(self, mouseEvent):
- lastSelection = self._selectionIndex
+ totalWeight = sum(child.weight() for child in self._children)
+ if totalWeight == 0:
+ totalWeight = 1
+ baseAngle = (2 * math.pi) / totalWeight
- lastMousePos = mouseEvent.pos()
- self._update_selection(lastMousePos)
+ angle = math.pi / 2
+ if isShifted:
+ if self._children:
+ angle -= (self._children[0].weight() * baseAngle) / 2
+ else:
+ angle -= baseAngle / 2
+ while angle < 0:
+ angle += 2*math.pi
- if lastSelection != self._selectionIndex:
- self.highlighted.emit(self._selectionIndex)
- self.update()
+ for i, child in enumerate(self._children):
+ if index < i:
+ break
+ angle += child.weight() * baseAngle
+ while (2*math.pi) < angle:
+ angle -= 2*math.pi
- def mouseReleaseEvent(self, mouseEvent):
- lastSelection = self._selectionIndex
+ return angle
- lastMousePos = mouseEvent.pos()
- self._update_selection(lastMousePos)
- self._mouseButtonPressed = False
- self._mousePosition = ()
+ def _angle_to_index(self, angle):
+ numChildren = len(self._children)
+ if numChildren == 0:
+ return self.SELECTION_CENTER
- self._activate_at(self._selectionIndex)
- self.update()
+ totalWeight = sum(child.weight() for child in self._children)
+ if totalWeight == 0:
+ totalWeight = 1
+ baseAngle = (2 * math.pi) / totalWeight
- def showEvent(self, showEvent):
- self.aboutToShow.emit()
+ iterAngle = math.pi / 2 - (self.itemAt(0).weight() * baseAngle) / 2
+ while iterAngle < 0:
+ iterAngle += 2 * math.pi
+
+ oldIterAngle = iterAngle
+ for index, child in enumerate(self._children):
+ iterAngle += child.weight() * baseAngle
+ if oldIterAngle < angle and angle <= iterAngle:
+ return index - 1 if index != 0 else numChildren - 1
+ elif oldIterAngle < (angle + 2*math.pi) and (angle + 2*math.pi <= iterAngle):
+ return index - 1 if index != 0 else numChildren - 1
+ oldIterAngle = iterAngle
+
+ def _radius_at(self, pos):
+ xDelta = pos.x() - self._centerPos.x()
+ yDelta = pos.y() - self._centerPos.y()
+
+ radius = math.sqrt(xDelta ** 2 + yDelta ** 2)
+ return radius
+
+ def _angle_at(self, pos):
+ xDelta = pos.x() - self._centerPos.x()
+ yDelta = pos.y() - self._centerPos.y()
+
+ radius = math.sqrt(xDelta ** 2 + yDelta ** 2)
+ angle = math.acos(xDelta / radius)
+ if 0 <= yDelta:
+ angle = 2*math.pi - angle
+
+ return angle
+
+
+class PieArtist(object):
+
+ ICON_SIZE_DEFAULT = 32
+
+ def __init__(self, filing):
+ self._filing = filing
+
+ self._cachedOuterRadius = self._filing.outerRadius()
+ self._cachedInnerRadius = self._filing.innerRadius()
+ canvasSize = self._cachedOuterRadius * 2 + 1
+ self._canvas = QtGui.QPixmap(canvasSize, canvasSize)
+ self._mask = None
+ self.palette = None
+
+ def sizeHint(self):
+ diameter = self._cachedOuterRadius * 2 + 1
+ return QtCore.QSize(diameter, diameter)
+
+ def show(self, palette):
+ self.palette = palette
+
+ if (
+ self._cachedOuterRadius != self._filing.outerRadius() or
+ self._cachedInnerRadius != self._filing.innerRadius()
+ ):
+ self._cachedOuterRadius = self._filing.outerRadius()
+ self._cachedInnerRadius = self._filing.innerRadius()
+ self._canvas = self._canvas.scaled(self.sizeHint())
if self._mask is None:
self._mask = QtGui.QBitmap(self._canvas.size())
self._mask.fill(QtCore.Qt.color0)
self._generate_mask(self._mask)
self._canvas.setMask(self._mask)
- self.setMask(self._mask)
+ return self._mask
- lastMousePos = self.mapFromGlobal(QtGui.QCursor.pos())
- self._update_selection(lastMousePos)
-
- QtGui.QWidget.showEvent(self, showEvent)
+ def hide(self):
+ self.palette = None
- def hideEvent(self, hideEvent):
- self.canceled.emit()
- self._selectionIndex = self.SELECTION_NONE
- QtGui.QWidget.hideEvent(self, hideEvent)
-
- def paintEvent(self, paintEvent):
+ def paint(self, selectionIndex):
painter = QtGui.QPainter(self._canvas)
painter.setRenderHint(QtGui.QPainter.Antialiasing, True)
adjustmentRect = self._canvas.rect().adjusted(0, 0, -1, -1)
- numChildren = len(self._children)
+ numChildren = len(self._filing)
if numChildren < 2:
- if self._selectionIndex == 0 and self._children[0].isEnabled():
- painter.setBrush(self.palette().highlight())
+ if selectionIndex == 0 and self._filing[0].isEnabled():
+ painter.setBrush(self.palette.highlight())
else:
- painter.setBrush(self.palette().background())
+ painter.setBrush(self.palette.background())
painter.fillRect(self.rect(), painter.brush())
else:
- for i in xrange(len(self._children)):
- self._paint_slice_background(painter, adjustmentRect, i)
+ for i in xrange(len(self._filing)):
+ self._paint_slice_background(painter, adjustmentRect, i, selectionIndex)
- self._paint_center_background(painter, adjustmentRect)
- self._paint_center_foreground(painter)
+ self._paint_center_background(painter, adjustmentRect, selectionIndex)
+ self._paint_center_foreground(painter, selectionIndex)
- for i in xrange(len(self._children)):
- self._paint_slice_foreground(painter, i)
+ for i in xrange(len(self._filing)):
+ self._paint_slice_foreground(painter, i, selectionIndex)
- screen = QtGui.QPainter(self)
- screen.drawPixmap(QtCore.QPoint(0, 0), self._canvas)
-
- QtGui.QWidget.paintEvent(self, paintEvent)
-
- def __len__(self):
- return len(self._children)
+ return self._canvas
def _generate_mask(self, mask):
"""
painter.setBrush(QtCore.Qt.color1)
painter.drawEllipse(mask.rect().adjusted(0, 0, -1, -1))
- def _paint_slice_background(self, painter, adjustmentRect, i):
- if i == self._selectionIndex and self._children[i].isEnabled():
- painter.setBrush(self.palette().highlight())
+ def _paint_slice_background(self, painter, adjustmentRect, i, selectionIndex):
+ if i == selectionIndex and self._filing[i].isEnabled():
+ painter.setBrush(self.palette.highlight())
else:
- painter.setBrush(self.palette().background())
- painter.setPen(self.palette().mid().color())
+ painter.setBrush(self.palette.background())
+ painter.setPen(self.palette.mid().color())
- a = self._index_to_angle(i, True)
- b = self._index_to_angle(i + 1, True)
+ a = self._filing._index_to_angle(i, True)
+ b = self._filing._index_to_angle(i + 1, True)
if b < a:
b += 2*math.pi
size = b - a
sizeInDeg = (size * 360 * 16) / (2*math.pi)
painter.drawPie(adjustmentRect, int(startAngleInDeg), int(sizeInDeg))
- def _paint_slice_foreground(self, painter, i):
- child = self._children[i]
+ def _paint_slice_foreground(self, painter, i, selectionIndex):
+ child = self._filing[i]
- a = self._index_to_angle(i, True)
- b = self._index_to_angle(i + 1, True)
+ a = self._filing._index_to_angle(i, True)
+ b = self._filing._index_to_angle(i + 1, True)
if b < a:
b += 2*math.pi
middleAngle = (a + b) / 2
- averageRadius = (self._innerRadius + self._outerRadius) / 2
+ averageRadius = (self._cachedInnerRadius + self._cachedOuterRadius) / 2
sliceX = averageRadius * math.cos(middleAngle)
sliceY = - averageRadius * math.sin(middleAngle)
pieX = self._canvas.rect().center().x()
pieY = self._canvas.rect().center().y()
self._paint_label(
- painter, child.action(), i == self._selectionIndex, pieX+sliceX, pieY+sliceY
+ painter, child.action(), i == selectionIndex, pieX+sliceX, pieY+sliceY
)
def _paint_label(self, painter, action, isSelected, x, y):
if text:
if isSelected:
if action.isEnabled():
- pen = self.palette().highlightedText()
- brush = self.palette().highlight()
+ pen = self.palette.highlightedText()
+ brush = self.palette.highlight()
else:
- pen = self.palette().mid()
- brush = self.palette().background()
+ pen = self.palette.mid()
+ brush = self.palette.background()
else:
if action.isEnabled():
- pen = self.palette().text()
+ pen = self.palette.text()
else:
- pen = self.palette().mid()
- brush = self.palette().background()
+ pen = self.palette.mid()
+ brush = self.palette.background()
leftX = x - averageWidth + icon.width()
topY = y + textHeight/2
painter.setBrush(brush)
painter.drawText(leftX, topY, text)
- def _paint_center_background(self, painter, adjustmentRect):
- dark = self.palette().dark().color()
- light = self.palette().light().color()
- if self._selectionIndex == self.SELECTION_CENTER and self._center.isEnabled():
- background = self.palette().highlight().color()
+ def _paint_center_background(self, painter, adjustmentRect, selectionIndex):
+ dark = self.palette.dark().color()
+ light = self.palette.light().color()
+ if selectionIndex == PieFiling.SELECTION_CENTER and self._filing.center().isEnabled():
+ background = self.palette.highlight().color()
else:
- background = self.palette().background().color()
+ background = self.palette.background().color()
+ innerRadius = self._cachedInnerRadius
innerRect = QtCore.QRect(
- adjustmentRect.center().x() - self._innerRadius,
- adjustmentRect.center().y() - self._innerRadius,
- self._innerRadius * 2 + 1,
- self._innerRadius * 2 + 1,
+ adjustmentRect.center().x() - innerRadius,
+ adjustmentRect.center().y() - innerRadius,
+ innerRadius * 2 + 1,
+ innerRadius * 2 + 1,
)
painter.setPen(QtCore.Qt.NoPen)
innerRect.setTop(r.center().y() + ((r.top() - r.center().y()) / 3) * 1)
innerRect.setBottom(r.center().y() + ((r.bottom() - r.center().y()) / 3) * 1)
- def _paint_center_foreground(self, painter):
+ def _paint_center_foreground(self, painter, selectionIndex):
pieX = self._canvas.rect().center().x()
pieY = self._canvas.rect().center().y()
y = pieY
self._paint_label(
- painter, self._center.action(), self._selectionIndex == self.SELECTION_CENTER, x, y
+ painter,
+ self._filing.center().action(),
+ selectionIndex == PieFiling.SELECTION_CENTER,
+ x, y
)
+
+class QPieMenu(QtGui.QWidget):
+
+ activated = QtCore.pyqtSignal(int)
+ highlighted = QtCore.pyqtSignal(int)
+ canceled = QtCore.pyqtSignal()
+ aboutToShow = QtCore.pyqtSignal()
+ aboutToHide = QtCore.pyqtSignal()
+
+ def __init__(self, parent = None):
+ QtGui.QWidget.__init__(self, parent)
+ self._filing = PieFiling(self.rect().center())
+ self._artist = PieArtist(self._filing)
+ self._selectionIndex = PieFiling.SELECTION_NONE
+
+ self._mouseButtonPressed = False
+ self._mousePosition = ()
+
+ def popup(self, pos):
+ index = self.indexAt(self.mapFromGlobal(QtGui.QCursor.pos()))
+ self._mousePosition = pos
+ self.show()
+
+ def insertItem(self, item, index = -1):
+ self._filing.insertItem(item, index)
+ self.update()
+
+ def removeItemAt(self, index):
+ self._filing.removeItemAt(index)
+ self.update()
+
+ def set_center(self, item):
+ self._filing.set_center(item)
+
+ def clear(self):
+ self._filing.clear()
+ self.update()
+
+ def itemAt(self, index):
+ return self._filing.itemAt(index)
+
+ def indexAt(self, point):
+ return self._filing.indexAt(point)
+
+ def innerRadius(self):
+ return self._filing.innerRadius()
+
+ def setInnerRadius(self, radius):
+ self._filing.setInnerRadius(radius)
+ self.update()
+
+ def outerRadius(self):
+ return self._filing.outerRadius()
+
+ def setOuterRadius(self, radius):
+ self._filing.setOuterRadius(radius)
+ self.update()
+
+ def sizeHint(self):
+ return self._artist.sizeHint()
+
+ def mousePressEvent(self, mouseEvent):
+ lastSelection = self._selectionIndex
+
+ lastMousePos = mouseEvent.pos()
+ self._update_selection(lastMousePos)
+ self._mouseButtonPressed = True
+ self._mousePosition = lastMousePos
+
+ if lastSelection != self._selectionIndex:
+ self.highlighted.emit(self._selectionIndex)
+ self.update()
+
+ def mouseMoveEvent(self, mouseEvent):
+ lastSelection = self._selectionIndex
+
+ lastMousePos = mouseEvent.pos()
+ self._update_selection(lastMousePos)
+
+ if lastSelection != self._selectionIndex:
+ self.highlighted.emit(self._selectionIndex)
+ self.update()
+
+ def mouseReleaseEvent(self, mouseEvent):
+ lastSelection = self._selectionIndex
+
+ lastMousePos = mouseEvent.pos()
+ self._update_selection(lastMousePos)
+ self._mouseButtonPressed = False
+ self._mousePosition = ()
+
+ self._activate_at(self._selectionIndex)
+ self.update()
+
+ def showEvent(self, showEvent):
+ self.aboutToShow.emit()
+
+ self._filing.setCenterPosition(self.rect().center())
+ mask = self._artist.show(self.palette())
+ self.setMask(mask)
+
+ lastMousePos = self.mapFromGlobal(QtGui.QCursor.pos())
+ self._update_selection(lastMousePos)
+
+ QtGui.QWidget.showEvent(self, showEvent)
+
+ def hideEvent(self, hideEvent):
+ self.canceled.emit()
+ self._artist.hide()
+ self._selectionIndex = PieFiling.SELECTION_NONE
+ QtGui.QWidget.hideEvent(self, hideEvent)
+
+ def paintEvent(self, paintEvent):
+ canvas = self._artist.paint(self._selectionIndex)
+
+ screen = QtGui.QPainter(self)
+ screen.drawPixmap(QtCore.QPoint(0, 0), canvas)
+
+ QtGui.QWidget.paintEvent(self, paintEvent)
+
def _select_at(self, index):
self._selectionIndex = index
- numChildren = len(self._children)
+ numChildren = len(self._filing)
loopDelta = max(numChildren, 1)
while self._selectionIndex < 0:
self._selectionIndex += loopDelta
self._selectionIndex -= loopDelta
def _update_selection(self, lastMousePos):
- radius = self._radius_at(lastMousePos)
- if radius < self._innerRadius:
- self._selectionIndex = self.SELECTION_CENTER
- elif radius <= self._outerRadius:
- self._select_at(self._angle_to_index(self._angle_at(lastMousePos)))
+ radius = self._filing._radius_at(lastMousePos)
+ if radius < self._filing.innerRadius():
+ self._selectionIndex = PieFiling.SELECTION_CENTER
+ elif radius <= self._filing.outerRadius():
+ self._select_at(self.indexAt(lastMousePos))
else:
- self._selectionIndex = self.SELECTION_NONE
+ self._selectionIndex = PieFiling.SELECTION_NONE
def _activate_at(self, index):
- if index == self.SELECTION_NONE:
+ if index == PieFiling.SELECTION_NONE:
print "Nothing selected"
return
- elif index == self.SELECTION_CENTER:
- child = self._center
+ elif index == PieFiling.SELECTION_CENTER:
+ child = self._filing.center()
else:
child = self.itemAt(index)
if child.action().isEnabled():
self.aboutToHide.emit()
self.hide()
- def _index_to_angle(self, index, isShifted):
- index = index % len(self._children)
-
- totalWeight = sum(child.weight() for child in self._children)
- if totalWeight == 0:
- totalWeight = 1
- baseAngle = (2 * math.pi) / totalWeight
-
- angle = math.pi / 2
- if isShifted:
- if self._children:
- angle -= (self._children[0].weight() * baseAngle) / 2
- else:
- angle -= baseAngle / 2
- while angle < 0:
- angle += 2*math.pi
-
- for i, child in enumerate(self._children):
- if index < i:
- break
- angle += child.weight() * baseAngle
- while (2*math.pi) < angle:
- angle -= 2*math.pi
-
- return angle
-
- def _angle_to_index(self, angle):
- numChildren = len(self._children)
- if numChildren == 0:
- return self.SELECTION_CENTER
-
- totalWeight = sum(child.weight() for child in self._children)
- if totalWeight == 0:
- totalWeight = 1
- baseAngle = (2 * math.pi) / totalWeight
-
- iterAngle = math.pi / 2 - (self.itemAt(0).weight() * baseAngle) / 2
- while iterAngle < 0:
- iterAngle += 2 * math.pi
-
- oldIterAngle = iterAngle
- for index, child in enumerate(self._children):
- iterAngle += child.weight() * baseAngle
- if oldIterAngle < angle and angle <= iterAngle:
- return index - 1 if index != 0 else numChildren - 1
- elif oldIterAngle < (angle + 2*math.pi) and (angle + 2*math.pi <= iterAngle):
- return index - 1 if index != 0 else numChildren - 1
- oldIterAngle = iterAngle
-
- def _radius_at(self, pos):
- xDelta = pos.x() - self.rect().center().x()
- yDelta = pos.y() - self.rect().center().y()
-
- radius = math.sqrt(xDelta ** 2 + yDelta ** 2)
- return radius
-
- def _angle_at(self, pos):
- xDelta = pos.x() - self.rect().center().x()
- yDelta = pos.y() - self.rect().center().y()
-
- radius = math.sqrt(xDelta ** 2 + yDelta ** 2)
- angle = math.acos(xDelta / radius)
- if 0 <= yDelta:
- angle = 2*math.pi - angle
-
- return angle
-
def _on_key_press(self, keyEvent):
if keyEvent.key in [QtCore.Qt.Key_Right, QtCore.Qt.Key_Down, QtCore.Qt.Key_Tab]:
self._select_at(self._selectionIndex + 1)
elif keyEvent.key in [QtCore.Qt.Key_Return, QtCore.Qt.Key_Enter, QtCore.Qt.Key_Space]:
self._activate_at(self._selectionIndex)
elif keyEvent.key in [QtCore.Qt.Key_Escape, QtCore.Qt.Key_Backspace]:
- self._activate_at(self.SELECTION_NONE)
-
- def _on_mouse_press(self, mouseEvent):
- self._mouseButtonPressed = True
+ self._activate_at(PieFiling.SELECTION_NONE)
def _print(msg):
if __name__ == "__main__":
app = QtGui.QApplication([])
- QPieMenu.NULL_CENTER.setEnabled(False)
+ PieFiling.NULL_CENTER.setEnabled(False)
if False:
pie = QPieMenu()