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