FF-1785: Created a track remover based on the TrackSelectorFlowElement.cs

2026-02-16 07:08:58 -06:00 · 2024-09-26 16:58:28 +12:00
parent 5dbe70efd5
commit 2896bd3d61
3 changed files with 132 additions and 0 deletions
--- a/FileFlows.Plugin.dll
+++ b/FileFlows.Plugin.dll
--- a/FileFlows.Plugin.pdb
+++ b/FileFlows.Plugin.pdb
--- a/VideoNodes/Tests/FfmpegBuilderTests/FFmpegBuilder_TrackRemoverTests.cs
+++ b/VideoNodes/Tests/FfmpegBuilderTests/FFmpegBuilder_TrackRemoverTests.cs
@@ -0,0 +1,132 @@
+#if(DEBUG)
+
+using FileFlows.VideoNodes.FfmpegBuilderNodes;
+using FileFlows.VideoNodes.FfmpegBuilderNodes.Models;
+using Microsoft.VisualStudio.TestTools.UnitTesting;
+using VideoNodes.Tests;
+
+namespace FileFlows.VideoNodes.Tests.FfmpegBuilderTests;
+
+[TestClass]
+public class FFmpegBuilder_TrackRemoverTests : VideoTestBase
+{
+    VideoInfo vii;
+    NodeParameters args;
+    FfmpegModel Model;
+    
+    protected override void TestStarting()
+    {
+        args = GetVideoNodeParameters();
+        VideoFile vf = new VideoFile();
+        vf.PreExecute(args);
+        vf.Execute(args);
+        vii = (VideoInfo)args.Parameters["VideoInfo"];
+        
+        FfmpegBuilderStart ffStart = new();
+        ffStart.PreExecute(args);
+        Assert.AreEqual(1, ffStart.Execute(args));
+        Model = ffStart.GetModel();
+    }
+    
+    /// <summary>
+    /// Test German is matched and removed
+    /// </summary>
+    [TestMethod]
+    public void KeepEngAndOriginalSubtitles()
+    {
+        int originaluNonDeletedSubtitles = Model.SubtitleStreams.Count(x => x.Deleted == false);
+        FfmpegBuilderTrackRemover ffRemover = new();
+        args.Variables["OriginalLanguage"] = "fre";
+        ffRemover.CustomTrackSelection = true;
+        ffRemover.TrackSelectionOptions = new();
+        ffRemover.TrackSelectionOptions.Add(new ("Language", "!/orig|eng/"));
+        ffRemover.StreamType = "Subtitle";
+        ffRemover.PreExecute(args);
+        Assert.AreEqual(1, ffRemover.Execute(args));
+        List<FfmpegSubtitleStream> nonDeletedSubtitles = new();
+
+        Logger.ILog(new string('-', 100));
+        foreach (var sub in Model.SubtitleStreams.Where(x => x.Deleted))
+        {
+            Logger.ILog("Deleted subtitle: " + sub);
+        }
+        Logger.ILog(new string('-', 100));
+        foreach (var sub in Model.SubtitleStreams.Where(x => x.Deleted == false))
+        {
+            Logger.ILog("Non deleted subtitle: " + sub);
+            nonDeletedSubtitles.Add(sub);
+        }
+        Assert.AreNotEqual(originaluNonDeletedSubtitles, nonDeletedSubtitles.Count);
+        Assert.AreEqual(nonDeletedSubtitles.Count, 2);
+        Assert.AreEqual("eng", nonDeletedSubtitles[0].Language);
+        Assert.AreEqual("fre", nonDeletedSubtitles[1].Language);
+    }
+    /// <summary>
+    /// Test German is matched and removed
+    /// </summary>
+    [TestMethod]
+    public void KeepOriginalSubtitles()
+    {
+        int originaluNonDeletedSubtitles = Model.SubtitleStreams.Count(x => x.Deleted == false);
+        FfmpegBuilderTrackRemover ffRemover = new();
+        args.Variables["OriginalLanguage"] = "fre";
+        ffRemover.CustomTrackSelection = true;
+        ffRemover.TrackSelectionOptions = new();
+        ffRemover.TrackSelectionOptions.Add(new ("Language", "!orig"));
+        ffRemover.StreamType = "Subtitle";
+        ffRemover.PreExecute(args);
+        Assert.AreEqual(1, ffRemover.Execute(args));
+        List<FfmpegSubtitleStream> nonDeletedSubtitles = new();
+
+        Logger.ILog(new string('-', 100));
+        foreach (var sub in Model.SubtitleStreams.Where(x => x.Deleted))
+        {
+            Logger.ILog("Deleted subtitle: " + sub);
+        }
+        Logger.ILog(new string('-', 100));
+        foreach (var sub in Model.SubtitleStreams.Where(x => x.Deleted == false))
+        {
+            Logger.ILog("Non deleted subtitle: " + sub);
+            nonDeletedSubtitles.Add(sub);
+        }
+        Assert.AreNotEqual(originaluNonDeletedSubtitles, nonDeletedSubtitles.Count);
+        Assert.AreEqual(nonDeletedSubtitles.Count, 1);
+        Assert.AreEqual("fre", nonDeletedSubtitles[0].Language);
+    }
+    
+    /// <summary>
+    /// Tests english subtitles are removed
+    /// </summary>
+    [TestMethod]
+    public void RemoveEnglishSubtitles()
+    {
+        int originaluNonDeletedSubtitles = Model.SubtitleStreams.Count(x => x.Deleted == false);
+        FfmpegBuilderTrackRemover ffRemover = new();
+        args.Variables["OriginalLanguage"] = "fre";
+        ffRemover.CustomTrackSelection = true;
+        ffRemover.TrackSelectionOptions = new();
+        ffRemover.TrackSelectionOptions.Add(new ("Language", "eng"));
+        ffRemover.StreamType = "Subtitle";
+        ffRemover.PreExecute(args);
+        Assert.AreEqual(1, ffRemover.Execute(args));
+        List<FfmpegSubtitleStream> nonDeletedSubtitles = new();
+
+        Logger.ILog(new string('-', 100));
+        foreach (var sub in Model.SubtitleStreams.Where(x => x.Deleted))
+        {
+            Logger.ILog("Deleted subtitle: " + sub);
+        }
+        Logger.ILog(new string('-', 100));
+        foreach (var sub in Model.SubtitleStreams.Where(x => x.Deleted == false))
+        {
+            Logger.ILog("Non deleted subtitle: " + sub);
+            nonDeletedSubtitles.Add(sub);
+        }
+        Assert.AreNotEqual(originaluNonDeletedSubtitles, nonDeletedSubtitles.Count);
+        Assert.AreEqual(nonDeletedSubtitles.Count, originaluNonDeletedSubtitles - 1);
+        Assert.IsFalse(nonDeletedSubtitles.Any(x =>
+            x.Language.Contains("en", StringComparison.InvariantCultureIgnoreCase)));
+    }
+}
+
+#endif