/[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 1927 by bh, Mon Nov 10 16:57:35 2003 UTC revision 2712 by bernhard, Sun Oct 15 23:27:05 2006 UTC
# Line 1  Line 1 
1  # Copyright (c) 2001, 2002, 2003 by Intevation GmbH  # Copyright (c) 2001-2006 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-2005)
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.
# Line 15  __version__ = "$Revision$" Line 16  __version__ = "$Revision$"
16    
17  import cStringIO  import cStringIO
18    
19    import array
20    
21    import traceback
22    
23  from Thuban import _  from Thuban import _
24    
25  from wxPython.wx import wxPoint, wxRect, wxPen, wxBrush, wxFont, \  import wx
     wxTRANSPARENT_PEN, wxTRANSPARENT_BRUSH, \  
     wxBLACK_PEN, wxBLACK, wxSOLID, wxCROSS_HATCH, wxSWISS, wxNORMAL, \  
     wxBitmapFromImage, wxImageFromStream, wxBITMAP_TYPE_BMP, wxBITMAP_TYPE_JPEG  
26    
27  from wxproj import draw_polygon_shape, draw_polygon_init  from wxproj import draw_polygon_shape, draw_polygon_init
28    
# Line 36  import Thuban.Model.resource Line 38  import Thuban.Model.resource
38    
39  from baserenderer import BaseRenderer  from baserenderer import BaseRenderer
40    
41    from math import floor
42    
43    from types import StringType
44    
45    from Thuban.version import versions
46    
47    if Thuban.Model.resource.has_gdal_support():
48        from gdalwarp import ProjectRasterFile
49    
50    verbose = 0 # whether to talk more on stdout
51    
52  # Map the strings used for the format parameter of the draw_raster_data  # Map the strings used for the format parameter of the draw_raster_data
53  # method to the appropriate wxWindows constants  # method to the appropriate wxWindows constants
54  raster_format_map = {  raster_format_map = {
55      "BMP": wxBITMAP_TYPE_BMP,      "BMP": wx.BITMAP_TYPE_BMP,
56      "JPEG": wxBITMAP_TYPE_JPEG,      "JPEG": wx.BITMAP_TYPE_JPEG,
57        "PNG": wx.BITMAP_TYPE_PNG,
58        "TIFF": wx.BITMAP_TYPE_TIF,
59        "GIF": wx.BITMAP_TYPE_GIF,
60      }      }
61    
62  class MapRenderer(BaseRenderer):  class MapRenderer(BaseRenderer):
63    
64      """Class to render a map onto a wxDC"""      """Class to render a map onto a wxDC"""
65    
66      TRANSPARENT_PEN = wxTRANSPARENT_PEN      TRANSPARENT_PEN = wx.TRANSPARENT_PEN
67      TRANSPARENT_BRUSH = wxTRANSPARENT_BRUSH      TRANSPARENT_BRUSH = wx.TRANSPARENT_BRUSH
68    
69      make_point = wxPoint      def make_point(self, x, y):
70            return wx.Point(int(round(x)), int(round(y)))
71    
72      def tools_for_property(self, prop):      def tools_for_property(self, prop):
73          fill = prop.GetFill()          fill = prop.GetFill()
74          if fill is Transparent:          if fill is Transparent:
75              brush = self.TRANSPARENT_BRUSH              brush = self.TRANSPARENT_BRUSH
76          else:          else:
77              brush = wxBrush(Color2wxColour(fill), wxSOLID)              brush = wx.Brush(Color2wxColour(fill), wx.SOLID)
78    
79          stroke = prop.GetLineColor()          stroke = prop.GetLineColor()
80          if stroke is Transparent:          if stroke is Transparent:
81              pen = self.TRANSPARENT_PEN              pen = self.TRANSPARENT_PEN
82          else:          else:
83              pen = wxPen(Color2wxColour(stroke), prop.GetLineWidth(), wxSOLID)              pen = wx.Pen(Color2wxColour(stroke), prop.GetLineWidth(), wx.SOLID)
84          return pen, brush          return pen, brush
85    
86      def low_level_renderer(self, layer):      def low_level_renderer(self, layer):
# Line 81  class MapRenderer(BaseRenderer): Line 97  class MapRenderer(BaseRenderer):
97          if (layer.ShapeStore().RawShapeFormat() == RAW_SHAPEFILE          if (layer.ShapeStore().RawShapeFormat() == RAW_SHAPEFILE
98              and layer.ShapeType() in (SHAPETYPE_ARC, SHAPETYPE_POLYGON)):              and layer.ShapeType() in (SHAPETYPE_ARC, SHAPETYPE_POLYGON)):
99              offx, offy = self.offset              offx, offy = self.offset
100                x = lambda a, b, c, d: None
101                #return (True, x, None)
102              return (True, draw_polygon_shape,              return (True, draw_polygon_shape,
103                      draw_polygon_init(layer.ShapeStore().Shapefile(),                      draw_polygon_init(layer.ShapeStore().Shapefile(),
104                                        self.dc, self.map.projection,                                        self.dc, self.map.projection,
# Line 90  class MapRenderer(BaseRenderer): Line 108  class MapRenderer(BaseRenderer):
108              return BaseRenderer.low_level_renderer(self, layer)              return BaseRenderer.low_level_renderer(self, layer)
109    
110      def label_font(self):      def label_font(self):
111          return wxFont(self.resolution * 10, wxSWISS, wxNORMAL, wxNORMAL)          return wx.Font(int(round(self.resolution * 10)), wx.SWISS, wx.NORMAL,
112                          wx.NORMAL)
113    
114        def projected_raster_layer(self, layer, srcProj, dstProj, extents,
115                                   resolution, dimensions, options):
116            """Returns a raster layer image in projected space
117    
118            Based on a given filename.  This method must be implemented in
119            classes derived from BaseRenderer.
120            """
121    
122            ret = None
123    
124            if Thuban.Model.resource.has_gdal_support():
125    
126                if versions['wxPython-tuple'] < (2,5,3):
127                    options = options | 4  # INVERT_MASK_BITS
128                    options = options & ~2 # ALPHA_MASK not supported
129    
130                try:
131                    if verbose > 0:
132                        print "doing ProjectRasterFile '%s' -> '%s'" % \
133                                    (srcProj, dstProj)
134                        print "extents:", extents, "resolution:", resolution
135                        print "dimensions:", dimensions, "options:", options
136                    ret = ProjectRasterFile(layer.GetImageFilename(),
137                                            srcProj, dstProj,
138                                            extents, resolution, dimensions,
139                                            options)
140                except (MemoryError, IOError, AttributeError, ValueError):
141                    # Why does this catch AttributeError and ValueError?
142                    # FIXME: The exception should be communicated to the user
143                    # better.
144                    traceback.print_exc()
145    
146            return ret
147    
148        def draw_raster_data(self, x,y, data, format = 'BMP', opacity=1.0):
149    
150            mask = None
151            alpha = None
152            width = data[0]
153            height = data[1]
154            image_data, mask_data, alpha_data = data[2]
155    
156            if versions['wxPython-tuple'] < (2,5,3):
157                alpha_data = None
158    
159            if format == 'RAW':
160                image = wx.EmptyImage(width, height)
161                image.SetData(image_data)
162                if mask_data is not None:
163                    mask = wx.BitmapFromBits(mask_data, width, height, 1)
164                    mask = wx.Mask(mask)
165                elif alpha_data is not None:
166                    # alpha_data is already in the right format
167                    alpha = alpha_data
168    
169      def draw_raster_data(self, data, format = 'BMP'):          else:
170          stream = cStringIO.StringIO(data)              stream = cStringIO.StringIO(image_data)
171          image = wxImageFromStream(stream, raster_format_map[format])              image = wx.ImageFromStream(stream, raster_format_map[format])
172          bitmap = wxBitmapFromImage(image)  
173          self.dc.DrawBitmap(bitmap, 0, 0)              if mask_data is not None:
174                    stream = cStringIO.StringIO(mask_data)
175                    mask = wx.ImageFromStream(stream, raster_format_map[format])
176                    mask = wx.Mask(wx.BitmapFromImage(mask, 1))
177                elif alpha_data is not None:
178                    stream = cStringIO.StringIO(alpha_data)
179                    alpha = wx.ImageFromStream(stream, raster_format_map[format])
180                    alpha = alpha.GetData() #[:] # XXX: do we need to copy this?
181                elif image.HasAlpha():
182                    alpha = image.GetAlphaData()
183    
184            #
185            # scale down the alpha values the opacity level using a string
186            # translation table for efficiency.
187            #
188            if alpha is not None:
189                if opacity == 0:
190                    return
191                elif opacity == 1:
192                    a = alpha
193                else:
194                    tr = [int(i*opacity) for i in range(256)]
195                    table = array.array('B', tr).tostring()
196                    a = alpha.translate(table)
197    
198                image.SetAlphaData(a)
199    
200            bitmap = wx.BitmapFromImage(image)
201    
202            if mask is not None:
203                bitmap.SetMask(mask)
204    
205            self.dc.DrawBitmap(bitmap, int(round(x)), int(round(y)), True)
206    
207    
208  class ScreenRenderer(MapRenderer):  class ScreenRenderer(MapRenderer):
# Line 127  class ScreenRenderer(MapRenderer): Line 233  class ScreenRenderer(MapRenderer):
233          return self.render_map_incrementally()          return self.render_map_incrementally()
234    
235      def draw_selection_incrementally(self, layer, selected_shapes):      def draw_selection_incrementally(self, layer, selected_shapes):
236          pen = wxPen(wxBLACK, 3, wxSOLID)          """Draw the selected shapes in a emphasized way (i.e.
237          brush = wxBrush(wxBLACK, wxCROSS_HATCH)          with a special pen and brush.
238            The drawing is performed incrementally, that means every
239            n shapes, the user can have interactions with the map.
240            n is currently fixed to 500.
241    
242            layer -- the layer where the shapes belong to.
243            selected_shapes -- a list of the shape-ids representing the
244                               selected shapes for the given layer.
245            """
246            pen = wx.Pen(wx.BLACK, 3, wx.SOLID)
247            brush = wx.Brush(wx.BLACK, wx.CROSS_HATCH)
248    
249          shapetype = layer.ShapeType()          shapetype = layer.ShapeType()
250          useraw, func, param = self.low_level_renderer(layer)          useraw, func, param = self.low_level_renderer(layer)
251          args = (pen, brush)          args = (pen, brush)
252    
253            # for point shapes we need to find out the properties
254            # to determine the size. Based on table and field,
255            # we can find out the properties for object - see below.
256            if shapetype == SHAPETYPE_POINT:
257                lc = layer.GetClassification()
258                field = layer.GetClassificationColumn()
259                table = layer.ShapeStore().Table()
260    
261          count = 0          count = 0
262          for index in selected_shapes:          for index in selected_shapes:
263              count += 1              count += 1
264              shape = layer.Shape(index)              shape = layer.Shape(index)
265    
266                # Get the size of the specific property for this
267                # point
268                if shapetype == SHAPETYPE_POINT:
269                    if field is not None:
270                        value = table.ReadValue(shape.ShapeID(), field)
271                        group = lc.FindGroup(value)
272                        size = group.GetProperties().GetSize()
273                    else:
274                        size = lc.GetDefaultGroup().GetProperties().GetSize()
275                    args = (pen, brush, size)
276    
277              if useraw:              if useraw:
278                  data = shape.RawData()                  data = shape.RawData()
279              else:              else:
# Line 233  class ExportRenderer(ScreenRenderer): Line 370  class ExportRenderer(ScreenRenderer):
370          self.shifty = (mmaxy - mminy)*0.5 - (ury - lly)*0.5          self.shifty = (mmaxy - mminy)*0.5 - (ury - lly)*0.5
371    
372          self.offset = (offx+self.shiftx, offy+self.shifty)          self.offset = (offx+self.shiftx, offy+self.shifty)
373            self.region = (llx + self.shiftx, lly + self.shifty, urx, ury)
374    
375          # Draw the map          # Draw the map
376          self.dc.BeginDrawing()          self.dc.BeginDrawing()
# Line 247  class ExportRenderer(ScreenRenderer): Line 385  class ExportRenderer(ScreenRenderer):
385          self.dc.DestroyClippingRegion()          self.dc.DestroyClippingRegion()
386    
387          # Force the font for Legend drawing          # Force the font for Legend drawing
388          font = wxFont(self.resolution * 10, wxSWISS, wxNORMAL, wxNORMAL)          font = wx.Font(self.resolution * 10, wx.SWISS, wx.NORMAL, wx.NORMAL)
389          self.dc.SetFont(font)          self.dc.SetFont(font)
390    
391          self.render_frame()          self.render_frame()
# Line 259  class ExportRenderer(ScreenRenderer): Line 397  class ExportRenderer(ScreenRenderer):
397          """Render the frames for map and legend/scalebar."""          """Render the frames for map and legend/scalebar."""
398    
399          dc = self.dc          dc = self.dc
400          dc.SetPen(wxBLACK_PEN)          dc.SetPen(wx.BLACK_PEN)
401          dc.SetBrush(wxTRANSPARENT_BRUSH)          dc.SetBrush(wx.TRANSPARENT_BRUSH)
402    
403          # Dimension stuff          # Dimension stuff
404          width, height = dc.GetSizeTuple()          width, height = dc.GetSizeTuple()
# Line 285  class ExportRenderer(ScreenRenderer): Line 423  class ExportRenderer(ScreenRenderer):
423    
424          previewer = ClassDataPreviewer()          previewer = ClassDataPreviewer()
425          dc = self.dc          dc = self.dc
426          dc.SetPen(wxBLACK_PEN)          dc.SetPen(wx.BLACK_PEN)
427          dc.SetBrush(wxTRANSPARENT_BRUSH)          dc.SetBrush(wx.TRANSPARENT_BRUSH)
428    
429          # Dimension stuff          # Dimension stuff
430          width, height = dc.GetSizeTuple()          width, height = dc.GetSizeTuple()
# Line 301  class ExportRenderer(ScreenRenderer): Line 439  class ExportRenderer(ScreenRenderer):
439          posy = mminy + 5        # 5 pix inside legend frame          posy = mminy + 5        # 5 pix inside legend frame
440    
441          # Render the legend          # Render the legend
442          dc.SetTextForeground(wxBLACK)          dc.SetTextForeground(wx.BLACK)
443          if self.map.HasLayers():          if self.map.HasLayers():
444              layers = self.map.Layers()[:]              layers = self.map.Layers()[:]
445              layers.reverse()              layers.reverse()
# Line 317  class ExportRenderer(ScreenRenderer): Line 455  class ExportRenderer(ScreenRenderer):
455                          for g in clazz:                          for g in clazz:
456                              if g.IsVisible():                              if g.IsVisible():
457                                  previewer.Draw(dc,                                  previewer.Draw(dc,
458                                      wxRect(posx+dx, posy,                                      wx.Rect(posx+dx, posy,
459                                             iconwidth, iconheight),                                             iconwidth, iconheight),
460                                      g.GetProperties(), shapeType)                                      g.GetProperties(), shapeType)
461                                  dc.DrawText(g.GetDisplayText(),                                  dc.DrawText(g.GetDisplayText(),
Legend:
Removed from v.1927  
changed lines
  Added in v.2712
[email protected]
 ViewVC Help
Powered by ViewVC 1.1.26