/[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 2562 by jonathan, Wed Feb 16 21:14:47 2005 UTC revision 2602 by joey, Mon Apr 25 15:01:48 2005 UTC
# Line 16  __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, \  from wxPython.wx import wxPoint, wxRect, wxPen, wxBrush, wxFont, \
# Line 45  from types import StringType Line 49  from types import StringType
49    
50  from Thuban.version import versions  from Thuban.version import versions
51    
52    if Thuban.Model.resource.has_gdal_support():
53        from gdalwarp import ProjectRasterFile
54    
55    
56  # 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
57  # method to the appropriate wxWindows constants  # method to the appropriate wxWindows constants
# Line 108  class MapRenderer(BaseRenderer): Line 115  class MapRenderer(BaseRenderer):
115          return wxFont(int(round(self.resolution * 10)), wxSWISS, wxNORMAL,          return wxFont(int(round(self.resolution * 10)), wxSWISS, wxNORMAL,
116                        wxNORMAL)                        wxNORMAL)
117    
118      def draw_raster_data(self, x,y, data, format = 'BMP'):      def projected_raster_layer(self, layer, srcProj, dstProj, extents,
119                                   resolution, dimensions, options):
120            """Returns a raster layer image in projected space
121    
122            Based on a given filename.  This method must be implemented in
123            classes derived from BaseRenderer.
124            """
125    
126            ret = None
127    
128            if Thuban.Model.resource.has_gdal_support():
129    
130                if versions['wxPython-tuple'] < (2,5,3):
131                    options = options | 4  # INVERT_MASK_BITS
132                    options = options & ~2 # ALPHA_MASK not supported
133    
134                try:
135                    ret = ProjectRasterFile(layer.GetImageFilename(),
136                                            srcProj, dstProj,
137                                            extents, resolution, dimensions,
138                                            options)
139                except (MemoryError, IOError, AttributeError, ValueError):
140                    # Why does this catch AttributeError and ValueError?
141                    # FIXME: The exception should be communicated to the user
142                    # better.
143                    traceback.print_exc()
144    
145            return ret
146    
147        def draw_raster_data(self, layer, x,y, data, format = 'BMP'):
148    
149          mask = None          mask = None
150            alpha = None
151          width = data[0]          width = data[0]
152          height = data[1]          height = data[1]
153          image_data, mask_data, alpha_data = data[2]          image_data, mask_data, alpha_data = data[2]
# Line 125  class MapRenderer(BaseRenderer): Line 162  class MapRenderer(BaseRenderer):
162                  mask = wxBitmapFromBits(mask_data, width, height, 1)                  mask = wxBitmapFromBits(mask_data, width, height, 1)
163                  mask = wxMask(mask)                  mask = wxMask(mask)
164              elif alpha_data is not None:              elif alpha_data is not None:
165                  image.SetAlphaData(alpha_data)                  # alpha_data is already in the right format
166                    alpha = alpha_data
167    
168          else:          else:
169              stream = cStringIO.StringIO(image_data)              stream = cStringIO.StringIO(image_data)
170              image = wxImageFromStream(stream, raster_format_map[format])              image = wxImageFromStream(stream, raster_format_map[format])
# Line 136  class MapRenderer(BaseRenderer): Line 175  class MapRenderer(BaseRenderer):
175              elif alpha_data is not None:              elif alpha_data is not None:
176                  stream = cStringIO.StringIO(alpha_data)                  stream = cStringIO.StringIO(alpha_data)
177                  alpha = wxImageFromStream(stream, raster_format_map[format])                  alpha = wxImageFromStream(stream, raster_format_map[format])
178                  image.SetAlpha(alpha.GetData()[:])                  alpha = alpha.GetData()[:] # XXX: do we need to copy this?
179    
180            #
181            # if we are using the alpha_data then scale down the alpha values
182            # by the layer's opacity using a string translation table
183            #
184            if alpha is not None:
185                lo = layer.Opacity()
186                if lo == 0:
187                    return
188                elif lo == 1:
189                    a = alpha
190                else:
191                    tr = [int(i*lo) for i in range(256)]
192                    table = array.array('B', tr).tostring()
193                    a = alpha.translate(table)
194    
195                image.SetAlphaData(a)
196    
197          bitmap = wxBitmapFromImage(image)          bitmap = wxBitmapFromImage(image)
198          bitmap.SetMask(mask)  
199            if mask is not None:
200                bitmap.SetMask(mask)
201    
202          self.dc.DrawBitmap(bitmap, int(round(x)), int(round(y)), True)          self.dc.DrawBitmap(bitmap, int(round(x)), int(round(y)), True)
203    
Legend:
Removed from v.2562  
changed lines
  Added in v.2602
[email protected]
 ViewVC Help
Powered by ViewVC 1.1.26