/[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 771 by jonathan, Tue Apr 29 14:34:02 2003 UTC revision 1219 by bh, Mon Jun 16 17:42:54 2003 UTC
# Line 9  Line 9 
9  __version__ = "$Revision$"  __version__ = "$Revision$"
10    
11  from math import log, ceil  from math import log, ceil
12    import warnings
13    
14  from Thuban import _  from Thuban import _
   
15  import shapelib, shptree  import shapelib, shptree
16    
17  from messages import LAYER_PROJECTION_CHANGED, LAYER_VISIBILITY_CHANGED, \  from messages import LAYER_PROJECTION_CHANGED, LAYER_VISIBILITY_CHANGED, \
18       LAYER_CHANGED       LAYER_CHANGED, LAYER_SHAPESTORE_REPLACED
   
 from color import Color  
19    
20  import classification  import classification
21    
22    from color import Color
23  from base import TitledObject, Modifiable  from base import TitledObject, Modifiable
24    
25    import resource
26    
27    
28  class Shape:  class Shape:
29    
# Line 31  class Shape: Line 32  class Shape:
32      def __init__(self, points):      def __init__(self, points):
33          self.points = points          self.points = points
34          #self.compute_bbox()          #self.compute_bbox()
35            self.bbox = None
36    
37      def compute_bbox(self):      def compute_bbox(self):
38            if self.bbox is not None:
39                return self.bbox
40    
41          xs = []          xs = []
42          ys = []          ys = []
43          for x, y in self.points:          for x, y in self.points:
# Line 43  class Shape: Line 48  class Shape:
48          self.urx = max(xs)          self.urx = max(xs)
49          self.ury = max(ys)          self.ury = max(ys)
50    
51            self.bbox = (self.llx, self.lly, self.urx, self.ury)
52    
53            return self.bbox
54    
55      def Points(self):      def Points(self):
56          return self.points          return self.points
57    
# Line 66  class BaseLayer(TitledObject, Modifiable Line 75  class BaseLayer(TitledObject, Modifiable
75    
76      """Base class for the layers."""      """Base class for the layers."""
77    
78      def __init__(self, title, visible = True):      def __init__(self, title, visible = True, projection = None):
79          """Initialize the layer.          """Initialize the layer.
80    
81          title -- the title          title -- the title
# Line 75  class BaseLayer(TitledObject, Modifiable Line 84  class BaseLayer(TitledObject, Modifiable
84          TitledObject.__init__(self, title)          TitledObject.__init__(self, title)
85          Modifiable.__init__(self)          Modifiable.__init__(self)
86          self.visible = visible          self.visible = visible
87            self.projection = projection
88    
89      def Visible(self):      def Visible(self):
90          """Return true if layer is visible"""          """Return true if layer is visible"""
# Line 85  class BaseLayer(TitledObject, Modifiable Line 95  class BaseLayer(TitledObject, Modifiable
95          self.visible = visible          self.visible = visible
96          self.issue(LAYER_VISIBILITY_CHANGED, self)          self.issue(LAYER_VISIBILITY_CHANGED, self)
97    
98        def HasClassification(self):
99            """Determine if this layer support classifications."""
100            return False
101    
102        def GetProjection(self):
103            """Return the layer's projection."""
104            return self.projection
105    
106        def SetProjection(self, projection):
107            """Set the layer's projection"""
108            self.projection = projection
109            self.changed(LAYER_PROJECTION_CHANGED, self)
110    
111  class Layer(BaseLayer):  class Layer(BaseLayer):
112    
# Line 122  class Layer(BaseLayer): Line 144  class Layer(BaseLayer):
144    
145          colors are expected to be instances of Color class          colors are expected to be instances of Color class
146          """          """
147          BaseLayer.__init__(self, title, visible = visible)          BaseLayer.__init__(self, title,
148                                     visible = visible,
149          self.projection = projection                                   projection = projection)
150    
151          #          #
152          # this is really important so that when the classification class          # this is really important so that when the classification class
# Line 144  class Layer(BaseLayer): Line 166  class Layer(BaseLayer):
166    
167          self.UnsetModified()          self.UnsetModified()
168    
169        def __getattr__(self, attr):
170            """Access to some attributes for backwards compatibility
171    
172            The attributes implemented here are now held by the shapestore
173            if at all. For backwards compatibility pretend that they are
174            still there but issue a DeprecationWarning when they are
175            accessed.
176            """
177            if attr in ("table", "shapetable"):
178                value = self.store.Table()
179            elif attr == "shapefile":
180                value = self.store.Shapefile()
181            elif attr == "filename":
182                value = self.store.FileName()
183            else:
184                raise AttributeError, attr
185            warnings.warn("The Layer attribute %r is deprecated."
186                          " It's value can be accessed through the shapestore"
187                          % attr, DeprecationWarning, stacklevel = 2)
188            return value
189    
190      def SetShapeStore(self, store):      def SetShapeStore(self, store):
191          self.store = store          self.store = store
192          self.shapefile = self.store.Shapefile()          shapefile = self.store.Shapefile()
         self.shapetable = self.store.Table()  
         self.filename = self.store.filename  
         self.table = self.shapetable  
193    
194          numshapes, shapetype, mins, maxs = self.shapefile.info()          numshapes, shapetype, mins, maxs = shapefile.info()
195          self.numshapes = numshapes          self.numshapes = numshapes
196          self.shapetype = shapelib_shapetypes[shapetype]          self.shapetype = shapelib_shapetypes[shapetype]
197    
# Line 172  class Layer(BaseLayer): Line 211  class Layer(BaseLayer):
211          else:          else:
212              maxdepth = int(ceil(log(self.numshapes / 4.0) / log(4)))              maxdepth = int(ceil(log(self.numshapes / 4.0) / log(4)))
213    
214          self.shapetree = shptree.SHPTree(self.shapefile.cobject(), 2,          self.shapetree = shptree.SHPTree(shapefile.cobject(), 2,
215                                           maxdepth)                                           maxdepth)
216            # Set the classification to None if there is a classification
217            # and the new shapestore doesn't have a table with a suitable
218            # column, i.e one with the same name and type as before
219            # FIXME: Maybe we should keep it the same if the type is
220            # compatible enough such as FIELDTYPE_DOUBLE and FIELDTYPE_INT
221          if self.__classification is not None:          if self.__classification is not None:
222              fieldname = self.__classification.GetField()              fieldname = self.__classification.GetField()
223              if not self.store.Table().field_info_by_name(fieldname):              fieldtype = self.__classification.GetFieldType()
224                table = self.store.Table()
225                if (fieldname is not None
226                    and (not table.HasColumn(fieldname)
227                         or table.Column(fieldname).type != fieldtype)):
228                  self.SetClassification(None)                  self.SetClassification(None)
229          self.changed(LAYER_CHANGED, self)          self.changed(LAYER_SHAPESTORE_REPLACED, self)
230    
231      def ShapeStore(self):      def ShapeStore(self):
232          return self.store          return self.store
# Line 209  class Layer(BaseLayer): Line 257  class Layer(BaseLayer):
257          else:          else:
258              return None              return None
259    
260      def GetFieldType(self, fieldName):      def ShapesBoundingBox(self, shapes):
261          info = self.table.field_info_by_name(fieldName)          """Return a bounding box in lat/long coordinates for the given
262          if info is not None:          list of shape ids.
263              return info[0]  
264            If shapes is None or empty, return None.
265            """
266    
267            if shapes is None or len(shapes) == 0: return None
268    
269            llx = []
270            lly = []
271            urx = []
272            ury = []
273    
274            if self.projection is not None:
275                inverse = lambda x, y: self.projection.Inverse(x, y)
276          else:          else:
277              return None              inverse = lambda x, y: (x, y)
278    
279            for id in shapes:
280                left, bottom, right, top = self.Shape(id).compute_bbox()
281    
282                left, bottom = inverse(left, bottom)
283                right, top   = inverse(right, top)
284    
285                llx.append(left)
286                lly.append(bottom)
287                urx.append(right)
288                ury.append(top)
289    
290            return (min(llx), min(lly), max(urx), max(ury))
291    
292        def GetFieldType(self, fieldName):
293            table = self.store.Table()
294            if table.HasColumn(fieldName):
295                return table.Column(fieldName).type
296            return None
297    
298      def NumShapes(self):      def NumShapes(self):
299          """Return the number of shapes in the layer"""          """Return the number of shapes in the layer"""
# Line 228  class Layer(BaseLayer): Line 307  class Layer(BaseLayer):
307    
308      def Shape(self, index):      def Shape(self, index):
309          """Return the shape with index index"""          """Return the shape with index index"""
310          shape = self.shapefile.read_object(index)          shape = self.store.Shapefile().read_object(index)
311    
312          if self.shapetype == SHAPETYPE_POINT:          if self.shapetype == SHAPETYPE_POINT:
313              points = shape.vertices()              points = shape.vertices()
# Line 244  class Layer(BaseLayer): Line 323  class Layer(BaseLayer):
323      def ShapesInRegion(self, box):      def ShapesInRegion(self, box):
324          """Return the ids of the shapes that overlap the box.          """Return the ids of the shapes that overlap the box.
325    
326          Box is a tuple (left, bottom, right, top) in the coordinate          Box is a tuple (left, bottom, right, top) in unprojected coordinates.
         system used by the layer's shapefile.  
327          """          """
328          left, bottom, right, top = box          left, bottom, right, top = box
         return self.shapetree.find_shapes((left, bottom), (right, top))  
329    
330      def GetProjection(self):          if self.projection is not None:
331          return self.projection              left,  bottom = self.projection.Forward(left, bottom)
332                right, top    = self.projection.Forward(right, top)
333    
334      def SetProjection(self, projection):          return self.shapetree.find_shapes((left, bottom), (right, top))
335          """Set the layer's projection"""  
336          self.projection = projection      def HasClassification(self):
337          self.changed(LAYER_PROJECTION_CHANGED, self)          return True
338    
339      def GetClassification(self):      def GetClassification(self):
340          return self.__classification          return self.__classification
# Line 295  class Layer(BaseLayer): Line 373  class Layer(BaseLayer):
373      def TreeInfo(self):      def TreeInfo(self):
374          items = []          items = []
375    
376            if hasattr(self, 'filename'):
377                items.append(_("Filename: %s") % self.filename)
378    
379          if self.Visible():          if self.Visible():
380              items.append(_("Shown"))              items.append(_("Shown"))
381          else:          else:
# Line 317  class Layer(BaseLayer): Line 398  class Layer(BaseLayer):
398          return (_("Layer '%s'") % self.Title(), items)          return (_("Layer '%s'") % self.Title(), items)
399    
400    
401    if resource.has_gdal_support():
402        import gdal
403        from gdalconst import GA_ReadOnly
404    
405    class RasterLayer(BaseLayer):
406    
407        def __init__(self, title, filename, projection = None, visible = True):
408            """Initialize the Raster Layer.
409    
410            title -- title for the layer.
411    
412            filename -- file name of the source image.
413    
414            projection -- Projection object describing the projection which
415                          the source image is in.
416    
417            visible -- True is the layer should initially be visible.
418    
419            Throws IOError if the filename is invalid or points to a file that
420            is not in a format GDAL can use.
421            """
422    
423            BaseLayer.__init__(self, title, visible = visible)
424    
425            self.projection = projection
426            self.filename = filename
427    
428            self.bbox = -1
429    
430            if resource.has_gdal_support():
431                #
432                # temporarily open the file so that GDAL can test if it's valid.
433                #
434                dataset = gdal.Open(self.filename, GA_ReadOnly)
435    
436                if dataset is None:
437                    raise IOError()
438    
439            self.UnsetModified()
440    
441        def BoundingBox(self):
442            """Return the layer's bounding box in the intrinsic coordinate system.
443    
444            If the layer has no shapes, return None.
445            """
446            if not resource.has_gdal_support():
447                return None
448    
449            if self.bbox == -1:
450                dataset = gdal.Open(self.filename, GA_ReadOnly)
451                if dataset is None:
452                    self.bbox = None
453                else:
454                    geotransform = dataset.GetGeoTransform()
455                    if geotransform is None:
456                        return None
457    
458                    x = 0
459                    y = dataset.RasterYSize
460                    left = geotransform[0] +        \
461                           geotransform[1] * x +    \
462                           geotransform[2] * y
463    
464                    bottom = geotransform[3] +      \
465                             geotransform[4] * x +  \
466                             geotransform[5] * y
467    
468                    x = dataset.RasterXSize
469                    y = 0
470                    right = geotransform[0] +       \
471                            geotransform[1] * x +   \
472                            geotransform[2] * y
473    
474                    top = geotransform[3] +         \
475                          geotransform[4] * x +     \
476                          geotransform[5] * y
477    
478                    self.bbox = (left, bottom, right, top)
479    
480            return self.bbox
481    
482        def LatLongBoundingBox(self):
483            bbox = self.BoundingBox()
484            if bbox is None:
485                return None
486    
487            llx, lly, urx, ury = bbox
488            if self.projection is not None:
489                llx, lly = self.projection.Inverse(llx, lly)
490                urx, ury = self.projection.Inverse(urx, ury)
491    
492            return llx, lly, urx, ury
493    
494        def GetImageFilename(self):
495            return self.filename
496    
497        def TreeInfo(self):
498            items = []
499    
500            if self.Visible():
501                items.append(_("Shown"))
502            else:
503                items.append(_("Hidden"))
504            items.append(_("Shapes: %d") % self.NumShapes())
505    
506            bbox = self.LatLongBoundingBox()
507            if bbox is not None:
508                items.append(_("Extent (lat-lon): (%g, %g, %g, %g)") % bbox)
509            else:
510                items.append(_("Extent (lat-lon):"))
511    
512            if self.projection and len(self.projection.params) > 0:
513                items.append((_("Projection"),
514                            [str(param) for param in self.projection.params]))
515    
516            return (_("Layer '%s'") % self.Title(), items)
517    
Legend:
Removed from v.771  
changed lines
  Added in v.1219
[email protected]
 ViewVC Help
Powered by ViewVC 1.1.26