cv.cpp

Matthew Baker, 2011-04-11 01:49 am

Download (112.2 kB)

 
1
#include <Python.h>
2
3
#include <assert.h>
4
5
#include <opencv/cxcore.h>
6
#include <opencv/cv.h>
7
#include <opencv/cvaux.h>
8
#include <opencv/cvwimage.h>
9
#include <opencv/highgui.h>
10
11
#define MODULESTR "cv"
12
13
static PyObject *opencv_error;
14
15
struct memtrack_t {
16
  PyObject_HEAD
17
  void *ptr;
18
  Py_ssize_t size;
19
};
20
21
struct iplimage_t {
22
  PyObject_HEAD
23
  IplImage *a;
24
  PyObject *data;
25
  size_t offset;
26
};
27
28
struct cvmat_t {
29
  PyObject_HEAD
30
  CvMat *a;
31
  PyObject *data;
32
  size_t offset;
33
};
34
35
struct cvmatnd_t {
36
  PyObject_HEAD
37
  CvMatND *a;
38
  PyObject *data;
39
  size_t offset;
40
};
41
42
struct cvhistogram_t {
43
  PyObject_HEAD
44
  CvHistogram h;
45
  PyObject *bins;
46
};
47
48
struct cvmemstorage_t {
49
  PyObject_HEAD
50
  CvMemStorage *a;
51
};
52
53
struct cvseq_t {
54
  PyObject_HEAD
55
  CvSeq *a;
56
  PyObject *container;  // Containing cvmemstorage_t
57
};
58
59
struct cvset_t {
60
  PyObject_HEAD
61
  CvSet *a;
62
  PyObject *container;  // Containing cvmemstorage_t
63
  int i;
64
};
65
66
struct cvsubdiv2d_t {
67
  PyObject_HEAD
68
  CvSubdiv2D *a;
69
  PyObject *container;  // Containing cvmemstorage_t
70
};
71
72
struct cvsubdiv2dpoint_t {
73
  PyObject_HEAD
74
  CvSubdiv2DPoint *a;
75
  PyObject *container;  // Containing cvmemstorage_t
76
};
77
78
struct cvsubdiv2dedge_t {
79
  PyObject_HEAD
80
  CvSubdiv2DEdge a;
81
  PyObject *container;  // Containing cvmemstorage_t
82
};
83
84
struct cvlineiterator_t {
85
  PyObject_HEAD
86
  CvLineIterator iter;
87
  int count;
88
  int type;
89
};
90
91
typedef IplImage ROIplImage;
92
typedef const CvMat ROCvMat;
93
typedef PyObject PyCallableObject;
94
95
struct cvfont_t {
96
  PyObject_HEAD
97
  CvFont a;
98
};
99
100
struct cvcontourtree_t {
101
  PyObject_HEAD
102
  CvContourTree *a;
103
};
104
105
struct cvrng_t {
106
  PyObject_HEAD
107
  CvRNG a;
108
};
109
110
static int is_iplimage(PyObject *o);
111
static int is_cvmat(PyObject *o);
112
static int is_cvmatnd(PyObject *o);
113
static int convert_to_CvArr(PyObject *o, CvArr **dst, const char *name = "no_name");
114
static int convert_to_IplImage(PyObject *o, IplImage **dst, const char *name = "no_name");
115
static int convert_to_CvMat(PyObject *o, CvMat **dst, const char *name = "no_name");
116
static int convert_to_CvMatND(PyObject *o, CvMatND **dst, const char *name = "no_name");
117
static PyObject *what_data(PyObject *o);
118
static PyObject *FROM_CvMat(CvMat *r);
119
static PyObject *FROM_ROCvMatPTR(ROCvMat *r);
120
static PyObject *shareDataND(PyObject *donor, CvMatND *pdonor, CvMatND *precipient);
121
122
#define FROM_double(r)  PyFloat_FromDouble(r)
123
#define FROM_float(r)  PyFloat_FromDouble(r)
124
#define FROM_int(r)  PyInt_FromLong(r)
125
#define FROM_int64(r)  PyLong_FromLongLong(r)
126
#define FROM_unsigned(r)  PyLong_FromUnsignedLong(r)
127
#define FROM_CvBox2D(r) Py_BuildValue("(ff)(ff)f", r.center.x, r.center.y, r.size.width, r.size.height, r.angle)
128
#define FROM_CvScalar(r)  Py_BuildValue("(ffff)", r.val[0], r.val[1], r.val[2], r.val[3])
129
#define FROM_CvPoint(r)  Py_BuildValue("(ii)", r.x, r.y)
130
#define FROM_CvPoint2D32f(r) Py_BuildValue("(ff)", r.x, r.y)
131
#define FROM_CvPoint3D64f(r) Py_BuildValue("(fff)", r.x, r.y, r.z)
132
#define FROM_CvSize(r) Py_BuildValue("(ii)", r.width, r.height)
133
#define FROM_CvRect(r) Py_BuildValue("(iiii)", r.x, r.y, r.width, r.height)
134
#define FROM_CvSeqPTR(r) _FROM_CvSeqPTR(r, pyobj_storage)
135
#define FROM_CvSubdiv2DPTR(r) _FROM_CvSubdiv2DPTR(r, pyobj_storage)
136
#define FROM_CvPoint2D64f(r) Py_BuildValue("(ff)", r.x, r.y)
137
#define FROM_CvConnectedComp(r) Py_BuildValue("(fNN)", (r).area, FROM_CvScalar((r).value), FROM_CvRect((r).rect))
138
139
#if PYTHON_USE_NUMPY
140
static PyObject *fromarray(PyObject *o, int allowND);
141
#endif
142
143
static void translate_error_to_exception(void)
144
{
145
  PyErr_SetString(opencv_error, cvErrorStr(cvGetErrStatus()));
146
  cvSetErrStatus(0);
147
}
148
149
#define ERRCHK do { if (cvGetErrStatus() != 0) { translate_error_to_exception(); return NULL; } } while (0)
150
#define ERRWRAP(F) \
151
    do { \
152
        try \
153
        { \
154
            F; \
155
        } \
156
        catch (const cv::Exception &e) \
157
        { \
158
           PyErr_SetString(opencv_error, e.err.c_str()); \
159
           return NULL; \
160
        } \
161
        ERRCHK; \
162
    } while(0)
163
164
/************************************************************************/
165
166
static int failmsg(const char *fmt, ...)
167
{
168
  char str[1000];
169
170
  va_list ap;
171
  va_start(ap, fmt);
172
  vsnprintf(str, sizeof(str), fmt, ap);
173
  va_end(ap);
174
175
  PyErr_SetString(PyExc_TypeError, str);
176
  return 0;
177
}
178
179
/************************************************************************/
180
181
/* These get/setters are polymorphic, used in both iplimage and cvmat */
182
183
static PyObject *PyObject_FromCvScalar(CvScalar s, int type)
184
{
185
  int i, spe = CV_MAT_CN(type);
186
  PyObject *r;
187
  if (spe > 1) {
188
    r = PyTuple_New(spe);
189
    for (i = 0; i < spe; i++)
190
      PyTuple_SET_ITEM(r, i, PyFloat_FromDouble(s.val[i]));
191
  } else {
192
    r = PyFloat_FromDouble(s.val[0]);
193
  }
194
  return r;
195
}
196
197
static PyObject *cvarr_GetItem(PyObject *o, PyObject *key);
198
static int cvarr_SetItem(PyObject *o, PyObject *key, PyObject *v);
199
200
// o is a Python string or buffer object.  Return its size.
201
202
static Py_ssize_t what_size(PyObject *o)
203
{
204
  void *buffer;
205
  Py_ssize_t buffer_len;
206
207
  if (PyString_Check(o)) {
208
    return PyString_Size(o);
209
  } else if (PyObject_AsWriteBuffer(o, &buffer, &buffer_len) == 0) {
210
    return buffer_len;
211
  } else {
212
    assert(0);  // argument must be string or buffer.
213
    return 0;
214
  }
215
}
216
217
218
/************************************************************************/
219
220
CvMat *PyCvMat_AsCvMat(PyObject *o)
221
{
222
  assert(0); // not yet implemented: reference counting for CvMat in Kalman is unclear...
223
  return NULL;
224
}
225
226
#define cvReleaseIplConvKernel(x) cvReleaseStructuringElement(x)
227
#include "generated3.i"
228
229
/* iplimage */
230
231
static void iplimage_dealloc(PyObject *self)
232
{
233
  iplimage_t *pc = (iplimage_t*)self;
234
  cvReleaseImageHeader((IplImage**)&pc->a);
235
  Py_DECREF(pc->data);
236
  PyObject_Del(self);
237
}
238
239
static PyObject *iplimage_repr(PyObject *self)
240
{
241
  iplimage_t *cva = (iplimage_t*)self;
242
  IplImage* ipl = (IplImage*)(cva->a);
243
  char str[1000];
244
  sprintf(str, "<iplimage(");
245
  char *d = str + strlen(str);
246
  sprintf(d, "nChannels=%d ", ipl->nChannels);
247
  d += strlen(d);
248
  sprintf(d, "width=%d ", ipl->width);
249
  d += strlen(d);
250
  sprintf(d, "height=%d ", ipl->height);
251
  d += strlen(d);
252
  sprintf(d, "widthStep=%d ", ipl->widthStep);
253
  d += strlen(d);
254
  sprintf(d, ")>");
255
  return PyString_FromString(str);
256
}
257
258
static PyObject *iplimage_tostring(PyObject *self, PyObject *args)
259
{
260
  iplimage_t *pc = (iplimage_t*)self;
261
  IplImage *i;
262
  if (!convert_to_IplImage(self, &i, "self"))
263
    return NULL;
264
  if (i == NULL)
265
    return NULL;
266
  int bps;
267
  switch (i->depth) {
268
  case IPL_DEPTH_8U:
269
  case IPL_DEPTH_8S:
270
    bps = 1;
271
    break;
272
  case IPL_DEPTH_16U:
273
  case IPL_DEPTH_16S:
274
    bps = 2;
275
    break;
276
  case IPL_DEPTH_32S:
277
  case IPL_DEPTH_32F:
278
    bps = 4;
279
    break;
280
  case IPL_DEPTH_64F:
281
    bps = 8;
282
    break;
283
  default:
284
    return (PyObject*)failmsg("Unrecognised depth %d", i->depth);
285
  }
286
  int bpl = i->width * i->nChannels * bps;
287
  if (PyString_Check(pc->data) && bpl == i->widthStep && pc->offset == 0 && ((bpl * i->height) == what_size(pc->data))) {
288
    Py_INCREF(pc->data);
289
    return pc->data;
290
  } else {
291
    int l = bpl * i->height;
292
    char *s = new char[l];
293
    int y;
294
    for (y = 0; y < i->height; y++) {
295
      memcpy(s + y * bpl, i->imageData + y * i->widthStep, bpl);
296
    }
297
    PyObject *r = PyString_FromStringAndSize(s, l);
298
    delete s;
299
    return r;
300
  }
301
}
302
303
static struct PyMethodDef iplimage_methods[] =
304
{
305
  {"tostring", iplimage_tostring, METH_VARARGS},
306
  {NULL,          NULL}
307
};
308
309
static PyObject *iplimage_getnChannels(iplimage_t *cva)
310
{
311
  return PyInt_FromLong(((IplImage*)(cva->a))->nChannels);
312
}
313
static PyObject *iplimage_getwidth(iplimage_t *cva)
314
{
315
  return PyInt_FromLong(((IplImage*)(cva->a))->width);
316
}
317
static PyObject *iplimage_getheight(iplimage_t *cva)
318
{
319
  return PyInt_FromLong(((IplImage*)(cva->a))->height);
320
}
321
static PyObject *iplimage_getdepth(iplimage_t *cva)
322
{
323
  return PyLong_FromUnsignedLong((unsigned)((IplImage*)(cva->a))->depth);
324
}
325
static PyObject *iplimage_getorigin(iplimage_t *cva)
326
{
327
  return PyInt_FromLong(((IplImage*)(cva->a))->origin);
328
}
329
static void iplimage_setorigin(iplimage_t *cva, PyObject *v)
330
{
331
  ((IplImage*)(cva->a))->origin = PyInt_AsLong(v);
332
}
333
334
static PyGetSetDef iplimage_getseters[] = {
335
  {(char*)"nChannels", (getter)iplimage_getnChannels, (setter)NULL, (char*)"nChannels", NULL},
336
  {(char*)"channels", (getter)iplimage_getnChannels, (setter)NULL, (char*)"nChannels", NULL},
337
  {(char*)"width", (getter)iplimage_getwidth, (setter)NULL, (char*)"width", NULL},
338
  {(char*)"height", (getter)iplimage_getheight, (setter)NULL, (char*)"height", NULL},
339
  {(char*)"depth", (getter)iplimage_getdepth, (setter)NULL, (char*)"depth", NULL},
340
  {(char*)"origin", (getter)iplimage_getorigin, (setter)iplimage_setorigin, (char*)"origin", NULL},
341
  {NULL}  /* Sentinel */
342
};
343
344
static PyMappingMethods iplimage_as_map = {
345
  NULL,
346
  &cvarr_GetItem,
347
  &cvarr_SetItem,
348
};
349
350
static PyTypeObject iplimage_Type = {
351
  PyObject_HEAD_INIT(&PyType_Type)
352
  0,                                      /*size*/
353
  MODULESTR".iplimage",                          /*name*/
354
  sizeof(iplimage_t),                        /*basicsize*/
355
};
356
357
static void iplimage_specials(void)
358
{
359
  iplimage_Type.tp_dealloc = iplimage_dealloc;
360
  iplimage_Type.tp_as_mapping = &iplimage_as_map;
361
  iplimage_Type.tp_repr = iplimage_repr;
362
  iplimage_Type.tp_methods = iplimage_methods;
363
  iplimage_Type.tp_getset = iplimage_getseters;
364
}
365
366
static int is_iplimage(PyObject *o)
367
{
368
  return PyType_IsSubtype(o->ob_type, &iplimage_Type);
369
}
370
371
/************************************************************************/
372
373
/* cvmat */
374
375
static void cvmat_dealloc(PyObject *self)
376
{
377
  cvmat_t *pc = (cvmat_t*)self;
378
  if (pc->data) {
379
    Py_DECREF(pc->data);
380
  }
381
  cvFree(&pc->a);
382
  PyObject_Del(self);
383
}
384
385
static PyObject *cvmat_repr(PyObject *self)
386
{
387
  CvMat *m = ((cvmat_t*)self)->a;
388
  char str[1000];
389
  sprintf(str, "<cvmat(");
390
  char *d = str + strlen(str);
391
  sprintf(d, "type=%08x ", m->type);
392
  d += strlen(d);
393
  switch (CV_MAT_DEPTH(m->type)) {
394
  case CV_8U: strcpy(d, "8U"); break;
395
  case CV_8S: strcpy(d, "8S"); break;
396
  case CV_16U: strcpy(d, "16U"); break;
397
  case CV_16S: strcpy(d, "16S"); break;
398
  case CV_32S: strcpy(d, "32S"); break;
399
  case CV_32F: strcpy(d, "32F"); break;
400
  case CV_64F: strcpy(d, "64F"); break;
401
  }
402
  d += strlen(d);
403
  sprintf(d, "C%d ", CV_MAT_CN(m->type));
404
  d += strlen(d);
405
  sprintf(d, "rows=%d ", m->rows);
406
  d += strlen(d);
407
  sprintf(d, "cols=%d ", m->cols);
408
  d += strlen(d);
409
  sprintf(d, "step=%d ", m->step);
410
  d += strlen(d);
411
  sprintf(d, ")>");
412
  return PyString_FromString(str);
413
}
414
415
static PyObject *cvmat_tostring(PyObject *self, PyObject *args)
416
{
417
  CvMat *m;
418
  if (!convert_to_CvMat(self, &m, "self"))
419
    return NULL;
420
421
  int bps;                     // bytes per sample
422
423
  switch (CV_MAT_DEPTH(m->type)) {
424
  case CV_8U:
425
  case CV_8S:
426
    bps = CV_MAT_CN(m->type) * 1;
427
    break;
428
  case CV_16U:
429
  case CV_16S:
430
    bps = CV_MAT_CN(m->type) * 2;
431
    break;
432
  case CV_32S:
433
  case CV_32F:
434
    bps = CV_MAT_CN(m->type) * 4;
435
    break;
436
  case CV_64F:
437
    bps = CV_MAT_CN(m->type) * 8;
438
    break;
439
  default:
440
    return (PyObject*)failmsg("Unrecognised depth %d", CV_MAT_DEPTH(m->type));
441
  }
442
443
  int bpl = m->cols * bps; // bytes per line
444
  cvmat_t *pc = (cvmat_t*)self;
445
  if (PyString_Check(pc->data) && bpl == m->step && pc->offset == 0 && ((bpl * m->rows) == what_size(pc->data))) {
446
    Py_INCREF(pc->data);
447
    return pc->data;
448
  } else {
449
    int l = bpl * m->rows;
450
    char *s = new char[l];
451
    int y;
452
    for (y = 0; y < m->rows; y++) {
453
      memcpy(s + y * bpl, m->data.ptr + y * m->step, bpl);
454
    }
455
    PyObject *r = PyString_FromStringAndSize(s, l);
456
    delete s;
457
    return r;
458
  }
459
}
460
461
static struct PyMethodDef cvmat_methods[] =
462
{
463
  {"tostring", cvmat_tostring, METH_VARARGS},
464
  {NULL,          NULL}
465
};
466
467
static PyObject *cvmat_gettype(cvmat_t *cva)
468
{
469
  return PyInt_FromLong(cvGetElemType(cva->a));
470
}
471
472
static PyObject *cvmat_getstep(cvmat_t *cva)
473
{
474
  return PyInt_FromLong(cva->a->step);
475
}
476
477
static PyObject *cvmat_getrows(cvmat_t *cva)
478
{
479
  return PyInt_FromLong(cva->a->rows);
480
}
481
482
static PyObject *cvmat_getcols(cvmat_t *cva)
483
{
484
  return PyInt_FromLong(cva->a->cols);
485
}
486
487
static PyObject *cvmat_getchannels(cvmat_t *cva)
488
{
489
  return PyInt_FromLong(CV_MAT_CN(cva->a->type));
490
}
491
492
#if PYTHON_USE_NUMPY
493
#include "numpy/ndarrayobject.h"
494
495
// A PyArrayInterface, with an associated python object that should be DECREF'ed on release
496
struct arrayTrack {
497
  PyArrayInterface s;
498
  PyObject *o;
499
};
500
501
static void arrayTrackDtor(void *p)
502
{
503
  struct arrayTrack *at = (struct arrayTrack *)p;
504
  delete at->s.shape;
505
  delete at->s.strides;
506
  if (at->s.descr)
507
    Py_DECREF(at->s.descr);
508
  Py_DECREF(at->o);
509
}
510
511
// Fill in fields of PyArrayInterface s using mtype.  This code is common
512
// to cvmat and cvmatnd
513
514
static void arrayinterface_common(PyArrayInterface *s, int mtype)
515
{
516
  s->two = 2;
517
518
  switch (CV_MAT_DEPTH(mtype)) {
519
  case CV_8U:
520
    s->typekind = 'u';
521
    s->itemsize = 1;
522
    break;
523
  case CV_8S:
524
    s->typekind = 'i';
525
    s->itemsize = 1;
526
    break;
527
  case CV_16U:
528
    s->typekind = 'u';
529
    s->itemsize = 2;
530
    break;
531
  case CV_16S:
532
    s->typekind = 'i';
533
    s->itemsize = 2;
534
    break;
535
  case CV_32S:
536
    s->typekind = 'i';
537
    s->itemsize = 4;
538
    break;
539
  case CV_32F:
540
    s->typekind = 'f';
541
    s->itemsize = 4;
542
    break;
543
  case CV_64F:
544
    s->typekind = 'f';
545
    s->itemsize = 8;
546
    break;
547
  default:
548
    assert(0);
549
  }
550
551
  s->flags = NPY_WRITEABLE | NPY_NOTSWAPPED;
552
}
553
554
static PyObject *cvmat_array_struct(cvmat_t *cva)
555
{
556
  CvMat *m;
557
  convert_to_CvMat((PyObject *)cva, &m, "");
558
559
  arrayTrack *at = new arrayTrack;
560
  PyArrayInterface *s = &at->s;
561
562
  at->o = cva->data;
563
  Py_INCREF(at->o);
564
565
  arrayinterface_common(s, m->type);
566
567
  if (CV_MAT_CN(m->type) == 1) {
568
    s->nd = 2;
569
    s->shape = new npy_intp[2];
570
    s->shape[0] = m->rows;
571
    s->shape[1] = m->cols;
572
    s->strides = new npy_intp[2];
573
    s->strides[0] = m->step;
574
    s->strides[1] = s->itemsize;
575
  } else {
576
    s->nd = 3;
577
    s->shape = new npy_intp[3];
578
    s->shape[0] = m->rows;
579
    s->shape[1] = m->cols;
580
    s->shape[2] = CV_MAT_CN(m->type);
581
    s->strides = new npy_intp[3];
582
    s->strides[0] = m->step;
583
    s->strides[1] = s->itemsize * CV_MAT_CN(m->type);
584
    s->strides[2] = s->itemsize;
585
  }
586
  s->data = (void*)(m->data.ptr);
587
  s->descr = PyList_New(1);
588
  char typestr[10];
589
  sprintf(typestr, "<%c%d", s->typekind, s->itemsize);
590
  PyList_SetItem(s->descr, 0, Py_BuildValue("(ss)", "x", typestr));
591
592
  return PyCObject_FromVoidPtr(s, arrayTrackDtor);
593
}
594
595
static PyObject *cvmatnd_array_struct(cvmatnd_t *cva)
596
{
597
  CvMatND *m;
598
  convert_to_CvMatND((PyObject *)cva, &m, "");
599
600
  arrayTrack *at = new arrayTrack;
601
  PyArrayInterface *s = &at->s;
602
603
  at->o = cva->data;
604
  Py_INCREF(at->o);
605
606
  arrayinterface_common(s, m->type);
607
608
  int i;
609
  if (CV_MAT_CN(m->type) == 1) {
610
    s->nd = m->dims;
611
    s->shape = new npy_intp[s->nd];
612
    for (i = 0; i < s->nd; i++)
613
      s->shape[i] = m->dim[i].size;
614
    s->strides = new npy_intp[s->nd];
615
    for (i = 0; i < (s->nd - 1); i++)
616
      s->strides[i] = m->dim[i].step;
617
    s->strides[s->nd - 1] = s->itemsize;
618
  } else {
619
    s->nd = m->dims + 1;
620
    s->shape = new npy_intp[s->nd];
621
    for (i = 0; i < (s->nd - 1); i++)
622
      s->shape[i] = m->dim[i].size;
623
    s->shape[s->nd - 1] = CV_MAT_CN(m->type);
624
625
    s->strides = new npy_intp[s->nd];
626
    for (i = 0; i < (s->nd - 2); i++)
627
      s->strides[i] = m->dim[i].step;
628
    s->strides[s->nd - 2] = s->itemsize * CV_MAT_CN(m->type);
629
    s->strides[s->nd - 1] = s->itemsize;
630
  }
631
  s->data = (void*)(m->data.ptr);
632
  s->descr = PyList_New(1);
633
  char typestr[10];
634
  sprintf(typestr, "<%c%d", s->typekind, s->itemsize);
635
  PyList_SetItem(s->descr, 0, Py_BuildValue("(ss)", "x", typestr));
636
637
  return PyCObject_FromVoidPtr(s, arrayTrackDtor);
638
}
639
#endif
640
641
static PyGetSetDef cvmat_getseters[] = {
642
  {(char*)"type",   (getter)cvmat_gettype, (setter)NULL, (char*)"type",   NULL},
643
  {(char*)"step",   (getter)cvmat_getstep, (setter)NULL, (char*)"step",   NULL},
644
  {(char*)"rows",   (getter)cvmat_getrows, (setter)NULL, (char*)"rows",   NULL},
645
  {(char*)"cols",   (getter)cvmat_getcols, (setter)NULL, (char*)"cols",   NULL},
646
  {(char*)"channels",(getter)cvmat_getchannels, (setter)NULL, (char*)"channels",   NULL},
647
  {(char*)"width",  (getter)cvmat_getcols, (setter)NULL, (char*)"width",  NULL},
648
  {(char*)"height", (getter)cvmat_getrows, (setter)NULL, (char*)"height", NULL},
649
#if PYTHON_USE_NUMPY
650
  {(char*)"__array_struct__", (getter)cvmat_array_struct, (setter)NULL, (char*)"__array_struct__", NULL},
651
#endif
652
  {NULL}  /* Sentinel */
653
};
654
655
static PyMappingMethods cvmat_as_map = {
656
  NULL,
657
  &cvarr_GetItem,
658
  &cvarr_SetItem,
659
};
660
661
static PyTypeObject cvmat_Type = {
662
  PyObject_HEAD_INIT(&PyType_Type)
663
  0,                                      /*size*/
664
  MODULESTR".cvmat",                      /*name*/
665
  sizeof(cvmat_t),                        /*basicsize*/
666
};
667
668
static int illegal_init(PyObject *self, PyObject *args, PyObject *kwds)
669
{
670
  PyErr_SetString(opencv_error, "Cannot create cvmat directly; use CreateMat() instead");
671
  return -1;
672
}
673
674
static void cvmat_specials(void)
675
{
676
  cvmat_Type.tp_dealloc = cvmat_dealloc;
677
  cvmat_Type.tp_as_mapping = &cvmat_as_map;
678
  cvmat_Type.tp_repr = cvmat_repr;
679
  cvmat_Type.tp_methods = cvmat_methods;
680
  cvmat_Type.tp_getset = cvmat_getseters;
681
  cvmat_Type.tp_init = illegal_init;
682
}
683
684
static int is_cvmat(PyObject *o)
685
{
686
  return PyType_IsSubtype(o->ob_type, &cvmat_Type);
687
}
688
689
/************************************************************************/
690
691
/* cvmatnd */
692
693
static void cvmatnd_dealloc(PyObject *self)
694
{
695
  cvmatnd_t *pc = (cvmatnd_t*)self;
696
  Py_DECREF(pc->data);
697
  cvDecRefData(pc->a);
698
  cvFree(&pc->a);
699
  PyObject_Del(self);
700
}
701
702
static PyObject *cvmatnd_repr(PyObject *self)
703
{
704
  CvMatND *m = ((cvmatnd_t*)self)->a;
705
  char str[1000];
706
  sprintf(str, "<cvmatnd(");
707
  char *d = str + strlen(str);
708
  sprintf(d, "type=%08x ", m->type);
709
  d += strlen(d);
710
  sprintf(d, ")>");
711
  return PyString_FromString(str);
712
}
713
714
static size_t cvmatnd_size(CvMatND *m)
715
{
716
  int bps = 1;
717
  switch (CV_MAT_DEPTH(m->type)) {
718
  case CV_8U:
719
  case CV_8S:
720
    bps = CV_MAT_CN(m->type) * 1;
721
    break;
722
  case CV_16U:
723
  case CV_16S:
724
    bps = CV_MAT_CN(m->type) * 2;
725
    break;
726
  case CV_32S:
727
  case CV_32F:
728
    bps = CV_MAT_CN(m->type) * 4;
729
    break;
730
  case CV_64F:
731
    bps = CV_MAT_CN(m->type) * 8;
732
    break;
733
  default:
734
    assert(0);
735
  }
736
  size_t l = bps;
737
  for (int d = 0; d < m->dims; d++) {
738
    l *= m->dim[d].size;
739
  }
740
  return l;
741
}
742
743
static PyObject *cvmatnd_tostring(PyObject *self, PyObject *args)
744
{
745
  CvMatND *m;
746
  if (!convert_to_CvMatND(self, &m, "self"))
747
    return NULL;
748
749
  int bps;
750
  switch (CV_MAT_DEPTH(m->type)) {
751
  case CV_8U:
752
  case CV_8S:
753
    bps = CV_MAT_CN(m->type) * 1;
754
    break;
755
  case CV_16U:
756
  case CV_16S:
757
    bps = CV_MAT_CN(m->type) * 2;
758
    break;
759
  case CV_32S:
760
  case CV_32F:
761
    bps = CV_MAT_CN(m->type) * 4;
762
    break;
763
  case CV_64F:
764
    bps = CV_MAT_CN(m->type) * 8;
765
    break;
766
  default:
767
    return (PyObject*)failmsg("Unrecognised depth %d", CV_MAT_DEPTH(m->type));
768
  }
769
770
  int l = bps;
771
  for (int d = 0; d < m->dims; d++) {
772
    l *= m->dim[d].size;
773
  }
774
  int i[CV_MAX_DIM];
775
  int d;
776
  for (d = 0; d < m->dims; d++) {
777
    i[d] = 0;
778
  }
779
  int rowsize = m->dim[m->dims-1].size * bps;
780
  char *s = new char[l];
781
  char *ps = s;
782
783
  int finished = 0;
784
  while (!finished) {
785
    memcpy(ps, cvPtrND(m, i), rowsize);
786
    ps += rowsize;
787
    for (d = m->dims - 2; 0 <= d; d--) {
788
      if (++i[d] < cvGetDimSize(m, d)) {
789
        break;
790
      } else {
791
        i[d] = 0;
792
      }
793
    }
794
    if (d < 0)
795
      finished = 1;
796
  }
797
798
  return PyString_FromStringAndSize(s, ps - s);
799
}
800
801
static struct PyMethodDef cvmatnd_methods[] =
802
{
803
  {"tostring", cvmatnd_tostring, METH_VARARGS},
804
  {NULL,          NULL}
805
};
806
807
static PyObject *cvmatnd_getchannels(cvmatnd_t *cva)
808
{
809
  return PyInt_FromLong(CV_MAT_CN(cva->a->type));
810
}
811
812
static PyGetSetDef cvmatnd_getseters[] = {
813
#if PYTHON_USE_NUMPY
814
  {(char*)"__array_struct__", (getter)cvmatnd_array_struct, (setter)NULL, (char*)"__array_struct__", NULL},
815
#endif
816
  {(char*)"channels",(getter)cvmatnd_getchannels, (setter)NULL, (char*)"channels",   NULL},
817
  {NULL}  /* Sentinel */
818
};
819
820
static PyMappingMethods cvmatnd_as_map = {
821
  NULL,
822
  &cvarr_GetItem,
823
  &cvarr_SetItem,
824
};
825
826
static PyTypeObject cvmatnd_Type = {
827
  PyObject_HEAD_INIT(&PyType_Type)
828
  0,                                      /*size*/
829
  MODULESTR".cvmatnd",                          /*name*/
830
  sizeof(cvmatnd_t),                        /*basicsize*/
831
};
832
833
static void cvmatnd_specials(void)
834
{
835
  cvmatnd_Type.tp_dealloc = cvmatnd_dealloc;
836
  cvmatnd_Type.tp_as_mapping = &cvmatnd_as_map;
837
  cvmatnd_Type.tp_repr = cvmatnd_repr;
838
  cvmatnd_Type.tp_methods = cvmatnd_methods;
839
  cvmatnd_Type.tp_getset = cvmatnd_getseters;
840
}
841
842
static int is_cvmatnd(PyObject *o)
843
{
844
  return PyType_IsSubtype(o->ob_type, &cvmatnd_Type);
845
}
846
847
/************************************************************************/
848
849
/* cvhistogram */
850
851
static void cvhistogram_dealloc(PyObject *self)
852
{
853
  cvhistogram_t *cvh = (cvhistogram_t*)self;
854
  Py_DECREF(cvh->bins);
855
  PyObject_Del(self);
856
}
857
858
static PyTypeObject cvhistogram_Type = {
859
  PyObject_HEAD_INIT(&PyType_Type)
860
  0,                                      /*size*/
861
  MODULESTR".cvhistogram",                /*name*/
862
  sizeof(cvhistogram_t),                  /*basicsize*/
863
};
864
865
static PyObject *cvhistogram_getbins(cvhistogram_t *cvh)
866
{
867
  Py_INCREF(cvh->bins);
868
  return cvh->bins;
869
}
870
871
static PyGetSetDef cvhistogram_getseters[] = {
872
  {(char*)"bins", (getter)cvhistogram_getbins, (setter)NULL, (char*)"bins", NULL},
873
  {NULL}  /* Sentinel */
874
};
875
876
static void cvhistogram_specials(void)
877
{
878
  cvhistogram_Type.tp_dealloc = cvhistogram_dealloc;
879
  cvhistogram_Type.tp_getset = cvhistogram_getseters;
880
}
881
882
/************************************************************************/
883
884
/* cvlineiterator */
885
886
static PyObject *cvlineiterator_iter(PyObject *o)
887
{
888
  Py_INCREF(o);
889
  return o;
890
}
891
892
static PyObject *cvlineiterator_next(PyObject *o)
893
{
894
  cvlineiterator_t *pi = (cvlineiterator_t*)o;
895
896
  if (pi->count) {
897
      pi->count--;
898
899
      CvScalar r;
900
      cvRawDataToScalar( (void*)(pi->iter.ptr), pi->type, &r);
901
      PyObject *pr = PyObject_FromCvScalar(r, pi->type);
902
903
      CV_NEXT_LINE_POINT(pi->iter);
904
905
      return pr;
906
  } else {
907
    return NULL;
908
  }
909
}
910
911
static PyTypeObject cvlineiterator_Type = {
912
  PyObject_HEAD_INIT(&PyType_Type)
913
  0,                                      /*size*/
914
  MODULESTR".cvlineiterator",             /*name*/
915
  sizeof(cvlineiterator_t),               /*basicsize*/
916
};
917
918
static void cvlineiterator_specials(void)
919
{
920
  cvlineiterator_Type.tp_iter = cvlineiterator_iter;
921
  cvlineiterator_Type.tp_iternext = cvlineiterator_next;
922
}
923
924
/************************************************************************/
925
926
/* memtrack */
927
928
static void memtrack_dealloc(PyObject *self)
929
{
930
  memtrack_t *pi = (memtrack_t*)self;
931
  // printf("===> memtrack_dealloc %p!\n", pi->ptr);
932
  cvFree(&pi->ptr);
933
  PyObject_Del(self);
934
}
935
936
static PyTypeObject memtrack_Type = {
937
  PyObject_HEAD_INIT(&PyType_Type)
938
  0,                                      /*size*/
939
  MODULESTR".memtrack",                          /*name*/
940
  sizeof(memtrack_t),                        /*basicsize*/
941
};
942
943
Py_ssize_t memtrack_getreadbuffer(PyObject *self, Py_ssize_t segment, void **ptrptr)
944
{
945
  *ptrptr = &((memtrack_t*)self)->ptr;
946
  return ((memtrack_t*)self)->size;
947
}
948
949
Py_ssize_t memtrack_getwritebuffer(PyObject *self, Py_ssize_t segment, void **ptrptr)
950
{
951
  *ptrptr = ((memtrack_t*)self)->ptr;
952
  return ((memtrack_t*)self)->size;
953
}
954
955
Py_ssize_t memtrack_getsegcount(PyObject *self, Py_ssize_t *lenp)
956
{
957
  return (Py_ssize_t)1;
958
}
959
960
PyBufferProcs memtrack_as_buffer = {
961
  memtrack_getreadbuffer,
962
  memtrack_getwritebuffer,
963
  memtrack_getsegcount
964
};
965
966
static void memtrack_specials(void)
967
{
968
  memtrack_Type.tp_dealloc = memtrack_dealloc;
969
  memtrack_Type.tp_as_buffer = &memtrack_as_buffer;
970
}
971
972
/************************************************************************/
973
974
/* cvmemstorage */
975
976
static void cvmemstorage_dealloc(PyObject *self)
977
{
978
  cvmemstorage_t *ps = (cvmemstorage_t*)self;
979
  cvReleaseMemStorage(&(ps->a));
980
  PyObject_Del(self);
981
}
982
983
static PyTypeObject cvmemstorage_Type = {
984
  PyObject_HEAD_INIT(&PyType_Type)
985
  0,                                      /*size*/
986
  MODULESTR".cvmemstorage",               /*name*/
987
  sizeof(cvmemstorage_t),                 /*basicsize*/
988
};
989
990
static void cvmemstorage_specials(void)
991
{
992
  cvmemstorage_Type.tp_dealloc = cvmemstorage_dealloc;
993
}
994
995
/************************************************************************/
996
997
/* cvfont */
998
999
static PyTypeObject cvfont_Type = {
1000
  PyObject_HEAD_INIT(&PyType_Type)
1001
  0,                                      /*size*/
1002
  MODULESTR".cvfont",                     /*name*/
1003
  sizeof(cvfont_t),                       /*basicsize*/
1004
};
1005
1006
static void cvfont_specials(void) { }
1007
1008
/************************************************************************/
1009
1010
/* cvrng */
1011
1012
static PyTypeObject cvrng_Type = {
1013
  PyObject_HEAD_INIT(&PyType_Type)
1014
  0,                                      /*size*/
1015
  MODULESTR".cvrng",                     /*name*/
1016
  sizeof(cvrng_t),                       /*basicsize*/
1017
};
1018
1019
static void cvrng_specials(void)
1020
{
1021
}
1022
1023
/************************************************************************/
1024
1025
/* cvcontourtree */
1026
1027
static PyTypeObject cvcontourtree_Type = {
1028
  PyObject_HEAD_INIT(&PyType_Type)
1029
  0,                                      /*size*/
1030
  MODULESTR".cvcontourtree",                     /*name*/
1031
  sizeof(cvcontourtree_t),                       /*basicsize*/
1032
};
1033
1034
static void cvcontourtree_specials(void) { }
1035
1036
1037
/************************************************************************/
1038
1039
/* cvsubdiv2dedge */
1040
1041
static PyTypeObject cvsubdiv2dedge_Type = {
1042
  PyObject_HEAD_INIT(&PyType_Type)
1043
  0,                                      /*size*/
1044
  MODULESTR".cvsubdiv2dedge",                     /*name*/
1045
  sizeof(cvsubdiv2dedge_t),                       /*basicsize*/
1046
};
1047
1048
static int cvsubdiv2dedge_compare(PyObject *o1, PyObject *o2)
1049
{
1050
  cvsubdiv2dedge_t *e1 = (cvsubdiv2dedge_t*)o1;
1051
  cvsubdiv2dedge_t *e2 = (cvsubdiv2dedge_t*)o2;
1052
  if (e1->a < e2->a)
1053
    return -1;
1054
  else if (e1->a > e2->a)
1055
    return 1;
1056
  else
1057
    return 0;
1058
}
1059
1060
static PyObject *cvquadedge_repr(PyObject *self)
1061
{
1062
  CvSubdiv2DEdge m = ((cvsubdiv2dedge_t*)self)->a;
1063
  char str[1000];
1064
  sprintf(str, "<cvsubdiv2dedge(");
1065
  char *d = str + strlen(str);
1066
  sprintf(d, "%zx.%d", m & ~3, (int)(m & 3));
1067
  d += strlen(d);
1068
  sprintf(d, ")>");
1069
  return PyString_FromString(str);
1070
}
1071
1072
static void cvsubdiv2dedge_specials(void) {
1073
  cvsubdiv2dedge_Type.tp_compare = cvsubdiv2dedge_compare;
1074
  cvsubdiv2dedge_Type.tp_repr = cvquadedge_repr;
1075
}
1076
1077
/************************************************************************/
1078
1079
/* cvseq */
1080
1081
static void cvseq_dealloc(PyObject *self)
1082
{
1083
  cvseq_t *ps = (cvseq_t*)self;
1084
  Py_DECREF(ps->container);
1085
  PyObject_Del(self);
1086
}
1087
1088
static PyObject *cvseq_h_next(PyObject *self, PyObject *args);
1089
static PyObject *cvseq_h_prev(PyObject *self, PyObject *args);
1090
static PyObject *cvseq_v_next(PyObject *self, PyObject *args);
1091
static PyObject *cvseq_v_prev(PyObject *self, PyObject *args);
1092
1093
static struct PyMethodDef cvseq_methods[] =
1094
{
1095
  {"h_next", cvseq_h_next, METH_VARARGS},
1096
  {"h_prev", cvseq_h_prev, METH_VARARGS},
1097
  {"v_next", cvseq_v_next, METH_VARARGS},
1098
  {"v_prev", cvseq_v_prev, METH_VARARGS},
1099
  {NULL,          NULL}
1100
};
1101
1102
static Py_ssize_t cvseq_seq_length(PyObject *o)
1103
{
1104
  cvseq_t *ps = (cvseq_t*)o;
1105
  if (ps->a == NULL)
1106
    return (Py_ssize_t)0;
1107
  else
1108
    return (Py_ssize_t)(ps->a->total);
1109
}
1110
1111
static PyObject* cvseq_seq_getitem(PyObject *o, Py_ssize_t i)
1112
{
1113
  cvseq_t *ps = (cvseq_t*)o;
1114
  CvPoint *pt;
1115
  struct pointpair{
1116
    CvPoint a, b;
1117
  } *pp;
1118
  CvPoint2D32f *pt2;
1119
  CvPoint3D32f *pt3;
1120
1121
  if (i < cvseq_seq_length(o)) {
1122
    switch (CV_SEQ_ELTYPE(ps->a)) {
1123
1124
    case CV_SEQ_ELTYPE_POINT:
1125
      pt = CV_GET_SEQ_ELEM(CvPoint, ps->a, i);
1126
      return Py_BuildValue("ii", pt->x, pt->y);
1127
1128
    case CV_SEQ_ELTYPE_GENERIC:
1129
      switch (ps->a->elem_size) {
1130
      case sizeof(CvQuadEdge2D):
1131
        {
1132
          cvsubdiv2dedge_t *r = PyObject_NEW(cvsubdiv2dedge_t, &cvsubdiv2dedge_Type);
1133
          r->a = (CvSubdiv2DEdge)CV_GET_SEQ_ELEM(CvQuadEdge2D, ps->a, i);
1134
          r->container = ps->container;
1135
          Py_INCREF(r->container);
1136
          return (PyObject*)r;
1137
        }
1138
      case sizeof(CvConnectedComp):
1139
        {
1140
          CvConnectedComp *cc = CV_GET_SEQ_ELEM(CvConnectedComp, ps->a, i);
1141
          return FROM_CvConnectedComp(*cc);
1142
        }
1143
      default:
1144
        printf("seq elem size is %d\n", ps->a->elem_size);
1145
        printf("KIND %d\n", CV_SEQ_KIND(ps->a));
1146
        assert(0);
1147
      }
1148
      return PyInt_FromLong(*CV_GET_SEQ_ELEM(unsigned char, ps->a, i));
1149
1150
    case CV_SEQ_ELTYPE_PTR:
1151
    case CV_SEQ_ELTYPE_INDEX:
1152
      return PyInt_FromLong(*CV_GET_SEQ_ELEM(int, ps->a, i));
1153
1154
    case CV_32SC4:
1155
      pp = CV_GET_SEQ_ELEM(pointpair, ps->a, i);
1156
      return Py_BuildValue("(ii),(ii)", pp->a.x, pp->a.y, pp->b.x, pp->b.y);
1157
1158
    case CV_32FC2:
1159
      pt2 = CV_GET_SEQ_ELEM(CvPoint2D32f, ps->a, i);
1160
      return Py_BuildValue("ff", pt2->x, pt2->y);
1161
1162
    case CV_SEQ_ELTYPE_POINT3D:
1163
      pt3 = CV_GET_SEQ_ELEM(CvPoint3D32f, ps->a, i);
1164
      return Py_BuildValue("fff", pt3->x, pt3->y, pt3->z);
1165
1166
    default:
1167
      printf("Unknown element type %08x\n", CV_SEQ_ELTYPE(ps->a));
1168
      assert(0);
1169
      return NULL;
1170
    }
1171
  } else
1172
    return NULL;
1173
}
1174
1175
static PyObject* cvseq_map_getitem(PyObject *o, PyObject *item)
1176
{
1177
  if (PyInt_Check(item)) {
1178
    long i = PyInt_AS_LONG(item);
1179
    if (i < 0)
1180
      i += cvseq_seq_length(o);
1181
    return cvseq_seq_getitem(o, i);
1182
  } else if (PySlice_Check(item)) {
1183
    Py_ssize_t start, stop, step, slicelength, cur, i;
1184
    PyObject* result;
1185
1186
    if (PySlice_GetIndicesEx((PySliceObject*)item, cvseq_seq_length(o),
1187
         &start, &stop, &step, &slicelength) < 0) {
1188
      return NULL;
1189
    }
1190
1191
    if (slicelength <= 0) {
1192
      return PyList_New(0);
1193
    } else {
1194
      result = PyList_New(slicelength);
1195
      if (!result) return NULL;
1196
1197
      for (cur = start, i = 0; i < slicelength;
1198
           cur += step, i++) {
1199
        PyList_SET_ITEM(result, i, cvseq_seq_getitem(o, cur));
1200
      }
1201
1202
      return result;
1203
    }
1204
  } else {
1205
    PyErr_SetString(PyExc_TypeError, "CvSeq indices must be integers");
1206
    return NULL;
1207
  }
1208
}
1209
1210
static 
1211
PySequenceMethods cvseq_sequence = {
1212
  cvseq_seq_length,
1213
  NULL,
1214
  NULL,
1215
  cvseq_seq_getitem
1216
};
1217
1218
static PyMappingMethods cvseq_mapping = {
1219
  cvseq_seq_length,
1220
  cvseq_map_getitem,
1221
  NULL,
1222
};
1223
1224
static PyTypeObject cvseq_Type = {
1225
  PyObject_HEAD_INIT(&PyType_Type)
1226
  0,                                      /*size*/
1227
  MODULESTR".cvseq",                          /*name*/
1228
  sizeof(cvseq_t),                        /*basicsize*/
1229
};
1230
1231
static void cvseq_specials(void)
1232
{
1233
  cvseq_Type.tp_dealloc = cvseq_dealloc;
1234
  cvseq_Type.tp_as_sequence = &cvseq_sequence;
1235
  cvseq_Type.tp_as_mapping = &cvseq_mapping;
1236
  cvseq_Type.tp_methods = cvseq_methods;
1237
}
1238
1239
#define MK_ACCESSOR(FIELD) \
1240
static PyObject *cvseq_##FIELD(PyObject *self, PyObject *args) \
1241
{ \
1242
  cvseq_t *ps = (cvseq_t*)self; \
1243
  CvSeq *s = ps->a; \
1244
  if (s->FIELD == NULL) { \
1245
    Py_RETURN_NONE; \
1246
  } else { \
1247
    cvseq_t *r = PyObject_NEW(cvseq_t, &cvseq_Type); \
1248
    r->a = s->FIELD; \
1249
    r->container = ps->container; \
1250
    Py_INCREF(r->container); \
1251
    return (PyObject*)r; \
1252
  } \
1253
}
1254
1255
MK_ACCESSOR(h_next)
1256
MK_ACCESSOR(h_prev)
1257
MK_ACCESSOR(v_next)
1258
MK_ACCESSOR(v_prev)
1259
#undef MK_ACCESSOR
1260
1261
/************************************************************************/
1262
1263
/* cvset */
1264
1265
static void cvset_dealloc(PyObject *self)
1266
{
1267
  cvset_t *ps = (cvset_t*)self;
1268
  Py_DECREF(ps->container);
1269
  PyObject_Del(self);
1270
}
1271
1272
static PyTypeObject cvset_Type = {
1273
  PyObject_HEAD_INIT(&PyType_Type)
1274
  0,                                      /*size*/
1275
  MODULESTR".cvset",                          /*name*/
1276
  sizeof(cvset_t),                        /*basicsize*/
1277
};
1278
1279
static PyObject *cvset_iter(PyObject *o)
1280
{
1281
  Py_INCREF(o);
1282
  cvset_t *ps = (cvset_t*)o;
1283
  ps->i = 0;
1284
  return o;
1285
}
1286
1287
static PyObject *cvset_next(PyObject *o)
1288
{
1289
  cvset_t *ps = (cvset_t*)o;
1290
1291
  while (ps->i < ps->a->total) {
1292
    CvSetElem *e = cvGetSetElem(ps->a, ps->i);
1293
    int prev_i = ps->i++;
1294
    if (e != NULL) {
1295
      return cvseq_seq_getitem(o, prev_i);
1296
    }
1297
  }
1298
  return NULL;
1299
}
1300
1301
static void cvset_specials(void)
1302
{
1303
  cvset_Type.tp_dealloc = cvset_dealloc;
1304
  cvset_Type.tp_iter = cvset_iter;
1305
  cvset_Type.tp_iternext = cvset_next;
1306
}
1307
1308
/************************************************************************/
1309
1310
/* cvsubdiv2d */
1311
1312
static PyTypeObject cvsubdiv2d_Type = {
1313
  PyObject_HEAD_INIT(&PyType_Type)
1314
  0,                                          /*size*/
1315
  MODULESTR".cvsubdiv2d",                     /*name*/
1316
  sizeof(cvsubdiv2d_t),                       /*basicsize*/
1317
};
1318
1319
static PyObject *cvsubdiv2d_getattro(PyObject *o, PyObject *name)
1320
{
1321
  cvsubdiv2d_t *p = (cvsubdiv2d_t*)o;
1322
  if (strcmp(PyString_AsString(name), "edges") == 0) {
1323
    cvset_t *r = PyObject_NEW(cvset_t, &cvset_Type);
1324
    r->a = p->a->edges;
1325
    r->container = p->container;
1326
    Py_INCREF(r->container);
1327
    return (PyObject*)r;
1328
  } else {
1329
    PyErr_SetString(PyExc_TypeError, "cvsubdiv2d has no such attribute");
1330
    return NULL;
1331
  }
1332
}
1333
1334
static void cvsubdiv2d_specials(void)
1335
{
1336
  cvsubdiv2d_Type.tp_getattro = cvsubdiv2d_getattro;
1337
}
1338
1339
/************************************************************************/
1340
1341
/* cvsubdiv2dpoint */
1342
1343
static PyTypeObject cvsubdiv2dpoint_Type = {
1344
  PyObject_HEAD_INIT(&PyType_Type)
1345
  0,                                      /*size*/
1346
  MODULESTR".cvsubdiv2dpoint",                     /*name*/
1347
  sizeof(cvsubdiv2dpoint_t),                       /*basicsize*/
1348
};
1349
1350
static PyObject *cvsubdiv2dpoint_getattro(PyObject *o, PyObject *name)
1351
{
1352
  cvsubdiv2dpoint_t *p = (cvsubdiv2dpoint_t*)o;
1353
  if (strcmp(PyString_AsString(name), "first") == 0) {
1354
    cvsubdiv2dedge_t *r = PyObject_NEW(cvsubdiv2dedge_t, &cvsubdiv2dedge_Type);
1355
    r->a = p->a->first;
1356
    r->container = p->container;
1357
    Py_INCREF(r->container);
1358
    return (PyObject*)r;
1359
  } else if (strcmp(PyString_AsString(name), "pt") == 0) {
1360
    return Py_BuildValue("(ff)", p->a->pt.x, p->a->pt.y);
1361
  } else {
1362
    PyErr_SetString(PyExc_TypeError, "cvsubdiv2dpoint has no such attribute");
1363
    return NULL;
1364
  }
1365
}
1366
1367
static void cvsubdiv2dpoint_specials(void)
1368
{
1369
  cvsubdiv2dpoint_Type.tp_getattro = cvsubdiv2dpoint_getattro;
1370
}
1371
1372
/************************************************************************/
1373
/* convert_to_X: used after PyArg_ParseTuple in the generated code  */
1374
1375
static int convert_to_PyObjectPTR(PyObject *o, PyObject **dst, const char *name = "no_name")
1376
{
1377
  *dst = o;
1378
  return 1;
1379
}
1380
1381
static int convert_to_PyCallableObjectPTR(PyObject *o, PyObject **dst, const char *name = "no_name")
1382
{
1383
  *dst = o;
1384
  return 1;
1385
}
1386
1387
static int convert_to_char(PyObject *o, char *dst, const char *name = "no_name")
1388
{
1389
  if (PyString_Check(o) && PyString_Size(o) == 1) {
1390
    *dst = PyString_AsString(o)[0];
1391
    return 1;
1392
  } else {
1393
    (*dst) = 0;
1394
    return failmsg("Expected single character string for argument '%s'", name);
1395
  }
1396
}
1397
1398
static int convert_to_CvMemStorage(PyObject *o, CvMemStorage **dst, const char *name = "no_name")
1399
{
1400
  if (PyType_IsSubtype(o->ob_type, &cvmemstorage_Type)) {
1401
    (*dst) = (((cvmemstorage_t*)o)->a);
1402
    return 1;
1403
  } else {
1404
    (*dst) = (CvMemStorage*)NULL;
1405
    return failmsg("Expected CvMemStorage for argument '%s'", name);
1406
  }
1407
}
1408
1409
static int convert_to_CvSeq(PyObject *o, CvSeq **dst, const char *name = "no_name")
1410
{
1411
  if (PyType_IsSubtype(o->ob_type, &cvseq_Type)) {
1412
    (*dst) = (((cvseq_t*)o)->a);
1413
    return 1;
1414
  } else {
1415
    (*dst) = (CvSeq*)NULL;
1416
    return failmsg("Expected CvSeq for argument '%s'", name);
1417
  }
1418
}
1419
1420
static int convert_to_CvSize(PyObject *o, CvSize *dst, const char *name = "no_name")
1421
{
1422
  if (!PyArg_ParseTuple(o, "ii", &dst->width, &dst->height))
1423
    return failmsg("CvSize argument '%s' expects two integers", name);
1424
  else
1425
    return 1;
1426
}
1427
1428
static int convert_to_CvScalar(PyObject *o, CvScalar *s, const char *name = "no_name")
1429
{
1430
  if (PySequence_Check(o)) {
1431
    PyObject *fi = PySequence_Fast(o, name);
1432
    if (fi == NULL)
1433
      return 0;
1434
    if (4 < PySequence_Fast_GET_SIZE(fi))
1435
        return failmsg("CvScalar value for argument '%s' is longer than 4", name);
1436
    for (Py_ssize_t i = 0; i < PySequence_Fast_GET_SIZE(fi); i++) {
1437
      PyObject *item = PySequence_Fast_GET_ITEM(fi, i);
1438
      if (PyFloat_Check(item) || PyInt_Check(item)) {
1439
        s->val[i] = PyFloat_AsDouble(item);
1440
      } else {
1441
        return failmsg("CvScalar value for argument '%s' is not numeric", name);
1442
      }
1443
    }
1444
    Py_DECREF(fi);
1445
  } else {
1446
    if (PyFloat_Check(o) || PyInt_Check(o)) {
1447
      s->val[0] = PyFloat_AsDouble(o);
1448
    } else {
1449
      return failmsg("CvScalar value for argument '%s' is not numeric", name);
1450
    }
1451
  }
1452
  return 1;
1453
}
1454
1455
static int convert_to_CvPointPTR(PyObject *o, CvPoint **p, const char *name = "no_name")
1456
{
1457
  if (!PySequence_Check(o))
1458
    return failmsg("Expected sequence for point list argument '%s'", name);
1459
  PyObject *fi = PySequence_Fast(o, name);
1460
  if (fi == NULL)
1461
    return 0;
1462
  *p = new CvPoint[PySequence_Fast_GET_SIZE(fi)];
1463
  for (Py_ssize_t i = 0; i < PySequence_Fast_GET_SIZE(fi); i++) {
1464
    PyObject *item = PySequence_Fast_GET_ITEM(fi, i);
1465
    if (!PyTuple_Check(item))
1466
      return failmsg("Expected tuple for element in point list argument '%s'", name);
1467
    if (!PyArg_ParseTuple(item, "ii", &((*p)[i].x), &((*p)[i].y))) {
1468
      return 0;
1469
    }
1470
  }
1471
  Py_DECREF(fi);
1472
  return 1;
1473
}
1474
1475
static int convert_to_CvPoint2D32fPTR(PyObject *o, CvPoint2D32f **p, const char *name = "no_name")
1476
{
1477
  PyObject *fi = PySequence_Fast(o, name);
1478
  if (fi == NULL)
1479
    return 0;
1480
  *p = new CvPoint2D32f[PySequence_Fast_GET_SIZE(fi)];
1481
  for (Py_ssize_t i = 0; i < PySequence_Fast_GET_SIZE(fi); i++) {
1482
    PyObject *item = PySequence_Fast_GET_ITEM(fi, i);
1483
    if (!PyTuple_Check(item))
1484
      return failmsg("Expected tuple for CvPoint2D32f argument '%s'", name);
1485
    if (!PyArg_ParseTuple(item, "ff", &((*p)[i].x), &((*p)[i].y))) {
1486
      return 0;
1487
    }
1488
  }
1489
  Py_DECREF(fi);
1490
  return 1;
1491
}
1492
1493
#if 0 // not used
1494
static int convert_to_CvPoint3D32fPTR(PyObject *o, CvPoint3D32f **p, const char *name = "no_name")
1495
{
1496
  PyObject *fi = PySequence_Fast(o, name);
1497
  if (fi == NULL)
1498
    return 0;
1499
  *p = new CvPoint3D32f[PySequence_Fast_GET_SIZE(fi)];
1500
  for (Py_ssize_t i = 0; i < PySequence_Fast_GET_SIZE(fi); i++) {
1501
    PyObject *item = PySequence_Fast_GET_ITEM(fi, i);
1502
    if (!PyTuple_Check(item))
1503
      return failmsg("Expected tuple for CvPoint3D32f argument '%s'", name);
1504
    if (!PyArg_ParseTuple(item, "fff", &((*p)[i].x), &((*p)[i].y), &((*p)[i].z))) {
1505
      return 0;
1506
    }
1507
  }
1508
  Py_DECREF(fi);
1509
  return 1;
1510
}
1511
#endif
1512
1513
static int convert_to_CvStarDetectorParams(PyObject *o, CvStarDetectorParams *dst, const char *name = "no_name")
1514
{
1515
  if (!PyArg_ParseTuple(o,
1516
                        "iiiii",
1517
                        &dst->maxSize,
1518
                        &dst->responseThreshold,
1519
                        &dst->lineThresholdProjected,
1520
                        &dst->lineThresholdBinarized,
1521
                        &dst->suppressNonmaxSize))
1522
    return failmsg("CvRect argument '%s' expects four integers", name);
1523
  else
1524
    return 1;
1525
}
1526
1527
static int convert_to_CvRect(PyObject *o, CvRect *dst, const char *name = "no_name")
1528
{
1529
  if (!PyArg_ParseTuple(o, "iiii", &dst->x, &dst->y, &dst->width, &dst->height))
1530
    return failmsg("CvRect argument '%s' expects four integers", name);
1531
  else
1532
    return 1;
1533
}
1534
1535
static int convert_to_CvRectPTR(PyObject *o, CvRect **dst, const char *name = "no_name")
1536
{
1537
  *dst = new CvRect;
1538
  if (!PyArg_ParseTuple(o, "iiii", &(*dst)->x, &(*dst)->y, &(*dst)->width, &(*dst)->height))
1539
    return failmsg("CvRect argument '%s' expects four integers", name);
1540
  else
1541
    return 1;
1542
}
1543
1544
static int convert_to_CvSlice(PyObject *o, CvSlice *dst, const char *name = "no_name")
1545
{
1546
  if (!PyArg_ParseTuple(o, "ii", &dst->start_index, &dst->end_index))
1547
    return failmsg("CvSlice argument '%s' expects two integers", name);
1548
  else
1549
    return 1;
1550
}
1551
1552
static int convert_to_CvPoint(PyObject *o, CvPoint *dst, const char *name = "no_name")
1553
{
1554
  if (!PyArg_ParseTuple(o, "ii", &dst->x, &dst->y))
1555
    return failmsg("CvPoint argument '%s' expects two integers", name);
1556
  else
1557
    return 1;
1558
}
1559
1560
static int convert_to_CvPoint2D32f(PyObject *o, CvPoint2D32f *dst, const char *name = "no_name")
1561
{
1562
  if (!PyArg_ParseTuple(o, "ff", &dst->x, &dst->y))
1563
    return failmsg("CvPoint2D32f argument '%s' expects two floats", name);
1564
  else
1565
    return 1;
1566
}
1567
1568
static int convert_to_CvPoint3D32f(PyObject *o, CvPoint3D32f *dst, const char *name = "no_name")
1569
{
1570
  if (!PyArg_ParseTuple(o, "fff", &dst->x, &dst->y, &dst->z))
1571
    return failmsg("CvPoint3D32f argument '%s' expects three floats", name);
1572
  else
1573
    return 1;
1574
}
1575
1576
static int convert_to_IplImage(PyObject *o, IplImage **dst, const char *name)
1577
{
1578
  iplimage_t *ipl = (iplimage_t*)o;
1579
  void *buffer;
1580
  Py_ssize_t buffer_len;
1581
1582
  if (!is_iplimage(o)) {
1583
    return failmsg("Argument '%s' must be IplImage", name);
1584
  } else if (PyString_Check(ipl->data)) {
1585
    cvSetData(ipl->a, PyString_AsString(ipl->data) + ipl->offset, ipl->a->widthStep);
1586
    assert(cvGetErrStatus() == 0);
1587
    *dst = ipl->a;
1588
    return 1;
1589
  } else if (ipl->data && PyObject_AsWriteBuffer(ipl->data, &buffer, &buffer_len) == 0) {
1590
    cvSetData(ipl->a, (void*)((char*)buffer + ipl->offset), ipl->a->widthStep);
1591
    assert(cvGetErrStatus() == 0);
1592
    *dst = ipl->a;
1593
    return 1;
1594
  } else {
1595
    return failmsg("IplImage argument '%s' has no data", name);
1596
  }
1597
}
1598
1599
static int convert_to_CvMat(PyObject *o, CvMat **dst, const char *name)
1600
{
1601
  cvmat_t *m = (cvmat_t*)o;
1602
  void *buffer;
1603
  Py_ssize_t buffer_len;
1604
1605
  if (!is_cvmat(o)) {
1606
#if !PYTHON_USE_NUMPY
1607
    return failmsg("Argument '%s' must be CvMat", name);
1608
#else
1609
    PyObject *asmat = fromarray(o, 0);
1610
    if (asmat == NULL)
1611
      return failmsg("Argument '%s' must be CvMat", name);
1612
    // now have the array obect as a cvmat, can use regular conversion
1613
    return convert_to_CvMat(asmat, dst, name);
1614
#endif
1615
  } else {
1616
    m->a->refcount = NULL;
1617
    if (m->data && PyString_Check(m->data)) {
1618
      assert(cvGetErrStatus() == 0);
1619
      char *ptr = PyString_AsString(m->data) + m->offset;
1620
      cvSetData(m->a, ptr, m->a->step);
1621
      assert(cvGetErrStatus() == 0);
1622
      *dst = m->a;
1623
      return 1;
1624
    } else if (m->data && PyObject_AsWriteBuffer(m->data, &buffer, &buffer_len) == 0) {
1625
      cvSetData(m->a, (void*)((char*)buffer + m->offset), m->a->step);
1626
      assert(cvGetErrStatus() == 0);
1627
      *dst = m->a;
1628
      return 1;
1629
    } else {
1630
      return failmsg("CvMat argument '%s' has no data", name);
1631
    }
1632
  }
1633
}
1634
1635
static int convert_to_CvMatND(PyObject *o, CvMatND **dst, const char *name)
1636
{
1637
  cvmatnd_t *m = (cvmatnd_t*)o;
1638
  void *buffer;
1639
  Py_ssize_t buffer_len;
1640
1641
  if (!is_cvmatnd(o)) {
1642
    return failmsg("Argument '%s' must be CvMatND", name);
1643
  } else if (m->data && PyString_Check(m->data)) {
1644
    m->a->data.ptr = ((uchar*)PyString_AsString(m->data)) + m->offset;
1645
    *dst = m->a;
1646
    return 1;
1647
  } else if (m->data && PyObject_AsWriteBuffer(m->data, &buffer, &buffer_len) == 0) {
1648
    m->a->data.ptr = ((uchar*)buffer + m->offset);
1649
    *dst = m->a;
1650
    return 1;
1651
  } else {
1652
    return failmsg("CvMatND argument '%s' has no data", name);
1653
  }
1654
}
1655
1656
static int convert_to_CvArr(PyObject *o, CvArr **dst, const char *name)
1657
{
1658
  if (o == Py_None) {
1659
    *dst = (void*)NULL;
1660
    return 1;
1661
  } else if (is_iplimage(o)) {
1662
    return convert_to_IplImage(o, (IplImage**)dst, name);
1663
  } else if (is_cvmat(o)) {
1664
    return convert_to_CvMat(o, (CvMat**)dst, name);
1665
  } else if (is_cvmatnd(o)) {
1666
    return convert_to_CvMatND(o, (CvMatND**)dst, name);
1667
  } else {
1668
#if !PYTHON_USE_NUMPY
1669
    return failmsg("CvArr argument '%s' must be IplImage, CvMat or CvMatND", name);
1670
#else
1671
    PyObject *asmat = fromarray(o, 0);
1672
    if (asmat == NULL)
1673
      return failmsg("CvArr argument '%s' must be IplImage, CvMat, CvMatND, or support the array interface", name);
1674
    // now have the array obect as a cvmat, can use regular conversion
1675
    return convert_to_CvArr(asmat, dst, name);
1676
#endif
1677
  }
1678
}
1679
1680
static int convert_to_CvHistogram(PyObject *o, CvHistogram **dst, const char *name = "no_name")
1681
{
1682
  if (PyType_IsSubtype(o->ob_type, &cvhistogram_Type)) {
1683
    cvhistogram_t *ht = (cvhistogram_t*)o;
1684
    *dst = &ht->h;
1685
    return convert_to_CvArr(ht->bins, &(ht->h.bins), "bins");
1686
  } else {
1687
    *dst = (CvHistogram *)NULL;
1688
    return failmsg("Expected CvHistogram for argument '%s'", name);
1689
  }
1690
}
1691
1692
// Used by FillPoly, FillConvexPoly, PolyLine
1693
struct pts_npts_contours {
1694
  CvPoint** pts;
1695
  int* npts;
1696
  int contours;
1697
};
1698
1699
static int convert_to_pts_npts_contours(PyObject *o, pts_npts_contours *dst, const char *name = "no_name")
1700
{
1701
  PyObject *fi = PySequence_Fast(o, name);
1702
  if (fi == NULL)
1703
    return 0;
1704
  dst->contours = PySequence_Fast_GET_SIZE(fi);
1705
  dst->pts = new CvPoint*[dst->contours];
1706
  dst->npts = new int[dst->contours];
1707
  for (Py_ssize_t i = 0; i < PySequence_Fast_GET_SIZE(fi); i++) {
1708
    if (!convert_to_CvPointPTR(PySequence_Fast_GET_ITEM(fi, i), &dst->pts[i], name))
1709
      return 0;
1710
    dst->npts[i] = PySequence_Size(PySequence_Fast_GET_ITEM(fi, i)); // safe because convert_ just succeeded
1711
  }
1712
  Py_DECREF(fi);
1713
  return 1;
1714
}
1715
1716
struct cvarrseq {
1717
  union {
1718
    CvSeq *seq;
1719
    CvArr *mat;
1720
  };
1721
};
1722
1723
static int is_convertible_to_mat(PyObject *o)
1724
{
1725
#if PYTHON_USE_NUMPY
1726
  if (PyObject_HasAttrString(o, "__array_struct__")) {
1727
    PyObject *ao = PyObject_GetAttrString(o, "__array_struct__");
1728
    if (ao != NULL &&
1729
        PyCObject_Check(ao) &&
1730
        ((PyArrayInterface*)PyCObject_AsVoidPtr(ao))->two == 2) {
1731
      return 1;
1732
    }
1733
  }
1734
#endif
1735
  return is_iplimage(o) && is_cvmat(o) && is_cvmatnd(o);
1736
}
1737
1738
static int convert_to_cvarrseq(PyObject *o, cvarrseq *dst, const char *name = "no_name")
1739
{
1740
  if (PyType_IsSubtype(o->ob_type, &cvseq_Type)) {
1741
    return convert_to_CvSeq(o, &(dst->seq), name);
1742
  } else if (is_convertible_to_mat(o)) {
1743
    int r = convert_to_CvArr(o, &(dst->mat), name);
1744
    return r;
1745
  } else if (PySequence_Check(o)) {
1746
    PyObject *fi = PySequence_Fast(o, name);
1747
    if (fi == NULL)
1748
      return 0;
1749
    Py_ssize_t size = -1;
1750
    // Make a pass through the sequence, checking that each element is
1751
    // a sequence and that they are all the same size
1752
    for (Py_ssize_t i = 0; i < PySequence_Fast_GET_SIZE(fi); i++) {
1753
      PyObject *e = PySequence_Fast_GET_ITEM(fi, i);
1754
1755
      if (!PySequence_Check(e))
1756
        return failmsg("Sequence '%s' must contain sequences", name);
1757
      if (i == 0)
1758
        size = (int)PySequence_Size(e);
1759
      else if (size != PySequence_Size(e))
1760
        return failmsg("All elements of sequence '%s' must be same size", name);
1761
    }
1762
    assert(size != -1);
1763
    CvMat *mt = cvCreateMat((int)PySequence_Fast_GET_SIZE(fi), 1, CV_32SC(size));
1764
    for (Py_ssize_t i = 0; i < PySequence_Fast_GET_SIZE(fi); i++) {
1765
      PyObject *e = PySequence_Fast_GET_ITEM(fi, i);
1766
      PyObject *fe = PySequence_Fast(e, name);
1767
      assert(fe != NULL);
1768
      int *pdst = (int*)cvPtr2D(mt, i, 0);
1769
      for (Py_ssize_t j = 0; j < size; j++) {
1770
        PyObject *num = PySequence_Fast_GET_ITEM(fe, j);
1771
        if (!PyNumber_Check(num)) {
1772
          return failmsg("Sequence must contain numbers", name);
1773
        }
1774
        *pdst++ = PyInt_AsLong(num);
1775
      }
1776
      Py_DECREF(fe);
1777
    }
1778
    Py_DECREF(fi);
1779
    dst->mat = mt;
1780
    return 1;
1781
  } else {
1782
    return failmsg("Argument '%s' must be CvSeq, CvArr, or a sequence of numbers");
1783
  }
1784
}
1785
1786
struct cvarr_count {
1787
  CvArr **cvarr;
1788
  int count;
1789
};
1790
1791
static int convert_to_cvarr_count(PyObject *o, cvarr_count *dst, const char *name = "no_name")
1792
{
1793
  PyObject *fi = PySequence_Fast(o, name);
1794
  if (fi == NULL)
1795
    return 0;
1796
  dst->count = PySequence_Fast_GET_SIZE(fi);
1797
  dst->cvarr = new CvArr*[dst->count];
1798
  for (Py_ssize_t i = 0; i < PySequence_Fast_GET_SIZE(fi); i++) {
1799
    if (!convert_to_CvArr(PySequence_Fast_GET_ITEM(fi, i), &dst->cvarr[i], name))
1800
      return 0;
1801
  }
1802
  Py_DECREF(fi);
1803
  return 1;
1804
}
1805
1806
struct intpair
1807
{
1808
  int *pairs;
1809
  int count;
1810
};
1811
1812
static int convert_to_intpair(PyObject *o, intpair *dst, const char *name = "no_name")
1813
{
1814
  PyObject *fi = PySequence_Fast(o, name);
1815
  if (fi == NULL)
1816
    return 0;
1817
  dst->count = PySequence_Fast_GET_SIZE(fi);
1818
  dst->pairs = new int[2 * dst->count];
1819
  for (Py_ssize_t i = 0; i < PySequence_Fast_GET_SIZE(fi); i++) {
1820
    PyObject *item = PySequence_Fast_GET_ITEM(fi, i);
1821
    if (!PyArg_ParseTuple(item, "ii", &dst->pairs[2 * i], &dst->pairs[2 * i + 1])) {
1822
      return 0;
1823
    }
1824
  }
1825
  Py_DECREF(fi);
1826
  return 1;
1827
}
1828
1829
struct cvpoint2d32f_count {
1830
  CvPoint2D32f* points;
1831
  int count;
1832
};
1833
1834
static int convert_to_cvpoint2d32f_count(PyObject *o, cvpoint2d32f_count *dst, const char *name = "no_name")
1835
{
1836
  if (PyInt_Check(o)) {
1837
    dst->count = PyInt_AsLong(o);
1838
    dst->points = new CvPoint2D32f[dst->count];
1839
    return 1;
1840
  } else {
1841
    return failmsg("Expected integer for CvPoint2D32f count");
1842
  }
1843
}
1844
1845
struct floats {
1846
  float *f;
1847
  int count;
1848
};
1849
static int convert_to_floats(PyObject *o, floats *dst, const char *name = "no_name")
1850
{
1851
  if (PySequence_Check(o)) {
1852
    PyObject *fi = PySequence_Fast(o, name);
1853
    if (fi == NULL)
1854
      return 0;
1855
    dst->count = PySequence_Fast_GET_SIZE(fi);
1856
    dst->f = new float[dst->count];
1857
    for (Py_ssize_t i = 0; i < PySequence_Fast_GET_SIZE(fi); i++) {
1858
      PyObject *item = PySequence_Fast_GET_ITEM(fi, i);
1859
      dst->f[i] = (float)PyFloat_AsDouble(item);
1860
    }
1861
    Py_DECREF(fi);
1862
  } else if (PyNumber_Check(o)) {
1863
    dst->count = 1;
1864
    dst->f = new float[1];
1865
    dst->f[0] = (float)PyFloat_AsDouble(o);
1866
  } else {
1867
    return failmsg("Expected list of floats, or float for argument '%s'", name);
1868
  }
1869
  return 1;
1870
}
1871
1872
struct chars {
1873
  char *f;
1874
  int count;
1875
};
1876
/// convert_to_chars not used
1877
1878
struct CvPoints {
1879
  CvPoint *p;
1880
  int count;
1881
};
1882
static int convert_to_CvPoints(PyObject *o, CvPoints *dst, const char *name = "no_name")
1883
{
1884
  PyObject *fi = PySequence_Fast(o, name);
1885
  if (fi == NULL)
1886
    return 0;
1887
  dst->count = PySequence_Fast_GET_SIZE(fi);
1888
  dst->p = new CvPoint[dst->count];
1889
  for (Py_ssize_t i = 0; i < PySequence_Fast_GET_SIZE(fi); i++) {
1890
    PyObject *item = PySequence_Fast_GET_ITEM(fi, i);
1891
    convert_to_CvPoint(item, &dst->p[i], name);
1892
  }
1893
  Py_DECREF(fi);
1894
  return 1;
1895
}
1896
1897
struct CvPoint3D32fs {
1898
  CvPoint3D32f *p;
1899
  int count;
1900
};
1901
static int convert_to_CvPoint3D32fs(PyObject *o, CvPoint3D32fs *dst, const char *name = "no_name")
1902
{
1903
  PyObject *fi = PySequence_Fast(o, name);
1904
  if (fi == NULL)
1905
    return 0;
1906
  dst->count = PySequence_Fast_GET_SIZE(fi);
1907
  dst->p = new CvPoint3D32f[dst->count];
1908
  for (Py_ssize_t i = 0; i < PySequence_Fast_GET_SIZE(fi); i++) {
1909
    PyObject *item = PySequence_Fast_GET_ITEM(fi, i);
1910
    convert_to_CvPoint3D32f(item, &dst->p[i], name);
1911
  }
1912
  Py_DECREF(fi);
1913
  return 1;
1914
}
1915
1916
struct CvPoint2D32fs {
1917
  CvPoint2D32f *p;
1918
  int count;
1919
};
1920
static int convert_to_CvPoint2D32fs(PyObject *o, CvPoint2D32fs *dst, const char *name = "no_name")
1921
{
1922
  PyObject *fi = PySequence_Fast(o, name);
1923
  if (fi == NULL)
1924
    return 0;
1925
  dst->count = PySequence_Fast_GET_SIZE(fi);
1926
  dst->p = new CvPoint2D32f[dst->count];
1927
  for (Py_ssize_t i = 0; i < PySequence_Fast_GET_SIZE(fi); i++) {
1928
    PyObject *item = PySequence_Fast_GET_ITEM(fi, i);
1929
    convert_to_CvPoint2D32f(item, &dst->p[i], name);
1930
  }
1931
  Py_DECREF(fi);
1932
  return 1;
1933
}
1934
1935
struct ints {
1936
  int *i;
1937
  int count;
1938
};
1939
static int convert_to_ints(PyObject *o, ints *dst, const char *name = "no_name")
1940
{
1941
  PyObject *fi = PySequence_Fast(o, name);
1942
  if (fi == NULL)
1943
    return 0;
1944
  dst->count = PySequence_Fast_GET_SIZE(fi);
1945
  dst->i = new int[dst->count];
1946
  for (Py_ssize_t i = 0; i < PySequence_Fast_GET_SIZE(fi); i++) {
1947
    PyObject *item = PySequence_Fast_GET_ITEM(fi, i);
1948
    dst->i[i] = PyInt_AsLong(item);
1949
  }
1950
  Py_DECREF(fi);
1951
  return 1;
1952
}
1953
1954
struct ints0 {
1955
  int *i;
1956
  int count;
1957
};
1958
static int convert_to_ints0(PyObject *o, ints0 *dst, const char *name = "no_name")
1959
{
1960
  PyObject *fi = PySequence_Fast(o, name);
1961
  if (fi == NULL)
1962
    return 0;
1963
  dst->count = PySequence_Fast_GET_SIZE(fi);
1964
  dst->i = new int[dst->count + 1];
1965
  for (Py_ssize_t i = 0; i < PySequence_Fast_GET_SIZE(fi); i++) {
1966
    PyObject *item = PySequence_Fast_GET_ITEM(fi, i);
1967
    dst->i[i] = PyInt_AsLong(item);
1968
  }
1969
  dst->i[dst->count] = 0;
1970
  Py_DECREF(fi);
1971
  return 1;
1972
}
1973
1974
struct dims
1975
{
1976
  int count;
1977
  int i[CV_MAX_DIM];
1978
  int step[CV_MAX_DIM];
1979
  int length[CV_MAX_DIM];
1980
};
1981
1982
static int convert_to_dim(PyObject *item, int i, dims *dst, CvArr *cva, const char *name = "no_name")
1983
{
1984
  if (PySlice_Check(item)) {
1985
    Py_ssize_t start, stop, step, slicelength;
1986
    PySlice_GetIndicesEx((PySliceObject*)item, cvGetDimSize(cva, i), &start, &stop, &step, &slicelength);
1987
    dst->i[i] = start;
1988
    dst->step[i] = step;
1989
    dst->length[i] = slicelength;
1990
  } else {
1991
    int index = PyInt_AsLong(item);
1992
    if (0 <= index)
1993
      dst->i[i] = index;
1994
    else
1995
      dst->i[i] = cvGetDimSize(cva, i) + index;
1996
    dst->step[i] = 0;
1997
    dst->length[i] = 1;
1998
  }
1999
  return 1;
2000
}
2001
2002
static int convert_to_dims(PyObject *o, dims *dst, CvArr *cva, const char *name = "no_name")
2003
{
2004
  if (!PyTuple_Check(o)) {
2005
    dst->count = 1;
2006
    return convert_to_dim(o, 0, dst, cva, name);
2007
  } else {
2008
    PyObject *fi = PySequence_Fast(o, name);
2009
    if (fi == NULL) {
2010
      PyErr_SetString(PyExc_TypeError, "Expected tuple for index");
2011
      return 0;
2012
    }
2013
    dst->count = PySequence_Fast_GET_SIZE(fi);
2014
    for (Py_ssize_t i = 0; i < PySequence_Fast_GET_SIZE(fi); i++) {
2015
      if (i >= cvGetDims(cva)) {
2016
        return failmsg("Access specifies %d dimensions, but array only has %d", PySequence_Fast_GET_SIZE(fi), cvGetDims(cva));
2017
      }
2018
      PyObject *item = PySequence_Fast_GET_ITEM(fi, i);
2019
      if (!convert_to_dim(item, i, dst, cva, name))
2020
        return 0;
2021
    }
2022
    Py_DECREF(fi);
2023
    return 1;
2024
  }
2025
}
2026
2027
struct IplImages {
2028
  IplImage **ims;
2029
  int count;
2030
};
2031
static int convert_to_IplImages(PyObject *o, IplImages *dst, const char *name = "no_name")
2032
{
2033
  PyObject *fi = PySequence_Fast(o, name);
2034
  if (fi == NULL)
2035
    return 0;
2036
  dst->count = PySequence_Fast_GET_SIZE(fi);
2037
  dst->ims = new IplImage*[dst->count];
2038
  for (Py_ssize_t i = 0; i < PySequence_Fast_GET_SIZE(fi); i++) {
2039
    PyObject *item = PySequence_Fast_GET_ITEM(fi, i);
2040
    if (!convert_to_IplImage(item, &dst->ims[i]))
2041
      return 0;
2042
  }
2043
  Py_DECREF(fi);
2044
  return 1;
2045
}
2046
2047
struct CvArrs {
2048
  CvArr **ims;
2049
  int count;
2050
};
2051
static int convert_to_CvArrs(PyObject *o, CvArrs *dst, const char *name = "no_name")
2052
{
2053
  PyObject *fi = PySequence_Fast(o, name);
2054
  if (fi == NULL)
2055
    return 0;
2056
  dst->count = PySequence_Fast_GET_SIZE(fi);
2057
  dst->ims = new CvArr*[dst->count];
2058
  for (Py_ssize_t i = 0; i < PySequence_Fast_GET_SIZE(fi); i++) {
2059
    PyObject *item = PySequence_Fast_GET_ITEM(fi, i);
2060
    if (!convert_to_CvArr(item, &dst->ims[i]))
2061
      return 0;
2062
  }
2063
  Py_DECREF(fi);
2064
  return 1;
2065
}
2066
2067
static int convert_to_floatPTRPTR(PyObject *o, float*** dst, const char *name = "no_name")
2068
{
2069
  PyObject *fi = PySequence_Fast(o, name);
2070
  if (fi == NULL)
2071
    return 0;
2072
  Py_ssize_t sz = PySequence_Fast_GET_SIZE(fi);
2073
  float **r = new float*[sz];
2074
  for (Py_ssize_t i = 0; i < PySequence_Fast_GET_SIZE(fi); i++) {
2075
    PyObject *item = PySequence_Fast_GET_ITEM(fi, i);
2076
    floats ff;
2077
    if (!convert_to_floats(item, &ff))
2078
      return 0;
2079
    r[i] = ff.f;
2080
  }
2081
  *dst = r;
2082
  return 1;
2083
}
2084
2085
static int convert_to_CvFontPTR(PyObject *o, CvFont** dst, const char *name = "no_name")
2086
{
2087
  if (PyType_IsSubtype(o->ob_type, &cvfont_Type)) {
2088
    (*dst) = &(((cvfont_t*)o)->a);
2089
    return 1;
2090
  } else {
2091
    (*dst) = (CvFont*)NULL;
2092
    return failmsg("Expected CvFont for argument '%s'", name);
2093
  }
2094
}
2095
2096
static int convert_to_CvContourTreePTR(PyObject *o, CvContourTree** dst, const char *name = "no_name")
2097
{
2098
  if (PyType_IsSubtype(o->ob_type, &cvcontourtree_Type)) {
2099
    (*dst) = ((cvcontourtree_t*)o)->a;
2100
    return 1;
2101
  } else {
2102
    (*dst) = NULL;
2103
    return failmsg("Expected CvContourTree for argument '%s'", name);
2104
  }
2105
}
2106
2107
static int convert_to_CvRNGPTR(PyObject *o, CvRNG** dst, const char *name = "no_name")
2108
{
2109
  if (PyType_IsSubtype(o->ob_type, &cvrng_Type)) {
2110
    (*dst) = &(((cvrng_t*)o)->a);
2111
    return 1;
2112
  } else {
2113
    (*dst) = (CvRNG*)NULL;
2114
    return failmsg("Expected CvRNG for argument '%s'", name);
2115
  }
2116
}
2117
2118
typedef void* generic;
2119
static int convert_to_generic(PyObject *o, generic *dst, const char *name = "no_name")
2120
{
2121
  if (PyType_IsSubtype(o->ob_type, &iplimage_Type))
2122
    return convert_to_IplImage(o, (IplImage**)dst, name);
2123
  else if (PyType_IsSubtype(o->ob_type, &cvmat_Type))
2124
    return convert_to_CvMat(o, (CvMat**)dst, name);
2125
  else if (PyType_IsSubtype(o->ob_type, &cvmatnd_Type))
2126
    return convert_to_CvMatND(o, (CvMatND**)dst, name);
2127
  else {
2128
    return failmsg("Cannot identify type of '%s'", name);
2129
  }
2130
}
2131
2132
static int convert_to_CvTermCriteria(PyObject *o, CvTermCriteria* dst, const char *name = "no_name")
2133
{
2134
  if (!PyArg_ParseTuple(o, "iid", &dst->type, &dst->max_iter, &dst->epsilon))
2135
    return 0;
2136
  return 1;
2137
}
2138
2139
static int convert_to_CvBox2D(PyObject *o, CvBox2D* dst, const char *name = "no_name")
2140
{
2141
  if (!PyArg_ParseTuple(o, "(ff)(ff)f", &dst->center.x, &dst->center.y, &dst->size.width, &dst->size.height, &dst->angle))
2142
    return 0;
2143
  return 1;
2144
}
2145
2146
static int convert_to_CvSubdiv2DPTR(PyObject *o, CvSubdiv2D** dst, const char *name = "no_name")
2147
{
2148
  if (PyType_IsSubtype(o->ob_type, &cvsubdiv2d_Type)) {
2149
    (*dst) = (((cvsubdiv2d_t*)o)->a);
2150
    return 1;
2151
  } else {
2152
    (*dst) = (CvSubdiv2D*)NULL;
2153
    return failmsg("Expected CvSubdiv2D for argument '%s'", name);
2154
  }
2155
}
2156
2157
static int convert_to_CvNextEdgeType(PyObject *o, CvNextEdgeType *dst, const char *name = "no_name")
2158
{
2159
  if (!PyInt_Check(o)) {
2160
    *dst = (CvNextEdgeType)NULL;
2161
    return failmsg("Expected number for CvNextEdgeType argument '%s'", name);
2162
  } else {
2163
    *dst = (CvNextEdgeType)PyInt_AsLong(o);
2164
    return 1;
2165
  }
2166
}
2167
2168
static int convert_to_CvSubdiv2DEdge(PyObject *o, CvSubdiv2DEdge *dst, const char *name = "no_name")
2169
{
2170
  if (PyType_IsSubtype(o->ob_type, &cvsubdiv2dedge_Type)) {
2171
    (*dst) = (((cvsubdiv2dedge_t*)o)->a);
2172
    return 1;
2173
  } else {
2174
    *dst = 0L;
2175
    return failmsg("Expected CvSubdiv2DEdge for argument '%s'", name);
2176
  }
2177
}
2178
2179
/************************************************************************/
2180
2181
static PyObject *pythonize_CvMat(cvmat_t *m)
2182
{
2183
  // Need to make this CvMat look like any other, with a Python 
2184
  // buffer object as its data.
2185
  CvMat *mat = m->a;
2186
  assert(mat->step != 0);
2187
#if 0
2188
  PyObject *data = PyString_FromStringAndSize((char*)(mat->data.ptr), mat->rows * mat->step);
2189
#else
2190
  memtrack_t *o = PyObject_NEW(memtrack_t, &memtrack_Type);
2191
  size_t gap = mat->data.ptr - (uchar*)mat->refcount;
2192
  o->ptr = mat->refcount;
2193
  o->size = gap + mat->rows * mat->step;
2194
  PyObject *data = PyBuffer_FromReadWriteObject((PyObject*)o, (size_t)gap, mat->rows * mat->step);
2195
  if (data == NULL)
2196
    return NULL;
2197
#endif
2198
  m->data = data;
2199
  m->offset = 0;
2200
  Py_DECREF(o);
2201
2202
  // Now m has a reference to data, which has a reference to o.
2203
2204
  return (PyObject*)m;
2205
}
2206
2207
static PyObject *pythonize_foreign_CvMat(cvmat_t *m)
2208
{
2209
  // Need to make this CvMat look like any other, with a Python 
2210
  // buffer object as its data.
2211
  // Difference here is that the buffer is 'foreign' (from NumPy, for example)
2212
  CvMat *mat = m->a;
2213
  assert(mat->step != 0);
2214
#if 0
2215
  PyObject *data = PyString_FromStringAndSize((char*)(mat->data.ptr), mat->rows * mat->step);
2216
#else
2217
  memtrack_t *o = PyObject_NEW(memtrack_t, &memtrack_Type);
2218
  o->ptr = mat->data.ptr;
2219
  o->size = mat->rows * mat->step;
2220
  PyObject *data = PyBuffer_FromReadWriteObject((PyObject*)o, (size_t)0, mat->rows * mat->step);
2221
  if (data == NULL)
2222
    return NULL;
2223
  Py_INCREF(o);   // XXX - hack to prevent free of this foreign memory
2224
#endif
2225
  m->data = data;
2226
  m->offset = 0;
2227
  Py_DECREF(o);
2228
2229
  // Now m has a reference to data, which has a reference to o.
2230
2231
  return (PyObject*)m;
2232
}
2233
2234
static PyObject *pythonize_IplImage(iplimage_t *cva)
2235
{
2236
  // Need to make this iplimage look like any other, with a Python 
2237
  // string as its data.
2238
  // So copy the image data into a Python string object, then release 
2239
  // it.
2240
2241
  IplImage *ipl = (IplImage*)(cva->a);
2242
  // PyObject *data = PyString_FromStringAndSize(ipl->imageData, ipl->imageSize);
2243
2244
  memtrack_t *o = PyObject_NEW(memtrack_t, &memtrack_Type);
2245
  assert(ipl->imageDataOrigin == ipl->imageData);
2246
  o->ptr = ipl->imageDataOrigin;
2247
  o->size = ipl->height * ipl->widthStep;
2248
  PyObject *data = PyBuffer_FromReadWriteObject((PyObject*)o, (size_t)0, o->size);
2249
  if (data == NULL)
2250
    return NULL;
2251
  Py_DECREF(o);
2252
  cva->data = data;
2253
  cva->offset = 0;
2254
2255
  return (PyObject*)cva;
2256
}
2257
2258
static PyObject *pythonize_CvMatND(cvmatnd_t *m)
2259
{
2260
  //
2261
  // Need to make this CvMatND look like any other, with a Python 
2262
  // string as its data.
2263
  // So copy the image data into a Python string object, then release 
2264
  // it.
2265
  //
2266
2267
  CvMatND *mat = m->a;
2268
  assert(mat->dim[0].step != 0);
2269
#if 0
2270
  PyObject *data = PyString_FromStringAndSize((char*)(mat->data.ptr), mat->dim[0].size * mat->dim[0].step);
2271
#else
2272
  memtrack_t *o = PyObject_NEW(memtrack_t, &memtrack_Type);
2273
  o->ptr = cvPtr1D(mat, 0);
2274
  o->size = cvmatnd_size(mat);
2275
  PyObject *data = PyBuffer_FromReadWriteObject((PyObject*)o, (size_t)0, o->size);
2276
  if (data == NULL)
2277
    return NULL;
2278
#endif
2279
  m->data = data;
2280
  m->offset = 0;
2281
  // cvDecRefData(mat); // Ref count should be zero here, so this is a release
2282
2283
  return (PyObject*)m;
2284
}
2285
2286
/************************************************************************/
2287
/* FROM_xxx:   C -> Python converters.
2288
 *
2289
 * Turn various OpenCV types (and some aggregate types above)
2290
 * into Python objects.  Used by the generated code.
2291
 *
2292
 * All these functions and macros return a new reference.
2293
 */
2294
2295
2296
static PyObject *_FROM_CvSeqPTR(CvSeq *s, PyObject *storage)
2297
{
2298
  cvseq_t *ps = PyObject_NEW(cvseq_t, &cvseq_Type);
2299
  ps->a = s;
2300
  ps->container = storage;
2301
  Py_INCREF(ps->container);
2302
  return (PyObject*)ps;
2303
}
2304
2305
static PyObject *_FROM_CvSubdiv2DPTR(CvSubdiv2D *s, PyObject *storage)
2306
{
2307
  cvsubdiv2d_t *ps = PyObject_NEW(cvsubdiv2d_t, &cvsubdiv2d_Type);
2308
  ps->a = s;
2309
  ps->container = storage;
2310
  Py_INCREF(ps->container);
2311
  return (PyObject*)ps;
2312
}
2313
2314
static PyObject *FROM_floats(floats r)
2315
{
2316
  PyObject *pr;
2317
2318
  pr = PyList_New(r.count);
2319
  for (Py_ssize_t i = 0; i < (Py_ssize_t)r.count; i++) {
2320
    PyList_SetItem(pr, i, PyFloat_FromDouble(r.f[i]));
2321
  }
2322
  return pr;
2323
}
2324
2325
static PyObject *FROM_chars(chars r)
2326
{
2327
  PyObject *pr;
2328
2329
  pr = PyList_New(r.count);
2330
  for (Py_ssize_t i = 0; i < (Py_ssize_t)r.count; i++) {
2331
    PyList_SetItem(pr, i, PyInt_FromLong(r.f[i]));
2332
  }
2333
  return pr;
2334
}
2335
2336
static PyObject *FROM_cvpoint2d32f_count(cvpoint2d32f_count r)
2337
{
2338
  PyObject *pr;
2339
2340
  pr = PyList_New(r.count);
2341
  for (Py_ssize_t i = 0; i < (Py_ssize_t)r.count; i++) {
2342
    PyList_SetItem(pr, i, FROM_CvPoint2D32f(r.points[i]));
2343
  }
2344
  return pr;
2345
}
2346
2347
static PyObject *FROM_CvPoint2D32fs(CvPoint2D32fs r)
2348
{
2349
  PyObject *pr;
2350
2351
  pr = PyList_New(r.count);
2352
  for (Py_ssize_t i = 0; i < (Py_ssize_t)r.count; i++) {
2353
    PyList_SetItem(pr, i, FROM_CvPoint2D32f(r.p[i]));
2354
  }
2355
  return pr;
2356
}
2357
2358
typedef CvSeq CvSeqOfCvConvexityDefect;
2359
static PyObject *FROM_CvSeqOfCvConvexityDefectPTR(CvSeqOfCvConvexityDefect *r)
2360
{
2361
  PyObject *pr;
2362
  pr = PyList_New(r->total);
2363
  for (int i = 0; i < r->total; i++) {
2364
    CvConvexityDefect *pd = CV_GET_SEQ_ELEM(CvConvexityDefect, r, i);
2365
    PyList_SetItem(pr, i, Py_BuildValue("(ii)(ii)(ii)f",
2366
                                        pd->start->x, pd->start->y, 
2367
                                        pd->end->x, pd->end->y, 
2368
                                        pd->depth_point->x, pd->depth_point->y, 
2369
                                        pd->depth));
2370
  }
2371
  // This function has copied the CvSeq data into a list.  Hence the
2372
  // CvSeq is not being returned to the caller.  Hence, no reference
2373
  // count increase for the storage, unlike _FROM_CvSeqPTR.
2374
  return pr;
2375
}
2376
2377
typedef CvSeq CvSeqOfCvAvgComp;
2378
static PyObject *FROM_CvSeqOfCvAvgCompPTR(CvSeqOfCvAvgComp *r)
2379
{
2380
  PyObject *pr;
2381
  pr = PyList_New(r->total);
2382
  for (int i = 0; i < r->total; i++) {
2383
    CvAvgComp *pd = CV_GET_SEQ_ELEM(CvAvgComp, r, i);
2384
    PyList_SetItem(pr, i, Py_BuildValue("(iiii)i",
2385
                                        pd->rect.x, pd->rect.y, 
2386
                                        pd->rect.width, pd->rect.height, 
2387
                                        pd->neighbors));
2388
  }
2389
  // This function has copied the CvSeq data into a list.  Hence the
2390
  // CvSeq is not being returned to the caller.  Hence, no reference
2391
  // count increase for the storage, unlike _FROM_CvSeqPTR.
2392
  return pr;
2393
}
2394
2395
typedef CvSeq CvSeqOfCvStarKeypoint;
2396
static PyObject *FROM_CvSeqOfCvStarKeypointPTR(CvSeqOfCvStarKeypoint *r)
2397
{
2398
  PyObject *pr;
2399
  pr = PyList_New(r->total);
2400
  for (int i = 0; i < r->total; i++) {
2401
    CvStarKeypoint *pd = CV_GET_SEQ_ELEM(CvStarKeypoint, r, i);
2402
    PyList_SetItem(pr, i, Py_BuildValue("(ii)if",
2403
                                        pd->pt.x, pd->pt.y, 
2404
                                        pd->size,
2405
                                        pd->response));
2406
  }
2407
  // This function has copied the CvSeq data into a list.  Hence the
2408
  // CvSeq is not being returned to the caller.  Hence, no reference
2409
  // count increase for the storage, unlike _FROM_CvSeqPTR.
2410
  return pr;
2411
}
2412
2413
typedef CvSeq CvSeqOfCvSURFPoint;
2414
static PyObject *FROM_CvSeqOfCvSURFPointPTR(CvSeqOfCvSURFPoint *r)
2415
{
2416
  PyObject *pr;
2417
  pr = PyList_New(r->total);
2418
  for (int i = 0; i < r->total; i++) {
2419
    CvSURFPoint *pd = CV_GET_SEQ_ELEM(CvSURFPoint, r, i);
2420
    PyList_SetItem(pr, i, Py_BuildValue("(ff)iiff",
2421
                                        pd->pt.x, pd->pt.y, 
2422
                                        pd->laplacian,
2423
                                        pd->size,
2424
                                        pd->dir,
2425
                                        pd->hessian));
2426
  }
2427
  // This function has copied the CvSeq data into a list.  Hence the
2428
  // CvSeq is not being returned to the caller.  Hence, no reference
2429
  // count increase for the storage, unlike _FROM_CvSeqPTR.
2430
  return pr;
2431
}
2432
2433
typedef CvSeq CvSeqOfCvSURFDescriptor;
2434
static PyObject *FROM_CvSeqOfCvSURFDescriptorPTR(CvSeqOfCvSURFDescriptor *r)
2435
{
2436
  PyObject *pr;
2437
  pr = PyList_New(r->total);
2438
  for (int i = 0; i < r->total; i++) {
2439
    float *pd = (float*)cvGetSeqElem(r, i);
2440
    int count = r->elem_size / sizeof(float);
2441
    PyObject *oi = PyList_New(count);
2442
    for (int j = 0; j < count; j++) {
2443
      PyList_SetItem(oi, j, PyFloat_FromDouble(pd[j]));
2444
    }
2445
    PyList_SetItem(pr, i, oi);
2446
  }
2447
  // This function has copied the CvSeq data into a list.  Hence the
2448
  // CvSeq is not being returned to the caller.  Hence, no reference
2449
  // count increase for the storage, unlike _FROM_CvSeqPTR.
2450
  return pr;
2451
}
2452
2453
typedef CvPoint2D32f CvPoint2D32f_4[4];
2454
static PyObject *FROM_CvPoint2D32f_4(CvPoint2D32f* r)
2455
{
2456
  return Py_BuildValue("(ff)(ff)(ff)(ff)",
2457
                       r[0].x, r[0].y,
2458
                       r[1].x, r[1].y,
2459
                       r[2].x, r[2].y,
2460
                       r[3].x, r[3].y);
2461
}
2462
2463
typedef float CvMatr32f_i[9];
2464
2465
static PyObject *FROM_CvMatr32f_i(CvMatr32f_i r)
2466
{
2467
  return Py_BuildValue("(fff)(fff)(fff)",
2468
    r[0], r[1], r[2],
2469
    r[3], r[4], r[5],
2470
    r[6], r[7], r[8]);
2471
}
2472
2473
typedef float CvVect32f_i[3];
2474
static PyObject *FROM_CvVect32f_i(CvVect32f_i r)
2475
{
2476
  return Py_BuildValue("fff",
2477
    r[0], r[1], r[2]);
2478
}
2479
2480
static PyObject *FROM_CvFont(CvFont r)
2481
{
2482
  cvfont_t *cf = PyObject_NEW(cvfont_t, &cvfont_Type);
2483
  cf->a = r;
2484
  return (PyObject*)cf;
2485
}
2486
2487
static PyObject *FROM_CvSubdiv2DPointPTR(CvSubdiv2DPoint* r)
2488
{
2489
  if (r != NULL) {
2490
    cvsubdiv2dpoint_t *cf = PyObject_NEW(cvsubdiv2dpoint_t, &cvsubdiv2dpoint_Type);
2491
    cf->a = r;
2492
    return (PyObject*)cf;
2493
  } else {
2494
    Py_INCREF(Py_None);
2495
    return Py_None;
2496
  }
2497
}
2498
2499
static PyObject *FROM_IplImagePTR(IplImage *r)
2500
{
2501
  iplimage_t *cva = PyObject_NEW(iplimage_t, &iplimage_Type);
2502
  cva->a = r;
2503
  return pythonize_IplImage(cva);
2504
}
2505
2506
static PyObject *FROM_ROIplImagePTR(ROIplImage *r)
2507
{
2508
  if (r != NULL) {
2509
    iplimage_t *cva = PyObject_NEW(iplimage_t, &iplimage_Type);
2510
    cva->a = cvCreateImageHeader(cvSize(100,100), 8, 1);
2511
    *(cva->a) = *r;
2512
    cva->data = PyBuffer_FromReadWriteMemory(r->imageData, r->height * r->widthStep);
2513
    cva->offset = 0;
2514
    return (PyObject*)cva;
2515
  } else {
2516
    Py_RETURN_NONE;
2517
  }
2518
}
2519
2520
static PyObject *FROM_ROCvMatPTR(ROCvMat *r)
2521
{
2522
  if (r != NULL) {
2523
    cvmat_t *cva = PyObject_NEW(cvmat_t, &cvmat_Type);
2524
    cva->a = cvCreateMatHeader(100, 100, CV_8U);
2525
    *(cva->a) = *r;
2526
    cva->data = PyBuffer_FromReadWriteMemory(r->data.ptr, r->rows * r->step);
2527
    cva->offset = 0;
2528
    return (PyObject*)cva;
2529
  } else {
2530
    Py_RETURN_NONE;
2531
  }
2532
}
2533
2534
static PyObject *FROM_CvMatPTR(CvMat *r)
2535
{
2536
  cvmat_t *cvm = PyObject_NEW(cvmat_t, &cvmat_Type);
2537
  cvm->a = r;
2538
2539
  return pythonize_CvMat(cvm);
2540
}
2541
2542
static PyObject *FROM_CvMat(CvMat *r)
2543
{
2544
  cvmat_t *m = PyObject_NEW(cvmat_t, &cvmat_Type);
2545
  m->a = r;
2546
  return pythonize_CvMat(m);
2547
}
2548
2549
static PyObject *FROM_CvMatNDPTR(CvMatND *r)
2550
{
2551
  cvmatnd_t *m = PyObject_NEW(cvmatnd_t, &cvmatnd_Type);
2552
  m->a = r;
2553
  return pythonize_CvMatND(m);
2554
}
2555
2556
static PyObject *FROM_CvRNG(CvRNG r)
2557
{
2558
  cvrng_t *m = PyObject_NEW(cvrng_t, &cvrng_Type);
2559
  m->a = r;
2560
  return (PyObject*)m;
2561
}
2562
2563
static PyObject *FROM_CvContourTreePTR(CvContourTree *r)
2564
{
2565
  cvcontourtree_t *m = PyObject_NEW(cvcontourtree_t, &cvcontourtree_Type);
2566
  m->a = r;
2567
  return (PyObject*)m;
2568
}
2569
2570
static PyObject *FROM_generic(generic r)
2571
{
2572
  CvTypeInfo* t = cvTypeOf(r);
2573
  if (r == NULL) {
2574
    failmsg("OpenCV returned NULL");
2575
    return NULL;
2576
  } if (strcmp(t->type_name, "opencv-image") == 0)
2577
    return FROM_IplImagePTR((IplImage*)r);
2578
  else if (strcmp(t->type_name, "opencv-matrix") == 0)
2579
    return FROM_CvMat((CvMat*)r);
2580
  else if (strcmp(t->type_name, "opencv-nd-matrix") == 0)
2581
    return FROM_CvMatNDPTR((CvMatND*)r);
2582
  else if (strcmp(t->type_name, "opencv-haar-classifier") == 0)
2583
    return FROM_CvHaarClassifierCascadePTR((CvHaarClassifierCascade*)r);
2584
  else {
2585
    failmsg("Unknown OpenCV type '%s'", t->type_name);
2586
    return NULL;
2587
  }
2588
}
2589
2590
static PyObject *FROM_CvSubdiv2DEdge(CvSubdiv2DEdge r)
2591
{
2592
  cvsubdiv2dedge_t *m = PyObject_NEW(cvsubdiv2dedge_t, &cvsubdiv2dedge_Type);
2593
  m->a = r;
2594
  m->container = Py_None; // XXX
2595
  Py_INCREF(m->container);
2596
  return (PyObject*)m;
2597
}
2598
2599
static PyObject *FROM_CvPoints(CvPoints src)
2600
{
2601
  PyObject *pr;
2602
  pr = PyList_New(src.count);
2603
  for (int i = 0; i < src.count; i++) {
2604
    PyList_SetItem(pr, i, FROM_CvPoint(src.p[i]));
2605
  }
2606
  return pr;
2607
}
2608
2609
/************************************************************************/
2610
2611
/* A few functions are too odd to be generated, 
2612
 * so are handwritten here */
2613
2614
static PyObject *pycvWaitKey(PyObject *self, PyObject *args, PyObject *kw)
2615
{
2616
  int delay = 0;
2617
2618
  const char *keywords[] = { "delay", NULL };
2619
  if (!PyArg_ParseTupleAndKeywords(args, kw, "|i", (char**)keywords, &delay))
2620
    return NULL;
2621
  int r;
2622
  Py_BEGIN_ALLOW_THREADS
2623
  r = cvWaitKey(delay);
2624
  Py_END_ALLOW_THREADS
2625
  return FROM_int(r);
2626
}
2627
2628
static PyObject *pycvLoadImage(PyObject *self, PyObject *args, PyObject *kw)
2629
{
2630
  const char *keywords[] = { "filename", "iscolor", NULL };
2631
  char *filename;
2632
  int iscolor = CV_LOAD_IMAGE_COLOR;
2633
2634
  if (!PyArg_ParseTupleAndKeywords(args, kw, "s|i", (char**)keywords, &filename, &iscolor))
2635
    return NULL;
2636
2637
  // Inside ALLOW_THREADS, must not reference 'filename' because it might move.
2638
  // So make a local copy 'filename_copy'.
2639
  char filename_copy[2048];
2640
  strncpy(filename_copy, filename, sizeof(filename_copy));
2641
2642
  IplImage *r;
2643
  Py_BEGIN_ALLOW_THREADS
2644
  r = cvLoadImage(filename_copy, iscolor);
2645
  Py_END_ALLOW_THREADS
2646
2647
  if (r == NULL) {
2648
    PyErr_SetFromErrnoWithFilename(PyExc_IOError, filename);
2649
    return NULL;
2650
  } else {
2651
    return FROM_IplImagePTR(r);
2652
  }
2653
}
2654
2655
static PyObject *pycvLoadImageM(PyObject *self, PyObject *args, PyObject *kw)
2656
{
2657
  const char *keywords[] = { "filename", "iscolor", NULL };
2658
  char *filename;
2659
  int iscolor = CV_LOAD_IMAGE_COLOR;
2660
2661
  if (!PyArg_ParseTupleAndKeywords(args, kw, "s|i", (char**)keywords, &filename, &iscolor))
2662
    return NULL;
2663
2664
  // Inside ALLOW_THREADS, must not reference 'filename' because it might move.
2665
  // So make a local copy 'filename_copy'.
2666
  char filename_copy[2048];
2667
  strncpy(filename_copy, filename, sizeof(filename_copy));
2668
2669
  CvMat *r;
2670
  Py_BEGIN_ALLOW_THREADS
2671
  r = cvLoadImageM(filename_copy, iscolor);
2672
  Py_END_ALLOW_THREADS
2673
2674
  if (r == NULL) {
2675
    PyErr_SetFromErrnoWithFilename(PyExc_IOError, filename);
2676
    return NULL;
2677
  } else {
2678
    return FROM_CvMatPTR(r);
2679
  }
2680
}
2681
2682
static PyObject *pycvCreateImageHeader(PyObject *self, PyObject *args)
2683
{
2684
  int w, h, depth, channels;
2685
  if (!PyArg_ParseTuple(args, "(ii)Ii", &w, &h, &depth, &channels))
2686
    return NULL;
2687
  iplimage_t *cva = PyObject_NEW(iplimage_t, &iplimage_Type);
2688
  cva->a = cvCreateImageHeader(cvSize(w, h), depth, channels);
2689
  if (cva->a == NULL) {
2690
    PyErr_SetString(PyExc_TypeError, "CreateImage failed");
2691
    return NULL;
2692
  } else {
2693
    cva->data = Py_None;
2694
    Py_INCREF(cva->data);
2695
    cva->offset = 0;
2696
2697
    return (PyObject*)cva;
2698
  }
2699
}
2700
2701
static PyObject *pycvCreateImage(PyObject *self, PyObject *args)
2702
{
2703
  int w, h, depth, channels;
2704
  if (!PyArg_ParseTuple(args, "(ii)Ii:CreateImage", &w, &h, &depth, &channels))
2705
    return NULL;
2706
  iplimage_t *cva = PyObject_NEW(iplimage_t, &iplimage_Type);
2707
  ERRWRAP(cva->a = cvCreateImage(cvSize(w, h), depth, channels));
2708
  if (cva->a == NULL) {
2709
    PyErr_SetString(PyExc_TypeError, "CreateImage failed");
2710
    return NULL;
2711
  } else {
2712
    return pythonize_IplImage(cva);
2713
  }
2714
}
2715
2716
static PyObject *pycvCreateMatHeader(PyObject *self, PyObject *args)
2717
{
2718
  int rows, cols, type;
2719
  if (!PyArg_ParseTuple(args, "iii", &rows, &cols, &type))
2720
    return NULL;
2721
  cvmat_t *m = PyObject_NEW(cvmat_t, &cvmat_Type);
2722
  ERRWRAP(m->a = cvCreateMatHeader(rows, cols, type));
2723
  if (m->a == NULL) {
2724
    PyErr_SetString(PyExc_TypeError, "CreateMat failed");
2725
    return NULL;
2726
  } else {
2727
    m->data = Py_None;
2728
    Py_INCREF(m->data);
2729
    m->offset = 0;
2730
    return (PyObject*)m;
2731
  }
2732
}
2733
2734
static PyObject *pycvCreateMat(PyObject *self, PyObject *args)
2735
{
2736
  int rows, cols, type;
2737
  if (!PyArg_ParseTuple(args, "iii", &rows, &cols, &type))
2738
    return NULL;
2739
  cvmat_t *m = PyObject_NEW(cvmat_t, &cvmat_Type);
2740
  ERRWRAP(m->a = cvCreateMat(rows, cols, type));
2741
  if (m->a == NULL) {
2742
    PyErr_SetString(PyExc_TypeError, "CreateMat failed");
2743
    return NULL;
2744
  } else {
2745
    return pythonize_CvMat(m);
2746
  }
2747
}
2748
2749
static PyObject *pycvCreateMatNDHeader(PyObject *self, PyObject *args)
2750
{
2751
  ints dims;
2752
  int type;
2753
2754
  if (!PyArg_ParseTuple(args, "O&i", convert_to_ints, (void*)&dims, &type))
2755
    return NULL;
2756
  cvmatnd_t *m = PyObject_NEW(cvmatnd_t, &cvmatnd_Type);
2757
  ERRWRAP(m->a = cvCreateMatNDHeader(dims.count, dims.i, type));
2758
2759
  m->data = Py_None;
2760
  Py_INCREF(m->data);
2761
  return (PyObject*)m;
2762
}
2763
2764
2765
static PyObject *pycvCreateMatND(PyObject *self, PyObject *args)
2766
{
2767
  ints dims;
2768
  int type;
2769
2770
  if (!PyArg_ParseTuple(args, "O&i", convert_to_ints, (void*)&dims, &type))
2771
    return NULL;
2772
  cvmatnd_t *m = PyObject_NEW(cvmatnd_t, &cvmatnd_Type);
2773
  ERRWRAP(m->a = cvCreateMatND(dims.count, dims.i, type));
2774
  return pythonize_CvMatND(m);
2775
}
2776
2777
#if PYTHON_USE_NUMPY
2778
static PyObject *pycvfromarray(PyObject *self, PyObject *args, PyObject *kw)
2779
{
2780
  const char *keywords[] = { "arr", "allowND", NULL };
2781
  PyObject *o;
2782
  int allowND = 0;
2783
2784
  if (!PyArg_ParseTupleAndKeywords(args, kw, "O|i", (char**)keywords, &o, &allowND))
2785
    return NULL;
2786
  return fromarray(o, allowND);
2787
}
2788
2789
static PyObject *fromarray(PyObject *o, int allowND)
2790
{
2791
  PyObject *retval;
2792
  PyObject *ao = PyObject_GetAttrString(o, "__array_struct__");
2793
  if ((ao == NULL) || !PyCObject_Check(ao)) {
2794
    PyErr_SetString(PyExc_TypeError, "object does not have array interface");
2795
    return NULL;
2796
  }
2797
  PyArrayInterface *pai = (PyArrayInterface*)PyCObject_AsVoidPtr(ao);
2798
  if (pai->two != 2) {
2799
    PyErr_SetString(PyExc_TypeError, "object does not have array interface");
2800
    Py_DECREF(ao);
2801
        return NULL;
2802
  }
2803
2804
  int type = -1;
2805
2806
  switch (pai->typekind) {
2807
  case 'i':
2808
    if (pai->itemsize == 1)
2809
      type = CV_8SC1;
2810
    else if (pai->itemsize == 2)
2811
      type = CV_16SC1;
2812
    else if (pai->itemsize == 4)
2813
      type = CV_32SC1;
2814
    else if (pai->itemsize == 8) {
2815
      PyErr_SetString(PyExc_TypeError, "OpenCV cannot handle 64-bit integer arrays");
2816
      Py_DECREF(ao);
2817
          return NULL;
2818
    }
2819
    break;
2820
2821
  case 'u':
2822
    if (pai->itemsize == 1)
2823
      type = CV_8UC1;
2824
    else if (pai->itemsize == 2)
2825
      type = CV_16UC1;
2826
    break;
2827
2828
  case 'f':
2829
    if (pai->itemsize == 4)
2830
      type = CV_32FC1;
2831
    else if (pai->itemsize == 8)
2832
      type = CV_64FC1;
2833
    break;
2834
2835
  }
2836
  assert(type != -1);
2837
2838
  if (!allowND) {
2839
    cvmat_t *m = PyObject_NEW(cvmat_t, &cvmat_Type);
2840
    if (pai->nd == 2) {
2841
      if (pai->strides[1] != pai->itemsize) {
2842
            Py_DECREF(ao);
2843
        return (PyObject*)failmsg("cv.fromarray array can only accept arrays with contiguous data");
2844
      }
2845
      ERRWRAP(m->a = cvCreateMatHeader(pai->shape[0], pai->shape[1], type));
2846
      m->a->step = pai->strides[0];
2847
    } else if (pai->nd == 3) {
2848
                if (pai->shape[2] > CV_CN_MAX){
2849
                        Py_DECREF(ao);
2850
                        return (PyObject*)failmsg("cv.fromarray too many channels, see allowND argument");
2851
                }
2852
      ERRWRAP(m->a = cvCreateMatHeader(pai->shape[0], pai->shape[1], type + ((pai->shape[2] - 1) << CV_CN_SHIFT)));
2853
      m->a->step = pai->strides[0];
2854
    } else {
2855
          Py_DECREF(ao);
2856
      return (PyObject*)failmsg("cv.fromarray array can be 2D or 3D only, see allowND argument");
2857
    }
2858
    m->a->data.ptr = (uchar*)pai->data;
2859
    m->data = o;
2860
    m->offset = 0;
2861
    retval = (PyObject*) m;
2862
  } else {
2863
    int dims[CV_MAX_DIM];
2864
    int i;
2865
    for (i = 0; i < pai->nd; i++)
2866
      dims[i] = pai->shape[i];
2867
    cvmatnd_t *m = PyObject_NEW(cvmatnd_t, &cvmatnd_Type);
2868
    ERRWRAP(m->a = cvCreateMatND(pai->nd, dims, type));
2869
    m->a->data.ptr = (uchar*)pai->data;
2870
        m->data = o;
2871
        m->offset = 0;
2872
    retval = (PyObject*) m;
2873
  }
2874
  Py_DECREF(ao);
2875
  Py_INCREF(o);
2876
  return retval;
2877
}
2878
#endif
2879
2880
static PyObject *pycvCreateHist(PyObject *self, PyObject *args, PyObject *kw)
2881
{
2882
  const char *keywords[] = { "dims", "type", "ranges", "uniform", NULL };
2883
  PyObject *dims;
2884
  int type;
2885
  float **ranges = NULL;
2886
  int uniform = 1;
2887
2888
  if (!PyArg_ParseTupleAndKeywords(args, kw, "Oi|O&i", (char**)keywords, &dims, &type, convert_to_floatPTRPTR, (void*)&ranges, &uniform)) {
2889
    return NULL;
2890
  }
2891
  cvhistogram_t *h = PyObject_NEW(cvhistogram_t, &cvhistogram_Type);
2892
  args = Py_BuildValue("Oi", dims, CV_32FC1);
2893
  h->bins = pycvCreateMatND(self, args);
2894
  Py_DECREF(args);
2895
  if (h->bins == NULL) {
2896
    return NULL;
2897
  }
2898
  h->h.type = CV_HIST_MAGIC_VAL;
2899
  if (!convert_to_CvArr(h->bins, &(h->h.bins), "bins"))
2900
    return NULL;
2901
2902
  ERRWRAP(cvSetHistBinRanges(&(h->h), ranges, uniform));
2903
2904
  return (PyObject*)h;
2905
}
2906
2907
static PyObject *pycvInitLineIterator(PyObject *self, PyObject *args, PyObject *kw)
2908
{
2909
  const char *keywords[] = { "image", "pt1", "pt2", "connectivity", "left_to_right", NULL };
2910
  CvArr *image;
2911
  CvPoint pt1;
2912
  CvPoint pt2;
2913
  int connectivity = 8;
2914
  int left_to_right = 0;
2915
2916
  if (!PyArg_ParseTupleAndKeywords(args, kw, "O&O&O&|ii", (char**)keywords,
2917
                        convert_to_CvArr, &image,
2918
                        convert_to_CvPoint, &pt1,
2919
                        convert_to_CvPoint, &pt2,
2920
                        &connectivity,
2921
                        &left_to_right))
2922
    return NULL;
2923
2924
  cvlineiterator_t *pi = PyObject_NEW(cvlineiterator_t, &cvlineiterator_Type);
2925
  pi->count = cvInitLineIterator(image, pt1, pt2, &pi->iter, connectivity, left_to_right);
2926
  ERRWRAP(pi->type = cvGetElemType(image));
2927
  return (PyObject*)pi;
2928
}
2929
2930
static PyObject *pycvCreateMemStorage(PyObject *self, PyObject *args)
2931
{
2932
  int block_size = 0;
2933
  if (!PyArg_ParseTuple(args, "|i", &block_size))
2934
    return NULL;
2935
  cvmemstorage_t *pm = PyObject_NEW(cvmemstorage_t, &cvmemstorage_Type);
2936
  pm->a = cvCreateMemStorage(block_size);
2937
  return (PyObject*)pm;
2938
}
2939
2940
// single index: return row
2941
// 2 indices: row, column
2942
// both row and column can be slices.  column slice must have a step of 1.
2943
//
2944
// returns a scalar when all dimensions are specified and all are integers.  Otherwise returns a CvMat.
2945
//
2946
static PyObject *cvarr_GetItem(PyObject *o, PyObject *key)
2947
{
2948
  dims dd;
2949
2950
  CvArr *cva;
2951
  if (!convert_to_CvArr(o, &cva, "src"))
2952
    return NULL;
2953
2954
  if (!convert_to_dims(key, &dd, cva, "key")) {
2955
    return NULL;
2956
  }
2957
2958
  // Figure out if all supplied indices have a stride of zero - means they are not slices
2959
  // and if all indices are positive
2960
  int all0 = 1;
2961
  for (int i = 0; i < dd.count; i++) {
2962
    all0 &= (dd.step[i] == 0) && (0 <= dd.i[i]);
2963
  }
2964
2965
  // if every dimension supplied, and none are slices, return the scalar
2966
  if ((cvGetDims(cva) == dd.count) && all0) {
2967
    CvScalar s;
2968
    ERRWRAP(s = cvGetND(cva, dd.i));
2969
    return PyObject_FromCvScalar(s, cvGetElemType(cva));
2970
  } else {
2971
    // pad missing dimensions
2972
    for (int i = dd.count; i < cvGetDims(cva); i++) {
2973
      dd.i[i] = 0;
2974
      dd.step[i] = 1;
2975
      dd.length[i] = cvGetDimSize(cva, i);
2976
    }
2977
    dd.count = cvGetDims(cva);
2978
2979
    // negative steps are illegal for OpenCV
2980
    for (int i = 0; i < dd.count; i++) {
2981
      if (dd.step[i] < 0)
2982
        return (PyObject*)failmsg("Negative step is illegal");
2983
    }
2984
2985
    // zero length illegal for OpenCV
2986
    for (int i = 0; i < dd.count; i++) {
2987
      if (dd.length[i] == 0)
2988
        return (PyObject*)failmsg("Zero sized dimension is illegal");
2989
    }
2990
2991
    // column step can only be 0 or 1
2992
    if ((dd.step[dd.count-1] != 0) && (dd.step[dd.count-1] != 1))
2993
        return (PyObject*)failmsg("Column step is illegal");
2994
2995
    if (is_cvmat(o) || is_iplimage(o)) {
2996
      cvmat_t *sub = PyObject_NEW(cvmat_t, &cvmat_Type);
2997
      sub->a = cvCreateMatHeader(dd.length[0], dd.length[1], cvGetElemType(cva));
2998
      uchar *old0;  // pointer to first element in old mat
2999
      int oldstep;
3000
      cvGetRawData(cva, &old0, &oldstep);
3001
      uchar *new0;  // pointer to first element in new mat
3002
      ERRWRAP(new0 = cvPtrND(cva, dd.i));
3003
3004
      sub->a->step = oldstep * dd.step[0];
3005
      sub->data = what_data(o);
3006
      Py_INCREF(sub->data);
3007
      sub->offset = new0 - old0;
3008
      return (PyObject*)sub;
3009
    } else {
3010
      cvmatnd_t *sub = PyObject_NEW(cvmatnd_t, &cvmatnd_Type);
3011
      sub->a = cvCreateMatNDHeader(dd.count, dd.length, cvGetElemType(cva));
3012
      uchar *old0;  // pointer to first element in old mat
3013
      cvGetRawData(cva, &old0);
3014
      uchar *new0;  // pointer to first element in new mat
3015
      ERRWRAP(new0 = cvPtrND(cva, dd.i));
3016
3017
      for (int d = 0; d < dd.count; d++) {
3018
        int stp = dd.step[d];
3019
        sub->a->dim[d].step = ((CvMatND*)cva)->dim[d].step * ((stp == 0) ? 1 : stp);
3020
        sub->a->dim[d].size = dd.length[d];
3021
      }
3022
      sub->data = what_data(o);
3023
      Py_INCREF(sub->data);
3024
      sub->offset = new0 - old0;
3025
      return (PyObject*)sub;
3026
    }
3027
  }
3028
}
3029
3030
static int cvarr_SetItem(PyObject *o, PyObject *key, PyObject *v)
3031
{
3032
  dims dd;
3033
3034
  CvArr *cva;
3035
  if (!convert_to_CvArr(o, &cva, "src"))
3036
    return -1;
3037
3038
  if (!convert_to_dims(key, &dd, cva, "key")) {
3039
    return -1;
3040
  }
3041
3042
  if (cvGetDims(cva) != dd.count) {
3043
    PyErr_SetString(PyExc_TypeError, "key length does not match array dimension");
3044
    return -1;
3045
  }
3046
3047
  CvScalar s;
3048
  if (PySequence_Check(v)) {
3049
    PyObject *fi = PySequence_Fast(v, "v");
3050
    if (fi == NULL)
3051
      return -1;
3052
    if (PySequence_Fast_GET_SIZE(fi) != CV_MAT_CN(cvGetElemType(cva))) {
3053
      PyErr_SetString(PyExc_TypeError, "sequence size must be same as channel count");
3054
      return -1;
3055
    }
3056
    for (Py_ssize_t i = 0; i < PySequence_Fast_GET_SIZE(fi); i++)
3057
      s.val[i] = PyFloat_AsDouble(PySequence_Fast_GET_ITEM(fi, i));
3058
    Py_DECREF(fi);
3059
  } else {
3060
    if (1 != CV_MAT_CN(cvGetElemType(cva))) {
3061
      PyErr_SetString(PyExc_TypeError, "scalar supplied but channel count does not equal 1");
3062
      return -1;
3063
    }
3064
    s.val[0] = PyFloat_AsDouble(v);
3065
  }
3066
  switch (dd.count) {
3067
  case 1:
3068
    cvSet1D(cva, dd.i[0], s);
3069
    break;
3070
  case 2:
3071
    cvSet2D(cva, dd.i[0], dd.i[1], s);
3072
    break;
3073
  case 3:
3074
    cvSet3D(cva, dd.i[0], dd.i[1], dd.i[2], s);
3075
    break;
3076
  default:
3077
    cvSetND(cva, dd.i, s);
3078
    // XXX - OpenCV bug? - seems as if an error in cvSetND does not set error status?
3079
    break;
3080
  }
3081
  if (cvGetErrStatus() != 0) {
3082
    translate_error_to_exception();
3083
    return -1;
3084
  }
3085
3086
  return 0;
3087
}
3088
3089
3090
static PyObject *pycvSetData(PyObject *self, PyObject *args)
3091
{
3092
  PyObject *o, *s;
3093
  int step = CV_AUTO_STEP;
3094
3095
  if (!PyArg_ParseTuple(args, "OO|i", &o, &s, &step))
3096
    return NULL;
3097
  if (is_iplimage(o)) {
3098
    iplimage_t *ipl = (iplimage_t*)o;
3099
    ipl->a->widthStep = step;
3100
    Py_DECREF(ipl->data);
3101
    ipl->data = s;
3102
    Py_INCREF(ipl->data);
3103
  } else if (is_cvmat(o)) {
3104
    cvmat_t *m = (cvmat_t*)o;
3105
    m->a->step = step;
3106
    Py_DECREF(m->data);
3107
    m->data = s;
3108
    Py_INCREF(m->data);
3109
  } else if (is_cvmatnd(o)) {
3110
    cvmatnd_t *m = (cvmatnd_t*)o;
3111
    Py_DECREF(m->data);
3112
    m->data = s;
3113
    Py_INCREF(m->data);
3114
  } else {
3115
    PyErr_SetString(PyExc_TypeError, "SetData argument must be either IplImage, CvMat or CvMatND");
3116
    return NULL;
3117
  }
3118
3119
  Py_RETURN_NONE;
3120
}
3121
3122
static PyObject *what_data(PyObject *o)
3123
{
3124
  if (is_iplimage(o)) {
3125
    iplimage_t *ipl = (iplimage_t*)o;
3126
    return ipl->data;
3127
  } else if (is_cvmat(o)) {
3128
    cvmat_t *m = (cvmat_t*)o;
3129
    return m->data;
3130
  } else if (is_cvmatnd(o)) {
3131
    cvmatnd_t *m = (cvmatnd_t*)o;
3132
    return m->data;
3133
  } else {
3134
    assert(0);
3135
    return NULL;
3136
  }
3137
}
3138
3139
static PyObject *pycvCreateData(PyObject *self, PyObject *args)
3140
{
3141
  PyObject *o;
3142
3143
  if (!PyArg_ParseTuple(args, "O", &o))
3144
    return NULL;
3145
3146
  CvArr *a;
3147
  if (!convert_to_CvArr(o, &a, "arr"))
3148
    return NULL;
3149
  ERRWRAP(cvCreateData(a));
3150
3151
  Py_DECREF(what_data(o));
3152
  if (is_iplimage(o)) {
3153
    iplimage_t *ipl = (iplimage_t*)o;
3154
    pythonize_IplImage(ipl);
3155
  } else if (is_cvmat(o)) {
3156
    cvmat_t *m = (cvmat_t*)o;
3157
    pythonize_CvMat(m);
3158
  } else if (is_cvmatnd(o)) {
3159
    cvmatnd_t *m = (cvmatnd_t*)o;
3160
    pythonize_CvMatND(m);
3161
  } else {
3162
    PyErr_SetString(PyExc_TypeError, "CreateData argument must be either IplImage, CvMat or CvMatND");
3163
    return NULL;
3164
  }
3165
3166
  Py_RETURN_NONE;
3167
}
3168
3169
static PyObject *pycvGetDims(PyObject *self, PyObject *args)
3170
{
3171
  PyObject *o;
3172
3173
  if (!PyArg_ParseTuple(args, "O", &o))
3174
    return NULL;
3175
  CvArr *cva;
3176
  if (!convert_to_CvArr(o, &cva, "src"))
3177
    return NULL;
3178
3179
  int i, nd;
3180
  ERRWRAP(nd = cvGetDims(cva));
3181
  PyObject *r = PyTuple_New(nd);
3182
  for (i = 0; i < nd; i++)
3183
    PyTuple_SetItem(r, i, PyInt_FromLong(cvGetDimSize(cva, i)));
3184
  return r;
3185
}
3186
3187
static PyObject *pycvGetImage(PyObject *self, PyObject *args)
3188
{
3189
  PyObject *o, *r;
3190
3191
  if (!PyArg_ParseTuple(args, "O", &o))
3192
    return NULL;
3193
  if (is_iplimage(o)) {
3194
    r = o;
3195
    Py_INCREF(o);
3196
  } else {
3197
    IplImage *ipl = cvCreateImageHeader(cvSize(100,100), 8, 1); // these args do not matter, because overwritten
3198
    CvArr *cva;
3199
    if (!convert_to_CvArr(o, &cva, "src"))
3200
      return NULL;
3201
    ERRWRAP(cvGetImage(cva, ipl));
3202
3203
    iplimage_t *oipl = PyObject_NEW(iplimage_t, &iplimage_Type);
3204
    oipl->a = ipl;
3205
    oipl->data = what_data(o);
3206
    Py_INCREF(oipl->data);
3207
    oipl->offset = 0;
3208
3209
    r = (PyObject*)oipl;
3210
  }
3211
  return r;
3212
}
3213
3214
static PyObject *pycvGetMat(PyObject *self, PyObject *args, PyObject *kw)
3215
{
3216
  const char *keywords[] = { "arr", "allowND", NULL };
3217
  PyObject *o, *r;
3218
  int allowND = 0;
3219
3220
  if (!PyArg_ParseTupleAndKeywords(args, kw, "O|i", (char**)keywords, &o, &allowND))
3221
    return NULL;
3222
  if (is_cvmat(o)) {
3223
    r = o;
3224
    Py_INCREF(o);
3225
  } else {
3226
    CvMat *m = cvCreateMatHeader(100,100, 1); // these args do not matter, because overwritten
3227
    CvArr *cva;
3228
    if (!convert_to_CvArr(o, &cva, "src"))
3229
      return NULL;
3230
    ERRWRAP(cvGetMat(cva, m, NULL, allowND));
3231
3232
    cvmat_t *om = PyObject_NEW(cvmat_t, &cvmat_Type);
3233
    om->a = m;
3234
    om->data = what_data(o);
3235
    Py_INCREF(om->data);
3236
    om->offset = 0;
3237
3238
    r = (PyObject*)om;
3239
  }
3240
  return r;
3241
}
3242
3243
static PyObject *pycvReshape(PyObject *self, PyObject *args)
3244
{
3245
  PyObject *o;
3246
  int new_cn;
3247
  int new_rows = 0;
3248
3249
  if (!PyArg_ParseTuple(args, "Oi|i", &o, &new_cn, &new_rows))
3250
    return NULL;
3251
3252
  CvMat *m = cvCreateMatHeader(100,100, 1); // these args do not matter, because overwritten
3253
  CvArr *cva;
3254
  if (!convert_to_CvArr(o, &cva, "src"))
3255
    return NULL;
3256
  ERRWRAP(cvReshape(cva, m, new_cn, new_rows));
3257
3258
  cvmat_t *om = PyObject_NEW(cvmat_t, &cvmat_Type);
3259
  om->a = m;
3260
  om->data = what_data(o);
3261
  Py_INCREF(om->data);
3262
  om->offset = 0;
3263
3264
  return (PyObject*)om;
3265
}
3266
3267
static PyObject *pycvReshapeMatND(PyObject *self, PyObject *args)
3268
{
3269
  PyObject *o;
3270
  int new_cn = 0;
3271
  PyObject *new_dims = NULL;
3272
3273
  if (!PyArg_ParseTuple(args, "OiO", &o, &new_cn, &new_dims))
3274
    return NULL;
3275
3276
  CvMatND *cva;
3277
  if (!convert_to_CvMatND(o, &cva, "src"))
3278
    return NULL;
3279
  ints dims;
3280
  if (new_dims != NULL) {
3281
    if (!convert_to_ints(new_dims, &dims, "new_dims"))
3282
      return NULL;
3283
  }
3284
3285
  if (new_cn == 0)
3286
    new_cn = CV_MAT_CN(cvGetElemType(cva));
3287
3288
  int i;
3289
  int count = CV_MAT_CN(cvGetElemType(cva));
3290
  for (i = 0; i < cva->dims; i++)
3291
    count *= cva->dim[i].size;
3292
3293
  int newcount = new_cn;
3294
  for (i = 0; i < dims.count; i++)
3295
    newcount *= dims.i[i];
3296
3297
  if (count != newcount) {
3298
    PyErr_SetString(PyExc_TypeError, "Total number of elements must be unchanged");
3299
    return NULL;
3300
  }
3301
3302
  CvMatND *pn = cvCreateMatNDHeader(dims.count, dims.i, CV_MAKETYPE(CV_MAT_TYPE(cva->type), new_cn));
3303
  return shareDataND(o, cva, pn);
3304
}
3305
3306
static void OnMouse(int event, int x, int y, int flags, void* param)
3307
{
3308
  PyGILState_STATE gstate;
3309
  gstate = PyGILState_Ensure();
3310
3311
  PyObject *o = (PyObject*)param;
3312
  PyObject *args = Py_BuildValue("iiiiO", event, x, y, flags, PyTuple_GetItem(o, 1));
3313
3314
  PyObject *r = PyObject_Call(PyTuple_GetItem(o, 0), args, NULL);
3315
  if (r == NULL)
3316
    PyErr_Print();
3317
  else
3318
    Py_DECREF(r);
3319
  Py_DECREF(args);
3320
  PyGILState_Release(gstate);
3321
}
3322
3323
static PyObject *pycvSetMouseCallback(PyObject *self, PyObject *args, PyObject *kw)
3324
{
3325
  const char *keywords[] = { "window_name", "on_mouse", "param", NULL };
3326
  char* name;
3327
  PyObject *on_mouse;
3328
  PyObject *param = NULL;
3329
3330
  if (!PyArg_ParseTupleAndKeywords(args, kw, "sO|O", (char**)keywords, &name, &on_mouse, &param))
3331
    return NULL;
3332
  if (!PyCallable_Check(on_mouse)) {
3333
    PyErr_SetString(PyExc_TypeError, "on_mouse must be callable");
3334
    return NULL;
3335
  }
3336
  if (param == NULL) {
3337
    param = Py_None;
3338
  }
3339
  ERRWRAP(cvSetMouseCallback(name, OnMouse, Py_BuildValue("OO", on_mouse, param)));
3340
  Py_RETURN_NONE;
3341
}
3342
3343
void OnChange(int pos, void *param)
3344
{
3345
  PyGILState_STATE gstate;
3346
  gstate = PyGILState_Ensure();
3347
3348
  PyObject *o = (PyObject*)param;
3349
  PyObject *args = Py_BuildValue("(i)", pos);
3350
  PyObject *r = PyObject_Call(PyTuple_GetItem(o, 0), args, NULL);
3351
  if (r == NULL)
3352
    PyErr_Print();
3353
  Py_DECREF(args);
3354
  PyGILState_Release(gstate);
3355
}
3356
3357
static PyObject *pycvCreateTrackbar(PyObject *self, PyObject *args)
3358
{
3359
  PyObject *on_change;
3360
  char* trackbar_name;
3361
  char* window_name;
3362
  int *value = new int;
3363
  int count;
3364
3365
  if (!PyArg_ParseTuple(args, "ssiiO", &trackbar_name, &window_name, value, &count, &on_change))
3366
    return NULL;
3367
  if (!PyCallable_Check(on_change)) {
3368
    PyErr_SetString(PyExc_TypeError, "on_change must be callable");
3369
    return NULL;
3370
  }
3371
  ERRWRAP(cvCreateTrackbar2(trackbar_name, window_name, value, count, OnChange, Py_BuildValue("OO", on_change, Py_None)));
3372
  Py_RETURN_NONE;
3373
}
3374
3375
static PyObject *pycvFindContours(PyObject *self, PyObject *args, PyObject *kw)
3376
{
3377
  CvArr* image;
3378
  PyObject *pyobj_image = NULL;
3379
  CvMemStorage* storage;
3380
  PyObject *pyobj_storage = NULL;
3381
  CvSeq* first_contour;
3382
  int header_size = sizeof(CvContour);
3383
  int mode = CV_RETR_LIST;
3384
  int method = CV_CHAIN_APPROX_SIMPLE;
3385
  CvPoint offset = cvPoint(0,0);
3386
  PyObject *pyobj_offset = NULL;
3387
3388
  const char *keywords[] = { "image", "storage", "mode", "method", "offset", NULL };
3389
  if (!PyArg_ParseTupleAndKeywords(args, kw, "OO|iiO", (char**)keywords, &pyobj_image, &pyobj_storage, &mode, &method, &pyobj_offset))
3390
    return NULL;
3391
  if (!convert_to_CvArr(pyobj_image, &image, "image")) return NULL;
3392
  if (!convert_to_CvMemStorage(pyobj_storage, &storage, "storage")) return NULL;
3393
  if ((pyobj_offset != NULL) && !convert_to_CvPoint(pyobj_offset, &offset, "offset")) return NULL;
3394
  ERRWRAP(cvFindContours(image, storage, &first_contour, header_size, mode, method, offset));
3395
  cvseq_t *ps = PyObject_NEW(cvseq_t, &cvseq_Type);
3396
  ps->a = first_contour;
3397
  ps->container = PyTuple_GetItem(args, 1); // storage
3398
  Py_INCREF(ps->container);
3399
  return (PyObject*)ps;
3400
}
3401
3402
static PyObject *pycvApproxPoly(PyObject *self, PyObject *args, PyObject *kw)
3403
{
3404
  cvarrseq src_seq;
3405
  PyObject *pyobj_src_seq = NULL;
3406
  int header_size = sizeof(CvContour);
3407
  CvMemStorage* storage;
3408
  PyObject *pyobj_storage = NULL;
3409
  int method;
3410
  double parameter = 0;
3411
  int parameter2 = 0;
3412
3413
  const char *keywords[] = { "src_seq", "storage", "method", "parameter", "parameter2", NULL };
3414
  if (!PyArg_ParseTupleAndKeywords(args, kw, "OOi|di", (char**)keywords, &pyobj_src_seq, &pyobj_storage, &method, &parameter, &parameter2))
3415
    return NULL;
3416
  if (!convert_to_cvarrseq(pyobj_src_seq, &src_seq, "src_seq")) return NULL;
3417
  if (!convert_to_CvMemStorage(pyobj_storage, &storage, "storage")) return NULL;
3418
  CvSeq* r;
3419
  ERRWRAP(r = cvApproxPoly(src_seq.mat, header_size, storage, method, parameter, parameter2));
3420
  return FROM_CvSeqPTR(r);
3421
}
3422
3423
static float distance_function_glue( const float* a, const float* b, void* user_param )
3424
{
3425
  PyObject *o = (PyObject*)user_param;
3426
  PyObject *args = Py_BuildValue("(ff)(ff)O", a[0], a[1], b[0], b[1], PyTuple_GetItem(o, 1));
3427
  PyObject *r = PyObject_Call(PyTuple_GetItem(o, 0), args, NULL);
3428
  Py_DECREF(args);
3429
  return (float)PyFloat_AsDouble(r);
3430
}
3431
3432
static PyObject *pycvCalcEMD2(PyObject *self, PyObject *args, PyObject *kw)
3433
{
3434
  const char *keywords[] = { "signature1", "signature2", "distance_type", "distance_func", "cost_matrix", "flow", "lower_bound", "userdata", NULL };
3435
  CvArr* signature1;
3436
  PyObject *pyobj_signature1;
3437
  CvArr* signature2;
3438
  PyObject *pyobj_signature2;
3439
  int distance_type;
3440
  PyObject *distance_func = NULL;
3441
  CvArr* cost_matrix=NULL;
3442
  PyObject *pyobj_cost_matrix = NULL;
3443
  CvArr* flow=NULL;
3444
  PyObject *pyobj_flow = NULL;
3445
  float lower_bound = 0.0;
3446
  PyObject *userdata = NULL;
3447
3448
  if (!PyArg_ParseTupleAndKeywords(args, kw, "OOi|OOOfO", (char**)keywords,
3449
                                   &pyobj_signature1,
3450
                                   &pyobj_signature2,
3451
                                   &distance_type,
3452
                                   &distance_func,
3453
                                   &pyobj_cost_matrix,
3454
                                   &pyobj_flow,
3455
                                   &lower_bound,
3456
                                   &userdata))
3457
    return NULL;
3458
  if (!convert_to_CvArr(pyobj_signature1, &signature1, "signature1")) return NULL;
3459
  if (!convert_to_CvArr(pyobj_signature2, &signature2, "signature2")) return NULL;
3460
  if (pyobj_cost_matrix && !convert_to_CvArr(pyobj_cost_matrix, &cost_matrix, "cost_matrix")) return NULL;
3461
  if (pyobj_flow && !convert_to_CvArr(pyobj_flow, &flow, "flow")) return NULL;
3462
3463
  if (distance_func == NULL) {
3464
    distance_func = Py_None;
3465
  }
3466
  if (userdata == NULL) {
3467
    userdata = Py_None;
3468
  }
3469
3470
  PyObject *ud = Py_BuildValue("OO", distance_func, userdata);
3471
  float r;
3472
  ERRWRAP(r = cvCalcEMD2(signature1, signature2, distance_type, distance_function_glue, cost_matrix, flow, &lower_bound, (void*)ud));
3473
  Py_DECREF(ud);
3474
3475
  return PyFloat_FromDouble(r);
3476
}
3477
3478
static PyObject *pycvSubdiv2DLocate(PyObject *self, PyObject *args)
3479
{
3480
  PyObject *pyobj_subdiv;
3481
  PyObject *pyobj_pt;
3482
  CvSubdiv2D *subdiv;
3483
  CvPoint2D32f pt;
3484
  CvSubdiv2DEdge edge;
3485
  CvSubdiv2DPoint* vertex;
3486
3487
  if (!PyArg_ParseTuple(args, "OO", &pyobj_subdiv, &pyobj_pt))
3488
    return NULL;
3489
  if (!convert_to_CvSubdiv2DPTR(pyobj_subdiv, &subdiv, "subdiv"))
3490
    return NULL;
3491
  if (!convert_to_CvPoint2D32f(pyobj_pt, &pt, "pt"))
3492
    return NULL;
3493
3494
  CvSubdiv2DPointLocation loc = cvSubdiv2DLocate(subdiv, pt, &edge, &vertex);
3495
  PyObject *r;
3496
  switch (loc) {
3497
  case CV_PTLOC_INSIDE:
3498
  case CV_PTLOC_ON_EDGE:
3499
    r = FROM_CvSubdiv2DEdge(edge);
3500
    break;
3501
  case CV_PTLOC_VERTEX:
3502
    r = FROM_CvSubdiv2DPointPTR(vertex);
3503
    break;
3504
  case CV_PTLOC_OUTSIDE_RECT:
3505
    r = Py_None;
3506
    Py_INCREF(Py_None);
3507
    break;
3508
  default:
3509
    return (PyObject*)failmsg("Unexpected loc from cvSubdiv2DLocate");
3510
  }
3511
  return Py_BuildValue("iO", (int)loc, r);
3512
}
3513
3514
static PyObject *pycvCalcOpticalFlowPyrLK(PyObject *self, PyObject *args)
3515
{
3516
  CvArr* prev;
3517
  PyObject *pyobj_prev = NULL;
3518
  CvArr* curr;
3519
  PyObject *pyobj_curr = NULL;
3520
  CvArr* prev_pyr;
3521
  PyObject *pyobj_prev_pyr = NULL;
3522
  CvArr* curr_pyr;
3523
  PyObject *pyobj_curr_pyr = NULL;
3524
  CvPoint2D32f* prev_features;
3525
  PyObject *pyobj_prev_features = NULL;
3526
  PyObject *pyobj_curr_features = NULL;
3527
  CvPoint2D32f* curr_features;
3528
  CvSize win_size;
3529
  int level;
3530
  CvTermCriteria criteria;
3531
  int flags;
3532
3533
  if (!PyArg_ParseTuple(args, "OOOOO(ii)i(iif)i|O",
3534
    &pyobj_prev, &pyobj_curr, &pyobj_prev_pyr, &pyobj_curr_pyr,
3535
    &pyobj_prev_features,
3536
    &win_size.width, &win_size.height, &level,
3537
    &criteria.type, &criteria.max_iter, &criteria.epsilon,
3538
    &flags,
3539
    &pyobj_curr_features))
3540
    return NULL;
3541
  if (!convert_to_CvArr(pyobj_prev, &prev, "prev")) return NULL;
3542
  if (!convert_to_CvArr(pyobj_curr, &curr, "curr")) return NULL;
3543
  if (!convert_to_CvArr(pyobj_prev_pyr, &prev_pyr, "prev_pyr")) return NULL;
3544
  if (!convert_to_CvArr(pyobj_curr_pyr, &curr_pyr, "curr_pyr")) return NULL;
3545
  if (!convert_to_CvPoint2D32fPTR(pyobj_prev_features, &prev_features, "prev_features")) return NULL;
3546
  int count = (int)PySequence_Length(pyobj_prev_features);
3547
  if (flags & CV_LKFLOW_INITIAL_GUESSES) {
3548
    failmsg("flag CV_LKFLOW_INITIAL_GUESSES is determined automatically from function arguments - it is not required");
3549
    return NULL;
3550
  }
3551
  if (!pyobj_curr_features) {
3552
    curr_features = new CvPoint2D32f[count];
3553
  } else {
3554
    if (PySequence_Length(pyobj_curr_features) != count) {
3555
      failmsg("curr_features must have same length as prev_features");
3556
      return NULL;
3557
    }
3558
    if (!convert_to_CvPoint2D32fPTR(pyobj_curr_features, &curr_features, "curr_features")) return NULL;
3559
    flags |= CV_LKFLOW_INITIAL_GUESSES;
3560
  }
3561
  float *track_error = new float[count];
3562
  char* status = new char[count];
3563
  ERRWRAP(cvCalcOpticalFlowPyrLK(prev, curr, prev_pyr, curr_pyr, prev_features, curr_features, count, win_size, level, status, track_error, criteria, flags));
3564
3565
  cvpoint2d32f_count r0;
3566
  r0.points = curr_features;
3567
  r0.count = count;
3568
3569
  chars r1;
3570
  r1.f = status;
3571
  r1.count = count;
3572
3573
  floats r2;
3574
  r2.f = track_error;
3575
  r2.count = count;
3576
3577
  return Py_BuildValue("NNN", FROM_cvpoint2d32f_count(r0), FROM_chars(r1), FROM_floats(r2));
3578
}
3579
3580
// pt1,pt2 are input and output arguments here
3581
3582
static PyObject *pycvClipLine(PyObject *self, PyObject *args)
3583
{
3584
  CvSize img_size;
3585
  PyObject *pyobj_img_size = NULL;
3586
  CvPoint pt1;
3587
  PyObject *pyobj_pt1 = NULL;
3588
  CvPoint pt2;
3589
  PyObject *pyobj_pt2 = NULL;
3590
3591
  if (!PyArg_ParseTuple(args, "OOO", &pyobj_img_size, &pyobj_pt1, &pyobj_pt2))
3592
    return NULL;
3593
  if (!convert_to_CvSize(pyobj_img_size, &img_size, "img_size")) return NULL;
3594
  if (!convert_to_CvPoint(pyobj_pt1, &pt1, "pt1")) return NULL;
3595
  if (!convert_to_CvPoint(pyobj_pt2, &pt2, "pt2")) return NULL;
3596
  int r;
3597
  ERRWRAP(r = cvClipLine(img_size, &pt1, &pt2));
3598
  if (r == 0) {
3599
    Py_RETURN_NONE;
3600
  } else {
3601
    return Py_BuildValue("NN", FROM_CvPoint(pt1), FROM_CvPoint(pt2));
3602
  }
3603
}
3604
3605
static PyObject *temp_test(PyObject *self, PyObject *args)
3606
{
3607
#if 0
3608
  CvArr *im = cvLoadImage("../samples/c/lena.jpg", 0);
3609
  printf("im=%p\n", im);
3610
  CvMat *m = cvEncodeImage(".jpeg", im);
3611
#endif
3612
#if 0
3613
  CvArr *im = cvLoadImage("lena.jpg", 0);
3614
  float r0[] = { 0, 255 };
3615
  float *ranges[] = { r0 };
3616
  int hist_size[] = { 256 };
3617
  CvHistogram *hist = cvCreateHist(1, hist_size, CV_HIST_ARRAY, ranges, 1);
3618
  cvCalcHist(im, hist, 0, 0);
3619
#endif
3620
3621
#if 0
3622
  CvMat* mat = cvCreateMat( 3, 3, CV_32F );
3623
  CvMat row_header, *row;
3624
  row = cvReshape( mat, &row_header, 0, 1 );
3625
  printf("%d,%d\n", row_header.rows, row_header.cols);
3626
  printf("ge %08x\n", cvGetElemType(mat));
3627
#endif
3628
3629
#if 0
3630
  CvMat *m = cvCreateMat(1, 10, CV_8UC1);
3631
  printf("CvMat stride ===> %d\n", m->step);
3632
#endif
3633
3634
#if 0
3635
  CvPoint2D32f src[3] = { { 0,0 }, { 1,0 }, { 0,1 } };
3636
  CvPoint2D32f dst[3] = { { 0,0 }, { 17,0 }, { 0,17 } };
3637
3638
  CvMat* mapping = cvCreateMat(2, 3, CV_32FC1);
3639
  cvGetAffineTransform(src, dst, mapping);
3640
  printf("===> %f\n", cvGetReal2D(mapping, 0, 0));
3641
#endif
3642
3643
#if 0
3644
  CvArr *im = cvLoadImage("checker77.png");
3645
  CvPoint2D32f corners[49];
3646
  int count;
3647
  cvFindChessboardCorners(im, cvSize(7,7), corners, &count, 0);
3648
  printf("count=%d\n", count);
3649
#endif
3650
3651
#if 0
3652
  CvMat *src = cvCreateMat(512, 512, CV_8UC3);
3653
  CvMat *dst = cvCreateMat(512, 512, CV_8UC3);
3654
  cvPyrMeanShiftFiltering(src, dst, 5, 5);
3655
  return FROM_CvMat(src);
3656
#endif
3657
3658
  return PyFloat_FromDouble(0.0);
3659
}
3660
3661
static PyObject *pycvFindChessboardCorners(PyObject *self, PyObject *args, PyObject *kw)
3662
{
3663
  CvArr* image;
3664
  PyObject *pyobj_image = NULL;
3665
  CvSize pattern_size;
3666
  PyObject *pyobj_pattern_size = NULL;
3667
  cvpoint2d32f_count corners;
3668
  int flags = CV_CALIB_CB_ADAPTIVE_THRESH;
3669
3670
  const char *keywords[] = { "image", "pattern_size", "flags", NULL };
3671
  if (!PyArg_ParseTupleAndKeywords(args, kw, "OO|i", (char**)keywords, &pyobj_image, &pyobj_pattern_size, &flags))
3672
    return NULL;
3673
  if (!convert_to_CvArr(pyobj_image, &image, "image")) return NULL;
3674
  if (!convert_to_CvSize(pyobj_pattern_size, &pattern_size, "pattern_size")) return NULL;
3675
  int r;
3676
  corners.points = new CvPoint2D32f[pattern_size.width * pattern_size.height];
3677
  ERRWRAP(r = cvFindChessboardCorners(image, pattern_size, corners.points,&corners.count, flags));
3678
  return Py_BuildValue("NN", FROM_int(r), FROM_cvpoint2d32f_count(corners));
3679
}
3680
3681
// For functions GetSubRect, GetRow, GetCol.
3682
// recipient has a view into donor's data, and needs to share it.
3683
// make recipient use the donor's data, compute the offset,
3684
// and manage reference counts.
3685
3686
static void preShareData(CvArr *donor, CvMat **recipient)
3687
{
3688
  *recipient = cvCreateMatHeader(4, 4, cvGetElemType(donor));
3689
}
3690
3691
static PyObject *shareData(PyObject *donor, CvArr *pdonor, CvMat *precipient)
3692
{
3693
  PyObject *recipient = (PyObject*)PyObject_NEW(cvmat_t, &cvmat_Type);
3694
  ((cvmat_t*)recipient)->a = precipient;
3695
  ((cvmat_t*)recipient)->offset = cvPtr1D(precipient, 0) - cvPtr1D(pdonor, 0);
3696
3697
  PyObject *arr_data;
3698
  if (is_cvmat(donor)) {
3699
    arr_data = ((cvmat_t*)donor)->data;
3700
    ((cvmat_t*)recipient)->offset += ((cvmat_t*)donor)->offset;
3701
  } else if (is_iplimage(donor)) {
3702
    arr_data = ((iplimage_t*)donor)->data;
3703
    ((cvmat_t*)recipient)->offset += ((iplimage_t*)donor)->offset;
3704
  } else {
3705
    return (PyObject*)failmsg("Argument 'mat' must be either IplImage or CvMat");
3706
  }
3707
  ((cvmat_t*)recipient)->data = arr_data;
3708
  Py_INCREF(arr_data);
3709
  return recipient;
3710
}
3711
3712
static PyObject *shareDataND(PyObject *donor, CvMatND *pdonor, CvMatND *precipient)
3713
{
3714
  PyObject *recipient = (PyObject*)PyObject_NEW(cvmatnd_t, &cvmatnd_Type);
3715
  ((cvmatnd_t*)recipient)->a = precipient;
3716
  ((cvmatnd_t*)recipient)->offset = 0;
3717
3718
  PyObject *arr_data;
3719
  arr_data = ((cvmatnd_t*)donor)->data;
3720
  ((cvmatnd_t*)recipient)->data = arr_data;
3721
  Py_INCREF(arr_data);
3722
  return recipient;
3723
}
3724
3725
static PyObject *pycvGetHuMoments(PyObject *self, PyObject *args)
3726
{
3727
  CvMoments* moments;
3728
  PyObject *pyobj_moments = NULL;
3729
3730
  if (!PyArg_ParseTuple(args, "O", &pyobj_moments))
3731
    return NULL;
3732
  if (!convert_to_CvMomentsPTR(pyobj_moments, &moments, "moments")) return NULL;
3733
  CvHuMoments r;
3734
  ERRWRAP(cvGetHuMoments(moments, &r));
3735
  return Py_BuildValue("ddddddd", r.hu1, r.hu2, r.hu3, r.hu4, r.hu5, r.hu6, r.hu7);
3736
}
3737
3738
static PyObject *pycvFitLine(PyObject *self, PyObject *args)
3739
{
3740
  cvarrseq points;
3741
  PyObject *pyobj_points = NULL;
3742
  int dist_type;
3743
  float param;
3744
  float reps;
3745
  float aeps;
3746
  float r[6];
3747
3748
  if (!PyArg_ParseTuple(args, "Oifff", &pyobj_points, &dist_type, &param, &reps, &aeps))
3749
    return NULL;
3750
  if (!convert_to_cvarrseq(pyobj_points, &points, "points")) return NULL;
3751
  ERRWRAP(cvFitLine(points.mat, dist_type, param, reps, aeps, r));
3752
  int dimension;
3753
  if (strcmp("opencv-matrix", cvTypeOf(points.mat)->type_name) == 0)
3754
    dimension = CV_MAT_CN(cvGetElemType(points.mat));
3755
  else {
3756
    // sequence case... don't think there is a sequence of 3d points,
3757
    // so assume 2D
3758
    dimension = 2;
3759
  }
3760
  if (dimension == 2)
3761
    return Py_BuildValue("dddd", r[0], r[1], r[2], r[3]);
3762
  else
3763
    return Py_BuildValue("dddddd", r[0], r[1], r[2], r[3], r[4], r[5]);
3764
}
3765
3766
static PyObject *pycvGetMinMaxHistValue(PyObject *self, PyObject *args)
3767
{
3768
  CvHistogram* hist;
3769
  PyObject *pyobj_hist = NULL;
3770
  float min_val;
3771
  float max_val;
3772
  int min_loc[CV_MAX_DIM];
3773
  int max_loc[CV_MAX_DIM];
3774
3775
  if (!PyArg_ParseTuple(args, "O", &pyobj_hist))
3776
    return NULL;
3777
  if (!convert_to_CvHistogram(pyobj_hist, &hist, "hist")) return NULL;
3778
  ERRWRAP(cvGetMinMaxHistValue(hist, &min_val, &max_val, min_loc, max_loc));
3779
  int d = cvGetDims(hist->bins);
3780
  PyObject *pminloc = PyTuple_New(d), *pmaxloc = PyTuple_New(d);
3781
  for (int i = 0; i < d; i++) {
3782
    PyTuple_SetItem(pminloc, i, PyInt_FromLong(min_loc[i]));
3783
    PyTuple_SetItem(pmaxloc, i, PyInt_FromLong(max_loc[i]));
3784
  }
3785
  return Py_BuildValue("ffNN", min_val, max_val, pminloc, pmaxloc);
3786
}
3787
3788
static CvSeq* cvHOGDetectMultiScale( const CvArr* image, CvMemStorage* storage,
3789
  const CvArr* svm_classifier=NULL, CvSize win_stride=cvSize(0,0),
3790
  double hit_threshold=0, double scale=1.05,
3791
  int group_threshold=2, CvSize padding=cvSize(0,0),
3792
  CvSize win_size=cvSize(64,128), CvSize block_size=cvSize(16,16),
3793
  CvSize block_stride=cvSize(8,8), CvSize cell_size=cvSize(8,8),
3794
  int nbins=9, int gammaCorrection=1 )
3795
{
3796
    cv::HOGDescriptor hog(win_size, block_size, block_stride, cell_size, nbins, 1, -1, cv::HOGDescriptor::L2Hys, 0.2, gammaCorrection!=0);
3797
    if(win_stride.width == 0 && win_stride.height == 0)
3798
        win_stride = block_stride;
3799
    cv::Mat img = cv::cvarrToMat(image);
3800
    std::vector<cv::Rect> found;
3801
    if(svm_classifier)
3802
    {
3803
        CvMat stub, *m = cvGetMat(svm_classifier, &stub);
3804
        int sz = m->cols*m->rows;
3805
        CV_Assert(CV_IS_MAT_CONT(m->type) && (m->cols == 1 || m->rows == 1) && CV_MAT_TYPE(m->type) == CV_32FC1);
3806
        std::vector<float> w(sz);
3807
        std::copy(m->data.fl, m->data.fl + sz, w.begin());
3808
        hog.setSVMDetector(w);
3809
    }
3810
    else
3811
        hog.setSVMDetector(cv::HOGDescriptor::getDefaultPeopleDetector());
3812
    hog.detectMultiScale(img, found, hit_threshold, win_stride, padding, scale, group_threshold);
3813
    CvSeq* seq = cvCreateSeq(cv::DataType<cv::Rect>::type, sizeof(CvSeq), sizeof(cv::Rect), storage);
3814
    if(found.size())
3815
        cvSeqPushMulti(seq, &found[0], (int)found.size());
3816
    return seq;
3817
}
3818
3819
static void cvGrabCut(CvArr *image,
3820
                      CvArr *mask,
3821
                      CvRect rect,
3822
                      CvArr *bgdModel,
3823
                      CvArr *fgdModel,
3824
                      int iterCount,
3825
                      int mode)
3826
{
3827
  cv::Mat _image = cv::cvarrToMat(image);
3828
  cv::Mat _mask = cv::cvarrToMat(mask);
3829
  cv::Mat _bgdModel = cv::cvarrToMat(bgdModel);
3830
  cv::Mat _fgdModel = cv::cvarrToMat(fgdModel);
3831
  grabCut(_image, _mask, rect, _bgdModel, _fgdModel, iterCount, mode);
3832
}
3833
3834
static int zero = 0;
3835
3836
/************************************************************************/
3837
/* Custom Validators */
3838
3839
#define CVPY_VALIDATE_DrawChessboardCorners() do { \
3840
  if ((patternSize.width * patternSize.height) != corners.count) \
3841
    return (PyObject*)failmsg("Size is %dx%d, but corner list is length %d", patternSize.width, patternSize.height, corners.count); \
3842
  } while (0)
3843
3844
#define cvGetRotationMatrix2D cv2DRotationMatrix
3845
3846
/************************************************************************/
3847
/* Generated functions */
3848
3849
#define constCvMat const CvMat
3850
#define FROM_constCvMatPTR(x) FROM_CvMatPTR((CvMat*)x)
3851
3852
#define cvSnakeImage(image, points, length, a, b, g, win, criteria, calc_gradient) \
3853
  do { \
3854
    int coeff_usage; \
3855
    if ((alpha.count == 1) && (beta.count == 1) && (gamma.count == 1)) \
3856
      coeff_usage = CV_VALUE; \
3857
    else if ((length == alpha.count) && (alpha.count == beta.count) && (beta.count == gamma.count)) \
3858
      coeff_usage = CV_ARRAY; \
3859
    else \
3860
      return (PyObject*)failmsg("SnakeImage weights invalid"); \
3861
    cvSnakeImage(image, points, length, a, b, g, coeff_usage, win, criteria, calc_gradient); \
3862
  } while (0)
3863
3864
static double cppKMeans(const CvArr* _samples, int cluster_count, CvArr* _labels,
3865
           CvTermCriteria termcrit, int attempts, int flags, CvArr* _centers)
3866
{
3867
    cv::Mat data = cv::cvarrToMat(_samples), labels = cv::cvarrToMat(_labels), centers;
3868
    if( _centers )
3869
        centers = cv::cvarrToMat(_centers);
3870
    CV_Assert( labels.isContinuous() && labels.type() == CV_32S &&
3871
        (labels.cols == 1 || labels.rows == 1) &&
3872
        labels.cols + labels.rows - 1 == data.rows );
3873
    return cv::kmeans(data, cluster_count, labels, termcrit, attempts,
3874
                                    flags, _centers ? &centers : 0 );
3875
}
3876
3877
#define cvKMeans2(samples, nclusters, labels, termcrit, attempts, flags, centers) \
3878
    cppKMeans(samples, nclusters, labels, termcrit, attempts, flags, centers)
3879
3880
#include "generated0.i"
3881
3882
#if PYTHON_USE_NUMPY
3883
#include "opencv2x.h"
3884
#include "pyopencv_generated_types.h"
3885
#include "pyopencv_generated_funcs.h"
3886
#endif
3887
3888
static PyMethodDef methods[] = {
3889
3890
#if PYTHON_USE_NUMPY
3891
    {"fromarray", (PyCFunction)pycvfromarray, METH_KEYWORDS, "fromarray(array) -> cvmatnd"},
3892
#endif
3893
3894
  //{"CalcOpticalFlowFarneback", (PyCFunction)pycvCalcOpticalFlowFarneback, METH_KEYWORDS, "CalcOpticalFlowFarneback(prev, next, flow, pyr_scale=0.5, levels=3, win_size=15, iterations=3, poly_n=7, poly_sigma=1.5, flags=0) -> None"},
3895
  //{"_HOGComputeDescriptors", (PyCFunction)pycvHOGComputeDescriptors, METH_KEYWORDS, "_HOGComputeDescriptors(image, win_stride=block_stride, locations=None, padding=(0,0), win_size=(64,128), block_size=(16,16), block_stride=(8,8), cell_size=(8,8), nbins=9, gammaCorrection=true) -> list_of_descriptors"},
3896
  //{"_HOGDetect", (PyCFunction)pycvHOGDetect, METH_KEYWORDS, "_HOGDetect(image, svm_classifier, win_stride=block_stride, locations=None, padding=(0,0), win_size=(64,128), block_size=(16,16), block_stride=(8,8), cell_size=(8,8), nbins=9, gammaCorrection=true) -> list_of_points"},
3897
  //{"_HOGDetectMultiScale", (PyCFunction)pycvHOGDetectMultiScale, METH_KEYWORDS, "_HOGDetectMultiScale(image, svm_classifier, win_stride=block_stride, scale=1.05, group_threshold=2, padding=(0,0), win_size=(64,128), block_size=(16,16), block_stride=(8,8), cell_size=(8,8), nbins=9, gammaCorrection=true) -> list_of_points"},
3898
3899
  {"temp_test", temp_test, METH_VARARGS},
3900
3901
#include "generated1.i"
3902
3903
#if PYTHON_USE_NUMPY
3904
#include "pyopencv_generated_func_tab.h"
3905
#endif
3906
3907
  {NULL, NULL},
3908
};
3909
3910
/************************************************************************/
3911
/* Module init */
3912
3913
static int to_ok(PyTypeObject *to)
3914
{
3915
  to->tp_alloc = PyType_GenericAlloc;
3916
  to->tp_new = PyType_GenericNew;
3917
  to->tp_flags = Py_TPFLAGS_DEFAULT | Py_TPFLAGS_BASETYPE;
3918
  return (PyType_Ready(to) == 0);
3919
}
3920
3921
#define MKTYPE(NAME)  NAME##_specials(); if (!to_ok(&NAME##_Type)) return
3922
#define MKTYPE2(NAME) pyopencv_##NAME##_specials(); if (!to_ok(&pyopencv_##NAME##_Type)) return
3923
3924
using namespace cv;
3925
3926
extern "C"
3927
#if defined WIN32 || defined _WIN32
3928
__declspec(dllexport)
3929
#endif
3930
3931
void initcv()
3932
{
3933
#if PYTHON_USE_NUMPY
3934
    import_array();
3935
#endif
3936
    
3937
  PyObject *m, *d;
3938
3939
  cvSetErrMode(CV_ErrModeParent);
3940
3941
  MKTYPE(cvcontourtree);
3942
  MKTYPE(cvfont);
3943
  MKTYPE(cvhistogram);
3944
  MKTYPE(cvlineiterator);
3945
  MKTYPE(cvmat);
3946
  MKTYPE(cvmatnd);
3947
  MKTYPE(cvmemstorage);
3948
  MKTYPE(cvsubdiv2dedge);
3949
  MKTYPE(cvrng);
3950
  MKTYPE(cvseq);
3951
  MKTYPE(cvset);
3952
  MKTYPE(cvsubdiv2d);
3953
  MKTYPE(cvsubdiv2dpoint);
3954
  MKTYPE(iplimage);
3955
  MKTYPE(memtrack);
3956
3957
#include "generated4.i"
3958
3959
#if PYTHON_USE_NUMPY
3960
#include "pyopencv_generated_type_reg.h"
3961
#endif
3962
3963
  m = Py_InitModule(MODULESTR"", methods);
3964
  d = PyModule_GetDict(m);
3965
3966
  PyDict_SetItemString(d, "__version__", PyString_FromString("$Rev: 3057 $"));
3967
3968
  opencv_error = PyErr_NewException((char*)MODULESTR".error", NULL, NULL);
3969
  PyDict_SetItemString(d, "error", opencv_error);
3970
3971
  // Couple of warnings about strict aliasing here.  Not clear how to fix.
3972
  union {
3973
    PyObject *o;
3974
    PyTypeObject *to;
3975
  } convert;
3976
  convert.to = &iplimage_Type;
3977
  PyDict_SetItemString(d, "iplimage", convert.o);
3978
  convert.to = &cvmat_Type;
3979
  PyDict_SetItemString(d, "cvmat", convert.o);
3980
3981
#define PUBLISH(I) PyDict_SetItemString(d, #I, PyInt_FromLong(I))
3982
#define PUBLISHU(I) PyDict_SetItemString(d, #I, PyLong_FromUnsignedLong(I))
3983
#define PUBLISH2(I, value) PyDict_SetItemString(d, #I, PyLong_FromLong(value))
3984
3985
  PUBLISHU(IPL_DEPTH_8U);
3986
  PUBLISHU(IPL_DEPTH_8S);
3987
  PUBLISHU(IPL_DEPTH_16U);
3988
  PUBLISHU(IPL_DEPTH_16S);
3989
  PUBLISHU(IPL_DEPTH_32S);
3990
  PUBLISHU(IPL_DEPTH_32F);
3991
  PUBLISHU(IPL_DEPTH_64F);
3992
3993
  PUBLISH(CV_LOAD_IMAGE_COLOR);
3994
  PUBLISH(CV_LOAD_IMAGE_GRAYSCALE);
3995
  PUBLISH(CV_LOAD_IMAGE_UNCHANGED);
3996
  PUBLISH(CV_HIST_ARRAY);
3997
  PUBLISH(CV_HIST_SPARSE);
3998
  PUBLISH(CV_8U);
3999
  PUBLISH(CV_8UC1);
4000
  PUBLISH(CV_8UC2);
4001
  PUBLISH(CV_8UC3);
4002
  PUBLISH(CV_8UC4);
4003
  PUBLISH(CV_8S);
4004
  PUBLISH(CV_8SC1);
4005
  PUBLISH(CV_8SC2);
4006
  PUBLISH(CV_8SC3);
4007
  PUBLISH(CV_8SC4);
4008
  PUBLISH(CV_16U);
4009
  PUBLISH(CV_16UC1);
4010
  PUBLISH(CV_16UC2);
4011
  PUBLISH(CV_16UC3);
4012
  PUBLISH(CV_16UC4);
4013
  PUBLISH(CV_16S);
4014
  PUBLISH(CV_16SC1);
4015
  PUBLISH(CV_16SC2);
4016
  PUBLISH(CV_16SC3);
4017
  PUBLISH(CV_16SC4);
4018
  PUBLISH(CV_32S);
4019
  PUBLISH(CV_32SC1);
4020
  PUBLISH(CV_32SC2);
4021
  PUBLISH(CV_32SC3);
4022
  PUBLISH(CV_32SC4);
4023
  PUBLISH(CV_32F);
4024
  PUBLISH(CV_32FC1);
4025
  PUBLISH(CV_32FC2);
4026
  PUBLISH(CV_32FC3);
4027
  PUBLISH(CV_32FC4);
4028
  PUBLISH(CV_64F);
4029
  PUBLISH(CV_64FC1);
4030
  PUBLISH(CV_64FC2);
4031
  PUBLISH(CV_64FC3);
4032
  PUBLISH(CV_64FC4);
4033
  PUBLISH(CV_NEXT_AROUND_ORG);
4034
  PUBLISH(CV_NEXT_AROUND_DST);
4035
  PUBLISH(CV_PREV_AROUND_ORG);
4036
  PUBLISH(CV_PREV_AROUND_DST);
4037
  PUBLISH(CV_NEXT_AROUND_LEFT);
4038
  PUBLISH(CV_NEXT_AROUND_RIGHT);
4039
  PUBLISH(CV_PREV_AROUND_LEFT);
4040
  PUBLISH(CV_PREV_AROUND_RIGHT);
4041
4042
  PUBLISH(CV_WINDOW_AUTOSIZE);
4043
4044
  PUBLISH(CV_PTLOC_INSIDE);
4045
  PUBLISH(CV_PTLOC_ON_EDGE);
4046
  PUBLISH(CV_PTLOC_VERTEX);
4047
  PUBLISH(CV_PTLOC_OUTSIDE_RECT);
4048
4049
  PUBLISH(GC_BGD);
4050
  PUBLISH(GC_FGD);
4051
  PUBLISH(GC_PR_BGD);
4052
  PUBLISH(GC_PR_FGD);
4053
  PUBLISH(GC_INIT_WITH_RECT);
4054
  PUBLISH(GC_INIT_WITH_MASK);
4055
  PUBLISH(GC_EVAL);
4056
4057
#include "generated2.i"
4058
4059
#if PYTHON_USE_NUMPY
4060
#include "pyopencv_generated_const_reg.h"
4061
#endif
4062
4063
#if 0
4064
  {
4065
    int sizes[] = { 10 } ;
4066
    float ranges[] = { 0.0, 1.0 };
4067
    // CvHistogram*h = cvCreateHist(1, sizes, CV_HIST_ARRAY);
4068
    CvHistogram H;
4069
    float data[10];
4070
    CvHistogram*h = cvMakeHistHeaderForArray(1, sizes, &H, data);
4071
    printf("h->type = %08x\n", h->type);
4072
    printf("h->bins = %p\n", h->bins);
4073
    printf("h->mat = %p\n", &(h->mat));
4074
  }
4075
#endif
4076
}
4077