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