#include "SynTree/Mutator.h"
#include "SynTree/Statement.h"
#include "ForExprMutator.h"

namespace ControlStruct {
  Statement* ForExprMutator::mutate(ForStmt *forStmt)
  {
    DeclStmt *decl;
    if (( decl = dynamic_cast< DeclStmt * > ( forStmt->get_initialization() )) != 0 )
      {
	// create compound statement, move declaration outside, leave _for_ as-is
	CompoundStmt *block = new CompoundStmt( std::list< Label >() );
	std::list<Statement *> &stmts = block->get_kids();

	stmts.push_back( decl );
	forStmt->set_initialization( 0 );
	stmts.push_back( forStmt );

	return block;
      }
    // ForStmt still needs to be fixed
    else
      return forStmt;
  }

} // namespace ControlStruct