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

Comparing CV/CV.xs (file contents):
Revision 1.31 by root, Wed Feb 15 08:08:37 2006 UTC vs.
Revision 1.65 by root, Thu Dec 2 03:11:45 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_MIME);
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
171# 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);
172 363
173gboolean 364gboolean
174gdk_net_wm_supports (GdkAtom property) 365gdk_net_wm_supports (GdkAtom property)
175 CODE: 366 CODE:
176#if defined(GDK_WINDOWING_X11) && !defined(GDK_MULTIHEAD_SAFE) 367#if defined(GDK_WINDOWING_X11) && !defined(GDK_MULTIHEAD_SAFE)
182 RETVAL 373 RETVAL
183 374
184GdkPixbuf_noinc * 375GdkPixbuf_noinc *
185dealpha_expose (GdkPixbuf *pb) 376dealpha_expose (GdkPixbuf *pb)
186 CODE: 377 CODE:
378 perlinterp_release ();
187{ 379{
188 int w = gdk_pixbuf_get_width (pb); 380 int w = gdk_pixbuf_get_width (pb);
189 int h = gdk_pixbuf_get_height (pb); 381 int h = gdk_pixbuf_get_height (pb);
190 int bpp = gdk_pixbuf_get_n_channels (pb); 382 int bpp = gdk_pixbuf_get_n_channels (pb);
191 int x, y, i; 383 int x, y, i;
200 for (x = 0; x < w; x++) 392 for (x = 0; x < w; x++)
201 for (y = 0; y < h; y++) 393 for (y = 0; y < h; y++)
202 for (i = 0; i < 3; i++) 394 for (i = 0; i < 3; i++)
203 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];
204} 396}
397 perlinterp_acquire ();
205 OUTPUT: 398 OUTPUT:
206 RETVAL 399 RETVAL
207 400
208GdkPixbuf_noinc * 401GdkPixbuf_noinc *
209rotate (GdkPixbuf *pb, int angle) 402rotate (GdkPixbuf *pb, int angle)
210 CODE: 403 CODE:
404 perlinterp_release ();
405 if (angle < 0)
406 angle += 360;
211 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
212 : angle == 90 ? GDK_PIXBUF_ROTATE_COUNTERCLOCKWISE 408 : angle == 90 ? GDK_PIXBUF_ROTATE_COUNTERCLOCKWISE
213 : angle == 180 ? GDK_PIXBUF_ROTATE_UPSIDEDOWN 409 : angle == 180 ? GDK_PIXBUF_ROTATE_UPSIDEDOWN
214 : angle == 270 ? GDK_PIXBUF_ROTATE_CLOCKWISE 410 : angle == 270 ? GDK_PIXBUF_ROTATE_CLOCKWISE
215 : angle); 411 : angle);
412 perlinterp_acquire ();
216 OUTPUT: 413 OUTPUT:
217 RETVAL 414 RETVAL
218 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, 0x0d, 0xa, 0x87, 0x0a
424 };
425
426 if (data_len >= 20
427 && data[0] == 0xff
428 && data[1] == 0xd8
429 && data[2] == 0xff)
430 RETVAL = "image/jpeg";
431 else if (data_len >= 12
432 && data[ 0] == (U8)'R'
433 && data[ 1] == (U8)'I'
434 && data[ 2] == (U8)'F'
435 && data[ 3] == (U8)'F'
436 && data[ 8] == (U8)'W'
437 && data[ 9] == (U8)'E'
438 && data[10] == (U8)'B'
439 && data[11] == (U8)'P')
440 RETVAL = "image/webp";
441 else if (data_len >= 16
442 && data[ 0] == 0x89
443 && data[ 1] == (U8)'P'
444 && data[ 2] == (U8)'N'
445 && data[ 3] == (U8)'G'
446 && data[ 4] == 0x0d
447 && data[ 5] == 0x0a
448 && data[ 6] == 0x1a
449 && data[ 7] == 0x0a)
450 RETVAL = "image/png";
451 else if (data_len >= sizeof (jxl_header) && memcmp (data, jxl_header, sizeof (jxl_header)) == 0)
452 RETVAL = "image/jxl"; // todo: might want to use JxlSignatureCheck
453 else if (data_len >= 2
454 && data[0] == 0xff
455 && data[1] == 0x0a)
456 RETVAL = "image/jxl";
457 else if (data_len >= 13
458 && data[0] == 'G'
459 && data[1] == 'I'
460 && data[2] == 'F'
461 && data[3] == '8'
462 //&& (data[4] == '7' || data[4] == '9')
463 && data[5] == 'a')
464 {
465 RETVAL = "image/gif";
466
467 // now see if its animated - we require the netscape application header for this
468 int ofs = 13;
469
470 if (data[10] & 0x80)
471 ofs += (1 << ((data[10] & 7) + 1)) * 3;
472
473 if (data_len >= ofs + 2 + 1 + 11)
474 {
475
476 // skip a graphic control extension block. we assume
477 // there is at most one such block - while the NAB
478 // has to come firstz, some files do not obey this
479 if (data[ofs] == 0x21 && data[ofs + 1] == 0xf9)
480 ofs += 3 + data[ofs + 2] + 1;
481
482 if (data_len >= ofs + 2 + 1 + 11)
483 if (!memcmp (data + ofs, "\x21\xff\x0bNETSCAPE2.0", sizeof ("\x21\xff\x0bNETSCAPE2.0") - 1))
484 RETVAL = "video/gif";
485 }
486 }
487
488 else
489 XSRETURN_UNDEF;
490}
491 OUTPUT:
492 RETVAL
493
219GdkPixbuf_noinc * 494GdkPixbuf_noinc *
220load_jpeg (SV *path, int thumbnail=0) 495decode_webp (SV *image_data, int thumbnail = 0, int iw = 0, int ih = 0)
496 CODE:
497{
498#if WEBP
499 STRLEN data_size;
500 int alpha;
501 WebPData data;
502 WebPDemuxer *demux;
503 WebPIterator iter;
504 WebPDecoderConfig config;
505 int inw, inh;
506
507 data.bytes = (uint8_t *)SvPVbyte (image_data, data_size);
508 data.size = data_size;
509
510 perlinterp_release ();
511
512 RETVAL = 0;
513
514 if (!(demux = WebPDemux (&data)))
515 goto err_demux;
516
517 if (!WebPDemuxGetFrame (demux, 1, &iter))
518 goto err_iter;
519
520 if (!WebPInitDecoderConfig (&config))
521 goto err_iter;
522
523 config.options.use_threads = 1;
524
525 if (WebPGetFeatures (iter.fragment.bytes, iter.fragment.size, &config.input) != VP8_STATUS_OK)
526 goto err_iter;
527
528 inw = config.input.width;
529 inh = config.input.height;
530
531 if (thumbnail)
532 {
533 if (inw * ih > inh * iw)
534 ih = (iw * inh + inw - 1) / inw;
535 else
536 iw = (ih * inw + inh - 1) / inh;
537
538 config.options.bypass_filtering = 1;
539 config.options.no_fancy_upsampling = 1;
540
541 config.options.use_scaling = 1;
542 config.options.scaled_width = iw;
543 config.options.scaled_height = ih;
544 }
545 else
546 {
547 iw = inw;
548 ih = inh;
549 }
550
551 alpha = !!config.input.has_alpha;
552
553 RETVAL = gdk_pixbuf_new (GDK_COLORSPACE_RGB, alpha, 8, iw, ih);
554 if (!RETVAL)
555 goto err_iter;
556
557 config.output.colorspace = alpha ? MODE_RGBA : MODE_RGB;
558 config.output.u.RGBA.rgba = gdk_pixbuf_get_pixels (RETVAL);
559 config.output.u.RGBA.stride = gdk_pixbuf_get_rowstride (RETVAL);
560 config.output.u.RGBA.size = gdk_pixbuf_get_byte_length (RETVAL);
561 config.output.is_external_memory = 1;
562
563 if (WebPDecode (iter.fragment.bytes, iter.fragment.size, &config) != VP8_STATUS_OK)
564 {
565 g_object_unref (RETVAL);
566 RETVAL = 0;
567 goto err_iter;
568 }
569
570 err_iter:
571 WebPDemuxReleaseIterator (&iter);
572 err_demux:
573 WebPDemuxDelete (demux);
574
575 perlinterp_acquire ();
576#else
577 croak ("load_webp: webp not enabled at compile time");
578#endif
579}
580 OUTPUT:
581 RETVAL
582
583GdkPixbuf_noinc *
584decode_jxl (SV *image_data, int thumbnail = 0, int iw = 0, int ih = 0)
585 CODE:
586{
587#if JXL
588 JxlDecoder *dec;
589 JxlBasicInfo info;
590 const uint8_t *next_in = (uint8_t *)SvPVbyte_nolen (image_data);
591 size_t avail_in = SvCUR (image_data);
592 const char *error = 0;
593 JxlDecoderStatus status;
594 static void *runner_cache;
595 void *runner = 0;
596
597 RETVAL = 0;
598
599 if (runner_cache)
600 runner = runner_cache;
601 else
602 runner = JxlThreadParallelRunnerCreate (0, JxlThreadParallelRunnerDefaultNumWorkerThreads ());
603
604 runner_cache = 0;
605
606 perlinterp_release ();
607
608 dec = JxlDecoderCreate (0);
609
610 error = "JxlDecoderCreate failed";
611 if (!dec)
612 goto done;
613
614 status = JxlDecoderSetParallelRunner (dec, JxlThreadParallelRunner, runner);
615 error = "JxlDecoderSetParallelRunner failed";
616 if (status != JXL_DEC_SUCCESS)
617 goto done;
618
619 error = "JxlDecoderSubscribeEvents failed";
620 status = JxlDecoderSubscribeEvents (dec, JXL_DEC_BASIC_INFO | JXL_DEC_FULL_IMAGE);
621 if (status != JXL_DEC_SUCCESS)
622 goto done;
623
624 status = JxlDecoderSetInput (dec, next_in, avail_in);
625 error = "JxlDecoderSetInput failed";
626 if (status != JXL_DEC_SUCCESS)
627 goto done;
628
629 for (;;)
630 {
631 status = JxlDecoderProcessInput (dec);
632
633 switch (status)
634 {
635 case JXL_DEC_FULL_IMAGE:
636 error = 0;
637 goto done;
638
639 case JXL_DEC_ERROR:
640 error = "JxlDecoderProcessInput failed";
641 goto done;
642
643 case JXL_DEC_NEED_MORE_INPUT:
644 error = "incomplete file";
645 goto done;
646
647 case JXL_DEC_SUCCESS:
648 error = "incomplete decode";
649 goto done;
650
651 case JXL_DEC_BASIC_INFO:
652 {
653 status = JxlDecoderGetBasicInfo (dec, &info);
654 error = "JxlDecoderGetBasicInfo failed";
655 if (status != JXL_DEC_SUCCESS)
656 goto done;
657
658 RETVAL = gdk_pixbuf_new (GDK_COLORSPACE_RGB, !!info.alpha_bits, 8, info.xsize, info.ysize);
659 error = "unable to allocate pixbuf";
660 if (!RETVAL)
661 goto done;
662
663 JxlPixelFormat format = {
664 info.alpha_bits ? 4 : 3,
665 JXL_TYPE_UINT8,
666 JXL_NATIVE_ENDIAN,
667 gdk_pixbuf_get_rowstride (RETVAL)
668 };
669
670 // cannot use gdk_pixbuf_get_byte_length because that does
671 // not return the size of the buffer, but the size of the buffer without
672 // the last padding bytes. the internal buffer is rowstride * ysize,
673 // and this is what the jxl decoder needs. none of this is documented
674 // in either library, of course.
675
676 status = JxlDecoderSetImageOutBuffer (
677 dec,
678 &format,
679 gdk_pixbuf_get_pixels (RETVAL),
680 gdk_pixbuf_get_rowstride (RETVAL) * info.ysize
681 );
682 error = "JxlDecoderSetImageOutBuffer failed";
683 if (status != JXL_DEC_SUCCESS)
684 goto done;
685 }
686 break;
687
688 default:
689 error = "unexpected event";
690 goto done;
691 }
692 }
693
694 done:
695 if (dec)
696 JxlDecoderDestroy (dec);
697
698 perlinterp_acquire ();
699
700 if (runner_cache)
701 JxlThreadParallelRunnerDestroy (runner);
702
703 runner_cache = runner;
704
705 if (error)
706 {
707 if (RETVAL)
708 g_object_unref (RETVAL);
709
710 croak ("load_jxl: %s (status %d)", error, status);
711 }
712#else
713 croak ("load_jxl: jpeg-xl not enabled at compile time");
714#endif
715}
716 OUTPUT:
717 RETVAL
718
719GdkPixbuf_noinc *
720decode_jpeg (SV *image_data, int thumbnail = 0, int iw = 0, int ih = 0)
221 CODE: 721 CODE:
222{ 722{
223 struct jpeg_decompress_struct cinfo; 723 struct jpeg_decompress_struct cinfo;
224 struct jpg_err_mgr jerr; 724 struct jpg_err_mgr jerr;
225 guchar *data;
226 int rs; 725 int rs;
227 FILE *fp;
228 volatile GdkPixbuf *pb = 0; 726 volatile GdkPixbuf *pb = 0;
727 STRLEN data_len;
728 guchar *data = SvPVbyte (image_data, data_len);
229 729
230 RETVAL = 0; 730 RETVAL = 0;
231 731
232 fp = fopen (SvPVbyte_nolen (path), "rb"); 732 perlinterp_release ();
233
234 if (!fp)
235 XSRETURN_UNDEF;
236 733
237 cinfo.err = jpeg_std_error (&jerr.err); 734 cinfo.err = jpeg_std_error (&jerr.err);
238 735
239 jerr.err.error_exit = cv_error_exit; 736 jerr.err.error_exit = cv_error_exit;
240 jerr.err.output_message = cv_error_output; 737 jerr.err.output_message = cv_error_output;
241 738
242 if ((rs = setjmp (jerr.setjmp_buffer))) 739 if ((rs = setjmp (jerr.setjmp_buffer)))
243 { 740 {
244 fclose (fp);
245 jpeg_destroy_decompress (&cinfo); 741 jpeg_destroy_decompress (&cinfo);
246 742
247 if (pb) 743 if (pb)
248 g_object_unref ((gpointer)pb); 744 g_object_unref ((gpointer)pb);
249 745
746 perlinterp_acquire ();
250 XSRETURN_UNDEF; 747 XSRETURN_UNDEF;
251 } 748 }
252 749
750 if (!data_len)
751 longjmp (jerr.setjmp_buffer, 4);
752
253 jpeg_create_decompress (&cinfo); 753 jpeg_create_decompress (&cinfo);
754 cv_jpeg_mem_src (&cinfo, data, data_len);
254 755
255 jpeg_stdio_src (&cinfo, fp);
256 jpeg_read_header (&cinfo, TRUE); 756 jpeg_read_header (&cinfo, TRUE);
257 757
258 cinfo.dct_method = JDCT_DEFAULT; 758 cinfo.dct_method = JDCT_DEFAULT;
259 cinfo.do_fancy_upsampling = FALSE; /* worse quality, but nobody compained so far, and gdk-pixbuf does the same */ 759 cinfo.do_fancy_upsampling = FALSE; /* worse quality, but nobody compained so far, and gdk-pixbuf does the same */
260 cinfo.do_block_smoothing = FALSE; 760 cinfo.do_block_smoothing = FALSE;
270 { 770 {
271 cinfo.dct_method = JDCT_FASTEST; 771 cinfo.dct_method = JDCT_FASTEST;
272 cinfo.do_fancy_upsampling = FALSE; 772 cinfo.do_fancy_upsampling = FALSE;
273 773
274 while (cinfo.scale_denom < 8 774 while (cinfo.scale_denom < 8
275 && cinfo.output_width >= IW*4 775 && cinfo.output_width >= iw*4
276 && cinfo.output_height >= IH*4) 776 && cinfo.output_height >= ih*4)
277 { 777 {
278 cinfo.scale_denom <<= 1; 778 cinfo.scale_denom <<= 1;
279 jpeg_calc_output_dimensions (&cinfo); 779 jpeg_calc_output_dimensions (&cinfo);
280 } 780 }
281 } 781 }
282 782
783 if (cinfo.output_components != 3)
784 longjmp (jerr.setjmp_buffer, 3);
785
786 if (cinfo.jpeg_color_space == JCS_YCCK || cinfo.jpeg_color_space == JCS_CMYK)
787 {
788 cinfo.out_color_space = JCS_CMYK;
789 cinfo.output_components = 4;
790 }
791
283 pb = RETVAL = gdk_pixbuf_new (GDK_COLORSPACE_RGB, 0, 8, cinfo.output_width, cinfo.output_height); 792 pb = RETVAL = gdk_pixbuf_new (GDK_COLORSPACE_RGB, cinfo.output_components == 4, 8, cinfo.output_width, cinfo.output_height);
284 if (!RETVAL) 793 if (!RETVAL)
285 longjmp (jerr.setjmp_buffer, 2); 794 longjmp (jerr.setjmp_buffer, 2);
286 795
287 data = gdk_pixbuf_get_pixels (RETVAL); 796 data = gdk_pixbuf_get_pixels (RETVAL);
288 rs = gdk_pixbuf_get_rowstride (RETVAL); 797 rs = gdk_pixbuf_get_rowstride (RETVAL);
289
290 if (cinfo.output_components != 3)
291 longjmp (jerr.setjmp_buffer, 3);
292 798
293 jpeg_start_decompress (&cinfo); 799 jpeg_start_decompress (&cinfo);
294 800
295 while (cinfo.output_scanline < cinfo.output_height) 801 while (cinfo.output_scanline < cinfo.output_height)
296 { 802 {
303 rp [3] = (guchar *)rp [2] + rs; 809 rp [3] = (guchar *)rp [2] + rs;
304 810
305 jpeg_read_scanlines (&cinfo, rp, remaining < 4 ? remaining : 4); 811 jpeg_read_scanlines (&cinfo, rp, remaining < 4 ? remaining : 4);
306 } 812 }
307 813
814 if (cinfo.out_color_space == JCS_CMYK)
815 {
816 guchar *end = data + cinfo.output_height * rs;
817
818 while (data < end)
819 {
820 U32 c = data [0];
821 U32 m = data [1];
822 U32 y = data [2];
823 U32 k = data [3];
824
825 if (0)
826 if (cinfo.Adobe_transform == 2)
827 {
828 c ^= 0xff;
829 m ^= 0xff;
830 y ^= 0xff;
831 k ^= 0xff;
832 }
833
834 data [0] = (c * k + 0x80) / 0xff;
835 data [1] = (m * k + 0x80) / 0xff;
836 data [2] = (y * k + 0x80) / 0xff;
837 data [3] = 0xff;
838
839 data += 4;
840 }
841 }
842
308 jpeg_finish_decompress (&cinfo); 843 jpeg_finish_decompress (&cinfo);
309 fclose (fp);
310 jpeg_destroy_decompress (&cinfo); 844 jpeg_destroy_decompress (&cinfo);
845 perlinterp_acquire ();
311} 846}
312 OUTPUT: 847 OUTPUT:
313 RETVAL 848 RETVAL
314 849
850void
851compare (GdkPixbuf *a, GdkPixbuf *b)
852 PPCODE:
853 perlinterp_release ();
854{
855 int w = gdk_pixbuf_get_width (a);
856 int h = gdk_pixbuf_get_height (a);
857
858 int sa = gdk_pixbuf_get_rowstride (a);
859 int sb = gdk_pixbuf_get_rowstride (b);
860
861 guchar *pa = gdk_pixbuf_get_pixels (a);
862 guchar *pb = gdk_pixbuf_get_pixels (b);
863
864 int x, y;
865
866 assert (w == gdk_pixbuf_get_width (b));
867 assert (h == gdk_pixbuf_get_height (b));
868
869 assert (gdk_pixbuf_get_n_channels (a) == 3);
870 assert (gdk_pixbuf_get_n_channels (b) == 3);
871
872 double diff = 0.;
873 int peak = 0;
874
875 if (w && h)
876 for (y = 0; y < h; y++)
877 {
878 guchar *pa_ = pa + y * sa;
879 guchar *pb_ = pb + y * sb;
880
881 for (x = 0; x < w; x++)
882 {
883 int d;
884
885 d = ((int)*pa_++) - ((int)*pb_++); diff += d*d; peak = MAX (peak, abs (d));
886 d = ((int)*pa_++) - ((int)*pb_++); diff += d*d; peak = MAX (peak, abs (d));
887 d = ((int)*pa_++) - ((int)*pb_++); diff += d*d; peak = MAX (peak, abs (d));
888 }
889 }
890
891 perlinterp_acquire ();
892
893 EXTEND (SP, 2);
894 PUSHs (sv_2mortal (newSVnv (sqrt (diff / (w * h * 3. * 255. * 255.)))));
895 PUSHs (sv_2mortal (newSVnv (peak / 255.)));
896}
897
315############################################################################# 898#############################################################################
316 899
317MODULE = Gtk2::CV PACKAGE = Gtk2::CV::Schnauzer 900MODULE = Gtk2::CV PACKAGE = Gtk2::CV::Schnauzer
901
902# currently only works for filenames (octet strings)
318 903
319SV * 904SV *
320foldcase (SV *pathsv) 905foldcase (SV *pathsv)
321 PROTOTYPE: $ 906 PROTOTYPE: $
322 CODE: 907 CODE:
323{ 908{
324 STRLEN plen; 909 STRLEN plen;
325 U8 *path = (U8 *)SvPVutf8 (pathsv, plen); 910 U8 *path = (U8 *)SvPV (pathsv, plen);
326 U8 *pend = path + plen; 911 U8 *pend = path + plen;
327 U8 dst [plen * 6 * 3], *dstp = dst; 912 U8 dst [plen * 8 * 3], *dstp = dst;
328 913
329 while (path < pend) 914 while (path < pend)
330 { 915 {
331 U8 ch = *path; 916 U8 ch = *path;
332 917
333 if (ch >= 'a' && ch <= 'z') 918 if (ch >= 'a' && ch <= 'z')
334 *dstp++ = *path++; 919 *dstp++ = *path++;
920 else if (ch >= 'A' && ch <= 'Z')
921 *dstp++ = *path++ + ('a' - 'A');
335 else if (ch >= '0' && ch <= '9') 922 else if (ch >= '0' && ch <= '9')
336 { 923 {
924 /* version sort, up to 8 digits */
337 STRLEN el, nl = 0; 925 STRLEN el, nl = 0;
338 while (*path >= '0' && *path <= '9' && path < pend) 926 while (*path >= '0' && *path <= '9' && path < pend)
339 path++, nl++; 927 path++, nl++;
340 928
341 for (el = nl; el < 6; el++) 929 for (el = nl; el < 8; el++)
342 *dstp++ = '0'; 930 *dstp++ = '0';
343 931
344 memcpy (dstp, path - nl, nl); 932 memcpy (dstp, path - nl, nl);
345 dstp += nl; 933 dstp += nl;
346 } 934 }
935 else
936 *dstp++ = *path++;
937#if 0
347 else 938 else
348 { 939 {
349 STRLEN cl; 940 STRLEN cl;
350 to_utf8_fold (path, dstp, &cl); 941 to_utf8_fold (path, dstp, &cl);
351 dstp += cl; 942 dstp += cl;
352 path += is_utf8_char (path); 943 path += is_utf8_char (path);
353 } 944 }
945#endif
354 } 946 }
355 947
356 RETVAL = newSVpvn ((const char *)dst, dstp - dst); 948 RETVAL = newSVpvn ((const char *)dst, dstp - dst);
357} 949}
358 OUTPUT: 950 OUTPUT:
520############################################################################# 1112#############################################################################
521 1113
522MODULE = Gtk2::CV PACKAGE = Gtk2::CV::Plugin::RCluster 1114MODULE = Gtk2::CV PACKAGE = Gtk2::CV::Plugin::RCluster
523 1115
524SV * 1116SV *
525make_histograms (SV *ar) 1117extract_features (SV *ar)
526 CODE: 1118 CODE:
527{ 1119{
528 int i; 1120 int i;
529 AV *av, *result; 1121 AV *av, *result;
530 1122
531 if (!SvROK (ar) || SvTYPE (SvRV (ar)) != SVt_PVAV) 1123 if (!SvROK (ar) || SvTYPE (SvRV (ar)) != SVt_PVAV)
532 croak ("Not an array ref as first argument to make_histogram"); 1124 croak ("Not an array ref as first argument to extract_features");
533 1125
534 av = (AV *) SvRV (ar); 1126 av = (AV *) SvRV (ar);
535 result = newAV (); 1127 result = newAV ();
536 1128
537 for (i = 0; i <= av_len (av); ++i) 1129 for (i = 0; i <= av_len (av); ++i)
538 { 1130 {
539 const int HISTSIZE = 64;
540
541 int j;
542 SV *sv = *av_fetch (av, i, 1); 1131 SV *sv = *av_fetch (av, i, 1);
543 STRLEN len;
544 char *buf = SvPVbyte (sv, len);
545
546 int tmphist[HISTSIZE];
547 float *hist;
548
549 SV *histsv = newSV (HISTSIZE * sizeof (float) + 1); 1132 SV *histsv = newSV (9 * sizeof (float) + 1);
1133
550 SvPOK_on (histsv); 1134 SvPOK_on (histsv);
551 SvCUR_set (histsv, HISTSIZE * sizeof (float)); 1135 SvCUR_set (histsv, 9 * sizeof (float));
552 hist = (float *)SvPVX (histsv); 1136 float *hist = (float *)SvPVX (histsv);
553 1137
554 Zero (tmphist, sizeof (tmphist), char); 1138 struct feature f_h, f_s, f_v;
1139 feature_init (&f_h);
1140 feature_init (&f_s);
1141 feature_init (&f_v);
555 1142
556 for (j = len; j--; )
557 {
558 unsigned int idx
559 = ((*buf & 0xc0) >> 2)
560 | ((*buf & 0x18) >> 1)
561 | (*buf & 0x03);
562
563 ++tmphist[idx];
564 ++buf;
565 }
566 1143 {
567 for (j = 0; j < HISTSIZE; ++j) 1144 STRLEN len;
568 hist[j] = (float)tmphist[j] / (len + 1e-30); 1145 unsigned char *buf = (unsigned char *)SvPVbyte (sv, len);
1146 while (len >= 3)
1147 {
1148 unsigned int r, g, b, h, s, v;
1149 r = *buf++; g = *buf++; b = *buf++;
1150 rgb_to_hsv (r, g, b, &h, &s, &v);
1151
1152 feature_update_pass_1 (&f_h, h);
1153 feature_update_pass_1 (&f_s, s);
1154 feature_update_pass_1 (&f_v, v);
1155
1156 len -= 3;
1157 }
1158
1159 feature_finish_pass_1 (&f_h);
1160 feature_finish_pass_1 (&f_s);
1161 feature_finish_pass_1 (&f_v);
1162 }
1163
1164 {
1165 STRLEN len;
1166 unsigned char *buf = (unsigned char *)SvPVbyte (sv, len);
1167 while (len >= 3)
1168 {
1169 unsigned int r, g, b, h, s, v;
1170 r = *buf++; g = *buf++; b = *buf++;
1171 rgb_to_hsv (r, g, b, &h, &s, &v);
1172
1173 feature_update_pass_2 (&f_h, h);
1174 feature_update_pass_2 (&f_s, s);
1175 feature_update_pass_2 (&f_v, v);
1176
1177 len -= 3;
1178 }
1179
1180 feature_finish_pass_2 (&f_h);
1181 feature_finish_pass_2 (&f_s);
1182 feature_finish_pass_2 (&f_v);
1183 }
1184
1185 hist [0] = f_h.v1 * 2.; hist [1] = f_h.v2 * 2.; hist [2] = f_h.v3 * 2.;
1186 hist [3] = f_s.v1 ; hist [4] = f_s.v2 ; hist [5] = f_s.v3 ;
1187 hist [6] = f_v.v1 * .5; hist [7] = f_v.v2 * .5; hist [8] = f_v.v3 * .5;
569 1188
570 av_push (result, histsv); 1189 av_push (result, histsv);
571 } 1190 }
572 1191
573 RETVAL = newRV_noinc ((SV *)result); 1192 RETVAL = newRV_noinc ((SV *)result);
574} 1193}
575 OUTPUT: 1194 OUTPUT:
576 RETVAL 1195 RETVAL
577 1196
578

Diff Legend

Removed lines
+ Added lines
< Changed lines
> Changed lines