/[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 2454 by bh, Mon Dec 13 18:26:11 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, \
25        wxBITMAP_TYPE_JPEG, wxBITMAP_TYPE_PNG, wxBITMAP_TYPE_TIF, wxBITMAP_TYPE_GIF
26    
27  from wxproj import draw_polygon_shape, draw_polygon_init  from wxproj import draw_polygon_shape, draw_polygon_init
28    
# Line 24  from Thuban.UI.common import Color2wxCol Line 30  from Thuban.UI.common import Color2wxCol
30  from Thuban.UI.classifier import ClassDataPreviewer  from Thuban.UI.classifier import ClassDataPreviewer
31  from Thuban.UI.scalebar import ScaleBar  from Thuban.UI.scalebar import ScaleBar
32    
33  from Thuban.Model.data import SHAPETYPE_POLYGON, SHAPETYPE_ARC, SHAPETYPE_POINT  from Thuban.Model.data import SHAPETYPE_POLYGON, SHAPETYPE_ARC, \
34         SHAPETYPE_POINT, RAW_SHAPEFILE
35    
36  from Thuban.Model.color import Transparent  from Thuban.Model.color import Transparent
37  import Thuban.Model.resource  import Thuban.Model.resource
38    
39  from baserenderer import BaseRenderer  from baserenderer import BaseRenderer
40    
41    
42    # Map the strings used for the format parameter of the draw_raster_data
43    # method to the appropriate wxWindows constants
44    raster_format_map = {
45        "BMP": wxBITMAP_TYPE_BMP,
46        "JPEG": wxBITMAP_TYPE_JPEG,
47        "PNG": wxBITMAP_TYPE_PNG,
48        "TIFF": wxBITMAP_TYPE_TIF,
49        "GIF": wxBITMAP_TYPE_GIF,
50        }
51    
52  class MapRenderer(BaseRenderer):  class MapRenderer(BaseRenderer):
53    
54      """Class to render a map onto a wxDC"""      """Class to render a map onto a wxDC"""
# Line 38  class MapRenderer(BaseRenderer): Line 56  class MapRenderer(BaseRenderer):
56      TRANSPARENT_PEN = wxTRANSPARENT_PEN      TRANSPARENT_PEN = wxTRANSPARENT_PEN
57      TRANSPARENT_BRUSH = wxTRANSPARENT_BRUSH      TRANSPARENT_BRUSH = wxTRANSPARENT_BRUSH
58    
59      make_point = wxPoint      def make_point(self, x, y):
60            return wxPoint(int(round(x)), int(round(y)))
61    
62      def tools_for_property(self, prop):      def tools_for_property(self, prop):
63          fill = prop.GetFill()          fill = prop.GetFill()
# Line 57  class MapRenderer(BaseRenderer): Line 76  class MapRenderer(BaseRenderer):
76      def low_level_renderer(self, layer):      def low_level_renderer(self, layer):
77          """Override inherited method to provide more efficient renderers          """Override inherited method to provide more efficient renderers
78    
79          For point shapes, return self.draw_point_shape and layer just as          If the underlying data format is not a shapefile or the layer
80          the base class method. For arc and polygon use the more          contains points shapes, simply use what the inherited method
81          efficient wxproj.draw_polygon_shape and its corresponding          returns.
82          parameter created with wxproj.draw_polygon_init.  
83            Otherwise, i.e. for arc and polygon use the more efficient
84            wxproj.draw_polygon_shape and its corresponding parameter
85            created with wxproj.draw_polygon_init.
86          """          """
87          shapetype = layer.ShapeType()          if (layer.ShapeStore().RawShapeFormat() == RAW_SHAPEFILE
88          if shapetype == SHAPETYPE_POINT:              and layer.ShapeType() in (SHAPETYPE_ARC, SHAPETYPE_POLYGON)):
             func = self.draw_point_shape  
             param = layer  
         else:  
89              offx, offy = self.offset              offx, offy = self.offset
90              param = draw_polygon_init(layer.ShapeStore().Shapefile(), self.dc,              return (True, draw_polygon_shape,
91                                        self.map.projection,                      draw_polygon_init(layer.ShapeStore().Shapefile(),
92                                          self.dc, self.map.projection,
93                                        layer.projection,                                        layer.projection,
94                                        self.scale, -self.scale,                                        self.scale, -self.scale, offx, offy))
95                                        offx, offy)          else:
96              func = draw_polygon_shape              return BaseRenderer.low_level_renderer(self, layer)
         return func, param  
97    
98      def label_font(self):      def label_font(self):
99          return wxFont(self.resolution * 10, wxSWISS, wxNORMAL, wxNORMAL)          return wxFont(int(round(self.resolution * 10)), wxSWISS, wxNORMAL,
100                          wxNORMAL)
101    
102      def draw_raster_data(self, data):      def draw_raster_data(self, data, format = 'BMP'):
103          stream = cStringIO.StringIO(data)          stream = cStringIO.StringIO(data)
104          image = wxImageFromStream(stream, wxBITMAP_TYPE_BMP)          image = wxImageFromStream(stream, raster_format_map[format])
105          bitmap = wxBitmapFromImage(image)          bitmap = wxBitmapFromImage(image)
106          self.dc.DrawBitmap(bitmap, 0, 0)          self.dc.DrawBitmap(bitmap, 0, 0)
107    
# Line 91  class ScreenRenderer(MapRenderer): Line 111  class ScreenRenderer(MapRenderer):
111      # On the screen we want to see only visible layers by default      # On the screen we want to see only visible layers by default
112      honor_visibility = 1      honor_visibility = 1
113    
114      def RenderMap(self, map, region, selected_layer, selected_shapes):      def RenderMap(self, selected_layer, selected_shapes):
115          """Render the map.          """Render the map.
116    
117          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 119  class ScreenRenderer(MapRenderer):
119          shapes given by the ids in selected_shapes in the          shapes given by the ids in selected_shapes in the
120          selected_layer.          selected_layer.
121          """          """
         self.update_region = region  
122          self.selected_layer = selected_layer          self.selected_layer = selected_layer
123          self.selected_shapes = selected_shapes          self.selected_shapes = selected_shapes
124          self.render_map(map)          self.render_map()
125    
126      def draw_shape_layer(self, layer):      def RenderMapIncrementally(self):
127          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)  
128    
129      def layer_ids(self, layer):          Only the given region (a tuple in window coordinates as returned
130            by a wxrect's asTuple method) needs to be redrawn. Highlight the
131            shapes given by the ids in selected_shapes in the
132            selected_layer.
133            """
134            return self.render_map_incrementally()
135    
136        def draw_selection_incrementally(self, layer, selected_shapes):
137            """Draw the selected shapes in a emphasized way (i.e.
138            with a special pen and brush.
139            The drawing is performed incrementally, that means every
140            n shapes, the user can have interactions with the map.
141            n is currently fixed to 500.
142    
143            layer -- the layer where the shapes belong to.
144            selected_shapes -- a list of the shape-ids representing the
145                               selected shapes for the given layer.
146            """
147            pen = wxPen(wxBLACK, 3, wxSOLID)
148            brush = wxBrush(wxBLACK, wxCROSS_HATCH)
149    
150            shapetype = layer.ShapeType()
151            useraw, func, param = self.low_level_renderer(layer)
152            args = (pen, brush)
153    
154            # for point shapes we need to find out the properties
155            # to determine the size. Based on table and field,
156            # we can find out the properties for object - see below.
157            if shapetype == SHAPETYPE_POINT:
158                lc = layer.GetClassification()
159                field = layer.GetClassificationColumn()
160                table = layer.ShapeStore().Table()
161    
162            count = 0
163            for index in selected_shapes:
164                count += 1
165                shape = layer.Shape(index)
166    
167                # Get the size of the specific property for this
168                # point
169                if shapetype == SHAPETYPE_POINT:
170                    value = table.ReadValue(shape.ShapeID(), field)
171                    group = lc.FindGroup(value)
172                    size = group.GetProperties().GetSize()
173                    args = (pen, brush, size)
174    
175                if useraw:
176                    data = shape.RawData()
177                else:
178                    data = shape.Points()
179                func(param, data, *args)
180                if count % 500 == 0:
181                    yield True
182    
183        def layer_shapes(self, layer):
184          """Return the shapeids covered by the region that has to be redrawn          """Return the shapeids covered by the region that has to be redrawn
185    
186          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 201  class ScreenRenderer(MapRenderer):
201          offx, offy = self.offset          offx, offy = self.offset
202          xs = []          xs = []
203          ys = []          ys = []
204          x, y, width, height = self.update_region          x, y, width, height = self.region
205          for winx, winy in ((x, y), (x + width, y),          for winx, winy in ((x, y), (x + width, y),
206                             (x + width, y + height), (x, y + height)):                             (x + width, y + height), (x, y + height)):
207              px = (winx - offx) / scale              px = (winx - offx) / scale
# Line 158  class ExportRenderer(ScreenRenderer): Line 222  class ExportRenderer(ScreenRenderer):
222    
223      honor_visibility = 1      honor_visibility = 1
224    
225      def RenderMap(self, map, region, mapregion,      def __init__(self, *args, **kw):
226                    selected_layer, selected_shapes ):          """Initialize the ExportRenderer.
227    
228            In addition to all parameters of the the ScreenRender
229            constructor, this class requires and additional keyword argument
230            destination_region with a tuple (minx, miny, maxx, maxy) giving
231            the region in dc coordinates which is to contain the map.
232            """
233            self.destination_region = kw["destination_region"]
234            del kw["destination_region"]
235            ScreenRenderer.__init__(self, *args, **kw)
236    
237        def RenderMap(self, selected_layer, selected_shapes):
238          """Render the map.          """Render the map.
239    
240          The rendering device has been specified during initialisation.          The rendering device has been specified during initialisation.
241          The device border distance was set in Thuban.UI.view.OutputTranform().          The device border distance was set in
242            Thuban.UI.viewport.output_transform().
243    
244          RenderMap renders a frame set (one page frame, one around          RenderMap renders a frame set (one page frame, one around
245          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
246          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
247          in the canvas view, centered on the area available for map display.          region displayed in the canvas view, centered on the area
248            available for map display.
249          """          """
250    
         self.update_region = region  
251          self.selected_layer = selected_layer          self.selected_layer = selected_layer
252          self.selected_shapes = selected_shapes          self.selected_shapes = selected_shapes
253    
254          # Get some dimensions          # Get some dimensions
255          llx, lly, urx, ury = region          llx, lly, urx, ury = self.region
256          self.mapregion = mapregion          mminx, mminy, mmaxx, mmaxy = self.destination_region
         mminx, mminy, mmaxx, mmaxy = self.mapregion  
257    
258          # Manipulate the offset to position the map          # Manipulate the offset to position the map
259          offx, offy = self.offset          offx, offy = self.offset
# Line 193  class ExportRenderer(ScreenRenderer): Line 268  class ExportRenderer(ScreenRenderer):
268          self.shifty = (mmaxy - mminy)*0.5 - (ury - lly)*0.5          self.shifty = (mmaxy - mminy)*0.5 - (ury - lly)*0.5
269    
270          self.offset = (offx+self.shiftx, offy+self.shifty)          self.offset = (offx+self.shiftx, offy+self.shifty)
271            self.region = (llx + self.shiftx, lly + self.shifty, urx, ury)
272    
273          # Draw the map          # Draw the map
274          self.dc.BeginDrawing()          self.dc.BeginDrawing()
275          self.dc.DestroyClippingRegion()          self.dc.DestroyClippingRegion()
276          self.dc.SetClippingRegion(mminx+self.shiftx, mminy+self.shifty,          self.dc.SetClippingRegion(mminx+self.shiftx, mminy+self.shifty,
277                                    urx, ury)                                    urx, ury)
278          self.render_map(map)          self.render_map()
279          self.dc.EndDrawing()          self.dc.EndDrawing()
280    
281          # Draw the rest (frames, legend, scalebar)          # Draw the rest (frames, legend, scalebar)
# Line 210  class ExportRenderer(ScreenRenderer): Line 286  class ExportRenderer(ScreenRenderer):
286          font = wxFont(self.resolution * 10, wxSWISS, wxNORMAL, wxNORMAL)          font = wxFont(self.resolution * 10, wxSWISS, wxNORMAL, wxNORMAL)
287          self.dc.SetFont(font)          self.dc.SetFont(font)
288    
289          self.render_frame(region)          self.render_frame()
290          self.render_legend(map)          self.render_legend()
291          self.render_scalebar(map)          self.render_scalebar()
292          self.dc.EndDrawing()          self.dc.EndDrawing()
293    
294      def render_frame(self, region):      def render_frame(self):
295          """Render the frames for map and legend/scalebar."""          """Render the frames for map and legend/scalebar."""
296    
297          dc = self.dc          dc = self.dc
# Line 224  class ExportRenderer(ScreenRenderer): Line 300  class ExportRenderer(ScreenRenderer):
300    
301          # Dimension stuff          # Dimension stuff
302          width, height = dc.GetSizeTuple()          width, height = dc.GetSizeTuple()
303          mminx, mminy, mmaxx, mmaxy = self.mapregion          mminx, mminy, mmaxx, mmaxy = self.destination_region
304    
305          # Page Frame          # Page Frame
306          dc.DrawRectangle(15,15,width-30, (mmaxy-mminy)+10)          dc.DrawRectangle(15,15,width-30, (mmaxy-mminy)+10)
307    
308          # Map Frame          # Map Frame
309          llx, lly, urx, ury = region          llx, lly, urx, ury = self.region
310          dc.DrawRectangle(mminx + self.shiftx, mminy + self.shifty, urx, ury)          dc.DrawRectangle(mminx + self.shiftx, mminy + self.shifty, urx, ury)
311    
312          # Legend Frame          # Legend Frame
# Line 240  class ExportRenderer(ScreenRenderer): Line 316  class ExportRenderer(ScreenRenderer):
316          dc.SetClippingRegion(mmaxx+10,mminy,          dc.SetClippingRegion(mmaxx+10,mminy,
317                               (width-20) - (mmaxx+10), mmaxy-mminy)                               (width-20) - (mmaxx+10), mmaxy-mminy)
318    
319      def render_legend(self, map):      def render_legend(self):
320          """Render the legend on the Map."""          """Render the legend on the Map."""
321    
322          previewer = ClassDataPreviewer()          previewer = ClassDataPreviewer()
# Line 250  class ExportRenderer(ScreenRenderer): Line 326  class ExportRenderer(ScreenRenderer):
326    
327          # Dimension stuff          # Dimension stuff
328          width, height = dc.GetSizeTuple()          width, height = dc.GetSizeTuple()
329          mminx, mminy, mmaxx, mmaxy = self.mapregion          mminx, mminy, mmaxx, mmaxy = self.destination_region
330          textwidth, textheight = dc.GetTextExtent("0")          textwidth, textheight = dc.GetTextExtent("0")
331          iconwidth  = textheight          iconwidth  = textheight
332          iconheight = textheight          iconheight = textheight
# Line 262  class ExportRenderer(ScreenRenderer): Line 338  class ExportRenderer(ScreenRenderer):
338    
339          # Render the legend          # Render the legend
340          dc.SetTextForeground(wxBLACK)          dc.SetTextForeground(wxBLACK)
341          if map.HasLayers():          if self.map.HasLayers():
342              layers = map.Layers()              layers = self.map.Layers()[:]
343              layers.reverse()              layers.reverse()
344              for l in layers:              for l in layers:
345                  if l.Visible():                  if l.Visible():
# Line 284  class ExportRenderer(ScreenRenderer): Line 360  class ExportRenderer(ScreenRenderer):
360                                              posx+2*dx+iconwidth, posy)                                              posx+2*dx+iconwidth, posy)
361                                  posy+=stepy                                  posy+=stepy
362    
363      def render_scalebar(self, map):      def render_scalebar(self):
364          """Render the scalebar."""          """Render the scalebar."""
365    
366          scalebar = ScaleBar(map)          scalebar = ScaleBar(self.map)
367    
368          # Dimension stuff          # Dimension stuff
369          width, height = self.dc.GetSizeTuple()          width, height = self.dc.GetSizeTuple()
370          mminx, mminy, mmaxx, mmaxy = self.mapregion          mminx, mminy, mmaxx, mmaxy = self.destination_region
371    
372          # Render the scalebar          # Render the scalebar
373          scalebar.DrawScaleBar(self.scale, self.dc,          scalebar.DrawScaleBar(self.scale, self.dc,
Legend:
Removed from v.1552  
changed lines
  Added in v.2454
[email protected]
 ViewVC Help
Powered by ViewVC 1.1.26