def _set_labels(self, labels: List[str]):
for i, (label, _) in enumerate(zip(labels, self.__violin_items)):
text = VariableItem(self, label)
item = SimpleLayoutItem(text)
item.setSizePolicy(QSizePolicy.Fixed, QSizePolicy.Fixed)
self.__layout.addItem(item, i, ViolinPlot.LABEL_COLUMN,
Qt.AlignRight | Qt.AlignVCenter)
self.__variable_items.append(item)
def _set_labels(self, labels: List[str]):
for i, (label, _) in enumerate(zip(labels, self.__violin_items)):
short = f"{label[:self.MAX_ATTR_LEN - 1]}..." \
if len(label) > self.MAX_ATTR_LEN else label
text = QGraphicsSimpleTextItem(short, self)
text.setToolTip(label)
item = SimpleLayoutItem(text)
item.setSizePolicy(QSizePolicy.Fixed, QSizePolicy.Fixed)
self.__layout.addItem(item, i, ViolinPlot.LABEL_COLUMN,
Qt.AlignRight | Qt.AlignVCenter)
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
layout = self.__layout
assert layout is self.layout()
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)
layout.addItem(silhouettegroup, i + 1, 2)
if group.label:
layout.addItem(Line(orientation=Qt.Vertical), i + 1, 1)
label = QGraphicsSimpleTextItem(
"{} ({})".format(group.label, len(group.scores)), self
)
label.setRotation(-90)
item = SimpleLayoutItem(
label,
anchor=(0., 1.0),
anchorItem=(0., 0.),
)
item.setSizePolicy(QSizePolicy.Fixed, QSizePolicy.Fixed)
layout.addItem(item, i + 1, 0, Qt.AlignCenter)
textlist = _SilhouettePlotTextListWidget(
self, font=font, elideMode=Qt.ElideRight,
alignment=Qt.AlignLeft | Qt.AlignVCenter
)
textlist.setMaximumWidth(750)
textlist.setFlag(TextListWidget.ItemClipsChildrenToShape, False)
sp = textlist.sizePolicy()
sp.setVerticalPolicy(QSizePolicy.Ignored)
textlist.setSizePolicy(sp)
if group.rownames is not None:
textlist.setItems(group.items)
textlist.setVisible(self.__rowNamesVisible)
else:
textlist.setVisible(False)
layout.addItem(textlist, i + 1, 3)
ax = pg.AxisItem(parent=self, orientation="bottom", maxTickLength=7,
pen=axispen)
ax.setRange(smin, smax)
self.__bottomScale = ax
layout.addItem(ax, len(self.__groups) + 1, 2)