wxPython/demo/CustomTreeCtrl.py

   1 import wx
   2 import string
   3 import os
   4 import wx.lib.colourselect as csel
   5
   6 import wx.lib.customtreectrl as CT
   7
   8 from Main import opj
   9 import images
  10
  11 #---------------------------------------------------------------------------
  12
  13
  14 penstyle = ["wx.SOLID", "wx.TRANSPARENT", "wx.DOT", "wx.LONG_DASH", "wx.DOT_DASH", "wx.USER_DASH",
  15            "wx.BDIAGONAL_HATCH", "wx.CROSSDIAG_HATCH", "wx.FDIAGONAL_HATCH", "wx.CROSS_HATCH",
  16            "wx.HORIZONTAL_HATCH", "wx.VERTICAL_HATCH"]
  17
  18 ArtIDs = [ "None",
  19            "wx.ART_ADD_BOOKMARK",
  20            "wx.ART_DEL_BOOKMARK",
  21            "wx.ART_HELP_SIDE_PANEL",
  22            "wx.ART_HELP_SETTINGS",
  23            "wx.ART_HELP_BOOK",
  24            "wx.ART_HELP_FOLDER",
  25            "wx.ART_HELP_PAGE",
  26            "wx.ART_GO_BACK",
  27            "wx.ART_GO_FORWARD",
  28            "wx.ART_GO_UP",
  29            "wx.ART_GO_DOWN",
  30            "wx.ART_GO_TO_PARENT",
  31            "wx.ART_GO_HOME",
  32            "wx.ART_FILE_OPEN",
  33            "wx.ART_PRINT",
  34            "wx.ART_HELP",
  35            "wx.ART_TIP",
  36            "wx.ART_REPORT_VIEW",
  37            "wx.ART_LIST_VIEW",
  38            "wx.ART_NEW_DIR",
  39            "wx.ART_HARDDISK",
  40            "wx.ART_FLOPPY",
  41            "wx.ART_CDROM",
  42            "wx.ART_REMOVABLE",
  43            "wx.ART_FOLDER",
  44            "wx.ART_FOLDER_OPEN",
  45            "wx.ART_GO_DIR_UP",
  46            "wx.ART_EXECUTABLE_FILE",
  47            "wx.ART_NORMAL_FILE",
  48            "wx.ART_TICK_MARK",
  49            "wx.ART_CROSS_MARK",
  50            "wx.ART_ERROR",
  51            "wx.ART_QUESTION",
  52            "wx.ART_WARNING",
  53            "wx.ART_INFORMATION",
  54            "wx.ART_MISSING_IMAGE",
  55            "SmileBitmap"
  56            ]
  57
  58 keyMap = {
  59     wx.WXK_BACK : "WXK_BACK",
  60     wx.WXK_TAB : "WXK_TAB",
  61     wx.WXK_RETURN : "WXK_RETURN",
  62     wx.WXK_ESCAPE : "WXK_ESCAPE",
  63     wx.WXK_SPACE : "WXK_SPACE",
  64     wx.WXK_DELETE : "WXK_DELETE",
  65     wx.WXK_START : "WXK_START",
  66     wx.WXK_LBUTTON : "WXK_LBUTTON",
  67     wx.WXK_RBUTTON : "WXK_RBUTTON",
  68     wx.WXK_CANCEL : "WXK_CANCEL",
  69     wx.WXK_MBUTTON : "WXK_MBUTTON",
  70     wx.WXK_CLEAR : "WXK_CLEAR",
  71     wx.WXK_SHIFT : "WXK_SHIFT",
  72     wx.WXK_ALT : "WXK_ALT",
  73     wx.WXK_CONTROL : "WXK_CONTROL",
  74     wx.WXK_MENU : "WXK_MENU",
  75     wx.WXK_PAUSE : "WXK_PAUSE",
  76     wx.WXK_CAPITAL : "WXK_CAPITAL",
  77     wx.WXK_PRIOR : "WXK_PRIOR",
  78     wx.WXK_NEXT : "WXK_NEXT",
  79     wx.WXK_END : "WXK_END",
  80     wx.WXK_HOME : "WXK_HOME",
  81     wx.WXK_LEFT : "WXK_LEFT",
  82     wx.WXK_UP : "WXK_UP",
  83     wx.WXK_RIGHT : "WXK_RIGHT",
  84     wx.WXK_DOWN : "WXK_DOWN",
  85     wx.WXK_SELECT : "WXK_SELECT",
  86     wx.WXK_PRINT : "WXK_PRINT",
  87     wx.WXK_EXECUTE : "WXK_EXECUTE",
  88     wx.WXK_SNAPSHOT : "WXK_SNAPSHOT",
  89     wx.WXK_INSERT : "WXK_INSERT",
  90     wx.WXK_HELP : "WXK_HELP",
  91     wx.WXK_NUMPAD0 : "WXK_NUMPAD0",
  92     wx.WXK_NUMPAD1 : "WXK_NUMPAD1",
  93     wx.WXK_NUMPAD2 : "WXK_NUMPAD2",
  94     wx.WXK_NUMPAD3 : "WXK_NUMPAD3",
  95     wx.WXK_NUMPAD4 : "WXK_NUMPAD4",
  96     wx.WXK_NUMPAD5 : "WXK_NUMPAD5",
  97     wx.WXK_NUMPAD6 : "WXK_NUMPAD6",
  98     wx.WXK_NUMPAD7 : "WXK_NUMPAD7",
  99     wx.WXK_NUMPAD8 : "WXK_NUMPAD8",
 100     wx.WXK_NUMPAD9 : "WXK_NUMPAD9",
 101     wx.WXK_MULTIPLY : "WXK_MULTIPLY",
 102     wx.WXK_ADD : "WXK_ADD",
 103     wx.WXK_SEPARATOR : "WXK_SEPARATOR",
 104     wx.WXK_SUBTRACT : "WXK_SUBTRACT",
 105     wx.WXK_DECIMAL : "WXK_DECIMAL",
 106     wx.WXK_DIVIDE : "WXK_DIVIDE",
 107     wx.WXK_F1 : "WXK_F1",
 108     wx.WXK_F2 : "WXK_F2",
 109     wx.WXK_F3 : "WXK_F3",
 110     wx.WXK_F4 : "WXK_F4",
 111     wx.WXK_F5 : "WXK_F5",
 112     wx.WXK_F6 : "WXK_F6",
 113     wx.WXK_F7 : "WXK_F7",
 114     wx.WXK_F8 : "WXK_F8",
 115     wx.WXK_F9 : "WXK_F9",
 116     wx.WXK_F10 : "WXK_F10",
 117     wx.WXK_F11 : "WXK_F11",
 118     wx.WXK_F12 : "WXK_F12",
 119     wx.WXK_F13 : "WXK_F13",
 120     wx.WXK_F14 : "WXK_F14",
 121     wx.WXK_F15 : "WXK_F15",
 122     wx.WXK_F16 : "WXK_F16",
 123     wx.WXK_F17 : "WXK_F17",
 124     wx.WXK_F18 : "WXK_F18",
 125     wx.WXK_F19 : "WXK_F19",
 126     wx.WXK_F20 : "WXK_F20",
 127     wx.WXK_F21 : "WXK_F21",
 128     wx.WXK_F22 : "WXK_F22",
 129     wx.WXK_F23 : "WXK_F23",
 130     wx.WXK_F24 : "WXK_F24",
 131     wx.WXK_NUMLOCK : "WXK_NUMLOCK",
 132     wx.WXK_SCROLL : "WXK_SCROLL",
 133     wx.WXK_PAGEUP : "WXK_PAGEUP",
 134     wx.WXK_PAGEDOWN : "WXK_PAGEDOWN",
 135     wx.WXK_NUMPAD_SPACE : "WXK_NUMPAD_SPACE",
 136     wx.WXK_NUMPAD_TAB : "WXK_NUMPAD_TAB",
 137     wx.WXK_NUMPAD_ENTER : "WXK_NUMPAD_ENTER",
 138     wx.WXK_NUMPAD_F1 : "WXK_NUMPAD_F1",
 139     wx.WXK_NUMPAD_F2 : "WXK_NUMPAD_F2",
 140     wx.WXK_NUMPAD_F3 : "WXK_NUMPAD_F3",
 141     wx.WXK_NUMPAD_F4 : "WXK_NUMPAD_F4",
 142     wx.WXK_NUMPAD_HOME : "WXK_NUMPAD_HOME",
 143     wx.WXK_NUMPAD_LEFT : "WXK_NUMPAD_LEFT",
 144     wx.WXK_NUMPAD_UP : "WXK_NUMPAD_UP",
 145     wx.WXK_NUMPAD_RIGHT : "WXK_NUMPAD_RIGHT",
 146     wx.WXK_NUMPAD_DOWN : "WXK_NUMPAD_DOWN",
 147     wx.WXK_NUMPAD_PRIOR : "WXK_NUMPAD_PRIOR",
 148     wx.WXK_NUMPAD_PAGEUP : "WXK_NUMPAD_PAGEUP",
 149     wx.WXK_NUMPAD_NEXT : "WXK_NUMPAD_NEXT",
 150     wx.WXK_NUMPAD_PAGEDOWN : "WXK_NUMPAD_PAGEDOWN",
 151     wx.WXK_NUMPAD_END : "WXK_NUMPAD_END",
 152     wx.WXK_NUMPAD_BEGIN : "WXK_NUMPAD_BEGIN",
 153     wx.WXK_NUMPAD_INSERT : "WXK_NUMPAD_INSERT",
 154     wx.WXK_NUMPAD_DELETE : "WXK_NUMPAD_DELETE",
 155     wx.WXK_NUMPAD_EQUAL : "WXK_NUMPAD_EQUAL",
 156     wx.WXK_NUMPAD_MULTIPLY : "WXK_NUMPAD_MULTIPLY",
 157     wx.WXK_NUMPAD_ADD : "WXK_NUMPAD_ADD",
 158     wx.WXK_NUMPAD_SEPARATOR : "WXK_NUMPAD_SEPARATOR",
 159     wx.WXK_NUMPAD_SUBTRACT : "WXK_NUMPAD_SUBTRACT",
 160     wx.WXK_NUMPAD_DECIMAL : "WXK_NUMPAD_DECIMAL",
 161     wx.WXK_NUMPAD_DIVIDE : "WXK_NUMPAD_DIVIDE"
 162     }
 163
 164
 165
 166 #---------------------------------------------------------------------------
 167 # Just A Dialog To Select Pen Styles
 168 #---------------------------------------------------------------------------
 169 class PenDialog(wx.Dialog):
 170
 171     def __init__(self, parent=None, id=-1, title="", pos=wx.DefaultPosition,
 172                  size=wx.DefaultSize, style=wx.DEFAULT_DIALOG_STYLE, oldpen=None,
 173                  pentype=0):
 174
 175         wx.Dialog.__init__(self, parent, id, title, pos, size, style)
 176
 177         self.colourbutton = csel.ColourSelect(self)
 178         self.spinwidth = wx.SpinCtrl(self, -1, "1", min=1, max=3, style=wx.SP_ARROW_KEYS)
 179
 180         self.combostyle = wx.ComboBox(self, -1, choices=penstyle, style=wx.CB_DROPDOWN|wx.CB_READONLY)
 181
 182         choices = ["[1, 1]", "[2, 2]", "[3, 3]", "[4, 4]"]
 183         self.combodash = wx.ComboBox(self, -1, choices=choices, style=wx.CB_DROPDOWN|wx.CB_READONLY)
 184
 185         self.okbutton = wx.Button(self, wx.ID_OK)
 186         self.cancelbutton = wx.Button(self, wx.ID_CANCEL)
 187
 188         self.oldpen = oldpen
 189         self.parent = parent
 190         self.pentype = pentype
 191
 192         self.__set_properties()
 193         self.__do_layout()
 194
 195         self.Bind(wx.EVT_COMBOBOX, self.OnStyle, self.combostyle)
 196         self.Bind(wx.EVT_BUTTON, self.OnOk, self.okbutton)
 197         self.Bind(wx.EVT_BUTTON, self.OnCancel, self.cancelbutton)
 198
 199
 200     def __set_properties(self):
 201
 202         self.SetTitle("Pen Dialog Selector")
 203         self.colourbutton.SetMinSize((25, 25))
 204         self.colourbutton.SetColour(self.oldpen.GetColour())
 205
 206         style = self.oldpen.GetStyle()
 207         for count, st in enumerate(penstyle):
 208             if eval(st) == style:
 209                 self.combostyle.SetSelection(count)
 210                 if count == 5:
 211                     self.combodash.Enable(True)
 212                 else:
 213                     self.combodash.Enable(False)
 214                 break
 215
 216         if self.combodash.IsEnabled():
 217             dashes = repr(self.oldpen.GetDashes())
 218             self.combodash.SetValue(dashes)
 219
 220         self.spinwidth.SetValue(self.oldpen.GetWidth())
 221         self.okbutton.SetDefault()
 222
 223         if self.pentype == 1:
 224             self.spinwidth.Enable(False)
 225
 226
 227     def __do_layout(self):
 228
 229         mainsizer = wx.BoxSizer(wx.VERTICAL)
 230         bottomsizer = wx.BoxSizer(wx.HORIZONTAL)
 231         middlesizer = wx.BoxSizer(wx.VERTICAL)
 232         stylesizer = wx.BoxSizer(wx.HORIZONTAL)
 233         widthsizer = wx.BoxSizer(wx.HORIZONTAL)
 234         coloursizer = wx.BoxSizer(wx.HORIZONTAL)
 235         label_1 = wx.StaticText(self, -1, "Please Choose The Pen Settings:")
 236         label_1.SetFont(wx.Font(8, wx.DEFAULT, wx.NORMAL, wx.BOLD, 0, ""))
 237         mainsizer.Add(label_1, 0, wx.ALL|wx.ADJUST_MINSIZE, 10)
 238         label_2 = wx.StaticText(self, -1, "Pen Colour")
 239         coloursizer.Add(label_2, 0, wx.ALL|wx.ALIGN_CENTER_VERTICAL|wx.ADJUST_MINSIZE, 5)
 240         coloursizer.Add((5, 5), 1, wx.ADJUST_MINSIZE, 0)
 241         coloursizer.Add(self.colourbutton, 0, wx.ALL|wx.ADJUST_MINSIZE, 5)
 242         middlesizer.Add(coloursizer, 0, wx.EXPAND, 0)
 243         label_3 = wx.StaticText(self, -1, "Pen Width")
 244         widthsizer.Add(label_3, 0, wx.ALL|wx.ALIGN_CENTER_VERTICAL|wx.ADJUST_MINSIZE, 5)
 245         widthsizer.Add((5, 5), 1, wx.ADJUST_MINSIZE, 0)
 246         widthsizer.Add(self.spinwidth, 0, wx.ALL|wx.ALIGN_CENTER_VERTICAL|wx.ADJUST_MINSIZE, 5)
 247         middlesizer.Add(widthsizer, 0, wx.EXPAND, 0)
 248         label_4 = wx.StaticText(self, -1, "Pen Style")
 249         stylesizer.Add(label_4, 0, wx.ALL|wx.ALIGN_CENTER_VERTICAL|wx.ADJUST_MINSIZE, 5)
 250         stylesizer.Add((5, 5), 1, wx.ADJUST_MINSIZE, 0)
 251         stylesizer.Add(self.combostyle, 0, wx.ALL|wx.ALIGN_CENTER_VERTICAL|wx.ADJUST_MINSIZE, 5)
 252         stylesizer.Add(self.combodash, 0, wx.ALL|wx.ALIGN_CENTER_VERTICAL|wx.ADJUST_MINSIZE, 5)
 253         middlesizer.Add(stylesizer, 0, wx.BOTTOM|wx.EXPAND, 5)
 254         mainsizer.Add(middlesizer, 1, wx.EXPAND, 0)
 255         bottomsizer.Add(self.okbutton, 0, wx.ALL|wx.ALIGN_CENTER_VERTICAL|wx.ADJUST_MINSIZE, 20)
 256         bottomsizer.Add((20, 20), 1, wx.ADJUST_MINSIZE, 0)
 257         bottomsizer.Add(self.cancelbutton, 0, wx.ALL|wx.ALIGN_CENTER_VERTICAL|wx.ADJUST_MINSIZE, 20)
 258         mainsizer.Add(bottomsizer, 0, wx.EXPAND, 0)
 259         self.SetAutoLayout(True)
 260         self.SetSizer(mainsizer)
 261         mainsizer.Fit(self)
 262         mainsizer.SetSizeHints(self)
 263         self.Layout()
 264         self.Centre()
 265
 266
 267
 268
 269     def OnStyle(self, event):
 270
 271         choice = event.GetEventObject().GetValue()
 272         self.combodash.Enable(choice==5)
 273         event.Skip()
 274
 275
 276     def OnOk(self, event):
 277
 278         colour = self.colourbutton.GetColour()
 279         style = eval(self.combostyle.GetValue())
 280         width = int(self.spinwidth.GetValue())
 281
 282         dashes = None
 283         if self.combostyle.GetSelection() == 5:
 284             dashes = eval(self.combodash.GetValue())
 285
 286         pen = wx.Pen(colour, width, style)
 287
 288         if dashes:
 289             pen.SetDashes(dashes)
 290
 291         pen.SetCap(wx.CAP_BUTT)
 292
 293         if self.pentype == 0:
 294             self.parent.SetConnectionPen(pen)
 295         else:
 296             self.parent.SetBorderPen(pen)
 297
 298         self.Destroy()
 299         event.Skip()
 300
 301
 302     def OnCancel(self, event):
 303
 304         self.Destroy()
 305         event.Skip()
 306
 307
 308 #---------------------------------------------------------------------------
 309 # Just A Dialog To Select Tree Buttons Icons
 310 #---------------------------------------------------------------------------
 311 class TreeButtonsDialog(wx.Dialog):
 312
 313     def __init__(self, parent=None, id=-1, title="", pos=wx.DefaultPosition,
 314                  size=wx.DefaultSize, style=wx.DEFAULT_DIALOG_STYLE, oldicons=None):
 315
 316         wx.Dialog.__init__(self, parent, id, title, pos, size, style)
 317
 318         self.listicons = wx.ListBox(self, -1, choices=["Set 1", "Set 2", "Set 3", "Set 4", "Set 5"], style=wx.LB_SINGLE)
 319
 320         bitmap_plus = opj("bitmaps/plus" + str(oldicons+1) + ".ico")
 321         bitmap_minus = opj("bitmaps/minus" + str(oldicons+1) + ".ico")
 322
 323         bitmap_plus = wx.Image(bitmap_plus, wx.BITMAP_TYPE_ICO)
 324         bitmap_plus.Rescale(24, 24)
 325         bitmap_plus = bitmap_plus.ConvertToBitmap()
 326         bitmap_minus = wx.Image(bitmap_minus, wx.BITMAP_TYPE_ICO)
 327         bitmap_minus.Rescale(24, 24)
 328         bitmap_minus = bitmap_minus.ConvertToBitmap()
 329
 330         self.bitmap_plus = wx.StaticBitmap(self, -1, bitmap_plus)
 331         self.bitmap_minus = wx.StaticBitmap(self, -1, bitmap_minus)
 332
 333         self.okbutton = wx.Button(self, wx.ID_OK)
 334         self.cancelbutton = wx.Button(self, wx.ID_CANCEL)
 335
 336         self.parent = parent
 337
 338         self.__set_properties()
 339         self.__do_layout()
 340
 341         self.Bind(wx.EVT_BUTTON, self.OnOk, self.okbutton)
 342         self.Bind(wx.EVT_BUTTON, self.OnCancel, self.cancelbutton)
 343         self.Bind(wx.EVT_LISTBOX, self.OnListBox, self.listicons)
 344
 345
 346     def __set_properties(self):
 347
 348         self.SetTitle("Tree Buttons Selector")
 349         self.listicons.SetSelection(0)
 350         self.okbutton.SetDefault()
 351
 352
 353     def __do_layout(self):
 354
 355         mainsizer = wx.BoxSizer(wx.VERTICAL)
 356         bottomsizer = wx.BoxSizer(wx.HORIZONTAL)
 357         topsizer = wx.BoxSizer(wx.HORIZONTAL)
 358         rightsizer = wx.BoxSizer(wx.VERTICAL)
 359         sizer_2 = wx.BoxSizer(wx.HORIZONTAL)
 360         sizer_1 = wx.BoxSizer(wx.HORIZONTAL)
 361         label_1 = wx.StaticText(self, -1, "Please Choose One Of These Sets Of Icons:")
 362         label_1.SetFont(wx.Font(8, wx.DEFAULT, wx.NORMAL, wx.BOLD, 0, ""))
 363         mainsizer.Add(label_1, 0, wx.ALL|wx.ADJUST_MINSIZE, 10)
 364         topsizer.Add(self.listicons, 0, wx.ALL|wx.EXPAND|wx.ADJUST_MINSIZE, 5)
 365         label_2 = wx.StaticText(self, -1, "Collapsed")
 366         sizer_1.Add(label_2, 0, wx.RIGHT|wx.ALIGN_CENTER_VERTICAL|wx.ADJUST_MINSIZE, 10)
 367         sizer_1.Add((20, 20), 1, wx.ALIGN_RIGHT|wx.ADJUST_MINSIZE, 0)
 368         sizer_1.Add(self.bitmap_plus, 1, wx.RIGHT|wx.EXPAND|wx.ALIGN_CENTER_VERTICAL|wx.ADJUST_MINSIZE, 10)
 369         rightsizer.Add(sizer_1, 0, wx.EXPAND, 0)
 370         label_3 = wx.StaticText(self, -1, "Expanded")
 371         sizer_2.Add(label_3, 0, wx.RIGHT|wx.ALIGN_CENTER_VERTICAL|wx.ADJUST_MINSIZE, 10)
 372         sizer_2.Add((20, 20), 1, wx.ADJUST_MINSIZE, 0)
 373         sizer_2.Add(self.bitmap_minus, 1, wx.RIGHT|wx.EXPAND|wx.ALIGN_CENTER_VERTICAL|wx.ADJUST_MINSIZE, 10)
 374         rightsizer.Add(sizer_2, 0, wx.EXPAND, 0)
 375         topsizer.Add(rightsizer, 0, wx.ALL|wx.EXPAND, 5)
 376         mainsizer.Add(topsizer, 1, wx.EXPAND, 0)
 377         bottomsizer.Add(self.okbutton, 0, wx.ALL|wx.ADJUST_MINSIZE, 20)
 378         bottomsizer.Add((20, 20), 1, wx.ADJUST_MINSIZE, 0)
 379         bottomsizer.Add(self.cancelbutton, 0, wx.ALL|wx.ADJUST_MINSIZE, 20)
 380         mainsizer.Add(bottomsizer, 0, wx.EXPAND, 0)
 381         self.SetAutoLayout(True)
 382         self.SetSizer(mainsizer)
 383         mainsizer.Fit(self)
 384         mainsizer.SetSizeHints(self)
 385         self.Layout()
 386
 387
 388     def OnListBox(self, event):
 389
 390         selection = self.listicons.GetSelection()
 391         bitmap_plus = opj("bitmaps/plus" + str(selection+1) + ".ico")
 392         bitmap_minus = opj("bitmaps/minus" + str(selection+1) + ".ico")
 393
 394         bitmap_plus = wx.Image(bitmap_plus, wx.BITMAP_TYPE_ICO)
 395         bitmap_plus.Rescale(24, 24)
 396         bitmap_plus = bitmap_plus.ConvertToBitmap()
 397         bitmap_minus = wx.Image(bitmap_minus, wx.BITMAP_TYPE_ICO)
 398         bitmap_minus.Rescale(24, 24)
 399         bitmap_minus = bitmap_minus.ConvertToBitmap()
 400
 401         self.bitmap_plus.SetBitmap(bitmap_plus)
 402         self.bitmap_minus.SetBitmap(bitmap_minus)
 403
 404         self.bitmap_plus.Refresh()
 405         self.bitmap_minus.Refresh()
 406         event.Skip()
 407
 408
 409     def OnOk(self, event):
 410
 411         selection = self.listicons.GetSelection()
 412         self.parent.SetTreeButtons(selection)
 413         self.Destroy()
 414         event.Skip()
 415
 416
 417     def OnCancel(self, event):
 418
 419         self.Destroy()
 420         event.Skip()
 421
 422
 423 #---------------------------------------------------------------------------
 424 # Just A Dialog To Select Tree Check/Radio Item Icons
 425 #---------------------------------------------------------------------------
 426 class CheckDialog(wx.Dialog):
 427
 428     def __init__(self, parent=None, id=-1, title="", pos=wx.DefaultPosition,
 429                  size=wx.DefaultSize, style=wx.DEFAULT_DIALOG_STYLE):
 430
 431         wx.Dialog.__init__(self, parent, id, title, pos, size, style)
 432
 433         self.listicons = wx.ListBox(self, -1, choices=["Set 1", "Set 2"], style=wx.LB_SINGLE)
 434
 435         bitmap_check = wx.Bitmap(opj("bitmaps/checked.ico"), wx.BITMAP_TYPE_ICO)
 436         bitmap_uncheck = wx.Bitmap(opj("bitmaps/notchecked.ico"), wx.BITMAP_TYPE_ICO)
 437         bitmap_flag = wx.Bitmap(opj("bitmaps/flagged.ico"), wx.BITMAP_TYPE_ICO)
 438         bitmap_unflag = wx.Bitmap(opj("bitmaps/notflagged.ico"), wx.BITMAP_TYPE_ICO)
 439
 440         self.bitmap_check = wx.StaticBitmap(self, -1, bitmap_check)
 441         self.bitmap_uncheck = wx.StaticBitmap(self, -1, bitmap_uncheck)
 442         self.bitmap_flag = wx.StaticBitmap(self, -1, bitmap_flag)
 443         self.bitmap_unflag = wx.StaticBitmap(self, -1, bitmap_unflag)
 444
 445         self.okbutton = wx.Button(self, wx.ID_OK)
 446         self.cancelbutton = wx.Button(self, wx.ID_CANCEL)
 447
 448         self.parent = parent
 449
 450         self.__set_properties()
 451         self.__do_layout()
 452
 453         self.Bind(wx.EVT_BUTTON, self.OnOk, self.okbutton)
 454         self.Bind(wx.EVT_BUTTON, self.OnCancel, self.cancelbutton)
 455         self.Bind(wx.EVT_LISTBOX, self.OnListBox, self.listicons)
 456
 457
 458     def __set_properties(self):
 459
 460         self.SetTitle("Check/Radio Icon Selector")
 461         self.listicons.SetSelection(0)
 462         self.okbutton.SetDefault()
 463
 464
 465     def __do_layout(self):
 466
 467         mainsizer = wx.BoxSizer(wx.VERTICAL)
 468         bottomsizer = wx.BoxSizer(wx.HORIZONTAL)
 469         topsizer = wx.BoxSizer(wx.HORIZONTAL)
 470         rightsizer = wx.BoxSizer(wx.VERTICAL)
 471         sizer_2 = wx.BoxSizer(wx.HORIZONTAL)
 472         sizer_1 = wx.BoxSizer(wx.HORIZONTAL)
 473         sizer_3 = wx.BoxSizer(wx.HORIZONTAL)
 474         sizer_4 = wx.BoxSizer(wx.HORIZONTAL)
 475         label_1 = wx.StaticText(self, -1, "Please Choose One Of These Sets Of Icons:")
 476         label_1.SetFont(wx.Font(8, wx.DEFAULT, wx.NORMAL, wx.BOLD, 0, ""))
 477         mainsizer.Add(label_1, 0, wx.ALL|wx.ADJUST_MINSIZE, 10)
 478         topsizer.Add(self.listicons, 0, wx.ALL|wx.EXPAND|wx.ADJUST_MINSIZE, 5)
 479         label_2 = wx.StaticText(self, -1, "Checked")
 480         sizer_1.Add(label_2, 0, wx.RIGHT|wx.ALIGN_CENTER_VERTICAL|wx.ADJUST_MINSIZE, 10)
 481         sizer_1.Add((20, 20), 1, wx.ALIGN_RIGHT|wx.ADJUST_MINSIZE, 0)
 482         sizer_1.Add(self.bitmap_check, 0, wx.RIGHT|wx.ALIGN_CENTER_VERTICAL|wx.ADJUST_MINSIZE, 10)
 483         rightsizer.Add(sizer_1, 0, wx.EXPAND, 0)
 484         label_3 = wx.StaticText(self, -1, "Not Checked")
 485         sizer_2.Add(label_3, 0, wx.RIGHT|wx.ALIGN_CENTER_VERTICAL|wx.ADJUST_MINSIZE, 10)
 486         sizer_2.Add((20, 20), 1, wx.ADJUST_MINSIZE, 0)
 487         sizer_2.Add(self.bitmap_uncheck, 0, wx.RIGHT|wx.ALIGN_CENTER_VERTICAL|wx.ADJUST_MINSIZE, 10)
 488         rightsizer.Add(sizer_2, 0, wx.EXPAND, 0)
 489         label_4 = wx.StaticText(self, -1, "Flagged")
 490         sizer_3.Add(label_4, 0, wx.RIGHT|wx.ALIGN_CENTER_VERTICAL|wx.ADJUST_MINSIZE, 10)
 491         sizer_3.Add((20, 20), 1, wx.ADJUST_MINSIZE, 0)
 492         sizer_3.Add(self.bitmap_flag, 0, wx.RIGHT|wx.ALIGN_CENTER_VERTICAL|wx.ADJUST_MINSIZE, 10)
 493         rightsizer.Add(sizer_3, 0, wx.EXPAND, 0)
 494         label_5 = wx.StaticText(self, -1, "Not Flagged")
 495         sizer_4.Add(label_5, 0, wx.RIGHT|wx.ALIGN_CENTER_VERTICAL|wx.ADJUST_MINSIZE, 10)
 496         sizer_4.Add((20, 20), 1, wx.ADJUST_MINSIZE, 0)
 497         sizer_4.Add(self.bitmap_unflag, 0, wx.RIGHT|wx.ALIGN_CENTER_VERTICAL|wx.ADJUST_MINSIZE, 10)
 498         rightsizer.Add(sizer_4, 0, wx.EXPAND, 0)
 499
 500         topsizer.Add(rightsizer, 0, wx.ALL|wx.EXPAND, 5)
 501         mainsizer.Add(topsizer, 1, wx.EXPAND, 0)
 502         bottomsizer.Add(self.okbutton, 0, wx.ALL|wx.ADJUST_MINSIZE, 20)
 503         bottomsizer.Add((20, 20), 1, wx.ADJUST_MINSIZE, 0)
 504         bottomsizer.Add(self.cancelbutton, 0, wx.ALL|wx.ADJUST_MINSIZE, 20)
 505         mainsizer.Add(bottomsizer, 0, wx.EXPAND, 0)
 506         self.SetAutoLayout(True)
 507         self.SetSizer(mainsizer)
 508         mainsizer.Fit(self)
 509         mainsizer.SetSizeHints(self)
 510         self.Layout()
 511
 512
 513     def OnListBox(self, event):
 514
 515         selection = self.listicons.GetSelection()
 516
 517         if selection == 0:
 518             bitmap_check = opj("bitmaps/checked.ico")
 519             bitmap_uncheck = opj("bitmaps/notchecked.ico")
 520             bitmap_flag = opj("bitmaps/flagged.ico")
 521             bitmap_unflag = opj("bitmaps/notflagged.ico")
 522         else:
 523             bitmap_check = opj("bitmaps/aquachecked.ico")
 524             bitmap_uncheck = opj("bitmaps/aquanotchecked.ico")
 525             bitmap_flag = opj("bitmaps/aquaflagged.ico")
 526             bitmap_unflag = opj("bitmaps/aquanotflagged.ico")
 527
 528         bitmap_check = wx.Bitmap(bitmap_check, wx.BITMAP_TYPE_ICO)
 529         bitmap_uncheck = wx.Bitmap(bitmap_uncheck, wx.BITMAP_TYPE_ICO)
 530         bitmap_flag = wx.Bitmap(bitmap_flag, wx.BITMAP_TYPE_ICO)
 531         bitmap_unflag = wx.Bitmap(bitmap_unflag, wx.BITMAP_TYPE_ICO)
 532
 533         self.bitmap_uncheck.SetBitmap(bitmap_uncheck)
 534         self.bitmap_check.SetBitmap(bitmap_check)
 535         self.bitmap_unflag.SetBitmap(bitmap_unflag)
 536         self.bitmap_flag.SetBitmap(bitmap_flag)
 537
 538         self.bitmap_check.Refresh()
 539         self.bitmap_uncheck.Refresh()
 540         self.bitmap_flag.Refresh()
 541         self.bitmap_unflag.Refresh()
 542
 543         event.Skip()
 544
 545
 546     def OnOk(self, event):
 547
 548         selection = self.listicons.GetSelection()
 549         self.parent.SetCheckRadio(selection)
 550         self.Destroy()
 551         event.Skip()
 552
 553
 554     def OnCancel(self, event):
 555
 556         self.Destroy()
 557         event.Skip()
 558
 559
 560 #---------------------------------------------------------------------------
 561 # Just A Dialog To Select Tree Items Icons
 562 #---------------------------------------------------------------------------
 563 class TreeIcons(wx.Dialog):
 564
 565     def __init__(self, parent=None, id=-1, title="", pos=wx.DefaultPosition,
 566                  size=wx.DefaultSize, style=wx.DEFAULT_DIALOG_STYLE, oldpen=None,
 567                  bitmaps=None):
 568
 569         wx.Dialog.__init__(self, parent, id, title, pos, size, style)
 570
 571         self.bitmaps = [None, None, None, None]
 572         empty = wx.EmptyBitmap(16, 16)
 573         self.parent = parent
 574
 575         self.bitmaps[0] = wx.StaticBitmap(self, -1, empty)
 576         self.combonormal = wx.ComboBox(self, -1, choices=ArtIDs, style=wx.CB_DROPDOWN|wx.CB_READONLY)
 577         self.bitmaps[1] = wx.StaticBitmap(self, -1, empty)
 578         self.comboselected = wx.ComboBox(self, -1, choices=ArtIDs, style=wx.CB_DROPDOWN|wx.CB_READONLY)
 579         self.bitmaps[2] = wx.StaticBitmap(self, -1, empty)
 580         self.comboexpanded = wx.ComboBox(self, -1, choices=ArtIDs, style=wx.CB_DROPDOWN|wx.CB_READONLY)
 581         self.bitmaps[3] = wx.StaticBitmap(self, -1, empty)
 582         self.comboselectedexpanded = wx.ComboBox(self, -1, choices=ArtIDs, style=wx.CB_DROPDOWN|wx.CB_READONLY)
 583         self.okbutton = wx.Button(self, wx.ID_OK)
 584         self.cancelbutton = wx.Button(self, wx.ID_CANCEL)
 585
 586         self.combonormal.SetSelection(bitmaps[0] >= 0 and bitmaps[0]+1 or 0)
 587         self.comboselected.SetSelection(bitmaps[1] >= 0 and bitmaps[1]+1 or 0)
 588         self.comboexpanded.SetSelection(bitmaps[2] >= 0 and bitmaps[2]+1 or 0)
 589         self.comboselectedexpanded.SetSelection(bitmaps[3] >= 0 and bitmaps[3]+1 or 0)
 590
 591         self.GetBitmaps(bitmaps)
 592
 593         self.__set_properties()
 594         self.__do_layout()
 595
 596         self.Bind(wx.EVT_COMBOBOX, self.OnComboNormal, self.combonormal)
 597         self.Bind(wx.EVT_COMBOBOX, self.OnComboSelected, self.comboselected)
 598         self.Bind(wx.EVT_COMBOBOX, self.OnComboExpanded, self.comboexpanded)
 599         self.Bind(wx.EVT_COMBOBOX, self.OnComboSelectedExpanded, self.comboselectedexpanded)
 600         self.Bind(wx.EVT_BUTTON, self.OnOk, self.okbutton)
 601         self.Bind(wx.EVT_BUTTON, self.OnCancel, self.cancelbutton)
 602
 603
 604     def __set_properties(self):
 605
 606         self.SetTitle("Item Icon Selector")
 607         self.okbutton.SetDefault()
 608
 609
 610     def __do_layout(self):
 611
 612         mainsizer = wx.BoxSizer(wx.VERTICAL)
 613         sizer_2 = wx.BoxSizer(wx.HORIZONTAL)
 614         gridsizer = wx.FlexGridSizer(4, 3, 5, 5)
 615         label_1 = wx.StaticText(self, -1, "Please Choose The Icons For This Item (All Are Optional):")
 616         label_1.SetFont(wx.Font(8, wx.DEFAULT, wx.NORMAL, wx.BOLD, 0, ""))
 617         mainsizer.Add(label_1, 0, wx.ALL|wx.ADJUST_MINSIZE, 10)
 618         label_2 = wx.StaticText(self, -1, "TreeIcon_Normal:")
 619         gridsizer.Add(label_2, 0, wx.LEFT|wx.ALIGN_CENTER_VERTICAL|wx.ADJUST_MINSIZE, 5)
 620         gridsizer.Add(self.bitmaps[0], 0, wx.LEFT|wx.RIGHT|wx.ALIGN_CENTER_VERTICAL|wx.ADJUST_MINSIZE, 5)
 621         gridsizer.Add(self.combonormal, 0, wx.ALIGN_CENTER_VERTICAL|wx.ADJUST_MINSIZE, 0)
 622         label_3 = wx.StaticText(self, -1, "TreeIcon_Selected:")
 623         gridsizer.Add(label_3, 0, wx.LEFT|wx.ALIGN_CENTER_VERTICAL|wx.ADJUST_MINSIZE, 5)
 624         gridsizer.Add(self.bitmaps[1], 0, wx.LEFT|wx.RIGHT|wx.ALIGN_CENTER_VERTICAL|wx.ADJUST_MINSIZE, 5)
 625         gridsizer.Add(self.comboselected, 0, wx.ALIGN_CENTER_VERTICAL|wx.ADJUST_MINSIZE, 0)
 626         label_4 = wx.StaticText(self, -1, "TreeIcon_Expanded:")
 627         gridsizer.Add(label_4, 0, wx.LEFT|wx.ALIGN_CENTER_VERTICAL|wx.ADJUST_MINSIZE, 5)
 628         gridsizer.Add(self.bitmaps[2], 0, wx.LEFT|wx.RIGHT|wx.ALIGN_CENTER_VERTICAL|wx.ADJUST_MINSIZE, 5)
 629         gridsizer.Add(self.comboexpanded, 0, wx.ALIGN_CENTER_VERTICAL|wx.ADJUST_MINSIZE, 0)
 630         label_5 = wx.StaticText(self, -1, "TreeIcon_SelectedExpanded:")
 631         gridsizer.Add(label_5, 0, wx.LEFT|wx.ALIGN_CENTER_VERTICAL|wx.ADJUST_MINSIZE, 5)
 632         gridsizer.Add(self.bitmaps[3], 0, wx.LEFT|wx.RIGHT|wx.ALIGN_CENTER_VERTICAL|wx.ADJUST_MINSIZE, 5)
 633         gridsizer.Add(self.comboselectedexpanded, 0, wx.ALIGN_CENTER_VERTICAL|wx.ADJUST_MINSIZE, 0)
 634         gridsizer.AddGrowableCol(0)
 635         gridsizer.AddGrowableCol(1)
 636         gridsizer.AddGrowableCol(2)
 637         mainsizer.Add(gridsizer, 0, wx.ALL|wx.EXPAND, 5)
 638         sizer_2.Add(self.okbutton, 0, wx.ALL|wx.ALIGN_CENTER_VERTICAL|wx.ADJUST_MINSIZE, 20)
 639         sizer_2.Add((20, 20), 1, wx.ADJUST_MINSIZE, 0)
 640         sizer_2.Add(self.cancelbutton, 0, wx.ALL|wx.ALIGN_CENTER_VERTICAL|wx.ADJUST_MINSIZE, 20)
 641         mainsizer.Add(sizer_2, 1, wx.EXPAND, 0)
 642         self.SetAutoLayout(True)
 643         self.SetSizer(mainsizer)
 644         mainsizer.Fit(self)
 645         mainsizer.SetSizeHints(self)
 646         self.Layout()
 647         self.Centre()
 648
 649
 650     def OnComboNormal(self, event):
 651
 652         input = event.GetSelection()
 653         self.GetBitmap(input, 0)
 654         event.Skip()
 655
 656
 657     def OnComboSelected(self, event):
 658
 659         input = event.GetSelection()
 660         self.GetBitmap(input, 1)
 661         event.Skip()
 662
 663
 664     def OnComboExpanded(self, event):
 665
 666         input = event.GetSelection()
 667         self.GetBitmap(input, 2)
 668         event.Skip()
 669
 670
 671     def OnComboSelectedExpanded(self, event):
 672
 673         input = event.GetSelection()
 674         self.GetBitmap(input, 3)
 675         event.Skip()
 676
 677
 678     def OnOk(self, event):
 679
 680         bitmaps = [-1, -1, -1, -1]
 681         normal = self.combonormal.GetSelection()
 682         selected = self.comboselected.GetSelection()
 683         expanded = self.comboexpanded.GetSelection()
 684         selexp = self.comboselectedexpanded.GetSelection()
 685
 686         bitmaps = [(normal > 0 and normal or -1), (selected > 0 and selected or -1),
 687                    (expanded > 0 and expanded or -1), (selexp > 0 and selexp or -1)]
 688
 689         newbitmaps = []
 690
 691         for bmp in bitmaps:
 692             if bmp > 0:
 693                 newbitmaps.append(bmp-1)
 694             else:
 695                 newbitmaps.append(bmp)
 696
 697         self.parent.SetNewIcons(newbitmaps)
 698
 699         self.Destroy()
 700         event.Skip()
 701
 702
 703     def OnCancel(self, event):
 704
 705         self.Destroy()
 706         event.Skip()
 707
 708
 709     def GetBitmap(self, input, which):
 710
 711         if input == 0:
 712             bmp = wx.EmptyBitmap(16,16)
 713             self.ClearBmp(bmp)
 714         elif input > 36:
 715             bmp = images.getSmilesBitmap()
 716         else:
 717             bmp = wx.ArtProvider_GetBitmap(eval(ArtIDs[input]), wx.ART_OTHER, (16,16))
 718             if not bmp.Ok():
 719                 bmp = wx.EmptyBitmap(16,16)
 720                 self.ClearBmp(bmp)
 721
 722         self.bitmaps[which].SetBitmap(bmp)
 723         self.bitmaps[which].Refresh()
 724
 725
 726     def GetBitmaps(self, bitmaps):
 727
 728         output = []
 729
 730         for count, input in enumerate(bitmaps):
 731             if input < 0:
 732                 bmp = wx.EmptyBitmap(16,16)
 733                 self.ClearBmp(bmp)
 734             elif input > 35:
 735                 bmp = images.getSmilesBitmap()
 736             else:
 737                 bmp = wx.ArtProvider_GetBitmap(eval(ArtIDs[input+1]), wx.ART_OTHER, (16,16))
 738                 if not bmp.Ok():
 739                     bmp = wx.EmptyBitmap(16,16)
 740                     self.ClearBmp(bmp)
 741
 742             self.bitmaps[count].SetBitmap(bmp)
 743
 744
 745     def ClearBmp(self, bmp):
 746
 747         dc = wx.MemoryDC()
 748         dc.SelectObject(bmp)
 749         dc.SetBackground(wx.Brush("white"))
 750         dc.Clear()
 751
 752
 753 #---------------------------------------------------------------------------
 754 # CustomTreeCtrl Demo Implementation
 755 #---------------------------------------------------------------------------
 756 class CustomTreeCtrlDemo(wx.Panel):
 757
 758     def __init__(self, parent, log):
 759         wx.Panel.__init__(self, parent)
 760
 761         self.log = log
 762         self.oldicons = 0
 763
 764         splitter = wx.SplitterWindow(self, -1, style=wx.CLIP_CHILDREN | wx.SP_LIVE_UPDATE | wx.SP_3D)
 765
 766         # Create the CustomTreeCtrl, using a derived class defined below
 767         self.tree = CustomTreeCtrl(splitter, -1, log=self.log, style=wx.SUNKEN_BORDER)
 768
 769         self.leftpanel = wx.ScrolledWindow(splitter, -1, style=wx.SUNKEN_BORDER)
 770         self.leftpanel.SetScrollRate(20,20)
 771         width = self.PopulateLeftPanel(self.tree.styles, self.tree.events)
 772
 773         # add the windows to the splitter and split it.
 774         splitter.SplitVertically(self.leftpanel, self.tree, 300)
 775         splitter.SetMinimumPaneSize(width+5)
 776
 777         sizer = wx.BoxSizer()
 778         sizer.Add(splitter, 1, wx.EXPAND)
 779         self.SetSizer(sizer)
 780
 781
 782     def PopulateLeftPanel(self, styles, events):
 783         pnl = wx.Panel(self.leftpanel)
 784         mainsizer = wx.BoxSizer(wx.VERTICAL)
 785         recreatetree = wx.Button(pnl, -1, "Recreate CustomTreeCtrl")
 786         mainsizer.Add(recreatetree, 0, wx.ALL|wx.ALIGN_CENTER, 10)
 787         recreatetree.Bind(wx.EVT_BUTTON, self.OnRecreateTree)
 788
 789         staticboxstyles = wx.StaticBox(pnl, -1, "CustomTreeCtrl Styles")
 790         stylesizer = wx.StaticBoxSizer(staticboxstyles, wx.VERTICAL)
 791         staticboxevents = wx.StaticBox(pnl, -1, "CustomTreeCtrl Events")
 792         eventssizer = wx.StaticBoxSizer(staticboxevents, wx.VERTICAL)
 793         staticboxcolours = wx.StaticBox(pnl, -1, "CustomTreeCtrl Images/Colours")
 794         colourssizer = wx.StaticBoxSizer(staticboxcolours, wx.VERTICAL)
 795         staticboxthemes = wx.StaticBox(pnl, -1, "CustomTreeCtrl Themes/Gradients")
 796         themessizer = wx.StaticBoxSizer(staticboxthemes, wx.VERTICAL)
 797
 798         self.treestyles = []
 799         self.treeevents = []
 800
 801         for count, style in enumerate(styles):
 802
 803             if count == 0:
 804                 tags = wx.ALL
 805             else:
 806                 tags = wx.LEFT|wx.RIGHT|wx.BOTTOM
 807
 808             if style != "TR_DEFAULT_STYLE":
 809                 check = wx.CheckBox(pnl, -1, style)
 810                 stylesizer.Add(check, 0, tags, 3)
 811
 812                 if style in ["TR_HAS_BUTTONS", "TR_HAS_VARIABLE_ROW_HEIGHT"]:
 813                     check.SetValue(1)
 814
 815                 if style == "TR_HAS_VARIABLE_ROW_HEIGHT":
 816                     check.Enable(False)
 817
 818                 check.Bind(wx.EVT_CHECKBOX, self.OnCheckStyle)
 819                 self.treestyles.append(check)
 820
 821
 822         for count, event in enumerate(events):
 823
 824             if count == 0:
 825                 tags = wx.ALL
 826             else:
 827                 tags = wx.LEFT|wx.RIGHT|wx.BOTTOM
 828
 829             if count not in [6, 17, 22, 23]:
 830                 check = wx.CheckBox(pnl, -1, event)
 831                 eventssizer.Add(check, 0, tags, 3)
 832
 833                 if event in ["EVT_TREE_ITEM_EXPANDED", "EVT_TREE_ITEM_COLLAPSED",
 834                              "EVT_TREE_SEL_CHANGED", "EVT_TREE_SEL_CHANGING"]:
 835
 836                     check.SetValue(1)
 837
 838                 check.Bind(wx.EVT_CHECKBOX, self.OnCheckEvent)
 839                 self.treeevents.append(check)
 840
 841         sizer1 = wx.BoxSizer(wx.HORIZONTAL)
 842         label = wx.StaticText(pnl, -1, "Connection Pen")
 843         font = label.GetFont()
 844         font.SetWeight(wx.FONTWEIGHT_BOLD)
 845         label.SetFont(font)
 846         buttonconnection = wx.Button(pnl, -1, "Choose...")
 847         buttonconnection.Bind(wx.EVT_BUTTON, self.OnButtonConnection)
 848         sizer1.Add(label, 0, wx.ALL|wx.ALIGN_CENTER, 5)
 849         sizer1.Add((1,0), 1, wx.EXPAND)
 850         sizer1.Add(buttonconnection, 0, wx.ALL|wx.ALIGN_CENTER_HORIZONTAL|wx.ALIGN_RIGHT, 5)
 851
 852         sizer2 = wx.BoxSizer(wx.HORIZONTAL)
 853         label = wx.StaticText(pnl, -1, "Border Pen")
 854         font = label.GetFont()
 855         font.SetWeight(wx.FONTWEIGHT_BOLD)
 856         label.SetFont(font)
 857         buttonborder = wx.Button(pnl, -1, "Choose...")
 858         buttonborder.Bind(wx.EVT_BUTTON, self.OnButtonBorder)
 859         sizer2.Add(label, 0, wx.LEFT|wx.RIGHT|wx.BOTTOM|wx.ALIGN_CENTER, 5)
 860         sizer2.Add((1,0), 1, wx.EXPAND)
 861         sizer2.Add(buttonborder, 0, wx.LEFT|wx.RIGHT|wx.BOTTOM|wx.ALIGN_CENTER_HORIZONTAL|wx.ALIGN_RIGHT, 5)
 862
 863         sizer3 = wx.BoxSizer(wx.HORIZONTAL)
 864         label = wx.StaticText(pnl, -1, "Tree Buttons")
 865         font = label.GetFont()
 866         font.SetWeight(wx.FONTWEIGHT_BOLD)
 867         label.SetFont(font)
 868         buttontree = wx.Button(pnl, -1, "Choose...")
 869         buttontree.Bind(wx.EVT_BUTTON, self.OnButtonTree)
 870         sizer3.Add(label, 0, wx.LEFT|wx.RIGHT|wx.BOTTOM|wx.ALIGN_CENTER, 5)
 871         sizer3.Add((1,0), 1, wx.EXPAND)
 872         sizer3.Add(buttontree, 0, wx.LEFT|wx.RIGHT|wx.BOTTOM|wx.ALIGN_CENTER_HORIZONTAL|wx.ALIGN_RIGHT, 5)
 873
 874         sizer4 = wx.BoxSizer(wx.HORIZONTAL)
 875         label = wx.StaticText(pnl, -1, "Check/Radio Buttons")
 876         font = label.GetFont()
 877         font.SetWeight(wx.FONTWEIGHT_BOLD)
 878         label.SetFont(font)
 879         buttoncr = wx.Button(pnl, -1, "Choose...")
 880         buttoncr.Bind(wx.EVT_BUTTON, self.OnButtonCheckRadio)
 881         sizer4.Add(label, 0, wx.LEFT|wx.RIGHT|wx.BOTTOM|wx.ALIGN_CENTER, 5)
 882         sizer4.Add((1,0), 1, wx.EXPAND)
 883         sizer4.Add(buttoncr, 0, wx.LEFT|wx.RIGHT|wx.BOTTOM|wx.ALIGN_CENTER_HORIZONTAL|wx.ALIGN_RIGHT, 5)
 884
 885         sizer5 = wx.BoxSizer(wx.HORIZONTAL)
 886         radioimage = wx.RadioButton(pnl, -1, "Image Background", style=wx.RB_GROUP)
 887         radioimage.Bind(wx.EVT_RADIOBUTTON, self.OnBackgroundImage)
 888         self.imagebutton = wx.Button(pnl, -1, "Choose...")
 889         self.imagebutton.Bind(wx.EVT_BUTTON, self.OnChooseImage)
 890         sizer5.Add(radioimage, 0, wx.ALL|wx.ALIGN_CENTER, 5)
 891         sizer5.Add((1,0), 1, wx.EXPAND)
 892         sizer5.Add(self.imagebutton, 0, wx.ALL|wx.ALIGN_CENTER_HORIZONTAL|wx.ALIGN_RIGHT, 5)
 893
 894         sizer6 = wx.BoxSizer(wx.HORIZONTAL)
 895         radiobackground = wx.RadioButton(pnl, -1, "Background Colour")
 896         radiobackground.Bind(wx.EVT_RADIOBUTTON, self.OnBackgroundColour)
 897         self.backbutton = csel.ColourSelect(pnl, -1, "Choose...", wx.WHITE)
 898         self.backbutton.Bind(csel.EVT_COLOURSELECT, self.OnChooseBackground)
 899         sizer6.Add(radiobackground, 0, wx.LEFT|wx.RIGHT|wx.BOTTOM|wx.ALIGN_CENTER, 5)
 900         sizer6.Add((1,0), 1, wx.EXPAND)
 901         sizer6.Add(self.backbutton, 0, wx.LEFT|wx.RIGHT|wx.BOTTOM|wx.ALIGN_CENTER, 5)
 902
 903         colourssizer.Add(sizer1, 0, wx.EXPAND)
 904         colourssizer.Add(sizer2, 0, wx.EXPAND)
 905         colourssizer.Add(sizer3, 0, wx.EXPAND)
 906         colourssizer.Add(sizer4, 0, wx.EXPAND)
 907         colourssizer.Add(sizer5, 0, wx.EXPAND)
 908         colourssizer.Add(sizer6, 0, wx.EXPAND)
 909
 910         sizera = wx.BoxSizer(wx.HORIZONTAL)
 911         self.checknormal = wx.CheckBox(pnl, -1, "Standard Colours")
 912         self.focus = csel.ColourSelect(pnl, -1, "Focus",
 913                                        self.tree.GetHilightFocusColour())
 914         self.unfocus = csel.ColourSelect(pnl, -1, "Non-Focus",
 915                                          self.tree.GetHilightNonFocusColour())
 916         self.checknormal.Bind(wx.EVT_CHECKBOX, self.OnCheckNormal)
 917         self.focus.Bind(csel.EVT_COLOURSELECT, self.OnFocusColour)
 918         self.unfocus.Bind(csel.EVT_COLOURSELECT, self.OnNonFocusColour)
 919         sizera1 = wx.BoxSizer(wx.VERTICAL)
 920         sizera1.Add(self.focus, 0, wx.BOTTOM, 2)
 921         sizera1.Add(self.unfocus, 0)
 922         sizera.Add(self.checknormal, 0, wx.ALL, 3)
 923         sizera.Add((1, 0), 1, wx.EXPAND)
 924         sizera.Add(sizera1, 0, wx.ALL|wx.ALIGN_CENTER_HORIZONTAL|wx.EXPAND, 3)
 925
 926         sizerb = wx.BoxSizer(wx.VERTICAL)
 927         self.checkgradient = wx.CheckBox(pnl, -1, "Gradient Theme")
 928         self.checkgradient.Bind(wx.EVT_CHECKBOX, self.OnCheckGradient)
 929         sizerb1 = wx.BoxSizer(wx.HORIZONTAL)
 930         sizerb1.Add((10, 0))
 931         self.radiohorizontal = wx.RadioButton(pnl, -1, "Horizontal", style=wx.RB_GROUP)
 932         self.radiohorizontal.Bind(wx.EVT_RADIOBUTTON, self.OnHorizontal)
 933         sizerb1.Add(self.radiohorizontal, 0, wx.TOP|wx.BOTTOM, 3)
 934         sizerb2 = wx.BoxSizer(wx.HORIZONTAL)
 935         sizerb2.Add((10, 0))
 936         self.radiovertical = wx.RadioButton(pnl, -1, "Vertical")
 937         self.radiovertical.Bind(wx.EVT_RADIOBUTTON, self.OnVertical)
 938         sizerb2.Add(self.radiovertical, 0, wx.BOTTOM, 3)
 939         sizerb3 = wx.BoxSizer(wx.HORIZONTAL)
 940         self.firstcolour = csel.ColourSelect(pnl, -1, "First Colour",
 941                                              self.tree.GetFirstGradientColour())
 942         self.secondcolour = csel.ColourSelect(pnl, -1, "Second Colour",
 943                                               self.tree.GetSecondGradientColour())
 944         self.firstcolour.Bind(csel.EVT_COLOURSELECT, self.OnFirstColour)
 945         self.secondcolour.Bind(csel.EVT_COLOURSELECT, self.OnSecondColour)
 946         sizerb3.Add(self.firstcolour, 0, wx.TOP|wx.BOTTOM|wx.ALIGN_CENTER_HORIZONTAL, 3)
 947         sizerb3.Add(self.secondcolour, 0, wx.LEFT|wx.TOP|wx.BOTTOM|wx.ALIGN_CENTER_HORIZONTAL, 3)
 948         sizerb.Add(self.checkgradient, 0, wx.ALL, 3)
 949         sizerb.Add(sizerb1, 0)
 950         sizerb.Add(sizerb2, 0)
 951         sizerb.Add(sizerb3, 0, wx.ALIGN_CENTER)
 952
 953         self.checkvista = wx.CheckBox(pnl, -1, "Windows Vista Theme")
 954         self.checkvista.Bind(wx.EVT_CHECKBOX, self.OnVista)
 955
 956         themessizer.Add(sizera, 0, wx.EXPAND)
 957         themessizer.Add(sizerb, 0, wx.EXPAND)
 958         themessizer.Add((0, 5))
 959         themessizer.Add(self.checkvista, 0, wx.EXPAND|wx.ALL, 3)
 960
 961         mainsizer.Add(stylesizer, 0, wx.EXPAND|wx.ALL, 5)
 962         mainsizer.Add(colourssizer, 0, wx.EXPAND|wx.ALL, 5)
 963         mainsizer.Add(themessizer, 0, wx.EXPAND|wx.ALL, 5)
 964         mainsizer.Add(eventssizer, 0, wx.EXPAND|wx.ALL, 5)
 965
 966         pnl.SetSizer(mainsizer)
 967         pnl.Fit()
 968
 969         swsizer = wx.BoxSizer(wx.VERTICAL)
 970         swsizer.Add(pnl, 0, wx.EXPAND)
 971         self.leftpanel.SetSizer(swsizer)
 972         swsizer.Layout()
 973
 974         radiobackground.SetValue(1)
 975         self.checknormal.SetValue(1)
 976         self.radiohorizontal.Enable(False)
 977         self.radiovertical.Enable(False)
 978         self.firstcolour.Enable(False)
 979         self.secondcolour.Enable(False)
 980         self.imagebutton.Enable(False)
 981
 982         return mainsizer.CalcMin().width + wx.SystemSettings.GetMetric(wx.SYS_VSCROLL_X)
 983
 984
 985     def OnRecreateTree(self, event):
 986
 987         splitter = self.tree.GetParent()
 988         newtree = CustomTreeCtrl(splitter, -1, log=self.log)
 989         splitter.ReplaceWindow(self.tree, newtree)
 990         self.tree.Destroy()
 991         self.tree = newtree
 992         # Todo:  The settings in the leftpanel should be reset too
 993
 994
 995     def OnCheckStyle(self, event):
 996
 997         self.tree.ChangeStyle(self.treestyles)
 998         event.Skip()
 999
1000
1001     def OnCheckEvent(self, event):
1002
1003         obj = event.GetEventObject()
1004         self.tree.BindEvents(obj)
1005
1006         event.Skip()
1007
1008
1009     def OnButtonConnection(self, event):
1010
1011         pen = self.tree.GetConnectionPen()
1012         dlg = PenDialog(self, -1, oldpen=pen, pentype=0)
1013
1014         dlg.ShowModal()
1015
1016         event.Skip()
1017
1018
1019     def SetConnectionPen(self, pen):
1020
1021         self.tree.SetConnectionPen(pen)
1022
1023
1024     def OnButtonBorder(self, event):
1025
1026         pen = self.tree.GetBorderPen()
1027         dlg = PenDialog(self, -1, oldpen=pen, pentype=1)
1028
1029         dlg.ShowModal()
1030         event.Skip()
1031
1032
1033     def SetBorderPen(self, pen):
1034
1035         self.tree.SetBorderPen(pen)
1036
1037
1038     def OnButtonTree(self, event):
1039
1040         dlg = TreeButtonsDialog(self, -1, oldicons=self.oldicons)
1041         dlg.ShowModal()
1042
1043         event.Skip()
1044
1045
1046     def OnButtonCheckRadio(self, event):
1047
1048         dlg = CheckDialog(self, -1)
1049         dlg.ShowModal()
1050
1051         event.Skip()
1052
1053
1054     def SetTreeButtons(self, selection):
1055
1056         bitmap_plus = opj("bitmaps/plus" + str(selection+1) + ".ico")
1057         bitmap_minus = opj("bitmaps/minus" + str(selection+1) + ".ico")
1058
1059         bitmap = wx.Bitmap(bitmap_plus, wx.BITMAP_TYPE_ICO)
1060         width = bitmap.GetWidth()
1061
1062         il = wx.ImageList(width, width)
1063
1064         il.Add(wx.Bitmap(bitmap_plus, wx.BITMAP_TYPE_ICO))
1065         il.Add(wx.Bitmap(bitmap_plus, wx.BITMAP_TYPE_ICO))
1066         il.Add(wx.Bitmap(bitmap_minus, wx.BITMAP_TYPE_ICO))
1067         il.Add(wx.Bitmap(bitmap_minus, wx.BITMAP_TYPE_ICO))
1068
1069         self.il = il
1070         self.tree.SetButtonsImageList(il)
1071
1072
1073     def SetCheckRadio(self, selection):
1074
1075         if selection == 0:
1076             self.tree.SetImageListCheck(13, 13)
1077         else:
1078             bitmap_check = opj("bitmaps/aquachecked.ico")
1079             bitmap_uncheck = opj("bitmaps/aquanotchecked.ico")
1080             bitmap_flag = opj("bitmaps/aquaflagged.ico")
1081             bitmap_unflag = opj("bitmaps/aquanotflagged.ico")
1082
1083             il = wx.ImageList(16, 16)
1084
1085             il.Add(wx.Bitmap(bitmap_check, wx.BITMAP_TYPE_ICO))
1086             il.Add(wx.Bitmap(bitmap_uncheck, wx.BITMAP_TYPE_ICO))
1087             il.Add(wx.Bitmap(bitmap_flag, wx.BITMAP_TYPE_ICO))
1088             il.Add(wx.Bitmap(bitmap_unflag, wx.BITMAP_TYPE_ICO))
1089             self.tree.SetImageListCheck(16, 16, il)
1090
1091
1092     def OnBackgroundImage(self, event):
1093
1094         if hasattr(self, "backgroundimage"):
1095             self.tree.SetBackgroundImage(self.backgroundimage)
1096
1097         self.backbutton.Enable(False)
1098         self.imagebutton.Enable(True)
1099
1100         event.Skip()
1101
1102
1103     def OnChooseImage(self, event):
1104
1105         wildcard = "JPEG Files (*.jpg)|*.jpg|"     \
1106                    "Bitmap Files (*.bmp)|*.bmp|" \
1107                    "PNG Files (*.png)|*.png|"    \
1108                    "Icon Files (*.ico)|*.ico|"        \
1109                    "GIF Files (*.gif)|*.gif|"        \
1110                    "All files (*.*)|*.*"
1111
1112         dlg = wx.FileDialog(self, "Choose An Image File", ".", "", wildcard, wx.OPEN)
1113
1114         if dlg.ShowModal() == wx.ID_OK:
1115             path = dlg.GetPath()
1116         else:
1117             dlg.Destroy()
1118             return
1119
1120         dlg.Destroy()
1121         bitmap = wx.Bitmap(path, wx.BITMAP_TYPE_ANY)
1122         self.tree.SetBackgroundImage(bitmap)
1123         self.backgroundimage = bitmap
1124
1125         event.Skip()
1126
1127
1128     def OnBackgroundColour(self, event):
1129
1130         self.imagebutton.Enable(False)
1131         self.backbutton.Enable(True)
1132         self.tree.SetBackgroundImage(None)
1133
1134         event.Skip()
1135
1136
1137     def OnChooseBackground(self, event):
1138
1139         col1 = event.GetValue()
1140         self.tree.SetBackgroundColour(col1)
1141         event.Skip()
1142
1143
1144     def OnCheckNormal(self, event):
1145
1146         self.radiohorizontal.Enable(False)
1147         self.radiovertical.Enable(False)
1148         self.firstcolour.Enable(False)
1149         self.secondcolour.Enable(False)
1150         self.focus.Enable(True)
1151         self.unfocus.Enable(True)
1152         self.checkgradient.SetValue(0)
1153         self.checkvista.SetValue(0)
1154         self.tree.EnableSelectionGradient(False)
1155         self.tree.EnableSelectionVista(False)
1156         event.Skip()
1157
1158
1159     def OnFocusColour(self, event):
1160
1161         col1 = event.GetValue()
1162         self.tree.SetHilightFocusColour(col1)
1163         event.Skip()
1164
1165
1166     def OnNonFocusColour(self, event):
1167
1168         col1 = event.GetValue()
1169         self.tree.SetHilightNonFocusColour(col1)
1170         event.Skip()
1171
1172
1173     def OnCheckGradient(self, event):
1174
1175         self.radiohorizontal.Enable(True)
1176         self.radiovertical.Enable(True)
1177         self.firstcolour.Enable(True)
1178         self.secondcolour.Enable(True)
1179         self.checknormal.SetValue(0)
1180         self.checkvista.SetValue(0)
1181         self.focus.Enable(False)
1182         self.unfocus.Enable(False)
1183         self.tree.SetGradientStyle(self.radiovertical.GetValue())
1184         self.tree.EnableSelectionVista(False)
1185         self.tree.EnableSelectionGradient(True)
1186
1187         event.Skip()
1188
1189
1190     def OnHorizontal(self, event):
1191
1192         self.tree.SetGradientStyle(self.radiovertical.GetValue())
1193         event.Skip()
1194
1195
1196     def OnVertical(self, event):
1197
1198         self.tree.SetGradientStyle(self.radiovertical.GetValue())
1199         event.Skip()
1200
1201
1202     def OnFirstColour(self, event):
1203
1204         col1 = event.GetValue()
1205         self.tree.SetFirstGradientColour(wx.Colour(col1[0], col1[1], col1[2]))
1206         event.Skip()
1207
1208
1209     def OnSecondColour(self, event):
1210
1211         col1 = event.GetValue()
1212         self.tree.SetSecondGradientColour(wx.Colour(col1[0], col1[1], col1[2]))
1213         event.Skip()
1214
1215
1216     def OnVista(self, event):
1217
1218         self.radiohorizontal.Enable(False)
1219         self.radiovertical.Enable(False)
1220         self.firstcolour.Enable(False)
1221         self.secondcolour.Enable(False)
1222         self.checknormal.SetValue(0)
1223         self.checkgradient.SetValue(0)
1224         self.focus.Enable(False)
1225         self.unfocus.Enable(False)
1226         self.tree.EnableSelectionGradient(False)
1227         self.tree.EnableSelectionVista(True)
1228
1229         event.Skip()
1230
1231
1232
1233
1234
1235 #---------------------------------------------------------------------------
1236 # CustomTreeCtrl Demo Implementation
1237 #---------------------------------------------------------------------------
1238 class CustomTreeCtrl(CT.CustomTreeCtrl):
1239
1240     def __init__(self, parent, id=wx.ID_ANY, pos=wx.DefaultPosition,
1241                  size=wx.DefaultSize,
1242                  style=wx.SUNKEN_BORDER,
1243                  ctstyle=CT.TR_HAS_BUTTONS | CT.TR_HAS_VARIABLE_ROW_HEIGHT,
1244                  log=None):
1245
1246         CT.CustomTreeCtrl.__init__(self, parent, id, pos, size, style, ctstyle)
1247
1248         alldata = dir(CT)
1249
1250         treestyles = []
1251         events = []
1252         for data in alldata:
1253             if data.startswith("TR_"):
1254                 treestyles.append(data)
1255             elif data.startswith("EVT_"):
1256                 events.append(data)
1257
1258         self.events = events
1259         self.styles = treestyles
1260         self.item = None
1261
1262         il = wx.ImageList(16, 16)
1263
1264         for items in ArtIDs[1:-1]:
1265             bmp = wx.ArtProvider_GetBitmap(eval(items), wx.ART_TOOLBAR, (16, 16))
1266             il.Add(bmp)
1267
1268         smileidx = il.Add(images.getSmilesBitmap())
1269         numicons = il.GetImageCount()
1270
1271         self.AssignImageList(il)
1272         self.count = 0
1273         self.log = log
1274
1275         # NOTE:  For some reason tree items have to have a data object in
1276         #        order to be sorted.  Since our compare just uses the labels
1277         #        we don't need any real data, so we'll just use None below for
1278         #        the item data.
1279
1280         self.root = self.AddRoot("The Root Item")
1281
1282         if not(self.GetTreeStyle() & CT.TR_HIDE_ROOT):
1283             self.SetPyData(self.root, None)
1284             self.SetItemImage(self.root, 24, CT.TreeItemIcon_Normal)
1285             self.SetItemImage(self.root, 13, CT.TreeItemIcon_Expanded)
1286
1287         textctrl = wx.TextCtrl(self, -1, "I Am A Simple\nMultiline wx.TexCtrl", style=wx.TE_MULTILINE)
1288         self.gauge = wx.Gauge(self, -1, 50, style=wx.GA_HORIZONTAL|wx.GA_SMOOTH)
1289         self.gauge.SetValue(0)
1290         combobox = wx.ComboBox(self, -1, choices=["That", "Was", "A", "Nice", "Holyday!"], style=wx.CB_READONLY|wx.CB_DROPDOWN)
1291
1292         textctrl.Bind(wx.EVT_CHAR, self.OnTextCtrl)
1293         combobox.Bind(wx.EVT_COMBOBOX, self.OnComboBox)
1294
1295         for x in range(15):
1296             if x == 1:
1297                 child = self.AppendItem(self.root, "Item %d" % x + "\nHello World\nHappy wxPython-ing!")
1298                 self.SetItemBold(child, True)
1299             else:
1300                 child = self.AppendItem(self.root, "Item %d" % x)
1301             self.SetPyData(child, None)
1302             self.SetItemImage(child, 24, CT.TreeItemIcon_Normal)
1303             self.SetItemImage(child, 13, CT.TreeItemIcon_Expanded)
1304
1305             for y in range(5):
1306                 if y == 0 and x == 1:
1307                     last = self.AppendItem(child, "item %d-%s" % (x, chr(ord("a")+y)), ct_type=2, wnd=self.gauge)
1308                 elif y == 1 and x == 2:
1309                     last = self.AppendItem(child, "Item %d-%s" % (x, chr(ord("a")+y)), ct_type=1, wnd=textctrl)
1310                 elif 2 < y < 4:
1311                     last = self.AppendItem(child, "item %d-%s" % (x, chr(ord("a")+y)))
1312                 elif y == 4 and x == 1:
1313                     last = self.AppendItem(child, "item %d-%s" % (x, chr(ord("a")+y)), wnd=combobox)
1314                 else:
1315                     last = self.AppendItem(child, "item %d-%s" % (x, chr(ord("a")+y)), ct_type=2)
1316
1317                 self.SetPyData(last, None)
1318                 self.SetItemImage(last, 24, CT.TreeItemIcon_Normal)
1319                 self.SetItemImage(last, 13, CT.TreeItemIcon_Expanded)
1320
1321                 for z in range(5):
1322                     if z > 2:
1323                         item = self.AppendItem(last,  "item %d-%s-%d" % (x, chr(ord("a")+y), z), ct_type=1)
1324                     elif 0 < z <= 2:
1325                         item = self.AppendItem(last,  "item %d-%s-%d" % (x, chr(ord("a")+y), z), ct_type=2)
1326                     elif z == 0:
1327                         item = self.AppendItem(last,  "item %d-%s-%d" % (x, chr(ord("a")+y), z))
1328                         self.SetItemHyperText(item, True)
1329                     self.SetPyData(item, None)
1330                     self.SetItemImage(item, 28, CT.TreeItemIcon_Normal)
1331                     self.SetItemImage(item, numicons-1, CT.TreeItemIcon_Selected)
1332
1333         self.Bind(wx.EVT_LEFT_DCLICK, self.OnLeftDClick)
1334         self.Bind(wx.EVT_IDLE, self.OnIdle)
1335
1336         self.eventdict = {'EVT_TREE_BEGIN_DRAG': self.OnBeginDrag, 'EVT_TREE_BEGIN_LABEL_EDIT': self.OnBeginEdit,
1337                           'EVT_TREE_BEGIN_RDRAG': self.OnBeginRDrag, 'EVT_TREE_DELETE_ITEM': self.OnDeleteItem,
1338                           'EVT_TREE_END_DRAG': self.OnEndDrag, 'EVT_TREE_END_LABEL_EDIT': self.OnEndEdit,
1339                           'EVT_TREE_ITEM_ACTIVATED': self.OnActivate, 'EVT_TREE_ITEM_CHECKED': self.OnItemCheck,
1340                           'EVT_TREE_ITEM_CHECKING': self.OnItemChecking, 'EVT_TREE_ITEM_COLLAPSED': self.OnItemCollapsed,
1341                           'EVT_TREE_ITEM_COLLAPSING': self.OnItemCollapsing, 'EVT_TREE_ITEM_EXPANDED': self.OnItemExpanded,
1342                           'EVT_TREE_ITEM_EXPANDING': self.OnItemExpanding, 'EVT_TREE_ITEM_GETTOOLTIP': self.OnToolTip,
1343                           'EVT_TREE_ITEM_MENU': self.OnItemMenu, 'EVT_TREE_ITEM_RIGHT_CLICK': self.OnRightDown,
1344                           'EVT_TREE_KEY_DOWN': self.OnKey, 'EVT_TREE_SEL_CHANGED': self.OnSelChanged,
1345                           'EVT_TREE_SEL_CHANGING': self.OnSelChanging, "EVT_TREE_ITEM_HYPERLINK": self.OnHyperLink}
1346
1347         mainframe = wx.GetTopLevelParent(self)
1348
1349         if not hasattr(mainframe, "leftpanel"):
1350             self.Bind(CT.EVT_TREE_ITEM_EXPANDED, self.OnItemExpanded)
1351             self.Bind(CT.EVT_TREE_ITEM_COLLAPSED, self.OnItemCollapsed)
1352             self.Bind(CT.EVT_TREE_SEL_CHANGED, self.OnSelChanged)
1353             self.Bind(CT.EVT_TREE_SEL_CHANGING, self.OnSelChanging)
1354             self.Bind(wx.EVT_RIGHT_DOWN, self.OnRightDown)
1355             self.Bind(wx.EVT_RIGHT_UP, self.OnRightUp)
1356         else:
1357             for combos in mainframe.treeevents:
1358                 self.BindEvents(combos)
1359
1360         if hasattr(mainframe, "leftpanel"):
1361             self.ChangeStyle(mainframe.treestyles)
1362
1363         if not(self.GetTreeStyle() & CT.TR_HIDE_ROOT):
1364             self.SelectItem(self.root)
1365             self.Expand(self.root)
1366
1367
1368     def BindEvents(self, choice, recreate=False):
1369
1370         value = choice.GetValue()
1371         text = choice.GetLabel()
1372
1373         evt = "CT." + text
1374         binder = self.eventdict[text]
1375
1376         if value == 1:
1377             if evt == "CT.EVT_TREE_BEGIN_RDRAG":
1378                 self.Bind(wx.EVT_RIGHT_DOWN, None)
1379                 self.Bind(wx.EVT_RIGHT_UP, None)
1380             self.Bind(eval(evt), binder)
1381         else:
1382             self.Bind(eval(evt), None)
1383             if evt == "CT.EVT_TREE_BEGIN_RDRAG":
1384                 self.Bind(wx.EVT_RIGHT_DOWN, self.OnRightDown)
1385                 self.Bind(wx.EVT_RIGHT_UP, self.OnRightUp)
1386
1387
1388     def ChangeStyle(self, combos):
1389
1390         style = 0
1391         for combo in combos:
1392             if combo.GetValue() == 1:
1393                 style = style | eval("CT." + combo.GetLabel())
1394
1395         if self.GetTreeStyle() != style:
1396             self.SetTreeStyle(style)
1397
1398
1399     def OnCompareItems(self, item1, item2):
1400
1401         t1 = self.GetItemText(item1)
1402         t2 = self.GetItemText(item2)
1403
1404         self.log.write('compare: ' + t1 + ' <> ' + t2 + "\n")
1405
1406         if t1 < t2:
1407             return -1
1408         if t1 == t2:
1409             return 0
1410
1411         return 1
1412
1413
1414     def OnIdle(self, event):
1415
1416         if self.gauge:
1417             try:
1418                 if self.gauge.IsEnabled() and self.gauge.IsShown():
1419                     self.count = self.count + 1
1420
1421                     if self.count >= 50:
1422                         self.count = 0
1423
1424                     self.gauge.SetValue(self.count)
1425
1426             except:
1427                 self.gauge = None
1428
1429         event.Skip()
1430
1431
1432     def OnRightDown(self, event):
1433
1434         pt = event.GetPosition()
1435         item, flags = self.HitTest(pt)
1436
1437         if item:
1438             self.item = item
1439             self.log.write("OnRightClick: %s, %s, %s" % (self.GetItemText(item), type(item), item.__class__) + "\n")
1440             self.SelectItem(item)
1441
1442
1443     def OnRightUp(self, event):
1444
1445         item = self.item
1446
1447         if not item:
1448             event.Skip()
1449             return
1450
1451         if not self.IsEnabled(item):
1452             event.Skip()
1453             return
1454
1455         # Item Text Appearance
1456         ishtml = self.IsItemHyperText(item)
1457         back = self.GetItemBackgroundColour(item)
1458         fore = self.GetItemTextColour(item)
1459         isbold = self.IsBold(item)
1460         font = self.GetItemFont(item)
1461
1462         # Icons On Item
1463         normal = self.GetItemImage(item, CT.TreeItemIcon_Normal)
1464         selected = self.GetItemImage(item, CT.TreeItemIcon_Selected)
1465         expanded = self.GetItemImage(item, CT.TreeItemIcon_Expanded)
1466         selexp = self.GetItemImage(item, CT.TreeItemIcon_SelectedExpanded)
1467
1468         # Enabling/Disabling Windows Associated To An Item
1469         haswin = self.GetItemWindow(item)
1470
1471         # Enabling/Disabling Items
1472         enabled = self.IsEnabled(item)
1473
1474         # Generic Item's Info
1475         children = self.GetChildrenCount(item)
1476         itemtype = self.GetItemType(item)
1477         text = self.GetItemText(item)
1478         pydata = self.GetPyData(item)
1479
1480         self.current = item
1481         self.itemdict = {"ishtml": ishtml, "back": back, "fore": fore, "isbold": isbold,
1482                          "font": font, "normal": normal, "selected": selected, "expanded": expanded,
1483                          "selexp": selexp, "haswin": haswin, "children": children,
1484                          "itemtype": itemtype, "text": text, "pydata": pydata, "enabled": enabled}
1485
1486         menu = wx.Menu()
1487
1488         item1 = menu.Append(wx.ID_ANY, "Change Item Background Colour")
1489         item2 = menu.Append(wx.ID_ANY, "Modify Item Text Colour")
1490         menu.AppendSeparator()
1491         if isbold:
1492             strs = "Make Item Text Not Bold"
1493         else:
1494             strs = "Make Item Text Bold"
1495         item3 = menu.Append(wx.ID_ANY, strs)
1496         item4 = menu.Append(wx.ID_ANY, "Change Item Font")
1497         menu.AppendSeparator()
1498         if ishtml:
1499             strs = "Set Item As Non-Hyperlink"
1500         else:
1501             strs = "Set Item As Hyperlink"
1502         item5 = menu.Append(wx.ID_ANY, strs)
1503         menu.AppendSeparator()
1504         if haswin:
1505             enabled = self.GetItemWindowEnabled(item)
1506             if enabled:
1507                 strs = "Disable Associated Widget"
1508             else:
1509                 strs = "Enable Associated Widget"
1510         else:
1511             strs = "Enable Associated Widget"
1512         item6 = menu.Append(wx.ID_ANY, strs)
1513
1514         if not haswin:
1515             item6.Enable(False)
1516
1517         item7 = menu.Append(wx.ID_ANY, "Disable Item")
1518
1519         menu.AppendSeparator()
1520         item8 = menu.Append(wx.ID_ANY, "Change Item Icons")
1521         menu.AppendSeparator()
1522         item9 = menu.Append(wx.ID_ANY, "Get Other Information For This Item")
1523         menu.AppendSeparator()
1524
1525         item10 = menu.Append(wx.ID_ANY, "Delete Item")
1526         if item == self.GetRootItem():
1527             item10.Enable(False)
1528         item11 = menu.Append(wx.ID_ANY, "Prepend An Item")
1529         item12 = menu.Append(wx.ID_ANY, "Append An Item")
1530
1531         self.Bind(wx.EVT_MENU, self.OnItemBackground, item1)
1532         self.Bind(wx.EVT_MENU, self.OnItemForeground, item2)
1533         self.Bind(wx.EVT_MENU, self.OnItemBold, item3)
1534         self.Bind(wx.EVT_MENU, self.OnItemFont, item4)
1535         self.Bind(wx.EVT_MENU, self.OnItemHyperText, item5)
1536         self.Bind(wx.EVT_MENU, self.OnEnableWindow, item6)
1537         self.Bind(wx.EVT_MENU, self.OnDisableItem, item7)
1538         self.Bind(wx.EVT_MENU, self.OnItemIcons, item8)
1539         self.Bind(wx.EVT_MENU, self.OnItemInfo, item9)
1540         self.Bind(wx.EVT_MENU, self.OnItemDelete, item10)
1541         self.Bind(wx.EVT_MENU, self.OnItemPrepend, item11)
1542         self.Bind(wx.EVT_MENU, self.OnItemAppend, item12)
1543
1544         self.PopupMenu(menu)
1545         menu.Destroy()
1546         event.Skip()
1547
1548
1549     def OnItemBackground(self, event):
1550
1551         colourdata = wx.ColourData()
1552         colourdata.SetColour(self.itemdict["back"])
1553         dlg = wx.ColourDialog(self, colourdata)
1554
1555         dlg.GetColourData().SetChooseFull(True)
1556
1557         if dlg.ShowModal() == wx.ID_OK:
1558             data = dlg.GetColourData()
1559             col1 = data.GetColour().Get()
1560             self.SetItemBackgroundColour(self.current, col1)
1561         dlg.Destroy()
1562         event.Skip()
1563
1564
1565     def OnItemForeground(self, event):
1566
1567         colourdata = wx.ColourData()
1568         colourdata.SetColour(self.itemdict["fore"])
1569         dlg = wx.ColourDialog(self, colourdata)
1570
1571         dlg.GetColourData().SetChooseFull(True)
1572
1573         if dlg.ShowModal() == wx.ID_OK:
1574             data = dlg.GetColourData()
1575             col1 = data.GetColour().Get()
1576             self.SetItemTextColour(self.current, col1)
1577         dlg.Destroy()
1578         event.Skip()
1579
1580
1581     def OnItemBold(self, event):
1582
1583         self.SetItemBold(self.current, not self.itemdict["isbold"])
1584         event.Skip()
1585
1586
1587     def OnItemFont(self, event):
1588
1589         data = wx.FontData()
1590         font = self.itemdict["font"]
1591
1592         if font is None:
1593             font = wx.SystemSettings_GetFont(wx.SYS_DEFAULT_GUI_FONT)
1594
1595         data.SetInitialFont(font)
1596
1597         dlg = wx.FontDialog(self, data)
1598
1599         if dlg.ShowModal() == wx.ID_OK:
1600             data = dlg.GetFontData()
1601             font = data.GetChosenFont()
1602             self.SetItemFont(self.current, font)
1603
1604         dlg.Destroy()
1605         event.Skip()
1606
1607
1608     def OnItemHyperText(self, event):
1609
1610         self.SetItemHyperText(self.current, not self.itemdict["ishtml"])
1611         event.Skip()
1612
1613
1614     def OnEnableWindow(self, event):
1615
1616         enable = self.GetItemWindowEnabled(self.current)
1617         self.SetItemWindowEnabled(self.current, not enable)
1618
1619         event.Skip()
1620
1621
1622     def OnDisableItem(self, event):
1623
1624         self.EnableItem(self.current, False)
1625         event.Skip()
1626
1627
1628     def OnItemIcons(self, event):
1629
1630         bitmaps = [self.itemdict["normal"], self.itemdict["selected"],
1631                    self.itemdict["expanded"], self.itemdict["selexp"]]
1632
1633         wx.BeginBusyCursor()
1634         dlg = TreeIcons(self, -1, bitmaps=bitmaps)
1635         wx.EndBusyCursor()
1636         dlg.ShowModal()
1637         event.Skip()
1638
1639
1640     def SetNewIcons(self, bitmaps):
1641
1642         self.SetItemImage(self.current, bitmaps[0], CT.TreeItemIcon_Normal)
1643         self.SetItemImage(self.current, bitmaps[1], CT.TreeItemIcon_Selected)
1644         self.SetItemImage(self.current, bitmaps[2], CT.TreeItemIcon_Expanded)
1645         self.SetItemImage(self.current, bitmaps[3], CT.TreeItemIcon_SelectedExpanded)
1646
1647
1648     def OnItemInfo(self, event):
1649
1650         itemtext = self.itemdict["text"]
1651         numchildren = str(self.itemdict["children"])
1652         itemtype = self.itemdict["itemtype"]
1653         pydata = repr(type(self.itemdict["pydata"]))
1654
1655         if itemtype == 0:
1656             itemtype = "Normal"
1657         elif itemtype == 1:
1658             itemtype = "CheckBox"
1659         else:
1660             itemtype = "RadioButton"
1661
1662         strs = "Information On Selected Item:\n\n" + "Text: " + itemtext + "\n" \
1663                "Number Of Children: " + numchildren + "\n" \
1664                "Item Type: " + itemtype + "\n" \
1665                "Item Data Type: " + pydata + "\n"
1666
1667         dlg = wx.MessageDialog(self, strs, "CustomTreeCtrlDemo Info", wx.OK | wx.ICON_INFORMATION)
1668         dlg.ShowModal()
1669         dlg.Destroy()
1670
1671         event.Skip()
1672
1673
1674     def OnItemDelete(self, event):
1675
1676         strs = "Are You Sure You Want To Delete Item " + self.GetItemText(self.current) + "?"
1677         dlg = wx.MessageDialog(None, strs, 'Deleting Item', wx.YES_NO | wx.NO_DEFAULT | wx.CANCEL | wx.ICON_QUESTION)
1678
1679         if dlg.ShowModal() in [wx.ID_NO, wx.ID_CANCEL]:
1680             dlg.Destroy()
1681             return
1682
1683         dlg.Destroy()
1684
1685         self.DeleteChildren(self.current)
1686         self.Delete(self.current)
1687         self.current = None
1688
1689         event.Skip()
1690
1691
1692     def OnItemPrepend(self, event):
1693
1694         dlg = wx.TextEntryDialog(self, "Please Enter The New Item Name", 'Item Naming', 'Python')
1695
1696         if dlg.ShowModal() == wx.ID_OK:
1697             newname = dlg.GetValue()
1698             newitem = self.PrependItem(self.current, newname)
1699             self.EnsureVisible(newitem)
1700
1701         dlg.Destroy()
1702         event.Skip()
1703
1704
1705     def OnItemAppend(self, event):
1706
1707         dlg = wx.TextEntryDialog(self, "Please Enter The New Item Name", 'Item Naming', 'Python')
1708
1709         if dlg.ShowModal() == wx.ID_OK:
1710             newname = dlg.GetValue()
1711             newitem = self.AppendItem(self.current, newname)
1712             self.EnsureVisible(newitem)
1713
1714         dlg.Destroy()
1715         event.Skip()
1716
1717
1718     def OnBeginEdit(self, event):
1719
1720         self.log.write("OnBeginEdit" + "\n")
1721         # show how to prevent edit...
1722         item = event.GetItem()
1723         if item and self.GetItemText(item) == "The Root Item":
1724             wx.Bell()
1725             self.log.write("You can't edit this one..." + "\n")
1726
1727             # Lets just see what's visible of its children
1728             cookie = 0
1729             root = event.GetItem()
1730             (child, cookie) = self.GetFirstChild(root)
1731
1732             while child:
1733                 self.log.write("Child [%s] visible = %d" % (self.GetItemText(child), self.IsVisible(child)) + "\n")
1734                 (child, cookie) = self.GetNextChild(root, cookie)
1735
1736             event.Veto()
1737
1738
1739     def OnEndEdit(self, event):
1740
1741         self.log.write("OnEndEdit: %s %s" %(event.IsEditCancelled(), event.GetLabel()))
1742         # show how to reject edit, we'll not allow any digits
1743         for x in event.GetLabel():
1744             if x in string.digits:
1745                 self.log.write(", You can't enter digits..." + "\n")
1746                 event.Veto()
1747                 return
1748
1749         self.log.write("\n")
1750
1751
1752     def OnLeftDClick(self, event):
1753
1754         pt = event.GetPosition()
1755         item, flags = self.HitTest(pt)
1756         if item and (flags & CT.TREE_HITTEST_ONITEMLABEL):
1757             if self.GetTreeStyle() & CT.TR_EDIT_LABELS:
1758                 self.log.write("OnLeftDClick: %s (manually starting label edit)"% self.GetItemText(item) + "\n")
1759                 self.EditLabel(item)
1760             else:
1761                 self.log.write("OnLeftDClick: Cannot Start Manual Editing, Missing Style TR_EDIT_LABELS\n")
1762
1763         event.Skip()
1764
1765
1766     def OnItemExpanded(self, event):
1767
1768         item = event.GetItem()
1769         if item:
1770             self.log.write("OnItemExpanded: %s" % self.GetItemText(item) + "\n")
1771
1772
1773     def OnItemExpanding(self, event):
1774
1775         item = event.GetItem()
1776         if item:
1777             self.log.write("OnItemExpanding: %s" % self.GetItemText(item) + "\n")
1778
1779         event.Skip()
1780
1781
1782     def OnItemCollapsed(self, event):
1783
1784         item = event.GetItem()
1785         if item:
1786             self.log.write("OnItemCollapsed: %s" % self.GetItemText(item) + "\n")
1787
1788
1789     def OnItemCollapsing(self, event):
1790
1791         item = event.GetItem()
1792         if item:
1793             self.log.write("OnItemCollapsing: %s" % self.GetItemText(item) + "\n")
1794
1795         event.Skip()
1796
1797
1798     def OnSelChanged(self, event):
1799
1800         self.item = event.GetItem()
1801         if self.item:
1802             self.log.write("OnSelChanged: %s" % self.GetItemText(self.item))
1803             if wx.Platform == '__WXMSW__':
1804                 self.log.write(", BoundingRect: %s" % self.GetBoundingRect(self.item, True) + "\n")
1805             else:
1806                 self.log.write("\n")
1807
1808         event.Skip()
1809
1810
1811     def OnSelChanging(self, event):
1812
1813         item = event.GetItem()
1814         olditem = event.GetOldItem()
1815
1816         if item:
1817             if not olditem:
1818                 olditemtext = "None"
1819             else:
1820                 olditemtext = self.GetItemText(olditem)
1821             self.log.write("OnSelChanging: From %s" % olditemtext + " To %s" % self.GetItemText(item) + "\n")
1822
1823         event.Skip()
1824
1825
1826     def OnBeginDrag(self, event):
1827
1828         self.item = event.GetItem()
1829         if self.item:
1830             self.log.write("Beginning Drag..." + "\n")
1831
1832             event.Allow()
1833             event.Skip()
1834
1835
1836     def OnBeginRDrag(self, event):
1837
1838         self.item = event.GetItem()
1839         if self.item:
1840             self.log.write("Beginning Right Drag..." + "\n")
1841
1842             event.Allow()
1843             event.Skip()
1844
1845
1846     def OnEndDrag(self, event):
1847
1848         self.item = event.GetItem()
1849         if self.item:
1850             self.log.write("Ending Drag!" + "\n")
1851
1852         event.Skip()
1853
1854
1855     def OnDeleteItem(self, event):
1856
1857         item = event.GetItem()
1858
1859         if not item:
1860             return
1861
1862         self.log.write("Deleting Item: %s" % self.GetItemText(item) + "\n")
1863         event.Skip()
1864
1865
1866     def OnItemCheck(self, event):
1867
1868         item = event.GetItem()
1869         self.log.write("Item " + self.GetItemText(item) + " Has Been Checked!\n")
1870         event.Skip()
1871
1872
1873     def OnItemChecking(self, event):
1874
1875         item = event.GetItem()
1876         self.log.write("Item " + self.GetItemText(item) + " Is Being Checked...\n")
1877         event.Skip()
1878
1879
1880     def OnToolTip(self, event):
1881
1882         item = event.GetItem()
1883         if item:
1884             event.SetToolTip(wx.ToolTip(self.GetItemText(item)))
1885
1886
1887     def OnItemMenu(self, event):
1888
1889         item = event.GetItem()
1890         if item:
1891             self.log.write("OnItemMenu: %s" % self.GetItemText(item) + "\n")
1892
1893         event.Skip()
1894
1895
1896     def OnKey(self, event):
1897
1898         keycode = event.GetKeyCode()
1899         keyname = keyMap.get(keycode, None)
1900
1901         if keycode == wx.WXK_BACK:
1902             self.log.write("OnKeyDown: HAHAHAHA! I Vetoed Your Backspace! HAHAHAHA\n")
1903             return
1904
1905         if keyname is None:
1906             if "unicode" in wx.PlatformInfo:
1907                 keycode = event.GetUnicodeKey()
1908                 if keycode <= 127:
1909                     keycode = event.GetKeyCode()
1910                 keyname = "\"" + unichr(event.GetUnicodeKey()) + "\""
1911                 if keycode < 27:
1912                     keyname = "Ctrl-%s" % chr(ord('A') + keycode-1)
1913
1914             elif keycode < 256:
1915                 if keycode == 0:
1916                     keyname = "NUL"
1917                 elif keycode < 27:
1918                     keyname = "Ctrl-%s" % chr(ord('A') + keycode-1)
1919                 else:
1920                     keyname = "\"%s\"" % chr(keycode)
1921             else:
1922                 keyname = "unknown (%s)" % keycode
1923
1924         self.log.write("OnKeyDown: You Pressed '" + keyname + "'\n")
1925
1926         event.Skip()
1927
1928
1929     def OnActivate(self, event):
1930
1931         if self.item:
1932             self.log.write("OnActivate: %s" % self.GetItemText(self.item) + "\n")
1933
1934         event.Skip()
1935
1936
1937     def OnHyperLink(self, event):
1938
1939         item = event.GetItem()
1940         if item:
1941             self.log.write("OnHyperLink: %s" % self.GetItemText(self.item) + "\n")
1942
1943
1944     def OnTextCtrl(self, event):
1945
1946         char = chr(event.GetKeyCode())
1947         self.log.write("EDITING THE TEXTCTRL: You Wrote '" + char + \
1948                        "' (KeyCode = " + str(event.GetKeyCode()) + ")\n")
1949         event.Skip()
1950
1951
1952     def OnComboBox(self, event):
1953
1954         selection = event.GetEventObject().GetValue()
1955         self.log.write("CHOICE FROM COMBOBOX: You Chose '" + selection + "'\n")
1956         event.Skip()
1957
1958
1959
1960 #----------------------------------------------------------------------
1961
1962 def runTest(frame, nb, log):
1963     win = CustomTreeCtrlDemo(nb, log)
1964     return win
1965
1966 #----------------------------------------------------------------------
1967
1968
1969
1970 overview = CT.__doc__
1971
1972
1973 if __name__ == '__main__':
1974     import sys,os
1975     import run
1976     run.main(['', os.path.basename(sys.argv[0])] + sys.argv[1:])
1977