diff options
Diffstat (limited to 'Tools/scripts/deepfreeze.py')
| -rw-r--r-- | Tools/scripts/deepfreeze.py | 5 |
1 files changed, 5 insertions, 0 deletions
diff --git a/Tools/scripts/deepfreeze.py b/Tools/scripts/deepfreeze.py index a1ef85e..7884939 100644 --- a/Tools/scripts/deepfreeze.py +++ b/Tools/scripts/deepfreeze.py @@ -109,6 +109,7 @@ class Printer: self.cache: Dict[tuple[type, object, str], str] = {} self.hits, self.misses = 0, 0 self.patchups: list[str] = [] + self.deallocs: list[str] = [] self.write('#include "Python.h"') self.write('#include "internal/pycore_gc.h"') self.write('#include "internal/pycore_code.h"') @@ -277,6 +278,7 @@ class Printer: self.write(f".co_varnames = {co_varnames},") self.write(f".co_cellvars = {co_cellvars},") self.write(f".co_freevars = {co_freevars},") + self.deallocs.append(f"_PyStaticCode_Dealloc(&{name}, (_Py_CODEUNIT *) {removesuffix(co_code, '.ob_base.ob_base')}.ob_sval);") return f"& {name}.ob_base" def generate_tuple(self, name: str, t: Tuple[object, ...]) -> str: @@ -440,6 +442,9 @@ def generate(args: list[str], output: TextIO) -> None: else: code = compile(fd.read(), f"<frozen {modname}>", "exec") printer.generate_file(modname, code) + with printer.block(f"void\n_Py_Deepfreeze_Fini(void)"): + for p in printer.deallocs: + printer.write(p) if verbose: print(f"Cache hits: {printer.hits}, misses: {printer.misses}") |