/[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 45 by bh, Fri Sep 7 15:00:21 2001 UTC revision 831 by jonathan, Tue May 6 12:07:21 2003 UTC
# Line 1  Line 1 
1  # Copyright (c) 2001 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_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
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  from messages import LAYER_SELECTED, SHAPES_SELECTED, VIEW_POSITION
47    
48    
49  #  #
# Line 127  class ZoomInTool(RectTool): Line 135  class ZoomInTool(RectTool):
135      def MouseUp(self, event):      def MouseUp(self, event):
136          if self.dragging:          if self.dragging:
137              Tool.MouseUp(self, event)              Tool.MouseUp(self, event)
138              self.view.FitRectToWindow(self.proj_rect())              sx, sy = self.start
139                cx, cy = self.current
140                if sx == cx or sy == cy:
141                    # 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))
147                else:
148                    # A drag. Zoom in to the rectangle
149                    self.view.FitRectToWindow(self.proj_rect())
150    
151    
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 142  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              self.view.ZoomOutToRect((min(sx, cx), min(sy, cy),              if sx == cx or sy == cy:
165                                       max(sx, cx), max(sy, cy)))                  # Just a mouse click or a degenerate rectangle. Simply
166                    # 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:
172                    # A drag. Zoom out to the rectangle
173                    self.view.ZoomOutToRect((min(sx, cx), min(sy, cy),
174                                             max(sx, cx), max(sy, cy)))
175    
176    
177  class PanTool(Tool):  class PanTool(Tool):
# Line 155  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 168  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 226  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  class MapCanvas(wxWindow):  
263    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                             "HasSelectedShapes": "selection"}
282    
283        def __init__(self, parent, winid):
284          wxWindow.__init__(self, parent, winid)          wxWindow.__init__(self, parent, winid)
285          self.SetBackgroundColour(wxColour(255, 255, 255))          self.SetBackgroundColour(wxColour(255, 255, 255))
286    
287            # the map displayed in this canvas. Set with SetMap()
288          self.map = None          self.map = None
289    
290            # scale and offset describe the transformation from projected
291            # coordinates to window coordinates.
292          self.scale = 1.0          self.scale = 1.0
293          self.offset = (0, 0)          self.offset = (0, 0)
294    
295            # whether the user is currently dragging the mouse, i.e. moving
296            # the mouse while pressing a mouse button
297          self.dragging = 0          self.dragging = 0
298    
299            # the currently active tool
300          self.tool = None          self.tool = None
301          self.redraw_on_idle = 0  
302            # The current mouse position of the last OnMotion event or None
303            # if the mouse is outside the window.
304            self.current_position = None
305    
306            # the bitmap serving as backing store
307            self.bitmap = None
308    
309            # the selection
310            self.selection = Selection()
311            self.selection.Subscribe(SHAPES_SELECTED , self.shape_selected)
312    
313            # keep track of which layers/shapes are selected to make sure we
314            # only redraw when necessary
315            self.last_selected_layer = None
316            self.last_selected_shape = None
317    
318            # subscribe the WX events we're interested in
319          EVT_PAINT(self, self.OnPaint)          EVT_PAINT(self, self.OnPaint)
320          EVT_LEFT_DOWN(self, self.OnLeftDown)          EVT_LEFT_DOWN(self, self.OnLeftDown)
321          EVT_LEFT_UP(self, self.OnLeftUp)          EVT_LEFT_UP(self, self.OnLeftUp)
322          EVT_MOTION(self, self.OnMotion)          EVT_MOTION(self, self.OnMotion)
323          wx.EVT_IDLE(self, self.OnIdle)          EVT_LEAVE_WINDOW(self, self.OnLeaveWindow)
324          self.interactor = interactor          wx.EVT_SIZE(self, self.OnSize)
325          self.interactor.Subscribe(SELECTED_SHAPE, self.shape_selected)  
326        def __del__(self):
327            wxWindow.__del__(self)
328            Publisher.__del__(self)
329    
330        def Subscribe(self, channel, *args):
331            """Extend the inherited method to handle delegated messages.
332    
333            If channel is one of the delegated messages call the appropriate
334            object's Subscribe method. Otherwise just call the inherited
335            method.
336            """
337            if channel in self.delegated_messages:
338                object = getattr(self, self.delegated_messages[channel])
339                object.Subscribe(channel, *args)
340            else:
341                Publisher.Subscribe(self, channel, *args)
342    
343        def Unsubscribe(self, channel, *args):
344            """Extend the inherited method to handle delegated messages.
345    
346            If channel is one of the delegated messages call the appropriate
347            object's Unsubscribe method. Otherwise just call the inherited
348            method.
349            """
350            if channel in self.delegated_messages:
351                object = getattr(self, self.delegated_messages[channel])
352                object.Unsubscribe(channel, *args)
353            else:
354                Publisher.Unsubscribe(self, channel, *args)
355    
356        def __getattr__(self, attr):
357            if attr in self.delegated_methods:
358                return getattr(getattr(self, self.delegated_methods[attr]), attr)
359            raise AttributeError(attr)
360    
361      def OnPaint(self, event):      def OnPaint(self, event):
362          dc = wxPaintDC(self)          dc = wxPaintDC(self)
363          if self.map is None or not self.map.HasLayers():          clear = self.map is None or not self.map.HasLayers()
             return  
         self.redraw_on_idle = 1  
364    
365      def do_redraw(self):          #wxBeginBusyCursor()
         width, height = self.GetSizeTuple()  
         bitmap = wx.wxEmptyBitmap(width, height)  
366    
367          dc = wx.wxMemoryDC()          if not clear:
368          dc.SelectObject(bitmap)              try:
369                    self.do_redraw()
370                except:
371                    print "Error during drawing:", sys.exc_info()[0]
372                    clear = True
373    
374            if clear:
375                # If we've got no map or if the map is empty, simply clear
376                # the screen.
377    
378                # XXX it's probably possible to get rid of this. The
379                # background color of the window is already white and the
380                # only thing we may have to do is to call self.Refresh()
381                # with a true argument in the right places.
382                dc.BeginDrawing()
383                dc.Clear()
384                dc.EndDrawing()
385    
386            #wxEndBusyCursor()
387    
388          dc.BeginDrawing()      def do_redraw(self):
389            # This should only be called if we have a non-empty map.
390    
391            # Get the window size.
392            width, height = self.GetSizeTuple()
393    
394          dc.SetBrush(wx.wxWHITE_BRUSH)          # If self.bitmap's still there, reuse it. Otherwise redraw it
395          dc.SetPen(wx.wxTRANSPARENT_PEN)          if self.bitmap is not None:
396          dc.DrawRectangle(0, 0, width, height)              bitmap = self.bitmap
   
         if 1: #self.interactor.selected_map is self.map:  
             selected_layer = self.interactor.selected_layer  
             selected_shape = self.interactor.selected_shape  
397          else:          else:
398              selected_layer = None              bitmap = wx.wxEmptyBitmap(width, height)
399              selected_shape = None              dc = wx.wxMemoryDC()
400                            dc.SelectObject(bitmap)
401          renderer = ScreenRenderer(dc, self.scale, self.offset)              dc.BeginDrawing()
402          renderer.RenderMap(self.map, selected_layer, selected_shape)  
403                # clear the background
404                #dc.SetBrush(wx.wxWHITE_BRUSH)
405                #dc.SetPen(wx.wxTRANSPARENT_PEN)
406                #dc.DrawRectangle(0, 0, width, height)
407                dc.SetBackground(wx.wxWHITE_BRUSH)
408                dc.Clear()
409    
410                selected_layer = self.selection.SelectedLayer()
411                selected_shapes = self.selection.SelectedShapes()
412    
413                # draw the map into the bitmap
414                renderer = ScreenRenderer(dc, self.scale, self.offset)
415    
416                # Pass the entire bitmap as update region to the renderer.
417                # We're redrawing the whole bitmap, after all.
418                renderer.RenderMap(self.map, (0, 0, width, height),
419                                   selected_layer, selected_shapes)
420    
421                dc.EndDrawing()
422                dc.SelectObject(wx.wxNullBitmap)
423                self.bitmap = bitmap
424    
425            # blit the bitmap to the screen
426            dc = wx.wxMemoryDC()
427            dc.SelectObject(bitmap)
428          clientdc = wxClientDC(self)          clientdc = wxClientDC(self)
429          clientdc.BeginDrawing()          clientdc.BeginDrawing()
430          clientdc.Blit(0, 0, width, height, dc, 0, 0)          clientdc.Blit(0, 0, width, height, dc, 0, 0)
431          clientdc.EndDrawing()          clientdc.EndDrawing()
432    
   
433      def Print(self):      def Print(self):
434          printer = wx.wxPrinter()          printer = wx.wxPrinter()
435          printout = MapPrintout(self.map)          printout = MapPrintout(self.map)
436          printer.Print(self, printout, wx.true)          printer.Print(self, printout, wx.true)
437          printout.Destroy()          printout.Destroy()
438            
439      def SetMap(self, map):      def SetMap(self, map):
440          redraw_channels = (LAYERS_CHANGED, LAYER_LEGEND_CHANGED,          redraw_channels = (MAP_LAYERS_CHANGED, LAYER_CHANGED,
441                             LAYER_VISIBILITY_CHANGED)                             LAYER_VISIBILITY_CHANGED)
442          if self.map is not None:          if self.map is not None:
443              for channel in redraw_channels:              for channel in redraw_channels:
444                  self.map.Unsubscribe(channel, self.redraw)                  self.map.Unsubscribe(channel, self.full_redraw)
445              self.map.Unsubscribe(MAP_PROJECTION_CHANGED,              self.map.Unsubscribe(MAP_PROJECTION_CHANGED,
446                                   self.projection_changed)                                   self.projection_changed)
447          self.map = map          self.map = map
448            self.selection.ClearSelection()
449          if self.map is not None:          if self.map is not None:
450              for channel in redraw_channels:              for channel in redraw_channels:
451                  self.map.Subscribe(channel, self.redraw)                  self.map.Subscribe(channel, self.full_redraw)
452              self.map.Subscribe(MAP_PROJECTION_CHANGED, self.projection_changed)              self.map.Subscribe(MAP_PROJECTION_CHANGED, self.projection_changed)
453          self.FitMapToWindow()          self.FitMapToWindow()
454            # force a redraw. If map is not empty, it's already been called
455            # by FitMapToWindow but if map is empty it hasn't been called
456            # yet so we have to explicitly call it.
457            self.full_redraw()
458    
459      def Map(self):      def Map(self):
460            """Return the map displayed by this canvas"""
461          return self.map          return self.map
462    
463      def redraw(self, *args):      def redraw(self, *args):
464          self.Refresh(0)          self.Refresh(0)
465    
466        def full_redraw(self, *args):
467            self.bitmap = None
468            self.redraw()
469    
470      def projection_changed(self, *args):      def projection_changed(self, *args):
471          self.FitMapToWindow()          self.FitMapToWindow()
472          self.redraw()          self.full_redraw()
473    
474      def set_view_transform(self, scale, offset):      def set_view_transform(self, scale, offset):
475          self.scale = scale          self.scale = scale
476          self.offset = offset          self.offset = offset
477          self.redraw()          self.full_redraw()
478    
479      def proj_to_win(self, x, y):      def proj_to_win(self, x, y):
480          """\          """\
# Line 335  class MapCanvas(wxWindow): Line 491  class MapCanvas(wxWindow):
491          return ((x - offx) / self.scale, (offy - y) / self.scale)          return ((x - offx) / self.scale, (offy - y) / self.scale)
492    
493      def FitRectToWindow(self, rect):      def FitRectToWindow(self, rect):
494            """Fit the rectangular region given by rect into the window.
495    
496            Set scale so that rect (in projected coordinates) just fits into
497            the window and center it.
498            """
499          width, height = self.GetSizeTuple()          width, height = self.GetSizeTuple()
500          llx, lly, urx, ury = rect          llx, lly, urx, ury = rect
501          if llx == urx or lly == ury:          if llx == urx or lly == ury:
502              # zero with or zero height. Do Nothing              # zero width or zero height. Do Nothing
503              return              return
504          scalex = width / (urx - llx)          scalex = width / (urx - llx)
505          scaley = height / (ury - lly)          scaley = height / (ury - lly)
# Line 348  class MapCanvas(wxWindow): Line 509  class MapCanvas(wxWindow):
509          self.set_view_transform(scale, (offx, offy))          self.set_view_transform(scale, (offx, offy))
510    
511      def FitMapToWindow(self):      def FitMapToWindow(self):
512          """\          """Fit the map to the window
513          Set the scale and offset so that the map is centered in the  
514          window          Set the scale so that the map fits exactly into the window and
515            center it in the window.
516          """          """
517          bbox = self.map.ProjectedBoundingBox()          bbox = self.map.ProjectedBoundingBox()
518          if bbox is not None:          if bbox is not None:
519              self.FitRectToWindow(bbox)              self.FitRectToWindow(bbox)
520    
521      def ZoomFactor(self, factor):      def FitLayerToWindow(self, layer):
522            """Fit the given layer to the window.
523    
524            Set the scale so that the layer fits exactly into the window and
525            center it in the window.
526            """
527            
528            bbox = layer.LatLongBoundingBox()
529            if bbox is not None:
530                proj = self.map.GetProjection()
531                if proj is not None:
532                    bbox = proj.ForwardBBox(bbox)
533    
534                if bbox is not None:
535                    self.FitRectToWindow(bbox)
536    
537        def FitSelectedToWindow(self):
538            layer = self.selection.SelectedLayer()
539            shapes = self.selection.SelectedShapes()
540    
541            bbox = layer.ShapesBoundingBox(shapes)
542            if bbox is not None:
543                proj = self.map.GetProjection()
544                if proj is not None:
545                    bbox = proj.ForwardBBox(bbox)
546    
547                if bbox is not None:
548                    self.FitRectToWindow(bbox)
549    
550        def ZoomFactor(self, factor, center = None):
551            """Multiply the zoom by factor and center on center.
552    
553            The optional parameter center is a point in window coordinates
554            that should be centered. If it is omitted, it defaults to the
555            center of the window
556            """
557          width, height = self.GetSizeTuple()          width, height = self.GetSizeTuple()
558          scale = self.scale * factor          scale = self.scale * factor
559          offx, offy = self.offset          offx, offy = self.offset
560          offset = (factor * (offx - width / 2) + width / 2,          if center is not None:
561                    factor * (offy - height / 2) + height / 2)              cx, cy = center
562            else:
563                cx = width / 2
564                cy = height / 2
565            offset = (factor * (offx - cx) + width / 2,
566                      factor * (offy - cy) + height / 2)
567          self.set_view_transform(scale, offset)          self.set_view_transform(scale, offset)
568    
569      def ZoomOutToRect(self, rect):      def ZoomOutToRect(self, rect):
570          # rect is given in window coordinates          """Zoom out to fit the currently visible region into rect.
571    
572            The rect parameter is given in window coordinates
573            """
574          # determine the bbox of the displayed region in projected          # determine the bbox of the displayed region in projected
575          # coordinates          # coordinates
576          width, height = self.GetSizeTuple()          width, height = self.GetSizeTuple()
# Line 383  class MapCanvas(wxWindow): Line 587  class MapCanvas(wxWindow):
587          self.set_view_transform(scale, (offx, offy))          self.set_view_transform(scale, (offx, offy))
588    
589      def Translate(self, dx, dy):      def Translate(self, dx, dy):
590            """Move the map by dx, dy pixels"""
591          offx, offy = self.offset          offx, offy = self.offset
592          self.set_view_transform(self.scale, (offx + dx, offy + dy))          self.set_view_transform(self.scale, (offx + dx, offy + dy))
593    
594        def SelectTool(self, tool):
595            """Make tool the active tool.
596    
597            The parameter should be an instance of Tool or None to indicate
598            that no tool is active.
599            """
600            self.tool = tool
601    
602      def ZoomInTool(self):      def ZoomInTool(self):
603          self.tool = ZoomInTool(self)          """Start the zoom in tool"""
604            self.SelectTool(ZoomInTool(self))
605    
606      def ZoomOutTool(self):      def ZoomOutTool(self):
607          self.tool = ZoomOutTool(self)          """Start the zoom out tool"""
608            self.SelectTool(ZoomOutTool(self))
609    
610      def PanTool(self):      def PanTool(self):
611          self.tool = PanTool(self)          """Start the pan tool"""
612            self.SelectTool(PanTool(self))
613            #img = resource.GetImageResource("pan", wxBITMAP_TYPE_XPM)
614            #bmp = resource.GetBitmapResource("pan", wxBITMAP_TYPE_XPM)
615            #print bmp
616            #img = wxImageFromBitmap(bmp)
617            #print img
618            #cur = wxCursor(img)
619            #print cur
620            #self.SetCursor(cur)
621    
622      def IdentifyTool(self):      def IdentifyTool(self):
623          self.tool = IdentifyTool(self)          """Start the identify tool"""
624            self.SelectTool(IdentifyTool(self))
625    
626      def LabelTool(self):      def LabelTool(self):
627          self.tool = LabelTool(self)          """Start the label tool"""
628            self.SelectTool(LabelTool(self))
629    
630      def CurrentTool(self):      def CurrentTool(self):
631            """Return the name of the current tool or None if no tool is active"""
632          return self.tool and self.tool.Name() or None          return self.tool and self.tool.Name() or None
633    
634        def CurrentPosition(self):
635            """Return current position of the mouse in projected coordinates.
636    
637            The result is a 2-tuple of floats with the coordinates. If the
638            mouse is not in the window, the result is None.
639            """
640            if self.current_position is not None:
641                x, y = self.current_position
642                return self.win_to_proj(x, y)
643            else:
644                return None
645    
646        def set_current_position(self, event):
647            """Set the current position from event
648    
649            Should be called by all events that contain mouse positions
650            especially EVT_MOTION. The event paramete may be None to
651            indicate the the pointer left the window.
652            """
653            if event is not None:
654                self.current_position = (event.m_x, event.m_y)
655            else:
656                self.current_position = None
657            self.issue(VIEW_POSITION)
658    
659      def OnLeftDown(self, event):      def OnLeftDown(self, event):
660            self.set_current_position(event)
661          if self.tool is not None:          if self.tool is not None:
662              self.drag_dc = wxClientDC(self)              self.drag_dc = wxClientDC(self)
663              self.drag_dc.SetLogicalFunction(wxINVERT)              self.drag_dc.SetLogicalFunction(wxINVERT)
# Line 413  class MapCanvas(wxWindow): Line 666  class MapCanvas(wxWindow):
666              self.tool.MouseDown(event)              self.tool.MouseDown(event)
667              self.tool.Show(self.drag_dc)              self.tool.Show(self.drag_dc)
668              self.dragging = 1              self.dragging = 1
669            
670      def OnLeftUp(self, event):      def OnLeftUp(self, event):
671          self.ReleaseMouse()          self.set_current_position(event)
672          if self.dragging:          if self.dragging:
673              self.tool.Hide(self.drag_dc)              self.ReleaseMouse()
674              self.tool.MouseUp(event)              try:
675              self.drag_dc = None                  self.tool.Hide(self.drag_dc)
676          self.dragging = 0                  self.tool.MouseUp(event)
677                finally:
678                    self.drag_dc = None
679                    self.dragging = 0
680    
681      def OnMotion(self, event):      def OnMotion(self, event):
682            self.set_current_position(event)
683          if self.dragging:          if self.dragging:
684              self.tool.Hide(self.drag_dc)              self.tool.Hide(self.drag_dc)
685              self.tool.MouseMove(event)              self.tool.MouseMove(event)
686              self.tool.Show(self.drag_dc)              self.tool.Show(self.drag_dc)
687    
688      def OnIdle(self, event):      def OnLeaveWindow(self, event):
689          if self.redraw_on_idle:          self.set_current_position(None)
690              self.do_redraw()  
691          self.redraw_on_idle = 0      def OnSize(self, event):
692            # the window's size has changed. We have to get a new bitmap. If
693            # we want to be clever we could try to get by without throwing
694            # everything away. E.g. when the window gets smaller, we could
695            # either keep the bitmap or create the new one from the old one.
696            # Even when the window becomes larger some parts of the bitmap
697            # could be reused.
698            self.full_redraw()
699            pass
700    
701      def shape_selected(self, layer, shape):      def shape_selected(self, layer, shape):
702          self.redraw()          """Receiver for the SHAPES_SELECTED messages. Redraw the map."""
703            # The selection object takes care that it only issues
704            # SHAPES_SELECTED messages when the set of selected shapes has
705            # actually changed, so we can do a full redraw unconditionally.
706            # FIXME: We should perhaps try to limit the redraw to the are
707            # actually covered by the shapes before and after the selection
708            # change.
709            self.full_redraw()
710    
711        def unprojected_rect_around_point(self, x, y, dist):
712            """return a rect dist pixels around (x, y) in unprojected corrdinates
713    
714            The return value is a tuple (minx, miny, maxx, maxy) suitable a
715            parameter to a layer's ShapesInRegion method.
716            """
717            map_proj = self.map.projection
718            if map_proj is not None:
719                inverse = map_proj.Inverse
720            else:
721                inverse = None
722    
723      def find_shape_at(self, px, py, select_labels = 0, selected_layer = 1):          xs = []
724            ys = []
725            for dx, dy in ((-1, -1), (1, -1), (1, 1), (-1, 1)):
726                px, py = self.win_to_proj(x + dist * dx, y + dist * dy)
727                if inverse:
728                    px, py = inverse(px, py)
729                xs.append(px)
730                ys.append(py)
731            return (min(xs), min(ys), max(xs), max(ys))
732    
733        def find_shape_at(self, px, py, select_labels = 0, searched_layer = None):
734          """Determine the shape at point px, py in window coords          """Determine the shape at point px, py in window coords
735    
736          Return the shape and the corresponding layer as a tuple (layer,          Return the shape and the corresponding layer as a tuple (layer,
# Line 446  class MapCanvas(wxWindow): Line 740  class MapCanvas(wxWindow):
740          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
741          as the shape and None as the layer.          as the shape and None as the layer.
742    
743          If the optional parameter selected_layer is true (default), only          If the optional parameter searched_layer is given (or not None
744          search in the currently selected layer.          which it defaults to), only search in that layer.
745          """          """
746          map_proj = self.map.projection          map_proj = self.map.projection
747          if map_proj is not None:          if map_proj is not None:
# Line 460  class MapCanvas(wxWindow): Line 754  class MapCanvas(wxWindow):
754    
755          if select_labels:          if select_labels:
756              labels = self.map.LabelLayer().Labels()              labels = self.map.LabelLayer().Labels()
757                
758              if labels:              if labels:
759                  dc = wxClientDC(self)                  dc = wxClientDC(self)
760                  font = wxFont(10, wx.wxSWISS, wx.wxNORMAL, wx.wxNORMAL)                  font = wxFont(10, wx.wxSWISS, wx.wxNORMAL, wx.wxNORMAL)
761                  dc.SetFont(font)                  dc.SetFont(font)
762                  for i in range(len(labels)):                  for i in range(len(labels) - 1, -1, -1):
763                      label = labels[i]                      label = labels[i]
764                      x = label.x                      x = label.x
765                      y = label.y                      y = label.y
# Line 492  class MapCanvas(wxWindow): Line 786  class MapCanvas(wxWindow):
786                      if x <= px < x + width and y <= py <= y + height:                      if x <= px < x + width and y <= py <= y + height:
787                          return None, i                          return None, i
788    
789          if selected_layer:          if searched_layer:
790              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 = []  
791          else:          else:
792              layers = self.map.Layers()              layers = self.map.Layers()
793    
# Line 510  class MapCanvas(wxWindow): Line 798  class MapCanvas(wxWindow):
798              if not layer.Visible():              if not layer.Visible():
799                  continue                  continue
800    
801              filled = layer.fill is not None              filled = layer.GetClassification().GetDefaultFill() \
802              stroked = layer.stroke is not None                       is not Color.Transparent
803                                stroked = layer.GetClassification().GetDefaultLineColor() \
804                          is not Color.Transparent
805    
806              layer_proj = layer.projection              layer_proj = layer.projection
807              if layer_proj is not None:              if layer_proj is not None:
808                  inverse = layer_proj.Inverse                  inverse = layer_proj.Inverse
809              else:              else:
810                  inverse = None                  inverse = None
811                    
812              shapetype = layer.ShapeType()              shapetype = layer.ShapeType()
813    
814              select_shape = -1              select_shape = -1
815    
816                # Determine the ids of the shapes that overlap a tiny area
817                # around the point. For layers containing points we have to
818                # choose a larger size of the box we're testing agains so
819                # that we take the size of the markers into account
820                # FIXME: Once the markers are more flexible this part has to
821                # become more flexible too, of course
822                if shapetype == SHAPETYPE_POINT:
823                    box = self.unprojected_rect_around_point(px, py, 5)
824                else:
825                    box = self.unprojected_rect_around_point(px, py, 1)
826                shape_ids = layer.ShapesInRegion(box)
827                shape_ids.reverse()
828    
829              if shapetype == SHAPETYPE_POLYGON:              if shapetype == SHAPETYPE_POLYGON:
830                  for i in range(layer.NumShapes()):                  for i in shape_ids:
831                      result = point_in_polygon_shape(layer.shapefile.cobject(),                      result = point_in_polygon_shape(layer.shapefile.cobject(),
832                                                      i,                                                      i,
833                                                      filled, stroked,                                                      filled, stroked,
# Line 534  class MapCanvas(wxWindow): Line 838  class MapCanvas(wxWindow):
838                          select_shape = i                          select_shape = i
839                          break                          break
840              elif shapetype == SHAPETYPE_ARC:              elif shapetype == SHAPETYPE_ARC:
841                  for i in range(layer.NumShapes()):                  for i in shape_ids:
842                      result = point_in_polygon_shape(layer.shapefile.cobject(),                      result = point_in_polygon_shape(layer.shapefile.cobject(),
843                                                      i, 0, 1,                                                      i, 0, 1,
844                                                      map_proj, layer_proj,                                                      map_proj, layer_proj,
# Line 544  class MapCanvas(wxWindow): Line 848  class MapCanvas(wxWindow):
848                          select_shape = i                          select_shape = i
849                          break                          break
850              elif shapetype == SHAPETYPE_POINT:              elif shapetype == SHAPETYPE_POINT:
851                  for i in range(layer.NumShapes()):                  for i in shape_ids:
852                      shape = layer.Shape(i)                      shape = layer.Shape(i)
853                      x, y = shape.Points()[0]                      x, y = shape.Points()[0]
854                      if inverse:                      if inverse:
# Line 561  class MapCanvas(wxWindow): Line 865  class MapCanvas(wxWindow):
865                  return layer, select_shape                  return layer, select_shape
866          return None, None          return None, None
867    
868      def SelectShapeAt(self, x, y):      def SelectShapeAt(self, x, y, layer = None):
869          layer, shape = self.find_shape_at(x, y)          """\
870            Select and return the shape and its layer at window position (x, y)
871    
872            If layer is given, only search in that layer. If no layer is
873            given, search through all layers.
874    
875            Return a tuple (layer, shapeid). If no shape is found, return
876            (None, None).
877            """
878            layer, shape = result = self.find_shape_at(x, y, searched_layer=layer)
879          # 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
880          # to deselect the currently selected layer, so we simply select          # to deselect the currently selected layer, so we simply select
881          # the already selected layer again.          # the already selected layer again.
882          if layer is None:          if layer is None:
883              layer = self.interactor.SelectedLayer()              layer = self.selection.SelectedLayer()
884          self.interactor.SelectLayerAndShape(layer, shape)              shapes = []
885            else:
886                shapes = [shape]
887            self.selection.SelectShapes(layer, shapes)
888            return result
889    
890      def LabelShapeAt(self, x, y):      def LabelShapeAt(self, x, y):
891            """Add or remove a label at window position x, y.
892    
893            If there's a label at the given position, remove it. Otherwise
894            determine the shape at the position, run the label dialog and
895            unless the user cancels the dialog, add a laber.
896            """
897          ox = x; oy = y          ox = x; oy = y
898          label_layer = self.map.LabelLayer()          label_layer = self.map.LabelLayer()
899          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.45  
changed lines
  Added in v.831
[email protected]
 ViewVC Help
Powered by ViewVC 1.1.26