/[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 125 by bh, Thu May 2 18:55:33 2002 UTC revision 910 by frank, Fri May 16 16:23:43 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    # Frank Koormann <[email protected]>
5  #  #
6  # This program is free software under the GPL (>=v2)  # This program is free software under the GPL (>=v2)
7  # Read the file COPYING coming with Thuban for details.  # Read the file COPYING coming with Thuban for details.
# Line 11  Classes for display of a map and interac Line 12  Classes for display of a map and interac
12    
13  __version__ = "$Revision$"  __version__ = "$Revision$"
14    
15    from Thuban import _
16    
17    import sys
18    import os.path
19    
20  from math import hypot  from math import hypot
21    
22  from wxPython.wx import wxWindow,\  from wxPython.wx import wxWindow,\
23       wxPaintDC, wxColour, wxClientDC, wxINVERT, wxTRANSPARENT_BRUSH, wxFont,\       wxPaintDC, wxColour, wxClientDC, wxINVERT, wxTRANSPARENT_BRUSH, wxFont,\
24       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, \
25         wxBITMAP_TYPE_XPM, wxBeginBusyCursor, wxEndBusyCursor, wxCursor, \
26         wxImageFromBitmap, wxPlatform
27    
28    # Export related stuff
29    if wxPlatform == '__WXMSW__':
30        from wxPython.wx import wxMetaFileDC
31    from wxPython.wx import wxFileDialog, wxSAVE, wxOVERWRITE_PROMPT, wxID_OK
32    
33  from wxPython import wx  from wxPython import wx
34    
# Line 24  from wxproj import point_in_polygon_shap Line 36  from wxproj import point_in_polygon_shap
36    
37    
38  from Thuban.Model.messages import MAP_PROJECTION_CHANGED, \  from Thuban.Model.messages import MAP_PROJECTION_CHANGED, \
39       LAYERS_CHANGED, LAYER_LEGEND_CHANGED, LAYER_VISIBILITY_CHANGED       MAP_LAYERS_CHANGED, LAYER_CHANGED, LAYER_VISIBILITY_CHANGED
40  from Thuban.Model.layer import SHAPETYPE_POLYGON, SHAPETYPE_ARC, \  from Thuban.Model.layer import SHAPETYPE_POLYGON, SHAPETYPE_ARC, \
41       SHAPETYPE_POINT       SHAPETYPE_POINT
42  from Thuban.Model.label import ALIGN_CENTER, ALIGN_TOP, ALIGN_BOTTOM, \  from Thuban.Model.label import ALIGN_CENTER, ALIGN_TOP, ALIGN_BOTTOM, \
43       ALIGN_LEFT, ALIGN_RIGHT       ALIGN_LEFT, ALIGN_RIGHT
44  from Thuban.Lib.connector import Publisher  from Thuban.Lib.connector import Publisher
45    from Thuban.Model.color import Color
46    
47    import resource
48    
49  from renderer import ScreenRenderer, PrinterRender  from selection import Selection
50    from renderer import ScreenRenderer, ExportRenderer, PrinterRenderer
51    
52  import labeldialog  import labeldialog
53    
54  from messages import SELECTED_SHAPE, VIEW_POSITION  from messages import LAYER_SELECTED, SHAPES_SELECTED, VIEW_POSITION, \
55                         SCALE_CHANGED
56    
57    
58  #  #
# Line 129  class ZoomInTool(RectTool): Line 146  class ZoomInTool(RectTool):
146              Tool.MouseUp(self, event)              Tool.MouseUp(self, event)
147              sx, sy = self.start              sx, sy = self.start
148              cx, cy = self.current              cx, cy = self.current
149              if sx == cx and sy == cy:              if sx == cx or sy == cy:
150                  # Just a mouse click. Simply zoom in by a factor of two                  # Just a mouse click or a degenerate rectangle. Simply
151                    # zoom in by a factor of two
152                    # FIXME: For a click this is the desired behavior but should we
153                    # really do this for degenrate rectagles as well or
154                    # should we ignore them?
155                  self.view.ZoomFactor(2, center = (cx, cy))                  self.view.ZoomFactor(2, center = (cx, cy))
156              else:              else:
157                  # A drag. Zoom in to the rectangle                  # A drag. Zoom in to the rectangle
# Line 140  class ZoomInTool(RectTool): Line 161  class ZoomInTool(RectTool):
161  class ZoomOutTool(RectTool):  class ZoomOutTool(RectTool):
162    
163      """The Zoom-Out Tool"""      """The Zoom-Out Tool"""
164        
165      def Name(self):      def Name(self):
166          return "ZoomOutTool"          return "ZoomOutTool"
167    
# Line 149  class ZoomOutTool(RectTool): Line 170  class ZoomOutTool(RectTool):
170              Tool.MouseUp(self, event)              Tool.MouseUp(self, event)
171              sx, sy = self.start              sx, sy = self.start
172              cx, cy = self.current              cx, cy = self.current
173              if sx == cx and sy == cy:              if sx == cx or sy == cy:
174                  # Just a mouse click. Simply zoom out by a factor of two                  # Just a mouse click or a degenerate rectangle. Simply
175                  self.view.ZoomFactor(0.5, center = (cy, cy))                  # zoom out by a factor of two.
176                    # FIXME: For a click this is the desired behavior but should we
177                    # really do this for degenrate rectagles as well or
178                    # should we ignore them?
179                    self.view.ZoomFactor(0.5, center = (cx, cy))
180              else:              else:
181                  # A drag. Zoom out to the rectangle                  # A drag. Zoom out to the rectangle
182                  self.view.ZoomOutToRect((min(sx, cx), min(sy, cy),                  self.view.ZoomOutToRect((min(sx, cx), min(sy, cy),
# Line 167  class PanTool(Tool): Line 192  class PanTool(Tool):
192    
193      def MouseMove(self, event):      def MouseMove(self, event):
194          if self.dragging:          if self.dragging:
             x0, y0 = self.current  
195              Tool.MouseMove(self, event)              Tool.MouseMove(self, event)
196                sx, sy = self.start
197              x, y = self.current              x, y = self.current
198              width, height = self.view.GetSizeTuple()              width, height = self.view.GetSizeTuple()
199    
200                bitmapdc = wx.wxMemoryDC()
201                bitmapdc.SelectObject(self.view.bitmap)
202    
203              dc = self.view.drag_dc              dc = self.view.drag_dc
204              dc.Blit(0, 0, width, height, dc, x0 - x, y0 - y)              dc.Blit(0, 0, width, height, bitmapdc, sx - x, sy - y)
205    
206      def MouseUp(self, event):      def MouseUp(self, event):
207          if self.dragging:          if self.dragging:
# Line 180  class PanTool(Tool): Line 209  class PanTool(Tool):
209              sx, sy = self.start              sx, sy = self.start
210              cx, cy = self.current              cx, cy = self.current
211              self.view.Translate(cx - sx, cy - sy)              self.view.Translate(cx - sx, cy - sy)
212            
213  class IdentifyTool(Tool):  class IdentifyTool(Tool):
214    
215      """The "Identify" Tool"""      """The "Identify" Tool"""
216        
217      def Name(self):      def Name(self):
218          return "IdentifyTool"          return "IdentifyTool"
219    
# Line 203  class LabelTool(Tool): Line 232  class LabelTool(Tool):
232          self.view.LabelShapeAt(event.m_x, event.m_y)          self.view.LabelShapeAt(event.m_x, event.m_y)
233    
234    
   
   
235  class MapPrintout(wx.wxPrintout):  class MapPrintout(wx.wxPrintout):
236    
237      """      """
238      wxPrintout class for printing Thuban maps      wxPrintout class for printing Thuban maps
239      """      """
240    
241      def __init__(self, map):      def __init__(self, canvas, map, region, selected_layer, selected_shapes):
242          wx.wxPrintout.__init__(self)          wx.wxPrintout.__init__(self)
243            self.canvas = canvas
244          self.map = map          self.map = map
245            self.region = region
246            self.selected_layer = selected_layer
247            self.selected_shapes = selected_shapes
248    
249      def GetPageInfo(self):      def GetPageInfo(self):
250          return (1, 1, 1, 1)          return (1, 1, 1, 1)
# Line 227  class MapPrintout(wx.wxPrintout): Line 258  class MapPrintout(wx.wxPrintout):
258    
259      def draw_on_dc(self, dc):      def draw_on_dc(self, dc):
260          width, height = self.GetPageSizePixels()          width, height = self.GetPageSizePixels()
261          llx, lly, urx, ury = self.map.ProjectedBoundingBox()          scale, offset, mapregion = OutputTransform(self.canvas.scale,
262          scalex = width / (urx - llx)                                                     self.canvas.offset,
263          scaley = height / (ury - lly)                                                     self.canvas.GetSizeTuple(),
264          scale = min(scalex, scaley)                                                     self.GetPageSizePixels())
         offx = 0.5 * (width - (urx + llx) * scale)  
         offy = 0.5 * (height + (ury + lly) * scale)  
   
265          resx, resy = self.GetPPIPrinter()          resx, resy = self.GetPPIPrinter()
266          renderer = PrinterRender(dc, scale, (offx, offy), resolution = resx)          renderer = PrinterRenderer(dc, scale, offset, resolution = resy)
267          renderer.RenderMap(self.map)          x, y, width, height = self.region
268            canvas_scale = self.canvas.scale
269            renderer.RenderMap(self.map,
270                               (0,0,
271                                    (width/canvas_scale)*scale,
272                                    (height/canvas_scale)*scale),
273                                    mapregion,
274                               self.selected_layer, self.selected_shapes)
275          return wx.true          return wx.true
276            
277    
278  class MapCanvas(wxWindow, Publisher):  class MapCanvas(wxWindow, Publisher):
279    
280      """A widget that displays a map and offers some interaction"""      """A widget that displays a map and offers some interaction"""
281    
282      def __init__(self, parent, winid, interactor):      # Some messages that can be subscribed/unsubscribed directly through
283        # the MapCanvas come in fact from other objects. This is a dict
284        # mapping those messages to the names of the instance variables they
285        # actually come from. The delegation is implemented in the Subscribe
286        # and Unsubscribe methods
287        delegated_messages = {LAYER_SELECTED: "selection",
288                              SHAPES_SELECTED: "selection"}
289    
290        # Methods delegated to some instance variables. The delegation is
291        # implemented in the __getattr__ method.
292        delegated_methods = {"SelectLayer": "selection",
293                             "SelectShapes": "selection",
294                             "SelectedLayer": "selection",
295                             "HasSelectedLayer": "selection",
296                             "HasSelectedShapes": "selection",
297                             "SelectedShapes": "selection"}
298    
299        def __init__(self, parent, winid):
300          wxWindow.__init__(self, parent, winid)          wxWindow.__init__(self, parent, winid)
301          self.SetBackgroundColour(wxColour(255, 255, 255))          self.SetBackgroundColour(wxColour(255, 255, 255))
302    
# Line 267  class MapCanvas(wxWindow, Publisher): Line 319  class MapCanvas(wxWindow, Publisher):
319          # if the mouse is outside the window.          # if the mouse is outside the window.
320          self.current_position = None          self.current_position = None
321    
   
         # 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  
   
322          # the bitmap serving as backing store          # the bitmap serving as backing store
323          self.bitmap = None          self.bitmap = None
324    
325          # the interactor          # the selection
326          self.interactor = interactor          self.selection = Selection()
327          self.interactor.Subscribe(SELECTED_SHAPE, self.shape_selected)          self.selection.Subscribe(SHAPES_SELECTED , self.shape_selected)
328    
329            # keep track of which layers/shapes are selected to make sure we
330            # only redraw when necessary
331            self.last_selected_layer = None
332            self.last_selected_shape = None
333    
334          # subscribe the WX events we're interested in          # subscribe the WX events we're interested in
335          EVT_PAINT(self, self.OnPaint)          EVT_PAINT(self, self.OnPaint)
# Line 287  class MapCanvas(wxWindow, Publisher): Line 338  class MapCanvas(wxWindow, Publisher):
338          EVT_MOTION(self, self.OnMotion)          EVT_MOTION(self, self.OnMotion)
339          EVT_LEAVE_WINDOW(self, self.OnLeaveWindow)          EVT_LEAVE_WINDOW(self, self.OnLeaveWindow)
340          wx.EVT_SIZE(self, self.OnSize)          wx.EVT_SIZE(self, self.OnSize)
         wx.EVT_IDLE(self, self.OnIdle)  
341    
342      def __del__(self):      def __del__(self):
343          wxWindow.__del__(self)          wxWindow.__del__(self)
344          Publisher.__del__(self)          Publisher.__del__(self)
345    
346        def Subscribe(self, channel, *args):
347            """Extend the inherited method to handle delegated messages.
348    
349            If channel is one of the delegated messages call the appropriate
350            object's Subscribe method. Otherwise just call the inherited
351            method.
352            """
353            if channel in self.delegated_messages:
354                object = getattr(self, self.delegated_messages[channel])
355                object.Subscribe(channel, *args)
356            else:
357                Publisher.Subscribe(self, channel, *args)
358    
359        def Unsubscribe(self, channel, *args):
360            """Extend the inherited method to handle delegated messages.
361    
362            If channel is one of the delegated messages call the appropriate
363            object's Unsubscribe method. Otherwise just call the inherited
364            method.
365            """
366            if channel in self.delegated_messages:
367                object = getattr(self, self.delegated_messages[channel])
368                object.Unsubscribe(channel, *args)
369            else:
370                Publisher.Unsubscribe(self, channel, *args)
371    
372        def __getattr__(self, attr):
373            if attr in self.delegated_methods:
374                return getattr(getattr(self, self.delegated_methods[attr]), attr)
375            raise AttributeError(attr)
376    
377      def OnPaint(self, event):      def OnPaint(self, event):
378          dc = wxPaintDC(self)          dc = wxPaintDC(self)
379          if self.map is not None and self.map.HasLayers():          clear = self.map is None or not self.map.HasLayers()
380              # We have a non-empty map. Redraw it in idle time  
381              self.redraw_on_idle = 1          #wxBeginBusyCursor()
382          else:  
383            if not clear:
384                try:
385                    self.do_redraw()
386                except:
387                    print "Error during drawing:", sys.exc_info()[0]
388                    clear = True
389    
390            if clear:
391              # 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
392              # the screen.              # the screen.
393                
394              # XXX it's probably possible to get rid of this. The              # XXX it's probably possible to get rid of this. The
395              # background color of the window is already white and the              # background color of the window is already white and the
396              # only thing we may have to do is to call self.Refresh()              # only thing we may have to do is to call self.Refresh()
397              # with a true argument in the right places.              # with a true argument in the right places.
398              dc.BeginDrawing()              dc.BeginDrawing()
399              dc.Clear()                          dc.Clear()
400              dc.EndDrawing()              dc.EndDrawing()
401    
402            #wxEndBusyCursor()
403    
404      def do_redraw(self):      def do_redraw(self):
405          # This should only be called if we have a non-empty map. We draw          # This should only be called if we have a non-empty map.
         # it into a memory DC and then blit it to the screen.  
406    
407            # Get the window size.
408          width, height = self.GetSizeTuple()          width, height = self.GetSizeTuple()
409    
410          # If self.bitmap's still there, reuse it. Otherwise redraw it          # If self.bitmap's still there, reuse it. Otherwise redraw it
# Line 326  class MapCanvas(wxWindow, Publisher): Line 417  class MapCanvas(wxWindow, Publisher):
417              dc.BeginDrawing()              dc.BeginDrawing()
418    
419              # clear the background              # clear the background
420              dc.SetBrush(wx.wxWHITE_BRUSH)              #dc.SetBrush(wx.wxWHITE_BRUSH)
421              dc.SetPen(wx.wxTRANSPARENT_PEN)              #dc.SetPen(wx.wxTRANSPARENT_PEN)
422              dc.DrawRectangle(0, 0, width, height)              #dc.DrawRectangle(0, 0, width, height)
423                dc.SetBackground(wx.wxWHITE_BRUSH)
424              if 1: #self.interactor.selected_map is self.map:              dc.Clear()
425                  selected_layer = self.interactor.selected_layer  
426                  selected_shape = self.interactor.selected_shape              selected_layer = self.selection.SelectedLayer()
427              else:              selected_shapes = self.selection.SelectedShapes()
                 selected_layer = None  
                 selected_shape = None  
428    
429              # draw the map into the bitmap              # draw the map into the bitmap
430              renderer = ScreenRenderer(dc, self.scale, self.offset)              renderer = ScreenRenderer(dc, self.scale, self.offset)
431              renderer.RenderMap(self.map, selected_layer, selected_shape)  
432                # Pass the entire bitmap as update region to the renderer.
433                # We're redrawing the whole bitmap, after all.
434                renderer.RenderMap(self.map, (0, 0, width, height),
435                                   selected_layer, selected_shapes)
436    
437              dc.EndDrawing()              dc.EndDrawing()
438              dc.SelectObject(wx.wxNullBitmap)              dc.SelectObject(wx.wxNullBitmap)
# Line 353  class MapCanvas(wxWindow, Publisher): Line 446  class MapCanvas(wxWindow, Publisher):
446          clientdc.Blit(0, 0, width, height, dc, 0, 0)          clientdc.Blit(0, 0, width, height, dc, 0, 0)
447          clientdc.EndDrawing()          clientdc.EndDrawing()
448    
449        def Export(self):
450            if hasattr(self, "export_path"):
451                export_path = self.export_path
452            else:
453                export_path="."
454            dlg = wxFileDialog(self, _("Export Map"), export_path, "",
455                               "Enhanced Metafile (*.wmf)|*.wmf",
456                               wxSAVE|wxOVERWRITE_PROMPT)
457            if dlg.ShowModal() == wxID_OK:
458                self.export_path = os.path.dirname(dlg.GetPath())
459                dc = wxMetaFileDC(dlg.GetPath())
460        
461                scale, offset, mapregion = OutputTransform(self.scale,
462                                                           self.canvas.offset,
463                                                           self.GetSizeTuple(),
464                                                           dc.GetSizeTuple())
465    
466                selected_layer = self.selection.SelectedLayer()
467                selected_shapes = self.selection.SelectedShapes()
468    
469                renderer = ExportRenderer(dc, scale, offset)
470    
471                # Pass the entire bitmap as update region to the renderer.
472                # We're redrawing the whole bitmap, after all.
473                width, height = self.GetSizeTuple()
474                renderer.RenderMap(self.map,
475                                    (0,0,
476                                        (width/self.scale)*scale,
477                                        (height/self.scale)*scale),
478                                    mapregion,
479                                    selected_layer, selected_shapes)
480                dc.EndDrawing()
481                dc.Close()
482            dlg.Destroy()
483            
484      def Print(self):      def Print(self):
485          printer = wx.wxPrinter()          printer = wx.wxPrinter()
486          printout = MapPrintout(self.map)          width, height = self.GetSizeTuple()
487            selected_layer = self.selection.SelectedLayer()
488            selected_shapes = self.selection.SelectedShapes()
489            
490            printout = MapPrintout(self, self.map, (0, 0, width, height),
491                                   selected_layer, selected_shapes)
492          printer.Print(self, printout, wx.true)          printer.Print(self, printout, wx.true)
493          printout.Destroy()          printout.Destroy()
494            
495      def SetMap(self, map):      def SetMap(self, map):
496          redraw_channels = (LAYERS_CHANGED, LAYER_LEGEND_CHANGED,          redraw_channels = (MAP_LAYERS_CHANGED, LAYER_CHANGED,
497                             LAYER_VISIBILITY_CHANGED)                             LAYER_VISIBILITY_CHANGED)
498          if self.map is not None:          if self.map is not None:
499              for channel in redraw_channels:              for channel in redraw_channels:
# Line 368  class MapCanvas(wxWindow, Publisher): Line 501  class MapCanvas(wxWindow, Publisher):
501              self.map.Unsubscribe(MAP_PROJECTION_CHANGED,              self.map.Unsubscribe(MAP_PROJECTION_CHANGED,
502                                   self.projection_changed)                                   self.projection_changed)
503          self.map = map          self.map = map
504            self.selection.ClearSelection()
505          if self.map is not None:          if self.map is not None:
506              for channel in redraw_channels:              for channel in redraw_channels:
507                  self.map.Subscribe(channel, self.full_redraw)                  self.map.Subscribe(channel, self.full_redraw)
# Line 379  class MapCanvas(wxWindow, Publisher): Line 513  class MapCanvas(wxWindow, Publisher):
513          self.full_redraw()          self.full_redraw()
514    
515      def Map(self):      def Map(self):
516            """Return the map displayed by this canvas"""
517          return self.map          return self.map
518    
519      def redraw(self, *args):      def redraw(self, *args):
# Line 396  class MapCanvas(wxWindow, Publisher): Line 531  class MapCanvas(wxWindow, Publisher):
531          self.scale = scale          self.scale = scale
532          self.offset = offset          self.offset = offset
533          self.full_redraw()          self.full_redraw()
534            self.issue(SCALE_CHANGED, scale)
535    
536      def proj_to_win(self, x, y):      def proj_to_win(self, x, y):
537          """\          """\
# Line 412  class MapCanvas(wxWindow, Publisher): Line 548  class MapCanvas(wxWindow, Publisher):
548          return ((x - offx) / self.scale, (offy - y) / self.scale)          return ((x - offx) / self.scale, (offy - y) / self.scale)
549    
550      def FitRectToWindow(self, rect):      def FitRectToWindow(self, rect):
551            """Fit the rectangular region given by rect into the window.
552    
553            Set scale so that rect (in projected coordinates) just fits into
554            the window and center it.
555            """
556          width, height = self.GetSizeTuple()          width, height = self.GetSizeTuple()
557          llx, lly, urx, ury = rect          llx, lly, urx, ury = rect
558          if llx == urx or lly == ury:          if llx == urx or lly == ury:
559              # zero with or zero height. Do Nothing              # zero width or zero height. Do Nothing
560              return              return
561          scalex = width / (urx - llx)          scalex = width / (urx - llx)
562          scaley = height / (ury - lly)          scaley = height / (ury - lly)
# Line 425  class MapCanvas(wxWindow, Publisher): Line 566  class MapCanvas(wxWindow, Publisher):
566          self.set_view_transform(scale, (offx, offy))          self.set_view_transform(scale, (offx, offy))
567    
568      def FitMapToWindow(self):      def FitMapToWindow(self):
569          """\          """Fit the map to the window
570          Set the scale and offset so that the map is centered in the  
571          window          Set the scale so that the map fits exactly into the window and
572            center it in the window.
573          """          """
574          bbox = self.map.ProjectedBoundingBox()          bbox = self.map.ProjectedBoundingBox()
575          if bbox is not None:          if bbox is not None:
576              self.FitRectToWindow(bbox)              self.FitRectToWindow(bbox)
577    
578        def FitLayerToWindow(self, layer):
579            """Fit the given layer to the window.
580    
581            Set the scale so that the layer fits exactly into the window and
582            center it in the window.
583            """
584            
585            bbox = layer.LatLongBoundingBox()
586            if bbox is not None:
587                proj = self.map.GetProjection()
588                if proj is not None:
589                    bbox = proj.ForwardBBox(bbox)
590    
591                if bbox is not None:
592                    self.FitRectToWindow(bbox)
593    
594        def FitSelectedToWindow(self):
595            layer = self.selection.SelectedLayer()
596            shapes = self.selection.SelectedShapes()
597    
598            bbox = layer.ShapesBoundingBox(shapes)
599            if bbox is not None:
600                proj = self.map.GetProjection()
601                if proj is not None:
602                    bbox = proj.ForwardBBox(bbox)
603    
604                if bbox is not None:
605                    self.FitRectToWindow(bbox)
606    
607      def ZoomFactor(self, factor, center = None):      def ZoomFactor(self, factor, center = None):
608          """Multiply the zoom by factor and center on center.          """Multiply the zoom by factor and center on center.
609    
# Line 453  class MapCanvas(wxWindow, Publisher): Line 624  class MapCanvas(wxWindow, Publisher):
624          self.set_view_transform(scale, offset)          self.set_view_transform(scale, offset)
625    
626      def ZoomOutToRect(self, rect):      def ZoomOutToRect(self, rect):
627          # rect is given in window coordinates          """Zoom out to fit the currently visible region into rect.
628    
629            The rect parameter is given in window coordinates
630            """
631          # determine the bbox of the displayed region in projected          # determine the bbox of the displayed region in projected
632          # coordinates          # coordinates
633          width, height = self.GetSizeTuple()          width, height = self.GetSizeTuple()
# Line 471  class MapCanvas(wxWindow, Publisher): Line 644  class MapCanvas(wxWindow, Publisher):
644          self.set_view_transform(scale, (offx, offy))          self.set_view_transform(scale, (offx, offy))
645    
646      def Translate(self, dx, dy):      def Translate(self, dx, dy):
647            """Move the map by dx, dy pixels"""
648          offx, offy = self.offset          offx, offy = self.offset
649          self.set_view_transform(self.scale, (offx + dx, offy + dy))          self.set_view_transform(self.scale, (offx + dx, offy + dy))
650    
651        def SelectTool(self, tool):
652            """Make tool the active tool.
653    
654            The parameter should be an instance of Tool or None to indicate
655            that no tool is active.
656            """
657            self.tool = tool
658    
659      def ZoomInTool(self):      def ZoomInTool(self):
660          self.tool = ZoomInTool(self)          """Start the zoom in tool"""
661            self.SelectTool(ZoomInTool(self))
662    
663      def ZoomOutTool(self):      def ZoomOutTool(self):
664          self.tool = ZoomOutTool(self)          """Start the zoom out tool"""
665            self.SelectTool(ZoomOutTool(self))
666    
667      def PanTool(self):      def PanTool(self):
668          self.tool = PanTool(self)          """Start the pan tool"""
669            self.SelectTool(PanTool(self))
670            #img = resource.GetImageResource("pan", wxBITMAP_TYPE_XPM)
671            #bmp = resource.GetBitmapResource("pan", wxBITMAP_TYPE_XPM)
672            #print bmp
673            #img = wxImageFromBitmap(bmp)
674            #print img
675            #cur = wxCursor(img)
676            #print cur
677            #self.SetCursor(cur)
678    
679      def IdentifyTool(self):      def IdentifyTool(self):
680          self.tool = IdentifyTool(self)          """Start the identify tool"""
681            self.SelectTool(IdentifyTool(self))
682    
683      def LabelTool(self):      def LabelTool(self):
684          self.tool = LabelTool(self)          """Start the label tool"""
685            self.SelectTool(LabelTool(self))
686    
687      def CurrentTool(self):      def CurrentTool(self):
688            """Return the name of the current tool or None if no tool is active"""
689          return self.tool and self.tool.Name() or None          return self.tool and self.tool.Name() or None
690    
691      def CurrentPosition(self):      def CurrentPosition(self):
# Line 527  class MapCanvas(wxWindow, Publisher): Line 723  class MapCanvas(wxWindow, Publisher):
723              self.tool.MouseDown(event)              self.tool.MouseDown(event)
724              self.tool.Show(self.drag_dc)              self.tool.Show(self.drag_dc)
725              self.dragging = 1              self.dragging = 1
726            
727      def OnLeftUp(self, event):      def OnLeftUp(self, event):
         self.ReleaseMouse()  
728          self.set_current_position(event)          self.set_current_position(event)
729          if self.dragging:          if self.dragging:
730              self.tool.Hide(self.drag_dc)              self.ReleaseMouse()
731              self.tool.MouseUp(event)              try:
732              self.drag_dc = None                  self.tool.Hide(self.drag_dc)
733          self.dragging = 0                  self.tool.MouseUp(event)
734                finally:
735                    self.drag_dc = None
736                    self.dragging = 0
737    
738      def OnMotion(self, event):      def OnMotion(self, event):
739          self.set_current_position(event)          self.set_current_position(event)
# Line 547  class MapCanvas(wxWindow, Publisher): Line 745  class MapCanvas(wxWindow, Publisher):
745      def OnLeaveWindow(self, event):      def OnLeaveWindow(self, event):
746          self.set_current_position(None)          self.set_current_position(None)
747    
     def OnIdle(self, event):  
         if self.redraw_on_idle:  
             self.do_redraw()  
         self.redraw_on_idle = 0  
   
748      def OnSize(self, event):      def OnSize(self, event):
749          # 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
750          # 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 560  class MapCanvas(wxWindow, Publisher): Line 753  class MapCanvas(wxWindow, Publisher):
753          # Even when the window becomes larger some parts of the bitmap          # Even when the window becomes larger some parts of the bitmap
754          # could be reused.          # could be reused.
755          self.full_redraw()          self.full_redraw()
756            pass
757    
758      def shape_selected(self, layer, shape):      def shape_selected(self, layer, shape):
759            """Receiver for the SHAPES_SELECTED messages. Redraw the map."""
760            # The selection object takes care that it only issues
761            # SHAPES_SELECTED messages when the set of selected shapes has
762            # actually changed, so we can do a full redraw unconditionally.
763            # FIXME: We should perhaps try to limit the redraw to the are
764            # actually covered by the shapes before and after the selection
765            # change.
766          self.full_redraw()          self.full_redraw()
767    
768      def find_shape_at(self, px, py, select_labels = 0, selected_layer = 1):      def unprojected_rect_around_point(self, x, y, dist):
769            """return a rect dist pixels around (x, y) in unprojected corrdinates
770    
771            The return value is a tuple (minx, miny, maxx, maxy) suitable a
772            parameter to a layer's ShapesInRegion method.
773            """
774            map_proj = self.map.projection
775            if map_proj is not None:
776                inverse = map_proj.Inverse
777            else:
778                inverse = None
779    
780            xs = []
781            ys = []
782            for dx, dy in ((-1, -1), (1, -1), (1, 1), (-1, 1)):
783                px, py = self.win_to_proj(x + dist * dx, y + dist * dy)
784                if inverse:
785                    px, py = inverse(px, py)
786                xs.append(px)
787                ys.append(py)
788            return (min(xs), min(ys), max(xs), max(ys))
789    
790        def find_shape_at(self, px, py, select_labels = 0, searched_layer = None):
791          """Determine the shape at point px, py in window coords          """Determine the shape at point px, py in window coords
792    
793          Return the shape and the corresponding layer as a tuple (layer,          Return the shape and the corresponding layer as a tuple (layer,
# Line 574  class MapCanvas(wxWindow, Publisher): Line 797  class MapCanvas(wxWindow, Publisher):
797          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
798          as the shape and None as the layer.          as the shape and None as the layer.
799    
800          If the optional parameter selected_layer is true (default), only          If the optional parameter searched_layer is given (or not None
801          search in the currently selected layer.          which it defaults to), only search in that layer.
802          """          """
803          map_proj = self.map.projection          map_proj = self.map.projection
804          if map_proj is not None:          if map_proj is not None:
# Line 588  class MapCanvas(wxWindow, Publisher): Line 811  class MapCanvas(wxWindow, Publisher):
811    
812          if select_labels:          if select_labels:
813              labels = self.map.LabelLayer().Labels()              labels = self.map.LabelLayer().Labels()
814                
815              if labels:              if labels:
816                  dc = wxClientDC(self)                  dc = wxClientDC(self)
817                  font = wxFont(10, wx.wxSWISS, wx.wxNORMAL, wx.wxNORMAL)                  font = wxFont(10, wx.wxSWISS, wx.wxNORMAL, wx.wxNORMAL)
# Line 620  class MapCanvas(wxWindow, Publisher): Line 843  class MapCanvas(wxWindow, Publisher):
843                      if x <= px < x + width and y <= py <= y + height:                      if x <= px < x + width and y <= py <= y + height:
844                          return None, i                          return None, i
845    
846          if selected_layer:          if searched_layer:
847              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 = []  
848          else:          else:
849              layers = self.map.Layers()              layers = self.map.Layers()
850    
# Line 638  class MapCanvas(wxWindow, Publisher): Line 855  class MapCanvas(wxWindow, Publisher):
855              if not layer.Visible():              if not layer.Visible():
856                  continue                  continue
857    
858              filled = layer.fill is not None              filled = layer.GetClassification().GetDefaultFill() \
859              stroked = layer.stroke is not None                       is not Color.Transparent
860                                stroked = layer.GetClassification().GetDefaultLineColor() \
861                          is not Color.Transparent
862    
863              layer_proj = layer.projection              layer_proj = layer.projection
864              if layer_proj is not None:              if layer_proj is not None:
865                  inverse = layer_proj.Inverse                  inverse = layer_proj.Inverse
866              else:              else:
867                  inverse = None                  inverse = None
868                    
869              shapetype = layer.ShapeType()              shapetype = layer.ShapeType()
870    
871              select_shape = -1              select_shape = -1
872    
873                # Determine the ids of the shapes that overlap a tiny area
874                # around the point. For layers containing points we have to
875                # choose a larger size of the box we're testing agains so
876                # that we take the size of the markers into account
877                # FIXME: Once the markers are more flexible this part has to
878                # become more flexible too, of course
879                if shapetype == SHAPETYPE_POINT:
880                    box = self.unprojected_rect_around_point(px, py, 5)
881                else:
882                    box = self.unprojected_rect_around_point(px, py, 1)
883                shape_ids = layer.ShapesInRegion(box)
884                shape_ids.reverse()
885    
886              if shapetype == SHAPETYPE_POLYGON:              if shapetype == SHAPETYPE_POLYGON:
887                  for i in range(layer.NumShapes() - 1, -1, -1):                  for i in shape_ids:
888                      result = point_in_polygon_shape(layer.shapefile.cobject(),                      result = point_in_polygon_shape(layer.shapefile.cobject(),
889                                                      i,                                                      i,
890                                                      filled, stroked,                                                      filled, stroked,
# Line 662  class MapCanvas(wxWindow, Publisher): Line 895  class MapCanvas(wxWindow, Publisher):
895                          select_shape = i                          select_shape = i
896                          break                          break
897              elif shapetype == SHAPETYPE_ARC:              elif shapetype == SHAPETYPE_ARC:
898                  for i in range(layer.NumShapes() - 1, -1, -1):                  for i in shape_ids:
899                      result = point_in_polygon_shape(layer.shapefile.cobject(),                      result = point_in_polygon_shape(layer.shapefile.cobject(),
900                                                      i, 0, 1,                                                      i, 0, 1,
901                                                      map_proj, layer_proj,                                                      map_proj, layer_proj,
# Line 672  class MapCanvas(wxWindow, Publisher): Line 905  class MapCanvas(wxWindow, Publisher):
905                          select_shape = i                          select_shape = i
906                          break                          break
907              elif shapetype == SHAPETYPE_POINT:              elif shapetype == SHAPETYPE_POINT:
908                  for i in range(layer.NumShapes() - 1, -1, -1):                  for i in shape_ids:
909                      shape = layer.Shape(i)                      shape = layer.Shape(i)
910                      x, y = shape.Points()[0]                      x, y = shape.Points()[0]
911                      if inverse:                      if inverse:
# Line 689  class MapCanvas(wxWindow, Publisher): Line 922  class MapCanvas(wxWindow, Publisher):
922                  return layer, select_shape                  return layer, select_shape
923          return None, None          return None, None
924    
925      def SelectShapeAt(self, x, y):      def SelectShapeAt(self, x, y, layer = None):
926          layer, shape = self.find_shape_at(x, y, selected_layer = 0)          """\
927            Select and return the shape and its layer at window position (x, y)
928    
929            If layer is given, only search in that layer. If no layer is
930            given, search through all layers.
931    
932            Return a tuple (layer, shapeid). If no shape is found, return
933            (None, None).
934            """
935            layer, shape = result = self.find_shape_at(x, y, searched_layer=layer)
936          # 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
937          # to deselect the currently selected layer, so we simply select          # to deselect the currently selected layer, so we simply select
938          # the already selected layer again.          # the already selected layer again.
939          if layer is None:          if layer is None:
940              layer = self.interactor.SelectedLayer()              layer = self.selection.SelectedLayer()
941          self.interactor.SelectLayerAndShape(layer, shape)              shapes = []
942            else:
943                shapes = [shape]
944            self.selection.SelectShapes(layer, shapes)
945            return result
946    
947      def LabelShapeAt(self, x, y):      def LabelShapeAt(self, x, y):
948            """Add or remove a label at window position x, y.
949    
950            If there's a label at the given position, remove it. Otherwise
951            determine the shape at the position, run the label dialog and
952            unless the user cancels the dialog, add a laber.
953            """
954          ox = x; oy = y          ox = x; oy = y
955          label_layer = self.map.LabelLayer()          label_layer = self.map.LabelLayer()
956          layer, shape_index = self.find_shape_at(x, y, select_labels = 1)          layer, shape_index = self.find_shape_at(x, y, select_labels = 1)
# Line 747  class MapCanvas(wxWindow, Publisher): Line 999  class MapCanvas(wxWindow, Publisher):
999                      valign = ALIGN_CENTER                      valign = ALIGN_CENTER
1000                  label_layer.AddLabel(x, y, text,                  label_layer.AddLabel(x, y, text,
1001                                       halign = halign, valign = valign)                                       halign = halign, valign = valign)
1002    
1003    def OutputTransform(canvas_scale, canvas_offset, canvas_size, device_extend):
1004        """Calculate dimensions to transform canvas content to output device."""
1005        width, height = device_extend
1006    
1007        # Only 80 % of the with are available for the map
1008        width = width * 0.8
1009    
1010        # Define the distance of the map from DC border
1011        distance = 20
1012    
1013        if height < width:
1014            # landscape
1015            map_height = height - 2*distance
1016            map_width = map_height
1017        else:
1018            # portrait, recalibrate width (usually the legend width is too
1019            # small
1020            width = width * 0.9
1021            map_height = width - 2*distance
1022            map_width = map_height
1023        
1024        mapregion = (distance, distance,
1025                     distance+map_width, distance+map_height)
1026    
1027        canvas_width, canvas_height = canvas_size
1028        
1029        scalex = map_width / (canvas_width/canvas_scale)
1030        scaley = map_height / (canvas_height/canvas_scale)
1031        scale = min(scalex, scaley)
1032        canvas_offx, canvas_offy = canvas_offset
1033        offx = scale*canvas_offx/canvas_scale
1034        offy = scale*canvas_offy/canvas_scale
1035    
1036        return scale, (offx, offy), mapregion
Legend:
Removed from v.125  
changed lines
  Added in v.910
[email protected]
 ViewVC Help
Powered by ViewVC 1.1.26