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