blob: 6b1a9e7a501ba692889b12e9d5ac5567be4ef1ee (
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
|
/*
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 "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();
}
|
