TestNodes.hpp

Go to the documentation of this file.

/*
 * Copyright 2025 Nico Reißmann <nico.reissmann@gmail.com>
 * See COPYING for terms of redistribution.
 */
 
#ifndef JLM_RVSDG_TESTNODES_HPP
#define JLM_RVSDG_TESTNODES_HPP
 
#include <jlm/rvsdg/operation.hpp>
#include <jlm/rvsdg/structural-node.hpp>
 
namespace jlm::rvsdg
{
 
class TestStructuralOperation final : public StructuralOperation
{
public:
  ~TestStructuralOperation() noexcept override;
 
  [[nodiscard]] std::string
  debug_string() const override;
 
  [[nodiscard]] std::unique_ptr<Operation>
  copy() const override;
};
 
class TestStructuralNode final : public StructuralNode
{
public:
  ~TestStructuralNode() noexcept override;
 
private:
  TestStructuralNode(Region * parent, size_t numSubregions)
      : StructuralNode(parent, numSubregions)
  {}
 
public:
  struct InputVar
  {
    StructuralInput * input{};
    std::vector<RegionArgument *> argument{};
  };
 
  struct OutputVar
  {
    StructuralOutput * output{};
    std::vector<RegionResult *> result{};
  };
 
  StructuralInput &
  addInputOnly(Output & origin);
 
  InputVar
  addInputWithArguments(Output & origin);
 
  void
  removeInputAndArguments(size_t index);
 
  InputVar
  addArguments(const std::shared_ptr<const Type> & type);
 
  StructuralOutput &
  addOutputOnly(std::shared_ptr<const Type> type);
 
  OutputVar
  addOutputWithResults(const std::vector<Output *> & origins);
 
  void
  removeOutputAndResults(size_t index);
 
  OutputVar
  addResults(const std::vector<Output *> & origins);
 
  InputVar
  mapInput(const Input & input) const;
 
  OutputVar
  mapOutput(const Output & output) const;
 
  [[nodiscard]] const TestStructuralOperation &
  GetOperation() const noexcept override;
 
  static TestStructuralNode *
  create(Region * parent, const size_t numSubregions)
  {
    return new TestStructuralNode(parent, numSubregions);
  }
 
  TestStructuralNode *
  copy(Region * region, SubstitutionMap & smap) const override;
};
 
}
 
#endif