[wxWidgets.git] / wxPython / demo / wxCalendar.py

#----------------------------------------------------------------------------
# Name:         wxCalendar.py
# Purpose:      Calendar control display testing on panel for wxPython demo
#
# Author:       Lorne White (email: lwhite1@planet.eon.net)
#
# Created:
# Version       0.8 2000/04/16
# Licence:      wxWindows license
#----------------------------------------------------------------------------

from wxPython.wx           import *
from wxPython.lib.calendar import wxCalendar, Month, PrtCalDraw

import os
dir_path = os.getcwd()


# highlighted days in month

test_days ={ 0: [],
            1: [3, 7, 9, 21],
            2: [2, 10, 4, 9],
            3: [4, 20, 29],
            4: [1, 12, 22],
            5: [2, 10, 15],
            6: [4, 8, 17],
            7: [6, 7, 8],
            8: [5, 10, 20],
            9: [1, 2, 5, 29],
           10: [2, 4, 6, 22],
           11: [6, 9, 12, 28, 29],
           12: [8, 9, 10, 11, 20] }

# test of full window calendar control functions

def GetMonthList():
    monthlist = []
    for i in range(13):
        name = Month[i]
        if name != None:
            monthlist.append(name)
    return monthlist

class TestPanel(wxPanel):
    def __init__(self, parent, log, frame):
        wxPanel.__init__(self, parent, -1)

        self.log = log
        self.frame = frame

        self.calend = wxCalendar(self, -1, wxPoint(100, 50), wxSize(200, 180))

        start_month = 11
        start_year = 1999

# month list from DateTime module

        monthlist = GetMonthList()

        self.date = wxComboBox(self, 10, Month[start_month], wxPoint(100, 20), wxSize(90, -1), monthlist, wxCB_DROPDOWN)
        EVT_COMBOBOX(self, 10, self.EvtComboBox)

# set start month and year

        self.calend.SetMonth(start_month)
        self.calend.SetYear(start_year)

# set attributes of calendar

        self.calend.hide_title = TRUE
        self.calend.HideGrid()
        self.calend.SetWeekColor('WHITE', 'BLACK')

# display routine

        self.ResetDisplay()

# mouse click event

        self.Connect(self.calend.GetId(), -1, 2100, self.MouseClick)

# scroll bar for month selection

        self.scroll = wxScrollBar(self, 40, wxPoint(100, 240), wxSize(200, 20), wxSB_HORIZONTAL)
        self.scroll.SetScrollbar(start_month-1, 1, 12, 1, TRUE)
        EVT_COMMAND_SCROLL(self, 40, self.Scroll)

# spin control for year selection

        self.dtext = wxTextCtrl(self, -1, str(start_year), wxPoint(200, 20), wxSize(60, -1))
        h = self.dtext.GetSize().height

        self.spin = wxSpinButton(self, 20, wxPoint(270, 20), wxSize(h*2, h))
        self.spin.SetRange(1980, 2010)
        self.spin.SetValue(start_year)
        EVT_SPIN(self, 20, self.OnSpin)

# button for calendar dialog test

        wxStaticText(self, -1, "Test Calendar Dialog", wxPoint(350, 50), wxSize(150, -1))

        bmp = wxBitmap('bitmaps/Calend.bmp', wxBITMAP_TYPE_BMP)
        self.but = wxBitmapButton(self, 60, bmp, wxPoint(380, 80))#, wxSize(30, 30))
        EVT_BUTTON(self, 60, self.TestDlg)

# button for calendar window test

        wxStaticText(self, -1, "Test Calendar Window", wxPoint(350, 150), wxSize(150, -1))

        bmp = wxBitmap('bitmaps/Calend.bmp', wxBITMAP_TYPE_BMP)
        self.but = wxBitmapButton(self, 160, bmp, wxPoint(380, 180))#, wxSize(30, 30))
        EVT_BUTTON(self, 160, self.TestFrame)

        wxStaticText(self, -1, "Test Calendar Print", wxPoint(350, 250), wxSize(150, -1))

        bmp = wxBitmap('bitmaps/Calend.bmp', wxBITMAP_TYPE_BMP)
        self.but = wxBitmapButton(self, 170, bmp, wxPoint(380, 280))#, wxSize(30, 30))
        EVT_BUTTON(self, 170, self.OnPreview)

# calendar dialog

    def TestDlg(self, event):
        dlg = CalenDlg(self, self.log)
        dlg.Centre()
        dlg.ShowModal()
        dlg.Destroy()

# calendar window test

    def TestFrame(self, event):
        frame = CalendFrame(self, -1, "Test Calendar", self.log)
        frame.Show(true)
        return true

# calendar print preview

    def OnPreview(self, event):
        month = self.calend.GetMonth()
        year = self.calend.GetYear()

        prt = PrintCalend(self.frame, month, year)
        prt.Preview()

# month and year control events

    def OnSpin(self, event):
        year = event.GetPosition()
        self.dtext.SetValue(str(year))
        self.calend.SetYear(year)
        self.calend.Refresh()

    def EvtComboBox(self, event):
        name = event.GetString()
        self.log.WriteText('EvtComboBox: %s\n' % name)
        monthval = self.date.FindString(name)
        self.scroll.SetScrollbar(monthval, 1, 12, 1, TRUE)

        self.calend.SetMonth(monthval+1)
        self.ResetDisplay()

    def Scroll(self, event):
        value = self.scroll.GetThumbPosition()
        monthval = int(value)+1
        self.calend.SetMonth(monthval)
        self.ResetDisplay()
        self.log.WriteText('Month: %s\n' % value)

        name = Month[monthval]
        self.date.SetValue(name)

# log mouse events

    def MouseClick(self, evt):
        text = '%s CLICK   %02d/%02d/%d' % (evt.click, evt.day, evt.month, evt.year)  # format date
        self.log.WriteText('Date Selected: ' + text + '\n')


# set the highlighted days for the calendar

    def ResetDisplay(self):
        month = self.calend.GetMonth()
        try:
            set_days = test_days[month]
        except:
            set_days = [1, 5, 12]

        self.calend.AddSelect([4, 11], 'BLUE', 'WHITE')
        self.calend.SetSelDay(set_days)
        self.calend.Refresh()

# increment and decrement toolbar controls

    def OnIncYear(self, event):
        self.calend.IncYear()
        self.ResetDisplay()

    def OnDecYear(self, event):
        self.calend.DecYear()
        self.ResetDisplay()

    def OnIncMonth(self, event):
        self.calend.IncMonth()
        self.ResetDisplay()

    def OnDecMonth(self, event):
        self.calend.DecMonth()
        self.ResetDisplay()

    def OnCurrent(self, event):
        self.calend.SetCurrentDay()
        self.ResetDisplay()

# test the calendar control in a dialog

class CalenDlg(wxDialog):
    def __init__(self, parent, log):
        self.log = log
        wxDialog.__init__(self, parent, -1, "Test Calendar", wxPyDefaultPosition, wxSize(280, 300))

        start_month = 2
        start_year = 1999

# get month list from DateTime

        monthlist = GetMonthList()

# select the month

        self.date = wxComboBox(self, 100, Month[start_month], wxPoint(20, 20), wxSize(90, -1), monthlist, wxCB_DROPDOWN)
        EVT_COMBOBOX(self, 100, self.EvtComboBox)

# alternate spin button to control the month

        h = self.date.GetSize().height
        self.m_spin = wxSpinButton(self, 120, wxPoint(120, 20), wxSize(h*2, h), wxSP_VERTICAL)
        self.m_spin.SetRange(1, 12)
        self.m_spin.SetValue(start_month)

        EVT_SPIN(self, 120, self.OnMonthSpin)

# spin button to conrol the year

        self.dtext = wxTextCtrl(self, -1, str(start_year), wxPoint(160, 20), wxSize(60, -1))
        h = self.dtext.GetSize().height

        self.y_spin = wxSpinButton(self, 20, wxPoint(220, 20), wxSize(h*2, h), wxSP_VERTICAL)
        self.y_spin.SetRange(1980, 2010)
        self.y_spin.SetValue(start_year)

        EVT_SPIN(self, 20, self.OnYrSpin)

# set the calendar and attributes

        self.calend = wxCalendar(self, -1, wxPoint(20, 60), wxSize(240, 200))
        self.calend.SetMonth(start_month)
        self.calend.SetYear(start_year)

        self.calend.HideTitle()
        self.calend.ShowWeekEnd()

        self.ResetDisplay()

        self.Connect(self.calend.GetId(), -1, 2100, self.MouseClick)

# log the mouse clicks

    def MouseClick(self, evt):
        text = '%s CLICK   %02d/%02d/%d' % (evt.click, evt.day, evt.month, evt.year)  # format date
        self.log.WriteText('Date Selected: ' + text + '\n')

        if evt.click == 'DLEFT':
            self.EndModal(wxID_OK)

# month and year spin selection routines

    def OnMonthSpin(self, event):
        month = event.GetPosition()
        if month >= 0 and month <= 12:
            self.date.SetValue(Month[month])
            self.calend.SetMonth(month)
            self.calend.Refresh()

    def OnYrSpin(self, event):
        year = event.GetPosition()
        self.dtext.SetValue(str(year))
        self.calend.SetYear(year)
        self.calend.Refresh()

    def EvtComboBox(self, event):
        name = event.GetString()
        self.log.WriteText('EvtComboBox: %s\n' % name)
        monthval = self.date.FindString(name)
        self.m_spin.SetValue(monthval+1)

        self.calend.SetMonth(monthval+1)
        self.ResetDisplay()

# set the calendar for highlighted days

    def ResetDisplay(self):
        month = self.calend.GetMonth()
        try:
            set_days = test_days[month]
        except:
            set_days = [1, 5, 12]

        self.calend.AddSelect([4, 11], 'BLUE', 'WHITE')

        self.calend.SetSelDay(set_days)
        self.calend.Refresh()

# test of full window calendar control functions

class CalendFrame(wxFrame):
    def __init__(self, parent, id, title, log):
        wxFrame.__init__(self, parent, id, title, wxPyDefaultPosition, wxSize(400, 400))
        EVT_CLOSE(self, self.OnCloseWindow)

        self.log = log
        self.CreateStatusBar()
        self.mainmenu = wxMenuBar()
        menu = wxMenu()

        menu = self.MakeFileMenu()
        self.mainmenu.Append(menu, '&File')

        self.MakeToolMenu()             # toolbar

        self.SetMenuBar(self.mainmenu)
        self.calend = wxCalendar(self, -1)
        self.calend.SetCurrentDay()
        self.calend.grid_color = 'BLUE'
        self.calend.SetBusType()
#        self.calend.ShowWeekEnd()

        self.ResetDisplay()

        self.Connect(self.calend.GetId(), -1, 2100, self.MouseClick)

    def MouseClick(self, evt):
        text = '%s CLICK   %02d/%02d/%d' % (evt.click, evt.day, evt.month, evt.year)  # format date
        self.log.WriteText('Date Selected: ' + text + '\n')

    def OnCloseWindow(self, event):
        self.Destroy()

    def ResetDisplay(self):
        month = self.calend.GetMonth()
        try:
            set_days = test_days[month]
        except:
            set_days = [1, 5, 12]

        self.calend.AddSelect([2, 16], 'GREEN', 'WHITE')

        self.calend.SetSelDay(set_days)
        self.calend.Refresh()

    def OnIncYear(self, event):
        self.calend.IncYear()
        self.ResetDisplay()

    def OnDecYear(self, event):
        self.calend.DecYear()
        self.ResetDisplay()

    def OnIncMonth(self, event):
        self.calend.IncMonth()
        self.ResetDisplay()

    def OnDecMonth(self, event):
        self.calend.DecMonth()
        self.ResetDisplay()

    def OnCurrent(self, event):
        self.calend.SetCurrentDay()
        self.ResetDisplay()

    def MakeFileMenu(self):
        menu = wxMenu()

        mID = NewId()
        menu.Append(mID, 'Decrement', 'Next')
        EVT_MENU(self, mID, self.OnDecMonth)

        mID = NewId()
        menu.Append(mID, 'Increment', 'Dec')
        EVT_MENU(self, mID, self.OnIncMonth)

        menu.AppendSeparator()

        mID = NewId()
        menu.Append(mID, 'E&xit', 'Exit')
        EVT_MENU(self, mID, self.OnCloseWindow)

        return menu

    def MakeToolMenu(self):
        tb = self.CreateToolBar(wxTB_HORIZONTAL|wxNO_BORDER)

        bmp_path = 'bitmaps/'
        SetToolPath(self, tb, 10, bmp_path + 'DbDec.bmp', 'Dec Year')
        EVT_TOOL(self, 10, self.OnDecYear)

        SetToolPath(self, tb, 15, bmp_path + 'Dec.bmp', 'Dec Month')
        EVT_TOOL(self, 15, self.OnDecMonth)

        SetToolPath(self, tb, 30, bmp_path + 'Pt.bmp', 'Current Month')
        EVT_TOOL(self, 30, self.OnCurrent)

        SetToolPath(self, tb, 40, bmp_path + 'Inc.bmp', 'Inc Month')
        EVT_TOOL(self, 40, self.OnIncMonth)

        SetToolPath(self, tb, 45, bmp_path + 'DbInc.bmp', 'Inc Year')
        EVT_TOOL(self, 45, self.OnIncYear)

        tb.Realize()

#---------------------------------------------------------------------------

# example class for printing/previewing calendars

class PrintCalend:
    def __init__(self, parent, month, year):
        self.frame = parent
        self.month = month
        self.year = year

        self.SetParms()
        self.SetCal()
        self.printData = wxPrintData()

    def SetCal(self):
        self.grid_color = 'BLUE'
        self.back_color = 'WHITE'
        self.sel_color = 'RED'
        self.high_color = 'LIGHT BLUE'
        self.font = wxSWISS
        self.bold = wxNORMAL

        self.sel_key = None      #  last used by
        self.sel_lst = []        # highlighted selected days

        self.size = None
        self.hide_title = FALSE
        self.hide_grid = FALSE
        self.set_day = None

    def SetParms(self):
        self.ymax = 1
        self.xmax = 1
        self.page = 1
        self.total_pg = 1

        self.preview = None
        self.scale = 1.0

        self.pagew = 8.5
        self.pageh = 11.0

        self.txt_marg = 0.1
        self.lf_marg = 0
        self.top_marg = 0

        self.page = 0

    def SetDates(self, month, year):
        self.month = month
        self.year = year

    def SetStyleDef(self, desc):
        self.style = desc

    def SetCopies(self, copies):        # number of copies of label
        self.copies = copies

    def SetStart(self, start):          # start position of label
        self.start = start

    def Preview(self):
        printout = SetPrintout(self)
        printout2 = SetPrintout(self)
        self.preview = wxPrintPreview(printout, printout2, self.printData)
        if not self.preview.Ok():
            wxMessageBox("There was a problem printing!", "Printing", wxOK)
            return

        self.preview.SetZoom(60)        # initial zoom value

        frame = wxPreviewFrame(self.preview, self.frame, "Print preview")

        frame.Initialize()
        frame.SetPosition(self.frame.GetPosition())
        frame.SetSize(self.frame.GetSize())
        frame.Show(true)

    def Print(self):
        pdd = wxPrintDialogData()
        pdd.SetPrintData(self.printData)
        printer = wxPrinter(pdd)
        printout = SetPrintout(self)
        frame = wxFrame(NULL, -1, "Test")
        if not printer.Print(frame, printout):
            wxMessageBox("There was a problem printing.\nPerhaps your current printer is not set correctly?", "Printing", wxOK)
        else:
            self.printData = printer.GetPrintDialogData().GetPrintData()
        printout.Destroy()

    def DoDrawing(self, DC):
        size = DC.GetSizeTuple()
        DC.BeginDrawing()

        cal = PrtCalDraw(self)

        if self.preview is None:
            cal.SetPSize(size[0]/self.pagew, size[1]/self.pageh)
            cal.SetPreview(FALSE)

        else:
            if self.preview == 1:
                cal.SetPSize(size[0]/self.pagew, size[1]/self.pageh)
            else:
                cal.SetPSize(self.pwidth, self.pheight)

            cal.SetPreview(self.preview)

        cal.hide_title = self.hide_title        # set the calendar parameters
        cal.hide_grid = self.hide_grid

        cal.grid_color = self.grid_color
        cal.high_color = self.high_color
        cal.back_color = self.back_color
        cal.outer_border = FALSE
        cal.font = self.font
        cal.bold = self.bold

        cal_size = wxSize(3.0, 3.0)
        cal.SetSize(cal_size)

        year, month = self.year, self.month

        x = 1.0
        for i in range(2):
            y = 0.5
            for j in range(3):
                cal.SetCal(year, month)       # current month
                cal.SetPos(x, y)

                try:
                    set_days = test_days[month]
                except:
                    set_days = [1, 5, 12]

                cal.AddSelect([2, 16], 'GREEN', 'WHITE')

                cal.DrawCal(DC, set_days)

                year, month = self.IncMonth(year, month)
                y = y + 3.5
            x = x + 4.0     # next colum

        DC.EndDrawing()

        self.ymax = DC.MaxY()
        self.xmax = DC.MaxX()

    def IncMonth(self, year, month):     # next month
        month = month + 1
        if month > 12:
            month = 1
            year = year + 1

        return year, month

    def GetTotalPages(self):
        self.pg_cnt = 1
        return self.pg_cnt

    def SetPage(self, page):
        self.page = page

    def SetPageSize(self, width, height):
        self.pwidth, self.pheight = width, height

    def SetTotalSize(self, width, height):
        self.ptwidth, self.ptheight = width, height

    def SetPreview(self, preview, scale):
        self.preview = preview
        self.scale = scale

    def SetTotalSize(self, width, height):
        self.ptwidth = width
        self.ptheight = height

def SetToolPath(self, tb, id, bmp, title):
    global dir_path
    tb.AddSimpleTool(id, wxBitmap(os.path.join(dir_path, bmp), wxBITMAP_TYPE_BMP),
                     title, title)

class SetPrintout(wxPrintout):
    def __init__(self, canvas):
        wxPrintout.__init__(self)
        self.canvas = canvas
        self.end_pg = 1

    def OnBeginDocument(self, start, end):
        return self.base_OnBeginDocument(start, end)

    def OnEndDocument(self):
        self.base_OnEndDocument()

    def HasPage(self, page):
        if page <= self.end_pg:
            return true
        else:
            return false

    def GetPageInfo(self):
        self.end_pg = self.canvas.GetTotalPages()
        str_pg = 1
        try:
            end_pg = self.end_pg
        except:
            end_pg = 1
        return (str_pg, end_pg, str_pg, end_pg)

    def OnPreparePrinting(self):
        self.base_OnPreparePrinting()

    def OnBeginPrinting(self):
        dc = self.GetDC()

        self.preview = self.IsPreview()
        if (self.preview):
            self.pixelsPerInch = self.GetPPIScreen()
        else:
            self.pixelsPerInch = self.GetPPIPrinter()

        (w, h) = dc.GetSizeTuple()
        scaleX = float(w) / 1000
        scaleY = float(h) / 1000
        self.printUserScale = min(scaleX, scaleY)

        self.base_OnBeginPrinting()

    def GetSize(self):
        self.psizew, self.psizeh = self.GetPPIPrinter()
        return self.psizew, self.psizeh

    def GetTotalSize(self):
        self.ptsizew, self.ptsizeh = self.GetPageSizePixels()
        return self.ptsizew, self.ptsizeh

    def OnPrintPage(self, page):
        dc = self.GetDC()
        (w, h) = dc.GetSizeTuple()
        scaleX = float(w) / 1000
        scaleY = float(h) / 1000
        self.printUserScale = min(scaleX, scaleY)
        dc.SetUserScale(self.printUserScale, self.printUserScale)

        self.preview = self.IsPreview()

        self.canvas.SetPreview(self.preview, self.printUserScale)
        self.canvas.SetPage(page)

        self.ptsizew, self.ptsizeh = self.GetPageSizePixels()
        self.canvas.SetTotalSize(self.ptsizew, self.ptsizeh)

        self.psizew, self.psizeh = self.GetPPIPrinter()
        self.canvas.SetPageSize(self.psizew, self.psizeh)

        self.canvas.DoDrawing(dc)
        return true

class MyApp(wxApp):
    def OnInit(self):
        frame = CalendFrame(NULL, -1, "Test Calendar")
        frame.Show(true)
        self.SetTopWindow(frame)
        return true

#---------------------------------------------------------------------------

def MessageDlg(self, message, type = 'Message'):
    dlg = wxMessageDialog(self, message, type, wxOK | wxICON_INFORMATION)
    dlg.ShowModal()
    dlg.Destroy()

#---------------------------------------------------------------------------

def main():
    app = MyApp(0)
    app.MainLoop()


if __name__ == '__main__':
    main()


#---------------------------------------------------------------------------

def runTest(frame, nb, log):
    win = TestPanel(nb, log, frame)
    return win

#---------------------------------------------------------------------------


overview = """\
This control provides a calendar control class for displaying and selecting dates.  In addition, the class is extended and can now be used for printing/previewing.

Additional features include weekend highlighting and business type Monday-Sunday format.

See example for various methods used to set display month, year, and highlighted dates (different font and background colours).

by Lorne White

"""
Commit	Line	Data
944930d5	1	#----------------------------------------------------------------------------
f6bcfd97 BP	2	# Name: wxCalendar.py
f6bcfd97 BP	3	# Purpose: Calendar control display testing on panel for wxPython demo
944930d5 RD	4	#
	5	# Author: Lorne White (email: lwhite1@planet.eon.net)
	6	#
	7	# Created:
f6bcfd97	8	# Version 0.8 2000/04/16
944930d5 RD	9	# Licence: wxWindows license
	10	#----------------------------------------------------------------------------
	11
	12	from wxPython.wx import *
f6bcfd97	13	from wxPython.lib.calendar import wxCalendar, Month, PrtCalDraw
944930d5 RD	14
	15	import os
	16	dir_path = os.getcwd()
	17
	18
	19	# highlighted days in month
	20
	21	test_days ={ 0: [],
	22	1: [3, 7, 9, 21],
	23	2: [2, 10, 4, 9],
	24	3: [4, 20, 29],
	25	4: [1, 12, 22],
	26	5: [2, 10, 15],
	27	6: [4, 8, 17],
	28	7: [6, 7, 8],
	29	8: [5, 10, 20],
	30	9: [1, 2, 5, 29],
	31	10: [2, 4, 6, 22],
	32	11: [6, 9, 12, 28, 29],
	33	12: [8, 9, 10, 11, 20] }
	34
	35	# test of full window calendar control functions
	36
	37	def GetMonthList():
	38	monthlist = []
	39	for i in range(13):
	40	name = Month[i]
	41	if name != None:
	42	monthlist.append(name)
	43	return monthlist
	44
	45	class TestPanel(wxPanel):
f6bcfd97	46	def __init__(self, parent, log, frame):
944930d5 RD	47	wxPanel.__init__(self, parent, -1)
	48
	49	self.log = log
f6bcfd97	50	self.frame = frame
944930d5	51
e0473f5f	52	self.calend = wxCalendar(self, -1, wxPoint(100, 50), wxSize(200, 180))
944930d5 RD	53
	54	start_month = 11
	55	start_year = 1999
	56
	57	# month list from DateTime module
	58
	59	monthlist = GetMonthList()
	60
	61	self.date = wxComboBox(self, 10, Month[start_month], wxPoint(100, 20), wxSize(90, -1), monthlist, wxCB_DROPDOWN)
	62	EVT_COMBOBOX(self, 10, self.EvtComboBox)
	63
	64	# set start month and year
	65
	66	self.calend.SetMonth(start_month)
	67	self.calend.SetYear(start_year)
	68
	69	# set attributes of calendar
	70
f6bcfd97	71	self.calend.hide_title = TRUE
944930d5	72	self.calend.HideGrid()
f6bcfd97	73	self.calend.SetWeekColor('WHITE', 'BLACK')
944930d5 RD	74
	75	# display routine
	76
	77	self.ResetDisplay()
	78
	79	# mouse click event
	80
	81	self.Connect(self.calend.GetId(), -1, 2100, self.MouseClick)
	82
	83	# scroll bar for month selection
	84
	85	self.scroll = wxScrollBar(self, 40, wxPoint(100, 240), wxSize(200, 20), wxSB_HORIZONTAL)
	86	self.scroll.SetScrollbar(start_month-1, 1, 12, 1, TRUE)
	87	EVT_COMMAND_SCROLL(self, 40, self.Scroll)
	88
	89	# spin control for year selection
	90
	91	self.dtext = wxTextCtrl(self, -1, str(start_year), wxPoint(200, 20), wxSize(60, -1))
	92	h = self.dtext.GetSize().height
	93
	94	self.spin = wxSpinButton(self, 20, wxPoint(270, 20), wxSize(h*2, h))
	95	self.spin.SetRange(1980, 2010)
	96	self.spin.SetValue(start_year)
	97	EVT_SPIN(self, 20, self.OnSpin)
	98
	99	# button for calendar dialog test
	100
f6bcfd97	101	wxStaticText(self, -1, "Test Calendar Dialog", wxPoint(350, 50), wxSize(150, -1))
944930d5	102
6230a84f	103	bmp = wxBitmap('bitmaps/Calend.bmp', wxBITMAP_TYPE_BMP)
e0473f5f	104	self.but = wxBitmapButton(self, 60, bmp, wxPoint(380, 80))#, wxSize(30, 30))
944930d5 RD	105	EVT_BUTTON(self, 60, self.TestDlg)
	106
	107	# button for calendar window test
	108
f6bcfd97	109	wxStaticText(self, -1, "Test Calendar Window", wxPoint(350, 150), wxSize(150, -1))
944930d5	110
6230a84f	111	bmp = wxBitmap('bitmaps/Calend.bmp', wxBITMAP_TYPE_BMP)
e0473f5f	112	self.but = wxBitmapButton(self, 160, bmp, wxPoint(380, 180))#, wxSize(30, 30))
944930d5 RD	113	EVT_BUTTON(self, 160, self.TestFrame)
944930d5 RD	114
f6bcfd97 BP	115	wxStaticText(self, -1, "Test Calendar Print", wxPoint(350, 250), wxSize(150, -1))
	116
	117	bmp = wxBitmap('bitmaps/Calend.bmp', wxBITMAP_TYPE_BMP)
	118	self.but = wxBitmapButton(self, 170, bmp, wxPoint(380, 280))#, wxSize(30, 30))
	119	EVT_BUTTON(self, 170, self.OnPreview)
	120
944930d5 RD	121	# calendar dialog
	122
	123	def TestDlg(self, event):
	124	dlg = CalenDlg(self, self.log)
	125	dlg.Centre()
	126	dlg.ShowModal()
	127	dlg.Destroy()
	128
	129	# calendar window test
	130
	131	def TestFrame(self, event):
e0473f5f	132	frame = CalendFrame(self, -1, "Test Calendar", self.log)
944930d5	133	frame.Show(true)
944930d5 RD	134	return true
944930d5 RD	135
f6bcfd97 BP	136	# calendar print preview
	137
	138	def OnPreview(self, event):
	139	month = self.calend.GetMonth()
	140	year = self.calend.GetYear()
	141
	142	prt = PrintCalend(self.frame, month, year)
	143	prt.Preview()
	144
944930d5 RD	145	# month and year control events
	146
	147	def OnSpin(self, event):
	148	year = event.GetPosition()
	149	self.dtext.SetValue(str(year))
	150	self.calend.SetYear(year)
	151	self.calend.Refresh()
	152
	153	def EvtComboBox(self, event):
	154	name = event.GetString()
	155	self.log.WriteText('EvtComboBox: %s\n' % name)
	156	monthval = self.date.FindString(name)
	157	self.scroll.SetScrollbar(monthval, 1, 12, 1, TRUE)
	158
	159	self.calend.SetMonth(monthval+1)
	160	self.ResetDisplay()
	161
	162	def Scroll(self, event):
	163	value = self.scroll.GetThumbPosition()
	164	monthval = int(value)+1
	165	self.calend.SetMonth(monthval)
	166	self.ResetDisplay()
	167	self.log.WriteText('Month: %s\n' % value)
	168
	169	name = Month[monthval]
	170	self.date.SetValue(name)
	171
	172	# log mouse events
	173
	174	def MouseClick(self, evt):
	175	text = '%s CLICK %02d/%02d/%d' % (evt.click, evt.day, evt.month, evt.year) # format date
	176	self.log.WriteText('Date Selected: ' + text + '\n')
	177
944930d5 RD	178
	179	# set the highlighted days for the calendar
	180
	181	def ResetDisplay(self):
	182	month = self.calend.GetMonth()
	183	try:
	184	set_days = test_days[month]
	185	except:
	186	set_days = [1, 5, 12]
	187
f6bcfd97	188	self.calend.AddSelect([4, 11], 'BLUE', 'WHITE')
944930d5 RD	189	self.calend.SetSelDay(set_days)
	190	self.calend.Refresh()
	191
	192	# increment and decrement toolbar controls
	193
	194	def OnIncYear(self, event):
	195	self.calend.IncYear()
	196	self.ResetDisplay()
	197
	198	def OnDecYear(self, event):
	199	self.calend.DecYear()
	200	self.ResetDisplay()
	201
	202	def OnIncMonth(self, event):
	203	self.calend.IncMonth()
	204	self.ResetDisplay()
	205
	206	def OnDecMonth(self, event):
	207	self.calend.DecMonth()
	208	self.ResetDisplay()
	209
	210	def OnCurrent(self, event):
	211	self.calend.SetCurrentDay()
	212	self.ResetDisplay()
	213
	214	# test the calendar control in a dialog
	215
	216	class CalenDlg(wxDialog):
	217	def __init__(self, parent, log):
	218	self.log = log
	219	wxDialog.__init__(self, parent, -1, "Test Calendar", wxPyDefaultPosition, wxSize(280, 300))
	220
	221	start_month = 2
	222	start_year = 1999
	223
	224	# get month list from DateTime
	225
	226	monthlist = GetMonthList()
	227
	228	# select the month
	229
	230	self.date = wxComboBox(self, 100, Month[start_month], wxPoint(20, 20), wxSize(90, -1), monthlist, wxCB_DROPDOWN)
	231	EVT_COMBOBOX(self, 100, self.EvtComboBox)
	232
	233	# alternate spin button to control the month
	234
	235	h = self.date.GetSize().height
f6bcfd97	236	self.m_spin = wxSpinButton(self, 120, wxPoint(120, 20), wxSize(h*2, h), wxSP_VERTICAL)
944930d5 RD	237	self.m_spin.SetRange(1, 12)
	238	self.m_spin.SetValue(start_month)
	239
	240	EVT_SPIN(self, 120, self.OnMonthSpin)
	241
	242	# spin button to conrol the year
	243
	244	self.dtext = wxTextCtrl(self, -1, str(start_year), wxPoint(160, 20), wxSize(60, -1))
	245	h = self.dtext.GetSize().height
	246
	247	self.y_spin = wxSpinButton(self, 20, wxPoint(220, 20), wxSize(h*2, h), wxSP_VERTICAL)
	248	self.y_spin.SetRange(1980, 2010)
	249	self.y_spin.SetValue(start_year)
	250
	251	EVT_SPIN(self, 20, self.OnYrSpin)
	252
	253	# set the calendar and attributes
	254
e0473f5f	255	self.calend = wxCalendar(self, -1, wxPoint(20, 60), wxSize(240, 200))
944930d5 RD	256	self.calend.SetMonth(start_month)
	257	self.calend.SetYear(start_year)
	258
	259	self.calend.HideTitle()
f6bcfd97 BP	260	self.calend.ShowWeekEnd()
f6bcfd97 BP	261
944930d5 RD	262	self.ResetDisplay()
	263
	264	self.Connect(self.calend.GetId(), -1, 2100, self.MouseClick)
	265
	266	# log the mouse clicks
	267
	268	def MouseClick(self, evt):
	269	text = '%s CLICK %02d/%02d/%d' % (evt.click, evt.day, evt.month, evt.year) # format date
	270	self.log.WriteText('Date Selected: ' + text + '\n')
	271
	272	if evt.click == 'DLEFT':
	273	self.EndModal(wxID_OK)
	274
	275	# month and year spin selection routines
	276
	277	def OnMonthSpin(self, event):
	278	month = event.GetPosition()
f6bcfd97 BP	279	if month >= 0 and month <= 12:
	280	self.date.SetValue(Month[month])
	281	self.calend.SetMonth(month)
	282	self.calend.Refresh()
944930d5 RD	283
	284	def OnYrSpin(self, event):
	285	year = event.GetPosition()
	286	self.dtext.SetValue(str(year))
	287	self.calend.SetYear(year)
	288	self.calend.Refresh()
	289
	290	def EvtComboBox(self, event):
	291	name = event.GetString()
	292	self.log.WriteText('EvtComboBox: %s\n' % name)
	293	monthval = self.date.FindString(name)
	294	self.m_spin.SetValue(monthval+1)
	295
	296	self.calend.SetMonth(monthval+1)
	297	self.ResetDisplay()
	298
	299	# set the calendar for highlighted days
	300
	301	def ResetDisplay(self):
	302	month = self.calend.GetMonth()
	303	try:
	304	set_days = test_days[month]
	305	except:
	306	set_days = [1, 5, 12]
	307
f6bcfd97 BP	308	self.calend.AddSelect([4, 11], 'BLUE', 'WHITE')
f6bcfd97 BP	309
944930d5 RD	310	self.calend.SetSelDay(set_days)
	311	self.calend.Refresh()
	312
	313	# test of full window calendar control functions
	314
	315	class CalendFrame(wxFrame):
	316	def __init__(self, parent, id, title, log):
	317	wxFrame.__init__(self, parent, id, title, wxPyDefaultPosition, wxSize(400, 400))
f6bcfd97	318	EVT_CLOSE(self, self.OnCloseWindow)
944930d5 RD	319
	320	self.log = log
	321	self.CreateStatusBar()
	322	self.mainmenu = wxMenuBar()
	323	menu = wxMenu()
	324
	325	menu = self.MakeFileMenu()
	326	self.mainmenu.Append(menu, '&File')
	327
	328	self.MakeToolMenu() # toolbar
	329
	330	self.SetMenuBar(self.mainmenu)
e0473f5f	331	self.calend = wxCalendar(self, -1)
944930d5 RD	332	self.calend.SetCurrentDay()
944930d5 RD	333	self.calend.grid_color = 'BLUE'
f6bcfd97 BP	334	self.calend.SetBusType()
	335	# self.calend.ShowWeekEnd()
	336
944930d5 RD	337	self.ResetDisplay()
	338
	339	self.Connect(self.calend.GetId(), -1, 2100, self.MouseClick)
	340
	341	def MouseClick(self, evt):
	342	text = '%s CLICK %02d/%02d/%d' % (evt.click, evt.day, evt.month, evt.year) # format date
	343	self.log.WriteText('Date Selected: ' + text + '\n')
	344
	345	def OnCloseWindow(self, event):
	346	self.Destroy()
	347
	348	def ResetDisplay(self):
	349	month = self.calend.GetMonth()
	350	try:
	351	set_days = test_days[month]
	352	except:
	353	set_days = [1, 5, 12]
	354
f6bcfd97 BP	355	self.calend.AddSelect([2, 16], 'GREEN', 'WHITE')
f6bcfd97 BP	356
944930d5 RD	357	self.calend.SetSelDay(set_days)
	358	self.calend.Refresh()
	359
	360	def OnIncYear(self, event):
	361	self.calend.IncYear()
	362	self.ResetDisplay()
	363
	364	def OnDecYear(self, event):
	365	self.calend.DecYear()
	366	self.ResetDisplay()
	367
	368	def OnIncMonth(self, event):
	369	self.calend.IncMonth()
	370	self.ResetDisplay()
	371
	372	def OnDecMonth(self, event):
	373	self.calend.DecMonth()
	374	self.ResetDisplay()
	375
	376	def OnCurrent(self, event):
	377	self.calend.SetCurrentDay()
	378	self.ResetDisplay()
	379
	380	def MakeFileMenu(self):
	381	menu = wxMenu()
	382
	383	mID = NewId()
	384	menu.Append(mID, 'Decrement', 'Next')
	385	EVT_MENU(self, mID, self.OnDecMonth)
	386
	387	mID = NewId()
	388	menu.Append(mID, 'Increment', 'Dec')
	389	EVT_MENU(self, mID, self.OnIncMonth)
	390
	391	menu.AppendSeparator()
	392
	393	mID = NewId()
	394	menu.Append(mID, 'E&xit', 'Exit')
	395	EVT_MENU(self, mID, self.OnCloseWindow)
	396
	397	return menu
	398
	399	def MakeToolMenu(self):
	400	tb = self.CreateToolBar(wxTB_HORIZONTAL\|wxNO_BORDER)
	401
6230a84f	402	bmp_path = 'bitmaps/'
944930d5 RD	403	SetToolPath(self, tb, 10, bmp_path + 'DbDec.bmp', 'Dec Year')
	404	EVT_TOOL(self, 10, self.OnDecYear)
	405
	406	SetToolPath(self, tb, 15, bmp_path + 'Dec.bmp', 'Dec Month')
	407	EVT_TOOL(self, 15, self.OnDecMonth)
	408
	409	SetToolPath(self, tb, 30, bmp_path + 'Pt.bmp', 'Current Month')
	410	EVT_TOOL(self, 30, self.OnCurrent)
	411
	412	SetToolPath(self, tb, 40, bmp_path + 'Inc.bmp', 'Inc Month')
	413	EVT_TOOL(self, 40, self.OnIncMonth)
	414
	415	SetToolPath(self, tb, 45, bmp_path + 'DbInc.bmp', 'Inc Year')
	416	EVT_TOOL(self, 45, self.OnIncYear)
	417
	418	tb.Realize()
	419
f6bcfd97 BP	420	#---------------------------------------------------------------------------
	421
	422	# example class for printing/previewing calendars
	423
	424	class PrintCalend:
	425	def __init__(self, parent, month, year):
	426	self.frame = parent
	427	self.month = month
	428	self.year = year
	429
	430	self.SetParms()
	431	self.SetCal()
	432	self.printData = wxPrintData()
	433
	434	def SetCal(self):
	435	self.grid_color = 'BLUE'
	436	self.back_color = 'WHITE'
	437	self.sel_color = 'RED'
	438	self.high_color = 'LIGHT BLUE'
	439	self.font = wxSWISS
	440	self.bold = wxNORMAL
	441
	442	self.sel_key = None # last used by
	443	self.sel_lst = [] # highlighted selected days
	444
	445	self.size = None
	446	self.hide_title = FALSE
	447	self.hide_grid = FALSE
	448	self.set_day = None
	449
	450	def SetParms(self):
	451	self.ymax = 1
	452	self.xmax = 1
	453	self.page = 1
	454	self.total_pg = 1
	455
	456	self.preview = None
	457	self.scale = 1.0
	458
	459	self.pagew = 8.5
	460	self.pageh = 11.0
	461
	462	self.txt_marg = 0.1
	463	self.lf_marg = 0
	464	self.top_marg = 0
	465
	466	self.page = 0
	467
	468	def SetDates(self, month, year):
	469	self.month = month
	470	self.year = year
	471
	472	def SetStyleDef(self, desc):
	473	self.style = desc
	474
	475	def SetCopies(self, copies): # number of copies of label
	476	self.copies = copies
	477
	478	def SetStart(self, start): # start position of label
	479	self.start = start
	480
	481	def Preview(self):
	482	printout = SetPrintout(self)
	483	printout2 = SetPrintout(self)
484	self.preview = wxPrintPreview(printout, printout2, self.printData)
485	if not self.preview.Ok():
486	wxMessageBox("There was a problem printing!", "Printing", wxOK)
487	return
488
489	self.preview.SetZoom(60) # initial zoom value
490
491	frame = wxPreviewFrame(self.preview, self.frame, "Print preview")
492
493	frame.Initialize()
494	frame.SetPosition(self.frame.GetPosition())
495	frame.SetSize(self.frame.GetSize())
496	frame.Show(true)
497
498	def Print(self):
499	pdd = wxPrintDialogData()
500	pdd.SetPrintData(self.printData)
501	printer = wxPrinter(pdd)
502	printout = SetPrintout(self)
503	frame = wxFrame(NULL, -1, "Test")
504	if not printer.Print(frame, printout):
505	wxMessageBox("There was a problem printing.\nPerhaps your current printer is not set correctly?", "Printing", wxOK)
506	else:
507	self.printData = printer.GetPrintDialogData().GetPrintData()
508	printout.Destroy()
509
510	def DoDrawing(self, DC):
511	size = DC.GetSizeTuple()
512	DC.BeginDrawing()
513
514	cal = PrtCalDraw(self)
515
516	if self.preview is None:
517	cal.SetPSize(size[0]/self.pagew, size[1]/self.pageh)
518	cal.SetPreview(FALSE)
519
520	else:
521	if self.preview == 1:
522	cal.SetPSize(size[0]/self.pagew, size[1]/self.pageh)
523	else:
524	cal.SetPSize(self.pwidth, self.pheight)
525
526	cal.SetPreview(self.preview)
527
528	cal.hide_title = self.hide_title # set the calendar parameters
529	cal.hide_grid = self.hide_grid
530
531	cal.grid_color = self.grid_color
532	cal.high_color = self.high_color
533	cal.back_color = self.back_color
534	cal.outer_border = FALSE
535	cal.font = self.font
536	cal.bold = self.bold
537
538	cal_size = wxSize(3.0, 3.0)
539	cal.SetSize(cal_size)
540
541	year, month = self.year, self.month
542
543	x = 1.0
544	for i in range(2):
545	y = 0.5
546	for j in range(3):
547	cal.SetCal(year, month) # current month
548	cal.SetPos(x, y)
549
550	try:
551	set_days = test_days[month]
552	except:
553	set_days = [1, 5, 12]
554
555	cal.AddSelect([2, 16], 'GREEN', 'WHITE')
556
557	cal.DrawCal(DC, set_days)
558
559	year, month = self.IncMonth(year, month)
560	y = y + 3.5
561	x = x + 4.0 # next colum
562
563	DC.EndDrawing()
564
565	self.ymax = DC.MaxY()
566	self.xmax = DC.MaxX()
567
568	def IncMonth(self, year, month): # next month
569	month = month + 1
570	if month > 12:
571	month = 1
572	year = year + 1
573
574	return year, month
575
576	def GetTotalPages(self):
577	self.pg_cnt = 1
578	return self.pg_cnt
579
580	def SetPage(self, page):
581	self.page = page
582
583	def SetPageSize(self, width, height):
584	self.pwidth, self.pheight = width, height
585
586	def SetTotalSize(self, width, height):
587	self.ptwidth, self.ptheight = width, height
588
589	def SetPreview(self, preview, scale):
590	self.preview = preview
591	self.scale = scale
592
593	def SetTotalSize(self, width, height):
594	self.ptwidth = width
595	self.ptheight = height
596
944930d5 RD	597	def SetToolPath(self, tb, id, bmp, title):
944930d5 RD	598	global dir_path
3af4e610 RD	599	tb.AddSimpleTool(id, wxBitmap(os.path.join(dir_path, bmp), wxBITMAP_TYPE_BMP),
3af4e610 RD	600	title, title)
944930d5	601
f6bcfd97 BP	602	class SetPrintout(wxPrintout):
	603	def __init__(self, canvas):
	604	wxPrintout.__init__(self)
	605	self.canvas = canvas
	606	self.end_pg = 1
	607
	608	def OnBeginDocument(self, start, end):
	609	return self.base_OnBeginDocument(start, end)
	610
	611	def OnEndDocument(self):
	612	self.base_OnEndDocument()
	613
	614	def HasPage(self, page):
	615	if page <= self.end_pg:
	616	return true
	617	else:
	618	return false
	619
	620	def GetPageInfo(self):
	621	self.end_pg = self.canvas.GetTotalPages()
	622	str_pg = 1
	623	try:
	624	end_pg = self.end_pg
	625	except:
	626	end_pg = 1
	627	return (str_pg, end_pg, str_pg, end_pg)
	628
	629	def OnPreparePrinting(self):
	630	self.base_OnPreparePrinting()
	631
	632	def OnBeginPrinting(self):
	633	dc = self.GetDC()
	634
	635	self.preview = self.IsPreview()
	636	if (self.preview):
	637	self.pixelsPerInch = self.GetPPIScreen()
	638	else:
	639	self.pixelsPerInch = self.GetPPIPrinter()
	640
	641	(w, h) = dc.GetSizeTuple()
	642	scaleX = float(w) / 1000
	643	scaleY = float(h) / 1000
	644	self.printUserScale = min(scaleX, scaleY)
	645
	646	self.base_OnBeginPrinting()
	647
	648	def GetSize(self):
	649	self.psizew, self.psizeh = self.GetPPIPrinter()
	650	return self.psizew, self.psizeh
	651
	652	def GetTotalSize(self):
	653	self.ptsizew, self.ptsizeh = self.GetPageSizePixels()
	654	return self.ptsizew, self.ptsizeh
	655
	656	def OnPrintPage(self, page):
	657	dc = self.GetDC()
	658	(w, h) = dc.GetSizeTuple()
	659	scaleX = float(w) / 1000
	660	scaleY = float(h) / 1000
	661	self.printUserScale = min(scaleX, scaleY)
	662	dc.SetUserScale(self.printUserScale, self.printUserScale)
	663
	664	self.preview = self.IsPreview()
	665
666	self.canvas.SetPreview(self.preview, self.printUserScale)
667	self.canvas.SetPage(page)
668
669	self.ptsizew, self.ptsizeh = self.GetPageSizePixels()
670	self.canvas.SetTotalSize(self.ptsizew, self.ptsizeh)
671
672	self.psizew, self.psizeh = self.GetPPIPrinter()
673	self.canvas.SetPageSize(self.psizew, self.psizeh)
674
675	self.canvas.DoDrawing(dc)
676	return true
e0473f5f	677
944930d5 RD	678	class MyApp(wxApp):
	679	def OnInit(self):
	680	frame = CalendFrame(NULL, -1, "Test Calendar")
	681	frame.Show(true)
	682	self.SetTopWindow(frame)
	683	return true
	684
	685	#---------------------------------------------------------------------------
	686
	687	def MessageDlg(self, message, type = 'Message'):
	688	dlg = wxMessageDialog(self, message, type, wxOK \| wxICON_INFORMATION)
	689	dlg.ShowModal()
	690	dlg.Destroy()
	691
	692	#---------------------------------------------------------------------------
	693
	694	def main():
	695	app = MyApp(0)
	696	app.MainLoop()
	697
	698
	699	if __name__ == '__main__':
	700	main()
	701
	702
	703	#---------------------------------------------------------------------------
	704
	705	def runTest(frame, nb, log):
f6bcfd97	706	win = TestPanel(nb, log, frame)
944930d5 RD	707	return win
	708
	709	#---------------------------------------------------------------------------
	710
	711
	712	overview = """\
f6bcfd97 BP	713	This control provides a calendar control class for displaying and selecting dates. In addition, the class is extended and can now be used for printing/previewing.
	714
	715	Additional features include weekend highlighting and business type Monday-Sunday format.
944930d5	716
f6bcfd97	717	See example for various methods used to set display month, year, and highlighted dates (different font and background colours).
944930d5 RD	718
	719	by Lorne White
	720
	721	"""