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 shape(self):
path = QtGui.QPainterPath()
# path.addEllipse(0, 0, 5, 5)
path.addPolygon(self.portShape)
self.update()
return path
