Changeset 834b892

Timestamp:

Feb 6, 2018, 5:11:22 PM (7 years ago)

Author:

Rob Schluntz <rschlunt@…>

Branches:

ADT, aaron-thesis, arm-eh, ast-experimental, cleanup-dtors, deferred_resn, demangler, enum, forall-pointer-decay, jacob/cs343-translation, jenkins-sandbox, master, new-ast, new-ast-unique-expr, new-env, no_list, persistent-indexer, pthread-emulation, qualifiedEnum, resolv-new, with_gc

Children:

10f8142

Parents:

86c934a

Message:

Expand VISIT/MUTATE_BODY for WaitForStmt?

File:

• src/Common/PassVisitor.impl.h (modified) (1 diff)

src/Common/PassVisitor.impl.h

@@ -1018 +1018 @@
 template< typename pass_type >
 void PassVisitor< pass_type >::visit( WaitForStmt * node ) {
-        VISIT_BODY( node );
+        VISIT_START( node );
+
+        for( auto & clause : node->clauses ) {
+                maybeAccept_impl( clause.target.function, *this );
+                maybeAccept_impl( clause.target.arguments, *this );
+
+                maybeAccept_impl( clause.statement, *this );
+                maybeAccept_impl( clause.condition, *this );
+        }
+
+        maybeAccept_impl( node->timeout.time, *this );
+        maybeAccept_impl( node->timeout.statement, *this );
+        maybeAccept_impl( node->timeout.condition, *this );
+        maybeAccept_impl( node->orelse.statement, *this );
+        maybeAccept_impl( node->orelse.condition, *this );
+
+        VISIT_END( node );
 }
 
 template< typename pass_type >
 Statement * PassVisitor< pass_type >::mutate( WaitForStmt * node ) {
-        MUTATE_BODY( Statement, node );
+        MUTATE_START( node );
+
+        for( auto & clause : node->clauses ) {
+                maybeMutate_impl( clause.target.function, *this );
+                maybeMutate_impl( clause.target.arguments, *this );
+
+                maybeMutate_impl( clause.statement, *this );
+                maybeMutate_impl( clause.condition, *this );
+        }
+
+        maybeMutate_impl( node->timeout.time, *this );
+        maybeMutate_impl( node->timeout.statement, *this );
+        maybeMutate_impl( node->timeout.condition, *this );
+        maybeMutate_impl( node->orelse.statement, *this );
+        maybeMutate_impl( node->orelse.condition, *this );
+
+        MUTATE_END( Statement, node );
 }