From 9402a7eeb9cea93a37834a0fcb76a7aaf67f3683 Mon Sep 17 00:00:00 2001
From: LordBaryhobal <lordbaryhobal@gmail.com>
Date: Fri, 28 Jun 2024 14:20:16 +0200
Subject: [PATCH] added script to split map image

---
 utils/map_splitter.py | 82 +++++++++++++++++++++++++++++++++++++++++++
 1 file changed, 82 insertions(+)
 create mode 100644 utils/map_splitter.py

diff --git a/utils/map_splitter.py b/utils/map_splitter.py
new file mode 100644
index 0000000..11c2686
--- /dev/null
+++ b/utils/map_splitter.py
@@ -0,0 +1,82 @@
+from math import ceil, log10
+import os.path
+
+from PIL import Image
+
+Image.MAX_IMAGE_PIXELS = 200000000
+MAP_SIZE = 1024
+
+
+def clamp(mn, value, mx):
+    return max(mn, min(mx, value))
+
+
+def main():
+    # 2024-06-27_21.01.45_Lycacraft_minecraft~overworld_day.png
+    # 6144,10240
+    input_path = input("Input image: ")
+    output_path = input("Output dir (default: /tmp/lycacraft_maps/): ")
+    if output_path.strip() == "":
+        output_path = "/tmp/lycacraft_maps"
+
+    if not os.path.exists(output_path):
+        os.mkdir(output_path)
+    elif not os.path.isdir(output_path):
+        print("Output path must be a directory")
+        return
+
+    center_pos = input("(0, 0) = ").split(",")
+    cx, cy = list(map(lambda p: int(p), center_pos))
+    image = Image.open(input_path)
+    width, height = image.width, image.height
+
+    left_maps = ceil(cx / MAP_SIZE)
+    right_maps = ceil((width - cx) / MAP_SIZE)
+    top_maps = ceil(cy / MAP_SIZE)
+    bottom_maps = ceil((height - cy) / MAP_SIZE)
+    h_maps = right_maps + left_maps
+    v_maps = bottom_maps + top_maps
+
+    total_maps = h_maps * v_maps
+    digits = ceil(log10(total_maps - 1))
+    map_i_str = f"{{:0{digits}d}}"
+    print(f"{total_maps} maps will be created")
+    proceed = input("Do you want to proceed ? y/[n] ")
+
+    if proceed.lower().strip() != "y":
+        return
+
+    for y in range(v_maps):
+        y0 = y * MAP_SIZE
+        y1 = y0 + MAP_SIZE
+        # y0 = clamp(0, y0, height)
+        # y1 = clamp(0, y1, height)
+
+        start_y = clamp(0, y0, height) - cy
+        end_y = clamp(0, y1, height) - cy
+
+        for x in range(h_maps):
+            x0 = x * MAP_SIZE
+            x1 = x0 + MAP_SIZE
+            # x0 = clamp(0, x0, width)
+            # x1 = clamp(0, x1, width)
+            start_x = clamp(0, x0, width) - cx
+            end_x = clamp(0, x1, width) - cx
+
+            map_i = x + y * h_maps
+            print(f"\r{map_i + 1}/{total_maps} ({map_i / total_maps * 100:.2f}%)", end="")
+            map_image = image.crop((x0, y0, x1, y1))
+            # map_image.save(
+            #     os.path.join(
+            #         output_path,
+            #         f"map{map_i_str.format(map_i)}_{start_x},{start_y}_{end_x},{end_y}.png"
+            #     )
+            # )
+            map_path = os.path.join(output_path, f"map{map_i_str.format(map_i)}_{x - left_maps}_{y - top_maps}.png")
+            map_image.save(map_path)
+
+    print(f"\r{total_maps}/{total_maps} (100%)")
+
+
+if __name__ == '__main__':
+    main()