def paint(self, painter, option, index):
painter.save()
self.drawBackground(painter, option, index)
ratio = index.data(TableBarItem.BarRole)
if isinstance(ratio, float):
if math.isnan(ratio):
ratio = None
color = None
if ratio is not None:
if self.color_schema is not None:
class_ = index.data(TableClassValueRole)
if isinstance(class_, Orange.data.Value) and \
class_.variable.is_discrete and \
not math.isnan(class_):
color = self.color_schema[int(class_)]
else:
color = index.data(self.BarColorRole)
if color is None:
color = self.color
rect = option.rect
if ratio is not None:
pw = 5
hmargin = 3 + pw / 2 # + half pen width for the round line cap
vmargin = 1
textoffset = pw + vmargin * 2
baseline = rect.bottom() - textoffset / 2
width = (rect.width() - 2 * hmargin) * ratio
painter.save()
painter.setRenderHint(QtGui.QPainter.Antialiasing)
painter.setPen(QtGui.QPen(QtGui.QBrush(color), pw,
Qt.SolidLine, Qt.RoundCap))
line = QtCore.QLineF(
rect.left() + hmargin, baseline,
rect.left() + hmargin + width, baseline
)
painter.drawLine(line)
painter.restore()
text_rect = rect.adjusted(0, 0, 0, -textoffset)
else:
text_rect = rect
text = str(index.data(Qt.DisplayRole))
self.drawDisplay(painter, option, text_rect, text)
painter.restore()
def paint(self, painter, option, index):
dist = self.distribution(index)
if dist is None or self.__colors is None:
return super().paint(painter, option, index)
if not numpy.isfinite(numpy.sum(dist)):
return super().paint(painter, option, index)
nvalues = len(dist)
if len(self.__colors) < nvalues:
colors = colorpalette.ColorPaletteGenerator(nvalues)
colors = [colors[i] for i in range(nvalues)]
else:
colors = self.__colors
if option.widget is not None:
style = option.widget.style()
else:
style = QApplication.style()
self.initStyleOption(option, index)
text = option.text
metrics = option.fontMetrics
margin = style.pixelMetric(QStyle.PM_FocusFrameHMargin, option,
option.widget) + 1
bottommargin = min(margin, 1)
rect = option.rect.adjusted(margin, margin, -margin, -bottommargin)
textrect = style.subElementRect(QStyle.SE_ItemViewItemText, option,
option.widget)
# Are the margins included in the subElementRect?? -> No!
textrect = textrect.adjusted(margin, margin, -margin, -bottommargin)
text = option.fontMetrics.elidedText(text, option.textElideMode,
textrect.width())
spacing = max(metrics.leading(), 1)
distheight = rect.height() - metrics.height() - spacing
distheight = numpy.clip(distheight, 2, metrics.height())
painter.save()
painter.setClipRect(option.rect)
painter.setFont(option.font)
painter.setRenderHint(QPainter.Antialiasing)
style.drawPrimitive(QStyle.PE_PanelItemViewRow, option, painter,
option.widget)
style.drawPrimitive(QStyle.PE_PanelItemViewItem, option, painter,
option.widget)
if option.state & QStyle.State_Selected:
color = option.palette.highlightedText().color()
else:
color = option.palette.text().color()
painter.setPen(QtGui.QPen(color))
textrect = textrect.adjusted(0, 0, 0, -distheight - spacing)
distrect = QtCore.QRect(
textrect.bottomLeft() + QtCore.QPoint(0, spacing),
QtCore.QSize(rect.width(), distheight))
painter.setPen(QtGui.QPen(Qt.lightGray, 0.3))
drawDistBar(painter, distrect, dist, colors)
painter.restore()
if text:
style.drawItemText(painter, textrect, option.displayAlignment,
option.palette,
option.state & QStyle.State_Enabled, text)
