1 files changed, 9 insertions, 7 deletions
diff --git a/mappingtreemodel.h b/mappingtreemodel.h
index 38c0c7b..a4a1ae8 100644
--- a/mappingtreemodel.h
+++ b/mappingtreemodel.h
@@ -96,16 +96,16 @@ class MappingTreeModel : public SmTreeModel {
 class MappingTreeResultModel : public SmTreeModel {
     Q_OBJECT
     public:
-        enum Roles { NameRole = Qt::UserRole + 1, IdRole = Qt::UserRole + 2 };
-        enum Fields { Name = 0, Id = 1 };
-        enum { NumFields = 2 };
+        enum Roles { NameRole = Qt::UserRole + 1, MappingIdRole = Qt::UserRole + 2, ParentIdRole = Qt::UserRole + 3, MyIdRole = Qt::UserRole + 4 };
+        enum Fields { Name = 0, MappingId = 1, ParentId = 2, MyId = 3 };
+        enum { NumFields = 4 };
         explicit MappingTreeResultModel(const QStringList &headers, QObject *parent = 0);
 
         //data + flags
         Qt::ItemFlags flags(const QModelIndex &index) const;
         bool setData(const QModelIndex &index, const QVariant &value, int role);
-        QModelIndex insertChild(const QVariant &data, int id, SmTreeItem *parent);
-        QList<int> mappingsIds() const;
+        QModelIndex insertChild(const QVariant &data, int mappingId, int parentId, int myId, SmTreeItem *parent);
+        QList<QVariant> columnValues(int column) const;
         void clearData();
 
     public slots:
@@ -113,11 +113,13 @@ class MappingTreeResultModel : public SmTreeModel {
 
     private:
         int hasChild(SmTreeItem *item, const QVariant &name, int column = 0) const;
-        QList<int> mappingIdsRecursive(SmTreeItem *parent) const;
+        QList<QVariant> columnValuesRecursive(SmTreeItem *parent, int column) const;
 };
 
 struct MappingData {
-    int id;
+    int mappingId;
+    int parentId;
+    int myId;
     QString name;
     QList<QStringList> path;
 };