Minor fix for another change in Google's movie pages and a fix in
[movie-schedule] / src / ui / movieschedulemodel.h
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 #ifndef MOVIESCHEDULEMODEL_H
19 #define MOVIESCHEDULEMODEL_H
20
21 #include "data/scheduleentrykey.h"
22 #include "data/moviekey.h"
23
24 #include <QAbstractListModel>
25
26 class CinemaSchedule;
27
28 class MovieScheduleModel : public QAbstractListModel
29 {
30     Q_OBJECT
31 private:
32     typedef QList<QDate> ScheduleDates;
33     typedef QList<ScheduleEntryKey> ScheduleKeys;
34 public:
35     enum {
36         SortRole = Qt::UserRole,
37         MovieScheduleItemRole,
38         MovieScheduleKeyRole
39     };
40 public:
41     MovieScheduleModel(const CinemaSchedule *, const MovieKey &, QObject *parent = 0);
42
43     void Update();
44
45     int rowCount(const QModelIndex &parent = QModelIndex()) const;
46     QVariant data(const QModelIndex &index, int role = Qt::DisplayRole) const;
47
48 private:
49     const CinemaSchedule *_cinema_schedule;
50     MovieKey _movie_key;
51     ScheduleDates _schedule_dates;
52     ScheduleKeys _schedule_keys;
53 };
54
55 #endif // MOVIESCHEDULEMODEL_H