Changes in src/Common/PassVisitor.h [6ca154b:5dd0704]

File:

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

src/Common/PassVisitor.h

@@ -25 +25 @@
 // | WithStmtsToAdd       - provides the ability to insert statements before or after the current statement by adding new statements into
 //                          stmtsToAddBefore or stmtsToAddAfter respectively.
-// | WithShortCircuiting  - provides the ability to skip visiting child nodes; set visit_children to false in pre{visit,mutate} to skip visiting children
+// | WithShortCircuiting  - provides the ability to skip visiting child nodes; set skip_children to true if pre{visit,mutate} to skip visiting children
 // | WithScopes           - provides the ability to save/restore data like a LIFO stack; to save, call GuardValue with the variable to save, the variable
 //                          will automatically be restored to its previous value after the corresponding postvisit/postmutate teminates.
@@ -37 +37 @@
         PassVisitor(Args &&... args)
                 : pass( std::forward<Args>( args )... )
-        {
-                typedef PassVisitor<pass_type> this_t;
-                this_t * const * visitor = visitor_impl(pass, 0);
-                if(visitor) {
-                        *const_cast<this_t **>( visitor ) = this;
-                }
-        }
+        {}
 
         virtual ~PassVisitor() = default;
@@ -220 +214 @@
 
 private:
-        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 node_type> void call_previsit ( node_type * node ) { previsit_impl ( pass, node, 0 ); }
         template<typename node_type> void call_postvisit( node_type * node ) { postvisit_impl( pass, node, 0 ); }
@@ -234 +225 @@
         void set_env( TypeSubstitution * env ) { set_env_impl( pass, env, 0); }
 
-        template< typename func_t >
-        void handleStatementList( std::list< Statement * > & statements, func_t func );
-        void visitStatementList ( std::list< Statement* > &statements );
+        void visitStatementList( std::list< Statement* > &statements );
         void mutateStatementList( std::list< Statement* > &statements );
 
-        template< typename func_t >
-        Statement * handleStatement( Statement * stmt, func_t func );
-        Statement * visitStatement ( Statement * stmt );
+        Statement * visitStatement( Statement * stmt );
         Statement * mutateStatement( Statement * stmt );
 
-        template< typename func_t >
-        Expression * handleExpression( Expression * expr, func_t func );
-        Expression * visitExpression ( Expression * expr );
+        void visitExpression( Expression * expr );
         Expression * mutateExpression( Expression * expr );
 
@@ -253 +238 @@
         std::list< Statement* > *       get_beforeStmts() { return stmtsToAddBefore_impl( pass, 0); }
         std::list< Statement* > *       get_afterStmts () { return stmtsToAddAfter_impl ( pass, 0); }
-        std::list< Declaration* > *     get_beforeDecls() { return declsToAddBefore_impl( pass, 0); }
-        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 visit_children() { bool* skip = skip_children_impl(pass, 0); return ! (skip && *skip); }
+        void reset_visit() { bool* skip = skip_children_impl(pass, 0); if(skip) *skip = false; }
 
         guard_value_impl init_guard() {
@@ -295 +278 @@
         std::list< Statement* > stmtsToAddAfter;
 };
+
 class WithShortCircuiting {
 protected:
@@ -301 +285 @@
 
 public:
-        bool_ref visit_children;
+        bool skip_children;
 };
 
@@ -318 +302 @@
                 }, static_cast< void * >( & val ) );
         }
-};
-
-template<typename pass_type>
-class WithVisitorRef {
-protected:
-        WithVisitorRef() = default;
-        ~WithVisitorRef() = default;
-
-public:
-        PassVisitor<pass_type> * const visitor;
-};
+
+        template< typename Func >
+        void GuardAction( Func&& func ) {
+                at_cleanup( std::forward<Func>( func ) );
+        }
+};
+
 
 #include "PassVisitor.impl.h"