1 //===-------------- IRTransformLayer.cpp - IR Transform Layer -------------===//
2 //
3 //                     The LLVM Compiler Infrastructure
4 //
5 // This file is distributed under the University of Illinois Open Source
6 // License. See LICENSE.TXT for details.
7 //
8 //===----------------------------------------------------------------------===//
9 
10 #include "llvm/ExecutionEngine/Orc/IRTransformLayer.h"
11 #include "llvm/Support/MemoryBuffer.h"
12 
13 namespace llvm {
14 namespace orc {
15 
16 IRTransformLayer::IRTransformLayer(ExecutionSession &ES,
17                                      IRLayer &BaseLayer,
18                                      TransformFunction Transform)
19     : IRLayer(ES), BaseLayer(BaseLayer), Transform(std::move(Transform)) {}
20 
21 void IRTransformLayer::emit(MaterializationResponsibility R,
22                             ThreadSafeModule TSM) {
23   assert(TSM.getModule() && "Module must not be null");
24 
25   if (auto TransformedTSM = Transform(std::move(TSM), R))
26     BaseLayer.emit(std::move(R), std::move(*TransformedTSM));
27   else {
28     R.failMaterialization();
29     getExecutionSession().reportError(TransformedTSM.takeError());
30   }
31 }
32 
33 } // End namespace orc.
34 } // End namespace llvm.
35