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

Comparing CV/CV.xs (file contents):
Revision 1.32 by root, Wed Feb 15 08:36:55 2006 UTC vs.
Revision 1.62 by root, Sun Jul 25 11:32:46 2021 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>
7 8
8#include <magic.h> 9#include <magic.h>
9 10
10#include <jpeglib.h> 11#include <jpeglib.h>
12#include <jerror.h>
13
11#include <glib.h> 14#include <glib.h>
12#include <gtk/gtk.h> 15#include <gtk/gtk.h>
16#include <gdk/gdkx.h>
13#include <gdk-pixbuf/gdk-pixbuf.h> 17#include <gdk-pixbuf/gdk-pixbuf.h>
14 18
15#include <gperl.h> 19#include <gperl.h>
16#include <gtk2perl.h> 20#include <gtk2perl.h>
17 21
22#include <assert.h>
23
24#if WEBP
25#include <webp/demux.h>
26#include <webp/decode.h>
27#endif
28
29#if JXL
30#include <jxl/decode.h>
31#include "jxl/thread_parallel_runner.h"
32#endif
33
34#include "perlmulticore.h"
35
18#define IW 80 /* MUST match Schnauer.pm! */ 36#define IW 80 /* MUST match Schnauzer.pm! */
19#define IH 60 /* MUST match Schnauer.pm! */ 37#define IH 60 /* MUST match Schnauzer.pm! */
20
21#define RAND (seed = (seed + 7141) * 54773 % 134456)
22 38
23#define LINELENGTH 240 39#define LINELENGTH 240
24 40
25#define ELLIPSIS "\xe2\x80\xa6" 41#define ELLIPSIS "\xe2\x80\xa6"
42
43typedef char *octet_string;
44
45static magic_t magic_cookie[2]; /* !mime, mime */
26 46
27struct jpg_err_mgr 47struct jpg_err_mgr
28{ 48{
29 struct jpeg_error_mgr err; 49 struct jpeg_error_mgr err;
30 jmp_buf setjmp_buffer; 50 jmp_buf setjmp_buffer;
31}; 51};
32 52
33static void 53static void
34cv_error_exit (j_common_ptr cinfo) 54cv_error_exit (j_common_ptr cinfo)
35{ 55{
56 cinfo->err->output_message (cinfo);
36 longjmp (((struct jpg_err_mgr *)cinfo->err)->setjmp_buffer, 99); 57 longjmp (((struct jpg_err_mgr *)cinfo->err)->setjmp_buffer, 99);
37} 58}
38 59
39static void 60static void
40cv_error_output (j_common_ptr cinfo) 61cv_error_output (j_common_ptr cinfo)
41{ 62{
63 char msg[JMSG_LENGTH_MAX];
64
65 cinfo->err->format_message (cinfo, msg);
66
67 fprintf (stderr, "JPEG decoding error: %s\n", msg);
42 return; 68 return;
43} 69}
44 70
45static void 71static void
46rgb_to_hsv (unsigned int r, unsigned int g, unsigned int b, 72rgb_to_hsv (unsigned int r, unsigned int g, unsigned int b,
67 93
68 *h &= 255; 94 *h &= 255;
69 } 95 }
70} 96}
71 97
98struct feature {
99 float v1, v2, v3; // mean, square, cube
100 int n;
101};
102
103static void
104feature_init (struct feature *f)
105{
106 f->v1 = 0.;
107 f->v2 = 0.;
108 f->v3 = 0.;
109 f->n = 0;
110}
111
112// didn't find an algorithm to neatly do mean, variance and skew in one pass.
113// elmex ist schuld.
114static void
115feature_update_pass_1 (struct feature *f, unsigned int v)
116{
117 f->v1 += v;
118 f->n += 1;
119}
120
121static void
122feature_finish_pass_1 (struct feature *f)
123{
124 if (f->n < 1)
125 return;
126
127 f->v1 /= f->n;
128}
129
130static void
131feature_update_pass_2 (struct feature *f, unsigned int v)
132{
133 float d = v - f->v1;
134
135 f->v2 += d * d;
136 f->v3 += d * d * d;
137}
138
139static void
140feature_finish_pass_2 (struct feature *f)
141{
142 if (f->n < 1)
143 return;
144
145 f->v2 /= f->n;
146 f->v3 /= f->n;
147
148 f->v1 /= 255.;
149 f->v2 /= 255. * 255.; f->v2 = sqrtf (f->v2);
150 f->v3 /= 255. * 255. * 255.; f->v3 = powf (fabsf (f->v3), 1./3.);
151}
152
72static guint32 a85_val; 153static guint32 a85_val;
73static guint a85_cnt; 154static guint a85_cnt;
74static guchar a85_buf[LINELENGTH], *a85_ptr; 155static guchar a85_buf[LINELENGTH], *a85_ptr;
75 156
76static void 157static void
88 if (!--a85_cnt) 169 if (!--a85_cnt)
89 { 170 {
90 a85_cnt = 4; 171 a85_cnt = 4;
91 if (a85_val) 172 if (a85_val)
92 { 173 {
93 a85_ptr[4] = (a85_val % 85) + 33; a85_val /= 85; 174 a85_ptr[4] = (a85_val % 85) + 33; a85_val /= 85;
94 a85_ptr[3] = (a85_val % 85) + 33; a85_val /= 85; 175 a85_ptr[3] = (a85_val % 85) + 33; a85_val /= 85;
95 a85_ptr[2] = (a85_val % 85) + 33; a85_val /= 85; 176 a85_ptr[2] = (a85_val % 85) + 33; a85_val /= 85;
96 a85_ptr[1] = (a85_val % 85) + 33; a85_val /= 85; 177 a85_ptr[1] = (a85_val % 85) + 33; a85_val /= 85;
97 a85_ptr[0] = (a85_val ) + 33; 178 a85_ptr[0] = (a85_val ) + 33;
98 179
99 a85_ptr += 5; 180 a85_ptr += 5;
100 } 181 }
123 204
124 PerlIO_write (fp, a85_buf, a85_ptr - a85_buf); 205 PerlIO_write (fp, a85_buf, a85_ptr - a85_buf);
125} 206}
126 207
127///////////////////////////////////////////////////////////////////////////// 208/////////////////////////////////////////////////////////////////////////////
209// memory source for libjpeg
210
211static void cv_ms_init (j_decompress_ptr cinfo)
212{
213}
214
215static void cv_ms_term (j_decompress_ptr cinfo)
216{
217}
218
219static boolean cv_ms_fill (j_decompress_ptr cinfo)
220{
221 // unexpected EOF, warn and generate fake EOI marker
222
223 WARNMS (cinfo, JWRN_JPEG_EOF);
224
225 struct jpeg_source_mgr *src = (struct jpeg_source_mgr *)cinfo->src;
226
227 static const JOCTET eoi[] = { 0xFF, JPEG_EOI };
228
229 src->next_input_byte = eoi;
230 src->bytes_in_buffer = sizeof (eoi);
231
232 return TRUE;
233}
234
235static void cv_ms_skip (j_decompress_ptr cinfo, long num_bytes)
236{
237 struct jpeg_source_mgr *src = (struct jpeg_source_mgr *)cinfo->src;
238
239 src->next_input_byte += num_bytes;
240 src->bytes_in_buffer -= num_bytes;
241}
242
243static void cv_jpeg_mem_src (j_decompress_ptr cinfo, void *buf, size_t buflen)
244{
245 struct jpeg_source_mgr *src;
246
247 if (!cinfo->src)
248 cinfo->src = (struct jpeg_source_mgr *)
249 (*cinfo->mem->alloc_small) (
250 (j_common_ptr) cinfo, JPOOL_PERMANENT, sizeof (struct jpeg_source_mgr)
251 );
252
253 src = (struct jpeg_source_mgr *)cinfo->src;
254 src->init_source = cv_ms_init;
255 src->fill_input_buffer = cv_ms_fill;
256 src->skip_input_data = cv_ms_skip;
257 src->resync_to_restart = jpeg_resync_to_restart;
258 src->term_source = cv_ms_term;
259 src->next_input_byte = (JOCTET *)buf;
260 src->bytes_in_buffer = buflen;
261}
262
263/////////////////////////////////////////////////////////////////////////////
264
265/* great, the jpeg-xl reference implementaton requires us to parse bmff files */
266
267struct bmff_box
268{
269 char type[4];
270 const uint8_t *ptr;
271 size_t size;
272};
273
274static int
275bmff_parse_box (struct bmff_box *box, const uint8_t **next_in, size_t *avail_in)
276{
277 if (*avail_in < 8)
278 return 0;
279
280 box->size = ((*next_in)[0] << 24)
281 | ((*next_in)[1] << 16)
282 | ((*next_in)[2] << 8)
283 | ((*next_in)[3] );
284
285 if (box->size < 8)
286 return 0;
287
288 if (*avail_in < box->size)
289 return 0;
290
291 memcpy (box->type, *next_in + 4, 4);
292 box->ptr = *next_in + 8;
293
294 *next_in += box->size;
295 *avail_in -= box->size;
296
297 box->size -= 8;
298
299 return 1;
300}
301
302/////////////////////////////////////////////////////////////////////////////
128 303
129MODULE = Gtk2::CV PACKAGE = Gtk2::CV 304MODULE = Gtk2::CV PACKAGE = Gtk2::CV
130 305
131PROTOTYPES: ENABLE 306PROTOTYPES: ENABLE
132 307
308# calculate the common prefix length of two strings
133# missing function in perl. really :) 309# missing function in perl. really :)
134int 310int
135common_prefix_length (a, b) 311common_prefix_length (a, b)
136 unsigned char *a = (unsigned char *)SvPVutf8_nolen ($arg); 312 unsigned char *a = (unsigned char *)SvPVutf8_nolen ($arg);
137 unsigned char *b = (unsigned char *)SvPVutf8_nolen ($arg); 313 unsigned char *b = (unsigned char *)SvPVutf8_nolen ($arg);
146 322
147 OUTPUT: 323 OUTPUT:
148 RETVAL 324 RETVAL
149 325
150const char * 326const char *
151magic (const char *path) 327magic (SV *path_or_data)
328 ALIAS:
329 magic = 0
330 magic_mime = 1
331 magic_buffer = 2
332 magic_buffer_mime = 3
152 CODE: 333 CODE:
153{ 334{
154 static magic_t cookie; 335 STRLEN len;
336 char *data = SvPVbyte (path_or_data, len);
155 337
156 if (!cookie) 338 if (!magic_cookie[0])
339 {
340 magic_cookie[0] = magic_open (MAGIC_SYMLINK);
341 magic_cookie[1] = magic_open (MAGIC_SYMLINK | MAGIC_MIME_TYPE);
342 magic_load (magic_cookie[0], 0);
343 magic_load (magic_cookie[1], 0);
157 { 344 }
158 cookie = magic_open (MAGIC_NONE);
159 345
160 if (cookie) 346 perlinterp_release ();
161 magic_load (cookie, 0);
162 else
163 XSRETURN_UNDEF;
164 }
165 347
166 RETVAL = magic_file (cookie, path); 348 RETVAL = ix & 2
349 ? magic_buffer (magic_cookie[ix & 1], data, len)
350 : magic_file (magic_cookie[ix & 1], data);
351
352 perlinterp_acquire ();
167} 353}
168 OUTPUT: 354 OUTPUT:
169 RETVAL 355 RETVAL
170 356
171const char *
172magic_mime (const char *path)
173 CODE:
174{
175 static magic_t cookie;
176
177 if (!cookie)
178 {
179 cookie = magic_open (MAGIC_MIME);
180
181 if (cookie)
182 magic_load (cookie, 0);
183 else
184 XSRETURN_UNDEF;
185 }
186
187 RETVAL = magic_file (cookie, path);
188}
189 OUTPUT:
190 RETVAL
191
192# missing in Gtk2 perl module 357# missing/broken in Gtk2 perl module
358
359void
360gdk_window_clear_hints (GdkWindow *window)
361 CODE:
362 gdk_window_set_geometry_hints (window, 0, 0);
193 363
194gboolean 364gboolean
195gdk_net_wm_supports (GdkAtom property) 365gdk_net_wm_supports (GdkAtom property)
196 CODE: 366 CODE:
197#if defined(GDK_WINDOWING_X11) && !defined(GDK_MULTIHEAD_SAFE) 367#if defined(GDK_WINDOWING_X11) && !defined(GDK_MULTIHEAD_SAFE)
203 RETVAL 373 RETVAL
204 374
205GdkPixbuf_noinc * 375GdkPixbuf_noinc *
206dealpha_expose (GdkPixbuf *pb) 376dealpha_expose (GdkPixbuf *pb)
207 CODE: 377 CODE:
378 perlinterp_release ();
208{ 379{
209 int w = gdk_pixbuf_get_width (pb); 380 int w = gdk_pixbuf_get_width (pb);
210 int h = gdk_pixbuf_get_height (pb); 381 int h = gdk_pixbuf_get_height (pb);
211 int bpp = gdk_pixbuf_get_n_channels (pb); 382 int bpp = gdk_pixbuf_get_n_channels (pb);
212 int x, y, i; 383 int x, y, i;
221 for (x = 0; x < w; x++) 392 for (x = 0; x < w; x++)
222 for (y = 0; y < h; y++) 393 for (y = 0; y < h; y++)
223 for (i = 0; i < 3; i++) 394 for (i = 0; i < 3; i++)
224 dst[x * 3 + y * dstr + i] = src[x * bpp + y * sstr + i]; 395 dst[x * 3 + y * dstr + i] = src[x * bpp + y * sstr + i];
225} 396}
397 perlinterp_acquire ();
226 OUTPUT: 398 OUTPUT:
227 RETVAL 399 RETVAL
228 400
229GdkPixbuf_noinc * 401GdkPixbuf_noinc *
230rotate (GdkPixbuf *pb, int angle) 402rotate (GdkPixbuf *pb, int angle)
231 CODE: 403 CODE:
404 perlinterp_release ();
405 if (angle < 0)
406 angle += 360;
232 RETVAL = gdk_pixbuf_rotate_simple (pb, angle == 0 ? GDK_PIXBUF_ROTATE_NONE 407 RETVAL = gdk_pixbuf_rotate_simple (pb, angle == 0 ? GDK_PIXBUF_ROTATE_NONE
233 : angle == 90 ? GDK_PIXBUF_ROTATE_COUNTERCLOCKWISE 408 : angle == 90 ? GDK_PIXBUF_ROTATE_COUNTERCLOCKWISE
234 : angle == 180 ? GDK_PIXBUF_ROTATE_UPSIDEDOWN 409 : angle == 180 ? GDK_PIXBUF_ROTATE_UPSIDEDOWN
235 : angle == 270 ? GDK_PIXBUF_ROTATE_CLOCKWISE 410 : angle == 270 ? GDK_PIXBUF_ROTATE_CLOCKWISE
236 : angle); 411 : angle);
412 perlinterp_acquire ();
237 OUTPUT: 413 OUTPUT:
238 RETVAL 414 RETVAL
239 415
416const char *
417filetype (SV *image_data)
418 CODE:
419{
420 STRLEN data_len;
421 U8 *data = SvPVbyte (image_data, data_len);
422 static const unsigned char jxl_header[] = {
423 0, 0, 0, 0x0c, 0x4a, 0x58, 0x4c, 0x20,
424 0x0d, 0xa, 0x87, 0x0a, 0, 0, 0, 0x14,
425 0x66, 0x74, 0x79, 0x70, 0x6a, 0x78, 0x6c, 0x20,
426 0, 0, 0, 0, 0x6a, 0x78, 0x6c, 0x20,
427 };
428
429 if (data_len >= 20
430 && data[0] == 0xff
431 && data[1] == 0xd8
432 && data[2] == 0xff)
433 RETVAL = "image/jpeg";
434 else if (data_len >= 12
435 && data[ 0] == (U8)'R'
436 && data[ 1] == (U8)'I'
437 && data[ 2] == (U8)'F'
438 && data[ 3] == (U8)'F'
439 && data[ 8] == (U8)'W'
440 && data[ 9] == (U8)'E'
441 && data[10] == (U8)'B'
442 && data[11] == (U8)'P')
443 RETVAL = "image/webp";
444 else if (data_len >= 16
445 && data[ 0] == 0x89
446 && data[ 1] == (U8)'P'
447 && data[ 2] == (U8)'N'
448 && data[ 3] == (U8)'G'
449 && data[ 4] == 0x0d
450 && data[ 5] == 0x0a
451 && data[ 6] == 0x1a
452 && data[ 7] == 0x0a)
453 RETVAL = "image/png";
454 else if (data_len >= sizeof (jxl_header) && memcmp (data, jxl_header, sizeof (jxl_header)) == 0)
455 RETVAL = "image/jxl";
456 else if (data_len >= 13
457 && data[0] == 'G'
458 && data[1] == 'I'
459 && data[2] == 'F'
460 && data[3] == '8'
461 //&& (data[4] == '7' || data[4] == '9')
462 && data[5] == 'a')
463 {
464 RETVAL = "image/gif";
465
466 // now see if its animated - we require the netscape application header for this
467 int ofs = 13;
468
469 if (data[10] & 0x80)
470 ofs += (1 << ((data[10] & 7) + 1)) * 3;
471
472 if (data_len >= ofs + 2 + 1 + 11)
473 {
474
475 // skip a graphic control extension block. we assume
476 // there is at most one such block - while the NAB
477 // has to come firstz, some files do not obey this
478 if (data[ofs] == 0x21 && data[ofs + 1] == 0xf9)
479 ofs += 3 + data[ofs + 2] + 1;
480
481 if (data_len >= ofs + 2 + 1 + 11)
482 if (!memcmp (data + ofs, "\x21\xff\x0bNETSCAPE2.0", sizeof ("\x21\xff\x0bNETSCAPE2.0") - 1))
483 RETVAL = "video/gif";
484 }
485 }
486
487 else
488 XSRETURN_UNDEF;
489}
490 OUTPUT:
491 RETVAL
492
240GdkPixbuf_noinc * 493GdkPixbuf_noinc *
241load_jpeg (SV *path, int thumbnail=0) 494decode_webp (SV *image_data, int thumbnail = 0, int iw = 0, int ih = 0)
495 CODE:
496{
497#if WEBP
498 STRLEN data_size;
499 int alpha;
500 WebPData data;
501 WebPDemuxer *demux;
502 WebPIterator iter;
503 WebPDecoderConfig config;
504 int inw, inh;
505
506 data.bytes = (uint8_t *)SvPVbyte (image_data, data_size);
507 data.size = data_size;
508
509 perlinterp_release ();
510
511 RETVAL = 0;
512
513 if (!(demux = WebPDemux (&data)))
514 goto err_demux;
515
516 if (!WebPDemuxGetFrame (demux, 1, &iter))
517 goto err_iter;
518
519 if (!WebPInitDecoderConfig (&config))
520 goto err_iter;
521
522 config.options.use_threads = 1;
523
524 if (WebPGetFeatures (iter.fragment.bytes, iter.fragment.size, &config.input) != VP8_STATUS_OK)
525 goto err_iter;
526
527 inw = config.input.width;
528 inh = config.input.height;
529
530 if (thumbnail)
531 {
532 if (inw * ih > inh * iw)
533 ih = (iw * inh + inw - 1) / inw;
534 else
535 iw = (ih * inw + inh - 1) / inh;
536
537 config.options.bypass_filtering = 1;
538 config.options.no_fancy_upsampling = 1;
539
540 config.options.use_scaling = 1;
541 config.options.scaled_width = iw;
542 config.options.scaled_height = ih;
543 }
544 else
545 {
546 iw = inw;
547 ih = inh;
548 }
549
550 alpha = !!config.input.has_alpha;
551
552 RETVAL = gdk_pixbuf_new (GDK_COLORSPACE_RGB, alpha, 8, iw, ih);
553 if (!RETVAL)
554 goto err_iter;
555
556 config.output.colorspace = alpha ? MODE_RGBA : MODE_RGB;
557 config.output.u.RGBA.rgba = gdk_pixbuf_get_pixels (RETVAL);
558 config.output.u.RGBA.stride = gdk_pixbuf_get_rowstride (RETVAL);
559 config.output.u.RGBA.size = gdk_pixbuf_get_byte_length (RETVAL);
560 config.output.is_external_memory = 1;
561
562 if (WebPDecode (iter.fragment.bytes, iter.fragment.size, &config) != VP8_STATUS_OK)
563 {
564 g_object_unref (RETVAL);
565 RETVAL = 0;
566 goto err_iter;
567 }
568
569 err_iter:
570 WebPDemuxReleaseIterator (&iter);
571 err_demux:
572 WebPDemuxDelete (demux);
573
574 perlinterp_acquire ();
575#else
576 croak ("load_webp: webp not enabled at compile time");
577#endif
578}
579 OUTPUT:
580 RETVAL
581
582GdkPixbuf_noinc *
583decode_jxl (SV *image_data, int thumbnail = 0, int iw = 0, int ih = 0)
584 CODE:
585{
586#if JXL
587 JxlDecoder *dec = JxlDecoderCreate (0);
588 JxlBasicInfo info;
589 const uint8_t *next_in = (uint8_t *)SvPVbyte_nolen (image_data);
590 size_t avail_in = SvCUR (image_data);
591 const char *error = 0;
592 void *runner = 0;
593 struct bmff_box box;
594
595 perlinterp_release ();
596
597 RETVAL = 0;
598
599 error = "JxlDecoderCreate failed";
600 if (!dec)
601 goto done;
602
603 runner = JxlThreadParallelRunnerCreate (0, JxlThreadParallelRunnerDefaultNumWorkerThreads ());
604
605 error = "JxlDecoderSetParallelRunner failed";
606 if (JxlDecoderSetParallelRunner (dec, JxlThreadParallelRunner, runner) != JXL_DEC_SUCCESS)
607 goto done;
608
609 error = "JxlDecoderSubscribeEvents failed";
610 if (JxlDecoderSubscribeEvents (dec, JXL_DEC_FULL_IMAGE | JXL_DEC_BASIC_INFO) != JXL_DEC_SUCCESS)
611 goto done;
612
613 error = "JxlDecoderSetInput failed";
614 if (JxlDecoderSetInput (dec, next_in, avail_in) != JXL_DEC_SUCCESS)
615 goto done;
616
617 for (;;)
618 {
619 JxlDecoderStatus status = JxlDecoderProcessInput (dec);
620
621 printf ("status %d\n",status);
622
623 switch (status)
624 {
625 case JXL_DEC_ERROR:
626 error = "JxlDecoderProcessInput failed";
627 goto done;
628
629 case JXL_DEC_NEED_MORE_INPUT:
630 error = "incomplete file";
631 goto done;
632
633 case JXL_DEC_SUCCESS:
634 goto done;
635
636 case JXL_DEC_NEED_IMAGE_OUT_BUFFER:
637 {
638 error = "JxlDecoderGetBasicInfo failed";
639 if (JxlDecoderGetBasicInfo (dec, &info) != JXL_DEC_SUCCESS)
640 goto done;
641
642 RETVAL = gdk_pixbuf_new (GDK_COLORSPACE_RGB, !!info.alpha_bits, 8, info.xsize, info.ysize);
643 error = "unable to allocate pixbuf";
644 if (!RETVAL)
645 goto done;
646
647 JxlPixelFormat format = {
648 info.alpha_bits ? 4 : 3,
649 JXL_TYPE_UINT8,
650 JXL_NATIVE_ENDIAN,
651 gdk_pixbuf_get_rowstride (RETVAL)
652 };
653
654 error = "JxlDecoderSetImageOutBuffer failed";
655 if (JxlDecoderSetImageOutBuffer (
656 dec,
657 &format,
658 gdk_pixbuf_get_pixels (RETVAL),
659 gdk_pixbuf_get_byte_length (RETVAL)
660 ) != JXL_DEC_SUCCESS)
661 goto done;
662 }
663 break;
664
665 default:
666 error = "unexpected event";
667 goto done;
668 }
669 }
670
671 done:
672 if (dec)
673 JxlDecoderDestroy (dec);
674
675 if (runner)
676 JxlThreadParallelRunnerDestroy (runner);
677
678 perlinterp_acquire ();
679
680 if (error)
681 {
682 if (RETVAL)
683 g_object_unref (RETVAL);
684
685 croak ("load_jxl: %s", error);
686 }
687#else
688 croak ("load_jxl: jpeg-xl not enabled at compile time");
689#endif
690}
691 OUTPUT:
692 RETVAL
693
694GdkPixbuf_noinc *
695decode_jpeg (SV *image_data, int thumbnail = 0, int iw = 0, int ih = 0)
242 CODE: 696 CODE:
243{ 697{
244 struct jpeg_decompress_struct cinfo; 698 struct jpeg_decompress_struct cinfo;
245 struct jpg_err_mgr jerr; 699 struct jpg_err_mgr jerr;
246 guchar *data;
247 int rs; 700 int rs;
248 FILE *fp;
249 volatile GdkPixbuf *pb = 0; 701 volatile GdkPixbuf *pb = 0;
702 STRLEN data_len;
703 guchar *data = SvPVbyte (image_data, data_len);
250 704
251 RETVAL = 0; 705 RETVAL = 0;
252 706
253 fp = fopen (SvPVbyte_nolen (path), "rb"); 707 perlinterp_release ();
254
255 if (!fp)
256 XSRETURN_UNDEF;
257 708
258 cinfo.err = jpeg_std_error (&jerr.err); 709 cinfo.err = jpeg_std_error (&jerr.err);
259 710
260 jerr.err.error_exit = cv_error_exit; 711 jerr.err.error_exit = cv_error_exit;
261 jerr.err.output_message = cv_error_output; 712 jerr.err.output_message = cv_error_output;
262 713
263 if ((rs = setjmp (jerr.setjmp_buffer))) 714 if ((rs = setjmp (jerr.setjmp_buffer)))
264 { 715 {
265 fclose (fp);
266 jpeg_destroy_decompress (&cinfo); 716 jpeg_destroy_decompress (&cinfo);
267 717
268 if (pb) 718 if (pb)
269 g_object_unref ((gpointer)pb); 719 g_object_unref ((gpointer)pb);
270 720
721 perlinterp_acquire ();
271 XSRETURN_UNDEF; 722 XSRETURN_UNDEF;
272 } 723 }
273 724
725 if (!data_len)
726 longjmp (jerr.setjmp_buffer, 4);
727
274 jpeg_create_decompress (&cinfo); 728 jpeg_create_decompress (&cinfo);
729 cv_jpeg_mem_src (&cinfo, data, data_len);
275 730
276 jpeg_stdio_src (&cinfo, fp);
277 jpeg_read_header (&cinfo, TRUE); 731 jpeg_read_header (&cinfo, TRUE);
278 732
279 cinfo.dct_method = JDCT_DEFAULT; 733 cinfo.dct_method = JDCT_DEFAULT;
280 cinfo.do_fancy_upsampling = FALSE; /* worse quality, but nobody compained so far, and gdk-pixbuf does the same */ 734 cinfo.do_fancy_upsampling = FALSE; /* worse quality, but nobody compained so far, and gdk-pixbuf does the same */
281 cinfo.do_block_smoothing = FALSE; 735 cinfo.do_block_smoothing = FALSE;
291 { 745 {
292 cinfo.dct_method = JDCT_FASTEST; 746 cinfo.dct_method = JDCT_FASTEST;
293 cinfo.do_fancy_upsampling = FALSE; 747 cinfo.do_fancy_upsampling = FALSE;
294 748
295 while (cinfo.scale_denom < 8 749 while (cinfo.scale_denom < 8
296 && cinfo.output_width >= IW*4 750 && cinfo.output_width >= iw*4
297 && cinfo.output_height >= IH*4) 751 && cinfo.output_height >= ih*4)
298 { 752 {
299 cinfo.scale_denom <<= 1; 753 cinfo.scale_denom <<= 1;
300 jpeg_calc_output_dimensions (&cinfo); 754 jpeg_calc_output_dimensions (&cinfo);
301 } 755 }
302 } 756 }
303 757
758 if (cinfo.output_components != 3)
759 longjmp (jerr.setjmp_buffer, 3);
760
761 if (cinfo.jpeg_color_space == JCS_YCCK || cinfo.jpeg_color_space == JCS_CMYK)
762 {
763 cinfo.out_color_space = JCS_CMYK;
764 cinfo.output_components = 4;
765 }
766
304 pb = RETVAL = gdk_pixbuf_new (GDK_COLORSPACE_RGB, 0, 8, cinfo.output_width, cinfo.output_height); 767 pb = RETVAL = gdk_pixbuf_new (GDK_COLORSPACE_RGB, cinfo.output_components == 4, 8, cinfo.output_width, cinfo.output_height);
305 if (!RETVAL) 768 if (!RETVAL)
306 longjmp (jerr.setjmp_buffer, 2); 769 longjmp (jerr.setjmp_buffer, 2);
307 770
308 data = gdk_pixbuf_get_pixels (RETVAL); 771 data = gdk_pixbuf_get_pixels (RETVAL);
309 rs = gdk_pixbuf_get_rowstride (RETVAL); 772 rs = gdk_pixbuf_get_rowstride (RETVAL);
310
311 if (cinfo.output_components != 3)
312 longjmp (jerr.setjmp_buffer, 3);
313 773
314 jpeg_start_decompress (&cinfo); 774 jpeg_start_decompress (&cinfo);
315 775
316 while (cinfo.output_scanline < cinfo.output_height) 776 while (cinfo.output_scanline < cinfo.output_height)
317 { 777 {
324 rp [3] = (guchar *)rp [2] + rs; 784 rp [3] = (guchar *)rp [2] + rs;
325 785
326 jpeg_read_scanlines (&cinfo, rp, remaining < 4 ? remaining : 4); 786 jpeg_read_scanlines (&cinfo, rp, remaining < 4 ? remaining : 4);
327 } 787 }
328 788
789 if (cinfo.out_color_space == JCS_CMYK)
790 {
791 guchar *end = data + cinfo.output_height * rs;
792
793 while (data < end)
794 {
795 U32 c = data [0];
796 U32 m = data [1];
797 U32 y = data [2];
798 U32 k = data [3];
799
800 if (0)
801 if (cinfo.Adobe_transform == 2)
802 {
803 c ^= 0xff;
804 m ^= 0xff;
805 y ^= 0xff;
806 k ^= 0xff;
807 }
808
809 data [0] = (c * k + 0x80) / 0xff;
810 data [1] = (m * k + 0x80) / 0xff;
811 data [2] = (y * k + 0x80) / 0xff;
812 data [3] = 0xff;
813
814 data += 4;
815 }
816 }
817
329 jpeg_finish_decompress (&cinfo); 818 jpeg_finish_decompress (&cinfo);
330 fclose (fp);
331 jpeg_destroy_decompress (&cinfo); 819 jpeg_destroy_decompress (&cinfo);
820 perlinterp_acquire ();
332} 821}
333 OUTPUT: 822 OUTPUT:
334 RETVAL 823 RETVAL
335 824
825void
826compare (GdkPixbuf *a, GdkPixbuf *b)
827 PPCODE:
828 perlinterp_release ();
829{
830 int w = gdk_pixbuf_get_width (a);
831 int h = gdk_pixbuf_get_height (a);
832
833 int sa = gdk_pixbuf_get_rowstride (a);
834 int sb = gdk_pixbuf_get_rowstride (b);
835
836 guchar *pa = gdk_pixbuf_get_pixels (a);
837 guchar *pb = gdk_pixbuf_get_pixels (b);
838
839 int x, y;
840
841 assert (w == gdk_pixbuf_get_width (b));
842 assert (h == gdk_pixbuf_get_height (b));
843
844 assert (gdk_pixbuf_get_n_channels (a) == 3);
845 assert (gdk_pixbuf_get_n_channels (b) == 3);
846
847 double diff = 0.;
848 int peak = 0;
849
850 if (w && h)
851 for (y = 0; y < h; y++)
852 {
853 guchar *pa_ = pa + y * sa;
854 guchar *pb_ = pb + y * sb;
855
856 for (x = 0; x < w; x++)
857 {
858 int d;
859
860 d = ((int)*pa_++) - ((int)*pb_++); diff += d*d; peak = MAX (peak, abs (d));
861 d = ((int)*pa_++) - ((int)*pb_++); diff += d*d; peak = MAX (peak, abs (d));
862 d = ((int)*pa_++) - ((int)*pb_++); diff += d*d; peak = MAX (peak, abs (d));
863 }
864 }
865
866 perlinterp_acquire ();
867
868 EXTEND (SP, 2);
869 PUSHs (sv_2mortal (newSVnv (sqrt (diff / (w * h * 3. * 255. * 255.)))));
870 PUSHs (sv_2mortal (newSVnv (peak / 255.)));
871}
872
336############################################################################# 873#############################################################################
337 874
338MODULE = Gtk2::CV PACKAGE = Gtk2::CV::Schnauzer 875MODULE = Gtk2::CV PACKAGE = Gtk2::CV::Schnauzer
876
877# currently only works for filenames (octet strings)
339 878
340SV * 879SV *
341foldcase (SV *pathsv) 880foldcase (SV *pathsv)
342 PROTOTYPE: $ 881 PROTOTYPE: $
343 CODE: 882 CODE:
344{ 883{
345 STRLEN plen; 884 STRLEN plen;
346 U8 *path = (U8 *)SvPVutf8 (pathsv, plen); 885 U8 *path = (U8 *)SvPV (pathsv, plen);
347 U8 *pend = path + plen; 886 U8 *pend = path + plen;
348 U8 dst [plen * 6 * 3], *dstp = dst; 887 U8 dst [plen * 8 * 3], *dstp = dst;
349 888
350 while (path < pend) 889 while (path < pend)
351 { 890 {
352 U8 ch = *path; 891 U8 ch = *path;
353 892
354 if (ch >= 'a' && ch <= 'z') 893 if (ch >= 'a' && ch <= 'z')
355 *dstp++ = *path++; 894 *dstp++ = *path++;
895 else if (ch >= 'A' && ch <= 'Z')
896 *dstp++ = *path++ + ('a' - 'A');
356 else if (ch >= '0' && ch <= '9') 897 else if (ch >= '0' && ch <= '9')
357 { 898 {
899 /* version sort, up to 8 digits */
358 STRLEN el, nl = 0; 900 STRLEN el, nl = 0;
359 while (*path >= '0' && *path <= '9' && path < pend) 901 while (*path >= '0' && *path <= '9' && path < pend)
360 path++, nl++; 902 path++, nl++;
361 903
362 for (el = nl; el < 6; el++) 904 for (el = nl; el < 8; el++)
363 *dstp++ = '0'; 905 *dstp++ = '0';
364 906
365 memcpy (dstp, path - nl, nl); 907 memcpy (dstp, path - nl, nl);
366 dstp += nl; 908 dstp += nl;
367 } 909 }
910 else
911 *dstp++ = *path++;
912#if 0
368 else 913 else
369 { 914 {
370 STRLEN cl; 915 STRLEN cl;
371 to_utf8_fold (path, dstp, &cl); 916 to_utf8_fold (path, dstp, &cl);
372 dstp += cl; 917 dstp += cl;
373 path += is_utf8_char (path); 918 path += is_utf8_char (path);
374 } 919 }
920#endif
375 } 921 }
376 922
377 RETVAL = newSVpvn ((const char *)dst, dstp - dst); 923 RETVAL = newSVpvn ((const char *)dst, dstp - dst);
378} 924}
379 OUTPUT: 925 OUTPUT:
541############################################################################# 1087#############################################################################
542 1088
543MODULE = Gtk2::CV PACKAGE = Gtk2::CV::Plugin::RCluster 1089MODULE = Gtk2::CV PACKAGE = Gtk2::CV::Plugin::RCluster
544 1090
545SV * 1091SV *
546make_histograms (SV *ar) 1092extract_features (SV *ar)
547 CODE: 1093 CODE:
548{ 1094{
549 int i; 1095 int i;
550 AV *av, *result; 1096 AV *av, *result;
551 1097
552 if (!SvROK (ar) || SvTYPE (SvRV (ar)) != SVt_PVAV) 1098 if (!SvROK (ar) || SvTYPE (SvRV (ar)) != SVt_PVAV)
553 croak ("Not an array ref as first argument to make_histogram"); 1099 croak ("Not an array ref as first argument to extract_features");
554 1100
555 av = (AV *) SvRV (ar); 1101 av = (AV *) SvRV (ar);
556 result = newAV (); 1102 result = newAV ();
557 1103
558 for (i = 0; i <= av_len (av); ++i) 1104 for (i = 0; i <= av_len (av); ++i)
559 { 1105 {
560 const int HISTSIZE = 64;
561
562 int j;
563 SV *sv = *av_fetch (av, i, 1); 1106 SV *sv = *av_fetch (av, i, 1);
564 STRLEN len;
565 char *buf = SvPVbyte (sv, len);
566
567 int tmphist[HISTSIZE];
568 float *hist;
569
570 SV *histsv = newSV (HISTSIZE * sizeof (float) + 1); 1107 SV *histsv = newSV (9 * sizeof (float) + 1);
1108
571 SvPOK_on (histsv); 1109 SvPOK_on (histsv);
572 SvCUR_set (histsv, HISTSIZE * sizeof (float)); 1110 SvCUR_set (histsv, 9 * sizeof (float));
573 hist = (float *)SvPVX (histsv); 1111 float *hist = (float *)SvPVX (histsv);
574 1112
575 Zero (tmphist, sizeof (tmphist), char); 1113 struct feature f_h, f_s, f_v;
1114 feature_init (&f_h);
1115 feature_init (&f_s);
1116 feature_init (&f_v);
576 1117
577 for (j = len; j--; )
578 {
579 unsigned int idx
580 = ((*buf & 0xc0) >> 2)
581 | ((*buf & 0x18) >> 1)
582 | (*buf & 0x03);
583
584 ++tmphist[idx];
585 ++buf;
586 }
587 1118 {
588 for (j = 0; j < HISTSIZE; ++j) 1119 STRLEN len;
589 hist[j] = (float)tmphist[j] / (len + 1e-30); 1120 unsigned char *buf = (unsigned char *)SvPVbyte (sv, len);
1121 while (len >= 3)
1122 {
1123 unsigned int r, g, b, h, s, v;
1124 r = *buf++; g = *buf++; b = *buf++;
1125 rgb_to_hsv (r, g, b, &h, &s, &v);
1126
1127 feature_update_pass_1 (&f_h, h);
1128 feature_update_pass_1 (&f_s, s);
1129 feature_update_pass_1 (&f_v, v);
1130
1131 len -= 3;
1132 }
1133
1134 feature_finish_pass_1 (&f_h);
1135 feature_finish_pass_1 (&f_s);
1136 feature_finish_pass_1 (&f_v);
1137 }
1138
1139 {
1140 STRLEN len;
1141 unsigned char *buf = (unsigned char *)SvPVbyte (sv, len);
1142 while (len >= 3)
1143 {
1144 unsigned int r, g, b, h, s, v;
1145 r = *buf++; g = *buf++; b = *buf++;
1146 rgb_to_hsv (r, g, b, &h, &s, &v);
1147
1148 feature_update_pass_2 (&f_h, h);
1149 feature_update_pass_2 (&f_s, s);
1150 feature_update_pass_2 (&f_v, v);
1151
1152 len -= 3;
1153 }
1154
1155 feature_finish_pass_2 (&f_h);
1156 feature_finish_pass_2 (&f_s);
1157 feature_finish_pass_2 (&f_v);
1158 }
1159
1160 hist [0] = f_h.v1 * 2.; hist [1] = f_h.v2 * 2.; hist [2] = f_h.v3 * 2.;
1161 hist [3] = f_s.v1 ; hist [4] = f_s.v2 ; hist [5] = f_s.v3 ;
1162 hist [6] = f_v.v1 * .5; hist [7] = f_v.v2 * .5; hist [8] = f_v.v3 * .5;
590 1163
591 av_push (result, histsv); 1164 av_push (result, histsv);
592 } 1165 }
593 1166
594 RETVAL = newRV_noinc ((SV *)result); 1167 RETVAL = newRV_noinc ((SV *)result);
595} 1168}
596 OUTPUT: 1169 OUTPUT:
597 RETVAL 1170 RETVAL
598 1171
599

Diff Legend

Removed lines
+ Added lines
< Changed lines
> Changed lines