675e209f5e
The the presence of a value in the Optional[OpDef] already indicated whether the loop was in an op. The `in_op` variable was decoupling the check for a current op from the usage of the current op, so type checking couldn't determine that None was impossible where it was used.
94 lines
2.5 KiB
Python
94 lines
2.5 KiB
Python
#!/usr/bin/env python3
|
|
from __future__ import annotations
|
|
|
|
from dataclasses import dataclass
|
|
from dataclasses import field
|
|
from typing import List
|
|
from typing import Optional
|
|
|
|
|
|
@dataclass
|
|
class Field:
|
|
name: str
|
|
type: str
|
|
is_array: bool = False
|
|
|
|
|
|
@dataclass
|
|
class OpDef:
|
|
name: str
|
|
base: str
|
|
fields: List[Field] = field(default_factory=list)
|
|
is_terminator: bool = False # @terminator
|
|
is_nothrow: bool = False # @nothrow
|
|
|
|
|
|
def parse_bytecode_def(path: str) -> List[OpDef]:
|
|
with open(path, "r", encoding="utf-8") as f:
|
|
lines = f.readlines()
|
|
|
|
ops: List[OpDef] = []
|
|
current_op: Optional[OpDef] = None
|
|
|
|
for raw_line in lines:
|
|
stripped = raw_line.strip()
|
|
if not stripped:
|
|
continue
|
|
if stripped.startswith("//") or stripped.startswith("#"):
|
|
continue
|
|
|
|
if stripped.startswith("op "):
|
|
if current_op is not None:
|
|
raise RuntimeError("Nested op blocks are not allowed")
|
|
|
|
rest = stripped[len("op ") :].strip()
|
|
if "<" in rest:
|
|
name_part, base_part = rest.split("<", 1)
|
|
name = name_part.strip()
|
|
base = base_part.strip()
|
|
else:
|
|
name = rest.strip()
|
|
base = "Instruction"
|
|
|
|
current_op = OpDef(name=name, base=base)
|
|
continue
|
|
|
|
if stripped == "endop":
|
|
if current_op is None:
|
|
raise RuntimeError("endop without corresponding op")
|
|
ops.append(current_op)
|
|
current_op = None
|
|
continue
|
|
|
|
if current_op is None:
|
|
continue
|
|
|
|
if stripped.startswith("@"):
|
|
if stripped == "@terminator":
|
|
current_op.is_terminator = True
|
|
elif stripped == "@nothrow":
|
|
current_op.is_nothrow = True
|
|
continue
|
|
|
|
if ":" not in stripped:
|
|
raise RuntimeError(f"Malformed field line: {stripped!r}")
|
|
lhs, rhs = stripped.split(":", 1)
|
|
field_name = lhs.strip()
|
|
field_type = rhs.strip()
|
|
is_array = False
|
|
if field_type.endswith("[]"):
|
|
is_array = True
|
|
field_type = field_type[:-2].strip()
|
|
current_op.fields.append(Field(name=field_name, type=field_type, is_array=is_array))
|
|
|
|
if current_op is not None:
|
|
raise RuntimeError("Unclosed op block at end of file")
|
|
|
|
return ops
|
|
|
|
|
|
__all__ = [
|
|
"Field",
|
|
"OpDef",
|
|
"parse_bytecode_def",
|
|
]
|