maps4fs 2.1.2__py3-none-any.whl → 2.1.3__py3-none-any.whl

maps4fs/generator/component/background.py

@@ -21,6 +21,8 @@ from maps4fs.generator.component.dem import DEM
 from maps4fs.generator.component.texture import Texture
 from maps4fs.generator.settings import Parameters
 
+SEGMENT_LENGTH = 2
+
 
 class Background(MeshComponent, ImageComponent):
     """Component for creating 3D obj files based on DEM data around the map.
@@ -97,6 +99,10 @@ class Background(MeshComponent, ImageComponent):
         cutted_dem_path = self.save_map_dem(self.output_path)
         shutil.copyfile(self.output_path, self.not_substracted_path)
         self.save_map_dem(self.output_path, save_path=self.not_resized_path)
+
+        if self.map.background_settings.flatten_roads:
+            self.flatten_roads()
+
         if self.game.additional_dem_name is not None:
             self.make_copy(cutted_dem_path, self.game.additional_dem_name)
 
@@ -723,3 +729,94 @@ class Background(MeshComponent, ImageComponent):
         self.assets.water_mesh = elevated_save_path
 
         self.plane_from_np(elevated_water, elevated_save_path, include_zeros=False)
+
+    def flatten_roads(self) -> None:
+        """Flattens the roads in the DEM data by averaging the height values along the road polylines."""
+        if not self.not_resized_path or not os.path.isfile(self.not_resized_path):
+            self.logger.warning("No DEM data found for flattening roads.")
+            return
+
+        dem_image = cv2.imread(self.not_resized_path, cv2.IMREAD_UNCHANGED)
+        if dem_image is None:
+            self.logger.warning("Failed to read DEM data.")
+            return
+
+        roads_polylines = self.get_infolayer_data(Parameters.TEXTURES, Parameters.ROADS_POLYLINES)
+        if not roads_polylines:
+            self.logger.warning("No roads polylines found in textures info layer.")
+            return
+
+        self.logger.debug("Found %s roads polylines in textures info layer.", len(roads_polylines))
+
+        full_mask = np.zeros(dem_image.shape, dtype=np.uint8)
+
+        for road_polyline in tqdm(roads_polylines, desc="Flattening roads", unit="road"):
+            points = road_polyline.get("points")
+            width = road_polyline.get("width")
+            if not points or not width:
+                self.logger.warning("Skipping road with insufficient data: %s", road_polyline)
+                continue
+
+            try:
+                fitted_road = self.fit_object_into_bounds(
+                    linestring_points=points, angle=self.rotation
+                )
+            except ValueError as e:
+                self.logger.debug(
+                    "Road polyline could not be fitted into the map bounds with error: %s",
+                    e,
+                )
+                continue
+
+            polyline = shapely.LineString(fitted_road)
+
+            total_length = polyline.length
+            self.logger.debug("Total length of the road polyline: %s", total_length)
+
+            current_distance = 0
+            segments: list[shapely.LineString] = []
+            while current_distance < total_length:
+                start_point = polyline.interpolate(current_distance)
+                end_distance = min(current_distance + SEGMENT_LENGTH, total_length)
+                end_point = polyline.interpolate(end_distance)
+
+                # Create a small segment LineString
+                segment = shapely.LineString([start_point, end_point])
+                segments.append(segment)
+                current_distance += SEGMENT_LENGTH
+
+            self.logger.debug("Number of segments created: %s", len(segments))
+
+            road_polygons: list[shapely.Polygon] = []
+            for segment in segments:
+                polygon = segment.buffer(
+                    width * 2, resolution=4, cap_style="flat", join_style="mitre"
+                )
+                road_polygons.append(polygon)
+
+            for polygon in road_polygons:
+                polygon_points = polygon.exterior.coords
+                road_np = self.polygon_points_to_np(polygon_points)
+                mask = np.zeros(dem_image.shape, dtype=np.uint8)
+
+                try:
+                    cv2.fillPoly(mask, [road_np], 255)  # type: ignore
+                except Exception as e:
+                    self.logger.debug("Could not create mask for road with error: %s", e)
+                    continue
+
+                mean_value = cv2.mean(dem_image, mask=mask)[0]  # type: ignore
+                dem_image[mask == 255] = mean_value
+                full_mask[mask == 255] = 255
+
+        main_dem_path = self.game.dem_file_path(self.map_directory)
+        dem_image = self.blur_by_mask(dem_image, full_mask)
+        dem_image = self.blur_edges_by_mask(dem_image, full_mask)
+
+        output_size = dem_image.shape[0] + 1
+        resized_dem = cv2.resize(
+            dem_image, (output_size, output_size), interpolation=cv2.INTER_NEAREST
+        )
+
+        cv2.imwrite(main_dem_path, resized_dem)
+        self.logger.debug("Flattened roads saved to DEM file: %s", main_dem_path)

maps4fs/generator/component/base/component_image.py

@@ -175,3 +175,59 @@ class ImageComponent(Component):
         elif blur_radius % 2 == 0:
             blur_radius += 1
         return blur_radius
+
+    def blur_by_mask(self, data: np.ndarray, mask: np.ndarray, blur_radius: int = 3) -> np.ndarray:
+        """Blurs the provided image only where the mask is set.
+
+        Arguments:
+            data (np.ndarray): The image data to be blurred.
+            mask (np.ndarray): The mask where the blur should be applied.
+
+        Returns:
+            np.ndarray: The image with the blur applied according to the mask.
+        """
+        if data.shape[:2] != mask.shape[:2]:
+            raise ValueError("Data and mask must have the same dimensions.")
+
+        # Create a blurred version of the data
+        blurred_data = cv2.GaussianBlur(data, (blur_radius, blur_radius), sigmaX=3)
+
+        # Combine the blurred data with the original data using the mask
+        result = np.where(mask == 255, blurred_data, data)
+
+        return result
+
+    def blur_edges_by_mask(
+        self,
+        data: np.ndarray,
+        mask: np.ndarray,
+        bigger_kernel: int = 3,
+        smaller_kernel: int = 1,
+        iterations: int = 1,
+    ) -> np.ndarray:
+        """Blurs the edges of the edge region where changes were made by mask earlier.
+        Creates a slightly bigger mask, a slightly smaller mask, subtract them
+        and obtains the mask for the edges.
+        Then applies blur to the image data only where the mask is set.
+
+        Arguments:
+            data (np.ndarray): The image data to be blurred.
+            mask (np.ndarray): The mask where changes were made.
+            blur_radius (int): The radius of the blur.
+
+        Returns:
+            np.ndarray: The image with the edges blurred according to the mask.
+        """
+        if data.shape[:2] != mask.shape[:2]:
+            raise ValueError("Data and mask must have the same dimensions.")
+
+        bigger_mask = cv2.dilate(
+            mask, np.ones((bigger_kernel, bigger_kernel), np.uint8), iterations=iterations
+        )
+        smaller_mask = cv2.erode(
+            mask, np.ones((smaller_kernel, smaller_kernel), np.uint8), iterations=iterations
+        )
+
+        edge_mask = cv2.subtract(bigger_mask, smaller_mask)
+
+        return self.blur_by_mask(data, edge_mask)

maps4fs/generator/settings.py

@@ -163,12 +163,14 @@ class BackgroundSettings(SettingsModel):
         water_blurriness (int): blurriness of the water.
         remove_center (bool): remove the center of the background terrain.
             It will be used to remove the center of the map where the player starts.
+        flatten_roads (bool): if True, roads will be flattened in the DEM data.
     """
 
     generate_background: bool = False
     generate_water: bool = False
     water_blurriness: int = 20
     remove_center: bool = True
+    flatten_roads: bool = False
 
 
 class GRLESettings(SettingsModel):

{maps4fs-2.1.2.dist-info → maps4fs-2.1.3.dist-info}/METADATA

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

{maps4fs-2.1.2.dist-info → maps4fs-2.1.3.dist-info}/RECORD

@@ -4,10 +4,10 @@ maps4fs/generator/__init__.py,sha256=zZMLEkGzb4z0xql650gOtGSvcgX58DnJ2yN3vC2daRk
 maps4fs/generator/game.py,sha256=Qqeh1rg0RFghhu9uhBMn_XKXYkTkg7fsjY1xO2eIZRw,13777
 maps4fs/generator/map.py,sha256=x_NjUwKkDnh2cC61QpTfX88mFHAkQG1sxVwZckoV0mE,16357
 maps4fs/generator/qgis.py,sha256=Es8hLuqN_KH8lDfnJE6He2rWYbAKJ3RGPn-o87S6CPI,6116
-maps4fs/generator/settings.py,sha256=OXgCjpmcqvTENRcKKYcEhES3GD3KPpvrbeJDv8g25yY,7676
+maps4fs/generator/settings.py,sha256=bjEhn1918lxSsIRgmOXVpYWS3E4y72eqx2Ty7f99JKE,7788
 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/background.py,sha256=0fTmghJZzmZMqleTmFc5C6A53eZm7nhMpLYmkELAHmQ,33507
 maps4fs/generator/component/config.py,sha256=uL76h9UwyhZKZmbxz0mBmWtEPN6qYay4epTEqqtej60,8601
 maps4fs/generator/component/dem.py,sha256=SH_2Zu5O4dhWtZeOkCwzDF4RU04XhTdpGFYaRYJkdjc,11905
 maps4fs/generator/component/grle.py,sha256=3BKGlR0q0t0NvmqeT81WieS6MIc_UlMQjIDDZiqTiws,27243
@@ -17,11 +17,11 @@ maps4fs/generator/component/satellite.py,sha256=OsxoNOCgkUtRzL7Geuqubsf6uoKXAIN8
 maps4fs/generator/component/texture.py,sha256=bpHBwUksEbwPQrdQ8K94m_MIKXnWYnUMvP987JVabAA,34987
 maps4fs/generator/component/base/__init__.py,sha256=zZMLEkGzb4z0xql650gOtGSvcgX58DnJ2yN3vC2daRk,43
 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_image.py,sha256=n_tHyq8J1zWnb1_8O6fhNNd_4Wfj2VU2QqSAEzJXf60,8119
 maps4fs/generator/component/base/component_mesh.py,sha256=_thzgjJDroMn-9SBsBmAWizcSsnV9U5445SD18Tx1kc,9090
 maps4fs/generator/component/base/component_xml.py,sha256=MT-VhU2dEckLFxAgmxg6V3gnv11di_94Qq6atfpOLdc,5342
-maps4fs-2.1.2.dist-info/licenses/LICENSE.md,sha256=pTKD_oUexcn-yccFCTrMeLkZy0ifLRa-VNcDLqLZaIw,10749
-maps4fs-2.1.2.dist-info/METADATA,sha256=L6AJvi0CMTG_RfUP9isMXDYTnugeBPWQzWQwwy2NyTg,45431
-maps4fs-2.1.2.dist-info/WHEEL,sha256=_zCd3N1l69ArxyTb8rzEoP9TpbYXkqRFSNOD5OuxnTs,91
-maps4fs-2.1.2.dist-info/top_level.txt,sha256=Ue9DSRlejRQRCaJueB0uLcKrWwsEq9zezfv5dI5mV1M,8
-maps4fs-2.1.2.dist-info/RECORD,,
+maps4fs-2.1.3.dist-info/licenses/LICENSE.md,sha256=pTKD_oUexcn-yccFCTrMeLkZy0ifLRa-VNcDLqLZaIw,10749
+maps4fs-2.1.3.dist-info/METADATA,sha256=qJ4vDMMwhoupUdurZjg1aROKauLLhWLx9845zyynngY,45431
+maps4fs-2.1.3.dist-info/WHEEL,sha256=_zCd3N1l69ArxyTb8rzEoP9TpbYXkqRFSNOD5OuxnTs,91
+maps4fs-2.1.3.dist-info/top_level.txt,sha256=Ue9DSRlejRQRCaJueB0uLcKrWwsEq9zezfv5dI5mV1M,8
+maps4fs-2.1.3.dist-info/RECORD,,