xref: /freebsd/contrib/llvm-project/llvm/lib/ExecutionEngine/Orc/ObjectTransformLayer.cpp (revision 0fcececbac9880b092aeb56a41a16f1ec8ac1ae6)
1  //===---------- ObjectTransformLayer.cpp - Object Transform Layer ---------===//
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/ExecutionEngine/Orc/ObjectTransformLayer.h"
10  #include "llvm/Support/MemoryBuffer.h"
11  
12  namespace llvm {
13  namespace orc {
14  
15  char ObjectTransformLayer::ID;
16  
17  using BaseT = RTTIExtends<ObjectTransformLayer, ObjectLayer>;
18  
19  ObjectTransformLayer::ObjectTransformLayer(ExecutionSession &ES,
20                                             ObjectLayer &BaseLayer,
21                                             TransformFunction Transform)
22      : BaseT(ES), BaseLayer(BaseLayer), Transform(std::move(Transform)) {}
23  
24  void ObjectTransformLayer::emit(
25      std::unique_ptr<MaterializationResponsibility> R,
26      std::unique_ptr<MemoryBuffer> O) {
27    assert(O && "Module must not be null");
28  
29    // If there is a transform set then apply it.
30    if (Transform) {
31      if (auto TransformedObj = Transform(std::move(O)))
32        O = std::move(*TransformedObj);
33      else {
34        R->failMaterialization();
35        getExecutionSession().reportError(TransformedObj.takeError());
36        return;
37      }
38    }
39  
40    BaseLayer.emit(std::move(R), std::move(O));
41  }
42  
43  } // End namespace orc.
44  } // End namespace llvm.
45