]>
Commit | Line | Data |
---|---|---|
1 | ///////////////////////////////////////////////////////////////////////////// | |
2 | // Name: image.cpp | |
3 | // Purpose: wxImage | |
4 | // Author: Robert Roebling | |
5 | // RCS-ID: $Id$ | |
6 | // Copyright: (c) Robert Roebling | |
7 | // Licence: wxWindows licence | |
8 | ///////////////////////////////////////////////////////////////////////////// | |
9 | ||
10 | #ifdef __GNUG__ | |
11 | #pragma implementation "image.h" | |
12 | #endif | |
13 | ||
14 | // For compilers that support precompilation, includes "wx.h". | |
15 | #include "wx/wxprec.h" | |
16 | ||
17 | #ifdef __BORLANDC__ | |
18 | #pragma hdrstop | |
19 | #endif | |
20 | ||
21 | #include "wx/image.h" | |
22 | #include "wx/debug.h" | |
23 | #include "wx/log.h" | |
24 | #include "../png/png.h" | |
25 | #include "wx/filefn.h" | |
26 | ||
27 | //----------------------------------------------------------------------------- | |
28 | // wxImage | |
29 | //----------------------------------------------------------------------------- | |
30 | ||
31 | class wxImageRefData: public wxObjectRefData | |
32 | { | |
33 | ||
34 | public: | |
35 | wxImageRefData(void); | |
36 | ~wxImageRefData(void); | |
37 | ||
38 | int m_width; | |
39 | int m_height; | |
40 | unsigned char *m_data; | |
41 | bool m_hasMask; | |
42 | unsigned char m_maskRed,m_maskGreen,m_maskBlue; | |
43 | bool m_ok; | |
44 | }; | |
45 | ||
46 | wxImageRefData::wxImageRefData(void) | |
47 | { | |
48 | m_width = 0; | |
49 | m_height = 0; | |
50 | m_data = (unsigned char*) NULL; | |
51 | m_ok = FALSE; | |
52 | m_maskRed = 0; | |
53 | m_maskGreen = 0; | |
54 | m_maskBlue = 0; | |
55 | m_hasMask = FALSE; | |
56 | } | |
57 | ||
58 | wxImageRefData::~wxImageRefData(void) | |
59 | { | |
60 | if (m_data) free( m_data ); | |
61 | } | |
62 | ||
63 | wxList wxImage::sm_handlers; | |
64 | ||
65 | //----------------------------------------------------------------------------- | |
66 | ||
67 | #define M_IMGDATA ((wxImageRefData *)m_refData) | |
68 | ||
69 | #if !USE_SHARED_LIBRARIES | |
70 | IMPLEMENT_DYNAMIC_CLASS(wxImage, wxObject) | |
71 | #endif | |
72 | ||
73 | wxImage::wxImage() | |
74 | { | |
75 | } | |
76 | ||
77 | wxImage::wxImage( int width, int height ) | |
78 | { | |
79 | Create( width, height ); | |
80 | } | |
81 | ||
82 | wxImage::wxImage( const wxString& name, long type ) | |
83 | { | |
84 | LoadFile( name, type ); | |
85 | } | |
86 | ||
87 | wxImage::wxImage( const wxImage& image ) | |
88 | { | |
89 | Ref(image); | |
90 | } | |
91 | ||
92 | wxImage::wxImage( const wxImage* image ) | |
93 | { | |
94 | if (image) Ref(*image); | |
95 | } | |
96 | ||
97 | void wxImage::Create( int width, int height ) | |
98 | { | |
99 | m_refData = new wxImageRefData(); | |
100 | ||
101 | M_IMGDATA->m_data = (unsigned char *) malloc( width*height*3 ); | |
102 | if (M_IMGDATA->m_data) | |
103 | { | |
104 | for (int l = 0; l < width*height*3; l++) M_IMGDATA->m_data[l] = 0; | |
105 | ||
106 | M_IMGDATA->m_width = width; | |
107 | M_IMGDATA->m_height = height; | |
108 | M_IMGDATA->m_ok = TRUE; | |
109 | } | |
110 | else | |
111 | { | |
112 | UnRef(); | |
113 | } | |
114 | } | |
115 | ||
116 | void wxImage::Destroy() | |
117 | { | |
118 | UnRef(); | |
119 | } | |
120 | ||
121 | wxImage wxImage::Scale( int width, int height ) | |
122 | { | |
123 | wxImage image; | |
124 | ||
125 | wxCHECK_MSG( Ok(), image, "invlaid image" ); | |
126 | ||
127 | wxCHECK_MSG( (width > 0) && (height > 0), image, "invalid image size" ); | |
128 | ||
129 | image.Create( width, height ); | |
130 | ||
131 | char unsigned *data = image.GetData(); | |
132 | ||
133 | wxCHECK_MSG( data, image, "unable to create image" ); | |
134 | ||
135 | if (M_IMGDATA->m_hasMask) | |
136 | image.SetMaskColour( M_IMGDATA->m_maskRed, M_IMGDATA->m_maskGreen, M_IMGDATA->m_maskBlue ); | |
137 | ||
138 | double xscale = (double)width / (double)M_IMGDATA->m_width; | |
139 | double yscale = (double)height / (double)M_IMGDATA->m_height; | |
140 | ||
141 | for (int j = 0; j < height; j++) | |
142 | { | |
143 | for (int i = 0; i < width; i++) | |
144 | { | |
145 | int new_pos = 3*(j*width + i); | |
146 | int old_pos = 3*((long)(j/yscale)*M_IMGDATA->m_width + (long)(i/xscale)); | |
147 | data[ new_pos ] = M_IMGDATA->m_data[ old_pos ]; | |
148 | data[ new_pos+1 ] = M_IMGDATA->m_data[ old_pos+1 ]; | |
149 | data[ new_pos+2 ] = M_IMGDATA->m_data[ old_pos+2 ]; | |
150 | } | |
151 | } | |
152 | ||
153 | return image; | |
154 | } | |
155 | ||
156 | void wxImage::SetRGB( int x, int y, unsigned char r, unsigned char g, unsigned char b ) | |
157 | { | |
158 | wxCHECK_RET( Ok(), "invalid image" ); | |
159 | ||
160 | int w = M_IMGDATA->m_width; | |
161 | int h = M_IMGDATA->m_height; | |
162 | ||
163 | wxCHECK_RET( (x>=0) && (y>=0) && (x<w) && (y<h), "invalid image index" ); | |
164 | ||
165 | long pos = (y * w + x) * 3; | |
166 | ||
167 | M_IMGDATA->m_data[ pos ] = r; | |
168 | M_IMGDATA->m_data[ pos+1 ] = g; | |
169 | M_IMGDATA->m_data[ pos+2 ] = b; | |
170 | } | |
171 | ||
172 | unsigned char wxImage::GetRed( int x, int y ) | |
173 | { | |
174 | wxCHECK_MSG( Ok(), 0, "invalid image" ); | |
175 | ||
176 | int w = M_IMGDATA->m_width; | |
177 | int h = M_IMGDATA->m_height; | |
178 | ||
179 | wxCHECK_MSG( (x>=0) && (y>=0) && (x<w) && (y<h), 0, "invalid image index" ); | |
180 | ||
181 | long pos = (y * w + x) * 3; | |
182 | ||
183 | return M_IMGDATA->m_data[pos]; | |
184 | } | |
185 | ||
186 | unsigned char wxImage::GetGreen( int x, int y ) | |
187 | { | |
188 | wxCHECK_MSG( Ok(), 0, "invalid image" ); | |
189 | ||
190 | int w = M_IMGDATA->m_width; | |
191 | int h = M_IMGDATA->m_height; | |
192 | ||
193 | wxCHECK_MSG( (x>=0) && (y>=0) && (x<w) && (y<h), 0, "invalid image index" ); | |
194 | ||
195 | long pos = (y * w + x) * 3; | |
196 | ||
197 | return M_IMGDATA->m_data[pos+1]; | |
198 | } | |
199 | ||
200 | unsigned char wxImage::GetBlue( int x, int y ) | |
201 | { | |
202 | wxCHECK_MSG( Ok(), 0, "invalid image" ); | |
203 | ||
204 | int w = M_IMGDATA->m_width; | |
205 | int h = M_IMGDATA->m_height; | |
206 | ||
207 | wxCHECK_MSG( (x>=0) && (y>=0) && (x<w) && (y<h), 0, "invalid image index" ); | |
208 | ||
209 | long pos = (y * w + x) * 3; | |
210 | ||
211 | return M_IMGDATA->m_data[pos+2]; | |
212 | } | |
213 | ||
214 | bool wxImage::Ok() const | |
215 | { | |
216 | return (M_IMGDATA && M_IMGDATA->m_ok); | |
217 | } | |
218 | ||
219 | char unsigned *wxImage::GetData() const | |
220 | { | |
221 | wxCHECK_MSG( Ok(), (char unsigned *)NULL, "invalid image" ); | |
222 | ||
223 | return M_IMGDATA->m_data; | |
224 | } | |
225 | ||
226 | void wxImage::SetData( char unsigned *WXUNUSED(data) ) | |
227 | { | |
228 | wxCHECK_RET( Ok(), "invalid image" ); | |
229 | } | |
230 | ||
231 | void wxImage::SetMaskColour( unsigned char r, unsigned char g, unsigned char b ) | |
232 | { | |
233 | wxCHECK_RET( Ok(), "invalid image" ); | |
234 | ||
235 | M_IMGDATA->m_maskRed = r; | |
236 | M_IMGDATA->m_maskGreen = g; | |
237 | M_IMGDATA->m_maskBlue = b; | |
238 | M_IMGDATA->m_hasMask = TRUE; | |
239 | } | |
240 | ||
241 | unsigned char wxImage::GetMaskRed() const | |
242 | { | |
243 | wxCHECK_MSG( Ok(), 0, "invalid image" ); | |
244 | ||
245 | return M_IMGDATA->m_maskRed; | |
246 | } | |
247 | ||
248 | unsigned char wxImage::GetMaskGreen() const | |
249 | { | |
250 | wxCHECK_MSG( Ok(), 0, "invalid image" ); | |
251 | ||
252 | return M_IMGDATA->m_maskGreen; | |
253 | } | |
254 | ||
255 | unsigned char wxImage::GetMaskBlue() const | |
256 | { | |
257 | wxCHECK_MSG( Ok(), 0, "invalid image" ); | |
258 | ||
259 | return M_IMGDATA->m_maskBlue; | |
260 | } | |
261 | ||
262 | void wxImage::SetMask( bool mask ) | |
263 | { | |
264 | wxCHECK_RET( Ok(), "invalid image" ); | |
265 | ||
266 | M_IMGDATA->m_hasMask = mask; | |
267 | } | |
268 | ||
269 | bool wxImage::HasMask() const | |
270 | { | |
271 | wxCHECK_MSG( Ok(), FALSE, "invalid image" ); | |
272 | ||
273 | return M_IMGDATA->m_hasMask; | |
274 | } | |
275 | ||
276 | int wxImage::GetWidth() const | |
277 | { | |
278 | wxCHECK_MSG( Ok(), 0, "invalid image" ); | |
279 | ||
280 | return M_IMGDATA->m_width; | |
281 | } | |
282 | ||
283 | int wxImage::GetHeight() const | |
284 | { | |
285 | wxCHECK_MSG( Ok(), 0, "invalid image" ); | |
286 | ||
287 | return M_IMGDATA->m_height; | |
288 | } | |
289 | ||
290 | bool wxImage::LoadFile( const wxString& filename, long type ) | |
291 | { | |
292 | UnRef(); | |
293 | ||
294 | if (!wxFileExists(filename)) | |
295 | { | |
296 | wxLogWarning( "Image file does not exist." ); | |
297 | ||
298 | return FALSE; | |
299 | } | |
300 | ||
301 | m_refData = new wxImageRefData; | |
302 | ||
303 | wxImageHandler *handler = FindHandler(type); | |
304 | ||
305 | if (handler == NULL) | |
306 | { | |
307 | wxLogWarning( "No image handler for type %d defined.", type ); | |
308 | ||
309 | return FALSE; | |
310 | } | |
311 | ||
312 | return handler->LoadFile( this, filename ); | |
313 | } | |
314 | ||
315 | bool wxImage::SaveFile( const wxString& filename, int type ) | |
316 | { | |
317 | wxCHECK_MSG( Ok(), FALSE, "invalid image" ); | |
318 | ||
319 | wxImageHandler *handler = FindHandler(type); | |
320 | ||
321 | if (handler == NULL) | |
322 | { | |
323 | wxLogWarning( "No image handler for type %d defined.", type ); | |
324 | ||
325 | return FALSE; | |
326 | } | |
327 | ||
328 | return handler->SaveFile( this, filename ); | |
329 | } | |
330 | ||
331 | void wxImage::AddHandler( wxImageHandler *handler ) | |
332 | { | |
333 | sm_handlers.Append( handler ); | |
334 | } | |
335 | ||
336 | void wxImage::InsertHandler( wxImageHandler *handler ) | |
337 | { | |
338 | sm_handlers.Insert( handler ); | |
339 | } | |
340 | ||
341 | bool wxImage::RemoveHandler( const wxString& name ) | |
342 | { | |
343 | wxImageHandler *handler = FindHandler(name); | |
344 | if (handler) | |
345 | { | |
346 | sm_handlers.DeleteObject(handler); | |
347 | return TRUE; | |
348 | } | |
349 | else | |
350 | return FALSE; | |
351 | } | |
352 | ||
353 | wxImageHandler *wxImage::FindHandler( const wxString& name ) | |
354 | { | |
355 | wxNode *node = sm_handlers.First(); | |
356 | while (node) | |
357 | { | |
358 | wxImageHandler *handler = (wxImageHandler*)node->Data(); | |
359 | if (handler->GetName() == name) return handler; | |
360 | node = node->Next(); | |
361 | } | |
362 | return (wxImageHandler *)NULL; | |
363 | } | |
364 | ||
365 | wxImageHandler *wxImage::FindHandler( const wxString& extension, long bitmapType ) | |
366 | { | |
367 | wxNode *node = sm_handlers.First(); | |
368 | while (node) | |
369 | { | |
370 | wxImageHandler *handler = (wxImageHandler*)node->Data(); | |
371 | if ( handler->GetExtension() == extension && | |
372 | (bitmapType == -1 || handler->GetType() == bitmapType) ) | |
373 | return handler; | |
374 | node = node->Next(); | |
375 | } | |
376 | return (wxImageHandler*)NULL; | |
377 | } | |
378 | ||
379 | wxImageHandler *wxImage::FindHandler( long bitmapType ) | |
380 | { | |
381 | wxNode *node = sm_handlers.First(); | |
382 | while (node) | |
383 | { | |
384 | wxImageHandler *handler = (wxImageHandler *)node->Data(); | |
385 | if (handler->GetType() == bitmapType) return handler; | |
386 | node = node->Next(); | |
387 | } | |
388 | return NULL; | |
389 | } | |
390 | ||
391 | void wxImage::InitStandardHandlers() | |
392 | { | |
393 | AddHandler( new wxBMPHandler ); | |
394 | AddHandler( new wxPNGHandler ); | |
395 | } | |
396 | ||
397 | void wxImage::CleanUpHandlers() | |
398 | { | |
399 | wxNode *node = sm_handlers.First(); | |
400 | while (node) | |
401 | { | |
402 | wxImageHandler *handler = (wxImageHandler *)node->Data(); | |
403 | wxNode *next = node->Next(); | |
404 | delete handler; | |
405 | delete node; | |
406 | node = next; | |
407 | } | |
408 | } | |
409 | ||
410 | //----------------------------------------------------------------------------- | |
411 | // wxImageHandler | |
412 | //----------------------------------------------------------------------------- | |
413 | ||
414 | #if !USE_SHARED_LIBRARIES | |
415 | IMPLEMENT_DYNAMIC_CLASS(wxImageHandler,wxObject) | |
416 | #endif | |
417 | ||
418 | bool wxImageHandler::LoadFile( wxImage *WXUNUSED(image), const wxString& WXUNUSED(name) ) | |
419 | { | |
420 | return FALSE; | |
421 | } | |
422 | ||
423 | bool wxImageHandler::SaveFile( wxImage *WXUNUSED(image), const wxString& WXUNUSED(name) ) | |
424 | { | |
425 | return FALSE; | |
426 | } | |
427 | ||
428 | //----------------------------------------------------------------------------- | |
429 | // wxPNGHandler | |
430 | //----------------------------------------------------------------------------- | |
431 | ||
432 | #if !USE_SHARED_LIBRARIES | |
433 | IMPLEMENT_DYNAMIC_CLASS(wxPNGHandler,wxImageHandler) | |
434 | #endif | |
435 | ||
436 | bool wxPNGHandler::LoadFile( wxImage *image, const wxString& name ) | |
437 | { | |
438 | FILE *f; | |
439 | png_structp png_ptr; | |
440 | png_infop info_ptr; | |
441 | unsigned char *ptr, **lines, *ptr2; | |
442 | int transp,bit_depth,color_type,interlace_type; | |
443 | png_uint_32 width, height; | |
444 | unsigned int i; | |
445 | ||
446 | image->Destroy(); | |
447 | ||
448 | transp = 0; | |
449 | png_ptr = png_create_read_struct( PNG_LIBPNG_VER_STRING, NULL, NULL, NULL ); | |
450 | if (!png_ptr) return FALSE; | |
451 | ||
452 | info_ptr = png_create_info_struct( png_ptr ); | |
453 | if (!info_ptr) | |
454 | { | |
455 | png_destroy_read_struct( &png_ptr, NULL, NULL ); | |
456 | return FALSE; | |
457 | } | |
458 | ||
459 | if (setjmp(png_ptr->jmpbuf)) | |
460 | { | |
461 | png_destroy_read_struct( &png_ptr, &info_ptr, NULL ); | |
462 | return FALSE; | |
463 | } | |
464 | ||
465 | if (info_ptr->color_type == PNG_COLOR_TYPE_RGB_ALPHA) | |
466 | { | |
467 | png_destroy_read_struct( &png_ptr, &info_ptr, NULL ); | |
468 | return FALSE; | |
469 | } | |
470 | ||
471 | f = fopen( name, "rb" ); | |
472 | png_init_io( png_ptr, f ); | |
473 | ||
474 | png_read_info( png_ptr, info_ptr ); | |
475 | png_get_IHDR( png_ptr, info_ptr, &width, &height, &bit_depth, &color_type, &interlace_type, NULL, NULL ); | |
476 | ||
477 | if (color_type == PNG_COLOR_TYPE_PALETTE) png_set_expand( png_ptr ); | |
478 | ||
479 | png_set_strip_16( png_ptr ); | |
480 | png_set_packing( png_ptr ); | |
481 | if (png_get_valid( png_ptr, info_ptr, PNG_INFO_tRNS)) png_set_expand( png_ptr ); | |
482 | png_set_filler( png_ptr, 0xff, PNG_FILLER_AFTER ); | |
483 | ||
484 | image->Create( width, height ); | |
485 | ||
486 | if (!image->Ok()) | |
487 | { | |
488 | png_destroy_read_struct( &png_ptr, &info_ptr, NULL ); | |
489 | return FALSE; | |
490 | } | |
491 | ||
492 | lines = (unsigned char **)malloc( height * sizeof(unsigned char *) ); | |
493 | if (lines == NULL) | |
494 | { | |
495 | image->Destroy(); | |
496 | png_destroy_read_struct( &png_ptr, &info_ptr, NULL ); | |
497 | return FALSE; | |
498 | } | |
499 | ||
500 | for (i = 0; i < height; i++) | |
501 | { | |
502 | if ((lines[i] = (unsigned char *)malloc(width * (sizeof(unsigned char) * 4))) == NULL) | |
503 | { | |
504 | image->Destroy(); | |
505 | for (unsigned int n = 0; n < i; n++) free( lines[n] ); | |
506 | free( lines ); | |
507 | png_destroy_read_struct( &png_ptr, &info_ptr, NULL ); | |
508 | return FALSE; | |
509 | } | |
510 | } | |
511 | ||
512 | png_read_image( png_ptr, lines ); | |
513 | png_destroy_read_struct( &png_ptr, &info_ptr, NULL ); | |
514 | ptr = image->GetData(); | |
515 | if ((color_type == PNG_COLOR_TYPE_GRAY) || | |
516 | (color_type == PNG_COLOR_TYPE_GRAY_ALPHA)) | |
517 | { | |
518 | for (unsigned int y = 0; y < height; y++) | |
519 | { | |
520 | ptr2 = lines[y]; | |
521 | for (unsigned int x = 0; x < width; x++) | |
522 | { | |
523 | unsigned char r = *ptr2++; | |
524 | unsigned char a = *ptr2++; | |
525 | if (a < 128) | |
526 | { | |
527 | *ptr++ = 255; | |
528 | *ptr++ = 0; | |
529 | *ptr++ = 255; | |
530 | transp = 1; | |
531 | } | |
532 | else | |
533 | { | |
534 | *ptr++ = r; | |
535 | *ptr++ = r; | |
536 | *ptr++ = r; | |
537 | } | |
538 | } | |
539 | } | |
540 | } | |
541 | else | |
542 | { | |
543 | for (unsigned int y = 0; y < height; y++) | |
544 | { | |
545 | ptr2 = lines[y]; | |
546 | for (unsigned int x = 0; x < width; x++) | |
547 | { | |
548 | unsigned char r = *ptr2++; | |
549 | unsigned char g = *ptr2++; | |
550 | unsigned char b = *ptr2++; | |
551 | unsigned char a = *ptr2++; | |
552 | if (a < 128) | |
553 | { | |
554 | *ptr++ = 255; | |
555 | *ptr++ = 0; | |
556 | *ptr++ = 255; | |
557 | transp = 1; | |
558 | } | |
559 | else | |
560 | { | |
561 | if ((r == 255) && (g == 0) && (b == 255)) r = 254; | |
562 | *ptr++ = r; | |
563 | *ptr++ = g; | |
564 | *ptr++ = b; | |
565 | } | |
566 | } | |
567 | } | |
568 | } | |
569 | for (i = 0; i < height; i++) free( lines[i] ); | |
570 | free( lines ); | |
571 | if (transp) | |
572 | image->SetMaskColour( 255, 0, 255 ); | |
573 | else | |
574 | image->SetMask( FALSE ); | |
575 | ||
576 | return TRUE; | |
577 | } | |
578 | ||
579 | ||
580 | bool wxPNGHandler::SaveFile( wxImage *image, const wxString& name ) | |
581 | { | |
582 | FILE *f = fopen( name, "wb" ); | |
583 | if (f) | |
584 | { | |
585 | png_structp png_ptr = png_create_write_struct( PNG_LIBPNG_VER_STRING, NULL, NULL, NULL); | |
586 | if (!png_ptr) | |
587 | { | |
588 | fclose( f ); | |
589 | return FALSE; | |
590 | } | |
591 | ||
592 | png_infop info_ptr = png_create_info_struct(png_ptr); | |
593 | if (info_ptr == NULL) | |
594 | { | |
595 | fclose(f); | |
596 | png_destroy_write_struct( &png_ptr, (png_infopp)NULL ); | |
597 | return FALSE; | |
598 | } | |
599 | ||
600 | if (setjmp(png_ptr->jmpbuf)) | |
601 | { | |
602 | fclose( f ); | |
603 | png_destroy_write_struct( &png_ptr, (png_infopp)NULL ); | |
604 | return FALSE; | |
605 | } | |
606 | ||
607 | png_init_io( png_ptr, f ); | |
608 | png_set_IHDR( png_ptr, info_ptr, image->GetWidth(), image->GetHeight(), 8, | |
609 | PNG_COLOR_TYPE_RGB_ALPHA, PNG_INTERLACE_NONE, | |
610 | PNG_COMPRESSION_TYPE_BASE, PNG_FILTER_TYPE_BASE); | |
611 | ||
612 | png_color_8 sig_bit; | |
613 | sig_bit.red = 8; | |
614 | sig_bit.green = 8; | |
615 | sig_bit.blue = 8; | |
616 | sig_bit.alpha = 8; | |
617 | png_set_sBIT( png_ptr, info_ptr, &sig_bit ); | |
618 | png_write_info( png_ptr, info_ptr ); | |
619 | png_set_shift( png_ptr, &sig_bit ); | |
620 | png_set_packing( png_ptr ); | |
621 | ||
622 | unsigned char *data = (unsigned char *)malloc( image->GetWidth()*4 ); | |
623 | if (!data) | |
624 | { | |
625 | fclose( f ); | |
626 | png_destroy_write_struct( &png_ptr, (png_infopp)NULL ); | |
627 | return FALSE; | |
628 | } | |
629 | ||
630 | for (int y = 0; y < image->GetHeight(); y++) | |
631 | { | |
632 | unsigned char *ptr = image->GetData() + (y * image->GetWidth() * 3); | |
633 | for (int x = 0; x < image->GetWidth(); x++) | |
634 | { | |
635 | data[(x << 2) + 0] = *ptr++; | |
636 | data[(x << 2) + 1] = *ptr++; | |
637 | data[(x << 2) + 2] = *ptr++; | |
638 | if ((data[(x << 2) + 0] == image->GetMaskRed()) && | |
639 | (data[(x << 2) + 1] == image->GetMaskGreen()) && | |
640 | (data[(x << 2) + 2] == image->GetMaskBlue())) | |
641 | data[(x << 2) + 3] = 0; | |
642 | else | |
643 | data[(x << 2) + 3] = 255; | |
644 | } | |
645 | png_bytep row_ptr = data; | |
646 | png_write_rows( png_ptr, &row_ptr, 1 ); | |
647 | } | |
648 | free(data); | |
649 | png_write_end( png_ptr, info_ptr ); | |
650 | png_destroy_write_struct( &png_ptr, (png_infopp)NULL ); | |
651 | ||
652 | fclose(f); | |
653 | } | |
654 | return TRUE; | |
655 | } | |
656 | ||
657 | //----------------------------------------------------------------------------- | |
658 | // wxBMPHandler | |
659 | //----------------------------------------------------------------------------- | |
660 | ||
661 | #if !USE_SHARED_LIBRARIES | |
662 | IMPLEMENT_DYNAMIC_CLASS(wxBMPHandler,wxImageHandler) | |
663 | #endif | |
664 | ||
665 | bool wxBMPHandler::LoadFile( wxImage *image, const wxString& name ) | |
666 | { | |
667 | FILE *file; | |
668 | unsigned char *data, *ptr; | |
669 | int done, i, bpp, planes, comp, ncolors, line, column, | |
670 | linesize, linepos, rshift = 0, gshift = 0, bshift = 0; | |
671 | unsigned char byte; | |
672 | short int word; | |
673 | long int dbuf[4], dword, rmask = 0, gmask = 0, bmask = 0, offset, | |
674 | size; | |
675 | signed char bbuf[4]; | |
676 | struct _cmap | |
677 | { | |
678 | unsigned char r, g, b; | |
679 | } | |
680 | *cmap = NULL; | |
681 | #ifndef BI_RGB | |
682 | #define BI_RGB 0 | |
683 | #define BI_RLE8 1 | |
684 | #define BI_RLE4 2 | |
685 | #define BI_BITFIELDS 3 | |
686 | #endif | |
687 | ||
688 | image->Destroy(); | |
689 | ||
690 | file = fopen(name, "r"); | |
691 | if (!file) | |
692 | return NULL; | |
693 | ||
694 | done = 0; | |
695 | /* | |
696 | * Reading the bmp header | |
697 | */ | |
698 | ||
699 | fread(&bbuf, 1, 2, file); | |
700 | ||
701 | fread(dbuf, 4, 4, file); | |
702 | ||
703 | size = dbuf[0]; | |
704 | offset = dbuf[2]; | |
705 | ||
706 | fread(dbuf, 4, 2, file); | |
707 | int width = (int)dbuf[0]; | |
708 | int height = (int)dbuf[1]; | |
709 | if (width > 32767) | |
710 | { | |
711 | fprintf(stderr, "IMLIB ERROR: Image width > 32767 pixels for file\n"); | |
712 | fclose(file); | |
713 | return FALSE; | |
714 | } | |
715 | if (height > 32767) | |
716 | { | |
717 | fprintf(stderr, "IMLIB ERROR: Image height > 32767 pixels for file\n"); | |
718 | fclose(file); | |
719 | return FALSE; | |
720 | } | |
721 | fread(&word, 2, 1, file); | |
722 | planes = (int)word; | |
723 | fread(&word, 2, 1, file); | |
724 | bpp = (int)word; | |
725 | if (bpp != 1 && bpp != 4 && bpp != 8 && bpp && 16 && bpp != 24 && bpp != 32) | |
726 | { | |
727 | fprintf(stderr, "IMLIB ERROR: unknown bitdepth in file\n"); | |
728 | fclose(file); | |
729 | return FALSE; | |
730 | } | |
731 | fread(dbuf, 4, 4, file); | |
732 | comp = (int)dbuf[0]; | |
733 | if (comp != BI_RGB && comp != BI_RLE4 && comp != BI_RLE8 && comp != BI_BITFIELDS) | |
734 | { | |
735 | fprintf(stderr, "IMLIB ERROR: unknown encoding in Windows BMP file\n"); | |
736 | fclose(file); | |
737 | return FALSE; | |
738 | } | |
739 | fread(dbuf, 4, 2, file); | |
740 | ncolors = (int)dbuf[0]; | |
741 | if (ncolors == 0) | |
742 | ncolors = 1 << bpp; | |
743 | /* some more sanity checks */ | |
744 | if (((comp == BI_RLE4) && (bpp != 4)) || ((comp == BI_RLE8) && (bpp != 8)) || ((comp == BI_BITFIELDS) && (bpp != 16 && bpp != 32))) | |
745 | { | |
746 | fprintf(stderr, "IMLIB ERROR: encoding of BMP doesn't match bitdepth\n"); | |
747 | fclose(file); | |
748 | return FALSE; | |
749 | } | |
750 | if (bpp < 16) | |
751 | { | |
752 | cmap = (struct _cmap *)malloc(sizeof(struct _cmap) * ncolors); | |
753 | ||
754 | if (!cmap) | |
755 | { | |
756 | fprintf(stderr, "IMLIB ERROR: Cannot allocate RAM for color map in BMP file\n"); | |
757 | fclose(file); | |
758 | return FALSE; | |
759 | } | |
760 | } | |
761 | else | |
762 | cmap = NULL; | |
763 | ||
764 | image->Create( width, height ); | |
765 | ptr = image->GetData(); | |
766 | if (!ptr) | |
767 | { | |
768 | fprintf(stderr, "IMLIB ERROR: Cannot allocate RAM for RGB data in file\n"); | |
769 | fclose(file); | |
770 | if (cmap) | |
771 | free(cmap); | |
772 | return FALSE; | |
773 | } | |
774 | ||
775 | /* | |
776 | * Reading the palette, if it exists. | |
777 | */ | |
778 | if (bpp < 16 && ncolors != 0) | |
779 | { | |
780 | for (i = 0; i < ncolors; i++) | |
781 | { | |
782 | fread(bbuf, 1, 4, file); | |
783 | cmap[i].b = bbuf[0]; | |
784 | cmap[i].g = bbuf[1]; | |
785 | cmap[i].r = bbuf[2]; | |
786 | } | |
787 | } | |
788 | else if (bpp == 16 || bpp == 32) | |
789 | { | |
790 | if (comp == BI_BITFIELDS) | |
791 | { | |
792 | int bit = 0; | |
793 | ||
794 | fread(dbuf, 4, 3, file); | |
795 | bmask = dbuf[0]; | |
796 | gmask = dbuf[1]; | |
797 | rmask = dbuf[2]; | |
798 | /* find shift amount.. ugly, but i can't think of a better way */ | |
799 | for (bit = 0; bit < bpp; bit++) | |
800 | { | |
801 | if (bmask & (1 << bit)) | |
802 | bshift = bit; | |
803 | if (gmask & (1 << bit)) | |
804 | gshift = bit; | |
805 | if (rmask & (1 << bit)) | |
806 | rshift = bit; | |
807 | } | |
808 | } | |
809 | else if (bpp == 16) | |
810 | { | |
811 | rmask = 0x7C00; | |
812 | gmask = 0x03E0; | |
813 | bmask = 0x001F; | |
814 | rshift = 10; | |
815 | gshift = 5; | |
816 | bshift = 0; | |
817 | } | |
818 | else if (bpp == 32) | |
819 | { | |
820 | rmask = 0x00FF0000; | |
821 | gmask = 0x0000FF00; | |
822 | bmask = 0x000000FF; | |
823 | rshift = 16; | |
824 | gshift = 8; | |
825 | bshift = 0; | |
826 | } | |
827 | } | |
828 | ||
829 | /* | |
830 | * REading the image data | |
831 | */ | |
832 | fseek(file, offset, SEEK_SET); | |
833 | data = ptr; | |
834 | ||
835 | /* set the whole image to the background color */ | |
836 | if (bpp < 16 && (comp == BI_RLE4 || comp == BI_RLE8)) | |
837 | { | |
838 | for (i = 0; i < width * height; i++) | |
839 | { | |
840 | *ptr++ = cmap[0].r; | |
841 | *ptr++ = cmap[0].g; | |
842 | *ptr++ = cmap[0].b; | |
843 | } | |
844 | ptr = data; | |
845 | } | |
846 | line = 0; | |
847 | column = 0; | |
848 | #define poffset (line * width * 3 + column * 3) | |
849 | ||
850 | /* | |
851 | * BMPs are stored upside down... hmmmmmmmmmm.... | |
852 | */ | |
853 | ||
854 | linesize = ((width * bpp + 31) / 32) * 4; | |
855 | for (line = (height - 1); line >= 0; line--) | |
856 | { | |
857 | linepos = 0; | |
858 | for (column = 0; column < width;) | |
859 | { | |
860 | if (bpp < 16) | |
861 | { | |
862 | int index; | |
863 | ||
864 | linepos++; | |
865 | byte = getc(file); | |
866 | if (bpp == 1) | |
867 | { | |
868 | int bit = 0; | |
869 | ||
870 | for (bit = 0; bit < 8; bit++) | |
871 | { | |
872 | index = ((byte & (0x80 >> bit)) ? 1 : 0); | |
873 | ptr[poffset] = cmap[index].r; | |
874 | ptr[poffset + 1] = cmap[index].g; | |
875 | ptr[poffset + 2] = cmap[index].b; | |
876 | column++; | |
877 | } | |
878 | } | |
879 | else if (bpp == 4) | |
880 | { | |
881 | if (comp == BI_RLE4) | |
882 | { | |
883 | fprintf(stderr, "can't deal with 4bit encoded yet.\n"); | |
884 | image->Destroy(); | |
885 | free(cmap); | |
886 | return FALSE; | |
887 | } | |
888 | else | |
889 | { | |
890 | int nibble = 0; | |
891 | ||
892 | for (nibble = 0; nibble < 2; nibble++) | |
893 | { | |
894 | index = ((byte & (0xF0 >> nibble * 4)) >> (!nibble * 4)); | |
895 | if (index >= 16) | |
896 | index = 15; | |
897 | ptr[poffset] = cmap[index].r; | |
898 | ptr[poffset + 1] = cmap[index].g; | |
899 | ptr[poffset + 2] = cmap[index].b; | |
900 | column++; | |
901 | } | |
902 | } | |
903 | } | |
904 | else if (bpp == 8) | |
905 | { | |
906 | if (comp == BI_RLE8) | |
907 | { | |
908 | unsigned char first; | |
909 | ||
910 | first = byte; | |
911 | byte = getc(file); | |
912 | if (first == 0) | |
913 | { | |
914 | if (byte == 0) | |
915 | { | |
916 | /* column = width; */ | |
917 | } | |
918 | else if (byte == 1) | |
919 | { | |
920 | column = width; | |
921 | line = -1; | |
922 | } | |
923 | else if (byte == 2) | |
924 | { | |
925 | byte = getc(file); | |
926 | column += byte; | |
927 | linepos = column * bpp / 8; | |
928 | byte = getc(file); | |
929 | line += byte; | |
930 | } | |
931 | else | |
932 | { | |
933 | int absolute = byte; | |
934 | ||
935 | for (i = 0; i < absolute; i++) | |
936 | { | |
937 | linepos++; | |
938 | byte = getc(file); | |
939 | ptr[poffset] = cmap[byte].r; | |
940 | ptr[poffset + 1] = cmap[byte].g; | |
941 | ptr[poffset + 2] = cmap[byte].b; | |
942 | column++; | |
943 | } | |
944 | if (absolute & 0x01) | |
945 | byte = getc(file); | |
946 | } | |
947 | } | |
948 | else | |
949 | { | |
950 | for (i = 0; i < first; i++) | |
951 | { | |
952 | ptr[poffset] = cmap[byte].r; | |
953 | ptr[poffset + 1] = cmap[byte].g; | |
954 | ptr[poffset + 2] = cmap[byte].b; | |
955 | column++; | |
956 | linepos++; | |
957 | } | |
958 | } | |
959 | } | |
960 | else | |
961 | { | |
962 | ptr[poffset] = cmap[byte].r; | |
963 | ptr[poffset + 1] = cmap[byte].g; | |
964 | ptr[poffset + 2] = cmap[byte].b; | |
965 | column++; | |
966 | linepos += size; | |
967 | } | |
968 | } | |
969 | } | |
970 | else if (bpp == 24) | |
971 | { | |
972 | linepos += fread(&bbuf, 1, 3, file); | |
973 | ptr[poffset] = (unsigned char)bbuf[2]; | |
974 | ptr[poffset + 1] = (unsigned char)bbuf[1]; | |
975 | ptr[poffset + 2] = (unsigned char)bbuf[0]; | |
976 | column++; | |
977 | } | |
978 | else if (bpp == 16) | |
979 | { | |
980 | unsigned char temp; | |
981 | ||
982 | linepos += fread(&word, 2, 1, file); | |
983 | temp = (word & rmask) >> rshift; | |
984 | ptr[poffset] = temp; | |
985 | temp = (word & gmask) >> gshift; | |
986 | ptr[poffset + 1] = temp; | |
987 | temp = (word & bmask) >> gshift; | |
988 | ptr[poffset + 2] = temp; | |
989 | column++; | |
990 | } | |
991 | else | |
992 | { | |
993 | unsigned char temp; | |
994 | ||
995 | linepos += fread(&dword, 4, 1, file); | |
996 | temp = (dword & rmask) >> rshift; | |
997 | ptr[poffset] = temp; | |
998 | temp = (dword & gmask) >> gshift; | |
999 | ptr[poffset + 1] = temp; | |
1000 | temp = (dword & bmask) >> bshift; | |
1001 | ptr[poffset + 2] = temp; | |
1002 | column++; | |
1003 | } | |
1004 | } | |
1005 | while ((linepos < linesize) && (comp != 1) && (comp != 2)) | |
1006 | { | |
1007 | int temp = fread(&byte, 1, 1, file); | |
1008 | ||
1009 | linepos += temp; | |
1010 | if (!temp) | |
1011 | break; | |
1012 | } | |
1013 | } | |
1014 | if (cmap) free(cmap); | |
1015 | ||
1016 | image->SetMask( FALSE ); | |
1017 | ||
1018 | fclose(file); | |
1019 | return TRUE; | |
1020 | } | |
1021 |