git.maemo.org Git - quandoparte/blob - 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(ScheduleType type READ type WRITE setType NOTIFY typeChanged)
  36     Q_ENUMS(ScheduleType)
  37
  38     enum StationRoles {
  39         TrainRole = Qt::UserRole +1,
  40         DepartureStationRole,
  41         DepartureTimeRole,
  42         ArrivalStationRole,
  43         ArrivalTimeRole,
  44         DetailsUrlRole,
  45         DelayRole,
  46         DelayClassRole
  47     };
  48
  49 public:
  50     enum ScheduleType {
  51         DepartureSchedule,
  52         ArrivalSchedule
  53     };
  54
  55     explicit StationScheduleModel(const QString &name = "", QObject *parent = 0);
  56
  57     QString &name();
  58     void setName(const QString &name);
  59
  60     ScheduleType type();
  61     void setType(ScheduleType type);
  62
  63     int rowCount(const QModelIndex &parent) const;
  64     QVariant data(const QModelIndex &index, int role) const;
  65
  66 signals:
  67     void nameChanged();
  68     void typeChanged();
  69
  70 public slots:
  71     void fetch(const QString &name);
  72
  73 private slots:
  74     void parse(const QByteArray &htmlReply, const QUrl &baseUrl);
  75
  76 private:
  77     QString m_name;
  78     QList<StationScheduleItem> m_departureSchedules;
  79     QList<StationScheduleItem> m_arrivalSchedules;
  80     ScheduleType m_scheduleType;
  81 };
  82
  83 #endif // STATIONSCHEDULEMODEL_H