/[thuban]/branches/WIP-pyshapelib-bramz/Thuban/UI/renderer.py
ViewVC logotype

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

Parent Directory Parent Directory | Revision Log Revision Log | View Patch Patch

revision 1552 by bh, Wed Aug 6 17:21:32 2003 UTC revision 2394 by jan, Wed Nov 17 22:02:29 2004 UTC
# Line 1  Line 1 
1  # Copyright (c) 2001, 2002, 2003 by Intevation GmbH  # Copyright (c) 2001-2004 by Intevation GmbH
2  # Authors:  # Authors:
3  # Bernhard Herzog <[email protected]>  # Bernhard Herzog <[email protected]> (2001-2003)
4  # Jonathan Coles <[email protected]>  # Jonathan Coles <[email protected]> (2003)
5  # Frank Koormann <[email protected]>  # Frank Koormann <[email protected]> (2003)
6    # Jan-Oliver Wagner <[email protected]> (2003, 2004)
7  #  #
8  # This program is free software under the GPL (>=v2)  # This program is free software under the GPL (>=v2)
9  # Read the file COPYING coming with Thuban for details.  # Read the file COPYING coming with Thuban for details.
10    
11    from __future__ import generators
12    
13  __version__ = "$Revision$"  __version__ = "$Revision$"
14    # $Source$
15    # $Id$
16    
17  import cStringIO  import cStringIO
18    
# Line 16  from Thuban import _ Line 21  from Thuban import _
21  from wxPython.wx import wxPoint, wxRect, wxPen, wxBrush, wxFont, \  from wxPython.wx import wxPoint, wxRect, wxPen, wxBrush, wxFont, \
22      wxTRANSPARENT_PEN, wxTRANSPARENT_BRUSH, \      wxTRANSPARENT_PEN, wxTRANSPARENT_BRUSH, \
23      wxBLACK_PEN, wxBLACK, wxSOLID, wxCROSS_HATCH, wxSWISS, wxNORMAL, \      wxBLACK_PEN, wxBLACK, wxSOLID, wxCROSS_HATCH, wxSWISS, wxNORMAL, \
24      wxBitmapFromImage, wxImageFromStream, wxBITMAP_TYPE_BMP      wxBitmapFromImage, wxImageFromStream, wxBITMAP_TYPE_BMP, wxBITMAP_TYPE_JPEG
25    
26  from wxproj import draw_polygon_shape, draw_polygon_init  from wxproj import draw_polygon_shape, draw_polygon_init
27    
# Line 24  from Thuban.UI.common import Color2wxCol Line 29  from Thuban.UI.common import Color2wxCol
29  from Thuban.UI.classifier import ClassDataPreviewer  from Thuban.UI.classifier import ClassDataPreviewer
30  from Thuban.UI.scalebar import ScaleBar  from Thuban.UI.scalebar import ScaleBar
31    
32  from Thuban.Model.data import SHAPETYPE_POLYGON, SHAPETYPE_ARC, SHAPETYPE_POINT  from Thuban.Model.data import SHAPETYPE_POLYGON, SHAPETYPE_ARC, \
33         SHAPETYPE_POINT, RAW_SHAPEFILE
34    
35  from Thuban.Model.color import Transparent  from Thuban.Model.color import Transparent
36  import Thuban.Model.resource  import Thuban.Model.resource
37    
38  from baserenderer import BaseRenderer  from baserenderer import BaseRenderer
39    
40    
41    # Map the strings used for the format parameter of the draw_raster_data
42    # method to the appropriate wxWindows constants
43    raster_format_map = {
44        "BMP": wxBITMAP_TYPE_BMP,
45        "JPEG": wxBITMAP_TYPE_JPEG,
46        }
47    
48  class MapRenderer(BaseRenderer):  class MapRenderer(BaseRenderer):
49    
50      """Class to render a map onto a wxDC"""      """Class to render a map onto a wxDC"""
# Line 38  class MapRenderer(BaseRenderer): Line 52  class MapRenderer(BaseRenderer):
52      TRANSPARENT_PEN = wxTRANSPARENT_PEN      TRANSPARENT_PEN = wxTRANSPARENT_PEN
53      TRANSPARENT_BRUSH = wxTRANSPARENT_BRUSH      TRANSPARENT_BRUSH = wxTRANSPARENT_BRUSH
54    
55      make_point = wxPoint      def make_point(self, x, y):
56            return wxPoint(int(round(x)), int(round(y)))
57    
58      def tools_for_property(self, prop):      def tools_for_property(self, prop):
59          fill = prop.GetFill()          fill = prop.GetFill()
# Line 57  class MapRenderer(BaseRenderer): Line 72  class MapRenderer(BaseRenderer):
72      def low_level_renderer(self, layer):      def low_level_renderer(self, layer):
73          """Override inherited method to provide more efficient renderers          """Override inherited method to provide more efficient renderers
74    
75          For point shapes, return self.draw_point_shape and layer just as          If the underlying data format is not a shapefile or the layer
76          the base class method. For arc and polygon use the more          contains points shapes, simply use what the inherited method
77          efficient wxproj.draw_polygon_shape and its corresponding          returns.
78          parameter created with wxproj.draw_polygon_init.  
79            Otherwise, i.e. for arc and polygon use the more efficient
80            wxproj.draw_polygon_shape and its corresponding parameter
81            created with wxproj.draw_polygon_init.
82          """          """
83          shapetype = layer.ShapeType()          if (layer.ShapeStore().RawShapeFormat() == RAW_SHAPEFILE
84          if shapetype == SHAPETYPE_POINT:              and layer.ShapeType() in (SHAPETYPE_ARC, SHAPETYPE_POLYGON)):
             func = self.draw_point_shape  
             param = layer  
         else:  
85              offx, offy = self.offset              offx, offy = self.offset
86              param = draw_polygon_init(layer.ShapeStore().Shapefile(), self.dc,              return (True, draw_polygon_shape,
87                                        self.map.projection,                      draw_polygon_init(layer.ShapeStore().Shapefile(),
88                                          self.dc, self.map.projection,
89                                        layer.projection,                                        layer.projection,
90                                        self.scale, -self.scale,                                        self.scale, -self.scale, offx, offy))
91                                        offx, offy)          else:
92              func = draw_polygon_shape              return BaseRenderer.low_level_renderer(self, layer)
         return func, param  
93    
94      def label_font(self):      def label_font(self):
95          return wxFont(self.resolution * 10, wxSWISS, wxNORMAL, wxNORMAL)          return wxFont(int(round(self.resolution * 10)), wxSWISS, wxNORMAL,
96                          wxNORMAL)
97    
98      def draw_raster_data(self, data):      def draw_raster_data(self, data, format = 'BMP'):
99          stream = cStringIO.StringIO(data)          stream = cStringIO.StringIO(data)
100          image = wxImageFromStream(stream, wxBITMAP_TYPE_BMP)          image = wxImageFromStream(stream, raster_format_map[format])
101          bitmap = wxBitmapFromImage(image)          bitmap = wxBitmapFromImage(image)
102          self.dc.DrawBitmap(bitmap, 0, 0)          self.dc.DrawBitmap(bitmap, 0, 0)
103    
# Line 91  class ScreenRenderer(MapRenderer): Line 107  class ScreenRenderer(MapRenderer):
107      # On the screen we want to see only visible layers by default      # On the screen we want to see only visible layers by default
108      honor_visibility = 1      honor_visibility = 1
109    
110      def RenderMap(self, map, region, selected_layer, selected_shapes):      def RenderMap(self, selected_layer, selected_shapes):
111          """Render the map.          """Render the map.
112    
113          Only the given region (a tuple in window coordinates as returned          Only the given region (a tuple in window coordinates as returned
# Line 99  class ScreenRenderer(MapRenderer): Line 115  class ScreenRenderer(MapRenderer):
115          shapes given by the ids in selected_shapes in the          shapes given by the ids in selected_shapes in the
116          selected_layer.          selected_layer.
117          """          """
         self.update_region = region  
118          self.selected_layer = selected_layer          self.selected_layer = selected_layer
119          self.selected_shapes = selected_shapes          self.selected_shapes = selected_shapes
120          self.render_map(map)          self.render_map()
121    
122      def draw_shape_layer(self, layer):      def RenderMapIncrementally(self):
123          MapRenderer.draw_shape_layer(self, layer)          """Render the map.
         if layer is self.selected_layer and self.selected_shapes:  
             pen = wxPen(wxBLACK, 3, wxSOLID)  
             brush = wxBrush(wxBLACK, wxCROSS_HATCH)  
   
             shapetype = layer.ShapeType()  
             func, param = self.low_level_renderer(layer)  
             args = (pen, brush)  
             for index in self.selected_shapes:  
                 func(param, index, *args)  
124    
125      def layer_ids(self, layer):          Only the given region (a tuple in window coordinates as returned
126            by a wxrect's asTuple method) needs to be redrawn. Highlight the
127            shapes given by the ids in selected_shapes in the
128            selected_layer.
129            """
130            return self.render_map_incrementally()
131    
132        def draw_selection_incrementally(self, layer, selected_shapes):
133            """Draw the selected shapes in a emphasized way (i.e.
134            with a special pen and brush.
135            The drawing is performed incrementally, that means every
136            n shapes, the user can have interactions with the map.
137            n is currently fixed to 500.
138    
139            layer -- the layer where the shapes belong to.
140            selected_shapes -- a list of the shape-ids representing the
141                               selected shapes for the given layer.
142            """
143            pen = wxPen(wxBLACK, 3, wxSOLID)
144            brush = wxBrush(wxBLACK, wxCROSS_HATCH)
145    
146            shapetype = layer.ShapeType()
147            useraw, func, param = self.low_level_renderer(layer)
148            args = (pen, brush)
149    
150            # for point shapes we need to find out the properties
151            # to determine the size. Based on table and field,
152            # we can find out the properties for object - see below.
153            if shapetype == SHAPETYPE_POINT:
154                lc = layer.GetClassification()
155                field = layer.GetClassificationColumn()
156                table = layer.ShapeStore().Table()
157    
158            count = 0
159            for index in selected_shapes:
160                count += 1
161                shape = layer.Shape(index)
162    
163                # Get the size of the specific property for this
164                # point
165                if shapetype == SHAPETYPE_POINT:
166                    value = table.ReadValue(shape.ShapeID(), field)
167                    group = lc.FindGroup(value)
168                    size = group.GetProperties().GetSize()
169                    args = (pen, brush, size)
170    
171                if useraw:
172                    data = shape.RawData()
173                else:
174                    data = shape.Points()
175                func(param, data, *args)
176                if count % 500 == 0:
177                    yield True
178    
179        def layer_shapes(self, layer):
180          """Return the shapeids covered by the region that has to be redrawn          """Return the shapeids covered by the region that has to be redrawn
181    
182          Call the layer's ShapesInRegion method to determine the ids so          Call the layer's ShapesInRegion method to determine the ids so
# Line 137  class ScreenRenderer(MapRenderer): Line 197  class ScreenRenderer(MapRenderer):
197          offx, offy = self.offset          offx, offy = self.offset
198          xs = []          xs = []
199          ys = []          ys = []
200          x, y, width, height = self.update_region          x, y, width, height = self.region
201          for winx, winy in ((x, y), (x + width, y),          for winx, winy in ((x, y), (x + width, y),
202                             (x + width, y + height), (x, y + height)):                             (x + width, y + height), (x, y + height)):
203              px = (winx - offx) / scale              px = (winx - offx) / scale
# Line 158  class ExportRenderer(ScreenRenderer): Line 218  class ExportRenderer(ScreenRenderer):
218    
219      honor_visibility = 1      honor_visibility = 1
220    
221      def RenderMap(self, map, region, mapregion,      def __init__(self, *args, **kw):
222                    selected_layer, selected_shapes ):          """Initialize the ExportRenderer.
223    
224            In addition to all parameters of the the ScreenRender
225            constructor, this class requires and additional keyword argument
226            destination_region with a tuple (minx, miny, maxx, maxy) giving
227            the region in dc coordinates which is to contain the map.
228            """
229            self.destination_region = kw["destination_region"]
230            del kw["destination_region"]
231            ScreenRenderer.__init__(self, *args, **kw)
232    
233        def RenderMap(self, selected_layer, selected_shapes):
234          """Render the map.          """Render the map.
235    
236          The rendering device has been specified during initialisation.          The rendering device has been specified during initialisation.
237          The device border distance was set in Thuban.UI.view.OutputTranform().          The device border distance was set in
238            Thuban.UI.viewport.output_transform().
239    
240          RenderMap renders a frame set (one page frame, one around          RenderMap renders a frame set (one page frame, one around
241          legend/scalebar and one around the map), the map, the legend and the          legend/scalebar and one around the map), the map, the legend and
242          scalebar on the given DC. The map is rendered with the region displayed          the scalebar on the given DC. The map is rendered with the
243          in the canvas view, centered on the area available for map display.          region displayed in the canvas view, centered on the area
244            available for map display.
245          """          """
246    
         self.update_region = region  
247          self.selected_layer = selected_layer          self.selected_layer = selected_layer
248          self.selected_shapes = selected_shapes          self.selected_shapes = selected_shapes
249    
250          # Get some dimensions          # Get some dimensions
251          llx, lly, urx, ury = region          llx, lly, urx, ury = self.region
252          self.mapregion = mapregion          mminx, mminy, mmaxx, mmaxy = self.destination_region
         mminx, mminy, mmaxx, mmaxy = self.mapregion  
253    
254          # Manipulate the offset to position the map          # Manipulate the offset to position the map
255          offx, offy = self.offset          offx, offy = self.offset
# Line 199  class ExportRenderer(ScreenRenderer): Line 270  class ExportRenderer(ScreenRenderer):
270          self.dc.DestroyClippingRegion()          self.dc.DestroyClippingRegion()
271          self.dc.SetClippingRegion(mminx+self.shiftx, mminy+self.shifty,          self.dc.SetClippingRegion(mminx+self.shiftx, mminy+self.shifty,
272                                    urx, ury)                                    urx, ury)
273          self.render_map(map)          self.render_map()
274          self.dc.EndDrawing()          self.dc.EndDrawing()
275    
276          # Draw the rest (frames, legend, scalebar)          # Draw the rest (frames, legend, scalebar)
# Line 210  class ExportRenderer(ScreenRenderer): Line 281  class ExportRenderer(ScreenRenderer):
281          font = wxFont(self.resolution * 10, wxSWISS, wxNORMAL, wxNORMAL)          font = wxFont(self.resolution * 10, wxSWISS, wxNORMAL, wxNORMAL)
282          self.dc.SetFont(font)          self.dc.SetFont(font)
283    
284          self.render_frame(region)          self.render_frame()
285          self.render_legend(map)          self.render_legend()
286          self.render_scalebar(map)          self.render_scalebar()
287          self.dc.EndDrawing()          self.dc.EndDrawing()
288    
289      def render_frame(self, region):      def render_frame(self):
290          """Render the frames for map and legend/scalebar."""          """Render the frames for map and legend/scalebar."""
291    
292          dc = self.dc          dc = self.dc
# Line 224  class ExportRenderer(ScreenRenderer): Line 295  class ExportRenderer(ScreenRenderer):
295    
296          # Dimension stuff          # Dimension stuff
297          width, height = dc.GetSizeTuple()          width, height = dc.GetSizeTuple()
298          mminx, mminy, mmaxx, mmaxy = self.mapregion          mminx, mminy, mmaxx, mmaxy = self.destination_region
299    
300          # Page Frame          # Page Frame
301          dc.DrawRectangle(15,15,width-30, (mmaxy-mminy)+10)          dc.DrawRectangle(15,15,width-30, (mmaxy-mminy)+10)
302    
303          # Map Frame          # Map Frame
304          llx, lly, urx, ury = region          llx, lly, urx, ury = self.region
305          dc.DrawRectangle(mminx + self.shiftx, mminy + self.shifty, urx, ury)          dc.DrawRectangle(mminx + self.shiftx, mminy + self.shifty, urx, ury)
306    
307          # Legend Frame          # Legend Frame
# Line 240  class ExportRenderer(ScreenRenderer): Line 311  class ExportRenderer(ScreenRenderer):
311          dc.SetClippingRegion(mmaxx+10,mminy,          dc.SetClippingRegion(mmaxx+10,mminy,
312                               (width-20) - (mmaxx+10), mmaxy-mminy)                               (width-20) - (mmaxx+10), mmaxy-mminy)
313    
314      def render_legend(self, map):      def render_legend(self):
315          """Render the legend on the Map."""          """Render the legend on the Map."""
316    
317          previewer = ClassDataPreviewer()          previewer = ClassDataPreviewer()
# Line 250  class ExportRenderer(ScreenRenderer): Line 321  class ExportRenderer(ScreenRenderer):
321    
322          # Dimension stuff          # Dimension stuff
323          width, height = dc.GetSizeTuple()          width, height = dc.GetSizeTuple()
324          mminx, mminy, mmaxx, mmaxy = self.mapregion          mminx, mminy, mmaxx, mmaxy = self.destination_region
325          textwidth, textheight = dc.GetTextExtent("0")          textwidth, textheight = dc.GetTextExtent("0")
326          iconwidth  = textheight          iconwidth  = textheight
327          iconheight = textheight          iconheight = textheight
# Line 262  class ExportRenderer(ScreenRenderer): Line 333  class ExportRenderer(ScreenRenderer):
333    
334          # Render the legend          # Render the legend
335          dc.SetTextForeground(wxBLACK)          dc.SetTextForeground(wxBLACK)
336          if map.HasLayers():          if self.map.HasLayers():
337              layers = map.Layers()              layers = self.map.Layers()[:]
338              layers.reverse()              layers.reverse()
339              for l in layers:              for l in layers:
340                  if l.Visible():                  if l.Visible():
# Line 284  class ExportRenderer(ScreenRenderer): Line 355  class ExportRenderer(ScreenRenderer):
355                                              posx+2*dx+iconwidth, posy)                                              posx+2*dx+iconwidth, posy)
356                                  posy+=stepy                                  posy+=stepy
357    
358      def render_scalebar(self, map):      def render_scalebar(self):
359          """Render the scalebar."""          """Render the scalebar."""
360    
361          scalebar = ScaleBar(map)          scalebar = ScaleBar(self.map)
362    
363          # Dimension stuff          # Dimension stuff
364          width, height = self.dc.GetSizeTuple()          width, height = self.dc.GetSizeTuple()
365          mminx, mminy, mmaxx, mmaxy = self.mapregion          mminx, mminy, mmaxx, mmaxy = self.destination_region
366    
367          # Render the scalebar          # Render the scalebar
368          scalebar.DrawScaleBar(self.scale, self.dc,          scalebar.DrawScaleBar(self.scale, self.dc,
Legend:
Removed from v.1552  
changed lines
  Added in v.2394
[email protected]
 ViewVC Help
Powered by ViewVC 1.1.26