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