1
2
3
4
5
6
7
8
9
10
11
12
13
14
15
16
17
18
19
20
21
22
23
24
25
26
27
28
29
30
31
32
33
34
35
36
37
38
39
40
41
42
43
44
45
46
47
48
49
50
51
52
53
54
55
56
57
58
59
60
61
62
63
64
65
66
67
68
69
70
71
72
|
from PyQt5.QtGui import QFont, QPen, QPainterPath
from PyQt5.QtWidgets import QGraphicsLineItem, QLineEdit, QGraphicsProxyWidget, QGraphicsItem
from PyQt5.QtCore import Qt, QPointF, QRectF
class Line(QGraphicsItem):
def __init__(self, startPoint, endPoint, **args):
QGraphicsItem.__init__(self, **args)
self.startPoint = startPoint
self.endPoint = endPoint
self.startGripItem = None
self.endGripItem = None
self.setFlag(QGraphicsItem.ItemIsSelectable, True)
# self.setFlag(QGraphicsItem.ItemIsMovable, True)
self.setFlag(QGraphicsItem.ItemSendsGeometryChanges, True)
self.setAcceptHoverEvents(True)
def setStartGripItem(self, item):
self.startGripItem = item
def setEndGripItem(self, item):
self.endGripItem = item
def shape(self):
x0, y0 = self.startPoint.x(), self.startPoint.y()
x1, y1 = self.endPoint.x(), self.endPoint.y()
path = QPainterPath(QPointF(x0,y0))
path.lineTo((x0 + x1) / 2, y0)
path.moveTo((x0 + x1) / 2, y0)
path.lineTo((x0 + x1) / 2, y1)
path.moveTo((x0 + x1) / 2, y1)
path.lineTo(x1, y1)
return path
def boundingRect(self):
x0, y0 = self.startPoint.x(), self.startPoint.y()
x1, y1 = self.endPoint.x(), self.endPoint.y()
return QRectF(min(x0, x1), min(y0, y1), abs(x1 - x0), abs(y1 - y0))
def paint(self, painter, style, widget=None):
# x0, y0 = self.startPoint.x(), self.startPoint.y()
# x1, y1 = self.endPoint.x(), self.endPoint.y()
# painter.drawLine(x0, y0, (x0 + x1) / 2, y0)
# painter.drawLine((x0 + x1) / 2, y0, (x0 + x1) / 2, y1)
# painter.drawLine((x0 + x1) / 2, y1, x1, y1)
painter.drawPath(self.shape())
def updateLine(self, startPoint=None, endPoint=None):
"""This function is used to update connecting line when it add on
canvas and when it's grip item moves
:return:
"""
if self.startGripItem and self.endGripItem:
item = self.startGripItem
self.startPoint = item.parentItem().mapToScene(item.pos())
item = self.endGripItem
self.endPoint = item.parentItem().mapToScene(item.pos())
if startPoint:
self.startPoint = startPoint
if endPoint:
self.endPoint = endPoint
self.prepareGeometryChange()
self.update(self.boundingRect())
def removeFromCanvas(self):
"""This function is used to remove connecting line from canvas
:return:
"""
if self.scene():
self.scene().removeItem(self)
|
