1 files changed, 35 insertions, 0 deletions
diff --git a/tikzit/src/data/edge.h b/tikzit/src/data/edge.h
index 9655e98..e392c56 100644
--- a/tikzit/src/data/edge.h
+++ b/tikzit/src/data/edge.h
@@ -5,6 +5,7 @@
 #include "node.h"
 
 #include <QObject>
+#include <QPointF>
 
 class Edge : public QObject
 {
@@ -16,6 +17,9 @@ public:
     Node *source() const;
     Node *target() const;
 
+    bool isSelfLoop();
+    bool isStraight();
+
     GraphElementData *data() const;
     void setData(GraphElementData *data);
 
@@ -29,6 +33,23 @@ public:
     void setEdgeNode(Node *edgeNode);
     bool hasEdgeNode();
 
+    void updateControls();
+    void setAttributesFromData();
+    void updateData();
+
+    QPointF head() const;
+    QPointF tail() const;
+    QPointF cp1() const;
+    QPointF cp2() const;
+    QPointF mid() const;
+
+    int bend() const;
+    int inAngle() const;
+    int outAngle() const;
+    float weight() const;
+    bool basicBendMode() const;
+    float cpDist() const;
+
 signals:
 
 public slots:
@@ -44,6 +65,20 @@ private:
     // referenced
     Node *_source;
     Node *_target;
+
+    bool _dirty;
+    bool _basicBendMode;
+    int _bend;
+    int _inAngle;
+    int _outAngle;
+    float _weight;
+    float _cpDist;
+
+    QPointF _head;
+    QPointF _tail;
+    QPointF _cp1;
+    QPointF _cp2;
+    QPointF _mid;
 };
 
 #endif // EDGE_H