Merge pull request #233 from alex6dj/feature/support-subtitles-stream

Basic ffprobe subtitle streams support

Former-commit-id: 1a87b0b133

FFMpegCore.Test/FFMpegCore.Test.csproj

@@ -83,6 +83,9 @@
     <None Update="Resources\mute.mp4">
       <CopyToOutputDirectory>Always</CopyToOutputDirectory>
     </None>
+    <None Update="Resources\sample.srt">
+      <CopyToOutputDirectory>PreserveNewest</CopyToOutputDirectory>
+    </None>
   </ItemGroup>
 
   <ItemGroup>

FFMpegCore.Test/FFProbeTests.cs

@@ -100,5 +100,15 @@ public async Task Probe_Success_FromStream_Async()
             var info = await FFProbe.AnalyseAsync(stream);
             Assert.AreEqual(3, info.Duration.Seconds);
         }
+
+        [TestMethod, Timeout(10000)]
+        public async Task Probe_Success_Subtitle_Async()
+        {
+            var info = await FFProbe.AnalyseAsync(TestResources.SrtSubtitle);
+            Assert.IsNotNull(info.PrimarySubtitleStream);
+            Assert.AreEqual(1, info.SubtitleStreams.Count);
+            Assert.AreEqual(0, info.AudioStreams.Count);
+            Assert.AreEqual(0, info.VideoStreams.Count);
+        }
     }
 }

FFMpegCore.Test/Resources/TestResources.cs

@@ -20,5 +20,6 @@ public static class TestResources
         public static readonly string Mp3Audio = "./Resources/audio.mp3";
         public static readonly string PngImage = "./Resources/cover.png";
         public static readonly string ImageCollection = "./Resources/images";
+        public static readonly string SrtSubtitle = "./Resources/sample.srt";
     }
 }

FFMpegCore.Test/Resources/sample.srt

@@ -0,0 +1,12 @@
+1
+00:00:00,000 --> 00:00:01,500
+For www.forom.com
+
+2
+00:00:01,500 --> 00:00:02,500
+<i>Tonight's the night.</i>
+
+3
+00:00:03,000 --> 00:00:15,000
+<i>And it's going to happen
+again and again --</i>

FFMpegCore/FFProbe/IMediaAnalysis.cs

@@ -9,7 +9,9 @@ public interface IMediaAnalysis
         MediaFormat Format { get; }
         AudioStream? PrimaryAudioStream { get; }
         VideoStream? PrimaryVideoStream { get; }
+        SubtitleStream? PrimarySubtitleStream { get; }
         List<VideoStream> VideoStreams { get; }
         List<AudioStream> AudioStreams { get; }
+        List<SubtitleStream> SubtitleStreams { get; }
     }
 }

FFMpegCore/FFProbe/MediaAnalysis.cs

@@ -12,6 +12,7 @@ internal MediaAnalysis(FFProbeAnalysis analysis)
             Format = ParseFormat(analysis.Format);
             VideoStreams = analysis.Streams.Where(stream => stream.CodecType == "video").Select(ParseVideoStream).ToList();
             AudioStreams = analysis.Streams.Where(stream => stream.CodecType == "audio").Select(ParseAudioStream).ToList();
+            SubtitleStreams = analysis.Streams.Where(stream => stream.CodecType == "subtitle").Select(ParseSubtitleStream).ToList();
         }
         
         private MediaFormat ParseFormat(Format analysisFormat)
@@ -36,12 +37,14 @@ private MediaFormat ParseFormat(Format analysisFormat)
         }.Max();
 
         public MediaFormat Format { get; }
-        public AudioStream? PrimaryAudioStream => AudioStreams.OrderBy(stream => stream.Index).FirstOrDefault();
         
+        public AudioStream? PrimaryAudioStream => AudioStreams.OrderBy(stream => stream.Index).FirstOrDefault();
         public VideoStream? PrimaryVideoStream => VideoStreams.OrderBy(stream => stream.Index).FirstOrDefault();
+        public SubtitleStream? PrimarySubtitleStream => SubtitleStreams.OrderBy(stream => stream.Index).FirstOrDefault();
 
         public List<VideoStream> VideoStreams { get; }
         public List<AudioStream> AudioStreams { get; }
+        public List<SubtitleStream> SubtitleStreams { get; }
 
         private VideoStream ParseVideoStream(FFProbeStream stream)
         {
@@ -84,7 +87,19 @@ private AudioStream ParseAudioStream(FFProbeStream stream)
             };
         }
 
-
+        private SubtitleStream ParseSubtitleStream(FFProbeStream stream)
+        {
+            return new SubtitleStream
+            {
+                Index = stream.Index,
+                BitRate = !string.IsNullOrEmpty(stream.BitRate) ? MediaAnalysisUtils.ParseIntInvariant(stream.BitRate) : default,
+                CodecName = stream.CodecName,
+                CodecLongName = stream.CodecLongName,
+                Duration = MediaAnalysisUtils.ParseDuration(stream),
+                Language = stream.GetLanguage(),
+                Tags = stream.Tags,
+            };
+        }
     }
 
     public static class MediaAnalysisUtils

FFMpegCore/FFProbe/SubtitleStream.cs

@@ -0,0 +1,7 @@
+namespace FFMpegCore
+{
+    public class SubtitleStream : MediaStream
+    {
+
+    }
+}