annotate python/c3/codegenerator.py @ 231:521567d17388

simplify blink.c3
author Windel Bouwman
date Sat, 13 Jul 2013 20:20:44 +0200
parents 88a1e0baef65
children e621e3ba78d2
rev   line source
155
b28a11c01dbe Simplified IR classes
Windel Bouwman
parents: 151
diff changeset
1 import ir
b28a11c01dbe Simplified IR classes
Windel Bouwman
parents: 151
diff changeset
2 from . import astnodes
222
c3f1ce8b638f Fixup of parser
Windel Bouwman
parents: 221
diff changeset
3 from .scope import boolType, intType
220
3f6c30a5d234 Major change in expression parsing to enable pointers and structs
Windel Bouwman
parents: 217
diff changeset
4 from ppci import CompilerError
231
521567d17388 simplify blink.c3
Windel Bouwman
parents: 230
diff changeset
5 from .typecheck import theType
230
88a1e0baef65 Added some tests for IR-code
Windel Bouwman
parents: 228
diff changeset
6
88a1e0baef65 Added some tests for IR-code
Windel Bouwman
parents: 228
diff changeset
7 tmpnames = {'+':'add', '-':'sub', '*': 'mul', '/':'div', '|':'or', '&':'and'}
228
7f18ed9b6b7e Removal of emptystatement class
Windel Bouwman
parents: 227
diff changeset
8
170
4348da5ca307 Cleanup of ir dir
Windel Bouwman
parents: 169
diff changeset
9 class CodeGenerator:
217
8b2e5f3cd579 Removed some stale python source files
Windel Bouwman
parents: 205
diff changeset
10 """ Generates intermediate code from a package """
8b2e5f3cd579 Removed some stale python source files
Windel Bouwman
parents: 205
diff changeset
11 def gencode(self, pkg):
8b2e5f3cd579 Removed some stale python source files
Windel Bouwman
parents: 205
diff changeset
12 assert type(pkg) is astnodes.Package
8b2e5f3cd579 Removed some stale python source files
Windel Bouwman
parents: 205
diff changeset
13 self.varMap = {} # Maps variables to storage locations.
8b2e5f3cd579 Removed some stale python source files
Windel Bouwman
parents: 205
diff changeset
14 self.funcMap = {}
8b2e5f3cd579 Removed some stale python source files
Windel Bouwman
parents: 205
diff changeset
15 self.builder = ir.Builder()
8b2e5f3cd579 Removed some stale python source files
Windel Bouwman
parents: 205
diff changeset
16 m = ir.Module(pkg.name)
8b2e5f3cd579 Removed some stale python source files
Windel Bouwman
parents: 205
diff changeset
17 self.builder.setModule(m)
8b2e5f3cd579 Removed some stale python source files
Windel Bouwman
parents: 205
diff changeset
18 self.genModule(pkg)
8b2e5f3cd579 Removed some stale python source files
Windel Bouwman
parents: 205
diff changeset
19 return m
170
4348da5ca307 Cleanup of ir dir
Windel Bouwman
parents: 169
diff changeset
20
217
8b2e5f3cd579 Removed some stale python source files
Windel Bouwman
parents: 205
diff changeset
21 # inner helpers:
8b2e5f3cd579 Removed some stale python source files
Windel Bouwman
parents: 205
diff changeset
22 def genModule(self, pkg):
222
c3f1ce8b638f Fixup of parser
Windel Bouwman
parents: 221
diff changeset
23 # Take care of forward declarations:
c3f1ce8b638f Fixup of parser
Windel Bouwman
parents: 221
diff changeset
24 for s in pkg.innerScope.Functions:
204
de3a68f677a5 Added long comment to c3 parser
Windel Bouwman
parents: 186
diff changeset
25 f = self.builder.newFunction(s.name)
175
a51b3c956386 Added function call in expressions
Windel Bouwman
parents: 174
diff changeset
26 self.funcMap[s] = f
222
c3f1ce8b638f Fixup of parser
Windel Bouwman
parents: 221
diff changeset
27 for s in pkg.innerScope:
170
4348da5ca307 Cleanup of ir dir
Windel Bouwman
parents: 169
diff changeset
28 if type(s) is astnodes.Variable:
230
88a1e0baef65 Added some tests for IR-code
Windel Bouwman
parents: 228
diff changeset
29 v = self.builder.newTmp(s.name)
186
46d62dadd61b Improved testsuite
Windel Bouwman
parents: 177
diff changeset
30 #self.builder.addIns(ir.Alloc(v))
46d62dadd61b Improved testsuite
Windel Bouwman
parents: 177
diff changeset
31 self.varMap[s] = v
170
4348da5ca307 Cleanup of ir dir
Windel Bouwman
parents: 169
diff changeset
32 elif type(s) is astnodes.Function:
4348da5ca307 Cleanup of ir dir
Windel Bouwman
parents: 169
diff changeset
33 # TODO: handle arguments
175
a51b3c956386 Added function call in expressions
Windel Bouwman
parents: 174
diff changeset
34 f = self.funcMap[s]
204
de3a68f677a5 Added long comment to c3 parser
Windel Bouwman
parents: 186
diff changeset
35 self.builder.setFunction(f)
171
3eb9b9e2958d Improved IR code
Windel Bouwman
parents: 170
diff changeset
36 bb = self.builder.newBB()
172
5a7d37d615ee Added function to IR
Windel Bouwman
parents: 171
diff changeset
37 f.entry = bb
171
3eb9b9e2958d Improved IR code
Windel Bouwman
parents: 170
diff changeset
38 self.builder.setBB(bb)
174
3eb06f5fb987 Added memory alloc for locals
Windel Bouwman
parents: 172
diff changeset
39 # generate room for locals:
3eb06f5fb987 Added memory alloc for locals
Windel Bouwman
parents: 172
diff changeset
40
217
8b2e5f3cd579 Removed some stale python source files
Windel Bouwman
parents: 205
diff changeset
41 for sym in s.innerScope:
175
a51b3c956386 Added function call in expressions
Windel Bouwman
parents: 174
diff changeset
42 #print(sym, sym.isParameter)
a51b3c956386 Added function call in expressions
Windel Bouwman
parents: 174
diff changeset
43 # TODO: handle parameters different
174
3eb06f5fb987 Added memory alloc for locals
Windel Bouwman
parents: 172
diff changeset
44 v = self.builder.newTmp(sym.name)
3eb06f5fb987 Added memory alloc for locals
Windel Bouwman
parents: 172
diff changeset
45 self.builder.addIns(ir.Alloc(v))
3eb06f5fb987 Added memory alloc for locals
Windel Bouwman
parents: 172
diff changeset
46 self.varMap[sym] = v
3eb06f5fb987 Added memory alloc for locals
Windel Bouwman
parents: 172
diff changeset
47
170
4348da5ca307 Cleanup of ir dir
Windel Bouwman
parents: 169
diff changeset
48 self.genCode(s.body)
172
5a7d37d615ee Added function to IR
Windel Bouwman
parents: 171
diff changeset
49 # TODO handle return?
171
3eb9b9e2958d Improved IR code
Windel Bouwman
parents: 170
diff changeset
50 self.builder.addIns(ir.Return())
204
de3a68f677a5 Added long comment to c3 parser
Windel Bouwman
parents: 186
diff changeset
51 self.builder.setFunction(None)
170
4348da5ca307 Cleanup of ir dir
Windel Bouwman
parents: 169
diff changeset
52 else:
4348da5ca307 Cleanup of ir dir
Windel Bouwman
parents: 169
diff changeset
53 print(s)
158
9683a4cd848f Added some functions for code generation
Windel Bouwman
parents: 157
diff changeset
54
217
8b2e5f3cd579 Removed some stale python source files
Windel Bouwman
parents: 205
diff changeset
55 def genCode(self, code):
222
c3f1ce8b638f Fixup of parser
Windel Bouwman
parents: 221
diff changeset
56 assert isinstance(code, astnodes.Statement)
c3f1ce8b638f Fixup of parser
Windel Bouwman
parents: 221
diff changeset
57 if type(code) is astnodes.CompoundStatement:
221
848c4b15fd0b pointers
Windel Bouwman
parents: 220
diff changeset
58 for s in code.statements:
848c4b15fd0b pointers
Windel Bouwman
parents: 220
diff changeset
59 self.genCode(s)
222
c3f1ce8b638f Fixup of parser
Windel Bouwman
parents: 221
diff changeset
60 elif type(code) is astnodes.Assignment:
228
7f18ed9b6b7e Removal of emptystatement class
Windel Bouwman
parents: 227
diff changeset
61 re = self.genExprCode(code.rval)
7f18ed9b6b7e Removal of emptystatement class
Windel Bouwman
parents: 227
diff changeset
62 loc = self.genExprCode(code.lval)
7f18ed9b6b7e Removal of emptystatement class
Windel Bouwman
parents: 227
diff changeset
63 self.builder.addIns(ir.Store(loc, re))
222
c3f1ce8b638f Fixup of parser
Windel Bouwman
parents: 221
diff changeset
64 elif type(code) is astnodes.ExpressionStatement:
c3f1ce8b638f Fixup of parser
Windel Bouwman
parents: 221
diff changeset
65 self.genExprCode(code.ex)
c3f1ce8b638f Fixup of parser
Windel Bouwman
parents: 221
diff changeset
66 elif type(code) is astnodes.IfStatement:
228
7f18ed9b6b7e Removal of emptystatement class
Windel Bouwman
parents: 227
diff changeset
67 bbtrue = self.builder.newBB()
7f18ed9b6b7e Removal of emptystatement class
Windel Bouwman
parents: 227
diff changeset
68 bbfalse = self.builder.newBB()
7f18ed9b6b7e Removal of emptystatement class
Windel Bouwman
parents: 227
diff changeset
69 te = self.builder.newBB()
7f18ed9b6b7e Removal of emptystatement class
Windel Bouwman
parents: 227
diff changeset
70 self.genCondCode(code.condition, bbtrue, bbfalse)
7f18ed9b6b7e Removal of emptystatement class
Windel Bouwman
parents: 227
diff changeset
71 self.builder.setBB(bbtrue)
7f18ed9b6b7e Removal of emptystatement class
Windel Bouwman
parents: 227
diff changeset
72 self.genCode(code.truestatement)
7f18ed9b6b7e Removal of emptystatement class
Windel Bouwman
parents: 227
diff changeset
73 self.builder.addIns(ir.Branch(te))
7f18ed9b6b7e Removal of emptystatement class
Windel Bouwman
parents: 227
diff changeset
74 self.builder.setBB(bbfalse)
7f18ed9b6b7e Removal of emptystatement class
Windel Bouwman
parents: 227
diff changeset
75 if code.falsestatement:
7f18ed9b6b7e Removal of emptystatement class
Windel Bouwman
parents: 227
diff changeset
76 self.genCode(code.falsestatement)
7f18ed9b6b7e Removal of emptystatement class
Windel Bouwman
parents: 227
diff changeset
77 self.builder.addIns(ir.Branch(te))
7f18ed9b6b7e Removal of emptystatement class
Windel Bouwman
parents: 227
diff changeset
78 self.builder.setBB(te)
222
c3f1ce8b638f Fixup of parser
Windel Bouwman
parents: 221
diff changeset
79 elif type(code) is astnodes.ReturnStatement:
228
7f18ed9b6b7e Removal of emptystatement class
Windel Bouwman
parents: 227
diff changeset
80 if code.expr:
7f18ed9b6b7e Removal of emptystatement class
Windel Bouwman
parents: 227
diff changeset
81 re = self.genExprCode(code.expr)
7f18ed9b6b7e Removal of emptystatement class
Windel Bouwman
parents: 227
diff changeset
82 self.builder.addIns(ir.Return(re))
7f18ed9b6b7e Removal of emptystatement class
Windel Bouwman
parents: 227
diff changeset
83 else:
7f18ed9b6b7e Removal of emptystatement class
Windel Bouwman
parents: 227
diff changeset
84 self.builder.addIns(ir.Return())
222
c3f1ce8b638f Fixup of parser
Windel Bouwman
parents: 221
diff changeset
85 elif type(code) is astnodes.WhileStatement:
228
7f18ed9b6b7e Removal of emptystatement class
Windel Bouwman
parents: 227
diff changeset
86 bbdo = self.builder.newBB()
7f18ed9b6b7e Removal of emptystatement class
Windel Bouwman
parents: 227
diff changeset
87 bbtest = self.builder.newBB()
7f18ed9b6b7e Removal of emptystatement class
Windel Bouwman
parents: 227
diff changeset
88 te = self.builder.newBB()
7f18ed9b6b7e Removal of emptystatement class
Windel Bouwman
parents: 227
diff changeset
89 self.builder.addIns(ir.Branch(bbtest))
7f18ed9b6b7e Removal of emptystatement class
Windel Bouwman
parents: 227
diff changeset
90 self.builder.setBB(bbtest)
7f18ed9b6b7e Removal of emptystatement class
Windel Bouwman
parents: 227
diff changeset
91 self.genCondCode(code.condition, bbdo, te)
7f18ed9b6b7e Removal of emptystatement class
Windel Bouwman
parents: 227
diff changeset
92 self.builder.setBB(bbdo)
7f18ed9b6b7e Removal of emptystatement class
Windel Bouwman
parents: 227
diff changeset
93 self.genCode(code.statement)
7f18ed9b6b7e Removal of emptystatement class
Windel Bouwman
parents: 227
diff changeset
94 self.builder.addIns(ir.Branch(bbtest))
7f18ed9b6b7e Removal of emptystatement class
Windel Bouwman
parents: 227
diff changeset
95 self.builder.setBB(te)
222
c3f1ce8b638f Fixup of parser
Windel Bouwman
parents: 221
diff changeset
96 else:
228
7f18ed9b6b7e Removal of emptystatement class
Windel Bouwman
parents: 227
diff changeset
97 print('Unknown stmt:', code)
230
88a1e0baef65 Added some tests for IR-code
Windel Bouwman
parents: 228
diff changeset
98
217
8b2e5f3cd579 Removed some stale python source files
Windel Bouwman
parents: 205
diff changeset
99 def genCondCode(self, expr, bbtrue, bbfalse):
228
7f18ed9b6b7e Removal of emptystatement class
Windel Bouwman
parents: 227
diff changeset
100 # Implement sequential logical operators
7f18ed9b6b7e Removal of emptystatement class
Windel Bouwman
parents: 227
diff changeset
101 assert expr.typ == boolType
7f18ed9b6b7e Removal of emptystatement class
Windel Bouwman
parents: 227
diff changeset
102 if type(expr) is astnodes.Binop:
7f18ed9b6b7e Removal of emptystatement class
Windel Bouwman
parents: 227
diff changeset
103 if expr.op == 'or':
7f18ed9b6b7e Removal of emptystatement class
Windel Bouwman
parents: 227
diff changeset
104 l2 = self.builder.newBB()
7f18ed9b6b7e Removal of emptystatement class
Windel Bouwman
parents: 227
diff changeset
105 self.genCondCode(expr.a, bbtrue, l2)
7f18ed9b6b7e Removal of emptystatement class
Windel Bouwman
parents: 227
diff changeset
106 self.builder.setBB(l2)
7f18ed9b6b7e Removal of emptystatement class
Windel Bouwman
parents: 227
diff changeset
107 self.genCondCode(expr.b, bbtrue, bbfalse)
7f18ed9b6b7e Removal of emptystatement class
Windel Bouwman
parents: 227
diff changeset
108 elif expr.op == 'and':
7f18ed9b6b7e Removal of emptystatement class
Windel Bouwman
parents: 227
diff changeset
109 l2 = self.builder.newBB()
7f18ed9b6b7e Removal of emptystatement class
Windel Bouwman
parents: 227
diff changeset
110 self.genCondCode(expr.a, l2, bbfalse)
7f18ed9b6b7e Removal of emptystatement class
Windel Bouwman
parents: 227
diff changeset
111 self.builder.setBB(l2)
7f18ed9b6b7e Removal of emptystatement class
Windel Bouwman
parents: 227
diff changeset
112 self.genCondCode(expr.b, bbtrue, bbfalse)
7f18ed9b6b7e Removal of emptystatement class
Windel Bouwman
parents: 227
diff changeset
113 elif expr.op in ['==', '>', '<']:
7f18ed9b6b7e Removal of emptystatement class
Windel Bouwman
parents: 227
diff changeset
114 ta = self.genExprCode(expr.a)
7f18ed9b6b7e Removal of emptystatement class
Windel Bouwman
parents: 227
diff changeset
115 tb = self.genExprCode(expr.b)
7f18ed9b6b7e Removal of emptystatement class
Windel Bouwman
parents: 227
diff changeset
116 i = ir.ConditionalBranch(ta, expr.op, tb, bbtrue, bbfalse)
7f18ed9b6b7e Removal of emptystatement class
Windel Bouwman
parents: 227
diff changeset
117 self.builder.addIns(i)
7f18ed9b6b7e Removal of emptystatement class
Windel Bouwman
parents: 227
diff changeset
118 else:
7f18ed9b6b7e Removal of emptystatement class
Windel Bouwman
parents: 227
diff changeset
119 raise NotImlementedError('Unknown condition {0}'.format(expr))
7f18ed9b6b7e Removal of emptystatement class
Windel Bouwman
parents: 227
diff changeset
120 elif type(expr) is astnodes.Literal:
7f18ed9b6b7e Removal of emptystatement class
Windel Bouwman
parents: 227
diff changeset
121 if expr.val:
7f18ed9b6b7e Removal of emptystatement class
Windel Bouwman
parents: 227
diff changeset
122 self.builder.addIns(ir.Branch(bbtrue))
7f18ed9b6b7e Removal of emptystatement class
Windel Bouwman
parents: 227
diff changeset
123 else:
7f18ed9b6b7e Removal of emptystatement class
Windel Bouwman
parents: 227
diff changeset
124 self.builder.addIns(ir.Branch(bbfalse))
7f18ed9b6b7e Removal of emptystatement class
Windel Bouwman
parents: 227
diff changeset
125 else:
7f18ed9b6b7e Removal of emptystatement class
Windel Bouwman
parents: 227
diff changeset
126 print('Unknown cond', expr)
230
88a1e0baef65 Added some tests for IR-code
Windel Bouwman
parents: 228
diff changeset
127
88a1e0baef65 Added some tests for IR-code
Windel Bouwman
parents: 228
diff changeset
128 def cast_to_rvalue(self, expr, loc):
88a1e0baef65 Added some tests for IR-code
Windel Bouwman
parents: 228
diff changeset
129 """ Cast lvalue to rvalue if required """
88a1e0baef65 Added some tests for IR-code
Windel Bouwman
parents: 228
diff changeset
130 if hasattr(expr, 'expect_rvalue'):
88a1e0baef65 Added some tests for IR-code
Windel Bouwman
parents: 228
diff changeset
131 # Load value from memory location:
88a1e0baef65 Added some tests for IR-code
Windel Bouwman
parents: 228
diff changeset
132 tmp = self.builder.newTmp()
88a1e0baef65 Added some tests for IR-code
Windel Bouwman
parents: 228
diff changeset
133 i = ir.Load(loc, tmp)
88a1e0baef65 Added some tests for IR-code
Windel Bouwman
parents: 228
diff changeset
134 self.builder.addIns(i)
88a1e0baef65 Added some tests for IR-code
Windel Bouwman
parents: 228
diff changeset
135 return tmp
88a1e0baef65 Added some tests for IR-code
Windel Bouwman
parents: 228
diff changeset
136 if expr.lvalue:
88a1e0baef65 Added some tests for IR-code
Windel Bouwman
parents: 228
diff changeset
137 return loc
88a1e0baef65 Added some tests for IR-code
Windel Bouwman
parents: 228
diff changeset
138
217
8b2e5f3cd579 Removed some stale python source files
Windel Bouwman
parents: 205
diff changeset
139 def genExprCode(self, expr):
230
88a1e0baef65 Added some tests for IR-code
Windel Bouwman
parents: 228
diff changeset
140 assert isinstance(expr, astnodes.Expression)
225
1c7364bd74c7 Fixed pointer deref
Windel Bouwman
parents: 222
diff changeset
141 if type(expr) is astnodes.Binop:
228
7f18ed9b6b7e Removal of emptystatement class
Windel Bouwman
parents: 227
diff changeset
142 ra = self.genExprCode(expr.a)
7f18ed9b6b7e Removal of emptystatement class
Windel Bouwman
parents: 227
diff changeset
143 rb = self.genExprCode(expr.b)
7f18ed9b6b7e Removal of emptystatement class
Windel Bouwman
parents: 227
diff changeset
144 ops = ['+', '-', '*', '/', '|', '&']
7f18ed9b6b7e Removal of emptystatement class
Windel Bouwman
parents: 227
diff changeset
145 if expr.op in ops:
7f18ed9b6b7e Removal of emptystatement class
Windel Bouwman
parents: 227
diff changeset
146 tmp = self.builder.newTmp(tmpnames[expr.op])
7f18ed9b6b7e Removal of emptystatement class
Windel Bouwman
parents: 227
diff changeset
147 op = expr.op
7f18ed9b6b7e Removal of emptystatement class
Windel Bouwman
parents: 227
diff changeset
148 ins = ir.BinaryOperator(tmp, op, ra, rb)
7f18ed9b6b7e Removal of emptystatement class
Windel Bouwman
parents: 227
diff changeset
149 self.builder.addIns(ins)
7f18ed9b6b7e Removal of emptystatement class
Windel Bouwman
parents: 227
diff changeset
150 return tmp
7f18ed9b6b7e Removal of emptystatement class
Windel Bouwman
parents: 227
diff changeset
151 else:
7f18ed9b6b7e Removal of emptystatement class
Windel Bouwman
parents: 227
diff changeset
152 print('Unknown {0}'.format(expr))
7f18ed9b6b7e Removal of emptystatement class
Windel Bouwman
parents: 227
diff changeset
153 tmp = self.builder.newTmp()
7f18ed9b6b7e Removal of emptystatement class
Windel Bouwman
parents: 227
diff changeset
154 # TODO
7f18ed9b6b7e Removal of emptystatement class
Windel Bouwman
parents: 227
diff changeset
155 return tmp
225
1c7364bd74c7 Fixed pointer deref
Windel Bouwman
parents: 222
diff changeset
156 elif type(expr) is astnodes.Unop:
1c7364bd74c7 Fixed pointer deref
Windel Bouwman
parents: 222
diff changeset
157 ra = self.genExprCode(expr.a)
1c7364bd74c7 Fixed pointer deref
Windel Bouwman
parents: 222
diff changeset
158 if expr.op == '&':
1c7364bd74c7 Fixed pointer deref
Windel Bouwman
parents: 222
diff changeset
159 # Address of operator
1c7364bd74c7 Fixed pointer deref
Windel Bouwman
parents: 222
diff changeset
160 tmp = self.builder.newTmp('addr')
1c7364bd74c7 Fixed pointer deref
Windel Bouwman
parents: 222
diff changeset
161 return tmp
1c7364bd74c7 Fixed pointer deref
Windel Bouwman
parents: 222
diff changeset
162 #self.builder.addIns(ins)
1c7364bd74c7 Fixed pointer deref
Windel Bouwman
parents: 222
diff changeset
163 else:
1c7364bd74c7 Fixed pointer deref
Windel Bouwman
parents: 222
diff changeset
164 raise Exception('Unknown {0}'.format(expr))
1c7364bd74c7 Fixed pointer deref
Windel Bouwman
parents: 222
diff changeset
165 elif type(expr) is astnodes.Constant:
1c7364bd74c7 Fixed pointer deref
Windel Bouwman
parents: 222
diff changeset
166 tmp = self.builder.newTmp()
1c7364bd74c7 Fixed pointer deref
Windel Bouwman
parents: 222
diff changeset
167 # TODO
220
3f6c30a5d234 Major change in expression parsing to enable pointers and structs
Windel Bouwman
parents: 217
diff changeset
168 return tmp
225
1c7364bd74c7 Fixed pointer deref
Windel Bouwman
parents: 222
diff changeset
169 elif type(expr) is astnodes.VariableUse:
1c7364bd74c7 Fixed pointer deref
Windel Bouwman
parents: 222
diff changeset
170 loc = self.varMap[expr.target]
230
88a1e0baef65 Added some tests for IR-code
Windel Bouwman
parents: 228
diff changeset
171 return self.cast_to_rvalue(expr, loc)
225
1c7364bd74c7 Fixed pointer deref
Windel Bouwman
parents: 222
diff changeset
172 elif type(expr) is astnodes.Deref:
222
c3f1ce8b638f Fixup of parser
Windel Bouwman
parents: 221
diff changeset
173 # dereference pointer type:
225
1c7364bd74c7 Fixed pointer deref
Windel Bouwman
parents: 222
diff changeset
174 addr = self.genExprCode(expr.ptr)
222
c3f1ce8b638f Fixup of parser
Windel Bouwman
parents: 221
diff changeset
175 tmp = self.builder.newTmp('deref')
225
1c7364bd74c7 Fixed pointer deref
Windel Bouwman
parents: 222
diff changeset
176 ins = ir.Load(addr, tmp)
222
c3f1ce8b638f Fixup of parser
Windel Bouwman
parents: 221
diff changeset
177 self.builder.addIns(ins)
c3f1ce8b638f Fixup of parser
Windel Bouwman
parents: 221
diff changeset
178 return tmp
227
82dfe6a32717 Fixed tests
Windel Bouwman
parents: 225
diff changeset
179 elif type(expr) is astnodes.FieldRef:
230
88a1e0baef65 Added some tests for IR-code
Windel Bouwman
parents: 228
diff changeset
180 b = self.genExprCode(expr.base)
88a1e0baef65 Added some tests for IR-code
Windel Bouwman
parents: 228
diff changeset
181 offset = self.builder.newTmp('off_' + expr.field)
231
521567d17388 simplify blink.c3
Windel Bouwman
parents: 230
diff changeset
182 bt = theType(expr.base.typ)
230
88a1e0baef65 Added some tests for IR-code
Windel Bouwman
parents: 228
diff changeset
183 ofs = bt.fieldOffset(expr.field)
88a1e0baef65 Added some tests for IR-code
Windel Bouwman
parents: 228
diff changeset
184 ins = ir.ImmLoad(offset, ofs)
88a1e0baef65 Added some tests for IR-code
Windel Bouwman
parents: 228
diff changeset
185 self.builder.addIns(ins)
88a1e0baef65 Added some tests for IR-code
Windel Bouwman
parents: 228
diff changeset
186 tmp2 = self.builder.newTmp('adr_' + expr.field)
88a1e0baef65 Added some tests for IR-code
Windel Bouwman
parents: 228
diff changeset
187 ins = ir.BinaryOperator(tmp2, '+', b, offset)
88a1e0baef65 Added some tests for IR-code
Windel Bouwman
parents: 228
diff changeset
188 self.builder.addIns(ins)
88a1e0baef65 Added some tests for IR-code
Windel Bouwman
parents: 228
diff changeset
189 return self.cast_to_rvalue(expr, tmp2)
225
1c7364bd74c7 Fixed pointer deref
Windel Bouwman
parents: 222
diff changeset
190 elif type(expr) is astnodes.Literal:
1c7364bd74c7 Fixed pointer deref
Windel Bouwman
parents: 222
diff changeset
191 tmp = self.builder.newTmp()
1c7364bd74c7 Fixed pointer deref
Windel Bouwman
parents: 222
diff changeset
192 ins = ir.ImmLoad(tmp, expr.val)
1c7364bd74c7 Fixed pointer deref
Windel Bouwman
parents: 222
diff changeset
193 self.builder.addIns(ins)
1c7364bd74c7 Fixed pointer deref
Windel Bouwman
parents: 222
diff changeset
194 return tmp
1c7364bd74c7 Fixed pointer deref
Windel Bouwman
parents: 222
diff changeset
195 elif type(expr) is astnodes.TypeCast:
222
c3f1ce8b638f Fixup of parser
Windel Bouwman
parents: 221
diff changeset
196 ar = self.genExprCode(expr.a)
231
521567d17388 simplify blink.c3
Windel Bouwman
parents: 230
diff changeset
197 tt = theType(expr.to_type)
230
88a1e0baef65 Added some tests for IR-code
Windel Bouwman
parents: 228
diff changeset
198 if isinstance(tt, astnodes.PointerType):
222
c3f1ce8b638f Fixup of parser
Windel Bouwman
parents: 221
diff changeset
199 if expr.a.typ is intType:
c3f1ce8b638f Fixup of parser
Windel Bouwman
parents: 221
diff changeset
200 return ar
c3f1ce8b638f Fixup of parser
Windel Bouwman
parents: 221
diff changeset
201 elif isinstance(expr.a.typ, astnodes.PointerType):
c3f1ce8b638f Fixup of parser
Windel Bouwman
parents: 221
diff changeset
202 return ar
c3f1ce8b638f Fixup of parser
Windel Bouwman
parents: 221
diff changeset
203 else:
c3f1ce8b638f Fixup of parser
Windel Bouwman
parents: 221
diff changeset
204 raise Exception()
230
88a1e0baef65 Added some tests for IR-code
Windel Bouwman
parents: 228
diff changeset
205 else:
88a1e0baef65 Added some tests for IR-code
Windel Bouwman
parents: 228
diff changeset
206 raise Exception("not implemented")
225
1c7364bd74c7 Fixed pointer deref
Windel Bouwman
parents: 222
diff changeset
207 elif type(expr) is astnodes.FunctionCall:
228
7f18ed9b6b7e Removal of emptystatement class
Windel Bouwman
parents: 227
diff changeset
208 tmp = self.builder.newTmp("res")
7f18ed9b6b7e Removal of emptystatement class
Windel Bouwman
parents: 227
diff changeset
209 args = []
7f18ed9b6b7e Removal of emptystatement class
Windel Bouwman
parents: 227
diff changeset
210 for arg in expr.args:
7f18ed9b6b7e Removal of emptystatement class
Windel Bouwman
parents: 227
diff changeset
211 ar = self.genExprCode(arg)
7f18ed9b6b7e Removal of emptystatement class
Windel Bouwman
parents: 227
diff changeset
212 args.append(ar)
7f18ed9b6b7e Removal of emptystatement class
Windel Bouwman
parents: 227
diff changeset
213 fn = self.funcMap[expr.proc]
7f18ed9b6b7e Removal of emptystatement class
Windel Bouwman
parents: 227
diff changeset
214 ins = ir.Call(fn, args, tmp)
7f18ed9b6b7e Removal of emptystatement class
Windel Bouwman
parents: 227
diff changeset
215 self.builder.addIns(ins)
7f18ed9b6b7e Removal of emptystatement class
Windel Bouwman
parents: 227
diff changeset
216 return tmp
225
1c7364bd74c7 Fixed pointer deref
Windel Bouwman
parents: 222
diff changeset
217 else:
228
7f18ed9b6b7e Removal of emptystatement class
Windel Bouwman
parents: 227
diff changeset
218 raise CompilerError('Unknown expr {}'.format(expr))
157
8f3924b6076e Added some code generator things
Windel Bouwman
parents: 156
diff changeset
219