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

File:

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

src/AST/Pass.hpp

@@ -112 +112 @@
         const ast::Stmt *             visit( const ast::FinallyStmt          * ) override final;
         const ast::Stmt *             visit( const ast::WaitForStmt          * ) override final;
-        const ast::Stmt *             visit( const ast::WithStmt             * ) override final;
+        const ast::Decl *             visit( const ast::WithStmt             * ) override final;
         const ast::NullStmt *         visit( const ast::NullStmt             * ) override final;
         const ast::Stmt *             visit( const ast::DeclStmt             * ) override final;
@@ -293 +293 @@
                 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 ) );
+        }
 };