]> git.saurik.com Git - wxWidgets.git/blobdiff - wxPython/wx/tools/XRCed/tree.py
test needed for wxGTK
[wxWidgets.git] / wxPython / wx / tools / XRCed / tree.py
index d287a1d33adf245ec2b2064a39f01adca4015428..1e77eac3a4554da23f5ecea7c69332ee609c4a68 100644 (file)
@@ -11,12 +11,6 @@ import traceback
 # Constant to define standart window name
 STD_NAME = '_XRCED_T_W'
 
-COLOUR_COMMENT  = 'Blue'
-COLOUR_REF      = 'DarkGreen'
-COLOUR_HIDDEN   = 'Grey'
-COLOUR_HL       = 'Red'
-COLOUR_DT       = 'Green'
-
 # Icons
 import images
 
@@ -420,22 +414,30 @@ def SetMenu(m, list, shift=False):
 class HighLightBox:
     colour = None
     def __init__(self, pos, size):
-        if not self.colour: colour = self.colour = wx.NamedColour(COLOUR_HL)
-        else: colour = self.colour
+        colour = g.tree.COLOUR_HL
         if size.width == -1: size.width = 0
         if size.height == -1: size.height = 0
         w = g.testWin.panel
         l1 = wx.Window(w, -1, pos, wx.Size(size.width, 2))
-        l1.SetBackgroundColour(self.colour)
+        l1.SetBackgroundColour(colour)
         l2 = wx.Window(w, -1, pos, wx.Size(2, size.height))
-        l2.SetBackgroundColour(self.colour)
+        l2.SetBackgroundColour(colour)
         l3 = wx.Window(w, -1, wx.Point(pos.x + size.width - 2, pos.y), wx.Size(2, size.height))
-        l3.SetBackgroundColour(self.colour)
+        l3.SetBackgroundColour(colour)
         l4 = wx.Window(w, -1, wx.Point(pos.x, pos.y + size.height - 2), wx.Size(size.width, 2))
-        l4.SetBackgroundColour(self.colour)
+        l4.SetBackgroundColour(colour)
         self.lines = [l1, l2, l3, l4]
-        self.size = size
+        if wx.Platform == '__WXMSW__':
+            for l in self.lines:
+                l.Bind(wx.EVT_PAINT, self.OnPaint)
         g.testWin.highLight = self
+        self.size = size
+    # Repainting is not always done for these windows on Windows
+    def OnPaint(self, evt):
+        w = evt.GetEventObject()
+        dc = wx.PaintDC(w)
+        w.ClearBackground()
+        dc.Destroy()
     # Move highlight to a new position
     def Replace(self, pos, size):
         if size.width == -1: size.width = 0
@@ -445,7 +447,6 @@ class HighLightBox:
         self.lines[2].SetDimensions(pos.x + size.width - 2, pos.y, 2, size.height)
         self.lines[3].SetDimensions(pos.x, pos.y + size.height - 2, size.width, 2)
         self.size = size
-    # Remove it
     def Remove(self):
         map(wx.Window.Destroy, self.lines)
         g.testWin.highLight = None
@@ -456,8 +457,7 @@ class HighLightBox:
 class HighLightDTBox(HighLightBox):
     colour = None
     def __init__(self, pos, size):
-        if not self.colour: colour = self.colour = wx.NamedColour(COLOUR_DT)
-        else: colour = self.colour
+        colour = g.tree.COLOUR_DT
         if size.width == -1: size.width = 0
         if size.height == -1: size.height = 0
         w = g.testWin.panel
@@ -470,28 +470,36 @@ class HighLightDTBox(HighLightBox):
         l4 = wx.Window(w, -1, wx.Point(pos.x, pos.y + size.height - 2), wx.Size(size.width, 2))
         l4.SetBackgroundColour(colour)
         self.lines = [l1, l2, l3, l4]
-        self.size = size
         self.item = None
+        self.size = size
     # Remove it
     def Remove(self):
         map(wx.Window.Destroy, self.lines)
         g.testWin.highLightDT = None
 
 def updateHL(hl, hlClass, pos, size):
+    # Need to recreate window if size did not change to force update
     if hl and hl.size == size:
         hl.Remove()
         hl = None
     if hl:
         hl.Replace(pos, size)
-        hl.Refresh()
     else:
         hl = hlClass(pos, size)
+    hl.Refresh()
     return hl
 
 ################################################################################
 
 class XML_Tree(wx.TreeCtrl):
     def __init__(self, parent, id):
+        # Item colour
+        self.COLOUR_COMMENT  = wx.Colour(0, 0, 255)
+        self.COLOUR_REF      = wx.Colour(0, 0, 128)
+        self.COLOUR_HIDDEN   = wx.Colour(128, 128, 128)
+        self.COLOUR_HL       = wx.Colour(255, 0, 0)
+        self.COLOUR_DT       = wx.Colour(0, 64, 0)
+
         wx.TreeCtrl.__init__(self, parent, id,
                              style = wx.TR_HAS_BUTTONS | wx.TR_MULTIPLE | wx.TR_EDIT_LABELS)
         self.SetBackgroundColour(wx.Colour(224, 248, 224))
@@ -501,7 +509,7 @@ class XML_Tree(wx.TreeCtrl):
         # Register events
         wx.EVT_TREE_SEL_CHANGED(self, self.GetId(), self.OnSelChanged)
         # One works on Linux, another on Windows
-        if wx.Platform == '__WXGTK__':
+        if wx.Platform == '__WXGTK__': # !!! MAC too?
             wx.EVT_TREE_ITEM_ACTIVATED(self, self.GetId(), self.OnItemActivated)
         else:
             wx.EVT_LEFT_DCLICK(self, self.OnDClick)
@@ -569,7 +577,6 @@ class XML_Tree(wx.TreeCtrl):
 
     # Clear tree
     def Clear(self):
-        self.selection = None
         self.UnselectAll()
         self.DeleteAllItems()
         # Add minimal structure
@@ -582,6 +589,7 @@ class XML_Tree(wx.TreeCtrl):
         self.rootObj = xxxMainNode(self.dom)
         self.root = self.AddRoot('XML tree', self.rootImage,
                                  data=wx.TreeItemData(self.rootObj))
+        self.itemColour = self.GetItemTextColour(self.root)
         self.SetItemHasChildren(self.root)
         self.testElem = self.dom.createElement('dummy')
         self.mainNode.appendChild(self.testElem)
@@ -589,7 +597,6 @@ class XML_Tree(wx.TreeCtrl):
 
     # Clear old data and set new
     def SetData(self, dom):
-        self.selection = None
         self.UnselectAll()
         self.DeleteAllItems()
         # Add minimal structure
@@ -601,7 +608,6 @@ class XML_Tree(wx.TreeCtrl):
         self.rootObj = xxxMainNode(self.dom)
         self.root = self.AddRoot('XML tree', self.rootImage,
                                  data=wx.TreeItemData(self.rootObj))
-        self.itemColour = self.GetItemTextColour(self.root)
         self.SetItemHasChildren(self.root)
         nodes = self.mainNode.childNodes[:]
         for node in nodes:
@@ -634,12 +640,12 @@ class XML_Tree(wx.TreeCtrl):
                                data=wx.TreeItemData(xxx))
         # Different color for comments and references
         if xxx.className == 'comment':
-            self.SetItemTextColour(item, COLOUR_COMMENT)
+            self.SetItemTextColour(item, self.COLOUR_COMMENT)
             self.SetItemFont(item, self.fontComment)
         elif treeObj.ref:
-            self.SetItemTextColour(item, COLOUR_REF)
+            self.SetItemTextColour(item, self.COLOUR_REF)
         elif treeObj.hasStyle and treeObj.params.get('hidden', False):
-            self.SetItemTextColour(item, COLOUR_HIDDEN)
+            self.SetItemTextColour(item, self.COLOUR_HIDDEN)
         # Try to find children objects
         if treeObj.hasChildren:
             nodes = treeObj.node.childNodes[:]
@@ -670,12 +676,12 @@ class XML_Tree(wx.TreeCtrl):
         treeObj = xxx.treeObject()
         # Different color for references and comments
         if xxx.className == 'comment':
-            self.SetItemTextColour(newItem, COLOUR_COMMENT)
+            self.SetItemTextColour(newItem, self.COLOUR_COMMENT)
             self.SetItemFont(newItem, self.fontComment)
         elif treeObj.ref:
-            self.SetItemTextColour(newItem, COLOUR_REF)
+            self.SetItemTextColour(newItem, self.COLOUR_REF)
         elif treeObj.hasStyle and treeObj.params.get('hidden', False):
-            self.SetItemTextColour(newItem, COLOUR_HIDDEN)
+            self.SetItemTextColour(newItem, self.COLOUR_HIDDEN)
         # Add children items
         if xxx.hasChildren:
             treeObj = xxx.treeObject()
@@ -691,8 +697,6 @@ class XML_Tree(wx.TreeCtrl):
         parent = node.parentNode
         parent.removeChild(node)
         self.Delete(leaf)
-        # Reset selection object
-        self.selection = None
         return node
     
     # Find position relative to the top-level window
@@ -768,10 +772,27 @@ class XML_Tree(wx.TreeCtrl):
     def OnSelChanged(self, evt):
         if self.selectionChanging: return
         self.selectionChanging = True
-        self.UnselectAll()
-        self.SelectItem(evt.GetItem())
+        wx.TreeCtrl.UnselectAll(self)
+        self.ChangeSelection(evt.GetItem())
+        wx.TreeCtrl.SelectItem(self, evt.GetItem())
         self.selectionChanging = False
         g.frame.SetStatusText('')
+        evt.Skip()
+
+    # Override to use like single-selection tree
+    def GetSelection(self):
+        return self.selection
+
+    def SelectItem(self, item):
+        self.UnselectAll()
+        self.ChangeSelection(item)
+        wx.TreeCtrl.SelectItem(self, item)
+
+    def UnselectAll(self):
+        self.selection = None
+        g.tools.UpdateUI()
+        wx.TreeCtrl.UnselectAll(self)
+        wx.Yield()
 
     def ChangeSelection(self, item):
         # Apply changes
@@ -791,11 +812,12 @@ class XML_Tree(wx.TreeCtrl):
                 status = 'Changes were applied'
         if status: g.frame.SetStatusText(status)
         # Generate view
-        self.selection = item
-        if not self.selection.IsOk():
+        if not item:
             self.selection = None
             return
-        xxx = self.GetPyData(self.selection)
+        else:
+            self.selection = item
+        xxx = self.GetPyData(item)
         # Update panel
         g.panel.SetData(xxx)
         # Update tools
@@ -838,6 +860,7 @@ class XML_Tree(wx.TreeCtrl):
         # so we must remove the old HL window
         g.testWin.highLight = updateHL(g.testWin.highLight, HighLightBox, pos, size)
         g.testWin.highLight.item = item
+        g.testWin.highLight.obj = obj
 
     def ShowTestWindow(self, item):
         xxx = self.GetPyData(item)
@@ -975,9 +998,10 @@ class XML_Tree(wx.TreeCtrl):
                 # Create new frame
                 if not testWin:
                     testWin = g.testWin = wx.Frame(g.frame, -1, 'Panel: ' + name,
-                                                  pos=pos, name=STD_NAME)
+                                                   pos=pos, name=STD_NAME)
                 testWin.panel = res.LoadPanel(testWin, STD_NAME)
-                testWin.SetClientSize(testWin.GetBestSize())
+                testWin.panel.SetSize(testWin.GetClientSize())
+                #testWin.SetClientSize(testWin.GetSize())
                 testWin.Show(True)
             elif xxx.__class__ == xxxDialog:
                 testWin = g.testWin = res.LoadDialog(g.frame, STD_NAME)
@@ -1038,7 +1062,10 @@ class XML_Tree(wx.TreeCtrl):
                 wx.EVT_CLOSE(testWin, self.OnCloseTestWin)
                 wx.EVT_SIZE(testWin, self.OnSizeTestWin)
                 # Add drop target
-                testWin.SetDropTarget(DropTarget())
+                if testWin.panel:
+                    testWin.panel.SetDropTarget(DropTarget())
+                else:
+                    testWin.SetDropTarget(DropTarget())
                 # Reset highlights
                 testWin.highLight = testWin.highLightDT = None
                 if highLight and not self.pendingHighLight:
@@ -1069,8 +1096,12 @@ class XML_Tree(wx.TreeCtrl):
         self.CloseTestWindow()
 
     def OnSizeTestWin(self, evt):
-        if g.testWin.highLight:
-            self.HighLight(g.testWin.highLight.item)
+        # Update highlight after size change
+        hl = g.testWin.highLight
+        if hl:
+            hl.Replace(self.FindNodePos(hl.item), hl.obj.GetSize())
+            hl.Refresh()
+            #self.HighLight(g.testWin.highLight.item)
         evt.Skip()
 
     # Return index in parent, for real window children
@@ -1118,14 +1149,6 @@ class XML_Tree(wx.TreeCtrl):
             return False
         return not (self.IsExpanded(item) and self.GetChildrenCount(item, False))
 
-    # Override to use like single-selection tree
-    def GetSelection(self):
-        return self.selection
-    def SelectItem(self, item):
-        self.UnselectAll()
-        self.ChangeSelection(item)
-        wx.TreeCtrl.SelectItem(self, item)
-
     # Pull-down
     def OnRightDown(self, evt):
         pullDownMenu = g.pullDownMenu
@@ -1341,13 +1364,17 @@ class DropTarget(wx.PyDropTarget):
         if d != wx.DragNone:
             obj,parent,parentItem,item = other
             pos, size = g.tree.FindNodePos(parentItem, obj), obj.GetSize()
-            # Change tree item colour
             hl = g.testWin.highLightDT
-            if hl and hl.item and hl.item != parentItem:
-                g.tree.SetItemTextColour(hl.item, g.tree.itemColour)
-            g.testWin.highLightDT = updateHL(hl, HighLightDTBox, pos, size)
-            g.testWin.highLightDT.item = parentItem
-            g.tree.SetItemTextColour(parentItem, COLOUR_DT)
+            # Set color of highlighted item back to normal
+            if hl and hl.item:
+                if hl.item != parentItem:
+                    g.tree.SetItemTextColour(hl.item, g.tree.itemColour)
+                    # Highlight future parent
+                    g.tree.itemColour = g.tree.GetItemTextColour(parentItem) # save current
+            if not hl or hl.item != parentItem:
+                g.testWin.highLightDT = updateHL(hl, HighLightDTBox, pos, size)
+                g.testWin.highLightDT.item = parentItem
+            g.tree.SetItemTextColour(parentItem, g.tree.COLOUR_DT)
             g.tree.EnsureVisible(parentItem)
             g.frame.SetStatusText('Drop target: %s' % parent.treeName())
         else:
@@ -1361,6 +1388,7 @@ class DropTarget(wx.PyDropTarget):
     def RemoveHL(self):
         hl = g.testWin.highLightDT
         if hl:
-            if hl.item: g.tree.SetItemTextColour(hl.item, g.tree.itemColour)
+            if hl.item:
+                g.tree.SetItemTextColour(hl.item, g.tree.itemColour)
             hl.Remove()