1 files changed, 40 insertions, 3 deletions

diff --git a/tikzit/graphelementdata.h b/tikzit/graphelementdata.h
index e67d740..fee65e7 100644
--- a/tikzit/graphelementdata.h
+++ b/tikzit/graphelementdata.h
@@ -1,24 +1,61 @@
 #ifndef GRAPHELEMENTDATA_H
 #define GRAPHELEMENTDATA_H
 
-#include <QObject>
+#include "graphelementproperty.h"
+
+#include <QAbstractItemModel>
 #include <QString>
+#include <QVariant>
+#include <QModelIndex>
+#include <QVector>
 
-class GraphElementData : public QObject
+class GraphElementData : public QAbstractItemModel
 {
     Q_OBJECT
 public:
     explicit GraphElementData(QObject *parent = 0);
+    ~GraphElementData();
     void setProperty(QString key, QString value);
     void unsetProperty(QString key);
     void setAtom(QString atom);
     void unsetAtom(QString atom);
     QString property(QString key);
-    QString atom(QString atom);
+    bool atom(QString atom);
+
+    QVariant data(const QModelIndex &index, int role) const Q_DECL_OVERRIDE;
+    QVariant headerData(int section, Qt::Orientation orientation,
+                        int role = Qt::DisplayRole) const Q_DECL_OVERRIDE;
+
+    QModelIndex index(int row, int column,
+                      const QModelIndex &parent = QModelIndex()) const Q_DECL_OVERRIDE;
+    QModelIndex parent(const QModelIndex &index) const Q_DECL_OVERRIDE;
+
+    int rowCount(const QModelIndex &parent = QModelIndex()) const Q_DECL_OVERRIDE;
+    int columnCount(const QModelIndex &parent = QModelIndex()) const Q_DECL_OVERRIDE;
+
+    Qt::ItemFlags flags(const QModelIndex &index) const Q_DECL_OVERRIDE;
+
+//    bool setData(const QModelIndex &index, const QVariant &value,
+//                 int role = Qt::EditRole) Q_DECL_OVERRIDE;
+//    bool setHeaderData(int section, Qt::Orientation orientation,
+//                       const QVariant &value, int role = Qt::EditRole) Q_DECL_OVERRIDE;
+
+//    bool insertColumns(int position, int columns,
+//                       const QModelIndex &parent = QModelIndex()) Q_DECL_OVERRIDE;
+//    bool removeColumns(int position, int columns,
+//                       const QModelIndex &parent = QModelIndex()) Q_DECL_OVERRIDE;
+//    bool insertRows(int position, int rows,
+//                    const QModelIndex &parent = QModelIndex()) Q_DECL_OVERRIDE;
+//    bool removeRows(int position, int rows,
+//                    const QModelIndex &parent = QModelIndex()) Q_DECL_OVERRIDE;
 
 signals:
 
 public slots:
+
+private:
+    QVector<GraphElementProperty> _properties;
+    GraphElementProperty *root;
 };
 
 #endif // GRAPHELEMENTDATA_H