ModuleBuilder.cpp revision 285830 
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 "CGDebugInfo.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 "clang/Frontend/CodeGenOptions.h"
23#include "llvm/ADT/OwningPtr.h"
24#include "llvm/ADT/StringRef.h"
25#include "llvm/IR/DataLayout.h"
26#include "llvm/IR/LLVMContext.h"
27#include "llvm/IR/Module.h"
28using namespace clang;
29
30namespace {
31  class CodeGeneratorImpl : public CodeGenerator {
32    DiagnosticsEngine &Diags;
33    OwningPtr<const llvm::DataLayout> TD;
34    ASTContext *Ctx;
35    const CodeGenOptions CodeGenOpts;  // Intentionally copied in.
36  protected:
37    OwningPtr<llvm::Module> M;
38    OwningPtr<CodeGen::CodeGenModule> Builder;
39  public:
40    CodeGeneratorImpl(DiagnosticsEngine &diags, const std::string& ModuleName,
41                      const CodeGenOptions &CGO, llvm::LLVMContext& C)
42      : Diags(diags), CodeGenOpts(CGO),
43        M(new llvm::Module(ModuleName, C)) {}
44
45    virtual ~CodeGeneratorImpl() {}
46
47    virtual llvm::Module* GetModule() {
48      return M.get();
49    }
50
51    virtual llvm::Module* ReleaseModule() {
52      return M.take();
53    }
54
55    virtual void Initialize(ASTContext &Context) {
56      Ctx = &Context;
57
58      M->setTargetTriple(Ctx->getTargetInfo().getTriple().getTriple());
59      M->setDataLayout(Ctx->getTargetInfo().getTargetDescription());
60      TD.reset(new llvm::DataLayout(Ctx->getTargetInfo().getTargetDescription()));
61      Builder.reset(new CodeGen::CodeGenModule(Context, CodeGenOpts, *M, *TD,
62                                               Diags));
63
64      for (size_t i = 0, e = CodeGenOpts.DependentLibraries.size(); i < e; ++i)
65        HandleDependentLibrary(CodeGenOpts.DependentLibraries[i]);
66    }
67
68    virtual void HandleCXXStaticMemberVarInstantiation(VarDecl *VD) {
69      if (Diags.hasErrorOccurred())
70        return;
71
72      Builder->HandleCXXStaticMemberVarInstantiation(VD);
73    }
74
75    virtual bool HandleTopLevelDecl(DeclGroupRef DG) {
76      if (Diags.hasErrorOccurred())
77        return true;
78
79      // Make sure to emit all elements of a Decl.
80      for (DeclGroupRef::iterator I = DG.begin(), E = DG.end(); I != E; ++I)
81        Builder->EmitTopLevelDecl(*I);
82      return true;
83    }
84
85    /// HandleTagDeclDefinition - This callback is invoked each time a TagDecl
86    /// to (e.g. struct, union, enum, class) is completed. This allows the
87    /// client hack on the type, which can occur at any point in the file
88    /// (because these can be defined in declspecs).
89    virtual void HandleTagDeclDefinition(TagDecl *D) {
90      if (Diags.hasErrorOccurred())
91        return;
92
93      Builder->UpdateCompletedType(D);
94
95      // In C++, we may have member functions that need to be emitted at this
96      // point.
97      if (Ctx->getLangOpts().CPlusPlus && !D->isDependentContext()) {
98        for (DeclContext::decl_iterator M = D->decls_begin(),
99                                     MEnd = D->decls_end();
100             M != MEnd; ++M)
101          if (CXXMethodDecl *Method = dyn_cast<CXXMethodDecl>(*M))
102            if (Method->doesThisDeclarationHaveABody() &&
103                (Method->hasAttr<UsedAttr>() ||
104                 Method->hasAttr<ConstructorAttr>()))
105              Builder->EmitTopLevelDecl(Method);
106      }
107    }
108
109    virtual void HandleTagDeclRequiredDefinition(const TagDecl *D) LLVM_OVERRIDE {
110      if (Diags.hasErrorOccurred())
111        return;
112
113      if (CodeGen::CGDebugInfo *DI = Builder->getModuleDebugInfo())
114        if (const RecordDecl *RD = dyn_cast<RecordDecl>(D))
115          DI->completeRequiredType(RD);
116    }
117
118    virtual void HandleTranslationUnit(ASTContext &Ctx) {
119      if (Diags.hasErrorOccurred()) {
120        M.reset();
121        return;
122      }
123
124      if (Builder)
125        Builder->Release();
126    }
127
128    virtual void CompleteTentativeDefinition(VarDecl *D) {
129      if (Diags.hasErrorOccurred())
130        return;
131
132      Builder->EmitTentativeDefinition(D);
133    }
134
135    virtual void HandleVTable(CXXRecordDecl *RD, bool DefinitionRequired) {
136      if (Diags.hasErrorOccurred())
137        return;
138
139      Builder->EmitVTable(RD, DefinitionRequired);
140    }
141
142    virtual void HandleLinkerOptionPragma(llvm::StringRef Opts) {
143      Builder->AppendLinkerOptions(Opts);
144    }
145
146    virtual void HandleDetectMismatch(llvm::StringRef Name,
147                                      llvm::StringRef Value) {
148      Builder->AddDetectMismatch(Name, Value);
149    }
150
151    virtual void HandleDependentLibrary(llvm::StringRef Lib) {
152      Builder->AddDependentLib(Lib);
153    }
154  };
155}
156
157void CodeGenerator::anchor() { }
158
159CodeGenerator *clang::CreateLLVMCodeGen(DiagnosticsEngine &Diags,
160                                        const std::string& ModuleName,
161                                        const CodeGenOptions &CGO,
162                                        const TargetOptions &/*TO*/,
163                                        llvm::LLVMContext& C) {
164  return new CodeGeneratorImpl(Diags, ModuleName, CGO, C);
165}
166