]> mj.ucw.cz Git - libucw.git/blob - images/io-libjpeg.c
f4a72e29a70228afb7d98681e246347d531605b2
[libucw.git] / images / io-libjpeg.c
1 /*
2  *      Image Library -- libjpeg
3  *
4  *      (c) 2006 Pavel Charvat <pchar@ucw.cz>
5  *
6  *      This software may be freely distributed and used according to the terms
7  *      of the GNU Lesser General Public License.
8  */
9
10 #define LOCAL_DEBUG
11
12 #include "lib/lib.h"
13 #include "lib/mempool.h"
14 #include "lib/fastbuf.h"
15 #include "images/images.h"
16 #include "images/error.h"
17 #include "images/color.h"
18 #include "images/io-main.h"
19
20 #include <stdio.h>
21 #include <sys/types.h>
22 #include <jpeglib.h>
23 #include <jerror.h>
24 #include <setjmp.h>
25
26 struct libjpeg_err {
27   struct jpeg_error_mgr pub;
28   jmp_buf setjmp_buf;
29   struct image_io *io;
30 };
31
32 struct libjpeg_read_internals {
33   struct jpeg_decompress_struct cinfo;
34   struct jpeg_source_mgr src;
35   struct libjpeg_err err;
36   struct fastbuf *fastbuf;
37   byte *fastbuf_pos;
38 };
39
40 struct libjpeg_write_internals {
41   struct jpeg_compress_struct cinfo;
42   struct jpeg_destination_mgr dest;
43   struct libjpeg_err err;
44   struct fastbuf *fastbuf;
45   byte *fastbuf_pos;
46 };
47
48 static void NONRET
49 libjpeg_read_error_exit(j_common_ptr cinfo)
50 {
51   DBG("libjpeg_error_exit()");
52   struct libjpeg_err *e = (struct libjpeg_err *)cinfo->err;
53   byte buf[JMSG_LENGTH_MAX];
54   e->pub.format_message(cinfo, buf);
55   IMAGE_ERROR(e->io->context, IMAGE_ERROR_READ_FAILED, "%s", buf);
56   longjmp(e->setjmp_buf, 1);
57 }
58
59 static void NONRET
60 libjpeg_write_error_exit(j_common_ptr cinfo)
61 {
62   DBG("libjpeg_error_exit()");
63   struct libjpeg_err *e = (struct libjpeg_err *)cinfo->err;
64   byte buf[JMSG_LENGTH_MAX];
65   e->pub.format_message(cinfo, buf);
66   IMAGE_ERROR(e->io->context, IMAGE_ERROR_WRITE_FAILED, "%s", buf);
67   longjmp(e->setjmp_buf, 1);
68 }
69
70 static void
71 libjpeg_emit_message(j_common_ptr cinfo UNUSED, int msg_level UNUSED)
72 {
73 #ifdef LOCAL_DEBUG
74   byte buf[JMSG_LENGTH_MAX];
75   cinfo->err->format_message(cinfo, buf);
76   DBG("libjpeg_emit_message(): [%d] %s", msg_level, buf);
77 #endif
78   if (unlikely(msg_level == -1))
79     longjmp(((struct libjpeg_err *)(cinfo)->err)->setjmp_buf, 1);
80 }
81
82 static inline uns
83 libjpeg_fastbuf_read_prepare(struct libjpeg_read_internals *i)
84 {
85   byte *start;
86   uns len = bdirect_read_prepare(i->fastbuf, &start);
87   i->fastbuf_pos = start + len;
88   i->src.next_input_byte = start;
89   i->src.bytes_in_buffer = len;
90   return len;
91 }
92
93 static inline void
94 libjpeg_fastbuf_read_commit(struct libjpeg_read_internals *i)
95 {
96   bdirect_read_commit(i->fastbuf, i->fastbuf_pos);
97 }
98
99 static void
100 libjpeg_init_source(j_decompress_ptr cinfo)
101 {
102   DBG("libjpeg_init_source()");
103   libjpeg_fastbuf_read_prepare((struct libjpeg_read_internals *)cinfo);
104 }
105
106 static void
107 libjpeg_term_source(j_decompress_ptr cinfo UNUSED)
108 {
109   DBG("libjpeg_term_source()");
110   //libjpeg_fastbuf_read_commit((struct libjpeg_read_internals *)cinfo);
111 }
112
113 static boolean
114 libjpeg_fill_input_buffer(j_decompress_ptr cinfo)
115 {
116   DBG("libjpeg_fill_input_buffer()");
117   struct libjpeg_read_internals *i = (struct libjpeg_read_internals *)cinfo;
118   libjpeg_fastbuf_read_commit(i);
119   return !!libjpeg_fastbuf_read_prepare(i);
120 }
121
122 static void
123 libjpeg_skip_input_data(j_decompress_ptr cinfo, long num_bytes)
124 {
125   DBG("libjpeg_skip_input_data(num_bytes=%d)", (int)num_bytes);
126   if (num_bytes > 0)
127     {
128       struct libjpeg_read_internals *i = (struct libjpeg_read_internals *)cinfo;
129       if ((unsigned long)num_bytes <= i->src.bytes_in_buffer)
130         {
131           i->src.next_input_byte += num_bytes;
132           i->src.bytes_in_buffer -= num_bytes;
133         }
134       else
135         {
136           num_bytes -= i->src.bytes_in_buffer;
137           libjpeg_fastbuf_read_commit(i);
138           if (!bskip(i->fastbuf, num_bytes))
139             {
140               IMAGE_ERROR(i->err.io->context, IMAGE_ERROR_READ_FAILED, "Incomplete JPEG file");
141               longjmp(i->err.setjmp_buf, 1);
142             }
143           libjpeg_fastbuf_read_prepare(i);
144         }
145     }
146 }
147
148 static inline void
149 libjpeg_fastbuf_write_prepare(struct libjpeg_write_internals *i)
150 {
151   byte *start;
152   uns len = bdirect_write_prepare(i->fastbuf, &start);
153   i->fastbuf_pos = start + len;
154   i->dest.next_output_byte = start;
155   i->dest.free_in_buffer = len;
156   if (!len)
157     {
158       IMAGE_ERROR(i->err.io->context, IMAGE_ERROR_WRITE_FAILED, "Unexpected end of stream");
159       longjmp(i->err.setjmp_buf, 1);
160     }
161 }
162
163 static void
164 libjpeg_init_destination(j_compress_ptr cinfo)
165 {
166   DBG("libjpeg_init_destination()");
167   libjpeg_fastbuf_write_prepare((struct libjpeg_write_internals *)cinfo);
168 }
169
170 static void
171 libjpeg_term_destination(j_compress_ptr cinfo)
172 {
173   DBG("libjpeg_term_destination()");
174   struct libjpeg_write_internals *i = (struct libjpeg_write_internals *)cinfo;
175   bdirect_write_commit(i->fastbuf, (byte *)i->dest.next_output_byte);
176 }
177
178 static boolean
179 libjpeg_empty_output_buffer(j_compress_ptr cinfo)
180 {
181   DBG("libjpeg_empty_output_buffer()");
182   struct libjpeg_write_internals *i = (struct libjpeg_write_internals *)cinfo;
183   bdirect_write_commit(i->fastbuf, i->fastbuf_pos);
184   libjpeg_fastbuf_write_prepare(i);
185   return TRUE;
186 }
187
188 static inline uns
189 libjpeg_read_byte(struct libjpeg_read_internals *i)
190 {
191   DBG("libjpeg_read_byte()");
192   if (!i->src.bytes_in_buffer)
193     if (!libjpeg_fill_input_buffer(&i->cinfo))
194       ERREXIT(&i->cinfo, JERR_CANT_SUSPEND);
195   i->src.bytes_in_buffer--;
196   return *i->src.next_input_byte++;
197 }
198
199 static inline void
200 libjpeg_read_buf(struct libjpeg_read_internals *i, byte *buf, uns len)
201 {
202   DBG("libjpeg_read_buf(len=%u)", len);
203   while (len)
204     {
205       if (!i->src.bytes_in_buffer)
206         if (!libjpeg_fill_input_buffer(&i->cinfo))
207           ERREXIT(&i->cinfo, JERR_CANT_SUSPEND);
208       uns buf_size = i->src.bytes_in_buffer;
209       uns read_size = MIN(buf_size, len);
210       memcpy(buf, i->src.next_input_byte, read_size);
211       i->src.bytes_in_buffer -= read_size;
212       i->src.next_input_byte += read_size;
213       len -= read_size;
214     }
215 }
216
217 static byte libjpeg_exif_header[6] = { 'E', 'x', 'i', 'f', 0, 0 };
218
219 static boolean
220 libjpeg_app1_preprocessor(j_decompress_ptr cinfo)
221 {
222   struct libjpeg_read_internals *i = (struct libjpeg_read_internals *)cinfo;
223   struct image_io *io = i->err.io;
224   uns len = libjpeg_read_byte(i) << 8;
225   len += libjpeg_read_byte(i);
226   DBG("Found APP1 marker, len=%u", len);
227   if (len < 2)
228     return TRUE;
229   len -= 2;
230   if (len < 7 /*|| io->exif_size*/)
231     {
232       libjpeg_skip_input_data(cinfo, len);
233       return TRUE;
234     }
235   byte header[6];
236   libjpeg_read_buf(i, header, 6);
237   if (memcmp(header, libjpeg_exif_header, 6))
238     {
239       libjpeg_skip_input_data(cinfo, len - 6);
240       return TRUE;
241     }
242   io->exif_size = len;
243   io->exif_data = mp_alloc(io->internal_pool, len);
244   memcpy(io->exif_data, header, 6);
245   libjpeg_read_buf(i, io->exif_data + 6, len - 6);
246   DBG("Parsed EXIF of length %u", len);
247   return TRUE;
248 }
249
250 static void
251 libjpeg_read_cancel(struct image_io *io)
252 {
253   DBG("libjpeg_read_cancel()");
254   struct libjpeg_read_internals *i = io->read_data;
255   jpeg_destroy_decompress(&i->cinfo);
256 }
257
258 int
259 libjpeg_read_header(struct image_io *io)
260 {
261   DBG("libjpeg_read_header()");
262   struct libjpeg_read_internals *i = io->read_data = mp_alloc(io->internal_pool, sizeof(*i));
263   i->fastbuf = io->fastbuf;
264
265   /* Create libjpeg read structure */
266   DBG("Creating libjpeg read structure");
267   i->cinfo.err = jpeg_std_error(&i->err.pub);
268   i->err.pub.error_exit = libjpeg_read_error_exit;
269   i->err.pub.emit_message = libjpeg_emit_message;
270   i->err.io = io;
271   if (setjmp(i->err.setjmp_buf))
272     {
273       DBG("Libjpeg failed to read the image, longjump saved us");
274       jpeg_destroy_decompress(&i->cinfo);
275       return 0;
276     }
277   jpeg_create_decompress(&i->cinfo);
278
279   /* Initialize source manager */
280   i->cinfo.src = &i->src;
281   i->src.init_source = libjpeg_init_source;
282   i->src.fill_input_buffer = libjpeg_fill_input_buffer;
283   i->src.skip_input_data = libjpeg_skip_input_data;
284   i->src.resync_to_restart = jpeg_resync_to_restart;
285   i->src.term_source = libjpeg_term_source;
286
287   if (io->flags & IMAGE_IO_WANT_EXIF)
288     jpeg_set_marker_processor(&i->cinfo, JPEG_APP0 + 1, libjpeg_app1_preprocessor);
289
290   /* Read JPEG header and setup decompression options */
291   DBG("Reading image header");
292   jpeg_read_header(&i->cinfo, TRUE);
293   switch (i->cinfo.jpeg_color_space)
294     {
295       case JCS_GRAYSCALE:
296         io->flags = COLOR_SPACE_GRAYSCALE;
297         break;
298       case JCS_RGB:
299         io->flags = COLOR_SPACE_RGB;
300         break;
301       case JCS_YCbCr:
302         io->flags = COLOR_SPACE_YCBCR;
303         break;
304       case JCS_CMYK:
305         io->flags = COLOR_SPACE_CMYK;
306         break;
307       case JCS_YCCK:
308         io->flags = COLOR_SPACE_YCCK;
309         break;
310       default:
311         if (unlikely(i->cinfo.num_components < 1 || i->cinfo.num_components > 4))
312           {
313             jpeg_destroy_decompress(&i->cinfo);
314             IMAGE_ERROR(io->context, IMAGE_ERROR_INVALID_PIXEL_FORMAT, "Invalid color space.");
315             return 0;
316           }
317         io->flags = COLOR_SPACE_UNKNOWN + i->cinfo.num_components;
318         break;
319     }
320   if (unlikely(i->cinfo.num_components != (int)color_space_channels[io->flags]))
321     {
322       jpeg_destroy_decompress(&i->cinfo);
323       IMAGE_ERROR(io->context, IMAGE_ERROR_INVALID_PIXEL_FORMAT, "Invalid number of color channels.");
324       return 0;
325     }
326   io->cols = i->cinfo.image_width;
327   io->rows = i->cinfo.image_height;
328   io->number_of_colors = (i->cinfo.num_components < 4) ? (1U << (i->cinfo.num_components * 8)) : 0xffffffff;
329   io->read_cancel = libjpeg_read_cancel;
330   return 1;
331 }
332
333 int
334 libjpeg_read_data(struct image_io *io)
335 {
336   DBG("libjpeg_read_data()");
337
338   struct libjpeg_read_internals *i = io->read_data;
339
340   /* Select color space */
341   switch (io->flags & IMAGE_COLOR_SPACE)
342     {
343       case COLOR_SPACE_GRAYSCALE:
344         i->cinfo.out_color_space = JCS_GRAYSCALE;
345         break;
346       case COLOR_SPACE_RGB:
347         i->cinfo.out_color_space = JCS_RGB;
348         break;
349       case COLOR_SPACE_YCBCR:
350         i->cinfo.out_color_space = JCS_YCbCr;
351         break;
352       case COLOR_SPACE_CMYK:
353         i->cinfo.out_color_space = JCS_CMYK;
354         break;
355       case COLOR_SPACE_YCCK:
356         i->cinfo.out_color_space = JCS_YCCK;
357         break;
358       default:
359         jpeg_destroy_decompress(&i->cinfo);
360         IMAGE_ERROR(io->context, IMAGE_ERROR_INVALID_PIXEL_FORMAT, "Unsupported color space.");
361         return 0;
362     }
363
364   /* Prepare the image  */
365   struct image_io_read_data_internals rdi;
366   if (io->cols <= (i->cinfo.image_width >> 3) && io->rows <= (i->cinfo.image_height >> 3))
367     {
368       DBG("Scaling to 1/8");
369       i->cinfo.scale_num = 1;
370       i->cinfo.scale_denom = 8;
371     }
372   else if (io->cols <= (i->cinfo.image_width >> 2) && io->rows <= (i->cinfo.image_height >> 2))
373     {
374       DBG("Scaling to 1/4");
375       i->cinfo.scale_num = 1;
376       i->cinfo.scale_denom = 4;
377     }
378   else if (io->cols <= (i->cinfo.image_width >> 1) && io->rows <= (i->cinfo.image_height >> 1))
379     {
380       DBG("Scaling to 1/2");
381       i->cinfo.scale_num = 1;
382       i->cinfo.scale_denom = 2;
383     }
384   jpeg_calc_output_dimensions(&i->cinfo);
385   DBG("Output dimensions %ux%u", (uns)i->cinfo.output_width, (uns)i->cinfo.output_height);
386   if (unlikely(!image_io_read_data_prepare(&rdi, io, i->cinfo.output_width, i->cinfo.output_height, io->flags)))
387     {
388       jpeg_destroy_decompress(&i->cinfo);
389       return 0;
390     }
391
392   /* Setup fallback */
393   if (setjmp(i->err.setjmp_buf))
394     {
395       DBG("Libjpeg failed to read the image, longjump saved us");
396       jpeg_destroy_decompress(&i->cinfo);
397       image_io_read_data_break(&rdi, io);
398       return 0;
399     }
400
401   /* Decompress the image */
402   struct image *img = rdi.image;
403   jpeg_start_decompress(&i->cinfo);
404   if ((int)img->pixel_size == i->cinfo.num_components)
405     {
406       byte *pixels = img->pixels;
407       for (uns r = img->rows; r--; )
408         {
409           jpeg_read_scanlines(&i->cinfo, (JSAMPLE **)&pixels, 1);
410           pixels += img->row_size;
411         }
412     }
413   else
414     {
415       switch (img->pixel_size)
416         {
417           case 2: /* Grayscale -> Grayscale+Alpha */
418             {
419               ASSERT(i->cinfo.num_components == 1);
420               byte buf[img->cols], *src;
421 #             define IMAGE_WALK_PREFIX(x) walk_##x
422 #             define IMAGE_WALK_INLINE
423 #             define IMAGE_WALK_IMAGE img
424 #             define IMAGE_WALK_UNROLL 4
425 #             define IMAGE_WALK_COL_STEP 2
426 #             define IMAGE_WALK_DO_ROW_START do{ src = buf; jpeg_read_scanlines(&i->cinfo, (JSAMPLE **)&src, 1); }while(0)
427 #             define IMAGE_WALK_DO_STEP do{ walk_pos[0] = *src++; walk_pos[1] = 255; }while(0)
428 #             include "images/image-walk.h"
429             }
430             break;
431           case 4: /* * -> *+Alpha or aligned * */
432             {
433               ASSERT(i->cinfo.num_components == 3);
434               byte buf[img->cols * 3], *src;
435 #             define IMAGE_WALK_PREFIX(x) walk_##x
436 #             define IMAGE_WALK_INLINE
437 #             define IMAGE_WALK_IMAGE img
438 #             define IMAGE_WALK_UNROLL 4
439 #             define IMAGE_WALK_COL_STEP 4
440 #             define IMAGE_WALK_DO_ROW_START do{ src = buf; jpeg_read_scanlines(&i->cinfo, (JSAMPLE **)&src, 1); }while(0)
441 #             define IMAGE_WALK_DO_STEP do{ *(u32 *)walk_pos = *(u32 *)src; walk_pos[3] = 255; src += 3; }while(0)
442 #             include "images/image-walk.h"
443             }
444             break;
445           default:
446             ASSERT(0);
447         }
448
449     }
450
451   ASSERT(i->cinfo.output_scanline == i->cinfo.output_height);
452
453   /* Destroy libjpeg object */
454   jpeg_finish_decompress(&i->cinfo);
455   jpeg_destroy_decompress(&i->cinfo);
456
457   /* Finish the image */
458   return image_io_read_data_finish(&rdi, io);
459 }
460
461 int
462 libjpeg_write(struct image_io *io)
463 {
464   DBG("libjpeg_write()");
465   struct libjpeg_write_internals i;
466   i.fastbuf = io->fastbuf;
467
468   /* Create libjpeg write structure */
469   DBG("Creating libjpeg write structure");
470   i.cinfo.err = jpeg_std_error(&i.err.pub);
471   i.err.pub.error_exit = libjpeg_write_error_exit;
472   i.err.pub.emit_message = libjpeg_emit_message;
473   i.err.io = io;
474   if (setjmp(i.err.setjmp_buf))
475     {
476       DBG("Libjpeg failed to write the image, longjump saved us");
477       jpeg_destroy_compress(&i.cinfo);
478       return 0;
479     }
480   jpeg_create_compress(&i.cinfo);
481
482   /* Initialize destination manager */
483   i.cinfo.dest = &i.dest;
484   i.dest.init_destination = libjpeg_init_destination;
485   i.dest.term_destination = libjpeg_term_destination;
486   i.dest.empty_output_buffer = libjpeg_empty_output_buffer;
487
488   /* Set output parameters */
489   struct image *img = io->image;
490   i.cinfo.image_width = img->cols;
491   i.cinfo.image_height = img->rows;
492   switch (img->flags & IMAGE_COLOR_SPACE)
493     {
494       case COLOR_SPACE_GRAYSCALE:
495         i.cinfo.in_color_space = JCS_GRAYSCALE;
496         break;
497       case COLOR_SPACE_RGB:
498         i.cinfo.in_color_space = JCS_RGB;
499         break;
500       case COLOR_SPACE_YCBCR:
501         i.cinfo.in_color_space = JCS_YCbCr;
502         break;
503       case COLOR_SPACE_CMYK:
504         i.cinfo.in_color_space = JCS_CMYK;
505         break;
506       case COLOR_SPACE_YCCK:
507         i.cinfo.in_color_space = JCS_YCCK;
508         break;
509       default:
510         jpeg_destroy_compress(&i.cinfo);
511         IMAGE_ERROR(io->context, IMAGE_ERROR_INVALID_PIXEL_FORMAT, "Unsupported pixel format.");
512         return 0;
513     }
514   i.cinfo.input_components = color_space_channels[img->flags & IMAGE_COLOR_SPACE];
515   jpeg_set_defaults(&i.cinfo);
516   if (io->jpeg_quality)
517     jpeg_set_quality(&i.cinfo, MIN(io->jpeg_quality, 100), 1);
518   if (io->exif_size)
519     {
520       /* According to the Exif specification, the Exif APP1 marker has to follow immediately after the SOI,
521        * just as the JFIF specification requires the same for the JFIF APP0 marker!
522        * Therefore a JPEG file cannot legally be both Exif and JFIF.  */
523       i.cinfo.write_JFIF_header = FALSE;
524       i.cinfo.write_Adobe_marker = FALSE;
525     }
526
527   /* Compress the image */
528   jpeg_start_compress(&i.cinfo, TRUE);
529   if (io->exif_size)
530     {
531       DBG("Writing EXIF");
532       jpeg_write_marker(&i.cinfo, JPEG_APP0 + 1, io->exif_data, io->exif_size);
533     }
534   if ((int)img->pixel_size == i.cinfo.input_components)
535     {
536       byte *pixels = img->pixels;
537       for (uns r = img->rows; r--; )
538         {
539           jpeg_write_scanlines(&i.cinfo, (JSAMPLE **)&pixels, 1);
540           pixels += img->row_size;
541         }
542     }
543   else
544     {
545       switch (img->pixel_size)
546         {
547           case 2: /* Grayscale+Alpha -> Grayscale */
548             {
549               ASSERT(i.cinfo.input_components == 1);
550               byte buf[img->cols], *dest = buf;
551 #             define IMAGE_WALK_PREFIX(x) walk_##x
552 #             define IMAGE_WALK_INLINE
553 #             define IMAGE_WALK_IMAGE img
554 #             define IMAGE_WALK_UNROLL 4
555 #             define IMAGE_WALK_COL_STEP 2
556 #             define IMAGE_WALK_DO_ROW_END do{ dest = buf; jpeg_write_scanlines(&i.cinfo, (JSAMPLE **)&dest, 1); }while(0)
557 #             define IMAGE_WALK_DO_STEP do{ *dest++ = walk_pos[0]; }while(0)
558 #             include "images/image-walk.h"
559               break;
560             }
561           case 4: /* *+Alpha or aligned * -> * */
562             {
563               ASSERT(i.cinfo.input_components == 3);
564               byte buf[img->cols * 3], *dest = buf;
565 #             define IMAGE_WALK_PREFIX(x) walk_##x
566 #             define IMAGE_WALK_INLINE
567 #             define IMAGE_WALK_IMAGE img
568 #             define IMAGE_WALK_UNROLL 4
569 #             define IMAGE_WALK_COL_STEP 4
570 #             define IMAGE_WALK_DO_ROW_END do{ dest = buf; jpeg_write_scanlines(&i.cinfo, (JSAMPLE **)&dest, 1); }while(0)
571 #             define IMAGE_WALK_DO_STEP do{ *dest++ = walk_pos[0]; *dest++ = walk_pos[1]; *dest++ = walk_pos[2]; }while(0)
572 #             include "images/image-walk.h"
573               break;
574             }
575           default:
576             ASSERT(0);
577         }
578     }
579   ASSERT(i.cinfo.next_scanline == i.cinfo.image_height);
580   jpeg_finish_compress(&i.cinfo);
581   jpeg_destroy_compress(&i.cinfo);
582   return 1;
583 }