Read layers from project, include only visible layers.

Read layers from project. Include only visible layers when creating GeoTIFF. Added destination crs.

Author: Mtk112

arho_feature_template/core/geotiff_creator.py

@@ -1,17 +1,18 @@
 import os
 
 from osgeo import gdal
-from qgis.core import QgsMapRendererParallelJob, QgsMapSettings, QgsMapSettingsUtils
+from qgis.core import (
+    QgsCoordinateReferenceSystem,
+    QgsMapRendererParallelJob,
+    QgsMapSettings,
+    QgsMapSettingsUtils,
+    QgsProject,
+)
 from qgis.PyQt.QtCore import QSize, Qt
 from qgis.PyQt.QtGui import QColor
 from qgis.PyQt.QtWidgets import QApplication, QFileDialog
 
 from arho_feature_template.project.layers.plan_layers import (
-    LandUseAreaLayer,
-    LandUsePointLayer,
-    LineLayer,
-    OtherAreaLayer,
-    OtherPointLayer,
     PlanLayer,
 )
 from arho_feature_template.utils.misc_utils import get_active_plan_id, iface
@@ -60,16 +61,13 @@ def create_geotiff(self, geotiff_path):
 
         # Rendering settings
         settings = QgsMapSettings()
-        settings.setLayers(
-            [
-                PlanLayer.get_from_project(),
-                LandUsePointLayer.get_from_project(),
-                OtherPointLayer.get_from_project(),
-                LineLayer.get_from_project(),
-                OtherAreaLayer.get_from_project(),
-                LandUseAreaLayer.get_from_project(),
-            ]
-        )
+        layer_tree = QgsProject.instance().layerTreeRoot()
+
+        # Filter only visible layers
+        layers = [layer for layer in layer_tree.layerOrder() if layer_tree.findLayer(layer.id()).isVisible()]
+
+        settings.setLayers(layers)
+        settings.setDestinationCrs(QgsCoordinateReferenceSystem("EPSG:3067"))
         settings.setBackgroundColor(QColor(255, 255, 255))
         width = int(buffered_bbox.width() / self.desired_pixel_size) * self.desired_pixel_size
         height = int(buffered_bbox.height() / self.desired_pixel_size) * self.desired_pixel_size