/[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 725 by jonathan, Thu Apr 24 16:05:36 2003 UTC revision 1012 by jan, Fri May 23 09:13:58 2003 UTC
# Line 14  from Thuban import _ Line 14  from Thuban import _
14    
15  import shapelib, shptree  import shapelib, shptree
16    
17    import gdal
18    from gdalconst import GA_ReadOnly
19    
20  from messages import LAYER_PROJECTION_CHANGED, LAYER_VISIBILITY_CHANGED, \  from messages import LAYER_PROJECTION_CHANGED, LAYER_VISIBILITY_CHANGED, \
21       LAYER_CHANGED       LAYER_CHANGED
22    
# Line 31  class Shape: Line 34  class Shape:
34      def __init__(self, points):      def __init__(self, points):
35          self.points = points          self.points = points
36          #self.compute_bbox()          #self.compute_bbox()
37            self.bbox = None
38    
39      def compute_bbox(self):      def compute_bbox(self):
40            if self.bbox is not None:
41                return self.bbox
42    
43          xs = []          xs = []
44          ys = []          ys = []
45          for x, y in self.points:          for x, y in self.points:
# Line 43  class Shape: Line 50  class Shape:
50          self.urx = max(xs)          self.urx = max(xs)
51          self.ury = max(ys)          self.ury = max(ys)
52    
53            self.bbox = (self.llx, self.lly, self.urx, self.ury)
54    
55            return self.bbox
56    
57      def Points(self):      def Points(self):
58          return self.points          return self.points
59    
# Line 66  class BaseLayer(TitledObject, Modifiable Line 77  class BaseLayer(TitledObject, Modifiable
77    
78      """Base class for the layers."""      """Base class for the layers."""
79    
80      def __init__(self, title, visible = 1):      def __init__(self, title, visible = True, projection = None):
81          """Initialize the layer.          """Initialize the layer.
82    
83          title -- the title          title -- the title
# Line 75  class BaseLayer(TitledObject, Modifiable Line 86  class BaseLayer(TitledObject, Modifiable
86          TitledObject.__init__(self, title)          TitledObject.__init__(self, title)
87          Modifiable.__init__(self)          Modifiable.__init__(self)
88          self.visible = visible          self.visible = visible
89            self.projection = projection
90    
91      def Visible(self):      def Visible(self):
92          """Return true if layer is visible"""          """Return true if layer is visible"""
# Line 85  class BaseLayer(TitledObject, Modifiable Line 97  class BaseLayer(TitledObject, Modifiable
97          self.visible = visible          self.visible = visible
98          self.issue(LAYER_VISIBILITY_CHANGED, self)          self.issue(LAYER_VISIBILITY_CHANGED, self)
99    
100        def HasClassification(self):
101            """Determine if this layer support classifications."""
102            return False
103    
104        def GetProjection(self):
105            """Return the layer's projection."""
106            return self.projection
107    
108        def SetProjection(self, projection):
109            """Set the layer's projection"""
110            self.projection = projection
111            self.changed(LAYER_PROJECTION_CHANGED, self)
112    
113  class Layer(BaseLayer):  class Layer(BaseLayer):
114    
# Line 106  class Layer(BaseLayer): Line 130  class Layer(BaseLayer):
130                   fill = Color.Transparent,                   fill = Color.Transparent,
131                   stroke = Color.Black,                   stroke = Color.Black,
132                   lineWidth = 1,                   lineWidth = 1,
133                   visible = 1):                   visible = True):
134          """Initialize the layer.          """Initialize the layer.
135    
136          title -- the title          title -- the title
# Line 122  class Layer(BaseLayer): Line 146  class Layer(BaseLayer):
146    
147          colors are expected to be instances of Color class          colors are expected to be instances of Color class
148          """          """
149          BaseLayer.__init__(self, title, visible = visible)          BaseLayer.__init__(self, title,
150                                     visible = visible,
151          self.projection = projection                                   projection = projection)
152    
153          #          #
154          # this is really important so that when the classification class          # this is really important so that when the classification class
# Line 149  class Layer(BaseLayer): Line 173  class Layer(BaseLayer):
173          self.store = store          self.store = store
174          self.shapefile = self.store.Shapefile()          self.shapefile = self.store.Shapefile()
175          self.shapetable = self.store.Table()          self.shapetable = self.store.Table()
176          self.filename = self.store.filename          if hasattr(self.store, "FileName"):
177                self.filename = self.store.FileName()
178          self.table = self.shapetable          self.table = self.shapetable
179    
180          numshapes, shapetype, mins, maxs = self.shapefile.info()          numshapes, shapetype, mins, maxs = self.shapefile.info()
# Line 176  class Layer(BaseLayer): Line 201  class Layer(BaseLayer):
201                                           maxdepth)                                           maxdepth)
202          if self.__classification is not None:          if self.__classification is not None:
203              fieldname = self.__classification.GetField()              fieldname = self.__classification.GetField()
204              if not self.store.Table().field_info_by_name(fieldname):              if fieldname is not None and \
205                   not self.store.Table().HasColumn(fieldname):
206                  self.SetClassification(None)                  self.SetClassification(None)
207          self.changed(LAYER_CHANGED, self)          self.changed(LAYER_CHANGED, self)
208    
# Line 209  class Layer(BaseLayer): Line 235  class Layer(BaseLayer):
235          else:          else:
236              return None              return None
237    
238      def GetFieldType(self, fieldName):      def ShapesBoundingBox(self, shapes):
239          info = self.table.field_info_by_name(fieldName)          """Return a bounding box in lat/long coordinates for the given
240          if info is not None:          list of shape ids.
241              return info[0]  
242            If shapes is None or empty, return None.
243            """
244    
245            if shapes is None or len(shapes) == 0: return None
246    
247            llx = []
248            lly = []
249            urx = []
250            ury = []
251    
252            if self.projection is not None:
253                inverse = lambda x, y: self.projection.Inverse(x, y)
254          else:          else:
255              return None              inverse = lambda x, y: (x, y)
256    
257            for id in shapes:
258                left, bottom, right, top = self.Shape(id).compute_bbox()
259    
260                left, bottom = inverse(left, bottom)
261                right, top   = inverse(right, top)
262    
263                llx.append(left)
264                lly.append(bottom)
265                urx.append(right)
266                ury.append(top)
267    
268            return (min(llx), min(lly), max(urx), max(ury))
269    
270        def GetFieldType(self, fieldName):
271            if self.table.HasColumn(fieldName):
272                return self.table.Column(fieldName).type
273            return None
274    
275      def NumShapes(self):      def NumShapes(self):
276          """Return the number of shapes in the layer"""          """Return the number of shapes in the layer"""
# Line 244  class Layer(BaseLayer): Line 300  class Layer(BaseLayer):
300      def ShapesInRegion(self, box):      def ShapesInRegion(self, box):
301          """Return the ids of the shapes that overlap the box.          """Return the ids of the shapes that overlap the box.
302    
303          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.  
304          """          """
305          left, bottom, right, top = box          left, bottom, right, top = box
         return self.shapetree.find_shapes((left, bottom), (right, top))  
306    
307      def GetProjection(self):          if self.projection is not None:
308          return self.projection              left,  bottom = self.projection.Forward(left, bottom)
309                right, top    = self.projection.Forward(right, top)
310    
311      def SetProjection(self, projection):          return self.shapetree.find_shapes((left, bottom), (right, top))
312          """Set the layer's projection"""  
313          self.projection = projection      def HasClassification(self):
314          self.changed(LAYER_PROJECTION_CHANGED, self)          return True
315    
316      def GetClassification(self):      def GetClassification(self):
317          return self.__classification          return self.__classification
# Line 295  class Layer(BaseLayer): Line 350  class Layer(BaseLayer):
350      def TreeInfo(self):      def TreeInfo(self):
351          items = []          items = []
352    
353            items.append(_("Filename: %s") % self.filename)
354    
355          if self.Visible():          if self.Visible():
356              items.append(_("Shown"))              items.append(_("Shown"))
357          else:          else:
# Line 308  class Layer(BaseLayer): Line 365  class Layer(BaseLayer):
365              items.append(_("Extent (lat-lon):"))              items.append(_("Extent (lat-lon):"))
366          items.append(_("Shapetype: %s") % shapetype_names[self.ShapeType()])          items.append(_("Shapetype: %s") % shapetype_names[self.ShapeType()])
367    
368            if self.projection and len(self.projection.params) > 0:
369                items.append((_("Projection"),
370                            [str(param) for param in self.projection.params]))
371    
372          items.append(self.__classification)          items.append(self.__classification)
373    
374          return (_("Layer '%s'") % self.Title(), items)          return (_("Layer '%s'") % self.Title(), items)
375    
376    
377    class RasterLayer(BaseLayer):
378    
379        def __init__(self, title, filename, projection = None, visible = True):
380            """Initialize the Raster Layer.
381    
382            title -- title for the layer.
383    
384            filename -- file name of the source image.
385    
386            projection -- Projection object describing the projection which
387                          the source image is in.
388    
389            visible -- True is the layer should initially be visible.
390    
391            Throws IOError if the filename is invalid or points to a file that
392            is not in a format GDAL can use.
393            """
394    
395            BaseLayer.__init__(self, title, visible = visible)
396    
397            self.projection = projection
398            self.filename = filename
399    
400            self.bbox = -1
401    
402            #
403            # temporarily open the file so that GDAL can test if it's valid.
404            #
405            dataset = gdal.Open(self.filename, GA_ReadOnly)
406    
407            if dataset is None:
408                raise IOError()
409    
410            self.UnsetModified()
411    
412        def BoundingBox(self):
413            """Return the layer's bounding box in the intrinsic coordinate system.
414    
415            If the layer has no shapes, return None.
416            """
417            if self.bbox == -1:
418                dataset = gdal.Open(self.filename, GA_ReadOnly)
419                if dataset is None:
420                    self.bbox = None
421                else:
422                    geotransform = dataset.GetGeoTransform()
423                    if geotransform is None:
424                        return None
425    
426                    x = 0
427                    y = dataset.RasterYSize
428                    left = geotransform[0] +        \
429                           geotransform[1] * x +    \
430                           geotransform[2] * y
431    
432                    bottom = geotransform[3] +      \
433                             geotransform[4] * x +  \
434                             geotransform[5] * y
435    
436                    x = dataset.RasterXSize
437                    y = 0
438                    right = geotransform[0] +       \
439                            geotransform[1] * x +   \
440                            geotransform[2] * y
441    
442                    top = geotransform[3] +         \
443                          geotransform[4] * x +     \
444                          geotransform[5] * y
445    
446                    self.bbox = (left, bottom, right, top)
447    
448            return self.bbox
449    
450        def LatLongBoundingBox(self):
451            bbox = self.BoundingBox()
452            if bbox is None:
453                return None
454    
455            llx, lly, urx, ury = bbox
456            if self.projection is not None:
457                llx, lly = self.projection.Inverse(llx, lly)
458                urx, ury = self.projection.Inverse(urx, ury)
459    
460            return llx, lly, urx, ury
461    
462        def GetImageFilename(self):
463            return self.filename
464    
465        def TreeInfo(self):
466            items = []
467    
468            if self.Visible():
469                items.append(_("Shown"))
470            else:
471                items.append(_("Hidden"))
472            items.append(_("Shapes: %d") % self.NumShapes())
473    
474            bbox = self.LatLongBoundingBox()
475            if bbox is not None:
476                items.append(_("Extent (lat-lon): (%g, %g, %g, %g)") % bbox)
477            else:
478                items.append(_("Extent (lat-lon):"))
479    
480            if self.projection and len(self.projection.params) > 0:
481                items.append((_("Projection"),
482                            [str(param) for param in self.projection.params]))
483    
484            return (_("Layer '%s'") % self.Title(), items)
485    
Legend:
Removed from v.725  
changed lines
  Added in v.1012
[email protected]
 ViewVC Help
Powered by ViewVC 1.1.26