ViewVC Help
View File | Revision Log | Show Annotations | Download File
/cvs/CV/CV.xs
Revision: 1.41
Committed: Sat Mar 1 04:11:26 2008 UTC (16 years, 2 months ago) by root
Branch: MAIN
CVS Tags: rel-1_53
Changes since 1.40: +9 -1 lines
Log Message:
*** empty log message ***

File Contents

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