source: src/AST/ForallSubstitutor.hpp@ 7ff3e522

//
// 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.
//
// ForallSubstitutor.hpp --
//
// Author           : Aaron B. Moss
// Created On       : Wed Jun 26 15:00:00 2019
// Last Modified By : Aaron B. Moss
// Last Modified On : Wed Jun 26 15:00:00 2019
// Update Count     : 1
//

#include "Pass.hpp"

namespace ast {

class Expr;

/// Visitor that correctly substitutes TypeDecl while maintaining TypeInstType bindings.
/// Also has some convenience methods to mutate fields.
struct ForallSubstitutor : public WithForallSubstitutor, public WithVisitorRef<ForallSubstitutor> {
        /// Substitute TypeInstType base type
        readonly< TypeDecl > operator() ( const readonly< TypeDecl > & o ) {
                return subs.replace( o );
        }
        
        /// Make new forall-list clone
        ParameterizedType::ForallList operator() ( const ParameterizedType::ForallList & o ) {
                return subs.clone( o, *visitor );
        }

        /// Substitute parameter/return type
        std::vector< ptr< DeclWithType > > operator() ( const std::vector< ptr< DeclWithType > > & o ) {
                std::vector< ptr< DeclWithType > > n;
                n.reserve( o.size() );
                for ( const DeclWithType * d : o ) { n.emplace_back( d->accept( *visitor ) ); }
                return n;
        }

        /// Substitute type parameter list
        std::vector< ptr< Expr > > operator() ( const std::vector< ptr< Expr > > & o ) {
                std::vector< ptr< Expr > > n;
                n.reserve( o.size() );
                for ( const Expr * d : o ) { n.emplace_back( d->accept( *visitor ) ); }
                return n;
        }
};

} // namespace ast

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