Init Repo

2026-06-03 09:00:11 +08:00
commit 9be48d8b9e
155 changed files with 14827 additions and 0 deletions
@@ -0,0 +1,106 @@
+#pragma once
+
+#include "../core/PropertyManager.h"
+#include "../core/CommandHistory.h"
+#include "IconManager.h"
+#include <pxr/usd/usd/stage.h>
+#include <pxr/usd/sdf/path.h>
+#include <pxr/usd/sdf/layer.h>
+#include <pxr/usd/sdf/reference.h>
+#include <imgui.h>
+#include <string>
+#include <vector>
+#include <unordered_set>
+#include <functional>
+
+PXR_NAMESPACE_USING_DIRECTIVE
+
+namespace UsdLayerManager {
+
+class SceneHierarchyPanel {
+public:
+    SceneHierarchyPanel();
+    ~SceneHierarchyPanel();
+    
+    void SetPropertyManager(PropertyManager* manager);
+    void SetCommandHistory(CommandHistory* history) { m_commandHistory = history; }
+    void SetStage(UsdStageRefPtr stage);
+    void SetIconManager(IconManager* iconManager) { m_iconManager = iconManager; }
+    void Render();
+    
+    std::string GetSelectedPrimPath() const { return m_primarySelectedPath; }
+    UsdPrim GetSelectedPrim() const;
+
+    /// Set single selection from hierarchy click (fires callback).
+    /// Also clears any rect multi-selection.
+    void SetSelectedPathFromClick(const std::string& path);
+
+    /// Set single selection from viewport single-click (no callback, scroll-to).
+    void SetSelectedPath(const std::string& path) {
+        m_selectedPaths.clear();
+        m_primarySelectedPath  = path;
+        m_primarySdfPath       = path.empty() ? SdfPath() : SdfPath(path);
+        if (!path.empty()) m_selectedPaths.insert(path);
+        m_scrollToSelected     = !path.empty();
+    }
+
+    /// Set multi-selection from viewport rect pick (no callback, scroll-to first).
+    void SetSelectedPaths(const std::vector<std::string>& paths) {
+        m_selectedPaths.clear();
+        m_primarySelectedPath.clear();
+        m_primarySdfPath = SdfPath();
+        for (const auto& p : paths) m_selectedPaths.insert(p);
+        if (!paths.empty()) {
+            m_primarySelectedPath = paths.front();
+            m_primarySdfPath      = SdfPath(paths.front());
+            m_scrollToSelected    = true;
+        }
+    }
+
+    using PrimSelectCallback = std::function<void(const std::string& path)>;
+    void SetOnPrimSelected(PrimSelectCallback callback) { m_onPrimSelected = callback; }
+
+    /// Called when stage-level metadata (e.g. up axis) is changed via the hierarchy panel.
+    using StageMetadataChangedCallback = std::function<void()>;
+    void SetOnStageMetadataChanged(StageMetadataChangedCallback callback) { m_onStageMetadataChanged = callback; }
+    
+private:
+    void RenderPrimNode(const UsdPrim& prim);
+    const char* GetPrimTypeIcon(const UsdPrim& prim) const;
+    Icon        GetPrimTypeIconEnum(const UsdPrim& prim) const;
+    void RenderContextMenu(const UsdPrim& prim);
+    void RenderRemovePrimModal();
+    void ProcessPendingReplaceRef();
+    
+    PropertyManager*             m_propertyManager;
+    CommandHistory*              m_commandHistory = nullptr;
+    IconManager*                 m_iconManager = nullptr;
+    UsdStageRefPtr               m_stage;
+
+    /// Primary path: the scroll/frame target; also used for F-to-frame.
+    std::string                  m_primarySelectedPath;
+    SdfPath                      m_primarySdfPath;
+    /// Full set of selected paths (supports multi-select from rect pick).
+    std::unordered_set<std::string> m_selectedPaths;
+
+    bool                         m_scrollToSelected  = false;
+
+    /// Stage-local layer identifiers rebuilt once per Render() for override detection.
+    /// Contains only the stage's own layers (root + sublayers + session),
+    /// NOT layers that came in through references or payloads.
+    std::unordered_set<std::string> m_localLayers;
+
+    PrimSelectCallback           m_onPrimSelected;
+    StageMetadataChangedCallback m_onStageMetadataChanged;
+
+    /// Remove-prim confirmation state.
+    bool                         m_showRemovePrimConfirm = false;
+    SdfPath                      m_pendingRemovePrimPath;
+
+    /// Replace-reference deferred state (file dialog must run outside popup stack).
+    bool                         m_doReplaceRefPick  = false;
+    SdfPath                      m_pendingReplaceRefPrim;
+    pxr::SdfReference            m_pendingReplaceRef;
+};
+
+} // namespace UsdLayerManager