graph.cpp

Go to the documentation of this file.

/*
 * Copyright 2010 2011 2012 2013 2014 2015 Helge Bahmann <hcb@chaoticmind.net>
 * Copyright 2013 2014 2016 Nico Reißmann <nico.reissmann@gmail.com>
 * See COPYING for terms of redistribution.
 */
 
#include <jlm/rvsdg/graph.hpp>
#include <jlm/rvsdg/substitution.hpp>
#include <jlm/util/strfmt.hpp>
 
#include <algorithm>
 
namespace jlm::rvsdg
{
 
GraphImport::GraphImport(Graph & graph, std::shared_ptr<const rvsdg::Type> type, std::string name)
    : RegionArgument(&graph.GetRootRegion(), nullptr, std::move(type)),
      Name_(std::move(name))
{}
 
std::string
GraphImport::debug_string() const
{
  return util::strfmt("import[", Name_, "]");
}
 
GraphImport &
GraphImport::Copy(Region & region, StructuralInput *) const
{
  // FIXME: A import should never be created on a region, but only on a graph. This interface is
  // broken as it suggest to the user that it should be created on \p region.
  return Create(*region.graph(), Type(), Name());
}
 
GraphImport &
GraphImport::Create(Graph & graph, std::shared_ptr<const rvsdg::Type> type, std::string name)
{
  std::unique_ptr<GraphImport> graphImport(
      new GraphImport(graph, std::move(type), std::move(name)));
  return static_cast<GraphImport &>(graph.GetRootRegion().addArgument(std::move(graphImport)));
}
 
GraphExport::GraphExport(rvsdg::Output & origin, std::string name)
    : RegionResult(origin.region(), &origin, nullptr, origin.Type()),
      Name_(std::move(name))
{}
 
std::string
GraphExport::debug_string() const
{
  return util::strfmt("export[", Name_, "]");
}
 
GraphExport &
GraphExport::Copy(Output & origin, StructuralOutput * output) const
{
  JLM_ASSERT(output == nullptr);
  return Create(origin, Name());
}
 
GraphExport &
GraphExport::Create(Output & origin, std::string name)
{
  if (!origin.region()->IsRootRegion())
    throw std::logic_error("Cannot create a GraphExport on a non-root region.");
 
  std::unique_ptr<GraphExport> graphExport(new GraphExport(origin, std::move(name)));
  return static_cast<GraphExport &>(graphExport->region()->addResult(std::move(graphExport)));
}
 
Graph::~Graph() noexcept = default;
 
Graph::Graph()
    : nextRegionId_(0),
      RootRegion_(new Region(this))
{}
 
std::unique_ptr<Graph>
Graph::Copy() const
{
  SubstitutionMap substitutionMap;
  auto graph = std::make_unique<Graph>();
 
  // copy graph imports
  for (auto import : GetRootRegion().Arguments())
  {
    auto & importCopy = import->Copy(graph->GetRootRegion(), nullptr);
    substitutionMap.insert(import, &importCopy);
  }
 
  // copy root region
  GetRootRegion().copy(&graph->GetRootRegion(), substitutionMap);
 
  // copy graph exports
  for (auto xport : GetRootRegion().Results())
  {
    auto & newOrigin = substitutionMap.lookup(*xport->origin());
    xport->Copy(newOrigin, nullptr);
  }
 
  return graph;
}
 
std::vector<Node *>
Graph::ExtractTailNodes(const Graph & rvsdg)
{
  auto IsOnlyExported = [](const rvsdg::Output & output)
  {
    auto IsRootRegionExport = [](const rvsdg::Input * input)
    {
      if (!input->region()->IsRootRegion())
      {
        return false;
      }
 
      if (TryGetOwnerNode<Node>(*input))
      {
        return false;
      }
 
      return true;
    };
 
    for (auto & user : output.Users())
    {
      if (!IsRootRegionExport(&user))
      {
        return false;
      }
    }
 
    return true;
  };
 
  auto & rootRegion = rvsdg.GetRootRegion();
 
  std::vector<Node *> nodes;
  for (auto & bottomNode : rootRegion.BottomNodes())
  {
    nodes.push_back(&bottomNode);
  }
 
  for (size_t n = 0; n < rootRegion.nresults(); n++)
  {
    auto output = rootRegion.result(n)->origin();
    if (IsOnlyExported(*output))
    {
      nodes.push_back(rvsdg::TryGetOwnerNode<Node>(*output));
    }
  }
 
  return nodes;
}
 
}