source: src/ControlStruct/ChooseMutator.cc @ 09a9a8b

//
// 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.
//
// ChooseMutator.cc -- 
//
// Author           : Rodolfo G. Esteves
// Created On       : Mon May 18 07:44:20 2015
// Last Modified By : Peter A. Buhr
// Last Modified On : Tue May 19 15:31:39 2015
// Update Count     : 2
//

#include <list>

#include "SynTree/Statement.h"
#include "ChooseMutator.h"

namespace ControlStruct {
        Statement *ChooseMutator::mutate( ChooseStmt *chooseStmt ) {
                bool enclosingChoose = insideChoose;
                insideChoose = true;
                mutateAll( chooseStmt->get_branches(), *this );
                insideChoose = enclosingChoose;
                return new SwitchStmt( chooseStmt->get_labels(),  chooseStmt->get_condition(), chooseStmt->get_branches() );
        }

        Statement *ChooseMutator::mutate( SwitchStmt *switchStmt ) {
                bool enclosingChoose = insideChoose;
                insideChoose = false;
                mutateAll( switchStmt->get_branches(), *this );
                insideChoose = enclosingChoose;
                return switchStmt;
        }

        Statement *ChooseMutator::mutate( FallthruStmt *fallthruStmt ) {
                delete fallthruStmt;
                return new NullStmt();
        }

        Statement* ChooseMutator::mutate( CaseStmt *caseStmt ) {
                std::list< Statement * > &stmts = caseStmt->get_statements();

                if ( insideChoose ) {
                        BranchStmt *posBrk;
                        if ( (( posBrk = dynamic_cast< BranchStmt * > ( stmts.back() ) ) && 
                                  ( posBrk->get_type() == BranchStmt::Break ))  // last statement in the list is a (superfluous) 'break' 
                                 || dynamic_cast< FallthruStmt * > ( stmts.back() ) )
                                ; 
                        else {
                                stmts.push_back( new BranchStmt( std::list< Label >(), "", BranchStmt::Break ) );
                        } // if
                } // if

                mutateAll ( stmts, *this );
                return caseStmt;
        }
} // namespace ControlStruct

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