61 lines · cpp
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-exception6//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 for (const auto &Member : Doc.Members) {22 object::OffloadBinary::OffloadingImage Image{};23 if (Member.ImageKind)24 Image.TheImageKind = *Member.ImageKind;25 if (Member.OffloadKind)26 Image.TheOffloadKind = *Member.OffloadKind;27 if (Member.Flags)28 Image.Flags = *Member.Flags;29 30 if (Member.StringEntries)31 for (const auto &Entry : *Member.StringEntries)32 Image.StringData[Entry.Key] = Entry.Value;33 34 SmallVector<char, 1024> Data;35 raw_svector_ostream OS(Data);36 if (Member.Content)37 Member.Content->writeAsBinary(OS);38 Image.Image = MemoryBuffer::getMemBufferCopy(OS.str());39 40 // Copy the data to a new buffer so we can modify the bytes directly.41 auto Buffer = object::OffloadBinary::write(Image);42 auto *TheHeader =43 reinterpret_cast<object::OffloadBinary::Header *>(&Buffer[0]);44 if (Doc.Version)45 TheHeader->Version = *Doc.Version;46 if (Doc.Size)47 TheHeader->Size = *Doc.Size;48 if (Doc.EntryOffset)49 TheHeader->EntryOffset = *Doc.EntryOffset;50 if (Doc.EntrySize)51 TheHeader->EntrySize = *Doc.EntrySize;52 53 Out.write(Buffer.begin(), Buffer.size());54 }55 56 return true;57}58 59} // namespace yaml60} // namespace llvm61