]> git.saurik.com Git - wxWidgets.git/blob - wxPython/wx/lib/buttonpanel.py
More updates from Andrea
[wxWidgets.git] / 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._alignment = 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
1260 self._art = BPArt(style)
1261
1262 self._controlCreated = False
1263
1264 direction = (self.IsVertical() and [wx.VERTICAL] or [wx.HORIZONTAL])[0]
1265 self._mainsizer = BoxSizer(direction)
1266 self.SetSizer(self._mainsizer)
1267
1268 margins = self._art.GetMetric(BP_MARGINS_SIZE)
1269
1270 # First spacer to create some room before the first text/button/control
1271 self._mainsizer.Add((margins.x, margins.y), 0)
1272
1273 # Last spacer to create some room before the last text/button/control
1274 self._mainsizer.Add((margins.x, margins.y), 0)
1275
1276 self.Bind(wx.EVT_SIZE, self.OnSize)
1277 self.Bind(wx.EVT_PAINT, self.OnPaint)
1278 self.Bind(wx.EVT_ERASE_BACKGROUND, self.OnEraseBackground)
1279 self.Bind(wx.EVT_MOTION, self.OnMouseMove)
1280 self.Bind(wx.EVT_LEFT_DOWN, self.OnLeftDown)
1281 self.Bind(wx.EVT_LEFT_UP, self.OnLeftUp)
1282 self.Bind(wx.EVT_LEAVE_WINDOW, self.OnMouseLeave)
1283 self.Bind(wx.EVT_ENTER_WINDOW, self.OnMouseEnterWindow)
1284
1285 self.SetBarText(text)
1286 self.LayoutItems()
1287
1288
1289 def SetBarText(self, text):
1290 """ Sets the main caption text (leave text="" for no text). """
1291
1292 self.Freeze()
1293
1294 text = text.strip()
1295
1296 if self._controlCreated:
1297 self.RemoveText()
1298
1299 self._text = ButtonPanelText(self, text)
1300 lenChildren = len(self._mainsizer.GetChildren())
1301
1302 if text == "":
1303 # Even if we have no text, we insert it an empty spacer anyway
1304 # it is easier to handle if you have to recreate the sizer after.
1305 if self.IsStandard():
1306 self._mainsizer.Insert(1, self._text, 0, wx.ALIGN_CENTER,
1307 userData=self._text, realIndex=0)
1308 else:
1309 self._mainsizer.Insert(lenChildren-1, self._text, 0, wx.ALIGN_CENTER,
1310 userData=self._text, realIndex=lenChildren)
1311
1312 return
1313
1314 # We have text, so insert the text and an expandable spacer
1315 # alongside it. "Standard" ButtonPanel are left or top aligned.
1316 if self.IsStandard():
1317 self._mainsizer.Insert(1, self._text, 0, wx.ALIGN_CENTER,
1318 userData=self._text, realIndex=0)
1319 self._mainsizer.Insert(2, (0, 0), 1, wx.EXPAND)
1320
1321 else:
1322 self._mainsizer.Insert(lenChildren-1, self._text, 0, wx.ALIGN_CENTER,
1323 userData=self._text, realIndex=lenChildren)
1324 self._mainsizer.Insert(lenChildren-1, (0, 0), 1, wx.EXPAND)
1325
1326
1327 def RemoveText(self):
1328 """ Removes the main caption text. """
1329
1330 lenChildren = len(self._mainsizer.GetChildren())
1331 lenCustom = len(self._vButtons) + len(self._vSeparators) + 1
1332
1333 if self.IsStandard():
1334 # Detach the text
1335 self._mainsizer.Remove(1, 0)
1336 if self.HasBarText():
1337 # Detach the expandable spacer
1338 self._mainsizer.Remove(1, -1)
1339 else:
1340 # Detach the text
1341 self._mainsizer.Remove(lenChildren-2, lenCustom-1)
1342 if self.HasBarText():
1343 # Detach the expandable spacer
1344 self._mainsizer.Remove(lenChildren-3, -1)
1345
1346
1347 def GetBarText(self):
1348 """ Returns the main caption text. """
1349
1350 return self._text.GetText()
1351
1352
1353 def HasBarText(self):
1354 """ Returns whether ButtonPanel has a main caption text or not. """
1355
1356 return hasattr(self, "_text") and self._text.GetText() != ""
1357
1358
1359 def AddButton(self, btnInfo):
1360 """
1361 Adds a button to ButtonPanel. Remember to pass a ButtonInfo instance to
1362 this method. See the demo for details.
1363 """
1364
1365 lenChildren = len(self._mainsizer.GetChildren())
1366 self._mainsizer.Insert(lenChildren-1, btnInfo, 0, wx.ALIGN_CENTER|wx.EXPAND, userData=btnInfo)
1367
1368 self._vButtons.append(btnInfo)
1369
1370
1371 def AddSpacer(self, size=(0, 0), proportion=1, flag=wx.EXPAND):
1372 """ Adds a spacer (stretchable or fixed-size) to ButtonPanel. """
1373
1374 lenChildren = len(self._mainsizer.GetChildren())
1375 self._mainsizer.Insert(lenChildren-1, size, proportion, flag)
1376
1377
1378 def AddControl(self, control, proportion=0, flag=wx.ALIGN_CENTER|wx.ALL, border=None):
1379 """ Adds a wxPython control to ButtonPanel. """
1380
1381 lenChildren = len(self._mainsizer.GetChildren())
1382
1383 if border is None:
1384 border = self._art.GetMetric(BP_PADDING_SIZE)
1385 border = max(border.x, border.y)
1386
1387 self._mainsizer.Insert(lenChildren-1, control, proportion, flag, border)
1388
1389
1390 def AddSeparator(self):
1391 """ Adds a separator line to ButtonPanel. """
1392
1393 lenChildren = len(self._mainsizer.GetChildren())
1394 separator = Separator(self)
1395
1396 self._mainsizer.Insert(lenChildren-1, separator, 0, wx.EXPAND)
1397 self._vSeparators.append(separator)
1398
1399
1400 def RemoveAllButtons(self):
1401 """ Remove all the buttons from ButtonPanel. """
1402
1403 self._vButtons = []
1404
1405
1406 def RemoveAllSeparators(self):
1407 """ Remove all the separators from ButtonPanel. """
1408
1409 self._vSeparators = []
1410
1411
1412 def GetAlignment(self):
1413 """ Returns the button alignment (left, right, top, bottom). """
1414
1415 return self._alignment
1416
1417
1418 def SetAlignment(self, alignment):
1419 """ Sets the button alignment (left, right, top, bottom). """
1420
1421 if alignment == self._alignment:
1422 return
1423
1424 self.Freeze()
1425
1426 text = self.GetBarText()
1427
1428 # Remove the text in any case
1429 self.RemoveText()
1430
1431 # Remove the first and last spacers
1432 self._mainsizer.Remove(0, -1)
1433 self._mainsizer.Remove(len(self._mainsizer.GetChildren())-1, -1)
1434
1435 self._alignment = alignment
1436
1437 # Recreate the sizer accordingly to the new alignment
1438 self.ReCreateSizer(text)
1439
1440
1441 def IsVertical(self):
1442 """ Returns whether ButtonPanel is vertically aligned or not. """
1443
1444 return self._alignment not in [BP_ALIGN_RIGHT, BP_ALIGN_LEFT]
1445
1446
1447 def IsStandard(self):
1448 """ Returns whether ButtonPanel is aligned "Standard" (left/top) or not. """
1449
1450 return self._alignment in [BP_ALIGN_LEFT, BP_ALIGN_TOP]
1451
1452
1453 def DoLayout(self):
1454 """
1455 Do the Layout for ButtonPanel.
1456 NB: Call this method every time you make a modification to the layout
1457 or to the customizable sizes of the pseudo controls.
1458 """
1459
1460 margins = self._art.GetMetric(BP_MARGINS_SIZE)
1461 lenChildren = len(self._mainsizer.GetChildren())
1462
1463 self._mainsizer.SetItemMinSize(0, (margins.x, margins.y))
1464 self._mainsizer.SetItemMinSize(lenChildren-1, (margins.x, margins.y))
1465
1466 self._controlCreated = True
1467 self.LayoutItems()
1468
1469 # *VERY* WEIRD: the sizer seems not to respond to any layout until I
1470 # change the ButtonPanel size and restore it back
1471 size = self.GetSize()
1472 self.SetSize((size.x+1, size.y+1))
1473 self.SetSize((size.x, size.y))
1474
1475 if self.IsFrozen():
1476 self.Thaw()
1477
1478
1479 def ReCreateSizer(self, text):
1480 """ Recreates the ButtonPanel sizer accordingly to the alignment specified. """
1481
1482 children = self._mainsizer.GetChildren()
1483 self.RemoveAllButtons()
1484 self.RemoveAllSeparators()
1485
1486 # Create a new sizer depending on the alignment chosen
1487 direction = (self.IsVertical() and [wx.VERTICAL] or [wx.HORIZONTAL])[0]
1488 self._mainsizer = BoxSizer(direction)
1489
1490 margins = self._art.GetMetric(BP_MARGINS_SIZE)
1491 # First spacer to create some room before the first text/button/control
1492 self._mainsizer.Add((margins.x, margins.y), 0)
1493
1494 # Last spacer to create some room before the last text/button/control
1495 self._mainsizer.Add((margins.x, margins.y), 0)
1496
1497 # This is needed otherwise SetBarText goes mad
1498 self._controlCreated = False
1499
1500 for child in children:
1501 userData = child.GetUserData()
1502 if userData:
1503 if isinstance(userData, ButtonInfo):
1504 # It is a ButtonInfo, can't be anything else
1505 self.AddButton(child.GetUserData())
1506 elif isinstance(userData, Separator):
1507 self.AddSeparator()
1508
1509 else:
1510 if child.IsSpacer():
1511 # This is a spacer, expandable or not
1512 self.AddSpacer(child.GetSize(), child.GetProportion(),
1513 child.GetFlag())
1514 else:
1515 # This is a wxPython control
1516 self.AddControl(child.GetWindow(), child.GetProportion(),
1517 child.GetFlag(), child.GetBorder())
1518
1519 self.SetSizer(self._mainsizer)
1520
1521 # Now add the text. It doesn't matter if there is no text
1522 self.SetBarText(text)
1523
1524 self.DoLayout()
1525
1526 self.Thaw()
1527
1528
1529 def DoGetBestSize(self):
1530 """ Returns the best size of ButtonPanel. """
1531
1532 w = h = btnWidth = btnHeight = 0
1533 isVertical = self.IsVertical()
1534
1535 padding = self._art.GetMetric(BP_PADDING_SIZE)
1536 border = self._art.GetMetric(BP_BORDER_SIZE)
1537 margins = self._art.GetMetric(BP_MARGINS_SIZE)
1538 separator_size = self._art.GetMetric(BP_SEPARATOR_SIZE)
1539
1540 # Add the space required for the main caption
1541 if self.HasBarText():
1542 w, h = self._text.GetBestSize()
1543 if isVertical:
1544 h += padding.y
1545 else:
1546 w += padding.x
1547 else:
1548 w = h = border
1549
1550 # Add the button's sizes
1551 for btn in self._vButtons:
1552
1553 bw, bh = btn.GetBestSize()
1554 btnWidth = max(btnWidth, bw)
1555 btnHeight = max(btnHeight, bh)
1556
1557 if isVertical:
1558 w = max(w, btnWidth)
1559 h += bh
1560 else:
1561 h = max(h, btnHeight)
1562 w += bw
1563
1564 # Add the control's sizes
1565 for control in self.GetControls():
1566 cw, ch = control.GetSize()
1567 if isVertical:
1568 h += ch
1569 w = max(w, cw)
1570 else:
1571 w += cw
1572 h = max(h, ch)
1573
1574 # Add the separator's sizes and the 2 SizerItems at the beginning
1575 # and at the end
1576 if self.IsVertical():
1577 h += 2*margins.y + len(self._vSeparators)*separator_size
1578 else:
1579 w += 2*margins.x + len(self._vSeparators)*separator_size
1580
1581 return wx.Size(w, h)
1582
1583
1584 def OnPaint(self, event):
1585 """ Handles the wx.EVT_PAINT event for ButtonPanel. """
1586
1587 dc = wx.BufferedPaintDC(self)
1588 rect = self.GetClientRect()
1589
1590 self._art.DrawButtonPanel(dc, rect, self._nStyle)
1591 self._mainsizer.Draw(dc)
1592
1593
1594 def OnEraseBackground(self, event):
1595 """ Handles the wx.EVT_ERASE_BACKGROUND event for ButtonPanel (does nothing). """
1596
1597 pass
1598
1599
1600 def OnSize(self, event):
1601 """ Handles the wx.EVT_SIZE event for ButtonPanel. """
1602
1603 # NOTE: It seems like LayoutItems number of calls can be optimized in some way.
1604 # Currently every DoLayout (or every parent Layout()) calls about 3 times
1605 # the LayoutItems method. Any idea on how to improve it?
1606 self.LayoutItems()
1607 self.Refresh()
1608
1609 event.Skip()
1610
1611
1612 def LayoutItems(self):
1613 """
1614 Layout the items using a different algorithm depending on the existance
1615 of the main caption.
1616 """
1617
1618 nonspacers, allchildren = self.GetNonFlexibleChildren()
1619
1620 if self.HasBarText():
1621 self.FlexibleLayout(nonspacers, allchildren)
1622 else:
1623 self.SizeLayout(nonspacers, allchildren)
1624
1625 self._mainsizer.Layout()
1626
1627
1628 def SizeLayout(self, nonspacers, children):
1629 """ Layout the items when no main caption exists. """
1630
1631 size = self.GetSize()
1632 isVertical = self.IsVertical()
1633
1634 corner = 0
1635 indx1 = len(nonspacers)
1636
1637 for item in nonspacers:
1638 corner += self.GetItemSize(item, isVertical)
1639 if corner > size[isVertical]:
1640 indx1 = nonspacers.index(item)
1641 break
1642
1643 # Leave out the last spacer, it has to be there always
1644 for ii in xrange(len(nonspacers)-1):
1645 indx = children.index(nonspacers[ii])
1646 self._mainsizer.Show(indx, ii < indx1)
1647
1648
1649 def GetItemSize(self, item, isVertical):
1650 """ Returns the size of an item in the main ButtonPanel sizer. """
1651
1652 if item.GetUserData():
1653 return item.GetUserData().GetBestSize()[isVertical]
1654 else:
1655 return item.GetSize()[isVertical]
1656
1657
1658 def FlexibleLayout(self, nonspacers, allchildren):
1659 """ Layout the items when the main caption exists. """
1660
1661 if len(nonspacers) < 2:
1662 return
1663
1664 isVertical = self.IsVertical()
1665 isStandard = self.IsStandard()
1666
1667 size = self.GetSize()[isVertical]
1668 padding = self._art.GetMetric(BP_PADDING_SIZE)
1669
1670 fixed = (isStandard and [nonspacers[1]] or [nonspacers[-2]])[0]
1671
1672 if isStandard:
1673 nonspacers.reverse()
1674 leftendx = fixed.GetSize()[isVertical] + padding.x
1675 else:
1676 rightstartx = size - fixed.GetSize()[isVertical]
1677 size = 0
1678
1679 count = lennonspacers = len(nonspacers)
1680
1681 for item in nonspacers:
1682 if isStandard:
1683 size -= self.GetItemSize(item, isVertical)
1684 if size < leftendx:
1685 break
1686 else:
1687 size += self.GetItemSize(item, isVertical)
1688 if size > rightstartx:
1689 break
1690
1691 count = count - 1
1692
1693 nonspacers.reverse()
1694
1695 for jj in xrange(2, lennonspacers):
1696 indx = allchildren.index(nonspacers[jj])
1697 self._mainsizer.Show(indx, jj >= count)
1698
1699
1700 def GetNonFlexibleChildren(self):
1701 """
1702 Returns all the ButtonPanel main sizer's children that are not
1703 flexible spacers.
1704 """
1705
1706 children1 = []
1707 children2 = self._mainsizer.GetChildren()
1708
1709 for child in children2:
1710 if child.IsSpacer():
1711 if child.GetUserData() or child.GetProportion() == 0:
1712 children1.append(child)
1713 else:
1714 children1.append(child)
1715
1716 return children1, children2
1717
1718
1719 def GetControls(self):
1720 """ Returns the wxPython controls that belongs to ButtonPanel. """
1721
1722 children2 = self._mainsizer.GetChildren()
1723 children1 = [child for child in children2 if not child.IsSpacer()]
1724
1725 return children1
1726
1727
1728 def SetStyle(self, style):
1729 """ Sets ButtonPanel style. """
1730
1731 if style == self._nStyle:
1732 return
1733
1734 self._nStyle = style
1735 self.Refresh()
1736
1737
1738 def GetStyle(self):
1739 """ Returns the ButtonPanel style. """
1740
1741 return self._nStyle
1742
1743
1744 def OnMouseMove(self, event):
1745 """ Handles the wx.EVT_MOTION event for ButtonPanel. """
1746
1747 # Check to see if we are hovering a button
1748 tabId, flags = self.HitTest(event.GetPosition())
1749
1750 if flags != BP_HT_BUTTON:
1751 self.RemoveHelp()
1752 self.RepaintOldSelection()
1753 self._currentButton = -1
1754 return
1755
1756 btn = self._vButtons[tabId]
1757
1758 if not btn.IsEnabled():
1759 self.RemoveHelp()
1760 self.RepaintOldSelection()
1761 return
1762
1763 if tabId != self._currentButton:
1764 self.RepaintOldSelection()
1765
1766 if btn.GetRect().Contains(event.GetPosition()):
1767 btn.SetStatus("Hover")
1768 else:
1769 btn.SetStatus("Normal")
1770
1771 if tabId != self._currentButton:
1772 self.RemoveHelp()
1773 self.DoGiveHelp(btn)
1774
1775 self._currentButton = tabId
1776
1777 event.Skip()
1778
1779
1780 def OnLeftDown(self, event):
1781 """ Handles the wx.EVT_LEFT_DOWN event for ButtonPanel. """
1782
1783 tabId, hit = self.HitTest(event.GetPosition())
1784
1785 if hit == BP_HT_BUTTON:
1786 btn = self._vButtons[tabId]
1787 if btn.IsEnabled():
1788 btn.SetStatus("Pressed")
1789 self._currentButton = tabId
1790
1791
1792 def OnLeftUp(self, event):
1793 """ Handles the wx.EVT_LEFT_UP event for ButtonPanel. """
1794
1795 tabId, flags = self.HitTest(event.GetPosition())
1796
1797 if flags != BP_HT_BUTTON:
1798 return
1799
1800 hit = self._vButtons[tabId]
1801
1802 if hit.GetStatus() == "Disabled":
1803 return
1804
1805 for btn in self._vButtons:
1806 if btn != hit:
1807 btn.SetFocus(False)
1808
1809 if hit.GetStatus() == "Pressed":
1810 # Fire a button click event
1811 btnEvent = wx.CommandEvent(wx.wxEVT_COMMAND_BUTTON_CLICKED, hit.GetId())
1812 self.GetEventHandler().ProcessEvent(btnEvent)
1813
1814 hit.SetToggled(not hit.GetToggled())
1815
1816 # Update the button status to be hovered
1817 hit.SetStatus("Hover")
1818 hit.SetFocus()
1819 self._currentButton = tabId
1820
1821
1822 def OnMouseLeave(self, event):
1823 """ Handles the wx.EVT_LEAVE_WINDOW event for ButtonPanel. """
1824
1825 # Reset all buttons statuses
1826 for btn in self._vButtons:
1827 if not btn.IsEnabled():
1828 continue
1829 btn.SetStatus("Normal")
1830
1831 self.RemoveHelp()
1832
1833 event.Skip()
1834
1835
1836 def OnMouseEnterWindow(self, event):
1837 """ Handles the wx.EVT_ENTER_WINDOW event for ButtonPanel. """
1838
1839 tabId, flags = self.HitTest(event.GetPosition())
1840
1841 if flags == BP_HT_BUTTON:
1842
1843 hit = self._vButtons[tabId]
1844
1845 if hit.GetStatus() == "Disabled":
1846 event.Skip()
1847 return
1848
1849 self.DoGiveHelp(hit)
1850 self._currentButton = tabId
1851
1852 event.Skip()
1853
1854
1855 def DoGiveHelp(self, hit):
1856 """ Gives tooltips and help in StatusBar. """
1857
1858 if not self.GetUseHelp():
1859 return
1860
1861 shortHelp = hit.GetShortHelp()
1862 if shortHelp:
1863 self.SetToolTipString(shortHelp)
1864
1865 longHelp = hit.GetLongHelp()
1866 if not longHelp:
1867 return
1868
1869 topLevel = wx.GetTopLevelParent(self)
1870
1871 if isinstance(topLevel, wx.Frame) and topLevel.GetStatusBar():
1872 statusBar = topLevel.GetStatusBar()
1873
1874 if self._statusTimer and self._statusTimer.IsRunning():
1875 self._statusTimer.Stop()
1876 statusBar.PopStatusText(0)
1877
1878 statusBar.PushStatusText(longHelp, 0)
1879 self._statusTimer = StatusBarTimer(self)
1880 self._statusTimer.Start(_DELAY, wx.TIMER_ONE_SHOT)
1881
1882
1883 def RemoveHelp(self):
1884 """ Removes the tooltips and statusbar help (if any) for a button. """
1885
1886 if not self.GetUseHelp():
1887 return
1888
1889 self.SetToolTipString("")
1890
1891 if self._statusTimer and self._statusTimer.IsRunning():
1892 topLevel = wx.GetTopLevelParent(self)
1893 statusBar = topLevel.GetStatusBar()
1894 self._statusTimer.Stop()
1895 statusBar.PopStatusText(0)
1896 self._statusTimer = None
1897
1898
1899 def RepaintOldSelection(self):
1900 """ Repaints the old selected/hovered button. """
1901
1902 current = self._currentButton
1903
1904 if current == -1:
1905 return
1906
1907 btn = self._vButtons[current]
1908 if not btn.IsEnabled():
1909 return
1910
1911 btn.SetStatus("Normal")
1912
1913
1914 def OnStatusBarTimer(self):
1915 """ Handles the timer expiring to delete the longHelp in the StatusBar. """
1916
1917 topLevel = wx.GetTopLevelParent(self)
1918 statusBar = topLevel.GetStatusBar()
1919 statusBar.PopStatusText(0)
1920
1921
1922 def SetUseHelp(self, useHelp=True):
1923 """ Sets whether or not shortHelp and longHelp should be displayed. """
1924
1925 self._useHelp = useHelp
1926
1927
1928 def GetUseHelp(self):
1929 """ Returns whether or not shortHelp and longHelp should be displayed. """
1930
1931 return self._useHelp
1932
1933
1934 def HitTest(self, pt):
1935 """
1936 HitTest method for ButtonPanel. Returns the button (if any) and
1937 a flag (if any).
1938 """
1939
1940 for ii in xrange(len(self._vButtons)):
1941 if not self._vButtons[ii].IsEnabled():
1942 continue
1943 if self._vButtons[ii].GetRect().Contains(pt):
1944 return ii, BP_HT_BUTTON
1945
1946 return -1, BP_HT_NONE
1947
1948
1949 def GetBPArt(self):
1950 """ Returns the associated BPArt art provider. """
1951
1952 return self._art
1953
1954
1955 def SetBPArt(self, art):
1956 """ Sets a new BPArt to ButtonPanel. Useful only if another BPArt class is used. """
1957
1958 self._art = art
1959 self.Refresh()
1960
1961 if wx.VERSION < (2,7,1,1):
1962 def Freeze(self):
1963 """Freeze ButtonPanel."""
1964
1965 self._freezeCount = self._freezeCount + 1
1966 wx.PyPanel.Freeze(self)
1967
1968
1969 def Thaw(self):
1970 """Thaw ButtonPanel."""
1971
1972 if self._freezeCount == 0:
1973 raise "\nERROR: Thawing Unfrozen ButtonPanel?"
1974
1975 self._freezeCount = self._freezeCount - 1
1976 wx.PyPanel.Thaw(self)
1977
1978
1979 def IsFrozen(self):
1980 """ Returns whether a call to Freeze() has been done. """
1981
1982 return self._freezeCount != 0
1983
1984