1 files changed, 25 insertions, 0 deletions
diff --git a/mappingtreewidget.h b/mappingtreewidget.h
index 734d943..7050743 100644
--- a/mappingtreewidget.h
+++ b/mappingtreewidget.h
@@ -18,6 +18,7 @@ class QPushButton;
 class QSortFilterProxyModel;
 class QStringListModel;
 class QAction;
+class MappingTreeResultModel;
 
 struct MappingData;
 
@@ -69,4 +70,28 @@ class MappingTreeView : public QTreeView {
         virtual void contextMenuEvent(QContextMenuEvent *e);
 };
 
+class MappingTreeResultView : public QTreeView {
+    Q_OBJECT
+    public:
+        explicit MappingTreeResultView(QWidget *parent = 0);
+};
+
+class MappingEditWidget : public QWidget {
+    Q_OBJECT
+    public:
+        explicit MappingEditWidget(QWidget *parent = 0);
+        QList<int> mappingIds() const;
+
+    public slots:
+        void addMapping();
+        void removeMapping();
+
+    private:
+        MappingTreeWidget *mMappingTree;
+        MappingTreeResultView *mMappingResult;
+        MappingTreeResultModel *mResultModel;
+        QPushButton *mAddMapping;
+        QPushButton *mRemoveMapping;
+};
+
 #endif // MAPPINGTREEWIDGET_H