1 files changed, 40 insertions, 0 deletions
diff --git a/src/data/nodestylelist.h b/src/data/nodestylelist.h
new file mode 100644
index 0000000..03f7ed5
--- /dev/null
+++ b/src/data/nodestylelist.h
@@ -0,0 +1,40 @@
+#ifndef NODESTYLELIST_H
+#define NODESTYLELIST_H
+
+#include "nodestyle.h"
+
+#include <QAbstractListModel>
+
+class NodeStyleList : public QAbstractListModel
+{
+    Q_OBJECT
+public:
+    explicit NodeStyleList(QObject *parent = nullptr);
+    NodeStyle *style(QString name);
+    NodeStyle *style(int i);
+    int length() const;
+    void addStyle(NodeStyle *s);
+    void clear();
+    QString tikz();
+
+    int numInCategory() const;
+    int nthInCategory(int n) const;
+    NodeStyle *styleInCategory(int n) const;
+
+    QVariant data(const QModelIndex &index, int role) const override;
+    int rowCount(const QModelIndex &/*parent*/) const override;
+
+
+    QString category() const;
+    void setCategory(const QString &category);
+
+signals:
+
+public slots:
+
+private:
+    QVector<NodeStyle*> _styles;
+    QString _category;
+};
+
+#endif // NODESTYLELIST_H