Changes in src/Common/PassVisitor.h [5ea7a22:3c398b6]

File:

• src/Common/PassVisitor.h (modified) (3 diffs)

src/Common/PassVisitor.h

@@ -4 +4 @@
 
 #include <stack>
+
+#include "Common/utility.h"
 
 #include "SynTree/Mutator.h"
@@ -239 +241 @@
         template<typename pass_t> friend void acceptAll( std::list< Declaration* > &decls, PassVisitor< pass_t >& visitor );
         template<typename pass_t> friend void mutateAll( std::list< Declaration* > &decls, PassVisitor< pass_t >& visitor );
+        template< typename TreeType, typename pass_t > friend void maybeAccept_impl( TreeType * tree, PassVisitor< pass_t > & visitor );
+        template< typename TreeType, typename pass_t > friend void maybeMutate_impl( TreeType *& tree, PassVisitor< pass_t > & mutator );
+        template< typename Container, typename pass_t > friend void maybeAccept_impl( Container & container, PassVisitor< pass_t > & visitor );
+        template< typename Container, typename pass_t > friend void maybeMutate_impl( Container & container, PassVisitor< pass_t > & mutator );
 
         template<typename node_type> void call_previsit ( node_type * node ) { previsit_impl ( pass, node, 0 ); }
@@ -273 +279 @@
         std::list< Declaration* > *     get_afterDecls () { return declsToAddAfter_impl ( pass, 0); }
 
-        void set_visit_children( bool& ref ) { bool_ref * ptr = visit_children_impl(pass, 0); if(ptr) ptr->set( ref ); }
+        bool       get_visit_children    () { bool_ref * ptr = visit_children_impl(pass, 0); return ptr ? *ptr : true; }
+        bool_ref * get_visit_children_ptr() { return visit_children_impl(pass, 0); }
 
         void indexerScopeEnter  ()                             { indexer_impl_enterScope  ( pass, 0       ); }