src/msw/imaglist.cpp

   1 /////////////////////////////////////////////////////////////////////////////
   2 // Name:        src/msw/imaglist.cpp
   3 // Purpose:     wxImageList implementation for Win32
   4 // Author:      Julian Smart
   5 // Modified by:
   6 // Created:     04/01/98
   7 // RCS-ID:      $Id$
   8 // Copyright:   (c) Julian Smart and Markus Holzem
   9 // Licence:     wxWindows license
  10 /////////////////////////////////////////////////////////////////////////////
  11
  12 // ============================================================================
  13 // declarations
  14 // ============================================================================
  15
  16 // ----------------------------------------------------------------------------
  17 // headers
  18 // ----------------------------------------------------------------------------
  19
  20 #ifdef __GNUG__
  21     #pragma implementation "imaglist.h"
  22 #endif
  23
  24 // For compilers that support precompilation, includes "wx.h".
  25 #include "wx/wxprec.h"
  26
  27 #ifdef __BORLANDC__
  28     #pragma hdrstop
  29 #endif
  30
  31 #if defined(__WIN95__)
  32
  33 #ifndef WX_PRECOMP
  34     #include "wx/window.h"
  35     #include "wx/icon.h"
  36     #include "wx/dc.h"
  37     #include "wx/string.h"
  38     #include "wx/dcmemory.h"
  39
  40     #include <stdio.h>
  41 #endif
  42
  43 #include "wx/log.h"
  44 #include "wx/intl.h"
  45
  46 #include "wx/msw/imaglist.h"
  47 #include "wx/msw/private.h"
  48
  49 #if !defined(__GNUWIN32_OLD__) && !defined(__TWIN32__)
  50     #include <commctrl.h>
  51 #endif
  52
  53 // ----------------------------------------------------------------------------
  54 // wxWin macros
  55 // ----------------------------------------------------------------------------
  56
  57 IMPLEMENT_DYNAMIC_CLASS(wxImageList, wxObject)
  58
  59 #define GetHImageList()     ((HIMAGELIST)m_hImageList)
  60
  61 // ----------------------------------------------------------------------------
  62 // private functions
  63 // ----------------------------------------------------------------------------
  64
  65 // returns the mask if it's valid, otherwise the bitmap mask and, if it's not
  66 // valid neither, a "solid" mask (no transparent zones at all)
  67 static HBITMAP GetMaskForImage(const wxBitmap& bitmap, const wxBitmap& mask);
  68
  69 // ============================================================================
  70 // implementation
  71 // ============================================================================
  72
  73 // ----------------------------------------------------------------------------
  74 // wxImageList creation/destruction
  75 // ----------------------------------------------------------------------------
  76
  77 wxImageList::wxImageList()
  78 {
  79     m_hImageList = 0;
  80 }
  81
  82 // Creates an image list
  83 bool wxImageList::Create(int width, int height, bool mask, int initial)
  84 {
  85     UINT flags = 0;
  86
  87     // set appropriate color depth
  88     int dd = wxDisplayDepth();
  89     if (dd <= 4)       flags |= ILC_COLOR;      // 16 color
  90     else if (dd <= 8)  flags |= ILC_COLOR8;     // 256 color
  91     else if (dd <= 16) flags |= ILC_COLOR16;    // 64k hi-color
  92     else if (dd <= 24) flags |= ILC_COLOR24;    // 16m truecolor
  93     else if (dd <= 32) flags |= ILC_COLOR32;    // 16m truecolor
  94
  95     if ( mask )
  96         flags |= ILC_MASK;
  97
  98     // Grow by 1, I guess this is reasonable behaviour most of the time
  99     m_hImageList = (WXHIMAGELIST) ImageList_Create(width, height, flags,
 100                                                    initial, 1);
 101     if ( !m_hImageList )
 102     {
 103         wxLogLastError(wxT("ImageList_Create()"));
 104     }
 105
 106     return m_hImageList != 0;
 107 }
 108
 109 wxImageList::~wxImageList()
 110 {
 111     if ( m_hImageList )
 112     {
 113         ImageList_Destroy(GetHImageList());
 114         m_hImageList = 0;
 115     }
 116 }
 117
 118 // ----------------------------------------------------------------------------
 119 // wxImageList attributes
 120 // ----------------------------------------------------------------------------
 121
 122 // Returns the number of images in the image list.
 123 int wxImageList::GetImageCount() const
 124 {
 125     wxASSERT_MSG( m_hImageList, _T("invalid image list") );
 126
 127     return ImageList_GetImageCount(GetHImageList());
 128 }
 129
 130 // Returns the size (same for all images) of the images in the list
 131 bool wxImageList::GetSize(int WXUNUSED(index), int &width, int &height) const
 132 {
 133     wxASSERT_MSG( m_hImageList, _T("invalid image list") );
 134
 135     return ImageList_GetIconSize(GetHImageList(), &width, &height) != 0;
 136 }
 137
 138 // ----------------------------------------------------------------------------
 139 // wxImageList operations
 140 // ----------------------------------------------------------------------------
 141
 142 // Adds a bitmap, and optionally a mask bitmap.
 143 // Note that wxImageList creates new bitmaps, so you may delete
 144 // 'bitmap' and 'mask'.
 145 int wxImageList::Add(const wxBitmap& bitmap, const wxBitmap& mask)
 146 {
 147     HBITMAP hbmpMask = GetMaskForImage(bitmap, mask);
 148
 149     int index = ImageList_Add(GetHImageList(), GetHbitmapOf(bitmap), hbmpMask);
 150     if ( index == -1 )
 151     {
 152         wxLogError(_("Couldn't add an image to the image list."));
 153     }
 154
 155     ::DeleteObject(hbmpMask);
 156
 157     return index;
 158 }
 159
 160 // Adds a bitmap, using the specified colour to create the mask bitmap
 161 // Note that wxImageList creates new bitmaps, so you may delete
 162 // 'bitmap'.
 163 int wxImageList::Add(const wxBitmap& bitmap, const wxColour& maskColour)
 164 {
 165 #ifdef __TWIN32__
 166     wxFAIL_MSG(_T("ImageList_AddMasked not implemented in TWIN32"));
 167     return -1;
 168 #else
 169     int index = ImageList_AddMasked(GetHImageList(),
 170                                     GetHbitmapOf(bitmap),
 171                                     wxColourToRGB(maskColour));
 172     if ( index == -1 )
 173     {
 174         wxLogError(_("Couldn't add an image to the image list."));
 175     }
 176
 177     return index;
 178 #endif
 179 }
 180
 181 // Adds a bitmap and mask from an icon.
 182 int wxImageList::Add(const wxIcon& icon)
 183 {
 184     int index = ImageList_AddIcon(GetHImageList(), GetHiconOf(icon));
 185     if ( index == -1 )
 186     {
 187         wxLogError(_("Couldn't add an image to the image list."));
 188     }
 189
 190     return index;
 191 }
 192
 193 // Replaces a bitmap, optionally passing a mask bitmap.
 194 // Note that wxImageList creates new bitmaps, so you may delete
 195 // 'bitmap' and 'mask'.
 196 bool wxImageList::Replace(int index,
 197                           const wxBitmap& bitmap, const wxBitmap& mask)
 198 {
 199 #ifdef __TWIN32__
 200     wxFAIL_MSG(_T("ImageList_Replace not implemented in TWIN32"));
 201     return FALSE;
 202 #else
 203     HBITMAP hbmpMask = GetMaskForImage(bitmap, mask);
 204
 205     bool ok = ImageList_Replace(GetHImageList(), index,
 206                                 GetHbitmapOf(bitmap), hbmpMask) != 0;
 207     if ( !ok )
 208     {
 209         wxLogLastError(wxT("ImageList_Add()"));
 210     }
 211
 212     ::DeleteObject(hbmpMask);
 213
 214     return ok;
 215 #endif
 216 }
 217
 218 // Replaces a bitmap and mask from an icon.
 219 bool wxImageList::Replace(int i, const wxIcon& icon)
 220 {
 221     bool ok = ImageList_ReplaceIcon(GetHImageList(), i, GetHiconOf(icon)) != 0;
 222     if ( !ok )
 223     {
 224         wxLogLastError(wxT("ImageList_ReplaceIcon()"));
 225     }
 226
 227     return ok;
 228 }
 229
 230 // Removes the image at the given index.
 231 bool wxImageList::Remove(int index)
 232 {
 233 #ifdef __TWIN32__
 234     wxFAIL_MSG(_T("ImageList_Replace not implemented in TWIN32"));
 235     return FALSE;
 236 #else
 237     bool ok = ImageList_Remove(GetHImageList(), index) != 0;
 238     if ( !ok )
 239     {
 240         wxLogLastError(wxT("ImageList_Remove()"));
 241     }
 242
 243     return ok;
 244 #endif
 245 }
 246
 247 // Remove all images
 248 bool wxImageList::RemoveAll()
 249 {
 250     // don't use ImageList_RemoveAll() because mingw32 headers don't have it
 251     int count = ImageList_GetImageCount(GetHImageList());
 252     for ( int i = 0; i < count; i++ )
 253     {
 254         // the image indexes are shifted, so we should always remove the first
 255         // one
 256         (void)Remove(0);
 257     }
 258
 259     return TRUE;
 260 }
 261
 262 // Draws the given image on a dc at the specified position.
 263 // If 'solidBackground' is TRUE, Draw sets the image list background
 264 // colour to the background colour of the wxDC, to speed up
 265 // drawing by eliminating masked drawing where possible.
 266 bool wxImageList::Draw(int index,
 267                        wxDC& dc,
 268                        int x, int y,
 269                        int flags,
 270                        bool solidBackground)
 271 {
 272 #ifdef __TWIN32__
 273     wxFAIL_MSG(_T("ImageList_Replace not implemented in TWIN32"));
 274     return FALSE;
 275 #else
 276     HDC hDC = GetHdcOf(dc);
 277     wxCHECK_MSG( hDC, FALSE, _T("invalid wxDC in wxImageList::Draw") );
 278
 279     COLORREF clr = CLR_NONE;    // transparent by default
 280     if ( solidBackground )
 281     {
 282         wxBrush *brush = & dc.GetBackground();
 283         if ( brush && brush->Ok() )
 284         {
 285             clr = wxColourToRGB(brush->GetColour());
 286         }
 287     }
 288
 289     ImageList_SetBkColor(GetHImageList(), clr);
 290
 291     UINT style = 0;
 292     if ( flags & wxIMAGELIST_DRAW_NORMAL )
 293         style |= ILD_NORMAL;
 294     if ( flags & wxIMAGELIST_DRAW_TRANSPARENT )
 295         style |= ILD_TRANSPARENT;
 296     if ( flags & wxIMAGELIST_DRAW_SELECTED )
 297         style |= ILD_SELECTED;
 298     if ( flags & wxIMAGELIST_DRAW_FOCUSED )
 299         style |= ILD_FOCUS;
 300
 301     bool ok = ImageList_Draw(GetHImageList(), index, hDC, x, y, style) != 0;
 302     if ( !ok )
 303     {
 304         wxLogLastError(wxT("ImageList_Draw()"));
 305     }
 306
 307     return ok;
 308 #endif
 309 }
 310
 311 // ----------------------------------------------------------------------------
 312 // helpers
 313 // ----------------------------------------------------------------------------
 314
 315 static HBITMAP GetMaskForImage(const wxBitmap& bitmap, const wxBitmap& mask)
 316 {
 317     HBITMAP hbmpMask;
 318     wxBitmap *bmpMask = NULL;
 319
 320     if ( mask.Ok() )
 321     {
 322         hbmpMask = GetHbitmapOf(mask);
 323     }
 324     else
 325     {
 326         wxMask *pMask = bitmap.GetMask();
 327         if ( pMask )
 328         {
 329             hbmpMask = (HBITMAP)pMask->GetMaskBitmap();
 330         }
 331         else
 332         {
 333             // create a non transparent mask - apparently, this is needed under
 334             // Win9x (it doesn't behave correctly if it's passed 0 mask)
 335             bmpMask = new wxBitmap(bitmap.GetWidth(), bitmap.GetHeight(), 1);
 336
 337             wxMemoryDC dcMem;
 338             dcMem.SelectObject(*bmpMask);
 339             dcMem.Clear();
 340             dcMem.SelectObject(wxNullBitmap);
 341
 342             hbmpMask = GetHbitmapOf(*bmpMask);
 343         }
 344     }
 345
 346     // windows mask convention is opposite to the wxWindows one
 347     HBITMAP hbmpMaskInv = wxInvertMask(hbmpMask);
 348     delete bmpMask;
 349
 350     return hbmpMaskInv;
 351 }
 352
 353 #endif // Win95
 354