Display number of joints in skeleton asset

00mjk · Feb 25, 2021 · f93af19 · f93af19
1 parent 82c20d5
commit f93af19
Show file tree

Hide file tree

Showing 4 changed files with 28 additions and 3 deletions.
diff --git a/Code/EditorPlugins/Assets/EnginePluginAssets/SkeletonAsset/SkeletonContext.cpp b/Code/EditorPlugins/Assets/EnginePluginAssets/SkeletonAsset/SkeletonContext.cpp
@@ -64,8 +64,8 @@ void ezSkeletonContext::OnInitialize()
     ezSkeletonComponent::CreateComponent(m_pGameObject, pVisSkeleton);
     ezStringBuilder sSkeletonGuid;
     ezConversionUtils::ToString(GetDocumentGuid(), sSkeletonGuid);
-    ezSkeletonResourceHandle hSkeleton = ezResourceManager::LoadResource<ezSkeletonResource>(sSkeletonGuid);
-    pVisSkeleton->SetSkeleton(hSkeleton);
+    m_hSkeleton = ezResourceManager::LoadResource<ezSkeletonResource>(sSkeletonGuid);
+    pVisSkeleton->SetSkeleton(m_hSkeleton);
     pVisSkeleton->m_bVisualizeSkeleton = true;
   }
 

diff --git a/Code/EditorPlugins/Assets/EnginePluginAssets/SkeletonAsset/SkeletonContext.h b/Code/EditorPlugins/Assets/EnginePluginAssets/SkeletonAsset/SkeletonContext.h
@@ -14,6 +14,8 @@ class EZ_ENGINEPLUGINASSETS_DLL ezSkeletonContext : public ezEngineProcessDocume
 
   virtual void HandleMessage(const ezEditorEngineDocumentMsg* pMsg) override;
 
+  ezSkeletonResourceHandle GetSkeleton() const { return m_hSkeleton; }
+
 protected:
   virtual void OnInitialize() override;
 
@@ -25,5 +27,5 @@ class EZ_ENGINEPLUGINASSETS_DLL ezSkeletonContext : public ezEngineProcessDocume
   void QuerySelectionBBox(const ezEditorEngineDocumentMsg* pMsg);
 
   ezGameObject* m_pGameObject = nullptr;
-  ezSkeletonResourceHandle m_hMesh;
+  ezSkeletonResourceHandle m_hSkeleton;
 };
diff --git a/Code/EditorPlugins/Assets/EnginePluginAssets/SkeletonAsset/SkeletonView.cpp b/Code/EditorPlugins/Assets/EnginePluginAssets/SkeletonAsset/SkeletonView.cpp
@@ -10,6 +10,7 @@
 #include <EnginePluginAssets/SkeletonAsset/SkeletonView.h>
 #include <Foundation/Utilities/GraphicsUtils.h>
 #include <GameEngine/GameApplication/GameApplication.h>
+#include <RendererCore/Debug/DebugRenderer.h>
 #include <RendererCore/Pipeline/Implementation/RenderPipelineResourceLoader.h>
 #include <RendererCore/Pipeline/View.h>
 #include <RendererCore/RenderContext/RenderContext.h>
@@ -47,3 +48,24 @@ ezViewHandle ezSkeletonViewContext::CreateView()
   pView->SetCamera(&m_Camera);
   return pView->GetHandle();
 }
+
+void ezSkeletonViewContext::SetCamera(const ezViewRedrawMsgToEngine* pMsg)
+{
+  ezEngineProcessViewContext::SetCamera(pMsg);
+
+  const ezUInt32 viewHeight = pMsg->m_uiWindowHeight;
+
+  auto hSkeleton = m_pContext->GetSkeleton();
+  if (hSkeleton.IsValid())
+  {
+    ezResourceLock<ezSkeletonResource> pSkeleton(hSkeleton, ezResourceAcquireMode::AllowLoadingFallback);
+
+    ezUInt32 uiNumJoints = pSkeleton->GetDescriptor().m_Skeleton.GetJointCount();
+
+    ezStringBuilder sText;
+    sText.AppendFormat("Joints: {}\n", uiNumJoints);
+
+    ezDebugRenderer::Draw2DText(m_hView, sText, ezVec2I32(10, viewHeight - 10), ezColor::White, 16, ezDebugRenderer::HorizontalAlignment::Left,
+      ezDebugRenderer::VerticalAlignment::Bottom);
+  }
+}
diff --git a/Code/EditorPlugins/Assets/EnginePluginAssets/SkeletonAsset/SkeletonView.h b/Code/EditorPlugins/Assets/EnginePluginAssets/SkeletonAsset/SkeletonView.h
@@ -14,6 +14,7 @@ class ezSkeletonViewContext : public ezEngineProcessViewContext
 
 protected:
   virtual ezViewHandle CreateView() override;
+  virtual void SetCamera(const ezViewRedrawMsgToEngine* pMsg) override;
 
   ezSkeletonContext* m_pContext;
 };