Welcome to the NetCologne GmbH open source mirroring service!

This machine mirrors various open-source projects. 20 Gbit/s uplink.

If there are any issues or you want another project mirrored, please contact mirror-service -=AT=- netcologne DOT de !

include/comma/ast/AstRewriter.h

Go to the documentation of this file.

//===-- ast/AstRewriter.h ------------------------------------- -*- C++ -*-===//
//
// This file is distributed under the MIT license.  See LICENSE.txt for details.
//
// Copyright (C) 2008-2010, Stephen Wilson
//
//===----------------------------------------------------------------------===//

//===----------------------------------------------------------------------===//
//===----------------------------------------------------------------------===//

#ifndef COMMA_AST_ASTREWRITER_HDR_GUARD
#define COMMA_AST_ASTREWRITER_HDR_GUARD

#include "comma/ast/AstBase.h"

#include <map>

namespace comma {

class AstRewriter {

public:
    AstRewriter(AstResource &resource) : resource(resource) { }

    void addTypeRewrite(Type *source, Type *target) {
        rewrites[source] = target;
    }

    template <class Iter>
    void addTypeRewrites(Iter I, Iter E) {
        for ( ; I != E; ++I)
            rewrites[I->first] = I->second;
    }

    void installRewrites(DomainType *context);

    void installRewrites(SigInstanceDecl *context);

    bool hasRewriteRule(Type *source) const {
        return getRewrite(source) != source;
    }

    void clear() { rewrites.clear(); }


    Type *rewriteType(Type *type) const;

    DomainType *rewriteType(DomainType *dom) const;

    SubroutineType *rewriteType(SubroutineType *srType) const;

    FunctionType *rewriteType(FunctionType *ftype) const;

    ProcedureType *rewriteType(ProcedureType *ftype) const;

    SigInstanceDecl *rewriteSigInstance(SigInstanceDecl *sig) const;

    // Returns the AstResource used to construct rewritten nodes.
    AstResource &getAstResource() const { return resource; }

protected:
    Type *findRewrite(Type *source) const;

private:
    AstResource &resource;

    typedef std::map<Type*, Type*> RewriteMap;
    RewriteMap rewrites;

    Type *&operator [](Type *source) {
        return rewrites[source];
    }

    Type *getRewrite(Type *source) const;

    void rewriteParameters(SubroutineType *srType,
                           unsigned count, Type **params) const;
};

} // End comma namespace

#endif

---