From 9c4f2575962867357d9f258f611074fdebb25474 Mon Sep 17 00:00:00 2001
From: Bartosz Taudul <wolf@nereid.pl>
Date: Mon, 15 Aug 2022 21:27:05 +0200
Subject: [PATCH] Normalize frame names in wait stacks tree.

---
 server/TracyView_FrameTree.cpp | 30 +++++++++++++++++++++++++++---
 1 file changed, 27 insertions(+), 3 deletions(-)

diff --git a/server/TracyView_FrameTree.cpp b/server/TracyView_FrameTree.cpp
index 5fcb5b33..f9eb7d04 100644
--- a/server/TracyView_FrameTree.cpp
+++ b/server/TracyView_FrameTree.cpp
@@ -590,10 +590,16 @@ void View::DrawFrameTreeLevel( const unordered_flat_map<uint64_t, CallstackFrame
                 {
                     TextColoredUnformatted( 0xFF8888FF, frameName );
                 }
-                else
+                else if( m_shortenName == ShortenName::Never )
                 {
                     ImGui::TextUnformatted( frameName );
                 }
+                else
+                {
+                    const auto normalized = ShortenZoneName( ShortenName::OnlyNormalize, frameName );
+                    expand = ImGui::TreeNodeEx( normalized, ImGuiTreeNodeFlags_DefaultOpen );
+                    TooltipNormalizedName( frameName, normalized );
+                }
                 ImGui::Unindent( ImGui::GetTreeNodeToLabelSpacing() );
             }
             else
@@ -603,11 +609,29 @@ void View::DrawFrameTreeLevel( const unordered_flat_map<uint64_t, CallstackFrame
                 else if( isKernel ) ImGui::PushStyleColor( ImGuiCol_Text, 0xFF8888FF );
                 if( tree.size() == 1 )
                 {
-                    expand = ImGui::TreeNodeEx( frameName, ImGuiTreeNodeFlags_DefaultOpen );
+                    if( m_shortenName == ShortenName::Never )
+                    {
+                        expand = ImGui::TreeNodeEx( frameName, ImGuiTreeNodeFlags_DefaultOpen );
+                    }
+                    else
+                    {
+                        const auto normalized = ShortenZoneName( ShortenName::OnlyNormalize, frameName );
+                        expand = ImGui::TreeNodeEx( normalized, ImGuiTreeNodeFlags_DefaultOpen );
+                        TooltipNormalizedName( frameName, normalized );
+                    }
                 }
                 else
                 {
-                    expand = ImGui::TreeNode( frameName );
+                    if( m_shortenName == ShortenName::Never )
+                    {
+                        expand = ImGui::TreeNode( frameName );
+                    }
+                    else
+                    {
+                        const auto normalized = ShortenZoneName( ShortenName::OnlyNormalize, frameName );
+                        expand = ImGui::TreeNode( normalized );
+                        TooltipNormalizedName( frameName, normalized );
+                    }
                 }
                 if( isKernel || frameName[0] == '[' ) ImGui::PopStyleColor();
                 ImGui::PopID();