A few more stations
[quandoparte] / application / stationlistmodel.cpp
1 /*
2
3 Copyright (C) 2011 Luciano Montanaro <mikelima@cirulla.net>
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 #include "stationlistmodel.h"
23
24 #include <QFile>
25 #include <QFileInfo>
26 #include <QDebug>
27 #include <QStandardItem>
28 #include <QGeoCoordinate>
29
30 #if (QT_VERSION < QT_VERSION_CHECK(5, 0, 0))
31 QTM_USE_NAMESPACE
32 Q_DECLARE_METATYPE(QGeoCoordinate)
33 #endif
34
35 StationListModel::StationListModel(QObject *parent) :
36     QStandardItemModel(parent)
37 {
38     setRowCount(0);
39 #if (QT_VERSION < QT_VERSION_CHECK(5, 0, 0))
40     setRoleNames(roleNames());
41 #endif
42 }
43
44 bool StationListModel::load(const QString &filename)
45 {
46     QFile file(filename);
47     QFileInfo fi(file);
48
49     qDebug() << "loading file:" << fi.absoluteFilePath();
50
51     emit layoutAboutToBeChanged();
52     beginResetModel();
53     if (!file.open(QFile::ReadOnly | QFile::Text)) {
54         qDebug() << "cannot open file:" << filename;
55         return false;
56     }
57     m_reader.setDevice(&file);
58     m_reader.readNext();
59     while (!m_reader.atEnd()) {
60         if (m_reader.isStartElement()) {
61             if(m_reader.name() == "stations") {
62                 readStationsElement();
63             } else {
64                 m_reader.raiseError(tr("Not a qpl file"));
65             }
66         } else {
67             m_reader.readNext();
68         }
69     }
70     file.close();
71     qDebug() << rowCount() << "stations loaded";
72     if (m_reader.hasError()) {
73         qDebug() << "parser error for:" << filename;
74         return false;
75     } else if (file.error() != QFile::NoError) {
76         qDebug() << "file error for:" << filename;
77         return false;
78     }
79     endResetModel();
80     emit layoutChanged();
81     return true;
82 }
83
84 QHash<int, QByteArray> StationListModel::roleNames() const
85 {
86     QHash<int, QByteArray> roles;
87     roles[Qt::DisplayRole] = "name";
88     roles[StationListModel::PositionRole] = "position";
89     roles[StationListModel::StationCodeRole] = "code";
90     return roles;
91 }
92
93 int StationListModel::rowCount(const QModelIndex &parent) const
94 {
95     return QStandardItemModel::rowCount(parent);
96
97 }
98
99 void StationListModel::readStationsElement()
100 {
101     m_reader.readNext();
102     while (!m_reader.atEnd()) {
103         if (m_reader.isEndElement()) {
104             m_reader.readNext();
105             break;
106         } else if (m_reader.isStartElement()) {
107             if (m_reader.name() == "station") {
108                 readStationElement();
109             } else {
110                 skipUnknownElement(m_reader.name().toString());
111             }
112         } else {
113             m_reader.readNext();
114         }
115     }
116 }
117
118 void StationListModel::readStationElement()
119 {
120     QStandardItem *item = new QStandardItem;
121     m_reader.readNext();
122     while (!m_reader.atEnd()) {
123         if (m_reader.isEndElement()) {
124             this->appendRow(item);
125             m_reader.readNext();
126             break;
127         } else if (m_reader.isStartElement()) {
128             if (m_reader.name() == "pos") {
129                 readPosElement(item);
130             } else  if (m_reader.name() == "name") {
131                 readNameElement(item);
132             } else  if (m_reader.name() == "code") {
133                 readCodeElement(item);
134             } else {
135                 skipUnknownElement(m_reader.name().toString());
136             }
137         } else {
138             m_reader.readNext();
139         }
140     }
141 }
142
143 void StationListModel::readPosElement(QStandardItem *item)
144 {
145     QStringList coordinates = m_reader.readElementText().split(",");
146     QGeoCoordinate pos = QGeoCoordinate(coordinates[0].toDouble(), coordinates[1].toDouble());
147     item->setData(QVariant::fromValue(pos), PositionRole);
148     m_reader.readElementText();
149     if (m_reader.isEndElement()) {
150         m_reader.readNext();
151     }
152 }
153
154 void StationListModel::readNameElement(QStandardItem *item)
155 {
156     item->setText(m_reader.readElementText());
157     if (m_reader.isEndElement()) {
158         m_reader.readNext();
159     }
160 }
161
162 void StationListModel::readCodeElement(QStandardItem *item)
163 {
164     const QString code = m_reader.readElementText();
165     qDebug() << "reading code element" << code;
166
167     item->setData(QVariant::fromValue(code), StationCodeRole);
168     if (m_reader.isEndElement()) {
169         m_reader.readNext();
170     }
171 }
172
173 void StationListModel::skipUnknownElement(const QString &name)
174 {
175     qDebug() << "skipping unknown element" << name << "at line" << m_reader.lineNumber();
176
177     m_reader.readNext();
178     while (!m_reader.atEnd()) {
179         if (m_reader.isEndElement()) {
180             m_reader.readNext();
181             break;
182         } else if (!m_reader.isStartElement()) {
183             skipUnknownElement(m_reader.name().toString());
184         } else {
185             m_reader.readNext();
186         }
187     }
188 }