/[thuban]/branches/WIP-pyshapelib-bramz/Thuban/UI/view.py

Diff of /branches/WIP-pyshapelib-bramz/Thuban/UI/view.py

Parent Directory | Revision Log | View Patch Patch

-revision 45 by bh,
Fri Sep  7 15:00:21 2001 UTC
+revision 433 by jonathan,
Mon Feb 24 18:47:49 2003 UTC
 Line 1
- # Copyright (c) 2001 by Intevation GmbH
+ # Copyright (c) 2001, 2002, 2003 by Intevation GmbH
  # Authors:
  # Bernhard Herzog <[email protected]>
  #
 Line 15 
 from math import hypot
  from wxPython.wx import wxWindow,\
       wxPaintDC, wxColour, wxClientDC, wxINVERT, wxTRANSPARENT_BRUSH, wxFont,\
-      EVT_PAINT, EVT_LEFT_DOWN, EVT_LEFT_UP, EVT_MOTION
+      EVT_PAINT, EVT_LEFT_DOWN, EVT_LEFT_UP, EVT_MOTION, EVT_LEAVE_WINDOW
  from wxPython import wx
 Line 29 
 from Thuban.Model.layer import SHAPETYPE
       SHAPETYPE_POINT
  from Thuban.Model.label import ALIGN_CENTER, ALIGN_TOP, ALIGN_BOTTOM, \
       ALIGN_LEFT, ALIGN_RIGHT
+ from Thuban.Lib.connector import Publisher
+ from Thuban.Model.color import Color
  from renderer import ScreenRenderer, PrinterRender
  import labeldialog
- from messages import SELECTED_SHAPE
+ from messages import SELECTED_SHAPE, VIEW_POSITION
  #
-Line 127 
 class ZoomInTool(RectTool):
+Line 128 
 class ZoomInTool(RectTool):
      def MouseUp(self, event):
          if self.dragging:
              Tool.MouseUp(self, event)
-             self.view.FitRectToWindow(self.proj_rect())
+             sx, sy = self.start
+             cx, cy = self.current
+             if sx == cx or sy == cy:
+                 # Just a mouse click or a degenerate rectangle. Simply
+                 # zoom in by a factor of two
+                 # FIXME: For a click this is the desired behavior but should we
+                 # really do this for degenrate rectagles as well or
+                 # should we ignore them?
+                 self.view.ZoomFactor(2, center = (cx, cy))
+             else:
+                 # A drag. Zoom in to the rectangle
+                 self.view.FitRectToWindow(self.proj_rect())
  class ZoomOutTool(RectTool):
      """The Zoom-Out Tool"""
      def Name(self):
          return "ZoomOutTool"
-Line 142 
 class ZoomOutTool(RectTool):
+Line 154 
 class ZoomOutTool(RectTool):
              Tool.MouseUp(self, event)
              sx, sy = self.start
              cx, cy = self.current
-             self.view.ZoomOutToRect((min(sx, cx), min(sy, cy),
+             if sx == cx or sy == cy:
-                                      max(sx, cx), max(sy, cy)))
+                 # Just a mouse click or a degenerate rectangle. Simply
+                 # zoom out by a factor of two.
+                 # FIXME: For a click this is the desired behavior but should we
+                 # really do this for degenrate rectagles as well or
+                 # should we ignore them?
+                 self.view.ZoomFactor(0.5, center = (cx, cy))
+             else:
+                 # A drag. Zoom out to the rectangle
+                 self.view.ZoomOutToRect((min(sx, cx), min(sy, cy),
+                                          max(sx, cx), max(sy, cy)))
  class PanTool(Tool):
-Line 155 
 class PanTool(Tool):
+Line 176 
 class PanTool(Tool):
      def MouseMove(self, event):
          if self.dragging:
-             x0, y0 = self.current
              Tool.MouseMove(self, event)
+             sx, sy = self.start
              x, y = self.current
              width, height = self.view.GetSizeTuple()
+             bitmapdc = wx.wxMemoryDC()
+             bitmapdc.SelectObject(self.view.bitmap)
              dc = self.view.drag_dc
-             dc.Blit(0, 0, width, height, dc, x0 - x, y0 - y)
+             dc.Blit(0, 0, width, height, bitmapdc, sx - x, sy - y)
      def MouseUp(self, event):
          if self.dragging:
-Line 168 
 class PanTool(Tool):
+Line 193 
 class PanTool(Tool):
              sx, sy = self.start
              cx, cy = self.current
              self.view.Translate(cx - sx, cy - sy)
  class IdentifyTool(Tool):
      """The "Identify" Tool"""
      def Name(self):
          return "IdentifyTool"
-Line 226 
 class MapPrintout(wx.wxPrintout):
+Line 251 
 class MapPrintout(wx.wxPrintout):
          renderer = PrinterRender(dc, scale, (offx, offy), resolution = resx)
          renderer.RenderMap(self.map)
          return wx.true
- class MapCanvas(wxWindow):
+ class MapCanvas(wxWindow, Publisher):
      """A widget that displays a map and offers some interaction"""
      def __init__(self, parent, winid, interactor):
          wxWindow.__init__(self, parent, winid)
          self.SetBackgroundColour(wxColour(255, 255, 255))
+         # the map displayed in this canvas. Set with SetMap()
          self.map = None
+         # scale and offset describe the transformation from projected
+         # coordinates to window coordinates.
          self.scale = 1.0
          self.offset = (0, 0)
+         # whether the user is currently dragging the mouse, i.e. moving
+         # the mouse while pressing a mouse button
          self.dragging = 0
+         # the currently active tool
          self.tool = None
-         self.redraw_on_idle = 0
+         # The current mouse position of the last OnMotion event or None
+         # if the mouse is outside the window.
+         self.current_position = None
+         # the bitmap serving as backing store
+         self.bitmap = None
+         # the interactor
+         self.interactor = interactor
+         self.interactor.Subscribe(SELECTED_SHAPE, self.shape_selected)
+         # keep track of which layers/shapes are selected to make sure we
+         # only redraw when necessary
+         self.last_selected_layer = None
+         self.last_selected_shape = None
+         # subscribe the WX events we're interested in
          EVT_PAINT(self, self.OnPaint)
          EVT_LEFT_DOWN(self, self.OnLeftDown)
          EVT_LEFT_UP(self, self.OnLeftUp)
          EVT_MOTION(self, self.OnMotion)
-         wx.EVT_IDLE(self, self.OnIdle)
+         EVT_LEAVE_WINDOW(self, self.OnLeaveWindow)
-         self.interactor = interactor
+         wx.EVT_SIZE(self, self.OnSize)
-         self.interactor.Subscribe(SELECTED_SHAPE, self.shape_selected)
+     def __del__(self):
+         wxWindow.__del__(self)
+         Publisher.__del__(self)
      def OnPaint(self, event):
          dc = wxPaintDC(self)
-         if self.map is None or not self.map.HasLayers():
+         if self.map is not None and self.map.HasLayers():
-             return
+             self.do_redraw()
-         self.redraw_on_idle = 1
+         else:
+             # If we've got no map or if the map is empty, simply clear
+             # the screen.
+             # XXX it's probably possible to get rid of this. The
+             # background color of the window is already white and the
+             # only thing we may have to do is to call self.Refresh()
+             # with a true argument in the right places.
+             dc.BeginDrawing()
+             dc.Clear()
+             dc.EndDrawing()
      def do_redraw(self):
+         # This should only be called if we have a non-empty map.
+         # Get the window size.
          width, height = self.GetSizeTuple()
-         bitmap = wx.wxEmptyBitmap(width, height)
-         dc = wx.wxMemoryDC()
+         # If self.bitmap's still there, reuse it. Otherwise redraw it
-         dc.SelectObject(bitmap)
+         if self.bitmap is not None:
+             bitmap = self.bitmap
+         else:
+             bitmap = wx.wxEmptyBitmap(width, height)
+             dc = wx.wxMemoryDC()
+             dc.SelectObject(bitmap)
+             dc.BeginDrawing()
+             # clear the background
+             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
-         dc.BeginDrawing()
+             # draw the map into the bitmap
+             renderer = ScreenRenderer(dc, self.scale, self.offset)
-         dc.SetBrush(wx.wxWHITE_BRUSH)
+             # Pass the entire bitmap as update region to the renderer.
-         dc.SetPen(wx.wxTRANSPARENT_PEN)
+             # We're redrawing the whole bitmap, after all.
-         dc.DrawRectangle(0, 0, width, height)
+             renderer.RenderMap(self.map, (0, 0, width, height),
+                                selected_layer, selected_shape)
-         if 1: #self.interactor.selected_map is self.map:
-             selected_layer = self.interactor.selected_layer
+             dc.EndDrawing()
-             selected_shape = self.interactor.selected_shape
+             dc.SelectObject(wx.wxNullBitmap)
-         else:
+             self.bitmap = bitmap
-             selected_layer = None
-             selected_shape = None
-         renderer = ScreenRenderer(dc, self.scale, self.offset)
-         renderer.RenderMap(self.map, selected_layer, selected_shape)
+         # blit the bitmap to the screen
+         dc = wx.wxMemoryDC()
+         dc.SelectObject(bitmap)
          clientdc = wxClientDC(self)
          clientdc.BeginDrawing()
          clientdc.Blit(0, 0, width, height, dc, 0, 0)
          clientdc.EndDrawing()
      def Print(self):
          printer = wx.wxPrinter()
          printout = MapPrintout(self.map)
          printer.Print(self, printout, wx.true)
          printout.Destroy()
      def SetMap(self, map):
          redraw_channels = (LAYERS_CHANGED, LAYER_LEGEND_CHANGED,
                             LAYER_VISIBILITY_CHANGED)
          if self.map is not None:
              for channel in redraw_channels:
-                 self.map.Unsubscribe(channel, self.redraw)
+                 self.map.Unsubscribe(channel, self.full_redraw)
              self.map.Unsubscribe(MAP_PROJECTION_CHANGED,
                                   self.projection_changed)
          self.map = map
          if self.map is not None:
              for channel in redraw_channels:
-                 self.map.Subscribe(channel, self.redraw)
+                 self.map.Subscribe(channel, self.full_redraw)
              self.map.Subscribe(MAP_PROJECTION_CHANGED, self.projection_changed)
          self.FitMapToWindow()
+         # force a redraw. If map is not empty, it's already been called
+         # by FitMapToWindow but if map is empty it hasn't been called
+         # yet so we have to explicitly call it.
+         self.full_redraw()
      def Map(self):
+         """Return the map displayed by this canvas"""
          return self.map
      def redraw(self, *args):
          self.Refresh(0)
+     def full_redraw(self, *args):
+         self.bitmap = None
+         self.redraw()
      def projection_changed(self, *args):
          self.FitMapToWindow()
-         self.redraw()
+         self.full_redraw()
      def set_view_transform(self, scale, offset):
          self.scale = scale
          self.offset = offset
-         self.redraw()
+         self.full_redraw()
      def proj_to_win(self, x, y):
          """\
-Line 335 
 class MapCanvas(wxWindow):
+Line 427 
 class MapCanvas(wxWindow):
          return ((x - offx) / self.scale, (offy - y) / self.scale)
      def FitRectToWindow(self, rect):
+         """Fit the rectangular region given by rect into the window.
+         Set scale so that rect (in projected coordinates) just fits into
+         the window and center it.
+         """
          width, height = self.GetSizeTuple()
          llx, lly, urx, ury = rect
          if llx == urx or lly == ury:
-Line 348 
 class MapCanvas(wxWindow):
+Line 445 
 class MapCanvas(wxWindow):
          self.set_view_transform(scale, (offx, offy))
      def FitMapToWindow(self):
-         """\
+         """Fit the map to the window
-         Set the scale and offset so that the map is centered in the
-         window
+         Set the scale so that the map fits exactly into the window and
+         center it in the window.
          """
          bbox = self.map.ProjectedBoundingBox()
          if bbox is not None:
              self.FitRectToWindow(bbox)
-     def ZoomFactor(self, factor):
+     def ZoomFactor(self, factor, center = None):
+         """Multiply the zoom by factor and center on center.
+         The optional parameter center is a point in window coordinates
+         that should be centered. If it is omitted, it defaults to the
+         center of the window
+         """
          width, height = self.GetSizeTuple()
          scale = self.scale * factor
          offx, offy = self.offset
-         offset = (factor * (offx - width / 2) + width / 2,
+         if center is not None:
-                   factor * (offy - height / 2) + height / 2)
+             cx, cy = center
+         else:
+             cx = width / 2
+             cy = height / 2
+         offset = (factor * (offx - cx) + width / 2,
+                   factor * (offy - cy) + height / 2)
          self.set_view_transform(scale, offset)
      def ZoomOutToRect(self, rect):
-         # rect is given in window coordinates
+         """Zoom out to fit the currently visible region into rect.
+         The rect parameter is given in window coordinates
+         """
          # determine the bbox of the displayed region in projected
          # coordinates
          width, height = self.GetSizeTuple()
-Line 383 
 class MapCanvas(wxWindow):
+Line 494 
 class MapCanvas(wxWindow):
          self.set_view_transform(scale, (offx, offy))
      def Translate(self, dx, dy):
+         """Move the map by dx, dy pixels"""
          offx, offy = self.offset
          self.set_view_transform(self.scale, (offx + dx, offy + dy))
+     def SelectTool(self, tool):
+         """Make tool the active tool.
+         The parameter should be an instance of Tool or None to indicate
+         that no tool is active.
+         """
+         self.tool = tool
      def ZoomInTool(self):
-         self.tool = ZoomInTool(self)
+         """Start the zoom in tool"""
+         self.SelectTool(ZoomInTool(self))
      def ZoomOutTool(self):
-         self.tool = ZoomOutTool(self)
+         """Start the zoom out tool"""
+         self.SelectTool(ZoomOutTool(self))
      def PanTool(self):
-         self.tool = PanTool(self)
+         """Start the pan tool"""
+         self.SelectTool(PanTool(self))
      def IdentifyTool(self):
-         self.tool = IdentifyTool(self)
+         """Start the identify tool"""
+         self.SelectTool(IdentifyTool(self))
      def LabelTool(self):
-         self.tool = LabelTool(self)
+         """Start the label tool"""
+         self.SelectTool(LabelTool(self))
      def CurrentTool(self):
+         """Return the name of the current tool or None if no tool is active"""
          return self.tool and self.tool.Name() or None
+     def CurrentPosition(self):
+         """Return current position of the mouse in projected coordinates.
+         The result is a 2-tuple of floats with the coordinates. If the
+         mouse is not in the window, the result is None.
+         """
+         if self.current_position is not None:
+             x, y = self.current_position
+             return self.win_to_proj(x, y)
+         else:
+             return None
+     def set_current_position(self, event):
+         """Set the current position from event
+         Should be called by all events that contain mouse positions
+         especially EVT_MOTION. The event paramete may be None to
+         indicate the the pointer left the window.
+         """
+         if event is not None:
+             self.current_position = (event.m_x, event.m_y)
+         else:
+             self.current_position = None
+         self.issue(VIEW_POSITION)
      def OnLeftDown(self, event):
+         self.set_current_position(event)
          if self.tool is not None:
              self.drag_dc = wxClientDC(self)
              self.drag_dc.SetLogicalFunction(wxINVERT)
-Line 413 
 class MapCanvas(wxWindow):
+Line 565 
 class MapCanvas(wxWindow):
              self.tool.MouseDown(event)
              self.tool.Show(self.drag_dc)
              self.dragging = 1
      def OnLeftUp(self, event):
-         self.ReleaseMouse()
+         self.set_current_position(event)
          if self.dragging:
-             self.tool.Hide(self.drag_dc)
+             self.ReleaseMouse()
-             self.tool.MouseUp(event)
+             try:
-             self.drag_dc = None
+                 self.tool.Hide(self.drag_dc)
-         self.dragging = 0
+                 self.tool.MouseUp(event)
+             finally:
+                 self.drag_dc = None
+                 self.dragging = 0
      def OnMotion(self, event):
+         self.set_current_position(event)
          if self.dragging:
              self.tool.Hide(self.drag_dc)
              self.tool.MouseMove(event)
              self.tool.Show(self.drag_dc)
-     def OnIdle(self, event):
+     def OnLeaveWindow(self, event):
-         if self.redraw_on_idle:
+         self.set_current_position(None)
-             self.do_redraw()
-         self.redraw_on_idle = 0
+     def OnSize(self, event):
+         # the window's size has changed. We have to get a new bitmap. If
+         # we want to be clever we could try to get by without throwing
+         # everything away. E.g. when the window gets smaller, we could
+         # either keep the bitmap or create the new one from the old one.
+         # Even when the window becomes larger some parts of the bitmap
+         # could be reused.
+         self.full_redraw()
      def shape_selected(self, layer, shape):
-         self.redraw()
+         """Redraw the map.
+         Receiver for the SELECTED_SHAPE messages. Try to redraw only
+         when necessary.
+         """
+         # A redraw is necessary when the display has to change, which
+         # means that either the status changes from having no selection
+         # to having a selection shape or vice versa, or when the fact
+         # whether there is a selection at all doesn't change, when the
+         # shape which is selected has changed (which means that layer or
+         # shapeid changes).
+         if ((shape is not None or self.last_selected_shape is not None)
+             and (shape != self.last_selected_shape
+                  or layer != self.last_selected_layer)):
+             self.full_redraw()
+         # remember the selection so we can compare when it changes again.
+         self.last_selected_layer = layer
+         self.last_selected_shape = shape
+     def unprojected_rect_around_point(self, x, y, dist):
+         """return a rect dist pixels around (x, y) in unprojected corrdinates
+         The return value is a tuple (minx, miny, maxx, maxy) suitable a
+         parameter to a layer's ShapesInRegion method.
+         """
+         map_proj = self.map.projection
+         if map_proj is not None:
+             inverse = map_proj.Inverse
+         else:
+             inverse = None
-     def find_shape_at(self, px, py, select_labels = 0, selected_layer = 1):
+         xs = []
+         ys = []
+         for dx, dy in ((-1, -1), (1, -1), (1, 1), (-1, 1)):
+             px, py = self.win_to_proj(x + dist * dx, y + dist * dy)
+             if inverse:
+                 px, py = inverse(px, py)
+             xs.append(px)
+             ys.append(py)
+         return (min(xs), min(ys), max(xs), max(ys))
+     def find_shape_at(self, px, py, select_labels = 0, searched_layer = None):
          """Determine the shape at point px, py in window coords
          Return the shape and the corresponding layer as a tuple (layer,
-Line 446 
 class MapCanvas(wxWindow):
+Line 649 
 class MapCanvas(wxWindow):
          search through the labels. If a label is found return it's index
          as the shape and None as the layer.
-         If the optional parameter selected_layer is true (default), only
+         If the optional parameter searched_layer is given (or not None
-         search in the currently selected layer.
+         which it defaults to), only search in that layer.
          """
          map_proj = self.map.projection
          if map_proj is not None:
-Line 460 
 class MapCanvas(wxWindow):
+Line 663 
 class MapCanvas(wxWindow):
          if select_labels:
              labels = self.map.LabelLayer().Labels()
              if labels:
                  dc = wxClientDC(self)
                  font = wxFont(10, wx.wxSWISS, wx.wxNORMAL, wx.wxNORMAL)
                  dc.SetFont(font)
-                 for i in range(len(labels)):
+                 for i in range(len(labels) - 1, -1, -1):
                      label = labels[i]
                      x = label.x
                      y = label.y
-Line 492 
 class MapCanvas(wxWindow):
+Line 695 
 class MapCanvas(wxWindow):
                      if x <= px < x + width and y <= py <= y + height:
                          return None, i
-         if selected_layer:
+         if searched_layer:
-             layer = self.interactor.SelectedLayer()
+             layers = [searched_layer]
-             if layer is not None:
-                 layers = [layer]
-             else:
-                 # no layer selected. Use an empty list to effectively
-                 # ignore all layers.
-                 layers = []
          else:
              layers = self.map.Layers()
-Line 510 
 class MapCanvas(wxWindow):
+Line 707 
 class MapCanvas(wxWindow):
              if not layer.Visible():
                  continue
-             filled = layer.fill is not None
+             filled = layer.GetClassification().GetDefaultFill() \
-             stroked = layer.stroke is not None
+                      is not Color.None
+             stroked = layer.GetClassification().GetDefaultStroke() \
+                       is not Color.None
              layer_proj = layer.projection
              if layer_proj is not None:
                  inverse = layer_proj.Inverse
              else:
                  inverse = None
              shapetype = layer.ShapeType()
              select_shape = -1
+             # Determine the ids of the shapes that overlap a tiny area
+             # around the point. For layers containing points we have to
+             # choose a larger size of the box we're testing agains so
+             # that we take the size of the markers into account
+             # FIXME: Once the markers are more flexible this part has to
+             # become more flexible too, of course
+             if shapetype == SHAPETYPE_POINT:
+                 box = self.unprojected_rect_around_point(px, py, 5)
+             else:
+                 box = self.unprojected_rect_around_point(px, py, 1)
+             shape_ids = layer.ShapesInRegion(box)
+             shape_ids.reverse()
              if shapetype == SHAPETYPE_POLYGON:
-                 for i in range(layer.NumShapes()):
+                 for i in shape_ids:
                      result = point_in_polygon_shape(layer.shapefile.cobject(),
                                                      i,
                                                      filled, stroked,
-Line 534 
 class MapCanvas(wxWindow):
+Line 747 
 class MapCanvas(wxWindow):
                          select_shape = i
                          break
              elif shapetype == SHAPETYPE_ARC:
-                 for i in range(layer.NumShapes()):
+                 for i in shape_ids:
                      result = point_in_polygon_shape(layer.shapefile.cobject(),
                                                      i, 0, 1,
                                                      map_proj, layer_proj,
-Line 544 
 class MapCanvas(wxWindow):
+Line 757 
 class MapCanvas(wxWindow):
                          select_shape = i
                          break
              elif shapetype == SHAPETYPE_POINT:
-                 for i in range(layer.NumShapes()):
+                 for i in shape_ids:
                      shape = layer.Shape(i)
                      x, y = shape.Points()[0]
                      if inverse:
-Line 561 
 class MapCanvas(wxWindow):
+Line 774 
 class MapCanvas(wxWindow):
                  return layer, select_shape
          return None, None
-     def SelectShapeAt(self, x, y):
+     def SelectShapeAt(self, x, y, layer = None):
-         layer, shape = self.find_shape_at(x, y)
+         """\
+         Select and return the shape and its layer at window position (x, y)
+         If layer is given, only search in that layer. If no layer is
+         given, search through all layers.
+         Return a tuple (layer, shapeid). If no shape is found, return
+         (None, None).
+         """
+         layer, shape = result = self.find_shape_at(x, y, searched_layer=layer)
          # If layer is None, then shape will also be None. We don't want
          # to deselect the currently selected layer, so we simply select
          # the already selected layer again.
          if layer is None:
              layer = self.interactor.SelectedLayer()
          self.interactor.SelectLayerAndShape(layer, shape)
+         return result
      def LabelShapeAt(self, x, y):
+         """Add or remove a label at window position x, y.
+         If there's a label at the given position, remove it. Otherwise
+         determine the shape at the position, run the label dialog and
+         unless the user cancels the dialog, add a laber.
+         """
          ox = x; oy = y
          label_layer = self.map.LabelLayer()
          layer, shape_index = self.find_shape_at(x, y, select_labels = 1)

 Legend:



Removed from v.45
 


changed lines


 
Added in v.433
 Legend:



Removed from v.45
 


changed lines


 
Added in v.433
-Removed from v.45
+Added in v.433

[email protected]	ViewVC Help
Powered by ViewVC 1.1.26