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