/[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 1562 by bh, Thu Aug 7 17:53:37 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 32  import Thuban.Model.resource Line 39  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 39  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 69  class MapRenderer(BaseRenderer): Line 94  class MapRenderer(BaseRenderer):
94          if (layer.ShapeStore().RawShapeFormat() == RAW_SHAPEFILE          if (layer.ShapeStore().RawShapeFormat() == RAW_SHAPEFILE
95              and layer.ShapeType() in (SHAPETYPE_ARC, SHAPETYPE_POLYGON)):              and layer.ShapeType() in (SHAPETYPE_ARC, SHAPETYPE_POLYGON)):
96              offx, offy = self.offset              offx, offy = self.offset
97              return (draw_polygon_shape,              x = lambda a, b, c, d: None
98                #return (True, x, None)
99                return (True, draw_polygon_shape,
100                      draw_polygon_init(layer.ShapeStore().Shapefile(),                      draw_polygon_init(layer.ShapeStore().Shapefile(),
101                                        self.dc, self.map.projection,                                        self.dc, self.map.projection,
102                                        layer.projection,                                        layer.projection,
# Line 78  class MapRenderer(BaseRenderer): Line 105  class MapRenderer(BaseRenderer):
105              return BaseRenderer.low_level_renderer(self, layer)              return BaseRenderer.low_level_renderer(self, layer)
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 92  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 100  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 draw_shape_layer(self, layer):      def RenderMapIncrementally(self):
165          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)  
166    
167      def layer_ids(self, layer):          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_selection_incrementally(self, layer, selected_shapes):
175            """Draw the selected shapes in a emphasized way (i.e.
176            with a special pen and brush.
177            The drawing is performed incrementally, that means every
178            n shapes, the user can have interactions with the map.
179            n is currently fixed to 500.
180    
181            layer -- the layer where the shapes belong to.
182            selected_shapes -- a list of the shape-ids representing the
183                               selected shapes for the given layer.
184            """
185            pen = wxPen(wxBLACK, 3, wxSOLID)
186            brush = wxBrush(wxBLACK, wxCROSS_HATCH)
187    
188            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 138  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 159  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 194  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 211  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 225  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 241  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 251  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 263  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 285  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.1562  
changed lines
  Added in v.2562
[email protected]
 ViewVC Help
Powered by ViewVC 1.1.26