Cache now updates if script is different

YashikMoloka · Sep 6, 2023 · e9f7f77 · e9f7f77
1 parent 30294af
commit e9f7f77
Showing 1 changed file with 15 additions and 1 deletion.
diff --git a/HierarchyDecorator/Scripts/Editor/Data/Types/ComponentGroup.cs b/HierarchyDecorator/Scripts/Editor/Data/Types/ComponentGroup.cs
@@ -167,9 +167,18 @@ public void Update(ComponentType component, MonoScript script)
                 return;
             }
 
+            // Remove old component cached
+
+            if (Contains(component))
+            {
+                lookup.Remove(component.Type);
+            }
+
+            // Update type and recache 
+
             component.UpdateType(script);
 
-            if (!lookup.ContainsKey(component.Type))
+            if (!Contains(component.Type))
             {
                 lookup.Add(component.Type, component);
             }
@@ -340,6 +349,11 @@ public bool Contains(ComponentType component)
 
         public bool Contains(Type type)
         {
+            if (hasCached && lookup.ContainsKey(type))
+            {
+                return true;
+            }
+
             for (int i = 0; i < components.Count; i++)
             {
                 if (components[i].Type == type)