/[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 652 by jonathan, Fri Apr 11 14:28:08 2003 UTC revision 882 by jonathan, Fri May 9 16:34:28 2003 UTC
# Line 15  from wxPython.wx import * Line 15  from wxPython.wx import *
15    
16  from Thuban.Model.layer import Layer  from Thuban.Model.layer import Layer
17  from Thuban.Model.map import Map  from Thuban.Model.map import Map
 from Thuban.Model.messages import *  
18  from Thuban.Model.classification import ClassGroup  from Thuban.Model.classification import ClassGroup
19    
20  from Thuban.UI.messages import *  from Thuban.Model.messages import \
21        MAP_STACKING_CHANGED, MAP_LAYERS_ADDED, MAP_LAYERS_REMOVED, LAYER_CHANGED,\
22        LAYER_VISIBILITY_CHANGED, TITLE_CHANGED
23    
24    from Thuban.UI.messages import SCALE_CHANGED
25    
26  from Thuban.UI.classifier import ClassDataPreviewer  from Thuban.UI.classifier import ClassDataPreviewer
27  from Thuban.UI.dock import DockPanel  from Thuban.UI.dock import DockPanel
28    from Thuban.UI.scalebar import ScaleBar
29    
30  from Thuban.Lib.connector import ConnectorError  from Thuban.Lib.connector import ConnectorError
31    
# Line 54  class LegendPanel(DockPanel): Line 59  class LegendPanel(DockPanel):
59          panelBox = wxBoxSizer(wxVERTICAL)          panelBox = wxBoxSizer(wxVERTICAL)
60    
61          self.toolBar = wxToolBar(self, -1)          self.toolBar = wxToolBar(self, -1)
62            self.toolBar.SetToolBitmapSize(wxSize(24, 24))
63    
64          bmp = resource.GetBitmapResource(RAISE_BMP, wxBITMAP_TYPE_XPM)          bmp = resource.GetBitmapResource(RAISE_BMP, wxBITMAP_TYPE_XPM)
65          self.toolBar.AddTool(ID_LEGEND_RAISE, bmp,          self.toolBar.AddTool(ID_LEGEND_RAISE, bmp,
# Line 76  class LegendPanel(DockPanel): Line 82  class LegendPanel(DockPanel):
82              shortHelpString=_("Edit Layer Properties"))              shortHelpString=_("Edit Layer Properties"))
83    
84          self.toolBar.Realize()          self.toolBar.Realize()
85          panelBox.Add(self.toolBar, 0, wxALL, 0)          panelBox.Add(self.toolBar, 0, wxGROW, 0)
86    
87          EVT_TOOL(self, ID_LEGEND_RAISE, self._OnMoveUp)          EVT_TOOL(self, ID_LEGEND_RAISE, self._OnMoveUp)
88          EVT_TOOL(self, ID_LEGEND_LOWER, self._OnMoveDown)          EVT_TOOL(self, ID_LEGEND_LOWER, self._OnMoveDown)
# Line 86  class LegendPanel(DockPanel): Line 92  class LegendPanel(DockPanel):
92    
93          self.tree = LegendTree(self, ID_LEGEND_TREE, map, mainWindow)          self.tree = LegendTree(self, ID_LEGEND_TREE, map, mainWindow)
94    
95          panelBox.Add(self.tree, 1, wxGROW, 4)          panelBox.Add(self.tree, 1, wxGROW, 0)
96    
97          panelBox.Fit(self)          self.scalebarbitmap = ScaleBarBitmap(self, map, mainWindow)
98            panelBox.Add(self.scalebarbitmap, 0, wxGROW, 0)
99    
100          self.SetAutoLayout(True)          self.SetAutoLayout(True)
101          self.SetSizer(panelBox)          self.SetSizer(panelBox)
102          panelBox.SetSizeHints(self)          panelBox.SetSizeHints(self)
103    
104    
105          self.panelBox = panelBox          self.panelBox = panelBox
106    
107          self.__EnableButtons(False)          self.__EnableButtons(False)
108    
109            self.Create()
110    
111          EVT_CLOSE(self, self._OnClose)          EVT_CLOSE(self, self._OnClose)
112    
113    
# Line 106  class LegendPanel(DockPanel): Line 116  class LegendPanel(DockPanel):
116    
117      def SetMap(self, map):      def SetMap(self, map):
118          self.tree.SetMap(map)          self.tree.SetMap(map)
119            self.scalebarbitmap.SetCanvas(self.mainWindow.canvas)
120    
121      def DoOnSelChanged(self, layer, group):      def DoOnSelChanged(self, layer, group):
122    
123          ok = isinstance(layer, Layer)          ok = isinstance(layer, Layer)
124          self.__EnableButtons(ok)          self.__EnableButtons(ok)
125    
126          if ok:          self.mainWindow.SelectLayer(layer)
             self.mainWindow.SelectLayer(layer)  
127    
128      def DoOnProperties(self):      def DoOnProperties(self):
129          list = self.tree.GetSelectedHierarchy()          list = self.tree.GetSelectedHierarchy()
# Line 215  class LegendTree(wxTreeCtrl): Line 225  class LegendTree(wxTreeCtrl):
225                  #self._OnMsgMapsChanged)                  #self._OnMsgMapsChanged)
226              self.__FillTree(self.map)              self.__FillTree(self.map)
227    
               
228      def MoveCurrentItemUp(self):      def MoveCurrentItemUp(self):
229          cur_id = self.GetSelection()          layer, group = self.GetSelectedHierarchy()
         assert cur_id.IsOk()  
   
         cur_data = self.GetPyData(cur_id)  
   
         #prev_id = self.GetPrevSibling(cur_id)  
   
         #  
         # Get out if there's nowhere to go  
         #  
         #if prev_id == INVALID_TREE_ID: return  
230    
231          if isinstance(cur_data, Layer):          if layer is not None:
232              self.map.RaiseLayer(cur_data)              self.map.RaiseLayer(layer)
         elif isinstance(cur_data, ClassGroup):  
             pass  
233          else:          else:
234              assert False, "Shouldn't be here."              assert False, "Shouldn't be allowed."
235              pass              pass
236    
237      def MoveCurrentItemDown(self):      def MoveCurrentItemDown(self):
238          cur_id = self.GetSelection()          layer, group = self.GetSelectedHierarchy()
         assert cur_id.IsOk()  
   
         cur_data = self.GetPyData(cur_id)  
239    
240          if isinstance(cur_data, Layer):          if layer is not None:
241              self.map.LowerLayer(cur_data)              self.map.LowerLayer(layer)
         elif isinstance(cur_data, ClassGroup):  
             pass  
242          else:          else:
243              assert False, "Shouldn't be here."              assert False, "Shouldn't be allowed."
244              pass              pass
245    
   
246      def OnCompareItems(self, item1, item2):      def OnCompareItems(self, item1, item2):
247    
248          data1 = self.GetPyData(item1)          data1 = self.GetPyData(item1)
# Line 263  class LegendTree(wxTreeCtrl): Line 254  class LegendTree(wxTreeCtrl):
254          else:          else:
255              return wxTreeCtrl.OnCompareItems(self, item1, item2)              return wxTreeCtrl.OnCompareItems(self, item1, item2)
256    
   
257      def DoOnShowLayer(self):      def DoOnShowLayer(self):
258          #self.__ShowHideLayer(True)          #self.__ShowHideLayer(True)
259          layer, group = self.GetSelectedHierarchy()          layer, group = self.GetSelectedHierarchy()
# Line 277  class LegendTree(wxTreeCtrl): Line 267  class LegendTree(wxTreeCtrl):
267      def Sort(self):      def Sort(self):
268          self.SortChildren(self.GetRootItem())          self.SortChildren(self.GetRootItem())
269    
270        def GetSelectedHierarchy(self):
271            id = self.GetSelection()
272    
273            if not id.IsOk():
274                return (None, None)
275    
276            layer = self.GetPyData(id)
277            group = None
278    
279            if isinstance(layer, ClassGroup):
280                id = self.GetItemParent(id)
281                assert id.IsOk()
282                group = layer
283                layer = self.GetPyData(id)
284    
285            return (layer, group)
286    
287      def _OnMsgMapsChanged(self):      def _OnMsgMapsChanged(self):
288          #print self.map is self.mainWindow.Map()          #print self.map is self.mainWindow.Map()
289          self.SetMap(self.mainWindow.Map())          self.SetMap(self.mainWindow.Map())
290                    
291      def _OnSelChanged(self, event):      def _OnSelChanged(self, event):
292            self.__UpdateSelection()
         layer, group = self.GetSelectedHierarchy()  
         self.parent.DoOnSelChanged(layer, group)  
293    
294      def _OnItemActivated(self, event):      def _OnItemActivated(self, event):
295          self.parent.DoOnProperties()          self.parent.DoOnProperties()
# Line 300  class LegendTree(wxTreeCtrl): Line 305  class LegendTree(wxTreeCtrl):
305          #      track of the images in the image list when we replace          #      track of the images in the image list when we replace
306          #      a layer. it ends up causing a seg fault.          #      a layer. it ends up causing a seg fault.
307          self.__FillTree(self.map)          self.__FillTree(self.map)
308            self.__UpdateSelection()
309          #self.__FillTreeLayer(id)          #self.__FillTreeLayer(id)
310    
311      def _OnMsgMapStackingChanged(self, *args):      def _OnMsgMapStackingChanged(self, *args):
# Line 308  class LegendTree(wxTreeCtrl): Line 314  class LegendTree(wxTreeCtrl):
314    
315          if id.IsOk():          if id.IsOk():
316              self.EnsureVisible(id)              self.EnsureVisible(id)
317            self.__UpdateSelection()
318    
319      def _OnMsgMapLayersAddedRemoved(self, map):      def _OnMsgMapLayersAddedRemoved(self, map):
320          assert map is self.map          assert map is self.map
321    
322          self.__FillTree(self.map)          self.__FillTree(self.map)
323            self.__UpdateSelection()
324    
325      def _OnMsgLayerVisibilityChanged(self, layer):      def _OnMsgLayerVisibilityChanged(self, layer):
326          assert isinstance(layer, Layer)          assert isinstance(layer, Layer)
327    
328          self.__ShowHideLayer(layer)          self.__ShowHideLayer(layer)
329            self.__UpdateSelection()
330    
331      def _OnMsgLayerTitleChanged(self, layer):      def _OnMsgLayerTitleChanged(self, layer):
332    
333          id = self.layer2id[layer]          id = self.layer2id[layer]
334          if id.IsOk():          if id.IsOk():
335              self.SetItemText(id, layer.Title())              self.SetItemText(id, layer.Title())
336            self.__UpdateSelection()
337    
338      def GetSelectedHierarchy(self):      def __UpdateSelection(self):
339          id = self.GetSelection()          layer, group = self.GetSelectedHierarchy()
340          assert id.IsOk()          self.parent.DoOnSelChanged(layer, group)
341            
         layer = self.GetPyData(id)  
         group = None  
   
         if isinstance(layer, ClassGroup):  
             id = self.GetItemParent(id)  
             assert id.IsOk()  
             group = layer  
             layer = self.GetPyData(id)  
   
         return (layer, group)  
   
342      def __FillTree(self, map):      def __FillTree(self, map):
343    
         assert isinstance(map, Map)  
   
344          self.Freeze()          self.Freeze()
345    
346          self.__DeleteAllItems()          self.__DeleteAllItems()
# Line 411  class LegendTree(wxTreeCtrl): Line 408  class LegendTree(wxTreeCtrl):
408          self.Thaw()          self.Thaw()
409    
410      def __BuildGroupImage(self, group, shapeType):      def __BuildGroupImage(self, group, shapeType):
         assert isinstance(group, ClassGroup)  
411    
412          bmp = wxEmptyBitmap(BMP_SIZE_W, BMP_SIZE_H)          bmp = wxEmptyBitmap(BMP_SIZE_W, BMP_SIZE_H)
413          #brush = wxBrush(Color2wxColour(item[1]), wxSOLID)          #brush = wxBrush(Color2wxColour(item[1]), wxSOLID)
# Line 462  class LegendTree(wxTreeCtrl): Line 458  class LegendTree(wxTreeCtrl):
458              self.__SetVisibilityStyle(visible, id)              self.__SetVisibilityStyle(visible, id)
459              id, cookie = self.GetNextChild(parent, cookie)              id, cookie = self.GetNextChild(parent, cookie)
460                            
461    class ScaleBarBitmap(wxBoxSizer):
462    
463        def __init__(self, parent, map, mainWindow):
464            # While the width is fixed, get the height _now_.
465            dc = wxMemoryDC()
466            textwidth, textheight = dc.GetTextExtent("%d"%0)
467            self.width = 200
468            self.height = textheight + 3*2 + 8
469    
470            wxBoxSizer.__init__(self, wxVERTICAL)
471            bmp=wxEmptyBitmap(self.width, self.height)
472            self.scalebarBitmap = wxStaticBitmap(parent, -1, bmp)
473            self.Add(self.scalebarBitmap, 0, wxALIGN_CENTER|wxLEFT|wxTOP|wxRIGHT, 1)
474    
475            self.mainWindow = mainWindow
476            self.parent = parent
477            self.canvas = None
478            self.SetCanvas(self.mainWindow.canvas)
479    
480        def SetCanvas(self, canvas):
481            sub_list = [(SCALE_CHANGED, self._OnMsgScaleChanged)]
482    
483            if self.canvas is not None:
484                for msg, func in sub_list: self.canvas.Unsubscribe(msg, func)
485            
486            self.canvas = canvas
487            self.scalebar = ScaleBar(canvas.map)
488    
489            if self.canvas is not None:
490                for msg, func in sub_list: self.canvas.Subscribe(msg, func)
491                self.__SetScale(self.canvas.scale)
492    
493        def _OnMsgScaleChanged(self, scale):
494            self.__SetScale(scale)
495    
496        def __SetScale(self, scale):
497            bmp = wxEmptyBitmap(self.width, self.height)
498            dc = wxMemoryDC()
499            dc.SelectObject(bmp)
500            dc.Clear()
501    
502            self.scalebar.DrawScalebar(scale, dc)
503    
504            self.scalebarBitmap.SetBitmap(bmp)
505    
Legend:
Removed from v.652  
changed lines
  Added in v.882
[email protected]
 ViewVC Help
Powered by ViewVC 1.1.26