Code/EditorPlugins/Assets/EditorPluginAssets/StateMachineAsset/StateMachineGraphQt.cpp - dev - redgpu/ezEngine

Code/EditorPlugins/Assets/EditorPluginAssets/StateMachineAsset/StateMachineGraphQt.cpp
240 строк6 KB
Jan Krassnigg
Visual graph infrastructure cleanup (#1821)
19 фев 2026, 09:02
19 фев 2026, 09:027f7f12f
#include <EditorPluginAssets/EditorPluginAssetsPCH.h>

#include <EditorPluginAssets/StateMachineAsset/StateMachineAsset.h>
#include <EditorPluginAssets/StateMachineAsset/StateMachineGraph.h>
#include <EditorPluginAssets/StateMachineAsset/StateMachineGraphQt.moc.h>
#include <Foundation/Math/ColorScheme.h>

ezQtStateMachinePin::ezQtStateMachinePin() = default;

void ezQtStateMachinePin::SetPin(const ezVisualGraphPin& pin)
{
  ezQtVisualGraphPin::SetPin(pin);

  constexpr int padding = 3;

  m_pLabel->setPlainText("     +     ");
  m_pLabel->setPos(0, 0);

  auto bounds = m_pLabel->boundingRect();
  m_PinCenter = bounds.center();

  QPainterPath p;
  p.addRect(bounds);
  setPath(p);

  if (pin.GetType() == ezVisualGraphPin::Type::Input)
  {
    m_pLabel->setPlainText("");
  }
  else
  {
    m_pLabel->setToolTip("Add Transition");
  }
}

QRectF ezQtStateMachinePin::GetPinRect() const
{
  auto rect = path().boundingRect();
  rect.translate(pos());
  return rect;
}

//////////////////////////////////////////////////////////////////////////

ezQtStateMachineConnection::ezQtStateMachineConnection()
{
  setFlag(QGraphicsItem::ItemIsSelectable);
}

//////////////////////////////////////////////////////////////////////////

ezQtStateMachineNode::ezQtStateMachineNode() = default;

void ezQtStateMachineNode::InitNode(const ezVisualGraphObjectManager* pManager, const ezDocumentObject* pObject)
{
  ezQtVisualGraphNode::InitNode(pManager, pObject);

  UpdateHeaderColor();
}

void ezQtStateMachineNode::UpdateGeometry()
{
  prepareGeometryChange();

  auto labelRect = m_pTitleLabel->boundingRect();

  constexpr int padding = 5;
  const int headerWidth = labelRect.width();
  const int headerHeight = labelRect.height() + padding * 2;

  int h = headerHeight;
  int w = headerWidth;

  for (ezQtVisualGraphPin* pQtPin : GetInputPins())
  {
    auto rectPin = pQtPin->GetPinRect();
    w = ezMath::Max(w, (int)rectPin.width());

    pQtPin->setPos((w - rectPin.width()) / 2.0, h);
  }

  for (ezQtVisualGraphPin* pQtPin : GetOutputPins())
  {
    auto rectPin = pQtPin->GetPinRect();
    w = ezMath::Max(w, (int)rectPin.width());

    pQtPin->setPos((w - rectPin.width()) / 2.0, h);
    h += rectPin.height();
  }

  w += padding * 2;
  h += padding * 2;

  m_HeaderRect = QRectF(-padding, -padding, w, headerHeight);

  {
    QPainterPath p;
    p.addRoundedRect(-padding, -padding, w, h, padding, padding);
    setPath(p);
  }
}

void ezQtStateMachineNode::UpdateState()
{
  UpdateHeaderColor();

  if (IsAnyState())
  {
    m_pTitleLabel->setPlainText("Any State");
  }
  else
  {
    ezStringBuilder sName;

    auto& typeAccessor = GetObject()->GetTypeAccessor();

    ezVariant name = typeAccessor.GetValue("Name");
    if (name.IsA<ezString>() && name.Get<ezString>().IsEmpty() == false)
    {
      sName = name.Get<ezString>();
    }
    else
    {
      sName = typeAccessor.GetType()->GetTypeName();
    }

    if (IsInitialState())
    {
      sName.Append(" [Initial State]");
    }

    m_pTitleLabel->setPlainText(sName.GetData());
  }
}

void ezQtStateMachineNode::ExtendContextMenu(QMenu& ref_menu)
{
  if (IsAnyState())
    return;

  QAction* pAction = new QAction("Set as Initial State", &ref_menu);
  pAction->setEnabled(IsInitialState() == false);
  pAction->connect(pAction, &QAction::triggered,
    [this]()
    {
      auto pScene = static_cast<ezQtStateMachineAssetScene*>(scene());
      pScene->SetInitialState(this);
    });

  ref_menu.addAction(pAction);
}

bool ezQtStateMachineNode::IsInitialState() const
{
  auto pManager = static_cast<const ezStateMachineNodeManager*>(GetObject()->GetDocumentObjectManager());
  return pManager->IsInitialState(GetObject());
}

bool ezQtStateMachineNode::IsAnyState() const
{
  auto pManager = static_cast<const ezStateMachineNodeManager*>(GetObject()->GetDocumentObjectManager());
  return pManager->IsAnyState(GetObject());
}

void ezQtStateMachineNode::UpdateHeaderColor()
{
  ezColorScheme::Enum schemeColor = ezColorScheme::Gray;

  if (IsAnyState())
  {
    schemeColor = ezColorScheme::Violet;
  }
  else if (IsInitialState())
  {
    schemeColor = ezColorScheme::Teal;
  }

  m_HeaderColor = ezToQtColor(ezColorScheme::DarkUI(schemeColor));

  update();
}

//////////////////////////////////////////////////////////////////////////

ezQtStateMachineAssetScene::ezQtStateMachineAssetScene(QObject* pParent /*= nullptr*/)
  : ezQtVisualGraphScene(pParent)
{
  SetConnectionStyle(ezQtVisualGraphScene::ConnectionStyle::StraightLine);
  SetConnectionDecorationFlags(ezQtVisualGraphScene::ConnectionDecorationFlags::DirectionArrows);
}

ezQtStateMachineAssetScene::~ezQtStateMachineAssetScene() = default;

void ezQtStateMachineAssetScene::SetInitialState(ezQtStateMachineNode* pNode)
{
  ezCommandHistory* history = GetDocumentNodeManager()->GetDocument()->GetCommandHistory();
  history->StartTransaction("Set Initial State");

  ezStateMachine_SetInitialStateCommand cmd;
  cmd.m_NewInitialStateObject = pNode->GetObject()->GetGuid();

  ezStatus res = history->AddCommand(cmd);

  if (res.Failed())
    history->CancelTransaction();
  else
    history->FinishTransaction();
}

ezStatus ezQtStateMachineAssetScene::RemoveNode(ezQtVisualGraphNode* pNode)
{
  auto pManager = static_cast<const ezStateMachineNodeManager*>(GetDocumentNodeManager());
  const bool bWasInitialState = pManager->IsInitialState(pNode->GetObject());

  auto res = ezQtVisualGraphScene::RemoveNode(pNode);
  if (res.Succeeded() && bWasInitialState)
  {
    // Find another node
    ezUuid newInitialStateObject;
    for (auto it : m_Nodes)
    {
      if (it.Value() != pNode && pManager->IsAnyState(it.Key()) == false)
      {
        newInitialStateObject = it.Key()->GetGuid();
      }
    }

    if (newInitialStateObject.IsValid())
    {
      ezCommandHistory* history = pManager->GetDocument()->GetCommandHistory();

      ezStateMachine_SetInitialStateCommand cmd;
      cmd.m_NewInitialStateObject = newInitialStateObject;

      res = history->AddCommand(cmd);
    }
  }

  return res;
}