video_transcoding 0.1.0

data/bin/convert-video

@@ -0,0 +1,290 @@
+#!/usr/bin/env ruby -W
+#
+# convert-video
+#
+# Copyright (c) 2013-2015 Don Melton
+#
+
+$LOAD_PATH.unshift File.expand_path(File.dirname(__FILE__) + '/../lib')
+
+require 'video_transcoding/cli'
+
+module VideoTranscoding
+  class Command
+    include CLI
+
+    def about
+      <<HERE
+convert-video #{VERSION}
+#{COPYRIGHT}
+HERE
+    end
+
+    def usage
+      <<HERE
+Convert video file from Matroska to MP4 format or from MP4 to Matroksa format
+WITHOUT TRANSCODING VIDEO.
+
+Usage: #{$PROGRAM_NAME} [OPTION]... [FILE]...
+
+-o, --output DIRECTORY
+                    set output path
+                      (default: input filename with output format extension
+                        in current working directory)
+    --use-m4v       use `.m4v` extension instead of `.mp4` for MP4 output
+
+-v, --verbose       increase diagnostic information
+-q, --quiet         decrease     "           "
+
+-h, --help          display this help and exit
+    --version       output version information and exit
+
+Requires `HandBrakeCLI`, `mp4track`, `ffmpeg` and `mkvmerge`.
+HERE
+    end
+
+    def initialize
+      super
+      @output   = nil
+      @use_m4v  = false
+    end
+
+    def define_options(opts)
+      opts.on('-o', '--output ARG') { |arg| @output = arg }
+      opts.on('--use-m4v')          { @use_m4v = true }
+    end
+
+    def configure
+      unless @output.nil? or File.directory? @output
+        fail UsageError, "not a directory: #{@output}"
+      end
+
+      HandBrake.setup
+      MP4track.setup
+      FFmpeg.setup
+      MKVmerge.setup
+    end
+
+    def process_input(arg)
+      Console.info "Processing: #{arg}..."
+      media = Media.new(path: arg, allow_directory: false)
+      Console.debug media.info
+      output = resolve_output(media)
+      fail "no H.264 format video track: #{arg}" unless media.info[:h264]
+      fail "no video stream: #{arg}" unless media.info.has_key? :stream
+
+      if media.info[:mkv]
+        convert_to_mp4(media, output)
+      else
+        convert_to_mkv(media, output)
+      end
+    end
+
+    def resolve_output(media)
+      if media.info[:mkv]
+        ext = @use_m4v ? '.m4v' : '.mp4'
+      elsif media.info[:mp4]
+        ext = '.mkv'
+      else
+        fail "unsupported input container format: #{media.path}"
+      end
+
+      output = File.basename(media.path, '.*') + ext
+
+      unless @output.nil?
+        output = File.absolute_path(@output) + File::SEPARATOR + output
+      end
+
+      fail "output file exists: #{media.path}" if File.exist? output
+      output
+    end
+
+    def convert_to_mp4(media, output)
+      map_options = [
+        '-map', "0:#{media.info[:stream]}"
+      ]
+      copy_options = [
+        '-c:v', 'copy'
+      ]
+
+      unless media.info[:audio].empty?
+        track_order = media.info[:audio].keys
+        stream = 0
+
+        if media.info[:audio][1][:channels] > 2.0
+          if track_order.size > 1
+            if  media.info[:audio][1][:language] == media.info[:audio][2][:language] and
+                media.info[:audio][2][:format] == 'AAC' and
+                media.info[:audio][2][:channels] <= 2.0
+              first = track_order[0]
+              track_order[0] = track_order[1]
+              track_order[1] = first
+            end
+          else
+            map_options.concat([
+              '-map', "0:#{media.info[:audio][1][:stream]}"
+            ])
+
+            if FFmpeg.aac_encoder == 'aac'
+              copy_options.concat([
+                '-strict', 'experimental'
+              ])
+            end
+
+            copy_options.concat([
+              '-ac', '2',
+              '-c:a:0', FFmpeg.aac_encoder,
+              '-b:a:0', '160k'
+            ])
+            stream += 1
+          end
+        end
+
+        track_order.each do |track|
+          map_options.concat([
+            '-map', "0:#{media.info[:audio][track][:stream]}"
+          ])
+
+          if  media.info[:audio][track][:channels] > 2.0 and
+              media.info[:audio][track][:format] != 'AC3'
+            copy_options.concat([
+              '-ac', '6',
+              "-c:a:#{stream}", 'ac3',
+              "-b:a:#{stream}", '384k'
+            ])
+          elsif media.info[:audio][track][:channels] <= 2.0 and
+                media.info[:audio][track][:format] != 'AAC' and
+                media.info[:audio][track][:format] != 'AC3'
+            if FFmpeg.aac_encoder == 'aac'
+              copy_options.concat([
+                '-strict', 'experimental'
+              ])
+            end
+
+            copy_options.concat([
+              '-ac', '2',
+              "-c:a:#{stream}", FFmpeg.aac_encoder,
+              "-b:a:#{stream}", '160k'
+            ])
+          else
+            copy_options.concat([
+              "-c:a:#{stream}", 'copy'
+            ])
+          end
+
+          stream += 1
+        end
+      end
+
+      ffmpeg_command = [
+        FFmpeg.command_name,
+        '-hide_banner',
+        '-nostdin',
+        '-i', media.path,
+        *map_options,
+        *copy_options,
+        output
+      ]
+      Console.debug ffmpeg_command
+      Console.info 'Converting with ffmpeg...'
+
+      begin
+        IO.popen(ffmpeg_command, :err=>[:child, :out]) do |io|
+          Signal.trap 'INT' do
+            Process.kill 'INT', io.pid
+          end
+
+          io.each_char do |char|
+            print char
+          end
+        end
+      rescue SystemCallError => e
+        raise "conversion failed: #{e}"
+      end
+
+      fail "conversion failed: #{media.path}" unless $CHILD_STATUS.exitstatus == 0
+      mp4_media = Media.new(path: output, allow_directory: false)
+      Console.debug media.info
+
+      mp4_media.info[:audio].each do |track, info|
+        if track == 1 and not info[:default]
+          enabled = 'true'
+        elsif track != 1 and info[:default]
+          enabled = 'false'
+        else
+          enabled = nil
+        end
+
+        unless enabled.nil?
+          begin
+            IO.popen([
+              MP4track.command_name,
+              '--track-index', track.to_s,
+              '--enabled', enabled,
+              output,
+            ], :err=>[:child, :out]) do |io|
+              io.each do |line|
+                Console.debug line
+              end
+            end
+          rescue SystemCallError => e
+            raise "adjusting audio enabled failed: #{e}"
+          end
+
+          fail "adjusting audio enabled failed: #{output}" unless $CHILD_STATUS.exitstatus == 0
+        end
+      end
+    end
+
+    def convert_to_mkv(media, output)
+      track_order = ['0:' + media.info[:stream].to_s]
+      track_name_options = []
+
+      media.info[:audio].each do |_, info|
+        track_order << '0:' + info[:stream].to_s
+        track_name_options.concat([
+          '--track-name', "#{info[:stream]}:#{info[:name]}"
+        ])
+      end
+
+      if  track_order.size > 2 and
+          media.info[:audio][1][:language] == media.info[:audio][2][:language] and
+          media.info[:audio][1][:format] == 'AAC' and
+          media.info[:audio][1][:channels] <= 2.0 and
+          media.info[:audio][2][:channels] > 2.0
+        first = track_order[1]
+        track_order[1] = track_order[2]
+        track_order[2] = first
+      end
+
+      mkvmerge_command = [
+        MKVmerge.command_name,
+        '--output', output,
+        '--track-order', track_order.join(','),
+        '--disable-track-statistics-tags'
+      ]
+      mkvmerge_command += track_name_options unless track_name_options.empty?
+      mkvmerge_command << media.path
+      Console.debug mkvmerge_command
+      Console.info 'Converting with mkvmerge...'
+
+      begin
+        IO.popen(mkvmerge_command, :err=>[:child, :out]) do |io|
+          Signal.trap 'INT' do
+            Process.kill 'INT', io.pid
+          end
+
+          io.each_char do |char|
+            print char
+          end
+        end
+      rescue SystemCallError => e
+        raise "conversion failed: #{e}"
+      end
+
+      fail "conversion failed: #{media.path}" if $CHILD_STATUS.exitstatus == 2
+    end
+  end
+end
+
+VideoTranscoding::Command.new.run

data/bin/detect-crop

@@ -0,0 +1,130 @@
+#!/usr/bin/env ruby -W
+#
+# detect-crop
+#
+# Copyright (c) 2013-2015 Don Melton
+#
+
+$LOAD_PATH.unshift File.expand_path(File.dirname(__FILE__) + '/../lib')
+
+require 'video_transcoding/cli'
+
+module VideoTranscoding
+  class Command
+    include CLI
+
+    def about
+      <<HERE
+detect-crop #{VERSION}
+#{COPYRIGHT}
+HERE
+    end
+
+    def usage
+      <<HERE
+Detect optimal crop values for video file or disc image directory.
+
+Usage: #{$PROGRAM_NAME} [OPTION]... [FILE|DIRECTORY]...
+
+    --scan          list title(s) and tracks in video media and exit
+    --title NUMBER  select numbered title in video media
+                      (default: main feature or first listed)
+    --no-constrain  don't constrain crop to optimal shape
+    --values-only   output only unambiguous crop values, not commands
+
+-v, --verbose       increase diagnostic information
+-q, --quiet         decrease     "           "
+
+-h, --help          display this help and exit
+    --version       output version information and exit
+
+Requires `HandBrakeCLI` and `mplayer`.
+HERE
+    end
+
+    def initialize
+      super
+      @scan         = false
+      @title        = nil
+      @constrain    = true
+      @values_only  = false
+    end
+
+    def define_options(opts)
+      opts.on('--scan')               { @scan = true }
+      opts.on('--title ARG', Integer) { |arg| @title = arg }
+      opts.on('--no-constrain')       { @constrain = false }
+      opts.on('--values-only')        { @values_only = true }
+    end
+
+    def configure
+      HandBrake.setup
+      MPlayer.setup
+    end
+
+    def process_input(arg)
+      Console.info "Processing: #{arg}..."
+
+      if @scan
+        media = Media.new(path: arg, title: @title)
+        Console.debug media.info
+        puts media.summary
+        return
+      end
+
+      media = Media.new(path: arg, title: @title, autocrop: true, extended: false)
+      Console.debug media.info
+      width, height = media.info[:width], media.info[:height]
+      directory = media.info[:directory]
+      hb_crop = media.info[:autocrop]
+      hb_crop = Crop.constrain(hb_crop, width, height) if @constrain
+      shell_path = arg.shellescape
+
+      print_transcode = ->(crop) do
+        puts
+        print 'transcode-video '
+        print "--title #{media.info[:title]} " if directory
+        puts "--crop #{Crop.handbrake_string(crop)} #{shell_path}"
+        puts
+      end
+
+      print_all = ->(crop) do
+        str = Crop.mplayer_string(crop, width, height)
+        puts
+        puts "mplayer -really-quiet -nosound -vf rectangle=#{str} #{shell_path}"
+        puts "mplayer -really-quiet -nosound -vf crop=#{str} #{shell_path}"
+        print_transcode.call crop
+      end
+
+      if directory
+        if @values_only
+          puts Crop.handbrake_string(hb_crop)
+        else
+          print_transcode.call hb_crop
+        end
+      else
+        mp_crop = Crop.detect(arg, media.info[:duration], width, height)
+        mp_crop = Crop.constrain(mp_crop, width, height) if @constrain
+
+        if hb_crop == mp_crop
+          if @values_only
+            puts Crop.handbrake_string(hb_crop)
+          else
+            Console.info 'Results from HandBrakeCLI and mplayer are identical...'
+            print_all.call hb_crop
+          end
+        else
+          fail "results from HandBrakeCLI and mplayer differ: #{arg}" if @values_only
+          Console.warn 'Results differ...'
+          puts
+          puts '# From HandBrakeCLI:'
+          print_all.call hb_crop
+          puts '# From mplayer:'
+          print_all.call mp_crop
+        end
+      end
+    end
+  end
+end
+
+VideoTranscoding::Command.new.run

data/bin/query-handbrake-log

@@ -0,0 +1,248 @@
+#!/usr/bin/env ruby -W
+#
+# query-handbrake-log
+#
+# Copyright (c) 2013-2015 Don Melton
+#
+
+$LOAD_PATH.unshift File.expand_path(File.dirname(__FILE__) + '/../lib')
+
+require 'abbrev'
+require 'video_transcoding/cli'
+
+module VideoTranscoding
+  class Command
+    include CLI
+
+    def about
+      <<HERE
+query-handbrake-log #{VERSION}
+#{COPYRIGHT}
+HERE
+    end
+
+    def usage
+      <<HERE
+Report information from HandBrake-generated `.log` files.
+
+Usage: #{$PROGRAM_NAME} INFO [OPTION]... [FILE|DIRECTORY]...
+
+Information types:
+t,  time            time spent during transcoding
+                      (sorted from short to long)
+s,  speed           speed of transcoding in frames per second
+                      (sorted from fast to slow)
+b,  bitrate         video bitrate of transcoded output
+                      (sorted from low to high)
+r,  ratefactor      average P-frame quantizer for transcoding
+                      (sorted from low to high)
+
+Options:
+    --reverse       reverse direction of sort
+
+-v, --verbose       increase diagnostic information
+-q, --quiet         decrease     "           "
+
+-h, --help          display this help and exit
+    --version       output version information and exit
+HERE
+    end
+
+    def initialize
+      super
+      @reverse  = false
+      @info     = nil
+      @logs     = []
+      @paths    = []
+    end
+
+    def define_options(opts)
+      opts.on('--reverse') { @reverse = true }
+    end
+
+    def configure
+      fail UsageError, 'missing argument' if ARGV.empty?
+      arg = $ARGV.shift
+
+      @info = case arg
+      when 'time', 't'
+        :time
+      when 'speed', 's'
+        @reverse = !@reverse
+        :speed
+      when 'bitrate', 'b'
+        :bitrate
+      when 'ratefactor', 'r'
+        :ratefactor
+      else
+        fail UsageError, "invalid information type: #{arg}"
+      end
+    end
+
+    def process_input(arg)
+      Console.info "Processing: #{arg}..."
+      input = File.absolute_path(arg)
+
+      if File.directory? input
+        logs = Dir[input + File::SEPARATOR + '*.log']
+        fail "does not contain `.log` files: #{input}" if logs.empty?
+        @logs += logs
+        @paths << input
+      else
+        fail "not a `.log` file: #{input}" unless File.extname(input) == '.log'
+        @logs << File.absolute_path(input)
+        @paths << File.dirname(input)
+      end
+    end
+
+    def complete
+      @logs.uniq!
+      @paths.uniq!
+
+      if @paths.size > 1
+        prefix = File.dirname(@paths.abbrev.keys.min_by { |key| key.size }) + File::SEPARATOR
+      else
+        prefix = ''
+      end
+
+      report = []
+
+      @logs.each do |log|
+        Console.info "Reading: #{log}..."
+        video = File.basename(log, '.log')
+        video += " (#{File.dirname(log).sub(prefix, '')})" unless prefix.empty?
+        found = false
+        count = 0
+        duration_line = nil
+        rate_line = nil
+        fps_line = nil
+        fps_line_2 = nil
+        ratefactor_line = nil
+        bitrate_line = nil
+
+        begin
+          File.foreach(log) do |line|
+            found = true if not found and line =~ /^HandBrake/
+            count += 1
+            fail "not a HandBrake-generated `.log` file: #{log}" if count > 5 and not found
+            duration_line = line if duration_line.nil? and line =~ /\+ duration: /
+            rate_line = line if line =~ /\+ frame rate: /
+
+            if line =~ /average encoding speed/
+              if fps_line.nil?
+                fps_line = line
+              elsif fps_line_2.nil?
+                fps_line_2 = line
+              end
+            end
+
+            ratefactor_line = line if line =~ /x26[45] \[info\]: frame P:/
+            bitrate_line = line if bitrate_line.nil? and line =~ /mux: track 0/
+          end
+        rescue SystemCallError => e
+          raise "reading failed: #{e}"
+        end
+
+        fail "not a HandBrake-generated `.log` file: #{log}" unless found
+
+        case @info
+        when :time, :speed
+          if fps_line.nil?
+            if @info == :time
+              report << "00:00:00 #{video}"
+            else
+              report << "00.000000 fps #{video}"
+            end
+          else
+            Console.debug fps_line
+
+            if fps_line =~ / ([0-9.]+) fps$/
+              fps = $1
+            else
+              fail "fps not found: #{log}"
+            end
+
+            unless fps_line_2.nil?
+              Console.debug fps_line_2
+
+              if fps_line_2 =~ / ([0-9.]+) fps$/
+                fps_2 = $1
+                fps = (1 / ((1 / fps.to_f) + (1 / fps_2.to_f))).round(6).to_s
+              else
+                fail "fps not found: #{log}"
+              end
+            end
+
+            if @info == :time
+              fail "duration not found: #{log}" if duration_line.nil?
+              Console.debug duration_line
+
+              if duration_line =~ /([0-9]{2}):([0-9]{2}):([0-9]{2})/
+                duration = ($1.to_i * 60 * 60) + ($2.to_i * 60) + $3.to_i
+                fail "frame rate not found: #{log}" if rate_line.nil?
+                Console.debug rate_line
+                rate = rate_line.sub(/^.*: /, '').sub(/^.*constant /, '').sub(/ fps.*$/, '').to_f
+                seconds = ((duration * rate) / fps.to_f).to_i
+                hours   = seconds / (60 * 60)
+                minutes = (seconds / 60) % 60
+                seconds = seconds % 60
+                report << format("%02d:%02d:%02d %s", hours, minutes, seconds, video)
+              else
+                fail "duration not found: #{log}"
+              end
+              # TODO
+            else
+              report << "#{fps} fps #{video}"
+            end
+          end
+        when :bitrate
+          if bitrate_line.nil?
+            report << "0000.00 kbps #{video}"
+          else
+            Console.debug bitrate_line
+
+            if bitrate_line =~ /[0-9.]+ kbps/
+              report << "#{$MATCH} #{video}"
+            else
+              fail "bitrate not found: #{log}"
+            end
+          end
+        when :ratefactor
+          if ratefactor_line.nil?
+            report << "00.00 #{video}"
+          else
+            Console.debug ratefactor_line
+
+            if ratefactor_line =~ /Avg QP:([0-9.]+)/
+              report << "#{$1} #{video}"
+            else
+              fail "ratefactor not found: #{log}"
+            end
+          end
+        end
+      end
+
+      if @info == :time
+        report.sort!
+      else
+        report.sort! do |a, b|
+          number_a = a.sub(/ .*$/, '').to_f
+          number_b = b.sub(/ .*$/, '').to_f
+
+          if number_a < number_b
+            -1
+          elsif number_a > number_b
+            1
+          else
+            a <=> b
+          end
+        end
+      end
+
+      report.reverse! if @reverse
+      puts report
+    end
+  end
+end
+
+VideoTranscoding::Command.new.run