added testing

1 files changed, 15 insertions, 0 deletions

diff --git a/tikzit/src/data/edge.h b/tikzit/src/data/edge.h
index 2153f30..5884b90 100644
--- a/tikzit/src/data/edge.h
+++ b/tikzit/src/data/edge.h
@@ -1,6 +1,8 @@
 #ifndef EDGE_H
 #define EDGE_H
 
+#include "graphelementdata.h"
+
 #include <QObject>
 
 class Node;
@@ -10,10 +12,20 @@ class Edge : public QObject
     Q_OBJECT
 public:
     explicit Edge(Node *s, Node *t, QObject *parent = 0);
+    ~Edge();
 
     Node *source() const;
     Node *target() const;
 
+    GraphElementData *data() const;
+    void setData(GraphElementData *data);
+
+    QString sourceAnchor() const;
+    void setSourceAnchor(const QString &sourceAnchor);
+
+    QString targetAnchor() const;
+    void setTargetAnchor(const QString &targetAnchor);
+
 signals:
 
 public slots:
@@ -21,6 +33,9 @@ public slots:
 private:
     Node *_source;
     Node *_target;
+    GraphElementData *_data;
+    QString _sourceAnchor;
+    QString _targetAnchor;
 };
 
 #endif // EDGE_H