From: Daniel Dunbar <daniel@zuster.org>
Date: Tue, 21 Oct 2008 19:55:09 +0000 (+0000)
Subject: Add GetModule accessor to ModuleBuilder
X-Git-Url: https://granicus.if.org/sourcecode?a=commitdiff_plain;h=d8c0ea1a80f27045c5371f6f97eadb38e8ca4279;p=clang

Add GetModule accessor to ModuleBuilder


git-svn-id: https://llvm.org/svn/llvm-project/cfe/trunk@57924 91177308-0d34-0410-b5e6-96231b3b80d8
---

diff --git a/include/clang/CodeGen/ModuleBuilder.h b/include/clang/CodeGen/ModuleBuilder.h
index 0f9b39341b..4642e4eeea 100644
--- a/include/clang/CodeGen/ModuleBuilder.h
+++ b/include/clang/CodeGen/ModuleBuilder.h
@@ -26,8 +26,9 @@ namespace clang {
   struct LangOptions;
   
   class CodeGenerator : public ASTConsumer {
-    public:
-      virtual llvm::Module* ReleaseModule() = 0;    
+  public:
+    virtual llvm::Module* GetModule() = 0;
+    virtual llvm::Module* ReleaseModule() = 0;    
   };
   
   CodeGenerator *CreateLLVMCodeGen(Diagnostic &Diags,
diff --git a/lib/CodeGen/ModuleBuilder.cpp b/lib/CodeGen/ModuleBuilder.cpp
index dafc0f5e25..be6e171bc4 100644
--- a/lib/CodeGen/ModuleBuilder.cpp
+++ b/lib/CodeGen/ModuleBuilder.cpp
@@ -48,6 +48,10 @@ namespace {
     
     virtual ~CodeGeneratorImpl() {}
     
+    virtual llvm::Module* GetModule() {
+      return M.get();
+    }
+    
     virtual llvm::Module* ReleaseModule() {
       return M.take();
     }