1 files changed, 18 insertions, 0 deletions
diff --git a/newpicsdialog.h b/newpicsdialog.h
index de4275a..9a11675 100644
--- a/newpicsdialog.h
+++ b/newpicsdialog.h
@@ -10,6 +10,7 @@
 
 #include <QWidget>
 #include <QSqlDatabase>
+#include <QTreeView>
 
 #include "smtreemodel.h"
 
@@ -20,6 +21,9 @@ class QTreeView;
 class NewPicFilesModel;
 class QSortFilterProxyModel;
 class QPushButton;
+class MappingTreeWidget;
+class MappingTreeResultView;
+class MappingTreeResultModel;
 
 struct FileData;
 
@@ -31,15 +35,29 @@ class NewPicsDialog : public QWidget {
     public slots:
         void addFiles();
         void removeFiles();
+        void addMapping();
+        void removeMapping();
 
     private:
         QTabWidget *mTab;
         QWidget *mFilesWidget;
+        QWidget *mMappingWidget;
         QTreeView *mFilesV;
         NewPicFilesModel *mFilesModel;
         QSortFilterProxyModel *mFilesProxy;
         QPushButton *mAddFiles;
         QPushButton *mRemoveFiles;
+        MappingTreeWidget *mMappingTreeWidget;
+        QPushButton *mAddMapping;
+        QPushButton *mRemoveMapping;
+        MappingTreeResultView *mResultView;
+        MappingTreeResultModel *mResultModel;
+};
+
+class MappingTreeResultView : public QTreeView {
+    Q_OBJECT
+    public:
+        explicit MappingTreeResultView(QWidget *parent = 0);
 };
 
 class NewPicFilesModel : public SmTreeModel {