Subtitle metadata reading support

lib/ffmpeg/movie.rb

@@ -8,6 +8,7 @@ class Movie
     attr_reader :path, :duration, :time, :bitrate, :rotation, :creation_time
     attr_reader :video_stream, :video_codec, :video_bitrate, :colorspace, :width, :height, :sar, :dar, :frame_rate
     attr_reader :audio_streams, :audio_stream, :audio_codec, :audio_bitrate, :audio_sample_rate, :audio_channels, :audio_tags
+    attr_reader :subtitle_streams, :subtitle_stream, :subtitle_codec, :subtitle_language
     attr_reader :container
     attr_reader :metadata, :format_tags
 
@@ -53,6 +54,7 @@ def initialize(path)
       else
         video_streams = @metadata[:streams].select { |stream| stream.key?(:codec_type) and stream[:codec_type] === 'video' }
         audio_streams = @metadata[:streams].select { |stream| stream.key?(:codec_type) and stream[:codec_type] === 'audio' }
+        subtitle_streams = @metadata[:streams].select { |stream| stream.key?(:codec_type) and stream[:codec_type] === 'subtitle' }
 
         @container = @metadata[:format][:format_name]
 
@@ -123,6 +125,22 @@ def initialize(path)
           @audio_tags = audio_stream[:audio_tags]
           @audio_stream = audio_stream[:overview]
         end
+
+        @subtitle_streams = subtitle_streams.map do |stream|
+          {
+            :index => stream[:index],
+            :language => stream[:tags][:language],
+            :codec_name => stream[:codec_tag_string],
+            :overview => "#{stream[:codec_tag_string]} (#{stream[:tags][:language]})"
+          }
+        end
+
+        subtitle_stream = @subtitle_streams.first
+        unless subtitle_stream.nil?
+          @subtitle_language = subtitle_stream[:language]
+          @subtitle_codec = subtitle_stream[:codec_name]
+          @subtitle_stream = subtitle_stream[:overview]
+        end
 
       end
 

spec/ffmpeg/movie_spec.rb

@@ -213,6 +213,14 @@ module FFMPEG
         end
       end
 
+      context "given a file with a subtitle stream" do
+        let(:movie) {Movie.new("#{fixture_path}/movies/awesome movie with subtitles.m4v") }
+
+        it "should be valid" do
+          expect(movie).to be_valid
+        end
+      end
+
       context "given a file named with URL characters" do
         let(:movie) { Movie.new("#{fixture_path}/movies/file+with+data&streams=works?.mp4") }
 
@@ -505,6 +513,30 @@ module FFMPEG
       end
     end
 
+    context "given a file with subtitle streams" do
+      let(:movie) {Movie.new("#{fixture_path}/movies/awesome movie with subtitles.m4v") }
+
+      it "should identify both subtitle streams" do
+        expect(movie.subtitle_streams.length).to eq(2)
+      end
+
+      it "should assign subtitle_codec to the format of the first stream" do
+        subtitle_codec = movie.subtitle_streams[0][:codec_name]
+        expect(movie.subtitle_codec).to eq(subtitle_codec)
+      end
+
+      it "should assign subtitle_language to the language of the first stream" do
+        subtitle_language = movie.subtitle_streams[0][:language]
+        expect(movie.subtitle_language).to eq(subtitle_language)
+      end
+
+      it "should assign subtitle_stream to the properties of the first stream" do
+        stream_overview = movie.subtitle_streams[0][:overview]
+        expect(movie.subtitle_stream).to eq stream_overview
+      end
+
+    end
+
     describe "transcode" do
       let(:movie) { Movie.new("#{fixture_path}/movies/awesome movie.mov")}