source: translator/InitTweak/DeclarationHoister.cc @ 1ead581

#include <list>
#include <cassert>
#include <iostream>
#include <iterator>
#include <algorithm>

#include "utility.h"

#include "SynTree/Statement.h"
#include "SynTree/Expression.h"
#include "SynTree/Declaration.h"

#include "DeclarationHoister.h"


namespace InitTweak {

  CompoundStmt* DeclarationHoister::mutate(CompoundStmt *compoundStmt) {
    typedef std::list<Statement *>::iterator stmt_it;
    // 1. collect Declaration Statements  in this scope
    std::list<Statement *> &kids = compoundStmt->get_kids();
    std::list<Statement *>::iterator result = kids.begin();
    std::list< stmt_it > decls;

    while ( result !=  kids.end() ) {
      result = std::find_if(result, kids.end(), cast_ptr< Statement, DeclStmt > );

      if ( result != kids.end() ) {
        decls.push_back( result );
        std::advance( result, 1 );
      } 
    } 

    for( std::list< stmt_it >::reverse_iterator i = decls.rbegin(); i!= decls.rend(); i++ ){
      kids.push_front( **i );
      kids.erase( *i );
    }

    return compoundStmt;
  }
} // namespace InitTweak