1 /** 2 * Utility to visit every variable in an expression. 3 * 4 * Copyright: Copyright (C) 1999-2020 by The D Language Foundation, All Rights Reserved 5 * Authors: $(LINK2 http://www.digitalmars.com, Walter Bright) 6 * License: $(LINK2 http://www.boost.org/LICENSE_1_0.txt, Boost License 1.0) 7 * Source: $(LINK2 https://github.com/dlang/dmd/blob/master/src/dmd/foreachvar.d, _foreachvar.d) 8 * Documentation: https://dlang.org/phobos/dmd_foreachvar.html 9 * Coverage: https://codecov.io/gh/dlang/dmd/src/master/src/dmd/foreachvar.d 10 */ 11 12 module dmd.foreachvar; 13 14 import core.stdc.stdio; 15 import core.stdc.stdlib; 16 import core.stdc..string; 17 18 import dmd.apply; 19 import dmd.arraytypes; 20 import dmd.attrib; 21 import dmd.dclass; 22 import dmd.declaration; 23 import dmd.dstruct; 24 import dmd.dsymbol; 25 import dmd.dsymbolsem; 26 import dmd.dtemplate; 27 import dmd.errors; 28 import dmd.expression; 29 import dmd.func; 30 import dmd.id; 31 import dmd.identifier; 32 import dmd.init; 33 import dmd.initsem; 34 import dmd.mtype; 35 import dmd.printast; 36 import dmd.root.array; 37 import dmd.root.rootobject; 38 import dmd.statement; 39 import dmd.tokens; 40 import dmd.visitor; 41 42 /********************************************* 43 * Visit each Expression in e, and call dgVar() on each variable declared in it. 44 * Params: 45 * e = expression tree to visit 46 * dgVar = call when a variable is declared 47 */ 48 void foreachVar(Expression e, void delegate(VarDeclaration) dgVar) 49 { 50 if (!e) 51 return; 52 53 extern (C++) final class VarWalker : StoppableVisitor 54 { 55 alias visit = typeof(super).visit; 56 extern (D) void delegate(VarDeclaration) dgVar; 57 58 extern (D) this(void delegate(VarDeclaration) dgVar) 59 { 60 this.dgVar = dgVar; 61 } 62 63 override void visit(Expression e) 64 { 65 } 66 67 override void visit(ErrorExp e) 68 { 69 } 70 71 override void visit(DeclarationExp e) 72 { 73 VarDeclaration v = e.declaration.isVarDeclaration(); 74 if (!v) 75 return; 76 if (TupleDeclaration td = v.toAlias().isTupleDeclaration()) 77 { 78 if (!td.objects) 79 return; 80 foreach (o; *td.objects) 81 { 82 Expression ex = isExpression(o); 83 DsymbolExp s = ex ? ex.isDsymbolExp() : null; 84 assert(s); 85 VarDeclaration v2 = s.s.isVarDeclaration(); 86 assert(v2); 87 dgVar(v2); 88 } 89 } 90 else 91 dgVar(v); 92 Dsymbol s = v.toAlias(); 93 if (s == v && !v.isStatic() && v._init) 94 { 95 if (auto ie = v._init.isExpInitializer()) 96 ie.exp.foreachVar(dgVar); 97 } 98 } 99 100 override void visit(IndexExp e) 101 { 102 if (e.lengthVar) 103 dgVar(e.lengthVar); 104 } 105 106 override void visit(SliceExp e) 107 { 108 if (e.lengthVar) 109 dgVar(e.lengthVar); 110 } 111 } 112 113 scope VarWalker v = new VarWalker(dgVar); 114 walkPostorder(e, v); 115 } 116 117 /*************** 118 * Transitively walk Statement s, pass Expressions to dgExp(), VarDeclarations to dgVar(). 119 * Params: 120 * s = Statement to traverse 121 * dgExp = delegate to pass found Expressions to 122 * dgVar = delegate to pass found VarDeclarations to 123 */ 124 void foreachExpAndVar(Statement s, 125 void delegate(Expression) dgExp, 126 void delegate(VarDeclaration) dgVar) 127 { 128 void visit(Statement s) 129 { 130 void visitExp(ExpStatement s) 131 { 132 if (s.exp) 133 dgExp(s.exp); 134 } 135 136 void visitDtorExp(DtorExpStatement s) 137 { 138 if (s.exp) 139 dgExp(s.exp); 140 } 141 142 void visitIf(IfStatement s) 143 { 144 dgExp(s.condition); 145 visit(s.ifbody); 146 visit(s.elsebody); 147 } 148 149 void visitDo(DoStatement s) 150 { 151 dgExp(s.condition); 152 visit(s._body); 153 } 154 155 void visitFor(ForStatement s) 156 { 157 visit(s._init); 158 if (s.condition) 159 dgExp(s.condition); 160 if (s.increment) 161 dgExp(s.increment); 162 visit(s._body); 163 } 164 165 void visitSwitch(SwitchStatement s) 166 { 167 dgExp(s.condition); 168 // Note that the body contains the Case and Default 169 // statements, so we only need to compile the expressions 170 foreach (cs; *s.cases) 171 { 172 dgExp(cs.exp); 173 } 174 visit(s._body); 175 } 176 177 void visitCase(CaseStatement s) 178 { 179 visit(s.statement); 180 } 181 182 void visitReturn(ReturnStatement s) 183 { 184 if (s.exp) 185 dgExp(s.exp); 186 } 187 188 void visitCompound(CompoundStatement s) 189 { 190 if (s.statements) 191 { 192 foreach (s2; *s.statements) 193 { 194 visit(s2); 195 } 196 } 197 } 198 199 void visitCompoundDeclaration(CompoundDeclarationStatement s) 200 { 201 visitCompound(s); 202 } 203 204 void visitUnrolledLoop(UnrolledLoopStatement s) 205 { 206 foreach (s2; *s.statements) 207 { 208 visit(s2); 209 } 210 } 211 212 void visitScope(ScopeStatement s) 213 { 214 visit(s.statement); 215 } 216 217 void visitDefault(DefaultStatement s) 218 { 219 visit(s.statement); 220 } 221 222 void visitWith(WithStatement s) 223 { 224 // If it is with(Enum) {...}, just execute the body. 225 if (s.exp.op == TOK.scope_ || s.exp.op == TOK.type) 226 { 227 } 228 else 229 { 230 dgVar(s.wthis); 231 dgExp(s.exp); 232 } 233 visit(s._body); 234 } 235 236 void visitTryCatch(TryCatchStatement s) 237 { 238 visit(s._body); 239 foreach (ca; *s.catches) 240 { 241 if (ca.var) 242 dgVar(ca.var); 243 visit(ca.handler); 244 } 245 } 246 247 void visitTryFinally(TryFinallyStatement s) 248 { 249 visit(s._body); 250 visit(s.finalbody); 251 } 252 253 void visitThrow(ThrowStatement s) 254 { 255 dgExp(s.exp); 256 } 257 258 void visitLabel(LabelStatement s) 259 { 260 visit(s.statement); 261 } 262 263 if (!s) 264 return; 265 266 final switch (s.stmt) 267 { 268 case STMT.Exp: visitExp(s.isExpStatement()); break; 269 case STMT.DtorExp: visitDtorExp(s.isDtorExpStatement()); break; 270 case STMT.Compound: visitCompound(s.isCompoundStatement()); break; 271 case STMT.CompoundDeclaration: visitCompoundDeclaration(s.isCompoundDeclarationStatement()); break; 272 case STMT.UnrolledLoop: visitUnrolledLoop(s.isUnrolledLoopStatement()); break; 273 case STMT.Scope: visitScope(s.isScopeStatement()); break; 274 case STMT.Do: visitDo(s.isDoStatement()); break; 275 case STMT.For: visitFor(s.isForStatement()); break; 276 case STMT.If: visitIf(s.isIfStatement()); break; 277 case STMT.Switch: visitSwitch(s.isSwitchStatement()); break; 278 case STMT.Case: visitCase(s.isCaseStatement()); break; 279 case STMT.Default: visitDefault(s.isDefaultStatement()); break; 280 case STMT.Return: visitReturn(s.isReturnStatement()); break; 281 case STMT.With: visitWith(s.isWithStatement()); break; 282 case STMT.TryCatch: visitTryCatch(s.isTryCatchStatement()); break; 283 case STMT.TryFinally: visitTryFinally(s.isTryFinallyStatement()); break; 284 case STMT.Throw: visitThrow(s.isThrowStatement()); break; 285 case STMT.Label: visitLabel(s.isLabelStatement()); break; 286 287 case STMT.CompoundAsm: 288 case STMT.Asm: 289 case STMT.InlineAsm: 290 case STMT.GccAsm: 291 292 case STMT.Break: 293 case STMT.Continue: 294 case STMT.GotoDefault: 295 case STMT.GotoCase: 296 case STMT.SwitchError: 297 case STMT.Goto: 298 case STMT.Pragma: 299 case STMT.Import: 300 case STMT.Error: 301 break; // ignore these 302 303 case STMT.ScopeGuard: 304 case STMT.Foreach: 305 case STMT.ForeachRange: 306 case STMT.Debug: 307 case STMT.CaseRange: 308 case STMT.StaticForeach: 309 case STMT.StaticAssert: 310 case STMT.Conditional: 311 case STMT.While: 312 case STMT.Forwarding: 313 case STMT.Compile: 314 case STMT.Peel: 315 case STMT.Synchronized: 316 assert(0); // should have been rewritten 317 } 318 } 319 320 visit(s); 321 } 322