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, \ |
139 |
|
|
140 |
return ret |
return ret |
141 |
|
|
142 |
def draw_raster_data(self, x,y, data, format = 'BMP'): |
def draw_raster_data(self, layer, x,y, data, format = 'BMP'): |
143 |
|
|
144 |
mask = None |
mask = None |
145 |
|
alpha = None |
146 |
width = data[0] |
width = data[0] |
147 |
height = data[1] |
height = data[1] |
148 |
image_data, mask_data, alpha_data = data[2] |
image_data, mask_data, alpha_data = data[2] |
157 |
mask = wxBitmapFromBits(mask_data, width, height, 1) |
mask = wxBitmapFromBits(mask_data, width, height, 1) |
158 |
mask = wxMask(mask) |
mask = wxMask(mask) |
159 |
elif alpha_data is not None: |
elif alpha_data is not None: |
160 |
image.SetAlphaData(alpha_data) |
# alpha_data is already in the right format |
161 |
|
alpha = alpha_data |
162 |
|
|
163 |
else: |
else: |
164 |
stream = cStringIO.StringIO(image_data) |
stream = cStringIO.StringIO(image_data) |
165 |
image = wxImageFromStream(stream, raster_format_map[format]) |
image = wxImageFromStream(stream, raster_format_map[format]) |
170 |
elif alpha_data is not None: |
elif alpha_data is not None: |
171 |
stream = cStringIO.StringIO(alpha_data) |
stream = cStringIO.StringIO(alpha_data) |
172 |
alpha = wxImageFromStream(stream, raster_format_map[format]) |
alpha = wxImageFromStream(stream, raster_format_map[format]) |
173 |
image.SetAlpha(alpha.GetData()[:]) |
alpha = alpha.GetData()[:] # XXX: do we need to copy this? |
174 |
|
|
175 |
|
# |
176 |
|
# if we are using the alpha_data then scale down the alpha values |
177 |
|
# by the layer's opacity using a string translation table |
178 |
|
# |
179 |
|
if alpha is not None: |
180 |
|
lo = layer.Opacity() |
181 |
|
if lo == 0: |
182 |
|
return |
183 |
|
elif lo == 1: |
184 |
|
a = alpha |
185 |
|
else: |
186 |
|
tr = [int(i*lo) for i in range(256)] |
187 |
|
table = array.array('B', tr).tostring() |
188 |
|
a = alpha.translate(table) |
189 |
|
|
190 |
|
image.SetAlphaData(a) |
191 |
|
|
192 |
bitmap = wxBitmapFromImage(image) |
bitmap = wxBitmapFromImage(image) |
193 |
bitmap.SetMask(mask) |
|
194 |
|
if mask is not None: |
195 |
|
bitmap.SetMask(mask) |
196 |
|
|
197 |
self.dc.DrawBitmap(bitmap, int(round(x)), int(round(y)), True) |
self.dc.DrawBitmap(bitmap, int(round(x)), int(round(y)), True) |
198 |
|
|