]> git.saurik.com Git - wxWidgets.git/blame - wxPython/wx/lib/calendar.py
Use "import wx". wxPython.wx now has backwards compatible DC's and so
[wxWidgets.git] / wxPython / wx / lib / calendar.py
CommitLineData
d14a1e28
RD
1#----------------------------------------------------------------------------
2# Name: calendar.py
3# Purpose: Calendar display control
4#
5# Author: Lorne White (email: lorne.white@telusplanet.net)
6#
7# Created:
8# Version 0.92
9# Date: Nov 26, 2001
10# Licence: wxWindows license
11#----------------------------------------------------------------------------
b881fc78
RD
12# 12/01/2003 - Jeff Grimmett (grimmtooth@softhome.net)
13#
14# o Updated for wx namespace
15# o Tested with updated demo
16# o Added new event type EVT_CALENDAR. The reason for this is that the original
17# library used a hardcoded ID of 2100 for generating events. This makes it
18# very difficult to fathom when trying to decode the code since there's no
19# published API. Creating the new event binder might seem like overkill -
20# after all, you might ask, why not just use a new event ID and be done with
21# it? However, a consistent interface is very useful at times; also it makes
22# it clear that we're not just hunting for mouse clicks -- we're hunting
23# wabbit^H^H^H^H (sorry bout that) for calender-driven mouse clicks. So
24# that's my sad story. Shoot me if you must :-)
25# o There's still one deprecation warning buried in here somewhere, but I
26# haven't been able to find it yet. It only occurs when displaying a
27# print preview, and only the first time. It *could* be an error in the
28# demo, I suppose.
29#
30# Here's the traceback:
31#
32# C:\Python\lib\site-packages\wx\core.py:949: DeprecationWarning:
33# integer argument expected, got float
34# newobj = _core.new_Rect(*args, **kwargs)
35#
c1ea08d3
RD
36# 12/17/2003 - Jeff Grimmett (grimmtooth@softhome.net)
37#
38# o A few style-guide nips and tucks
39# o Renamed wxCalendar to Calendar
40# o Couple of bugfixes
41#
d14a1e28 42
b881fc78 43import wx
d14a1e28
RD
44
45from CDate import *
46
d14a1e28
RD
47CalDays = [6, 0, 1, 2, 3, 4, 5]
48AbrWeekday = {6:"Sun", 0:"Mon", 1:"Tue", 2:"Wed", 3:"Thu", 4:"Fri", 5:"Sat"}
49_MIDSIZE = 180
50
51BusCalDays = [0, 1, 2, 3, 4, 5, 6]
52
b881fc78
RD
53# Calendar click event - added 12/1/03 by jmg (see above)
54wxEVT_COMMAND_PYCALENDAR_DAY_CLICKED = wx.NewEventType()
55EVT_CALENDAR = wx.PyEventBinder(wxEVT_COMMAND_PYCALENDAR_DAY_CLICKED, 1)
d14a1e28
RD
56
57def GetMonthList():
58 monthlist = []
59 for i in range(13):
60 name = Month[i]
61 if name != None:
62 monthlist.append(name)
63 return monthlist
64
b881fc78
RD
65# calendar drawing routing
66
d14a1e28
RD
67class CalDraw:
68 def __init__(self, parent):
69 self.pwidth = 1
70 self.pheight = 1
71 try:
72 self.scale = parent.scale
73 except:
74 self.scale = 1
75
76 self.DefParms()
77
78 def DefParms(self):
c1ea08d3
RD
79 self.num_auto = True # auto scale of the cal number day size
80 self.num_size = 12 # default size of calendar if no auto size
81 self.max_num_size = 12 # maximum size for calendar number
d14a1e28 82
b881fc78
RD
83 self.num_align_horz = wx.ALIGN_CENTRE # alignment of numbers
84 self.num_align_vert = wx.ALIGN_CENTRE
d14a1e28
RD
85 self.num_indent_horz = 0 # points indent from position, used to offset if not centered
86 self.num_indent_vert = 0
87
88 self.week_auto = True # auto scale of week font text
89 self.week_size = 10
90 self.max_week_size = 12
91
92 self.grid_color = 'BLACK' # grid and selection colors
93 self.back_color = 'WHITE'
94 self.sel_color = 'RED'
95
96 self.high_color = 'LIGHT BLUE'
97 self.border_color = 'BLACK'
98 self.week_color = 'LIGHT GREY'
99
100 self.week_font_color = 'BLACK' # font colors
101 self.day_font_color = 'BLACK'
102
b881fc78
RD
103 self.font = wx.SWISS
104 self.bold = wx.NORMAL
d14a1e28
RD
105
106 self.hide_title = False
107 self.hide_grid = False
108 self.outer_border = True
109
110 self.title_offset = 0
111 self.cal_week_scale = 0.7
112 self.show_weekend = False
113 self.cal_type = "NORMAL"
114
115 def SetWeekColor(self, font_color, week_color): # set font and background color for week title
116 self.week_font_color = font_color
117 self.week_color = week_color
118
119 def SetSize(self, size):
b881fc78
RD
120 self.set_sizew = size[0]
121 self.set_sizeh = size[1]
d14a1e28
RD
122
123 def InitValues(self): # default dimensions of various elements of the calendar
124 self.rg = {}
125 self.cal_sel = {}
126 self.set_cy_st = 0 # start position
127 self.set_cx_st = 0
128
129 self.set_y_mrg = 15 # start of vertical draw default
130 self.set_x_mrg = 10
131 self.set_y_end = 10
132
133 def SetPos(self, xpos, ypos):
134 self.set_cx_st = xpos
135 self.set_cy_st = ypos
136
137 def SetMarg(self, xmarg, ymarg):
138 self.set_x_st = xmarg
139 self.set_y_st = ymarg
140 self.set_y_end = ymarg
141
142 def InitScale(self): # scale position values
c8431295
RD
143 self.sizew = int(self.set_sizew * self.pwidth)
144 self.sizeh = int(self.set_sizeh * self.pheight)
d14a1e28 145
c8431295
RD
146 self.cx_st = int(self.set_cx_st * self.pwidth) # draw start position
147 self.cy_st = int(self.set_cy_st * self.pheight)
d14a1e28 148
c8431295
RD
149 self.x_mrg = int(self.set_x_mrg * self.pwidth) # calendar draw margins
150 self.y_mrg = int(self.set_y_mrg * self.pheight)
151 self.y_end = int(self.set_y_end * self.pheight)
d14a1e28
RD
152
153 def DrawCal(self, DC, sel_lst=[]):
154 self.DC = DC
155 self.InitScale()
156
157 self.DrawBorder()
b881fc78 158
d14a1e28
RD
159 if self.hide_title is False:
160 self.DrawMonth()
161
162 self.Center()
163
164 self.DrawGrid()
165 self.GetRect()
166
167 if self.show_weekend is True: # highlight weekend dates
168 self.SetWeekEnd()
169
170 self.AddSelect(sel_lst) # overrides the weekend highlight
171
172 self.DrawSel() # highlighted days
173 self.DrawWeek()
174 self.DrawNum()
175
176 def AddSelect(self, list, cfont=None, cbackgrd = None):
177 if cfont is None:
178 cfont = self.sel_color # font digit color
b881fc78 179
d14a1e28
RD
180 if cbackgrd is None:
181 cbackgrd = self.high_color # select background color
182
183 for val in list:
184 self.cal_sel[val] = (cfont, cbackgrd)
185
b881fc78
RD
186 # draw border around the outside of the main display rectangle
187 def DrawBorder(self):
188 brush = wx.Brush(wx.NamedColour(self.back_color), wx.SOLID)
d14a1e28 189 self.DC.SetBrush(brush)
b881fc78 190 self.DC.SetPen(wx.Pen(wx.NamedColour(self.border_color), 1))
d14a1e28
RD
191
192 if self.outer_border is True:
b881fc78
RD
193 # full display window area
194 rect = wx.Rect(self.cx_st, self.cy_st, self.sizew, self.sizeh)
fd3f2efe 195 self.DC.DrawRectangleRect(rect)
d14a1e28
RD
196
197 def DrawNumVal(self):
198 self.DrawNum()
199
c1ea08d3 200 # calculate the calendar days and offset position
d14a1e28
RD
201
202 def SetCal(self, year, month):
203 self.InitValues() # reset initial values
204
205 self.year = year
206 self.month = month
207
208 day = 1
209 t = Date(year, month, day)
210 dow = self.dow = t.day_of_week # start day in month
211 dim = self.dim = t.days_in_month # number of days in month
b881fc78 212
d14a1e28
RD
213 if self.cal_type == "NORMAL":
214 start_pos = dow+1
215 else:
216 start_pos = dow
217
218 self.st_pos = start_pos
219
220 self.cal = []
221 for i in range(start_pos):
222 self.cal.append('')
b881fc78 223
d14a1e28
RD
224 i = 1
225 while i <= dim:
226 self.cal.append(str(i))
227 i = i + 1
b881fc78 228
d14a1e28
RD
229 return start_pos
230
231 def SetWeekEnd(self, font_color='BLACK', backgrd = 'LIGHT GREY'):
232 date = 6 - int(self.dow) # start day of first saturday
b881fc78 233
d14a1e28
RD
234 while date <= self.dim:
235 self.cal_sel[date] = (font_color, backgrd) # Saturday
236 date = date + 1
b881fc78 237
d14a1e28
RD
238 if date <= self.dim:
239 self.cal_sel[date] = (font_color, backgrd) # Sunday
240 date = date + 6
241 else:
242 date = date + 7
243
c1ea08d3
RD
244 # get the display rectange list of the day grid
245 def GetRect(self):
d14a1e28
RD
246 cnt = 0
247 for y in self.gridy[1:-1]:
248 for x in self.gridx[:-1]:
c8431295
RD
249 assert type(y) == int
250 assert type(x) == int
b881fc78 251 rect = wx.Rect(x, y, self.dl_w, self.dl_h) # create rect region
d14a1e28
RD
252 self.rg[cnt] = rect
253 cnt = cnt + 1
b881fc78 254
d14a1e28
RD
255 return self.rg
256
257 def GetCal(self):
258 return self.cal
259
260 def GetOffset(self):
261 return self.st_pos
262
c1ea08d3
RD
263 # month and year title
264 def DrawMonth(self):
d14a1e28
RD
265 month = Month[self.month]
266
267 sizef = 11
268 if self.sizeh < _MIDSIZE:
269 sizef = 10
270
b881fc78 271 f = wx.Font(sizef, self.font, wx.NORMAL, self.bold)
d14a1e28
RD
272 self.DC.SetFont(f)
273
274 tw,th = self.DC.GetTextExtent(month)
275 adjust = self.cx_st + (self.sizew-tw)/2
fd3f2efe 276 self.DC.DrawText(month, (adjust, self.cy_st + th))
d14a1e28
RD
277
278 year = str(self.year)
279 tw,th = self.DC.GetTextExtent(year)
280 adjust = self.sizew - tw - self.x_mrg
281
282 self.title_offset = th * 2
283
b881fc78 284 f = wx.Font(sizef, self.font, wx.NORMAL, self.bold)
d14a1e28 285 self.DC.SetFont(f)
fd3f2efe 286 self.DC.DrawText(year, (self.cx_st + adjust, self.cy_st + th))
d14a1e28
RD
287
288 def DrawWeek(self): # draw the week days
289 width = self.gridx[1]-self.gridx[0]
290 height = self.gridy[1] - self.gridy[0]
291 rect_w = self.gridx[7]-self.gridx[0]
292
b881fc78
RD
293 f = wx.Font(10, self.font, wx.NORMAL, self.bold) # initial font setting
294
d14a1e28
RD
295 if self.week_auto == True:
296 test_size = self.max_week_size # max size
297 test_day = ' Sun '
298 while test_size > 2:
299 f.SetPointSize(test_size)
300 self.DC.SetFont(f)
301 tw,th = self.DC.GetTextExtent(test_day)
b881fc78 302
d14a1e28
RD
303 if tw < width and th < height:
304 break
b881fc78 305
d14a1e28
RD
306 test_size = test_size - 1
307 else:
308 f.SetPointSize(self.week_size) # set fixed size
309 self.DC.SetFont(f)
310
b881fc78 311 self.DC.SetTextForeground(wx.NamedColour(self.week_font_color))
d14a1e28
RD
312
313 cnt_x = 0
314 cnt_y = 0
315
b881fc78 316 brush = wx.Brush(wx.NamedColour(self.week_color), wx.SOLID)
d14a1e28 317 self.DC.SetBrush(brush)
b881fc78 318 self.DC.DrawRectangle((self.gridx[0], self.gridy[0]), (rect_w+1, height))
d14a1e28
RD
319
320 if self.cal_type == "NORMAL":
321 cal_days = CalDays
322 else:
323 cal_days = BusCalDays
324
325 for val in cal_days:
326 day = AbrWeekday[val]
b881fc78 327
d14a1e28
RD
328 if self.sizew < 200:
329 day = day[0]
b881fc78 330
d14a1e28
RD
331 dw,dh = self.DC.GetTextExtent(day)
332 diffx = (width-dw)/2
333 diffy = (height-dh)/2
334
335 x = self.gridx[cnt_x]
336 y = self.gridy[cnt_y]
fd3f2efe
RD
337 self.DC.DrawRectangle((self.gridx[cnt_x], self.gridy[0]), (width+1, height))
338 self.DC.DrawText(day, (x+diffx, y+diffy))
d14a1e28
RD
339 cnt_x = cnt_x + 1
340
c1ea08d3
RD
341 # draw the day numbers
342 def DrawNum(self):
b881fc78
RD
343 f = wx.Font(10, self.font, wx.NORMAL, self.bold) # initial font setting
344
d14a1e28
RD
345 if self.num_auto == True:
346 test_size = self.max_num_size # max size
347 test_day = ' 99 '
b881fc78 348
d14a1e28
RD
349 while test_size > 2:
350 f.SetPointSize(test_size)
351 self.DC.SetFont(f)
352 tw,th = self.DC.GetTextExtent(test_day)
b881fc78 353
d14a1e28
RD
354 if tw < self.dl_w and th < self.dl_h:
355 sizef = test_size
356 break
357 test_size = test_size - 1
358 else:
359 f.SetPointSize(self.num_size) # set fixed size
360 self.DC.SetFont(f)
361
362 cnt_x = 0
363 cnt_y = 1
364 for val in self.cal:
365 x = self.gridx[cnt_x]
366 y = self.gridy[cnt_y]
367
368 try:
369 num_val = int(val)
370 num_color = self.cal_sel[num_val][0]
371 except:
372 num_color = self.day_font_color
373
b881fc78 374 self.DC.SetTextForeground(wx.NamedColour(num_color))
d14a1e28
RD
375 self.DC.SetFont(f)
376
377 tw,th = self.DC.GetTextExtent(val)
b881fc78
RD
378
379 if self.num_align_horz == wx.ALIGN_CENTRE:
d14a1e28 380 adj_h = (self.dl_w - tw)/2
b881fc78 381 elif self.num_align_horz == wx.ALIGN_RIGHT:
d14a1e28
RD
382 adj_h = self.dl_w - tw
383 else:
384 adj_h = 0 # left alignment
385
386 adj_h = adj_h + self.num_indent_horz
387
b881fc78 388 if self.num_align_vert == wx.ALIGN_CENTRE:
d14a1e28 389 adj_v = (self.dl_h - th)/2
b881fc78 390 elif self.num_align_horz == wx.ALIGN_RIGHT:
d14a1e28
RD
391 adj_v = self.dl_h - th
392 else:
393 adj_v = 0 # left alignment
394
395 adj_v = adj_v + self.num_indent_vert
396
fd3f2efe 397 self.DC.DrawText(val, (x+adj_h, y+adj_v))
b881fc78 398
d14a1e28
RD
399 if cnt_x < 6:
400 cnt_x = cnt_x + 1
401 else:
402 cnt_x = 0
403 cnt_y = cnt_y + 1
404
c1ea08d3
RD
405 # calculate the dimensions in the center of the drawing area
406 def Center(self):
d14a1e28
RD
407 bdw = self.x_mrg * 2
408 bdh = self.y_mrg + self.y_end + self.title_offset
409
410 self.dl_w = int((self.sizew-bdw)/7)
411 self.dl_h = int((self.sizeh-bdh)/7)
412
c1ea08d3
RD
413 # week title adjustment
414 self.dl_th = int(self.dl_h*self.cal_week_scale)
d14a1e28
RD
415 self.cwidth = self.dl_w * 7
416 self.cheight = self.dl_h * 6 + self.dl_th
417
c1ea08d3
RD
418 # highlighted selected days
419 def DrawSel(self):
d14a1e28
RD
420 for key in self.cal_sel.keys():
421 sel_color = self.cal_sel[key][1]
b881fc78 422 brush = wx.Brush(wx.NamedColour(sel_color), wx.SOLID)
d14a1e28
RD
423 self.DC.SetBrush(brush)
424
425 if self.hide_grid is False:
b881fc78 426 self.DC.SetPen(wx.Pen(wx.NamedColour(self.grid_color), 0))
d14a1e28 427 else:
b881fc78
RD
428 self.DC.SetPen(wx.Pen(wx.NamedColour(self.back_color), 0))
429
d14a1e28
RD
430 nkey = key + self.st_pos -1
431 rect = self.rg[nkey]
fd3f2efe 432 self.DC.DrawRectangle((rect.x, rect.y), (rect.width+1, rect.height+1))
d14a1e28 433
c1ea08d3
RD
434 # calculate and draw the grid lines
435 def DrawGrid(self):
b881fc78 436 self.DC.SetPen(wx.Pen(wx.NamedColour(self.grid_color), 0))
d14a1e28
RD
437
438 self.gridx = []
439 self.gridy = []
440
441 self.x_st = self.cx_st + self.x_mrg
c1ea08d3
RD
442 # start postion of draw
443 self.y_st = self.cy_st + self.y_mrg + self.title_offset
d14a1e28
RD
444
445 x1 = self.x_st
446 y1 = self.y_st
d14a1e28 447 y2 = y1 + self.cheight
b881fc78 448
d14a1e28
RD
449 for i in range(8):
450 if self.hide_grid is False:
fd3f2efe 451 self.DC.DrawLine((x1, y1), (x1, y2))
d14a1e28
RD
452 self.gridx.append(x1)
453 x1 = x1 + self.dl_w
454
455 x1 = self.x_st
456 y1 = self.y_st
d14a1e28 457 x2 = x1 + self.cwidth
b881fc78 458
d14a1e28
RD
459 for i in range(8):
460 if self.hide_grid is False:
fd3f2efe 461 self.DC.DrawLine((x1, y1), (x2, y1))
b881fc78 462
d14a1e28 463 self.gridy.append(y1)
b881fc78 464
d14a1e28
RD
465 if i == 0:
466 y1 = y1 + self.dl_th
467 else:
468 y1 = y1 + self.dl_h
469
470
471class PrtCalDraw(CalDraw):
472 def InitValues(self):
473 self.rg = {}
474 self.cal_sel = {}
c1ea08d3
RD
475 # start draw border location
476 self.set_cx_st = 1.0
d14a1e28
RD
477 self.set_cy_st = 1.0
478
c1ea08d3
RD
479 # draw offset position
480 self.set_y_mrg = 0.2
d14a1e28
RD
481 self.set_x_mrg = 0.2
482 self.set_y_end = 0.2
483
b881fc78
RD
484 # calculate the dimensions in the center of the drawing area
485 def SetPSize(self, pwidth, pheight):
d14a1e28
RD
486 self.pwidth = int(pwidth)/self.scale
487 self.pheight = int(pheight)/self.scale
488
489 def SetPreview(self, preview):
490 self.preview = preview
491
c1ea08d3 492class Calendar(wx.Window):
b881fc78
RD
493 def __init__(self, parent, id, pos=wx.DefaultPosition, size=wx.DefaultSize):
494 wx.Window.__init__(self, parent, id, pos, size)
d14a1e28 495
c1ea08d3 496 # set the calendar control attributes
d14a1e28
RD
497
498 self.grid_color = 'BLACK'
499 self.back_color = 'WHITE'
500 self.hide_grid = False
501 self.sel_color = 'RED'
502 self.hide_title = False
503 self.show_weekend = False
504 self.cal_type = "NORMAL"
505
c1ea08d3 506 # font colors
d14a1e28 507 self.week_color = 'LIGHT GREY'
c1ea08d3 508 self.week_font_color = 'BLACK'
d14a1e28
RD
509
510 self.select_list = []
511
b881fc78
RD
512 self.SetBackgroundColour(self.back_color)
513 self.Bind(wx.EVT_LEFT_DOWN, self.OnLeftEvent)
514 self.Bind(wx.EVT_LEFT_DCLICK, self.OnLeftDEvent)
515 self.Bind(wx.EVT_RIGHT_DOWN, self.OnRightEvent)
516 self.Bind(wx.EVT_RIGHT_DCLICK, self.OnRightDEvent)
d14a1e28
RD
517
518 self.sel_key = None # last used by
519 self.sel_lst = [] # highlighted selected days
520
c1ea08d3
RD
521 # default calendar for current month
522 self.SetNow()
d14a1e28
RD
523
524 self.size = None
525 self.set_day = None
526
b881fc78
RD
527 self.Bind(wx.EVT_PAINT, self.OnPaint)
528 self.Bind(wx.EVT_SIZE, self.OnSize)
d14a1e28 529
c1ea08d3 530 # control some of the main calendar attributes
d14a1e28
RD
531
532 def HideTitle(self):
533 self.hide_title = True
534
535 def HideGrid(self):
536 self.hide_grid = True
537
c1ea08d3 538 # determine the calendar rectangle click area and draw a selection
d14a1e28
RD
539
540 def ProcessClick(self, event):
541 self.x, self.y = event.GetX(), event.GetY()
542 key = self.GetDayHit(self.x, self.y)
543 self.SelectDay(key)
544
c1ea08d3 545 # tab mouse click events and process
d14a1e28
RD
546
547 def OnLeftEvent(self, event):
548 self.click = 'LEFT'
549 self.shiftkey = event.ShiftDown()
550 self.ctrlkey = event.ControlDown()
551 self.ProcessClick(event)
552
553 def OnLeftDEvent(self, event):
554 self.click = 'DLEFT'
555 self.ProcessClick(event)
556
557 def OnRightEvent(self, event):
558 self.click = 'RIGHT'
559 self.ProcessClick(event)
560
561 def OnRightDEvent(self, event):
562 self.click = 'DRIGHT'
563 self.ProcessClick(event)
564
565 def SetSize(self, set_size):
566 self.size = set_size
567
568 def SetSelDay(self, sel):
c1ea08d3
RD
569 # list of highlighted days
570 self.sel_lst = sel
d14a1e28 571
c1ea08d3 572 # get the current date
d14a1e28
RD
573 def SetNow(self):
574 dt = now()
575 self.month = dt.month
576 self.year = dt.year
577 self.day = dt.day
578
c1ea08d3 579 # set the current day
d14a1e28
RD
580 def SetCurrentDay(self):
581 self.SetNow()
582 self.set_day = self.day
583
c1ea08d3 584 # get the date, day, month, year set in calendar
d14a1e28
RD
585
586 def GetDate(self):
587 return self.day, self.month, self.year
588
589 def GetDay(self):
590 return self.day
591
592 def GetMonth(self):
593 return self.month
594
595 def GetYear(self):
596 return self.year
597
c1ea08d3 598 # set the day, month, and year
d14a1e28
RD
599
600 def SetDayValue(self, day):
601 self.set_day = day
602
603 def SetMonth(self, month):
604 if month >= 1 and month <= 12:
605 self.month = month
606 else:
607 self.month = 1
608 self.set_day = None
609
610 def SetYear(self, year):
611 self.year = year
612
c1ea08d3 613 # increment year and month
d14a1e28
RD
614
615 def IncYear(self):
616 self.year = self.year + 1
617 self.set_day = None
618
619 def DecYear(self):
620 self.year = self.year - 1
621 self.set_day = None
622
623 def IncMonth(self):
624 self.month = self.month + 1
625 if self.month > 12:
626 self.month = 1
627 self.year = self.year + 1
628 self.set_day = None
629
630 def DecMonth(self):
631 self.month = self.month - 1
632 if self.month < 1:
633 self.month = 12
634 self.year = self.year - 1
635 self.set_day = None
636
c1ea08d3 637 # test to see if the selection has a date and create event
d14a1e28
RD
638
639 def TestDay(self, key):
640 try:
641 self.day = int(self.cal[key])
642 except:
643 return None
644 if self.day == "":
645 return None
646 else:
b881fc78
RD
647 # Changed 12/1/03 by jmg (see above) to support 2.5 event binding
648 evt = wx.PyCommandEvent(wxEVT_COMMAND_PYCALENDAR_DAY_CLICKED, self.GetId())
d14a1e28
RD
649 evt.click, evt.day, evt.month, evt.year = self.click, self.day, self.month, self.year
650 evt.shiftkey = self.shiftkey
651 evt.ctrlkey = self.ctrlkey
652 self.GetEventHandler().ProcessEvent(evt)
653
654 self.set_day = self.day
655 return key
656
c1ea08d3 657 # find the clicked area rectangle
d14a1e28
RD
658
659 def GetDayHit(self, mx, my):
660 for key in self.rg.keys():
661 val = self.rg[key]
b881fc78
RD
662 ms_rect = wx.Rect(mx, my, 1, 1)
663 if wx.IntersectRect(ms_rect, val) is not None:
d14a1e28
RD
664 result = self.TestDay(key)
665 return result
b881fc78 666
d14a1e28
RD
667 return None
668
c1ea08d3 669 # calendar drawing
d14a1e28 670
c1ea08d3
RD
671 def SetWeekColor(self, font_color, week_color):
672 # set font and background color for week title
d14a1e28
RD
673 self.week_font_color = font_color
674 self.week_color = week_color
675
676 def AddSelect(self, list, font_color, back_color):
677 list_val = [list, font_color, back_color]
678 self.select_list.append(list_val)
679
680 def ShowWeekEnd(self):
c1ea08d3
RD
681 # highlight weekend
682 self.show_weekend = True
d14a1e28
RD
683
684 def SetBusType(self):
685 self.cal_type = "BUS"
686
687 def OnSize(self, evt):
688 self.Refresh(False)
689 evt.Skip()
690
691 def OnPaint(self, event):
b881fc78 692 DC = wx.PaintDC(self)
d14a1e28
RD
693 self.DoDrawing(DC)
694
695 def DoDrawing(self, DC):
b881fc78 696 DC = wx.PaintDC(self)
d14a1e28
RD
697 DC.BeginDrawing()
698
699 self.cal = cal = CalDraw(self)
700
701 cal.grid_color = self.grid_color
702 cal.back_color = self.back_color
703 cal.hide_grid = self.hide_grid
704 cal.grid_color = self.grid_color
705 cal.hide_title = self.hide_title
706 cal.show_weekend = self.show_weekend
707 cal.cal_type = self.cal_type
708
709 if self.size is None:
710 size = self.GetClientSize()
711 else:
712 size = self.size
713
c1ea08d3 714 # drawing attributes
d14a1e28
RD
715
716 cal.week_font_color = self.week_font_color
717 cal.week_color = self.week_color
718
719 cal.SetSize(size)
720 cal.SetCal(self.year, self.month)
b881fc78 721
d14a1e28
RD
722 for val in self.select_list:
723 cal.AddSelect(val[0], val[1], val[2])
724
725 cal.DrawCal(DC, self.sel_lst)
726
727 self.rg = cal.GetRect()
728 self.cal = cal.GetCal()
729 self.st_pos = cal.GetOffset()
730 self.ymax = DC.MaxY()
731
732 if self.set_day != None:
733 self.SetDay(self.set_day)
b881fc78 734
d14a1e28
RD
735 DC.EndDrawing()
736
c1ea08d3 737 # draw the selection rectangle
d14a1e28
RD
738 def DrawRect(self, key, color = 'BLACK', width = 0):
739 if key == None:
740 return
b881fc78
RD
741
742 DC = wx.ClientDC(self)
d14a1e28
RD
743 DC.BeginDrawing()
744
b881fc78 745 brush = wx.Brush(wx.Colour(0, 0xFF, 0x80), wx.TRANSPARENT)
d14a1e28 746 DC.SetBrush(brush)
b881fc78 747 DC.SetPen(wx.Pen(wx.NamedColour(color), width))
d14a1e28
RD
748
749 rect = self.rg[key]
fd3f2efe 750 DC.DrawRectangle((rect.x, rect.y), (rect.width+1, rect.height+1))
d14a1e28
RD
751
752 DC.EndDrawing()
753
c1ea08d3 754 # set the day selection
d14a1e28
RD
755
756 def SetDay(self, day):
757 day = day + self.st_pos - 1
758 self.SelectDay(day)
759
760 def SelectDay(self, key):
761 sel_size = 1
c1ea08d3
RD
762 # clear large selection
763 self.DrawRect(self.sel_key, self.back_color, sel_size)
b881fc78 764
d14a1e28
RD
765 if self.hide_grid is False:
766 self.DrawRect(self.sel_key, self.grid_color)
767
768 self.DrawRect(key, self.sel_color, sel_size)
c1ea08d3
RD
769 # store last used by
770 self.sel_key = key
d14a1e28
RD
771 self.select_day = None
772
773 def ClearDsp(self):
774 self.Clear()
775
b881fc78 776class CalenDlg(wx.Dialog):
d14a1e28 777 def __init__(self, parent, month=None, day = None, year=None):
b881fc78 778 wx.Dialog.__init__(self, parent, -1, "Event Calendar", wx.DefaultPosition, (280, 360))
d14a1e28 779
c1ea08d3
RD
780 # set the calendar and attributes
781 self.calend = Calendar(self, -1, (20, 60), (240, 200))
b881fc78 782
d14a1e28
RD
783 if month == None:
784 self.calend.SetCurrentDay()
785 start_month = self.calend.GetMonth()
786 start_year = self.calend.GetYear()
787 else:
788 self.calend.month = start_month = month
789 self.calend.year = start_year = year
790 self.calend.SetDayValue(day)
791
792 self.calend.HideTitle()
793 self.ResetDisplay()
794
c1ea08d3 795 # get month list from DateTime
d14a1e28
RD
796 monthlist = GetMonthList()
797
c1ea08d3 798 # select the month
b881fc78
RD
799 self.date = wx.ComboBox(self, -1, Month[start_month], (20, 20), (90, -1),
800 monthlist, wx.CB_DROPDOWN)
801 self.Bind(wx.EVT_COMBOBOX, self.EvtComboBox, self.date)
d14a1e28 802
c1ea08d3 803 # alternate spin button to control the month
d14a1e28 804 h = self.date.GetSize().height
b881fc78 805 self.m_spin = wx.SpinButton(self, -1, (130, 20), (h*2, h), wx.SP_VERTICAL)
d14a1e28
RD
806 self.m_spin.SetRange(1, 12)
807 self.m_spin.SetValue(start_month)
b881fc78 808 self.Bind(wx.EVT_SPIN, self.OnMonthSpin, self.m_spin)
d14a1e28 809
c1ea08d3 810 # spin button to control the year
b881fc78 811 self.dtext = wx.TextCtrl(self, -1, str(start_year), (160, 20), (60, -1))
d14a1e28
RD
812 h = self.dtext.GetSize().height
813
b881fc78 814 self.y_spin = wx.SpinButton(self, -1, (220, 20), (h*2, h), wx.SP_VERTICAL)
d14a1e28
RD
815 self.y_spin.SetRange(1980, 2010)
816 self.y_spin.SetValue(start_year)
817
b881fc78
RD
818 self.Bind(wx.EVT_SPIN, self.OnYrSpin, self.y_spin)
819 self.Bind(EVT_CALENDAR, self.MouseClick, self.calend)
d14a1e28
RD
820
821 x_pos = 50
822 y_pos = 280
b881fc78 823 but_size = (60, 25)
d14a1e28 824
b881fc78
RD
825 btn = wx.Button(self, -1, ' Ok ', (x_pos, y_pos), but_size)
826 self.Bind(wx.EVT_BUTTON, self.OnOk, btn)
d14a1e28 827
c1ea08d3 828 btn = wx.Button(self, -1, ' Close ', (x_pos + 120, y_pos), but_size)
b881fc78 829 self.Bind(wx.EVT_BUTTON, self.OnCancel, btn)
d14a1e28
RD
830
831 def OnOk(self, event):
b881fc78 832 self.EndModal(wx.ID_OK)
d14a1e28
RD
833
834 def OnCancel(self, event):
b881fc78 835 self.EndModal(wx.ID_CANCEL)
d14a1e28 836
c1ea08d3 837 # log the mouse clicks
d14a1e28
RD
838 def MouseClick(self, evt):
839 self.month = evt.month
c1ea08d3
RD
840 # result click type and date
841 self.result = [evt.click, str(evt.day), Month[evt.month], str(evt.year)]
d14a1e28
RD
842
843 if evt.click == 'DLEFT':
b881fc78 844 self.EndModal(wx.ID_OK)
d14a1e28 845
c1ea08d3 846 # month and year spin selection routines
d14a1e28
RD
847 def OnMonthSpin(self, event):
848 month = event.GetPosition()
849 self.date.SetValue(Month[month])
850 self.calend.SetMonth(month)
851 self.calend.Refresh()
852
853 def OnYrSpin(self, event):
854 year = event.GetPosition()
855 self.dtext.SetValue(str(year))
856 self.calend.SetYear(year)
857 self.calend.Refresh()
858
859 def EvtComboBox(self, event):
860 name = event.GetString()
861 monthval = self.date.FindString(name)
862 self.m_spin.SetValue(monthval+1)
863
864 self.calend.SetMonth(monthval+1)
865 self.ResetDisplay()
866
c1ea08d3 867 # set the calendar for highlighted days
d14a1e28
RD
868
869 def ResetDisplay(self):
870 month = self.calend.GetMonth()
871 self.calend.Refresh()
872
1fded56b 873
1fded56b 874