ModuleBuilder.cpp revision 201361
1//===--- ModuleBuilder.cpp - Emit LLVM Code from ASTs ---------------------===// 2// 3// The LLVM Compiler Infrastructure 4// 5// This file is distributed under the University of Illinois Open Source 6// License. See LICENSE.TXT for details. 7// 8//===----------------------------------------------------------------------===// 9// 10// This builds an AST and converts it to LLVM Code. 11// 12//===----------------------------------------------------------------------===// 13 14#include "clang/CodeGen/ModuleBuilder.h" 15#include "CodeGenModule.h" 16#include "clang/CodeGen/CodeGenOptions.h" 17#include "clang/AST/ASTContext.h" 18#include "clang/AST/DeclObjC.h" 19#include "clang/AST/Expr.h" 20#include "clang/Basic/Diagnostic.h" 21#include "clang/Basic/TargetInfo.h" 22#include "llvm/LLVMContext.h" 23#include "llvm/Module.h" 24#include "llvm/Target/TargetData.h" 25#include "llvm/ADT/OwningPtr.h" 26using namespace clang; 27 28namespace { 29 class CodeGeneratorImpl : public CodeGenerator { 30 Diagnostic &Diags; 31 llvm::OwningPtr<const llvm::TargetData> TD; 32 ASTContext *Ctx; 33 const CodeGenOptions CodeGenOpts; // Intentionally copied in. 34 protected: 35 llvm::OwningPtr<llvm::Module> M; 36 llvm::OwningPtr<CodeGen::CodeGenModule> Builder; 37 public: 38 CodeGeneratorImpl(Diagnostic &diags, const std::string& ModuleName, 39 const CodeGenOptions &CGO, llvm::LLVMContext& C) 40 : Diags(diags), CodeGenOpts(CGO), M(new llvm::Module(ModuleName, C)) {} 41 42 virtual ~CodeGeneratorImpl() {} 43 44 virtual llvm::Module* GetModule() { 45 return M.get(); 46 } 47 48 virtual llvm::Module* ReleaseModule() { 49 return M.take(); 50 } 51 52 virtual void Initialize(ASTContext &Context) { 53 Ctx = &Context; 54 55 M->setTargetTriple(Ctx->Target.getTriple().getTriple()); 56 M->setDataLayout(Ctx->Target.getTargetDescription()); 57 TD.reset(new llvm::TargetData(Ctx->Target.getTargetDescription())); 58 Builder.reset(new CodeGen::CodeGenModule(Context, CodeGenOpts, 59 *M, *TD, Diags)); 60 } 61 62 virtual void HandleTopLevelDecl(DeclGroupRef DG) { 63 // Make sure to emit all elements of a Decl. 64 for (DeclGroupRef::iterator I = DG.begin(), E = DG.end(); I != E; ++I) 65 Builder->EmitTopLevelDecl(*I); 66 } 67 68 /// HandleTagDeclDefinition - This callback is invoked each time a TagDecl 69 /// to (e.g. struct, union, enum, class) is completed. This allows the 70 /// client hack on the type, which can occur at any point in the file 71 /// (because these can be defined in declspecs). 72 virtual void HandleTagDeclDefinition(TagDecl *D) { 73 Builder->UpdateCompletedType(D); 74 } 75 76 virtual void HandleTranslationUnit(ASTContext &Ctx) { 77 if (Diags.hasErrorOccurred()) { 78 M.reset(); 79 return; 80 } 81 82 if (Builder) 83 Builder->Release(); 84 } 85 86 virtual void CompleteTentativeDefinition(VarDecl *D) { 87 if (Diags.hasErrorOccurred()) 88 return; 89 90 Builder->EmitTentativeDefinition(D); 91 } 92 }; 93} 94 95CodeGenerator *clang::CreateLLVMCodeGen(Diagnostic &Diags, 96 const std::string& ModuleName, 97 const CodeGenOptions &CGO, 98 llvm::LLVMContext& C) { 99 return new CodeGeneratorImpl(Diags, ModuleName, CGO, C); 100} 101