Implement mechanism to generate survey items from geometries

Author: nirvn

xlsformconverter/XLSFormConverter.py

@@ -18,6 +18,8 @@
     QgsEditorWidgetSetup,
     QgsExpression,
     QgsFeature,
+    QgsFeatureRequest,
+    QgsFeatureSink,
     QgsField,
     QgsFieldConstraints,
     QgsFields,
@@ -33,6 +35,8 @@
     QgsRelationContext,
     QgsVectorFileWriter,
     QgsVectorLayer,
+    QgsVectorLayerUtils,
+    QgsWkbTypes,
 )
 from qgis.PyQt.QtCore import QMetaType, QObject, QSize, pyqtSignal
 
@@ -690,13 +694,13 @@ def detect_geometry(self, child_name=None):
                         " "
                     )[0]
                     if type_details == "geopoint" or type_details == "start-geopoint":
-                        geometry = Qgis.WkbType.Point
+                        geometry = Qgis.WkbType.MultiPoint
                         break
                     if type_details == "geotrace" or type_details == "start-geotrace":
-                        geometry = Qgis.WkbType.LineString
+                        geometry = Qgis.WkbType.MultiLineString
                         break
                     if type_details == "geoshape" or type_details == "start-geoshape":
-                        geometry = Qgis.WkbType.Polygon
+                        geometry = Qgis.WkbType.MultiPolygon
                         break
 
         return geometry
@@ -990,7 +994,12 @@ def process_project_write(self, document):
         ms.writeXml(mapcanvasNode, document)
 
     def convert(
-        self, output_directory, title=None, language=None, basemap="OpenStreetMap"
+        self,
+        output_directory,
+        title=None,
+        language=None,
+        basemap="OpenStreetMap",
+        geometries=None,
     ):
         if not self.survey_layer:
             return ""
@@ -1494,4 +1503,39 @@ def convert(
         )
         self.output_project.writeProject.connect(self.process_project_write)
         self.output_project.write(output_project_file)
+
+        if (
+            geometries
+            and current_layer[0].geometryType() != Qgis.GeometryType.Null
+            and current_layer[0].geometryType() != Qgis.GeometryType.Unknown
+        ):
+            if current_layer[0].geometryType() == QgsWkbTypes.geometryType(
+                geometries.wkbType()
+            ):
+                current_layer[0].startEditing()
+
+                request = QgsFeatureRequest()
+                request.setDestinationCrs(
+                    current_layer[0].crs(), self.output_project.transformContext()
+                )
+                geometries_iterator = geometries.getFeatures(request)
+                for feature in geometries_iterator:
+                    output_features = QgsVectorLayerUtils.makeFeatureCompatible(
+                        feature,
+                        current_layer[0],
+                        QgsFeatureSink.SinkFlag.RegeneratePrimaryKey,
+                    )
+                    if output_features:
+                        current_layer[0].addFeature(
+                            output_features[0], QgsFeatureSink.Flag.FastInsert
+                        )
+
+                current_layer[0].commitChanges()
+            else:
+                self.warning.emit(
+                    self.tr(
+                        "The geometries' type does not match the output survey layer, skipping"
+                    )
+                )
+
         return output_project_file

xlsformconverter/XLSFormConverterAlgorithms.py

@@ -1,9 +1,12 @@
 import os
 
 from qgis.core import (
+    QgsProcessing,
     QgsProcessingAlgorithm,
     QgsProcessingParameterBoolean,
+    QgsProcessingParameterDefinition,
     QgsProcessingParameterEnum,
+    QgsProcessingParameterFeatureSource,
     QgsProcessingParameterFile,
     QgsProcessingParameterFolderDestination,
     QgsProcessingParameterString,
@@ -31,6 +34,7 @@ class XLSFormConverterAlgorithm(QgsProcessingAlgorithm):
     LANGUAGE = "LANGUAGE"
     BASEMAP = "BASEMAP"
     UPLOAD_TO_QFIELDCLOUD = "UPLOAD_TO_QFIELDCLOUD"
+    GEOMETRIES = "GEOMETRIES"
     OUTPUT = "OUTPUT"
 
     def tr(self, string):
@@ -110,6 +114,17 @@ def initAlgorithm(self, config=None):
             )
         )
 
+        param = QgsProcessingParameterFeatureSource(
+            self.GEOMETRIES,
+            self.tr(
+                "Pre-fill project with features' geometries and matching attributes"
+            ),
+            types=[QgsProcessing.SourceType.VectorAnyGeometry],
+            optional=True,
+        )
+        param.setFlags(param.flags() | QgsProcessingParameterDefinition.Flag.Advanced)
+        self.addParameter(param)
+
         self.addParameter(
             QgsProcessingParameterFolderDestination(
                 self.OUTPUT,
@@ -121,6 +136,7 @@ def processAlgorithm(self, parameters, context, feedback):
         xlsform_file = self.parameterAsString(parameters, self.INPUT, context)
         title = self.parameterAsString(parameters, self.TITLE, context)
         language = self.parameterAsString(parameters, self.LANGUAGE, context)
+        geometries = self.parameterAsSource(parameters, self.GEOMETRIES, context)
 
         basemap = "OpenStreetMap"
         basemap_index = self.parameterAsEnum(parameters, self.BASEMAP, context)
@@ -141,7 +157,9 @@ def processAlgorithm(self, parameters, context, feedback):
         converter.warning.connect(lambda message: feedback.pushWarning(message))
         converter.error.connect(lambda message: feedback.reportError(message))
 
-        project_file = converter.convert(output_directory, title, language, basemap)
+        project_file = converter.convert(
+            output_directory, title, language, basemap, geometries
+        )
 
         if project_file and upload_to_qfieldcloud:
             for root, dirs, files in os.walk(output_directory):