video_screenshoter 0.0.6 → 0.1.0

data/lib/video_screenshoter/abstract.rb

@@ -0,0 +1,44 @@
+# encoding: utf-8
+
+module VideoScreenshoter
+  class Abstract
+    attr_accessor :ffmpeg, :output_dir, :output_file, :input, :times, :duration, :verbose, :size
+
+    def initialize params
+      [:ffmpeg, :output_dir, :output_file, :verbose].each do |attr|
+        self.send("#{attr}=".to_sym, params[attr].nil? ? VideoScreenshoter.send(attr) : params[attr])
+      end
+      FileUtils.mkdir_p self.output_dir
+      self.input = params[:input] or raise ArgumentError.new('Input is needed')
+      self.duration = input_duration
+      raise ArgumentError.new('Incorrect or empty m3u8 playlist') if duration.to_i == 0
+      self.times = params[:times].to_a.map do |time|
+        if time.is_a?(String) && matches = time.match(/(.*)%$/)
+          time = matches[1].to_f / 100 * duration
+        end
+        time = duration + time if time < 0
+        time = time.to_i
+        time
+      end.uniq
+      self.size = params[:size] ? "-s #{params[:size]}" : ''
+    end
+
+    def output_fullpath time
+      sprintf(File.join(output_dir, output_file), time)
+    end
+
+    def command input, output, time
+      "#{ffmpeg} -i #{input} -acodec -an -ss #{time} #{size} -f image2 -vframes 1 -y #{output} 2>/dev/null 1>&2"
+    end
+
+    def run
+      raise NotImplementedError
+    end
+
+    protected
+
+    def input_duration
+      raise NotImplementedError
+    end
+  end
+end

data/lib/video_screenshoter/hls.rb

@@ -0,0 +1,42 @@
+# encoding: utf-8
+
+module VideoScreenshoter
+  class Hls < Abstract
+
+    def initialize params
+      super
+    end
+
+    def run
+      chunk_limits = []
+      chunks.inject(0.0) do |time, chunk|
+        chunk_limits.push({:start => time, :end => time + chunk.first, :path => chunk.last})
+        time + chunk.first
+      end
+      times.each do |time|
+        chunk_limit = chunk_limits.select { |c| time > c[:start] && time <= c[:end] }.first
+        path = File.join(File.dirname(input), chunk_limit[:path])
+        rel_time = time - chunk_limit[:start]
+        cmd = command(path, output_fullpath(time), rel_time)
+        puts cmd if verbose
+        `#{cmd}`
+      end
+    end
+
+    alias :make_screenshots :run
+    alias :make_thumbnails :run
+
+    private
+    
+    attr_accessor :chunks
+
+    def input_duration
+      (self.chunks = parse_playlist).map(&:first).inject(:+)
+    end
+    
+    def parse_playlist
+      require 'open-uri'
+      open(input) { |f| f.read }.scan(/EXTINF:([\d.]+).*?\n(.*?)\n/).map { |c| [c.first.to_f, c.last] }
+    end
+  end
+end

data/lib/video_screenshoter/version.rb

@@ -1,3 +1,3 @@
 module VideoScreenshoter
-  VERSION = "0.0.6"
+  VERSION = "0.1.0"
 end

data/lib/video_screenshoter/video.rb

@@ -1,34 +1,15 @@
 # encoding: utf-8
 
 module VideoScreenshoter
-  class Video
-    attr_accessor :ffmpeg, :output_dir, :output_file, :input, :times, :duration, :verbose, :size
+  class Video < Abstract
 
     def initialize params
-      [:ffmpeg, :output_dir, :output_file, :verbose].each do |attr|
-        self.send("#{attr}=".to_sym, params[attr].nil? ? VideoScreenshoter.send(attr) : params[attr])
-      end
-      FileUtils.mkdir_p self.output_dir
-      self.input = params[:input] or raise ArgumentError.new('Input is needed')
-      self.duration = input_duration or raise ArgumentError.new('Incorrect video file')
-      self.times = params[:times].to_a.map do |time|
-        if time.is_a?(String) && matches = time.match(/(.*)%$/)
-          time = matches[1].to_f / 100 * duration
-        end
-        time = duration + time if time < 0
-        time = time.to_i
-        time
-      end
-      self.size = params[:size] ? "-s #{params[:size]}" : ''
-    end
-
-    def output_fullpath time
-      sprintf(File.join(output_dir, output_file), time)
+      super
     end
 
     def run
       times.each do |time|
-        cmd = "#{ffmpeg} -i #{input} -acodec -an -ss #{time} #{size} -f image2 -vframes 1 -y #{output_fullpath(time)} 2>/dev/null 1>&2"
+        cmd = command(input, output_fullpath(time), time)
         puts cmd if verbose
         `#{cmd}`
       end

data/lib/video_screenshoter.rb

@@ -1,6 +1,8 @@
 require "video_screenshoter/version"
 require "fileutils"
+require "video_screenshoter/abstract"
 require "video_screenshoter/video"
+require "video_screenshoter/hls"
 
 module VideoScreenshoter
   class << self
@@ -12,6 +14,8 @@ module VideoScreenshoter
   self.verbose = false
 
   def self.new params
-    VideoScreenshoter::Video.new params
+    raise ArgumentError.new('Incorrect type param') unless [nil, 'hls', 'video'].include? params[:type]
+    params[:type] ||= File.extname(params[:input]) == '.m3u8' ? 'hls' : 'video'
+    params[:type] == 'hls' ? VideoScreenshoter::Hls.new(params) : VideoScreenshoter::Video.new(params)
   end
 end

data/test/video_screenshoter_test.rb

@@ -1,10 +1,10 @@
 require 'test_helper'
 
 class VideoScreenshoterTest < Test::Unit::TestCase
-  context 'make screenshots' do
+  context 'video' do
     setup do
       @input = 'http://techslides.com/demos/sample-videos/small.mp4'
-      @output_dir = '/tmp/screenshots_test'
+      @output_dir = '/tmp/video_screenshots_test'
       VideoScreenshoter.new(:input => @input, :output_dir => @output_dir, :times => [1, -2, '50%', '-10%']).make_screenshots
     end
     should 'create screenshots' do
@@ -14,4 +14,30 @@ class VideoScreenshoterTest < Test::Unit::TestCase
       end
     end
   end
+
+  context 'hls' do
+    context 'with 1 level playlist' do
+      setup do
+        @input = 'http://devimages.apple.com.edgekey.net/streaming/examples/bipbop_4x3/bipbop_4x3_variant.m3u8'
+        @output_dir = '/tmp/hls_screenshots_test'
+      end
+      should 'raise error' do
+        assert_raises(ArgumentError) { VideoScreenshoter.new(:input => @input, :output_dir => @output_dir, :times => ['10%', '50%']).make_thumbnails }
+      end
+    end
+
+    context 'with 2 level playlist' do
+      setup do
+        @input = 'http://devimages.apple.com.edgekey.net/streaming/examples/bipbop_4x3/gear4/prog_index.m3u8'
+        @output_dir = '/tmp/hls_screenshots_test'
+        VideoScreenshoter.new(:input => @input, :output_dir => @output_dir, :times => ['10%', '50%', '-10%']).make_thumbnails
+      end
+      should 'create screenshots' do
+        ["scr1620.jpg", "scr180.jpg", "scr900.jpg"].each do |scr|
+          assert File.exists?(File.join(@output_dir, scr))
+          assert File.size(File.join(@output_dir, scr)) > 0
+        end
+      end
+    end
+  end
 end

metadata

@@ -1,7 +1,7 @@
 --- !ruby/object:Gem::Specification
 name: video_screenshoter
 version: !ruby/object:Gem::Version
-  version: 0.0.6
+  version: 0.1.0
   prerelease: 
 platform: ruby
 authors:
@@ -9,7 +9,7 @@ authors:
 autorequire: 
 bindir: bin
 cert_chain: []
-date: 2013-05-15 00:00:00.000000000 Z
+date: 2013-05-16 00:00:00.000000000 Z
 dependencies:
 - !ruby/object:Gem::Dependency
   name: bundler
@@ -88,6 +88,8 @@ files:
 - README.md
 - Rakefile
 - lib/video_screenshoter.rb
+- lib/video_screenshoter/abstract.rb
+- lib/video_screenshoter/hls.rb
 - lib/video_screenshoter/version.rb
 - lib/video_screenshoter/video.rb
 - test/test_helper.rb
@@ -108,7 +110,7 @@ required_ruby_version: !ruby/object:Gem::Requirement
       version: '0'
       segments:
       - 0
-      hash: 1571913647041927257
+      hash: -4324957323078991671
 required_rubygems_version: !ruby/object:Gem::Requirement
   none: false
   requirements:
@@ -117,7 +119,7 @@ required_rubygems_version: !ruby/object:Gem::Requirement
       version: '0'
       segments:
       - 0
-      hash: 1571913647041927257
+      hash: -4324957323078991671
 requirements:
 - ffmpeg
 rubyforge_project: