1 //===--- IncrementalExecutor.h - Incremental Execution ----------*- C++ -*-===//
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 // This file implements the class which performs incremental code execution.
10 //
11 //===----------------------------------------------------------------------===//
12 
13 #ifndef LLVM_CLANG_LIB_INTERPRETER_INCREMENTALEXECUTOR_H
14 #define LLVM_CLANG_LIB_INTERPRETER_INCREMENTALEXECUTOR_H
15 
16 #include "llvm/ADT/DenseMap.h"
17 #include "llvm/ADT/StringRef.h"
18 #include "llvm/ExecutionEngine/Orc/ExecutionUtils.h"
19 
20 #include <memory>
21 
22 namespace llvm {
23 class Error;
24 namespace orc {
25 class LLJIT;
26 class ThreadSafeContext;
27 } // namespace orc
28 } // namespace llvm
29 
30 namespace clang {
31 
32 struct PartialTranslationUnit;
33 class TargetInfo;
34 
35 class IncrementalExecutor {
36   using CtorDtorIterator = llvm::orc::CtorDtorIterator;
37   std::unique_ptr<llvm::orc::LLJIT> Jit;
38   llvm::orc::ThreadSafeContext &TSCtx;
39 
40   llvm::DenseMap<const PartialTranslationUnit *, llvm::orc::ResourceTrackerSP>
41       ResourceTrackers;
42 
43 public:
44   enum SymbolNameKind { IRName, LinkerName };
45 
46   IncrementalExecutor(llvm::orc::ThreadSafeContext &TSC, llvm::Error &Err,
47                       const clang::TargetInfo &TI);
48   ~IncrementalExecutor();
49 
50   llvm::Error addModule(PartialTranslationUnit &PTU);
51   llvm::Error removeModule(PartialTranslationUnit &PTU);
52   llvm::Error runCtors() const;
53   llvm::Error cleanUp();
54   llvm::Expected<llvm::JITTargetAddress>
55   getSymbolAddress(llvm::StringRef Name, SymbolNameKind NameKind) const;
56   llvm::orc::LLJIT *getExecutionEngine() const { return Jit.get(); }
57 };
58 
59 } // end namespace clang
60 
61 #endif // LLVM_CLANG_LIB_INTERPRETER_INCREMENTALEXECUTOR_H
62