revng-revng/lib/Pipes/FunctionPass.cpp

/// \file FunctionPass.cpp
/// Contains the implementation of pipeline passes.

///
//
// This file is distributed under the MIT License. See LICENSE.md for details.
//

#include "llvm/Support/Progress.h"

#include "revng/Model/LoadModelPass.h"
#include "revng/Pipeline/Context.h"
#include "revng/Pipeline/ExecutionContext.h"
#include "revng/Pipeline/LLVMKind.h"
#include "revng/Pipes/FunctionPass.h"
#include "revng/Pipes/TaggedFunctionKind.h"
#include "revng/Support/FunctionTags.h"

bool pipeline::detail::runOnModule(llvm::Module &Module,
                                   FunctionPassImpl &Pipe) {
  auto &Analysis = Pipe.getAnalysis<pipeline::LoadExecutionContextPass>();

  // Obtain the context
  ExecutionContext *EC = Analysis.get();
  revng_assert(EC != nullptr);

  // Run the prologue
  auto &ModelWrapper = Pipe.getAnalysis<LoadModelWrapperPass>().get();
  bool Result = Pipe.prologue();

  // Run on individual functions
  using Type = revng::kinds::TaggedFunctionKind;
  auto ContainerName = Analysis.getContainerName();
  auto ToIterOn = Type::getFunctionsAndCommit(*EC, Module, ContainerName);
  llvm::Task T(Analysis.getRequestedTargets().size(), "Running FunctionPass");
  for (const auto &[ModelFunction, LLVMFunction] : ToIterOn) {
    T.advance(ModelFunction->Entry().toString(), true);
    Result = Pipe.runOnFunction(*ModelFunction, *LLVMFunction) or Result;
  }

  Result = Pipe.epilogue() or Result;

  return Result;
}