1 files changed, 60 insertions, 0 deletions
diff --git a/line.py b/line.py
new file mode 100644
index 0000000..941016d
--- /dev/null
+++ b/line.py
@@ -0,0 +1,60 @@
+from PyQt5.QtGui import QFont, QPen
+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 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)
+
+    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)