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

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

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

revision 356 by bh, Mon Dec 9 10:32:15 2002 UTC revision 883 by jonathan, Fri May 9 16:34:39 2003 UTC
# Line 1  Line 1 
1  # Copyright (c) 2001, 2002 by Intevation GmbH  # Copyright (c) 2001, 2002, 2003 by Intevation GmbH
2  # Authors:  # Authors:
3  # Bernhard Herzog <[email protected]>  # Bernhard Herzog <[email protected]>
4  #  #
# Line 11  Classes for display of a map and interac Line 11  Classes for display of a map and interac
11    
12  __version__ = "$Revision$"  __version__ = "$Revision$"
13    
14    import sys
15    
16  from math import hypot  from math import hypot
17    
18  from wxPython.wx import wxWindow,\  from wxPython.wx import wxWindow,\
19       wxPaintDC, wxColour, wxClientDC, wxINVERT, wxTRANSPARENT_BRUSH, wxFont,\       wxPaintDC, wxColour, wxClientDC, wxINVERT, wxTRANSPARENT_BRUSH, wxFont,\
20       EVT_PAINT, EVT_LEFT_DOWN, EVT_LEFT_UP, EVT_MOTION, EVT_LEAVE_WINDOW       EVT_PAINT, EVT_LEFT_DOWN, EVT_LEFT_UP, EVT_MOTION, EVT_LEAVE_WINDOW, \
21         wxBITMAP_TYPE_XPM, wxBeginBusyCursor, wxEndBusyCursor, wxCursor, \
22         wxImageFromBitmap
23    
24    
25  from wxPython import wx  from wxPython import wx
# Line 24  from wxproj import point_in_polygon_shap Line 28  from wxproj import point_in_polygon_shap
28    
29    
30  from Thuban.Model.messages import MAP_PROJECTION_CHANGED, \  from Thuban.Model.messages import MAP_PROJECTION_CHANGED, \
31       LAYERS_CHANGED, LAYER_LEGEND_CHANGED, LAYER_VISIBILITY_CHANGED       MAP_LAYERS_CHANGED, LAYER_CHANGED, LAYER_VISIBILITY_CHANGED
32  from Thuban.Model.layer import SHAPETYPE_POLYGON, SHAPETYPE_ARC, \  from Thuban.Model.layer import SHAPETYPE_POLYGON, SHAPETYPE_ARC, \
33       SHAPETYPE_POINT       SHAPETYPE_POINT
34  from Thuban.Model.label import ALIGN_CENTER, ALIGN_TOP, ALIGN_BOTTOM, \  from Thuban.Model.label import ALIGN_CENTER, ALIGN_TOP, ALIGN_BOTTOM, \
35       ALIGN_LEFT, ALIGN_RIGHT       ALIGN_LEFT, ALIGN_RIGHT
36  from Thuban.Lib.connector import Publisher  from Thuban.Lib.connector import Publisher
37    from Thuban.Model.color import Color
38    
39    import resource
40    
41    from selection import Selection
42  from renderer import ScreenRenderer, PrinterRender  from renderer import ScreenRenderer, PrinterRender
43    
44  import labeldialog  import labeldialog
45    
46  from messages import SELECTED_SHAPE, VIEW_POSITION  from messages import LAYER_SELECTED, SHAPES_SELECTED, VIEW_POSITION, \
47                         SCALE_CHANGED
48    
49    
50  #  #
# Line 256  class MapCanvas(wxWindow, Publisher): Line 265  class MapCanvas(wxWindow, Publisher):
265    
266      """A widget that displays a map and offers some interaction"""      """A widget that displays a map and offers some interaction"""
267    
268      def __init__(self, parent, winid, interactor):      # Some messages that can be subscribed/unsubscribed directly through
269        # the MapCanvas come in fact from other objects. This is a dict
270        # mapping those messages to the names of the instance variables they
271        # actually come from. The delegation is implemented in the Subscribe
272        # and Unsubscribe methods
273        delegated_messages = {LAYER_SELECTED: "selection",
274                              SHAPES_SELECTED: "selection"}
275    
276        # Methods delegated to some instance variables. The delegation is
277        # implemented in the __getattr__ method.
278        delegated_methods = {"SelectLayer": "selection",
279                             "SelectShapes": "selection",
280                             "SelectedLayer": "selection",
281                             "HasSelectedLayer": "selection",
282                             "HasSelectedShapes": "selection",
283                             "SelectedShapes": "selection"}
284    
285        def __init__(self, parent, winid):
286          wxWindow.__init__(self, parent, winid)          wxWindow.__init__(self, parent, winid)
287          self.SetBackgroundColour(wxColour(255, 255, 255))          self.SetBackgroundColour(wxColour(255, 255, 255))
288    
# Line 282  class MapCanvas(wxWindow, Publisher): Line 308  class MapCanvas(wxWindow, Publisher):
308          # the bitmap serving as backing store          # the bitmap serving as backing store
309          self.bitmap = None          self.bitmap = None
310    
311          # the interactor          # the selection
312          self.interactor = interactor          self.selection = Selection()
313          self.interactor.Subscribe(SELECTED_SHAPE, self.shape_selected)          self.selection.Subscribe(SHAPES_SELECTED , self.shape_selected)
314    
315          # keep track of which layers/shapes are selected to make sure we          # keep track of which layers/shapes are selected to make sure we
316          # only redraw when necessary          # only redraw when necessary
# Line 303  class MapCanvas(wxWindow, Publisher): Line 329  class MapCanvas(wxWindow, Publisher):
329          wxWindow.__del__(self)          wxWindow.__del__(self)
330          Publisher.__del__(self)          Publisher.__del__(self)
331    
332        def Subscribe(self, channel, *args):
333            """Extend the inherited method to handle delegated messages.
334    
335            If channel is one of the delegated messages call the appropriate
336            object's Subscribe method. Otherwise just call the inherited
337            method.
338            """
339            if channel in self.delegated_messages:
340                object = getattr(self, self.delegated_messages[channel])
341                object.Subscribe(channel, *args)
342            else:
343                Publisher.Subscribe(self, channel, *args)
344    
345        def Unsubscribe(self, channel, *args):
346            """Extend the inherited method to handle delegated messages.
347    
348            If channel is one of the delegated messages call the appropriate
349            object's Unsubscribe method. Otherwise just call the inherited
350            method.
351            """
352            if channel in self.delegated_messages:
353                object = getattr(self, self.delegated_messages[channel])
354                object.Unsubscribe(channel, *args)
355            else:
356                Publisher.Unsubscribe(self, channel, *args)
357    
358        def __getattr__(self, attr):
359            if attr in self.delegated_methods:
360                return getattr(getattr(self, self.delegated_methods[attr]), attr)
361            raise AttributeError(attr)
362    
363      def OnPaint(self, event):      def OnPaint(self, event):
364          dc = wxPaintDC(self)          dc = wxPaintDC(self)
365          if self.map is not None and self.map.HasLayers():          clear = self.map is None or not self.map.HasLayers()
366              self.do_redraw()  
367          else:          #wxBeginBusyCursor()
368    
369            if not clear:
370                try:
371                    self.do_redraw()
372                except:
373                    print "Error during drawing:", sys.exc_info()[0]
374                    clear = True
375    
376            if clear:
377              # If we've got no map or if the map is empty, simply clear              # If we've got no map or if the map is empty, simply clear
378              # the screen.              # the screen.
379    
# Line 319  class MapCanvas(wxWindow, Publisher): Line 385  class MapCanvas(wxWindow, Publisher):
385              dc.Clear()              dc.Clear()
386              dc.EndDrawing()              dc.EndDrawing()
387    
388            #wxEndBusyCursor()
389    
390      def do_redraw(self):      def do_redraw(self):
391          # This should only be called if we have a non-empty map.          # This should only be called if we have a non-empty map.
392    
# Line 335  class MapCanvas(wxWindow, Publisher): Line 403  class MapCanvas(wxWindow, Publisher):
403              dc.BeginDrawing()              dc.BeginDrawing()
404    
405              # clear the background              # clear the background
406              dc.SetBrush(wx.wxWHITE_BRUSH)              #dc.SetBrush(wx.wxWHITE_BRUSH)
407              dc.SetPen(wx.wxTRANSPARENT_PEN)              #dc.SetPen(wx.wxTRANSPARENT_PEN)
408              dc.DrawRectangle(0, 0, width, height)              #dc.DrawRectangle(0, 0, width, height)
409                dc.SetBackground(wx.wxWHITE_BRUSH)
410              if 1: #self.interactor.selected_map is self.map:              dc.Clear()
411                  selected_layer = self.interactor.selected_layer  
412                  selected_shape = self.interactor.selected_shape              selected_layer = self.selection.SelectedLayer()
413              else:              selected_shapes = self.selection.SelectedShapes()
                 selected_layer = None  
                 selected_shape = None  
414    
415              # draw the map into the bitmap              # draw the map into the bitmap
416              renderer = ScreenRenderer(dc, self.scale, self.offset)              renderer = ScreenRenderer(dc, self.scale, self.offset)
# Line 352  class MapCanvas(wxWindow, Publisher): Line 418  class MapCanvas(wxWindow, Publisher):
418              # Pass the entire bitmap as update region to the renderer.              # Pass the entire bitmap as update region to the renderer.
419              # We're redrawing the whole bitmap, after all.              # We're redrawing the whole bitmap, after all.
420              renderer.RenderMap(self.map, (0, 0, width, height),              renderer.RenderMap(self.map, (0, 0, width, height),
421                                 selected_layer, selected_shape)                                 selected_layer, selected_shapes)
422    
423              dc.EndDrawing()              dc.EndDrawing()
424              dc.SelectObject(wx.wxNullBitmap)              dc.SelectObject(wx.wxNullBitmap)
# Line 373  class MapCanvas(wxWindow, Publisher): Line 439  class MapCanvas(wxWindow, Publisher):
439          printout.Destroy()          printout.Destroy()
440    
441      def SetMap(self, map):      def SetMap(self, map):
442          redraw_channels = (LAYERS_CHANGED, LAYER_LEGEND_CHANGED,          redraw_channels = (MAP_LAYERS_CHANGED, LAYER_CHANGED,
443                             LAYER_VISIBILITY_CHANGED)                             LAYER_VISIBILITY_CHANGED)
444          if self.map is not None:          if self.map is not None:
445              for channel in redraw_channels:              for channel in redraw_channels:
# Line 381  class MapCanvas(wxWindow, Publisher): Line 447  class MapCanvas(wxWindow, Publisher):
447              self.map.Unsubscribe(MAP_PROJECTION_CHANGED,              self.map.Unsubscribe(MAP_PROJECTION_CHANGED,
448                                   self.projection_changed)                                   self.projection_changed)
449          self.map = map          self.map = map
450            self.selection.ClearSelection()
451          if self.map is not None:          if self.map is not None:
452              for channel in redraw_channels:              for channel in redraw_channels:
453                  self.map.Subscribe(channel, self.full_redraw)                  self.map.Subscribe(channel, self.full_redraw)
# Line 410  class MapCanvas(wxWindow, Publisher): Line 477  class MapCanvas(wxWindow, Publisher):
477          self.scale = scale          self.scale = scale
478          self.offset = offset          self.offset = offset
479          self.full_redraw()          self.full_redraw()
480            self.issue(SCALE_CHANGED, scale)
481    
482      def proj_to_win(self, x, y):      def proj_to_win(self, x, y):
483          """\          """\
# Line 427  class MapCanvas(wxWindow, Publisher): Line 495  class MapCanvas(wxWindow, Publisher):
495    
496      def FitRectToWindow(self, rect):      def FitRectToWindow(self, rect):
497          """Fit the rectangular region given by rect into the window.          """Fit the rectangular region given by rect into the window.
498            
499          Set scale so that rect (in projected coordinates) just fits into          Set scale so that rect (in projected coordinates) just fits into
500          the window and center it.          the window and center it.
501          """          """
502          width, height = self.GetSizeTuple()          width, height = self.GetSizeTuple()
503          llx, lly, urx, ury = rect          llx, lly, urx, ury = rect
504          if llx == urx or lly == ury:          if llx == urx or lly == ury:
505              # zero with or zero height. Do Nothing              # zero width or zero height. Do Nothing
506              return              return
507          scalex = width / (urx - llx)          scalex = width / (urx - llx)
508          scaley = height / (ury - lly)          scaley = height / (ury - lly)
# Line 445  class MapCanvas(wxWindow, Publisher): Line 513  class MapCanvas(wxWindow, Publisher):
513    
514      def FitMapToWindow(self):      def FitMapToWindow(self):
515          """Fit the map to the window          """Fit the map to the window
516            
517          Set the scale so that the map fits exactly into the window and          Set the scale so that the map fits exactly into the window and
518          center it in the window.          center it in the window.
519          """          """
# Line 453  class MapCanvas(wxWindow, Publisher): Line 521  class MapCanvas(wxWindow, Publisher):
521          if bbox is not None:          if bbox is not None:
522              self.FitRectToWindow(bbox)              self.FitRectToWindow(bbox)
523    
524        def FitLayerToWindow(self, layer):
525            """Fit the given layer to the window.
526    
527            Set the scale so that the layer fits exactly into the window and
528            center it in the window.
529            """
530            
531            bbox = layer.LatLongBoundingBox()
532            if bbox is not None:
533                proj = self.map.GetProjection()
534                if proj is not None:
535                    bbox = proj.ForwardBBox(bbox)
536    
537                if bbox is not None:
538                    self.FitRectToWindow(bbox)
539    
540        def FitSelectedToWindow(self):
541            layer = self.selection.SelectedLayer()
542            shapes = self.selection.SelectedShapes()
543    
544            bbox = layer.ShapesBoundingBox(shapes)
545            if bbox is not None:
546                proj = self.map.GetProjection()
547                if proj is not None:
548                    bbox = proj.ForwardBBox(bbox)
549    
550                if bbox is not None:
551                    self.FitRectToWindow(bbox)
552    
553      def ZoomFactor(self, factor, center = None):      def ZoomFactor(self, factor, center = None):
554          """Multiply the zoom by factor and center on center.          """Multiply the zoom by factor and center on center.
555    
# Line 516  class MapCanvas(wxWindow, Publisher): Line 613  class MapCanvas(wxWindow, Publisher):
613      def PanTool(self):      def PanTool(self):
614          """Start the pan tool"""          """Start the pan tool"""
615          self.SelectTool(PanTool(self))          self.SelectTool(PanTool(self))
616            #img = resource.GetImageResource("pan", wxBITMAP_TYPE_XPM)
617            #bmp = resource.GetBitmapResource("pan", wxBITMAP_TYPE_XPM)
618            #print bmp
619            #img = wxImageFromBitmap(bmp)
620            #print img
621            #cur = wxCursor(img)
622            #print cur
623            #self.SetCursor(cur)
624    
625      def IdentifyTool(self):      def IdentifyTool(self):
626          """Start the identify tool"""          """Start the identify tool"""
# Line 569  class MapCanvas(wxWindow, Publisher): Line 674  class MapCanvas(wxWindow, Publisher):
674          self.set_current_position(event)          self.set_current_position(event)
675          if self.dragging:          if self.dragging:
676              self.ReleaseMouse()              self.ReleaseMouse()
677              self.tool.Hide(self.drag_dc)              try:
678              self.tool.MouseUp(event)                  self.tool.Hide(self.drag_dc)
679              self.drag_dc = None                  self.tool.MouseUp(event)
680          self.dragging = 0              finally:
681                    self.drag_dc = None
682                    self.dragging = 0
683    
684      def OnMotion(self, event):      def OnMotion(self, event):
685          self.set_current_position(event)          self.set_current_position(event)
# Line 592  class MapCanvas(wxWindow, Publisher): Line 699  class MapCanvas(wxWindow, Publisher):
699          # Even when the window becomes larger some parts of the bitmap          # Even when the window becomes larger some parts of the bitmap
700          # could be reused.          # could be reused.
701          self.full_redraw()          self.full_redraw()
702            pass
703    
704      def shape_selected(self, layer, shape):      def shape_selected(self, layer, shape):
705          """Redraw the map.          """Receiver for the SHAPES_SELECTED messages. Redraw the map."""
706            # The selection object takes care that it only issues
707          Receiver for the SELECTED_SHAPE messages. Try to redraw only          # SHAPES_SELECTED messages when the set of selected shapes has
708          when necessary.          # actually changed, so we can do a full redraw unconditionally.
709          """          # FIXME: We should perhaps try to limit the redraw to the are
710          # A redraw is necessary when the display has to change, which          # actually covered by the shapes before and after the selection
711          # means that either the status changes from having no selection          # change.
712          # to having a selection shape or vice versa, or when the fact          self.full_redraw()
         # whether there is a selection at all doesn't change, when the  
         # shape which is selected has changed (which means that layer or  
         # shapeid changes).  
         if ((shape is not None or self.last_selected_shape is not None)  
             and (shape != self.last_selected_shape  
                  or layer != self.last_selected_layer)):  
             self.full_redraw()  
   
         # remember the selection so we can compare when it changes again.  
         self.last_selected_layer = layer  
         self.last_selected_shape = shape  
713    
714      def unprojected_rect_around_point(self, x, y, dist):      def unprojected_rect_around_point(self, x, y, dist):
715          """return a rect dist pixels around (x, y) in unprojected corrdinates          """return a rect dist pixels around (x, y) in unprojected corrdinates
# Line 704  class MapCanvas(wxWindow, Publisher): Line 801  class MapCanvas(wxWindow, Publisher):
801              if not layer.Visible():              if not layer.Visible():
802                  continue                  continue
803    
804              filled = layer.fill is not None              filled = layer.GetClassification().GetDefaultFill() \
805              stroked = layer.stroke is not None                       is not Color.Transparent
806                stroked = layer.GetClassification().GetDefaultLineColor() \
807                          is not Color.Transparent
808    
809              layer_proj = layer.projection              layer_proj = layer.projection
810              if layer_proj is not None:              if layer_proj is not None:
# Line 784  class MapCanvas(wxWindow, Publisher): Line 883  class MapCanvas(wxWindow, Publisher):
883          # to deselect the currently selected layer, so we simply select          # to deselect the currently selected layer, so we simply select
884          # the already selected layer again.          # the already selected layer again.
885          if layer is None:          if layer is None:
886              layer = self.interactor.SelectedLayer()              layer = self.selection.SelectedLayer()
887          self.interactor.SelectLayerAndShape(layer, shape)              shapes = []
888            else:
889                shapes = [shape]
890            self.selection.SelectShapes(layer, shapes)
891          return result          return result
892    
893      def LabelShapeAt(self, x, y):      def LabelShapeAt(self, x, y):
Legend:
Removed from v.356  
changed lines
  Added in v.883
[email protected]
 ViewVC Help
Powered by ViewVC 1.1.26