Remodel section handling: Use constants

1 files changed, 9 insertions, 4 deletions

diff --git a/src/sessiontreeitem.h b/src/sessiontreeitem.h
index a95fb45..6202b62 100644
--- a/src/sessiontreeitem.h
+++ b/src/sessiontreeitem.h
@@ -4,13 +4,14 @@
 #include <QList>
 #include <QVariant>
 
-// class ... instead of include?
 #include "session.h"
+#include "globals.h"
 #include <QString>
 
 class SessionTreeItem {
   public:
     SessionTreeItem(const Session* session, SessionTreeItem *parent = 0);
+    SessionTreeItem(const SectionType type, SessionTreeItem *parent = 0);
     SessionTreeItem(const QString& text, SessionTreeItem *parent = 0);
     ~SessionTreeItem();
 
@@ -19,17 +20,21 @@ class SessionTreeItem {
 
     SessionTreeItem *child(int row);
     int childCount() const;
-    int columnCount() const;
+    int columnCount() const { return 1; }
     int row() const;
     SessionTreeItem *parent();
-    const Session* session() const;
-    const QString text() const;
+    const Session* session() const { return session_; }
+    const QString text() const { return text_; }
+    SectionType sectionType() const { return sectionType_; }
 
   private:
     QList<SessionTreeItem*> children_;
     SessionTreeItem *parent_;
     const Session *session_;
+    const SectionType sectionType_;
     const QString text_;
+
+    static QString typeToName(const SectionType type);
 };
 
 #endif // SESSIONTREEITEM_H