1 |
# Copyright (c) 2001, 2002 by Intevation GmbH |
# Copyright (c) 2001, 2002 by Intevation GmbH |
2 |
# Authors: |
# Authors: |
3 |
# Bernhard Herzog <[email protected]> |
# Bernhard Herzog <[email protected]> |
4 |
|
# Jonathan Coles <[email protected]> |
5 |
# |
# |
6 |
# This program is free software under the GPL (>=v2) |
# This program is free software under the GPL (>=v2) |
7 |
# Read the file COPYING coming with Thuban for details. |
# Read the file COPYING coming with Thuban for details. |
19 |
LAYER_VISIBILITY_CHANGED |
LAYER_VISIBILITY_CHANGED |
20 |
|
|
21 |
from color import Color |
from color import Color |
|
# Some predefined colors for internal use |
|
|
_black = Color(0, 0, 0) |
|
22 |
|
|
23 |
from classification import Classification |
import classification |
24 |
|
|
25 |
from table import Table |
from table import Table |
26 |
|
|
107 |
""" |
""" |
108 |
|
|
109 |
def __init__(self, title, filename, projection = None, |
def __init__(self, title, filename, projection = None, |
110 |
fill = None, stroke = _black, stroke_width = 1, visible = 1): |
fill = Color.None, |
111 |
|
stroke = Color.Black, |
112 |
|
lineWidth = 1, |
113 |
|
visible = 1): |
114 |
"""Initialize the layer. |
"""Initialize the layer. |
115 |
|
|
116 |
title -- the title |
title -- the title |
118 |
projection -- the projection object. Its Inverse method is |
projection -- the projection object. Its Inverse method is |
119 |
assumed to map the layer's coordinates to lat/long |
assumed to map the layer's coordinates to lat/long |
120 |
coordinates |
coordinates |
121 |
fill -- the fill color or None if the shapes are not filled |
fill -- the fill color or Color.None if the shapes are not filled |
122 |
stroke -- the stroke color or None if the shapes are not stroked |
stroke -- the stroke color or Color.None if the shapes are not stroked |
123 |
visible -- boolean. If true the layer is visible. |
visible -- boolean. If true the layer is visible. |
124 |
|
|
125 |
colors are expected to be instances of Color class |
colors are expected to be instances of Color class |
133 |
self.filename = os.path.abspath(filename) |
self.filename = os.path.abspath(filename) |
134 |
|
|
135 |
self.projection = projection |
self.projection = projection |
|
self.fill = fill |
|
|
self.stroke = stroke |
|
|
self.stroke_width = stroke_width |
|
136 |
self.shapefile = None |
self.shapefile = None |
137 |
self.shapetree = None |
self.shapetree = None |
138 |
self.open_shapefile() |
self.open_shapefile() |
142 |
self.shapetable = Table(filename) |
self.shapetable = Table(filename) |
143 |
self.table = self.shapetable |
self.table = self.shapetable |
144 |
|
|
145 |
self.classification = Classification(self) |
# |
146 |
self.classification.SetDefaultStroke(stroke) |
# this is really important so that when the classification class |
147 |
self.classification.SetDefaultStrokeWidth(stroke_width) |
# tries to set its parent layer the variable will exist |
148 |
self.classification.SetDefaultFill(fill) |
# |
149 |
|
self.__classification = None |
150 |
|
|
151 |
|
|
152 |
|
self.SetClassification(None) |
153 |
|
|
154 |
|
self.__classification.SetDefaultLineColor(stroke) |
155 |
|
self.__classification.SetDefaultLineWidth(lineWidth) |
156 |
|
self.__classification.SetDefaultFill(fill) |
157 |
|
self.__classification.SetLayer(self) |
158 |
|
|
159 |
self.UnsetModified() |
self.UnsetModified() |
160 |
|
|
217 |
else: |
else: |
218 |
return None |
return None |
219 |
|
|
220 |
|
def GetFieldType(self, fieldName): |
221 |
|
self.open_shapefile() |
222 |
|
info = self.table.field_info_by_name(fieldName) |
223 |
|
if info is not None: |
224 |
|
return info[0] |
225 |
|
else: |
226 |
|
return None |
227 |
|
|
228 |
def NumShapes(self): |
def NumShapes(self): |
229 |
"""Return the number of shapes in the layer""" |
"""Return the number of shapes in the layer""" |
230 |
self.open_shapefile() |
self.open_shapefile() |
267 |
self.projection = projection |
self.projection = projection |
268 |
self.changed(LAYER_PROJECTION_CHANGED, self) |
self.changed(LAYER_PROJECTION_CHANGED, self) |
269 |
|
|
270 |
|
def GetClassification(self): |
271 |
|
return self.__classification |
272 |
|
|
273 |
|
def SetClassification(self, clazz): |
274 |
|
"""Set the classification to 'clazz' |
275 |
|
|
276 |
|
If 'clazz' is None a default classification is created |
277 |
|
""" |
278 |
|
|
279 |
|
if clazz is None: |
280 |
|
self.__classification = classification.Classification() |
281 |
|
else: |
282 |
|
|
283 |
|
# prevent infinite recursion when calling SetLayer() |
284 |
|
if clazz == self.__classification: |
285 |
|
return |
286 |
|
|
287 |
|
self.__classification = clazz |
288 |
|
self.__classification.SetLayer(self) |
289 |
|
|
290 |
|
self.changed(LAYER_LEGEND_CHANGED, self) |
291 |
|
|
292 |
|
def ClassChanged(self): |
293 |
|
"""Called from the classification object when it has changed.""" |
294 |
|
self.changed(LAYER_LEGEND_CHANGED, self) |
295 |
|
|
296 |
def TreeInfo(self): |
def TreeInfo(self): |
297 |
items = [] |
items = [] |
298 |
|
|
309 |
items.append(_("Extent (lat-lon):")) |
items.append(_("Extent (lat-lon):")) |
310 |
items.append(_("Shapetype: %s") % shapetype_names[self.ShapeType()]) |
items.append(_("Shapetype: %s") % shapetype_names[self.ShapeType()]) |
311 |
|
|
312 |
def color_string(color): |
items.append(self.__classification) |
|
if color is None: |
|
|
return "None" |
|
|
return "(%.3f, %.3f, %.3f)" % (color.red, color.green, color.blue) |
|
|
|
|
|
# layers will always have a classification with at least a NULL data set |
|
|
|
|
|
#items.append((_("Fill: %s") % color_string(self.fill), self.fill)) |
|
|
#items.append((_("Outline: %s") % color_string(self.stroke), self.stroke)) |
|
|
|
|
|
items.append(self.classification) |
|
313 |
|
|
314 |
return (_("Layer '%s'") % self.Title(), items) |
return (_("Layer '%s'") % self.Title(), items) |
315 |
|
|