maps4fs 1.8.236__py3-none-any.whl → 1.8.238__py3-none-any.whl

maps4fs/generator/component/background.py

@@ -3,13 +3,13 @@ around the map."""
 
 from __future__ import annotations
 
-import json
 import os
 import shutil
 from copy import deepcopy
 
 import cv2
 import numpy as np
+from tqdm import tqdm
 
 from maps4fs.generator.component.base.component_image import ImageComponent
 from maps4fs.generator.component.base.component_mesh import MeshComponent
@@ -101,6 +101,56 @@ class Background(MeshComponent, ImageComponent):
         if self.map.background_settings.generate_water:
             self.generate_water_resources_obj()
 
+    def create_foundations(self, dem_image: np.ndarray) -> np.ndarray:
+        """Creates foundations for buildings based on the DEM data.
+
+        Arguments:
+            dem_image (np.ndarray): The DEM data as a numpy array.
+
+        Returns:
+            np.ndarray: The DEM data with the foundations added.
+        """
+        buildings = self.get_infolayer_data(Parameters.TEXTURES, Parameters.BUILDINGS)
+        if not buildings:
+            self.logger.warning("Buildings data not found in textures info layer.")
+            return dem_image
+
+        self.logger.debug("Found %s buildings in textures info layer.", len(buildings))
+
+        for building in tqdm(buildings, desc="Creating foundations", unit="building"):
+            try:
+                fitted_building = self.fit_object_into_bounds(
+                    polygon_points=building, angle=self.rotation
+                )
+            except ValueError as e:
+                self.logger.debug(
+                    "Building could not be fitted into the map bounds with error: %s",
+                    e,
+                )
+                continue
+
+            # 1. Read the pixel values from the DEM image.
+            # 2. Calculate the average pixel value of the building area.
+            # 3. Set the pixel values in the DEM to the average pixel value.
+
+            building_np = self.polygon_points_to_np(fitted_building)
+            mask = np.zeros(dem_image.shape, dtype=np.uint8)
+
+            try:
+                cv2.fillPoly(mask, [building_np], 255)  # type: ignore
+            except Exception as e:
+                self.logger.debug("Could not create mask for building with error: %s", e)
+                continue
+
+            mean_value = cv2.mean(dem_image, mask=mask)[0]
+            mean_value = np.round(mean_value).astype(dem_image.dtype)
+            self.logger.info("Mean value of the building area: %s", mean_value)
+
+            # Set the pixel values in the DEM to the average pixel value.
+            dem_image[mask == 255] = mean_value
+
+        return dem_image
+
     def make_copy(self, dem_path: str, dem_name: str) -> None:
         """Copies DEM data to additional DEM file.
 
@@ -198,6 +248,9 @@ class Background(MeshComponent, ImageComponent):
             self.logger.debug("Not resized DEM saved: %s", save_path)
             return save_path
 
+        if self.map.dem_settings.add_foundations:
+            dem_data = self.create_foundations(dem_data)
+
         output_size = self.map_size + 1
 
         main_dem_path = self.game.dem_file_path(self.map_directory)
@@ -347,13 +400,11 @@ class Background(MeshComponent, ImageComponent):
 
     def create_background_textures(self) -> None:
         """Creates background textures for the map."""
-        if not os.path.isfile(self.game.texture_schema):
-            self.logger.warning("Texture schema file not found: %s", self.game.texture_schema)
+        layers_schema = self.map.texture_schema
+        if not layers_schema:
+            self.logger.warning("No texture schema found.")
             return
 
-        with open(self.game.texture_schema, "r", encoding="utf-8") as f:
-            layers_schema = json.load(f)
-
         background_layers = []
         for layer in layers_schema:
             if layer.get("background") is True:

maps4fs/generator/map.py

@@ -184,6 +184,18 @@ class Map:
         except Exception as e:
             raise RuntimeError(f"Can not unpack map template due to error: {e}") from e
 
+    @property
+    def texture_schema(self) -> list[dict[str, Any]] | None:
+        """Return texture schema (custom if provided, default otherwise).
+
+        Returns:
+            list[dict[str, Any]] | None: Texture schema.
+        """
+        if self.texture_custom_schema:
+            return self.texture_custom_schema
+        with open(self.game.texture_schema, "r", encoding="utf-8") as file:
+            return json.load(file)
+
     def generate(self) -> Generator[str, None, None]:
         """Launch map generation using all components. Yield component names during the process.
 

maps4fs/generator/settings.py

@@ -13,6 +13,7 @@ class Parameters:
 
     FIELD = "field"
     FIELDS = "fields"
+    BUILDINGS = "buildings"
     TEXTURES = "textures"
     FOREST = "forest"
     ROADS_POLYLINES = "roads_polylines"
@@ -133,6 +134,7 @@ class DEMSettings(SettingsModel):
     ceiling: int = 0
     water_depth: int = 0
     blur_radius: int = 3
+    add_foundations: bool = False
 
 
 class BackgroundSettings(SettingsModel):

{maps4fs-1.8.236.dist-info → maps4fs-1.8.238.dist-info}/METADATA

@@ -1,6 +1,6 @@
 Metadata-Version: 2.2
 Name: maps4fs
-Version: 1.8.236
+Version: 1.8.238
 Summary: Generate map templates for Farming Simulator from real places.
 Author-email: iwatkot <iwatkot@gmail.com>
 License: MIT License
@@ -590,6 +590,8 @@ You can also apply some advanced settings to the map generation process.<br>
 
 - Water depth: Water depth value (in meters) will be subtracted from the DEM image, making the water deeper. The pixel value used for this is calculated based on the heightScale value for your map.
 
+- Add foundations: If enabled the terrain under the buildings will be flattened to the average height of the building.
+
 ### Background terrain Advanced settings
 
 - Generate background - if enabled, the obj files for the background terrain will be generated. You can turn it off if you already have those files or don't need them. By default, it's set to True.

{maps4fs-1.8.236.dist-info → maps4fs-1.8.238.dist-info}/RECORD

@@ -3,12 +3,12 @@ maps4fs/logger.py,sha256=HQrDyj72mUjVYo25aR_-_SxVn2rfFjDCNbj-JKJdSnE,1488
 maps4fs/generator/__init__.py,sha256=zZMLEkGzb4z0xql650gOtGSvcgX58DnJ2yN3vC2daRk,43
 maps4fs/generator/dem.py,sha256=a9B3tatj7pzvvdLIyLw7BA3JoDTibFczpqiXJnx054U,12864
 maps4fs/generator/game.py,sha256=NZaxj5z7WzMiHzAvQyr-TvVjGoHgqGldM6ZsItuYyzA,11292
-maps4fs/generator/map.py,sha256=tgbpu3xxtFFWMualvaOnfnBXjDYWqAa_xIpu-OvcnbQ,12636
+maps4fs/generator/map.py,sha256=9tQTEMYvh4RSC5ACp7fE7RhVDGa7Mc-P5W95h-Zineg,13069
 maps4fs/generator/qgis.py,sha256=Es8hLuqN_KH8lDfnJE6He2rWYbAKJ3RGPn-o87S6CPI,6116
-maps4fs/generator/settings.py,sha256=DJCG--Z__JBEc-fllWOvnb4roYuNFv7bSkAclUZV2gE,6786
+maps4fs/generator/settings.py,sha256=-QdpmIoMiWXPeqYX5QQ5Pplz9xHk2OnyBdzRIc7aB44,6848
 maps4fs/generator/statistics.py,sha256=aynS3zbAtiwnU_YLKHPTiiaKW98_suvQUhy1SGBA6mc,2448
 maps4fs/generator/component/__init__.py,sha256=s01yVVVi8R2xxNvflu2D6wTd9I_g73AMM2x7vAC7GX4,490
-maps4fs/generator/component/background.py,sha256=7YkSFrbPHN_dTh4xB1lM60mloLJXOdXZw19T0Gs4x1k,18971
+maps4fs/generator/component/background.py,sha256=RpEjOW7ray0h_Jd6Mf8XEzGjKtJeYPzze9bPT8M1fhA,21011
 maps4fs/generator/component/config.py,sha256=RitKgFDZPzjA1fi8GcEi1na75qqaueUvpcITHjBvCXc,3674
 maps4fs/generator/component/grle.py,sha256=C6wpYQrJBVRb3vVbEAY5BYTRYU9BaefmoWCXoPrVn2A,19454
 maps4fs/generator/component/i3d.py,sha256=yWGs5eBuBT7V4F6SbTNAmF-8fj2Zy4kcC5OKZ_XVICs,21329
@@ -53,8 +53,8 @@ maps4fs/toolbox/__init__.py,sha256=zZMLEkGzb4z0xql650gOtGSvcgX58DnJ2yN3vC2daRk,4
 maps4fs/toolbox/background.py,sha256=RclEqxEWLbMxuEkkegQP8jybzugwQ1_R3rdfDe0s21U,2104
 maps4fs/toolbox/custom_osm.py,sha256=X6ZlPqiOhNjkmdD_qVroIfdOl9Rb90cDwVSLDVYgx80,1892
 maps4fs/toolbox/dem.py,sha256=z9IPFNmYbjiigb3t02ZenI3Mo8odd19c5MZbjDEovTo,3525
-maps4fs-1.8.236.dist-info/LICENSE.md,sha256=pTKD_oUexcn-yccFCTrMeLkZy0ifLRa-VNcDLqLZaIw,10749
-maps4fs-1.8.236.dist-info/METADATA,sha256=mZfVF18tG8BlX96UWN_T35hCIGaveTOGUpAjwAXsIQY,46039
-maps4fs-1.8.236.dist-info/WHEEL,sha256=In9FTNxeP60KnTkGw7wk6mJPYd_dQSjEZmXdBdMCI-8,91
-maps4fs-1.8.236.dist-info/top_level.txt,sha256=Ue9DSRlejRQRCaJueB0uLcKrWwsEq9zezfv5dI5mV1M,8
-maps4fs-1.8.236.dist-info/RECORD,,
+maps4fs-1.8.238.dist-info/LICENSE.md,sha256=pTKD_oUexcn-yccFCTrMeLkZy0ifLRa-VNcDLqLZaIw,10749
+maps4fs-1.8.238.dist-info/METADATA,sha256=5zXEOsIxjrknQLlwShN1CsM7Br62aJUMaDJ2F2UlGec,46159
+maps4fs-1.8.238.dist-info/WHEEL,sha256=In9FTNxeP60KnTkGw7wk6mJPYd_dQSjEZmXdBdMCI-8,91
+maps4fs-1.8.238.dist-info/top_level.txt,sha256=Ue9DSRlejRQRCaJueB0uLcKrWwsEq9zezfv5dI5mV1M,8
+maps4fs-1.8.238.dist-info/RECORD,,