xref: /freebsd-src/contrib/llvm-project/llvm/lib/ExecutionEngine/Orc/ObjectTransformLayer.cpp (revision fe6060f10f634930ff71b7c50291ddc610da2475)
10b57cec5SDimitry Andric //===---------- ObjectTransformLayer.cpp - Object Transform Layer ---------===//
20b57cec5SDimitry Andric //
30b57cec5SDimitry Andric // Part of the LLVM Project, under the Apache License v2.0 with LLVM Exceptions.
40b57cec5SDimitry Andric // See https://llvm.org/LICENSE.txt for license information.
50b57cec5SDimitry Andric // SPDX-License-Identifier: Apache-2.0 WITH LLVM-exception
60b57cec5SDimitry Andric //
70b57cec5SDimitry Andric //===----------------------------------------------------------------------===//
80b57cec5SDimitry Andric 
90b57cec5SDimitry Andric #include "llvm/ExecutionEngine/Orc/ObjectTransformLayer.h"
100b57cec5SDimitry Andric #include "llvm/Support/MemoryBuffer.h"
110b57cec5SDimitry Andric 
120b57cec5SDimitry Andric namespace llvm {
130b57cec5SDimitry Andric namespace orc {
140b57cec5SDimitry Andric 
15*fe6060f1SDimitry Andric char ObjectTransformLayer::ID;
16*fe6060f1SDimitry Andric 
17*fe6060f1SDimitry Andric using BaseT = RTTIExtends<ObjectTransformLayer, ObjectLayer>;
18*fe6060f1SDimitry Andric 
ObjectTransformLayer(ExecutionSession & ES,ObjectLayer & BaseLayer,TransformFunction Transform)190b57cec5SDimitry Andric ObjectTransformLayer::ObjectTransformLayer(ExecutionSession &ES,
200b57cec5SDimitry Andric                                            ObjectLayer &BaseLayer,
210b57cec5SDimitry Andric                                            TransformFunction Transform)
22*fe6060f1SDimitry Andric     : BaseT(ES), BaseLayer(BaseLayer), Transform(std::move(Transform)) {}
230b57cec5SDimitry Andric 
emit(std::unique_ptr<MaterializationResponsibility> R,std::unique_ptr<MemoryBuffer> O)24e8d8bef9SDimitry Andric void ObjectTransformLayer::emit(
25e8d8bef9SDimitry Andric     std::unique_ptr<MaterializationResponsibility> R,
260b57cec5SDimitry Andric     std::unique_ptr<MemoryBuffer> O) {
270b57cec5SDimitry Andric   assert(O && "Module must not be null");
280b57cec5SDimitry Andric 
29480093f4SDimitry Andric   // If there is a transform set then apply it.
30480093f4SDimitry Andric   if (Transform) {
310b57cec5SDimitry Andric     if (auto TransformedObj = Transform(std::move(O)))
32480093f4SDimitry Andric       O = std::move(*TransformedObj);
330b57cec5SDimitry Andric     else {
34e8d8bef9SDimitry Andric       R->failMaterialization();
350b57cec5SDimitry Andric       getExecutionSession().reportError(TransformedObj.takeError());
36480093f4SDimitry Andric       return;
370b57cec5SDimitry Andric     }
380b57cec5SDimitry Andric   }
390b57cec5SDimitry Andric 
40480093f4SDimitry Andric   BaseLayer.emit(std::move(R), std::move(O));
41480093f4SDimitry Andric }
42480093f4SDimitry Andric 
430b57cec5SDimitry Andric } // End namespace orc.
440b57cec5SDimitry Andric } // End namespace llvm.
45