def paint(self, painter, option, widget):
"""
Draws the node base not the ports.
Args:
painter (QtGui.QPainter): painter used for drawing the item.
option (QtGui.QStyleOptionGraphicsItem):
used to describe the parameters needed to draw.
widget (QtWidgets.QWidget): not used.
"""
painter.save()
bg_border = 1.0
rect = QtCore.QRectF(0.5 - (bg_border / 2),
0.5 - (bg_border / 2),
self._width + bg_border,
self._height + bg_border)
radius = 2
border_color = QtGui.QColor(*self.border_color)
path = QtGui.QPainterPath()
path.addRoundedRect(rect, radius, radius)
rect = self.boundingRect()
bg_color = QtGui.QColor(*self.color)
painter.setBrush(bg_color)
painter.setPen(QtCore.Qt.NoPen)
painter.drawRoundedRect(rect, radius, radius)
if self.selected and NODE_SEL_COLOR:
painter.setBrush(QtGui.QColor(*NODE_SEL_COLOR))
painter.drawRoundedRect(rect, radius, radius)
label_rect = QtCore.QRectF(rect.left() + (radius / 2),
rect.top() + (radius / 2),
self._width - (radius / 1.25),
28)
path = QtGui.QPainterPath()
path.addRoundedRect(label_rect, radius / 1.5, radius / 1.5)
painter.setBrush(QtGui.QColor(0, 0, 0, 50))
painter.fillPath(path, painter.brush())
border_width = 0.8
if self.selected and NODE_SEL_BORDER_COLOR:
border_width = 1.2
border_color = QtGui.QColor(*NODE_SEL_BORDER_COLOR)
border_rect = QtCore.QRectF(rect.left() - (border_width / 2),
rect.top() - (border_width / 2),
rect.width() + border_width,
rect.height() + border_width)
pen = QtGui.QPen(border_color, border_width)
pen.setCosmetic(self.viewer().get_zoom() < 0.0)
path = QtGui.QPainterPath()
path.addRoundedRect(border_rect, radius, radius)
painter.setBrush(QtCore.Qt.NoBrush)
painter.setPen(pen)
painter.drawPath(path)
painter.restore()
def paint(self, painter, option, widget):
"""
Draws the backdrop rect.
Args:
painter (QtGui.QPainter): painter used for drawing the item.
option (QtGui.QStyleOptionGraphicsItem):
used to describe the parameters needed to draw.
widget (QtWidgets.QWidget): not used.
"""
painter.save()
rect = self.boundingRect()
color = (self.color[0], self.color[1], self.color[2], 50)
painter.setBrush(QtGui.QColor(*color))
painter.setPen(QtCore.Qt.NoPen)
painter.drawRect(rect)
top_rect = QtCore.QRectF(0.0, 0.0, rect.width(), 20.0)
painter.setBrush(QtGui.QColor(*self.color))
painter.setPen(QtCore.Qt.NoPen)
painter.drawRect(top_rect)
if self.backdrop_text:
painter.setPen(QtGui.QColor(*self.text_color))
txt_rect = QtCore.QRectF(top_rect.x() + 5.0,
top_rect.height() + 2.0,
rect.width() - 5.0, rect.height())
painter.setPen(QtGui.QColor(*self.text_color))
painter.drawText(txt_rect,
QtCore.Qt.AlignLeft | QtCore.Qt.TextWordWrap,
self.backdrop_text)
if self.selected and NODE_SEL_COLOR:
sel_color = [x for x in NODE_SEL_COLOR]
sel_color[-1] = 10
painter.setBrush(QtGui.QColor(*sel_color))
painter.setPen(QtCore.Qt.NoPen)
painter.drawRect(rect)
txt_rect = QtCore.QRectF(top_rect.x(),
top_rect.y() + 1.2, rect.width(),
top_rect.height())
painter.setPen(QtGui.QColor(*self.text_color))
painter.drawText(txt_rect, QtCore.Qt.AlignCenter, self.name)
path = QtGui.QPainterPath()
path.addRect(rect)
border_color = self.color
if self.selected and NODE_SEL_BORDER_COLOR:
border_color = NODE_SEL_BORDER_COLOR
painter.setBrush(QtCore.Qt.NoBrush)
painter.setPen(QtGui.QPen(QtGui.QColor(*border_color), 1))
painter.drawPath(path)
painter.restore()
def paint(self, painter, option, widget):
"""
Draws the circular port.
Args:
painter (QtGui.QPainter): painter used for drawing the item.
option (QtGui.QStyleOptionGraphicsItem):
used to describe the parameters needed to draw.
widget (QtWidgets.QWidget): not used.
"""
painter.save()
rect = QtCore.QRectF(0.0, 0.8, self._width, self._height)
path = QtGui.QPainterPath()
path.addEllipse(rect)
if self._hovered:
color = QtGui.QColor(*PORT_HOVER_COLOR)
border_color = QtGui.QColor(*PORT_HOVER_BORDER_COLOR)
elif self.connected_pipes:
color = QtGui.QColor(*PORT_ACTIVE_COLOR)
border_color = QtGui.QColor(*PORT_ACTIVE_BORDER_COLOR)
else:
color = QtGui.QColor(*self.color)
border_color = QtGui.QColor(*self.border_color)
painter.setBrush(color)
pen = QtGui.QPen(border_color, 1.5)
painter.setPen(pen)
painter.drawEllipse(self.boundingRect())
if self.connected_pipes and not self._hovered:
painter.setBrush(border_color)
w = self._width / 2.5
h = self._height / 2.5
rect = QtCore.QRectF(self.boundingRect().center().x() - w / 2,
self.boundingRect().center().y() - h / 2, w,
h)
painter.drawEllipse(rect)
elif self._hovered:
if self.multi_connection:
painter.setBrush(color)
w = self._width / 1.8
h = self._height / 1.8
else:
painter.setBrush(border_color)
w = self._width / 3.5
h = self._height / 3.5
rect = QtCore.QRectF(self.boundingRect().center().x() - w / 2,
self.boundingRect().center().y() - h / 2, w,
h)
painter.drawEllipse(rect)
painter.restore()
def reset_zoom(self, cent=None):
self._scene_range = QtCore.QRectF(0, 0,
self.size().width(),
self.size().height())
if cent:
self._scene_range.translate(cent - self._scene_range.center())
self._update_scene()
def paint(self, painter, option, widget):
"""
Draws the circular port.
Args:
painter (QtGui.QPainter): painter used for drawing the item.
option (QtGui.QStyleOptionGraphicsItem):
used to describe the parameters needed to draw.
widget (QtWidgets.QWidget): not used.
"""
painter.save()
rect = QtCore.QRectF(0.0, 0.8, self._width, self._height)
painter.setBrush(QtGui.QColor(0, 0, 0, 200))
painter.setPen(QtGui.QPen(QtGui.QColor(0, 0, 0, 255), 1.8))
path = QtGui.QPainterPath()
path.addEllipse(rect)
painter.drawPath(path)
if self._hovered:
color = QtGui.QColor(*PORT_HOVER_COLOR)
border_color = QtGui.QColor(*PORT_HOVER_BORDER_COLOR)
elif self.connected_pipes:
color = QtGui.QColor(*PORT_ACTIVE_COLOR)
border_color = QtGui.QColor(*PORT_ACTIVE_BORDER_COLOR)
else:
color = QtGui.QColor(*self.color)
border_color = QtGui.QColor(*self.border_color)
painter.setBrush(color)
pen = QtGui.QPen(border_color, 1.5)
painter.setPen(pen)
painter.drawEllipse(self.boundingRect())
painter.restore()
def _items_near(self, pos, item_type=None, width=20, height=20):
x, y = pos.x() - width, pos.y() - height
rect = QtCore.QRectF(x, y, width, height)
items = []
for item in self.scene().items(rect):
if not item_type or isinstance(item, item_type):
items.append(item)
return items
def paint(self, painter, option, widget):
"""
Draws the slicer pipe.
Args:
painter (QtGui.QPainter): painter used for drawing the item.
option (QtGui.QStyleOptionGraphicsItem):
used to describe the parameters needed to draw.
widget (QtWidgets.QWidget): not used.
"""
color = QtGui.QColor(*PIPE_SLICER_COLOR)
p1 = self.path().pointAtPercent(0)
p2 = self.path().pointAtPercent(1)
size = 6.0
offset = size / 2
painter.save()
painter.setRenderHint(painter.Antialiasing, True)
font = painter.font()
font.setPointSize(12)
painter.setFont(font)
text = 'slice'
text_x = painter.fontMetrics().width(text) / 2
text_y = painter.fontMetrics().height() / 1.5
text_pos = QtCore.QPointF(p1.x() - text_x, p1.y() - text_y)
text_color = QtGui.QColor(*PIPE_SLICER_COLOR)
text_color.setAlpha(80)
painter.setPen(QtGui.QPen(text_color, 1.5, QtCore.Qt.SolidLine))
painter.drawText(text_pos, text)
painter.setPen(QtGui.QPen(color, 1.5, QtCore.Qt.DashLine))
painter.drawPath(self.path())
painter.setPen(QtGui.QPen(color, 1.5, QtCore.Qt.SolidLine))
painter.setBrush(color)
rect = QtCore.QRectF(p1.x() - offset, p1.y() - offset, size, size)
painter.drawEllipse(rect)
rect = QtCore.QRectF(p2.x() - offset, p2.y() - offset, size, size)
painter.drawEllipse(rect)
painter.restore()
def _items_near(self, pos, item_type=None, width=20, height=20):
x, y = pos.x() - width, pos.y() - height
rect = QtCore.QRectF(x, y, width, height)
items = []
excl = [self._LIVE_PIPE, self._SLICER_PIPE]
for item in self.scene().items(rect):
if item in excl:
continue
if not item_type or isinstance(item, item_type):
items.append(item)
return items
def scale(self, sx, sy, pos=None):
scale = [sx, sx]
center = pos or self._scene_range.center()
w = self._scene_range.width() / scale[0]
h = self._scene_range.height() / scale[1]
self._scene_range = QtCore.QRectF(
center.x() - (center.x() - self._scene_range.left()) / scale[0],
center.y() - (center.y() - self._scene_range.top()) / scale[1], w,
h)
self._update_scene()
def mousePressEvent(self, event):
if event.button() == QtCore.Qt.LeftButton:
pos = event.scenePos()
rect = QtCore.QRectF(pos.x() - 5, pos.y() - 5, 10, 10)
item = self.scene().items(rect)[0]
if isinstance(item, (PortItem, Pipe)):
self.setFlag(self.ItemIsMovable, False)
return
if self.selected:
return
viewer = self.viewer()
[n.setSelected(False) for n in viewer.selected_nodes()]
self._nodes += self.get_nodes(False)
[n.setSelected(True) for n in self._nodes]
def boundingRect(self):
return QtCore.QRectF(0.0, 0.0, self._width + PORT_FALLOFF,
self._height)
def boundingRect(self):
return QtCore.QRectF(0.0, 0.0, self._width, self._height)
def paint(self, painter, option, widget):
"""
Draws the overlay disabled X item on top of a node item.
Args:
painter (QtGui.QPainter): painter used for drawing the item.
option (QtGui.QStyleOptionGraphicsItem):
used to describe the parameters needed to draw.
widget (QtWidgets.QWidget): not used.
"""
painter.save()
margin = 20
rect = self.boundingRect()
dis_rect = QtCore.QRectF(rect.left() - (margin / 2),
rect.top() - (margin / 2),
rect.width() + margin,
rect.height() + margin)
pen = QtGui.QPen(QtGui.QColor(*self.color), 8)
pen.setCapStyle(QtCore.Qt.RoundCap)
painter.setPen(pen)
painter.drawLine(dis_rect.topLeft(), dis_rect.bottomRight())
painter.drawLine(dis_rect.topRight(), dis_rect.bottomLeft())
bg_color = QtGui.QColor(*self.color)
bg_color.setAlpha(100)
bg_margin = -0.5
bg_rect = QtCore.QRectF(dis_rect.left() - (bg_margin / 2),
dis_rect.top() - (bg_margin / 2),
dis_rect.width() + bg_margin,
dis_rect.height() + bg_margin)
painter.setPen(QtGui.QPen(QtGui.QColor(0, 0, 0, 0)))
painter.setBrush(bg_color)
painter.drawRoundedRect(bg_rect, 5, 5)
pen = QtGui.QPen(QtGui.QColor(155, 0, 0, 255), 0.7)
painter.setPen(pen)
painter.drawLine(dis_rect.topLeft(), dis_rect.bottomRight())
painter.drawLine(dis_rect.topRight(), dis_rect.bottomLeft())
point_size = 4.0
point_pos = (dis_rect.topLeft(), dis_rect.topRight(),
dis_rect.bottomLeft(), dis_rect.bottomRight())
painter.setBrush(QtGui.QColor(255, 0, 0, 255))
for p in point_pos:
p.setX(p.x() - (point_size / 2))
p.setY(p.y() - (point_size / 2))
point_rect = QtCore.QRectF(
p, QtCore.QSizeF(point_size, point_size))
painter.drawEllipse(point_rect)
if self.text:
font = painter.font()
font.setPointSize(10)
painter.setFont(font)
font_metrics = QtGui.QFontMetrics(font)
font_width = font_metrics.width(self.text)
font_height = font_metrics.height()
txt_w = font_width * 1.25
txt_h = font_height * 2.25
text_bg_rect = QtCore.QRectF((rect.width() / 2) - (txt_w / 2),
(rect.height() / 2) - (txt_h / 2),
txt_w, txt_h)
painter.setPen(QtGui.QPen(QtGui.QColor(255, 0, 0), 0.5))
painter.setBrush(QtGui.QColor(*self.color))
painter.drawRoundedRect(text_bg_rect, 2, 2)
text_rect = QtCore.QRectF((rect.width() / 2) - (font_width / 2),
(rect.height() / 2) - (font_height / 2),
txt_w * 2, font_height * 2)
painter.setPen(QtGui.QPen(QtGui.QColor(255, 0, 0), 1))
painter.drawText(text_rect, self.text)
painter.restore()
def __init__(self, parent=None):
super(NodeViewer, self).__init__(parent)
self.setScene(NodeScene(self))
self.setRenderHint(QtGui.QPainter.Antialiasing, True)
self.setHorizontalScrollBarPolicy(QtCore.Qt.ScrollBarAlwaysOff)
self.setVerticalScrollBarPolicy(QtCore.Qt.ScrollBarAlwaysOff)
self.setViewportUpdateMode(QtWidgets.QGraphicsView.FullViewportUpdate)
self.setCacheMode(QtWidgets.QGraphicsView.CacheBackground)
self.setOptimizationFlag(
QtWidgets.QGraphicsView.DontAdjustForAntialiasing)
self.setAcceptDrops(True)
self.resize(850, 800)
self._scene_range = QtCore.QRectF(0, 0,
self.size().width(),
self.size().height())
self._update_scene()
self._last_size = self.size()
self._pipe_layout = PIPE_LAYOUT_CURVED
self._detached_port = None
self._start_port = None
self._origin_pos = None
self._previous_pos = QtCore.QPoint(self.width(), self.height())
self._prev_selection_nodes = []
self._prev_selection_pipes = []
self._node_positions = {}
self._rubber_band = QtWidgets.QRubberBand(
QtWidgets.QRubberBand.Rectangle, self)
self._LIVE_PIPE = LivePipe()
self._LIVE_PIPE.setVisible(False)
self.scene().addItem(self._LIVE_PIPE)
self._SLICER_PIPE = SlicerPipe()
self._SLICER_PIPE.setVisible(False)
self.scene().addItem(self._SLICER_PIPE)
self._undo_stack = QtWidgets.QUndoStack(self)
self._search_widget = TabSearchMenuWidget(self)
self._search_widget.search_submitted.connect(self._on_search_submitted)
# workaround fix for shortcuts from the non-native menu actions
# don't seem to trigger so we create a hidden menu bar.
menu_bar = QtWidgets.QMenuBar(self)
menu_bar.setNativeMenuBar(False)
# shortcuts don't work with "setVisibility(False)".
menu_bar.setMaximumWidth(0)
self._ctx_menu = BaseMenu('NodeGraph', self)
self._ctx_node_menu = BaseMenu('Nodes', self)
menu_bar.addMenu(self._ctx_menu)
menu_bar.addMenu(self._ctx_node_menu)
self._ctx_node_menu.setDisabled(True)
self.acyclic = True
self.LMB_state = False
self.RMB_state = False
self.MMB_state = False
self.ALT_state = False
self.CTRL_state = False
self.SHIFT_state = False
self.COLLIDING_state = False
def paint(self, painter, option, widget):
"""
Draws the circular port.
Args:
painter (QtGui.QPainter): painter used for drawing the item.
option (QtGui.QStyleOptionGraphicsItem):
used to describe the parameters needed to draw.
widget (QtWidgets.QWidget): not used.
"""
painter.save()
### display the falloff colision ###
# pen = QtGui.QPen(QtGui.QColor(255, 255, 255, 80), 0.8)
# pen.setStyle(QtCore.Qt.DotLine)
# painter.setPen(pen)
# painter.drawRect(self.boundingRect())
rect_w = self._width / 1.8
rect_h = self._height / 1.8
rect_x = self.boundingRect().center().x() - (rect_w / 2)
rect_y = self.boundingRect().center().y() - (rect_h / 2)
port_rect = QtCore.QRectF(rect_x, rect_y, rect_w, rect_h)
if self._hovered:
color = QtGui.QColor(*PORT_HOVER_COLOR)
border_color = QtGui.QColor(*PORT_HOVER_BORDER_COLOR)
elif self.connected_pipes:
color = QtGui.QColor(*PORT_ACTIVE_COLOR)
border_color = QtGui.QColor(*PORT_ACTIVE_BORDER_COLOR)
else:
color = QtGui.QColor(*self.color)
border_color = QtGui.QColor(*self.border_color)
painter.setBrush(color)
pen = QtGui.QPen(border_color, 1.8)
painter.setPen(pen)
painter.drawEllipse(port_rect)
if self.connected_pipes and not self._hovered:
painter.setBrush(border_color)
w = port_rect.width() / 2.5
h = port_rect.height() / 2.5
rect = QtCore.QRectF(port_rect.center().x() - w / 2,
port_rect.center().y() - h / 2, w, h)
painter.drawEllipse(rect)
elif self._hovered:
if self.multi_connection:
pen = QtGui.QPen(border_color, 1.4)
painter.setPen(pen)
painter.setBrush(color)
w = port_rect.width() / 1.8
h = port_rect.height() / 1.8
else:
painter.setBrush(border_color)
w = port_rect.width() / 3.5
h = port_rect.height() / 3.5
rect = QtCore.QRectF(port_rect.center().x() - w / 2,
port_rect.center().y() - h / 2, w, h)
painter.drawEllipse(rect)
painter.restore()
def boundingRect(self):
return QtCore.QRectF(0.5, 0.5, self._size, self._size)
def paint(self, painter, option, widget):
"""
Draws the connection line.
Args:
painter (QtGui.QPainter): painter used for drawing the item.
option (QtGui.QStyleOptionGraphicsItem):
used to describe the parameters needed to draw.
widget (QtWidgets.QWidget): not used.
"""
color = QtGui.QColor(*PIPE_ACTIVE_COLOR)
pen_style = PIPE_STYLES.get(PIPE_STYLE_DASHED)
pen_width = PIPE_WIDTH + 0.35
pen = QtGui.QPen(color, pen_width)
pen.setStyle(pen_style)
pen.setCapStyle(QtCore.Qt.RoundCap)
painter.save()
painter.setPen(pen)
painter.setRenderHint(painter.Antialiasing, True)
painter.drawPath(self.path())
cen_x = self.path().pointAtPercent(0.5).x()
cen_y = self.path().pointAtPercent(0.5).y()
loc_pt = self.path().pointAtPercent(0.9)
tgt_pt = self.path().pointAtPercent(1.0)
dist = math.hypot(tgt_pt.x() - cen_x, tgt_pt.y() - cen_y)
if dist < 0.05:
painter.restore()
return
# draw circle
size = 10.0
if dist < 50.0:
size *= (dist / 50.0)
rect = QtCore.QRectF(cen_x-(size/2), cen_y-(size/2), size, size)
painter.setBrush(color)
painter.setPen(QtGui.QPen(color.darker(130), pen_width))
painter.drawEllipse(rect)
# draw arrow
color.setAlpha(255)
painter.setBrush(color.darker(200))
pen_width = 0.6
if dist < 1.0:
pen_width *= 1.0 + dist
painter.setPen(QtGui.QPen(color, pen_width))
transform = QtGui.QTransform()
transform.translate(tgt_pt.x(), tgt_pt.y())
radians = math.atan2(tgt_pt.y() - loc_pt.y(),
tgt_pt.x() - loc_pt.x())
degrees = math.degrees(radians) + 90
transform.rotate(degrees)
scale = 1.0
if dist < 20.0:
scale = dist / 20.0
transform.scale(scale, scale)
painter.drawPolygon(transform.map(self._arrow))
painter.restore()