class _SelectionItem(QGraphicsItemGroup):
def __init__(self, parent, path, unscaled_path, label=""):
super().__init__(parent)
self.path = QGraphicsPathItem(path, self)
self.path.setPen(make_pen(width=1, cosmetic=True))
self.addToGroup(self.path)
self.label = QGraphicsSimpleTextItem(label)
self._update_label_pos()
self.addToGroup(self.label)
self.unscaled_path = unscaled_path
def set_path(self, path):
self.path.setPath(path)
self._update_label_pos()
def set_label(self, label):
self.label.setText(label)
self._update_label_pos()
def set_color(self, color):
self.path.setBrush(QColor(color))
def _update_label_pos(self):
path = self.path.path()
elements = (path.elementAt(i) for i in range(path.elementCount()))
points = ((p.x, p.y) for p in elements)
p1, p2, *rest = sorted(points)
x, y = p1[0], (p1[1] + p2[1]) / 2
brect = self.label.boundingRect()
# leaf nodes' paths are 4 pixels higher; leafs are `len(rest) == 3`
self.label.setPos(x - brect.width() - 4,
y - brect.height() + 4 * (len(rest) == 3))
def __setup(self):
# Setup the subwidgets/groups/layout
smax = max((np.nanmax(g.scores) for g in self.__groups
if g.scores.size),
default=1)
smax = 1 if np.isnan(smax) else smax
smin = min((np.nanmin(g.scores) for g in self.__groups
if g.scores.size),
default=-1)
smin = -1 if np.isnan(smin) else smin
smin = min(smin, 0)
font = self.font()
font.setPixelSize(self.__barHeight)
axispen = QPen(Qt.black)
ax = pg.AxisItem(parent=self, orientation="top", maxTickLength=7,
pen=axispen)
ax.setRange(smin, smax)
self.__topScale = ax
self.layout().addItem(ax, 0, 2)
for i, group in enumerate(self.__groups):
silhouettegroup = BarPlotItem(parent=self)
silhouettegroup.setBrush(QBrush(QColor(*group.color)))
silhouettegroup.setPen(self.__pen)
silhouettegroup.setDataRange(smin, smax)
silhouettegroup.setPlotData(group.scores)
silhouettegroup.setPreferredBarSize(self.__barHeight)
silhouettegroup.setData(0, group.indices)
self.layout().addItem(silhouettegroup, i + 1, 2)
if group.label:
self.layout().addItem(Line(orientation=Qt.Vertical), i + 1, 1)
label = QGraphicsSimpleTextItem(self)
label.setText("{} ({})".format(escape(group.label),
len(group.scores)))
item = WrapperLayoutItem(label, Qt.Vertical, parent=self)
self.layout().addItem(item, i + 1, 0, Qt.AlignCenter)
textlist = TextListWidget(self, font=font)
sp = textlist.sizePolicy()
sp.setVerticalPolicy(QSizePolicy.Ignored)
textlist.setSizePolicy(sp)
textlist.setParent(self)
if group.rownames is not None:
textlist.setItems(group.items)
textlist.setVisible(self.__rowNamesVisible)
else:
textlist.setVisible(False)
self.layout().addItem(textlist, i + 1, 3)
ax = pg.AxisItem(parent=self, orientation="bottom", maxTickLength=7,
pen=axispen)
ax.setRange(smin, smax)
self.__bottomScale = ax
self.layout().addItem(ax, len(self.__groups) + 1, 2)
def __setup(self):
# Setup the subwidgets/groups/layout
smax = max((numpy.max(g.scores) for g in self.__groups
if g.scores.size),
default=1)
smin = min((numpy.min(g.scores) for g in self.__groups
if g.scores.size),
default=-1)
smin = min(smin, 0)
font = self.font()
font.setPixelSize(self.__barHeight)
axispen = QPen(Qt.black)
ax = pg.AxisItem(parent=self, orientation="top", maxTickLength=7,
pen=axispen)
ax.setRange(smin, smax)
self.layout().addItem(ax, 0, 2)
for i, group in enumerate(self.__groups):
silhouettegroup = BarPlotItem(parent=self)
silhouettegroup.setBrush(self.__brush)
silhouettegroup.setPen(self.__pen)
silhouettegroup.setDataRange(smin, smax)
silhouettegroup.setPlotData(group.scores)
silhouettegroup.setPreferredBarSize(self.__barHeight)
silhouettegroup.setData(0, group.indices)
self.layout().addItem(silhouettegroup, i + 1, 2)
if group.label:
line = QFrame(frameShape=QFrame.VLine)
proxy = QGraphicsProxyWidget(self)
proxy.setWidget(line)
self.layout().addItem(proxy, i + 1, 1)
label = QGraphicsSimpleTextItem(self)
label.setText("{} ({})".format(escape(group.label),
len(group.scores)))
item = WrapperLayoutItem(label, Qt.Vertical, parent=self)
self.layout().addItem(item, i + 1, 0, Qt.AlignCenter)
textlist = TextListWidget(self, font=font)
sp = textlist.sizePolicy()
sp.setVerticalPolicy(QSizePolicy.Ignored)
textlist.setSizePolicy(sp)
textlist.setParent(self)
if group.rownames is not None:
textlist.setItems(group.items)
textlist.setVisible(self.__rowNamesVisible)
else:
textlist.setVisible(False)
self.layout().addItem(textlist, i + 1, 3)
ax = pg.AxisItem(parent=self, orientation="bottom", maxTickLength=7,
pen=axispen)
ax.setRange(smin, smax)
self.layout().addItem(ax, len(self.__groups) + 1, 2)
def __setup(self) -> None:
self.__clear()
font = self.__effectiveFont if self.__autoScale else self.font()
assert self.__group is None
group = QGraphicsItemGroup()
for text in self.__items:
t = QGraphicsSimpleTextItem(group)
t.setFont(font)
t.setText(text)
t.setData(0, text)
self.__textitems.append(t)
group.setParentItem(self)
self.__group = group
class VariableItem(QGraphicsItemGroup):
MAX_ATTR_LEN = 25
MAX_LABEL_LEN = 150
VALUE_FONT_SETTING = {Updater.SIZE_LABEL: 12,
Updater.IS_ITALIC_LABEL: True}
def __init__(self, parent, label: str):
self.__name: str = None
self.__value: Optional[str] = None
self.__name_item = QGraphicsSimpleTextItem()
self.__value_item = QGraphicsSimpleTextItem()
font = Updater.change_font(QFont(), self.VALUE_FONT_SETTING)
self.__value_item.setFont(font)
self.__max_len = self.MAX_LABEL_LEN
super().__init__(parent)
self._set_data(label)
@property
def items(self):
return self.__name_item, self.__value_item
def boundingRect(self):
name_br = self.__name_item.boundingRect()
width = name_br.width()
height = name_br.height()
if self.__value_item is not None:
value_br = self.__value_item.boundingRect()
width = max(width, value_br.width())
height += value_br.height()
return QRectF(-width, 0, width, height)
def updateGeometry(self):
self.__elide()
self.__align_center()
self.__align_right()
def set_max_len(self, length: int):
self.__max_len = length
self.updateGeometry()
def _set_data(self, label: str):
split = label.split("=")
self.__name = split[0]
self.__name_item.setToolTip(self.__name)
self.updateGeometry() # align before adding to group
self.addToGroup(self.__name_item)
if len(split) > 1:
self.__value = split[1]
self.__value_item.setToolTip(self.__value)
self.updateGeometry() # align before adding to group
self.addToGroup(self.__value_item)
def __elide(self):
fm = QFontMetrics(self.__name_item.font())
text = fm.elidedText(self.__name, Qt.ElideRight, self.__max_len)
self.__name_item.setText(text)
if self.__value is not None:
fm = QFontMetrics(self.__value_item.font())
text = fm.elidedText(self.__value, Qt.ElideRight, self.__max_len)
self.__value_item.setText(text)
def __align_center(self):
if self.__value is not None:
self.__value_item.setY(self.__name_item.boundingRect().height())
def __align_right(self):
self.__name_item.setX(-self.__name_item.boundingRect().width())
if self.__value is not None:
self.__value_item.setX(-self.__value_item.boundingRect().width())
