ViewVC Help
View File | Revision Log | Show Annotations | Download File
/cvs/CV/CV.xs
(Generate patch)

Comparing CV/CV.xs (file contents):
Revision 1.33 by root, Fri Feb 17 08:48:06 2006 UTC vs.
Revision 1.52 by root, Sat Dec 23 04:25:50 2017 UTC

9#include <magic.h> 9#include <magic.h>
10 10
11#include <jpeglib.h> 11#include <jpeglib.h>
12#include <glib.h> 12#include <glib.h>
13#include <gtk/gtk.h> 13#include <gtk/gtk.h>
14#include <gdk/gdkx.h>
14#include <gdk-pixbuf/gdk-pixbuf.h> 15#include <gdk-pixbuf/gdk-pixbuf.h>
15 16
16#include <gperl.h> 17#include <gperl.h>
17#include <gtk2perl.h> 18#include <gtk2perl.h>
18 19
20#include <assert.h>
21
22#if WEBP
23#include <webp/decode.h>
24#endif
25
26#include "perlmulticore.h"
27
19#define IW 80 /* MUST match Schnauzer.pm! */ 28#define IW 80 /* MUST match Schnauzer.pm! */
20#define IH 60 /* MUST match Schnauzer.pm! */ 29#define IH 60 /* MUST match Schnauzer.pm! */
21 30
22#define RAND (seed = (seed + 7141) * 54773 % 134456) 31#define RAND (seed = (seed + 7141) * 54773 % 134456)
23 32
24#define LINELENGTH 240 33#define LINELENGTH 240
25 34
26#define ELLIPSIS "\xe2\x80\xa6" 35#define ELLIPSIS "\xe2\x80\xa6"
36
37typedef char *octet_string;
38
39static magic_t magic_cookie[2]; /* !mime, mime */
27 40
28struct jpg_err_mgr 41struct jpg_err_mgr
29{ 42{
30 struct jpeg_error_mgr err; 43 struct jpeg_error_mgr err;
31 jmp_buf setjmp_buffer; 44 jmp_buf setjmp_buffer;
32}; 45};
33 46
34static void 47static void
35cv_error_exit (j_common_ptr cinfo) 48cv_error_exit (j_common_ptr cinfo)
36{ 49{
50 cinfo->err->output_message (cinfo);
37 longjmp (((struct jpg_err_mgr *)cinfo->err)->setjmp_buffer, 99); 51 longjmp (((struct jpg_err_mgr *)cinfo->err)->setjmp_buffer, 99);
38} 52}
39 53
40static void 54static void
41cv_error_output (j_common_ptr cinfo) 55cv_error_output (j_common_ptr cinfo)
42{ 56{
57 char msg[JMSG_LENGTH_MAX];
58
59 cinfo->err->format_message (cinfo, msg);
60
61 fprintf (stderr, "JPEG decoding error: %s\n", msg);
43 return; 62 return;
44} 63}
45 64
46static void 65static void
47rgb_to_hsv (unsigned int r, unsigned int g, unsigned int b, 66rgb_to_hsv (unsigned int r, unsigned int g, unsigned int b,
68 87
69 *h &= 255; 88 *h &= 255;
70 } 89 }
71} 90}
72 91
92struct feature {
93 float v1, v2, v3; // mean, square, cube
94 int n;
95};
96
97static void
98feature_init (struct feature *f)
99{
100 f->v1 = 0.;
101 f->v2 = 0.;
102 f->v3 = 0.;
103 f->n = 0;
104}
105
106// didn't find an algorithm to neatly do mean, variance and skew in one pass.
107// elmex ist schuld.
108static void
109feature_update_pass_1 (struct feature *f, unsigned int v)
110{
111 f->v1 += v;
112 f->n += 1;
113}
114
115static void
116feature_finish_pass_1 (struct feature *f)
117{
118 if (f->n < 1)
119 return;
120
121 f->v1 /= f->n;
122}
123
124static void
125feature_update_pass_2 (struct feature *f, unsigned int v)
126{
127 float d = v - f->v1;
128
129 f->v2 += d * d;
130 f->v3 += d * d * d;
131}
132
133static void
134feature_finish_pass_2 (struct feature *f)
135{
136 if (f->n < 1)
137 return;
138
139 f->v2 /= f->n;
140 f->v3 /= f->n;
141
142 f->v1 /= 255.;
143 f->v2 /= 255. * 255.; f->v2 = sqrtf (f->v2);
144 f->v3 /= 255. * 255. * 255.; f->v3 = powf (fabsf (f->v3), 1./3.);
145}
146
73static guint32 a85_val; 147static guint32 a85_val;
74static guint a85_cnt; 148static guint a85_cnt;
75static guchar a85_buf[LINELENGTH], *a85_ptr; 149static guchar a85_buf[LINELENGTH], *a85_ptr;
76 150
77static void 151static void
89 if (!--a85_cnt) 163 if (!--a85_cnt)
90 { 164 {
91 a85_cnt = 4; 165 a85_cnt = 4;
92 if (a85_val) 166 if (a85_val)
93 { 167 {
94 a85_ptr[4] = (a85_val % 85) + 33; a85_val /= 85; 168 a85_ptr[4] = (a85_val % 85) + 33; a85_val /= 85;
95 a85_ptr[3] = (a85_val % 85) + 33; a85_val /= 85; 169 a85_ptr[3] = (a85_val % 85) + 33; a85_val /= 85;
96 a85_ptr[2] = (a85_val % 85) + 33; a85_val /= 85; 170 a85_ptr[2] = (a85_val % 85) + 33; a85_val /= 85;
97 a85_ptr[1] = (a85_val % 85) + 33; a85_val /= 85; 171 a85_ptr[1] = (a85_val % 85) + 33; a85_val /= 85;
98 a85_ptr[0] = (a85_val ) + 33; 172 a85_ptr[0] = (a85_val ) + 33;
99 173
100 a85_ptr += 5; 174 a85_ptr += 5;
101 } 175 }
147 221
148 OUTPUT: 222 OUTPUT:
149 RETVAL 223 RETVAL
150 224
151const char * 225const char *
152magic (const char *path) 226magic (SV *path_or_data)
227 ALIAS:
228 magic = 0
229 magic_mime = 1
230 magic_buffer = 2
231 magic_buffer_mime = 3
153 CODE: 232 CODE:
154{ 233{
155 static magic_t cookie; 234 STRLEN len;
235 char *data = SvPVbyte (path_or_data, len);
156 236
157 if (!cookie) 237 if (!magic_cookie[0])
238 {
239 magic_cookie[0] = magic_open (MAGIC_SYMLINK);
240 magic_cookie[1] = magic_open (MAGIC_SYMLINK | MAGIC_MIME_TYPE);
241 magic_load (magic_cookie[0], 0);
242 magic_load (magic_cookie[1], 0);
158 { 243 }
159 cookie = magic_open (MAGIC_NONE);
160 244
161 if (cookie) 245 perlinterp_release ();
162 magic_load (cookie, 0);
163 else
164 XSRETURN_UNDEF;
165 }
166 246
167 RETVAL = magic_file (cookie, path); 247 RETVAL = ix & 2
248 ? magic_buffer (magic_cookie[ix & 1], data, len)
249 : magic_file (magic_cookie[ix & 1], data);
250
251 perlinterp_acquire ();
168} 252}
169 OUTPUT: 253 OUTPUT:
170 RETVAL 254 RETVAL
171 255
172const char *
173magic_mime (const char *path)
174 CODE:
175{
176 static magic_t cookie;
177
178 if (!cookie)
179 {
180 cookie = magic_open (MAGIC_MIME);
181
182 if (cookie)
183 magic_load (cookie, 0);
184 else
185 XSRETURN_UNDEF;
186 }
187
188 RETVAL = magic_file (cookie, path);
189}
190 OUTPUT:
191 RETVAL
192
193# missing in Gtk2 perl module 256# missing/broken in Gtk2 perl module
257
258void
259gdk_window_clear_hints (GdkWindow *window)
260 CODE:
261 gdk_window_set_geometry_hints (window, 0, 0);
194 262
195gboolean 263gboolean
196gdk_net_wm_supports (GdkAtom property) 264gdk_net_wm_supports (GdkAtom property)
197 CODE: 265 CODE:
198#if defined(GDK_WINDOWING_X11) && !defined(GDK_MULTIHEAD_SAFE) 266#if defined(GDK_WINDOWING_X11) && !defined(GDK_MULTIHEAD_SAFE)
204 RETVAL 272 RETVAL
205 273
206GdkPixbuf_noinc * 274GdkPixbuf_noinc *
207dealpha_expose (GdkPixbuf *pb) 275dealpha_expose (GdkPixbuf *pb)
208 CODE: 276 CODE:
277 perlinterp_release ();
209{ 278{
210 int w = gdk_pixbuf_get_width (pb); 279 int w = gdk_pixbuf_get_width (pb);
211 int h = gdk_pixbuf_get_height (pb); 280 int h = gdk_pixbuf_get_height (pb);
212 int bpp = gdk_pixbuf_get_n_channels (pb); 281 int bpp = gdk_pixbuf_get_n_channels (pb);
213 int x, y, i; 282 int x, y, i;
222 for (x = 0; x < w; x++) 291 for (x = 0; x < w; x++)
223 for (y = 0; y < h; y++) 292 for (y = 0; y < h; y++)
224 for (i = 0; i < 3; i++) 293 for (i = 0; i < 3; i++)
225 dst[x * 3 + y * dstr + i] = src[x * bpp + y * sstr + i]; 294 dst[x * 3 + y * dstr + i] = src[x * bpp + y * sstr + i];
226} 295}
296 perlinterp_acquire ();
227 OUTPUT: 297 OUTPUT:
228 RETVAL 298 RETVAL
229 299
230GdkPixbuf_noinc * 300GdkPixbuf_noinc *
231rotate (GdkPixbuf *pb, int angle) 301rotate (GdkPixbuf *pb, int angle)
232 CODE: 302 CODE:
303 perlinterp_release ();
304 if (angle < 0)
305 angle += 360;
233 RETVAL = gdk_pixbuf_rotate_simple (pb, angle == 0 ? GDK_PIXBUF_ROTATE_NONE 306 RETVAL = gdk_pixbuf_rotate_simple (pb, angle == 0 ? GDK_PIXBUF_ROTATE_NONE
234 : angle == 90 ? GDK_PIXBUF_ROTATE_COUNTERCLOCKWISE 307 : angle == 90 ? GDK_PIXBUF_ROTATE_COUNTERCLOCKWISE
235 : angle == 180 ? GDK_PIXBUF_ROTATE_UPSIDEDOWN 308 : angle == 180 ? GDK_PIXBUF_ROTATE_UPSIDEDOWN
236 : angle == 270 ? GDK_PIXBUF_ROTATE_CLOCKWISE 309 : angle == 270 ? GDK_PIXBUF_ROTATE_CLOCKWISE
237 : angle); 310 : angle);
311 perlinterp_acquire ();
238 OUTPUT: 312 OUTPUT:
239 RETVAL 313 RETVAL
240 314
241GdkPixbuf_noinc * 315GdkPixbuf_noinc *
242load_jpeg (SV *path, int thumbnail=0) 316decode_webp (SV *image_data, int thumbnail = 0, int iw = 0, int ih = 0)
317 CODE:
318{
319#if WEBP
320 guchar *data;
321 STRLEN data_size;
322 int alpha;
323 WebPDecoderConfig config;
324 int inw, inh;
325
326 data = SvPVbyte (image_data, data_size);
327
328 perlinterp_release ();
329
330 RETVAL = 0;
331
332 if (!WebPInitDecoderConfig (&config))
333 goto err;
334
335 config.options.use_threads = 1;
336
337 if (WebPGetFeatures (data, data_size, &config.input) != VP8_STATUS_OK)
338 goto err;
339
340 inw = config.input.width;
341 inh = config.input.height;
342
343 if (thumbnail)
344 {
345 if (inw * ih > inh * iw)
346 ih = (iw * inh + inw - 1) / inw;
347 else
348 iw = (ih * inw + inh - 1) / inh;
349
350 config.options.bypass_filtering = 1;
351 config.options.no_fancy_upsampling = 1;
352
353 config.options.use_scaling = 1;
354 config.options.scaled_width = iw;
355 config.options.scaled_height = ih;
356 }
357 else
358 {
359 iw = inw;
360 ih = inh;
361 }
362
363 alpha = config.input.has_alpha;
364
365 RETVAL = gdk_pixbuf_new (GDK_COLORSPACE_RGB, !!alpha, 8, iw, ih);
366 if (!RETVAL)
367 goto err;
368
369 config.output.colorspace = alpha ? MODE_RGBA : MODE_RGB;
370 config.output.u.RGBA.rgba = gdk_pixbuf_get_pixels (RETVAL);
371 config.output.u.RGBA.stride = gdk_pixbuf_get_rowstride (RETVAL);
372 config.output.u.RGBA.size = gdk_pixbuf_get_byte_length (RETVAL);
373 config.output.is_external_memory = 1;
374
375 if (WebPDecode (data, data_size, &config) != VP8_STATUS_OK)
376 {
377 g_object_unref (RETVAL);
378 RETVAL = 0;
379 goto err;
380 }
381
382 err:
383 perlinterp_acquire ();
384#else
385 croak ("load_webp: webp not enabled at compile time");
386#endif
387}
388 OUTPUT:
389 RETVAL
390
391GdkPixbuf_noinc *
392load_jpeg (SV *path, int thumbnail = 0, int iw = 0, int ih = 0)
243 CODE: 393 CODE:
244{ 394{
245 struct jpeg_decompress_struct cinfo; 395 struct jpeg_decompress_struct cinfo;
246 struct jpg_err_mgr jerr; 396 struct jpg_err_mgr jerr;
247 guchar *data; 397 guchar *data;
254 fp = fopen (SvPVbyte_nolen (path), "rb"); 404 fp = fopen (SvPVbyte_nolen (path), "rb");
255 405
256 if (!fp) 406 if (!fp)
257 XSRETURN_UNDEF; 407 XSRETURN_UNDEF;
258 408
409 perlinterp_release ();
410
259 cinfo.err = jpeg_std_error (&jerr.err); 411 cinfo.err = jpeg_std_error (&jerr.err);
260 412
261 jerr.err.error_exit = cv_error_exit; 413 jerr.err.error_exit = cv_error_exit;
262 jerr.err.output_message = cv_error_output; 414 jerr.err.output_message = cv_error_output;
263 415
267 jpeg_destroy_decompress (&cinfo); 419 jpeg_destroy_decompress (&cinfo);
268 420
269 if (pb) 421 if (pb)
270 g_object_unref ((gpointer)pb); 422 g_object_unref ((gpointer)pb);
271 423
424 perlinterp_acquire ();
272 XSRETURN_UNDEF; 425 XSRETURN_UNDEF;
273 } 426 }
274 427
275 jpeg_create_decompress (&cinfo); 428 jpeg_create_decompress (&cinfo);
276 429
292 { 445 {
293 cinfo.dct_method = JDCT_FASTEST; 446 cinfo.dct_method = JDCT_FASTEST;
294 cinfo.do_fancy_upsampling = FALSE; 447 cinfo.do_fancy_upsampling = FALSE;
295 448
296 while (cinfo.scale_denom < 8 449 while (cinfo.scale_denom < 8
297 && cinfo.output_width >= IW*4 450 && cinfo.output_width >= iw*4
298 && cinfo.output_height >= IH*4) 451 && cinfo.output_height >= ih*4)
299 { 452 {
300 cinfo.scale_denom <<= 1; 453 cinfo.scale_denom <<= 1;
301 jpeg_calc_output_dimensions (&cinfo); 454 jpeg_calc_output_dimensions (&cinfo);
302 } 455 }
303 } 456 }
304 457
458 if (cinfo.output_components != 3)
459 longjmp (jerr.setjmp_buffer, 3);
460
461 if (cinfo.jpeg_color_space == JCS_YCCK || cinfo.jpeg_color_space == JCS_CMYK)
462 {
463 cinfo.out_color_space = JCS_CMYK;
464 cinfo.output_components = 4;
465 }
466
305 pb = RETVAL = gdk_pixbuf_new (GDK_COLORSPACE_RGB, 0, 8, cinfo.output_width, cinfo.output_height); 467 pb = RETVAL = gdk_pixbuf_new (GDK_COLORSPACE_RGB, cinfo.output_components == 4, 8, cinfo.output_width, cinfo.output_height);
306 if (!RETVAL) 468 if (!RETVAL)
307 longjmp (jerr.setjmp_buffer, 2); 469 longjmp (jerr.setjmp_buffer, 2);
308 470
309 data = gdk_pixbuf_get_pixels (RETVAL); 471 data = gdk_pixbuf_get_pixels (RETVAL);
310 rs = gdk_pixbuf_get_rowstride (RETVAL); 472 rs = gdk_pixbuf_get_rowstride (RETVAL);
311
312 if (cinfo.output_components != 3)
313 longjmp (jerr.setjmp_buffer, 3);
314 473
315 jpeg_start_decompress (&cinfo); 474 jpeg_start_decompress (&cinfo);
316 475
317 while (cinfo.output_scanline < cinfo.output_height) 476 while (cinfo.output_scanline < cinfo.output_height)
318 { 477 {
325 rp [3] = (guchar *)rp [2] + rs; 484 rp [3] = (guchar *)rp [2] + rs;
326 485
327 jpeg_read_scanlines (&cinfo, rp, remaining < 4 ? remaining : 4); 486 jpeg_read_scanlines (&cinfo, rp, remaining < 4 ? remaining : 4);
328 } 487 }
329 488
489 if (cinfo.out_color_space == JCS_CMYK)
490 {
491 guchar *end = data + cinfo.output_height * rs;
492
493 while (data < end)
494 {
495 U32 c = data [0];
496 U32 m = data [1];
497 U32 y = data [2];
498 U32 k = data [3];
499
500 if (0)
501 if (cinfo.Adobe_transform == 2)
502 {
503 c ^= 0xff;
504 m ^= 0xff;
505 y ^= 0xff;
506 k ^= 0xff;
507 }
508
509 data [0] = (c * k + 0x80) / 0xff;
510 data [1] = (m * k + 0x80) / 0xff;
511 data [2] = (y * k + 0x80) / 0xff;
512 data [3] = 0xff;
513
514 data += 4;
515 }
516 }
517
330 jpeg_finish_decompress (&cinfo); 518 jpeg_finish_decompress (&cinfo);
331 fclose (fp); 519 fclose (fp);
332 jpeg_destroy_decompress (&cinfo); 520 jpeg_destroy_decompress (&cinfo);
521 perlinterp_acquire ();
333} 522}
334 OUTPUT: 523 OUTPUT:
335 RETVAL 524 RETVAL
336 525
337void 526void
338compare (GdkPixbuf *a, GdkPixbuf *b) 527compare (GdkPixbuf *a, GdkPixbuf *b)
339 PPCODE: 528 PPCODE:
529 perlinterp_release ();
340{ 530{
341 int w = gdk_pixbuf_get_width (a); 531 int w = gdk_pixbuf_get_width (a);
342 int h = gdk_pixbuf_get_height (a); 532 int h = gdk_pixbuf_get_height (a);
533
343 int sa = gdk_pixbuf_get_rowstride (a); 534 int sa = gdk_pixbuf_get_rowstride (a);
344 int sb = gdk_pixbuf_get_rowstride (b); 535 int sb = gdk_pixbuf_get_rowstride (b);
345 536
346 guchar *pa = gdk_pixbuf_get_pixels (a); 537 guchar *pa = gdk_pixbuf_get_pixels (a);
347 guchar *pb = gdk_pixbuf_get_pixels (b); 538 guchar *pb = gdk_pixbuf_get_pixels (b);
355 assert (gdk_pixbuf_get_n_channels (b) == 3); 546 assert (gdk_pixbuf_get_n_channels (b) == 3);
356 547
357 double diff = 0.; 548 double diff = 0.;
358 int peak = 0; 549 int peak = 0;
359 550
551 if (w && h)
360 for (y = 0; y < h; y++) 552 for (y = 0; y < h; y++)
361 { 553 {
362 guchar *pa_ = pa + y * sa; 554 guchar *pa_ = pa + y * sa;
363 guchar *pb_ = pb + y * sb; 555 guchar *pb_ = pb + y * sb;
364 556
365 for (x = 0; x < w; x ++) 557 for (x = 0; x < w; x++)
366 { 558 {
367 int d; 559 int d;
368 560
369 d = ((int)*pa_++) - ((int)*pb_++); diff += d*d; peak = MAX (peak, abs (d)); 561 d = ((int)*pa_++) - ((int)*pb_++); diff += d*d; peak = MAX (peak, abs (d));
370 d = ((int)*pa_++) - ((int)*pb_++); diff += d*d; peak = MAX (peak, abs (d)); 562 d = ((int)*pa_++) - ((int)*pb_++); diff += d*d; peak = MAX (peak, abs (d));
371 d = ((int)*pa_++) - ((int)*pb_++); diff += d*d; peak = MAX (peak, abs (d)); 563 d = ((int)*pa_++) - ((int)*pb_++); diff += d*d; peak = MAX (peak, abs (d));
372 } 564 }
373 } 565 }
566
567 perlinterp_acquire ();
374 568
375 EXTEND (SP, 2); 569 EXTEND (SP, 2);
376 PUSHs (sv_2mortal (newSVnv (sqrt (diff / (w * h * 3. * 255. * 255.))))); 570 PUSHs (sv_2mortal (newSVnv (sqrt (diff / (w * h * 3. * 255. * 255.)))));
377 PUSHs (sv_2mortal (newSVnv (peak / 255.))); 571 PUSHs (sv_2mortal (newSVnv (peak / 255.)));
378} 572}
379 573
380############################################################################# 574#############################################################################
381 575
382MODULE = Gtk2::CV PACKAGE = Gtk2::CV::Schnauzer 576MODULE = Gtk2::CV PACKAGE = Gtk2::CV::Schnauzer
383 577
578# currently only works for filenames (octet strings)
579
384SV * 580SV *
385foldcase (SV *pathsv) 581foldcase (SV *pathsv)
386 PROTOTYPE: $ 582 PROTOTYPE: $
387 CODE: 583 CODE:
388{ 584{
389 STRLEN plen; 585 STRLEN plen;
390 U8 *path = (U8 *)SvPVutf8 (pathsv, plen); 586 U8 *path = (U8 *)SvPV (pathsv, plen);
391 U8 *pend = path + plen; 587 U8 *pend = path + plen;
392 U8 dst [plen * 6 * 3], *dstp = dst; 588 U8 dst [plen * 6 * 3], *dstp = dst;
393 589
394 while (path < pend) 590 while (path < pend)
395 { 591 {
396 U8 ch = *path; 592 U8 ch = *path;
397 593
398 if (ch >= 'a' && ch <= 'z') 594 if (ch >= 'a' && ch <= 'z')
399 *dstp++ = *path++; 595 *dstp++ = *path++;
596 else if (ch >= 'A' && ch <= 'Z')
597 *dstp++ = *path++ + ('a' - 'A');
400 else if (ch >= '0' && ch <= '9') 598 else if (ch >= '0' && ch <= '9')
401 { 599 {
402 STRLEN el, nl = 0; 600 STRLEN el, nl = 0;
403 while (*path >= '0' && *path <= '9' && path < pend) 601 while (*path >= '0' && *path <= '9' && path < pend)
404 path++, nl++; 602 path++, nl++;
407 *dstp++ = '0'; 605 *dstp++ = '0';
408 606
409 memcpy (dstp, path - nl, nl); 607 memcpy (dstp, path - nl, nl);
410 dstp += nl; 608 dstp += nl;
411 } 609 }
610 else
611 *dstp++ = *path++;
612#if 0
412 else 613 else
413 { 614 {
414 STRLEN cl; 615 STRLEN cl;
415 to_utf8_fold (path, dstp, &cl); 616 to_utf8_fold (path, dstp, &cl);
416 dstp += cl; 617 dstp += cl;
417 path += is_utf8_char (path); 618 path += is_utf8_char (path);
418 } 619 }
620#endif
419 } 621 }
420 622
421 RETVAL = newSVpvn ((const char *)dst, dstp - dst); 623 RETVAL = newSVpvn ((const char *)dst, dstp - dst);
422} 624}
423 OUTPUT: 625 OUTPUT:
585############################################################################# 787#############################################################################
586 788
587MODULE = Gtk2::CV PACKAGE = Gtk2::CV::Plugin::RCluster 789MODULE = Gtk2::CV PACKAGE = Gtk2::CV::Plugin::RCluster
588 790
589SV * 791SV *
590make_histograms (SV *ar) 792extract_features (SV *ar)
591 CODE: 793 CODE:
592{ 794{
593 int i; 795 int i;
594 AV *av, *result; 796 AV *av, *result;
595 797
596 if (!SvROK (ar) || SvTYPE (SvRV (ar)) != SVt_PVAV) 798 if (!SvROK (ar) || SvTYPE (SvRV (ar)) != SVt_PVAV)
597 croak ("Not an array ref as first argument to make_histogram"); 799 croak ("Not an array ref as first argument to extract_features");
598 800
599 av = (AV *) SvRV (ar); 801 av = (AV *) SvRV (ar);
600 result = newAV (); 802 result = newAV ();
601 803
602 for (i = 0; i <= av_len (av); ++i) 804 for (i = 0; i <= av_len (av); ++i)
603 { 805 {
604 const int HISTSIZE = 64;
605
606 int j;
607 SV *sv = *av_fetch (av, i, 1); 806 SV *sv = *av_fetch (av, i, 1);
608 STRLEN len;
609 char *buf = SvPVbyte (sv, len);
610
611 int tmphist[HISTSIZE];
612 float *hist;
613
614 SV *histsv = newSV (HISTSIZE * sizeof (float) + 1); 807 SV *histsv = newSV (9 * sizeof (float) + 1);
808
615 SvPOK_on (histsv); 809 SvPOK_on (histsv);
616 SvCUR_set (histsv, HISTSIZE * sizeof (float)); 810 SvCUR_set (histsv, 9 * sizeof (float));
617 hist = (float *)SvPVX (histsv); 811 float *hist = (float *)SvPVX (histsv);
618 812
619 Zero (tmphist, sizeof (tmphist), char); 813 struct feature f_h, f_s, f_v;
814 feature_init (&f_h);
815 feature_init (&f_s);
816 feature_init (&f_v);
620 817
621 for (j = len; j--; )
622 {
623 unsigned int idx
624 = ((*buf & 0xc0) >> 2)
625 | ((*buf & 0x18) >> 1)
626 | (*buf & 0x03);
627
628 ++tmphist[idx];
629 ++buf;
630 }
631 818 {
632 for (j = 0; j < HISTSIZE; ++j) 819 STRLEN len;
633 hist[j] = (float)tmphist[j] / (len + 1e-30); 820 unsigned char *buf = (unsigned char *)SvPVbyte (sv, len);
821 while (len >= 3)
822 {
823 unsigned int r, g, b, h, s, v;
824 r = *buf++; g = *buf++; b = *buf++;
825 rgb_to_hsv (r, g, b, &h, &s, &v);
826
827 feature_update_pass_1 (&f_h, h);
828 feature_update_pass_1 (&f_s, s);
829 feature_update_pass_1 (&f_v, v);
830
831 len -= 3;
832 }
833
834 feature_finish_pass_1 (&f_h);
835 feature_finish_pass_1 (&f_s);
836 feature_finish_pass_1 (&f_v);
837 }
838
839 {
840 STRLEN len;
841 unsigned char *buf = (unsigned char *)SvPVbyte (sv, len);
842 while (len >= 3)
843 {
844 unsigned int r, g, b, h, s, v;
845 r = *buf++; g = *buf++; b = *buf++;
846 rgb_to_hsv (r, g, b, &h, &s, &v);
847
848 feature_update_pass_2 (&f_h, h);
849 feature_update_pass_2 (&f_s, s);
850 feature_update_pass_2 (&f_v, v);
851
852 len -= 3;
853 }
854
855 feature_finish_pass_2 (&f_h);
856 feature_finish_pass_2 (&f_s);
857 feature_finish_pass_2 (&f_v);
858 }
859
860 hist [0] = f_h.v1 * 2.; hist [1] = f_h.v2 * 2.; hist [2] = f_h.v3 * 2.;
861 hist [3] = f_s.v1 ; hist [4] = f_s.v2 ; hist [5] = f_s.v3 ;
862 hist [6] = f_v.v1 * .5; hist [7] = f_v.v2 * .5; hist [8] = f_v.v3 * .5;
634 863
635 av_push (result, histsv); 864 av_push (result, histsv);
636 } 865 }
637 866
638 RETVAL = newRV_noinc ((SV *)result); 867 RETVAL = newRV_noinc ((SV *)result);
639} 868}
640 OUTPUT: 869 OUTPUT:
641 RETVAL 870 RETVAL
642 871
643

Diff Legend

Removed lines
+ Added lines
< Changed lines
> Changed lines