577014123b9bfdbecaacb232cecbaecc70e78f56
[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 void StationListModel::readStationsElement()
94 {
95     m_reader.readNext();
96     while (!m_reader.atEnd()) {
97         if (m_reader.isEndElement()) {
98             m_reader.readNext();
99             break;
100         } else if (m_reader.isStartElement()) {
101             if (m_reader.name() == "station") {
102                 readStationElement();
103             } else {
104                 skipUnknownElement();
105             }
106         } else {
107             m_reader.readNext();
108         }
109     }
110 }
111
112 void StationListModel::readStationElement()
113 {
114     QStandardItem *item = new QStandardItem;
115     m_reader.readNext();
116     while (!m_reader.atEnd()) {
117         if (m_reader.isEndElement()) {
118             this->appendRow(item);
119             m_reader.readNext();
120             break;
121         } else if (m_reader.isStartElement()) {
122             if (m_reader.name() == "pos") {
123                 readPosElement(item);
124             } else  if (m_reader.name() == "name") {
125                 readNameElement(item);
126             } else  if (m_reader.name() == "code") {
127                 readCodeElement(item);
128             } else {
129                 skipUnknownElement();
130             }
131         } else {
132             m_reader.readNext();
133         }
134     }
135 }
136
137 void StationListModel::readPosElement(QStandardItem *item)
138 {
139     QStringList coordinates = m_reader.readElementText().split(",");
140     QGeoCoordinate pos = QGeoCoordinate(coordinates[0].toDouble(), coordinates[1].toDouble());
141     item->setData(QVariant::fromValue(pos), PositionRole);
142     m_reader.readElementText();
143     if (m_reader.isEndElement()) {
144         m_reader.readNext();
145     }
146 }
147
148 void StationListModel::readNameElement(QStandardItem *item)
149 {
150     item->setText(m_reader.readElementText());
151     if (m_reader.isEndElement()) {
152         m_reader.readNext();
153     }
154 }
155
156 void StationListModel::readCodeElement(QStandardItem *item)
157 {
158     const QString code = m_reader.readElementText();
159     qDebug() << "reading code element" << code;
160
161     item->setData(QVariant::fromValue(code), StationCodeRole);
162     if (m_reader.isEndElement()) {
163         m_reader.readNext();
164     }
165 }
166
167 void StationListModel::skipUnknownElement()
168 {
169     qDebug() << "skipping unknown element";
170
171     m_reader.readNext();
172     while (!m_reader.atEnd()) {
173         if (m_reader.isEndElement()) {
174             m_reader.readNext();
175             break;
176         } else if (!m_reader.isStartElement()) {
177             skipUnknownElement();
178         } else {
179             m_reader.readNext();
180         }
181     }
182 }