lib/CodeGen/ModuleBuilder.cpp

   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/Frontend/CompileOptions.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/Support/Compiler.h"
  26 #include "llvm/ADT/OwningPtr.h"
  27 using namespace clang;
  28
  29
  30 namespace {
  31   class VISIBILITY_HIDDEN CodeGeneratorImpl : public CodeGenerator {
  32     Diagnostic &Diags;
  33     llvm::OwningPtr<const llvm::TargetData> TD;
  34     ASTContext *Ctx;
  35     const CompileOptions CompileOpts;  // Intentionally copied in.
  36   protected:
  37     llvm::OwningPtr<llvm::Module> M;
  38     llvm::OwningPtr<CodeGen::CodeGenModule> Builder;
  39   public:
  40     CodeGeneratorImpl(Diagnostic &diags, const std::string& ModuleName,
  41                       const CompileOptions &CO, llvm::LLVMContext& C)
  42       : Diags(diags), CompileOpts(CO), M(new llvm::Module(ModuleName, C)) {}
  43
  44     virtual ~CodeGeneratorImpl() {}
  45
  46     virtual llvm::Module* GetModule() {
  47       return M.get();
  48     }
  49
  50     virtual llvm::Module* ReleaseModule() {
  51       return M.take();
  52     }
  53
  54     virtual void Initialize(ASTContext &Context) {
  55       Ctx = &Context;
  56
  57       M->setTargetTriple(Ctx->Target.getTargetTriple());
  58       M->setDataLayout(Ctx->Target.getTargetDescription());
  59       TD.reset(new llvm::TargetData(Ctx->Target.getTargetDescription()));
  60       Builder.reset(new CodeGen::CodeGenModule(Context, CompileOpts,
  61                                                *M, *TD, Diags));
  62     }
  63
  64     virtual void HandleTopLevelDecl(DeclGroupRef DG) {
  65       // Make sure to emit all elements of a Decl.
  66       for (DeclGroupRef::iterator I = DG.begin(), E = DG.end(); I != E; ++I)
  67         Builder->EmitTopLevelDecl(*I);
  68     }
  69
  70     /// HandleTagDeclDefinition - This callback is invoked each time a TagDecl
  71     /// to (e.g. struct, union, enum, class) is completed. This allows the
  72     /// client hack on the type, which can occur at any point in the file
  73     /// (because these can be defined in declspecs).
  74     virtual void HandleTagDeclDefinition(TagDecl *D) {
  75       Builder->UpdateCompletedType(D);
  76     }
  77
  78     virtual void HandleTranslationUnit(ASTContext &Ctx) {
  79       if (Diags.hasErrorOccurred()) {
  80         M.reset();
  81         return;
  82       }
  83
  84       if (Builder)
  85         Builder->Release();
  86     };
  87
  88     virtual void CompleteTentativeDefinition(VarDecl *D) {
  89       if (Diags.hasErrorOccurred())
  90         return;
  91
  92       Builder->EmitTentativeDefinition(D);
  93     }
  94   };
  95 }
  96
  97 CodeGenerator *clang::CreateLLVMCodeGen(Diagnostic &Diags,
  98                                         const std::string& ModuleName,
  99                                         const CompileOptions &CO,
 100                                         llvm::LLVMContext& C) {
 101   return new CodeGeneratorImpl(Diags, ModuleName, CO, C);
 102 }