aviglitch 0.1.3 → 0.2.0

data/lib/aviglitch.rb

@@ -1,12 +1,10 @@
 require 'tempfile'
-require 'fileutils'
-require 'readline'
 require 'pathname'
 require 'stringio'
+require 'aviglitch/avi'
 require 'aviglitch/base'
 require 'aviglitch/frame'
 require 'aviglitch/frames'
-require 'aviglitch/tempfile'
 
 # AviGlitch provides the ways to glitch AVI formatted video files.
 #
@@ -15,7 +13,7 @@ require 'aviglitch/tempfile'
 # You can manipulate each frame, like:
 #
 #   avi = AviGlitch.open '/path/to/your.avi'
-#   avi.frames.each |frame|
+#   avi.frames.each do |frame|
 #     if frame.is_keyframe?
 #       frame.data = frame.data.gsub(/\d/, '0')
 #     end
@@ -30,18 +28,17 @@ require 'aviglitch/tempfile'
 #   end
 #   avi.output '/path/to/broken.avi'
 #
-#--
-# It does not support AVI2, interleave format.
-#
 module AviGlitch
 
-  VERSION = '0.1.3'
+  VERSION = '0.2.0'
+
+  BUFFER_SIZE = 2 ** 24
 
   class << self
     ##
     # Returns AviGlitch::Base instance.
     # It requires +path_or_frames+ as String or Pathname, or Frames instance.
-    def AviGlitch.open path_or_frames
+    def open path_or_frames
       if path_or_frames.kind_of?(Frames)
         path_or_frames.to_avi
       else

data/spec/avi2_spec.rb

@@ -0,0 +1,40 @@
+require File.expand_path(File.dirname(__FILE__) + '/spec_helper')
+
+describe AviGlitch, 'AVI2.0' do
+
+  it 'should save same file when nothing has changed' do
+    avi = AviGlitch.open @in2
+    avi.glitch do |d|
+      d
+    end
+    avi.output @out
+    FileUtils.cmp(@in2, @out).should be true
+  end
+
+  it 'should be AVI1.0 when its size has reduced less than 1GB' do
+    a = AviGlitch.open @in2
+    size = 0
+    a.glitch do |d|
+      size += d.size
+      size < 1024 ** 3 ? d : nil
+    end
+    a.output @out
+    b = AviGlitch.open @out
+    b.avi.was_avi2?.should be false
+    b.close
+  end
+
+  it 'should be AVI2.0 when its size has increased over 1GB' do
+    a = AviGlitch.open @in
+    n = Math.log(1024.0 ** 3 / a.frames.data_size.to_f, 2).ceil
+    f = a.frames[0..-1]
+    n.times do
+      fx = f[0..-1]
+      f.concat fx
+    end
+    f.to_avi.output @out
+    b = AviGlitch.open @out
+    b.avi.was_avi2?.should be true
+    b.close
+  end
+end

data/spec/aviglitch_spec.rb

@@ -2,24 +2,10 @@ require File.expand_path(File.dirname(__FILE__) + '/spec_helper')
 
 describe AviGlitch do
 
-  before :all do
-    FileUtils.mkdir OUTPUT_DIR unless File.exist? OUTPUT_DIR
-    @in = FILES_DIR + 'sample.avi'
-    @out = OUTPUT_DIR + 'out.avi'
-  end
-
-  after :each do
-    FileUtils.rm Dir.glob((OUTPUT_DIR + '*').to_s)
-  end
-
-  after :all do
-    FileUtils.rmdir OUTPUT_DIR
-  end
-
   it 'should raise an error against unsupported files' do
     lambda {
       avi = AviGlitch.open __FILE__
-    }.should raise_error
+    }.should raise_error(RuntimeError)
   end
 
   it 'should return AviGlitch::Base object through the method #open' do
@@ -52,13 +38,24 @@ describe AviGlitch do
 
   it 'can glitch each keyframe with index' do
     avi = AviGlitch.open @in
+
+    a_size = 0
+    avi.glitch :keyframe do |f|
+      a_size += 1
+      f
+    end
+
+    b_size = 0
     avi.glitch_with_index :keyframe do |kf, idx|
+      b_size += 1
       if idx < 25
         kf.slice(10..kf.size)
       else
         kf
       end
     end
+    expect(a_size).to be == b_size
+
     avi.output @out
     i_size = File.stat(@in).size
     o_size = File.stat(@out).size
@@ -143,9 +140,9 @@ describe AviGlitch do
     AviGlitch::Base.surely_formatted?(@out, true).should be true
   end
 
-  it 'should clear keyframes with one method' do
+  it 'should mutate keyframes into deltaframes' do
     a = AviGlitch.open @in
-    a.clear_keyframes!
+    a.mutate_keyframes_into_deltaframes!
     a.output @out
     a = AviGlitch.open @out
     a.frames.each do |f|
@@ -153,7 +150,7 @@ describe AviGlitch do
     end
 
     a = AviGlitch.open @in
-    a.clear_keyframes! 0..50
+    a.mutate_keyframes_into_deltaframes! 0..50
     a.output @out
     a = AviGlitch.open @out
     a.frames.each_with_index do |f, i|
@@ -172,4 +169,26 @@ describe AviGlitch do
     a.has_keyframe?.should be false
   end
 
+  it 'should #remove_all_keyframes!' do
+    a = AviGlitch.open @in
+    a.has_keyframe?.should be true
+    a.remove_all_keyframes!
+    a.has_keyframe?.should be false
+  end
+
+  it 'should count same number of specific frames' do
+    a = AviGlitch.open @in
+    dc1 = 0
+    dc2 = 0
+    a.frames.each do |f|
+      dc1 += 1 if f.is_deltaframe?
+    end
+    a.glitch(:deltaframe) do |d|
+      dc2 += 1
+      d
+    end
+
+    expect(dc1).to eq(dc2)
+  end
+  
 end

data/spec/datamosh_spec.rb

@@ -3,23 +3,12 @@ require File.expand_path(File.dirname(__FILE__) + '/spec_helper')
 describe AviGlitch, 'datamosh cli' do
 
   before :all do
-    FileUtils.mkdir OUTPUT_DIR unless File.exist? OUTPUT_DIR
-    @in = FILES_DIR + 'sample.avi'
-    @out = OUTPUT_DIR + 'out.avi'
     here = File.dirname(__FILE__)
     lib = Pathname.new(File.join(here, '..', 'lib')).realpath
     datamosh = Pathname.new(File.join(here, '..', 'bin/datamosh')).realpath
     @cmd = "ruby -I%s %s -o %s " % [lib, datamosh, @out]
   end
 
-  after :each do
-    FileUtils.rm Dir.glob((OUTPUT_DIR + '*').to_s)
-  end
-
-  after :all do
-    FileUtils.rmdir OUTPUT_DIR
-  end
-
   it 'should correctly process files' do
     a = AviGlitch.open @in
     keys = a.frames.inject(0) do |c, f|
@@ -27,12 +16,13 @@ describe AviGlitch, 'datamosh cli' do
       c
     end
     total = a.frames.size
+    first_keyframe = a.frames.index(a.frames.first_of(:keyframe))
     a.close
 
     system [@cmd, @in].join(' ')
     o = AviGlitch.open @out
     o.frames.size.should == total
-    o.frames.first.is_keyframe?.should be true
+    o.frames[first_keyframe].is_keyframe?.should be true
     o.has_keyframe?.should be true
     o.close
     AviGlitch::Base.surely_formatted?(@out, true).should be true
@@ -40,7 +30,7 @@ describe AviGlitch, 'datamosh cli' do
     system [@cmd, '-a', @in].join(' ')
     o = AviGlitch.open @out
     o.frames.size.should == total
-    o.frames.first.is_keyframe?.should be false
+    o.frames[first_keyframe].is_keyframe?.should be false
     o.has_keyframe?.should be false
     o.close
     AviGlitch::Base.surely_formatted?(@out, true).should be true
@@ -48,7 +38,7 @@ describe AviGlitch, 'datamosh cli' do
     system [@cmd, @in, @in, @in].join(' ')
     o = AviGlitch.open @out
     o.frames.size.should == total * 3
-    o.frames.first.is_keyframe?.should be true
+    o.frames[first_keyframe].is_keyframe?.should be true
     o.close
     AviGlitch::Base.surely_formatted?(@out, true).should be true
 

data/spec/frames_spec.rb

@@ -5,26 +5,21 @@ describe AviGlitch::Frames do
   before :all do
     AviGlitch::Frames.class_eval do
       define_method(:get_real_id_with) do |frame|
-        pos = @io.pos
-        @io.pos -= frame.data.size
-        @io.pos -= 8
-        id = @io.read 4
-        @io.pos = pos
+        movi = @avi.get_movi
+        pos = movi.pos
+        movi.pos -= frame.data.size
+        movi.pos -= 8
+        id = movi.read 4
+        movi.pos = pos
         id
       end
     end
 
-    FileUtils.mkdir OUTPUT_DIR unless File.exist? OUTPUT_DIR
-    @in = FILES_DIR + 'sample.avi'
-    @out = OUTPUT_DIR + 'out.avi'
-  end
-
-  after :each do
-    FileUtils.rm Dir.glob((OUTPUT_DIR + '*').to_s)
-  end
-
-  after :all do
-    FileUtils.rmdir OUTPUT_DIR
+    AviGlitch::Avi.class_eval do
+      define_method(:get_movi) do
+        @movi
+      end
+    end
   end
 
   it 'should save the same file when nothing is changed' do
@@ -85,15 +80,6 @@ describe AviGlitch::Frames do
     avi.close
   end
 
-  it 'should hide the inner variables' do
-    avi = AviGlitch.open @in
-    frames = avi.frames
-    lambda { frames.meta }.should raise_error(NoMethodError)
-    lambda { frames.io }.should raise_error(NoMethodError)
-    lambda { frames.frames_data_as_io }.should raise_error(NoMethodError)
-    avi.close
-  end
-
   it 'should save video frames count in header' do
     avi = AviGlitch.open @in
     c = 0
@@ -460,9 +446,9 @@ describe AviGlitch::Frames do
     end
   end
 
-  it 'should clear keyframes with one method' do
+  it 'should mutate keyframes into deltaframe' do
     a = AviGlitch.open @in
-    a.frames.clear_keyframes!
+    a.frames.mutate_keyframes_into_deltaframes!
     a.output @out
     a = AviGlitch.open @out
     a.frames.each do |f|
@@ -470,7 +456,7 @@ describe AviGlitch::Frames do
     end
 
     a = AviGlitch.open @in
-    a.frames.clear_keyframes! 0..50
+    a.frames.mutate_keyframes_into_deltaframes! 0..50
     a.output @out
     a = AviGlitch.open @out
     a.frames.each_with_index do |f, i|
@@ -480,4 +466,104 @@ describe AviGlitch::Frames do
     end
   end
 
+  it 'should return Enumerator with #each' do
+    a = AviGlitch.open @in
+    enum = a.frames.each
+    enum.each do |f, i|
+      if f.is_keyframe?
+        f.data = f.data.gsub(/\d/, '')
+      end
+    end
+    a.output @out
+    AviGlitch::Base.surely_formatted?(@out, true).should be true
+    expect(File.size(@out)).to be < File.size(@in)
+  end
+
+  it 'should use Enumerator as an external iterator',
+     :skip => Gem::Version.new(RUBY_VERSION.dup) < Gem::Version.new('1.9.0') || RUBY_PLATFORM == 'java' do
+    a = AviGlitch.open @in
+    e = a.frames.each
+    expect {
+      while f = e.next do
+        expect(f).to be_a(AviGlitch::Frame)
+        if f.is_keyframe?
+          f.data = f.data.gsub(/\d/, '')
+        end
+      end
+    }.to raise_error(StopIteration)
+    a.output @out
+    AviGlitch::Base.surely_formatted?(@out, true).should be true
+    expect(File.size(@out)).to be < File.size(@in)
+  end
+
+  it 'should count the size of specific frames' do
+    a = AviGlitch.open @in
+    f = a.frames
+
+    kc1 = f.size_of :keyframes
+    kc2 = f.size_of :keyframe
+    kc3 = f.size_of :iframes
+    kc4 = f.size_of :iframe
+
+    dc1 = f.size_of :deltaframes
+    dc2 = f.size_of :deltaframe
+    dc3 = f.size_of :pframes
+    dc4 = f.size_of :pframe
+
+    vc1 = f.size_of :videoframes
+    vc2 = f.size_of :videoframe
+
+    ac1 = f.size_of :audioframes
+    ac2 = f.size_of :audioframe
+
+    kc = dc = vc = ac = 0
+    a.frames.each do |x|
+      vc += x.is_videoframe? ? 1 : 0
+      kc += x.is_keyframe? ? 1 : 0
+      dc += x.is_deltaframe? ? 1 : 0
+      ac += x.is_audioframe? ? 1 : 0
+    end
+
+    a.close
+
+    expect(kc1).to eq(kc)
+    expect(kc2).to eq(kc)
+    expect(kc3).to eq(kc)
+    expect(kc4).to eq(kc)
+
+    expect(dc1).to eq(dc)
+    expect(dc2).to eq(dc)
+    expect(dc3).to eq(dc)
+    expect(dc4).to eq(dc)
+
+    expect(vc1).to eq(vc)
+    expect(vc2).to eq(vc)
+
+    expect(ac1).to eq(ac)
+    expect(ac2).to eq(ac)
+  end
+
+  it 'should pick the first / last frame with a method' do
+    a = AviGlitch.open @in
+    fkidx = -1
+    lkidx = -1
+    faidx = -1
+    laidx = -1
+    a.frames.each_with_index do |f, i|
+      if f.is_keyframe?
+        fkidx = i if fkidx == -1
+        lkidx = i
+      end
+      if f.is_audioframe?
+        faidx = i if faidx == -1
+        laidx = i
+      end
+    end
+    a.frames.index(a.frames.first_of(:keyframe)).should eq(fkidx)
+    a.frames.rindex(a.frames.last_of(:keyframe)).should eq(lkidx)
+    a.frames.index(a.frames.first_of(:audioframe)).should eq(faidx)
+    a.frames.rindex(a.frames.last_of(:audioframe)).should eq(laidx)
+    a.close
+  end
+
 end

data/spec/spec_helper.rb

@@ -2,10 +2,54 @@ require 'rspec'
 require 'aviglitch'
 require 'pathname'
 require 'fileutils'
+require 'net/http'
 
 FILES_DIR = Pathname.new(File.dirname(__FILE__)).realpath + 'files'
 OUTPUT_DIR = FILES_DIR + 'output'
 
 RSpec.configure do |config|
-  
+  config.filter_run_excluding :skip => true
+  config.expect_with :rspec do |c|
+    c.syntax = [:should, :expect]
+  end
+
+  config.before(:all) do
+    FileUtils.mkdir FILES_DIR unless File.exist? FILES_DIR
+    FileUtils.mkdir OUTPUT_DIR unless File.exist? OUTPUT_DIR
+    @in = FILES_DIR + 'sample1.avi'
+    @in2 = FILES_DIR + 'sample2.avi'
+    @out = OUTPUT_DIR + 'out.avi'
+    [
+      [@in2, 'http://a.ucnv.org/sample2.avi'], [@in, 'http://a.ucnv.org/sample1.avi']
+    ].each do |file, url|
+      unless File.exist? file
+        if file == @in2
+          puts 'At first test it needs to download a file over 1GB. It will take a while.'
+        end
+        puts 'Downloading ' + url
+        $stdout.sync = true
+        u = URI.parse url
+        Net::HTTP.start(u.host, u.port) do |http|
+          res = http.request_head u.path
+          max = res['content-length'].to_i
+          len = 0
+          bl = 75
+          File.open(file, 'w') do |file|
+            http.get(u.path) do |chunk|
+              file.write chunk
+              len += chunk.length
+              pct = '%3.1f' % (100.0 * len / max)
+              bar = ('#' * (bl * len / max)).ljust(bl)
+              print "\r#{bar} #{'%5s' % pct}%" unless ENV['CI']
+            end
+          end
+        end
+        puts
+      end
+    end
+  end
+
+  config.after(:each) do
+    FileUtils.rm Dir.glob((OUTPUT_DIR + '*').to_s)
+  end
 end