From b6d459b664a25bdc558f1ec0c82997ec73dc7731 Mon Sep 17 00:00:00 2001
From: originalnicodr <niconavall@gmail.com>
Date: Wed, 7 Aug 2024 19:39:02 -0300
Subject: [PATCH] Search AnimatorCell-associated meshes again when refreshing
 the animator's list. Refreshes the bone panel with the new meshes.

---
 src/Cinematic/Cells/AnimatorPlayer.cs | 17 +++++++++++++----
 src/UI/Panels/AnimatorPanel.cs        |  4 ++++
 src/UI/Panels/BonesPanel.cs           |  7 +++++++
 3 files changed, 24 insertions(+), 4 deletions(-)

diff --git a/src/Cinematic/Cells/AnimatorPlayer.cs b/src/Cinematic/Cells/AnimatorPlayer.cs
index 7adf2ba..fe4fb6d 100644
--- a/src/Cinematic/Cells/AnimatorPlayer.cs
+++ b/src/Cinematic/Cells/AnimatorPlayer.cs
@@ -48,8 +48,12 @@ namespace UnityExplorer.UI.Panels
 
             this.favAnimations = new List<IAnimationClip>();
 
-            this.skinnedMeshes.AddRange(this.animator.wrappedObject.gameObject.GetComponentsInChildren<SkinnedMeshRenderer>(false));
-            this.extraMeshes.AddRange(this.animator.wrappedObject.gameObject.GetComponentsInChildren<MeshRenderer>(false));
+            SearchMeshes();
+        }
+
+        public void SearchMeshes(){
+            skinnedMeshes = new List<SkinnedMeshRenderer>(animator.wrappedObject.gameObject.GetComponentsInChildren<SkinnedMeshRenderer>(false));
+            extraMeshes = new List<MeshRenderer>(animator.wrappedObject.gameObject.GetComponentsInChildren<MeshRenderer>(false));
         }
 
         // Include the animations being played in other layers
@@ -125,7 +129,7 @@ namespace UnityExplorer.UI.Panels
             }
         }
 
-        private List<Transform> GetMeshes(){
+        private List<Transform> GetMeshesTransforms(){
             List<Transform> meshes = new List<Transform>();
 
             foreach (SkinnedMeshRenderer skinnedMesh in skinnedMeshes) {
@@ -139,11 +143,16 @@ namespace UnityExplorer.UI.Panels
         public void OpenBonesPanel(){
             if (skinnedMeshes.Count == 0 && extraMeshes.Count == 0) return;
             if (bonesManager == null){
-                bonesManager = new BonesManager(UIManager.GetPanel<UnityExplorer.UI.Panels.AnimatorPanel>(UIManager.Panels.AnimatorPanel).Owner, GetMeshes(), animator);
+                bonesManager = new BonesManager(UIManager.GetPanel<UnityExplorer.UI.Panels.AnimatorPanel>(UIManager.Panels.AnimatorPanel).Owner, GetMeshesTransforms(), animator);
             }
             bonesManager.SetActive(true);
         }
 
+        public void MaybeResetBonesPanel(){
+            if (bonesManager == null) return;
+            bonesManager.RefreshBones(GetMeshesTransforms());
+        }
+
         public void SetMeshesEnabled(bool value){
             foreach (SkinnedMeshRenderer skinnedMesh in skinnedMeshes) {
                 skinnedMesh.TryCast<Renderer>().enabled = value;
diff --git a/src/UI/Panels/AnimatorPanel.cs b/src/UI/Panels/AnimatorPanel.cs
index 02e21d2..1388ed4 100644
--- a/src/UI/Panels/AnimatorPanel.cs
+++ b/src/UI/Panels/AnimatorPanel.cs
@@ -76,6 +76,10 @@ namespace UnityExplorer.UI.Panels
                                 animators[i].animations.Add(animationClip);
                         }
                         newAnimators[newAnimatorsIndex] = animators[i];
+
+                        // Reset meshes
+                        newAnimators[newAnimatorsIndex].SearchMeshes();
+                        newAnimators[newAnimatorsIndex].MaybeResetBonesPanel();
                     }
                 }
             }
diff --git a/src/UI/Panels/BonesPanel.cs b/src/UI/Panels/BonesPanel.cs
index 9a6b57c..7e970a6 100644
--- a/src/UI/Panels/BonesPanel.cs
+++ b/src/UI/Panels/BonesPanel.cs
@@ -35,6 +35,13 @@ namespace UnityExplorer.UI.Panels
             BuildBoneTrees();
         }
 
+        public void RefreshBones(List<Transform> bones) {
+            this.bones = bones;
+            boneTrees.Clear();
+            BuildBoneTrees();
+            boneScrollPool.Refresh(true, true);
+        }
+
         private void BuildBoneTrees(){
             BoneTree root = new BoneTree(animator.wrappedObject.gameObject, bones);
             if (root.obj != null){