diff options
author | Douglas Gregor <dgregor@apple.com> | 2010-08-17 18:31:01 +0000 |
---|---|---|
committer | Douglas Gregor <dgregor@apple.com> | 2010-08-17 18:31:01 +0000 |
commit | 256a7282e41a67bea753c054627619b0e35d80a7 (patch) | |
tree | 6a9223337c2729f8d9d733aeb60874b90a19297b /clang/lib/Frontend/GeneratePCH.cpp | |
parent | 1afba421a9bde494fe7f994ab74280056007edbb (diff) | |
download | bcm5719-llvm-256a7282e41a67bea753c054627619b0e35d80a7.tar.gz bcm5719-llvm-256a7282e41a67bea753c054627619b0e35d80a7.zip |
Revert Sebastian's build-breaking patch.
llvm-svn: 111265
Diffstat (limited to 'clang/lib/Frontend/GeneratePCH.cpp')
-rw-r--r-- | clang/lib/Frontend/GeneratePCH.cpp | 64 |
1 files changed, 64 insertions, 0 deletions
diff --git a/clang/lib/Frontend/GeneratePCH.cpp b/clang/lib/Frontend/GeneratePCH.cpp new file mode 100644 index 00000000000..33f5ef56ae3 --- /dev/null +++ b/clang/lib/Frontend/GeneratePCH.cpp @@ -0,0 +1,64 @@ +//===--- GeneratePCH.cpp - AST Consumer for PCH Generation ------*- C++ -*-===// +// +// The LLVM Compiler Infrastructure +// +// This file is distributed under the University of Illinois Open Source +// License. See LICENSE.TXT for details. +// +//===----------------------------------------------------------------------===// +// +// This file defines the CreatePCHGenerate function, which creates an +// ASTConsume that generates a PCH file. +// +//===----------------------------------------------------------------------===// + +#include "clang/Frontend/ASTConsumers.h" +#include "clang/Frontend/PCHWriter.h" +#include "clang/Sema/SemaConsumer.h" +#include "clang/AST/ASTContext.h" +#include "clang/AST/ASTConsumer.h" +#include "clang/Lex/Preprocessor.h" +#include "clang/Basic/FileManager.h" +#include "llvm/Bitcode/BitstreamWriter.h" +#include "llvm/Support/raw_ostream.h" +#include <string> + +using namespace clang; + +PCHGenerator::PCHGenerator(const Preprocessor &PP, + bool Chaining, + const char *isysroot, + llvm::raw_ostream *OS) + : PP(PP), isysroot(isysroot), Out(OS), SemaPtr(0), + StatCalls(0), Stream(Buffer), Writer(Stream) { + + // Install a stat() listener to keep track of all of the stat() + // calls. + StatCalls = new MemorizeStatCalls; + // If we have a chain, we want new stat calls only, so install the memorizer + // *after* the already installed PCHReader's stat cache. + PP.getFileManager().addStatCache(StatCalls, + /*AtBeginning=*/!Chaining); +} + +void PCHGenerator::HandleTranslationUnit(ASTContext &Ctx) { + if (PP.getDiagnostics().hasErrorOccurred()) + return; + + // Emit the PCH file + assert(SemaPtr && "No Sema?"); + Writer.WritePCH(*SemaPtr, StatCalls, isysroot); + + // Write the generated bitstream to "Out". + Out->write((char *)&Buffer.front(), Buffer.size()); + + // Make sure it hits disk now. + Out->flush(); + + // Free up some memory, in case the process is kept alive. + Buffer.clear(); +} + +PCHDeserializationListener *PCHGenerator::GetPCHDeserializationListener() { + return &Writer; +} |