Home | History | Annotate | Line # | Download | only in Frontend 
      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     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         MinorVersion(MinorVersion), Hashed(Hashed), UserInfo(UserInfo) {}
     54   ~TestModuleFileExtension() override;
     55 
     56   ModuleFileExtensionMetadata getExtensionMetadata() const override;
     57 
     58   llvm::hash_code hashExtension(llvm::hash_code Code) 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
     74