1 files changed, 8 insertions, 1 deletions
diff --git a/newmoviewizard.h b/newmoviewizard.h
index a92d91d..e28a3a9 100644
--- a/newmoviewizard.h
+++ b/newmoviewizard.h
@@ -21,11 +21,13 @@ class QLineEdit;
 class QSpinBox;
 class QPushButton;
 class QComboBox;
+class QCheckBox;
 
 class NewMovieWizard : public QWizard {
 	Q_OBJECT
 	public:
 		explicit NewMovieWizard(QWidget *parent = 0);
+		void accept();
 
 };
 
@@ -33,11 +35,14 @@ class MovieInfoPage : public QWizardPage {
 	Q_OBJECT
 	public:
 		explicit MovieInfoPage(QWidget *parent = 0);
+		WizardTreeModel *model() { return mFileModel; }
 
 	private slots:
 		void addFiles();
 		void removeFile();
+		void onDvd(int);
 		void typeChanged(QString);
+		void setNextDvdNo();
 
 	private:
 		QTreeView *mFileView;
@@ -49,6 +54,8 @@ class MovieInfoPage : public QWizardPage {
 		QPushButton *mAddFile;
 		QPushButton *mRemoveFile;
 		QComboBox *mFileType;
+		QCheckBox *mOnDvd;
+		QPushButton *mNextDvdNo;
 		WizardTreeModel *mFileModel;
 		SmTreeItem *mMoviesItem;
 		SmTreeItem *mCoversItem;
@@ -58,7 +65,7 @@ class WizardTreeModel : public SmTreeModel {
 	Q_OBJECT
 	public:
 		enum CustomRoles { FileNameRole = Qt::UserRole + 1, FileSizeRole = Qt::UserRole + 2, FileTypeRole = Qt::UserRole + 3, FullPathRole = Qt::UserRole + 4 };
-		enum Fields { FileName = 0, FileSize = 1, FileType = 3, FullPath = 4 };
+		enum Fields { FileName = 0, FileSize = 1, FileType = 2, FullPath = 3 };
 		enum Types { Movie = 1, FrontCover = 2, BackCover = 3, GeneralCover = 4 };
 		explicit WizardTreeModel(QStringList &headers, QObject *parent = 0);
 		virtual ~WizardTreeModel() {}