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):
self.__clear()
font = self.font()
group = QGraphicsItemGroup(self)
for text in self.__items:
t = QGraphicsSimpleTextItem(text, group)
t.setData(0, text)
t.setFont(font)
t.setToolTip(text)
self.__textitems.append(t)
def __setup(self):
self.__clear()
font = self.font()
group = QGraphicsItemGroup(self)
for text in self.__items:
t = QGraphicsSimpleTextItem(text, group)
t.setData(0, text)
t.setFont(font)
t.setToolTip(text)
self.__textitems.append(t)
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.setToolTip(text)
t.setData(0, text)
self.__textitems.append(t)
group.setParentItem(self)
self.__group = group
class PartItem(QGraphicsPathItem):
COLOR = NotImplemented
TIP_LEN = 13
def __init__(self, value: float, label: Tuple[str, float],
norm_value: float):
super().__init__()
color = QColor(*self.light_rgb)
self.value = value
self.norm_value = norm_value
self.setPath(self.get_path())
pen = QPen(color)
pen.setWidth(2)
self.setPen(pen)
value = np.abs(value)
self.value_item = item = QGraphicsSimpleTextItem(_str(value))
item.setToolTip(_str(value, 3))
font = item.font()
font.setPixelSize(11)
item.setFont(font)
width = item.boundingRect().width()
item.setX(StripeItem.WIDTH / 2 - width / 2)
item.setPen(color)
item.setBrush(color)
self.label_item = QGraphicsSimpleTextItem(
f"{label[0]} = {_str(label[1])}")
self.label_item.setToolTip(f"{label[0]} = {_str(label[1], 3)}")
self.label_item.setX(StripeItem.WIDTH + StripePlot.SPACING)
@property
def light_rgb(self) -> List[int]:
rgb = np.array(self.COLOR)
return list((rgb + (255 - rgb) * 0.7).astype(int))
@property
def value_height(self) -> float:
return self.value_item.boundingRect().height()
@property
def label_height(self) -> float:
return self.label_item.boundingRect().height()
def get_path(self) -> QPainterPath:
raise NotImplementedError
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())
