Changes in src/SynTree/DeclReplacer.h [7862059:546e712]

File:

• src/SynTree/DeclReplacer.h (modified) (1 diff)

src/SynTree/DeclReplacer.h

@@ -26 +26 @@
         typedef std::map< DeclarationWithType *, DeclarationWithType * > DeclMap;
         typedef std::map< TypeDecl *, TypeDecl * > TypeMap;
+        typedef std::map< DeclarationWithType *, Expression * > ExprMap;
 
-        void replace( BaseSyntaxNode * node, const DeclMap & declMap, bool debug = false );
-        void replace( BaseSyntaxNode * node, const TypeMap & typeMap, bool debug = false );
-        void replace( BaseSyntaxNode * node, const DeclMap & declMap, const TypeMap & typeMap, bool debug = false );
+        size_t replace( BaseSyntaxNode * node, const DeclMap & declMap, bool debug = false );
+        size_t replace( BaseSyntaxNode * node, const TypeMap & typeMap, bool debug = false );
+        size_t replace( BaseSyntaxNode * node, const DeclMap & declMap, const TypeMap & typeMap, bool debug = false );
+
+        size_t replace( BaseSyntaxNode *& node, const ExprMap & exprMap, bool debug = false);
+
+        template<typename T>
+        size_t replace( T *& node, const ExprMap & exprMap, bool debug = false ) {
+                if ( ! node ) return 0ul;
+                BaseSyntaxNode * arg = node;
+                size_t replaced = replace( arg, exprMap, debug );
+                node = dynamic_cast<T *>( arg );
+                assertf( node, "DeclReplacer fundamentally changed the type of its argument." );
+                return replaced;
+        }
 }