source: src/SynTree/CompoundStmt.cc @ e464759

//
// 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.
//
// XXX.cc --
//
// Author           : Richard C. Bilson
// Created On       : Mon May 18 07:44:20 2015
// Last Modified By : Rob Schluntz
// Last Modified On : Mon May 02 15:19:17 2016
// Update Count     : 3
//

#include "Statement.h"
#include "Common/utility.h"
#include <algorithm>
#include <functional>
#include "Expression.h"
#include "Declaration.h"
#include "SynTree/VarExprReplacer.h"

using std::string;
using std::endl;

CompoundStmt::CompoundStmt( std::list<Label> labels ) : Statement( labels ) {
}

CompoundStmt::CompoundStmt( const CompoundStmt &other ) : Statement( other ) {
        cloneAll( other.kids, kids );

        // when cloning a compound statement, we may end up cloning declarations which
        // are referred to by VariableExprs throughout the block. Cloning a VariableExpr
        // does a shallow copy, so the VariableExpr will end up pointing to the original
        // declaration. If the original declaration is deleted, e.g. because the original
        // CompoundStmt is deleted, then we have a dangling pointer. To avoid this case,
        // find all DeclarationWithType nodes (since a VariableExpr must point to a
        // DeclarationWithType) in the original CompoundStmt and map them to the cloned
        // node in the new CompoundStmt ('this'), then replace the Declarations referred to
        // by each VariableExpr according to the constructed map. Note that only the declarations
        // in the current level are collected into the map, because child CompoundStmts will
        // recursively execute this routine. There may be more efficient ways of doing
        // this.
        VarExprReplacer::DeclMap declMap;
        std::list< Statement * >::const_iterator origit = other.kids.begin();
        for ( Statement * s : kids ) {
                assert( origit != other.kids.end() );
                Statement * origStmt = *origit++;
                if ( DeclStmt * declStmt = dynamic_cast< DeclStmt * >( s ) ) {
                        DeclStmt * origDeclStmt = safe_dynamic_cast< DeclStmt * >( origStmt );
                        if ( DeclarationWithType * dwt = dynamic_cast< DeclarationWithType * > ( declStmt->get_decl() ) ) {
                                DeclarationWithType * origdwt = safe_dynamic_cast< DeclarationWithType * > ( origDeclStmt->get_decl() );
                                assert( dwt->get_name() == origdwt->get_name() );
                                declMap[ origdwt ] = dwt;
                        } else assert( ! dynamic_cast< DeclarationWithType * > ( origDeclStmt->get_decl() ) );
                } else assert( ! dynamic_cast< DeclStmt * > ( s ) );
        }
        if ( ! declMap.empty() ) {
                VarExprReplacer replacer( declMap );
                accept( replacer );
        }
}

CompoundStmt::~CompoundStmt() {
        deleteAll( kids );
}

void CompoundStmt::print( std::ostream &os, int indent ) const {
        os << "CompoundStmt" << endl ;
        printAll( kids, os, indent + 2 );
}

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