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

Comparing CV/CV.xs (file contents):
Revision 1.30 by root, Wed Dec 28 20:56:20 2005 UTC vs.
Revision 1.39 by root, Fri Jun 23 09:09:14 2006 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>
9#include <glib.h> 12#include <glib.h>
10#include <gtk/gtk.h> 13#include <gtk/gtk.h>
11#include <gdk-pixbuf/gdk-pixbuf.h> 14#include <gdk-pixbuf/gdk-pixbuf.h>
12 15
13#include <gperl.h> 16#include <gperl.h>
14#include <gtk2perl.h> 17#include <gtk2perl.h>
15 18
16#define IW 80 /* MUST match Schnauer.pm! */ 19#define IW 80 /* MUST match Schnauzer.pm! */
17#define IH 60 /* MUST match Schnauer.pm! */ 20#define IH 60 /* MUST match Schnauzer.pm! */
18 21
19#define RAND (seed = (seed + 7141) * 54773 % 134456) 22#define RAND (seed = (seed + 7141) * 54773 % 134456)
20 23
21#define LINELENGTH 240 24#define LINELENGTH 240
22 25
65 68
66 *h &= 255; 69 *h &= 255;
67 } 70 }
68} 71}
69 72
73struct feature {
74 float v1, v2, v3; // mean, square, cube
75 int n;
76};
77
78static void
79feature_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.
89static void
90feature_update_pass_1 (struct feature *f, unsigned int v)
91{
92 f->v1 += v;
93 f->n += 1;
94}
95
96static void
97feature_finish_pass_1 (struct feature *f)
98{
99 if (f->n < 1)
100 return;
101
102 f->v1 /= f->n;
103}
104
105static void
106feature_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
114static void
115feature_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
70static guint32 a85_val; 128static guint32 a85_val;
71static guint a85_cnt; 129static guint a85_cnt;
72static guchar a85_buf[LINELENGTH], *a85_ptr; 130static guchar a85_buf[LINELENGTH], *a85_ptr;
73 131
74static void 132static void
143 } 201 }
144 202
145 OUTPUT: 203 OUTPUT:
146 RETVAL 204 RETVAL
147 205
206const char *
207magic (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
227const char *
228magic_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
148# missing in Gtk2 perl module 248# missing in Gtk2 perl module
149 249
150gboolean 250gboolean
151gdk_net_wm_supports (GdkAtom property) 251gdk_net_wm_supports (GdkAtom property)
152 CODE: 252 CODE:
286 fclose (fp); 386 fclose (fp);
287 jpeg_destroy_decompress (&cinfo); 387 jpeg_destroy_decompress (&cinfo);
288} 388}
289 OUTPUT: 389 OUTPUT:
290 RETVAL 390 RETVAL
391
392void
393compare (GdkPixbuf *a, GdkPixbuf *b)
394 PPCODE:
395{
396 int w = gdk_pixbuf_get_width (a);
397 int h = gdk_pixbuf_get_height (a);
398 int sa = gdk_pixbuf_get_rowstride (a);
399 int sb = gdk_pixbuf_get_rowstride (b);
400
401 guchar *pa = gdk_pixbuf_get_pixels (a);
402 guchar *pb = gdk_pixbuf_get_pixels (b);
403
404 int x, y;
405
406 assert (w == gdk_pixbuf_get_width (b));
407 assert (h == gdk_pixbuf_get_height (b));
408
409 assert (gdk_pixbuf_get_n_channels (a) == 3);
410 assert (gdk_pixbuf_get_n_channels (b) == 3);
411
412 double diff = 0.;
413 int peak = 0;
414
415 if (w && h)
416 for (y = 0; y < h; y++)
417 {
418 guchar *pa_ = pa + y * sa;
419 guchar *pb_ = pb + y * sb;
420
421 for (x = 0; x < w; x++)
422 {
423 int d;
424
425 d = ((int)*pa_++) - ((int)*pb_++); diff += d*d; peak = MAX (peak, abs (d));
426 d = ((int)*pa_++) - ((int)*pb_++); diff += d*d; peak = MAX (peak, abs (d));
427 d = ((int)*pa_++) - ((int)*pb_++); diff += d*d; peak = MAX (peak, abs (d));
428 }
429 }
430
431 EXTEND (SP, 2);
432 PUSHs (sv_2mortal (newSVnv (sqrt (diff / (w * h * 3. * 255. * 255.)))));
433 PUSHs (sv_2mortal (newSVnv (peak / 255.)));
434}
291 435
292############################################################################# 436#############################################################################
293 437
294MODULE = Gtk2::CV PACKAGE = Gtk2::CV::Schnauzer 438MODULE = Gtk2::CV PACKAGE = Gtk2::CV::Schnauzer
295 439
497############################################################################# 641#############################################################################
498 642
499MODULE = Gtk2::CV PACKAGE = Gtk2::CV::Plugin::RCluster 643MODULE = Gtk2::CV PACKAGE = Gtk2::CV::Plugin::RCluster
500 644
501SV * 645SV *
502make_histograms (SV *ar) 646extract_features (SV *ar)
503 CODE: 647 CODE:
504{ 648{
505 int i; 649 int i;
506 AV *av, *result; 650 AV *av, *result;
507 651
508 if (!SvROK (ar) || SvTYPE (SvRV (ar)) != SVt_PVAV) 652 if (!SvROK (ar) || SvTYPE (SvRV (ar)) != SVt_PVAV)
509 croak ("Not an array ref as first argument to make_histogram"); 653 croak ("Not an array ref as first argument to extract_features");
510 654
511 av = (AV *) SvRV (ar); 655 av = (AV *) SvRV (ar);
512 result = newAV (); 656 result = newAV ();
513 657
514 for (i = 0; i <= av_len (av); ++i) 658 for (i = 0; i <= av_len (av); ++i)
515 { 659 {
516 const int HISTSIZE = 64;
517
518 int j;
519 SV *sv = *av_fetch (av, i, 1); 660 SV *sv = *av_fetch (av, i, 1);
520 STRLEN len;
521 char *buf = SvPVbyte (sv, len);
522
523 int tmphist[HISTSIZE];
524 float *hist;
525
526 SV *histsv = newSV (HISTSIZE * sizeof (float) + 1); 661 SV *histsv = newSV (9 * sizeof (float) + 1);
662
527 SvPOK_on (histsv); 663 SvPOK_on (histsv);
528 SvCUR_set (histsv, HISTSIZE * sizeof (float)); 664 SvCUR_set (histsv, 9 * sizeof (float));
529 hist = (float *)SvPVX (histsv); 665 float *hist = (float *)SvPVX (histsv);
530 666
531 Zero (tmphist, sizeof (tmphist), char); 667 struct feature f_h, f_s, f_v;
668 feature_init (&f_h);
669 feature_init (&f_s);
670 feature_init (&f_v);
532 671
533 for (j = len; j--; )
534 {
535 unsigned int idx
536 = ((*buf & 0xc0) >> 2)
537 | ((*buf & 0x18) >> 1)
538 | (*buf & 0x03);
539
540 ++tmphist[idx];
541 ++buf;
542 }
543 672 {
544 for (j = 0; j < HISTSIZE; ++j) 673 STRLEN len;
545 hist[j] = (float)tmphist[j] / (len + 1e-30); 674 unsigned char *buf = (unsigned char *)SvPVbyte (sv, len);
675 while (len >= 3)
676 {
677 unsigned int r, g, b, h, s, v;
678 r = *buf++; g = *buf++; b = *buf++;
679 rgb_to_hsv (r, g, b, &h, &s, &v);
680
681 feature_update_pass_1 (&f_h, h);
682 feature_update_pass_1 (&f_s, s);
683 feature_update_pass_1 (&f_v, v);
684
685 len -= 3;
686 }
687
688 feature_finish_pass_1 (&f_h);
689 feature_finish_pass_1 (&f_s);
690 feature_finish_pass_1 (&f_v);
691 }
692
693 {
694 STRLEN len;
695 unsigned char *buf = (unsigned char *)SvPVbyte (sv, len);
696 while (len >= 3)
697 {
698 unsigned int r, g, b, h, s, v;
699 r = *buf++; g = *buf++; b = *buf++;
700 rgb_to_hsv (r, g, b, &h, &s, &v);
701
702 feature_update_pass_2 (&f_h, h);
703 feature_update_pass_2 (&f_s, s);
704 feature_update_pass_2 (&f_v, v);
705
706 len -= 3;
707 }
708
709 feature_finish_pass_2 (&f_h);
710 feature_finish_pass_2 (&f_s);
711 feature_finish_pass_2 (&f_v);
712 }
713
714 hist [0] = f_h.v1 * 2.; hist [1] = f_h.v2 * 2.; hist [2] = f_h.v3 * 2.;
715 hist [3] = f_s.v1 ; hist [4] = f_s.v2 ; hist [5] = f_s.v3 ;
716 hist [6] = f_v.v1 * .5; hist [7] = f_v.v2 * .5; hist [8] = f_v.v3 * .5;
546 717
547 av_push (result, histsv); 718 av_push (result, histsv);
548 } 719 }
549 720
550 RETVAL = newRV_noinc ((SV *)result); 721 RETVAL = newRV_noinc ((SV *)result);
551} 722}
552 OUTPUT: 723 OUTPUT:
553 RETVAL 724 RETVAL
554 725
555

Diff Legend

Removed lines
+ Added lines
< Changed lines
> Changed lines