| //===-- TestModuleFileExtension.h - Module Extension Tester -----*- C++ -*-===// |
| // |
| // Part of the LLVM Project, under the Apache License v2.0 with LLVM Exceptions. |
| // See https://llvm.org/LICENSE.txt for license information. |
| // SPDX-License-Identifier: Apache-2.0 WITH LLVM-exception |
| // |
| //===----------------------------------------------------------------------===// |
| #ifndef LLVM_CLANG_FRONTEND_TESTMODULEFILEEXTENSION_H |
| #define LLVM_CLANG_FRONTEND_TESTMODULEFILEEXTENSION_H |
| |
| #include "clang/Serialization/ModuleFileExtension.h" |
| #include "clang/Basic/LLVM.h" |
| #include "llvm/ADT/StringRef.h" |
| #include "llvm/Bitstream/BitstreamReader.h" |
| #include <string> |
| |
| namespace clang { |
| |
| /// A module file extension used for testing purposes. |
| class TestModuleFileExtension |
| : public llvm::RTTIExtends<TestModuleFileExtension, ModuleFileExtension> { |
| std::string BlockName; |
| unsigned MajorVersion; |
| unsigned MinorVersion; |
| bool Hashed; |
| std::string UserInfo; |
| |
| class Writer : public ModuleFileExtensionWriter { |
| public: |
| Writer(ModuleFileExtension *Ext) : ModuleFileExtensionWriter(Ext) { } |
| ~Writer() override; |
| |
| void writeExtensionContents(Sema &SemaRef, |
| llvm::BitstreamWriter &Stream) override; |
| }; |
| |
| class Reader : public ModuleFileExtensionReader { |
| llvm::BitstreamCursor Stream; |
| |
| public: |
| ~Reader() override; |
| |
| Reader(ModuleFileExtension *Ext, const llvm::BitstreamCursor &InStream); |
| }; |
| |
| public: |
| static char ID; |
| |
| TestModuleFileExtension(StringRef BlockName, unsigned MajorVersion, |
| unsigned MinorVersion, bool Hashed, |
| StringRef UserInfo) |
| : BlockName(BlockName), MajorVersion(MajorVersion), |
| MinorVersion(MinorVersion), Hashed(Hashed), UserInfo(UserInfo) {} |
| ~TestModuleFileExtension() override; |
| |
| ModuleFileExtensionMetadata getExtensionMetadata() const override; |
| |
| void hashExtension(ExtensionHashBuilder &HBuilder) const override; |
| |
| std::unique_ptr<ModuleFileExtensionWriter> |
| createExtensionWriter(ASTWriter &Writer) override; |
| |
| std::unique_ptr<ModuleFileExtensionReader> |
| createExtensionReader(const ModuleFileExtensionMetadata &Metadata, |
| ASTReader &Reader, serialization::ModuleFile &Mod, |
| const llvm::BitstreamCursor &Stream) override; |
| |
| std::string str() const; |
| }; |
| |
| } // end namespace clang |
| |
| #endif // LLVM_CLANG_FRONTEND_TESTMODULEFILEEXTENSION_H |