8 |
|
|
9 |
__version__ = "$Revision$" |
__version__ = "$Revision$" |
10 |
|
|
11 |
from wxPython.wx import wxPoint, wxColour, wxPen, wxBrush, wxFont, \ |
from wxPython.wx import wxPoint, wxPen, wxBrush, wxFont, \ |
12 |
wxTRANSPARENT_PEN, wxTRANSPARENT_BRUSH, \ |
wxTRANSPARENT_PEN, wxTRANSPARENT_BRUSH, \ |
13 |
wxBLACK, wxSOLID, wxCROSS_HATCH, wxSWISS, wxNORMAL |
wxBLACK, wxSOLID, wxCROSS_HATCH, wxSWISS, wxNORMAL |
14 |
|
|
15 |
from wxproj import draw_polygon_shape |
from wxproj import draw_polygon_shape, draw_polygon_init |
16 |
|
|
17 |
from Thuban import _ |
from Thuban import _ |
18 |
|
from Thuban.UI.common import * |
19 |
|
|
20 |
from Thuban.Model.layer import SHAPETYPE_POLYGON, SHAPETYPE_ARC, \ |
from Thuban.Model.layer import SHAPETYPE_POLYGON, SHAPETYPE_ARC, \ |
21 |
SHAPETYPE_POINT |
SHAPETYPE_POINT |
48 |
the renderer's default. |
the renderer's default. |
49 |
""" |
""" |
50 |
# resolution in pixel/inch |
# resolution in pixel/inch |
51 |
|
|
52 |
|
assert scale > 0 |
53 |
|
|
54 |
self.dc = dc |
self.dc = dc |
55 |
self.scale = scale |
self.scale = scale |
56 |
self.offset = offset |
self.offset = offset |
82 |
pen = wxTRANSPARENT_PEN |
pen = wxTRANSPARENT_PEN |
83 |
|
|
84 |
old_prop = None |
old_prop = None |
85 |
|
lc = layer.GetClassification() |
86 |
|
field = lc.GetField() |
87 |
|
|
88 |
|
|
89 |
|
if shapetype != SHAPETYPE_POINT: |
90 |
|
polygon_render_param = self.polygon_render_param(layer) |
91 |
|
|
92 |
for i in self.layer_ids(layer): |
for i in self.layer_ids(layer): |
93 |
value = None |
value = None |
|
lc = layer.GetClassification() |
|
|
field = lc.field |
|
94 |
|
|
95 |
if field is not None: |
if field is not None: |
96 |
record = layer.table.read_record(i) |
try: |
97 |
if record is not None: |
record = layer.table.read_record(i) |
98 |
value = record[field] |
if record is not None: |
99 |
|
value = record[field] |
100 |
# |
except: |
101 |
# if the above statements fail 'value' should |
pass |
102 |
# be null, at which point this call will |
|
103 |
# at least retreive the NullData |
# |
104 |
# |
# if the above statements fail 'value' should |
105 |
prop = lc.GetClassData(value) |
# be null, at which point this call will |
106 |
|
# at least retreive the NullData |
107 |
|
# |
108 |
|
|
109 |
|
group = lc.FindGroup(value) |
110 |
|
|
111 |
|
#prop = lc.GetProperties(value) |
112 |
|
else: |
113 |
|
group = lc.GetDefaultGroup() |
114 |
|
|
115 |
|
|
116 |
|
if not group.IsVisible(): |
117 |
|
continue |
118 |
|
|
119 |
|
prop = group.GetProperties() |
120 |
|
|
121 |
|
# don't recreate new objects if they are the same as before |
122 |
if prop != old_prop: |
if prop != old_prop: |
123 |
old_prop = prop |
old_prop = prop |
124 |
|
|
125 |
if shapetype == SHAPETYPE_ARC: |
if shapetype == SHAPETYPE_ARC: |
126 |
fill = Color.None |
fill = Color.Transparent |
127 |
else: |
else: |
128 |
fill = prop.GetFill() |
fill = prop.GetFill() |
129 |
|
|
130 |
if fill is Color.None: |
|
131 |
|
if fill is Color.Transparent: |
132 |
brush = wxTRANSPARENT_BRUSH |
brush = wxTRANSPARENT_BRUSH |
133 |
else: |
else: |
134 |
color = wxColour(fill.red * 255, |
color = Color2wxColour(fill) |
|
fill.green * 255, |
|
|
fill.blue * 255) |
|
135 |
brush = wxBrush(color, wxSOLID) |
brush = wxBrush(color, wxSOLID) |
136 |
|
|
137 |
stroke = prop.GetStroke() |
stroke = prop.GetLineColor() |
138 |
stroke_width = prop.GetStrokeWidth() |
stroke_width = prop.GetLineWidth() |
139 |
if stroke is Color.None: |
if stroke is Color.Transparent: |
140 |
pen = wxTRANSPARENT_PEN |
pen = wxTRANSPARENT_PEN |
141 |
else: |
else: |
142 |
color = wxColour(stroke.red * 255, |
color = Color2wxColour(stroke) |
|
stroke.green * 255, |
|
|
stroke.blue * 255) |
|
143 |
pen = wxPen(color, stroke_width, wxSOLID) |
pen = wxPen(color, stroke_width, wxSOLID) |
144 |
|
|
145 |
|
|
146 |
if shapetype == SHAPETYPE_POINT: |
if shapetype == SHAPETYPE_POINT: |
147 |
self.dc.SetBrush(brush) |
self.dc.SetBrush(brush) |
148 |
self.dc.SetPen(pen) |
self.dc.SetPen(pen) |
149 |
self.draw_point_shape(layer, i) |
self.draw_point_shape(layer, i) |
150 |
else: |
else: |
151 |
self.draw_polygon_shape(layer, i, pen, brush) |
self.draw_polygon_shape(polygon_render_param, i, pen, brush) |
152 |
|
|
153 |
def layer_ids(self, layer): |
def layer_ids(self, layer): |
154 |
"""Return the shape ids of the given layer that have to be drawn. |
"""Return the shape ids of the given layer that have to be drawn. |
155 |
|
|
156 |
The default implementation simply returns all ids in the layer. |
The default implementation simply returns all ids in the layer. |
157 |
Override in derived classes to be more precise. |
Override in derived classes to be more precise. |
158 |
""" |
""" |
159 |
return range(layer.NumShapes()) |
return range(layer.NumShapes()) |
160 |
|
|
161 |
def draw_polygon_shape(self, layer, index, pen, brush): |
def polygon_render_param(self, layer): |
162 |
offx, offy = self.offset |
"""Return the low-lever render parameter for the layer""" |
163 |
draw_polygon_shape(layer.shapefile.cobject(), index, |
offx, offy = self.offset |
164 |
self.dc, pen, brush, |
return draw_polygon_init(layer.shapefile, self.dc, |
165 |
self.map.projection, layer.projection, |
self.map.projection, |
166 |
self.scale, -self.scale, offx, offy) |
layer.projection, |
167 |
|
self.scale, -self.scale, |
168 |
|
offx, offy) |
169 |
|
|
170 |
|
def draw_polygon_shape(self, draw_polygon_info, index, pen, brush): |
171 |
|
draw_polygon_shape(draw_polygon_info, index, pen, brush) |
172 |
|
|
173 |
def projected_points(self, layer, index): |
def projected_points(self, layer, index): |
174 |
proj = self.map.projection |
proj = self.map.projection |
199 |
self.dc.DrawLines(points) |
self.dc.DrawLines(points) |
200 |
|
|
201 |
def draw_point_shape(self, layer, index): |
def draw_point_shape(self, layer, index): |
202 |
p = self.projected_points(layer, index)[0] |
pp = self.projected_points(layer, index) |
203 |
|
|
204 |
|
if len(pp) == 0: return # ignore Null Shapes which have no points |
205 |
|
|
206 |
|
p = pp[0] |
207 |
radius = self.resolution * 5 |
radius = self.resolution * 5 |
208 |
self.dc.DrawEllipse(p.x - radius, p.y - radius, 2*radius, 2*radius) |
self.dc.DrawEllipse(p.x - radius, p.y - radius, 2*radius, 2*radius) |
209 |
|
|
251 |
# On the screen we want to see only visible layers by default |
# On the screen we want to see only visible layers by default |
252 |
honor_visibility = 1 |
honor_visibility = 1 |
253 |
|
|
254 |
def RenderMap(self, map, region, selected_layer, selected_shape): |
def RenderMap(self, map, region, selected_layer, selected_shapes): |
255 |
"""Render the map. |
"""Render the map. |
256 |
|
|
257 |
Only the given region (a tuple in window coordinates as returned |
Only the given region (a tuple in window coordinates as returned |
258 |
by a wxrect's asTuple method) needs to be redrawn. Highlight the |
by a wxrect's asTuple method) needs to be redrawn. Highlight the |
259 |
shape with id selected_shape in the selected_layer. |
shapes given by the ids in selected_shapes in the |
260 |
|
selected_layer. |
261 |
""" |
""" |
262 |
self.update_region = region |
self.update_region = region |
263 |
self.selected_layer = selected_layer |
self.selected_layer = selected_layer |
264 |
self.selected_shape = selected_shape |
self.selected_shapes = selected_shapes |
265 |
self.render_map(map) |
self.render_map(map) |
266 |
|
|
267 |
def draw_shape_layer(self, layer): |
def draw_shape_layer(self, layer): |
268 |
MapRenderer.draw_shape_layer(self, layer) |
MapRenderer.draw_shape_layer(self, layer) |
269 |
if layer is self.selected_layer and self.selected_shape is not None: |
if layer is self.selected_layer and self.selected_shapes: |
270 |
pen = wxPen(wxBLACK, 3, wxSOLID) |
pen = wxPen(wxBLACK, 3, wxSOLID) |
271 |
brush = wxBrush(wxBLACK, wxCROSS_HATCH) |
brush = wxBrush(wxBLACK, wxCROSS_HATCH) |
272 |
|
|
273 |
shapetype = layer.ShapeType() |
shapetype = layer.ShapeType() |
|
index = self.selected_shape |
|
274 |
if shapetype == SHAPETYPE_POLYGON: |
if shapetype == SHAPETYPE_POLYGON: |
275 |
self.draw_polygon_shape(layer, index, pen, brush) |
offx, offy = self.offset |
276 |
|
renderparam = self.polygon_render_param(layer) |
277 |
|
func = self.draw_polygon_shape |
278 |
|
args = (pen, brush) |
279 |
elif shapetype == SHAPETYPE_ARC: |
elif shapetype == SHAPETYPE_ARC: |
280 |
self.draw_polygon_shape(layer, index, pen, None) |
renderparam = self.polygon_render_param(layer) |
281 |
else: |
func = self.draw_polygon_shape |
282 |
|
args = (pen, None) |
283 |
|
elif shapetype == SHAPETYPE_POINT: |
284 |
|
renderparam = layer |
285 |
self.dc.SetBrush(brush) |
self.dc.SetBrush(brush) |
286 |
self.dc.SetPen(pen) |
self.dc.SetPen(pen) |
287 |
if shapetype == SHAPETYPE_POINT: |
func = self.draw_point_shape |
288 |
self.draw_point_shape(layer, index) |
args = () |
289 |
else: |
else: |
290 |
raise TypeError(_("Unhandled shape type %s") % shapetype) |
raise TypeError(_("Unhandled shape type %s") % shapetype) |
291 |
|
|
292 |
|
for index in self.selected_shapes: |
293 |
|
func(renderparam, index, *args) |
294 |
|
|
295 |
|
|
296 |
def layer_ids(self, layer): |
def layer_ids(self, layer): |
297 |
"""Return the shapeids covered by the region that has to be redrawn |
"""Return the shapeids covered by the region that has to be redrawn |