Changes in src/AST/Pass.hpp [73f1b1c:e6b42e7]

File:

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

src/AST/Pass.hpp

@@ -67 +67 @@
 // | WithSymbolTable       - provides symbol table functionality
 // | WithForallSubstitutor - maintains links between TypeInstType and TypeDecl under mutation
+//
+// Other Special Members:
+// | result                - Either a method that takes no parameters or a field. If a method (or
+//                           callable field) get_result calls it, otherwise the value is returned.
 //-------------------------------------------------------------------------------------------------
 template< typename core_t >
@@ -89 +93 @@
         virtual ~Pass() = default;
 
+        /// Storage for the actual pass.
+        core_t core;
+
+        /// If the core defines a result, call it if possible, otherwise return it.
+        inline auto get_result() -> decltype( __pass::get_result( core, '0' ) ) {
+                return __pass::get_result( core, '0' );
+        }
+
         /// Construct and run a pass on a translation unit.
         template< typename... Args >
@@ -96 +108 @@
         }
 
-        template< typename... Args >
+        /// Contruct and run a pass on a pointer to extract a value.
+        template< typename node_type, typename... Args >
+        static auto read( node_type const * node, Args&&... args ) {
+                Pass<core_t> visitor( std::forward<Args>( args )... );
+                node_type const * temp = node->accept( visitor );
+                assert( temp == node );
+                return visitor.get_result();
+        }
+
+        // Versions of the above for older compilers.
         static void run( std::list< ptr<Decl> > & decls ) {
                 Pass<core_t> visitor;
@@ -102 +123 @@
         }
 
-        /// Storage for the actual pass
-        core_t core;
+        static auto read( Node const * node ) {
+                Pass<core_t> visitor;
+                Node const * temp = node->accept( visitor );
+                assert( temp == node );
+                return visitor.get_result();
+        }
 
         /// Visit function declarations