annotate python/ppci/c3/codegenerator.py @ 308:2e7f55319858

Merged analyse into codegenerator
author Windel Bouwman
date Fri, 13 Dec 2013 11:53:29 +0100
parents e609d5296ee9
children ff665880a6b0
rev   line source
255
7416c923a02a Added more logging
Windel Bouwman
parents: 252
diff changeset
1 import logging
301
6753763d3bec merge codegen into ppci package
Windel Bouwman
parents: 300
diff changeset
2 from .. import ir
307
e609d5296ee9 Massive rewrite of codegenerator
Windel Bouwman
parents: 306
diff changeset
3 from .. import irutils
308
2e7f55319858 Merged analyse into codegenerator
Windel Bouwman
parents: 307
diff changeset
4 from . import astnodes as ast
230
88a1e0baef65 Added some tests for IR-code
Windel Bouwman
parents: 228
diff changeset
5
228
7f18ed9b6b7e Removal of emptystatement class
Windel Bouwman
parents: 227
diff changeset
6
307
e609d5296ee9 Massive rewrite of codegenerator
Windel Bouwman
parents: 306
diff changeset
7 class SemanticError(Exception):
308
2e7f55319858 Merged analyse into codegenerator
Windel Bouwman
parents: 307
diff changeset
8 """ Error thrown when a semantic issue is observed """
307
e609d5296ee9 Massive rewrite of codegenerator
Windel Bouwman
parents: 306
diff changeset
9 def __init__(self, msg, loc):
308
2e7f55319858 Merged analyse into codegenerator
Windel Bouwman
parents: 307
diff changeset
10 super().__init__()
307
e609d5296ee9 Massive rewrite of codegenerator
Windel Bouwman
parents: 306
diff changeset
11 self.msg = msg
e609d5296ee9 Massive rewrite of codegenerator
Windel Bouwman
parents: 306
diff changeset
12 self.loc = loc
e609d5296ee9 Massive rewrite of codegenerator
Windel Bouwman
parents: 306
diff changeset
13
e609d5296ee9 Massive rewrite of codegenerator
Windel Bouwman
parents: 306
diff changeset
14
e609d5296ee9 Massive rewrite of codegenerator
Windel Bouwman
parents: 306
diff changeset
15 class CodeGenerator(irutils.Builder):
288
a747a45dcd78 Various styling work
Windel Bouwman
parents: 287
diff changeset
16 """
274
ea93e0a7a31e Move docs
Windel Bouwman
parents: 272
diff changeset
17 Generates intermediate (IR) code from a package. The entry function is
ea93e0a7a31e Move docs
Windel Bouwman
parents: 272
diff changeset
18 'genModule'. The main task of this part is to rewrite complex control
ea93e0a7a31e Move docs
Windel Bouwman
parents: 272
diff changeset
19 structures, such as while and for loops into simple conditional
ea93e0a7a31e Move docs
Windel Bouwman
parents: 272
diff changeset
20 jump statements. Also complex conditional statements are simplified.
ea93e0a7a31e Move docs
Windel Bouwman
parents: 272
diff changeset
21 Such as 'and' and 'or' statements are rewritten in conditional jumps.
ea93e0a7a31e Move docs
Windel Bouwman
parents: 272
diff changeset
22 And structured datatypes are rewritten.
307
e609d5296ee9 Massive rewrite of codegenerator
Windel Bouwman
parents: 306
diff changeset
23
e609d5296ee9 Massive rewrite of codegenerator
Windel Bouwman
parents: 306
diff changeset
24 Type checking is done in one run with code generation.
274
ea93e0a7a31e Move docs
Windel Bouwman
parents: 272
diff changeset
25 """
307
e609d5296ee9 Massive rewrite of codegenerator
Windel Bouwman
parents: 306
diff changeset
26 def __init__(self, diag):
261
444b9df2ed99 try to split up code generation
Windel Bouwman
parents: 259
diff changeset
27 self.logger = logging.getLogger('c3cgen')
307
e609d5296ee9 Massive rewrite of codegenerator
Windel Bouwman
parents: 306
diff changeset
28 self.diag = diag
261
444b9df2ed99 try to split up code generation
Windel Bouwman
parents: 259
diff changeset
29
217
8b2e5f3cd579 Removed some stale python source files
Windel Bouwman
parents: 205
diff changeset
30 def gencode(self, pkg):
308
2e7f55319858 Merged analyse into codegenerator
Windel Bouwman
parents: 307
diff changeset
31 """ Generate code for a single module """
268
5ec7580976d9 Op naar tree-IR
Windel Bouwman
parents: 261
diff changeset
32 self.prepare()
308
2e7f55319858 Merged analyse into codegenerator
Windel Bouwman
parents: 307
diff changeset
33 assert type(pkg) is ast.Package
307
e609d5296ee9 Massive rewrite of codegenerator
Windel Bouwman
parents: 306
diff changeset
34 self.pkg = pkg
e609d5296ee9 Massive rewrite of codegenerator
Windel Bouwman
parents: 306
diff changeset
35 self.intType = pkg.scope['int']
e609d5296ee9 Massive rewrite of codegenerator
Windel Bouwman
parents: 306
diff changeset
36 self.boolType = pkg.scope['bool']
255
7416c923a02a Added more logging
Windel Bouwman
parents: 252
diff changeset
37 self.logger.info('Generating ir-code for {}'.format(pkg.name))
288
a747a45dcd78 Various styling work
Windel Bouwman
parents: 287
diff changeset
38 self.varMap = {} # Maps variables to storage locations.
217
8b2e5f3cd579 Removed some stale python source files
Windel Bouwman
parents: 205
diff changeset
39 self.funcMap = {}
268
5ec7580976d9 Op naar tree-IR
Windel Bouwman
parents: 261
diff changeset
40 self.m = ir.Module(pkg.name)
307
e609d5296ee9 Massive rewrite of codegenerator
Windel Bouwman
parents: 306
diff changeset
41 try:
e609d5296ee9 Massive rewrite of codegenerator
Windel Bouwman
parents: 306
diff changeset
42 for s in pkg.innerScope.Functions:
e609d5296ee9 Massive rewrite of codegenerator
Windel Bouwman
parents: 306
diff changeset
43 f = self.newFunction(s.name)
e609d5296ee9 Massive rewrite of codegenerator
Windel Bouwman
parents: 306
diff changeset
44 self.funcMap[s] = f
e609d5296ee9 Massive rewrite of codegenerator
Windel Bouwman
parents: 306
diff changeset
45 for v in pkg.innerScope.Variables:
e609d5296ee9 Massive rewrite of codegenerator
Windel Bouwman
parents: 306
diff changeset
46 self.varMap[v] = self.newTemp()
e609d5296ee9 Massive rewrite of codegenerator
Windel Bouwman
parents: 306
diff changeset
47 for s in pkg.innerScope.Functions:
308
2e7f55319858 Merged analyse into codegenerator
Windel Bouwman
parents: 307
diff changeset
48 self.gen_function(s)
307
e609d5296ee9 Massive rewrite of codegenerator
Windel Bouwman
parents: 306
diff changeset
49 except SemanticError as e:
e609d5296ee9 Massive rewrite of codegenerator
Windel Bouwman
parents: 306
diff changeset
50 self.error(e.msg, e.loc)
308
2e7f55319858 Merged analyse into codegenerator
Windel Bouwman
parents: 307
diff changeset
51 return self.m
2e7f55319858 Merged analyse into codegenerator
Windel Bouwman
parents: 307
diff changeset
52
2e7f55319858 Merged analyse into codegenerator
Windel Bouwman
parents: 307
diff changeset
53 def error(self, msg, loc=None):
2e7f55319858 Merged analyse into codegenerator
Windel Bouwman
parents: 307
diff changeset
54 self.pkg.ok = False
2e7f55319858 Merged analyse into codegenerator
Windel Bouwman
parents: 307
diff changeset
55 self.diag.error(msg, loc)
307
e609d5296ee9 Massive rewrite of codegenerator
Windel Bouwman
parents: 306
diff changeset
56
e609d5296ee9 Massive rewrite of codegenerator
Windel Bouwman
parents: 306
diff changeset
57 def checkType(self, t):
e609d5296ee9 Massive rewrite of codegenerator
Windel Bouwman
parents: 306
diff changeset
58 """ Verify the type is correct """
e609d5296ee9 Massive rewrite of codegenerator
Windel Bouwman
parents: 306
diff changeset
59 t = self.theType(t)
174
3eb06f5fb987 Added memory alloc for locals
Windel Bouwman
parents: 172
diff changeset
60
308
2e7f55319858 Merged analyse into codegenerator
Windel Bouwman
parents: 307
diff changeset
61 def gen_function(self, fn):
268
5ec7580976d9 Op naar tree-IR
Windel Bouwman
parents: 261
diff changeset
62 # TODO: handle arguments
5ec7580976d9 Op naar tree-IR
Windel Bouwman
parents: 261
diff changeset
63 f = self.funcMap[fn]
272
e64bae57cda8 refactor ir
Windel Bouwman
parents: 269
diff changeset
64 f.return_value = self.newTemp()
268
5ec7580976d9 Op naar tree-IR
Windel Bouwman
parents: 261
diff changeset
65 self.setFunction(f)
269
5f8c04a8d26b Towards better modularity
Windel Bouwman
parents: 268
diff changeset
66 l2 = self.newBlock()
5f8c04a8d26b Towards better modularity
Windel Bouwman
parents: 268
diff changeset
67 self.emit(ir.Jump(l2))
5f8c04a8d26b Towards better modularity
Windel Bouwman
parents: 268
diff changeset
68 self.setBlock(l2)
268
5ec7580976d9 Op naar tree-IR
Windel Bouwman
parents: 261
diff changeset
69 # generate room for locals:
174
3eb06f5fb987 Added memory alloc for locals
Windel Bouwman
parents: 172
diff changeset
70
268
5ec7580976d9 Op naar tree-IR
Windel Bouwman
parents: 261
diff changeset
71 for sym in fn.innerScope:
5ec7580976d9 Op naar tree-IR
Windel Bouwman
parents: 261
diff changeset
72 # TODO: handle parameters different
272
e64bae57cda8 refactor ir
Windel Bouwman
parents: 269
diff changeset
73 if sym.isParameter:
307
e609d5296ee9 Massive rewrite of codegenerator
Windel Bouwman
parents: 306
diff changeset
74 self.checkType(sym.typ)
308
2e7f55319858 Merged analyse into codegenerator
Windel Bouwman
parents: 307
diff changeset
75 variable = ir.Parameter(sym.name)
2e7f55319858 Merged analyse into codegenerator
Windel Bouwman
parents: 307
diff changeset
76 f.addParameter(variable)
274
ea93e0a7a31e Move docs
Windel Bouwman
parents: 272
diff changeset
77 elif sym.isLocal:
307
e609d5296ee9 Massive rewrite of codegenerator
Windel Bouwman
parents: 306
diff changeset
78 self.checkType(sym.typ)
308
2e7f55319858 Merged analyse into codegenerator
Windel Bouwman
parents: 307
diff changeset
79 variable = ir.LocalVariable(sym.name)
2e7f55319858 Merged analyse into codegenerator
Windel Bouwman
parents: 307
diff changeset
80 f.addLocal(variable)
2e7f55319858 Merged analyse into codegenerator
Windel Bouwman
parents: 307
diff changeset
81 elif isinstance(sym, ast.Variable):
307
e609d5296ee9 Massive rewrite of codegenerator
Windel Bouwman
parents: 306
diff changeset
82 self.checkType(sym.typ)
308
2e7f55319858 Merged analyse into codegenerator
Windel Bouwman
parents: 307
diff changeset
83 variable = ir.LocalVariable(sym.name)
2e7f55319858 Merged analyse into codegenerator
Windel Bouwman
parents: 307
diff changeset
84 f.addLocal(variable)
274
ea93e0a7a31e Move docs
Windel Bouwman
parents: 272
diff changeset
85 else:
275
6f2423df0675 Fixed serve arm-as
Windel Bouwman
parents: 274
diff changeset
86 raise NotImplementedError('{}'.format(sym))
308
2e7f55319858 Merged analyse into codegenerator
Windel Bouwman
parents: 307
diff changeset
87 self.varMap[sym] = variable
268
5ec7580976d9 Op naar tree-IR
Windel Bouwman
parents: 261
diff changeset
88
5ec7580976d9 Op naar tree-IR
Windel Bouwman
parents: 261
diff changeset
89 self.genCode(fn.body)
272
e64bae57cda8 refactor ir
Windel Bouwman
parents: 269
diff changeset
90 self.emit(ir.Move(f.return_value, ir.Const(0)))
269
5f8c04a8d26b Towards better modularity
Windel Bouwman
parents: 268
diff changeset
91 self.emit(ir.Jump(f.epiloog))
268
5ec7580976d9 Op naar tree-IR
Windel Bouwman
parents: 261
diff changeset
92 self.setFunction(None)
158
9683a4cd848f Added some functions for code generation
Windel Bouwman
parents: 157
diff changeset
93
217
8b2e5f3cd579 Removed some stale python source files
Windel Bouwman
parents: 205
diff changeset
94 def genCode(self, code):
308
2e7f55319858 Merged analyse into codegenerator
Windel Bouwman
parents: 307
diff changeset
95 """ Wrapper around gen_stmt to catch errors """
307
e609d5296ee9 Massive rewrite of codegenerator
Windel Bouwman
parents: 306
diff changeset
96 try:
308
2e7f55319858 Merged analyse into codegenerator
Windel Bouwman
parents: 307
diff changeset
97 self.gen_stmt(code)
307
e609d5296ee9 Massive rewrite of codegenerator
Windel Bouwman
parents: 306
diff changeset
98 except SemanticError as e:
e609d5296ee9 Massive rewrite of codegenerator
Windel Bouwman
parents: 306
diff changeset
99 self.error(e.msg, e.loc)
e609d5296ee9 Massive rewrite of codegenerator
Windel Bouwman
parents: 306
diff changeset
100
308
2e7f55319858 Merged analyse into codegenerator
Windel Bouwman
parents: 307
diff changeset
101 def gen_stmt(self, code):
2e7f55319858 Merged analyse into codegenerator
Windel Bouwman
parents: 307
diff changeset
102 """ Generate code for a statement """
2e7f55319858 Merged analyse into codegenerator
Windel Bouwman
parents: 307
diff changeset
103 assert isinstance(code, ast.Statement)
268
5ec7580976d9 Op naar tree-IR
Windel Bouwman
parents: 261
diff changeset
104 self.setLoc(code.loc)
308
2e7f55319858 Merged analyse into codegenerator
Windel Bouwman
parents: 307
diff changeset
105 if type(code) is ast.Compound:
221
848c4b15fd0b pointers
Windel Bouwman
parents: 220
diff changeset
106 for s in code.statements:
848c4b15fd0b pointers
Windel Bouwman
parents: 220
diff changeset
107 self.genCode(s)
308
2e7f55319858 Merged analyse into codegenerator
Windel Bouwman
parents: 307
diff changeset
108 elif type(code) is ast.Empty:
306
b145f8e6050b Start on c3 rewrite
Windel Bouwman
parents: 305
diff changeset
109 pass
308
2e7f55319858 Merged analyse into codegenerator
Windel Bouwman
parents: 307
diff changeset
110 elif type(code) is ast.Assignment:
268
5ec7580976d9 Op naar tree-IR
Windel Bouwman
parents: 261
diff changeset
111 lval = self.genExprCode(code.lval)
307
e609d5296ee9 Massive rewrite of codegenerator
Windel Bouwman
parents: 306
diff changeset
112 rval = self.genExprCode(code.rval)
e609d5296ee9 Massive rewrite of codegenerator
Windel Bouwman
parents: 306
diff changeset
113 if not self.equalTypes(code.lval.typ, code.rval.typ):
e609d5296ee9 Massive rewrite of codegenerator
Windel Bouwman
parents: 306
diff changeset
114 msg = 'Cannot assign {} to {}'.format(code.lval.typ, code.rval.typ)
e609d5296ee9 Massive rewrite of codegenerator
Windel Bouwman
parents: 306
diff changeset
115 raise SemanticError(msg, code.loc)
e609d5296ee9 Massive rewrite of codegenerator
Windel Bouwman
parents: 306
diff changeset
116 if not code.lval.lvalue:
e609d5296ee9 Massive rewrite of codegenerator
Windel Bouwman
parents: 306
diff changeset
117 raise SemanticError('No valid lvalue {}'.format(code.lval), code.lval.loc)
268
5ec7580976d9 Op naar tree-IR
Windel Bouwman
parents: 261
diff changeset
118 self.emit(ir.Move(lval, rval))
308
2e7f55319858 Merged analyse into codegenerator
Windel Bouwman
parents: 307
diff changeset
119 elif type(code) is ast.ExpressionStatement:
275
6f2423df0675 Fixed serve arm-as
Windel Bouwman
parents: 274
diff changeset
120 self.emit(ir.Exp(self.genExprCode(code.ex)))
308
2e7f55319858 Merged analyse into codegenerator
Windel Bouwman
parents: 307
diff changeset
121 elif type(code) is ast.If:
268
5ec7580976d9 Op naar tree-IR
Windel Bouwman
parents: 261
diff changeset
122 bbtrue = self.newBlock()
5ec7580976d9 Op naar tree-IR
Windel Bouwman
parents: 261
diff changeset
123 bbfalse = self.newBlock()
5ec7580976d9 Op naar tree-IR
Windel Bouwman
parents: 261
diff changeset
124 te = self.newBlock()
308
2e7f55319858 Merged analyse into codegenerator
Windel Bouwman
parents: 307
diff changeset
125 self.gen_cond_code(code.condition, bbtrue, bbfalse)
268
5ec7580976d9 Op naar tree-IR
Windel Bouwman
parents: 261
diff changeset
126 self.setBlock(bbtrue)
5ec7580976d9 Op naar tree-IR
Windel Bouwman
parents: 261
diff changeset
127 self.genCode(code.truestatement)
5ec7580976d9 Op naar tree-IR
Windel Bouwman
parents: 261
diff changeset
128 self.emit(ir.Jump(te))
5ec7580976d9 Op naar tree-IR
Windel Bouwman
parents: 261
diff changeset
129 self.setBlock(bbfalse)
306
b145f8e6050b Start on c3 rewrite
Windel Bouwman
parents: 305
diff changeset
130 self.genCode(code.falsestatement)
268
5ec7580976d9 Op naar tree-IR
Windel Bouwman
parents: 261
diff changeset
131 self.emit(ir.Jump(te))
5ec7580976d9 Op naar tree-IR
Windel Bouwman
parents: 261
diff changeset
132 self.setBlock(te)
308
2e7f55319858 Merged analyse into codegenerator
Windel Bouwman
parents: 307
diff changeset
133 elif type(code) is ast.Return:
303
be7f60545368 Final fixups
Windel Bouwman
parents: 301
diff changeset
134 re = self.genExprCode(code.expr)
be7f60545368 Final fixups
Windel Bouwman
parents: 301
diff changeset
135 self.emit(ir.Move(self.fn.return_value, re))
be7f60545368 Final fixups
Windel Bouwman
parents: 301
diff changeset
136 self.emit(ir.Jump(self.fn.epiloog))
be7f60545368 Final fixups
Windel Bouwman
parents: 301
diff changeset
137 b = self.newBlock()
be7f60545368 Final fixups
Windel Bouwman
parents: 301
diff changeset
138 self.setBlock(b)
308
2e7f55319858 Merged analyse into codegenerator
Windel Bouwman
parents: 307
diff changeset
139 elif type(code) is ast.While:
268
5ec7580976d9 Op naar tree-IR
Windel Bouwman
parents: 261
diff changeset
140 bbdo = self.newBlock()
5ec7580976d9 Op naar tree-IR
Windel Bouwman
parents: 261
diff changeset
141 bbtest = self.newBlock()
5ec7580976d9 Op naar tree-IR
Windel Bouwman
parents: 261
diff changeset
142 te = self.newBlock()
5ec7580976d9 Op naar tree-IR
Windel Bouwman
parents: 261
diff changeset
143 self.emit(ir.Jump(bbtest))
5ec7580976d9 Op naar tree-IR
Windel Bouwman
parents: 261
diff changeset
144 self.setBlock(bbtest)
308
2e7f55319858 Merged analyse into codegenerator
Windel Bouwman
parents: 307
diff changeset
145 self.gen_cond_code(code.condition, bbdo, te)
268
5ec7580976d9 Op naar tree-IR
Windel Bouwman
parents: 261
diff changeset
146 self.setBlock(bbdo)
228
7f18ed9b6b7e Removal of emptystatement class
Windel Bouwman
parents: 227
diff changeset
147 self.genCode(code.statement)
268
5ec7580976d9 Op naar tree-IR
Windel Bouwman
parents: 261
diff changeset
148 self.emit(ir.Jump(bbtest))
5ec7580976d9 Op naar tree-IR
Windel Bouwman
parents: 261
diff changeset
149 self.setBlock(te)
222
c3f1ce8b638f Fixup of parser
Windel Bouwman
parents: 221
diff changeset
150 else:
268
5ec7580976d9 Op naar tree-IR
Windel Bouwman
parents: 261
diff changeset
151 raise NotImplementedError('Unknown stmt {}'.format(code))
230
88a1e0baef65 Added some tests for IR-code
Windel Bouwman
parents: 228
diff changeset
152
308
2e7f55319858 Merged analyse into codegenerator
Windel Bouwman
parents: 307
diff changeset
153 def gen_cond_code(self, expr, bbtrue, bbfalse):
2e7f55319858 Merged analyse into codegenerator
Windel Bouwman
parents: 307
diff changeset
154 """ Generate conditional logic.
2e7f55319858 Merged analyse into codegenerator
Windel Bouwman
parents: 307
diff changeset
155 Implement sequential logical operators. """
2e7f55319858 Merged analyse into codegenerator
Windel Bouwman
parents: 307
diff changeset
156 if type(expr) is ast.Binop:
268
5ec7580976d9 Op naar tree-IR
Windel Bouwman
parents: 261
diff changeset
157 if expr.op == 'or':
5ec7580976d9 Op naar tree-IR
Windel Bouwman
parents: 261
diff changeset
158 l2 = self.newBlock()
308
2e7f55319858 Merged analyse into codegenerator
Windel Bouwman
parents: 307
diff changeset
159 self.gen_cond_code(expr.a, bbtrue, l2)
2e7f55319858 Merged analyse into codegenerator
Windel Bouwman
parents: 307
diff changeset
160 if not self.equalTypes(expr.a.typ, self.boolType):
307
e609d5296ee9 Massive rewrite of codegenerator
Windel Bouwman
parents: 306
diff changeset
161 raise SemanticError('Must be boolean', expr.a.loc)
268
5ec7580976d9 Op naar tree-IR
Windel Bouwman
parents: 261
diff changeset
162 self.setBlock(l2)
308
2e7f55319858 Merged analyse into codegenerator
Windel Bouwman
parents: 307
diff changeset
163 self.gen_cond_code(expr.b, bbtrue, bbfalse)
2e7f55319858 Merged analyse into codegenerator
Windel Bouwman
parents: 307
diff changeset
164 if not self.equalTypes(expr.b.typ, self.boolType):
307
e609d5296ee9 Massive rewrite of codegenerator
Windel Bouwman
parents: 306
diff changeset
165 raise SemanticError('Must be boolean', expr.b.loc)
268
5ec7580976d9 Op naar tree-IR
Windel Bouwman
parents: 261
diff changeset
166 elif expr.op == 'and':
5ec7580976d9 Op naar tree-IR
Windel Bouwman
parents: 261
diff changeset
167 l2 = self.newBlock()
308
2e7f55319858 Merged analyse into codegenerator
Windel Bouwman
parents: 307
diff changeset
168 self.gen_cond_code(expr.a, l2, bbfalse)
2e7f55319858 Merged analyse into codegenerator
Windel Bouwman
parents: 307
diff changeset
169 if not self.equalTypes(expr.a.typ, self.boolType):
307
e609d5296ee9 Massive rewrite of codegenerator
Windel Bouwman
parents: 306
diff changeset
170 self.error('Must be boolean', expr.a.loc)
268
5ec7580976d9 Op naar tree-IR
Windel Bouwman
parents: 261
diff changeset
171 self.setBlock(l2)
308
2e7f55319858 Merged analyse into codegenerator
Windel Bouwman
parents: 307
diff changeset
172 self.gen_cond_code(expr.b, bbtrue, bbfalse)
2e7f55319858 Merged analyse into codegenerator
Windel Bouwman
parents: 307
diff changeset
173 if not self.equalTypes(expr.b.typ, self.boolType):
307
e609d5296ee9 Massive rewrite of codegenerator
Windel Bouwman
parents: 306
diff changeset
174 raise SemanticError('Must be boolean', expr.b.loc)
305
0615b5308710 Updated docs
Windel Bouwman
parents: 303
diff changeset
175 elif expr.op in ['==', '>', '<', '!=', '<=', '>=']:
228
7f18ed9b6b7e Removal of emptystatement class
Windel Bouwman
parents: 227
diff changeset
176 ta = self.genExprCode(expr.a)
7f18ed9b6b7e Removal of emptystatement class
Windel Bouwman
parents: 227
diff changeset
177 tb = self.genExprCode(expr.b)
307
e609d5296ee9 Massive rewrite of codegenerator
Windel Bouwman
parents: 306
diff changeset
178 if not self.equalTypes(expr.a.typ, expr.b.typ):
e609d5296ee9 Massive rewrite of codegenerator
Windel Bouwman
parents: 306
diff changeset
179 raise SemanticError('Types unequal {} != {}'
e609d5296ee9 Massive rewrite of codegenerator
Windel Bouwman
parents: 306
diff changeset
180 .format(expr.a.typ, expr.b.typ), expr.loc)
268
5ec7580976d9 Op naar tree-IR
Windel Bouwman
parents: 261
diff changeset
181 self.emit(ir.CJump(ta, expr.op, tb, bbtrue, bbfalse))
5ec7580976d9 Op naar tree-IR
Windel Bouwman
parents: 261
diff changeset
182 else:
5ec7580976d9 Op naar tree-IR
Windel Bouwman
parents: 261
diff changeset
183 raise NotImplementedError('Unknown condition {}'.format(expr))
307
e609d5296ee9 Massive rewrite of codegenerator
Windel Bouwman
parents: 306
diff changeset
184 expr.typ = self.boolType
308
2e7f55319858 Merged analyse into codegenerator
Windel Bouwman
parents: 307
diff changeset
185 elif type(expr) is ast.Literal:
2e7f55319858 Merged analyse into codegenerator
Windel Bouwman
parents: 307
diff changeset
186 self.genExprCode(expr)
288
a747a45dcd78 Various styling work
Windel Bouwman
parents: 287
diff changeset
187 if expr.val:
268
5ec7580976d9 Op naar tree-IR
Windel Bouwman
parents: 261
diff changeset
188 self.emit(ir.Jump(bbtrue))
288
a747a45dcd78 Various styling work
Windel Bouwman
parents: 287
diff changeset
189 else:
268
5ec7580976d9 Op naar tree-IR
Windel Bouwman
parents: 261
diff changeset
190 self.emit(ir.Jump(bbfalse))
228
7f18ed9b6b7e Removal of emptystatement class
Windel Bouwman
parents: 227
diff changeset
191 else:
288
a747a45dcd78 Various styling work
Windel Bouwman
parents: 287
diff changeset
192 raise NotImplementedError('Unknown cond {}'.format(expr))
307
e609d5296ee9 Massive rewrite of codegenerator
Windel Bouwman
parents: 306
diff changeset
193 if not self.equalTypes(expr.typ, self.boolType):
e609d5296ee9 Massive rewrite of codegenerator
Windel Bouwman
parents: 306
diff changeset
194 self.error('Condition must be boolean', expr.loc)
230
88a1e0baef65 Added some tests for IR-code
Windel Bouwman
parents: 228
diff changeset
195
217
8b2e5f3cd579 Removed some stale python source files
Windel Bouwman
parents: 205
diff changeset
196 def genExprCode(self, expr):
308
2e7f55319858 Merged analyse into codegenerator
Windel Bouwman
parents: 307
diff changeset
197 """ Generate code for an expression. Return the generated ir-value """
2e7f55319858 Merged analyse into codegenerator
Windel Bouwman
parents: 307
diff changeset
198 assert isinstance(expr, ast.Expression)
2e7f55319858 Merged analyse into codegenerator
Windel Bouwman
parents: 307
diff changeset
199 if type(expr) is ast.Binop:
307
e609d5296ee9 Massive rewrite of codegenerator
Windel Bouwman
parents: 306
diff changeset
200 expr.lvalue = False
e609d5296ee9 Massive rewrite of codegenerator
Windel Bouwman
parents: 306
diff changeset
201 if expr.op in ['+', '-', '*', '/', '<<', '>>', '|', '&']:
e609d5296ee9 Massive rewrite of codegenerator
Windel Bouwman
parents: 306
diff changeset
202 ra = self.genExprCode(expr.a)
e609d5296ee9 Massive rewrite of codegenerator
Windel Bouwman
parents: 306
diff changeset
203 rb = self.genExprCode(expr.b)
e609d5296ee9 Massive rewrite of codegenerator
Windel Bouwman
parents: 306
diff changeset
204 if self.equalTypes(expr.a.typ, self.intType) and \
e609d5296ee9 Massive rewrite of codegenerator
Windel Bouwman
parents: 306
diff changeset
205 self.equalTypes(expr.b.typ, self.intType):
e609d5296ee9 Massive rewrite of codegenerator
Windel Bouwman
parents: 306
diff changeset
206 expr.typ = expr.a.typ
e609d5296ee9 Massive rewrite of codegenerator
Windel Bouwman
parents: 306
diff changeset
207 else:
e609d5296ee9 Massive rewrite of codegenerator
Windel Bouwman
parents: 306
diff changeset
208 raise SemanticError('Can only add integers', expr.loc)
e609d5296ee9 Massive rewrite of codegenerator
Windel Bouwman
parents: 306
diff changeset
209 else:
e609d5296ee9 Massive rewrite of codegenerator
Windel Bouwman
parents: 306
diff changeset
210 raise NotImplementedError("Cannot use equality as expressions")
268
5ec7580976d9 Op naar tree-IR
Windel Bouwman
parents: 261
diff changeset
211 return ir.Binop(ra, expr.op, rb)
308
2e7f55319858 Merged analyse into codegenerator
Windel Bouwman
parents: 307
diff changeset
212 elif type(expr) is ast.Unop:
307
e609d5296ee9 Massive rewrite of codegenerator
Windel Bouwman
parents: 306
diff changeset
213 if expr.op == '&':
e609d5296ee9 Massive rewrite of codegenerator
Windel Bouwman
parents: 306
diff changeset
214 ra = self.genExprCode(expr.a)
308
2e7f55319858 Merged analyse into codegenerator
Windel Bouwman
parents: 307
diff changeset
215 expr.typ = ast.PointerType(expr.a.typ)
307
e609d5296ee9 Massive rewrite of codegenerator
Windel Bouwman
parents: 306
diff changeset
216 if not expr.a.lvalue:
e609d5296ee9 Massive rewrite of codegenerator
Windel Bouwman
parents: 306
diff changeset
217 raise SemanticError('No valid lvalue', expr.a.loc)
e609d5296ee9 Massive rewrite of codegenerator
Windel Bouwman
parents: 306
diff changeset
218 expr.lvalue = False
e609d5296ee9 Massive rewrite of codegenerator
Windel Bouwman
parents: 306
diff changeset
219 assert type(ra) is ir.Mem
e609d5296ee9 Massive rewrite of codegenerator
Windel Bouwman
parents: 306
diff changeset
220 return ra.e
e609d5296ee9 Massive rewrite of codegenerator
Windel Bouwman
parents: 306
diff changeset
221 else:
e609d5296ee9 Massive rewrite of codegenerator
Windel Bouwman
parents: 306
diff changeset
222 raise NotImplementedError('Unknown unop {0}'.format(expr.op))
308
2e7f55319858 Merged analyse into codegenerator
Windel Bouwman
parents: 307
diff changeset
223 elif type(expr) is ast.Identifier:
307
e609d5296ee9 Massive rewrite of codegenerator
Windel Bouwman
parents: 306
diff changeset
224 # Generate code for this identifier.
e609d5296ee9 Massive rewrite of codegenerator
Windel Bouwman
parents: 306
diff changeset
225 expr.lvalue = True
e609d5296ee9 Massive rewrite of codegenerator
Windel Bouwman
parents: 306
diff changeset
226 tg = self.resolveSymbol(expr)
e609d5296ee9 Massive rewrite of codegenerator
Windel Bouwman
parents: 306
diff changeset
227 expr.kind = type(tg)
e609d5296ee9 Massive rewrite of codegenerator
Windel Bouwman
parents: 306
diff changeset
228 expr.typ = tg.typ
279
2ccd57b1d78c Fix register allocator to do burn2 OK
Windel Bouwman
parents: 275
diff changeset
229 # This returns the dereferenced variable.
308
2e7f55319858 Merged analyse into codegenerator
Windel Bouwman
parents: 307
diff changeset
230 if isinstance(tg, ast.Variable):
307
e609d5296ee9 Massive rewrite of codegenerator
Windel Bouwman
parents: 306
diff changeset
231 return ir.Mem(self.varMap[tg])
280
02385f62f250 Rework from str interface to Instruction interface
Windel Bouwman
parents: 279
diff changeset
232 else:
308
2e7f55319858 Merged analyse into codegenerator
Windel Bouwman
parents: 307
diff changeset
233 raise NotImplementedError(str(tg))
2e7f55319858 Merged analyse into codegenerator
Windel Bouwman
parents: 307
diff changeset
234 elif type(expr) is ast.Deref:
222
c3f1ce8b638f Fixup of parser
Windel Bouwman
parents: 221
diff changeset
235 # dereference pointer type:
225
1c7364bd74c7 Fixed pointer deref
Windel Bouwman
parents: 222
diff changeset
236 addr = self.genExprCode(expr.ptr)
307
e609d5296ee9 Massive rewrite of codegenerator
Windel Bouwman
parents: 306
diff changeset
237 ptr_typ = self.theType(expr.ptr.typ)
e609d5296ee9 Massive rewrite of codegenerator
Windel Bouwman
parents: 306
diff changeset
238 expr.lvalue = True
308
2e7f55319858 Merged analyse into codegenerator
Windel Bouwman
parents: 307
diff changeset
239 if type(ptr_typ) is ast.PointerType:
307
e609d5296ee9 Massive rewrite of codegenerator
Windel Bouwman
parents: 306
diff changeset
240 expr.typ = ptr_typ.ptype
e609d5296ee9 Massive rewrite of codegenerator
Windel Bouwman
parents: 306
diff changeset
241 return ir.Mem(addr)
e609d5296ee9 Massive rewrite of codegenerator
Windel Bouwman
parents: 306
diff changeset
242 else:
e609d5296ee9 Massive rewrite of codegenerator
Windel Bouwman
parents: 306
diff changeset
243 raise SemanticError('Cannot deref non-pointer', expr.loc)
308
2e7f55319858 Merged analyse into codegenerator
Windel Bouwman
parents: 307
diff changeset
244 elif type(expr) is ast.Member:
279
2ccd57b1d78c Fix register allocator to do burn2 OK
Windel Bouwman
parents: 275
diff changeset
245 base = self.genExprCode(expr.base)
307
e609d5296ee9 Massive rewrite of codegenerator
Windel Bouwman
parents: 306
diff changeset
246 expr.lvalue = expr.base.lvalue
e609d5296ee9 Massive rewrite of codegenerator
Windel Bouwman
parents: 306
diff changeset
247 basetype = self.theType(expr.base.typ)
308
2e7f55319858 Merged analyse into codegenerator
Windel Bouwman
parents: 307
diff changeset
248 if type(basetype) is ast.StructureType:
307
e609d5296ee9 Massive rewrite of codegenerator
Windel Bouwman
parents: 306
diff changeset
249 if basetype.hasField(expr.field):
e609d5296ee9 Massive rewrite of codegenerator
Windel Bouwman
parents: 306
diff changeset
250 expr.typ = basetype.fieldType(expr.field)
e609d5296ee9 Massive rewrite of codegenerator
Windel Bouwman
parents: 306
diff changeset
251 else:
e609d5296ee9 Massive rewrite of codegenerator
Windel Bouwman
parents: 306
diff changeset
252 raise SemanticError('{} does not contain field {}'
308
2e7f55319858 Merged analyse into codegenerator
Windel Bouwman
parents: 307
diff changeset
253 .format(basetype, expr.field), expr.loc)
307
e609d5296ee9 Massive rewrite of codegenerator
Windel Bouwman
parents: 306
diff changeset
254 else:
308
2e7f55319858 Merged analyse into codegenerator
Windel Bouwman
parents: 307
diff changeset
255 raise SemanticError('Cannot select {} of non-structure type {}'
2e7f55319858 Merged analyse into codegenerator
Windel Bouwman
parents: 307
diff changeset
256 .format(expr.field, basetype), expr.loc)
307
e609d5296ee9 Massive rewrite of codegenerator
Windel Bouwman
parents: 306
diff changeset
257
279
2ccd57b1d78c Fix register allocator to do burn2 OK
Windel Bouwman
parents: 275
diff changeset
258 assert type(base) is ir.Mem, type(base)
307
e609d5296ee9 Massive rewrite of codegenerator
Windel Bouwman
parents: 306
diff changeset
259 bt = self.theType(expr.base.typ)
268
5ec7580976d9 Op naar tree-IR
Windel Bouwman
parents: 261
diff changeset
260 offset = ir.Const(bt.fieldOffset(expr.field))
308
2e7f55319858 Merged analyse into codegenerator
Windel Bouwman
parents: 307
diff changeset
261 return ir.Mem(ir.Add(base.e, offset))
2e7f55319858 Merged analyse into codegenerator
Windel Bouwman
parents: 307
diff changeset
262 elif type(expr) is ast.Literal:
307
e609d5296ee9 Massive rewrite of codegenerator
Windel Bouwman
parents: 306
diff changeset
263 expr.lvalue = False
e609d5296ee9 Massive rewrite of codegenerator
Windel Bouwman
parents: 306
diff changeset
264 typemap = {int: 'int', float: 'double', bool: 'bool'}
e609d5296ee9 Massive rewrite of codegenerator
Windel Bouwman
parents: 306
diff changeset
265 if type(expr.val) in typemap:
e609d5296ee9 Massive rewrite of codegenerator
Windel Bouwman
parents: 306
diff changeset
266 expr.typ = self.pkg.scope[typemap[type(expr.val)]]
e609d5296ee9 Massive rewrite of codegenerator
Windel Bouwman
parents: 306
diff changeset
267 else:
e609d5296ee9 Massive rewrite of codegenerator
Windel Bouwman
parents: 306
diff changeset
268 raise SemanticError('Unknown literal type {}'.format(expr.val))
268
5ec7580976d9 Op naar tree-IR
Windel Bouwman
parents: 261
diff changeset
269 return ir.Const(expr.val)
308
2e7f55319858 Merged analyse into codegenerator
Windel Bouwman
parents: 307
diff changeset
270 elif type(expr) is ast.TypeCast:
2e7f55319858 Merged analyse into codegenerator
Windel Bouwman
parents: 307
diff changeset
271 return self.gen_type_cast(expr)
2e7f55319858 Merged analyse into codegenerator
Windel Bouwman
parents: 307
diff changeset
272 elif type(expr) is ast.FunctionCall:
2e7f55319858 Merged analyse into codegenerator
Windel Bouwman
parents: 307
diff changeset
273 return self.gen_function_call(expr)
225
1c7364bd74c7 Fixed pointer deref
Windel Bouwman
parents: 222
diff changeset
274 else:
259
ac603eb66b63 Added function call
Windel Bouwman
parents: 255
diff changeset
275 raise NotImplementedError('Unknown expr {}'.format(expr))
307
e609d5296ee9 Massive rewrite of codegenerator
Windel Bouwman
parents: 306
diff changeset
276
308
2e7f55319858 Merged analyse into codegenerator
Windel Bouwman
parents: 307
diff changeset
277 def gen_type_cast(self, expr):
2e7f55319858 Merged analyse into codegenerator
Windel Bouwman
parents: 307
diff changeset
278 """ Generate code for type casting """
2e7f55319858 Merged analyse into codegenerator
Windel Bouwman
parents: 307
diff changeset
279 ar = self.genExprCode(expr.a)
2e7f55319858 Merged analyse into codegenerator
Windel Bouwman
parents: 307
diff changeset
280 from_type = self.theType(expr.a.typ)
2e7f55319858 Merged analyse into codegenerator
Windel Bouwman
parents: 307
diff changeset
281 to_type = self.theType(expr.to_type)
2e7f55319858 Merged analyse into codegenerator
Windel Bouwman
parents: 307
diff changeset
282 if isinstance(from_type, ast.PointerType) and isinstance(to_type, ast.PointerType):
2e7f55319858 Merged analyse into codegenerator
Windel Bouwman
parents: 307
diff changeset
283 expr.typ = expr.to_type
2e7f55319858 Merged analyse into codegenerator
Windel Bouwman
parents: 307
diff changeset
284 return ar
2e7f55319858 Merged analyse into codegenerator
Windel Bouwman
parents: 307
diff changeset
285 elif type(from_type) is ast.BaseType and from_type.name == 'int' and \
2e7f55319858 Merged analyse into codegenerator
Windel Bouwman
parents: 307
diff changeset
286 isinstance(to_type, ast.PointerType):
2e7f55319858 Merged analyse into codegenerator
Windel Bouwman
parents: 307
diff changeset
287 expr.typ = expr.to_type
2e7f55319858 Merged analyse into codegenerator
Windel Bouwman
parents: 307
diff changeset
288 return ar
2e7f55319858 Merged analyse into codegenerator
Windel Bouwman
parents: 307
diff changeset
289 else:
2e7f55319858 Merged analyse into codegenerator
Windel Bouwman
parents: 307
diff changeset
290 raise SemanticError('Cannot cast {} to {}'
2e7f55319858 Merged analyse into codegenerator
Windel Bouwman
parents: 307
diff changeset
291 .format(from_type, to_type), expr.loc)
2e7f55319858 Merged analyse into codegenerator
Windel Bouwman
parents: 307
diff changeset
292
2e7f55319858 Merged analyse into codegenerator
Windel Bouwman
parents: 307
diff changeset
293 def gen_function_call(self, expr):
2e7f55319858 Merged analyse into codegenerator
Windel Bouwman
parents: 307
diff changeset
294 """ Generate code for a function call """
2e7f55319858 Merged analyse into codegenerator
Windel Bouwman
parents: 307
diff changeset
295 # Evaluate the arguments:
2e7f55319858 Merged analyse into codegenerator
Windel Bouwman
parents: 307
diff changeset
296 args = [self.genExprCode(e) for e in expr.args]
2e7f55319858 Merged analyse into codegenerator
Windel Bouwman
parents: 307
diff changeset
297 # Check arguments:
2e7f55319858 Merged analyse into codegenerator
Windel Bouwman
parents: 307
diff changeset
298 tg = self.resolveSymbol(expr.proc)
2e7f55319858 Merged analyse into codegenerator
Windel Bouwman
parents: 307
diff changeset
299 if type(tg) is not ast.Function:
2e7f55319858 Merged analyse into codegenerator
Windel Bouwman
parents: 307
diff changeset
300 raise SemanticError('cannot call {}'.format(tg))
2e7f55319858 Merged analyse into codegenerator
Windel Bouwman
parents: 307
diff changeset
301 ftyp = tg.typ
2e7f55319858 Merged analyse into codegenerator
Windel Bouwman
parents: 307
diff changeset
302 fname = tg.name
2e7f55319858 Merged analyse into codegenerator
Windel Bouwman
parents: 307
diff changeset
303 ptypes = ftyp.parametertypes
2e7f55319858 Merged analyse into codegenerator
Windel Bouwman
parents: 307
diff changeset
304 if len(expr.args) != len(ptypes):
2e7f55319858 Merged analyse into codegenerator
Windel Bouwman
parents: 307
diff changeset
305 raise SemanticError('{} requires {} arguments, {} given'
2e7f55319858 Merged analyse into codegenerator
Windel Bouwman
parents: 307
diff changeset
306 .format(fname, len(ptypes), len(expr.args)), expr.loc)
2e7f55319858 Merged analyse into codegenerator
Windel Bouwman
parents: 307
diff changeset
307 for arg, at in zip(expr.args, ptypes):
2e7f55319858 Merged analyse into codegenerator
Windel Bouwman
parents: 307
diff changeset
308 if not self.equalTypes(arg.typ, at):
2e7f55319858 Merged analyse into codegenerator
Windel Bouwman
parents: 307
diff changeset
309 raise SemanticError('Got {}, expected {}'
2e7f55319858 Merged analyse into codegenerator
Windel Bouwman
parents: 307
diff changeset
310 .format(arg.typ, at), arg.loc)
2e7f55319858 Merged analyse into codegenerator
Windel Bouwman
parents: 307
diff changeset
311 # determine return type:
2e7f55319858 Merged analyse into codegenerator
Windel Bouwman
parents: 307
diff changeset
312 expr.typ = ftyp.returntype
2e7f55319858 Merged analyse into codegenerator
Windel Bouwman
parents: 307
diff changeset
313 return ir.Call(fname, args)
2e7f55319858 Merged analyse into codegenerator
Windel Bouwman
parents: 307
diff changeset
314
307
e609d5296ee9 Massive rewrite of codegenerator
Windel Bouwman
parents: 306
diff changeset
315 def resolveSymbol(self, sym):
308
2e7f55319858 Merged analyse into codegenerator
Windel Bouwman
parents: 307
diff changeset
316 if type(sym) is ast.Member:
307
e609d5296ee9 Massive rewrite of codegenerator
Windel Bouwman
parents: 306
diff changeset
317 base = self.resolveSymbol(sym.base)
308
2e7f55319858 Merged analyse into codegenerator
Windel Bouwman
parents: 307
diff changeset
318 if type(base) is not ast.Package:
2e7f55319858 Merged analyse into codegenerator
Windel Bouwman
parents: 307
diff changeset
319 raise SemanticError('Base is not a package', sym.loc)
307
e609d5296ee9 Massive rewrite of codegenerator
Windel Bouwman
parents: 306
diff changeset
320 scope = base.innerScope
e609d5296ee9 Massive rewrite of codegenerator
Windel Bouwman
parents: 306
diff changeset
321 name = sym.field
308
2e7f55319858 Merged analyse into codegenerator
Windel Bouwman
parents: 307
diff changeset
322 elif type(sym) is ast.Identifier:
307
e609d5296ee9 Massive rewrite of codegenerator
Windel Bouwman
parents: 306
diff changeset
323 scope = sym.scope
e609d5296ee9 Massive rewrite of codegenerator
Windel Bouwman
parents: 306
diff changeset
324 name = sym.target
e609d5296ee9 Massive rewrite of codegenerator
Windel Bouwman
parents: 306
diff changeset
325 else:
e609d5296ee9 Massive rewrite of codegenerator
Windel Bouwman
parents: 306
diff changeset
326 raise NotImplementedError(str(sym))
e609d5296ee9 Massive rewrite of codegenerator
Windel Bouwman
parents: 306
diff changeset
327 if name in scope:
e609d5296ee9 Massive rewrite of codegenerator
Windel Bouwman
parents: 306
diff changeset
328 s = scope[name]
e609d5296ee9 Massive rewrite of codegenerator
Windel Bouwman
parents: 306
diff changeset
329 else:
e609d5296ee9 Massive rewrite of codegenerator
Windel Bouwman
parents: 306
diff changeset
330 raise SemanticError('{} undefined'.format(name), sym.loc)
308
2e7f55319858 Merged analyse into codegenerator
Windel Bouwman
parents: 307
diff changeset
331 assert isinstance(s, ast.Symbol)
307
e609d5296ee9 Massive rewrite of codegenerator
Windel Bouwman
parents: 306
diff changeset
332 return s
e609d5296ee9 Massive rewrite of codegenerator
Windel Bouwman
parents: 306
diff changeset
333
e609d5296ee9 Massive rewrite of codegenerator
Windel Bouwman
parents: 306
diff changeset
334 def theType(self, t):
e609d5296ee9 Massive rewrite of codegenerator
Windel Bouwman
parents: 306
diff changeset
335 """ Recurse until a 'real' type is found """
308
2e7f55319858 Merged analyse into codegenerator
Windel Bouwman
parents: 307
diff changeset
336 if type(t) is ast.DefinedType:
307
e609d5296ee9 Massive rewrite of codegenerator
Windel Bouwman
parents: 306
diff changeset
337 t = self.theType(t.typ)
308
2e7f55319858 Merged analyse into codegenerator
Windel Bouwman
parents: 307
diff changeset
338 elif type(t) in [ast.Identifier, ast.Member]:
307
e609d5296ee9 Massive rewrite of codegenerator
Windel Bouwman
parents: 306
diff changeset
339 t = self.theType(self.resolveSymbol(t))
308
2e7f55319858 Merged analyse into codegenerator
Windel Bouwman
parents: 307
diff changeset
340 elif isinstance(t, ast.Type):
307
e609d5296ee9 Massive rewrite of codegenerator
Windel Bouwman
parents: 306
diff changeset
341 pass
e609d5296ee9 Massive rewrite of codegenerator
Windel Bouwman
parents: 306
diff changeset
342 else:
e609d5296ee9 Massive rewrite of codegenerator
Windel Bouwman
parents: 306
diff changeset
343 raise NotImplementedError(str(t))
308
2e7f55319858 Merged analyse into codegenerator
Windel Bouwman
parents: 307
diff changeset
344 assert isinstance(t, ast.Type)
307
e609d5296ee9 Massive rewrite of codegenerator
Windel Bouwman
parents: 306
diff changeset
345 return t
e609d5296ee9 Massive rewrite of codegenerator
Windel Bouwman
parents: 306
diff changeset
346
e609d5296ee9 Massive rewrite of codegenerator
Windel Bouwman
parents: 306
diff changeset
347 def equalTypes(self, a, b):
e609d5296ee9 Massive rewrite of codegenerator
Windel Bouwman
parents: 306
diff changeset
348 """ Compare types a and b for structural equavalence. """
e609d5296ee9 Massive rewrite of codegenerator
Windel Bouwman
parents: 306
diff changeset
349 # Recurse into named types:
e609d5296ee9 Massive rewrite of codegenerator
Windel Bouwman
parents: 306
diff changeset
350 a = self.theType(a)
e609d5296ee9 Massive rewrite of codegenerator
Windel Bouwman
parents: 306
diff changeset
351 b = self.theType(b)
308
2e7f55319858 Merged analyse into codegenerator
Windel Bouwman
parents: 307
diff changeset
352 assert isinstance(a, ast.Type)
2e7f55319858 Merged analyse into codegenerator
Windel Bouwman
parents: 307
diff changeset
353 assert isinstance(b, ast.Type)
307
e609d5296ee9 Massive rewrite of codegenerator
Windel Bouwman
parents: 306
diff changeset
354
e609d5296ee9 Massive rewrite of codegenerator
Windel Bouwman
parents: 306
diff changeset
355 if type(a) is type(b):
308
2e7f55319858 Merged analyse into codegenerator
Windel Bouwman
parents: 307
diff changeset
356 if type(a) is ast.BaseType:
307
e609d5296ee9 Massive rewrite of codegenerator
Windel Bouwman
parents: 306
diff changeset
357 return a.name == b.name
308
2e7f55319858 Merged analyse into codegenerator
Windel Bouwman
parents: 307
diff changeset
358 elif type(a) is ast.PointerType:
307
e609d5296ee9 Massive rewrite of codegenerator
Windel Bouwman
parents: 306
diff changeset
359 return self.equalTypes(a.ptype, b.ptype)
308
2e7f55319858 Merged analyse into codegenerator
Windel Bouwman
parents: 307
diff changeset
360 elif type(a) is ast.StructureType:
307
e609d5296ee9 Massive rewrite of codegenerator
Windel Bouwman
parents: 306
diff changeset
361 if len(a.mems) != len(b.mems):
e609d5296ee9 Massive rewrite of codegenerator
Windel Bouwman
parents: 306
diff changeset
362 return False
e609d5296ee9 Massive rewrite of codegenerator
Windel Bouwman
parents: 306
diff changeset
363 return all(self.equalTypes(am.typ, bm.typ) for am, bm in
e609d5296ee9 Massive rewrite of codegenerator
Windel Bouwman
parents: 306
diff changeset
364 zip(a.mems, b.mems))
e609d5296ee9 Massive rewrite of codegenerator
Windel Bouwman
parents: 306
diff changeset
365 else:
e609d5296ee9 Massive rewrite of codegenerator
Windel Bouwman
parents: 306
diff changeset
366 raise NotImplementedError('{} not implemented'.format(type(a)))
e609d5296ee9 Massive rewrite of codegenerator
Windel Bouwman
parents: 306
diff changeset
367 return False