def _draw_grid(self, painter, rect, pen, grid_size):
lines = []
left = int(rect.left()) - (int(rect.left()) % grid_size)
top = int(rect.top()) - (int(rect.top()) % grid_size)
x = left
while x < rect.right():
x += grid_size
lines.append(QtCore.QLineF(x, rect.top(), x, rect.bottom()))
y = top
while y < rect.bottom():
y += grid_size
lines.append(QtCore.QLineF(rect.left(), y, rect.right(), y))
painter.setPen(pen)
painter.drawLines(lines)
def draw_path(self, start_port, end_port, cursor_pos=None):
"""
Draws the path between ports.
Args:
start_port (PortItem): port used to draw the starting point.
end_port (PortItem): port used to draw the end point.
cursor_pos (QtCore.QPointF): cursor position if specified this
will be the draw end point.
"""
if not start_port:
return
offset = (start_port.boundingRect().width() / 2)
pos1 = start_port.scenePos()
pos1.setX(pos1.x() + offset)
pos1.setY(pos1.y() + offset)
if cursor_pos:
pos2 = cursor_pos
elif end_port:
offset = start_port.boundingRect().width() / 2
pos2 = end_port.scenePos()
pos2.setX(pos2.x() + offset)
pos2.setY(pos2.y() + offset)
else:
return
line = QtCore.QLineF(pos1, pos2)
path = QtGui.QPainterPath()
path.moveTo(line.x1(), line.y1())
if self.viewer_pipe_layout() == PIPE_LAYOUT_STRAIGHT:
path.lineTo(pos2)
self.setPath(path)
return
ctr_offset_x1, ctr_offset_x2 = pos1.x(), pos2.x()
tangent = ctr_offset_x1 - ctr_offset_x2
tangent = (tangent * -1) if tangent < 0 else tangent
max_width = start_port.node.boundingRect().width() / 2
tangent = max_width if tangent > max_width else tangent
if start_port.port_type == IN_PORT:
ctr_offset_x1 -= tangent
ctr_offset_x2 += tangent
else:
ctr_offset_x1 += tangent
ctr_offset_x2 -= tangent
ctr_point1 = QtCore.QPointF(ctr_offset_x1, pos1.y())
ctr_point2 = QtCore.QPointF(ctr_offset_x2, pos2.y())
path.cubicTo(ctr_point1, ctr_point2, pos2)
self.setPath(path)
