maps4fs 1.8.235__py3-none-any.whl → 1.8.237__py3-none-any.whl

maps4fs/generator/component/background.py

@@ -10,6 +10,7 @@ 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 +102,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 +249,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)

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.235.dist-info → maps4fs-1.8.237.dist-info}/METADATA

@@ -1,6 +1,6 @@
 Metadata-Version: 2.2
 Name: maps4fs
-Version: 1.8.235
+Version: 1.8.237
 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.235.dist-info → maps4fs-1.8.237.dist-info}/RECORD

@@ -5,10 +5,10 @@ maps4fs/generator/dem.py,sha256=a9B3tatj7pzvvdLIyLw7BA3JoDTibFczpqiXJnx054U,1286
 maps4fs/generator/game.py,sha256=NZaxj5z7WzMiHzAvQyr-TvVjGoHgqGldM6ZsItuYyzA,11292
 maps4fs/generator/map.py,sha256=tgbpu3xxtFFWMualvaOnfnBXjDYWqAa_xIpu-OvcnbQ,12636
 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=RhUDOTfo7Ph9B2bM5tp85wTg6FKycpsf1L_pHYa2TLc,21152
 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.235.dist-info/LICENSE.md,sha256=pTKD_oUexcn-yccFCTrMeLkZy0ifLRa-VNcDLqLZaIw,10749
-maps4fs-1.8.235.dist-info/METADATA,sha256=1xbg950DUENzpMbqA1MXVSnwzdq6f5Ypb6nb_2Z4oEk,46039
-maps4fs-1.8.235.dist-info/WHEEL,sha256=In9FTNxeP60KnTkGw7wk6mJPYd_dQSjEZmXdBdMCI-8,91
-maps4fs-1.8.235.dist-info/top_level.txt,sha256=Ue9DSRlejRQRCaJueB0uLcKrWwsEq9zezfv5dI5mV1M,8
-maps4fs-1.8.235.dist-info/RECORD,,
+maps4fs-1.8.237.dist-info/LICENSE.md,sha256=pTKD_oUexcn-yccFCTrMeLkZy0ifLRa-VNcDLqLZaIw,10749
+maps4fs-1.8.237.dist-info/METADATA,sha256=cYwHiebwnB_vFE-JNMqZBw2xXiU4XNL5rJjWAOpldCY,46159
+maps4fs-1.8.237.dist-info/WHEEL,sha256=In9FTNxeP60KnTkGw7wk6mJPYd_dQSjEZmXdBdMCI-8,91
+maps4fs-1.8.237.dist-info/top_level.txt,sha256=Ue9DSRlejRQRCaJueB0uLcKrWwsEq9zezfv5dI5mV1M,8
+maps4fs-1.8.237.dist-info/RECORD,,