opencv-ffi 0.0.1

data/lib/opencv-ffi-wrappers/core/misc_draw.rb

@@ -0,0 +1,44 @@
+
+require 'opencv-ffi-wrappers/core'
+require 'opencv-ffi-wrappers/core/point'
+
+module CVFFI
+
+  def self.draw_circle( img, point, opts={} )
+    color = opts[:color] || CVFFI::CvScalar.new( {:w=>255, :x=>255, :y=>255, :z=>0} )
+    thickness = opts[:thickness] || 5
+    radius = opts[:radius] || 1
+
+    CVFFI::cvCircle( img.to_IplImage, point.to_CvPoint, radius, color, thickness,8,0 )
+  end
+
+  def self.draw_point( img, point, opts={} )
+    opts[:thickness] = -1
+    draw_circle( img, point, opts )
+  end
+
+
+  def self.draw_line( img, aPoint, bPoint, opts )
+    color = opts[:color] || CVFFI::CvScalar.new( {:w=>255, :x=>255, :y=>255, :z=>0} )
+    thickness = opts[:thickness] || 5
+
+    CVFFI::cvLine( img.to_IplImage, aPoint.to_CvPoint, bPoint.to_CvPoint, color, thickness, 8, 0 )
+  end
+
+  def self.put_text( img, text, point, opts = {} )
+    color = opts[:color] || CVFFI::Scalar.new( 255,255,255,0 )
+    thickness = opts[:thickness] || 2
+    face = opts[:face] || opts[:typeface] || :CV_FONT_HERSHEY_SIMPLEX
+    font = opts[:font] || nil
+    hscale = opts[:hscale] || opts[:scale] || 1.0
+    vscale = opts[:vscale] || opts[:scale] || hscale
+    shear = opts[:shear] || 0.0
+
+    unless font
+      font = CVFFI::CvFont.new '\0'
+      CVFFI::cvInitFont( font, face, hscale, vscale, shear, thickness, :CV_AA )
+    end
+
+    CVFFI::cvPutText( img.to_IplImage, text, point.to_CvPoint, font, color.to_CvScalar )
+  end
+end

data/lib/opencv-ffi-wrappers/core/point.rb

@@ -0,0 +1,286 @@
+
+require 'opencv-ffi/core/types'
+
+module CVFFI
+
+  module CvPointMethods
+    def got_what_i_need(a)
+      a.class.method_defined?(:x) and a.class.method_defined?(:y) 
+    end
+
+    def /(a)
+      if got_what_i_need a
+        self.class.new( [ x.to_f/a.x.to_f, y.to_f/a.y.to_f ] )
+      else
+        self.class.new( [ x.to_f/a, y.to_f/a ] )
+      end
+    end
+
+    def *(a)
+      if got_what_i_need a
+        self.class.new( [ x*a.x, y*a.y ] )
+      else
+        self.class.new( [ x*a, y*a ] )
+      end
+    end
+
+    def -(a)
+      if got_what_i_need a
+        self.class.new( [ x.to_f-a.x, y.to_f-a.y ] )
+      else
+        self.class.new( [ x.to_f-a, y.to_f-a ] )
+      end
+    end
+ 
+    def +(a)
+      if  got_what_i_need a
+        self.class.new( [ x+a.x, y+a.y ] )
+      else
+        self.class.new( [ x+a, y+a ] )
+      end
+    end
+
+    def rotate( rads )
+      sa = Math::sin rads
+      ca = Math::cos rads
+      self.class.new( x*ca - y*sa,
+                      x*sa + y*ca )
+    end
+    
+    def ==(b)
+      @x == b.x and @y == b.y
+    end
+    def ===(b)
+      @x === b.x and @y === b.y
+    end
+
+    def to_Vector( homogeneous = true )
+      Vector.elements( to_a( homogeneous ) )
+    end
+    alias :to_vector :to_Vector
+
+    def to_a(homogeneous=true)
+      if homogeneous
+        [ @x/@w, @y/@w, 1.0 ]
+      else
+        [ @x, @y ]
+      end
+    end
+
+    def neighbor?( p, radius )
+      return false if (x-p.x).abs > radius or (y-p.y).abs > radius
+      return false if l2distance(p) > radius
+      true
+    end
+
+    def neighbor_rsquared?( p, rsquared )
+      return false if l2_squared_distance(p) > rsquared
+      true
+    end
+
+    def l2distance( b )
+      Math::sqrt( l2_squared_distance(b) )
+    end
+    alias :distance_to :l2distance
+
+    def l2_squared_distance( b )
+      (x-b.x)**2 + (y-b.y)**2
+    end
+ end
+
+  module CvPointCastMethods
+    def to_CvPoint2D64f
+      CvPoint2D64f.new( :x => x, :y => y )
+    end
+
+    def to_CvPoint2D32f
+      CvPoint2D32f.new( :x => x, :y => y )
+    end
+
+    def to_CvPoint
+      CvPoint.new( :x => x.to_i, :y => y.to_i )
+    end
+
+    def to_Point
+      Point.new( x, y )
+    end
+
+    def to_a(homogeneous=true)
+      if homogeneous
+        [ x, y, 1 ]
+      else
+        [x,y]
+      end
+    end
+  end
+
+  class CvPointBase
+    include CvPointMethods
+    include CvPointCastMethods
+  end
+
+  class CvPoint; def to_CvPoint; self; end; end
+  class CvPoint2D32f; def to_CvPoint2D32f; self; end; end
+  class CvPoint2D64f; def to_CvPoint2D64f; self; end; end
+
+  class Point 
+    include CvPointCastMethods
+    include CvPointMethods
+
+    attr_accessor :w, :y, :x
+
+    def initialize( *args )
+      if args.length == 2 and args[1] != nil
+        @x = args[0]
+        @y = args[1]
+      else
+        args = args.shift
+
+        case args
+        when Hash
+          @y = args[:y]
+          @x = args[:x]
+        when Array
+          @x = args[0]
+          @y = args[1]
+      else
+        @x = args.x
+        @y = args.y
+      end
+      end
+
+      @x = @x.to_f
+      @y = @y.to_f
+      @w = 1
+    end
+
+    def area
+      @y*@x
+    end
+ end
+
+#===========================================================
+  module CvPoint3DMethods
+    def got_what_i_need(a)
+      a.method_defined?(:x) and a.method_defined?(:y) and a.method_defined(:z)
+    end
+
+    def /(a)
+      if a.is_a? Point3D
+        self.class.new( [ x.to_f/a.x.to_f, y.to_f/a.y.to_f, z.to_f/a.z.to_f ] )
+      else
+        self.class.new( [ x.to_f/a, y.to_f/a, z.to_f/z ] )
+      end
+    end
+
+    def *(a)
+      if a.is_a? Point3D
+        self.class.new( [ x*a.x, y*a.y, z*a.z ] )
+      else
+        self.class.new( [ x*a, y*a, z*a.z ] )
+      end
+    end
+
+    def -(a)
+      if a.is_a? Point3D
+        self.class.new( [ x.to_f-a.x, y.to_f-a.y, z.to_f-a.z ] )
+      else
+        self.class.new( [ x.to_f-a, y.to_f-a, z.to_f-a ] )
+      end
+    end
+ 
+    def +(a)
+      if a.is_a? Point3D
+        self.class.new( [ x+a.x, y+a.y, z+a.z ] )
+      else
+        self.class.new( [ x+a, y+a, z+a ] )
+      end
+    end
+
+    def ==(b)
+      @x == b.x and @y == b.y and @z == b.z
+    end
+    def ===(b)
+      @x === b.x and @y === b.y and @z === b.z
+    end
+
+    def to_Vector( homogeneous = true )
+      Vector.elements( to_a(homogeneous) )
+    end
+    alias :to_vector :to_Vector
+
+    def to_a(homogeneous=true)
+      if homogeneous
+        [@x/@w, @y/@w, @z/@w, 1]
+      else
+        [@x, @y, @z]
+      end
+    end
+
+end
+  
+  module CvPoint3DCastMethods
+    def to_CvPoint3D64f
+      CvPoint3D64f.new( :x => x, :y => y, :z => z )
+    end
+
+    def to_CvPoint3D32f
+      CvPoint3D32f.new( :x => x, :y => y, :z => z )
+    end
+
+    def to_a( homogeneous = true )
+      if homogeneous
+        [x,y,z,1]
+      else
+        [x,y,z]
+      end
+    end
+  end
+
+  class CvPoint3DBase
+    include CvPoint3DMethods
+    include CvPoint3DCastMethods
+  end
+
+  class CvPoint3D32f; def to_CvPoint3D32f; self; end; end
+  class CvPoint3D64f; def to_CvPoint3D64f; self; end; end
+
+  class Point3D
+    include CvPoint3DCastMethods
+    include CvPoint3DMethods
+
+    attr_accessor :w, :z, :y, :x
+
+    def initialize( *args )
+      if args.length == 3
+        @x = args[0]
+        @y = args[1]
+        @z = args[2]
+      else
+        args = args.shift
+
+        case args
+        when Hash
+          @z = args[:z]
+          @y = args[:y]
+          @x = args[:x]
+        when Array
+          @x = args[0]
+          @y = args[1]
+          @z = args[2]
+      else
+        @x = args.x
+        @y = args.y
+        @z = args.z
+      end
+      end
+
+      @w = 1
+    end
+
+  end
+
+
+
+
+end

data/lib/opencv-ffi-wrappers/core/rect.rb

@@ -0,0 +1,40 @@
+
+require 'opencv-ffi-wrappers/core'
+
+module CVFFI
+
+  class Rect
+    attr_accessor :size, :origin
+
+    def initialize( args )
+      case args
+      when Array
+        @origin = Point.new( args[0..1] )
+        @size   = Size.new(  args[2..3] )
+      when Hash
+        if args[:size]
+          @size = Size.new args[:size]
+        else
+          @size = Size.new(args)
+        end
+
+        if args[:origin] 
+          @origin = Point.new( args[:origin] )
+        elsif args[:center]
+          @origin = Point.new( args[:center] - @size/2.0 )
+        else
+          @origin = Point.new(args)
+        end
+
+      else
+        @size = Size.new args.size
+        @origin = Point.new args.origin
+      end
+    end
+
+    def to_CvRect
+      CvRect.new( :x => @origin.x, :y => @origin.y,
+              :width => @size.width, :height => @size.height )
+    end
+  end
+end

data/lib/opencv-ffi-wrappers/core/scalar.rb

@@ -0,0 +1,104 @@
+
+require 'opencv-ffi/core/types'
+require 'opencv-ffi-wrappers/core'
+
+
+module CVFFI
+
+  module CvScalarFunctions
+    def self.included(base)
+      base.extend(ClassMethods)
+    end
+
+    def to_CvScalar
+      self
+    end
+
+    module ClassMethods 
+   end
+
+
+  end
+
+  class CvMat
+    include CvMatFunctions
+  end
+
+
+
+  class Scalar
+
+    def initialize( *args )
+      @order = :BGR
+      w=x=y=z=0
+
+      if args[0].is_a?( Hash )
+        args = args[0]
+
+        @order = args[:channel_order] if args[:channel_order]
+        a,b,c,d = color_symbols
+
+        w = args[:w] || args[a] || args[a.upcase] || 0
+        x = args[:x] || args[b] || args[b.upcase] || 0
+        y = args[:y] || args[c] || args[c.upcase] || 0
+        z = args[:z] || args[d] || args[d.upcase] || 0
+      else
+        w,x,y,z = args
+      end
+
+      @s = CVFFI::CvScalar.new( :w => w, :x => x, :y => y, :z => z )
+    end
+
+    def color_symbols
+      case @order
+      when :BGR then
+        [ :b, :g, :r, :a ]
+      else
+        raise "Hm, the Scalar::order is undefined"
+      end
+    end
+
+    def to_CvScalar
+      @s
+    end
+
+    def []=(i,x)
+      a,b,c,d = color_symbols
+
+      case i.downcase
+      when :w,a then
+        @s.w = x
+      when :x,b then
+        @s.x = x
+      when :y,c then
+        @s.y = x
+      when :z,d then
+        @s.z = x
+      else 
+        raise "Hm, don't understand index to CVFFI::Scalar[]="
+      end
+    end
+
+
+    def [](i)
+      a,b,c,d = color_symbols
+
+      case i.downcase
+      when :w,a then
+        @s.w
+      when :x,b then
+        @s.x
+      when :y,c then
+        @s.y
+      when :z,d then
+        @s.z
+      else
+        nil
+      end
+    end
+
+    def to_s
+      "CvScalar(#{ [@s.w, @s.w, @s.y, @s.z].join(',') })"
+    end
+  end
+end