1 files changed, 17 insertions, 0 deletions
diff --git a/filestreewidget.h b/filestreewidget.h
index 06f1d3c..b6ce951 100644
--- a/filestreewidget.h
+++ b/filestreewidget.h
@@ -11,12 +11,15 @@
 #include <QWidget>
 #include <QTreeView>
 #include <QSortFilterProxyModel>
+#include <QDialog>
 
 class FilesTreeView;
 class FilesTreeModel;
 class FilesTreeSortModel;
 class SeriesTreeModel;
 class QContextMenuEvent;
+class QSpinBox;
+class QPushButton;
 
 class FilesTreeWidget : public QWidget {
 	Q_OBJECT
@@ -27,6 +30,7 @@ class FilesTreeWidget : public QWidget {
 
 	public slots:
 		void moveToBurn();
+		void setDvdNo();
 
 	private slots:
 		void fileSelectionChanged(const QModelIndex &current, const QModelIndex &previous);
@@ -64,4 +68,17 @@ class FilesTreeSortModel : public QSortFilterProxyModel {
 		virtual bool lessThan(const QModelIndex &left, const QModelIndex &right) const;
 };
 
+class DvdNoDialog : public QDialog {
+	Q_OBJECT
+	public:
+		DvdNoDialog(QWidget *parent = 0, Qt::WindowFlags f = 0);
+		~DvdNoDialog() {}
+		int dvdNo() const;
+
+	private:
+		QSpinBox *mDvdNo;
+		QPushButton *mOk;
+		QPushButton *mCancel;
+};
+
 #endif