/Users/buildslave/jenkins/workspace/coverage/llvm-project/clang/lib/Frontend/TestModuleFileExtension.h
Line | Count | Source |
1 | | //===-- TestModuleFileExtension.h - Module Extension Tester -----*- 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 | | #ifndef LLVM_CLANG_FRONTEND_TESTMODULEFILEEXTENSION_H |
9 | | #define LLVM_CLANG_FRONTEND_TESTMODULEFILEEXTENSION_H |
10 | | |
11 | | #include "clang/Serialization/ModuleFileExtension.h" |
12 | | #include "clang/Basic/LLVM.h" |
13 | | #include "llvm/ADT/StringRef.h" |
14 | | #include "llvm/Bitstream/BitstreamReader.h" |
15 | | #include <string> |
16 | | |
17 | | namespace clang { |
18 | | |
19 | | /// A module file extension used for testing purposes. |
20 | | class TestModuleFileExtension |
21 | | : public llvm::RTTIExtends<TestModuleFileExtension, ModuleFileExtension> { |
22 | | std::string BlockName; |
23 | | unsigned MajorVersion; |
24 | | unsigned MinorVersion; |
25 | | bool Hashed; |
26 | | std::string UserInfo; |
27 | | |
28 | | class Writer : public ModuleFileExtensionWriter { |
29 | | public: |
30 | 12 | Writer(ModuleFileExtension *Ext) : ModuleFileExtensionWriter(Ext) { } |
31 | | ~Writer() override; |
32 | | |
33 | | void writeExtensionContents(Sema &SemaRef, |
34 | | llvm::BitstreamWriter &Stream) override; |
35 | | }; |
36 | | |
37 | | class Reader : public ModuleFileExtensionReader { |
38 | | llvm::BitstreamCursor Stream; |
39 | | |
40 | | public: |
41 | | ~Reader() override; |
42 | | |
43 | | Reader(ModuleFileExtension *Ext, const llvm::BitstreamCursor &InStream); |
44 | | }; |
45 | | |
46 | | public: |
47 | | static char ID; |
48 | | |
49 | | TestModuleFileExtension(StringRef BlockName, unsigned MajorVersion, |
50 | | unsigned MinorVersion, bool Hashed, |
51 | | StringRef UserInfo) |
52 | | : BlockName(BlockName), MajorVersion(MajorVersion), |
53 | 36 | MinorVersion(MinorVersion), Hashed(Hashed), UserInfo(UserInfo) {} |
54 | | ~TestModuleFileExtension() override; |
55 | | |
56 | | ModuleFileExtensionMetadata getExtensionMetadata() const override; |
57 | | |
58 | | void hashExtension(ExtensionHashBuilder &HBuilder) const override; |
59 | | |
60 | | std::unique_ptr<ModuleFileExtensionWriter> |
61 | | createExtensionWriter(ASTWriter &Writer) override; |
62 | | |
63 | | std::unique_ptr<ModuleFileExtensionReader> |
64 | | createExtensionReader(const ModuleFileExtensionMetadata &Metadata, |
65 | | ASTReader &Reader, serialization::ModuleFile &Mod, |
66 | | const llvm::BitstreamCursor &Stream) override; |
67 | | |
68 | | std::string str() const; |
69 | | }; |
70 | | |
71 | | } // end namespace clang |
72 | | |
73 | | #endif // LLVM_CLANG_FRONTEND_TESTMODULEFILEEXTENSION_H |