--- CV/CV.xs 2005/07/20 02:26:46 1.16 +++ CV/CV.xs 2009/05/03 08:16:47 1.43 @@ -2,12 +2,24 @@ #include "perl.h" #include "XSUB.h" +#include +#include +#include + +#include + +#include +#include +#include #include #include #include -#define IW 80 +#include + +#define IW 80 /* MUST match Schnauzer.pm! */ +#define IH 60 /* MUST match Schnauzer.pm! */ #define RAND (seed = (seed + 7141) * 54773 % 134456) @@ -15,6 +27,108 @@ #define ELLIPSIS "\xe2\x80\xa6" +typedef char *octet_string; + +struct jpg_err_mgr +{ + struct jpeg_error_mgr err; + jmp_buf setjmp_buffer; +}; + +static void +cv_error_exit (j_common_ptr cinfo) +{ + longjmp (((struct jpg_err_mgr *)cinfo->err)->setjmp_buffer, 99); +} + +static void +cv_error_output (j_common_ptr cinfo) +{ + return; +} + +static void +rgb_to_hsv (unsigned int r, unsigned int g, unsigned int b, + unsigned int *h, unsigned int *s, unsigned int *v) +{ + unsigned int mx = r; if (g > mx) mx = g; if (b > mx) mx = b; + unsigned int mn = r; if (g < mn) mn = g; if (b < mn) mn = b; + unsigned int delta = mx - mn; + + *v = mx; + + *s = mx ? delta * 255 / mx : 0; + + if (delta == 0) + *h = 0; + else + { + if (r == mx) + *h = ((int)g - (int)b) * 255 / (int)(delta * 3); + else if (g == mx) + *h = ((int)b - (int)r) * 255 / (int)(delta * 3) + 52; + else if (b == mx) + *h = ((int)r - (int)g) * 255 / (int)(delta * 3) + 103; + + *h &= 255; + } +} + +struct feature { + float v1, v2, v3; // mean, square, cube + int n; +}; + +static void +feature_init (struct feature *f) +{ + f->v1 = 0.; + f->v2 = 0.; + f->v3 = 0.; + f->n = 0; +} + +// didn't find an algorithm to neatly do mean, variance and skew in one pass. +// elmex ist schuld. +static void +feature_update_pass_1 (struct feature *f, unsigned int v) +{ + f->v1 += v; + f->n += 1; +} + +static void +feature_finish_pass_1 (struct feature *f) +{ + if (f->n < 1) + return; + + f->v1 /= f->n; +} + +static void +feature_update_pass_2 (struct feature *f, unsigned int v) +{ + float d = v - f->v1; + + f->v2 += d * d; + f->v3 += d * d * d; +} + +static void +feature_finish_pass_2 (struct feature *f) +{ + if (f->n < 1) + return; + + f->v2 /= f->n; + f->v3 /= f->n; + + f->v1 /= 255.; + f->v2 /= 255. * 255.; f->v2 = sqrtf (f->v2); + f->v3 /= 255. * 255. * 255.; f->v3 = powf (fabsf (f->v3), 1./3.); +} + static guint32 a85_val; static guint a85_cnt; static guchar a85_buf[LINELENGTH], *a85_ptr; @@ -70,65 +184,91 @@ PerlIO_write (fp, a85_buf, a85_ptr - a85_buf); } -static void -rgb_to_hsv (unsigned int r, unsigned int g, unsigned int b, - unsigned int *h, unsigned int *s, unsigned int *v) -{ - unsigned int mx = r; if (g > mx) mx = g; if (b > mx) mx = b; - unsigned int mn = r; if (g < mn) mn = g; if (b < mn) mn = b; - unsigned int delta = mx - mn; +///////////////////////////////////////////////////////////////////////////// - *v = mx; +MODULE = Gtk2::CV PACKAGE = Gtk2::CV - *s = mx ? delta * 255 / mx : 0; +PROTOTYPES: ENABLE - if (delta == 0) - *h = 0; - else - { - if (r == mx) - *h = ((int)g - (int)b) * 255 / (int)(delta * 3); - else if (g == mx) - *h = ((int)b - (int)r) * 255 / (int)(delta * 3) + 52; - else if (b == mx) - *h = ((int)r - (int)g) * 255 / (int)(delta * 3) + 103; +# missing function in perl. really :) +int +common_prefix_length (a, b) + unsigned char *a = (unsigned char *)SvPVutf8_nolen ($arg); + unsigned char *b = (unsigned char *)SvPVutf8_nolen ($arg); + CODE: + RETVAL = 0; - *h &= 255; - } -} + while (*a == *b && *a) + { + RETVAL += (*a & 0xc0) != 0x80; + a++, b++; + } -///////////////////////////////////////////////////////////////////////////// + OUTPUT: + RETVAL -MODULE = Gtk2::CV PACKAGE = Gtk2::CV +const char * +magic (octet_string path) + CODE: +{ + static magic_t cookie; -PROTOTYPES: ENABLE + if (!cookie) + { + cookie = magic_open (MAGIC_NONE); -GdkPixbuf_noinc * -transpose (GdkPixbuf *pb) + if (cookie) + magic_load (cookie, 0); + else + XSRETURN_UNDEF; + } + + RETVAL = magic_file (cookie, path); +} + OUTPUT: + RETVAL + +const char * +magic_mime (octet_string path) CODE: { - int w = gdk_pixbuf_get_width (pb); - int h = gdk_pixbuf_get_height (pb); - int bpp = gdk_pixbuf_get_n_channels (pb); - int x, y, i; - guchar *src = gdk_pixbuf_get_pixels (pb), *dst; - int sstr = gdk_pixbuf_get_rowstride (pb), dstr; + static magic_t cookie; - RETVAL = gdk_pixbuf_new (GDK_COLORSPACE_RGB, bpp == 4, 8, h, w); + if (!cookie) + { + cookie = magic_open (MAGIC_MIME); - dst = gdk_pixbuf_get_pixels (RETVAL); - dstr = gdk_pixbuf_get_rowstride (RETVAL); + if (cookie) + magic_load (cookie, 0); + else + XSRETURN_UNDEF; + } - for (y = 0; y < h; y++) - for (x = 0; x < w; x++) - for (i = 0; i < bpp; i++) - dst[y * bpp + x * dstr + i] = src[x * bpp + y * sstr + i]; + RETVAL = magic_file (cookie, path); } OUTPUT: RETVAL +# missing/broken in Gtk2 perl module + +void +gdk_window_clear_hints (GdkWindow *window) + CODE: + gdk_window_set_geometry_hints (window, 0, 0); + +gboolean +gdk_net_wm_supports (GdkAtom property) + CODE: +#if defined(GDK_WINDOWING_X11) && !defined(GDK_MULTIHEAD_SAFE) + RETVAL = gdk_net_wm_supports (property); +#else + RETVAL = 0; +#endif + OUTPUT: + RETVAL + GdkPixbuf_noinc * -flop (GdkPixbuf *pb) +dealpha_expose (GdkPixbuf *pb) CODE: { int w = gdk_pixbuf_get_width (pb); @@ -138,161 +278,251 @@ guchar *src = gdk_pixbuf_get_pixels (pb), *dst; int sstr = gdk_pixbuf_get_rowstride (pb), dstr; - RETVAL = gdk_pixbuf_new (GDK_COLORSPACE_RGB, bpp == 4, 8, w, h); + RETVAL = gdk_pixbuf_new (GDK_COLORSPACE_RGB, 0, 8, w, h); dst = gdk_pixbuf_get_pixels (RETVAL); dstr = gdk_pixbuf_get_rowstride (RETVAL); - for (y = 0; y < h; y++) - for (x = 0; x < w; x++) - for (i = 0; i < bpp; i++) - dst[(w - 1 - x) * bpp + y * dstr + i] = src[x * bpp + y * sstr + i]; + for (x = 0; x < w; x++) + for (y = 0; y < h; y++) + for (i = 0; i < 3; i++) + dst[x * 3 + y * dstr + i] = src[x * bpp + y * sstr + i]; } OUTPUT: RETVAL -############################################################################# - -MODULE = Gtk2::CV PACKAGE = Gtk2::CV::Schnauzer +GdkPixbuf_noinc * +rotate (GdkPixbuf *pb, int angle) + CODE: + RETVAL = gdk_pixbuf_rotate_simple (pb, angle == 0 ? GDK_PIXBUF_ROTATE_NONE + : angle == 90 ? GDK_PIXBUF_ROTATE_COUNTERCLOCKWISE + : angle == 180 ? GDK_PIXBUF_ROTATE_UPSIDEDOWN + : angle == 270 ? GDK_PIXBUF_ROTATE_CLOCKWISE + : angle); + OUTPUT: + RETVAL GdkPixbuf_noinc * -p7_to_pb (int w, int h, guchar *src) +load_jpeg (SV *path, int thumbnail=0) CODE: { - int x, y; - guchar *dst, *d; - int dstr; + struct jpeg_decompress_struct cinfo; + struct jpg_err_mgr jerr; + guchar *data; + int rs; + FILE *fp; + volatile GdkPixbuf *pb = 0; - RETVAL = gdk_pixbuf_new (GDK_COLORSPACE_RGB, 0, 8, w, h); - dst = gdk_pixbuf_get_pixels (RETVAL); - dstr = gdk_pixbuf_get_rowstride (RETVAL); + RETVAL = 0; - for (y = 0; y < h; y++) - for (d = dst + y * dstr, x = 0; x < w; x++) - { - *d++ = (((*src >> 5) & 7) * 255 + 4) / 7; - *d++ = (((*src >> 2) & 7) * 255 + 4) / 7; - *d++ = (((*src >> 0) & 3) * 255 + 2) / 3; + fp = fopen (SvPVbyte_nolen (path), "rb"); - src++; - } -} - OUTPUT: - RETVAL + if (!fp) + XSRETURN_UNDEF; -SV * -pb_to_p7 (GdkPixbuf *pb) - CODE: -{ - int w = gdk_pixbuf_get_width (pb); - int h = gdk_pixbuf_get_height (pb); - int x, y; - guchar *dst; - int bpp = gdk_pixbuf_get_n_channels (pb); - guchar *src = gdk_pixbuf_get_pixels (pb); - int sstr = gdk_pixbuf_get_rowstride (pb); - int Er[IW], Eg[IW], Eb[IW]; - int seed = 77; + cinfo.err = jpeg_std_error (&jerr.err); - RETVAL = newSV (w * h); - SvPOK_only (RETVAL); - SvCUR_set (RETVAL, w * h); + jerr.err.error_exit = cv_error_exit; + jerr.err.output_message = cv_error_output; - dst = SvPVX (RETVAL); + if ((rs = setjmp (jerr.setjmp_buffer))) + { + fclose (fp); + jpeg_destroy_decompress (&cinfo); - memset (Er, 0, sizeof (int) * IW); - memset (Eg, 0, sizeof (int) * IW); - memset (Eb, 0, sizeof (int) * IW); + if (pb) + g_object_unref ((gpointer)pb); - /* some primitive error distribution + random dithering */ + XSRETURN_UNDEF; + } - for (y = 0; y < h; y++) + jpeg_create_decompress (&cinfo); + + jpeg_stdio_src (&cinfo, fp); + jpeg_read_header (&cinfo, TRUE); + + cinfo.dct_method = JDCT_DEFAULT; + cinfo.do_fancy_upsampling = FALSE; /* worse quality, but nobody compained so far, and gdk-pixbuf does the same */ + cinfo.do_block_smoothing = FALSE; + cinfo.out_color_space = JCS_RGB; + cinfo.quantize_colors = FALSE; + + cinfo.scale_num = 1; + cinfo.scale_denom = 1; + + jpeg_calc_output_dimensions (&cinfo); + + if (thumbnail) { - int er = 0, eg = 0, eb = 0; + cinfo.dct_method = JDCT_FASTEST; + cinfo.do_fancy_upsampling = FALSE; - for (x = 0; x < w; x++) + while (cinfo.scale_denom < 8 + && cinfo.output_width >= IW*4 + && cinfo.output_height >= IH*4) { - int r, g, b; - guchar *p = src + x * bpp + y * sstr; + cinfo.scale_denom <<= 1; + jpeg_calc_output_dimensions (&cinfo); + } + } - r = ((p[0] + er + Er[x]) * 7 + 128) / 255; - g = ((p[1] + eg + Eg[x]) * 7 + 128) / 255; - b = ((p[2] + eb + Eb[x]) * 3 + 128) / 255; - - r = r > 7 ? 7 : r < 0 ? 0 : r; - g = g > 7 ? 7 : g < 0 ? 0 : g; - b = b > 3 ? 3 : b < 0 ? 0 : b; - - er += p[0] - (r * 255 + 4) / 7; - eg += p[1] - (g * 255 + 4) / 7; - eb += p[2] - (b * 255 + 2) / 3; - - Er[x] = er / 2; er -= er / 2 + RAND % 5 - 2; - Eg[x] = eg / 2; eg -= eg / 2 + RAND % 5 - 2; - Eb[x] = eb / 2; eb -= eb / 2 + RAND % 5 - 2; + pb = RETVAL = gdk_pixbuf_new (GDK_COLORSPACE_RGB, 0, 8, cinfo.output_width, cinfo.output_height); + if (!RETVAL) + longjmp (jerr.setjmp_buffer, 2); - *dst++ = r << 5 | g << 2 | b; - } + data = gdk_pixbuf_get_pixels (RETVAL); + rs = gdk_pixbuf_get_rowstride (RETVAL); + + if (cinfo.output_components != 3) + longjmp (jerr.setjmp_buffer, 3); + + jpeg_start_decompress (&cinfo); + + while (cinfo.output_scanline < cinfo.output_height) + { + int remaining = cinfo.output_height - cinfo.output_scanline; + JSAMPROW rp[4]; + + rp [0] = data + cinfo.output_scanline * rs; + rp [1] = (guchar *)rp [0] + rs; + rp [2] = (guchar *)rp [1] + rs; + rp [3] = (guchar *)rp [2] + rs; + + jpeg_read_scanlines (&cinfo, rp, remaining < 4 ? remaining : 4); } + + jpeg_finish_decompress (&cinfo); + fclose (fp); + jpeg_destroy_decompress (&cinfo); } OUTPUT: RETVAL -SV * -make_histogram (SV *ar) - CODE: +void +compare (GdkPixbuf *a, GdkPixbuf *b) + PPCODE: { - int i; - AV *av, *result; + int w = gdk_pixbuf_get_width (a); + int h = gdk_pixbuf_get_height (a); - if (!SvROK (ar) || SvTYPE (SvRV (ar)) != SVt_PVAV) - croak ("Not an array ref as first argument to make_histogram"); + int sa = gdk_pixbuf_get_rowstride (a); + int sb = gdk_pixbuf_get_rowstride (b); - av = (AV *) SvRV (ar); - result = newAV (); + guchar *pa = gdk_pixbuf_get_pixels (a); + guchar *pb = gdk_pixbuf_get_pixels (b); - for (i = 0; i <= av_len (av); ++i) - { - const int HISTSIZE = 64; + int x, y; - int j; - SV *sv = *av_fetch (av, i, 1); - STRLEN len; - char *buf = SvPVbyte (sv, len); + assert (w == gdk_pixbuf_get_width (b)); + assert (h == gdk_pixbuf_get_height (b)); - int tmphist[HISTSIZE]; - float *hist; + assert (gdk_pixbuf_get_n_channels (a) == 3); + assert (gdk_pixbuf_get_n_channels (b) == 3); - SV *histsv = newSV (HISTSIZE * sizeof (float) + 1); - SvPOK_on (histsv); - SvCUR_set (histsv, HISTSIZE * sizeof (float)); - hist = (float *)SvPVX (histsv); + double diff = 0.; + int peak = 0; - Zero (tmphist, sizeof (tmphist), char); + if (w && h) + for (y = 0; y < h; y++) + { + guchar *pa_ = pa + y * sa; + guchar *pb_ = pb + y * sb; + + for (x = 0; x < w; x++) + { + int d; + + d = ((int)*pa_++) - ((int)*pb_++); diff += d*d; peak = MAX (peak, abs (d)); + d = ((int)*pa_++) - ((int)*pb_++); diff += d*d; peak = MAX (peak, abs (d)); + d = ((int)*pa_++) - ((int)*pb_++); diff += d*d; peak = MAX (peak, abs (d)); + } + } - for (j = len; j--; ) + EXTEND (SP, 2); + PUSHs (sv_2mortal (newSVnv (sqrt (diff / (w * h * 3. * 255. * 255.))))); + PUSHs (sv_2mortal (newSVnv (peak / 255.))); +} + +############################################################################# + +MODULE = Gtk2::CV PACKAGE = Gtk2::CV::Schnauzer + +# currently only works for filenames (octet strings) + +SV * +foldcase (SV *pathsv) + PROTOTYPE: $ + CODE: +{ + STRLEN plen; + U8 *path = (U8 *)SvPV (pathsv, plen); + U8 *pend = path + plen; + U8 dst [plen * 6 * 3], *dstp = dst; + + while (path < pend) + { + U8 ch = *path; + + if (ch >= 'a' && ch <= 'z') + *dstp++ = *path++; + else if (ch >= 'A' && ch <= 'Z') + *dstp++ = *path++ + ('a' - 'A'); + else if (ch >= '0' && ch <= '9') { - unsigned int idx - = ((*buf & 0xc0) >> 2) - | ((*buf & 0x18) >> 1) - | (*buf & 0x03); + STRLEN el, nl = 0; + while (*path >= '0' && *path <= '9' && path < pend) + path++, nl++; - ++tmphist[idx]; - ++buf; - } - - for (j = 0; j < HISTSIZE; ++j) - hist[j] = (float)tmphist[j] / (len + 1e-30); + for (el = nl; el < 6; el++) + *dstp++ = '0'; - av_push (result, histsv); + memcpy (dstp, path - nl, nl); + dstp += nl; + } + else + *dstp++ = *path++; +#if 0 + else + { + STRLEN cl; + to_utf8_fold (path, dstp, &cl); + dstp += cl; + path += is_utf8_char (path); + } +#endif } - RETVAL = newRV_noinc ((SV *)result); + RETVAL = newSVpvn ((const char *)dst, dstp - dst); } - OUTPUT: + OUTPUT: RETVAL +GdkPixbuf_noinc * +p7_to_pb (int w, int h, SV *src_sv) + PROTOTYPE: @ + CODE: +{ + int x, y; + guchar *dst, *d; + int dstr; + guchar *src = (guchar *)SvPVbyte_nolen (src_sv); + + RETVAL = gdk_pixbuf_new (GDK_COLORSPACE_RGB, 0, 8, w, h); + dst = gdk_pixbuf_get_pixels (RETVAL); + dstr = gdk_pixbuf_get_rowstride (RETVAL); + for (y = 0; y < h; y++) + for (d = dst + y * dstr, x = 0; x < w; x++) + { + *d++ = (((*src >> 5) & 7) * 255 + 4) / 7; + *d++ = (((*src >> 2) & 7) * 255 + 4) / 7; + *d++ = (((*src >> 0) & 3) * 255 + 2) / 3; + + src++; + } +} + OUTPUT: + RETVAL ############################################################################# @@ -358,7 +588,7 @@ SvPOK_only (RETVAL); SvCUR_set (RETVAL, 6 * 8 * 12 / 8); - dst = SvPVX (RETVAL); + dst = (guchar *)SvPVX (RETVAL); /* some primitive error distribution + random dithering */ @@ -426,5 +656,88 @@ OUTPUT: RETVAL +############################################################################# + +MODULE = Gtk2::CV PACKAGE = Gtk2::CV::Plugin::RCluster + +SV * +extract_features (SV *ar) + CODE: +{ + int i; + AV *av, *result; + + if (!SvROK (ar) || SvTYPE (SvRV (ar)) != SVt_PVAV) + croak ("Not an array ref as first argument to extract_features"); + + av = (AV *) SvRV (ar); + result = newAV (); + + for (i = 0; i <= av_len (av); ++i) + { + SV *sv = *av_fetch (av, i, 1); + SV *histsv = newSV (9 * sizeof (float) + 1); + SvPOK_on (histsv); + SvCUR_set (histsv, 9 * sizeof (float)); + float *hist = (float *)SvPVX (histsv); + + struct feature f_h, f_s, f_v; + feature_init (&f_h); + feature_init (&f_s); + feature_init (&f_v); + + { + STRLEN len; + unsigned char *buf = (unsigned char *)SvPVbyte (sv, len); + while (len >= 3) + { + unsigned int r, g, b, h, s, v; + r = *buf++; g = *buf++; b = *buf++; + rgb_to_hsv (r, g, b, &h, &s, &v); + + feature_update_pass_1 (&f_h, h); + feature_update_pass_1 (&f_s, s); + feature_update_pass_1 (&f_v, v); + + len -= 3; + } + + feature_finish_pass_1 (&f_h); + feature_finish_pass_1 (&f_s); + feature_finish_pass_1 (&f_v); + } + + { + STRLEN len; + unsigned char *buf = (unsigned char *)SvPVbyte (sv, len); + while (len >= 3) + { + unsigned int r, g, b, h, s, v; + r = *buf++; g = *buf++; b = *buf++; + rgb_to_hsv (r, g, b, &h, &s, &v); + + feature_update_pass_2 (&f_h, h); + feature_update_pass_2 (&f_s, s); + feature_update_pass_2 (&f_v, v); + + len -= 3; + } + + feature_finish_pass_2 (&f_h); + feature_finish_pass_2 (&f_s); + feature_finish_pass_2 (&f_v); + } + + hist [0] = f_h.v1 * 2.; hist [1] = f_h.v2 * 2.; hist [2] = f_h.v3 * 2.; + hist [3] = f_s.v1 ; hist [4] = f_s.v2 ; hist [5] = f_s.v3 ; + hist [6] = f_v.v1 * .5; hist [7] = f_v.v2 * .5; hist [8] = f_v.v3 * .5; + + av_push (result, histsv); + } + + RETVAL = newRV_noinc ((SV *)result); +} + OUTPUT: + RETVAL