From c91847d549cc1c30eb15504a15ea9a6d5aa48165 Mon Sep 17 00:00:00 2001 From: "chirs241097@gmail.com" Date: Sun, 12 Jan 2014 14:43:14 +0000 Subject: --- hge/CxImage/ximapng.cpp | 644 ++++++++++++++++++++++++++++++++++++++++++++++++ 1 file changed, 644 insertions(+) create mode 100644 hge/CxImage/ximapng.cpp (limited to 'hge/CxImage/ximapng.cpp') diff --git a/hge/CxImage/ximapng.cpp b/hge/CxImage/ximapng.cpp new file mode 100644 index 0000000..b732491 --- /dev/null +++ b/hge/CxImage/ximapng.cpp @@ -0,0 +1,644 @@ +/* + * File: ximapng.cpp + * Purpose: Platform Independent PNG Image Class Loader and Writer + * 07/Aug/2001 Davide Pizzolato - www.xdp.it + * CxImage version 7.0.0 31/Dec/2010 + */ + +#include "ximapng.h" + +#if CXIMAGE_SUPPORT_PNG + +#include "ximaiter.h" + +//////////////////////////////////////////////////////////////////////////////// +void CxImagePNG::ima_png_error(png_struct *png_ptr, char *message) +{ + strcpy(info.szLastError,message); +#if PNG_LIBPNG_VER > 10399 + longjmp(png_jmpbuf(png_ptr), 1); +#else + longjmp(png_ptr->jmpbuf, 1); +#endif +} +//////////////////////////////////////////////////////////////////////////////// +#if CXIMAGE_SUPPORT_DECODE +//////////////////////////////////////////////////////////////////////////////// +void CxImagePNG::expand2to4bpp(uint8_t* prow) +{ + uint8_t *psrc,*pdst; + uint8_t pos,idx; + for(int32_t x=head.biWidth-1;x>=0;x--){ + psrc = prow + ((2*x)>>3); + pdst = prow + ((4*x)>>3); + pos = (uint8_t)(2*(3-x%4)); + idx = (uint8_t)((*psrc & (0x03<>pos); + pos = (uint8_t)(4*(1-x%2)); + *pdst &= ~(0x0F< 10399 + if (setjmp(png_jmpbuf(png_ptr))) { +#else + if (setjmp(png_ptr->jmpbuf)) { +#endif + /* Free all of the memory associated with the png_ptr and info_ptr */ + delete [] row_pointers; + png_destroy_read_struct(&png_ptr, &info_ptr, (png_infopp)NULL); + cx_throw(""); + } + + // use custom I/O functions + png_set_read_fn(png_ptr, hFile, /*(png_rw_ptr)*/user_read_data); + png_set_error_fn(png_ptr,info.szLastError,/*(png_error_ptr)*/user_error_fn,NULL); + + /* read the file information */ + png_read_info(png_ptr, info_ptr); + + png_uint_32 _width,_height; + int _bit_depth,_color_type,_interlace_type,_compression_type,_filter_type; +#if PNG_LIBPNG_VER > 10399 + png_get_IHDR(png_ptr,info_ptr,&_width,&_height,&_bit_depth,&_color_type, + &_interlace_type,&_compression_type,&_filter_type); +#else + _width=info_ptr->width; + _height=info_ptr->height; + _bit_depth=info_ptr->bit_depth; + _color_type=info_ptr->color_type; + _interlace_type=info_ptr->interlace_type; + _compression_type=info_ptr->compression_type; + _filter_type=info_ptr->filter_type; +#endif + + if (info.nEscape == -1){ + head.biWidth = _width; + head.biHeight= _height; + info.dwType = CXIMAGE_FORMAT_PNG; +#if PNG_LIBPNG_VER > 10399 + longjmp(png_jmpbuf(png_ptr), 1); +#else + longjmp(png_ptr->jmpbuf, 1); +#endif + } + + /* calculate new number of channels */ + int32_t channels=0; + switch(_color_type){ + case PNG_COLOR_TYPE_GRAY: + case PNG_COLOR_TYPE_PALETTE: + channels = 1; + break; + case PNG_COLOR_TYPE_GRAY_ALPHA: + channels = 2; + break; + case PNG_COLOR_TYPE_RGB: + channels = 3; + break; + case PNG_COLOR_TYPE_RGB_ALPHA: + channels = 4; + break; + default: + strcpy(info.szLastError,"unknown PNG color type"); +#if PNG_LIBPNG_VER > 10399 + longjmp(png_jmpbuf(png_ptr), 1); +#else + longjmp(png_ptr->jmpbuf, 1); +#endif + } + + //find the right pixel depth used for cximage +#if PNG_LIBPNG_VER > 10399 + int pixel_depth = _bit_depth * png_get_channels(png_ptr,info_ptr); +#else + int pixel_depth = info_ptr->pixel_depth; +#endif + if (channels == 1 && pixel_depth>8) pixel_depth=8; + if (channels == 2) pixel_depth=8; + if (channels >= 3) pixel_depth=24; + + if (!Create(_width, _height, pixel_depth, CXIMAGE_FORMAT_PNG)){ +#if PNG_LIBPNG_VER > 10399 + longjmp(png_jmpbuf(png_ptr), 1); +#else + longjmp(png_ptr->jmpbuf, 1); +#endif + } + + /* get metrics */ + png_uint_32 _x_pixels_per_unit,_y_pixels_per_unit; + int _phys_unit_type; +#if PNG_LIBPNG_VER > 10399 + png_get_pHYs(png_ptr,info_ptr,&_x_pixels_per_unit,&_y_pixels_per_unit,&_phys_unit_type); +#else + _x_pixels_per_unit=info_ptr->x_pixels_per_unit; + _y_pixels_per_unit=info_ptr->y_pixels_per_unit; + _phys_unit_type=info_ptr->phys_unit_type; +#endif + switch (_phys_unit_type) + { + case PNG_RESOLUTION_UNKNOWN: + SetXDPI(_x_pixels_per_unit); + SetYDPI(_y_pixels_per_unit); + break; + case PNG_RESOLUTION_METER: + SetXDPI((long)floor(_x_pixels_per_unit * 254.0 / 10000.0 + 0.5)); + SetYDPI((long)floor(_y_pixels_per_unit * 254.0 / 10000.0 + 0.5)); + break; + } + + int _num_palette; + png_colorp _palette; +#if PNG_LIBPNG_VER > 10399 + png_get_PLTE(png_ptr,info_ptr,&_palette,&_num_palette); +#else + _num_palette=info_ptr->num_palette; + _palette=info_ptr->palette; +#endif + if (_num_palette>0){ + SetPalette((rgb_color*)_palette,_num_palette); + SetClrImportant(_num_palette); + } else if (_bit_depth ==2) { // needed for 2 bpp grayscale PNGs + SetPaletteColor(0,0,0,0); + SetPaletteColor(1,85,85,85); + SetPaletteColor(2,170,170,170); + SetPaletteColor(3,255,255,255); + } else SetGrayPalette(); // needed for grayscale PNGs + + int nshift = max(0,(_bit_depth>>3)-1)<<3; + + png_bytep _trans_alpha; + int _num_trans; + png_color_16p _trans_color; +#if PNG_LIBPNG_VER > 10399 + png_get_tRNS(png_ptr,info_ptr,&_trans_alpha,&_num_trans,&_trans_color); +#else + _num_trans=info_ptr->num_trans; +#endif + if (_num_trans!=0){ //palette transparency + if (_num_trans==1){ + if (_color_type == PNG_COLOR_TYPE_PALETTE){ + info.nBkgndIndex = _trans_color->index; + } else{ + info.nBkgndIndex = _trans_color->gray>>nshift; + } + } + if (_num_trans>1){ + RGBQUAD* pal=GetPalette(); + if (pal){ + uint32_t ip; + if(_trans_alpha != 0) { + for (ip=0;ipred>>nshift); + info.nBkgndColor.rgbGreen = (uint8_t)(_trans_color->green>>nshift); + info.nBkgndColor.rgbBlue = (uint8_t)(_trans_color->blue>>nshift); + info.nBkgndColor.rgbReserved = 0; + info.nBkgndIndex = 0; + } + } + + int32_t alpha_present = (channels - 1) % 2; + if (alpha_present){ +#if CXIMAGE_SUPPORT_ALPHA // + AlphaCreate(); +#else + png_set_strip_alpha(png_ptr); +#endif //CXIMAGE_SUPPORT_ALPHA + } + + // - flip the RGB pixels to BGR (or RGBA to BGRA) + if (_color_type & PNG_COLOR_MASK_COLOR){ + png_set_bgr(png_ptr); + } + + // - handle cancel + if (info.nEscape) +#if PNG_LIBPNG_VER > 10399 + longjmp(png_jmpbuf(png_ptr), 1); +#else + longjmp(png_ptr->jmpbuf, 1); +#endif + + // row_bytes is the width x number of channels x (bit-depth / 8) +#if PNG_LIBPNG_VER > 10399 + row_pointers = new uint8_t[png_get_rowbytes(png_ptr,info_ptr) + 8]; +#else + row_pointers = new uint8_t[info_ptr->rowbytes + 8]; +#endif + + + // turn on interlace handling + int32_t number_passes = png_set_interlace_handling(png_ptr); + + if (number_passes>1){ + SetCodecOption( (ENCODE_INTERLACE) | GetCodecOption(CXIMAGE_FORMAT_PNG)); + } else { + SetCodecOption(~(ENCODE_INTERLACE) & GetCodecOption(CXIMAGE_FORMAT_PNG)); + } + + int chan_offset = _bit_depth >> 3; +#if PNG_LIBPNG_VER > 10399 + int pixel_offset = (_bit_depth * png_get_channels(png_ptr,info_ptr)) >> 3; +#else + int pixel_offset = info_ptr->pixel_depth >> 3; +#endif + + for (int32_t pass=0; pass < number_passes; pass++) { + iter.Upset(); + int32_t y=0; + do { + + // - handle cancel + if (info.nEscape) +#if PNG_LIBPNG_VER > 10399 + longjmp(png_jmpbuf(png_ptr), 1); +#else + longjmp(png_ptr->jmpbuf, 1); +#endif + +#if CXIMAGE_SUPPORT_ALPHA // + if (AlphaIsValid()) { + + //compute the correct position of the line + int32_t ax,ay; + ay = head.biHeight-1-y; + uint8_t* prow= iter.GetRow(ay); + + //recover data from previous scan + if (_interlace_type && pass>0 && pass!=7){ + for(ax=0;ax RGB + A + for(ax=0;ax0){ +#if PNG_LIBPNG_VER > 10399 + iter.GetRow(row_pointers, png_get_rowbytes(png_ptr,info_ptr)); +#else + iter.GetRow(row_pointers, info_ptr->rowbytes); +#endif + //re-expand buffer for images with bit depth > 8 + if (_bit_depth > 8){ + for(int32_t ax=(head.biWidth*channels-1);ax>=0;ax--) + row_pointers[ax*chan_offset] = row_pointers[ax]; + } + } + + //read next row + png_read_row(png_ptr, row_pointers, NULL); + + //shrink 16 bit depth images down to 8 bits + if (_bit_depth > 8){ + for(int32_t ax=0;ax<(head.biWidth*channels);ax++) + row_pointers[ax] = row_pointers[ax*chan_offset]; + } + + //copy the pixels +#if PNG_LIBPNG_VER > 10399 + iter.SetRow(row_pointers, png_get_rowbytes(png_ptr,info_ptr)); +#else + iter.SetRow(row_pointers, info_ptr->rowbytes); +#endif + // expand 2 bpp images only in the last pass + if (_bit_depth==2 && pass==(number_passes-1)) + expand2to4bpp(iter.GetRow()); + + //go on + iter.PrevRow(); + } + + y++; + } while(yjmpbuf)){ + /* If we get here, we had a problem reading the file */ + if (info_ptr->palette) free(info_ptr->palette); + png_destroy_write_struct(&png_ptr, (png_infopp)&info_ptr); + cx_throw("Error saving PNG file"); + } + + /* set up the output control */ + //png_init_io(png_ptr, hFile); + + // use custom I/O functions + png_set_write_fn(png_ptr,hFile,/*(png_rw_ptr)*/user_write_data,/*(png_flush_ptr)*/user_flush_data); + + /* set the file information here */ + info_ptr->width = GetWidth(); + info_ptr->height = GetHeight(); + info_ptr->pixel_depth = (uint8_t)GetBpp(); + info_ptr->channels = (GetBpp()>8) ? (uint8_t)3: (uint8_t)1; + info_ptr->bit_depth = (uint8_t)(GetBpp()/info_ptr->channels); + info_ptr->compression_type = info_ptr->filter_type = 0; + info_ptr->valid = 0; + + // set interlace type + DWORD codec_opt = GetCodecOption(CXIMAGE_FORMAT_PNG); + if (codec_opt & CxImagePNG::ENCODE_INTERLACE) + info_ptr->interlace_type = PNG_INTERLACE_ADAM7; + else + info_ptr->interlace_type = PNG_INTERLACE_NONE; + + /* set compression level */ + int32_t compress_level; + switch (codec_opt & CxImagePNG::ENCODE_COMPRESSION_MASK) + { + case ENCODE_NO_COMPRESSION: + compress_level = Z_NO_COMPRESSION; + break; + case ENCODE_BEST_SPEED: + compress_level = Z_BEST_SPEED; + break; + case ENCODE_BEST_COMPRESSION: + compress_level = Z_BEST_COMPRESSION; + break; + default: + compress_level = Z_DEFAULT_COMPRESSION; + break; + } + png_set_compression_level(png_ptr, compress_level); + + bool bGrayScale = IsGrayScale(); + + if (GetNumColors()){ + if (bGrayScale){ + info_ptr->color_type = PNG_COLOR_TYPE_GRAY; + } else { + info_ptr->color_type = PNG_COLOR_TYPE_PALETTE; + } + } else { + info_ptr->color_type = PNG_COLOR_TYPE_RGB; + } +#if CXIMAGE_SUPPORT_ALPHA + if (AlphaIsValid()){ + info_ptr->color_type |= PNG_COLOR_MASK_ALPHA; + info_ptr->channels++; + info_ptr->bit_depth = 8; + info_ptr->pixel_depth += 8; + } +#endif + + /* set background */ + png_color_16 image_background={ 0, 255, 255, 255, 0 }; + RGBQUAD tc = GetTransColor(); + if (info.nBkgndIndex>=0) { + image_background.blue = tc.rgbBlue; + image_background.green = tc.rgbGreen; + image_background.red = tc.rgbRed; + } + png_set_bKGD(png_ptr, info_ptr, &image_background); + + /* set metrics */ + png_set_pHYs(png_ptr, info_ptr, head.biXPelsPerMeter, head.biYPelsPerMeter, PNG_RESOLUTION_METER); + + png_set_IHDR(png_ptr, info_ptr, info_ptr->width, info_ptr->height, info_ptr->bit_depth, + info_ptr->color_type, info_ptr->interlace_type, + PNG_COMPRESSION_TYPE_BASE, PNG_FILTER_TYPE_BASE); + + // simple transparency + if (info.nBkgndIndex >= 0){ + info_ptr->num_trans = 1; + info_ptr->valid |= PNG_INFO_tRNS; + info_ptr->trans_alpha = trans; + info_ptr->trans_color.index = (uint8_t)info.nBkgndIndex; + info_ptr->trans_color.red = tc.rgbRed; + info_ptr->trans_color.green = tc.rgbGreen; + info_ptr->trans_color.blue = tc.rgbBlue; + info_ptr->trans_color.gray = info_ptr->trans_color.index; + + // the transparency indexes start from 0 for non grayscale palette + if (!bGrayScale && head.biClrUsed && info.nBkgndIndex) + SwapIndex(0,(uint8_t)info.nBkgndIndex); + } + + /* set the palette if there is one */ + if (GetPalette()){ + if (!bGrayScale){ + info_ptr->valid |= PNG_INFO_PLTE; + } + + int32_t nc = GetClrImportant(); + if (nc==0) nc = GetNumColors(); + + if (info.bAlphaPaletteEnabled){ + for(uint16_t ip=0; ipnum_trans = (uint16_t)nc; + info_ptr->valid |= PNG_INFO_tRNS; + info_ptr->trans_alpha = trans; + } + + // copy the palette colors + info_ptr->palette = new png_color[nc]; + info_ptr->num_palette = (png_uint_16) nc; + for (int32_t i=0; ipalette[i].red, &info_ptr->palette[i].green, &info_ptr->palette[i].blue); + } + +#if CXIMAGE_SUPPORT_ALPHA // + //Merge the transparent color with the alpha channel + if (AlphaIsValid() && head.biBitCount==24 && info.nBkgndIndex>=0){ + for(int32_t y=0; y < head.biHeight; y++){ + for(int32_t x=0; x < head.biWidth ; x++){ + RGBQUAD c=GetPixelColor(x,y,false); + if (*(int32_t*)&c==*(int32_t*)&tc) + AlphaSet(x,y,0); + } } } +#endif // CXIMAGE_SUPPORT_ALPHA // + + int32_t row_size = max(info.dwEffWidth, info_ptr->width*info_ptr->channels*(info_ptr->bit_depth/8)); + info_ptr->rowbytes = row_size; + uint8_t *row_pointers = new uint8_t[row_size]; + + /* write the file information */ + png_write_info(png_ptr, info_ptr); + + //interlace handling + int32_t num_pass = png_set_interlace_handling(png_ptr); + for (int32_t pass = 0; pass < num_pass; pass++){ + //write image + iter.Upset(); + int32_t ay=head.biHeight-1; + RGBQUAD c; + do { +#if CXIMAGE_SUPPORT_ALPHA // + if (AlphaIsValid()){ + for (int32_t ax=head.biWidth-1; ax>=0;ax--){ + c = BlindGetPixelColor(ax,ay); + int32_t px = ax * info_ptr->channels; + if (!bGrayScale){ + row_pointers[px++]=c.rgbRed; + row_pointers[px++]=c.rgbGreen; + } + row_pointers[px++]=c.rgbBlue; + row_pointers[px] = AlphaGet(ax,ay); + } + png_write_row(png_ptr, row_pointers); + ay--; + } + else +#endif //CXIMAGE_SUPPORT_ALPHA // + { + iter.GetRow(row_pointers, row_size); + if (info_ptr->color_type == PNG_COLOR_TYPE_RGB) //HACK BY OP + RGBtoBGR(row_pointers, row_size); + png_write_row(png_ptr, row_pointers); + } + } while(iter.PrevRow()); + } + + delete [] row_pointers; + row_pointers = NULL; + + //if necessary, restore the original palette + if (!bGrayScale && head.biClrUsed && info.nBkgndIndex>0) + SwapIndex((uint8_t)info.nBkgndIndex,0); + + /* It is REQUIRED to call this to finish writing the rest of the file */ + png_write_end(png_ptr, info_ptr); + + /* if you malloced the palette, free it here */ + if (info_ptr->palette){ + delete [] (info_ptr->palette); + info_ptr->palette = NULL; + } + + /* clean up after the write, and free any memory allocated */ + png_destroy_write_struct(&png_ptr, (png_infopp)&info_ptr); + + } cx_catch { + if (strcmp(message,"")) strncpy(info.szLastError,message,255); + return FALSE; + } + /* that's it */ + return TRUE; +} +//////////////////////////////////////////////////////////////////////////////// +#endif // CXIMAGE_SUPPORT_ENCODE +//////////////////////////////////////////////////////////////////////////////// +#endif // CXIMAGE_SUPPORT_PNG -- cgit v1.2.3