/[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 2562 by jonathan, Wed Feb 16 21:14:47 2005 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, \
26        wxBITMAP_TYPE_GIF, wxEmptyImage, wxMask, wxBitmapFromBits
27    
28  from wxproj import draw_polygon_shape, draw_polygon_init  from wxproj import draw_polygon_shape, draw_polygon_init
29    
# Line 24  from Thuban.UI.common import Color2wxCol Line 31  from Thuban.UI.common import Color2wxCol
31  from Thuban.UI.classifier import ClassDataPreviewer  from Thuban.UI.classifier import ClassDataPreviewer
32  from Thuban.UI.scalebar import ScaleBar  from Thuban.UI.scalebar import ScaleBar
33    
34  from Thuban.Model.data import SHAPETYPE_POLYGON, SHAPETYPE_ARC, SHAPETYPE_POINT  from Thuban.Model.data import SHAPETYPE_POLYGON, SHAPETYPE_ARC, \
35         SHAPETYPE_POINT, RAW_SHAPEFILE
36    
37  from Thuban.Model.color import Transparent  from Thuban.Model.color import Transparent
38  import Thuban.Model.resource  import Thuban.Model.resource
39    
40  from baserenderer import BaseRenderer  from baserenderer import BaseRenderer
41    
42    from math import floor
43    
44    from types import StringType
45    
46    from Thuban.version import versions
47    
48    
49    # Map the strings used for the format parameter of the draw_raster_data
50    # method to the appropriate wxWindows constants
51    raster_format_map = {
52        "BMP": wxBITMAP_TYPE_BMP,
53        "JPEG": wxBITMAP_TYPE_JPEG,
54        "PNG": wxBITMAP_TYPE_PNG,
55        "TIFF": wxBITMAP_TYPE_TIF,
56        "GIF": wxBITMAP_TYPE_GIF,
57        }
58    
59  class MapRenderer(BaseRenderer):  class MapRenderer(BaseRenderer):
60    
61      """Class to render a map onto a wxDC"""      """Class to render a map onto a wxDC"""
# Line 38  class MapRenderer(BaseRenderer): Line 63  class MapRenderer(BaseRenderer):
63      TRANSPARENT_PEN = wxTRANSPARENT_PEN      TRANSPARENT_PEN = wxTRANSPARENT_PEN
64      TRANSPARENT_BRUSH = wxTRANSPARENT_BRUSH      TRANSPARENT_BRUSH = wxTRANSPARENT_BRUSH
65    
66      make_point = wxPoint      def make_point(self, x, y):
67            return wxPoint(int(round(x)), int(round(y)))
68    
69      def tools_for_property(self, prop):      def tools_for_property(self, prop):
70          fill = prop.GetFill()          fill = prop.GetFill()
# Line 57  class MapRenderer(BaseRenderer): Line 83  class MapRenderer(BaseRenderer):
83      def low_level_renderer(self, layer):      def low_level_renderer(self, layer):
84          """Override inherited method to provide more efficient renderers          """Override inherited method to provide more efficient renderers
85    
86          For point shapes, return self.draw_point_shape and layer just as          If the underlying data format is not a shapefile or the layer
87          the base class method. For arc and polygon use the more          contains points shapes, simply use what the inherited method
88          efficient wxproj.draw_polygon_shape and its corresponding          returns.
89          parameter created with wxproj.draw_polygon_init.  
90            Otherwise, i.e. for arc and polygon use the more efficient
91            wxproj.draw_polygon_shape and its corresponding parameter
92            created with wxproj.draw_polygon_init.
93          """          """
94          shapetype = layer.ShapeType()          if (layer.ShapeStore().RawShapeFormat() == RAW_SHAPEFILE
95          if shapetype == SHAPETYPE_POINT:              and layer.ShapeType() in (SHAPETYPE_ARC, SHAPETYPE_POLYGON)):
             func = self.draw_point_shape  
             param = layer  
         else:  
96              offx, offy = self.offset              offx, offy = self.offset
97              param = draw_polygon_init(layer.ShapeStore().Shapefile(), self.dc,              x = lambda a, b, c, d: None
98                                        self.map.projection,              #return (True, x, None)
99                return (True, draw_polygon_shape,
100                        draw_polygon_init(layer.ShapeStore().Shapefile(),
101                                          self.dc, self.map.projection,
102                                        layer.projection,                                        layer.projection,
103                                        self.scale, -self.scale,                                        self.scale, -self.scale, offx, offy))
104                                        offx, offy)          else:
105              func = draw_polygon_shape              return BaseRenderer.low_level_renderer(self, layer)
         return func, param  
106    
107      def label_font(self):      def label_font(self):
108          return wxFont(self.resolution * 10, wxSWISS, wxNORMAL, wxNORMAL)          return wxFont(int(round(self.resolution * 10)), wxSWISS, wxNORMAL,
109                          wxNORMAL)
110    
111        def draw_raster_data(self, x,y, data, format = 'BMP'):
112    
113            mask = None
114            width = data[0]
115            height = data[1]
116            image_data, mask_data, alpha_data = data[2]
117    
118            if versions['wxPython-tuple'] < (2,5,3):
119                alpha_data = None
120    
121            if format == 'RAW':
122                image = wxEmptyImage(width, height)
123                image.SetData(image_data)
124                if mask_data is not None:
125                    mask = wxBitmapFromBits(mask_data, width, height, 1)
126                    mask = wxMask(mask)
127                elif alpha_data is not None:
128                    image.SetAlphaData(alpha_data)
129            else:
130                stream = cStringIO.StringIO(image_data)
131                image = wxImageFromStream(stream, raster_format_map[format])
132                if mask_data is not None:
133                    stream = cStringIO.StringIO(mask_data)
134                    mask = wxImageFromStream(stream, raster_format_map[format])
135                    mask = wxMask(wxBitmapFromImage(mask, 1))
136                elif alpha_data is not None:
137                    stream = cStringIO.StringIO(alpha_data)
138                    alpha = wxImageFromStream(stream, raster_format_map[format])
139                    image.SetAlpha(alpha.GetData()[:])
140    
     def draw_raster_data(self, data):  
         stream = cStringIO.StringIO(data)  
         image = wxImageFromStream(stream, wxBITMAP_TYPE_BMP)  
141          bitmap = wxBitmapFromImage(image)          bitmap = wxBitmapFromImage(image)
142          self.dc.DrawBitmap(bitmap, 0, 0)          bitmap.SetMask(mask)
143    
144            self.dc.DrawBitmap(bitmap, int(round(x)), int(round(y)), True)
145    
146    
147  class ScreenRenderer(MapRenderer):  class ScreenRenderer(MapRenderer):
# Line 91  class ScreenRenderer(MapRenderer): Line 149  class ScreenRenderer(MapRenderer):
149      # On the screen we want to see only visible layers by default      # On the screen we want to see only visible layers by default
150      honor_visibility = 1      honor_visibility = 1
151    
152      def RenderMap(self, map, region, selected_layer, selected_shapes):      def RenderMap(self, selected_layer, selected_shapes):
153          """Render the map.          """Render the map.
154    
155          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 157  class ScreenRenderer(MapRenderer):
157          shapes given by the ids in selected_shapes in the          shapes given by the ids in selected_shapes in the
158          selected_layer.          selected_layer.
159          """          """
         self.update_region = region  
160          self.selected_layer = selected_layer          self.selected_layer = selected_layer
161          self.selected_shapes = selected_shapes          self.selected_shapes = selected_shapes
162          self.render_map(map)          self.render_map()
163    
164        def RenderMapIncrementally(self):
165            """Render the map.
166    
167            Only the given region (a tuple in window coordinates as returned
168            by a wxrect's asTuple method) needs to be redrawn. Highlight the
169            shapes given by the ids in selected_shapes in the
170            selected_layer.
171            """
172            return self.render_map_incrementally()
173    
174      def draw_shape_layer(self, layer):      def draw_selection_incrementally(self, layer, selected_shapes):
175          MapRenderer.draw_shape_layer(self, layer)          """Draw the selected shapes in a emphasized way (i.e.
176          if layer is self.selected_layer and self.selected_shapes:          with a special pen and brush.
177              pen = wxPen(wxBLACK, 3, wxSOLID)          The drawing is performed incrementally, that means every
178              brush = wxBrush(wxBLACK, wxCROSS_HATCH)          n shapes, the user can have interactions with the map.
179            n is currently fixed to 500.
180              shapetype = layer.ShapeType()  
181              func, param = self.low_level_renderer(layer)          layer -- the layer where the shapes belong to.
182              args = (pen, brush)          selected_shapes -- a list of the shape-ids representing the
183              for index in self.selected_shapes:                             selected shapes for the given layer.
184                  func(param, index, *args)          """
185            pen = wxPen(wxBLACK, 3, wxSOLID)
186            brush = wxBrush(wxBLACK, wxCROSS_HATCH)
187    
188      def layer_ids(self, layer):          shapetype = layer.ShapeType()
189            useraw, func, param = self.low_level_renderer(layer)
190            args = (pen, brush)
191    
192            # for point shapes we need to find out the properties
193            # to determine the size. Based on table and field,
194            # we can find out the properties for object - see below.
195            if shapetype == SHAPETYPE_POINT:
196                lc = layer.GetClassification()
197                field = layer.GetClassificationColumn()
198                table = layer.ShapeStore().Table()
199    
200            count = 0
201            for index in selected_shapes:
202                count += 1
203                shape = layer.Shape(index)
204    
205                # Get the size of the specific property for this
206                # point
207                if shapetype == SHAPETYPE_POINT and field is not None:
208                    value = table.ReadValue(shape.ShapeID(), field)
209                    group = lc.FindGroup(value)
210                    size = group.GetProperties().GetSize()
211                    args = (pen, brush, size)
212    
213                if useraw:
214                    data = shape.RawData()
215                else:
216                    data = shape.Points()
217                func(param, data, *args)
218                if count % 500 == 0:
219                    yield True
220    
221        def layer_shapes(self, layer):
222          """Return the shapeids covered by the region that has to be redrawn          """Return the shapeids covered by the region that has to be redrawn
223    
224          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 239  class ScreenRenderer(MapRenderer):
239          offx, offy = self.offset          offx, offy = self.offset
240          xs = []          xs = []
241          ys = []          ys = []
242          x, y, width, height = self.update_region          x, y, width, height = self.region
243          for winx, winy in ((x, y), (x + width, y),          for winx, winy in ((x, y), (x + width, y),
244                             (x + width, y + height), (x, y + height)):                             (x + width, y + height), (x, y + height)):
245              px = (winx - offx) / scale              px = (winx - offx) / scale
# Line 158  class ExportRenderer(ScreenRenderer): Line 260  class ExportRenderer(ScreenRenderer):
260    
261      honor_visibility = 1      honor_visibility = 1
262    
263      def RenderMap(self, map, region, mapregion,      def __init__(self, *args, **kw):
264                    selected_layer, selected_shapes ):          """Initialize the ExportRenderer.
265    
266            In addition to all parameters of the the ScreenRender
267            constructor, this class requires and additional keyword argument
268            destination_region with a tuple (minx, miny, maxx, maxy) giving
269            the region in dc coordinates which is to contain the map.
270            """
271            self.destination_region = kw["destination_region"]
272            del kw["destination_region"]
273            ScreenRenderer.__init__(self, *args, **kw)
274    
275        def RenderMap(self, selected_layer, selected_shapes):
276          """Render the map.          """Render the map.
277    
278          The rendering device has been specified during initialisation.          The rendering device has been specified during initialisation.
279          The device border distance was set in Thuban.UI.view.OutputTranform().          The device border distance was set in
280            Thuban.UI.viewport.output_transform().
281    
282          RenderMap renders a frame set (one page frame, one around          RenderMap renders a frame set (one page frame, one around
283          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
284          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
285          in the canvas view, centered on the area available for map display.          region displayed in the canvas view, centered on the area
286            available for map display.
287          """          """
288    
         self.update_region = region  
289          self.selected_layer = selected_layer          self.selected_layer = selected_layer
290          self.selected_shapes = selected_shapes          self.selected_shapes = selected_shapes
291    
292          # Get some dimensions          # Get some dimensions
293          llx, lly, urx, ury = region          llx, lly, urx, ury = self.region
294          self.mapregion = mapregion          mminx, mminy, mmaxx, mmaxy = self.destination_region
         mminx, mminy, mmaxx, mmaxy = self.mapregion  
295    
296          # Manipulate the offset to position the map          # Manipulate the offset to position the map
297          offx, offy = self.offset          offx, offy = self.offset
# Line 193  class ExportRenderer(ScreenRenderer): Line 306  class ExportRenderer(ScreenRenderer):
306          self.shifty = (mmaxy - mminy)*0.5 - (ury - lly)*0.5          self.shifty = (mmaxy - mminy)*0.5 - (ury - lly)*0.5
307    
308          self.offset = (offx+self.shiftx, offy+self.shifty)          self.offset = (offx+self.shiftx, offy+self.shifty)
309            self.region = (llx + self.shiftx, lly + self.shifty, urx, ury)
310    
311          # Draw the map          # Draw the map
312          self.dc.BeginDrawing()          self.dc.BeginDrawing()
313          self.dc.DestroyClippingRegion()          self.dc.DestroyClippingRegion()
314          self.dc.SetClippingRegion(mminx+self.shiftx, mminy+self.shifty,          self.dc.SetClippingRegion(mminx+self.shiftx, mminy+self.shifty,
315                                    urx, ury)                                    urx, ury)
316          self.render_map(map)          self.render_map()
317          self.dc.EndDrawing()          self.dc.EndDrawing()
318    
319          # Draw the rest (frames, legend, scalebar)          # Draw the rest (frames, legend, scalebar)
# Line 210  class ExportRenderer(ScreenRenderer): Line 324  class ExportRenderer(ScreenRenderer):
324          font = wxFont(self.resolution * 10, wxSWISS, wxNORMAL, wxNORMAL)          font = wxFont(self.resolution * 10, wxSWISS, wxNORMAL, wxNORMAL)
325          self.dc.SetFont(font)          self.dc.SetFont(font)
326    
327          self.render_frame(region)          self.render_frame()
328          self.render_legend(map)          self.render_legend()
329          self.render_scalebar(map)          self.render_scalebar()
330          self.dc.EndDrawing()          self.dc.EndDrawing()
331    
332      def render_frame(self, region):      def render_frame(self):
333          """Render the frames for map and legend/scalebar."""          """Render the frames for map and legend/scalebar."""
334    
335          dc = self.dc          dc = self.dc
# Line 224  class ExportRenderer(ScreenRenderer): Line 338  class ExportRenderer(ScreenRenderer):
338    
339          # Dimension stuff          # Dimension stuff
340          width, height = dc.GetSizeTuple()          width, height = dc.GetSizeTuple()
341          mminx, mminy, mmaxx, mmaxy = self.mapregion          mminx, mminy, mmaxx, mmaxy = self.destination_region
342    
343          # Page Frame          # Page Frame
344          dc.DrawRectangle(15,15,width-30, (mmaxy-mminy)+10)          dc.DrawRectangle(15,15,width-30, (mmaxy-mminy)+10)
345    
346          # Map Frame          # Map Frame
347          llx, lly, urx, ury = region          llx, lly, urx, ury = self.region
348          dc.DrawRectangle(mminx + self.shiftx, mminy + self.shifty, urx, ury)          dc.DrawRectangle(mminx + self.shiftx, mminy + self.shifty, urx, ury)
349    
350          # Legend Frame          # Legend Frame
# Line 240  class ExportRenderer(ScreenRenderer): Line 354  class ExportRenderer(ScreenRenderer):
354          dc.SetClippingRegion(mmaxx+10,mminy,          dc.SetClippingRegion(mmaxx+10,mminy,
355                               (width-20) - (mmaxx+10), mmaxy-mminy)                               (width-20) - (mmaxx+10), mmaxy-mminy)
356    
357      def render_legend(self, map):      def render_legend(self):
358          """Render the legend on the Map."""          """Render the legend on the Map."""
359    
360          previewer = ClassDataPreviewer()          previewer = ClassDataPreviewer()
# Line 250  class ExportRenderer(ScreenRenderer): Line 364  class ExportRenderer(ScreenRenderer):
364    
365          # Dimension stuff          # Dimension stuff
366          width, height = dc.GetSizeTuple()          width, height = dc.GetSizeTuple()
367          mminx, mminy, mmaxx, mmaxy = self.mapregion          mminx, mminy, mmaxx, mmaxy = self.destination_region
368          textwidth, textheight = dc.GetTextExtent("0")          textwidth, textheight = dc.GetTextExtent("0")
369          iconwidth  = textheight          iconwidth  = textheight
370          iconheight = textheight          iconheight = textheight
# Line 262  class ExportRenderer(ScreenRenderer): Line 376  class ExportRenderer(ScreenRenderer):
376    
377          # Render the legend          # Render the legend
378          dc.SetTextForeground(wxBLACK)          dc.SetTextForeground(wxBLACK)
379          if map.HasLayers():          if self.map.HasLayers():
380              layers = map.Layers()              layers = self.map.Layers()[:]
381              layers.reverse()              layers.reverse()
382              for l in layers:              for l in layers:
383                  if l.Visible():                  if l.Visible():
# Line 284  class ExportRenderer(ScreenRenderer): Line 398  class ExportRenderer(ScreenRenderer):
398                                              posx+2*dx+iconwidth, posy)                                              posx+2*dx+iconwidth, posy)
399                                  posy+=stepy                                  posy+=stepy
400    
401      def render_scalebar(self, map):      def render_scalebar(self):
402          """Render the scalebar."""          """Render the scalebar."""
403    
404          scalebar = ScaleBar(map)          scalebar = ScaleBar(self.map)
405    
406          # Dimension stuff          # Dimension stuff
407          width, height = self.dc.GetSizeTuple()          width, height = self.dc.GetSizeTuple()
408          mminx, mminy, mmaxx, mmaxy = self.mapregion          mminx, mminy, mmaxx, mmaxy = self.destination_region
409    
410          # Render the scalebar          # Render the scalebar
411          scalebar.DrawScaleBar(self.scale, self.dc,          scalebar.DrawScaleBar(self.scale, self.dc,
Legend:
Removed from v.1552  
changed lines
  Added in v.2562
[email protected]
 ViewVC Help
Powered by ViewVC 1.1.26