ruby-opencv 0.0.8-x86-mingw32 → 0.0.9.pre2-x86-mingw32

checksums.yaml

@@ -0,0 +1,7 @@
+---
+SHA1:
+  metadata.gz: 0a4168af4998ed69e69f9cc4f65cc4823f0bc5b9
+  data.tar.gz: 9cc4ad8a1fe60f3ae95c70837151381f4a6f9d13
+SHA512:
+  metadata.gz: 1f78f622de1d449ce2d9882887f04ed35181a9394d8e856a6f939fa4a28892c6311a2996d5992cd59353e81ea07e661d81086d2ae5c8d5b77d83ad1c5f522931
+  data.tar.gz: 94ecb941ed131422d4ba69c0d067907c569915ca70817b62852ed6d5d34b16c6bbc645e35ddd553575781d23496c253adebd27df63f863ee840ba8f01a2f1c2c

data/DEVELOPERS_NOTE.md

@@ -14,12 +14,11 @@
   * [hoe](https://github.com/seattlerb/hoe)
   * [hoe-gemspec](https://github.com/flavorjones/hoe-gemspec)
   * [rake-compiler](https://github.com/luislavena/rake-compiler)
-  * [gem-compile](https://github.com/frsyuki/gem-compile)
 
 
 ## Create ruby-opencv gem
 Run the following commands.
-When you use mingw32, use **MSYS console**, or when you use mswin32, 
+When you use mingw32, use **MSYS console**, or when you use mswin32,
 use [**Visual Studio Command Prompt**](http://msdn.microsoft.com/en-us/library/ms229859.aspx)
 instead of cmd.exe.
 
@@ -32,16 +31,34 @@ $ git ls-files > Manifest.txt
 $ rake gem:spec
 $ rake gem
 ```
-**ruby-opencv-x.y.z.gem** will be created in pkg/ directory.
+**ruby-opencv-x.y.z.gem** will be created in **pkg** directory.
 
-To create pre-build binaries, run the following commands in Windows.
+To create pre-build binaries, create a config file firstly:
 
+```yml
+# config.yml
+platform: mingw32
+rubies:
+  - C:/ruby-1.9.3-p392-mingw32/bin/ruby.exe
+  - C:/ruby-2.0.0-p0-mingw32/bin/ruby.exe
+extopts:
+  - --with-opencv-include=C:/opencv/build/include
+  - --with-opencv-lib=C:/opencv/build/x86/mingw/lib
 ```
-$ cd pkg
-$ gem compile ruby-opencv-*.gem
+
+Entries are below:
+
+- **platform**: Target platform (e.g. mingw32, mswin32)
+- **rubies**: Array of target versions of ruby's paths (You can create fat gems if you specify multiple versions of ruby)
+- **extopts**: Array of options to be passed to **extconf.rb**
+
+Then, run the following command:
+
+```
+$ rake gem:precompile CONFIG=config.yml
 ```
 
-**ruby-opencv-x.y.z-x86-mingw32.gem** will be created when you use mingw32, or
+**ruby-opencv-x.y.z-mingw32.gem** will be created when you use mingw32, or
 **ruby-opencv-x.y.z-x86-mswin32.gem** when you use mswin32.
 
 
@@ -52,7 +69,7 @@ $ gem compile ruby-opencv-*.gem
 $ git clone git://github.com/ruby-opencv/ruby-opencv.git
 $ cd ruby-opencv
 $ git checkout master
-$ ruby extconf.rb --with-opencv-dir=/path/to/opencvdir
+$ ruby ext/opencv/extconf.rb --with-opencv-dir=/path/to/opencvdir
 $ make
 $ make install
 ```
@@ -68,7 +85,7 @@ Run the following commands on [**Visual Studio Command Prompt**](http://msdn.mic
 $ git clone git://github.com/ruby-opencv/ruby-opencv.git
 $ cd ruby-opencv
 $ git checkout master
-$ ruby extconf.rb --with-opencv-dir=C:\path\to\opencvdir\install  # for your own built OpenCV library
+$ ruby ext/opencv/extconf.rb --with-opencv-dir=C:\path\to\opencvdir\install  # for your own built OpenCV library
 $ nmake
 $ nmake install
 ```
@@ -76,7 +93,7 @@ $ nmake install
 To use pre-built OpenCV libraries, set the following option to extconf.rb.
 
 ```
-$ ruby extconf.rb --with-opencv-include=C:\path\to\opencvdir\build\include --with-opencv-lib=C:\path\to\opencvdir\build\x86\vc10\lib
+$ ruby ext/opencv/extconf.rb --with-opencv-include=C:\path\to\opencvdir\build\include --with-opencv-lib=C:\path\to\opencvdir\build\x86\vc10\lib
 ```
 
 
@@ -88,7 +105,7 @@ Run the following commands on **MSYS console**.
 $ git clone git://github.com/ruby-opencv/ruby-opencv.git
 $ cd ruby-opencv
 $ git checkout master
-$ ruby extconf.rb --with-opencv-dir=/C/path/to/opencvdir/install  # for your own built OpenCV library
+$ ruby ext/opencv/extconf.rb --with-opencv-dir=/C/path/to/opencvdir/install  # for your own built OpenCV library
 $ make
 $ make install
 ```
@@ -96,7 +113,7 @@ $ make install
 To use pre-built OpenCV libraries, set the following option to extconf.rb.
 
 ```
-$ ruby extconf.rb --with-opencv-include=/c/path/to/opencvdir/build/include --with-opencv-lib=/c/path/to/opencvdir/build/x86/mingw/lib
+$ ruby ext/opencv/extconf.rb --with-opencv-include=/c/path/to/opencvdir/build/include --with-opencv-lib=/c/path/to/opencvdir/build/x86/mingw/lib
 ```
 
 

data/Gemfile

@@ -1,9 +1,8 @@
-source :gemcutter
+source 'https://rubygems.org'
 
 group :development do
   gem "hoe"
   gem "hoe-gemspec"
   gem "rake-compiler"
-  gem "gem-compile"
 end
 

data/License.txt

@@ -1,30 +1,30 @@
-The BSD Liscense
-
-Copyright (c) 2008, Masakazu Yonekura
-All rights reserved.
-
-Redistribution and use of this software in source and binary forms, with or without modification, are
-permitted provided that the following conditions are met:
-
-* Redistributions of source code must retain the above
-  copyright notice, this list of conditions and the
-  following disclaimer.
-
-* Redistributions in binary form must reproduce the above
-  copyright notice, this list of conditions and the
-  following disclaimer in the documentation and/or other
-  materials provided with the distribution.
-
-* Neither the name of Masakazu Yonekura. nor the names of its
-  contributors may be used to endorse or promote products
-  derived from this software without specific prior
-  written permission of Masakazu Yonekura.
-
-THIS SOFTWARE IS PROVIDED BY THE COPYRIGHT HOLDERS AND CONTRIBUTORS "AS IS" AND ANY EXPRESS OR IMPLIED
-WARRANTIES, INCLUDING, BUT NOT LIMITED TO, THE IMPLIED WARRANTIES OF MERCHANTABILITY AND FITNESS FOR A
-PARTICULAR PURPOSE ARE DISCLAIMED. IN NO EVENT SHALL THE COPYRIGHT OWNER OR CONTRIBUTORS BE LIABLE FOR
-ANY DIRECT, INDIRECT, INCIDENTAL, SPECIAL, EXEMPLARY, OR CONSEQUENTIAL DAMAGES (INCLUDING, BUT NOT
-LIMITED TO, PROCUREMENT OF SUBSTITUTE GOODS OR SERVICES; LOSS OF USE, DATA, OR PROFITS; OR BUSINESS
-INTERRUPTION) HOWEVER CAUSED AND ON ANY THEORY OF LIABILITY, WHETHER IN CONTRACT, STRICT LIABILITY, OR
-TORT (INCLUDING NEGLIGENCE OR OTHERWISE) ARISING IN ANY WAY OUT OF THE USE OF THIS SOFTWARE, EVEN IF
-ADVISED OF THE POSSIBILITY OF SUCH DAMAGE.
+The BSD License
+
+Copyright (c) 2008, Masakazu Yonekura
+All rights reserved.
+
+Redistribution and use of this software in source and binary forms, with or without modification, are
+permitted provided that the following conditions are met:
+
+* Redistributions of source code must retain the above
+  copyright notice, this list of conditions and the
+  following disclaimer.
+
+* Redistributions in binary form must reproduce the above
+  copyright notice, this list of conditions and the
+  following disclaimer in the documentation and/or other
+  materials provided with the distribution.
+
+* Neither the name of Masakazu Yonekura. nor the names of its
+  contributors may be used to endorse or promote products
+  derived from this software without specific prior
+  written permission of Masakazu Yonekura.
+
+THIS SOFTWARE IS PROVIDED BY THE COPYRIGHT HOLDERS AND CONTRIBUTORS "AS IS" AND ANY EXPRESS OR IMPLIED
+WARRANTIES, INCLUDING, BUT NOT LIMITED TO, THE IMPLIED WARRANTIES OF MERCHANTABILITY AND FITNESS FOR A
+PARTICULAR PURPOSE ARE DISCLAIMED. IN NO EVENT SHALL THE COPYRIGHT OWNER OR CONTRIBUTORS BE LIABLE FOR
+ANY DIRECT, INDIRECT, INCIDENTAL, SPECIAL, EXEMPLARY, OR CONSEQUENTIAL DAMAGES (INCLUDING, BUT NOT
+LIMITED TO, PROCUREMENT OF SUBSTITUTE GOODS OR SERVICES; LOSS OF USE, DATA, OR PROFITS; OR BUSINESS
+INTERRUPTION) HOWEVER CAUSED AND ON ANY THEORY OF LIABILITY, WHETHER IN CONTRACT, STRICT LIABILITY, OR
+TORT (INCLUDING NEGLIGENCE OR OTHERWISE) ARISING IN ANY WAY OUT OF THE USE OF THIS SOFTWARE, EVEN IF
+ADVISED OF THE POSSIBILITY OF SUCH DAMAGE.

data/Manifest.txt

@@ -6,6 +6,7 @@ License.txt
 Manifest.txt
 README.md
 Rakefile
+config.yml
 examples/alpha_blend.rb
 examples/box.png
 examples/box_in_scene.png
@@ -109,15 +110,13 @@ ext/opencv/cvutils.cpp
 ext/opencv/cvutils.h
 ext/opencv/cvvideowriter.cpp
 ext/opencv/cvvideowriter.h
+ext/opencv/extconf.rb
 ext/opencv/gui.cpp
 ext/opencv/gui.h
 ext/opencv/iplconvkernel.cpp
 ext/opencv/iplconvkernel.h
 ext/opencv/iplimage.cpp
 ext/opencv/iplimage.h
-ext/opencv/lib/opencv.rb
-ext/opencv/lib/opencv/psyched_yaml.rb
-ext/opencv/lib/opencv/version.rb
 ext/opencv/mouseevent.cpp
 ext/opencv/mouseevent.h
 ext/opencv/opencv.cpp
@@ -128,11 +127,13 @@ ext/opencv/trackbar.cpp
 ext/opencv/trackbar.h
 ext/opencv/window.cpp
 ext/opencv/window.h
-extconf.rb
 images/CvMat_sobel.png
 images/CvMat_sub_rect.png
 images/CvSeq_relationmap.png
 images/face_detect_from_lena.jpg
+lib/opencv.rb
+lib/opencv/psyched_yaml.rb
+lib/opencv/version.rb
 ruby-opencv.gemspec
 test/helper.rb
 test/runner.rb

data/README.md

@@ -3,7 +3,7 @@
 An OpenCV wrapper for Ruby.
 
 * Web site: <https://github.com/ruby-opencv/ruby-opencv>
-* Ruby 1.8.7, 1.9.3 and OpenCV 2.4.3 are supported.
+* Ruby 1.9.3, 2.0.0 and OpenCV 2.4.5 are supported.
 
 ## Requirement
 

data/Rakefile

@@ -1,27 +1,36 @@
-# -*- mode: ruby; coding: utf-8-unix -*-
+# -*- mode: ruby; coding: utf-8 -*-
 require 'rubygems'
-require './ext/opencv/lib/opencv/psyched_yaml'
+require "rubygems/ext"
+require "rubygems/installer"
 require 'hoe'
 require 'rake/extensiontask'
+require 'fileutils'
+require './lib/opencv/psyched_yaml'
+
+SO_FILE = 'opencv.so'
 
 Hoe.plugin :gemspec
 
 hoespec = Hoe.spec 'ruby-opencv' do |s|
   s.summary = 'OpenCV wrapper for Ruby'
+  s.description = 'ruby-opencv is a wrapper of OpenCV for Ruby. It helps you to write computer vision programs (e.g. detecting faces from pictures) with Ruby.'
+  s.licenses = ['The BSD License']
   s.developer('lsxi', 'masakazu.yonekura@gmail.com')
   s.developer('ser1zw', 'azariahsawtikes@gmail.com')
   s.developer('pcting', 'pcting@gmail.com')
 
   s.readme_file  = 'README.md'
   s.history_file = 'History.txt'
-  s.spec_extras = { :extensions => ['extconf.rb'] }
+
+  s.spec_extras = { :extensions => ['ext/opencv/extconf.rb'] }
+
   s.test_globs = ['test/test_*.rb']
   s.urls = ['https://github.com/ruby-opencv/ruby-opencv/']
 
   s.extra_dev_deps << ['rake-compiler', '>= 0'] << ['hoe-gemspec']
 
   Rake::ExtensionTask.new('opencv', spec) do |ext|
-    ext.lib_dir = File.join('lib', 'opencv')
+    ext.lib_dir = 'lib'
   end
 end
 
@@ -29,4 +38,53 @@ hoespec.spec.files.delete('.gemtest')
 
 Rake::Task[:test].prerequisites << :compile
 
+desc 'Create a pre-compiled gem'
+task 'gem:precompile' => ['gem'] do
+  tmp_dir = Dir.mktmpdir('tmp', '.')
+  gemfile = Dir.glob("pkg/*.gem")[0]
+  target_dir = File.join(tmp_dir, File.basename(gemfile, '.gem'))
+
+  installer = Gem::Installer.new(gemfile)
+  installer.unpack(target_dir)
+
+  gemspec = installer.spec
+  extension = gemspec.extensions[0]
+  gemspec.extensions.clear
+
+  config = ENV['CONFIG'] ?  YAML.load_file(ENV['CONFIG']) : {}
+  rubies = config['rubies'] || [Gem.ruby]
+  args = config['extopts'] || []
+  gemspec.platform = config['platform'] || Gem::Platform::CURRENT
+
+  multi = rubies.size > 1
+  rubies.each { |ruby|
+    lib_dir = 'lib'
+    if multi
+      major, minor, _ = `#{ruby} -e "print RUBY_VERSION"`.chomp.split('.')
+      lib_dir = File.join(lib_dir, [major, minor].join('.'))
+    end
+
+    make_cmd = (`#{ruby} -e "print RUBY_PLATFORM"` =~ /mswin/) ? 'nmake' : 'make'
+    Dir.chdir(target_dir) {
+      cmd = [ruby, extension, *args].join(' ')
+      results = []
+      Gem::Ext::ExtConfBuilder.run(cmd, results)
+      Gem::Ext::ExtConfBuilder.make('', results)
+
+      FileUtils.mkdir_p lib_dir
+      FileUtils.mv SO_FILE, lib_dir
+      sh "#{make_cmd} clean"
+    }
+
+    gemspec.files << File.join(lib_dir, SO_FILE)
+  }
+
+  Dir.chdir(target_dir) {
+    gemfile = Gem::Package.build(gemspec)
+    FileUtils.mv gemfile, File.dirname(__FILE__)
+  }
+
+  FileUtils.rm_rf tmp_dir
+end
+
 # vim: syntax=ruby

data/config.yml

@@ -0,0 +1,7 @@
+platform: mingw32
+rubies:
+  - C:/ruby-1.9.3-p392-mingw32/bin/ruby.exe
+  - C:/ruby-2.0.0-p0-mingw32/bin/ruby.exe
+extopts:
+  - --with-opencv-include=C:/opencv/build/include
+  - --with-opencv-lib=C:/opencv/build/x86/mingw/lib

data/examples/alpha_blend.rb

@@ -1,21 +1,21 @@
-#!/usr/bin/env ruby
-# -*- mode: ruby; coding: utf-8-unix -*-
-
-# Alpha blending sample with GUI
-
-require 'opencv'
-include OpenCV
-
-img1 = IplImage.load('lenna.jpg', CV_LOAD_IMAGE_ANYCOLOR | CV_LOAD_IMAGE_ANYDEPTH)
-img2 = IplImage.load('tiffany.jpg', CV_LOAD_IMAGE_ANYCOLOR | CV_LOAD_IMAGE_ANYDEPTH)
-
-window = GUI::Window.new('Alpha blend')
-max = 100.0
-val = max / 2.0
-window.set_trackbar("Alpha", max, val) { |v|
-  a = v.to_f / max
-  window.show CvMat.add_weighted(img1, a, img2, 1.0 - a, 0)
-}
-window.show CvMat.add_weighted(img1, val / max, img2, 1.0 - val / max, 0)
-GUI::wait_key
-
+#!/usr/bin/env ruby
+# -*- mode: ruby; coding: utf-8 -*-
+
+# Alpha blending sample with GUI
+
+require 'opencv'
+include OpenCV
+
+img1 = IplImage.load('lenna.jpg', CV_LOAD_IMAGE_ANYCOLOR | CV_LOAD_IMAGE_ANYDEPTH)
+img2 = IplImage.load('tiffany.jpg', CV_LOAD_IMAGE_ANYCOLOR | CV_LOAD_IMAGE_ANYDEPTH)
+
+window = GUI::Window.new('Alpha blend')
+max = 100.0
+val = max / 2.0
+window.set_trackbar("Alpha", max, val) { |v|
+  a = v.to_f / max
+  window.show CvMat.add_weighted(img1, a, img2, 1.0 - a, 0)
+}
+window.show CvMat.add_weighted(img1, val / max, img2, 1.0 - val / max, 0)
+GUI::wait_key
+

data/examples/find_obj.rb

@@ -1,169 +1,169 @@
-#!/usr/bin/env ruby
-# -*- mode: ruby; coding: utf-8-unix -*- 
-
-# A Demo Ruby/OpenCV Implementation of SURF
-# See https://code.ros.org/trac/opencv/browser/tags/2.3.1/opencv/samples/c/find_obj.cpp
-require 'opencv'
-require 'benchmark'
-include OpenCV
-
-def compare_surf_descriptors(d1, d2, best, length)
-  raise ArgumentError unless (length % 4) == 0
-  total_cost = 0
-  0.step(length - 1, 4) { |i|
-    t0 = d1[i] - d2[i]
-    t1 = d1[i + 1] - d2[i + 1]
-    t2 = d1[i + 2] - d2[i + 2]
-    t3 = d1[i + 3] - d2[i + 3]
-    total_cost += t0 * t0 + t1 * t1 + t2 * t2 + t3 * t3
-    break if total_cost > best
-  }
-  total_cost
-end
-
-def naive_nearest_neighbor(vec, laplacian, model_keypoints, model_descriptors)
-  length = model_descriptors[0].size
-  neighbor = nil
-  dist1 = 1e6
-  dist2 = 1e6
-
-  model_descriptors.size.times { |i|
-    kp = model_keypoints[i]
-    mvec = model_descriptors[i]
-    next if laplacian != kp.laplacian
-    
-    d = compare_surf_descriptors(vec, mvec, dist2, length)
-    if d < dist1
-      dist2 = dist1
-      dist1 = d
-      neighbor = i
-    elsif d < dist2
-      dist2 = d
-    end
-  }
-
-  return (dist1 < 0.6 * dist2) ? neighbor : nil
-end
-
-def find_pairs(object_keypoints, object_descriptors,
-               image_keypoints, image_descriptors)
-  ptpairs = []
-  object_descriptors.size.times { |i|
-    kp = object_keypoints[i]
-    descriptor = object_descriptors[i]
-    nearest_neighbor = naive_nearest_neighbor(descriptor, kp.laplacian, image_keypoints, image_descriptors)
-    unless nearest_neighbor.nil?
-      ptpairs << i
-      ptpairs << nearest_neighbor
-    end
-  }
-  ptpairs
-end
-
-def locate_planar_object(object_keypoints, object_descriptors,
-                         image_keypoints, image_descriptors, src_corners)
-  ptpairs = find_pairs(object_keypoints, object_descriptors, image_keypoints, image_descriptors)
-  n = ptpairs.size / 2
-  return nil if n < 4
-
-  pt1 = []
-  pt2 = []
-  n.times { |i|
-    pt1 << object_keypoints[ptpairs[i * 2]].pt
-    pt2 << image_keypoints[ptpairs[i * 2 + 1]].pt
-  }
-
-  _pt1 = CvMat.new(1, n, CV_32F, 2)
-  _pt2 = CvMat.new(1, n, CV_32F, 2)
-  _pt1.set_data(pt1)
-  _pt2.set_data(pt2)
-  h = CvMat.find_homography(_pt1, _pt2, :ransac, 5)
-
-  dst_corners = []
-  4.times { |i|
-    x = src_corners[i].x
-    y = src_corners[i].y
-    z = 1.0 / (h[6][0] * x + h[7][0] * y + h[8][0])
-    x = (h[0][0] * x + h[1][0] * y + h[2][0]) * z
-    y = (h[3][0] * x + h[4][0] * y + h[5][0]) * z
-    dst_corners << CvPoint.new(x.to_i, y.to_i)
-  }
-
-  dst_corners
-end
-
-
-##### Main #####
-puts 'This program demonstrated the use of the SURF Detector and Descriptor using'
-puts 'brute force matching on planar objects.'
-puts 'Usage:'
-puts "ruby #{__FILE__} <object_filename> <scene_filename>, default is box.png and box_in_scene.png"
-puts
-
-object_filename = (ARGV.size == 2) ? ARGV[0] : 'box.png'
-scene_filename = (ARGV.size == 2) ? ARGV[1] : 'box_in_scene.png'
-
-object, image = nil, nil
-begin
-  object = IplImage.load(object_filename, CV_LOAD_IMAGE_GRAYSCALE)
-  image = IplImage.load(scene_filename, CV_LOAD_IMAGE_GRAYSCALE)
-rescue
-  puts "Can not load #{object_filename} and/or #{scene_filename}"
-  puts "Usage: ruby #{__FILE__} [<object_filename> <scene_filename>]"
-  exit
-end
-object_color = object.GRAY2BGR
-
-param = CvSURFParams.new(1500)
-
-object_keypoints, object_descriptors = nil, nil
-image_keypoints, image_descriptors = nil, nil
-tms = Benchmark.measure {
-  object_keypoints, object_descriptors = object.extract_surf(param)
-  puts "Object Descriptors: #{object_descriptors.size}"
-
-  image_keypoints, image_descriptors = image.extract_surf(param)
-  puts "Image Descriptors: #{image_descriptors.size}"
-}
-puts "Extraction time = #{tms.real * 1000} ms"
-
-correspond = IplImage.new(image.width, object.height + image.height, CV_8U, 1);
-correspond.set_roi(CvRect.new(0, 0, object.width, object.height))
-object.copy(correspond)
-correspond.set_roi(CvRect.new(0, object.height, image.width, image.height))
-image.copy(correspond)
-correspond.reset_roi
-
-src_corners = [CvPoint.new(0, 0), CvPoint.new(object.width, 0),
-               CvPoint.new(object.width, object.height), CvPoint.new(0, object.height)]
-dst_corners = locate_planar_object(object_keypoints, object_descriptors,
-                                   image_keypoints, image_descriptors, src_corners)
-
-correspond = correspond.GRAY2BGR
-if dst_corners
-  4.times { |i|
-    r1 = dst_corners[i % 4]
-    r2 = dst_corners[(i + 1) % 4]
-    correspond.line!(CvPoint.new(r1.x, r1.y + object.height), CvPoint.new(r2.x, r2.y + object.height),
-                     :color => CvColor::Red, :thickness => 2, :line_type => :aa)
-  }
-end
-
-ptpairs = find_pairs(object_keypoints, object_descriptors, image_keypoints, image_descriptors)
-
-0.step(ptpairs.size - 1, 2) { |i|
-  r1 = object_keypoints[ptpairs[i]]
-  r2 = image_keypoints[ptpairs[i + 1]]
-  correspond.line!(r1.pt, CvPoint.new(r2.pt.x, r2.pt.y + object.height),
-                   :color => CvColor::Red, :line_type => :aa)
-}
-
-object_keypoints.each { |r|
-  radius = (r.size * 1.2 / 9.0 * 2).to_i
-  object_color.circle!(r.pt, radius, :color => CvColor::Red, :line_type => :aa)
-}
-
-GUI::Window.new('Object Correspond').show correspond
-GUI::Window.new('Object').show object_color
-GUI::wait_key
-
+#!/usr/bin/env ruby
+# -*- mode: ruby; coding: utf-8 -*-
+
+# A Demo Ruby/OpenCV Implementation of SURF
+# See https://code.ros.org/trac/opencv/browser/tags/2.3.1/opencv/samples/c/find_obj.cpp
+require 'opencv'
+require 'benchmark'
+include OpenCV
+
+def compare_surf_descriptors(d1, d2, best, length)
+  raise ArgumentError unless (length % 4) == 0
+  total_cost = 0
+  0.step(length - 1, 4) { |i|
+    t0 = d1[i] - d2[i]
+    t1 = d1[i + 1] - d2[i + 1]
+    t2 = d1[i + 2] - d2[i + 2]
+    t3 = d1[i + 3] - d2[i + 3]
+    total_cost += t0 * t0 + t1 * t1 + t2 * t2 + t3 * t3
+    break if total_cost > best
+  }
+  total_cost
+end
+
+def naive_nearest_neighbor(vec, laplacian, model_keypoints, model_descriptors)
+  length = model_descriptors[0].size
+  neighbor = nil
+  dist1 = 1e6
+  dist2 = 1e6
+
+  model_descriptors.size.times { |i|
+    kp = model_keypoints[i]
+    mvec = model_descriptors[i]
+    next if laplacian != kp.laplacian
+    
+    d = compare_surf_descriptors(vec, mvec, dist2, length)
+    if d < dist1
+      dist2 = dist1
+      dist1 = d
+      neighbor = i
+    elsif d < dist2
+      dist2 = d
+    end
+  }
+
+  return (dist1 < 0.6 * dist2) ? neighbor : nil
+end
+
+def find_pairs(object_keypoints, object_descriptors,
+               image_keypoints, image_descriptors)
+  ptpairs = []
+  object_descriptors.size.times { |i|
+    kp = object_keypoints[i]
+    descriptor = object_descriptors[i]
+    nearest_neighbor = naive_nearest_neighbor(descriptor, kp.laplacian, image_keypoints, image_descriptors)
+    unless nearest_neighbor.nil?
+      ptpairs << i
+      ptpairs << nearest_neighbor
+    end
+  }
+  ptpairs
+end
+
+def locate_planar_object(object_keypoints, object_descriptors,
+                         image_keypoints, image_descriptors, src_corners)
+  ptpairs = find_pairs(object_keypoints, object_descriptors, image_keypoints, image_descriptors)
+  n = ptpairs.size / 2
+  return nil if n < 4
+
+  pt1 = []
+  pt2 = []
+  n.times { |i|
+    pt1 << object_keypoints[ptpairs[i * 2]].pt
+    pt2 << image_keypoints[ptpairs[i * 2 + 1]].pt
+  }
+
+  _pt1 = CvMat.new(1, n, CV_32F, 2)
+  _pt2 = CvMat.new(1, n, CV_32F, 2)
+  _pt1.set_data(pt1)
+  _pt2.set_data(pt2)
+  h = CvMat.find_homography(_pt1, _pt2, :ransac, 5)
+
+  dst_corners = []
+  4.times { |i|
+    x = src_corners[i].x
+    y = src_corners[i].y
+    z = 1.0 / (h[6][0] * x + h[7][0] * y + h[8][0])
+    x = (h[0][0] * x + h[1][0] * y + h[2][0]) * z
+    y = (h[3][0] * x + h[4][0] * y + h[5][0]) * z
+    dst_corners << CvPoint.new(x.to_i, y.to_i)
+  }
+
+  dst_corners
+end
+
+
+##### Main #####
+puts 'This program demonstrated the use of the SURF Detector and Descriptor using'
+puts 'brute force matching on planar objects.'
+puts 'Usage:'
+puts "ruby #{__FILE__} <object_filename> <scene_filename>, default is box.png and box_in_scene.png"
+puts
+
+object_filename = (ARGV.size == 2) ? ARGV[0] : 'box.png'
+scene_filename = (ARGV.size == 2) ? ARGV[1] : 'box_in_scene.png'
+
+object, image = nil, nil
+begin
+  object = IplImage.load(object_filename, CV_LOAD_IMAGE_GRAYSCALE)
+  image = IplImage.load(scene_filename, CV_LOAD_IMAGE_GRAYSCALE)
+rescue
+  puts "Can not load #{object_filename} and/or #{scene_filename}"
+  puts "Usage: ruby #{__FILE__} [<object_filename> <scene_filename>]"
+  exit
+end
+object_color = object.GRAY2BGR
+
+param = CvSURFParams.new(1500)
+
+object_keypoints, object_descriptors = nil, nil
+image_keypoints, image_descriptors = nil, nil
+tms = Benchmark.measure {
+  object_keypoints, object_descriptors = object.extract_surf(param)
+  puts "Object Descriptors: #{object_descriptors.size}"
+
+  image_keypoints, image_descriptors = image.extract_surf(param)
+  puts "Image Descriptors: #{image_descriptors.size}"
+}
+puts "Extraction time = #{tms.real * 1000} ms"
+
+correspond = IplImage.new(image.width, object.height + image.height, CV_8U, 1);
+correspond.set_roi(CvRect.new(0, 0, object.width, object.height))
+object.copy(correspond)
+correspond.set_roi(CvRect.new(0, object.height, image.width, image.height))
+image.copy(correspond)
+correspond.reset_roi
+
+src_corners = [CvPoint.new(0, 0), CvPoint.new(object.width, 0),
+               CvPoint.new(object.width, object.height), CvPoint.new(0, object.height)]
+dst_corners = locate_planar_object(object_keypoints, object_descriptors,
+                                   image_keypoints, image_descriptors, src_corners)
+
+correspond = correspond.GRAY2BGR
+if dst_corners
+  4.times { |i|
+    r1 = dst_corners[i % 4]
+    r2 = dst_corners[(i + 1) % 4]
+    correspond.line!(CvPoint.new(r1.x, r1.y + object.height), CvPoint.new(r2.x, r2.y + object.height),
+                     :color => CvColor::Red, :thickness => 2, :line_type => :aa)
+  }
+end
+
+ptpairs = find_pairs(object_keypoints, object_descriptors, image_keypoints, image_descriptors)
+
+0.step(ptpairs.size - 1, 2) { |i|
+  r1 = object_keypoints[ptpairs[i]]
+  r2 = image_keypoints[ptpairs[i + 1]]
+  correspond.line!(r1.pt, CvPoint.new(r2.pt.x, r2.pt.y + object.height),
+                   :color => CvColor::Red, :line_type => :aa)
+}
+
+object_keypoints.each { |r|
+  radius = (r.size * 1.2 / 9.0 * 2).to_i
+  object_color.circle!(r.pt, radius, :color => CvColor::Red, :line_type => :aa)
+}
+
+GUI::Window.new('Object Correspond').show correspond
+GUI::Window.new('Object').show object_color
+GUI::wait_key
+