[ATOM-5389] MeshComponent stats

2021-06-08 19:21:40 -07:00
parent 1f48985a0e
commit 672b9fd956
7 changed files with 145 additions and 2 deletions
@@ -25,12 +25,16 @@ namespace AZ
        void EditorMeshComponent::Reflect(AZ::ReflectContext* context)
        {
            BaseClass::Reflect(context);
+            EditorMeshStats::Reflect(context);

            if (AZ::SerializeContext* serializeContext = azrtti_cast<AZ::SerializeContext*>(context))
            {
+                serializeContext->RegisterGenericType<EditorMeshStats>();
+
                serializeContext->Class<EditorMeshComponent, BaseClass>()
                    ->Version(2, ConvertToEditorRenderComponentAdapter<1>)
                    ->Field("addMaterialComponentFlag", &EditorMeshComponent::m_addMaterialComponentFlag)
+                    ->Field("meshStats", &EditorMeshComponent::m_stats)
                    ;

                // This shouldn't be registered here, but is required to make a vector from EditorMeshComponentTypeId. This can be removed when one of the following happens:
@@ -55,6 +59,8 @@ namespace AZ
                            ->Attribute(AZ::Edit::Attributes::ButtonText, "Add Material Component")
                            ->Attribute(AZ::Edit::Attributes::ChangeNotify, &EditorMeshComponent::AddEditorMaterialComponent)
                            ->Attribute(AZ::Edit::Attributes::Visibility, &EditorMeshComponent::GetEditorMaterialComponentVisibility)
+                        ->DataElement(AZ::Edit::UIHandlers::Default, &EditorMeshComponent::m_stats, "Mesh Stats", "Mesh statistics.")
+                            ->Attribute(AZ::Edit::Attributes::AutoExpand, false)
                        ;

                    editContext->Class<MeshComponentController>(
@@ -204,6 +210,25 @@ namespace AZ

        void EditorMeshComponent::OnModelReady(const Data::Asset<RPI::ModelAsset>& /*modelAsset*/, const Data::Instance<RPI::Model>& /*model*/)
        {
+            m_stats.m_meshStatsForLod.clear();
+            if (m_controller.GetConfiguration().IsAssetSet())
+            {
+                auto lods = m_controller.GetConfiguration().m_modelAsset->GetLodAssets();
+                for (auto& lod : lods)
+                {
+                    EditorMeshStatsForLod stats;
+                    auto meshes = lod->GetMeshes();
+                    stats.m_meshCount = lod->GetMeshes().size();
+                    for (auto& mesh : meshes)
+                    {
+                        stats.m_vertCount += mesh.GetVertexCount();
+                        stats.m_triCount += mesh.GetIndexCount() / 3;
+                    }
+                    m_stats.m_meshStatsForLod.push_back(stats);
+                }
+            }
+            m_stats.UpdateStringRepresentation();
+
            // Refresh the tree when the model loads to update UI based on the model.
            AzToolsFramework::ToolsApplicationEvents::Bus::Broadcast(
                &AzToolsFramework::ToolsApplicationEvents::InvalidatePropertyDisplay,
@@ -216,6 +241,11 @@ namespace AZ
            // places it in a bad state, which happens in OnConfigurationChanged base function.
            // This is a bug with AssetManager [LYN-2249]
            auto temp = m_controller.m_configuration.m_modelAsset;
+
+            m_stats.m_meshStatsForLod.clear();
+            m_stats.UpdateStringRepresentation();
+            SetDirty();
+
            return BaseClass::OnConfigurationChanged();
        }
    } // namespace Render
@@ -18,6 +18,7 @@
 #include <Atom/Feature/Utils/EditorRenderComponentAdapter.h>
 #include <AtomLyIntegration/CommonFeatures/Mesh/MeshComponentConstants.h>
 #include <Mesh/MeshComponent.h>
+#include <Mesh/EditorMeshStats.h>

 namespace AZ
 {
@@ -33,6 +34,7 @@ namespace AZ
            , private AzToolsFramework::EditorComponentSelectionRequestsBus::Handler
            , private AzFramework::EntityDebugDisplayEventBus::Handler
            , private MeshComponentNotificationBus::Handler
+            , private Data::AssetBus::Handler
        {
        public:
            using BaseClass = EditorRenderComponentAdapter<MeshComponentController, MeshComponent, MeshComponentConfig>;
@@ -71,6 +73,8 @@ namespace AZ

            // Flag used for button placement
            bool m_addMaterialComponentFlag = false;
+
+            EditorMeshStats m_stats;
        };
    } // namespace Render
 } // namespace AZ
@@ -0,0 +1,58 @@
+/*
+* All or portions of this file Copyright (c) Amazon.com, Inc. or its affiliates or
+* its licensors.
+*
+* For complete copyright and license terms please see the LICENSE at the root of this
+* distribution (the "License"). All use of this software is governed by the License,
+* or, if provided, by the license below or the license accompanying this file. Do not
+* remove or modify any license notices. This file is distributed on an "AS IS" BASIS,
+* WITHOUT WARRANTIES OR CONDITIONS OF ANY KIND, either express or implied.
+*
+*/
+
+#include <Mesh/EditorMeshStats.h>
+#include <AzCore/Serialization/SerializeContext.h>
+#include <AzCore/Serialization/EditContext.h>
+#include <AzCore/StringFunc/StringFunc.h>
+
+namespace AZ
+{
+    namespace Render
+    {
+        void EditorMeshStats::Reflect(ReflectContext* context)
+        {
+            if (AZ::SerializeContext* serializeContext = azrtti_cast<AZ::SerializeContext*>(context))
+            {
+                serializeContext->Class<EditorMeshStats>()
+                    ->Field("stringRepresentation", &EditorMeshStats::m_stringRepresentation)
+                    ;
+
+                if (AZ::EditContext* editContext = serializeContext->GetEditContext())
+                {
+                    editContext->Class<EditorMeshStats>(
+                        "EditorMeshStats", "")
+                        ->ClassElement(AZ::Edit::ClassElements::EditorData, "")
+                            ->Attribute(AZ::Edit::Attributes::AutoExpand, false)
+                        ->DataElement(AZ::Edit::UIHandlers::MultiLineEdit, &EditorMeshStats::m_stringRepresentation, "Mesh Stats", "")
+                            ->Attribute(AZ::Edit::Attributes::NameLabelOverride, "")
+                            ->Attribute(AZ::Edit::Attributes::ReadOnly, true)
+                        ;
+                }
+            }
+        }
+
+        void EditorMeshStats::UpdateStringRepresentation()
+        {
+            m_stringRepresentation = "";
+            int lodIndex = 0;
+            for (auto& meshStatsForLod : m_meshStatsForLod)
+            {
+                m_stringRepresentation += AZStd::string::format("LOD: %d:\n", lodIndex++);
+                m_stringRepresentation += AZStd::string::format("\tMesh Count: %d\n", meshStatsForLod.m_meshCount);
+                m_stringRepresentation += AZStd::string::format("\tVert Count: %d\n", meshStatsForLod.m_vertCount);
+                m_stringRepresentation += AZStd::string::format("\tTriangle Count: %d\n", meshStatsForLod.m_triCount);
+            }
+            AZ::StringFunc::TrimWhiteSpace(m_stringRepresentation, true, true);
+        };        
+    } // namespace Render
+} // namespace AZ
@@ -0,0 +1,49 @@
+/*
+* All or portions of this file Copyright (c) Amazon.com, Inc. or its affiliates or
+* its licensors.
+*
+* For complete copyright and license terms please see the LICENSE at the root of this
+* distribution (the "License"). All use of this software is governed by the License,
+* or, if provided, by the license below or the license accompanying this file. Do not
+* remove or modify any license notices. This file is distributed on an "AS IS" BASIS,
+* WITHOUT WARRANTIES OR CONDITIONS OF ANY KIND, either express or implied.
+*
+*/
+
+#pragma once
+
+#include <AzCore/Memory/SystemAllocator.h>
+#include <AzCore/RTTI/ReflectContext.h>
+#include <AzCore/std/containers/vector.h>
+#include <AzCore/std/string/string.h>
+
+namespace AZ
+{
+    namespace Render
+    {
+        struct EditorMeshStatsForLod final
+        {
+            AZ_RTTI(EditorMeshStatsForLod, "{626E3AEB-0F7A-4777-BAF1-2BBA8C1857ED}");
+            AZ_CLASS_ALLOCATOR(EditorMeshStatsForLod, SystemAllocator, 0);
+
+
+            int m_meshCount = 0;
+            int m_vertCount = 0;
+            int m_triCount = 0;
+        };
+
+        struct EditorMeshStats final
+        {
+            AZ_RTTI(EditorMeshStats, "{68D0D3EF-17BB-46EA-B98F-51355402CCD6}");
+            AZ_CLASS_ALLOCATOR(EditorMeshStats, SystemAllocator, 0);
+
+            static void Reflect(ReflectContext* context);
+
+            AZStd::vector<EditorMeshStatsForLod> m_meshStatsForLod;
+            
+            void UpdateStringRepresentation();
+
+            AZStd::string m_stringRepresentation = {};
+        };
+    } // namespace Render
+} // namespace AZ
@@ -49,7 +49,7 @@ namespace AZ
            }
        }

-        bool MeshComponentConfig::IsAssetSet()
+        bool MeshComponentConfig::IsAssetSet() const
        {
            return m_modelAsset.GetId().IsValid();
        }
@@ -44,7 +44,7 @@ namespace AZ
            static void Reflect(AZ::ReflectContext* context);

            // Editor helper functions
-            bool IsAssetSet();
+            bool IsAssetSet() const;
            AZStd::vector<AZStd::pair<RPI::Cullable::LodOverride, AZStd::string>> GetLodOverrideValues();

            Data::Asset<RPI::ModelAsset> m_modelAsset = { AZ::Data::AssetLoadBehavior::QueueLoad };
@@ -50,6 +50,8 @@ set(FILES
    Source/Material/MaterialThumbnail.h
    Source/Mesh/EditorMeshComponent.h
    Source/Mesh/EditorMeshComponent.cpp
+    Source/Mesh/EditorMeshStats.h
+    Source/Mesh/EditorMeshStats.cpp
    Source/Mesh/EditorMeshSystemComponent.cpp
    Source/Mesh/EditorMeshSystemComponent.h
    Source/Mesh/MeshThumbnail.h