1 files changed, 14 insertions, 2 deletions
diff --git a/newmoviewizard.h b/newmoviewizard.h
index e28a3a9..a0f80b4 100644
--- a/newmoviewizard.h
+++ b/newmoviewizard.h
@@ -13,15 +13,17 @@
 #include <QHash>
 
 #include "smtreemodel.h"
+#include "mappingtablewidget.h"
 
 class QTreeView;
-class WizardTreeModel;
-class SmTreeItem;
 class QLineEdit;
 class QSpinBox;
 class QPushButton;
 class QComboBox;
 class QCheckBox;
+class QStringListModel;
+class WizardTreeModel;
+class SmTreeItem;
 
 class NewMovieWizard : public QWizard {
 	Q_OBJECT
@@ -61,6 +63,16 @@ class MovieInfoPage : public QWizardPage {
 		SmTreeItem *mCoversItem;
 };
 
+class MovieMappingPage : public QWizardPage {
+	Q_OBJECT
+	public:
+		explicit MovieMappingPage(const QString &table, QWidget *parent = 0);
+		MappingTableItemModel *model() { return mWidget->model(); }
+
+	private:
+		MappingTableWidget *mWidget;
+};
+
 class WizardTreeModel : public SmTreeModel {
 	Q_OBJECT
 	public: