hornetseye-linalg 0.1.0

data/ext/node.hh

@@ -0,0 +1,29 @@
+/* HornetsEye - Computer Vision with Ruby
+   Copyright (C) 2006, 2007, 2008, 2009, 2010, 2011 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_NODE_HH
+#define HORNETSEYE_NODE_HH
+#include "rubyinc.hh"
+
+class Node {
+public:
+  static VALUE cRubyClass;
+  static VALUE mModule;
+  static VALUE registerRubyClass(VALUE module);
+  static VALUE wrapToDMatrix(VALUE rbSelf);
+  static VALUE wrapToSMatrix(VALUE rbSelf);
+};
+
+#endif

data/ext/rubyinc.hh

@@ -0,0 +1,54 @@
+/* HornetsEye - Computer Vision with Ruby
+   Copyright (C) 2006, 2007   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_RUBYINC_HH
+#define HORNETSEYE_RUBYINC_HH
+
+#ifdef RSHIFT
+#undef RSHIFT
+#endif
+
+#define gettimeofday rubygettimeofday
+#define timezone rubygettimezone
+#include <ruby.h>
+// #include <version.h>
+#undef timezone
+#undef gettimeofday
+#ifdef read
+#undef read
+#endif
+#ifdef write
+#undef write
+#endif
+#ifdef RGB
+#undef RGB
+#endif
+
+#ifndef RUBY_VERSION_NUMBER
+#define RUBY_VERSION_NUMBER ( RUBY_VERSION_MAJOR * 10000 + \
+                              RUBY_VERSION_MINOR * 100 + \
+                              RUBY_VERSION_TEENY )
+#endif
+
+#ifndef RUBY_METHOD_FUNC
+#define RUBY_METHOD_FUNC(func) ((VALUE (*)(ANYARGS))func)
+#endif
+
+#ifndef xfree
+#define xfree free
+#endif
+
+#endif
+

data/ext/rubytools.hh

@@ -0,0 +1,33 @@
+/* HornetsEye - Computer Vision with Ruby
+   Copyright (C) 2006, 2007   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_RUBYTOOLS_HH
+#define HORNETSEYE_RUBYTOOLS_HH
+
+#include <complex>
+#include "rubyinc.hh"
+
+void checkType( VALUE rbValue, VALUE rbClass );
+
+void checkStruct( VALUE rbValue, VALUE rbClass );
+
+#define dataGetStruct(obj,klass,type,sval) { \
+  checkStruct( obj, klass );               \
+  Data_Get_Struct( obj, type, sval );      \
+}
+
+#include "rubytools.tcc"
+
+#endif

data/ext/rubytools.tcc

@@ -0,0 +1,32 @@
+/* HornetsEye - Computer Vision with Ruby
+   Copyright (C) 2006, 2007   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 <cassert>
+#include "error.hh"
+
+inline void checkType( VALUE rbValue, VALUE rbClass )
+{
+  ERRORMACRO( rb_funcall( rbValue, rb_intern( "kind_of?" ), 1, rbClass ) ==
+              Qtrue, Error, ,
+              "Argument must be of class \"" << rb_class2name( rbClass )
+              << "\"." );
+}
+
+inline void checkStruct( VALUE rbValue, VALUE rbClass )
+{
+  Check_Type( rbValue, T_DATA );
+  checkType( rbValue, rbClass );
+}
+

data/ext/smatrix.cc

@@ -0,0 +1,60 @@
+/* HornetsEye - Computer Vision with Ruby
+   Copyright (C) 2006, 2007, 2008, 2009, 2010, 2011 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 "error.hh"
+#include "rubytools.hh"
+#include "smatrix.hh"
+
+using namespace std;
+
+VALUE SMatrix::mModule = Qnil;
+
+VALUE SMatrix::cRubyClass = Qnil;
+
+VALUE SMatrix::registerRubyClass(VALUE module)
+{
+  mModule = module;
+  cRubyClass = rb_define_class_under(module, "SMatrix", rb_cObject);
+  rb_define_method(cRubyClass, "to_multiarray", RUBY_METHOD_FUNC(wrapToMultiArray), 0);
+  return cRubyClass;
+}
+
+void SMatrix::deleteRubyObject(void *ptr)
+{
+  free((SMatrix *)ptr);
+}
+
+VALUE SMatrix::wrapToMultiArray(VALUE rbSelf)
+{
+  VALUE rbRetVal = Qnil;
+  try {
+    VALUE mHornetseye = rb_define_module( "Hornetseye" );
+    SMatrix_ *smatrix;
+    dataGetStruct(rbSelf, cRubyClass, SMatrix_, smatrix);
+    int width = smatrix->hsize;
+    int height = smatrix->vsize;
+    VALUE cMalloc = rb_define_class_under(mHornetseye, "Malloc", rb_cObject);
+    VALUE rbMemory = Data_Wrap_Struct(cMalloc, 0, 0, smatrix->data);
+    rb_ivar_set(rbMemory, rb_intern("@size"), INT2NUM(width * height * sizeof(float)));
+    VALUE rbTypecode = rb_const_get(mHornetseye, rb_intern("SFLOAT"));
+    rbRetVal = rb_funcall(rb_const_get(mHornetseye, rb_intern("MultiArray")),
+      rb_intern("import"), 4, rbTypecode, rbMemory, INT2NUM(height), INT2NUM(width));
+  } catch( std::exception &e ) {
+    rb_raise( rb_eRuntimeError, "%s", e.what() );
+  };
+  return rbRetVal;
+}

data/ext/smatrix.hh

@@ -0,0 +1,36 @@
+/* HornetsEye - Computer Vision with Ruby
+   Copyright (C) 2006, 2007, 2008, 2009, 2010, 2011 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_SMATRIX_HH
+#define HORNETSEYE_SMATRIX_HH
+#include "rubyinc.hh"
+
+typedef struct {
+  int vsize;
+  int hsize;
+  float *data;
+} SMatrix_;
+
+class SMatrix {
+public:
+  static VALUE cRubyClass;
+  static VALUE mModule;
+  static VALUE registerRubyClass(VALUE module);
+  static void deleteRubyObject(void *ptr);
+  static VALUE wrapToMultiArray(VALUE rbSelf);
+};
+
+#endif
+

data/lib/hornetseye-linalg/dmatrix.rb

@@ -0,0 +1,33 @@
+# hornetseye-linalg - Linalg integration for Hornetseye
+# Copyright (C) 2011 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/>.
+module Linalg
+
+  class DMatrix
+
+    alias_method :orig_to_multiarray, :to_multiarray
+
+    def to_multiarray
+      source = self
+      retval = orig_to_multiarray.roll
+      retval.instance_eval { @source = source }
+      retval
+    end
+
+  end
+
+end
+
+

data/lib/hornetseye-linalg/node.rb

@@ -0,0 +1,41 @@
+# hornetseye-linalg - Linalg integration for Hornetseye
+# Copyright (C) 2011 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/>.
+module Hornetseye
+
+  class Node
+
+    alias_method :orig_to_dmatrix, :to_dmatrix
+
+    def to_dmatrix
+      source = roll.to_dfloat.memorise
+      retval = source.orig_to_dmatrix
+      retval.instance_eval { @source = source }
+      retval
+    end
+
+    alias_method :orig_to_smatrix, :to_smatrix
+
+    def to_smatrix
+      source = roll.to_sfloat.memorise
+      retval = source.orig_to_smatrix
+      retval.instance_eval { @source = source }
+      retval
+    end
+
+  end
+
+end
+

data/lib/hornetseye-linalg/smatrix.rb

@@ -0,0 +1,34 @@
+# hornetseye-linalg - Linalg integration for Hornetseye
+# Copyright (C) 2011 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/>.
+module Linalg
+
+  class SMatrix
+
+    alias_method :orig_to_multiarray, :to_multiarray
+
+    def to_multiarray
+      source = self
+      retval = orig_to_multiarray.roll
+      retval.instance_eval { @source = source }
+      retval
+    end
+
+  end
+
+end
+
+
+

data/lib/hornetseye_linalg_ext.rb

@@ -0,0 +1,24 @@
+# hornetseye-hypercomplex - Linalg integration for Hornetseye
+# Copyright (C) 2011 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/>.
+
+require 'malloc'
+require 'multiarray'
+require 'linalg'
+
+require 'hornetseye-linalg/node'
+require 'hornetseye-linalg/dmatrix'
+require 'hornetseye-linalg/smatrix'
+

data/test/tc_hornetseye_linalg.rb

@@ -0,0 +1,62 @@
+# hornetseye-hypercomplex - Linalg integration for Hornetseye
+# Copyright (C) 2011 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/>.
+
+require 'test/unit'
+begin
+  require 'rubygems'
+rescue LoadError
+end
+Kernel::require 'hornetseye_linalg'
+
+class TC_Hornetseye_Linalg < Test::Unit::TestCase
+
+  D = Hornetseye::DFLOAT
+  S = Hornetseye::SFLOAT
+
+  def M( *args )
+    Hornetseye::MultiArray *args
+  end
+
+  def setup
+    @m1 = M(D, 2)[[1, 2, 3], [4, 5, 6]]
+    @m2 = M(S, 2)[[1, 2, 3], [4, 5, 6]]
+  end
+
+  def teardown
+  end
+
+  def test_to_dmatrix
+    d = @m1.to_dmatrix
+    for j in 0 ... @m1.shape[1]
+      for i in 0 ... @m1.shape[0]
+        assert_equal @m1[j][i], d[j, i]
+      end
+    end
+    assert_equal @m1, d.to_multiarray
+  end
+
+  def test_to_smatrix
+    s = @m2.to_smatrix
+    for j in 0 ... @m2.shape[1]
+      for i in 0 ... @m2.shape[0]
+        assert_equal @m2[j][i], s[j, i]
+      end
+    end
+    assert_equal @m2, s.to_multiarray
+  end
+
+end
+