class ScaleLine:
def __init__(self,
parent,
color,
text_color,
text_start="",
margins=(0, 0, 0, 0),
dp=1,
is_upload=False):
self._parent = parent
self._color = color
self._text_color = text_color
self._text_start = text_start
self._left, self._top, self._right, self._bottom = margins
self._dp = dp
self._is_upload = is_upload
self._line = QGraphicsLineItem(self._parent)
self._line.setZValue(12)
pen = QPen(self._color)
pen.setWidth(1)
pen.setStyle(Qt.DashLine)
self._line.setPen(pen)
if not self._is_upload:
self._text_item = QGraphicsTextItem(self._parent)
self._text_item.setZValue(11)
self._text_item.setDefaultTextColor(self._text_color)
font = self._text_item.font()
font_size = 10 * self._dp
if font_size > 0:
self._text_item.setFont(QFont(font.family(), font_size))
def set_line(self, max_value=0, resize=False):
height = self._parent.size().height() + self._top + self._bottom
shift = int(height - height / 1.1)
y = -self._top + shift
if not self._is_upload:
value = 0
max_value = int(max_value / 1.1)
megabyte = 1024 * 1024
if max_value > megabyte:
value = "{} MB".format(round(max_value / megabyte, 1))
elif max_value > 1024:
max_value //= 1024
if max_value >= 10:
max_value = max_value // 10 * 10
value = "{} KB".format(max_value)
elif max_value > 0:
if max_value >= 10:
max_value = max_value // 10 * 10
value = "{} B".format(max_value)
scale_text = self._text_start if not value \
else "{}{}/s".format(self._text_start, value)
font_height = QFontMetrics(self._text_item.font())\
.boundingRect(scale_text).height()
x = 10
self._text_item.setPos(x, y - font_height - 10)
self._text_item.setPlainText(scale_text)
if not resize:
return
self._line.setLine(QLineF(0, y, self._parent.size().width() + 30, y))
class Node(SchemeItem):
"""The main component of the graph scene.
A node represents a single vertex of the graph, with its associated information and inputs/outputs.
"""
def __init__(self, parent=None):
super().__init__(parent)
self.setFlag(self.ItemIsMovable)
self.setFlag(self.ItemIsSelectable)
# Default size and text
self._size = QSizeF(0, 115)
self._body_item = QGraphicsPathItem(self)
self._title_item = TextLineItem("Node", self)
self._divider = QGraphicsLineItem(self)
self._description_item = TextRectItem("No description", self)
# Inputs and outputs
self.inputs = []
self.outputs = []
self.messages = SortedList(key=lambda item: item.severity())
self._config_widget = None
# Set proper style and color for the item
self.styleChange()
self.paletteChange()
# Operations =======================================================================================================
def showConnectionText(self):
for input in self.inputs:
input.showText()
for output in self.outputs:
output.showText()
def hideConnectionText(self):
for input in self.inputs:
input.hideText()
for output in self.outputs:
output.hideText()
# Input/output management ==========================================================================================
def addInput(self, obj: Input):
self.insertInput(len(self.inputs), obj)
def insertInput(self, index, obj: Input):
index = clamp(index, 0, len(self.inputs))
self.inputs.insert(index, obj)
obj.setParentItem(self)
self._updateInputPositions()
def removeInput(self, index):
removed = self.inputs.pop(index)
removed.setParentItem(None)
self._updateInputPositions()
return removed
def addOutput(self, obj: Output):
self.insertOutput(len(self.outputs), obj)
def insertOutput(self, index, obj: Output):
index = clamp(index, 0, len(self.outputs))
self.outputs.insert(index, obj)
obj.setParentItem(self)
self._updateOutputPositions()
def removeOutput(self, index):
removed = self.outputs.pop(index)
removed.setParentItem(None)
self._updateOutputPositions()
return removed
# Message management ===============================================================================================
def addMessage(self, obj: Message):
self.messages.add(obj)
obj.setParentItem(self)
self._updateMessagePositions()
def removeMessage(self, index):
removed = self.messages.pop(index)
removed.setParentItem(None)
self._updateMessagePositions()
return removed
# Member variables =================================================================================================
def setTitle(self, title):
self._title_item.setText(title)
def setDescription(self, description):
self._description_item.setText(description)
def setConfigWidget(self, w):
self._config_widget = w
def size(self):
return self._size
def title(self):
return self._title_item.text()
def description(self):
return self._description_item.text()
def configWidget(self):
"""Return a widget for configuring this node, or None if it does not exist."""
return self._config_widget
# Updating functions ===============================================================================================
def _updateInputPositions(self):
padding = self.style().pixelMetric(Style.NodeConnectionPadding)
i = 0
for item in self.inputs:
item.setPos(0, padding * (2 * i + 1))
i += 1
def _updateOutputPositions(self):
padding = self.style().pixelMetric(Style.NodeConnectionPadding)
i = 0
for item in self.outputs:
item.setPos(self.size().width(), padding * (2 * i + 1))
i += 1
def _updateMessagePositions(self):
padding = self.style().pixelMetric(Style.NodeMessagePadding)
icon_size = self.style().pixelMetric(Style.MessageIconSize)
i = 0
for msg in self.messages:
msg.setPos(
0,
self.size().height() + padding * (i + 1) + icon_size * i)
i += 1
# Geometry and drawing =============================================================================================
def boundingRect(self) -> QRectF:
frame_width = self.style().pixelMetric(Style.NodeFrameWidth)
return QRectF(QPointF(-frame_width / 2, -frame_width / 2),
self.size() + QSizeF(frame_width, frame_width))
def shape(self):
frame_corner_radius = self.style().pixelMetric(
Style.NodeFrameCornerRadius)
path = QPainterPath()
path.addRoundedRect(QRectF(QPointF(0, 0), self.size()),
frame_corner_radius, frame_corner_radius)
return path
def paint(self, painter: QPainter, option, widget=...) -> None:
pass
# Event handlers ===================================================================================================
def styleChange(self):
self.prepareGeometryChange()
style = self.style()
# Size and Frame
frame_width = style.pixelMetric(Style.NodeWidth)
self._size.setWidth(frame_width)
self._body_item.setPen(style.framePen(self.palette()))
path = QPainterPath()
path.addRoundedRect(QRectF(QPointF(), self.size()),
style.pixelMetric(Style.NodeFrameCornerRadius),
style.pixelMetric(Style.NodeFrameCornerRadius))
self._body_item.setPath(path)
# Title item
title_font = style.font(Style.NodeTitleFont)
title_metrics = QFontMetricsF(title_font)
padding = style.pixelMetric(Style.NodeFrameTextPadding)
self._title_item.setFont(title_font)
self._title_item.setPos(padding, padding)
self._title_item.moveBy(0, title_metrics.capHeight())
self._title_item.setMaximumWidth(frame_width - padding * 2)
# Divider item
div_margin = style.pixelMetric(Style.NodeDividerTextMargin)
div_voffset = padding + title_metrics.capHeight(
) + title_metrics.descent() + div_margin
self._divider.setLine(0, div_voffset, frame_width, div_voffset)
self._divider.setPen(style.framePen(self.palette()))
# Description item
self._description_item.setFont(style.font(Style.NodeDescriptionFont))
self._description_item.setPos(padding, div_voffset + div_margin)
self._description_item.setFrame(
QRectF(
QPointF(0, 0),
QSizeF(
px(frame_width) - padding * 2,
(px(self.size().height()) - padding) -
(div_voffset + padding))))
def paletteChange(self):
self._body_item.setPen(self.style().framePen(self.palette()))
self._body_item.setBrush(self.palette().base())
self._title_item.setBrush(self.palette().text())
self._description_item.setBrush(self.palette().text())
self._divider.setPen(self.style().framePen(self.palette()))
self.update(self.boundingRect())
def itemChange(self, change, value):
if change == self.ItemSelectedHasChanged:
selection = self.scene().selection()
if len(selection.nodes) == 1:
for edge in selection.edges:
edge.autoSelect()
# Update selection status for all connections
for input in self.inputs:
input.autoSelect()
for output in self.outputs:
output.autoSelect()
return super().itemChange(change, value)
def mouseDoubleClickEvent(self, event):
if self.configWidget() is not None:
view = event.widget().parent()
view.configRequested.emit(self)
# Serialization ====================================================================================================
def serialize(self) -> dict:
return {
"title": self.title(),
"description": self.description(),
"inputs": self.inputs,
"outputs": self.outputs,
"messages": list(self.messages),
"position": inch(self.pos())
}
@classmethod
def deserialize(cls, data: dict):
obj = cls()
obj.setTitle(data["title"])
obj.setDescription(data["description"])
obj.setPos(px(data["position"]))
for i in range(len(obj.inputs)):
obj.removeInput(0)
for i in range(len(obj.outputs)):
obj.removeOutput(0)
for i in range(len(obj.messages)):
obj.removeMessage(0)
for input in data["inputs"]:
obj.addInput(input)
for output in data["outputs"]:
obj.addOutput(output)
for message in data["messages"]:
obj.addMessage(message)
return obj