1 /*
   2  * Copyright 2005-2006 Sun Microsystems, Inc.  All Rights Reserved.
   3  * DO NOT ALTER OR REMOVE COPYRIGHT NOTICES OR THIS FILE HEADER.
   4  *
   5  * This code is free software; you can redistribute it and/or modify it
   6  * under the terms of the GNU General Public License version 2 only, as
   7  * published by the Free Software Foundation.  Sun designates this
   8  * particular file as subject to the "Classpath" exception as provided
   9  * by Sun in the LICENSE file that accompanied this code.
  10  *
  11  * This code is distributed in the hope that it will be useful, but WITHOUT
  12  * ANY WARRANTY; without even the implied warranty of MERCHANTABILITY or
  13  * FITNESS FOR A PARTICULAR PURPOSE.  See the GNU General Public License
  14  * version 2 for more details (a copy is included in the LICENSE file that
  15  * accompanied this code).
  16  *
  17  * You should have received a copy of the GNU General Public License version
  18  * 2 along with this work; if not, write to the Free Software Foundation,
  19  * Inc., 51 Franklin St, Fifth Floor, Boston, MA 02110-1301 USA.
  20  *
  21  * Please contact Sun Microsystems, Inc., 4150 Network Circle, Santa Clara,
  22  * CA 95054 USA or visit www.sun.com if you need additional information or
  23  * have any questions.
  24  */
  25 
  26 #include "splashscreen_impl.h"
  27 
  28 #include "../libpng/png.h"
  29 
  30 #include <setjmp.h>
  31 
  32 #define SIG_BYTES 8
  33 
  34 void PNGAPI
  35 my_png_read_stream(png_structp png_ptr, png_bytep data, png_size_t length)
  36 {
  37     png_uint_32 check;
  38 
  39     SplashStream * stream = (SplashStream*)png_ptr->io_ptr;
  40     check = stream->read(stream, data, length);
  41     if (check != length)
  42         png_error(png_ptr, "Read Error");
  43 }
  44 
  45 int
  46 SplashDecodePng(Splash * splash, png_rw_ptr read_func, void *io_ptr)
  47 {
  48     int stride;
  49     ImageFormat srcFormat;
  50     png_uint_32 i, rowbytes;
  51     png_bytepp row_pointers = NULL;
  52     png_bytep image_data = NULL;
  53     int success = 0;
  54     double gamma;
  55 
  56     png_structp png_ptr = NULL;
  57     png_infop info_ptr = NULL;
  58 
  59     png_uint_32 width, height;
  60     int bit_depth, color_type;
  61 
  62     ImageRect srcRect, dstRect;
  63 
  64     png_ptr = png_create_read_struct(PNG_LIBPNG_VER_STRING, NULL, NULL, NULL);
  65     if (!png_ptr) {
  66         goto done;
  67     }
  68 
  69     info_ptr = png_create_info_struct(png_ptr);
  70     if (!info_ptr) {
  71         goto done;
  72     }
  73 
  74     if (setjmp(png_ptr->jmpbuf)) {
  75         goto done;
  76     }
  77 
  78     png_ptr->io_ptr = io_ptr;
  79     png_ptr->read_data_fn = read_func;
  80 
  81     png_set_sig_bytes(png_ptr, SIG_BYTES);      /* we already read the 8 signature bytes */
  82 
  83     png_read_info(png_ptr, info_ptr);   /* read all PNG info up to image data */
  84 
  85     png_get_IHDR(png_ptr, info_ptr, &width, &height, &bit_depth, &color_type,
  86         NULL, NULL, NULL);
  87 
  88     /* expand palette images to RGB, low-bit-depth grayscale images to 8 bits,
  89      * transparency chunks to full alpha channel; strip 16-bit-per-sample
  90      * images to 8 bits per sample; and convert grayscale to RGB[A]
  91      * this may be sub-optimal but this simplifies implementation */
  92 
  93     png_set_expand(png_ptr);
  94     png_set_tRNS_to_alpha(png_ptr);
  95     png_set_filler(png_ptr, 0xff, PNG_FILLER_AFTER);
  96     png_set_strip_16(png_ptr);
  97     png_set_gray_to_rgb(png_ptr);
  98 
  99     if (png_get_gAMA(png_ptr, info_ptr, &gamma))
 100         png_set_gamma(png_ptr, 2.2, gamma);
 101 
 102     png_read_update_info(png_ptr, info_ptr);
 103 
 104     rowbytes = png_get_rowbytes(png_ptr, info_ptr);
 105 
 106     if (!SAFE_TO_ALLOC(rowbytes, height)) {
 107         goto done;
 108     }
 109 
 110     if ((image_data = (unsigned char *) malloc(rowbytes * height)) == NULL) {
 111         goto done;
 112     }
 113 
 114     if (!SAFE_TO_ALLOC(height, sizeof(png_bytep))) {
 115         goto done;
 116     }
 117     if ((row_pointers = (png_bytepp) malloc(height * sizeof(png_bytep)))
 118             == NULL) {
 119         goto done;
 120     }
 121 
 122     for (i = 0; i < height; ++i)
 123         row_pointers[i] = image_data + i * rowbytes;
 124 
 125     png_read_image(png_ptr, row_pointers);
 126 
 127     SplashCleanup(splash);
 128 
 129     splash->width = width;
 130     splash->height = height;
 131 
 132     if (!SAFE_TO_ALLOC(splash->width, splash->imageFormat.depthBytes)) {
 133         goto done;
 134     }
 135     stride = splash->width * splash->imageFormat.depthBytes;
 136 
 137     if (!SAFE_TO_ALLOC(splash->height, stride)) {
 138         goto done;
 139     }
 140     splash->frameCount = 1;
 141     splash->frames = (SplashImage *)
 142         malloc(sizeof(SplashImage) * splash->frameCount);
 143 
 144     if (splash->frames == NULL) {
 145         goto done;
 146     }
 147 
 148     splash->loopCount = 1;
 149     splash->frames[0].bitmapBits = malloc(stride * splash->height);
 150     if (splash->frames[0].bitmapBits == NULL) {
 151         free(splash->frames);
 152         goto done;
 153     }
 154     splash->frames[0].delay = 0;
 155 
 156     /* FIXME: sort out the real format */
 157     initFormat(&srcFormat, 0xFF000000, 0x00FF0000, 0x0000FF00, 0x000000FF);
 158     srcFormat.byteOrder = BYTE_ORDER_MSBFIRST;
 159 
 160     initRect(&srcRect, 0, 0, width, height, 1, rowbytes,
 161         image_data, &srcFormat);
 162     initRect(&dstRect, 0, 0, width, height, 1, stride,
 163         splash->frames[0].bitmapBits, &splash->imageFormat);
 164     convertRect(&srcRect, &dstRect, CVT_COPY);
 165 
 166     SplashInitFrameShape(splash, 0);
 167 
 168     png_read_end(png_ptr, NULL);
 169     success = 1;
 170 
 171   done:
 172     free(row_pointers);
 173     free(image_data);
 174     png_destroy_read_struct(&png_ptr, &info_ptr, NULL);
 175     return success;
 176 }
 177 
 178 int
 179 SplashDecodePngStream(Splash * splash, SplashStream * stream)
 180 {
 181     unsigned char sig[SIG_BYTES];
 182     int success = 0;
 183 
 184     stream->read(stream, sig, SIG_BYTES);
 185     if (png_sig_cmp(sig, 0, SIG_BYTES)) {
 186         goto done;
 187     }
 188     success = SplashDecodePng(splash, my_png_read_stream, stream);
 189 
 190   done:
 191     return success;
 192 }