Details | Last modification | View Log | RSS feed
| Rev | Author | Line No. | Line | 
|---|---|---|---|
| 14 | pmbaty | 1 | //===- OffloadYAML.h - Offload Binary YAMLIO implementation -----*- 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 | /// \file | ||
| 10 | /// This file declares classes for handling the YAML representation of | ||
| 11 | /// offloading binaries. | ||
| 12 | /// | ||
| 13 | //===----------------------------------------------------------------------===// | ||
| 14 | |||
| 15 | #ifndef LLVM_OBJECTYAML_OFFLOADYAML_H | ||
| 16 | #define LLVM_OBJECTYAML_OFFLOADYAML_H | ||
| 17 | |||
| 18 | #include "llvm/ADT/MapVector.h" | ||
| 19 | #include "llvm/Object/OffloadBinary.h" | ||
| 20 | #include "llvm/ObjectYAML/YAML.h" | ||
| 21 | #include "llvm/Support/YAMLTraits.h" | ||
| 22 | #include <optional> | ||
| 23 | |||
| 24 | namespace llvm { | ||
| 25 | namespace OffloadYAML { | ||
| 26 | |||
| 27 | struct Binary { | ||
| 28 | struct StringEntry { | ||
| 29 |     StringRef Key; | ||
| 30 |     StringRef Value; | ||
| 31 | }; | ||
| 32 | |||
| 33 | struct Member { | ||
| 34 | std::optional<object::ImageKind> ImageKind; | ||
| 35 | std::optional<object::OffloadKind> OffloadKind; | ||
| 36 | std::optional<uint32_t> Flags; | ||
| 37 | std::optional<std::vector<StringEntry>> StringEntries; | ||
| 38 | std::optional<yaml::BinaryRef> Content; | ||
| 39 | }; | ||
| 40 | |||
| 41 | std::optional<uint32_t> Version; | ||
| 42 | std::optional<uint64_t> Size; | ||
| 43 | std::optional<uint64_t> EntryOffset; | ||
| 44 | std::optional<uint64_t> EntrySize; | ||
| 45 | std::vector<Member> Members; | ||
| 46 | }; | ||
| 47 | |||
| 48 | } // end namespace OffloadYAML | ||
| 49 | } // end namespace llvm | ||
| 50 | |||
| 51 | LLVM_YAML_IS_SEQUENCE_VECTOR(llvm::OffloadYAML::Binary::Member) | ||
| 52 | LLVM_YAML_IS_SEQUENCE_VECTOR(llvm::OffloadYAML::Binary::StringEntry) | ||
| 53 | |||
| 54 | namespace llvm { | ||
| 55 | namespace yaml { | ||
| 56 | |||
| 57 | template <> struct ScalarEnumerationTraits<object::ImageKind> { | ||
| 58 | static void enumeration(IO &IO, object::ImageKind &Value); | ||
| 59 | }; | ||
| 60 | |||
| 61 | template <> struct ScalarEnumerationTraits<object::OffloadKind> { | ||
| 62 | static void enumeration(IO &IO, object::OffloadKind &Value); | ||
| 63 | }; | ||
| 64 | |||
| 65 | template <> struct MappingTraits<OffloadYAML::Binary> { | ||
| 66 | static void mapping(IO &IO, OffloadYAML::Binary &O); | ||
| 67 | }; | ||
| 68 | |||
| 69 | template <> struct MappingTraits<OffloadYAML::Binary::StringEntry> { | ||
| 70 | static void mapping(IO &IO, OffloadYAML::Binary::StringEntry &M); | ||
| 71 | }; | ||
| 72 | |||
| 73 | template <> struct MappingTraits<OffloadYAML::Binary::Member> { | ||
| 74 | static void mapping(IO &IO, OffloadYAML::Binary::Member &M); | ||
| 75 | }; | ||
| 76 | |||
| 77 | } // end namespace yaml | ||
| 78 | } // end namespace llvm | ||
| 79 | |||
| 80 | #endif // LLVM_OBJECTYAML_OFFLOADYAML_H |