source: src/SynTree/AddStmtVisitor.cc @ 1f44196

//
// Cforall Version 1.0.0 Copyright (C) 2015 University of Waterloo
//
// The contents of this file are covered under the licence agreement in the
// file "LICENCE" distributed with Cforall.
//
// AddStmtVisitor.cc --
//
// Author           : Rob Schluntz
// Created On       : Wed Jun 22 12:11:17 2016
// Last Modified By : Peter A. Buhr
// Last Modified On : Thu Aug  4 11:23:47 2016
// Update Count     : 16
//

#include "AddStmtVisitor.h"
#include "Statement.h"
#include "Declaration.h"
#include "Expression.h"
#include "Common/utility.h"

void AddStmtVisitor::visitStatementList( std::list< Statement* > &statements ) {
        for ( std::list< Statement* >::iterator i = statements.begin(); i != statements.end(); ++i ) {
                if ( ! stmtsToAddAfter.empty() ) {
                        statements.splice( i, stmtsToAddAfter );
                } // if
                (*i)->accept( *this );
                if ( ! stmtsToAdd.empty() ) {
                        statements.splice( i, stmtsToAdd );
                } // if
        } // for
        if ( ! stmtsToAddAfter.empty() ) {
                statements.splice( statements.end(), stmtsToAddAfter );
        } // if
}

Statement * AddStmtVisitor::visitStatement( Statement *stmt ) {
        maybeAccept( stmt, *this );
        if ( ! stmtsToAdd.empty() || ! stmtsToAddAfter.empty() ) {
                CompoundStmt *compound = new CompoundStmt( noLabels );
                compound->get_kids().splice( compound->get_kids().end(), stmtsToAdd );
                compound->get_kids().push_back( stmt );
                compound->get_kids().splice( compound->get_kids().end(), stmtsToAddAfter );
                return compound;
        } else {
                return stmt;
        }
}

void AddStmtVisitor::visit(CompoundStmt *compoundStmt) {
        visitStatementList( compoundStmt->get_kids() );
}

void AddStmtVisitor::visit(IfStmt *ifStmt) {
        ifStmt->set_thenPart( visitStatement( ifStmt->get_thenPart() ) );
        ifStmt->set_elsePart( visitStatement( ifStmt->get_elsePart() ) );
        maybeAccept( ifStmt->get_condition(), *this );
}

void AddStmtVisitor::visit(WhileStmt *whileStmt) {
        whileStmt->set_body( visitStatement( whileStmt->get_body() ) );
        maybeAccept( whileStmt->get_condition(), *this );
}

void AddStmtVisitor::visit(ForStmt *forStmt) {
        forStmt->set_body( visitStatement( forStmt->get_body() ) );
        acceptAll( forStmt->get_initialization(), *this );
        maybeAccept( forStmt->get_condition(), *this );
        maybeAccept( forStmt->get_increment(), *this );
}

void AddStmtVisitor::visit(SwitchStmt *switchStmt) {
        visitStatementList( switchStmt->get_statements() );
        maybeAccept( switchStmt->get_condition(), *this );
}

void AddStmtVisitor::visit(CaseStmt *caseStmt) {
        visitStatementList( caseStmt->get_statements() );
        maybeAccept( caseStmt->get_condition(), *this );
}

void AddStmtVisitor::visit(CatchStmt *catchStmt) {
        catchStmt->set_body( visitStatement( catchStmt->get_body() ) );
        maybeAccept( catchStmt->get_decl(), *this );
}

// Local Variables: //
// tab-width: 4 //
// mode: c++ //
// compile-command: "make install" //
// End: //