/[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 822 by jonathan, Mon May 5 18:20:28 2003 UTC revision 1105 by jonathan, Fri May 30 06:30:15 2003 UTC
# Line 1  Line 1 
1  # Copyright (c) 2001, 2002, 2003 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  import sys
18    import os.path
19    
20  from math import hypot  from math import hypot
21    
# Line 19  from wxPython.wx import wxWindow,\ Line 23  from wxPython.wx import wxWindow,\
23       wxPaintDC, wxColour, wxClientDC, wxINVERT, wxTRANSPARENT_BRUSH, wxFont,\       wxPaintDC, wxColour, wxClientDC, wxINVERT, wxTRANSPARENT_BRUSH, wxFont,\
24       EVT_PAINT, EVT_LEFT_DOWN, EVT_LEFT_UP, EVT_MOTION, EVT_LEAVE_WINDOW, \       EVT_PAINT, EVT_LEFT_DOWN, EVT_LEFT_UP, EVT_MOTION, EVT_LEAVE_WINDOW, \
25       wxBITMAP_TYPE_XPM, wxBeginBusyCursor, wxEndBusyCursor, wxCursor, \       wxBITMAP_TYPE_XPM, wxBeginBusyCursor, wxEndBusyCursor, wxCursor, \
26       wxImageFromBitmap       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       MAP_LAYERS_CHANGED, LAYER_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, \
# Line 39  from Thuban.Model.color import Color Line 46  from Thuban.Model.color import Color
46  import resource  import resource
47    
48  from selection import Selection  from selection import Selection
49  from renderer import ScreenRenderer, PrinterRender  from renderer import ScreenRenderer, ExportRenderer, PrinterRenderer
50    
51  import labeldialog  import labeldialog
52    
53  from messages import LAYER_SELECTED, SHAPES_SELECTED, VIEW_POSITION  from messages import LAYER_SELECTED, SHAPES_SELECTED, VIEW_POSITION, \
54                         SCALE_CHANGED
55    
56    
57  #  #
# Line 223  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 247  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    
277  class MapCanvas(wxWindow, Publisher):  class MapCanvas(wxWindow, Publisher):
# Line 277  class MapCanvas(wxWindow, Publisher): Line 291  class MapCanvas(wxWindow, Publisher):
291      delegated_methods = {"SelectLayer": "selection",      delegated_methods = {"SelectLayer": "selection",
292                           "SelectShapes": "selection",                           "SelectShapes": "selection",
293                           "SelectedLayer": "selection",                           "SelectedLayer": "selection",
294                           "HasSelectedLayer": "selection"}                           "HasSelectedLayer": "selection",
295                             "HasSelectedShapes": "selection",
296                             "SelectedShapes": "selection"}
297    
298      def __init__(self, parent, winid):      def __init__(self, parent, winid):
299          wxWindow.__init__(self, parent, winid)          wxWindow.__init__(self, parent, winid)
# Line 361  class MapCanvas(wxWindow, Publisher): Line 377  class MapCanvas(wxWindow, Publisher):
377          dc = wxPaintDC(self)          dc = wxPaintDC(self)
378          clear = self.map is None or not self.map.HasLayers()          clear = self.map is None or not self.map.HasLayers()
379    
380          #wxBeginBusyCursor()          wxBeginBusyCursor()
381            try:
382          if not clear:              if not clear:
             try:  
383                  self.do_redraw()                  self.do_redraw()
384              except:                  try:
385                  print "Error during drawing:", sys.exc_info()[0]                      pass
386                  clear = True                  except:
387                        print "Error during drawing:", sys.exc_info()[0]
388          if clear:                      clear = True
389              # If we've got no map or if the map is empty, simply clear  
390              # the screen.              if clear:
391                    # If we've got no map or if the map is empty, simply clear
392              # XXX it's probably possible to get rid of this. The                  # the screen.
393              # background color of the window is already white and the  
394              # only thing we may have to do is to call self.Refresh()                  # XXX it's probably possible to get rid of this. The
395              # with a true argument in the right places.                  # background color of the window is already white and the
396              dc.BeginDrawing()                  # only thing we may have to do is to call self.Refresh()
397              dc.Clear()                  # with a true argument in the right places.
398              dc.EndDrawing()                  dc.BeginDrawing()
399                    dc.Clear()
400          #wxEndBusyCursor()                  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.          # This should only be called if we have a non-empty map.
# Line 429  class MapCanvas(wxWindow, Publisher): Line 446  class MapCanvas(wxWindow, Publisher):
446          clientdc.Blit(0, 0, width, height, dc, 0, 0)          clientdc.Blit(0, 0, width, height, dc, 0, 0)
447          clientdc.EndDrawing()          clientdc.EndDrawing()
448    
449        def Export(self):
450            if 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            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()          printout.Destroy()
497    
498      def SetMap(self, map):      def SetMap(self, map):
# Line 472  class MapCanvas(wxWindow, Publisher): Line 532  class MapCanvas(wxWindow, Publisher):
532    
533      def set_view_transform(self, scale, offset):      def set_view_transform(self, scale, offset):
534          self.scale = scale          self.scale = scale
535    
536          self.offset = offset          self.offset = offset
537          self.full_redraw()          self.full_redraw()
538            self.issue(SCALE_CHANGED, scale)
539    
540      def proj_to_win(self, x, y):      def proj_to_win(self, x, y):
541          """\          """\
542          Return the point in  window coords given by projected coordinates x y          Return the point in  window coords given by projected coordinates x y
543          """          """
544            if self.scale == 0:
545                return (0, 0)
546    
547          offx, offy = self.offset          offx, offy = self.offset
548          return (self.scale * x + offx, -self.scale * y + offy)          return (self.scale * x + offx, -self.scale * y + offy)
549    
# Line 486  class MapCanvas(wxWindow, Publisher): Line 551  class MapCanvas(wxWindow, Publisher):
551          """\          """\
552          Return the point in projected coordinates given by window coords x y          Return the point in projected coordinates given by window coords x y
553          """          """
554            if self.scale == 0:
555                return (0, 0)
556    
557          offx, offy = self.offset          offx, offy = self.offset
558          return ((x - offx) / self.scale, (offy - y) / self.scale)          return ((x - offx) / self.scale, (offy - y) / self.scale)
559    
# Line 529  class MapCanvas(wxWindow, Publisher): Line 597  class MapCanvas(wxWindow, Publisher):
597              proj = self.map.GetProjection()              proj = self.map.GetProjection()
598              if proj is not None:              if proj is not None:
599                  bbox = proj.ForwardBBox(bbox)                  bbox = proj.ForwardBBox(bbox)
600                  if bbox is not None:  
601                if bbox is not None:
602                    self.FitRectToWindow(bbox)
603    
604        def FitSelectedToWindow(self):
605            layer = self.selection.SelectedLayer()
606            shapes = self.selection.SelectedShapes()
607    
608            bbox = layer.ShapesBoundingBox(shapes)
609            if bbox is not None:
610                proj = self.map.GetProjection()
611                if proj is not None:
612                    bbox = proj.ForwardBBox(bbox)
613    
614                if bbox is not None:
615                    if len(shapes) == 1 and layer.ShapeType() == SHAPETYPE_POINT:
616                        self.ZoomFactor(1, self.proj_to_win(bbox[0], bbox[1]))
617                    else:
618                      self.FitRectToWindow(bbox)                      self.FitRectToWindow(bbox)
619    
620      def ZoomFactor(self, factor, center = None):      def ZoomFactor(self, factor, center = None):
# Line 539  class MapCanvas(wxWindow, Publisher): Line 624  class MapCanvas(wxWindow, Publisher):
624          that should be centered. If it is omitted, it defaults to the          that should be centered. If it is omitted, it defaults to the
625          center of the window          center of the window
626          """          """
627          width, height = self.GetSizeTuple()          if self.scale > 0:
628          scale = self.scale * factor              width, height = self.GetSizeTuple()
629          offx, offy = self.offset              scale = self.scale * factor
630          if center is not None:              offx, offy = self.offset
631              cx, cy = center              if center is not None:
632          else:                  cx, cy = center
633              cx = width / 2              else:
634              cy = height / 2                  cx = width / 2
635          offset = (factor * (offx - cx) + width / 2,                  cy = height / 2
636                    factor * (offy - cy) + height / 2)              offset = (factor * (offx - cx) + width / 2,
637          self.set_view_transform(scale, offset)                      factor * (offy - cy) + height / 2)
638                self.set_view_transform(scale, offset)
639    
640      def ZoomOutToRect(self, rect):      def ZoomOutToRect(self, rect):
641          """Zoom out to fit the currently visible region into rect.          """Zoom out to fit the currently visible region into rect.
# Line 735  class MapCanvas(wxWindow, Publisher): Line 821  class MapCanvas(wxWindow, Publisher):
821              forward = None              forward = None
822    
823          scale = self.scale          scale = self.scale
824    
825            if scale == 0:
826                return None, None
827    
828          offx, offy = self.offset          offx, offy = self.offset
829    
830          if select_labels:          if select_labels:
# Line 927  class MapCanvas(wxWindow, Publisher): Line 1017  class MapCanvas(wxWindow, Publisher):
1017                      valign = ALIGN_CENTER                      valign = ALIGN_CENTER
1018                  label_layer.AddLabel(x, y, text,                  label_layer.AddLabel(x, y, text,
1019                                       halign = halign, valign = valign)                                       halign = halign, valign = valign)
1020    
1021    def OutputTransform(canvas_scale, canvas_offset, canvas_size, device_extend):
1022        """Calculate dimensions to transform canvas content to output device."""
1023        width, height = device_extend
1024    
1025        # Only 80 % of the with are available for the map
1026        width = width * 0.8
1027    
1028        # Define the distance of the map from DC border
1029        distance = 20
1030    
1031        if height < width:
1032            # landscape
1033            map_height = height - 2*distance
1034            map_width = map_height
1035        else:
1036            # portrait, recalibrate width (usually the legend width is too
1037            # small
1038            width = width * 0.9
1039            map_height = width - 2*distance
1040            map_width = map_height
1041        
1042        mapregion = (distance, distance,
1043                     distance+map_width, distance+map_height)
1044    
1045        canvas_width, canvas_height = canvas_size
1046        
1047        scalex = map_width / (canvas_width/canvas_scale)
1048        scaley = map_height / (canvas_height/canvas_scale)
1049        scale = min(scalex, scaley)
1050        canvas_offx, canvas_offy = canvas_offset
1051        offx = scale*canvas_offx/canvas_scale
1052        offy = scale*canvas_offy/canvas_scale
1053    
1054        return scale, (offx, offy), mapregion
Legend:
Removed from v.822  
changed lines
  Added in v.1105
[email protected]
 ViewVC Help
Powered by ViewVC 1.1.26