mirror of
https://github.com/revng/revng
synced 2026-06-21 14:07:57 +00:00
Giacomo Vercesi
19f67c8223
Add a mixin class for piperuns that work on an LLVM module and need both the model function and the `llvm::Function`. Pipe that inherit this mixin will receive both the model and LLVM function in the `runOnLLVMFunction` method.
44 lines
1.5 KiB
C++
44 lines
1.5 KiB
C++
#pragma once
|
|
|
|
//
|
|
// This file is distributed under the MIT License. See LICENSE.md for details.
|
|
//
|
|
|
|
#include "llvm/IR/Function.h"
|
|
|
|
#include "revng/Model/IRHelpers.h"
|
|
#include "revng/PipeboxCommon/LLVMContainer.h"
|
|
|
|
namespace revng::pypeline::piperuns {
|
|
|
|
template<typename T>
|
|
concept HasRunOnLLVMFunctionFunction = requires(T &PipeRun,
|
|
const model::Function &Function,
|
|
llvm::Function &LLVMFunction) {
|
|
{ PipeRun.runOnLLVMFunction(Function, LLVMFunction) } -> std::same_as<void>;
|
|
};
|
|
|
|
/// Mixin CRTP class that allows a pipe run to define
|
|
/// `runOnLLVMFunction(const model::Function &, llvm::Function &)` instead of
|
|
/// `runOnFunction`, which this class will take care to implement.
|
|
template<typename Derived>
|
|
class LLVMFunctionMixin {
|
|
private:
|
|
revng::pypeline::LLVMFunctionContainer &ModuleContainer;
|
|
|
|
public:
|
|
LLVMFunctionMixin(revng::pypeline::LLVMFunctionContainer &ModuleContainer) :
|
|
ModuleContainer(ModuleContainer) {}
|
|
|
|
void runOnFunction(const model::Function &Function) {
|
|
static_assert(HasRunOnLLVMFunctionFunction<Derived>);
|
|
llvm::Module &Module = ModuleContainer
|
|
.getModule(ObjectID(Function.Entry()));
|
|
llvm::Function &LLVMFunction = getUniqueIsolatedFunction(Module,
|
|
Function.Entry());
|
|
static_cast<Derived *>(this)->runOnLLVMFunction(Function, LLVMFunction);
|
|
}
|
|
};
|
|
|
|
} // namespace revng::pypeline::piperuns
|