Code/EditorPlugins/Assets/EditorPluginAssets/SkeletonAsset/SkeletonActions.cpp - dev - redgpu/ezEngine

Code/EditorPlugins/Assets/EditorPluginAssets/SkeletonAsset/SkeletonActions.cpp
180 строк6 KB
Jan Krassnigg
New color design for the editor user interface (#1024)
08 авг 2023, 08:42
08 авг 2023, 08:42b7bf6f5
#include <EditorPluginAssets/EditorPluginAssetsPCH.h>

#include <EditorPluginAssets/SkeletonAsset/SkeletonActions.h>
#include <EditorPluginAssets/SkeletonAsset/SkeletonAsset.h>
#include <GuiFoundation/Action/ActionManager.h>
#include <GuiFoundation/Action/ActionMapManager.h>

EZ_BEGIN_DYNAMIC_REFLECTED_TYPE(ezSkeletonAction, 1, ezRTTINoAllocator)
EZ_END_DYNAMIC_REFLECTED_TYPE;

ezActionDescriptorHandle ezSkeletonActions::s_hCategory;
ezActionDescriptorHandle ezSkeletonActions::s_hRenderBones;
ezActionDescriptorHandle ezSkeletonActions::s_hRenderColliders;
ezActionDescriptorHandle ezSkeletonActions::s_hRenderJoints;
ezActionDescriptorHandle ezSkeletonActions::s_hRenderSwingLimits;
ezActionDescriptorHandle ezSkeletonActions::s_hRenderTwistLimits;
ezActionDescriptorHandle ezSkeletonActions::s_hRenderPreviewMesh;

void ezSkeletonActions::RegisterActions()
{
  s_hCategory = EZ_REGISTER_CATEGORY("SkeletonCategory");
  s_hRenderBones = EZ_REGISTER_ACTION_1("Skeleton.RenderBones", ezActionScope::Document, "Skeletons", "", ezSkeletonAction, ezSkeletonAction::ActionType::RenderBones);
  s_hRenderColliders = EZ_REGISTER_ACTION_1("Skeleton.RenderColliders", ezActionScope::Document, "Skeletons", "", ezSkeletonAction, ezSkeletonAction::ActionType::RenderColliders);
  s_hRenderJoints = EZ_REGISTER_ACTION_1("Skeleton.RenderJoints", ezActionScope::Document, "Skeletons", "", ezSkeletonAction, ezSkeletonAction::ActionType::RenderJoints);
  s_hRenderSwingLimits = EZ_REGISTER_ACTION_1("Skeleton.RenderSwingLimits", ezActionScope::Document, "Skeletons", "", ezSkeletonAction, ezSkeletonAction::ActionType::RenderSwingLimits);
  s_hRenderTwistLimits = EZ_REGISTER_ACTION_1("Skeleton.RenderTwistLimits", ezActionScope::Document, "Skeletons", "", ezSkeletonAction, ezSkeletonAction::ActionType::RenderTwistLimits);
  s_hRenderPreviewMesh = EZ_REGISTER_ACTION_1("Skeleton.RenderPreviewMesh", ezActionScope::Document, "Skeletons", "", ezSkeletonAction, ezSkeletonAction::ActionType::RenderPreviewMesh);
}

void ezSkeletonActions::UnregisterActions()
{
  ezActionManager::UnregisterAction(s_hCategory);
  ezActionManager::UnregisterAction(s_hRenderBones);
  ezActionManager::UnregisterAction(s_hRenderColliders);
  ezActionManager::UnregisterAction(s_hRenderJoints);
  ezActionManager::UnregisterAction(s_hRenderSwingLimits);
  ezActionManager::UnregisterAction(s_hRenderTwistLimits);
  ezActionManager::UnregisterAction(s_hRenderPreviewMesh);
}

void ezSkeletonActions::MapActions(ezStringView sMapping)
{
  ezActionMap* pMap = ezActionMapManager::GetActionMap(sMapping);
  EZ_ASSERT_DEV(pMap != nullptr, "The given mapping ('{0}') does not exist, mapping the actions failed!", sMapping);

  pMap->MapAction(s_hCategory, "", 11.0f);

  const char* szSubPath = "SkeletonCategory";

  pMap->MapAction(s_hRenderBones, szSubPath, 1.0f);
  pMap->MapAction(s_hRenderColliders, szSubPath, 2.0f);
  // pMap->MapAction(s_hRenderJoints, szSubPath, 3.0f);
  pMap->MapAction(s_hRenderSwingLimits, szSubPath, 4.0f);
  pMap->MapAction(s_hRenderTwistLimits, szSubPath, 5.0f);
  pMap->MapAction(s_hRenderPreviewMesh, szSubPath, 6.0f);
}

ezSkeletonAction::ezSkeletonAction(const ezActionContext& context, const char* szName, ezSkeletonAction::ActionType type)
  : ezButtonAction(context, szName, false, "")
{
  m_Type = type;

  m_pSkeletonpDocument = const_cast<ezSkeletonAssetDocument*>(static_cast<const ezSkeletonAssetDocument*>(context.m_pDocument));
  m_pSkeletonpDocument->Events().AddEventHandler(ezMakeDelegate(&ezSkeletonAction::AssetEventHandler, this));

  switch (m_Type)
  {
    case ActionType::RenderBones:
      SetIconPath(":/EditorPluginAssets/SkeletonBones.svg");
      break;

    case ActionType::RenderColliders:
      SetIconPath(":/EditorPluginAssets/SkeletonColliders.svg");
      break;

    case ActionType::RenderJoints:
      SetIconPath(":/EditorPluginAssets/SkeletonJoints.svg");
      break;

    case ActionType::RenderSwingLimits:
      SetIconPath(":/EditorPluginAssets/JointSwingLimits.svg");
      break;

    case ActionType::RenderTwistLimits:
      SetIconPath(":/EditorPluginAssets/JointTwistLimits.svg");
      break;

    case ActionType::RenderPreviewMesh:
      SetIconPath(":/EditorPluginAssets/PreviewMesh.svg");
      break;
  }

  UpdateState();
}

ezSkeletonAction::~ezSkeletonAction()
{
  m_pSkeletonpDocument->Events().RemoveEventHandler(ezMakeDelegate(&ezSkeletonAction::AssetEventHandler, this));
}

void ezSkeletonAction::Execute(const ezVariant& value)
{
  switch (m_Type)
  {
    case ActionType::RenderBones:
      m_pSkeletonpDocument->SetRenderBones(!m_pSkeletonpDocument->GetRenderBones());
      return;

    case ActionType::RenderColliders:
      m_pSkeletonpDocument->SetRenderColliders(!m_pSkeletonpDocument->GetRenderColliders());
      return;

    case ActionType::RenderJoints:
      m_pSkeletonpDocument->SetRenderJoints(!m_pSkeletonpDocument->GetRenderJoints());
      return;

    case ActionType::RenderSwingLimits:
      m_pSkeletonpDocument->SetRenderSwingLimits(!m_pSkeletonpDocument->GetRenderSwingLimits());
      return;

    case ActionType::RenderTwistLimits:
      m_pSkeletonpDocument->SetRenderTwistLimits(!m_pSkeletonpDocument->GetRenderTwistLimits());
      return;

    case ActionType::RenderPreviewMesh:
      m_pSkeletonpDocument->SetRenderPreviewMesh(!m_pSkeletonpDocument->GetRenderPreviewMesh());
      return;
  }
}

void ezSkeletonAction::AssetEventHandler(const ezSkeletonAssetEvent& e)
{
  switch (e.m_Type)
  {
    case ezSkeletonAssetEvent::RenderStateChanged:
      UpdateState();
      break;
    default:
      break;
  }
}

void ezSkeletonAction::UpdateState()
{
  if (m_Type == ActionType::RenderBones)
  {
    SetCheckable(true);
    SetChecked(m_pSkeletonpDocument->GetRenderBones());
  }

  if (m_Type == ActionType::RenderColliders)
  {
    SetCheckable(true);
    SetChecked(m_pSkeletonpDocument->GetRenderColliders());
  }

  if (m_Type == ActionType::RenderJoints)
  {
    SetCheckable(true);
    SetChecked(m_pSkeletonpDocument->GetRenderJoints());
  }

  if (m_Type == ActionType::RenderSwingLimits)
  {
    SetCheckable(true);
    SetChecked(m_pSkeletonpDocument->GetRenderSwingLimits());
  }

  if (m_Type == ActionType::RenderTwistLimits)
  {
    SetCheckable(true);
    SetChecked(m_pSkeletonpDocument->GetRenderTwistLimits());
  }

  if (m_Type == ActionType::RenderPreviewMesh)
  {
    SetCheckable(true);
    SetChecked(m_pSkeletonpDocument->GetRenderPreviewMesh());
  }
}