summaryrefslogtreecommitdiff
path: root/plugins/AdvaImg/src/FreeImage/PluginPNG.cpp
diff options
context:
space:
mode:
authorGeorge Hazan <ghazan@miranda.im>2018-01-26 17:38:31 +0300
committerGeorge Hazan <ghazan@miranda.im>2018-01-26 17:38:31 +0300
commitdea9c030340e50324eba97c72a27c151bed12e1c (patch)
tree6fc156f40f52a9fc6e6b29e60001959477ee1a5e /plugins/AdvaImg/src/FreeImage/PluginPNG.cpp
parentc6e8f8223cab9d799593b7b2cfa22134aa9745d6 (diff)
AdvaImg:
- freeimage extracted to the separate library; - FI_INTERFACE removed, all references to it are replaced with direct calls of FreeImage_* functions; - unified project for AdvaImg
Diffstat (limited to 'plugins/AdvaImg/src/FreeImage/PluginPNG.cpp')
-rw-r--r--plugins/AdvaImg/src/FreeImage/PluginPNG.cpp1115
1 files changed, 0 insertions, 1115 deletions
diff --git a/plugins/AdvaImg/src/FreeImage/PluginPNG.cpp b/plugins/AdvaImg/src/FreeImage/PluginPNG.cpp
deleted file mode 100644
index fe80a2b533..0000000000
--- a/plugins/AdvaImg/src/FreeImage/PluginPNG.cpp
+++ /dev/null
@@ -1,1115 +0,0 @@
-// ==========================================================
-// PNG Loader and Writer
-//
-// Design and implementation by
-// - Floris van den Berg (flvdberg@wxs.nl)
-// - Herve Drolon (drolon@infonie.fr)
-// - Detlev Vendt (detlev.vendt@brillit.de)
-// - Aaron Shumate (trek@startreker.com)
-// - Tanner Helland (tannerhelland@users.sf.net)
-//
-// This file is part of FreeImage 3
-//
-// COVERED CODE IS PROVIDED UNDER THIS LICENSE ON AN "AS IS" BASIS, WITHOUT WARRANTY
-// OF ANY KIND, EITHER EXPRESSED OR IMPLIED, INCLUDING, WITHOUT LIMITATION, WARRANTIES
-// THAT THE COVERED CODE IS FREE OF DEFECTS, MERCHANTABLE, FIT FOR A PARTICULAR PURPOSE
-// OR NON-INFRINGING. THE ENTIRE RISK AS TO THE QUALITY AND PERFORMANCE OF THE COVERED
-// CODE IS WITH YOU. SHOULD ANY COVERED CODE PROVE DEFECTIVE IN ANY RESPECT, YOU (NOT
-// THE INITIAL DEVELOPER OR ANY OTHER CONTRIBUTOR) ASSUME THE COST OF ANY NECESSARY
-// SERVICING, REPAIR OR CORRECTION. THIS DISCLAIMER OF WARRANTY CONSTITUTES AN ESSENTIAL
-// PART OF THIS LICENSE. NO USE OF ANY COVERED CODE IS AUTHORIZED HEREUNDER EXCEPT UNDER
-// THIS DISCLAIMER.
-//
-// Use at your own risk!
-// ==========================================================
-
-#ifdef _MSC_VER
-#pragma warning (disable : 4786) // identifier was truncated to 'number' characters
-#endif
-
-#include "FreeImage.h"
-#include "Utilities.h"
-
-#include "../Metadata/FreeImageTag.h"
-
-// ----------------------------------------------------------
-
-#define PNG_BYTES_TO_CHECK 8
-
-#undef PNG_Z_DEFAULT_COMPRESSION // already used in ../LibPNG/pnglibconf.h
-
-// ----------------------------------------------------------
-
-#include "zlib.h"
-#include "../LibPNG/png.h"
-
-// ----------------------------------------------------------
-
-typedef struct {
- FreeImageIO *s_io;
- fi_handle s_handle;
-} fi_ioStructure, *pfi_ioStructure;
-
-// ==========================================================
-// libpng interface
-// ==========================================================
-
-static void
-_ReadProc(png_structp png_ptr, unsigned char *data, png_size_t size) {
- pfi_ioStructure pfio = (pfi_ioStructure)png_get_io_ptr(png_ptr);
- unsigned n = pfio->s_io->read_proc(data, (unsigned int)size, 1, pfio->s_handle);
- if(size && (n == 0)) {
- throw "Read error: invalid or corrupted PNG file";
- }
-}
-
-static void
-_WriteProc(png_structp png_ptr, unsigned char *data, png_size_t size) {
- pfi_ioStructure pfio = (pfi_ioStructure)png_get_io_ptr(png_ptr);
- pfio->s_io->write_proc(data, (unsigned int)size, 1, pfio->s_handle);
-}
-
-static void
-_FlushProc(png_structp png_ptr) {
- (png_structp)png_ptr;
- // empty flush implementation
-}
-
-static void
-error_handler(png_structp png_ptr, const char *error) {
- (png_structp)png_ptr;
- throw error;
-}
-
-// in FreeImage warnings disabled
-
-static void
-warning_handler(png_structp png_ptr, const char *warning) {
- (png_structp)png_ptr;
- (char*)warning;
-}
-
-// ==========================================================
-// Metadata routines
-// ==========================================================
-
-static BOOL
-ReadMetadata(png_structp png_ptr, png_infop info_ptr, FIBITMAP *dib) {
- // XMP keyword
- const char *g_png_xmp_keyword = "XML:com.adobe.xmp";
-
- FITAG *tag = NULL;
- png_textp text_ptr = NULL;
- png_timep mod_time = NULL;
- int num_text = 0;
-
- // iTXt/tEXt/zTXt chuncks
- if(png_get_text(png_ptr, info_ptr, &text_ptr, &num_text) > 0) {
- for(int i = 0; i < num_text; i++) {
- // create a tag
- tag = FreeImage_CreateTag();
- if(!tag) return FALSE;
-
- DWORD tag_length = (DWORD) MAX(text_ptr[i].text_length, text_ptr[i].itxt_length);
-
- FreeImage_SetTagLength(tag, tag_length);
- FreeImage_SetTagCount(tag, tag_length);
- FreeImage_SetTagType(tag, FIDT_ASCII);
- FreeImage_SetTagValue(tag, text_ptr[i].text);
-
- if(strcmp(text_ptr[i].key, g_png_xmp_keyword) == 0) {
- // store the tag as XMP
- FreeImage_SetTagKey(tag, g_TagLib_XMPFieldName);
- FreeImage_SetMetadata(FIMD_XMP, dib, FreeImage_GetTagKey(tag), tag);
- } else {
- // store the tag as a comment
- FreeImage_SetTagKey(tag, text_ptr[i].key);
- FreeImage_SetMetadata(FIMD_COMMENTS, dib, FreeImage_GetTagKey(tag), tag);
- }
-
- // destroy the tag
- FreeImage_DeleteTag(tag);
- }
- }
-
- // timestamp chunk
- if(png_get_tIME(png_ptr, info_ptr, &mod_time)) {
- char timestamp[32];
- // create a tag
- tag = FreeImage_CreateTag();
- if(!tag) return FALSE;
-
- // convert as 'yyyy:MM:dd hh:mm:ss'
- sprintf(timestamp, "%4d:%02d:%02d %2d:%02d:%02d", mod_time->year, mod_time->month, mod_time->day, mod_time->hour, mod_time->minute, mod_time->second);
-
- DWORD tag_length = (DWORD)strlen(timestamp) + 1;
- FreeImage_SetTagLength(tag, tag_length);
- FreeImage_SetTagCount(tag, tag_length);
- FreeImage_SetTagType(tag, FIDT_ASCII);
- FreeImage_SetTagID(tag, TAG_DATETIME);
- FreeImage_SetTagValue(tag, timestamp);
-
- // store the tag as Exif-TIFF
- FreeImage_SetTagKey(tag, "DateTime");
- FreeImage_SetMetadata(FIMD_EXIF_MAIN, dib, FreeImage_GetTagKey(tag), tag);
-
- // destroy the tag
- FreeImage_DeleteTag(tag);
- }
-
- return TRUE;
-}
-
-static BOOL
-WriteMetadata(png_structp png_ptr, png_infop info_ptr, FIBITMAP *dib) {
- // XMP keyword
- const char *g_png_xmp_keyword = "XML:com.adobe.xmp";
-
- FITAG *tag = NULL;
- FIMETADATA *mdhandle = NULL;
- BOOL bResult = TRUE;
-
- png_text text_metadata;
- png_time mod_time;
-
- // set the 'Comments' metadata as iTXt chuncks
-
- mdhandle = FreeImage_FindFirstMetadata(FIMD_COMMENTS, dib, &tag);
-
- if(mdhandle) {
- do {
- memset(&text_metadata, 0, sizeof(png_text));
- text_metadata.compression = 1; // iTXt, none
- text_metadata.key = (char*)FreeImage_GetTagKey(tag); // keyword, 1-79 character description of "text"
- text_metadata.text = (char*)FreeImage_GetTagValue(tag); // comment, may be an empty string (ie "")
- text_metadata.text_length = FreeImage_GetTagLength(tag);// length of the text string
- text_metadata.itxt_length = FreeImage_GetTagLength(tag);// length of the itxt string
- text_metadata.lang = 0; // language code, 0-79 characters or a NULL pointer
- text_metadata.lang_key = 0; // keyword translated UTF-8 string, 0 or more chars or a NULL pointer
-
- // set the tag
- png_set_text(png_ptr, info_ptr, &text_metadata, 1);
-
- } while(FreeImage_FindNextMetadata(mdhandle, &tag));
-
- FreeImage_FindCloseMetadata(mdhandle);
- bResult &= TRUE;
- }
-
- // set the 'XMP' metadata as iTXt chuncks
- tag = NULL;
- FreeImage_GetMetadata(FIMD_XMP, dib, g_TagLib_XMPFieldName, &tag);
- if(tag && FreeImage_GetTagLength(tag)) {
- memset(&text_metadata, 0, sizeof(png_text));
- text_metadata.compression = 1; // iTXt, none
- text_metadata.key = (char*)g_png_xmp_keyword; // keyword, 1-79 character description of "text"
- text_metadata.text = (char*)FreeImage_GetTagValue(tag); // comment, may be an empty string (ie "")
- text_metadata.text_length = FreeImage_GetTagLength(tag);// length of the text string
- text_metadata.itxt_length = FreeImage_GetTagLength(tag);// length of the itxt string
- text_metadata.lang = 0; // language code, 0-79 characters or a NULL pointer
- text_metadata.lang_key = 0; // keyword translated UTF-8 string, 0 or more chars or a NULL pointer
-
- // set the tag
- png_set_text(png_ptr, info_ptr, &text_metadata, 1);
- bResult &= TRUE;
- }
-
- // set the Exif-TIFF 'DateTime' metadata as a tIME chunk
- tag = NULL;
- FreeImage_GetMetadata(FIMD_EXIF_MAIN, dib, "DateTime", &tag);
- if(tag && FreeImage_GetTagLength(tag)) {
- int year, month, day, hour, minute, second;
- const char *value = (char*)FreeImage_GetTagValue(tag);
- if(sscanf(value, "%4d:%02d:%02d %2d:%02d:%02d", &year, &month, &day, &hour, &minute, &second) == 6) {
- mod_time.year = (png_uint_16)year;
- mod_time.month = (png_byte)month;
- mod_time.day = (png_byte)day;
- mod_time.hour = (png_byte)hour;
- mod_time.minute = (png_byte)minute;
- mod_time.second = (png_byte)second;
- png_set_tIME (png_ptr, info_ptr, &mod_time);
- }
- }
-
- return bResult;
-}
-
-// ==========================================================
-// Plugin Interface
-// ==========================================================
-
-static int s_format_id;
-
-// ==========================================================
-// Plugin Implementation
-// ==========================================================
-
-static const char * DLL_CALLCONV
-Format() {
- return "PNG";
-}
-
-static const char * DLL_CALLCONV
-Description() {
- return "Portable Network Graphics";
-}
-
-static const char * DLL_CALLCONV
-Extension() {
- return "png";
-}
-
-static const char * DLL_CALLCONV
-RegExpr() {
- return "^.PNG\r";
-}
-
-static const char * DLL_CALLCONV
-MimeType() {
- return "image/png";
-}
-
-static BOOL DLL_CALLCONV
-Validate(FreeImageIO *io, fi_handle handle) {
- BYTE png_signature[8] = { 137, 80, 78, 71, 13, 10, 26, 10 };
- BYTE signature[8] = { 0, 0, 0, 0, 0, 0, 0, 0 };
-
- io->read_proc(&signature, 1, 8, handle);
-
- return (memcmp(png_signature, signature, 8) == 0);
-}
-
-static BOOL DLL_CALLCONV
-SupportsExportDepth(int depth) {
- return (
- (depth == 1) ||
- (depth == 4) ||
- (depth == 8) ||
- (depth == 24) ||
- (depth == 32)
- );
-}
-
-static BOOL DLL_CALLCONV
-SupportsExportType(FREE_IMAGE_TYPE type) {
- return (
- (type == FIT_BITMAP) ||
- (type == FIT_UINT16) ||
- (type == FIT_RGB16) ||
- (type == FIT_RGBA16)
- );
-}
-
-static BOOL DLL_CALLCONV
-SupportsICCProfiles() {
- return TRUE;
-}
-
-static BOOL DLL_CALLCONV
-SupportsNoPixels() {
- return TRUE;
-}
-
-// --------------------------------------------------------------------------
-
-/**
-Configure the decoder so that decoded pixels are compatible with a FREE_IMAGE_TYPE format.
-Set conversion instructions as needed.
-@param png_ptr PNG handle
-@param info_ptr PNG info handle
-@param flags Decoder flags
-@param output_image_type Returned FreeImage converted image type
-@return Returns TRUE if successful, returns FALSE otherwise
-@see png_read_update_info
-*/
-static BOOL
-ConfigureDecoder(png_structp png_ptr, png_infop info_ptr, int flags, FREE_IMAGE_TYPE *output_image_type) {
- // get original image info
- const int color_type = png_get_color_type(png_ptr, info_ptr);
- const int bit_depth = png_get_bit_depth(png_ptr, info_ptr);
- const int pixel_depth = bit_depth * png_get_channels(png_ptr, info_ptr);
-
- FREE_IMAGE_TYPE image_type = FIT_BITMAP; // assume standard image type
-
- // check for transparency table or single transparent color
- BOOL bIsTransparent = png_get_valid(png_ptr, info_ptr, PNG_INFO_tRNS) == PNG_INFO_tRNS ? TRUE : FALSE;
-
- // check allowed combinations of colour type and bit depth
- // then get converted FreeImage type
-
- switch(color_type) {
- case PNG_COLOR_TYPE_GRAY: // color type '0', bitdepth = 1, 2, 4, 8, 16
- switch(bit_depth) {
- case 1:
- case 2:
- case 4:
- case 8:
- // expand grayscale images to the full 8-bit from 2-bit/pixel
- if (pixel_depth == 2) {
- png_set_expand_gray_1_2_4_to_8(png_ptr);
- }
-
- // if a tRNS chunk is provided, we must also expand the grayscale data to 8-bits,
- // this allows us to make use of the transparency table with existing FreeImage methods
- if (bIsTransparent && (pixel_depth < 8)) {
- png_set_expand_gray_1_2_4_to_8(png_ptr);
- }
- break;
-
- case 16:
- image_type = (pixel_depth == 16) ? FIT_UINT16 : FIT_UNKNOWN;
-
- // 16-bit grayscale images can contain a transparent value (shade)
- // if found, expand the transparent value to a full alpha channel
- if (bIsTransparent && (image_type != FIT_UNKNOWN)) {
- // expand tRNS to a full alpha channel
- png_set_tRNS_to_alpha(png_ptr);
-
- // expand new 16-bit gray + 16-bit alpha to full 64-bit RGBA
- png_set_gray_to_rgb(png_ptr);
-
- image_type = FIT_RGBA16;
- }
- break;
-
- default:
- image_type = FIT_UNKNOWN;
- break;
- }
- break;
-
- case PNG_COLOR_TYPE_RGB: // color type '2', bitdepth = 8, 16
- switch(bit_depth) {
- case 8:
- image_type = (pixel_depth == 24) ? FIT_BITMAP : FIT_UNKNOWN;
- break;
- case 16:
- image_type = (pixel_depth == 48) ? FIT_RGB16 : FIT_UNKNOWN;
- break;
- default:
- image_type = FIT_UNKNOWN;
- break;
- }
- // sometimes, 24- or 48-bit images may contain transparency information
- // check for this use case and convert to an alpha-compatible format
- if (bIsTransparent && (image_type != FIT_UNKNOWN)) {
- // if the image is 24-bit RGB, mark it as 32-bit; if it is 48-bit, mark it as 64-bit
- image_type = (pixel_depth == 24) ? FIT_BITMAP : (pixel_depth == 48) ? FIT_RGBA16 : FIT_UNKNOWN;
- // expand tRNS chunk to alpha channel
- png_set_tRNS_to_alpha(png_ptr);
- }
- break;
-
- case PNG_COLOR_TYPE_PALETTE: // color type '3', bitdepth = 1, 2, 4, 8
- switch(bit_depth) {
- case 1:
- case 2:
- case 4:
- case 8:
- // expand palette images to the full 8 bits from 2 bits/pixel
- if (pixel_depth == 2) {
- png_set_packing(png_ptr);
- }
-
- // if a tRNS chunk is provided, we must also expand the palletized data to 8-bits,
- // this allows us to make use of the transparency table with existing FreeImage methods
- if (bIsTransparent && (pixel_depth < 8)) {
- png_set_packing(png_ptr);
- }
- break;
-
- default:
- image_type = FIT_UNKNOWN;
- break;
- }
- break;
-
- case PNG_COLOR_TYPE_GRAY_ALPHA: // color type '4', bitdepth = 8, 16
- switch(bit_depth) {
- case 8:
- // 8-bit grayscale + 8-bit alpha => convert to 32-bit RGBA
- image_type = (pixel_depth == 16) ? FIT_BITMAP : FIT_UNKNOWN;
- break;
- case 16:
- // 16-bit grayscale + 16-bit alpha => convert to 64-bit RGBA
- image_type = (pixel_depth == 32) ? FIT_RGBA16 : FIT_UNKNOWN;
- break;
- default:
- image_type = FIT_UNKNOWN;
- break;
- }
- // expand 8-bit greyscale + 8-bit alpha to 32-bit
- // expand 16-bit greyscale + 16-bit alpha to 64-bit
- png_set_gray_to_rgb(png_ptr);
- break;
-
- case PNG_COLOR_TYPE_RGB_ALPHA: // color type '6', bitdepth = 8, 16
- switch(bit_depth) {
- case 8:
- break;
- case 16:
- image_type = (pixel_depth == 64) ? FIT_RGBA16 : FIT_UNKNOWN;
- break;
- default:
- image_type = FIT_UNKNOWN;
- break;
- }
- break;
- }
-
- // check for unknown or invalid formats
- if(image_type == FIT_UNKNOWN) {
- *output_image_type = image_type;
- return FALSE;
- }
-
-#ifndef FREEIMAGE_BIGENDIAN
- if((image_type == FIT_UINT16) || (image_type == FIT_RGB16) || (image_type == FIT_RGBA16)) {
- // turn on 16-bit byte swapping
- png_set_swap(png_ptr);
- }
-#endif
-
-#if FREEIMAGE_COLORORDER == FREEIMAGE_COLORORDER_BGR
- if((image_type == FIT_BITMAP) && ((color_type == PNG_COLOR_TYPE_RGB) || (color_type == PNG_COLOR_TYPE_RGB_ALPHA))) {
- // flip the RGB pixels to BGR (or RGBA to BGRA)
- png_set_bgr(png_ptr);
- }
-#endif
-
- // gamma correction
- // unlike the example in the libpng documentation, we have *no* idea where
- // this file may have come from--so if it doesn't have a file gamma, don't
- // do any correction ("do no harm")
-
- if (png_get_valid(png_ptr, info_ptr, PNG_INFO_gAMA)) {
- double gamma = 0;
- double screen_gamma = 2.2;
-
- if (png_get_gAMA(png_ptr, info_ptr, &gamma) && ( flags & PNG_IGNOREGAMMA ) != PNG_IGNOREGAMMA) {
- png_set_gamma(png_ptr, screen_gamma, gamma);
- }
- }
-
- // all transformations have been registered; now update info_ptr data
- png_read_update_info(png_ptr, info_ptr);
-
- // return the output image type
- *output_image_type = image_type;
-
- return TRUE;
-}
-
-static FIBITMAP * DLL_CALLCONV
-Load(FreeImageIO *io, fi_handle handle, int page, int flags, void *data) {
- png_structp png_ptr = NULL;
- png_infop info_ptr = NULL;
- png_uint_32 width, height;
- int color_type;
- int bit_depth;
- int pixel_depth = 0; // pixel_depth = bit_depth * channels
-
- FIBITMAP *dib = NULL;
- png_bytepp row_pointers = NULL;
-
- fi_ioStructure fio;
- fio.s_handle = handle;
- fio.s_io = io;
-
- if (handle) {
- BOOL header_only = (flags & FIF_LOAD_NOPIXELS) == FIF_LOAD_NOPIXELS;
-
- try {
- // check to see if the file is in fact a PNG file
-
- BYTE png_check[PNG_BYTES_TO_CHECK];
-
- io->read_proc(png_check, PNG_BYTES_TO_CHECK, 1, handle);
-
- if (png_sig_cmp(png_check, (png_size_t)0, PNG_BYTES_TO_CHECK) != 0) {
- return NULL; // Bad signature
- }
-
- // create the chunk manage structure
-
- png_ptr = png_create_read_struct(PNG_LIBPNG_VER_STRING, (png_voidp)NULL, error_handler, warning_handler);
-
- if (!png_ptr) {
- return NULL;
- }
-
- // create the info structure
-
- info_ptr = png_create_info_struct(png_ptr);
-
- if (!info_ptr) {
- png_destroy_read_struct(&png_ptr, (png_infopp)NULL, (png_infopp)NULL);
- return NULL;
- }
-
- // init the IO
-
- png_set_read_fn(png_ptr, &fio, _ReadProc);
-
- if (setjmp(png_jmpbuf(png_ptr))) {
- png_destroy_read_struct(&png_ptr, &info_ptr, NULL);
- return NULL;
- }
-
- // because we have already read the signature...
-
- png_set_sig_bytes(png_ptr, PNG_BYTES_TO_CHECK);
-
- // read the IHDR chunk
-
- png_read_info(png_ptr, info_ptr);
- png_get_IHDR(png_ptr, info_ptr, &width, &height, &bit_depth, &color_type, NULL, NULL, NULL);
-
- // configure the decoder
-
- FREE_IMAGE_TYPE image_type = FIT_BITMAP;
-
- if(!ConfigureDecoder(png_ptr, info_ptr, flags, &image_type)) {
- throw FI_MSG_ERROR_UNSUPPORTED_FORMAT;
- }
-
- // update image info
-
- color_type = png_get_color_type(png_ptr, info_ptr);
- bit_depth = png_get_bit_depth(png_ptr, info_ptr);
- pixel_depth = bit_depth * png_get_channels(png_ptr, info_ptr);
-
- // create a dib and write the bitmap header
- // set up the dib palette, if needed
-
- switch (color_type) {
- case PNG_COLOR_TYPE_RGB:
- case PNG_COLOR_TYPE_RGB_ALPHA:
- dib = FreeImage_AllocateHeaderT(header_only, image_type, width, height, pixel_depth, FI_RGBA_RED_MASK, FI_RGBA_GREEN_MASK, FI_RGBA_BLUE_MASK);
- break;
-
- case PNG_COLOR_TYPE_PALETTE:
- dib = FreeImage_AllocateHeaderT(header_only, image_type, width, height, pixel_depth, FI_RGBA_RED_MASK, FI_RGBA_GREEN_MASK, FI_RGBA_BLUE_MASK);
- if(dib) {
- png_colorp png_palette = NULL;
- int palette_entries = 0;
-
- png_get_PLTE(png_ptr,info_ptr, &png_palette, &palette_entries);
-
- palette_entries = MIN((unsigned)palette_entries, FreeImage_GetColorsUsed(dib));
-
- // store the palette
-
- RGBQUAD *palette = FreeImage_GetPalette(dib);
- for(int i = 0; i < palette_entries; i++) {
- palette[i].rgbRed = png_palette[i].red;
- palette[i].rgbGreen = png_palette[i].green;
- palette[i].rgbBlue = png_palette[i].blue;
- }
- }
- break;
-
- case PNG_COLOR_TYPE_GRAY:
- dib = FreeImage_AllocateHeaderT(header_only, image_type, width, height, pixel_depth, FI_RGBA_RED_MASK, FI_RGBA_GREEN_MASK, FI_RGBA_BLUE_MASK);
-
- if(dib && (pixel_depth <= 8)) {
- RGBQUAD *palette = FreeImage_GetPalette(dib);
- const int palette_entries = 1 << pixel_depth;
-
- for(int i = 0; i < palette_entries; i++) {
- palette[i].rgbRed =
- palette[i].rgbGreen =
- palette[i].rgbBlue = (BYTE)((i * 255) / (palette_entries - 1));
- }
- }
- break;
-
- default:
- throw FI_MSG_ERROR_UNSUPPORTED_FORMAT;
- }
-
- if(!dib) {
- throw FI_MSG_ERROR_DIB_MEMORY;
- }
-
- // store the transparency table
-
- if (png_get_valid(png_ptr, info_ptr, PNG_INFO_tRNS)) {
- // array of alpha (transparency) entries for palette
- png_bytep trans_alpha = NULL;
- // number of transparent entries
- int num_trans = 0;
- // graylevel or color sample values of the single transparent color for non-paletted images
- png_color_16p trans_color = NULL;
-
- png_get_tRNS(png_ptr, info_ptr, &trans_alpha, &num_trans, &trans_color);
-
- if((color_type == PNG_COLOR_TYPE_GRAY) && trans_color) {
- // single transparent color
- if (trans_color->gray < 256) {
- BYTE table[256];
- memset(table, 0xFF, 256);
- table[trans_color->gray] = 0;
- FreeImage_SetTransparencyTable(dib, table, 256);
- }
- // check for a full transparency table, too
- else if ((trans_alpha) && (pixel_depth <= 8)) {
- FreeImage_SetTransparencyTable(dib, (BYTE *)trans_alpha, num_trans);
- }
-
- } else if((color_type == PNG_COLOR_TYPE_PALETTE) && trans_alpha) {
- // transparency table
- FreeImage_SetTransparencyTable(dib, (BYTE *)trans_alpha, num_trans);
- }
- }
-
- // store the background color (only supported for FIT_BITMAP types)
-
- if ((image_type == FIT_BITMAP) && png_get_valid(png_ptr, info_ptr, PNG_INFO_bKGD)) {
- // Get the background color to draw transparent and alpha images over.
- // Note that even if the PNG file supplies a background, you are not required to
- // use it - you should use the (solid) application background if it has one.
-
- png_color_16p image_background = NULL;
- RGBQUAD rgbBkColor;
-
- if (png_get_bKGD(png_ptr, info_ptr, &image_background)) {
- rgbBkColor.rgbRed = (BYTE)image_background->red;
- rgbBkColor.rgbGreen = (BYTE)image_background->green;
- rgbBkColor.rgbBlue = (BYTE)image_background->blue;
- rgbBkColor.rgbReserved = 0;
-
- FreeImage_SetBackgroundColor(dib, &rgbBkColor);
- }
- }
-
- // get physical resolution
-
- if (png_get_valid(png_ptr, info_ptr, PNG_INFO_pHYs)) {
- png_uint_32 res_x, res_y;
-
- // we'll overload this var and use 0 to mean no phys data,
- // since if it's not in meters we can't use it anyway
-
- int res_unit_type = PNG_RESOLUTION_UNKNOWN;
-
- png_get_pHYs(png_ptr,info_ptr, &res_x, &res_y, &res_unit_type);
-
- if (res_unit_type == PNG_RESOLUTION_METER) {
- FreeImage_SetDotsPerMeterX(dib, res_x);
- FreeImage_SetDotsPerMeterY(dib, res_y);
- }
- }
-
- // get possible ICC profile
-
- if (png_get_valid(png_ptr, info_ptr, PNG_INFO_iCCP)) {
- png_charp profile_name = NULL;
- png_bytep profile_data = NULL;
- png_uint_32 profile_length = 0;
- int compression_type;
-
- png_get_iCCP(png_ptr, info_ptr, &profile_name, &compression_type, &profile_data, &profile_length);
-
- // copy ICC profile data (must be done after FreeImage_AllocateHeader)
-
- FreeImage_CreateICCProfile(dib, profile_data, profile_length);
- }
-
- // --- header only mode => clean-up and return
-
- if (header_only) {
- // get possible metadata (it can be located both before and after the image data)
- ReadMetadata(png_ptr, info_ptr, dib);
- if (png_ptr) {
- // clean up after the read, and free any memory allocated - REQUIRED
- png_destroy_read_struct(&png_ptr, &info_ptr, (png_infopp)NULL);
- }
- return dib;
- }
-
- // set the individual row_pointers to point at the correct offsets
-
- row_pointers = (png_bytepp)malloc(height * sizeof(png_bytep));
-
- if (!row_pointers) {
- png_destroy_read_struct(&png_ptr, &info_ptr, NULL);
- FreeImage_Unload(dib);
- return NULL;
- }
-
- // read in the bitmap bits via the pointer table
- // allow loading of PNG with minor errors (such as images with several IDAT chunks)
-
- for (png_uint_32 k = 0; k < height; k++) {
- row_pointers[height - 1 - k] = FreeImage_GetScanLine(dib, k);
- }
-
- png_set_benign_errors(png_ptr, 1);
- png_read_image(png_ptr, row_pointers);
-
- // check if the bitmap contains transparency, if so enable it in the header
-
- if (FreeImage_GetBPP(dib) == 32) {
- if (FreeImage_GetColorType(dib) == FIC_RGBALPHA) {
- FreeImage_SetTransparent(dib, TRUE);
- } else {
- FreeImage_SetTransparent(dib, FALSE);
- }
- }
-
- // cleanup
-
- if (row_pointers) {
- free(row_pointers);
- row_pointers = NULL;
- }
-
- // read the rest of the file, getting any additional chunks in info_ptr
-
- png_read_end(png_ptr, info_ptr);
-
- // get possible metadata (it can be located both before and after the image data)
-
- ReadMetadata(png_ptr, info_ptr, dib);
-
- if (png_ptr) {
- // clean up after the read, and free any memory allocated - REQUIRED
- png_destroy_read_struct(&png_ptr, &info_ptr, (png_infopp)NULL);
- }
-
- return dib;
-
- } catch (const char *text) {
- if (png_ptr) {
- png_destroy_read_struct(&png_ptr, &info_ptr, (png_infopp)NULL);
- }
- if (row_pointers) {
- free(row_pointers);
- }
- if (dib) {
- FreeImage_Unload(dib);
- }
- FreeImage_OutputMessageProc(s_format_id, text);
-
- return NULL;
- }
- }
-
- return NULL;
-}
-
-// --------------------------------------------------------------------------
-
-static BOOL DLL_CALLCONV
-Save(FreeImageIO *io, FIBITMAP *dib, fi_handle handle, int page, int flags, void *data) {
- png_structp png_ptr;
- png_infop info_ptr;
- png_colorp palette = NULL;
- png_uint_32 width, height;
- BOOL has_alpha_channel = FALSE;
-
- RGBQUAD *pal; // pointer to dib palette
- int bit_depth, pixel_depth; // pixel_depth = bit_depth * channels
- int palette_entries;
- int interlace_type;
-
- fi_ioStructure fio;
- fio.s_handle = handle;
- fio.s_io = io;
-
- if ((dib) && (handle)) {
- try {
- // create the chunk manage structure
-
- png_ptr = png_create_write_struct(PNG_LIBPNG_VER_STRING, (png_voidp)NULL, error_handler, warning_handler);
-
- if (!png_ptr) {
- return FALSE;
- }
-
- // allocate/initialize the image information data.
-
- info_ptr = png_create_info_struct(png_ptr);
-
- if (!info_ptr) {
- png_destroy_write_struct(&png_ptr, (png_infopp)NULL);
- return FALSE;
- }
-
- // Set error handling. REQUIRED if you aren't supplying your own
- // error handling functions in the png_create_write_struct() call.
-
- if (setjmp(png_jmpbuf(png_ptr))) {
- // if we get here, we had a problem reading the file
-
- png_destroy_write_struct(&png_ptr, &info_ptr);
-
- return FALSE;
- }
-
- // init the IO
-
- png_set_write_fn(png_ptr, &fio, _WriteProc, _FlushProc);
-
- // set physical resolution
-
- png_uint_32 res_x = (png_uint_32)FreeImage_GetDotsPerMeterX(dib);
- png_uint_32 res_y = (png_uint_32)FreeImage_GetDotsPerMeterY(dib);
-
- if ((res_x > 0) && (res_y > 0)) {
- png_set_pHYs(png_ptr, info_ptr, res_x, res_y, PNG_RESOLUTION_METER);
- }
-
- // Set the image information here. Width and height are up to 2^31,
- // bit_depth is one of 1, 2, 4, 8, or 16, but valid values also depend on
- // the color_type selected. color_type is one of PNG_COLOR_TYPE_GRAY,
- // PNG_COLOR_TYPE_GRAY_ALPHA, PNG_COLOR_TYPE_PALETTE, PNG_COLOR_TYPE_RGB,
- // or PNG_COLOR_TYPE_RGB_ALPHA. interlace is either PNG_INTERLACE_NONE or
- // PNG_INTERLACE_ADAM7, and the compression_type and filter_type MUST
- // currently be PNG_COMPRESSION_TYPE_BASE and PNG_FILTER_TYPE_BASE. REQUIRED
-
- width = FreeImage_GetWidth(dib);
- height = FreeImage_GetHeight(dib);
- pixel_depth = FreeImage_GetBPP(dib);
-
- BOOL bInterlaced = FALSE;
- if( (flags & PNG_INTERLACED) == PNG_INTERLACED) {
- interlace_type = PNG_INTERLACE_ADAM7;
- bInterlaced = TRUE;
- } else {
- interlace_type = PNG_INTERLACE_NONE;
- }
-
- // set the ZLIB compression level or default to PNG default compression level (ZLIB level = 6)
- int zlib_level = flags & 0x0F;
- if((zlib_level >= 1) && (zlib_level <= 9)) {
- png_set_compression_level(png_ptr, zlib_level);
- } else if((flags & PNG_Z_NO_COMPRESSION) == PNG_Z_NO_COMPRESSION) {
- png_set_compression_level(png_ptr, Z_NO_COMPRESSION);
- }
-
- // filtered strategy works better for high color images
- if(pixel_depth >= 16){
- png_set_compression_strategy(png_ptr, Z_FILTERED);
- png_set_filter(png_ptr, 0, PNG_FILTER_NONE|PNG_FILTER_SUB|PNG_FILTER_PAETH);
- } else {
- png_set_compression_strategy(png_ptr, Z_DEFAULT_STRATEGY);
- }
-
- FREE_IMAGE_TYPE image_type = FreeImage_GetImageType(dib);
- if(image_type == FIT_BITMAP) {
- // standard image type
- bit_depth = (pixel_depth > 8) ? 8 : pixel_depth;
- } else {
- // 16-bit greyscale or 16-bit RGB(A)
- bit_depth = 16;
- }
-
- // check for transparent images
- BOOL bIsTransparent =
- (image_type == FIT_BITMAP) && FreeImage_IsTransparent(dib) && (FreeImage_GetTransparencyCount(dib) > 0) ? TRUE : FALSE;
-
- switch (FreeImage_GetColorType(dib)) {
- case FIC_MINISWHITE:
- if(!bIsTransparent) {
- // Invert monochrome files to have 0 as black and 1 as white (no break here)
- png_set_invert_mono(png_ptr);
- }
- // (fall through)
-
- case FIC_MINISBLACK:
- if(!bIsTransparent) {
- png_set_IHDR(png_ptr, info_ptr, width, height, bit_depth,
- PNG_COLOR_TYPE_GRAY, interlace_type,
- PNG_COMPRESSION_TYPE_BASE, PNG_FILTER_TYPE_BASE);
- break;
- }
- // If a monochrome image is transparent, save it with a palette
- // (fall through)
-
- case FIC_PALETTE:
- {
- png_set_IHDR(png_ptr, info_ptr, width, height, bit_depth,
- PNG_COLOR_TYPE_PALETTE, interlace_type,
- PNG_COMPRESSION_TYPE_BASE, PNG_FILTER_TYPE_BASE);
-
- // set the palette
-
- palette_entries = 1 << bit_depth;
- palette = (png_colorp)png_malloc(png_ptr, palette_entries * sizeof (png_color));
- pal = FreeImage_GetPalette(dib);
-
- for (int i = 0; i < palette_entries; i++) {
- palette[i].red = pal[i].rgbRed;
- palette[i].green = pal[i].rgbGreen;
- palette[i].blue = pal[i].rgbBlue;
- }
-
- png_set_PLTE(png_ptr, info_ptr, palette, palette_entries);
-
- // You must not free palette here, because png_set_PLTE only makes a link to
- // the palette that you malloced. Wait until you are about to destroy
- // the png structure.
-
- break;
- }
-
- case FIC_RGBALPHA :
- has_alpha_channel = TRUE;
-
- png_set_IHDR(png_ptr, info_ptr, width, height, bit_depth,
- PNG_COLOR_TYPE_RGBA, interlace_type,
- PNG_COMPRESSION_TYPE_BASE, PNG_FILTER_TYPE_BASE);
-
-#if FREEIMAGE_COLORORDER == FREEIMAGE_COLORORDER_BGR
- // flip BGR pixels to RGB
- if(image_type == FIT_BITMAP) {
- png_set_bgr(png_ptr);
- }
-#endif
- break;
-
- case FIC_RGB:
- png_set_IHDR(png_ptr, info_ptr, width, height, bit_depth,
- PNG_COLOR_TYPE_RGB, interlace_type,
- PNG_COMPRESSION_TYPE_BASE, PNG_FILTER_TYPE_BASE);
-
-#if FREEIMAGE_COLORORDER == FREEIMAGE_COLORORDER_BGR
- // flip BGR pixels to RGB
- if(image_type == FIT_BITMAP) {
- png_set_bgr(png_ptr);
- }
-#endif
- break;
-
- case FIC_CMYK:
- break;
- }
-
- // write possible ICC profile
-
- FIICCPROFILE *iccProfile = FreeImage_GetICCProfile(dib);
- if (iccProfile->size && iccProfile->data) {
- png_set_iCCP(png_ptr, info_ptr, "Embedded Profile", 0, (png_const_bytep)iccProfile->data, iccProfile->size);
- }
-
- // write metadata
-
- WriteMetadata(png_ptr, info_ptr, dib);
-
- // Optional gamma chunk is strongly suggested if you have any guess
- // as to the correct gamma of the image.
- // png_set_gAMA(png_ptr, info_ptr, gamma);
-
- // set the transparency table
-
- if (bIsTransparent) {
- png_set_tRNS(png_ptr, info_ptr, FreeImage_GetTransparencyTable(dib), FreeImage_GetTransparencyCount(dib), NULL);
- }
-
- // set the background color
-
- if(FreeImage_HasBackgroundColor(dib)) {
- png_color_16 image_background;
- RGBQUAD rgbBkColor;
-
- FreeImage_GetBackgroundColor(dib, &rgbBkColor);
- memset(&image_background, 0, sizeof(png_color_16));
- image_background.blue = rgbBkColor.rgbBlue;
- image_background.green = rgbBkColor.rgbGreen;
- image_background.red = rgbBkColor.rgbRed;
- image_background.index = rgbBkColor.rgbReserved;
-
- png_set_bKGD(png_ptr, info_ptr, &image_background);
- }
-
- // Write the file header information.
-
- png_write_info(png_ptr, info_ptr);
-
- // write out the image data
-
-#ifndef FREEIMAGE_BIGENDIAN
- if (bit_depth == 16) {
- // turn on 16 bit byte swapping
- png_set_swap(png_ptr);
- }
-#endif
-
- int number_passes = 1;
- if (bInterlaced) {
- number_passes = png_set_interlace_handling(png_ptr);
- }
-
- if ((pixel_depth == 32) && (!has_alpha_channel)) {
- BYTE *buffer = (BYTE *)malloc(width * 3);
-
- // transparent conversion to 24-bit
- // the number of passes is either 1 for non-interlaced images, or 7 for interlaced images
- for (int pass = 0; pass < number_passes; pass++) {
- for (png_uint_32 k = 0; k < height; k++) {
- FreeImage_ConvertLine32To24(buffer, FreeImage_GetScanLine(dib, height - k - 1), width);
- png_write_row(png_ptr, buffer);
- }
- }
- free(buffer);
- } else {
- // the number of passes is either 1 for non-interlaced images, or 7 for interlaced images
- for (int pass = 0; pass < number_passes; pass++) {
- for (png_uint_32 k = 0; k < height; k++) {
- png_write_row(png_ptr, FreeImage_GetScanLine(dib, height - k - 1));
- }
- }
- }
-
- // It is REQUIRED to call this to finish writing the rest of the file
- // Bug with png_flush
-
- png_write_end(png_ptr, info_ptr);
-
- // clean up after the write, and free any memory allocated
- if (palette) {
- png_free(png_ptr, palette);
- }
-
- png_destroy_write_struct(&png_ptr, &info_ptr);
-
- return TRUE;
-
- } catch (const char *text) {
- if(png_ptr) {
- png_destroy_write_struct(&png_ptr, &info_ptr);
- }
- FreeImage_OutputMessageProc(s_format_id, text);
- }
- }
-
- return FALSE;
-}
-
-// ==========================================================
-// Init
-// ==========================================================
-
-void DLL_CALLCONV
-InitPNG(Plugin *plugin, int format_id) {
- s_format_id = format_id;
-
- plugin->format_proc = Format;
- plugin->description_proc = Description;
- plugin->extension_proc = Extension;
- plugin->regexpr_proc = RegExpr;
- plugin->open_proc = NULL;
- plugin->close_proc = NULL;
- plugin->pagecount_proc = NULL;
- plugin->pagecapability_proc = NULL;
- plugin->load_proc = Load;
- plugin->save_proc = Save;
- plugin->validate_proc = Validate;
- plugin->mime_proc = MimeType;
- plugin->supports_export_bpp_proc = SupportsExportDepth;
- plugin->supports_export_type_proc = SupportsExportType;
- plugin->supports_icc_profiles_proc = SupportsICCProfiles;
- plugin->supports_no_pixels_proc = SupportsNoPixels;
-}