Changes in src/GenPoly/Specialize.cc [d5baf0c:07de76b]

File:

• src/GenPoly/Specialize.cc (modified) (4 diffs)

src/GenPoly/Specialize.cc

@@ -9 +9 @@
 // Author           : Richard C. Bilson
 // Created On       : Mon May 18 07:44:20 2015
-// Last Modified By : Andrew Beach
-// Last Modified On : Thr Jul  2 17:42:00 2020
-// Update Count     : 33
+// Last Modified By : Peter A. Buhr
+// Last Modified On : Fri Dec 13 23:40:49 2019
+// Update Count     : 32
 //
 
@@ -42 +42 @@
 
 namespace GenPoly {
-        struct Specialize final : public WithConstTypeSubstitution,
-                        public WithDeclsToAdd, public WithVisitorRef<Specialize> {
+        struct Specialize final : public WithConstTypeSubstitution, public WithStmtsToAdd, public WithVisitorRef<Specialize> {
                 Expression * postmutate( ApplicationExpr *applicationExpr );
                 Expression * postmutate( CastExpr *castExpr );
@@ -249 +248 @@
                 } // if
 
-                // Handle any specializations that may still be present.
-                {
-                        std::string oldParamPrefix = paramPrefix;
-                        paramPrefix += "p";
-                        std::list< Declaration * > oldDecls;
-                        oldDecls.splice( oldDecls.end(), declsToAddBefore );
-
-                        appExpr->acceptMutator( *visitor );
-                        // Write recursive specializations into the thunk body.
-                        for ( Declaration * decl : declsToAddBefore ) {
-                                thunkFunc->statements->kids.push_back( new DeclStmt( decl ) );
-                        }
-
-                        declsToAddBefore = std::move( oldDecls );
-                        paramPrefix = oldParamPrefix;
-                }
+                // handle any specializations that may still be present
+                std::string oldParamPrefix = paramPrefix;
+                paramPrefix += "p";
+                // save stmtsToAddBefore in oldStmts
+                std::list< Statement* > oldStmts;
+                oldStmts.splice( oldStmts.end(), stmtsToAddBefore );
+                appExpr->acceptMutator( *visitor );
+                paramPrefix = oldParamPrefix;
+                // write any statements added for recursive specializations into the thunk body
+                thunkFunc->statements->kids.splice( thunkFunc->statements->kids.end(), stmtsToAddBefore );
+                // restore oldStmts into stmtsToAddBefore
+                stmtsToAddBefore.splice( stmtsToAddBefore.end(), oldStmts );
 
                 // add return (or valueless expression) to the thunk
@@ -275 +270 @@
                 thunkFunc->statements->kids.push_back( appStmt );
 
-                // Add the thunk definition (converted to DeclStmt if appproprate).
-                declsToAddBefore.push_back( thunkFunc );
+                // add thunk definition to queue of statements to add
+                stmtsToAddBefore.push_back( new DeclStmt( thunkFunc ) );
                 // return address of thunk function as replacement expression
                 return new AddressExpr( new VariableExpr( thunkFunc ) );