aviallon/compiler
1
1
Fork 0
Code Issues Pull requests Projects Releases Packages Wiki Activity Actions

main: refactor IR printing

Browse source
This commit is contained in:
Antoine Viallon 2024-01-05 23:31:47 +01:00
parent c29d4cd9ec
commit 5ed4c55522
Signed by: aviallon
GPG key ID: 186FC35EDEB25716
1 changed files with 18 additions and 12 deletions
Download patch file Download diff file Expand all files Collapse all files

30
compiler/__main__.py
View file

@ -53,21 +53,15 @@ def main():
context.check()
intermediate_representation = ast.intermediate_representation()
pseudo_asm = [x.codegen() for x in intermediate_representation]
for i, action in enumerate(intermediate_representation):
messages = []
for ir_item in intermediate_representation:
ir_item.location.source = data
prefix = f"{str(ir_item.location) + ':':<30}"
source_info = ir_item.location.source_substring.splitlines(keepends=False)
messages += [f"# {prefix} {source_info.pop(0)}"]
while len(source_info) > 0:
messages += [f"# {' ' * len(prefix)} {source_info.pop(0)}"]
messages += [f"{repr(ir_item)}\n"]
print("\n".join(messages))
action.location.source = data
print("\n---\n", repr(context))
print_ir(intermediate_representation)
except CompilationError as e:
e.location.source = data
print(f"{e}\n{e.location.show_in_source()}", file=sys.stderr)
@ -76,6 +70,18 @@ def main():
raise e.__cause__
print(f"Caused by:\n{e.__cause__.__class__.__name__}: {e.__cause__}", file=sys.stderr)
def print_ir(intermediate_representation: list[ir.IRAction]):
messages = []
for i, ir_item in enumerate(intermediate_representation):
prefix = f"{str(ir_item.location) + ':':<30}"
source_info = ir_item.location.source_substring.splitlines(keepends=False)
messages += [f"# {prefix} {source_info.pop(0)}"]
while len(source_info) > 0:
messages += [f"# {' ' * len(prefix)} {source_info.pop(0)}"]
messages += [f"{repr(ir_item)}\n"]
print("\n".join(messages))
if __name__ == "__main__":
main()