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