parsing

1 files changed, 15 insertions, 0 deletions

diff --git a/tikzit/src/data/edge.cpp b/tikzit/src/data/edge.cpp
index c8d1cfc..67e2061 100644
--- a/tikzit/src/data/edge.cpp
+++ b/tikzit/src/data/edge.cpp
@@ -1,14 +1,18 @@
 #include "edge.h"
 
+#include <QDebug>
+
 Edge::Edge(Node *s, Node *t, QObject *parent) :
     QObject(parent), _source(s), _target(t)
 {
     _data = new GraphElementData();
+    _edgeNode = 0;
 }
 
 Edge::~Edge()
 {
     delete _data;
+    delete _edgeNode;
 }
 
 Node *Edge::source() const
@@ -52,4 +56,15 @@ void Edge::setTargetAnchor(const QString &targetAnchor)
     _targetAnchor = targetAnchor;
 }
 
+Node *Edge::edgeNode() const
+{
+    return _edgeNode;
+}
+
+void Edge::setEdgeNode(Node *edgeNode)
+{
+    if (_edgeNode != 0) delete _edgeNode;
+    _edgeNode = edgeNode;
+}
+