From: Daniel Dunbar <daniel@zuster.org>
Date: Thu, 12 Nov 2009 02:53:20 +0000 (+0000)
Subject: clang-cc: Factor ReadPCHFile out of ProcessInputFile.
X-Git-Url: https://granicus.if.org/sourcecode?a=commitdiff_plain;h=975790e8f88c427e2d3377f6cb03351a33e1f8c4;p=clang

clang-cc: Factor ReadPCHFile out of ProcessInputFile.

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

diff --git a/tools/clang-cc/clang-cc.cpp b/tools/clang-cc/clang-cc.cpp
index 2648a0b4d8..6981783ba9 100644
--- a/tools/clang-cc/clang-cc.cpp
+++ b/tools/clang-cc/clang-cc.cpp
@@ -631,6 +631,41 @@ static ASTConsumer *CreateConsumerAction(const CompilerInvocation &CompOpts,
   }
 }
 
+/// ReadPCHFile - Load a PCH file from disk, and initialize the preprocessor for
+/// reading from the PCH file.
+///
+/// \return The AST source, or null on failure.
+static ExternalASTSource *ReadPCHFile(llvm::StringRef Path,
+                                      const CompilerInvocation &CompOpts,
+                                      Preprocessor &PP,
+                                      ASTContext &Context) {
+  // If the user specified -isysroot, it will be used for relocatable PCH files.
+  const char *isysrootPCH = CompOpts.getHeaderSearchOpts().Sysroot.c_str();
+  if (isysrootPCH[0] == '\0')
+    isysrootPCH = 0;
+
+  llvm::OwningPtr<PCHReader> Reader;
+  Reader.reset(new PCHReader(PP, &Context, isysrootPCH));
+
+  switch (Reader->ReadPCH(Path)) {
+  case PCHReader::Success:
+    // Set the predefines buffer as suggested by the PCH reader. Typically, the
+    // predefines buffer will be empty.
+    PP.setPredefines(Reader->getSuggestedPredefines());
+    return Reader.take();
+
+  case PCHReader::Failure:
+    // Unrecoverable failure: don't even try to process the input file.
+    break;
+
+  case PCHReader::IgnorePCH:
+    // No suitable PCH file could be found. Return an error.
+    break;
+  }
+
+  return 0;
+}
+
 /// ProcessInputFile - Process a single input file with the specified state.
 ///
 static void ProcessInputFile(const CompilerInvocation &CompOpts,
@@ -731,40 +766,13 @@ static void ProcessInputFile(const CompilerInvocation &CompOpts,
                                       /* FreeMemory = */ !DisableFree,
                                       /* size_reserve = */0));
   if (Consumer && !ImplicitPCHInclude.empty()) {
-    // If the user specified -isysroot, it will be used for relocatable PCH
-    // files.
-    const char *isysrootPCH = CompOpts.getHeaderSearchOpts().Sysroot.c_str();
-    if (isysrootPCH[0] == '\0')
-      isysrootPCH = 0;
-
-    llvm::OwningPtr<PCHReader> Reader;
-    Reader.reset(new PCHReader(PP, ContextOwner.get(), isysrootPCH));
-
-    // The user has asked us to include a precompiled header. Load
-    // the precompiled header into the AST context.
-    switch (Reader->ReadPCH(ImplicitPCHInclude)) {
-    case PCHReader::Success: {
-      // Set the predefines buffer as suggested by the PCH
-      // reader. Typically, the predefines buffer will be empty.
-      PP.setPredefines(Reader->getSuggestedPredefines());
-
-      // Attach the PCH reader to the AST context as an external AST
-      // source, so that declarations will be deserialized from the
-      // PCH file as needed.
-      Source.reset(Reader.take());
-      ContextOwner->setExternalSource(Source);
-      break;
-    }
-
-    case PCHReader::Failure:
-      // Unrecoverable failure: don't even try to process the input
-      // file.
+    Source.reset(ReadPCHFile(ImplicitPCHInclude, CompOpts, PP, *ContextOwner));
+    if (!Source)
       return;
 
-    case PCHReader::IgnorePCH:
-      // No suitable PCH file could be found. Return an error.
-      return;
-    }
+    // Attach the PCH reader to the AST context as an external AST source, so
+    // that declarations will be deserialized from the PCH file as needed.
+    ContextOwner->setExternalSource(Source);
   }
 
   // Initialize the main file entry. This needs to be delayed until after PCH