Merge https://vcs.maemo.org/git/movie-schedule
[movie-schedule] / src / ui / moviemodel.cpp
1 // Copyright 2010 Jochen Becher
2 //
3 // This file is part of MovieSchedule.
4 //
5 // MovieSchedule is free software: you can redistribute it and/or modify
6 // it under the terms of the GNU General Public License as published by
7 // the Free Software Foundation, either version 3 of the License, or
8 // (at your option) any later version.
9 //
10 // MovieSchedule is distributed in the hope that it will be useful,
11 // but WITHOUT ANY WARRANTY; without even the implied warranty of
12 // MERCHANTABILITY or FITNESS FOR A PARTICULAR PURPOSE.  See the
13 // GNU General Public License for more details.
14 //
15 // You should have received a copy of the GNU General Public License
16 // along with MovieSchedule.  If not, see <http://www.gnu.org/licenses/>.
17
18 #include "moviemodel.h"
19
20 #include "data/cinemaschedule.h"
21 #include "data/movie.h"
22 #include "utils/assertedlocker.h"
23
24 #include <iostream>
25
26 MovieModel::MovieModel(const CinemaSchedule *cinema_schedule, QObject *parent)
27     : QAbstractListModel(parent),
28     _cinema_schedule(cinema_schedule)
29 {
30 }
31
32 int MovieModel::rowCount(const QModelIndex &parent) const
33 {
34     Q_UNUSED(parent);
35     AssertedReadLocker locker(_cinema_schedule->GetLock());
36     return _cinema_schedule->GetMovies().values().size();
37 }
38
39 QVariant MovieModel::data(const QModelIndex &index, int role) const
40 {
41     AssertedReadLocker locker(_cinema_schedule->GetLock());
42     if (!(index.isValid() && index.row() >= 0 && index.row() < _cinema_schedule->GetMovies().values().size())) {
43         std::cout << "MovieModel: invalid index" << std::endl;
44         return QVariant();
45     }
46     Movie *movie = _cinema_schedule->GetMovies().values().operator [](index.row());
47     if (movie == 0) {
48         std::cout << "movie is 0 (row = " << index.row() << ", size = " << _cinema_schedule->GetMovies().values().size() << ")" << std::endl;
49         return QVariant();
50     }
51     switch (role) {
52     case Qt::DisplayRole:
53     case MovieItemRole:
54         return qVariantFromValue(MovieItem(*movie));
55     case MovieKeyRole:
56         return qVariantFromValue(movie->GetKey());
57     case SortRole:
58         return QVariant(movie->GetName());
59     }
60     return QVariant();
61 }