Mercurial > lcfOS
annotate python/c3/typecheck.py @ 225:1c7364bd74c7
Fixed pointer deref
author | Windel Bouwman |
---|---|
date | Thu, 11 Jul 2013 07:42:30 +0200 |
parents | c3f1ce8b638f |
children | 240111e0456f |
rev | line source |
---|---|
163 | 1 from .astnodes import * |
2 from .scope import * | |
3 from .visitor import Visitor | |
4 | |
5 def equalTypes(a, b): | |
6 """ Compare types a and b for equality. Not equal until proven otherwise. """ | |
7 if type(a) is type(b): | |
8 if type(a) is BaseType: | |
9 return a.name == b.name | |
220
3f6c30a5d234
Major change in expression parsing to enable pointers and structs
Windel Bouwman
parents:
215
diff
changeset
|
10 elif type(a) is PointerType: |
3f6c30a5d234
Major change in expression parsing to enable pointers and structs
Windel Bouwman
parents:
215
diff
changeset
|
11 return equalTypes(a.ptype, b.ptype) |
163 | 12 return False |
150 | 13 |
222 | 14 def canCast(fromT, toT): |
15 if isinstance(fromT, PointerType) and isinstance(toT, PointerType): | |
16 return True | |
17 elif fromT is intType and isinstance(toT, PointerType): | |
18 return True | |
19 return False | |
20 | |
150 | 21 class TypeChecker: |
220
3f6c30a5d234
Major change in expression parsing to enable pointers and structs
Windel Bouwman
parents:
215
diff
changeset
|
22 def __init__(self, diag): |
3f6c30a5d234
Major change in expression parsing to enable pointers and structs
Windel Bouwman
parents:
215
diff
changeset
|
23 self.diag = diag |
3f6c30a5d234
Major change in expression parsing to enable pointers and structs
Windel Bouwman
parents:
215
diff
changeset
|
24 |
3f6c30a5d234
Major change in expression parsing to enable pointers and structs
Windel Bouwman
parents:
215
diff
changeset
|
25 def error(self, msg, loc): |
186 | 26 """ Wrapper that registers the message and marks the result invalid """ |
27 self.diag.error(msg, loc) | |
28 self.ok = False | |
220
3f6c30a5d234
Major change in expression parsing to enable pointers and structs
Windel Bouwman
parents:
215
diff
changeset
|
29 |
3f6c30a5d234
Major change in expression parsing to enable pointers and structs
Windel Bouwman
parents:
215
diff
changeset
|
30 def checkPackage(self, pkg): |
186 | 31 self.ok = True |
215 | 32 visitor = Visitor() |
33 visitor.visit(pkg, f_post=self.check2) | |
186 | 34 return self.ok |
220
3f6c30a5d234
Major change in expression parsing to enable pointers and structs
Windel Bouwman
parents:
215
diff
changeset
|
35 |
3f6c30a5d234
Major change in expression parsing to enable pointers and structs
Windel Bouwman
parents:
215
diff
changeset
|
36 def check2(self, sym): |
225 | 37 if type(sym) in [IfStatement, WhileStatement]: |
38 if not equalTypes(sym.condition.typ, boolType): | |
39 self.error('Condition must be of type {0}'.format(boolType), sym.condition.loc) | |
40 elif type(sym) is Assignment: | |
221 | 41 if not equalTypes(sym.lval.typ, sym.rval.typ): |
42 self.error('Cannot assign {0} to {1}'.format(sym.rval.typ, sym.lval.typ), sym.loc) | |
43 if not sym.lval.lvalue: | |
44 self.error('No valid lvalue {}'.format(sym.lval), sym.lval.loc) | |
45 #if sym.rval.lvalue: | |
46 # self.error('Right hand side must be an rvalue', sym.rval.loc) | |
225 | 47 elif type(sym) is ReturnStatement: |
48 pass | |
49 elif type(sym) is FunctionCall: | |
166 | 50 if sym.proc: |
167 | 51 # Check arguments: |
52 ngiv = len(sym.args) | |
53 ptypes = sym.proc.typ.parametertypes | |
54 nreq = len(ptypes) | |
55 if ngiv != nreq: | |
186 | 56 self.error('Function {2}: {0} arguments required, {1} given'.format(nreq, ngiv, sym.proc.name), sym.loc) |
167 | 57 else: |
58 for a, at in zip(sym.args, ptypes): | |
59 if not equalTypes(a.typ, at): | |
186 | 60 self.error('Got {0}, expected {1}'.format(a.typ, at), a.loc) |
167 | 61 # determine return type: |
62 sym.typ = sym.proc.typ.returntype | |
63 else: | |
64 sym.typ = intType | |
225 | 65 elif type(sym) is VariableUse: |
221 | 66 sym.lvalue = True |
225 | 67 if type(sym.target) is Variable: |
221 | 68 sym.typ = sym.target.typ |
69 else: | |
225 | 70 print('warning {} has no target, defaulting to int'.format(sym)) |
221 | 71 sym.typ = intType |
225 | 72 elif type(sym) is Literal: |
221 | 73 sym.lvalue = False |
74 if type(sym.val) is int: | |
75 sym.typ = intType | |
76 elif type(sym.val) is float: | |
77 sym.typ = doubleType | |
78 elif type(sym.val) is bool: | |
79 sym.typ = boolType | |
80 else: | |
225 | 81 raise Exception('Unknown literal type'.format(sym.val)) |
82 elif type(sym) is Unop: | |
220
3f6c30a5d234
Major change in expression parsing to enable pointers and structs
Windel Bouwman
parents:
215
diff
changeset
|
83 if sym.op == '&': |
3f6c30a5d234
Major change in expression parsing to enable pointers and structs
Windel Bouwman
parents:
215
diff
changeset
|
84 sym.typ = PointerType(sym.a.typ) |
221 | 85 sym.lvalue = False |
225 | 86 else: |
87 raise Exception('Unknown unop {0}'.format(sym.op)) | |
88 elif type(sym) is Deref: | |
89 # pointer deref | |
90 sym.lvalue = True | |
91 # check if the to be dereferenced variable is a pointer type: | |
92 if type(sym.ptr.typ) is PointerType: | |
93 sym.typ = sym.ptr.typ.ptype | |
94 else: | |
95 self.error('Cannot dereference non-pointer type {}'.format(sym.ptr.typ), sym.loc) | |
96 sym.typ = intType | |
97 elif type(sym) is FieldRef: | |
98 basetype = sym.base.typ | |
99 sym.lvalue = True | |
100 if type(basetype) is StructureType: | |
101 if basetype.hasField(sym.field): | |
102 sym.typ = basetype.fieldType(sym.field) | |
220
3f6c30a5d234
Major change in expression parsing to enable pointers and structs
Windel Bouwman
parents:
215
diff
changeset
|
103 else: |
225 | 104 self.error('{} does not contain field {}'.format(basetype, symfield), sym.loc) |
221 | 105 sym.typ = intType |
220
3f6c30a5d234
Major change in expression parsing to enable pointers and structs
Windel Bouwman
parents:
215
diff
changeset
|
106 else: |
225 | 107 self.error('Cannot select field {} of non-structure type {}'.format(sym.field, basetype), sym.loc) |
108 sym.typ = intType | |
109 elif type(sym) is Binop: | |
221 | 110 sym.lvalue = False |
163 | 111 if sym.op in ['+', '-', '*', '/']: |
112 if equalTypes(sym.a.typ, sym.b.typ): | |
170 | 113 if equalTypes(sym.a.typ, intType): |
114 sym.typ = sym.a.typ | |
115 else: | |
186 | 116 self.error('Can only add integers', sym.loc) |
170 | 117 sym.typ = intType |
163 | 118 else: |
165 | 119 # assume void here? TODO: throw exception! |
120 sym.typ = intType | |
225 | 121 self.error('Types unequal {} != {}'.format(sym.a.typ, sym.b.typ), sym.loc) |
170 | 122 elif sym.op in ['>', '<', '==', '<=', '>=']: |
165 | 123 sym.typ = boolType |
124 if not equalTypes(sym.a.typ, sym.b.typ): | |
225 | 125 self.error('Types unequal {} != {}'.format(sym.a.typ, sym.b.typ), sym.loc) |
166 | 126 elif sym.op in ['or', 'and']: |
127 sym.typ = boolType | |
128 if not equalTypes(sym.a.typ, boolType): | |
186 | 129 self.error('Must be {0}'.format(boolType), sym.a.loc) |
166 | 130 if not equalTypes(sym.b.typ, boolType): |
186 | 131 self.error('Must be {0}'.format(boolType), sym.b.loc) |
221 | 132 elif sym.op in ['|', '&']: |
133 sym.typ = intType | |
225 | 134 raise Exception('Not implemented') |
221 | 135 # TODO: elaborate? |
163 | 136 else: |
225 | 137 raise Exception('Unknown binop {0}'.format(sym.op)) |
138 elif type(sym) is Variable: | |
165 | 139 # check initial value type: |
140 # TODO | |
141 pass | |
225 | 142 elif type(sym) is TypeCast: |
222 | 143 if canCast(sym.a.typ, sym.to_type): |
144 sym.typ = sym.to_type | |
145 else: | |
146 self.error('Cannot cast {} to {}'.format(sym.a.typ, sym.to_type)) | |
225 | 147 elif type(sym) is Constant: |
148 if not equalTypes(sym.typ, sym.value.typ): | |
149 self.error('Cannot assign {0} to {1}'.format(sym.value.typ, sym.typ), sym.loc) | |
150 elif type(sym) in [EmptyStatement, CompoundStatement, Package, Function, FunctionType, ExpressionStatement, DefinedType]: | |
165 | 151 pass |
225 | 152 else: |
186 | 153 raise Exception('Unknown type check {0}'.format(sym)) |
150 | 154 |