add GetNode method to output (symmetric to input) (#644)

phate · Oct 1, 2024 · 8e11f8c · 8e11f8c
1 parent 99fe50b
commit 8e11f8c
Show file tree

Hide file tree

Showing 2 changed files with 18 additions and 0 deletions.
diff --git a/jlm/rvsdg/node.cpp b/jlm/rvsdg/node.cpp
@@ -95,6 +95,13 @@ output::debug_string() const
   return jlm::util::strfmt(index());
 }
 
+rvsdg::node *
+output::GetNode(const rvsdg::output & output) noexcept
+{
+  auto nodeOutput = dynamic_cast<const rvsdg::node_output *>(&output);
+  return nodeOutput ? nodeOutput->node() : nullptr;
+}
+
 void
 output::remove_user(jlm::rvsdg::input * user)
 {

diff --git a/jlm/rvsdg/node.hpp b/jlm/rvsdg/node.hpp
@@ -374,6 +374,17 @@ class output
   virtual std::string
   debug_string() const;
 
+  /**
+   * Retrieve the associated node from \p output if \p output is derived from
+   * jlm::rvsdg::node_output.
+   *
+   * @param output The output from which to retrieve the node.
+   * @return The node associated with \p output if output is derived from jlm::rvsdg::node_output,
+   * otherwise nullptr.
+   */
+  [[nodiscard]] static rvsdg::node *
+  GetNode(const rvsdg::output & output) noexcept;
+
   template<class T>
   class iterator
   {