/[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 6 by bh, Tue Aug 28 15:41:52 2001 UTC revision 926 by frank, Mon May 19 12:09:01 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    # 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_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
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  from messages import LAYER_SELECTED, SHAPES_SELECTED, VIEW_POSITION, \
55                         SCALE_CHANGED
56    
57    
58  #  #
# Line 127  class ZoomInTool(RectTool): Line 144  class ZoomInTool(RectTool):
144      def MouseUp(self, event):      def MouseUp(self, event):
145          if self.dragging:          if self.dragging:
146              Tool.MouseUp(self, event)              Tool.MouseUp(self, event)
147              self.view.FitRectToWindow(self.proj_rect())              sx, sy = self.start
148                cx, cy = self.current
149                if sx == cx or sy == cy:
150                    # 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))
156                else:
157                    # A drag. Zoom in to the rectangle
158                    self.view.FitRectToWindow(self.proj_rect())
159    
160    
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 142  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              self.view.ZoomOutToRect((min(sx, cx), min(sy, cy),              if sx == cx or sy == cy:
174                                       max(sx, cx), max(sy, cy)))                  # Just a mouse click or a degenerate rectangle. Simply
175                    # 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:
181                    # A drag. Zoom out to the rectangle
182                    self.view.ZoomOutToRect((min(sx, cx), min(sy, cy),
183                                             max(sx, cx), max(sy, cy)))
184    
185    
186  class PanTool(Tool):  class PanTool(Tool):
# Line 155  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 168  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 191  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 215  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  class MapCanvas(wxWindow):  
278    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        # 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):      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    
303            # the map displayed in this canvas. Set with SetMap()
304          self.map = None          self.map = None
305    
306            # scale and offset describe the transformation from projected
307            # coordinates to window coordinates.
308          self.scale = 1.0          self.scale = 1.0
309          self.offset = (0, 0)          self.offset = (0, 0)
310    
311            # whether the user is currently dragging the mouse, i.e. moving
312            # the mouse while pressing a mouse button
313          self.dragging = 0          self.dragging = 0
314    
315            # the currently active tool
316          self.tool = None          self.tool = None
317          self.redraw_on_idle = 0  
318            # The current mouse position of the last OnMotion event or None
319            # if the mouse is outside the window.
320            self.current_position = None
321    
322            # the bitmap serving as backing store
323            self.bitmap = None
324    
325            # the selection
326            self.selection = Selection()
327            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
335          EVT_PAINT(self, self.OnPaint)          EVT_PAINT(self, self.OnPaint)
336          EVT_LEFT_DOWN(self, self.OnLeftDown)          EVT_LEFT_DOWN(self, self.OnLeftDown)
337          EVT_LEFT_UP(self, self.OnLeftUp)          EVT_LEFT_UP(self, self.OnLeftUp)
338          EVT_MOTION(self, self.OnMotion)          EVT_MOTION(self, self.OnMotion)
339          wx.EVT_IDLE(self, self.OnIdle)          EVT_LEAVE_WINDOW(self, self.OnLeaveWindow)
340          import main          wx.EVT_SIZE(self, self.OnSize)
341          self.interactor = main.app.interactor  
342          self.interactor.Subscribe(SELECTED_SHAPE, self.shape_selected)      def __del__(self):
343            wxWindow.__del__(self)
344            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 None or not self.map.HasLayers():          clear = self.map is None or not self.map.HasLayers()
380              return  
381          self.redraw_on_idle = 1          #wxBeginBusyCursor()
382    
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
392                # the screen.
393    
394                # XXX it's probably possible to get rid of this. The
395                # background color of the window is already white and the
396                # only thing we may have to do is to call self.Refresh()
397                # with a true argument in the right places.
398                dc.BeginDrawing()
399                dc.Clear()
400                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.
406    
407            # Get the window size.
408          width, height = self.GetSizeTuple()          width, height = self.GetSizeTuple()
         bitmap = wx.wxEmptyBitmap(width, height)  
409    
410            # If self.bitmap's still there, reuse it. Otherwise redraw it
411            if self.bitmap is not None:
412                bitmap = self.bitmap
413            else:
414                bitmap = wx.wxEmptyBitmap(width, height)
415                dc = wx.wxMemoryDC()
416                dc.SelectObject(bitmap)
417                dc.BeginDrawing()
418    
419                # clear the background
420                #dc.SetBrush(wx.wxWHITE_BRUSH)
421                #dc.SetPen(wx.wxTRANSPARENT_PEN)
422                #dc.DrawRectangle(0, 0, width, height)
423                dc.SetBackground(wx.wxWHITE_BRUSH)
424                dc.Clear()
425    
426                selected_layer = self.selection.SelectedLayer()
427                selected_shapes = self.selection.SelectedShapes()
428    
429                # draw the map into the bitmap
430                renderer = ScreenRenderer(dc, self.scale, self.offset)
431    
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()
438                dc.SelectObject(wx.wxNullBitmap)
439                self.bitmap = bitmap
440    
441            # blit the bitmap to the screen
442          dc = wx.wxMemoryDC()          dc = wx.wxMemoryDC()
443          dc.SelectObject(bitmap)          dc.SelectObject(bitmap)
   
         dc.BeginDrawing()  
   
         dc.SetBrush(wx.wxWHITE_BRUSH)  
         dc.SetPen(wx.wxTRANSPARENT_PEN)  
         dc.DrawRectangle(0, 0, width, height)  
   
         if 1: #self.interactor.selected_map is self.map:  
             selected_layer = self.interactor.selected_layer  
             selected_shape = self.interactor.selected_shape  
         else:  
             selected_layer = None  
             selected_shape = None  
               
         renderer = ScreenRenderer(dc, self.scale, self.offset)  
         renderer.RenderMap(self.map, selected_layer, selected_shape)  
   
444          clientdc = wxClientDC(self)          clientdc = wxClientDC(self)
445          clientdc.BeginDrawing()          clientdc.BeginDrawing()
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.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:
500                  self.map.Unsubscribe(channel, self.redraw)                  self.map.Unsubscribe(channel, self.full_redraw)
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.redraw)                  self.map.Subscribe(channel, self.full_redraw)
508              self.map.Subscribe(MAP_PROJECTION_CHANGED, self.projection_changed)              self.map.Subscribe(MAP_PROJECTION_CHANGED, self.projection_changed)
509          self.FitMapToWindow()          self.FitMapToWindow()
510            # force a redraw. If map is not empty, it's already been called
511            # by FitMapToWindow but if map is empty it hasn't been called
512            # yet so we have to explicitly call it.
513            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):
520          self.Refresh(0)          self.Refresh(0)
521    
522        def full_redraw(self, *args):
523            self.bitmap = None
524            self.redraw()
525    
526      def projection_changed(self, *args):      def projection_changed(self, *args):
527          self.FitMapToWindow()          self.FitMapToWindow()
528          self.redraw()          self.full_redraw()
529    
530      def set_view_transform(self, scale, offset):      def set_view_transform(self, scale, offset):
531          self.scale = scale          self.scale = scale
532          self.offset = offset          self.offset = offset
533          self.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 336  class MapCanvas(wxWindow): Line 548  class MapCanvas(wxWindow):
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:
559                # zero width or zero height. Do Nothing
560                return
561          scalex = width / (urx - llx)          scalex = width / (urx - llx)
562          scaley = height / (ury - lly)          scaley = height / (ury - lly)
563          scale = min(scalex, scaley)          scale = min(scalex, scaley)
# Line 346  class MapCanvas(wxWindow): Line 566  class MapCanvas(wxWindow):
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 ZoomFactor(self, factor):      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):
608            """Multiply the zoom by factor and center on center.
609    
610            The optional parameter center is a point in window coordinates
611            that should be centered. If it is omitted, it defaults to the
612            center of the window
613            """
614          width, height = self.GetSizeTuple()          width, height = self.GetSizeTuple()
615          scale = self.scale * factor          scale = self.scale * factor
616          offx, offy = self.offset          offx, offy = self.offset
617          offset = (factor * (offx - width / 2) + width / 2,          if center is not None:
618                    factor * (offy - height / 2) + height / 2)              cx, cy = center
619            else:
620                cx = width / 2
621                cy = height / 2
622            offset = (factor * (offx - cx) + width / 2,
623                      factor * (offy - cy) + height / 2)
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 381  class MapCanvas(wxWindow): Line 644  class MapCanvas(wxWindow):
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):
692            """Return current position of the mouse in projected coordinates.
693    
694            The result is a 2-tuple of floats with the coordinates. If the
695            mouse is not in the window, the result is None.
696            """
697            if self.current_position is not None:
698                x, y = self.current_position
699                return self.win_to_proj(x, y)
700            else:
701                return None
702    
703        def set_current_position(self, event):
704            """Set the current position from event
705    
706            Should be called by all events that contain mouse positions
707            especially EVT_MOTION. The event paramete may be None to
708            indicate the the pointer left the window.
709            """
710            if event is not None:
711                self.current_position = (event.m_x, event.m_y)
712            else:
713                self.current_position = None
714            self.issue(VIEW_POSITION)
715    
716      def OnLeftDown(self, event):      def OnLeftDown(self, event):
717            self.set_current_position(event)
718          if self.tool is not None:          if self.tool is not None:
719              self.drag_dc = wxClientDC(self)              self.drag_dc = wxClientDC(self)
720              self.drag_dc.SetLogicalFunction(wxINVERT)              self.drag_dc.SetLogicalFunction(wxINVERT)
# Line 411  class MapCanvas(wxWindow): Line 723  class MapCanvas(wxWindow):
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):
728          self.ReleaseMouse()          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)
740          if self.dragging:          if self.dragging:
741              self.tool.Hide(self.drag_dc)              self.tool.Hide(self.drag_dc)
742              self.tool.MouseMove(event)              self.tool.MouseMove(event)
743              self.tool.Show(self.drag_dc)              self.tool.Show(self.drag_dc)
744    
745      def OnIdle(self, event):      def OnLeaveWindow(self, event):
746          if self.redraw_on_idle:          self.set_current_position(None)
747              self.do_redraw()  
748          self.redraw_on_idle = 0      def OnSize(self, event):
749            # 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
751            # everything away. E.g. when the window gets smaller, we could
752            # either keep the bitmap or create the new one from the old one.
753            # Even when the window becomes larger some parts of the bitmap
754            # could be reused.
755            self.full_redraw()
756            pass
757    
758      def shape_selected(self, layer, shape):      def shape_selected(self, layer, shape):
759          self.redraw()          """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()
767    
768        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      def find_shape_at(self, px, py, select_labels = 0):          xs = []
781          """Return a tuple shape at point px, py in window coords."""          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
792    
793            Return the shape and the corresponding layer as a tuple (layer,
794            shape).
795    
796            If the optional parameter select_labels is true (default false)
797            search through the labels. If a label is found return it's index
798            as the shape and None as the layer.
799    
800            If the optional parameter searched_layer is given (or not None
801            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:
805              forward = map_proj.Forward              forward = map_proj.Forward
# Line 447  class MapCanvas(wxWindow): Line 811  class MapCanvas(wxWindow):
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)
818                  dc.SetFont(font)                  dc.SetFont(font)
819                  for i in range(len(labels)):                  for i in range(len(labels) - 1, -1, -1):
820                      label = labels[i]                      label = labels[i]
821                      x = label.x                      x = label.x
822                      y = label.y                      y = label.y
# Line 478  class MapCanvas(wxWindow): Line 842  class MapCanvas(wxWindow):
842                          y = y - height/2                          y = y - height/2
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          layers = self.map.Layers()          if searched_layer:
847                layers = [searched_layer]
848            else:
849                layers = self.map.Layers()
850    
851          for layer_index in range(len(layers) - 1, -1, -1):          for layer_index in range(len(layers) - 1, -1, -1):
852              layer = layers[layer_index]              layer = layers[layer_index]
853    
# Line 487  class MapCanvas(wxWindow): Line 855  class MapCanvas(wxWindow):
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()):                  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 511  class MapCanvas(wxWindow): Line 895  class MapCanvas(wxWindow):
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()):                  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 521  class MapCanvas(wxWindow): Line 905  class MapCanvas(wxWindow):
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()):                  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 538  class MapCanvas(wxWindow): Line 922  class MapCanvas(wxWindow):
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)          """\
927          self.interactor.SelectLayerAndShape(layer, shape)          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
937            # to deselect the currently selected layer, so we simply select
938            # the already selected layer again.
939            if layer is None:
940                layer = self.selection.SelectedLayer()
941                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 591  class MapCanvas(wxWindow): Line 999  class MapCanvas(wxWindow):
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.6  
changed lines
  Added in v.926

[email protected]
ViewVC Help
Powered by ViewVC 1.1.26