calender tab change

#!/usr/bin/env python

### BITPIM
###
### Copyright (C) 2003-2004 Roger Binns <[email protected]>
###
### This program is free software; you can redistribute it and/or modify
### it under the terms of the BitPim license as detailed in the LICENSE file.
###
### $Id: calendarcontrol.py,v 1.40 2005/03/03 07:10:31 rogerb Exp $

"""A calendar control that shows several weeks in one go

The design is inspired by the Alan Cooper article U{http://www.cooper.com/articles/art_goal_directed_design.htm}
about goal directed design.  I also have to apologise for it not quite living up to that vision :-)

It is fairly feature complete and supports all sorts of interaction, scrolling and customization
of appearance"""

import wx
import wx.lib.rcsizer
import wx.calendar
import cStringIO
import calendar
import time


class FontscaleCache(dict):
    """A cache used internally to remember how much to shrink fonts by"""
    # cache results of what the scale factor is to fit a number of lines in a space is
    def get(self, y, attr, numentries):
        return dict.get(self, (y, id(attr), numentries), 1)
    def set(self, y, attr, numentries, scale):
        self[(y, id(attr),  numentries)]=scale
    def uncache(self, *args):
        # clear out any cached attrs listed in args (eg when they are changed)
        keys=self.keys()
        l2=[id(x) for x in args]
        for y, idattr, numentries in keys:
            if idattr in l2:
                del self[ (y, idattr, numentries) ]

thefontscalecache=FontscaleCache()

class CalendarCellAttributes:
    """A class represnting appearance attributes for an individual day.

    You should subclass this if you wish to change the appearance from
    the defaults"""
    def __init__(self):
        # Set some defaults
        #self.cellbackground=wx.TheBrushList.FindOrCreateBrush(wx.Colour(230,255,255), wx.SOLID)
        self.cellbackground=wx.Brush(wx.Colour(197,255,255), wx.SOLID)
        self.labelfont=wx.Font(14, wx.SWISS, wx.NORMAL, wx.NORMAL )
        self.labelforeground=wx.NamedColour("CORNFLOWER BLUE")
        self.labelalign=wx.ALIGN_RIGHT
        self.timefont=wx.Font(8, wx.SWISS, wx.NORMAL, wx.NORMAL )
        self.timeforeground=wx.NamedColour("ORCHID")
        self.entryfont=wx.Font(9, wx.SWISS, wx.NORMAL, wx.NORMAL )
        self.entryforeground=wx.NamedColour("BLACK")
        self.miltime=False
        self.initdone=True

    def __setattr__(self, k, v):
        self.__dict__[k]=v
        if hasattr(self, 'initdone'):
            thefontscalecache.uncache(self)

    def isrightaligned(self):
        """Is the number representing the day right aligned within the cell?

        @rtype: Bool
        @return:  True is it should be shown right aligned"""
        return self.labelalign==wx.ALIGN_RIGHT

    def ismiltime(self):
        """Are times shown in military (aka 24 hour) time?

        @rtype: Bool
        @return: True is militart/24 hour format should be used"""
        return self.miltime

    def setforcellbackground(self, dc):
        """Set the cell background attributes

        Colour
        @type dc: wx.DC"""
        dc.SetBackground(self.cellbackground)

    def setforlabel(self, dc, fontscale=1):
        """Set the attributes for the day label

        Colour, font
        @type dc: wx.DC
        @param fontscale: You should multiply the font point size
                          by this number
        @type fontscale: float
        """
        
        dc.SetTextForeground(self.labelforeground)
        return self.setscaledfont(dc, self.labelfont, fontscale)

    def setfortime(self,dc, fontscale=1):
        """Set the attributes for the time of an event text
 
        Colour, font
        @type dc: wx.DC
        @param fontscale: You should multiply the font point size
                          by this number
        @type fontscale: float
        """
        dc.SetTextForeground(self.timeforeground)
        return self.setscaledfont(dc, self.timefont, fontscale)

    def setforentry(self, dc, fontscale=1):
        """Set the attributes for the label of an event text
 
        Colour, font
        @type dc: wx.DC
        @param fontscale: You should multiply the font point size
                          by this number
        @type fontscale: float
        """        
        dc.SetTextForeground(self.entryforeground)
        return self.setscaledfont(dc, self.entryfont, fontscale)
                
    def setscaledfont(self, dc, font, fontscale):
        """Changes the in the device context to the supplied font suitably scaled

        @type dc: wx.DC
        @type font: wx.Font
        @type fontscale: float
        @return: Returns True if the scaling succeeded, and False if the font was already
                 too small to scale smaller (the smallest size will still have been
                 selected into the device context)
        @rtype: Bool"""
        if fontscale==1:
            dc.SetFont(font)
            return True
        ps=int(font.GetPointSize()*fontscale)
        if ps<2:
            ps=2
        f=wx.TheFontList.FindOrCreateFont(int(ps), font.GetFamily(), font.GetStyle(), font.GetWeight())
        dc.SetFont(f)
        if ps==2:
            return False
        return True


# a hack - this used to be an instance, but wx 2.5 doesn't allow using brushes/pens etc until
# app instance is created
DefaultCalendarCellAttributes=None

def GetCalendarCellAttributes(attr=None):
    if attr is not None:
        return attr
    global DefaultCalendarCellAttributes
    if DefaultCalendarCellAttributes is None:
        DefaultCalendarCellAttributes=CalendarCellAttributes()
    return DefaultCalendarCellAttributes
                
class CalendarCell(wx.PyWindow):
    """A control that is used for each day in the calendar

    As the user scrolls around the calendar, each cell is updated with new dates rather
    than creating new CalendarCell objects.  Internally it uses a backing buffer so
    that redraws are quick and flicker free."""
    
    fontscalecache=FontscaleCache()

    def __init__(self, parent, id, attr=DefaultCalendarCellAttributes, style=wx.SIMPLE_BORDER):
        wx.PyWindow.__init__(self, parent, id, style=style|wx.WANTS_CHARS|wx.FULL_REPAINT_ON_RESIZE)
        self.attr=GetCalendarCellAttributes(attr)
        self.day=33
        self.year=2033
        self.month=3
        self.buffer=None
        self.needsupdate=True
        self.entries=()

        wx.EVT_PAINT(self, self.OnPaint)
        wx.EVT_SIZE(self, self.OnSize)
        wx.EVT_ERASE_BACKGROUND(self, self.OnEraseBackground)
        self.OnSize(None)

    def DoGetBestSize(self):
        return (10,10)

    def OnEraseBackground(self, _):
        pass

    def setdate(self, year, month, day):
        """Set the date we are"""
        self.year=year
        self.month=month
        self.day=day
        self.needsupdate=True
        self.Refresh(False)

    def setattr(self, attr):
        """Sets what CalendarCellAtrributes we use for appearance

        @type attr: CalendarCellAtrributes"""
        self.attr=GetCalendarCellAttributes(attr)
        self.needsupdate=True
        self.Refresh(False)

    def setentries(self, entries):
        """Sets the entries we will display

        @type entries: list
        @param entries: A list of entries.  Format is ( ( hour, minute, description), (hour, minute, decription) ... ).  hour is in 24 hour
        """
        self.entries=entries
        self.needsupdate=True
        self.Refresh(False)

    def getdate(self):
        """Returns what date we are currently displaying

        @rtype: tuple
        @return: tuple of (year, month, day)"""
        return (self.year, self.month, self.day)

    def OnSize(self, evt=None):
        """Callback for when we are resized"""
        self.width, self.height = self.GetClientSizeTuple()
        self.needsupdate=True
        if evt is not None:
            evt.Skip()
            

    def redraw(self):
        """Causes a forced redraw into our back buffer"""
        if self.buffer is None or \
           self.buffer.GetWidth()!=self.width or \
           self.buffer.GetHeight()!=self.height:
            if self.buffer is not None:
                del self.buffer
            self.buffer=wx.EmptyBitmap(self.width, self.height)

        mdc=wx.MemoryDC()
        mdc.SelectObject(self.buffer)
        self.attr.setforcellbackground(mdc)
        mdc.Clear()
        self.draw(mdc)
        mdc.SelectObject(wx.NullBitmap)
        del mdc

    def OnPaint(self, _=None):
        """Callback for when we need to repaint"""
        if self.needsupdate:
            self.needsupdate=False
            self.redraw()
        dc=wx.PaintDC(self)
        dc.DrawBitmap(self.buffer, 0, 0, False)

    def draw(self, dc):
        """Draw ourselves

        @type dc: wx.DC"""
        
        # do the label
        self.attr.setforlabel(dc)
        w,h=dc.GetTextExtent(`self.day`)
        x=1
        if self.attr.isrightaligned():
            x=self.width-(w+5)
        dc.DrawText(`self.day`, x, 0)

        if len(self.entries)==0:
            return
        
        entrystart=h # +5
        dc.DestroyClippingRegion()
        dc.SetClippingRegion( 0, entrystart, self.width, self.height-entrystart)

        fontscale=thefontscalecache.get(self.height-entrystart, self.attr, len(self.entries))
        iteration=0
         # this loop scales the contents to fit the space available
         # we do it as a loop because even when we ask for a smaller font
         # after finding out that it was too big the first time, the
         # smaller font may not be as small as we requested
        while 1:
            y=entrystart
            iteration+=1
            # now calculate how much space is needed for the time fields
            self.attr.setfortime(dc, fontscale)
            boundingspace=2
            space,_=dc.GetTextExtent("i")
            timespace,timeheight=dc.GetTextExtent("mm:mm")
            if self.attr.ismiltime():
                ampm=0
            else:
                ampm,_=dc.GetTextExtent("a")

            r=self.attr.setforentry(dc, fontscale)
            if not r: iteration=-1 # font can't be made this small
            _,entryheight=dc.GetTextExtent("I")
            firstrowheight=max(timeheight, entryheight)

            # Now draw each item
            lastap=""
            for h,m,desc in self.entries:
                x=0
                if h is not None:
                    self.attr.setfortime(dc, fontscale)
                    # bounding
                    x+=boundingspace # we don't draw anything yet
                    timey=y
                    if timeheight<firstrowheight:
                        timey+=(firstrowheight-timeheight)/2
                    text=""
                    
                    if self.attr.ismiltime():
                        ap=""
                    else:
                        ap="a"
                        if h>=12: ap="p"
                        h%=12
                        if h==0: h=12
                        if ap==lastap:
                            ap=""
                        else:
                            lastap=ap
                    if h<10: text+=" "
                    
                    text+="%d:%02d%s" % (h,m,ap)
                    dc.DrawText(text, x, timey)
                    x+=timespace
                    if not self.attr.ismiltime: x+=ampm
                
                self.attr.setforentry(dc, fontscale)
                ey=y
                if entryheight<firstrowheight:
                    ey+=(firstrowheight-entryheight)/2
                dc.DrawText(desc, x, ey)
                # that row is dealt with!
                y+=firstrowheight
            if iteration==1 and fontscale!=1:
                # came from cache
                break
            if iteration==-1:
                # update cache
                thefontscalecache.set(self.height-entrystart, self.attr, len(self.entries), fontscale)
                break # reached limit of font scaling
            if iteration<10 and y>self.height:
                dc.Clear()
                # how much too big were we?
                fontscale=fontscale*float(self.height-entrystart)/(y-entrystart)
                # print iteration, y, self.height, fontscale
            else:
                thefontscalecache.set(self.height-entrystart, self.attr, len(self.entries), fontscale)
                break

class CalendarLabel(wx.PyWindow):
    """The label window on the left of the day cells that shows the month with rotated text

    It uses double buffering etc for a flicker free experience"""
    
    def __init__(self, parent, cells, id=-1):
        wx.PyWindow.__init__(self, parent, id, style=wx.FULL_REPAINT_ON_RESIZE)
        self.needsupdate=True
        self.buffer=None
        self.cells=cells
        wx.EVT_PAINT(self, self.OnPaint)
        wx.EVT_SIZE(self, self.OnSize)
        wx.EVT_ERASE_BACKGROUND(self, self.OnEraseBackground)
        self.setfont(wx.Font(20, wx.SWISS, wx.NORMAL, wx.BOLD ))
        self.settextcolour(wx.NamedColour("BLACK"))
        self.OnSize(None)

    def DoGetBestSize(self):
        return (10,10)

    def OnEraseBackground(self, _):
        pass

    def OnSize(self, _=None):
        self.width, self.height = self.GetClientSizeTuple()
        self.needsupdate=True
        
    def OnPaint(self, _=None):
        if self.needsupdate:
            self.needsupdate=False
            self.redraw()
        dc=wx.PaintDC(self)
        dc.DrawBitmap(self.buffer, 0, 0, False)

    def setfont(self, font):
        self.font=font

    def settextcolour(self, colour):
        self.colour=colour

    def changenotify(self):
        self.needsupdate=True
        self.Refresh()

    def redraw(self):
        if self.buffer is None or \
           self.buffer.GetWidth()!=self.width or \
           self.buffer.GetHeight()!=self.height:
            if self.buffer is not None:
                del self.buffer
            self.buffer=wx.EmptyBitmap(self.width, self.height)

        mdc=wx.MemoryDC()
        mdc.SelectObject(self.buffer)
        mdc.SetBackground(wx.TheBrushList.FindOrCreateBrush(self.GetBackgroundColour(), wx.SOLID))
        mdc.Clear()
        self.draw(mdc)
        mdc.SelectObject(wx.NullBitmap)
        del mdc

    def draw(self, dc):
        # find the lines for each cell
        row=0
        while row<len(self.cells):
            month=self.cells[row].month
            endrow=row
            for r2 in range(row+1,len(self.cells)):
                if month==self.cells[r2].month:
                    endrow=r2
                else:
                    break
            # row is begining row, endrow is end, inclusive

            # find the space available.  we do lots of lovely math
            # in order to translate the coordinates from the rows
            # into our window
            x=0
            y=self.cells[row].GetPositionTuple()[1]-self.cells[0].GetPositionTuple()[1]
            w=self.width
            h=self.cells[endrow].GetPositionTuple()[1]+self.cells[endrow].GetRect().height \
               -self.cells[row].GetPositionTuple()[1]

            
            dc.DestroyClippingRegion()
            dc.SetClippingRegion(x,y,w,h)
            dc.SetPen(wx.ThePenList.FindOrCreatePen("BLACK", 3, wx.SOLID))
            # draw line at top and bottom
            if row!=0:
                dc.DrawLine(x, y, x+w, y)
            if endrow!=len(self.cells)-1:
                dc.DrawLine(x, y+h, x+w, y+h)
            month=calendar.month_name[month]
            dc.SetFont(self.font)
            dc.SetTextForeground(self.colour)
            tw,th=dc.GetTextExtent(month)
            # Now figure out where to draw it
            if tw<h:
                # it fits, so centre
                dc.DrawRotatedText(month, w/2-th/2, y + h/2 + tw/2, 90)
            else:
                 # it doesn't fit
                 if row==0:
                     # top one shows start of text
                     dc.DrawRotatedText(month, w/2-th/2, y + h -5, 90)
                 else:
                     # show end of text at bottom
                     dc.DrawRotatedText(month, w/2-th/2, y + 5 + tw, 90)

            # Loop around
            row=endrow+1


class Calendar(wx.Panel):
    """The main calendar control.

    You should subclass this clas and need to
    implement the following methods:

    L{OnGetEntries}
    L{OnEdit}

    The following methods you may want to call at some point:

    L{RefreshEntry}
    L{RefreshAllEntries}

    
"""

    # All the horrible date code is an excellent case for metric time!
    ID_UP=wx.NewId()
    ID_DOWN=wx.NewId()
    ID_YEARBUTTON=wx.NewId()
    ID_TODAYBUTTON=wx.NewId()

    attrevenmonth=None
    attroddmonth=None
    attrselectedcell=None

    def _initvars(self):
        # this is needed to avoid issues with the wx. library not being initialised
        # as the module is imported.  We initialise the values when the first
        # calendar constructor is run
        if Calendar.attrevenmonth is None:
            Calendar.attrevenmonth=CalendarCellAttributes()
        if Calendar.attroddmonth is None:
            Calendar.attroddmonth=CalendarCellAttributes()
            Calendar.attroddmonth.cellbackground=wx.TheBrushList.FindOrCreateBrush( wx.Colour(255, 255, 230), wx.SOLID)
        if Calendar.attrselectedcell is None:
            Calendar.attrselectedcell=CalendarCellAttributes()
            Calendar.attrselectedcell.cellbackground=wx.TheBrushList.FindOrCreateBrush( wx.Colour(240,240,240), wx.SOLID)
            Calendar.attrselectedcell.labelfont=wx.Font(17, wx.SWISS, wx.NORMAL, wx.BOLD )
            Calendar.attrselectedcell.labelforeground=wx.NamedColour("BLACK")
    
    def __init__(self, parent, rows=5, id=-1):
        self._initvars()
        wx.Panel.__init__(self, parent, id, style=wx.WANTS_CHARS|wx.FULL_REPAINT_ON_RESIZE)
        sizer=wx.GridBagSizer()
        self.upbutt=wx.BitmapButton(self, self.ID_UP, getupbitmapBitmap())
        sizer.Add(self.upbutt, flag=wx.EXPAND, pos=(0,1), span=(1,7))
        self.year=wx.Button(self, self.ID_YEARBUTTON, "2003")
        sizer.Add(self.year, flag=wx.EXPAND, pos=(1,0))
        sizer.Add(wx.Button(self, self.ID_TODAYBUTTON, "Today"), flag=wx.EXPAND, pos=(0,0))
        p=1
        calendar.setfirstweekday(calendar.SUNDAY)
        for i in ( "Sun", "Mon", "Tue", "Wed" , "Thu", "Fri", "Sat" ):
           sizer.Add(  wx.StaticText( self, -1, i, style=wx.ALIGN_CENTER|wx.ALIGN_CENTER_VERTICAL),
                       flag=wx.ALIGN_CENTER_VERTICAL|wx.ALIGN_CENTER_HORIZONTAL|wx.EXPAND, pos=(1,p))
           sizer.AddGrowableCol(p)
           p+=1
        self.numrows=rows
        self.showrow=rows/2
        self.rows=[]
        for i in range(0, rows):
            self.rows.append( self.makerow(sizer, i+2) )
        self.downbutt=wx.BitmapButton(self, self.ID_DOWN, getdownbitmapBitmap())
        sizer.Add(self.downbutt, flag=wx.EXPAND, pos=(2+rows, 0), span=(1,8))
        self.label=CalendarLabel(self, map(lambda x: x[0], self.rows))
        sizer.Add(self.label, flag=wx.EXPAND, pos=(2,0), span=(self.numrows,1))
        self.SetSizer(sizer)
        self.sizer=sizer
        self.SetAutoLayout(True)
        self.bgcolor=self.GetBackgroundColour()

        self.popupcalendar=PopupCalendar(self, self)

        wx.EVT_PAINT(self, self.OnPaint)
        wx.EVT_BUTTON(self, self.ID_UP, self.OnScrollUp)
        wx.EVT_BUTTON(self, self.ID_DOWN, self.OnScrollDown)
        wx.EVT_BUTTON(self, self.ID_YEARBUTTON, self.OnYearButton)
        wx.EVT_BUTTON(self, self.ID_TODAYBUTTON, self.OnTodayButton)
        wx.EVT_ERASE_BACKGROUND(self, self.OnEraseBackground)
        # grab key down from all children
        map(lambda child: wx.EVT_KEY_DOWN(child, self.OnKeyDown), self.GetChildren())
        # and mousewheel
        map(lambda child: wx.EVT_MOUSEWHEEL(child, self.OnMouseWheel), self.GetChildren())
        # grab left down, left dclick from all cells
        for r in self.rows:
            map(lambda cell: wx.EVT_LEFT_DOWN(cell, self.OnLeftDown), r)
            map(lambda cell: wx.EVT_LEFT_DCLICK(cell, self.OnLeftDClick), r)

        self.selectedcell=(-1,-1)
        self.selecteddate=(-1,-1,-1)

        self.showday(*time.localtime()[:3]+(self.showrow,))
        self.setday(*time.localtime()[:3])

    def OnPaint(self, _=None):
        DC = wx.BufferedPaintDC(self)
        self.Layout
        #calculate the gap at the edge of the sizer control
        width, height = self.GetClientSizeTuple()
        #can't use sizer.GetSize(), have to add up the individual row column sizes to get the real width
        l = self.sizer.GetColWidths()
        swidth=0
        for i in range(len(l)):
            swidth+=l[i]
        l= self.sizer.GetRowHeights()
        sheight=0
        for i in range(len(l)):
            sheight+=l[i]
        wfill=width-swidth
        hfill=height-sheight
        if wfill or hfill:
            brush = wx.Brush(self.bgcolor)
            DC.SetBrush(brush)
            DC.SetPen(wx.TRANSPARENT_PEN)
            if wfill:
                DC.DrawRectangle(width-wfill, 0, wfill, height)
            if hfill:
                DC.DrawRectangle(0, height-hfill, width, hfill)
        return wx.Window.OnPaint(self, _)

    def OnKeyDown(self, event):
        key=event.GetKeyCode()
        if key==wx.WXK_NEXT:
           self.scrollby( (self.numrows-1)*7)
        elif key==wx.WXK_PRIOR:
           self.scrollby( (self.numrows-1)*-7)
        elif key==wx.WXK_LEFT:
           self.setday(*normalizedate( self.selecteddate[0], self.selecteddate[1], self.selecteddate[2]-1) )
        elif key==wx.WXK_RIGHT:
           self.setday(*normalizedate( self.selecteddate[0], self.selecteddate[1], self.selecteddate[2]+1) )
        elif key==wx.WXK_UP:
           self.setday(*normalizedate( self.selecteddate[0], self.selecteddate[1], self.selecteddate[2]-7) )
        elif key==wx.WXK_DOWN:
           self.setday(*normalizedate( self.selecteddate[0], self.selecteddate[1], self.selecteddate[2]+7) )
        elif key==wx.WXK_HOME:  # back a month
           self.setday(*normalizedate( self.selecteddate[0], self.selecteddate[1]-1, self.selecteddate[2]) )
        elif key==wx.WXK_END: # forward a month
           self.setday(*normalizedate( self.selecteddate[0], self.selecteddate[1]+1, self.selecteddate[2]) )
        # ::TODO:: activate edit code for return or space on a calendarcell
        else:
           event.Skip()  # control can have it

    def OnMouseWheel(self, event):
        delta=event.GetWheelDelta()
        if delta==0: # as it does on linux
            delta=120
        lines=event.GetWheelRotation()/delta
        self.scrollby(-7*lines)

    def OnLeftDown(self, event):
        cell=event.GetEventObject()
        self.setselection(cell.year, cell.month, cell.day)

    def OnLeftDClick(self,event):
        cell=event.GetEventObject()
        self.OnEdit(cell.year, cell.month, cell.day)

    def OnYearButton(self, event):
        self.popupcalendar.Popup( * (self.selecteddate + (event,)) )

    def OnTodayButton(self, _):
        self.setday(*time.localtime()[:3])

    def OnEraseBackground(self, _):
        pass

    def makerow(self, sizer, row):
        res=[]
        sizer.AddGrowableRow(row)
        for i in range(0,7):
            res.append( CalendarCell(self, -1) )
            sizer.Add( res[-1], flag=wx.EXPAND, pos=(row,i+1))
        return res

    def scrollby(self, amount):
        assert abs(amount)%7==0
        for row in range(0, self.numrows):
                y,m,d=self.rows[row][0].getdate()
                y,m,d=normalizedate(y,m,d+amount)
                self.updaterow(row, y,m,d)
        self.setselection(*self.selecteddate)
        self.label.changenotify()
        self.ensureallpainted()

    def ensureallpainted(self):
        # doesn't return until cells have been painted
        self.Update()

    def OnScrollDown(self, _=None):
        # user has pressed scroll down button
        self.scrollby(7)
        
    def OnScrollUp(self, _=None):
       # user has pressed scroll up button
       self.scrollby(-7)

    def setday(self, year, month, day):
       # makes specified day be shown and selected
       self.showday(year, month, day)
       self.setselection(year, month, day)
       
    def showday(self, year, month, day, rowtoshow=-1):
       """Ensures specified date is onscreen

       @param rowtoshow:   if is >=0 then it will be forced to appear in that row
       """
       # check first cell
       y,m,d=self.rows[0][0].year, self.rows[0][0].month, self.rows[0][0].day
       if rowtoshow==-1:
          if year<y or (year<=y and month<m) or (year<=y and month<=m and day<d):
             rowtoshow=0
       # check last cell   
       y,m,d=self.rows[-1][-1].year, self.rows[-1][-1].month, self.rows[-1][-1].day
       if rowtoshow==-1:
          if year>y or (year>=y and month>m) or (year>=y and month>=m and day>d):
             rowtoshow=self.numrows-1
       if rowtoshow!=-1:
          d=calendar.weekday(year, month, day)
          d=(d+1)%7
          
          d=day-d # go back to begining of week
          d-=7*rowtoshow # then begining of screen
          y,m,d=normalizedate(year, month, d)
          for row in range(0,self.numrows):
             self.updaterow(row, *normalizedate(y, m, d+7*row))
          self.label.changenotify()
          self.ensureallpainted()

    def isvisible(self, year, month, day):
       """Tests if the date is visible to the user

       @rtype: Bool
       """
       y,m,d=self.rows[0][0].year, self.rows[0][0].month, self.rows[0][0].day
       if year<y or (year<=y and month<m) or (year<=y and month<=m and day<d):
           return False
       y,m,d=self.rows[-1][-1].year, self.rows[-1][-1].month, self.rows[-1][-1].day
       if year>y or (year>=y and month>m) or (year>=y and month>=m and day>d):
           return False
       return True

    def RefreshEntry(self, year, month, day):
       """Causes that date's entries to be refreshed.

       Call this if you have changed the data for one day.
       Note that your OnGetEntries will only be called if the date is
       currently visible."""
       if self.isvisible(year,month,day):
           # ::TODO:: find correct cell and only update that
           self.RefreshAllEntries()
   
    def RefreshAllEntries(self):
       """Call this if you have completely changed all your data.

       OnGetEntries will be called for each visible day."""

       for row in self.rows:
           for cell in row:
               cell.setentries(self.OnGetEntries(cell.year, cell.month, cell.day))

   
    def setselection(self, year, month, day):
       """Selects the specifed date if it is visible"""
       self.selecteddate=(year,month,day)
       d=calendar.weekday(year, month, day)
       d=(d+1)%7
       for row in range(0, self.numrows):
          cell=self.rows[row][d]
          if cell.year==year and cell.month==month and cell.day==day:
             self._unselect()
             self.rows[row][d].setattr(self.attrselectedcell)
             self.selectedcell=(row,d)
             self.ensureallpainted()
             return

    def _unselect(self):
       if self.selectedcell!=(-1,-1):
          self.updatecell(*self.selectedcell)
          self.selectedcell=(-1,-1)

    def updatecell(self, row, column, y=-1, m=-1, d=-1):
       if y!=-1:
          self.rows[row][column].setdate(y,m,d)
       if self.rows[row][column].month%2:
          self.rows[row][column].setattr(self.attroddmonth)
       else:
          self.rows[row][column].setattr(self.attrevenmonth)
       if y!=-1 and row==0 and column==0:
          self.year.SetLabel(`y`)
       if y!=-1:
           self.rows[row][column].setentries(self.OnGetEntries(y,m,d))

    def updaterow(self, row, y, m, d):
        daysinmonth=monthrange(y, m)
        for c in range(0,7):
            self.updatecell(row, c, y, m, d)
            if d==daysinmonth:
                d=1
                m+=1
                if m==13:
                    m=1
                    y+=1
                daysinmonth=monthrange(y, m)
            else:
                d+=1
                
    # The following methods should be implemented in derived class.
    # Implementations here are to make it be a nice demo if not subclassed
    
    def OnGetEntries(self, year, month, day):
        """Return a list of entries for the specified y,m,d.

        B{You must implement this method in a derived class}

        The format is ( (hour,min,desc), (hour,min,desc)... )  Hour
        should be in 24 hour format.  You should sort the entries.

        Note that Calendar does not cache any results so you will be
        asked for the same dates as the user scrolls around."""
        
        return (
            (1, 88, "Early morning"),
            (10,15, "Some explanatory text" ),
            (10,30, "It is %04d-%02d-%02d" % (year,month,day)),
            (11,11, "Look at me!"),
            (12,30, "More text here"),
            (15,30, "A very large amount of text that will never fit"),
            (20,30, "Evening drinks"),
            )

    def OnEdit(self, year, month, day):
        """The user wishes to edit the entries for the specified date

        B{You should implement this method in a derived class}
        """
        print "The user wants to edit %04d-%02d-%02d" % (year,month,day)


class PopupCalendar(wx.Dialog):
    """The control that pops up when you click the year button"""
    def __init__(self, parent, calendar, style=wx.SIMPLE_BORDER):
        wx.Dialog.__init__(self, parent, -1, '', style=wx.STAY_ON_TOP|style)
        self.calendar=calendar
        self.control=wx.calendar.CalendarCtrl(self, 1, style=wx.calendar.CAL_SUNDAY_FIRST, pos=(0,0))
        sz=self.control.GetBestSize()
        self.SetSize(sz)
        wx.calendar.EVT_CALENDAR(self, self.control.GetId(), self.OnCalSelected)

    def Popup(self, year, month, day, event):
        d=wx.DateTimeFromDMY(day, month, year)
        self.control.SetDate(d)
        btn=event.GetEventObject()
        pos=btn.ClientToScreen( (0,0) )
        sz=btn.GetSize()
        self.Move( (pos[0], pos[1]+sz.height ) )
        self.ShowModal()

    def OnCalSelected(self, evt):
        dt=evt.GetDate()
        self.calendar.setday(dt.GetYear(), dt.GetMonth()+1, dt.GetDay())
        self.calendar.ensureallpainted()
        self.EndModal(1)
        

_monthranges=[0, 31, -1, 31, 30, 31, 30, 31, 31, 30, 31, 30, 31]

def monthrange(year, month):
    """How many days are in the specified month?

    @rtype: int"""
    if month==2:
        return calendar.monthrange(year, month)[1]
    return _monthranges[month]

def normalizedate(year, month, day):
    """Return a valid date (and an excellent case for metric time)

    And example is the 32nd of January is first of Feb, or Jan -2 is
    December 29th of previous year.  You should call this after doing
    arithmetic on dates (for example you can just subtract 14 from the
    current day and then call this to get the correct date for two weeks
    ago.

    @rtype: tuple
    @return: (year, month, day)
    """

    while day<1 or month<1 or month>12 or (day>28 and day>monthrange(year, month)):
        if day<1:
            month-=1
            if month<1:
                month=12
                year-=1
            num=monthrange(year, month)
            day=num+day
            continue
        if day>28 and day>monthrange(year, month):
            num=calendar.monthrange(year, month)[1]
            month+=1
            if month>12:
                month=1
                year+=1
            day=day-num
            continue    
        if month<1:
            year-=1
            month=month+12
            continue
        if month>12:
            year+=1
            month-=12
            continue
        assert False, "can't get here"

    return year, month, day
            
# Up and down bitmap icons

def getupbitmapData():
    """Returns raw data for the up icon"""
    return \
'\x89PNG\r\n\x1a\n\x00\x00\x00\rIHDR\x00\x00\x00 \x00\x00\x00\x10\x08\x06\
\x00\x00\x00w\x00}Y\x00\x00\x00\x04sBIT\x08\x08\x08\x08|\x08d\x88\x00\x00\
\x00}IDATx\x9c\xbd\xd5K\x0e\xc0 \x08\x04P\xe0\x04\xdc\xff\x94\xdc\xa0\xdd6\
\xad\xca\xf0)n\\\xa83/1FVU\xca\x0e3\xbbT\x95\xd3\x01D$\x95\xf2\xe7<\nx\x97V\
\x10a\xc0\xae,\x8b\x08\x01\xbc\x92\x0c\x02\x06\xa0\xe1Q\x04\x04\x88\x86F\xf6\
\xbb\x80\xec\xdd\xa2\xe7\x8e\x80\xea\x13C\xceo\x01\xd5r4g\t\xe8*G\xf2>\x80\
\xeer/W\x90M\x7f"\xe4\xb48\x81\x90\xc9\xf2\x15\x82+\xdfq\xc7\xb8\x01;]o#\xdc\
D \x03\x00\x00\x00\x00IEND\xaeB`\x82' 

def getupbitmapBitmap():
    """Returns a wx.Bitmap of the up icon"""
    return wx.BitmapFromImage(getupbitmapImage())

def getupbitmapImage():
    """Returns wx.Image of the up icon"""
    stream = cStringIO.StringIO(getupbitmapData())
    return wx.ImageFromStream(stream)

def getdownbitmapData():
    """Returns raw data for the down icon"""
    return \
'\x89PNG\r\n\x1a\n\x00\x00\x00\rIHDR\x00\x00\x00 \x00\x00\x00\x10\x08\x06\
\x00\x00\x00w\x00}Y\x00\x00\x00\x04sBIT\x08\x08\x08\x08|\x08d\x88\x00\x00\
\x00\x80IDATx\x9c\xc5\xd3\xd1\r\x80 \x0c\x04\xd0B\x1c\xe0\xf6\x9f\xb2\x1b\
\xe8\x97\t\x91R\xda\x02\x95/!r\xf7bj\x01@\x7f\xae\xeb}`\xe6;\xbb\x1c@\xa9\
\xed&\xbb\x9c\x88\xa8J\x87Y\xe5\x1d \x03\xf1\xcd\xef\x00\'\x11R\xae\x088\x81\
\x18\xe5\r\x01;\x11Z\x8e\n\xd8\x81\x98\xdd\x9f\x02V\x10\x96{&@\x04a}\xdf\x0c\
\xf0\x84z\xb0.\x80%\xdc\xfb\xa5\xdc\x00\xad$2+!\x80T\x16\x1d\xd40\xa0-]\xf9U\
\x1f\xf8\xca\t\xael-\x16\x86\x00\x00\x00\x00IEND\xaeB`\x82' 

def getdownbitmapBitmap():
    """Returns a wx.Bitmap of the down icon"""
    return wx.BitmapFromImage(getdownbitmapImage())

def getdownbitmapImage():
    """Returns wx.Image of the down icon"""
    stream = cStringIO.StringIO(getdownbitmapData())
    return wx.ImageFromStream(stream)


 

# If run by self, then is a nice demo

if __name__=="__main__":
    class MainWindow(wx.Frame):
        def __init__(self, parent, id, title):
            wx.Frame.__init__(self, parent, id, title, size=(640,480),
                             style=wx.DEFAULT_FRAME_STYLE)
            #self.control=CalendarCell(self, 1) # test just the cell
            #hbs=wx.BoxSizer(wx.VERTICAL)
            self.control=Calendar(self)
            #hbs.Add(self.control, 1, wx.EXPAND)
            #self.SetSizer(hbs)
            self.Show(True)
    
    app=wx.PySimpleApp()
    frame=MainWindow(None, -1, "Calendar Test")
    if False: # change to True to do profiling
        import profile
        profile.run("app.MainLoop()", "fooprof")
    else:
        app.MainLoop()

Index: calendarcontrol.py
===================================================================
RCS file: /cvsroot/bitpim/bitpim/calendarcontrol.py,v
retrieving revision 1.40
diff -u -r1.40 calendarcontrol.py
--- calendarcontrol.py	3 Mar 2005 07:10:31 -0000	1.40
+++ calendarcontrol.py	29 Sep 2005 08:49:40 -0000
@@ -535,9 +535,11 @@
         self.SetSizer(sizer)
         self.sizer=sizer
         self.SetAutoLayout(True)
+        self.bgcolor=self.GetBackgroundColour()
 
         self.popupcalendar=PopupCalendar(self, self)
 
+        wx.EVT_PAINT(self, self.OnPaint)
         wx.EVT_BUTTON(self, self.ID_UP, self.OnScrollUp)
         wx.EVT_BUTTON(self, self.ID_DOWN, self.OnScrollDown)
         wx.EVT_BUTTON(self, self.ID_YEARBUTTON, self.OnYearButton)
@@ -558,6 +560,32 @@
         self.showday(*time.localtime()[:3]+(self.showrow,))
         self.setday(*time.localtime()[:3])
 
+    def OnPaint(self, _=None):
+        DC = wx.BufferedPaintDC(self)
+        self.Layout
+        #calculate the gap at the edge of the sizer control
+        width, height = self.GetClientSizeTuple()
+        #can't use sizer.GetSize(), have to add up the individual row column sizes to get the real width
+        l = self.sizer.GetColWidths()
+        swidth=0
+        for i in range(len(l)):
+            swidth+=l[i]
+        l= self.sizer.GetRowHeights()
+        sheight=0
+        for i in range(len(l)):
+            sheight+=l[i]
+        wfill=width-swidth
+        hfill=height-sheight
+        if wfill or hfill:
+            brush = wx.Brush(self.bgcolor)
+            DC.SetBrush(brush)
+            DC.SetPen(wx.TRANSPARENT_PEN)
+            if wfill:
+                DC.DrawRectangle(width-wfill, 0, wfill, height)
+            if hfill:
+                DC.DrawRectangle(0, height-hfill, width, hfill)
+        return wx.Window.OnPaint(self, _)
+
     def OnKeyDown(self, event):
         key=event.GetKeyCode()
         if key==wx.WXK_NEXT: