Changeset dbd8652 for src/GenPoly

Timestamp:

Feb 29, 2016, 5:42:02 PM (9 years ago)

Author:

Aaron Moss <a3moss@…>

Branches:

ADT, aaron-thesis, arm-eh, ast-experimental, cleanup-dtors, ctor, deferred_resn, demangler, enum, forall-pointer-decay, gc_noraii, jacob/cs343-translation, jenkins-sandbox, master, memory, new-ast, new-ast-unique-expr, new-env, no_list, persistent-indexer, pthread-emulation, qualifiedEnum, resolv-new, string, with_gc

Children:

9d7b3ea

Parents:

540ddb7d

Message:

Add capability to add declarations after current one to DeclMutator?

Location:

src/GenPoly

Files:

• DeclMutator.cc (modified) (8 diffs)
• DeclMutator.h (modified) (1 diff)

src/GenPoly/DeclMutator.cc

@@ -24 +24 @@
         }
 
-        DeclMutator::DeclMutator() : Mutator(), declsToAdd(1) {}
+        DeclMutator::DeclMutator() : Mutator(), declsToAdd(1), declsToAddAfter(1) {}
 
         DeclMutator::~DeclMutator() {}
         
         void DeclMutator::mutateDeclarationList( std::list< Declaration* > &decls ) {
-                for ( std::list< Declaration* >::iterator decl = decls.begin(); decl != decls.end(); ++decl ) {
+                for ( std::list< Declaration* >::iterator decl = decls.begin(); ; ++decl ) {
+                        // splice in new declarations after previous decl
+                        decls.splice( decl, declsToAddAfter.back() );
+
+                        if ( decl == decls.end() ) break;
+                        
                         // run mutator on declaration
                         *decl = maybeMutate( *decl, *this );
@@ -39 +44 @@
 
         void DeclMutator::doBeginScope() {
-                // add new decl list for inside of scope
+                // add new decl lists for inside of scope
                 declsToAdd.resize( declsToAdd.size()+1 );
+                declsToAddAfter.resize( declsToAddAfter.size()+1 );
         }
 
@@ -49 +55 @@
                 newBack->splice( newBack->end(), *back );
                 declsToAdd.pop_back();
+                
+                back = declsToAddAfter.rbegin();
+                newBack = back + 1;
+                newBack->splice( newBack->end(), *back );
+                declsToAddAfter.pop_back();
         }
 
@@ -61 +72 @@
                 stmt = maybeMutate( stmt, *this );
                 // return if no declarations to add
-                if ( declsToAdd.back().empty() ) return stmt;
+                if ( declsToAdd.back().empty() && declsToAddAfter.back().empty() ) {
+                        doEndScope();
+                        return stmt;
+                }
 
                 // otherwise add declarations to new compound statement
@@ -71 +85 @@
                 declsToAdd.back().clear();
 
+                // add mutated statement
+                compound->get_kids().push_back( stmt );
+
+                // add declarations after to new compound statement
+                for ( std::list< Declaration* >::iterator decl = declsToAddAfter.back().begin(); decl != declsToAddAfter.back().end(); ++decl ) {
+                        DeclStmt *declStmt = new DeclStmt( noLabels, *decl );
+                        compound->get_kids().push_back( declStmt );
+                }
+                declsToAddAfter.back().clear();
+
                 doEndScope();
-
-                // add mutated statement and return
-                compound->get_kids().push_back( stmt );
                 return compound;
         }
@@ -80 +101 @@
         void DeclMutator::mutateStatementList( std::list< Statement* > &stmts ) {
                 doBeginScope();
+
                 
-                for ( std::list< Statement* >::iterator stmt = stmts.begin(); stmt != stmts.end(); ++stmt ) {
+                for ( std::list< Statement* >::iterator stmt = stmts.begin(); ; ++stmt ) {
+                        // add any new declarations after the previous statement
+                        for ( std::list< Declaration* >::iterator decl = declsToAddAfter.back().begin(); decl != declsToAddAfter.back().end(); ++decl ) {
+                                DeclStmt *declStmt = new DeclStmt( noLabels, *decl );
+                                stmts.insert( stmt, declStmt );
+                        }
+                        declsToAddAfter.back().clear();
+
+                        if ( stmt == stmts.end() ) break;
+                        
                         // run mutator on statement
                         *stmt = maybeMutate( *stmt, *this );
@@ -92 +123 @@
                         declsToAdd.back().clear();
                 }
-
+                
                 doEndScope();
         }
@@ -98 +129 @@
         void DeclMutator::addDeclaration( Declaration *decl ) {
                 declsToAdd.back().push_back( decl );
+        }
+
+        void DeclMutator::addDeclarationAfter( Declaration *decl ) {
+                declsToAddAfter.back().push_back( decl );
         }
 

src/GenPoly/DeclMutator.h

@@ -55 +55 @@
                 /// Add a declaration to the list to be added before the current position
                 void addDeclaration( Declaration* decl );
+                /// Add a declaration to the list to be added after the current position
+                void addDeclarationAfter( Declaration* decl );
         private:
                 /// A stack of declarations to add before the current declaration or statement
                 std::vector< std::list< Declaration* > > declsToAdd;
+                /// A stack of declarations to add after the current declaration or statement
+                std::vector< std::list< Declaration* > > declsToAddAfter;
         };
 } // namespace