Branch data Line data Source code
1 : : /***************************************************************************
2 : : qgsvectorfilewriter.cpp
3 : : generic vector file writer
4 : : -------------------
5 : : begin : Sat Jun 16 2004
6 : : copyright : (C) 2004 by Tim Sutton
7 : : email : tim at linfiniti.com
8 : : ***************************************************************************/
9 : :
10 : : /***************************************************************************
11 : : * *
12 : : * This program is free software; you can redistribute it and/or modify *
13 : : * it under the terms of the GNU General Public License as published by *
14 : : * the Free Software Foundation; either version 2 of the License, or *
15 : : * (at your option) any later version. *
16 : : * *
17 : : ***************************************************************************/
18 : :
19 : : #include "qgsapplication.h"
20 : : #include "qgsfields.h"
21 : : #include "qgsfeature.h"
22 : : #include "qgsfeatureiterator.h"
23 : : #include "qgsgeometry.h"
24 : : #include "qgslogger.h"
25 : : #include "qgsmessagelog.h"
26 : : #include "qgscoordinatereferencesystem.h"
27 : : #include "qgsvectorfilewriter.h"
28 : : #include "qgsrenderer.h"
29 : : #include "qgssymbollayer.h"
30 : : #include "qgsvectordataprovider.h"
31 : : #include "qgsvectorlayer.h"
32 : : #include "qgslocalec.h"
33 : : #include "qgsexception.h"
34 : : #include "qgssettings.h"
35 : : #include "qgsgeometryengine.h"
36 : : #include "qgsproviderregistry.h"
37 : : #include "qgsexpressioncontextutils.h"
38 : : #include "qgsreadwritelocker.h"
39 : :
40 : : #include <QFile>
41 : : #include <QFileInfo>
42 : : #include <QDir>
43 : : #include <QTextCodec>
44 : : #include <QTextStream>
45 : : #include <QSet>
46 : : #include <QMetaType>
47 : : #include <QMutex>
48 : : #include <QRegularExpression>
49 : :
50 : : #include <cassert>
51 : : #include <cstdlib> // size_t
52 : : #include <limits> // std::numeric_limits
53 : :
54 : : #include <ogr_srs_api.h>
55 : : #include <cpl_error.h>
56 : : #include <cpl_conv.h>
57 : : #include <cpl_string.h>
58 : : #include <gdal.h>
59 : :
60 : : // Thin wrapper around OGROpen() to workaround a bug in GDAL < 2.3.1
61 : : // where a existing BNA file is wrongly reported to be openable in update mode
62 : : // but attempting to add features in it crashes the BNA driver.
63 : 0 : static OGRDataSourceH myOGROpen( const char *pszName, int bUpdate, OGRSFDriverH *phDriver )
64 : : {
65 : 0 : OGRSFDriverH hDriver = nullptr;
66 : 0 : OGRDataSourceH hDS = OGROpen( pszName, bUpdate, &hDriver );
67 : 0 : if ( hDS && bUpdate )
68 : : {
69 : 0 : QString drvName = OGR_Dr_GetName( hDriver );
70 : 0 : if ( drvName == "BNA" )
71 : : {
72 : 0 : OGR_DS_Destroy( hDS );
73 : 0 : if ( phDriver )
74 : 0 : *phDriver = nullptr;
75 : 0 : return nullptr;
76 : : }
77 : 0 : }
78 : 0 : if ( phDriver )
79 : 0 : *phDriver = hDriver;
80 : 0 : return hDS;
81 : 0 : }
82 : :
83 : 0 : QgsField QgsVectorFileWriter::FieldValueConverter::fieldDefinition( const QgsField &field )
84 : : {
85 : 0 : return field;
86 : : }
87 : :
88 : 0 : QVariant QgsVectorFileWriter::FieldValueConverter::convert( int /*fieldIdxInLayer*/, const QVariant &value )
89 : : {
90 : 0 : return value;
91 : : }
92 : :
93 : 0 : QgsVectorFileWriter::FieldValueConverter *QgsVectorFileWriter::FieldValueConverter::clone() const
94 : : {
95 : 0 : return new FieldValueConverter( *this );
96 : : }
97 : :
98 : 0 : QgsVectorFileWriter::QgsVectorFileWriter(
99 : : const QString &vectorFileName,
100 : : const QString &fileEncoding,
101 : : const QgsFields &fields,
102 : : QgsWkbTypes::Type geometryType,
103 : : const QgsCoordinateReferenceSystem &srs,
104 : : const QString &driverName,
105 : : const QStringList &datasourceOptions,
106 : : const QStringList &layerOptions,
107 : : QString *newFilename,
108 : : SymbologyExport symbologyExport,
109 : : QgsFeatureSink::SinkFlags sinkFlags,
110 : : QString *newLayer,
111 : : QgsCoordinateTransformContext transformContext,
112 : : FieldNameSource fieldNameSource
113 : : )
114 : 0 : : mError( NoError )
115 : 0 : , mWkbType( geometryType )
116 : 0 : , mSymbologyExport( symbologyExport )
117 : 0 : , mSymbologyScale( 1.0 )
118 : 0 : {
119 : 0 : init( vectorFileName, fileEncoding, fields, geometryType,
120 : 0 : srs, driverName, datasourceOptions, layerOptions, newFilename, nullptr,
121 : 0 : QString(), CreateOrOverwriteFile, newLayer, sinkFlags, transformContext, fieldNameSource );
122 : 0 : }
123 : :
124 : 0 : QgsVectorFileWriter::QgsVectorFileWriter(
125 : : const QString &vectorFileName,
126 : : const QString &fileEncoding,
127 : : const QgsFields &fields,
128 : : QgsWkbTypes::Type geometryType,
129 : : const QgsCoordinateReferenceSystem &srs,
130 : : const QString &driverName,
131 : : const QStringList &datasourceOptions,
132 : : const QStringList &layerOptions,
133 : : QString *newFilename,
134 : : QgsVectorFileWriter::SymbologyExport symbologyExport,
135 : : FieldValueConverter *fieldValueConverter,
136 : : const QString &layerName,
137 : : ActionOnExistingFile action,
138 : : QString *newLayer,
139 : : QgsCoordinateTransformContext transformContext,
140 : : QgsFeatureSink::SinkFlags sinkFlags,
141 : : FieldNameSource fieldNameSource
142 : : )
143 : 0 : : mError( NoError )
144 : 0 : , mWkbType( geometryType )
145 : 0 : , mSymbologyExport( symbologyExport )
146 : 0 : , mSymbologyScale( 1.0 )
147 : 0 : {
148 : 0 : init( vectorFileName, fileEncoding, fields, geometryType, srs, driverName,
149 : 0 : datasourceOptions, layerOptions, newFilename, fieldValueConverter,
150 : 0 : layerName, action, newLayer, sinkFlags, transformContext, fieldNameSource );
151 : 0 : }
152 : :
153 : 0 : QgsVectorFileWriter *QgsVectorFileWriter::create(
154 : : const QString &fileName,
155 : : const QgsFields &fields,
156 : : QgsWkbTypes::Type geometryType,
157 : : const QgsCoordinateReferenceSystem &srs,
158 : : const QgsCoordinateTransformContext &transformContext,
159 : : const QgsVectorFileWriter::SaveVectorOptions &options,
160 : : QgsFeatureSink::SinkFlags sinkFlags,
161 : : QString *newFilename,
162 : : QString *newLayer
163 : : )
164 : : {
165 : : Q_NOWARN_DEPRECATED_PUSH
166 : 0 : return new QgsVectorFileWriter( fileName, options.fileEncoding, fields, geometryType, srs,
167 : 0 : options.driverName, options.datasourceOptions, options.layerOptions,
168 : 0 : newFilename, options.symbologyExport, options.fieldValueConverter, options.layerName,
169 : 0 : options.actionOnExistingFile, newLayer, transformContext, sinkFlags, options.fieldNameSource );
170 : : Q_NOWARN_DEPRECATED_POP
171 : 0 : }
172 : :
173 : 0 : bool QgsVectorFileWriter::supportsFeatureStyles( const QString &driverName )
174 : : {
175 : 0 : if ( driverName == QLatin1String( "MapInfo MIF" ) )
176 : : {
177 : 0 : return true;
178 : : }
179 : 0 : GDALDriverH gdalDriver = GDALGetDriverByName( driverName.toLocal8Bit().constData() );
180 : 0 : if ( !gdalDriver )
181 : 0 : return false;
182 : :
183 : 0 : char **driverMetadata = GDALGetMetadata( gdalDriver, nullptr );
184 : 0 : if ( !driverMetadata )
185 : 0 : return false;
186 : :
187 : 0 : return CSLFetchBoolean( driverMetadata, GDAL_DCAP_FEATURE_STYLES, false );
188 : 0 : }
189 : :
190 : 0 : void QgsVectorFileWriter::init( QString vectorFileName,
191 : : QString fileEncoding,
192 : : const QgsFields &fields,
193 : : QgsWkbTypes::Type geometryType,
194 : : QgsCoordinateReferenceSystem srs,
195 : : const QString &driverName,
196 : : QStringList datasourceOptions,
197 : : QStringList layerOptions,
198 : : QString *newFilename,
199 : : FieldValueConverter *fieldValueConverter,
200 : : const QString &layerNameIn,
201 : : ActionOnExistingFile action,
202 : : QString *newLayer, SinkFlags sinkFlags,
203 : : const QgsCoordinateTransformContext &transformContext, FieldNameSource fieldNameSource )
204 : : {
205 : 0 : mRenderContext.setRendererScale( mSymbologyScale );
206 : :
207 : 0 : if ( vectorFileName.isEmpty() )
208 : : {
209 : 0 : mErrorMessage = QObject::tr( "Empty filename given" );
210 : 0 : mError = ErrCreateDataSource;
211 : 0 : return;
212 : : }
213 : :
214 : 0 : if ( driverName == QLatin1String( "MapInfo MIF" ) )
215 : : {
216 : 0 : mOgrDriverName = QStringLiteral( "MapInfo File" );
217 : 0 : }
218 : 0 : else if ( driverName == QLatin1String( "SpatiaLite" ) )
219 : : {
220 : 0 : mOgrDriverName = QStringLiteral( "SQLite" );
221 : 0 : if ( !datasourceOptions.contains( QStringLiteral( "SPATIALITE=YES" ) ) )
222 : : {
223 : 0 : datasourceOptions.append( QStringLiteral( "SPATIALITE=YES" ) );
224 : 0 : }
225 : 0 : }
226 : 0 : else if ( driverName == QLatin1String( "DBF file" ) )
227 : : {
228 : 0 : mOgrDriverName = QStringLiteral( "ESRI Shapefile" );
229 : 0 : if ( !layerOptions.contains( QStringLiteral( "SHPT=NULL" ) ) )
230 : : {
231 : 0 : layerOptions.append( QStringLiteral( "SHPT=NULL" ) );
232 : 0 : }
233 : 0 : srs = QgsCoordinateReferenceSystem();
234 : 0 : }
235 : : else
236 : : {
237 : 0 : mOgrDriverName = driverName;
238 : : }
239 : :
240 : : // find driver in OGR
241 : : OGRSFDriverH poDriver;
242 : 0 : QgsApplication::registerOgrDrivers();
243 : :
244 : 0 : poDriver = OGRGetDriverByName( mOgrDriverName.toLocal8Bit().constData() );
245 : :
246 : 0 : if ( !poDriver )
247 : : {
248 : 0 : mErrorMessage = QObject::tr( "OGR driver for '%1' not found (OGR error: %2)" )
249 : 0 : .arg( driverName,
250 : 0 : QString::fromUtf8( CPLGetLastErrorMsg() ) );
251 : 0 : mError = ErrDriverNotFound;
252 : 0 : return;
253 : : }
254 : :
255 : 0 : MetaData metadata;
256 : 0 : bool metadataFound = driverMetadata( driverName, metadata );
257 : :
258 : 0 : if ( mOgrDriverName == QLatin1String( "ESRI Shapefile" ) )
259 : : {
260 : 0 : if ( layerOptions.join( QString() ).toUpper().indexOf( QLatin1String( "ENCODING=" ) ) == -1 )
261 : : {
262 : 0 : layerOptions.append( "ENCODING=" + convertCodecNameForEncodingOption( fileEncoding ) );
263 : 0 : }
264 : :
265 : 0 : if ( driverName == QLatin1String( "ESRI Shapefile" ) && !vectorFileName.endsWith( QLatin1String( ".shp" ), Qt::CaseInsensitive ) )
266 : : {
267 : 0 : vectorFileName += QLatin1String( ".shp" );
268 : 0 : }
269 : 0 : else if ( driverName == QLatin1String( "DBF file" ) && !vectorFileName.endsWith( QLatin1String( ".dbf" ), Qt::CaseInsensitive ) )
270 : : {
271 : 0 : vectorFileName += QLatin1String( ".dbf" );
272 : 0 : }
273 : :
274 : 0 : if ( action == CreateOrOverwriteFile || action == CreateOrOverwriteLayer )
275 : 0 : deleteShapeFile( vectorFileName );
276 : 0 : }
277 : : else
278 : : {
279 : 0 : if ( metadataFound )
280 : : {
281 : : #if QT_VERSION < QT_VERSION_CHECK(5, 15, 0)
282 : : QStringList allExts = metadata.ext.split( ' ', QString::SkipEmptyParts );
283 : : #else
284 : 0 : QStringList allExts = metadata.ext.split( ' ', Qt::SkipEmptyParts );
285 : : #endif
286 : 0 : bool found = false;
287 : 0 : const auto constAllExts = allExts;
288 : 0 : for ( const QString &ext : constAllExts )
289 : : {
290 : 0 : if ( vectorFileName.endsWith( '.' + ext, Qt::CaseInsensitive ) )
291 : : {
292 : 0 : found = true;
293 : 0 : break;
294 : : }
295 : : }
296 : :
297 : 0 : if ( !found )
298 : : {
299 : 0 : vectorFileName += '.' + allExts[0];
300 : 0 : }
301 : 0 : }
302 : :
303 : 0 : if ( action == CreateOrOverwriteFile )
304 : : {
305 : 0 : if ( vectorFileName.endsWith( QLatin1String( ".gdb" ), Qt::CaseInsensitive ) )
306 : : {
307 : 0 : QDir dir( vectorFileName );
308 : 0 : if ( dir.exists() )
309 : : {
310 : 0 : QFileInfoList fileList = dir.entryInfoList(
311 : 0 : QDir::NoDotAndDotDot | QDir::System | QDir::Hidden | QDir::AllDirs | QDir::Files, QDir::DirsFirst );
312 : 0 : const auto constFileList = fileList;
313 : 0 : for ( const QFileInfo &info : constFileList )
314 : : {
315 : 0 : QFile::remove( info.absoluteFilePath() );
316 : : }
317 : 0 : }
318 : 0 : QDir().rmdir( vectorFileName );
319 : 0 : }
320 : : else
321 : : {
322 : 0 : QFile::remove( vectorFileName );
323 : : }
324 : 0 : }
325 : : }
326 : :
327 : 0 : if ( metadataFound && !metadata.compulsoryEncoding.isEmpty() )
328 : : {
329 : 0 : if ( fileEncoding.compare( metadata.compulsoryEncoding, Qt::CaseInsensitive ) != 0 )
330 : : {
331 : 0 : QgsDebugMsgLevel( QStringLiteral( "forced %1 encoding for %2" ).arg( metadata.compulsoryEncoding, driverName ), 2 );
332 : 0 : fileEncoding = metadata.compulsoryEncoding;
333 : 0 : }
334 : :
335 : 0 : }
336 : :
337 : 0 : char **options = nullptr;
338 : 0 : if ( !datasourceOptions.isEmpty() )
339 : : {
340 : 0 : options = new char *[ datasourceOptions.size() + 1 ];
341 : 0 : for ( int i = 0; i < datasourceOptions.size(); i++ )
342 : : {
343 : 0 : QgsDebugMsgLevel( QStringLiteral( "-dsco=%1" ).arg( datasourceOptions[i] ), 2 );
344 : 0 : options[i] = CPLStrdup( datasourceOptions[i].toLocal8Bit().constData() );
345 : 0 : }
346 : 0 : options[ datasourceOptions.size()] = nullptr;
347 : 0 : }
348 : 0 : mAttrIdxToOgrIdx.remove( 0 );
349 : :
350 : : // create the data source
351 : 0 : if ( action == CreateOrOverwriteFile )
352 : 0 : mDS.reset( OGR_Dr_CreateDataSource( poDriver, vectorFileName.toUtf8().constData(), options ) );
353 : : else
354 : 0 : mDS.reset( myOGROpen( vectorFileName.toUtf8().constData(), TRUE, nullptr ) );
355 : :
356 : 0 : if ( options )
357 : : {
358 : 0 : for ( int i = 0; i < datasourceOptions.size(); i++ )
359 : 0 : CPLFree( options[i] );
360 : 0 : delete [] options;
361 : 0 : options = nullptr;
362 : 0 : }
363 : :
364 : 0 : if ( !mDS )
365 : : {
366 : 0 : mError = ErrCreateDataSource;
367 : 0 : if ( action == CreateOrOverwriteFile )
368 : 0 : mErrorMessage = QObject::tr( "Creation of data source failed (OGR error: %1)" )
369 : 0 : .arg( QString::fromUtf8( CPLGetLastErrorMsg() ) );
370 : : else
371 : 0 : mErrorMessage = QObject::tr( "Opening of data source in update mode failed (OGR error: %1)" )
372 : 0 : .arg( QString::fromUtf8( CPLGetLastErrorMsg() ) );
373 : 0 : return;
374 : : }
375 : :
376 : 0 : QString layerName( layerNameIn );
377 : 0 : if ( layerName.isEmpty() )
378 : 0 : layerName = QFileInfo( vectorFileName ).baseName();
379 : :
380 : 0 : if ( action == CreateOrOverwriteLayer )
381 : : {
382 : 0 : const int layer_count = OGR_DS_GetLayerCount( mDS.get() );
383 : 0 : for ( int i = 0; i < layer_count; i++ )
384 : : {
385 : 0 : OGRLayerH hLayer = OGR_DS_GetLayer( mDS.get(), i );
386 : 0 : if ( EQUAL( OGR_L_GetName( hLayer ), layerName.toUtf8().constData() ) )
387 : : {
388 : 0 : if ( OGR_DS_DeleteLayer( mDS.get(), i ) != OGRERR_NONE )
389 : : {
390 : 0 : mError = ErrCreateLayer;
391 : 0 : mErrorMessage = QObject::tr( "Overwriting of existing layer failed (OGR error: %1)" )
392 : 0 : .arg( QString::fromUtf8( CPLGetLastErrorMsg() ) );
393 : 0 : return;
394 : : }
395 : 0 : break;
396 : : }
397 : 0 : }
398 : 0 : }
399 : :
400 : 0 : if ( action == CreateOrOverwriteFile )
401 : : {
402 : 0 : QgsDebugMsgLevel( QStringLiteral( "Created data source" ), 2 );
403 : 0 : }
404 : : else
405 : : {
406 : 0 : QgsDebugMsgLevel( QStringLiteral( "Opened data source in update mode" ), 2 );
407 : : }
408 : :
409 : : // use appropriate codec
410 : 0 : mCodec = QTextCodec::codecForName( fileEncoding.toLocal8Bit().constData() );
411 : 0 : if ( !mCodec )
412 : : {
413 : 0 : QgsDebugMsg( "error finding QTextCodec for " + fileEncoding );
414 : :
415 : 0 : QgsSettings settings;
416 : 0 : QString enc = settings.value( QStringLiteral( "UI/encoding" ), "System" ).toString();
417 : 0 : mCodec = QTextCodec::codecForName( enc.toLocal8Bit().constData() );
418 : 0 : if ( !mCodec )
419 : : {
420 : 0 : QgsDebugMsg( "error finding QTextCodec for " + enc );
421 : 0 : mCodec = QTextCodec::codecForLocale();
422 : : Q_ASSERT( mCodec );
423 : 0 : }
424 : 0 : }
425 : :
426 : : // consider spatial reference system of the layer
427 : 0 : if ( driverName == QLatin1String( "KML" ) || driverName == QLatin1String( "LIBKML" ) || driverName == QLatin1String( "GPX" ) )
428 : : {
429 : 0 : if ( srs.authid() != QLatin1String( "EPSG:4326" ) )
430 : : {
431 : : // Those drivers outputs WGS84 geometries, let's align our output CRS to have QGIS take charge of geometry transformation
432 : 0 : QgsCoordinateReferenceSystem wgs84 = QgsCoordinateReferenceSystem::fromEpsgId( 4326 );
433 : 0 : mCoordinateTransform.reset( new QgsCoordinateTransform( srs, wgs84, transformContext ) );
434 : 0 : srs = wgs84;
435 : 0 : }
436 : 0 : }
437 : :
438 : 0 : if ( srs.isValid() )
439 : : {
440 : 0 : QString srsWkt = srs.toWkt( QgsCoordinateReferenceSystem::WKT_PREFERRED_GDAL );
441 : 0 : QgsDebugMsgLevel( "WKT to save as is " + srsWkt, 2 );
442 : 0 : mOgrRef = OSRNewSpatialReference( srsWkt.toLocal8Bit().constData() );
443 : : #if GDAL_VERSION_MAJOR >= 3
444 : 0 : if ( mOgrRef )
445 : : {
446 : 0 : OSRSetAxisMappingStrategy( mOgrRef, OAMS_TRADITIONAL_GIS_ORDER );
447 : 0 : }
448 : : #endif
449 : 0 : }
450 : :
451 : : // datasource created, now create the output layer
452 : 0 : OGRwkbGeometryType wkbType = ogrTypeFromWkbType( geometryType );
453 : :
454 : : // Remove FEATURE_DATASET layer option (used for ESRI File GDB driver) if its value is not set
455 : 0 : int optIndex = layerOptions.indexOf( QLatin1String( "FEATURE_DATASET=" ) );
456 : 0 : if ( optIndex != -1 )
457 : : {
458 : 0 : layerOptions.removeAt( optIndex );
459 : 0 : }
460 : :
461 : 0 : if ( !layerOptions.isEmpty() )
462 : : {
463 : 0 : options = new char *[ layerOptions.size() + 1 ];
464 : 0 : for ( int i = 0; i < layerOptions.size(); i++ )
465 : : {
466 : 0 : QgsDebugMsgLevel( QStringLiteral( "-lco=%1" ).arg( layerOptions[i] ), 2 );
467 : 0 : options[i] = CPLStrdup( layerOptions[i].toLocal8Bit().constData() );
468 : 0 : }
469 : 0 : options[ layerOptions.size()] = nullptr;
470 : 0 : }
471 : :
472 : : // disable encoding conversion of OGR Shapefile layer
473 : 0 : CPLSetConfigOption( "SHAPE_ENCODING", "" );
474 : :
475 : 0 : if ( action == CreateOrOverwriteFile || action == CreateOrOverwriteLayer )
476 : : {
477 : 0 : mLayer = OGR_DS_CreateLayer( mDS.get(), layerName.toUtf8().constData(), mOgrRef, wkbType, options );
478 : 0 : if ( newLayer && mLayer )
479 : : {
480 : 0 : *newLayer = OGR_L_GetName( mLayer );
481 : 0 : if ( driverName == QLatin1String( "GPX" ) )
482 : 0 : {
483 : : // See logic in GDAL ogr/ogrsf_frmts/gpx/ogrgpxdatasource.cpp ICreateLayer()
484 : 0 : switch ( QgsWkbTypes::flatType( geometryType ) )
485 : : {
486 : : case QgsWkbTypes::Point:
487 : : {
488 : 0 : if ( !EQUAL( layerName.toUtf8().constData(), "track_points" ) &&
489 : 0 : !EQUAL( layerName.toUtf8().constData(), "route_points" ) )
490 : : {
491 : 0 : *newLayer = QStringLiteral( "waypoints" );
492 : 0 : }
493 : : }
494 : 0 : break;
495 : :
496 : : case QgsWkbTypes::LineString:
497 : 0 : {
498 : 0 : const char *pszForceGPXTrack
499 : 0 : = CSLFetchNameValue( options, "FORCE_GPX_TRACK" );
500 : 0 : if ( pszForceGPXTrack && CPLTestBool( pszForceGPXTrack ) )
501 : 0 : *newLayer = QStringLiteral( "tracks" );
502 : : else
503 : 0 : *newLayer = QStringLiteral( "routes" );
504 : :
505 : : }
506 : 0 : break;
507 : :
508 : : case QgsWkbTypes::MultiLineString:
509 : 0 : {
510 : 0 : const char *pszForceGPXRoute
511 : 0 : = CSLFetchNameValue( options, "FORCE_GPX_ROUTE" );
512 : 0 : if ( pszForceGPXRoute && CPLTestBool( pszForceGPXRoute ) )
513 : 0 : *newLayer = QStringLiteral( "routes" );
514 : : else
515 : 0 : *newLayer = QStringLiteral( "tracks" );
516 : : }
517 : 0 : break;
518 : :
519 : : default:
520 : 0 : break;
521 : : }
522 : 0 : }
523 : 0 : }
524 : 0 : }
525 : 0 : else if ( driverName == QLatin1String( "DGN" ) )
526 : 0 : {
527 : 0 : mLayer = OGR_DS_GetLayerByName( mDS.get(), "elements" );
528 : 0 : }
529 : : else
530 : : {
531 : 0 : mLayer = OGR_DS_GetLayerByName( mDS.get(), layerName.toUtf8().constData() );
532 : : }
533 : 0 :
534 : 0 : if ( options )
535 : : {
536 : 0 : for ( int i = 0; i < layerOptions.size(); i++ )
537 : 0 : CPLFree( options[i] );
538 : 0 : delete [] options;
539 : 0 : options = nullptr;
540 : 0 : }
541 : :
542 : 0 : if ( srs.isValid() )
543 : : {
544 : 0 : if ( mOgrDriverName == QLatin1String( "ESRI Shapefile" ) )
545 : : {
546 : 0 : QString layerName = vectorFileName.left( vectorFileName.indexOf( QLatin1String( ".shp" ), Qt::CaseInsensitive ) );
547 : 0 : QFile prjFile( layerName + ".qpj" );
548 : 0 : if ( prjFile.exists() )
549 : 0 : prjFile.remove();
550 : 0 : }
551 : 0 : }
552 : :
553 : 0 : if ( !mLayer )
554 : : {
555 : 0 : if ( action == CreateOrOverwriteFile || action == CreateOrOverwriteLayer )
556 : 0 : mErrorMessage = QObject::tr( "Creation of layer failed (OGR error: %1)" )
557 : 0 : .arg( QString::fromUtf8( CPLGetLastErrorMsg() ) );
558 : : else
559 : 0 : mErrorMessage = QObject::tr( "Opening of layer failed (OGR error: %1)" )
560 : 0 : .arg( QString::fromUtf8( CPLGetLastErrorMsg() ) );
561 : 0 : mError = ErrCreateLayer;
562 : 0 : return;
563 : : }
564 : :
565 : 0 : OGRFeatureDefnH defn = OGR_L_GetLayerDefn( mLayer );
566 : :
567 : 0 : QgsDebugMsgLevel( QStringLiteral( "created layer" ), 2 );
568 : :
569 : : // create the fields
570 : 0 : QgsDebugMsgLevel( "creating " + QString::number( fields.size() ) + " fields", 2 );
571 : :
572 : 0 : mFields = fields;
573 : 0 : mAttrIdxToOgrIdx.clear();
574 : 0 : QSet<int> existingIdxs;
575 : :
576 : 0 : mFieldValueConverter = fieldValueConverter;
577 : :
578 : 0 : switch ( action )
579 : : {
580 : : case CreateOrOverwriteFile:
581 : : case CreateOrOverwriteLayer:
582 : : case AppendToLayerAddFields:
583 : : {
584 : 0 : for ( int fldIdx = 0; fldIdx < fields.count(); ++fldIdx )
585 : : {
586 : 0 : QgsField attrField = fields.at( fldIdx );
587 : :
588 : 0 : if ( fieldValueConverter )
589 : : {
590 : 0 : attrField = fieldValueConverter->fieldDefinition( fields.at( fldIdx ) );
591 : 0 : }
592 : :
593 : 0 : if ( action == AppendToLayerAddFields )
594 : : {
595 : 0 : int ogrIdx = OGR_FD_GetFieldIndex( defn, mCodec->fromUnicode( attrField.name() ) );
596 : 0 : if ( ogrIdx >= 0 )
597 : : {
598 : 0 : mAttrIdxToOgrIdx.insert( fldIdx, ogrIdx );
599 : 0 : continue;
600 : : }
601 : 0 : }
602 : :
603 : 0 : QString name;
604 : 0 : switch ( fieldNameSource )
605 : : {
606 : : case Original:
607 : 0 : name = attrField.name();
608 : 0 : break;
609 : :
610 : : case PreferAlias:
611 : 0 : name = !attrField.alias().isEmpty() ? attrField.alias() : attrField.name();
612 : 0 : break;
613 : : }
614 : :
615 : 0 : OGRFieldType ogrType = OFTString; //default to string
616 : 0 : int ogrWidth = attrField.length();
617 : 0 : int ogrPrecision = attrField.precision();
618 : 0 : if ( ogrPrecision > 0 )
619 : 0 : ++ogrWidth;
620 : :
621 : 0 : switch ( attrField.type() )
622 : : {
623 : : case QVariant::LongLong:
624 : : {
625 : 0 : const char *pszDataTypes = GDALGetMetadataItem( poDriver, GDAL_DMD_CREATIONFIELDDATATYPES, nullptr );
626 : 0 : if ( pszDataTypes && strstr( pszDataTypes, "Integer64" ) )
627 : 0 : ogrType = OFTInteger64;
628 : : else
629 : 0 : ogrType = OFTReal;
630 : 0 : ogrWidth = ogrWidth > 0 && ogrWidth <= 20 ? ogrWidth : 20;
631 : 0 : ogrPrecision = 0;
632 : 0 : break;
633 : : }
634 : : case QVariant::String:
635 : 0 : ogrType = OFTString;
636 : 0 : if ( ( ogrWidth <= 0 || ogrWidth > 255 ) && mOgrDriverName == QLatin1String( "ESRI Shapefile" ) )
637 : 0 : ogrWidth = 255;
638 : 0 : break;
639 : :
640 : : case QVariant::Int:
641 : 0 : ogrType = OFTInteger;
642 : 0 : ogrWidth = ogrWidth > 0 && ogrWidth <= 10 ? ogrWidth : 10;
643 : 0 : ogrPrecision = 0;
644 : 0 : break;
645 : :
646 : : case QVariant::Bool:
647 : 0 : ogrType = OFTInteger;
648 : 0 : ogrWidth = 1;
649 : 0 : ogrPrecision = 0;
650 : 0 : break;
651 : :
652 : : case QVariant::Double:
653 : 0 : ogrType = OFTReal;
654 : 0 : break;
655 : :
656 : : case QVariant::Date:
657 : 0 : ogrType = OFTDate;
658 : 0 : break;
659 : :
660 : : case QVariant::Time:
661 : 0 : if ( mOgrDriverName == QLatin1String( "ESRI Shapefile" ) )
662 : : {
663 : 0 : ogrType = OFTString;
664 : 0 : ogrWidth = 12; // %02d:%02d:%06.3f
665 : 0 : }
666 : : else
667 : : {
668 : 0 : ogrType = OFTTime;
669 : : }
670 : 0 : break;
671 : :
672 : : case QVariant::DateTime:
673 : 0 : if ( mOgrDriverName == QLatin1String( "ESRI Shapefile" ) )
674 : : {
675 : 0 : ogrType = OFTString;
676 : 0 : ogrWidth = 24; // "%04d/%02d/%02d %02d:%02d:%06.3f"
677 : 0 : }
678 : : else
679 : : {
680 : 0 : ogrType = OFTDateTime;
681 : : }
682 : 0 : break;
683 : :
684 : : case QVariant::ByteArray:
685 : 0 : ogrType = OFTBinary;
686 : 0 : break;
687 : :
688 : : case QVariant::List:
689 : : // only string list supported at the moment, fall through to default for other types
690 : 0 : if ( attrField.subType() == QVariant::String )
691 : : {
692 : 0 : const char *pszDataTypes = GDALGetMetadataItem( poDriver, GDAL_DMD_CREATIONFIELDDATATYPES, nullptr );
693 : 0 : if ( pszDataTypes && strstr( pszDataTypes, "StringList" ) )
694 : : {
695 : 0 : ogrType = OFTStringList;
696 : 0 : supportsStringList = true;
697 : 0 : }
698 : : else
699 : : {
700 : 0 : ogrType = OFTString;
701 : 0 : ogrWidth = 255;
702 : : }
703 : 0 : break;
704 : : }
705 : : //intentional fall-through
706 : : FALLTHROUGH
707 : :
708 : : default:
709 : : //assert(0 && "invalid variant type!");
710 : 0 : mErrorMessage = QObject::tr( "Unsupported type for field %1" )
711 : 0 : .arg( attrField.name() );
712 : 0 : mError = ErrAttributeTypeUnsupported;
713 : 0 : return;
714 : : }
715 : :
716 : 0 : if ( mOgrDriverName == QLatin1String( "SQLite" ) && name.compare( QLatin1String( "ogc_fid" ), Qt::CaseInsensitive ) == 0 )
717 : : {
718 : : int i;
719 : 0 : for ( i = 0; i < 10; i++ )
720 : : {
721 : 0 : name = QStringLiteral( "ogc_fid%1" ).arg( i );
722 : :
723 : : int j;
724 : 0 : for ( j = 0; j < fields.size() && name.compare( fields.at( j ).name(), Qt::CaseInsensitive ) != 0; j++ )
725 : : ;
726 : :
727 : 0 : if ( j == fields.size() )
728 : 0 : break;
729 : 0 : }
730 : :
731 : 0 : if ( i == 10 )
732 : : {
733 : 0 : mErrorMessage = QObject::tr( "No available replacement for internal fieldname ogc_fid found" ).arg( attrField.name() );
734 : 0 : mError = ErrAttributeCreationFailed;
735 : 0 : return;
736 : : }
737 : :
738 : 0 : QgsMessageLog::logMessage( QObject::tr( "Reserved attribute name ogc_fid replaced with %1" ).arg( name ), QObject::tr( "OGR" ) );
739 : 0 : }
740 : :
741 : : // create field definition
742 : 0 : gdal::ogr_field_def_unique_ptr fld( OGR_Fld_Create( mCodec->fromUnicode( name ), ogrType ) );
743 : 0 : if ( ogrWidth > 0 )
744 : : {
745 : 0 : OGR_Fld_SetWidth( fld.get(), ogrWidth );
746 : 0 : }
747 : :
748 : 0 : if ( ogrPrecision >= 0 )
749 : : {
750 : 0 : OGR_Fld_SetPrecision( fld.get(), ogrPrecision );
751 : 0 : }
752 : :
753 : 0 : switch ( attrField.type() )
754 : : {
755 : : case QVariant::Bool:
756 : 0 : OGR_Fld_SetSubType( fld.get(), OFSTBoolean );
757 : 0 : break;
758 : : default:
759 : 0 : break;
760 : : }
761 : :
762 : : // create the field
763 : 0 : QgsDebugMsgLevel( "creating field " + attrField.name() +
764 : : " type " + QString( QVariant::typeToName( attrField.type() ) ) +
765 : : " width " + QString::number( ogrWidth ) +
766 : : " precision " + QString::number( ogrPrecision ), 2 );
767 : 0 : if ( OGR_L_CreateField( mLayer, fld.get(), true ) != OGRERR_NONE )
768 : : {
769 : 0 : QgsDebugMsg( "error creating field " + attrField.name() );
770 : 0 : mErrorMessage = QObject::tr( "Creation of field %1 failed (OGR error: %2)" )
771 : 0 : .arg( attrField.name(),
772 : 0 : QString::fromUtf8( CPLGetLastErrorMsg() ) );
773 : 0 : mError = ErrAttributeCreationFailed;
774 : 0 : return;
775 : : }
776 : :
777 : 0 : int ogrIdx = OGR_FD_GetFieldIndex( defn, mCodec->fromUnicode( name ) );
778 : 0 : QgsDebugMsgLevel( QStringLiteral( "returned field index for %1: %2" ).arg( name ).arg( ogrIdx ), 2 );
779 : 0 : if ( ogrIdx < 0 || existingIdxs.contains( ogrIdx ) )
780 : : {
781 : : // GDAL 1.7 not just truncates, but launders more aggressivly.
782 : 0 : ogrIdx = OGR_FD_GetFieldCount( defn ) - 1;
783 : :
784 : 0 : if ( ogrIdx < 0 )
785 : : {
786 : 0 : QgsDebugMsg( "error creating field " + attrField.name() );
787 : 0 : mErrorMessage = QObject::tr( "Created field %1 not found (OGR error: %2)" )
788 : 0 : .arg( attrField.name(),
789 : 0 : QString::fromUtf8( CPLGetLastErrorMsg() ) );
790 : 0 : mError = ErrAttributeCreationFailed;
791 : 0 : return;
792 : : }
793 : 0 : }
794 : :
795 : 0 : existingIdxs.insert( ogrIdx );
796 : 0 : mAttrIdxToOgrIdx.insert( fldIdx, ogrIdx );
797 : 0 : }
798 : : }
799 : 0 : break;
800 : :
801 : : case AppendToLayerNoNewFields:
802 : : {
803 : 0 : for ( int fldIdx = 0; fldIdx < fields.count(); ++fldIdx )
804 : : {
805 : 0 : QgsField attrField = fields.at( fldIdx );
806 : 0 : QString name( attrField.name() );
807 : 0 : int ogrIdx = OGR_FD_GetFieldIndex( defn, mCodec->fromUnicode( name ) );
808 : 0 : if ( ogrIdx >= 0 )
809 : 0 : mAttrIdxToOgrIdx.insert( fldIdx, ogrIdx );
810 : 0 : }
811 : : }
812 : 0 : break;
813 : : }
814 : :
815 : : // Geopackages require a unique feature id. If the input feature stream cannot guarantee
816 : : // the uniqueness of the FID column, we drop it and let OGR generate new ones
817 : 0 : if ( sinkFlags.testFlag( QgsFeatureSink::RegeneratePrimaryKey ) && driverName == QLatin1String( "GPKG" ) )
818 : : {
819 : 0 : int fidIdx = fields.lookupField( QStringLiteral( "FID" ) );
820 : :
821 : 0 : if ( fidIdx >= 0 )
822 : 0 : mAttrIdxToOgrIdx.remove( fidIdx );
823 : 0 : }
824 : :
825 : 0 : QgsDebugMsgLevel( QStringLiteral( "Done creating fields" ), 2 );
826 : :
827 : 0 : mWkbType = geometryType;
828 : :
829 : 0 : if ( newFilename )
830 : 0 : *newFilename = vectorFileName;
831 : :
832 : : // enabling transaction on databases that support it
833 : 0 : mUsingTransaction = true;
834 : 0 : if ( OGRERR_NONE != OGR_L_StartTransaction( mLayer ) )
835 : : {
836 : 0 : mUsingTransaction = false;
837 : 0 : }
838 : 0 : }
839 : :
840 : 0 : OGRGeometryH QgsVectorFileWriter::createEmptyGeometry( QgsWkbTypes::Type wkbType )
841 : : {
842 : 0 : return OGR_G_CreateGeometry( ogrTypeFromWkbType( wkbType ) );
843 : : }
844 : :
845 : : ///@cond PRIVATE
846 : : class QgsVectorFileWriterMetadataContainer
847 : : {
848 : : public:
849 : :
850 : 0 : QgsVectorFileWriterMetadataContainer()
851 : : {
852 : 0 : QMap<QString, QgsVectorFileWriter::Option *> datasetOptions;
853 : 0 : QMap<QString, QgsVectorFileWriter::Option *> layerOptions;
854 : :
855 : : // Arc/Info ASCII Coverage
856 : 0 : datasetOptions.clear();
857 : 0 : layerOptions.clear();
858 : :
859 : 0 : driverMetadata.insert( QStringLiteral( "AVCE00" ),
860 : 0 : QgsVectorFileWriter::MetaData(
861 : 0 : QStringLiteral( "Arc/Info ASCII Coverage" ),
862 : 0 : QObject::tr( "Arc/Info ASCII Coverage" ),
863 : 0 : QStringLiteral( "*.e00" ),
864 : 0 : QStringLiteral( "e00" ),
865 : : datasetOptions,
866 : : layerOptions
867 : : )
868 : : );
869 : :
870 : : // Atlas BNA
871 : 0 : datasetOptions.clear();
872 : 0 : layerOptions.clear();
873 : :
874 : 0 : datasetOptions.insert( QStringLiteral( "LINEFORMAT" ), new QgsVectorFileWriter::SetOption(
875 : 0 : QObject::tr( "New BNA files are created by the "
876 : : "systems default line termination conventions. "
877 : : "This may be overridden here." ),
878 : 0 : QStringList()
879 : 0 : << QStringLiteral( "CRLF" )
880 : 0 : << QStringLiteral( "LF" ),
881 : 0 : QString(), // Default value
882 : 0 : true // Allow None
883 : 0 : ) );
884 : :
885 : 0 : datasetOptions.insert( QStringLiteral( "MULTILINE" ), new QgsVectorFileWriter::BoolOption(
886 : 0 : QObject::tr( "By default, BNA files are created in multi-line format. "
887 : : "For each record, the first line contains the identifiers and the "
888 : : "type/number of coordinates to follow. Each following line contains "
889 : : "a pair of coordinates." ),
890 : : true // Default value
891 : 0 : ) );
892 : :
893 : 0 : datasetOptions.insert( QStringLiteral( "NB_IDS" ), new QgsVectorFileWriter::SetOption(
894 : 0 : QObject::tr( "BNA records may contain from 2 to 4 identifiers per record. "
895 : : "Some software packages only support a precise number of identifiers. "
896 : : "You can override the default value (2) by a precise value." ),
897 : 0 : QStringList()
898 : 0 : << QStringLiteral( "2" )
899 : 0 : << QStringLiteral( "3" )
900 : 0 : << QStringLiteral( "4" )
901 : 0 : << QStringLiteral( "NB_SOURCE_FIELDS" ),
902 : 0 : QStringLiteral( "2" ) // Default value
903 : : ) );
904 : :
905 : 0 : datasetOptions.insert( QStringLiteral( "ELLIPSES_AS_ELLIPSES" ), new QgsVectorFileWriter::BoolOption(
906 : 0 : QObject::tr( "The BNA writer will try to recognize ellipses and circles when writing a polygon. "
907 : : "This will only work if the feature has previously been read from a BNA file. "
908 : : "As some software packages do not support ellipses/circles in BNA data file, "
909 : 0 : "it may be useful to tell the writer by specifying ELLIPSES_AS_ELLIPSES=NO not "
910 : : "to export them as such, but keep them as polygons." ),
911 : : true // Default value
912 : : ) );
913 : :
914 : 0 : datasetOptions.insert( QStringLiteral( "NB_PAIRS_PER_LINE" ), new QgsVectorFileWriter::IntOption(
915 : 0 : QObject::tr( "Limit the number of coordinate pairs per line in multiline format." ),
916 : : 2 // Default value
917 : : ) );
918 : :
919 : 0 : datasetOptions.insert( QStringLiteral( "COORDINATE_PRECISION" ), new QgsVectorFileWriter::IntOption(
920 : 0 : QObject::tr( "Set the number of decimal for coordinates. Default value is 10." ),
921 : : 10 // Default value
922 : : ) );
923 : :
924 : 0 : driverMetadata.insert( QStringLiteral( "BNA" ),
925 : 0 : QgsVectorFileWriter::MetaData(
926 : 0 : QStringLiteral( "Atlas BNA" ),
927 : 0 : QObject::tr( "Atlas BNA" ),
928 : 0 : QStringLiteral( "*.bna" ),
929 : 0 : QStringLiteral( "bna" ),
930 : : datasetOptions,
931 : : layerOptions
932 : : )
933 : : );
934 : :
935 : : // Comma Separated Value
936 : 0 : datasetOptions.clear();
937 : 0 : layerOptions.clear();
938 : :
939 : 0 : layerOptions.insert( QStringLiteral( "LINEFORMAT" ), new QgsVectorFileWriter::SetOption(
940 : 0 : QObject::tr( "By default when creating new .csv files they "
941 : : "are created with the line termination conventions "
942 : : "of the local platform (CR/LF on Win32 or LF on all other systems). "
943 : : "This may be overridden through the use of the LINEFORMAT option." ),
944 : 0 : QStringList()
945 : 0 : << QStringLiteral( "CRLF" )
946 : 0 : << QStringLiteral( "LF" ),
947 : 0 : QString(), // Default value
948 : : true // Allow None
949 : : ) );
950 : :
951 : 0 : layerOptions.insert( QStringLiteral( "GEOMETRY" ), new QgsVectorFileWriter::SetOption(
952 : 0 : QObject::tr( "By default, the geometry of a feature written to a .csv file is discarded. "
953 : : "It is possible to export the geometry in its WKT representation by "
954 : : "specifying GEOMETRY=AS_WKT. It is also possible to export point geometries "
955 : : "into their X,Y,Z components by specifying GEOMETRY=AS_XYZ, GEOMETRY=AS_XY "
956 : : "or GEOMETRY=AS_YX." ),
957 : 0 : QStringList()
958 : 0 : << QStringLiteral( "AS_WKT" )
959 : 0 : << QStringLiteral( "AS_XYZ" )
960 : 0 : << QStringLiteral( "AS_XY" )
961 : 0 : << QStringLiteral( "AS_YX" ),
962 : 0 : QString(), // Default value
963 : : true // Allow None
964 : : ) );
965 : :
966 : 0 : layerOptions.insert( QStringLiteral( "CREATE_CSVT" ), new QgsVectorFileWriter::BoolOption(
967 : 0 : QObject::tr( "Create the associated .csvt file to describe the type of each "
968 : : "column of the layer and its optional width and precision." ),
969 : : false // Default value
970 : : ) );
971 : :
972 : 0 : layerOptions.insert( QStringLiteral( "SEPARATOR" ), new QgsVectorFileWriter::SetOption(
973 : 0 : QObject::tr( "Field separator character." ),
974 : 0 : QStringList()
975 : 0 : << QStringLiteral( "COMMA" )
976 : 0 : << QStringLiteral( "SEMICOLON" )
977 : 0 : << QStringLiteral( "TAB" ),
978 : 0 : QStringLiteral( "COMMA" ) // Default value
979 : : ) );
980 : :
981 : 0 : layerOptions.insert( QStringLiteral( "STRING_QUOTING" ), new QgsVectorFileWriter::SetOption(
982 : 0 : QObject::tr( "Double-quote strings. IF_AMBIGUOUS means that string values that look like numbers will be quoted." ),
983 : 0 : QStringList()
984 : 0 : << QStringLiteral( "IF_NEEDED" )
985 : 0 : << QStringLiteral( "IF_AMBIGUOUS" )
986 : 0 : << QStringLiteral( "ALWAYS" ),
987 : 0 : QStringLiteral( "IF_AMBIGUOUS" ) // Default value
988 : : ) );
989 : :
990 : 0 : layerOptions.insert( QStringLiteral( "WRITE_BOM" ), new QgsVectorFileWriter::BoolOption(
991 : 0 : QObject::tr( "Write a UTF-8 Byte Order Mark (BOM) at the start of the file." ),
992 : : false // Default value
993 : : ) );
994 : :
995 : 0 : driverMetadata.insert( QStringLiteral( "CSV" ),
996 : 0 : QgsVectorFileWriter::MetaData(
997 : 0 : QStringLiteral( "Comma Separated Value [CSV]" ),
998 : 0 : QObject::tr( "Comma Separated Value [CSV]" ),
999 : 0 : QStringLiteral( "*.csv" ),
1000 : 0 : QStringLiteral( "csv" ),
1001 : 0 : datasetOptions,
1002 : : layerOptions
1003 : : )
1004 : : );
1005 : :
1006 : : #if defined(GDAL_COMPUTE_VERSION) && GDAL_VERSION_NUM >= GDAL_COMPUTE_VERSION(3,1,0)
1007 : : // FlatGeobuf
1008 : 0 : datasetOptions.clear();
1009 : 0 : layerOptions.clear();
1010 : :
1011 : 0 : driverMetadata.insert( QStringLiteral( "FlatGeobuf" ),
1012 : 0 : QgsVectorFileWriter::MetaData(
1013 : 0 : QStringLiteral( "FlatGeobuf" ),
1014 : 0 : QObject::tr( "FlatGeobuf" ),
1015 : 0 : QStringLiteral( "*.fgb" ),
1016 : 0 : QStringLiteral( "fgb" ),
1017 : : datasetOptions,
1018 : : layerOptions
1019 : : )
1020 : : );
1021 : : #endif
1022 : :
1023 : : // ESRI Shapefile
1024 : 0 : datasetOptions.clear();
1025 : 0 : layerOptions.clear();
1026 : :
1027 : 0 : layerOptions.insert( QStringLiteral( "SHPT" ), new QgsVectorFileWriter::SetOption(
1028 : 0 : QObject::tr( "Override the type of shapefile created. "
1029 : : "Can be one of NULL for a simple .dbf file with no .shp file, POINT, "
1030 : : "ARC, POLYGON or MULTIPOINT for 2D, or POINTZ, ARCZ, POLYGONZ or "
1031 : 0 : "MULTIPOINTZ for 3D;" ) +
1032 : 0 : QObject::tr( " POINTM, ARCM, POLYGONM or MULTIPOINTM for measured geometries"
1033 : : " and POINTZM, ARCZM, POLYGONZM or MULTIPOINTZM for 3D measured"
1034 : 0 : " geometries." ) +
1035 : 0 : QObject::tr( " MULTIPATCH files are supported since GDAL 2.2." ) +
1036 : : ""
1037 : 0 : , QStringList()
1038 : 0 : << QStringLiteral( "NULL" )
1039 : 0 : << QStringLiteral( "POINT" )
1040 : 0 : << QStringLiteral( "ARC" )
1041 : 0 : << QStringLiteral( "POLYGON" )
1042 : 0 : << QStringLiteral( "MULTIPOINT" )
1043 : 0 : << QStringLiteral( "POINTZ" )
1044 : 0 : << QStringLiteral( "ARCZ" )
1045 : 0 : << QStringLiteral( "POLYGONZ" )
1046 : 0 : << QStringLiteral( "MULTIPOINTZ" )
1047 : 0 : << QStringLiteral( "POINTM" )
1048 : 0 : << QStringLiteral( "ARCM" )
1049 : 0 : << QStringLiteral( "POLYGONM" )
1050 : 0 : << QStringLiteral( "MULTIPOINTM" )
1051 : 0 : << QStringLiteral( "POINTZM" )
1052 : 0 : << QStringLiteral( "ARCZM" )
1053 : 0 : << QStringLiteral( "POLYGONZM" )
1054 : 0 : << QStringLiteral( "MULTIPOINTZM" )
1055 : 0 : << QStringLiteral( "MULTIPATCH" )
1056 : 0 : << QString(),
1057 : 0 : QString(), // Default value
1058 : : true // Allow None
1059 : : ) );
1060 : :
1061 : : // there does not seem to be a reason to provide this option to the user again
1062 : : // as we set encoding for shapefiles based on "fileEncoding" parameter passed to the writer
1063 : : #if 0
1064 : : layerOptions.insert( "ENCODING", new QgsVectorFileWriter::SetOption(
1065 : : QObject::tr( "Set the encoding value in the DBF file. "
1066 : : "The default value is LDID/87. It is not clear "
1067 : : "what other values may be appropriate." ),
1068 : : QStringList()
1069 : : << "LDID/87",
1070 : : "LDID/87" // Default value
1071 : : ) );
1072 : : #endif
1073 : :
1074 : 0 : layerOptions.insert( QStringLiteral( "RESIZE" ), new QgsVectorFileWriter::BoolOption(
1075 : 0 : QObject::tr( "Set to YES to resize fields to their optimal size." ),
1076 : : false // Default value
1077 : : ) );
1078 : :
1079 : 0 : driverMetadata.insert( QStringLiteral( "ESRI" ),
1080 : 0 : QgsVectorFileWriter::MetaData(
1081 : 0 : QStringLiteral( "ESRI Shapefile" ),
1082 : 0 : QObject::tr( "ESRI Shapefile" ),
1083 : 0 : QStringLiteral( "*.shp" ),
1084 : 0 : QStringLiteral( "shp" ),
1085 : : datasetOptions,
1086 : : layerOptions
1087 : : )
1088 : : );
1089 : :
1090 : : // DBF File
1091 : 0 : datasetOptions.clear();
1092 : 0 : layerOptions.clear();
1093 : :
1094 : 0 : driverMetadata.insert( QStringLiteral( "DBF File" ),
1095 : 0 : QgsVectorFileWriter::MetaData(
1096 : 0 : QStringLiteral( "DBF File" ),
1097 : 0 : QObject::tr( "DBF File" ),
1098 : 0 : QStringLiteral( "*.dbf" ),
1099 : 0 : QStringLiteral( "dbf" ),
1100 : : datasetOptions,
1101 : : layerOptions
1102 : : )
1103 : : );
1104 : :
1105 : : // FMEObjects Gateway
1106 : 0 : datasetOptions.clear();
1107 : 0 : layerOptions.clear();
1108 : :
1109 : 0 : driverMetadata.insert( QStringLiteral( "FMEObjects Gateway" ),
1110 : 0 : QgsVectorFileWriter::MetaData(
1111 : 0 : QStringLiteral( "FMEObjects Gateway" ),
1112 : 0 : QObject::tr( "FMEObjects Gateway" ),
1113 : 0 : QStringLiteral( "*.fdd" ),
1114 : 0 : QStringLiteral( "fdd" ),
1115 : : datasetOptions,
1116 : : layerOptions
1117 : : )
1118 : : );
1119 : :
1120 : : // GeoJSON
1121 : 0 : datasetOptions.clear();
1122 : 0 : layerOptions.clear();
1123 : :
1124 : 0 : layerOptions.insert( QStringLiteral( "WRITE_BBOX" ), new QgsVectorFileWriter::BoolOption(
1125 : 0 : QObject::tr( "Set to YES to write a bbox property with the bounding box "
1126 : : "of the geometries at the feature and feature collection level." ),
1127 : : false // Default value
1128 : : ) );
1129 : :
1130 : 0 : layerOptions.insert( QStringLiteral( "COORDINATE_PRECISION" ), new QgsVectorFileWriter::IntOption(
1131 : 0 : QObject::tr( "Maximum number of figures after decimal separator to write in coordinates. "
1132 : : "Defaults to 15. Truncation will occur to remove trailing zeros." ),
1133 : : 15 // Default value
1134 : : ) );
1135 : :
1136 : 0 : layerOptions.insert( QStringLiteral( "RFC7946" ), new QgsVectorFileWriter::BoolOption(
1137 : 0 : QObject::tr( "Whether to use RFC 7946 standard. "
1138 : : "If disabled GeoJSON 2008 initial version will be used. "
1139 : : "Default is NO (thus GeoJSON 2008). See also Documentation (via Help button)" ),
1140 : : false // Default value
1141 : : ) );
1142 : :
1143 : 0 : driverMetadata.insert( QStringLiteral( "GeoJSON" ),
1144 : 0 : QgsVectorFileWriter::MetaData(
1145 : 0 : QStringLiteral( "GeoJSON" ),
1146 : 0 : QObject::tr( "GeoJSON" ),
1147 : 0 : QStringLiteral( "*.geojson" ),
1148 : 0 : QStringLiteral( "geojson" ),
1149 : : datasetOptions,
1150 : : layerOptions,
1151 : 0 : QStringLiteral( "UTF-8" )
1152 : : )
1153 : : );
1154 : :
1155 : : // GeoJSONSeq
1156 : 0 : datasetOptions.clear();
1157 : 0 : layerOptions.clear();
1158 : :
1159 : 0 : layerOptions.insert( QStringLiteral( "COORDINATE_PRECISION" ), new QgsVectorFileWriter::IntOption(
1160 : 0 : QObject::tr( "Maximum number of figures after decimal separator to write in coordinates. "
1161 : : "Defaults to 15. Truncation will occur to remove trailing zeros." ),
1162 : : 15 // Default value
1163 : : ) );
1164 : :
1165 : 0 : layerOptions.insert( QStringLiteral( "RS" ), new QgsVectorFileWriter::BoolOption(
1166 : 0 : QObject::tr( "Whether to start records with the RS=0x1E character (RFC 8142 standard). "
1167 : : "Defaults to NO: Newline Delimited JSON (geojsonl). \n"
1168 : : "If set to YES: RFC 8142 standard: GeoJSON Text Sequences (geojsons)." ),
1169 : : false // Default value = NO
1170 : : ) );
1171 : :
1172 : 0 : driverMetadata.insert( QStringLiteral( "GeoJSONSeq" ),
1173 : 0 : QgsVectorFileWriter::MetaData(
1174 : 0 : QStringLiteral( "GeoJSON - Newline Delimited" ),
1175 : 0 : QObject::tr( "GeoJSON - Newline Delimited" ),
1176 : 0 : QStringLiteral( "*.geojsonl *.geojsons *.json" ),
1177 : 0 : QStringLiteral( "json" ), // add json for now
1178 : : datasetOptions,
1179 : : layerOptions,
1180 : 0 : QStringLiteral( "UTF-8" )
1181 : : )
1182 : : );
1183 : :
1184 : : // GeoRSS
1185 : 0 : datasetOptions.clear();
1186 : 0 : layerOptions.clear();
1187 : :
1188 : 0 : datasetOptions.insert( QStringLiteral( "FORMAT" ), new QgsVectorFileWriter::SetOption(
1189 : 0 : QObject::tr( "whether the document must be in RSS 2.0 or Atom 1.0 format. "
1190 : : "Default value : RSS" ),
1191 : 0 : QStringList()
1192 : 0 : << QStringLiteral( "RSS" )
1193 : 0 : << QStringLiteral( "ATOM" ),
1194 : 0 : QStringLiteral( "RSS" ) // Default value
1195 : : ) );
1196 : :
1197 : 0 : datasetOptions.insert( QStringLiteral( "GEOM_DIALECT" ), new QgsVectorFileWriter::SetOption(
1198 : 0 : QObject::tr( "The encoding of location information. Default value : SIMPLE. "
1199 : : "W3C_GEO only supports point geometries. "
1200 : : "SIMPLE or W3C_GEO only support geometries in geographic WGS84 coordinates." ),
1201 : 0 : QStringList()
1202 : 0 : << QStringLiteral( "SIMPLE" )
1203 : 0 : << QStringLiteral( "GML" )
1204 : 0 : << QStringLiteral( "W3C_GEO" ),
1205 : 0 : QStringLiteral( "SIMPLE" ) // Default value
1206 : : ) );
1207 : :
1208 : 0 : datasetOptions.insert( QStringLiteral( "USE_EXTENSIONS" ), new QgsVectorFileWriter::BoolOption(
1209 : 0 : QObject::tr( "If defined to YES, extension fields will be written. "
1210 : : "If the field name not found in the base schema matches "
1211 : : "the foo_bar pattern, foo will be considered as the namespace "
1212 : : "of the element, and a <foo:bar> element will be written. "
1213 : : "Otherwise, elements will be written in the <ogr:> namespace." ),
1214 : : false // Default value
1215 : : ) );
1216 : :
1217 : 0 : datasetOptions.insert( QStringLiteral( "WRITE_HEADER_AND_FOOTER" ), new QgsVectorFileWriter::BoolOption(
1218 : 0 : QObject::tr( "If defined to NO, only <entry> or <item> elements will be written. "
1219 : : "The user will have to provide the appropriate header and footer of the document." ),
1220 : : true // Default value
1221 : : ) );
1222 : :
1223 : 0 : datasetOptions.insert( QStringLiteral( "HEADER" ), new QgsVectorFileWriter::StringOption(
1224 : 0 : QObject::tr( "XML content that will be put between the <channel> element and the "
1225 : : "first <item> element for a RSS document, or between the xml tag and "
1226 : : "the first <entry> element for an Atom document." ),
1227 : 0 : QString() // Default value
1228 : : ) );
1229 : :
1230 : 0 : datasetOptions.insert( QStringLiteral( "TITLE" ), new QgsVectorFileWriter::StringOption(
1231 : 0 : QObject::tr( "Value put inside the <title> element in the header. "
1232 : : "If not provided, a dummy value will be used as that element is compulsory." ),
1233 : 0 : QString() // Default value
1234 : : ) );
1235 : :
1236 : 0 : datasetOptions.insert( QStringLiteral( "DESCRIPTION" ), new QgsVectorFileWriter::StringOption(
1237 : 0 : QObject::tr( "Value put inside the <description> element in the header. "
1238 : : "If not provided, a dummy value will be used as that element is compulsory." ),
1239 : 0 : QString() // Default value
1240 : : ) );
1241 : :
1242 : 0 : datasetOptions.insert( QStringLiteral( "LINK" ), new QgsVectorFileWriter::StringOption(
1243 : 0 : QObject::tr( "Value put inside the <link> element in the header. "
1244 : : "If not provided, a dummy value will be used as that element is compulsory." ),
1245 : 0 : QString() // Default value
1246 : : ) );
1247 : :
1248 : 0 : datasetOptions.insert( QStringLiteral( "UPDATED" ), new QgsVectorFileWriter::StringOption(
1249 : 0 : QObject::tr( "Value put inside the <updated> element in the header. "
1250 : : "Should be formatted as a XML datetime. "
1251 : : "If not provided, a dummy value will be used as that element is compulsory." ),
1252 : 0 : QString() // Default value
1253 : : ) );
1254 : :
1255 : 0 : datasetOptions.insert( QStringLiteral( "AUTHOR_NAME" ), new QgsVectorFileWriter::StringOption(
1256 : 0 : QObject::tr( "Value put inside the <author><name> element in the header. "
1257 : : "If not provided, a dummy value will be used as that element is compulsory." ),
1258 : 0 : QString() // Default value
1259 : : ) );
1260 : :
1261 : 0 : datasetOptions.insert( QStringLiteral( "ID" ), new QgsVectorFileWriter::StringOption(
1262 : 0 : QObject::tr( "Value put inside the <id> element in the header. "
1263 : : "If not provided, a dummy value will be used as that element is compulsory." ),
1264 : 0 : QString() // Default value
1265 : : ) );
1266 : :
1267 : 0 : driverMetadata.insert( QStringLiteral( "GeoRSS" ),
1268 : 0 : QgsVectorFileWriter::MetaData(
1269 : 0 : QStringLiteral( "GeoRSS" ),
1270 : 0 : QObject::tr( "GeoRSS" ),
1271 : 0 : QStringLiteral( "*.xml" ),
1272 : 0 : QStringLiteral( "xml" ),
1273 : : datasetOptions,
1274 : : layerOptions,
1275 : 0 : QStringLiteral( "UTF-8" )
1276 : : )
1277 : : );
1278 : :
1279 : : // Geography Markup Language [GML]
1280 : 0 : datasetOptions.clear();
1281 : 0 : layerOptions.clear();
1282 : :
1283 : 0 : datasetOptions.insert( QStringLiteral( "XSISCHEMAURI" ), new QgsVectorFileWriter::StringOption(
1284 : 0 : QObject::tr( "If provided, this URI will be inserted as the schema location. "
1285 : : "Note that the schema file isn't actually accessed by OGR, so it "
1286 : : "is up to the user to ensure it will match the schema of the OGR "
1287 : : "produced GML data file." ),
1288 : 0 : QString() // Default value
1289 : : ) );
1290 : :
1291 : 0 : datasetOptions.insert( QStringLiteral( "XSISCHEMA" ), new QgsVectorFileWriter::SetOption(
1292 : 0 : QObject::tr( "This writes a GML application schema file to a corresponding "
1293 : : ".xsd file (with the same basename). If INTERNAL is used the "
1294 : : "schema is written within the GML file, but this is experimental "
1295 : : "and almost certainly not valid XML. "
1296 : : "OFF disables schema generation (and is implicit if XSISCHEMAURI is used)." ),
1297 : 0 : QStringList()
1298 : 0 : << QStringLiteral( "EXTERNAL" )
1299 : 0 : << QStringLiteral( "INTERNAL" )
1300 : 0 : << QStringLiteral( "OFF" ),
1301 : 0 : QStringLiteral( "EXTERNAL" ) // Default value
1302 : : ) );
1303 : :
1304 : 0 : datasetOptions.insert( QStringLiteral( "PREFIX" ), new QgsVectorFileWriter::StringOption(
1305 : 0 : QObject::tr( "This is the prefix for the application target namespace." ),
1306 : 0 : QStringLiteral( "ogr" ) // Default value
1307 : : ) );
1308 : :
1309 : 0 : datasetOptions.insert( QStringLiteral( "STRIP_PREFIX" ), new QgsVectorFileWriter::BoolOption(
1310 : 0 : QObject::tr( "Can be set to TRUE to avoid writing the prefix of the "
1311 : : "application target namespace in the GML file." ),
1312 : : false // Default value
1313 : : ) );
1314 : :
1315 : 0 : datasetOptions.insert( QStringLiteral( "TARGET_NAMESPACE" ), new QgsVectorFileWriter::StringOption(
1316 : 0 : QObject::tr( "Defaults to 'http://ogr.maptools.org/'. "
1317 : : "This is the application target namespace." ),
1318 : 0 : QStringLiteral( "http://ogr.maptools.org/" ) // Default value
1319 : : ) );
1320 : :
1321 : 0 : datasetOptions.insert( QStringLiteral( "FORMAT" ), new QgsVectorFileWriter::SetOption(
1322 : 0 : QObject::tr( "If not specified, GML2 will be used." ),
1323 : 0 : QStringList()
1324 : 0 : << QStringLiteral( "GML3" )
1325 : 0 : << QStringLiteral( "GML3Deegree" )
1326 : 0 : << QStringLiteral( "GML3.2" ),
1327 : 0 : QString(), // Default value
1328 : : true // Allow None
1329 : : ) );
1330 : :
1331 : 0 : datasetOptions.insert( QStringLiteral( "GML3_LONGSRS" ), new QgsVectorFileWriter::BoolOption(
1332 : 0 : QObject::tr( "Only valid when FORMAT=GML3/GML3Degree/GML3.2. Default to YES. " //needs review here
1333 : : "If YES, SRS with EPSG authority will be written with the "
1334 : : "'urn:ogc:def:crs:EPSG::' prefix. In the case the SRS is a "
1335 : : "geographic SRS without explicit AXIS order, but that the same "
1336 : : "SRS authority code imported with ImportFromEPSGA() should be "
1337 : : "treated as lat/long, then the function will take care of coordinate "
1338 : : "order swapping. If set to NO, SRS with EPSG authority will be "
1339 : : "written with the 'EPSG:' prefix, even if they are in lat/long order." ),
1340 : : true // Default value
1341 : : ) );
1342 : :
1343 : 0 : datasetOptions.insert( QStringLiteral( "WRITE_FEATURE_BOUNDED_BY" ), new QgsVectorFileWriter::BoolOption(
1344 : 0 : QObject::tr( "only valid when FORMAT=GML3/GML3Degree/GML3.2) Default to YES. "
1345 : : "If set to NO, the <gml:boundedBy> element will not be written for "
1346 : : "each feature." ),
1347 : : true // Default value
1348 : : ) );
1349 : :
1350 : 0 : datasetOptions.insert( QStringLiteral( "SPACE_INDENTATION" ), new QgsVectorFileWriter::BoolOption(
1351 : 0 : QObject::tr( "Default to YES. If YES, the output will be indented with spaces "
1352 : : "for more readability, but at the expense of file size." ),
1353 : : true // Default value
1354 : : ) );
1355 : :
1356 : :
1357 : 0 : driverMetadata.insert( QStringLiteral( "GML" ),
1358 : 0 : QgsVectorFileWriter::MetaData(
1359 : 0 : QStringLiteral( "Geography Markup Language [GML]" ),
1360 : 0 : QObject::tr( "Geography Markup Language [GML]" ),
1361 : 0 : QStringLiteral( "*.gml" ),
1362 : 0 : QStringLiteral( "gml" ),
1363 : : datasetOptions,
1364 : : layerOptions,
1365 : 0 : QStringLiteral( "UTF-8" )
1366 : : )
1367 : : );
1368 : :
1369 : : // GeoPackage
1370 : 0 : datasetOptions.clear();
1371 : 0 : layerOptions.clear();
1372 : :
1373 : 0 : layerOptions.insert( QStringLiteral( "IDENTIFIER" ), new QgsVectorFileWriter::StringOption(
1374 : 0 : QObject::tr( "Human-readable identifier (e.g. short name) for the layer content" ),
1375 : 0 : QString() // Default value
1376 : : ) );
1377 : :
1378 : 0 : layerOptions.insert( QStringLiteral( "DESCRIPTION" ), new QgsVectorFileWriter::StringOption(
1379 : 0 : QObject::tr( "Human-readable description for the layer content" ),
1380 : 0 : QString() // Default value
1381 : : ) );
1382 : :
1383 : 0 : layerOptions.insert( QStringLiteral( "FID" ), new QgsVectorFileWriter::StringOption(
1384 : 0 : QObject::tr( "Name for the feature identifier column" ),
1385 : 0 : QStringLiteral( "fid" ) // Default value
1386 : : ) );
1387 : :
1388 : 0 : layerOptions.insert( QStringLiteral( "GEOMETRY_NAME" ), new QgsVectorFileWriter::StringOption(
1389 : 0 : QObject::tr( "Name for the geometry column" ),
1390 : 0 : QStringLiteral( "geom" ) // Default value
1391 : : ) );
1392 : :
1393 : 0 : layerOptions.insert( QStringLiteral( "SPATIAL_INDEX" ), new QgsVectorFileWriter::BoolOption(
1394 : 0 : QObject::tr( "If a spatial index must be created." ),
1395 : : true // Default value
1396 : : ) );
1397 : :
1398 : 0 : driverMetadata.insert( QStringLiteral( "GPKG" ),
1399 : 0 : QgsVectorFileWriter::MetaData(
1400 : 0 : QStringLiteral( "GeoPackage" ),
1401 : 0 : QObject::tr( "GeoPackage" ),
1402 : 0 : QStringLiteral( "*.gpkg" ),
1403 : 0 : QStringLiteral( "gpkg" ),
1404 : : datasetOptions,
1405 : : layerOptions,
1406 : 0 : QStringLiteral( "UTF-8" )
1407 : : )
1408 : : );
1409 : :
1410 : : // Generic Mapping Tools [GMT]
1411 : 0 : datasetOptions.clear();
1412 : 0 : layerOptions.clear();
1413 : :
1414 : 0 : driverMetadata.insert( QStringLiteral( "GMT" ),
1415 : 0 : QgsVectorFileWriter::MetaData(
1416 : 0 : QStringLiteral( "Generic Mapping Tools [GMT]" ),
1417 : 0 : QObject::tr( "Generic Mapping Tools [GMT]" ),
1418 : 0 : QStringLiteral( "*.gmt" ),
1419 : 0 : QStringLiteral( "gmt" ),
1420 : : datasetOptions,
1421 : : layerOptions
1422 : : )
1423 : : );
1424 : :
1425 : : // GPS eXchange Format [GPX]
1426 : 0 : datasetOptions.clear();
1427 : 0 : layerOptions.clear();
1428 : :
1429 : 0 : layerOptions.insert( QStringLiteral( "FORCE_GPX_TRACK" ), new QgsVectorFileWriter::BoolOption(
1430 : 0 : QObject::tr( "By default when writing a layer whose features are of "
1431 : : "type wkbLineString, the GPX driver chooses to write "
1432 : : "them as routes. If FORCE_GPX_TRACK=YES is specified, "
1433 : : "they will be written as tracks." ),
1434 : : false // Default value
1435 : : ) );
1436 : :
1437 : 0 : layerOptions.insert( QStringLiteral( "FORCE_GPX_ROUTE" ), new QgsVectorFileWriter::BoolOption(
1438 : 0 : QObject::tr( "By default when writing a layer whose features are of "
1439 : : "type wkbMultiLineString, the GPX driver chooses to write "
1440 : : "them as tracks. If FORCE_GPX_ROUTE=YES is specified, "
1441 : : "they will be written as routes, provided that the multilines "
1442 : : "are composed of only one single line." ),
1443 : : false // Default value
1444 : : ) );
1445 : :
1446 : 0 : datasetOptions.insert( QStringLiteral( "GPX_USE_EXTENSIONS" ), new QgsVectorFileWriter::BoolOption(
1447 : 0 : QObject::tr( "If GPX_USE_EXTENSIONS=YES is specified, "
1448 : : "extra fields will be written inside the <extensions> tag." ),
1449 : : false // Default value
1450 : : ) );
1451 : :
1452 : 0 : datasetOptions.insert( QStringLiteral( "GPX_EXTENSIONS_NS" ), new QgsVectorFileWriter::StringOption(
1453 : 0 : QObject::tr( "Only used if GPX_USE_EXTENSIONS=YES and GPX_EXTENSIONS_NS_URL "
1454 : : "is set. The namespace value used for extension tags. By default, 'ogr'." ),
1455 : 0 : QStringLiteral( "ogr" ) // Default value
1456 : : ) );
1457 : :
1458 : 0 : datasetOptions.insert( QStringLiteral( "GPX_EXTENSIONS_NS_URL" ), new QgsVectorFileWriter::StringOption(
1459 : 0 : QObject::tr( "Only used if GPX_USE_EXTENSIONS=YES and GPX_EXTENSIONS_NS "
1460 : : "is set. The namespace URI. By default, 'http://osgeo.org/gdal'." ),
1461 : 0 : QStringLiteral( "http://osgeo.org/gdal" ) // Default value
1462 : : ) );
1463 : :
1464 : 0 : datasetOptions.insert( QStringLiteral( "LINEFORMAT" ), new QgsVectorFileWriter::SetOption(
1465 : 0 : QObject::tr( "By default files are created with the line termination "
1466 : : "conventions of the local platform (CR/LF on win32 or LF "
1467 : : "on all other systems). This may be overridden through use "
1468 : : "of the LINEFORMAT layer creation option which may have a value "
1469 : : "of CRLF (DOS format) or LF (Unix format)." ),
1470 : 0 : QStringList()
1471 : 0 : << QStringLiteral( "CRLF" )
1472 : 0 : << QStringLiteral( "LF" ),
1473 : 0 : QString(), // Default value
1474 : : true // Allow None
1475 : : ) );
1476 : :
1477 : 0 : driverMetadata.insert( QStringLiteral( "GPX" ),
1478 : 0 : QgsVectorFileWriter::MetaData(
1479 : 0 : QStringLiteral( "GPS eXchange Format [GPX]" ),
1480 : 0 : QObject::tr( "GPS eXchange Format [GPX]" ),
1481 : 0 : QStringLiteral( "*.gpx" ),
1482 : 0 : QStringLiteral( "gpx" ),
1483 : : datasetOptions,
1484 : : layerOptions,
1485 : 0 : QStringLiteral( "UTF-8" )
1486 : : )
1487 : : );
1488 : :
1489 : : // INTERLIS 1
1490 : 0 : datasetOptions.clear();
1491 : 0 : layerOptions.clear();
1492 : :
1493 : 0 : driverMetadata.insert( QStringLiteral( "Interlis 1" ),
1494 : 0 : QgsVectorFileWriter::MetaData(
1495 : 0 : QStringLiteral( "INTERLIS 1" ),
1496 : 0 : QObject::tr( "INTERLIS 1" ),
1497 : 0 : QStringLiteral( "*.itf *.xml *.ili" ),
1498 : 0 : QStringLiteral( "ili" ),
1499 : : datasetOptions,
1500 : : layerOptions
1501 : : )
1502 : : );
1503 : :
1504 : : // INTERLIS 2
1505 : 0 : datasetOptions.clear();
1506 : 0 : layerOptions.clear();
1507 : :
1508 : 0 : driverMetadata.insert( QStringLiteral( "Interlis 2" ),
1509 : 0 : QgsVectorFileWriter::MetaData(
1510 : 0 : QStringLiteral( "INTERLIS 2" ),
1511 : 0 : QObject::tr( "INTERLIS 2" ),
1512 : 0 : QStringLiteral( "*.xtf *.xml *.ili" ),
1513 : 0 : QStringLiteral( "ili" ),
1514 : : datasetOptions,
1515 : : layerOptions
1516 : : )
1517 : : );
1518 : :
1519 : : // Keyhole Markup Language [KML]
1520 : 0 : datasetOptions.clear();
1521 : 0 : layerOptions.clear();
1522 : :
1523 : 0 : datasetOptions.insert( QStringLiteral( "NameField" ), new QgsVectorFileWriter::StringOption(
1524 : 0 : QObject::tr( "Allows you to specify the field to use for the KML <name> element." ),
1525 : 0 : QStringLiteral( "Name" ) // Default value
1526 : : ) );
1527 : :
1528 : 0 : datasetOptions.insert( QStringLiteral( "DescriptionField" ), new QgsVectorFileWriter::StringOption(
1529 : 0 : QObject::tr( "Allows you to specify the field to use for the KML <description> element." ),
1530 : 0 : QStringLiteral( "Description" ) // Default value
1531 : : ) );
1532 : :
1533 : 0 : datasetOptions.insert( QStringLiteral( "AltitudeMode" ), new QgsVectorFileWriter::SetOption(
1534 : 0 : QObject::tr( "Allows you to specify the AltitudeMode to use for KML geometries. "
1535 : : "This will only affect 3D geometries and must be one of the valid KML options." ),
1536 : 0 : QStringList()
1537 : 0 : << QStringLiteral( "clampToGround" )
1538 : 0 : << QStringLiteral( "relativeToGround" )
1539 : 0 : << QStringLiteral( "absolute" ),
1540 : 0 : QStringLiteral( "relativeToGround" ) // Default value
1541 : : ) );
1542 : :
1543 : 0 : datasetOptions.insert( QStringLiteral( "DOCUMENT_ID" ), new QgsVectorFileWriter::StringOption(
1544 : 0 : QObject::tr( "The DOCUMENT_ID datasource creation option can be used to specified "
1545 : : "the id of the root <Document> node. The default value is root_doc." ),
1546 : 0 : QStringLiteral( "root_doc" ) // Default value
1547 : : ) );
1548 : :
1549 : 0 : driverMetadata.insert( QStringLiteral( "KML" ),
1550 : 0 : QgsVectorFileWriter::MetaData(
1551 : 0 : QStringLiteral( "Keyhole Markup Language [KML]" ),
1552 : 0 : QObject::tr( "Keyhole Markup Language [KML]" ),
1553 : 0 : QStringLiteral( "*.kml" ),
1554 : 0 : QStringLiteral( "kml" ),
1555 : : datasetOptions,
1556 : : layerOptions,
1557 : 0 : QStringLiteral( "UTF-8" )
1558 : : )
1559 : : );
1560 : :
1561 : : // Mapinfo
1562 : 0 : datasetOptions.clear();
1563 : 0 : layerOptions.clear();
1564 : :
1565 : 0 : auto insertMapInfoOptions = []( QMap<QString, QgsVectorFileWriter::Option *> &datasetOptions, QMap<QString, QgsVectorFileWriter::Option *> &layerOptions )
1566 : : {
1567 : 0 : datasetOptions.insert( QStringLiteral( "SPATIAL_INDEX_MODE" ), new QgsVectorFileWriter::SetOption(
1568 : 0 : QObject::tr( "Use this to turn on 'quick spatial index mode'. "
1569 : : "In this mode writing files can be about 5 times faster, "
1570 : : "but spatial queries can be up to 30 times slower." ),
1571 : 0 : QStringList()
1572 : 0 : << QStringLiteral( "QUICK" )
1573 : 0 : << QStringLiteral( "OPTIMIZED" ),
1574 : 0 : QStringLiteral( "QUICK" ), // Default value
1575 : : true // Allow None
1576 : : ) );
1577 : :
1578 : 0 : datasetOptions.insert( QStringLiteral( "BLOCK_SIZE" ), new QgsVectorFileWriter::IntOption(
1579 : 0 : QObject::tr( "(multiples of 512): Block size for .map files. Defaults "
1580 : : "to 512. MapInfo 15.2 and above creates .tab files with a "
1581 : : "blocksize of 16384 bytes. Any MapInfo version should be "
1582 : : "able to handle block sizes from 512 to 32256." ),
1583 : : 512
1584 : : ) );
1585 : 0 : layerOptions.insert( QStringLiteral( "BOUNDS" ), new QgsVectorFileWriter::StringOption(
1586 : 0 : QObject::tr( "xmin,ymin,xmax,ymax: Define custom layer bounds to increase the "
1587 : : "accuracy of the coordinates. Note: the geometry of written "
1588 : : "features must be within the defined box." ),
1589 : 0 : QString() // Default value
1590 : : ) );
1591 : 0 : };
1592 : 0 : insertMapInfoOptions( datasetOptions, layerOptions );
1593 : :
1594 : 0 : driverMetadata.insert( QStringLiteral( "MapInfo File" ),
1595 : 0 : QgsVectorFileWriter::MetaData(
1596 : 0 : QStringLiteral( "Mapinfo" ),
1597 : 0 : QObject::tr( "Mapinfo TAB" ),
1598 : 0 : QStringLiteral( "*.tab" ),
1599 : 0 : QStringLiteral( "tab" ),
1600 : : datasetOptions,
1601 : : layerOptions
1602 : : )
1603 : : );
1604 : 0 : datasetOptions.clear();
1605 : 0 : layerOptions.clear();
1606 : 0 : insertMapInfoOptions( datasetOptions, layerOptions );
1607 : :
1608 : : // QGIS internal alias for MIF files
1609 : 0 : driverMetadata.insert( QStringLiteral( "MapInfo MIF" ),
1610 : 0 : QgsVectorFileWriter::MetaData(
1611 : 0 : QStringLiteral( "Mapinfo" ),
1612 : 0 : QObject::tr( "Mapinfo MIF" ),
1613 : 0 : QStringLiteral( "*.mif" ),
1614 : 0 : QStringLiteral( "mif" ),
1615 : : datasetOptions,
1616 : : layerOptions
1617 : : )
1618 : : );
1619 : :
1620 : : // Microstation DGN
1621 : 0 : datasetOptions.clear();
1622 : 0 : layerOptions.clear();
1623 : :
1624 : 0 : datasetOptions.insert( QStringLiteral( "3D" ), new QgsVectorFileWriter::BoolOption(
1625 : 0 : QObject::tr( "Determine whether 2D (seed_2d.dgn) or 3D (seed_3d.dgn) "
1626 : : "seed file should be used. This option is ignored if the SEED option is provided." ),
1627 : : false // Default value
1628 : : ) );
1629 : :
1630 : 0 : datasetOptions.insert( QStringLiteral( "SEED" ), new QgsVectorFileWriter::StringOption(
1631 : 0 : QObject::tr( "Override the seed file to use." ),
1632 : 0 : QString() // Default value
1633 : : ) );
1634 : :
1635 : 0 : datasetOptions.insert( QStringLiteral( "COPY_WHOLE_SEED_FILE" ), new QgsVectorFileWriter::BoolOption(
1636 : 0 : QObject::tr( "Indicate whether the whole seed file should be copied. "
1637 : : "If not, only the first three elements will be copied." ),
1638 : : false // Default value
1639 : : ) );
1640 : :
1641 : 0 : datasetOptions.insert( QStringLiteral( "COPY_SEED_FILE_COLOR_TABLE" ), new QgsVectorFileWriter::BoolOption(
1642 : 0 : QObject::tr( "Indicates whether the color table should be copied from the seed file." ),
1643 : : false // Default value
1644 : : ) );
1645 : :
1646 : 0 : datasetOptions.insert( QStringLiteral( "MASTER_UNIT_NAME" ), new QgsVectorFileWriter::StringOption(
1647 : 0 : QObject::tr( "Override the master unit name from the seed file with "
1648 : : "the provided one or two character unit name." ),
1649 : 0 : QString() // Default value
1650 : : ) );
1651 : :
1652 : 0 : datasetOptions.insert( QStringLiteral( "SUB_UNIT_NAME" ), new QgsVectorFileWriter::StringOption(
1653 : 0 : QObject::tr( "Override the sub unit name from the seed file with the provided "
1654 : : "one or two character unit name." ),
1655 : 0 : QString() // Default value
1656 : : ) );
1657 : :
1658 : 0 : datasetOptions.insert( QStringLiteral( "SUB_UNITS_PER_MASTER_UNIT" ), new QgsVectorFileWriter::IntOption(
1659 : 0 : QObject::tr( "Override the number of subunits per master unit. "
1660 : : "By default the seed file value is used." ),
1661 : : 0 // Default value
1662 : : ) );
1663 : :
1664 : 0 : datasetOptions.insert( QStringLiteral( "UOR_PER_SUB_UNIT" ), new QgsVectorFileWriter::IntOption(
1665 : 0 : QObject::tr( "Override the number of UORs (Units of Resolution) "
1666 : : "per sub unit. By default the seed file value is used." ),
1667 : : 0 // Default value
1668 : : ) );
1669 : :
1670 : 0 : datasetOptions.insert( QStringLiteral( "ORIGIN" ), new QgsVectorFileWriter::StringOption(
1671 : 0 : QObject::tr( "ORIGIN=x,y,z: Override the origin of the design plane. "
1672 : : "By default the origin from the seed file is used." ),
1673 : 0 : QString() // Default value
1674 : : ) );
1675 : :
1676 : 0 : driverMetadata.insert( QStringLiteral( "DGN" ),
1677 : 0 : QgsVectorFileWriter::MetaData(
1678 : 0 : QStringLiteral( "Microstation DGN" ),
1679 : 0 : QObject::tr( "Microstation DGN" ),
1680 : 0 : QStringLiteral( "*.dgn" ),
1681 : 0 : QStringLiteral( "dgn" ),
1682 : : datasetOptions,
1683 : : layerOptions
1684 : : )
1685 : : );
1686 : :
1687 : : // S-57 Base file
1688 : 0 : datasetOptions.clear();
1689 : 0 : layerOptions.clear();
1690 : :
1691 : 0 : datasetOptions.insert( QStringLiteral( "UPDATES" ), new QgsVectorFileWriter::SetOption(
1692 : 0 : QObject::tr( "Should update files be incorporated into the base data on the fly." ),
1693 : 0 : QStringList()
1694 : 0 : << QStringLiteral( "APPLY" )
1695 : 0 : << QStringLiteral( "IGNORE" ),
1696 : 0 : QStringLiteral( "APPLY" ) // Default value
1697 : : ) );
1698 : :
1699 : 0 : datasetOptions.insert( QStringLiteral( "SPLIT_MULTIPOINT" ), new QgsVectorFileWriter::BoolOption(
1700 : 0 : QObject::tr( "Should multipoint soundings be split into many single point sounding features. "
1701 : : "Multipoint geometries are not well handled by many formats, "
1702 : : "so it can be convenient to split single sounding features with many points "
1703 : : "into many single point features." ),
1704 : : false // Default value
1705 : : ) );
1706 : :
1707 : 0 : datasetOptions.insert( QStringLiteral( "ADD_SOUNDG_DEPTH" ), new QgsVectorFileWriter::BoolOption(
1708 : 0 : QObject::tr( "Should a DEPTH attribute be added on SOUNDG features and assign the depth "
1709 : : "of the sounding. This should only be enabled when SPLIT_MULTIPOINT is "
1710 : : "also enabled." ),
1711 : : false // Default value
1712 : : ) );
1713 : :
1714 : 0 : datasetOptions.insert( QStringLiteral( "RETURN_PRIMITIVES" ), new QgsVectorFileWriter::BoolOption(
1715 : 0 : QObject::tr( "Should all the low level geometry primitives be returned as special "
1716 : : "IsolatedNode, ConnectedNode, Edge and Face layers." ),
1717 : : false // Default value
1718 : : ) );
1719 : :
1720 : 0 : datasetOptions.insert( QStringLiteral( "PRESERVE_EMPTY_NUMBERS" ), new QgsVectorFileWriter::BoolOption(
1721 : 0 : QObject::tr( "If enabled, numeric attributes assigned an empty string as a value will "
1722 : : "be preserved as a special numeric value. This option should not generally "
1723 : : "be needed, but may be useful when translated S-57 to S-57 losslessly." ),
1724 : : false // Default value
1725 : : ) );
1726 : :
1727 : 0 : datasetOptions.insert( QStringLiteral( "LNAM_REFS" ), new QgsVectorFileWriter::BoolOption(
1728 : 0 : QObject::tr( "Should LNAM and LNAM_REFS fields be attached to features capturing "
1729 : : "the feature to feature relationships in the FFPT group of the S-57 file." ),
1730 : : true // Default value
1731 : : ) );
1732 : :
1733 : 0 : datasetOptions.insert( QStringLiteral( "RETURN_LINKAGES" ), new QgsVectorFileWriter::BoolOption(
1734 : 0 : QObject::tr( "Should additional attributes relating features to their underlying "
1735 : : "geometric primitives be attached. These are the values of the FSPT group, "
1736 : : "and are primarily needed when doing S-57 to S-57 translations." ),
1737 : : false // Default value
1738 : : ) );
1739 : :
1740 : 0 : datasetOptions.insert( QStringLiteral( "RECODE_BY_DSSI" ), new QgsVectorFileWriter::BoolOption(
1741 : 0 : QObject::tr( "Should attribute values be recoded to UTF-8 from the character encoding "
1742 : : "specified in the S57 DSSI record." ),
1743 : : false // Default value
1744 : : ) );
1745 : :
1746 : : // set OGR_S57_OPTIONS = "RETURN_PRIMITIVES=ON,RETURN_LINKAGES=ON,LNAM_REFS=ON"
1747 : :
1748 : 0 : driverMetadata.insert( QStringLiteral( "S57" ),
1749 : 0 : QgsVectorFileWriter::MetaData(
1750 : 0 : QStringLiteral( "S-57 Base file" ),
1751 : 0 : QObject::tr( "S-57 Base file" ),
1752 : 0 : QStringLiteral( "*.000" ),
1753 : 0 : QStringLiteral( "000" ),
1754 : : datasetOptions,
1755 : : layerOptions
1756 : : )
1757 : : );
1758 : :
1759 : : // Spatial Data Transfer Standard [SDTS]
1760 : 0 : datasetOptions.clear();
1761 : 0 : layerOptions.clear();
1762 : :
1763 : 0 : driverMetadata.insert( QStringLiteral( "SDTS" ),
1764 : 0 : QgsVectorFileWriter::MetaData(
1765 : 0 : QStringLiteral( "Spatial Data Transfer Standard [SDTS]" ),
1766 : 0 : QObject::tr( "Spatial Data Transfer Standard [SDTS]" ),
1767 : 0 : QStringLiteral( "*catd.ddf" ),
1768 : 0 : QStringLiteral( "ddf" ),
1769 : : datasetOptions,
1770 : : layerOptions
1771 : : )
1772 : : );
1773 : :
1774 : : // SQLite
1775 : 0 : datasetOptions.clear();
1776 : 0 : layerOptions.clear();
1777 : :
1778 : 0 : datasetOptions.insert( QStringLiteral( "METADATA" ), new QgsVectorFileWriter::BoolOption(
1779 : 0 : QObject::tr( "Can be used to avoid creating the geometry_columns and spatial_ref_sys "
1780 : : "tables in a new database. By default these metadata tables are created "
1781 : : "when a new database is created." ),
1782 : : true // Default value
1783 : : ) );
1784 : :
1785 : : // Will handle the SpatiaLite alias
1786 : 0 : datasetOptions.insert( QStringLiteral( "SPATIALITE" ), new QgsVectorFileWriter::HiddenOption(
1787 : 0 : QStringLiteral( "NO" )
1788 : : ) );
1789 : :
1790 : :
1791 : 0 : datasetOptions.insert( QStringLiteral( "INIT_WITH_EPSG" ), new QgsVectorFileWriter::HiddenOption(
1792 : 0 : QStringLiteral( "NO" )
1793 : : ) );
1794 : :
1795 : 0 : layerOptions.insert( QStringLiteral( "FORMAT" ), new QgsVectorFileWriter::SetOption(
1796 : 0 : QObject::tr( "Controls the format used for the geometry column. Defaults to WKB. "
1797 : : "This is generally more space and processing efficient, but harder "
1798 : : "to inspect or use in simple applications than WKT (Well Known Text)." ),
1799 : 0 : QStringList()
1800 : 0 : << QStringLiteral( "WKB" )
1801 : 0 : << QStringLiteral( "WKT" ),
1802 : 0 : QStringLiteral( "WKB" ) // Default value
1803 : : ) );
1804 : :
1805 : 0 : layerOptions.insert( QStringLiteral( "LAUNDER" ), new QgsVectorFileWriter::BoolOption(
1806 : 0 : QObject::tr( "Controls whether layer and field names will be laundered for easier use "
1807 : : "in SQLite. Laundered names will be converted to lower case and some special "
1808 : : "characters(' - #) will be changed to underscores." ),
1809 : : true // Default value
1810 : : ) );
1811 : :
1812 : 0 : layerOptions.insert( QStringLiteral( "SPATIAL_INDEX" ), new QgsVectorFileWriter::HiddenOption(
1813 : 0 : QStringLiteral( "NO" )
1814 : : ) );
1815 : :
1816 : 0 : layerOptions.insert( QStringLiteral( "COMPRESS_GEOM" ), new QgsVectorFileWriter::HiddenOption(
1817 : 0 : QStringLiteral( "NO" )
1818 : : ) );
1819 : :
1820 : 0 : layerOptions.insert( QStringLiteral( "SRID" ), new QgsVectorFileWriter::HiddenOption(
1821 : 0 : QString()
1822 : : ) );
1823 : :
1824 : 0 : layerOptions.insert( QStringLiteral( "COMPRESS_COLUMNS" ), new QgsVectorFileWriter::StringOption(
1825 : 0 : QObject::tr( "column_name1[,column_name2, …] A list of (String) columns that "
1826 : : "must be compressed with ZLib DEFLATE algorithm. This might be beneficial "
1827 : : "for databases that have big string blobs. However, use with care, since "
1828 : : "the value of such columns will be seen as compressed binary content with "
1829 : : "other SQLite utilities (or previous OGR versions). With OGR, when inserting, "
1830 : : "modifying or querying compressed columns, compression/decompression is "
1831 : : "done transparently. However, such columns cannot be (easily) queried with "
1832 : : "an attribute filter or WHERE clause. Note: in table definition, such columns "
1833 : : "have the 'VARCHAR_deflate' declaration type." ),
1834 : 0 : QString() // Default value
1835 : : ) );
1836 : :
1837 : 0 : driverMetadata.insert( QStringLiteral( "SQLite" ),
1838 : 0 : QgsVectorFileWriter::MetaData(
1839 : 0 : QStringLiteral( "SQLite" ),
1840 : 0 : QObject::tr( "SQLite" ),
1841 : 0 : QStringLiteral( "*.sqlite" ),
1842 : 0 : QStringLiteral( "sqlite" ),
1843 : : datasetOptions,
1844 : : layerOptions,
1845 : 0 : QStringLiteral( "UTF-8" )
1846 : : )
1847 : : );
1848 : :
1849 : : // SpatiaLite
1850 : 0 : datasetOptions.clear();
1851 : 0 : layerOptions.clear();
1852 : :
1853 : 0 : datasetOptions.insert( QStringLiteral( "METADATA" ), new QgsVectorFileWriter::BoolOption(
1854 : 0 : QObject::tr( "Can be used to avoid creating the geometry_columns and spatial_ref_sys "
1855 : : "tables in a new database. By default these metadata tables are created "
1856 : : "when a new database is created." ),
1857 : : true // Default value
1858 : : ) );
1859 : :
1860 : 0 : datasetOptions.insert( QStringLiteral( "SPATIALITE" ), new QgsVectorFileWriter::HiddenOption(
1861 : 0 : QStringLiteral( "YES" )
1862 : : ) );
1863 : :
1864 : 0 : datasetOptions.insert( QStringLiteral( "INIT_WITH_EPSG" ), new QgsVectorFileWriter::BoolOption(
1865 : 0 : QObject::tr( "Insert the content of the EPSG CSV files into the spatial_ref_sys table. "
1866 : : "Set to NO for regular SQLite databases." ),
1867 : : true // Default value
1868 : : ) );
1869 : :
1870 : 0 : layerOptions.insert( QStringLiteral( "FORMAT" ), new QgsVectorFileWriter::HiddenOption(
1871 : 0 : QStringLiteral( "SPATIALITE" )
1872 : : ) );
1873 : :
1874 : 0 : layerOptions.insert( QStringLiteral( "LAUNDER" ), new QgsVectorFileWriter::BoolOption(
1875 : 0 : QObject::tr( "Controls whether layer and field names will be laundered for easier use "
1876 : : "in SQLite. Laundered names will be converted to lower case and some special "
1877 : : "characters(' - #) will be changed to underscores." ),
1878 : : true // Default value
1879 : : ) );
1880 : :
1881 : 0 : layerOptions.insert( QStringLiteral( "SPATIAL_INDEX" ), new QgsVectorFileWriter::BoolOption(
1882 : 0 : QObject::tr( "If the database is of the SpatiaLite flavor, and if OGR is linked "
1883 : : "against libspatialite, this option can be used to control if a spatial "
1884 : : "index must be created." ),
1885 : : true // Default value
1886 : : ) );
1887 : :
1888 : 0 : layerOptions.insert( QStringLiteral( "COMPRESS_GEOM" ), new QgsVectorFileWriter::BoolOption(
1889 : 0 : QObject::tr( "If the format of the geometry BLOB is of the SpatiaLite flavor, "
1890 : : "this option can be used to control if the compressed format for "
1891 : : "geometries (LINESTRINGs, POLYGONs) must be used." ),
1892 : : false // Default value
1893 : : ) );
1894 : :
1895 : 0 : layerOptions.insert( QStringLiteral( "SRID" ), new QgsVectorFileWriter::StringOption(
1896 : 0 : QObject::tr( "Used to force the SRID number of the SRS associated with the layer. "
1897 : : "When this option isn't specified and that a SRS is associated with the "
1898 : : "layer, a search is made in the spatial_ref_sys to find a match for the "
1899 : : "SRS, and, if there is no match, a new entry is inserted for the SRS in "
1900 : : "the spatial_ref_sys table. When the SRID option is specified, this "
1901 : : "search (and the eventual insertion of a new entry) will not be done: "
1902 : : "the specified SRID is used as such." ),
1903 : 0 : QString() // Default value
1904 : : ) );
1905 : :
1906 : 0 : layerOptions.insert( QStringLiteral( "COMPRESS_COLUMNS" ), new QgsVectorFileWriter::StringOption(
1907 : 0 : QObject::tr( "column_name1[,column_name2, …] A list of (String) columns that "
1908 : : "must be compressed with ZLib DEFLATE algorithm. This might be beneficial "
1909 : : "for databases that have big string blobs. However, use with care, since "
1910 : : "the value of such columns will be seen as compressed binary content with "
1911 : : "other SQLite utilities (or previous OGR versions). With OGR, when inserting, "
1912 : : "modifying or queryings compressed columns, compression/decompression is "
1913 : : "done transparently. However, such columns cannot be (easily) queried with "
1914 : : "an attribute filter or WHERE clause. Note: in table definition, such columns "
1915 : : "have the 'VARCHAR_deflate' declaration type." ),
1916 : 0 : QString() // Default value
1917 : : ) );
1918 : :
1919 : 0 : driverMetadata.insert( QStringLiteral( "SpatiaLite" ),
1920 : 0 : QgsVectorFileWriter::MetaData(
1921 : 0 : QStringLiteral( "SpatiaLite" ),
1922 : 0 : QObject::tr( "SpatiaLite" ),
1923 : 0 : QStringLiteral( "*.sqlite" ),
1924 : 0 : QStringLiteral( "sqlite" ),
1925 : : datasetOptions,
1926 : : layerOptions,
1927 : 0 : QStringLiteral( "UTF-8" )
1928 : : )
1929 : : );
1930 : : // AutoCAD DXF
1931 : 0 : datasetOptions.clear();
1932 : 0 : layerOptions.clear();
1933 : :
1934 : 0 : datasetOptions.insert( QStringLiteral( "HEADER" ), new QgsVectorFileWriter::StringOption(
1935 : 0 : QObject::tr( "Override the header file used - in place of header.dxf." ),
1936 : 0 : QString() // Default value
1937 : : ) );
1938 : :
1939 : 0 : datasetOptions.insert( QStringLiteral( "TRAILER" ), new QgsVectorFileWriter::StringOption(
1940 : 0 : QObject::tr( "Override the trailer file used - in place of trailer.dxf." ),
1941 : 0 : QString() // Default value
1942 : : ) );
1943 : :
1944 : 0 : driverMetadata.insert( QStringLiteral( "DXF" ),
1945 : 0 : QgsVectorFileWriter::MetaData(
1946 : 0 : QStringLiteral( "AutoCAD DXF" ),
1947 : 0 : QObject::tr( "AutoCAD DXF" ),
1948 : 0 : QStringLiteral( "*.dxf" ),
1949 : 0 : QStringLiteral( "dxf" ),
1950 : : datasetOptions,
1951 : : layerOptions
1952 : : )
1953 : : );
1954 : :
1955 : : // Geoconcept
1956 : 0 : datasetOptions.clear();
1957 : 0 : layerOptions.clear();
1958 : :
1959 : 0 : datasetOptions.insert( QStringLiteral( "EXTENSION" ), new QgsVectorFileWriter::SetOption(
1960 : 0 : QObject::tr( "Indicates the GeoConcept export file extension. "
1961 : : "TXT was used by earlier releases of GeoConcept. GXT is currently used." ),
1962 : 0 : QStringList()
1963 : 0 : << QStringLiteral( "GXT" )
1964 : 0 : << QStringLiteral( "TXT" ),
1965 : 0 : QStringLiteral( "GXT" ) // Default value
1966 : : ) );
1967 : :
1968 : 0 : datasetOptions.insert( QStringLiteral( "CONFIG" ), new QgsVectorFileWriter::StringOption(
1969 : 0 : QObject::tr( "Path to the GCT: the GCT file describes the GeoConcept types definitions: "
1970 : : "In this file, every line must start with //# followed by a keyword. "
1971 : : "Lines starting with // are comments." ),
1972 : 0 : QString() // Default value
1973 : : ) );
1974 : :
1975 : 0 : datasetOptions.insert( QStringLiteral( "FEATURETYPE" ), new QgsVectorFileWriter::StringOption(
1976 : 0 : QObject::tr( "Defines the feature to be created. The TYPE corresponds to one of the Name "
1977 : : "found in the GCT file for a type section. The SUBTYPE corresponds to one of "
1978 : : "the Name found in the GCT file for a sub-type section within the previous "
1979 : : "type section." ),
1980 : 0 : QString() // Default value
1981 : : ) );
1982 : :
1983 : 0 : driverMetadata.insert( QStringLiteral( "Geoconcept" ),
1984 : 0 : QgsVectorFileWriter::MetaData(
1985 : 0 : QStringLiteral( "Geoconcept" ),
1986 : 0 : QObject::tr( "Geoconcept" ),
1987 : 0 : QStringLiteral( "*.gxt *.txt" ),
1988 : 0 : QStringLiteral( "gxt" ),
1989 : : datasetOptions,
1990 : : layerOptions
1991 : : )
1992 : : );
1993 : :
1994 : : // ESRI FileGDB
1995 : 0 : datasetOptions.clear();
1996 : 0 : layerOptions.clear();
1997 : :
1998 : 0 : layerOptions.insert( QStringLiteral( "FEATURE_DATASET" ), new QgsVectorFileWriter::StringOption(
1999 : 0 : QObject::tr( "When this option is set, the new layer will be created inside the named "
2000 : : "FeatureDataset folder. If the folder does not already exist, it will be created." ),
2001 : 0 : QString() // Default value
2002 : : ) );
2003 : :
2004 : 0 : layerOptions.insert( QStringLiteral( "GEOMETRY_NAME" ), new QgsVectorFileWriter::StringOption(
2005 : 0 : QObject::tr( "Set name of geometry column in new layer. Defaults to 'SHAPE'." ),
2006 : 0 : QStringLiteral( "SHAPE" ) // Default value
2007 : : ) );
2008 : :
2009 : 0 : layerOptions.insert( QStringLiteral( "FID" ), new QgsVectorFileWriter::StringOption(
2010 : 0 : QObject::tr( "Name of the OID column to create. Defaults to 'OBJECTID'." ),
2011 : 0 : QStringLiteral( "OBJECTID" ) // Default value
2012 : : ) );
2013 : :
2014 : 0 : driverMetadata.insert( QStringLiteral( "FileGDB" ),
2015 : 0 : QgsVectorFileWriter::MetaData(
2016 : 0 : QStringLiteral( "ESRI FileGDB" ),
2017 : 0 : QObject::tr( "ESRI FileGDB" ),
2018 : 0 : QStringLiteral( "*.gdb" ),
2019 : 0 : QStringLiteral( "gdb" ),
2020 : : datasetOptions,
2021 : : layerOptions,
2022 : 0 : QStringLiteral( "UTF-8" )
2023 : : )
2024 : : );
2025 : :
2026 : : // XLSX
2027 : 0 : datasetOptions.clear();
2028 : 0 : layerOptions.clear();
2029 : :
2030 : 0 : layerOptions.insert( QStringLiteral( "OGR_XLSX_FIELD_TYPES" ), new QgsVectorFileWriter::SetOption(
2031 : 0 : QObject::tr( "By default, the driver will try to detect the data type of fields. If set "
2032 : : "to STRING, all fields will be of String type." ),
2033 : 0 : QStringList()
2034 : 0 : << QStringLiteral( "AUTO" )
2035 : 0 : << QStringLiteral( "STRING" ),
2036 : 0 : QStringLiteral( "AUTO" ), // Default value
2037 : : false // Allow None
2038 : : ) );
2039 : :
2040 : 0 : layerOptions.insert( QStringLiteral( "OGR_XLSX_HEADERS" ), new QgsVectorFileWriter::SetOption(
2041 : 0 : QObject::tr( "By default, the driver will read the first lines of each sheet to detect "
2042 : : "if the first line might be the name of columns. If set to FORCE, the driver "
2043 : : "will consider the first line as the header line. If set to "
2044 : : "DISABLE, it will be considered as the first feature. Otherwise "
2045 : : "auto-detection will occur." ),
2046 : 0 : QStringList()
2047 : 0 : << QStringLiteral( "FORCE" )
2048 : 0 : << QStringLiteral( "DISABLE" )
2049 : 0 : << QStringLiteral( "AUTO" ),
2050 : 0 : QStringLiteral( "AUTO" ), // Default value
2051 : : false // Allow None
2052 : : ) );
2053 : :
2054 : 0 : driverMetadata.insert( QStringLiteral( "XLSX" ),
2055 : 0 : QgsVectorFileWriter::MetaData(
2056 : 0 : QStringLiteral( "MS Office Open XML spreadsheet" ),
2057 : 0 : QObject::tr( "MS Office Open XML spreadsheet [XLSX]" ),
2058 : 0 : QStringLiteral( "*.xlsx" ),
2059 : 0 : QStringLiteral( "xlsx" ),
2060 : : datasetOptions,
2061 : : layerOptions,
2062 : 0 : QStringLiteral( "UTF-8" )
2063 : : )
2064 : : );
2065 : :
2066 : : // ODS
2067 : 0 : datasetOptions.clear();
2068 : 0 : layerOptions.clear();
2069 : :
2070 : 0 : layerOptions.insert( QStringLiteral( "OGR_ODS_FIELD_TYPES" ), new QgsVectorFileWriter::SetOption(
2071 : 0 : QObject::tr( "By default, the driver will try to detect the data type of fields. If set "
2072 : : "to STRING, all fields will be of String type." ),
2073 : 0 : QStringList()
2074 : 0 : << QStringLiteral( "AUTO" )
2075 : 0 : << QStringLiteral( "STRING" ),
2076 : 0 : QStringLiteral( "AUTO" ), // Default value
2077 : : false // Allow None
2078 : : ) );
2079 : :
2080 : 0 : layerOptions.insert( QStringLiteral( "OGR_ODS_HEADERS" ), new QgsVectorFileWriter::SetOption(
2081 : 0 : QObject::tr( "By default, the driver will read the first lines of each sheet to detect "
2082 : : "if the first line might be the name of columns. If set to FORCE, the driver "
2083 : : "will consider the first line as the header line. If set to "
2084 : : "DISABLE, it will be considered as the first feature. Otherwise "
2085 : : "auto-detection will occur." ),
2086 : 0 : QStringList()
2087 : 0 : << QStringLiteral( "FORCE" )
2088 : 0 : << QStringLiteral( "DISABLE" )
2089 : 0 : << QStringLiteral( "AUTO" ),
2090 : 0 : QStringLiteral( "AUTO" ), // Default value
2091 : : false // Allow None
2092 : : ) );
2093 : :
2094 : 0 : driverMetadata.insert( QStringLiteral( "ODS" ),
2095 : 0 : QgsVectorFileWriter::MetaData(
2096 : 0 : QStringLiteral( "Open Document Spreadsheet" ),
2097 : 0 : QObject::tr( "Open Document Spreadsheet [ODS]" ),
2098 : 0 : QStringLiteral( "*.ods" ),
2099 : 0 : QStringLiteral( "ods" ),
2100 : : datasetOptions,
2101 : : layerOptions,
2102 : 0 : QStringLiteral( "UTF-8" )
2103 : : )
2104 : : );
2105 : :
2106 : : // PGDump
2107 : 0 : datasetOptions.clear();
2108 : 0 : layerOptions.clear();
2109 : :
2110 : 0 : datasetOptions.insert( QStringLiteral( "LINEFORMAT" ), new QgsVectorFileWriter::SetOption(
2111 : 0 : QObject::tr( "Line termination character sequence." ),
2112 : 0 : QStringList()
2113 : 0 : << QStringLiteral( "CRLF" )
2114 : 0 : << QStringLiteral( "LF" ),
2115 : 0 : QString( "LF" ), // Default value
2116 : : false // Allow None
2117 : : ) );
2118 : :
2119 : :
2120 : 0 : layerOptions.insert( QStringLiteral( "GEOM_TYPE" ), new QgsVectorFileWriter::SetOption(
2121 : 0 : QObject::tr( "Format of geometry columns." ),
2122 : 0 : QStringList()
2123 : 0 : << QStringLiteral( "geometry" )
2124 : 0 : << QStringLiteral( "geography" ),
2125 : 0 : QStringLiteral( "geometry" ), // Default value
2126 : : false // Allow None
2127 : : ) );
2128 : :
2129 : 0 : layerOptions.insert( QStringLiteral( "LAUNDER" ), new QgsVectorFileWriter::BoolOption(
2130 : 0 : QObject::tr( "Controls whether layer and field names will be laundered for easier use. "
2131 : : "Laundered names will be converted to lower case and some special "
2132 : : "characters(' - #) will be changed to underscores." ),
2133 : : true // Default value
2134 : : ) );
2135 : :
2136 : 0 : layerOptions.insert( QStringLiteral( "GEOMETRY_NAME" ), new QgsVectorFileWriter::StringOption(
2137 : 0 : QObject::tr( "Name for the geometry column. Defaults to wkb_geometry "
2138 : : "for GEOM_TYPE=geometry or the_geog for GEOM_TYPE=geography" ) ) );
2139 : :
2140 : 0 : layerOptions.insert( QStringLiteral( "SCHEMA" ), new QgsVectorFileWriter::StringOption(
2141 : 0 : QObject::tr( "Name of schema into which to create the new table" ) ) );
2142 : :
2143 : 0 : layerOptions.insert( QStringLiteral( "CREATE_SCHEMA" ), new QgsVectorFileWriter::BoolOption(
2144 : 0 : QObject::tr( "Whether to explicitly emit the CREATE SCHEMA statement to create the specified schema." ),
2145 : : true // Default value
2146 : : ) );
2147 : :
2148 : 0 : layerOptions.insert( QStringLiteral( "CREATE_TABLE" ), new QgsVectorFileWriter::BoolOption(
2149 : 0 : QObject::tr( "Whether to explicitly recreate the table if necessary." ),
2150 : : true // Default value
2151 : : ) );
2152 : :
2153 : 0 : layerOptions.insert( QStringLiteral( "DROP_TABLE" ), new QgsVectorFileWriter::SetOption(
2154 : 0 : QObject::tr( "Whether to explicitly destroy tables before recreating them." ),
2155 : 0 : QStringList()
2156 : 0 : << QStringLiteral( "YES" )
2157 : 0 : << QStringLiteral( "NO" )
2158 : 0 : << QStringLiteral( "IF_EXISTS" ),
2159 : 0 : QStringLiteral( "YES" ), // Default value
2160 : : false // Allow None
2161 : : ) );
2162 : :
2163 : 0 : layerOptions.insert( QStringLiteral( "SRID" ), new QgsVectorFileWriter::StringOption(
2164 : 0 : QObject::tr( "Used to force the SRID number of the SRS associated with the layer. "
2165 : : "When this option isn't specified and that a SRS is associated with the "
2166 : : "layer, a search is made in the spatial_ref_sys to find a match for the "
2167 : : "SRS, and, if there is no match, a new entry is inserted for the SRS in "
2168 : : "the spatial_ref_sys table. When the SRID option is specified, this "
2169 : : "search (and the eventual insertion of a new entry) will not be done: "
2170 : : "the specified SRID is used as such." ),
2171 : 0 : QString() // Default value
2172 : : ) );
2173 : :
2174 : 0 : layerOptions.insert( QStringLiteral( "POSTGIS_VERSION" ), new QgsVectorFileWriter::StringOption(
2175 : 0 : QObject::tr( "Can be set to 2.0 or 2.2 for PostGIS 2.0/2.2 compatibility. "
2176 : : "Important to set it correctly if using non-linear geometry types" ),
2177 : 0 : QString( "2.2" ) // Default value
2178 : : ) );
2179 : :
2180 : 0 : driverMetadata.insert( QStringLiteral( "PGDUMP" ),
2181 : 0 : QgsVectorFileWriter::MetaData(
2182 : 0 : QStringLiteral( "PostgreSQL SQL dump" ),
2183 : 0 : QObject::tr( "PostgreSQL SQL dump" ),
2184 : 0 : QStringLiteral( "*.sql" ),
2185 : 0 : QStringLiteral( "sql" ),
2186 : : datasetOptions,
2187 : : layerOptions,
2188 : 0 : QStringLiteral( "UTF-8" )
2189 : : )
2190 : : );
2191 : :
2192 : 0 : }
2193 : :
2194 : : QgsVectorFileWriterMetadataContainer( const QgsVectorFileWriterMetadataContainer &other ) = delete;
2195 : : QgsVectorFileWriterMetadataContainer &operator=( const QgsVectorFileWriterMetadataContainer &other ) = delete;
2196 : 0 : ~QgsVectorFileWriterMetadataContainer()
2197 : : {
2198 : 0 : for ( auto it = driverMetadata.constBegin(); it != driverMetadata.constEnd(); ++it )
2199 : : {
2200 : 0 : for ( auto optionIt = it.value().driverOptions.constBegin(); optionIt != it.value().driverOptions.constEnd(); ++optionIt )
2201 : 0 : delete optionIt.value();
2202 : 0 : for ( auto optionIt = it.value().layerOptions.constBegin(); optionIt != it.value().layerOptions.constEnd(); ++optionIt )
2203 : 0 : delete optionIt.value();
2204 : 0 : }
2205 : 0 : }
2206 : :
2207 : : QMap<QString, QgsVectorFileWriter::MetaData> driverMetadata;
2208 : :
2209 : : };
2210 : : ///@endcond
2211 : :
2212 : 0 : bool QgsVectorFileWriter::driverMetadata( const QString &driverName, QgsVectorFileWriter::MetaData &driverMetadata )
2213 : : {
2214 : 0 : static QgsVectorFileWriterMetadataContainer sDriverMetadata;
2215 : 0 : QMap<QString, MetaData>::ConstIterator it = sDriverMetadata.driverMetadata.constBegin();
2216 : :
2217 : 0 : for ( ; it != sDriverMetadata.driverMetadata.constEnd(); ++it )
2218 : : {
2219 : 0 : if ( it.key() == QLatin1String( "PGDUMP" ) &&
2220 : 0 : driverName != QLatin1String( "PGDUMP" ) &&
2221 : 0 : driverName != QLatin1String( "PostgreSQL SQL dump" ) )
2222 : : {
2223 : : // We do not want the 'PG' driver to be wrongly identified with PGDUMP
2224 : 0 : continue;
2225 : : }
2226 : 0 : if ( it.key().startsWith( driverName ) || it.value().longName.startsWith( driverName ) )
2227 : : {
2228 : 0 : driverMetadata = it.value();
2229 : 0 : return true;
2230 : : }
2231 : 0 : }
2232 : :
2233 : 0 : return false;
2234 : 0 : }
2235 : :
2236 : 0 : QStringList QgsVectorFileWriter::defaultDatasetOptions( const QString &driverName )
2237 : : {
2238 : 0 : MetaData metadata;
2239 : 0 : bool ok = driverMetadata( driverName, metadata );
2240 : 0 : if ( !ok )
2241 : 0 : return QStringList();
2242 : 0 : return concatenateOptions( metadata.driverOptions );
2243 : 0 : }
2244 : :
2245 : 0 : QStringList QgsVectorFileWriter::defaultLayerOptions( const QString &driverName )
2246 : : {
2247 : 0 : MetaData metadata;
2248 : 0 : bool ok = driverMetadata( driverName, metadata );
2249 : 0 : if ( !ok )
2250 : 0 : return QStringList();
2251 : 0 : return concatenateOptions( metadata.layerOptions );
2252 : 0 : }
2253 : :
2254 : 0 : OGRwkbGeometryType QgsVectorFileWriter::ogrTypeFromWkbType( QgsWkbTypes::Type type )
2255 : : {
2256 : :
2257 : 0 : OGRwkbGeometryType ogrType = static_cast<OGRwkbGeometryType>( type );
2258 : :
2259 : 0 : if ( type >= QgsWkbTypes::PointZ && type <= QgsWkbTypes::GeometryCollectionZ )
2260 : : {
2261 : 0 : ogrType = static_cast<OGRwkbGeometryType>( QgsWkbTypes::to25D( type ) );
2262 : 0 : }
2263 : 0 : return ogrType;
2264 : : }
2265 : :
2266 : 0 : QgsVectorFileWriter::WriterError QgsVectorFileWriter::hasError()
2267 : : {
2268 : 0 : return mError;
2269 : : }
2270 : :
2271 : 0 : QString QgsVectorFileWriter::errorMessage()
2272 : : {
2273 : 0 : return mErrorMessage;
2274 : : }
2275 : :
2276 : 0 : bool QgsVectorFileWriter::addFeature( QgsFeature &feature, QgsFeatureSink::Flags )
2277 : : {
2278 : 0 : return addFeatureWithStyle( feature, nullptr, QgsUnitTypes::DistanceMeters );
2279 : : }
2280 : :
2281 : 0 : bool QgsVectorFileWriter::addFeatures( QgsFeatureList &features, QgsFeatureSink::Flags )
2282 : : {
2283 : 0 : QgsFeatureList::iterator fIt = features.begin();
2284 : 0 : bool result = true;
2285 : 0 : for ( ; fIt != features.end(); ++fIt )
2286 : : {
2287 : 0 : result = result && addFeatureWithStyle( *fIt, nullptr, QgsUnitTypes::DistanceMeters );
2288 : 0 : }
2289 : 0 : return result;
2290 : : }
2291 : :
2292 : 0 : QString QgsVectorFileWriter::lastError() const
2293 : : {
2294 : 0 : return mErrorMessage;
2295 : : }
2296 : :
2297 : 0 : bool QgsVectorFileWriter::addFeatureWithStyle( QgsFeature &feature, QgsFeatureRenderer *renderer, QgsUnitTypes::DistanceUnit outputUnit )
2298 : : {
2299 : : // create the feature
2300 : 0 : gdal::ogr_feature_unique_ptr poFeature = createFeature( feature );
2301 : 0 : if ( !poFeature )
2302 : 0 : return false;
2303 : :
2304 : : //add OGR feature style type
2305 : 0 : if ( mSymbologyExport != NoSymbology && renderer )
2306 : : {
2307 : 0 : mRenderContext.expressionContext().setFeature( feature );
2308 : : //SymbolLayerSymbology: concatenate ogr styles of all symbollayers
2309 : 0 : QgsSymbolList symbols = renderer->symbolsForFeature( feature, mRenderContext );
2310 : 0 : QString styleString;
2311 : 0 : QString currentStyle;
2312 : :
2313 : 0 : QgsSymbolList::const_iterator symbolIt = symbols.constBegin();
2314 : 0 : for ( ; symbolIt != symbols.constEnd(); ++symbolIt )
2315 : : {
2316 : 0 : int nSymbolLayers = ( *symbolIt )->symbolLayerCount();
2317 : 0 : for ( int i = 0; i < nSymbolLayers; ++i )
2318 : : {
2319 : : #if 0
2320 : : QMap< QgsSymbolLayer *, QString >::const_iterator it = mSymbolLayerTable.find( ( *symbolIt )->symbolLayer( i ) );
2321 : : if ( it == mSymbolLayerTable.constEnd() )
2322 : : {
2323 : : continue;
2324 : : }
2325 : : #endif
2326 : 0 : double mmsf = mmScaleFactor( mSymbologyScale, ( *symbolIt )->outputUnit(), outputUnit );
2327 : 0 : double musf = mapUnitScaleFactor( mSymbologyScale, ( *symbolIt )->outputUnit(), outputUnit );
2328 : :
2329 : 0 : currentStyle = ( *symbolIt )->symbolLayer( i )->ogrFeatureStyle( mmsf, musf );//"@" + it.value();
2330 : :
2331 : 0 : if ( mSymbologyExport == FeatureSymbology )
2332 : : {
2333 : 0 : if ( symbolIt != symbols.constBegin() || i != 0 )
2334 : : {
2335 : 0 : styleString.append( ';' );
2336 : 0 : }
2337 : 0 : styleString.append( currentStyle );
2338 : 0 : }
2339 : 0 : else if ( mSymbologyExport == SymbolLayerSymbology )
2340 : : {
2341 : 0 : OGR_F_SetStyleString( poFeature.get(), currentStyle.toLocal8Bit().constData() );
2342 : 0 : if ( !writeFeature( mLayer, poFeature.get() ) )
2343 : : {
2344 : 0 : return false;
2345 : : }
2346 : 0 : }
2347 : 0 : }
2348 : 0 : }
2349 : 0 : OGR_F_SetStyleString( poFeature.get(), styleString.toLocal8Bit().constData() );
2350 : 0 : }
2351 : :
2352 : 0 : if ( mSymbologyExport == NoSymbology || mSymbologyExport == FeatureSymbology )
2353 : : {
2354 : 0 : if ( !writeFeature( mLayer, poFeature.get() ) )
2355 : : {
2356 : 0 : return false;
2357 : : }
2358 : 0 : }
2359 : :
2360 : 0 : return true;
2361 : 0 : }
2362 : :
2363 : 0 : gdal::ogr_feature_unique_ptr QgsVectorFileWriter::createFeature( const QgsFeature &feature )
2364 : : {
2365 : 0 : QgsLocaleNumC l; // Make sure the decimal delimiter is a dot
2366 : : Q_UNUSED( l )
2367 : :
2368 : 0 : gdal::ogr_feature_unique_ptr poFeature( OGR_F_Create( OGR_L_GetLayerDefn( mLayer ) ) );
2369 : :
2370 : 0 : qint64 fid = FID_TO_NUMBER( feature.id() );
2371 : 0 : if ( fid > std::numeric_limits<int>::max() )
2372 : : {
2373 : 0 : QgsDebugMsg( QStringLiteral( "feature id %1 too large." ).arg( fid ) );
2374 : 0 : OGRErr err = OGR_F_SetFID( poFeature.get(), static_cast<long>( fid ) );
2375 : 0 : if ( err != OGRERR_NONE )
2376 : : {
2377 : 0 : QgsDebugMsg( QStringLiteral( "Failed to set feature id to %1: %2 (OGR error: %3)" )
2378 : : .arg( feature.id() )
2379 : : .arg( err ).arg( CPLGetLastErrorMsg() )
2380 : : );
2381 : 0 : }
2382 : 0 : }
2383 : :
2384 : : // attribute handling
2385 : 0 : for ( QMap<int, int>::const_iterator it = mAttrIdxToOgrIdx.constBegin(); it != mAttrIdxToOgrIdx.constEnd(); ++it )
2386 : : {
2387 : 0 : int fldIdx = it.key();
2388 : 0 : int ogrField = it.value();
2389 : :
2390 : 0 : QVariant attrValue = feature.attribute( fldIdx );
2391 : 0 : QgsField field = mFields.at( fldIdx );
2392 : :
2393 : 0 : if ( !attrValue.isValid() || attrValue.isNull() )
2394 : : {
2395 : : // Starting with GDAL 2.2, there are 2 concepts: unset fields and null fields
2396 : : // whereas previously there was only unset fields. For a GeoJSON output,
2397 : : // leaving a field unset will cause it to not appear at all in the output
2398 : : // feature.
2399 : : // When all features of a layer have a field unset, this would cause the
2400 : : // field to not be present at all in the output, and thus on reading to
2401 : : // have disappeared. #16812
2402 : : #ifdef OGRNullMarker
2403 : 0 : OGR_F_SetFieldNull( poFeature.get(), ogrField );
2404 : : #endif
2405 : 0 : continue;
2406 : : }
2407 : :
2408 : 0 : if ( mFieldValueConverter )
2409 : : {
2410 : 0 : field = mFieldValueConverter->fieldDefinition( field );
2411 : 0 : attrValue = mFieldValueConverter->convert( fldIdx, attrValue );
2412 : 0 : }
2413 : :
2414 : : // Check type compatibility before passing attribute value to OGR
2415 : 0 : QString errorMessage;
2416 : 0 : if ( ! field.convertCompatible( attrValue, &errorMessage ) )
2417 : : {
2418 : 0 : mErrorMessage = QObject::tr( "Error converting value (%1) for attribute field %2: %3" )
2419 : 0 : .arg( feature.attribute( fldIdx ).toString(),
2420 : 0 : mFields.at( fldIdx ).name(), errorMessage );
2421 : 0 : QgsMessageLog::logMessage( mErrorMessage, QObject::tr( "OGR" ) );
2422 : 0 : mError = ErrFeatureWriteFailed;
2423 : 0 : return nullptr;
2424 : : }
2425 : :
2426 : 0 : switch ( field.type() )
2427 : : {
2428 : : case QVariant::Int:
2429 : 0 : OGR_F_SetFieldInteger( poFeature.get(), ogrField, attrValue.toInt() );
2430 : 0 : break;
2431 : : case QVariant::LongLong:
2432 : 0 : OGR_F_SetFieldInteger64( poFeature.get(), ogrField, attrValue.toLongLong() );
2433 : 0 : break;
2434 : : case QVariant::Bool:
2435 : 0 : OGR_F_SetFieldInteger( poFeature.get(), ogrField, attrValue.toInt() );
2436 : 0 : break;
2437 : : case QVariant::String:
2438 : 0 : OGR_F_SetFieldString( poFeature.get(), ogrField, mCodec->fromUnicode( attrValue.toString() ).constData() );
2439 : 0 : break;
2440 : : case QVariant::Double:
2441 : 0 : OGR_F_SetFieldDouble( poFeature.get(), ogrField, attrValue.toDouble() );
2442 : 0 : break;
2443 : : case QVariant::Date:
2444 : 0 : OGR_F_SetFieldDateTime( poFeature.get(), ogrField,
2445 : 0 : attrValue.toDate().year(),
2446 : 0 : attrValue.toDate().month(),
2447 : 0 : attrValue.toDate().day(),
2448 : : 0, 0, 0, 0 );
2449 : 0 : break;
2450 : : case QVariant::DateTime:
2451 : 0 : if ( mOgrDriverName == QLatin1String( "ESRI Shapefile" ) )
2452 : : {
2453 : 0 : OGR_F_SetFieldString( poFeature.get(), ogrField, mCodec->fromUnicode( attrValue.toDateTime().toString( QStringLiteral( "yyyy/MM/dd hh:mm:ss.zzz" ) ) ).constData() );
2454 : 0 : }
2455 : : else
2456 : : {
2457 : 0 : OGR_F_SetFieldDateTime( poFeature.get(), ogrField,
2458 : 0 : attrValue.toDateTime().date().year(),
2459 : 0 : attrValue.toDateTime().date().month(),
2460 : 0 : attrValue.toDateTime().date().day(),
2461 : 0 : attrValue.toDateTime().time().hour(),
2462 : 0 : attrValue.toDateTime().time().minute(),
2463 : 0 : attrValue.toDateTime().time().second(),
2464 : : 0 );
2465 : : }
2466 : 0 : break;
2467 : : case QVariant::Time:
2468 : 0 : if ( mOgrDriverName == QLatin1String( "ESRI Shapefile" ) )
2469 : : {
2470 : 0 : OGR_F_SetFieldString( poFeature.get(), ogrField, mCodec->fromUnicode( attrValue.toString() ).constData() );
2471 : 0 : }
2472 : : else
2473 : : {
2474 : 0 : OGR_F_SetFieldDateTime( poFeature.get(), ogrField,
2475 : : 0, 0, 0,
2476 : 0 : attrValue.toTime().hour(),
2477 : 0 : attrValue.toTime().minute(),
2478 : 0 : attrValue.toTime().second(),
2479 : : 0 );
2480 : : }
2481 : 0 : break;
2482 : :
2483 : : case QVariant::ByteArray:
2484 : : {
2485 : 0 : const QByteArray ba = attrValue.toByteArray();
2486 : 0 : OGR_F_SetFieldBinary( poFeature.get(), ogrField, ba.size(), const_cast< GByte * >( reinterpret_cast< const GByte * >( ba.data() ) ) );
2487 : : break;
2488 : 0 : }
2489 : :
2490 : : case QVariant::Invalid:
2491 : 0 : break;
2492 : :
2493 : : case QVariant::List:
2494 : : // only string list supported at the moment, fall through to default for other types
2495 : 0 : if ( field.subType() == QVariant::String )
2496 : : {
2497 : 0 : QStringList list = attrValue.toStringList();
2498 : 0 : if ( supportsStringList )
2499 : : {
2500 : 0 : int count = list.count();
2501 : 0 : char **lst = new char *[count + 1];
2502 : 0 : if ( count > 0 )
2503 : : {
2504 : 0 : int pos = 0;
2505 : 0 : for ( QString string : list )
2506 : : {
2507 : 0 : lst[pos] = mCodec->fromUnicode( string ).data();
2508 : 0 : pos++;
2509 : 0 : }
2510 : 0 : }
2511 : 0 : lst[count] = nullptr;
2512 : 0 : OGR_F_SetFieldStringList( poFeature.get(), ogrField, lst );
2513 : 0 : }
2514 : : else
2515 : : {
2516 : 0 : OGR_F_SetFieldString( poFeature.get(), ogrField, mCodec->fromUnicode( list.join( ',' ) ).constData() );
2517 : : }
2518 : : break;
2519 : 0 : }
2520 : : //intentional fall-through
2521 : : FALLTHROUGH
2522 : :
2523 : : default:
2524 : 0 : mErrorMessage = QObject::tr( "Invalid variant type for field %1[%2]: received %3 with type %4" )
2525 : 0 : .arg( mFields.at( fldIdx ).name() )
2526 : 0 : .arg( ogrField )
2527 : 0 : .arg( attrValue.typeName(),
2528 : 0 : attrValue.toString() );
2529 : 0 : QgsMessageLog::logMessage( mErrorMessage, QObject::tr( "OGR" ) );
2530 : 0 : mError = ErrFeatureWriteFailed;
2531 : 0 : return nullptr;
2532 : : }
2533 : 0 : }
2534 : :
2535 : 0 : if ( mWkbType != QgsWkbTypes::NoGeometry )
2536 : : {
2537 : 0 : if ( feature.hasGeometry() )
2538 : : {
2539 : : // build geometry from WKB
2540 : 0 : QgsGeometry geom = feature.geometry();
2541 : 0 : if ( mCoordinateTransform )
2542 : : {
2543 : : // output dataset requires coordinate transform
2544 : : try
2545 : : {
2546 : 0 : geom.transform( *mCoordinateTransform );
2547 : 0 : }
2548 : : catch ( QgsCsException & )
2549 : : {
2550 : 0 : QgsLogger::warning( QObject::tr( "Feature geometry failed to transform" ) );
2551 : 0 : return nullptr;
2552 : 0 : }
2553 : 0 : }
2554 : :
2555 : : // turn single geometry to multi geometry if needed
2556 : 0 : if ( QgsWkbTypes::flatType( geom.wkbType() ) != QgsWkbTypes::flatType( mWkbType ) &&
2557 : 0 : QgsWkbTypes::flatType( geom.wkbType() ) == QgsWkbTypes::flatType( QgsWkbTypes::singleType( mWkbType ) ) )
2558 : : {
2559 : 0 : geom.convertToMultiType();
2560 : 0 : }
2561 : :
2562 : 0 : if ( geom.wkbType() != mWkbType )
2563 : : {
2564 : 0 : OGRGeometryH mGeom2 = nullptr;
2565 : :
2566 : : // If requested WKB type is 25D and geometry WKB type is 3D,
2567 : : // we must force the use of 25D.
2568 : 0 : if ( mWkbType >= QgsWkbTypes::Point25D && mWkbType <= QgsWkbTypes::MultiPolygon25D )
2569 : : {
2570 : : //ND: I suspect there's a bug here, in that this is NOT converting the geometry's WKB type,
2571 : : //so the exported WKB has a different type to what the OGRGeometry is expecting.
2572 : : //possibly this is handled already in OGR, but it should be fixed regardless by actually converting
2573 : : //geom to the correct WKB type
2574 : 0 : QgsWkbTypes::Type wkbType = geom.wkbType();
2575 : 0 : if ( wkbType >= QgsWkbTypes::PointZ && wkbType <= QgsWkbTypes::MultiPolygonZ )
2576 : : {
2577 : 0 : QgsWkbTypes::Type wkbType25d = static_cast<QgsWkbTypes::Type>( geom.wkbType() - QgsWkbTypes::PointZ + QgsWkbTypes::Point25D );
2578 : 0 : mGeom2 = createEmptyGeometry( wkbType25d );
2579 : 0 : }
2580 : 0 : }
2581 : :
2582 : : // drop m/z value if not present in output wkb type
2583 : 0 : if ( !QgsWkbTypes::hasZ( mWkbType ) && QgsWkbTypes::hasZ( geom.wkbType() ) )
2584 : 0 : geom.get()->dropZValue();
2585 : 0 : if ( !QgsWkbTypes::hasM( mWkbType ) && QgsWkbTypes::hasM( geom.wkbType() ) )
2586 : 0 : geom.get()->dropMValue();
2587 : :
2588 : : // add m/z values if not present in the input wkb type -- this is needed for formats which determine
2589 : : // geometry type based on features, e.g. geojson
2590 : 0 : if ( QgsWkbTypes::hasZ( mWkbType ) && !QgsWkbTypes::hasZ( geom.wkbType() ) )
2591 : 0 : geom.get()->addZValue( 0 );
2592 : 0 : if ( QgsWkbTypes::hasM( mWkbType ) && !QgsWkbTypes::hasM( geom.wkbType() ) )
2593 : 0 : geom.get()->addMValue( 0 );
2594 : :
2595 : 0 : if ( !mGeom2 )
2596 : : {
2597 : : // there's a problem when layer type is set as wkbtype Polygon
2598 : : // although there are also features of type MultiPolygon
2599 : : // (at least in OGR provider)
2600 : : // If the feature's wkbtype is different from the layer's wkbtype,
2601 : : // try to export it too.
2602 : : //
2603 : : // Btw. OGRGeometry must be exactly of the type of the geometry which it will receive
2604 : : // i.e. Polygons can't be imported to OGRMultiPolygon
2605 : 0 : mGeom2 = createEmptyGeometry( geom.wkbType() );
2606 : 0 : }
2607 : :
2608 : 0 : if ( !mGeom2 )
2609 : : {
2610 : 0 : mErrorMessage = QObject::tr( "Feature geometry not imported (OGR error: %1)" )
2611 : 0 : .arg( QString::fromUtf8( CPLGetLastErrorMsg() ) );
2612 : 0 : mError = ErrFeatureWriteFailed;
2613 : 0 : QgsMessageLog::logMessage( mErrorMessage, QObject::tr( "OGR" ) );
2614 : 0 : return nullptr;
2615 : : }
2616 : :
2617 : 0 : QByteArray wkb( geom.asWkb() );
2618 : 0 : OGRErr err = OGR_G_ImportFromWkb( mGeom2, reinterpret_cast<unsigned char *>( const_cast<char *>( wkb.constData() ) ), wkb.length() );
2619 : 0 : if ( err != OGRERR_NONE )
2620 : : {
2621 : 0 : mErrorMessage = QObject::tr( "Feature geometry not imported (OGR error: %1)" )
2622 : 0 : .arg( QString::fromUtf8( CPLGetLastErrorMsg() ) );
2623 : 0 : mError = ErrFeatureWriteFailed;
2624 : 0 : QgsMessageLog::logMessage( mErrorMessage, QObject::tr( "OGR" ) );
2625 : 0 : return nullptr;
2626 : : }
2627 : :
2628 : : // pass ownership to geometry
2629 : 0 : OGR_F_SetGeometryDirectly( poFeature.get(), mGeom2 );
2630 : 0 : }
2631 : : else // wkb type matches
2632 : : {
2633 : 0 : QByteArray wkb( geom.asWkb( QgsAbstractGeometry::FlagExportTrianglesAsPolygons ) );
2634 : 0 : OGRGeometryH ogrGeom = createEmptyGeometry( mWkbType );
2635 : 0 : OGRErr err = OGR_G_ImportFromWkb( ogrGeom, reinterpret_cast<unsigned char *>( const_cast<char *>( wkb.constData() ) ), wkb.length() );
2636 : 0 : if ( err != OGRERR_NONE )
2637 : : {
2638 : 0 : mErrorMessage = QObject::tr( "Feature geometry not imported (OGR error: %1)" )
2639 : 0 : .arg( QString::fromUtf8( CPLGetLastErrorMsg() ) );
2640 : 0 : mError = ErrFeatureWriteFailed;
2641 : 0 : QgsMessageLog::logMessage( mErrorMessage, QObject::tr( "OGR" ) );
2642 : 0 : return nullptr;
2643 : : }
2644 : :
2645 : : // set geometry (ownership is passed to OGR)
2646 : 0 : OGR_F_SetGeometryDirectly( poFeature.get(), ogrGeom );
2647 : 0 : }
2648 : 0 : }
2649 : : else
2650 : : {
2651 : 0 : OGR_F_SetGeometryDirectly( poFeature.get(), createEmptyGeometry( mWkbType ) );
2652 : : }
2653 : 0 : }
2654 : 0 : return poFeature;
2655 : 0 : }
2656 : :
2657 : 0 : void QgsVectorFileWriter::resetMap( const QgsAttributeList &attributes )
2658 : : {
2659 : 0 : QMap<int, int> omap( mAttrIdxToOgrIdx );
2660 : 0 : mAttrIdxToOgrIdx.clear();
2661 : 0 : for ( int i = 0; i < attributes.size(); i++ )
2662 : : {
2663 : 0 : if ( omap.find( i ) != omap.end() )
2664 : 0 : mAttrIdxToOgrIdx.insert( attributes[i], omap[i] );
2665 : 0 : }
2666 : 0 : }
2667 : :
2668 : 0 : bool QgsVectorFileWriter::writeFeature( OGRLayerH layer, OGRFeatureH feature )
2669 : : {
2670 : 0 : if ( OGR_L_CreateFeature( layer, feature ) != OGRERR_NONE )
2671 : : {
2672 : 0 : mErrorMessage = QObject::tr( "Feature creation error (OGR error: %1)" ).arg( QString::fromUtf8( CPLGetLastErrorMsg() ) );
2673 : 0 : mError = ErrFeatureWriteFailed;
2674 : 0 : QgsMessageLog::logMessage( mErrorMessage, QObject::tr( "OGR" ) );
2675 : 0 : return false;
2676 : : }
2677 : 0 : return true;
2678 : 0 : }
2679 : :
2680 : 0 : QgsVectorFileWriter::~QgsVectorFileWriter()
2681 : 0 : {
2682 : 0 : if ( mUsingTransaction )
2683 : : {
2684 : 0 : if ( OGRERR_NONE != OGR_L_CommitTransaction( mLayer ) )
2685 : : {
2686 : 0 : QgsDebugMsg( QStringLiteral( "Error while committing transaction on OGRLayer." ) );
2687 : 0 : }
2688 : 0 : }
2689 : :
2690 : : #if GDAL_VERSION_NUM >= GDAL_COMPUTE_VERSION(3,1,0) && GDAL_VERSION_NUM <= GDAL_COMPUTE_VERSION(3,1,3)
2691 : : if ( mDS )
2692 : : {
2693 : : // Workaround bug in GDAL 3.1.0 to 3.1.3 that creates XLSX and ODS files incompatible with LibreOffice due to use of ZIP64
2694 : : QString drvName = GDALGetDriverShortName( GDALGetDatasetDriver( mDS.get() ) );
2695 : : if ( drvName == QLatin1String( "XLSX" ) ||
2696 : : drvName == QLatin1String( "ODS" ) )
2697 : : {
2698 : : CPLSetThreadLocalConfigOption( "CPL_CREATE_ZIP64", "NO" );
2699 : : mDS.reset();
2700 : : CPLSetThreadLocalConfigOption( "CPL_CREATE_ZIP64", nullptr );
2701 : : }
2702 : : }
2703 : : #endif
2704 : :
2705 : 0 : mDS.reset();
2706 : :
2707 : 0 : if ( mOgrRef )
2708 : : {
2709 : 0 : OSRDestroySpatialReference( mOgrRef );
2710 : 0 : }
2711 : 0 : }
2712 : :
2713 : : QgsVectorFileWriter::WriterError
2714 : 0 : QgsVectorFileWriter::writeAsVectorFormat( QgsVectorLayer *layer,
2715 : : const QString &fileName,
2716 : : const QString &fileEncoding,
2717 : : const QgsCoordinateReferenceSystem &destCRS,
2718 : : const QString &driverName,
2719 : : bool onlySelected,
2720 : : QString *errorMessage,
2721 : : const QStringList &datasourceOptions,
2722 : : const QStringList &layerOptions,
2723 : : bool skipAttributeCreation,
2724 : : QString *newFilename,
2725 : : SymbologyExport symbologyExport,
2726 : : double symbologyScale,
2727 : : const QgsRectangle *filterExtent,
2728 : : QgsWkbTypes::Type overrideGeometryType,
2729 : : bool forceMulti,
2730 : : bool includeZ,
2731 : : const QgsAttributeList &attributes,
2732 : : FieldValueConverter *fieldValueConverter,
2733 : : QString *newLayer )
2734 : : {
2735 : 0 : QgsCoordinateTransform ct;
2736 : 0 : if ( destCRS.isValid() && layer )
2737 : : {
2738 : 0 : ct = QgsCoordinateTransform( layer->crs(), destCRS, layer->transformContext() );
2739 : 0 : }
2740 : :
2741 : 0 : SaveVectorOptions options;
2742 : 0 : options.fileEncoding = fileEncoding;
2743 : 0 : options.ct = ct;
2744 : 0 : options.driverName = driverName;
2745 : 0 : options.onlySelectedFeatures = onlySelected;
2746 : 0 : options.datasourceOptions = datasourceOptions;
2747 : 0 : options.layerOptions = layerOptions;
2748 : 0 : options.skipAttributeCreation = skipAttributeCreation;
2749 : 0 : options.symbologyExport = symbologyExport;
2750 : 0 : options.symbologyScale = symbologyScale;
2751 : 0 : if ( filterExtent )
2752 : 0 : options.filterExtent = *filterExtent;
2753 : 0 : options.overrideGeometryType = overrideGeometryType;
2754 : 0 : options.forceMulti = forceMulti;
2755 : 0 : options.includeZ = includeZ;
2756 : 0 : options.attributes = attributes;
2757 : 0 : options.fieldValueConverter = fieldValueConverter;
2758 : 0 : return writeAsVectorFormatV2( layer, fileName, layer->transformContext(), options, newFilename, newLayer, errorMessage );
2759 : 0 : }
2760 : :
2761 : 0 : QgsVectorFileWriter::WriterError QgsVectorFileWriter::writeAsVectorFormat( QgsVectorLayer *layer,
2762 : : const QString &fileName,
2763 : : const QString &fileEncoding,
2764 : : const QgsCoordinateTransform &ct,
2765 : : const QString &driverName,
2766 : : bool onlySelected,
2767 : : QString *errorMessage,
2768 : : const QStringList &datasourceOptions,
2769 : : const QStringList &layerOptions,
2770 : : bool skipAttributeCreation,
2771 : : QString *newFilename,
2772 : : SymbologyExport symbologyExport,
2773 : : double symbologyScale,
2774 : : const QgsRectangle *filterExtent,
2775 : : QgsWkbTypes::Type overrideGeometryType,
2776 : : bool forceMulti,
2777 : : bool includeZ,
2778 : : const QgsAttributeList &attributes,
2779 : : FieldValueConverter *fieldValueConverter,
2780 : : QString *newLayer )
2781 : : {
2782 : 0 : SaveVectorOptions options;
2783 : 0 : options.fileEncoding = fileEncoding;
2784 : 0 : options.ct = ct;
2785 : 0 : options.driverName = driverName;
2786 : 0 : options.onlySelectedFeatures = onlySelected;
2787 : 0 : options.datasourceOptions = datasourceOptions;
2788 : 0 : options.layerOptions = layerOptions;
2789 : 0 : options.skipAttributeCreation = skipAttributeCreation;
2790 : 0 : options.symbologyExport = symbologyExport;
2791 : 0 : options.symbologyScale = symbologyScale;
2792 : 0 : if ( filterExtent )
2793 : 0 : options.filterExtent = *filterExtent;
2794 : 0 : options.overrideGeometryType = overrideGeometryType;
2795 : 0 : options.forceMulti = forceMulti;
2796 : 0 : options.includeZ = includeZ;
2797 : 0 : options.attributes = attributes;
2798 : 0 : options.fieldValueConverter = fieldValueConverter;
2799 : 0 : return writeAsVectorFormatV2( layer, fileName, layer->transformContext(), options, newFilename, newLayer, errorMessage );
2800 : 0 : }
2801 : :
2802 : :
2803 : 0 : QgsVectorFileWriter::SaveVectorOptions::SaveVectorOptions()
2804 : 0 : : driverName( QStringLiteral( "GPKG" ) )
2805 : 0 : {
2806 : 0 : }
2807 : :
2808 : :
2809 : :
2810 : 0 : QgsVectorFileWriter::WriterError QgsVectorFileWriter::prepareWriteAsVectorFormat( QgsVectorLayer *layer, const QgsVectorFileWriter::SaveVectorOptions &options, QgsVectorFileWriter::PreparedWriterDetails &details )
2811 : : {
2812 : 0 : if ( !layer || !layer->isValid() )
2813 : : {
2814 : 0 : return ErrInvalidLayer;
2815 : : }
2816 : :
2817 : 0 : if ( layer->renderer() )
2818 : 0 : details.renderer.reset( layer->renderer()->clone() );
2819 : 0 : details.sourceCrs = layer->crs();
2820 : 0 : details.sourceWkbType = layer->wkbType();
2821 : 0 : details.sourceFields = layer->fields();
2822 : 0 : details.providerType = layer->providerType();
2823 : 0 : details.featureCount = options.onlySelectedFeatures ? layer->selectedFeatureCount() : layer->featureCount();
2824 : 0 : if ( layer->dataProvider() )
2825 : 0 : details.dataSourceUri = layer->dataProvider()->dataSourceUri();
2826 : 0 : details.storageType = layer->storageType();
2827 : 0 : details.selectedFeatureIds = layer->selectedFeatureIds();
2828 : 0 : details.providerUriParams = QgsProviderRegistry::instance()->decodeUri( layer->providerType(), layer->dataProvider()->dataSourceUri() );
2829 : :
2830 : 0 : if ( details.storageType == QLatin1String( "ESRI Shapefile" ) )
2831 : : {
2832 : 0 : QgsFeatureRequest req;
2833 : 0 : if ( options.onlySelectedFeatures )
2834 : : {
2835 : 0 : req.setFilterFids( details.selectedFeatureIds );
2836 : 0 : }
2837 : 0 : req.setNoAttributes();
2838 : 0 : details.geometryTypeScanIterator = layer->getFeatures( req );
2839 : 0 : }
2840 : :
2841 : 0 : details.expressionContext = QgsExpressionContext( QgsExpressionContextUtils::globalProjectLayerScopes( layer ) );
2842 : 0 : details.renderContext.setExpressionContext( details.expressionContext );
2843 : 0 : details.renderContext.setRendererScale( options.symbologyScale );
2844 : :
2845 : 0 : details.shallTransform = false;
2846 : 0 : if ( options.ct.isValid() )
2847 : : {
2848 : : // This means we should transform
2849 : 0 : details.outputCrs = options.ct.destinationCrs();
2850 : 0 : details.shallTransform = true;
2851 : 0 : }
2852 : : else
2853 : : {
2854 : : // This means we shouldn't transform, use source CRS as output (if defined)
2855 : 0 : details.outputCrs = details.sourceCrs;
2856 : : }
2857 : :
2858 : 0 : details.destWkbType = details.sourceWkbType;
2859 : 0 : if ( options.overrideGeometryType != QgsWkbTypes::Unknown )
2860 : : {
2861 : 0 : details.destWkbType = QgsWkbTypes::flatType( options.overrideGeometryType );
2862 : 0 : if ( QgsWkbTypes::hasZ( options.overrideGeometryType ) || options.includeZ )
2863 : 0 : details.destWkbType = QgsWkbTypes::addZ( details.destWkbType );
2864 : 0 : }
2865 : 0 : if ( options.forceMulti )
2866 : : {
2867 : 0 : details.destWkbType = QgsWkbTypes::multiType( details.destWkbType );
2868 : 0 : }
2869 : :
2870 : 0 : details.attributes = options.attributes;
2871 : 0 : if ( options.skipAttributeCreation )
2872 : 0 : details.attributes.clear();
2873 : 0 : else if ( details.attributes.isEmpty() )
2874 : : {
2875 : 0 : const QgsAttributeList allAttributes = details.sourceFields.allAttributesList();
2876 : 0 : for ( int idx : allAttributes )
2877 : : {
2878 : 0 : QgsField fld = details.sourceFields.at( idx );
2879 : 0 : if ( details.providerType == QLatin1String( "oracle" ) && fld.typeName().contains( QLatin1String( "SDO_GEOMETRY" ) ) )
2880 : 0 : continue;
2881 : 0 : details.attributes.append( idx );
2882 : 0 : }
2883 : 0 : }
2884 : :
2885 : 0 : if ( !details.attributes.isEmpty() )
2886 : : {
2887 : 0 : for ( int attrIdx : std::as_const( details.attributes ) )
2888 : : {
2889 : 0 : details.outputFields.append( details.sourceFields.at( attrIdx ) );
2890 : : }
2891 : 0 : }
2892 : :
2893 : : // not ideal - would be nice to avoid this happening in the preparation step if possible,
2894 : : // but currently requires access to the layer's minimumValue/maximumValue methods
2895 : 0 : if ( details.providerType == QLatin1String( "spatialite" ) )
2896 : : {
2897 : 0 : for ( int i = 0; i < details.outputFields.size(); i++ )
2898 : : {
2899 : 0 : if ( details.outputFields.at( i ).type() == QVariant::LongLong )
2900 : : {
2901 : 0 : QVariant min = layer->minimumValue( i );
2902 : 0 : QVariant max = layer->maximumValue( i );
2903 : 0 : if ( std::max( std::llabs( min.toLongLong() ), std::llabs( max.toLongLong() ) ) < std::numeric_limits<int>::max() )
2904 : : {
2905 : 0 : details.outputFields[i].setType( QVariant::Int );
2906 : 0 : }
2907 : 0 : }
2908 : 0 : }
2909 : 0 : }
2910 : :
2911 : :
2912 : : //add possible attributes needed by renderer
2913 : 0 : addRendererAttributes( details.renderer.get(), details.renderContext, details.sourceFields, details.attributes );
2914 : :
2915 : 0 : QgsFeatureRequest req;
2916 : 0 : req.setSubsetOfAttributes( details.attributes );
2917 : 0 : if ( options.onlySelectedFeatures )
2918 : 0 : req.setFilterFids( details.selectedFeatureIds );
2919 : :
2920 : 0 : if ( !options.filterExtent.isNull() )
2921 : : {
2922 : 0 : QgsRectangle filterRect = options.filterExtent;
2923 : 0 : bool useFilterRect = true;
2924 : 0 : if ( details.shallTransform )
2925 : : {
2926 : : try
2927 : : {
2928 : : // map filter rect back from destination CRS to layer CRS
2929 : 0 : filterRect = options.ct.transformBoundingBox( filterRect, QgsCoordinateTransform::ReverseTransform );
2930 : 0 : }
2931 : : catch ( QgsCsException & )
2932 : : {
2933 : 0 : useFilterRect = false;
2934 : 0 : }
2935 : 0 : }
2936 : 0 : if ( useFilterRect )
2937 : : {
2938 : 0 : req.setFilterRect( filterRect );
2939 : 0 : }
2940 : 0 : details.filterRectGeometry = QgsGeometry::fromRect( options.filterExtent );
2941 : 0 : details.filterRectEngine.reset( QgsGeometry::createGeometryEngine( details.filterRectGeometry.constGet() ) );
2942 : 0 : details.filterRectEngine->prepareGeometry();
2943 : 0 : }
2944 : 0 : details.sourceFeatureIterator = layer->getFeatures( req );
2945 : :
2946 : 0 : return NoError;
2947 : 0 : }
2948 : :
2949 : 0 : QgsVectorFileWriter::WriterError QgsVectorFileWriter::writeAsVectorFormat( PreparedWriterDetails &details, const QString &fileName, const QgsVectorFileWriter::SaveVectorOptions &options, QString *newFilename, QString *errorMessage, QString *newLayer )
2950 : : {
2951 : 0 : return writeAsVectorFormatV2( details, fileName, QgsCoordinateTransformContext(), options, newFilename, newLayer, errorMessage );
2952 : 0 : }
2953 : :
2954 : 0 : QgsVectorFileWriter::WriterError QgsVectorFileWriter::writeAsVectorFormatV2( PreparedWriterDetails &details, const QString &fileName, const QgsCoordinateTransformContext &transformContext, const QgsVectorFileWriter::SaveVectorOptions &options, QString *newFilename, QString *newLayer, QString *errorMessage )
2955 : : {
2956 : 0 : QgsWkbTypes::Type destWkbType = details.destWkbType;
2957 : :
2958 : 0 : int lastProgressReport = 0;
2959 : 0 : long total = details.featureCount;
2960 : :
2961 : : // Special rules for OGR layers
2962 : 0 : if ( details.providerType == QLatin1String( "ogr" ) && !details.dataSourceUri.isEmpty() )
2963 : : {
2964 : 0 : QString srcFileName( details.providerUriParams.value( QLatin1String( "path" ) ).toString() );
2965 : 0 : if ( QFile::exists( srcFileName ) && QFileInfo( fileName ).canonicalFilePath() == QFileInfo( srcFileName ).canonicalFilePath() )
2966 : : {
2967 : : // Check the layer name too if it's a GPKG/SpatiaLite/SQLite OGR driver (pay attention: camel case in layerName)
2968 : 0 : QgsDataSourceUri uri( details.dataSourceUri );
2969 : 0 : if ( !( ( options.driverName == QLatin1String( "GPKG" ) ||
2970 : 0 : options.driverName == QLatin1String( "SpatiaLite" ) ||
2971 : 0 : options.driverName == QLatin1String( "SQLite" ) ) &&
2972 : 0 : options.layerName != details.providerUriParams.value( QLatin1String( "layerName" ) ) ) )
2973 : : {
2974 : 0 : if ( errorMessage )
2975 : 0 : *errorMessage = QObject::tr( "Cannot overwrite a OGR layer in place" );
2976 : 0 : return ErrCreateDataSource;
2977 : : }
2978 : 0 : }
2979 : :
2980 : : // Shapefiles might contain multi types although wkbType() only reports singles
2981 : 0 : if ( details.storageType == QLatin1String( "ESRI Shapefile" ) && !QgsWkbTypes::isMultiType( destWkbType ) )
2982 : : {
2983 : 0 : QgsFeatureIterator fit = details.geometryTypeScanIterator;
2984 : 0 : QgsFeature fet;
2985 : 0 : long scanned = 0;
2986 : 0 : while ( fit.nextFeature( fet ) )
2987 : : {
2988 : 0 : if ( options.feedback && options.feedback->isCanceled() )
2989 : : {
2990 : 0 : return Canceled;
2991 : : }
2992 : 0 : if ( options.feedback )
2993 : : {
2994 : : //dedicate first 5% of progress bar to this scan
2995 : 0 : int newProgress = static_cast<int>( ( 5.0 * scanned ) / total );
2996 : 0 : if ( newProgress != lastProgressReport )
2997 : : {
2998 : 0 : lastProgressReport = newProgress;
2999 : 0 : options.feedback->setProgress( lastProgressReport );
3000 : 0 : }
3001 : 0 : }
3002 : :
3003 : 0 : if ( fet.hasGeometry() && QgsWkbTypes::isMultiType( fet.geometry().wkbType() ) )
3004 : : {
3005 : 0 : destWkbType = QgsWkbTypes::multiType( destWkbType );
3006 : 0 : break;
3007 : : }
3008 : 0 : scanned++;
3009 : : }
3010 : 0 : }
3011 : 0 : }
3012 : :
3013 : 0 : std::unique_ptr< QgsVectorFileWriter > writer( create( fileName, details.outputFields, destWkbType, details.outputCrs, transformContext, options, QgsFeatureSink::SinkFlags(), newFilename, newLayer ) );
3014 : 0 : writer->setSymbologyScale( options.symbologyScale );
3015 : :
3016 : 0 : if ( newFilename )
3017 : : {
3018 : 0 : QgsDebugMsgLevel( "newFilename = " + *newFilename, 2 );
3019 : 0 : }
3020 : :
3021 : : // check whether file creation was successful
3022 : 0 : WriterError err = writer->hasError();
3023 : 0 : if ( err != NoError )
3024 : : {
3025 : 0 : if ( errorMessage )
3026 : 0 : *errorMessage = writer->errorMessage();
3027 : 0 : return err;
3028 : : }
3029 : :
3030 : 0 : if ( errorMessage )
3031 : : {
3032 : 0 : errorMessage->clear();
3033 : 0 : }
3034 : :
3035 : 0 : QgsFeature fet;
3036 : :
3037 : : //create symbol table if needed
3038 : 0 : if ( writer->symbologyExport() != NoSymbology )
3039 : : {
3040 : : //writer->createSymbolLayerTable( layer, writer->mDS );
3041 : 0 : }
3042 : :
3043 : 0 : if ( writer->symbologyExport() == SymbolLayerSymbology )
3044 : : {
3045 : 0 : QgsFeatureRenderer *r = details.renderer.get();
3046 : 0 : if ( r && r->capabilities() & QgsFeatureRenderer::SymbolLevels
3047 : 0 : && r->usingSymbolLevels() )
3048 : : {
3049 : 0 : QgsVectorFileWriter::WriterError error = writer->exportFeaturesSymbolLevels( details, details.sourceFeatureIterator, options.ct, errorMessage );
3050 : 0 : return ( error == NoError ) ? NoError : ErrFeatureWriteFailed;
3051 : : }
3052 : 0 : }
3053 : :
3054 : 0 : int n = 0, errors = 0;
3055 : :
3056 : : //unit type
3057 : 0 : QgsUnitTypes::DistanceUnit mapUnits = details.sourceCrs.mapUnits();
3058 : 0 : if ( options.ct.isValid() )
3059 : : {
3060 : 0 : mapUnits = options.ct.destinationCrs().mapUnits();
3061 : 0 : }
3062 : :
3063 : 0 : writer->startRender( details.renderer.get(), details.sourceFields );
3064 : :
3065 : 0 : writer->resetMap( details.attributes );
3066 : : // Reset mFields to layer fields, and not just exported fields
3067 : 0 : writer->mFields = details.sourceFields;
3068 : :
3069 : : // write all features
3070 : 0 : long saved = 0;
3071 : 0 : int initialProgress = lastProgressReport;
3072 : 0 : while ( details.sourceFeatureIterator.nextFeature( fet ) )
3073 : : {
3074 : 0 : if ( options.feedback && options.feedback->isCanceled() )
3075 : : {
3076 : 0 : return Canceled;
3077 : : }
3078 : :
3079 : 0 : saved++;
3080 : 0 : if ( options.feedback )
3081 : : {
3082 : : //avoid spamming progress reports
3083 : 0 : int newProgress = static_cast<int>( initialProgress + ( ( 100.0 - initialProgress ) * saved ) / total );
3084 : 0 : if ( newProgress < 100 && newProgress != lastProgressReport )
3085 : : {
3086 : 0 : lastProgressReport = newProgress;
3087 : 0 : options.feedback->setProgress( lastProgressReport );
3088 : 0 : }
3089 : 0 : }
3090 : :
3091 : 0 : if ( details.shallTransform )
3092 : : {
3093 : : try
3094 : : {
3095 : 0 : if ( fet.hasGeometry() )
3096 : : {
3097 : 0 : QgsGeometry g = fet.geometry();
3098 : 0 : g.transform( options.ct );
3099 : 0 : fet.setGeometry( g );
3100 : 0 : }
3101 : 0 : }
3102 : : catch ( QgsCsException &e )
3103 : : {
3104 : 0 : QString msg = QObject::tr( "Failed to transform a point while drawing a feature with ID '%1'. Writing stopped. (Exception: %2)" )
3105 : 0 : .arg( fet.id() ).arg( e.what() );
3106 : 0 : QgsLogger::warning( msg );
3107 : 0 : if ( errorMessage )
3108 : 0 : *errorMessage = msg;
3109 : :
3110 : 0 : return ErrProjection;
3111 : 0 : }
3112 : 0 : }
3113 : :
3114 : 0 : if ( fet.hasGeometry() && details.filterRectEngine && !details.filterRectEngine->intersects( fet.geometry().constGet() ) )
3115 : 0 : continue;
3116 : :
3117 : 0 : if ( details.attributes.empty() && options.skipAttributeCreation )
3118 : : {
3119 : 0 : fet.initAttributes( 0 );
3120 : 0 : }
3121 : :
3122 : 0 : if ( !writer->addFeatureWithStyle( fet, writer->mRenderer.get(), mapUnits ) )
3123 : : {
3124 : 0 : WriterError err = writer->hasError();
3125 : 0 : if ( err != NoError && errorMessage )
3126 : : {
3127 : 0 : if ( errorMessage->isEmpty() )
3128 : : {
3129 : 0 : *errorMessage = QObject::tr( "Feature write errors:" );
3130 : 0 : }
3131 : 0 : *errorMessage += '\n' + writer->errorMessage();
3132 : 0 : }
3133 : 0 : errors++;
3134 : :
3135 : 0 : if ( errors > 1000 )
3136 : : {
3137 : 0 : if ( errorMessage )
3138 : : {
3139 : 0 : *errorMessage += QObject::tr( "Stopping after %1 errors" ).arg( errors );
3140 : 0 : }
3141 : :
3142 : 0 : n = -1;
3143 : 0 : break;
3144 : : }
3145 : 0 : }
3146 : 0 : n++;
3147 : : }
3148 : :
3149 : 0 : writer->stopRender();
3150 : :
3151 : 0 : if ( errors > 0 && errorMessage && n > 0 )
3152 : : {
3153 : 0 : *errorMessage += QObject::tr( "\nOnly %1 of %2 features written." ).arg( n - errors ).arg( n );
3154 : 0 : }
3155 : :
3156 : 0 : return errors == 0 ? NoError : ErrFeatureWriteFailed;
3157 : 0 : }
3158 : :
3159 : 0 : QgsVectorFileWriter::WriterError QgsVectorFileWriter::writeAsVectorFormat( QgsVectorLayer *layer,
3160 : : const QString &fileName,
3161 : : const SaveVectorOptions &options,
3162 : : QString *newFilename,
3163 : : QString *errorMessage,
3164 : : QString *newLayer )
3165 : : {
3166 : 0 : QgsVectorFileWriter::PreparedWriterDetails details;
3167 : 0 : WriterError err = prepareWriteAsVectorFormat( layer, options, details );
3168 : 0 : if ( err != NoError )
3169 : 0 : return err;
3170 : :
3171 : 0 : return writeAsVectorFormatV2( details, fileName, layer->transformContext(), options, newFilename, newLayer, errorMessage );
3172 : 0 : }
3173 : :
3174 : 0 : QgsVectorFileWriter::WriterError QgsVectorFileWriter::writeAsVectorFormatV2( QgsVectorLayer *layer,
3175 : : const QString &fileName,
3176 : : const QgsCoordinateTransformContext &transformContext,
3177 : : const SaveVectorOptions &options,
3178 : : QString *newFilename,
3179 : : QString *newLayer,
3180 : : QString *errorMessage )
3181 : : {
3182 : 0 : QgsVectorFileWriter::PreparedWriterDetails details;
3183 : 0 : WriterError err = prepareWriteAsVectorFormat( layer, options, details );
3184 : 0 : if ( err != NoError )
3185 : 0 : return err;
3186 : :
3187 : 0 : return writeAsVectorFormatV2( details, fileName, transformContext, options, newFilename, newLayer, errorMessage );
3188 : 0 : }
3189 : :
3190 : 0 : bool QgsVectorFileWriter::deleteShapeFile( const QString &fileName )
3191 : : {
3192 : 0 : QFileInfo fi( fileName );
3193 : 0 : QDir dir = fi.dir();
3194 : :
3195 : 0 : QStringList filter;
3196 : 0 : const char *suffixes[] = { ".shp", ".shx", ".dbf", ".prj", ".qix", ".qpj", ".cpg", ".sbn", ".sbx", ".idm", ".ind" };
3197 : 0 : for ( std::size_t i = 0; i < sizeof( suffixes ) / sizeof( *suffixes ); i++ )
3198 : : {
3199 : 0 : filter << fi.completeBaseName() + suffixes[i];
3200 : 0 : }
3201 : :
3202 : 0 : bool ok = true;
3203 : 0 : const auto constEntryList = dir.entryList( filter );
3204 : 0 : for ( const QString &file : constEntryList )
3205 : : {
3206 : 0 : QFile f( dir.canonicalPath() + '/' + file );
3207 : 0 : if ( !f.remove() )
3208 : : {
3209 : 0 : QgsDebugMsg( QStringLiteral( "Removing file %1 failed: %2" ).arg( file, f.errorString() ) );
3210 : 0 : ok = false;
3211 : 0 : }
3212 : 0 : }
3213 : :
3214 : 0 : return ok;
3215 : 0 : }
3216 : :
3217 : 0 : void QgsVectorFileWriter::setSymbologyScale( double d )
3218 : : {
3219 : 0 : mSymbologyScale = d;
3220 : 0 : mRenderContext.setRendererScale( mSymbologyScale );
3221 : 0 : }
3222 : :
3223 : 0 : QList< QgsVectorFileWriter::FilterFormatDetails > QgsVectorFileWriter::supportedFiltersAndFormats( const VectorFormatOptions options )
3224 : : {
3225 : 0 : static QReadWriteLock sFilterLock;
3226 : 0 : static QMap< VectorFormatOptions, QList< QgsVectorFileWriter::FilterFormatDetails > > sFilters;
3227 : :
3228 : 0 : QgsReadWriteLocker locker( sFilterLock, QgsReadWriteLocker::Read );
3229 : :
3230 : 0 : const auto it = sFilters.constFind( options );
3231 : 0 : if ( it != sFilters.constEnd() )
3232 : 0 : return it.value();
3233 : :
3234 : 0 : locker.changeMode( QgsReadWriteLocker::Write );
3235 : 0 : QList< QgsVectorFileWriter::FilterFormatDetails > results;
3236 : :
3237 : 0 : QgsApplication::registerOgrDrivers();
3238 : 0 : int const drvCount = OGRGetDriverCount();
3239 : :
3240 : 0 : for ( int i = 0; i < drvCount; ++i )
3241 : : {
3242 : 0 : OGRSFDriverH drv = OGRGetDriver( i );
3243 : 0 : if ( drv )
3244 : : {
3245 : 0 : QString drvName = OGR_Dr_GetName( drv );
3246 : :
3247 : 0 : GDALDriverH gdalDriver = GDALGetDriverByName( drvName.toLocal8Bit().constData() );
3248 : 0 : char **metadata = nullptr;
3249 : 0 : if ( gdalDriver )
3250 : : {
3251 : 0 : metadata = GDALGetMetadata( gdalDriver, nullptr );
3252 : 0 : }
3253 : :
3254 : 0 : bool nonSpatialFormat = CSLFetchBoolean( metadata, GDAL_DCAP_NONSPATIAL, false );
3255 : :
3256 : 0 : if ( OGR_Dr_TestCapability( drv, "CreateDataSource" ) != 0 )
3257 : : {
3258 : 0 : if ( options & SkipNonSpatialFormats )
3259 : : {
3260 : : // skip non-spatial formats
3261 : 0 : if ( nonSpatialFormat )
3262 : 0 : continue;
3263 : 0 : }
3264 : :
3265 : 0 : QString filterString = filterForDriver( drvName );
3266 : 0 : if ( filterString.isEmpty() )
3267 : 0 : continue;
3268 : :
3269 : 0 : MetaData metadata;
3270 : 0 : QStringList globs;
3271 : 0 : if ( driverMetadata( drvName, metadata ) && !metadata.glob.isEmpty() )
3272 : : {
3273 : 0 : globs = metadata.glob.toLower().split( ' ' );
3274 : 0 : }
3275 : :
3276 : 0 : FilterFormatDetails details;
3277 : 0 : details.driverName = drvName;
3278 : 0 : details.filterString = filterString;
3279 : 0 : details.globs = globs;
3280 : :
3281 : 0 : results << details;
3282 : 0 : }
3283 : 0 : }
3284 : 0 : }
3285 : :
3286 : 0 : std::sort( results.begin(), results.end(), [options]( const FilterFormatDetails & a, const FilterFormatDetails & b ) -> bool
3287 : : {
3288 : 0 : if ( options & SortRecommended )
3289 : : {
3290 : 0 : if ( a.driverName == QLatin1String( "GPKG" ) )
3291 : 0 : return true; // Make https://twitter.com/shapefiIe a sad little fellow
3292 : 0 : else if ( b.driverName == QLatin1String( "GPKG" ) )
3293 : 0 : return false;
3294 : 0 : else if ( a.driverName == QLatin1String( "ESRI Shapefile" ) )
3295 : 0 : return true;
3296 : 0 : else if ( b.driverName == QLatin1String( "ESRI Shapefile" ) )
3297 : 0 : return false;
3298 : 0 : }
3299 : :
3300 : 0 : return a.filterString.toLower().localeAwareCompare( b.filterString.toLower() ) < 0;
3301 : 0 : } );
3302 : :
3303 : 0 : sFilters.insert( options, results );
3304 : 0 : return results;
3305 : 0 : }
3306 : :
3307 : 0 : QStringList QgsVectorFileWriter::supportedFormatExtensions( const VectorFormatOptions options )
3308 : : {
3309 : 0 : const auto formats = supportedFiltersAndFormats( options );
3310 : 0 : QSet< QString > extensions;
3311 : :
3312 : 0 : const QRegularExpression rx( QStringLiteral( "\\*\\.(.*)$" ) );
3313 : :
3314 : 0 : for ( const FilterFormatDetails &format : formats )
3315 : : {
3316 : 0 : for ( const QString &glob : format.globs )
3317 : : {
3318 : 0 : const QRegularExpressionMatch match = rx.match( glob );
3319 : 0 : if ( !match.hasMatch() )
3320 : 0 : continue;
3321 : :
3322 : 0 : const QString matched = match.captured( 1 );
3323 : 0 : extensions.insert( matched );
3324 : 0 : }
3325 : : }
3326 : :
3327 : 0 : QStringList extensionList = qgis::setToList( extensions );
3328 : :
3329 : 0 : std::sort( extensionList.begin(), extensionList.end(), [options]( const QString & a, const QString & b ) -> bool
3330 : : {
3331 : 0 : if ( options & SortRecommended )
3332 : : {
3333 : 0 : if ( a == QLatin1String( "gpkg" ) )
3334 : 0 : return true; // Make https://twitter.com/shapefiIe a sad little fellow
3335 : 0 : else if ( b == QLatin1String( "gpkg" ) )
3336 : 0 : return false;
3337 : 0 : else if ( a == QLatin1String( "shp" ) )
3338 : 0 : return true;
3339 : 0 : else if ( b == QLatin1String( "shp" ) )
3340 : 0 : return false;
3341 : 0 : }
3342 : :
3343 : 0 : return a.toLower().localeAwareCompare( b.toLower() ) < 0;
3344 : 0 : } );
3345 : :
3346 : 0 : return extensionList;
3347 : 0 : }
3348 : :
3349 : 0 : QList< QgsVectorFileWriter::DriverDetails > QgsVectorFileWriter::ogrDriverList( const VectorFormatOptions options )
3350 : : {
3351 : 0 : QList< QgsVectorFileWriter::DriverDetails > results;
3352 : :
3353 : 0 : QgsApplication::registerOgrDrivers();
3354 : 0 : const int drvCount = OGRGetDriverCount();
3355 : :
3356 : 0 : QStringList writableDrivers;
3357 : 0 : for ( int i = 0; i < drvCount; ++i )
3358 : : {
3359 : 0 : OGRSFDriverH drv = OGRGetDriver( i );
3360 : 0 : if ( drv )
3361 : : {
3362 : 0 : QString drvName = OGR_Dr_GetName( drv );
3363 : :
3364 : 0 : if ( options & SkipNonSpatialFormats )
3365 : : {
3366 : : // skip non-spatial formats
3367 : : // TODO - use GDAL metadata to determine this, when support exists in GDAL
3368 : 0 : if ( drvName == QLatin1String( "ODS" ) || drvName == QLatin1String( "XLSX" ) || drvName == QLatin1String( "XLS" ) )
3369 : 0 : continue;
3370 : 0 : }
3371 : :
3372 : 0 : if ( drvName == QLatin1String( "ESRI Shapefile" ) )
3373 : : {
3374 : 0 : writableDrivers << QStringLiteral( "DBF file" );
3375 : 0 : }
3376 : 0 : if ( OGR_Dr_TestCapability( drv, "CreateDataSource" ) != 0 )
3377 : : {
3378 : : // Add separate format for Mapinfo MIF (MITAB is OGR default)
3379 : 0 : if ( drvName == QLatin1String( "MapInfo File" ) )
3380 : : {
3381 : 0 : writableDrivers << QStringLiteral( "MapInfo MIF" );
3382 : 0 : }
3383 : 0 : else if ( drvName == QLatin1String( "SQLite" ) )
3384 : : {
3385 : : // Unfortunately it seems that there is no simple way to detect if
3386 : : // OGR SQLite driver is compiled with SpatiaLite support.
3387 : : // We have HAVE_SPATIALITE in QGIS, but that may differ from OGR
3388 : : // http://lists.osgeo.org/pipermail/gdal-dev/2012-November/034580.html
3389 : : // -> test if creation fails
3390 : 0 : QString option = QStringLiteral( "SPATIALITE=YES" );
3391 : 0 : char *options[2] = { CPLStrdup( option.toLocal8Bit().constData() ), nullptr };
3392 : : OGRSFDriverH poDriver;
3393 : 0 : QgsApplication::registerOgrDrivers();
3394 : 0 : poDriver = OGRGetDriverByName( drvName.toLocal8Bit().constData() );
3395 : 0 : if ( poDriver )
3396 : : {
3397 : 0 : gdal::ogr_datasource_unique_ptr ds( OGR_Dr_CreateDataSource( poDriver, QStringLiteral( "/vsimem/spatialitetest.sqlite" ).toUtf8().constData(), options ) );
3398 : 0 : if ( ds )
3399 : : {
3400 : 0 : writableDrivers << QStringLiteral( "SpatiaLite" );
3401 : 0 : OGR_Dr_DeleteDataSource( poDriver, QStringLiteral( "/vsimem/spatialitetest.sqlite" ).toUtf8().constData() );
3402 : 0 : }
3403 : 0 : }
3404 : 0 : CPLFree( options[0] );
3405 : 0 : }
3406 : 0 : writableDrivers << drvName;
3407 : 0 : }
3408 : 0 : }
3409 : 0 : }
3410 : :
3411 : 0 : results.reserve( writableDrivers.count() );
3412 : 0 : for ( const QString &drvName : std::as_const( writableDrivers ) )
3413 : : {
3414 : 0 : MetaData metadata;
3415 : 0 : if ( driverMetadata( drvName, metadata ) && !metadata.trLongName.isEmpty() )
3416 : : {
3417 : 0 : DriverDetails details;
3418 : 0 : details.driverName = drvName;
3419 : 0 : details.longName = metadata.trLongName;
3420 : 0 : results << details;
3421 : 0 : }
3422 : 0 : }
3423 : :
3424 : 0 : std::sort( results.begin(), results.end(), [options]( const DriverDetails & a, const DriverDetails & b ) -> bool
3425 : : {
3426 : 0 : if ( options & SortRecommended )
3427 : : {
3428 : 0 : if ( a.driverName == QLatin1String( "GPKG" ) )
3429 : 0 : return true; // Make https://twitter.com/shapefiIe a sad little fellow
3430 : 0 : else if ( b.driverName == QLatin1String( "GPKG" ) )
3431 : 0 : return false;
3432 : 0 : else if ( a.driverName == QLatin1String( "ESRI Shapefile" ) )
3433 : 0 : return true;
3434 : 0 : else if ( b.driverName == QLatin1String( "ESRI Shapefile" ) )
3435 : 0 : return false;
3436 : 0 : }
3437 : :
3438 : 0 : return a.longName.toLower().localeAwareCompare( b.longName.toLower() ) < 0;
3439 : 0 : } );
3440 : 0 : return results;
3441 : 0 : }
3442 : :
3443 : 0 : QString QgsVectorFileWriter::driverForExtension( const QString &extension )
3444 : : {
3445 : 0 : QString ext = extension.trimmed();
3446 : 0 : if ( ext.isEmpty() )
3447 : 0 : return QString();
3448 : :
3449 : 0 : if ( ext.startsWith( '.' ) )
3450 : 0 : ext.remove( 0, 1 );
3451 : :
3452 : 0 : GDALAllRegister();
3453 : 0 : int const drvCount = GDALGetDriverCount();
3454 : :
3455 : 0 : for ( int i = 0; i < drvCount; ++i )
3456 : : {
3457 : 0 : GDALDriverH drv = GDALGetDriver( i );
3458 : 0 : if ( drv )
3459 : : {
3460 : 0 : char **driverMetadata = GDALGetMetadata( drv, nullptr );
3461 : 0 : if ( CSLFetchBoolean( driverMetadata, GDAL_DCAP_CREATE, false ) && CSLFetchBoolean( driverMetadata, GDAL_DCAP_VECTOR, false ) )
3462 : : {
3463 : 0 : QString drvName = GDALGetDriverShortName( drv );
3464 : 0 : QStringList driverExtensions = QString( GDALGetMetadataItem( drv, GDAL_DMD_EXTENSIONS, nullptr ) ).split( ' ' );
3465 : :
3466 : 0 : const auto constDriverExtensions = driverExtensions;
3467 : 0 : for ( const QString &driver : constDriverExtensions )
3468 : : {
3469 : 0 : if ( driver.compare( ext, Qt::CaseInsensitive ) == 0 )
3470 : 0 : return drvName;
3471 : : }
3472 : 0 : }
3473 : 0 : }
3474 : 0 : }
3475 : 0 : return QString();
3476 : 0 : }
3477 : :
3478 : 0 : QString QgsVectorFileWriter::fileFilterString( const VectorFormatOptions options )
3479 : : {
3480 : 0 : QString filterString;
3481 : 0 : const auto driverFormats = supportedFiltersAndFormats( options );
3482 : 0 : for ( const FilterFormatDetails &details : driverFormats )
3483 : : {
3484 : 0 : if ( !filterString.isEmpty() )
3485 : 0 : filterString += QLatin1String( ";;" );
3486 : :
3487 : 0 : filterString += details.filterString;
3488 : : }
3489 : 0 : return filterString;
3490 : 0 : }
3491 : :
3492 : 0 : QString QgsVectorFileWriter::filterForDriver( const QString &driverName )
3493 : : {
3494 : 0 : MetaData metadata;
3495 : 0 : if ( !driverMetadata( driverName, metadata ) || metadata.trLongName.isEmpty() || metadata.glob.isEmpty() )
3496 : 0 : return QString();
3497 : :
3498 : 0 : return QStringLiteral( "%1 (%2 %3)" ).arg( metadata.trLongName,
3499 : 0 : metadata.glob.toLower(),
3500 : 0 : metadata.glob.toUpper() );
3501 : 0 : }
3502 : :
3503 : 0 : QString QgsVectorFileWriter::convertCodecNameForEncodingOption( const QString &codecName )
3504 : : {
3505 : 0 : if ( codecName == QLatin1String( "System" ) )
3506 : 0 : return QStringLiteral( "LDID/0" );
3507 : :
3508 : 0 : QRegExp re = QRegExp( QString( "(CP|windows-|ISO[ -])(.+)" ), Qt::CaseInsensitive );
3509 : 0 : if ( re.exactMatch( codecName ) )
3510 : : {
3511 : 0 : QString c = re.cap( 2 ).remove( '-' );
3512 : : bool isNumber;
3513 : 0 : ( void ) c.toInt( &isNumber );
3514 : 0 : if ( isNumber )
3515 : 0 : return c;
3516 : 0 : }
3517 : 0 : return codecName;
3518 : 0 : }
3519 : :
3520 : 0 : void QgsVectorFileWriter::createSymbolLayerTable( QgsVectorLayer *vl, const QgsCoordinateTransform &ct, OGRDataSourceH ds )
3521 : : {
3522 : 0 : if ( !vl || !ds )
3523 : : {
3524 : 0 : return;
3525 : : }
3526 : :
3527 : 0 : QgsFeatureRenderer *renderer = vl->renderer();
3528 : 0 : if ( !renderer )
3529 : : {
3530 : 0 : return;
3531 : : }
3532 : :
3533 : : //unit type
3534 : 0 : QgsUnitTypes::DistanceUnit mapUnits = vl->crs().mapUnits();
3535 : 0 : if ( ct.isValid() )
3536 : : {
3537 : 0 : mapUnits = ct.destinationCrs().mapUnits();
3538 : 0 : }
3539 : :
3540 : 0 : mSymbolLayerTable.clear();
3541 : 0 : OGRStyleTableH ogrStyleTable = OGR_STBL_Create();
3542 : 0 : OGRStyleMgrH styleManager = OGR_SM_Create( ogrStyleTable );
3543 : :
3544 : : //get symbols
3545 : 0 : int nTotalLevels = 0;
3546 : 0 : QgsSymbolList symbolList = renderer->symbols( mRenderContext );
3547 : 0 : QgsSymbolList::iterator symbolIt = symbolList.begin();
3548 : 0 : for ( ; symbolIt != symbolList.end(); ++symbolIt )
3549 : : {
3550 : 0 : double mmsf = mmScaleFactor( mSymbologyScale, ( *symbolIt )->outputUnit(), mapUnits );
3551 : 0 : double musf = mapUnitScaleFactor( mSymbologyScale, ( *symbolIt )->outputUnit(), mapUnits );
3552 : :
3553 : 0 : int nLevels = ( *symbolIt )->symbolLayerCount();
3554 : 0 : for ( int i = 0; i < nLevels; ++i )
3555 : : {
3556 : 0 : mSymbolLayerTable.insert( ( *symbolIt )->symbolLayer( i ), QString::number( nTotalLevels ) );
3557 : 0 : OGR_SM_AddStyle( styleManager, QString::number( nTotalLevels ).toLocal8Bit(),
3558 : 0 : ( *symbolIt )->symbolLayer( i )->ogrFeatureStyle( mmsf, musf ).toLocal8Bit() );
3559 : 0 : ++nTotalLevels;
3560 : 0 : }
3561 : 0 : }
3562 : 0 : OGR_DS_SetStyleTableDirectly( ds, ogrStyleTable );
3563 : 0 : }
3564 : :
3565 : 0 : QgsVectorFileWriter::WriterError QgsVectorFileWriter::exportFeaturesSymbolLevels( const PreparedWriterDetails &details, QgsFeatureIterator &fit,
3566 : : const QgsCoordinateTransform &ct, QString *errorMessage )
3567 : : {
3568 : 0 : if ( !details.renderer )
3569 : 0 : return ErrInvalidLayer;
3570 : :
3571 : 0 : mRenderContext.expressionContext() = details.expressionContext;
3572 : :
3573 : 0 : QHash< QgsSymbol *, QList<QgsFeature> > features;
3574 : :
3575 : : //unit type
3576 : 0 : QgsUnitTypes::DistanceUnit mapUnits = details.sourceCrs.mapUnits();
3577 : 0 : if ( ct.isValid() )
3578 : : {
3579 : 0 : mapUnits = ct.destinationCrs().mapUnits();
3580 : 0 : }
3581 : :
3582 : 0 : startRender( details.renderer.get(), details.sourceFields );
3583 : :
3584 : : //fetch features
3585 : 0 : QgsFeature fet;
3586 : 0 : QgsSymbol *featureSymbol = nullptr;
3587 : 0 : while ( fit.nextFeature( fet ) )
3588 : : {
3589 : 0 : if ( ct.isValid() )
3590 : : {
3591 : : try
3592 : : {
3593 : 0 : if ( fet.hasGeometry() )
3594 : : {
3595 : 0 : QgsGeometry g = fet.geometry();
3596 : 0 : g.transform( ct );
3597 : 0 : fet.setGeometry( g );
3598 : 0 : }
3599 : 0 : }
3600 : : catch ( QgsCsException &e )
3601 : : {
3602 : 0 : QString msg = QObject::tr( "Failed to transform, writing stopped. (Exception: %1)" )
3603 : 0 : .arg( e.what() );
3604 : 0 : QgsLogger::warning( msg );
3605 : 0 : if ( errorMessage )
3606 : 0 : *errorMessage = msg;
3607 : :
3608 : 0 : return ErrProjection;
3609 : 0 : }
3610 : 0 : }
3611 : 0 : mRenderContext.expressionContext().setFeature( fet );
3612 : :
3613 : 0 : featureSymbol = mRenderer->symbolForFeature( fet, mRenderContext );
3614 : 0 : if ( !featureSymbol )
3615 : : {
3616 : 0 : continue;
3617 : : }
3618 : :
3619 : 0 : QHash< QgsSymbol *, QList<QgsFeature> >::iterator it = features.find( featureSymbol );
3620 : 0 : if ( it == features.end() )
3621 : : {
3622 : 0 : it = features.insert( featureSymbol, QList<QgsFeature>() );
3623 : 0 : }
3624 : 0 : it.value().append( fet );
3625 : : }
3626 : :
3627 : : //find out order
3628 : 0 : QgsSymbolLevelOrder levels;
3629 : 0 : QgsSymbolList symbols = mRenderer->symbols( mRenderContext );
3630 : 0 : for ( int i = 0; i < symbols.count(); i++ )
3631 : : {
3632 : 0 : QgsSymbol *sym = symbols[i];
3633 : 0 : for ( int j = 0; j < sym->symbolLayerCount(); j++ )
3634 : : {
3635 : 0 : int level = sym->symbolLayer( j )->renderingPass();
3636 : 0 : if ( level < 0 || level >= 1000 ) // ignore invalid levels
3637 : 0 : continue;
3638 : 0 : QgsSymbolLevelItem item( sym, j );
3639 : 0 : while ( level >= levels.count() ) // append new empty levels
3640 : 0 : levels.append( QgsSymbolLevel() );
3641 : 0 : levels[level].append( item );
3642 : 0 : }
3643 : 0 : }
3644 : :
3645 : 0 : int nErrors = 0;
3646 : 0 : int nTotalFeatures = 0;
3647 : :
3648 : : //export symbol layers and symbology
3649 : 0 : for ( int l = 0; l < levels.count(); l++ )
3650 : : {
3651 : 0 : QgsSymbolLevel &level = levels[l];
3652 : 0 : for ( int i = 0; i < level.count(); i++ )
3653 : : {
3654 : 0 : QgsSymbolLevelItem &item = level[i];
3655 : 0 : QHash< QgsSymbol *, QList<QgsFeature> >::iterator levelIt = features.find( item.symbol() );
3656 : 0 : if ( levelIt == features.end() )
3657 : : {
3658 : 0 : ++nErrors;
3659 : 0 : continue;
3660 : : }
3661 : :
3662 : 0 : double mmsf = mmScaleFactor( mSymbologyScale, levelIt.key()->outputUnit(), mapUnits );
3663 : 0 : double musf = mapUnitScaleFactor( mSymbologyScale, levelIt.key()->outputUnit(), mapUnits );
3664 : :
3665 : 0 : int llayer = item.layer();
3666 : 0 : QList<QgsFeature> &featureList = levelIt.value();
3667 : 0 : QList<QgsFeature>::iterator featureIt = featureList.begin();
3668 : 0 : for ( ; featureIt != featureList.end(); ++featureIt )
3669 : : {
3670 : 0 : ++nTotalFeatures;
3671 : 0 : gdal::ogr_feature_unique_ptr ogrFeature = createFeature( *featureIt );
3672 : 0 : if ( !ogrFeature )
3673 : : {
3674 : 0 : ++nErrors;
3675 : 0 : continue;
3676 : : }
3677 : :
3678 : 0 : QString styleString = levelIt.key()->symbolLayer( llayer )->ogrFeatureStyle( mmsf, musf );
3679 : 0 : if ( !styleString.isEmpty() )
3680 : : {
3681 : 0 : OGR_F_SetStyleString( ogrFeature.get(), styleString.toLocal8Bit().constData() );
3682 : 0 : if ( !writeFeature( mLayer, ogrFeature.get() ) )
3683 : : {
3684 : 0 : ++nErrors;
3685 : 0 : }
3686 : 0 : }
3687 : 0 : }
3688 : 0 : }
3689 : 0 : }
3690 : :
3691 : 0 : stopRender();
3692 : :
3693 : 0 : if ( nErrors > 0 && errorMessage )
3694 : : {
3695 : 0 : *errorMessage += QObject::tr( "\nOnly %1 of %2 features written." ).arg( nTotalFeatures - nErrors ).arg( nTotalFeatures );
3696 : 0 : }
3697 : :
3698 : 0 : return ( nErrors > 0 ) ? QgsVectorFileWriter::ErrFeatureWriteFailed : QgsVectorFileWriter::NoError;
3699 : 0 : }
3700 : :
3701 : 0 : double QgsVectorFileWriter::mmScaleFactor( double scale, QgsUnitTypes::RenderUnit symbolUnits, QgsUnitTypes::DistanceUnit mapUnits )
3702 : : {
3703 : 0 : if ( symbolUnits == QgsUnitTypes::RenderMillimeters )
3704 : : {
3705 : 0 : return 1.0;
3706 : : }
3707 : : else
3708 : : {
3709 : : //conversion factor map units -> mm
3710 : 0 : if ( mapUnits == QgsUnitTypes::DistanceMeters )
3711 : : {
3712 : 0 : return 1000 / scale;
3713 : : }
3714 : :
3715 : : }
3716 : 0 : return 1.0; //todo: map units
3717 : 0 : }
3718 : :
3719 : 0 : double QgsVectorFileWriter::mapUnitScaleFactor( double scale, QgsUnitTypes::RenderUnit symbolUnits, QgsUnitTypes::DistanceUnit mapUnits )
3720 : : {
3721 : 0 : if ( symbolUnits == QgsUnitTypes::RenderMapUnits )
3722 : : {
3723 : 0 : return 1.0;
3724 : : }
3725 : : else
3726 : : {
3727 : 0 : if ( symbolUnits == QgsUnitTypes::RenderMillimeters && mapUnits == QgsUnitTypes::DistanceMeters )
3728 : : {
3729 : 0 : return scale / 1000;
3730 : : }
3731 : : }
3732 : 0 : return 1.0;
3733 : 0 : }
3734 : :
3735 : 0 : void QgsVectorFileWriter::startRender( QgsFeatureRenderer *sourceRenderer, const QgsFields &fields )
3736 : : {
3737 : 0 : mRenderer = createSymbologyRenderer( sourceRenderer );
3738 : 0 : if ( !mRenderer )
3739 : : {
3740 : 0 : return;
3741 : : }
3742 : :
3743 : 0 : mRenderer->startRender( mRenderContext, fields );
3744 : 0 : }
3745 : :
3746 : 0 : void QgsVectorFileWriter::stopRender()
3747 : : {
3748 : 0 : if ( !mRenderer )
3749 : : {
3750 : 0 : return;
3751 : : }
3752 : :
3753 : 0 : mRenderer->stopRender( mRenderContext );
3754 : 0 : }
3755 : :
3756 : 0 : std::unique_ptr<QgsFeatureRenderer> QgsVectorFileWriter::createSymbologyRenderer( QgsFeatureRenderer *sourceRenderer ) const
3757 : : {
3758 : 0 : if ( mSymbologyExport == NoSymbology )
3759 : : {
3760 : 0 : return nullptr;
3761 : : }
3762 : 0 : if ( !sourceRenderer )
3763 : : {
3764 : 0 : return nullptr;
3765 : : }
3766 : :
3767 : 0 : return std::unique_ptr< QgsFeatureRenderer >( sourceRenderer->clone() );
3768 : 0 : }
3769 : :
3770 : 0 : void QgsVectorFileWriter::addRendererAttributes( QgsFeatureRenderer *renderer, QgsRenderContext &context, const QgsFields &fields, QgsAttributeList &attList )
3771 : : {
3772 : 0 : if ( renderer )
3773 : : {
3774 : 0 : const QSet<QString> rendererAttributes = renderer->usedAttributes( context );
3775 : 0 : for ( const QString &attr : rendererAttributes )
3776 : : {
3777 : 0 : int index = fields.lookupField( attr );
3778 : 0 : if ( index != -1 )
3779 : : {
3780 : 0 : attList.append( index );
3781 : 0 : }
3782 : : }
3783 : 0 : }
3784 : 0 : }
3785 : :
3786 : 0 : QStringList QgsVectorFileWriter::concatenateOptions( const QMap<QString, QgsVectorFileWriter::Option *> &options )
3787 : : {
3788 : 0 : QStringList list;
3789 : 0 : QMap<QString, QgsVectorFileWriter::Option *>::ConstIterator it;
3790 : :
3791 : 0 : for ( it = options.constBegin(); it != options.constEnd(); ++it )
3792 : : {
3793 : 0 : QgsVectorFileWriter::Option *option = it.value();
3794 : 0 : switch ( option->type )
3795 : : {
3796 : : case QgsVectorFileWriter::Int:
3797 : : {
3798 : 0 : QgsVectorFileWriter::IntOption *opt = dynamic_cast<QgsVectorFileWriter::IntOption *>( option );
3799 : 0 : if ( opt )
3800 : : {
3801 : 0 : list.append( QStringLiteral( "%1=%2" ).arg( it.key() ).arg( opt->defaultValue ) );
3802 : 0 : }
3803 : 0 : break;
3804 : : }
3805 : :
3806 : : case QgsVectorFileWriter::Set:
3807 : : {
3808 : 0 : QgsVectorFileWriter::SetOption *opt = dynamic_cast<QgsVectorFileWriter::SetOption *>( option );
3809 : 0 : if ( opt && !opt->defaultValue.isEmpty() )
3810 : : {
3811 : 0 : list.append( QStringLiteral( "%1=%2" ).arg( it.key(), opt->defaultValue ) );
3812 : 0 : }
3813 : 0 : break;
3814 : : }
3815 : :
3816 : : case QgsVectorFileWriter::String:
3817 : : {
3818 : 0 : QgsVectorFileWriter::StringOption *opt = dynamic_cast<QgsVectorFileWriter::StringOption *>( option );
3819 : 0 : if ( opt && !opt->defaultValue.isNull() )
3820 : : {
3821 : 0 : list.append( QStringLiteral( "%1=%2" ).arg( it.key(), opt->defaultValue ) );
3822 : 0 : }
3823 : 0 : break;
3824 : : }
3825 : :
3826 : : case QgsVectorFileWriter::Hidden:
3827 : 0 : QgsVectorFileWriter::HiddenOption *opt = dynamic_cast<QgsVectorFileWriter::HiddenOption *>( option );
3828 : 0 : if ( opt )
3829 : : {
3830 : 0 : list.append( QStringLiteral( "%1=%2" ).arg( it.key(), opt->mValue ) );
3831 : 0 : }
3832 : 0 : break;
3833 : : }
3834 : 0 : }
3835 : :
3836 : 0 : return list;
3837 : 0 : }
3838 : :
3839 : 0 : QgsVectorFileWriter::EditionCapabilities QgsVectorFileWriter::editionCapabilities( const QString &datasetName )
3840 : : {
3841 : 0 : OGRSFDriverH hDriver = nullptr;
3842 : 0 : gdal::ogr_datasource_unique_ptr hDS( myOGROpen( datasetName.toUtf8().constData(), TRUE, &hDriver ) );
3843 : 0 : if ( !hDS )
3844 : 0 : return QgsVectorFileWriter::EditionCapabilities();
3845 : 0 : QString drvName = OGR_Dr_GetName( hDriver );
3846 : 0 : QgsVectorFileWriter::EditionCapabilities caps = QgsVectorFileWriter::EditionCapabilities();
3847 : 0 : if ( OGR_DS_TestCapability( hDS.get(), ODsCCreateLayer ) )
3848 : : {
3849 : : // Shapefile driver returns True for a "foo.shp" dataset name,
3850 : : // creating "bar.shp" new layer, but this would be a bit confusing
3851 : : // for the user, so pretent that it does not support that
3852 : 0 : if ( !( drvName == QLatin1String( "ESRI Shapefile" ) && QFile::exists( datasetName ) ) )
3853 : 0 : caps |= CanAddNewLayer;
3854 : 0 : }
3855 : 0 : if ( OGR_DS_TestCapability( hDS.get(), ODsCDeleteLayer ) )
3856 : : {
3857 : 0 : caps |= CanDeleteLayer;
3858 : 0 : }
3859 : 0 : int layer_count = OGR_DS_GetLayerCount( hDS.get() );
3860 : 0 : if ( layer_count )
3861 : : {
3862 : 0 : OGRLayerH hLayer = OGR_DS_GetLayer( hDS.get(), 0 );
3863 : 0 : if ( hLayer )
3864 : : {
3865 : 0 : if ( OGR_L_TestCapability( hLayer, OLCSequentialWrite ) )
3866 : : {
3867 : 0 : caps |= CanAppendToExistingLayer;
3868 : 0 : if ( OGR_L_TestCapability( hLayer, OLCCreateField ) )
3869 : : {
3870 : 0 : caps |= CanAddNewFieldsToExistingLayer;
3871 : 0 : }
3872 : 0 : }
3873 : 0 : }
3874 : 0 : }
3875 : 0 : return caps;
3876 : 0 : }
3877 : :
3878 : 0 : bool QgsVectorFileWriter::targetLayerExists( const QString &datasetName,
3879 : : const QString &layerNameIn )
3880 : : {
3881 : 0 : OGRSFDriverH hDriver = nullptr;
3882 : 0 : gdal::ogr_datasource_unique_ptr hDS( myOGROpen( datasetName.toUtf8().constData(), TRUE, &hDriver ) );
3883 : 0 : if ( !hDS )
3884 : 0 : return false;
3885 : :
3886 : 0 : QString layerName( layerNameIn );
3887 : 0 : if ( layerName.isEmpty() )
3888 : 0 : layerName = QFileInfo( datasetName ).baseName();
3889 : :
3890 : 0 : return OGR_DS_GetLayerByName( hDS.get(), layerName.toUtf8().constData() );
3891 : 0 : }
3892 : :
3893 : :
3894 : 0 : bool QgsVectorFileWriter::areThereNewFieldsToCreate( const QString &datasetName,
3895 : : const QString &layerName,
3896 : : QgsVectorLayer *layer,
3897 : : const QgsAttributeList &attributes )
3898 : : {
3899 : 0 : OGRSFDriverH hDriver = nullptr;
3900 : 0 : gdal::ogr_datasource_unique_ptr hDS( myOGROpen( datasetName.toUtf8().constData(), TRUE, &hDriver ) );
3901 : 0 : if ( !hDS )
3902 : 0 : return false;
3903 : 0 : OGRLayerH hLayer = OGR_DS_GetLayerByName( hDS.get(), layerName.toUtf8().constData() );
3904 : 0 : if ( !hLayer )
3905 : : {
3906 : 0 : return false;
3907 : : }
3908 : 0 : bool ret = false;
3909 : 0 : OGRFeatureDefnH defn = OGR_L_GetLayerDefn( hLayer );
3910 : 0 : const auto constAttributes = attributes;
3911 : 0 : for ( int idx : constAttributes )
3912 : : {
3913 : 0 : QgsField fld = layer->fields().at( idx );
3914 : 0 : if ( OGR_FD_GetFieldIndex( defn, fld.name().toUtf8().constData() ) < 0 )
3915 : : {
3916 : 0 : ret = true;
3917 : 0 : break;
3918 : : }
3919 : 0 : }
3920 : 0 : return ret;
3921 : 0 : }
|
