Mercurial > lcfOS
annotate python/ppci/c3/parser.py @ 354:5477e499b039
Added some sort of string functionality
author | Windel Bouwman |
---|---|
date | Thu, 13 Mar 2014 18:59:06 +0100 |
parents | d1ecc493384e |
children | c05ab629976a |
rev | line source |
---|---|
254 | 1 import logging |
354 | 2 from .. import CompilerError |
306 | 3 from .astnodes import Member, Literal, TypeCast, Unop, Binop |
307 | 4 from .astnodes import Assignment, ExpressionStatement, Compound |
315 | 5 from .astnodes import Return, While, If, Empty, For |
296 | 6 from .astnodes import FunctionType, Function, FormalParameter |
354 | 7 from .astnodes import StructureType, DefinedType, PointerType, ArrayType |
296 | 8 from .astnodes import Constant, Variable |
354 | 9 from .astnodes import StructField, Deref, Index |
306 | 10 from .astnodes import Package |
11 from .astnodes import Identifier | |
12 from .astnodes import FunctionCall | |
148 | 13 |
288 | 14 |
148 | 15 class Parser: |
213 | 16 """ Parses sourcecode into an abstract syntax tree (AST) """ |
17 def __init__(self, diag): | |
254 | 18 self.logger = logging.getLogger('c3') |
213 | 19 self.diag = diag |
20 | |
306 | 21 def parseSource(self, tokens): |
334 | 22 self.logger.debug('Parsing source') |
306 | 23 self.tokens = tokens |
24 self.token = self.tokens.__next__() | |
254 | 25 try: |
26 self.parsePackage() | |
306 | 27 self.mod.ok = True # Valid until proven wrong :) |
254 | 28 return self.mod |
29 except CompilerError as e: | |
30 self.diag.addDiag(e) | |
220
3f6c30a5d234
Major change in expression parsing to enable pointers and structs
Windel Bouwman
parents:
219
diff
changeset
|
31 |
213 | 32 def Error(self, msg): |
220
3f6c30a5d234
Major change in expression parsing to enable pointers and structs
Windel Bouwman
parents:
219
diff
changeset
|
33 raise CompilerError(msg, self.token.loc) |
3f6c30a5d234
Major change in expression parsing to enable pointers and structs
Windel Bouwman
parents:
219
diff
changeset
|
34 |
213 | 35 # Lexer helpers: |
36 def Consume(self, typ): | |
220
3f6c30a5d234
Major change in expression parsing to enable pointers and structs
Windel Bouwman
parents:
219
diff
changeset
|
37 if self.Peak == typ: |
3f6c30a5d234
Major change in expression parsing to enable pointers and structs
Windel Bouwman
parents:
219
diff
changeset
|
38 return self.NextToken() |
3f6c30a5d234
Major change in expression parsing to enable pointers and structs
Windel Bouwman
parents:
219
diff
changeset
|
39 else: |
3f6c30a5d234
Major change in expression parsing to enable pointers and structs
Windel Bouwman
parents:
219
diff
changeset
|
40 self.Error('Excected: "{0}", got "{1}"'.format(typ, self.Peak)) |
3f6c30a5d234
Major change in expression parsing to enable pointers and structs
Windel Bouwman
parents:
219
diff
changeset
|
41 |
213 | 42 @property |
43 def Peak(self): | |
220
3f6c30a5d234
Major change in expression parsing to enable pointers and structs
Windel Bouwman
parents:
219
diff
changeset
|
44 return self.token.typ |
3f6c30a5d234
Major change in expression parsing to enable pointers and structs
Windel Bouwman
parents:
219
diff
changeset
|
45 |
228 | 46 @property |
47 def CurLoc(self): | |
48 return self.token.loc | |
49 | |
213 | 50 def hasConsumed(self, typ): |
220
3f6c30a5d234
Major change in expression parsing to enable pointers and structs
Windel Bouwman
parents:
219
diff
changeset
|
51 if self.Peak == typ: |
3f6c30a5d234
Major change in expression parsing to enable pointers and structs
Windel Bouwman
parents:
219
diff
changeset
|
52 self.Consume(typ) |
3f6c30a5d234
Major change in expression parsing to enable pointers and structs
Windel Bouwman
parents:
219
diff
changeset
|
53 return True |
3f6c30a5d234
Major change in expression parsing to enable pointers and structs
Windel Bouwman
parents:
219
diff
changeset
|
54 return False |
213 | 55 |
56 def NextToken(self): | |
219 | 57 t = self.token |
58 if t.typ != 'END': | |
59 self.token = self.tokens.__next__() | |
60 return t | |
213 | 61 |
215 | 62 def addDeclaration(self, decl): |
336
d1ecc493384e
Added spiffy armtoken class for bit fiddeling. Added cool test that checks for build repeatability
Windel Bouwman
parents:
334
diff
changeset
|
63 self.currentPart.add_declaration(decl) |
288 | 64 |
251
6ed3d3a82a63
Added another c3 example. First import attempt
Windel Bouwman
parents:
232
diff
changeset
|
65 def parseImport(self): |
6ed3d3a82a63
Added another c3 example. First import attempt
Windel Bouwman
parents:
232
diff
changeset
|
66 self.Consume('import') |
6ed3d3a82a63
Added another c3 example. First import attempt
Windel Bouwman
parents:
232
diff
changeset
|
67 name = self.Consume('ID').val |
6ed3d3a82a63
Added another c3 example. First import attempt
Windel Bouwman
parents:
232
diff
changeset
|
68 self.mod.imports.append(name) |
6ed3d3a82a63
Added another c3 example. First import attempt
Windel Bouwman
parents:
232
diff
changeset
|
69 self.Consume(';') |
213 | 70 |
71 def parsePackage(self): | |
284 | 72 self.Consume('module') |
220
3f6c30a5d234
Major change in expression parsing to enable pointers and structs
Windel Bouwman
parents:
219
diff
changeset
|
73 name = self.Consume('ID') |
3f6c30a5d234
Major change in expression parsing to enable pointers and structs
Windel Bouwman
parents:
219
diff
changeset
|
74 self.Consume(';') |
296 | 75 self.mod = Package(name.val, name.loc) |
220
3f6c30a5d234
Major change in expression parsing to enable pointers and structs
Windel Bouwman
parents:
219
diff
changeset
|
76 self.currentPart = self.mod |
3f6c30a5d234
Major change in expression parsing to enable pointers and structs
Windel Bouwman
parents:
219
diff
changeset
|
77 while self.Peak != 'END': |
3f6c30a5d234
Major change in expression parsing to enable pointers and structs
Windel Bouwman
parents:
219
diff
changeset
|
78 self.parseTopLevel() |
3f6c30a5d234
Major change in expression parsing to enable pointers and structs
Windel Bouwman
parents:
219
diff
changeset
|
79 self.Consume('END') |
148 | 80 |
213 | 81 def parseTopLevel(self): |
251
6ed3d3a82a63
Added another c3 example. First import attempt
Windel Bouwman
parents:
232
diff
changeset
|
82 if self.Peak == 'function': |
6ed3d3a82a63
Added another c3 example. First import attempt
Windel Bouwman
parents:
232
diff
changeset
|
83 self.parseFunctionDef() |
6ed3d3a82a63
Added another c3 example. First import attempt
Windel Bouwman
parents:
232
diff
changeset
|
84 elif self.Peak == 'var': |
6ed3d3a82a63
Added another c3 example. First import attempt
Windel Bouwman
parents:
232
diff
changeset
|
85 self.parseVarDef() |
306 | 86 # TODO handle variable initialization |
251
6ed3d3a82a63
Added another c3 example. First import attempt
Windel Bouwman
parents:
232
diff
changeset
|
87 elif self.Peak == 'const': |
6ed3d3a82a63
Added another c3 example. First import attempt
Windel Bouwman
parents:
232
diff
changeset
|
88 self.parseConstDef() |
6ed3d3a82a63
Added another c3 example. First import attempt
Windel Bouwman
parents:
232
diff
changeset
|
89 elif self.Peak == 'type': |
6ed3d3a82a63
Added another c3 example. First import attempt
Windel Bouwman
parents:
232
diff
changeset
|
90 self.parseTypeDef() |
6ed3d3a82a63
Added another c3 example. First import attempt
Windel Bouwman
parents:
232
diff
changeset
|
91 elif self.Peak == 'import': |
6ed3d3a82a63
Added another c3 example. First import attempt
Windel Bouwman
parents:
232
diff
changeset
|
92 self.parseImport() |
6ed3d3a82a63
Added another c3 example. First import attempt
Windel Bouwman
parents:
232
diff
changeset
|
93 else: |
6ed3d3a82a63
Added another c3 example. First import attempt
Windel Bouwman
parents:
232
diff
changeset
|
94 self.Error('Expected function, var, const or type') |
148 | 95 |
213 | 96 def parseDesignator(self): |
301 | 97 """ A designator designates an object with a name. """ |
225 | 98 name = self.Consume('ID') |
306 | 99 return Identifier(name.val, name.loc) |
100 | |
101 def parseIdSequence(self): | |
102 ids = [self.Consume('ID')] | |
103 while self.hasConsumed(','): | |
104 ids.append(self.Consume('ID')) | |
105 return ids | |
148 | 106 |
213 | 107 # Type system |
354 | 108 def PostFixId(self): |
109 pfe = self.PrimaryExpression_Id() | |
110 while self.Peak in ['.']: | |
111 if self.hasConsumed('.'): | |
112 field = self.Consume('ID') | |
113 pfe = Member(pfe, field.val, field.loc) | |
114 else: | |
115 raise Exception() | |
116 return pfe | |
117 | |
118 def PrimaryExpression_Id(self): | |
119 if self.Peak == 'ID': | |
120 return self.parseDesignator() | |
121 self.Error('Expected ID, got {0}'.format(self.Peak)) | |
122 | |
123 def parse_type_spec(self): | |
124 """ Parse type specification """ | |
213 | 125 if self.Peak == 'struct': |
126 self.Consume('struct') | |
127 self.Consume('{') | |
128 mems = [] | |
129 while self.Peak != '}': | |
354 | 130 mem_t = self.parse_type_spec() |
306 | 131 for i in self.parseIdSequence(): |
132 mems.append(StructField(i.val, mem_t)) | |
213 | 133 self.Consume(';') |
134 self.Consume('}') | |
296 | 135 theT = StructureType(mems) |
306 | 136 elif self.Peak == 'enum': |
137 # TODO) | |
138 raise NotImplementedError() | |
213 | 139 else: |
354 | 140 theT = self.PostFixId() |
141 | |
142 # Check for pointer or array suffix: | |
143 while self.Peak in ['*', '[']: | |
144 if self.hasConsumed('*'): | |
145 theT = PointerType(theT) | |
146 elif self.hasConsumed('['): | |
147 if self.Peak == ']': | |
148 size = 0 | |
149 self.Consume(']') | |
150 else: | |
151 size = self.Expression() | |
152 self.Consume(']') | |
153 theT = ArrayType(theT, size) | |
154 else: | |
155 raise Exception() | |
213 | 156 return theT |
157 | |
158 def parseTypeDef(self): | |
159 self.Consume('type') | |
354 | 160 newtype = self.parse_type_spec() |
213 | 161 typename = self.Consume('ID') |
162 self.Consume(';') | |
296 | 163 df = DefinedType(typename.val, newtype, typename.loc) |
225 | 164 self.addDeclaration(df) |
213 | 165 |
166 # Variable declarations: | |
167 def parseVarDef(self): | |
272 | 168 self.Consume('var') |
354 | 169 t = self.parse_type_spec() |
306 | 170 for name in self.parseIdSequence(): |
296 | 171 v = Variable(name.val, t) |
272 | 172 v.loc = name.loc |
173 self.addDeclaration(v) | |
174 self.Consume(';') | |
307 | 175 return Empty() |
148 | 176 |
213 | 177 def parseConstDef(self): |
280
02385f62f250
Rework from str interface to Instruction interface
Windel Bouwman
parents:
272
diff
changeset
|
178 self.Consume('const') |
354 | 179 t = self.parse_type_spec() |
306 | 180 while True: |
280
02385f62f250
Rework from str interface to Instruction interface
Windel Bouwman
parents:
272
diff
changeset
|
181 name = self.Consume('ID') |
02385f62f250
Rework from str interface to Instruction interface
Windel Bouwman
parents:
272
diff
changeset
|
182 self.Consume('=') |
02385f62f250
Rework from str interface to Instruction interface
Windel Bouwman
parents:
272
diff
changeset
|
183 val = self.Expression() |
296 | 184 c = Constant(name.val, t, val) |
313 | 185 self.addDeclaration(c) |
280
02385f62f250
Rework from str interface to Instruction interface
Windel Bouwman
parents:
272
diff
changeset
|
186 c.loc = name.loc |
306 | 187 if not self.hasConsumed(','): |
188 break | |
280
02385f62f250
Rework from str interface to Instruction interface
Windel Bouwman
parents:
272
diff
changeset
|
189 self.Consume(';') |
288 | 190 |
213 | 191 def parseFunctionDef(self): |
280
02385f62f250
Rework from str interface to Instruction interface
Windel Bouwman
parents:
272
diff
changeset
|
192 loc = self.Consume('function').loc |
354 | 193 returntype = self.parse_type_spec() |
280
02385f62f250
Rework from str interface to Instruction interface
Windel Bouwman
parents:
272
diff
changeset
|
194 fname = self.Consume('ID').val |
296 | 195 f = Function(fname, loc) |
280
02385f62f250
Rework from str interface to Instruction interface
Windel Bouwman
parents:
272
diff
changeset
|
196 self.addDeclaration(f) |
02385f62f250
Rework from str interface to Instruction interface
Windel Bouwman
parents:
272
diff
changeset
|
197 savePart = self.currentPart |
02385f62f250
Rework from str interface to Instruction interface
Windel Bouwman
parents:
272
diff
changeset
|
198 self.currentPart = f |
02385f62f250
Rework from str interface to Instruction interface
Windel Bouwman
parents:
272
diff
changeset
|
199 self.Consume('(') |
02385f62f250
Rework from str interface to Instruction interface
Windel Bouwman
parents:
272
diff
changeset
|
200 parameters = [] |
02385f62f250
Rework from str interface to Instruction interface
Windel Bouwman
parents:
272
diff
changeset
|
201 if not self.hasConsumed(')'): |
306 | 202 while True: |
354 | 203 typ = self.parse_type_spec() |
272 | 204 name = self.Consume('ID') |
296 | 205 param = FormalParameter(name.val, typ) |
272 | 206 param.loc = name.loc |
207 self.addDeclaration(param) | |
208 parameters.append(param) | |
306 | 209 if not self.hasConsumed(','): |
210 break | |
272 | 211 self.Consume(')') |
280
02385f62f250
Rework from str interface to Instruction interface
Windel Bouwman
parents:
272
diff
changeset
|
212 paramtypes = [p.typ for p in parameters] |
296 | 213 f.typ = FunctionType(paramtypes, returntype) |
307 | 214 f.body = self.parseCompound() |
280
02385f62f250
Rework from str interface to Instruction interface
Windel Bouwman
parents:
272
diff
changeset
|
215 self.currentPart = savePart |
148 | 216 |
354 | 217 def parse_if(self): |
272 | 218 loc = self.Consume('if').loc |
219 self.Consume('(') | |
220 condition = self.Expression() | |
221 self.Consume(')') | |
306 | 222 yes = self.Statement() |
307 | 223 no = self.Statement() if self.hasConsumed('else') else Empty() |
224 return If(condition, yes, no, loc) | |
148 | 225 |
307 | 226 def parseWhile(self): |
220
3f6c30a5d234
Major change in expression parsing to enable pointers and structs
Windel Bouwman
parents:
219
diff
changeset
|
227 loc = self.Consume('while').loc |
3f6c30a5d234
Major change in expression parsing to enable pointers and structs
Windel Bouwman
parents:
219
diff
changeset
|
228 self.Consume('(') |
3f6c30a5d234
Major change in expression parsing to enable pointers and structs
Windel Bouwman
parents:
219
diff
changeset
|
229 condition = self.Expression() |
3f6c30a5d234
Major change in expression parsing to enable pointers and structs
Windel Bouwman
parents:
219
diff
changeset
|
230 self.Consume(')') |
306 | 231 statements = self.Statement() |
307 | 232 return While(condition, statements, loc) |
149 | 233 |
354 | 234 def parse_for(self): |
315 | 235 loc = self.Consume('for').loc |
236 self.Consume('(') | |
237 init = self.Statement() | |
238 self.Consume(';') | |
239 condition = self.Expression() | |
240 self.Consume(';') | |
241 final = self.Statement() | |
242 self.Consume(')') | |
243 statements = self.Statement() | |
244 return For(init, condition, final, statements, loc) | |
245 | |
307 | 246 def parseReturn(self): |
220
3f6c30a5d234
Major change in expression parsing to enable pointers and structs
Windel Bouwman
parents:
219
diff
changeset
|
247 loc = self.Consume('return').loc |
280
02385f62f250
Rework from str interface to Instruction interface
Windel Bouwman
parents:
272
diff
changeset
|
248 if self.Peak == ';': |
296 | 249 expr = Literal(0, loc) |
280
02385f62f250
Rework from str interface to Instruction interface
Windel Bouwman
parents:
272
diff
changeset
|
250 else: |
02385f62f250
Rework from str interface to Instruction interface
Windel Bouwman
parents:
272
diff
changeset
|
251 expr = self.Expression() |
220
3f6c30a5d234
Major change in expression parsing to enable pointers and structs
Windel Bouwman
parents:
219
diff
changeset
|
252 self.Consume(';') |
307 | 253 return Return(expr, loc) |
148 | 254 |
307 | 255 def parseCompound(self): |
220
3f6c30a5d234
Major change in expression parsing to enable pointers and structs
Windel Bouwman
parents:
219
diff
changeset
|
256 self.Consume('{') |
3f6c30a5d234
Major change in expression parsing to enable pointers and structs
Windel Bouwman
parents:
219
diff
changeset
|
257 statements = [] |
3f6c30a5d234
Major change in expression parsing to enable pointers and structs
Windel Bouwman
parents:
219
diff
changeset
|
258 while not self.hasConsumed('}'): |
306 | 259 statements.append(self.Statement()) |
307 | 260 return Compound(statements) |
220
3f6c30a5d234
Major change in expression parsing to enable pointers and structs
Windel Bouwman
parents:
219
diff
changeset
|
261 |
3f6c30a5d234
Major change in expression parsing to enable pointers and structs
Windel Bouwman
parents:
219
diff
changeset
|
262 def Statement(self): |
3f6c30a5d234
Major change in expression parsing to enable pointers and structs
Windel Bouwman
parents:
219
diff
changeset
|
263 # Determine statement type based on the pending token: |
3f6c30a5d234
Major change in expression parsing to enable pointers and structs
Windel Bouwman
parents:
219
diff
changeset
|
264 if self.Peak == 'if': |
354 | 265 return self.parse_if() |
220
3f6c30a5d234
Major change in expression parsing to enable pointers and structs
Windel Bouwman
parents:
219
diff
changeset
|
266 elif self.Peak == 'while': |
307 | 267 return self.parseWhile() |
315 | 268 elif self.Peak == 'for': |
354 | 269 return self.parse_for() |
220
3f6c30a5d234
Major change in expression parsing to enable pointers and structs
Windel Bouwman
parents:
219
diff
changeset
|
270 elif self.Peak == '{': |
307 | 271 return self.parseCompound() |
220
3f6c30a5d234
Major change in expression parsing to enable pointers and structs
Windel Bouwman
parents:
219
diff
changeset
|
272 elif self.hasConsumed(';'): |
307 | 273 return Empty() |
220
3f6c30a5d234
Major change in expression parsing to enable pointers and structs
Windel Bouwman
parents:
219
diff
changeset
|
274 elif self.Peak == 'var': |
306 | 275 return self.parseVarDef() |
220
3f6c30a5d234
Major change in expression parsing to enable pointers and structs
Windel Bouwman
parents:
219
diff
changeset
|
276 elif self.Peak == 'return': |
307 | 277 return self.parseReturn() |
220
3f6c30a5d234
Major change in expression parsing to enable pointers and structs
Windel Bouwman
parents:
219
diff
changeset
|
278 else: |
306 | 279 x = self.UnaryExpression() |
280 if self.Peak == '=': | |
281 # We enter assignment mode here. | |
282 loc = self.Consume('=').loc | |
283 rhs = self.Expression() | |
284 return Assignment(x, rhs, loc) | |
285 else: | |
286 return ExpressionStatement(x, x.loc) | |
148 | 287 |
220
3f6c30a5d234
Major change in expression parsing to enable pointers and structs
Windel Bouwman
parents:
219
diff
changeset
|
288 # Expression section: |
3f6c30a5d234
Major change in expression parsing to enable pointers and structs
Windel Bouwman
parents:
219
diff
changeset
|
289 # We not implement these C constructs: |
3f6c30a5d234
Major change in expression parsing to enable pointers and structs
Windel Bouwman
parents:
219
diff
changeset
|
290 # a(2), f = 2 |
3f6c30a5d234
Major change in expression parsing to enable pointers and structs
Windel Bouwman
parents:
219
diff
changeset
|
291 # and this: |
3f6c30a5d234
Major change in expression parsing to enable pointers and structs
Windel Bouwman
parents:
219
diff
changeset
|
292 # a = 2 < x : 4 ? 1; |
3f6c30a5d234
Major change in expression parsing to enable pointers and structs
Windel Bouwman
parents:
219
diff
changeset
|
293 |
3f6c30a5d234
Major change in expression parsing to enable pointers and structs
Windel Bouwman
parents:
219
diff
changeset
|
294 def Expression(self): |
3f6c30a5d234
Major change in expression parsing to enable pointers and structs
Windel Bouwman
parents:
219
diff
changeset
|
295 exp = self.LogicalAndExpression() |
3f6c30a5d234
Major change in expression parsing to enable pointers and structs
Windel Bouwman
parents:
219
diff
changeset
|
296 while self.Peak == 'or': |
3f6c30a5d234
Major change in expression parsing to enable pointers and structs
Windel Bouwman
parents:
219
diff
changeset
|
297 loc = self.Consume('or').loc |
3f6c30a5d234
Major change in expression parsing to enable pointers and structs
Windel Bouwman
parents:
219
diff
changeset
|
298 e2 = self.LogicalAndExpression() |
296 | 299 exp = Binop(exp, 'or', e2, loc) |
220
3f6c30a5d234
Major change in expression parsing to enable pointers and structs
Windel Bouwman
parents:
219
diff
changeset
|
300 return exp |
148 | 301 |
220
3f6c30a5d234
Major change in expression parsing to enable pointers and structs
Windel Bouwman
parents:
219
diff
changeset
|
302 def LogicalAndExpression(self): |
3f6c30a5d234
Major change in expression parsing to enable pointers and structs
Windel Bouwman
parents:
219
diff
changeset
|
303 o = self.EqualityExpression() |
3f6c30a5d234
Major change in expression parsing to enable pointers and structs
Windel Bouwman
parents:
219
diff
changeset
|
304 while self.Peak == 'and': |
3f6c30a5d234
Major change in expression parsing to enable pointers and structs
Windel Bouwman
parents:
219
diff
changeset
|
305 loc = self.Consume('and').loc |
3f6c30a5d234
Major change in expression parsing to enable pointers and structs
Windel Bouwman
parents:
219
diff
changeset
|
306 o2 = self.EqualityExpression() |
296 | 307 o = Binop(o, 'and', o2, loc) |
220
3f6c30a5d234
Major change in expression parsing to enable pointers and structs
Windel Bouwman
parents:
219
diff
changeset
|
308 return o |
3f6c30a5d234
Major change in expression parsing to enable pointers and structs
Windel Bouwman
parents:
219
diff
changeset
|
309 |
3f6c30a5d234
Major change in expression parsing to enable pointers and structs
Windel Bouwman
parents:
219
diff
changeset
|
310 def EqualityExpression(self): |
3f6c30a5d234
Major change in expression parsing to enable pointers and structs
Windel Bouwman
parents:
219
diff
changeset
|
311 ee = self.SimpleExpression() |
300 | 312 while self.Peak in ['<', '==', '>', '>=', '<=', '!=']: |
220
3f6c30a5d234
Major change in expression parsing to enable pointers and structs
Windel Bouwman
parents:
219
diff
changeset
|
313 op = self.Consume(self.Peak) |
3f6c30a5d234
Major change in expression parsing to enable pointers and structs
Windel Bouwman
parents:
219
diff
changeset
|
314 ee2 = self.SimpleExpression() |
296 | 315 ee = Binop(ee, op.typ, ee2, op.loc) |
220
3f6c30a5d234
Major change in expression parsing to enable pointers and structs
Windel Bouwman
parents:
219
diff
changeset
|
316 return ee |
3f6c30a5d234
Major change in expression parsing to enable pointers and structs
Windel Bouwman
parents:
219
diff
changeset
|
317 |
3f6c30a5d234
Major change in expression parsing to enable pointers and structs
Windel Bouwman
parents:
219
diff
changeset
|
318 def SimpleExpression(self): |
232 | 319 """ Shift operations before + and - ? """ |
320 e = self.AddExpression() | |
321 while self.Peak in ['>>', '<<']: | |
322 op = self.Consume(self.Peak) | |
323 e2 = self.AddExpression() | |
296 | 324 e = Binop(e, op.typ, e2, op.loc) |
232 | 325 return e |
326 | |
327 def AddExpression(self): | |
220
3f6c30a5d234
Major change in expression parsing to enable pointers and structs
Windel Bouwman
parents:
219
diff
changeset
|
328 e = self.Term() |
3f6c30a5d234
Major change in expression parsing to enable pointers and structs
Windel Bouwman
parents:
219
diff
changeset
|
329 while self.Peak in ['+', '-']: |
3f6c30a5d234
Major change in expression parsing to enable pointers and structs
Windel Bouwman
parents:
219
diff
changeset
|
330 op = self.Consume(self.Peak) |
3f6c30a5d234
Major change in expression parsing to enable pointers and structs
Windel Bouwman
parents:
219
diff
changeset
|
331 e2 = self.Term() |
296 | 332 e = Binop(e, op.typ, e2, op.loc) |
220
3f6c30a5d234
Major change in expression parsing to enable pointers and structs
Windel Bouwman
parents:
219
diff
changeset
|
333 return e |
213 | 334 |
220
3f6c30a5d234
Major change in expression parsing to enable pointers and structs
Windel Bouwman
parents:
219
diff
changeset
|
335 def Term(self): |
221 | 336 t = self.BitwiseOr() |
220
3f6c30a5d234
Major change in expression parsing to enable pointers and structs
Windel Bouwman
parents:
219
diff
changeset
|
337 while self.Peak in ['*', '/']: |
3f6c30a5d234
Major change in expression parsing to enable pointers and structs
Windel Bouwman
parents:
219
diff
changeset
|
338 op = self.Consume(self.Peak) |
221 | 339 t2 = self.BitwiseOr() |
296 | 340 t = Binop(t, op.typ, t2, op.loc) |
220
3f6c30a5d234
Major change in expression parsing to enable pointers and structs
Windel Bouwman
parents:
219
diff
changeset
|
341 return t |
221 | 342 |
343 def BitwiseOr(self): | |
344 a = self.BitwiseAnd() | |
306 | 345 while self.Peak == '|': |
221 | 346 op = self.Consume(self.Peak) |
347 b = self.BitwiseAnd() | |
296 | 348 a = Binop(a, op.typ, b, op.loc) |
221 | 349 return a |
350 | |
351 def BitwiseAnd(self): | |
352 a = self.CastExpression() | |
306 | 353 while self.Peak == '&': |
221 | 354 op = self.Consume(self.Peak) |
355 b = self.CastExpression() | |
296 | 356 a = Binop(a, op.typ, b, op.loc) |
221 | 357 return a |
220
3f6c30a5d234
Major change in expression parsing to enable pointers and structs
Windel Bouwman
parents:
219
diff
changeset
|
358 |
3f6c30a5d234
Major change in expression parsing to enable pointers and structs
Windel Bouwman
parents:
219
diff
changeset
|
359 # Domain of unary expressions: |
3f6c30a5d234
Major change in expression parsing to enable pointers and structs
Windel Bouwman
parents:
219
diff
changeset
|
360 |
3f6c30a5d234
Major change in expression parsing to enable pointers and structs
Windel Bouwman
parents:
219
diff
changeset
|
361 def CastExpression(self): |
228 | 362 """ |
363 the C-style type cast conflicts with '(' expr ')' | |
364 so introduce extra keyword 'cast' | |
365 """ | |
221 | 366 if self.Peak == 'cast': |
230 | 367 loc = self.Consume('cast').loc |
221 | 368 self.Consume('<') |
354 | 369 t = self.parse_type_spec() |
221 | 370 self.Consume('>') |
371 self.Consume('(') | |
232 | 372 ce = self.Expression() |
220
3f6c30a5d234
Major change in expression parsing to enable pointers and structs
Windel Bouwman
parents:
219
diff
changeset
|
373 self.Consume(')') |
296 | 374 return TypeCast(t, ce, loc) |
220
3f6c30a5d234
Major change in expression parsing to enable pointers and structs
Windel Bouwman
parents:
219
diff
changeset
|
375 else: |
3f6c30a5d234
Major change in expression parsing to enable pointers and structs
Windel Bouwman
parents:
219
diff
changeset
|
376 return self.UnaryExpression() |
230 | 377 |
220
3f6c30a5d234
Major change in expression parsing to enable pointers and structs
Windel Bouwman
parents:
219
diff
changeset
|
378 def UnaryExpression(self): |
3f6c30a5d234
Major change in expression parsing to enable pointers and structs
Windel Bouwman
parents:
219
diff
changeset
|
379 if self.Peak in ['&', '*']: |
3f6c30a5d234
Major change in expression parsing to enable pointers and structs
Windel Bouwman
parents:
219
diff
changeset
|
380 op = self.Consume(self.Peak) |
3f6c30a5d234
Major change in expression parsing to enable pointers and structs
Windel Bouwman
parents:
219
diff
changeset
|
381 ce = self.CastExpression() |
225 | 382 if op.val == '*': |
296 | 383 return Deref(ce, op.loc) |
225 | 384 else: |
296 | 385 return Unop(op.typ, ce, op.loc) |
220
3f6c30a5d234
Major change in expression parsing to enable pointers and structs
Windel Bouwman
parents:
219
diff
changeset
|
386 else: |
3f6c30a5d234
Major change in expression parsing to enable pointers and structs
Windel Bouwman
parents:
219
diff
changeset
|
387 return self.PostFixExpression() |
3f6c30a5d234
Major change in expression parsing to enable pointers and structs
Windel Bouwman
parents:
219
diff
changeset
|
388 |
3f6c30a5d234
Major change in expression parsing to enable pointers and structs
Windel Bouwman
parents:
219
diff
changeset
|
389 def PostFixExpression(self): |
3f6c30a5d234
Major change in expression parsing to enable pointers and structs
Windel Bouwman
parents:
219
diff
changeset
|
390 pfe = self.PrimaryExpression() |
315 | 391 while self.Peak in ['[', '.', '->', '(', '++']: |
308 | 392 if self.hasConsumed('['): |
354 | 393 i = self.Expression() |
394 self.Consume(']') | |
395 pfe = Index(pfe, i, i.loc) | |
308 | 396 elif self.hasConsumed('->'): |
397 field = self.Consume('ID') | |
398 pfe = Deref(pfe, pfe.loc) | |
399 pfe = Member(pfe, field.val, field.loc) | |
400 elif self.hasConsumed('.'): | |
401 field = self.Consume('ID') | |
402 pfe = Member(pfe, field.val, field.loc) | |
315 | 403 elif self.Peak == '++': |
404 loc = self.Consume('++').loc | |
405 pfe = Unop('++', pfe, loc) | |
308 | 406 elif self.hasConsumed('('): |
407 # Function call | |
408 args = [] | |
409 if not self.hasConsumed(')'): | |
220
3f6c30a5d234
Major change in expression parsing to enable pointers and structs
Windel Bouwman
parents:
219
diff
changeset
|
410 args.append(self.Expression()) |
308 | 411 while self.hasConsumed(','): |
412 args.append(self.Expression()) | |
413 self.Consume(')') | |
414 pfe = FunctionCall(pfe, args, pfe.loc) | |
415 else: | |
416 raise Exception() | |
220
3f6c30a5d234
Major change in expression parsing to enable pointers and structs
Windel Bouwman
parents:
219
diff
changeset
|
417 return pfe |
3f6c30a5d234
Major change in expression parsing to enable pointers and structs
Windel Bouwman
parents:
219
diff
changeset
|
418 |
3f6c30a5d234
Major change in expression parsing to enable pointers and structs
Windel Bouwman
parents:
219
diff
changeset
|
419 def PrimaryExpression(self): |
3f6c30a5d234
Major change in expression parsing to enable pointers and structs
Windel Bouwman
parents:
219
diff
changeset
|
420 if self.hasConsumed('('): |
3f6c30a5d234
Major change in expression parsing to enable pointers and structs
Windel Bouwman
parents:
219
diff
changeset
|
421 e = self.Expression() |
3f6c30a5d234
Major change in expression parsing to enable pointers and structs
Windel Bouwman
parents:
219
diff
changeset
|
422 self.Consume(')') |
3f6c30a5d234
Major change in expression parsing to enable pointers and structs
Windel Bouwman
parents:
219
diff
changeset
|
423 return e |
3f6c30a5d234
Major change in expression parsing to enable pointers and structs
Windel Bouwman
parents:
219
diff
changeset
|
424 elif self.Peak == 'NUMBER': |
3f6c30a5d234
Major change in expression parsing to enable pointers and structs
Windel Bouwman
parents:
219
diff
changeset
|
425 val = self.Consume('NUMBER') |
296 | 426 return Literal(val.val, val.loc) |
220
3f6c30a5d234
Major change in expression parsing to enable pointers and structs
Windel Bouwman
parents:
219
diff
changeset
|
427 elif self.Peak == 'REAL': |
3f6c30a5d234
Major change in expression parsing to enable pointers and structs
Windel Bouwman
parents:
219
diff
changeset
|
428 val = self.Consume('REAL') |
296 | 429 return Literal(val.val, val.loc) |
220
3f6c30a5d234
Major change in expression parsing to enable pointers and structs
Windel Bouwman
parents:
219
diff
changeset
|
430 elif self.Peak == 'true': |
3f6c30a5d234
Major change in expression parsing to enable pointers and structs
Windel Bouwman
parents:
219
diff
changeset
|
431 val = self.Consume('true') |
296 | 432 return Literal(True, val.loc) |
220
3f6c30a5d234
Major change in expression parsing to enable pointers and structs
Windel Bouwman
parents:
219
diff
changeset
|
433 elif self.Peak == 'false': |
3f6c30a5d234
Major change in expression parsing to enable pointers and structs
Windel Bouwman
parents:
219
diff
changeset
|
434 val = self.Consume('false') |
296 | 435 return Literal(False, val.loc) |
311 | 436 elif self.Peak == 'STRING': |
437 val = self.Consume('STRING') | |
438 return Literal(val.val, val.loc) | |
220
3f6c30a5d234
Major change in expression parsing to enable pointers and structs
Windel Bouwman
parents:
219
diff
changeset
|
439 elif self.Peak == 'ID': |
306 | 440 return self.parseDesignator() |
220
3f6c30a5d234
Major change in expression parsing to enable pointers and structs
Windel Bouwman
parents:
219
diff
changeset
|
441 self.Error('Expected NUM, ID or (expr), got {0}'.format(self.Peak)) |
354 | 442 |