From 376d2fbb78114e746618f14afe7728cf9793630b Mon Sep 17 00:00:00 2001 From: Arno Date: Sat, 1 Jun 2013 11:17:42 +0200 Subject: Do something when SortOrder is changed in ArchiveView Actually do something when the sort order is changed. That resulted in various fixes to both model and view, since both had completely untested codepaths. Most notable fix: Only select those seriesparts which are actually mapped to the selected genre or actor. --- archivemodel.cpp | 60 ++++++++++++++++++++++++++++++++++++-------------------- 1 file changed, 39 insertions(+), 21 deletions(-) (limited to 'archivemodel.cpp') diff --git a/archivemodel.cpp b/archivemodel.cpp index 3379d67..5c3cb49 100644 --- a/archivemodel.cpp +++ b/archivemodel.cpp @@ -12,14 +12,14 @@ ArchiveModel::ArchiveModel(const QStringList &headers, QObject *parent) : SmTreeModel(headers, parent), mOrder(SeriesName){ mDb = QSqlDatabase::database("treedb"); - mAvailableOrders.insert(SeriesName, "Series Name"); - mAvailableOrders.insert(Actor, "Actor"); - mAvailableOrders.insert(Genre, "Genre"); + mAvailableOrders.insert("Series Name", SeriesName); + mAvailableOrders.insert("Actor", Actor); + mAvailableOrders.insert("Genre", Genre); setOrder(mOrder); } const QStringList ArchiveModel::availableOrders() const { - QStringList retval = mAvailableOrders.values(); + QStringList retval = mAvailableOrders.keys(); qSort(retval); return retval; } @@ -31,7 +31,7 @@ QVariant ArchiveModel::data(const QModelIndex &index, int role) const{ int part = item->data(SeriesPart).toInt(); int childCount = item->data(Count).toInt(); int nodeType = item->data(Type).toInt(); - if(nodeType == SeriesNode){ + if(nodeType == SeriesNode || nodeType == GenreNode || nodeType == ActorNode){ retval = QString("%1 (%2)").arg(item->data(Name).toString()).arg(QString::number(childCount)); }else if(nodeType == SeriesPartNode){ if(part > 0){ @@ -42,6 +42,8 @@ QVariant ArchiveModel::data(const QModelIndex &index, int role) const{ retval = QString("%1 - %2").arg(item->data(Name).toString()).arg(item->data(Subtitle).toString()); } } + }else{ + retval = item->data(Name).toString(); } return retval; } @@ -55,6 +57,7 @@ QVariant ArchiveModel::data(const QModelIndex &index, int role) const{ } void ArchiveModel::setOrder(int order) { + mOrder = order; switch (order) { case SeriesName: populateBySeriesName(); @@ -70,10 +73,18 @@ void ArchiveModel::setOrder(int order) { } } +void ArchiveModel::setOrder(const QString &order){ + int orderNum = mAvailableOrders.value(order); + setOrder(orderNum); +} + void ArchiveModel::populateBySeriesName() { SmTreeItem *rootItem = new SmTreeItem(NumFields); + fetchChildren(rootItem); fetchSeries(QVariant(), rootItem); - fetchParts(rootItem); + for(int i = 0; i < rootItem->childCount(); ++i){ + fetchParts(QVariant(), rootItem->child(i)); + } setRoot(rootItem); } @@ -105,10 +116,10 @@ void ArchiveModel::populateByActor(){ void ArchiveModel::fetchChildren(SmTreeItem *parent){ for(int i = 0; i < parent->childCount(); ++i){ - fetchSeries(parent->child(i)->data(SeriesId), parent->child(i)); + fetchSeries(parent->child(i)->data(GenericId), parent->child(i)); for(int j = 0; j < parent->child(i)->childCount(); ++j){ SmTreeItem *seriesItem = parent->child(i)->child(j); - fetchParts(seriesItem); + fetchParts(parent->child(i)->data(GenericId), seriesItem); } } } @@ -145,19 +156,26 @@ void ArchiveModel::fetchSeries(const QVariant &id, SmTreeItem *parent){ } -void ArchiveModel::fetchParts(SmTreeItem *parent){ +void ArchiveModel::fetchParts(const QVariant &id, SmTreeItem *parent){ QSqlQuery partsQuery(mDb); - partsQuery.prepare("SELECT iseriesparts_id, iseriespart, bfavorite, tsubtitle FROM seriesparts WHERE iseries_id = :id ORDER BY iseriespart"); - for(int i = 0; i < parent->childCount(); ++i){ - QVariant id = parent->child(i)->data(SeriesId); - partsQuery.bindValue(":id", id); - partsQuery.exec(); - while(partsQuery.next()){ - QList partData; - partData << parent->child(i)->data(Name) << parent->child(i)->data(SeriesId) << partsQuery.value(0) << partsQuery.value(1) << SeriesPartNode << partsQuery.value(2) << partsQuery.value(3) << QVariant(); - SmTreeItem *partItem = new SmTreeItem(partData, parent->child(i)); - parent->child(i)->appendChild(partItem); - } - parent->child(i)->setData(Count, parent->child(i)->childCount()); + if(mOrder == SeriesName){ + partsQuery.prepare("SELECT iseriesparts_id, iseriespart, bfavorite, tsubtitle FROM seriesparts WHERE iseries_id = :id ORDER BY iseriespart"); + partsQuery.bindValue(":id", parent->data(GenericId)); + }else if(mOrder == Genre){ + partsQuery.prepare("SELECT seriesparts.iseriesparts_id, iseriespart, bfavorite, tsubtitle FROM seriesparts, seriesparts_genremap WHERE iseries_id = :id AND seriesparts.iseriesparts_id = seriesparts_genremap.iseriesparts_id AND seriesparts_genremap.igenres_id = :genreid ORDER BY iseriespart"); + partsQuery.bindValue(":id", parent->data(GenericId)); + partsQuery.bindValue(":genreid", id); + }else if(mOrder == Actor){ + partsQuery.prepare("SELECT seriesparts.iseriesparts_id, iseriespart, bfavorite, tsubtitle FROM seriesparts, seriesparts_actormap WHERE iseries_id = :id AND seriesparts.iseriesparts_id = seriesparts_actormap.iseriesparts_id AND seriesparts_actormap.iactors_id = :actorid ORDER BY iseriespart"); + partsQuery.bindValue(":id", parent->data(GenericId)); + partsQuery.bindValue(":actorid", id); } + partsQuery.exec(); + while(partsQuery.next()){ + QList partData; + partData << parent->data(Name) << parent->data(GenericId) << partsQuery.value(0) << partsQuery.value(1) << SeriesPartNode << partsQuery.value(2) << partsQuery.value(3) << QVariant(); + SmTreeItem *partItem = new SmTreeItem(partData, parent); + parent->appendChild(partItem); + } + parent->setData(Count, parent->childCount()); } -- cgit v1.2.3-70-g09d2