Started implementation of FilesTreeModel

No visible representation yet. Implemented setData for quality and
dvdno, but not tested yet. Dunno yet if anything else has to be
editable.

1 files changed, 40 insertions, 0 deletions

diff --git a/filestreemodel.h b/filestreemodel.h
new file mode 100644
index 0000000..0db1f39
--- /dev/null
+++ b/filestreemodel.h
@@ -0,0 +1,40 @@
+/*
+  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 FILESTREEMODEL_H
+#define FILESTREEMODEL_H
+
+#include <QSqlDatabase>
+#include <QList>
+
+#include "smtreemodel.h"
+
+class QSqlQuery;
+
+class FilesTreeModel : public SmTreeModel {
+	Q_OBJECT
+	public:
+		enum CustomRoles { FileNameRole = Qt::UserRole + 1, FullPathRole = Qt::UserRole + 2, SizeRole = Qt::UserRole + 3, DvdNoRole = Qt::UserRole + 4,  SizeDisplayRole = Qt::UserRole + 5, FileTypeRole = Qt::UserRole + 6, Md5SumRole = Qt::UserRole + 7, PartNoRole = Qt::UserRole + 8, SeriesPartIdRole = Qt::UserRole + 9, QualityRole = Qt::UserRole + 10, FilesIdRole = Qt::UserRole + 11 };
+		enum FileTypes { Movie = 1, FrontCover = 2, BackCover = 3, GeneralCover = 4 };
+		enum Fields { FileName = 0, PartNo = 1, SizeDisplay = 2, Quality = 3, DvdNo = 4, FullPath = 5, Size = 6, FileType = 7, Md5Sum = 8, SeriesPartId = 9, FilesId = 10 };
+		explicit FilesTreeModel(QStringList &headers, QObject *parent = 0);
+		~FilesTreeModel();
+
+		//data + flags
+		void setIds(const QList<int> &seriesPartIds);
+		QVariant data(const QModelIndex &index, int role) const;
+		bool setData(const QModelIndex &index, const QVariant &value, int role);
+		Qt::ItemFlags flags(const QModelIndex &index) const;
+
+	private:
+		QSqlDatabase mDb;
+		QSqlQuery *mFilesQuery;
+		QSqlQuery *mUpdateDvdQuery;
+		QSqlQuery *mUpdateQualityQuery;
+};
+
+#endif // FILESTREEMODEL_H