mirror of https://github.com/python/cpython
96 lines
3.3 KiB
Python
96 lines
3.3 KiB
Python
|
"""Utilities for writing StencilGroups out to a C header file."""
|
||
|
|
import typing
|
||
|
|
|
||
|
|
import _schema
|
||
|
|
import _stencils
|
||
|
|
|
||
|
|
|
||
|
|
def _dump_header() -> typing.Iterator[str]:
|
||
|
|
yield "typedef enum {"
|
||
|
|
for kind in typing.get_args(_schema.HoleKind):
|
||
|
|
yield f" HoleKind_{kind},"
|
||
|
|
yield "} HoleKind;"
|
||
|
|
yield ""
|
||
|
|
yield "typedef enum {"
|
||
|
|
for value in _stencils.HoleValue:
|
||
|
|
yield f" HoleValue_{value.name},"
|
||
|
|
yield "} HoleValue;"
|
||
|
|
yield ""
|
||
|
|
yield "typedef struct {"
|
||
|
|
yield " const uint64_t offset;"
|
||
|
|
yield " const HoleKind kind;"
|
||
|
|
yield " const HoleValue value;"
|
||
|
|
yield " const void *symbol;"
|
||
|
|
yield " const uint64_t addend;"
|
||
|
|
yield "} Hole;"
|
||
|
|
yield ""
|
||
|
|
yield "typedef struct {"
|
||
|
|
yield " const size_t body_size;"
|
||
|
|
yield " const unsigned char * const body;"
|
||
|
|
yield " const size_t holes_size;"
|
||
|
|
yield " const Hole * const holes;"
|
||
|
|
yield "} Stencil;"
|
||
|
|
yield ""
|
||
|
|
yield "typedef struct {"
|
||
|
|
yield " const Stencil code;"
|
||
|
|
yield " const Stencil data;"
|
||
|
|
yield "} StencilGroup;"
|
||
|
|
yield ""
|
||
|
|
|
||
|
|
|
||
|
|
def _dump_footer(opnames: typing.Iterable[str]) -> typing.Iterator[str]:
|
||
|
|
yield "#define INIT_STENCIL(STENCIL) { \\"
|
||
|
|
yield " .body_size = Py_ARRAY_LENGTH(STENCIL##_body) - 1, \\"
|
||
|
|
yield " .body = STENCIL##_body, \\"
|
||
|
|
yield " .holes_size = Py_ARRAY_LENGTH(STENCIL##_holes) - 1, \\"
|
||
|
|
yield " .holes = STENCIL##_holes, \\"
|
||
|
|
yield "}"
|
||
|
|
yield ""
|
||
|
|
yield "#define INIT_STENCIL_GROUP(OP) { \\"
|
||
|
|
yield " .code = INIT_STENCIL(OP##_code), \\"
|
||
|
|
yield " .data = INIT_STENCIL(OP##_data), \\"
|
||
|
|
yield "}"
|
||
|
|
yield ""
|
||
|
|
yield "static const StencilGroup stencil_groups[512] = {"
|
||
|
|
for opname in opnames:
|
||
|
|
yield f" [{opname}] = INIT_STENCIL_GROUP({opname}),"
|
||
|
|
yield "};"
|
||
|
|
yield ""
|
||
|
|
yield "#define GET_PATCHES() { \\"
|
||
|
|
for value in _stencils.HoleValue:
|
||
|
|
yield f" [HoleValue_{value.name}] = (uint64_t)0xBADBADBADBADBADB, \\"
|
||
|
|
yield "}"
|
||
|
|
|
||
|
|
|
||
|
|
def _dump_stencil(opname: str, group: _stencils.StencilGroup) -> typing.Iterator[str]:
|
||
|
|
yield f"// {opname}"
|
||
|
|
for part, stencil in [("code", group.code), ("data", group.data)]:
|
||
|
|
for line in stencil.disassembly:
|
||
|
|
yield f"// {line}"
|
||
|
|
if stencil.body:
|
||
|
|
size = len(stencil.body) + 1
|
||
|
|
yield f"static const unsigned char {opname}_{part}_body[{size}] = {{"
|
||
|
|
for i in range(0, len(stencil.body), 8):
|
||
|
|
row = " ".join(f"{byte:#04x}," for byte in stencil.body[i : i + 8])
|
||
|
|
yield f" {row}"
|
||
|
|
yield "};"
|
||
|
|
else:
|
||
|
|
yield f"static const unsigned char {opname}_{part}_body[1];"
|
||
|
|
if stencil.holes:
|
||
|
|
size = len(stencil.holes) + 1
|
||
|
|
yield f"static const Hole {opname}_{part}_holes[{size}] = {{"
|
||
|
|
for hole in stencil.holes:
|
||
|
|
yield f" {hole.as_c()},"
|
||
|
|
yield "};"
|
||
|
|
else:
|
||
|
|
yield f"static const Hole {opname}_{part}_holes[1];"
|
||
|
|
yield ""
|
||
|
|
|
||
|
|
|
||
|
|
def dump(groups: dict[str, _stencils.StencilGroup]) -> typing.Iterator[str]:
|
||
|
|
"""Yield a JIT compiler line-by-line as a C header file."""
|
||
|
|
yield from _dump_header()
|
||
|
|
for opname, group in groups.items():
|
||
|
|
yield from _dump_stencil(opname, group)
|
||
|
|
yield from _dump_footer(groups)
|