lib/codegen/SRInfo.h

Go to the documentation of this file.

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

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

#ifndef COMMA_CODEGEN_SRINFO_HDR_GUARD
#define COMMA_CODEGEN_SRINFO_HDR_GUARD

#include "CodeGen.h"
#include "comma/ast/Decl.h"

#include "llvm/ADT/StringRef.h"

namespace comma {

class SRInfo {

public:


    SubroutineDecl *getDeclaration() { return srDecl; }
    const SubroutineDecl *getDeclaration() const { return srDecl; }

    llvm::StringRef getLinkName() const { return llvmFn->getName(); }

    llvm::Function *getLLVMFunction() const { return llvmFn; }

    llvm::Module *getLLVMModule() const { return llvmFn->getParent(); }

    bool isaFunction() const { return llvm::isa<FunctionDecl>(srDecl); }

    bool isaProcedure() const { return llvm::isa<ProcedureDecl>(srDecl); }

    bool hasSRet() const { return llvmFn->hasStructRetAttr(); }

    bool usesVRet() const {
        // If this is a function returning void but does not use the sret
        // convention, then the value must be returned on the vstack.
        //
        // FIXME: llvm-2.6-svn and up has Type::isVoidTy predicate.  Use it.
        if (isaFunction() &&
            (llvmFn->getReturnType()->getTypeID() == llvm::Type::VoidTyID))
            return !hasSRet();
        return false;
    }

    bool hasAggRet() const { return hasSRet() || usesVRet(); }

    bool isImported() const { return srDecl->hasPragma(pragma::Import); }

private:
    SRInfo(SubroutineDecl *decl, llvm::Function *fn)
        : srDecl(decl), llvmFn(fn) { }
    friend class InstanceInfo;

    SubroutineDecl *srDecl;
    llvm::Function *llvmFn;
};

} // end comma namespace.

#endif

---