1 files changed, 113 insertions, 0 deletions
diff --git a/src/data/nodestylelist.cpp b/src/data/nodestylelist.cpp
new file mode 100644
index 0000000..41749db
--- /dev/null
+++ b/src/data/nodestylelist.cpp
@@ -0,0 +1,113 @@
+#include "nodestylelist.h"
+
+#include <QTextStream>
+
+NodeStyleList::NodeStyleList(QObject *parent) : QAbstractListModel(parent)
+{
+}
+
+NodeStyle *NodeStyleList::style(QString name)
+{
+    foreach (NodeStyle *s, _styles)
+        if (s->name() == name) return s;
+    return nullptr;
+}
+
+NodeStyle *NodeStyleList::style(int i)
+{
+    return _styles[i];
+}
+
+int NodeStyleList::length() const
+{
+    return _styles.length();
+}
+
+void NodeStyleList::addStyle(NodeStyle *s)
+{
+    if (s->category() == _category) {
+        int n = numInCategory();
+        beginInsertRows(QModelIndex(), n, n);
+        _styles << s;
+        endInsertRows();
+    } else {
+        _styles << s;
+    }
+}
+
+void NodeStyleList::clear()
+{
+    int n = numInCategory();
+    if (n > 0) {
+        beginRemoveRows(QModelIndex(), 0, n - 1);
+        _styles.clear();
+        endRemoveRows();
+    } else {
+        _styles.clear();
+    }
+
+    _category = "";
+}
+
+QString NodeStyleList::tikz()
+{
+    QString str;
+    QTextStream code(&str);
+    foreach (NodeStyle *s, _styles) code << s->tikz() << "\n";
+    code.flush();
+    return str;
+}
+
+int NodeStyleList::numInCategory() const
+{
+    int c = 0;
+    foreach (NodeStyle *s, _styles) {
+        if (_category == "" || s->category() == _category) {
+            ++c;
+        }
+    }
+    return c;
+}
+
+int NodeStyleList::nthInCategory(int n) const
+{
+    int c = 0;
+    for (int j = 0; j < _styles.length(); ++j) {
+        if (_category == "" || _styles[j]->category() == _category) {
+            if (c == n) return j;
+            else ++c;
+        }
+    }
+    return -1;
+}
+
+NodeStyle *NodeStyleList::styleInCategory(int n) const
+{
+    return _styles[nthInCategory(n)];
+}
+
+QVariant NodeStyleList::data(const QModelIndex &index, int role) const
+{
+    if (role == Qt::DisplayRole) {
+        return QVariant(styleInCategory(index.row())->name());
+    } else if (role == Qt::DecorationRole) {
+        return QVariant(styleInCategory(index.row())->icon());
+    } else {
+        return QVariant();
+    }
+}
+
+int NodeStyleList::rowCount(const QModelIndex &/*parent*/) const
+{
+    return numInCategory();
+}
+
+QString NodeStyleList::category() const
+{
+    return _category;
+}
+
+void NodeStyleList::setCategory(const QString &category)
+{
+    _category = category;
+}