include/comma/ast/KeywordSelector.h

Go to the documentation of this file.

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


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

#ifndef COMMA_AST_KEYWORDSELECTOR_HDR_GUARD
#define COMMA_AST_KEYWORDSELECTOR_HDR_GUARD

#include "comma/ast/Expr.h"
#include "comma/ast/TypeRef.h"

namespace comma {

class KeywordSelector : public Ast {

public:
    KeywordSelector(IdentifierInfo *key, Location loc, Expr *expr)
        : Ast(AST_KeywordSelector),
          keyword(key), loc(loc), rhs(expr) { }

    KeywordSelector(IdentifierInfo *key, Location loc, TypeRef *tyRef)
        : Ast(AST_KeywordSelector),
          keyword(key), loc(loc), rhs(tyRef) { }

    IdentifierInfo *getKeyword() const { return keyword; }

    Location getLocation()  const { return loc; }

    bool isTypeSelector() const { return llvm::isa<TypeRef>(rhs); }

    bool isExprSelector() const { return llvm::isa<Expr>(rhs); }



    void setRHS(Expr *expr) { rhs = expr; }
    void setRHS(TypeRef *ref) { rhs = ref; }



    Expr *getExpression() { return llvm::dyn_cast<Expr>(rhs); }
    const Expr *getExpression() const { return llvm::dyn_cast<Expr>(rhs); }

    Type *getType() const {
        if (const Expr *expr = getExpression()) {
            if (expr->hasType())
                return expr->getType();
        }
        return 0;
    }



    TypeRef *getTypeRef() { return llvm::dyn_cast<TypeRef>(rhs); }
    const TypeRef *getTypeRef() const { return llvm::dyn_cast<TypeRef>(rhs); }

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

private:
    IdentifierInfo *keyword;
    Location loc;
    Ast *rhs;
};

} // end comma namespace.

#endif

---