1 files changed, 5 insertions, 0 deletions
diff --git a/mappingtreemodel.h b/mappingtreemodel.h
index fa2cfef..60c7bd4 100644
--- a/mappingtreemodel.h
+++ b/mappingtreemodel.h
@@ -40,6 +40,8 @@ class MappingTreeModel : public SmTreeModel {
         bool addChild(const QVariant &name, const QModelIndex &parent);
         bool deleteChild(const QModelIndex &idx);
         int childCount(const QModelIndex &idx) const;
+        MappingData mappingDataFromId(int mappingId) const;
+        void setSelectedMappings(const QList<int> &mappingData);
 
     public slots:
         void populate();
@@ -57,6 +59,7 @@ class MappingTreeModel : public SmTreeModel {
         void getMappingTypes();
         void getChildrenRecursive(SmTreeItem *item);
         int lowerBound(SmTreeItem *item, const QVariant &value, int column = 0) const;
+        QList<MappingData> mappingData(SmTreeItem *item);
         QSqlDatabase mDb;
         QSqlQuery *mTypesQ;
         QSqlQuery *mTypeParentsQ;
@@ -69,6 +72,8 @@ class MappingTreeModel : public SmTreeModel {
         QSqlQuery *mAddParentQ;
         QSqlQuery *mDeleteChildQ;
         QList<mappingType> mMappingTypes;
+        QList<MappingData> mValidMappings;
+        QList<int> mSelectedMappings;
         int mType;
 
 };