74 lines
2.4 KiB
C++
74 lines
2.4 KiB
C++
//===-- 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
|