llvm/lib/ExecutionEngine/Orc/ObjectTransformLayer.cpp

   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.