def paintSlider(self, painter, option, index):
"""
Draw the virtual slider for the item.
:type painter: QtWidgets.QPainter
:type option: QtWidgets.QStyleOptionViewItem
:type index: QtCore.QModelIndex
"""
if not self.PAINT_SLIDER:
return
if not self.itemsWidget().isIconView():
return
# Draw slider background
painter.setPen(QtGui.QPen(QtCore.Qt.NoPen))
rect = self.visualRect(option)
color = self.itemsWidget().backgroundColor().toRgb()
color.setAlpha(75)
painter.setBrush(QtGui.QBrush(color))
height = rect.height()
ratio = self.sliderValue()
if ratio < 0:
width = 0
elif ratio > 100:
width = rect.width()
else:
width = rect.width() * (float(ratio) / 100)
rect.setWidth(width)
rect.setHeight(height)
painter.drawRect(rect)
# Draw slider value
rect = self.visualRect(option)
rect.setY(rect.y() + (4 * self.dpi()))
color = self.itemsWidget().textColor().toRgb()
color.setAlpha(220)
pen = QtGui.QPen(color)
align = QtCore.Qt.AlignTop | QtCore.Qt.AlignHCenter
painter.setPen(pen)
painter.drawText(rect, align, str(self.sliderValue()) + "%")
def paintBackground(self, painter, option, index):
"""
Draw the background for the item.
:type painter: QtWidgets.QPainter
:type option: QtWidgets.QStyleOptionViewItem
:type index: QtCore.QModelIndex
"""
isSelected = option.state & QtWidgets.QStyle.State_Selected
isMouseOver = option.state & QtWidgets.QStyle.State_MouseOver
painter.setPen(QtGui.QPen(QtCore.Qt.NoPen))
visualRect = self.visualRect(option)
if isSelected:
color = self.backgroundSelectedColor()
painter.setBrush(QtGui.QBrush(color))
elif isMouseOver:
color = self.backgroundHoverColor()
painter.setBrush(QtGui.QBrush(color))
else:
color = self.backgroundColor()
painter.setBrush(QtGui.QBrush(color))
if not self.itemsWidget().isIconView():
spacing = 1 * self.dpi()
height = visualRect.height() - spacing
visualRect.setHeight(height)
painter.drawRect(visualRect)
def setBadge(self, x, y, w, h, color=None):
"""
Set a for the icon.
:type x: int
:type y: int
:type w: int
:type h: int
:type color: QtGui.QColor or None
"""
color = color or QtGui.QColor(240, 100, 100)
size = self.actualSize(QtCore.QSize(256, 256))
pixmap = self.pixmap(size)
painter = QtGui.QPainter(pixmap)
pen = QtGui.QPen(color)
pen.setWidth(0)
painter.setPen(pen)
painter.setBrush(color)
painter.setRenderHint(QtGui.QPainter.Antialiasing)
painter.drawEllipse(x, y, w, h)
painter.end()
icon = QtGui.QIcon(pixmap)
self.swap(icon)
def paintBackground(self, painter, option, index):
"""
Draw the background for the item.
:type painter: QtWidgets.QPainter
:type option: QtWidgets.QStyleOptionViewItem
:type index: QtCore.QModelIndex
:rtype: None
"""
super(GroupItem, self).paintBackground(painter, option, index)
painter.setPen(QtGui.QPen(QtCore.Qt.NoPen))
visualRect = self.visualRect(option)
text = self.name()
metrics = QtGui.QFontMetricsF(self._font)
textWidth = metrics.width(text)
padding = (25 * self.dpi())
visualRect.setX(textWidth + padding)
visualRect.setY(visualRect.y() + (visualRect.height() / 2))
visualRect.setHeight(2 * self.dpi())
visualRect.setWidth(visualRect.width() - padding)
color = QtGui.QColor(
self.textColor().red(),
self.textColor().green(),
self.textColor().blue(), 10
)
painter.setBrush(QtGui.QBrush(color))
painter.drawRect(visualRect)
def _paintText(self, painter, option, column):
if self.itemsWidget().isIconView():
text = self.name()
else:
label = self.labelFromColumn(column)
text = self.displayText(label)
isSelected = option.state & QtWidgets.QStyle.State_Selected
if isSelected:
color = self.textSelectedColor()
else:
color = self.textColor()
visualRect = self.visualRect(option)
width = visualRect.width()
height = visualRect.height()
padding = self.padding()
x = padding / 2
y = padding / 2
visualRect.translate(x, y)
visualRect.setWidth(width - padding)
visualRect.setHeight(height - padding)
font = self.font(column)
align = self.textAlignment(column)
metrics = QtGui.QFontMetricsF(font)
if text:
textWidth = metrics.width(text)
else:
textWidth = 1
# # Check if the current text fits within the rect.
if textWidth > visualRect.width() - padding:
visualWidth = visualRect.width()
text = metrics.elidedText(text, QtCore.Qt.ElideRight, visualWidth)
align = QtCore.Qt.AlignLeft
if self.itemsWidget().isIconView():
align = align | QtCore.Qt.AlignBottom
else:
align = align | QtCore.Qt.AlignVCenter
pen = QtGui.QPen(color)
painter.setPen(pen)
painter.setFont(font)
painter.drawText(visualRect, align, text)
def drawIconBorder(self, painter, pixmapRect):
"""
Draw a border around the icon.
:type painter: QtWidgets.QPainter
:type pixmapRect: QtWidgets.QRect
:rtype: None
"""
pixmapRect = QtCore.QRect(pixmapRect)
pixmapRect.setX(pixmapRect.x() - 5)
pixmapRect.setY(pixmapRect.y() - 5)
pixmapRect.setWidth(pixmapRect.width() + 5)
pixmapRect.setHeight(pixmapRect.height() + 5)
color = QtGui.QColor(255, 255, 255, 10)
painter.setPen(QtGui.QPen(QtCore.Qt.NoPen))
painter.setBrush(QtGui.QBrush(color))
painter.drawRect(pixmapRect)
def _paintText(self, painter, option, column):
if self.itemsWidget().isIconView():
text = self.name()
else:
label = self.labelFromColumn(column)
text = self.displayText(label)
isSelected = option.state & QtWidgets.QStyle.State_Selected
if isSelected:
color = self.textSelectedColor()
else:
color = self.textColor()
visualRect = self.visualRect(option)
width = visualRect.width()
height = visualRect.height()
padding = self.padding()
x = padding / 2
y = padding / 2
visualRect.translate(x, y)
visualRect.setWidth(width - padding)
visualRect.setHeight(height - padding)
font = self.font(column)
align = self.textAlignment(column)
metrics = QtGui.QFontMetricsF(font)
if text:
textWidth = metrics.width(text)
else:
textWidth = 1
# # Check if the current text fits within the rect.
if textWidth > visualRect.width() - padding:
visualWidth = visualRect.width()
text = metrics.elidedText(text, QtCore.Qt.ElideRight, visualWidth)
align = QtCore.Qt.AlignLeft
align = align | QtCore.Qt.AlignVCenter
rect = QtCore.QRect(visualRect)
if self.itemsWidget().isIconView():
if self.isLabelOverItem() or self.isLabelUnderItem():
padding = 8 if padding < 8 else padding
height = metrics.height() + (padding / 2)
y = (rect.y() + rect.height()) - height
rect.setY(y)
rect.setHeight(height)
if self.isLabelOverItem():
color2 = self.itemsWidget().backgroundColor().toRgb()
color2.setAlpha(200)
painter.setPen(QtCore.Qt.NoPen)
painter.setBrush(QtGui.QBrush(color2))
painter.drawRect(rect)
pen = QtGui.QPen(color)
painter.setPen(pen)
painter.setFont(font)
painter.drawText(rect, align, text)
