Fixed issue with templateCollections loading

Author: Cibbi

Editor/ShaderGenerator.cs

@@ -324,12 +324,24 @@ private static void AddFreshShaderToList(this Dictionary<TemplateAsset, Template
             if ((object)asset == null) return;
             if (dictionary.ContainsKey(asset)) return;
             string assetPath = AssetDatabase.GetAssetPath(asset);
-            dictionary.Add(asset, AssetDatabase.LoadAssetAtPath<TemplateAsset>(assetPath));
+            var genericAsset = AssetDatabase.LoadMainAssetAtPath(assetPath);
+            TemplateAsset template = null;
+            switch (genericAsset)
+            {
+                case TemplateCollectionAsset collection:
+                    template = collection.Templates.FirstOrDefault(x => x.name.Equals(asset.name));
+                    break;
+                case TemplateAsset t:
+                    template = t;
+                    break;
+            }
+            dictionary.Add(asset, template);
         }
 
         // Retrieves the new version of the asset from the dictionary if available
         private static TemplateAsset GetTemplate(this Dictionary<TemplateAsset, TemplateAsset> dictionary, TemplateAsset asset)
         {
+            if ((object)asset == null) return null;
             return dictionary.TryGetValue(asset, out TemplateAsset result) ? result : null;
         }