]> git.saurik.com Git - wxWidgets.git/blob - src/common/image.cpp
Doc tweaks
[wxWidgets.git] / src / common / image.cpp
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 #if defined(__GNUG__) && !defined(NO_GCC_PRAGMA)
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/defs.h"
22
23 #if wxUSE_IMAGE
24
25 #include "wx/image.h"
26 #include "wx/bitmap.h"
27 #include "wx/debug.h"
28 #include "wx/log.h"
29 #include "wx/app.h"
30 #include "wx/filefn.h"
31 #include "wx/wfstream.h"
32 #include "wx/intl.h"
33 #include "wx/module.h"
34 #include "wx/hash.h"
35 #include "wx/utils.h"
36 #include "wx/math.h"
37
38 #if wxUSE_XPM
39 #include "wx/xpmdecod.h"
40 #endif
41
42 // For memcpy
43 #include <string.h>
44
45 #ifdef __SALFORDC__
46 #undef FAR
47 #endif
48
49
50 //-----------------------------------------------------------------------------
51 // wxImage
52 //-----------------------------------------------------------------------------
53
54 class wxImageRefData: public wxObjectRefData
55 {
56 public:
57 wxImageRefData();
58 virtual ~wxImageRefData();
59
60 int m_width;
61 int m_height;
62 unsigned char *m_data;
63
64 bool m_hasMask;
65 unsigned char m_maskRed,m_maskGreen,m_maskBlue;
66
67 // alpha channel data, may be NULL for the formats without alpha support
68 unsigned char *m_alpha;
69
70 bool m_ok;
71 bool m_static;
72
73 #if wxUSE_PALETTE
74 wxPalette m_palette;
75 #endif // wxUSE_PALETTE
76
77 wxArrayString m_optionNames;
78 wxArrayString m_optionValues;
79
80 DECLARE_NO_COPY_CLASS(wxImageRefData)
81 };
82
83 wxImageRefData::wxImageRefData()
84 {
85 m_width = 0;
86 m_height = 0;
87 m_data =
88 m_alpha = (unsigned char *) NULL;
89
90 m_maskRed = 0;
91 m_maskGreen = 0;
92 m_maskBlue = 0;
93 m_hasMask = false;
94
95 m_ok = false;
96 m_static = false;
97 }
98
99 wxImageRefData::~wxImageRefData()
100 {
101 if (!m_static)
102 {
103 free( m_data );
104 free( m_alpha );
105 }
106 }
107
108 wxList wxImage::sm_handlers;
109
110 wxImage wxNullImage;
111
112 //-----------------------------------------------------------------------------
113
114 #define M_IMGDATA ((wxImageRefData *)m_refData)
115
116 IMPLEMENT_DYNAMIC_CLASS(wxImage, wxObject)
117
118 wxImage::wxImage( int width, int height, bool clear )
119 {
120 Create( width, height, clear );
121 }
122
123 wxImage::wxImage( int width, int height, unsigned char* data, bool static_data )
124 {
125 Create( width, height, data, static_data );
126 }
127
128 wxImage::wxImage( int width, int height, unsigned char* data, unsigned char* alpha, bool static_data )
129 {
130 Create( width, height, data, alpha, static_data );
131 }
132
133 wxImage::wxImage( const wxString& name, long type, int index )
134 {
135 LoadFile( name, type, index );
136 }
137
138 wxImage::wxImage( const wxString& name, const wxString& mimetype, int index )
139 {
140 LoadFile( name, mimetype, index );
141 }
142
143 #if wxUSE_STREAMS
144 wxImage::wxImage( wxInputStream& stream, long type, int index )
145 {
146 LoadFile( stream, type, index );
147 }
148
149 wxImage::wxImage( wxInputStream& stream, const wxString& mimetype, int index )
150 {
151 LoadFile( stream, mimetype, index );
152 }
153 #endif // wxUSE_STREAMS
154
155 wxImage::wxImage( const wxImage& image )
156 : wxObject()
157 {
158 Ref(image);
159 }
160
161 wxImage::wxImage( const wxImage* image )
162 {
163 if (image) Ref(*image);
164 }
165
166 wxImage::wxImage( const char** xpmData )
167 {
168 Create(xpmData);
169 }
170
171 wxImage::wxImage( char** xpmData )
172 {
173 Create((const char**) xpmData);
174 }
175
176 bool wxImage::Create( const char** xpmData )
177 {
178 #if wxUSE_XPM
179 UnRef();
180
181 wxXPMDecoder decoder;
182 (*this) = decoder.ReadData(xpmData);
183 return Ok();
184 #else
185 return false;
186 #endif
187 }
188
189 bool wxImage::Create( int width, int height, bool clear )
190 {
191 UnRef();
192
193 m_refData = new wxImageRefData();
194
195 M_IMGDATA->m_data = (unsigned char *) malloc( width*height*3 );
196 if (!M_IMGDATA->m_data)
197 {
198 UnRef();
199 return false;
200 }
201
202 if (clear)
203 memset(M_IMGDATA->m_data, 0, width*height*3);
204
205 M_IMGDATA->m_width = width;
206 M_IMGDATA->m_height = height;
207 M_IMGDATA->m_ok = true;
208
209 return true;
210 }
211
212 bool wxImage::Create( int width, int height, unsigned char* data, bool static_data )
213 {
214 UnRef();
215
216 wxCHECK_MSG( data, false, _T("NULL data in wxImage::Create") );
217
218 m_refData = new wxImageRefData();
219
220 M_IMGDATA->m_data = data;
221 M_IMGDATA->m_width = width;
222 M_IMGDATA->m_height = height;
223 M_IMGDATA->m_ok = true;
224 M_IMGDATA->m_static = static_data;
225
226 return true;
227 }
228
229 bool wxImage::Create( int width, int height, unsigned char* data, unsigned char* alpha, bool static_data )
230 {
231 UnRef();
232
233 wxCHECK_MSG( data, false, _T("NULL data in wxImage::Create") );
234
235 m_refData = new wxImageRefData();
236
237 M_IMGDATA->m_data = data;
238 M_IMGDATA->m_alpha = alpha;
239 M_IMGDATA->m_width = width;
240 M_IMGDATA->m_height = height;
241 M_IMGDATA->m_ok = true;
242 M_IMGDATA->m_static = static_data;
243
244 return true;
245 }
246
247 void wxImage::Destroy()
248 {
249 UnRef();
250 }
251
252 wxImage wxImage::Copy() const
253 {
254 wxImage image;
255
256 wxCHECK_MSG( Ok(), image, wxT("invalid image") );
257
258 image.Create( M_IMGDATA->m_width, M_IMGDATA->m_height, false );
259
260 unsigned char *data = image.GetData();
261
262 wxCHECK_MSG( data, image, wxT("unable to create image") );
263
264 image.SetMaskColour( M_IMGDATA->m_maskRed, M_IMGDATA->m_maskGreen, M_IMGDATA->m_maskBlue );
265 image.SetMask( M_IMGDATA->m_hasMask );
266
267 memcpy( data, GetData(), M_IMGDATA->m_width*M_IMGDATA->m_height*3 );
268
269 // also copy the image options
270 wxImageRefData *imgData = (wxImageRefData *)image.m_refData;
271 imgData->m_optionNames = M_IMGDATA->m_optionNames;
272 imgData->m_optionValues = M_IMGDATA->m_optionValues;
273
274 return image;
275 }
276
277 wxImage wxImage::ShrinkBy( int xFactor , int yFactor ) const
278 {
279 if( xFactor == 1 && yFactor == 1 )
280 return Copy() ;
281
282 wxImage image;
283
284 wxCHECK_MSG( Ok(), image, wxT("invalid image") );
285
286 // can't scale to/from 0 size
287 wxCHECK_MSG( (xFactor > 0) && (yFactor > 0), image,
288 wxT("invalid new image size") );
289
290 long old_height = M_IMGDATA->m_height,
291 old_width = M_IMGDATA->m_width;
292
293 wxCHECK_MSG( (old_height > 0) && (old_width > 0), image,
294 wxT("invalid old image size") );
295
296 long width = old_width / xFactor ;
297 long height = old_height / yFactor ;
298
299 image.Create( width, height, false );
300
301 char unsigned *data = image.GetData();
302
303 wxCHECK_MSG( data, image, wxT("unable to create image") );
304
305 bool hasMask = false ;
306 unsigned char maskRed = 0;
307 unsigned char maskGreen = 0;
308 unsigned char maskBlue =0 ;
309
310 unsigned char *source_data = M_IMGDATA->m_data;
311 unsigned char *target_data = data;
312 unsigned char *source_alpha = 0 ;
313 unsigned char *target_alpha = 0 ;
314 if (M_IMGDATA->m_hasMask)
315 {
316 hasMask = true ;
317 maskRed = M_IMGDATA->m_maskRed;
318 maskGreen = M_IMGDATA->m_maskGreen;
319 maskBlue =M_IMGDATA->m_maskBlue ;
320
321 image.SetMaskColour( M_IMGDATA->m_maskRed,
322 M_IMGDATA->m_maskGreen,
323 M_IMGDATA->m_maskBlue );
324 }
325 else
326 {
327 source_alpha = M_IMGDATA->m_alpha ;
328 if ( source_alpha )
329 {
330 image.SetAlpha() ;
331 target_alpha = image.GetAlpha() ;
332 }
333 }
334
335 for (long y = 0; y < height; y++)
336 {
337 for (long x = 0; x < width; x++)
338 {
339 unsigned long avgRed = 0 ;
340 unsigned long avgGreen = 0;
341 unsigned long avgBlue = 0;
342 unsigned long avgAlpha = 0 ;
343 unsigned long counter = 0 ;
344 // determine average
345 for ( int y1 = 0 ; y1 < yFactor ; ++y1 )
346 {
347 long y_offset = (y * yFactor + y1) * old_width;
348 for ( int x1 = 0 ; x1 < xFactor ; ++x1 )
349 {
350 unsigned char *pixel = source_data + 3 * ( y_offset + x * xFactor + x1 ) ;
351 unsigned char red = pixel[0] ;
352 unsigned char green = pixel[1] ;
353 unsigned char blue = pixel[2] ;
354 unsigned char alpha = 255 ;
355 if ( source_alpha )
356 alpha = *(source_alpha + y_offset + x * xFactor + x1) ;
357 if ( !hasMask || red != maskRed || green != maskGreen || blue != maskBlue )
358 {
359 if ( alpha > 0 )
360 {
361 avgRed += red ;
362 avgGreen += green ;
363 avgBlue += blue ;
364 }
365 avgAlpha += alpha ;
366 counter++ ;
367 }
368 }
369 }
370 if ( counter == 0 )
371 {
372 *(target_data++) = M_IMGDATA->m_maskRed ;
373 *(target_data++) = M_IMGDATA->m_maskGreen ;
374 *(target_data++) = M_IMGDATA->m_maskBlue ;
375 }
376 else
377 {
378 if ( source_alpha )
379 *(target_alpha++) = (unsigned char)(avgAlpha / counter ) ;
380 *(target_data++) = (unsigned char)(avgRed / counter);
381 *(target_data++) = (unsigned char)(avgGreen / counter);
382 *(target_data++) = (unsigned char)(avgBlue / counter);
383 }
384 }
385 }
386
387 // In case this is a cursor, make sure the hotspot is scalled accordingly:
388 if ( HasOption(wxIMAGE_OPTION_CUR_HOTSPOT_X) )
389 image.SetOption(wxIMAGE_OPTION_CUR_HOTSPOT_X,
390 (GetOptionInt(wxIMAGE_OPTION_CUR_HOTSPOT_X))/xFactor);
391 if ( HasOption(wxIMAGE_OPTION_CUR_HOTSPOT_Y) )
392 image.SetOption(wxIMAGE_OPTION_CUR_HOTSPOT_Y,
393 (GetOptionInt(wxIMAGE_OPTION_CUR_HOTSPOT_Y))/yFactor);
394
395 return image;
396 }
397
398 wxImage wxImage::Scale( int width, int height ) const
399 {
400 wxImage image;
401
402 wxCHECK_MSG( Ok(), image, wxT("invalid image") );
403
404 // can't scale to/from 0 size
405 wxCHECK_MSG( (width > 0) && (height > 0), image,
406 wxT("invalid new image size") );
407
408 long old_height = M_IMGDATA->m_height,
409 old_width = M_IMGDATA->m_width;
410 wxCHECK_MSG( (old_height > 0) && (old_width > 0), image,
411 wxT("invalid old image size") );
412
413 if ( old_width % width == 0 && old_width >= width &&
414 old_height % height == 0 && old_height >= height )
415 {
416 return ShrinkBy( old_width / width , old_height / height ) ;
417 }
418 image.Create( width, height, false );
419
420 unsigned char *data = image.GetData();
421
422 wxCHECK_MSG( data, image, wxT("unable to create image") );
423
424 unsigned char *source_data = M_IMGDATA->m_data;
425 unsigned char *target_data = data;
426 unsigned char *source_alpha = 0 ;
427 unsigned char *target_alpha = 0 ;
428
429 if (M_IMGDATA->m_hasMask)
430 {
431 image.SetMaskColour( M_IMGDATA->m_maskRed,
432 M_IMGDATA->m_maskGreen,
433 M_IMGDATA->m_maskBlue );
434 }
435 else
436 {
437 source_alpha = M_IMGDATA->m_alpha ;
438 if ( source_alpha )
439 {
440 image.SetAlpha() ;
441 target_alpha = image.GetAlpha() ;
442 }
443 }
444
445 long x_delta = (old_width<<16) / width;
446 long y_delta = (old_height<<16) / height;
447
448 unsigned char* dest_pixel = target_data;
449
450 long y = 0;
451 for ( long j = 0; j < height; j++ )
452 {
453 unsigned char* src_line = &source_data[(y>>16)*old_width*3];
454 unsigned char* src_alpha_line = source_alpha ? &source_alpha[(y>>16)*old_width] : 0 ;
455
456 long x = 0;
457 for ( long i = 0; i < width; i++ )
458 {
459 unsigned char* src_pixel = &src_line[(x>>16)*3];
460 unsigned char* src_alpha_pixel = source_alpha ? &src_alpha_line[(x>>16)] : 0 ;
461 dest_pixel[0] = src_pixel[0];
462 dest_pixel[1] = src_pixel[1];
463 dest_pixel[2] = src_pixel[2];
464 dest_pixel += 3;
465 if ( source_alpha )
466 *(target_alpha++) = *src_alpha_pixel ;
467 x += x_delta;
468 }
469
470 y += y_delta;
471 }
472
473 // In case this is a cursor, make sure the hotspot is scalled accordingly:
474 if ( HasOption(wxIMAGE_OPTION_CUR_HOTSPOT_X) )
475 image.SetOption(wxIMAGE_OPTION_CUR_HOTSPOT_X,
476 (GetOptionInt(wxIMAGE_OPTION_CUR_HOTSPOT_X)*width)/old_width);
477 if ( HasOption(wxIMAGE_OPTION_CUR_HOTSPOT_Y) )
478 image.SetOption(wxIMAGE_OPTION_CUR_HOTSPOT_Y,
479 (GetOptionInt(wxIMAGE_OPTION_CUR_HOTSPOT_Y)*height)/old_height);
480
481 return image;
482 }
483
484 wxImage wxImage::Rotate90( bool clockwise ) const
485 {
486 wxImage image;
487
488 wxCHECK_MSG( Ok(), image, wxT("invalid image") );
489
490 image.Create( M_IMGDATA->m_height, M_IMGDATA->m_width, false );
491
492 unsigned char *data = image.GetData();
493
494 wxCHECK_MSG( data, image, wxT("unable to create image") );
495
496 if (M_IMGDATA->m_hasMask)
497 image.SetMaskColour( M_IMGDATA->m_maskRed, M_IMGDATA->m_maskGreen, M_IMGDATA->m_maskBlue );
498
499 long height = M_IMGDATA->m_height;
500 long width = M_IMGDATA->m_width;
501
502 unsigned char *source_data = M_IMGDATA->m_data;
503 unsigned char *target_data;
504
505 for (long j = 0; j < height; j++)
506 {
507 for (long i = 0; i < width; i++)
508 {
509 if (clockwise)
510 target_data = data + (((i+1)*height) - j - 1)*3;
511 else
512 target_data = data + ((height*(width-1)) + j - (i*height))*3;
513 memcpy( target_data, source_data, 3 );
514 source_data += 3;
515 }
516 }
517
518 return image;
519 }
520
521 wxImage wxImage::Mirror( bool horizontally ) const
522 {
523 wxImage image;
524
525 wxCHECK_MSG( Ok(), image, wxT("invalid image") );
526
527 image.Create( M_IMGDATA->m_width, M_IMGDATA->m_height, false );
528
529 unsigned char *data = image.GetData();
530
531 wxCHECK_MSG( data, image, wxT("unable to create image") );
532
533 if (M_IMGDATA->m_hasMask)
534 image.SetMaskColour( M_IMGDATA->m_maskRed, M_IMGDATA->m_maskGreen, M_IMGDATA->m_maskBlue );
535
536 long height = M_IMGDATA->m_height;
537 long width = M_IMGDATA->m_width;
538
539 unsigned char *source_data = M_IMGDATA->m_data;
540 unsigned char *target_data;
541
542 if (horizontally)
543 {
544 for (long j = 0; j < height; j++)
545 {
546 data += width*3;
547 target_data = data-3;
548 for (long i = 0; i < width; i++)
549 {
550 memcpy( target_data, source_data, 3 );
551 source_data += 3;
552 target_data -= 3;
553 }
554 }
555 }
556 else
557 {
558 for (long i = 0; i < height; i++)
559 {
560 target_data = data + 3*width*(height-1-i);
561 memcpy( target_data, source_data, (size_t)3*width );
562 source_data += 3*width;
563 }
564 }
565
566 return image;
567 }
568
569 wxImage wxImage::GetSubImage( const wxRect &rect ) const
570 {
571 wxImage image;
572
573 wxCHECK_MSG( Ok(), image, wxT("invalid image") );
574
575 wxCHECK_MSG( (rect.GetLeft()>=0) && (rect.GetTop()>=0) && (rect.GetRight()<=GetWidth()) && (rect.GetBottom()<=GetHeight()),
576 image, wxT("invalid subimage size") );
577
578 int subwidth=rect.GetWidth();
579 const int subheight=rect.GetHeight();
580
581 image.Create( subwidth, subheight, false );
582
583 unsigned char *subdata = image.GetData(), *data=GetData();
584
585 wxCHECK_MSG( subdata, image, wxT("unable to create image") );
586
587 if (M_IMGDATA->m_hasMask)
588 image.SetMaskColour( M_IMGDATA->m_maskRed, M_IMGDATA->m_maskGreen, M_IMGDATA->m_maskBlue );
589
590 const int subleft=3*rect.GetLeft();
591 const int width=3*GetWidth();
592 subwidth*=3;
593
594 data+=rect.GetTop()*width+subleft;
595
596 for (long j = 0; j < subheight; ++j)
597 {
598 memcpy( subdata, data, subwidth);
599 subdata+=subwidth;
600 data+=width;
601 }
602
603 return image;
604 }
605
606 wxImage wxImage::Size( const wxSize& size, const wxPoint& pos,
607 int r_, int g_, int b_ ) const
608 {
609 wxImage image;
610
611 wxCHECK_MSG( Ok(), image, wxT("invalid image") );
612 wxCHECK_MSG( (size.GetWidth() > 0) && (size.GetHeight() > 0), image, wxT("invalid size") );
613
614 int width = GetWidth(), height = GetHeight();
615 image.Create(size.GetWidth(), size.GetHeight(), false);
616
617 unsigned char r = (unsigned char)r_;
618 unsigned char g = (unsigned char)g_;
619 unsigned char b = (unsigned char)b_;
620 if ((r_ == -1) && (g_ == -1) && (b_ == -1))
621 {
622 GetOrFindMaskColour( &r, &g, &b );
623 image.SetMaskColour(r, g, b);
624 }
625
626 image.SetRGB(wxRect(), r, g, b);
627
628 wxRect subRect(pos.x, pos.y, width, height);
629 wxRect finalRect(0, 0, size.GetWidth(), size.GetHeight());
630
631 subRect.Intersect(finalRect);
632
633 if (!subRect.IsEmpty())
634 {
635 if ((subRect.GetWidth() == width) && (subRect.GetHeight() == height))
636 image.Paste(*this, pos.x, pos.y);
637 else
638 image.Paste(GetSubImage(subRect), pos.x, pos.y);
639 }
640
641 return image;
642 }
643
644 void wxImage::Paste( const wxImage &image, int x, int y )
645 {
646 wxCHECK_RET( Ok(), wxT("invalid image") );
647 wxCHECK_RET( image.Ok(), wxT("invalid image") );
648
649 int xx = 0;
650 int yy = 0;
651 int width = image.GetWidth();
652 int height = image.GetHeight();
653
654 if (x < 0)
655 {
656 xx = -x;
657 width += x;
658 }
659 if (y < 0)
660 {
661 yy = -y;
662 height += y;
663 }
664
665 if ((x+xx)+width > M_IMGDATA->m_width)
666 width = M_IMGDATA->m_width - (x+xx);
667 if ((y+yy)+height > M_IMGDATA->m_height)
668 height = M_IMGDATA->m_height - (y+yy);
669
670 if (width < 1) return;
671 if (height < 1) return;
672
673 if ((!HasMask() && !image.HasMask()) ||
674 (HasMask() && !image.HasMask()) ||
675 ((HasMask() && image.HasMask() &&
676 (GetMaskRed()==image.GetMaskRed()) &&
677 (GetMaskGreen()==image.GetMaskGreen()) &&
678 (GetMaskBlue()==image.GetMaskBlue()))))
679 {
680 width *= 3;
681 unsigned char* source_data = image.GetData() + xx*3 + yy*3*image.GetWidth();
682 int source_step = image.GetWidth()*3;
683
684 unsigned char* target_data = GetData() + (x+xx)*3 + (y+yy)*3*M_IMGDATA->m_width;
685 int target_step = M_IMGDATA->m_width*3;
686 for (int j = 0; j < height; j++)
687 {
688 memcpy( target_data, source_data, width );
689 source_data += source_step;
690 target_data += target_step;
691 }
692 return;
693 }
694
695 if (!HasMask() && image.HasMask())
696 {
697 unsigned char r = image.GetMaskRed();
698 unsigned char g = image.GetMaskGreen();
699 unsigned char b = image.GetMaskBlue();
700
701 width *= 3;
702 unsigned char* source_data = image.GetData() + xx*3 + yy*3*image.GetWidth();
703 int source_step = image.GetWidth()*3;
704
705 unsigned char* target_data = GetData() + (x+xx)*3 + (y+yy)*3*M_IMGDATA->m_width;
706 int target_step = M_IMGDATA->m_width*3;
707
708 for (int j = 0; j < height; j++)
709 {
710 for (int i = 0; i < width; i+=3)
711 {
712 if ((source_data[i] != r) &&
713 (source_data[i+1] != g) &&
714 (source_data[i+2] != b))
715 {
716 memcpy( target_data+i, source_data+i, 3 );
717 }
718 }
719 source_data += source_step;
720 target_data += target_step;
721 }
722 }
723 }
724
725 void wxImage::Replace( unsigned char r1, unsigned char g1, unsigned char b1,
726 unsigned char r2, unsigned char g2, unsigned char b2 )
727 {
728 wxCHECK_RET( Ok(), wxT("invalid image") );
729
730 unsigned char *data = GetData();
731
732 const int w = GetWidth();
733 const int h = GetHeight();
734
735 for (int j = 0; j < h; j++)
736 for (int i = 0; i < w; i++)
737 {
738 if ((data[0] == r1) && (data[1] == g1) && (data[2] == b1))
739 {
740 data[0] = r2;
741 data[1] = g2;
742 data[2] = b2;
743 }
744 data += 3;
745 }
746 }
747
748 wxImage wxImage::ConvertToMono( unsigned char r, unsigned char g, unsigned char b ) const
749 {
750 wxImage image;
751
752 wxCHECK_MSG( Ok(), image, wxT("invalid image") );
753
754 image.Create( M_IMGDATA->m_width, M_IMGDATA->m_height, false );
755
756 unsigned char *data = image.GetData();
757
758 wxCHECK_MSG( data, image, wxT("unable to create image") );
759
760 if (M_IMGDATA->m_hasMask)
761 {
762 if (M_IMGDATA->m_maskRed == r && M_IMGDATA->m_maskGreen == g &&
763 M_IMGDATA->m_maskBlue == b)
764 image.SetMaskColour( 255, 255, 255 );
765 else
766 image.SetMaskColour( 0, 0, 0 );
767 }
768
769 long size = M_IMGDATA->m_height * M_IMGDATA->m_width;
770
771 unsigned char *srcd = M_IMGDATA->m_data;
772 unsigned char *tard = image.GetData();
773
774 for ( long i = 0; i < size; i++, srcd += 3, tard += 3 )
775 {
776 if (srcd[0] == r && srcd[1] == g && srcd[2] == b)
777 tard[0] = tard[1] = tard[2] = 255;
778 else
779 tard[0] = tard[1] = tard[2] = 0;
780 }
781
782 return image;
783 }
784
785 void wxImage::SetRGB( int x, int y, unsigned char r, unsigned char g, unsigned char b )
786 {
787 wxCHECK_RET( Ok(), wxT("invalid image") );
788
789 int w = M_IMGDATA->m_width;
790 int h = M_IMGDATA->m_height;
791
792 wxCHECK_RET( (x>=0) && (y>=0) && (x<w) && (y<h), wxT("invalid image index") );
793
794 long pos = (y * w + x) * 3;
795
796 M_IMGDATA->m_data[ pos ] = r;
797 M_IMGDATA->m_data[ pos+1 ] = g;
798 M_IMGDATA->m_data[ pos+2 ] = b;
799 }
800
801 void wxImage::SetRGB( const wxRect& rect_, unsigned char r, unsigned char g, unsigned char b )
802 {
803 wxCHECK_RET( Ok(), wxT("invalid image") );
804
805 wxRect rect(rect_);
806 wxRect imageRect(0, 0, GetWidth(), GetHeight());
807 if ( rect == wxRect() )
808 {
809 rect = imageRect;
810 }
811 else
812 {
813 wxCHECK_RET( imageRect.Inside(rect.GetTopLeft()) &&
814 imageRect.Inside(rect.GetBottomRight()),
815 wxT("invalid bounding rectangle") );
816 }
817
818 int x1 = rect.GetLeft(),
819 y1 = rect.GetTop(),
820 x2 = rect.GetRight() + 1,
821 y2 = rect.GetBottom() + 1;
822
823 unsigned char *data wxDUMMY_INITIALIZE(NULL);
824 int x, y, width = GetWidth();
825 for (y = y1; y < y2; y++)
826 {
827 data = M_IMGDATA->m_data + (y*width + x1)*3;
828 for (x = x1; x < x2; x++)
829 {
830 *data++ = r;
831 *data++ = g;
832 *data++ = b;
833 }
834 }
835 }
836
837 unsigned char wxImage::GetRed( int x, int y ) const
838 {
839 wxCHECK_MSG( Ok(), 0, wxT("invalid image") );
840
841 int w = M_IMGDATA->m_width;
842 int h = M_IMGDATA->m_height;
843
844 wxCHECK_MSG( (x>=0) && (y>=0) && (x<w) && (y<h), 0, wxT("invalid image index") );
845
846 long pos = (y * w + x) * 3;
847
848 return M_IMGDATA->m_data[pos];
849 }
850
851 unsigned char wxImage::GetGreen( int x, int y ) const
852 {
853 wxCHECK_MSG( Ok(), 0, wxT("invalid image") );
854
855 int w = M_IMGDATA->m_width;
856 int h = M_IMGDATA->m_height;
857
858 wxCHECK_MSG( (x>=0) && (y>=0) && (x<w) && (y<h), 0, wxT("invalid image index") );
859
860 long pos = (y * w + x) * 3;
861
862 return M_IMGDATA->m_data[pos+1];
863 }
864
865 unsigned char wxImage::GetBlue( int x, int y ) const
866 {
867 wxCHECK_MSG( Ok(), 0, wxT("invalid image") );
868
869 int w = M_IMGDATA->m_width;
870 int h = M_IMGDATA->m_height;
871
872 wxCHECK_MSG( (x>=0) && (y>=0) && (x<w) && (y<h), 0, wxT("invalid image index") );
873
874 long pos = (y * w + x) * 3;
875
876 return M_IMGDATA->m_data[pos+2];
877 }
878
879 bool wxImage::Ok() const
880 {
881 // image of 0 width or height can't be considered ok - at least because it
882 // causes crashes in ConvertToBitmap() if we don't catch it in time
883 wxImageRefData *data = M_IMGDATA;
884 return data && data->m_ok && data->m_width && data->m_height;
885 }
886
887 unsigned char *wxImage::GetData() const
888 {
889 wxCHECK_MSG( Ok(), (unsigned char *)NULL, wxT("invalid image") );
890
891 return M_IMGDATA->m_data;
892 }
893
894 void wxImage::SetData( unsigned char *data )
895 {
896 wxCHECK_RET( Ok(), wxT("invalid image") );
897
898 wxImageRefData *newRefData = new wxImageRefData();
899
900 newRefData->m_width = M_IMGDATA->m_width;
901 newRefData->m_height = M_IMGDATA->m_height;
902 newRefData->m_data = data;
903 newRefData->m_ok = true;
904 newRefData->m_maskRed = M_IMGDATA->m_maskRed;
905 newRefData->m_maskGreen = M_IMGDATA->m_maskGreen;
906 newRefData->m_maskBlue = M_IMGDATA->m_maskBlue;
907 newRefData->m_hasMask = M_IMGDATA->m_hasMask;
908
909 UnRef();
910
911 m_refData = newRefData;
912 }
913
914 void wxImage::SetData( unsigned char *data, int new_width, int new_height )
915 {
916 wxImageRefData *newRefData = new wxImageRefData();
917
918 if (m_refData)
919 {
920 newRefData->m_width = new_width;
921 newRefData->m_height = new_height;
922 newRefData->m_data = data;
923 newRefData->m_ok = true;
924 newRefData->m_maskRed = M_IMGDATA->m_maskRed;
925 newRefData->m_maskGreen = M_IMGDATA->m_maskGreen;
926 newRefData->m_maskBlue = M_IMGDATA->m_maskBlue;
927 newRefData->m_hasMask = M_IMGDATA->m_hasMask;
928 }
929 else
930 {
931 newRefData->m_width = new_width;
932 newRefData->m_height = new_height;
933 newRefData->m_data = data;
934 newRefData->m_ok = true;
935 }
936
937 UnRef();
938
939 m_refData = newRefData;
940 }
941
942 // ----------------------------------------------------------------------------
943 // alpha channel support
944 // ----------------------------------------------------------------------------
945
946 void wxImage::SetAlpha(int x, int y, unsigned char alpha)
947 {
948 wxCHECK_RET( Ok() && HasAlpha(), wxT("invalid image or no alpha channel") );
949
950 int w = M_IMGDATA->m_width,
951 h = M_IMGDATA->m_height;
952
953 wxCHECK_RET( x >=0 && y >= 0 && x < w && y < h, wxT("invalid image index") );
954
955 M_IMGDATA->m_alpha[y*w + x] = alpha;
956 }
957
958 unsigned char wxImage::GetAlpha(int x, int y) const
959 {
960 wxCHECK_MSG( Ok() && HasAlpha(), 0, wxT("invalid image or no alpha channel") );
961
962 int w = M_IMGDATA->m_width,
963 h = M_IMGDATA->m_height;
964
965 wxCHECK_MSG( x >=0 && y >= 0 && x < w && y < h, 0, wxT("invalid image index") );
966
967 return M_IMGDATA->m_alpha[y*w + x];
968 }
969
970 bool wxImage::ConvertColourToAlpha( unsigned char r, unsigned char g, unsigned char b )
971 {
972 SetAlpha( NULL );
973
974 int w = M_IMGDATA->m_width,
975 h = M_IMGDATA->m_height;
976
977 unsigned char *alpha = GetAlpha();
978 unsigned char *data = GetData();
979
980 int x,y;
981 for (y = 0; y < h; y++)
982 for (x = 0; x < w; x++)
983 {
984 *alpha = *data;
985 alpha++;
986 *data = r;
987 data++;
988 *data = g;
989 data++;
990 *data = b;
991 data++;
992 }
993
994 return true;
995 }
996
997 void wxImage::SetAlpha( unsigned char *alpha )
998 {
999 wxCHECK_RET( Ok(), wxT("invalid image") );
1000
1001 if ( !alpha )
1002 {
1003 alpha = (unsigned char *)malloc(M_IMGDATA->m_width*M_IMGDATA->m_height);
1004 }
1005
1006 free(M_IMGDATA->m_alpha);
1007 M_IMGDATA->m_alpha = alpha;
1008 }
1009
1010 unsigned char *wxImage::GetAlpha() const
1011 {
1012 wxCHECK_MSG( Ok(), (unsigned char *)NULL, wxT("invalid image") );
1013
1014 return M_IMGDATA->m_alpha;
1015 }
1016
1017 void wxImage::InitAlpha()
1018 {
1019 wxCHECK_RET( !HasAlpha(), wxT("image already has an alpha channel") );
1020
1021 // initialize memory for alpha channel
1022 SetAlpha();
1023
1024 unsigned char *alpha = M_IMGDATA->m_alpha;
1025 const size_t lenAlpha = M_IMGDATA->m_width * M_IMGDATA->m_height;
1026
1027 static const unsigned char ALPHA_TRANSPARENT = 0;
1028 static const unsigned char ALPHA_OPAQUE = 0xff;
1029 if ( HasMask() )
1030 {
1031 // use the mask to initialize the alpha channel.
1032 const unsigned char * const alphaEnd = alpha + lenAlpha;
1033
1034 const unsigned char mr = M_IMGDATA->m_maskRed;
1035 const unsigned char mg = M_IMGDATA->m_maskGreen;
1036 const unsigned char mb = M_IMGDATA->m_maskBlue;
1037 for ( unsigned char *src = M_IMGDATA->m_data;
1038 alpha < alphaEnd;
1039 src += 3, alpha++ )
1040 {
1041 *alpha = (src[0] == mr && src[1] == mg && src[2] == mb)
1042 ? ALPHA_TRANSPARENT
1043 : ALPHA_OPAQUE;
1044 }
1045
1046 M_IMGDATA->m_hasMask = false;
1047 }
1048 else // no mask
1049 {
1050 // make the image fully opaque
1051 memset(alpha, ALPHA_OPAQUE, lenAlpha);
1052 }
1053 }
1054
1055 // ----------------------------------------------------------------------------
1056 // mask support
1057 // ----------------------------------------------------------------------------
1058
1059 void wxImage::SetMaskColour( unsigned char r, unsigned char g, unsigned char b )
1060 {
1061 wxCHECK_RET( Ok(), wxT("invalid image") );
1062
1063 M_IMGDATA->m_maskRed = r;
1064 M_IMGDATA->m_maskGreen = g;
1065 M_IMGDATA->m_maskBlue = b;
1066 M_IMGDATA->m_hasMask = true;
1067 }
1068
1069 bool wxImage::GetOrFindMaskColour( unsigned char *r, unsigned char *g, unsigned char *b ) const
1070 {
1071 wxCHECK_MSG( Ok(), false, wxT("invalid image") );
1072
1073 if (M_IMGDATA->m_hasMask)
1074 {
1075 if (r) *r = M_IMGDATA->m_maskRed;
1076 if (g) *g = M_IMGDATA->m_maskGreen;
1077 if (b) *b = M_IMGDATA->m_maskBlue;
1078 return true;
1079 }
1080 else
1081 {
1082 FindFirstUnusedColour(r, g, b);
1083 return false;
1084 }
1085 }
1086
1087 unsigned char wxImage::GetMaskRed() const
1088 {
1089 wxCHECK_MSG( Ok(), 0, wxT("invalid image") );
1090
1091 return M_IMGDATA->m_maskRed;
1092 }
1093
1094 unsigned char wxImage::GetMaskGreen() const
1095 {
1096 wxCHECK_MSG( Ok(), 0, wxT("invalid image") );
1097
1098 return M_IMGDATA->m_maskGreen;
1099 }
1100
1101 unsigned char wxImage::GetMaskBlue() const
1102 {
1103 wxCHECK_MSG( Ok(), 0, wxT("invalid image") );
1104
1105 return M_IMGDATA->m_maskBlue;
1106 }
1107
1108 void wxImage::SetMask( bool mask )
1109 {
1110 wxCHECK_RET( Ok(), wxT("invalid image") );
1111
1112 M_IMGDATA->m_hasMask = mask;
1113 }
1114
1115 bool wxImage::HasMask() const
1116 {
1117 wxCHECK_MSG( Ok(), false, wxT("invalid image") );
1118
1119 return M_IMGDATA->m_hasMask;
1120 }
1121
1122 int wxImage::GetWidth() const
1123 {
1124 wxCHECK_MSG( Ok(), 0, wxT("invalid image") );
1125
1126 return M_IMGDATA->m_width;
1127 }
1128
1129 int wxImage::GetHeight() const
1130 {
1131 wxCHECK_MSG( Ok(), 0, wxT("invalid image") );
1132
1133 return M_IMGDATA->m_height;
1134 }
1135
1136 bool wxImage::SetMaskFromImage(const wxImage& mask,
1137 unsigned char mr, unsigned char mg, unsigned char mb)
1138 {
1139 // check that the images are the same size
1140 if ( (M_IMGDATA->m_height != mask.GetHeight() ) || (M_IMGDATA->m_width != mask.GetWidth () ) )
1141 {
1142 wxLogError( _("Image and mask have different sizes.") );
1143 return false;
1144 }
1145
1146 // find unused colour
1147 unsigned char r,g,b ;
1148 if (!FindFirstUnusedColour(&r, &g, &b))
1149 {
1150 wxLogError( _("No unused colour in image being masked.") );
1151 return false ;
1152 }
1153
1154 unsigned char *imgdata = GetData();
1155 unsigned char *maskdata = mask.GetData();
1156
1157 const int w = GetWidth();
1158 const int h = GetHeight();
1159
1160 for (int j = 0; j < h; j++)
1161 {
1162 for (int i = 0; i < w; i++)
1163 {
1164 if ((maskdata[0] == mr) && (maskdata[1] == mg) && (maskdata[2] == mb))
1165 {
1166 imgdata[0] = r;
1167 imgdata[1] = g;
1168 imgdata[2] = b;
1169 }
1170 imgdata += 3;
1171 maskdata += 3;
1172 }
1173 }
1174
1175 SetMaskColour(r, g, b);
1176 SetMask(true);
1177
1178 return true;
1179 }
1180
1181 bool wxImage::ConvertAlphaToMask(unsigned char threshold)
1182 {
1183 if (!HasAlpha())
1184 return true;
1185
1186 unsigned char mr, mg, mb;
1187 if (!FindFirstUnusedColour(&mr, &mg, &mb))
1188 {
1189 wxLogError( _("No unused colour in image being masked.") );
1190 return false;
1191 }
1192
1193 SetMask(true);
1194 SetMaskColour(mr, mg, mb);
1195
1196 unsigned char *imgdata = GetData();
1197 unsigned char *alphadata = GetAlpha();
1198
1199 int w = GetWidth();
1200 int h = GetHeight();
1201
1202 for (int y = 0; y < h; y++)
1203 {
1204 for (int x = 0; x < w; x++, imgdata += 3, alphadata++)
1205 {
1206 if (*alphadata < threshold)
1207 {
1208 imgdata[0] = mr;
1209 imgdata[1] = mg;
1210 imgdata[2] = mb;
1211 }
1212 }
1213 }
1214
1215 free(M_IMGDATA->m_alpha);
1216 M_IMGDATA->m_alpha = NULL;
1217
1218 return true;
1219 }
1220
1221 #if wxUSE_PALETTE
1222
1223 // Palette functions
1224
1225 bool wxImage::HasPalette() const
1226 {
1227 if (!Ok())
1228 return false;
1229
1230 return M_IMGDATA->m_palette.Ok();
1231 }
1232
1233 const wxPalette& wxImage::GetPalette() const
1234 {
1235 wxCHECK_MSG( Ok(), wxNullPalette, wxT("invalid image") );
1236
1237 return M_IMGDATA->m_palette;
1238 }
1239
1240 void wxImage::SetPalette(const wxPalette& palette)
1241 {
1242 wxCHECK_RET( Ok(), wxT("invalid image") );
1243
1244 M_IMGDATA->m_palette = palette;
1245 }
1246
1247 #endif // wxUSE_PALETTE
1248
1249 // Option functions (arbitrary name/value mapping)
1250 void wxImage::SetOption(const wxString& name, const wxString& value)
1251 {
1252 wxCHECK_RET( Ok(), wxT("invalid image") );
1253
1254 int idx = M_IMGDATA->m_optionNames.Index(name, false);
1255 if (idx == wxNOT_FOUND)
1256 {
1257 M_IMGDATA->m_optionNames.Add(name);
1258 M_IMGDATA->m_optionValues.Add(value);
1259 }
1260 else
1261 {
1262 M_IMGDATA->m_optionNames[idx] = name;
1263 M_IMGDATA->m_optionValues[idx] = value;
1264 }
1265 }
1266
1267 void wxImage::SetOption(const wxString& name, int value)
1268 {
1269 wxString valStr;
1270 valStr.Printf(wxT("%d"), value);
1271 SetOption(name, valStr);
1272 }
1273
1274 wxString wxImage::GetOption(const wxString& name) const
1275 {
1276 wxCHECK_MSG( Ok(), wxEmptyString, wxT("invalid image") );
1277
1278 int idx = M_IMGDATA->m_optionNames.Index(name, false);
1279 if (idx == wxNOT_FOUND)
1280 return wxEmptyString;
1281 else
1282 return M_IMGDATA->m_optionValues[idx];
1283 }
1284
1285 int wxImage::GetOptionInt(const wxString& name) const
1286 {
1287 return wxAtoi(GetOption(name));
1288 }
1289
1290 bool wxImage::HasOption(const wxString& name) const
1291 {
1292 wxCHECK_MSG( Ok(), false, wxT("invalid image") );
1293
1294 return (M_IMGDATA->m_optionNames.Index(name, false) != wxNOT_FOUND);
1295 }
1296
1297 bool wxImage::LoadFile( const wxString& filename, long type, int index )
1298 {
1299 #if wxUSE_STREAMS
1300 if (wxFileExists(filename))
1301 {
1302 wxFileInputStream stream(filename);
1303 wxBufferedInputStream bstream( stream );
1304 return LoadFile(bstream, type, index);
1305 }
1306 else
1307 {
1308 wxLogError( _("Can't load image from file '%s': file does not exist."), filename.c_str() );
1309
1310 return false;
1311 }
1312 #else // !wxUSE_STREAMS
1313 return false;
1314 #endif // wxUSE_STREAMS
1315 }
1316
1317 bool wxImage::LoadFile( const wxString& filename, const wxString& mimetype, int index )
1318 {
1319 #if wxUSE_STREAMS
1320 if (wxFileExists(filename))
1321 {
1322 wxFileInputStream stream(filename);
1323 wxBufferedInputStream bstream( stream );
1324 return LoadFile(bstream, mimetype, index);
1325 }
1326 else
1327 {
1328 wxLogError( _("Can't load image from file '%s': file does not exist."), filename.c_str() );
1329
1330 return false;
1331 }
1332 #else // !wxUSE_STREAMS
1333 return false;
1334 #endif // wxUSE_STREAMS
1335 }
1336
1337
1338
1339 bool wxImage::SaveFile( const wxString& filename ) const
1340 {
1341 wxString ext = filename.AfterLast('.').Lower();
1342
1343 wxImageHandler * pHandler = FindHandler(ext, -1);
1344 if (pHandler)
1345 {
1346 SaveFile(filename, pHandler->GetType());
1347 return true;
1348 }
1349
1350 wxLogError(_("Can't save image to file '%s': unknown extension."), filename.c_str());
1351
1352 return false;
1353 }
1354
1355 bool wxImage::SaveFile( const wxString& filename, int type ) const
1356 {
1357 #if wxUSE_STREAMS
1358 wxCHECK_MSG( Ok(), false, wxT("invalid image") );
1359
1360 ((wxImage*)this)->SetOption(wxIMAGE_OPTION_FILENAME, filename);
1361
1362 wxFileOutputStream stream(filename);
1363
1364 if ( stream.IsOk() )
1365 {
1366 wxBufferedOutputStream bstream( stream );
1367 return SaveFile(bstream, type);
1368 }
1369 #endif // wxUSE_STREAMS
1370
1371 return false;
1372 }
1373
1374 bool wxImage::SaveFile( const wxString& filename, const wxString& mimetype ) const
1375 {
1376 #if wxUSE_STREAMS
1377 wxCHECK_MSG( Ok(), false, wxT("invalid image") );
1378
1379 ((wxImage*)this)->SetOption(wxIMAGE_OPTION_FILENAME, filename);
1380
1381 wxFileOutputStream stream(filename);
1382
1383 if ( stream.IsOk() )
1384 {
1385 wxBufferedOutputStream bstream( stream );
1386 return SaveFile(bstream, mimetype);
1387 }
1388 #endif // wxUSE_STREAMS
1389
1390 return false;
1391 }
1392
1393 bool wxImage::CanRead( const wxString &name )
1394 {
1395 #if wxUSE_STREAMS
1396 wxFileInputStream stream(name);
1397 return CanRead(stream);
1398 #else
1399 return false;
1400 #endif
1401 }
1402
1403 int wxImage::GetImageCount( const wxString &name, long type )
1404 {
1405 #if wxUSE_STREAMS
1406 wxFileInputStream stream(name);
1407 if (stream.Ok())
1408 return GetImageCount(stream, type);
1409 #endif
1410
1411 return 0;
1412 }
1413
1414 #if wxUSE_STREAMS
1415
1416 bool wxImage::CanRead( wxInputStream &stream )
1417 {
1418 const wxList& list = GetHandlers();
1419
1420 for ( wxList::compatibility_iterator node = list.GetFirst(); node; node = node->GetNext() )
1421 {
1422 wxImageHandler *handler=(wxImageHandler*)node->GetData();
1423 if (handler->CanRead( stream ))
1424 return true;
1425 }
1426
1427 return false;
1428 }
1429
1430 int wxImage::GetImageCount( wxInputStream &stream, long type )
1431 {
1432 wxImageHandler *handler;
1433
1434 if ( type == wxBITMAP_TYPE_ANY )
1435 {
1436 wxList &list=GetHandlers();
1437
1438 for (wxList::compatibility_iterator node = list.GetFirst(); node; node = node->GetNext())
1439 {
1440 handler=(wxImageHandler*)node->GetData();
1441 if ( handler->CanRead(stream) )
1442 return handler->GetImageCount(stream);
1443
1444 }
1445
1446 wxLogWarning(_("No handler found for image type."));
1447 return 0;
1448 }
1449
1450 handler = FindHandler(type);
1451
1452 if ( !handler )
1453 {
1454 wxLogWarning(_("No image handler for type %d defined."), type);
1455 return false;
1456 }
1457
1458 if ( handler->CanRead(stream) )
1459 {
1460 return handler->GetImageCount(stream);
1461 }
1462 else
1463 {
1464 wxLogError(_("Image file is not of type %d."), type);
1465 return 0;
1466 }
1467 }
1468
1469 bool wxImage::LoadFile( wxInputStream& stream, long type, int index )
1470 {
1471 UnRef();
1472
1473 m_refData = new wxImageRefData;
1474
1475 wxImageHandler *handler;
1476
1477 if ( type == wxBITMAP_TYPE_ANY )
1478 {
1479 wxList &list=GetHandlers();
1480
1481 for ( wxList::compatibility_iterator node = list.GetFirst(); node; node = node->GetNext() )
1482 {
1483 handler=(wxImageHandler*)node->GetData();
1484 if ( handler->CanRead(stream) )
1485 return handler->LoadFile(this, stream, true/*verbose*/, index);
1486
1487 }
1488
1489 wxLogWarning( _("No handler found for image type.") );
1490 return false;
1491 }
1492
1493 handler = FindHandler(type);
1494
1495 if (handler == 0)
1496 {
1497 wxLogWarning( _("No image handler for type %d defined."), type );
1498
1499 return false;
1500 }
1501
1502 return handler->LoadFile(this, stream, true/*verbose*/, index);
1503 }
1504
1505 bool wxImage::LoadFile( wxInputStream& stream, const wxString& mimetype, int index )
1506 {
1507 UnRef();
1508
1509 m_refData = new wxImageRefData;
1510
1511 wxImageHandler *handler = FindHandlerMime(mimetype);
1512
1513 if (handler == 0)
1514 {
1515 wxLogWarning( _("No image handler for type %s defined."), mimetype.GetData() );
1516
1517 return false;
1518 }
1519
1520 return handler->LoadFile( this, stream, true/*verbose*/, index );
1521 }
1522
1523 bool wxImage::SaveFile( wxOutputStream& stream, int type ) const
1524 {
1525 wxCHECK_MSG( Ok(), false, wxT("invalid image") );
1526
1527 wxImageHandler *handler = FindHandler(type);
1528 if ( !handler )
1529 {
1530 wxLogWarning( _("No image handler for type %d defined."), type );
1531
1532 return false;
1533 }
1534
1535 return handler->SaveFile( (wxImage*)this, stream );
1536 }
1537
1538 bool wxImage::SaveFile( wxOutputStream& stream, const wxString& mimetype ) const
1539 {
1540 wxCHECK_MSG( Ok(), false, wxT("invalid image") );
1541
1542 wxImageHandler *handler = FindHandlerMime(mimetype);
1543 if ( !handler )
1544 {
1545 wxLogWarning( _("No image handler for type %s defined."), mimetype.GetData() );
1546
1547 return false;
1548 }
1549
1550 return handler->SaveFile( (wxImage*)this, stream );
1551 }
1552 #endif // wxUSE_STREAMS
1553
1554 void wxImage::AddHandler( wxImageHandler *handler )
1555 {
1556 // Check for an existing handler of the type being added.
1557 if (FindHandler( handler->GetType() ) == 0)
1558 {
1559 sm_handlers.Append( handler );
1560 }
1561 else
1562 {
1563 // This is not documented behaviour, merely the simplest 'fix'
1564 // for preventing duplicate additions. If someone ever has
1565 // a good reason to add and remove duplicate handlers (and they
1566 // may) we should probably refcount the duplicates.
1567 // also an issue in InsertHandler below.
1568
1569 wxLogDebug( _T("Adding duplicate image handler for '%s'"),
1570 handler->GetName().c_str() );
1571 delete handler;
1572 }
1573 }
1574
1575 void wxImage::InsertHandler( wxImageHandler *handler )
1576 {
1577 // Check for an existing handler of the type being added.
1578 if (FindHandler( handler->GetType() ) == 0)
1579 {
1580 sm_handlers.Insert( handler );
1581 }
1582 else
1583 {
1584 // see AddHandler for additional comments.
1585 wxLogDebug( _T("Inserting duplicate image handler for '%s'"),
1586 handler->GetName().c_str() );
1587 delete handler;
1588 }
1589 }
1590
1591 bool wxImage::RemoveHandler( const wxString& name )
1592 {
1593 wxImageHandler *handler = FindHandler(name);
1594 if (handler)
1595 {
1596 sm_handlers.DeleteObject(handler);
1597 delete handler;
1598 return true;
1599 }
1600 else
1601 return false;
1602 }
1603
1604 wxImageHandler *wxImage::FindHandler( const wxString& name )
1605 {
1606 wxList::compatibility_iterator node = sm_handlers.GetFirst();
1607 while (node)
1608 {
1609 wxImageHandler *handler = (wxImageHandler*)node->GetData();
1610 if (handler->GetName().Cmp(name) == 0) return handler;
1611
1612 node = node->GetNext();
1613 }
1614 return 0;
1615 }
1616
1617 wxImageHandler *wxImage::FindHandler( const wxString& extension, long bitmapType )
1618 {
1619 wxList::compatibility_iterator node = sm_handlers.GetFirst();
1620 while (node)
1621 {
1622 wxImageHandler *handler = (wxImageHandler*)node->GetData();
1623 if ( (handler->GetExtension().Cmp(extension) == 0) &&
1624 (bitmapType == -1 || handler->GetType() == bitmapType) )
1625 return handler;
1626 node = node->GetNext();
1627 }
1628 return 0;
1629 }
1630
1631 wxImageHandler *wxImage::FindHandler( long bitmapType )
1632 {
1633 wxList::compatibility_iterator node = sm_handlers.GetFirst();
1634 while (node)
1635 {
1636 wxImageHandler *handler = (wxImageHandler *)node->GetData();
1637 if (handler->GetType() == bitmapType) return handler;
1638 node = node->GetNext();
1639 }
1640 return 0;
1641 }
1642
1643 wxImageHandler *wxImage::FindHandlerMime( const wxString& mimetype )
1644 {
1645 wxList::compatibility_iterator node = sm_handlers.GetFirst();
1646 while (node)
1647 {
1648 wxImageHandler *handler = (wxImageHandler *)node->GetData();
1649 if (handler->GetMimeType().IsSameAs(mimetype, false)) return handler;
1650 node = node->GetNext();
1651 }
1652 return 0;
1653 }
1654
1655 void wxImage::InitStandardHandlers()
1656 {
1657 #if wxUSE_STREAMS
1658 AddHandler(new wxBMPHandler);
1659 #endif // wxUSE_STREAMS
1660 }
1661
1662 void wxImage::CleanUpHandlers()
1663 {
1664 wxList::compatibility_iterator node = sm_handlers.GetFirst();
1665 while (node)
1666 {
1667 wxImageHandler *handler = (wxImageHandler *)node->GetData();
1668 wxList::compatibility_iterator next = node->GetNext();
1669 delete handler;
1670 node = next;
1671 }
1672
1673 sm_handlers.Clear();
1674 }
1675
1676 wxString wxImage::GetImageExtWildcard()
1677 {
1678 wxString fmts;
1679
1680 wxList& Handlers = wxImage::GetHandlers();
1681 wxList::compatibility_iterator Node = Handlers.GetFirst();
1682 while ( Node )
1683 {
1684 wxImageHandler* Handler = (wxImageHandler*)Node->GetData();
1685 fmts += wxT("*.") + Handler->GetExtension();
1686 Node = Node->GetNext();
1687 if ( Node ) fmts += wxT(";");
1688 }
1689
1690 return wxT("(") + fmts + wxT(")|") + fmts;
1691 }
1692
1693 //-----------------------------------------------------------------------------
1694 // wxImageHandler
1695 //-----------------------------------------------------------------------------
1696
1697 IMPLEMENT_ABSTRACT_CLASS(wxImageHandler,wxObject)
1698
1699 #if wxUSE_STREAMS
1700 bool wxImageHandler::LoadFile( wxImage *WXUNUSED(image), wxInputStream& WXUNUSED(stream), bool WXUNUSED(verbose), int WXUNUSED(index) )
1701 {
1702 return false;
1703 }
1704
1705 bool wxImageHandler::SaveFile( wxImage *WXUNUSED(image), wxOutputStream& WXUNUSED(stream), bool WXUNUSED(verbose) )
1706 {
1707 return false;
1708 }
1709
1710 int wxImageHandler::GetImageCount( wxInputStream& WXUNUSED(stream) )
1711 {
1712 return 1;
1713 }
1714
1715 bool wxImageHandler::CanRead( const wxString& name )
1716 {
1717 if (wxFileExists(name))
1718 {
1719 wxFileInputStream stream(name);
1720 return CanRead(stream);
1721 }
1722
1723 wxLogError( _("Can't check image format of file '%s': file does not exist."), name.c_str() );
1724
1725 return false;
1726 }
1727
1728 bool wxImageHandler::CallDoCanRead(wxInputStream& stream)
1729 {
1730 wxFileOffset posOld = stream.TellI();
1731 if ( posOld == wxInvalidOffset )
1732 {
1733 // can't test unseekable stream
1734 return false;
1735 }
1736
1737 bool ok = DoCanRead(stream);
1738
1739 // restore the old position to be able to test other formats and so on
1740 if ( stream.SeekI(posOld) == wxInvalidOffset )
1741 {
1742 wxLogDebug(_T("Failed to rewind the stream in wxImageHandler!"));
1743
1744 // reading would fail anyhow as we're not at the right position
1745 return false;
1746 }
1747
1748 return ok;
1749 }
1750
1751 #endif // wxUSE_STREAMS
1752
1753 // ----------------------------------------------------------------------------
1754 // image histogram stuff
1755 // ----------------------------------------------------------------------------
1756
1757 bool
1758 wxImageHistogram::FindFirstUnusedColour(unsigned char *r,
1759 unsigned char *g,
1760 unsigned char *b,
1761 unsigned char r2,
1762 unsigned char b2,
1763 unsigned char g2) const
1764 {
1765 unsigned long key = MakeKey(r2, g2, b2);
1766
1767 while ( find(key) != end() )
1768 {
1769 // color already used
1770 r2++;
1771 if ( r2 >= 255 )
1772 {
1773 r2 = 0;
1774 g2++;
1775 if ( g2 >= 255 )
1776 {
1777 g2 = 0;
1778 b2++;
1779 if ( b2 >= 255 )
1780 {
1781 wxLogError(_("No unused colour in image.") );
1782 return false;
1783 }
1784 }
1785 }
1786
1787 key = MakeKey(r2, g2, b2);
1788 }
1789
1790 if ( r )
1791 *r = r2;
1792 if ( g )
1793 *g = g2;
1794 if ( b )
1795 *b = b2;
1796
1797 return true;
1798 }
1799
1800 bool
1801 wxImage::FindFirstUnusedColour(unsigned char *r,
1802 unsigned char *g,
1803 unsigned char *b,
1804 unsigned char r2,
1805 unsigned char b2,
1806 unsigned char g2) const
1807 {
1808 wxImageHistogram histogram;
1809
1810 ComputeHistogram(histogram);
1811
1812 return histogram.FindFirstUnusedColour(r, g, b, r2, g2, b2);
1813 }
1814
1815
1816
1817 // GRG, Dic/99
1818 // Counts and returns the number of different colours. Optionally stops
1819 // when it exceeds 'stopafter' different colours. This is useful, for
1820 // example, to see if the image can be saved as 8-bit (256 colour or
1821 // less, in this case it would be invoked as CountColours(256)). Default
1822 // value for stopafter is -1 (don't care).
1823 //
1824 unsigned long wxImage::CountColours( unsigned long stopafter ) const
1825 {
1826 wxHashTable h;
1827 wxObject dummy;
1828 unsigned char r, g, b;
1829 unsigned char *p;
1830 unsigned long size, nentries, key;
1831
1832 p = GetData();
1833 size = GetWidth() * GetHeight();
1834 nentries = 0;
1835
1836 for (unsigned long j = 0; (j < size) && (nentries <= stopafter) ; j++)
1837 {
1838 r = *(p++);
1839 g = *(p++);
1840 b = *(p++);
1841 key = wxImageHistogram::MakeKey(r, g, b);
1842
1843 if (h.Get(key) == NULL)
1844 {
1845 h.Put(key, &dummy);
1846 nentries++;
1847 }
1848 }
1849
1850 return nentries;
1851 }
1852
1853
1854 unsigned long wxImage::ComputeHistogram( wxImageHistogram &h ) const
1855 {
1856 unsigned char *p = GetData();
1857 unsigned long nentries = 0;
1858
1859 h.clear();
1860
1861 const unsigned long size = GetWidth() * GetHeight();
1862
1863 unsigned char r, g, b;
1864 for ( unsigned long n = 0; n < size; n++ )
1865 {
1866 r = *p++;
1867 g = *p++;
1868 b = *p++;
1869
1870 wxImageHistogramEntry& entry = h[wxImageHistogram::MakeKey(r, g, b)];
1871
1872 if ( entry.value++ == 0 )
1873 entry.index = nentries++;
1874 }
1875
1876 return nentries;
1877 }
1878
1879 /*
1880 * Rotation code by Carlos Moreno
1881 */
1882
1883 // GRG: I've removed wxRotationPoint - we already have wxRealPoint which
1884 // does exactly the same thing. And I also got rid of wxRotationPixel
1885 // bacause of potential problems in architectures where alignment
1886 // is an issue, so I had to rewrite parts of the code.
1887
1888 static const double gs_Epsilon = 1e-10;
1889
1890 static inline int wxCint (double x)
1891 {
1892 return (x > 0) ? (int) (x + 0.5) : (int) (x - 0.5);
1893 }
1894
1895
1896 // Auxiliary function to rotate a point (x,y) with respect to point p0
1897 // make it inline and use a straight return to facilitate optimization
1898 // also, the function receives the sine and cosine of the angle to avoid
1899 // repeating the time-consuming calls to these functions -- sin/cos can
1900 // be computed and stored in the calling function.
1901
1902 inline wxRealPoint rotated_point (const wxRealPoint & p, double cos_angle, double sin_angle, const wxRealPoint & p0)
1903 {
1904 return wxRealPoint (p0.x + (p.x - p0.x) * cos_angle - (p.y - p0.y) * sin_angle,
1905 p0.y + (p.y - p0.y) * cos_angle + (p.x - p0.x) * sin_angle);
1906 }
1907
1908 inline wxRealPoint rotated_point (double x, double y, double cos_angle, double sin_angle, const wxRealPoint & p0)
1909 {
1910 return rotated_point (wxRealPoint(x,y), cos_angle, sin_angle, p0);
1911 }
1912
1913 wxImage wxImage::Rotate(double angle, const wxPoint & centre_of_rotation, bool interpolating, wxPoint * offset_after_rotation) const
1914 {
1915 int i;
1916 angle = -angle; // screen coordinates are a mirror image of "real" coordinates
1917
1918 bool has_alpha = HasAlpha();
1919
1920 // Create pointer-based array to accelerate access to wxImage's data
1921 unsigned char ** data = new unsigned char * [GetHeight()];
1922 data[0] = GetData();
1923 for (i = 1; i < GetHeight(); i++)
1924 data[i] = data[i - 1] + (3 * GetWidth());
1925
1926 // Same for alpha channel
1927 unsigned char ** alpha = NULL;
1928 if (has_alpha)
1929 {
1930 alpha = new unsigned char * [GetHeight()];
1931 alpha[0] = GetAlpha();
1932 for (i = 1; i < GetHeight(); i++)
1933 alpha[i] = alpha[i - 1] + GetWidth();
1934 }
1935
1936 // precompute coefficients for rotation formula
1937 // (sine and cosine of the angle)
1938 const double cos_angle = cos(angle);
1939 const double sin_angle = sin(angle);
1940
1941 // Create new Image to store the result
1942 // First, find rectangle that covers the rotated image; to do that,
1943 // rotate the four corners
1944
1945 const wxRealPoint p0(centre_of_rotation.x, centre_of_rotation.y);
1946
1947 wxRealPoint p1 = rotated_point (0, 0, cos_angle, sin_angle, p0);
1948 wxRealPoint p2 = rotated_point (0, GetHeight(), cos_angle, sin_angle, p0);
1949 wxRealPoint p3 = rotated_point (GetWidth(), 0, cos_angle, sin_angle, p0);
1950 wxRealPoint p4 = rotated_point (GetWidth(), GetHeight(), cos_angle, sin_angle, p0);
1951
1952 int x1 = (int) floor (wxMin (wxMin(p1.x, p2.x), wxMin(p3.x, p4.x)));
1953 int y1 = (int) floor (wxMin (wxMin(p1.y, p2.y), wxMin(p3.y, p4.y)));
1954 int x2 = (int) ceil (wxMax (wxMax(p1.x, p2.x), wxMax(p3.x, p4.x)));
1955 int y2 = (int) ceil (wxMax (wxMax(p1.y, p2.y), wxMax(p3.y, p4.y)));
1956
1957 // Create rotated image
1958 wxImage rotated (x2 - x1 + 1, y2 - y1 + 1, false);
1959 // With alpha channel
1960 if (has_alpha)
1961 rotated.SetAlpha();
1962
1963 if (offset_after_rotation != NULL)
1964 {
1965 *offset_after_rotation = wxPoint (x1, y1);
1966 }
1967
1968 // GRG: The rotated (destination) image is always accessed
1969 // sequentially, so there is no need for a pointer-based
1970 // array here (and in fact it would be slower).
1971 //
1972 unsigned char * dst = rotated.GetData();
1973
1974 unsigned char * alpha_dst = NULL;
1975 if (has_alpha)
1976 alpha_dst = rotated.GetAlpha();
1977
1978 // GRG: if the original image has a mask, use its RGB values
1979 // as the blank pixel, else, fall back to default (black).
1980 //
1981 unsigned char blank_r = 0;
1982 unsigned char blank_g = 0;
1983 unsigned char blank_b = 0;
1984
1985 if (HasMask())
1986 {
1987 blank_r = GetMaskRed();
1988 blank_g = GetMaskGreen();
1989 blank_b = GetMaskBlue();
1990 rotated.SetMaskColour( blank_r, blank_g, blank_b );
1991 }
1992
1993 // Now, for each point of the rotated image, find where it came from, by
1994 // performing an inverse rotation (a rotation of -angle) and getting the
1995 // pixel at those coordinates
1996
1997 // GRG: I've taken the (interpolating) test out of the loops, so that
1998 // it is done only once, instead of repeating it for each pixel.
1999
2000 int x;
2001 if (interpolating)
2002 {
2003 for (int y = 0; y < rotated.GetHeight(); y++)
2004 {
2005 for (x = 0; x < rotated.GetWidth(); x++)
2006 {
2007 wxRealPoint src = rotated_point (x + x1, y + y1, cos_angle, -sin_angle, p0);
2008
2009 if (-0.25 < src.x && src.x < GetWidth() - 0.75 &&
2010 -0.25 < src.y && src.y < GetHeight() - 0.75)
2011 {
2012 // interpolate using the 4 enclosing grid-points. Those
2013 // points can be obtained using floor and ceiling of the
2014 // exact coordinates of the point
2015 // C.M. 2000-02-17: when the point is near the border, special care is required.
2016
2017 int x1, y1, x2, y2;
2018
2019 if (0 < src.x && src.x < GetWidth() - 1)
2020 {
2021 x1 = wxCint(floor(src.x));
2022 x2 = wxCint(ceil(src.x));
2023 }
2024 else // else means that x is near one of the borders (0 or width-1)
2025 {
2026 x1 = x2 = wxCint (src.x);
2027 }
2028
2029 if (0 < src.y && src.y < GetHeight() - 1)
2030 {
2031 y1 = wxCint(floor(src.y));
2032 y2 = wxCint(ceil(src.y));
2033 }
2034 else
2035 {
2036 y1 = y2 = wxCint (src.y);
2037 }
2038
2039 // get four points and the distances (square of the distance,
2040 // for efficiency reasons) for the interpolation formula
2041
2042 // GRG: Do not calculate the points until they are
2043 // really needed -- this way we can calculate
2044 // just one, instead of four, if d1, d2, d3
2045 // or d4 are < gs_Epsilon
2046
2047 const double d1 = (src.x - x1) * (src.x - x1) + (src.y - y1) * (src.y - y1);
2048 const double d2 = (src.x - x2) * (src.x - x2) + (src.y - y1) * (src.y - y1);
2049 const double d3 = (src.x - x2) * (src.x - x2) + (src.y - y2) * (src.y - y2);
2050 const double d4 = (src.x - x1) * (src.x - x1) + (src.y - y2) * (src.y - y2);
2051
2052 // Now interpolate as a weighted average of the four surrounding
2053 // points, where the weights are the distances to each of those points
2054
2055 // If the point is exactly at one point of the grid of the source
2056 // image, then don't interpolate -- just assign the pixel
2057
2058 if (d1 < gs_Epsilon) // d1,d2,d3,d4 are positive -- no need for abs()
2059 {
2060 unsigned char *p = data[y1] + (3 * x1);
2061 *(dst++) = *(p++);
2062 *(dst++) = *(p++);
2063 *(dst++) = *p;
2064
2065 if (has_alpha)
2066 {
2067 unsigned char *p = alpha[y1] + x1;
2068 *(alpha_dst++) = *p;
2069 }
2070 }
2071 else if (d2 < gs_Epsilon)
2072 {
2073 unsigned char *p = data[y1] + (3 * x2);
2074 *(dst++) = *(p++);
2075 *(dst++) = *(p++);
2076 *(dst++) = *p;
2077
2078 if (has_alpha)
2079 {
2080 unsigned char *p = alpha[y1] + x2;
2081 *(alpha_dst++) = *p;
2082 }
2083 }
2084 else if (d3 < gs_Epsilon)
2085 {
2086 unsigned char *p = data[y2] + (3 * x2);
2087 *(dst++) = *(p++);
2088 *(dst++) = *(p++);
2089 *(dst++) = *p;
2090
2091 if (has_alpha)
2092 {
2093 unsigned char *p = alpha[y2] + x2;
2094 *(alpha_dst++) = *p;
2095 }
2096 }
2097 else if (d4 < gs_Epsilon)
2098 {
2099 unsigned char *p = data[y2] + (3 * x1);
2100 *(dst++) = *(p++);
2101 *(dst++) = *(p++);
2102 *(dst++) = *p;
2103
2104 if (has_alpha)
2105 {
2106 unsigned char *p = alpha[y2] + x1;
2107 *(alpha_dst++) = *p;
2108 }
2109 }
2110 else
2111 {
2112 // weights for the weighted average are proportional to the inverse of the distance
2113 unsigned char *v1 = data[y1] + (3 * x1);
2114 unsigned char *v2 = data[y1] + (3 * x2);
2115 unsigned char *v3 = data[y2] + (3 * x2);
2116 unsigned char *v4 = data[y2] + (3 * x1);
2117
2118 const double w1 = 1/d1, w2 = 1/d2, w3 = 1/d3, w4 = 1/d4;
2119
2120 // GRG: Unrolled.
2121
2122 *(dst++) = (unsigned char)
2123 ( (w1 * *(v1++) + w2 * *(v2++) +
2124 w3 * *(v3++) + w4 * *(v4++)) /
2125 (w1 + w2 + w3 + w4) );
2126 *(dst++) = (unsigned char)
2127 ( (w1 * *(v1++) + w2 * *(v2++) +
2128 w3 * *(v3++) + w4 * *(v4++)) /
2129 (w1 + w2 + w3 + w4) );
2130 *(dst++) = (unsigned char)
2131 ( (w1 * *v1 + w2 * *v2 +
2132 w3 * *v3 + w4 * *v4) /
2133 (w1 + w2 + w3 + w4) );
2134
2135 if (has_alpha)
2136 {
2137 unsigned char *v1 = alpha[y1] + (x1);
2138 unsigned char *v2 = alpha[y1] + (x2);
2139 unsigned char *v3 = alpha[y2] + (x2);
2140 unsigned char *v4 = alpha[y2] + (x1);
2141
2142 *(alpha_dst++) = (unsigned char)
2143 ( (w1 * *v1 + w2 * *v2 +
2144 w3 * *v3 + w4 * *v4) /
2145 (w1 + w2 + w3 + w4) );
2146 }
2147 }
2148 }
2149 else
2150 {
2151 *(dst++) = blank_r;
2152 *(dst++) = blank_g;
2153 *(dst++) = blank_b;
2154
2155 if (has_alpha)
2156 *(alpha_dst++) = 0;
2157 }
2158 }
2159 }
2160 }
2161 else // not interpolating
2162 {
2163 for (int y = 0; y < rotated.GetHeight(); y++)
2164 {
2165 for (x = 0; x < rotated.GetWidth(); x++)
2166 {
2167 wxRealPoint src = rotated_point (x + x1, y + y1, cos_angle, -sin_angle, p0);
2168
2169 const int xs = wxCint (src.x); // wxCint rounds to the
2170 const int ys = wxCint (src.y); // closest integer
2171
2172 if (0 <= xs && xs < GetWidth() &&
2173 0 <= ys && ys < GetHeight())
2174 {
2175 unsigned char *p = data[ys] + (3 * xs);
2176 *(dst++) = *(p++);
2177 *(dst++) = *(p++);
2178 *(dst++) = *p;
2179
2180 if (has_alpha)
2181 {
2182 unsigned char *p = alpha[ys] + (xs);
2183 *(alpha_dst++) = *p;
2184 }
2185 }
2186 else
2187 {
2188 *(dst++) = blank_r;
2189 *(dst++) = blank_g;
2190 *(dst++) = blank_b;
2191
2192 if (has_alpha)
2193 *(alpha_dst++) = 255;
2194 }
2195 }
2196 }
2197 }
2198
2199 delete [] data;
2200
2201 if (has_alpha)
2202 delete [] alpha;
2203
2204 return rotated;
2205 }
2206
2207
2208
2209
2210
2211 // A module to allow wxImage initialization/cleanup
2212 // without calling these functions from app.cpp or from
2213 // the user's application.
2214
2215 class wxImageModule: public wxModule
2216 {
2217 DECLARE_DYNAMIC_CLASS(wxImageModule)
2218 public:
2219 wxImageModule() {}
2220 bool OnInit() { wxImage::InitStandardHandlers(); return true; };
2221 void OnExit() { wxImage::CleanUpHandlers(); };
2222 };
2223
2224 IMPLEMENT_DYNAMIC_CLASS(wxImageModule, wxModule)
2225
2226
2227 #endif // wxUSE_IMAGE