encoder-tools 0.0.2

data/.document

@@ -0,0 +1,5 @@
+README.rdoc
+lib/**/*.rb
+bin/*
+features/**/*.feature
+LICENSE

data/.gitignore

@@ -0,0 +1,23 @@
+## MAC OS
+.DS_Store
+
+## TEXTMATE
+*.tmproj
+tmtags
+
+## EMACS
+*~
+\#*
+.\#*
+
+## VIM
+*.swp
+
+## PROJECT::GENERAL
+coverage
+rdoc
+pkg
+
+## PROJECT::SPECIFIC
+.bundle
+encoder-tools.gemspec

data/Gemfile

@@ -0,0 +1,12 @@
+source "http://gemcutter.org"
+
+gem "thor"
+
+group :test do
+  gem "jeweler"
+  gem "rake"
+  gem "yard"
+  gem "rspec", "~>1.3.0"
+  gem "bundler", "~>0.9.4"
+  gem "ruby-debug"
+end

data/Gemfile.lock

@@ -0,0 +1,83 @@
+--- 
+dependencies: 
+  ruby-debug: 
+    group: 
+    - :test
+    version: ">= 0"
+    require: 
+    - ruby-debug
+  rake: 
+    group: 
+    - :test
+    version: ">= 0"
+    require: 
+    - rake
+  thor: 
+    group: 
+    - :default
+    version: ">= 0"
+    require: 
+    - thor
+  rspec: 
+    group: 
+    - :test
+    version: ~> 1.3.0
+    require: 
+    - rspec
+  yard: 
+    group: 
+    - :test
+    version: ">= 0"
+    require: 
+    - yard
+  bundler: 
+    group: 
+    - :test
+    version: ~> 0.9.4
+    require: 
+    - bundler
+  jeweler: 
+    group: 
+    - :test
+    version: ">= 0"
+    require: 
+    - jeweler
+specs: 
+- json_pure: 
+    version: 1.2.0
+- bundler: 
+    version: 0.9.4
+- git: 
+    version: 1.2.5
+- linecache: 
+    version: "0.43"
+- ruby-debug-base: 
+    version: 0.10.3
+- json: 
+    version: 1.1.9
+- net-ssh: 
+    version: 2.0.15
+- net-scp: 
+    version: 1.0.1
+- gemcutter: 
+    version: 0.1.3
+- rspec: 
+    version: 1.3.0
+- yard: 
+    version: 0.5.3
+- rubyforge: 
+    version: 2.0.3
+- jeweler: 
+    version: 1.4.0
+- thor: 
+    version: 0.13.1
+- rake: 
+    version: 0.8.7
+- columnize: 
+    version: 0.3.1
+- ruby-debug: 
+    version: 0.10.3
+hash: 075129d1fa532f8d303254e1ef70185e252d1ce4
+sources: 
+- Rubygems: 
+    uri: http://gemcutter.org

data/LICENSE

@@ -0,0 +1,20 @@
+Copyright (c) 2009 Brian Donovan
+
+Permission is hereby granted, free of charge, to any person obtaining
+a copy of this software and associated documentation files (the
+"Software"), to deal in the Software without restriction, including
+without limitation the rights to use, copy, modify, merge, publish,
+distribute, sublicense, and/or sell copies of the Software, and to
+permit persons to whom the Software is furnished to do so, subject to
+the following conditions:
+
+The above copyright notice and this permission notice shall be
+included in all copies or substantial portions of the Software.
+
+THE SOFTWARE IS PROVIDED "AS IS", WITHOUT WARRANTY OF ANY KIND,
+EXPRESS OR IMPLIED, INCLUDING BUT NOT LIMITED TO THE WARRANTIES OF
+MERCHANTABILITY, FITNESS FOR A PARTICULAR PURPOSE AND
+NONINFRINGEMENT. IN NO EVENT SHALL THE AUTHORS OR COPYRIGHT HOLDERS BE
+LIABLE FOR ANY CLAIM, DAMAGES OR OTHER LIABILITY, WHETHER IN AN ACTION
+OF CONTRACT, TORT OR OTHERWISE, ARISING FROM, OUT OF OR IN CONNECTION
+WITH THE SOFTWARE OR THE USE OR OTHER DEALINGS IN THE SOFTWARE.

data/README.rdoc

@@ -0,0 +1,17 @@
+= encoder-tools
+
+Description goes here.
+
+== Note on Patches/Pull Requests
+ 
+* Fork the project.
+* Make your feature addition or bug fix.
+* Add tests for it. This is important so I don't break it in a
+  future version unintentionally.
+* Commit, do not mess with rakefile, version, or history.
+  (if you want to have your own version, that is fine but bump version in a commit by itself I can ignore when I pull)
+* Send me a pull request. Bonus points for topic branches.
+
+== Copyright
+
+Copyright (c) 2010 Brian Donovan. See LICENSE for details.

data/Rakefile

@@ -0,0 +1,46 @@
+require 'rubygems'
+require 'rake'
+
+begin
+  require 'jeweler'
+  Jeweler::Tasks.new do |gem|
+    gem.name = "encoder-tools"
+    gem.summary = %Q{Some tools to make encoding from DVDs easier}
+    gem.description = %Q{Tools for ripping, encoding, and subtitling movies and TV shows}
+    gem.email = "brian.donovan@gmail.com"
+    gem.homepage = "http://github.com/eventualbuddha/encoder-tools"
+    gem.authors = ["Brian Donovan"]
+    gem.add_dependency "thor", ">= 0"
+    gem.add_development_dependency "rspec", ">= 1.2.9"
+    gem.add_development_dependency "yard", ">= 0"
+    # gem is a Gem::Specification... see http://www.rubygems.org/read/chapter/20 for additional settings
+  end
+  Jeweler::GemcutterTasks.new
+rescue LoadError
+  puts "Jeweler (or a dependency) not available. Install it with: gem install jeweler"
+end
+
+require 'spec/rake/spectask'
+Spec::Rake::SpecTask.new(:spec) do |spec|
+  spec.libs << 'lib' << 'spec'
+  spec.spec_files = FileList['spec/**/*_spec.rb']
+end
+
+Spec::Rake::SpecTask.new(:rcov) do |spec|
+  spec.libs << 'lib' << 'spec'
+  spec.pattern = 'spec/**/*_spec.rb'
+  spec.rcov = true
+end
+
+task :spec => :check_dependencies
+
+task :default => :spec
+
+begin
+  require 'yard'
+  YARD::Rake::YardocTask.new
+rescue LoadError
+  task :yardoc do
+    abort "YARD is not available. In order to run yardoc, you must: sudo gem install yard"
+  end
+end

data/VERSION

@@ -0,0 +1 @@
+0.0.2

data/bin/encoder-tools

@@ -0,0 +1,4 @@
+#!/usr/bin/env ruby
+$LOAD_PATH.unshift(File.expand_path('../../lib', __FILE__))
+require 'encoder-tools'
+EncoderTools::CLI.start

data/lib/encoder-tools/cli/base.rb

@@ -0,0 +1,40 @@
+module EncoderTools
+  class CLI
+    class Base
+      attr_reader :shell, :options
+
+      def initialize(shell, options={})
+        @shell, @options = shell, options
+      end
+
+      def run
+        # overridden in subclasses
+      end
+
+      private
+        def input
+          @input ||= @options[:input] ?
+            open(@options[:input]) :
+            $stdin
+        end
+
+        def output
+          @output ||= @options[:output] ?
+            open(@options[:output], 'w') :
+            $stdout
+        end
+
+        def open(stream_or_file, mode='r')
+          if stream_or_file.respond_to?(:eof?)
+            stream_or_file
+          else
+            File.open(stream_or_file, mode)
+          end
+        end
+
+      def self.run(shell, options={})
+        new(shell, options).run
+      end
+    end
+  end
+end

data/lib/encoder-tools/cli/subtitles/base.rb

@@ -0,0 +1,15 @@
+module EncoderTools
+  class CLI
+    module Subtitles
+      class Base < CLI::Base
+        def parse(text)
+          EncoderTools::Subtitles::List.load(text)
+        end
+
+        def parse_relaxed(text)
+          EncoderTools::Subtitles::List.load(text, EncoderTools::Subtitles::RelaxedParser)
+        end
+      end
+    end
+  end
+end

data/lib/encoder-tools/cli/subtitles/fix_lengths.rb

@@ -0,0 +1,46 @@
+module EncoderTools
+  class CLI
+    module Subtitles
+      class FixLengths < Base
+        THRESHOLD = 5
+
+        def run
+          if long_subtitles.empty?
+            shell.say "No subtitles found over #{THRESHOLD}s"
+            return
+          end
+
+          if not shell.yes?("Found #{long_subtitles.size} long subtitles. Would you like to fix them?")
+            return
+          end
+
+          long_subtitles.each do |subtitle|
+            lines = subtitle.to_s.to_a
+            range = lines.shift.chomp
+            range += " (#{subtitle.duration.to_i}s)\n"
+            lines.unshift(range)
+            lines << "\n" << "\n"
+
+            shell.say(lines.join)
+            subtitle.duration = shell.ask("How long should it be?").to_i
+          end
+
+          output << list.to_s
+
+          return nil
+        end
+
+        private
+          def long_subtitles
+            @long_subtitles ||= list.entries.select do |subtitle|
+              subtitle.duration > THRESHOLD
+            end
+          end
+
+          def list
+            @list ||= parse(input.read)
+          end
+      end
+    end
+  end
+end

data/lib/encoder-tools/cli/subtitles/offset.rb

@@ -0,0 +1,30 @@
+module EncoderTools
+  class CLI
+    module Subtitles
+      class Offset < Base
+        def run
+          output << offset(parse(input.read), options[:offset]).to_s
+        end
+
+        protected
+          def offset(list, offset)
+            list.offset = parse_offset(list, offset)
+            return list
+          end
+
+          def parse_offset(list, offset)
+            case offset
+            when Fixnum
+              offset
+            when /^\+(\d+)$/
+              list.offset + $1.to_i
+            when /^-(\d+)$/
+              list.offset - $1.to_i
+            when /^\d+$/
+              offset.to_i
+            end
+          end
+      end
+    end
+  end
+end

data/lib/encoder-tools/cli/subtitles/renumber.rb

@@ -0,0 +1,11 @@
+module EncoderTools
+  class CLI
+    module Subtitles
+      class Renumber < Base
+        def run
+          output << parse_relaxed(input.read).to_s
+        end
+      end
+    end
+  end
+end

data/lib/encoder-tools/cli/subtitles.rb

@@ -0,0 +1,10 @@
+module EncoderTools
+  class CLI
+    module Subtitles
+      autoload :Base,       'encoder-tools/cli/subtitles/base'
+      autoload :FixLengths, 'encoder-tools/cli/subtitles/fix_lengths'
+      autoload :Offset,     'encoder-tools/cli/subtitles/offset'
+      autoload :Renumber,   'encoder-tools/cli/subtitles/renumber'
+    end
+  end
+end

data/lib/encoder-tools/cli.rb

@@ -0,0 +1,30 @@
+require 'thor'
+require 'thor/group'
+
+module EncoderTools
+  class CLI < Thor
+    autoload :Base,      'encoder-tools/cli/base'
+    autoload :Subtitles, 'encoder-tools/cli/subtitles'
+
+    desc "renumber [--input FILE] [--output FILE]", "Renumber badly-numbered SRT subtitle text"
+    method_option :input,  :type => :string, :required => false, :aliases => %w[-i]
+    method_option :output, :type => :string, :required => false, :aliases => %w[-o]
+    def renumber
+      CLI::Subtitles::Renumber.run(self, options)
+    end
+
+    desc "offset [--input FILE] [--output FILE] [+-]OFFSET", "Change the SRT subtitle offset to OFFSET or by +/-OFFSET"
+    method_option :input,  :type => :string, :required => false, :aliases => %w[-i]
+    method_option :output, :type => :string, :required => false, :aliases => %w[-o]
+    def offset(offset)
+      CLI::Subtitles::Offset.run(self, options.merge(:offset => offset))
+    end
+
+    desc "fix-lengths --input FILE --output FILE", "Interactively fix subtitle lengths over a #{CLI::Subtitles::FixLengths::THRESHOLD}s threshold"
+    method_option :input,  :type => :string, :required => true, :aliases => %w[-i]
+    method_option :output, :type => :string, :required => true, :aliases => %w[-o]
+    def fix_lengths
+      CLI::Subtitles::FixLengths.run(self, options)
+    end
+  end
+end

data/lib/encoder-tools/options/title.rb

@@ -0,0 +1,34 @@
+require 'singleton'
+
+module EncoderTools
+  module Options
+    class Title
+      attr_accessor :number
+
+      def initialize(number)
+        @number = number
+      end
+
+      def ==(other)
+        other.is_a?(self.class) && other.number == self.number
+      end
+
+      def to_args
+        ['--title', number.to_s]
+      end
+
+      class Longest < Title
+        include Singleton
+
+        def initialize
+          super(nil)
+        end
+
+        def to_args
+          %w[--longest]
+        end
+      end
+      LONGEST = Longest.instance
+    end
+  end
+end

data/lib/encoder-tools/options.rb

@@ -0,0 +1,5 @@
+module EncoderTools
+  module Options
+    autoload :Title, 'encoder-tools/options/title'
+  end
+end

data/lib/encoder-tools/strategies/base.rb

@@ -0,0 +1,22 @@
+module EncoderTools
+  module Strategies
+    class Base
+      attr_reader :input_path, :title
+
+      def initialize(input_path)
+        @input_path = input_path
+      end
+
+      def title=(title)
+        case title
+        when Options::Title, nil
+          @title = title
+        when Fixnum
+          @title = Options::Title.new(title)
+        else
+          raise ArgumentError, "expected an #{Options::Title} or #{Fixnum}, got #{title.inspect}"
+        end
+      end
+    end
+  end
+end

data/lib/encoder-tools/strategies/movie.rb

@@ -0,0 +1,10 @@
+module EncoderTools
+  module Strategies
+    class Movie < Base
+      def initialize(input_path)
+        super
+        self.title = Options::Title::LONGEST
+      end
+    end
+  end
+end

data/lib/encoder-tools/strategies/tv.rb

@@ -0,0 +1,10 @@
+module EncoderTools
+  module Strategies
+    class TV < Base
+      def initialize(input_path)
+        super
+        self.title = 1
+      end
+    end
+  end
+end

data/lib/encoder-tools/strategies.rb

@@ -0,0 +1,7 @@
+module EncoderTools
+  module Strategies
+    autoload :Base,  'encoder-tools/strategies/base'
+    autoload :Movie, 'encoder-tools/strategies/movie'
+    autoload :TV,    'encoder-tools/strategies/tv'
+  end
+end

data/lib/encoder-tools/subtitles/list.rb

@@ -0,0 +1,35 @@
+module EncoderTools
+  module Subtitles
+    class List
+      attr_accessor :entries
+
+      def offset
+        return 0 if entries.empty?
+        entries.first.offset
+      end
+
+      def offset=(offset)
+        return nil if entries.empty?
+        diff = offset - self.offset
+        entries.each {|subtitle| subtitle.offset += diff}
+      end
+
+      def to_s
+        str = ""
+        i = 0
+        entries.each do |subtitle|
+          str << (i += 1).to_s << "\n" << subtitle.to_s << "\n\n"
+        end
+
+        return str
+      end
+
+      def self.load(input, parser_class=Parser)
+        result = new
+        parser = parser_class.new(input)
+        result.entries = parser.parse
+        return result
+      end
+    end
+  end
+end

data/lib/encoder-tools/subtitles/parser.rb

@@ -0,0 +1,96 @@
+require 'strscan'
+require 'bigdecimal'
+
+module EncoderTools
+  module Subtitles
+    class Parser
+      class ParseError < RuntimeError; end
+
+      def initialize(input)
+        @scanner = StringScanner.new(input)
+        @last_index = 0
+      end
+
+      def parse
+        result = []
+        result << scan_subtitle until @scanner.eos?
+        return result
+      end
+
+      protected
+      def scan_subtitle
+        index = scan_index
+        range = scan_timestamp_range
+        text  = scan_text
+        return Subtitle.new(range, text)
+      end
+
+      def scan_index
+        # 1\n
+        index = string @last_index += 1
+        newline
+        return index
+      end
+
+      def scan_timestamp_range
+        # 01:15:18,000 --> 01:15:20,300\n
+        rstart = timestamp
+        string ' --> '
+        rend = timestamp
+        newline
+        return rstart..rend
+      end
+
+      def scan_text
+        # No wonder you can't do it... you acquiesce\n
+        # to defeat... before you even begin.\n
+        # \n
+        text = ''
+        loop do
+          l = line
+          break if l.nil? || l.strip.empty?
+          text << l
+        end
+        text.strip!
+
+        return text
+      end
+
+      def string(str)
+        scan /#{Regexp.escape(str.to_s)}/
+      end
+
+      def timestamp
+        hours = scan /\d\d/, 'hours'
+        string ':'
+        minutes = scan /\d\d/, 'minutes'
+        string ':'
+        seconds = scan /\d\d/, 'seconds'
+        string ','
+        millis = scan /\d\d\d/, 'milliseconds'
+
+        return hours.to_i * 3600 + minutes.to_i * 60 + seconds.to_i + (BigDecimal(millis) / 1000)
+      end
+
+      def newline
+        scan /\r?\n/, 'newline'
+      end
+
+      def line
+        @scanner.scan_until(/\r?\n/)
+      end
+
+      def scan(scannable, name=nil)
+        if scanned = @scanner.scan(scannable)
+          return scanned
+        else
+          raise ParseError, "expected #{name || scannable.inspect} at character #{@scanner.pos}, got #{@scanner.string[@scanner.pos, 10].inspect}"
+        end
+      end
+
+      def upto(scannable)
+        @scanner.scan_until(scannable)
+      end
+    end
+  end
+end

data/lib/encoder-tools/subtitles/relaxed_parser.rb

@@ -0,0 +1,11 @@
+module EncoderTools
+  module Subtitles
+    class RelaxedParser < Parser
+      protected
+      def scan_index
+        scan /\d+/
+        newline
+      end
+    end
+  end
+end