ruby-opencv/ext/opencv/cvsurfpoint.cpp

/************************************************************

   cvsurfpoint.cpp -

   $Author: ser1zw $

   Copyright (C) 2011 ser1zw

************************************************************/
#include "cvsurfpoint.h"
/*
 * Document-class: OpenCV::CvSURFPoint
 *
 * C structure is here.
 *   typedef struct CvSURFPoint {
 *     CvPoint2D32f pt; // position of the feature within the image
 *     int laplacian;   // -1, 0 or +1. sign of the laplacian at the point.
 *                      // can be used to speedup feature comparison
 *                      // (normally features with laplacians of different
 *                      // signs can not match)
 *     int size;        // size of the feature
 *     float dir;       // orientation of the feature: 0..360 degrees
 *     float hessian;   // value of the hessian (can be used to
 *                      // approximately estimate the feature strengths)
 *   } CvSURFPoint;
 */
__NAMESPACE_BEGIN_OPENCV
__NAMESPACE_BEGIN_CVSURFPOINT

VALUE rb_klass;

VALUE
rb_class()
{
  return rb_klass;
}

void
define_ruby_class()
{
  if(rb_klass)
    return;
  /* 
   * opencv = rb_define_module("OpenCV");
   * 
   * note: this comment is used by rdoc.
   */
  VALUE opencv = rb_module_opencv();
  rb_klass = rb_define_class_under(opencv, "CvSURFPoint", rb_cObject);
  rb_define_alloc_func(rb_klass, rb_allocate);
  rb_define_private_method(rb_klass, "initialize", RUBY_METHOD_FUNC(rb_initialize), 5);
  rb_define_method(rb_klass, "pt", RUBY_METHOD_FUNC(rb_get_pt), 0);
  rb_define_method(rb_klass, "pt=", RUBY_METHOD_FUNC(rb_set_pt), 1);
  rb_define_method(rb_klass, "laplacian", RUBY_METHOD_FUNC(rb_get_laplacian), 0);
  rb_define_method(rb_klass, "laplacian=", RUBY_METHOD_FUNC(rb_set_laplacian), 1);
  rb_define_method(rb_klass, "size", RUBY_METHOD_FUNC(rb_get_size), 0);
  rb_define_method(rb_klass, "size=", RUBY_METHOD_FUNC(rb_set_size), 1);
  rb_define_method(rb_klass, "dir", RUBY_METHOD_FUNC(rb_get_dir), 0);
  rb_define_method(rb_klass, "dir=", RUBY_METHOD_FUNC(rb_set_dir), 1);
  rb_define_method(rb_klass, "hessian", RUBY_METHOD_FUNC(rb_get_hessian), 0);
  rb_define_method(rb_klass, "hessian=", RUBY_METHOD_FUNC(rb_set_hessian), 1);
}

VALUE
rb_allocate(VALUE klass)
{
  CvSURFPoint *ptr;
  return Data_Make_Struct(klass, CvSURFPoint, 0, -1, ptr);
}

/* 
 * call-seq:
 *   CvSURFPoint.new(<i>pt,laplacian,size,dir,hessian</i>) -> cvsurfpoint
 *
 * Create a CvSURFPoint
 */
VALUE
rb_initialize(VALUE self, VALUE pt, VALUE laplacian, VALUE size, VALUE dir, VALUE hessian)
{
  CvSURFPoint *self_ptr = CVSURFPOINT(self);
  self_ptr->pt = VALUE_TO_CVPOINT2D32F(pt);
  self_ptr->laplacian = FIX2INT(laplacian);
  self_ptr->size = FIX2INT(size);
  self_ptr->dir = (float)NUM2DBL(dir);
  self_ptr->hessian = (float)NUM2DBL(hessian);

  return self;
}

/*
 * call-seq:
 *   pt -> cvpoint2d32f
 * Return position of the feature as CvPoint2D32f.
 */
VALUE
rb_get_pt(VALUE self)
{
  return REFER_OBJECT(cCvPoint2D32f::rb_class(), &CVSURFPOINT(self)->pt, self);
}

/*
 * call-seq:
 *   pt = <i>value</i>
 *
 * Set position of the feature to <i>value</i>
 */
VALUE
rb_set_pt(VALUE self, VALUE value)
{
  if (!cCvPoint2D32f::rb_compatible_q(rb_klass, value))
    rb_raise(rb_eArgError, "object is not compatible %s.", rb_class2name(cCvPoint2D32f::rb_class()));
  CVSURFPOINT(self)->pt = VALUE_TO_CVPOINT2D32F(value);
  return self;
}

/*
 * call-seq:
 *   laplacian -> fixnum
 * Return sign of the laplacian at the point (-1, 0 or +1)
 */
VALUE
rb_get_laplacian(VALUE self)
{
  return INT2FIX(CVSURFPOINT(self)->laplacian);
}

/*
 * call-seq:
 *   laplacian = <i>value</i> (-1, 0 or +1)
 * Set sign of the laplacian at the point
 */
VALUE
rb_set_laplacian(VALUE self, VALUE value)
{
  int val = FIX2INT(value);
  CVSURFPOINT(self)->laplacian = (val > 0) ? 1 : (val < 0) ? -1 : 0;
  return self;
}

/*
 * call-seq:
 *   size -> fixnum
 * Return size of feature
 */
VALUE
rb_get_size(VALUE self)
{
  return INT2FIX(CVSURFPOINT(self)->size);
}

/*
 * call-seq:
 *   size = <i>value</i>
 * Return size of feature
 */
VALUE
rb_set_size(VALUE self, VALUE value)
{
  CVSURFPOINT(self)->size = FIX2INT(value);
  return self;
}

/*
 * call-seq:
 *   dir -> number
 * Return orientation of the feature: 0..360 degrees
 */
VALUE
rb_get_dir(VALUE self)
{
  return DBL2NUM((double)(CVSURFPOINT(self)->dir));
}

/*
 * call-seq:
 *   dir = <i>value</i>
 * Set orientation of the feature: 0..360 degrees
 */
VALUE
rb_set_dir(VALUE self, VALUE value)
{
  CVSURFPOINT(self)->dir = (float)NUM2DBL(value);
  return self;
}

/*
 * call-seq:
 *   hessian -> number
 * Return value of the hessian
 */
VALUE
rb_get_hessian(VALUE self)
{
  return DBL2NUM((double)(CVSURFPOINT(self)->hessian));
}

/*
 * call-seq:
 *   hessian = <i>value</i>
 * Set value of the hessian
 */
VALUE
rb_set_hessian(VALUE self, VALUE value)
{
  CVSURFPOINT(self)->hessian = (float)NUM2DBL(value);
  return self;
}

VALUE
new_object()
{
  return rb_allocate(rb_klass);
}

VALUE
new_object(CvSURFPoint* cvsurfpoint)
{
  VALUE object = rb_allocate(rb_klass);
  CvSURFPoint *ptr = CVSURFPOINT(object);
  ptr = cvsurfpoint;
  return object;
}

__NAMESPACE_END_CVSURFPOINT
__NAMESPACE_END_OPENCV