use blacklist and sorted

DREDGE-Mods · Sep 5, 2024 · 239d2fe · 239d2fe
1 parent 1eed710
commit 239d2fe
Showing 1 changed file with 2 additions and 2 deletions.
diff --git a/Winch/Util/AssetBundleUtil.cs b/Winch/Util/AssetBundleUtil.cs
@@ -33,7 +33,7 @@ internal static Shader CacheShader(this Shader shader)
 
         internal static Shader[] GetSortedShaders()
         {
-            var shaders = Resources.FindObjectsOfTypeAll<Shader>().Where(shader => shader.isSupported).Distinct(UnityObjectComparer<Shader>.Instance).ToList();
+            var shaders = Resources.FindObjectsOfTypeAll<Shader>().Where(shader => shader.isSupported && !blacklistedShaders.ContainsKey(shader.GetInstanceID())).Distinct(UnityObjectComparer<Shader>.Instance).ToList();
             shaders.Sort(UnityObjectComparer.Instance);
             shaders.Reverse();
             return shaders.ToArray();
@@ -44,7 +44,7 @@ public static Shader GetReplacementShader(string name)
             Shader replacementShader;
             if (cachedShaders.TryGetValue(name, out replacementShader) && replacementShader != null) return replacementShader;
 
-            replacementShader = Resources.FindObjectsOfTypeAll<Shader>().Where(shader => shader.isSupported).Reverse().FirstOrDefault(shader => shader.name == name);// Shader.Find(name);
+            replacementShader = GetSortedShaders().FirstOrDefault(shader => shader.name == name);// Shader.Find(name);
             if (replacementShader != null)
                 return replacementShader.CacheShader();