/[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 1750 by bh, Wed Sep 24 15:46:27 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                x = lambda a, b, c, d: None
98                #return (True, x, None)
99              return (True, draw_polygon_shape,              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,
# 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.
166          if layer is self.selected_layer and self.selected_shapes:  
167              pen = wxPen(wxBLACK, 3, wxSOLID)          Only the given region (a tuple in window coordinates as returned
168              brush = wxBrush(wxBLACK, wxCROSS_HATCH)          by a wxrect's asTuple method) needs to be redrawn. Highlight the
169            shapes given by the ids in selected_shapes in the
170              shapetype = layer.ShapeType()          selected_layer.
171              useraw, func, param = self.low_level_renderer(layer)          """
172              args = (pen, brush)          return self.render_map_incrementally()
173              for index in self.selected_shapes:  
174                  shape = layer.Shape(index)      def draw_selection_incrementally(self, layer, selected_shapes):
175                  if useraw:          """Draw the selected shapes in a emphasized way (i.e.
176                      data = shape.RawData()          with a special pen and brush.
177                  else:          The drawing is performed incrementally, that means every
178                      data = shape.Points()          n shapes, the user can have interactions with the map.
179                  func(param, data, *args)          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):      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
# Line 143  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 164  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 199  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 216  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 230  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 246  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 256  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 268  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 290  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.1750  
changed lines
  Added in v.2562
[email protected]
 ViewVC Help
Powered by ViewVC 1.1.26