From fd3bbf11398f9a3383ada61495096121fc25c581 Mon Sep 17 00:00:00 2001
From: LuaxY <yann.guineau@voidmx.net>
Date: Tue, 1 May 2018 14:13:46 +0200
Subject: [PATCH] Add subtitle support

---
 probedata.go | 14 ++++++++++++++
 1 file changed, 14 insertions(+)

diff --git a/probedata.go b/probedata.go
index 2ed0e54..76d2062 100644
--- a/probedata.go
+++ b/probedata.go
@@ -10,6 +10,7 @@ const (
 	STREAM_ANY   StreamType = ""
 	STREAM_VIDEO            = "video"
 	STREAM_AUDIO            = "audio"
+	STREAM_SUBTITLE         = "subtitle"
 )
 
 type ProbeData struct {
@@ -146,4 +147,17 @@ func (p *ProbeData) GetFirstAudioStream() (str *Stream) {
 		}
 	}
 	return
+}
+
+func (p *ProbeData) GetFirstSubtitleStream() (str *Stream) {
+	for _, s := range p.Streams {
+		if s == nil {
+			continue
+		}
+		if s.CodecType == STREAM_SUBTITLE {
+			str = s
+			return
+		}
+	}
+	return
 }
\ No newline at end of file