wxPython/demo/Treebook.py

   1
   2 import  wx
   3
   4 import ColorPanel
   5 import images
   6
   7 colourList = [ "Aquamarine", "Grey", "Blue", "Blue Violet", "Brown", "Cadet Blue",
   8                "Coral", "Wheat", "Cyan", "Dark Grey", "Dark Green",
   9                "Steel Blue",
  10                ]
  11
  12 #----------------------------------------------------------------------------
  13
  14 def getNextImageID(count):
  15     imID = 0
  16     while True:
  17         yield imID
  18         imID += 1
  19         if imID == count:
  20             imID = 0
  21
  22
  23 class TestTB(wx.Treebook):
  24     def __init__(self, parent, id, log):
  25         wx.Treebook.__init__(self, parent, id, style=
  26                              wx.BK_DEFAULT
  27                              #wx.BK_TOP
  28                              #wx.BK_BOTTOM
  29                              #wx.BK_LEFT
  30                              #wx.BK_RIGHT
  31                             )
  32         self.log = log
  33
  34         # make an image list using the LBXX images
  35         il = wx.ImageList(32, 32)
  36         for x in range(12):
  37             f = getattr(images, 'getLB%02dBitmap' % (x+1))
  38             bmp = f()
  39             il.Add(bmp)
  40         self.AssignImageList(il)
  41         imageIdGenerator = getNextImageID(il.GetImageCount())
  42
  43         # Now make a bunch of panels for the list book
  44         first = True
  45         for colour in colourList:
  46             win = self.makeColorPanel(colour)
  47             self.AddPage(win, colour, imageId=imageIdGenerator.next())
  48             if first:
  49                 st = wx.StaticText(win.win, -1,
  50                           "You can put nearly any type of window here,\n"
  51                           "and the wx.TreeCtrl can be on either side of the\n"
  52                           "Treebook",
  53                           wx.Point(10, 10))
  54                 first = False
  55
  56             win = self.makeColorPanel(colour)
  57             st = wx.StaticText(win.win, -1, "this is a sub-page", (10,10))
  58             self.AddSubPage(win, 'a sub-page', imageId=imageIdGenerator.next())
  59
  60         self.Bind(wx.EVT_TREEBOOK_PAGE_CHANGED, self.OnPageChanged)
  61         self.Bind(wx.EVT_TREEBOOK_PAGE_CHANGING, self.OnPageChanging)
  62
  63         wx.FutureCall(100, self.AdjustSize)
  64
  65     def AdjustSize(self):
  66         print self.GetTreeCtrl().GetBestSize()
  67         self.GetTreeCtrl().InvalidateBestSize()
  68         self.SendSizeEvent()
  69         print self.GetTreeCtrl().GetBestSize()
  70
  71
  72     def makeColorPanel(self, color):
  73         p = wx.Panel(self, -1)
  74         win = ColorPanel.ColoredPanel(p, color)
  75         p.win = win
  76         def OnCPSize(evt, win=win):
  77             win.SetPosition((0,0))
  78             win.SetSize(evt.GetSize())
  79         p.Bind(wx.EVT_SIZE, OnCPSize)
  80         return p
  81
  82
  83     def OnPageChanged(self, event):
  84         old = event.GetOldSelection()
  85         new = event.GetSelection()
  86         sel = self.GetSelection()
  87         self.log.write('OnPageChanged,  old:%d, new:%d, sel:%d\n' % (old, new, sel))
  88         event.Skip()
  89
  90     def OnPageChanging(self, event):
  91         old = event.GetOldSelection()
  92         new = event.GetSelection()
  93         sel = self.GetSelection()
  94         self.log.write('OnPageChanging, old:%d, new:%d, sel:%d\n' % (old, new, sel))
  95         event.Skip()
  96
  97 #----------------------------------------------------------------------------
  98
  99 def runTest(frame, nb, log):
 100     testWin = TestTB(nb, -1, log)
 101     return testWin
 102
 103 #----------------------------------------------------------------------------
 104
 105
 106 overview = """\
 107 <html><body>
 108 <h2>wx.Treebook</h2>
 109 <p>
 110 This class is a control similar to a notebook control, but with a
 111 wx.TreeCtrl instead of a set of tabs.
 112
 113 """
 114
 115
 116
 117 if __name__ == '__main__':
 118     import sys,os
 119     import run
 120     run.main(['', os.path.basename(sys.argv[0])] + sys.argv[1:])
 121
 122
 123