maps4fs 2.0.9__py3-none-any.whl → 2.1.0__py3-none-any.whl

maps4fs/generator/component/base/component_xml.py

@@ -4,6 +4,7 @@ import os
 from xml.etree import ElementTree as ET
 
 from maps4fs.generator.component.base.component import Component
+from maps4fs.generator.settings import Parameters
 
 
 class XMLComponent(Component):
@@ -57,6 +58,20 @@ class XMLComponent(Component):
 
         tree.write(xml_path, encoding="utf-8", xml_declaration=True)
 
+    def get_element_from_tree(self, path: str, xml_path: str | None = None) -> ET.Element | None:
+        """Finds an element in the XML tree by the path.
+
+        Arguments:
+            path (str): The path to the element.
+            xml_path (str, optional): The path to the XML file. Defaults to None.
+
+        Returns:
+            ET.Element | None: The found element or None if not found.
+        """
+        tree = self.get_tree(xml_path)
+        root = tree.getroot()
+        return root.find(path)  # type: ignore
+
     def get_and_update_element(self, root: ET.Element, path: str, data: dict[str, str]) -> None:
         """Finds the element by the path and updates it with the provided data.
 
@@ -106,3 +121,25 @@ class XMLComponent(Component):
         """
         element = ET.SubElement(parent, element_name)
         self.update_element(element, data)
+
+    def get_height_scale(self) -> int:
+        """Returns the height scale from the I3D file.
+
+        Returns:
+            int: The height scale value.
+
+        Raises:
+            ValueError: If the height scale element is not found in the I3D file.
+        """
+        height_scale_element = self.get_element_from_tree(
+            path=".//Scene/TerrainTransformGroup",
+            xml_path=self.game.i3d_file_path(self.map_directory),
+        )
+        if height_scale_element is None:
+            raise ValueError("Height scale element not found in the I3D file.")
+
+        height_scale = height_scale_element.get(Parameters.HEIGHT_SCALE)
+        if height_scale is None:
+            raise ValueError("Height scale not found in the I3D file.")
+
+        return int(height_scale)

maps4fs/generator/component/config.py

@@ -2,6 +2,10 @@
 
 from __future__ import annotations
 
+import os
+
+import cv2
+
 from maps4fs.generator.component.base.component_xml import XMLComponent
 
 
@@ -23,11 +27,15 @@ class Config(XMLComponent):
     def preprocess(self) -> None:
         """Gets the path to the map XML file and saves it to the instance variable."""
         self.xml_path = self.game.map_xml_path(self.map_directory)
+        self.fog_parameters: dict[str, int] = {}
 
     def process(self) -> None:
         """Sets the map size in the map.xml file."""
         self._set_map_size()
 
+        if self.game.fog_processing:
+            self._adjust_fog()
+
     def _set_map_size(self) -> None:
         """Edits map.xml file to set correct map size."""
         tree = self.get_tree()
@@ -77,6 +85,8 @@ class Config(XMLComponent):
         data = {
             "Overview": overview_data,
         }
+        if self.fog_parameters:
+            data["Fog"] = self.fog_parameters  # type: ignore
 
         return data  # type: ignore
 
@@ -92,3 +102,114 @@ class Config(XMLComponent):
         layers = qgis_layers + qgis_layers_with_margin
 
         self.create_qgis_scripts(layers)
+
+    def _adjust_fog(self) -> None:
+        """Adjusts the fog settings in the environment XML file based on the DEM and height scale."""
+        self.logger.debug("Adjusting fog settings based on DEM and height scale...")
+        try:
+            environment_xml_path = self.game.get_environment_xml_path(self.map_directory)
+        except NotImplementedError:
+            self.logger.warning(
+                "Game does not support environment XML file, fog adjustment will not be applied."
+            )
+            return
+
+        if not environment_xml_path or not os.path.isfile(environment_xml_path):
+            self.logger.warning(
+                "Environment XML file not found, fog adjustment will not be applied."
+            )
+            return
+
+        self.logger.debug("Will work with environment XML file: %s", environment_xml_path)
+
+        dem_params = self._get_dem_meter_params()
+        if not dem_params:
+            return
+        maximum_height, minimum_height = dem_params
+
+        tree = self.get_tree(xml_path=environment_xml_path)
+        root = tree.getroot()
+
+        # Find the <latitude>40.6</latitude> element in the XML file.
+        latitude_element = root.find("./latitude")  # type: ignore
+        if latitude_element is not None:
+            map_latitude = round(self.map.coordinates[0], 1)
+            latitude_element.text = str(map_latitude)
+            self.logger.debug(
+                "Found latitude element and set it to: %s",
+                latitude_element.text,
+            )
+
+        # The XML file contains 4 <fog> entries in different sections of <weather> representing
+        # different seasons, such as <season name="spring">, <season name="summer">, etc.
+        # We need to find them all and adjust the parameters accordingly.
+        for season in root.findall(".//weather/season"):  # type: ignore
+            # Example of the <heightFog> element:
+            # <heightFog>
+            #     <groundLevelDensity min="0.05" max="0.2" />
+            #     <maxHeight min="420" max="600" />
+            # </heightFog>
+            # We need to adjust the maxheight min and max attributes.
+            max_height_element = season.find("./fog/heightFog/maxHeight")
+            data = {
+                "min": str(minimum_height),
+                "max": str(maximum_height),
+            }
+            self.update_element(max_height_element, data)  # type: ignore
+            self.logger.debug(
+                "Adjusted fog settings for season '%s': min=%s, max=%s",
+                season.get("name", "unknown"),
+                minimum_height,
+                maximum_height,
+            )
+
+        self.logger.debug("Fog adjusted and file will be saved to %s", environment_xml_path)
+        self.save_tree(tree, xml_path=environment_xml_path)
+
+        self.fog_parameters = {
+            "minimum_height": minimum_height,
+            "maximum_height": maximum_height,
+        }
+
+    def _get_dem_meter_params(self) -> tuple[int, int] | None:
+        """Reads the DEM file and returns the maximum and minimum height in meters.
+
+        Returns:
+            tuple[int, int] | None: Maximum and minimum height in meters or None if the DEM file
+                is not found or cannot be read.
+        """
+        self.logger.debug("Reading DEM meter parameters...")
+        dem_path = self.game.dem_file_path(self.map_directory)
+        if not dem_path or not os.path.isfile(dem_path):
+            self.logger.warning("DEM file not found, fog adjustment will not be applied.")
+            return None
+
+        dem_image = cv2.imread(dem_path, cv2.IMREAD_UNCHANGED)
+        if dem_image is None:
+            self.logger.warning("Failed to read DEM image, fog adjustment will not be applied.")
+            return None
+        dem_maximum_pixel = dem_image.max()
+        dem_minimum_pixel = dem_image.min()
+
+        self.logger.debug(
+            "DEM read successfully. Max pixel: %d, Min pixel: %d",
+            dem_maximum_pixel,
+            dem_minimum_pixel,
+        )
+
+        try:
+            height_scale = self.get_height_scale()
+        except ValueError as e:
+            self.logger.warning("Error getting height scale from I3D file: %s", e)
+            return None
+        self.logger.debug("Height scale from I3D file: %d", height_scale)
+
+        dem_maximum_meter = int(dem_maximum_pixel / height_scale)
+        dem_minimum_meter = int(dem_minimum_pixel / height_scale)
+        self.logger.debug(
+            "DEM maximum height in meters: %d, minimum height in meters: %d",
+            dem_maximum_meter,
+            dem_minimum_meter,
+        )
+
+        return dem_maximum_meter, dem_minimum_meter

maps4fs/generator/component/i3d.py

@@ -83,7 +83,7 @@ class I3d(XMLComponent):
         root = tree.getroot()
         path = ".//Scene/TerrainTransformGroup"
 
-        data = {"heightScale": str(value)}
+        data = {Parameters.HEIGHT_SCALE: str(value)}
 
         self.get_and_update_element(root, path, data)  # type: ignore
         self.save_tree(tree)  # type: ignore

maps4fs/generator/game.py

@@ -40,6 +40,7 @@ class Game:
     _tree_schema: str | None = None
     _i3d_processing: bool = True
     _plants_processing: bool = True
+    _fog_processing: bool = True
     _dissolve: bool = True
 
     # Order matters! Some components depend on others.
@@ -189,6 +190,16 @@ class Game:
             str: The path to the farmlands xml file."""
         raise NotImplementedError
 
+    def get_environment_xml_path(self, map_directory: str) -> str:
+        """Returns the path to the environment xml file.
+
+        Arguments:
+            map_directory (str): The path to the map directory.
+
+        Returns:
+            str: The path to the environment xml file."""
+        raise NotImplementedError
+
     def i3d_file_path(self, map_directory: str) -> str:
         """Returns the path to the i3d file.
 
@@ -207,6 +218,14 @@ class Game:
             bool: True if the i3d file should be processed, False otherwise."""
         return self._i3d_processing
 
+    @property
+    def fog_processing(self) -> bool:
+        """Returns whether the fog should be processed.
+
+        Returns:
+            bool: True if the fog should be processed, False otherwise."""
+        return self._fog_processing
+
     @property
     def plants_processing(self) -> bool:
         """Returns whether the plants should be processed.
@@ -250,6 +269,7 @@ class FS22(Game):
     _map_template_path = os.path.join(working_directory, "data", "fs22-map-template.zip")
     _texture_schema = os.path.join(working_directory, "data", "fs22-texture-schema.json")
     _i3d_processing = False
+    _fog_processing = False
     _plants_processing = False
     _dissolve = False
 
@@ -345,3 +365,13 @@ class FS25(Game):
         Returns:
             str: The path to the farmlands xml file."""
         return os.path.join(map_directory, "map", "config", "farmlands.xml")
+
+    def get_environment_xml_path(self, map_directory: str) -> str:
+        """Returns the path to the environment xml file.
+
+        Arguments:
+            map_directory (str): The path to the map directory.
+
+        Returns:
+            str: The path to the environment xml file."""
+        return os.path.join(map_directory, "map", "config", "environment.xml")

maps4fs/generator/settings.py

@@ -40,6 +40,8 @@ class Parameters:
 
     WATER_ADD_WIDTH = 2
 
+    HEIGHT_SCALE = "heightScale"
+
 
 class SharedSettings(BaseModel):
     """Represents the shared settings for all components."""

{maps4fs-2.0.9.dist-info → maps4fs-2.1.0.dist-info}/METADATA

@@ -1,6 +1,6 @@
 Metadata-Version: 2.4
 Name: maps4fs
-Version: 2.0.9
+Version: 2.1.0
 Summary: Generate map templates for Farming Simulator from real places.
 Author-email: iwatkot <iwatkot@gmail.com>
 License: Apache License 2.0

{maps4fs-2.0.9.dist-info → maps4fs-2.1.0.dist-info}/RECORD

@@ -1,17 +1,17 @@
 maps4fs/__init__.py,sha256=Fy521EmVAWnhu6OvOInc97yrtJotFzcV0YfRB2b9O4s,314
 maps4fs/logger.py,sha256=WDfR14hxqy8b6xtwL6YIu2LGzFO1sbt0LxMgfsDTOkA,865
 maps4fs/generator/__init__.py,sha256=zZMLEkGzb4z0xql650gOtGSvcgX58DnJ2yN3vC2daRk,43
-maps4fs/generator/game.py,sha256=g8lMHuuRRmJLSDsQTAMv8p_-qntYMiZKnAqn7ru96i0,11645
+maps4fs/generator/game.py,sha256=Nl8xqEwW-E04l2D3e9ff7kkzPcolr1ndnXxQJlLMBDM,12627
 maps4fs/generator/map.py,sha256=x_NjUwKkDnh2cC61QpTfX88mFHAkQG1sxVwZckoV0mE,16357
 maps4fs/generator/qgis.py,sha256=Es8hLuqN_KH8lDfnJE6He2rWYbAKJ3RGPn-o87S6CPI,6116
-maps4fs/generator/settings.py,sha256=iEazY2goQlyYgJA6-jhCvDOh7EZ4bDzBDGocJ5K3cjc,7544
+maps4fs/generator/settings.py,sha256=FIE02u2gWwo1WC3rUIZmQziC994dhDH38HflLrXT_DM,7578
 maps4fs/generator/statistics.py,sha256=aynS3zbAtiwnU_YLKHPTiiaKW98_suvQUhy1SGBA6mc,2448
 maps4fs/generator/component/__init__.py,sha256=s01yVVVi8R2xxNvflu2D6wTd9I_g73AMM2x7vAC7GX4,490
 maps4fs/generator/component/background.py,sha256=wqTgndTfRIGlLMpMXqJMw_6eVdH1JtMei46ZIXlT9X4,29480
-maps4fs/generator/component/config.py,sha256=IP530sapLofFskSnBEB96G0aUSd6Sno0G9ET3ca0ZOQ,3696
+maps4fs/generator/component/config.py,sha256=uL76h9UwyhZKZmbxz0mBmWtEPN6qYay4epTEqqtej60,8601
 maps4fs/generator/component/dem.py,sha256=SH_2Zu5O4dhWtZeOkCwzDF4RU04XhTdpGFYaRYJkdjc,11905
 maps4fs/generator/component/grle.py,sha256=8K32pC_ar9CR6p0EhCe2X--wEoIxFzJCPcN9ydHQ1LE,19747
-maps4fs/generator/component/i3d.py,sha256=iGBDaCPC-EqoJ7C4PrKSel6mW4U1DPfgSp6pPNR-Xr8,23788
+maps4fs/generator/component/i3d.py,sha256=B0-YTeiIkon2v7OLFNMEw38ouuuu4G54WbYIQm6I5a8,23798
 maps4fs/generator/component/layer.py,sha256=-br4gAGcGeBNb3ldch9XFEK0lhXqb1IbArhFB5Owu54,6186
 maps4fs/generator/component/satellite.py,sha256=OsxoNOCgkUtRzL7Geuqubsf6uoKXAIN8jQvrJ7IFeAI,4958
 maps4fs/generator/component/texture.py,sha256=krtvOS0hH8BTzfxd2jsTo3bIJYRkIVbaz6FGhWY8L1o,33921
@@ -19,9 +19,9 @@ maps4fs/generator/component/base/__init__.py,sha256=zZMLEkGzb4z0xql650gOtGSvcgX5
 maps4fs/generator/component/base/component.py,sha256=Vgmdsn1ZC37EwWi4Va4uYVt0RnFLiARTtZ-R5GTSrrM,22877
 maps4fs/generator/component/base/component_image.py,sha256=2NYJgCU8deHl7O2FYFYk38WKZVJygFoc2gjBXwH6vjM,5970
 maps4fs/generator/component/base/component_mesh.py,sha256=lZuF-9hzDYodL4Ub4RgvFjsozxTPZJrOaLQmlNTrZ7I,9074
-maps4fs/generator/component/base/component_xml.py,sha256=V9pGUvHh6UF6BP0qFARqDq9vquoAgq1zJqhOgBoeS_Y,3983
-maps4fs-2.0.9.dist-info/licenses/LICENSE.md,sha256=pTKD_oUexcn-yccFCTrMeLkZy0ifLRa-VNcDLqLZaIw,10749
-maps4fs-2.0.9.dist-info/METADATA,sha256=JhBYuvnO_3NuZlvzWafXt94qffCQPdO5GXotONW69PE,44135
-maps4fs-2.0.9.dist-info/WHEEL,sha256=_zCd3N1l69ArxyTb8rzEoP9TpbYXkqRFSNOD5OuxnTs,91
-maps4fs-2.0.9.dist-info/top_level.txt,sha256=Ue9DSRlejRQRCaJueB0uLcKrWwsEq9zezfv5dI5mV1M,8
-maps4fs-2.0.9.dist-info/RECORD,,
+maps4fs/generator/component/base/component_xml.py,sha256=MT-VhU2dEckLFxAgmxg6V3gnv11di_94Qq6atfpOLdc,5342
+maps4fs-2.1.0.dist-info/licenses/LICENSE.md,sha256=pTKD_oUexcn-yccFCTrMeLkZy0ifLRa-VNcDLqLZaIw,10749
+maps4fs-2.1.0.dist-info/METADATA,sha256=-_6ei79gfvcNStmIBB3iu8zlf4Y39EOZWp1pvvIQdwI,44135
+maps4fs-2.1.0.dist-info/WHEEL,sha256=_zCd3N1l69ArxyTb8rzEoP9TpbYXkqRFSNOD5OuxnTs,91
+maps4fs-2.1.0.dist-info/top_level.txt,sha256=Ue9DSRlejRQRCaJueB0uLcKrWwsEq9zezfv5dI5mV1M,8
+maps4fs-2.1.0.dist-info/RECORD,,