Skip to content

Add script to parse tapovac map data #1329

New issue

Have a question about this project? Sign up for a free GitHub account to open an issue and contact its maintainers and the community.

Sign up for GitHub

By clicking “Sign up for GitHub”, you agree to our terms of service and privacy statement. We’ll occasionally send you account related emails.

Already on GitHub? Sign in to your account

Draft
rytilahti wants to merge 1 commit into
base: master
Choose a base branch
from
Draft

Add script to parse tapovac map data #1329

Changes from all commits
Commits
Show all changes
1 commit
Select commit
File filter

Filter by extension

Filter by extension
Conversations
Failed to load comments.
Loading
Jump to
Jump to file
Failed to load files.
Loading
Diff view
Diff view
96 changes: 96 additions & 0 deletions devtools/parse_tapovac_map.py
View file
Open in desktop
Original file line number Diff line number Diff line change
@@ -0,0 +1,96 @@
# /// script
# dependencies = [
# "mashumaro",
# "pillow",
# "lz4",
# ]
# ///

"""Simple tool to parse tapovac maps.

Use either get_image(<response for getMapData command>), or
show_image(<path to json file containing the response data).

This script can be also executed directly:

$ uv run parse_tapovac_map.py my_beautiful_getMapData.json

"""

import hashlib
import json
import sys
from base64 import b64decode
from dataclasses import dataclass, field

import lz4.block
from mashumaro import DataClassDictMixin
from PIL import Image, ImageShow


@dataclass
class MapData(DataClassDictMixin):
"""Class representing getMapData response."""

auto_area_flag: bool
path_id: int
version: str
map_id: int
resolution: int
resolution_unit: str
width: int
height: int
origin_coor: tuple[int, int, int]
real_origin_coor: tuple[int, int, int]
bitnum: int
bit_list: dict
pix_len: int
map_hash: str
pix_lz4len: int
map_data: bytes = field(metadata={"deserialize": b64decode})
area_list: list


def get_image(data) -> Image:
"""Return image object for getMapData response."""
data = data["result"]
d = MapData.from_dict(data)

# TODO: move assert checks to mashumaro checks

assert len(d.map_data) == d.pix_lz4len
assert d.width * d.height == d.pix_len

print(f"{d.resolution=} {d.resolution_unit}")
print(f"{d.width=} x {d.height=} = {d.width * d.height} ({d.pix_len=}) {d.bitnum=}")
print(f"{d.origin_coor=} {d.real_origin_coor=}")
print(f"{d.auto_area_flag=}")
print(f"{d.area_list=}")
print(f"{d.bit_list=}")

# TODO: handle orientation
# TODO: parse arealist for rooms/virtualwalls/other
# TODO: use nicer palette

img_data = lz4.block.decompress(d.map_data, uncompressed_size=d.pix_len)
assert d.map_hash == hashlib.md5(img_data).hexdigest().upper()

assert d.bitnum == 8
mode = "L" # L = 8bit gray

img = Image.frombytes(mode, (d.width, d.height), data=img_data)

return img


def show_image(filename):
"""Load and show image from getMapData json file."""
with open(filename) as f:
data = json.load(f)
img = get_image(data)
final = img.resize((4 * img.width, 4 * img.height))
ImageShow.show(final)


if __name__ == "__main__":
show_image(sys.argv[1])
Loading