]> git.saurik.com Git - wxWidgets.git/blob - src/mac/carbon/pnghand.cpp
Include wx/msgdlg.h according to precompiled headers of wx/wx.h (with other minor...
[wxWidgets.git] / src / mac / carbon / pnghand.cpp
1 /////////////////////////////////////////////////////////////////////////////
2 // Name: pnghand.cpp
3 // Purpose: Implements a PNG reader class + handler
4 // Author: Julian Smart
5 // Modified by:
6 // Created: 04/01/98
7 // RCS-ID: $Id$
8 // Copyright: (c) Julian Smart
9 // Licence: wxWindows licence
10 /////////////////////////////////////////////////////////////////////////////
11
12 // For compilers that support precompilation, includes "wx.h".
13 #include "wx/wxprec.h"
14
15 #ifdef __BORLANDC__
16 # pragma hdrstop
17 #endif
18
19 #if 0 // wxUSE_LIBPNG
20
21 #include <stdlib.h>
22 #include <stdio.h>
23 #include <string.h>
24
25 #if defined(__DARWIN__)
26 /* MW's math routines do certain things if __FP__ (the include guard around
27 CarbonCore's fp.h) is defined. CarbonCore's fp.h does certain things if
28 __cmath__ is defined so it seems the easy thing to do is to make sure
29 __cmath__ is effectively not defined which counteracts the MWERKS check
30 then when the real cmath is included everything will be okay.
31 */
32 #include <CoreServices/CoreServices.h>
33 //#include <cmath>
34 #endif //defined(__DARWIN__)
35
36 #if wxUSE_IOSTREAMH
37 # include <fstream.h>
38 #else
39 # include <fstream>
40 #endif
41
42 #ifndef __DARWIN__
43 # include <windows.h>
44 #endif
45 #include "wx/msgdlg.h"
46 #include "wx/palette.h"
47 #include "wx/bitmap.h"
48 #include "wx/mac/pnghand.h"
49 #include "wx/mac/pngread.h"
50 #include "wx/mac/private.h"
51
52 extern "C" {
53 #include "png.h"
54 }
55
56 extern "C" void png_read_init PNGARG((png_structp png_ptr));
57 extern "C" void png_write_init PNGARG((png_structp png_ptr));
58
59 extern CTabHandle wxMacCreateColorTable( int numColors ) ;
60 extern void wxMacDestroyColorTable( CTabHandle colors ) ;
61 extern void wxMacSetColorTableEntry( CTabHandle newColors , int index , int red , int green , int blue ) ;
62 extern GWorldPtr wxMacCreateGWorld( int width , int height , int depth ) ;
63 extern void wxMacDestroyGWorld( GWorldPtr gw ) ;
64
65 void
66 ima_png_error(png_struct *png_ptr, char *message)
67 {
68 wxMessageBox(wxString::FromAscii(message), wxT("PNG error"));
69 longjmp(png_ptr->jmpbuf, 1);
70 }
71
72
73 // static wxGifReaderIter* iter;
74 wxPalette *wxCopyPalette(const wxPalette *cmap);
75
76 wxPNGReader::wxPNGReader(void)
77 {
78 filetype = 0;
79 RawImage = NULL; // Image data
80
81 Width = 0; Height = 0; // Dimensions
82 Depth = 0; // (bits x pixel)
83 ColorType = 0; // Bit 1 = Palette used
84 // Bit 2 = Color used
85 // Bit 3 = Alpha used
86
87 EfeWidth = 0; // Efective Width
88
89 lpbi = NULL;
90 bgindex = -1;
91 m_palette = 0;
92 imageOK = FALSE;
93 }
94
95 wxPNGReader::wxPNGReader ( char* ImageFileName )
96 {
97 imageOK = FALSE;
98 filetype = 0;
99 RawImage = NULL; // Image data
100
101 Width = 0; Height = 0; // Dimensions
102 Depth = 0; // (bits x pixel)
103 ColorType = 0; // Bit 1 = m_palette used
104 // Bit 2 = Color used
105 // Bit 3 = Alpha used
106
107 EfeWidth = 0; // Efective Width
108
109 lpbi = NULL;
110 bgindex = -1;
111 m_palette = 0;
112
113 imageOK = ReadFile (ImageFileName);
114 }
115
116 void
117 wxPNGReader::Create(int width, int height, int depth, int colortype)
118 {
119 Width = width; Height = height; Depth = depth;
120 ColorType = (colortype>=0) ? colortype: ((Depth>8) ? COLORTYPE_COLOR: 0);
121 delete m_palette;
122 m_palette = NULL;
123 delete[] RawImage;
124 RawImage = NULL;
125
126 if (lpbi) {
127 wxMacDestroyGWorld( (GWorldPtr) lpbi ) ;
128 }
129 lpbi = wxMacCreateGWorld( Width , Height , Depth);
130 if (lpbi)
131 {
132 EfeWidth = (long)(((long)Width*Depth + 31) / 32) * 4;
133 int bitwidth = width ;
134 if ( EfeWidth > bitwidth )
135 bitwidth = EfeWidth ;
136
137 RawImage = (byte*) new char[ ( bitwidth * Height * ((Depth+7)>>3) ) ];
138 imageOK = TRUE;
139 }
140 }
141
142 wxPNGReader::~wxPNGReader ( )
143 {
144 if (RawImage != NULL) {
145 delete[] RawImage ;
146 RawImage = NULL;
147 }
148 if (lpbi) {
149 wxMacDestroyGWorld( (GWorldPtr) lpbi ) ;
150 lpbi = NULL;
151 }
152 if (m_palette != NULL) {
153 delete m_palette;
154 m_palette = NULL;
155 }
156 }
157
158
159 int wxPNGReader::GetIndex(int x, int y)
160 {
161 if (!Inside(x, y) || (Depth>8)) return -1;
162
163 ImagePointerType ImagePointer = RawImage + EfeWidth*y + (x*Depth >> 3);
164 int index = (int)(*ImagePointer);
165 return index;
166 }
167
168 bool wxPNGReader::GetRGB(int x, int y, byte* r, byte* g, byte* b)
169 {
170 if (!Inside(x, y)) return FALSE;
171
172 if (m_palette) {
173 return m_palette->GetRGB(GetIndex(x, y), r, g, b);
174 /* PALETTEENTRY entry;
175 ::GetPaletteEntries((HPALETTE) m_palette->GetHPALETTE(), GetIndex(x, y), 1, &entry);
176 *r = entry.peRed;
177 *g = entry.peGreen;
178 *b = entry.peBlue; */
179 } else {
180 ImagePointerType ImagePointer = RawImage + EfeWidth*y + (x*Depth >> 3);
181 *b = ImagePointer[0];
182 *g = ImagePointer[1];
183 *r = ImagePointer[2];
184 }
185 return TRUE;
186 }
187
188
189 bool wxPNGReader::SetIndex(int x, int y, int index)
190 {
191 if (!Inside(x, y) || (Depth>8)) return FALSE;
192
193 ImagePointerType ImagePointer = RawImage + EfeWidth*y + (x*Depth >> 3);
194 *ImagePointer = index;
195
196 return TRUE;
197 }
198
199 bool wxPNGReader::SetRGB(int x, int y, byte r, byte g, byte b)
200 {
201 if (!Inside(x, y)) return FALSE;
202
203 if (ColorType & COLORTYPE_PALETTE)
204 {
205 if (!m_palette) return FALSE;
206 SetIndex(x, y, m_palette->GetPixel(r, g, b));
207
208 } else {
209 ImagePointerType ImagePointer = RawImage + EfeWidth*y + (x*Depth >> 3);
210 ImagePointer[0] = b;
211 ImagePointer[1] = g;
212 ImagePointer[2] = r;
213 }
214
215 return TRUE;
216 }
217
218 bool wxPNGReader::SetPalette(wxPalette* colourmap)
219 {
220 delete m_palette ;
221 if (!colourmap)
222 return FALSE;
223 ColorType |= (COLORTYPE_PALETTE | COLORTYPE_COLOR);
224 m_palette = new wxPalette( *colourmap );
225 return true ;
226 // return (DibSetUsage(lpbi, (HPALETTE) m_palette->GetHPALETTE(), WXIMA_COLORS ) != 0);
227 }
228
229 bool
230 wxPNGReader::SetPalette(int n, byte *r, byte *g, byte *b)
231 {
232 delete m_palette ;
233 m_palette = new wxPalette();
234 if (!m_palette)
235 return FALSE;
236
237 if (!g) g = r;
238 if (!b) b = g;
239 m_palette->Create(n, r, g, b);
240 ColorType |= (COLORTYPE_PALETTE | COLORTYPE_COLOR);
241 return true ;
242 // return (DibSetUsage(lpbi, (HPALETTE) m_palette->GetHPALETTE(), WXIMA_COLORS ) != 0);
243 }
244
245 bool
246 wxPNGReader::SetPalette(int n, rgb_color_struct *rgb_struct)
247 {
248 delete m_palette ;
249 m_palette = new wxPalette();
250 if (!m_palette)
251 return FALSE;
252
253 byte r[256], g[256], b[256];
254
255 for(int i=0; i<n; i++)
256 {
257 r[i] = rgb_struct[i].red;
258 g[i] = rgb_struct[i].green;
259 b[i] = rgb_struct[i].blue;
260 }
261 // Added by JACS copying from Andrew Davison's additions
262 // to GIF-reading code
263 // Make transparency colour black...
264 if (bgindex != -1)
265 r[bgindex] = g[bgindex] = b[bgindex] = 0;
266
267 m_palette->Create(n, r, g, b);
268 ColorType |= (COLORTYPE_PALETTE | COLORTYPE_COLOR);
269 return true ;
270 // return (DibSetUsage(lpbi, (HPALETTE) m_palette->GetHPALETTE(), WXIMA_COLORS ) != 0);
271 }
272
273 void wxPNGReader::NullData()
274 {
275 if (lpbi) {
276 wxMacDestroyGWorld( (GWorldPtr) lpbi ) ;
277 lpbi = NULL;
278 }
279 if (m_palette != NULL) {
280 delete m_palette;
281 m_palette = NULL;
282 }
283 }
284
285 wxBitmap* wxPNGReader::GetBitmap(void)
286 {
287 wxBitmap *bitmap = new wxBitmap;
288 if ( InstantiateBitmap(bitmap) )
289 return bitmap;
290 else
291 {
292 delete bitmap;
293 return NULL;
294 }
295 }
296
297 bool wxPNGReader::InstantiateBitmap(wxBitmap *bitmap)
298 {
299 if ( lpbi )
300 {
301 bitmap->SetHBITMAP((WXHBITMAP) lpbi);
302 bitmap->SetWidth(GetWidth());
303 bitmap->SetHeight(GetHeight());
304 bitmap->SetDepth(GetDepth());
305 if ( GetDepth() > 1 && m_palette )
306 bitmap->SetPalette(*m_palette);
307 bitmap->SetOk(TRUE);
308
309
310 // Make a mask if appropriate
311 /*
312 if ( bgindex > -1 )
313 {
314 wxMask *mask = CreateMask();
315 bitmap->SetMask(mask);
316 }
317 */
318 lpbi = NULL ; // bitmap has taken over ownership
319 return TRUE;
320 }
321 else
322 {
323 return FALSE;
324 }
325 /*
326 HDC dc = ::CreateCompatibleDC(NULL);
327
328 if (dc)
329 {
330 // tmpBitmap is a dummy, to satisfy ::CreateCompatibleDC (it
331 // is a memory dc that must have a bitmap selected into it)
332 HDC dc2 = GetDC(NULL);
333 HBITMAP tmpBitmap = ::CreateCompatibleBitmap(dc2, GetWidth(), GetHeight());
334 ReleaseDC(NULL, dc2);
335 HBITMAP oldBitmap = (HBITMAP) ::SelectObject(dc, tmpBitmap);
336
337 if ( m_palette )
338 {
339 HPALETTE oldPal = ::SelectPalette(dc, (HPALETTE) m_palette->GetHPALETTE(), FALSE);
340 ::RealizePalette(dc);
341 }
342
343 HBITMAP hBitmap = ::CreateDIBitmap(dc, lpbi,
344 CBM_INIT, RawImage, (LPBITMAPINFO) lpbi, DIB_PAL_COLORS);
345
346 ::SelectPalette(dc, NULL, TRUE);
347 ::SelectObject(dc, oldBitmap);
348 ::DeleteObject(tmpBitmap);
349 ::DeleteDC(dc);
350
351 if ( hBitmap )
352 {
353 bitmap->SetHBITMAP((WXHBITMAP) hBitmap);
354 bitmap->SetWidth(GetWidth());
355 bitmap->SetHeight(GetHeight());
356 bitmap->SetDepth(GetDepth());
357 if ( GetDepth() > 1 && m_palette )
358 bitmap->SetPalette(*m_palette);
359 bitmap->SetOk(TRUE);
360
361
362 // Make a mask if appropriate
363 if ( bgindex > -1 )
364 {
365 wxMask *mask = CreateMask();
366 bitmap->SetMask(mask);
367 }
368 return TRUE;
369 }
370 else
371 {
372 return FALSE;
373 }
374 }
375 else
376 {
377 return FALSE;
378 }
379 */
380 return false ;
381 }
382
383 wxPalette *wxCopyPalette(const wxPalette *cmap)
384 {
385 wxPalette *newCmap = new wxPalette( *cmap ) ;
386 return newCmap;
387 }
388
389 wxMask *wxPNGReader::CreateMask(void)
390 {
391 /*
392 HBITMAP hBitmap = ::CreateBitmap(GetWidth(), GetHeight(), 1, 1, NULL);
393
394 HDC dc = ::CreateCompatibleDC(NULL);
395 HBITMAP oldBitmap = (HBITMAP) ::SelectObject(dc, hBitmap);
396
397 int bgIndex = GetBGIndex();
398
399 int x,y;
400
401 for (x=0; x<GetWidth(); x++)
402 {
403 for (y=0; y<GetHeight(); y++)
404 {
405 int index = GetIndex(x, y);
406 if ( index == bgIndex )
407 ::SetPixel(dc, x, GetHeight() - y - 1, RGB(0, 0, 0));
408 else
409 ::SetPixel(dc, x, GetHeight() - y - 1, RGB(255, 255, 255));
410
411 }
412 }
413 ::SelectObject(dc, oldBitmap);
414 wxMask *mask = new wxMask;
415 mask->SetMaskBitmap((WXHBITMAP) hBitmap);
416 return mask;
417 */
418 return NULL ;
419 }
420
421 bool wxPNGReader::ReadFile(char * ImageFileName)
422 {
423 int number_passes;
424
425 if (ImageFileName)
426 strcpy(filename, ImageFileName);
427
428 FILE *fp;
429 png_struct *png_ptr;
430 png_info *info_ptr;
431 wxPNGReaderIter iter(this);
432
433 /* open the file */
434 fp = fopen( ImageFileName , "rb" );
435
436 if (!fp)
437 return FALSE;
438
439 /* allocate the necessary structures */
440 png_ptr = new (png_struct);
441 if (!png_ptr)
442 {
443 fclose(fp);
444 return FALSE;
445 }
446
447 info_ptr = new (png_info);
448 if (!info_ptr)
449 {
450 fclose(fp);
451 delete png_ptr;
452 return FALSE;
453 }
454 /* set error handling */
455 if (setjmp(png_ptr->jmpbuf))
456 {
457 png_read_destroy(png_ptr, info_ptr, (png_info *)0);
458 fclose(fp);
459 delete png_ptr;
460 delete info_ptr;
461
462 /* If we get here, we had a problem reading the file */
463 return FALSE;
464 }
465 //png_set_error(ima_png_error, NULL);
466
467 /* initialize the structures, info first for error handling */
468 png_info_init(info_ptr);
469 png_read_init(png_ptr);
470
471 /* set up the input control */
472 png_init_io(png_ptr, fp);
473
474 /* read the file information */
475 png_read_info(png_ptr, info_ptr);
476
477 /* allocate the memory to hold the image using the fields
478 of png_info. */
479 png_color_16 my_background={ 0, 31, 127, 255, 0 };
480
481 if (info_ptr->valid & PNG_INFO_bKGD)
482 {
483 png_set_background(png_ptr, &(info_ptr->background),
484 PNG_BACKGROUND_GAMMA_FILE, 1, 1.0);
485 if ( info_ptr->num_palette > 0 )
486 bgindex = info_ptr->background.index;
487 }
488 else {
489 png_set_background(png_ptr, &my_background,
490 PNG_BACKGROUND_GAMMA_SCREEN, 0, 1.0);
491
492 // Added by JACS: guesswork!
493 if ( info_ptr->num_trans != 0 )
494 bgindex = info_ptr->num_trans - 1 ;
495 }
496
497 /* tell libpng to strip 16 bit depth files down to 8 bits */
498 if (info_ptr->bit_depth == 16)
499 png_set_strip_16(png_ptr);
500
501 int pixel_depth=(info_ptr->pixel_depth<24) ? info_ptr->pixel_depth: 24;
502 Create(info_ptr->width, info_ptr->height, pixel_depth,
503 info_ptr->color_type);
504
505 if (info_ptr->num_palette>0)
506 {
507 SetPalette((int)info_ptr->num_palette, (rgb_color_struct*)info_ptr->palette);
508 }
509
510 int row_stride = info_ptr->width * ((pixel_depth+7)>>3);
511 // printf("P = %d D = %d RS= %d ", info_ptr->num_palette, info_ptr->pixel_depth,row_stride);
512 // printf("CT = %d TRS = %d BD= %d ", info_ptr->color_type, info_ptr->valid & PNG_INFO_tRNS,info_ptr->bit_depth);
513
514 byte *row_pointers = new byte[row_stride];
515
516 /* turn on interlace handling */
517 if (info_ptr->interlace_type)
518 number_passes = png_set_interlace_handling(png_ptr);
519 else
520 number_passes = 1;
521 // printf("NP = %d ", number_passes);
522
523 for (int pass=0; pass< number_passes; pass++)
524 {
525 iter.upset();
526 int y=0;
527 CGrafPtr origPort ;
528 GDHandle origDevice ;
529
530 GetGWorld( &origPort , &origDevice ) ;
531 // ignore shapedc
532 SetGWorld( (GWorldPtr) lpbi , NULL ) ;
533 do
534 {
535 // (unsigned char *)iter.GetRow();
536 if (info_ptr->interlace_type)
537 {
538 if (pass>0)
539 iter.GetRow(row_pointers, row_stride);
540 png_read_row(png_ptr, row_pointers, NULL);
541 }
542 else
543 png_read_row(png_ptr, row_pointers, NULL);
544
545 if ( info_ptr->palette )
546 {
547 if ( pixel_depth == 8 )
548 {
549 for ( size_t i = 0 ; i < info_ptr->width ; ++i )
550 {
551 png_color_struct* color ;
552 RGBColor col ;
553
554 int index = row_pointers[i] ;
555 color = &info_ptr->palette[index] ;
556 col.red = (((int)color->red) << 8) | ((int)color->red) ;
557 col.green = (((int)color->green) << 8) | ((int)color->green) ;
558 col.blue = (((int)color->blue) << 8) | ((int)color->blue) ;
559 SetCPixel( i, y, &col);
560 }
561 /*
562 png_color_struct* color ;
563 RGBColor col ;
564 unsigned char* p = &row_pointers[0] ;
565 PenNormal() ;
566 MoveTo( 0 , y ) ;
567 int index = *p ;
568 color = &info_ptr->palette[index] ;
569 col.red = (color->red << 8) | color->red ;
570 col.green = (color->green << 8) | color->green ;
571 col.blue = (color->blue << 8) | color->blue ;
572 RGBForeColor( &col ) ;
573 col.red = col.green = col.blue = 0xFFFF ;
574 RGBBackColor( &col ) ;
575 for ( int i = 0 ; i < info_ptr->width ; ++i , ++p)
576 {
577 if ( *p != index )
578 {
579 LineTo( i , y ) ;
580 index = *p ;
581 color = &info_ptr->palette[index] ;
582 col.red = (((int)color->red) << 8) | ((int)color->red) ;
583 col.green = (((int)color->green) << 8) | ((int)color->green) ;
584 col.blue = (((int)color->blue) << 8) | ((int)color->blue) ;
585 RGBForeColor( &col ) ;
586 }
587 }
588 LineTo( info_ptr->width , y ) ;
589 */
590 }
591 else
592 {
593 for ( size_t i = 0 ; i < info_ptr->width ; ++i )
594 {
595 png_color_struct* color ;
596 RGBColor col ;
597
598 int byte = ( i * pixel_depth ) / 8 ;
599 int offset = ( 8 - pixel_depth ) - ( i * pixel_depth ) % 8 ;
600
601 int index = ( row_pointers[byte] >> offset ) & ( 0xFF >> ( 8 - pixel_depth ) );
602 color = &info_ptr->palette[index] ;
603 col.red = (((int)color->red) << 8) | ((int)color->red) ;
604 col.green = (((int)color->green) << 8) | ((int)color->green) ;
605 col.blue = (((int)color->blue) << 8) | ((int)color->blue) ;
606 SetCPixel( i, y, &col);
607 }
608 }
609 }
610 else
611 {
612 for ( size_t i = 0 ; i < info_ptr->width ; ++i )
613 {
614 png_color_struct* color ;
615 RGBColor col ;
616 color =(png_color_struct*) (&row_pointers[i*3]) ;
617 col.red = (((int)color->red) << 8) | ((int)color->red) ;
618 col.green = (((int)color->green) << 8) | ((int)color->green) ;
619 col.blue = (((int)color->blue) << 8) | ((int)color->blue) ;
620 SetCPixel( i, y, &col);
621 }
622 }
623 if (number_passes)
624 iter.SetRow(row_pointers, row_stride);
625 y++;
626 }
627 while(iter.PrevRow());
628 SetGWorld( origPort , origDevice ) ;
629
630 // printf("Y=%d ",y);
631 }
632 delete[] row_pointers;
633
634 /* read the rest of the file, getting any additional chunks
635 in info_ptr */
636 png_read_end(png_ptr, info_ptr);
637
638 /* clean up after the read, and free any memory allocated */
639 png_read_destroy(png_ptr, info_ptr, (png_info *)0);
640
641 /* free the structures */
642 delete png_ptr;
643 delete info_ptr;
644
645 /* close the file */
646 fclose(fp);
647
648 /* that's it */
649 return TRUE;
650 }
651
652
653 /* write a png file */
654
655 bool wxPNGReader::SaveFile(char * ImageFileName)
656 {
657 if (ImageFileName)
658 strcpy(filename, ImageFileName);
659
660 wxPNGReaderIter iter(this);
661 FILE *fp;
662 png_struct *png_ptr;
663 png_info *info_ptr;
664
665 /* open the file */
666 fp = fopen(filename, "wb");
667 if (!fp)
668 return FALSE;
669
670 /* allocate the necessary structures */
671 png_ptr = new (png_struct);
672 if (!png_ptr)
673 {
674 fclose(fp);
675 return FALSE;
676 }
677
678 info_ptr = new (png_info);
679 if (!info_ptr)
680 {
681 fclose(fp);
682 delete png_ptr;
683 return FALSE;
684 }
685
686 /* set error handling */
687 if (setjmp(png_ptr->jmpbuf))
688 {
689 png_write_destroy(png_ptr);
690 fclose(fp);
691 delete png_ptr;
692 delete info_ptr;
693
694 /* If we get here, we had a problem reading the file */
695 return FALSE;
696 }
697 //png_set_error(ima_png_error, NULL);
698
699 // printf("writig pg %s ", filename);
700 /* initialize the structures */
701 png_info_init(info_ptr);
702 png_write_init(png_ptr);
703
704 int row_stride = GetWidth() * ((GetDepth()+7)>>3);
705 /* set up the output control */
706 png_init_io(png_ptr, fp);
707
708 /* set the file information here */
709 info_ptr->width = GetWidth();
710 info_ptr->height = GetHeight();
711 info_ptr->pixel_depth = GetDepth();
712 info_ptr->channels = (GetDepth()>8) ? 3: 1;
713 info_ptr->bit_depth = GetDepth()/info_ptr->channels;
714 info_ptr->color_type = GetColorType();
715 info_ptr->compression_type = info_ptr->filter_type = info_ptr->interlace_type=0;
716 info_ptr->valid = 0;
717 info_ptr->rowbytes = row_stride;
718
719
720 // 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);
721 /* set the palette if there is one */
722 if ((GetColorType() & COLORTYPE_PALETTE) && GetPalette())
723 {
724 // printf("writing paleta[%d %d %x]",GetColorType() ,COLORTYPE_PALETTE, GetPalette());
725 info_ptr->valid |= PNG_INFO_PLTE;
726 info_ptr->palette = new png_color[256];
727 info_ptr->num_palette = 256;
728 for (int i=0; i<256; i++)
729 GetPalette()->GetRGB(i, &info_ptr->palette[i].red, &info_ptr->palette[i].green, &info_ptr->palette[i].blue);
730 }
731 // printf("Paleta [%d %d %x]",GetColorType() ,COLORTYPE_PALETTE, GetPalette());
732
733
734 /* optional significant bit chunk */
735 // info_ptr->valid |= PNG_INFO_sBIT;
736 // info_ptr->sig_bit = true_bit_depth;
737
738 /* optional gamma chunk */
739 // info_ptr->valid |= PNG_INFO_gAMA;
740 // info_ptr->gamma = gamma;
741
742 /* other optional chunks */
743
744 /* write the file information */
745 png_write_info(png_ptr, info_ptr);
746
747 /* set up the transformations you want. Note that these are
748 all optional. Only call them if you want them */
749
750 /* shift the pixels up to a legal bit depth and fill in
751 as appropriate to correctly scale the image */
752 // png_set_shift(png_ptr, &(info_ptr->sig_bit));
753
754 /* pack pixels into bytes */
755 // png_set_packing(png_ptr);
756
757 /* flip bgr pixels to rgb */
758 // png_set_bgr(png_ptr);
759
760 /* swap bytes of 16 bit files to most significant bit first */
761 // png_set_swap(png_ptr);
762
763 /* get rid of filler bytes, pack rgb into 3 bytes */
764 // png_set_rgbx(png_ptr);
765
766 /* If you are only writing one row at a time, this works */
767
768 byte *row_pointers = new byte[row_stride];
769 iter.upset();
770 do {
771 // (unsigned char *)iter.GetRow();
772 iter.GetRow(row_pointers, row_stride);
773 png_write_row(png_ptr, row_pointers);
774 } while(iter.PrevRow());
775
776 delete[] row_pointers;
777
778 /* write the rest of the file */
779 png_write_end(png_ptr, info_ptr);
780
781 /* clean up after the write, and free any memory allocated */
782 png_write_destroy(png_ptr);
783
784 /* if you malloced the palette, free it here */
785 if (info_ptr->palette)
786 delete[] (info_ptr->palette);
787
788 /* free the structures */
789 delete png_ptr;
790 delete info_ptr;
791
792 /* close the file */
793 fclose(fp);
794
795 /* that's it */
796 return TRUE;
797 }
798
799 static int Power(int x, int y)
800 {
801 int z = 1;
802 int i;
803 for ( i = 0; i < y; i++)
804 {
805 z *= x;
806 }
807 return z;
808 }
809
810 static char hexArray[] = { '0', '1', '2', '3', '4', '5', '6', '7', '8', '9', 'A', 'B',
811 'C', 'D', 'E', 'F' };
812
813 static void DecToHex(int dec, char *buf)
814 {
815 int firstDigit = (int)(dec/16.0);
816 int secondDigit = (int)(dec - (firstDigit*16.0));
817 buf[0] = hexArray[firstDigit];
818 buf[1] = hexArray[secondDigit];
819 buf[2] = 0;
820 }
821
822
823 bool wxPNGReader::SaveXPM(char *filename, char *name)
824 {
825 char nameStr[256];
826 if ( name )
827 strcpy(nameStr, name);
828 else
829 {
830 wxString str = wxString::FromAscii(filename) ;
831 wxStripExtension( str ) ;
832 strcpy(nameStr, str.ToAscii() );
833 }
834
835 if ( GetDepth() > 4 )
836 {
837 // Only a depth of 4 and below allowed
838 return FALSE;
839 }
840
841 if ( !GetPalette() )
842 return FALSE;
843
844 wxSTD ofstream str(filename);
845 if ( str.bad() )
846 return FALSE;
847
848 int noColours = Power(2, GetDepth());
849
850 // Output header
851 str << "/* XPM */\n";
852 str << "static char * " << nameStr << "_xpm[] = {\n";
853 str << "\"" << GetWidth() << " " << GetHeight() << " " << noColours << " 1\",\n";
854
855 // Output colourmap
856 int base = 97 ; // start from 'a'
857
858 unsigned char red, green, blue;
859 char hexBuf[4];
860 int i;
861 for ( i = 0; i < noColours; i ++)
862 {
863 str << "\"" << (char)(base + i) << " c #";
864 GetPalette()->GetRGB(i, &red, &green, &blue);
865 DecToHex(red, hexBuf);
866 str << hexBuf;
867 DecToHex(green, hexBuf);
868 str << hexBuf;
869 DecToHex(blue, hexBuf);
870 str << hexBuf;
871 str << "\",\n";
872 }
873
874 // Output the data
875 int x, y;
876 for ( y = 0; y < GetHeight(); y++)
877 {
878 str << "\"";
879 for ( x = 0; x < GetWidth(); x++)
880 {
881 int index = GetIndex(x, y);
882 str << (char)(base + index) ;
883 }
884 str << "\",\n";
885 }
886
887 str << "};\n";
888 str.flush();
889
890 return TRUE;
891 }
892
893
894 IMPLEMENT_DYNAMIC_CLASS(wxPNGFileHandler, wxBitmapHandler)
895
896 bool wxPNGFileHandler::LoadFile(wxBitmap *bitmap, const wxString& name, long flags,
897 int desiredWidth, int desiredHeight)
898 {
899 wxPNGReader reader;
900 if (reader.ReadFile( (char*)(const char*) name.ToAscii() ) )
901 {
902 return reader.InstantiateBitmap(bitmap);
903 }
904 else
905 return FALSE;
906 }
907
908 bool wxPNGFileHandler::SaveFile(const wxBitmap *bitmap, const wxString& name, int type, const wxPalette *pal)
909 {
910 return FALSE;
911 }
912
913 #endif //wxUSE_LIBPNG