Merge pull request #1 from LuaxY/master

Add subtitle support
vansante · May 1, 2018 · d08148e · d08148e
2 parents 6489095 + fd3bbf1
commit d08148e
Showing 1 changed file with 14 additions and 0 deletions.
diff --git 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
 }