Refined automatic map generation script

2025-10-29 16:56:34 +00:00 · 2024-02-23 15:54:49 +01:00
parent acb55044d1
commit 9a571132c8
11 changed files with 274 additions and 155 deletions
--- a/scripts/python/map_generator/main.py
+++ b/scripts/python/map_generator/main.py
@@ -0,0 +1,67 @@
+import sys
+import yaml
+from pyproj import Geod
+from fastkml import kml
+from shapely import wkt
+from datetime import timedelta
+
+import map_generator
+
+if len(sys.argv) == 1:
+    print("Please provide a configuration file as first argument. You can also drop the configuration file on this script to run it.")
+else:
+    config_file = sys.argv[1]
+    print(f"Using config file: {config_file}")
+    
+    with open('configs/screen_properties.yml', 'r') as sp:
+        with open(config_file, 'r') as cp:
+            screen_config = yaml.safe_load(sp)
+            map_config = yaml.safe_load(cp)
+
+            print("#################################################################################################################################################")
+            print("# IMPORTANT NOTE: the screen properties must be configured according to your screen and desired zoom level. Make sure you set them accordingly. #")
+            print("#################################################################################################################################################")
+
+            print("Screen parameters:")
+            print(f"-> Screen width: {screen_config["width"]}px")
+            print(f"-> Screen height: {screen_config["height"]}px")
+            print(f"-> Geographic resolution: {screen_config["geo_resolution"]} meters/pixel")
+            
+            print("Map parameters:")
+            print(f"-> Output directory: {map_config["output_directory"]}")
+            print(f"-> Boundary file: {map_config["boundary_file"]}")
+
+            with open(map_config["boundary_file"], 'rt', encoding="utf-8") as bp:
+                # Read the config file and compute the total area of the covered map
+                doc = bp.read()
+                k = kml.KML()
+                k.from_string(doc)
+
+                geod = Geod(ellps="WGS84")
+                features = []
+                area = 0
+                for feature in k.features():
+                    for sub_feature in list(feature.features()):
+                        geo = sub_feature.geometry
+                        area += abs(geod.geometry_area_perimeter(wkt.loads(geo.wkt))[0])
+                        features.append(sub_feature)
+
+                print(f"Found {len(features)} features in the provided kml file")
+
+                tile_size = 256 * screen_config["geo_resolution"] # meters
+                tiles_per_screenshot = int(screen_config["width"] / 256) * int(screen_config["height"] / 256)
+                tiles_num = int(area / (tile_size * tile_size))
+                screenshots_num = int(tiles_num / tiles_per_screenshot)
+                total_time = int(screenshots_num / 1.0)
+                
+                print(f"Total area: {int(area / 1e6)} square kilometers")
+                print(f"Estimated number of tiles: {tiles_num}")
+                print(f"Estimated number of screenshots: {screenshots_num}")
+                print(f"Estimated time to complete: {timedelta(seconds=total_time)} (hh:mm:ss)")
+                print("The script is ready to go. After you press enter, it will wait for 5 seconds, then it will start.")
+
+                input("Press enter to continue...")
+                map_generator.run(map_config)
+    
+
+