USD Viewer: scroll stage tree to selected prim

DiligentGraphics · Dec 2, 2023 · 4d71d10 · 4d71d10
1 parent d0211d1
commit 4d71d10
Show file tree

Hide file tree

Showing 2 changed files with 12 additions and 1 deletion.
diff --git a/Samples/USDViewer/src/USDViewer.cpp b/Samples/USDViewer/src/USDViewer.cpp
@@ -250,6 +250,14 @@ void USDViewer::PopulateSceneTree(const pxr::UsdPrim& Prim)
 
         ImGui::TreePop();
     }
+    else
+    {
+        if (m_ScrolllToSelectedTreeItem && (Flags & ImGuiTreeNodeFlags_Selected))
+        {
+            ImGui::SetScrollHereY();
+            m_ScrolllToSelectedTreeItem = false;
+        }
+    }
 }
 
 void USDViewer::UpdateUI()
@@ -579,6 +587,8 @@ void USDViewer::Update(double CurrTime, double ElapsedTime)
     if (SelectPrim && SelectedPrimId != nullptr)
     {
         SetSelectedPrim(*SelectedPrimId);
+        if (!SelectedPrimId->IsEmpty())
+            m_ScrolllToSelectedTreeItem = true;
     }
 
     m_PrevMouse = Mouse;

diff --git a/Samples/USDViewer/src/USDViewer.hpp b/Samples/USDViewer/src/USDViewer.hpp
@@ -141,7 +141,8 @@ class USDViewer final : public SampleBase
     };
     SelectionMode m_SelectMode = SelectionMode::OnClick;
     MouseState    m_PrevMouse;
-    bool          m_IsSelecting = false;
+    bool          m_IsSelecting               = false;
+    bool          m_ScrolllToSelectedTreeItem = false;
 };
 
 } // namespace Diligent