1 files changed, 12 insertions, 1 deletions
diff --git a/Project/ConnectionLine.h b/Project/ConnectionLine.h
index 8249df8..cc7b51e 100644
--- a/Project/ConnectionLine.h
+++ b/Project/ConnectionLine.h
@@ -6,12 +6,23 @@
 class ConnectionLine : public ControlElement
 {
 public:
-    ConnectionLine();
+    ConnectionLine(Node* firstNode);
     ~ConnectionLine();
     
     virtual void Draw(wxPoint2DDouble translation, double scale) const;
     virtual bool Contains(wxPoint2DDouble position) const;
     virtual bool Intersects(wxRect2DDouble rect) const;
+    virtual void StartMove(wxPoint2DDouble position);
+    virtual void Move(wxPoint2DDouble position);
+    virtual bool AppendNode(Node* node, ControlElement* parent);
+    virtual void UpdatePoints();
+    virtual void SetTemporarySecondPoint(wxPoint2DDouble point) { m_tmpSndPt = point; };
+    
+protected:
+    double m_lineOffset = 0.0;
+    double m_moveStartPtY = 0.0;
+    double m_moveStartOffset = 0.0;
+    wxPoint2DDouble m_tmpSndPt;
 };
 
 #endif // CONNECTIONLINE_H