]>
Commit | Line | Data |
---|---|---|
1 | #---------------------------------------------------------------------------- | |
2 | # Name: ColourSelect.py | |
3 | # Purpose: Colour Box Selection Control | |
4 | # | |
5 | # Author: Lorne White, Lorne.White@telusplanet.net | |
6 | # | |
7 | # Created: Feb 25, 2001 | |
8 | # Licence: wxWindows license | |
9 | #---------------------------------------------------------------------------- | |
10 | ||
11 | # creates a colour wxButton with selectable color | |
12 | # button click provides a colour selection box | |
13 | # button colour will change to new colour | |
14 | # GetColour method to get the selected colour | |
15 | ||
16 | # Updates: | |
17 | # call back to function if changes made | |
18 | ||
19 | # Cliff Wells, logiplexsoftware@earthlink.net: | |
20 | # - Made ColourSelect into "is a button" rather than "has a button" | |
21 | # - Added label parameter and logic to adjust the label colour according to the background | |
22 | # colour | |
23 | # - Added id argument | |
24 | # - Rearranged arguments to more closely follow wx conventions | |
25 | # - Simplified some of the code | |
26 | ||
27 | # Cliff Wells, 2002/02/07 | |
28 | # - Added ColourSelect Event | |
29 | ||
30 | # 12/01/2003 - Jeff Grimmett (grimmtooth@softhome.net) | |
31 | # | |
32 | # o Updated for 2.5 compatability. | |
33 | # | |
34 | ||
35 | #---------------------------------------------------------------------------- | |
36 | ||
37 | import wx | |
38 | ||
39 | #---------------------------------------------------------------------------- | |
40 | ||
41 | wxEVT_COMMAND_COLOURSELECT = wx.NewEventType() | |
42 | ||
43 | class ColourSelectEvent(wx.PyCommandEvent): | |
44 | def __init__(self, id, value): | |
45 | wx.PyCommandEvent.__init__(self, id = id) | |
46 | self.SetEventType(wxEVT_COMMAND_COLOURSELECT) | |
47 | self.value = value | |
48 | ||
49 | def GetValue(self): | |
50 | return self.value | |
51 | ||
52 | EVT_COLOURSELECT = wx.PyEventBinder(wxEVT_COMMAND_COLOURSELECT, 1) | |
53 | ||
54 | #---------------------------------------------------------------------------- | |
55 | ||
56 | class ColourSelect(wx.BitmapButton): | |
57 | def __init__(self, parent, id, label="", colour=wx.BLACK, | |
58 | pos=wx.DefaultPosition, size=wx.DefaultSize, | |
59 | callback=None, style=0): | |
60 | if label: | |
61 | w, h = parent.GetTextExtent(label) | |
62 | w += 6 | |
63 | h += 6 | |
64 | else: | |
65 | w, h = 20, 20 | |
66 | wx.BitmapButton.__init__(self, parent, id, wx.EmptyBitmap(w,h), | |
67 | pos=pos, size=size, style=style|wx.BU_AUTODRAW) | |
68 | ||
69 | if type(colour) == type( () ): | |
70 | colour = wx.Colour(*colour) | |
71 | self.colour = colour | |
72 | self.SetLabel(label) | |
73 | self.callback = callback | |
74 | bmp = self.MakeBitmap() | |
75 | self.SetBitmap(bmp) | |
76 | parent.Bind(wx.EVT_BUTTON, self.OnClick, self) | |
77 | ||
78 | ||
79 | def GetColour(self): | |
80 | return self.colour | |
81 | ||
82 | def GetValue(self): | |
83 | return self.colour | |
84 | ||
85 | def SetValue(self, colour): | |
86 | self.SetColour(colour) | |
87 | ||
88 | def SetColour(self, colour): | |
89 | if type(colour) == tuple: | |
90 | colour = wx.Colour(*colour) | |
91 | if type(colour) == str: | |
92 | colour = wx.NamedColour(colour) | |
93 | ||
94 | self.colour = colour | |
95 | bmp = self.MakeBitmap() | |
96 | self.SetBitmap(bmp) | |
97 | ||
98 | ||
99 | def MakeBitmap(self): | |
100 | bdr = 10 | |
101 | width, height = self.GetSize() | |
102 | bmp = wx.EmptyBitmap(width-bdr, height-bdr) | |
103 | dc = wx.MemoryDC() | |
104 | dc.SelectObject(bmp) | |
105 | dc.SetFont(self.GetFont()) | |
106 | label = self.GetLabel() | |
107 | # Just make a little colored bitmap | |
108 | dc.SetBackground(wx.Brush(self.colour)) | |
109 | dc.Clear() | |
110 | ||
111 | if label: | |
112 | # Add a label to it | |
113 | avg = reduce(lambda a, b: a + b, self.colour.Get()) / 3 | |
114 | fcolour = avg > 128 and wx.BLACK or wx.WHITE | |
115 | dc.SetTextForeground(fcolour) | |
116 | dc.DrawLabel(label, (0,0, width-bdr, height-bdr), | |
117 | wx.ALIGN_CENTER) | |
118 | ||
119 | dc.SelectObject(wx.NullBitmap) | |
120 | return bmp | |
121 | ||
122 | ||
123 | def SetBitmap(self, bmp): | |
124 | self.SetBitmapLabel(bmp) | |
125 | self.SetBitmapSelected(bmp) | |
126 | self.SetBitmapDisabled(bmp) | |
127 | self.SetBitmapFocus(bmp) | |
128 | self.SetBitmapSelected(bmp) | |
129 | ||
130 | ||
131 | def OnChange(self): | |
132 | wx.PostEvent(self, ColourSelectEvent(self.GetId(), self.GetValue())) | |
133 | if self.callback is not None: | |
134 | self.callback() | |
135 | ||
136 | def OnClick(self, event): | |
137 | data = wx.ColourData() | |
138 | data.SetChooseFull(True) | |
139 | data.SetColour(self.colour) | |
140 | dlg = wx.ColourDialog(self.GetParent(), data) | |
141 | changed = dlg.ShowModal() == wx.ID_OK | |
142 | ||
143 | if changed: | |
144 | data = dlg.GetColourData() | |
145 | self.SetColour(data.GetColour()) | |
146 | ||
147 | dlg.Destroy() | |
148 | ||
149 | # moved after dlg.Destroy, since who knows what the callback will do... | |
150 | if changed: | |
151 | self.OnChange() | |
152 |