1 /////////////////////////////////////////////////////////////////////////////
3 // Purpose: Implements a PNG reader class + handler
4 // Author: Julian Smart
8 // Copyright: (c) Julian Smart and Markus Holzem
9 // Licence: wxWindows license
10 /////////////////////////////////////////////////////////////////////////////
13 #pragma implementation "pngread.h"
14 #pragma implementation "pnghand.h"
17 // For compilers that support precompilation, includes "wx.h".
18 #include "wx/wxprec.h"
35 #include "wx/palette.h"
36 #include "wx/bitmap.h"
38 #include "wx/msw/pngread.h"
39 #include "wx/msw/dibutils.h"
42 #include "../png/png.h"
45 extern "C" void png_read_init
PNGARG((png_structp png_ptr
));
46 extern "C" void png_write_init
PNGARG((png_structp png_ptr
));
48 #ifndef GlobalAllocPtr
49 #define GlobalPtrHandle(lp) \
50 ((HGLOBAL)GlobalHandle(lp))
52 #define GlobalLockPtr(lp) \
53 ((BOOL)GlobalLock(GlobalPtrHandle(lp)))
54 #define GlobalUnlockPtr(lp) \
55 GlobalUnlock(GlobalPtrHandle(lp))
57 #define GlobalAllocPtr(flags, cb) \
58 (GlobalLock(GlobalAlloc((flags), (cb))))
59 #define GlobalReAllocPtr(lp, cbNew, flags) \
60 (GlobalUnlockPtr(lp), GlobalLock(GlobalReAlloc(GlobalPtrHandle(lp) , (cbNew), (flags))))
61 #define GlobalFreePtr(lp) \
62 (GlobalUnlockPtr(lp), (BOOL)GlobalFree(GlobalPtrHandle(lp)))
67 ima_png_error(png_struct
*png_ptr
, char *message
)
69 // wxMessageBox(message, "PNG error");
71 longjmp(png_ptr
->jmpbuf
, 1);
75 // static wxGifReaderIter* iter;
76 wxPalette
*wxCopyPalette(const wxPalette
*cmap
);
78 wxPNGReader::wxPNGReader()
81 RawImage
= NULL
; // Image data
83 Width
= 0; Height
= 0; // Dimensions
84 Depth
= 0; // (bits x pixel)
85 ColorType
= 0; // Bit 1 = Palette used
89 EfeWidth
= 0; // Efective Width
97 wxPNGReader::wxPNGReader ( wxChar
* ImageFileName
)
101 RawImage
= NULL
; // Image data
103 Width
= 0; Height
= 0; // Dimensions
104 Depth
= 0; // (bits x pixel)
105 ColorType
= 0; // Bit 1 = Palette used
106 // Bit 2 = Color used
107 // Bit 3 = Alpha used
109 EfeWidth
= 0; // Efective Width
115 imageOK
= ReadFile (ImageFileName
);
119 wxPNGReader::Create(int width
, int height
, int depth
, int colortype
)
121 Width
= width
; Height
= height
; Depth
= depth
;
122 ColorType
= (colortype
>=0) ? colortype
: ((Depth
>8) ? COLORTYPE_COLOR
: 0);
126 GlobalFreePtr((unsigned int) lpbi
);
134 lpbi
= wxDibCreate(Depth
, Width
, Height
);
136 RawImage
= (ImagePointerType
)wxDibPtr(lpbi
);
137 EfeWidth
= (long)(((long)Width
*Depth
+ 31) / 32) * 4;
142 wxPNGReader::~wxPNGReader ( )
146 GlobalFreePtr((unsigned int) lpbi
);
155 int wxPNGReader::GetIndex(int x
, int y
)
157 if (!Inside(x
, y
) || (Depth
>8)) return -1;
159 ImagePointerType ImagePointer
= RawImage
+ EfeWidth
*y
+ (x
*Depth
>> 3);
160 int index
= (int)(*ImagePointer
);
164 bool wxPNGReader::GetRGB(int x
, int y
, byte
* r
, byte
* g
, byte
* b
)
166 if (!Inside(x
, y
)) return FALSE
;
169 return Palette
->GetRGB(GetIndex(x
, y
), r
, g
, b
);
170 /* PALETTEENTRY entry;
171 ::GetPaletteEntries((HPALETTE) Palette->GetHPALETTE(), GetIndex(x, y), 1, &entry);
174 *b = entry.peBlue; */
176 ImagePointerType ImagePointer
= RawImage
+ EfeWidth
*y
+ (x
*Depth
>> 3);
177 *b
= ImagePointer
[0];
178 *g
= ImagePointer
[1];
179 *r
= ImagePointer
[2];
185 bool wxPNGReader::SetIndex(int x
, int y
, int index
)
187 if (!Inside(x
, y
) || (Depth
>8)) return FALSE
;
189 ImagePointerType ImagePointer
= RawImage
+ EfeWidth
*y
+ (x
*Depth
>> 3);
190 *ImagePointer
= index
;
195 bool wxPNGReader::SetRGB(int x
, int y
, byte r
, byte g
, byte b
)
197 if (!Inside(x
, y
)) return FALSE
;
199 if (ColorType
& COLORTYPE_PALETTE
)
201 if (!Palette
) return FALSE
;
202 SetIndex(x
, y
, Palette
->GetPixel(r
, g
, b
));
205 ImagePointerType ImagePointer
= RawImage
+ EfeWidth
*y
+ (x
*Depth
>> 3);
214 bool wxPNGReader::SetPalette(wxPalette
* colourmap
)
218 ColorType
|= (COLORTYPE_PALETTE
| COLORTYPE_COLOR
);
220 return (wxDibSetUsage(lpbi
, (HPALETTE
) Palette
->GetHPALETTE(), WXIMA_COLORS
) != 0);
224 wxPNGReader::SetPalette(int n
, byte
*r
, byte
*g
, byte
*b
)
226 Palette
= new wxPalette();
232 Palette
->Create(n
, r
, g
, b
);
233 ColorType
|= (COLORTYPE_PALETTE
| COLORTYPE_COLOR
);
234 return (wxDibSetUsage(lpbi
, (HPALETTE
) Palette
->GetHPALETTE(), WXIMA_COLORS
) != 0);
238 wxPNGReader::SetPalette(int n
, rgb_color_struct
*rgb_struct
)
240 Palette
= new wxPalette();
244 byte r
[256], g
[256], b
[256];
246 for(int i
=0; i
<n
; i
++)
248 r
[i
] = rgb_struct
[i
].red
;
249 g
[i
] = rgb_struct
[i
].green
;
250 b
[i
] = rgb_struct
[i
].blue
;
252 // Added by JACS copying from Andrew Davison's additions
253 // to GIF-reading code
254 // Make transparency colour black...
256 r
[bgindex
] = g
[bgindex
] = b
[bgindex
] = 0;
258 Palette
->Create(n
, r
, g
, b
);
259 ColorType
|= (COLORTYPE_PALETTE
| COLORTYPE_COLOR
);
260 return (wxDibSetUsage(lpbi
, (HPALETTE
) Palette
->GetHPALETTE(), WXIMA_COLORS
) != 0);
263 void wxPNGReader::NullData()
269 wxBitmap
* wxPNGReader::GetBitmap()
271 wxBitmap
*bitmap
= new wxBitmap
;
272 if ( InstantiateBitmap(bitmap
) )
281 bool wxPNGReader::InstantiateBitmap(wxBitmap
*bitmap
)
283 HDC dc
= ::CreateCompatibleDC(NULL
);
287 // tmpBitmap is a dummy, to satisfy ::CreateCompatibleDC (it
288 // is a memory dc that must have a bitmap selected into it)
289 HDC dc2
= GetDC(NULL
);
290 HBITMAP tmpBitmap
= ::CreateCompatibleBitmap(dc2
, GetWidth(), GetHeight());
291 ReleaseDC(NULL
, dc2
);
292 HBITMAP oldBitmap
= (HBITMAP
) ::SelectObject(dc
, tmpBitmap
);
296 ::SelectPalette(dc
, (HPALETTE
) Palette
->GetHPALETTE(), FALSE
);
297 ::RealizePalette(dc
);
300 HBITMAP hBitmap
= ::CreateDIBitmap(dc
, lpbi
,
301 CBM_INIT
, RawImage
, (LPBITMAPINFO
) lpbi
, DIB_PAL_COLORS
);
303 ::SelectPalette(dc
, NULL
, TRUE
);
304 ::SelectObject(dc
, oldBitmap
);
305 ::DeleteObject(tmpBitmap
);
310 bitmap
->SetHBITMAP((WXHBITMAP
) hBitmap
);
311 bitmap
->SetWidth(GetWidth());
312 bitmap
->SetHeight(GetHeight());
313 bitmap
->SetDepth(GetDepth());
314 if ( GetDepth() > 1 && Palette
)
315 bitmap
->SetPalette(*Palette
);
319 // Make a mask if appropriate
322 wxMask
*mask
= CreateMask();
323 bitmap
->SetMask(mask
);
338 wxPalette
*wxCopyPalette(const wxPalette
*cmap
)
340 // To get number of entries...
342 ::GetObject((HPALETTE
) cmap
->GetHPALETTE(), sizeof(WORD
), &count
);
344 LOGPALETTE
* logPal
= (LOGPALETTE
*)
345 new BYTE
[sizeof(LOGPALETTE
) + count
*sizeof(PALETTEENTRY
)];
346 logPal
->palVersion
= 0x300;
347 logPal
->palNumEntries
= count
;
348 ::GetPaletteEntries((HPALETTE
) cmap
->GetHPALETTE(), 0, count
, logPal
->palPalEntry
);
350 HPALETTE hPalette
= ::CreatePalette(logPal
);
353 wxPalette
*newCmap
= new wxPalette
;
354 newCmap
->SetHPALETTE((WXHPALETTE
) hPalette
);
358 wxMask
*wxPNGReader::CreateMask()
360 HBITMAP hBitmap
= ::CreateBitmap(GetWidth(), GetHeight(), 1, 1, NULL
);
362 HDC dc
= ::CreateCompatibleDC(NULL
);
363 HBITMAP oldBitmap
= (HBITMAP
) ::SelectObject(dc
, hBitmap
);
365 int bgIndex
= GetBGIndex();
369 for (x
=0; x
<GetWidth(); x
++)
371 for (y
=0; y
<GetHeight(); y
++)
373 int index
= GetIndex(x
, y
);
374 if ( index
== bgIndex
)
375 ::SetPixel(dc
, x
, GetHeight() - y
- 1, RGB(0, 0, 0));
377 ::SetPixel(dc
, x
, GetHeight() - y
- 1, RGB(255, 255, 255));
381 ::SelectObject(dc
, oldBitmap
);
382 wxMask
*mask
= new wxMask
;
383 mask
->SetMaskBitmap((WXHBITMAP
) hBitmap
);
387 bool wxPNGReader::ReadFile(wxChar
* ImageFileName
)
390 wxStrcpy(filename
, ImageFileName
);
393 FILE *fp
= fopen(wxConvFile
.cWX2MB(filename
), "rb");
397 /* allocate the necessary structures */
398 png_struct
*png_ptr
= new (png_struct
);
405 png_info
*info_ptr
= new (png_info
);
413 /* set error handling */
414 if (setjmp(png_ptr
->jmpbuf
))
416 png_read_destroy(png_ptr
, info_ptr
, (png_info
*)0);
421 /* If we get here, we had a problem reading the file */
424 //png_set_error(ima_png_error, NULL);
426 /* initialize the structures, info first for error handling */
427 png_info_init(info_ptr
);
428 png_read_init(png_ptr
);
430 /* set up the input control */
431 png_init_io(png_ptr
, fp
);
433 /* read the file information */
434 png_read_info(png_ptr
, info_ptr
);
436 /* allocate the memory to hold the image using the fields
438 png_color_16 my_background
={ 0, 31, 127, 255, 0 };
440 if (info_ptr
->valid
& PNG_INFO_bKGD
)
442 png_set_background(png_ptr
, &(info_ptr
->background
),
443 PNG_BACKGROUND_GAMMA_FILE
, 1, 1.0);
444 if ( info_ptr
->num_palette
> 0 )
445 bgindex
= info_ptr
->background
.index
;
448 png_set_background(png_ptr
, &my_background
,
449 PNG_BACKGROUND_GAMMA_SCREEN
, 0, 1.0);
451 // Added by JACS: guesswork!
452 if ( info_ptr
->num_trans
!= 0 )
453 bgindex
= info_ptr
->num_trans
- 1 ;
456 /* tell libpng to strip 16 bit depth files down to 8 bits */
457 if (info_ptr
->bit_depth
== 16)
458 png_set_strip_16(png_ptr
);
460 int pixel_depth
=(info_ptr
->pixel_depth
<24) ? info_ptr
->pixel_depth
: 24;
461 Create(info_ptr
->width
, info_ptr
->height
, pixel_depth
,
462 info_ptr
->color_type
);
464 if (info_ptr
->num_palette
>0)
466 SetPalette((int)info_ptr
->num_palette
, (rgb_color_struct
*)info_ptr
->palette
);
469 int row_stride
= info_ptr
->width
* ((pixel_depth
+7)>>3);
470 // printf("P = %d D = %d RS= %d ", info_ptr->num_palette, info_ptr->pixel_depth,row_stride);
471 // printf("CT = %d TRS = %d BD= %d ", info_ptr->color_type, info_ptr->valid & PNG_INFO_tRNS,info_ptr->bit_depth);
473 byte
*row_pointers
= new byte
[row_stride
];
475 /* turn on interlace handling */
477 if (info_ptr
->interlace_type
)
478 number_passes
= png_set_interlace_handling(png_ptr
);
481 // printf("NP = %d ", number_passes);
483 // don't use the object to prevent warnings from VC++ about "unportable
484 // interaction between setjmp and C++ object destruction" (this is a correct
485 // warning, of course!)
486 wxPNGReaderIter
*iter
= new wxPNGReaderIter(this);
487 for (int pass
=0; pass
< number_passes
; pass
++)
492 //(unsigned char *)iter.GetRow();
493 if (info_ptr
->interlace_type
) {
495 iter
->GetRow(row_pointers
, row_stride
);
496 png_read_row(png_ptr
, row_pointers
, NULL
);
499 png_read_row(png_ptr
, row_pointers
, NULL
);
501 iter
->SetRow(row_pointers
, row_stride
);
503 } while(iter
->PrevRow());
504 // printf("Y=%d ",y);
508 delete[] row_pointers
;
510 /* read the rest of the file, getting any additional chunks
512 png_read_end(png_ptr
, info_ptr
);
514 /* clean up after the read, and free any memory allocated */
515 png_read_destroy(png_ptr
, info_ptr
, (png_info
*)0);
517 /* free the structures */
529 /* write a png file */
531 bool wxPNGReader::SaveFile(wxChar
* ImageFileName
)
534 wxStrcpy(filename
, ImageFileName
);
536 wxPNGReaderIter
iter(this);
542 fp
= fopen(wxConvFile
.cWX2MB(filename
), "wb");
546 /* allocate the necessary structures */
547 png_ptr
= new (png_struct
);
554 info_ptr
= new (png_info
);
562 /* set error handling */
563 if (setjmp(png_ptr
->jmpbuf
))
565 png_write_destroy(png_ptr
);
570 /* If we get here, we had a problem reading the file */
573 //png_set_error(ima_png_error, NULL);
575 // printf("writig pg %s ", filename);
576 /* initialize the structures */
577 png_info_init(info_ptr
);
578 png_write_init(png_ptr
);
580 int row_stride
= GetWidth() * ((GetDepth()+7)>>3);
581 /* set up the output control */
582 png_init_io(png_ptr
, fp
);
584 /* set the file information here */
585 info_ptr
->width
= GetWidth();
586 info_ptr
->height
= GetHeight();
587 info_ptr
->pixel_depth
= GetDepth();
588 info_ptr
->channels
= (GetDepth()>8) ? 3: 1;
589 info_ptr
->bit_depth
= GetDepth()/info_ptr
->channels
;
590 info_ptr
->color_type
= GetColorType();
591 info_ptr
->compression_type
= info_ptr
->filter_type
= info_ptr
->interlace_type
=0;
593 info_ptr
->rowbytes
= row_stride
;
596 // printf("P = %d D = %d RS= %d GD= %d CH= %d ", info_ptr->pixel_depth, info_ptr->bit_depth, row_stride, GetDepth(), info_ptr->channels);
597 /* set the palette if there is one */
598 if ((GetColorType() & COLORTYPE_PALETTE
) && GetPalette())
600 // printf("writing paleta[%d %d %x]",GetColorType() ,COLORTYPE_PALETTE, GetPalette());
601 info_ptr
->valid
|= PNG_INFO_PLTE
;
602 info_ptr
->palette
= new png_color
[256];
603 info_ptr
->num_palette
= 256;
604 for (int i
=0; i
<256; i
++)
605 GetPalette()->GetRGB(i
, &info_ptr
->palette
[i
].red
, &info_ptr
->palette
[i
].green
, &info_ptr
->palette
[i
].blue
);
607 // printf("Paleta [%d %d %x]",GetColorType() ,COLORTYPE_PALETTE, GetPalette());
610 /* optional significant bit chunk */
611 // info_ptr->valid |= PNG_INFO_sBIT;
612 // info_ptr->sig_bit = true_bit_depth;
614 /* optional gamma chunk */
615 // info_ptr->valid |= PNG_INFO_gAMA;
616 // info_ptr->gamma = gamma;
618 /* other optional chunks */
620 /* write the file information */
621 png_write_info(png_ptr
, info_ptr
);
623 /* set up the transformations you want. Note that these are
624 all optional. Only call them if you want them */
626 /* shift the pixels up to a legal bit depth and fill in
627 as appropriate to correctly scale the image */
628 // png_set_shift(png_ptr, &(info_ptr->sig_bit));
630 /* pack pixels into bytes */
631 // png_set_packing(png_ptr);
633 /* flip bgr pixels to rgb */
634 // png_set_bgr(png_ptr);
636 /* swap bytes of 16 bit files to most significant bit first */
637 // png_set_swap(png_ptr);
639 /* get rid of filler bytes, pack rgb into 3 bytes */
640 // png_set_rgbx(png_ptr);
642 /* If you are only writing one row at a time, this works */
644 byte
*row_pointers
= new byte
[row_stride
];
647 // (unsigned char *)iter.GetRow();
648 iter
.GetRow(row_pointers
, row_stride
);
649 png_write_row(png_ptr
, row_pointers
);
650 } while(iter
.PrevRow());
652 delete[] row_pointers
;
654 /* write the rest of the file */
655 png_write_end(png_ptr
, info_ptr
);
657 /* clean up after the write, and free any memory allocated */
658 png_write_destroy(png_ptr
);
660 /* if you malloced the palette, free it here */
661 if (info_ptr
->palette
)
662 delete[] (info_ptr
->palette
);
664 /* free the structures */
675 static int Power(int x
, int y
)
679 for ( i
= 0; i
< y
; i
++)
686 static char hexArray
[] = { '0', '1', '2', '3', '4', '5', '6', '7', '8', '9', 'A', 'B',
687 'C', 'D', 'E', 'F' };
689 static void DecToHex(int dec
, char *buf
)
691 int firstDigit
= (int)(dec
/16.0);
692 int secondDigit
= (int)(dec
- (firstDigit
*16.0));
693 buf
[0] = hexArray
[firstDigit
];
694 buf
[1] = hexArray
[secondDigit
];
699 bool wxPNGReader::SaveXPM(wxChar
*filename
, wxChar
*name
)
703 wxStrcpy(nameStr
, name
);
706 wxStrcpy(nameStr
, filename
);
707 wxStripExtension(nameStr
);
710 if ( GetDepth() > 4 )
712 // Only a depth of 4 and below allowed
719 ofstream
str(wxConvFile
.cWX2MB(filename
));
723 int noColours
= Power(2, GetDepth());
726 str
<< "/* XPM */\n";
727 str
<< "static char * " << nameStr
<< "_xpm[] = {\n";
728 str
<< "\"" << GetWidth() << " " << GetHeight() << " " << noColours
<< " 1\",\n";
731 int base
= 97 ; // start from 'a'
733 unsigned char red
, green
, blue
;
736 for ( i
= 0; i
< noColours
; i
++)
738 str
<< "\"" << (char)(base
+ i
) << " c #";
739 GetPalette()->GetRGB(i
, &red
, &green
, &blue
);
740 DecToHex(red
, hexBuf
);
742 DecToHex(green
, hexBuf
);
744 DecToHex(blue
, hexBuf
);
751 for ( y
= 0; y
< GetHeight(); y
++)
754 for ( x
= 0; x
< GetWidth(); x
++)
756 int index
= GetIndex(x
, y
);
757 str
<< (char)(base
+ index
) ;
768 #include "wx/msw/pnghand.h"
770 IMPLEMENT_DYNAMIC_CLASS(wxPNGFileHandler
, wxBitmapHandler
)
772 bool wxPNGFileHandler::LoadFile(wxBitmap
*bitmap
, const wxString
& name
, long flags
,
773 int desiredWidth
, int desiredHeight
)
776 if (reader
.ReadFile(WXSTRINGCAST name
))
778 return reader
.InstantiateBitmap(bitmap
);
784 bool wxPNGFileHandler::SaveFile(wxBitmap
*bitmap
, const wxString
& name
, int type
, const wxPalette
*pal
)