| 0 | 0 |
# nascent! embryonic! inchoate! alpha! wip!
|
|
1 |
from castile.types import *
|
|
2 |
from castile.transformer import VarDeclTypeAssigner
|
| 1 | 3 |
|
| 2 | 4 |
OPS = {
|
| 3 | 5 |
}
|
|
| 6 | 8 |
class Compiler(object):
|
| 7 | 9 |
def __init__(self, out):
|
| 8 | 10 |
self.out = out
|
|
11 |
self.main_type = None
|
| 9 | 12 |
|
| 10 | 13 |
def commas(self, asts, sep=','):
|
| 11 | 14 |
if asts:
|
|
| 15 | 18 |
self.compile(asts[-1])
|
| 16 | 19 |
|
| 17 | 20 |
def c_type(self, type):
|
| 18 | |
return 'int'
|
|
21 |
if type == Integer():
|
|
22 |
return 'int'
|
|
23 |
elif type == String():
|
|
24 |
return 'char *'
|
|
25 |
elif type == Void():
|
|
26 |
return 'void'
|
|
27 |
elif isinstance(type, Struct):
|
|
28 |
return 'struct %s *' % type.name
|
|
29 |
elif isinstance(type, Function):
|
|
30 |
# oh dear
|
|
31 |
return 'void *'
|
|
32 |
elif isinstance(type, Union):
|
|
33 |
# oh dear
|
|
34 |
return 'void *'
|
|
35 |
else:
|
|
36 |
raise NotImplementedError(type)
|
| 19 | 37 |
|
| 20 | 38 |
def compile(self, ast):
|
| 21 | 39 |
if ast.tag == 'Program':
|
|
40 |
g = VarDeclTypeAssigner()
|
|
41 |
g.assign_types(ast)
|
| 22 | 42 |
self.out.write(r"""
|
| 23 | 43 |
/* AUTOMATICALLY GENERATED -- EDIT AT OWN RISK */
|
| 24 | 44 |
|
| 25 | 45 |
#include <stdio.h>
|
|
46 |
#include <stdlib.h>
|
| 26 | 47 |
|
| 27 | 48 |
void print(char *s)
|
| 28 | 49 |
{
|
|
| 32 | 53 |
""")
|
| 33 | 54 |
for child in ast.children:
|
| 34 | 55 |
self.compile(child)
|
| 35 | |
self.out.write(r"""
|
|
56 |
if self.main_type == Void():
|
|
57 |
self.out.write(r"""
|
|
58 |
|
|
59 |
int main(int argc, char **argv)
|
|
60 |
{
|
|
61 |
castile_main();
|
|
62 |
return 0;
|
|
63 |
}
|
|
64 |
|
|
65 |
""")
|
|
66 |
if self.main_type == Integer():
|
|
67 |
self.out.write(r"""
|
| 36 | 68 |
|
| 37 | 69 |
int main(int argc, char **argv)
|
| 38 | 70 |
{
|
|
| 47 | 79 |
name = ast.value
|
| 48 | 80 |
if name == 'main':
|
| 49 | 81 |
name = 'castile_main'
|
|
82 |
self.main_type = thing.type.return_type
|
| 50 | 83 |
if thing.tag == 'FunLit':
|
| 51 | 84 |
self.out.write('%s %s' % (self.c_type(thing.type.return_type), name))
|
| 52 | 85 |
self.compile(ast.children[0])
|
|
| 57 | 90 |
elif ast.tag == 'Forward':
|
| 58 | 91 |
self.out.write('extern %s;\n' % ast.value)
|
| 59 | 92 |
elif ast.tag == 'StructDefn':
|
| 60 | |
self.out.write('struct %s {}\n' % ast.value)
|
|
93 |
self.out.write('struct %s {' % ast.value)
|
|
94 |
for child in ast.children:
|
|
95 |
self.compile(child)
|
|
96 |
self.out.write('};\n')
|
|
97 |
elif ast.tag == 'FieldDefn':
|
|
98 |
self.out.write('%s %s;\n' % (self.c_type(ast.children[0].type), ast.value))
|
| 61 | 99 |
elif ast.tag == 'FunLit':
|
| 62 | 100 |
self.out.write('(')
|
| 63 | 101 |
self.compile(ast.children[0])
|
|
| 135 | 173 |
self.out.write(' = ')
|
| 136 | 174 |
self.compile(ast.children[1])
|
| 137 | 175 |
elif ast.tag == 'Make':
|
| 138 | |
self.out.write('malloc()<--{')
|
|
176 |
self.out.write('&(struct %s){ ' % ast.type.name)
|
| 139 | 177 |
self.commas(ast.children[1:])
|
| 140 | |
self.out.write('}')
|
|
178 |
self.out.write(' }')
|
| 141 | 179 |
elif ast.tag == 'FieldInit':
|
| 142 | |
self.out.write("'%s'," % ast.value)
|
| 143 | |
self.compile(ast.children[0])
|
|
180 |
self.commas(ast.children)
|
| 144 | 181 |
elif ast.tag == 'Index':
|
| 145 | 182 |
self.compile(ast.children[0])
|
| 146 | |
self.out.write('["%s"]' % ast.value)
|
|
183 |
self.out.write('->%s' % ast.value)
|
| 147 | 184 |
elif ast.tag == 'TypeCast':
|
| 148 | |
self.out.write("['%s'," % str(ast.children[0].type))
|
| 149 | |
self.compile(ast.children[0])
|
| 150 | |
self.out.write(']')
|
|
185 |
self.out.write('tag("%s",' % str(ast.children[0].type))
|
|
186 |
self.compile(ast.children[0])
|
|
187 |
self.out.write(')')
|
| 151 | 188 |
elif ast.tag == 'TypeCase':
|
| 152 | 189 |
self.out.write('if (')
|
| 153 | 190 |
self.compile(ast.children[0])
|