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 |
|
|
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 |
|
|
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""" |
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 |
|
|
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 |
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 RenderMapIncrementally(self): |
120 |
|
"""Render the map. |
121 |
|
|
122 |
|
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_shape_layer(self, layer): |
def draw_selection_incrementally(self, layer, selected_shapes): |
130 |
MapRenderer.draw_shape_layer(self, layer) |
pen = wxPen(wxBLACK, 3, wxSOLID) |
131 |
if layer is self.selected_layer and self.selected_shapes: |
brush = wxBrush(wxBLACK, wxCROSS_HATCH) |
132 |
pen = wxPen(wxBLACK, 3, wxSOLID) |
|
133 |
brush = wxBrush(wxBLACK, wxCROSS_HATCH) |
shapetype = layer.ShapeType() |
134 |
|
useraw, func, param = self.low_level_renderer(layer) |
135 |
shapetype = layer.ShapeType() |
args = (pen, brush) |
136 |
useraw, func, param = self.low_level_renderer(layer) |
count = 0 |
137 |
args = (pen, brush) |
for index in selected_shapes: |
138 |
for index in self.selected_shapes: |
count += 1 |
139 |
shape = layer.Shape(index) |
shape = layer.Shape(index) |
140 |
if useraw: |
if useraw: |
141 |
data = shape.RawData() |
data = shape.RawData() |
142 |
else: |
else: |
143 |
data = shape.Points() |
data = shape.Points() |
144 |
func(param, data, *args) |
func(param, data, *args) |
145 |
|
if count % 500 == 0: |
146 |
|
yield True |
147 |
|
|
148 |
def layer_ids(self, layer): |
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 |
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 |
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 |
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) |
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 |
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 |
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() |
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 |
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(): |
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, |