Code/Tools/Libs/ToolsFoundation/Command/VisualGraphCommands.h - dev - redgpu/ezEngine

Code/Tools/Libs/ToolsFoundation/Command/VisualGraphCommands.h
118 строк3 KB
Jan Krassnigg
Visual graph infrastructure cleanup (#1821)
19 фев 2026, 09:02
19 фев 2026, 09:027f7f12f
#pragma once

#include <ToolsFoundation/Command/Command.h>

class ezDocumentObject;
class ezCommandHistory;
class ezVisualGraphPin;

/// Command to remove a node from a visual graph.
///
/// Removes the specified node and all its connections. The operation is undoable.
class EZ_TOOLSFOUNDATION_DLL ezRemoveNodeCommand : public ezCommand
{
  EZ_ADD_DYNAMIC_REFLECTION(ezRemoveNodeCommand, ezCommand);

public:
  ezRemoveNodeCommand();

public: // Properties
  ezUuid m_Object;

private:
  virtual ezStatus DoInternal(bool bRedo) override;
  virtual ezStatus UndoInternal(bool bFireEvents) override;
  virtual void CleanupInternal(CommandState state) override;

private:
  ezDocumentObject* m_pObject = nullptr;
};

/// Command to move a node to a new position in the visual graph.
///
/// Changes the node's position in the graph editor. The operation is undoable.
class EZ_TOOLSFOUNDATION_DLL ezMoveNodeCommand : public ezCommand
{
  EZ_ADD_DYNAMIC_REFLECTION(ezMoveNodeCommand, ezCommand);

public:
  ezMoveNodeCommand();

public: // Properties
  ezUuid m_Object;
  ezVec2 m_NewPos = ezVec2::MakeZero();

private:
  virtual ezStatus DoInternal(bool bRedo) override;
  virtual ezStatus UndoInternal(bool bFireEvents) override;
  virtual void CleanupInternal(CommandState state) override {}

private:
  ezDocumentObject* m_pObject = nullptr;
  ezVec2 m_vOldPos = ezVec2::MakeZero();
};

/// Command to connect two pins in a visual graph.
///
/// Creates a connection between an output pin and an input pin. The operation is undoable.
class EZ_TOOLSFOUNDATION_DLL ezConnectNodePinsCommand : public ezCommand
{
  EZ_ADD_DYNAMIC_REFLECTION(ezConnectNodePinsCommand, ezCommand);

public:
  ezConnectNodePinsCommand();

public: // Properties
  ezUuid m_ConnectionObject;
  ezUuid m_ObjectSource;
  ezUuid m_ObjectTarget;
  ezString m_sSourcePin;
  ezString m_sTargetPin;

private:
  virtual ezStatus DoInternal(bool bRedo) override;
  virtual ezStatus UndoInternal(bool bFireEvents) override;
  virtual void CleanupInternal(CommandState state) override {}

private:
  ezDocumentObject* m_pConnectionObject = nullptr;
  ezDocumentObject* m_pObjectSource = nullptr;
  ezDocumentObject* m_pObjectTarget = nullptr;
};

/// Command to disconnect two pins in a visual graph.
///
/// Removes an existing connection between pins. The operation is undoable.
class EZ_TOOLSFOUNDATION_DLL ezDisconnectNodePinsCommand : public ezCommand
{
  EZ_ADD_DYNAMIC_REFLECTION(ezDisconnectNodePinsCommand, ezCommand);

public:
  ezDisconnectNodePinsCommand();

public: // Properties
  ezUuid m_ConnectionObject;

private:
  virtual ezStatus DoInternal(bool bRedo) override;
  virtual ezStatus UndoInternal(bool bFireEvents) override;
  virtual void CleanupInternal(CommandState state) override {}

private:
  ezDocumentObject* m_pConnectionObject = nullptr;
  const ezDocumentObject* m_pObjectSource = nullptr;
  const ezDocumentObject* m_pObjectTarget = nullptr;
  ezString m_sSourcePin;
  ezString m_sTargetPin;
};

/// Helper class for executing common visual graph command operations.
class EZ_TOOLSFOUNDATION_DLL ezNodeCommands
{
public:
  /// Creates a connection between two pins by adding a connection object and connecting it.
  static ezStatus AddAndConnectCommand(ezCommandHistory* pHistory, const ezRTTI* pConnectionType, const ezVisualGraphPin& sourcePin, const ezVisualGraphPin& targetPin);

  /// Disconnects pins and removes the connection object.
  static ezStatus DisconnectAndRemoveCommand(ezCommandHistory* pHistory, const ezUuid& connectionObject);
};