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, \ |
117 |
|
|
118 |
def projected_raster_layer(self, layer, srcProj, dstProj, extents, |
def projected_raster_layer(self, layer, srcProj, dstProj, extents, |
119 |
resolution, dimensions, options): |
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 |
ret = None |
127 |
|
|
144 |
|
|
145 |
return ret |
return ret |
146 |
|
|
147 |
def draw_raster_data(self, x,y, data, format = 'BMP'): |
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] |
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]) |
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 |
|
|