src/common/image.cpp

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