wxPython/demo/SplitTree.py

   1 from wxPython.wx import *
   2 from wxPython.gizmos import *
   3
   4 import images
   5
   6 #----------------------------------------------------------------------
   7
   8 class TestTree(wxRemotelyScrolledTreeCtrl):
   9     def __init__(self, parent, ID, pos=wxDefaultPosition, size=wxDefaultSize,
  10                  style=wxTR_HAS_BUTTONS):
  11         wxRemotelyScrolledTreeCtrl.__init__(self, parent, ID, pos, size, style)
  12         self.SetBackgroundColour("LIGHT BLUE")
  13
  14         # make an image list
  15         self.il = wxImageList(16, 16)
  16         im1 = self.il.AddIcon(wxIconFromXPMData(images.getFolder1Data()))
  17         im2 = self.il.AddIcon(wxIconFromXPMData(images.getFile1Data()))
  18
  19         # Add some items
  20         root = self.AddRoot("Root")
  21         for i in range(30):
  22             item = self.AppendItem(root, "Item %d" % i, im1)
  23             for j in range(10):
  24                 self.AppendItem(item, "Child %d" % j, im2)
  25
  26         self.Expand(root)
  27
  28
  29
  30 class TestValueWindow(wxTreeCompanionWindow):
  31     def __init__(self, parent, ID, pos=wxDefaultPosition, size=wxDefaultSize, style=0):
  32         wxTreeCompanionWindow.__init__(self, parent, ID, pos, size, style)
  33         self.SetBackgroundColour("WHITE")
  34
  35     # This method is called to draw each item in the value window
  36     def DrawItem(self, dc, itemId, rect):
  37         tree = self.GetTreeCtrl()
  38         if tree:
  39             text = "This is "
  40             parent = tree.GetItemParent(itemId)
  41             if parent.IsOk():
  42                 ptext = tree.GetItemText(parent)
  43                 text = text + ptext + " --> "
  44             text = text + tree.GetItemText(itemId)
  45             dc.SetTextForeground("BLACK")
  46             dc.SetBackgroundMode(wxTRANSPARENT)
  47             tw, th = dc.GetTextExtent(text)
  48             x = 5
  49             y = rect.y + max(0, (rect.height - th) / 2)
  50             dc.DrawText(text, x, y)
  51
  52
  53
  54 class TestPanel(wxPanel):
  55     def __init__(self, parent, log):
  56         wxPanel.__init__(self, parent, -1)
  57         self.log = log
  58
  59         scroller = wxSplitterScrolledWindow(self, -1, (50,50), (350, 250),
  60                                       style=wxNO_BORDER | wxCLIP_CHILDREN | wxVSCROLL)
  61         splitter = wxThinSplitterWindow(scroller, -1, style=wxSP_3DBORDER | wxCLIP_CHILDREN)
  62         splitter.SetSashSize(2)
  63         self.tree = TestTree(splitter, -1, style=wxTR_HAS_BUTTONS | wxTR_NO_LINES | wxNO_BORDER)
  64         valueWindow = TestValueWindow(splitter, -1, style=wxNO_BORDER)
  65
  66         splitter.SplitVertically(self.tree, valueWindow)
  67         splitter.SetSashPosition(150)
  68         scroller.SetTargetWindow(self.tree)
  69         scroller.EnableScrolling(FALSE, FALSE)
  70
  71         valueWindow.SetTreeCtrl(self.tree)
  72         self.tree.SetCompanionWindow(valueWindow)
  73
  74
  75
  76 #----------------------------------------------------------------------
  77
  78 def runTest(frame, nb, log):
  79     win = TestPanel(nb, log)
  80     return win
  81
  82
  83 #----------------------------------------------------------------------
  84
  85
  86
  87
  88
  89 overview = """\
  90 This demo shows a collection of classes that were designed to operate
  91 together and provide a tree control with additional columns for each
  92 item.  The classes are wxRemotelyScrolledTreeCtrl, wxTreeCompanionWindow,
  93 wxThinSplitterWindow, and wxSplitterScrolledWindow, some of which may
  94 also be useful by themselves.
  95 """
  96
  97