/[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 23 by bh, Wed Sep 5 13:35:22 2001 UTC revision 1221 by jonathan, Tue Jun 17 15:24:45 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    
35  from wxproj import point_in_polygon_shape, shape_centroid  from wxproj import point_in_polygon_shape, shape_centroid
36    
   
37  from Thuban.Model.messages import MAP_PROJECTION_CHANGED, \  from Thuban.Model.messages import MAP_PROJECTION_CHANGED, \
38       LAYERS_CHANGED, LAYER_LEGEND_CHANGED, LAYER_VISIBILITY_CHANGED       LAYER_PROJECTION_CHANGED, \
39         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          return wx.true          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 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      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    
303            # the map displayed in this canvas. Set with SetMap()
304          self.map = None          self.map = None
305    
306            # current map projection. should only differ from map.projection
307            # when the map's projection is changing and we need access to the
308            # old projection.
309            self.current_map_proj = None
310    
311            # scale and offset describe the transformation from projected
312            # coordinates to window coordinates.
313          self.scale = 1.0          self.scale = 1.0
314          self.offset = (0, 0)          self.offset = (0, 0)
315    
316            # whether the user is currently dragging the mouse, i.e. moving
317            # the mouse while pressing a mouse button
318          self.dragging = 0          self.dragging = 0
319    
320            # the currently active tool
321          self.tool = None          self.tool = None
322          self.redraw_on_idle = 0  
323            # The current mouse position of the last OnMotion event or None
324            # if the mouse is outside the window.
325            self.current_position = None
326    
327            # the bitmap serving as backing store
328            self.bitmap = None
329    
330            # the selection
331            self.selection = Selection()
332            self.selection.Subscribe(SHAPES_SELECTED , self.shape_selected)
333    
334            # keep track of which layers/shapes are selected to make sure we
335            # only redraw when necessary
336            self.last_selected_layer = None
337            self.last_selected_shape = None
338    
339            # subscribe the WX events we're interested in
340          EVT_PAINT(self, self.OnPaint)          EVT_PAINT(self, self.OnPaint)
341          EVT_LEFT_DOWN(self, self.OnLeftDown)          EVT_LEFT_DOWN(self, self.OnLeftDown)
342          EVT_LEFT_UP(self, self.OnLeftUp)          EVT_LEFT_UP(self, self.OnLeftUp)
343          EVT_MOTION(self, self.OnMotion)          EVT_MOTION(self, self.OnMotion)
344          wx.EVT_IDLE(self, self.OnIdle)          EVT_LEAVE_WINDOW(self, self.OnLeaveWindow)
345          self.interactor = interactor          wx.EVT_SIZE(self, self.OnSize)
346          self.interactor.Subscribe(SELECTED_SHAPE, self.shape_selected)  
347        def __del__(self):
348            wxWindow.__del__(self)
349            Publisher.__del__(self)
350    
351        def Subscribe(self, channel, *args):
352            """Extend the inherited method to handle delegated messages.
353    
354            If channel is one of the delegated messages call the appropriate
355            object's Subscribe method. Otherwise just call the inherited
356            method.
357            """
358            if channel in self.delegated_messages:
359                object = getattr(self, self.delegated_messages[channel])
360                object.Subscribe(channel, *args)
361            else:
362                Publisher.Subscribe(self, channel, *args)
363    
364        def Unsubscribe(self, channel, *args):
365            """Extend the inherited method to handle delegated messages.
366    
367            If channel is one of the delegated messages call the appropriate
368            object's Unsubscribe method. Otherwise just call the inherited
369            method.
370            """
371            if channel in self.delegated_messages:
372                object = getattr(self, self.delegated_messages[channel])
373                object.Unsubscribe(channel, *args)
374            else:
375                Publisher.Unsubscribe(self, channel, *args)
376    
377        def __getattr__(self, attr):
378            if attr in self.delegated_methods:
379                return getattr(getattr(self, self.delegated_methods[attr]), attr)
380            raise AttributeError(attr)
381    
382      def OnPaint(self, event):      def OnPaint(self, event):
383          dc = wxPaintDC(self)          dc = wxPaintDC(self)
384          if self.map is None or not self.map.HasLayers():          clear = self.map is None or not self.map.HasLayers()
385              return  
386          self.redraw_on_idle = 1          wxBeginBusyCursor()
387            try:
388                if not clear:
389                    self.do_redraw()
390                    try:
391                        pass
392                    except:
393                        print "Error during drawing:", sys.exc_info()[0]
394                        clear = True
395    
396                if clear:
397                    # If we've got no map or if the map is empty, simply clear
398                    # the screen.
399    
400                    # XXX it's probably possible to get rid of this. The
401                    # background color of the window is already white and the
402                    # only thing we may have to do is to call self.Refresh()
403                    # with a true argument in the right places.
404                    dc.BeginDrawing()
405                    dc.Clear()
406                    dc.EndDrawing()
407            finally:
408                wxEndBusyCursor()
409    
410      def do_redraw(self):      def do_redraw(self):
411            # This should only be called if we have a non-empty map.
412    
413            # Get the window size.
414          width, height = self.GetSizeTuple()          width, height = self.GetSizeTuple()
         bitmap = wx.wxEmptyBitmap(width, height)  
415    
416            # If self.bitmap's still there, reuse it. Otherwise redraw it
417            if self.bitmap is not None:
418                bitmap = self.bitmap
419            else:
420                bitmap = wx.wxEmptyBitmap(width, height)
421                dc = wx.wxMemoryDC()
422                dc.SelectObject(bitmap)
423                dc.BeginDrawing()
424    
425                # clear the background
426                #dc.SetBrush(wx.wxWHITE_BRUSH)
427                #dc.SetPen(wx.wxTRANSPARENT_PEN)
428                #dc.DrawRectangle(0, 0, width, height)
429                dc.SetBackground(wx.wxWHITE_BRUSH)
430                dc.Clear()
431    
432                selected_layer = self.selection.SelectedLayer()
433                selected_shapes = self.selection.SelectedShapes()
434    
435                # draw the map into the bitmap
436                renderer = ScreenRenderer(dc, self.scale, self.offset)
437    
438                # Pass the entire bitmap as update region to the renderer.
439                # We're redrawing the whole bitmap, after all.
440                renderer.RenderMap(self.map, (0, 0, width, height),
441                                   selected_layer, selected_shapes)
442    
443                dc.EndDrawing()
444                dc.SelectObject(wx.wxNullBitmap)
445                self.bitmap = bitmap
446    
447            # blit the bitmap to the screen
448          dc = wx.wxMemoryDC()          dc = wx.wxMemoryDC()
449          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)  
   
450          clientdc = wxClientDC(self)          clientdc = wxClientDC(self)
451          clientdc.BeginDrawing()          clientdc.BeginDrawing()
452          clientdc.Blit(0, 0, width, height, dc, 0, 0)          clientdc.Blit(0, 0, width, height, dc, 0, 0)
453          clientdc.EndDrawing()          clientdc.EndDrawing()
454    
455        def Export(self):
456            if self.scale == 0:
457                return
458    
459            if hasattr(self, "export_path"):
460                export_path = self.export_path
461            else:
462                export_path="."
463            dlg = wxFileDialog(self, _("Export Map"), export_path, "",
464                               "Enhanced Metafile (*.wmf)|*.wmf",
465                               wxSAVE|wxOVERWRITE_PROMPT)
466            if dlg.ShowModal() == wxID_OK:
467                self.export_path = os.path.dirname(dlg.GetPath())
468                dc = wxMetaFileDC(dlg.GetPath())
469        
470                scale, offset, mapregion = OutputTransform(self.scale,
471                                                           self.offset,
472                                                           self.GetSizeTuple(),
473                                                           dc.GetSizeTuple())
474    
475                selected_layer = self.selection.SelectedLayer()
476                selected_shapes = self.selection.SelectedShapes()
477    
478                renderer = ExportRenderer(dc, scale, offset)
479    
480                # Pass the entire bitmap as update region to the renderer.
481                # We're redrawing the whole bitmap, after all.
482                width, height = self.GetSizeTuple()
483                renderer.RenderMap(self.map,
484                                    (0,0,
485                                        (width/self.scale)*scale,
486                                        (height/self.scale)*scale),
487                                    mapregion,
488                                    selected_layer, selected_shapes)
489                dc.EndDrawing()
490                dc.Close()
491            dlg.Destroy()
492            
493      def Print(self):      def Print(self):
494          printer = wx.wxPrinter()          printer = wx.wxPrinter()
495          printout = MapPrintout(self.map)          width, height = self.GetSizeTuple()
496          printer.Print(self, printout, wx.true)          selected_layer = self.selection.SelectedLayer()
497          printout.Destroy()          selected_shapes = self.selection.SelectedShapes()
498                    
499            printout = MapPrintout(self, self.map, (0, 0, width, height),
500                                   selected_layer, selected_shapes)
501            printer.Print(self, printout, True)
502            printout.Destroy()
503    
504      def SetMap(self, map):      def SetMap(self, map):
505          redraw_channels = (LAYERS_CHANGED, LAYER_LEGEND_CHANGED,          redraw_channels = (MAP_LAYERS_CHANGED, LAYER_CHANGED,
506                             LAYER_VISIBILITY_CHANGED)                             LAYER_VISIBILITY_CHANGED)
507          if self.map is not None:          if self.map is not None:
508              for channel in redraw_channels:              for channel in redraw_channels:
509                  self.map.Unsubscribe(channel, self.redraw)                  self.map.Unsubscribe(channel, self.full_redraw)
510              self.map.Unsubscribe(MAP_PROJECTION_CHANGED,              self.map.Unsubscribe(MAP_PROJECTION_CHANGED,
511                                   self.projection_changed)                                   self.map_projection_changed)
512                self.map.Unsubscribe(LAYER_PROJECTION_CHANGED,
513                                     self.layer_projection_changed)
514          self.map = map          self.map = map
515            self.current_map_proj = self.map.GetProjection()
516            self.selection.ClearSelection()
517          if self.map is not None:          if self.map is not None:
518              for channel in redraw_channels:              for channel in redraw_channels:
519                  self.map.Subscribe(channel, self.redraw)                  self.map.Subscribe(channel, self.full_redraw)
520              self.map.Subscribe(MAP_PROJECTION_CHANGED, self.projection_changed)              self.map.Subscribe(MAP_PROJECTION_CHANGED, self.map_projection_changed)
521                self.map.Subscribe(LAYER_PROJECTION_CHANGED, self.layer_projection_changed)
522          self.FitMapToWindow()          self.FitMapToWindow()
523            # force a redraw. If map is not empty, it's already been called
524            # by FitMapToWindow but if map is empty it hasn't been called
525            # yet so we have to explicitly call it.
526            self.full_redraw()
527    
528      def Map(self):      def Map(self):
529            """Return the map displayed by this canvas"""
530          return self.map          return self.map
531    
532      def redraw(self, *args):      def redraw(self, *args):
533          self.Refresh(0)          self.Refresh(0)
534    
535      def projection_changed(self, *args):      def full_redraw(self, *args):
536          self.FitMapToWindow()          self.bitmap = None
537          self.redraw()          self.redraw()
538    
539        def map_projection_changed(self, *args):
540    
541            proj = self.current_map_proj
542            self.current_map_proj = self.map.GetProjection()
543    
544            bbox = None
545    
546            if proj is not None and self.current_map_proj is not None:
547                width, height = self.GetSizeTuple()
548                llx, lly = self.win_to_proj(0, height)
549                urx, ury = self.win_to_proj(width, 0)
550                bbox = proj.Inverse(llx, lly) + proj.Inverse(urx, ury)
551                bbox = self.current_map_proj.ForwardBBox(bbox)
552    
553            if bbox is not None:
554                self.FitRectToWindow(bbox)
555            else:
556                self.FitMapToWindow()
557    
558            self.full_redraw()
559    
560        def layer_projection_changed(self, *args):
561            self.full_redraw()
562    
563      def set_view_transform(self, scale, offset):      def set_view_transform(self, scale, offset):
564            # width/height of the projected bbox
565            llx, lly, urx, ury = bbox = self.map.ProjectedBoundingBox()
566            pwidth = float(urx - llx)
567            pheight = float(ury - lly)
568    
569            # width/height of the window
570            wwidth, wheight = self.GetSizeTuple()
571    
572            # The window's center in projected coordinates assuming the new
573            # scale/offset
574            pcenterx = (wwidth/2 - offset[0]) / scale
575            pcentery = (offset[1] - wheight/2) / scale
576    
577            # The window coordinates used when drawing the shapes must fit
578            # into 16bit signed integers.
579            max_len = max(pwidth, pheight)
580            if max_len:
581                max_scale = 32000.0 / max_len
582            else:
583                # FIXME: What to do in this case? The bbox is effectively
584                # empty so any scale should work.
585                max_scale = scale
586    
587            # The minimal scale is somewhat arbitrarily set to half that of
588            # the bbox fit into the window
589            scales = []
590            if pwidth:
591                scales.append(wwidth / pwidth)
592            if pheight:
593                scales.append(wheight / pheight)
594            if scales:
595                min_scale = 0.5 * min(scales)
596            else:
597                min_scale = scale
598    
599            if scale > max_scale:
600                scale = max_scale
601            elif scale < min_scale:
602                scale = min_scale
603    
604          self.scale = scale          self.scale = scale
605          self.offset = offset  
606          self.redraw()          # determine new offset to preserve the center
607            self.offset = (wwidth/2 - scale * pcenterx,
608                           wheight/2 + scale * pcentery)
609            self.full_redraw()
610            self.issue(SCALE_CHANGED, scale)
611    
612      def proj_to_win(self, x, y):      def proj_to_win(self, x, y):
613          """\          """\
614          Return the point in  window coords given by projected coordinates x y          Return the point in  window coords given by projected coordinates x y
615          """          """
616            if self.scale == 0:
617                return (0, 0)
618    
619          offx, offy = self.offset          offx, offy = self.offset
620          return (self.scale * x + offx, -self.scale * y + offy)          return (self.scale * x + offx, -self.scale * y + offy)
621    
# Line 331  class MapCanvas(wxWindow): Line 623  class MapCanvas(wxWindow):
623          """\          """\
624          Return the point in projected coordinates given by window coords x y          Return the point in projected coordinates given by window coords x y
625          """          """
626            if self.scale == 0:
627                return (0, 0)
628    
629          offx, offy = self.offset          offx, offy = self.offset
630          return ((x - offx) / self.scale, (offy - y) / self.scale)          return ((x - offx) / self.scale, (offy - y) / self.scale)
631    
632      def FitRectToWindow(self, rect):      def FitRectToWindow(self, rect):
633            """Fit the rectangular region given by rect into the window.
634    
635            Set scale so that rect (in projected coordinates) just fits into
636            the window and center it.
637            """
638          width, height = self.GetSizeTuple()          width, height = self.GetSizeTuple()
639          llx, lly, urx, ury = rect          llx, lly, urx, ury = rect
640            if llx == urx or lly == ury:
641                # zero width or zero height. Do Nothing
642                return
643          scalex = width / (urx - llx)          scalex = width / (urx - llx)
644          scaley = height / (ury - lly)          scaley = height / (ury - lly)
645          scale = min(scalex, scaley)          scale = min(scalex, scaley)
# Line 345  class MapCanvas(wxWindow): Line 648  class MapCanvas(wxWindow):
648          self.set_view_transform(scale, (offx, offy))          self.set_view_transform(scale, (offx, offy))
649    
650      def FitMapToWindow(self):      def FitMapToWindow(self):
651          """\          """Fit the map to the window
652          Set the scale and offset so that the map is centered in the  
653          window          Set the scale so that the map fits exactly into the window and
654            center it in the window.
655          """          """
656          bbox = self.map.ProjectedBoundingBox()          bbox = self.map.ProjectedBoundingBox()
657          if bbox is not None:          if bbox is not None:
658              self.FitRectToWindow(bbox)              self.FitRectToWindow(bbox)
659    
660      def ZoomFactor(self, factor):      def FitLayerToWindow(self, layer):
661          width, height = self.GetSizeTuple()          """Fit the given layer to the window.
662          scale = self.scale * factor  
663          offx, offy = self.offset          Set the scale so that the layer fits exactly into the window and
664          offset = (factor * (offx - width / 2) + width / 2,          center it in the window.
665                    factor * (offy - height / 2) + height / 2)          """
666          self.set_view_transform(scale, offset)          
667            bbox = layer.LatLongBoundingBox()
668            if bbox is not None:
669                proj = self.map.GetProjection()
670                if proj is not None:
671                    bbox = proj.ForwardBBox(bbox)
672    
673                if bbox is not None:
674                    self.FitRectToWindow(bbox)
675    
676        def FitSelectedToWindow(self):
677            layer = self.selection.SelectedLayer()
678            shapes = self.selection.SelectedShapes()
679    
680            bbox = layer.ShapesBoundingBox(shapes)
681            if bbox is not None:
682                proj = self.map.GetProjection()
683                if proj is not None:
684                    bbox = proj.ForwardBBox(bbox)
685    
686                if bbox is not None:
687                    if len(shapes) == 1 and layer.ShapeType() == SHAPETYPE_POINT:
688                        self.ZoomFactor(1, self.proj_to_win(bbox[0], bbox[1]))
689                    else:
690                        self.FitRectToWindow(bbox)
691    
692        def ZoomFactor(self, factor, center = None):
693            """Multiply the zoom by factor and center on center.
694    
695            The optional parameter center is a point in window coordinates
696            that should be centered. If it is omitted, it defaults to the
697            center of the window
698            """
699            if self.scale > 0:
700                width, height = self.GetSizeTuple()
701                scale = self.scale * factor
702                offx, offy = self.offset
703                if center is not None:
704                    cx, cy = center
705                else:
706                    cx = width / 2
707                    cy = height / 2
708                offset = (factor * (offx - cx) + width / 2,
709                        factor * (offy - cy) + height / 2)
710                self.set_view_transform(scale, offset)
711    
712      def ZoomOutToRect(self, rect):      def ZoomOutToRect(self, rect):
713          # rect is given in window coordinates          """Zoom out to fit the currently visible region into rect.
714    
715            The rect parameter is given in window coordinates
716            """
717          # determine the bbox of the displayed region in projected          # determine the bbox of the displayed region in projected
718          # coordinates          # coordinates
719          width, height = self.GetSizeTuple()          width, height = self.GetSizeTuple()
# Line 380  class MapCanvas(wxWindow): Line 730  class MapCanvas(wxWindow):
730          self.set_view_transform(scale, (offx, offy))          self.set_view_transform(scale, (offx, offy))
731    
732      def Translate(self, dx, dy):      def Translate(self, dx, dy):
733            """Move the map by dx, dy pixels"""
734          offx, offy = self.offset          offx, offy = self.offset
735          self.set_view_transform(self.scale, (offx + dx, offy + dy))          self.set_view_transform(self.scale, (offx + dx, offy + dy))
736    
737        def SelectTool(self, tool):
738            """Make tool the active tool.
739    
740            The parameter should be an instance of Tool or None to indicate
741            that no tool is active.
742            """
743            self.tool = tool
744    
745      def ZoomInTool(self):      def ZoomInTool(self):
746          self.tool = ZoomInTool(self)          """Start the zoom in tool"""
747            self.SelectTool(ZoomInTool(self))
748    
749      def ZoomOutTool(self):      def ZoomOutTool(self):
750          self.tool = ZoomOutTool(self)          """Start the zoom out tool"""
751            self.SelectTool(ZoomOutTool(self))
752    
753      def PanTool(self):      def PanTool(self):
754          self.tool = PanTool(self)          """Start the pan tool"""
755            self.SelectTool(PanTool(self))
756            #img = resource.GetImageResource("pan", wxBITMAP_TYPE_XPM)
757            #bmp = resource.GetBitmapResource("pan", wxBITMAP_TYPE_XPM)
758            #print bmp
759            #img = wxImageFromBitmap(bmp)
760            #print img
761            #cur = wxCursor(img)
762            #print cur
763            #self.SetCursor(cur)
764    
765      def IdentifyTool(self):      def IdentifyTool(self):
766          self.tool = IdentifyTool(self)          """Start the identify tool"""
767            self.SelectTool(IdentifyTool(self))
768    
769      def LabelTool(self):      def LabelTool(self):
770          self.tool = LabelTool(self)          """Start the label tool"""
771            self.SelectTool(LabelTool(self))
772    
773      def CurrentTool(self):      def CurrentTool(self):
774            """Return the name of the current tool or None if no tool is active"""
775          return self.tool and self.tool.Name() or None          return self.tool and self.tool.Name() or None
776    
777        def CurrentPosition(self):
778            """Return current position of the mouse in projected coordinates.
779    
780            The result is a 2-tuple of floats with the coordinates. If the
781            mouse is not in the window, the result is None.
782            """
783            if self.current_position is not None:
784                x, y = self.current_position
785                return self.win_to_proj(x, y)
786            else:
787                return None
788    
789        def set_current_position(self, event):
790            """Set the current position from event
791    
792            Should be called by all events that contain mouse positions
793            especially EVT_MOTION. The event paramete may be None to
794            indicate the the pointer left the window.
795            """
796            if event is not None:
797                self.current_position = (event.m_x, event.m_y)
798            else:
799                self.current_position = None
800            self.issue(VIEW_POSITION)
801    
802      def OnLeftDown(self, event):      def OnLeftDown(self, event):
803            self.set_current_position(event)
804          if self.tool is not None:          if self.tool is not None:
805              self.drag_dc = wxClientDC(self)              self.drag_dc = wxClientDC(self)
806              self.drag_dc.SetLogicalFunction(wxINVERT)              self.drag_dc.SetLogicalFunction(wxINVERT)
# Line 410  class MapCanvas(wxWindow): Line 809  class MapCanvas(wxWindow):
809              self.tool.MouseDown(event)              self.tool.MouseDown(event)
810              self.tool.Show(self.drag_dc)              self.tool.Show(self.drag_dc)
811              self.dragging = 1              self.dragging = 1
812            
813      def OnLeftUp(self, event):      def OnLeftUp(self, event):
814          self.ReleaseMouse()          self.set_current_position(event)
815          if self.dragging:          if self.dragging:
816              self.tool.Hide(self.drag_dc)              self.ReleaseMouse()
817              self.tool.MouseUp(event)              try:
818              self.drag_dc = None                  self.tool.Hide(self.drag_dc)
819          self.dragging = 0                  self.tool.MouseUp(event)
820                finally:
821                    self.drag_dc = None
822                    self.dragging = 0
823    
824      def OnMotion(self, event):      def OnMotion(self, event):
825            self.set_current_position(event)
826          if self.dragging:          if self.dragging:
827              self.tool.Hide(self.drag_dc)              self.tool.Hide(self.drag_dc)
828              self.tool.MouseMove(event)              self.tool.MouseMove(event)
829              self.tool.Show(self.drag_dc)              self.tool.Show(self.drag_dc)
830    
831      def OnIdle(self, event):      def OnLeaveWindow(self, event):
832          if self.redraw_on_idle:          self.set_current_position(None)
833              self.do_redraw()  
834          self.redraw_on_idle = 0      def OnSize(self, event):
835            # the window's size has changed. We have to get a new bitmap. If
836            # we want to be clever we could try to get by without throwing
837            # everything away. E.g. when the window gets smaller, we could
838            # either keep the bitmap or create the new one from the old one.
839            # Even when the window becomes larger some parts of the bitmap
840            # could be reused.
841            self.full_redraw()
842            pass
843    
844      def shape_selected(self, layer, shape):      def shape_selected(self, layer, shape):
845          self.redraw()          """Receiver for the SHAPES_SELECTED messages. Redraw the map."""
846            # The selection object takes care that it only issues
847            # SHAPES_SELECTED messages when the set of selected shapes has
848            # actually changed, so we can do a full redraw unconditionally.
849            # FIXME: We should perhaps try to limit the redraw to the are
850            # actually covered by the shapes before and after the selection
851            # change.
852            self.full_redraw()
853    
854        def unprojected_rect_around_point(self, x, y, dist):
855            """return a rect dist pixels around (x, y) in unprojected corrdinates
856    
857            The return value is a tuple (minx, miny, maxx, maxy) suitable a
858            parameter to a layer's ShapesInRegion method.
859            """
860            map_proj = self.map.projection
861            if map_proj is not None:
862                inverse = map_proj.Inverse
863            else:
864                inverse = None
865    
866            xs = []
867            ys = []
868            for dx, dy in ((-1, -1), (1, -1), (1, 1), (-1, 1)):
869                px, py = self.win_to_proj(x + dist * dx, y + dist * dy)
870                if inverse:
871                    px, py = inverse(px, py)
872                xs.append(px)
873                ys.append(py)
874            return (min(xs), min(ys), max(xs), max(ys))
875    
876        def find_shape_at(self, px, py, select_labels = 0, searched_layer = None):
877            """Determine the shape at point px, py in window coords
878    
879            Return the shape and the corresponding layer as a tuple (layer,
880            shape).
881    
882            If the optional parameter select_labels is true (default false)
883            search through the labels. If a label is found return it's index
884            as the shape and None as the layer.
885    
886      def find_shape_at(self, px, py, select_labels = 0):          If the optional parameter searched_layer is given (or not None
887          """Return a tuple shape at point px, py in window coords."""          which it defaults to), only search in that layer.
888            """
889          map_proj = self.map.projection          map_proj = self.map.projection
890          if map_proj is not None:          if map_proj is not None:
891              forward = map_proj.Forward              forward = map_proj.Forward
# Line 442  class MapCanvas(wxWindow): Line 893  class MapCanvas(wxWindow):
893              forward = None              forward = None
894    
895          scale = self.scale          scale = self.scale
896    
897            if scale == 0:
898                return None, None
899    
900          offx, offy = self.offset          offx, offy = self.offset
901    
902          if select_labels:          if select_labels:
903              labels = self.map.LabelLayer().Labels()              labels = self.map.LabelLayer().Labels()
904                
905              if labels:              if labels:
906                  dc = wxClientDC(self)                  dc = wxClientDC(self)
907                  font = wxFont(10, wx.wxSWISS, wx.wxNORMAL, wx.wxNORMAL)                  font = wxFont(10, wx.wxSWISS, wx.wxNORMAL, wx.wxNORMAL)
908                  dc.SetFont(font)                  dc.SetFont(font)
909                  for i in range(len(labels)):                  for i in range(len(labels) - 1, -1, -1):
910                      label = labels[i]                      label = labels[i]
911                      x = label.x                      x = label.x
912                      y = label.y                      y = label.y
# Line 477  class MapCanvas(wxWindow): Line 932  class MapCanvas(wxWindow):
932                          y = y - height/2                          y = y - height/2
933                      if x <= px < x + width and y <= py <= y + height:                      if x <= px < x + width and y <= py <= y + height:
934                          return None, i                          return None, i
935                    
936          layers = self.map.Layers()          if searched_layer:
937                layers = [searched_layer]
938            else:
939                layers = self.map.Layers()
940    
941          for layer_index in range(len(layers) - 1, -1, -1):          for layer_index in range(len(layers) - 1, -1, -1):
942              layer = layers[layer_index]              layer = layers[layer_index]
943    
# Line 486  class MapCanvas(wxWindow): Line 945  class MapCanvas(wxWindow):
945              if not layer.Visible():              if not layer.Visible():
946                  continue                  continue
947    
948              filled = layer.fill is not None              filled = layer.GetClassification().GetDefaultFill() \
949              stroked = layer.stroke is not None                       is not Color.Transparent
950                                stroked = layer.GetClassification().GetDefaultLineColor() \
951                          is not Color.Transparent
952    
953              layer_proj = layer.projection              layer_proj = layer.projection
954              if layer_proj is not None:              if layer_proj is not None:
955                  inverse = layer_proj.Inverse                  inverse = layer_proj.Inverse
956              else:              else:
957                  inverse = None                  inverse = None
958                    
959              shapetype = layer.ShapeType()              shapetype = layer.ShapeType()
960    
961              select_shape = -1              select_shape = -1
962    
963                # Determine the ids of the shapes that overlap a tiny area
964                # around the point. For layers containing points we have to
965                # choose a larger size of the box we're testing agains so
966                # that we take the size of the markers into account
967                # FIXME: Once the markers are more flexible this part has to
968                # become more flexible too, of course
969                if shapetype == SHAPETYPE_POINT:
970                    box = self.unprojected_rect_around_point(px, py, 5)
971                else:
972                    box = self.unprojected_rect_around_point(px, py, 1)
973                shape_ids = layer.ShapesInRegion(box)
974                shape_ids.reverse()
975    
976              if shapetype == SHAPETYPE_POLYGON:              if shapetype == SHAPETYPE_POLYGON:
977                  for i in range(layer.NumShapes()):                  for i in shape_ids:
978                      result = point_in_polygon_shape(layer.shapefile.cobject(),                      shapefile = layer.ShapeStore().Shapefile().cobject()
979                                                      i,                      result = point_in_polygon_shape(shapefile, i,
980                                                      filled, stroked,                                                      filled, stroked,
981                                                      map_proj, layer_proj,                                                      map_proj, layer_proj,
982                                                      scale, -scale, offx, offy,                                                      scale, -scale, offx, offy,
# Line 510  class MapCanvas(wxWindow): Line 985  class MapCanvas(wxWindow):
985                          select_shape = i                          select_shape = i
986                          break                          break
987              elif shapetype == SHAPETYPE_ARC:              elif shapetype == SHAPETYPE_ARC:
988                  for i in range(layer.NumShapes()):                  for i in shape_ids:
989                      result = point_in_polygon_shape(layer.shapefile.cobject(),                      shapefile = layer.ShapeStore().Shapefile().cobject()
990                        result = point_in_polygon_shape(shapefile,
991                                                      i, 0, 1,                                                      i, 0, 1,
992                                                      map_proj, layer_proj,                                                      map_proj, layer_proj,
993                                                      scale, -scale, offx, offy,                                                      scale, -scale, offx, offy,
# Line 520  class MapCanvas(wxWindow): Line 996  class MapCanvas(wxWindow):
996                          select_shape = i                          select_shape = i
997                          break                          break
998              elif shapetype == SHAPETYPE_POINT:              elif shapetype == SHAPETYPE_POINT:
999                  for i in range(layer.NumShapes()):                  for i in shape_ids:
1000                      shape = layer.Shape(i)                      shape = layer.Shape(i)
1001                      x, y = shape.Points()[0]                      x, y = shape.Points()[0]
1002                      if inverse:                      if inverse:
# Line 537  class MapCanvas(wxWindow): Line 1013  class MapCanvas(wxWindow):
1013                  return layer, select_shape                  return layer, select_shape
1014          return None, None          return None, None
1015    
1016      def SelectShapeAt(self, x, y):      def SelectShapeAt(self, x, y, layer = None):
1017          layer, shape = self.find_shape_at(x, y)          """\
1018          self.interactor.SelectLayerAndShape(layer, shape)          Select and return the shape and its layer at window position (x, y)
1019    
1020            If layer is given, only search in that layer. If no layer is
1021            given, search through all layers.
1022    
1023            Return a tuple (layer, shapeid). If no shape is found, return
1024            (None, None).
1025            """
1026            layer, shape = result = self.find_shape_at(x, y, searched_layer=layer)
1027            # If layer is None, then shape will also be None. We don't want
1028            # to deselect the currently selected layer, so we simply select
1029            # the already selected layer again.
1030            if layer is None:
1031                layer = self.selection.SelectedLayer()
1032                shapes = []
1033            else:
1034                shapes = [shape]
1035            self.selection.SelectShapes(layer, shapes)
1036            return result
1037    
1038      def LabelShapeAt(self, x, y):      def LabelShapeAt(self, x, y):
1039            """Add or remove a label at window position x, y.
1040    
1041            If there's a label at the given position, remove it. Otherwise
1042            determine the shape at the position, run the label dialog and
1043            unless the user cancels the dialog, add a laber.
1044            """
1045          ox = x; oy = y          ox = x; oy = y
1046          label_layer = self.map.LabelLayer()          label_layer = self.map.LabelLayer()
1047          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 549  class MapCanvas(wxWindow): Line 1049  class MapCanvas(wxWindow):
1049              # a label was selected              # a label was selected
1050              label_layer.RemoveLabel(shape_index)              label_layer.RemoveLabel(shape_index)
1051          elif layer is not None:          elif layer is not None:
1052              text = labeldialog.run_label_dialog(self, layer.table, shape_index)              text = labeldialog.run_label_dialog(self,
1053                                                    layer.ShapeStore().Table(),
1054                                                    shape_index)
1055              if text:              if text:
1056                  proj = self.map.projection                  proj = self.map.projection
1057                  if proj is not None:                  if proj is not None:
# Line 564  class MapCanvas(wxWindow): Line 1066  class MapCanvas(wxWindow):
1066    
1067                  shapetype = layer.ShapeType()                  shapetype = layer.ShapeType()
1068                  if shapetype == SHAPETYPE_POLYGON:                  if shapetype == SHAPETYPE_POLYGON:
1069                      x, y = shape_centroid(layer.shapefile.cobject(),                      shapefile = layer.ShapeStore().Shapefile().cobject()
1070                                            shape_index,                      x, y = shape_centroid(shapefile, shape_index,
1071                                            map_proj, layer_proj, 1, 1, 0, 0)                                            map_proj, layer_proj, 1, 1, 0, 0)
1072                      if map_proj is not None:                      if map_proj is not None:
1073                          x, y = map_proj.Inverse(x, y)                          x, y = map_proj.Inverse(x, y)
# Line 590  class MapCanvas(wxWindow): Line 1092  class MapCanvas(wxWindow):
1092                      valign = ALIGN_CENTER                      valign = ALIGN_CENTER
1093                  label_layer.AddLabel(x, y, text,                  label_layer.AddLabel(x, y, text,
1094                                       halign = halign, valign = valign)                                       halign = halign, valign = valign)
1095    
1096    def OutputTransform(canvas_scale, canvas_offset, canvas_size, device_extend):
1097        """Calculate dimensions to transform canvas content to output device."""
1098        width, height = device_extend
1099    
1100        # Only 80 % of the with are available for the map
1101        width = width * 0.8
1102    
1103        # Define the distance of the map from DC border
1104        distance = 20
1105    
1106        if height < width:
1107            # landscape
1108            map_height = height - 2*distance
1109            map_width = map_height
1110        else:
1111            # portrait, recalibrate width (usually the legend width is too
1112            # small
1113            width = width * 0.9
1114            map_height = width - 2*distance
1115            map_width = map_height
1116        
1117        mapregion = (distance, distance,
1118                     distance+map_width, distance+map_height)
1119    
1120        canvas_width, canvas_height = canvas_size
1121        
1122        scalex = map_width / (canvas_width/canvas_scale)
1123        scaley = map_height / (canvas_height/canvas_scale)
1124        scale = min(scalex, scaley)
1125        canvas_offx, canvas_offy = canvas_offset
1126        offx = scale*canvas_offx/canvas_scale
1127        offy = scale*canvas_offy/canvas_scale
1128    
1129        return scale, (offx, offy), mapregion
Legend:
Removed from v.23  
changed lines
  Added in v.1221
[email protected]
 ViewVC Help
Powered by ViewVC 1.1.26