Implement support for precompiled headers, preambles, and serialized

"main" files that import modules. When loading any of these kinds of
AST files, we make the modules that were imported visible into the
translation unit that loaded the PCH file or preamble.


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

1 files changed, 21 insertions, 0 deletions

diff --git a/lib/Serialization/ASTWriter.cpp b/lib/Serialization/ASTWriter.cpp
index 00dbfabb45..15db7050e9 100644
--- a/lib/Serialization/ASTWriter.cpp
+++ b/lib/Serialization/ASTWriter.cpp
@@ -3335,6 +3335,27 @@ void ASTWriter::WriteASTCore(Sema &SemaRef, MemorizeStatCalls *StatCalls,
        I != E; ++I)
     WriteDeclContextVisibleUpdate(*I);
 
+  // Write the submodules that were imported, if any.
+  RecordData ImportedModules;
+  for (ASTContext::import_iterator I = Context.local_import_begin(),
+                                IEnd = Context.local_import_end();
+       I != IEnd; ++I) {
+    assert(SubmoduleIDs.find(I->getImportedModule()) != SubmoduleIDs.end());
+    ImportedModules.push_back(SubmoduleIDs[I->getImportedModule()]);
+  }
+  if (!ImportedModules.empty()) {
+    // Sort module IDs.
+    llvm::array_pod_sort(ImportedModules.begin(), ImportedModules.end());
+    
+    // Unique module IDs.
+    ImportedModules.erase(std::unique(ImportedModules.begin(), 
+                                      ImportedModules.end()),
+                          ImportedModules.end());
+    
+    Stream.EmitRecord(IMPORTED_MODULES, ImportedModules);
+    ImportedModules.clear();
+  }
+  
   WriteDeclUpdatesBlocks();
   WriteDeclReplacementsBlock();
   WriteChainedObjCCategories();