Function main

Scripts/codegen/build_mjxmacro_snapshot.py:331–378

()

def main() -> int:
    src = sys.argv[1] if len(sys.argv) > 1 else DEFAULT_MJXMACRO
    dst = sys.argv[2] if len(sys.argv) > 2 else DEFAULT_OUTPUT
    if not os.path.isfile(src):
        print(f"error: mjxmacro.h not found at {src}", file=sys.stderr)
        return 1
    snapshot = parse_mjxmacro(src)
    # Track which source each struct-field block came from so a snapshot
    # reader can tell mjmodel-side (mjxmacro.h) blocks apart from mjspec-side
    # (mjspecmacro.h) blocks without re-parsing the headers.
    block_sources: Dict[str, str] = {}
    for block_name in snapshot["struct_fields"]:
        block_sources[block_name] = snapshot["_meta"]["source"]
    # Also fold in mjspecmacro.h (MJSCOMPILER_FIELDS + MJSPEC_FIELDS) from the
    # installed MuJoCo headers — both live in the same ``struct_fields`` dict so
    # synthetic_categories rules can reference either by block name.
    if os.path.isfile(DEFAULT_MJSPECMACRO):
        spec_snapshot = parse_mjxmacro(DEFAULT_MJSPECMACRO)
        spec_rel = os.path.relpath(DEFAULT_MJSPECMACRO, PLUGIN_ROOT)
        for block_name, entries in spec_snapshot["struct_fields"].items():
            snapshot["struct_fields"][block_name] = entries
            block_sources[block_name] = spec_rel
    else:
        # Loud rather than silent — MJSCOMPILER_FIELDS / MJSPEC_FIELDS blocks
        # WILL be missing if the installed headers predate mjspecmacro.h, and
        # that surfaces only as a downstream KeyError much later in
        # generate_ue_components.py.
        print(
            f"warning: mjspecmacro.h not found at {DEFAULT_MJSPECMACRO}; "
            f"MJSCOMPILER_FIELDS + MJSPEC_FIELDS will be absent from the snapshot. "
            f"Rebuild the MuJoCo install (needs a version that ships mjspecmacro.h).",
            file=sys.stderr,
        )
    snapshot["_meta"]["struct_field_sources"] = block_sources
    # Stats for sanity.
    total_model = sum(len(v) for v in snapshot["mjmodel_pointers"].values())
    total_data = sum(len(v) for v in snapshot["mjdata_pointers"].values())
    total_structs = sum(len(v) for v in snapshot["struct_fields"].values())
    print(
        f"parsed mjxmacro.h: {len(snapshot['mjmodel_pointers'])} model blocks "
        f"({total_model} entries), "
        f"{len(snapshot['mjdata_pointers'])} data blocks ({total_data} entries), "
        f"{len(snapshot['struct_fields'])} struct blocks ({total_structs} entries)"
    )
    with open(dst, "w", encoding="utf-8") as f:
        json.dump(snapshot, f, indent=2)
    print(f"wrote {dst}")
    return 0


if __name__ == "__main__":