From eb58ab5327dc78032c2dfaa0e41d29108fae10b9 Mon Sep 17 00:00:00 2001
From: Sinai <outlander.sinai@gmail.com>
Date: Thu, 22 Apr 2021 22:57:04 +1000
Subject: [PATCH] Fix TransformCell for destroyed transforms

---
 src/UI/Widgets/TransformTree/TransformCell.cs | 39 ++++++++++++-------
 1 file changed, 24 insertions(+), 15 deletions(-)

diff --git a/src/UI/Widgets/TransformTree/TransformCell.cs b/src/UI/Widgets/TransformTree/TransformCell.cs
index 53bf16a..5646bfb 100644
--- a/src/UI/Widgets/TransformTree/TransformCell.cs
+++ b/src/UI/Widgets/TransformTree/TransformCell.cs
@@ -48,8 +48,11 @@ namespace UnityExplorer.UI.Widgets
         //This is called from the SetCell method in DataSource
         public void ConfigureCell(CachedTransform cached, int cellIndex)
         {
-            if (cached == null || !cached.Value)
+            if (cached == null)
+            {
+                ExplorerCore.LogWarning("Setting TransformTree cell but the CachedTransform is null!");
                 return;
+            }
 
             if (!Enabled)
                 Enable();
@@ -59,25 +62,31 @@ namespace UnityExplorer.UI.Widgets
 
             spacer.minWidth = cached.Depth * 15;
 
-            nameLabel.text = cached.Value.name;
-            nameLabel.color = cached.Value.gameObject.activeSelf ? Color.white : Color.grey;
-
-            int childCount = cached.Value.childCount;
-            if (childCount > 0)
+            if (cached.Value)
             {
-                nameLabel.text = $"<color=grey>[{childCount}]</color> {nameLabel.text}";
+                nameLabel.text = cached.Value.name;
+                nameLabel.color = cached.Value.gameObject.activeSelf ? Color.white : Color.grey;
 
-                expandButton.interactable = true;
-                //expandLabel.enabled = true;
-                expandLabel.text = cached.Expanded ? "▼" : "►";
-                expandLabel.color = cached.Expanded ? new Color(0.5f, 0.5f, 0.5f) : new Color(0.3f, 0.3f, 0.3f);
+                int childCount = cached.Value.childCount;
+                if (childCount > 0)
+                {
+                    nameLabel.text = $"<color=grey>[{childCount}]</color> {nameLabel.text}";
+
+                    expandButton.interactable = true;
+                    expandLabel.text = cached.Expanded ? "▼" : "►";
+                    expandLabel.color = cached.Expanded ? new Color(0.5f, 0.5f, 0.5f) : new Color(0.3f, 0.3f, 0.3f);
+                }
+                else
+                {
+                    expandButton.interactable = false;
+                    expandLabel.text = "▪";
+                    expandLabel.color = new Color(0.3f, 0.3f, 0.3f);
+                }
             }
             else
             {
-                expandButton.interactable = false;
-                expandLabel.text = "▪";
-                expandLabel.color = new Color(0.3f, 0.3f, 0.3f);
-                //expandLabel.enabled = false;
+                nameLabel.text = $"[Destroyed]";
+                nameLabel.color = Color.red;
             }
         }