/[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 149 by bh, Tue May 7 16:41:07 2002 UTC revision 824 by jonathan, Tue May 6 08:09:27 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 129  class ZoomInTool(RectTool): Line 137  class ZoomInTool(RectTool):
137              Tool.MouseUp(self, event)              Tool.MouseUp(self, event)
138              sx, sy = self.start              sx, sy = self.start
139              cx, cy = self.current              cx, cy = self.current
140              if sx == cx and sy == cy:              if sx == cx or sy == cy:
141                  # Just a mouse click. Simply zoom in by a factor of two                  # Just a mouse click or a degenerate rectangle. Simply
142                    # zoom in by a factor of two
143                    # FIXME: For a click this is the desired behavior but should we
144                    # really do this for degenrate rectagles as well or
145                    # should we ignore them?
146                  self.view.ZoomFactor(2, center = (cx, cy))                  self.view.ZoomFactor(2, center = (cx, cy))
147              else:              else:
148                  # A drag. Zoom in to the rectangle                  # A drag. Zoom in to the rectangle
# Line 140  class ZoomInTool(RectTool): Line 152  class ZoomInTool(RectTool):
152  class ZoomOutTool(RectTool):  class ZoomOutTool(RectTool):
153    
154      """The Zoom-Out Tool"""      """The Zoom-Out Tool"""
155        
156      def Name(self):      def Name(self):
157          return "ZoomOutTool"          return "ZoomOutTool"
158    
# Line 149  class ZoomOutTool(RectTool): Line 161  class ZoomOutTool(RectTool):
161              Tool.MouseUp(self, event)              Tool.MouseUp(self, event)
162              sx, sy = self.start              sx, sy = self.start
163              cx, cy = self.current              cx, cy = self.current
164              if sx == cx and sy == cy:              if sx == cx or sy == cy:
165                  # Just a mouse click. Simply zoom out by a factor of two                  # Just a mouse click or a degenerate rectangle. Simply
166                  self.view.ZoomFactor(0.5, center = (cy, cy))                  # zoom out by a factor of two.
167                    # FIXME: For a click this is the desired behavior but should we
168                    # really do this for degenrate rectagles as well or
169                    # should we ignore them?
170                    self.view.ZoomFactor(0.5, center = (cx, cy))
171              else:              else:
172                  # A drag. Zoom out to the rectangle                  # A drag. Zoom out to the rectangle
173                  self.view.ZoomOutToRect((min(sx, cx), min(sy, cy),                  self.view.ZoomOutToRect((min(sx, cx), min(sy, cy),
# Line 167  class PanTool(Tool): Line 183  class PanTool(Tool):
183    
184      def MouseMove(self, event):      def MouseMove(self, event):
185          if self.dragging:          if self.dragging:
             x0, y0 = self.current  
186              Tool.MouseMove(self, event)              Tool.MouseMove(self, event)
187                sx, sy = self.start
188              x, y = self.current              x, y = self.current
189              width, height = self.view.GetSizeTuple()              width, height = self.view.GetSizeTuple()
190    
191                bitmapdc = wx.wxMemoryDC()
192                bitmapdc.SelectObject(self.view.bitmap)
193    
194              dc = self.view.drag_dc              dc = self.view.drag_dc
195              dc.Blit(0, 0, width, height, dc, x0 - x, y0 - y)              dc.Blit(0, 0, width, height, bitmapdc, sx - x, sy - y)
196    
197      def MouseUp(self, event):      def MouseUp(self, event):
198          if self.dragging:          if self.dragging:
# Line 180  class PanTool(Tool): Line 200  class PanTool(Tool):
200              sx, sy = self.start              sx, sy = self.start
201              cx, cy = self.current              cx, cy = self.current
202              self.view.Translate(cx - sx, cy - sy)              self.view.Translate(cx - sx, cy - sy)
203            
204  class IdentifyTool(Tool):  class IdentifyTool(Tool):
205    
206      """The "Identify" Tool"""      """The "Identify" Tool"""
207        
208      def Name(self):      def Name(self):
209          return "IdentifyTool"          return "IdentifyTool"
210    
# Line 238  class MapPrintout(wx.wxPrintout): Line 258  class MapPrintout(wx.wxPrintout):
258          renderer = PrinterRender(dc, scale, (offx, offy), resolution = resx)          renderer = PrinterRender(dc, scale, (offx, offy), resolution = resx)
259          renderer.RenderMap(self.map)          renderer.RenderMap(self.map)
260          return wx.true          return wx.true
261            
262    
263  class MapCanvas(wxWindow, Publisher):  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 267  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
313            # only redraw when necessary
314            self.last_selected_layer = None
315            self.last_selected_shape = None
316    
317          # subscribe the WX events we're interested in          # subscribe the WX events we're interested in
318          EVT_PAINT(self, self.OnPaint)          EVT_PAINT(self, self.OnPaint)
# Line 289  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                
377              # XXX it's probably possible to get rid of this. The              # XXX it's probably possible to get rid of this. The
378              # background color of the window is already white and the              # background color of the window is already white and the
379              # only thing we may have to do is to call self.Refresh()              # only thing we may have to do is to call self.Refresh()
380              # with a true argument in the right places.              # with a true argument in the right places.
381              dc.BeginDrawing()              dc.BeginDrawing()
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 338  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 374  class MapCanvas(wxWindow, Publisher): Line 434  class MapCanvas(wxWindow, Publisher):
434          printout = MapPrintout(self.map)          printout = MapPrintout(self.map)
435          printer.Print(self, printout, wx.true)          printer.Print(self, printout, wx.true)
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 384  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 395  class MapCanvas(wxWindow, Publisher): Line 456  class MapCanvas(wxWindow, Publisher):
456          self.full_redraw()          self.full_redraw()
457    
458      def Map(self):      def Map(self):
459            """Return the map displayed by this canvas"""
460          return self.map          return self.map
461    
462      def redraw(self, *args):      def redraw(self, *args):
# Line 428  class MapCanvas(wxWindow, Publisher): Line 490  class MapCanvas(wxWindow, Publisher):
490          return ((x - offx) / self.scale, (offy - y) / self.scale)          return ((x - offx) / self.scale, (offy - y) / self.scale)
491    
492      def FitRectToWindow(self, rect):      def FitRectToWindow(self, rect):
493            """Fit the rectangular region given by rect into the window.
494    
495            Set scale so that rect (in projected coordinates) just fits into
496            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 441  class MapCanvas(wxWindow, Publisher): Line 508  class MapCanvas(wxWindow, Publisher):
508          self.set_view_transform(scale, (offx, offy))          self.set_view_transform(scale, (offx, offy))
509    
510      def FitMapToWindow(self):      def FitMapToWindow(self):
511          """\          """Fit the map to the window
512          Set the scale and offset so that the map is centered in the  
513          window          Set the scale so that the map fits exactly into the window and
514            center it in the window.
515          """          """
516          bbox = self.map.ProjectedBoundingBox()          bbox = self.map.ProjectedBoundingBox()
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    
533                if bbox is not None:
534                    self.FitRectToWindow(bbox)
535    
536      def ZoomFactor(self, factor, center = None):      def ZoomFactor(self, factor, center = None):
537          """Multiply the zoom by factor and center on center.          """Multiply the zoom by factor and center on center.
538    
# Line 469  class MapCanvas(wxWindow, Publisher): Line 553  class MapCanvas(wxWindow, Publisher):
553          self.set_view_transform(scale, offset)          self.set_view_transform(scale, offset)
554    
555      def ZoomOutToRect(self, rect):      def ZoomOutToRect(self, rect):
556          # rect is given in window coordinates          """Zoom out to fit the currently visible region into rect.
557    
558            The rect parameter is given in window coordinates
559            """
560          # determine the bbox of the displayed region in projected          # determine the bbox of the displayed region in projected
561          # coordinates          # coordinates
562          width, height = self.GetSizeTuple()          width, height = self.GetSizeTuple()
# Line 487  class MapCanvas(wxWindow, Publisher): Line 573  class MapCanvas(wxWindow, Publisher):
573          self.set_view_transform(scale, (offx, offy))          self.set_view_transform(scale, (offx, offy))
574    
575      def Translate(self, dx, dy):      def Translate(self, dx, dy):
576            """Move the map by dx, dy pixels"""
577          offx, offy = self.offset          offx, offy = self.offset
578          self.set_view_transform(self.scale, (offx + dx, offy + dy))          self.set_view_transform(self.scale, (offx + dx, offy + dy))
579    
580        def SelectTool(self, tool):
581            """Make tool the active tool.
582    
583            The parameter should be an instance of Tool or None to indicate
584            that no tool is active.
585            """
586            self.tool = tool
587    
588      def ZoomInTool(self):      def ZoomInTool(self):
589          self.tool = ZoomInTool(self)          """Start the zoom in tool"""
590            self.SelectTool(ZoomInTool(self))
591    
592      def ZoomOutTool(self):      def ZoomOutTool(self):
593          self.tool = ZoomOutTool(self)          """Start the zoom out tool"""
594            self.SelectTool(ZoomOutTool(self))
595    
596      def PanTool(self):      def PanTool(self):
597          self.tool = PanTool(self)          """Start the pan tool"""
598            self.SelectTool(PanTool(self))
599            #img = resource.GetImageResource("pan", wxBITMAP_TYPE_XPM)
600            #bmp = resource.GetBitmapResource("pan", wxBITMAP_TYPE_XPM)
601            #print bmp
602            #img = wxImageFromBitmap(bmp)
603            #print img
604            #cur = wxCursor(img)
605            #print cur
606            #self.SetCursor(cur)
607    
608      def IdentifyTool(self):      def IdentifyTool(self):
609          self.tool = IdentifyTool(self)          """Start the identify tool"""
610            self.SelectTool(IdentifyTool(self))
611    
612      def LabelTool(self):      def LabelTool(self):
613          self.tool = LabelTool(self)          """Start the label tool"""
614            self.SelectTool(LabelTool(self))
615    
616      def CurrentTool(self):      def CurrentTool(self):
617            """Return the name of the current tool or None if no tool is active"""
618          return self.tool and self.tool.Name() or None          return self.tool and self.tool.Name() or None
619    
620      def CurrentPosition(self):      def CurrentPosition(self):
# Line 543  class MapCanvas(wxWindow, Publisher): Line 652  class MapCanvas(wxWindow, Publisher):
652              self.tool.MouseDown(event)              self.tool.MouseDown(event)
653              self.tool.Show(self.drag_dc)              self.tool.Show(self.drag_dc)
654              self.dragging = 1              self.dragging = 1
655            
656      def OnLeftUp(self, event):      def OnLeftUp(self, event):
         self.ReleaseMouse()  
657          self.set_current_position(event)          self.set_current_position(event)
658          if self.dragging:          if self.dragging:
659              self.tool.Hide(self.drag_dc)              self.ReleaseMouse()
660              self.tool.MouseUp(event)              try:
661              self.drag_dc = None                  self.tool.Hide(self.drag_dc)
662          self.dragging = 0                  self.tool.MouseUp(event)
663                finally:
664                    self.drag_dc = None
665                    self.dragging = 0
666    
667      def OnMotion(self, event):      def OnMotion(self, event):
668          self.set_current_position(event)          self.set_current_position(event)
# Line 563  class MapCanvas(wxWindow, Publisher): Line 674  class MapCanvas(wxWindow, Publisher):
674      def OnLeaveWindow(self, event):      def OnLeaveWindow(self, event):
675          self.set_current_position(None)          self.set_current_position(None)
676    
     def OnIdle(self, event):  
         if self.redraw_on_idle:  
             self.do_redraw()  
         self.redraw_on_idle = 0  
   
677      def OnSize(self, event):      def OnSize(self, event):
678          # 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
679          # 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 576  class MapCanvas(wxWindow, Publisher): Line 682  class MapCanvas(wxWindow, Publisher):
682          # Even when the window becomes larger some parts of the bitmap          # Even when the window becomes larger some parts of the bitmap
683          # could be reused.          # could be reused.
684          self.full_redraw()          self.full_redraw()
685            pass
686    
687      def shape_selected(self, layer, shape):      def shape_selected(self, layer, shape):
688            """Receiver for the SHAPES_SELECTED messages. Redraw the map."""
689            # The selection object takes care that it only issues
690            # SHAPES_SELECTED messages when the set of selected shapes has
691            # actually changed, so we can do a full redraw unconditionally.
692            # FIXME: We should perhaps try to limit the redraw to the are
693            # actually covered by the shapes before and after the selection
694            # change.
695          self.full_redraw()          self.full_redraw()
696    
697      def find_shape_at(self, px, py, select_labels = 0, selected_layer = 1):      def unprojected_rect_around_point(self, x, y, dist):
698            """return a rect dist pixels around (x, y) in unprojected corrdinates
699    
700            The return value is a tuple (minx, miny, maxx, maxy) suitable a
701            parameter to a layer's ShapesInRegion method.
702            """
703            map_proj = self.map.projection
704            if map_proj is not None:
705                inverse = map_proj.Inverse
706            else:
707                inverse = None
708    
709            xs = []
710            ys = []
711            for dx, dy in ((-1, -1), (1, -1), (1, 1), (-1, 1)):
712                px, py = self.win_to_proj(x + dist * dx, y + dist * dy)
713                if inverse:
714                    px, py = inverse(px, py)
715                xs.append(px)
716                ys.append(py)
717            return (min(xs), min(ys), max(xs), max(ys))
718    
719        def find_shape_at(self, px, py, select_labels = 0, searched_layer = None):
720          """Determine the shape at point px, py in window coords          """Determine the shape at point px, py in window coords
721    
722          Return the shape and the corresponding layer as a tuple (layer,          Return the shape and the corresponding layer as a tuple (layer,
# Line 590  class MapCanvas(wxWindow, Publisher): Line 726  class MapCanvas(wxWindow, Publisher):
726          search through the labels. If a label is found return it's index          search through the labels. If a label is found return it's index
727          as the shape and None as the layer.          as the shape and None as the layer.
728    
729          If the optional parameter selected_layer is true (default), only          If the optional parameter searched_layer is given (or not None
730          search in the currently selected layer.          which it defaults to), only search in that layer.
731          """          """
732          map_proj = self.map.projection          map_proj = self.map.projection
733          if map_proj is not None:          if map_proj is not None:
# Line 604  class MapCanvas(wxWindow, Publisher): Line 740  class MapCanvas(wxWindow, Publisher):
740    
741          if select_labels:          if select_labels:
742              labels = self.map.LabelLayer().Labels()              labels = self.map.LabelLayer().Labels()
743                
744              if labels:              if labels:
745                  dc = wxClientDC(self)                  dc = wxClientDC(self)
746                  font = wxFont(10, wx.wxSWISS, wx.wxNORMAL, wx.wxNORMAL)                  font = wxFont(10, wx.wxSWISS, wx.wxNORMAL, wx.wxNORMAL)
# Line 636  class MapCanvas(wxWindow, Publisher): Line 772  class MapCanvas(wxWindow, Publisher):
772                      if x <= px < x + width and y <= py <= y + height:                      if x <= px < x + width and y <= py <= y + height:
773                          return None, i                          return None, i
774    
775          if selected_layer:          if searched_layer:
776              layer = self.interactor.SelectedLayer()              layers = [searched_layer]
             if layer is not None:  
                 layers = [layer]  
             else:  
                 # no layer selected. Use an empty list to effectively  
                 # ignore all layers.  
                 layers = []  
777          else:          else:
778              layers = self.map.Layers()              layers = self.map.Layers()
779    
# Line 654  class MapCanvas(wxWindow, Publisher): Line 784  class MapCanvas(wxWindow, Publisher):
784              if not layer.Visible():              if not layer.Visible():
785                  continue                  continue
786    
787              filled = layer.fill is not None              filled = layer.GetClassification().GetDefaultFill() \
788              stroked = layer.stroke is not None                       is not Color.Transparent
789                                stroked = layer.GetClassification().GetDefaultLineColor() \
790                          is not Color.Transparent
791    
792              layer_proj = layer.projection              layer_proj = layer.projection
793              if layer_proj is not None:              if layer_proj is not None:
794                  inverse = layer_proj.Inverse                  inverse = layer_proj.Inverse
795              else:              else:
796                  inverse = None                  inverse = None
797                    
798              shapetype = layer.ShapeType()              shapetype = layer.ShapeType()
799    
800              select_shape = -1              select_shape = -1
801    
802                # Determine the ids of the shapes that overlap a tiny area
803                # around the point. For layers containing points we have to
804                # choose a larger size of the box we're testing agains so
805                # that we take the size of the markers into account
806                # FIXME: Once the markers are more flexible this part has to
807                # become more flexible too, of course
808                if shapetype == SHAPETYPE_POINT:
809                    box = self.unprojected_rect_around_point(px, py, 5)
810                else:
811                    box = self.unprojected_rect_around_point(px, py, 1)
812                shape_ids = layer.ShapesInRegion(box)
813                shape_ids.reverse()
814    
815              if shapetype == SHAPETYPE_POLYGON:              if shapetype == SHAPETYPE_POLYGON:
816                  for i in range(layer.NumShapes() - 1, -1, -1):                  for i in shape_ids:
817                      result = point_in_polygon_shape(layer.shapefile.cobject(),                      result = point_in_polygon_shape(layer.shapefile.cobject(),
818                                                      i,                                                      i,
819                                                      filled, stroked,                                                      filled, stroked,
# Line 678  class MapCanvas(wxWindow, Publisher): Line 824  class MapCanvas(wxWindow, Publisher):
824                          select_shape = i                          select_shape = i
825                          break                          break
826              elif shapetype == SHAPETYPE_ARC:              elif shapetype == SHAPETYPE_ARC:
827                  for i in range(layer.NumShapes() - 1, -1, -1):                  for i in shape_ids:
828                      result = point_in_polygon_shape(layer.shapefile.cobject(),                      result = point_in_polygon_shape(layer.shapefile.cobject(),
829                                                      i, 0, 1,                                                      i, 0, 1,
830                                                      map_proj, layer_proj,                                                      map_proj, layer_proj,
# Line 688  class MapCanvas(wxWindow, Publisher): Line 834  class MapCanvas(wxWindow, Publisher):
834                          select_shape = i                          select_shape = i
835                          break                          break
836              elif shapetype == SHAPETYPE_POINT:              elif shapetype == SHAPETYPE_POINT:
837                  for i in range(layer.NumShapes() - 1, -1, -1):                  for i in shape_ids:
838                      shape = layer.Shape(i)                      shape = layer.Shape(i)
839                      x, y = shape.Points()[0]                      x, y = shape.Points()[0]
840                      if inverse:                      if inverse:
# Line 705  class MapCanvas(wxWindow, Publisher): Line 851  class MapCanvas(wxWindow, Publisher):
851                  return layer, select_shape                  return layer, select_shape
852          return None, None          return None, None
853    
854      def SelectShapeAt(self, x, y):      def SelectShapeAt(self, x, y, layer = None):
855          layer, shape = self.find_shape_at(x, y, selected_layer = 0)          """\
856            Select and return the shape and its layer at window position (x, y)
857    
858            If layer is given, only search in that layer. If no layer is
859            given, search through all layers.
860    
861            Return a tuple (layer, shapeid). If no shape is found, return
862            (None, None).
863            """
864            layer, shape = result = self.find_shape_at(x, y, searched_layer=layer)
865          # If layer is None, then shape will also be None. We don't want          # If layer is None, then shape will also be None. We don't want
866          # to deselect the currently selected layer, so we simply select          # to deselect the currently selected layer, so we simply select
867          # the already selected layer again.          # the already selected layer again.
868          if layer is None:          if layer is None:
869              layer = self.interactor.SelectedLayer()              layer = self.selection.SelectedLayer()
870          self.interactor.SelectLayerAndShape(layer, shape)              shapes = []
871            else:
872                shapes = [shape]
873            self.selection.SelectShapes(layer, shapes)
874            return result
875    
876      def LabelShapeAt(self, x, y):      def LabelShapeAt(self, x, y):
877            """Add or remove a label at window position x, y.
878    
879            If there's a label at the given position, remove it. Otherwise
880            determine the shape at the position, run the label dialog and
881            unless the user cancels the dialog, add a laber.
882            """
883          ox = x; oy = y          ox = x; oy = y
884          label_layer = self.map.LabelLayer()          label_layer = self.map.LabelLayer()
885          layer, shape_index = self.find_shape_at(x, y, select_labels = 1)          layer, shape_index = self.find_shape_at(x, y, select_labels = 1)
Legend:
Removed from v.149  
changed lines
  Added in v.824
[email protected]
 ViewVC Help
Powered by ViewVC 1.1.26