ruby-opencv/ext/cvrect.cpp

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

   cvrect.cpp -

   $Author: lsxi $

   Copyright (C) 2005-2006 Masakazu Yonekura

************************************************************/
#include "cvrect.h"
/*
 * Document-class: OpenCV::CvRect
 *
 * This class have coordinate of top-left point(x, y) and size, width and height.
 * 
 * 
 * C stracture is here, very simple.
 *   typdef struct CvRect{
 *     int x;
 *     int y;
 *     int width;
 *     int height;
 *   }
 * 
 */
__NAMESPACE_BEGIN_OPENCV
__NAMESPACE_BEGIN_CVRECT

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, "CvRect", rb_cObject);            
  rb_define_alloc_func(rb_klass, rb_allocate);
  rb_define_singleton_method(rb_klass, "compatible?", RUBY_METHOD_FUNC(rb_compatible_q), 1);
  rb_define_private_method(rb_klass, "initialize", RUBY_METHOD_FUNC(rb_initialize), -1);

  rb_define_method(rb_klass, "x", RUBY_METHOD_FUNC(rb_x), 0);
  rb_define_method(rb_klass, "x=", RUBY_METHOD_FUNC(rb_set_x), 1);
  rb_define_method(rb_klass, "y", RUBY_METHOD_FUNC(rb_y), 0);
  rb_define_method(rb_klass, "y=", RUBY_METHOD_FUNC(rb_set_y), 1);
  rb_define_method(rb_klass, "width", RUBY_METHOD_FUNC(rb_width), 0);
  rb_define_method(rb_klass, "width=", RUBY_METHOD_FUNC(rb_set_width), 1);
  rb_define_method(rb_klass, "height", RUBY_METHOD_FUNC(rb_height), 0);
  rb_define_method(rb_klass, "height=", RUBY_METHOD_FUNC(rb_set_height), 1);      
  rb_define_method(rb_klass, "center", RUBY_METHOD_FUNC(rb_center), 0);
  rb_define_method(rb_klass, "points", RUBY_METHOD_FUNC(rb_points), 0);
  rb_define_method(rb_klass, "top_left", RUBY_METHOD_FUNC(rb_top_left), 0);
  rb_define_method(rb_klass, "top_right", RUBY_METHOD_FUNC(rb_top_right), 0);
  rb_define_method(rb_klass, "bottom_left", RUBY_METHOD_FUNC(rb_bottom_left), 0);
  rb_define_method(rb_klass, "bottom_right", RUBY_METHOD_FUNC(rb_bottom_right), 0);
  rb_define_method(rb_klass, "or", RUBY_METHOD_FUNC(rb_or), 1);
  rb_define_alias(rb_klass, "|", "or");
}

/*
 * call-seq:
 *   combatible?(obj)
 *
 * Return compatibility to CvRect. Return true if object have method #x and #y and #width and #height.
 *
 * For example.
 *   class MyRect
 *     def x
 *       1
 *     end
 *     def y
 *       2
 *     end
 *     def width
 *       10
 *     end
 *     def height
 *       20
 *     end
 *   end
 *   mr = MyRect.new
 *   CvRect.compatible?(mp)  #=> true
 *   CvRect.new(mp)          #=> same as CvRect(1, 2, 10, 20)
 */
VALUE
rb_compatible_q(VALUE klass, VALUE object)
{
  return (rb_respond_to(object, rb_intern("x")) && rb_respond_to(object, rb_intern("y")) && rb_respond_to(object, rb_intern("width")) && rb_respond_to(object, rb_intern("height"))) ? Qtrue : Qfalse;
}

/*
 * call-seq:
 *   CvRect.bounding
 *
 */
VALUE
rb_bounding(VALUE klass, VALUE points)
{
  /* not yet */
  return Qnil;
}

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

/*
 * call-seq:
 *   new -> CvRect.new(0, 0, 0, 0)
 *   new(obj) -> CvRect.new(obj.x.to_i, obj.y.to_i, obj.width.to_i, obj.height.to_i)
 *   new(x, y, width, height)
 *
 * Create new rectangle area. (x, y) is top-left point, and width, height is size of area.
 * It is dropped below the decimal point.
 *
 * new() is same as new(0, 0, 0, 0)
 *
 * new(obj) is same as new(obj.x.to_i, obj.y.to_i, obj.width.to_i, obj.height.to_i)
 */
VALUE
rb_initialize(int argc, VALUE *argv, VALUE self)
{
  VALUE object, x, y, width, height;
  switch (argc) {
  case 0:
    break;
  case 1:
    object = argv[0];
    if(rb_compatible_q(rb_klass, object)) {
      CVRECT(self)->x = NUM2INT(rb_funcall(rb_funcall(object, rb_intern("x"), 0), rb_intern("to_i"), 0));
      CVRECT(self)->y = NUM2INT(rb_funcall(rb_funcall(object, rb_intern("y"), 0), rb_intern("to_i"), 0));
      CVRECT(self)->width = NUM2INT(rb_funcall(rb_funcall(object, rb_intern("width"), 0), rb_intern("to_i"), 0));
      CVRECT(self)->height = NUM2INT(rb_funcall(rb_funcall(object, rb_intern("height"), 0), rb_intern("to_i"), 0));
    }else{
      rb_raise(rb_eArgError, "object is not compatible %s.", rb_class2name(rb_klass));
    }
    break;
  case 4:
    x = argv[0], y = argv[1], width = argv[2], height = argv[3];
    CVRECT(self)->x = NUM2INT(x);
    CVRECT(self)->y = NUM2INT(y);
    CVRECT(self)->width = NUM2INT(width);
    CVRECT(self)->height = NUM2INT(height);
    break;
  default:
    rb_raise(rb_eArgError, "wrong number of arguments (%d for 0..2)", argc);
  }
  return Qnil;      
}

/*
 * Return parameter on x-axis of top-left point.
 */
VALUE
rb_x(VALUE self)
{
  return INT2FIX(CVRECT(self)->x);
}

/*
 * call-seq:
 *   x = val
 *
 * Set x-axis parameter of top-left point, return self.
 * It is dropped below the decimal point.
 */
VALUE
rb_set_x(VALUE self, VALUE x)
{
  CVRECT(self)->x = NUM2INT(x);
  return self;
}

/*
 * Return parameter on y-axis of top-left point.
 */
VALUE
rb_y(VALUE self)
{
  return INT2FIX(CVRECT(self)->y);
}

/*
 * call-seq:
 *   y = val
 *
 * Set y-axis parameter of top-left point, return self.
 * It is dropped below the decimal point.
 */
VALUE
rb_set_y(VALUE self, VALUE y)
{
  CVRECT(self)->y = NUM2INT(y);
  return self;
}

/*
 * Return size of x-axis.
 */
VALUE
rb_width(VALUE self)
{
  return INT2FIX(CVRECT(self)->width);
}

/*
 * call-seq:
 *   width = val
 *
 * Set x-axis size, return self.
 * It is dropped below the decimal point.
 */
VALUE
rb_set_width(VALUE self, VALUE x)
{
  CVRECT(self)->width = NUM2INT(x);
  return self;
}

/*
 * Return size of y-axis.
 */
VALUE
rb_height(VALUE self)
{
  return INT2FIX(CVRECT(self)->height);
}

/*
 * call-seq:
 *   height = val
 *
 * Set y-axis size, return self.
 * It is dropped below the decimal point.
 */
VALUE
rb_set_height(VALUE self, VALUE y)
{
  CVRECT(self)->height = NUM2INT(y);
  return self;
}

/*
 * Return center point of rectangle.
 */
VALUE
rb_center(VALUE self)
{
  CvRect *rect = CVRECT(self);  
  return cCvPoint2D32f::new_object(cvPoint2D32f((float)rect->x + (float)rect->width / 2.0,
                                                (float)rect->y + (float)rect->height / 2.0));
}

/*
 * Return 4 points (top-left, bottom-left, bottom-right, top-right)
 */
VALUE
rb_points(VALUE self)
{
  CvRect *rect = CVRECT(self);
  return rb_ary_new3(4,
                     cCvPoint::new_object(cvPoint(rect->x, rect->y)),
                     cCvPoint::new_object(cvPoint(rect->x, rect->y + rect->height)),
                     cCvPoint::new_object(cvPoint(rect->x + rect->width, rect->y + rect->height)),
                     cCvPoint::new_object(cvPoint(rect->x + rect->width, rect->y))
                     );
}

/*
 * Return top-left point of rectangle.
 */
VALUE
rb_top_left(VALUE self)
{
  return cCvPoint::new_object(cvPoint(CVRECT(self)->x, CVRECT(self)->y));
}

/*
 * Return top-right point of rectangle.
 */
VALUE
rb_top_right(VALUE self)
{
  return cCvPoint::new_object(cvPoint(CVRECT(self)->x + CVRECT(self)->width, CVRECT(self)->y));
}

/*
 * Return bottom-left point of rectangle.
 */
VALUE
rb_bottom_left(VALUE self)
{
  return cCvPoint::new_object(cvPoint(CVRECT(self)->x, CVRECT(self)->y + CVRECT(self)->height));
}

/*
 * Return bottom-right point of rectangle.
 */
VALUE
rb_bottom_right(VALUE self)
{
  return cCvPoint::new_object(cvPoint(CVRECT(self)->x + CVRECT(self)->width, CVRECT(self)->y + CVRECT(self)->height));
}

/*
 * call-seq:
 *   or(rect) -> cvrect
 *
 * Finds bounding rectangle for self and given rectangles.
 */
VALUE
rb_or(VALUE self, VALUE rect)
{
  return cCvRect::new_object(cvMaxRect(CVRECT(self), CVRECT(rect)));
}

VALUE
new_object(CvRect rect)
{
  VALUE object = rb_allocate(rb_klass);
  *CVRECT(object) = rect;
  return object;
}

__NAMESPACE_END_CVRECT
__NAMESPACE_END_OPENCV