diff options
Diffstat (limited to 'filestreemodel.cpp')
| -rw-r--r-- | filestreemodel.cpp | 15 |
1 files changed, 14 insertions, 1 deletions
diff --git a/filestreemodel.cpp b/filestreemodel.cpp index 38e86b8..b02c8ac 100644 --- a/filestreemodel.cpp +++ b/filestreemodel.cpp @@ -15,7 +15,7 @@ #include "smtreeitem.h" #include "helper.h" -FilesTreeModel::FilesTreeModel(QStringList &headers, QObject *parent) : SmTreeModel(headers, parent){ +FilesTreeModel::FilesTreeModel(QStringList &headers, QObject *parent) : SmTreeModel(headers, parent), mMode(Normal){ //database setup mDb = QSqlDatabase::database("treedb"); mUpdateDvdQuery = new QSqlQuery(mDb); @@ -38,6 +38,11 @@ FilesTreeModel::FilesTreeModel(QStringList &headers, QObject *parent) : SmTreeMo //cover types mCoverTypes = mFileTypes; mCoverTypes.remove(1); + + //mode names + mModeNames.insert(Normal, "Normal"); + mModeNames.insert(Archived, "Archived"); + mModeNames.insert(Local, "Local"); } FilesTreeModel::~FilesTreeModel(){ @@ -49,6 +54,13 @@ FilesTreeModel::~FilesTreeModel(){ mDb = QSqlDatabase(); } +const QString FilesTreeModel::modeName(int mode) const{ + if(mode == -1){ + return mModeNames.value(mMode); + } + return mModeNames.value(mode); +} + void FilesTreeModel::setIds(const QList<int> &seriesPartIds){ QStringList ids; foreach(int s, seriesPartIds){ @@ -76,6 +88,7 @@ void FilesTreeModel::setMode(int mode){ if(queryOk){ populate(modeQuery); } + mMode = mode; } QVariant FilesTreeModel::data(const QModelIndex &index, int role) const{ |