Mercurial > projects > dang
annotate sema/Visitor.d @ 37:858b9805843d new_gen
Bug-fixes
Void can now be used and is recognized as a keyword by lexer
Fixed a problem with casting on pointer types
The expression is now optional for a ReturnStmt (only legal in void funcs)
| author | Anders Halager <halager@gmail.com> |
|---|---|
| date | Sun, 20 Apr 2008 23:53:05 +0200 |
| parents | ce17bea8e9bd |
| children | 495188f9078e |
| rev | line source |
|---|---|
| 1 | 1 module sema.Visitor; |
| 2 | |
| 3 import tango.io.Stdout; | |
| 4 | |
| 5 public | |
| 6 import ast.Decl, | |
| 7 ast.Stmt, | |
| 8 ast.Exp; | |
| 9 | |
| 10 import lexer.Token; | |
| 11 | |
| 12 class Visitor(FinalT = int, DeclT = FinalT, StmtT = DeclT, ExpT = StmtT) | |
| 13 { | |
| 14 public: | |
| 15 FinalT visit(Decl[] decls) | |
| 16 { | |
| 17 foreach (decl; decls) | |
| 18 visitDecl(decl); | |
| 19 static if (is(FinalT == void)) | |
| 20 return; | |
| 21 else | |
| 22 return FinalT.init; | |
| 23 } | |
| 24 | |
| 25 DeclT visitDecl(Decl decl) | |
| 26 { | |
| 27 switch(decl.declType) | |
| 28 { | |
| 29 case DeclType.FuncDecl: | |
| 30 return visitFuncDecl(cast(FuncDecl)decl); | |
| 31 case DeclType.VarDecl: | |
| 32 return visitVarDecl(cast(VarDecl)decl); | |
| 22 | 33 case DeclType.StructDecl: |
| 34 return visitStructDecl(cast(StructDecl)decl); | |
| 1 | 35 default: |
| 36 throw new Exception("Unknown declaration type"); | |
| 37 } | |
| 38 } | |
| 39 | |
| 40 StmtT visitStmt(Stmt stmt) | |
| 41 { | |
| 42 switch(stmt.stmtType) | |
| 43 { | |
| 44 case StmtType.Return: | |
| 45 return visitReturnStmt(cast(ReturnStmt)stmt); | |
| 46 case StmtType.Decl: | |
| 47 return visitDeclStmt(cast(DeclStmt)stmt); | |
| 48 case StmtType.Exp: | |
| 49 return visitExpStmt(cast(ExpStmt)stmt); | |
|
5
2c5a8f4c254a
Added very simple if support.
Anders Halager <halager@gmail.com>
parents:
1
diff
changeset
|
50 case StmtType.If: |
|
2c5a8f4c254a
Added very simple if support.
Anders Halager <halager@gmail.com>
parents:
1
diff
changeset
|
51 return visitIfStmt(cast(IfStmt)stmt); |
|
11
642c6a998fd9
Support for while statements and fixed scope for if
Anders Halager <halager@gmail.com>
parents:
5
diff
changeset
|
52 case StmtType.While: |
|
642c6a998fd9
Support for while statements and fixed scope for if
Anders Halager <halager@gmail.com>
parents:
5
diff
changeset
|
53 return visitWhileStmt(cast(WhileStmt)stmt); |
|
36
ce17bea8e9bd
Switch statements support
Anders Halager <halager@gmail.com>
parents:
28
diff
changeset
|
54 case StmtType.Switch: |
|
ce17bea8e9bd
Switch statements support
Anders Halager <halager@gmail.com>
parents:
28
diff
changeset
|
55 return visitSwitchStmt(cast(SwitchStmt)stmt); |
| 1 | 56 default: |
| 57 throw new Exception("Unknown statement type"); | |
| 58 } | |
| 59 } | |
| 60 | |
| 61 ExpT visitExp(Exp exp) | |
| 62 { | |
| 63 switch(exp.expType) | |
| 64 { | |
| 65 case ExpType.Binary: | |
| 66 return visitBinaryExp(cast(BinaryExp)exp); | |
| 67 case ExpType.IntegerLit: | |
| 68 return visitIntegerLit(cast(IntegerLit)exp); | |
| 69 case ExpType.Negate: | |
| 70 return visitNegateExp(cast(NegateExp)exp); | |
| 71 case ExpType.AssignExp: | |
| 72 return visitAssignExp(cast(AssignExp)exp); | |
| 73 case ExpType.CallExp: | |
| 74 return visitCallExp(cast(CallExp)exp); | |
| 75 case ExpType.Identifier: | |
| 76 return visitIdentifier(cast(Identifier)exp); | |
|
28
69464d465284
Now supporting structs - both read and write. Still a few errors though, so watch out.
Anders Johnsen <skabet@gmail.com>
parents:
22
diff
changeset
|
77 case ExpType.MemberLookup: |
|
69464d465284
Now supporting structs - both read and write. Still a few errors though, so watch out.
Anders Johnsen <skabet@gmail.com>
parents:
22
diff
changeset
|
78 return visitMemberLookup(cast(MemberLookup)exp); |
| 1 | 79 default: |
| 80 throw new Exception("Unknown expression type"); | |
| 81 } | |
| 82 } | |
| 83 | |
| 84 // Declarations: | |
| 85 DeclT visitVarDecl(VarDecl d) | |
| 86 { | |
| 87 visitExp(d.type); | |
| 88 visitExp(d.identifier); | |
| 89 if (d.init) | |
| 90 visitExp(d.init); | |
| 91 | |
| 92 static if (is(DeclT == void)) | |
| 93 return; | |
| 94 else | |
| 95 return DeclT.init; | |
| 96 } | |
| 97 | |
| 98 DeclT visitFuncDecl(FuncDecl f) | |
| 99 { | |
| 100 visitExp(f.type); | |
| 101 visitExp(f.identifier); | |
| 102 foreach (arg; f.funcArgs) | |
| 103 visitDecl(arg); | |
| 104 foreach (stmt; f.statements) | |
| 105 visitStmt(stmt); | |
| 106 | |
| 107 static if (is(DeclT == void)) | |
| 108 return; | |
| 109 else | |
| 110 return DeclT.init; | |
| 111 } | |
| 112 | |
| 22 | 113 DeclT visitStructDecl(StructDecl s) |
| 114 { | |
| 115 visitExp(s.identifier); | |
| 116 | |
| 117 foreach (arg; s.vars) | |
| 118 visitDecl(arg); | |
| 119 | |
| 120 static if (is(DeclT == void)) | |
| 121 return; | |
| 122 else | |
| 123 return DeclT.init; | |
| 124 } | |
| 125 | |
| 1 | 126 // Statements: |
| 127 StmtT visitReturnStmt(ReturnStmt s) | |
| 128 { | |
| 37 | 129 if (s.exp) |
| 130 visitExp(s.exp); | |
| 1 | 131 static if (is(StmtT == void)) |
| 132 return; | |
| 133 else | |
| 134 return StmtT.init; | |
| 135 } | |
| 136 | |
| 137 StmtT visitDeclStmt(DeclStmt d) | |
| 138 { | |
| 139 visitDecl(d.decl); | |
| 140 static if (is(StmtT == void)) | |
| 141 return; | |
| 142 else | |
| 143 return StmtT.init; | |
| 144 } | |
| 145 | |
|
5
2c5a8f4c254a
Added very simple if support.
Anders Halager <halager@gmail.com>
parents:
1
diff
changeset
|
146 StmtT visitIfStmt(IfStmt s) |
|
2c5a8f4c254a
Added very simple if support.
Anders Halager <halager@gmail.com>
parents:
1
diff
changeset
|
147 { |
|
2c5a8f4c254a
Added very simple if support.
Anders Halager <halager@gmail.com>
parents:
1
diff
changeset
|
148 visitExp(s.cond); |
|
11
642c6a998fd9
Support for while statements and fixed scope for if
Anders Halager <halager@gmail.com>
parents:
5
diff
changeset
|
149 foreach (stmt; s.then_body) |
|
642c6a998fd9
Support for while statements and fixed scope for if
Anders Halager <halager@gmail.com>
parents:
5
diff
changeset
|
150 visitStmt(stmt); |
|
642c6a998fd9
Support for while statements and fixed scope for if
Anders Halager <halager@gmail.com>
parents:
5
diff
changeset
|
151 foreach (stmt; s.else_body) |
|
642c6a998fd9
Support for while statements and fixed scope for if
Anders Halager <halager@gmail.com>
parents:
5
diff
changeset
|
152 visitStmt(stmt); |
|
642c6a998fd9
Support for while statements and fixed scope for if
Anders Halager <halager@gmail.com>
parents:
5
diff
changeset
|
153 static if (is(StmtT == void)) |
|
642c6a998fd9
Support for while statements and fixed scope for if
Anders Halager <halager@gmail.com>
parents:
5
diff
changeset
|
154 return; |
|
642c6a998fd9
Support for while statements and fixed scope for if
Anders Halager <halager@gmail.com>
parents:
5
diff
changeset
|
155 else |
|
642c6a998fd9
Support for while statements and fixed scope for if
Anders Halager <halager@gmail.com>
parents:
5
diff
changeset
|
156 return StmtT.init; |
|
642c6a998fd9
Support for while statements and fixed scope for if
Anders Halager <halager@gmail.com>
parents:
5
diff
changeset
|
157 } |
|
642c6a998fd9
Support for while statements and fixed scope for if
Anders Halager <halager@gmail.com>
parents:
5
diff
changeset
|
158 |
|
642c6a998fd9
Support for while statements and fixed scope for if
Anders Halager <halager@gmail.com>
parents:
5
diff
changeset
|
159 StmtT visitWhileStmt(WhileStmt s) |
|
642c6a998fd9
Support for while statements and fixed scope for if
Anders Halager <halager@gmail.com>
parents:
5
diff
changeset
|
160 { |
|
642c6a998fd9
Support for while statements and fixed scope for if
Anders Halager <halager@gmail.com>
parents:
5
diff
changeset
|
161 visitExp(s.cond); |
|
642c6a998fd9
Support for while statements and fixed scope for if
Anders Halager <halager@gmail.com>
parents:
5
diff
changeset
|
162 foreach (stmt; s.stmts) |
|
5
2c5a8f4c254a
Added very simple if support.
Anders Halager <halager@gmail.com>
parents:
1
diff
changeset
|
163 visitStmt(stmt); |
|
2c5a8f4c254a
Added very simple if support.
Anders Halager <halager@gmail.com>
parents:
1
diff
changeset
|
164 static if (is(StmtT == void)) |
|
2c5a8f4c254a
Added very simple if support.
Anders Halager <halager@gmail.com>
parents:
1
diff
changeset
|
165 return; |
|
2c5a8f4c254a
Added very simple if support.
Anders Halager <halager@gmail.com>
parents:
1
diff
changeset
|
166 else |
|
2c5a8f4c254a
Added very simple if support.
Anders Halager <halager@gmail.com>
parents:
1
diff
changeset
|
167 return StmtT.init; |
|
2c5a8f4c254a
Added very simple if support.
Anders Halager <halager@gmail.com>
parents:
1
diff
changeset
|
168 } |
|
2c5a8f4c254a
Added very simple if support.
Anders Halager <halager@gmail.com>
parents:
1
diff
changeset
|
169 |
|
36
ce17bea8e9bd
Switch statements support
Anders Halager <halager@gmail.com>
parents:
28
diff
changeset
|
170 StmtT visitSwitchStmt(SwitchStmt s) |
|
ce17bea8e9bd
Switch statements support
Anders Halager <halager@gmail.com>
parents:
28
diff
changeset
|
171 { |
|
ce17bea8e9bd
Switch statements support
Anders Halager <halager@gmail.com>
parents:
28
diff
changeset
|
172 visitExp(s.cond); |
|
ce17bea8e9bd
Switch statements support
Anders Halager <halager@gmail.com>
parents:
28
diff
changeset
|
173 foreach(stmt; s.defaultBlock) |
|
ce17bea8e9bd
Switch statements support
Anders Halager <halager@gmail.com>
parents:
28
diff
changeset
|
174 visitStmt(stmt); |
|
ce17bea8e9bd
Switch statements support
Anders Halager <halager@gmail.com>
parents:
28
diff
changeset
|
175 foreach (c; s.cases) |
|
ce17bea8e9bd
Switch statements support
Anders Halager <halager@gmail.com>
parents:
28
diff
changeset
|
176 { |
|
ce17bea8e9bd
Switch statements support
Anders Halager <halager@gmail.com>
parents:
28
diff
changeset
|
177 foreach(lit; c.values) |
|
ce17bea8e9bd
Switch statements support
Anders Halager <halager@gmail.com>
parents:
28
diff
changeset
|
178 visitIntegerLit(lit); |
|
ce17bea8e9bd
Switch statements support
Anders Halager <halager@gmail.com>
parents:
28
diff
changeset
|
179 foreach(stmt; c.stmts) |
|
ce17bea8e9bd
Switch statements support
Anders Halager <halager@gmail.com>
parents:
28
diff
changeset
|
180 visitStmt(stmt); |
|
ce17bea8e9bd
Switch statements support
Anders Halager <halager@gmail.com>
parents:
28
diff
changeset
|
181 } |
|
ce17bea8e9bd
Switch statements support
Anders Halager <halager@gmail.com>
parents:
28
diff
changeset
|
182 static if (is(StmtT == void)) |
|
ce17bea8e9bd
Switch statements support
Anders Halager <halager@gmail.com>
parents:
28
diff
changeset
|
183 return; |
|
ce17bea8e9bd
Switch statements support
Anders Halager <halager@gmail.com>
parents:
28
diff
changeset
|
184 else |
|
ce17bea8e9bd
Switch statements support
Anders Halager <halager@gmail.com>
parents:
28
diff
changeset
|
185 return StmtT.init; |
|
ce17bea8e9bd
Switch statements support
Anders Halager <halager@gmail.com>
parents:
28
diff
changeset
|
186 } |
|
ce17bea8e9bd
Switch statements support
Anders Halager <halager@gmail.com>
parents:
28
diff
changeset
|
187 |
| 1 | 188 StmtT visitExpStmt(ExpStmt s) |
| 189 { | |
| 190 visitExp(s.exp); | |
| 191 static if (is(StmtT == void)) | |
| 192 return; | |
| 193 else | |
| 194 return StmtT.init; | |
| 195 } | |
| 196 | |
| 197 // Expressions: | |
| 198 ExpT visitAssignExp(AssignExp exp) | |
| 199 { | |
| 200 visitExp(exp.identifier); | |
| 201 visitExp(exp.exp); | |
| 202 static if (is(ExpT == void)) | |
| 203 return; | |
| 204 else | |
| 205 return ExpT.init; | |
| 206 } | |
| 207 | |
| 208 ExpT visitBinaryExp(BinaryExp exp) | |
| 209 { | |
| 210 visitExp(exp.left); | |
| 211 visitExp(exp.right); | |
| 212 static if (is(ExpT == void)) | |
| 213 return; | |
| 214 else | |
| 215 return ExpT.init; | |
| 216 } | |
| 217 | |
| 218 ExpT visitCallExp(CallExp exp) | |
| 219 { | |
| 220 visitExp(exp.exp); | |
| 221 foreach (arg; exp.args) | |
| 222 visitExp(arg); | |
| 223 static if (is(ExpT == void)) | |
| 224 return; | |
| 225 else | |
| 226 return ExpT.init; | |
| 227 } | |
| 228 | |
| 229 ExpT visitNegateExp(NegateExp exp) | |
| 230 { | |
| 231 visitExp(exp.exp); | |
| 232 static if (is(ExpT == void)) | |
| 233 return; | |
| 234 else | |
| 235 return ExpT.init; | |
| 236 } | |
| 237 | |
| 238 ExpT visitIntegerLit(IntegerLit exp) | |
| 239 { | |
| 240 static if (is(ExpT == void)) | |
| 241 return; | |
| 242 else | |
| 243 return ExpT.init; | |
| 244 } | |
| 245 | |
| 246 ExpT visitIdentifier(Identifier exp) | |
| 247 { | |
| 248 static if (is(ExpT == void)) | |
| 249 return; | |
| 250 else | |
| 251 return ExpT.init; | |
| 252 } | |
|
28
69464d465284
Now supporting structs - both read and write. Still a few errors though, so watch out.
Anders Johnsen <skabet@gmail.com>
parents:
22
diff
changeset
|
253 |
|
69464d465284
Now supporting structs - both read and write. Still a few errors though, so watch out.
Anders Johnsen <skabet@gmail.com>
parents:
22
diff
changeset
|
254 ExpT visitMemberLookup(MemberLookup mem) |
|
69464d465284
Now supporting structs - both read and write. Still a few errors though, so watch out.
Anders Johnsen <skabet@gmail.com>
parents:
22
diff
changeset
|
255 { |
|
69464d465284
Now supporting structs - both read and write. Still a few errors though, so watch out.
Anders Johnsen <skabet@gmail.com>
parents:
22
diff
changeset
|
256 visitExp(mem.target); |
|
69464d465284
Now supporting structs - both read and write. Still a few errors though, so watch out.
Anders Johnsen <skabet@gmail.com>
parents:
22
diff
changeset
|
257 visitExp(mem.child); |
|
69464d465284
Now supporting structs - both read and write. Still a few errors though, so watch out.
Anders Johnsen <skabet@gmail.com>
parents:
22
diff
changeset
|
258 |
|
69464d465284
Now supporting structs - both read and write. Still a few errors though, so watch out.
Anders Johnsen <skabet@gmail.com>
parents:
22
diff
changeset
|
259 static if (is(ExpT == void)) |
|
69464d465284
Now supporting structs - both read and write. Still a few errors though, so watch out.
Anders Johnsen <skabet@gmail.com>
parents:
22
diff
changeset
|
260 return; |
|
69464d465284
Now supporting structs - both read and write. Still a few errors though, so watch out.
Anders Johnsen <skabet@gmail.com>
parents:
22
diff
changeset
|
261 else |
|
69464d465284
Now supporting structs - both read and write. Still a few errors though, so watch out.
Anders Johnsen <skabet@gmail.com>
parents:
22
diff
changeset
|
262 return ExpT.init; |
|
69464d465284
Now supporting structs - both read and write. Still a few errors though, so watch out.
Anders Johnsen <skabet@gmail.com>
parents:
22
diff
changeset
|
263 } |
| 1 | 264 } |
| 265 |