puremotion 0.0.1

data/lib/puremotion/tools/ffmpeg.rb

@@ -0,0 +1,141 @@
+module PureMotion::Tools
+  
+  class FFmpeg
+    
+    class Status
+      
+      NOT_STARTED       = 1
+      INITIALIZING      = 2
+      ANALYZING         = 4
+      PREPARING_OUTPUT  = 8
+      ENCODING          = 16
+      ERROR             = 32
+      
+    end
+    
+    attr_reader :args
+    attr_accessor :output
+    
+    event :line
+    event :complete
+    event :exited
+    event :status_change
+    
+    @status = Status::NOT_STARTED
+    
+    def self.run(args = {})
+      FFmpeg.new( args )
+    end
+    
+    def initialize(params = {})
+      if params.class != Hash then raise ArgumentError, "Invalid parameters" end
+      
+      @defaults = {
+        :ffmpeg => 'ffmpeg',
+        :options => nil
+      }
+      
+      @params = params
+      
+      @options = @params[:options]
+      
+      if @options.nil? then
+        raise ArgumentError, "No options given"
+      end
+      
+      @output = []
+      
+      go
+    end
+    
+    def go
+      
+      @pio = IO.popen("ffmpeg #{@options} 2>&1")
+      
+      temp = ''
+      
+      PureMotion::Thread.new do
+        @pio.each_byte do |l|
+          if l == 10 or l == 13 then
+            @output.push temp
+            line(temp)
+            temp = ''
+            l = ''
+          else
+            l = l.chr
+          end
+          temp = temp + l
+          #@output << l.gsub("\r","\n")
+        end
+        if @pio.eof? then
+          @done = true
+          complete(true)
+        end
+      end
+    
+    end
+    
+    def ran?
+      @pio.eof? unless @pio.nil?
+      false
+      #!find(/^FFmpeg/).nil?
+    end
+    
+    def ended?
+      return false unless ran?
+      @pio.eof?
+    end
+    
+    def version
+      return nil unless ran?
+      find(/^FFmpeg version (\S+)/)
+    end
+    
+    def find(regexp)
+      m = regexp.match(@output)
+      return m[1] if m
+      nil
+    end
+    
+    def status
+      @status
+    end
+    
+    private
+    
+    def prepare_args
+      @args = []
+      @args << ['i', @input]
+      
+      args = []
+    
+      strings = []
+    
+      @args.each do |arg, value|
+        if value.nil? then value = '' end
+        value = '"' + value.to_s + '"' unless value.index(' ').nil?
+        
+        args << [ arg, value]
+        
+        if value.empty? or value =~ /^\s*$/ then
+          value = ''
+        else
+          value = ' ' << value
+        end
+        
+        strings << ( '-' << arg << value )
+      end
+      
+      if !@output.nil? then strings << '"' + @output + '"' end
+      
+      @args = strings.join(' ')
+    end
+    
+    def status=(status_const)
+      @status = status_const
+      status_change(status_const)
+    end
+    
+  end
+  
+end

data/lib/puremotion/transcode/recipe.rb

@@ -0,0 +1,250 @@
+require 'yaml'
+
+module PureMotion::Transcode
+  
+  class Recipe
+    
+    def self.from_file(file)
+      
+      file.sub!(/\.yml/, '') if file =~ /\.yml/
+      
+      built_in_path = File.join( File.expand_path(File.dirname(__FILE__)), '../recipes', file + '.yml')
+
+      where = nil
+      path = nil
+      
+      if File.exists? built_in_path then where = :internal end
+      if File.exists? file then where = :external end
+      
+      raise(ArgumentError, "Recipe file '#{file}' not found") if where.nil?
+      
+      case where
+      when :internal
+        path = built_in_path
+      when :external
+        path = file
+      end
+      
+      return self.new({ :source => :file, :recipe => path })
+    end
+    
+    def self.from_name(name)
+      
+      n = name.to_s
+      
+      raise(ArgumentError, 'Invalid recipe name') if not n.is_a?(String) 
+      
+      return from_file(n)
+      
+    end
+    
+    def self.build(hash)
+      return self.new({ :source => :hash, :recipe => hash})
+    end
+    
+    @recipe = {
+      'format' => :skip
+    }
+    @video = {}
+    @audio = {}
+    @container = nil
+    
+    attr_accessor :audio, :video, :arg_string, :args
+    
+    def initialize(opts)
+      
+      if not opts.is_a? Hash then raise(ArgumentError, "opts should be a hash") end
+        
+      defaults = {
+        :source => :hash,
+        :recipe => nil
+      }
+      
+      @opts = defaults.merge!(opts)
+      
+      @recipe = @opts[:recipe]
+      
+      if @recipe.nil? then raise(ArgumentError, "No recipe given") end
+      
+      case @opts[:source]
+        when :hash
+          hash_build
+        when :file
+          file_build @recipe
+        when :name
+          from_name @recipe
+        else
+          raise(ArgumentError, 'Invalid recipe source')
+      end
+    
+    end
+  
+    def parse o
+      
+      recipe = nil
+      
+      recipe = o['recipe'] if o.has_key? 'recipe'
+      recipe = o if o.has_key? 'video' or o.has_key? 'audio'
+      
+      raise ArgumentError, 'Bad recipe' if recipe.nil?
+      
+      video = recipe['video']
+      audio = recipe['audio']
+      
+      parse_video = lambda do |source_video|
+        
+        @video = {}
+        @video.merge! source_video
+        
+        # Video codec
+        
+        # End video codec
+
+        # Video bitrate
+        if @video['bitrate'].nil? then @video['bitrate'] = :skip else
+          begin
+            @video['bitrate'] = @video['bitrate'].to_i if not @video['bitrate'].is_a? Integer
+          rescue
+            raise ArgumentError, 'Video bitrate must be a number'
+          end
+          
+          @video['bitrate'] *= 1024
+          raise ArgumentError, 'Recipe: video bitrate too low' if @video['bitrate'] < 1024
+        end
+        # End video
+        
+        # Video resolution
+        if @video['resolution'].nil? then @video['resolution'] = :skip else
+          if not @video['resolution'] =~ /\d*x\d*/ then
+            raise ArgumentError, "Invalid resolution"
+          end
+        end
+        # End video resolution
+        
+        
+        
+      end
+      
+      parse_audio = lambda do |source_audio|
+        
+        @audio = {
+          'codec' => nil,
+          'bitrate' => nil,
+        }
+        
+        @audio.merge! source_audio
+        
+        # Audio codec
+        if @audio['codec'].nil? then @audio['codec'] = :skip else
+          raise ArgumentError, 'Recipe: Invalid Audio Codec' unless PureMotion::Codecs.valid? @audio['codec'], :audio
+          @audio['codec'] = PureMotion::Codecs.find(@audio['codec']).ffmpeg_name
+        end
+        # End Audio codec
+        
+        # Audio bitrate
+        if @audio['bitrate'].nil? then @audio['bitrate'] = :skip else
+          begin
+            @audio['bitrate'] = @audio['bitrate'].to_i if not @audio['bitrate'].is_a? Integer
+          rescue
+            raise ArgumentError, 'Audio bitrate must be a number'
+          end
+          
+          @audio['bitrate'] *= 1024
+          raise ArgumentError, 'Recipe: Audio bitrate too low' if @audio['bitrate'] < 1024
+        end
+        # End Audio Bitrate
+        
+        # Audio channels
+        if @audio['channels'].nil? then @audio['channels'] = :skip else
+          channel_types = { 'mono' => 1, 'stereo' => 2 }
+          if @audio['channels'].is_a? String then
+            @audio['channels'].downcase!
+            if channel_types[@audio['channels']].nil? then
+              @audio['channels'] = :skip
+            else
+              @audio['channels'] = channel_types[@audio['channels']]
+            end
+          end
+        end
+        # End audio channels
+        
+      end
+      
+      parse_video.call video unless video.nil?
+      parse_audio.call audio unless audio.nil?
+      
+      @recipe = {}
+      
+      if recipe['format'].nil? then recipe['format'] = :skip else
+        @recipe.store('format', recipe['format'])
+      end
+      
+    end
+  
+    def valid?
+      valid = true
+      if audio? and !valid_audio? then valid = false end
+      if video? and !valid_video? then valid = false end
+      valid
+    end
+  
+    def valid_video?
+      
+    end
+  
+    def valid_audio?
+      
+    end
+  
+    def to_args
+      
+      args = []
+      arg_strings = []
+      
+      args[0] = ['f',       @recipe['format']   ]
+      args[1] = ['acodec',  @audio['codec']     ]
+      args[2] = ['vcodec',  @video['codec']     ]
+      args[3] = ['vb',      @video['bitrate']   ]
+      args[4] = ['s',       @video['resolution']]
+      args[5] = ['r',       @video['fps']       ]
+      args[6] = ['ab',      @audio['bitrate']   ]
+      args[7] = ['ac',      @audio['channels']  ]
+      
+      required_value_args = ['f', 'vcodec', 'acodec', 'vb', 's', 'r', 'ab', 'ac']
+      
+      args.each do |arg|
+        next if required_value_args.include? arg[0] and arg[1].nil?
+        arg_strings << format_arg(arg[0], arg[1])
+      end
+      
+      @args = arg_strings.join(' ')
+      
+      @args
+      
+    end
+  
+    private
+    
+    def format_arg name, value
+      name = '' if name.nil?
+      value = '' if value.nil?
+      value = value.to_s
+      value.gsub! '"', ''
+      value = '"' + value + '"' if value =~ /\s/
+      return value if name == :non
+      return name if value == :non
+      '-' + name.to_s + ' ' + value.to_s
+    end
+    
+    def hash_build
+      
+    end
+    
+    def file_build(file)
+      @yaml = ::YAML.load( File.read(file) )
+      parse @yaml
+    end
+    
+  end
+  
+end

data/lib/puremotion/transcode/transcode.rb

@@ -0,0 +1,153 @@
+module PureMotion::Transcode
+  
+  @@recipes = []
+  
+  def self.do(media, params)
+    return Transcode.new(media, params)
+  end
+  
+  class Transcode < Object
+    
+    event :progress
+    event :complete
+    event :error
+    event :status_change
+    
+    @status
+    @media = nil
+    
+    @ffmpeg = nil
+    
+    def initialize(media, params)
+      
+      @status = Status::NOT_STARTED
+      
+      @params = {
+        :output => nil,
+        :overwrite => false,
+        :recipe => nil
+      }
+      
+      raise ArgumentError, "Invalid paramters given" unless params.is_a? Hash
+      
+      @params.merge!(params)
+      
+      raise ArgumentError, "No output file given" if @params[:output].nil?
+      
+      case @params[:recipe]
+      when String
+        set_recipe Recipe.from_file(@params[:recipe])
+      when Hash
+        set_recipe Recipe.build(@params[:recipe])
+      when Recipe
+        set_recipe @params[:recipe]
+      when Symbol
+        set_recipe Recipe.from_name(@params[:recipe].to_s)
+      else
+        raise ArgumentError, "Invalid recipe"
+      end
+
+      @media = find_media(media)
+      
+      if @media.analysed?
+        prepare
+      else
+        
+        @media.analysed + lambda do |media, raw|
+          prepare
+        end
+        
+      end
+      
+      raise ArgumentError, "Output file #{@params[:output]} exists and overwrite disabled" if File.exists?(@params[:output]) and !@params[:overwrite]
+      
+      @ffmpeg = PureMotion::Tools::FFmpeg.run :options => "-i #{@media.file} #{'-y' if @params[:overwrite]} " + @recipe.to_args + " \"#{@params[:output]}\""
+      
+      @ffmpeg.line + lambda do |ffmpeg, line|
+        handle_output line
+      end
+      
+    end
+    
+    def run
+      
+    end
+    
+    def cancel
+      
+    end
+    
+    def cancel!
+      
+    end
+    
+    private
+    
+    def set_recipe(recipe)
+      @params[:recipe] = recipe
+      @recipe = recipe
+    end
+    
+    def set_status(status)
+      @status = status
+      status_change(@status)
+    end
+  
+    def find_media(media)
+      
+      case media
+      when String
+        return PureMotion::Media.new(media)
+      when PureMotion::Media
+        return media
+      else
+        raise Error, "Invalid media given"
+      end
+      
+    end
+    
+    def handle_output line
+      puts line
+      # /^frame=\s*(?<frame>\d*)\s*fps=\s*(?<fps>\d*)\s*q=(?<q>\S*)\s*size=\s*(?<size>[\d]*)(?<size_unit>[kmg]B)\s*time=\s*(?<time>[\d.]*)\s*bitrate=\s*(?<bitrate>[\d.]*)(?<bitrate_unit>[km]b).*$/
+      progress_line = /frame=\s*(\d*)\s*fps=\s*(\d*)\s*q=(\S*)\s*[L]?size=\s*([\d]*)([kmg]B)\s*time=\s*([\d.]*)\s*bitrate=\s*([\d.]*)([km]b)/
+      # 1 => frame
+      # 2 => fps
+      # 3 => q
+      # 4 => size
+      # 5 => size_unit
+      # 6 => time
+      # 7 => bitrate
+      # 8 => bitrate_unit
+      handle_progress(progress_line.match(line)) if progress_line.match(line)
+    end
+    
+    def handle_progress line
+      p = {
+        :frame    => line[1].to_i,
+        :fps      => line[2].to_i,
+        :q        => line[3].to_f,
+        :size     => line[4].to_i,
+        :time     => line[6].to_f,
+        :bitrate  => line[7].to_f,
+        :precent  => ((100.00 / @media.detail(:duration, :seconds)) * line[6].to_f).to_i
+      }
+      p[:percent] = 100 if p[:percent] > 100
+      progress(p)
+    end
+    
+    # Fire progress event
+    # complete(p)
+  end
+  
+  class Error < Exception
+      
+  
+  end  
+  
+  class Status
+    
+    NOT_STARTED = -1
+    
+  end
+  
+end

data/lib/puremotion.rb

@@ -0,0 +1,22 @@
+req = File.expand_path(File.dirname(__FILE__)) + '/'
+
+require 'rubygems'
+
+module PureMotion
+  
+end
+
+require req + 'puremotion/events/event'
+require req + 'puremotion/events/generator'
+
+Object.extend(PureMotion::Events::Generator)
+
+require req + 'puremotion/thread'
+require req + 'puremotion/media'
+require req + 'puremotion/media/stream'
+require req + 'puremotion/media/stream/base'
+require req + 'puremotion/media/stream/video'
+require req + 'puremotion/codecs'
+require req + 'puremotion/transcode/transcode'
+require req + 'puremotion/transcode/recipe'
+require req + 'puremotion/tools/ffmpeg'

data/puremotion.gemspec

@@ -0,0 +1,68 @@
+# Generated by jeweler
+# DO NOT EDIT THIS FILE DIRECTLY
+# Instead, edit Jeweler::Tasks in Rakefile, and run the gemspec command
+# -*- encoding: utf-8 -*-
+
+Gem::Specification.new do |s|
+  s.name = %q{puremotion}
+  s.version = "0.0.1"
+
+  s.required_rubygems_version = Gem::Requirement.new(">= 0") if s.respond_to? :required_rubygems_version=
+  s.authors = ["Ominiom"]
+  s.date = %q{2009-11-14}
+  s.description = %q{A Ruby wrapper for analysis and conversion of media using FFmpeg}
+  s.email = %q{iain.iw.wilson@googlemail.com}
+  s.extra_rdoc_files = [
+    "LICENSE",
+     "README.rdoc"
+  ]
+  s.files = [
+    ".document",
+     ".gitignore",
+     "LICENSE",
+     "README.rdoc",
+     "Rakefile",
+     "VERSION",
+     "lib/puremotion.rb",
+     "lib/puremotion/codecs.rb",
+     "lib/puremotion/events/event.rb",
+     "lib/puremotion/events/generator.rb",
+     "lib/puremotion/media.rb",
+     "lib/puremotion/media/stream.rb",
+     "lib/puremotion/media/stream/audio.rb",
+     "lib/puremotion/media/stream/base.rb",
+     "lib/puremotion/media/stream/collection.rb",
+     "lib/puremotion/media/stream/video.rb",
+     "lib/puremotion/recipes/ipod.yml",
+     "lib/puremotion/thread.rb",
+     "lib/puremotion/tools/ffmpeg.rb",
+     "lib/puremotion/transcode/recipe.rb",
+     "lib/puremotion/transcode/transcode.rb",
+     "puremotion.gemspec",
+     "test/helper.rb",
+     "test/test_puremotion.rb"
+  ]
+  s.homepage = %q{http://github.com/ominiom/puremotion}
+  s.rdoc_options = ["--charset=UTF-8"]
+  s.require_paths = ["lib"]
+  s.rubygems_version = %q{1.3.5}
+  s.summary = %q{PureMotion}
+  s.test_files = [
+    "test/helper.rb",
+     "test/test_puremotion.rb"
+  ]
+
+  if s.respond_to? :specification_version then
+    current_version = Gem::Specification::CURRENT_SPECIFICATION_VERSION
+    s.specification_version = 3
+
+    if Gem::Version.new(Gem::RubyGemsVersion) >= Gem::Version.new('1.2.0') then
+      s.add_development_dependency(%q<thoughtbot-shoulda>, [">= 0"])
+    else
+      s.add_dependency(%q<thoughtbot-shoulda>, [">= 0"])
+    end
+  else
+    s.add_dependency(%q<thoughtbot-shoulda>, [">= 0"])
+  end
+end
+

data/test/helper.rb

@@ -0,0 +1,10 @@
+require 'rubygems'
+require 'test/unit'
+require 'shoulda'
+
+$LOAD_PATH.unshift(File.join(File.dirname(__FILE__), '..', 'lib'))
+$LOAD_PATH.unshift(File.dirname(__FILE__))
+require 'puremotion'
+
+class Test::Unit::TestCase
+end

data/test/test_puremotion.rb

@@ -0,0 +1,7 @@
+require 'helper'
+
+class TestPuremotion < Test::Unit::TestCase
+  should "probably rename this file and start testing for real" do
+    flunk "hey buddy, you should probably rename this file and start testing for real"
+  end
+end