1 files changed, 109 insertions, 0 deletions
diff --git a/moviemodel.cpp b/moviemodel.cpp
new file mode 100644
index 0000000..a4f47dc
--- /dev/null
+++ b/moviemodel.cpp
@@ -0,0 +1,109 @@
+/*
+  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.
+*/
+
+#include <QSqlQuery>
+
+#include "moviemodel.h"
+
+MovieModel::MovieModel(QObject *parent) : QAbstractItemModel(parent) {
+	mHeaders << tr("Title") << tr("Filename") << tr("MD5Sum") << tr("Size") << tr("Genre") << tr("Quality") << tr("Archived");
+	populate();
+}
+
+MovieModel::~MovieModel(){
+	qDeleteAll(mItems);
+}
+
+QModelIndex MovieModel::index(int row, int column, const QModelIndex &parent) const{
+	if((parent != QModelIndex()) || (!hasIndex(row, column, parent)) || (row > mItems.size())){
+		return QModelIndex();
+	}
+	MovieItem *internal = mItems.at(row);
+	return createIndex(row, column, internal);
+}
+
+QVariant MovieModel::data(const QModelIndex &index, int role) const{
+	if(!index.isValid()){
+		return QVariant();
+	}
+	if(role == Qt::DisplayRole){
+		MovieItem *item = static_cast<MovieItem*>(index.internalPointer());
+		Q_ASSERT(item != 0);
+		switch (index.column()){
+			case MovieItem::Dvd:
+				return QVariant(QString(tr("DVD %1")).arg(QString::number(item->dataAt(MovieItem::Dvd).toInt())));
+		}
+		return item->dataAt(index.column());
+	}
+	return QVariant();
+}
+
+Qt::ItemFlags MovieModel::flags(const QModelIndex &index) const{
+	if(!index.isValid()){
+		return 0;
+	}
+	return Qt::ItemIsEnabled | Qt::ItemIsSelectable;
+}
+
+QVariant MovieModel::headerData(int section, Qt::Orientation o, int role) const{
+	if((o == Qt::Horizontal) && (role == Qt::DisplayRole)){
+		if(section < mHeaders.size()){
+			return mHeaders.at(section);
+		}
+	}
+	return QVariant();
+}
+
+bool MovieModel::insertRows(int row, int count, const QModelIndex &){
+	if(row > rowCount(QModelIndex())){
+		return false;
+	}
+	Q_ASSERT(count == 1);
+	beginInsertRows(QModelIndex(), row, row);
+	MovieItem *newItem = new MovieItem(-1, 0);
+	mItems.insert(row, newItem);
+	endInsertRows();
+	return true;
+}
+
+bool MovieModel::removeRows(int row, int count, const QModelIndex &){
+	if(row >= rowCount(QModelIndex())){
+		return false;
+	}
+	beginRemoveRows(QModelIndex(), row, row + count - 1);
+	for(int i = row; i < (row + count); ++i){
+		MovieItem *item = mItems.at(i);
+		delete item;
+		mItems.removeAt(i);
+	}
+	endRemoveRows();
+	return true;
+}
+
+bool MovieModel::setDataAt(const QModelIndex &index, const QList<QVariant> &data, qint32 genre, const QList<QVariant> &actors, const QList<QVariant> &covers){
+	if(!index.isValid() || (data.size() > MovieItem::NumRows)){
+		return false;
+	}
+	MovieItem *item = static_cast<MovieItem*>(index.internalPointer());
+	Q_ASSERT(item != 0);
+	if(item->id() != -1){
+		qWarning("ID not as expected: item->id() == %d", item->id());
+		return false;
+	}
+	return true;
+}
+
+void MovieModel::populate(){
+	QSqlQuery movieQuery("SELECT imovid FROM movies");
+	movieQuery.exec();
+	while(movieQuery.next()){
+		int id = movieQuery.value(0).toInt();
+		MovieItem *item = new MovieItem(id);
+		mItems << item;
+	}
+}
+