opencv 0.0.6

Sign up to get free protection for your applications and to get access to all the features.
Files changed (117) hide show
  1. data/History.txt +5 -0
  2. data/License.txt +30 -0
  3. data/Manifest.txt +115 -0
  4. data/README.txt +47 -0
  5. data/Rakefile +34 -0
  6. data/examples/convexhull.rb +41 -0
  7. data/examples/face_detect.rb +25 -0
  8. data/examples/houghcircle.rb +23 -0
  9. data/examples/inpaint.png +0 -0
  10. data/examples/inpaint.rb +43 -0
  11. data/examples/paint.rb +72 -0
  12. data/examples/snake.rb +43 -0
  13. data/examples/stuff.jpg +0 -0
  14. data/ext/curve.cpp +103 -0
  15. data/ext/curve.h +34 -0
  16. data/ext/cvavgcomp.cpp +67 -0
  17. data/ext/cvavgcomp.h +39 -0
  18. data/ext/cvbox2d.cpp +114 -0
  19. data/ext/cvbox2d.h +53 -0
  20. data/ext/cvcapture.cpp +276 -0
  21. data/ext/cvcapture.h +54 -0
  22. data/ext/cvchain.cpp +184 -0
  23. data/ext/cvchain.h +43 -0
  24. data/ext/cvchaincode.cpp +49 -0
  25. data/ext/cvchaincode.h +43 -0
  26. data/ext/cvcircle32f.cpp +90 -0
  27. data/ext/cvcircle32f.h +53 -0
  28. data/ext/cvcondensation.cpp +230 -0
  29. data/ext/cvcondensation.h +49 -0
  30. data/ext/cvconnectedcomp.cpp +115 -0
  31. data/ext/cvconnectedcomp.h +46 -0
  32. data/ext/cvcontour.cpp +219 -0
  33. data/ext/cvcontour.h +47 -0
  34. data/ext/cvcontourtree.cpp +86 -0
  35. data/ext/cvcontourtree.h +41 -0
  36. data/ext/cvconvexitydefect.cpp +103 -0
  37. data/ext/cvconvexitydefect.h +42 -0
  38. data/ext/cverror.cpp +140 -0
  39. data/ext/cverror.h +79 -0
  40. data/ext/cvfont.cpp +173 -0
  41. data/ext/cvfont.h +56 -0
  42. data/ext/cvhaarclassifiercascade.cpp +159 -0
  43. data/ext/cvhaarclassifiercascade.h +41 -0
  44. data/ext/cvhistogram.cpp +200 -0
  45. data/ext/cvhistogram.h +51 -0
  46. data/ext/cvindex.cpp +73 -0
  47. data/ext/cvindex.h +40 -0
  48. data/ext/cvline.cpp +106 -0
  49. data/ext/cvline.h +52 -0
  50. data/ext/cvmat.cpp +4809 -0
  51. data/ext/cvmat.h +286 -0
  52. data/ext/cvmatnd.cpp +44 -0
  53. data/ext/cvmatnd.h +28 -0
  54. data/ext/cvmemstorage.cpp +64 -0
  55. data/ext/cvmemstorage.h +53 -0
  56. data/ext/cvmoments.cpp +204 -0
  57. data/ext/cvmoments.h +48 -0
  58. data/ext/cvpoint.cpp +229 -0
  59. data/ext/cvpoint.h +59 -0
  60. data/ext/cvpoint2d32f.cpp +213 -0
  61. data/ext/cvpoint2d32f.h +61 -0
  62. data/ext/cvpoint3d32f.cpp +245 -0
  63. data/ext/cvpoint3d32f.h +64 -0
  64. data/ext/cvrect.cpp +340 -0
  65. data/ext/cvrect.h +79 -0
  66. data/ext/cvscalar.cpp +227 -0
  67. data/ext/cvscalar.h +63 -0
  68. data/ext/cvseq.cpp +583 -0
  69. data/ext/cvseq.h +71 -0
  70. data/ext/cvset.cpp +63 -0
  71. data/ext/cvset.h +39 -0
  72. data/ext/cvsize.cpp +223 -0
  73. data/ext/cvsize.h +63 -0
  74. data/ext/cvsize2d32f.cpp +180 -0
  75. data/ext/cvsize2d32f.h +59 -0
  76. data/ext/cvslice.cpp +82 -0
  77. data/ext/cvslice.h +53 -0
  78. data/ext/cvsparsemat.cpp +44 -0
  79. data/ext/cvsparsemat.h +28 -0
  80. data/ext/cvtermcriteria.cpp +183 -0
  81. data/ext/cvtermcriteria.h +71 -0
  82. data/ext/cvtwopoints.cpp +98 -0
  83. data/ext/cvtwopoints.h +50 -0
  84. data/ext/cvvector.cpp +206 -0
  85. data/ext/cvvector.h +54 -0
  86. data/ext/cvvideowriter.cpp +116 -0
  87. data/ext/cvvideowriter.h +41 -0
  88. data/ext/extconf.rb +61 -0
  89. data/ext/gui.cpp +65 -0
  90. data/ext/gui.h +33 -0
  91. data/ext/iplconvkernel.cpp +177 -0
  92. data/ext/iplconvkernel.h +52 -0
  93. data/ext/iplimage.cpp +238 -0
  94. data/ext/iplimage.h +54 -0
  95. data/ext/mouseevent.cpp +184 -0
  96. data/ext/mouseevent.h +59 -0
  97. data/ext/opencv.cpp +481 -0
  98. data/ext/opencv.h +356 -0
  99. data/ext/point3dset.cpp +41 -0
  100. data/ext/point3dset.h +31 -0
  101. data/ext/pointset.cpp +238 -0
  102. data/ext/pointset.h +69 -0
  103. data/ext/trackbar.cpp +122 -0
  104. data/ext/trackbar.h +65 -0
  105. data/ext/window.cpp +368 -0
  106. data/ext/window.h +56 -0
  107. data/images/CvMat_sobel.png +0 -0
  108. data/images/CvMat_sub_rect.png +0 -0
  109. data/images/CvSeq_relationmap.png +0 -0
  110. data/images/face_detect_from_lena.jpg +0 -0
  111. data/lib/opencv.rb +3 -0
  112. data/lib/version.rb +3 -0
  113. data/setup/setup.cygwin.rb +120 -0
  114. data/setup/setup.mingw.rb +99 -0
  115. data/setup/setup.mswin32.rb +103 -0
  116. data/test/test_opencv.rb +4 -0
  117. metadata +191 -0
@@ -0,0 +1,227 @@
1
+ /************************************************************
2
+
3
+ cvscalar.cpp -
4
+
5
+ $Author: lsxi $
6
+
7
+ Copyright (C) 2005 Masakazu Yonekura
8
+
9
+ ************************************************************/
10
+ #include "cvscalar.h"
11
+ /*
12
+ * Document-class: OpenCV::CvScalar
13
+ *
14
+ * Element-value of one pixel.
15
+ * OpenCV supports the image of 4-channels in the maximum.
16
+ * Therefore, CvScalar has 4-values.
17
+ *
18
+ * C structure is here, very simple.
19
+ * typdef struct CvScalar{
20
+ * double val[4];
21
+ * }CvScalar;
22
+ *
23
+ * If obtain CvScalar-object from the method of CvMat(or IplImage),
24
+ * the channel outside the range is obtained as all 0.
25
+ *
26
+ * image = IplImage::load("opencv.jpg") #=> 3-channel 8bit-depth BGR image
27
+ * pixel = image[10, 20] #=> Get pixel value of (10, 20) of image. pixel is CvScalar-object.
28
+ * blue, green, red = pixel[0], pixel[1], pixel[2]
29
+ * # pixel[3] always 0.
30
+ *
31
+ * CvColor is alias of CvScalar.
32
+ */
33
+ __NAMESPACE_BEGIN_OPENCV
34
+ __NAMESPACE_BEGIN_CVSCALAR
35
+
36
+
37
+ VALUE rb_klass;
38
+
39
+ VALUE
40
+ rb_class()
41
+ {
42
+ return rb_klass;
43
+ }
44
+
45
+ void
46
+ define_ruby_class()
47
+ {
48
+ if (rb_klass)
49
+ return;
50
+ /*
51
+ * opencv = rb_define_module("OpenCV");
52
+ *
53
+ * note: this comment is used by rdoc.
54
+ */
55
+ VALUE opencv = rb_module_opencv();
56
+
57
+ rb_klass = rb_define_class_under(opencv, "CvScalar", rb_cObject);
58
+ /* CvScalar: class */
59
+ rb_define_const(opencv, "CvColor", rb_klass);
60
+ rb_define_alloc_func(rb_klass, rb_allocate);
61
+ rb_define_method(rb_klass, "initialize", RUBY_METHOD_FUNC(rb_initialize), -1);
62
+ rb_define_method(rb_klass, "[]", RUBY_METHOD_FUNC(rb_aref), 1);
63
+ rb_define_method(rb_klass, "[]=", RUBY_METHOD_FUNC(rb_aset), 2);
64
+ rb_define_method(rb_klass, "sub", RUBY_METHOD_FUNC(rb_sub), -1);
65
+ rb_define_alias(rb_klass, "-", "sub");
66
+
67
+ rb_define_method(rb_klass, "to_s", RUBY_METHOD_FUNC(rb_to_s), 0);
68
+ rb_define_method(rb_klass, "to_ary", RUBY_METHOD_FUNC(rb_to_ary), 0);
69
+
70
+ rb_define_const(rb_klass, "Black", cCvScalar::new_object(cvScalar(0x0,0x0,0x0)));
71
+ rb_define_const(rb_klass, "Silver", cCvScalar::new_object(cvScalar(0x0c,0x0c,0x0c)));
72
+ rb_define_const(rb_klass, "Gray", cCvScalar::new_object(cvScalar(0x80,0x80,0x80)));
73
+ rb_define_const(rb_klass, "White", cCvScalar::new_object(cvScalar(0xff,0xff,0xff)));
74
+ rb_define_const(rb_klass, "Maroon", cCvScalar::new_object(cvScalar(0x0,0x0,0x80)));
75
+ rb_define_const(rb_klass, "Red", cCvScalar::new_object(cvScalar(0x0,0x0,0xff)));
76
+ rb_define_const(rb_klass, "Purple", cCvScalar::new_object(cvScalar(0x80,0x0,0x80)));
77
+ rb_define_const(rb_klass, "Fuchsia", cCvScalar::new_object(cvScalar(0xff,0x0,0xff)));
78
+ rb_define_const(rb_klass, "Green", cCvScalar::new_object(cvScalar(0x0,0x80,0x0)));
79
+ rb_define_const(rb_klass, "Lime", cCvScalar::new_object(cvScalar(0x0,0xff,0x0)));
80
+ rb_define_const(rb_klass, "Olive", cCvScalar::new_object(cvScalar(0x0,0x80,0x80)));
81
+ rb_define_const(rb_klass, "Yellow", cCvScalar::new_object(cvScalar(0x0,0xff,0xff)));
82
+ rb_define_const(rb_klass, "Navy", cCvScalar::new_object(cvScalar(0x80,0x0,0x0)));
83
+ rb_define_const(rb_klass, "Blue", cCvScalar::new_object(cvScalar(0xff,0x0,0x0)));
84
+ rb_define_const(rb_klass, "Teal", cCvScalar::new_object(cvScalar(0x80,0x80,0x0)));
85
+ rb_define_const(rb_klass, "Aqua", cCvScalar::new_object(cvScalar(0xff,0xff,0x0)));
86
+ }
87
+
88
+ VALUE
89
+ rb_allocate(VALUE klass)
90
+ {
91
+ CvScalar *ptr;
92
+ return Data_Make_Struct(klass, CvScalar, 0, -1, ptr);
93
+ }
94
+
95
+ /*
96
+ * call-seq:
97
+ * new(<i>[d1][,d2][,d3][,d4]</i>)
98
+ *
99
+ * Create new Scalar. Argument should be Fixnum (or nil as 0).
100
+ */
101
+ VALUE
102
+ rb_initialize(int argc, VALUE *argv, VALUE self)
103
+ {
104
+ VALUE val[4];
105
+ rb_scan_args(argc, argv, "04", &val[0], &val[1], &val[2], &val[3]);
106
+ for (int i = 0; i < 4; i++) {
107
+ CVSCALAR(self)->val[i] = NIL_P(val[i]) ? 0 : NUM2DBL(val[i]);
108
+ }
109
+ return self;
110
+ }
111
+
112
+ /*
113
+ * call-seq:
114
+ * [<i>index</i>]
115
+ *
116
+ * Return value of <i>index</i> dimension.
117
+ */
118
+ VALUE
119
+ rb_aref(VALUE self, VALUE index)
120
+ {
121
+ int idx = NUM2INT(index);
122
+ if (!(idx < 0) && idx < 4) {
123
+ return rb_float_new(CVSCALAR(self)->val[idx]);
124
+ }else{
125
+ rb_raise(rb_eIndexError, "scalar index should be 0...4");
126
+ }
127
+ }
128
+
129
+ /*
130
+ * call-seq:
131
+ * [<i>index</i>] = <i>value</i>
132
+ *
133
+ * Set value of <i>index</i> dimension to <i>value</i>
134
+ */
135
+ VALUE
136
+ rb_aset(VALUE self, VALUE index, VALUE value)
137
+ {
138
+ int idx = NUM2INT(index);
139
+ if (!(idx < 0) && idx < 4) {
140
+ CVSCALAR(self)->val[idx] = NUM2DBL(value);
141
+ return self;
142
+ }else{
143
+ rb_raise(rb_eIndexError, "scalar index should be 0...4");
144
+ }
145
+ }
146
+
147
+ /*
148
+ * call-seq:
149
+ * sub(val[,mask])
150
+ *
151
+ * Return new CvScalar if <i>val</i> is CvScalar or compatible object.
152
+ * self[I] - val[I]
153
+ * Or return new CvMat if <i>val</i> is CvMat or subclass.
154
+ */
155
+ VALUE
156
+ rb_sub(int argc, VALUE *argv, VALUE self)
157
+ {
158
+ VALUE val, mask;
159
+ rb_scan_args(argc, argv, "11", &val, &mask);
160
+ if(rb_obj_is_kind_of(val, cCvMat::rb_class())){
161
+ VALUE dest = cCvMat::new_object(cvGetSize(CVARR(val)), cvGetElemType(CVARR(val)));
162
+ cvSubRS(CVARR(val), *CVSCALAR(self), CVARR(dest), MASK(mask));
163
+ return dest;
164
+ }else{
165
+ CvScalar *src = CVSCALAR(self), scl = VALUE_TO_CVSCALAR(val);
166
+ return new_object(cvScalar(src->val[0] - scl.val[0],
167
+ src->val[1] - scl.val[1],
168
+ src->val[2] - scl.val[2],
169
+ src->val[3] - scl.val[3]));
170
+ }
171
+ }
172
+
173
+ /*
174
+ * call-seq:
175
+ * to_s -> "<OpeCV::CvScalar:#{self[0]},#{self[1]},#{self[2]},#{self[3]}>"
176
+ *
177
+ * Return values by String.
178
+ */
179
+ VALUE
180
+ rb_to_s(VALUE self)
181
+ {
182
+ const int i = 6;
183
+ VALUE str[i];
184
+ str[0] = rb_str_new2("<%s:%d,%d,%d,%d>");
185
+ str[1] = rb_str_new2(rb_class2name(CLASS_OF(self)));
186
+ str[2] = rb_aref(self, INT2FIX(0));
187
+ str[3] = rb_aref(self, INT2FIX(1));
188
+ str[4] = rb_aref(self, INT2FIX(2));
189
+ str[5] = rb_aref(self, INT2FIX(3));
190
+ return rb_f_sprintf(i, str);
191
+ }
192
+
193
+ /*
194
+ * call-seq:
195
+ * to_ary -> [self[0],self[1],self[2],self[3]]
196
+ *
197
+ * Return values by Array.
198
+ */
199
+ VALUE
200
+ rb_to_ary(VALUE self)
201
+ {
202
+ return rb_ary_new3(4,
203
+ rb_aref(self, INT2FIX(0)),
204
+ rb_aref(self, INT2FIX(1)),
205
+ rb_aref(self, INT2FIX(2)),
206
+ rb_aref(self, INT2FIX(3)));
207
+ }
208
+
209
+ VALUE
210
+ new_object()
211
+ {
212
+ VALUE object = rb_allocate(rb_klass);
213
+ *CVSCALAR(object) = cvScalar(0);
214
+ return object;
215
+ }
216
+
217
+ VALUE
218
+ new_object(CvScalar scalar)
219
+ {
220
+ VALUE object = rb_allocate(rb_klass);
221
+ *CVSCALAR(object) = scalar;
222
+ return object;
223
+ }
224
+
225
+ __NAMESPACE_END_CVSCALAR
226
+ __NAMESPACE_END_OPENCV
227
+
@@ -0,0 +1,63 @@
1
+ /************************************************************
2
+
3
+ cvscalar.h -
4
+
5
+ $Author: lsxi $
6
+
7
+ Copyright (C) 2005 Masakazu Yonekura
8
+
9
+ ************************************************************/
10
+ #ifndef RUBY_OPENCV_CVSCALAR_H
11
+ #define RUBY_OPENCV_CVSCALAR_H
12
+
13
+ #include "opencv.h"
14
+
15
+ #define __NAMESPACE_BEGIN_CVSCALAR namespace cCvScalar{
16
+ #define __NAMESPACE_END_CVSCALAR }
17
+
18
+ __NAMESPACE_BEGIN_OPENCV
19
+ __NAMESPACE_BEGIN_CVSCALAR
20
+
21
+ VALUE rb_class();
22
+
23
+ void define_ruby_class();
24
+
25
+ VALUE rb_compatible_q(VALUE klass, VALUE object);
26
+
27
+ VALUE rb_allocate(VALUE klass);
28
+ VALUE rb_initialize(int argc, VALUE *argv, VALUE self);
29
+
30
+ VALUE rb_aref(VALUE self, VALUE index);
31
+ VALUE rb_aset(VALUE self, VALUE index, VALUE value);
32
+ VALUE rb_sub(int argc, VALUE *argv, VALUE self);
33
+
34
+ VALUE rb_to_s(VALUE self);
35
+ VALUE rb_to_ary(VALUE self);
36
+
37
+ VALUE new_object();
38
+ VALUE new_object(CvScalar scalar);
39
+
40
+ __NAMESPACE_END_CVSCALAR
41
+
42
+ inline CvScalar*
43
+ CVSCALAR(VALUE object)
44
+ {
45
+ CvScalar *ptr;
46
+ Data_Get_Struct(object, CvScalar, ptr);
47
+ return ptr;
48
+ }
49
+
50
+ inline CvScalar
51
+ VALUE_TO_CVSCALAR(VALUE object)
52
+ {
53
+ if(FIXNUM_P(object))
54
+ return cvScalarAll(FIX2INT(object));
55
+ return cvScalar(NUM2INT(rb_funcall(object, rb_intern("[]"), 1, INT2FIX(0))),
56
+ NUM2INT(rb_funcall(object, rb_intern("[]"), 1, INT2FIX(1))),
57
+ NUM2INT(rb_funcall(object, rb_intern("[]"), 1, INT2FIX(2))),
58
+ NUM2INT(rb_funcall(object, rb_intern("[]"), 1, INT2FIX(3))));
59
+ }
60
+
61
+ __NAMESPACE_END_OPENCV
62
+
63
+ #endif // RUBY_OPENCV_CVSCALAR_H
@@ -0,0 +1,583 @@
1
+ /************************************************************
2
+
3
+ cvseq.cpp -
4
+
5
+ $Author: lsxi $
6
+
7
+ Copyright (C) 2005-2006 Masakazu Yonekura
8
+
9
+ ************************************************************/
10
+ #include "cvseq.h"
11
+ /*
12
+ * Document-class: OpenCV::CvSeq
13
+ *
14
+ * Generic Sequence class. CvSeq has the method like Array (push, pop, select, etc...).
15
+ * But, CvSeq cannot store the object of a different class.
16
+ * When storing object in CvSeq, conversion is automatically tried,
17
+ * and the object occurs the error if it cannot be done.
18
+ *
19
+ * e.g.
20
+ * seq = CvSeq.new(CvPoint) # Argument mean "this sequence contain only this class's object"
21
+ * seq.push(CvPoint.new(0, 0)) # => it's ok
22
+ * seq.push("hello") # => try convert "hello" to CvPoint. but can't do it. raise error.
23
+ *
24
+ * If the sequecne contain object of class A.
25
+ * When storing object(named "obj") of class B to the sequence.
26
+ * Try automatically : A.from_B(obj) => object of class A.
27
+ *
28
+ * CvSeq has the circulation structure internally.
29
+ * That is, when the sequence has three values ("a","b","c"),
30
+ * seq[0] and seq[3] are same "a", and seq[-1] and seq[2] are same "c".
31
+ *
32
+ * The sequence might have another sequence outside. see below.
33
+ * Each sequece has h_prev, h_next, v_prev, v_next method.
34
+ * If the adjoining sequence exists, each method return the adjoining sequence.
35
+ * Otherwise return nil.
36
+ *
37
+ * link:../images/CvSeq_relationmap.png
38
+ */
39
+ __NAMESPACE_BEGIN_OPENCV
40
+ __NAMESPACE_BEGIN_CVSEQ
41
+
42
+ // contain sequence-block class
43
+ st_table *seqblock_klass = st_init_numtable();
44
+
45
+ VALUE
46
+ seqblock_class(void *ptr)
47
+ {
48
+ VALUE klass;
49
+ if(!st_lookup(seqblock_klass, (st_data_t)ptr, (st_data_t*)&klass)){
50
+ rb_raise(rb_eTypeError, "Invalid sequence error.");
51
+ }
52
+ return klass;
53
+ }
54
+
55
+ VALUE rb_klass;
56
+
57
+ VALUE
58
+ rb_class()
59
+ {
60
+ return rb_klass;
61
+ }
62
+
63
+ void
64
+ define_ruby_class()
65
+ {
66
+ if(rb_klass)
67
+ return;
68
+ /*
69
+ * opencv = rb_define_module("OpenCV");
70
+ *
71
+ * note: this comment is used by rdoc.
72
+ */
73
+ VALUE opencv = rb_module_opencv();
74
+ rb_klass = rb_define_class_under(opencv, "CvSeq", rb_cObject);
75
+ rb_define_alloc_func(rb_klass, rb_allocate);
76
+ rb_define_private_method(rb_klass, "initialize", RUBY_METHOD_FUNC(rb_initialize), -1);
77
+ rb_define_method(rb_klass, "total", RUBY_METHOD_FUNC(rb_total), 0);
78
+ rb_define_alias(rb_klass, "length", "total");
79
+ rb_define_alias(rb_klass, "size", "total");
80
+ rb_define_method(rb_klass, "empty?", RUBY_METHOD_FUNC(rb_empty_q), 0);
81
+ rb_define_method(rb_klass, "[]", RUBY_METHOD_FUNC(rb_aref), 1);
82
+ rb_define_method(rb_klass, "first", RUBY_METHOD_FUNC(rb_first), 0);
83
+ rb_define_method(rb_klass, "last", RUBY_METHOD_FUNC(rb_last), 0);
84
+ rb_define_method(rb_klass, "index", RUBY_METHOD_FUNC(rb_index), 1);
85
+
86
+ rb_define_method(rb_klass, "h_prev", RUBY_METHOD_FUNC(rb_h_prev), 0);
87
+ rb_define_method(rb_klass, "h_next", RUBY_METHOD_FUNC(rb_h_next), 0);
88
+ rb_define_method(rb_klass, "v_prev", RUBY_METHOD_FUNC(rb_v_prev), 0);
89
+ rb_define_method(rb_klass, "v_next", RUBY_METHOD_FUNC(rb_v_next), 0);
90
+
91
+ rb_define_method(rb_klass, "push", RUBY_METHOD_FUNC(rb_push), -2);
92
+ rb_define_alias(rb_klass, "<<", "push");
93
+ rb_define_method(rb_klass, "pop", RUBY_METHOD_FUNC(rb_pop), 0);
94
+ rb_define_method(rb_klass, "unshift", RUBY_METHOD_FUNC(rb_unshift), -2);
95
+ rb_define_alias(rb_klass, "push_front", "unshift");
96
+ rb_define_method(rb_klass, "shift", RUBY_METHOD_FUNC(rb_shift), 0);
97
+ rb_define_alias(rb_klass, "pop_front", "shift");
98
+ rb_define_method(rb_klass, "each", RUBY_METHOD_FUNC(rb_each), 0);
99
+ rb_define_method(rb_klass, "each_index", RUBY_METHOD_FUNC(rb_each_index), 0);
100
+ rb_define_method(rb_klass, "each_with_index", RUBY_METHOD_FUNC(rb_each_with_index), 0);
101
+ rb_define_method(rb_klass, "insert", RUBY_METHOD_FUNC(rb_insert), 2);
102
+ rb_define_method(rb_klass, "remove", RUBY_METHOD_FUNC(rb_remove), 1);
103
+ rb_define_method(rb_klass, "clear", RUBY_METHOD_FUNC(rb_clear), 0);
104
+
105
+ rb_define_alias(rb_klass, "delete_at", "remove");
106
+ }
107
+
108
+ VALUE
109
+ rb_allocate(VALUE klass)
110
+ {
111
+ return Data_Wrap_Struct(klass, mark_root_object, free, 0);
112
+ }
113
+
114
+ void
115
+ free(void *ptr)
116
+ {
117
+ if(ptr){
118
+ unresist_object(ptr);
119
+ st_delete(seqblock_klass, (st_data_t*)&ptr, 0);
120
+ }
121
+ }
122
+
123
+ void
124
+ resist_class_information_of_sequence(CvSeq *seq, VALUE klass)
125
+ {
126
+ st_insert(seqblock_klass, (st_data_t)seq, (st_data_t)klass);
127
+ }
128
+
129
+ /*
130
+ * call-seq:
131
+ * CvSeq.new(<i>type[,storage]</i>)
132
+ *
133
+ * Return a new CvSeq. <i>type</i> should be following classes.
134
+ *
135
+ * * CvIndex
136
+ * * CvPoint
137
+ */
138
+ VALUE
139
+ rb_initialize(int argc, VALUE *argv, VALUE self)
140
+ {
141
+ VALUE klass, storage;
142
+ rb_scan_args(argc, argv, "11", &klass, &storage);
143
+ if(!rb_obj_is_kind_of(klass, rb_cClass))
144
+ rb_raise(rb_eTypeError, "argument 1 (sequence-block class) should be %s.", rb_class2name(rb_cClass));
145
+ CvSeq *seq = 0;
146
+ storage = CHECK_CVMEMSTORAGE(storage);
147
+ int type = 0, size = 0;
148
+ if(klass == cCvIndex::rb_class()){
149
+ type = CV_SEQ_ELTYPE_INDEX;
150
+ size = sizeof(CvIndex);
151
+ }else if(klass == cCvPoint::rb_class()){
152
+ type = CV_SEQ_ELTYPE_POINT;
153
+ size = sizeof(CvPoint);
154
+ }else if(klass == cCvPoint2D32f::rb_class()){
155
+ type = CV_SEQ_ELTYPE_POINT;
156
+ size = sizeof(CvPoint2D32f);
157
+ }else if(klass == cCvPoint3D32f::rb_class()){
158
+ type = CV_SEQ_ELTYPE_POINT3D;
159
+ size = sizeof(CvPoint3D32f);
160
+ }
161
+ auto_extend(self);
162
+ // todo: more various class will be support.
163
+ if(!size)
164
+ rb_raise(rb_eTypeError, "unsupport %s class for sequence-block.", rb_class2name(klass));
165
+ seq = cvCreateSeq(type, sizeof(CvSeq), size, CVMEMSTORAGE(storage));
166
+ DATA_PTR(self) = seq;
167
+ resist_root_object(seq, storage);
168
+ // resist class information of this sequence.
169
+ st_insert(seqblock_klass, (st_data_t)seq, (st_data_t)klass);
170
+ return self;
171
+ }
172
+
173
+ /*
174
+ * call-seq:
175
+ * total -> int
176
+ *
177
+ * Return total number of sequence-block.
178
+ */
179
+ VALUE
180
+ rb_total(VALUE self)
181
+ {
182
+ return INT2FIX(CVSEQ(self)->total);
183
+ }
184
+
185
+ /*
186
+ * call-seq:
187
+ * empty? -> true or false.
188
+ *
189
+ * Return <tt>true</tt> if contain no object, otherwize return <tt>false</tt>.
190
+ */
191
+ VALUE
192
+ rb_empty_q(VALUE self)
193
+ {
194
+ return CVSEQ(self)->total == 0 ? Qtrue : Qfalse;
195
+ }
196
+
197
+ /*
198
+ * call-seq:
199
+ * [<i>index</i>] -> obj or nil
200
+ *
201
+ * Return sequence-block at <i>index</i>.
202
+ */
203
+ VALUE
204
+ rb_aref(VALUE self, VALUE index)
205
+ {
206
+ CvSeq *seq = CVSEQ(self);
207
+ if(!(seq->total > 0))
208
+ return Qnil;
209
+ return REFER_OBJECT(seqblock_class(seq), cvGetSeqElem(seq, NUM2INT(index) % seq->total), self);
210
+ }
211
+
212
+ /*
213
+ * call-seq:
214
+ * first -> obj or nil
215
+ *
216
+ * Return first sequence-block.
217
+ */
218
+ VALUE
219
+ rb_first(VALUE self)
220
+ {
221
+ CvSeq *seq = CVSEQ(self);
222
+ if(!(seq->total > 0))
223
+ return Qnil;
224
+ return REFER_OBJECT(seqblock_class(seq), cvGetSeqElem(seq, 0), self);
225
+ }
226
+
227
+ /*
228
+ * call-seq:
229
+ * last -> obj or nil
230
+ *
231
+ * Return last sequence-block.
232
+ */
233
+ VALUE
234
+ rb_last(VALUE self)
235
+ {
236
+ CvSeq *seq = CVSEQ(self);
237
+ if(!(seq->total > 0))
238
+ return Qnil;
239
+ return REFER_OBJECT(seqblock_class(seq), cvGetSeqElem(seq, -1), self);
240
+ }
241
+
242
+ /*when storing it in CvSeq.
243
+ * call-seq:
244
+ * index(<i>obj</i>) -> int or nil
245
+ *
246
+ * Return the index of the first object in <i>self</i>. Return <tt>nil</tt> if no match is found.
247
+ */
248
+ VALUE
249
+ rb_index(VALUE self, VALUE object)
250
+ {
251
+ CvSeq *seq = CVSEQ(self);
252
+ int index;
253
+ if(CLASS_OF(object) == seqblock_class(seq)){
254
+ index = cvSeqElemIdx(seq, DATA_PTR(object));
255
+ if(!(index < 0))
256
+ return INT2FIX(index);
257
+ }else{
258
+ rb_warn("sequence-block class unmatch.");
259
+ }
260
+ return Qnil;
261
+ }
262
+
263
+ /*
264
+ * call-seq:
265
+ * h_prev -> seq or nil
266
+ *
267
+ * Return the sequence horizontally located in previous.
268
+ * Return <tt>nil</tt> if not existing.
269
+ */
270
+ VALUE
271
+ rb_h_prev(VALUE self)
272
+ {
273
+ CvSeq *seq = CVSEQ(self);
274
+ if (seq->h_prev) {
275
+ return new_sequence(CLASS_OF(self), seq->h_prev, seqblock_class(seq), lookup_root_object(seq));
276
+ //new_sequence(seq->h_prev, CLASS_OF(self), seqblock_class(seq), lookup_root_object(seq));
277
+ } else
278
+ return Qnil;
279
+ }
280
+
281
+ /*
282
+ * call-seq:
283
+ * h_next -> seq or nil
284
+ *
285
+ * Return the sequence horizontally located in next.
286
+ * Return <tt>nil</tt> if not existing.
287
+ */
288
+ VALUE
289
+ rb_h_next(VALUE self)
290
+ {
291
+ CvSeq *seq = CVSEQ(self);
292
+ if (seq->h_next) {
293
+ return new_sequence(CLASS_OF(self), seq->h_next, seqblock_class(seq), lookup_root_object(seq));
294
+ //new_sequence(seq->h_next, CLASS_OF(self), seqblock_class(seq), lookup_root_object(seq));
295
+ } else
296
+ return Qnil;
297
+ }
298
+
299
+ /*
300
+ * call-seq:
301
+ * v_prev -> seq or nil
302
+ *
303
+ * Return the sequence vertically located in previous.
304
+ * Return <tt>nil</tt> if not existing.
305
+ */
306
+ VALUE
307
+ rb_v_prev(VALUE self)
308
+ {
309
+ CvSeq *seq = CVSEQ(self);
310
+ if (seq->v_prev) {
311
+ return new_sequence(CLASS_OF(self), seq->v_prev, seqblock_class(seq), lookup_root_object(seq));
312
+ } else
313
+ return Qnil;
314
+ }
315
+
316
+ /*
317
+ * call-seq:
318
+ * v_prev -> seq or nil
319
+ *
320
+ * Return the sequence vertically located in next.
321
+ * Return <tt>nil</tt> if not existing.
322
+ */
323
+ VALUE
324
+ rb_v_next(VALUE self)
325
+ {
326
+ CvSeq *seq = CVSEQ(self);
327
+ if (seq->v_next) {
328
+ return new_sequence(CLASS_OF(self), seq->v_next, seqblock_class(seq), lookup_root_object(seq));
329
+ } else
330
+ return Qnil;
331
+ }
332
+
333
+ /*
334
+ * call-seq:
335
+ * push(<i>obj, ...</i>) -> self
336
+ *
337
+ * Append - Pushes the given object(s) on the end of this sequence. This expression return the sequence itself,
338
+ * so several append may be chainded together.
339
+ */
340
+ VALUE
341
+ rb_push(VALUE self, VALUE args)
342
+ {
343
+ CvSeq *seq = CVSEQ(self);
344
+ VALUE klass = seqblock_class(seq), object;
345
+ void *buffer = 0;
346
+ for(int i = 0; i < RARRAY(args)->len; i++){
347
+ object = RARRAY(args)->ptr[i];
348
+ if(CLASS_OF(object) == klass){
349
+ cvSeqPush(seq, DATA_PTR(object));
350
+ }else if(rb_obj_is_kind_of(object, rb_klass) && CLASS_OF(object) == klass){ // object is CvSeq
351
+ buffer = cvCvtSeqToArray(CVSEQ(object), cvAlloc(CVSEQ(object)->total * CVSEQ(object)->elem_size));
352
+ cvSeqPushMulti(seq, buffer, CVSEQ(object)->total);
353
+ cvFree((void**)&buffer);
354
+ }else{
355
+ object = CONVERT(object, klass);
356
+ cvSeqPush(seq, DATA_PTR(object));
357
+ }
358
+ }
359
+ return self;
360
+ }
361
+
362
+ /*
363
+ * call-seq:
364
+ * pop -> obj or nil
365
+ *
366
+ * Remove the last sequence-block from <i>self</i> and return it,
367
+ * or <tt>nil</tt> if the sequence is empty.
368
+ */
369
+ VALUE
370
+ rb_pop(VALUE self)
371
+ {
372
+ CvSeq *seq = CVSEQ(self);
373
+ if(!(seq->total > 0)){
374
+ return Qnil;
375
+ }
376
+ VALUE object = GENERIC_OBJECT(seqblock_class(seq), malloc(seq->elem_size));
377
+ cvSeqPop(seq, DATA_PTR(object));
378
+ return object;
379
+ }
380
+
381
+ /*
382
+ * call-seq:
383
+ * clear -> self
384
+ *
385
+ * Clears sequence. Removes all elements from the sequence.
386
+ */
387
+ VALUE
388
+ rb_clear(VALUE self)
389
+ {
390
+ cvClearSeq(CVSEQ(self));
391
+ return self;
392
+ }
393
+
394
+ /*
395
+ * call-seq:
396
+ * unshift -> self
397
+ *
398
+ * Prepends objects to the front of sequence. other elements up one.
399
+ */
400
+ VALUE
401
+ rb_unshift(VALUE self, VALUE args)
402
+ {
403
+ CvSeq *seq = CVSEQ(self);
404
+ VALUE klass = seqblock_class(seq), object;
405
+ void *buffer = 0;
406
+ for(int i = 0; i < RARRAY(args)->len; i++){
407
+ object = RARRAY(args)->ptr[i];
408
+ if(CLASS_OF(object) == klass){
409
+ cvSeqPushFront(seq, DATA_PTR(object));
410
+ }else if(rb_obj_is_kind_of(object, rb_klass) && CLASS_OF(object) == klass){
411
+ buffer = cvCvtSeqToArray(CVSEQ(object), cvAlloc(CVSEQ(object)->total * CVSEQ(object)->elem_size));
412
+ cvSeqPushMulti(seq, buffer, CVSEQ(object)->total, 1);
413
+ cvFree((void**)&buffer);
414
+ }else{
415
+ object = CONVERT(object, klass);
416
+ cvSeqPushFront(seq, DATA_PTR(object));
417
+ }
418
+ }
419
+ return self;
420
+ }
421
+
422
+ /*
423
+ * call-seq:
424
+ * shift -> obj or nil
425
+ *
426
+ * Returns the first element of <i>self</i> and removes it (shifting all other elements down by one). Returns <tt>nil</tt> if the array is empty.
427
+ */
428
+ VALUE
429
+ rb_shift(VALUE self)
430
+ {
431
+ CvSeq *seq = CVSEQ(self);
432
+ if(!(seq->total > 0)){
433
+ return Qnil;
434
+ }
435
+ VALUE object = GENERIC_OBJECT(seqblock_class(seq), malloc(seq->elem_size));
436
+ cvSeqPopFront(seq, DATA_PTR(object));
437
+ return object;
438
+ }
439
+
440
+ /*
441
+ * call-seq:
442
+ * each{|obj| ... } -> self
443
+ *
444
+ * Calls block once for each sequence-block in <i>self</i>,
445
+ * passing that sequence-block as a parameter.
446
+ * seq = CvSeq.new(CvIndex)
447
+ * seq.push(5, 6, 7)
448
+ * seq.each{|x| print x, " -- "
449
+ * produces:
450
+ * 5 -- 6 -- 7 --
451
+ */
452
+ VALUE
453
+ rb_each(VALUE self)
454
+ {
455
+ CvSeq *seq = CVSEQ(self);
456
+ if(seq->total > 0){
457
+ VALUE klass = seqblock_class(seq);
458
+ for(int i = 0; i < seq->total; i++){
459
+ rb_yield(REFER_OBJECT(klass, cvGetSeqElem(seq, i), self));
460
+ }
461
+ }
462
+ return self;
463
+ }
464
+
465
+ /*
466
+ * call-seq:
467
+ * each_index{|index| ... } -> self
468
+ *
469
+ * Same as CvSeq#each, but passes the index of the element instead of the element itself.
470
+ */
471
+ VALUE
472
+ rb_each_index(VALUE self)
473
+ {
474
+ CvSeq *seq = CVSEQ(self);
475
+ for(int i = 0; i < seq->total; i++)
476
+ rb_yield(INT2FIX(i));
477
+ return self;
478
+ }
479
+
480
+ /*
481
+ * call-seq:
482
+ * each_with_index{|obj, i| ... } -> self
483
+ *
484
+ * Calls block with two arguments, the sequence-block and its index, for each sequence-block in sequence.
485
+ */
486
+ VALUE
487
+ rb_each_with_index(VALUE self)
488
+ {
489
+ CvSeq *seq = CVSEQ(self);
490
+ VALUE klass = seqblock_class(seq);
491
+ for(int i = 0; i < seq->total; i++)
492
+ rb_yield_values(2, REFER_OBJECT(klass, cvGetSeqElem(seq, i), self), INT2FIX(i));
493
+ return self;
494
+ }
495
+
496
+ /*
497
+ * call-seq:
498
+ * insert(<i>index,obj</i>) -> self
499
+ *
500
+ * Inserts the given values before element with the given index (which may be negative).
501
+ */
502
+ VALUE
503
+ rb_insert(VALUE self, VALUE index, VALUE object)
504
+ {
505
+ Check_Type(index, T_FIXNUM);
506
+ CvSeq *seq = CVSEQ(self);
507
+ VALUE klass = seqblock_class(seq);
508
+ if(CLASS_OF(object) != klass)
509
+ object = CONVERT(object, klass);
510
+ cvSeqInsert(seq, FIX2INT(index), DATA_PTR(object));
511
+ return self;
512
+ }
513
+
514
+ /*
515
+ * call-seq:
516
+ * remove(<i>index</i>) -> obj or nil
517
+ *
518
+ * Deletes the elements at the specified index.
519
+ */
520
+ VALUE
521
+ rb_remove(VALUE self, VALUE index)
522
+ {
523
+ cvSeqRemove(CVSEQ(self), FIX2INT(index));
524
+ return self;
525
+ }
526
+
527
+ /*
528
+ * call-seq:
529
+ * clone
530
+ *
531
+ * Return copy of sequence.
532
+ */
533
+ VALUE
534
+ rb_clone(VALUE self)
535
+ {
536
+ CvSeq *seq = CVSEQ(self);
537
+ VALUE storage = cCvMemStorage::new_object();
538
+ return new_sequence(CLASS_OF(self), cvCloneSeq(seq), seqblock_class(seq), storage);
539
+ }
540
+
541
+ /*
542
+ VALUE
543
+ new_object(CvSeq *seq, VALUE klass)
544
+ {
545
+ VALUE storage = cCvMemStorage::new_object();
546
+ VALUE object = REFER_OBJECT(rb_klass, seq, storage);
547
+ st_insert(seqblock_klass, (st_data_t)seq, (st_data_t)klass);
548
+ return object;
549
+ }
550
+
551
+ VALUE
552
+ new_object(CvSeq *seq, VALUE klass, VALUE storage)
553
+ {
554
+ VALUE object = REFER_OBJECT(rb_klass, seq, storage);
555
+ st_insert(seqblock_klass, (st_data_t)seq, (st_data_t)klass);
556
+ return object;
557
+ }
558
+ */
559
+
560
+
561
+ VALUE
562
+ new_sequence(VALUE klass, CvSeq *seq, VALUE element_klass, VALUE storage)
563
+ {
564
+ resist_root_object(seq, storage);
565
+ if (!NIL_P(element_klass))
566
+ st_insert(seqblock_klass, (st_data_t)seq, (st_data_t)element_klass);
567
+ VALUE object = Data_Wrap_Struct(klass, mark_root_object, free, seq);
568
+ auto_extend(object);
569
+ return object;
570
+ }
571
+
572
+ VALUE
573
+ auto_extend(VALUE object)
574
+ {
575
+ CvSeq *seq = CVSEQ(object);
576
+ if(CV_IS_SEQ_POINT_SET(seq)){
577
+ rb_extend_object(object, mPointSet::rb_module());
578
+ }
579
+ return object;
580
+ }
581
+
582
+ __NAMESPACE_END_CVSEQ
583
+ __NAMESPACE_END_OPENCV