updated scripts to create minimap and map image

scripts/generate_track_minimap.py

@@ -0,0 +1,34 @@
+"""
+Note : use this script from the ../scripts/ directory
+"""
+import glob
+import subprocess
+import os
+import shutil
+
+from source.wszst import szs
+from scripts import obj_to_png
+
+
+def get_track_minimap(directory: str, sha1: str):
+    os.makedirs(tmp_dir := f"./scripts/tmp/{sha1}/", exist_ok=True)
+
+    szs.extract(f"{directory}{sha1}.szs", tmp_dir + "track.szs")
+    subprocess.run(["abmatt", "convert", tmp_dir + "track.szs.d/map_model.brres", "to", tmp_dir + "map_model.obj"])
+
+    try: img = obj_to_png.render_top_view(obj_file=tmp_dir + "map_model.obj")
+    except Exception as e:
+        print(e)
+        return None
+    finally:
+        shutil.rmtree(tmp_dir, ignore_errors=True)
+
+    return img
+
+
+directory = "./file/Track/"
+
+for track in glob.glob("*.szs", root_dir=directory):
+    sha1 = track.replace(".szs", "")
+    if (image := get_track_minimap(directory, sha1)) is not None:
+        image.save(f"./scripts/minimap/{sha1}.png")

scripts/ask_map_preview.py → scripts/generate_track_preview.py

@@ -20,8 +20,10 @@
 driver.fullscreen_window()
 time.sleep(5)
 
-driver.execute_script("var element = arguments[0]; element.parentNode.removeChild(element);",
-                      driver.find_element_by_id("Panel"))
+driver.execute_script(
+    "var element = arguments[0]; element.parentNode.removeChild(element);",
+    driver.find_element_by_id("Panel")
+)
 
 tracks = glob.iglob("../file/Track/*.szs")
 track = "none.png"

scripts/requirements.txt

@@ -1,6 +1,7 @@
-discord.py
 pygame
 PyOpenGL
 pywavefront
+numpy
+
 selenium
 keyboard

setup.py

@@ -9,8 +9,8 @@
     "./version",
     "./translation.json",
 
+    "./assets",
     "./tools",
-    "./source",
 
     sys.exec_prefix + "\\DLLs\\tcl86t.dll",
     sys.exec_prefix + "\\DLLs\\tk86t.dll",
@@ -22,7 +22,6 @@
         "includes": ["tkinter", "requests", "PIL", "distutils"],
         "include_msvcr": True,
         "packages": ["tkinter", "distutils"],
-        "excludes": ["source"],
     }
 }
 
@@ -37,7 +36,7 @@
     license='Apache-2.0',
     author='Faraphel',
     author_email='[email protected]',
-    description='Installateur pour Mario Kart Wii Faraphel.',
+    description='Mario Kart Wii Mod Installer.',
     executables=[
         Executable(
             "./main.pyw",

source/Game.py

@@ -444,6 +444,7 @@ def patch_file(self):
         Prepare all files to install the mod (track, bmg text, descriptive image, ...)
         """
         try:
+            os.makedirs(f"./file/", exist_ok=True)
             os.makedirs(f"{self.common.ct_config.pack_path}/file/Track-WU8/", exist_ok=True)
 
             max_step = (

source/wszst/__init__.py

@@ -1 +1,3 @@
+from source.wszst import bmg, ctc, img, lec, wstrt, szs, wit
+
 __all__ = ["bmg", "ctc", "img", "lec", "wstrt", "szs", "wit"]