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

Comparing CV/CV.xs (file contents):
Revision 1.20 by root, Fri Jul 22 06:14:25 2005 UTC vs.
Revision 1.42 by root, Sun Sep 14 12:47:59 2008 UTC

2#include "perl.h" 2#include "perl.h"
3#include "XSUB.h" 3#include "XSUB.h"
4 4
5#include <string.h> 5#include <string.h>
6#include <setjmp.h> 6#include <setjmp.h>
7#include <math.h>
8
9#include <magic.h>
7 10
8#include <jpeglib.h> 11#include <jpeglib.h>
12#include <glib.h>
13#include <gtk/gtk.h>
9#include <gdk-pixbuf/gdk-pixbuf.h> 14#include <gdk-pixbuf/gdk-pixbuf.h>
10 15
11#include <gperl.h> 16#include <gperl.h>
12#include <gtk2perl.h> 17#include <gtk2perl.h>
13 18
14#define IW 80 /* MUST match Schnauer.pm! */ 19#define IW 80 /* MUST match Schnauzer.pm! */
15#define IH 60 /* MUST match Schnauer.pm! */ 20#define IH 60 /* MUST match Schnauzer.pm! */
16 21
17#define RAND (seed = (seed + 7141) * 54773 % 134456) 22#define RAND (seed = (seed + 7141) * 54773 % 134456)
18 23
19#define LINELENGTH 240 24#define LINELENGTH 240
20 25
21#define ELLIPSIS "\xe2\x80\xa6" 26#define ELLIPSIS "\xe2\x80\xa6"
27
28typedef char *octet_string;
22 29
23struct jpg_err_mgr 30struct jpg_err_mgr
24{ 31{
25 struct jpeg_error_mgr err; 32 struct jpeg_error_mgr err;
26 jmp_buf setjmp_buffer; 33 jmp_buf setjmp_buffer;
63 70
64 *h &= 255; 71 *h &= 255;
65 } 72 }
66} 73}
67 74
75struct feature {
76 float v1, v2, v3; // mean, square, cube
77 int n;
78};
79
80static void
81feature_init (struct feature *f)
82{
83 f->v1 = 0.;
84 f->v2 = 0.;
85 f->v3 = 0.;
86 f->n = 0;
87}
88
89// didn't find an algorithm to neatly do mean, variance and skew in one pass.
90// elmex ist schuld.
91static void
92feature_update_pass_1 (struct feature *f, unsigned int v)
93{
94 f->v1 += v;
95 f->n += 1;
96}
97
98static void
99feature_finish_pass_1 (struct feature *f)
100{
101 if (f->n < 1)
102 return;
103
104 f->v1 /= f->n;
105}
106
107static void
108feature_update_pass_2 (struct feature *f, unsigned int v)
109{
110 float d = v - f->v1;
111
112 f->v2 += d * d;
113 f->v3 += d * d * d;
114}
115
116static void
117feature_finish_pass_2 (struct feature *f)
118{
119 if (f->n < 1)
120 return;
121
122 f->v2 /= f->n;
123 f->v3 /= f->n;
124
125 f->v1 /= 255.;
126 f->v2 /= 255. * 255.; f->v2 = sqrtf (f->v2);
127 f->v3 /= 255. * 255. * 255.; f->v3 = powf (fabsf (f->v3), 1./3.);
128}
129
68static guint32 a85_val; 130static guint32 a85_val;
69static guint a85_cnt; 131static guint a85_cnt;
70static guchar a85_buf[LINELENGTH], *a85_ptr; 132static guchar a85_buf[LINELENGTH], *a85_ptr;
71 133
72static void 134static void
124 186
125MODULE = Gtk2::CV PACKAGE = Gtk2::CV 187MODULE = Gtk2::CV PACKAGE = Gtk2::CV
126 188
127PROTOTYPES: ENABLE 189PROTOTYPES: ENABLE
128 190
191# missing function in perl. really :)
192int
193common_prefix_length (a, b)
194 unsigned char *a = (unsigned char *)SvPVutf8_nolen ($arg);
195 unsigned char *b = (unsigned char *)SvPVutf8_nolen ($arg);
196 CODE:
197 RETVAL = 0;
198
199 while (*a == *b && *a)
200 {
201 RETVAL += (*a & 0xc0) != 0x80;
202 a++, b++;
203 }
204
205 OUTPUT:
206 RETVAL
207
208const char *
209magic (octet_string path)
210 CODE:
211{
212 static magic_t cookie;
213
214 if (!cookie)
215 {
216 cookie = magic_open (MAGIC_NONE);
217
218 if (cookie)
219 magic_load (cookie, 0);
220 else
221 XSRETURN_UNDEF;
222 }
223
224 RETVAL = magic_file (cookie, path);
225}
226 OUTPUT:
227 RETVAL
228
229const char *
230magic_mime (octet_string path)
231 CODE:
232{
233 static magic_t cookie;
234
235 if (!cookie)
236 {
237 cookie = magic_open (MAGIC_MIME);
238
239 if (cookie)
240 magic_load (cookie, 0);
241 else
242 XSRETURN_UNDEF;
243 }
244
245 RETVAL = magic_file (cookie, path);
246}
247 OUTPUT:
248 RETVAL
249
250# missing/broken in Gtk2 perl module
251
252void
253gdk_window_clear_hints (GdkWindow *window)
254 CODE:
255 gdk_window_set_geometry_hints (window, 0, 0);
256
257gboolean
258gdk_net_wm_supports (GdkAtom property)
259 CODE:
260#if defined(GDK_WINDOWING_X11) && !defined(GDK_MULTIHEAD_SAFE)
261 RETVAL = gdk_net_wm_supports (property);
262#else
263 RETVAL = 0;
264#endif
265 OUTPUT:
266 RETVAL
267
129GdkPixbuf_noinc * 268GdkPixbuf_noinc *
130transpose (GdkPixbuf *pb) 269dealpha_expose (GdkPixbuf *pb)
131 CODE: 270 CODE:
132{ 271{
133 int w = gdk_pixbuf_get_width (pb); 272 int w = gdk_pixbuf_get_width (pb);
134 int h = gdk_pixbuf_get_height (pb); 273 int h = gdk_pixbuf_get_height (pb);
135 int bpp = gdk_pixbuf_get_n_channels (pb); 274 int bpp = gdk_pixbuf_get_n_channels (pb);
136 int x, y, i; 275 int x, y, i;
137 guchar *src = gdk_pixbuf_get_pixels (pb), *dst; 276 guchar *src = gdk_pixbuf_get_pixels (pb), *dst;
138 int sstr = gdk_pixbuf_get_rowstride (pb), dstr; 277 int sstr = gdk_pixbuf_get_rowstride (pb), dstr;
139 278
140 RETVAL = gdk_pixbuf_new (GDK_COLORSPACE_RGB, bpp == 4, 8, h, w); 279 RETVAL = gdk_pixbuf_new (GDK_COLORSPACE_RGB, 0, 8, w, h);
141 280
142 dst = gdk_pixbuf_get_pixels (RETVAL); 281 dst = gdk_pixbuf_get_pixels (RETVAL);
143 dstr = gdk_pixbuf_get_rowstride (RETVAL); 282 dstr = gdk_pixbuf_get_rowstride (RETVAL);
144 283
145 for (y = 0; y < h; y++)
146 for (x = 0; x < w; x++) 284 for (x = 0; x < w; x++)
285 for (y = 0; y < h; y++)
147 for (i = 0; i < bpp; i++) 286 for (i = 0; i < 3; i++)
148 dst[y * bpp + x * dstr + i] = src[x * bpp + y * sstr + i]; 287 dst[x * 3 + y * dstr + i] = src[x * bpp + y * sstr + i];
149} 288}
150 OUTPUT: 289 OUTPUT:
151 RETVAL 290 RETVAL
152 291
153GdkPixbuf_noinc * 292GdkPixbuf_noinc *
154flop (GdkPixbuf *pb) 293rotate (GdkPixbuf *pb, int angle)
155 CODE: 294 CODE:
156{ 295 RETVAL = gdk_pixbuf_rotate_simple (pb, angle == 0 ? GDK_PIXBUF_ROTATE_NONE
157 int w = gdk_pixbuf_get_width (pb); 296 : angle == 90 ? GDK_PIXBUF_ROTATE_COUNTERCLOCKWISE
158 int h = gdk_pixbuf_get_height (pb); 297 : angle == 180 ? GDK_PIXBUF_ROTATE_UPSIDEDOWN
159 int bpp = gdk_pixbuf_get_n_channels (pb); 298 : angle == 270 ? GDK_PIXBUF_ROTATE_CLOCKWISE
160 int x, y, i; 299 : angle);
161 guchar *src = gdk_pixbuf_get_pixels (pb), *dst;
162 int sstr = gdk_pixbuf_get_rowstride (pb), dstr;
163
164 RETVAL = gdk_pixbuf_new (GDK_COLORSPACE_RGB, bpp == 4, 8, w, h);
165
166 dst = gdk_pixbuf_get_pixels (RETVAL);
167 dstr = gdk_pixbuf_get_rowstride (RETVAL);
168
169 for (y = 0; y < h; y++)
170 for (x = 0; x < w; x++)
171 for (i = 0; i < bpp; i++)
172 dst[(w - 1 - x) * bpp + y * dstr + i] = src[x * bpp + y * sstr + i];
173}
174 OUTPUT: 300 OUTPUT:
175 RETVAL 301 RETVAL
176 302
177GdkPixbuf_noinc * 303GdkPixbuf_noinc *
178load_jpeg (char *path, int thumbnail=0) 304load_jpeg (SV *path, int thumbnail=0)
179 CODE: 305 CODE:
180{ 306{
181 struct jpeg_decompress_struct cinfo; 307 struct jpeg_decompress_struct cinfo;
182 struct jpg_err_mgr jerr; 308 struct jpg_err_mgr jerr;
183 guchar *data; 309 guchar *data;
184 int rs; 310 int rs;
185 FILE *fp; 311 FILE *fp;
186 volatile GdkPixbuf *pb = 0; 312 volatile GdkPixbuf *pb = 0;
313
187 RETVAL = 0; 314 RETVAL = 0;
188 315
189 if (!(fp = fopen (path, "rb"))) 316 fp = fopen (SvPVbyte_nolen (path), "rb");
317
318 if (!fp)
190 XSRETURN_UNDEF; 319 XSRETURN_UNDEF;
191 320
192 cinfo.err = jpeg_std_error (&jerr.err); 321 cinfo.err = jpeg_std_error (&jerr.err);
193 322
194 jerr.err.error_exit = cv_error_exit; 323 jerr.err.error_exit = cv_error_exit;
225 { 354 {
226 cinfo.dct_method = JDCT_FASTEST; 355 cinfo.dct_method = JDCT_FASTEST;
227 cinfo.do_fancy_upsampling = FALSE; 356 cinfo.do_fancy_upsampling = FALSE;
228 357
229 while (cinfo.scale_denom < 8 358 while (cinfo.scale_denom < 8
230 && (cinfo.output_width >> 1) >= IW 359 && cinfo.output_width >= IW*4
231 && (cinfo.output_height >> 1) >= IH) 360 && cinfo.output_height >= IH*4)
232 { 361 {
233 cinfo.scale_denom <<= 1; 362 cinfo.scale_denom <<= 1;
234 jpeg_calc_output_dimensions (&cinfo); 363 jpeg_calc_output_dimensions (&cinfo);
235 } 364 }
236 } 365 }
265 jpeg_destroy_decompress (&cinfo); 394 jpeg_destroy_decompress (&cinfo);
266} 395}
267 OUTPUT: 396 OUTPUT:
268 RETVAL 397 RETVAL
269 398
399void
400compare (GdkPixbuf *a, GdkPixbuf *b)
401 PPCODE:
402{
403 int w = gdk_pixbuf_get_width (a);
404 int h = gdk_pixbuf_get_height (a);
405 int sa = gdk_pixbuf_get_rowstride (a);
406 int sb = gdk_pixbuf_get_rowstride (b);
407
408 guchar *pa = gdk_pixbuf_get_pixels (a);
409 guchar *pb = gdk_pixbuf_get_pixels (b);
410
411 int x, y;
412
413 assert (w == gdk_pixbuf_get_width (b));
414 assert (h == gdk_pixbuf_get_height (b));
415
416 assert (gdk_pixbuf_get_n_channels (a) == 3);
417 assert (gdk_pixbuf_get_n_channels (b) == 3);
418
419 double diff = 0.;
420 int peak = 0;
421
422 if (w && h)
423 for (y = 0; y < h; y++)
424 {
425 guchar *pa_ = pa + y * sa;
426 guchar *pb_ = pb + y * sb;
427
428 for (x = 0; x < w; x++)
429 {
430 int d;
431
432 d = ((int)*pa_++) - ((int)*pb_++); diff += d*d; peak = MAX (peak, abs (d));
433 d = ((int)*pa_++) - ((int)*pb_++); diff += d*d; peak = MAX (peak, abs (d));
434 d = ((int)*pa_++) - ((int)*pb_++); diff += d*d; peak = MAX (peak, abs (d));
435 }
436 }
437
438 EXTEND (SP, 2);
439 PUSHs (sv_2mortal (newSVnv (sqrt (diff / (w * h * 3. * 255. * 255.)))));
440 PUSHs (sv_2mortal (newSVnv (peak / 255.)));
441}
442
270############################################################################# 443#############################################################################
271 444
272MODULE = Gtk2::CV PACKAGE = Gtk2::CV::Schnauzer 445MODULE = Gtk2::CV PACKAGE = Gtk2::CV::Schnauzer
446
447# currently only works for filenames (octet strings)
273 448
274SV * 449SV *
275foldcase (SV *pathsv) 450foldcase (SV *pathsv)
276 PROTOTYPE: $ 451 PROTOTYPE: $
277 CODE: 452 CODE:
278{ 453{
279 STRLEN plen; 454 STRLEN plen;
280 U8 *path = SvPVutf8 (pathsv, plen); 455 U8 *path = (U8 *)SvPV (pathsv, plen);
281 U8 *pend = path + plen; 456 U8 *pend = path + plen;
282 U8 dst [plen * 6 * 3], *dstp = dst; 457 U8 dst [plen * 6 * 3], *dstp = dst;
283 458
284 while (path < pend) 459 while (path < pend)
285 { 460 {
286 U8 ch = *path; 461 U8 ch = *path;
287 462
288 if (ch >= 'a' && ch <= 'z') 463 if (ch >= 'a' && ch <= 'z')
289 *dstp++ = *path++; 464 *dstp++ = *path++;
465 else if (ch >= 'A' && ch <= 'Z')
466 *dstp++ = *path++ + ('a' - 'A');
290 else if (ch >= '0' && ch <= '9') 467 else if (ch >= '0' && ch <= '9')
291 { 468 {
292 STRLEN el, nl = 0; 469 STRLEN el, nl = 0;
293 while (*path >= '0' && *path <= '9' && path < pend) 470 while (*path >= '0' && *path <= '9' && path < pend)
294 path++, nl++; 471 path++, nl++;
297 *dstp++ = '0'; 474 *dstp++ = '0';
298 475
299 memcpy (dstp, path - nl, nl); 476 memcpy (dstp, path - nl, nl);
300 dstp += nl; 477 dstp += nl;
301 } 478 }
479 else
480 *dstp++ = *path++;
481#if 0
302 else 482 else
303 { 483 {
304 STRLEN cl; 484 STRLEN cl;
305 to_utf8_fold (path, dstp, &cl); 485 to_utf8_fold (path, dstp, &cl);
306 dstp += cl; 486 dstp += cl;
307 path += is_utf8_char (path); 487 path += is_utf8_char (path);
308 } 488 }
489#endif
309 } 490 }
310 491
311 RETVAL = newSVpvn (dst, dstp - dst); 492 RETVAL = newSVpvn ((const char *)dst, dstp - dst);
312} 493}
313 OUTPUT: 494 OUTPUT:
314 RETVAL 495 RETVAL
315 496
316GdkPixbuf_noinc * 497GdkPixbuf_noinc *
317p7_to_pb (int w, int h, guchar *src) 498p7_to_pb (int w, int h, SV *src_sv)
499 PROTOTYPE: @
318 CODE: 500 CODE:
319{ 501{
320 int x, y; 502 int x, y;
321 guchar *dst, *d; 503 guchar *dst, *d;
322 int dstr; 504 int dstr;
505 guchar *src = (guchar *)SvPVbyte_nolen (src_sv);
323 506
324 RETVAL = gdk_pixbuf_new (GDK_COLORSPACE_RGB, 0, 8, w, h); 507 RETVAL = gdk_pixbuf_new (GDK_COLORSPACE_RGB, 0, 8, w, h);
325 dst = gdk_pixbuf_get_pixels (RETVAL); 508 dst = gdk_pixbuf_get_pixels (RETVAL);
326 dstr = gdk_pixbuf_get_rowstride (RETVAL); 509 dstr = gdk_pixbuf_get_rowstride (RETVAL);
327 510
336 } 519 }
337} 520}
338 OUTPUT: 521 OUTPUT:
339 RETVAL 522 RETVAL
340 523
524#############################################################################
525
526MODULE = Gtk2::CV PACKAGE = Gtk2::CV::PostScript
527
528void
529dump_ascii85 (PerlIO *fp, GdkPixbuf *pb)
530 CODE:
531{
532 int w = gdk_pixbuf_get_width (pb);
533 int h = gdk_pixbuf_get_height (pb);
534 int x, y, i;
535 guchar *dst;
536 int bpp = gdk_pixbuf_get_n_channels (pb);
537 guchar *src = gdk_pixbuf_get_pixels (pb);
538 int sstr = gdk_pixbuf_get_rowstride (pb);
539
540 a85_init ();
541
542 for (y = 0; y < h; y++)
543 for (x = 0; x < w; x++)
544 for (i = 0; i < (bpp < 3 ? 1 : 3); i++)
545 a85_push (fp, src [x * bpp + y * sstr + i]);
546
547 a85_finish (fp);
548}
549
550void
551dump_binary (PerlIO *fp, GdkPixbuf *pb)
552 CODE:
553{
554 int w = gdk_pixbuf_get_width (pb);
555 int h = gdk_pixbuf_get_height (pb);
556 int x, y, i;
557 guchar *dst;
558 int bpp = gdk_pixbuf_get_n_channels (pb);
559 guchar *src = gdk_pixbuf_get_pixels (pb);
560 int sstr = gdk_pixbuf_get_rowstride (pb);
561
562 for (y = 0; y < h; y++)
563 for (x = 0; x < w; x++)
564 for (i = 0; i < (bpp < 3 ? 1 : 3); i++)
565 PerlIO_putc (fp, src [x * bpp + y * sstr + i]);
566}
567
568#############################################################################
569
570MODULE = Gtk2::CV PACKAGE = Gtk2::CV
571
341SV * 572SV *
342pb_to_p7 (GdkPixbuf *pb) 573pb_to_hv84 (GdkPixbuf *pb)
343 CODE: 574 CODE:
344{ 575{
345 int w = gdk_pixbuf_get_width (pb); 576 int w = gdk_pixbuf_get_width (pb);
346 int h = gdk_pixbuf_get_height (pb); 577 int h = gdk_pixbuf_get_height (pb);
347 int x, y; 578 int x, y;
348 guchar *dst; 579 guchar *dst;
349 int bpp = gdk_pixbuf_get_n_channels (pb); 580 int bpp = gdk_pixbuf_get_n_channels (pb);
350 guchar *src = gdk_pixbuf_get_pixels (pb); 581 guchar *src = gdk_pixbuf_get_pixels (pb);
351 int sstr = gdk_pixbuf_get_rowstride (pb); 582 int sstr = gdk_pixbuf_get_rowstride (pb);
352 int Er[IW], Eg[IW], Eb[IW];
353 int seed = 77;
354
355 RETVAL = newSV (w * h);
356 SvPOK_only (RETVAL);
357 SvCUR_set (RETVAL, w * h);
358
359 dst = SvPVX (RETVAL);
360
361 memset (Er, 0, sizeof (int) * IW);
362 memset (Eg, 0, sizeof (int) * IW);
363 memset (Eb, 0, sizeof (int) * IW);
364
365 /* some primitive error distribution + random dithering */
366
367 for (y = 0; y < h; y++)
368 {
369 int er = 0, eg = 0, eb = 0;
370
371 for (x = 0; x < w; x++)
372 {
373 int r, g, b;
374 guchar *p = src + x * bpp + y * sstr;
375
376 r = ((p[0] + er + Er[x]) * 7 + (RAND & 127) + 64) / 255;
377 g = ((p[1] + eg + Eg[x]) * 7 + (RAND & 127) + 64) / 255;
378 b = ((p[2] + eb + Eb[x]) * 3 + (RAND & 127) + 64) / 255;
379
380 r = r > 7 ? 7 : r < 0 ? 0 : r;
381 g = g > 7 ? 7 : g < 0 ? 0 : g;
382 b = b > 3 ? 3 : b < 0 ? 0 : b;
383
384 er += p[0] - (r * 255 + 4) / 7;
385 eg += p[1] - (g * 255 + 4) / 7;
386 eb += p[2] - (b * 255 + 2) / 3;
387
388 Er[x] = er >> 1; er -= (er + 1) >> 1;
389 Eg[x] = eg >> 1; eg -= (eg + 1) >> 1;
390 Eb[x] = eb >> 1; eb -= (eb + 1) >> 1;
391
392 *dst++ = r << 5 | g << 2 | b;
393 }
394 }
395}
396 OUTPUT:
397 RETVAL
398
399#############################################################################
400
401MODULE = Gtk2::CV PACKAGE = Gtk2::CV::PostScript
402
403void
404dump_ascii85 (PerlIO *fp, GdkPixbuf *pb)
405 CODE:
406{
407 int w = gdk_pixbuf_get_width (pb);
408 int h = gdk_pixbuf_get_height (pb);
409 int x, y, i;
410 guchar *dst;
411 int bpp = gdk_pixbuf_get_n_channels (pb);
412 guchar *src = gdk_pixbuf_get_pixels (pb);
413 int sstr = gdk_pixbuf_get_rowstride (pb);
414
415 a85_init ();
416
417 for (y = 0; y < h; y++)
418 for (x = 0; x < w; x++)
419 for (i = 0; i < (bpp < 3 ? 1 : 3); i++)
420 a85_push (fp, src [x * bpp + y * sstr + i]);
421
422 a85_finish (fp);
423}
424
425void
426dump_binary (PerlIO *fp, GdkPixbuf *pb)
427 CODE:
428{
429 int w = gdk_pixbuf_get_width (pb);
430 int h = gdk_pixbuf_get_height (pb);
431 int x, y, i;
432 guchar *dst;
433 int bpp = gdk_pixbuf_get_n_channels (pb);
434 guchar *src = gdk_pixbuf_get_pixels (pb);
435 int sstr = gdk_pixbuf_get_rowstride (pb);
436
437 for (y = 0; y < h; y++)
438 for (x = 0; x < w; x++)
439 for (i = 0; i < (bpp < 3 ? 1 : 3); i++)
440 PerlIO_putc (fp, src [x * bpp + y * sstr + i]);
441}
442
443#############################################################################
444
445MODULE = Gtk2::CV PACKAGE = Gtk2::CV
446
447SV *
448pb_to_hv84 (GdkPixbuf *pb)
449 CODE:
450{
451 int w = gdk_pixbuf_get_width (pb);
452 int h = gdk_pixbuf_get_height (pb);
453 int x, y;
454 guchar *dst;
455 int bpp = gdk_pixbuf_get_n_channels (pb);
456 guchar *src = gdk_pixbuf_get_pixels (pb);
457 int sstr = gdk_pixbuf_get_rowstride (pb);
458 583
459 RETVAL = newSV (6 * 8 * 12 / 8); 584 RETVAL = newSV (6 * 8 * 12 / 8);
460 SvPOK_only (RETVAL); 585 SvPOK_only (RETVAL);
461 SvCUR_set (RETVAL, 6 * 8 * 12 / 8); 586 SvCUR_set (RETVAL, 6 * 8 * 12 / 8);
462 587
463 dst = SvPVX (RETVAL); 588 dst = (guchar *)SvPVX (RETVAL);
464 589
465 /* some primitive error distribution + random dithering */ 590 /* some primitive error distribution + random dithering */
466 591
467 for (y = 0; y < h; y++) 592 for (y = 0; y < h; y++)
468 { 593 {
531############################################################################# 656#############################################################################
532 657
533MODULE = Gtk2::CV PACKAGE = Gtk2::CV::Plugin::RCluster 658MODULE = Gtk2::CV PACKAGE = Gtk2::CV::Plugin::RCluster
534 659
535SV * 660SV *
536make_histograms (SV *ar) 661extract_features (SV *ar)
537 CODE: 662 CODE:
538{ 663{
539 int i; 664 int i;
540 AV *av, *result; 665 AV *av, *result;
541 666
542 if (!SvROK (ar) || SvTYPE (SvRV (ar)) != SVt_PVAV) 667 if (!SvROK (ar) || SvTYPE (SvRV (ar)) != SVt_PVAV)
543 croak ("Not an array ref as first argument to make_histogram"); 668 croak ("Not an array ref as first argument to extract_features");
544 669
545 av = (AV *) SvRV (ar); 670 av = (AV *) SvRV (ar);
546 result = newAV (); 671 result = newAV ();
547 672
548 for (i = 0; i <= av_len (av); ++i) 673 for (i = 0; i <= av_len (av); ++i)
549 { 674 {
550 const int HISTSIZE = 64;
551
552 int j;
553 SV *sv = *av_fetch (av, i, 1); 675 SV *sv = *av_fetch (av, i, 1);
554 STRLEN len;
555 char *buf = SvPVbyte (sv, len);
556
557 int tmphist[HISTSIZE];
558 float *hist;
559
560 SV *histsv = newSV (HISTSIZE * sizeof (float) + 1); 676 SV *histsv = newSV (9 * sizeof (float) + 1);
677
561 SvPOK_on (histsv); 678 SvPOK_on (histsv);
562 SvCUR_set (histsv, HISTSIZE * sizeof (float)); 679 SvCUR_set (histsv, 9 * sizeof (float));
563 hist = (float *)SvPVX (histsv); 680 float *hist = (float *)SvPVX (histsv);
564 681
565 Zero (tmphist, sizeof (tmphist), char); 682 struct feature f_h, f_s, f_v;
683 feature_init (&f_h);
684 feature_init (&f_s);
685 feature_init (&f_v);
566 686
567 for (j = len; j--; )
568 {
569 unsigned int idx
570 = ((*buf & 0xc0) >> 2)
571 | ((*buf & 0x18) >> 1)
572 | (*buf & 0x03);
573
574 ++tmphist[idx];
575 ++buf;
576 }
577 687 {
578 for (j = 0; j < HISTSIZE; ++j) 688 STRLEN len;
579 hist[j] = (float)tmphist[j] / (len + 1e-30); 689 unsigned char *buf = (unsigned char *)SvPVbyte (sv, len);
690 while (len >= 3)
691 {
692 unsigned int r, g, b, h, s, v;
693 r = *buf++; g = *buf++; b = *buf++;
694 rgb_to_hsv (r, g, b, &h, &s, &v);
695
696 feature_update_pass_1 (&f_h, h);
697 feature_update_pass_1 (&f_s, s);
698 feature_update_pass_1 (&f_v, v);
699
700 len -= 3;
701 }
702
703 feature_finish_pass_1 (&f_h);
704 feature_finish_pass_1 (&f_s);
705 feature_finish_pass_1 (&f_v);
706 }
707
708 {
709 STRLEN len;
710 unsigned char *buf = (unsigned char *)SvPVbyte (sv, len);
711 while (len >= 3)
712 {
713 unsigned int r, g, b, h, s, v;
714 r = *buf++; g = *buf++; b = *buf++;
715 rgb_to_hsv (r, g, b, &h, &s, &v);
716
717 feature_update_pass_2 (&f_h, h);
718 feature_update_pass_2 (&f_s, s);
719 feature_update_pass_2 (&f_v, v);
720
721 len -= 3;
722 }
723
724 feature_finish_pass_2 (&f_h);
725 feature_finish_pass_2 (&f_s);
726 feature_finish_pass_2 (&f_v);
727 }
728
729 hist [0] = f_h.v1 * 2.; hist [1] = f_h.v2 * 2.; hist [2] = f_h.v3 * 2.;
730 hist [3] = f_s.v1 ; hist [4] = f_s.v2 ; hist [5] = f_s.v3 ;
731 hist [6] = f_v.v1 * .5; hist [7] = f_v.v2 * .5; hist [8] = f_v.v3 * .5;
580 732
581 av_push (result, histsv); 733 av_push (result, histsv);
582 } 734 }
583 735
584 RETVAL = newRV_noinc ((SV *)result); 736 RETVAL = newRV_noinc ((SV *)result);
585} 737}
586 OUTPUT: 738 OUTPUT:
587 RETVAL 739 RETVAL
588 740
589

Diff Legend

Removed lines
+ Added lines
< Changed lines
> Changed lines