/[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 295 by bh, Fri Aug 30 10:39:17 2002 UTC revision 822 by jonathan, Mon May 5 18:20:28 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    
48    
49  #  #
# Line 256  class MapCanvas(wxWindow, Publisher): Line 264  class MapCanvas(wxWindow, Publisher):
264    
265      """A widget that displays a map and offers some interaction"""      """A widget that displays a map and offers some interaction"""
266    
267      def __init__(self, parent, winid, interactor):      # Some messages that can be subscribed/unsubscribed directly through
268        # the MapCanvas come in fact from other objects. This is a dict
269        # mapping those messages to the names of the instance variables they
270        # actually come from. The delegation is implemented in the Subscribe
271        # and Unsubscribe methods
272        delegated_messages = {LAYER_SELECTED: "selection",
273                              SHAPES_SELECTED: "selection"}
274    
275        # Methods delegated to some instance variables. The delegation is
276        # implemented in the __getattr__ method.
277        delegated_methods = {"SelectLayer": "selection",
278                             "SelectShapes": "selection",
279                             "SelectedLayer": "selection",
280                             "HasSelectedLayer": "selection"}
281    
282        def __init__(self, parent, winid):
283          wxWindow.__init__(self, parent, winid)          wxWindow.__init__(self, parent, winid)
284          self.SetBackgroundColour(wxColour(255, 255, 255))          self.SetBackgroundColour(wxColour(255, 255, 255))
285    
# Line 279  class MapCanvas(wxWindow, Publisher): Line 302  class MapCanvas(wxWindow, Publisher):
302          # if the mouse is outside the window.          # if the mouse is outside the window.
303          self.current_position = None          self.current_position = None
304    
         # If true, OnIdle will call do_redraw to do the actual  
         # redrawing. Set by OnPaint to avoid some unnecessary redraws.  
         # To force a redraw call full_redraw().  
         self.redraw_on_idle = 0  
   
         # The region to update when idle  
         self.update_region = wx.wxRegion()  
   
305          # the bitmap serving as backing store          # the bitmap serving as backing store
306          self.bitmap = None          self.bitmap = None
307    
308          # the interactor          # the selection
309          self.interactor = interactor          self.selection = Selection()
310          self.interactor.Subscribe(SELECTED_SHAPE, self.shape_selected)          self.selection.Subscribe(SHAPES_SELECTED , self.shape_selected)
311    
312          # keep track of which layers/shapes are selected to make sure we          # keep track of which layers/shapes are selected to make sure we
313          # only redraw when necessary          # only redraw when necessary
# Line 306  class MapCanvas(wxWindow, Publisher): Line 321  class MapCanvas(wxWindow, Publisher):
321          EVT_MOTION(self, self.OnMotion)          EVT_MOTION(self, self.OnMotion)
322          EVT_LEAVE_WINDOW(self, self.OnLeaveWindow)          EVT_LEAVE_WINDOW(self, self.OnLeaveWindow)
323          wx.EVT_SIZE(self, self.OnSize)          wx.EVT_SIZE(self, self.OnSize)
         wx.EVT_IDLE(self, self.OnIdle)  
324    
325      def __del__(self):      def __del__(self):
326          wxWindow.__del__(self)          wxWindow.__del__(self)
327          Publisher.__del__(self)          Publisher.__del__(self)
328    
329        def Subscribe(self, channel, *args):
330            """Extend the inherited method to handle delegated messages.
331    
332            If channel is one of the delegated messages call the appropriate
333            object's Subscribe method. Otherwise just call the inherited
334            method.
335            """
336            if channel in self.delegated_messages:
337                object = getattr(self, self.delegated_messages[channel])
338                object.Subscribe(channel, *args)
339            else:
340                Publisher.Subscribe(self, channel, *args)
341    
342        def Unsubscribe(self, channel, *args):
343            """Extend the inherited method to handle delegated messages.
344    
345            If channel is one of the delegated messages call the appropriate
346            object's Unsubscribe method. Otherwise just call the inherited
347            method.
348            """
349            if channel in self.delegated_messages:
350                object = getattr(self, self.delegated_messages[channel])
351                object.Unsubscribe(channel, *args)
352            else:
353                Publisher.Unsubscribe(self, channel, *args)
354    
355        def __getattr__(self, attr):
356            if attr in self.delegated_methods:
357                return getattr(getattr(self, self.delegated_methods[attr]), attr)
358            raise AttributeError(attr)
359    
360      def OnPaint(self, event):      def OnPaint(self, event):
361          dc = wxPaintDC(self)          dc = wxPaintDC(self)
362          if self.map is not None and self.map.HasLayers():          clear = self.map is None or not self.map.HasLayers()
363              # We have a non-empty map. Redraw it in idle time  
364              self.redraw_on_idle = 1          #wxBeginBusyCursor()
365              # update the region that has to be redrawn  
366              self.update_region.UnionRegion(self.GetUpdateRegion())          if not clear:
367          else:              try:
368                    self.do_redraw()
369                except:
370                    print "Error during drawing:", sys.exc_info()[0]
371                    clear = True
372    
373            if clear:
374              # 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
375              # the screen.              # the screen.
376    
# Line 331  class MapCanvas(wxWindow, Publisher): Line 382  class MapCanvas(wxWindow, Publisher):
382              dc.Clear()              dc.Clear()
383              dc.EndDrawing()              dc.EndDrawing()
384    
385              # clear the region          #wxEndBusyCursor()
             self.update_region = wx.wxRegion()  
386    
387      def do_redraw(self):      def do_redraw(self):
388          # This should only be called if we have a non-empty map.          # This should only be called if we have a non-empty map.
389    
         # get the update region and reset it. We're not actually using  
         # it anymore, though.  
         update_box = self.update_region.GetBox()  
         self.update_region = wx.wxRegion()  
   
390          # Get the window size.          # Get the window size.
391          width, height = self.GetSizeTuple()          width, height = self.GetSizeTuple()
392    
# Line 355  class MapCanvas(wxWindow, Publisher): Line 400  class MapCanvas(wxWindow, Publisher):
400              dc.BeginDrawing()              dc.BeginDrawing()
401    
402              # clear the background              # clear the background
403              dc.SetBrush(wx.wxWHITE_BRUSH)              #dc.SetBrush(wx.wxWHITE_BRUSH)
404              dc.SetPen(wx.wxTRANSPARENT_PEN)              #dc.SetPen(wx.wxTRANSPARENT_PEN)
405              dc.DrawRectangle(0, 0, width, height)              #dc.DrawRectangle(0, 0, width, height)
406                dc.SetBackground(wx.wxWHITE_BRUSH)
407              if 1: #self.interactor.selected_map is self.map:              dc.Clear()
408                  selected_layer = self.interactor.selected_layer  
409                  selected_shape = self.interactor.selected_shape              selected_layer = self.selection.SelectedLayer()
410              else:              selected_shapes = self.selection.SelectedShapes()
                 selected_layer = None  
                 selected_shape = None  
411    
412              # draw the map into the bitmap              # draw the map into the bitmap
413              renderer = ScreenRenderer(dc, self.scale, self.offset)              renderer = ScreenRenderer(dc, self.scale, self.offset)
414    
415              # Pass the entire bitmap as update_region to the renderer.              # Pass the entire bitmap as update region to the renderer.
416              # We're redrawing the whole bitmap, after all.              # We're redrawing the whole bitmap, after all.
417              renderer.RenderMap(self.map, (0, 0, width, height),              renderer.RenderMap(self.map, (0, 0, width, height),
418                                 selected_layer, selected_shape)                                 selected_layer, selected_shapes)
419    
420              dc.EndDrawing()              dc.EndDrawing()
421              dc.SelectObject(wx.wxNullBitmap)              dc.SelectObject(wx.wxNullBitmap)
# Line 393  class MapCanvas(wxWindow, Publisher): Line 436  class MapCanvas(wxWindow, Publisher):
436          printout.Destroy()          printout.Destroy()
437    
438      def SetMap(self, map):      def SetMap(self, map):
439          redraw_channels = (LAYERS_CHANGED, LAYER_LEGEND_CHANGED,          redraw_channels = (MAP_LAYERS_CHANGED, LAYER_CHANGED,
440                             LAYER_VISIBILITY_CHANGED)                             LAYER_VISIBILITY_CHANGED)
441          if self.map is not None:          if self.map is not None:
442              for channel in redraw_channels:              for channel in redraw_channels:
# Line 401  class MapCanvas(wxWindow, Publisher): Line 444  class MapCanvas(wxWindow, Publisher):
444              self.map.Unsubscribe(MAP_PROJECTION_CHANGED,              self.map.Unsubscribe(MAP_PROJECTION_CHANGED,
445                                   self.projection_changed)                                   self.projection_changed)
446          self.map = map          self.map = map
447            self.selection.ClearSelection()
448          if self.map is not None:          if self.map is not None:
449              for channel in redraw_channels:              for channel in redraw_channels:
450                  self.map.Subscribe(channel, self.full_redraw)                  self.map.Subscribe(channel, self.full_redraw)
# Line 447  class MapCanvas(wxWindow, Publisher): Line 491  class MapCanvas(wxWindow, Publisher):
491    
492      def FitRectToWindow(self, rect):      def FitRectToWindow(self, rect):
493          """Fit the rectangular region given by rect into the window.          """Fit the rectangular region given by rect into the window.
494            
495          Set scale so that rect (in projected coordinates) just fits into          Set scale so that rect (in projected coordinates) just fits into
496          the window and center it.          the window and center it.
497          """          """
498          width, height = self.GetSizeTuple()          width, height = self.GetSizeTuple()
499          llx, lly, urx, ury = rect          llx, lly, urx, ury = rect
500          if llx == urx or lly == ury:          if llx == urx or lly == ury:
501              # zero with or zero height. Do Nothing              # zero width or zero height. Do Nothing
502              return              return
503          scalex = width / (urx - llx)          scalex = width / (urx - llx)
504          scaley = height / (ury - lly)          scaley = height / (ury - lly)
# Line 465  class MapCanvas(wxWindow, Publisher): Line 509  class MapCanvas(wxWindow, Publisher):
509    
510      def FitMapToWindow(self):      def FitMapToWindow(self):
511          """Fit the map to the window          """Fit the map to the window
512            
513          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
514          center it in the window.          center it in the window.
515          """          """
# Line 473  class MapCanvas(wxWindow, Publisher): Line 517  class MapCanvas(wxWindow, Publisher):
517          if bbox is not None:          if bbox is not None:
518              self.FitRectToWindow(bbox)              self.FitRectToWindow(bbox)
519    
520        def FitLayerToWindow(self, layer):
521            """Fit the given layer to the window.
522    
523            Set the scale so that the layer fits exactly into the window and
524            center it in the window.
525            """
526            
527            bbox = layer.LatLongBoundingBox()
528            if bbox is not None:
529                proj = self.map.GetProjection()
530                if proj is not None:
531                    bbox = proj.ForwardBBox(bbox)
532                    if bbox is not None:
533                        self.FitRectToWindow(bbox)
534    
535      def ZoomFactor(self, factor, center = None):      def ZoomFactor(self, factor, center = None):
536          """Multiply the zoom by factor and center on center.          """Multiply the zoom by factor and center on center.
537    
# Line 517  class MapCanvas(wxWindow, Publisher): Line 576  class MapCanvas(wxWindow, Publisher):
576          offx, offy = self.offset          offx, offy = self.offset
577          self.set_view_transform(self.scale, (offx + dx, offy + dy))          self.set_view_transform(self.scale, (offx + dx, offy + dy))
578    
579        def SelectTool(self, tool):
580            """Make tool the active tool.
581    
582            The parameter should be an instance of Tool or None to indicate
583            that no tool is active.
584            """
585            self.tool = tool
586    
587      def ZoomInTool(self):      def ZoomInTool(self):
588          """Start the zoom in tool"""          """Start the zoom in tool"""
589          self.tool = ZoomInTool(self)          self.SelectTool(ZoomInTool(self))
590    
591      def ZoomOutTool(self):      def ZoomOutTool(self):
592          """Start the zoom out tool"""          """Start the zoom out tool"""
593          self.tool = ZoomOutTool(self)          self.SelectTool(ZoomOutTool(self))
594    
595      def PanTool(self):      def PanTool(self):
596          """Start the pan tool"""          """Start the pan tool"""
597          self.tool = PanTool(self)          self.SelectTool(PanTool(self))
598            #img = resource.GetImageResource("pan", wxBITMAP_TYPE_XPM)
599            #bmp = resource.GetBitmapResource("pan", wxBITMAP_TYPE_XPM)
600            #print bmp
601            #img = wxImageFromBitmap(bmp)
602            #print img
603            #cur = wxCursor(img)
604            #print cur
605            #self.SetCursor(cur)
606    
607      def IdentifyTool(self):      def IdentifyTool(self):
608          """Start the identify tool"""          """Start the identify tool"""
609          self.tool = IdentifyTool(self)          self.SelectTool(IdentifyTool(self))
610    
611      def LabelTool(self):      def LabelTool(self):
612          """Start the label tool"""          """Start the label tool"""
613          self.tool = LabelTool(self)          self.SelectTool(LabelTool(self))
614    
615      def CurrentTool(self):      def CurrentTool(self):
616          """Return the name of the current tool or None if no tool is active"""          """Return the name of the current tool or None if no tool is active"""
# Line 581  class MapCanvas(wxWindow, Publisher): Line 656  class MapCanvas(wxWindow, Publisher):
656          self.set_current_position(event)          self.set_current_position(event)
657          if self.dragging:          if self.dragging:
658              self.ReleaseMouse()              self.ReleaseMouse()
659              self.tool.Hide(self.drag_dc)              try:
660              self.tool.MouseUp(event)                  self.tool.Hide(self.drag_dc)
661              self.drag_dc = None                  self.tool.MouseUp(event)
662          self.dragging = 0              finally:
663                    self.drag_dc = None
664                    self.dragging = 0
665    
666      def OnMotion(self, event):      def OnMotion(self, event):
667          self.set_current_position(event)          self.set_current_position(event)
# Line 596  class MapCanvas(wxWindow, Publisher): Line 673  class MapCanvas(wxWindow, Publisher):
673      def OnLeaveWindow(self, event):      def OnLeaveWindow(self, event):
674          self.set_current_position(None)          self.set_current_position(None)
675    
     def OnIdle(self, event):  
         if self.redraw_on_idle:  
             self.do_redraw()  
         self.redraw_on_idle = 0  
   
676      def OnSize(self, event):      def OnSize(self, event):
677          # the window's size has changed. We have to get a new bitmap. If          # the window's size has changed. We have to get a new bitmap. If
678          # we want to be clever we could try to get by without throwing          # we want to be clever we could try to get by without throwing
# Line 609  class MapCanvas(wxWindow, Publisher): Line 681  class MapCanvas(wxWindow, Publisher):
681          # Even when the window becomes larger some parts of the bitmap          # Even when the window becomes larger some parts of the bitmap
682          # could be reused.          # could be reused.
683          self.full_redraw()          self.full_redraw()
684            pass
685    
686      def shape_selected(self, layer, shape):      def shape_selected(self, layer, shape):
687          """Redraw the map.          """Receiver for the SHAPES_SELECTED messages. Redraw the map."""
688            # The selection object takes care that it only issues
689          Receiver for the SELECTED_SHAPE messages. Try to redraw only          # SHAPES_SELECTED messages when the set of selected shapes has
690          when necessary.          # actually changed, so we can do a full redraw unconditionally.
691          """          # FIXME: We should perhaps try to limit the redraw to the are
692          # A redraw is necessary when the display has to change, which          # actually covered by the shapes before and after the selection
693          # means that either the status changes from having no selection          # change.
694          # 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  
695    
696      def unprojected_rect_around_point(self, x, y):      def unprojected_rect_around_point(self, x, y, dist):
697          """return a rect a few pixels around (x, y) in unprojected corrdinates          """return a rect dist pixels around (x, y) in unprojected corrdinates
698    
699          The return value is a tuple (minx, miny, maxx, maxy) suitable a          The return value is a tuple (minx, miny, maxx, maxy) suitable a
700          parameter to a layer's ShapesInRegion method.          parameter to a layer's ShapesInRegion method.
# Line 646  class MapCanvas(wxWindow, Publisher): Line 708  class MapCanvas(wxWindow, Publisher):
708          xs = []          xs = []
709          ys = []          ys = []
710          for dx, dy in ((-1, -1), (1, -1), (1, 1), (-1, 1)):          for dx, dy in ((-1, -1), (1, -1), (1, 1), (-1, 1)):
711              px, py = self.win_to_proj(x + dx, y + dy)              px, py = self.win_to_proj(x + dist * dx, y + dist * dy)
712              if inverse:              if inverse:
713                  px, py = inverse(px, py)                  px, py = inverse(px, py)
714              xs.append(px)              xs.append(px)
# Line 675  class MapCanvas(wxWindow, Publisher): Line 737  class MapCanvas(wxWindow, Publisher):
737          scale = self.scale          scale = self.scale
738          offx, offy = self.offset          offx, offy = self.offset
739    
         box = self.unprojected_rect_around_point(px, py)  
   
740          if select_labels:          if select_labels:
741              labels = self.map.LabelLayer().Labels()              labels = self.map.LabelLayer().Labels()
742    
# Line 723  class MapCanvas(wxWindow, Publisher): Line 783  class MapCanvas(wxWindow, Publisher):
783              if not layer.Visible():              if not layer.Visible():
784                  continue                  continue
785    
786              filled = layer.fill is not None              filled = layer.GetClassification().GetDefaultFill() \
787              stroked = layer.stroke is not None                       is not Color.Transparent
788                stroked = layer.GetClassification().GetDefaultLineColor() \
789                          is not Color.Transparent
790    
791              layer_proj = layer.projection              layer_proj = layer.projection
792              if layer_proj is not None:              if layer_proj is not None:
# Line 736  class MapCanvas(wxWindow, Publisher): Line 798  class MapCanvas(wxWindow, Publisher):
798    
799              select_shape = -1              select_shape = -1
800    
801                # Determine the ids of the shapes that overlap a tiny area
802                # around the point. For layers containing points we have to
803                # choose a larger size of the box we're testing agains so
804                # that we take the size of the markers into account
805                # FIXME: Once the markers are more flexible this part has to
806                # become more flexible too, of course
807                if shapetype == SHAPETYPE_POINT:
808                    box = self.unprojected_rect_around_point(px, py, 5)
809                else:
810                    box = self.unprojected_rect_around_point(px, py, 1)
811              shape_ids = layer.ShapesInRegion(box)              shape_ids = layer.ShapesInRegion(box)
812              shape_ids.reverse()              shape_ids.reverse()
813    
# Line 793  class MapCanvas(wxWindow, Publisher): Line 865  class MapCanvas(wxWindow, Publisher):
865          # to deselect the currently selected layer, so we simply select          # to deselect the currently selected layer, so we simply select
866          # the already selected layer again.          # the already selected layer again.
867          if layer is None:          if layer is None:
868              layer = self.interactor.SelectedLayer()              layer = self.selection.SelectedLayer()
869          self.interactor.SelectLayerAndShape(layer, shape)              shapes = []
870            else:
871                shapes = [shape]
872            self.selection.SelectShapes(layer, shapes)
873          return result          return result
874    
875      def LabelShapeAt(self, x, y):      def LabelShapeAt(self, x, y):
Legend:
Removed from v.295  
changed lines
  Added in v.822
[email protected]
 ViewVC Help
Powered by ViewVC 1.1.26