1//===- OffloadEmitter.cpp -------------------------------------------------===//
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#include "llvm/Object/OffloadBinary.h"
10#include "llvm/ObjectYAML/OffloadYAML.h"
11#include "llvm/ObjectYAML/yaml2obj.h"
12#include "llvm/Support/raw_ostream.h"
13
14using namespace llvm;
15using namespace OffloadYAML;
16
17namespace llvm {
18namespace yaml {
19
20bool yaml2offload(Binary &Doc, raw_ostream &Out, ErrorHandler EH) {
21 SmallVector<object::OffloadBinary::OffloadingImage> Images;
22 for (const auto &Member : Doc.Members) {
23 object::OffloadBinary::OffloadingImage Image{};
24 if (Member.ImageKind)
25 Image.TheImageKind = *Member.ImageKind;
26 if (Member.OffloadKind)
27 Image.TheOffloadKind = *Member.OffloadKind;
28 if (Member.Flags)
29 Image.Flags = *Member.Flags;
30
31 if (Member.StringEntries)
32 for (const auto &Entry : *Member.StringEntries)
33 Image.StringData[Entry.Key] = Entry.Value;
34
35 SmallVector<char, 1024> Data;
36 raw_svector_ostream OS(Data);
37 if (Member.Content)
38 Member.Content->writeAsBinary(OS);
39 Image.Image = MemoryBuffer::getMemBufferCopy(InputData: OS.str());
40 Images.push_back(Elt: std::move(Image));
41 }
42
43 // Copy the data to a new buffer so we can modify the bytes directly.
44 auto Buffer = object::OffloadBinary::write(OffloadingData: Images);
45 auto *TheHeader =
46 reinterpret_cast<object::OffloadBinary::Header *>(&Buffer[0]);
47 if (Doc.Version)
48 TheHeader->Version = *Doc.Version;
49 if (Doc.Size)
50 TheHeader->Size = *Doc.Size;
51 if (Doc.EntriesOffset)
52 TheHeader->EntriesOffset = *Doc.EntriesOffset;
53 if (Doc.EntriesCount)
54 TheHeader->EntriesCount = *Doc.EntriesCount;
55
56 Out.write(Ptr: Buffer.begin(), Size: Buffer.size());
57
58 return true;
59}
60
61} // namespace yaml
62} // namespace llvm
63