def midLineDraw(self, event):
newLine = QtCore.QLineF(self.line.line().p1(), event.scenePos())
self.line.setLine(newLine)
# attach source or sink connection
#if QtCore.Qt.AltModifier == getKeyboardModifiers():
# self.line.setPen(QtGui.QPen(QtCore.Qt.blue, 2))
# self.dumpCursorStack()
# QtGui.QApplication.setOverrideCursor(
# QtGui.QCursor(QtCore.Qt.CrossCursor))
# return
#else:
# self.line.setPen(QtGui.QPen(QtCore.Qt.red, 2))
startItems = self.items(event.scenePos())
if len(startItems) and startItems[0] == self.line:
startItems.pop(0)
self.dumpCursorStack()
QtGui.QApplication.setOverrideCursor(
QtGui.QCursor(QtCore.Qt.ForbiddenCursor))
if len(startItems):
if isinstance(startItems[0], Port):
self.dumpCursorStack()
QtGui.QApplication.setOverrideCursor(
QtGui.QCursor(QtCore.Qt.CrossCursor))
def startLineDraw(self, event):
# highlight all ports that match
startItems = self.items(event.scenePos())
if len(startItems):
if isinstance(startItems[0], OutPort):
if QtCore.Qt.AltModifier == getKeyboardModifiers():
startItems[0].toggleMemSaver()
return
self.portMatches = startItems[0].findMatchingInPorts()
for port in self.portMatches:
port.scaleUp()
self.graph.viewAndSceneForcedUpdate()
if isinstance(startItems[0], InPort):
self.portMatches = startItems[0].findMatchingOutPorts()
for port in self.portMatches:
port.scaleUp()
self.graph.viewAndSceneForcedUpdate()
self.line = QtGui.QGraphicsLineItem(
QtCore.QLineF(event.scenePos(), event.scenePos()))
fade = QtGui.QColor(QtCore.Qt.red)
fade.setAlpha(150)
self.line.setPen(QtGui.QPen(fade, 2))
self.line.setZValue(10)
self.addItem(self.line)
def adjust(self):
if not self.source or not self.dest:
return
# position of pipe end based on port type
bindout_y = 5
bindin_y = 0
if isinstance(self.source, InPort):
line = QtCore.QLineF(self.mapFromItem(self.source, 3.5, bindin_y),
self.mapFromItem(self.dest, 3.5, bindout_y))
else:
line = QtCore.QLineF(self.mapFromItem(self.source, 3.5, bindout_y),
self.mapFromItem(self.dest, 3.5, bindin_y))
self.prepareGeometryChange()
self.sourcePoint = line.p1()
self.destPoint = line.p2()
def adjust(self):
if not self.source or not self.dest:
return
# line = QtCore.QLineF(self.mapFromItem(self.source, 0, 0)+self.source.line().p1(),
# self.mapFromItem(self.dest, 0, 0)+self.dest.line().p1())
line = QtCore.QLineF(self.mapFromItem(self.source, 0, 0),
self.mapFromItem(self.dest, 0, 0))
self.prepareGeometryChange()
self.sourcePoint = line.p1()
self.destPoint = line.p2()
def __init__(self, graph, destPort, sourcePort):
super(EdgeTracer, self).__init__()
# show a faux copy of the delete menu
menu = QtWidgets.QMenu()
menu.addAction("Delete")
# position of pipe end based on port type
bindout_y = 5
bindin_y = -1
p1 = self.mapFromItem(sourcePort, 3.5, bindin_y)
p2 = self.mapFromItem(destPort, 3.5, bindout_y)
pos = graph.mapToGlobal(graph.mapFromScene((p1 - p2) / 2 + p2))
# render the menu without executing it
try:
# PyQt4
menupixmap = QtGui.QPixmap().grabWidget(menu)
except AttributeError:
menupixmap = menu.grab() # QtGui.QPixmap().grabWidget(menu)
# round edges
#mask = menupixmap.createMaskFromColor(QtGui.QColor(255, 255, 255), QtCore.Qt.MaskOutColor)
#p = QtGui.QPainter(menupixmap)
#p.setRenderHint(QtGui.QPainter.Antialiasing)
#p.drawRoundedRect(0,0,menupixmap.width(),menupixmap.height(), 5,5)
#p.drawPixmap(menupixmap.rect(), mask, mask.rect())
#p.end()
# display the menu image (as a dummy menu as its being built)
# TODO: this could probably be moved to the FauxMenu
self._tracer = QtWidgets.QLabel()
self._tracer.setWindowFlags(QtCore.Qt.Tool
| QtCore.Qt.FramelessWindowHint)
self._tracer.move(pos)
self._tracer.setPixmap(menupixmap)
self._tracer.show()
self._tracer.raise_()
# draw a faux selected line
line = QtCore.QLineF(p1, p2)
self.setPen(
QtGui.QPen(QtGui.QColor(QtCore.Qt.red), 2, QtCore.Qt.DashLine,
QtCore.Qt.RoundCap, QtCore.Qt.RoundJoin))
self.setLine(line)
self.setZValue(0)
# cleanup both menu item and line by removing from scene (parent).
self._timer = QtCore.QTimer()
self._timer.singleShot(300, lambda: graph.scene().removeItem(self))
def paint(self, painter, option, widget): # EDGE
if not self.source or not self.dest:
return
# Draw the line itself.
line = QtCore.QLineF(self.sourcePoint, self.destPoint)
if line.length() == 0.0:
return
if self.isSelected(
) or self._beingHovered or self.connectedPortIsHovered():
fade = QtGui.QColor(QtCore.Qt.red)
fade.setAlpha(200)
#painter.setPen(QtGui.QPen(QtCore.Qt.red, 1, QtCore.Qt.DashLine,
painter.setPen(
QtGui.QPen(fade, 2, QtCore.Qt.SolidLine, QtCore.Qt.RoundCap,
QtCore.Qt.RoundJoin))
elif self.isCyclicConnection():
painter.setPen(
QtGui.QPen(QtCore.Qt.red, 2, QtCore.Qt.SolidLine,
QtCore.Qt.RoundCap, QtCore.Qt.RoundJoin))
else:
fade = QtGui.QColor(QtCore.Qt.black)
fade.setAlpha(150)
#painter.setPen(QtGui.QPen(QtCore.Qt.black, 2, QtCore.Qt.SolidLine,
painter.setPen(
QtGui.QPen(fade, 2, QtCore.Qt.SolidLine, QtCore.Qt.RoundCap,
QtCore.Qt.RoundJoin))
painter.drawLine(line)
x = (line.x1() + line.x2()) / 2.0
y = (line.y1() + line.y2()) / 2.0
xa = (line.x1() - line.x2())
ya = (line.y1() - line.y2())
m = math.sqrt(xa * xa + ya * ya)
a = math.atan2(ya, xa) * 180.0 / math.pi
buf = self.source.getDataString()
if self._beingHovered:
f = QtGui.QFont("Times New Roman", 8)
else:
f = QtGui.QFont("Times New Roman", 6)
fm = QtGui.QFontMetricsF(f)
bw = fm.width(buf)
bw2 = -bw * 0.5
#bh = fm.height()
# bezier curves
if False:
sa = (a + 90.) * 0.5
path = QtGui.QPainterPath(line.p1())
path.cubicTo(x - sa, y - sa, x + sa, y + sa, line.x2(), line.y2())
painter.drawPath(path)
# bezier curves, change direction on the angle
if False:
sa = (a + 90.) * 0.5
if a > 90 or a < -90:
path = QtGui.QPainterPath(line.p1())
path.cubicTo(x - sa, y - sa, x + sa, y + sa, line.x2(),
line.y2())
painter.drawPath(path)
else:
path = QtGui.QPainterPath(line.p1())
path.cubicTo(x + sa, y + sa, x - sa, y - sa, line.x2(),
line.y2())
painter.drawPath(path)
painter.setFont(f)
if self._beingHovered:
painter.setPen(QtGui.QPen(QtCore.Qt.red, 1))
else:
painter.setPen(QtGui.QPen(QtCore.Qt.darkGray, 1))
painter.save()
painter.translate(QtCore.QPointF(x, y))
if m > bw * 1.1 or self._beingHovered:
if a > 90 or a < -90:
painter.rotate(a + 180.0)
painter.drawText(QtCore.QPointF(bw2, -2.0), buf)
else:
painter.rotate(a)
painter.drawText(QtCore.QPointF(bw2, -2.0), buf)
else:
painter.drawText(QtCore.QPointF(bw2, -2.0), '')
painter.restore()
def paint(self, painter, option, widget):
if not self.source or not self.dest:
return
# Draw the line itself.
line = QtCore.QLineF(self.sourcePoint, self.destPoint)
if line.length() == 0.0:
return
painter.setPen(
QtGui.QPen(QtCore.Qt.gray, self.penWidth, QtCore.Qt.SolidLine,
QtCore.Qt.RoundCap, QtCore.Qt.RoundJoin))
painter.drawLine(line)
# drawing text
x = (line.x1() + line.x2()) / 2.0
y = (line.y1() + line.y2()) / 2.0
xa = (line.x1() - line.x2())
ya = (line.y1() - line.y2())
m = math.sqrt(xa * xa + ya * ya)
a = math.atan2(ya, xa) * 180.0 / math.pi
buf = "Macro"
f = QtGui.QFont("times", 20)
fm = QtGui.QFontMetricsF(f)
bw = fm.width(buf)
bw2 = -bw * 0.5
# bh = fm.height()
# Draw the arrows if there's enough room.
angle = math.acos(line.dx() / line.length())
if line.dy() >= 0:
angle = self.TwoPi - angle
sourceArrowP1 = self.sourcePoint + QtCore.QPointF(
math.sin(angle + self.Pi / 3) * self.arrowSize,
math.cos(angle + self.Pi / 3) * self.arrowSize)
sourceArrowP2 = self.sourcePoint + QtCore.QPointF(
math.sin(angle + self.Pi - self.Pi / 3) * self.arrowSize,
math.cos(angle + self.Pi - self.Pi / 3) * self.arrowSize)
destArrowP1 = self.destPoint + QtCore.QPointF(
math.sin(angle - self.Pi / 3) * self.arrowSize,
math.cos(angle - self.Pi / 3) * self.arrowSize)
destArrowP2 = self.destPoint + QtCore.QPointF(
math.sin(angle - self.Pi + self.Pi / 3) * self.arrowSize,
math.cos(angle - self.Pi + self.Pi / 3) * self.arrowSize)
painter.setBrush(QtCore.Qt.gray)
painter.drawPolygon(
QtGui.QPolygonF([line.p1(), sourceArrowP1, sourceArrowP2]))
painter.drawPolygon(
QtGui.QPolygonF([line.p2(), destArrowP1, destArrowP2]))
# drawing text
painter.setFont(f)
painter.setPen(QtGui.QPen(QtCore.Qt.darkGray, 1))
painter.save()
painter.translate(QtCore.QPointF(x, y))
if m > bw * 1.1:
if a > 90 or a < -90:
painter.rotate(a + 180.0)
painter.drawText(QtCore.QPointF(bw2, -5.0), buf)
else:
painter.rotate(a)
painter.drawText(QtCore.QPointF(bw2, -5.0), buf)
else:
painter.drawText(QtCore.QPointF(bw2, -5.0), '')
painter.restore()
