1 /////////////////////////////////////////////////////////////////////////////
2 // Name: src/common/imagepng.cpp
3 // Purpose: wxImage PNG handler
4 // Author: Robert Roebling
6 // Copyright: (c) Robert Roebling
7 // Licence: wxWindows licence
8 /////////////////////////////////////////////////////////////////////////////
10 // ============================================================================
12 // ============================================================================
14 // ----------------------------------------------------------------------------
16 // ----------------------------------------------------------------------------
18 // For compilers that support precompilation, includes "wx.h".
19 #include "wx/wxprec.h"
29 #if wxUSE_IMAGE && wxUSE_LIBPNG
31 #include "wx/imagpng.h"
32 #include "wx/bitmap.h"
37 #include "wx/filefn.h"
38 #include "wx/wfstream.h"
40 #include "wx/module.h"
51 // ----------------------------------------------------------------------------
53 // ----------------------------------------------------------------------------
55 // image can not have any transparent pixels at all, have only 100% opaque
56 // and/or 100% transparent pixels in which case a simple mask is enough to
57 // store this information in wxImage or have a real alpha channel in which case
58 // we need to have it in wxImage as well
66 // ----------------------------------------------------------------------------
68 // ----------------------------------------------------------------------------
70 // return the kind of transparency needed for this image assuming that it does
71 // have transparent pixels, i.e. either Transparency_Alpha or Transparency_Mask
73 CheckTransparency(unsigned char **lines
,
74 png_uint_32 x
, png_uint_32 y
, png_uint_32 w
, png_uint_32 h
,
77 // init the alpha channel for the image and fill it with 1s up to (x, y)
78 static unsigned char *InitAlpha(wxImage
*image
, png_uint_32 x
, png_uint_32 y
);
80 // find a free colour for the mask in the PNG data array
82 FindMaskColour(unsigned char **lines
, png_uint_32 width
, png_uint_32 height
,
83 unsigned char& rMask
, unsigned char& gMask
, unsigned char& bMask
);
85 // is the pixel with this value of alpha a fully opaque one?
87 bool IsOpaque(unsigned char a
)
92 // is the pixel with this value of alpha a fully transparent one?
94 bool IsTransparent(unsigned char a
)
99 // ============================================================================
100 // wxPNGHandler implementation
101 // ============================================================================
103 IMPLEMENT_DYNAMIC_CLASS(wxPNGHandler
,wxImageHandler
)
107 #ifndef PNGLINKAGEMODE
108 #if defined(__WATCOMC__) && ( defined(__WXMSW__) || defined(__WXMGL__) )
109 // we need an explicit cdecl for Watcom, at least according to
111 // http://sf.net/tracker/index.php?func=detail&aid=651492&group_id=9863&atid=109863
113 // more testing is needed for this however, please remove this comment
114 // if you can confirm that my fix works with Watcom 11
115 #define PNGLINKAGEMODE cdecl
117 #define PNGLINKAGEMODE LINKAGEMODE
122 // VS: wxPNGInfoStruct declared below is a hack that needs some explanation.
123 // First, let me describe what's the problem: libpng uses jmp_buf in
124 // its png_struct structure. Unfortunately, this structure is
125 // compiler-specific and may vary in size, so if you use libpng compiled
126 // as DLL with another compiler than the main executable, it may not work
127 // (this is for example the case with wxMGL port and SciTech MGL library
128 // that provides custom runtime-loadable libpng implementation with jmpbuf
129 // disabled altogether). Luckily, it is still possible to use setjmp() &
130 // longjmp() as long as the structure is not part of png_struct.
132 // Sadly, there's no clean way to attach user-defined data to png_struct.
133 // There is only one customizable place, png_struct.io_ptr, which is meant
134 // only for I/O routines and is set with png_set_read_fn or
135 // png_set_write_fn. The hacky part is that we use io_ptr to store
136 // a pointer to wxPNGInfoStruct that holds I/O structures _and_ jmp_buf.
138 struct wxPNGInfoStruct
150 #define WX_PNG_INFO(png_ptr) ((wxPNGInfoStruct*)png_get_io_ptr(png_ptr))
152 // ----------------------------------------------------------------------------
154 // ----------------------------------------------------------------------------
159 void PNGLINKAGEMODE
wx_PNG_stream_reader( png_structp png_ptr
, png_bytep data
,
162 WX_PNG_INFO(png_ptr
)->stream
.in
->Read(data
, length
);
165 void PNGLINKAGEMODE
wx_PNG_stream_writer( png_structp png_ptr
, png_bytep data
,
168 WX_PNG_INFO(png_ptr
)->stream
.out
->Write(data
, length
);
172 PNGLINKAGEMODE
wx_png_warning(png_structp png_ptr
, png_const_charp message
)
174 wxPNGInfoStruct
*info
= png_ptr
? WX_PNG_INFO(png_ptr
) : NULL
;
175 if ( !info
|| info
->verbose
)
176 wxLogWarning( wxString::FromAscii(message
) );
180 // so that the libpng doesn't send anything on stderr
182 PNGLINKAGEMODE
wx_png_error(png_structp png_ptr
, png_const_charp message
)
184 wx_png_warning(NULL
, message
);
186 // we're not using libpng built-in jump buffer (see comment before
187 // wxPNGInfoStruct above) so we have to return ourselves, otherwise libpng
189 longjmp(WX_PNG_INFO(png_ptr
)->jmpbuf
, 1);
194 // ----------------------------------------------------------------------------
195 // LoadFile() helpers
196 // ----------------------------------------------------------------------------
198 // determine the kind of transparency we need for this image: if the only alpha
199 // values it has are 0 (transparent) and 0xff (opaque) then we can simply
200 // create a mask for it, we should be ok with a simple mask but otherwise we
201 // need a full blown alpha channel in wxImage
204 // lines raw PNG data
205 // x, y starting position
206 // w, h size of the image
207 // numColBytes number of colour bytes (1 for grey scale, 3 for RGB)
208 // (NB: alpha always follows the colour bytes)
210 CheckTransparency(unsigned char **lines
,
211 png_uint_32 x
, png_uint_32 y
, png_uint_32 w
, png_uint_32 h
,
214 // suppose that a mask will suffice and check all the remaining alpha
215 // values to see if it does
218 // each pixel is numColBytes+1 bytes, offset into the current line by
219 // the current x position
220 unsigned const char *ptr
= lines
[y
] + (x
* (numColBytes
+ 1));
222 for ( png_uint_32 x2
= x
; x2
< w
; x2
++ )
224 // skip the grey or colour byte(s)
227 unsigned char a2
= *ptr
++;
229 if ( !IsTransparent(a2
) && !IsOpaque(a2
) )
231 // not fully opaque nor fully transparent, hence need alpha
232 return Transparency_Alpha
;
236 // during the next loop iteration check all the pixels in the row
240 // mask will be enough
241 return Transparency_Mask
;
244 unsigned char *InitAlpha(wxImage
*image
, png_uint_32 x
, png_uint_32 y
)
246 // create alpha channel
249 unsigned char *alpha
= image
->GetAlpha();
251 // set alpha for the pixels we had so far
252 png_uint_32 end
= y
* image
->GetWidth() + x
;
253 for ( png_uint_32 i
= 0; i
< end
; i
++ )
255 // all the previous pixels were opaque
263 FindMaskColour(unsigned char **lines
, png_uint_32 width
, png_uint_32 height
,
264 unsigned char& rMask
, unsigned char& gMask
, unsigned char& bMask
)
266 // choosing the colour for the mask is more
267 // difficult: we need to iterate over the entire
268 // image for this in order to choose an unused
269 // colour (this is not very efficient but what else
272 unsigned nentries
= 0;
273 unsigned char r2
, g2
, b2
;
274 for ( png_uint_32 y2
= 0; y2
< height
; y2
++ )
276 const unsigned char *p
= lines
[y2
];
277 for ( png_uint_32 x2
= 0; x2
< width
; x2
++ )
283 wxImageHistogramEntry
&
284 entry
= h
[wxImageHistogram:: MakeKey(r2
, g2
, b2
)];
286 if ( entry
.value
++ == 0 )
287 entry
.index
= nentries
++;
291 if ( !h
.FindFirstUnusedColour(&rMask
, &gMask
, &bMask
) )
293 wxLogWarning(_("Too many colours in PNG, the image may be slightly blurred."));
295 // use a fixed mask colour and we'll fudge
296 // the real pixels with this colour (see
304 // ----------------------------------------------------------------------------
306 // ----------------------------------------------------------------------------
308 bool wxPNGHandler::DoCanRead( wxInputStream
& stream
)
310 unsigned char hdr
[4];
312 if ( !stream
.Read(hdr
, WXSIZEOF(hdr
)) )
315 return memcmp(hdr
, "\211PNG", WXSIZEOF(hdr
)) == 0;
318 // convert data from RGB to wxImage format
320 void CopyDataFromPNG(wxImage
*image
,
321 unsigned char **lines
,
326 Transparency transparency
= Transparency_None
;
328 // only non NULL if transparency == Transparency_Alpha
329 unsigned char *alpha
= NULL
;
331 // RGB of the mask colour if transparency == Transparency_Mask
332 // (but init them anyhow to avoid compiler warnings)
333 unsigned char rMask
= 0,
337 unsigned char *ptrDst
= image
->GetData();
338 if ( !(color_type
& PNG_COLOR_MASK_COLOR
) )
340 // grey image: GAGAGA... where G == grey component and A == alpha
341 for ( png_uint_32 y
= 0; y
< height
; y
++ )
343 const unsigned char *ptrSrc
= lines
[y
];
344 for ( png_uint_32 x
= 0; x
< width
; x
++ )
346 unsigned char g
= *ptrSrc
++;
347 unsigned char a
= *ptrSrc
++;
349 // the first time we encounter a transparent pixel we must
350 // decide about what to do about them
351 if ( !IsOpaque(a
) && transparency
== Transparency_None
)
353 // we'll need at least the mask for this image and
354 // maybe even full alpha channel info: the former is
355 // only enough if we have alpha values of 0 and 0xff
356 // only, otherwisewe need the latter
357 transparency
= CheckTransparency
365 if ( transparency
== Transparency_Mask
)
367 // let's choose this colour for the mask: this is
368 // not a problem here as all the other pixels are
369 // grey, i.e. R == G == B which is not the case for
370 // this one so no confusion is possible
375 else // transparency == Transparency_Alpha
377 alpha
= InitAlpha(image
, x
, y
);
381 switch ( transparency
)
383 case Transparency_Mask
:
384 if ( IsTransparent(a
) )
391 // else: !transparent
393 // must be opaque then as otherwise we shouldn't be
394 // using the mask at all
395 wxASSERT_MSG( IsOpaque(a
), _T("logic error") );
399 case Transparency_Alpha
:
404 case Transparency_None
:
413 else // colour image: RGBRGB...
415 for ( png_uint_32 y
= 0; y
< height
; y
++ )
417 const unsigned char *ptrSrc
= lines
[y
];
418 for ( png_uint_32 x
= 0; x
< width
; x
++ )
420 unsigned char r
= *ptrSrc
++;
421 unsigned char g
= *ptrSrc
++;
422 unsigned char b
= *ptrSrc
++;
423 unsigned char a
= *ptrSrc
++;
425 // the logic here is the same as for the grey case except
427 if ( !IsOpaque(a
) && transparency
== Transparency_None
)
429 transparency
= CheckTransparency
437 if ( transparency
== Transparency_Mask
)
439 FindMaskColour(lines
, width
, height
,
440 rMask
, gMask
, bMask
);
442 else // transparency == Transparency_Alpha
444 alpha
= InitAlpha(image
, x
, y
);
449 switch ( transparency
)
451 case Transparency_Mask
:
452 if ( IsTransparent(a
) )
461 // must be opaque then as otherwise we shouldn't be
462 // using the mask at all
463 wxASSERT_MSG( IsOpaque(a
), _T("logic error") );
465 // if we couldn't find a unique colour for the
466 // mask, we can have real pixels with the same
467 // value as the mask and it's better to slightly
468 // change their colour than to make them
470 if ( r
== rMask
&& g
== gMask
&& b
== bMask
)
478 case Transparency_Alpha
:
483 case Transparency_None
:
493 if ( transparency
== Transparency_Mask
)
495 image
->SetMaskColour(rMask
, gMask
, bMask
);
499 // temporarily disable the warning C4611 (interaction between '_setjmp' and
500 // C++ object destruction is non-portable) - I don't see any dtors here
502 #pragma warning(disable:4611)
506 wxPNGHandler::LoadFile(wxImage
*image
,
507 wxInputStream
& stream
,
511 // VZ: as this function uses setjmp() the only fool-proof error handling
512 // method is to use goto (setjmp is not really C++ dtors friendly...)
514 unsigned char **lines
= NULL
;
515 png_infop info_ptr
= (png_infop
) NULL
;
516 wxPNGInfoStruct wxinfo
;
518 png_uint_32 i
, width
, height
= 0;
519 int bit_depth
, color_type
, interlace_type
;
521 wxinfo
.verbose
= verbose
;
522 wxinfo
.stream
.in
= &stream
;
526 png_structp png_ptr
= png_create_read_struct
528 PNG_LIBPNG_VER_STRING
,
536 // NB: please see the comment near wxPNGInfoStruct declaration for
537 // explanation why this line is mandatory
538 png_set_read_fn( png_ptr
, &wxinfo
, wx_PNG_stream_reader
);
540 info_ptr
= png_create_info_struct( png_ptr
);
544 if (setjmp(wxinfo
.jmpbuf
))
547 png_read_info( png_ptr
, info_ptr
);
548 png_get_IHDR( png_ptr
, info_ptr
, &width
, &height
, &bit_depth
, &color_type
, &interlace_type
, (int*) NULL
, (int*) NULL
);
550 if (color_type
== PNG_COLOR_TYPE_PALETTE
)
551 png_set_expand( png_ptr
);
553 // Fix for Bug [ 439207 ] Monochrome PNG images come up black
555 png_set_expand( png_ptr
);
557 png_set_strip_16( png_ptr
);
558 png_set_packing( png_ptr
);
559 if (png_get_valid( png_ptr
, info_ptr
, PNG_INFO_tRNS
))
560 png_set_expand( png_ptr
);
561 png_set_filler( png_ptr
, 0xff, PNG_FILLER_AFTER
);
563 image
->Create((int)width
, (int)height
, (bool) false /* no need to init pixels */);
568 lines
= (unsigned char **)malloc( (size_t)(height
* sizeof(unsigned char *)) );
572 for (i
= 0; i
< height
; i
++)
574 if ((lines
[i
] = (unsigned char *)malloc( (size_t)(width
* (sizeof(unsigned char) * 4)))) == NULL
)
576 for ( unsigned int n
= 0; n
< i
; n
++ )
582 png_read_image( png_ptr
, lines
);
583 png_read_end( png_ptr
, info_ptr
);
584 png_destroy_read_struct( &png_ptr
, &info_ptr
, (png_infopp
) NULL
);
586 // loaded successfully, now init wxImage with this data
587 CopyDataFromPNG(image
, lines
, width
, height
, color_type
);
589 for ( i
= 0; i
< height
; i
++ )
597 wxLogError(_("Couldn't load a PNG image - file is corrupted or not enough memory."));
606 for ( unsigned int n
= 0; n
< height
; n
++ )
616 png_destroy_read_struct( &png_ptr
, &info_ptr
, (png_infopp
) NULL
);
620 png_destroy_read_struct( &png_ptr
, (png_infopp
) NULL
, (png_infopp
) NULL
);
625 // ----------------------------------------------------------------------------
627 // ----------------------------------------------------------------------------
629 bool wxPNGHandler::SaveFile( wxImage
*image
, wxOutputStream
& stream
, bool verbose
)
631 wxPNGInfoStruct wxinfo
;
633 wxinfo
.verbose
= verbose
;
634 wxinfo
.stream
.out
= &stream
;
636 png_structp png_ptr
= png_create_write_struct
638 PNG_LIBPNG_VER_STRING
,
646 wxLogError(_("Couldn't save PNG image."));
650 png_infop info_ptr
= png_create_info_struct(png_ptr
);
651 if (info_ptr
== NULL
)
653 png_destroy_write_struct( &png_ptr
, (png_infopp
)NULL
);
655 wxLogError(_("Couldn't save PNG image."));
659 if (setjmp(wxinfo
.jmpbuf
))
661 png_destroy_write_struct( &png_ptr
, (png_infopp
)NULL
);
663 wxLogError(_("Couldn't save PNG image."));
667 // NB: please see the comment near wxPNGInfoStruct declaration for
668 // explanation why this line is mandatory
669 png_set_write_fn( png_ptr
, &wxinfo
, wx_PNG_stream_writer
, NULL
);
671 const int iColorType
= image
->HasOption(wxIMAGE_OPTION_PNG_FORMAT
)
672 ? image
->GetOptionInt(wxIMAGE_OPTION_PNG_FORMAT
)
674 const int iBitDepth
= image
->HasOption(wxIMAGE_OPTION_PNG_BITDEPTH
)
675 ? image
->GetOptionInt(wxIMAGE_OPTION_PNG_BITDEPTH
)
678 wxASSERT_MSG( iBitDepth
== 8 || iBitDepth
== 16,
679 _T("PNG bit depth must be 8 or 16") );
681 bool bHasAlpha
= image
->HasAlpha();
682 bool bHasMask
= image
->HasMask();
683 bool bUseAlpha
= bHasAlpha
|| bHasMask
;
686 if ( iColorType
==wxPNG_TYPE_COLOUR
)
688 iPngColorType
= bUseAlpha
? PNG_COLOR_TYPE_RGB_ALPHA
689 : PNG_COLOR_TYPE_RGB
;
693 iPngColorType
= bUseAlpha
? PNG_COLOR_TYPE_GRAY_ALPHA
694 : PNG_COLOR_TYPE_GRAY
;
697 png_set_IHDR( png_ptr
, info_ptr
, image
->GetWidth(), image
->GetHeight(),
698 iBitDepth
, iPngColorType
,
699 PNG_INTERLACE_NONE
, PNG_COMPRESSION_TYPE_BASE
,
700 PNG_FILTER_TYPE_BASE
);
705 if ( iPngColorType
& PNG_COLOR_MASK_COLOR
)
709 sig_bit
.blue
= (png_byte
)iBitDepth
;
714 sig_bit
.gray
= (png_byte
)iBitDepth
;
718 if ( iPngColorType
& PNG_COLOR_MASK_ALPHA
)
720 sig_bit
.alpha
= (png_byte
)iBitDepth
;
724 if ( iBitDepth
== 16 )
727 png_set_sBIT( png_ptr
, info_ptr
, &sig_bit
);
728 png_write_info( png_ptr
, info_ptr
);
729 png_set_shift( png_ptr
, &sig_bit
);
730 png_set_packing( png_ptr
);
733 data
= (unsigned char *)malloc( image
->GetWidth() * iElements
);
736 png_destroy_write_struct( &png_ptr
, (png_infopp
)NULL
);
741 pAlpha
= (unsigned char *)(bHasAlpha
? image
->GetAlpha() : NULL
);
742 int iHeight
= image
->GetHeight();
743 int iWidth
= image
->GetWidth();
745 unsigned char uchMaskRed
= 0, uchMaskGreen
= 0, uchMaskBlue
= 0;
749 uchMaskRed
= image
->GetMaskRed();
750 uchMaskGreen
= image
->GetMaskGreen();
751 uchMaskBlue
= image
->GetMaskBlue();
754 unsigned char *pColors
= image
->GetData();
756 for (int y
= 0; y
!= iHeight
; ++y
)
758 unsigned char *pData
= data
;
759 for (int x
= 0; x
!= iWidth
; x
++)
761 unsigned char uchRed
= *pColors
++;
762 unsigned char uchGreen
= *pColors
++;
763 unsigned char uchBlue
= *pColors
++;
765 switch ( iColorType
)
768 wxFAIL_MSG( _T("unknown wxPNG_TYPE_XXX") );
771 case wxPNG_TYPE_COLOUR
:
773 if ( iBitDepth
== 16 )
776 if ( iBitDepth
== 16 )
779 if ( iBitDepth
== 16 )
783 case wxPNG_TYPE_GREY
:
785 // where do these coefficients come from? maybe we
786 // should have image options for them as well?
788 (unsigned) (76.544*(unsigned)uchRed
+
789 150.272*(unsigned)uchGreen
+
790 36.864*(unsigned)uchBlue
);
792 *pData
++ = (unsigned char)((uiColor
>> 8) & 0xFF);
793 if ( iBitDepth
== 16 )
794 *pData
++ = (unsigned char)(uiColor
& 0xFF);
798 case wxPNG_TYPE_GREY_RED
:
800 if ( iBitDepth
== 16 )
807 unsigned char uchAlpha
= 255;
809 uchAlpha
= *pAlpha
++;
813 if ( (uchRed
== uchMaskRed
)
814 && (uchGreen
== uchMaskGreen
)
815 && (uchBlue
== uchMaskBlue
) )
820 if ( iBitDepth
== 16 )
825 png_bytep row_ptr
= data
;
826 png_write_rows( png_ptr
, &row_ptr
, 1 );
830 png_write_end( png_ptr
, info_ptr
);
831 png_destroy_write_struct( &png_ptr
, (png_infopp
)&info_ptr
);
837 #pragma warning(default:4611)
840 #endif // wxUSE_STREAMS
842 #endif // wxUSE_LIBPNG