111 lines
3.2 KiB
Python
111 lines
3.2 KiB
Python
"""
|
|
将多个独立 tile PNG 按 JSON 配置打包为 tile atlas，生成 C++ header。
|
|
|
|
用法:
|
|
python tools/pack_tile_atlas.py config.json output.h
|
|
|
|
JSON 格式:
|
|
{
|
|
"tile_size": 32,
|
|
"columns": 4,
|
|
"tiles": [
|
|
{ "id": 0, "file": "empty.png" },
|
|
{ "id": 1, "file": "ground_top.png" },
|
|
...
|
|
]
|
|
}
|
|
|
|
- tile_size: 每个 tile 的像素尺寸（正方形）
|
|
- columns: atlas 每行放几个 tile
|
|
- tiles: 按 tile_id 排序的列表，id 从 0 开始连续
|
|
- file: 相对于 config.json 所在目录的 PNG 路径
|
|
- 未指定的 tile_id 自动填充透明
|
|
"""
|
|
|
|
import json
|
|
import os
|
|
import sys
|
|
import math
|
|
|
|
from PIL import Image
|
|
|
|
|
|
def pack(config_path: str, output_path: str):
|
|
config_dir = os.path.dirname(os.path.abspath(config_path))
|
|
|
|
with open(config_path, "r", encoding="utf-8") as f:
|
|
cfg = json.load(f)
|
|
|
|
tile_size = cfg["tile_size"]
|
|
columns = cfg["columns"]
|
|
tiles = cfg["tiles"]
|
|
|
|
max_id = max(t["id"] for t in tiles)
|
|
tile_count = max_id + 1
|
|
rows = (tile_count + columns - 1) // columns
|
|
|
|
atlas_w = columns * tile_size
|
|
atlas_h = rows * tile_size
|
|
|
|
atlas = Image.new("RGBA", (atlas_w, atlas_h), (0, 0, 0, 0))
|
|
|
|
tile_map = {t["id"]: t for t in tiles}
|
|
|
|
for tid in range(tile_count):
|
|
col = tid % columns
|
|
row = tid // columns
|
|
x = col * tile_size
|
|
y = row * tile_size
|
|
|
|
if tid in tile_map:
|
|
png_path = os.path.join(config_dir, tile_map[tid]["file"])
|
|
img = Image.open(png_path).convert("RGBA")
|
|
if img.size != (tile_size, tile_size):
|
|
img = img.resize((tile_size, tile_size), Image.NEAREST)
|
|
atlas.paste(img, (x, y))
|
|
|
|
var_name = os.path.splitext(os.path.basename(output_path))[0]
|
|
|
|
pixels = list(atlas.getdata())
|
|
packed = []
|
|
for r, g, b, a in pixels:
|
|
packed.append(
|
|
((r >> 3) << 11)
|
|
| ((g >> 3) << 6)
|
|
| ((b >> 3) << 1)
|
|
| (1 if a else 0)
|
|
)
|
|
|
|
with open(output_path, "w", encoding="utf-8") as f:
|
|
f.write("// Auto-generated by tools/pack_tile_atlas.py\n")
|
|
f.write(f"#pragma once\n#include <cstdint>\n#include \"Image.h\"\n\n")
|
|
f.write(f"static const int32_t {var_name}_width = {atlas_w};\n")
|
|
f.write(f"static const int32_t {var_name}_height = {atlas_h};\n")
|
|
f.write(f"static const int32_t {var_name}_tile_size = {tile_size};\n")
|
|
f.write(f"static const int32_t {var_name}_columns = {columns};\n\n")
|
|
f.write(f"static const uint16_t {var_name}_pixels[] = {{\n")
|
|
|
|
for i in range(0, len(packed), 16):
|
|
chunk = packed[i : i + 16]
|
|
line = ", ".join(f"0x{p:04X}" for p in chunk)
|
|
f.write(f" {line},\n")
|
|
|
|
f.write("};\n\n")
|
|
f.write(f"static const RenderData::Image {var_name}_image({var_name}_pixels, {var_name}_width, {var_name}_height);\n")
|
|
|
|
print(f"已生成: {output_path}")
|
|
print(f" atlas: {atlas_w}x{atlas_h}, {tile_count} tiles ({columns}x{rows})")
|
|
print(f" var: {var_name}")
|
|
|
|
|
|
def main():
|
|
if len(sys.argv) < 3:
|
|
print(f"用法: {sys.argv[0]} config.json output.h")
|
|
sys.exit(1)
|
|
|
|
pack(sys.argv[1], sys.argv[2])
|
|
|
|
|
|
if __name__ == "__main__":
|
|
main()
|