109467b48Spatrick //===---------- ObjectTransformLayer.cpp - Object Transform Layer ---------===//
209467b48Spatrick //
309467b48Spatrick // Part of the LLVM Project, under the Apache License v2.0 with LLVM Exceptions.
409467b48Spatrick // See https://llvm.org/LICENSE.txt for license information.
509467b48Spatrick // SPDX-License-Identifier: Apache-2.0 WITH LLVM-exception
609467b48Spatrick //
709467b48Spatrick //===----------------------------------------------------------------------===//
809467b48Spatrick
909467b48Spatrick #include "llvm/ExecutionEngine/Orc/ObjectTransformLayer.h"
1009467b48Spatrick #include "llvm/Support/MemoryBuffer.h"
1109467b48Spatrick
1209467b48Spatrick namespace llvm {
1309467b48Spatrick namespace orc {
1409467b48Spatrick
15*73471bf0Spatrick char ObjectTransformLayer::ID;
16*73471bf0Spatrick
17*73471bf0Spatrick using BaseT = RTTIExtends<ObjectTransformLayer, ObjectLayer>;
18*73471bf0Spatrick
ObjectTransformLayer(ExecutionSession & ES,ObjectLayer & BaseLayer,TransformFunction Transform)1909467b48Spatrick ObjectTransformLayer::ObjectTransformLayer(ExecutionSession &ES,
2009467b48Spatrick ObjectLayer &BaseLayer,
2109467b48Spatrick TransformFunction Transform)
22*73471bf0Spatrick : BaseT(ES), BaseLayer(BaseLayer), Transform(std::move(Transform)) {}
2309467b48Spatrick
emit(std::unique_ptr<MaterializationResponsibility> R,std::unique_ptr<MemoryBuffer> O)24*73471bf0Spatrick void ObjectTransformLayer::emit(
25*73471bf0Spatrick std::unique_ptr<MaterializationResponsibility> R,
2609467b48Spatrick std::unique_ptr<MemoryBuffer> O) {
2709467b48Spatrick assert(O && "Module must not be null");
2809467b48Spatrick
2909467b48Spatrick // If there is a transform set then apply it.
3009467b48Spatrick if (Transform) {
3109467b48Spatrick if (auto TransformedObj = Transform(std::move(O)))
3209467b48Spatrick O = std::move(*TransformedObj);
3309467b48Spatrick else {
34*73471bf0Spatrick R->failMaterialization();
3509467b48Spatrick getExecutionSession().reportError(TransformedObj.takeError());
3609467b48Spatrick return;
3709467b48Spatrick }
3809467b48Spatrick }
3909467b48Spatrick
4009467b48Spatrick BaseLayer.emit(std::move(R), std::move(O));
4109467b48Spatrick }
4209467b48Spatrick
4309467b48Spatrick } // End namespace orc.
4409467b48Spatrick } // End namespace llvm.
45