1 files changed, 12 insertions, 0 deletions

diff --git a/newmoviewizard.h b/newmoviewizard.h
index 9f476ed..93da626 100644
--- a/newmoviewizard.h
+++ b/newmoviewizard.h
@@ -24,16 +24,26 @@ class QCheckBox;
 class QStringListModel;
 class WizardTreeModel;
 class SmTreeItem;
+class MovieInfoPage;
+class MovieMappingPage;
 
 class NewMovieWizard : public QWizard {
 	Q_OBJECT
 	public:
 		explicit NewMovieWizard(QWidget *parent = 0);
 		void accept();
+		MovieInfoPage *infoPage() { return mInfoPage; }
+		MovieMappingPage *actorPage() { return mActorPage; }
+		MovieMappingPage *genrePage() { return mGenrePage; }
 
 	signals:
 		void seriesAdded(const QString series, int seriesPart);
 
+	private:
+		MovieInfoPage *mInfoPage;
+		MovieMappingPage *mActorPage;
+		MovieMappingPage *mGenrePage;
+
 };
 
 class MovieInfoPage : public QWizardPage {
@@ -42,6 +52,7 @@ class MovieInfoPage : public QWizardPage {
 		explicit MovieInfoPage(QWidget *parent = 0);
 		WizardTreeModel *model() { return mFileModel; }
 		virtual void initializePage();
+		void addFile(const QString &file);
 
 	private slots:
 		void addFiles();
@@ -54,6 +65,7 @@ class MovieInfoPage : public QWizardPage {
 		void fileSelectionChanged(const QModelIndex &current, const QModelIndex &previous);
 
 	private:
+		void setupGui();
 		QTreeView *mFileView;
 		QLineEdit *mTitle;
 		QSpinBox *mSeriesNo;