| // ========================================================== |
| // Bitmap conversion routines |
| // |
| // Design and implementation by |
| // - Floris van den Berg (flvdberg@wxs.nl) |
| // - Hervé Drolon (drolon@infonie.fr) |
| // - Jani Kajala (janik@remedy.fi) |
| // - Mihail Naydenov (mnaydenov@users.sourceforge.net) |
| // - Carsten Klein (cklein05@users.sourceforge.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! |
| // ========================================================== |
| |
| #include "FreeImage.h" |
| #include "Utilities.h" |
| #include "Quantizers.h" |
| |
| // ---------------------------------------------------------- |
| |
| #define CONVERT(from, to) case to : FreeImage_ConvertLine##from##To##to(bits, scanline, FreeImage_GetWidth(dib)); break; |
| #define CONVERTWITHPALETTE(from, to) case to : FreeImage_ConvertLine##from##To##to(bits, scanline, FreeImage_GetWidth(dib), FreeImage_GetPalette(dib)); break; |
| |
| #define CONVERTTO16(from) \ |
| case 16 : \ |
| if ((red_mask == FI16_555_RED_MASK) && (green_mask == FI16_555_GREEN_MASK) && (blue_mask == FI16_555_BLUE_MASK)) { \ |
| FreeImage_ConvertLine##from##To16_555(bits, scanline, FreeImage_GetWidth(dib)); \ |
| } else { \ |
| FreeImage_ConvertLine##from##To16_565(bits, scanline, FreeImage_GetWidth(dib)); \ |
| } \ |
| break; |
| |
| #define CONVERTTO16WITHPALETTE(from) \ |
| case 16 : \ |
| if ((red_mask == FI16_555_RED_MASK) && (green_mask == FI16_555_GREEN_MASK) && (blue_mask == FI16_555_BLUE_MASK)) { \ |
| FreeImage_ConvertLine##from##To16_555(bits, scanline, FreeImage_GetWidth(dib), FreeImage_GetPalette(dib)); \ |
| } else { \ |
| FreeImage_ConvertLine##from##To16_565(bits, scanline, FreeImage_GetWidth(dib), FreeImage_GetPalette(dib)); \ |
| } \ |
| break; |
| |
| // ========================================================== |
| // Utility functions declared in Utilities.h |
| |
| BOOL SwapRedBlue32(FIBITMAP* dib) { |
| if(FreeImage_GetImageType(dib) != FIT_BITMAP) { |
| return FALSE; |
| } |
| |
| const unsigned bytesperpixel = FreeImage_GetBPP(dib) / 8; |
| if(bytesperpixel > 4 || bytesperpixel < 3) { |
| return FALSE; |
| } |
| |
| const unsigned height = FreeImage_GetHeight(dib); |
| const unsigned pitch = FreeImage_GetPitch(dib); |
| const unsigned lineSize = FreeImage_GetLine(dib); |
| |
| BYTE* line = FreeImage_GetBits(dib); |
| for(unsigned y = 0; y < height; ++y, line += pitch) { |
| for(BYTE* pixel = line; pixel < line + lineSize ; pixel += bytesperpixel) { |
| INPLACESWAP(pixel[0], pixel[2]); |
| } |
| } |
| |
| return TRUE; |
| } |
| |
| // ---------------------------------------------------------- |
| |
| static inline void |
| assignRGB(WORD r, WORD g, WORD b, WORD* out) { |
| out[0] = r; |
| out[1] = g; |
| out[2] = b; |
| } |
| |
| static inline void |
| assignRGB(BYTE r, BYTE g, BYTE b, BYTE* out) { |
| out[FI_RGBA_RED] = r; |
| out[FI_RGBA_GREEN] = g; |
| out[FI_RGBA_BLUE] = b; |
| } |
| |
| /** |
| CMYK -> CMY -> RGB conversion from http://www.easyrgb.com/ |
| |
| CMYK to CMY [0-1]: C,M,Y * (1 - K) + K |
| CMY to RGB [0-1]: (1 - C,M,Y) |
| |
| => R,G,B = (1 - C,M,Y) * (1 - K) |
| mapped to [0-MAX_VAL]: |
| (MAX_VAL - C,M,Y) * (MAX_VAL - K) / MAX_VAL |
| */ |
| template <class T> |
| static inline void |
| CMYKToRGB(T C, T M, T Y, T K, T* out) { |
| unsigned max_val = std::numeric_limits<T>::max(); |
| |
| unsigned r = (max_val - C) * (max_val - K) / max_val; |
| unsigned g = (max_val - M) * (max_val - K) / max_val; |
| unsigned b = (max_val - Y) * (max_val - K) / max_val; |
| |
| // clamp values to [0..max_val] |
| T red = (T)CLAMP(r, (unsigned)0, max_val); |
| T green = (T)CLAMP(g, (unsigned)0, max_val); |
| T blue = (T)CLAMP(b, (unsigned)0, max_val); |
| |
| assignRGB(red, green, blue, out); |
| } |
| |
| template <class T> |
| static void |
| _convertCMYKtoRGBA(unsigned width, unsigned height, BYTE* line_start, unsigned pitch, unsigned samplesperpixel) { |
| const BOOL hasBlack = (samplesperpixel > 3) ? TRUE : FALSE; |
| const T MAX_VAL = std::numeric_limits<T>::max(); |
| |
| T K = 0; |
| for(unsigned y = 0; y < height; y++) { |
| T *line = (T*)line_start; |
| |
| for(unsigned x = 0; x < width; x++) { |
| if(hasBlack) { |
| K = line[FI_RGBA_ALPHA]; |
| line[FI_RGBA_ALPHA] = MAX_VAL; // TODO write the first extra channel as alpha! |
| } |
| |
| CMYKToRGB<T>(line[0], line[1], line[2], K, line); |
| |
| line += samplesperpixel; |
| } |
| line_start += pitch; |
| } |
| } |
| |
| BOOL |
| ConvertCMYKtoRGBA(FIBITMAP* dib) { |
| if(!FreeImage_HasPixels(dib)) { |
| return FALSE; |
| } |
| |
| const FREE_IMAGE_TYPE image_type = FreeImage_GetImageType(dib); |
| const unsigned bytesperpixel = FreeImage_GetBPP(dib)/8; |
| |
| unsigned channelSize = 1; |
| if (image_type == FIT_RGBA16 || image_type == FIT_RGB16) { |
| channelSize = sizeof(WORD); |
| } else if (!(image_type == FIT_BITMAP && (bytesperpixel > 2))) { |
| return FALSE; |
| } |
| |
| const unsigned width = FreeImage_GetWidth(dib); |
| const unsigned height = FreeImage_GetHeight(dib); |
| BYTE *line_start = FreeImage_GetScanLine(dib, 0); |
| const unsigned pitch = FreeImage_GetPitch(dib); |
| |
| unsigned samplesperpixel = FreeImage_GetLine(dib) / width / channelSize; |
| |
| if(channelSize == sizeof(WORD)) { |
| _convertCMYKtoRGBA<WORD>(width, height, line_start, pitch, samplesperpixel); |
| } else { |
| _convertCMYKtoRGBA<BYTE>(width, height, line_start, pitch, samplesperpixel); |
| } |
| |
| return TRUE; |
| } |
| |
| // ---------------------------------------------------------- |
| |
| /** |
| CIELab -> XYZ conversion from http://www.easyrgb.com/ |
| */ |
| static void |
| CIELabToXYZ(float L, float a, float b, float *X, float *Y, float *Z) { |
| float pow_3; |
| |
| // CIELab -> XYZ conversion |
| // ------------------------ |
| float var_Y = (L + 16.F ) / 116.F; |
| float var_X = a / 500.F + var_Y; |
| float var_Z = var_Y - b / 200.F; |
| |
| pow_3 = powf(var_Y, 3); |
| if(pow_3 > 0.008856F) { |
| var_Y = pow_3; |
| } else { |
| var_Y = ( var_Y - 16.F / 116.F ) / 7.787F; |
| } |
| pow_3 = powf(var_X, 3); |
| if(pow_3 > 0.008856F) { |
| var_X = pow_3; |
| } else { |
| var_X = ( var_X - 16.F / 116.F ) / 7.787F; |
| } |
| pow_3 = powf(var_Z, 3); |
| if(pow_3 > 0.008856F) { |
| var_Z = pow_3; |
| } else { |
| var_Z = ( var_Z - 16.F / 116.F ) / 7.787F; |
| } |
| |
| static const float ref_X = 95.047F; |
| static const float ref_Y = 100.000F; |
| static const float ref_Z = 108.883F; |
| |
| *X = ref_X * var_X; // ref_X = 95.047 (Observer= 2°, Illuminant= D65) |
| *Y = ref_Y * var_Y; // ref_Y = 100.000 |
| *Z = ref_Z * var_Z; // ref_Z = 108.883 |
| } |
| |
| /** |
| XYZ -> RGB conversion from http://www.easyrgb.com/ |
| */ |
| static void |
| XYZToRGB(float X, float Y, float Z, float *R, float *G, float *B) { |
| float var_X = X / 100; // X from 0 to 95.047 (Observer = 2°, Illuminant = D65) |
| float var_Y = Y / 100; // Y from 0 to 100.000 |
| float var_Z = Z / 100; // Z from 0 to 108.883 |
| |
| float var_R = var_X * 3.2406F + var_Y * -1.5372F + var_Z * -0.4986F; |
| float var_G = var_X * -0.9689F + var_Y * 1.8758F + var_Z * 0.0415F; |
| float var_B = var_X * 0.0557F + var_Y * -0.2040F + var_Z * 1.0570F; |
| |
| float exponent = 1.F / 2.4F; |
| |
| if(var_R > 0.0031308F) { |
| var_R = 1.055F * powf(var_R, exponent) - 0.055F; |
| } else { |
| var_R = 12.92F * var_R; |
| } |
| if(var_G > 0.0031308F) { |
| var_G = 1.055F * powf(var_G, exponent) - 0.055F; |
| } else { |
| var_G = 12.92F * var_G; |
| } |
| if(var_B > 0.0031308F) { |
| var_B = 1.055F * powf(var_B, exponent) - 0.055F; |
| } else { |
| var_B = 12.92F * var_B; |
| } |
| |
| *R = var_R; |
| *G = var_G; |
| *B = var_B; |
| } |
| |
| template<class T> |
| static void |
| CIELabToRGB(float L, float a, float b, T *rgb) { |
| float X, Y, Z; |
| float R, G, B; |
| const float max_val = std::numeric_limits<T>::max(); |
| |
| CIELabToXYZ(L, a, b, &X, &Y, &Z); |
| XYZToRGB(X, Y, Z, &R, &G, &B); |
| |
| // clamp values to [0..max_val] |
| T red = (T)CLAMP(R * max_val, 0.0F, max_val); |
| T green = (T)CLAMP(G * max_val, 0.0F, max_val); |
| T blue = (T)CLAMP(B * max_val, 0.0F, max_val); |
| |
| assignRGB(red, green, blue, rgb); |
| } |
| |
| template<class T> |
| static void |
| _convertLABtoRGB(unsigned width, unsigned height, BYTE* line_start, unsigned pitch, unsigned samplesperpixel) { |
| const unsigned max_val = std::numeric_limits<T>::max(); |
| const float sL = 100.F / max_val; |
| const float sa = 256.F / max_val; |
| const float sb = 256.F / max_val; |
| |
| for(unsigned y = 0; y < height; y++) { |
| T *line = (T*)line_start; |
| |
| for(unsigned x = 0; x < width; x++) { |
| CIELabToRGB(line[0]* sL, line[1]* sa - 128.F, line[2]* sb - 128.F, line); |
| |
| line += samplesperpixel; |
| } |
| line_start += pitch; |
| } |
| } |
| |
| BOOL |
| ConvertLABtoRGB(FIBITMAP* dib) { |
| if(!FreeImage_HasPixels(dib)) { |
| return FALSE; |
| } |
| |
| const FREE_IMAGE_TYPE image_type = FreeImage_GetImageType(dib); |
| const unsigned bytesperpixel = FreeImage_GetBPP(dib) / 8; |
| |
| unsigned channelSize = 1; |
| if (image_type == FIT_RGBA16 || image_type == FIT_RGB16) { |
| channelSize = sizeof(WORD); |
| } else if (!(image_type == FIT_BITMAP && (bytesperpixel > 2))) { |
| return FALSE; |
| } |
| |
| const unsigned width = FreeImage_GetWidth(dib); |
| const unsigned height = FreeImage_GetHeight(dib); |
| BYTE *line_start = FreeImage_GetScanLine(dib, 0); |
| const unsigned pitch = FreeImage_GetPitch(dib); |
| |
| unsigned samplesperpixel = FreeImage_GetLine(dib) / width / channelSize; |
| |
| if(channelSize == 1) { |
| _convertLABtoRGB<BYTE>(width, height, line_start, pitch, samplesperpixel); |
| } |
| else { |
| _convertLABtoRGB<WORD>(width, height, line_start, pitch, samplesperpixel); |
| } |
| |
| return TRUE; |
| } |
| |
| // ---------------------------------------------------------- |
| |
| FIBITMAP* |
| RemoveAlphaChannel(FIBITMAP* src) { |
| |
| if(!FreeImage_HasPixels(src)) { |
| return NULL; |
| } |
| |
| const FREE_IMAGE_TYPE image_type = FreeImage_GetImageType(src); |
| |
| switch(image_type) { |
| case FIT_BITMAP: |
| if(FreeImage_GetBPP(src) == 32) { |
| // convert to 24-bit |
| return FreeImage_ConvertTo24Bits(src); |
| } |
| break; |
| case FIT_RGBA16: |
| // convert to RGB16 |
| return FreeImage_ConvertToRGB16(src); |
| case FIT_RGBAF: |
| // convert to RGBF |
| return FreeImage_ConvertToRGBF(src); |
| default: |
| // unsupported image type |
| return NULL; |
| } |
| |
| return NULL; |
| } |
| |
| |
| // ========================================================== |
| |
| FIBITMAP * DLL_CALLCONV |
| FreeImage_ColorQuantize(FIBITMAP *dib, FREE_IMAGE_QUANTIZE quantize) { |
| return FreeImage_ColorQuantizeEx(dib, quantize); |
| } |
| |
| FIBITMAP * DLL_CALLCONV |
| FreeImage_ColorQuantizeEx(FIBITMAP *dib, FREE_IMAGE_QUANTIZE quantize, int PaletteSize, int ReserveSize, RGBQUAD *ReservePalette) { |
| if( PaletteSize < 2 ) PaletteSize = 2; |
| if( PaletteSize > 256 ) PaletteSize = 256; |
| if( ReserveSize < 0 ) ReserveSize = 0; |
| if( ReserveSize > PaletteSize ) ReserveSize = PaletteSize; |
| if (FreeImage_HasPixels(dib)) { |
| const unsigned bpp = FreeImage_GetBPP(dib); |
| if((FreeImage_GetImageType(dib) == FIT_BITMAP) && (bpp == 24 || bpp == 32)) { |
| switch(quantize) { |
| case FIQ_WUQUANT : |
| { |
| try { |
| WuQuantizer Q (dib); |
| FIBITMAP *dst = Q.Quantize(PaletteSize, ReserveSize, ReservePalette); |
| if(dst) { |
| // copy metadata from src to dst |
| FreeImage_CloneMetadata(dst, dib); |
| } |
| return dst; |
| } catch (const char *) { |
| return NULL; |
| } |
| break; |
| } |
| case FIQ_NNQUANT : |
| { |
| if (bpp == 32) { |
| // 32-bit images not supported by NNQUANT |
| return NULL; |
| } |
| // sampling factor in range 1..30. |
| // 1 => slower (but better), 30 => faster. Default value is 1 |
| const int sampling = 1; |
| |
| NNQuantizer Q(PaletteSize); |
| FIBITMAP *dst = Q.Quantize(dib, ReserveSize, ReservePalette, sampling); |
| if(dst) { |
| // copy metadata from src to dst |
| FreeImage_CloneMetadata(dst, dib); |
| } |
| return dst; |
| } |
| case FIQ_LFPQUANT : |
| { |
| LFPQuantizer Q(PaletteSize); |
| FIBITMAP *dst = Q.Quantize(dib, ReserveSize, ReservePalette); |
| if(dst) { |
| // copy metadata from src to dst |
| FreeImage_CloneMetadata(dst, dib); |
| } |
| return dst; |
| } |
| } |
| } |
| } |
| |
| return NULL; |
| } |
| |
| // ========================================================== |
| |
| FIBITMAP * DLL_CALLCONV |
| FreeImage_ConvertFromRawBitsEx(BOOL copySource, BYTE *bits, FREE_IMAGE_TYPE type, int width, int height, int pitch, unsigned bpp, unsigned red_mask, unsigned green_mask, unsigned blue_mask, BOOL topdown) { |
| FIBITMAP *dib = NULL; |
| |
| if(copySource) { |
| // allocate a FIBITMAP with internally managed pixel buffer |
| dib = FreeImage_AllocateT(type, width, height, bpp, red_mask, green_mask, blue_mask); |
| if(!dib) { |
| return NULL; |
| } |
| // copy user provided pixel buffer into the dib |
| const unsigned linesize = FreeImage_GetLine(dib); |
| for(int y = 0; y < height; y++) { |
| memcpy(FreeImage_GetScanLine(dib, y), bits, linesize); |
| // next line in user's buffer |
| bits += pitch; |
| } |
| // flip pixels vertically if needed |
| if(topdown) { |
| FreeImage_FlipVertical(dib); |
| } |
| } |
| else { |
| // allocate a FIBITMAP using a wrapper to user provided pixel buffer |
| dib = FreeImage_AllocateHeaderForBits(bits, pitch, type, width, height, bpp, red_mask, green_mask, blue_mask); |
| if(!dib) { |
| return NULL; |
| } |
| // flip pixels vertically if needed |
| if(topdown) { |
| FreeImage_FlipVertical(dib); |
| } |
| } |
| |
| return dib; |
| } |
| |
| FIBITMAP * DLL_CALLCONV |
| FreeImage_ConvertFromRawBits(BYTE *bits, int width, int height, int pitch, unsigned bpp, unsigned red_mask, unsigned green_mask, unsigned blue_mask, BOOL topdown) { |
| return FreeImage_ConvertFromRawBitsEx(TRUE /* copySource */, bits, FIT_BITMAP, width, height, pitch, bpp, red_mask, green_mask, blue_mask, topdown); |
| } |
| |
| void DLL_CALLCONV |
| FreeImage_ConvertToRawBits(BYTE *bits, FIBITMAP *dib, int pitch, unsigned bpp, unsigned red_mask, unsigned green_mask, unsigned blue_mask, BOOL topdown) { |
| if (FreeImage_HasPixels(dib) && (bits != NULL)) { |
| for (unsigned i = 0; i < FreeImage_GetHeight(dib); ++i) { |
| BYTE *scanline = FreeImage_GetScanLine(dib, topdown ? (FreeImage_GetHeight(dib) - i - 1) : i); |
| |
| if ((bpp == 16) && (FreeImage_GetBPP(dib) == 16)) { |
| // convert 555 to 565 or vice versa |
| |
| if ((red_mask == FI16_555_RED_MASK) && (green_mask == FI16_555_GREEN_MASK) && (blue_mask == FI16_555_BLUE_MASK)) { |
| if ((FreeImage_GetRedMask(dib) == FI16_565_RED_MASK) && (FreeImage_GetGreenMask(dib) == FI16_565_GREEN_MASK) && (FreeImage_GetBlueMask(dib) == FI16_565_BLUE_MASK)) { |
| FreeImage_ConvertLine16_565_To16_555(bits, scanline, FreeImage_GetWidth(dib)); |
| } else { |
| memcpy(bits, scanline, FreeImage_GetLine(dib)); |
| } |
| } else { |
| if ((FreeImage_GetRedMask(dib) == FI16_555_RED_MASK) && (FreeImage_GetGreenMask(dib) == FI16_555_GREEN_MASK) && (FreeImage_GetBlueMask(dib) == FI16_555_BLUE_MASK)) { |
| FreeImage_ConvertLine16_555_To16_565(bits, scanline, FreeImage_GetWidth(dib)); |
| } else { |
| memcpy(bits, scanline, FreeImage_GetLine(dib)); |
| } |
| } |
| } else if (FreeImage_GetBPP(dib) != bpp) { |
| switch(FreeImage_GetBPP(dib)) { |
| case 1 : |
| switch(bpp) { |
| CONVERT(1, 8) |
| CONVERTTO16WITHPALETTE(1) |
| CONVERTWITHPALETTE(1, 24) |
| CONVERTWITHPALETTE(1, 32) |
| } |
| |
| break; |
| |
| case 4 : |
| switch(bpp) { |
| CONVERT(4, 8) |
| CONVERTTO16WITHPALETTE(4) |
| CONVERTWITHPALETTE(4, 24) |
| CONVERTWITHPALETTE(4, 32) |
| } |
| |
| break; |
| |
| case 8 : |
| switch(bpp) { |
| CONVERTTO16WITHPALETTE(8) |
| CONVERTWITHPALETTE(8, 24) |
| CONVERTWITHPALETTE(8, 32) |
| } |
| |
| break; |
| |
| case 24 : |
| switch(bpp) { |
| CONVERT(24, 8) |
| CONVERTTO16(24) |
| CONVERT(24, 32) |
| } |
| |
| break; |
| |
| case 32 : |
| switch(bpp) { |
| CONVERT(32, 8) |
| CONVERTTO16(32) |
| CONVERT(32, 24) |
| } |
| |
| break; |
| } |
| } else { |
| memcpy(bits, scanline, FreeImage_GetLine(dib)); |
| } |
| |
| bits += pitch; |
| } |
| } |
| } |