/[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 1591 by bh, Fri Aug 15 14:00:53 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 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    
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 39  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 78  class MapRenderer(BaseRenderer): Line 101  class MapRenderer(BaseRenderer):
101              return BaseRenderer.low_level_renderer(self, layer)              return BaseRenderer.low_level_renderer(self, layer)
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 92  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 100  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 draw_shape_layer(self, layer):      def RenderMapIncrementally(self):
149          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()  
             useraw, func, param = self.low_level_renderer(layer)  
             args = (pen, brush)  
             for index in self.selected_shapes:  
                 shape = layer.Shape(index)  
                 if useraw:  
                     data = shape.RawData()  
                 else:  
                     data = shape.Points()  
                 func(param, data, *args)  
150    
151      def layer_ids(self, layer):          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_selection_incrementally(self, layer, selected_shapes):
159            """Draw the selected shapes in a emphasized way (i.e.
160            with a special pen and brush.
161            The drawing is performed incrementally, that means every
162            n shapes, the user can have interactions with the map.
163            n is currently fixed to 500.
164    
165            layer -- the layer where the shapes belong to.
166            selected_shapes -- a list of the shape-ids representing the
167                               selected shapes for the given layer.
168            """
169            pen = wxPen(wxBLACK, 3, wxSOLID)
170            brush = wxBrush(wxBLACK, wxCROSS_HATCH)
171    
172            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 143  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 164  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 199  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 216  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 230  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 246  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 256  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 268  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 290  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.1591  
changed lines
  Added in v.2552
[email protected]
 ViewVC Help
Powered by ViewVC 1.1.26