/[thuban]/trunk/thuban/Thuban/Model/layer.py
ViewVC logotype

Diff of /trunk/thuban/Thuban/Model/layer.py

Parent Directory Parent Directory | Revision Log Revision Log | View Patch Patch

revision 171 by bh, Tue May 14 14:16:24 2002 UTC revision 725 by jonathan, Thu Apr 24 16:05:36 2003 UTC
# Line 1  Line 1 
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]>
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.
# Line 9  __version__ = "$Revision$" Line 10  __version__ = "$Revision$"
10    
11  from math import log, ceil  from math import log, ceil
12    
13    from Thuban import _
14    
15  import shapelib, shptree  import shapelib, shptree
16    
17  from messages import LAYER_PROJECTION_CHANGED, LAYER_LEGEND_CHANGED, \  from messages import LAYER_PROJECTION_CHANGED, LAYER_VISIBILITY_CHANGED, \
18       LAYER_VISIBILITY_CHANGED       LAYER_CHANGED
19    
20  from color import Color  from color import Color
 # Some predefined colors for internal use  
 _black = Color(0, 0, 0)  
   
21    
22  from table import Table  import classification
23    
24  from base import TitledObject, Modifiable  from base import TitledObject, Modifiable
25    
26    
27  class Shape:  class Shape:
28    
29      """Represent one shape"""      """Represent one shape"""
# Line 83  class BaseLayer(TitledObject, Modifiable Line 84  class BaseLayer(TitledObject, Modifiable
84          """Set the layer's visibility."""          """Set the layer's visibility."""
85          self.visible = visible          self.visible = visible
86          self.issue(LAYER_VISIBILITY_CHANGED, self)          self.issue(LAYER_VISIBILITY_CHANGED, self)
87            
88            
89  class Layer(BaseLayer):  class Layer(BaseLayer):
90    
91      """Represent the information of one geodata file (currently a shapefile)      """Represent the information of one geodata file (currently a shapefile)
# Line 99  class Layer(BaseLayer): Line 100  class Layer(BaseLayer):
100    
101          TITLE_CHANGED -- The title has changed.          TITLE_CHANGED -- The title has changed.
102          LAYER_PROJECTION_CHANGED -- the projection has changed.          LAYER_PROJECTION_CHANGED -- the projection has changed.
         LAYER_LEGEND_CHANGED -- the fill or stroke attributes have changed  
   
103      """      """
104    
105      def __init__(self, title, filename, projection = None,      def __init__(self, title, data, projection = None,
106                   fill = None, stroke = _black, stroke_width = 1, visible = 1):                   fill = Color.Transparent,
107                     stroke = Color.Black,
108                     lineWidth = 1,
109                     visible = 1):
110          """Initialize the layer.          """Initialize the layer.
111    
112          title -- the title          title -- the title
113          filename -- the name of the shapefile          data -- datastore object for the shape data shown by the layer
114          projection -- the projection object. Its Inverse method is          projection -- the projection object. Its Inverse method is
115                 assumed to map the layer's coordinates to lat/long                 assumed to map the layer's coordinates to lat/long
116                 coordinates                 coordinates
117          fill -- the fill color or None if the shapes are not filled          fill -- the fill color or Color.Transparent if the shapes are
118          stroke -- the stroke color or None if the shapes are not stroked                  not filled
119            stroke -- the stroke color or Color.Transparent if the shapes
120                    are not stroked
121          visible -- boolean. If true the layer is visible.          visible -- boolean. If true the layer is visible.
122    
123          colors are expected to be instances of Color class          colors are expected to be instances of Color class
124          """          """
125          BaseLayer.__init__(self, title, visible = visible)          BaseLayer.__init__(self, title, visible = visible)
126          self.filename = filename  
127          self.projection = projection          self.projection = projection
128          self.fill = fill  
129          self.stroke = stroke          #
130          self.stroke_width = stroke_width          # this is really important so that when the classification class
131          self.shapefile = None          # tries to set its parent layer the variable will exist
132          self.shapetree = None          #
133          self.open_shapefile()          self.__classification = None
134          # shapetable is the table associated with the shapefile, while          self.__setClassLock = False
135          # table is the default table used to look up attributes for  
136          # display          self.SetShapeStore(data)
137          self.shapetable = Table(filename)  
138            self.SetClassification(None)
139    
140            self.__classification.SetDefaultLineColor(stroke)
141            self.__classification.SetDefaultLineWidth(lineWidth)
142            self.__classification.SetDefaultFill(fill)
143            self.__classification.SetLayer(self)
144    
145            self.UnsetModified()
146    
147    
148        def SetShapeStore(self, store):
149            self.store = store
150            self.shapefile = self.store.Shapefile()
151            self.shapetable = self.store.Table()
152            self.filename = self.store.filename
153          self.table = self.shapetable          self.table = self.shapetable
154    
155      def open_shapefile(self):          numshapes, shapetype, mins, maxs = self.shapefile.info()
156          if self.shapefile is None:          self.numshapes = numshapes
157              self.shapefile = shapelib.ShapeFile(self.filename)          self.shapetype = shapelib_shapetypes[shapetype]
158              numshapes, shapetype, mins, maxs = self.shapefile.info()  
159              self.numshapes = numshapes          # if there are shapes, set the bbox accordingly. Otherwise
160              self.shapetype = shapelib_shapetypes[shapetype]          # set it to None.
161            if self.numshapes:
162              self.bbox = mins[:2] + maxs[:2]              self.bbox = mins[:2] + maxs[:2]
163            else:
164                self.bbox = None
165    
166              # estimate a good depth for the quad tree. Each depth          # estimate a good depth for the quad tree. Each depth
167              # multiplies the number of nodes by four, therefore we          # multiplies the number of nodes by four, therefore we
168              # basically take the base 4 logarithm of the number of          # basically take the base 4 logarithm of the number of
169              # shapes.          # shapes.
170              if self.numshapes < 4:          if self.numshapes < 4:
171                  maxdepth = 1              maxdepth = 1
172              else:          else:
173                  maxdepth = int(ceil(log(self.numshapes / 4.0) / log(4)))              maxdepth = int(ceil(log(self.numshapes / 4.0) / log(4)))
174    
175              self.shapetree = shptree.SHPTree(self.shapefile.cobject(), 2,          self.shapetree = shptree.SHPTree(self.shapefile.cobject(), 2,
176                                               maxdepth)                                           maxdepth)
177            if self.__classification is not None:
178                fieldname = self.__classification.GetField()
179                if not self.store.Table().field_info_by_name(fieldname):
180                    self.SetClassification(None)
181            self.changed(LAYER_CHANGED, self)
182    
183        def ShapeStore(self):
184            return self.store
185    
186        def Destroy(self):
187            BaseLayer.Destroy(self)
188            self.SetClassification(None)
189    
190      def BoundingBox(self):      def BoundingBox(self):
191          """Return the bounding box of the layer's shapes in their default          """Return the layer's bounding box in the intrinsic coordinate system.
192          coordinate system"""  
193          self.open_shapefile()          If the layer has no shapes, return None.
194            """
195          return self.bbox          return self.bbox
196    
197      def LatLongBoundingBox(self):      def LatLongBoundingBox(self):
198          """Return the layer's bounding box in lat/long coordinates"""          """Return the layer's bounding box in lat/long coordinates.
199          llx, lly, urx, ury = self.BoundingBox()  
200          if self.projection is not None:          Return None, if the layer doesn't contain any shapes.
201              llx, lly = self.projection.Inverse(llx, lly)          """
202              urx, ury = self.projection.Inverse(urx, ury)          bbox = self.BoundingBox()
203          return llx, lly, urx, ury          if bbox is not None:
204                llx, lly, urx, ury = bbox
205                if self.projection is not None:
206                    llx, lly = self.projection.Inverse(llx, lly)
207                    urx, ury = self.projection.Inverse(urx, ury)
208                return llx, lly, urx, ury
209            else:
210                return None
211    
212        def GetFieldType(self, fieldName):
213            info = self.table.field_info_by_name(fieldName)
214            if info is not None:
215                return info[0]
216            else:
217                return None
218    
219      def NumShapes(self):      def NumShapes(self):
220          """Return the number of shapes in the layer"""          """Return the number of shapes in the layer"""
         self.open_shapefile()  
221          return self.numshapes          return self.numshapes
222    
223      def ShapeType(self):      def ShapeType(self):
224          """Return the type of the shapes in the layer.          """Return the type of the shapes in the layer.
225          This is either SHAPETYPE_POINT, SHAPETYPE_ARC or SHAPETYPE_POLYGON.          This is either SHAPETYPE_POINT, SHAPETYPE_ARC or SHAPETYPE_POLYGON.
226          """          """
         self.open_shapefile()  
227          return self.shapetype          return self.shapetype
228    
229      def Shape(self, index):      def Shape(self, index):
230          """Return the shape with index index"""          """Return the shape with index index"""
         self.open_shapefile()  
231          shape = self.shapefile.read_object(index)          shape = self.shapefile.read_object(index)
232    
233          if self.shapetype == SHAPETYPE_POINT:          if self.shapetype == SHAPETYPE_POINT:
234              points = shape.vertices()              points = shape.vertices()
235          else:          else:
# Line 191  class Layer(BaseLayer): Line 238  class Layer(BaseLayer):
238              points = []              points = []
239              for x, y in poly:              for x, y in poly:
240                  points.append((x, y))                  points.append((x, y))
241    
242          return Shape(points)          return Shape(points)
243    
244      def ShapesInRegion(self, box):      def ShapesInRegion(self, box):
# Line 202  class Layer(BaseLayer): Line 250  class Layer(BaseLayer):
250          left, bottom, right, top = box          left, bottom, right, top = box
251          return self.shapetree.find_shapes((left, bottom), (right, top))          return self.shapetree.find_shapes((left, bottom), (right, top))
252    
253        def GetProjection(self):
254            return self.projection
255    
256      def SetProjection(self, projection):      def SetProjection(self, projection):
257          """Set the layer's projection"""          """Set the layer's projection"""
258          self.projection = projection          self.projection = projection
259          self.changed(LAYER_PROJECTION_CHANGED, self)          self.changed(LAYER_PROJECTION_CHANGED, self)
260    
261      def SetFill(self, fill):      def GetClassification(self):
262          """Set the layer's fill color. None means the shapes are not filled"""          return self.__classification
263          self.fill = fill  
264          self.changed(LAYER_LEGEND_CHANGED, self)      def SetClassification(self, clazz):
265            """Set the classification to 'clazz'
266      def SetStroke(self, stroke):  
267          """Set the layer's stroke color. None means the shapes are not          If 'clazz' is None a default classification is created
268          stroked."""          """
269          self.stroke = stroke  
270          self.changed(LAYER_LEGEND_CHANGED, self)          # prevent infinite recursion when calling SetLayer()
271            if self.__setClassLock: return
272      def SetStrokeWidth(self, width):  
273          """Set the layer's stroke width."""          self.__setClassLock = True
274          self.stroke_width = width  
275          self.changed(LAYER_LEGEND_CHANGED, self)          if clazz is None:
276                if self.__classification is not None:
277                    self.__classification.SetLayer(None)
278                self.__classification = classification.Classification()
279            else:
280                self.__classification = clazz
281                try:
282                    self.__classification.SetLayer(self)
283                except ValueError:
284                    self.__setClassLock = False
285                    raise ValueError
286    
287            self.changed(LAYER_CHANGED, self)
288    
289            self.__setClassLock = False
290    
291        def ClassChanged(self):
292            """Called from the classification object when it has changed."""
293            self.changed(LAYER_CHANGED, self)
294    
295        def TreeInfo(self):
296            items = []
297    
298            if self.Visible():
299                items.append(_("Shown"))
300            else:
301                items.append(_("Hidden"))
302            items.append(_("Shapes: %d") % self.NumShapes())
303    
304            bbox = self.LatLongBoundingBox()
305            if bbox is not None:
306                items.append(_("Extent (lat-lon): (%g, %g, %g, %g)") % bbox)
307            else:
308                items.append(_("Extent (lat-lon):"))
309            items.append(_("Shapetype: %s") % shapetype_names[self.ShapeType()])
310    
311            items.append(self.__classification)
312    
313            return (_("Layer '%s'") % self.Title(), items)
314    
Legend:
Removed from v.171  
changed lines
  Added in v.725
[email protected]
 ViewVC Help
Powered by ViewVC 1.1.26