Trace.hpp

Go to the documentation of this file.

/*
 * Copyright 2025 Håvard Krogstie <krogstie.havard@gmail.com>
 * See COPYING for terms of redistribution.
 */
 
#ifndef JLM_RVSDG_TRACE_HPP
#define JLM_RVSDG_TRACE_HPP
 
#include <jlm/rvsdg/node.hpp>
 
namespace jlm::rvsdg
{
class GammaNode;
class ThetaNode;
 
class OutputTracer
{
public:
  virtual ~OutputTracer();
 
  OutputTracer();
 
  [[nodiscard]] bool
  isTracingThroughStructuralNodes() const noexcept
  {
    return traceThroughStrucutalNodes_;
  }
 
  void
  setTraceThroughStructuralNodes(bool value) noexcept
  {
    traceThroughStrucutalNodes_ = value;
  }
 
  [[nodiscard]] bool
  isEnteringPhiNodes() const noexcept
  {
    return enterPhiNodes_;
  }
 
  void
  setEnterPhiNodes(bool value) noexcept
  {
    enterPhiNodes_ = value;
  }
 
  [[nodiscard]] bool
  isInterprocedural() const noexcept
  {
    return isInterprocedural_;
  }
 
  void
  setInterprocedural(bool value) noexcept
  {
    isInterprocedural_ = value;
  }
 
  [[nodiscard]] Output &
  trace(Output & output);
 
  [[nodiscard]] Output *
  tryTraceThroughGamma(GammaNode & gammaNode, Output & output);
 
  [[nodiscard]] Output *
  tryTraceThroughTheta(ThetaNode & thetaNode, Output & output);
 
protected:
  [[nodiscard]] Output &
  trace(Output & output, bool mayLeaveRegion);
 
  [[nodiscard]] virtual Output &
  traceStep(Output & output, bool mayLeaveRegion);
 
  // When true, tracing enters subregions of structural nodes to check if the value is invariant.
  // When false, values are only considered invariant if they are directly connected to arguments.
  bool traceThroughStrucutalNodes_ = true;
 
  // When true, tracing can go from the output of a Phi node into its subregion.
  // When false, tracing will stop at the Phi output.
  bool enterPhiNodes_ = true;
 
  // When true, tracing is allowed to continue outside of lambda nodes.
  // When false, tracing will stop at the lambda's context arguments.
  bool isInterprocedural_ = true;
};
 
Output &
traceOutputIntraProcedurally(Output & output);
 
inline const Output &
traceOutputIntraProcedurally(const Output & output)
{
  return traceOutputIntraProcedurally(const_cast<Output &>(output));
}
 
Output &
traceOutput(Output & output);
 
inline const Output &
traceOutput(const Output & output)
{
  return traceOutput(const_cast<Output &>(output));
}
 
}
 
#endif // JLM_RVSDG_TRACE_HPP