1
2
3
4
5
6
7
8
9
10
11
12
13
14
15
16
17
18
19
20
21
22
23
24
25
26
27
28
29
30
31
32
33
34
35
36
37
38
39
40
41
42
43
44
45
46
47
48
49
50
51
52
53
54
55
56
57
58
59
60
|
from .parser import parse_header, parse_implementation
from .scanner import scan
from .ssagen import compile_to_ssa
def main():
import argparse
import subprocess
import sys
args = argparse.ArgumentParser(description='The reference compiler for the Crowbar programming language')
args.add_argument('-V', '--version', action='version', version='%(prog)s 0.0.4')
args.add_argument('-g', '--include-debug-info', action='store_true')
args.add_argument('--stop-at-parse-tree', action='store_true')
args.add_argument('--stop-at-qbe-ssa', action='store_true')
args.add_argument('-S', '--stop-at-assembly', action='store_true')
args.add_argument('-o', '--out', help='output file')
args.add_argument('input', help='input file')
args = args.parse_args()
with open(args.input, 'r', encoding='utf-8') as input_file:
input_code = input_file.read()
tokens = scan(input_code)
parse_tree = parse_implementation(tokens)
if args.stop_at_parse_tree:
if args.out is None:
args.out = args.input.replace('.cro', '.cro.txt')
with open(args.out, 'w', encoding='utf-8') as output_file:
output_file.write(str(parse_tree))
return
ssa = compile_to_ssa(parse_tree)
if args.stop_at_qbe_ssa:
if args.out is None:
args.out = args.input.replace('.cro', '.ssa')
with open(args.out, 'w', encoding='utf-8') as output_file:
output_file.write(ssa)
return
# TODO bundle the qbe binary or something
qbe_result = subprocess.run(['qbe', '-'], input=ssa, capture_output=True, text=True)
if qbe_result.returncode != 0:
if args.out is None:
args.out = args.input.replace('.cro', '.s')
print(qbe_result.stderr, file=sys.stderr)
sys.exit(1)
asm = qbe_result.stdout
if args.stop_at_assembly:
if args.out is None:
args.out = args.input.replace('.cro', '.o')
with open(args.out, 'w', encoding='utf-8') as output_file:
output_file.write(asm)
return
if args.out is None:
args.out = args.input.replace('.cro', '.out')
# TODO don't assume gcc is always the right thing
gcc_result = subprocess.run(['gcc', '-x', 'assembler', '-o', args.out, '-'], input=asm, text=True)
sys.exit(gcc_result.returncode)
if __name__ == '__main__':
main()
|
