/[thuban]/branches/WIP-pyshapelib-bramz/Thuban/UI/legend.py
ViewVC logotype

Diff of /branches/WIP-pyshapelib-bramz/Thuban/UI/legend.py

Parent Directory Parent Directory | Revision Log Revision Log | View Patch Patch

revision 1102 by jonathan, Fri May 30 06:29:05 2003 UTC revision 2562 by jonathan, Wed Feb 16 21:14:47 2005 UTC
# Line 1  Line 1 
1  # Copyright (c) 2001, 2002, 2003 by Intevation GmbH  # Copyright (c) 2001, 2002, 2003, 2005 by Intevation GmbH
2  # Authors:  # Authors:
3  # Jonathan Coles <[email protected]>  # Jonathan Coles <[email protected]>
4  # Frank Koormann <[email protected]>  # Frank Koormann <[email protected]>
# Line 8  Line 8 
8    
9  __version__ = "$Revision$"  __version__ = "$Revision$"
10    
11    from  math import fabs, cos, pi
12    
13  from Thuban import _  from Thuban import _
14    
15  import resource  import resource
16    
17  from wxPython.wx import *  from wxPython.wx import *
18    import wxPython
19    
20  from Thuban.Model.layer import BaseLayer  from Thuban.Model.layer import BaseLayer
21  from Thuban.Model.map import Map  from Thuban.Model.map import Map
22  from Thuban.Model.classification import ClassGroup  from Thuban.Model.classification import ClassGroup
23    from Thuban.Model.proj import PROJ_UNITS_DEGREES
24    
25  from Thuban.Model.messages import \  from Thuban.Model.messages import \
26      MAP_STACKING_CHANGED, MAP_LAYERS_ADDED, MAP_LAYERS_REMOVED, LAYER_CHANGED,\      MAP_STACKING_CHANGED, MAP_LAYERS_ADDED, MAP_LAYERS_REMOVED, LAYER_CHANGED,\
# Line 28  from Thuban.UI.classifier import ClassDa Line 32  from Thuban.UI.classifier import ClassDa
32  from Thuban.UI.dock import DockPanel  from Thuban.UI.dock import DockPanel
33  from Thuban.UI.scalebar import ScaleBar  from Thuban.UI.scalebar import ScaleBar
34    
35    from Thuban.UI.menu import Menu
36    
37  from Thuban.Lib.connector import ConnectorError  from Thuban.Lib.connector import ConnectorError
38    
39  ID_LEGEND_TOP = 4001  ID_LEGEND_TOP = 4001
# Line 50  SHOW_BMP  = "show_layer" Line 56  SHOW_BMP  = "show_layer"
56  HIDE_BMP  = "hide_layer"  HIDE_BMP  = "hide_layer"
57  PROPS_BMP = "layer_properties"  PROPS_BMP = "layer_properties"
58    
   
59  class LegendPanel(DockPanel):  class LegendPanel(DockPanel):
60    
61      def __init__(self, parent, map, mainWindow):      def __init__(self, parent, map, mainWindow):
# Line 179  class LegendPanel(DockPanel): Line 184  class LegendPanel(DockPanel):
184          self.tree.DoOnHideLayer()          self.tree.DoOnHideLayer()
185          pass          pass
186    
187        def _OnToggleVisibility(self, event):
188            self.tree.ToggleVisibility()
189    
190        def _OnProjection(self, event):
191            self.tree.LayerProjection()
192    
193        def _OnRemoveLayer(self, event):
194            self.mainWindow.RemoveLayer()
195    
196        def _OnShowTable(self, event):
197            self.mainWindow.LayerShowTable()
198    
199      def __EnableButtons(self, on):      def __EnableButtons(self, on):
200          self.toolBar.EnableTool(ID_LEGEND_TOP, on)          self.toolBar.EnableTool(ID_LEGEND_TOP, on)
201          self.toolBar.EnableTool(ID_LEGEND_RAISE, on)          self.toolBar.EnableTool(ID_LEGEND_RAISE, on)
# Line 201  class LegendTree(wxTreeCtrl): Line 218  class LegendTree(wxTreeCtrl):
218          self.mainWindow = mainWindow          self.mainWindow = mainWindow
219          self.map = None          self.map = None
220          self.parent = parent          self.parent = parent
221          self.layer2id = {}          self.changing_selection = 0
222    
223          #          #
224          # The image list used by the wxTreeCtrl causes problems when          # The image list used by the wxTreeCtrl causes problems when
# Line 220  class LegendTree(wxTreeCtrl): Line 237  class LegendTree(wxTreeCtrl):
237    
238          self.previewer = ClassDataPreviewer()          self.previewer = ClassDataPreviewer()
239    
240            self.preventExpandCollapse = False
241            self.raiseProperties = False
242    
243          EVT_TREE_ITEM_ACTIVATED(self, ID_LEGEND_TREE, self._OnItemActivated)          EVT_TREE_ITEM_ACTIVATED(self, ID_LEGEND_TREE, self._OnItemActivated)
244          EVT_TREE_SEL_CHANGED(self, ID_LEGEND_TREE, self._OnSelChanged)          EVT_TREE_SEL_CHANGED(self, ID_LEGEND_TREE, self._OnSelChanged)
245            EVT_TREE_ITEM_EXPANDING(self, ID_LEGEND_TREE, self.OnItemExpandCollapse)
246            EVT_TREE_ITEM_COLLAPSING(self, ID_LEGEND_TREE, self.OnItemExpandCollapse)
247            EVT_TREE_ITEM_RIGHT_CLICK(self, ID_LEGEND_TREE, self._OnRightClick)
248    
249          EVT_CLOSE(self, self._OnClose)          EVT_CLOSE(self, self._OnClose)
250    
251          self.SetMap(map)          self.SetMap(map)
252    
253        def _OnRightClick(self, event):
254            """Select item and pop up a context menu"""
255    
256            # The pop up menu is related to the legend tree, so we have direct
257            # access on the tree items. The events issued by the menu are handled
258            # by the legend panel, since most of the handlers are already
259            # implemented there.
260    
261            # Update item selection to the right click
262            item = event.GetItem()
263            self.SelectItem(item)
264    
265            # Define the menu
266            popup_menu = Menu("PopUp", "",
267                              [ "layer_visibility",
268                                None,
269                                "layer_properties",
270                                "layer_projection",
271                                "layer_remove",
272                                "layer_show_table",
273                                None,
274                                "layer_to_top",
275                                "layer_raise",
276                                "layer_lower",
277                                "layer_to_bottom"
278                                ])
279    
280            # Display the menu
281            pos = event.GetPoint()
282            shift = self.ClientToScreen((0,0))
283            self.PopupMenu(self.mainWindow.build_menu(popup_menu), pos)
284    
285        def find_layer(self, layer):
286            """Return the tree item for the layer"""
287            root = self.GetRootItem()
288            id, cookie = self.GetFirstChild(root)
289            while id.IsOk():
290                if self.GetPyData(id) is layer:
291                    return id
292                id, cookie = self.GetNextChild(root, cookie)
293            return None
294    
295      def _OnClose(self, event):      def _OnClose(self, event):
296          self.SetMap(None)          self.SetMap(None)
297    
# Line 315  class LegendTree(wxTreeCtrl): Line 380  class LegendTree(wxTreeCtrl):
380          layer, group = self.GetSelectedHierarchy()          layer, group = self.GetSelectedHierarchy()
381          layer.SetVisible(False)          layer.SetVisible(False)
382    
383        def ToggleVisibility(self):
384            layer, group = self.GetSelectedHierarchy()
385    
386            layer.SetVisible(not layer.Visible())
387    
388        def LayerProjection(self):
389            self.parent.mainWindow.LayerProjection()
390    
391      def Sort(self):      def Sort(self):
392          self.SortChildren(self.GetRootItem())          self.SortChildren(self.GetRootItem())
393    
# Line 340  class LegendTree(wxTreeCtrl): Line 413  class LegendTree(wxTreeCtrl):
413          self.SetMap(self.mainWindow.Map())          self.SetMap(self.mainWindow.Map())
414                    
415      def _OnSelChanged(self, event):      def _OnSelChanged(self, event):
416            # If we change the selection from normalize_selection do nothing.
417            if self.changing_selection:
418                return
419    
420            self.normalize_selection()
421          self.__UpdateSelection()          self.__UpdateSelection()
422    
423        def normalize_selection(self):
424            """Select the layer containing currently selected item"""
425            # This is really a workaround for a bug in wx where deleting a
426            # subtree with DeleteChildren does not update the selection
427            # properly and can lead to segfaults later because the return
428            # value of GetSelection points to invalid data.
429            item = self.GetSelection()
430            while item.IsOk():
431                object = self.GetPyData(item)
432                if isinstance(object, BaseLayer):
433                    break
434                item = self.GetItemParent(item)
435            else:
436                # No layer was found in the chain of parents, so there's
437                # nothing we can do.
438                return
439    
440            self.changing_selection = 1
441            try:
442                self.SelectItem(item)
443            finally:
444                self.changing_selection = 0
445    
446    
447        def OnItemExpandCollapse(self, event):
448            if self.preventExpandCollapse:
449                event.Veto()
450                self.preventExpandCollapse = False
451    
452      def _OnItemActivated(self, event):      def _OnItemActivated(self, event):
453          self.parent.DoOnProperties()          # The following looks strange but is need under Windows to
454            # raise the Properties on double-click: The tree control
455            # always gets an Expanded / Collapsed event after the ItemActivated
456            # on double click, which raises the main window again. We add a second
457            # ItemActivated event to the queue, which simply raises the already
458            # displayed window.
459            if self.raiseProperties:
460                self.parent.DoOnProperties()
461                self.raiseProperties = False
462            else:
463                self.raiseProperties = True
464                self.preventExpandCollapse = True
465                self.parent.DoOnProperties()
466                self.AddPendingEvent(event)
467    
468      def _OnMsgLayerChanged(self, layer):      def _OnMsgLayerChanged(self, layer):
469          assert isinstance(layer, BaseLayer)          assert isinstance(layer, BaseLayer)
470    
471          id = self.layer2id[layer]          id = self.find_layer(layer)
472          assert id.IsOk()          assert id is not None
473    
474          self.__FillTreeLayer(id)          self.__FillTreeLayer(id)
475          self.__UpdateSelection()          self.__UpdateSelection()
# Line 365  class LegendTree(wxTreeCtrl): Line 485  class LegendTree(wxTreeCtrl):
485      def _OnMsgMapLayersAdded(self, map):      def _OnMsgMapLayersAdded(self, map):
486          assert map is self.map          assert map is self.map
487    
488            # Build a dict with all layers known by the the tree as keys
489            layers = {}
490          root = self.GetRootItem()          root = self.GetRootItem()
491            id, cookie = self.GetFirstChild(root)
492            while id.IsOk():
493                layers[self.GetPyData(id)] = 1
494                id, cookie = self.GetNextChild(root, cookie)
495    
496            # Add layers in the map but not in the dict
497          i = 0          i = 0
498          for l in map.Layers():          for l in map.Layers():
499              if not self.layer2id.has_key(l):              if not l in layers:
500                  self.__AddLayer(i, l)                  self.__AddLayer(i, l)
501    
502          self.__UpdateSelection()          self.__UpdateSelection()
# Line 377  class LegendTree(wxTreeCtrl): Line 504  class LegendTree(wxTreeCtrl):
504      def _OnMsgMapLayersRemoved(self, map):      def _OnMsgMapLayersRemoved(self, map):
505          assert map is self.map          assert map is self.map
506    
         layer, group = self.GetSelectedHierarchy()  
   
         if layer is None:  
             assert False, "Shouldn't be allowed."  
             return  
   
507          layers = map.Layers()          layers = map.Layers()
508    
509          for layer, id in self.layer2id.items():          root = self.GetRootItem()
510              if layer not in layers:          id, cookie = self.GetFirstChild(root)
511                  if id.IsOk():          while id.IsOk():
512                      self.__RemoveLayer(id)              if self.GetPyData(id) not in layers:
513                    self.__RemoveLayer(id)
514                id, cookie = self.GetNextChild(root, cookie)
515    
516    
517          self.__UpdateSelection()          self.__UpdateSelection()
518    
# Line 400  class LegendTree(wxTreeCtrl): Line 524  class LegendTree(wxTreeCtrl):
524    
525      def _OnMsgLayerTitleChanged(self, layer):      def _OnMsgLayerTitleChanged(self, layer):
526    
527          id = self.layer2id[layer]          id = self.find_layer(layer)
528          if id.IsOk():          if id.IsOk():
529              self.SetItemText(id, layer.Title())              self.SetItemText(id, layer.Title())
530          self.__UpdateSelection()          self.__UpdateSelection()
# Line 423  class LegendTree(wxTreeCtrl): Line 547  class LegendTree(wxTreeCtrl):
547          self.Thaw()          self.Thaw()
548    
549      def __FillTreeLayer(self, pid):      def __FillTreeLayer(self, pid):
550    
551          layer = self.GetPyData(pid)          layer = self.GetPyData(pid)
552    
553          self.Freeze()          self.Freeze()
# Line 445  class LegendTree(wxTreeCtrl): Line 570  class LegendTree(wxTreeCtrl):
570                      bmp = self.__BuildGroupImage(g, shapeType)                      bmp = self.__BuildGroupImage(g, shapeType)
571    
572                      if bmp is None:                      if bmp is None:
573                          self.SetItemImage(id, -1)                          self.SetItemImage(id, -1, wxTreeItemIcon_Normal)
574                          self.SetItemSelectedImage(id, -1)                          self.SetItemImage(id, -1, wxTreeItemIcon_Selected)
575                            #self.SetItemSelectedImage(id, -1)
576                      else:                      else:
577                          if self.availImgListIndices:                          if self.availImgListIndices:
578                              i = self.availImgListIndices.pop(0)                              i = self.availImgListIndices.pop(0)
# Line 454  class LegendTree(wxTreeCtrl): Line 580  class LegendTree(wxTreeCtrl):
580                          else:                          else:
581                              i = self.image_list.Add(bmp)                              i = self.image_list.Add(bmp)
582    
583                          self.SetItemImage(id, i)                          self.SetItemImage(id, i, wxTreeItemIcon_Normal)
584                          self.SetItemSelectedImage(id, i)                          self.SetItemImage(id, i, wxTreeItemIcon_Selected)
585                            #self.SetItemlectedImage(id, i)
586    
587          self.Thaw()          self.Thaw()
588    
# Line 475  class LegendTree(wxTreeCtrl): Line 602  class LegendTree(wxTreeCtrl):
602    
603          pid = self.GetRootItem()          pid = self.GetRootItem()
604    
605          id, cookie = self.GetFirstChild(pid, 123)          id, cookie = self.GetFirstChild(pid)
606          while id.IsOk():          while id.IsOk():
607              self.__RemoveLayer(id)              self.__RemoveLayer(id)
608              id, cookie = self.GetNextChild(pid, cookie)              id, cookie = self.GetNextChild(pid, cookie)
# Line 492  class LegendTree(wxTreeCtrl): Line 619  class LegendTree(wxTreeCtrl):
619          self.SetPyData(id, l)          self.SetPyData(id, l)
620          self.__SetVisibilityStyle(l.Visible(), id)          self.__SetVisibilityStyle(l.Visible(), id)
621    
         self.layer2id[l] = id  
   
622          self.__FillTreeLayer(id)          self.__FillTreeLayer(id)
623          self.Expand(id)          self.Expand(id)
624    
# Line 513  class LegendTree(wxTreeCtrl): Line 638  class LegendTree(wxTreeCtrl):
638          layer.Unsubscribe(TITLE_CHANGED, self._OnMsgLayerTitleChanged)          layer.Unsubscribe(TITLE_CHANGED, self._OnMsgLayerTitleChanged)
639    
640          self.Delete(id)          self.Delete(id)
         del self.layer2id[layer]  
641    
642      def DeleteChildren(self, pid):      def DeleteChildren(self, pid):
643          id, cookie = self.GetFirstChild(pid, 123)          id, cookie = self.GetFirstChild(pid)
644          while id.IsOk():          while id.IsOk():
645              self.availImgListIndices.append(self.GetItemImage(id))              self.availImgListIndices.append(self.GetItemImage(id))
646              id, cookie = self.GetNextChild(pid, cookie)              id, cookie = self.GetNextChild(pid, cookie)
# Line 527  class LegendTree(wxTreeCtrl): Line 651  class LegendTree(wxTreeCtrl):
651    
652          if not root.IsOk():          if not root.IsOk():
653              self.image_list = wxImageList(BMP_SIZE_W, BMP_SIZE_H, False, 0)              self.image_list = wxImageList(BMP_SIZE_W, BMP_SIZE_H, False, 0)
654                                                                                    
655              bmp = wxEmptyBitmap(BMP_SIZE_W, BMP_SIZE_H)              bmp = wxEmptyBitmap(BMP_SIZE_W, BMP_SIZE_H)
656              dc = wxMemoryDC()              dc = wxMemoryDC()
657              dc.SelectObject(bmp)              dc.SelectObject(bmp)
658              dc.SetBrush(wxBLACK_BRUSH)              dc.SetBrush(wxBLACK_BRUSH)
659              dc.Clear()              dc.Clear()
660              dc.SelectObject(wxNullBitmap)              dc.SelectObject(wxNullBitmap)
661                                                                                    
662              self.emptyImageIndex = \              self.emptyImageIndex = \
663                  self.image_list.AddWithColourMask(bmp, wxColour(0, 0, 0))                  self.image_list.AddWithColourMask(bmp, wxColour(0, 0, 0))
664                                                                                    
665              bmp = resource.GetBitmapResource("legend_icon_layer",              bmp = resource.GetBitmapResource("legend_icon_layer",
666                                                wxBITMAP_TYPE_XPM)                                                wxBITMAP_TYPE_XPM)
667              self.mapImageIndex = \              self.mapImageIndex = \
668                  self.image_list.Add(bmp)                  self.image_list.Add(bmp)
669    
670              self.AssignImageList(self.image_list)              self.AssignImageList(self.image_list)
671                self.availImgListIndices = []
672    
673              root = self.AddRoot("")              root = self.AddRoot("")
674    
# Line 564  class LegendTree(wxTreeCtrl): Line 689  class LegendTree(wxTreeCtrl):
689          self.SetItemFont(id, font)          self.SetItemFont(id, font)
690                    
691      def __ShowHideLayer(self, layer):      def __ShowHideLayer(self, layer):
692          parent = self.layer2id[layer]          parent = self.find_layer(layer)
693          assert parent.IsOk()          assert parent.IsOk()
694    
695          visible = layer.Visible()          visible = layer.Visible()
696    
697          self.__SetVisibilityStyle(visible, parent)          self.__SetVisibilityStyle(visible, parent)
698    
699          id, cookie = self.GetFirstChild(parent, 123)          id, cookie = self.GetFirstChild(parent)
700    
701          while id.IsOk():          while id.IsOk():
702              self.__SetVisibilityStyle(visible, id)              self.__SetVisibilityStyle(visible, id)
703              id, cookie = self.GetNextChild(parent, cookie)              id, cookie = self.GetNextChild(parent, cookie)
704                
705        # In wxPython 2.4 the GetFirstChild method has to be called with a
706        # second argument and in 2.5 it must not.  Reading the code of
707        # wxPython 2.4 it seems that the second parameter was intended to be
708        # optional there but due to a bug in the C++ code it doesn't work
709        # and omitting the second argument leads to a segfault.  To cope
710        # with this and to make the code usable with both 2.5 and 2.4 we
711        # overwrite the inherited method when running with 2.4 to provide a
712        # default value for the second argument.
713        if map(int, wxPython.__version__.split(".")[:2]) < [2, 5]:
714            def GetFirstChild(self, item):
715                return wxTreeCtrl.GetFirstChild(self, item, 0)
716    
717    
718  class ScaleBarBitmap(wxBoxSizer):  class ScaleBarBitmap(wxBoxSizer):
719    
720      def __init__(self, parent, map, mainWindow):      def __init__(self, parent, map, mainWindow):
# Line 618  class ScaleBarBitmap(wxBoxSizer): Line 756  class ScaleBarBitmap(wxBoxSizer):
756          dc.SelectObject(bmp)          dc.SelectObject(bmp)
757          dc.Clear()          dc.Clear()
758    
759          self.scalebar.DrawScaleBar(scale, dc, (0,0), dc.GetSizeTuple())          if self.canvas.map is not None \
760                and self.canvas.map.projection is not None:
761    
762                # if we are using a projection with geographics coordinates
763                # we need to change the scale value based on where we are
764                # on the globe.
765                if self.canvas.map.projection.GetProjectedUnits() \
766                    == PROJ_UNITS_DEGREES:
767    
768                    width, height = self.canvas.GetSizeTuple()
769                    long, lat = self.canvas.win_to_proj(width/2, height/2)
770    
771                    # slightly inaccurate, but if we are looking at
772                    # the north/south pole we could end up dividing by zero
773                    #
774                    # it shouldn't matter for our purposes that we ignore
775                    # the original sign of lat.
776                    if fabs(lat) > 89.9: lat = 89.9
777    
778                    #
779                    # one degree is about 111,133m at the equator
780                    # we need to adjust that for latitude as
781                    # we move north/south. use the center of the map
782                    # as the point to scale the length to.
783                    #
784                    scale = scale / (111133.0 * fabs(cos(lat * pi/180)))
785    
786                self.scalebar.DrawScaleBar(scale, dc, (0,0), dc.GetSizeTuple())
787    
788          self.scalebarBitmap.SetBitmap(bmp)          self.scalebarBitmap.SetBitmap(bmp)
789    
Legend:
Removed from v.1102  
changed lines
  Added in v.2562
[email protected]
 ViewVC Help
Powered by ViewVC 1.1.26