maps4fs 1.8.172__py3-none-any.whl → 1.8.173__py3-none-any.whl

maps4fs/generator/component/background.py

@@ -235,24 +235,16 @@ class Background(MeshComponent, ImageComponent):
                 This setting is used for a Background Terrain, where the center part where the
                 playable area is will be cut out.
         """
-        resize_factor = 1 / self.map.background_settings.resize_factor
-        dem_data = cv2.resize(dem_data, (0, 0), fx=resize_factor, fy=resize_factor)
-        if remove_center:
-            half_size = int(self.map_size // 2 * resize_factor)
-            dem_data = self.cut_out_np(dem_data, half_size, set_zeros=True)
-            self.logger.debug("Center removed from DEM data.")
-        self.logger.debug(
-            "DEM data resized to shape: %s with factor: %s", dem_data.shape, resize_factor
-        )
-
         mesh = self.mesh_from_np(
             dem_data,
             include_zeros=include_zeros,
             z_scaling_factor=self.get_z_scaling_factor(),
-            resize_factor=resize_factor,
+            resize_factor=self.map.background_settings.resize_factor,
             apply_decimation=self.map.background_settings.apply_decimation,
             decimation_percent=self.map.background_settings.decimation_percent,
             decimation_agression=self.map.background_settings.decimation_agression,
+            remove_center=remove_center,
+            remove_size=self.map_size,
         )
 
         mesh.export(save_path)
@@ -440,6 +432,12 @@ class Background(MeshComponent, ImageComponent):
 
         # Single channeled 8 bit image, where the water have values of 255, and the rest 0.
         plane_water = cv2.imread(self.water_resources_path, cv2.IMREAD_UNCHANGED)
+
+        # Check if the image contains non-zero values.
+        if not np.any(plane_water):
+            self.logger.warning("Water resources image is empty, skipping water generation.")
+            return
+
         dilated_plane_water = cv2.dilate(
             plane_water.astype(np.uint8), np.ones((5, 5), np.uint8), iterations=5
         ).astype(np.uint8)

maps4fs/generator/component/base/component_mesh.py

@@ -53,10 +53,12 @@ class MeshComponent(Component):
         image: np.ndarray,
         include_zeros: bool,
         z_scaling_factor: float,
-        resize_factor: float,
+        resize_factor: int,
         apply_decimation: bool,
         decimation_percent: int,
         decimation_agression: int,
+        remove_center: bool,
+        remove_size: int,
     ) -> trimesh.Trimesh:
         """Generates a mesh from the given numpy array.
 
@@ -64,16 +66,21 @@ class MeshComponent(Component):
             image (np.ndarray): The numpy array to generate the mesh from.
             include_zeros (bool): Whether to include zero values in the mesh.
             z_scaling_factor (float): The scaling factor for the Z-axis.
-            resize_factor (float): The resizing factor.
+            resize_factor (int): The resizing factor.
             apply_decimation (bool): Whether to apply decimation to the mesh.
             decimation_percent (int): The percent of the decimation.
             decimation_agression (int): The agression of the decimation.
+            remove_center (bool): Whether to remove the center from the mesh.
+            remove_size (int): The size of the center to remove.
 
         Returns:
             trimesh.Trimesh: The generated mesh.
         """
+        output_x_size, _ = image.shape
         image = image.max() - image
 
+        image = image[::resize_factor, ::resize_factor]
+
         rows, cols = image.shape
         x = np.linspace(0, cols - 1, cols)
         y = np.linspace(0, rows - 1, rows)
@@ -106,15 +113,7 @@ class MeshComponent(Component):
 
         faces_np = np.array(faces)
         mesh = trimesh.Trimesh(vertices=vertices, faces=faces_np)
-
-        # Apply rotation: 180 degrees around Y-axis and Z-axis
-        rotation_matrix_y = trimesh.transformations.rotation_matrix(np.pi, [0, 1, 0])
-        rotation_matrix_z = trimesh.transformations.rotation_matrix(np.pi, [0, 0, 1])
-        mesh.apply_transform(rotation_matrix_y)
-        mesh.apply_transform(rotation_matrix_z)
-
-        # if not include_zeros:
-        mesh.apply_scale([1 / resize_factor, 1 / resize_factor, z_scaling_factor])
+        mesh = MeshComponent.rotate_mesh(mesh)
 
         if apply_decimation:
             percent = decimation_percent / 100
@@ -122,4 +121,122 @@ class MeshComponent(Component):
                 percent=percent, aggression=decimation_agression
             )
 
+        try:
+            if not mesh.is_watertight:
+                mesh = MeshComponent.fix_mesh(mesh)
+        except Exception:
+            pass
+
+        mesh = MeshComponent.mesh_to_output_size(
+            mesh,
+            resize_factor,
+            z_scaling_factor,
+            output_x_size,
+            skip_resize_to_expected_size=not include_zeros,
+        )
+
+        if remove_center:
+            mesh = MeshComponent.remove_center_from_mesh(mesh, remove_size)
+
         return mesh
+
+    @staticmethod
+    def rotate_mesh(mesh: trimesh.Trimesh) -> trimesh.Trimesh:
+        """Rotates the given mesh by 180 degrees around the Y-axis and Z-axis.
+
+        Arguments:
+            mesh (trimesh.Trimesh): The mesh to rotate.
+
+        Returns:
+            trimesh.Trimesh: The rotated mesh.
+        """
+        mesh_copy = mesh.copy()
+
+        rotation_matrices = [
+            trimesh.transformations.rotation_matrix(np.pi, [0, 1, 0]),
+            trimesh.transformations.rotation_matrix(np.pi, [0, 0, 1]),
+        ]
+
+        for rotation_matrix in tqdm(rotation_matrices, desc="Rotating mesh", unit="rotation"):
+            mesh_copy.apply_transform(rotation_matrix)
+
+        return mesh_copy
+
+    @staticmethod
+    def fix_mesh(mesh: trimesh.Trimesh) -> trimesh.Trimesh:
+        """Fixes the given mesh by filling holes, fixing normals, fixing winding, fixing inversion,
+        fixing broken faces, and stitching.
+
+        Arguments:
+            mesh (trimesh.Trimesh): The mesh to fix.
+
+        Returns:
+            trimesh.Trimesh: The fixed mesh.
+        """
+        mesh_copy = mesh.copy()
+
+        fix_methods = [
+            trimesh.repair.fill_holes,
+            trimesh.repair.fix_normals,
+            trimesh.repair.fix_winding,
+            trimesh.repair.fix_inversion,
+            trimesh.repair.broken_faces,
+            trimesh.repair.stitch,
+        ]
+
+        for method in tqdm(fix_methods, desc="Fixing mesh", unit="method"):
+            method(mesh_copy)  # type: ignore
+
+        return mesh_copy
+
+    @staticmethod
+    def mesh_to_output_size(
+        mesh: trimesh.Trimesh,
+        resize_factor: int,
+        z_scaling_factor: float,
+        expected_size: int,
+        skip_resize_to_expected_size: bool = False,
+    ) -> trimesh.Trimesh:
+        """Resizes the given mesh to the expected size.
+
+        Arguments:
+            mesh (trimesh.Trimesh): The mesh to resize.
+            resize_factor (int): The resizing factor.
+            z_scaling_factor (float): The scaling factor for the Z-axis.
+            expected_size (int): The expected size.
+            skip_resize_to_expected_size (bool): Whether to skip resizing to the expected size.
+
+        Returns:
+            trimesh.Trimesh: The resized mesh.
+        """
+        mesh_copy = mesh.copy()
+
+        mesh_copy.apply_scale([resize_factor / 1, resize_factor / 1, z_scaling_factor])
+
+        if not skip_resize_to_expected_size:
+            x_size, y_size, _ = mesh_copy.extents
+            x_resize_factor = expected_size / x_size
+            y_resize_factor = expected_size / y_size
+
+            mesh_copy.apply_scale([x_resize_factor, y_resize_factor, 1])
+        return mesh_copy
+
+    @staticmethod
+    def remove_center_from_mesh(mesh: trimesh.Trimesh, remove_size: int) -> trimesh.Trimesh:
+        """Removes the center from the given mesh.
+
+        Arguments:
+            mesh (trimesh.Trimesh): The mesh to remove the center from.
+            remove_size (int): The size of the center to remove.
+
+        Returns:
+            trimesh.Trimesh: The mesh with the center removed.
+        """
+        mesh_copy = mesh.copy()
+
+        _, _, z_size = mesh_copy.extents
+
+        cube_mesh = trimesh.creation.box([remove_size, remove_size, z_size * 4])
+        cube_mesh.apply_translation(mesh_copy.centroid - cube_mesh.centroid)
+
+        return trimesh.boolean.difference([mesh_copy, cube_mesh], check_volume=False)

maps4fs/generator/settings.py

@@ -133,7 +133,7 @@ class BackgroundSettings(SettingsModel):
     generate_background: bool = False
     generate_water: bool = False
     resize_factor: int = 8
-    remove_center: bool = False
+    remove_center: bool = True
     apply_decimation: bool = False
     decimation_percent: int = 25
     decimation_agression: int = 3

{maps4fs-1.8.172.dist-info → maps4fs-1.8.173.dist-info}/METADATA

@@ -1,6 +1,6 @@
 Metadata-Version: 2.2
 Name: maps4fs
-Version: 1.8.172
+Version: 1.8.173
 Summary: Generate map templates for Farming Simulator from real places.
 Author-email: iwatkot <iwatkot@gmail.com>
 License: MIT License
@@ -582,7 +582,7 @@ You can also apply some advanced settings to the map generation process.<br>
 
 - Resize factor - the factor by which the background terrain will be resized. It will be used as 1 / resize_factor while generating the models. Which means that the larger the value the more the terrain will be resized. The lowest value is 1, in this case background terrain will not be resized. Note, than low values will lead to long processing and enormous size of the obj files.
 
-- Remove center - if enabled, the playable region (map terrain) will be removed from the background terrain. Note, that it will require low resize factors, to avoid gaps between the map and the background terrain.
+- Remove center - if enabled, the playable region (map terrain) will be removed from the background terrain. By default, it's set to True.
 
 - Apply decimation - if enabled, the mesh will be simplified to reduce the number of faces.
 

{maps4fs-1.8.172.dist-info → maps4fs-1.8.173.dist-info}/RECORD

@@ -6,17 +6,17 @@ maps4fs/generator/game.py,sha256=YwtdzqwcBJm1FHtc1ehDd3WbfTTW_gBvxK-pzfU48zs,110
 maps4fs/generator/map.py,sha256=c5GMhr5iTRdXvTXoBKyVYa2V1tocR3ZCc7D_hpU21k8,11523
 maps4fs/generator/qgis.py,sha256=Es8hLuqN_KH8lDfnJE6He2rWYbAKJ3RGPn-o87S6CPI,6116
 maps4fs/generator/satellite.py,sha256=t33EfaxWTCjcKYnqoppSlFdQQSekiBe7UFmLgTVhFQI,3650
-maps4fs/generator/settings.py,sha256=sMyDY29fweF_Yo_buaUELFSYBYkOqf8VsZPL8lM8j1c,6271
+maps4fs/generator/settings.py,sha256=cFlN-gK8QcySqyPtcGm-2fLnxQnlmC3Y9kQufJxwI3Y,6270
 maps4fs/generator/texture.py,sha256=_IfqIuxH4934VJXKtdABHa6ToPWk3T0xknvlu-rZ5Uc,36570
 maps4fs/generator/component/__init__.py,sha256=zZMLEkGzb4z0xql650gOtGSvcgX58DnJ2yN3vC2daRk,43
-maps4fs/generator/component/background.py,sha256=HWwQnZ5k-D4MdDSUdRvvNmWGUTtqps942DVGdpEQRoE,18947
+maps4fs/generator/component/background.py,sha256=cBBp-ULrJzuNG9wfE_MaxzAn_mR24neqtL7WjxXEb3I,18749
 maps4fs/generator/component/config.py,sha256=RitKgFDZPzjA1fi8GcEi1na75qqaueUvpcITHjBvCXc,3674
 maps4fs/generator/component/grle.py,sha256=aKMjVJNuKHJJ2gsXaH00bz10kWaIbbZXU_JbP-ZAGw4,18990
 maps4fs/generator/component/i3d.py,sha256=3x38yL-kSJ8ylBwICBb6wPYzRSky4gVj8XCk2jzYSeo,19861
 maps4fs/generator/component/base/__init__.py,sha256=zZMLEkGzb4z0xql650gOtGSvcgX58DnJ2yN3vC2daRk,43
 maps4fs/generator/component/base/component.py,sha256=U3XJae0GUvHN08psv2j24Y4PBAAytSkSK3AmD-VjjXE,21404
 maps4fs/generator/component/base/component_image.py,sha256=vuiiJqsrKZgYsnK8Qa5yL2Pejqa6nJSTtXNAcBPL29c,3099
-maps4fs/generator/component/base/component_mesh.py,sha256=-9kmMX5KUULAIScygcrzmq8Cg643hiZ2IfSZOKiFfRs,4634
+maps4fs/generator/component/base/component_mesh.py,sha256=jtsEQZkqiCkycYpueq1M1eqgxdShL_bHaJT6Upu8PLI,8412
 maps4fs/generator/component/base/component_xml.py,sha256=6OO1dKoceO1ACk7-k1oGtnkfNud8ZN3u3ZNjdNMpTqw,3967
 maps4fs/generator/dtm/__init__.py,sha256=47DEQpj8HBSa-_TImW-5JCeuQeRkm5NMpJWZG3hSuFU,0
 maps4fs/generator/dtm/bavaria.py,sha256=7njrEvSCYAC8ZVyvS-_84iXHhWA0oHKrEqSzxdnZuGs,4293
@@ -42,8 +42,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.172.dist-info/LICENSE.md,sha256=pTKD_oUexcn-yccFCTrMeLkZy0ifLRa-VNcDLqLZaIw,10749
-maps4fs-1.8.172.dist-info/METADATA,sha256=E5nc98xim9woImXF8JqFv44PLOjeqiBHhqdFfB8H3WA,43012
-maps4fs-1.8.172.dist-info/WHEEL,sha256=In9FTNxeP60KnTkGw7wk6mJPYd_dQSjEZmXdBdMCI-8,91
-maps4fs-1.8.172.dist-info/top_level.txt,sha256=Ue9DSRlejRQRCaJueB0uLcKrWwsEq9zezfv5dI5mV1M,8
-maps4fs-1.8.172.dist-info/RECORD,,
+maps4fs-1.8.173.dist-info/LICENSE.md,sha256=pTKD_oUexcn-yccFCTrMeLkZy0ifLRa-VNcDLqLZaIw,10749
+maps4fs-1.8.173.dist-info/METADATA,sha256=juPm5cwfn3XLlaUemJjP43yHQbW-9NYEljrHscMlGN4,42937
+maps4fs-1.8.173.dist-info/WHEEL,sha256=In9FTNxeP60KnTkGw7wk6mJPYd_dQSjEZmXdBdMCI-8,91
+maps4fs-1.8.173.dist-info/top_level.txt,sha256=Ue9DSRlejRQRCaJueB0uLcKrWwsEq9zezfv5dI5mV1M,8
+maps4fs-1.8.173.dist-info/RECORD,,