annotate python/c3/codegenerator.py @ 232:e621e3ba78d2

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