Code/Engine/Foundation/Utilities/Implementation/DGMLWriter.cpp - dev - redgpu/ezEngine

Code/Engine/Foundation/Utilities/Implementation/DGMLWriter.cpp
271 строка8 KB
Jan Krassnigg
Command history now resets modified state when undoing to a saved state (#1824)
22 фев 2026, 20:03
22 фев 2026, 20:03288fa6b
#include <Foundation/FoundationPCH.h>

#include <Foundation/IO/FileSystem/FileWriter.h>
#include <Foundation/Strings/FormatString.h>
#include <Foundation/Utilities/DGMLWriter.h>

ezDGMLGraph::ezDGMLGraph(ezDGMLGraph::Direction graphDirection /*= LeftToRight*/, ezDGMLGraph::Layout graphLayout /*= Tree*/)
  : m_Direction(graphDirection)
  , m_Layout(graphLayout)
{
}

ezDGMLGraph::NodeId ezDGMLGraph::AddNode(ezStringView sTitle, const NodeDesc* pDesc)
{
  return AddGroup(sTitle, GroupType::None, pDesc);
}

ezDGMLGraph::NodeId ezDGMLGraph::AddGroup(ezStringView sTitle, GroupType type, const NodeDesc* pDesc /*= nullptr*/)
{
  ezDGMLGraph::Node& Node = m_Nodes.ExpandAndGetRef();

  Node.m_Title = sTitle;
  Node.m_GroupType = type;

  if (pDesc)
  {
    Node.m_Desc = *pDesc;
  }

  return m_Nodes.GetCount() - 1;
}

void ezDGMLGraph::AddNodeToGroup(NodeId node, NodeId group)
{
  EZ_ASSERT_DEBUG(m_Nodes[group].m_GroupType != GroupType::None, "The given group node has not been created as a group node");

  m_Nodes[node].m_ParentGroup = group;
}

ezDGMLGraph::CategoryId ezDGMLGraph::AddConnectionCategory(ezStringView sName, const ezColor& color)
{
  auto& cat = m_ConnectionCategories.ExpandAndGetRef();
  cat.m_sName = sName;
  cat.m_Color = color;
  return m_ConnectionCategories.GetCount() - 1;
}

ezDGMLGraph::ConnectionId ezDGMLGraph::AddConnection(ezDGMLGraph::NodeId source, ezDGMLGraph::NodeId target, ezStringView sLabel, CategoryId category)
{
  ezDGMLGraph::Connection& connection = m_Connections.ExpandAndGetRef();

  connection.m_Source = source;
  connection.m_Target = target;
  connection.m_sLabel = sLabel;
  connection.m_uiCategory = category;

  return m_Connections.GetCount() - 1;
}

ezDGMLGraph::PropertyId ezDGMLGraph::AddPropertyType(ezStringView sName)
{
  auto& prop = m_PropertyTypes.ExpandAndGetRef();
  prop.m_Name = sName;
  return m_PropertyTypes.GetCount() - 1;
}

void ezDGMLGraph::AddNodeProperty(NodeId node, PropertyId property, const ezFormatString& fmt)
{
  ezStringBuilder tmp;

  auto& prop = m_Nodes[node].m_Properties.ExpandAndGetRef();
  prop.m_PropertyId = property;
  prop.m_sValue = fmt.GetText(tmp);
}

ezResult ezDGMLGraphWriter::WriteGraphToFile(ezStringView sFileName, const ezDGMLGraph& graph)
{
  ezStringBuilder sGraph;

  // Write to memory object and then to file
  if (WriteGraphToString(sGraph, graph).Succeeded())
  {
    ezStringBuilder sTemp;

    ezFileWriter fileWriter;
    if (!fileWriter.Open(sFileName.GetData(sTemp)).Succeeded())
      return EZ_FAILURE;

    fileWriter.WriteBytes(sGraph.GetData(), sGraph.GetElementCount()).IgnoreResult();

    fileWriter.Close();

    return EZ_SUCCESS;
  }

  return EZ_FAILURE;
}

ezResult ezDGMLGraphWriter::WriteGraphToString(ezStringBuilder& ref_sStringBuilder, const ezDGMLGraph& graph)
{
  const char* szDirection = nullptr;
  const char* szLayout = nullptr;

  switch (graph.m_Direction)
  {
    case ezDGMLGraph::Direction::TopToBottom:
      szDirection = "TopToBottom";
      break;
    case ezDGMLGraph::Direction::BottomToTop:
      szDirection = "BottomToTop";
      break;
    case ezDGMLGraph::Direction::LeftToRight:
      szDirection = "LeftToRight";
      break;
    case ezDGMLGraph::Direction::RightToLeft:
      szDirection = "RightToLeft";
      break;
  }

  switch (graph.m_Layout)
  {
    case ezDGMLGraph::Layout::Free:
      szLayout = "None";
      break;
    case ezDGMLGraph::Layout::Tree:
      szLayout = "Sugiyama";
      break;
    case ezDGMLGraph::Layout::DependencyMatrix:
      szLayout = "DependencyMatrix";
      break;
  }

  ref_sStringBuilder.AppendFormat("<DirectedGraph xmlns=\"http://schemas.microsoft.com/vs/2009/dgml\" GraphDirection=\"{0}\" Layout=\"{1}\">\n", szDirection, szLayout);

  // Write out all connection categories
  if (!graph.m_ConnectionCategories.IsEmpty())
  {
    ref_sStringBuilder.Append("\t<Categories>\n");
    for (ezUInt32 i = 0; i < graph.m_ConnectionCategories.GetCount(); ++i)
    {
      const auto& cat = graph.m_ConnectionCategories[i];
      ezColorGammaUB rgba(cat.m_Color);
      ezStringBuilder sStroke;
      sStroke.SetFormat("#FF{0}{1}{2}", ezArgU(rgba.r, 2, true, 16, true), ezArgU(rgba.g, 2, true, 16, true), ezArgU(rgba.b, 2, true, 16, true));
      ref_sStringBuilder.AppendFormat("\t\t<Category Id=\"C_{0}\" Label=\"{1}\" Stroke=\"{2}\" />\n", i, cat.m_sName, sStroke);
    }
    ref_sStringBuilder.Append("\t</Categories>\n");
  }

  // Write out all the properties
  if (!graph.m_PropertyTypes.IsEmpty())
  {
    ref_sStringBuilder.Append("\t<Properties>\n");

    for (ezUInt32 i = 0; i < graph.m_PropertyTypes.GetCount(); ++i)
    {
      const auto& prop = graph.m_PropertyTypes[i];

      ref_sStringBuilder.AppendFormat("\t\t<Property Id=\"P_{0}\" Label=\"{1}\" DataType=\"String\"/>\n", i, prop.m_Name);
    }

    ref_sStringBuilder.Append("\t</Properties>\n");
  }

  // Write out all the nodes
  if (!graph.m_Nodes.IsEmpty())
  {
    ezStringBuilder ColorValue;
    ezStringBuilder PropertiesString;
    ezStringBuilder SanitizedName;
    const char* szGroupString;

    ref_sStringBuilder.Append("\t<Nodes>\n");
    for (ezUInt32 i = 0; i < graph.m_Nodes.GetCount(); ++i)
    {
      const ezDGMLGraph::Node& node = graph.m_Nodes[i];

      SanitizedName = node.m_Title;
      SanitizedName.ReplaceAll("&", "&#038;");
      SanitizedName.ReplaceAll("<", "&lt;");
      SanitizedName.ReplaceAll(">", "&gt;");
      SanitizedName.ReplaceAll("\"", "&quot;");
      SanitizedName.ReplaceAll("'", "&apos;");
      SanitizedName.ReplaceAll("\n", "&#xA;");

      ColorValue = "#FF";
      ezColorGammaUB RGBA(node.m_Desc.m_Color);
      ColorValue.AppendFormat("{0}{1}{2}", ezArgU(RGBA.r, 2, true, 16, true), ezArgU(RGBA.g, 2, true, 16, true), ezArgU(RGBA.b, 2, true, 16, true));

      ezStringBuilder StyleString;
      switch (node.m_Desc.m_Shape)
      {
        case ezDGMLGraph::NodeShape::None:
          StyleString = "Shape=\"None\"";
          break;
        case ezDGMLGraph::NodeShape::Rectangle:
          StyleString = "NodeRadius=\"0\"";
          break;
        case ezDGMLGraph::NodeShape::RoundedRectangle:
          StyleString = "NodeRadius=\"4\"";
          break;
        case ezDGMLGraph::NodeShape::Button:
          StyleString = "";
          break;
      }

      switch (node.m_GroupType)
      {

        case ezDGMLGraph::GroupType::Expanded:
          szGroupString = " Group=\"Expanded\"";
          break;

        case ezDGMLGraph::GroupType::Collapsed:
          szGroupString = " Group=\"Collapsed\"";
          break;

        case ezDGMLGraph::GroupType::None:
        default:
          szGroupString = nullptr;
          break;
      }

      PropertiesString.Clear();
      for (const auto& prop : node.m_Properties)
      {
        PropertiesString.AppendFormat(" {0}=\"{1}\"", graph.m_PropertyTypes[prop.m_PropertyId].m_Name, prop.m_sValue);
      }

      ref_sStringBuilder.AppendFormat("\t\t<Node Id=\"N_{0}\" Label=\"{1}\" Background=\"{2}\" {3}{4}{5}/>\n", i, SanitizedName, ColorValue, StyleString, szGroupString, PropertiesString);
    }
    ref_sStringBuilder.Append("\t</Nodes>\n");
  }

  // Write out the links
  if (!graph.m_Connections.IsEmpty())
  {
    ref_sStringBuilder.Append("\t<Links>\n");
    {
      for (ezUInt32 i = 0; i < graph.m_Connections.GetCount(); ++i)
      {
        const auto& conn = graph.m_Connections[i];
        if (conn.m_uiCategory != ezInvalidIndex)
        {
          ref_sStringBuilder.AppendFormat("\t\t<Link Source=\"N_{0}\" Target=\"N_{1}\" Label=\"{2}\" Category=\"C_{3}\" />\n",
            conn.m_Source, conn.m_Target, conn.m_sLabel, conn.m_uiCategory);
        }
        else
        {
          ref_sStringBuilder.AppendFormat("\t\t<Link Source=\"N_{0}\" Target=\"N_{1}\" Label=\"{2}\" />\n",
            conn.m_Source, conn.m_Target, conn.m_sLabel);
        }
      }

      for (ezUInt32 i = 0; i < graph.m_Nodes.GetCount(); ++i)
      {
        const ezDGMLGraph::Node& node = graph.m_Nodes[i];

        if (node.m_ParentGroup != 0xFFFFFFFF)
        {
          ref_sStringBuilder.AppendFormat("\t\t<Link Category=\"Contains\" Source=\"N_{0}\" Target=\"N_{1}\" />\n", node.m_ParentGroup, i);
        }
      }
    }
    ref_sStringBuilder.Append("\t</Links>\n");
  }

  ref_sStringBuilder.Append("</DirectedGraph>\n");

  return EZ_SUCCESS;
}