Changes in src/AST/Pass.hpp [e16e27e1:be8518f]

File:

• src/AST/Pass.hpp (modified) (2 diffs)

src/AST/Pass.hpp

@@ -262 +262 @@
 /// Used to restore values/functions/etc. when the Pass finishes visiting this node
 class WithGuards {
-        __pass::at_cleanup_t at_cleanup = [](__pass::cleanup_func_t, void*) {
-                std::cerr << "No cleanup function was set" << std::endl;
-                abort();
-        };
-
-        template< typename pass_t>
-        friend auto __pass::at_cleanup( pass_t & pass, int ) -> decltype( &pass.at_cleanup );
+        __pass::at_cleanup_t at_cleanup;
+
 public:
-
         /// When this node is finished being visited, restore the value of a variable
         template< typename T >
@@ -293 +287 @@
                 at_cleanup( [func](void *) { func(); }, nullptr );
         }
+
+        /// When this node is finished being visited, call a member of an object.
+        template<typename T>
+        void GuardMethod( T * obj, void (T::*method)() ) {
+                at_cleanup( [ method ]( void * object ) {
+                        static_cast< T * >( object )->method();
+                }, static_cast< void * >( obj ) );
+        }
 };