]>
Commit | Line | Data |
---|---|---|
944930d5 RD |
1 | #! /usr/local/bin/python |
2 | #---------------------------------------------------------------------------- | |
e0473f5f | 3 | # Name: calendar.py |
944930d5 RD |
4 | # Purpose: Calendar display control |
5 | # | |
6 | # Author: Lorne White (email: lwhite1@planet.eon.net) | |
7 | # | |
8 | # Created: | |
9 | # Version 0.5 1999/11/03 | |
10 | # Licence: wxWindows license | |
11 | #---------------------------------------------------------------------------- | |
12 | ||
13 | from wxPython.wx import * | |
14 | ||
e0473f5f | 15 | from CDate import * |
944930d5 RD |
16 | import string, time |
17 | ||
e0473f5f | 18 | |
944930d5 RD |
19 | CalDays = [6, 0, 1, 2, 3, 4, 5] |
20 | AbrWeekday = {6:"Sun", 0:"Mon", 1:"Tue", 2:"Wed", 3:"Thu", 4:"Fri", 5:"Sat"} | |
e0473f5f | 21 | _MIDSIZE = 100 |
944930d5 RD |
22 | |
23 | ||
24 | # calendar drawing routing | |
25 | ||
26 | class CalDraw: | |
27 | def __init__(self): | |
28 | self.rg = {} | |
29 | self.y_st = 15 # start of vertical draw default | |
30 | ||
31 | def SetSize(self, size): | |
32 | self.sizew = size.width | |
33 | self.sizeh = size.height | |
34 | ||
35 | # draw the various elements of the calendar | |
36 | ||
37 | def DrawCal(self, DC, sel_lst): | |
38 | self.DC = DC | |
39 | ||
40 | self.DrawBorder() | |
41 | ||
42 | if self.hide_title is FALSE: | |
43 | self.DrawMonth() | |
44 | ||
45 | self.Center() | |
46 | ||
47 | self.DrawGrid() | |
48 | self.GetRect() | |
49 | ||
50 | self.DrawSel(sel_lst) # highlighted days | |
51 | self.DrawWeek() | |
52 | self.DrawNum() | |
53 | ||
54 | # draw border around the outside of the main display rectangle | |
55 | ||
56 | def DrawBorder(self): | |
57 | rect = wxRect(0, 0, self.sizew, self.sizeh) # full display window area | |
58 | self.DC.DrawRectangle(rect.x, rect.y, rect.width, rect.height) | |
59 | ||
60 | def DrawNumVal(self): | |
61 | self.DrawNum() | |
62 | ||
63 | # calculate the calendar days and offset position | |
64 | ||
65 | def SetCal(self, year, month): | |
66 | self.year = year | |
67 | self.month = month | |
68 | ||
69 | day = 1 | |
70 | t = Date(year, month, day) | |
71 | dow = t.day_of_week # start day in month | |
72 | dim = t.days_in_month # number of days in month | |
73 | start_pos = dow+1 | |
74 | self.st_pos = start_pos | |
75 | ||
76 | self.cal = [] | |
77 | for i in range(start_pos): | |
78 | self.cal.append('') | |
79 | i = 1 | |
80 | while i <= dim: | |
81 | self.cal.append(str(i)) | |
82 | i = i + 1 | |
83 | return start_pos | |
84 | ||
85 | # get the display rectange list of the day grid | |
86 | ||
87 | def GetRect(self): | |
88 | cnt = 0 | |
89 | for y in self.gridy[1:-1]: | |
90 | for x in self.gridx[:-1]: | |
91 | rect = wxRect(x, y, self.dl_w, self.dl_h) # create rect region | |
92 | self.rg[cnt] = rect | |
93 | cnt = cnt + 1 | |
94 | return self.rg | |
95 | ||
96 | def GetCal(self): | |
97 | return self.cal | |
98 | ||
99 | def GetOffset(self): | |
100 | return self.st_pos | |
101 | ||
102 | # month and year title | |
103 | ||
104 | def DrawMonth(self): | |
105 | month = Month[self.month] | |
106 | ||
107 | sizef = 12 | |
e0473f5f RD |
108 | if self.sizeh < _MIDSIZE: |
109 | sizef = 10 | |
944930d5 | 110 | |
e0473f5f | 111 | f = wxFont(sizef, self.font, wxNORMAL, self.bold) |
944930d5 RD |
112 | self.DC.SetFont(f) |
113 | ||
114 | tw,th = self.DC.GetTextExtent(month) | |
115 | adjust = (self.sizew-tw)/2 | |
116 | self.DC.DrawText(month, adjust, 10) | |
117 | ||
118 | year = str(self.year) | |
119 | tw,th = self.DC.GetTextExtent(year) | |
120 | adjust = self.sizew-tw-20 | |
121 | ||
122 | self.y_st = th * 3 | |
123 | ||
e0473f5f | 124 | f = wxFont(sizef, self.font, wxNORMAL, self.bold) |
944930d5 RD |
125 | self.DC.SetFont(f) |
126 | self.DC.DrawText(year, adjust, 10) | |
127 | ||
128 | # draw the week days | |
129 | ||
130 | def DrawWeek(self): | |
e0473f5f RD |
131 | sizef = 10 |
132 | if self.sizeh < _MIDSIZE: | |
133 | sizef = 8 | |
944930d5 | 134 | |
e0473f5f | 135 | f = wxFont(sizef, self.font, wxNORMAL, self.bold) |
944930d5 RD |
136 | self.DC.SetFont(f) |
137 | ||
138 | cnt_x = 0 | |
139 | cnt_y = 0 | |
140 | width = self.gridx[1]-self.gridx[0] | |
141 | height = self.gridy[1] - self.gridy[0] | |
142 | ||
143 | for val in CalDays: | |
144 | day = AbrWeekday[val] | |
145 | if self.sizew < 200: | |
146 | day = day[0] | |
147 | dw,dh = self.DC.GetTextExtent(day) | |
148 | diffx = (width-dw)/2 | |
149 | diffy = (height-dh)/2 | |
150 | ||
151 | x = self.gridx[cnt_x] | |
152 | y = self.gridy[cnt_y] | |
153 | self.DC.DrawText(day, x+diffx, y+diffy) | |
154 | cnt_x = cnt_x + 1 | |
155 | ||
156 | # draw the day numbers | |
157 | ||
158 | def DrawNum(self): | |
e0473f5f RD |
159 | sizef = 10 |
160 | if self.sizeh < _MIDSIZE: | |
161 | sizef = 8 | |
162 | f = wxFont(sizef, self.font, wxNORMAL, self.bold) | |
944930d5 RD |
163 | self.DC.SetFont(f) |
164 | ||
165 | cnt_x = 0 | |
166 | cnt_y = 1 | |
167 | for val in self.cal: | |
168 | x = self.gridx[cnt_x] | |
169 | y = self.gridy[cnt_y] | |
170 | self.DC.DrawText(val, x+5, y+5) | |
171 | if cnt_x < 6: | |
172 | cnt_x = cnt_x + 1 | |
173 | else: | |
174 | cnt_x = 0 | |
175 | cnt_y = cnt_y + 1 | |
176 | ||
177 | # calculate the dimensions in the center of the drawing area | |
178 | ||
179 | def Center(self): | |
180 | self.x_st = 10 | |
181 | self.y_end = 10 | |
182 | ||
183 | bdw = self.x_st * 2 | |
184 | bdh = self.y_st + self.y_end | |
185 | ||
186 | self.dl_w = (self.sizew-bdw)/7 | |
187 | self.dl_h = (self.sizeh-bdh)/7 | |
188 | ||
189 | self.cwidth = self.dl_w * 7 | |
190 | self.cheight = self.dl_h * 6 + self.dl_h/2 | |
191 | ||
192 | # highlighted selectioned days | |
193 | ||
194 | def DrawSel(self, sel_lst): | |
195 | for key in sel_lst: | |
196 | brush = wxBrush(wxNamedColour(self.high_color), wxSOLID) | |
197 | self.DC.SetBrush(brush) | |
198 | if self.hide_grid is FALSE: | |
199 | self.DC.SetPen(wxPen(wxNamedColour(self.grid_color), 0)) | |
200 | else: | |
201 | self.DC.SetPen(wxPen(wxNamedColour(self.back_color), 0)) | |
202 | nkey = key + self.st_pos -1 | |
203 | rect = self.rg[nkey] | |
204 | self.DC.DrawRectangle(rect.x, rect.y, rect.width+1, rect.height+1) | |
205 | ||
206 | # calculate and draw the grid lines | |
207 | ||
208 | def DrawGrid(self): | |
209 | self.DC.SetPen(wxPen(wxNamedColour(self.grid_color), 0)) | |
210 | ||
211 | self.gridx = [] | |
212 | self.gridy = [] | |
213 | ||
214 | x1 = self.x_st | |
215 | y1 = self.y_st | |
216 | y1 = self.y_st | |
217 | y2 = self.y_st + self.cheight | |
218 | for i in range(8): | |
219 | if self.hide_grid is FALSE: | |
220 | self.DC.DrawLine(x1, y1, x1, y2) | |
221 | self.gridx.append(x1) | |
222 | x1 = x1 + self.dl_w | |
223 | ||
224 | x1 = self.x_st | |
225 | y1 = self.y_st | |
226 | x2 = self.x_st + self.cwidth | |
227 | for i in range(8): | |
228 | if self.hide_grid is FALSE: | |
229 | self.DC.DrawLine(x1, y1, x2, y1) | |
230 | self.gridy.append(y1) | |
231 | if i == 0: | |
232 | y1 = y1 + self.dl_h/2 | |
233 | else: | |
234 | y1 = y1 + self.dl_h | |
235 | ||
236 | ||
e0473f5f | 237 | class wxCalendar(wxWindow): |
944930d5 RD |
238 | def __init__(self, parent, id, pos=wxDefaultPosition, size=wxDefaultSize): |
239 | wxWindow.__init__(self, parent, id, pos, size) | |
240 | ||
241 | # set the calendar control attributes | |
242 | ||
243 | self.grid_color = 'BLACK' | |
244 | self.back_color = 'WHITE' | |
245 | self.sel_color = 'RED' | |
246 | self.high_color = 'LIGHT BLUE' | |
247 | self.font = wxSWISS | |
e0473f5f | 248 | self.bold = wxNORMAL |
944930d5 RD |
249 | |
250 | self.SetBackgroundColour(wxNamedColor(self.back_color)) | |
251 | self.Connect(-1, -1, wxEVT_LEFT_DOWN, self.OnLeftEvent) | |
252 | self.Connect(-1, -1, wxEVT_LEFT_DCLICK, self.OnLeftDEvent) | |
253 | self.Connect(-1, -1, wxEVT_RIGHT_DOWN, self.OnRightEvent) | |
254 | self.Connect(-1, -1, wxEVT_RIGHT_DCLICK, self.OnRightDEvent) | |
255 | ||
256 | self.sel_key = None # last used by | |
257 | self.sel_lst = [] # highlighted selected days | |
258 | ||
259 | self.SetNow() # default calendar for current month | |
260 | ||
261 | self.size = None | |
262 | self.hide_title = FALSE | |
263 | self.hide_grid = FALSE | |
264 | self.set_day = None | |
265 | ||
266 | # control some of the main calendar attributes | |
267 | ||
268 | def HideTitle(self): | |
269 | self.hide_title = TRUE | |
270 | ||
271 | def HideGrid(self): | |
272 | self.hide_grid = TRUE | |
273 | ||
274 | # determine the calendar rectangle click area and draw a selection | |
275 | ||
276 | def ProcessClick(self, event): | |
277 | self.x, self.y = event.GetX(), event.GetY() | |
278 | key = self.GetDayHit(self.x, self.y) | |
279 | self.SelectDay(key) | |
280 | ||
281 | # tab mouse click events and process | |
282 | ||
283 | def OnLeftEvent(self, event): | |
284 | self.click = 'LEFT' | |
285 | self.ProcessClick(event) | |
286 | ||
287 | def OnLeftDEvent(self, event): | |
288 | self.click = 'DLEFT' | |
289 | self.ProcessClick(event) | |
290 | ||
291 | def OnRightEvent(self, event): | |
292 | self.click = 'RIGHT' | |
293 | self.ProcessClick(event) | |
294 | ||
295 | def OnRightDEvent(self, event): | |
296 | self.click = 'DRIGHT' | |
297 | self.ProcessClick(event) | |
298 | ||
299 | def SetSize(self, set_size): | |
300 | self.size = set_size | |
301 | ||
302 | def SetSelDay(self, sel): | |
303 | self.sel_lst = sel # list of highlighted days | |
304 | ||
305 | # get the current date | |
306 | ||
307 | def SetNow(self): | |
308 | dt = now() | |
309 | self.month = dt.month | |
310 | self.year = dt.year | |
311 | self.day = dt.day | |
312 | ||
313 | # set the current day | |
314 | ||
315 | def SetCurrentDay(self): | |
316 | self.SetNow() | |
317 | self.set_day = self.day | |
318 | ||
319 | # get the date, day, month, year set in calendar | |
320 | ||
321 | def GetDate(self): | |
322 | return self.day, self.month, self.year | |
323 | ||
324 | def GetDay(self): | |
325 | return self.day | |
326 | ||
327 | def GetMonth(self): | |
328 | return self.month | |
329 | ||
330 | def GetYear(self): | |
331 | return self.year | |
332 | ||
333 | # set the day, month, and year | |
334 | ||
335 | def SetDayValue(self, day): | |
336 | self.set_day = day | |
337 | ||
338 | def SetMonth(self, month): | |
339 | if month >= 1 and month <= 12: | |
340 | self.month = month | |
341 | else: | |
342 | self.month = 1 | |
343 | self.set_day = None | |
344 | ||
345 | def SetYear(self, year): | |
346 | self.year = year | |
347 | ||
348 | # increment year and month | |
349 | ||
350 | def IncYear(self): | |
351 | self.year = self.year + 1 | |
352 | self.set_day = None | |
353 | ||
354 | def DecYear(self): | |
355 | self.year = self.year - 1 | |
356 | self.set_day = None | |
357 | ||
358 | def IncMonth(self): | |
359 | self.month = self.month + 1 | |
360 | if self.month > 12: | |
361 | self.month = 1 | |
362 | self.year = self.year + 1 | |
363 | self.set_day = None | |
364 | ||
365 | def DecMonth(self): | |
366 | self.month = self.month - 1 | |
367 | if self.month < 1: | |
368 | self.month = 12 | |
369 | self.year = self.year - 1 | |
370 | self.set_day = None | |
371 | ||
372 | # test to see if the selection has a date and create event | |
373 | ||
374 | def TestDay(self, key): | |
375 | try: | |
376 | self.day = int(self.cal[key]) | |
377 | except: | |
378 | return None | |
379 | if self.day == "": | |
380 | return None | |
381 | else: | |
382 | evt = wxPyCommandEvent(2100, self.GetId()) | |
383 | evt.click, evt.day, evt.month, evt.year = self.click, self.day, self.month, self.year | |
384 | self.GetEventHandler().ProcessEvent(evt) | |
385 | ||
386 | self.set_day = self.day | |
387 | return key | |
388 | ||
389 | # find the clicked area rectangle | |
390 | ||
391 | def GetDayHit(self, mx, my): | |
392 | for key in self.rg.keys(): | |
393 | val = self.rg[key] | |
394 | rt = wxRegion() | |
3af4e610 | 395 | rt.UnionRect(val) |
944930d5 RD |
396 | if rt.Contains(mx, my) != 0: |
397 | result = self.TestDay(key) | |
398 | return result | |
399 | return None | |
400 | ||
401 | # calendar drawing | |
402 | ||
403 | def OnPaint(self, event): | |
404 | DC = wxPaintDC(self) | |
405 | self.DoDrawing(DC) | |
406 | ||
407 | def DoDrawing(self, DC): | |
408 | DC = wxPaintDC(self) | |
409 | DC.BeginDrawing() | |
410 | ||
411 | self.cal = cal = CalDraw() | |
412 | if self.size is None: | |
413 | size = self.GetClientSize() | |
414 | else: | |
415 | size = self.size | |
416 | ||
417 | # drawing attributes | |
418 | ||
419 | cal.hide_title = self.hide_title | |
420 | cal.hide_grid = self.hide_grid | |
421 | ||
422 | cal.grid_color = self.grid_color | |
423 | cal.high_color = self.high_color | |
424 | cal.back_color = self.back_color | |
425 | cal.font = self.font | |
e0473f5f | 426 | cal.bold = self.bold |
944930d5 RD |
427 | |
428 | cal.SetSize(size) | |
429 | cal.SetCal(self.year, self.month) | |
430 | cal.DrawCal(DC, self.sel_lst) | |
431 | ||
432 | self.rg = cal.GetRect() | |
433 | self.cal = cal.GetCal() | |
434 | self.st_pos = cal.GetOffset() | |
435 | self.ymax = DC.MaxY() | |
436 | ||
437 | if self.set_day != None: | |
438 | self.SetDay(self.set_day) | |
439 | DC.EndDrawing() | |
440 | ||
441 | # draw the selection rectangle | |
442 | ||
443 | def DrawRect(self, key, color = 'BLACK', width = 0): | |
444 | if key == None: | |
445 | return | |
446 | DC = wxClientDC(self) | |
447 | DC.BeginDrawing() | |
448 | ||
449 | brush = wxBrush(wxColour(0, 0xFF, 0x80), wxTRANSPARENT) | |
450 | DC.SetBrush(brush) | |
451 | DC.SetPen(wxPen(wxNamedColour(color), width)) | |
452 | ||
453 | rect = self.rg[key] | |
454 | DC.DrawRectangle(rect.x, rect.y, rect.width+1, rect.height+1) | |
455 | ||
456 | DC.EndDrawing() | |
457 | ||
458 | # set the day selection | |
459 | ||
460 | def SetDay(self, day): | |
461 | day = day + self.st_pos - 1 | |
462 | self.SelectDay(day) | |
463 | ||
464 | def SelectDay(self, key): | |
465 | sel_size = 1 | |
466 | self.DrawRect(self.sel_key, self.back_color, sel_size) # clear large selection | |
467 | if self.hide_grid is FALSE: | |
468 | self.DrawRect(self.sel_key, self.grid_color) | |
469 | ||
470 | self.DrawRect(key, self.sel_color, sel_size) | |
471 | self.sel_key = key # store last used by | |
472 | self.select_day = None | |
473 | ||
474 | def ClearDsp(self): | |
475 | self.Clear() | |
476 | ||
477 |