1 files changed, 36 insertions, 0 deletions
diff --git a/src/sessiontreemodel.h b/src/sessiontreemodel.h
new file mode 100644
index 0000000..31611cb
--- /dev/null
+++ b/src/sessiontreemodel.h
@@ -0,0 +1,36 @@
+#ifndef VMCHOOSER_SESSIONTREEMODEL_H
+#define VMCHOOSER_SESSIONTREEMODEL_H
+
+#include <QAbstractItemModel>
+#include <QModelIndex>
+#include <QVariant>
+
+class SessionTreeItem;
+class Session;
+
+class SessionTreeModel : public QAbstractItemModel
+{
+    Q_OBJECT
+
+public:
+    SessionTreeModel(QObject *parent = 0);
+    ~SessionTreeModel();
+
+    QVariant data(const QModelIndex &index, int role) const;
+    Qt::ItemFlags flags(const QModelIndex &index) const;
+    QVariant headerData(int section, Qt::Orientation orientation,
+                        int role = Qt::DisplayRole) const;
+    QModelIndex index(int row, int column,
+                      const QModelIndex &parent = QModelIndex()) const;
+    QModelIndex parent(const QModelIndex &index) const;
+    int rowCount(const QModelIndex &parent = QModelIndex()) const;
+    int columnCount(const QModelIndex &parent = QModelIndex()) const;
+
+    void addItems(const QList<Session*>& sessions, const QString& section);
+
+
+private:
+    SessionTreeItem* root_;
+};
+
+#endif // VMCHOOSER_SESSIONTREEMODEL_H