include/comma/ast/TypeRef.h

Go to the documentation of this file.

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

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

#ifndef COMMA_AST_TYPEREF_HDR_GUARD
#define COMMA_AST_TYPEREF_HDR_GUARD

namespace comma {

#include "comma/ast/Decl.h"

class TypeRef : public Ast {

public:
    TypeRef(Location loc, FunctorDecl *functor)
        : Ast(AST_TypeRef), loc(loc), target(functor) { }

    TypeRef(Location loc, VarietyDecl *variety)
        : Ast(AST_TypeRef), loc(loc), target(variety) { }

    TypeRef(Location loc, SigInstanceDecl *instance)
        : Ast(AST_TypeRef), loc(loc), target(instance) { }

    TypeRef(Location loc, TypeDecl *tyDecl)
        : Ast(AST_TypeRef), loc(loc), target(tyDecl) { }

    Location getLocation() const { return loc; }

    IdentifierInfo *getIdInfo() const { return target->getIdInfo(); }

    bool isIncomplete() const {
        return llvm::isa<FunctorDecl>(target) || llvm::isa<VarietyDecl>(target);
    }

    bool isComplete() const { return !isIncomplete(); }

    bool referencesVariety() const {
        return llvm::isa<VarietyDecl>(target);
    }

    bool referencesFunctor() const {
        return llvm::isa<FunctorDecl>(target);
    }

    bool referencesSigInstance() const {
        return llvm::isa<SigInstanceDecl>(target);
    }

    bool referencesTypeDecl() const {
        return llvm::isa<TypeDecl>(target);
    }

    Decl *getDecl() const { return target; }

    ModelDecl *getModelDecl() const {
        return llvm::dyn_cast<ModelDecl>(target);
    }

    VarietyDecl *getVarietyDecl() const {
        return llvm::dyn_cast<VarietyDecl>(target);
    }

    FunctorDecl *getFunctorDecl() const {
        return llvm::dyn_cast<FunctorDecl>(target);
    }

    SigInstanceDecl *getSigInstanceDecl() const {
        return llvm::dyn_cast<SigInstanceDecl>(target);
    }

    TypeDecl *getTypeDecl() const {
        return llvm::dyn_cast<TypeDecl>(target);
    }

    // Support isa and dyn_cast.
    static bool classof(const TypeRef *node) { return true; }
    static bool classof(const Ast *node) {
        return node->getKind() == AST_TypeRef;
    }

private:
    Location loc;
    Decl *target;
};

} // end comma namespace.

#endif

---