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