Branch data Line data Source code
1 : : /***************************************************************************
2 : : qgsogrprovider.cpp Data provider for OGR supported formats
3 : : Formerly known as qgsshapefileprovider.cpp
4 : : begin : Oct 29, 2003
5 : : copyright : (C) 2003 by Gary E.Sherman
6 : : email : sherman at mrcc.com
7 : : ***************************************************************************/
8 : :
9 : : /***************************************************************************
10 : : * *
11 : : * This program is free software; you can redistribute it and/or modify *
12 : : * it under the terms of the GNU General Public License as published by *
13 : : * the Free Software Foundation; either version 2 of the License, or *
14 : : * (at your option) any later version. *
15 : : * *
16 : : ***************************************************************************/
17 : :
18 : : #include "qgsogrprovider.h"
19 : : ///@cond PRIVATE
20 : :
21 : : #include "qgscplerrorhandler.h"
22 : : #include "qgsgdalutils.h"
23 : : #include "qgsogrfeatureiterator.h"
24 : : #include "qgslogger.h"
25 : : #include "qgsmessagelog.h"
26 : : #include "qgslocalec.h"
27 : : #include "qgsfeedback.h"
28 : : #include "qgssettings.h"
29 : : #include "qgsapplication.h"
30 : : #include "qgsauthmanager.h"
31 : : #include "qgscplhttpfetchoverrider.h"
32 : : #include "qgsdataitem.h"
33 : : #include "qgsdataprovider.h"
34 : : #include "qgsfeature.h"
35 : : #include "qgsfields.h"
36 : : #include "qgsgeometry.h"
37 : : #include "qgscoordinatereferencesystem.h"
38 : : #include "qgsvectorlayerexporter.h"
39 : : #include "qgsdataitemprovider.h"
40 : : #include "qgsogrdataitems.h"
41 : : #include "qgsgeopackagedataitems.h"
42 : : #include "qgswkbtypes.h"
43 : : #include "qgsnetworkaccessmanager.h"
44 : : #include "qgsogrtransaction.h"
45 : : #include "qgsgeopackageprojectstorage.h"
46 : : #include "qgsprojectstorageregistry.h"
47 : : #include "qgsprovidermetadata.h"
48 : : #include "qgsogrdbconnection.h"
49 : : #include "qgsgeopackageproviderconnection.h"
50 : : #include "qgis.h"
51 : : #include "qgsembeddedsymbolrenderer.h"
52 : :
53 : : #define CPL_SUPRESS_CPLUSPLUS //#spellok
54 : : #include <gdal.h> // to collect version information
55 : : #include <ogr_api.h>
56 : : #include <ogr_srs_api.h>
57 : : #include <cpl_string.h>
58 : :
59 : : #if GDAL_VERSION_NUM < GDAL_COMPUTE_VERSION(3, 2, 1)
60 : : // Temporary solution for gdal < 3.2.1 without GDAL Unique support
61 : : #include "qgssqliteutils.h"
62 : : #include <sqlite3.h>
63 : : // end temporary
64 : : #endif
65 : :
66 : : #include <limits>
67 : : #include <memory>
68 : :
69 : : #include <QtDebug>
70 : : #include <QFile>
71 : : #include <QDir>
72 : : #include <QFileInfo>
73 : : #include <QMap>
74 : : #include <QMessageBox>
75 : : #include <QString>
76 : : #include <QTextCodec>
77 : : #include <QStorageInfo>
78 : : #include <QRegularExpression>
79 : :
80 : :
81 : : #ifdef Q_OS_WIN
82 : : #include <windows.h>
83 : : #endif
84 : : #ifdef Q_OS_LINUX
85 : : #include <sys/vfs.h>
86 : : #endif
87 : :
88 : : // Starting with GDAL 2.2, there are 2 concepts: unset fields and null fields
89 : : // whereas previously there was only unset fields. For QGIS purposes, both
90 : : // states (unset/null) are equivalent.
91 : : #ifndef OGRNullMarker
92 : : #define OGR_F_IsFieldSetAndNotNull OGR_F_IsFieldSet
93 : : #endif
94 : :
95 : : #define TEXT_PROVIDER_KEY QStringLiteral( "ogr" )
96 : : #define TEXT_PROVIDER_DESCRIPTION QStringLiteral( "OGR data provider" )
97 : :
98 : : static OGRwkbGeometryType ogrWkbGeometryTypeFromName( const QString &typeName );
99 : :
100 : : static bool IsLocalFile( const QString &path );
101 : :
102 : 505 : Q_GLOBAL_STATIC_WITH_ARGS( QMutex, sGlobalMutex, ( QMutex::Recursive ) )
103 : :
104 : : //! Map a dataset name to the number of opened GDAL dataset objects on it (if opened with GDALOpenWrapper, only for GPKG)
105 : : typedef QMap< QString, int > OpenedDsCountMap;
106 : 10 : Q_GLOBAL_STATIC( OpenedDsCountMap, sMapCountOpenedDS )
107 : :
108 : : QMap< QgsOgrProviderUtils::DatasetIdentification,
109 : 5 : QList<QgsOgrProviderUtils::DatasetWithLayers *> > QgsOgrProviderUtils::sMapSharedDS;
110 : :
111 : : typedef QHash< GDALDatasetH, bool> DsHandleToUpdateModeHash;
112 : 8 : Q_GLOBAL_STATIC( DsHandleToUpdateModeHash, sMapDSHandleToUpdateMode )
113 : :
114 : : typedef QMap< QString, QDateTime > DsNameToLastModifiedDateMap;
115 : 0 : Q_GLOBAL_STATIC( DsNameToLastModifiedDateMap, sMapDSNameToLastModifiedDate )
116 : :
117 : 0 : bool QgsOgrProvider::convertField( QgsField &field, const QTextCodec &encoding )
118 : : {
119 : 0 : OGRFieldType ogrType = OFTString; //default to string
120 : 0 : OGRFieldSubType ogrSubType = OFSTNone;
121 : 0 : int ogrWidth = field.length();
122 : 0 : int ogrPrecision = field.precision();
123 : 0 : if ( ogrPrecision > 0 )
124 : 0 : ogrWidth += 1;
125 : 0 : switch ( field.type() )
126 : : {
127 : : case QVariant::LongLong:
128 : 0 : ogrType = OFTInteger64;
129 : 0 : ogrPrecision = 0;
130 : 0 : ogrWidth = ogrWidth > 0 && ogrWidth <= 21 ? ogrWidth : 21;
131 : 0 : break;
132 : :
133 : : case QVariant::String:
134 : 0 : ogrType = OFTString;
135 : 0 : if ( ogrWidth < 0 || ogrWidth > 255 )
136 : 0 : ogrWidth = 255;
137 : 0 : break;
138 : :
139 : : case QVariant::Int:
140 : 0 : ogrType = OFTInteger;
141 : 0 : ogrWidth = ogrWidth > 0 && ogrWidth <= 10 ? ogrWidth : 10;
142 : 0 : ogrPrecision = 0;
143 : 0 : break;
144 : :
145 : : case QVariant::Bool:
146 : 0 : ogrType = OFTInteger;
147 : 0 : ogrSubType = OFSTBoolean;
148 : 0 : ogrWidth = 1;
149 : 0 : ogrPrecision = 0;
150 : 0 : break;
151 : :
152 : : case QVariant::Double:
153 : 0 : ogrType = OFTReal;
154 : 0 : break;
155 : :
156 : : case QVariant::Date:
157 : 0 : ogrType = OFTDate;
158 : 0 : break;
159 : :
160 : : case QVariant::Time:
161 : 0 : ogrType = OFTTime;
162 : 0 : break;
163 : :
164 : : case QVariant::DateTime:
165 : 0 : ogrType = OFTDateTime;
166 : 0 : break;
167 : :
168 : : case QVariant::List:
169 : 0 : if ( field.subType() == QVariant::String )
170 : : {
171 : 0 : ogrType = OFTStringList;
172 : 0 : }
173 : : else
174 : : {
175 : : // only string lists are supported at this moment
176 : 0 : return false;
177 : : }
178 : 0 : break;
179 : :
180 : : case QVariant::Map:
181 : 0 : ogrType = OFTString;
182 : 0 : ogrSubType = OFSTJSON;
183 : 0 : break;
184 : :
185 : : default:
186 : 0 : return false;
187 : : }
188 : :
189 : 0 : if ( ogrSubType != OFSTNone )
190 : 0 : field.setTypeName( encoding.toUnicode( OGR_GetFieldSubTypeName( ogrSubType ) ) );
191 : : else
192 : 0 : field.setTypeName( encoding.toUnicode( OGR_GetFieldTypeName( ogrType ) ) );
193 : :
194 : 0 : field.setLength( ogrWidth );
195 : 0 : field.setPrecision( ogrPrecision );
196 : 0 : return true;
197 : 0 : }
198 : :
199 : 60 : void QgsOgrProvider::repack()
200 : : {
201 : 60 : if ( !mValid || mGDALDriverName != QLatin1String( "ESRI Shapefile" ) || !mOgrOrigLayer )
202 : 0 : return;
203 : :
204 : : // run REPACK on shape files
205 : 60 : QByteArray sql = QByteArray( "REPACK " ) + mOgrOrigLayer->name(); // don't quote the layer name as it works with spaces in the name and won't work if the name is quoted
206 : 60 : QgsDebugMsgLevel( QStringLiteral( "SQL: %1" ).arg( QString::fromUtf8( sql ) ), 2 );
207 : 60 : CPLErrorReset();
208 : 60 : mOgrOrigLayer->ExecuteSQLNoReturn( sql );
209 : 60 : if ( CPLGetLastErrorType() != CE_None )
210 : : {
211 : 0 : pushError( tr( "OGR[%1] error %2: %3" ).arg( CPLGetLastErrorType() ).arg( CPLGetLastErrorNo() ).arg( CPLGetLastErrorMsg() ) );
212 : 0 : }
213 : :
214 : 60 : if ( mFilePath.endsWith( QLatin1String( ".shp" ), Qt::CaseInsensitive ) || mFilePath.endsWith( QLatin1String( ".dbf" ), Qt::CaseInsensitive ) )
215 : : {
216 : 60 : QString packedDbf( mFilePath.left( mFilePath.size() - 4 ) + "_packed.dbf" );
217 : 60 : if ( QFile::exists( packedDbf ) )
218 : : {
219 : 0 : QgsMessageLog::logMessage( tr( "Possible corruption after REPACK detected. %1 still exists. This may point to a permission or locking problem of the original DBF." ).arg( packedDbf ), tr( "OGR" ), Qgis::Critical );
220 : :
221 : 0 : mOgrSqlLayer.reset();
222 : 0 : mOgrOrigLayer.reset();
223 : :
224 : 0 : QString errCause;
225 : 0 : if ( mLayerName.isNull() )
226 : : {
227 : 0 : mOgrOrigLayer = QgsOgrProviderUtils::getLayer( mFilePath, true, mOpenOptions, mLayerIndex, errCause, true );
228 : 0 : }
229 : : else
230 : : {
231 : 0 : mOgrOrigLayer = QgsOgrProviderUtils::getLayer( mFilePath, true, mOpenOptions, mLayerName, errCause, true );
232 : : }
233 : :
234 : 0 : if ( !mOgrOrigLayer )
235 : : {
236 : 0 : QgsMessageLog::logMessage( tr( "Original layer could not be reopened." ) + " " + errCause, tr( "OGR" ), Qgis::Critical );
237 : 0 : mValid = false;
238 : 0 : }
239 : 64 :
240 : 0 : mOgrLayer = mOgrOrigLayer.get();
241 : 64 : }
242 : :
243 : 60 : }
244 : :
245 : 60 : if ( mFeaturesCounted != QgsVectorDataProvider::Uncounted &&
246 : 7 : mFeaturesCounted != QgsVectorDataProvider::UnknownCount )
247 : : {
248 : 7 : long oldcount = mFeaturesCounted;
249 : 7 : recalculateFeatureCount();
250 : 7 : if ( oldcount != mFeaturesCounted )
251 : 5 : emit dataChanged();
252 : 7 : }
253 : 60 : }
254 : :
255 : 64 :
256 : 0 : QgsVectorLayerExporter::ExportError QgsOgrProviderMetadata::createEmptyLayer( const QString &uri,
257 : : const QgsFields &fields,
258 : : QgsWkbTypes::Type wkbType,
259 : : const QgsCoordinateReferenceSystem &srs,
260 : : bool overwrite,
261 : : QMap<int, int> &oldToNewAttrIdxMap,
262 : : QString &errorMessage,
263 : : const QMap<QString, QVariant> *options )
264 : : {
265 : 0 : return QgsOgrProvider::createEmptyLayer(
266 : 0 : uri, fields, wkbType, srs, overwrite,
267 : 0 : &oldToNewAttrIdxMap, &errorMessage, options
268 : : );
269 : : }
270 : :
271 : 65 : static QString AnalyzeURI( QString const &uri,
272 : : bool &isSubLayer,
273 : : int &layerIndex,
274 : : QString &layerName,
275 : 64 : QString &subsetString,
276 : : OGRwkbGeometryType &ogrGeometryTypeFilter,
277 : : QStringList &openOptions )
278 : : {
279 : 65 : isSubLayer = false;
280 : 65 : layerIndex = 0;
281 : 129 : layerName = QString();
282 : 65 : subsetString = QString();
283 : 65 : ogrGeometryTypeFilter = wkbUnknown;
284 : 65 : openOptions.clear();
285 : :
286 : 65 : QgsDebugMsgLevel( "Data source uri is [" + uri + ']', 2 );
287 : :
288 : 129 : QVariantMap parts = QgsOgrProviderMetadata().decodeUri( uri );
289 : :
290 : 130 : if ( parts.contains( QStringLiteral( "layerName" ) ) )
291 : : {
292 : 130 : layerName = parts.value( QStringLiteral( "layerName" ) ).toString();
293 : 65 : isSubLayer = !layerName.isEmpty();
294 : 65 : }
295 : :
296 : 195 : if ( parts.contains( QStringLiteral( "layerId" ) ) &&
297 : 130 : parts.value( QStringLiteral( "layerId" ) ) != QVariant() )
298 : : {
299 : : bool ok;
300 : 64 : layerIndex = parts.value( QStringLiteral( "layerId" ) ).toInt( &ok );
301 : 0 : if ( ok && layerIndex >= 0 )
302 : 64 : isSubLayer = true;
303 : : else
304 : 64 : layerIndex = -1;
305 : 0 : }
306 : :
307 : 194 : if ( parts.contains( QStringLiteral( "subset" ) ) )
308 : : {
309 : 64 : subsetString = parts.value( QStringLiteral( "subset" ) ).toString();
310 : 0 : }
311 : :
312 : 130 : if ( parts.contains( QStringLiteral( "geometryType" ) ) )
313 : : {
314 : 0 : ogrGeometryTypeFilter = ogrWkbGeometryTypeFromName( parts.value( QStringLiteral( "geometryType" ) ).toString() );
315 : 0 : }
316 : :
317 : 130 : if ( parts.contains( QStringLiteral( "openOptions" ) ) )
318 : : {
319 : 0 : openOptions = parts.value( QStringLiteral( "openOptions" ) ).toStringList();
320 : 0 : }
321 : :
322 : 195 : const QString fullPath = parts.value( QStringLiteral( "vsiPrefix" ) ).toString()
323 : 130 : + parts.value( QStringLiteral( "path" ) ).toString()
324 : 130 : + parts.value( QStringLiteral( "vsiSuffix" ) ).toString();
325 : 65 : return fullPath;
326 : 65 : }
327 : :
328 : 64 : QgsVectorLayerExporter::ExportError QgsOgrProvider::createEmptyLayer( const QString &uri,
329 : : const QgsFields &fields,
330 : : QgsWkbTypes::Type wkbType,
331 : 64 : const QgsCoordinateReferenceSystem &srs,
332 : : bool overwrite,
333 : : QMap<int, int> *oldToNewAttrIdxMap,
334 : 64 : QString *errorMessage,
335 : : const QMap<QString, QVariant> *options )
336 : 64 : {
337 : 0 : QString encoding;
338 : 0 : QString driverName = QStringLiteral( "GPKG" );
339 : 0 : QStringList dsOptions, layerOptions;
340 : 0 : QString layerName;
341 : 64 :
342 : 0 : if ( options )
343 : 64 : {
344 : 0 : if ( options->contains( QStringLiteral( "fileEncoding" ) ) )
345 : 0 : encoding = options->value( QStringLiteral( "fileEncoding" ) ).toString();
346 : :
347 : 64 : if ( options->contains( QStringLiteral( "driverName" ) ) )
348 : 0 : driverName = options->value( QStringLiteral( "driverName" ) ).toString();
349 : :
350 : 0 : if ( options->contains( QStringLiteral( "datasourceOptions" ) ) )
351 : 0 : dsOptions << options->value( QStringLiteral( "datasourceOptions" ) ).toStringList();
352 : :
353 : 64 : if ( options->contains( QStringLiteral( "layerOptions" ) ) )
354 : 0 : layerOptions << options->value( QStringLiteral( "layerOptions" ) ).toStringList();
355 : :
356 : 0 : if ( options->contains( QStringLiteral( "layerName" ) ) )
357 : 0 : layerName = options->value( QStringLiteral( "layerName" ) ).toString();
358 : 0 : }
359 : :
360 : 0 : oldToNewAttrIdxMap->clear();
361 : 0 : if ( errorMessage )
362 : 0 : errorMessage->clear();
363 : :
364 : 0 : QgsVectorFileWriter::ActionOnExistingFile action( QgsVectorFileWriter::CreateOrOverwriteFile );
365 : :
366 : 0 : bool update = false;
367 : 0 : if ( options && options->contains( QStringLiteral( "update" ) ) )
368 : : {
369 : 0 : update = options->value( QStringLiteral( "update" ) ).toBool();
370 : 0 : if ( update )
371 : : {
372 : 0 : if ( !overwrite && !layerName.isEmpty() )
373 : : {
374 : 0 : gdal::dataset_unique_ptr hDS( GDALOpenEx( uri.toUtf8().constData(), GDAL_OF_VECTOR, nullptr, nullptr, nullptr ) );
375 : 0 : if ( hDS )
376 : : {
377 : 0 : if ( GDALDatasetGetLayerByName( hDS.get(), layerName.toUtf8().constData() ) )
378 : : {
379 : 0 : if ( errorMessage )
380 : 0 : *errorMessage += QObject::tr( "Layer %2 of %1 exists and overwrite flag is false." )
381 : 0 : .arg( uri, layerName );
382 : 0 : return QgsVectorLayerExporter::ErrCreateDataSource;
383 : : }
384 : 0 : }
385 : 0 : }
386 : 0 : if ( QFileInfo::exists( uri ) )
387 : 0 : action = QgsVectorFileWriter::CreateOrOverwriteLayer;
388 : 0 : }
389 : 0 : }
390 : :
391 : 0 : if ( !overwrite && !update )
392 : : {
393 : 0 : if ( QFileInfo::exists( uri ) )
394 : : {
395 : 0 : if ( errorMessage )
396 : 0 : *errorMessage += QObject::tr( "Unable to create the datasource. %1 exists and overwrite flag is false." )
397 : 0 : .arg( uri );
398 : 0 : return QgsVectorLayerExporter::ErrCreateDataSource;
399 : : }
400 : 0 : }
401 : :
402 : 0 : QString newLayerName( layerName );
403 : :
404 : 0 : QgsVectorFileWriter::SaveVectorOptions saveOptions;
405 : 0 : saveOptions.layerName = layerName;
406 : 0 : saveOptions.fileEncoding = encoding;
407 : 0 : saveOptions.driverName = driverName;
408 : 0 : saveOptions.datasourceOptions = dsOptions;
409 : 0 : saveOptions.layerOptions = layerOptions;
410 : 0 : saveOptions.actionOnExistingFile = action;
411 : 0 : saveOptions.symbologyExport = QgsVectorFileWriter::NoSymbology;
412 : 0 : std::unique_ptr< QgsVectorFileWriter > writer( QgsVectorFileWriter::create( uri, fields, wkbType, srs, QgsCoordinateTransformContext(), saveOptions, QgsFeatureSink::SinkFlags(), nullptr, &newLayerName ) );
413 : 0 : layerName = newLayerName;
414 : :
415 : 0 : QgsVectorFileWriter::WriterError error = writer->hasError();
416 : 0 : if ( error )
417 : : {
418 : 0 : if ( errorMessage )
419 : 0 : *errorMessage += writer->errorMessage();
420 : :
421 : 0 : return static_cast<QgsVectorLayerExporter::ExportError>( error );
422 : : }
423 : :
424 : 0 : QMap<int, int> attrIdxMap = writer->attrIdxToOgrIdx();
425 : 0 : writer.reset();
426 : :
427 : : {
428 : 0 : bool firstFieldIsFid = false;
429 : 0 : bool fidColumnIsField = false;
430 : 0 : if ( !layerName.isEmpty() )
431 : : {
432 : 0 : gdal::dataset_unique_ptr hDS( GDALOpenEx( uri.toUtf8().constData(), GDAL_OF_VECTOR, nullptr, nullptr, nullptr ) );
433 : 0 : if ( hDS )
434 : : {
435 : 0 : OGRLayerH hLayer = GDALDatasetGetLayerByName( hDS.get(), layerName.toUtf8().constData() );
436 : 0 : if ( hLayer )
437 : : {
438 : : // Expose the OGR FID if it comes from a "real" column (typically GPKG)
439 : : // and make sure that this FID column is not exposed as a regular OGR field (shouldn't happen normally)
440 : 0 : const QString ogrFidColumnName { OGR_L_GetFIDColumn( hLayer ) };
441 : 0 : firstFieldIsFid = !( EQUAL( OGR_L_GetFIDColumn( hLayer ), "" ) ) &&
442 : 0 : OGR_FD_GetFieldIndex( OGR_L_GetLayerDefn( hLayer ), ogrFidColumnName.toUtf8() ) < 0 &&
443 : 0 : fields.indexFromName( ogrFidColumnName.toUtf8() ) < 0;
444 : : // At this point we must check if there is a real FID field in the the fields argument,
445 : : // because in that case we don't want to shift all fields (see issue GH #34333)
446 : : // Check for unique values should be performed in client code.
447 : 0 : for ( const auto &f : std::as_const( fields ) )
448 : : {
449 : 0 : if ( f.name().compare( ogrFidColumnName, Qt::CaseSensitivity::CaseInsensitive ) == 0 )
450 : : {
451 : 0 : fidColumnIsField = true;
452 : 0 : break;
453 : : }
454 : : }
455 : 0 : }
456 : 0 : }
457 : 0 : }
458 : :
459 : 0 : const bool shiftColumnsByOne { firstFieldIsFid &&( ! fidColumnIsField ) };
460 : :
461 : 0 : for ( QMap<int, int>::const_iterator attrIt = attrIdxMap.constBegin(); attrIt != attrIdxMap.constEnd(); ++attrIt )
462 : : {
463 : 0 : oldToNewAttrIdxMap->insert( attrIt.key(), *attrIt + ( shiftColumnsByOne ? 1 : 0 ) );
464 : 0 : }
465 : : }
466 : :
467 : 0 : QgsOgrProviderUtils::invalidateCachedLastModifiedDate( uri );
468 : :
469 : 0 : return QgsVectorLayerExporter::NoError;
470 : 0 : }
471 : :
472 : 64 : QgsOgrProvider::QgsOgrProvider( QString const &uri, const ProviderOptions &options, QgsDataProvider::ReadFlags flags )
473 : 64 : : QgsVectorDataProvider( uri, options, flags )
474 : 128 : {
475 : 64 : QgsApplication::registerOgrDrivers();
476 : :
477 : 64 : QgsSettings settings;
478 : : // we always disable GDAL side shapefile encoding handling, and do it on the QGIS side.
479 : : // why? it's not the ideal choice, but...
480 : : // - if we DON'T disable GDAL side encoding support, then there's NO way to change the encoding used when reading
481 : : // shapefiles. And unfortunately the embedded encoding (which is read by GDAL) is sometimes wrong, so we need
482 : : // to expose support for users to be able to change and correct this
483 : : // - we can't change this setting on-the-fly. If we don't set it upfront, we can't reverse this decision later when
484 : : // a user does want/need to manually specify the encoding
485 : 64 : CPLSetConfigOption( "SHAPE_ENCODING", "" );
486 : :
487 : : #ifndef QT_NO_NETWORKPROXY
488 : 64 : QgsGdalUtils::setupProxy();
489 : : #endif
490 : :
491 : : // make connection to the data source
492 : :
493 : 64 : QgsDebugMsgLevel( "Data source uri is [" + uri + ']', 2 );
494 : :
495 : 128 : mFilePath = AnalyzeURI( uri,
496 : 64 : mIsSubLayer,
497 : 64 : mLayerIndex,
498 : 64 : mLayerName,
499 : 64 : mSubsetString,
500 : 64 : mOgrGeometryTypeFilter,
501 : 64 : mOpenOptions );
502 : :
503 : 64 : if ( mFilePath.contains( QLatin1String( "authcfg" ) ) )
504 : : {
505 : 0 : QRegularExpression authcfgRe( " authcfg='([^']+)'" );
506 : 0 : QRegularExpressionMatch match;
507 : 0 : if ( mFilePath.contains( authcfgRe, &match ) )
508 : : {
509 : 0 : mAuthCfg = match.captured( 1 );
510 : 0 : }
511 : 0 : }
512 : 64 : QgsCPLHTTPFetchOverrider oCPLHTTPFetcher( mAuthCfg );
513 : 128 : QgsSetCPLHTTPFetchOverriderInitiatorClass( oCPLHTTPFetcher, QStringLiteral( "QgsOgrProvider" ) );
514 : :
515 : 64 : open( OpenModeInitial );
516 : :
517 : 64 : int nMaxIntLen = 11;
518 : 64 : int nMaxInt64Len = 21;
519 : 64 : int nMaxDoubleLen = 20;
520 : 64 : int nMaxDoublePrec = 15;
521 : 64 : int nDateLen = 8;
522 : 64 : if ( mGDALDriverName == QLatin1String( "GPKG" ) )
523 : : {
524 : : // GPKG only supports field length for text (and binary)
525 : 1 : nMaxIntLen = 0;
526 : 1 : nMaxInt64Len = 0;
527 : 1 : nMaxDoubleLen = 0;
528 : 1 : nMaxDoublePrec = 0;
529 : 1 : nDateLen = 0;
530 : 1 : }
531 : :
532 : 64 : QList<NativeType> nativeTypes;
533 : 64 : nativeTypes
534 : 128 : << QgsVectorDataProvider::NativeType( tr( "Whole number (integer)" ), QStringLiteral( "integer" ), QVariant::Int, 0, nMaxIntLen )
535 : 128 : << QgsVectorDataProvider::NativeType( tr( "Whole number (integer 64 bit)" ), QStringLiteral( "integer64" ), QVariant::LongLong, 0, nMaxInt64Len )
536 : 128 : << QgsVectorDataProvider::NativeType( tr( "Decimal number (real)" ), QStringLiteral( "double" ), QVariant::Double, 0, nMaxDoubleLen, 0, nMaxDoublePrec )
537 : 128 : << QgsVectorDataProvider::NativeType( tr( "Text (string)" ), QStringLiteral( "string" ), QVariant::String, 0, 65535 );
538 : :
539 : 64 : if ( mGDALDriverName == QLatin1String( "GPKG" ) )
540 : 2 : nativeTypes << QgsVectorDataProvider::NativeType( tr( "JSON (string)" ), QStringLiteral( "JSON" ), QVariant::Map, 0, 0, 0, 0, QVariant::String );
541 : :
542 : 64 : bool supportsDate = true;
543 : 64 : bool supportsTime = mGDALDriverName != QLatin1String( "ESRI Shapefile" ) && mGDALDriverName != QLatin1String( "GPKG" );
544 : 64 : bool supportsDateTime = mGDALDriverName != QLatin1String( "ESRI Shapefile" );
545 : 64 : bool supportsBinary = false;
546 : 64 : bool supportsStringList = false;
547 : 64 : const char *pszDataTypes = nullptr;
548 : 64 : if ( mOgrOrigLayer )
549 : : {
550 : 64 : pszDataTypes = GDALGetMetadataItem( mOgrOrigLayer->driver(), GDAL_DMD_CREATIONFIELDDATATYPES, nullptr );
551 : 64 : }
552 : : // For drivers that advertise their data type, use that instead of the
553 : : // above hardcoded defaults.
554 : 64 : if ( pszDataTypes )
555 : : {
556 : 64 : char **papszTokens = CSLTokenizeString2( pszDataTypes, " ", 0 );
557 : 64 : supportsDate = CSLFindString( papszTokens, "Date" ) >= 0;
558 : 64 : supportsTime = CSLFindString( papszTokens, "Time" ) >= 0;
559 : 64 : supportsDateTime = CSLFindString( papszTokens, "DateTime" ) >= 0;
560 : 64 : supportsBinary = CSLFindString( papszTokens, "Binary" ) >= 0;
561 : 64 : supportsStringList = CSLFindString( papszTokens, "StringList" ) >= 0;
562 : 64 : CSLDestroy( papszTokens );
563 : 64 : }
564 : :
565 : : // Older versions of GDAL incorrectly report that shapefiles support
566 : : // DateTime.
567 : : #if GDAL_VERSION_NUM < GDAL_COMPUTE_VERSION(3,2,0)
568 : : if ( mGDALDriverName == QLatin1String( "ESRI Shapefile" ) )
569 : : {
570 : : supportsDateTime = false;
571 : : }
572 : : #endif
573 : :
574 : 64 : if ( supportsDate )
575 : : {
576 : 64 : nativeTypes
577 : 128 : << QgsVectorDataProvider::NativeType( tr( "Date" ), QStringLiteral( "date" ), QVariant::Date, nDateLen, nDateLen );
578 : 64 : }
579 : 64 : if ( supportsTime )
580 : : {
581 : 0 : nativeTypes
582 : 0 : << QgsVectorDataProvider::NativeType( tr( "Time" ), QStringLiteral( "time" ), QVariant::Time );
583 : 0 : }
584 : 64 : if ( supportsDateTime )
585 : : {
586 : 1 : nativeTypes
587 : 2 : << QgsVectorDataProvider::NativeType( tr( "Date & Time" ), QStringLiteral( "datetime" ), QVariant::DateTime );
588 : 1 : }
589 : 64 : if ( supportsBinary )
590 : : {
591 : 1 : nativeTypes
592 : 2 : << QgsVectorDataProvider::NativeType( tr( "Binary object (BLOB)" ), QStringLiteral( "binary" ), QVariant::ByteArray );
593 : 1 : }
594 : 64 : if ( supportsStringList )
595 : : {
596 : 0 : nativeTypes
597 : 0 : << QgsVectorDataProvider::NativeType( tr( "String List" ), QStringLiteral( "stringlist" ), QVariant::List, 0, 0, 0, 0, QVariant::String );
598 : 0 : }
599 : :
600 : 64 : bool supportsBoolean = false;
601 : :
602 : : // layer metadata
603 : 128 : mLayerMetadata.setType( QStringLiteral( "dataset" ) );
604 : 64 : if ( mOgrOrigLayer )
605 : : {
606 : 64 : QMutex *mutex = nullptr;
607 : 64 : OGRLayerH layer = mOgrOrigLayer->getHandleAndMutex( mutex );
608 : 64 : QMutexLocker locker( mutex );
609 : 64 : const QString identifier = GDALGetMetadataItem( layer, "IDENTIFIER", nullptr );
610 : 64 : if ( !identifier.isEmpty() )
611 : 0 : mLayerMetadata.setTitle( identifier ); // see geopackage specs -- "'identifier' is analogous to 'title'"
612 : 64 : const QString abstract = GDALGetMetadataItem( layer, "DESCRIPTION", nullptr );
613 : 64 : if ( !abstract.isEmpty() )
614 : 0 : mLayerMetadata.setAbstract( abstract );
615 : 64 : }
616 : :
617 : 64 : if ( mOgrOrigLayer )
618 : : {
619 : 64 : const char *pszDataSubTypes = GDALGetMetadataItem( mOgrOrigLayer->driver(), GDAL_DMD_CREATIONFIELDDATASUBTYPES, nullptr );
620 : 64 : if ( pszDataSubTypes && strstr( pszDataSubTypes, "Boolean" ) )
621 : 1 : supportsBoolean = true;
622 : 64 : }
623 : :
624 : 64 : if ( supportsBoolean )
625 : 250 : {
626 : : // boolean data type
627 : 1 : nativeTypes
628 : 252 : << QgsVectorDataProvider::NativeType( tr( "Boolean" ), QStringLiteral( "bool" ), QVariant::Bool );
629 : 251 : }
630 : :
631 : 64 : setNativeTypes( nativeTypes );
632 : :
633 : 64 : QgsOgrConnPool::instance()->ref( QgsOgrProviderUtils::connectionPoolId( dataSourceUri( true ), mShareSameDatasetAmongLayers ) );
634 : 64 : }
635 : :
636 : 122 : QgsOgrProvider::~QgsOgrProvider()
637 : 122 : {
638 : 61 : QgsOgrConnPool::instance()->unref( QgsOgrProviderUtils::connectionPoolId( dataSourceUri( true ), mShareSameDatasetAmongLayers ) );
639 : : // We must also make sure to flush unusef cached connections so that
640 : : // the file can be removed (#15137)
641 : 61 : QgsOgrConnPool::instance()->invalidateConnections( QgsOgrProviderUtils::connectionPoolId( dataSourceUri( true ), mShareSameDatasetAmongLayers ) );
642 : :
643 : : // Do that as last step for final cleanup that might be prevented by
644 : : // still opened datasets.
645 : 61 : close();
646 : 122 : }
647 : :
648 : 1074 : QString QgsOgrProvider::dataSourceUri( bool expandAuthConfig ) const
649 : : {
650 : 1717 : if ( expandAuthConfig && QgsDataProvider::dataSourceUri( ).contains( QLatin1String( "authcfg" ) ) )
651 : : {
652 : 0 : return QgsOgrProviderUtils::expandAuthConfig( QgsDataProvider::dataSourceUri( ) );
653 : : }
654 : : else
655 : : {
656 : 1074 : return QgsDataProvider::dataSourceUri( );
657 : : }
658 : 1074 : }
659 : :
660 : 0 : QgsTransaction *QgsOgrProvider::transaction() const
661 : : {
662 : 0 : return static_cast<QgsTransaction *>( mTransaction );
663 : : }
664 : :
665 : 0 : void QgsOgrProvider::setTransaction( QgsTransaction *transaction )
666 : : {
667 : 0 : QgsDebugMsgLevel( QStringLiteral( "set transaction %1" ).arg( transaction != nullptr ), 1 );
668 : : // static_cast since layers cannot be added to a transaction of a non-matching provider
669 : 0 : mTransaction = static_cast<QgsOgrTransaction *>( transaction );
670 : 0 : }
671 : :
672 : 133 : QgsAbstractFeatureSource *QgsOgrProvider::featureSource() const
673 : : {
674 : 133 : return new QgsOgrFeatureSource( this );
675 : 0 : }
676 : :
677 : 0 : bool QgsOgrProvider::setSubsetString( const QString &theSQL, bool updateFeatureCount )
678 : : {
679 : 0 : return _setSubsetString( theSQL, updateFeatureCount, true );
680 : : }
681 : :
682 : 0 : QString QgsOgrProvider::subsetString() const
683 : : {
684 : 0 : return mSubsetString;
685 : : }
686 : :
687 : 0 : QString QgsOgrProvider::ogrWkbGeometryTypeName( OGRwkbGeometryType type ) const
688 : : {
689 : 0 : QString geom;
690 : :
691 : : // GDAL 2.1 can return M/ZM geometries
692 : 0 : if ( wkbHasM( type ) )
693 : : {
694 : 0 : geom = ogrWkbGeometryTypeName( wkbFlatten( type ) );
695 : 0 : if ( wkbHasZ( type ) )
696 : 0 : geom += QLatin1Char( 'Z' );
697 : 0 : if ( wkbHasM( type ) )
698 : 0 : geom += QLatin1Char( 'M' );
699 : 0 : return geom;
700 : : }
701 : :
702 : 0 : switch ( static_cast<unsigned>( type ) )
703 : : {
704 : : case wkbUnknown:
705 : 0 : geom = QStringLiteral( "Unknown" );
706 : 0 : break;
707 : : case wkbPoint:
708 : 0 : geom = QStringLiteral( "Point" );
709 : 0 : break;
710 : : case wkbLineString:
711 : 0 : geom = QStringLiteral( "LineString" );
712 : 0 : break;
713 : : case wkbPolygon:
714 : 0 : geom = QStringLiteral( "Polygon" );
715 : 0 : break;
716 : : case wkbMultiPoint:
717 : 0 : geom = QStringLiteral( "MultiPoint" );
718 : 0 : break;
719 : : case wkbMultiLineString:
720 : 0 : geom = QStringLiteral( "MultiLineString" );
721 : 0 : break;
722 : : case wkbMultiPolygon:
723 : 0 : geom = QStringLiteral( "MultiPolygon" );
724 : 0 : break;
725 : : case wkbGeometryCollection:
726 : 0 : geom = QStringLiteral( "GeometryCollection" );
727 : 0 : break;
728 : : case wkbCircularString:
729 : 0 : geom = QStringLiteral( "CircularString" );
730 : 0 : break;
731 : : case wkbCompoundCurve:
732 : 0 : geom = QStringLiteral( "CompoundCurve" );
733 : 0 : break;
734 : : case wkbCurvePolygon:
735 : 0 : geom = QStringLiteral( "CurvePolygon" );
736 : 0 : break;
737 : : case wkbMultiCurve:
738 : 0 : geom = QStringLiteral( "MultiCurve" );
739 : 0 : break;
740 : : case wkbMultiSurface:
741 : 0 : geom = QStringLiteral( "MultiSurface" );
742 : 0 : break;
743 : : case wkbCircularStringZ:
744 : 0 : geom = QStringLiteral( "CircularStringZ" );
745 : 0 : break;
746 : : case wkbCompoundCurveZ:
747 : 0 : geom = QStringLiteral( "CompoundCurveZ" );
748 : 0 : break;
749 : : case wkbCurvePolygonZ:
750 : 0 : geom = QStringLiteral( "CurvePolygonZ" );
751 : 0 : break;
752 : : case wkbMultiCurveZ:
753 : 0 : geom = QStringLiteral( "MultiCurveZ" );
754 : 0 : break;
755 : : case wkbMultiSurfaceZ:
756 : 0 : geom = QStringLiteral( "MultiSurfaceZ" );
757 : 0 : break;
758 : : case wkbNone:
759 : 0 : geom = QStringLiteral( "None" );
760 : 0 : break;
761 : : case static_cast<unsigned>( wkbUnknown ) | static_cast<unsigned>( wkb25DBit ):
762 : 0 : geom = QStringLiteral( "Unknown25D" );
763 : 0 : break;
764 : : case static_cast<unsigned>( wkbPoint25D ):
765 : 0 : geom = QStringLiteral( "Point25D" );
766 : 0 : break;
767 : : case static_cast<unsigned>( wkbLineString25D ):
768 : 0 : geom = QStringLiteral( "LineString25D" );
769 : 0 : break;
770 : : case static_cast<unsigned>( wkbPolygon25D ):
771 : 0 : geom = QStringLiteral( "Polygon25D" );
772 : 0 : break;
773 : : case static_cast<unsigned>( wkbMultiPoint25D ):
774 : 0 : geom = QStringLiteral( "MultiPoint25D" );
775 : 0 : break;
776 : : case static_cast<unsigned>( wkbMultiLineString25D ):
777 : 0 : geom = QStringLiteral( "MultiLineString25D" );
778 : 0 : break;
779 : : case static_cast<unsigned>( wkbMultiPolygon25D ):
780 : 0 : geom = QStringLiteral( "MultiPolygon25D" );
781 : 0 : break;
782 : : case static_cast<unsigned>( wkbGeometryCollection25D ):
783 : 0 : geom = QStringLiteral( "GeometryCollection25D" );
784 : 0 : break;
785 : : default:
786 : : // Do not use ':', as it will mess with the separator used by QgsSublayersDialog::populateLayers()
787 : 0 : geom = QStringLiteral( "Unknown WKB (%1)" ).arg( type );
788 : 0 : }
789 : 0 : return geom;
790 : 0 : }
791 : :
792 : 0 : static OGRwkbGeometryType ogrWkbGeometryTypeFromName( const QString &typeName )
793 : : {
794 : 0 : if ( typeName == QLatin1String( "Point" ) ) return wkbPoint;
795 : 0 : else if ( typeName == QLatin1String( "LineString" ) ) return wkbLineString;
796 : 0 : else if ( typeName == QLatin1String( "Polygon" ) ) return wkbPolygon;
797 : 0 : else if ( typeName == QLatin1String( "MultiPoint" ) ) return wkbMultiPoint;
798 : 0 : else if ( typeName == QLatin1String( "MultiLineString" ) ) return wkbMultiLineString;
799 : 0 : else if ( typeName == QLatin1String( "MultiPolygon" ) ) return wkbMultiPolygon;
800 : 0 : else if ( typeName == QLatin1String( "GeometryCollection" ) ) return wkbGeometryCollection;
801 : 0 : else if ( typeName == QLatin1String( "None" ) ) return wkbNone;
802 : 0 : else if ( typeName == QLatin1String( "Point25D" ) ) return wkbPoint25D;
803 : 0 : else if ( typeName == QLatin1String( "LineString25D" ) ) return wkbLineString25D;
804 : 0 : else if ( typeName == QLatin1String( "Polygon25D" ) ) return wkbPolygon25D;
805 : 0 : else if ( typeName == QLatin1String( "MultiPoint25D" ) ) return wkbMultiPoint25D;
806 : 0 : else if ( typeName == QLatin1String( "MultiLineString25D" ) ) return wkbMultiLineString25D;
807 : 0 : else if ( typeName == QLatin1String( "MultiPolygon25D" ) ) return wkbMultiPolygon25D;
808 : 0 : else if ( typeName == QLatin1String( "GeometryCollection25D" ) ) return wkbGeometryCollection25D;
809 : 0 : QgsDebugMsg( QStringLiteral( "unknown geometry type: %1" ).arg( typeName ) );
810 : 0 : return wkbUnknown;
811 : 0 : }
812 : :
813 : 0 : void QgsOgrProvider::addSubLayerDetailsToSubLayerList( int i, QgsOgrLayer *layer, bool withFeatureCount ) const
814 : : {
815 : 0 : QString layerName = QString::fromUtf8( layer->name() );
816 : :
817 : 0 : if ( !mIsSubLayer && ( layerName == QLatin1String( "layer_styles" ) ||
818 : 0 : layerName == QLatin1String( "qgis_projects" ) ) )
819 : : {
820 : : // Ignore layer_styles (coming from QGIS styling support) and
821 : : // qgis_projects (coming from http://plugins.qgis.org/plugins/QgisGeopackage/)
822 : 0 : return;
823 : : }
824 : : // Get first column name,
825 : : // TODO: add support for multiple
826 : 0 : QString geometryColumnName;
827 : 0 : OGRwkbGeometryType layerGeomType = wkbUnknown;
828 : 0 : const bool slowGeomTypeRetrieval =
829 : 0 : mGDALDriverName == QLatin1String( "OAPIF" ) || mGDALDriverName == QLatin1String( "WFS3" ) || mGDALDriverName == QLatin1String( "PGeo" );
830 : 0 : if ( !slowGeomTypeRetrieval )
831 : : {
832 : 0 : QgsOgrFeatureDefn &fdef = layer->GetLayerDefn();
833 : 0 : if ( fdef.GetGeomFieldCount() )
834 : : {
835 : 0 : OGRGeomFieldDefnH geomH = fdef.GetGeomFieldDefn( 0 );
836 : 0 : geometryColumnName = QString::fromUtf8( OGR_GFld_GetNameRef( geomH ) );
837 : 0 : }
838 : 0 : layerGeomType = fdef.GetGeomType();
839 : 0 : }
840 : :
841 : 0 : QString longDescription;
842 : 0 : if ( mGDALDriverName == QLatin1String( "OAPIF" ) || mGDALDriverName == QLatin1String( "WFS3" ) )
843 : : {
844 : 0 : longDescription = layer->GetMetadataItem( "TITLE" );
845 : 0 : }
846 : :
847 : 0 : QgsDebugMsgLevel( QStringLiteral( "id = %1 name = %2 layerGeomType = %3 longDescription = %4" ).arg( i ).arg( layerName ).arg( layerGeomType ). arg( longDescription ), 2 );
848 : :
849 : 0 : if ( slowGeomTypeRetrieval || wkbFlatten( layerGeomType ) != wkbUnknown )
850 : : {
851 : 0 : int layerFeatureCount = withFeatureCount ? layer->GetApproxFeatureCount() : -1;
852 : :
853 : 0 : QString geom = ogrWkbGeometryTypeName( layerGeomType );
854 : :
855 : : // For feature count, -1 indicates an unknown count state
856 : 0 : QStringList parts = QStringList()
857 : 0 : << QString::number( i )
858 : 0 : << layerName
859 : 0 : << QString::number( layerFeatureCount )
860 : 0 : << geom
861 : 0 : << geometryColumnName
862 : 0 : << longDescription;
863 : :
864 : 0 : mSubLayerList << parts.join( sublayerSeparator() );
865 : 0 : }
866 : : else
867 : : {
868 : 0 : QgsDebugMsgLevel( QStringLiteral( "Unknown geometry type, count features for each geometry type" ), 2 );
869 : : // Add virtual sublayers for supported geometry types if layer type is unknown
870 : : // Count features for geometry types
871 : 0 : QMap<OGRwkbGeometryType, int> fCount;
872 : : // TODO: avoid reading attributes, setRelevantFields cannot be called here because it is not constant
873 : :
874 : 0 : layer->ResetReading();
875 : 0 : gdal::ogr_feature_unique_ptr fet;
876 : 0 : while ( fet.reset( layer->GetNextFeature() ), fet )
877 : : {
878 : 0 : OGRGeometryH geom = OGR_F_GetGeometryRef( fet.get() );
879 : 0 : if ( geom )
880 : : {
881 : 0 : OGRwkbGeometryType gType = ogrWkbSingleFlatten( OGR_G_GetGeometryType( geom ) );
882 : 0 : fCount[gType] = fCount.value( gType ) + 1;
883 : 0 : }
884 : : }
885 : 0 : layer->ResetReading();
886 : : // it may happen that there are no features in the layer, in that case add unknown type
887 : : // to show to user that the layer exists but it is empty
888 : 0 : if ( fCount.isEmpty() )
889 : : {
890 : 0 : fCount[wkbUnknown] = 0;
891 : 0 : }
892 : :
893 : : // List TIN and PolyhedralSurface as Polygon
894 : 0 : if ( fCount.contains( wkbTIN ) )
895 : : {
896 : 0 : fCount[wkbPolygon] = fCount.value( wkbPolygon ) + fCount[wkbTIN];
897 : 0 : fCount.remove( wkbTIN );
898 : 0 : }
899 : 0 : if ( fCount.contains( wkbPolyhedralSurface ) )
900 : : {
901 : 0 : fCount[wkbPolygon] = fCount.value( wkbPolygon ) + fCount[wkbPolyhedralSurface];
902 : 0 : fCount.remove( wkbPolyhedralSurface );
903 : 0 : }
904 : : // When there are CurvePolygons, promote Polygons
905 : 0 : if ( fCount.contains( wkbPolygon ) && fCount.contains( wkbCurvePolygon ) )
906 : : {
907 : 0 : fCount[wkbCurvePolygon] += fCount.value( wkbPolygon );
908 : 0 : fCount.remove( wkbPolygon );
909 : 0 : }
910 : : // When there are CompoundCurves, promote LineStrings and CircularStrings
911 : 0 : if ( fCount.contains( wkbLineString ) && fCount.contains( wkbCompoundCurve ) )
912 : : {
913 : 0 : fCount[wkbCompoundCurve] += fCount.value( wkbLineString );
914 : 0 : fCount.remove( wkbLineString );
915 : 0 : }
916 : 0 : if ( fCount.contains( wkbCircularString ) && fCount.contains( wkbCompoundCurve ) )
917 : : {
918 : 0 : fCount[wkbCompoundCurve] += fCount.value( wkbCircularString );
919 : 0 : fCount.remove( wkbCircularString );
920 : 0 : }
921 : :
922 : 0 : bool bIs25D = wkbHasZ( layerGeomType );
923 : 0 : QMap<OGRwkbGeometryType, int>::const_iterator countIt = fCount.constBegin();
924 : 0 : for ( ; countIt != fCount.constEnd(); ++countIt )
925 : : {
926 : 0 : QString geom = ogrWkbGeometryTypeName( ( bIs25D ) ? wkbSetZ( countIt.key() ) : countIt.key() );
927 : :
928 : 0 : QStringList parts = QStringList()
929 : 0 : << QString::number( i )
930 : 0 : << layerName
931 : 0 : << QString::number( fCount.value( countIt.key() ) )
932 : 0 : << geom
933 : 0 : << geometryColumnName
934 : 0 : << longDescription;
935 : :
936 : 0 : QString sl = parts.join( sublayerSeparator() );
937 : 0 : QgsDebugMsgLevel( "sub layer: " + sl, 2 );
938 : 0 : mSubLayerList << sl;
939 : 0 : }
940 : 0 : }
941 : 0 : }
942 : :
943 : 0 : uint QgsOgrProvider::subLayerCount() const
944 : : {
945 : 0 : uint count = layerCount();
946 : :
947 : 0 : QString errCause;
948 : 0 : QgsOgrLayerUniquePtr layerStyles = QgsOgrProviderUtils::getLayer( mFilePath, "layer_styles", errCause );
949 : 0 : if ( layerStyles )
950 : : {
951 : 0 : count--;
952 : 0 : }
953 : 0 : return count;
954 : 0 : }
955 : :
956 : 0 : QStringList QgsOgrProvider::subLayers() const
957 : : {
958 : 0 : const bool withFeatureCount = ( mReadFlags & QgsDataProvider::SkipFeatureCount ) == 0;
959 : 0 : return _subLayers( withFeatureCount );
960 : : }
961 : :
962 : 64 : QgsLayerMetadata QgsOgrProvider::layerMetadata() const
963 : : {
964 : 64 : return mLayerMetadata;
965 : : }
966 : :
967 : 0 : QStringList QgsOgrProvider::subLayersWithoutFeatureCount() const
968 : : {
969 : 0 : return _subLayers( false );
970 : : }
971 : :
972 : 0 : QStringList QgsOgrProvider::_subLayers( bool withFeatureCount ) const
973 : : {
974 : 0 : QgsCPLHTTPFetchOverrider oCPLHTTPFetcher( mAuthCfg );
975 : 0 : QgsSetCPLHTTPFetchOverriderInitiatorClass( oCPLHTTPFetcher, QStringLiteral( "QgsOgrProvider" ) );
976 : :
977 : 0 : if ( !mValid )
978 : : {
979 : 0 : return QStringList();
980 : : }
981 : :
982 : 0 : if ( !mSubLayerList.isEmpty() )
983 : 0 : return mSubLayerList;
984 : :
985 : 0 : if ( mOgrLayer && ( mIsSubLayer || layerCount() == 1 ) )
986 : : {
987 : 0 : addSubLayerDetailsToSubLayerList( mLayerIndex, mOgrLayer, withFeatureCount );
988 : 0 : }
989 : : else
990 : : {
991 : : // In case there is no free opened dataset in the cache, keep the first
992 : : // layer alive while we iterate over the other layers, so that we can
993 : : // reuse the same dataset. Can help in a particular with a FileGDB with
994 : : // the FileGDB driver
995 : 0 : QgsOgrLayerUniquePtr firstLayer;
996 : 0 : for ( unsigned int i = 0; i < layerCount() ; i++ )
997 : : {
998 : 0 : QString errCause;
999 : 0 : QgsOgrLayerUniquePtr layer = QgsOgrProviderUtils::getLayer( mOgrOrigLayer->datasetName(),
1000 : 0 : mOgrOrigLayer->updateMode(),
1001 : 0 : mOgrOrigLayer->options(),
1002 : 0 : i,
1003 : : errCause,
1004 : : // do not check timestamp beyond the first
1005 : : // layer
1006 : 0 : firstLayer == nullptr );
1007 : 0 : if ( !layer )
1008 : 0 : continue;
1009 : :
1010 : 0 : addSubLayerDetailsToSubLayerList( i, layer.get(), withFeatureCount );
1011 : 0 : if ( firstLayer == nullptr )
1012 : : {
1013 : 0 : firstLayer = std::move( layer );
1014 : 0 : }
1015 : 0 : }
1016 : 0 : }
1017 : 0 : return mSubLayerList;
1018 : 0 : }
1019 : :
1020 : 187 : void QgsOgrProvider::setEncoding( const QString &e )
1021 : : {
1022 : 187 : QgsCPLHTTPFetchOverrider oCPLHTTPFetcher( mAuthCfg );
1023 : 374 : QgsSetCPLHTTPFetchOverriderInitiatorClass( oCPLHTTPFetcher, QStringLiteral( "QgsOgrProvider" ) );
1024 : :
1025 : 187 : QgsSettings settings;
1026 : :
1027 : : // if the layer has the OLCStringsAsUTF8 capability, we CANNOT override the
1028 : : // encoding on the QGIS side!
1029 : 187 : if ( mOgrLayer && !mOgrLayer->TestCapability( OLCStringsAsUTF8 ) )
1030 : : {
1031 : 186 : QgsVectorDataProvider::setEncoding( e );
1032 : 186 : }
1033 : : else
1034 : : {
1035 : 2 : QgsVectorDataProvider::setEncoding( QStringLiteral( "UTF-8" ) );
1036 : : }
1037 : 187 : loadFields();
1038 : 187 : }
1039 : :
1040 : : // This is reused by dataItem
1041 : 187 : OGRwkbGeometryType QgsOgrProvider::getOgrGeomType( const QString &driverName, OGRLayerH ogrLayer )
1042 : : {
1043 : 187 : OGRFeatureDefnH fdef = OGR_L_GetLayerDefn( ogrLayer );
1044 : 187 : OGRwkbGeometryType geomType = wkbUnknown;
1045 : 187 : if ( fdef )
1046 : : {
1047 : 187 : geomType = OGR_FD_GetGeomType( fdef );
1048 : :
1049 : : // Handle wkbUnknown and its Z/M variants. QGIS has no unknown Z/M variants,
1050 : : // so just use flat wkbUnknown
1051 : 187 : if ( wkbFlatten( geomType ) == wkbUnknown )
1052 : 0 : geomType = wkbUnknown;
1053 : :
1054 : : // Some ogr drivers (e.g. GML) are not able to determine the geometry type of a layer like this.
1055 : : // In such cases, we use virtual sublayers for each geometry if the layer contains
1056 : : // multiple geometries (see subLayers) otherwise we guess geometry type from the first
1057 : : // feature that has a geometry (limit us to a few features, not the whole layer)
1058 : 187 : if ( geomType == wkbUnknown )
1059 : : {
1060 : 0 : geomType = wkbNone;
1061 : 0 : OGR_L_ResetReading( ogrLayer );
1062 : 0 : for ( int i = 0; i < 10; i++ )
1063 : : {
1064 : 0 : gdal::ogr_feature_unique_ptr nextFeature( OGR_L_GetNextFeature( ogrLayer ) );
1065 : 0 : if ( !nextFeature )
1066 : 0 : break;
1067 : :
1068 : 0 : OGRGeometryH geometry = OGR_F_GetGeometryRef( nextFeature.get() );
1069 : 0 : if ( geometry )
1070 : : {
1071 : 0 : geomType = OGR_G_GetGeometryType( geometry );
1072 : :
1073 : : // Shapefile MultiPatch can be reported as GeometryCollectionZ of TINZ
1074 : 0 : if ( wkbFlatten( geomType ) == wkbGeometryCollection &&
1075 : 0 : driverName == QLatin1String( "ESRI Shapefile" ) &&
1076 : 0 : OGR_G_GetGeometryCount( geometry ) >= 1 &&
1077 : 0 : wkbFlatten( OGR_G_GetGeometryType( OGR_G_GetGeometryRef( geometry, 0 ) ) ) == wkbTIN )
1078 : : {
1079 : 0 : geomType = wkbMultiPolygon25D;
1080 : 0 : }
1081 : 0 : }
1082 : 0 : if ( geomType != wkbNone )
1083 : 0 : break;
1084 : 0 : }
1085 : 0 : OGR_L_ResetReading( ogrLayer );
1086 : 0 : }
1087 : 187 : }
1088 : 187 : return geomType;
1089 : 0 : }
1090 : :
1091 : 187 : void QgsOgrProvider::loadFields()
1092 : : {
1093 : 187 : QgsOgrConnPool::instance()->invalidateConnections( QgsOgrProviderUtils::connectionPoolId( dataSourceUri( true ), mShareSameDatasetAmongLayers ) );
1094 : : //the attribute fields need to be read again when the encoding changes
1095 : 187 : mAttributeFields.clear();
1096 : 187 : mDefaultValues.clear();
1097 : 187 : mPrimaryKeyAttrs.clear();
1098 : 187 : if ( !mOgrLayer )
1099 : 0 : return;
1100 : :
1101 : 187 : if ( mOgrGeometryTypeFilter != wkbUnknown )
1102 : : {
1103 : 0 : mOGRGeomType = mOgrGeometryTypeFilter;
1104 : 0 : }
1105 : : else
1106 : : {
1107 : 187 : QMutex *mutex = nullptr;
1108 : 187 : OGRLayerH ogrLayer = mOgrLayer->getHandleAndMutex( mutex );
1109 : 187 : QMutexLocker locker( mutex );
1110 : 187 : mOGRGeomType = getOgrGeomType( mGDALDriverName, ogrLayer );
1111 : 187 : }
1112 : 187 : QgsOgrFeatureDefn &fdef = mOgrLayer->GetLayerDefn();
1113 : :
1114 : : // Expose the OGR FID if it comes from a "real" column (typically GPKG)
1115 : : // and make sure that this FID column is not exposed as a regular OGR field (shouldn't happen normally)
1116 : 187 : QByteArray fidColumn( mOgrLayer->GetFIDColumn() );
1117 : 188 : mFirstFieldIsFid = !fidColumn.isEmpty() &&
1118 : 1 : fdef.GetFieldIndex( fidColumn ) < 0;
1119 : :
1120 : : #if GDAL_VERSION_NUM < GDAL_COMPUTE_VERSION(3, 2, 1)
1121 : : // This is a temporary solution until GDAL Unique support is available
1122 : : QSet<QString> uniqueFieldNames;
1123 : :
1124 : :
1125 : : if ( mGDALDriverName == QLatin1String( "GPKG" ) )
1126 : : {
1127 : : sqlite3_database_unique_ptr dsPtr;
1128 : : if ( dsPtr.open_v2( mFilePath, SQLITE_OPEN_READONLY, nullptr ) == SQLITE_OK )
1129 : : {
1130 : : QString errMsg;
1131 : : uniqueFieldNames = QgsSqliteUtils::uniqueFields( dsPtr.get(), mOgrLayer->name(), errMsg );
1132 : : if ( ! errMsg.isEmpty() )
1133 : : {
1134 : : QgsMessageLog::logMessage( tr( "GPKG error searching for unique constraints on fields for table %1. (%2)" ).arg( QString( mOgrLayer->name() ), errMsg ), tr( "OGR" ) );
1135 : : }
1136 : : }
1137 : : }
1138 : : #endif
1139 : :
1140 : 187 : int createdFields = 0;
1141 : 187 : if ( mFirstFieldIsFid )
1142 : : {
1143 : 1 : QgsField fidField(
1144 : 1 : fidColumn,
1145 : : QVariant::LongLong,
1146 : 2 : QStringLiteral( "Integer64" )
1147 : : );
1148 : : // Set constraints for feature id
1149 : 1 : QgsFieldConstraints constraints = fidField.constraints();
1150 : 1 : constraints.setConstraint( QgsFieldConstraints::ConstraintUnique, QgsFieldConstraints::ConstraintOriginProvider );
1151 : 1 : constraints.setConstraint( QgsFieldConstraints::ConstraintNotNull, QgsFieldConstraints::ConstraintOriginProvider );
1152 : 1 : fidField.setConstraints( constraints );
1153 : 1 : mAttributeFields.append(
1154 : : fidField
1155 : : );
1156 : 1 : mDefaultValues.insert( 0, tr( "Autogenerate" ) );
1157 : 1 : createdFields++;
1158 : 1 : mPrimaryKeyAttrs << 0;
1159 : 1 : }
1160 : :
1161 : 432 : for ( int i = 0; i < fdef.GetFieldCount(); ++i )
1162 : : {
1163 : 245 : OGRFieldDefnH fldDef = fdef.GetFieldDefn( i );
1164 : 245 : OGRFieldType ogrType = OGR_Fld_GetType( fldDef );
1165 : 245 : OGRFieldSubType ogrSubType = OFSTNone;
1166 : :
1167 : : QVariant::Type varType;
1168 : 245 : QVariant::Type varSubType = QVariant::Invalid;
1169 : 245 : switch ( ogrType )
1170 : : {
1171 : : case OFTInteger:
1172 : 56 : if ( OGR_Fld_GetSubType( fldDef ) == OFSTBoolean )
1173 : : {
1174 : 0 : varType = QVariant::Bool;
1175 : 0 : ogrSubType = OFSTBoolean;
1176 : 0 : }
1177 : : else
1178 : 56 : varType = QVariant::Int;
1179 : 56 : break;
1180 : : case OFTInteger64:
1181 : 186 : varType = QVariant::LongLong;
1182 : 186 : break;
1183 : : case OFTReal:
1184 : 0 : varType = QVariant::Double;
1185 : 0 : break;
1186 : : case OFTDate:
1187 : 0 : varType = QVariant::Date;
1188 : 0 : break;
1189 : : case OFTTime:
1190 : 0 : varType = QVariant::Time;
1191 : 0 : break;
1192 : : case OFTDateTime:
1193 : 0 : varType = QVariant::DateTime;
1194 : 0 : break;
1195 : :
1196 : : case OFTBinary:
1197 : 0 : varType = QVariant::ByteArray;
1198 : 0 : break;
1199 : :
1200 : : case OFTString:
1201 : 3 : if ( OGR_Fld_GetSubType( fldDef ) == OFSTJSON )
1202 : : {
1203 : 0 : ogrSubType = OFSTJSON;
1204 : 0 : varType = QVariant::Map;
1205 : 0 : varSubType = QVariant::String;
1206 : 0 : }
1207 : : else
1208 : : {
1209 : 3 : varType = QVariant::String;
1210 : : }
1211 : 3 : break;
1212 : :
1213 : : case OFTStringList:
1214 : 0 : varType = QVariant::List;
1215 : 0 : varSubType = QVariant::String;
1216 : 0 : break;
1217 : :
1218 : : default:
1219 : 0 : varType = QVariant::String; // other unsupported, leave it as a string
1220 : 0 : }
1221 : :
1222 : : //TODO: fix this hack
1223 : : #ifdef ANDROID
1224 : : QString name = OGR_Fld_GetNameRef( fldDef );
1225 : : #else
1226 : 245 : QString name = textEncoding()->toUnicode( OGR_Fld_GetNameRef( fldDef ) );
1227 : : #endif
1228 : :
1229 : 245 : if ( mAttributeFields.indexFromName( name ) != -1 )
1230 : : {
1231 : :
1232 : 0 : QString tmpname = name + "_%1";
1233 : 0 : int fix = 0;
1234 : :
1235 : 0 : while ( mAttributeFields.indexFromName( name ) != -1 )
1236 : : {
1237 : 0 : name = tmpname.arg( ++fix );
1238 : : }
1239 : 0 : }
1240 : :
1241 : 245 : int width = OGR_Fld_GetWidth( fldDef );
1242 : 245 : int prec = OGR_Fld_GetPrecision( fldDef );
1243 : 245 : if ( prec > 0 )
1244 : 0 : width -= 1;
1245 : :
1246 : 245 : QString typeName = OGR_GetFieldTypeName( ogrType );
1247 : 245 : if ( ogrSubType != OFSTNone )
1248 : 0 : typeName = OGR_GetFieldSubTypeName( ogrSubType );
1249 : :
1250 : 245 : QgsField newField = QgsField(
1251 : : name,
1252 : 245 : varType,
1253 : : #ifdef ANDROID
1254 : : typeName,
1255 : : #else
1256 : 245 : textEncoding()->toUnicode( typeName.toStdString().c_str() ),
1257 : : #endif
1258 : 245 : width, prec, QString(), varSubType
1259 : : );
1260 : :
1261 : : #if GDAL_VERSION_NUM >= GDAL_COMPUTE_VERSION(3,2,0)
1262 : 245 : const QString alias = textEncoding()->toUnicode( OGR_Fld_GetAlternativeNameRef( fldDef ) );
1263 : 245 : if ( !alias.isEmpty() )
1264 : : {
1265 : 0 : newField.setAlias( alias );
1266 : 0 : }
1267 : : #endif
1268 : :
1269 : : // check if field is nullable
1270 : 245 : bool nullable = OGR_Fld_IsNullable( fldDef );
1271 : 245 : if ( !nullable )
1272 : : {
1273 : 0 : QgsFieldConstraints constraints;
1274 : 0 : constraints.setConstraint( QgsFieldConstraints::ConstraintNotNull, QgsFieldConstraints::ConstraintOriginProvider );
1275 : 0 : newField.setConstraints( constraints );
1276 : 0 : }
1277 : :
1278 : : #if GDAL_VERSION_NUM < GDAL_COMPUTE_VERSION(3, 2, 1)
1279 : : if ( uniqueFieldNames.contains( OGR_Fld_GetNameRef( fldDef ) ) )
1280 : : #else
1281 : 245 : if ( OGR_Fld_IsUnique( fldDef ) )
1282 : : #endif
1283 : : {
1284 : 0 : QgsFieldConstraints constraints = newField.constraints();
1285 : 0 : constraints.setConstraint( QgsFieldConstraints::ConstraintUnique, QgsFieldConstraints::ConstraintOriginProvider );
1286 : 0 : newField.setConstraints( constraints );
1287 : 0 : }
1288 : :
1289 : : // check if field has default value
1290 : 245 : QString defaultValue = textEncoding()->toUnicode( OGR_Fld_GetDefault( fldDef ) );
1291 : 245 : if ( !defaultValue.isEmpty() && !OGR_Fld_IsDefaultDriverSpecific( fldDef ) )
1292 : : {
1293 : 0 : if ( defaultValue.startsWith( '\'' ) )
1294 : : {
1295 : 0 : defaultValue = defaultValue.remove( 0, 1 );
1296 : 0 : defaultValue.chop( 1 );
1297 : 0 : defaultValue.replace( QLatin1String( "''" ), QLatin1String( "'" ) );
1298 : 0 : }
1299 : 0 : mDefaultValues.insert( createdFields, defaultValue );
1300 : 0 : }
1301 : :
1302 : 245 : mAttributeFields.append( newField );
1303 : 245 : createdFields++;
1304 : 245 : }
1305 : 187 : }
1306 : :
1307 : :
1308 : 0 : QString QgsOgrProvider::storageType() const
1309 : : {
1310 : : // Delegate to the driver loaded in by OGR
1311 : 0 : return mGDALDriverName;
1312 : : }
1313 : :
1314 : :
1315 : 32 : void QgsOgrProvider::setRelevantFields( bool fetchGeometry, const QgsAttributeList &fetchAttributes ) const
1316 : : {
1317 : 32 : QMutex *mutex = nullptr;
1318 : 32 : OGRLayerH ogrLayer = mOgrLayer->getHandleAndMutex( mutex );
1319 : 32 : QMutexLocker locker( mutex );
1320 : 32 : QgsOgrProviderUtils::setRelevantFields( ogrLayer, mAttributeFields.count(), fetchGeometry, fetchAttributes, mFirstFieldIsFid, mSubsetString );
1321 : 32 : }
1322 : :
1323 : :
1324 : 124 : void QgsOgrProviderUtils::setRelevantFields( OGRLayerH ogrLayer, int fieldCount,
1325 : : bool fetchGeometry,
1326 : : const QgsAttributeList &fetchAttributes,
1327 : : bool firstAttrIsFid,
1328 : : const QString &subsetString )
1329 : : {
1330 : 124 : if ( OGR_L_TestCapability( ogrLayer, OLCIgnoreFields ) )
1331 : : {
1332 : 124 : QVector<const char *> ignoredFields;
1333 : 124 : OGRFeatureDefnH featDefn = OGR_L_GetLayerDefn( ogrLayer );
1334 : 308 : for ( int i = ( firstAttrIsFid ? 1 : 0 ); i < fieldCount; i++ )
1335 : : {
1336 : 184 : if ( !fetchAttributes.contains( i ) )
1337 : : {
1338 : : // add to ignored fields
1339 : 10 : if ( OGRFieldDefnH field = OGR_FD_GetFieldDefn( featDefn, firstAttrIsFid ? i - 1 : i ) )
1340 : : {
1341 : 10 : const char *fieldName = OGR_Fld_GetNameRef( field );
1342 : : // This is implemented a bit in a hacky way, but in case we are acting on a layer
1343 : : // with a subset filter, do not ignore fields that are found in the
1344 : : // where clause. We do this in a rough way, by looking, in a case
1345 : : // insensitive way, if the current field name is in the subsetString,
1346 : : // so we potentially don't ignore fields we could, in situations like
1347 : : // subsetFilter == "foobar = 2", and there's a "foo" or "bar" field.
1348 : : // Better be safe than sorry.
1349 : : // We could argue that OGR_L_SetIgnoredFields() should be aware of
1350 : : // the fields of the attribute filter, and do not ignore them.
1351 : 10 : if ( subsetString.isEmpty() ||
1352 : 0 : subsetString.indexOf( QString::fromUtf8( fieldName ), 0, Qt::CaseInsensitive ) < 0 )
1353 : : {
1354 : 10 : ignoredFields.append( fieldName );
1355 : 10 : }
1356 : 10 : }
1357 : 10 : }
1358 : 184 : }
1359 : :
1360 : 124 : if ( !fetchGeometry )
1361 : 0 : ignoredFields.append( "OGR_GEOMETRY" );
1362 : 124 : ignoredFields.append( "OGR_STYLE" ); // not used by QGIS
1363 : 124 : ignoredFields.append( nullptr );
1364 : :
1365 : 124 : OGR_L_SetIgnoredFields( ogrLayer, ignoredFields.data() );
1366 : 124 : }
1367 : 124 : }
1368 : :
1369 : 1 : QgsFeatureIterator QgsOgrProvider::getFeatures( const QgsFeatureRequest &request ) const
1370 : : {
1371 : 1 : return QgsFeatureIterator( new QgsOgrFeatureIterator( static_cast<QgsOgrFeatureSource *>( featureSource() ), true, request, mTransaction ) );
1372 : 0 : }
1373 : :
1374 : :
1375 : 0 : unsigned char *QgsOgrProvider::getGeometryPointer( OGRFeatureH fet )
1376 : : {
1377 : 0 : OGRGeometryH geom = OGR_F_GetGeometryRef( fet );
1378 : 0 : unsigned char *gPtr = nullptr;
1379 : :
1380 : 0 : if ( !geom )
1381 : 0 : return nullptr;
1382 : :
1383 : : // get the wkb representation
1384 : 0 : gPtr = new unsigned char[OGR_G_WkbSize( geom )];
1385 : :
1386 : 0 : OGR_G_ExportToWkb( geom, ( OGRwkbByteOrder ) QgsApplication::endian(), gPtr );
1387 : 0 : return gPtr;
1388 : 0 : }
1389 : :
1390 : :
1391 : 4 : QgsRectangle QgsOgrProvider::extent() const
1392 : : {
1393 : 4 : if ( !mExtent )
1394 : : {
1395 : 4 : QgsCPLHTTPFetchOverrider oCPLHTTPFetcher( mAuthCfg );
1396 : 8 : QgsSetCPLHTTPFetchOverriderInitiatorClass( oCPLHTTPFetcher, QStringLiteral( "QgsOgrProvider" ) );
1397 : :
1398 : 4 : mExtent.reset( new OGREnvelope() );
1399 : :
1400 : : // get the extent_ (envelope) of the layer
1401 : 4 : QgsDebugMsgLevel( QStringLiteral( "Starting get extent" ), 3 );
1402 : :
1403 : 4 : if ( mForceRecomputeExtent && mValid && mGDALDriverName == QLatin1String( "GPKG" ) && mOgrOrigLayer )
1404 : : {
1405 : : // works with unquoted layerName
1406 : 0 : QByteArray sql = QByteArray( "RECOMPUTE EXTENT ON " ) + mOgrOrigLayer->name();
1407 : 0 : QgsDebugMsgLevel( QStringLiteral( "SQL: %1" ).arg( QString::fromUtf8( sql ) ), 2 );
1408 : 0 : mOgrOrigLayer->ExecuteSQLNoReturn( sql );
1409 : 0 : }
1410 : :
1411 : 4 : mExtent->MinX = std::numeric_limits<double>::max();
1412 : 4 : mExtent->MinY = std::numeric_limits<double>::max();
1413 : 4 : mExtent->MaxX = -std::numeric_limits<double>::max();
1414 : 4 : mExtent->MaxY = -std::numeric_limits<double>::max();
1415 : :
1416 : : // TODO: This can be expensive, do we really need it!
1417 : 4 : if ( mOgrLayer == mOgrOrigLayer.get() && mSubsetString.isEmpty() )
1418 : : {
1419 : 4 : if ( ( mGDALDriverName == QLatin1String( "OAPIF" ) || mGDALDriverName == QLatin1String( "WFS3" ) ) &&
1420 : 0 : !mOgrLayer->TestCapability( OLCFastGetExtent ) )
1421 : : {
1422 : : // When the extent is not in the metadata, retrieving it would be
1423 : : // super slow
1424 : 0 : mExtent->MinX = -180;
1425 : 0 : mExtent->MinY = -90;
1426 : 0 : mExtent->MaxX = 180;
1427 : 0 : mExtent->MaxY = 90;
1428 : 0 : }
1429 : : else
1430 : : {
1431 : 4 : mOgrLayer->GetExtent( mExtent.get(), true );
1432 : : }
1433 : 4 : }
1434 : : else
1435 : : {
1436 : 0 : gdal::ogr_feature_unique_ptr f;
1437 : :
1438 : 0 : mOgrLayer->ResetReading();
1439 : 0 : while ( f.reset( mOgrLayer->GetNextFeature() ), f )
1440 : : {
1441 : 0 : OGRGeometryH g = OGR_F_GetGeometryRef( f.get() );
1442 : 0 : if ( g && !OGR_G_IsEmpty( g ) )
1443 : : {
1444 : 0 : OGREnvelope env;
1445 : 0 : OGR_G_GetEnvelope( g, &env );
1446 : :
1447 : 0 : mExtent->MinX = std::min( mExtent->MinX, env.MinX );
1448 : 0 : mExtent->MinY = std::min( mExtent->MinY, env.MinY );
1449 : 0 : mExtent->MaxX = std::max( mExtent->MaxX, env.MaxX );
1450 : 0 : mExtent->MaxY = std::max( mExtent->MaxY, env.MaxY );
1451 : 0 : }
1452 : : }
1453 : 0 : mOgrLayer->ResetReading();
1454 : 0 : }
1455 : :
1456 : 4 : QgsDebugMsgLevel( QStringLiteral( "Finished get extent" ), 4 );
1457 : 4 : }
1458 : :
1459 : 4 : mExtentRect.set( mExtent->MinX, mExtent->MinY, mExtent->MaxX, mExtent->MaxY );
1460 : 4 : return mExtentRect;
1461 : 0 : }
1462 : :
1463 : 0 : QVariant QgsOgrProvider::defaultValue( int fieldId ) const
1464 : : {
1465 : 0 : QgsCPLHTTPFetchOverrider oCPLHTTPFetcher( mAuthCfg );
1466 : 0 : QgsSetCPLHTTPFetchOverriderInitiatorClass( oCPLHTTPFetcher, QStringLiteral( "QgsOgrProvider" ) );
1467 : :
1468 : 0 : if ( fieldId < 0 || fieldId >= mAttributeFields.count() )
1469 : 0 : return QVariant();
1470 : :
1471 : 0 : QString defaultVal = mDefaultValues.value( fieldId, QString() );
1472 : 0 : if ( defaultVal.isEmpty() )
1473 : 0 : return QVariant();
1474 : :
1475 : 0 : QVariant resultVar = defaultVal;
1476 : 0 : if ( defaultVal == QLatin1String( "CURRENT_TIMESTAMP" ) )
1477 : 0 : resultVar = QDateTime::currentDateTime();
1478 : 0 : else if ( defaultVal == QLatin1String( "CURRENT_DATE" ) )
1479 : 0 : resultVar = QDate::currentDate();
1480 : 0 : else if ( defaultVal == QLatin1String( "CURRENT_TIME" ) )
1481 : 0 : resultVar = QTime::currentTime();
1482 : :
1483 : : // Get next sequence value for sqlite in case we are inside a transaction
1484 : 0 : if ( mOgrOrigLayer &&
1485 : 0 : mTransaction &&
1486 : 0 : mDefaultValues.value( fieldId, QString() ) == tr( "Autogenerate" ) &&
1487 : 0 : providerProperty( EvaluateDefaultValues, false ).toBool() &&
1488 : 0 : ( mGDALDriverName == QLatin1String( "GPKG" ) ||
1489 : 0 : mGDALDriverName == QLatin1String( "SQLite" ) ) &&
1490 : 0 : mFirstFieldIsFid &&
1491 : 0 : fieldId == 0 )
1492 : : {
1493 : 0 : QgsOgrLayerUniquePtr resultLayer = mOgrOrigLayer->ExecuteSQL( QByteArray( "SELECT seq FROM sqlite_sequence WHERE name = " ) + QgsSqliteUtils::quotedValue( mOgrOrigLayer->name() ).toUtf8() );
1494 : 0 : if ( resultLayer )
1495 : : {
1496 : 0 : gdal::ogr_feature_unique_ptr f;
1497 : 0 : if ( f.reset( resultLayer->GetNextFeature() ), f )
1498 : : {
1499 : 0 : bool ok { true };
1500 : 0 : const QVariant res = QgsOgrUtils::getOgrFeatureAttribute( f.get(),
1501 : 0 : fields().at( 0 ),
1502 : 0 : 0, textEncoding(), &ok );
1503 : 0 : if ( ok )
1504 : : {
1505 : 0 : long long nextVal { res.toLongLong( &ok ) };
1506 : 0 : if ( ok )
1507 : : {
1508 : : // Increment
1509 : 0 : resultVar = ++nextVal;
1510 : 0 : mOgrOrigLayer->ExecuteSQLNoReturn( QByteArray( "UPDATE sqlite_sequence SET seq = seq + 1 WHERE name = " ) + QgsSqliteUtils::quotedValue( mOgrOrigLayer->name() ).toUtf8() );
1511 : 0 : }
1512 : 0 : }
1513 : :
1514 : 0 : if ( ! ok )
1515 : : {
1516 : 0 : QgsMessageLog::logMessage( tr( "Error retrieving next sequence value for %1" ).arg( QString::fromUtf8( mOgrOrigLayer->name() ) ), tr( "OGR" ) );
1517 : 0 : }
1518 : 0 : }
1519 : : else // no sequence!
1520 : : {
1521 : 0 : resultVar = 1;
1522 : 0 : mOgrOrigLayer->ExecuteSQLNoReturn( QByteArray( "INSERT INTO sqlite_sequence (name, seq) VALUES( " +
1523 : 0 : QgsSqliteUtils::quotedValue( mOgrOrigLayer->name() ).toUtf8() ) + ", 1)" );
1524 : : }
1525 : 0 : }
1526 : : else
1527 : : {
1528 : 0 : QgsMessageLog::logMessage( tr( "Error retrieving default value for %1" ).arg( mLayerName ), tr( "OGR" ) );
1529 : : }
1530 : 0 : }
1531 : :
1532 : 0 : ( void )mAttributeFields.at( fieldId ).convertCompatible( resultVar );
1533 : 0 : return resultVar;
1534 : 0 : }
1535 : :
1536 : 0 : QString QgsOgrProvider::defaultValueClause( int fieldIndex ) const
1537 : : {
1538 : : // Return empty clause to force defaultValue calls for sqlite in case we are inside a transaction
1539 : 0 : if ( mTransaction &&
1540 : 0 : mDefaultValues.value( fieldIndex, QString() ) == tr( "Autogenerate" ) &&
1541 : 0 : providerProperty( EvaluateDefaultValues, false ).toBool() &&
1542 : 0 : ( mGDALDriverName == QLatin1String( "GPKG" ) ||
1543 : 0 : mGDALDriverName == QLatin1String( "SQLite" ) ) &&
1544 : 0 : mFirstFieldIsFid &&
1545 : 0 : fieldIndex == 0 )
1546 : 0 : return QString();
1547 : : else
1548 : 0 : return mDefaultValues.value( fieldIndex, QString() );
1549 : 0 : }
1550 : :
1551 : 0 : bool QgsOgrProvider::skipConstraintCheck( int fieldIndex, QgsFieldConstraints::Constraint constraint, const QVariant &value ) const
1552 : : {
1553 : : Q_UNUSED( constraint )
1554 : 0 : if ( providerProperty( EvaluateDefaultValues, false ).toBool() )
1555 : : {
1556 : 0 : return ! mDefaultValues.value( fieldIndex ).isEmpty();
1557 : : }
1558 : : else
1559 : : {
1560 : : // stricter check
1561 : 0 : return mDefaultValues.contains( fieldIndex ) && mDefaultValues.value( fieldIndex ) == value.toString() && !value.isNull();
1562 : : }
1563 : 0 : }
1564 : :
1565 : 0 : void QgsOgrProvider::updateExtents()
1566 : : {
1567 : 0 : invalidateCachedExtent( true );
1568 : 0 : }
1569 : :
1570 : 387 : void QgsOgrProvider::invalidateCachedExtent( bool bForceRecomputeExtent )
1571 : : {
1572 : 387 : mForceRecomputeExtent = bForceRecomputeExtent;
1573 : 387 : mExtent.reset();
1574 : 387 : }
1575 : :
1576 : 0 : size_t QgsOgrProvider::layerCount() const
1577 : : {
1578 : 0 : if ( !mValid )
1579 : 0 : return 0;
1580 : 0 : return mOgrLayer->GetLayerCount();
1581 : 0 : }
1582 : :
1583 : : /**
1584 : : * Returns the feature type
1585 : : */
1586 : 197 : QgsWkbTypes::Type QgsOgrProvider::wkbType() const
1587 : : {
1588 : 197 : QgsWkbTypes::Type wkb = QgsOgrUtils::ogrGeometryTypeToQgsWkbType( mOGRGeomType );
1589 : 197 : const QgsWkbTypes::Type wkbFlat = QgsWkbTypes::flatType( wkb );
1590 : 197 : if ( mGDALDriverName == QLatin1String( "ESRI Shapefile" ) && ( wkbFlat == QgsWkbTypes::LineString || wkbFlat == QgsWkbTypes::Polygon ) )
1591 : : {
1592 : 137 : wkb = QgsWkbTypes::multiType( wkb );
1593 : 137 : }
1594 : 197 : if ( mOGRGeomType % 1000 == wkbPolyhedralSurface ) // is PolyhedralSurface, PolyhedralSurfaceZ, PolyhedralSurfaceM or PolyhedralSurfaceZM => map to MultiPolygon
1595 : : {
1596 : 0 : wkb = static_cast<QgsWkbTypes::Type>( mOGRGeomType - ( wkbPolyhedralSurface - wkbMultiPolygon ) );
1597 : 0 : }
1598 : 197 : else if ( mOGRGeomType % 1000 == wkbTIN ) // is TIN, TINZ, TINM or TINZM => map to MultiPolygon
1599 : : {
1600 : 0 : wkb = static_cast<QgsWkbTypes::Type>( mOGRGeomType - ( wkbTIN - wkbMultiPolygon ) );
1601 : 0 : }
1602 : 197 : return wkb;
1603 : : }
1604 : :
1605 : : /**
1606 : : * Returns the feature count
1607 : : */
1608 : 2 : long QgsOgrProvider::featureCount() const
1609 : : {
1610 : 2 : if ( ( mReadFlags & QgsDataProvider::SkipFeatureCount ) != 0 )
1611 : : {
1612 : 0 : return QgsVectorDataProvider::UnknownCount;
1613 : : }
1614 : 2 : if ( mRefreshFeatureCount )
1615 : : {
1616 : 2 : mRefreshFeatureCount = false;
1617 : 2 : recalculateFeatureCount();
1618 : 2 : }
1619 : 2 : return mFeaturesCounted;
1620 : 2 : }
1621 : :
1622 : :
1623 : 96 : QgsFields QgsOgrProvider::fields() const
1624 : : {
1625 : 96 : return mAttributeFields;
1626 : : }
1627 : :
1628 : :
1629 : : //TODO - add sanity check for shape file layers, to include checking to
1630 : : // see if the .shp, .dbf, .shx files are all present and the layer
1631 : : // actually has features
1632 : 68 : bool QgsOgrProvider::isValid() const
1633 : : {
1634 : 68 : return mValid;
1635 : : }
1636 : :
1637 : : // Drivers may be more tolerant than we really wish (e.g. GeoPackage driver
1638 : : // may accept any geometry type)
1639 : 17 : OGRGeometryH QgsOgrProvider::ConvertGeometryIfNecessary( OGRGeometryH hGeom )
1640 : : {
1641 : 17 : if ( !hGeom )
1642 : 0 : return hGeom;
1643 : 17 : OGRwkbGeometryType layerGeomType = mOgrLayer->GetLayerDefn().GetGeomType();
1644 : 17 : OGRwkbGeometryType flattenLayerGeomType = wkbFlatten( layerGeomType );
1645 : 17 : OGRwkbGeometryType geomType = OGR_G_GetGeometryType( hGeom );
1646 : 17 : OGRwkbGeometryType flattenGeomType = wkbFlatten( geomType );
1647 : :
1648 : 17 : if ( flattenLayerGeomType == wkbUnknown || flattenLayerGeomType == flattenGeomType )
1649 : : {
1650 : 4 : return hGeom;
1651 : : }
1652 : 13 : if ( flattenLayerGeomType == wkbMultiPolygon && flattenGeomType == wkbPolygon )
1653 : : {
1654 : 0 : return OGR_G_ForceToMultiPolygon( hGeom );
1655 : : }
1656 : 13 : if ( flattenLayerGeomType == wkbMultiLineString && flattenGeomType == wkbLineString )
1657 : : {
1658 : 0 : return OGR_G_ForceToMultiLineString( hGeom );
1659 : : }
1660 : :
1661 : 13 : return OGR_G_ForceTo( hGeom, layerGeomType, nullptr );
1662 : 17 : }
1663 : :
1664 : 0 : QString QgsOgrProvider::jsonStringValue( const QVariant &value ) const
1665 : : {
1666 : 0 : QString stringValue = QString::fromUtf8( QJsonDocument::fromVariant( value ).toJson().constData() );
1667 : 0 : if ( stringValue.isEmpty() )
1668 : : {
1669 : : //store as string, because it's no valid QJson value
1670 : 0 : stringValue = value.toString();
1671 : 0 : }
1672 : 0 : return stringValue;
1673 : 0 : }
1674 : :
1675 : 2 : bool QgsOgrProvider::addFeaturePrivate( QgsFeature &f, Flags flags, QgsFeatureId incrementalFeatureId )
1676 : : {
1677 : 2 : bool returnValue = true;
1678 : 2 : QgsOgrFeatureDefn &featureDefinition = mOgrLayer->GetLayerDefn();
1679 : 2 : gdal::ogr_feature_unique_ptr feature( featureDefinition.CreateFeature() );
1680 : :
1681 : 2 : if ( f.hasGeometry() )
1682 : : {
1683 : 2 : QByteArray wkb( f.geometry().asWkb() );
1684 : 2 : OGRGeometryH geom = nullptr;
1685 : :
1686 : 2 : if ( !wkb.isEmpty() )
1687 : : {
1688 : 2 : if ( OGR_G_CreateFromWkb( reinterpret_cast<unsigned char *>( const_cast<char *>( wkb.constData() ) ), nullptr, &geom, wkb.length() ) != OGRERR_NONE )
1689 : : {
1690 : 0 : pushError( tr( "OGR error creating wkb for feature %1: %2" ).arg( f.id() ).arg( CPLGetLastErrorMsg() ) );
1691 : 0 : return false;
1692 : : }
1693 : :
1694 : 2 : geom = ConvertGeometryIfNecessary( geom );
1695 : :
1696 : 2 : OGR_F_SetGeometryDirectly( feature.get(), geom );
1697 : 2 : }
1698 : 2 : }
1699 : :
1700 : 2 : QgsAttributes attributes = f.attributes();
1701 : 2 : const QgsFields qgisFields { f.fields() };
1702 : :
1703 : 2 : QgsLocaleNumC l;
1704 : :
1705 : 2 : int qgisAttributeId = ( mFirstFieldIsFid ) ? 1 : 0;
1706 : : // If the first attribute is the FID and the user has set it, then use it
1707 : 2 : if ( mFirstFieldIsFid && attributes.count() > 0 )
1708 : : {
1709 : 0 : QVariant attrFid = attributes.at( 0 );
1710 : 0 : if ( !attrFid.isNull() )
1711 : : {
1712 : 0 : bool ok = false;
1713 : 0 : qlonglong id = attrFid.toLongLong( &ok );
1714 : 0 : if ( ok )
1715 : : {
1716 : 0 : OGR_F_SetFID( feature.get(), static_cast<GIntBig>( id ) );
1717 : 0 : }
1718 : 0 : }
1719 : 0 : }
1720 : :
1721 : : //add possible attribute information
1722 : 5 : for ( int ogrAttributeId = 0; qgisAttributeId < attributes.count(); ++qgisAttributeId, ++ogrAttributeId )
1723 : : {
1724 : : // Skip fields that have no provider origin
1725 : 3 : if ( qgisFields.exists( qgisAttributeId ) && qgisFields.fieldOrigin( qgisAttributeId ) != QgsFields::FieldOrigin::OriginProvider )
1726 : : {
1727 : 0 : qgisAttributeId++;
1728 : 0 : continue;
1729 : : }
1730 : :
1731 : : // don't try to set field from attribute map if it's not present in layer
1732 : 3 : if ( ogrAttributeId >= featureDefinition.GetFieldCount() )
1733 : : {
1734 : 0 : pushError( tr( "Feature has too many attributes (expecting %1, received %2)" ).arg( featureDefinition.GetFieldCount() ).arg( f.attributes().count() ) );
1735 : 0 : continue;
1736 : : }
1737 : :
1738 : : //if(!s.isEmpty())
1739 : : // continue;
1740 : : //
1741 : 3 : OGRFieldDefnH fldDef = featureDefinition.GetFieldDefn( ogrAttributeId );
1742 : 3 : OGRFieldType type = OGR_Fld_GetType( fldDef );
1743 : :
1744 : 3 : QVariant attrVal = attributes.at( qgisAttributeId );
1745 : : // The field value is equal to the default (that might be a provider-side expression)
1746 : 3 : if ( mDefaultValues.contains( qgisAttributeId ) && attrVal.toString() == mDefaultValues.value( qgisAttributeId ) )
1747 : : {
1748 : 0 : OGR_F_UnsetField( feature.get(), ogrAttributeId );
1749 : 0 : }
1750 : 5 : else if ( attrVal.isNull() || ( type != OFTString && attrVal.toString().isEmpty() ) )
1751 : : {
1752 : : // Starting with GDAL 2.2, there are 2 concepts: unset fields and null fields
1753 : : // whereas previously there was only unset fields. For a GeoJSON output,
1754 : : // leaving a field unset will cause it to not appear at all in the output
1755 : : // feature.
1756 : : // When all features of a layer have a field unset, this would cause the
1757 : : // field to not be present at all in the output, and thus on reading to
1758 : : // have disappeared. #16812
1759 : : #ifdef OGRNullMarker
1760 : 1 : OGR_F_SetFieldNull( feature.get(), ogrAttributeId );
1761 : : #else
1762 : : OGR_F_UnsetField( feature.get(), ogrAttId );
1763 : : #endif
1764 : 1 : }
1765 : : else
1766 : : {
1767 : 2 : switch ( type )
1768 : : {
1769 : : case OFTInteger:
1770 : 0 : OGR_F_SetFieldInteger( feature.get(), ogrAttributeId, attrVal.toInt() );
1771 : 0 : break;
1772 : :
1773 : :
1774 : : case OFTInteger64:
1775 : 2 : OGR_F_SetFieldInteger64( feature.get(), ogrAttributeId, attrVal.toLongLong() );
1776 : 2 : break;
1777 : :
1778 : : case OFTReal:
1779 : 0 : OGR_F_SetFieldDouble( feature.get(), ogrAttributeId, attrVal.toDouble() );
1780 : 0 : break;
1781 : :
1782 : : case OFTDate:
1783 : 0 : OGR_F_SetFieldDateTime( feature.get(), ogrAttributeId,
1784 : 0 : attrVal.toDate().year(),
1785 : 0 : attrVal.toDate().month(),
1786 : 0 : attrVal.toDate().day(),
1787 : : 0, 0, 0,
1788 : : 0 );
1789 : 0 : break;
1790 : :
1791 : : case OFTTime:
1792 : 0 : OGR_F_SetFieldDateTime( feature.get(), ogrAttributeId,
1793 : : 0, 0, 0,
1794 : 0 : attrVal.toTime().hour(),
1795 : 0 : attrVal.toTime().minute(),
1796 : 0 : attrVal.toTime().second(),
1797 : : 0 );
1798 : 0 : break;
1799 : :
1800 : : case OFTDateTime:
1801 : 0 : OGR_F_SetFieldDateTime( feature.get(), ogrAttributeId,
1802 : 0 : attrVal.toDateTime().date().year(),
1803 : 0 : attrVal.toDateTime().date().month(),
1804 : 0 : attrVal.toDateTime().date().day(),
1805 : 0 : attrVal.toDateTime().time().hour(),
1806 : 0 : attrVal.toDateTime().time().minute(),
1807 : 0 : attrVal.toDateTime().time().second(),
1808 : : 0 );
1809 : 0 : break;
1810 : :
1811 : : case OFTString:
1812 : : {
1813 : 0 : QString stringValue;
1814 : :
1815 : 0 : if ( OGR_Fld_GetSubType( fldDef ) == OFSTJSON )
1816 : 0 : stringValue = jsonStringValue( attrVal );
1817 : : else
1818 : : {
1819 : 0 : stringValue = attrVal.toString();
1820 : : }
1821 : 0 : QgsDebugMsgLevel( QStringLiteral( "Writing string attribute %1 with %2, encoding %3" )
1822 : : .arg( qgisAttributeId )
1823 : : .arg( attrVal.toString(),
1824 : : textEncoding()->name().data() ), 3 );
1825 : 0 : OGR_F_SetFieldString( feature.get(), ogrAttributeId, textEncoding()->fromUnicode( stringValue ).constData() );
1826 : : break;
1827 : 0 : }
1828 : : case OFTBinary:
1829 : : {
1830 : 0 : const QByteArray ba = attrVal.toByteArray();
1831 : 0 : OGR_F_SetFieldBinary( feature.get(), ogrAttributeId, ba.size(), const_cast< GByte * >( reinterpret_cast< const GByte * >( ba.data() ) ) );
1832 : : break;
1833 : 0 : }
1834 : :
1835 : : case OFTStringList:
1836 : : {
1837 : 0 : QStringList list = attrVal.toStringList();
1838 : 0 : int count = list.count();
1839 : 0 : char **lst = new char *[count + 1];
1840 : 0 : if ( count > 0 )
1841 : : {
1842 : 0 : int pos = 0;
1843 : 0 : for ( QString string : list )
1844 : : {
1845 : 0 : lst[pos] = textEncoding()->fromUnicode( string ).data();
1846 : 0 : pos++;
1847 : 0 : }
1848 : 0 : }
1849 : 0 : lst[count] = nullptr;
1850 : 0 : OGR_F_SetFieldStringList( feature.get(), ogrAttributeId, lst );
1851 : : break;
1852 : 0 : }
1853 : :
1854 : : default:
1855 : 0 : QgsMessageLog::logMessage( tr( "type %1 for attribute %2 not found" ).arg( type ).arg( qgisAttributeId ), tr( "OGR" ) );
1856 : 0 : break;
1857 : : }
1858 : : }
1859 : 3 : }
1860 : :
1861 : 2 : if ( mOgrLayer->CreateFeature( feature.get() ) != OGRERR_NONE )
1862 : : {
1863 : 0 : pushError( tr( "OGR error creating feature %1: %2" ).arg( f.id() ).arg( CPLGetLastErrorMsg() ) );
1864 : 0 : returnValue = false;
1865 : 0 : }
1866 : : else
1867 : : {
1868 : 2 : if ( !( flags & QgsFeatureSink::FastInsert ) )
1869 : : {
1870 : 2 : QgsFeatureId id = static_cast<QgsFeatureId>( OGR_F_GetFID( feature.get() ) );
1871 : 2 : if ( id >= 0 )
1872 : : {
1873 : 2 : f.setId( id );
1874 : :
1875 : 2 : if ( mFirstFieldIsFid && attributes.count() > 0 )
1876 : : {
1877 : 0 : f.setAttribute( 0, id );
1878 : 0 : }
1879 : 2 : }
1880 : 0 : else if ( incrementalFeatureId >= 0 )
1881 : : {
1882 : 0 : f.setId( incrementalFeatureId );
1883 : 0 : }
1884 : 2 : }
1885 : : }
1886 : :
1887 : 2 : return returnValue;
1888 : 2 : }
1889 : :
1890 : :
1891 : 2 : bool QgsOgrProvider::addFeatures( QgsFeatureList &flist, Flags flags )
1892 : : {
1893 : 2 : QgsCPLHTTPFetchOverrider oCPLHTTPFetcher( mAuthCfg );
1894 : 4 : QgsSetCPLHTTPFetchOverriderInitiatorClass( oCPLHTTPFetcher, QStringLiteral( "QgsOgrProvider" ) );
1895 : :
1896 : 2 : if ( !doInitialActionsForEdition() )
1897 : 0 : return false;
1898 : :
1899 : 2 : setRelevantFields( true, attributeIndexes() );
1900 : :
1901 : 2 : const bool inTransaction = startTransaction();
1902 : :
1903 : 2 : QgsFeatureId incrementalFeatureId = -1;
1904 : 4 : if ( !( flags & QgsFeatureSink::FastInsert ) &&
1905 : 2 : ( mGDALDriverName == QLatin1String( "CSV" ) || mGDALDriverName == QLatin1String( "XLSX" ) || mGDALDriverName == QLatin1String( "ODS" ) ) )
1906 : : {
1907 : 0 : QMutex *mutex = nullptr;
1908 : 0 : OGRLayerH layer = mOgrOrigLayer->getHandleAndMutex( mutex );
1909 : : {
1910 : 0 : QMutexLocker locker( mutex );
1911 : :
1912 : 0 : if ( !mSubsetString.isEmpty() )
1913 : 0 : OGR_L_SetAttributeFilter( layer, nullptr );
1914 : :
1915 : 0 : incrementalFeatureId = static_cast< QgsFeatureId >( OGR_L_GetFeatureCount( layer, false ) ) + 1;
1916 : :
1917 : 0 : if ( !mSubsetString.isEmpty() )
1918 : 0 : OGR_L_SetAttributeFilter( layer, textEncoding()->fromUnicode( mSubsetString ).constData() );
1919 : 0 : }
1920 : 0 : }
1921 : :
1922 : 2 : bool returnvalue = true;
1923 : 4 : for ( QgsFeatureList::iterator it = flist.begin(); it != flist.end(); ++it )
1924 : : {
1925 : 2 : if ( !addFeaturePrivate( *it, flags, incrementalFeatureId ) )
1926 : : {
1927 : 0 : returnvalue = false;
1928 : 0 : }
1929 : 2 : if ( incrementalFeatureId >= 0 )
1930 : 0 : incrementalFeatureId++;
1931 : 2 : }
1932 : :
1933 : 2 : if ( inTransaction )
1934 : : {
1935 : 0 : if ( returnvalue )
1936 : 0 : returnvalue = commitTransaction();
1937 : : else
1938 : 0 : rollbackTransaction();
1939 : 0 : }
1940 : :
1941 : 2 : if ( !syncToDisc() )
1942 : : {
1943 : 0 : returnvalue = false;
1944 : 0 : }
1945 : :
1946 : 2 : if ( mFeaturesCounted != QgsVectorDataProvider::Uncounted &&
1947 : 2 : mFeaturesCounted != QgsVectorDataProvider::UnknownCount )
1948 : : {
1949 : 2 : if ( returnvalue )
1950 : 2 : mFeaturesCounted += flist.size();
1951 : : else
1952 : 0 : recalculateFeatureCount();
1953 : 2 : }
1954 : :
1955 : 2 : if ( returnvalue )
1956 : 2 : clearMinMaxCache();
1957 : :
1958 : 2 : if ( mTransaction )
1959 : 0 : mTransaction->dirtyLastSavePoint();
1960 : :
1961 : 2 : return returnvalue;
1962 : 2 : }
1963 : :
1964 : 0 : bool QgsOgrProvider::addAttributeOGRLevel( const QgsField &field, bool &ignoreErrorOut )
1965 : : {
1966 : 0 : ignoreErrorOut = false;
1967 : :
1968 : : OGRFieldType type;
1969 : :
1970 : 0 : switch ( field.type() )
1971 : : {
1972 : : case QVariant::Int:
1973 : : case QVariant::Bool:
1974 : 0 : type = OFTInteger;
1975 : 0 : break;
1976 : : case QVariant::LongLong:
1977 : : {
1978 : 0 : const char *pszDataTypes = GDALGetMetadataItem( mOgrLayer->driver(), GDAL_DMD_CREATIONFIELDDATATYPES, nullptr );
1979 : 0 : if ( pszDataTypes && strstr( pszDataTypes, "Integer64" ) )
1980 : 0 : type = OFTInteger64;
1981 : : else
1982 : : {
1983 : 0 : type = OFTReal;
1984 : : }
1985 : 0 : break;
1986 : : }
1987 : : case QVariant::Double:
1988 : 0 : type = OFTReal;
1989 : 0 : break;
1990 : : case QVariant::Date:
1991 : 0 : type = OFTDate;
1992 : 0 : break;
1993 : : case QVariant::Time:
1994 : 0 : type = OFTTime;
1995 : 0 : break;
1996 : : case QVariant::DateTime:
1997 : 0 : type = OFTDateTime;
1998 : 0 : break;
1999 : : case QVariant::String:
2000 : 0 : type = OFTString;
2001 : 0 : break;
2002 : : case QVariant::ByteArray:
2003 : 0 : type = OFTBinary;
2004 : 0 : break;
2005 : : case QVariant::Map:
2006 : 0 : type = OFTString;
2007 : 0 : break;
2008 : : case QVariant::List:
2009 : : // only string list supported at the moment, fall through to default for other types
2010 : 0 : if ( field.subType() == QVariant::String )
2011 : : {
2012 : 0 : type = OFTStringList;
2013 : 0 : break;
2014 : : }
2015 : : //intentional fall-through
2016 : : FALLTHROUGH
2017 : :
2018 : : default:
2019 : 0 : pushError( tr( "type %1 for field %2 not found" ).arg( field.typeName(), field.name() ) );
2020 : 0 : ignoreErrorOut = true;
2021 : 0 : return false;
2022 : : }
2023 : :
2024 : 0 : gdal::ogr_field_def_unique_ptr fielddefn( OGR_Fld_Create( textEncoding()->fromUnicode( field.name() ).constData(), type ) );
2025 : 0 : int width = field.length();
2026 : : // Increase width by 1 for OFTReal to make room for the decimal point
2027 : 0 : if ( type == OFTReal && field.precision() )
2028 : 0 : width += 1;
2029 : 0 : OGR_Fld_SetWidth( fielddefn.get(), width );
2030 : 0 : OGR_Fld_SetPrecision( fielddefn.get(), field.precision() );
2031 : :
2032 : 0 : switch ( field.type() )
2033 : : {
2034 : : case QVariant::Bool:
2035 : 0 : OGR_Fld_SetSubType( fielddefn.get(), OFSTBoolean );
2036 : 0 : break;
2037 : : case QVariant::Map:
2038 : 0 : OGR_Fld_SetSubType( fielddefn.get(), OFSTJSON );
2039 : 0 : break;
2040 : : default:
2041 : 0 : break;
2042 : : }
2043 : :
2044 : 0 : if ( mOgrLayer->CreateField( fielddefn.get(), true ) != OGRERR_NONE )
2045 : : {
2046 : 0 : pushError( tr( "OGR error creating field %1: %2" ).arg( field.name(), CPLGetLastErrorMsg() ) );
2047 : 0 : return false;
2048 : : }
2049 : 0 : return true;
2050 : 0 : }
2051 : :
2052 : 0 : bool QgsOgrProvider::addAttributes( const QList<QgsField> &attributes )
2053 : : {
2054 : 0 : QgsCPLHTTPFetchOverrider oCPLHTTPFetcher( mAuthCfg );
2055 : 0 : QgsSetCPLHTTPFetchOverriderInitiatorClass( oCPLHTTPFetcher, QStringLiteral( "QgsOgrProvider" ) );
2056 : :
2057 : 0 : if ( !doInitialActionsForEdition() )
2058 : 0 : return false;
2059 : :
2060 : 0 : if ( mGDALDriverName == QLatin1String( "MapInfo File" ) )
2061 : : {
2062 : : // adding attributes in mapinfo requires to be able to delete the .dat file
2063 : : // so drop any cached connections.
2064 : 0 : QgsOgrConnPool::instance()->invalidateConnections( QgsOgrProviderUtils::connectionPoolId( dataSourceUri( true ), mShareSameDatasetAmongLayers ) );
2065 : 0 : }
2066 : :
2067 : 0 : bool returnvalue = true;
2068 : :
2069 : 0 : QMap< QString, QgsField > mapFieldNameToOriginalField;
2070 : :
2071 : 0 : for ( const auto &field : attributes )
2072 : : {
2073 : 0 : mapFieldNameToOriginalField[ field.name()] = field;
2074 : :
2075 : 0 : bool ignoreErrorOut = false;
2076 : 0 : if ( !addAttributeOGRLevel( field, ignoreErrorOut ) )
2077 : : {
2078 : 0 : returnvalue = false;
2079 : 0 : if ( !ignoreErrorOut )
2080 : : {
2081 : 0 : break;
2082 : : }
2083 : 0 : }
2084 : : }
2085 : :
2086 : : // Backup existing fields. We need them to 'restore' field type, length, precision
2087 : 0 : QgsFields oldFields = mAttributeFields;
2088 : :
2089 : 0 : loadFields();
2090 : :
2091 : : // The check in QgsVectorLayerEditBuffer::commitChanges() is questionable with
2092 : : // real-world drivers that might only be able to satisfy request only partially.
2093 : : // So to avoid erroring out, patch field type, width and precision to match
2094 : : // what was requested.
2095 : : // For example in case of Integer64->Real mapping so that QVariant::LongLong is
2096 : : // still returned to the caller
2097 : : // Or if a field width was specified but not strictly enforced by the driver (#15614)
2098 : 0 : for ( QMap< QString, QgsField >::const_iterator it = mapFieldNameToOriginalField.constBegin();
2099 : 0 : it != mapFieldNameToOriginalField.constEnd(); ++it )
2100 : : {
2101 : 0 : int idx = mAttributeFields.lookupField( it.key() );
2102 : 0 : if ( idx >= 0 )
2103 : : {
2104 : 0 : mAttributeFields[ idx ].setType( it->type() );
2105 : 0 : mAttributeFields[ idx ].setLength( it->length() );
2106 : 0 : mAttributeFields[ idx ].setPrecision( it->precision() );
2107 : 0 : }
2108 : 0 : }
2109 : :
2110 : : // Restore field type, length, precision of existing fields as well
2111 : : // We need that in scenarios where the user adds a int field with length != 0
2112 : : // in a editing session, and repeat that again in another editing session
2113 : : // Without the below hack, the length of the first added field would have
2114 : : // been reset to zero, and QgsVectorLayerEditBuffer::commitChanges() would
2115 : : // error out because of this.
2116 : : // See https://github.com/qgis/QGIS/issues/26840
2117 : 0 : for ( auto field : oldFields )
2118 : : {
2119 : 0 : int idx = mAttributeFields.lookupField( field.name() );
2120 : 0 : if ( idx >= 0 )
2121 : : {
2122 : 0 : mAttributeFields[ idx ].setType( field.type() );
2123 : 0 : mAttributeFields[ idx ].setLength( field.length() );
2124 : 0 : mAttributeFields[ idx ].setPrecision( field.precision() );
2125 : 0 : }
2126 : 0 : }
2127 : :
2128 : 0 : if ( mTransaction )
2129 : 0 : mTransaction->dirtyLastSavePoint();
2130 : :
2131 : 0 : return returnvalue;
2132 : 0 : }
2133 : :
2134 : 0 : bool QgsOgrProvider::deleteAttributes( const QgsAttributeIds &attributes )
2135 : : {
2136 : 0 : if ( !doInitialActionsForEdition() )
2137 : 0 : return false;
2138 : :
2139 : 0 : bool res = true;
2140 : 0 : QList<int> attrsLst = qgis::setToList( attributes );
2141 : : // sort in descending order
2142 : 0 : std::sort( attrsLst.begin(), attrsLst.end(), std::greater<int>() );
2143 : 0 : const auto constAttrsLst = attrsLst;
2144 : 0 : for ( int attr : constAttrsLst )
2145 : : {
2146 : 0 : if ( mFirstFieldIsFid )
2147 : : {
2148 : 0 : if ( attr == 0 )
2149 : : {
2150 : 0 : pushError( tr( "Cannot delete feature id column" ) );
2151 : 0 : res = false;
2152 : 0 : break;
2153 : : }
2154 : : else
2155 : : {
2156 : 0 : --attr;
2157 : : }
2158 : 0 : }
2159 : 0 : if ( mOgrLayer->DeleteField( attr ) != OGRERR_NONE )
2160 : : {
2161 : 0 : pushError( tr( "OGR error deleting field %1: %2" ).arg( attr ).arg( CPLGetLastErrorMsg() ) );
2162 : 0 : res = false;
2163 : 0 : }
2164 : : }
2165 : 0 : loadFields();
2166 : :
2167 : 0 : if ( mTransaction )
2168 : 0 : mTransaction->dirtyLastSavePoint();
2169 : :
2170 : 0 : return res;
2171 : 0 : }
2172 : :
2173 : 0 : bool QgsOgrProvider::renameAttributes( const QgsFieldNameMap &renamedAttributes )
2174 : : {
2175 : 0 : QgsCPLHTTPFetchOverrider oCPLHTTPFetcher( mAuthCfg );
2176 : 0 : QgsSetCPLHTTPFetchOverriderInitiatorClass( oCPLHTTPFetcher, QStringLiteral( "QgsOgrProvider" ) );
2177 : :
2178 : 0 : if ( !doInitialActionsForEdition() )
2179 : 0 : return false;
2180 : :
2181 : 0 : QgsFieldNameMap::const_iterator renameIt = renamedAttributes.constBegin();
2182 : 0 : bool result = true;
2183 : 0 : for ( ; renameIt != renamedAttributes.constEnd(); ++renameIt )
2184 : : {
2185 : 0 : int fieldIndex = renameIt.key();
2186 : 0 : if ( fieldIndex < 0 || fieldIndex >= mAttributeFields.count() )
2187 : : {
2188 : 0 : pushError( tr( "Invalid attribute index" ) );
2189 : 0 : result = false;
2190 : 0 : continue;
2191 : : }
2192 : 0 : if ( mAttributeFields.indexFromName( renameIt.value() ) >= 0 )
2193 : : {
2194 : : //field name already in use
2195 : 0 : pushError( tr( "Error renaming field %1: name '%2' already exists" ).arg( fieldIndex ).arg( renameIt.value() ) );
2196 : 0 : result = false;
2197 : 0 : continue;
2198 : : }
2199 : 0 : int ogrFieldIndex = fieldIndex;
2200 : 0 : if ( mFirstFieldIsFid )
2201 : : {
2202 : 0 : ogrFieldIndex -= 1;
2203 : 0 : if ( ogrFieldIndex < 0 )
2204 : : {
2205 : 0 : pushError( tr( "Invalid attribute index" ) );
2206 : 0 : result = false;
2207 : 0 : continue;
2208 : : }
2209 : 0 : }
2210 : :
2211 : : //type does not matter, it will not be used
2212 : 0 : gdal::ogr_field_def_unique_ptr fld( OGR_Fld_Create( textEncoding()->fromUnicode( renameIt.value() ), OFTReal ) );
2213 : 0 : if ( mOgrLayer->AlterFieldDefn( ogrFieldIndex, fld.get(), ALTER_NAME_FLAG ) != OGRERR_NONE )
2214 : : {
2215 : 0 : pushError( tr( "OGR error renaming field %1: %2" ).arg( fieldIndex ).arg( CPLGetLastErrorMsg() ) );
2216 : 0 : result = false;
2217 : 0 : }
2218 : 0 : }
2219 : 0 : loadFields();
2220 : :
2221 : 0 : if ( mTransaction )
2222 : 0 : mTransaction->dirtyLastSavePoint();
2223 : :
2224 : 0 : return result;
2225 : 0 : }
2226 : :
2227 : 40 : bool QgsOgrProvider::startTransaction()
2228 : : {
2229 : 40 : bool inTransaction = false;
2230 : 40 : if ( mTransaction == nullptr && mOgrLayer->TestCapability( OLCTransactions ) )
2231 : : {
2232 : : // A transaction might already be active, so be robust on failed
2233 : : // StartTransaction.
2234 : 0 : CPLPushErrorHandler( CPLQuietErrorHandler );
2235 : 0 : inTransaction = ( mOgrLayer->StartTransaction() == OGRERR_NONE );
2236 : 0 : CPLPopErrorHandler();
2237 : 0 : }
2238 : 40 : return inTransaction;
2239 : : }
2240 : :
2241 : :
2242 : 0 : bool QgsOgrProvider::commitTransaction()
2243 : : {
2244 : 0 : if ( mOgrLayer->CommitTransaction() != OGRERR_NONE )
2245 : : {
2246 : 0 : pushError( tr( "OGR error committing transaction: %1" ).arg( CPLGetLastErrorMsg() ) );
2247 : 0 : return false;
2248 : : }
2249 : 0 : return true;
2250 : 0 : }
2251 : :
2252 : :
2253 : 0 : bool QgsOgrProvider::rollbackTransaction()
2254 : : {
2255 : 0 : if ( mOgrLayer->RollbackTransaction() != OGRERR_NONE )
2256 : : {
2257 : 0 : pushError( tr( "OGR error rolling back transaction: %1" ).arg( CPLGetLastErrorMsg() ) );
2258 : 0 : return false;
2259 : : }
2260 : 0 : return true;
2261 : 0 : }
2262 : :
2263 : 187 : bool QgsOgrProvider::_setSubsetString( const QString &theSQL, bool updateFeatureCount, bool updateCapabilities, bool hasExistingRef )
2264 : : {
2265 : 187 : QgsCPLErrorHandler handler;
2266 : :
2267 : 187 : QgsCPLHTTPFetchOverrider oCPLHTTPFetcher( mAuthCfg );
2268 : 374 : QgsSetCPLHTTPFetchOverriderInitiatorClass( oCPLHTTPFetcher, QStringLiteral( "QgsOgrProvider" ) );
2269 : :
2270 : 187 : if ( !mOgrOrigLayer )
2271 : 0 : return false;
2272 : :
2273 : 187 : if ( theSQL == mSubsetString && mFeaturesCounted != QgsVectorDataProvider::Uncounted )
2274 : 7 : return true;
2275 : :
2276 : 180 : const bool subsetStringHasChanged { theSQL != mSubsetString };
2277 : :
2278 : 180 : if ( !theSQL.isEmpty() )
2279 : : {
2280 : 0 : QMutex *mutex = nullptr;
2281 : 0 : OGRLayerH layer = mOgrOrigLayer->getHandleAndMutex( mutex );
2282 : 0 : GDALDatasetH ds = mOgrOrigLayer->getDatasetHandleAndMutex( mutex );
2283 : : OGRLayerH subsetLayerH;
2284 : : {
2285 : 0 : QMutexLocker locker( mutex );
2286 : 0 : subsetLayerH = QgsOgrProviderUtils::setSubsetString( layer, ds, textEncoding(), theSQL );
2287 : 0 : }
2288 : 0 : if ( !subsetLayerH )
2289 : : {
2290 : 0 : pushError( tr( "OGR[%1] error %2: %3" ).arg( CPLGetLastErrorType() ).arg( CPLGetLastErrorNo() ).arg( CPLGetLastErrorMsg() ) );
2291 : 0 : return false;
2292 : : }
2293 : 0 : if ( layer != subsetLayerH )
2294 : : {
2295 : 0 : mOgrSqlLayer = QgsOgrProviderUtils::getSqlLayer( mOgrOrigLayer.get(), subsetLayerH, theSQL );
2296 : : Q_ASSERT( mOgrSqlLayer.get() );
2297 : 0 : mOgrLayer = mOgrSqlLayer.get();
2298 : 0 : }
2299 : : else
2300 : : {
2301 : 0 : mOgrSqlLayer.reset();
2302 : 0 : mOgrLayer = mOgrOrigLayer.get();
2303 : : }
2304 : 0 : }
2305 : : else
2306 : : {
2307 : 180 : mOgrSqlLayer.reset();
2308 : 180 : mOgrLayer = mOgrOrigLayer.get();
2309 : 180 : QMutex *mutex = nullptr;
2310 : 180 : OGRLayerH layer = mOgrOrigLayer->getHandleAndMutex( mutex );
2311 : : {
2312 : 180 : QMutexLocker locker( mutex );
2313 : 180 : OGR_L_SetAttributeFilter( layer, nullptr );
2314 : 180 : }
2315 : : }
2316 : 180 : mSubsetString = theSQL;
2317 : :
2318 : 180 : QVariantMap parts;
2319 : 360 : parts.insert( QStringLiteral( "path" ), mFilePath );
2320 : :
2321 : 180 : if ( !mLayerName.isNull() )
2322 : : {
2323 : 0 : parts.insert( QStringLiteral( "layerName" ), mLayerName );
2324 : 0 : }
2325 : 180 : else if ( mIsSubLayer && mLayerIndex >= 0 )
2326 : : {
2327 : 0 : parts.insert( QStringLiteral( "layerId" ), mLayerIndex );
2328 : 0 : }
2329 : :
2330 : 180 : if ( !mSubsetString.isEmpty() )
2331 : : {
2332 : 0 : parts.insert( QStringLiteral( "subset" ), mSubsetString );
2333 : 0 : }
2334 : :
2335 : 180 : if ( mOgrGeometryTypeFilter != wkbUnknown )
2336 : : {
2337 : 0 : parts.insert( QStringLiteral( "geometryType" ), ogrWkbGeometryTypeName( mOgrGeometryTypeFilter ) );
2338 : 0 : }
2339 : :
2340 : 180 : if ( !mOpenOptions.isEmpty() )
2341 : : {
2342 : 0 : parts.insert( QStringLiteral( "openOptions" ), mOpenOptions );
2343 : 0 : }
2344 : :
2345 : 180 : QString uri = QgsOgrProviderMetadata().encodeUri( parts );
2346 : 180 : if ( uri != dataSourceUri() )
2347 : : {
2348 : 0 : if ( hasExistingRef )
2349 : 0 : QgsOgrConnPool::instance()->unref( QgsOgrProviderUtils::connectionPoolId( dataSourceUri( true ), mShareSameDatasetAmongLayers ) );
2350 : 0 : setDataSourceUri( uri );
2351 : 0 : if ( hasExistingRef )
2352 : 0 : QgsOgrConnPool::instance()->ref( QgsOgrProviderUtils::connectionPoolId( dataSourceUri( true ), mShareSameDatasetAmongLayers ) );
2353 : 0 : }
2354 : :
2355 : 180 : mOgrLayer->ResetReading();
2356 : :
2357 : 180 : mRefreshFeatureCount = updateFeatureCount;
2358 : :
2359 : : // check the validity of the layer if subset string has changed
2360 : 180 : if ( subsetStringHasChanged )
2361 : : {
2362 : 0 : loadFields();
2363 : 0 : }
2364 : :
2365 : 180 : invalidateCachedExtent( false );
2366 : :
2367 : : // Changing the filter may change capabilities
2368 : 180 : if ( updateCapabilities )
2369 : 0 : computeCapabilities();
2370 : :
2371 : 180 : emit dataChanged();
2372 : :
2373 : 180 : return true;
2374 : :
2375 : 187 : }
2376 : :
2377 : :
2378 : 15 : bool QgsOgrProvider::changeAttributeValues( const QgsChangedAttributesMap &attr_map )
2379 : : {
2380 : 15 : QgsCPLHTTPFetchOverrider oCPLHTTPFetcher( mAuthCfg );
2381 : 30 : QgsSetCPLHTTPFetchOverriderInitiatorClass( oCPLHTTPFetcher, QStringLiteral( "QgsOgrProvider" ) );
2382 : :
2383 : 15 : if ( !doInitialActionsForEdition() )
2384 : 0 : return false;
2385 : :
2386 : 15 : if ( attr_map.isEmpty() )
2387 : 0 : return true;
2388 : :
2389 : 15 : bool returnValue = true;
2390 : :
2391 : 15 : clearMinMaxCache();
2392 : :
2393 : 15 : setRelevantFields( true, attributeIndexes() );
2394 : :
2395 : 15 : const bool inTransaction = startTransaction();
2396 : :
2397 : : // Some drivers may need to call ResetReading() after GetFeature(), such
2398 : : // as GPKG in GDAL < 2.3.0 to avoid letting the database in a locked state.
2399 : : // But this is undesirable in general, so don't do this when we know that
2400 : : // we don't need to.
2401 : 15 : bool mayNeedResetReadingAfterGetFeature = true;
2402 : 15 : if ( mGDALDriverName == QLatin1String( "ESRI Shapefile" ) )
2403 : : {
2404 : 15 : mayNeedResetReadingAfterGetFeature = false;
2405 : 15 : }
2406 : 0 : else if ( mGDALDriverName == QLatin1String( "GPKG" ) )
2407 : : {
2408 : 0 : mayNeedResetReadingAfterGetFeature = false;
2409 : 0 : }
2410 : :
2411 : 30 : for ( QgsChangedAttributesMap::const_iterator it = attr_map.begin(); it != attr_map.end(); ++it )
2412 : : {
2413 : 15 : QgsFeatureId fid = it.key();
2414 : :
2415 : 15 : const QgsAttributeMap &attr = it.value();
2416 : 15 : if ( attr.isEmpty() )
2417 : 0 : continue;
2418 : :
2419 : 15 : gdal::ogr_feature_unique_ptr of( mOgrLayer->GetFeature( FID_TO_NUMBER( fid ) ) );
2420 : 15 : if ( !of )
2421 : : {
2422 : 0 : pushError( tr( "Feature %1 for attribute update not found." ).arg( fid ) );
2423 : 0 : continue;
2424 : : }
2425 : :
2426 : 15 : if ( mayNeedResetReadingAfterGetFeature )
2427 : : {
2428 : 0 : mOgrLayer->ResetReading();
2429 : 0 : }
2430 : :
2431 : 15 : QgsLocaleNumC l;
2432 : :
2433 : 39 : for ( QgsAttributeMap::const_iterator it2 = attr.begin(); it2 != attr.end(); ++it2 )
2434 : : {
2435 : 24 : int f = it2.key();
2436 : 24 : if ( mFirstFieldIsFid )
2437 : : {
2438 : 0 : if ( f == 0 )
2439 : : {
2440 : 0 : if ( it2->toLongLong() != fid )
2441 : : {
2442 : 0 : pushError( tr( "Changing feature id of feature %1 is not allowed." ).arg( fid ) );
2443 : 0 : continue;
2444 : : }
2445 : 0 : }
2446 : : else
2447 : : {
2448 : 0 : --f;
2449 : : }
2450 : 0 : }
2451 : :
2452 : 24 : OGRFieldDefnH fd = OGR_F_GetFieldDefnRef( of.get(), f );
2453 : 24 : if ( !fd )
2454 : : {
2455 : 0 : pushError( tr( "Field %1 of feature %2 doesn't exist." ).arg( f ).arg( fid ) );
2456 : 0 : continue;
2457 : : }
2458 : :
2459 : 24 : OGRFieldType type = OGR_Fld_GetType( fd );
2460 : :
2461 : 42 : if ( it2->isNull() || ( type != OFTString && it2->toString().isEmpty() ) )
2462 : : {
2463 : : // Starting with GDAL 2.2, there are 2 concepts: unset fields and null fields
2464 : : // whereas previously there was only unset fields. For a GeoJSON output,
2465 : : // leaving a field unset will cause it to not appear at all in the output
2466 : : // feature.
2467 : : // When all features of a layer have a field unset, this would cause the
2468 : : // field to not be present at all in the output, and thus on reading to
2469 : : // have disappeared. #16812
2470 : : #ifdef OGRNullMarker
2471 : 6 : OGR_F_SetFieldNull( of.get(), f );
2472 : : #else
2473 : : OGR_F_UnsetField( of.get(), f );
2474 : : #endif
2475 : 6 : }
2476 : : else
2477 : : {
2478 : :
2479 : 18 : switch ( type )
2480 : : {
2481 : : case OFTInteger:
2482 : 3 : OGR_F_SetFieldInteger( of.get(), f, it2->toInt() );
2483 : 3 : break;
2484 : : case OFTInteger64:
2485 : 15 : OGR_F_SetFieldInteger64( of.get(), f, it2->toLongLong() );
2486 : 15 : break;
2487 : : case OFTReal:
2488 : 0 : OGR_F_SetFieldDouble( of.get(), f, it2->toDouble() );
2489 : 0 : break;
2490 : : case OFTDate:
2491 : 0 : OGR_F_SetFieldDateTime( of.get(), f,
2492 : 0 : it2->toDate().year(),
2493 : 0 : it2->toDate().month(),
2494 : 0 : it2->toDate().day(),
2495 : : 0, 0, 0,
2496 : : 0 );
2497 : 0 : break;
2498 : : case OFTTime:
2499 : 0 : OGR_F_SetFieldDateTime( of.get(), f,
2500 : : 0, 0, 0,
2501 : 0 : it2->toTime().hour(),
2502 : 0 : it2->toTime().minute(),
2503 : 0 : it2->toTime().second(),
2504 : : 0 );
2505 : 0 : break;
2506 : : case OFTDateTime:
2507 : 0 : OGR_F_SetFieldDateTime( of.get(), f,
2508 : 0 : it2->toDateTime().date().year(),
2509 : 0 : it2->toDateTime().date().month(),
2510 : 0 : it2->toDateTime().date().day(),
2511 : 0 : it2->toDateTime().time().hour(),
2512 : 0 : it2->toDateTime().time().minute(),
2513 : 0 : it2->toDateTime().time().second(),
2514 : : 0 );
2515 : 0 : break;
2516 : : case OFTString:
2517 : : {
2518 : 0 : QString stringValue;
2519 : 0 : if ( OGR_Fld_GetSubType( fd ) == OFSTJSON )
2520 : 0 : stringValue = jsonStringValue( it2.value() );
2521 : : else
2522 : 0 : stringValue = it2->toString();
2523 : 0 : OGR_F_SetFieldString( of.get(), f, textEncoding()->fromUnicode( stringValue ).constData() );
2524 : : break;
2525 : 0 : }
2526 : :
2527 : : case OFTBinary:
2528 : : {
2529 : 0 : const QByteArray ba = it2->toByteArray();
2530 : 0 : OGR_F_SetFieldBinary( of.get(), f, ba.size(), const_cast< GByte * >( reinterpret_cast< const GByte * >( ba.data() ) ) );
2531 : : break;
2532 : 0 : }
2533 : :
2534 : : case OFTStringList:
2535 : : {
2536 : 0 : QStringList list = it2->toStringList();
2537 : 0 : int count = list.count();
2538 : 0 : char **lst = new char *[count + 1];
2539 : 0 : if ( count > 0 )
2540 : : {
2541 : 0 : int pos = 0;
2542 : 0 : for ( QString string : list )
2543 : : {
2544 : 0 : lst[pos] = textEncoding()->fromUnicode( string ).data();
2545 : 0 : pos++;
2546 : 0 : }
2547 : 0 : }
2548 : 0 : lst[count] = nullptr;
2549 : 0 : OGR_F_SetFieldStringList( of.get(), f, lst );
2550 : : break;
2551 : 0 : }
2552 : :
2553 : : default:
2554 : 0 : pushError( tr( "Type %1 of attribute %2 of feature %3 unknown." ).arg( type ).arg( fid ).arg( f ) );
2555 : 0 : break;
2556 : : }
2557 : : }
2558 : 24 : }
2559 : :
2560 : 15 : if ( mOgrLayer->SetFeature( of.get() ) != OGRERR_NONE )
2561 : : {
2562 : 0 : pushError( tr( "OGR error setting feature %1: %2" ).arg( fid ).arg( CPLGetLastErrorMsg() ) );
2563 : 0 : returnValue = false;
2564 : 0 : }
2565 : 15 : }
2566 : :
2567 : 15 : if ( inTransaction )
2568 : : {
2569 : 0 : if ( returnValue )
2570 : 0 : returnValue = commitTransaction();
2571 : : else
2572 : 0 : rollbackTransaction();
2573 : 0 : }
2574 : :
2575 : 15 : if ( mTransaction )
2576 : 0 : mTransaction->dirtyLastSavePoint();
2577 : :
2578 : 15 : if ( !syncToDisc() )
2579 : : {
2580 : 0 : pushError( tr( "OGR error syncing to disk: %1" ).arg( CPLGetLastErrorMsg() ) );
2581 : 0 : }
2582 : 15 : return returnValue;
2583 : 15 : }
2584 : :
2585 : 15 : bool QgsOgrProvider::changeGeometryValues( const QgsGeometryMap &geometry_map )
2586 : : {
2587 : 15 : QgsCPLHTTPFetchOverrider oCPLHTTPFetcher( mAuthCfg );
2588 : 30 : QgsSetCPLHTTPFetchOverriderInitiatorClass( oCPLHTTPFetcher, QStringLiteral( "QgsOgrProvider" ) );
2589 : :
2590 : 15 : if ( !doInitialActionsForEdition() )
2591 : 0 : return false;
2592 : :
2593 : 15 : setRelevantFields( true, attributeIndexes() );
2594 : :
2595 : 15 : const bool inTransaction = startTransaction();
2596 : :
2597 : : // Some drivers may need to call ResetReading() after GetFeature(), such
2598 : : // as GPKG in GDAL < 2.3.0 to avoid letting the database in a locked state.
2599 : : // But this is undesirable in general, so don't do this when we know that
2600 : : // we don't need to.
2601 : 15 : bool mayNeedResetReadingAfterGetFeature = true;
2602 : 15 : if ( mGDALDriverName == QLatin1String( "ESRI Shapefile" ) )
2603 : : {
2604 : 15 : mayNeedResetReadingAfterGetFeature = false;
2605 : 15 : }
2606 : 0 : else if ( mGDALDriverName == QLatin1String( "GPKG" ) )
2607 : : {
2608 : 0 : mayNeedResetReadingAfterGetFeature = false;
2609 : 0 : }
2610 : :
2611 : 15 : bool returnvalue = true;
2612 : 30 : for ( QgsGeometryMap::const_iterator it = geometry_map.constBegin(); it != geometry_map.constEnd(); ++it )
2613 : : {
2614 : 15 : gdal::ogr_feature_unique_ptr theOGRFeature( mOgrLayer->GetFeature( FID_TO_NUMBER( it.key() ) ) );
2615 : 15 : if ( !theOGRFeature )
2616 : : {
2617 : 0 : pushError( tr( "OGR error changing geometry: feature %1 not found" ).arg( it.key() ) );
2618 : 0 : continue;
2619 : : }
2620 : :
2621 : 15 : if ( mayNeedResetReadingAfterGetFeature )
2622 : : {
2623 : 0 : mOgrLayer->ResetReading(); // needed for SQLite-based to clear iterator, which could let the database in a locked state otherwise
2624 : 0 : }
2625 : :
2626 : 15 : OGRGeometryH newGeometry = nullptr;
2627 : 15 : QByteArray wkb = it->asWkb();
2628 : : // We might receive null geometries. It is OK, but don't go through the
2629 : : // OGR_G_CreateFromWkb() route then
2630 : 15 : if ( !wkb.isEmpty() )
2631 : : {
2632 : : //create an OGRGeometry
2633 : 30 : if ( OGR_G_CreateFromWkb( reinterpret_cast<unsigned char *>( const_cast<char *>( wkb.constData() ) ),
2634 : 15 : mOgrLayer->GetSpatialRef(),
2635 : : &newGeometry,
2636 : 30 : wkb.length() ) != OGRERR_NONE )
2637 : : {
2638 : 0 : pushError( tr( "OGR error creating geometry for feature %1: %2" ).arg( it.key() ).arg( CPLGetLastErrorMsg() ) );
2639 : 0 : OGR_G_DestroyGeometry( newGeometry );
2640 : 0 : newGeometry = nullptr;
2641 : 0 : continue;
2642 : : }
2643 : :
2644 : 15 : if ( !newGeometry )
2645 : : {
2646 : 0 : pushError( tr( "OGR error in feature %1: geometry is null" ).arg( it.key() ) );
2647 : 0 : continue;
2648 : : }
2649 : :
2650 : 15 : newGeometry = ConvertGeometryIfNecessary( newGeometry );
2651 : 15 : }
2652 : :
2653 : : //set the new geometry
2654 : 15 : if ( OGR_F_SetGeometryDirectly( theOGRFeature.get(), newGeometry ) != OGRERR_NONE )
2655 : : {
2656 : 0 : pushError( tr( "OGR error setting geometry of feature %1: %2" ).arg( it.key() ).arg( CPLGetLastErrorMsg() ) );
2657 : : // Shouldn't happen normally. If it happens, ownership of the geometry
2658 : : // may be not really well defined, so better not destroy it, but just
2659 : : // the feature.
2660 : 0 : continue;
2661 : : }
2662 : :
2663 : :
2664 : 15 : if ( mOgrLayer->SetFeature( theOGRFeature.get() ) != OGRERR_NONE )
2665 : : {
2666 : 0 : pushError( tr( "OGR error setting feature %1: %2" ).arg( it.key() ).arg( CPLGetLastErrorMsg() ) );
2667 : 0 : returnvalue = false;
2668 : 0 : continue;
2669 : : }
2670 : 15 : mShapefileMayBeCorrupted = true;
2671 : :
2672 : 15 : invalidateCachedExtent( true );
2673 : 15 : }
2674 : :
2675 : 15 : if ( inTransaction )
2676 : : {
2677 : 0 : if ( returnvalue )
2678 : 0 : returnvalue = commitTransaction();
2679 : : else
2680 : 0 : rollbackTransaction();
2681 : 0 : }
2682 : :
2683 : 15 : if ( mTransaction )
2684 : 0 : mTransaction->dirtyLastSavePoint();
2685 : :
2686 : 15 : if ( !syncToDisc() )
2687 : : {
2688 : 0 : pushError( tr( "OGR error syncing to disk: %1" ).arg( CPLGetLastErrorMsg() ) );
2689 : 0 : }
2690 : 15 : return returnvalue;
2691 : 15 : }
2692 : :
2693 : 0 : bool QgsOgrProvider::createSpatialIndex()
2694 : : {
2695 : 0 : QgsCPLHTTPFetchOverrider oCPLHTTPFetcher( mAuthCfg );
2696 : 0 : QgsSetCPLHTTPFetchOverriderInitiatorClass( oCPLHTTPFetcher, QStringLiteral( "QgsOgrProvider" ) );
2697 : :
2698 : 0 : if ( !mOgrOrigLayer )
2699 : 0 : return false;
2700 : 0 : if ( !doInitialActionsForEdition() )
2701 : 0 : return false;
2702 : :
2703 : 0 : QByteArray layerName = mOgrOrigLayer->name();
2704 : 0 : if ( mGDALDriverName == QLatin1String( "ESRI Shapefile" ) )
2705 : : {
2706 : 0 : QByteArray sql = QByteArray( "CREATE SPATIAL INDEX ON " ) + quotedIdentifier( layerName ); // quote the layer name so spaces are handled
2707 : 0 : QgsDebugMsgLevel( QStringLiteral( "SQL: %1" ).arg( QString::fromUtf8( sql ) ), 2 );
2708 : 0 : mOgrOrigLayer->ExecuteSQLNoReturn( sql );
2709 : :
2710 : 0 : if ( !mFilePath.endsWith( QLatin1String( ".shp" ), Qt::CaseInsensitive ) )
2711 : 0 : return true;
2712 : :
2713 : 0 : QFileInfo fi( mFilePath ); // to get the base name
2714 : : //find out, if the .qix file is there
2715 : 0 : return QFileInfo::exists( fi.path().append( '/' ).append( fi.completeBaseName() ).append( ".qix" ) );
2716 : 0 : }
2717 : 0 : else if ( mGDALDriverName == QLatin1String( "GPKG" ) ||
2718 : 0 : mGDALDriverName == QLatin1String( "SQLite" ) )
2719 : : {
2720 : 0 : QMutex *mutex = nullptr;
2721 : 0 : OGRLayerH layer = mOgrOrigLayer->getHandleAndMutex( mutex );
2722 : 0 : QByteArray sql = QByteArray( "SELECT CreateSpatialIndex(" + quotedIdentifier( layerName ) + ","
2723 : 0 : + quotedIdentifier( OGR_L_GetGeometryColumn( layer ) ) + ") " ); // quote the layer name so spaces are handled
2724 : 0 : mOgrOrigLayer->ExecuteSQLNoReturn( sql );
2725 : 0 : return true;
2726 : 0 : }
2727 : 0 : return false;
2728 : 0 : }
2729 : :
2730 : 0 : QString QgsOgrProvider::createIndexName( QString tableName, QString field )
2731 : : {
2732 : 0 : QRegularExpression safeExp( QStringLiteral( "[^a-zA-Z0-9]" ) );
2733 : 0 : tableName.replace( safeExp, QStringLiteral( "_" ) );
2734 : 0 : field.replace( safeExp, QStringLiteral( "_" ) );
2735 : 0 : return tableName + "_" + field + "_idx";
2736 : 0 : }
2737 : :
2738 : 0 : bool QgsOgrProvider::createAttributeIndex( int field )
2739 : : {
2740 : 0 : QgsCPLHTTPFetchOverrider oCPLHTTPFetcher( mAuthCfg );
2741 : 0 : QgsSetCPLHTTPFetchOverriderInitiatorClass( oCPLHTTPFetcher, QStringLiteral( "QgsOgrProvider" ) );
2742 : :
2743 : 0 : if ( field < 0 || field >= mAttributeFields.count() )
2744 : 0 : return false;
2745 : :
2746 : 0 : if ( !doInitialActionsForEdition() )
2747 : 0 : return false;
2748 : :
2749 : 0 : QByteArray quotedLayerName = quotedIdentifier( mOgrOrigLayer->name() );
2750 : 0 : if ( mGDALDriverName == QLatin1String( "GPKG" ) ||
2751 : 0 : mGDALDriverName == QLatin1String( "SQLite" ) )
2752 : : {
2753 : 0 : if ( field == 0 && mFirstFieldIsFid )
2754 : : {
2755 : : // already an index on this field, no need to re-created
2756 : 0 : return false;
2757 : : }
2758 : :
2759 : 0 : QString indexName = createIndexName( mOgrOrigLayer->name(), fields().at( field ).name() );
2760 : 0 : QByteArray createSql = "CREATE INDEX IF NOT EXISTS " + textEncoding()->fromUnicode( indexName ) + " ON " + quotedLayerName + " (" + textEncoding()->fromUnicode( fields().at( field ).name() ) + ")";
2761 : 0 : mOgrOrigLayer->ExecuteSQLNoReturn( createSql );
2762 : 0 : return true;
2763 : 0 : }
2764 : : else
2765 : : {
2766 : 0 : QByteArray dropSql = "DROP INDEX ON " + quotedLayerName;
2767 : 0 : mOgrOrigLayer->ExecuteSQLNoReturn( dropSql );
2768 : 0 : QByteArray createSql = "CREATE INDEX ON " + quotedLayerName + " USING " + textEncoding()->fromUnicode( fields().at( field ).name() );
2769 : 0 : mOgrOrigLayer->ExecuteSQLNoReturn( createSql );
2770 : :
2771 : 0 : QFileInfo fi( mFilePath ); // to get the base name
2772 : : //find out, if the .idm/.ind file is there
2773 : 0 : QString idmFile( fi.path().append( '/' ).append( fi.completeBaseName() ).append( ".idm" ) );
2774 : 0 : QString indFile( fi.path().append( '/' ).append( fi.completeBaseName() ).append( ".ind" ) );
2775 : 0 : return QFile::exists( idmFile ) || QFile::exists( indFile );
2776 : 0 : }
2777 : 0 : }
2778 : :
2779 : 8 : bool QgsOgrProvider::deleteFeatures( const QgsFeatureIds &id )
2780 : : {
2781 : 8 : QgsCPLHTTPFetchOverrider oCPLHTTPFetcher( mAuthCfg );
2782 : 16 : QgsSetCPLHTTPFetchOverriderInitiatorClass( oCPLHTTPFetcher, QStringLiteral( "QgsOgrProvider" ) );
2783 : :
2784 : 8 : if ( !doInitialActionsForEdition() )
2785 : 0 : return false;
2786 : :
2787 : 8 : const bool inTransaction = startTransaction();
2788 : :
2789 : 8 : bool returnvalue = true;
2790 : 16 : for ( QgsFeatureIds::const_iterator it = id.begin(); it != id.end(); ++it )
2791 : : {
2792 : 8 : if ( !deleteFeature( *it ) )
2793 : : {
2794 : 0 : returnvalue = false;
2795 : 0 : }
2796 : 8 : }
2797 : :
2798 : 8 : if ( inTransaction )
2799 : : {
2800 : 0 : if ( returnvalue )
2801 : 0 : returnvalue = commitTransaction();
2802 : : else
2803 : 0 : rollbackTransaction();
2804 : 0 : }
2805 : :
2806 : 8 : if ( mTransaction )
2807 : 0 : mTransaction->dirtyLastSavePoint();
2808 : :
2809 : 8 : if ( !syncToDisc() )
2810 : : {
2811 : 0 : returnvalue = false;
2812 : 0 : }
2813 : :
2814 : 8 : if ( mGDALDriverName == QLatin1String( "ESRI Shapefile" ) )
2815 : : {
2816 : : // Shapefile behaves in a special way due to possible recompaction
2817 : 8 : recalculateFeatureCount();
2818 : 8 : }
2819 : : else
2820 : : {
2821 : 0 : if ( mFeaturesCounted != QgsVectorDataProvider::Uncounted &&
2822 : 0 : mFeaturesCounted != QgsVectorDataProvider::UnknownCount )
2823 : : {
2824 : 0 : if ( returnvalue )
2825 : 0 : mFeaturesCounted -= id.size();
2826 : : else
2827 : 0 : recalculateFeatureCount();
2828 : 0 : }
2829 : : }
2830 : :
2831 : 8 : clearMinMaxCache();
2832 : :
2833 : 8 : invalidateCachedExtent( true );
2834 : :
2835 : 8 : return returnvalue;
2836 : 8 : }
2837 : :
2838 : 8 : bool QgsOgrProvider::deleteFeature( QgsFeatureId id )
2839 : : {
2840 : 8 : QgsCPLHTTPFetchOverrider oCPLHTTPFetcher( mAuthCfg );
2841 : 16 : QgsSetCPLHTTPFetchOverriderInitiatorClass( oCPLHTTPFetcher, QStringLiteral( "QgsOgrProvider" ) );
2842 : :
2843 : 8 : if ( !doInitialActionsForEdition() )
2844 : 0 : return false;
2845 : :
2846 : 8 : if ( mOgrLayer->DeleteFeature( FID_TO_NUMBER( id ) ) != OGRERR_NONE )
2847 : : {
2848 : 0 : pushError( tr( "OGR error deleting feature %1: %2" ).arg( id ).arg( CPLGetLastErrorMsg() ) );
2849 : 0 : return false;
2850 : : }
2851 : :
2852 : 8 : if ( mTransaction )
2853 : 0 : mTransaction->dirtyLastSavePoint();
2854 : :
2855 : 8 : mShapefileMayBeCorrupted = true;
2856 : :
2857 : 8 : return true;
2858 : 8 : }
2859 : :
2860 : 48 : bool QgsOgrProvider::doInitialActionsForEdition()
2861 : : {
2862 : 48 : if ( !mValid )
2863 : 0 : return false;
2864 : :
2865 : : // If mUpdateModeStackDepth > 0, it means that an updateMode is already active and that we have write access
2866 : 48 : if ( mUpdateModeStackDepth == 0 )
2867 : : {
2868 : 0 : QgsDebugMsgLevel( QStringLiteral( "Enter update mode implicitly" ), 1 );
2869 : 0 : if ( !_enterUpdateMode( true ) )
2870 : 0 : return false;
2871 : 0 : }
2872 : :
2873 : 48 : return true;
2874 : 48 : }
2875 : :
2876 : 256 : QgsVectorDataProvider::Capabilities QgsOgrProvider::capabilities() const
2877 : : {
2878 : 256 : return mCapabilities;
2879 : : }
2880 : :
2881 : 64 : void QgsOgrProvider::computeCapabilities()
2882 : : {
2883 : 64 : QgsVectorDataProvider::Capabilities ability = QgsVectorDataProvider::Capabilities();
2884 : 64 : bool updateModeActivated = false;
2885 : :
2886 : : // collect abilities reported by OGR
2887 : 64 : if ( mOgrLayer )
2888 : : {
2889 : :
2890 : : // We want the layer in rw mode or capabilities will be wrong
2891 : : // If mUpdateModeStackDepth > 0, it means that an updateMode is already active and that we have write access
2892 : 64 : if ( mUpdateModeStackDepth == 0 )
2893 : : {
2894 : 64 : updateModeActivated = _enterUpdateMode( true );
2895 : 64 : }
2896 : :
2897 : : // Whilst the OGR documentation (e.g. at
2898 : : // https://gdal.org/doxygen/classOGRLayer.html#aeedbda1a62f9b89b8e5f24332cf22286) states "The capability
2899 : : // codes that can be tested are represented as strings, but #defined
2900 : : // constants exists to ensure correct spelling", we always use strings
2901 : : // here. This is because older versions of OGR don't always have all
2902 : : // the #defines we want to test for here.
2903 : :
2904 : 64 : if ( mOgrLayer->TestCapability( "RandomRead" ) )
2905 : : // true if the GetFeature() method works *efficiently* for this layer.
2906 : : // TODO: Perhaps influence if QGIS caches into memory
2907 : : // (vs read from disk every time) based on this setting.
2908 : : {
2909 : : // the latter flag is here just for compatibility
2910 : 64 : ability |= QgsVectorDataProvider::SelectAtId;
2911 : 64 : }
2912 : :
2913 : 64 : if ( mWriteAccessPossible && mOgrLayer->TestCapability( "SequentialWrite" ) )
2914 : : // true if the CreateFeature() method works for this layer.
2915 : : {
2916 : 64 : ability |= QgsVectorDataProvider::AddFeatures;
2917 : 64 : }
2918 : :
2919 : 64 : if ( mWriteAccessPossible && mOgrLayer->TestCapability( "DeleteFeature" ) )
2920 : : // true if this layer can delete its features
2921 : : {
2922 : 64 : ability |= DeleteFeatures;
2923 : 64 : }
2924 : :
2925 : 64 : if ( mWriteAccessPossible && mOgrLayer->TestCapability( "RandomWrite" ) )
2926 : : // true if the SetFeature() method is operational on this layer.
2927 : : {
2928 : : // TODO According to http://shapelib.maptools.org/ (Shapefile C Library V1.2)
2929 : : // TODO "You can't modify the vertices of existing structures".
2930 : : // TODO Need to work out versions of shapelib vs versions of GDAL/OGR
2931 : : // TODO And test appropriately.
2932 : :
2933 : 64 : ability |= ChangeAttributeValues;
2934 : 64 : ability |= ChangeGeometries;
2935 : 64 : }
2936 : :
2937 : : #if 0
2938 : : if ( mOgrLayer->TestCapability( "FastFeatureCount" ) )
2939 : : // true if this layer can return a feature count
2940 : : // (via OGRLayer::GetFeatureCount()) efficiently ... ie. without counting
2941 : : // the features. In some cases this will return true until a spatial
2942 : : // filter is installed after which it will return false.
2943 : : {
2944 : : // TODO: Perhaps use as a clue by QGIS whether it should spawn a thread to count features.
2945 : : }
2946 : :
2947 : : if ( mOgrLayer->TestCapability( "FastSetNextByIndex" ) )
2948 : : // true if this layer can perform the SetNextByIndex() call efficiently.
2949 : : {
2950 : : // No use required for this QGIS release.
2951 : : }
2952 : : #endif
2953 : :
2954 : 64 : if ( mWriteAccessPossible && mOgrLayer->TestCapability( "CreateField" ) )
2955 : : {
2956 : 64 : ability |= AddAttributes;
2957 : 64 : }
2958 : :
2959 : 64 : if ( mWriteAccessPossible && mOgrLayer->TestCapability( "DeleteField" ) )
2960 : : {
2961 : 64 : ability |= DeleteAttributes;
2962 : 64 : }
2963 : :
2964 : 64 : if ( mWriteAccessPossible && mOgrLayer->TestCapability( "AlterFieldDefn" ) )
2965 : : {
2966 : 64 : ability |= RenameAttributes;
2967 : 64 : }
2968 : :
2969 : 64 : if ( !mOgrLayer->TestCapability( OLCStringsAsUTF8 ) )
2970 : : {
2971 : 63 : ability |= SelectEncoding;
2972 : 63 : }
2973 : :
2974 : : // OGR doesn't handle shapefiles without attributes, ie. missing DBFs well, fixes #803
2975 : 64 : if ( mGDALDriverName == QLatin1String( "ESRI Shapefile" ) )
2976 : : {
2977 : 63 : ability |= CreateSpatialIndex;
2978 : 63 : ability |= CreateAttributeIndex;
2979 : :
2980 : 63 : if ( ( ability & ChangeAttributeValues ) == 0 )
2981 : : {
2982 : : // on readonly shapes OGR reports that it can delete features although it can't RandomWrite
2983 : 0 : ability &= ~( AddAttributes | DeleteFeatures );
2984 : 0 : }
2985 : 63 : }
2986 : 1 : else if ( mGDALDriverName == QLatin1String( "GPKG" ) ||
2987 : 0 : mGDALDriverName == QLatin1String( "SQLite" ) )
2988 : : {
2989 : 1 : ability |= CreateSpatialIndex;
2990 : 1 : ability |= CreateAttributeIndex;
2991 : 1 : }
2992 : :
2993 : : /* Curve geometries are available in some drivers starting with GDAL 2.0 */
2994 : 64 : if ( mOgrLayer->TestCapability( "CurveGeometries" ) )
2995 : : {
2996 : 1 : ability |= CircularGeometries;
2997 : 1 : }
2998 : :
2999 : 64 : if ( mGDALDriverName == QLatin1String( "GPKG" ) )
3000 : : {
3001 : : //supports transactions
3002 : 1 : ability |= TransactionSupport;
3003 : 1 : }
3004 : :
3005 : 64 : if ( GDALGetMetadataItem( mOgrLayer->driver(), GDAL_DCAP_FEATURE_STYLES, nullptr ) != nullptr )
3006 : : {
3007 : 0 : ability |= FeatureSymbology;
3008 : 0 : ability |= CreateRenderer;
3009 : 0 : }
3010 : 64 : }
3011 : :
3012 : 64 : ability |= ReadLayerMetadata;
3013 : 64 : ability |= ReloadData;
3014 : :
3015 : 64 : if ( updateModeActivated )
3016 : 64 : leaveUpdateMode();
3017 : :
3018 : 64 : mCapabilities = ability;
3019 : 64 : }
3020 : :
3021 : :
3022 : 0 : QString QgsOgrProvider::name() const
3023 : : {
3024 : 0 : return TEXT_PROVIDER_KEY;
3025 : : }
3026 : :
3027 : 3 : QString QgsOgrProvider::providerKey()
3028 : : {
3029 : 6 : return TEXT_PROVIDER_KEY;
3030 : : }
3031 : :
3032 : 0 : QString QgsOgrProvider::description() const
3033 : : {
3034 : 0 : return TEXT_PROVIDER_DESCRIPTION;
3035 : : } // QgsOgrProvider::description()
3036 : :
3037 : :
3038 : : /**
3039 : :
3040 : : Convenience function for readily creating file filters.
3041 : :
3042 : : Given a long name for a file filter and a regular expression, return
3043 : : a file filter string suitable for use in a QFileDialog::OpenFiles()
3044 : : call. The regular express, glob, will have both all lower and upper
3045 : : case versions added.
3046 : :
3047 : : \note
3048 : :
3049 : : Copied from qgisapp.cpp.
3050 : :
3051 : : \todo XXX This should probably be generalized and moved to a standard
3052 : : utility type thingy.
3053 : :
3054 : : */
3055 : 174 : static QString createFileFilter_( QString const &longName, QString const &glob )
3056 : : {
3057 : : // return longName + " [OGR] (" + glob.toLower() + ' ' + glob.toUpper() + ");;";
3058 : 174 : return longName + " (" + glob.toLower() + ' ' + glob.toUpper() + ");;";
3059 : 0 : } // createFileFilter_
3060 : :
3061 : :
3062 : 12 : QString createFilters( const QString &type )
3063 : : {
3064 : : //! Database drivers available
3065 : 12 : static QString sDatabaseDrivers;
3066 : : //! Protocol drivers available
3067 : 12 : static QString sProtocolDrivers;
3068 : : //! File filters
3069 : 12 : static QString sFileFilters;
3070 : : //! Directory drivers
3071 : 12 : static QString sDirectoryDrivers;
3072 : : //! Extensions
3073 : 12 : static QStringList sExtensions;
3074 : : //! Directory extensions
3075 : 12 : static QStringList sDirectoryExtensions;
3076 : : //! Wildcards
3077 : 12 : static QStringList sWildcards;
3078 : :
3079 : : // if we've already built the supported vector string, just return what
3080 : : // we've already built
3081 : :
3082 : 12 : if ( sFileFilters.isEmpty() || sFileFilters.isNull() )
3083 : : {
3084 : : // register ogr plugins
3085 : 3 : QgsApplication::registerOgrDrivers();
3086 : :
3087 : : // first get the GDAL driver manager
3088 : : GDALDriverH driver; // current driver
3089 : 3 : QString driverName; // current driver name
3090 : :
3091 : : // Grind through all the drivers and their respective metadata.
3092 : : // We'll add a file filter for those drivers that have a file
3093 : : // extension defined for them; the others, welll, even though
3094 : : // theoreticaly we can open those files because there exists a
3095 : : // driver for them, the user will have to use the "All Files" to
3096 : : // open datasets with no explicitly defined file name extension.
3097 : 3 : QgsDebugMsgLevel( QStringLiteral( "Driver count: %1" ).arg( OGRGetDriverCount() ), 3 );
3098 : :
3099 : 3 : bool kmlFound = false;
3100 : 3 : bool dwgFound = false;
3101 : 3 : bool dgnFound = false;
3102 : :
3103 : 267 : for ( int i = 0; i < OGRGetDriverCount(); ++i )
3104 : : {
3105 : 264 : driver = OGRGetDriver( i );
3106 : :
3107 : 264 : Q_CHECK_PTR( driver ); // NOLINT
3108 : :
3109 : 264 : if ( !driver )
3110 : : {
3111 : 0 : QgsMessageLog::logMessage( QObject::tr( "Unable to get driver %1" ).arg( i ), QObject::tr( "OGR" ) );
3112 : 0 : continue;
3113 : : }
3114 : :
3115 : 264 : driverName = GDALGetDriverShortName( driver );
3116 : :
3117 : 264 : if ( driverName.startsWith( QLatin1String( "AVCBin" ) ) )
3118 : : {
3119 : 3 : sDirectoryDrivers += QObject::tr( "Arc/Info Binary Coverage" ) + ",AVCBin;";
3120 : 3 : }
3121 : 261 : else if ( driverName.startsWith( QLatin1String( "AVCE00" ) ) )
3122 : : {
3123 : 6 : sFileFilters += createFileFilter_( QObject::tr( "Arc/Info ASCII Coverage" ), QStringLiteral( "*.e00" ) );
3124 : 6 : sExtensions << QStringLiteral( "e00" );
3125 : 3 : }
3126 : 258 : else if ( driverName.startsWith( QLatin1String( "BNA" ) ) )
3127 : : {
3128 : 6 : sFileFilters += createFileFilter_( QObject::tr( "Atlas BNA" ), QStringLiteral( "*.bna" ) );
3129 : 6 : sExtensions << QStringLiteral( "bna" );
3130 : 3 : }
3131 : 255 : else if ( driverName.startsWith( QLatin1String( "CSV" ) ) )
3132 : : {
3133 : 6 : sFileFilters += createFileFilter_( QObject::tr( "Comma Separated Value" ), QStringLiteral( "*.csv" ) );
3134 : 6 : sExtensions << QStringLiteral( "csv" );
3135 : 3 : }
3136 : 252 : else if ( driverName.startsWith( QObject::tr( "DODS" ) ) )
3137 : : {
3138 : 0 : sProtocolDrivers += QLatin1String( "DODS/OPeNDAP,DODS;" );
3139 : 0 : }
3140 : 252 : else if ( driverName.startsWith( QObject::tr( "CouchDB" ) ) )
3141 : : {
3142 : 3 : sProtocolDrivers += QLatin1String( "CouchDB;" );
3143 : 3 : }
3144 : 249 : else if ( driverName.startsWith( QLatin1String( "FileGDB" ) ) )
3145 : : {
3146 : 0 : sDirectoryDrivers += QObject::tr( "ESRI FileGDB" ) + ",FileGDB;";
3147 : 0 : if ( !sDirectoryExtensions.contains( QStringLiteral( "gdb" ) ) )
3148 : 0 : sDirectoryExtensions << QStringLiteral( "gdb" );
3149 : 0 : }
3150 : : #if GDAL_VERSION_NUM >= GDAL_COMPUTE_VERSION(3,1,0)
3151 : 249 : else if ( driverName.startsWith( QLatin1String( "FlatGeobuf" ) ) )
3152 : : {
3153 : 3 : sProtocolDrivers += QLatin1String( "FlatGeobuf;" );
3154 : 6 : sFileFilters += createFileFilter_( QObject::tr( "FlatGeobuf" ), QStringLiteral( "*.fgb" ) );
3155 : 6 : sExtensions << QStringLiteral( "fgb" );
3156 : 3 : }
3157 : : #endif
3158 : 246 : else if ( driverName.startsWith( QLatin1String( "PGeo" ) ) )
3159 : : {
3160 : 0 : sDatabaseDrivers += QObject::tr( "ESRI Personal GeoDatabase" ) + ",PGeo;";
3161 : : #ifdef Q_OS_WIN
3162 : : sFileFilters += createFileFilter_( QObject::tr( "ESRI Personal GeoDatabase" ), "*.mdb" );
3163 : : sExtensions << "mdb";
3164 : : #endif
3165 : 0 : }
3166 : 246 : else if ( driverName.startsWith( QLatin1String( "SDE" ) ) )
3167 : : {
3168 : 0 : sDatabaseDrivers += QObject::tr( "ESRI ArcSDE" ) + ",SDE;";
3169 : 0 : }
3170 : 246 : else if ( driverName.startsWith( QLatin1String( "ESRI Shapefile" ) ) )
3171 : : {
3172 : 3 : QString exts = GDALGetMetadataItem( driver, GDAL_DMD_EXTENSIONS, "" );
3173 : 6 : sFileFilters += createFileFilter_( QObject::tr( "ESRI Shapefiles" ), exts.contains( "shz" ) ? QStringLiteral( "*.shp *.shz *.shp.zip" ) : QStringLiteral( "*.shp" ) );
3174 : 9 : sExtensions << QStringLiteral( "shp" ) << QStringLiteral( "dbf" );
3175 : 3 : if ( exts.contains( "shz" ) )
3176 : 9 : sExtensions << QStringLiteral( "shz" ) << QStringLiteral( "shp.zip" );
3177 : 3 : }
3178 : 243 : else if ( driverName.startsWith( QObject::tr( "FMEObjects Gateway" ) ) )
3179 : : {
3180 : 0 : sFileFilters += createFileFilter_( QObject::tr( "FMEObjects Gateway" ), QStringLiteral( "*.fdd" ) );
3181 : 0 : sExtensions << QStringLiteral( "fdd" );
3182 : 0 : }
3183 : 243 : else if ( driverName.startsWith( QLatin1String( "GeoJSONSeq" ) ) )
3184 : : {
3185 : 3 : sProtocolDrivers += QLatin1String( "GeoJSON - Newline Delimited;" );
3186 : 6 : sFileFilters += createFileFilter_( QObject::tr( "GeoJSON Newline Delimited JSON" ), QStringLiteral( "*.geojsonl *.geojsons *.nlgeojson *.json" ) );
3187 : 15 : sExtensions << QStringLiteral( "geojsonl" ) << QStringLiteral( "geojsons" ) << QStringLiteral( "nlgeojson" ) << QStringLiteral( "json" );
3188 : 3 : }
3189 : 240 : else if ( driverName.startsWith( QLatin1String( "GeoJSON" ) ) )
3190 : : {
3191 : 3 : sProtocolDrivers += QLatin1String( "GeoJSON;" );
3192 : 6 : sFileFilters += createFileFilter_( QObject::tr( "GeoJSON" ), QStringLiteral( "*.geojson" ) );
3193 : 6 : sExtensions << QStringLiteral( "geojson" );
3194 : 3 : }
3195 : 237 : else if ( driverName.startsWith( QLatin1String( "GeoRSS" ) ) )
3196 : : {
3197 : 6 : sFileFilters += createFileFilter_( QObject::tr( "GeoRSS" ), QStringLiteral( "*.xml" ) );
3198 : 6 : sExtensions << QStringLiteral( "xml" );
3199 : 3 : }
3200 : 234 : else if ( driverName == QLatin1String( "GML" ) )
3201 : : {
3202 : 6 : sFileFilters += createFileFilter_( QObject::tr( "Geography Markup Language [GML]" ), QStringLiteral( "*.gml" ) );
3203 : 6 : sExtensions << QStringLiteral( "gml" );
3204 : 3 : }
3205 : 231 : else if ( driverName == QLatin1String( "GMLAS" ) )
3206 : : {
3207 : 3 : continue;
3208 : : }
3209 : 228 : else if ( driverName.startsWith( QLatin1String( "GMT" ) ) )
3210 : : {
3211 : 0 : sFileFilters += createFileFilter_( QObject::tr( "Generic Mapping Tools [GMT]" ), QStringLiteral( "*.gmt" ) );
3212 : 0 : sExtensions << QStringLiteral( "gmt" );
3213 : 0 : }
3214 : 228 : else if ( driverName.startsWith( QLatin1String( "GPX" ) ) )
3215 : : {
3216 : 6 : sFileFilters += createFileFilter_( QObject::tr( "GPS eXchange Format [GPX]" ), QStringLiteral( "*.gpx" ) );
3217 : 6 : sExtensions << QStringLiteral( "gpx" );
3218 : 3 : }
3219 : 225 : else if ( driverName.startsWith( QLatin1String( "GPKG" ) ) )
3220 : : {
3221 : 6 : sFileFilters += createFileFilter_( QObject::tr( "GeoPackage" ), QStringLiteral( "*.gpkg" ) );
3222 : 6 : sExtensions << QStringLiteral( "gpkg" );
3223 : 3 : }
3224 : 222 : else if ( driverName.startsWith( QLatin1String( "GRASS" ) ) )
3225 : : {
3226 : 0 : sDirectoryDrivers += QObject::tr( "Grass Vector" ) + ",GRASS;";
3227 : 0 : }
3228 : 222 : else if ( driverName.startsWith( QLatin1String( "IDB" ) ) )
3229 : : {
3230 : 0 : sDatabaseDrivers += QObject::tr( "Informix DataBlade" ) + ",IDB;";
3231 : 0 : }
3232 : 222 : else if ( driverName.startsWith( QLatin1String( "Interlis 1" ) ) )
3233 : : {
3234 : 6 : sFileFilters += createFileFilter_( QObject::tr( "INTERLIS 1" ), QStringLiteral( "*.itf *.xml *.ili" ) );
3235 : 12 : sExtensions << QStringLiteral( "itf" ) << QStringLiteral( "xml" ) << QStringLiteral( "ili" );
3236 : 3 : }
3237 : 219 : else if ( driverName.startsWith( QLatin1String( "Interlis 2" ) ) )
3238 : : {
3239 : 6 : sFileFilters += createFileFilter_( QObject::tr( "INTERLIS 2" ), QStringLiteral( "*.xtf *.xml *.ili" ) );
3240 : 12 : sExtensions << QStringLiteral( "xtf" ) << QStringLiteral( "xml" ) << QStringLiteral( "ili" );
3241 : 3 : }
3242 : 216 : else if ( driverName.startsWith( QLatin1String( "Ingres" ) ) )
3243 : : {
3244 : 0 : sDatabaseDrivers += QObject::tr( "Ingres" ) + ",Ingres;";
3245 : 0 : }
3246 : 216 : else if ( driverName == QLatin1String( "KML" ) || driverName == QLatin1String( "LIBKML" ) )
3247 : : {
3248 : 6 : if ( kmlFound )
3249 : 3 : continue;
3250 : 3 : kmlFound = true;
3251 : 6 : sFileFilters += createFileFilter_( QObject::tr( "Keyhole Markup Language [KML]" ), QStringLiteral( "*.kml *.kmz" ) );
3252 : 9 : sExtensions << QStringLiteral( "kml" ) << QStringLiteral( "kmz" );
3253 : 3 : }
3254 : 210 : else if ( driverName.startsWith( QLatin1String( "MapInfo File" ) ) )
3255 : : {
3256 : 6 : sFileFilters += createFileFilter_( QObject::tr( "Mapinfo File" ), QStringLiteral( "*.mif *.tab" ) );
3257 : 9 : sExtensions << QStringLiteral( "mif" ) << QStringLiteral( "tab" );
3258 : 3 : }
3259 : 207 : else if ( driverName == QLatin1String( "DGN" ) || driverName == QLatin1String( "DGNV8" ) )
3260 : : {
3261 : 3 : if ( dgnFound )
3262 : 0 : continue;
3263 : 3 : dgnFound = true;
3264 : 6 : sFileFilters += createFileFilter_( QObject::tr( "Microstation DGN" ), QStringLiteral( "*.dgn" ) );
3265 : 6 : sExtensions << QStringLiteral( "dgn" );
3266 : 3 : }
3267 : 204 : else if ( driverName.startsWith( QLatin1String( "MySQL" ) ) )
3268 : : {
3269 : 0 : sDatabaseDrivers += QObject::tr( "MySQL" ) + ",MySQL;";
3270 : 0 : }
3271 : 204 : else if ( driverName.startsWith( QLatin1String( "MSSQL" ) ) )
3272 : : {
3273 : 0 : sDatabaseDrivers += QObject::tr( "MSSQL" ) + ",MSSQL;";
3274 : 0 : }
3275 : 204 : else if ( driverName.startsWith( QLatin1String( "OCI" ) ) )
3276 : : {
3277 : 0 : sDatabaseDrivers += QObject::tr( "Oracle Spatial" ) + ",OCI;";
3278 : 0 : }
3279 : 204 : else if ( driverName.startsWith( QLatin1String( "ODBC" ) ) )
3280 : : {
3281 : 0 : sDatabaseDrivers += QObject::tr( "ODBC" ) + ",ODBC;";
3282 : 0 : }
3283 : 204 : else if ( driverName.startsWith( QLatin1String( "OGDI" ) ) )
3284 : : {
3285 : 0 : sDatabaseDrivers += QObject::tr( "OGDI Vectors" ) + ",OGDI;";
3286 : 0 : }
3287 : 204 : else if ( driverName.startsWith( QLatin1String( "OpenFileGDB" ) ) )
3288 : : {
3289 : 3 : sDirectoryDrivers += QObject::tr( "OpenFileGDB" ) + ",OpenFileGDB;";
3290 : 6 : if ( !sDirectoryExtensions.contains( QStringLiteral( "gdb" ) ) )
3291 : 6 : sDirectoryExtensions << QStringLiteral( "gdb" );
3292 : 3 : }
3293 : 201 : else if ( driverName.startsWith( QLatin1String( "PostgreSQL" ) ) )
3294 : : {
3295 : 3 : sDatabaseDrivers += QObject::tr( "PostgreSQL" ) + ",PostgreSQL;";
3296 : 3 : }
3297 : 198 : else if ( driverName.startsWith( QLatin1String( "S57" ) ) )
3298 : : {
3299 : 3 : sFileFilters += createFileFilter_( QObject::tr( "S-57 Base file" ),
3300 : 6 : QStringLiteral( "*.000" ) );
3301 : 6 : sExtensions << QStringLiteral( "000" );
3302 : 3 : }
3303 : 195 : else if ( driverName.startsWith( QLatin1String( "SDTS" ) ) )
3304 : : {
3305 : 0 : sFileFilters += createFileFilter_( QObject::tr( "Spatial Data Transfer Standard [SDTS]" ),
3306 : 0 : QStringLiteral( "*catd.ddf" ) );
3307 : 0 : sWildcards << QStringLiteral( "*catd.ddf" );
3308 : 0 : }
3309 : 195 : else if ( driverName.startsWith( QLatin1String( "SOSI" ) ) )
3310 : : {
3311 : 0 : sFileFilters += createFileFilter_( QObject::tr( "Systematic Organization of Spatial Information [SOSI]" ), QStringLiteral( "*.sos" ) );
3312 : 0 : sExtensions << QStringLiteral( "sos" );
3313 : 0 : }
3314 : 195 : else if ( driverName.startsWith( QLatin1String( "SQLite" ) ) )
3315 : : {
3316 : 6 : sFileFilters += createFileFilter_( QObject::tr( "SQLite/SpatiaLite" ), QStringLiteral( "*.sqlite *.db *.sqlite3 *.db3 *.s3db *.sl3" ) );
3317 : 21 : sExtensions << QStringLiteral( "sqlite" ) << QStringLiteral( "db" ) << QStringLiteral( "sqlite3" ) << QStringLiteral( "db3" ) << QStringLiteral( "s3db" ) << QStringLiteral( "sl3" );
3318 : 3 : }
3319 : 192 : else if ( driverName.startsWith( QLatin1String( "SXF" ) ) )
3320 : : {
3321 : 6 : sFileFilters += createFileFilter_( QObject::tr( "Storage and eXchange Format" ), QStringLiteral( "*.sxf" ) );
3322 : 6 : sExtensions << QStringLiteral( "sxf" );
3323 : 3 : }
3324 : 189 : else if ( driverName.startsWith( QLatin1String( "UK .NTF" ) ) )
3325 : : {
3326 : 3 : sDirectoryDrivers += QObject::tr( "UK. NTF2" ) + ",UK. NTF;";
3327 : 3 : }
3328 : 186 : else if ( driverName.startsWith( QLatin1String( "TIGER" ) ) )
3329 : : {
3330 : 3 : sDirectoryDrivers += QObject::tr( "U.S. Census TIGER/Line" ) + ",TIGER;";
3331 : 3 : }
3332 : 183 : else if ( driverName.startsWith( QLatin1String( "OGR_VRT" ) ) )
3333 : : {
3334 : 3 : sFileFilters += createFileFilter_( QObject::tr( "VRT - Virtual Datasource" ),
3335 : 6 : QStringLiteral( "*.vrt *.ovf" ) );
3336 : 9 : sExtensions << QStringLiteral( "vrt" ) << QStringLiteral( "ovf" );
3337 : 3 : }
3338 : 180 : else if ( driverName.startsWith( QLatin1String( "XPlane" ) ) )
3339 : : {
3340 : 3 : sFileFilters += createFileFilter_( QObject::tr( "X-Plane/Flightgear" ),
3341 : 6 : QStringLiteral( "apt.dat nav.dat fix.dat awy.dat" ) );
3342 : 15 : sWildcards << QStringLiteral( "apt.dat" ) << QStringLiteral( "nav.dat" ) << QStringLiteral( "fix.dat" ) << QStringLiteral( "awy.dat" );
3343 : 3 : }
3344 : 177 : else if ( driverName.startsWith( QLatin1String( "Geoconcept" ) ) )
3345 : : {
3346 : 6 : sFileFilters += createFileFilter_( QObject::tr( "Geoconcept" ), QStringLiteral( "*.gxt *.txt" ) );
3347 : 9 : sExtensions << QStringLiteral( "gxt" ) << QStringLiteral( "txt" );
3348 : 3 : }
3349 : 174 : else if ( driverName.startsWith( QLatin1String( "DXF" ) ) )
3350 : : {
3351 : 6 : sFileFilters += createFileFilter_( QObject::tr( "AutoCAD DXF" ), QStringLiteral( "*.dxf" ) );
3352 : 6 : sExtensions << QStringLiteral( "dxf" );
3353 : 3 : }
3354 : 171 : else if ( driverName.startsWith( QLatin1String( "ODS" ) ) )
3355 : : {
3356 : 6 : sFileFilters += createFileFilter_( QObject::tr( "Open Document Spreadsheet" ), QStringLiteral( "*.ods" ) );
3357 : 6 : sExtensions << QStringLiteral( "ods" );
3358 : 3 : }
3359 : 168 : else if ( driverName.startsWith( QLatin1String( "XLSX" ) ) )
3360 : : {
3361 : 6 : sFileFilters += createFileFilter_( QObject::tr( "MS Office Open XML spreadsheet" ), QStringLiteral( "*.xlsx" ) );
3362 : 6 : sExtensions << QStringLiteral( "xlsx" );
3363 : 3 : }
3364 : 165 : else if ( driverName.endsWith( QLatin1String( "XLS" ) ) )
3365 : : {
3366 : 6 : sFileFilters += createFileFilter_( QObject::tr( "MS Excel format" ), QStringLiteral( "*.xls" ) );
3367 : 6 : sExtensions << QStringLiteral( "xls" );
3368 : 3 : }
3369 : 162 : else if ( driverName.startsWith( QLatin1String( "EDIGEO" ) ) )
3370 : : {
3371 : 6 : sFileFilters += createFileFilter_( QObject::tr( "EDIGEO" ), QStringLiteral( "*.thf" ) );
3372 : 6 : sExtensions << QStringLiteral( "thf" );
3373 : 3 : }
3374 : 159 : else if ( driverName.startsWith( QLatin1String( "NAS" ) ) )
3375 : : {
3376 : 6 : sFileFilters += createFileFilter_( QObject::tr( "NAS - ALKIS" ), QStringLiteral( "*.xml" ) );
3377 : 6 : sExtensions << QStringLiteral( "xml" );
3378 : 3 : }
3379 : 156 : else if ( driverName.startsWith( QLatin1String( "WAsP" ) ) )
3380 : : {
3381 : 6 : sFileFilters += createFileFilter_( QObject::tr( "WAsP" ), QStringLiteral( "*.map" ) );
3382 : 6 : sExtensions << QStringLiteral( "map" );
3383 : 3 : }
3384 : 153 : else if ( driverName.startsWith( QLatin1String( "PCIDSK" ) ) )
3385 : : {
3386 : 6 : sFileFilters += createFileFilter_( QObject::tr( "PCI Geomatics Database File" ), QStringLiteral( "*.pix" ) );
3387 : 6 : sExtensions << QStringLiteral( "pix" );
3388 : 3 : }
3389 : 150 : else if ( driverName.startsWith( QLatin1String( "GPSTrackMaker" ) ) )
3390 : : {
3391 : 6 : sFileFilters += createFileFilter_( QObject::tr( "GPSTrackMaker" ), QStringLiteral( "*.gtm *.gtz" ) );
3392 : 9 : sExtensions << QStringLiteral( "gtm" ) << QStringLiteral( "gtz" );
3393 : 3 : }
3394 : 147 : else if ( driverName.startsWith( QLatin1String( "VFK" ) ) )
3395 : : {
3396 : 6 : sFileFilters += createFileFilter_( QObject::tr( "Czech Cadastral Exchange Data Format" ), QStringLiteral( "*.vfk" ) );
3397 : 6 : sExtensions << QStringLiteral( "vfk" );
3398 : 3 : }
3399 : 144 : else if ( driverName.startsWith( QLatin1String( "OSM" ) ) )
3400 : : {
3401 : 6 : sFileFilters += createFileFilter_( QObject::tr( "OpenStreetMap" ), QStringLiteral( "*.osm *.pbf" ) );
3402 : 9 : sExtensions << QStringLiteral( "osm" ) << QStringLiteral( "pbf" );
3403 : 3 : }
3404 : 141 : else if ( driverName.startsWith( QLatin1String( "SUA" ) ) )
3405 : : {
3406 : 6 : sFileFilters += createFileFilter_( QObject::tr( "Special Use Airspace Format" ), QStringLiteral( "*.sua" ) );
3407 : 6 : sExtensions << QStringLiteral( "sua" );
3408 : 3 : }
3409 : 138 : else if ( driverName.startsWith( QLatin1String( "OpenAir" ) ) )
3410 : : {
3411 : 6 : sFileFilters += createFileFilter_( QObject::tr( "OpenAir Special Use Airspace Format" ), QStringLiteral( "*.txt" ) );
3412 : 6 : sExtensions << QStringLiteral( "txt" );
3413 : 3 : }
3414 : 135 : else if ( driverName.startsWith( QLatin1String( "PDS" ) ) )
3415 : : {
3416 : 6 : sFileFilters += createFileFilter_( QObject::tr( "Planetary Data Systems TABLE" ), QStringLiteral( "*.xml" ) );
3417 : 6 : sExtensions << QStringLiteral( "xml" );
3418 : 3 : }
3419 : 132 : else if ( driverName.startsWith( QLatin1String( "HTF" ) ) )
3420 : : {
3421 : 6 : sFileFilters += createFileFilter_( QObject::tr( "Hydrographic Transfer Format" ), QStringLiteral( "*.htf" ) );
3422 : 6 : sExtensions << QStringLiteral( "htf" );
3423 : 3 : }
3424 : 129 : else if ( driverName.startsWith( QLatin1String( "SVG" ) ) )
3425 : : {
3426 : 6 : sFileFilters += createFileFilter_( QObject::tr( "Scalable Vector Graphics" ), QStringLiteral( "*.svg" ) );
3427 : 6 : sExtensions << QStringLiteral( "svg" );
3428 : 3 : }
3429 : 126 : else if ( driverName.startsWith( QLatin1String( "ARCGEN" ) ) )
3430 : : {
3431 : 6 : sFileFilters += createFileFilter_( QObject::tr( "Arc/Info Generate" ), QStringLiteral( "*.gen" ) );
3432 : 6 : sExtensions << QStringLiteral( "gen" );
3433 : 3 : }
3434 : 123 : else if ( driverName.startsWith( QLatin1String( "PDF" ) ) )
3435 : : {
3436 : 6 : sFileFilters += createFileFilter_( QObject::tr( "Geospatial PDF" ), QStringLiteral( "*.pdf" ) );
3437 : 6 : sExtensions << QStringLiteral( "pdf" );
3438 : 3 : }
3439 : 120 : else if ( driverName.startsWith( QLatin1String( "SEGY" ) ) )
3440 : : {
3441 : 6 : sFileFilters += createFileFilter_( QObject::tr( "SEG-Y" ), QStringLiteral( "*.sgy *.segy" ) );
3442 : 9 : sExtensions << QStringLiteral( "sgy" ) << QStringLiteral( "segy" );
3443 : 3 : }
3444 : 117 : else if ( driverName.startsWith( QLatin1String( "SEGUKOOA" ) ) )
3445 : : {
3446 : 6 : sFileFilters += createFileFilter_( QObject::tr( "SEG-P1" ), QStringLiteral( "*.seg *.seg1 *.sp1" ) );
3447 : 6 : sFileFilters += createFileFilter_( QObject::tr( "UKOOA P1/90" ), QStringLiteral( "*.uko *.ukooa" ) );
3448 : 18 : sExtensions << QStringLiteral( "seg" ) << QStringLiteral( "seg1" ) << QStringLiteral( "sp1" ) << QStringLiteral( "uko" ) << QStringLiteral( "ukooa" );
3449 : 3 : }
3450 : : else
3451 : : {
3452 : 114 : if ( driverName == QLatin1String( "CAD" ) || driverName == QLatin1String( "DWG" ) )
3453 : : {
3454 : 3 : if ( dwgFound )
3455 : 0 : continue;
3456 : 3 : dwgFound = true;
3457 : 3 : }
3458 : :
3459 : 114 : QString myGdalDriverExtensions = GDALGetMetadataItem( driver, GDAL_DMD_EXTENSIONS, "" );
3460 : 114 : QString myGdalDriverLongName = GDALGetMetadataItem( driver, GDAL_DMD_LONGNAME, "" );
3461 : 114 : if ( !( myGdalDriverExtensions.isEmpty() || myGdalDriverLongName.isEmpty() ) )
3462 : : {
3463 : : #if QT_VERSION < QT_VERSION_CHECK(5, 15, 0)
3464 : : const QStringList splitExtensions = myGdalDriverExtensions.split( ' ', QString::SkipEmptyParts );
3465 : : #else
3466 : 51 : const QStringList splitExtensions = myGdalDriverExtensions.split( ' ', Qt::SkipEmptyParts );
3467 : : #endif
3468 : 51 : QString glob;
3469 : :
3470 : 117 : for ( const QString &ext : splitExtensions )
3471 : : {
3472 : 66 : sExtensions << ext;
3473 : 66 : if ( !glob.isEmpty() )
3474 : 15 : glob += QLatin1Char( ' ' );
3475 : 66 : glob += "*." + ext;
3476 : : }
3477 : :
3478 : 99 : if ( driverName == QLatin1String( "JPEG2000" ) ||
3479 : 48 : driverName.startsWith( QLatin1String( "JP2" ) ) )
3480 : : {
3481 : : // Skip over JPEG2000 drivers, as their vector capabilities are just
3482 : : // a marginal use case
3483 : 9 : continue;
3484 : : }
3485 : :
3486 : 42 : sFileFilters += createFileFilter_( myGdalDriverLongName, glob );
3487 : :
3488 : 51 : }
3489 : : else
3490 : : {
3491 : : // NOP, we don't know anything about the current driver
3492 : : // with regards to a proper file filter string
3493 : 63 : QgsDebugMsgLevel( QStringLiteral( "Unknown driver %1 for file filters." ).arg( driverName ), 2 );
3494 : : }
3495 : 114 : }
3496 : :
3497 : 249 : } // each loaded OGR driver
3498 : :
3499 : : // sort file filters alphabetically
3500 : 3 : QgsDebugMsgLevel( "myFileFilters: " + sFileFilters, 2 );
3501 : : #if QT_VERSION < QT_VERSION_CHECK(5, 15, 0)
3502 : : QStringList filters = sFileFilters.split( QStringLiteral( ";;" ), QString::SkipEmptyParts );
3503 : : #else
3504 : 6 : QStringList filters = sFileFilters.split( QStringLiteral( ";;" ), Qt::SkipEmptyParts );
3505 : : #endif
3506 : 3 : filters.sort();
3507 : 3 : sFileFilters = filters.join( QLatin1String( ";;" ) ) + ";;";
3508 : 3 : QgsDebugMsgLevel( "myFileFilters: " + sFileFilters, 2 );
3509 : :
3510 : : // VSIFileHandler (.zip and .gz files) - second
3511 : : // see http://trac.osgeo.org/gdal/wiki/UserDocs/ReadInZip
3512 : : // Requires GDAL>=1.6.0 with libz support, let's assume we have it.
3513 : : // This does not work for some file types, see VSIFileHandler doc.
3514 : 3 : QgsSettings settings;
3515 : 6 : if ( settings.value( QStringLiteral( "qgis/scanZipInBrowser2" ), "basic" ).toString() != QLatin1String( "no" ) )
3516 : : {
3517 : 6 : sFileFilters.prepend( createFileFilter_( QObject::tr( "GDAL/OGR VSIFileHandler" ), QStringLiteral( "*.zip *.gz *.tar *.tar.gz *.tgz" ) ) );
3518 : 18 : sExtensions << QStringLiteral( "zip" ) << QStringLiteral( "gz" ) << QStringLiteral( "tar" ) << QStringLiteral( "tar.gz" ) << QStringLiteral( "tgz" );
3519 : 3 : }
3520 : :
3521 : : // can't forget the all supported case
3522 : 3 : QStringList exts;
3523 : 285 : for ( const QString &ext : std::as_const( sExtensions ) )
3524 : 564 : exts << QStringLiteral( "*.%1 *.%2" ).arg( ext, ext.toUpper() );
3525 : 6 : sFileFilters.prepend( QObject::tr( "All supported files" ) + QStringLiteral( " (%1);;" ).arg( exts.join( QLatin1Char( ' ' ) ) ) );
3526 : :
3527 : : // can't forget the default case - first
3528 : 3 : sFileFilters.prepend( QObject::tr( "All files" ) + " (*);;" );
3529 : :
3530 : :
3531 : : // cleanup
3532 : 3 : if ( sFileFilters.endsWith( QLatin1String( ";;" ) ) ) sFileFilters.chop( 2 );
3533 : :
3534 : 3 : QgsDebugMsgLevel( "myFileFilters: " + sFileFilters, 2 );
3535 : 3 : }
3536 : :
3537 : 12 : if ( type == QLatin1String( "file" ) )
3538 : : {
3539 : 3 : return sFileFilters;
3540 : : }
3541 : 9 : if ( type == QLatin1String( "database" ) )
3542 : : {
3543 : 3 : return sDatabaseDrivers;
3544 : : }
3545 : 6 : if ( type == QLatin1String( "protocol" ) )
3546 : : {
3547 : 3 : return sProtocolDrivers;
3548 : : }
3549 : 3 : if ( type == QLatin1String( "directory" ) )
3550 : : {
3551 : 3 : return sDirectoryDrivers;
3552 : : }
3553 : 0 : if ( type == QLatin1String( "extensions" ) )
3554 : : {
3555 : 0 : return sExtensions.join( QLatin1Char( '|' ) );
3556 : : }
3557 : 0 : if ( type == QLatin1String( "directory_extensions" ) )
3558 : : {
3559 : 0 : return sDirectoryExtensions.join( QLatin1Char( '|' ) );
3560 : : }
3561 : 0 : if ( type == QLatin1String( "wildcards" ) )
3562 : : {
3563 : 0 : return sWildcards.join( QLatin1Char( '|' ) );
3564 : : }
3565 : : else
3566 : : {
3567 : 0 : return QString();
3568 : : }
3569 : 12 : }
3570 : :
3571 : 216 : QVariantMap QgsOgrProviderMetadata::decodeUri( const QString &uri ) const
3572 : : {
3573 : 216 : QString path = uri;
3574 : 216 : QString layerName;
3575 : 216 : QString subset;
3576 : 216 : QString geometryType;
3577 : 216 : QStringList openOptions;
3578 : 216 : QString databaseName;
3579 : :
3580 : 216 : int layerId = -1;
3581 : :
3582 : 216 : QString vsiPrefix = qgsVsiPrefix( path );
3583 : 216 : QString vsiSuffix;
3584 : 216 : if ( path.startsWith( vsiPrefix, Qt::CaseInsensitive ) )
3585 : : {
3586 : 216 : path = path.mid( vsiPrefix.count() );
3587 : 216 : if ( vsiPrefix == QLatin1String( "/vsizip/" ) )
3588 : : {
3589 : 0 : const QRegularExpression vsiRegex( QStringLiteral( "(?:\\.zip|\\.tar|\\.gz|\\.tar\\.gz|\\.tgz)([^|]*)" ) );
3590 : 0 : QRegularExpressionMatch match = vsiRegex.match( path );
3591 : 0 : if ( match.hasMatch() )
3592 : : {
3593 : 0 : vsiSuffix = match.captured( 1 );
3594 : 0 : path = path.remove( match.capturedStart( 1 ), match.capturedLength( 1 ) );
3595 : 0 : }
3596 : 0 : }
3597 : 216 : }
3598 : : else
3599 : : {
3600 : 0 : vsiPrefix.clear();
3601 : : }
3602 : :
3603 : 216 : if ( path.contains( '|' ) )
3604 : : {
3605 : 0 : const QRegularExpression geometryTypeRegex( QStringLiteral( "\\|geometrytype=([a-zA-Z0-9]*)" ), QRegularExpression::PatternOption::CaseInsensitiveOption );
3606 : 0 : const QRegularExpression layerNameRegex( QStringLiteral( "\\|layername=([^|]*)" ), QRegularExpression::PatternOption::CaseInsensitiveOption );
3607 : 0 : const QRegularExpression layerIdRegex( QStringLiteral( "\\|layerid=([^|]*)" ), QRegularExpression::PatternOption::CaseInsensitiveOption );
3608 : 0 : const QRegularExpression subsetRegex( QStringLiteral( "\\|subset=((?:.*[\r\n]*)*)\\Z" ) );
3609 : 0 : const QRegularExpression openOptionRegex( QStringLiteral( "\\|option:([^|]*)" ) );
3610 : :
3611 : :
3612 : : // we first try to split off the geometry type component, if that's present. That's a known quantity which
3613 : : // will never be more than a-z characters
3614 : 0 : QRegularExpressionMatch match = geometryTypeRegex.match( path );
3615 : 0 : if ( match.hasMatch() )
3616 : : {
3617 : 0 : geometryType = match.captured( 1 );
3618 : 0 : path = path.remove( match.capturedStart( 0 ), match.capturedLength( 0 ) );
3619 : 0 : }
3620 : :
3621 : : // next, we try to find and strip out the layerid/layername component. (This logic is based on the assumption
3622 : : // that a layer name doesn't contain a | character!)
3623 : : // we prefer layer names over layer ids, since they are persistent..
3624 : 0 : match = layerNameRegex.match( path );
3625 : 0 : if ( match.hasMatch() )
3626 : : {
3627 : 0 : layerName = match.captured( 1 );
3628 : 0 : path = path.remove( match.capturedStart( 0 ), match.capturedLength( 0 ) );
3629 : 0 : }
3630 : :
3631 : 0 : match = layerIdRegex.match( path );
3632 : 0 : if ( match.hasMatch() )
3633 : : {
3634 : 0 : layerId = match.captured( 1 ).toInt();
3635 : 0 : path = path.remove( match.capturedStart( 0 ), match.capturedLength( 0 ) );
3636 : 0 : }
3637 : :
3638 : 0 : while ( true )
3639 : : {
3640 : 0 : match = openOptionRegex.match( path );
3641 : 0 : if ( match.hasMatch() )
3642 : : {
3643 : 0 : openOptions << match.captured( 1 );
3644 : 0 : path = path.remove( match.capturedStart( 0 ), match.capturedLength( 0 ) );
3645 : 0 : }
3646 : : else
3647 : : {
3648 : 0 : break;
3649 : : }
3650 : : }
3651 : :
3652 : : // lastly, try to parse out the subset component. This is the biggest unknown, because it's
3653 : : // quite possible that a subset string will contain a | character. If we've already parsed
3654 : : // out all the other known |xxx=yyy tags, then we can safely assume that everything from "|subset=" to the
3655 : : // end of the path is part of the subset filter
3656 : 0 : match = subsetRegex.match( path );
3657 : 0 : if ( match.hasMatch() )
3658 : : {
3659 : 0 : subset = match.captured( 1 );
3660 : 0 : path = path.remove( match.capturedStart( 0 ), match.capturedLength( 0 ) );
3661 : 0 : }
3662 : 0 : }
3663 : :
3664 : : // Handles DB connections extracting database name if possible
3665 : : // Example: MySQL:database_name,host=localhost,port=3306 authcfg='f8wwfx8'
3666 : 648 : if ( uri.startsWith( QStringLiteral( "MySQL" ), Qt::CaseSensitivity::CaseInsensitive ) ||
3667 : 432 : uri.startsWith( QStringLiteral( "PostgreSQL" ), Qt::CaseSensitivity::CaseInsensitive ) ||
3668 : 432 : uri.startsWith( QStringLiteral( "MSSQL" ), Qt::CaseSensitivity::CaseInsensitive ) ||
3669 : 432 : uri.startsWith( QStringLiteral( "ODBC" ), Qt::CaseSensitivity::CaseInsensitive ) ||
3670 : 432 : uri.startsWith( QStringLiteral( "PGeo" ), Qt::CaseSensitivity::CaseInsensitive ) ||
3671 : 432 : uri.startsWith( QStringLiteral( "SDE" ), Qt::CaseSensitivity::CaseInsensitive ) ||
3672 : 432 : uri.startsWith( QStringLiteral( "OGDI" ), Qt::CaseSensitivity::CaseInsensitive ) ||
3673 : 432 : uri.startsWith( QStringLiteral( "Ingres" ), Qt::CaseSensitivity::CaseInsensitive ) ||
3674 : 648 : uri.startsWith( QStringLiteral( "IDB" ), Qt::CaseSensitivity::CaseInsensitive ) ||
3675 : 432 : uri.startsWith( QStringLiteral( "OCI" ), Qt::CaseSensitivity::CaseInsensitive ) )
3676 : : {
3677 : 0 : auto parts( path.split( ':' ) );
3678 : 0 : if ( parts.count( ) > 1 )
3679 : : {
3680 : 0 : auto dataParts( parts.at( 1 ).split( ',' ) );
3681 : 0 : if ( dataParts.count() > 0 )
3682 : 0 : databaseName = dataParts.at( 0 );
3683 : 0 : }
3684 : 0 : }
3685 : :
3686 : 216 : QVariantMap uriComponents;
3687 : 432 : uriComponents.insert( QStringLiteral( "path" ), path );
3688 : 432 : uriComponents.insert( QStringLiteral( "layerName" ), layerName );
3689 : 432 : uriComponents.insert( QStringLiteral( "layerId" ), layerId > -1 && layerName.isEmpty() ? layerId : QVariant() ) ;
3690 : 216 : if ( !subset.isEmpty() )
3691 : 0 : uriComponents.insert( QStringLiteral( "subset" ), subset );
3692 : 216 : if ( !geometryType.isEmpty() )
3693 : 0 : uriComponents.insert( QStringLiteral( "geometryType" ), geometryType );
3694 : 216 : if ( !databaseName.isEmpty() )
3695 : 0 : uriComponents.insert( QStringLiteral( "databaseName" ), databaseName );
3696 : 216 : if ( !openOptions.isEmpty() )
3697 : 0 : uriComponents.insert( QStringLiteral( "openOptions" ), openOptions );
3698 : 216 : if ( !vsiPrefix.isEmpty() )
3699 : 0 : uriComponents.insert( QStringLiteral( "vsiPrefix" ), vsiPrefix );
3700 : 216 : if ( !vsiSuffix.isEmpty() )
3701 : 0 : uriComponents.insert( QStringLiteral( "vsiSuffix" ), vsiSuffix );
3702 : 216 : return uriComponents;
3703 : 216 : }
3704 : :
3705 : 180 : QString QgsOgrProviderMetadata::encodeUri( const QVariantMap &parts ) const
3706 : : {
3707 : 360 : const QString vsiPrefix = parts.value( QStringLiteral( "vsiPrefix" ) ).toString();
3708 : 360 : const QString vsiSuffix = parts.value( QStringLiteral( "vsiSuffix" ) ).toString();
3709 : 360 : const QString path = parts.value( QStringLiteral( "path" ) ).toString();
3710 : 360 : const QString layerName = parts.value( QStringLiteral( "layerName" ) ).toString();
3711 : 360 : const QString layerId = parts.value( QStringLiteral( "layerId" ) ).toString();
3712 : 360 : const QString subset = parts.value( QStringLiteral( "subset" ) ).toString();
3713 : 360 : const QString geometryType = parts.value( QStringLiteral( "geometryType" ) ).toString();
3714 : 360 : const QStringList openOptions = parts.value( QStringLiteral( "openOptions" ) ).toStringList();
3715 : 360 : QString uri = vsiPrefix + path + vsiSuffix
3716 : 180 : + ( !layerName.isEmpty() ? QStringLiteral( "|layername=%1" ).arg( layerName ) : !layerId.isEmpty() ? QStringLiteral( "|layerid=%1" ).arg( layerId ) : QString() )
3717 : 180 : + ( !geometryType.isEmpty() ? QStringLiteral( "|geometrytype=%1" ).arg( geometryType ) : QString() );
3718 : 180 : for ( const QString &openOption : openOptions )
3719 : : {
3720 : 0 : uri += QLatin1String( "|option:" );
3721 : 0 : uri += openOption;
3722 : : }
3723 : 180 : if ( !subset.isEmpty() )
3724 : 0 : uri += QStringLiteral( "|subset=%1" ).arg( subset );
3725 : 180 : return uri;
3726 : 180 : }
3727 : :
3728 : 3 : QString QgsOgrProviderUtils::fileVectorFilters()
3729 : : {
3730 : 6 : return createFilters( QStringLiteral( "file" ) );
3731 : 0 : }
3732 : :
3733 : 3 : QString QgsOgrProviderUtils::databaseDrivers()
3734 : : {
3735 : 6 : return createFilters( QStringLiteral( "database" ) );
3736 : 0 : }
3737 : :
3738 : 3 : QString QgsOgrProviderUtils::protocolDrivers()
3739 : : {
3740 : 6 : return createFilters( QStringLiteral( "protocol" ) );
3741 : 0 : }
3742 : :
3743 : :
3744 : 3 : QString QgsOgrProviderUtils::directoryDrivers()
3745 : : {
3746 : 6 : return createFilters( QStringLiteral( "directory" ) );
3747 : 0 : }
3748 : :
3749 : 0 : QStringList QgsOgrProviderUtils::fileExtensions()
3750 : : {
3751 : 0 : return createFilters( QStringLiteral( "extensions" ) ).split( '|' );
3752 : 0 : }
3753 : :
3754 : 0 : QStringList QgsOgrProviderUtils::directoryExtensions()
3755 : : {
3756 : 0 : return createFilters( QStringLiteral( "directory_extensions" ) ).split( '|' );
3757 : 0 : }
3758 : :
3759 : 0 : QStringList QgsOgrProviderUtils::wildcards()
3760 : : {
3761 : 0 : return createFilters( QStringLiteral( "wildcards" ) ).split( '|' );
3762 : 0 : }
3763 : :
3764 : :
3765 : : /**
3766 : : * Class factory to return a pointer to a newly created
3767 : : * QgsOgrProvider object
3768 : : */
3769 : 64 : QgsOgrProvider *QgsOgrProviderMetadata::createProvider( const QString &uri, const QgsDataProvider::ProviderOptions &options,
3770 : : QgsDataProvider::ReadFlags flags )
3771 : : {
3772 : 64 : return new QgsOgrProvider( uri, options, flags );
3773 : 0 : }
3774 : :
3775 : : /**
3776 : : * Creates an empty data source
3777 : : \param uri location to store the file(s)
3778 : : \param format data format (e.g. "ESRI Shapefile")
3779 : : \param vectortype point/line/polygon or multitypes
3780 : : \param attributes a list of name/type pairs for the initial attributes
3781 : : \return true in case of success*/
3782 : 0 : bool QgsOgrProviderUtils::createEmptyDataSource( const QString &uri,
3783 : : const QString &format,
3784 : : const QString &encoding,
3785 : : QgsWkbTypes::Type vectortype,
3786 : : const QList< QPair<QString, QString> > &attributes,
3787 : : const QgsCoordinateReferenceSystem &srs,
3788 : : QString &errorMessage )
3789 : : {
3790 : 0 : QgsDebugMsgLevel( QStringLiteral( "Creating empty vector layer with format: %1" ).arg( format ), 2 );
3791 : 0 : errorMessage.clear();
3792 : :
3793 : 0 : QgsApplication::registerOgrDrivers();
3794 : 0 : OGRSFDriverH driver = OGRGetDriverByName( format.toLatin1() );
3795 : 0 : if ( !driver )
3796 : : {
3797 : 0 : return false;
3798 : : }
3799 : :
3800 : 0 : QString driverName = GDALGetDriverShortName( driver );
3801 : :
3802 : 0 : if ( driverName == QLatin1String( "ESRI Shapefile" ) )
3803 : : {
3804 : 0 : if ( !uri.endsWith( QLatin1String( ".shp" ), Qt::CaseInsensitive ) &&
3805 : 0 : !uri.endsWith( QLatin1String( ".shz" ), Qt::CaseInsensitive ) &&
3806 : 0 : !uri.endsWith( QLatin1String( ".dbf" ), Qt::CaseInsensitive ) )
3807 : : {
3808 : 0 : errorMessage = QObject::tr( "URI %1 doesn't end with .shp or .dbf" ).arg( uri );
3809 : 0 : QgsDebugMsg( errorMessage );
3810 : 0 : return false;
3811 : : }
3812 : :
3813 : : // check for duplicate fieldnames
3814 : 0 : QSet<QString> fieldNames;
3815 : 0 : QList<QPair<QString, QString> >::const_iterator fldIt;
3816 : 0 : for ( fldIt = attributes.begin(); fldIt != attributes.end(); ++fldIt )
3817 : : {
3818 : 0 : QString name = fldIt->first.left( 10 );
3819 : 0 : if ( fieldNames.contains( name ) )
3820 : : {
3821 : 0 : errorMessage = QObject::tr( "Duplicate field (10 significant characters): %1" ).arg( name );
3822 : 0 : QgsMessageLog::logMessage( errorMessage, QObject::tr( "OGR" ) );
3823 : 0 : return false;
3824 : : }
3825 : 0 : fieldNames << name;
3826 : 0 : }
3827 : :
3828 : 0 : QgsVectorFileWriter::deleteShapeFile( uri );
3829 : 0 : }
3830 : : else
3831 : : {
3832 : 0 : QFile::remove( uri );
3833 : : }
3834 : :
3835 : 0 : gdal::dataset_unique_ptr dataSource;
3836 : 0 : dataSource.reset( OGR_Dr_CreateDataSource( driver, uri.toUtf8().constData(), nullptr ) );
3837 : 0 : if ( !dataSource )
3838 : : {
3839 : 0 : errorMessage = QObject::tr( "Creating the data source %1 failed: %2" ).arg( uri, QString::fromUtf8( CPLGetLastErrorMsg() ) );
3840 : 0 : QgsMessageLog::logMessage( errorMessage, QObject::tr( "OGR" ) );
3841 : 0 : return false;
3842 : : }
3843 : :
3844 : : //consider spatial reference system
3845 : 0 : OGRSpatialReferenceH reference = nullptr;
3846 : :
3847 : 0 : QgsCoordinateReferenceSystem mySpatialRefSys;
3848 : 0 : if ( srs.isValid() )
3849 : : {
3850 : 0 : mySpatialRefSys = srs;
3851 : 0 : }
3852 : : else
3853 : : {
3854 : 0 : mySpatialRefSys.validate();
3855 : : }
3856 : :
3857 : 0 : QString myWkt = mySpatialRefSys.toWkt( QgsCoordinateReferenceSystem::WKT_PREFERRED_GDAL );
3858 : :
3859 : 0 : if ( !myWkt.isNull() && myWkt.length() != 0 )
3860 : : {
3861 : 0 : reference = OSRNewSpatialReference( myWkt.toLocal8Bit().data() );
3862 : 0 : }
3863 : :
3864 : : // Map the qgis geometry type to the OGR geometry type
3865 : 0 : OGRwkbGeometryType OGRvectortype = wkbUnknown;
3866 : 0 : switch ( vectortype )
3867 : : {
3868 : : case QgsWkbTypes::GeometryCollection:
3869 : : case QgsWkbTypes::GeometryCollectionZ:
3870 : : case QgsWkbTypes::GeometryCollectionM:
3871 : : case QgsWkbTypes::GeometryCollectionZM:
3872 : : case QgsWkbTypes::Unknown:
3873 : : {
3874 : 0 : errorMessage = QObject::tr( "Unknown vector type of %1" ).arg( static_cast< int >( vectortype ) );
3875 : 0 : QgsMessageLog::logMessage( errorMessage, QObject::tr( "OGR" ) );
3876 : 0 : return false;
3877 : : }
3878 : :
3879 : : default:
3880 : 0 : OGRvectortype = QgsOgrProviderUtils::ogrTypeFromQgisType( vectortype );
3881 : 0 : }
3882 : :
3883 : 0 : char **papszOptions = nullptr;
3884 : 0 : if ( driverName == QLatin1String( "ESRI Shapefile" ) )
3885 : : {
3886 : 0 : papszOptions = CSLSetNameValue( papszOptions, "ENCODING", QgsVectorFileWriter::convertCodecNameForEncodingOption( encoding ).toLocal8Bit().data() );
3887 : : // OGR Shapefile fails to create fields if given encoding is not supported by its side
3888 : : // so disable encoding conversion of OGR Shapefile layer
3889 : 0 : CPLSetConfigOption( "SHAPE_ENCODING", "" );
3890 : 0 : }
3891 : :
3892 : : OGRLayerH layer;
3893 : 0 : layer = GDALDatasetCreateLayer( dataSource.get(), QFileInfo( uri ).completeBaseName().toUtf8().constData(), reference, OGRvectortype, papszOptions );
3894 : 0 : CSLDestroy( papszOptions );
3895 : :
3896 : 0 : if ( !layer )
3897 : : {
3898 : 0 : errorMessage = QString::fromUtf8( CPLGetLastErrorMsg() );
3899 : 0 : QgsMessageLog::logMessage( errorMessage, QObject::tr( "OGR" ) );
3900 : 0 : return false;
3901 : : }
3902 : :
3903 : : //create the attribute fields
3904 : :
3905 : 0 : QTextCodec *codec = QTextCodec::codecForName( encoding.toLocal8Bit().data() );
3906 : 0 : if ( !codec )
3907 : : {
3908 : : // fall back to "System" codec
3909 : 0 : codec = QTextCodec::codecForLocale();
3910 : : Q_ASSERT( codec );
3911 : 0 : }
3912 : :
3913 : 0 : for ( QList<QPair<QString, QString> >::const_iterator it = attributes.begin(); it != attributes.end(); ++it )
3914 : : {
3915 : 0 : QStringList fields = it->second.split( ';' );
3916 : :
3917 : 0 : if ( fields.isEmpty() )
3918 : 0 : continue;
3919 : :
3920 : 0 : int width = fields.size() > 1 ? fields[1].toInt() : -1;
3921 : 0 : int precision = fields.size() > 2 ? fields[2].toInt() : -1;
3922 : 0 : if ( precision > 0 )
3923 : 0 : width += 1;
3924 : :
3925 : : OGRFieldDefnH field;
3926 : 0 : if ( fields[0] == QLatin1String( "Real" ) )
3927 : : {
3928 : 0 : if ( width < 0 )
3929 : 0 : width = 32;
3930 : 0 : if ( precision < 0 )
3931 : 0 : precision = 3;
3932 : :
3933 : 0 : field = OGR_Fld_Create( codec->fromUnicode( it->first ).constData(), OFTReal );
3934 : 0 : OGR_Fld_SetWidth( field, width );
3935 : 0 : OGR_Fld_SetPrecision( field, precision );
3936 : 0 : }
3937 : 0 : else if ( fields[0] == QLatin1String( "Integer" ) )
3938 : : {
3939 : 0 : if ( width < 0 || width > 10 )
3940 : 0 : width = 10;
3941 : :
3942 : 0 : field = OGR_Fld_Create( codec->fromUnicode( it->first ).constData(), OFTInteger );
3943 : : // limit to 10. otherwise OGR sets it to 11 and recognizes as OFTDouble later
3944 : 0 : OGR_Fld_SetWidth( field, width );
3945 : 0 : }
3946 : 0 : else if ( fields[0] == QLatin1String( "String" ) )
3947 : : {
3948 : 0 : if ( width < 0 || width > 255 )
3949 : 0 : width = 255;
3950 : :
3951 : 0 : field = OGR_Fld_Create( codec->fromUnicode( it->first ).constData(), OFTString );
3952 : 0 : OGR_Fld_SetWidth( field, width );
3953 : 0 : }
3954 : 0 : else if ( fields[0] == QLatin1String( "Date" ) )
3955 : : {
3956 : 0 : field = OGR_Fld_Create( codec->fromUnicode( it->first ).constData(), OFTDate );
3957 : 0 : }
3958 : 0 : else if ( fields[0] == QLatin1String( "Time" ) )
3959 : : {
3960 : 0 : field = OGR_Fld_Create( codec->fromUnicode( it->first ).constData(), OFTTime );
3961 : 0 : }
3962 : 0 : else if ( fields[0] == QLatin1String( "DateTime" ) )
3963 : : {
3964 : 0 : field = OGR_Fld_Create( codec->fromUnicode( it->first ).constData(), OFTDateTime );
3965 : 0 : }
3966 : : else
3967 : : {
3968 : 0 : QgsMessageLog::logMessage( QObject::tr( "field %1 with unsupported type %2 skipped" ).arg( it->first, fields[0] ), QObject::tr( "OGR" ) );
3969 : 0 : continue;
3970 : : }
3971 : :
3972 : 0 : if ( OGR_L_CreateField( layer, field, true ) != OGRERR_NONE )
3973 : : {
3974 : 0 : QgsMessageLog::logMessage( QObject::tr( "creation of field %1 failed" ).arg( it->first ), QObject::tr( "OGR" ) );
3975 : 0 : }
3976 : 0 : }
3977 : :
3978 : 0 : dataSource.reset();
3979 : :
3980 : 0 : if ( driverName == QLatin1String( "ESRI Shapefile" ) )
3981 : : {
3982 : 0 : int index = uri.indexOf( QLatin1String( ".shp" ), Qt::CaseInsensitive );
3983 : 0 : if ( index > 0 )
3984 : : {
3985 : 0 : QString layerName = uri.left( index );
3986 : 0 : QFile prjFile( layerName + ".qpj" );
3987 : 0 : if ( prjFile.exists() )
3988 : 0 : prjFile.remove();
3989 : 0 : }
3990 : 0 : }
3991 : :
3992 : 0 : QgsDebugMsgLevel( QStringLiteral( "GDAL Version number %1" ).arg( GDAL_VERSION_NUM ), 2 );
3993 : 0 : if ( reference )
3994 : : {
3995 : 0 : OSRRelease( reference );
3996 : 0 : }
3997 : 0 : return true;
3998 : 0 : }
3999 : :
4000 : 3 : QList<QgsDataItemProvider *> QgsOgrProviderMetadata::dataItemProviders() const
4001 : : {
4002 : 3 : QList< QgsDataItemProvider * > providers;
4003 : 3 : providers << new QgsOgrDataItemProvider;
4004 : 3 : providers << new QgsGeoPackageDataItemProvider;
4005 : 3 : return providers;
4006 : 3 : }
4007 : :
4008 : 197 : QgsCoordinateReferenceSystem QgsOgrProvider::crs() const
4009 : : {
4010 : 197 : QgsCoordinateReferenceSystem srs;
4011 : 197 : if ( !mValid || ( mOGRGeomType == wkbNone ) )
4012 : 0 : return srs;
4013 : :
4014 : 197 : if ( OGRSpatialReferenceH spatialRefSys = mOgrLayer->GetSpatialRef() )
4015 : : {
4016 : 3 : srs = QgsOgrUtils::OGRSpatialReferenceToCrs( spatialRefSys );
4017 : 3 : }
4018 : : else
4019 : : {
4020 : 194 : QgsDebugMsg( QStringLiteral( "no spatial reference found" ) );
4021 : : }
4022 : :
4023 : 197 : return srs;
4024 : 197 : }
4025 : :
4026 : 0 : QSet<QVariant> QgsOgrProvider::uniqueValues( int index, int limit ) const
4027 : : {
4028 : 0 : QSet<QVariant> uniqueValues;
4029 : :
4030 : 0 : if ( !mValid || index < 0 || index >= mAttributeFields.count() )
4031 : 0 : return uniqueValues;
4032 : :
4033 : 0 : const QgsField fld = mAttributeFields.at( index );
4034 : 0 : if ( fld.name().isNull() )
4035 : : {
4036 : 0 : return uniqueValues; //not a provider field
4037 : : }
4038 : :
4039 : 0 : QgsCPLHTTPFetchOverrider oCPLHTTPFetcher( mAuthCfg );
4040 : 0 : QgsSetCPLHTTPFetchOverriderInitiatorClass( oCPLHTTPFetcher, QStringLiteral( "QgsOgrProvider" ) );
4041 : :
4042 : 0 : QByteArray sql = "SELECT DISTINCT " + quotedIdentifier( textEncoding()->fromUnicode( fld.name() ) );
4043 : :
4044 : : // GPKG/SQLite fid
4045 : : // For GPKG and SQLITE drivers PK fields are not exposed as real fields, (and OGR_F_GetFID only
4046 : : // works with GPKG), so we are adding an extra column that will become index 0
4047 : : // See https://github.com/qgis/QGIS/issues/29129
4048 : 0 : if ( ( mGDALDriverName == QLatin1String( "GPKG" ) || mGDALDriverName == QLatin1String( "SQLite" ) )
4049 : 0 : && mFirstFieldIsFid && index == 0 )
4050 : : {
4051 : 0 : sql += ", " + quotedIdentifier( textEncoding()->fromUnicode( fld.name() ) ) + " AS fid2";
4052 : 0 : }
4053 : :
4054 : 0 : sql += " FROM " + quotedIdentifier( mOgrLayer->name() );
4055 : :
4056 : 0 : if ( !mSubsetString.isEmpty() )
4057 : : {
4058 : 0 : sql += " WHERE " + textEncoding()->fromUnicode( mSubsetString );
4059 : 0 : }
4060 : :
4061 : 0 : sql += " ORDER BY " + quotedIdentifier( textEncoding()->fromUnicode( fld.name() ) ) + " ASC";
4062 : :
4063 : 0 : QgsDebugMsgLevel( QStringLiteral( "SQL: %1" ).arg( textEncoding()->toUnicode( sql ) ), 2 );
4064 : 0 : QgsOgrLayerUniquePtr l = mOgrLayer->ExecuteSQL( sql );
4065 : 0 : if ( !l )
4066 : : {
4067 : 0 : QgsDebugMsg( QStringLiteral( "Failed to execute SQL" ) );
4068 : 0 : return QgsVectorDataProvider::uniqueValues( index, limit );
4069 : : }
4070 : :
4071 : 0 : gdal::ogr_feature_unique_ptr f;
4072 : 0 : bool ok = false;
4073 : 0 : while ( f.reset( l->GetNextFeature() ), f )
4074 : : {
4075 : 0 : const QVariant res = QgsOgrUtils::getOgrFeatureAttribute( f.get(), fld, 0, textEncoding(), &ok );
4076 : 0 : if ( ok )
4077 : 0 : uniqueValues << res;
4078 : :
4079 : 0 : if ( limit >= 0 && uniqueValues.size() >= limit )
4080 : 0 : break;
4081 : 0 : }
4082 : :
4083 : 0 : return uniqueValues;
4084 : 0 : }
4085 : :
4086 : 0 : QStringList QgsOgrProvider::uniqueStringsMatching( int index, const QString &substring, int limit, QgsFeedback *feedback ) const
4087 : : {
4088 : 0 : QStringList results;
4089 : :
4090 : 0 : if ( !mValid || index < 0 || index >= mAttributeFields.count() )
4091 : 0 : return results;
4092 : :
4093 : 0 : QgsField fld = mAttributeFields.at( index );
4094 : 0 : if ( fld.name().isNull() )
4095 : : {
4096 : 0 : return results; //not a provider field
4097 : : }
4098 : :
4099 : 0 : QgsCPLHTTPFetchOverrider oCPLHTTPFetcher( mAuthCfg );
4100 : 0 : QgsSetCPLHTTPFetchOverriderInitiatorClass( oCPLHTTPFetcher, QStringLiteral( "QgsOgrProvider" ) );
4101 : :
4102 : : // uniqueStringsMatching() is supposed to be case insensitive, so use the
4103 : : // ILIKE operator when it is available.
4104 : : // Prior to GDAL 3.1, with OGR SQL, LIKE behaved like ILIKE
4105 : 0 : bool supportsILIKE = false;
4106 : : {
4107 : 0 : QByteArray sql = "SELECT 1 FROM ";
4108 : 0 : sql += quotedIdentifier( mOgrLayer->name() );
4109 : 0 : sql += " WHERE 'a' ILIKE 'A' LIMIT 1";
4110 : 0 : QgsOgrLayerUniquePtr l = mOgrLayer->ExecuteSQL( sql );
4111 : 0 : if ( l )
4112 : : {
4113 : 0 : gdal::ogr_feature_unique_ptr f;
4114 : 0 : f.reset( l->GetNextFeature() );
4115 : 0 : supportsILIKE = f != nullptr;
4116 : 0 : }
4117 : 0 : }
4118 : :
4119 : 0 : QByteArray sql = "SELECT DISTINCT " + quotedIdentifier( textEncoding()->fromUnicode( fld.name() ) );
4120 : 0 : sql += " FROM " + quotedIdentifier( mOgrLayer->name() );
4121 : :
4122 : 0 : sql += " WHERE " + quotedIdentifier( textEncoding()->fromUnicode( fld.name() ) );
4123 : 0 : if ( supportsILIKE )
4124 : 0 : sql += " ILIKE '%";
4125 : : else
4126 : 0 : sql += " LIKE '%";
4127 : 0 : sql += textEncoding()->fromUnicode( substring ) + "%'";
4128 : :
4129 : 0 : if ( !mSubsetString.isEmpty() )
4130 : : {
4131 : 0 : sql += " AND (" + textEncoding()->fromUnicode( mSubsetString ) + ')';
4132 : 0 : }
4133 : :
4134 : 0 : sql += " ORDER BY " + quotedIdentifier( textEncoding()->fromUnicode( fld.name() ) ) + " ASC";
4135 : :
4136 : 0 : QgsDebugMsgLevel( QStringLiteral( "SQL: %1" ).arg( textEncoding()->toUnicode( sql ) ), 2 );
4137 : 0 : QgsOgrLayerUniquePtr l = mOgrLayer->ExecuteSQL( sql );
4138 : 0 : if ( !l )
4139 : : {
4140 : 0 : QgsDebugMsg( QStringLiteral( "Failed to execute SQL" ) );
4141 : 0 : return QgsVectorDataProvider::uniqueStringsMatching( index, substring, limit, feedback );
4142 : : }
4143 : :
4144 : 0 : gdal::ogr_feature_unique_ptr f;
4145 : 0 : while ( f.reset( l->GetNextFeature() ), f )
4146 : : {
4147 : 0 : if ( OGR_F_IsFieldSetAndNotNull( f.get(), 0 ) )
4148 : 0 : results << textEncoding()->toUnicode( OGR_F_GetFieldAsString( f.get(), 0 ) );
4149 : :
4150 : 0 : if ( ( limit >= 0 && results.size() >= limit ) || ( feedback && feedback->isCanceled() ) )
4151 : 0 : break;
4152 : : }
4153 : :
4154 : 0 : return results;
4155 : 0 : }
4156 : :
4157 : 0 : QgsFeatureSource::SpatialIndexPresence QgsOgrProvider::hasSpatialIndex() const
4158 : : {
4159 : 0 : QgsCPLHTTPFetchOverrider oCPLHTTPFetcher( mAuthCfg );
4160 : 0 : QgsSetCPLHTTPFetchOverriderInitiatorClass( oCPLHTTPFetcher, QStringLiteral( "QgsOgrProvider" ) );
4161 : :
4162 : 0 : if ( mOgrLayer && mOgrLayer->TestCapability( OLCFastSpatialFilter ) )
4163 : 0 : return QgsFeatureSource::SpatialIndexPresent;
4164 : 0 : else if ( mOgrLayer )
4165 : 0 : return QgsFeatureSource::SpatialIndexNotPresent;
4166 : : else
4167 : 0 : return QgsFeatureSource::SpatialIndexUnknown;
4168 : 0 : }
4169 : :
4170 : 0 : QVariant QgsOgrProvider::minimumValue( int index ) const
4171 : : {
4172 : 0 : if ( !mValid || index < 0 || index >= mAttributeFields.count() )
4173 : : {
4174 : 0 : return QVariant();
4175 : : }
4176 : :
4177 : 0 : QgsCPLHTTPFetchOverrider oCPLHTTPFetcher( mAuthCfg );
4178 : 0 : QgsSetCPLHTTPFetchOverriderInitiatorClass( oCPLHTTPFetcher, QStringLiteral( "QgsOgrProvider" ) );
4179 : :
4180 : 0 : const QgsField originalField = mAttributeFields.at( index );
4181 : 0 : QgsField fld = originalField;
4182 : :
4183 : : // can't use native date/datetime types -- OGR converts these to string in the MAX return value
4184 : 0 : if ( fld.type() == QVariant::DateTime || fld.type() == QVariant::Date )
4185 : : {
4186 : 0 : fld.setType( QVariant::String );
4187 : 0 : }
4188 : :
4189 : : // Don't quote column name (see https://trac.osgeo.org/gdal/ticket/5799#comment:9)
4190 : 0 : QByteArray sql = "SELECT MIN(" + quotedIdentifier( textEncoding()->fromUnicode( fld.name() ) );
4191 : 0 : sql += ") FROM " + quotedIdentifier( mOgrLayer->name() );
4192 : :
4193 : 0 : if ( !mSubsetString.isEmpty() )
4194 : : {
4195 : 0 : sql += " WHERE " + textEncoding()->fromUnicode( mSubsetString );
4196 : 0 : }
4197 : :
4198 : 0 : QgsOgrLayerUniquePtr l = mOgrLayer->ExecuteSQL( sql );
4199 : 0 : if ( !l )
4200 : : {
4201 : 0 : QgsDebugMsg( QStringLiteral( "Failed to execute SQL: %1" ).arg( textEncoding()->toUnicode( sql ) ) );
4202 : 0 : return QgsVectorDataProvider::minimumValue( index );
4203 : : }
4204 : :
4205 : 0 : gdal::ogr_feature_unique_ptr f( l->GetNextFeature() );
4206 : 0 : if ( !f )
4207 : : {
4208 : 0 : return QVariant();
4209 : : }
4210 : :
4211 : 0 : bool ok = false;
4212 : 0 : QVariant res = QgsOgrUtils::getOgrFeatureAttribute( f.get(), fld, 0, textEncoding(), &ok );
4213 : 0 : if ( !ok )
4214 : 0 : return QVariant();
4215 : :
4216 : 0 : if ( res.type() != originalField.type() )
4217 : 0 : res = convertValue( originalField.type(), res.toString() );
4218 : :
4219 : 0 : if ( originalField.type() == QVariant::DateTime )
4220 : : {
4221 : : // ensure that we treat times as local time, to match behavior when iterating features
4222 : 0 : QDateTime temp = res.toDateTime();
4223 : 0 : temp.setTimeSpec( Qt::LocalTime );
4224 : 0 : res = temp;
4225 : 0 : }
4226 : :
4227 : 0 : return res;
4228 : 0 : }
4229 : :
4230 : 0 : QVariant QgsOgrProvider::maximumValue( int index ) const
4231 : : {
4232 : 0 : if ( !mValid || index < 0 || index >= mAttributeFields.count() )
4233 : : {
4234 : 0 : return QVariant();
4235 : : }
4236 : :
4237 : 0 : QgsCPLHTTPFetchOverrider oCPLHTTPFetcher( mAuthCfg );
4238 : 0 : QgsSetCPLHTTPFetchOverriderInitiatorClass( oCPLHTTPFetcher, QStringLiteral( "QgsOgrProvider" ) );
4239 : :
4240 : 0 : const QgsField originalField = mAttributeFields.at( index );
4241 : 0 : QgsField fld = originalField;
4242 : :
4243 : : // can't use native date/datetime types -- OGR converts these to string in the MAX return value
4244 : 0 : if ( fld.type() == QVariant::DateTime || fld.type() == QVariant::Date )
4245 : : {
4246 : 0 : fld.setType( QVariant::String );
4247 : 0 : }
4248 : :
4249 : : // Don't quote column name (see https://trac.osgeo.org/gdal/ticket/5799#comment:9)
4250 : 0 : QByteArray sql = "SELECT MAX(" + quotedIdentifier( textEncoding()->fromUnicode( fld.name() ) );
4251 : 0 : sql += ") FROM " + quotedIdentifier( mOgrLayer->name() );
4252 : :
4253 : 0 : if ( !mSubsetString.isEmpty() )
4254 : : {
4255 : 0 : sql += " WHERE " + textEncoding()->fromUnicode( mSubsetString );
4256 : 0 : }
4257 : :
4258 : 0 : QgsOgrLayerUniquePtr l = mOgrLayer->ExecuteSQL( sql );
4259 : 0 : if ( !l )
4260 : : {
4261 : 0 : QgsDebugMsg( QStringLiteral( "Failed to execute SQL: %1" ).arg( textEncoding()->toUnicode( sql ) ) );
4262 : 0 : return QgsVectorDataProvider::maximumValue( index );
4263 : : }
4264 : :
4265 : 0 : gdal::ogr_feature_unique_ptr f( l->GetNextFeature() );
4266 : 0 : if ( !f )
4267 : : {
4268 : 0 : return QVariant();
4269 : : }
4270 : :
4271 : 0 : bool ok = false;
4272 : 0 : QVariant res = QgsOgrUtils::getOgrFeatureAttribute( f.get(), fld, 0, textEncoding(), &ok );
4273 : 0 : if ( !ok )
4274 : 0 : return QVariant();
4275 : :
4276 : 0 : if ( res.type() != originalField.type() )
4277 : 0 : res = convertValue( originalField.type(), res.toString() );
4278 : :
4279 : 0 : if ( originalField.type() == QVariant::DateTime )
4280 : : {
4281 : : // ensure that we treat times as local time, to match behavior when iterating features
4282 : 0 : QDateTime temp = res.toDateTime();
4283 : 0 : temp.setTimeSpec( Qt::LocalTime );
4284 : 0 : res = temp;
4285 : 0 : }
4286 : :
4287 : 0 : return res;
4288 : 0 : }
4289 : :
4290 : 0 : QByteArray QgsOgrProvider::quotedIdentifier( const QByteArray &field ) const
4291 : : {
4292 : 0 : return QgsOgrProviderUtils::quotedIdentifier( field, mGDALDriverName );
4293 : 0 : }
4294 : :
4295 : 865 : QString QgsOgrProviderUtils::connectionPoolId( const QString &dataSourceURI, bool shareSameDatasetAmongLayers )
4296 : : {
4297 : 865 : if ( shareSameDatasetAmongLayers )
4298 : : {
4299 : : // If the file part of the URI is really a file, then use it as the
4300 : : // connection pool id (for example, so that all layers of a .gpkg file can
4301 : : // use the same GDAL dataset object)
4302 : : // Otherwise use the datasourceURI
4303 : : // Not completely sure about this logic. But at least, for GeoPackage this
4304 : : // works fine with multi layer datasets.
4305 : 865 : QString filePath = dataSourceURI.left( dataSourceURI.indexOf( QLatin1Char( '|' ) ) );
4306 : 865 : QFileInfo fi( filePath );
4307 : 865 : if ( fi.isFile() )
4308 : 865 : return filePath;
4309 : 865 : }
4310 : 0 : return dataSourceURI;
4311 : 865 : }
4312 : :
4313 : 337 : GDALDatasetH QgsOgrProviderUtils::GDALOpenWrapper( const char *pszPath, bool bUpdate, char **papszOpenOptionsIn, GDALDriverH *phDriver )
4314 : : {
4315 : 337 : CPLErrorReset();
4316 : :
4317 : 337 : char **papszOpenOptions = CSLDuplicate( papszOpenOptionsIn );
4318 : :
4319 : 337 : QString filePath( QString::fromUtf8( pszPath ) );
4320 : :
4321 : 337 : bool bIsGpkg = QFileInfo( filePath ).suffix().compare( QLatin1String( "gpkg" ), Qt::CaseInsensitive ) == 0;
4322 : 337 : bool bIsLocalGpkg = false;
4323 : 339 : if ( bIsGpkg &&
4324 : 2 : IsLocalFile( filePath ) &&
4325 : 2 : !CPLGetConfigOption( "OGR_SQLITE_JOURNAL", nullptr ) &&
4326 : 4 : QgsSettings().value( QStringLiteral( "qgis/walForSqlite3" ), true ).toBool() )
4327 : : {
4328 : : // For GeoPackage, we force opening of the file in WAL (Write Ahead Log)
4329 : : // mode so as to avoid readers blocking writer(s), and vice-versa.
4330 : : // https://www.sqlite.org/wal.html
4331 : : // But only do that on a local file since WAL is advertised not to work
4332 : : // on network shares
4333 : 2 : CPLSetThreadLocalConfigOption( "OGR_SQLITE_JOURNAL", "WAL" );
4334 : 2 : bIsLocalGpkg = true;
4335 : 2 : }
4336 : 335 : else if ( bIsGpkg )
4337 : : {
4338 : : // If WAL isn't set, we explicitly disable it, as it is persistent and it
4339 : : // may have been set on a previous connection.
4340 : 0 : CPLSetThreadLocalConfigOption( "OGR_SQLITE_JOURNAL", "DELETE" );
4341 : 0 : }
4342 : :
4343 : 337 : bool modify_OGR_GPKG_FOREIGN_KEY_CHECK = !CPLGetConfigOption( "OGR_GPKG_FOREIGN_KEY_CHECK", nullptr );
4344 : 337 : if ( modify_OGR_GPKG_FOREIGN_KEY_CHECK )
4345 : : {
4346 : 337 : CPLSetThreadLocalConfigOption( "OGR_GPKG_FOREIGN_KEY_CHECK", "NO" );
4347 : 337 : }
4348 : :
4349 : 337 : const int nOpenFlags = GDAL_OF_VECTOR | ( bUpdate ? GDAL_OF_UPDATE : 0 );
4350 : 337 : GDALDatasetH hDS = GDALOpenEx( pszPath, nOpenFlags, nullptr, papszOpenOptions, nullptr );
4351 : 337 : CSLDestroy( papszOpenOptions );
4352 : :
4353 : 337 : CPLSetThreadLocalConfigOption( "OGR_SQLITE_JOURNAL", nullptr );
4354 : 337 : if ( modify_OGR_GPKG_FOREIGN_KEY_CHECK )
4355 : : {
4356 : 337 : CPLSetThreadLocalConfigOption( "OGR_GPKG_FOREIGN_KEY_CHECK", nullptr );
4357 : 337 : }
4358 : :
4359 : 337 : if ( !hDS )
4360 : : {
4361 : 0 : if ( phDriver )
4362 : 0 : *phDriver = nullptr;
4363 : 0 : return nullptr;
4364 : : }
4365 : 337 : GDALDriverH hDrv = GDALGetDatasetDriver( hDS );
4366 : 337 : if ( bIsLocalGpkg && strcmp( GDALGetDriverShortName( hDrv ), "GPKG" ) == 0 )
4367 : : {
4368 : 2 : QMutexLocker locker( sGlobalMutex() );
4369 : 2 : ( *sMapCountOpenedDS() )[ filePath ]++;
4370 : 2 : ( *sMapDSHandleToUpdateMode() )[ hDS ] = bUpdate;
4371 : 2 : }
4372 : 337 : if ( phDriver )
4373 : 0 : *phDriver = hDrv;
4374 : :
4375 : 337 : return hDS;
4376 : 337 : }
4377 : :
4378 : 4 : static bool IsLocalFile( const QString &path )
4379 : : {
4380 : 4 : QString dirName( QFileInfo( path ).absolutePath() );
4381 : : // Start with the OS specific methods since the QT >= 5.4 method just
4382 : : // return a string and not an enumerated type.
4383 : : #if defined(Q_OS_WIN)
4384 : : if ( dirName.startsWith( "\\\\" ) || dirName.startsWith( "//" ) )
4385 : : return false;
4386 : : if ( dirName.length() >= 3 && dirName[1] == ':' &&
4387 : : ( dirName[2] == '\\' || dirName[2] == '/' ) )
4388 : : {
4389 : : dirName.resize( 3 );
4390 : : return GetDriveType( dirName.toLatin1().constData() ) != DRIVE_REMOTE;
4391 : : }
4392 : : return true;
4393 : : #elif defined(Q_OS_LINUX)
4394 : : struct statfs sStatFS;
4395 : : if ( statfs( dirName.toLatin1().constData(), &sStatFS ) == 0 )
4396 : : {
4397 : : // Codes from http://man7.org/linux/man-pages/man2/statfs.2.html
4398 : : if ( sStatFS.f_type == 0x6969 /* NFS */ ||
4399 : : sStatFS.f_type == 0x517b /* SMB */ ||
4400 : : sStatFS.f_type == 0xff534d42ul /* CIFS */ ||
4401 : : sStatFS.f_type == 0xfe534d42ul /* CIFS */ )
4402 : : {
4403 : : return false;
4404 : : }
4405 : : }
4406 : : return true;
4407 : : #else
4408 : 4 : QStorageInfo info( dirName );
4409 : 4 : const QString fileSystem( info.fileSystemType() );
4410 : 4 : bool isLocal = path != QLatin1String( "nfs" ) && path != QLatin1String( "smbfs" );
4411 : 4 : if ( !isLocal )
4412 : 0 : QgsDebugMsgLevel( QStringLiteral( "Filesystem for %1 is %2" ).arg( path, fileSystem ), 2 );
4413 : 4 : return isLocal;
4414 : : #endif
4415 : 4 : }
4416 : :
4417 : 334 : void QgsOgrProviderUtils::GDALCloseWrapper( GDALDatasetH hDS )
4418 : : {
4419 : 334 : if ( !hDS )
4420 : 0 : return;
4421 : 334 : GDALDriverH mGDALDriver = GDALGetDatasetDriver( hDS );
4422 : 334 : QString mGDALDriverName = GDALGetDriverShortName( mGDALDriver );
4423 : 334 : QString datasetName( QString::fromUtf8( GDALGetDescription( hDS ) ) );
4424 : 336 : if ( mGDALDriverName == QLatin1String( "GPKG" ) &&
4425 : 2 : IsLocalFile( datasetName ) &&
4426 : 2 : !CPLGetConfigOption( "OGR_SQLITE_JOURNAL", nullptr ) )
4427 : : {
4428 : 2 : bool openedAsUpdate = false;
4429 : 2 : bool tryReturnToWall = false;
4430 : : {
4431 : 2 : QMutexLocker locker( sGlobalMutex() );
4432 : 2 : ( *sMapCountOpenedDS() )[ datasetName ] --;
4433 : 2 : if ( ( *sMapCountOpenedDS() )[ datasetName ] == 0 )
4434 : : {
4435 : 1 : sMapCountOpenedDS()->remove( datasetName );
4436 : 1 : openedAsUpdate = ( *sMapDSHandleToUpdateMode() )[hDS];
4437 : 1 : tryReturnToWall = true;
4438 : 1 : }
4439 : 2 : sMapDSHandleToUpdateMode()->remove( hDS );
4440 : 2 : }
4441 : 2 : if ( tryReturnToWall )
4442 : : {
4443 : 1 : bool bSuccess = false;
4444 : 1 : if ( openedAsUpdate )
4445 : : {
4446 : : // We need to reset all iterators on layers, otherwise we will not
4447 : : // be able to change journal_mode.
4448 : 1 : int layerCount = GDALDatasetGetLayerCount( hDS );
4449 : 2 : for ( int i = 0; i < layerCount; i ++ )
4450 : : {
4451 : 1 : OGR_L_ResetReading( GDALDatasetGetLayer( hDS, i ) );
4452 : 1 : }
4453 : :
4454 : 1 : CPLPushErrorHandler( CPLQuietErrorHandler );
4455 : 1 : QgsDebugMsgLevel( QStringLiteral( "GPKG: Trying to return to delete mode" ), 2 );
4456 : 1 : OGRLayerH hSqlLyr = GDALDatasetExecuteSQL( hDS,
4457 : : "PRAGMA journal_mode = delete",
4458 : : nullptr, nullptr );
4459 : 1 : if ( hSqlLyr )
4460 : : {
4461 : 1 : gdal::ogr_feature_unique_ptr hFeat( OGR_L_GetNextFeature( hSqlLyr ) );
4462 : 1 : if ( hFeat )
4463 : : {
4464 : 1 : const char *pszRet = OGR_F_GetFieldAsString( hFeat.get(), 0 );
4465 : 1 : bSuccess = EQUAL( pszRet, "delete" );
4466 : 1 : QgsDebugMsgLevel( QStringLiteral( "Return: %1" ).arg( pszRet ), 2 );
4467 : 1 : }
4468 : 1 : }
4469 : 0 : else if ( CPLGetLastErrorType() != CE_None )
4470 : : {
4471 : 0 : QgsDebugMsg( QStringLiteral( "Return: %1" ).arg( CPLGetLastErrorMsg() ) );
4472 : 0 : }
4473 : 1 : GDALDatasetReleaseResultSet( hDS, hSqlLyr );
4474 : 1 : CPLPopErrorHandler();
4475 : 1 : }
4476 : 1 : GDALClose( hDS );
4477 : :
4478 : : // If the file was opened in read-only mode, or if the above failed,
4479 : : // we need to reopen it in update mode
4480 : 1 : if ( !bSuccess )
4481 : : {
4482 : 0 : if ( openedAsUpdate )
4483 : : {
4484 : 0 : QgsDebugMsgLevel( QStringLiteral( "GPKG: Trying again" ), 1 );
4485 : 0 : }
4486 : : else
4487 : : {
4488 : 0 : QgsDebugMsgLevel( QStringLiteral( "GPKG: Trying to return to delete mode" ), 1 );
4489 : : }
4490 : 0 : CPLSetThreadLocalConfigOption( "OGR_SQLITE_JOURNAL", "DELETE" );
4491 : 0 : hDS = GDALOpenEx( datasetName.toUtf8().constData(), GDAL_OF_VECTOR | GDAL_OF_UPDATE, nullptr, nullptr, nullptr );
4492 : 0 : CPLSetThreadLocalConfigOption( "OGR_SQLITE_JOURNAL", nullptr );
4493 : 0 : if ( hDS )
4494 : : {
4495 : : #ifdef QGISDEBUG
4496 : : CPLPushErrorHandler( CPLQuietErrorHandler );
4497 : : OGRLayerH hSqlLyr = GDALDatasetExecuteSQL( hDS,
4498 : : "PRAGMA journal_mode",
4499 : : nullptr, nullptr );
4500 : : CPLPopErrorHandler();
4501 : : if ( hSqlLyr != nullptr )
4502 : : {
4503 : : gdal::ogr_feature_unique_ptr hFeat( OGR_L_GetNextFeature( hSqlLyr ) );
4504 : : if ( hFeat != nullptr )
4505 : : {
4506 : : const char *pszRet = OGR_F_GetFieldAsString( hFeat.get(), 0 );
4507 : : QgsDebugMsgLevel( QStringLiteral( "Return: %1" ).arg( pszRet ), 1 );
4508 : : }
4509 : : GDALDatasetReleaseResultSet( hDS, hSqlLyr );
4510 : : }
4511 : : #endif
4512 : 0 : GDALClose( hDS );
4513 : 0 : }
4514 : 0 : }
4515 : 1 : }
4516 : : else
4517 : : {
4518 : 1 : GDALClose( hDS );
4519 : : }
4520 : 2 : }
4521 : :
4522 : : #if GDAL_VERSION_NUM >= GDAL_COMPUTE_VERSION(3,1,0) && GDAL_VERSION_NUM <= GDAL_COMPUTE_VERSION(3,1,3)
4523 : : else if ( mGDALDriverName == QLatin1String( "XLSX" ) ||
4524 : : mGDALDriverName == QLatin1String( "ODS" ) )
4525 : : {
4526 : : // 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
4527 : : CPLSetThreadLocalConfigOption( "CPL_CREATE_ZIP64", "NO" );
4528 : : GDALClose( hDS );
4529 : : CPLSetThreadLocalConfigOption( "CPL_CREATE_ZIP64", nullptr );
4530 : : }
4531 : : #endif
4532 : :
4533 : : else
4534 : : {
4535 : 332 : GDALClose( hDS );
4536 : : }
4537 : 334 : }
4538 : :
4539 : 0 : QByteArray QgsOgrProviderUtils::quotedIdentifier( QByteArray field, const QString &driverName )
4540 : : {
4541 : 0 : if ( driverName == QLatin1String( "MySQL" ) )
4542 : : {
4543 : 0 : field.replace( '\\', "\\\\" );
4544 : 0 : field.replace( '`', "``" );
4545 : 0 : return field.prepend( '`' ).append( '`' );
4546 : : }
4547 : : else
4548 : : {
4549 : 0 : field.replace( '\\', "\\\\" );
4550 : 0 : field.replace( '"', "\\\"" );
4551 : 0 : field.replace( '\'', "\\'" );
4552 : 0 : return field.prepend( '\"' ).append( '\"' );
4553 : : }
4554 : 0 : }
4555 : :
4556 : 0 : QString QgsOgrProviderUtils::quotedValue( const QVariant &value )
4557 : : {
4558 : 0 : if ( value.isNull() )
4559 : 0 : return QStringLiteral( "NULL" );
4560 : :
4561 : 0 : switch ( value.type() )
4562 : : {
4563 : : case QVariant::Int:
4564 : : case QVariant::LongLong:
4565 : : case QVariant::Double:
4566 : 0 : return value.toString();
4567 : :
4568 : : case QVariant::Bool:
4569 : : //OGR does not support boolean literals
4570 : 0 : return value.toBool() ? "1" : "0";
4571 : :
4572 : : default:
4573 : : case QVariant::String:
4574 : 0 : QString v = value.toString();
4575 : 0 : v.replace( '\'', QLatin1String( "''" ) );
4576 : 0 : if ( v.contains( '\\' ) )
4577 : 0 : return v.replace( '\\', QLatin1String( "\\\\" ) ).prepend( "E'" ).append( '\'' );
4578 : : else
4579 : 0 : return v.prepend( '\'' ).append( '\'' );
4580 : 0 : }
4581 : 0 : }
4582 : :
4583 : 40 : bool QgsOgrProvider::syncToDisc()
4584 : : {
4585 : 40 : QgsOgrConnPool::instance()->invalidateConnections( QgsOgrProviderUtils::connectionPoolId( dataSourceUri( true ), mShareSameDatasetAmongLayers ) );
4586 : :
4587 : : //for shapefiles, remove spatial index files and create a new index
4588 : 40 : QgsOgrConnPool::instance()->unref( QgsOgrProviderUtils::connectionPoolId( dataSourceUri( true ), mShareSameDatasetAmongLayers ) );
4589 : 40 : bool shapeIndex = false;
4590 : 40 : if ( mGDALDriverName == QLatin1String( "ESRI Shapefile" ) )
4591 : : {
4592 : 40 : QString sbnIndexFile;
4593 : 40 : QFileInfo fi( mFilePath );
4594 : 40 : int suffixLength = fi.suffix().length();
4595 : 40 : sbnIndexFile = mFilePath;
4596 : 40 : sbnIndexFile.chop( suffixLength );
4597 : 40 : sbnIndexFile.append( "sbn" );
4598 : :
4599 : 40 : if ( QFile::exists( sbnIndexFile ) )
4600 : : {
4601 : 0 : shapeIndex = true;
4602 : 0 : close();
4603 : 0 : QgsOgrConnPool::instance()->invalidateConnections( QgsOgrProviderUtils::connectionPoolId( dataSourceUri( true ), mShareSameDatasetAmongLayers ) );
4604 : 0 : QFile::remove( sbnIndexFile );
4605 : 0 : open( OpenModeSameAsCurrent );
4606 : 0 : if ( !mValid )
4607 : 0 : return false;
4608 : 0 : }
4609 : 40 : }
4610 : :
4611 : 40 : if ( mOgrLayer->SyncToDisk() != OGRERR_NONE )
4612 : : {
4613 : 0 : pushError( tr( "OGR error syncing to disk: %1" ).arg( CPLGetLastErrorMsg() ) );
4614 : 0 : }
4615 : :
4616 : 40 : QgsOgrConnPool::instance()->ref( QgsOgrProviderUtils::connectionPoolId( dataSourceUri( true ), mShareSameDatasetAmongLayers ) );
4617 : 40 : if ( shapeIndex )
4618 : : {
4619 : 0 : return createSpatialIndex();
4620 : : }
4621 : :
4622 : 40 : return true;
4623 : 40 : }
4624 : :
4625 : 17 : void QgsOgrProvider::recalculateFeatureCount() const
4626 : : {
4627 : 17 : if ( !mOgrLayer )
4628 : : {
4629 : 0 : mFeaturesCounted = QgsVectorDataProvider::Uncounted;
4630 : 0 : return;
4631 : : }
4632 : :
4633 : 17 : OGRGeometryH filter = mOgrLayer->GetSpatialFilter();
4634 : 17 : if ( filter )
4635 : : {
4636 : 0 : filter = OGR_G_Clone( filter );
4637 : 0 : mOgrLayer->SetSpatialFilter( nullptr );
4638 : 0 : }
4639 : :
4640 : : // feature count returns number of features within current spatial filter
4641 : : // so we remove it if there's any and then put it back
4642 : 17 : if ( mOgrGeometryTypeFilter == wkbUnknown )
4643 : : {
4644 : 17 : mFeaturesCounted = mOgrLayer->GetApproxFeatureCount();
4645 : 17 : if ( mFeaturesCounted == -1 )
4646 : : {
4647 : 0 : mFeaturesCounted = QgsVectorDataProvider::UnknownCount;
4648 : 0 : }
4649 : 17 : }
4650 : : else
4651 : : {
4652 : 0 : mFeaturesCounted = 0;
4653 : 0 : setRelevantFields( true, QgsAttributeList() );
4654 : 0 : mOgrLayer->ResetReading();
4655 : 0 : gdal::ogr_feature_unique_ptr fet;
4656 : 0 : const OGRwkbGeometryType flattenGeomTypeFilter =
4657 : 0 : QgsOgrProvider::ogrWkbSingleFlatten( mOgrGeometryTypeFilter );
4658 : 0 : while ( fet.reset( mOgrLayer->GetNextFeature() ), fet )
4659 : : {
4660 : 0 : OGRGeometryH geom = OGR_F_GetGeometryRef( fet.get() );
4661 : 0 : if ( geom )
4662 : : {
4663 : 0 : OGRwkbGeometryType gType = OGR_G_GetGeometryType( geom );
4664 : 0 : gType = QgsOgrProvider::ogrWkbSingleFlatten( gType );
4665 : 0 : if ( gType == flattenGeomTypeFilter ) mFeaturesCounted++;
4666 : 0 : }
4667 : : }
4668 : 0 : mOgrLayer->ResetReading();
4669 : 0 : setRelevantFields( true, attributeIndexes() );
4670 : 0 : }
4671 : :
4672 : 17 : if ( filter )
4673 : : {
4674 : 0 : mOgrLayer->SetSpatialFilter( filter );
4675 : 0 : }
4676 : :
4677 : 17 : QgsOgrConnPool::instance()->invalidateConnections( QgsOgrProviderUtils::connectionPoolId( dataSourceUri( true ), mShareSameDatasetAmongLayers ) );
4678 : 17 : }
4679 : :
4680 : 0 : bool QgsOgrProvider::doesStrictFeatureTypeCheck() const
4681 : : {
4682 : : // FIXME probably other drivers too...
4683 : 0 : return mGDALDriverName != QLatin1String( "ESRI Shapefile" ) || ( mOGRGeomType == wkbPoint || mOGRGeomType == wkbPoint25D );
4684 : : }
4685 : :
4686 : 0 : QgsFeatureRenderer *QgsOgrProvider::createRenderer( const QVariantMap & ) const
4687 : : {
4688 : 0 : if ( !( mCapabilities & FeatureSymbology ) )
4689 : 0 : return nullptr;
4690 : :
4691 : 0 : std::unique_ptr< QgsSymbol > defaultSymbol( QgsSymbol::defaultSymbol( QgsWkbTypes::geometryType( wkbType() ) ) );
4692 : 0 : return new QgsEmbeddedSymbolRenderer( defaultSymbol.release() );
4693 : 0 : }
4694 : :
4695 : 133 : OGRwkbGeometryType QgsOgrProvider::ogrWkbSingleFlatten( OGRwkbGeometryType type )
4696 : : {
4697 : 133 : type = wkbFlatten( type );
4698 : 133 : switch ( type )
4699 : : {
4700 : : case wkbMultiPoint:
4701 : 0 : return wkbPoint;
4702 : : case wkbMultiLineString:
4703 : 0 : return wkbLineString;
4704 : : case wkbMultiPolygon:
4705 : 0 : return wkbPolygon;
4706 : : case wkbMultiCurve:
4707 : 0 : return wkbCompoundCurve;
4708 : : case wkbMultiSurface:
4709 : 0 : return wkbCurvePolygon;
4710 : : default:
4711 : 133 : return type;
4712 : : }
4713 : 133 : }
4714 : :
4715 : 0 : OGRLayerH QgsOgrProviderUtils::setSubsetString( OGRLayerH layer, GDALDatasetH ds, QTextCodec *encoding, const QString &subsetString )
4716 : : {
4717 : 0 : QByteArray layerName = OGR_FD_GetName( OGR_L_GetLayerDefn( layer ) );
4718 : 0 : GDALDriverH driver = GDALGetDatasetDriver( ds );
4719 : 0 : QString driverName = GDALGetDriverShortName( driver );
4720 : :
4721 : 0 : if ( driverName == QLatin1String( "ODBC" ) ) //the odbc driver does not like schema names for subset
4722 : : {
4723 : 0 : QString layerNameString = encoding->toUnicode( layerName );
4724 : 0 : int dotIndex = layerNameString.indexOf( '.' );
4725 : 0 : if ( dotIndex > 1 )
4726 : : {
4727 : 0 : QString modifiedLayerName = layerNameString.right( layerNameString.size() - dotIndex - 1 );
4728 : 0 : layerName = encoding->fromUnicode( modifiedLayerName );
4729 : 0 : }
4730 : 0 : }
4731 : 0 : OGRLayerH subsetLayer = nullptr;
4732 : 0 : if ( subsetString.startsWith( QLatin1String( "SELECT " ), Qt::CaseInsensitive ) )
4733 : : {
4734 : 0 : QByteArray sql = encoding->fromUnicode( subsetString );
4735 : :
4736 : 0 : QgsDebugMsgLevel( QStringLiteral( "SQL: %1" ).arg( encoding->toUnicode( sql ) ), 1 );
4737 : 0 : subsetLayer = GDALDatasetExecuteSQL( ds, sql.constData(), nullptr, nullptr );
4738 : 0 : }
4739 : : else
4740 : : {
4741 : 0 : if ( OGR_L_SetAttributeFilter( layer, encoding->fromUnicode( subsetString ).constData() ) != OGRERR_NONE )
4742 : : {
4743 : 0 : return nullptr;
4744 : : }
4745 : 0 : subsetLayer = layer;
4746 : : }
4747 : :
4748 : 0 : return subsetLayer;
4749 : 0 : }
4750 : :
4751 : 187 : void QgsOgrProvider::open( OpenMode mode )
4752 : : {
4753 : 187 : bool openReadOnly = false;
4754 : : Q_ASSERT( !mOgrSqlLayer );
4755 : : Q_ASSERT( !mOgrOrigLayer );
4756 : :
4757 : : // Try to open using VSIFileHandler
4758 : : // see http://trac.osgeo.org/gdal/wiki/UserDocs/ReadInZip
4759 : 187 : QString vsiPrefix = QgsZipItem::vsiPrefix( dataSourceUri() );
4760 : 187 : if ( !vsiPrefix.isEmpty() )
4761 : : {
4762 : : // GDAL>=1.8.0 has write support for zip, but read and write operations
4763 : : // cannot be interleaved, so for now just use read-only.
4764 : 0 : openReadOnly = true;
4765 : 0 : if ( !mFilePath.startsWith( vsiPrefix ) )
4766 : : {
4767 : 0 : mFilePath = vsiPrefix + mFilePath;
4768 : 0 : setDataSourceUri( mFilePath );
4769 : 0 : }
4770 : 0 : QgsDebugMsgLevel( QStringLiteral( "Trying %1 syntax, mFilePath= %2" ).arg( vsiPrefix, mFilePath ), 1 );
4771 : 0 : }
4772 : :
4773 : 187 : QgsDebugMsgLevel( "mFilePath: " + mFilePath, 3 );
4774 : 187 : QgsDebugMsgLevel( "mLayerIndex: " + QString::number( mLayerIndex ), 3 );
4775 : 187 : QgsDebugMsgLevel( "mLayerName: " + mLayerName, 3 );
4776 : 187 : QgsDebugMsgLevel( "mSubsetString: " + mSubsetString, 3 );
4777 : 187 : CPLSetConfigOption( "OGR_ORGANIZE_POLYGONS", "ONLY_CCW" ); // "SKIP" returns MULTIPOLYGONs for multiringed POLYGONs
4778 : 187 : CPLSetConfigOption( "GPX_ELE_AS_25D", "YES" ); // use GPX elevation as z values
4779 : 187 : CPLSetConfigOption( "LIBKML_RESOLVE_STYLE", "YES" ); // resolve kml style urls from style tables to feature style strings
4780 : 187 : if ( !CPLGetConfigOption( "OSM_USE_CUSTOM_INDEXING", nullptr ) )
4781 : : {
4782 : : // Disable custom/fast indexing by default, as it can prevent some .osm.pbf
4783 : : // files to be loaded.
4784 : : // See https://github.com/qgis/QGIS/issues/31062
4785 : 1 : CPLSetConfigOption( "OSM_USE_CUSTOM_INDEXING", "NO" );
4786 : 1 : }
4787 : :
4788 : 187 : if ( mFilePath.startsWith( QLatin1String( "MySQL:" ) ) && !mLayerName.isEmpty() && !mFilePath.endsWith( ",tables=" + mLayerName ) )
4789 : : {
4790 : 0 : mFilePath += ",tables=" + mLayerName;
4791 : 0 : }
4792 : :
4793 : 187 : if ( mode == OpenModeForceReadOnly )
4794 : 60 : openReadOnly = true;
4795 : 127 : else if ( mode == OpenModeSameAsCurrent && !mWriteAccess )
4796 : 0 : openReadOnly = true;
4797 : :
4798 : : // first try to open in update mode (unless specified otherwise)
4799 : 187 : QString errCause;
4800 : 187 : if ( !openReadOnly )
4801 : : {
4802 : 127 : QStringList options( mOpenOptions );
4803 : 127 : if ( mode == OpenModeForceUpdateRepackOff || ( mDeferRepack && OpenModeSameAsCurrent ) )
4804 : : {
4805 : 63 : options << "AUTO_REPACK=OFF";
4806 : 63 : }
4807 : : // We get the layer which was requested by the uri. The layername
4808 : : // has precedence over the layerid if both are given.
4809 : 127 : if ( !mLayerName.isNull() )
4810 : : {
4811 : 0 : mOgrOrigLayer = QgsOgrProviderUtils::getLayer( mFilePath, true, options, mLayerName, errCause, true );
4812 : 0 : }
4813 : : else
4814 : : {
4815 : 127 : mOgrOrigLayer = QgsOgrProviderUtils::getLayer( mFilePath, true, options, mLayerIndex, errCause, true );
4816 : : }
4817 : 127 : }
4818 : :
4819 : 187 : mValid = false;
4820 : 187 : if ( mOgrOrigLayer )
4821 : : {
4822 : 127 : mWriteAccess = true;
4823 : 127 : mWriteAccessPossible = true;
4824 : 127 : }
4825 : : else
4826 : : {
4827 : 60 : mWriteAccess = false;
4828 : 60 : if ( !openReadOnly )
4829 : : {
4830 : 0 : QgsDebugMsg( QStringLiteral( "OGR failed to opened in update mode, trying in read-only mode" ) );
4831 : 0 : }
4832 : :
4833 : 60 : QStringList options( mOpenOptions );
4834 : : #if GDAL_VERSION_NUM >= GDAL_COMPUTE_VERSION(3,1,0)
4835 : : // assume trusted data to get more speed
4836 : 60 : if ( mGDALDriverName == QLatin1String( "FlatGeobuf" ) &&
4837 : 0 : !options.contains( "VERIFY_BUFFERS=YES" ) )
4838 : : {
4839 : 0 : options << "VERIFY_BUFFERS=NO";
4840 : 0 : }
4841 : : #endif
4842 : :
4843 : : // try to open read-only
4844 : 60 : if ( !mLayerName.isNull() )
4845 : : {
4846 : 0 : mOgrOrigLayer = QgsOgrProviderUtils::getLayer( mFilePath, false, options, mLayerName, errCause, true );
4847 : 0 : }
4848 : : else
4849 : : {
4850 : 60 : mOgrOrigLayer = QgsOgrProviderUtils::getLayer( mFilePath, false, options, mLayerIndex, errCause, true );
4851 : : }
4852 : 60 : }
4853 : :
4854 : 187 : if ( mOgrOrigLayer )
4855 : : {
4856 : 187 : mGDALDriverName = mOgrOrigLayer->driverName();
4857 : 187 : mShareSameDatasetAmongLayers = QgsOgrProviderUtils::canDriverShareSameDatasetAmongLayers( mGDALDriverName );
4858 : :
4859 : 187 : QgsDebugMsgLevel( "OGR opened using Driver " + mGDALDriverName, 2 );
4860 : :
4861 : 187 : mOgrLayer = mOgrOrigLayer.get();
4862 : :
4863 : : // check that the initial encoding setting is fit for this layer
4864 : :
4865 : 187 : if ( mode == OpenModeInitial && mGDALDriverName == QLatin1String( "ESRI Shapefile" ) )
4866 : : {
4867 : : // determine encoding from shapefile cpg or LDID information, if possible
4868 : 63 : QString shpEncoding;
4869 : : #if GDAL_VERSION_NUM >= GDAL_COMPUTE_VERSION(3,1,0)
4870 : 189 : shpEncoding = mOgrLayer->GetMetadataItem( QStringLiteral( "ENCODING_FROM_CPG" ), QStringLiteral( "SHAPEFILE" ) );
4871 : 63 : if ( shpEncoding.isEmpty() )
4872 : 189 : shpEncoding = mOgrLayer->GetMetadataItem( QStringLiteral( "ENCODING_FROM_LDID" ), QStringLiteral( "SHAPEFILE" ) );
4873 : : #else
4874 : : shpEncoding = QgsOgrUtils::readShapefileEncoding( mFilePath );
4875 : : #endif
4876 : :
4877 : 63 : if ( !shpEncoding.isEmpty() )
4878 : 1 : setEncoding( shpEncoding );
4879 : : else
4880 : 62 : setEncoding( encoding() );
4881 : 63 : }
4882 : : else
4883 : : {
4884 : 124 : setEncoding( encoding() );
4885 : : }
4886 : :
4887 : : // Ensure subset is set (setSubsetString does nothing if the passed sql subset string is equal to mSubsetString, which is the case when reloading the dataset)
4888 : 187 : QString origSubsetString = mSubsetString;
4889 : 187 : mSubsetString.clear();
4890 : : // Block signals to avoid endless recursion reloadData -> emit dataChanged -> reloadData
4891 : 187 : blockSignals( true );
4892 : :
4893 : : // Do not update capabilities: it will be done later
4894 : :
4895 : : // WARNING if this is the initial open - we don't already have a connection ref, and will be creating one later. So we *mustn't* grab an extra connection ref
4896 : : // while setting the subset string, or we'll be left with an extra reference which is never cleared.
4897 : 187 : mValid = _setSubsetString( origSubsetString, false, false, mode != OpenModeInitial );
4898 : :
4899 : 187 : blockSignals( false );
4900 : 187 : if ( mValid )
4901 : : {
4902 : 187 : if ( mode == OpenModeInitial )
4903 : : {
4904 : 64 : computeCapabilities();
4905 : 64 : }
4906 : 187 : QgsDebugMsgLevel( QStringLiteral( "Data source is valid" ), 2 );
4907 : 187 : }
4908 : : else
4909 : : {
4910 : 0 : QgsMessageLog::logMessage( tr( "Data source is invalid (%1)" ).arg( QString::fromUtf8( CPLGetLastErrorMsg() ) ), tr( "OGR" ) );
4911 : : }
4912 : 187 : }
4913 : : else
4914 : : {
4915 : 0 : QgsMessageLog::logMessage( errCause + "(" + QString::fromUtf8( CPLGetLastErrorMsg() ) + ")", tr( "OGR" ) );
4916 : : }
4917 : :
4918 : : // For shapefiles or MapInfo .tab, so as to allow concurrent opening between
4919 : : // QGIS and MapInfo, we go back to read-only mode for now.
4920 : : // We limit to those drivers as re-opening is relatively cheap (other drivers
4921 : : // like GeoJSON might do full content ingestion for example)
4922 : 188 : if ( mValid && mode == OpenModeInitial && mWriteAccess &&
4923 : 64 : ( mGDALDriverName == QLatin1String( "ESRI Shapefile" ) || mGDALDriverName == QLatin1String( "MapInfo File" ) ) )
4924 : : {
4925 : 63 : mOgrSqlLayer.reset();
4926 : 63 : mOgrOrigLayer.reset();
4927 : 63 : mOgrLayer = nullptr;
4928 : 63 : mValid = false;
4929 : :
4930 : : // In the case where we deal with a shapefile, it is possible that it has
4931 : : // pre-existing holes in the DBF (see #15407), so if using a GDAL version
4932 : : // recent enough to have reliable packing, do a packing at the first edit
4933 : : // action.
4934 : 126 : if ( mGDALDriverName == QLatin1String( "ESRI Shapefile" ) &&
4935 : 63 : atoi( GDALVersionInfo( "VERSION_NUM" ) ) >= GDAL_COMPUTE_VERSION( 2, 1, 2 ) )
4936 : : {
4937 : 63 : mShapefileMayBeCorrupted = true;
4938 : 63 : }
4939 : :
4940 : : // try to open read-only
4941 : 63 : if ( !mLayerName.isNull() )
4942 : : {
4943 : 0 : mOgrOrigLayer = QgsOgrProviderUtils::getLayer( mFilePath, false, mOpenOptions, mLayerName, errCause, true );
4944 : 0 : }
4945 : : else
4946 : : {
4947 : 63 : mOgrOrigLayer = QgsOgrProviderUtils::getLayer( mFilePath, false, mOpenOptions, mLayerIndex, errCause, true );
4948 : : }
4949 : :
4950 : 63 : mWriteAccess = false;
4951 : 63 : mOgrLayer = mOgrOrigLayer.get();
4952 : 63 : if ( mOgrLayer )
4953 : : {
4954 : 63 : mValid = true;
4955 : 63 : mDynamicWriteAccess = true;
4956 : :
4957 : 63 : if ( !mSubsetString.isEmpty() )
4958 : : {
4959 : : // Do not update capabilities here
4960 : : // but ensure subset is set (setSubsetString does nothing if the passed sql subset string is equal to
4961 : : // mSubsetString, which is the case when reloading the dataset)
4962 : 0 : QString origSubsetString = mSubsetString;
4963 : 0 : mSubsetString.clear();
4964 : 0 : mValid = _setSubsetString( origSubsetString, false, false );
4965 : 0 : }
4966 : 63 : }
4967 : 63 : }
4968 : :
4969 : : // For debug/testing purposes
4970 : 187 : if ( !mValid )
4971 : 0 : setProperty( "_debug_open_mode", "invalid" );
4972 : 187 : else if ( mWriteAccess )
4973 : 64 : setProperty( "_debug_open_mode", "read-write" );
4974 : : else
4975 : 123 : setProperty( "_debug_open_mode", "read-only" );
4976 : :
4977 : 187 : mRefreshFeatureCount = true;
4978 : 187 : }
4979 : :
4980 : 184 : void QgsOgrProvider::close()
4981 : : {
4982 : 184 : mOgrSqlLayer.reset();
4983 : 184 : mOgrOrigLayer.reset();
4984 : 184 : mOgrLayer = nullptr;
4985 : 184 : mValid = false;
4986 : 184 : setProperty( "_debug_open_mode", "invalid" );
4987 : :
4988 : 184 : invalidateCachedExtent( false );
4989 : 184 : }
4990 : :
4991 : 0 : void QgsOgrProvider::reloadProviderData()
4992 : : {
4993 : 0 : mFeaturesCounted = QgsVectorDataProvider::Uncounted;
4994 : 0 : bool wasValid = mValid;
4995 : 0 : QgsOgrConnPool::instance()->invalidateConnections( QgsOgrProviderUtils::connectionPoolId( dataSourceUri( true ), mShareSameDatasetAmongLayers ) );
4996 : 0 : close();
4997 : 0 : open( OpenModeSameAsCurrent );
4998 : 0 : if ( !mValid && wasValid )
4999 : 0 : pushError( tr( "Cannot reopen datasource %1" ).arg( dataSourceUri() ) );
5000 : 0 : }
5001 : :
5002 : 128 : bool QgsOgrProvider::_enterUpdateMode( bool implicit )
5003 : : {
5004 : 128 : if ( !mWriteAccessPossible )
5005 : : {
5006 : 0 : return false;
5007 : : }
5008 : 128 : if ( mWriteAccess )
5009 : : {
5010 : 65 : ++mUpdateModeStackDepth;
5011 : 65 : return true;
5012 : : }
5013 : 63 : if ( mUpdateModeStackDepth == 0 )
5014 : : {
5015 : : Q_ASSERT( mDynamicWriteAccess );
5016 : 63 : QgsDebugMsgLevel( QStringLiteral( "Reopening %1 in update mode" ).arg( dataSourceUri() ), 1 );
5017 : 63 : close();
5018 : 63 : open( implicit ? OpenModeForceUpdate : OpenModeForceUpdateRepackOff );
5019 : 63 : if ( !mOgrLayer || !mWriteAccess )
5020 : : {
5021 : 0 : QgsMessageLog::logMessage( tr( "Cannot reopen datasource %1 in update mode" ).arg( dataSourceUri() ), tr( "OGR" ) );
5022 : 0 : pushError( tr( "Cannot reopen datasource %1 in update mode" ).arg( dataSourceUri() ) );
5023 : 0 : return false;
5024 : : }
5025 : 63 : }
5026 : 63 : ++mUpdateModeStackDepth;
5027 : : // For implicitly entered updateMode, don't defer repacking
5028 : 63 : mDeferRepack = !implicit;
5029 : 63 : return true;
5030 : 128 : }
5031 : :
5032 : 125 : bool QgsOgrProvider::leaveUpdateMode()
5033 : : {
5034 : 125 : if ( !mWriteAccessPossible )
5035 : : {
5036 : 0 : return false;
5037 : : }
5038 : 125 : --mUpdateModeStackDepth;
5039 : 125 : if ( mUpdateModeStackDepth < 0 )
5040 : : {
5041 : 0 : QgsMessageLog::logMessage( tr( "Unbalanced call to leaveUpdateMode() w.r.t. enterUpdateMode()" ), tr( "OGR" ) );
5042 : 0 : mUpdateModeStackDepth = 0;
5043 : 0 : return false;
5044 : : }
5045 : 125 : if ( mDeferRepack && mUpdateModeStackDepth == 0 )
5046 : : {
5047 : : // Only repack once update mode is inactive
5048 : 60 : if ( mShapefileMayBeCorrupted )
5049 : 60 : repack();
5050 : :
5051 : 60 : mShapefileMayBeCorrupted = false;
5052 : 60 : mDeferRepack = false;
5053 : 60 : }
5054 : 125 : if ( !mDynamicWriteAccess )
5055 : : {
5056 : : // The GeoJSON driver only properly flushes stuff in all situations by
5057 : : // closing and re-opening. Starting with GDAL 2.3.1, it should be safe to
5058 : : // use GDALDatasetFlush().
5059 : 65 : if ( mGDALDriverName == QLatin1String( "GeoJSON" ) )
5060 : : {
5061 : : // Backup fields since if we created new fields, but didn't populate it
5062 : : // with any feature yet, it will disappear.
5063 : 0 : QgsFields oldFields = mAttributeFields;
5064 : 0 : reloadData();
5065 : 0 : if ( mValid )
5066 : : {
5067 : : // Make sure that new fields we added, but didn't populate yet, are
5068 : : // recreated at the OGR level, otherwise we won't be able to populate
5069 : : // them.
5070 : 0 : for ( const auto &field : oldFields )
5071 : : {
5072 : 0 : int idx = mAttributeFields.lookupField( field.name() );
5073 : 0 : if ( idx < 0 )
5074 : : {
5075 : 0 : bool ignoreErrorOut = false;
5076 : 0 : addAttributeOGRLevel( field, ignoreErrorOut );
5077 : 0 : }
5078 : : }
5079 : 0 : mAttributeFields = oldFields;
5080 : 0 : }
5081 : 0 : }
5082 : 65 : return true;
5083 : : }
5084 : 60 : if ( mUpdateModeStackDepth == 0 )
5085 : : {
5086 : 60 : QgsDebugMsgLevel( QStringLiteral( "Reopening %1 in read-only mode" ).arg( dataSourceUri() ), 1 );
5087 : 60 : close();
5088 : 60 : open( OpenModeForceReadOnly );
5089 : 60 : if ( !mOgrLayer )
5090 : : {
5091 : 0 : QgsMessageLog::logMessage( tr( "Cannot reopen datasource %1 in read-only mode" ).arg( dataSourceUri() ), tr( "OGR" ) );
5092 : 0 : pushError( tr( "Cannot reopen datasource %1 in read-only mode" ).arg( dataSourceUri() ) );
5093 : 0 : return false;
5094 : : }
5095 : 60 : }
5096 : 60 : return true;
5097 : 125 : }
5098 : :
5099 : 64 : bool QgsOgrProvider::isSaveAndLoadStyleToDatabaseSupported() const
5100 : : {
5101 : : // We could potentially extend support for styling to other drivers
5102 : : // with multiple layer support.
5103 : 64 : return mGDALDriverName == QLatin1String( "GPKG" ) ||
5104 : 63 : mGDALDriverName == QLatin1String( "SQLite" );
5105 : : }
5106 : :
5107 : 0 : bool QgsOgrProvider::isDeleteStyleFromDatabaseSupported() const
5108 : : {
5109 : 0 : return isSaveAndLoadStyleToDatabaseSupported();
5110 : : }
5111 : :
5112 : 0 : QString QgsOgrProvider::fileVectorFilters() const
5113 : : {
5114 : 0 : return QgsOgrProviderUtils::fileVectorFilters();
5115 : : }
5116 : :
5117 : 4070 : QString QgsOgrProviderUtils::DatasetIdentification::toString() const
5118 : : {
5119 : 12210 : return dsName +
5120 : 4070 : ( updateMode ?
5121 : 12210 : QStringLiteral( "update" ) : QStringLiteral( "read-only" ) ) +
5122 : 4070 : options.join( ',' );
5123 : 0 : }
5124 : :
5125 : 2035 : bool QgsOgrProviderUtils::DatasetIdentification::operator<
5126 : : ( const QgsOgrProviderUtils::DatasetIdentification &other ) const
5127 : : {
5128 : 2035 : return toString() < other.toString();
5129 : 0 : }
5130 : :
5131 : 250 : static GDALDatasetH OpenHelper( const QString &dsName,
5132 : : bool updateMode,
5133 : : const QStringList &options )
5134 : : {
5135 : 250 : char **papszOpenOptions = nullptr;
5136 : 250 : const auto constOptions = options;
5137 : 313 : for ( QString option : constOptions )
5138 : : {
5139 : 126 : papszOpenOptions = CSLAddString( papszOpenOptions,
5140 : 63 : option.toUtf8().constData() );
5141 : 63 : }
5142 : 250 : GDALDatasetH hDS = QgsOgrProviderUtils::GDALOpenWrapper(
5143 : 250 : QgsOgrProviderUtils::expandAuthConfig( dsName ).toUtf8().constData(), updateMode, papszOpenOptions, nullptr );
5144 : 250 : CSLDestroy( papszOpenOptions );
5145 : 250 : return hDS;
5146 : 250 : }
5147 : :
5148 : 0 : void QgsOgrProviderUtils::invalidateCachedDatasets( const QString &dsName )
5149 : : {
5150 : 0 : QMutexLocker locker( sGlobalMutex() );
5151 : 0 : while ( true )
5152 : : {
5153 : 0 : bool erased = false;
5154 : 0 : for ( auto iter = sMapSharedDS.begin(); iter != sMapSharedDS.end(); ++iter )
5155 : : {
5156 : 0 : if ( iter.key().dsName == dsName )
5157 : : {
5158 : 0 : sMapSharedDS.erase( iter );
5159 : 0 : erased = true;
5160 : 0 : break;
5161 : : }
5162 : 0 : }
5163 : 0 : if ( !erased )
5164 : 0 : break;
5165 : : }
5166 : 0 : }
5167 : :
5168 : :
5169 : 0 : QgsOgrDatasetSharedPtr QgsOgrProviderUtils::getAlreadyOpenedDataset( const QString &dsName )
5170 : : {
5171 : 0 : QMutexLocker locker( sGlobalMutex() );
5172 : 0 : for ( auto iter = sMapSharedDS.begin(); iter != sMapSharedDS.end(); ++iter )
5173 : : {
5174 : 0 : auto ident = iter.key();
5175 : 0 : if ( ident.dsName == dsName && ident.updateMode )
5176 : : {
5177 : : // Browse through this list, to look for the first available DatasetWithLayers*
5178 : : // instance that is in update mode (hoping there's only one...)
5179 : 0 : auto &datasetList = iter.value();
5180 : 0 : for ( const auto &ds : datasetList )
5181 : : {
5182 : : Q_ASSERT( ds->refCount > 0 );
5183 : 0 : return QgsOgrDataset::create( ident, ds );
5184 : : }
5185 : 0 : }
5186 : 0 : }
5187 : 0 : return nullptr;
5188 : 0 : }
5189 : :
5190 : :
5191 : 0 : QgsOgrLayerUniquePtr QgsOgrProviderUtils::getLayer( const QString &dsName,
5192 : : int layerIndex,
5193 : : QString &errCause )
5194 : : {
5195 : 0 : QMutexLocker locker( sGlobalMutex() );
5196 : 0 : for ( auto iter = sMapSharedDS.begin(); iter != sMapSharedDS.end(); ++iter )
5197 : : {
5198 : 0 : if ( iter.key().dsName == dsName )
5199 : : {
5200 : : // Browse through this list, to look for a DatasetWithLayers*
5201 : : // instance that don't use yet our layer of interest
5202 : 0 : auto &datasetList = iter.value();
5203 : 0 : const auto constDatasetList = datasetList;
5204 : 0 : for ( QgsOgrProviderUtils::DatasetWithLayers *ds : constDatasetList )
5205 : : {
5206 : 0 : if ( !ds->canBeShared )
5207 : 0 : continue;
5208 : : Q_ASSERT( ds->refCount > 0 );
5209 : :
5210 : 0 : QString layerName;
5211 : : OGRLayerH hLayer;
5212 : : {
5213 : 0 : QMutexLocker lockerDS( &ds->mutex );
5214 : 0 : hLayer = GDALDatasetGetLayer(
5215 : 0 : ds->hDS, layerIndex );
5216 : 0 : if ( hLayer )
5217 : : {
5218 : 0 : OGR_L_SetAttributeFilter( hLayer, nullptr );
5219 : 0 : layerName = QString::fromUtf8( OGR_L_GetName( hLayer ) );
5220 : 0 : }
5221 : 0 : }
5222 : 0 : if ( !hLayer )
5223 : : {
5224 : 0 : errCause = QObject::tr( "Cannot find layer %1." ).arg( layerIndex );
5225 : 0 : return nullptr;
5226 : : }
5227 : 0 : return getLayer( dsName, iter.key().updateMode, iter.key().options, layerName, errCause, true );
5228 : 0 : }
5229 : 0 : }
5230 : 0 : }
5231 : 0 : return getLayer( dsName, false, QStringList(), layerIndex, errCause, true );
5232 : 0 : }
5233 : :
5234 : 250 : QgsOgrLayerUniquePtr QgsOgrProviderUtils::getLayer( const QString &dsName,
5235 : : bool updateMode,
5236 : : const QStringList &options,
5237 : : int layerIndex,
5238 : : QString &errCause,
5239 : : bool checkModificationDateAgainstCache )
5240 : : {
5241 : 250 : QMutexLocker locker( sGlobalMutex() );
5242 : :
5243 : : // The idea is that we want to minimize the number of GDALDatasetH
5244 : : // handles openeded. But we have constraints. We do not want that 2
5245 : : // callers of getLayer() with the same input parameters get the same
5246 : : // GDALDatasetH since iterators over features of that layer would conflict
5247 : :
5248 : 250 : QgsOgrProviderUtils::DatasetIdentification ident;
5249 : 250 : ident.dsName = dsName;
5250 : 250 : ident.updateMode = updateMode;
5251 : 250 : ident.options = options;
5252 : : // Find if there's a list of DatasetWithLayers* that match our
5253 : : // (dsName, updateMode, options) criteria
5254 : 250 : auto iter = sMapSharedDS.find( ident );
5255 : 250 : if ( iter != sMapSharedDS.end() )
5256 : : {
5257 : : // Browse through this list, to look for a DatasetWithLayers*
5258 : : // instance that don't use yet our layer of interest
5259 : 0 : auto datasetList = iter.value();
5260 : 0 : const auto constDatasetList = datasetList;
5261 : 0 : for ( QgsOgrProviderUtils::DatasetWithLayers *ds : constDatasetList )
5262 : : {
5263 : 0 : if ( !ds->canBeShared )
5264 : 0 : continue;
5265 : : Q_ASSERT( ds->refCount > 0 );
5266 : :
5267 : 0 : QString layerName;
5268 : : OGRLayerH hLayer;
5269 : : {
5270 : 0 : QMutexLocker lockerDS( &ds->mutex );
5271 : 0 : hLayer = GDALDatasetGetLayer(
5272 : 0 : ds->hDS, layerIndex );
5273 : 0 : if ( hLayer )
5274 : : {
5275 : 0 : OGR_L_SetAttributeFilter( hLayer, nullptr );
5276 : 0 : layerName = QString::fromUtf8( OGR_L_GetName( hLayer ) );
5277 : 0 : }
5278 : 0 : }
5279 : 0 : if ( !hLayer )
5280 : : {
5281 : 0 : errCause = QObject::tr( "Cannot find layer %1." ).arg( layerIndex );
5282 : 0 : return nullptr;
5283 : : }
5284 : 0 : return getLayer( dsName, updateMode, options, layerName, errCause,
5285 : 0 : checkModificationDateAgainstCache );
5286 : 0 : }
5287 : 0 : }
5288 : :
5289 : 250 : GDALDatasetH hDS = OpenHelper( dsName, updateMode, options );
5290 : 250 : if ( !hDS )
5291 : : {
5292 : 0 : errCause = QObject::tr( "Cannot open %1." ).arg( dsName );
5293 : 0 : return nullptr;
5294 : : }
5295 : :
5296 : 250 : OGRLayerH hLayer = GDALDatasetGetLayer(
5297 : 250 : hDS, layerIndex );
5298 : 250 : if ( !hLayer )
5299 : : {
5300 : 0 : QgsOgrProviderUtils::GDALCloseWrapper( hDS );
5301 : 0 : errCause = QObject::tr( "Cannot find layer %1." ).arg( layerIndex );
5302 : 0 : return nullptr;
5303 : : }
5304 : 250 : QString layerName = QString::fromUtf8( OGR_L_GetName( hLayer ) );
5305 : :
5306 : 250 : QgsOgrProviderUtils::DatasetWithLayers *ds =
5307 : 250 : new QgsOgrProviderUtils::DatasetWithLayers;
5308 : 250 : ds->hDS = hDS;
5309 : :
5310 : 250 : GDALDriverH driver = GDALGetDatasetDriver( hDS );
5311 : 250 : QString driverName = GDALGetDriverShortName( driver );
5312 : 250 : ds->canBeShared = canDriverShareSameDatasetAmongLayers( driverName, updateMode, dsName );
5313 : :
5314 : 250 : QgsOgrLayerUniquePtr layer = QgsOgrLayer::CreateForLayer(
5315 : 250 : ident, layerName, ds, hLayer );
5316 : 250 : ds->setLayers[layerName] = layer.get();
5317 : :
5318 : 250 : QList<DatasetWithLayers *> datasetList;
5319 : 250 : datasetList.push_back( ds );
5320 : 250 : sMapSharedDS[ident] = datasetList;
5321 : :
5322 : 250 : return layer;
5323 : 250 : }
5324 : :
5325 : 1 : QgsOgrLayerUniquePtr QgsOgrProviderUtils::getLayer( const QString &dsName,
5326 : : const QString &layerName,
5327 : : QString &errCause )
5328 : : {
5329 : 1 : QMutexLocker locker( sGlobalMutex() );
5330 : :
5331 : 1 : for ( auto iter = sMapSharedDS.begin(); iter != sMapSharedDS.end(); ++iter )
5332 : : {
5333 : 1 : if ( iter.key().dsName == dsName )
5334 : : {
5335 : : // Browse through this list, to look for a DatasetWithLayers*
5336 : : // instance that don't use yet our layer of interest
5337 : 1 : auto &datasetList = iter.value();
5338 : 1 : const auto constDatasetList = datasetList;
5339 : 1 : for ( QgsOgrProviderUtils::DatasetWithLayers *ds : constDatasetList )
5340 : : {
5341 : 1 : if ( !ds->canBeShared )
5342 : 0 : continue;
5343 : : Q_ASSERT( ds->refCount > 0 );
5344 : :
5345 : 1 : auto iter2 = ds->setLayers.find( layerName );
5346 : 1 : if ( iter2 == ds->setLayers.end() )
5347 : : {
5348 : : OGRLayerH hLayer;
5349 : : {
5350 : 1 : QMutexLocker lockerDS( &ds->mutex );
5351 : 1 : hLayer = GDALDatasetGetLayerByName(
5352 : 1 : ds->hDS, layerName.toUtf8().constData() );
5353 : 1 : }
5354 : 1 : if ( !hLayer )
5355 : : {
5356 : : // Shouldn't happen really !
5357 : 1 : errCause = QObject::tr( "Cannot find layer %1." ).arg( layerName );
5358 : 1 : return nullptr;
5359 : : }
5360 : 0 : OGR_L_SetAttributeFilter( hLayer, nullptr );
5361 : :
5362 : 0 : QgsOgrLayerUniquePtr layer = QgsOgrLayer::CreateForLayer(
5363 : 0 : iter.key(), layerName, ds, hLayer );
5364 : 0 : ds->setLayers[layerName] = layer.get();
5365 : 0 : return layer;
5366 : 0 : }
5367 : : }
5368 : 1 : }
5369 : 0 : }
5370 : 0 : return getLayer( dsName, false, QStringList(), layerName, errCause, true );
5371 : 1 : }
5372 : :
5373 : 0 : static QDateTime getLastModified( const QString &dsName )
5374 : : {
5375 : 0 : if ( dsName.endsWith( ".gpkg", Qt::CaseInsensitive ) )
5376 : : {
5377 : 0 : QFileInfo info( dsName + "-wal" );
5378 : 0 : if ( info.exists() )
5379 : 0 : return info.lastModified();
5380 : 0 : }
5381 : 0 : return QFileInfo( dsName ).lastModified();
5382 : 0 : }
5383 : :
5384 : : // In case we do very fast structural changes within the same second,
5385 : : // the last modified date might not change enough, so artificially
5386 : : // decrement the cache modified date, so that the file appears newer to it
5387 : 0 : void QgsOgrProviderUtils::invalidateCachedLastModifiedDate( const QString &dsName )
5388 : : {
5389 : 0 : QMutexLocker locker( sGlobalMutex() );
5390 : :
5391 : 0 : auto iter = sMapDSNameToLastModifiedDate()->find( dsName );
5392 : 0 : if ( iter != sMapDSNameToLastModifiedDate()->end() )
5393 : : {
5394 : 0 : QgsDebugMsgLevel( QStringLiteral( "invalidating last modified date for %1" ).arg( dsName ), 1 );
5395 : 0 : iter.value() = iter.value().addSecs( -10 );
5396 : 0 : }
5397 : 0 : }
5398 : :
5399 : 0 : OGRwkbGeometryType QgsOgrProviderUtils::ogrTypeFromQgisType( QgsWkbTypes::Type type )
5400 : : {
5401 : 0 : switch ( type )
5402 : : {
5403 : : case QgsWkbTypes::Point:
5404 : 0 : return wkbPoint;
5405 : : case QgsWkbTypes::Point25D:
5406 : : case QgsWkbTypes::PointZ:
5407 : 0 : return wkbPoint25D;
5408 : : case QgsWkbTypes::PointM:
5409 : 0 : return wkbPointM;
5410 : : case QgsWkbTypes::PointZM:
5411 : 0 : return wkbPointZM;
5412 : :
5413 : : case QgsWkbTypes::LineString:
5414 : 0 : return wkbLineString;
5415 : : case QgsWkbTypes::LineString25D:
5416 : : case QgsWkbTypes::LineStringZ:
5417 : 0 : return wkbLineString25D;
5418 : : case QgsWkbTypes::LineStringM:
5419 : 0 : return wkbLineStringM;
5420 : : case QgsWkbTypes::LineStringZM:
5421 : 0 : return wkbLineStringZM;
5422 : :
5423 : : case QgsWkbTypes::Polygon:
5424 : 0 : return wkbPolygon;
5425 : : case QgsWkbTypes::Polygon25D:
5426 : : case QgsWkbTypes::PolygonZ:
5427 : 0 : return wkbPolygon25D;
5428 : : case QgsWkbTypes::PolygonM:
5429 : 0 : return wkbPolygonM;
5430 : : case QgsWkbTypes::PolygonZM:
5431 : 0 : return wkbPolygonZM;
5432 : :
5433 : : case QgsWkbTypes::MultiPoint:
5434 : 0 : return wkbMultiPoint;
5435 : : case QgsWkbTypes::MultiPoint25D:
5436 : : case QgsWkbTypes::MultiPointZ:
5437 : 0 : return wkbMultiPoint25D;
5438 : : case QgsWkbTypes::MultiPointM:
5439 : 0 : return wkbMultiPointM;
5440 : : case QgsWkbTypes::MultiPointZM:
5441 : 0 : return wkbMultiPointZM;
5442 : :
5443 : : case QgsWkbTypes::MultiLineString:
5444 : 0 : return wkbMultiLineString;
5445 : : case QgsWkbTypes::MultiLineString25D:
5446 : : case QgsWkbTypes::MultiLineStringZ:
5447 : 0 : return wkbMultiLineString25D;
5448 : : case QgsWkbTypes::MultiLineStringM:
5449 : 0 : return wkbMultiLineStringM;
5450 : : case QgsWkbTypes::MultiLineStringZM:
5451 : 0 : return wkbMultiLineStringZM;
5452 : :
5453 : : case QgsWkbTypes::MultiPolygon:
5454 : 0 : return wkbMultiPolygon;
5455 : : case QgsWkbTypes::MultiPolygon25D:
5456 : : case QgsWkbTypes::MultiPolygonZ:
5457 : 0 : return wkbMultiPolygon25D;
5458 : : case QgsWkbTypes::MultiPolygonM:
5459 : 0 : return wkbMultiPolygonM;
5460 : : case QgsWkbTypes::MultiPolygonZM:
5461 : 0 : return wkbMultiPolygonZM;
5462 : :
5463 : : case QgsWkbTypes::CircularString:
5464 : 0 : return wkbCircularString;
5465 : : case QgsWkbTypes::CircularStringZ:
5466 : 0 : return wkbCircularStringZ;
5467 : : case QgsWkbTypes::CircularStringM:
5468 : 0 : return wkbCircularStringM;
5469 : : case QgsWkbTypes::CircularStringZM:
5470 : 0 : return wkbCircularStringZM;
5471 : :
5472 : : case QgsWkbTypes::CompoundCurve:
5473 : 0 : return wkbCompoundCurve;
5474 : : case QgsWkbTypes::CompoundCurveZ:
5475 : 0 : return wkbCompoundCurveZ;
5476 : : case QgsWkbTypes::CompoundCurveM:
5477 : 0 : return wkbCompoundCurveM;
5478 : : case QgsWkbTypes::CompoundCurveZM:
5479 : 0 : return wkbCompoundCurveZM;
5480 : :
5481 : : case QgsWkbTypes::CurvePolygon:
5482 : 0 : return wkbCurvePolygon;
5483 : : case QgsWkbTypes::CurvePolygonZ:
5484 : 0 : return wkbCurvePolygonZ;
5485 : : case QgsWkbTypes::CurvePolygonM:
5486 : 0 : return wkbCurvePolygonM;
5487 : : case QgsWkbTypes::CurvePolygonZM:
5488 : 0 : return wkbCurvePolygonZM;
5489 : :
5490 : : case QgsWkbTypes::MultiCurve:
5491 : 0 : return wkbMultiCurve;
5492 : : case QgsWkbTypes::MultiCurveZ:
5493 : 0 : return wkbMultiCurveZ;
5494 : : case QgsWkbTypes::MultiCurveM:
5495 : 0 : return wkbMultiCurveM;
5496 : : case QgsWkbTypes::MultiCurveZM:
5497 : 0 : return wkbMultiCurveZM;
5498 : :
5499 : : case QgsWkbTypes::MultiSurface:
5500 : 0 : return wkbMultiSurface;
5501 : : case QgsWkbTypes::MultiSurfaceZ:
5502 : 0 : return wkbMultiSurfaceZ;
5503 : : case QgsWkbTypes::MultiSurfaceM:
5504 : 0 : return wkbMultiSurfaceM;
5505 : : case QgsWkbTypes::MultiSurfaceZM:
5506 : 0 : return wkbMultiSurfaceZM;
5507 : :
5508 : : case QgsWkbTypes::Triangle:
5509 : 0 : return wkbTriangle;
5510 : : case QgsWkbTypes::TriangleZ:
5511 : 0 : return wkbTriangleZ;
5512 : : case QgsWkbTypes::TriangleM:
5513 : 0 : return wkbTriangleM;
5514 : : case QgsWkbTypes::TriangleZM:
5515 : 0 : return wkbTriangleZM;
5516 : :
5517 : : case QgsWkbTypes::NoGeometry:
5518 : 0 : return wkbNone;
5519 : :
5520 : : case QgsWkbTypes::GeometryCollection:
5521 : 0 : return wkbGeometryCollection;
5522 : : case QgsWkbTypes::GeometryCollectionZ:
5523 : 0 : return wkbGeometryCollection25D;
5524 : : case QgsWkbTypes::GeometryCollectionM:
5525 : 0 : return wkbGeometryCollectionM;
5526 : : case QgsWkbTypes::GeometryCollectionZM:
5527 : 0 : return wkbGeometryCollectionZM;
5528 : :
5529 : : case QgsWkbTypes::Unknown:
5530 : 0 : return wkbUnknown;
5531 : : }
5532 : : // no warnings!
5533 : 0 : return wkbUnknown;
5534 : 0 : }
5535 : :
5536 : 0 : QgsWkbTypes::Type QgsOgrProviderUtils::qgisTypeFromOgrType( OGRwkbGeometryType type )
5537 : : {
5538 : 0 : switch ( type )
5539 : : {
5540 : : case wkbUnknown:
5541 : 0 : return QgsWkbTypes::Unknown;
5542 : : case wkbPoint:
5543 : 0 : return QgsWkbTypes::Point;
5544 : : case wkbLineString:
5545 : 0 : return QgsWkbTypes::LineString;
5546 : : case wkbPolygon:
5547 : 0 : return QgsWkbTypes::Polygon;
5548 : : case wkbMultiPoint:
5549 : 0 : return QgsWkbTypes::MultiPoint;
5550 : : case wkbMultiLineString:
5551 : 0 : return QgsWkbTypes::MultiLineString;
5552 : : case wkbMultiPolygon:
5553 : 0 : return QgsWkbTypes::MultiPolygon;
5554 : : case wkbGeometryCollection:
5555 : 0 : return QgsWkbTypes::GeometryCollection;
5556 : : case wkbCircularString:
5557 : 0 : return QgsWkbTypes::CircularString;
5558 : : case wkbCompoundCurve:
5559 : 0 : return QgsWkbTypes::CompoundCurve;
5560 : : case wkbCurvePolygon:
5561 : 0 : return QgsWkbTypes::CurvePolygon;
5562 : : case wkbMultiCurve:
5563 : 0 : return QgsWkbTypes::MultiCurve;
5564 : : case wkbMultiSurface:
5565 : 0 : return QgsWkbTypes::MultiSurface;
5566 : : case wkbTriangle:
5567 : 0 : return QgsWkbTypes::Triangle;
5568 : : case wkbNone:
5569 : 0 : return QgsWkbTypes::NoGeometry;
5570 : :
5571 : : case wkbCircularStringZ:
5572 : 0 : return QgsWkbTypes::CircularStringZ;
5573 : : case wkbCompoundCurveZ:
5574 : 0 : return QgsWkbTypes::CompoundCurveZ;
5575 : : case wkbCurvePolygonZ:
5576 : 0 : return QgsWkbTypes::PolygonZ;
5577 : : case wkbMultiCurveZ:
5578 : 0 : return QgsWkbTypes::MultiCurveZ;
5579 : : case wkbMultiSurfaceZ:
5580 : 0 : return QgsWkbTypes::MultiSurfaceZ;
5581 : : case wkbTriangleZ:
5582 : 0 : return QgsWkbTypes::TriangleZ;
5583 : :
5584 : : case wkbPointM:
5585 : 0 : return QgsWkbTypes::PointM;
5586 : : case wkbLineStringM:
5587 : 0 : return QgsWkbTypes::LineStringM;
5588 : : case wkbPolygonM:
5589 : 0 : return QgsWkbTypes::PolygonM;
5590 : : case wkbMultiPointM:
5591 : 0 : return QgsWkbTypes::PointM;
5592 : : case wkbMultiLineStringM:
5593 : 0 : return QgsWkbTypes::LineStringM;
5594 : : case wkbMultiPolygonM:
5595 : 0 : return QgsWkbTypes::PolygonM;
5596 : : case wkbGeometryCollectionM:
5597 : 0 : return QgsWkbTypes::GeometryCollectionM;
5598 : : case wkbCircularStringM:
5599 : 0 : return QgsWkbTypes::CircularStringM;
5600 : : case wkbCompoundCurveM:
5601 : 0 : return QgsWkbTypes::CompoundCurveM;
5602 : : case wkbCurvePolygonM:
5603 : 0 : return QgsWkbTypes::PolygonM;
5604 : : case wkbMultiCurveM:
5605 : 0 : return QgsWkbTypes::MultiCurveM;
5606 : : case wkbMultiSurfaceM:
5607 : 0 : return QgsWkbTypes::MultiSurfaceM;
5608 : : case wkbTriangleM:
5609 : 0 : return QgsWkbTypes::TriangleM;
5610 : :
5611 : : case wkbPointZM:
5612 : 0 : return QgsWkbTypes::PointZM;
5613 : : case wkbLineStringZM:
5614 : 0 : return QgsWkbTypes::LineStringZM;
5615 : : case wkbPolygonZM:
5616 : 0 : return QgsWkbTypes::PolygonZM;
5617 : : case wkbMultiPointZM:
5618 : 0 : return QgsWkbTypes::MultiPointZM;
5619 : : case wkbMultiLineStringZM:
5620 : 0 : return QgsWkbTypes::MultiLineStringZM;
5621 : : case wkbMultiPolygonZM:
5622 : 0 : return QgsWkbTypes::MultiPolygonZM;
5623 : : case wkbGeometryCollectionZM:
5624 : 0 : return QgsWkbTypes::GeometryCollectionZM;
5625 : : case wkbCircularStringZM:
5626 : 0 : return QgsWkbTypes::CircularStringZM;
5627 : : case wkbCompoundCurveZM:
5628 : 0 : return QgsWkbTypes::CompoundCurveZM;
5629 : : case wkbCurvePolygonZM:
5630 : 0 : return QgsWkbTypes::CurvePolygonZM;
5631 : : case wkbMultiCurveZM:
5632 : 0 : return QgsWkbTypes::MultiCurveZM;
5633 : : case wkbMultiSurfaceZM:
5634 : 0 : return QgsWkbTypes::MultiSurfaceZM;
5635 : : case wkbTriangleZM:
5636 : 0 : return QgsWkbTypes::TriangleZM;
5637 : :
5638 : : case wkbPoint25D:
5639 : 0 : return QgsWkbTypes::Point25D;
5640 : : case wkbLineString25D:
5641 : 0 : return QgsWkbTypes::LineString25D;
5642 : : case wkbPolygon25D:
5643 : 0 : return QgsWkbTypes::Polygon25D;
5644 : : case wkbMultiPoint25D:
5645 : 0 : return QgsWkbTypes::MultiPoint25D;
5646 : : case wkbMultiLineString25D:
5647 : 0 : return QgsWkbTypes::MultiLineString25D;
5648 : : case wkbMultiPolygon25D:
5649 : 0 : return QgsWkbTypes::MultiPolygon25D;
5650 : : case wkbGeometryCollection25D:
5651 : 0 : return QgsWkbTypes::GeometryCollectionZ;
5652 : :
5653 : : case wkbCurve:
5654 : : case wkbSurface:
5655 : : case wkbCurveZ:
5656 : : case wkbSurfaceZ:
5657 : : case wkbCurveM:
5658 : : case wkbSurfaceM:
5659 : : case wkbCurveZM:
5660 : : case wkbSurfaceZM:
5661 : 0 : return QgsWkbTypes::Unknown; // abstract types - no direct mapping to QGIS types
5662 : :
5663 : : case wkbLinearRing:
5664 : : case wkbTIN:
5665 : : case wkbTINZ:
5666 : : case wkbTINM:
5667 : : case wkbTINZM:
5668 : : case wkbPolyhedralSurface:
5669 : : case wkbPolyhedralSurfaceZ:
5670 : : case wkbPolyhedralSurfaceM:
5671 : : case wkbPolyhedralSurfaceZM:
5672 : 0 : return QgsWkbTypes::Unknown; // unsupported types
5673 : : }
5674 : 0 : return QgsWkbTypes::Unknown;
5675 : 0 : }
5676 : :
5677 : :
5678 : 250 : QString QgsOgrProviderUtils::expandAuthConfig( const QString &dsName )
5679 : : {
5680 : 250 : QString uri( dsName );
5681 : : // Check for authcfg
5682 : 250 : QRegularExpression authcfgRe( " authcfg='([^']+)'" );
5683 : 250 : QRegularExpressionMatch match;
5684 : 250 : if ( uri.contains( authcfgRe, &match ) )
5685 : : {
5686 : 0 : uri = uri.replace( match.captured( 0 ), QString() );
5687 : 0 : QString configId( match.captured( 1 ) );
5688 : 0 : QStringList connectionItems;
5689 : 0 : connectionItems << uri;
5690 : 0 : if ( QgsApplication::authManager()->updateDataSourceUriItems( connectionItems, configId, QStringLiteral( "ogr" ) ) )
5691 : : {
5692 : 0 : uri = connectionItems.first( );
5693 : 0 : }
5694 : 0 : }
5695 : 250 : return uri;
5696 : 250 : }
5697 : :
5698 : : // Must be called under the globalMutex
5699 : 0 : bool QgsOgrProviderUtils::canUseOpenedDatasets( const QString &dsName )
5700 : : {
5701 : 0 : auto iter = sMapDSNameToLastModifiedDate()->find( dsName );
5702 : 0 : if ( iter == sMapDSNameToLastModifiedDate()->end() )
5703 : 0 : return true;
5704 : 0 : return getLastModified( dsName ) <= iter.value();
5705 : 0 : }
5706 : :
5707 : 0 : QgsOgrProviderUtils::DatasetWithLayers *QgsOgrProviderUtils::createDatasetWithLayers(
5708 : : const QString &dsName,
5709 : : bool updateMode,
5710 : : const QStringList &options,
5711 : : const QString &layerName,
5712 : : const DatasetIdentification &ident,
5713 : : QgsOgrLayerUniquePtr &layer,
5714 : : QString &errCause )
5715 : : {
5716 : 0 : GDALDatasetH hDS = OpenHelper( dsName, updateMode, options );
5717 : 0 : if ( !hDS )
5718 : : {
5719 : 0 : errCause = QObject::tr( "Cannot open %1." ).arg( dsName );
5720 : 0 : return nullptr;
5721 : : }
5722 : 0 : ( *sMapDSNameToLastModifiedDate() )[dsName] = getLastModified( dsName );
5723 : :
5724 : 0 : OGRLayerH hLayer = GDALDatasetGetLayerByName(
5725 : 0 : hDS, layerName.toUtf8().constData() );
5726 : 0 : if ( !hLayer )
5727 : : {
5728 : 0 : errCause = QObject::tr( "Cannot find layer %1." ).arg( layerName );
5729 : 0 : QgsOgrProviderUtils::GDALCloseWrapper( hDS );
5730 : 0 : return nullptr;
5731 : : }
5732 : :
5733 : 0 : QgsOgrProviderUtils::DatasetWithLayers *ds =
5734 : 0 : new QgsOgrProviderUtils::DatasetWithLayers;
5735 : 0 : ds->hDS = hDS;
5736 : :
5737 : 0 : GDALDriverH driver = GDALGetDatasetDriver( hDS );
5738 : 0 : QString driverName = GDALGetDriverShortName( driver );
5739 : 0 : ds->canBeShared = canDriverShareSameDatasetAmongLayers( driverName, updateMode, dsName );
5740 : :
5741 : 0 : layer = QgsOgrLayer::CreateForLayer(
5742 : 0 : ident, layerName, ds, hLayer );
5743 : 0 : ds->setLayers[layerName] = layer.get();
5744 : 0 : return ds;
5745 : 0 : }
5746 : :
5747 : 0 : QgsOgrLayerUniquePtr QgsOgrProviderUtils::getLayer( const QString &dsName,
5748 : : bool updateMode,
5749 : : const QStringList &options,
5750 : : const QString &layerName,
5751 : : QString &errCause,
5752 : : bool checkModificationDateAgainstCache )
5753 : : {
5754 : 0 : QMutexLocker locker( sGlobalMutex() );
5755 : :
5756 : : // The idea is that we want to minimize the number of GDALDatasetH
5757 : : // handles openeded. But we have constraints. We do not want that 2
5758 : : // callers of getLayer() with the same input parameters get the same
5759 : : // GDALDatasetH since iterators over features of that layer would conflict
5760 : :
5761 : 0 : QgsOgrProviderUtils::DatasetIdentification ident;
5762 : 0 : ident.dsName = dsName;
5763 : 0 : ident.updateMode = updateMode;
5764 : 0 : ident.options = options;
5765 : : // Find if there's a list of DatasetWithLayers* that match our
5766 : : // (dsName, updateMode, options) criteria
5767 : 0 : auto iter = sMapSharedDS.find( ident );
5768 : 0 : if ( iter != sMapSharedDS.end() )
5769 : : {
5770 : 0 : if ( checkModificationDateAgainstCache && !canUseOpenedDatasets( dsName ) )
5771 : : {
5772 : 0 : QgsDebugMsg( QStringLiteral( "Cannot reuse existing opened dataset(s) on %1 since it has been modified" ).arg( dsName ) );
5773 : 0 : invalidateCachedDatasets( dsName );
5774 : 0 : iter = sMapSharedDS.find( ident );
5775 : : Q_ASSERT( iter == sMapSharedDS.end() );
5776 : 0 : }
5777 : 0 : }
5778 : 0 : if ( iter != sMapSharedDS.end() )
5779 : : {
5780 : : // Browse through this list, to look for a DatasetWithLayers*
5781 : : // instance that don't use yet our layer of interest
5782 : 0 : auto &datasetList = iter.value();
5783 : 0 : const auto constDatasetList = datasetList;
5784 : 0 : for ( QgsOgrProviderUtils::DatasetWithLayers *ds : constDatasetList )
5785 : : {
5786 : 0 : if ( !ds->canBeShared )
5787 : 0 : continue;
5788 : : Q_ASSERT( ds->refCount > 0 );
5789 : :
5790 : 0 : auto iter2 = ds->setLayers.find( layerName );
5791 : 0 : if ( iter2 == ds->setLayers.end() )
5792 : : {
5793 : : OGRLayerH hLayer;
5794 : : {
5795 : 0 : QMutexLocker lockerDS( &ds->mutex );
5796 : 0 : hLayer = GDALDatasetGetLayerByName(
5797 : 0 : ds->hDS, layerName.toUtf8().constData() );
5798 : 0 : }
5799 : 0 : if ( !hLayer )
5800 : : {
5801 : : // Shouldn't happen really !
5802 : 0 : errCause = QObject::tr( "Cannot find layer %1." ).arg( layerName );
5803 : 0 : return nullptr;
5804 : : }
5805 : 0 : OGR_L_SetAttributeFilter( hLayer, nullptr );
5806 : :
5807 : 0 : QgsOgrLayerUniquePtr layer = QgsOgrLayer::CreateForLayer(
5808 : 0 : ident, layerName, ds, hLayer );
5809 : 0 : ds->setLayers[layerName] = layer.get();
5810 : 0 : return layer;
5811 : 0 : }
5812 : : }
5813 : :
5814 : : // All existing DatasetWithLayers* already reference our layer of
5815 : : // interest, so instantiate a new DatasetWithLayers*
5816 : 0 : QgsOgrLayerUniquePtr layer;
5817 : 0 : QgsOgrProviderUtils::DatasetWithLayers *ds =
5818 : 0 : createDatasetWithLayers( dsName, updateMode, options, layerName, ident, layer, errCause );
5819 : 0 : if ( !ds )
5820 : 0 : return nullptr;
5821 : :
5822 : 0 : datasetList.push_back( ds );
5823 : :
5824 : 0 : return layer;
5825 : 0 : }
5826 : :
5827 : 0 : QgsOgrLayerUniquePtr layer;
5828 : 0 : QgsOgrProviderUtils::DatasetWithLayers *ds =
5829 : 0 : createDatasetWithLayers( dsName, updateMode, options, layerName, ident, layer, errCause );
5830 : 0 : if ( !ds )
5831 : 0 : return nullptr;
5832 : :
5833 : 0 : QList<DatasetWithLayers *> datasetList;
5834 : 0 : datasetList.push_back( ds );
5835 : 0 : sMapSharedDS[ident] = datasetList;
5836 : :
5837 : 0 : return layer;
5838 : 0 : }
5839 : :
5840 : 0 : QgsOgrLayerUniquePtr QgsOgrProviderUtils::getSqlLayer( QgsOgrLayer *baseLayer,
5841 : : OGRLayerH hSqlLayer,
5842 : : const QString &sql )
5843 : : {
5844 : 0 : QgsOgrProviderUtils::DatasetIdentification ident;
5845 : 0 : ident.dsName = baseLayer->datasetName();
5846 : 0 : ident.updateMode = baseLayer->updateMode();
5847 : 0 : ident.options = baseLayer->options();
5848 : 0 : return QgsOgrLayer::CreateForSql( ident, sql, baseLayer->ds, hSqlLayer );
5849 : 0 : }
5850 : :
5851 : 247 : void QgsOgrProviderUtils::releaseInternal( const DatasetIdentification &ident,
5852 : : DatasetWithLayers *ds,
5853 : : bool removeFromDatasetList )
5854 : : {
5855 : :
5856 : 247 : ds->refCount --;
5857 : 247 : if ( ds->refCount == 0 )
5858 : : {
5859 : : Q_ASSERT( ds->setLayers.isEmpty() );
5860 : :
5861 : 247 : if ( removeFromDatasetList )
5862 : : {
5863 : 247 : auto iter = sMapSharedDS.find( ident );
5864 : 247 : if ( iter != sMapSharedDS.end() )
5865 : : {
5866 : 247 : auto &datasetList = iter.value();
5867 : 247 : int i = 0;
5868 : :
5869 : : // Normally there should be a match, except for datasets that
5870 : : // have been invalidated
5871 : 247 : const auto constDatasetList = datasetList;
5872 : 247 : for ( QgsOgrProviderUtils::DatasetWithLayers *dsIter : constDatasetList )
5873 : : {
5874 : 247 : if ( dsIter == ds )
5875 : : {
5876 : 247 : datasetList.removeAt( i );
5877 : 247 : break;
5878 : : }
5879 : 0 : i ++;
5880 : : }
5881 : :
5882 : 247 : if ( datasetList.isEmpty() )
5883 : 247 : sMapSharedDS.erase( iter );
5884 : 247 : }
5885 : 247 : }
5886 : 247 : QgsOgrProviderUtils::GDALCloseWrapper( ds->hDS );
5887 : 247 : delete ds;
5888 : 247 : }
5889 : 247 : }
5890 : :
5891 : 247 : void QgsOgrProviderUtils::release( QgsOgrLayer *&layer )
5892 : : {
5893 : 247 : if ( !layer )
5894 : 0 : return;
5895 : :
5896 : 247 : QMutexLocker locker( sGlobalMutex() );
5897 : :
5898 : 247 : if ( !layer->isSqlLayer )
5899 : : {
5900 : 247 : layer->ds->setLayers.remove( layer->layerName );
5901 : 247 : }
5902 : : else
5903 : : {
5904 : 0 : QMutexLocker lockerDS( &layer->ds->mutex );
5905 : 0 : GDALDatasetReleaseResultSet( layer->ds->hDS, layer->hLayer );
5906 : 0 : }
5907 : :
5908 : 247 : releaseInternal( layer->ident, layer->ds, !layer->isSqlLayer );
5909 : :
5910 : 247 : delete layer;
5911 : 247 : layer = nullptr;
5912 : 247 : }
5913 : :
5914 : :
5915 : 0 : void QgsOgrProviderUtils::releaseDataset( QgsOgrDataset *&ds )
5916 : : {
5917 : 0 : if ( !ds )
5918 : 0 : return;
5919 : :
5920 : 0 : QMutexLocker locker( sGlobalMutex() );
5921 : 0 : releaseInternal( ds->mIdent, ds->mDs, true );
5922 : 0 : delete ds;
5923 : 0 : ds = nullptr;
5924 : 0 : }
5925 : :
5926 : 1795 : bool QgsOgrProviderUtils::canDriverShareSameDatasetAmongLayers( const QString &driverName )
5927 : : {
5928 : 1795 : return driverName != QLatin1String( "OSM" );
5929 : : }
5930 : :
5931 : 250 : bool QgsOgrProviderUtils::canDriverShareSameDatasetAmongLayers( const QString &driverName,
5932 : : bool updateMode,
5933 : : const QString &dsName )
5934 : : {
5935 : : // For .shp.zip with multiple layers, in update mode, we want that each
5936 : : // layer has its own dataset, so that when its gets closed and reopened,
5937 : : // the .shp.zip is updated. Otherwise if we share the same dataset, the .shp.zip
5938 : : // would only be updated when all layers are unloaded, and thus readers will see
5939 : : // an outdated version of the .shp.zip. This works only if editing operations are
5940 : : // done separately on layers (which is how it works from the GUI)
5941 : 500 : return canDriverShareSameDatasetAmongLayers( driverName ) &&
5942 : 250 : !( updateMode && dsName.endsWith( QLatin1String( ".shp.zip" ), Qt::CaseInsensitive ) );
5943 : : }
5944 : :
5945 : :
5946 : 0 : QgsOgrDatasetSharedPtr QgsOgrDataset::create( const QgsOgrProviderUtils::DatasetIdentification &ident,
5947 : : QgsOgrProviderUtils::DatasetWithLayers *ds )
5948 : : {
5949 : 0 : QgsOgrDatasetSharedPtr dsRet = QgsOgrDatasetSharedPtr( new QgsOgrDataset(), QgsOgrProviderUtils::releaseDataset );
5950 : 0 : dsRet->mIdent = ident;
5951 : 0 : dsRet->mDs = ds;
5952 : 0 : dsRet->mDs->refCount ++;
5953 : 0 : return dsRet;
5954 : 0 : }
5955 : :
5956 : 0 : bool QgsOgrDataset::executeSQLNoReturn( const QString &sql )
5957 : : {
5958 : 0 : QMutexLocker locker( &mutex() );
5959 : 0 : CPLErrorReset();
5960 : 0 : OGRLayerH hSqlLayer = GDALDatasetExecuteSQL(
5961 : 0 : mDs->hDS, sql.toUtf8().constData(), nullptr, nullptr );
5962 : 0 : bool ret = CPLGetLastErrorType() == CE_None;
5963 : 0 : GDALDatasetReleaseResultSet( mDs->hDS, hSqlLayer );
5964 : 0 : return ret;
5965 : 0 : }
5966 : :
5967 : :
5968 : 0 : OGRLayerH QgsOgrDataset::getLayerFromNameOrIndex( const QString &layerName, int layerIndex )
5969 : : {
5970 : 0 : QMutexLocker locker( &mutex() );
5971 : :
5972 : : OGRLayerH layer;
5973 : 0 : if ( !layerName.isEmpty() )
5974 : : {
5975 : 0 : layer = GDALDatasetGetLayerByName( mDs->hDS, layerName.toUtf8().constData() );
5976 : 0 : }
5977 : : else
5978 : : {
5979 : 0 : layer = GDALDatasetGetLayer( mDs->hDS, layerIndex );
5980 : : }
5981 : 0 : return layer;
5982 : 0 : }
5983 : :
5984 : 0 : void QgsOgrDataset::releaseResultSet( OGRLayerH hSqlLayer )
5985 : : {
5986 : 0 : QMutexLocker locker( &mutex() );
5987 : 0 : GDALDatasetReleaseResultSet( mDs->hDS, hSqlLayer );
5988 : 0 : }
5989 : :
5990 : 250 : QgsOgrLayer::QgsOgrLayer()
5991 : : {
5992 : 250 : oFDefn.layer = this;
5993 : 250 : }
5994 : :
5995 : 250 : QgsOgrLayerUniquePtr QgsOgrLayer::CreateForLayer(
5996 : : const QgsOgrProviderUtils::DatasetIdentification &ident,
5997 : : const QString &layerName,
5998 : : QgsOgrProviderUtils::DatasetWithLayers *ds,
5999 : : OGRLayerH hLayer )
6000 : : {
6001 : 250 : QgsOgrLayerUniquePtr layer( new QgsOgrLayer() );
6002 : 250 : layer->ident = ident;
6003 : 250 : layer->isSqlLayer = false;
6004 : 250 : layer->layerName = layerName;
6005 : 250 : layer->ds = ds;
6006 : 250 : layer->hLayer = hLayer;
6007 : : {
6008 : 250 : QMutexLocker locker( &ds->mutex );
6009 : 250 : OGR_L_ResetReading( hLayer );
6010 : 250 : }
6011 : 250 : ds->refCount ++;
6012 : 250 : return layer;
6013 : 250 : }
6014 : :
6015 : 0 : QgsOgrLayerUniquePtr QgsOgrLayer::CreateForSql(
6016 : : const QgsOgrProviderUtils::DatasetIdentification &ident,
6017 : : const QString &sql,
6018 : : QgsOgrProviderUtils::DatasetWithLayers *ds,
6019 : : OGRLayerH hLayer )
6020 : : {
6021 : 0 : QgsOgrLayerUniquePtr layer( new QgsOgrLayer() );
6022 : 0 : layer->ident = ident;
6023 : 0 : layer->isSqlLayer = true;
6024 : 0 : layer->sql = sql;
6025 : 0 : layer->ds = ds;
6026 : 0 : layer->hLayer = hLayer;
6027 : : {
6028 : 0 : QMutexLocker locker( &ds->mutex );
6029 : 0 : OGR_L_ResetReading( hLayer );
6030 : 0 : }
6031 : 0 : ds->refCount ++;
6032 : 0 : return layer;
6033 : 0 : }
6034 : :
6035 : 0 : int QgsOgrLayer::GetLayerCount()
6036 : : {
6037 : 0 : QMutexLocker locker( &ds->mutex );
6038 : 0 : return GDALDatasetGetLayerCount( ds->hDS );
6039 : 0 : }
6040 : :
6041 : 192 : GDALDriverH QgsOgrLayer::driver()
6042 : : {
6043 : 192 : return GDALGetDatasetDriver( ds->hDS );
6044 : : }
6045 : :
6046 : 187 : QString QgsOgrLayer::driverName()
6047 : : {
6048 : 187 : return QString::fromUtf8( GDALGetDriverShortName( GDALGetDatasetDriver( ds->hDS ) ) );
6049 : : }
6050 : :
6051 : 60 : QByteArray QgsOgrLayer::name()
6052 : : {
6053 : 60 : QMutexLocker locker( &ds->mutex );
6054 : 60 : return OGR_L_GetName( hLayer );
6055 : 60 : }
6056 : :
6057 : 180 : void QgsOgrLayer::ResetReading()
6058 : : {
6059 : 180 : QMutexLocker locker( &ds->mutex );
6060 : 180 : OGR_L_ResetReading( hLayer );
6061 : 180 : }
6062 : :
6063 : 187 : QByteArray QgsOgrLayer::GetFIDColumn()
6064 : : {
6065 : 187 : QMutexLocker locker( &ds->mutex );
6066 : 187 : return OGR_L_GetFIDColumn( hLayer );
6067 : 187 : }
6068 : :
6069 : 212 : OGRSpatialReferenceH QgsOgrLayer::GetSpatialRef()
6070 : : {
6071 : 212 : QMutexLocker locker( &ds->mutex );
6072 : 212 : return OGR_L_GetSpatialRef( hLayer );
6073 : 212 : }
6074 : :
6075 : 0 : OGRFeatureH QgsOgrLayer::GetNextFeature()
6076 : : {
6077 : 0 : QMutexLocker locker( &ds->mutex );
6078 : 0 : return OGR_L_GetNextFeature( hLayer );
6079 : 0 : }
6080 : :
6081 : 30 : OGRFeatureH QgsOgrLayer::GetFeature( GIntBig fid )
6082 : : {
6083 : 30 : QMutexLocker locker( &ds->mutex );
6084 : 30 : return OGR_L_GetFeature( hLayer, fid );
6085 : 30 : }
6086 : :
6087 : 206 : QgsOgrFeatureDefn &QgsOgrLayer::GetLayerDefn()
6088 : : {
6089 : 206 : return oFDefn;
6090 : : }
6091 : :
6092 : 0 : GIntBig QgsOgrLayer::GetFeatureCount( bool force )
6093 : : {
6094 : 0 : QMutexLocker locker( &ds->mutex );
6095 : 0 : return OGR_L_GetFeatureCount( hLayer, force );
6096 : 0 : }
6097 : :
6098 : 17 : GIntBig QgsOgrLayer::GetApproxFeatureCount()
6099 : : {
6100 : 17 : QMutexLocker locker( &ds->mutex );
6101 : :
6102 : : // OGR_L_GetFeatureCount() can be super slow on huge geopackage files
6103 : : // so implement some approximation strategy that has reasonable runtime.
6104 : 17 : QString driverName = GDALGetDriverShortName( GDALGetDatasetDriver( ds->hDS ) );
6105 : 17 : if ( driverName == QLatin1String( "GPKG" ) )
6106 : : {
6107 : 0 : CPLPushErrorHandler( CPLQuietErrorHandler );
6108 : 0 : OGRLayerH hSqlLayer = GDALDatasetExecuteSQL(
6109 : 0 : ds->hDS, "SELECT 1 FROM gpkg_ogr_contents LIMIT 0", nullptr, nullptr );
6110 : 0 : CPLPopErrorHandler();
6111 : 0 : if ( hSqlLayer )
6112 : : {
6113 : 0 : GDALDatasetReleaseResultSet( ds->hDS, hSqlLayer );
6114 : 0 : return OGR_L_GetFeatureCount( hLayer, TRUE );
6115 : : }
6116 : :
6117 : : // Enumerate features up to a limit of 100000.
6118 : 0 : const GIntBig nLimit = CPLAtoGIntBig(
6119 : 0 : CPLGetConfigOption( "QGIS_GPKG_FC_THRESHOLD", "100000" ) );
6120 : 0 : QByteArray layerName = OGR_L_GetName( hLayer );
6121 : 0 : QByteArray sql( "SELECT COUNT(*) FROM (SELECT 1 FROM " );
6122 : 0 : sql += QgsOgrProviderUtils::quotedIdentifier( layerName, driverName );
6123 : 0 : sql += " LIMIT ";
6124 : 0 : sql += CPLSPrintf( CPL_FRMT_GIB, nLimit );
6125 : 0 : sql += ")";
6126 : 0 : hSqlLayer = GDALDatasetExecuteSQL( ds->hDS, sql, nullptr, nullptr );
6127 : 0 : GIntBig res = -1;
6128 : 0 : if ( hSqlLayer )
6129 : : {
6130 : 0 : gdal::ogr_feature_unique_ptr fet( OGR_L_GetNextFeature( hSqlLayer ) );
6131 : 0 : if ( fet )
6132 : : {
6133 : 0 : res = OGR_F_GetFieldAsInteger64( fet.get(), 0 );
6134 : 0 : }
6135 : 0 : GDALDatasetReleaseResultSet( ds->hDS, hSqlLayer );
6136 : 0 : }
6137 : 0 : if ( res >= 0 && res < nLimit )
6138 : : {
6139 : : // Less than 100000 features ? This is the final count
6140 : 0 : return res;
6141 : : }
6142 : 0 : if ( res == nLimit )
6143 : : {
6144 : : // If we reach the threshold, then use the min and max values of the rowid
6145 : : // hoping there are not a lot of holes.
6146 : : // Do it in 2 separate SQL queries otherwise SQLite apparently does a
6147 : : // full table scan...
6148 : 0 : sql = "SELECT MAX(ROWID) FROM ";
6149 : 0 : sql += QgsOgrProviderUtils::quotedIdentifier( layerName, driverName );
6150 : 0 : hSqlLayer = GDALDatasetExecuteSQL( ds->hDS, sql, nullptr, nullptr );
6151 : 0 : GIntBig maxrowid = -1;
6152 : 0 : if ( hSqlLayer )
6153 : : {
6154 : 0 : gdal::ogr_feature_unique_ptr fet( OGR_L_GetNextFeature( hSqlLayer ) );
6155 : 0 : if ( fet )
6156 : : {
6157 : 0 : maxrowid = OGR_F_GetFieldAsInteger64( fet.get(), 0 );
6158 : 0 : }
6159 : 0 : GDALDatasetReleaseResultSet( ds->hDS, hSqlLayer );
6160 : 0 : }
6161 : :
6162 : 0 : sql = "SELECT MIN(ROWID) FROM ";
6163 : 0 : sql += QgsOgrProviderUtils::quotedIdentifier( layerName, driverName );
6164 : 0 : hSqlLayer = GDALDatasetExecuteSQL( ds->hDS, sql, nullptr, nullptr );
6165 : 0 : GIntBig minrowid = 0;
6166 : 0 : if ( hSqlLayer )
6167 : : {
6168 : 0 : gdal::ogr_feature_unique_ptr fet( OGR_L_GetNextFeature( hSqlLayer ) );
6169 : 0 : if ( fet )
6170 : : {
6171 : 0 : minrowid = OGR_F_GetFieldAsInteger64( fet.get(), 0 );
6172 : 0 : }
6173 : 0 : GDALDatasetReleaseResultSet( ds->hDS, hSqlLayer );
6174 : 0 : }
6175 : :
6176 : 0 : if ( maxrowid >= minrowid )
6177 : : {
6178 : 0 : return maxrowid - minrowid + 1;
6179 : : }
6180 : 0 : }
6181 : 0 : }
6182 : 17 : if ( driverName == QLatin1String( "OAPIF" ) || driverName == QLatin1String( "OAPIF" ) )
6183 : : {
6184 : 0 : return -1;
6185 : : }
6186 : :
6187 : 17 : return OGR_L_GetFeatureCount( hLayer, TRUE );
6188 : 17 : }
6189 : :
6190 : 4 : OGRErr QgsOgrLayer::GetExtent( OGREnvelope *psExtent, bool bForce )
6191 : : {
6192 : 4 : QMutexLocker locker( &ds->mutex );
6193 : 4 : return OGR_L_GetExtent( hLayer, psExtent, bForce );
6194 : 4 : }
6195 : :
6196 : 17 : OGRGeometryH QgsOgrLayer::GetSpatialFilter()
6197 : : {
6198 : 17 : QMutexLocker locker( &ds->mutex );
6199 : 17 : return OGR_L_GetSpatialFilter( hLayer );
6200 : 17 : }
6201 : :
6202 : 0 : void QgsOgrLayer::SetSpatialFilter( OGRGeometryH hGeometry )
6203 : : {
6204 : 0 : QMutexLocker locker( &ds->mutex );
6205 : 0 : OGR_L_SetSpatialFilter( hLayer, hGeometry );
6206 : 0 : }
6207 : :
6208 : 0 : GDALDatasetH QgsOgrLayer::getDatasetHandleAndMutex( QMutex *&mutex )
6209 : : {
6210 : 0 : mutex = &( ds->mutex );
6211 : 0 : return ds->hDS;
6212 : : }
6213 : :
6214 : 463 : OGRLayerH QgsOgrLayer::getHandleAndMutex( QMutex *&mutex )
6215 : : {
6216 : 463 : mutex = &( ds->mutex );
6217 : 463 : return hLayer;
6218 : : }
6219 : :
6220 : 2 : OGRErr QgsOgrLayer::CreateFeature( OGRFeatureH hFeature )
6221 : : {
6222 : 2 : QMutexLocker locker( &ds->mutex );
6223 : 2 : return OGR_L_CreateFeature( hLayer, hFeature );
6224 : 2 : }
6225 : :
6226 : 30 : OGRErr QgsOgrLayer::SetFeature( OGRFeatureH hFeature )
6227 : : {
6228 : 30 : QMutexLocker locker( &ds->mutex );
6229 : 30 : return OGR_L_SetFeature( hLayer, hFeature );
6230 : 30 : }
6231 : :
6232 : 8 : OGRErr QgsOgrLayer::DeleteFeature( GIntBig fid )
6233 : : {
6234 : 8 : QMutexLocker locker( &ds->mutex );
6235 : 8 : return OGR_L_DeleteFeature( hLayer, fid );
6236 : 8 : }
6237 : :
6238 : 0 : OGRErr QgsOgrLayer::CreateField( OGRFieldDefnH hFieldDefn, bool bStrict )
6239 : : {
6240 : 0 : QMutexLocker locker( &ds->mutex );
6241 : 0 : return OGR_L_CreateField( hLayer, hFieldDefn, bStrict );
6242 : 0 : }
6243 : :
6244 : 0 : OGRErr QgsOgrLayer::DeleteField( int iField )
6245 : : {
6246 : 0 : QMutexLocker locker( &ds->mutex );
6247 : 0 : return OGR_L_DeleteField( hLayer, iField );
6248 : 0 : }
6249 : :
6250 : 0 : OGRErr QgsOgrLayer::AlterFieldDefn( int iField, OGRFieldDefnH hNewFieldDefn, int flags )
6251 : : {
6252 : 0 : QMutexLocker locker( &ds->mutex );
6253 : 0 : return OGR_L_AlterFieldDefn( hLayer, iField, hNewFieldDefn, flags );
6254 : 0 : }
6255 : :
6256 : 803 : int QgsOgrLayer::TestCapability( const char *cap )
6257 : : {
6258 : 803 : QMutexLocker locker( &ds->mutex );
6259 : 803 : return OGR_L_TestCapability( hLayer, cap );
6260 : 803 : }
6261 : :
6262 : 0 : OGRErr QgsOgrLayer::StartTransaction()
6263 : : {
6264 : 0 : QMutexLocker locker( &ds->mutex );
6265 : 0 : return OGR_L_StartTransaction( hLayer );
6266 : 0 : }
6267 : :
6268 : 0 : OGRErr QgsOgrLayer::CommitTransaction()
6269 : : {
6270 : 0 : QMutexLocker locker( &ds->mutex );
6271 : 0 : return OGR_L_CommitTransaction( hLayer );
6272 : 0 : }
6273 : :
6274 : 0 : OGRErr QgsOgrLayer::RollbackTransaction()
6275 : : {
6276 : 0 : QMutexLocker locker( &ds->mutex );
6277 : 0 : return OGR_L_RollbackTransaction( hLayer );
6278 : 0 : }
6279 : :
6280 : 40 : OGRErr QgsOgrLayer::SyncToDisk()
6281 : : {
6282 : 40 : QMutexLocker locker( &ds->mutex );
6283 : :
6284 : : OGRErr eErr;
6285 : : #if GDAL_VERSION_NUM >= GDAL_COMPUTE_VERSION(3,1,0) && GDAL_VERSION_NUM <= GDAL_COMPUTE_VERSION(3,1,3)
6286 : : // 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
6287 : : QString drvName = GDALGetDriverShortName( GDALGetDatasetDriver( ds->hDS ) );
6288 : : if ( drvName == QLatin1String( "XLSX" ) ||
6289 : : drvName == QLatin1String( "ODS" ) )
6290 : : {
6291 : : CPLSetThreadLocalConfigOption( "CPL_CREATE_ZIP64", "NO" );
6292 : : eErr = OGR_L_SyncToDisk( hLayer );
6293 : : CPLSetThreadLocalConfigOption( "CPL_CREATE_ZIP64", nullptr );
6294 : : }
6295 : : else
6296 : : #endif
6297 : : {
6298 : 40 : eErr = OGR_L_SyncToDisk( hLayer );
6299 : : }
6300 : :
6301 : 40 : return eErr;
6302 : 40 : }
6303 : :
6304 : 60 : void QgsOgrLayer::ExecuteSQLNoReturn( const QByteArray &sql )
6305 : : {
6306 : 60 : QMutexLocker locker( &ds->mutex );
6307 : 60 : OGRLayerH hSqlLayer = GDALDatasetExecuteSQL( ds->hDS,
6308 : 60 : sql.constData(),
6309 : : nullptr,
6310 : : nullptr );
6311 : 60 : GDALDatasetReleaseResultSet( ds->hDS, hSqlLayer );
6312 : 60 : }
6313 : :
6314 : 0 : QgsOgrLayerUniquePtr QgsOgrLayer::ExecuteSQL( const QByteArray &sql )
6315 : : {
6316 : 0 : QMutexLocker locker( &ds->mutex );
6317 : 0 : OGRLayerH hSqlLayer = GDALDatasetExecuteSQL( ds->hDS,
6318 : 0 : sql.constData(),
6319 : : nullptr,
6320 : : nullptr );
6321 : 0 : if ( !hSqlLayer )
6322 : 0 : return nullptr;
6323 : :
6324 : 0 : return QgsOgrLayer::CreateForSql( ident,
6325 : 0 : QString::fromUtf8( sql ),
6326 : 0 : ds,
6327 : 0 : hSqlLayer );
6328 : 0 : }
6329 : :
6330 : 126 : QString QgsOgrLayer::GetMetadataItem( const QString &key, const QString &domain )
6331 : : {
6332 : 126 : QMutexLocker locker( &ds->mutex );
6333 : 252 : return GDALGetMetadataItem( hLayer, key.toUtf8().constData(),
6334 : 126 : domain.toUtf8().constData() );
6335 : 126 : }
6336 : :
6337 : 890 : QMutex &QgsOgrFeatureDefn::mutex()
6338 : : {
6339 : 890 : return layer->mutex();
6340 : : }
6341 : :
6342 : 703 : OGRFeatureDefnH QgsOgrFeatureDefn::get()
6343 : : {
6344 : 703 : if ( !hDefn )
6345 : : {
6346 : 187 : QMutexLocker locker( &mutex() );
6347 : 187 : hDefn = OGR_L_GetLayerDefn( layer->hLayer );
6348 : 187 : }
6349 : 703 : return hDefn;
6350 : 0 : }
6351 : :
6352 : :
6353 : 435 : int QgsOgrFeatureDefn::GetFieldCount()
6354 : : {
6355 : 435 : QMutexLocker locker( &mutex() );
6356 : 435 : return OGR_FD_GetFieldCount( get() );
6357 : 435 : }
6358 : :
6359 : 248 : OGRFieldDefnH QgsOgrFeatureDefn::GetFieldDefn( int idx )
6360 : : {
6361 : 248 : QMutexLocker locker( &mutex() );
6362 : 248 : return OGR_FD_GetFieldDefn( get(), idx );
6363 : 248 : }
6364 : :
6365 : 1 : int QgsOgrFeatureDefn::GetFieldIndex( const QByteArray &name )
6366 : : {
6367 : 1 : QMutexLocker locker( &mutex() );
6368 : 1 : return OGR_FD_GetFieldIndex( get(), name.constData() );
6369 : 1 : }
6370 : :
6371 : 0 : int QgsOgrFeatureDefn::GetGeomFieldCount()
6372 : : {
6373 : 0 : QMutexLocker locker( &mutex() );
6374 : 0 : return OGR_FD_GetGeomFieldCount( get() );
6375 : 0 : }
6376 : :
6377 : 0 : OGRGeomFieldDefnH QgsOgrFeatureDefn::GetGeomFieldDefn( int idx )
6378 : : {
6379 : 0 : QMutexLocker locker( &mutex() );
6380 : 0 : return OGR_FD_GetGeomFieldDefn( get(), idx );
6381 : 0 : }
6382 : :
6383 : 17 : OGRwkbGeometryType QgsOgrFeatureDefn::GetGeomType()
6384 : : {
6385 : 17 : QMutexLocker locker( &mutex() );
6386 : 17 : return OGR_FD_GetGeomType( get() );
6387 : 17 : }
6388 : :
6389 : 2 : OGRFeatureH QgsOgrFeatureDefn::CreateFeature()
6390 : : {
6391 : 2 : QMutexLocker locker( &mutex() );
6392 : 2 : return OGR_F_Create( get() );
6393 : 2 : }
6394 : :
6395 : : // ---------------------------------------------------------------------------
6396 : :
6397 : : static
6398 : 0 : QgsOgrLayerUniquePtr LoadDataSourceAndLayer( const QString &uri,
6399 : : QString &errCause )
6400 : : {
6401 : : bool isSubLayer;
6402 : : int layerIndex;
6403 : 0 : QString layerName;
6404 : 0 : QString subsetString;
6405 : : OGRwkbGeometryType ogrGeometryType;
6406 : 0 : QStringList openOptions;
6407 : 0 : QString filePath = AnalyzeURI( uri,
6408 : : isSubLayer,
6409 : : layerIndex,
6410 : : layerName,
6411 : : subsetString,
6412 : : ogrGeometryType,
6413 : : openOptions );
6414 : :
6415 : 0 : if ( !layerName.isEmpty() )
6416 : : {
6417 : 0 : return QgsOgrProviderUtils::getLayer( filePath, true, QStringList(), layerName, errCause, true );
6418 : : }
6419 : : else
6420 : : {
6421 : 0 : return QgsOgrProviderUtils::getLayer( filePath, true, QStringList(), layerIndex, errCause, true );
6422 : : }
6423 : 0 : }
6424 : :
6425 : :
6426 : 0 : bool QgsOgrProviderMetadata::saveStyle(
6427 : : const QString &uri, const QString &qmlStyle, const QString &sldStyle,
6428 : : const QString &styleName, const QString &styleDescription,
6429 : : const QString &uiFileContent, bool useAsDefault, QString &errCause )
6430 : : {
6431 : 0 : QgsOgrLayerUniquePtr userLayer = LoadDataSourceAndLayer( uri, errCause );
6432 : 0 : if ( !userLayer )
6433 : 0 : return false;
6434 : :
6435 : 0 : QMutex *mutex = nullptr;
6436 : 0 : OGRLayerH hUserLayer = userLayer->getHandleAndMutex( mutex );
6437 : 0 : GDALDatasetH hDS = userLayer->getDatasetHandleAndMutex( mutex );
6438 : 0 : QMutexLocker locker( mutex );
6439 : :
6440 : : // check if layer_styles table already exist
6441 : 0 : OGRLayerH hLayer = GDALDatasetGetLayerByName( hDS, "layer_styles" );
6442 : 0 : if ( !hLayer )
6443 : : {
6444 : : // if not create it
6445 : : // Note: we use the same schema as in the SpatiaLite and postgres providers
6446 : : //for cross interoperability
6447 : :
6448 : 0 : char **options = nullptr;
6449 : : // TODO: might need change if other drivers than GPKG / SQLite
6450 : 0 : options = CSLSetNameValue( options, "FID", "id" );
6451 : 0 : hLayer = GDALDatasetCreateLayer( hDS, "layer_styles", nullptr, wkbNone, options );
6452 : 0 : QgsOgrProviderUtils::invalidateCachedDatasets( QString::fromUtf8( GDALGetDescription( hDS ) ) );
6453 : 0 : CSLDestroy( options );
6454 : 0 : if ( !hLayer )
6455 : : {
6456 : 0 : errCause = QObject::tr( "Unable to save layer style. It's not possible to create the destination table on the database." );
6457 : 0 : return false;
6458 : : }
6459 : 0 : bool ok = true;
6460 : : {
6461 : 0 : gdal::ogr_field_def_unique_ptr fld( OGR_Fld_Create( "f_table_catalog", OFTString ) );
6462 : 0 : OGR_Fld_SetWidth( fld.get(), 256 );
6463 : 0 : ok &= OGR_L_CreateField( hLayer, fld.get(), true ) == OGRERR_NONE;
6464 : 0 : }
6465 : : {
6466 : 0 : gdal::ogr_field_def_unique_ptr fld( OGR_Fld_Create( "f_table_schema", OFTString ) );
6467 : 0 : OGR_Fld_SetWidth( fld.get(), 256 );
6468 : 0 : ok &= OGR_L_CreateField( hLayer, fld.get(), true ) == OGRERR_NONE;
6469 : 0 : }
6470 : : {
6471 : 0 : gdal::ogr_field_def_unique_ptr fld( OGR_Fld_Create( "f_table_name", OFTString ) );
6472 : 0 : OGR_Fld_SetWidth( fld.get(), 256 );
6473 : 0 : ok &= OGR_L_CreateField( hLayer, fld.get(), true ) == OGRERR_NONE;
6474 : 0 : }
6475 : : {
6476 : 0 : gdal::ogr_field_def_unique_ptr fld( OGR_Fld_Create( "f_geometry_column", OFTString ) );
6477 : 0 : OGR_Fld_SetWidth( fld.get(), 256 );
6478 : 0 : ok &= OGR_L_CreateField( hLayer, fld.get(), true ) == OGRERR_NONE;
6479 : 0 : }
6480 : : {
6481 : 0 : gdal::ogr_field_def_unique_ptr fld( OGR_Fld_Create( "styleName", OFTString ) );
6482 : 0 : OGR_Fld_SetWidth( fld.get(), 30 );
6483 : 0 : ok &= OGR_L_CreateField( hLayer, fld.get(), true ) == OGRERR_NONE;
6484 : 0 : }
6485 : : {
6486 : 0 : gdal::ogr_field_def_unique_ptr fld( OGR_Fld_Create( "styleQML", OFTString ) );
6487 : 0 : ok &= OGR_L_CreateField( hLayer, fld.get(), true ) == OGRERR_NONE;
6488 : 0 : }
6489 : : {
6490 : 0 : gdal::ogr_field_def_unique_ptr fld( OGR_Fld_Create( "styleSLD", OFTString ) );
6491 : 0 : ok &= OGR_L_CreateField( hLayer, fld.get(), true ) == OGRERR_NONE;
6492 : 0 : }
6493 : : {
6494 : 0 : gdal::ogr_field_def_unique_ptr fld( OGR_Fld_Create( "useAsDefault", OFTInteger ) );
6495 : 0 : OGR_Fld_SetSubType( fld.get(), OFSTBoolean );
6496 : 0 : ok &= OGR_L_CreateField( hLayer, fld.get(), true ) == OGRERR_NONE;
6497 : 0 : }
6498 : : {
6499 : 0 : gdal::ogr_field_def_unique_ptr fld( OGR_Fld_Create( "description", OFTString ) );
6500 : 0 : ok &= OGR_L_CreateField( hLayer, fld.get(), true ) == OGRERR_NONE;
6501 : 0 : }
6502 : : {
6503 : 0 : gdal::ogr_field_def_unique_ptr fld( OGR_Fld_Create( "owner", OFTString ) );
6504 : 0 : OGR_Fld_SetWidth( fld.get(), 30 );
6505 : 0 : ok &= OGR_L_CreateField( hLayer, fld.get(), true ) == OGRERR_NONE;
6506 : 0 : }
6507 : : {
6508 : 0 : gdal::ogr_field_def_unique_ptr fld( OGR_Fld_Create( "ui", OFTString ) );
6509 : 0 : OGR_Fld_SetWidth( fld.get(), 30 );
6510 : 0 : ok &= OGR_L_CreateField( hLayer, fld.get(), true ) == OGRERR_NONE;
6511 : 0 : }
6512 : : {
6513 : 0 : gdal::ogr_field_def_unique_ptr fld( OGR_Fld_Create( "update_time", OFTDateTime ) );
6514 : 0 : OGR_Fld_SetDefault( fld.get(), "CURRENT_TIMESTAMP" );
6515 : 0 : ok &= OGR_L_CreateField( hLayer, fld.get(), true ) == OGRERR_NONE;
6516 : 0 : }
6517 : 0 : if ( !ok )
6518 : : {
6519 : 0 : errCause = QObject::tr( "Unable to save layer style. It's not possible to create the destination table on the database." );
6520 : 0 : return false;
6521 : : }
6522 : 0 : }
6523 : :
6524 : : QString realStyleName =
6525 : 0 : styleName.isEmpty() ? QString( OGR_L_GetName( hUserLayer ) ) : styleName;
6526 : :
6527 : 0 : OGRFeatureDefnH hLayerDefn = OGR_L_GetLayerDefn( hLayer );
6528 : :
6529 : 0 : if ( useAsDefault )
6530 : : {
6531 : 0 : QString oldDefaultQuery = QStringLiteral( "useAsDefault = 1 AND f_table_schema=''"
6532 : : " AND f_table_name=%1"
6533 : : " AND f_geometry_column=%2" )
6534 : 0 : .arg( QgsOgrProviderUtils::quotedValue( QString( OGR_L_GetName( hUserLayer ) ) ) )
6535 : 0 : .arg( QgsOgrProviderUtils::quotedValue( QString( OGR_L_GetGeometryColumn( hUserLayer ) ) ) );
6536 : 0 : OGR_L_SetAttributeFilter( hLayer, oldDefaultQuery.toUtf8().constData() );
6537 : 0 : gdal::ogr_feature_unique_ptr hFeature( OGR_L_GetNextFeature( hLayer ) );
6538 : 0 : if ( hFeature )
6539 : : {
6540 : 0 : OGR_F_SetFieldInteger( hFeature.get(),
6541 : 0 : OGR_FD_GetFieldIndex( hLayerDefn, "useAsDefault" ),
6542 : : 0 );
6543 : 0 : bool ok = OGR_L_SetFeature( hLayer, hFeature.get() ) == 0;
6544 : 0 : if ( !ok )
6545 : : {
6546 : 0 : QgsDebugMsg( QStringLiteral( "Could not unset previous useAsDefault style" ) );
6547 : 0 : }
6548 : 0 : }
6549 : 0 : }
6550 : :
6551 : 0 : QString checkQuery = QStringLiteral( "f_table_schema=''"
6552 : : " AND f_table_name=%1"
6553 : : " AND f_geometry_column=%2"
6554 : : " AND styleName=%3" )
6555 : 0 : .arg( QgsOgrProviderUtils::quotedValue( QString( OGR_L_GetName( hUserLayer ) ) ) )
6556 : 0 : .arg( QgsOgrProviderUtils::quotedValue( QString( OGR_L_GetGeometryColumn( hUserLayer ) ) ) )
6557 : 0 : .arg( QgsOgrProviderUtils::quotedValue( realStyleName ) );
6558 : 0 : OGR_L_SetAttributeFilter( hLayer, checkQuery.toUtf8().constData() );
6559 : 0 : OGR_L_ResetReading( hLayer );
6560 : 0 : gdal::ogr_feature_unique_ptr hFeature( OGR_L_GetNextFeature( hLayer ) );
6561 : 0 : bool bNew = true;
6562 : :
6563 : 0 : if ( hFeature )
6564 : : {
6565 : 0 : QgsSettings settings;
6566 : : // Only used in tests. Do not define it for interactive implication
6567 : 0 : QVariant overwriteStyle = settings.value( QStringLiteral( "qgis/overwriteStyle" ) );
6568 : 0 : if ( ( !overwriteStyle.isNull() && !overwriteStyle.toBool() ) ||
6569 : 0 : ( overwriteStyle.isNull() &&
6570 : 0 : QMessageBox::question( nullptr, QObject::tr( "Save style in database" ),
6571 : 0 : QObject::tr( "A style named \"%1\" already exists in the database for this layer. Do you want to overwrite it?" )
6572 : 0 : .arg( realStyleName ),
6573 : 0 : QMessageBox::Yes | QMessageBox::No ) == QMessageBox::No ) )
6574 : : {
6575 : 0 : errCause = QObject::tr( "Operation aborted" );
6576 : 0 : return false;
6577 : : }
6578 : 0 : bNew = false;
6579 : 0 : }
6580 : : else
6581 : : {
6582 : 0 : hFeature.reset( OGR_F_Create( hLayerDefn ) );
6583 : 0 : OGR_F_SetFieldString( hFeature.get(),
6584 : 0 : OGR_FD_GetFieldIndex( hLayerDefn, "f_table_catalog" ),
6585 : : "" );
6586 : 0 : OGR_F_SetFieldString( hFeature.get(),
6587 : 0 : OGR_FD_GetFieldIndex( hLayerDefn, "f_table_schema" ),
6588 : : "" );
6589 : 0 : OGR_F_SetFieldString( hFeature.get(),
6590 : 0 : OGR_FD_GetFieldIndex( hLayerDefn, "f_table_name" ),
6591 : 0 : OGR_L_GetName( hUserLayer ) );
6592 : 0 : OGR_F_SetFieldString( hFeature.get(),
6593 : 0 : OGR_FD_GetFieldIndex( hLayerDefn, "f_geometry_column" ),
6594 : 0 : OGR_L_GetGeometryColumn( hUserLayer ) );
6595 : 0 : OGR_F_SetFieldString( hFeature.get(),
6596 : 0 : OGR_FD_GetFieldIndex( hLayerDefn, "styleName" ),
6597 : 0 : realStyleName.toUtf8().constData() );
6598 : 0 : if ( !uiFileContent.isEmpty() )
6599 : : {
6600 : 0 : OGR_F_SetFieldString( hFeature.get(),
6601 : 0 : OGR_FD_GetFieldIndex( hLayerDefn, "ui" ),
6602 : 0 : uiFileContent.toUtf8().constData() );
6603 : 0 : }
6604 : : }
6605 : 0 : OGR_F_SetFieldString( hFeature.get(),
6606 : 0 : OGR_FD_GetFieldIndex( hLayerDefn, "styleQML" ),
6607 : 0 : qmlStyle.toUtf8().constData() );
6608 : 0 : OGR_F_SetFieldString( hFeature.get(),
6609 : 0 : OGR_FD_GetFieldIndex( hLayerDefn, "styleSLD" ),
6610 : 0 : sldStyle.toUtf8().constData() );
6611 : 0 : OGR_F_SetFieldInteger( hFeature.get(),
6612 : 0 : OGR_FD_GetFieldIndex( hLayerDefn, "useAsDefault" ),
6613 : 0 : useAsDefault ? 1 : 0 );
6614 : 0 : OGR_F_SetFieldString( hFeature.get(),
6615 : 0 : OGR_FD_GetFieldIndex( hLayerDefn, "description" ),
6616 : 0 : ( styleDescription.isEmpty() ? QDateTime::currentDateTime().toString() : styleDescription ).toUtf8().constData() );
6617 : 0 : OGR_F_SetFieldString( hFeature.get(),
6618 : 0 : OGR_FD_GetFieldIndex( hLayerDefn, "owner" ),
6619 : : "" );
6620 : :
6621 : : bool bFeatureOK;
6622 : 0 : if ( bNew )
6623 : 0 : bFeatureOK = OGR_L_CreateFeature( hLayer, hFeature.get() ) == OGRERR_NONE;
6624 : : else
6625 : 0 : bFeatureOK = OGR_L_SetFeature( hLayer, hFeature.get() ) == OGRERR_NONE;
6626 : :
6627 : 0 : if ( !bFeatureOK )
6628 : : {
6629 : 0 : QgsMessageLog::logMessage( QObject::tr( "Error updating style" ) );
6630 : 0 : errCause = QObject::tr( "Error looking for style. The query was logged" );
6631 : 0 : return false;
6632 : : }
6633 : :
6634 : 0 : return true;
6635 : 0 : }
6636 : :
6637 : :
6638 : 0 : bool QgsOgrProviderMetadata::deleteStyleById( const QString &uri, QString styleId, QString &errCause )
6639 : : {
6640 : 0 : QgsDataSourceUri dsUri( uri );
6641 : : bool deleted;
6642 : :
6643 : 0 : QgsOgrLayerUniquePtr userLayer = LoadDataSourceAndLayer( uri, errCause );
6644 : 0 : if ( !userLayer )
6645 : 0 : return false;
6646 : :
6647 : 0 : QMutex *mutex = nullptr;
6648 : 0 : GDALDatasetH hDS = userLayer->getDatasetHandleAndMutex( mutex );
6649 : 0 : QMutexLocker locker( mutex );
6650 : :
6651 : : // check if layer_styles table already exist
6652 : 0 : OGRLayerH hLayer = GDALDatasetGetLayerByName( hDS, "layer_styles" );
6653 : 0 : if ( !hLayer )
6654 : : {
6655 : 0 : errCause = QObject::tr( "Connection to database failed: %1" ).arg( dsUri.uri() );
6656 : 0 : deleted = false;
6657 : 0 : }
6658 : : else
6659 : : {
6660 : 0 : if ( OGR_L_DeleteFeature( hLayer, styleId.toInt() ) != OGRERR_NONE )
6661 : : {
6662 : 0 : errCause = QObject::tr( "Error executing the delete query." );
6663 : 0 : deleted = false;
6664 : 0 : }
6665 : : else
6666 : : {
6667 : 0 : deleted = true;
6668 : : }
6669 : : }
6670 : 0 : return deleted;
6671 : 0 : }
6672 : :
6673 : : static
6674 : 1 : bool LoadDataSourceLayerStylesAndLayer( const QString &uri,
6675 : : QgsOgrLayerUniquePtr &layerStyles,
6676 : : QgsOgrLayerUniquePtr &userLayer,
6677 : : QString &errCause )
6678 : : {
6679 : : bool isSubLayer;
6680 : : int layerIndex;
6681 : 1 : QString layerName;
6682 : 1 : QString subsetString;
6683 : : OGRwkbGeometryType ogrGeometryType;
6684 : 1 : QStringList openOptions;
6685 : 1 : QString filePath = AnalyzeURI( uri,
6686 : : isSubLayer,
6687 : : layerIndex,
6688 : : layerName,
6689 : : subsetString,
6690 : : ogrGeometryType,
6691 : : openOptions );
6692 : :
6693 : 1 : layerStyles =
6694 : 1 : QgsOgrProviderUtils::getLayer( filePath, "layer_styles", errCause );
6695 : 1 : userLayer = nullptr;
6696 : 1 : if ( !layerStyles )
6697 : : {
6698 : 1 : errCause = QObject::tr( "Cannot find layer_styles layer" );
6699 : 1 : return false;
6700 : : }
6701 : :
6702 : 0 : if ( !layerName.isEmpty() )
6703 : : {
6704 : 0 : userLayer = QgsOgrProviderUtils::getLayer( filePath, layerName, errCause );
6705 : 0 : }
6706 : : else
6707 : : {
6708 : 0 : userLayer = QgsOgrProviderUtils::getLayer( filePath, layerIndex, errCause );
6709 : : }
6710 : 0 : if ( !userLayer )
6711 : : {
6712 : 0 : layerStyles.reset();
6713 : 0 : return false;
6714 : : }
6715 : 0 : return true;
6716 : 1 : }
6717 : :
6718 : :
6719 : 1 : QString QgsOgrProviderMetadata::loadStyle( const QString &uri, QString &errCause )
6720 : : {
6721 : 1 : QgsOgrLayerUniquePtr layerStyles;
6722 : 1 : QgsOgrLayerUniquePtr userLayer;
6723 : 1 : if ( !LoadDataSourceLayerStylesAndLayer( uri, layerStyles, userLayer, errCause ) )
6724 : : {
6725 : 1 : return QString();
6726 : : }
6727 : :
6728 : 0 : QMutex *mutex1 = nullptr;
6729 : 0 : OGRLayerH hLayer = layerStyles->getHandleAndMutex( mutex1 );
6730 : 0 : QMutex *mutex2 = nullptr;
6731 : 0 : OGRLayerH hUserLayer = userLayer->getHandleAndMutex( mutex2 );
6732 : 0 : QMutexLocker lock1( mutex1 );
6733 : 0 : QMutexLocker lock2( mutex2 );
6734 : :
6735 : 0 : QString selectQmlQuery = QStringLiteral( "f_table_schema=''"
6736 : : " AND f_table_name=%1"
6737 : : " AND f_geometry_column=%2"
6738 : : " ORDER BY CASE WHEN useAsDefault THEN 1 ELSE 2 END"
6739 : : ",update_time DESC LIMIT 1" )
6740 : 0 : .arg( QgsOgrProviderUtils::quotedValue( QString( OGR_L_GetName( hUserLayer ) ) ) )
6741 : 0 : .arg( QgsOgrProviderUtils::quotedValue( QString( OGR_L_GetGeometryColumn( hUserLayer ) ) ) );
6742 : 0 : OGR_L_SetAttributeFilter( hLayer, selectQmlQuery.toUtf8().constData() );
6743 : 0 : OGR_L_ResetReading( hLayer );
6744 : 0 : OGRFeatureDefnH hLayerDefn = OGR_L_GetLayerDefn( hLayer );
6745 : 0 : QString styleQML;
6746 : 0 : qlonglong moreRecentTimestamp = 0;
6747 : 0 : while ( true )
6748 : : {
6749 : 0 : gdal::ogr_feature_unique_ptr hFeat( OGR_L_GetNextFeature( hLayer ) );
6750 : 0 : if ( !hFeat )
6751 : 0 : break;
6752 : 0 : if ( OGR_F_GetFieldAsInteger( hFeat.get(), OGR_FD_GetFieldIndex( hLayerDefn, "useAsDefault" ) ) )
6753 : : {
6754 : 0 : styleQML = QString::fromUtf8(
6755 : 0 : OGR_F_GetFieldAsString( hFeat.get(), OGR_FD_GetFieldIndex( hLayerDefn, "styleQML" ) ) );
6756 : 0 : break;
6757 : : }
6758 : :
6759 : : int year, month, day, hour, minute, second, TZ;
6760 : 0 : OGR_F_GetFieldAsDateTime( hFeat.get(), OGR_FD_GetFieldIndex( hLayerDefn, "update_time" ),
6761 : : &year, &month, &day, &hour, &minute, &second, &TZ );
6762 : 0 : qlonglong ts = second + minute * 60 + hour * 3600 + day * 24 * 3600 +
6763 : 0 : static_cast<qlonglong>( month ) * 31 * 24 * 3600 + static_cast<qlonglong>( year ) * 12 * 31 * 24 * 3600;
6764 : 0 : if ( ts > moreRecentTimestamp )
6765 : : {
6766 : 0 : moreRecentTimestamp = ts;
6767 : 0 : styleQML = QString::fromUtf8(
6768 : 0 : OGR_F_GetFieldAsString( hFeat.get(), OGR_FD_GetFieldIndex( hLayerDefn, "styleQML" ) ) );
6769 : :
6770 : 0 : }
6771 : 0 : }
6772 : :
6773 : 0 : return styleQML;
6774 : 1 : }
6775 : :
6776 : 0 : int QgsOgrProviderMetadata::listStyles(
6777 : : const QString &uri, QStringList &ids, QStringList &names,
6778 : : QStringList &descriptions, QString &errCause )
6779 : : {
6780 : : bool isSubLayer;
6781 : : int layerIndex;
6782 : 0 : QString layerName;
6783 : 0 : QString subsetString;
6784 : : OGRwkbGeometryType ogrGeometryType;
6785 : 0 : QStringList openOptions;
6786 : 0 : QString filePath = AnalyzeURI( uri,
6787 : : isSubLayer,
6788 : : layerIndex,
6789 : : layerName,
6790 : : subsetString,
6791 : : ogrGeometryType,
6792 : : openOptions );
6793 : :
6794 : 0 : QgsOgrLayerUniquePtr userLayer;
6795 : 0 : if ( !layerName.isEmpty() )
6796 : : {
6797 : 0 : userLayer = QgsOgrProviderUtils::getLayer( filePath, layerName, errCause );
6798 : 0 : }
6799 : : else
6800 : : {
6801 : 0 : userLayer = QgsOgrProviderUtils::getLayer( filePath, layerIndex, errCause );
6802 : : }
6803 : 0 : if ( !userLayer )
6804 : : {
6805 : 0 : return -1;
6806 : : }
6807 : :
6808 : : QgsOgrLayerUniquePtr layerStyles =
6809 : 0 : QgsOgrProviderUtils::getLayer( filePath, "layer_styles", errCause );
6810 : 0 : if ( !layerStyles )
6811 : : {
6812 : 0 : QgsMessageLog::logMessage( QObject::tr( "No styles available on DB" ) );
6813 : 0 : errCause = QObject::tr( "No styles available on DB" );
6814 : 0 : return 0;
6815 : : }
6816 : :
6817 : 0 : QMutex *mutex1 = nullptr;
6818 : 0 : OGRLayerH hLayer = layerStyles->getHandleAndMutex( mutex1 );
6819 : 0 : QMutexLocker lock1( mutex1 );
6820 : 0 : QMutex *mutex2 = nullptr;
6821 : 0 : OGRLayerH hUserLayer = userLayer->getHandleAndMutex( mutex2 );
6822 : 0 : QMutexLocker lock2( mutex2 );
6823 : :
6824 : 0 : if ( OGR_L_GetFeatureCount( hLayer, TRUE ) == 0 )
6825 : : {
6826 : 0 : QgsMessageLog::logMessage( QObject::tr( "No styles available on DB" ) );
6827 : 0 : errCause = QObject::tr( "No styles available on DB" );
6828 : 0 : return 0;
6829 : : }
6830 : :
6831 : 0 : OGRFeatureDefnH hLayerDefn = OGR_L_GetLayerDefn( hLayer );
6832 : :
6833 : 0 : OGR_L_ResetReading( hLayer );
6834 : :
6835 : 0 : QList<qlonglong> listTimestamp;
6836 : 0 : QMap<int, QString> mapIdToStyleName;
6837 : 0 : QMap<int, QString> mapIdToDescription;
6838 : 0 : QMap<qlonglong, QList<int> > mapTimestampToId;
6839 : 0 : int numberOfRelatedStyles = 0;
6840 : 0 : while ( true )
6841 : : {
6842 : 0 : gdal::ogr_feature_unique_ptr hFeature( OGR_L_GetNextFeature( hLayer ) );
6843 : 0 : if ( !hFeature )
6844 : 0 : break;
6845 : :
6846 : 0 : QString tableName( QString::fromUtf8(
6847 : 0 : OGR_F_GetFieldAsString( hFeature.get(),
6848 : 0 : OGR_FD_GetFieldIndex( hLayerDefn, "f_table_name" ) ) ) );
6849 : 0 : QString geometryColumn( QString::fromUtf8(
6850 : 0 : OGR_F_GetFieldAsString( hFeature.get(),
6851 : 0 : OGR_FD_GetFieldIndex( hLayerDefn, "f_geometry_column" ) ) ) );
6852 : 0 : QString styleName( QString::fromUtf8(
6853 : 0 : OGR_F_GetFieldAsString( hFeature.get(),
6854 : 0 : OGR_FD_GetFieldIndex( hLayerDefn, "styleName" ) ) ) );
6855 : 0 : QString description( QString::fromUtf8(
6856 : 0 : OGR_F_GetFieldAsString( hFeature.get(),
6857 : 0 : OGR_FD_GetFieldIndex( hLayerDefn, "description" ) ) ) );
6858 : 0 : int fid = static_cast<int>( OGR_F_GetFID( hFeature.get() ) );
6859 : 0 : if ( tableName == QString::fromUtf8( OGR_L_GetName( hUserLayer ) ) &&
6860 : 0 : geometryColumn == QString::fromUtf8( OGR_L_GetGeometryColumn( hUserLayer ) ) )
6861 : : {
6862 : : // Append first all related styles
6863 : 0 : QString id( QString::number( fid ) );
6864 : 0 : ids.append( id );
6865 : 0 : names.append( styleName );
6866 : 0 : descriptions.append( description );
6867 : 0 : ++ numberOfRelatedStyles;
6868 : 0 : }
6869 : : else
6870 : : {
6871 : : int year, month, day, hour, minute, second, TZ;
6872 : 0 : OGR_F_GetFieldAsDateTime( hFeature.get(), OGR_FD_GetFieldIndex( hLayerDefn, "update_time" ),
6873 : : &year, &month, &day, &hour, &minute, &second, &TZ );
6874 : 0 : qlonglong ts = second + minute * 60 + hour * 3600 + day * 24 * 3600 +
6875 : 0 : static_cast<qlonglong>( month ) * 31 * 24 * 3600 + static_cast<qlonglong>( year ) * 12 * 31 * 24 * 3600;
6876 : :
6877 : 0 : listTimestamp.append( ts );
6878 : 0 : mapIdToStyleName[fid] = styleName;
6879 : 0 : mapIdToDescription[fid] = description;
6880 : 0 : mapTimestampToId[ts].append( fid );
6881 : : }
6882 : 0 : }
6883 : :
6884 : 0 : std::sort( listTimestamp.begin(), listTimestamp.end() );
6885 : : // Sort from most recent to least recent
6886 : 0 : for ( int i = listTimestamp.size() - 1; i >= 0; i-- )
6887 : : {
6888 : 0 : const QList<int> &listId = mapTimestampToId[listTimestamp[i]];
6889 : 0 : for ( int j = 0; j < listId.size(); j++ )
6890 : : {
6891 : 0 : int fid = listId[j];
6892 : 0 : QString id( QString::number( fid ) );
6893 : 0 : ids.append( id );
6894 : 0 : names.append( mapIdToStyleName[fid] );
6895 : 0 : descriptions.append( mapIdToDescription[fid] );
6896 : 0 : }
6897 : 0 : }
6898 : :
6899 : 0 : return numberOfRelatedStyles;
6900 : 0 : }
6901 : :
6902 : 0 : QString QgsOgrProviderMetadata::getStyleById( const QString &uri, QString styleId, QString &errCause )
6903 : : {
6904 : 0 : QgsOgrLayerUniquePtr layerStyles;
6905 : 0 : QgsOgrLayerUniquePtr userLayer;
6906 : 0 : if ( !LoadDataSourceLayerStylesAndLayer( uri, layerStyles, userLayer, errCause ) )
6907 : : {
6908 : 0 : return QString();
6909 : : }
6910 : :
6911 : 0 : QMutex *mutex1 = nullptr;
6912 : 0 : OGRLayerH hLayer = layerStyles->getHandleAndMutex( mutex1 );
6913 : 0 : QMutexLocker lock1( mutex1 );
6914 : :
6915 : : bool ok;
6916 : 0 : int id = styleId.toInt( &ok );
6917 : 0 : if ( !ok )
6918 : : {
6919 : 0 : errCause = QObject::tr( "Invalid style identifier" );
6920 : 0 : return QString();
6921 : : }
6922 : :
6923 : 0 : gdal::ogr_feature_unique_ptr hFeature( OGR_L_GetFeature( hLayer, id ) );
6924 : 0 : if ( !hFeature )
6925 : : {
6926 : 0 : errCause = QObject::tr( "No style corresponding to style identifier" );
6927 : 0 : return QString();
6928 : : }
6929 : :
6930 : 0 : OGRFeatureDefnH hLayerDefn = OGR_L_GetLayerDefn( hLayer );
6931 : 0 : QString styleQML( QString::fromUtf8(
6932 : 0 : OGR_F_GetFieldAsString( hFeature.get(),
6933 : 0 : OGR_FD_GetFieldIndex( hLayerDefn, "styleQML" ) ) ) );
6934 : :
6935 : 0 : return styleQML;
6936 : 0 : }
6937 : :
6938 : :
6939 : : // ---------------------------------------------------------------------------
6940 : :
6941 : 0 : bool QgsOgrProviderUtils::deleteLayer( const QString &uri, QString &errCause )
6942 : : {
6943 : : bool isSubLayer;
6944 : : int layerIndex;
6945 : 0 : QString layerName;
6946 : 0 : QString subsetString;
6947 : : OGRwkbGeometryType ogrGeometryType;
6948 : 0 : QStringList openOptions;
6949 : 0 : QString filePath = AnalyzeURI( uri,
6950 : : isSubLayer,
6951 : : layerIndex,
6952 : : layerName,
6953 : : subsetString,
6954 : : ogrGeometryType,
6955 : : openOptions );
6956 : :
6957 : :
6958 : 0 : GDALDatasetH hDS = GDALOpenEx( filePath.toUtf8().constData(), GDAL_OF_RASTER | GDAL_OF_VECTOR | GDAL_OF_UPDATE, nullptr, nullptr, nullptr );
6959 : 0 : if ( hDS && ( ! layerName.isEmpty() || layerIndex != -1 ) )
6960 : : {
6961 : : // If we have got a name we convert it into an index
6962 : 0 : if ( ! layerName.isEmpty() )
6963 : : {
6964 : 0 : layerIndex = -1;
6965 : 0 : for ( int i = 0; i < GDALDatasetGetLayerCount( hDS ); i++ )
6966 : : {
6967 : 0 : OGRLayerH hL = GDALDatasetGetLayer( hDS, i );
6968 : 0 : if ( layerName == QString::fromUtf8( OGR_L_GetName( hL ) ) )
6969 : : {
6970 : 0 : layerIndex = i;
6971 : 0 : break;
6972 : : }
6973 : 0 : }
6974 : 0 : }
6975 : : // Do delete!
6976 : 0 : if ( layerIndex != -1 )
6977 : : {
6978 : 0 : OGRErr error = GDALDatasetDeleteLayer( hDS, layerIndex );
6979 : 0 : switch ( error )
6980 : : {
6981 : : case OGRERR_NOT_ENOUGH_DATA:
6982 : 0 : errCause = QObject::tr( "Not enough data to deserialize" );
6983 : 0 : break;
6984 : : case OGRERR_NOT_ENOUGH_MEMORY:
6985 : 0 : errCause = QObject::tr( "Not enough memory" );
6986 : 0 : break;
6987 : : case OGRERR_UNSUPPORTED_GEOMETRY_TYPE:
6988 : 0 : errCause = QObject::tr( "Unsupported geometry type" );
6989 : 0 : break;
6990 : : case OGRERR_UNSUPPORTED_OPERATION:
6991 : 0 : errCause = QObject::tr( "Unsupported operation" );
6992 : 0 : break;
6993 : : case OGRERR_CORRUPT_DATA:
6994 : 0 : errCause = QObject::tr( "Corrupt data" );
6995 : 0 : break;
6996 : : case OGRERR_FAILURE:
6997 : 0 : errCause = QObject::tr( "Failure" );
6998 : 0 : break;
6999 : : case OGRERR_UNSUPPORTED_SRS:
7000 : 0 : errCause = QObject::tr( "Unsupported SRS" );
7001 : 0 : break;
7002 : : case OGRERR_INVALID_HANDLE:
7003 : 0 : errCause = QObject::tr( "Invalid handle" );
7004 : 0 : break;
7005 : : case OGRERR_NON_EXISTING_FEATURE:
7006 : 0 : errCause = QObject::tr( "Non existing feature" );
7007 : 0 : break;
7008 : : default:
7009 : : case OGRERR_NONE:
7010 : 0 : errCause = QObject::tr( "Success" );
7011 : 0 : break;
7012 : : }
7013 : 0 : errCause = QObject::tr( "GDAL result code: %1" ).arg( errCause );
7014 : 0 : return error == OGRERR_NONE;
7015 : : }
7016 : 0 : }
7017 : : // This should never happen:
7018 : 0 : errCause = QObject::tr( "Layer not found: %1" ).arg( uri );
7019 : 0 : return false;
7020 : 0 : }
7021 : :
7022 : 247 : void QgsOgrLayerReleaser::operator()( QgsOgrLayer *layer )
7023 : : {
7024 : 247 : QgsOgrProviderUtils::release( layer );
7025 : 247 : }
7026 : :
7027 : 0 : QgsTransaction *QgsOgrProviderMetadata::createTransaction( const QString &connString )
7028 : : {
7029 : 0 : auto ds = QgsOgrProviderUtils::getAlreadyOpenedDataset( connString );
7030 : 0 : if ( !ds )
7031 : : {
7032 : 0 : QgsMessageLog::logMessage( QObject::tr( "Cannot open transaction on %1, since it is is not currently opened" ).arg( connString ),
7033 : 0 : QObject::tr( "OGR" ), Qgis::Critical );
7034 : 0 : return nullptr;
7035 : : }
7036 : :
7037 : 0 : return new QgsOgrTransaction( connString, ds );
7038 : 0 : }
7039 : :
7040 : : QgsGeoPackageProjectStorage *gGeoPackageProjectStorage = nullptr; // when not null it is owned by QgsApplication::projectStorageRegistry()
7041 : :
7042 : 3 : void QgsOgrProviderMetadata::initProvider()
7043 : : {
7044 : : Q_ASSERT( !gGeoPackageProjectStorage );
7045 : 3 : gGeoPackageProjectStorage = new QgsGeoPackageProjectStorage;
7046 : 3 : QgsApplication::projectStorageRegistry()->registerProjectStorage( gGeoPackageProjectStorage ); // takes ownership
7047 : 3 : }
7048 : :
7049 : :
7050 : 3 : void QgsOgrProviderMetadata::cleanupProvider()
7051 : : {
7052 : 3 : QgsApplication::projectStorageRegistry()->unregisterProjectStorage( gGeoPackageProjectStorage ); // destroys the object
7053 : 3 : gGeoPackageProjectStorage = nullptr;
7054 : 3 : QgsOgrConnPool::cleanupInstance();
7055 : : // NOTE: QgsApplication takes care of
7056 : : // calling OGRCleanupAll();
7057 : 3 : }
7058 : :
7059 : :
7060 : :
7061 : 335 : QgsOgrProviderMetadata::QgsOgrProviderMetadata()
7062 : 1005 : : QgsProviderMetadata( TEXT_PROVIDER_KEY, TEXT_PROVIDER_DESCRIPTION )
7063 : 335 : {
7064 : :
7065 : 335 : }
7066 : :
7067 : 15 : QString QgsOgrProviderMetadata::filters( FilterType type )
7068 : : {
7069 : 15 : switch ( type )
7070 : : {
7071 : : case QgsProviderMetadata::FilterType::FilterVector:
7072 : 3 : return QgsOgrProviderUtils::fileVectorFilters();
7073 : :
7074 : : case QgsProviderMetadata::FilterType::FilterRaster:
7075 : : case QgsProviderMetadata::FilterType::FilterMesh:
7076 : : case QgsProviderMetadata::FilterType::FilterMeshDataset:
7077 : : case QgsProviderMetadata::FilterType::FilterPointCloud:
7078 : 12 : return QString();
7079 : : }
7080 : 0 : return QString();
7081 : 15 : }
7082 : :
7083 : 0 : bool QgsOgrProviderMetadata::uriIsBlocklisted( const QString &uri ) const
7084 : : {
7085 : 0 : const QVariantMap parts = decodeUri( uri );
7086 : 0 : if ( !parts.contains( QStringLiteral( "path" ) ) )
7087 : 0 : return false;
7088 : :
7089 : 0 : QFileInfo fi( parts.value( QStringLiteral( "path" ) ).toString() );
7090 : 0 : const QString suffix = fi.completeSuffix();
7091 : :
7092 : : // internal details only
7093 : 0 : if ( suffix.compare( QLatin1String( "shp.xml" ), Qt::CaseInsensitive ) == 0 )
7094 : 0 : return true;
7095 : :
7096 : 0 : return false;
7097 : 0 : }
7098 : :
7099 : 0 : QMap<QString, QgsAbstractProviderConnection *> QgsOgrProviderMetadata::connections( bool cached )
7100 : : {
7101 : 0 : return connectionsProtected<QgsGeoPackageProviderConnection, QgsOgrDbConnection>( cached );
7102 : : }
7103 : :
7104 : 0 : QgsAbstractProviderConnection *QgsOgrProviderMetadata::createConnection( const QString &connName )
7105 : : {
7106 : 0 : return new QgsGeoPackageProviderConnection( connName );
7107 : 0 : }
7108 : :
7109 : 0 : QgsAbstractProviderConnection *QgsOgrProviderMetadata::createConnection( const QString &uri, const QVariantMap &configuration )
7110 : : {
7111 : 0 : return new QgsGeoPackageProviderConnection( uri, configuration );
7112 : 0 : }
7113 : :
7114 : 0 : void QgsOgrProviderMetadata::deleteConnection( const QString &name )
7115 : : {
7116 : 0 : deleteConnectionProtected<QgsGeoPackageProviderConnection>( name );
7117 : 0 : }
7118 : :
7119 : 0 : void QgsOgrProviderMetadata::saveConnection( const QgsAbstractProviderConnection *conn, const QString &name )
7120 : : {
7121 : 0 : saveConnectionProtected( conn, name );
7122 : 0 : }
7123 : :
7124 : 0 : QgsProviderMetadata::ProviderCapabilities QgsOgrProviderMetadata::providerCapabilities() const
7125 : : {
7126 : 0 : return FileBasedUris;
7127 : : }
7128 : : ///@endcond
|
