1 |
# Copyright (c) 2001, 2002 by Intevation GmbH |
# Copyright (c) 2001, 2002, 2003 by Intevation GmbH |
2 |
# Authors: |
# Authors: |
3 |
# Bernhard Herzog <[email protected]> |
# Bernhard Herzog <[email protected]> |
4 |
# Jonathan Coles <[email protected]> |
# Jonathan Coles <[email protected]> |
15 |
|
|
16 |
import shapelib, shptree |
import shapelib, shptree |
17 |
|
|
18 |
from messages import LAYER_PROJECTION_CHANGED, LAYER_LEGEND_CHANGED, \ |
from messages import LAYER_PROJECTION_CHANGED, LAYER_VISIBILITY_CHANGED, \ |
19 |
LAYER_VISIBILITY_CHANGED |
LAYER_CHANGED |
20 |
|
|
21 |
from color import Color |
from color import Color |
22 |
|
|
23 |
from classification import Classification |
import classification |
24 |
|
|
25 |
from table import Table |
from table import Table |
26 |
|
|
102 |
|
|
103 |
TITLE_CHANGED -- The title has changed. |
TITLE_CHANGED -- The title has changed. |
104 |
LAYER_PROJECTION_CHANGED -- the projection has changed. |
LAYER_PROJECTION_CHANGED -- the projection has changed. |
|
LAYER_LEGEND_CHANGED -- the fill or stroke attributes have changed |
|
|
|
|
105 |
""" |
""" |
106 |
|
|
107 |
def __init__(self, title, filename, projection = None, |
def __init__(self, title, filename, projection = None, |
108 |
fill = Color.None, |
fill = Color.Transparent, |
109 |
stroke = Color.Black, |
stroke = Color.Black, |
110 |
stroke_width = 1, |
lineWidth = 1, |
111 |
visible = 1): |
visible = 1): |
112 |
"""Initialize the layer. |
"""Initialize the layer. |
113 |
|
|
116 |
projection -- the projection object. Its Inverse method is |
projection -- the projection object. Its Inverse method is |
117 |
assumed to map the layer's coordinates to lat/long |
assumed to map the layer's coordinates to lat/long |
118 |
coordinates |
coordinates |
119 |
fill -- the fill color or None if the shapes are not filled |
fill -- the fill color or Color.Transparent if the shapes are |
120 |
stroke -- the stroke color or None if the shapes are not stroked |
not filled |
121 |
|
stroke -- the stroke color or Color.Transparent if the shapes |
122 |
|
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 |
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 |
|
# |
149 |
|
self.__classification = None |
150 |
|
self.__setClassLock = False |
151 |
|
|
152 |
|
|
153 |
|
self.SetClassification(None) |
154 |
|
|
155 |
|
self.__classification.SetDefaultLineColor(stroke) |
156 |
|
self.__classification.SetDefaultLineWidth(lineWidth) |
157 |
self.__classification.SetDefaultFill(fill) |
self.__classification.SetDefaultFill(fill) |
158 |
|
self.__classification.SetLayer(self) |
159 |
|
|
160 |
self.UnsetModified() |
self.UnsetModified() |
161 |
|
|
191 |
self.shapefile.close() |
self.shapefile.close() |
192 |
self.shapefile = None |
self.shapefile = None |
193 |
self.shapetree = None |
self.shapetree = None |
194 |
|
self.SetClassification(None) |
195 |
self.table.Destroy() |
self.table.Destroy() |
196 |
|
|
197 |
def BoundingBox(self): |
def BoundingBox(self): |
199 |
|
|
200 |
If the layer has no shapes, return None. |
If the layer has no shapes, return None. |
201 |
""" |
""" |
|
# The bbox will be set by open_shapefile just as we need it |
|
|
# here. |
|
|
self.open_shapefile() |
|
202 |
return self.bbox |
return self.bbox |
203 |
|
|
204 |
def LatLongBoundingBox(self): |
def LatLongBoundingBox(self): |
216 |
else: |
else: |
217 |
return None |
return None |
218 |
|
|
219 |
|
def GetFieldType(self, fieldName): |
220 |
|
info = self.table.field_info_by_name(fieldName) |
221 |
|
if info is not None: |
222 |
|
return info[0] |
223 |
|
else: |
224 |
|
return None |
225 |
|
|
226 |
def NumShapes(self): |
def NumShapes(self): |
227 |
"""Return the number of shapes in the layer""" |
"""Return the number of shapes in the layer""" |
|
self.open_shapefile() |
|
228 |
return self.numshapes |
return self.numshapes |
229 |
|
|
230 |
def ShapeType(self): |
def ShapeType(self): |
231 |
"""Return the type of the shapes in the layer. |
"""Return the type of the shapes in the layer. |
232 |
This is either SHAPETYPE_POINT, SHAPETYPE_ARC or SHAPETYPE_POLYGON. |
This is either SHAPETYPE_POINT, SHAPETYPE_ARC or SHAPETYPE_POLYGON. |
233 |
""" |
""" |
|
self.open_shapefile() |
|
234 |
return self.shapetype |
return self.shapetype |
235 |
|
|
236 |
def Shape(self, index): |
def Shape(self, index): |
237 |
"""Return the shape with index index""" |
"""Return the shape with index index""" |
|
self.open_shapefile() |
|
238 |
shape = self.shapefile.read_object(index) |
shape = self.shapefile.read_object(index) |
239 |
|
|
240 |
if self.shapetype == SHAPETYPE_POINT: |
if self.shapetype == SHAPETYPE_POINT: |
266 |
return self.__classification |
return self.__classification |
267 |
|
|
268 |
def SetClassification(self, clazz): |
def SetClassification(self, clazz): |
269 |
self.__classification = clazz |
"""Set the classification to 'clazz' |
270 |
self.changed(LAYER_LEGEND_CHANGED, self) |
|
271 |
|
If 'clazz' is None a default classification is created |
272 |
|
""" |
273 |
|
|
274 |
|
# prevent infinite recursion when calling SetLayer() |
275 |
|
if self.__setClassLock: return |
276 |
|
|
277 |
|
self.__setClassLock = True |
278 |
|
|
279 |
|
if clazz is None: |
280 |
|
if self.__classification is not None: |
281 |
|
self.__classification.SetLayer(None) |
282 |
|
self.__classification = classification.Classification() |
283 |
|
else: |
284 |
|
self.__classification = clazz |
285 |
|
try: |
286 |
|
self.__classification.SetLayer(self) |
287 |
|
except ValueError: |
288 |
|
self.__setClassLock = False |
289 |
|
raise ValueError |
290 |
|
|
291 |
|
self.changed(LAYER_CHANGED, self) |
292 |
|
|
293 |
|
self.__setClassLock = False |
294 |
|
|
295 |
|
def ClassChanged(self): |
296 |
|
"""Called from the classification object when it has changed.""" |
297 |
|
self.changed(LAYER_CHANGED, self) |
298 |
|
|
299 |
def TreeInfo(self): |
def TreeInfo(self): |
300 |
items = [] |
items = [] |
301 |
|
|