summaryrefslogtreecommitdiffstats
path: root/newmoviewizard.h
blob: abcf977184127b9735c2e5ffe631d2097f0ffaf0 (plain)
1
2
3
4
5
6
7
8
9
10
11
12
13
14
15
16
17
18
19
20
21
22
23
24
25
26
27
28
29
30
31
32
33
34
35
36
37
38
39
40
41
42
43
44
45
46
47
48
49
50
51
52
53
54
55
56
57
58
59
60
61
62
63
64
65
66
67
68
69
70
71
72
73
74
75
76
77
78
79
80
81
82
83
84
85
86
87
88
89
90
91
92
93
94
95
96
97
98
99
100
101
102
103
104
105
106
107
108
109
110
111
112
113
114
115
116
117
118
119
120
121
122
123
124
125
/*
  This program is free software; you can redistribute it and/or
  modify it under the terms of the GNU General Public License
  as published by the Free Software Foundation; either version
  2 of the License, or (at your option) any later version.
*/

#ifndef NEWMOVIEWIZARD_H
#define NEWMOVIEWIZARD_H

#include <QWizard>
#include <QWizardPage>
#include <QHash>

#include "smtreemodel.h"
#include "mappingtablewidget.h"

class QTreeView;
class QLineEdit;
class QSpinBox;
class QPushButton;
class QComboBox;
class QCheckBox;
class QStringListModel;
class WizardTreeModel;
class SmTreeItem;
class MovieInfoPage;
class MovieMappingPage;
class PictureViewer;

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 {
	Q_OBJECT
	public:
		explicit MovieInfoPage(QWidget *parent = 0);
		WizardTreeModel *model() { return mFileModel; }
		virtual void initializePage();
		void addFile(const QString &file);

	private slots:
		void addFiles();
		void removeFile();
		void onDvd(int);
		void typeChanged(QString);
		void seriesPartChanged(int);
		void setNextDvdNo();
		void initModel();
		void fileSelectionChanged(const QModelIndex &current, const QModelIndex &previous);
		void itemClicked(const QModelIndex &index);

	private:
		void setupGui();
		QTreeView *mFileView;
		QLineEdit *mTitle;
		QSpinBox *mSeriesNo;
		QSpinBox *mPartno;
		QSpinBox *mQuality;
		QSpinBox *mDvdNo;
		QPushButton *mAddFile;
		QPushButton *mRemoveFile;
		QComboBox *mFileType;
		QCheckBox *mOnDvd;
		QCheckBox *mUsePicViewer;
		QPushButton *mNextDvdNo;
		WizardTreeModel *mFileModel;
		SmTreeItem *mMoviesItem;
		SmTreeItem *mCoversItem;
		PictureViewer *mPicViewer;
};

class MovieMappingPage : public QWizardPage {
	Q_OBJECT
	public:
		explicit MovieMappingPage(const QString &table, QWidget *parent = 0);
		MappingTableItemModel *model() { return mWidget->model(); }
		virtual void initializePage();

	private:
		MappingTableWidget *mWidget;
};

class WizardTreeModel : public SmTreeModel {
	Q_OBJECT
	public:
		enum CustomRoles { FileNameRole = Qt::UserRole + 1, FileSizeRole = Qt::UserRole + 2, FileTypeRole = Qt::UserRole + 3, FilePartRole = Qt::UserRole + 4, FullPathRole = Qt::UserRole + 5 };
		enum Fields { FileName = 0, FileSize = 1, FileType = 2, FilePart = 3, FullPath = 4 };
		enum Types { Movie = 1, FrontCover = 2, BackCover = 3, GeneralCover = 4 };
		explicit WizardTreeModel(QStringList &headers, QObject *parent = 0);
		virtual ~WizardTreeModel() {}

		//data + flags
		virtual QVariant data(const QModelIndex &index, int role) const;
		virtual Qt::ItemFlags flags(const QModelIndex &index) const;
		QHash<QString, int> files();
		int filesPart(const QString &fullPath) const;

		//file types
		QStringList types() const;
		int typeId(const QString &value) const;
		QString typeName(int typeId) const;

	private:
		QHash<int, QString> mFileTypeMap;
		QHash<QString, int> mFilePartMap;
};

#endif