aboutsummaryrefslogtreecommitdiff
path: root/tools/terrain_map_doc_validator.py
diff options
context:
space:
mode:
Diffstat (limited to 'tools/terrain_map_doc_validator.py')
-rw-r--r--tools/terrain_map_doc_validator.py809
1 files changed, 0 insertions, 809 deletions
diff --git a/tools/terrain_map_doc_validator.py b/tools/terrain_map_doc_validator.py
deleted file mode 100644
index 63c3077..0000000
--- a/tools/terrain_map_doc_validator.py
+++ /dev/null
@@ -1,809 +0,0 @@
-#!/usr/bin/env python3
-"""
-Validate terrain/map documentation assumptions against real game data.
-
-Targets:
-- tmp/gamedata/DATA/MAPS/**/Land.msh
-- tmp/gamedata/DATA/MAPS/**/Land.map
-"""
-
-from __future__ import annotations
-
-import argparse
-import json
-import math
-import struct
-from collections import Counter, defaultdict
-from dataclasses import dataclass
-from pathlib import Path
-from typing import Any
-
-import archive_roundtrip_validator as arv
-
-MAGIC_NRES = b"NRes"
-
-REQUIRED_MSH_TYPES = (1, 2, 3, 4, 5, 11, 18, 21)
-OPTIONAL_MSH_TYPES = (14,)
-EXPECTED_MSH_ORDER = (1, 2, 3, 4, 5, 18, 14, 11, 21)
-
-MSH_STRIDES = {
- 1: 38,
- 3: 12,
- 4: 4,
- 5: 4,
- 11: 4,
- 14: 4,
- 18: 4,
- 21: 28,
-}
-
-SLOT_TABLE_OFFSET = 0x8C
-
-
-@dataclass
-class ValidationIssue:
- severity: str # error | warning
- category: str
- resource: str
- message: str
-
-
-class TerrainMapDocValidator:
- def __init__(self) -> None:
- self.issues: list[ValidationIssue] = []
- self.stats: dict[str, Any] = {
- "maps_total": 0,
- "msh_total": 0,
- "map_total": 0,
- "msh_type_orders": Counter(),
- "msh_attr_triplets": defaultdict(Counter), # type_id -> Counter[(a1,a2,a3)]
- "msh_type11_header_words": Counter(),
- "msh_type21_flags_top": Counter(),
- "map_logic_flags": Counter(),
- "map_class_ids": Counter(), # record +40
- "map_poly_count": Counter(),
- "map_vertex_count_min": None,
- "map_vertex_count_max": None,
- "map_cell_dims": Counter(),
- "map_reserved_u12": Counter(),
- "map_reserved_u36": Counter(),
- "map_reserved_u44": Counter(),
- "map_area_delta_abs_max": 0.0,
- "map_area_delta_rel_max": 0.0,
- "map_area_rel_gt_05_count": 0,
- "map_normal_len_min": None,
- "map_normal_len_max": None,
- "map_records_total": 0,
- }
-
- def add_issue(self, severity: str, category: str, resource: Path, message: str) -> None:
- self.issues.append(
- ValidationIssue(
- severity=severity,
- category=category,
- resource=str(resource),
- message=message,
- )
- )
-
- def _entry_payload(self, blob: bytes, entry: dict[str, Any]) -> bytes:
- start = int(entry["data_offset"])
- end = start + int(entry["size"])
- return blob[start:end]
-
- def _entry_by_type(self, entries: list[dict[str, Any]]) -> dict[int, list[dict[str, Any]]]:
- by_type: dict[int, list[dict[str, Any]]] = {}
- for item in entries:
- by_type.setdefault(int(item["type_id"]), []).append(item)
- return by_type
-
- def _expect_single_type(
- self,
- *,
- by_type: dict[int, list[dict[str, Any]]],
- type_id: int,
- label: str,
- resource: Path,
- required: bool,
- ) -> dict[str, Any] | None:
- rows = by_type.get(type_id, [])
- if not rows:
- if required:
- self.add_issue(
- "error",
- "msh-chunk",
- resource,
- f"missing required chunk type={type_id} ({label})",
- )
- return None
- if len(rows) > 1:
- self.add_issue(
- "warning",
- "msh-chunk",
- resource,
- f"multiple chunks type={type_id} ({label}); using first",
- )
- return rows[0]
-
- def _check_stride(
- self,
- *,
- resource: Path,
- entry: dict[str, Any],
- stride: int,
- label: str,
- ) -> int:
- size = int(entry["size"])
- attr1 = int(entry["attr1"])
- attr2 = int(entry["attr2"])
- attr3 = int(entry["attr3"])
- self.stats["msh_attr_triplets"][int(entry["type_id"])][(attr1, attr2, attr3)] += 1
-
- if size % stride != 0:
- self.add_issue(
- "error",
- "msh-stride",
- resource,
- f"{label}: size={size} is not divisible by stride={stride}",
- )
- return -1
-
- count = size // stride
- if attr1 != count:
- self.add_issue(
- "error",
- "msh-attr",
- resource,
- f"{label}: attr1={attr1} != size/stride={count}",
- )
- if attr3 != stride:
- self.add_issue(
- "error",
- "msh-attr",
- resource,
- f"{label}: attr3={attr3} != {stride}",
- )
- if attr2 != 0 and int(entry["type_id"]) not in (1,):
- # type 1 has non-zero attr2 in real assets, others are expected zero.
- self.add_issue(
- "warning",
- "msh-attr",
- resource,
- f"{label}: attr2={attr2} (expected 0 for this chunk type)",
- )
- return count
-
- def validate_msh(self, path: Path) -> None:
- self.stats["msh_total"] += 1
- blob = path.read_bytes()
- if blob[:4] != MAGIC_NRES:
- self.add_issue("error", "msh-container", path, "file is not NRes")
- return
-
- try:
- parsed = arv.parse_nres(blob, source=str(path))
- except Exception as exc: # pylint: disable=broad-except
- self.add_issue("error", "msh-container", path, f"failed to parse NRes: {exc}")
- return
-
- for issue in parsed.get("issues", []):
- self.add_issue("warning", "msh-nres", path, issue)
-
- entries = parsed["entries"]
- types_order = tuple(int(item["type_id"]) for item in entries)
- self.stats["msh_type_orders"][types_order] += 1
- if types_order != EXPECTED_MSH_ORDER:
- self.add_issue(
- "warning",
- "msh-order",
- path,
- f"unexpected chunk order {types_order}, expected {EXPECTED_MSH_ORDER}",
- )
-
- by_type = self._entry_by_type(entries)
-
- chunks: dict[int, dict[str, Any]] = {}
- for type_id in REQUIRED_MSH_TYPES:
- chunk = self._expect_single_type(
- by_type=by_type,
- type_id=type_id,
- label=f"type{type_id}",
- resource=path,
- required=True,
- )
- if chunk:
- chunks[type_id] = chunk
- for type_id in OPTIONAL_MSH_TYPES:
- chunk = self._expect_single_type(
- by_type=by_type,
- type_id=type_id,
- label=f"type{type_id}",
- resource=path,
- required=False,
- )
- if chunk:
- chunks[type_id] = chunk
-
- for type_id, stride in MSH_STRIDES.items():
- chunk = chunks.get(type_id)
- if not chunk:
- continue
- self._check_stride(resource=path, entry=chunk, stride=stride, label=f"type{type_id}")
-
- # type 2 includes 0x8C-byte header + 68-byte slot table entries.
- type2 = chunks.get(2)
- if type2:
- size = int(type2["size"])
- attr1 = int(type2["attr1"])
- attr2 = int(type2["attr2"])
- attr3 = int(type2["attr3"])
- self.stats["msh_attr_triplets"][2][(attr1, attr2, attr3)] += 1
- if attr3 != 68:
- self.add_issue(
- "error",
- "msh-attr",
- path,
- f"type2: attr3={attr3} != 68",
- )
- if attr2 != 0:
- self.add_issue(
- "warning",
- "msh-attr",
- path,
- f"type2: attr2={attr2} (expected 0)",
- )
- if size < SLOT_TABLE_OFFSET:
- self.add_issue(
- "error",
- "msh-size",
- path,
- f"type2: size={size} < header_size={SLOT_TABLE_OFFSET}",
- )
- elif (size - SLOT_TABLE_OFFSET) % 68 != 0:
- self.add_issue(
- "error",
- "msh-size",
- path,
- f"type2: (size - 0x8C) is not divisible by 68 (size={size})",
- )
- else:
- slots_by_size = (size - SLOT_TABLE_OFFSET) // 68
- if attr1 != slots_by_size:
- self.add_issue(
- "error",
- "msh-attr",
- path,
- f"type2: attr1={attr1} != (size-0x8C)/68={slots_by_size}",
- )
-
- verts = chunks.get(3)
- face = chunks.get(21)
- slots = chunks.get(2)
- nodes = chunks.get(1)
- type11 = chunks.get(11)
-
- if verts and face:
- vcount = int(verts["attr1"])
- face_payload = self._entry_payload(blob, face)
- fcount = int(face["attr1"])
- if len(face_payload) >= 28:
- for idx in range(fcount):
- off = idx * 28
- if off + 28 > len(face_payload):
- self.add_issue(
- "error",
- "msh-face",
- path,
- f"type21 truncated at face {idx}",
- )
- break
- flags = struct.unpack_from("<I", face_payload, off)[0]
- self.stats["msh_type21_flags_top"][flags] += 1
- i0, i1, i2 = struct.unpack_from("<HHH", face_payload, off + 8)
- for name, value in (("i0", i0), ("i1", i1), ("i2", i2)):
- if value >= vcount:
- self.add_issue(
- "error",
- "msh-face-index",
- path,
- f"type21[{idx}].{name}={value} out of range vertex_count={vcount}",
- )
- n0, n1, n2 = struct.unpack_from("<HHH", face_payload, off + 14)
- for name, value in (("n0", n0), ("n1", n1), ("n2", n2)):
- if value != 0xFFFF and value >= fcount:
- self.add_issue(
- "error",
- "msh-face-neighbour",
- path,
- f"type21[{idx}].{name}={value} out of range face_count={fcount}",
- )
-
- if slots and face:
- slot_count = int(slots["attr1"])
- face_count = int(face["attr1"])
- slot_payload = self._entry_payload(blob, slots)
- need = SLOT_TABLE_OFFSET + slot_count * 68
- if len(slot_payload) < need:
- self.add_issue(
- "error",
- "msh-slot",
- path,
- f"type2 payload too short: size={len(slot_payload)}, need_at_least={need}",
- )
- else:
- if len(slot_payload) != need:
- self.add_issue(
- "warning",
- "msh-slot",
- path,
- f"type2 payload has trailing bytes: size={len(slot_payload)}, expected={need}",
- )
- for idx in range(slot_count):
- off = SLOT_TABLE_OFFSET + idx * 68
- tri_start, tri_count = struct.unpack_from("<HH", slot_payload, off)
- if tri_start + tri_count > face_count:
- self.add_issue(
- "error",
- "msh-slot-range",
- path,
- f"type2 slot[{idx}] range [{tri_start}, {tri_start + tri_count}) exceeds face_count={face_count}",
- )
-
- if nodes and slots:
- node_payload = self._entry_payload(blob, nodes)
- slot_count = int(slots["attr1"])
- node_count = int(nodes["attr1"])
- for node_idx in range(node_count):
- off = node_idx * 38
- if off + 38 > len(node_payload):
- self.add_issue(
- "error",
- "msh-node",
- path,
- f"type1 truncated at node {node_idx}",
- )
- break
- for j in range(19):
- slot_id = struct.unpack_from("<H", node_payload, off + j * 2)[0]
- if slot_id != 0xFFFF and slot_id >= slot_count:
- self.add_issue(
- "error",
- "msh-node-slot",
- path,
- f"type1 node[{node_idx}] slot[{j}]={slot_id} out of range slot_count={slot_count}",
- )
-
- if type11:
- payload = self._entry_payload(blob, type11)
- if len(payload) >= 8:
- w0, w1 = struct.unpack_from("<II", payload, 0)
- self.stats["msh_type11_header_words"][(w0, w1)] += 1
- else:
- self.add_issue(
- "error",
- "msh-type11",
- path,
- f"type11 payload too short: {len(payload)}",
- )
-
- def _update_minmax(self, key_min: str, key_max: str, value: float) -> None:
- if self.stats[key_min] is None or value < self.stats[key_min]:
- self.stats[key_min] = value
- if self.stats[key_max] is None or value > self.stats[key_max]:
- self.stats[key_max] = value
-
- def validate_map(self, path: Path) -> None:
- self.stats["map_total"] += 1
- blob = path.read_bytes()
- if blob[:4] != MAGIC_NRES:
- self.add_issue("error", "map-container", path, "file is not NRes")
- return
-
- try:
- parsed = arv.parse_nres(blob, source=str(path))
- except Exception as exc: # pylint: disable=broad-except
- self.add_issue("error", "map-container", path, f"failed to parse NRes: {exc}")
- return
-
- for issue in parsed.get("issues", []):
- self.add_issue("warning", "map-nres", path, issue)
-
- entries = parsed["entries"]
- if len(entries) != 1 or int(entries[0]["type_id"]) != 12:
- self.add_issue(
- "error",
- "map-chunk",
- path,
- f"expected single chunk type=12, got {[int(e['type_id']) for e in entries]}",
- )
- return
-
- entry = entries[0]
- areal_count = int(entry["attr1"])
- if areal_count <= 0:
- self.add_issue("error", "map-areal", path, f"invalid areal_count={areal_count}")
- return
-
- payload = self._entry_payload(blob, entry)
- ptr = 0
- records: list[dict[str, Any]] = []
-
- for idx in range(areal_count):
- if ptr + 56 > len(payload):
- self.add_issue(
- "error",
- "map-record",
- path,
- f"truncated areal header at index={idx}, ptr={ptr}, size={len(payload)}",
- )
- return
-
- anchor_x, anchor_y, anchor_z = struct.unpack_from("<fff", payload, ptr)
- u12 = struct.unpack_from("<I", payload, ptr + 12)[0]
- area_f = struct.unpack_from("<f", payload, ptr + 16)[0]
- nx, ny, nz = struct.unpack_from("<fff", payload, ptr + 20)
- logic_flag = struct.unpack_from("<I", payload, ptr + 32)[0]
- u36 = struct.unpack_from("<I", payload, ptr + 36)[0]
- class_id = struct.unpack_from("<I", payload, ptr + 40)[0]
- u44 = struct.unpack_from("<I", payload, ptr + 44)[0]
- vertex_count, poly_count = struct.unpack_from("<II", payload, ptr + 48)
-
- self.stats["map_records_total"] += 1
- self.stats["map_logic_flags"][logic_flag] += 1
- self.stats["map_class_ids"][class_id] += 1
- self.stats["map_poly_count"][poly_count] += 1
- self.stats["map_reserved_u12"][u12] += 1
- self.stats["map_reserved_u36"][u36] += 1
- self.stats["map_reserved_u44"][u44] += 1
- self._update_minmax("map_vertex_count_min", "map_vertex_count_max", float(vertex_count))
-
- normal_len = math.sqrt(nx * nx + ny * ny + nz * nz)
- self._update_minmax("map_normal_len_min", "map_normal_len_max", normal_len)
- if abs(normal_len - 1.0) > 1e-3:
- self.add_issue(
- "warning",
- "map-normal",
- path,
- f"record[{idx}] normal length={normal_len:.6f} (expected ~1.0)",
- )
-
- vertices_off = ptr + 56
- vertices_size = 12 * vertex_count
- if vertices_off + vertices_size > len(payload):
- self.add_issue(
- "error",
- "map-vertices",
- path,
- f"record[{idx}] vertices out of bounds",
- )
- return
-
- vertices: list[tuple[float, float, float]] = []
- for i in range(vertex_count):
- vertices.append(struct.unpack_from("<fff", payload, vertices_off + i * 12))
-
- if vertex_count >= 3:
- # signed shoelace area in XY.
- shoelace = 0.0
- for i in range(vertex_count):
- x1, y1, _ = vertices[i]
- x2, y2, _ = vertices[(i + 1) % vertex_count]
- shoelace += x1 * y2 - x2 * y1
- area_xy = abs(shoelace) * 0.5
- delta = abs(area_xy - area_f)
- if delta > self.stats["map_area_delta_abs_max"]:
- self.stats["map_area_delta_abs_max"] = delta
- rel_delta = delta / max(1.0, area_xy)
- if rel_delta > self.stats["map_area_delta_rel_max"]:
- self.stats["map_area_delta_rel_max"] = rel_delta
- if rel_delta > 0.05:
- self.stats["map_area_rel_gt_05_count"] += 1
-
- links_off = vertices_off + vertices_size
- link_count = vertex_count + 3 * poly_count
- links_size = 8 * link_count
- if links_off + links_size > len(payload):
- self.add_issue(
- "error",
- "map-links",
- path,
- f"record[{idx}] link table out of bounds",
- )
- return
-
- edge_links: list[tuple[int, int]] = []
- for i in range(vertex_count):
- area_ref, edge_ref = struct.unpack_from("<ii", payload, links_off + i * 8)
- edge_links.append((area_ref, edge_ref))
-
- poly_links_off = links_off + 8 * vertex_count
- poly_links: list[tuple[int, int]] = []
- for i in range(3 * poly_count):
- area_ref, edge_ref = struct.unpack_from("<ii", payload, poly_links_off + i * 8)
- poly_links.append((area_ref, edge_ref))
-
- p = links_off + links_size
- for poly_idx in range(poly_count):
- if p + 4 > len(payload):
- self.add_issue(
- "error",
- "map-poly",
- path,
- f"record[{idx}] poly header truncated at poly_idx={poly_idx}",
- )
- return
- n = struct.unpack_from("<I", payload, p)[0]
- poly_size = 4 * (3 * n + 1)
- if p + poly_size > len(payload):
- self.add_issue(
- "error",
- "map-poly",
- path,
- f"record[{idx}] poly data out of bounds at poly_idx={poly_idx}",
- )
- return
- p += poly_size
-
- records.append(
- {
- "index": idx,
- "anchor": (anchor_x, anchor_y, anchor_z),
- "logic": logic_flag,
- "class_id": class_id,
- "vertex_count": vertex_count,
- "poly_count": poly_count,
- "edge_links": edge_links,
- "poly_links": poly_links,
- }
- )
- ptr = p
-
- vertex_counts = [int(item["vertex_count"]) for item in records]
- for rec in records:
- idx = int(rec["index"])
- for link_idx, (area_ref, edge_ref) in enumerate(rec["edge_links"]):
- if area_ref == -1:
- if edge_ref != -1:
- self.add_issue(
- "warning",
- "map-link",
- path,
- f"record[{idx}] edge_link[{link_idx}] has area_ref=-1 but edge_ref={edge_ref}",
- )
- continue
- if area_ref < 0 or area_ref >= areal_count:
- self.add_issue(
- "error",
- "map-link",
- path,
- f"record[{idx}] edge_link[{link_idx}] area_ref={area_ref} out of range",
- )
- continue
- dst_vcount = vertex_counts[area_ref]
- if edge_ref < 0 or edge_ref >= dst_vcount:
- self.add_issue(
- "error",
- "map-link",
- path,
- f"record[{idx}] edge_link[{link_idx}] edge_ref={edge_ref} out of range dst_vertex_count={dst_vcount}",
- )
-
- for link_idx, (area_ref, edge_ref) in enumerate(rec["poly_links"]):
- if area_ref == -1:
- if edge_ref != -1:
- self.add_issue(
- "warning",
- "map-poly-link",
- path,
- f"record[{idx}] poly_link[{link_idx}] has area_ref=-1 but edge_ref={edge_ref}",
- )
- continue
- if area_ref < 0 or area_ref >= areal_count:
- self.add_issue(
- "error",
- "map-poly-link",
- path,
- f"record[{idx}] poly_link[{link_idx}] area_ref={area_ref} out of range",
- )
-
- if ptr + 8 > len(payload):
- self.add_issue(
- "error",
- "map-cells",
- path,
- f"missing cells header at ptr={ptr}, size={len(payload)}",
- )
- return
-
- cells_x, cells_y = struct.unpack_from("<II", payload, ptr)
- self.stats["map_cell_dims"][(cells_x, cells_y)] += 1
- ptr += 8
- if cells_x <= 0 or cells_y <= 0:
- self.add_issue(
- "error",
- "map-cells",
- path,
- f"invalid cells dimensions {cells_x}x{cells_y}",
- )
- return
-
- for x in range(cells_x):
- for y in range(cells_y):
- if ptr + 2 > len(payload):
- self.add_issue(
- "error",
- "map-cells",
- path,
- f"truncated hitCount at cell ({x},{y})",
- )
- return
- hit_count = struct.unpack_from("<H", payload, ptr)[0]
- ptr += 2
- need = 2 * hit_count
- if ptr + need > len(payload):
- self.add_issue(
- "error",
- "map-cells",
- path,
- f"truncated areaIds at cell ({x},{y}), hitCount={hit_count}",
- )
- return
- for i in range(hit_count):
- area_id = struct.unpack_from("<H", payload, ptr + 2 * i)[0]
- if area_id >= areal_count:
- self.add_issue(
- "error",
- "map-cells",
- path,
- f"cell ({x},{y}) has area_id={area_id} out of range areal_count={areal_count}",
- )
- ptr += need
-
- if ptr != len(payload):
- self.add_issue(
- "error",
- "map-size",
- path,
- f"payload tail mismatch: consumed={ptr}, payload_size={len(payload)}",
- )
-
- def validate(self, maps_root: Path) -> None:
- msh_paths = sorted(maps_root.rglob("Land.msh"))
- map_paths = sorted(maps_root.rglob("Land.map"))
-
- msh_by_dir = {path.parent: path for path in msh_paths}
- map_by_dir = {path.parent: path for path in map_paths}
-
- all_dirs = sorted(set(msh_by_dir) | set(map_by_dir))
- self.stats["maps_total"] = len(all_dirs)
-
- for folder in all_dirs:
- msh_path = msh_by_dir.get(folder)
- map_path = map_by_dir.get(folder)
- if msh_path is None:
- self.add_issue("error", "pairing", folder, "missing Land.msh")
- continue
- if map_path is None:
- self.add_issue("error", "pairing", folder, "missing Land.map")
- continue
- self.validate_msh(msh_path)
- self.validate_map(map_path)
-
- def build_report(self) -> dict[str, Any]:
- errors = [i for i in self.issues if i.severity == "error"]
- warnings = [i for i in self.issues if i.severity == "warning"]
-
- # Convert counters/defaultdicts to JSON-friendly dicts.
- msh_orders = {
- str(list(order)): count
- for order, count in self.stats["msh_type_orders"].most_common()
- }
- msh_attrs = {
- str(type_id): {str(list(k)): v for k, v in counter.most_common()}
- for type_id, counter in self.stats["msh_attr_triplets"].items()
- }
- type11_hdr = {
- str(list(key)): value
- for key, value in self.stats["msh_type11_header_words"].most_common()
- }
- type21_flags = {
- f"0x{key:08X}": value
- for key, value in self.stats["msh_type21_flags_top"].most_common(32)
- }
-
- return {
- "summary": {
- "maps_total": self.stats["maps_total"],
- "msh_total": self.stats["msh_total"],
- "map_total": self.stats["map_total"],
- "issues_total": len(self.issues),
- "errors_total": len(errors),
- "warnings_total": len(warnings),
- },
- "stats": {
- "msh_type_orders": msh_orders,
- "msh_attr_triplets": msh_attrs,
- "msh_type11_header_words": type11_hdr,
- "msh_type21_flags_top": type21_flags,
- "map_logic_flags": dict(self.stats["map_logic_flags"]),
- "map_class_ids": dict(self.stats["map_class_ids"]),
- "map_poly_count": dict(self.stats["map_poly_count"]),
- "map_vertex_count_min": self.stats["map_vertex_count_min"],
- "map_vertex_count_max": self.stats["map_vertex_count_max"],
- "map_cell_dims": {str(list(k)): v for k, v in self.stats["map_cell_dims"].items()},
- "map_reserved_u12": dict(self.stats["map_reserved_u12"]),
- "map_reserved_u36": dict(self.stats["map_reserved_u36"]),
- "map_reserved_u44": dict(self.stats["map_reserved_u44"]),
- "map_area_delta_abs_max": self.stats["map_area_delta_abs_max"],
- "map_area_delta_rel_max": self.stats["map_area_delta_rel_max"],
- "map_area_rel_gt_05_count": self.stats["map_area_rel_gt_05_count"],
- "map_normal_len_min": self.stats["map_normal_len_min"],
- "map_normal_len_max": self.stats["map_normal_len_max"],
- "map_records_total": self.stats["map_records_total"],
- },
- "issues": [
- {
- "severity": item.severity,
- "category": item.category,
- "resource": item.resource,
- "message": item.message,
- }
- for item in self.issues
- ],
- }
-
-
-def parse_args() -> argparse.Namespace:
- parser = argparse.ArgumentParser(description="Validate terrain/map doc assumptions")
- parser.add_argument(
- "--maps-root",
- type=Path,
- default=Path("tmp/gamedata/DATA/MAPS"),
- help="Root directory containing MAPS/**/Land.msh and Land.map",
- )
- parser.add_argument(
- "--report-json",
- type=Path,
- default=None,
- help="Optional path to save full JSON report",
- )
- parser.add_argument(
- "--fail-on-warning",
- action="store_true",
- help="Return non-zero exit code on warnings too",
- )
- return parser.parse_args()
-
-
-def main() -> int:
- args = parse_args()
- validator = TerrainMapDocValidator()
- validator.validate(args.maps_root)
- report = validator.build_report()
-
- print(
- json.dumps(
- report["summary"],
- indent=2,
- ensure_ascii=False,
- )
- )
-
- if args.report_json:
- args.report_json.parent.mkdir(parents=True, exist_ok=True)
- with args.report_json.open("w", encoding="utf-8") as handle:
- json.dump(report, handle, indent=2, ensure_ascii=False)
- handle.write("\n")
- print(f"report written: {args.report_json}")
-
- has_errors = report["summary"]["errors_total"] > 0
- has_warnings = report["summary"]["warnings_total"] > 0
- if has_errors:
- return 1
- if args.fail_on_warning and has_warnings:
- return 1
- return 0
-
-
-if __name__ == "__main__":
- raise SystemExit(main())