feat: Batch Isolate Vocals should filter out songs with unsupported a…

…udio formarts
UltraStar-Deluxe · Oct 17, 2024 · ed475b5 · ed475b5
1 parent 9f9ea94
commit ed475b5
Showing 1 changed file with 24 additions and 3 deletions.
diff --git a/...ar Play/Assets/StreamingAssets/Mods/BatchIsolateVocals/1-BatchIsolateVocalsModSettings.cs b/...ar Play/Assets/StreamingAssets/Mods/BatchIsolateVocals/1-BatchIsolateVocalsModSettings.cs
@@ -1,5 +1,6 @@
 using System;
 using System.Collections.Generic;
+using System.IO;
 using System.Linq;
 using UniInject;
 using UniRx;
@@ -39,7 +40,7 @@ public List<IModSettingControl> GetModSettingControls()
 
     private void OnSelectAll()
     {
-        toggles.ForEach(toggle => toggle.value = true);
+        toggles.FindAll(toggle => toggle.enabledSelf).ForEach(toggle => toggle.value = true);
     }
 
     private void OnDeselectAll()
@@ -152,13 +153,33 @@ public VisualElement CreateVisualElement()
                 toggle.label = SongMetaUtils.GetArtistDashTitle(songMeta);
                 toggle.value = false;
                 toggle.userData = songMeta;
-
-                toggles.Add(toggle);
 
+                if (!IsSongSupported(songMeta))
+                {
+                    toggle.SetEnabled(false);
+                }
+                toggles.Add(toggle);
                 toggleContainer.Add(toggle);
             }
 
             return toggleContainer;
         }
+
+        private readonly string[] supportedExtensions = { "wav", "mp3", "ogg", "m4a", "wma", "flac" };
+
+        private bool IsSongSupported(SongMeta songMeta)
+        {
+            string audio = songMeta.Audio;
+            string extension = Path.GetExtension(audio).ToLower();
+            if (extension.StartsWith("."))
+            {
+                extension = extension.Substring(1);
+            }
+            bool result = Array.Exists(supportedExtensions, ext => ext.Equals(extension));
+            Debug.Log($"Checking support for {audio}, extension: {extension}, result: {result}");
+            return result;
+        }
+
+
     }
 }