forked from OSchip/llvm-project
121 lines
4.1 KiB
C++
121 lines
4.1 KiB
C++
//===--- ModuleBuilder.cpp - Emit LLVM Code from ASTs ---------------------===//
|
|
//
|
|
// The LLVM Compiler Infrastructure
|
|
//
|
|
// This file is distributed under the University of Illinois Open Source
|
|
// License. See LICENSE.TXT for details.
|
|
//
|
|
//===----------------------------------------------------------------------===//
|
|
//
|
|
// This builds an AST and converts it to LLVM Code.
|
|
//
|
|
//===----------------------------------------------------------------------===//
|
|
|
|
#include "clang/CodeGen/ModuleBuilder.h"
|
|
#include "CodeGenModule.h"
|
|
#include "clang/Frontend/CodeGenOptions.h"
|
|
#include "clang/AST/ASTContext.h"
|
|
#include "clang/AST/DeclObjC.h"
|
|
#include "clang/AST/Expr.h"
|
|
#include "clang/Basic/Diagnostic.h"
|
|
#include "clang/Basic/TargetInfo.h"
|
|
#include "llvm/LLVMContext.h"
|
|
#include "llvm/Module.h"
|
|
#include "llvm/Target/TargetData.h"
|
|
#include "llvm/ADT/OwningPtr.h"
|
|
using namespace clang;
|
|
|
|
namespace {
|
|
class CodeGeneratorImpl : public CodeGenerator {
|
|
Diagnostic &Diags;
|
|
llvm::OwningPtr<const llvm::TargetData> TD;
|
|
ASTContext *Ctx;
|
|
const CodeGenOptions CodeGenOpts; // Intentionally copied in.
|
|
protected:
|
|
llvm::OwningPtr<llvm::Module> M;
|
|
llvm::OwningPtr<CodeGen::CodeGenModule> Builder;
|
|
public:
|
|
CodeGeneratorImpl(Diagnostic &diags, const std::string& ModuleName,
|
|
const CodeGenOptions &CGO, llvm::LLVMContext& C)
|
|
: Diags(diags), CodeGenOpts(CGO), M(new llvm::Module(ModuleName, C)) {}
|
|
|
|
virtual ~CodeGeneratorImpl() {}
|
|
|
|
virtual llvm::Module* GetModule() {
|
|
return M.get();
|
|
}
|
|
|
|
virtual llvm::Module* ReleaseModule() {
|
|
return M.take();
|
|
}
|
|
|
|
virtual void Initialize(ASTContext &Context) {
|
|
Ctx = &Context;
|
|
|
|
M->setTargetTriple(Ctx->Target.getTriple().getTriple());
|
|
M->setDataLayout(Ctx->Target.getTargetDescription());
|
|
TD.reset(new llvm::TargetData(Ctx->Target.getTargetDescription()));
|
|
Builder.reset(new CodeGen::CodeGenModule(Context, CodeGenOpts,
|
|
*M, *TD, Diags));
|
|
}
|
|
|
|
virtual void HandleTopLevelDecl(DeclGroupRef DG) {
|
|
// Make sure to emit all elements of a Decl.
|
|
for (DeclGroupRef::iterator I = DG.begin(), E = DG.end(); I != E; ++I)
|
|
Builder->EmitTopLevelDecl(*I);
|
|
}
|
|
|
|
/// HandleTagDeclDefinition - This callback is invoked each time a TagDecl
|
|
/// to (e.g. struct, union, enum, class) is completed. This allows the
|
|
/// client hack on the type, which can occur at any point in the file
|
|
/// (because these can be defined in declspecs).
|
|
virtual void HandleTagDeclDefinition(TagDecl *D) {
|
|
Builder->UpdateCompletedType(D);
|
|
|
|
// In C++, we may have member functions that need to be emitted at this
|
|
// point.
|
|
if (Ctx->getLangOptions().CPlusPlus && !D->isDependentContext()) {
|
|
for (DeclContext::decl_iterator M = D->decls_begin(),
|
|
MEnd = D->decls_end();
|
|
M != MEnd; ++M)
|
|
if (CXXMethodDecl *Method = dyn_cast<CXXMethodDecl>(*M))
|
|
if (Method->isThisDeclarationADefinition() &&
|
|
(Method->hasAttr<UsedAttr>() ||
|
|
Method->hasAttr<ConstructorAttr>()))
|
|
Builder->EmitTopLevelDecl(Method);
|
|
}
|
|
}
|
|
|
|
virtual void HandleTranslationUnit(ASTContext &Ctx) {
|
|
if (Diags.hasErrorOccurred()) {
|
|
M.reset();
|
|
return;
|
|
}
|
|
|
|
if (Builder)
|
|
Builder->Release();
|
|
}
|
|
|
|
virtual void CompleteTentativeDefinition(VarDecl *D) {
|
|
if (Diags.hasErrorOccurred())
|
|
return;
|
|
|
|
Builder->EmitTentativeDefinition(D);
|
|
}
|
|
|
|
virtual void HandleVTable(CXXRecordDecl *RD, bool DefinitionRequired) {
|
|
if (Diags.hasErrorOccurred())
|
|
return;
|
|
|
|
Builder->EmitVTable(RD, DefinitionRequired);
|
|
}
|
|
};
|
|
}
|
|
|
|
CodeGenerator *clang::CreateLLVMCodeGen(Diagnostic &Diags,
|
|
const std::string& ModuleName,
|
|
const CodeGenOptions &CGO,
|
|
llvm::LLVMContext& C) {
|
|
return new CodeGeneratorImpl(Diags, ModuleName, CGO, C);
|
|
}
|