added script to split map image
This commit is contained in:
parent
430f619f2e
commit
9402a7eeb9
82
utils/map_splitter.py
Normal file
82
utils/map_splitter.py
Normal file
@ -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()
|
Loading…
Reference in New Issue
Block a user