clang 20.0.0git
PCHContainerOperations.h
Go to the documentation of this file.
1//===-- PCHContainerOperations.h - PCH Containers ---------------*- C++ -*-===//
2//
3// Part of the LLVM Project, under the Apache License v2.0 with LLVM Exceptions.
4// See https://llvm.org/LICENSE.txt for license information.
5// SPDX-License-Identifier: Apache-2.0 WITH LLVM-exception
6//
7//===----------------------------------------------------------------------===//
8
9#ifndef LLVM_CLANG_SERIALIZATION_PCHCONTAINEROPERATIONS_H
10#define LLVM_CLANG_SERIALIZATION_PCHCONTAINEROPERATIONS_H
11
12#include "clang/Basic/Module.h"
13#include "llvm/ADT/SmallVector.h"
14#include "llvm/ADT/StringMap.h"
15#include "llvm/Support/MemoryBufferRef.h"
16#include <memory>
17
18namespace llvm {
19class raw_pwrite_stream;
20}
21
22namespace clang {
23
24class ASTConsumer;
25class CompilerInstance;
26
27struct PCHBuffer {
31};
32
33/// This abstract interface provides operations for creating
34/// containers for serialized ASTs (precompiled headers and clang
35/// modules).
37public:
38 virtual ~PCHContainerWriter() = 0;
39 virtual llvm::StringRef getFormat() const = 0;
40
41 /// Return an ASTConsumer that can be chained with a
42 /// PCHGenerator that produces a wrapper file format containing a
43 /// serialized AST bitstream.
44 virtual std::unique_ptr<ASTConsumer>
46 const std::string &MainFileName,
47 const std::string &OutputFileName,
48 std::unique_ptr<llvm::raw_pwrite_stream> OS,
49 std::shared_ptr<PCHBuffer> Buffer) const = 0;
50};
51
52/// This abstract interface provides operations for unwrapping
53/// containers for serialized ASTs (precompiled headers and clang
54/// modules).
56public:
57 virtual ~PCHContainerReader() = 0;
58 /// Equivalent to the format passed to -fmodule-format=
60
61 /// Returns the serialized AST inside the PCH container Buffer.
62 virtual llvm::StringRef ExtractPCH(llvm::MemoryBufferRef Buffer) const = 0;
63};
64
65/// Implements write operations for a raw pass-through PCH container.
67 llvm::StringRef getFormat() const override { return "raw"; }
68
69 /// Return an ASTConsumer that can be chained with a
70 /// PCHGenerator that writes the module to a flat file.
71 std::unique_ptr<ASTConsumer>
72 CreatePCHContainerGenerator(CompilerInstance &CI,
73 const std::string &MainFileName,
74 const std::string &OutputFileName,
75 std::unique_ptr<llvm::raw_pwrite_stream> OS,
76 std::shared_ptr<PCHBuffer> Buffer) const override;
77};
78
79/// Implements read operations for a raw pass-through PCH container.
81 llvm::ArrayRef<llvm::StringRef> getFormats() const override;
82 /// Simply returns the buffer contained in Buffer.
83 llvm::StringRef ExtractPCH(llvm::MemoryBufferRef Buffer) const override;
84};
85
86/// A registry of PCHContainerWriter and -Reader objects for different formats.
88 llvm::StringMap<std::unique_ptr<PCHContainerWriter>> Writers;
89 llvm::StringMap<PCHContainerReader *> Readers;
91
92public:
93 /// Automatically registers a RawPCHContainerWriter and
94 /// RawPCHContainerReader.
96 void registerWriter(std::unique_ptr<PCHContainerWriter> Writer) {
97 Writers[Writer->getFormat()] = std::move(Writer);
98 }
99 void registerReader(std::unique_ptr<PCHContainerReader> Reader) {
100 assert(!Reader->getFormats().empty() &&
101 "PCHContainerReader must handle >=1 format");
102 for (llvm::StringRef Fmt : Reader->getFormats())
103 Readers[Fmt] = Reader.get();
104 OwnedReaders.push_back(std::move(Reader));
105 }
106 const PCHContainerWriter *getWriterOrNull(llvm::StringRef Format) {
107 return Writers[Format].get();
108 }
109 const PCHContainerReader *getReaderOrNull(llvm::StringRef Format) {
110 return Readers[Format];
111 }
113 return *getReaderOrNull("raw");
114 }
115};
116
117}
118
119#endif
Defines the clang::Module class, which describes a module in the source code.
CompilerInstance - Helper class for managing a single instance of the Clang compiler.
A registry of PCHContainerWriter and -Reader objects for different formats.
void registerReader(std::unique_ptr< PCHContainerReader > Reader)
PCHContainerOperations()
Automatically registers a RawPCHContainerWriter and RawPCHContainerReader.
const PCHContainerReader * getReaderOrNull(llvm::StringRef Format)
const PCHContainerWriter * getWriterOrNull(llvm::StringRef Format)
void registerWriter(std::unique_ptr< PCHContainerWriter > Writer)
const PCHContainerReader & getRawReader()
This abstract interface provides operations for unwrapping containers for serialized ASTs (precompile...
virtual llvm::ArrayRef< llvm::StringRef > getFormats() const =0
Equivalent to the format passed to -fmodule-format=.
virtual llvm::StringRef ExtractPCH(llvm::MemoryBufferRef Buffer) const =0
Returns the serialized AST inside the PCH container Buffer.
This abstract interface provides operations for creating containers for serialized ASTs (precompiled ...
virtual std::unique_ptr< ASTConsumer > CreatePCHContainerGenerator(CompilerInstance &CI, const std::string &MainFileName, const std::string &OutputFileName, std::unique_ptr< llvm::raw_pwrite_stream > OS, std::shared_ptr< PCHBuffer > Buffer) const =0
Return an ASTConsumer that can be chained with a PCHGenerator that produces a wrapper file format con...
virtual llvm::StringRef getFormat() const =0
Implements read operations for a raw pass-through PCH container.
Implements write operations for a raw pass-through PCH container.
The JSON file list parser is used to communicate input to InstallAPI.
Diagnostic wrappers for TextAPI types for error reporting.
Definition: Dominators.h:30
The signature of a module, which is a hash of the AST content.
Definition: Module.h:57
llvm::SmallVector< char, 0 > Data
ASTFileSignature Signature