added edgestyles, but cant apply to nodes yet

1 files changed, 4 insertions, 0 deletions

diff --git a/src/data/tikzstyles.h b/src/data/tikzstyles.h
index eaf7e64..4cd7d6e 100644
--- a/src/data/tikzstyles.h
+++ b/src/data/tikzstyles.h
@@ -3,6 +3,7 @@
 
 #include "graphelementdata.h"
 #include "nodestyle.h"
+#include "edgestyle.h"
 
 #include <QObject>
 #include <QString>
@@ -15,7 +16,9 @@ public:
     void addStyle(QString name, GraphElementData *data);
 
     NodeStyle *nodeStyle(QString name) const;
+    EdgeStyle *edgeStyle(QString name) const;
     QVector<NodeStyle *> nodeStyles() const;
+    QVector<EdgeStyle *> edgeStyles() const;
     void clear();
 
 signals:
@@ -24,6 +27,7 @@ public slots:
 
 private:
     QVector<NodeStyle*> _nodeStyles;
+    QVector<EdgeStyle*> _edgeStyles;
 };
 
 #endif // PROJECT_H