/[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 1927 by bh, Mon Nov 10 16:57:35 2003 UTC
# Line 7  Line 7 
7  # This program is free software under the GPL (>=v2)  # This program is free software under the GPL (>=v2)
8  # Read the file COPYING coming with Thuban for details.  # Read the file COPYING coming with Thuban for details.
9    
10    from __future__ import generators
11    
12  __version__ = "$Revision$"  __version__ = "$Revision$"
13    # $Source$
14    # $Id$
15    
16  import cStringIO  import cStringIO
17    
# Line 16  from Thuban import _ Line 20  from Thuban import _
20  from wxPython.wx import wxPoint, wxRect, wxPen, wxBrush, wxFont, \  from wxPython.wx import wxPoint, wxRect, wxPen, wxBrush, wxFont, \
21      wxTRANSPARENT_PEN, wxTRANSPARENT_BRUSH, \      wxTRANSPARENT_PEN, wxTRANSPARENT_BRUSH, \
22      wxBLACK_PEN, wxBLACK, wxSOLID, wxCROSS_HATCH, wxSWISS, wxNORMAL, \      wxBLACK_PEN, wxBLACK, wxSOLID, wxCROSS_HATCH, wxSWISS, wxNORMAL, \
23      wxBitmapFromImage, wxImageFromStream, wxBITMAP_TYPE_BMP      wxBitmapFromImage, wxImageFromStream, wxBITMAP_TYPE_BMP, wxBITMAP_TYPE_JPEG
24    
25  from wxproj import draw_polygon_shape, draw_polygon_init  from wxproj import draw_polygon_shape, draw_polygon_init
26    
# Line 24  from Thuban.UI.common import Color2wxCol Line 28  from Thuban.UI.common import Color2wxCol
28  from Thuban.UI.classifier import ClassDataPreviewer  from Thuban.UI.classifier import ClassDataPreviewer
29  from Thuban.UI.scalebar import ScaleBar  from Thuban.UI.scalebar import ScaleBar
30    
31  from Thuban.Model.data import SHAPETYPE_POLYGON, SHAPETYPE_ARC, SHAPETYPE_POINT  from Thuban.Model.data import SHAPETYPE_POLYGON, SHAPETYPE_ARC, \
32         SHAPETYPE_POINT, RAW_SHAPEFILE
33    
34  from Thuban.Model.color import Transparent  from Thuban.Model.color import Transparent
35  import Thuban.Model.resource  import Thuban.Model.resource
36    
37  from baserenderer import BaseRenderer  from baserenderer import BaseRenderer
38    
39    
40    # Map the strings used for the format parameter of the draw_raster_data
41    # method to the appropriate wxWindows constants
42    raster_format_map = {
43        "BMP": wxBITMAP_TYPE_BMP,
44        "JPEG": wxBITMAP_TYPE_JPEG,
45        }
46    
47  class MapRenderer(BaseRenderer):  class MapRenderer(BaseRenderer):
48    
49      """Class to render a map onto a wxDC"""      """Class to render a map onto a wxDC"""
# Line 57  class MapRenderer(BaseRenderer): Line 70  class MapRenderer(BaseRenderer):
70      def low_level_renderer(self, layer):      def low_level_renderer(self, layer):
71          """Override inherited method to provide more efficient renderers          """Override inherited method to provide more efficient renderers
72    
73          For point shapes, return self.draw_point_shape and layer just as          If the underlying data format is not a shapefile or the layer
74          the base class method. For arc and polygon use the more          contains points shapes, simply use what the inherited method
75          efficient wxproj.draw_polygon_shape and its corresponding          returns.
76          parameter created with wxproj.draw_polygon_init.  
77            Otherwise, i.e. for arc and polygon use the more efficient
78            wxproj.draw_polygon_shape and its corresponding parameter
79            created with wxproj.draw_polygon_init.
80          """          """
81          shapetype = layer.ShapeType()          if (layer.ShapeStore().RawShapeFormat() == RAW_SHAPEFILE
82          if shapetype == SHAPETYPE_POINT:              and layer.ShapeType() in (SHAPETYPE_ARC, SHAPETYPE_POLYGON)):
             func = self.draw_point_shape  
             param = layer  
         else:  
83              offx, offy = self.offset              offx, offy = self.offset
84              param = draw_polygon_init(layer.ShapeStore().Shapefile(), self.dc,              return (True, draw_polygon_shape,
85                                        self.map.projection,                      draw_polygon_init(layer.ShapeStore().Shapefile(),
86                                          self.dc, self.map.projection,
87                                        layer.projection,                                        layer.projection,
88                                        self.scale, -self.scale,                                        self.scale, -self.scale, offx, offy))
89                                        offx, offy)          else:
90              func = draw_polygon_shape              return BaseRenderer.low_level_renderer(self, layer)
         return func, param  
91    
92      def label_font(self):      def label_font(self):
93          return wxFont(self.resolution * 10, wxSWISS, wxNORMAL, wxNORMAL)          return wxFont(self.resolution * 10, wxSWISS, wxNORMAL, wxNORMAL)
94    
95      def draw_raster_data(self, data):      def draw_raster_data(self, data, format = 'BMP'):
96          stream = cStringIO.StringIO(data)          stream = cStringIO.StringIO(data)
97          image = wxImageFromStream(stream, wxBITMAP_TYPE_BMP)          image = wxImageFromStream(stream, raster_format_map[format])
98          bitmap = wxBitmapFromImage(image)          bitmap = wxBitmapFromImage(image)
99          self.dc.DrawBitmap(bitmap, 0, 0)          self.dc.DrawBitmap(bitmap, 0, 0)
100    
# Line 91  class ScreenRenderer(MapRenderer): Line 104  class ScreenRenderer(MapRenderer):
104      # On the screen we want to see only visible layers by default      # On the screen we want to see only visible layers by default
105      honor_visibility = 1      honor_visibility = 1
106    
107      def RenderMap(self, map, region, selected_layer, selected_shapes):      def RenderMap(self, selected_layer, selected_shapes):
108          """Render the map.          """Render the map.
109    
110          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 112  class ScreenRenderer(MapRenderer):
112          shapes given by the ids in selected_shapes in the          shapes given by the ids in selected_shapes in the
113          selected_layer.          selected_layer.
114          """          """
         self.update_region = region  
115          self.selected_layer = selected_layer          self.selected_layer = selected_layer
116          self.selected_shapes = selected_shapes          self.selected_shapes = selected_shapes
117          self.render_map(map)          self.render_map()
118    
119      def draw_shape_layer(self, layer):      def RenderMapIncrementally(self):
120          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)  
121    
122      def layer_ids(self, layer):          Only the given region (a tuple in window coordinates as returned
123            by a wxrect's asTuple method) needs to be redrawn. Highlight the
124            shapes given by the ids in selected_shapes in the
125            selected_layer.
126            """
127            return self.render_map_incrementally()
128    
129        def draw_selection_incrementally(self, layer, selected_shapes):
130            pen = wxPen(wxBLACK, 3, wxSOLID)
131            brush = wxBrush(wxBLACK, wxCROSS_HATCH)
132    
133            shapetype = layer.ShapeType()
134            useraw, func, param = self.low_level_renderer(layer)
135            args = (pen, brush)
136            count = 0
137            for index in selected_shapes:
138                count += 1
139                shape = layer.Shape(index)
140                if useraw:
141                    data = shape.RawData()
142                else:
143                    data = shape.Points()
144                func(param, data, *args)
145                if count % 500 == 0:
146                    yield True
147    
148        def layer_shapes(self, layer):
149          """Return the shapeids covered by the region that has to be redrawn          """Return the shapeids covered by the region that has to be redrawn
150    
151          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 166  class ScreenRenderer(MapRenderer):
166          offx, offy = self.offset          offx, offy = self.offset
167          xs = []          xs = []
168          ys = []          ys = []
169          x, y, width, height = self.update_region          x, y, width, height = self.region
170          for winx, winy in ((x, y), (x + width, y),          for winx, winy in ((x, y), (x + width, y),
171                             (x + width, y + height), (x, y + height)):                             (x + width, y + height), (x, y + height)):
172              px = (winx - offx) / scale              px = (winx - offx) / scale
# Line 158  class ExportRenderer(ScreenRenderer): Line 187  class ExportRenderer(ScreenRenderer):
187    
188      honor_visibility = 1      honor_visibility = 1
189    
190      def RenderMap(self, map, region, mapregion,      def __init__(self, *args, **kw):
191                    selected_layer, selected_shapes ):          """Initialize the ExportRenderer.
192    
193            In addition to all parameters of the the ScreenRender
194            constructor, this class requires and additional keyword argument
195            destination_region with a tuple (minx, miny, maxx, maxy) giving
196            the region in dc coordinates which is to contain the map.
197            """
198            self.destination_region = kw["destination_region"]
199            del kw["destination_region"]
200            ScreenRenderer.__init__(self, *args, **kw)
201    
202        def RenderMap(self, selected_layer, selected_shapes):
203          """Render the map.          """Render the map.
204    
205          The rendering device has been specified during initialisation.          The rendering device has been specified during initialisation.
206          The device border distance was set in Thuban.UI.view.OutputTranform().          The device border distance was set in
207            Thuban.UI.viewport.output_transform().
208    
209          RenderMap renders a frame set (one page frame, one around          RenderMap renders a frame set (one page frame, one around
210          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
211          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
212          in the canvas view, centered on the area available for map display.          region displayed in the canvas view, centered on the area
213            available for map display.
214          """          """
215    
         self.update_region = region  
216          self.selected_layer = selected_layer          self.selected_layer = selected_layer
217          self.selected_shapes = selected_shapes          self.selected_shapes = selected_shapes
218    
219          # Get some dimensions          # Get some dimensions
220          llx, lly, urx, ury = region          llx, lly, urx, ury = self.region
221          self.mapregion = mapregion          mminx, mminy, mmaxx, mmaxy = self.destination_region
         mminx, mminy, mmaxx, mmaxy = self.mapregion  
222    
223          # Manipulate the offset to position the map          # Manipulate the offset to position the map
224          offx, offy = self.offset          offx, offy = self.offset
# Line 199  class ExportRenderer(ScreenRenderer): Line 239  class ExportRenderer(ScreenRenderer):
239          self.dc.DestroyClippingRegion()          self.dc.DestroyClippingRegion()
240          self.dc.SetClippingRegion(mminx+self.shiftx, mminy+self.shifty,          self.dc.SetClippingRegion(mminx+self.shiftx, mminy+self.shifty,
241                                    urx, ury)                                    urx, ury)
242          self.render_map(map)          self.render_map()
243          self.dc.EndDrawing()          self.dc.EndDrawing()
244    
245          # Draw the rest (frames, legend, scalebar)          # Draw the rest (frames, legend, scalebar)
# Line 210  class ExportRenderer(ScreenRenderer): Line 250  class ExportRenderer(ScreenRenderer):
250          font = wxFont(self.resolution * 10, wxSWISS, wxNORMAL, wxNORMAL)          font = wxFont(self.resolution * 10, wxSWISS, wxNORMAL, wxNORMAL)
251          self.dc.SetFont(font)          self.dc.SetFont(font)
252    
253          self.render_frame(region)          self.render_frame()
254          self.render_legend(map)          self.render_legend()
255          self.render_scalebar(map)          self.render_scalebar()
256          self.dc.EndDrawing()          self.dc.EndDrawing()
257    
258      def render_frame(self, region):      def render_frame(self):
259          """Render the frames for map and legend/scalebar."""          """Render the frames for map and legend/scalebar."""
260    
261          dc = self.dc          dc = self.dc
# Line 224  class ExportRenderer(ScreenRenderer): Line 264  class ExportRenderer(ScreenRenderer):
264    
265          # Dimension stuff          # Dimension stuff
266          width, height = dc.GetSizeTuple()          width, height = dc.GetSizeTuple()
267          mminx, mminy, mmaxx, mmaxy = self.mapregion          mminx, mminy, mmaxx, mmaxy = self.destination_region
268    
269          # Page Frame          # Page Frame
270          dc.DrawRectangle(15,15,width-30, (mmaxy-mminy)+10)          dc.DrawRectangle(15,15,width-30, (mmaxy-mminy)+10)
271    
272          # Map Frame          # Map Frame
273          llx, lly, urx, ury = region          llx, lly, urx, ury = self.region
274          dc.DrawRectangle(mminx + self.shiftx, mminy + self.shifty, urx, ury)          dc.DrawRectangle(mminx + self.shiftx, mminy + self.shifty, urx, ury)
275    
276          # Legend Frame          # Legend Frame
# Line 240  class ExportRenderer(ScreenRenderer): Line 280  class ExportRenderer(ScreenRenderer):
280          dc.SetClippingRegion(mmaxx+10,mminy,          dc.SetClippingRegion(mmaxx+10,mminy,
281                               (width-20) - (mmaxx+10), mmaxy-mminy)                               (width-20) - (mmaxx+10), mmaxy-mminy)
282    
283      def render_legend(self, map):      def render_legend(self):
284          """Render the legend on the Map."""          """Render the legend on the Map."""
285    
286          previewer = ClassDataPreviewer()          previewer = ClassDataPreviewer()
# Line 250  class ExportRenderer(ScreenRenderer): Line 290  class ExportRenderer(ScreenRenderer):
290    
291          # Dimension stuff          # Dimension stuff
292          width, height = dc.GetSizeTuple()          width, height = dc.GetSizeTuple()
293          mminx, mminy, mmaxx, mmaxy = self.mapregion          mminx, mminy, mmaxx, mmaxy = self.destination_region
294          textwidth, textheight = dc.GetTextExtent("0")          textwidth, textheight = dc.GetTextExtent("0")
295          iconwidth  = textheight          iconwidth  = textheight
296          iconheight = textheight          iconheight = textheight
# Line 262  class ExportRenderer(ScreenRenderer): Line 302  class ExportRenderer(ScreenRenderer):
302    
303          # Render the legend          # Render the legend
304          dc.SetTextForeground(wxBLACK)          dc.SetTextForeground(wxBLACK)
305          if map.HasLayers():          if self.map.HasLayers():
306              layers = map.Layers()              layers = self.map.Layers()[:]
307              layers.reverse()              layers.reverse()
308              for l in layers:              for l in layers:
309                  if l.Visible():                  if l.Visible():
# Line 284  class ExportRenderer(ScreenRenderer): Line 324  class ExportRenderer(ScreenRenderer):
324                                              posx+2*dx+iconwidth, posy)                                              posx+2*dx+iconwidth, posy)
325                                  posy+=stepy                                  posy+=stepy
326    
327      def render_scalebar(self, map):      def render_scalebar(self):
328          """Render the scalebar."""          """Render the scalebar."""
329    
330          scalebar = ScaleBar(map)          scalebar = ScaleBar(self.map)
331    
332          # Dimension stuff          # Dimension stuff
333          width, height = self.dc.GetSizeTuple()          width, height = self.dc.GetSizeTuple()
334          mminx, mminy, mmaxx, mmaxy = self.mapregion          mminx, mminy, mmaxx, mmaxy = self.destination_region
335    
336          # Render the scalebar          # Render the scalebar
337          scalebar.DrawScaleBar(self.scale, self.dc,          scalebar.DrawScaleBar(self.scale, self.dc,
Legend:
Removed from v.1552  
changed lines
  Added in v.1927
[email protected]
 ViewVC Help
Powered by ViewVC 1.1.26