mirror of
https://github.com/llvm/llvm-project.git
synced 2025-05-16 16:37:11 +00:00
10b2368e9f
Store in PCH the directory that the PCH was originally created in. If a header file is not found at the path that we expect it to be and the PCH file was moved from its original location, try to resolve the file by assuming that header+PCH were moved together and the header is in the same place relative to the PCH. llvm-svn: 125576
79 lines
2.5 KiB
C++
79 lines
2.5 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,
|
|
const std::string &OutputFile,
|
|
bool Chaining,
|
|
const char *isysroot,
|
|
llvm::raw_ostream *OS)
|
|
: PP(PP), OutputFile(OutputFile), 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, OutputFile, 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;
|
|
}
|