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
|
/*
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 <QStringList>
#include <QRegExp>
#include <QList>
#include "archiveproxy.h"
#include "moviemodel.h"
ArchiveProxy::ArchiveProxy(QObject *parent) : QSortFilterProxyModel(parent){};
void ArchiveProxy::setFilter(const QString &filter, FilterMode mode){
mFilterMode = mode;
switch(mFilterMode){
case NoFilter:
mGenreFilter = QString();
mActorFilter = QString();
mTitleFilter = QString();
break;
case GenreFilter:
mGenreFilter = filter;
break;
case ActorFilter:
mActorFilter = filter;
break;
case TitleFilter:
mTitleFilter = filter.toLower();
break;
}
invalidate();
}
bool ArchiveProxy::filterAcceptsRow(int sourceRow, const QModelIndex & /*sourceParent*/) const{
if(mFilterMode == NoFilter){
return true;
}
MovieModel *model = static_cast<MovieModel*>(sourceModel());
Q_ASSERT(model != 0);
if(mFilterMode == ActorFilter){
QModelIndex idx = model->index(sourceRow, 0, QModelIndex());
QStringList actors = model->data(idx, MovieModel::ActorsRole).toStringList();
return actors.contains(mActorFilter);
}
if(mFilterMode == GenreFilter){
QModelIndex idx = model->index(sourceRow, 4, QModelIndex());
QString genre = idx.data().toString();
return (genre == mGenreFilter);
}
if(mFilterMode == TitleFilter){
QModelIndex idx = model->index(sourceRow, 0, QModelIndex());
QString title = idx.data().toString();
QRegExp re(mTitleFilter);
return( re.indexIn(title) != -1);
}
return true;
}
void ArchiveProxy::clearFilter(){
mFilterMode = NoFilter;
mGenreFilter = QString();
mActorFilter = QString();
mTitleFilter = QString();
invalidate();
}
bool ArchiveProxy::lessThan(const QModelIndex &left, const QModelIndex &right) const{
if(left.column() > 0){
return QSortFilterProxyModel::lessThan(left, right);
}
QString lbase = left.data(MovieModel::TitleBaseRole).toString();
QString rbase = right.data(MovieModel::TitleBaseRole).toString();
if(lbase == rbase){
int lseriesno = left.data(MovieModel::SeriesNoRole).toInt();
int rseriesno = right.data(MovieModel::SeriesNoRole).toInt();
if(lseriesno == rseriesno){
int lpartno = left.data(MovieModel::PartNoRole).toInt();
int rpartno = right.data(MovieModel::PartNoRole).toInt();
return (lpartno < rpartno);
}else{
return (lseriesno < rseriesno);
}
}else{
return (lbase < rbase);
}
return QSortFilterProxyModel::lessThan(left, right);
}
|
