smart_ass 0.2.0 → 0.3.0

checksums.yaml

@@ -0,0 +1,7 @@
+---
+SHA1:
+  metadata.gz: 03ac31d6a1df96a33bb982c9606d5bc571c2c378
+  data.tar.gz: 6b629af238f02738dd2bcc5da651fe89faf48b84
+SHA512:
+  metadata.gz: c62f50057fc6f6b3bfdf030a6f7511c7e4e4d0e2d745d515a8fef3152d997025b42440bb7e8631dc397e5922cf02d11fd7b26acfcee6cf1a7159a0905dd53f35
+  data.tar.gz: 139f5889b4507601c6acf28d8200afa20ae6bc5ca7666452e7c45ccf4598d25a83362a7b1c0132677e98c072d0ea266ba3bb83d7330b2ba86174a31154ce7f96

data/.ruby-version

@@ -0,0 +1 @@
+2.0.0-p195

data/README.md

@@ -34,9 +34,13 @@ Install the Ruby Gem:
 
 ## Usage
 
-This is really minimal. No options for now.
+This is really minimal. To process mkvs:
 
-    smart_ass path/to/movie.mkv
+    smart_ass --type mkv path/to/movie.mkv
+
+To process .ass scripts:
+
+    smart_ass --type ass --suffix bt709 path/to/movie.mkv
 
 ## Contributing
 

data/bin/smart_ass

@@ -3,11 +3,24 @@
 $:.unshift File.join(File.dirname(File.dirname(__FILE__)), 'lib')
 
 require 'smart_ass'
+require 'optparse'
+require 'ostruct'
 
-if arg = ARGV[0] then
-  app = SmartAss::App.new(arg)
-  app.run
-else
-  puts "Please provide an input file"
-  exit()
+options = OpenStruct.new
+
+parser = OptionParser.new do |opts|
+  opts.banner = "Usage: smart_ass [options] <file>"
+  opts.on('--type [TYPE]', SmartAss::InputFile.types,
+          'Type of the file you are trying to process') do |type|
+    options.type = type
+  end
+
+  opts.on('--suffix [STRING]', 'String to append to output file name') do |s|
+    options.suffix = s
+  end
 end
+
+parser.parse!
+
+app = SmartAss::App.new(options, ARGV[0])
+app.run

data/lib/smart_ass.rb

@@ -2,7 +2,8 @@ require "smart_ass/version"
 
 module SmartAss
   autoload :App,              'smart_ass/app'
-  autoload :Ass,              'smart_ass/ass'
+  autoload :MKV,              'smart_ass/mkv'
+  autoload :InputFile,        'smart_ass/input_file'
   autoload :BT601ColorMatrix, 'smart_ass/bt601_colormatrix'
   autoload :BT709ColorMatrix, 'smart_ass/bt709_colormatrix'
   autoload :RGBAColor,        'smart_ass/rgba_color'

data/lib/smart_ass/app.rb

@@ -1,22 +1,35 @@
 class SmartAss::App
-  def initialize(file)
-    @ass  = SmartAss::Ass.new(file)
-    @file = Pathname.new(file)
+  def initialize(options, file)
+    if options.type == :ass and not options.suffix
+      raise "Please provide a suffix for the output file!"
+    end
+
+    @ass           = SmartAss::InputFile.from_type(options.type, file)
+    @output_suffix = options.suffix
+    @file          = Pathname.new(file)
   end
 
   def run
     # function composition baby!
-    write_ass_script replace_colors extract track
+    write_ass_script replace_colors @ass.stream
   end
 
   private
   def write_ass_script(ass_script)
-    script_file_basename = "#{@file.basename(@file.extname)}.ass"
-    File.open(@file.expand_path.dirname + script_file_basename, "w") {|f|
+    File.open(@file.expand_path.dirname + output_file_basename, "w") {|f|
       f.puts(ass_script)
     }
   end
 
+  def output_file_basename
+    name = [
+      @file.basename(@file.extname),
+      @output_suffix
+    ].select{|a| a}.join('-')
+
+    "#{name}.ass"
+  end
+
   def replace_colors(ass_script)
     ass_script.gsub(/&H[\da-zA-Z]{6,8}/) {|c| convert_color(c)}
   end
@@ -41,27 +54,4 @@ class SmartAss::App
 
     @_colors_cache[c]
   end
-
-  def track
-    if @ass.tracks.size > 1
-      track = ask_for_track
-    elsif @ass.tracks.size == 1
-      track = @ass.tracks[0]
-    else
-      puts "No ASS tracks found. Aborting."
-      exit
-    end
-  end
-
-  def ask_for_track
-    puts "Please enter a track name - #{@ass.tracks.inspect}:"
-    readline.to_i
-  rescue
-    puts "Invalid track provided. Try again."
-    retry
-  end
-
-  def extract(track_id)
-    @ass.extract(track_id)
-  end
 end

data/lib/smart_ass/input_file.rb

@@ -0,0 +1,32 @@
+require 'pathname'
+
+class SmartAss::InputFile
+  attr_reader :file
+
+  def initialize(file)
+    @file = Pathname.new(file)
+  end
+
+  def stream
+    IO.read(@file)
+  end
+
+  def self.types
+    [:mkv, :ass]
+  end
+
+  def self.from_type(type, file)
+    klass(type).new(file)
+  end
+
+  def self.klass(kind)
+    case kind
+    when :mkv
+      SmartAss::MKV
+    when :ass
+      SmartAss::InputFile
+    else
+      raise "Unrecognized kind=#{kind}"
+    end
+  end
+end

data/lib/smart_ass/{ass.rb → mkv.rb}

@@ -1,27 +1,41 @@
-require 'pathname'
-
-class SmartAss::Ass
-  attr_reader :file
+class SmartAss::MKV < SmartAss::InputFile
+  def stream
+    extract track
+  end
 
-  def initialize(file)
-    @file = Pathname.new(file)
+  def track
+    if tracks.size > 1
+      track = ask_for_track
+    elsif tracks.size == 1
+      track = tracks[0]
+    else
+      puts "No ASS tracks found. Aborting."
+      exit
+    end
   end
 
   def tracks
     identify.each_line \
       .map    {|l| (l =~ identify_ass_regexp) ? $1 : nil } \
       .compact \
-      .map    &:to_i
+      .map(&:to_i)
+  end
+
+  private
+  def ask_for_track
+    puts "Please enter a track name - #{tracks.inspect}:"
+    readline.to_i
+  rescue
+    puts "Invalid track provided. Try again."
+    retry
   end
 
-  # TODO: test this!
   def extract(track_id)
     puts `mkvextract tracks "#{expanded_file}" #{track_id}:#{export_temp_file}`
     IO.read(export_temp_file, :encoding => 'UTF-8') \
       .tap {|_| File.delete(export_temp_file) }
   end
 
-  private
   def identify
     `mkvmerge -i "#{expanded_file}"`
   end

data/lib/smart_ass/version.rb

@@ -1,3 +1,3 @@
 module SmartAss
-  VERSION = "0.2.0"
+  VERSION = "0.3.0"
 end

data/spec/{ass_spec.rb → mkv_spec.rb}

@@ -1,9 +1,9 @@
 require "spec_helper"
 
-describe SmartAss::Ass do
+describe SmartAss::MKV do
   subject {
     @file = "path/to/file/movie.mkv"
-    SmartAss::Ass.new(@file)
+    SmartAss::MKV.new(@file)
   }
 
   it "holds onto a file" do

metadata

@@ -1,15 +1,14 @@
 --- !ruby/object:Gem::Specification
 name: smart_ass
 version: !ruby/object:Gem::Version
-  version: 0.2.0
-  prerelease: 
+  version: 0.3.0
 platform: ruby
 authors:
 - Stefano Pigozzi
 autorequire: 
 bindir: bin
 cert_chain: []
-date: 2012-11-06 00:00:00.000000000 Z
+date: 2013-06-27 00:00:00.000000000 Z
 dependencies: []
 description: Utility for YCbCr ColorMatrix conversion of ASS Subtitles.
 email:
@@ -21,6 +20,7 @@ extra_rdoc_files: []
 files:
 - .gitignore
 - .rspec
+- .ruby-version
 - Gemfile
 - LICENSE
 - README.md
@@ -28,50 +28,50 @@ files:
 - bin/smart_ass
 - lib/smart_ass.rb
 - lib/smart_ass/app.rb
-- lib/smart_ass/ass.rb
 - lib/smart_ass/bt601_colormatrix.rb
 - lib/smart_ass/bt709_colormatrix.rb
+- lib/smart_ass/input_file.rb
+- lib/smart_ass/mkv.rb
 - lib/smart_ass/rgba_color.rb
 - lib/smart_ass/version.rb
 - lib/smart_ass/ycbcr_colormatrix.rb
 - lib/smart_ass/ycbcr_converter.rb
 - smart_ass.gemspec
-- spec/ass_spec.rb
 - spec/fixtures/info_0
 - spec/fixtures/info_1_t_3
 - spec/fixtures/info_3_t_3_5_6
+- spec/mkv_spec.rb
 - spec/rgba_color_spec.rb
 - spec/spec_helper.rb
 - spec/ycbcr_converter_spec.rb
 homepage: http://github.com/pigoz/smart_ass
 licenses: []
+metadata: {}
 post_install_message: 
 rdoc_options: []
 require_paths:
 - lib
 required_ruby_version: !ruby/object:Gem::Requirement
-  none: false
   requirements:
-  - - ! '>='
+  - - '>='
     - !ruby/object:Gem::Version
       version: '0'
 required_rubygems_version: !ruby/object:Gem::Requirement
-  none: false
   requirements:
-  - - ! '>='
+  - - '>='
     - !ruby/object:Gem::Version
       version: '0'
 requirements: []
 rubyforge_project: 
-rubygems_version: 1.8.23
+rubygems_version: 2.0.2
 signing_key: 
-specification_version: 3
+specification_version: 4
 summary: Utility for YCbCr ColorMatrix conversion of ASS Subtitles.
 test_files:
-- spec/ass_spec.rb
 - spec/fixtures/info_0
 - spec/fixtures/info_1_t_3
 - spec/fixtures/info_3_t_3_5_6
+- spec/mkv_spec.rb
 - spec/rgba_color_spec.rb
 - spec/spec_helper.rb
 - spec/ycbcr_converter_spec.rb