Make the station code visible to QML
[quandoparte] / application / stationschedulemodel.h
1 /*
2
3 Copyright (C) 2011 mikelima
4
5 This program 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 2 of the License, or
8 (at your option) any later version.
9
10 This program 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 GNU
13 General Public License for more details.
14
15 You should have received a copy of the GNU General Public License
16 along with this program; see the file COPYING.  If not, write to
17 the Free Software Foundation, Inc., 51 Franklin Street, Fifth Floor,
18 Boston, MA 02110-1301, USA.
19
20 */
21
22 #ifndef STATIONSCHEDULEMODEL_H
23 #define STATIONSCHEDULEMODEL_H
24
25 #include "stationscheduleitem.h"
26
27 #include <QObject>
28 #include <QAbstractListModel>
29 #include <QUrl>
30
31 class StationScheduleModel : public QAbstractListModel
32 {
33     Q_OBJECT
34     Q_PROPERTY(QString name READ name WRITE setName NOTIFY nameChanged)
35     Q_PROPERTY(QString code READ code WRITE setCode NOTIFY codeChanged)
36     Q_PROPERTY(ScheduleType type READ type WRITE setType NOTIFY typeChanged)
37     Q_ENUMS(ScheduleType)
38
39     enum StationRoles {
40         TrainRole = Qt::UserRole +1,
41         DepartureStationRole,
42         DepartureTimeRole,
43         ArrivalStationRole,
44         ArrivalTimeRole,
45         DetailsUrlRole,
46         DelayRole,
47         DelayClassRole,
48         ExpectedPlatformRole,
49         ActualPlatformRole
50     };
51
52 public:
53     enum ScheduleType {
54         DepartureSchedule,
55         ArrivalSchedule
56     };
57
58     explicit StationScheduleModel(const QString &name = "", QObject *parent = 0);
59
60     QString &name();
61     void setName(const QString &name);
62
63     QString &code();
64     void setCode(const QString &code);
65
66     ScheduleType type();
67     void setType(ScheduleType type);
68
69     int rowCount(const QModelIndex &parent) const;
70     QVariant data(const QModelIndex &index, int role) const;
71
72 signals:
73     void nameChanged();
74     void codeChanged();
75     void typeChanged();
76
77 public slots:
78     void fetch(const QString &name, const QString &code = QString());
79
80 private slots:
81     void parse(const QByteArray &htmlReply, const QUrl &baseUrl);
82
83 private:
84     QString m_name;
85     QString m_code;
86     QList<StationScheduleItem> m_departureSchedules;
87     QList<StationScheduleItem> m_arrivalSchedules;
88     ScheduleType m_scheduleType;
89 };
90
91 #endif // STATIONSCHEDULEMODEL_H