rest of bbuf printf

[libucw.git] / images / io-libpng.c

diff --git a/images/io-libpng.c b/images/io-libpng.c
index 4b4864f61f5b31cbc9eaf98eb8e8dd9091ea31e3..f15955a64cf83c045bd953cb1afa33fd96908b96 100644 (file)
--- a/images/io-libpng.c
+++ b/images/io-libpng.c
@@ -13,6 +13,7 @@
 #include "lib/mempool.h"
 #include "lib/fastbuf.h"
 #include "images/images.h"
 #include "lib/mempool.h"
 #include "lib/fastbuf.h"
 #include "images/images.h"

+#include "images/io-main.h"

 #include <png.h>
 #include <setjmp.h>
 
 #include <png.h>
 #include <setjmp.h>
 


@@ -43,7 +44,7 @@ static void NONRET
 libpng_read_error(png_structp png_ptr, png_const_charp msg)
 {
   DBG("libpng_read_error()");
 libpng_read_error(png_structp png_ptr, png_const_charp msg)
 {
   DBG("libpng_read_error()");

-  image_thread_err(png_get_error_ptr(png_ptr), IMAGE_ERR_READ_FAILED, (byte *)msg);
+  image_thread_err_dup(png_get_error_ptr(png_ptr), IMAGE_ERR_READ_FAILED, (byte *)msg);

   longjmp(png_jmpbuf(png_ptr), 1);
 }
 
   longjmp(png_jmpbuf(png_ptr), 1);
 }
 


@@ -51,7 +52,7 @@ static void NONRET
 libpng_write_error(png_structp png_ptr, png_const_charp msg)
 {
   DBG("libpng_write_error()");
 libpng_write_error(png_structp png_ptr, png_const_charp msg)
 {
   DBG("libpng_write_error()");

-  image_thread_err(png_get_error_ptr(png_ptr), IMAGE_ERR_WRITE_FAILED, (byte *)msg);
+  image_thread_err_dup(png_get_error_ptr(png_ptr), IMAGE_ERR_WRITE_FAILED, (byte *)msg);

   longjmp(png_jmpbuf(png_ptr), 1);
 }
 
   longjmp(png_jmpbuf(png_ptr), 1);
 }
 


@@ -95,7 +96,7 @@ int
 libpng_read_header(struct image_io *io)
 {
   DBG("libpng_read_header()");
 libpng_read_header(struct image_io *io)
 {
   DBG("libpng_read_header()");

-  
+

   /* Create libpng structures */
   struct libpng_read_data *rd = io->read_data = mp_alloc(io->internal_pool, sizeof(*rd));
   rd->png_ptr = png_create_read_struct_2(PNG_LIBPNG_VER_STRING,
   /* Create libpng structures */
   struct libpng_read_data *rd = io->read_data = mp_alloc(io->internal_pool, sizeof(*rd));
   rd->png_ptr = png_create_read_struct_2(PNG_LIBPNG_VER_STRING,


@@ -120,7 +121,7 @@ libpng_read_header(struct image_io *io)
       png_destroy_read_struct(&rd->png_ptr, &rd->info_ptr, NULL);
       return 0;
     }
       png_destroy_read_struct(&rd->png_ptr, &rd->info_ptr, NULL);
       return 0;
     }

-  
+

   /* Setup libpng longjump */
   if (unlikely(setjmp(png_jmpbuf(rd->png_ptr))))
     {
   /* Setup libpng longjump */
   if (unlikely(setjmp(png_jmpbuf(rd->png_ptr))))
     {


@@ -128,7 +129,7 @@ libpng_read_header(struct image_io *io)
       png_destroy_read_struct(&rd->png_ptr, &rd->info_ptr, &rd->end_ptr);
       return 0;
     }
       png_destroy_read_struct(&rd->png_ptr, &rd->info_ptr, &rd->end_ptr);
       return 0;
     }

-  
+

   /* Setup libpng IO */
   png_set_read_fn(rd->png_ptr, io->fastbuf, libpng_read_fn);
   png_set_user_limits(rd->png_ptr, IMAGE_MAX_SIZE, IMAGE_MAX_SIZE);
   /* Setup libpng IO */
   png_set_read_fn(rd->png_ptr, io->fastbuf, libpng_read_fn);
   png_set_user_limits(rd->png_ptr, IMAGE_MAX_SIZE, IMAGE_MAX_SIZE);


@@ -138,31 +139,39 @@ libpng_read_header(struct image_io *io)
   png_get_IHDR(rd->png_ptr, rd->info_ptr, &rd->cols, &rd->rows, &rd->bit_depth, &rd->color_type, NULL, NULL, NULL);
 
   /* Fill image_io values */
   png_get_IHDR(rd->png_ptr, rd->info_ptr, &rd->cols, &rd->rows, &rd->bit_depth, &rd->color_type, NULL, NULL, NULL);
 
   /* Fill image_io values */

-  if (!io->cols)
-    io->cols = rd->cols;
-  if (!io->rows)
-    io->rows = rd->rows;
-  if (!(io->flags & IMAGE_CHANNELS_FORMAT))
-    switch (rd->color_type)
-      {
-       case PNG_COLOR_TYPE_GRAY:
-         io->flags |= COLOR_SPACE_GRAYSCALE;
-         break;
-       case PNG_COLOR_TYPE_GRAY_ALPHA:
-         io->flags |= COLOR_SPACE_GRAYSCALE | IMAGE_ALPHA;
-         break;
-       case PNG_COLOR_TYPE_RGB:
-         io->flags |= COLOR_SPACE_RGB;
-         break;
-       case PNG_COLOR_TYPE_RGB_ALPHA:
-       case PNG_COLOR_TYPE_PALETTE:
-         io->flags |= COLOR_SPACE_RGB | IMAGE_ALPHA;
-         break;
-       default:
-         png_destroy_read_struct(&rd->png_ptr, &rd->info_ptr, &rd->end_ptr);
-         image_thread_err(io->thread, IMAGE_ERR_READ_FAILED, "Unknown color type");
-         break;
-      }
+  io->cols = rd->cols;
+  io->rows = rd->rows;
+  switch (rd->color_type)
+    {
+      case PNG_COLOR_TYPE_GRAY:
+        io->flags = COLOR_SPACE_GRAYSCALE;
+       io->number_of_colors = 1 << 8;
+        break;
+      case PNG_COLOR_TYPE_GRAY_ALPHA:
+        io->flags = COLOR_SPACE_GRAYSCALE | IMAGE_ALPHA;
+       io->number_of_colors = 1 << 8;
+        break;
+      case PNG_COLOR_TYPE_RGB:
+        io->flags = COLOR_SPACE_RGB;
+       io->number_of_colors = 1 << 24;
+        break;
+      case PNG_COLOR_TYPE_RGB_ALPHA:
+       io->number_of_colors = 1 << 24;
+        io->flags = COLOR_SPACE_RGB | IMAGE_ALPHA;
+        break;
+      case PNG_COLOR_TYPE_PALETTE:
+        io->flags = COLOR_SPACE_RGB | IMAGE_ALPHA | IMAGE_IO_HAS_PALETTE;
+       int num_palette;
+       if (png_get_PLTE(rd->png_ptr, rd->info_ptr, NULL, &num_palette))
+         io->number_of_colors = num_palette;
+       else
+         io->number_of_colors = 1 << rd->bit_depth;
+        break;
+      default:
+        png_destroy_read_struct(&rd->png_ptr, &rd->info_ptr, &rd->end_ptr);
+        image_thread_err(io->thread, IMAGE_ERR_READ_FAILED, "Unknown color type");
+        break;
+    }

 
   /* Success */
   io->read_cancel = libpng_read_cancel;
 
   /* Success */
   io->read_cancel = libpng_read_cancel;


@@ -188,25 +197,20 @@ libpng_read_data(struct image_io *io)
         return 0;
     }
 
         return 0;
     }
 

-  volatile int need_scale = io->cols != rd->cols || io->rows != rd->rows;
-  struct image * volatile img = need_scale ? 
-    image_new(io->thread, rd->cols, rd->rows, io->flags & IMAGE_PIXEL_FORMAT, NULL) :
-    image_new(io->thread, rd->cols, rd->rows, io->flags, io->pool);
-  if (!img)
-    {
-      png_destroy_read_struct(&rd->png_ptr, &rd->info_ptr, &rd->end_ptr);
-      return 0;
-    }
+  struct image_io_read_data_internals rdi;
+  rdi.image = NULL;

 
   if (setjmp(png_jmpbuf(rd->png_ptr)))
     {
       DBG("Libpng failed to read the image, longjump saved us");
       png_destroy_read_struct(&rd->png_ptr, &rd->info_ptr, &rd->end_ptr);
 
   if (setjmp(png_jmpbuf(rd->png_ptr)))
     {
       DBG("Libpng failed to read the image, longjump saved us");
       png_destroy_read_struct(&rd->png_ptr, &rd->info_ptr, &rd->end_ptr);

-      if (need_scale || !io->pool)
-       image_destroy(io->thread, img);
+      if (rdi.image)
+        image_io_read_data_break(&rdi, io);

       return 0;
     }
 
       return 0;
     }
 

+  uns read_flags = io->flags;
+  

   /* Apply transformations */
   if (rd->bit_depth == 16)
     png_set_strip_16(rd->png_ptr);
   /* Apply transformations */
   if (rd->bit_depth == 16)
     png_set_strip_16(rd->png_ptr);


@@ -220,10 +224,20 @@ libpng_read_data(struct image_io *io)
          }
        else
          png_set_palette_to_rgb(rd->png_ptr);
          }
        else
          png_set_palette_to_rgb(rd->png_ptr);

-       if ((io->flags & IMAGE_ALPHA) || (io->flags & IMAGE_PIXEL_FORMAT) == (COLOR_SPACE_RGB | IMAGE_PIXELS_ALIGNED))
-          png_set_add_alpha(rd->png_ptr, 255, PNG_FILLER_AFTER);
+       if (!(io->flags & IMAGE_ALPHA))
+         {
+           if (io->flags & IMAGE_IO_USE_BACKGROUND)
+             {
+                png_set_add_alpha(rd->png_ptr, 255, PNG_FILLER_AFTER);
+               read_flags = (read_flags | IMAGE_ALPHA) & IMAGE_CHANNELS_FORMAT;
+             }
+           else if ((io->flags & IMAGE_PIXEL_FORMAT) == (COLOR_SPACE_RGB | IMAGE_PIXELS_ALIGNED))
+              png_set_add_alpha(rd->png_ptr, 255, PNG_FILLER_AFTER);
+            else
+             png_set_strip_alpha(rd->png_ptr);
+         }

        else
        else

-         png_set_strip_alpha(rd->png_ptr);
+          png_set_add_alpha(rd->png_ptr, 255, PNG_FILLER_AFTER);

        break;
       case PNG_COLOR_TYPE_GRAY:
        if ((io->flags & IMAGE_COLOR_SPACE) == COLOR_SPACE_RGB)
        break;
       case PNG_COLOR_TYPE_GRAY:
        if ((io->flags & IMAGE_COLOR_SPACE) == COLOR_SPACE_RGB)


@@ -235,7 +249,12 @@ libpng_read_data(struct image_io *io)
        if ((io->flags & IMAGE_COLOR_SPACE) == COLOR_SPACE_RGB)
           png_set_gray_to_rgb(rd->png_ptr);
        if (!(io->flags & IMAGE_ALPHA))
        if ((io->flags & IMAGE_COLOR_SPACE) == COLOR_SPACE_RGB)
           png_set_gray_to_rgb(rd->png_ptr);
        if (!(io->flags & IMAGE_ALPHA))

-          png_set_strip_alpha(rd->png_ptr);
+         {
+           if (io->flags & IMAGE_IO_USE_BACKGROUND)
+             read_flags = (read_flags | IMAGE_ALPHA) & IMAGE_CHANNELS_FORMAT;
+           else
+              png_set_strip_alpha(rd->png_ptr);
+         }  

        break;
       case PNG_COLOR_TYPE_RGB:
        if ((io->flags & IMAGE_COLOR_SPACE) == COLOR_SPACE_GRAYSCALE)
        break;
       case PNG_COLOR_TYPE_RGB:
        if ((io->flags & IMAGE_COLOR_SPACE) == COLOR_SPACE_GRAYSCALE)


@@ -246,16 +265,26 @@ libpng_read_data(struct image_io *io)
       case PNG_COLOR_TYPE_RGB_ALPHA:
        if ((io->flags & IMAGE_COLOR_SPACE) == COLOR_SPACE_GRAYSCALE)
          png_set_rgb_to_gray_fixed(rd->png_ptr, 1, 21267, 71514);
       case PNG_COLOR_TYPE_RGB_ALPHA:
        if ((io->flags & IMAGE_COLOR_SPACE) == COLOR_SPACE_GRAYSCALE)
          png_set_rgb_to_gray_fixed(rd->png_ptr, 1, 21267, 71514);

-       if (!(io->flags & IMAGE_ALPHA) && (io->flags & IMAGE_PIXEL_FORMAT) != (COLOR_SPACE_RGB | IMAGE_PIXELS_ALIGNED))
-          png_set_strip_alpha(rd->png_ptr);
+       if (!(io->flags & IMAGE_ALPHA))
+         if (io->flags & IMAGE_IO_USE_BACKGROUND)
+           read_flags = (read_flags | IMAGE_ALPHA) & IMAGE_CHANNELS_FORMAT;
+         else if ((io->flags & IMAGE_PIXEL_FORMAT) != (COLOR_SPACE_RGB | IMAGE_PIXELS_ALIGNED))
+            png_set_strip_alpha(rd->png_ptr);

        break;
       default:
        ASSERT(0);
     }
   png_read_update_info(rd->png_ptr, rd->info_ptr);
 
        break;
       default:
        ASSERT(0);
     }
   png_read_update_info(rd->png_ptr, rd->info_ptr);
 

+  /* Prepare the image */
+  if (unlikely(!image_io_read_data_prepare(&rdi, io, rd->cols, rd->rows, read_flags)))
+    {
+      png_destroy_read_struct(&rd->png_ptr, &rd->info_ptr, &rd->end_ptr);
+      return 0;
+    }

   /* Read image data */
   DBG("Reading image data");
   /* Read image data */
   DBG("Reading image data");

+  struct image *img = rdi.image;

   byte *pixels = img->pixels;
   png_bytep rows[img->rows];
   for (uns r = 0; r < img->rows; r++, pixels += img->row_size)
   byte *pixels = img->pixels;
   png_bytep rows[img->rows];
   for (uns r = 0; r < img->rows; r++, pixels += img->row_size)


@@ -266,29 +295,8 @@ libpng_read_data(struct image_io *io)
   /* Destroy libpng read structure */
   png_destroy_read_struct(&rd->png_ptr, &rd->info_ptr, &rd->end_ptr);
 
   /* Destroy libpng read structure */
   png_destroy_read_struct(&rd->png_ptr, &rd->info_ptr, &rd->end_ptr);
 

-  /* Scale and store the resulting image */
-  if (need_scale)
-    {
-      struct image *dest = image_new(io->thread, io->cols, io->rows, io->flags, io->pool);
-      if (!dest)
-        {
-         image_destroy(io->thread, img);
-         return 0;
-       }
-      if (!image_scale(io->thread, dest, img))
-        {
-         image_destroy(io->thread, img);
-         if (!io->pool)
-           image_destroy(io->thread, dest);
-         return 0;
-       }
-      io->image = dest;
-    }
-  else
-    io->image = img;
-  io->image_destroy = !io->pool;
-  
-  return 1;
+  /* Finish the image  */
+  return image_io_read_data_finish(&rdi, io);

 }
 
 int
 }
 
 int


@@ -312,7 +320,7 @@ libpng_write(struct image_io *io)
       png_destroy_write_struct(&png_ptr, NULL);
       return 0;
     }
       png_destroy_write_struct(&png_ptr, NULL);
       return 0;
     }

- 
+

   /* Setup libpng longjump */
   if (unlikely(setjmp(png_jmpbuf(png_ptr))))
     {
   /* Setup libpng longjump */
   if (unlikely(setjmp(png_jmpbuf(png_ptr))))
     {


@@ -323,7 +331,7 @@ libpng_write(struct image_io *io)
 
   /* Setup libpng IO */
   png_set_write_fn(png_ptr, io->fastbuf, libpng_write_fn, libpng_flush_fn);
 
   /* Setup libpng IO */
   png_set_write_fn(png_ptr, io->fastbuf, libpng_write_fn, libpng_flush_fn);

-  
+

   /* Setup PNG parameters */
   struct image *img = io->image;
   switch (img->flags & IMAGE_PIXEL_FORMAT)
   /* Setup PNG parameters */
   struct image *img = io->image;
   switch (img->flags & IMAGE_PIXEL_FORMAT)


@@ -350,7 +358,7 @@ libpng_write(struct image_io *io)
        png_set_filler(png_ptr, 0, PNG_FILLER_AFTER);
        break;
       default:
        png_set_filler(png_ptr, 0, PNG_FILLER_AFTER);
        break;
       default:

-        ASSERT(0);     
+        ASSERT(0);

     }
   png_write_info(png_ptr, info_ptr);
 
     }
   png_write_info(png_ptr, info_ptr);