src/generic/bmpcboxg.cpp

   1 /////////////////////////////////////////////////////////////////////////////
   2 // Name:        src/generic/bmpcboxg.cpp
   3 // Purpose:     wxBitmapComboBox
   4 // Author:      Jaakko Salli
   5 // Modified by:
   6 // Created:     Aug-31-2006
   7 // RCS-ID:      $Id:
   8 // Copyright:   (c) 2005 Jaakko Salli
   9 // Licence:     wxWindows licence
  10 /////////////////////////////////////////////////////////////////////////////
  11
  12 // ============================================================================
  13 // declarations
  14 // ============================================================================
  15
  16 // ----------------------------------------------------------------------------
  17 // headers
  18 // ----------------------------------------------------------------------------
  19
  20 #include "wx/wxprec.h"
  21
  22 #ifdef __BORLANDC__
  23     #pragma hdrstop
  24 #endif
  25
  26 #if wxUSE_BITMAPCOMBOBOX
  27
  28 #include "wx/bmpcbox.h"
  29
  30 #if defined(wxGENERIC_BITMAPCOMBOBOX)
  31
  32 #ifndef WX_PRECOMP
  33     #include "wx/log.h"
  34 #endif
  35
  36 #include "wx/odcombo.h"
  37 #include "wx/settings.h"
  38 #include "wx/dc.h"
  39
  40 #if wxUSE_IMAGE
  41     #include "wx/image.h"
  42 #endif
  43
  44
  45 const wxChar wxBitmapComboBoxNameStr[] = wxT("bitmapComboBox");
  46
  47
  48 // These macros allow wxArrayPtrVoid to be used in more convenient manner
  49 #define GetBitmapPtr(n)     ((wxBitmap*)m_bitmaps[n])
  50
  51
  52 #define IMAGE_SPACING_RIGHT         4  // Space left of image
  53
  54 #define IMAGE_SPACING_LEFT          4  // Space right of image, left of text
  55
  56 #define IMAGE_SPACING_VERTICAL      2  // Space top and bottom of image
  57
  58 #define IMAGE_SPACING_CTRL_VERTICAL 7  // Spacing used in control size calculation
  59
  60 #define EXTRA_FONT_HEIGHT           0  // Add to increase min. height of list items
  61
  62
  63 // ============================================================================
  64 // implementation
  65 // ============================================================================
  66
  67
  68 BEGIN_EVENT_TABLE(wxBitmapComboBox, wxOwnerDrawnComboBox)
  69     EVT_SIZE(wxBitmapComboBox::OnResize)
  70 END_EVENT_TABLE()
  71
  72
  73 IMPLEMENT_DYNAMIC_CLASS(wxBitmapComboBox, wxOwnerDrawnComboBox)
  74
  75 void wxBitmapComboBox::Init()
  76 {
  77     m_fontHeight = 0;
  78     m_imgAreaWidth = 0;
  79     m_inResize = false;
  80 }
  81
  82 wxBitmapComboBox::wxBitmapComboBox(wxWindow *parent,
  83                                   wxWindowID id,
  84                                   const wxString& value,
  85                                   const wxPoint& pos,
  86                                   const wxSize& size,
  87                                   const wxArrayString& choices,
  88                                   long style,
  89                                   const wxValidator& validator,
  90                                   const wxString& name)
  91     : wxOwnerDrawnComboBox(),
  92       wxBitmapComboBoxBase()
  93 {
  94     Init();
  95
  96     Create(parent,id,value,pos,size,choices,style,validator,name);
  97 }
  98
  99 bool wxBitmapComboBox::Create(wxWindow *parent,
 100                               wxWindowID id,
 101                               const wxString& value,
 102                               const wxPoint& pos,
 103                               const wxSize& size,
 104                               const wxArrayString& choices,
 105                               long style,
 106                               const wxValidator& validator,
 107                               const wxString& name)
 108 {
 109     if ( !wxOwnerDrawnComboBox::Create(parent, id, value,
 110                                        pos, size,
 111                                        choices, style,
 112                                        validator, name) )
 113     {
 114         return false;
 115     }
 116
 117     PostCreate();
 118
 119     return true;
 120 }
 121
 122 bool wxBitmapComboBox::Create(wxWindow *parent,
 123                               wxWindowID id,
 124                               const wxString& value,
 125                               const wxPoint& pos,
 126                               const wxSize& size,
 127                               int n,
 128                               const wxString choices[],
 129                               long style,
 130                               const wxValidator& validator,
 131                               const wxString& name)
 132 {
 133     if ( !wxOwnerDrawnComboBox::Create(parent, id, value,
 134                                        pos, size, n,
 135                                        choices, style,
 136                                        validator, name) )
 137     {
 138         return false;
 139     }
 140
 141     PostCreate();
 142
 143     return true;
 144 }
 145
 146 void wxBitmapComboBox::PostCreate()
 147 {
 148     m_fontHeight = GetCharHeight() + EXTRA_FONT_HEIGHT;
 149 }
 150
 151 wxBitmapComboBox::~wxBitmapComboBox()
 152 {
 153     Clear();
 154 }
 155
 156 // ----------------------------------------------------------------------------
 157 // Item manipulation
 158 // ----------------------------------------------------------------------------
 159
 160 void wxBitmapComboBox::SetItemBitmap(unsigned int n, const wxBitmap& bitmap)
 161 {
 162     wxCHECK_RET( n < m_bitmaps.size(), wxT("invalid item index") );
 163     OnAddBitmap(bitmap);
 164     *GetBitmapPtr(n) = bitmap;
 165
 166     if ( (int)n == GetSelection() )
 167         Refresh();
 168 }
 169
 170 wxBitmap wxBitmapComboBox::GetItemBitmap(unsigned int n) const
 171 {
 172     wxCHECK_MSG( n < m_bitmaps.size(), wxNullBitmap, wxT("invalid item index") );
 173     return *GetBitmapPtr(n);
 174 }
 175
 176 int wxBitmapComboBox::Insert(const wxString& item, const wxBitmap& bitmap,
 177                              unsigned int pos, void *clientData)
 178 {
 179     int n = DoInsertWithImage(item, bitmap, pos);
 180     if ( n != wxNOT_FOUND )
 181         SetClientData(n, clientData);
 182
 183     return n;
 184 }
 185
 186 int wxBitmapComboBox::Insert(const wxString& item, const wxBitmap& bitmap,
 187                              unsigned int pos, wxClientData *clientData)
 188 {
 189     int n = DoInsertWithImage(item, bitmap, pos);
 190     if ( n != wxNOT_FOUND )
 191         SetClientObject(n, clientData);
 192
 193     return n;
 194 }
 195
 196 bool wxBitmapComboBox::OnAddBitmap(const wxBitmap& bitmap)
 197 {
 198     if ( bitmap.Ok() )
 199     {
 200         int width = bitmap.GetWidth();
 201         int height = bitmap.GetHeight();
 202
 203         if ( m_usedImgSize.x <= 0 )
 204         {
 205             //
 206             // If size not yet determined, get it from this image.
 207             m_usedImgSize.x = width;
 208             m_usedImgSize.y = height;
 209
 210             InvalidateBestSize();
 211             wxSize newSz = GetBestSize();
 212             wxSize sz = GetSize();
 213             if ( newSz.y > sz.y )
 214                 SetSize(sz.x, newSz.y);
 215             else
 216                 DetermineIndent();
 217         }
 218
 219         wxCHECK_MSG(width == m_usedImgSize.x && height == m_usedImgSize.y,
 220                     false,
 221                     wxT("you can only add images of same size"));
 222     }
 223
 224     return true;
 225 }
 226
 227 bool wxBitmapComboBox::DoInsertBitmap(const wxBitmap& bitmap, unsigned int pos)
 228 {
 229     if ( !OnAddBitmap(bitmap) )
 230         return false;
 231
 232     // NB: We must try to set the image before DoInsert or
 233     //     DoAppend because OnMeasureItem might be called
 234     //     before it returns.
 235     m_bitmaps.Insert( new wxBitmap(bitmap), pos);
 236
 237     return true;
 238 }
 239
 240 int wxBitmapComboBox::DoAppendWithImage(const wxString& item, const wxBitmap& image)
 241 {
 242     unsigned int pos = m_bitmaps.size();
 243
 244     if ( !DoInsertBitmap(image, pos) )
 245         return wxNOT_FOUND;
 246
 247     int index = wxOwnerDrawnComboBox::DoAppend(item);
 248
 249     if ( index < 0 )
 250         index = m_bitmaps.size();
 251
 252     // Need to re-check the index incase DoAppend sorted
 253     if ( (unsigned int) index != pos )
 254     {
 255         wxBitmap* bmp = GetBitmapPtr(pos);
 256         m_bitmaps.RemoveAt(pos);
 257         m_bitmaps.Insert(bmp, index);
 258     }
 259
 260     return index;
 261 }
 262
 263 int wxBitmapComboBox::DoInsertWithImage(const wxString& item,
 264                                         const wxBitmap& image,
 265                                         unsigned int pos)
 266 {
 267     if ( !DoInsertBitmap(image, pos) )
 268         return wxNOT_FOUND;
 269
 270     return wxOwnerDrawnComboBox::DoInsert(item, pos);
 271 }
 272
 273 int wxBitmapComboBox::DoAppend(const wxString& item)
 274 {
 275     return DoAppendWithImage(item, wxNullBitmap);
 276 }
 277
 278 int wxBitmapComboBox::DoInsert(const wxString& item, unsigned int pos)
 279 {
 280     return DoInsertWithImage(item, wxNullBitmap, pos);
 281 }
 282
 283 void wxBitmapComboBox::Clear()
 284 {
 285     wxOwnerDrawnComboBox::Clear();
 286
 287     unsigned int i;
 288
 289     for ( i=0; i<m_bitmaps.size(); i++ )
 290         delete GetBitmapPtr(i);
 291
 292     m_bitmaps.Empty();
 293
 294     m_usedImgSize.x = 0;
 295     m_usedImgSize.y = 0;
 296
 297     DetermineIndent();
 298 }
 299
 300 void wxBitmapComboBox::Delete(unsigned int n)
 301 {
 302     wxOwnerDrawnComboBox::Delete(n);
 303     delete GetBitmapPtr(n);
 304     m_bitmaps.RemoveAt(n);
 305 }
 306
 307 // ----------------------------------------------------------------------------
 308 // wxBitmapComboBox event handlers and such
 309 // ----------------------------------------------------------------------------
 310
 311 void wxBitmapComboBox::DetermineIndent()
 312 {
 313     //
 314     // Recalculate amount of empty space needed in front of
 315     // text in control itself.
 316     int indent = m_imgAreaWidth = 0;
 317
 318     if ( m_usedImgSize.x > 0 )
 319     {
 320         indent = m_usedImgSize.y + IMAGE_SPACING_LEFT + IMAGE_SPACING_RIGHT;
 321         m_imgAreaWidth = indent;
 322
 323         indent -= 3;
 324     }
 325
 326     SetCustomPaintWidth(indent);
 327 }
 328
 329 void wxBitmapComboBox::OnResize(wxSizeEvent& event)
 330 {
 331     // Prevent infinite looping
 332     if ( !m_inResize )
 333     {
 334         m_inResize = true;
 335         DetermineIndent();
 336         m_inResize = false;
 337     }
 338
 339     event.Skip();
 340 }
 341
 342 wxSize wxBitmapComboBox::DoGetBestSize() const
 343 {
 344     wxSize sz = wxOwnerDrawnComboBox::DoGetBestSize();
 345
 346     // Scale control to match height of highest image.
 347     int h2 = m_usedImgSize.y + IMAGE_SPACING_CTRL_VERTICAL;
 348
 349     if ( h2 > sz.y )
 350         sz.y = h2;
 351
 352     CacheBestSize(sz);
 353     return sz;
 354 }
 355
 356 // ----------------------------------------------------------------------------
 357 // wxBitmapComboBox miscellaneous
 358 // ----------------------------------------------------------------------------
 359
 360 bool wxBitmapComboBox::SetFont(const wxFont& font)
 361 {
 362     bool res = wxOwnerDrawnComboBox::SetFont(font);
 363     m_fontHeight = GetCharHeight() + EXTRA_FONT_HEIGHT;
 364     return res;
 365 }
 366
 367 // ----------------------------------------------------------------------------
 368 // wxBitmapComboBox item drawing and measuring
 369 // ----------------------------------------------------------------------------
 370
 371 void wxBitmapComboBox::OnDrawBackground(wxDC& dc,
 372                                         const wxRect& rect,
 373                                         int item,
 374                                         int flags) const
 375 {
 376     if ( GetCustomPaintWidth() == 0 ||
 377          !(flags & wxODCB_PAINTING_SELECTED) ||
 378          item < 0 )
 379     {
 380         wxOwnerDrawnComboBox::OnDrawBackground(dc, rect, item, flags);
 381         return;
 382     }
 383
 384     //
 385     // Just paint simple selection background under where is text
 386     // (ie. emulate what MSW image choice does).
 387     //
 388
 389     int xPos = 0;  // Starting x of selection rectangle
 390     const int vSizeDec = 1;  // Vertical size reduction of selection rectangle edges
 391
 392     xPos = GetCustomPaintWidth() + 2;
 393
 394     wxCoord x, y;
 395     GetTextExtent(GetString(item), &x, &y, 0, 0);
 396
 397     dc.SetTextForeground(wxSystemSettings::GetColour(wxSYS_COLOUR_HIGHLIGHTTEXT));
 398
 399     wxColour selCol = wxSystemSettings::GetColour(wxSYS_COLOUR_HIGHLIGHT);
 400     dc.SetPen(selCol);
 401     dc.SetBrush(selCol);
 402     dc.DrawRectangle(rect.x+xPos,
 403                      rect.y+vSizeDec,
 404                      x + 4,
 405                      rect.height-(vSizeDec*2));
 406 }
 407
 408 void wxBitmapComboBox::OnDrawItem(wxDC& dc,
 409                                  const wxRect& rect,
 410                                  int item,
 411                                  int flags) const
 412 {
 413     wxString text;
 414     int imgAreaWidth = m_imgAreaWidth;
 415     bool drawText;
 416
 417     if ( imgAreaWidth == 0 )
 418     {
 419         wxOwnerDrawnComboBox::OnDrawItem(dc, rect, item, flags);
 420         return;
 421     }
 422
 423     if ( flags & wxODCB_PAINTING_CONTROL )
 424     {
 425         text = GetValue();
 426         if ( HasFlag(wxCB_READONLY) )
 427             drawText = true;
 428         else
 429             drawText = false;
 430     }
 431     else
 432     {
 433         text = GetString(item);
 434         drawText = true;
 435     }
 436
 437     const wxBitmap& bmp = *GetBitmapPtr(item);
 438     if ( bmp.Ok() )
 439     {
 440         wxCoord w = bmp.GetWidth();
 441         wxCoord h = bmp.GetHeight();
 442
 443         // Draw the image centered
 444         dc.DrawBitmap(bmp,
 445                       rect.x + (m_usedImgSize.x-w)/2 + IMAGE_SPACING_LEFT,
 446                       rect.y + (rect.height-h)/2,
 447                       true);
 448     }
 449
 450     if ( drawText )
 451         dc.DrawText(GetString(item),
 452                     rect.x + imgAreaWidth + 1,
 453                     rect.y + (rect.height-dc.GetCharHeight())/2);
 454 }
 455
 456 wxCoord wxBitmapComboBox::OnMeasureItem(size_t WXUNUSED(item)) const
 457 {
 458     int imgHeightArea = m_usedImgSize.y + 2;
 459     return imgHeightArea > m_fontHeight ? imgHeightArea : m_fontHeight;
 460 }
 461
 462 wxCoord wxBitmapComboBox::OnMeasureItemWidth(size_t item) const
 463 {
 464     wxCoord x, y;
 465     GetTextExtent(GetString(item), &x, &y, 0, 0);
 466     x += m_imgAreaWidth;
 467     return x;
 468 }
 469
 470 #endif // defined(wxGENERIC_BITMAPCOMBOBOX)
 471
 472 #endif // wxUSE_BITMAPCOMBOBOX