Function _scan_hand_enums

Scripts/codegen/_codegen_checks.py:41–66

Walk ``public_root`` for every ``.h`` file declaring ``enum class EMj* : uint8 { ... }`` and return ``{enum_name: [members]}``. Members are stripped of inline initialisers, UMETA macros, and comments.

(public_root: str)

def _scan_hand_enums(public_root: str) -> Dict[str, List[str]]:
    """Walk ``public_root`` for every ``.h`` file declaring
    ``enum class EMj* : uint8 { ... }`` and return
    ``{enum_name: [members]}``. Members are stripped of inline
    initialisers, UMETA macros, and comments."""
    out: Dict[str, List[str]] = {}
    for root, _dirs, files in os.walk(public_root):
        for name in files:
            if not name.endswith(".h"):
                continue
            with open(os.path.join(root, name), "r", encoding="utf-8") as f:
                text = f.read()
            for m in _HAND_ENUM_RE.finditer(text):
                enum_name = m.group(1)
                body = m.group(2)
                body = re.sub(r"UMETA\([^)]*\)", "", body)
                body = re.sub(r"//[^\n]*", "", body)
                body = re.sub(r"/\*.*?\*/", "", body, flags=re.DOTALL)
                members: List[str] = []
                for line in body.split(","):
                    mm = _HAND_ENUM_MEMBER_RE.match(line)
                    if mm:
                        members.append(mm.group(1))
                if members:
                    out[enum_name] = members
    return out


def _hand_enums_from_snapshot(mjspec: Optional[Dict[str, Any]]) -> Dict[str, List[str]]: