From f80d1fa3a66b9779a5d4353a2f6a784cba117b7a Mon Sep 17 00:00:00 2001
From: alex6dj <alexeiagueroalba@gmail.com>
Date: Sat, 31 Jul 2021 15:34:47 -0400
Subject: [PATCH] Basic ffprobe subtitle support

---
 FFMpegCore/FFProbe/IMediaAnalysis.cs |  2 ++
 FFMpegCore/FFProbe/MediaAnalysis.cs  | 21 ++++++++++++++++++---
 FFMpegCore/FFProbe/SubtitleStream.cs |  7 +++++++
 3 files changed, 27 insertions(+), 3 deletions(-)
 create mode 100644 FFMpegCore/FFProbe/SubtitleStream.cs

diff --git a/FFMpegCore/FFProbe/IMediaAnalysis.cs b/FFMpegCore/FFProbe/IMediaAnalysis.cs
index 4e67d4f..7be3b20 100644
--- a/FFMpegCore/FFProbe/IMediaAnalysis.cs
+++ b/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; }
     }
 }
diff --git a/FFMpegCore/FFProbe/MediaAnalysis.cs b/FFMpegCore/FFProbe/MediaAnalysis.cs
index 2602f86..d021813 100644
--- a/FFMpegCore/FFProbe/MediaAnalysis.cs
+++ b/FFMpegCore/FFProbe/MediaAnalysis.cs
@@ -12,8 +12,9 @@ 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)
         {
             return new MediaFormat
@@ -36,12 +37,14 @@ private MediaFormat ParseFormat(Format analysisFormat)
         }.Max();
 
         public MediaFormat Format { get; }
+        
         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
diff --git a/FFMpegCore/FFProbe/SubtitleStream.cs b/FFMpegCore/FFProbe/SubtitleStream.cs
new file mode 100644
index 0000000..80493f4
--- /dev/null
+++ b/FFMpegCore/FFProbe/SubtitleStream.cs
@@ -0,0 +1,7 @@
+namespace FFMpegCore
+{
+    public class SubtitleStream : MediaStream
+    {
+
+    }
+}
\ No newline at end of file