wxPython/wx/lib/buttonpanel.py

   1 # --------------------------------------------------------------------------- #
   2 # FANCYBUTTONPANEL Widget wxPython IMPLEMENTATION
   3 #
   4 # Original C++ Code From Eran. You Can Find It At:
   5 #
   6 # http://wxforum.shadonet.com/viewtopic.php?t=6619
   7 #
   8 # License: wxWidgets license
   9 #
  10 #
  11 # Python Code By:
  12 #
  13 # Andrea Gavana, @ 02 Oct 2006
  14 # Latest Revision: 17 Oct 2006, 17.00 GMT
  15 #
  16 #
  17 # For All Kind Of Problems, Requests Of Enhancements And Bug Reports, Please
  18 # Write To Me At:
  19 #
  20 # andrea.gavana@gmail.com
  21 # gavana@kpo.kz
  22 #
  23 # Or, Obviously, To The wxPython Mailing List!!!
  24 #
  25 #
  26 # End Of Comments
  27 # --------------------------------------------------------------------------- #
  28
  29 """
  30 With `ButtonPanel` class you have a panel with gradient coloring
  31 on it and with the possibility to place some buttons on it. Using a
  32 standard panel with normal wx.Buttons leads to an ugly result: the
  33 buttons are placed correctly on the panel - but with grey area around
  34 them.  Gradient coloring is kept behind the images - this was achieved
  35 due to the PNG format and the transparency of the bitmaps.
  36
  37 The image are functioning like a buttons and can be caught in your
  38 code using the usual self.Bind(wx.EVT_BUTTON, self.OnButton) method.
  39
  40 The control is generic, and support theming (well, I tested it under
  41 Windows with the three defauls themes: grey, blue, silver and the
  42 classic look).
  43
  44
  45 Usage
  46 -----
  47
  48 ButtonPanel supports 4 alignments: left, right, top, bottom, which have a
  49 different meaning and behavior wrt wx.Toolbar. The easiest thing is to try
  50 the demo to understand, but I'll try to explain how it works.
  51
  52 CASE 1: ButtonPanel has a main caption text
  53
  54 Left alignment means ButtonPanel is horizontal, with the text aligned to the
  55 left. When you shrink the demo frame, if there is not enough room for all
  56 the controls to be shown, the controls closest to the text are hidden;
  57
  58 Right alignment means ButtonPanel is horizontal, with the text aligned to the
  59 right. Item layout as above;
  60
  61 Top alignment means ButtonPanel is vertical, with the text aligned to the top.
  62 Item layout as above;
  63
  64 Bottom alignment means ButtonPanel is vertical, with the text aligned to the
  65 bottom. Item layout as above.
  66
  67
  68 CASE 2: ButtonPanel has *no* main caption text
  69 In this case, left and right alignment are the same (as top and bottom are the same),
  70 but the layout strategy changes: now if there is not enough room for all the controls
  71 to be shown, the last added items are hidden ("last" means on the far right for
  72 horizontal ButtonPanels and far bottom for vertical ButtonPanels).
  73
  74
  75 The following example shows a simple implementation that uses ButtonPanel
  76 inside a very simple frame::
  77
  78   class MyFrame(wx.Frame):
  79
  80       def __init__(self, parent, id=-1, title="ButtonPanel", pos=wx.DefaultPosition,
  81                    size=(800, 600), style=wx.DEFAULT_FRAME_STYLE):
  82
  83           wx.Frame.__init__(self, parent, id, title, pos, size, style)
  84
  85           mainPanel = wx.Panel(self, -1)
  86           self.logtext = wx.TextCtrl(mainPanel, -1, "", style=wx.TE_MULTILINE)
  87
  88           vSizer = wx.BoxSizer(wx.VERTICAL)
  89           mainPanel.SetSizer(vSizer)
  90
  91           alignment = BP_ALIGN_RIGHT
  92
  93           titleBar = ButtonPanel(mainPanel, -1, "A Simple Test & Demo")
  94
  95           btn1 = ButtonInfo(wx.NewId(), wx.Bitmap("png4.png", wx.BITMAP_TYPE_PNG))
  96           titleBar.AddButton(btn1)
  97           self.Bind(wx.EVT_BUTTON, self.OnButton, btn1)
  98
  99           btn2 = ButtonInfo(wx.NewId(), wx.Bitmap("png3.png", wx.BITMAP_TYPE_PNG))
 100           titleBar.AddButton(btn2)
 101           self.Bind(wx.EVT_BUTTON, self.OnButton, btn2)
 102
 103           btn3 = ButtonInfo(wx.NewId(), wx.Bitmap("png2.png", wx.BITMAP_TYPE_PNG))
 104           titleBar.AddButton(btn3)
 105           self.Bind(wx.EVT_BUTTON, self.OnButton, btn3)
 106
 107           btn4 = ButtonInfo(wx.NewId(), wx.Bitmap("png1.png", wx.BITMAP_TYPE_PNG))
 108           titleBar.AddButton(btn4)
 109           self.Bind(wx.EVT_BUTTON, self.OnButton, btn4)
 110
 111           vSizer.Add(titleBar, 0, wx.EXPAND)
 112           vSizer.Add((20, 20))
 113           vSizer.Add(self.logtext, 1, wx.EXPAND|wx.ALL, 5)
 114
 115           vSizer.Layout()
 116
 117   # our normal wxApp-derived class, as usual
 118
 119   app = wx.PySimpleApp()
 120
 121   frame = MyFrame(None)
 122   app.SetTopWindow(frame)
 123   frame.Show()
 124
 125   app.MainLoop()
 126
 127
 128 License And Version:
 129
 130 ButtonPanel Is Freeware And Distributed Under The wxPython License.
 131
 132 Latest Revision: Andrea Gavana @ 12 Oct 2006, 17.00 GMT
 133 Version 0.3.
 134
 135 """
 136
 137
 138 import wx
 139
 140 # Some constants to tune the BPArt class
 141 BP_BACKGROUND_COLOR = 0
 142 """ Background brush colour when no gradient shading exists. """
 143 BP_GRADIENT_COLOR_FROM = 1
 144 """ Starting gradient colour, used only when BP_USE_GRADIENT style is applied. """
 145 BP_GRADIENT_COLOR_TO = 2
 146 """ Ending gradient colour, used only when BP_USE_GRADIENT style is applied. """
 147 BP_BORDER_COLOR = 3
 148 """ Pen colour to paint the border of ButtonPanel. """
 149 BP_TEXT_COLOR = 4
 150 """ Main ButtonPanel caption colour. """
 151 BP_BUTTONTEXT_COLOR = 5
 152 """ Text colour for buttons with text. """
 153 BP_BUTTONTEXT_INACTIVE_COLOR = 6
 154 """ Text colour for inactive buttons with text. """
 155 BP_SELECTION_BRUSH_COLOR = 7
 156 """ Brush colour to be used when hovering or selecting a button. """
 157 BP_SELECTION_PEN_COLOR = 8
 158 """ Pen colour to be used when hovering or selecting a button. """
 159 BP_SEPARATOR_COLOR = 9
 160 """ Pen colour used to paint the separators. """
 161 BP_TEXT_FONT = 10
 162 """ Font of the ButtonPanel main caption. """
 163 BP_BUTTONTEXT_FONT = 11
 164 """ Text font for the buttons with text. """
 165
 166 BP_BUTTONTEXT_ALIGN_BOTTOM = 12
 167 """ Flag that indicates the image and text in buttons is stacked. """
 168 BP_BUTTONTEXT_ALIGN_RIGHT = 13
 169 """ Flag that indicates the text is shown alongside the image in buttons with text. """
 170
 171 BP_SEPARATOR_SIZE = 14
 172 """
 173 Separator size. NB: This is not the line width, but the sum of the space before
 174 and after the separator line plus the width of the line.
 175 """
 176 BP_MARGINS_SIZE = 15
 177 """
 178 Size of the left/right margins in ButtonPanel (top/bottom for vertically
 179 aligned ButtonPanels).
 180 """
 181 BP_BORDER_SIZE = 16
 182 """ Size of the border. """
 183 BP_PADDING_SIZE = 17
 184 """ Inter-tool separator size. """
 185
 186 # Caption Gradient Type
 187 BP_GRADIENT_NONE = 0
 188 """ No gradient shading should be used to paint the background. """
 189 BP_GRADIENT_VERTICAL = 1
 190 """ Vertical gradient shading should be used to paint the background. """
 191 BP_GRADIENT_HORIZONTAL = 2
 192 """ Horizontal gradient shading should be used to paint the background. """
 193
 194 # Flags for HitTest() method
 195 BP_HT_BUTTON = 200
 196 BP_HT_NONE = 201
 197
 198 # Alignment of buttons in the panel
 199 BP_ALIGN_RIGHT = 1
 200 BP_ALIGN_LEFT = 2
 201 BP_ALIGN_TOP = 4
 202 BP_ALIGN_BOTTOM = 8
 203
 204 # ButtonPanel styles
 205 BP_DEFAULT_STYLE = 1
 206 BP_USE_GRADIENT = 2
 207
 208 # Delay used to cancel the longHelp in the statusbar field
 209 _DELAY = 3000
 210
 211
 212 # Check for the new method in 2.7 (not present in 2.6.3.3)
 213 if wx.VERSION_STRING < "2.7":
 214     wx.Rect.Contains = lambda self, point: wx.Rect.Inside(self, point)
 215
 216
 217 def BrightenColour(color, factor):
 218     """ Bright the input colour by a factor."""
 219
 220     val = color.Red()*factor
 221     if val > 255:
 222         red = 255
 223     else:
 224         red = val
 225
 226     val = color.Green()*factor
 227     if val > 255:
 228         green = 255
 229     else:
 230         green = val
 231
 232     val = color.Blue()*factor
 233     if val > 255:
 234         blue = 255
 235     else:
 236         blue = val
 237
 238     return wx.Color(red, green, blue)
 239
 240
 241 def GrayOut(anImage):
 242     """
 243     Convert the given image (in place) to a grayed-out version,
 244     appropriate for a 'Disabled' appearance.
 245     """
 246
 247     factor = 0.7        # 0 < f < 1.  Higher Is Grayer
 248
 249     anImage = anImage.ConvertToImage()
 250     if anImage.HasAlpha():
 251         anImage.ConvertAlphaToMask(1)
 252
 253     if anImage.HasMask():
 254         maskColor = (anImage.GetMaskRed(), anImage.GetMaskGreen(), anImage.GetMaskBlue())
 255     else:
 256         maskColor = None
 257
 258     data = map(ord, list(anImage.GetData()))
 259
 260     for i in range(0, len(data), 3):
 261
 262         pixel = (data[i], data[i+1], data[i+2])
 263         pixel = MakeGray(pixel, factor, maskColor)
 264
 265         for x in range(3):
 266             data[i+x] = pixel[x]
 267
 268     anImage.SetData(''.join(map(chr, data)))
 269
 270     anImage = anImage.ConvertToBitmap()
 271
 272     return anImage
 273
 274
 275 def MakeGray((r,g,b), factor, maskColor):
 276     """
 277     Make a pixel grayed-out. If the pixel matches the maskColor, it won't be
 278     changed.
 279     """
 280
 281     if (r,g,b) != maskColor:
 282         return map(lambda x: int((230 - x) * factor) + x, (r,g,b))
 283     else:
 284         return (r,g,b)
 285
 286
 287 # ---------------------------------------------------------------------------- #
 288 # Class BPArt
 289 # Handles all the drawings for buttons, separators and text and allows the
 290 # programmer to set colours, sizes and gradient shadings for ButtonPanel
 291 # ---------------------------------------------------------------------------- #
 292
 293 class BPArt:
 294     """
 295     BPArt is an art provider class which does all of the drawing for ButtonPanel.
 296     This allows the library caller to customize the BPArt or to completely replace
 297     all drawing with custom BPArts.
 298     """
 299
 300     def __init__(self, parentStyle):
 301         """ Default class constructor. """
 302
 303         base_color = wx.SystemSettings_GetColour(wx.SYS_COLOUR_3DFACE)
 304
 305         self._background_brush = wx.Brush(base_color, wx.SOLID)
 306         self._gradient_color_to = wx.WHITE
 307         self._gradient_color_from = wx.SystemSettings_GetColour(wx.SYS_COLOUR_ACTIVECAPTION)
 308
 309         if parentStyle & BP_USE_GRADIENT:
 310             self._border_pen = wx.Pen(wx.WHITE, 3)
 311             self._caption_text_color = wx.WHITE
 312             self._buttontext_color = wx.Colour(70, 143, 255)
 313             self._separator_pen = wx.Pen(BrightenColour(self._gradient_color_from, 1.4))
 314             self._gradient_type = BP_GRADIENT_VERTICAL
 315         else:
 316             self._border_pen = wx.Pen(BrightenColour(base_color, 0.9), 3)
 317             self._caption_text_color = wx.BLACK
 318             self._buttontext_color = wx.SystemSettings_GetColour(wx.SYS_COLOUR_BTNTEXT)
 319             self._separator_pen = wx.Pen(BrightenColour(base_color, 0.9))
 320             self._gradient_type = BP_GRADIENT_NONE
 321
 322         self._buttontext_inactive_color = wx.SystemSettings_GetColour(wx.SYS_COLOUR_GRAYTEXT)
 323         self._selection_brush = wx.Brush(wx.Color(225, 225, 255))
 324         self._selection_pen = wx.Pen(wx.SystemSettings_GetColour(wx.SYS_COLOUR_ACTIVECAPTION))
 325
 326         sysfont = wx.SystemSettings.GetFont(wx.SYS_DEFAULT_GUI_FONT)
 327         self._caption_font = wx.Font(sysfont.GetPointSize(), wx.DEFAULT, wx.NORMAL, wx.BOLD,
 328                                      False, sysfont.GetFaceName())
 329         self._buttontext_font = wx.Font(sysfont.GetPointSize(), wx.DEFAULT, wx.NORMAL, wx.NORMAL,
 330                                         False, sysfont.GetFaceName())
 331
 332         self._separator_size = 7
 333         self._margins_size = wx.Size(6, 6)
 334         self._caption_border_size = 3
 335         self._padding_size = wx.Size(6, 6)
 336
 337
 338     def GetMetric(self, id):
 339         """ Returns sizes of customizable options. """
 340
 341         if id == BP_SEPARATOR_SIZE:
 342             return self._separator_size
 343         elif id == BP_MARGINS_SIZE:
 344             return self._margins_size
 345         elif id == BP_BORDER_SIZE:
 346             return self._caption_border_size
 347         elif id == BP_PADDING_SIZE:
 348             return self._padding_size
 349         else:
 350             raise "\nERROR: Invalid Metric Ordinal. "
 351
 352
 353     def SetMetric(self, id, new_val):
 354         """ Sets sizes for customizable options. """
 355
 356         if id == BP_SEPARATOR_SIZE:
 357             self._separator_size = new_val
 358         elif id == BP_MARGINS_SIZE:
 359             self._margins_size = new_val
 360         elif id == BP_BORDER_SIZE:
 361             self._caption_border_size = new_val
 362             self._border_pen.SetWidth(new_val)
 363         elif id == BP_PADDING_SIZE:
 364             self._padding_size = new_val
 365         else:
 366             raise "\nERROR: Invalid Metric Ordinal. "
 367
 368
 369     def GetColor(self, id):
 370         """ Returns colours of customizable options. """
 371
 372         if id == BP_BACKGROUND_COLOR:
 373             return self._background_brush.GetColour()
 374         elif id == BP_GRADIENT_COLOR_FROM:
 375             return self._gradient_color_from
 376         elif id == BP_GRADIENT_COLOR_TO:
 377             return self._gradient_color_to
 378         elif id == BP_BORDER_COLOR:
 379             return self._border_pen.GetColour()
 380         elif id == BP_TEXT_COLOR:
 381             return self._caption_text_color
 382         elif id == BP_BUTTONTEXT_COLOR:
 383             return self._buttontext_color
 384         elif id == BP_BUTTONTEXT_INACTIVE_COLOR:
 385             return self._buttontext_inactive_color
 386         elif id == BP_SELECTION_BRUSH_COLOR:
 387             return self._selection_brush.GetColour()
 388         elif id == BP_SELECTION_PEN_COLOR:
 389             return self._selection_pen.GetColour()
 390         elif id == BP_SEPARATOR_COLOR:
 391             return self._separator_pen.GetColour()
 392         else:
 393             raise "\nERROR: Invalid Colour Ordinal. "
 394
 395
 396     def SetColor(self, id, colour):
 397         """ Sets colours for customizable options. """
 398
 399         if id == BP_BACKGROUND_COLOR:
 400             self._background_brush.SetColour(colour)
 401         elif id == BP_GRADIENT_COLOR_FROM:
 402             self._gradient_color_from = colour
 403         elif id == BP_GRADIENT_COLOR_TO:
 404             self._gradient_color_to = colour
 405         elif id == BP_BORDER_COLOR:
 406             self._border_pen.SetColour(colour)
 407         elif id == BP_TEXT_COLOR:
 408             self._caption_text_color = colour
 409         elif id == BP_BUTTONTEXT_COLOR:
 410             self._buttontext_color = colour
 411         elif id == BP_BUTTONTEXT_INACTIVE_COLOR:
 412             self._buttontext_inactive_color = colour
 413         elif id == BP_SELECTION_BRUSH_COLOR:
 414             self._selection_brush.SetColour(colour)
 415         elif id == BP_SELECTION_PEN_COLOR:
 416             self._selection_pen.SetColour(colour)
 417         elif id == BP_SEPARATOR_COLOR:
 418             self._separator_pen.SetColour(colour)
 419         else:
 420             raise "\nERROR: Invalid Colour Ordinal. "
 421
 422
 423     GetColour = GetColor
 424     SetColour = SetColor
 425
 426
 427     def SetFont(self, id, font):
 428         """ Sets font for customizable options. """
 429
 430         if id == BP_TEXT_FONT:
 431             self._caption_font = font
 432         elif id == BP_BUTTONTEXT_FONT:
 433             self._buttontext_font = font
 434
 435
 436     def GetFont(self, id):
 437         """ Returns font of customizable options. """
 438
 439         if id == BP_TEXT_FONT:
 440             return self._caption_font
 441         elif id == BP_BUTTONTEXT_FONT:
 442             return self._buttontext_font
 443
 444         return wx.NoneFont
 445
 446
 447     def SetGradientType(self, gradient):
 448         """ Sets the gradient type for BPArt drawings. """
 449
 450         self._gradient_type = gradient
 451
 452
 453     def GetGradientType(self):
 454         """ Returns the gradient type for BPArt drawings. """
 455
 456         return self._gradient_type
 457
 458
 459     def DrawSeparator(self, dc, rect, isVertical):
 460         """ Draws a separator in ButtonPanel. """
 461
 462         dc.SetPen(self._separator_pen)
 463
 464         if isVertical:
 465             ystart = yend = rect.y + rect.height/2
 466             xstart = int(rect.x + 1.5*self._caption_border_size)
 467             xend = int(rect.x + rect.width - 1.5*self._caption_border_size)
 468             dc.DrawLine(xstart, ystart, xend, yend)
 469         else:
 470             xstart = xend = rect.x + rect.width/2
 471             ystart = int(rect.y + 1.5*self._caption_border_size)
 472             yend = int(rect.y + rect.height - 1.5*self._caption_border_size)
 473             dc.DrawLine(xstart, ystart, xend, yend)
 474
 475
 476     def DrawCaption(self, dc, rect, captionText):
 477         """ Draws the main caption text in ButtonPanel. """
 478
 479         textColour = self._caption_text_color
 480         textFont = self._caption_font
 481         padding = self._padding_size
 482
 483         dc.SetTextForeground(textColour)
 484         dc.SetFont(textFont)
 485
 486         dc.DrawText(captionText, rect.x + padding.x, rect.y+padding.y)
 487
 488
 489     def DrawButton(self, dc, rect, parentSize, buttonBitmap, isVertical,
 490                    buttonStatus, isToggled, textAlignment, text=""):
 491         """ Draws a button in ButtonPanel, together with its text (if any). """
 492
 493         bmpxsize, bmpysize = buttonBitmap.GetWidth(), buttonBitmap.GetHeight()
 494         dx = dy = focus = 0
 495
 496         borderw = self._caption_border_size
 497         padding = self._padding_size
 498
 499         buttonFont = self._buttontext_font
 500         dc.SetFont(buttonFont)
 501
 502         if isVertical:
 503
 504             rect = wx.Rect(borderw, rect.y, rect.width-2*borderw, rect.height)
 505
 506             if text != "":
 507
 508                 textW, textH = dc.GetTextExtent(text)
 509
 510                 if textAlignment == BP_BUTTONTEXT_ALIGN_RIGHT:
 511                     fullExtent = bmpxsize + padding.x/2 + textW
 512                     bmpypos = rect.y + (rect.height - bmpysize)/2
 513                     bmpxpos = rect.x + (rect.width - fullExtent)/2
 514                     textxpos = bmpxpos + padding.x/2 + bmpxsize
 515                     textypos = bmpypos + (bmpysize - textH)/2
 516                 else:
 517                     bmpxpos = rect.x + (rect.width - bmpxsize)/2
 518                     bmpypos = rect.y + padding.y
 519                     textxpos = rect.x + (rect.width - textW)/2
 520                     textypos = bmpypos + bmpysize + padding.y/2
 521             else:
 522                 bmpxpos = rect.x + (rect.width - bmpxsize)/2
 523                 bmpypos = rect.y + (rect.height - bmpysize)/2
 524
 525
 526         else:
 527
 528             rect = wx.Rect(rect.x, borderw, rect.width, rect.height-2*borderw)
 529
 530             if text != "":
 531
 532                 textW, textH = dc.GetTextExtent(text)
 533
 534                 if textAlignment == BP_BUTTONTEXT_ALIGN_RIGHT:
 535                     fullExtent = bmpxsize + padding.x/2 + textW
 536                     bmpypos = rect.y + (rect.height - bmpysize)/2
 537                     bmpxpos = rect.x + (rect.width - fullExtent)/2
 538                     textxpos = bmpxpos + padding.x/2 + bmpxsize
 539                     textypos = bmpypos + (bmpysize - textH)/2
 540                 else:
 541                     fullExtent = bmpysize + padding.y/2 + textH
 542                     bmpxpos = rect.x + (rect.width - bmpxsize)/2
 543                     bmpypos = rect.y + (rect.height - fullExtent)/2
 544                     textxpos = rect.x + (rect.width - textW)/2
 545                     textypos = bmpypos + bmpysize + padding.y/2
 546             else:
 547                 bmpxpos = rect.x + (rect.width - bmpxsize)/2
 548                 bmpypos = rect.y + (rect.height - bmpysize)/2
 549
 550         # Draw a button
 551         # [ Padding | Text | .. Buttons .. | Padding ]
 552
 553         if buttonStatus in ["Pressed", "Toggled", "Hover"]:
 554             dc.SetBrush(self._selection_brush)
 555             dc.SetPen(self._selection_pen)
 556             dc.DrawRoundedRectangleRect(rect, 4)
 557
 558         if buttonStatus == "Pressed" or isToggled:
 559             dx = dy = 1
 560
 561         dc.DrawBitmap(buttonBitmap, bmpxpos+dx, bmpypos+dy, True)
 562
 563         if text != "":
 564             isEnabled = buttonStatus != "Disabled"
 565             self.DrawLabel(dc, text, isEnabled, textxpos+dx, textypos+dy)
 566
 567
 568     def DrawLabel(self, dc, text, isEnabled, xpos, ypos):
 569         """ Draws the label for a button. """
 570
 571         if not isEnabled:
 572             dc.SetTextForeground(self._buttontext_inactive_color)
 573         else:
 574             dc.SetTextForeground(self._buttontext_color)
 575
 576         dc.DrawText(text, xpos, ypos)
 577
 578
 579     def DrawButtonPanel(self, dc, rect, style):
 580         """ Paint the ButtonPanel's background. """
 581
 582         if style & BP_USE_GRADIENT:
 583             # Draw gradient color in the backgroud of the panel
 584             self.FillGradientColor(dc, rect)
 585
 586         # Draw a rectangle around the panel
 587         backBrush = (style & BP_USE_GRADIENT and [wx.TRANSPARENT_BRUSH] or \
 588                      [self._background_brush])[0]
 589
 590         dc.SetBrush(backBrush)
 591         dc.SetPen(self._border_pen)
 592         dc.DrawRectangleRect(rect)
 593
 594
 595     def FillGradientColor(self, dc, rect):
 596         """ Gradient fill from colour 1 to colour 2 with top to bottom or left to right. """
 597
 598         if rect.height < 1 or rect.width < 1:
 599             return
 600
 601         isVertical = self._gradient_type == BP_GRADIENT_VERTICAL
 602         size = (isVertical and [rect.height] or [rect.width])[0]
 603         start = (isVertical and [rect.y] or [rect.x])[0]
 604
 605         # calculate gradient coefficients
 606
 607         col2 = self._gradient_color_from
 608         col1 = self._gradient_color_to
 609
 610         rf, gf, bf = 0, 0, 0
 611         rstep = float((col2.Red() - col1.Red()))/float(size)
 612         gstep = float((col2.Green() - col1.Green()))/float(size)
 613         bstep = float((col2.Blue() - col1.Blue()))/float(size)
 614
 615         for coord in xrange(start, start + size):
 616
 617             currCol = wx.Colour(col1.Red() + rf, col1.Green() + gf, col1.Blue() + bf)
 618             dc.SetBrush(wx.Brush(currCol, wx.SOLID))
 619             dc.SetPen(wx.Pen(currCol))
 620             if isVertical:
 621                 dc.DrawLine(rect.x, coord, rect.x + rect.width, coord)
 622             else:
 623                 dc.DrawLine(coord, rect.y, coord, rect.y + rect.width)
 624
 625             rf += rstep
 626             gf += gstep
 627             bf += bstep
 628
 629
 630 class StatusBarTimer(wx.Timer):
 631     """Timer used for deleting StatusBar long help after _DELAY seconds."""
 632
 633     def __init__(self, owner):
 634         """
 635         Default class constructor.
 636         For internal use: do not call it in your code!
 637         """
 638
 639         wx.Timer.__init__(self)
 640         self._owner = owner
 641
 642
 643     def Notify(self):
 644         """The timer has expired."""
 645
 646         self._owner.OnStatusBarTimer()
 647
 648
 649 class Control(wx.EvtHandler):
 650
 651     def __init__(self, parent, size=wx.Size(-1, -1)):
 652         """
 653         Default class constructor.
 654
 655         Base class for all pseudo controls
 656         parent = parent object
 657         size = (width, height)
 658         """
 659
 660         wx.EvtHandler.__init__(self)
 661
 662         self._parent = parent
 663         self._id = wx.NewId()
 664         self._size = size
 665         self._isshown = True
 666         self._focus = False
 667
 668
 669     def Show(self, show=True):
 670         """ Shows or hide the control. """
 671
 672         self._isshown = show
 673
 674
 675     def Hide(self):
 676         """ Hides the control. """
 677
 678         self.Show(False)
 679
 680
 681     def IsShown(self):
 682         """ Returns whether the control is shown or not. """
 683
 684         return self._isshown
 685
 686
 687     def GetId(self):
 688         """ Returns the control id. """
 689
 690         return self._id
 691
 692
 693     def GetBestSize(self):
 694         """ Returns the control best size. """
 695
 696         return self._size
 697
 698
 699     def Disable(self):
 700         """ Disables the control. """
 701
 702         self.Enable(False)
 703
 704
 705     def Enable(self, value=True):
 706         """ Enables or disables the control. """
 707
 708         self.disabled = not value
 709
 710
 711     def SetFocus(self, focus=True):
 712         """ Sets or kills the focus on the control. """
 713
 714         self._focus = focus
 715
 716
 717     def HasFocus(self):
 718         """ Returns whether the control has the focus or not. """
 719
 720         return self._focus
 721
 722
 723     def OnMouseEvent(self, x, y, event):
 724         pass
 725
 726     def Draw(self, rect):
 727         pass
 728
 729
 730
 731 class Sizer(object):
 732     """
 733     Sizer
 734
 735     This is a mix-in class to add pseudo support to a wx sizer.  Just create
 736     a new class that derives from this class and the wx sizer and intercepts
 737     any methods that add to the wx sizer.
 738     """
 739     def __init__(self):
 740         self.children = [] # list of child Pseudo Controls
 741
 742     # Sizer doesn't use the x1,y1,x2,y2 so allow it to
 743     # be called with or without the coordinates
 744     def Draw(self, dc, x1=0, y1=0, x2=0, y2=0):
 745         for item in self.children:
 746             # use sizer coordinates rather than
 747             # what is passed in
 748             c = item.GetUserData()
 749             c.Draw(dc, item.GetRect())
 750
 751     def GetBestSize(self):
 752         # this should be handled by the wx.Sizer based class
 753         return self.GetMinSize()
 754
 755
 756 # Pseudo BoxSizer
 757 class BoxSizer(Sizer, wx.BoxSizer):
 758     def __init__(self, orient=wx.HORIZONTAL):
 759         wx.BoxSizer.__init__(self, orient)
 760         Sizer.__init__(self)
 761
 762     #-------------------------------------------
 763     # sizer overrides (only called from Python)
 764     #-------------------------------------------
 765     # no support for user data if it's a pseudocontrol
 766     # since that is already used
 767     def Add(self, item, proportion=0, flag=0, border=0, userData=None):
 768         # check to see if it's a pseudo object or sizer
 769         if isinstance(item, Sizer):
 770             szitem = wx.BoxSizer.Add(self, item, proportion, flag, border, item)
 771             self.children.append(szitem)
 772         elif isinstance(item, Control): # Control should be what ever class your controls come from
 773             sz = item.GetBestSize()
 774             # add a spacer to track this object
 775             szitem = wx.BoxSizer.Add(self, sz, proportion, flag, border, item)
 776             self.children.append(szitem)
 777         else:
 778             wx.BoxSizer.Add(self, item, proportion, flag, border, userData)
 779
 780     def Prepend(self, item, proportion=0, flag=0, border=0, userData=None):
 781         # check to see if it's a pseudo object or sizer
 782         if isinstance(item, Sizer):
 783             szitem = wx.BoxSizer.Prepend(self, item, proportion, flag, border, item)
 784             self.children.append(szitem)
 785         elif isinstance(item, Control): # Control should be what ever class your controls come from
 786             sz = item.GetBestSize()
 787             # add a spacer to track this object
 788             szitem = wx.BoxSizer.Prepend(self, sz, proportion, flag, border, item)
 789             self.children.insert(0,szitem)
 790         else:
 791             wx.BoxSizer.Prepend(self, item, proportion, flag, border, userData)
 792
 793     def Insert(self, before, item, proportion=0, flag=0, border=0, userData=None, realIndex=None):
 794         # check to see if it's a pseudo object or sizer
 795         if isinstance(item, Sizer):
 796             szitem = wx.BoxSizer.Insert(self, before, item, proportion, flag, border, item)
 797             self.children.append(szitem)
 798         elif isinstance(item, Control): # Control should be what ever class your controls come from
 799             sz = item.GetBestSize()
 800             # add a spacer to track this object
 801             szitem = wx.BoxSizer.Insert(self, before, sz, proportion, flag, border, item)
 802             if realIndex is not None:
 803                 self.children.insert(realIndex,szitem)
 804             else:
 805                 self.children.insert(before,szitem)
 806
 807         else:
 808             wx.BoxSizer.Insert(self, before, item, proportion, flag, border, userData)
 809
 810
 811     def Remove(self, indx, pop=-1):
 812
 813         if pop >= 0:
 814             self.children.pop(pop)
 815
 816         wx.BoxSizer.Remove(self, indx)
 817
 818
 819     def Layout(self):
 820
 821         for ii, child in enumerate(self.GetChildren()):
 822             item = child.GetUserData()
 823             if item and child.IsShown():
 824                 self.SetItemMinSize(ii, *item.GetBestSize())
 825
 826         wx.BoxSizer.Layout(self)
 827
 828
 829     def Show(self, item, show=True):
 830
 831         child = self.GetChildren()[item]
 832         if child and child.GetUserData():
 833             child.GetUserData().Show(show)
 834
 835         wx.BoxSizer.Show(self, item, show)
 836
 837
 838 # ---------------------------------------------------------------------------- #
 839 # Class Separator
 840 # This class holds all the information to size and draw a separator inside
 841 # ButtonPanel
 842 # ---------------------------------------------------------------------------- #
 843
 844 class Separator(Control):
 845
 846     def __init__(self, parent):
 847         """ Default class constructor. """
 848
 849         self._isshown = True
 850         self._parent = parent
 851         Control.__init__(self, parent)
 852
 853
 854     def GetBestSize(self):
 855         """ Returns the separator best size. """
 856
 857         # 10 is completely arbitrary, but it works anyhow
 858         if self._parent.IsVertical():
 859             return wx.Size(10, self._parent._art.GetMetric(BP_SEPARATOR_SIZE))
 860         else:
 861             return wx.Size(self._parent._art.GetMetric(BP_SEPARATOR_SIZE), 10)
 862
 863
 864     def Draw(self, dc, rect):
 865         """ Draws the separator. Actually the drawing is done in BPArt. """
 866
 867         if not self.IsShown():
 868             return
 869
 870         isVertical = self._parent.IsVertical()
 871         self._parent._art.DrawSeparator(dc, rect, isVertical)
 872
 873
 874 # ---------------------------------------------------------------------------- #
 875 # Class ButtonPanelText
 876 # This class is used to hold data about the main caption in ButtonPanel
 877 # ---------------------------------------------------------------------------- #
 878
 879 class ButtonPanelText(Control):
 880
 881     def __init__(self, parent, text=""):
 882         """ Default class constructor. """
 883
 884         self._text = text
 885         self._isshown = True
 886         self._parent = parent
 887
 888         Control.__init__(self, parent)
 889
 890
 891     def GetText(self):
 892         """ Returns the caption text. """
 893
 894         return self._text
 895
 896
 897     def SetText(self, text=""):
 898         """ Sets the caption text. """
 899
 900         self._text = text
 901
 902
 903     def CreateDC(self):
 904         """ Convenience function to create a DC. """
 905
 906         dc = wx.ClientDC(self._parent)
 907         textFont = self._parent._art.GetFont(BP_TEXT_FONT)
 908         dc.SetFont(textFont)
 909
 910         return dc
 911
 912
 913     def GetBestSize(self):
 914         """ Returns the best size for the main caption in ButtonPanel. """
 915
 916         if self._text == "":
 917             return wx.Size(0, 0)
 918
 919         dc = self.CreateDC()
 920         rect = self._parent.GetClientRect()
 921
 922         tw, th = dc.GetTextExtent(self._text)
 923         padding = self._parent._art.GetMetric(BP_PADDING_SIZE)
 924         self._size = wx.Size(tw+2*padding.x, th+2*padding.y)
 925
 926         return self._size
 927
 928
 929     def Draw(self, dc, rect):
 930         """ Draws the main caption. Actually the drawing is done in BPArt. """
 931
 932         if not self.IsShown():
 933             return
 934
 935         captionText = self.GetText()
 936         self._parent._art.DrawCaption(dc, rect, captionText)
 937
 938
 939 # -- ButtonInfo class implementation ----------------------------------------
 940 # This class holds information about every button that is added to
 941 # ButtonPanel.  It is an auxiliary class that you should use
 942 # every time you add a button.
 943
 944 class ButtonInfo(Control):
 945
 946     def __init__(self, parent, id=wx.ID_ANY, bmp=wx.NullBitmap,
 947                  status="Normal", text="", kind=wx.ITEM_NORMAL,
 948                  shortHelp="", longHelp=""):
 949         """
 950         Default class constructor.
 951
 952         Parameters:
 953         - parent: the parent window (ButtonPanel);
 954         - id: the button id;
 955         - bmp: the associated bitmap;
 956         - status: button status (pressed, hovered, normal).
 957         - text: text to be displayed either below of to the right of the button
 958         - kind: button kind, may be wx.ITEM_NORMAL for standard buttons or
 959           wx.ITEM_CHECK for toggle buttons;
 960         - shortHelp: a short help to be shown in the button tooltip;
 961         - longHelp: this string is shown in the statusbar (if any) of the parent
 962           frame when the mouse pointer is inside the button.
 963         """
 964
 965         if id == wx.ID_ANY:
 966             id = wx.NewId()
 967
 968         self._status = status
 969         self._rect = wx.Rect()
 970         self._text = text
 971         self._kind = kind
 972         self._toggle = False
 973         self._textAlignment = BP_BUTTONTEXT_ALIGN_BOTTOM
 974         self._shortHelp = shortHelp
 975         self._longHelp = longHelp
 976
 977         disabledbmp = GrayOut(bmp)
 978
 979         self._bitmaps = {"Normal": bmp, "Toggled": None, "Disabled": disabledbmp,
 980                          "Hover": None, "Pressed": None}
 981
 982         Control.__init__(self, parent)
 983
 984
 985     def GetBestSize(self):
 986         """ Returns the best size for the button. """
 987
 988         xsize = self.GetBitmap().GetWidth()
 989         ysize = self.GetBitmap().GetHeight()
 990
 991         if self.HasText():
 992             # We have text in the button
 993             dc = wx.ClientDC(self._parent)
 994             normalFont = self._parent._art.GetFont(BP_BUTTONTEXT_FONT)
 995             dc.SetFont(normalFont)
 996             tw, th = dc.GetTextExtent(self.GetText())
 997
 998             if self.GetTextAlignment() == BP_BUTTONTEXT_ALIGN_BOTTOM:
 999                 xsize = max(xsize, tw)
1000                 ysize = ysize + th
1001             else:
1002                 xsize = xsize + tw
1003                 ysize = max(ysize, th)
1004
1005         border = self._parent._art.GetMetric(BP_BORDER_SIZE)
1006         padding = self._parent._art.GetMetric(BP_PADDING_SIZE)
1007
1008         if self._parent.IsVertical():
1009             xsize = xsize + 2*border
1010         else:
1011             ysize = ysize + 2*border
1012
1013         self._size = wx.Size(xsize+2*padding.x, ysize+2*padding.y)
1014
1015         return self._size
1016
1017
1018     def Draw(self, dc, rect):
1019         """ Draws the button on ButtonPanel. Actually the drawing is done in BPArt. """
1020
1021         if not self.IsShown():
1022             return
1023
1024         buttonBitmap = self.GetBitmap()
1025         isVertical = self._parent.IsVertical()
1026         text = self.GetText()
1027         parentSize = self._parent.GetSize()[not isVertical]
1028         buttonStatus = self.GetStatus()
1029         isToggled = self.GetToggled()
1030         textAlignment = self.GetTextAlignment()
1031
1032         self._parent._art.DrawButton(dc, rect, parentSize, buttonBitmap, isVertical,
1033                                      buttonStatus, isToggled, textAlignment, text)
1034
1035         self.SetRect(rect)
1036
1037
1038     def CheckRefresh(self, status):
1039         """ Checks whether a ButtonPanel repaint is needed or not. Convenience function. """
1040
1041         if status == self._status:
1042             self._parent.RefreshRect(self.GetRect())
1043
1044
1045     def SetBitmap(self, bmp, status="Normal"):
1046         """ Sets the associated bitmap. """
1047
1048         self._bitmaps[status] = bmp
1049         self.CheckRefresh(status)
1050
1051
1052     def GetBitmap(self, status=None):
1053         """ Returns the associated bitmap. """
1054
1055         if status is None:
1056             status = self._status
1057
1058         if not self.IsEnabled():
1059             status = "Disabled"
1060
1061         if self._bitmaps[status] is None:
1062             return self._bitmaps["Normal"]
1063
1064         return self._bitmaps[status]
1065
1066
1067     def GetRect(self):
1068         """ Returns the button rect. """
1069
1070         return self._rect
1071
1072
1073     def GetStatus(self):
1074         """ Returns the button status. """
1075
1076         return self._status
1077
1078
1079     def GetId(self):
1080         """ Returns the button id. """
1081
1082         return self._id
1083
1084
1085     def SetRect(self, rect):
1086         """ Sets the button rect. """
1087
1088         self._rect = rect
1089
1090
1091     def SetStatus(self, status):
1092         """ Sets the button status. """
1093
1094         if status == self._status:
1095             return
1096
1097         if self.GetToggled() and status == "Normal":
1098             status = "Toggled"
1099
1100         self._status = status
1101         self._parent.RefreshRect(self.GetRect())
1102
1103
1104     def GetTextAlignment(self):
1105         """ Returns the text alignment in the button (bottom or right). """
1106
1107         return self._textAlignment
1108
1109
1110     def SetTextAlignment(self, alignment):
1111         """ Sets the text alignment in the button (bottom or right). """
1112
1113         if alignment == self._textAlignment:
1114             return
1115
1116         self._textAlignment = alignment
1117
1118
1119     def GetToggled(self):
1120         """ Returns whether a wx.ITEM_CHECK button is toggled or not. """
1121
1122         if self._kind == wx.ITEM_NORMAL:
1123             return False
1124
1125         return self._toggle
1126
1127
1128     def SetToggled(self, toggle=True):
1129         """ Sets a wx.ITEM_CHECK button toggled/not toggled. """
1130
1131         if self._kind == wx.ITEM_NORMAL:
1132             return
1133
1134         self._toggle = toggle
1135
1136
1137     def SetId(self, id):
1138         """ Sets the button id. """
1139
1140         self._id = id
1141
1142
1143     def AddStatus(self, name="Custom", bmp=wx.NullBitmap):
1144         """
1145         Add a programmer-defined status in addition to the 5 default status:
1146         - Normal;
1147         - Disabled;
1148         - Hover;
1149         - Pressed;
1150         - Toggled.
1151         """
1152
1153         self._bitmaps.update({name: bmp})
1154
1155
1156     def Enable(self, enable=True):
1157
1158         if enable:
1159             self._status = "Normal"
1160         else:
1161             self._status = "Disabled"
1162
1163
1164     def IsEnabled(self):
1165
1166         return self._status != "Disabled"
1167
1168
1169     def SetText(self, text=""):
1170         """ Sets the text of the button. """
1171
1172         self._text = text
1173
1174
1175     def GetText(self):
1176         """ Returns the text associated to the button. """
1177
1178         return self._text
1179
1180
1181     def HasText(self):
1182         """ Returns whether the button has text or not. """
1183
1184         return self._text != ""
1185
1186
1187     def SetKind(self, kind=wx.ITEM_NORMAL):
1188         """ Sets the button type (standard or toggle). """
1189
1190         self._kind = kind
1191
1192
1193     def GetKind(self):
1194         """ Returns the button type (standard or toggle). """
1195
1196         return self._kind
1197
1198
1199     def SetShortHelp(self, help=""):
1200         """ Sets the help string to be shown in a tootip. """
1201
1202         self._shortHelp = help
1203
1204
1205     def GetShortHelp(self):
1206         """ Returns the help string shown in a tootip. """
1207
1208         return self._shortHelp
1209
1210
1211     def SetLongHelp(self, help=""):
1212         """ Sets the help string to be shown in the statusbar. """
1213
1214         self._longHelp = help
1215
1216
1217     def GetLongHelp(self):
1218         """ Returns the help string shown in the statusbar. """
1219
1220         return self._longHelp
1221
1222
1223     Bitmap = property(GetBitmap, SetBitmap)
1224     Id     = property(GetId, SetId)
1225     Rect   = property(GetRect, SetRect)
1226     Status = property(GetStatus, SetStatus)
1227
1228
1229 # -- ButtonPanel class implementation ----------------------------------
1230 # This is the main class.
1231
1232 class ButtonPanel(wx.PyPanel):
1233
1234     def __init__(self, parent, id=wx.ID_ANY, text="", style=BP_DEFAULT_STYLE,
1235                  alignment=BP_ALIGN_LEFT, name="buttonPanel"):
1236         """
1237         Default class constructor.
1238
1239         - parent: parent window
1240         - id: window ID
1241         - text: text to draw
1242         - style: window style
1243         - alignment: alignment of buttons (left or right)
1244         - name: window class name
1245         """
1246
1247         wx.PyPanel.__init__(self, parent, id, wx.DefaultPosition, wx.DefaultSize,
1248                             wx.NO_BORDER, name=name)
1249
1250         self._vButtons = []
1251         self._vSeparators = []
1252
1253         self._nStyle = style
1254         self._alignment = alignment
1255         self._statusTimer = None
1256         self._useHelp = True
1257         self._freezeCount = 0
1258         self._currentButton = -1
1259         self._haveTip = False
1260
1261         self._art = BPArt(style)
1262
1263         self._controlCreated = False
1264
1265         direction = (self.IsVertical() and [wx.VERTICAL] or [wx.HORIZONTAL])[0]
1266         self._mainsizer = BoxSizer(direction)
1267         self.SetSizer(self._mainsizer)
1268
1269         margins = self._art.GetMetric(BP_MARGINS_SIZE)
1270
1271         # First spacer to create some room before the first text/button/control
1272         self._mainsizer.Add((margins.x, margins.y), 0)
1273
1274         # Last spacer to create some room before the last text/button/control
1275         self._mainsizer.Add((margins.x, margins.y), 0)
1276
1277         self.Bind(wx.EVT_SIZE, self.OnSize)
1278         self.Bind(wx.EVT_PAINT, self.OnPaint)
1279         self.Bind(wx.EVT_ERASE_BACKGROUND, self.OnEraseBackground)
1280         self.Bind(wx.EVT_MOTION, self.OnMouseMove)
1281         self.Bind(wx.EVT_LEFT_DOWN, self.OnLeftDown)
1282         self.Bind(wx.EVT_LEFT_UP, self.OnLeftUp)
1283         self.Bind(wx.EVT_LEAVE_WINDOW, self.OnMouseLeave)
1284         self.Bind(wx.EVT_ENTER_WINDOW, self.OnMouseEnterWindow)
1285
1286         self.SetBarText(text)
1287         self.LayoutItems()
1288
1289
1290     def SetBarText(self, text):
1291         """ Sets the main caption text (leave text="" for no text). """
1292
1293         self.Freeze()
1294
1295         text = text.strip()
1296
1297         if self._controlCreated:
1298             self.RemoveText()
1299
1300         self._text = ButtonPanelText(self, text)
1301         lenChildren = len(self._mainsizer.GetChildren())
1302
1303         if text == "":
1304             # Even if we have no text, we insert it an empty spacer anyway
1305             # it is easier to handle if you have to recreate the sizer after.
1306             if self.IsStandard():
1307                 self._mainsizer.Insert(1, self._text, 0, wx.ALIGN_CENTER,
1308                                        userData=self._text, realIndex=0)
1309             else:
1310                 self._mainsizer.Insert(lenChildren-1, self._text, 0, wx.ALIGN_CENTER,
1311                                        userData=self._text, realIndex=lenChildren)
1312
1313             return
1314
1315         # We have text, so insert the text and an expandable spacer
1316         # alongside it. "Standard" ButtonPanel are left or top aligned.
1317         if self.IsStandard():
1318             self._mainsizer.Insert(1, self._text, 0, wx.ALIGN_CENTER,
1319                                     userData=self._text, realIndex=0)
1320             self._mainsizer.Insert(2, (0, 0), 1, wx.EXPAND)
1321
1322         else:
1323             self._mainsizer.Insert(lenChildren-1, self._text, 0, wx.ALIGN_CENTER,
1324                                    userData=self._text, realIndex=lenChildren)
1325             self._mainsizer.Insert(lenChildren-1, (0, 0), 1, wx.EXPAND)
1326
1327
1328     def RemoveText(self):
1329         """ Removes the main caption text. """
1330
1331         lenChildren = len(self._mainsizer.GetChildren())
1332         lenCustom = len(self._vButtons) + len(self._vSeparators) + 1
1333
1334         if self.IsStandard():
1335             # Detach the text
1336             self._mainsizer.Remove(1, 0)
1337             if self.HasBarText():
1338                 # Detach the expandable spacer
1339                 self._mainsizer.Remove(1, -1)
1340         else:
1341             # Detach the text
1342             self._mainsizer.Remove(lenChildren-2, lenCustom-1)
1343             if self.HasBarText():
1344                 # Detach the expandable spacer
1345                 self._mainsizer.Remove(lenChildren-3, -1)
1346
1347
1348     def GetBarText(self):
1349         """ Returns the main caption text. """
1350
1351         return self._text.GetText()
1352
1353
1354     def HasBarText(self):
1355         """ Returns whether ButtonPanel has a main caption text or not. """
1356
1357         return hasattr(self, "_text") and self._text.GetText() != ""
1358
1359
1360     def AddButton(self, btnInfo):
1361         """
1362         Adds a button to ButtonPanel. Remember to pass a ButtonInfo instance to
1363         this method. See the demo for details.
1364         """
1365
1366         lenChildren = len(self._mainsizer.GetChildren())
1367         self._mainsizer.Insert(lenChildren-1, btnInfo, 0, wx.ALIGN_CENTER|wx.EXPAND, userData=btnInfo)
1368
1369         self._vButtons.append(btnInfo)
1370
1371
1372     def AddSpacer(self, size=(0, 0), proportion=1, flag=wx.EXPAND):
1373         """ Adds a spacer (stretchable or fixed-size) to ButtonPanel. """
1374
1375         lenChildren = len(self._mainsizer.GetChildren())
1376         self._mainsizer.Insert(lenChildren-1, size, proportion, flag)
1377
1378
1379     def AddControl(self, control, proportion=0, flag=wx.ALIGN_CENTER|wx.ALL, border=None):
1380         """ Adds a wxPython control to ButtonPanel. """
1381
1382         lenChildren = len(self._mainsizer.GetChildren())
1383
1384         if border is None:
1385             border = self._art.GetMetric(BP_PADDING_SIZE)
1386             border = max(border.x, border.y)
1387
1388         self._mainsizer.Insert(lenChildren-1, control, proportion, flag, border)
1389
1390
1391     def AddSeparator(self):
1392         """ Adds a separator line to ButtonPanel. """
1393
1394         lenChildren = len(self._mainsizer.GetChildren())
1395         separator = Separator(self)
1396
1397         self._mainsizer.Insert(lenChildren-1, separator, 0, wx.EXPAND)
1398         self._vSeparators.append(separator)
1399
1400
1401     def RemoveAllButtons(self):
1402         """ Remove all the buttons from ButtonPanel. """
1403
1404         self._vButtons = []
1405
1406
1407     def RemoveAllSeparators(self):
1408         """ Remove all the separators from ButtonPanel. """
1409
1410         self._vSeparators = []
1411
1412
1413     def GetAlignment(self):
1414         """ Returns the button alignment (left, right, top, bottom). """
1415
1416         return self._alignment
1417
1418
1419     def SetAlignment(self, alignment):
1420         """ Sets the button alignment (left, right, top, bottom). """
1421
1422         if alignment == self._alignment:
1423             return
1424
1425         self.Freeze()
1426
1427         text = self.GetBarText()
1428
1429         # Remove the text in any case
1430         self.RemoveText()
1431
1432         # Remove the first and last spacers
1433         self._mainsizer.Remove(0, -1)
1434         self._mainsizer.Remove(len(self._mainsizer.GetChildren())-1, -1)
1435
1436         self._alignment = alignment
1437
1438         # Recreate the sizer accordingly to the new alignment
1439         self.ReCreateSizer(text)
1440
1441
1442     def IsVertical(self):
1443         """ Returns whether ButtonPanel is vertically aligned or not. """
1444
1445         return self._alignment not in [BP_ALIGN_RIGHT, BP_ALIGN_LEFT]
1446
1447
1448     def IsStandard(self):
1449         """ Returns whether ButtonPanel is aligned "Standard" (left/top) or not. """
1450
1451         return self._alignment in [BP_ALIGN_LEFT, BP_ALIGN_TOP]
1452
1453
1454     def DoLayout(self):
1455         """
1456         Do the Layout for ButtonPanel.
1457         NB: Call this method every time you make a modification to the layout
1458         or to the customizable sizes of the pseudo controls.
1459         """
1460
1461         margins = self._art.GetMetric(BP_MARGINS_SIZE)
1462         lenChildren = len(self._mainsizer.GetChildren())
1463
1464         self._mainsizer.SetItemMinSize(0, (margins.x, margins.y))
1465         self._mainsizer.SetItemMinSize(lenChildren-1, (margins.x, margins.y))
1466
1467         self._controlCreated = True
1468         self.LayoutItems()
1469
1470         # *VERY* WEIRD: the sizer seems not to respond to any layout until I
1471         # change the ButtonPanel size and restore it back
1472         size = self.GetSize()
1473         self.SetSize((size.x+1, size.y+1))
1474         self.SetSize((size.x, size.y))
1475
1476         if self.IsFrozen():
1477             self.Thaw()
1478
1479
1480     def ReCreateSizer(self, text):
1481         """ Recreates the ButtonPanel sizer accordingly to the alignment specified. """
1482
1483         children = self._mainsizer.GetChildren()
1484         self.RemoveAllButtons()
1485         self.RemoveAllSeparators()
1486
1487         # Create a new sizer depending on the alignment chosen
1488         direction = (self.IsVertical() and [wx.VERTICAL] or [wx.HORIZONTAL])[0]
1489         self._mainsizer = BoxSizer(direction)
1490
1491         margins = self._art.GetMetric(BP_MARGINS_SIZE)
1492         # First spacer to create some room before the first text/button/control
1493         self._mainsizer.Add((margins.x, margins.y), 0)
1494
1495         # Last spacer to create some room before the last text/button/control
1496         self._mainsizer.Add((margins.x, margins.y), 0)
1497
1498         # This is needed otherwise SetBarText goes mad
1499         self._controlCreated = False
1500
1501         for child in children:
1502             userData = child.GetUserData()
1503             if userData:
1504                 if isinstance(userData, ButtonInfo):
1505                     # It is a ButtonInfo, can't be anything else
1506                     self.AddButton(child.GetUserData())
1507                 elif isinstance(userData, Separator):
1508                     self.AddSeparator()
1509
1510             else:
1511                 if child.IsSpacer():
1512                     # This is a spacer, expandable or not
1513                     self.AddSpacer(child.GetSize(), child.GetProportion(),
1514                                    child.GetFlag())
1515                 else:
1516                     # This is a wxPython control
1517                     self.AddControl(child.GetWindow(), child.GetProportion(),
1518                                     child.GetFlag(), child.GetBorder())
1519
1520         self.SetSizer(self._mainsizer)
1521
1522         # Now add the text. It doesn't matter if there is no text
1523         self.SetBarText(text)
1524
1525         self.DoLayout()
1526
1527         self.Thaw()
1528
1529
1530     def DoGetBestSize(self):
1531         """ Returns the best size of ButtonPanel. """
1532
1533         w = h = btnWidth = btnHeight = 0
1534         isVertical = self.IsVertical()
1535
1536         padding = self._art.GetMetric(BP_PADDING_SIZE)
1537         border = self._art.GetMetric(BP_BORDER_SIZE)
1538         margins = self._art.GetMetric(BP_MARGINS_SIZE)
1539         separator_size = self._art.GetMetric(BP_SEPARATOR_SIZE)
1540
1541         # Add the space required for the main caption
1542         if self.HasBarText():
1543             w, h = self._text.GetBestSize()
1544             if isVertical:
1545                 h += padding.y
1546             else:
1547                 w += padding.x
1548         else:
1549             w = h = border
1550
1551         # Add the button's sizes
1552         for btn in self._vButtons:
1553
1554             bw, bh = btn.GetBestSize()
1555             btnWidth = max(btnWidth, bw)
1556             btnHeight = max(btnHeight, bh)
1557
1558             if isVertical:
1559                 w = max(w, btnWidth)
1560                 h += bh
1561             else:
1562                 h = max(h, btnHeight)
1563                 w += bw
1564
1565         # Add the control's sizes
1566         for control in self.GetControls():
1567             cw, ch = control.GetSize()
1568             if isVertical:
1569                 h += ch
1570                 w = max(w, cw)
1571             else:
1572                 w += cw
1573                 h = max(h, ch)
1574
1575         # Add the separator's sizes and the 2 SizerItems at the beginning
1576         # and at the end
1577         if self.IsVertical():
1578             h += 2*margins.y + len(self._vSeparators)*separator_size
1579         else:
1580             w += 2*margins.x + len(self._vSeparators)*separator_size
1581
1582         return wx.Size(w, h)
1583
1584
1585     def OnPaint(self, event):
1586         """ Handles the wx.EVT_PAINT event for ButtonPanel. """
1587
1588         dc = wx.BufferedPaintDC(self)
1589         rect = self.GetClientRect()
1590
1591         self._art.DrawButtonPanel(dc, rect, self._nStyle)
1592         self._mainsizer.Draw(dc)
1593
1594
1595     def OnEraseBackground(self, event):
1596         """ Handles the wx.EVT_ERASE_BACKGROUND event for ButtonPanel (does nothing). """
1597
1598         pass
1599
1600
1601     def OnSize(self, event):
1602         """ Handles the wx.EVT_SIZE event for ButtonPanel. """
1603
1604         # NOTE: It seems like LayoutItems number of calls can be optimized in some way.
1605         # Currently every DoLayout (or every parent Layout()) calls about 3 times
1606         # the LayoutItems method. Any idea on how to improve it?
1607         self.LayoutItems()
1608         self.Refresh()
1609
1610         event.Skip()
1611
1612
1613     def LayoutItems(self):
1614         """
1615         Layout the items using a different algorithm depending on the existance
1616         of the main caption.
1617         """
1618
1619         nonspacers, allchildren = self.GetNonFlexibleChildren()
1620
1621         if self.HasBarText():
1622             self.FlexibleLayout(nonspacers, allchildren)
1623         else:
1624             self.SizeLayout(nonspacers, allchildren)
1625
1626         self._mainsizer.Layout()
1627
1628
1629     def SizeLayout(self, nonspacers, children):
1630         """ Layout the items when no main caption exists. """
1631
1632         size = self.GetSize()
1633         isVertical = self.IsVertical()
1634
1635         corner = 0
1636         indx1 = len(nonspacers)
1637
1638         for item in nonspacers:
1639             corner += self.GetItemSize(item, isVertical)
1640             if corner > size[isVertical]:
1641                 indx1 = nonspacers.index(item)
1642                 break
1643
1644         # Leave out the last spacer, it has to be there always
1645         for ii in xrange(len(nonspacers)-1):
1646             indx = children.index(nonspacers[ii])
1647             self._mainsizer.Show(indx, ii < indx1)
1648
1649
1650     def GetItemSize(self, item, isVertical):
1651         """ Returns the size of an item in the main ButtonPanel sizer. """
1652
1653         if item.GetUserData():
1654             return item.GetUserData().GetBestSize()[isVertical]
1655         else:
1656             return item.GetSize()[isVertical]
1657
1658
1659     def FlexibleLayout(self, nonspacers, allchildren):
1660         """ Layout the items when the main caption exists. """
1661
1662         if len(nonspacers) < 2:
1663             return
1664
1665         isVertical = self.IsVertical()
1666         isStandard = self.IsStandard()
1667
1668         size = self.GetSize()[isVertical]
1669         padding = self._art.GetMetric(BP_PADDING_SIZE)
1670
1671         fixed = (isStandard and [nonspacers[1]] or [nonspacers[-2]])[0]
1672
1673         if isStandard:
1674             nonspacers.reverse()
1675             leftendx = fixed.GetSize()[isVertical] + padding.x
1676         else:
1677             rightstartx = size - fixed.GetSize()[isVertical]
1678             size = 0
1679
1680         count = lennonspacers = len(nonspacers)
1681
1682         for item in nonspacers:
1683             if isStandard:
1684                 size -= self.GetItemSize(item, isVertical)
1685                 if size < leftendx:
1686                     break
1687             else:
1688                 size += self.GetItemSize(item, isVertical)
1689                 if size > rightstartx:
1690                     break
1691
1692             count = count - 1
1693
1694         nonspacers.reverse()
1695
1696         for jj in xrange(2, lennonspacers):
1697             indx = allchildren.index(nonspacers[jj])
1698             self._mainsizer.Show(indx, jj >= count)
1699
1700
1701     def GetNonFlexibleChildren(self):
1702         """
1703         Returns all the ButtonPanel main sizer's children that are not
1704         flexible spacers.
1705         """
1706
1707         children1 = []
1708         children2 = self._mainsizer.GetChildren()
1709
1710         for child in children2:
1711             if child.IsSpacer():
1712                 if child.GetUserData() or child.GetProportion() == 0:
1713                     children1.append(child)
1714             else:
1715                 children1.append(child)
1716
1717         return children1, children2
1718
1719
1720     def GetControls(self):
1721         """ Returns the wxPython controls that belongs to ButtonPanel. """
1722
1723         children2 = self._mainsizer.GetChildren()
1724         children1 = [child for child in children2 if not child.IsSpacer()]
1725
1726         return children1
1727
1728
1729     def SetStyle(self, style):
1730         """ Sets ButtonPanel style. """
1731
1732         if style == self._nStyle:
1733             return
1734
1735         self._nStyle = style
1736         self.Refresh()
1737
1738
1739     def GetStyle(self):
1740         """ Returns the ButtonPanel style. """
1741
1742         return self._nStyle
1743
1744
1745     def OnMouseMove(self, event):
1746         """ Handles the wx.EVT_MOTION event for ButtonPanel. """
1747
1748         # Check to see if we are hovering a button
1749         tabId, flags = self.HitTest(event.GetPosition())
1750
1751         if flags != BP_HT_BUTTON:
1752             self.RemoveHelp()
1753             self.RepaintOldSelection()
1754             self._currentButton = -1
1755             return
1756
1757         btn = self._vButtons[tabId]
1758
1759         if not btn.IsEnabled():
1760             self.RemoveHelp()
1761             self.RepaintOldSelection()
1762             return
1763
1764         if tabId != self._currentButton:
1765             self.RepaintOldSelection()
1766
1767         if btn.GetRect().Contains(event.GetPosition()):
1768             btn.SetStatus("Hover")
1769         else:
1770             btn.SetStatus("Normal")
1771
1772         if tabId != self._currentButton:
1773             self.RemoveHelp()
1774             self.DoGiveHelp(btn)
1775
1776         self._currentButton = tabId
1777
1778         event.Skip()
1779
1780
1781     def OnLeftDown(self, event):
1782         """ Handles the wx.EVT_LEFT_DOWN event for ButtonPanel. """
1783
1784         tabId, hit = self.HitTest(event.GetPosition())
1785
1786         if hit == BP_HT_BUTTON:
1787             btn = self._vButtons[tabId]
1788             if btn.IsEnabled():
1789                 btn.SetStatus("Pressed")
1790                 self._currentButton = tabId
1791
1792
1793     def OnLeftUp(self, event):
1794         """ Handles the wx.EVT_LEFT_UP event for ButtonPanel. """
1795
1796         tabId, flags = self.HitTest(event.GetPosition())
1797
1798         if flags != BP_HT_BUTTON:
1799             return
1800
1801         hit = self._vButtons[tabId]
1802
1803         if hit.GetStatus() == "Disabled":
1804             return
1805
1806         for btn in self._vButtons:
1807             if btn != hit:
1808                 btn.SetFocus(False)
1809
1810         if hit.GetStatus() == "Pressed":
1811             # Fire a button click event
1812             btnEvent = wx.CommandEvent(wx.wxEVT_COMMAND_BUTTON_CLICKED, hit.GetId())
1813             self.GetEventHandler().ProcessEvent(btnEvent)
1814
1815             hit.SetToggled(not hit.GetToggled())
1816
1817             # Update the button status to be hovered
1818             hit.SetStatus("Hover")
1819             hit.SetFocus()
1820             self._currentButton = tabId
1821
1822
1823     def OnMouseLeave(self, event):
1824         """ Handles the wx.EVT_LEAVE_WINDOW event for ButtonPanel. """
1825
1826         # Reset all buttons statuses
1827         for btn in self._vButtons:
1828             if not btn.IsEnabled():
1829                 continue
1830             btn.SetStatus("Normal")
1831
1832         self.RemoveHelp()
1833
1834         event.Skip()
1835
1836
1837     def OnMouseEnterWindow(self, event):
1838         """ Handles the wx.EVT_ENTER_WINDOW event for ButtonPanel. """
1839
1840         tabId, flags = self.HitTest(event.GetPosition())
1841
1842         if flags == BP_HT_BUTTON:
1843
1844             hit = self._vButtons[tabId]
1845
1846             if hit.GetStatus() == "Disabled":
1847                 event.Skip()
1848                 return
1849
1850             self.DoGiveHelp(hit)
1851             self._currentButton = tabId
1852
1853         event.Skip()
1854
1855
1856     def DoGiveHelp(self, hit):
1857         """ Gives tooltips and help in StatusBar. """
1858
1859         if not self.GetUseHelp():
1860             return
1861
1862         shortHelp = hit.GetShortHelp()
1863         if shortHelp:
1864             self.SetToolTipString(shortHelp)
1865             self._haveTip = True
1866
1867         longHelp = hit.GetLongHelp()
1868         if not longHelp:
1869             return
1870
1871         topLevel = wx.GetTopLevelParent(self)
1872
1873         if isinstance(topLevel, wx.Frame) and topLevel.GetStatusBar():
1874             statusBar = topLevel.GetStatusBar()
1875
1876             if self._statusTimer and self._statusTimer.IsRunning():
1877                 self._statusTimer.Stop()
1878                 statusBar.PopStatusText(0)
1879
1880             statusBar.PushStatusText(longHelp, 0)
1881             self._statusTimer = StatusBarTimer(self)
1882             self._statusTimer.Start(_DELAY, wx.TIMER_ONE_SHOT)
1883
1884
1885     def RemoveHelp(self):
1886         """ Removes the tooltips and statusbar help (if any) for a button. """
1887
1888         if not self.GetUseHelp():
1889             return
1890
1891         if self._haveTip:
1892             self.SetToolTipString("")
1893             self._haveTip = False
1894
1895         if self._statusTimer and self._statusTimer.IsRunning():
1896             topLevel = wx.GetTopLevelParent(self)
1897             statusBar = topLevel.GetStatusBar()
1898             self._statusTimer.Stop()
1899             statusBar.PopStatusText(0)
1900             self._statusTimer = None
1901
1902
1903     def RepaintOldSelection(self):
1904         """ Repaints the old selected/hovered button. """
1905
1906         current = self._currentButton
1907
1908         if current == -1:
1909             return
1910
1911         btn = self._vButtons[current]
1912         if not btn.IsEnabled():
1913             return
1914
1915         btn.SetStatus("Normal")
1916
1917
1918     def OnStatusBarTimer(self):
1919         """ Handles the timer expiring to delete the longHelp in the StatusBar. """
1920
1921         topLevel = wx.GetTopLevelParent(self)
1922         statusBar = topLevel.GetStatusBar()
1923         statusBar.PopStatusText(0)
1924
1925
1926     def SetUseHelp(self, useHelp=True):
1927         """ Sets whether or not shortHelp and longHelp should be displayed. """
1928
1929         self._useHelp = useHelp
1930
1931
1932     def GetUseHelp(self):
1933         """ Returns whether or not shortHelp and longHelp should be displayed. """
1934
1935         return self._useHelp
1936
1937
1938     def HitTest(self, pt):
1939         """
1940         HitTest method for ButtonPanel. Returns the button (if any) and
1941         a flag (if any).
1942         """
1943
1944         for ii in xrange(len(self._vButtons)):
1945             if not self._vButtons[ii].IsEnabled():
1946                 continue
1947             if self._vButtons[ii].GetRect().Contains(pt):
1948                 return ii, BP_HT_BUTTON
1949
1950         return -1, BP_HT_NONE
1951
1952
1953     def GetBPArt(self):
1954         """ Returns the associated BPArt art provider. """
1955
1956         return self._art
1957
1958
1959     def SetBPArt(self, art):
1960         """ Sets a new BPArt to ButtonPanel. Useful only if another BPArt class is used. """
1961
1962         self._art = art
1963         self.Refresh()
1964
1965     if wx.VERSION < (2,7,1,1):
1966         def Freeze(self):
1967             """Freeze ButtonPanel."""
1968
1969             self._freezeCount = self._freezeCount + 1
1970             wx.PyPanel.Freeze(self)
1971
1972
1973         def Thaw(self):
1974             """Thaw ButtonPanel."""
1975
1976             if self._freezeCount == 0:
1977                 raise "\nERROR: Thawing Unfrozen ButtonPanel?"
1978
1979             self._freezeCount = self._freezeCount - 1
1980             wx.PyPanel.Thaw(self)
1981
1982
1983         def IsFrozen(self):
1984             """ Returns whether a call to Freeze() has been done. """
1985
1986             return self._freezeCount != 0
1987
1988