mirror of
https://github.com/llvm/llvm-project.git
synced 2025-05-10 01:46:07 +00:00
f88e35ba0b
trap the serialized preprocessing records (macro definitions, macro instantiations, macro definitions) from the generation of the precompiled preamble, then replay those when walking the list of preprocessed entities. This eliminates a bug where clang_getCursor() wasn't able to find preprocessed-entity cursors in the preamble. llvm-svn: 120396
78 lines
2.4 KiB
C++
78 lines
2.4 KiB
C++
//===--- 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
|
|
// ASTConsumer that generates a PCH file.
|
|
//
|
|
//===----------------------------------------------------------------------===//
|
|
|
|
#include "clang/Frontend/ASTConsumers.h"
|
|
#include "clang/Serialization/ASTWriter.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 "clang/Basic/FileSystemStatCache.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), Chaining(Chaining) {
|
|
// 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 ASTReader's stat cache.
|
|
PP.getFileManager().addStatCache(StatCalls,
|
|
/*AtBeginning=*/!Chaining);
|
|
}
|
|
|
|
void PCHGenerator::HandleTranslationUnit(ASTContext &Ctx) {
|
|
if (PP.getDiagnostics().hasErrorOccurred())
|
|
return;
|
|
|
|
// Set up the serialization listener.
|
|
Writer.SetSerializationListener(GetASTSerializationListener());
|
|
|
|
// Emit the PCH file
|
|
assert(SemaPtr && "No Sema?");
|
|
Writer.WriteAST(*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();
|
|
}
|
|
|
|
ASTMutationListener *PCHGenerator::GetASTMutationListener() {
|
|
if (Chaining)
|
|
return &Writer;
|
|
return 0;
|
|
}
|
|
|
|
ASTSerializationListener *PCHGenerator::GetASTSerializationListener() {
|
|
return 0;
|
|
}
|
|
|
|
ASTDeserializationListener *PCHGenerator::GetASTDeserializationListener() {
|
|
return &Writer;
|
|
}
|