/[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 1219 by bh, Mon Jun 16 17:42:54 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       MAP_LAYERS_CHANGED, LAYER_CHANGED, LAYER_VISIBILITY_CHANGED
39  from Thuban.Model.layer import SHAPETYPE_POLYGON, SHAPETYPE_ARC, \  from Thuban.Model.layer import SHAPETYPE_POLYGON, SHAPETYPE_ARC, \
40       SHAPETYPE_POINT       SHAPETYPE_POINT
41  from Thuban.Model.label import ALIGN_CENTER, ALIGN_TOP, ALIGN_BOTTOM, \  from Thuban.Model.label import ALIGN_CENTER, ALIGN_TOP, ALIGN_BOTTOM, \
42       ALIGN_LEFT, ALIGN_RIGHT       ALIGN_LEFT, ALIGN_RIGHT
43    from Thuban.Lib.connector import Publisher
44    from Thuban.Model.color import Color
45    
46    import resource
47    
48  from renderer import ScreenRenderer, PrinterRender  from selection import Selection
49    from renderer import ScreenRenderer, ExportRenderer, PrinterRenderer
50    
51  import labeldialog  import labeldialog
52    
53  from messages import SELECTED_SHAPE  from messages import LAYER_SELECTED, SHAPES_SELECTED, VIEW_POSITION, \
54                         SCALE_CHANGED
55    
56    
57  #  #
# Line 127  class ZoomInTool(RectTool): Line 143  class ZoomInTool(RectTool):
143      def MouseUp(self, event):      def MouseUp(self, event):
144          if self.dragging:          if self.dragging:
145              Tool.MouseUp(self, event)              Tool.MouseUp(self, event)
146              self.view.FitRectToWindow(self.proj_rect())              sx, sy = self.start
147                cx, cy = self.current
148                if sx == cx or sy == cy:
149                    # Just a mouse click or a degenerate rectangle. Simply
150                    # zoom in by a factor of two
151                    # FIXME: For a click this is the desired behavior but should we
152                    # really do this for degenrate rectagles as well or
153                    # should we ignore them?
154                    self.view.ZoomFactor(2, center = (cx, cy))
155                else:
156                    # A drag. Zoom in to the rectangle
157                    self.view.FitRectToWindow(self.proj_rect())
158    
159    
160  class ZoomOutTool(RectTool):  class ZoomOutTool(RectTool):
161    
162      """The Zoom-Out Tool"""      """The Zoom-Out Tool"""
163        
164      def Name(self):      def Name(self):
165          return "ZoomOutTool"          return "ZoomOutTool"
166    
# Line 142  class ZoomOutTool(RectTool): Line 169  class ZoomOutTool(RectTool):
169              Tool.MouseUp(self, event)              Tool.MouseUp(self, event)
170              sx, sy = self.start              sx, sy = self.start
171              cx, cy = self.current              cx, cy = self.current
172              self.view.ZoomOutToRect((min(sx, cx), min(sy, cy),              if sx == cx or sy == cy:
173                                       max(sx, cx), max(sy, cy)))                  # Just a mouse click or a degenerate rectangle. Simply
174                    # zoom out by a factor of two.
175                    # FIXME: For a click this is the desired behavior but should we
176                    # really do this for degenrate rectagles as well or
177                    # should we ignore them?
178                    self.view.ZoomFactor(0.5, center = (cx, cy))
179                else:
180                    # A drag. Zoom out to the rectangle
181                    self.view.ZoomOutToRect((min(sx, cx), min(sy, cy),
182                                             max(sx, cx), max(sy, cy)))
183    
184    
185  class PanTool(Tool):  class PanTool(Tool):
# Line 155  class PanTool(Tool): Line 191  class PanTool(Tool):
191    
192      def MouseMove(self, event):      def MouseMove(self, event):
193          if self.dragging:          if self.dragging:
             x0, y0 = self.current  
194              Tool.MouseMove(self, event)              Tool.MouseMove(self, event)
195                sx, sy = self.start
196              x, y = self.current              x, y = self.current
197              width, height = self.view.GetSizeTuple()              width, height = self.view.GetSizeTuple()
198    
199                bitmapdc = wx.wxMemoryDC()
200                bitmapdc.SelectObject(self.view.bitmap)
201    
202              dc = self.view.drag_dc              dc = self.view.drag_dc
203              dc.Blit(0, 0, width, height, dc, x0 - x, y0 - y)              dc.Blit(0, 0, width, height, bitmapdc, sx - x, sy - y)
204    
205      def MouseUp(self, event):      def MouseUp(self, event):
206          if self.dragging:          if self.dragging:
# Line 168  class PanTool(Tool): Line 208  class PanTool(Tool):
208              sx, sy = self.start              sx, sy = self.start
209              cx, cy = self.current              cx, cy = self.current
210              self.view.Translate(cx - sx, cy - sy)              self.view.Translate(cx - sx, cy - sy)
211            
212  class IdentifyTool(Tool):  class IdentifyTool(Tool):
213    
214      """The "Identify" Tool"""      """The "Identify" Tool"""
215        
216      def Name(self):      def Name(self):
217          return "IdentifyTool"          return "IdentifyTool"
218    
# Line 191  class LabelTool(Tool): Line 231  class LabelTool(Tool):
231          self.view.LabelShapeAt(event.m_x, event.m_y)          self.view.LabelShapeAt(event.m_x, event.m_y)
232    
233    
   
   
234  class MapPrintout(wx.wxPrintout):  class MapPrintout(wx.wxPrintout):
235    
236      """      """
237      wxPrintout class for printing Thuban maps      wxPrintout class for printing Thuban maps
238      """      """
239    
240      def __init__(self, map):      def __init__(self, canvas, map, region, selected_layer, selected_shapes):
241          wx.wxPrintout.__init__(self)          wx.wxPrintout.__init__(self)
242            self.canvas = canvas
243          self.map = map          self.map = map
244            self.region = region
245            self.selected_layer = selected_layer
246            self.selected_shapes = selected_shapes
247    
248      def GetPageInfo(self):      def GetPageInfo(self):
249          return (1, 1, 1, 1)          return (1, 1, 1, 1)
# Line 215  class MapPrintout(wx.wxPrintout): Line 257  class MapPrintout(wx.wxPrintout):
257    
258      def draw_on_dc(self, dc):      def draw_on_dc(self, dc):
259          width, height = self.GetPageSizePixels()          width, height = self.GetPageSizePixels()
260          llx, lly, urx, ury = self.map.ProjectedBoundingBox()          scale, offset, mapregion = OutputTransform(self.canvas.scale,
261          scalex = width / (urx - llx)                                                     self.canvas.offset,
262          scaley = height / (ury - lly)                                                     self.canvas.GetSizeTuple(),
263          scale = min(scalex, scaley)                                                     self.GetPageSizePixels())
         offx = 0.5 * (width - (urx + llx) * scale)  
         offy = 0.5 * (height + (ury + lly) * scale)  
   
264          resx, resy = self.GetPPIPrinter()          resx, resy = self.GetPPIPrinter()
265          renderer = PrinterRender(dc, scale, (offx, offy), resolution = resx)          renderer = PrinterRenderer(dc, scale, offset, resolution = resy)
266          renderer.RenderMap(self.map)          x, y, width, height = self.region
267          return wx.true          canvas_scale = self.canvas.scale
268                    renderer.RenderMap(self.map,
269                               (0,0,
270                                    (width/canvas_scale)*scale,
271                                    (height/canvas_scale)*scale),
272                                    mapregion,
273                               self.selected_layer, self.selected_shapes)
274            return True
275    
276  class MapCanvas(wxWindow):  
277    class MapCanvas(wxWindow, Publisher):
278    
279      """A widget that displays a map and offers some interaction"""      """A widget that displays a map and offers some interaction"""
280    
281      def __init__(self, parent, winid, interactor):      # Some messages that can be subscribed/unsubscribed directly through
282        # the MapCanvas come in fact from other objects. This is a dict
283        # mapping those messages to the names of the instance variables they
284        # actually come from. The delegation is implemented in the Subscribe
285        # and Unsubscribe methods
286        delegated_messages = {LAYER_SELECTED: "selection",
287                              SHAPES_SELECTED: "selection"}
288    
289        # Methods delegated to some instance variables. The delegation is
290        # implemented in the __getattr__ method.
291        delegated_methods = {"SelectLayer": "selection",
292                             "SelectShapes": "selection",
293                             "SelectedLayer": "selection",
294                             "HasSelectedLayer": "selection",
295                             "HasSelectedShapes": "selection",
296                             "SelectedShapes": "selection"}
297    
298        def __init__(self, parent, winid):
299          wxWindow.__init__(self, parent, winid)          wxWindow.__init__(self, parent, winid)
300          self.SetBackgroundColour(wxColour(255, 255, 255))          self.SetBackgroundColour(wxColour(255, 255, 255))
301    
302            # the map displayed in this canvas. Set with SetMap()
303          self.map = None          self.map = None
304    
305            # scale and offset describe the transformation from projected
306            # coordinates to window coordinates.
307          self.scale = 1.0          self.scale = 1.0
308          self.offset = (0, 0)          self.offset = (0, 0)
309    
310            # whether the user is currently dragging the mouse, i.e. moving
311            # the mouse while pressing a mouse button
312          self.dragging = 0          self.dragging = 0
313    
314            # the currently active tool
315          self.tool = None          self.tool = None
316          self.redraw_on_idle = 0  
317            # The current mouse position of the last OnMotion event or None
318            # if the mouse is outside the window.
319            self.current_position = None
320    
321            # the bitmap serving as backing store
322            self.bitmap = None
323    
324            # the selection
325            self.selection = Selection()
326            self.selection.Subscribe(SHAPES_SELECTED , self.shape_selected)
327    
328            # keep track of which layers/shapes are selected to make sure we
329            # only redraw when necessary
330            self.last_selected_layer = None
331            self.last_selected_shape = None
332    
333            # subscribe the WX events we're interested in
334          EVT_PAINT(self, self.OnPaint)          EVT_PAINT(self, self.OnPaint)
335          EVT_LEFT_DOWN(self, self.OnLeftDown)          EVT_LEFT_DOWN(self, self.OnLeftDown)
336          EVT_LEFT_UP(self, self.OnLeftUp)          EVT_LEFT_UP(self, self.OnLeftUp)
337          EVT_MOTION(self, self.OnMotion)          EVT_MOTION(self, self.OnMotion)
338          wx.EVT_IDLE(self, self.OnIdle)          EVT_LEAVE_WINDOW(self, self.OnLeaveWindow)
339          self.interactor = interactor          wx.EVT_SIZE(self, self.OnSize)
340          self.interactor.Subscribe(SELECTED_SHAPE, self.shape_selected)  
341        def __del__(self):
342            wxWindow.__del__(self)
343            Publisher.__del__(self)
344    
345        def Subscribe(self, channel, *args):
346            """Extend the inherited method to handle delegated messages.
347    
348            If channel is one of the delegated messages call the appropriate
349            object's Subscribe method. Otherwise just call the inherited
350            method.
351            """
352            if channel in self.delegated_messages:
353                object = getattr(self, self.delegated_messages[channel])
354                object.Subscribe(channel, *args)
355            else:
356                Publisher.Subscribe(self, channel, *args)
357    
358        def Unsubscribe(self, channel, *args):
359            """Extend the inherited method to handle delegated messages.
360    
361            If channel is one of the delegated messages call the appropriate
362            object's Unsubscribe method. Otherwise just call the inherited
363            method.
364            """
365            if channel in self.delegated_messages:
366                object = getattr(self, self.delegated_messages[channel])
367                object.Unsubscribe(channel, *args)
368            else:
369                Publisher.Unsubscribe(self, channel, *args)
370    
371        def __getattr__(self, attr):
372            if attr in self.delegated_methods:
373                return getattr(getattr(self, self.delegated_methods[attr]), attr)
374            raise AttributeError(attr)
375    
376      def OnPaint(self, event):      def OnPaint(self, event):
377          dc = wxPaintDC(self)          dc = wxPaintDC(self)
378          if self.map is None or not self.map.HasLayers():          clear = self.map is None or not self.map.HasLayers()
379              return  
380          self.redraw_on_idle = 1          wxBeginBusyCursor()
381            try:
382                if not clear:
383                    self.do_redraw()
384                    try:
385                        pass
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            finally:
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 self.scale == 0:
451                return
452    
453            if hasattr(self, "export_path"):
454                export_path = self.export_path
455            else:
456                export_path="."
457            dlg = wxFileDialog(self, _("Export Map"), export_path, "",
458                               "Enhanced Metafile (*.wmf)|*.wmf",
459                               wxSAVE|wxOVERWRITE_PROMPT)
460            if dlg.ShowModal() == wxID_OK:
461                self.export_path = os.path.dirname(dlg.GetPath())
462                dc = wxMetaFileDC(dlg.GetPath())
463        
464                scale, offset, mapregion = OutputTransform(self.scale,
465                                                           self.offset,
466                                                           self.GetSizeTuple(),
467                                                           dc.GetSizeTuple())
468    
469                selected_layer = self.selection.SelectedLayer()
470                selected_shapes = self.selection.SelectedShapes()
471    
472                renderer = ExportRenderer(dc, scale, offset)
473    
474                # Pass the entire bitmap as update region to the renderer.
475                # We're redrawing the whole bitmap, after all.
476                width, height = self.GetSizeTuple()
477                renderer.RenderMap(self.map,
478                                    (0,0,
479                                        (width/self.scale)*scale,
480                                        (height/self.scale)*scale),
481                                    mapregion,
482                                    selected_layer, selected_shapes)
483                dc.EndDrawing()
484                dc.Close()
485            dlg.Destroy()
486            
487      def Print(self):      def Print(self):
488          printer = wx.wxPrinter()          printer = wx.wxPrinter()
489          printout = MapPrintout(self.map)          width, height = self.GetSizeTuple()
490          printer.Print(self, printout, wx.true)          selected_layer = self.selection.SelectedLayer()
491          printout.Destroy()          selected_shapes = self.selection.SelectedShapes()
492                    
493            printout = MapPrintout(self, self.map, (0, 0, width, height),
494                                   selected_layer, selected_shapes)
495            printer.Print(self, printout, True)
496            printout.Destroy()
497    
498      def SetMap(self, map):      def SetMap(self, map):
499          redraw_channels = (LAYERS_CHANGED, LAYER_LEGEND_CHANGED,          redraw_channels = (MAP_LAYERS_CHANGED, LAYER_CHANGED,
500                             LAYER_VISIBILITY_CHANGED)                             LAYER_VISIBILITY_CHANGED)
501          if self.map is not None:          if self.map is not None:
502              for channel in redraw_channels:              for channel in redraw_channels:
503                  self.map.Unsubscribe(channel, self.redraw)                  self.map.Unsubscribe(channel, self.full_redraw)
504              self.map.Unsubscribe(MAP_PROJECTION_CHANGED,              self.map.Unsubscribe(MAP_PROJECTION_CHANGED,
505                                   self.projection_changed)                                   self.projection_changed)
506          self.map = map          self.map = map
507            self.selection.ClearSelection()
508          if self.map is not None:          if self.map is not None:
509              for channel in redraw_channels:              for channel in redraw_channels:
510                  self.map.Subscribe(channel, self.redraw)                  self.map.Subscribe(channel, self.full_redraw)
511              self.map.Subscribe(MAP_PROJECTION_CHANGED, self.projection_changed)              self.map.Subscribe(MAP_PROJECTION_CHANGED, self.projection_changed)
512          self.FitMapToWindow()          self.FitMapToWindow()
513            # force a redraw. If map is not empty, it's already been called
514            # by FitMapToWindow but if map is empty it hasn't been called
515            # yet so we have to explicitly call it.
516            self.full_redraw()
517    
518      def Map(self):      def Map(self):
519            """Return the map displayed by this canvas"""
520          return self.map          return self.map
521    
522      def redraw(self, *args):      def redraw(self, *args):
523          self.Refresh(0)          self.Refresh(0)
524    
525        def full_redraw(self, *args):
526            self.bitmap = None
527            self.redraw()
528    
529      def projection_changed(self, *args):      def projection_changed(self, *args):
530          self.FitMapToWindow()          self.FitMapToWindow()
531          self.redraw()          self.full_redraw()
532    
533      def set_view_transform(self, scale, offset):      def set_view_transform(self, scale, offset):
534            # width/height of the projected bbox
535            llx, lly, urx, ury = bbox = self.map.ProjectedBoundingBox()
536            pwidth = float(urx - llx)
537            pheight = float(ury - lly)
538    
539            # width/height of the window
540            wwidth, wheight = self.GetSizeTuple()
541    
542            # The window's center in projected coordinates assuming the new
543            # scale/offset
544            pcenterx = (wwidth/2 - offset[0]) / scale
545            pcentery = (offset[1] - wheight/2) / scale
546    
547            # The window coordinates used when drawing the shapes must fit
548            # into 16bit signed integers.
549            max_len = max(pwidth, pheight)
550            if max_len:
551                max_scale = 32000.0 / max_len
552            else:
553                # FIXME: What to do in this case? The bbox is effectively
554                # empty so any scale should work.
555                max_scale = scale
556    
557            # The minimal scale is somewhat arbitrarily set to half that of
558            # the bbox fit into the window
559            scales = []
560            if pwidth:
561                scales.append(wwidth / pwidth)
562            if pheight:
563                scales.append(wheight / pheight)
564            if scales:
565                min_scale = 0.5 * min(scales)
566            else:
567                min_scale = scale
568    
569            if scale > max_scale:
570                scale = max_scale
571            elif scale < min_scale:
572                scale = min_scale
573    
574          self.scale = scale          self.scale = scale
575          self.offset = offset  
576          self.redraw()          # determine new offset to preserve the center
577            self.offset = (wwidth/2 - scale * pcenterx,
578                           wheight/2 + scale * pcentery)
579            self.full_redraw()
580            self.issue(SCALE_CHANGED, scale)
581    
582      def proj_to_win(self, x, y):      def proj_to_win(self, x, y):
583          """\          """\
584          Return the point in  window coords given by projected coordinates x y          Return the point in  window coords given by projected coordinates x y
585          """          """
586            if self.scale == 0:
587                return (0, 0)
588    
589          offx, offy = self.offset          offx, offy = self.offset
590          return (self.scale * x + offx, -self.scale * y + offy)          return (self.scale * x + offx, -self.scale * y + offy)
591    
# Line 331  class MapCanvas(wxWindow): Line 593  class MapCanvas(wxWindow):
593          """\          """\
594          Return the point in projected coordinates given by window coords x y          Return the point in projected coordinates given by window coords x y
595          """          """
596            if self.scale == 0:
597                return (0, 0)
598    
599          offx, offy = self.offset          offx, offy = self.offset
600          return ((x - offx) / self.scale, (offy - y) / self.scale)          return ((x - offx) / self.scale, (offy - y) / self.scale)
601    
602      def FitRectToWindow(self, rect):      def FitRectToWindow(self, rect):
603            """Fit the rectangular region given by rect into the window.
604    
605            Set scale so that rect (in projected coordinates) just fits into
606            the window and center it.
607            """
608          width, height = self.GetSizeTuple()          width, height = self.GetSizeTuple()
609          llx, lly, urx, ury = rect          llx, lly, urx, ury = rect
610            if llx == urx or lly == ury:
611                # zero width or zero height. Do Nothing
612                return
613          scalex = width / (urx - llx)          scalex = width / (urx - llx)
614          scaley = height / (ury - lly)          scaley = height / (ury - lly)
615          scale = min(scalex, scaley)          scale = min(scalex, scaley)
# Line 345  class MapCanvas(wxWindow): Line 618  class MapCanvas(wxWindow):
618          self.set_view_transform(scale, (offx, offy))          self.set_view_transform(scale, (offx, offy))
619    
620      def FitMapToWindow(self):      def FitMapToWindow(self):
621          """\          """Fit the map to the window
622          Set the scale and offset so that the map is centered in the  
623          window          Set the scale so that the map fits exactly into the window and
624            center it in the window.
625          """          """
626          bbox = self.map.ProjectedBoundingBox()          bbox = self.map.ProjectedBoundingBox()
627          if bbox is not None:          if bbox is not None:
628              self.FitRectToWindow(bbox)              self.FitRectToWindow(bbox)
629    
630      def ZoomFactor(self, factor):      def FitLayerToWindow(self, layer):
631          width, height = self.GetSizeTuple()          """Fit the given layer to the window.
632          scale = self.scale * factor  
633          offx, offy = self.offset          Set the scale so that the layer fits exactly into the window and
634          offset = (factor * (offx - width / 2) + width / 2,          center it in the window.
635                    factor * (offy - height / 2) + height / 2)          """
636          self.set_view_transform(scale, offset)          
637            bbox = layer.LatLongBoundingBox()
638            if bbox is not None:
639                proj = self.map.GetProjection()
640                if proj is not None:
641                    bbox = proj.ForwardBBox(bbox)
642    
643                if bbox is not None:
644                    self.FitRectToWindow(bbox)
645    
646        def FitSelectedToWindow(self):
647            layer = self.selection.SelectedLayer()
648            shapes = self.selection.SelectedShapes()
649    
650            bbox = layer.ShapesBoundingBox(shapes)
651            if bbox is not None:
652                proj = self.map.GetProjection()
653                if proj is not None:
654                    bbox = proj.ForwardBBox(bbox)
655    
656                if bbox is not None:
657                    if len(shapes) == 1 and layer.ShapeType() == SHAPETYPE_POINT:
658                        self.ZoomFactor(1, self.proj_to_win(bbox[0], bbox[1]))
659                    else:
660                        self.FitRectToWindow(bbox)
661    
662        def ZoomFactor(self, factor, center = None):
663            """Multiply the zoom by factor and center on center.
664    
665            The optional parameter center is a point in window coordinates
666            that should be centered. If it is omitted, it defaults to the
667            center of the window
668            """
669            if self.scale > 0:
670                width, height = self.GetSizeTuple()
671                scale = self.scale * factor
672                offx, offy = self.offset
673                if center is not None:
674                    cx, cy = center
675                else:
676                    cx = width / 2
677                    cy = height / 2
678                offset = (factor * (offx - cx) + width / 2,
679                        factor * (offy - cy) + height / 2)
680                self.set_view_transform(scale, offset)
681    
682      def ZoomOutToRect(self, rect):      def ZoomOutToRect(self, rect):
683          # rect is given in window coordinates          """Zoom out to fit the currently visible region into rect.
684    
685            The rect parameter is given in window coordinates
686            """
687          # determine the bbox of the displayed region in projected          # determine the bbox of the displayed region in projected
688          # coordinates          # coordinates
689          width, height = self.GetSizeTuple()          width, height = self.GetSizeTuple()
# Line 380  class MapCanvas(wxWindow): Line 700  class MapCanvas(wxWindow):
700          self.set_view_transform(scale, (offx, offy))          self.set_view_transform(scale, (offx, offy))
701    
702      def Translate(self, dx, dy):      def Translate(self, dx, dy):
703            """Move the map by dx, dy pixels"""
704          offx, offy = self.offset          offx, offy = self.offset
705          self.set_view_transform(self.scale, (offx + dx, offy + dy))          self.set_view_transform(self.scale, (offx + dx, offy + dy))
706    
707        def SelectTool(self, tool):
708            """Make tool the active tool.
709    
710            The parameter should be an instance of Tool or None to indicate
711            that no tool is active.
712            """
713            self.tool = tool
714    
715      def ZoomInTool(self):      def ZoomInTool(self):
716          self.tool = ZoomInTool(self)          """Start the zoom in tool"""
717            self.SelectTool(ZoomInTool(self))
718    
719      def ZoomOutTool(self):      def ZoomOutTool(self):
720          self.tool = ZoomOutTool(self)          """Start the zoom out tool"""
721            self.SelectTool(ZoomOutTool(self))
722    
723      def PanTool(self):      def PanTool(self):
724          self.tool = PanTool(self)          """Start the pan tool"""
725            self.SelectTool(PanTool(self))
726            #img = resource.GetImageResource("pan", wxBITMAP_TYPE_XPM)
727            #bmp = resource.GetBitmapResource("pan", wxBITMAP_TYPE_XPM)
728            #print bmp
729            #img = wxImageFromBitmap(bmp)
730            #print img
731            #cur = wxCursor(img)
732            #print cur
733            #self.SetCursor(cur)
734    
735      def IdentifyTool(self):      def IdentifyTool(self):
736          self.tool = IdentifyTool(self)          """Start the identify tool"""
737            self.SelectTool(IdentifyTool(self))
738    
739      def LabelTool(self):      def LabelTool(self):
740          self.tool = LabelTool(self)          """Start the label tool"""
741            self.SelectTool(LabelTool(self))
742    
743      def CurrentTool(self):      def CurrentTool(self):
744            """Return the name of the current tool or None if no tool is active"""
745          return self.tool and self.tool.Name() or None          return self.tool and self.tool.Name() or None
746    
747        def CurrentPosition(self):
748            """Return current position of the mouse in projected coordinates.
749    
750            The result is a 2-tuple of floats with the coordinates. If the
751            mouse is not in the window, the result is None.
752            """
753            if self.current_position is not None:
754                x, y = self.current_position
755                return self.win_to_proj(x, y)
756            else:
757                return None
758    
759        def set_current_position(self, event):
760            """Set the current position from event
761    
762            Should be called by all events that contain mouse positions
763            especially EVT_MOTION. The event paramete may be None to
764            indicate the the pointer left the window.
765            """
766            if event is not None:
767                self.current_position = (event.m_x, event.m_y)
768            else:
769                self.current_position = None
770            self.issue(VIEW_POSITION)
771    
772      def OnLeftDown(self, event):      def OnLeftDown(self, event):
773            self.set_current_position(event)
774          if self.tool is not None:          if self.tool is not None:
775              self.drag_dc = wxClientDC(self)              self.drag_dc = wxClientDC(self)
776              self.drag_dc.SetLogicalFunction(wxINVERT)              self.drag_dc.SetLogicalFunction(wxINVERT)
# Line 410  class MapCanvas(wxWindow): Line 779  class MapCanvas(wxWindow):
779              self.tool.MouseDown(event)              self.tool.MouseDown(event)
780              self.tool.Show(self.drag_dc)              self.tool.Show(self.drag_dc)
781              self.dragging = 1              self.dragging = 1
782            
783      def OnLeftUp(self, event):      def OnLeftUp(self, event):
784          self.ReleaseMouse()          self.set_current_position(event)
785          if self.dragging:          if self.dragging:
786              self.tool.Hide(self.drag_dc)              self.ReleaseMouse()
787              self.tool.MouseUp(event)              try:
788              self.drag_dc = None                  self.tool.Hide(self.drag_dc)
789          self.dragging = 0                  self.tool.MouseUp(event)
790                finally:
791                    self.drag_dc = None
792                    self.dragging = 0
793    
794      def OnMotion(self, event):      def OnMotion(self, event):
795            self.set_current_position(event)
796          if self.dragging:          if self.dragging:
797              self.tool.Hide(self.drag_dc)              self.tool.Hide(self.drag_dc)
798              self.tool.MouseMove(event)              self.tool.MouseMove(event)
799              self.tool.Show(self.drag_dc)              self.tool.Show(self.drag_dc)
800    
801      def OnIdle(self, event):      def OnLeaveWindow(self, event):
802          if self.redraw_on_idle:          self.set_current_position(None)
803              self.do_redraw()  
804          self.redraw_on_idle = 0      def OnSize(self, event):
805            # the window's size has changed. We have to get a new bitmap. If
806            # we want to be clever we could try to get by without throwing
807            # everything away. E.g. when the window gets smaller, we could
808            # either keep the bitmap or create the new one from the old one.
809            # Even when the window becomes larger some parts of the bitmap
810            # could be reused.
811            self.full_redraw()
812            pass
813    
814      def shape_selected(self, layer, shape):      def shape_selected(self, layer, shape):
815          self.redraw()          """Receiver for the SHAPES_SELECTED messages. Redraw the map."""
816            # The selection object takes care that it only issues
817            # SHAPES_SELECTED messages when the set of selected shapes has
818            # actually changed, so we can do a full redraw unconditionally.
819            # FIXME: We should perhaps try to limit the redraw to the are
820            # actually covered by the shapes before and after the selection
821            # change.
822            self.full_redraw()
823    
824      def find_shape_at(self, px, py, select_labels = 0):      def unprojected_rect_around_point(self, x, y, dist):
825          """Return a tuple shape at point px, py in window coords."""          """return a rect dist pixels around (x, y) in unprojected corrdinates
826    
827            The return value is a tuple (minx, miny, maxx, maxy) suitable a
828            parameter to a layer's ShapesInRegion method.
829            """
830            map_proj = self.map.projection
831            if map_proj is not None:
832                inverse = map_proj.Inverse
833            else:
834                inverse = None
835    
836            xs = []
837            ys = []
838            for dx, dy in ((-1, -1), (1, -1), (1, 1), (-1, 1)):
839                px, py = self.win_to_proj(x + dist * dx, y + dist * dy)
840                if inverse:
841                    px, py = inverse(px, py)
842                xs.append(px)
843                ys.append(py)
844            return (min(xs), min(ys), max(xs), max(ys))
845    
846        def find_shape_at(self, px, py, select_labels = 0, searched_layer = None):
847            """Determine the shape at point px, py in window coords
848    
849            Return the shape and the corresponding layer as a tuple (layer,
850            shape).
851    
852            If the optional parameter select_labels is true (default false)
853            search through the labels. If a label is found return it's index
854            as the shape and None as the layer.
855    
856            If the optional parameter searched_layer is given (or not None
857            which it defaults to), only search in that layer.
858            """
859          map_proj = self.map.projection          map_proj = self.map.projection
860          if map_proj is not None:          if map_proj is not None:
861              forward = map_proj.Forward              forward = map_proj.Forward
# Line 442  class MapCanvas(wxWindow): Line 863  class MapCanvas(wxWindow):
863              forward = None              forward = None
864    
865          scale = self.scale          scale = self.scale
866    
867            if scale == 0:
868                return None, None
869    
870          offx, offy = self.offset          offx, offy = self.offset
871    
872          if select_labels:          if select_labels:
873              labels = self.map.LabelLayer().Labels()              labels = self.map.LabelLayer().Labels()
874                
875              if labels:              if labels:
876                  dc = wxClientDC(self)                  dc = wxClientDC(self)
877                  font = wxFont(10, wx.wxSWISS, wx.wxNORMAL, wx.wxNORMAL)                  font = wxFont(10, wx.wxSWISS, wx.wxNORMAL, wx.wxNORMAL)
878                  dc.SetFont(font)                  dc.SetFont(font)
879                  for i in range(len(labels)):                  for i in range(len(labels) - 1, -1, -1):
880                      label = labels[i]                      label = labels[i]
881                      x = label.x                      x = label.x
882                      y = label.y                      y = label.y
# Line 477  class MapCanvas(wxWindow): Line 902  class MapCanvas(wxWindow):
902                          y = y - height/2                          y = y - height/2
903                      if x <= px < x + width and y <= py <= y + height:                      if x <= px < x + width and y <= py <= y + height:
904                          return None, i                          return None, i
905                    
906          layers = self.map.Layers()          if searched_layer:
907                layers = [searched_layer]
908            else:
909                layers = self.map.Layers()
910    
911          for layer_index in range(len(layers) - 1, -1, -1):          for layer_index in range(len(layers) - 1, -1, -1):
912              layer = layers[layer_index]              layer = layers[layer_index]
913    
# Line 486  class MapCanvas(wxWindow): Line 915  class MapCanvas(wxWindow):
915              if not layer.Visible():              if not layer.Visible():
916                  continue                  continue
917    
918              filled = layer.fill is not None              filled = layer.GetClassification().GetDefaultFill() \
919              stroked = layer.stroke is not None                       is not Color.Transparent
920                                stroked = layer.GetClassification().GetDefaultLineColor() \
921                          is not Color.Transparent
922    
923              layer_proj = layer.projection              layer_proj = layer.projection
924              if layer_proj is not None:              if layer_proj is not None:
925                  inverse = layer_proj.Inverse                  inverse = layer_proj.Inverse
926              else:              else:
927                  inverse = None                  inverse = None
928                    
929              shapetype = layer.ShapeType()              shapetype = layer.ShapeType()
930    
931              select_shape = -1              select_shape = -1
932    
933                # Determine the ids of the shapes that overlap a tiny area
934                # around the point. For layers containing points we have to
935                # choose a larger size of the box we're testing agains so
936                # that we take the size of the markers into account
937                # FIXME: Once the markers are more flexible this part has to
938                # become more flexible too, of course
939                if shapetype == SHAPETYPE_POINT:
940                    box = self.unprojected_rect_around_point(px, py, 5)
941                else:
942                    box = self.unprojected_rect_around_point(px, py, 1)
943                shape_ids = layer.ShapesInRegion(box)
944                shape_ids.reverse()
945    
946              if shapetype == SHAPETYPE_POLYGON:              if shapetype == SHAPETYPE_POLYGON:
947                  for i in range(layer.NumShapes()):                  for i in shape_ids:
948                      result = point_in_polygon_shape(layer.shapefile.cobject(),                      shapefile = layer.ShapeStore().Shapefile().cobject()
949                                                      i,                      result = point_in_polygon_shape(shapefile, i,
950                                                      filled, stroked,                                                      filled, stroked,
951                                                      map_proj, layer_proj,                                                      map_proj, layer_proj,
952                                                      scale, -scale, offx, offy,                                                      scale, -scale, offx, offy,
# Line 510  class MapCanvas(wxWindow): Line 955  class MapCanvas(wxWindow):
955                          select_shape = i                          select_shape = i
956                          break                          break
957              elif shapetype == SHAPETYPE_ARC:              elif shapetype == SHAPETYPE_ARC:
958                  for i in range(layer.NumShapes()):                  for i in shape_ids:
959                      result = point_in_polygon_shape(layer.shapefile.cobject(),                      shapefile = layer.ShapeStore().Shapefile().cobject()
960                        result = point_in_polygon_shape(shapefile,
961                                                      i, 0, 1,                                                      i, 0, 1,
962                                                      map_proj, layer_proj,                                                      map_proj, layer_proj,
963                                                      scale, -scale, offx, offy,                                                      scale, -scale, offx, offy,
# Line 520  class MapCanvas(wxWindow): Line 966  class MapCanvas(wxWindow):
966                          select_shape = i                          select_shape = i
967                          break                          break
968              elif shapetype == SHAPETYPE_POINT:              elif shapetype == SHAPETYPE_POINT:
969                  for i in range(layer.NumShapes()):                  for i in shape_ids:
970                      shape = layer.Shape(i)                      shape = layer.Shape(i)
971                      x, y = shape.Points()[0]                      x, y = shape.Points()[0]
972                      if inverse:                      if inverse:
# Line 537  class MapCanvas(wxWindow): Line 983  class MapCanvas(wxWindow):
983                  return layer, select_shape                  return layer, select_shape
984          return None, None          return None, None
985    
986      def SelectShapeAt(self, x, y):      def SelectShapeAt(self, x, y, layer = None):
987          layer, shape = self.find_shape_at(x, y)          """\
988          self.interactor.SelectLayerAndShape(layer, shape)          Select and return the shape and its layer at window position (x, y)
989    
990            If layer is given, only search in that layer. If no layer is
991            given, search through all layers.
992    
993            Return a tuple (layer, shapeid). If no shape is found, return
994            (None, None).
995            """
996            layer, shape = result = self.find_shape_at(x, y, searched_layer=layer)
997            # If layer is None, then shape will also be None. We don't want
998            # to deselect the currently selected layer, so we simply select
999            # the already selected layer again.
1000            if layer is None:
1001                layer = self.selection.SelectedLayer()
1002                shapes = []
1003            else:
1004                shapes = [shape]
1005            self.selection.SelectShapes(layer, shapes)
1006            return result
1007    
1008      def LabelShapeAt(self, x, y):      def LabelShapeAt(self, x, y):
1009            """Add or remove a label at window position x, y.
1010    
1011            If there's a label at the given position, remove it. Otherwise
1012            determine the shape at the position, run the label dialog and
1013            unless the user cancels the dialog, add a laber.
1014            """
1015          ox = x; oy = y          ox = x; oy = y
1016          label_layer = self.map.LabelLayer()          label_layer = self.map.LabelLayer()
1017          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 1019  class MapCanvas(wxWindow):
1019              # a label was selected              # a label was selected
1020              label_layer.RemoveLabel(shape_index)              label_layer.RemoveLabel(shape_index)
1021          elif layer is not None:          elif layer is not None:
1022              text = labeldialog.run_label_dialog(self, layer.table, shape_index)              text = labeldialog.run_label_dialog(self,
1023                                                    layer.ShapeStore().Table(),
1024                                                    shape_index)
1025              if text:              if text:
1026                  proj = self.map.projection                  proj = self.map.projection
1027                  if proj is not None:                  if proj is not None:
# Line 564  class MapCanvas(wxWindow): Line 1036  class MapCanvas(wxWindow):
1036    
1037                  shapetype = layer.ShapeType()                  shapetype = layer.ShapeType()
1038                  if shapetype == SHAPETYPE_POLYGON:                  if shapetype == SHAPETYPE_POLYGON:
1039                      x, y = shape_centroid(layer.shapefile.cobject(),                      shapefile = layer.ShapeStore().Shapefile().cobject()
1040                                            shape_index,                      x, y = shape_centroid(shapefile, shape_index,
1041                                            map_proj, layer_proj, 1, 1, 0, 0)                                            map_proj, layer_proj, 1, 1, 0, 0)
1042                      if map_proj is not None:                      if map_proj is not None:
1043                          x, y = map_proj.Inverse(x, y)                          x, y = map_proj.Inverse(x, y)
# Line 590  class MapCanvas(wxWindow): Line 1062  class MapCanvas(wxWindow):
1062                      valign = ALIGN_CENTER                      valign = ALIGN_CENTER
1063                  label_layer.AddLabel(x, y, text,                  label_layer.AddLabel(x, y, text,
1064                                       halign = halign, valign = valign)                                       halign = halign, valign = valign)
1065    
1066    def OutputTransform(canvas_scale, canvas_offset, canvas_size, device_extend):
1067        """Calculate dimensions to transform canvas content to output device."""
1068        width, height = device_extend
1069    
1070        # Only 80 % of the with are available for the map
1071        width = width * 0.8
1072    
1073        # Define the distance of the map from DC border
1074        distance = 20
1075    
1076        if height < width:
1077            # landscape
1078            map_height = height - 2*distance
1079            map_width = map_height
1080        else:
1081            # portrait, recalibrate width (usually the legend width is too
1082            # small
1083            width = width * 0.9
1084            map_height = width - 2*distance
1085            map_width = map_height
1086        
1087        mapregion = (distance, distance,
1088                     distance+map_width, distance+map_height)
1089    
1090        canvas_width, canvas_height = canvas_size
1091        
1092        scalex = map_width / (canvas_width/canvas_scale)
1093        scaley = map_height / (canvas_height/canvas_scale)
1094        scale = min(scalex, scaley)
1095        canvas_offx, canvas_offy = canvas_offset
1096        offx = scale*canvas_offx/canvas_scale
1097        offy = scale*canvas_offy/canvas_scale
1098    
1099        return scale, (offx, offy), mapregion
Legend:
Removed from v.23  
changed lines
  Added in v.1219
[email protected]
 ViewVC Help
Powered by ViewVC 1.1.26