Differential D1917 Diff 4555 src/plugins/runner/gdal/GdalRunner.cpp

Changeset View

Standalone View

src/plugins/runner/gdal/GdalRunner.cpp

This file was added.

1		//
2		// This file is part of the Marble Virtual Globe.
3		//
4		// This program is free software licensed under the GNU LGPL. You can
5		// find a copy of this license in LICENSE.txt in the top directory of
6		// the source code.
7		//
8		// Copyright 2016 Chris Hills <chaz@chaz6.com>
9
10		#include "GdalRunner.h"
11
12		#include "GeoDataDocument.h"
13		#include "GeoDataPlacemark.h"
14		#include "GeoDataPolygon.h"
15		#include "GeoDataSchema.h"
16		#include "GeoDataSimpleField.h"
17		#include "GeoDataStyle.h"
18		#include "GeoDataPolyStyle.h"
19		#include "MarbleDebug.h"
20
21		#include <QtCore/QFileInfo>
22
	kossebauUnsubmitted Not Done No QtCore module prefix, please. kossebau: No QtCore module prefix, please.
23		#include <gdal_priv.h>
24		#include <ogrsf_frmts.h>
25
26		namespace Marble
27		{
28
29		GdalRunner::GdalRunner(QObject *parent) :
30		ParsingRunner(parent)
31		{
32		}
33
34		GdalRunner::~GdalRunner()
35		{
36		}
37
38		GeoDataDocument *GdalRunner::parseFile( const QString &fileName, DocumentRole role, QString &error )
39		{
40
41		GDALDataset *gDataset;
42		gDataset = static_cast<GDALDataset*>( GDALOpenEx( fileName.toStdString().c_str(), GDAL_OF_VECTOR, NULL, NULL, NULL ) );
43
44		if ( gDataset == NULL ) {
45		error = QString("Failed to read %1").arg(fileName);
	kossebauUnsubmitted Not Done No NULL please., this is C++ :) 0 should work here. kossebau: No NULL please., this is C++ :) 0 should work here.
46		mDebug() << error;
47		return nullptr;
48		}
49
50		if ( gDataset->GetLayerCount() < 1 ) {
51		mDebug() << "No layers found!";
52		return nullptr;
	nienhueserUnsubmitted Done Can you update error here, too? nienhueser: Can you update error here, too?
	chaz6AuthorUnsubmitted Done OK! chaz6: OK!
53		}
54
55		GeoDataDocument *document = new GeoDataDocument;
56		document->setDocumentRole( role );
57
58		OGRLayer *gLayer;
59		gLayer = gDataset->GetLayer(0); // TODO: support for multiple layers
60
61		OGRFeature *gFeature;
62		gLayer->ResetReading();
63		while( ( gFeature = gLayer->GetNextFeature() ) != NULL ) {
64		OGRGeometry *gGeometry;
65		gGeometry = gFeature->GetGeometryRef();
66		if( gGeometry != NULL ) {
67		GeoDataPlacemark *placemark = 0;
68		placemark = new GeoDataPlacemark;
69
70		switch( gGeometry->getGeometryType() ) {
	nienhueserUnsubmitted Done Should this only be done when the geometry is supported below? Otherwise empty placemarks would be added nienhueser: Should this only be done when the geometry is supported below? Otherwise empty placemarks would…
	chaz6AuthorUnsubmitted Not Done Understood, I will move it. chaz6: Understood, I will move it.
71		case wkbPoint: {
72		OGRPoint gPoint = static_cast<OGRPoint>( gGeometry );
73		GeoDataPoint *point = new GeoDataPoint( gPoint->getX(), gPoint->getY(), 0, GeoDataCoordinates::Degree );
74		document->append( placemark );
75		placemark->setGeometry( point );
	nienhueserUnsubmitted Done Use p.GetZ() instead of 0 here? Same for the other points below. nienhueser: Use p.GetZ() instead of 0 here? Same for the other points below.
76		mDebug() << "point " << placemark->name();
77		break;
78		}
79		case wkbLineString: {
80		OGRLineString gLineString = static_cast<OGRLineString>( gGeometry );
81		GeoDataLineString *line = new GeoDataLineString;
82		OGRPoint p;
83
84		for( int j=0; j<gLineString->getNumPoints(); ++j ) {
85		gLineString->getPoint( j, &p );
	kossebauUnsubmitted Not Done To transform the temporary string behind the const char* pointer returned by OGRFeature::GetFieldAsString* into a QString, use QString::fromLatin1(), or, if there is not only 7bit-latin chars in the string, but actually utf-8 encoded ones, QString::fromUtf8(). E.g. placemark->setName(QString::fromLatin1(gFeature->GetFieldAsString(nameField))); http://www.gdal.org/classOGRFeature.html#ac7a0224e13dce1f9479a6bf2111ea766 kossebau: To transform the temporary string behind the const char* pointer returned by OGRFeature…
86		line->append( GeoDataCoordinates(
87		p.getX(), p.getY(),
88		0, GeoDataCoordinates::Degree ) );
89		}
90
91		document->append( placemark );
92		placemark->setGeometry( line );
93		mDebug() << "arc " << placemark->name();
94		break;
95		}
96		case wkbPolygon: {
97		OGRPolygon gPolygon = static_cast<OGRPolygon>( gGeometry );
98
99		GeoDataPolygon *polygon = new GeoDataPolygon;
100
101		OGRLinearRing *gLinearRing = gPolygon->getExteriorRing();
102		GeoDataLinearRing outerRing;
103
104		for( int j=0; j<gLinearRing->getNumPoints(); ++j ) {
105		OGRPoint gPoint;
106		gLinearRing->getPoint( j, &gPoint );
107		outerRing.append( GeoDataCoordinates(
108		gPoint.getX(), gPoint.getY(),
109		0, GeoDataCoordinates::Degree ) );
110		}
111		polygon->setOuterBoundary( outerRing );
112
113		for( int j=0; j<gPolygon->getNumInteriorRings(); ++j ) {
114		GeoDataLinearRing ring;
115		gLinearRing = gPolygon->getInteriorRing( j );
116		for( int k=0; k<gLinearRing->getNumPoints(); ++k ) {
117		OGRPoint gPoint;
118		gLinearRing->getPoint( k, &gPoint );
119		ring.append( GeoDataCoordinates(
120		gPoint.getX(), gPoint.getY(),
121		0, GeoDataCoordinates::Degree ) );
122		}
123		polygon->appendInnerBoundary( ring );
124		}
125
126		document->append( placemark );
127		placemark->setGeometry( polygon );
128		mDebug() << "polygon " << placemark->name();
129		break;
130		}
131		case wkbMultiPoint: {
132		break;
133		}
134		case wkbMultiLineString: {
135		break;
136		}
137		case wkbMultiPolygon: {
138		break;
139		}
140		}
141		}
142		}
143
144		if ( document->size() ) {
145		document->setFileName( fileName );
146		return document;
147		} else {
148		delete document;
149		return nullptr;
150		}
151
152		}
153
154		}
155
156		#include "moc_GdalRunner.cpp"
	kossebauUnsubmitted Not Done Tabs sneaking in? :) Please use spaces for indentation. kossebau: Tabs sneaking in? :) Please use spaces for indentation.
	kossebauUnsubmitted Not Done No need for `else` after `return` kossebau: No need for `else` after `return`