26 |
|
|
27 |
from Thuban.Model.session import Session |
from Thuban.Model.session import Session |
28 |
from Thuban.Model.map import Map |
from Thuban.Model.map import Map |
29 |
from Thuban.Model.layer import Layer |
from Thuban.Model.layer import Layer, RasterLayer |
30 |
from Thuban.Model.color import Color |
from Thuban.Model.color import Color |
31 |
from Thuban.Model.proj import Projection |
from Thuban.Model.proj import Projection |
32 |
|
from Thuban.Model.range import Range |
33 |
from Thuban.Model.classification import Classification, \ |
from Thuban.Model.classification import Classification, \ |
34 |
ClassGroupDefault, ClassGroupSingleton, ClassGroupRange, ClassGroupMap, \ |
ClassGroupDefault, ClassGroupSingleton, ClassGroupRange, ClassGroupMap, \ |
35 |
ClassGroupProperties |
ClassGroupProperties |
165 |
if method_name is not None and method_name[1] is not None: |
if method_name is not None and method_name[1] is not None: |
166 |
getattr(self, method_name[1])(name, qname) |
getattr(self, method_name[1])(name, qname) |
167 |
|
|
168 |
|
def encode(self, str): |
169 |
|
"""Assume that str is in Unicode and encode it into Latin1. |
170 |
|
|
171 |
|
If str is None, return None |
172 |
|
""" |
173 |
|
|
174 |
|
if str is not None: |
175 |
|
return str.encode("latin1") |
176 |
|
else: |
177 |
|
return None |
178 |
|
|
179 |
class SessionLoader(XMLReader): |
class SessionLoader(XMLReader): |
180 |
|
|
181 |
def __init__(self): |
def __init__(self): |
192 |
'projection' : ("start_projection", "end_projection"), |
'projection' : ("start_projection", "end_projection"), |
193 |
'parameter' : ("start_parameter", None), |
'parameter' : ("start_parameter", None), |
194 |
'layer' : ("start_layer", "end_layer"), |
'layer' : ("start_layer", "end_layer"), |
195 |
|
'rasterlayer' : ("start_rasterlayer", "end_rasterlayer"), |
196 |
'classification': ("start_classification", "end_classification"), |
'classification': ("start_classification", "end_classification"), |
197 |
'clnull' : ("start_clnull", "end_clnull"), |
'clnull' : ("start_clnull", "end_clnull"), |
198 |
'clpoint' : ("start_clpoint", "end_clpoint"), |
'clpoint' : ("start_clpoint", "end_clpoint"), |
203 |
'label' : ("start_label", None)}) |
'label' : ("start_label", None)}) |
204 |
|
|
205 |
def start_session(self, name, qname, attrs): |
def start_session(self, name, qname, attrs): |
206 |
self.theSession = Session(attrs.get((None, 'title'), None)) |
self.theSession = Session(self.encode(attrs.get((None, 'title'), None))) |
207 |
|
|
208 |
def end_session(self, name, qname): |
def end_session(self, name, qname): |
209 |
pass |
pass |
211 |
def start_map(self, name, qname, attrs): |
def start_map(self, name, qname, attrs): |
212 |
"""Start a map.""" |
"""Start a map.""" |
213 |
self.aMap = Map(attrs.get((None, 'title'), None)) |
self.aMap = Map(attrs.get((None, 'title'), None)) |
|
self.__projReceiver = self.aMap |
|
214 |
|
|
215 |
def end_map(self, name, qname): |
def end_map(self, name, qname): |
216 |
self.theSession.AddMap(self.aMap) |
self.theSession.AddMap(self.aMap) |
217 |
self.__projReceiver = None |
self.aMap = None |
218 |
|
|
219 |
def start_projection(self, name, qname, attrs): |
def start_projection(self, name, qname, attrs): |
220 |
self.ProjectionName = attrs.get((None, 'name'), None) |
self.ProjectionName = self.encode(attrs.get((None, 'name'), None)) |
221 |
self.ProjectionParams = [ ] |
self.ProjectionParams = [ ] |
222 |
|
|
223 |
def end_projection(self, name, qname): |
def end_projection(self, name, qname): |
224 |
self.__projReceiver.SetProjection( |
if self.aLayer is not None: |
225 |
|
obj = self.aLayer |
226 |
|
elif self.aMap is not None: |
227 |
|
obj = self.aMap |
228 |
|
else: |
229 |
|
assert False, "projection tag out of context" |
230 |
|
pass |
231 |
|
|
232 |
|
obj.SetProjection( |
233 |
Projection(self.ProjectionParams, self.ProjectionName)) |
Projection(self.ProjectionParams, self.ProjectionName)) |
234 |
|
|
235 |
def start_parameter(self, name, qname, attrs): |
def start_parameter(self, name, qname, attrs): |
244 |
attrs which may be a dictionary as well as the normal SAX attrs |
attrs which may be a dictionary as well as the normal SAX attrs |
245 |
object and bind it to self.aLayer. |
object and bind it to self.aLayer. |
246 |
""" |
""" |
247 |
title = attrs.get((None, 'title'), "") |
title = self.encode(attrs.get((None, 'title'), "")) |
248 |
filename = attrs.get((None, 'filename'), "") |
filename = attrs.get((None, 'filename'), "") |
249 |
filename = os.path.join(self.GetDirectory(), filename) |
filename = os.path.join(self.GetDirectory(), filename) |
250 |
visible = attrs.get((None, 'visible'), "true") |
filename = self.encode(filename) |
251 |
|
visible = self.encode(attrs.get((None, 'visible'), "true")) != "false" |
252 |
fill = parse_color(attrs.get((None, 'fill'), "None")) |
fill = parse_color(attrs.get((None, 'fill'), "None")) |
253 |
stroke = parse_color(attrs.get((None, 'stroke'), "#000000")) |
stroke = parse_color(attrs.get((None, 'stroke'), "#000000")) |
254 |
stroke_width = int(attrs.get((None, 'stroke_width'), "1")) |
stroke_width = int(attrs.get((None, 'stroke_width'), "1")) |
256 |
self.theSession.OpenShapefile(filename), |
self.theSession.OpenShapefile(filename), |
257 |
fill = fill, stroke = stroke, |
fill = fill, stroke = stroke, |
258 |
lineWidth = stroke_width, |
lineWidth = stroke_width, |
259 |
visible = visible != "false") |
visible = visible) |
|
|
|
|
self.__projReceiver = self.aLayer |
|
260 |
|
|
261 |
def end_layer(self, name, qname): |
def end_layer(self, name, qname): |
262 |
self.aMap.AddLayer(self.aLayer) |
self.aMap.AddLayer(self.aLayer) |
263 |
self.__projReceiver = None |
self.aLayer = None |
264 |
|
|
265 |
|
def start_rasterlayer(self, name, qname, attrs, layer_class = RasterLayer): |
266 |
|
title = self.encode(attrs.get((None, 'title'), "")) |
267 |
|
filename = attrs.get((None, 'filename'), "") |
268 |
|
filename = os.path.join(self.GetDirectory(), filename) |
269 |
|
filename = self.encode(filename) |
270 |
|
visible = self.encode(attrs.get((None, 'visible'), "true")) != "false" |
271 |
|
|
272 |
|
self.aLayer = layer_class(title, filename, visible = visible) |
273 |
|
|
274 |
|
def end_rasterlayer(self, name, qname): |
275 |
|
self.aMap.AddLayer(self.aLayer) |
276 |
|
self.aLayer = None |
277 |
|
|
278 |
def start_classification(self, name, qname, attrs): |
def start_classification(self, name, qname, attrs): |
279 |
field = attrs.get((None, 'field'), None) |
field = attrs.get((None, 'field'), None) |
295 |
|
|
296 |
self.aLayer.GetClassification().SetField(field) |
self.aLayer.GetClassification().SetField(field) |
297 |
|
|
|
|
|
298 |
def end_classification(self, name, qname): |
def end_classification(self, name, qname): |
299 |
pass |
pass |
300 |
|
|
301 |
def start_clnull(self, name, qname, attrs): |
def start_clnull(self, name, qname, attrs): |
302 |
self.cl_group = ClassGroupDefault() |
self.cl_group = ClassGroupDefault() |
303 |
self.cl_group.SetLabel(attrs.get((None, 'label'), "")) |
self.cl_group.SetLabel(self.encode(attrs.get((None, 'label'), ""))) |
304 |
self.cl_prop = ClassGroupProperties() |
self.cl_prop = ClassGroupProperties() |
305 |
|
|
306 |
def end_clnull(self, name, qname): |
def end_clnull(self, name, qname): |
314 |
value = self.conv(attrib_value) |
value = self.conv(attrib_value) |
315 |
|
|
316 |
self.cl_group = ClassGroupSingleton(value) |
self.cl_group = ClassGroupSingleton(value) |
317 |
self.cl_group.SetLabel(attrs.get((None, 'label'), "")) |
self.cl_group.SetLabel(self.encode(attrs.get((None, 'label'), ""))) |
318 |
self.cl_prop = ClassGroupProperties() |
self.cl_prop = ClassGroupProperties() |
319 |
|
|
320 |
|
|
325 |
|
|
326 |
def start_clrange(self, name, qname, attrs): |
def start_clrange(self, name, qname, attrs): |
327 |
|
|
328 |
|
range = attrs.get((None, 'range'), None) |
329 |
|
# for backward compatibility (min/max are not saved) |
330 |
|
min = attrs.get((None, 'min'), None) |
331 |
|
max = attrs.get((None, 'max'), None) |
332 |
|
|
333 |
try: |
try: |
334 |
min = self.conv(attrs.get((None, 'min'), "0")) |
if range is not None: |
335 |
max = self.conv(attrs.get((None, 'max'), "0")) |
self.cl_group = ClassGroupRange(Range(range)) |
336 |
|
elif min is not None and max is not None: |
337 |
|
self.cl_group = ClassGroupRange(self.conv(min), self.conv(max)) |
338 |
|
else: |
339 |
|
self.cl_group = ClassGroupRange(Range(None)) |
340 |
|
|
341 |
except ValueError: |
except ValueError: |
342 |
raise ValueError(_("Classification range is not a number!")) |
raise ValueError(_("Classification range is not a number!")) |
343 |
|
|
|
self.cl_group = ClassGroupRange(min, max) |
|
344 |
self.cl_group.SetLabel(attrs.get((None, 'label'), "")) |
self.cl_group.SetLabel(attrs.get((None, 'label'), "")) |
345 |
self.cl_prop = ClassGroupProperties() |
self.cl_prop = ClassGroupProperties() |
346 |
|
|
373 |
def start_label(self, name, qname, attrs): |
def start_label(self, name, qname, attrs): |
374 |
x = float(attrs[(None, 'x')]) |
x = float(attrs[(None, 'x')]) |
375 |
y = float(attrs[(None, 'y')]) |
y = float(attrs[(None, 'y')]) |
376 |
text = attrs[(None, 'text')] |
text = self.encode(attrs[(None, 'text')]) |
377 |
halign = attrs[(None, 'halign')] |
halign = attrs[(None, 'halign')] |
378 |
valign = attrs[(None, 'valign')] |
valign = attrs[(None, 'valign')] |
379 |
self.aLayer.AddLabel(x, y, text, halign = halign, valign = valign) |
self.aLayer.AddLabel(x, y, text, halign = halign, valign = valign) |