From 721ee2aaa9df3a0c5d3494e2e51b091257daea7c Mon Sep 17 00:00:00 2001
From: =?UTF-8?q?Martin=20T=C5=AFma?= <tumic@cbox.cz>
Date: Tue, 25 Oct 2016 22:50:11 +0200
Subject: [PATCH] Improved KML parser

---
 src/kmlparser.cpp | 123 ++++++++++++++++++++++++++++++++++++++--------
 src/kmlparser.h   |   9 +++-
 2 files changed, 110 insertions(+), 22 deletions(-)

diff --git a/src/kmlparser.cpp b/src/kmlparser.cpp
index 78c7a2f8..a8888738 100644
--- a/src/kmlparser.cpp
+++ b/src/kmlparser.cpp
@@ -13,8 +13,12 @@ bool KMLParser::pointCoordinates()
 	sp = data.constData();
 	ep = sp + data.size();
 
-	for (cp = sp, vp = sp; cp <= ep; cp++) {
-		if (*cp == ',' || cp->isNull()) {
+	for (cp = sp; cp < ep; cp++)
+		if (!cp->isSpace())
+			break;
+
+	for (vp = cp; cp <= ep; cp++) {
+		if (*cp == ',') {
 			if (c > 2)
 				return false;
 
@@ -34,6 +38,26 @@ bool KMLParser::pointCoordinates()
 
 			c++;
 			vp = cp + 1;
+		} else if (cp->isSpace() || cp->isNull()) {
+			if (c < 1)
+				return false;
+
+#if QT_VERSION < QT_VERSION_CHECK(5, 0, 0)
+			val[c] = QString(vp, cp - vp).toDouble(&res);
+#else // QT_VERSION < 5
+			val[c] = QStringRef(&data, vp - sp, cp - vp).toDouble(&res);
+#endif // QT_VERSION < 5
+			if (!res)
+				return false;
+
+			Waypoint w(Coordinates(val[0], val[1]));
+			if (c == 2)
+				w.setElevation(val[2]);
+			_waypoints.append(w);
+
+			while (cp->isSpace())
+				cp++;
+			c = 3;
 		}
 	}
 
@@ -52,8 +76,12 @@ bool KMLParser::lineCoordinates()
 	sp = data.constData();
 	ep = sp + data.size();
 
-	for (cp = sp, vp = sp; cp <= ep; cp++) {
-		if (*cp == ',' || cp->isSpace() || cp->isNull()) {
+	for (cp = sp; cp < ep; cp++)
+		if (!cp->isSpace())
+			break;
+
+	for (vp = cp; cp <= ep; cp++) {
+		if (*cp == ',') {
 			if (c > 2)
 				return false;
 
@@ -65,25 +93,48 @@ bool KMLParser::lineCoordinates()
 			if (!res)
 				return false;
 
-			if (c == 2) {
-				Waypoint w(Coordinates(val[0], val[1]));
-				w.setElevation(val[2]);
-				_route->append(w);
-			}
-
 			c++;
 			vp = cp + 1;
-		}
-		if (cp->isSpace()) {
-			if (c < 3)
+		} else if (cp->isSpace() || cp->isNull()) {
+			if (c < 1 || c > 2)
 				return false;
+
+#if QT_VERSION < QT_VERSION_CHECK(5, 0, 0)
+			val[c] = QString(vp, cp - vp).toDouble(&res);
+#else // QT_VERSION < 5
+			val[c] = QStringRef(&data, vp - sp, cp - vp).toDouble(&res);
+#endif // QT_VERSION < 5
+			if (!res)
+				return false;
+
+			Waypoint w(Coordinates(val[0], val[1]));
+			if (c == 2)
+				w.setElevation(val[2]);
+			_route->append(w);
+
 			while (cp->isSpace())
 				cp++;
 			c = 0;
+			vp = cp;
 		}
 	}
 
-	return true;
+ 	return true;
+}
+
+QDateTime KMLParser::timeStamp()
+{
+	QDateTime date;
+
+	while (_reader.readNextStartElement()) {
+		if (_reader.name() == "when") {
+			date = QDateTime::fromString(_reader.readElementText(),
+			  Qt::ISODate);
+		} else
+			_reader.skipCurrentElement();
+	}
+
+	return date;
 }
 
 void KMLParser::lineString()
@@ -99,22 +150,33 @@ void KMLParser::lineString()
 	}
 }
 
-void KMLParser::point()
+void KMLParser::point(const QString &name, const QString &desc,
+  const QDateTime timestamp)
 {
 	while (_reader.readNextStartElement()) {
 		if (_reader.name() == "coordinates") {
 			if (!pointCoordinates())
 				_reader.raiseError("Invalid coordinates.");
+			else {
+				Waypoint &w = _waypoints.last();
+				if (!name.isNull())
+					w.setName(name);
+				if (!desc.isNull())
+					w.setDescription(desc);
+				if (!timestamp.isNull())
+					w.setTimestamp(timestamp);
+			}
 		} else
 			_reader.skipCurrentElement();
 	}
 }
 
-void KMLParser::multiGeometry()
+void KMLParser::multiGeometry(const QString &name, const QString &desc,
+  const QDateTime timestamp)
 {
 	while (_reader.readNextStartElement()) {
 		if (_reader.name() == "Point")
-			point();
+			point(name, desc, timestamp);
 		else if (_reader.name() == "LineString")
 			lineString();
 		else
@@ -124,13 +186,32 @@ void KMLParser::multiGeometry()
 
 void KMLParser::placemark()
 {
+	QString name, desc;
+	QDateTime date;
+
 	while (_reader.readNextStartElement()) {
-		if (_reader.name() == "Point")
-			point();
+		if (_reader.name() == "name")
+			name = _reader.readElementText();
+		else if (_reader.name() == "description")
+			desc = _reader.readElementText();
+		else if (_reader.name() == "TimeStamp")
+			date = timeStamp();
+		else if (_reader.name() == "Point")
+			point(name, desc, date);
 		else if (_reader.name() == "LineString")
 			lineString();
 		else if (_reader.name() == "MultiGeometry")
-			multiGeometry();
+			multiGeometry(name, desc, date);
+		else
+			_reader.skipCurrentElement();
+	}
+}
+
+void KMLParser::folder()
+{
+	while (_reader.readNextStartElement()) {
+		if (_reader.name() == "Placemark")
+			placemark();
 		else
 			_reader.skipCurrentElement();
 	}
@@ -141,6 +222,8 @@ void KMLParser::document()
 	while (_reader.readNextStartElement()) {
 		if (_reader.name() == "Placemark")
 			placemark();
+		else if (_reader.name() == "Folder")
+			folder();
 		else
 			_reader.skipCurrentElement();
 	}
diff --git a/src/kmlparser.h b/src/kmlparser.h
index b27f348f..228619c3 100644
--- a/src/kmlparser.h
+++ b/src/kmlparser.h
@@ -3,6 +3,7 @@
 
 #include <QXmlStreamReader>
 #include <QVector>
+#include <QDateTime>
 #include "parser.h"
 
 class KMLParser : public Parser
@@ -22,12 +23,16 @@ private:
 	bool parse();
 	void kml();
 	void document();
+	void folder();
 	void placemark();
-	void multiGeometry();
+	void multiGeometry(const QString &name, const QString &desc,
+	  const QDateTime timestamp);
 	void lineString();
-	void point();
+	void point(const QString &name, const QString &desc,
+	  const QDateTime timestamp);
 	bool pointCoordinates();
 	bool lineCoordinates();
+	QDateTime timeStamp();
 
 	QXmlStreamReader _reader;
 	QVector<Trackpoint> *_track;