Code/Editor/EditorFramework/Actions/TransformGizmoActions.h - dev - redgpu/ezEngine

Code/Editor/EditorFramework/Actions/TransformGizmoActions.h
120 строк3 KB
Jan Krassnigg
Added toolbar menus to select snapping options (#1504)
09 мар 2025, 18:11
09 мар 2025, 18:11087f036
#pragma once

#include <EditorFramework/EditorFrameworkDLL.h>
#include <GuiFoundation/Action/BaseActions.h>
#include <GuiFoundation/GuiFoundationDLL.h>

enum class ActiveGizmo;
class ezGameObjectDocument;
struct ezSnapProviderEvent;
struct ezGameObjectEvent;

///
class EZ_EDITORFRAMEWORK_DLL ezTransformGizmoActions
{
public:
  static void RegisterActions();
  static void UnregisterActions();

  static void MapMenuActions(ezStringView sMapping);
  static void MapToolbarActions(ezStringView sMapping);

  static ezActionDescriptorHandle s_hGizmoCategory;
  static ezActionDescriptorHandle s_hGizmoMenu;
  static ezActionDescriptorHandle s_hNoGizmo;
  static ezActionDescriptorHandle s_hTranslateGizmo;
  static ezActionDescriptorHandle s_hRotateGizmo;
  static ezActionDescriptorHandle s_hScaleGizmo;
  static ezActionDescriptorHandle s_hDragToPositionGizmo;
  static ezActionDescriptorHandle s_hWorldSpace;
  static ezActionDescriptorHandle s_hMoveParentOnly;
  static ezActionDescriptorHandle s_SnapSettings;
  static ezActionDescriptorHandle s_SnapTranslationMenu;
  static ezActionDescriptorHandle s_SnapRotationMenu;
  static ezActionDescriptorHandle s_SnapScaleMenu;
};

///
class EZ_EDITORFRAMEWORK_DLL ezGizmoAction : public ezButtonAction
{
  EZ_ADD_DYNAMIC_REFLECTION(ezGizmoAction, ezButtonAction);

public:
  ezGizmoAction(const ezActionContext& context, const char* szName, const ezRTTI* pGizmoType);
  ~ezGizmoAction();

  virtual void Execute(const ezVariant& value) override;

protected:
  void UpdateState();
  void GameObjectEventHandler(const ezGameObjectEvent& e);

  ezGameObjectDocument* m_pGameObjectDocument = nullptr;
  const ezRTTI* m_pGizmoType = nullptr;
};

///
class EZ_EDITORFRAMEWORK_DLL ezToggleWorldSpaceGizmo : public ezGizmoAction
{
public:
  ezToggleWorldSpaceGizmo(const ezActionContext& context, const char* szName, const ezRTTI* pGizmoType);
  virtual void Execute(const ezVariant& value) override;
};

///
class EZ_EDITORFRAMEWORK_DLL ezTransformGizmoAction : public ezButtonAction
{
  EZ_ADD_DYNAMIC_REFLECTION(ezTransformGizmoAction, ezButtonAction);

public:
  enum class ActionType
  {
    GizmoToggleWorldSpace,
    GizmoToggleMoveParentOnly,
    GizmoSnapSettings,
  };

  ezTransformGizmoAction(const ezActionContext& context, const char* szName, ActionType type);
  ~ezTransformGizmoAction();

  virtual void Execute(const ezVariant& value) override;
  void GameObjectEventHandler(const ezGameObjectEvent& e);

private:
  void UpdateState();

  ezGameObjectDocument* m_pGameObjectDocument;
  ActionType m_Type;
};

///
class EZ_EDITORFRAMEWORK_DLL ezTranslateGizmoAction : public ezButtonAction
{
  EZ_ADD_DYNAMIC_REFLECTION(ezTranslateGizmoAction, ezButtonAction);

public:
  static void RegisterActions();
  static void UnregisterActions();

  static void MapActions(ezStringView sMapping);

private:
  static ezActionDescriptorHandle s_hSnappingValueMenu;
  static ezActionDescriptorHandle s_hSnapPivotToGrid;
  static ezActionDescriptorHandle s_hSnapObjectsToGrid;

public:
  enum class ActionType
  {
    SnapSelectionPivotToGrid,
    SnapEachSelectedObjectToGrid,
  };

  ezTranslateGizmoAction(const ezActionContext& context, const char* szName, ActionType type);

  virtual void Execute(const ezVariant& value) override;

private:
  const ezGameObjectDocument* m_pSceneDocument;
  ActionType m_Type;
};