hornetseye-opencv 0.1.0

data/README.md

@@ -0,0 +1,5 @@
+hornetseye-opencv
+======
+This Ruby extension provides conversions from {Hornetseye::MultiArray} to {OpenCV::CvMat} and vice versa.
+
+

data/Rakefile

@@ -0,0 +1,183 @@
+#!/usr/bin/env ruby
+require 'date'
+require 'rake/clean'
+require 'rake/testtask'
+require 'rake/packagetask'
+require 'rake/loaders/makefile'
+require 'rbconfig'
+
+PKG_NAME = 'hornetseye-opencv'
+PKG_VERSION = '0.1.0'
+CXX = ENV[ 'CXX' ] || 'g++'
+STRIP = ENV[ 'STRIP' ] || 'strip'
+RB_FILES = FileList[ 'lib/**/*.rb' ]
+CC_FILES = FileList[ 'ext/*.cc' ]
+HH_FILES = FileList[ 'ext/*.hh' ] + FileList[ 'ext/*.tcc' ]
+TC_FILES = FileList[ 'test/tc_*.rb' ]
+TS_FILES = FileList[ 'test/ts_*.rb' ]
+SO_FILE = "ext/#{PKG_NAME.tr '\-', '_'}.so"
+PKG_FILES = [ 'Rakefile', 'README.md', 'COPYING', '.document' ] +
+            RB_FILES + CC_FILES + HH_FILES + TS_FILES + TC_FILES
+BIN_FILES = [ 'README.md', 'COPYING', '.document', SO_FILE ] +
+            RB_FILES + TS_FILES + TC_FILES
+SUMMARY = %q{OpenCV integration for Hornetseye}
+DESCRIPTION = %q{This Ruby extension provides conversions from Hornetseye::MultiArray to OpenCV::CvMat and vice versa.}
+AUTHOR = %q{Jan Wedekind}
+EMAIL = %q{jan@wedesoft.de}
+HOMEPAGE = %q{http://wedesoft.github.com/hornetseye-opencv/}
+
+OBJ = CC_FILES.ext 'o'
+$CXXFLAGS = ENV[ 'CXXFLAGS' ] || ''
+$CXXFLAGS = "#{$CXXFLAGS} -fPIC -DNDEBUG"
+if RbConfig::CONFIG[ 'rubyhdrdir' ]
+  $CXXFLAGS = "#{$CXXFLAGS} -I#{RbConfig::CONFIG[ 'rubyhdrdir' ]} " +
+              "-I#{RbConfig::CONFIG[ 'rubyhdrdir' ]}/#{RbConfig::CONFIG[ 'arch' ]}"
+else
+  $CXXFLAGS = "#{$CXXFLAGS} -I#{RbConfig::CONFIG[ 'archdir' ]}"
+end
+$LIBRUBYARG = RbConfig::CONFIG[ 'LIBRUBYARG' ]
+$SITELIBDIR = RbConfig::CONFIG[ 'sitelibdir' ]
+$SITEARCHDIR = RbConfig::CONFIG[ 'sitearchdir' ]
+
+task :default => :all
+
+desc 'Compile Ruby extension (default)'
+task :all => [ SO_FILE ]
+
+file SO_FILE => OBJ do |t|
+   sh "#{CXX} -shared -o #{t.name} #{OBJ} -lcv -lhighgui -lcvaux -lcxcore -lml #{$LIBRUBYARG}"
+   # sh "#{STRIP} --strip-all #{t.name}"
+end
+
+task :test => [ SO_FILE ]
+
+desc 'Install Ruby extension'
+task :install => :all do
+  verbose true do
+    for f in RB_FILES do
+      FileUtils.mkdir_p "#{$SITELIBDIR}/#{File.dirname f.gsub( /^lib\//, '' )}"
+      FileUtils.cp_r f, "#{$SITELIBDIR}/#{f.gsub( /^lib\//, '' )}"
+    end
+    FileUtils.mkdir_p $SITEARCHDIR
+    FileUtils.cp SO_FILE, "#{$SITEARCHDIR}/#{File.basename SO_FILE}"
+  end
+end
+
+desc 'Uninstall Ruby extension'
+task :uninstall do
+  verbose true do
+    for f in RB_FILES do
+      FileUtils.rm_f "#{$SITELIBDIR}/#{f.gsub /^lib\//, ''}"
+    end
+    FileUtils.rm_f "#{$SITEARCHDIR}/#{File.basename SO_FILE}"
+  end
+end
+
+Rake::TestTask.new do |t|
+  t.libs << 'ext'
+  t.test_files = TC_FILES
+end
+
+begin
+  require 'yard'
+  YARD::Rake::YardocTask.new :yard do |y|
+    y.options << '--no-private'
+    y.files << FileList[ 'lib/**/*.rb' ]
+  end
+rescue LoadError
+  STDERR.puts 'Please install \'yard\' if you want to generate documentation'
+end
+
+Rake::PackageTask.new PKG_NAME, PKG_VERSION do |p|
+  p.need_tar = true
+  p.package_files = PKG_FILES
+end
+
+begin
+  require 'rubygems'
+  require 'rubygems/builder'
+  $SPEC = Gem::Specification.new do |s|
+    s.name = PKG_NAME
+    s.version = PKG_VERSION
+    s.platform = Gem::Platform::RUBY
+    s.date = Date.today.to_s
+    s.summary = SUMMARY
+    s.description = DESCRIPTION
+    s.author = AUTHOR
+    s.email = EMAIL
+    s.homepage = HOMEPAGE
+    s.files = PKG_FILES
+    s.test_files = TC_FILES
+    s.require_paths = [ 'lib', 'ext' ]
+    s.rubyforge_project = %q{hornetseye}
+    s.extensions = %w{Rakefile}
+    s.has_rdoc = 'yard'
+    s.extra_rdoc_files = []
+    s.rdoc_options = %w{--no-private}
+    s.add_dependency %<malloc>, [ '~> 1.1' ]
+    s.add_dependency %<multiarray>, [ '~> 0.10' ]
+    s.add_development_dependency %q{rake}
+  end
+  GEM_SOURCE = "#{PKG_NAME}-#{PKG_VERSION}.gem"
+  $BINSPEC = Gem::Specification.new do |s|
+    s.name = PKG_NAME
+    s.version = PKG_VERSION
+    s.platform = Gem::Platform::CURRENT
+    s.date = Date.today.to_s
+    s.summary = SUMMARY
+    s.description = DESCRIPTION
+    s.author = AUTHOR
+    s.email = EMAIL
+    s.homepage = HOMEPAGE
+    s.files = BIN_FILES
+    s.test_files = TC_FILES
+    s.require_paths = [ 'lib', 'ext' ]
+    s.rubyforge_project = %q{hornetseye}
+    s.has_rdoc = 'yard'
+    s.extra_rdoc_files = []
+    s.rdoc_options = %w{--no-private}
+    s.add_dependency %<malloc>, [ '~> 1.1' ]
+    s.add_dependency %<multiarray>, [ '~> 0.10' ]
+  end
+  GEM_BINARY = "#{PKG_NAME}-#{PKG_VERSION}-#{$BINSPEC.platform}.gem"
+  desc "Build the gem file #{GEM_SOURCE}"
+  task :gem => [ "pkg/#{GEM_SOURCE}" ]
+  file "pkg/#{GEM_SOURCE}" => [ 'pkg' ] + $SPEC.files do
+    when_writing 'Creating GEM' do
+      Gem::Builder.new( $SPEC ).build
+      verbose true do
+        FileUtils.mv GEM_SOURCE, "pkg/#{GEM_SOURCE}"
+      end
+    end
+  end
+  desc "Build the gem file #{GEM_BINARY}"
+  task :gem_binary => [ "pkg/#{GEM_BINARY}" ]
+  file "pkg/#{GEM_BINARY}" => [ 'pkg' ] + $BINSPEC.files do
+    when_writing 'Creating binary GEM' do
+      Gem::Builder.new( $BINSPEC ).build
+      verbose true do
+        FileUtils.mv GEM_BINARY, "pkg/#{GEM_BINARY}"
+      end
+    end
+  end
+rescue LoadError
+  STDERR.puts 'Please install \'rubygems\' if you want to create Gem packages'
+end
+
+rule '.o' => '.cc' do |t|
+   sh "#{CXX} #{$CXXFLAGS} -c -o #{t.name} #{t.source}"
+end
+
+file ".depends.mf" do |t|
+  sh "g++ -MM #{$CXXFLAGS} #{CC_FILES.join ' '} | " +
+    "sed -e :a -e N -e 's/\\n/\\$/g' -e ta | " +
+    "sed -e 's/ *\\\\\\$ */ /g' -e 's/\\$/\\n/g' | sed -e 's/^/ext\\//' > #{t.name}"
+end
+CC_FILES.each do |t|
+  file t.ext(".o") => t
+end
+import ".depends.mf"
+
+CLEAN.include 'ext/*.o'
+CLOBBER.include SO_FILE, 'doc', '.yardoc', '.depends.mf'
+

data/ext/cvmat.cc

@@ -0,0 +1,96 @@
+/* HornetsEye - Computer Vision with Ruby
+   Copyright (C) 2006, 2007, 2008, 2009 Jan Wedekind
+
+   This program is free software: you can redistribute it and/or modify
+   it under the terms of the GNU General Public License as published by
+   the Free Software Foundation, either version 3 of the License, or
+   (at your option) any later version.
+
+   This program is distributed in the hope that it will be useful,
+   but WITHOUT ANY WARRANTY; without even the implied warranty of
+   MERCHANTABILITY or FITNESS FOR A PARTICULAR PURPOSE.  See the
+   GNU General Public License for more details.
+
+   You should have received a copy of the GNU General Public License
+   along with this program.  If not, see <http://www.gnu.org/licenses/>. */
+#include <algorithm>
+#include <opencv/cxcore.h>
+#include "cvmat.hh"
+#include "error.hh"
+#include "rubytools.hh"
+
+using namespace std;
+
+VALUE CvMatExt::cRubyClass = Qnil;
+
+VALUE CvMatExt::registerRubyClass( VALUE module )
+{
+  cRubyClass = rb_define_class_under( module, "CvMat", rb_cObject );
+  rb_define_method( cRubyClass, "to_multiarray",
+                    RUBY_METHOD_FUNC( wrapToMultiArray ), 0 );
+  return cRubyClass;
+}
+
+void CvMatExt::deleteRubyObject( void *ptr )
+{
+  CvMat *mat = (CvMat *)ptr;
+  cvReleaseMat( &mat );
+}
+
+VALUE CvMatExt::wrapToMultiArray( VALUE rbSelf )
+{
+  VALUE rbRetVal = Qnil;
+  try {
+    VALUE mHornetseye = rb_define_module( "Hornetseye" );
+    CvMat *cvMat;
+    dataGetStruct( rbSelf, cRubyClass, CvMat, cvMat );
+    int width = cvMat->width;
+    int height = cvMat->height;
+    int channels = CV_MAT_CN( cvMat->type );
+    VALUE rbTypecode;
+    switch ( CV_MAT_DEPTH( cvMat->type ) ) {
+    case CV_8U:
+      rbTypecode = rb_const_get( mHornetseye, rb_intern( "UBYTE" ) );
+      break;
+    case CV_8S:
+      rbTypecode = rb_const_get( mHornetseye, rb_intern( "BYTE" ) );
+      break;
+    case CV_16U:
+      rbTypecode = rb_const_get( mHornetseye, rb_intern( "USINT" ) );
+      break;
+    case CV_16S:
+      rbTypecode = rb_const_get( mHornetseye, rb_intern( "SINT" ) );
+      break;
+    case CV_32S:
+      rbTypecode = rb_const_get( mHornetseye, rb_intern( "INT" ) );
+      break;
+    case CV_32F:
+      rbTypecode = rb_const_get( mHornetseye, rb_intern( "SFLOAT" ) );
+      break;
+    case CV_64F:
+      rbTypecode = rb_const_get( mHornetseye, rb_intern( "DFLOAT" ) );
+      break;
+    default:
+      ERRORMACRO( false, Error, ,
+                  "Encountered unknown type number "
+                  << CV_MAT_DEPTH( cvMat->type )
+                  << " while trying to convert to MultiArray" );
+    };
+    VALUE rbTypeSize = rb_funcall( rbTypecode, rb_intern( "storage_size" ), 0 );
+    int typesize = NUM2INT( rbTypeSize );
+    int size = width * height * channels * typesize;
+    VALUE cMalloc = rb_define_class_under( mHornetseye, "Malloc", rb_cObject );
+    VALUE rbDest = rb_funcall( cMalloc, rb_intern( "new" ), 1, INT2NUM( size ) );
+    char *dest; Data_Get_Struct( rbDest, char, dest );
+    copy( cvMat->data.ptr, cvMat->data.ptr + size, dest );
+    VALUE rbMultiArray = rb_funcall( mHornetseye, rb_intern( "MultiArray" ), 4,
+                                     rbTypecode, INT2NUM( channels ),
+                                     INT2NUM( width ), INT2NUM( height ) );
+    VALUE rbImport = rb_funcall( rbMultiArray, rb_intern( "new" ), 1, rbDest );
+    rbRetVal = rbImport;
+  } catch( std::exception &e ) {
+    rb_raise( rb_eRuntimeError, "%s", e.what() );
+  };
+  return rbRetVal;
+}
+

data/ext/cvmat.hh

@@ -0,0 +1,28 @@
+/* HornetsEye - Computer Vision with Ruby
+   Copyright (C) 2006, 2007, 2008, 2009 Jan Wedekind
+
+   This program is free software: you can redistribute it and/or modify
+   it under the terms of the GNU General Public License as published by
+   the Free Software Foundation, either version 3 of the License, or
+   (at your option) any later version.
+
+   This program is distributed in the hope that it will be useful,
+   but WITHOUT ANY WARRANTY; without even the implied warranty of
+   MERCHANTABILITY or FITNESS FOR A PARTICULAR PURPOSE.  See the
+   GNU General Public License for more details.
+
+   You should have received a copy of the GNU General Public License
+   along with this program.  If not, see <http://www.gnu.org/licenses/>. */
+#ifndef HORNETSEYE_CVMAT_HH
+#define HORNETSEYE_CVMAT_HH
+#include "rubyinc.hh"
+
+class CvMatExt {
+public:
+  static VALUE cRubyClass;
+  static VALUE registerRubyClass( VALUE module );
+  static void deleteRubyObject( void *ptr );
+  static VALUE wrapToMultiArray( VALUE rbSelf );
+};
+
+#endif

data/ext/error.hh

@@ -0,0 +1,50 @@
+/* HornetsEye - Computer Vision with Ruby
+   Copyright (C) 2006, 2007, 2008, 2009, 2010   Jan Wedekind
+
+   This program is free software: you can redistribute it and/or modify
+   it under the terms of the GNU General Public License as published by
+   the Free Software Foundation, either version 3 of the License, or
+   (at your option) any later version.
+
+   This program is distributed in the hope that it will be useful,
+   but WITHOUT ANY WARRANTY; without even the implied warranty of
+   MERCHANTABILITY or FITNESS FOR A PARTICULAR PURPOSE.  See the
+   GNU General Public License for more details.
+
+   You should have received a copy of the GNU General Public License
+   along with this program.  If not, see <http://www.gnu.org/licenses/>. */
+#ifndef ERROR_HH
+#define ERROR_HH
+
+#include <exception>
+#include <sstream>
+
+class Error: public std::exception
+{
+public:
+  Error(void) {}
+  Error( Error &e ): std::exception( e )
+    { m_message << e.m_message.str(); }
+  virtual ~Error(void) throw() {}
+  template< typename T >
+  std::ostream &operator<<( const T &t )
+    { m_message << t; return m_message; }
+  std::ostream &operator<<( std::ostream& (*__pf)(std::ostream&) )
+    { (*__pf)( m_message ); return m_message; }
+  virtual const char* what(void) const throw() {
+    temp = m_message.str();
+    return temp.c_str();
+  }
+protected:
+  std::ostringstream m_message;
+  mutable std::string temp;
+};
+
+#define ERRORMACRO( condition, class, params, message ) \
+  if ( !( condition ) ) {                               \
+    class _e params;                                    \
+    _e << message;                                      \
+    throw _e;                                           \
+  };
+
+#endif

data/ext/init.cc

@@ -0,0 +1,43 @@
+/* HornetsEye - Computer Vision with Ruby
+   Copyright (C) 2006, 2007, 2008, 2009, 2010   Jan Wedekind
+
+   This program is free software: you can redistribute it and/or modify
+   it under the terms of the GNU General Public License as published by
+   the Free Software Foundation, either version 3 of the License, or
+   (at your option) any later version.
+
+   This program is distributed in the hope that it will be useful,
+   but WITHOUT ANY WARRANTY; without even the implied warranty of
+   MERCHANTABILITY or FITNESS FOR A PARTICULAR PURPOSE.  See the
+   GNU General Public License for more details.
+
+   You should have received a copy of the GNU General Public License
+   along with this program.  If not, see <http://www.gnu.org/licenses/>. */
+#include "rubyinc.hh"
+#include "cvmat.hh"
+#include "node.hh"
+
+#ifdef WIN32
+#define DLLEXPORT __declspec(dllexport)
+#define DLLLOCAL
+#else
+#define DLLEXPORT __attribute__ ((visibility("default")))
+#define DLLLOCAL __attribute__ ((visibility("hidden")))
+#endif
+
+extern "C" DLLEXPORT void Init_hornetseye_opencv(void);
+
+extern "C" {
+
+  void Init_hornetseye_opencv(void)
+  {
+    rb_require( "opencv" );
+    rb_require( "multiarray" );
+    VALUE mOpenCV = rb_define_module( "OpenCV" );
+    VALUE mHornetseye = rb_define_module( "Hornetseye" );
+    CvMatExt::registerRubyClass( mOpenCV );
+    Node::registerRubyClass( mHornetseye );
+    rb_require( "hornetseye_opencv_ext.rb" );
+  }
+
+}

data/ext/node.cc

@@ -0,0 +1,103 @@
+/* HornetsEye - Computer Vision with Ruby
+   Copyright (C) 2006, 2007, 2008, 2009, 2010 Jan Wedekind
+
+   This program is free software: you can redistribute it and/or modify
+   it under the terms of the GNU General Public License as published by
+   the Free Software Foundation, either version 3 of the License, or
+   (at your option) any later version.
+
+   This program is distributed in the hope that it will be useful,
+   but WITHOUT ANY WARRANTY; without even the implied warranty of
+   MERCHANTABILITY or FITNESS FOR A PARTICULAR PURPOSE.  See the
+   GNU General Public License for more details.
+
+   You should have received a copy of the GNU General Public License
+   along with this program.  If not, see <http://www.gnu.org/licenses/>. */
+#include <algorithm>
+#include <boost/shared_array.hpp>
+#include <opencv/cxcore.h>
+#include "cvmat.hh"
+#include "error.hh"
+#include "rubytools.hh"
+#include "node.hh"
+
+using namespace boost;
+using namespace std;
+
+VALUE Node::mModule = Qnil;
+
+VALUE Node::cRubyClass = Qnil;
+
+VALUE Node::registerRubyClass( VALUE module )
+{
+  mModule = module;
+  cRubyClass = rb_define_class_under( module, "Node", rb_cObject );
+  rb_define_method( cRubyClass, "to_cvmat", RUBY_METHOD_FUNC( wrapToCvMat ), 0 );
+  return cRubyClass;
+}
+
+VALUE Node::wrapToCvMat( VALUE rbSelf )
+{
+  VALUE rbRetVal = Qnil;
+  try {
+    VALUE rbMalloc = rb_funcall( rbSelf, rb_intern( "memory" ), 0 );
+    VALUE rbTypecode = rb_funcall( rbSelf, rb_intern( "typecode" ), 0 );
+    VALUE rbShape = rb_funcall( rbSelf, rb_intern( "shape" ), 0 );
+    Check_Type( rbShape, T_ARRAY );
+    int rank = RARRAY_LEN(rbShape);
+    ERRORMACRO( rank == 3, Error, ,
+                "Array needs to have three dimensions for conversion to "
+                "CvMat (but had " << rank << " dimension(s))" );
+    int
+      channels = NUM2INT( RARRAY_PTR(rbShape)[0] ),
+      width = NUM2INT( RARRAY_PTR(rbShape)[1] ),
+      height = NUM2INT( RARRAY_PTR(rbShape)[2] );
+    int type = -1;
+    if ( rb_funcall( rbTypecode, rb_intern( "==" ), 1,
+                     rb_const_get( mModule, rb_intern( "UBYTE" ) ) ) ==
+         Qtrue )
+      type = CV_8U;
+    else if ( rb_funcall( rbTypecode, rb_intern( "==" ), 1,
+                          rb_const_get( mModule, rb_intern( "BYTE" ) ) ) ==
+              Qtrue )
+      type = CV_8S;
+    else if ( rb_funcall( rbTypecode, rb_intern( "==" ), 1,
+                          rb_const_get( mModule, rb_intern( "USINT" ) ) ) ==
+              Qtrue )
+      type = CV_16U;
+    else if ( rb_funcall( rbTypecode, rb_intern( "==" ), 1,
+                          rb_const_get( mModule, rb_intern( "SINT" ) ) ) ==
+              Qtrue )
+      type = CV_16S;
+    else if ( rb_funcall( rbTypecode, rb_intern( "==" ), 1,
+                          rb_const_get( mModule, rb_intern( "INT" ) ) ) ==
+              Qtrue )
+      type = CV_32S;
+    else if ( rb_funcall( rbTypecode, rb_intern( "==" ), 1,
+                          rb_const_get( mModule, rb_intern( "SFLOAT" ) ) ) ==
+              Qtrue )
+      type = CV_32F;
+    else if ( rb_funcall( rbTypecode, rb_intern( "==" ), 1,
+                          rb_const_get( mModule, rb_intern( "DFLOAT" ) ) ) ==
+              Qtrue )
+      type = CV_64F;
+    else {
+      ERRORMACRO( false, Error, ,
+                  "Conversion to CvMat only supports UBYTE, BYTE, USINT, "
+                  "SINT, INT, SFLOAT, DFLOAT as basic types" );
+    };
+    VALUE rbTypeSize = rb_funcall( rbSelf, rb_intern( "storage_size" ), 0 );
+    int typesize = NUM2INT( rbTypeSize );
+    VALUE cMalloc = rb_define_class_under( mModule, "Malloc", rb_cObject );
+    unsigned char *source;
+    dataGetStruct( rbMalloc, cMalloc, unsigned char, source );
+    CvMat *cvMat = cvCreateMat( height, width, CV_MAKETYPE( type, channels ) );
+    rbRetVal =
+      Data_Wrap_Struct( CvMatExt::cRubyClass, 0, CvMatExt::deleteRubyObject, cvMat );
+    copy( source, source + typesize, cvMat->data.ptr );
+  } catch ( std::exception &e ) {
+    rb_raise( rb_eRuntimeError, "%s", e.what() );
+  };
+  return rbRetVal;
+}
+