fix: implementation

brunomikoski · Sep 4, 2023 · fba8180 · fba8180
1 parent 6a36c66
commit fba8180
Showing 1 changed file with 4 additions and 2 deletions.
diff --git a/Scripts/Runtime/Core/CollectionsRegistry.cs b/Scripts/Runtime/Core/CollectionsRegistry.cs
@@ -170,13 +170,15 @@ public bool TryGetCollectionsOfItemType<T>(out List<ScriptableObjectCollection<T
             return false;
         }
 
-        public bool TryGetCollectionsOfType<T>(out List<T> inputActionMapCollections) where T : ScriptableObjectCollection
+        public bool TryGetCollectionsOfType<T>(out List<T> inputActionMapCollections, bool allowSubclasses = true) where T : ScriptableObjectCollection
         {
             List<T> result = new List<T>();
+            Type targetType = typeof(T);
             for (int i = 0; i < collections.Count; i++)
             {
                 ScriptableObjectCollection scriptableObjectCollection = collections[i];
-                if (scriptableObjectCollection.GetType() == typeof(T) || scriptableObjectCollection.GetType().IsSubclassOf(typeof(T)))
+                Type collectionType = scriptableObjectCollection.GetType();
+                if (collectionType == targetType || (allowSubclasses && collectionType.IsSubclassOf(targetType)))
                     result.Add((T)scriptableObjectCollection);
             }