21 #include <QDirIterator>
23 #include <QRegularExpression>
27 QString QgsImportPhotosAlgorithm::name()
const
29 return QStringLiteral(
"importphotos" );
32 QString QgsImportPhotosAlgorithm::displayName()
const
34 return QObject::tr(
"Import geotagged photos" );
37 QStringList QgsImportPhotosAlgorithm::tags()
const
39 return QObject::tr(
"exif,metadata,gps,jpeg,jpg" ).split(
',' );
42 QString QgsImportPhotosAlgorithm::group()
const
44 return QObject::tr(
"Vector creation" );
47 QString QgsImportPhotosAlgorithm::groupId()
const
49 return QStringLiteral(
"vectorcreation" );
52 void QgsImportPhotosAlgorithm::initAlgorithm(
const QVariantMap & )
57 std::unique_ptr< QgsProcessingParameterFeatureSink > output = std::make_unique< QgsProcessingParameterFeatureSink >( QStringLiteral(
"OUTPUT" ), QObject::tr(
"Photos" ),
QgsProcessing::TypeVectorPoint, QVariant(),
true );
58 output->setCreateByDefault(
true );
59 addParameter( output.release() );
61 std::unique_ptr< QgsProcessingParameterFeatureSink > invalid = std::make_unique< QgsProcessingParameterFeatureSink >( QStringLiteral(
"INVALID" ), QObject::tr(
"Invalid photos table" ),
QgsProcessing::TypeVector, QVariant(),
true );
62 invalid->setCreateByDefault(
false );
63 addParameter( invalid.release() );
66 QString QgsImportPhotosAlgorithm::shortHelpString()
const
68 return QObject::tr(
"Creates a point layer corresponding to the geotagged locations from JPEG images from a source folder. Optionally the folder can be recursively scanned.\n\n"
69 "The point layer will contain a single PointZ feature per input file from which the geotags could be read. Any altitude information from the geotags will be used "
70 "to set the point's Z value.\n\n"
71 "Optionally, a table of unreadable or non-geotagged photos can also be created." );
74 QgsImportPhotosAlgorithm *QgsImportPhotosAlgorithm::createInstance()
const
76 return new QgsImportPhotosAlgorithm();
79 QVariant QgsImportPhotosAlgorithm::parseMetadataValue(
const QString &value )
81 const QRegularExpression numRx( QStringLiteral(
"^\\s*\\(\\s*([-\\.\\d]+)\\s*\\)\\s*$" ) );
82 const QRegularExpressionMatch numMatch = numRx.match( value );
83 if ( numMatch.hasMatch() )
85 return numMatch.captured( 1 ).toDouble();
90 bool QgsImportPhotosAlgorithm::extractGeoTagFromMetadata(
const QVariantMap &metadata,
QgsPointXY &tag )
93 if ( metadata.contains( QStringLiteral(
"EXIF_GPSLongitude" ) ) )
96 x = metadata.value( QStringLiteral(
"EXIF_GPSLongitude" ) ).toDouble( &ok );
100 #if QT_VERSION < QT_VERSION_CHECK(6, 0, 0)
101 if ( metadata.value( QStringLiteral(
"EXIF_GPSLongitudeRef" ) ).toString().rightRef( 1 ).compare( QLatin1String(
"W" ), Qt::CaseInsensitive ) == 0
102 || metadata.value( QStringLiteral(
"EXIF_GPSLongitudeRef" ) ).toDouble() < 0 )
104 if ( QStringView { metadata.value( QStringLiteral(
"EXIF_GPSLongitudeRef" ) ).toString() }.right( 1 ).compare( QLatin1String(
"W" ), Qt::CaseInsensitive ) == 0
105 || metadata.value( QStringLiteral(
"EXIF_GPSLongitudeRef" ) ).toDouble() < 0 )
117 if ( metadata.contains( QStringLiteral(
"EXIF_GPSLatitude" ) ) )
120 y = metadata.value( QStringLiteral(
"EXIF_GPSLatitude" ) ).toDouble( &ok );
124 #if QT_VERSION < QT_VERSION_CHECK(6, 0, 0)
125 if ( metadata.value( QStringLiteral(
"EXIF_GPSLatitudeRef" ) ).toString().rightRef( 1 ).compare( QLatin1String(
"S" ), Qt::CaseInsensitive ) == 0
126 || metadata.value( QStringLiteral(
"EXIF_GPSLatitudeRef" ) ).toDouble() < 0 )
128 if ( QStringView { metadata.value( QStringLiteral(
"EXIF_GPSLatitudeRef" ) ).toString() }.right( 1 ).compare( QLatin1String(
"S" ), Qt::CaseInsensitive ) == 0
129 || metadata.value( QStringLiteral(
"EXIF_GPSLatitudeRef" ) ).toDouble() < 0 )
144 QVariant QgsImportPhotosAlgorithm::extractAltitudeFromMetadata(
const QVariantMap &metadata )
147 if ( metadata.contains( QStringLiteral(
"EXIF_GPSAltitude" ) ) )
149 double alt = metadata.value( QStringLiteral(
"EXIF_GPSAltitude" ) ).toDouble();
150 if ( metadata.contains( QStringLiteral(
"EXIF_GPSAltitudeRef" ) ) &&
151 ( ( metadata.value( QStringLiteral(
"EXIF_GPSAltitudeRef" ) ).type() == QVariant::String && metadata.value( QStringLiteral(
"EXIF_GPSAltitudeRef" ) ).toString().right( 1 ) == QLatin1String(
"1" ) )
152 || metadata.value( QStringLiteral(
"EXIF_GPSAltitudeRef" ) ).toDouble() < 0 ) )
159 QVariant QgsImportPhotosAlgorithm::extractDirectionFromMetadata(
const QVariantMap &metadata )
162 if ( metadata.contains( QStringLiteral(
"EXIF_GPSImgDirection" ) ) )
164 direction = metadata.value( QStringLiteral(
"EXIF_GPSImgDirection" ) ).toDouble();
169 QVariant QgsImportPhotosAlgorithm::extractOrientationFromMetadata(
const QVariantMap &metadata )
171 QVariant orientation;
172 if ( metadata.contains( QStringLiteral(
"EXIF_Orientation" ) ) )
174 switch ( metadata.value( QStringLiteral(
"EXIF_Orientation" ) ).toInt() )
205 QVariant QgsImportPhotosAlgorithm::extractTimestampFromMetadata(
const QVariantMap &metadata )
208 if ( metadata.contains( QStringLiteral(
"EXIF_DateTimeOriginal" ) ) )
210 ts = metadata.value( QStringLiteral(
"EXIF_DateTimeOriginal" ) );
212 else if ( metadata.contains( QStringLiteral(
"EXIF_DateTimeDigitized" ) ) )
214 ts = metadata.value( QStringLiteral(
"EXIF_DateTimeDigitized" ) );
216 else if ( metadata.contains( QStringLiteral(
"EXIF_DateTime" ) ) )
218 ts = metadata.value( QStringLiteral(
"EXIF_DateTime" ) );
224 const QRegularExpression dsRegEx( QStringLiteral(
"(\\d+):(\\d+):(\\d+)\\s+(\\d+):(\\d+):(\\d+)" ) );
225 const QRegularExpressionMatch dsMatch = dsRegEx.match( ts.toString() );
226 if ( dsMatch.hasMatch() )
228 const int year = dsMatch.captured( 1 ).toInt();
229 const int month = dsMatch.captured( 2 ).toInt();
230 const int day = dsMatch.captured( 3 ).toInt();
231 const int hour = dsMatch.captured( 4 ).toInt();
232 const int min = dsMatch.captured( 5 ).toInt();
233 const int sec = dsMatch.captured( 6 ).toInt();
234 return QDateTime( QDate( year, month, day ), QTime( hour, min, sec ) );
242 QVariant QgsImportPhotosAlgorithm::parseCoord(
const QString &
string )
244 const QRegularExpression coordRx( QStringLiteral(
"^\\s*\\(\\s*([-\\.\\d]+)\\s*\\)\\s*\\(\\s*([-\\.\\d]+)\\s*\\)\\s*\\(\\s*([-\\.\\d]+)\\s*\\)\\s*$" ) );
245 const QRegularExpressionMatch coordMatch = coordRx.match(
string );
246 if ( coordMatch.hasMatch() )
248 const double hours = coordMatch.captured( 1 ).toDouble();
249 const double minutes = coordMatch.captured( 2 ).toDouble();
250 const double seconds = coordMatch.captured( 3 ).toDouble();
251 return hours + minutes / 60.0 + seconds / 3600.0;
259 QVariantMap QgsImportPhotosAlgorithm::parseMetadataList(
const QStringList &input )
262 const QRegularExpression splitRx( QStringLiteral(
"(.*?)=(.*)" ) );
263 for (
const QString &item : input )
265 const QRegularExpressionMatch match = splitRx.match( item );
266 if ( !match.hasMatch() )
269 const QString tag = match.captured( 1 );
270 QVariant value = parseMetadataValue( match.captured( 2 ) );
272 if ( tag == QLatin1String(
"EXIF_GPSLatitude" ) || tag == QLatin1String(
"EXIF_GPSLongitude" ) )
273 value = parseCoord( value.toString() );
274 results.insert( tag, value );
286 if (
QgsVectorLayer *vl = qobject_cast< QgsVectorLayer * >( layer ) )
290 config.insert( QStringLiteral(
"DocumentViewer" ), 1 );
291 config.insert( QStringLiteral(
"FileWidget" ),
true );
292 config.insert( QStringLiteral(
"UseLink" ),
true );
293 config.insert( QStringLiteral(
"FullUrl" ),
true );
294 vl->setEditorWidgetSetup( vl->fields().lookupField( QStringLiteral(
"photo" ) ),
QgsEditorWidgetSetup( QStringLiteral(
"ExternalResource" ), config ) );
298 config.insert( QStringLiteral(
"FileWidgetButton" ),
true );
299 config.insert( QStringLiteral(
"StorageMode" ), 1 );
300 config.insert( QStringLiteral(
"UseLink" ),
true );
301 config.insert( QStringLiteral(
"FullUrl" ),
true );
302 vl->setEditorWidgetSetup( vl->fields().lookupField( QStringLiteral(
"directory" ) ),
QgsEditorWidgetSetup( QStringLiteral(
"ExternalResource" ), config ) );
309 const QString folder = parameterAsFile( parameters, QStringLiteral(
"FOLDER" ), context );
311 const QDir importDir( folder );
312 if ( !importDir.exists() )
317 const bool recurse = parameterAsBoolean( parameters, QStringLiteral(
"RECURSIVE" ), context );
320 outFields.
append(
QgsField( QStringLiteral(
"photo" ), QVariant::String ) );
321 outFields.
append(
QgsField( QStringLiteral(
"filename" ), QVariant::String ) );
322 outFields.
append(
QgsField( QStringLiteral(
"directory" ), QVariant::String ) );
323 outFields.
append(
QgsField( QStringLiteral(
"altitude" ), QVariant::Double ) );
324 outFields.
append(
QgsField( QStringLiteral(
"direction" ), QVariant::Double ) );
325 outFields.
append(
QgsField( QStringLiteral(
"rotation" ), QVariant::Int ) );
326 outFields.
append(
QgsField( QStringLiteral(
"longitude" ), QVariant::String ) );
327 outFields.
append(
QgsField( QStringLiteral(
"latitude" ), QVariant::String ) );
328 outFields.
append(
QgsField( QStringLiteral(
"timestamp" ), QVariant::DateTime ) );
330 std::unique_ptr< QgsFeatureSink > outputSink( parameterAsSink( parameters, QStringLiteral(
"OUTPUT" ), context, outputDest, outFields,
334 invalidFields.
append(
QgsField( QStringLiteral(
"photo" ), QVariant::String ) );
335 invalidFields.
append(
QgsField( QStringLiteral(
"filename" ), QVariant::String ) );
336 invalidFields.
append(
QgsField( QStringLiteral(
"directory" ), QVariant::String ) );
337 invalidFields.
append(
QgsField( QStringLiteral(
"readable" ), QVariant::Bool ) );
339 std::unique_ptr< QgsFeatureSink > invalidSink( parameterAsSink( parameters, QStringLiteral(
"INVALID" ), context, invalidDest, invalidFields ) );
341 const QStringList nameFilters {
"*.jpeg",
"*.jpg" };
346 const QFileInfoList fileInfoList = importDir.entryInfoList( nameFilters, QDir::NoDotAndDotDot | QDir::Files );
347 for (
auto infoIt = fileInfoList.constBegin(); infoIt != fileInfoList.constEnd(); ++infoIt )
349 files.append( infoIt->absoluteFilePath() );
354 QDirIterator it( folder, nameFilters, QDir::NoDotAndDotDot | QDir::Files, QDirIterator::Subdirectories );
355 while ( it.hasNext() )
358 files.append( it.filePath() );
362 auto saveInvalidFile = [&invalidSink, ¶meters](
QgsAttributes & attributes,
bool readable )
367 attributes.append( readable );
370 throw QgsProcessingException( writeFeatureError( invalidSink.get(), parameters, QStringLiteral(
"INVALID" ) ) );
374 const double step = files.count() > 0 ? 100.0 / files.count() : 1;
376 for (
const QString &file : files )
386 const QFileInfo fi( file );
388 attributes << QDir::toNativeSeparators( file )
389 << fi.completeBaseName()
390 << QDir::toNativeSeparators( fi.absolutePath() );
395 feedback->
reportError( QObject::tr(
"Could not open %1" ).arg( QDir::toNativeSeparators( file ) ) );
396 saveInvalidFile( attributes,
false );
400 if (
char **GDALmetadata = GDALGetMetadata( hDS.get(),
nullptr ) )
409 if ( !extractGeoTagFromMetadata( metadata, tag ) )
412 feedback->
reportError( QObject::tr(
"Could not retrieve geotag for %1" ).arg( QDir::toNativeSeparators( file ) ) );
413 saveInvalidFile( attributes,
true );
417 const QVariant altitude = extractAltitudeFromMetadata( metadata );
423 << extractDirectionFromMetadata( metadata )
424 << extractOrientationFromMetadata( metadata )
427 << extractTimestampFromMetadata( metadata );
430 throw QgsProcessingException( writeFeatureError( outputSink.get(), parameters, QStringLiteral(
"OUTPUT" ) ) );
434 feedback->
reportError( QObject::tr(
"No metadata found in %1" ).arg( QDir::toNativeSeparators( file ) ) );
435 saveInvalidFile( attributes,
true );
442 outputs.insert( QStringLiteral(
"OUTPUT" ), outputDest );
451 outputs.insert( QStringLiteral(
"INVALID" ), invalidDest );