/[thuban]/branches/WIP-pyshapelib-bramz/Thuban/Model/layer.py
ViewVC logotype

Diff of /branches/WIP-pyshapelib-bramz/Thuban/Model/layer.py

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

revision 2228 by bh, Thu Jun 3 15:17:54 2004 UTC revision 2587 by jonathan, Wed Mar 23 15:30:27 2005 UTC
# Line 2  Line 2 
2  # Authors:  # Authors:
3  # Bernhard Herzog <[email protected]>  # Bernhard Herzog <[email protected]>
4  # Jonathan Coles <[email protected]>  # Jonathan Coles <[email protected]>
5    # Silke Reimer <[email protected]>
6  #  #
7  # This program is free software under the GPL (>=v2)  # This program is free software under the GPL (>=v2)
8  # Read the file COPYING coming with Thuban for details.  # Read the file COPYING coming with Thuban for details.
# Line 24  from data import SHAPETYPE_POLYGON, SHAP Line 25  from data import SHAPETYPE_POLYGON, SHAP
25    
26  import resource  import resource
27    
28    from color import Color
29    
30  shapetype_names = {SHAPETYPE_POINT: "Point",  shapetype_names = {SHAPETYPE_POINT: "Point",
31                     SHAPETYPE_ARC: "Arc",                     SHAPETYPE_ARC: "Arc",
# Line 54  class BaseLayer(TitledObject, Modifiable Line 56  class BaseLayer(TitledObject, Modifiable
56          self.issue(LAYER_VISIBILITY_CHANGED, self)          self.issue(LAYER_VISIBILITY_CHANGED, self)
57    
58      def HasClassification(self):      def HasClassification(self):
59          """Determine if this layer support classifications."""          """Determine if this layer supports classifications."""
60          return False          return False
61    
62      def HasShapes(self):      def HasShapes(self):
# Line 66  class BaseLayer(TitledObject, Modifiable Line 68  class BaseLayer(TitledObject, Modifiable
68          return self.projection          return self.projection
69    
70      def SetProjection(self, projection):      def SetProjection(self, projection):
71          """Set the layer's projection"""          """Set the layer's projection."""
72          self.projection = projection          self.projection = projection
73          self.changed(LAYER_PROJECTION_CHANGED, self)          self.changed(LAYER_PROJECTION_CHANGED, self)
74    
75        def Type(self):
76            return "Unknown"
77    
78  class Layer(BaseLayer):  class Layer(BaseLayer):
79    
80      """Represent the information of one geodata file (currently a shapefile)      """Represent the information of one geodata file (currently a shapefile)
# Line 170  class Layer(BaseLayer): Line 175  class Layer(BaseLayer):
175              bbox = self.projection.InverseBBox(bbox)              bbox = self.projection.InverseBBox(bbox)
176          return bbox          return bbox
177    
178        def Type(self):
179            return self.ShapeType();
180    
181      def ShapesBoundingBox(self, shapes):      def ShapesBoundingBox(self, shapes):
182          """Return a bounding box in lat/long coordinates for the given          """Return a bounding box in lat/long coordinates for the given
183          list of shape ids.          list of shape ids.
# Line 226  class Layer(BaseLayer): Line 234  class Layer(BaseLayer):
234          form (minx, miny, maxx, maxy) in unprojected coordinates.          form (minx, miny, maxx, maxy) in unprojected coordinates.
235          """          """
236          if self.projection is not None:          if self.projection is not None:
237              bbox = self.projection.ForwardBBox(bbox)              # Ensure that region lies within the layer's bounding box
238                # Otherwise projection of the region would lead to incorrect
239                # values.
240                clipbbox = self.ClipBoundingBox(bbox)
241                bbox = self.projection.ForwardBBox(clipbbox)
242          return self.store.ShapesInRegion(bbox)          return self.store.ShapesInRegion(bbox)
243    
244      def GetClassificationColumn(self):      def GetClassificationColumn(self):
# Line 302  class Layer(BaseLayer): Line 314  class Layer(BaseLayer):
314    
315          return (_("Layer '%s'") % self.Title(), items)          return (_("Layer '%s'") % self.Title(), items)
316    
317        def ClipBoundingBox(self, bbox):
318            """ Clip bbox to layer's bounding box.
319    
320            Returns that part of bbox that lies within the layers bounding box.
321            If bbox is completely outside of the layers bounding box, bbox is
322            returned.  It is assumed that bbox has sensible values, i.e. bminx
323            < bmaxx and bminy < bmaxy.
324            """
325            bminx, bminy, bmaxx, bmaxy = bbox
326            lminx, lminy, lmaxx, lmaxy = self.LatLongBoundingBox()
327            if bminx > lmaxx or bmaxx < lminx:
328                left, right = bminx, bmaxx
329            else:
330                left = max(lminx, bminx)
331                right = min(lmaxx, bmaxx)
332            if bminy > lmaxy or bmaxy < lminy:
333                bottom, top = bminy, bmaxy
334            else:
335                bottom = max(lminy, bminy)
336                top = min(lmaxy, bmaxy)
337            
338            return (left, bottom, right, top)
339    
340    
341  if resource.has_gdal_support():  if resource.has_gdal_support():
342      import gdal      import gdal
# Line 309  if resource.has_gdal_support(): Line 344  if resource.has_gdal_support():
344    
345  class RasterLayer(BaseLayer):  class RasterLayer(BaseLayer):
346    
347        MASK_NONE  = 0
348        MASK_BIT   = 1
349        MASK_ALPHA = 2
350    
351      def __init__(self, title, filename, projection = None, visible = True):      def __init__(self, title, filename, projection = None, visible = True):
352          """Initialize the Raster Layer.          """Initialize the Raster Layer.
353    
# Line 332  class RasterLayer(BaseLayer): Line 371  class RasterLayer(BaseLayer):
371    
372          self.bbox = -1          self.bbox = -1
373    
374            self.mask_type = self.MASK_BIT
375            self.opacity = 1
376    
377            self.image_info = None
378    
379          if resource.has_gdal_support():          if resource.has_gdal_support():
380              #              #
381              # temporarily open the file so that GDAL can test if it's valid.              # temporarily open the file so that GDAL can test if it's valid.
# Line 341  class RasterLayer(BaseLayer): Line 385  class RasterLayer(BaseLayer):
385              if dataset is None:              if dataset is None:
386                  raise IOError()                  raise IOError()
387    
388                #
389                # while we have the file, extract some basic information
390                # that we can display later
391                #
392                self.image_info = {}
393    
394                self.image_info["nBands"] = dataset.RasterCount
395                self.image_info["Size"] = (dataset.RasterXSize, dataset.RasterYSize)
396                self.image_info["Driver"] = dataset.GetDriver().ShortName
397    
398                # store some information about the individual bands
399                # [min_value, max_value]
400                a = self.image_info["BandData"] = []
401    
402                for i in range(1, dataset.RasterCount+1):
403                    band = dataset.GetRasterBand(i)
404                    a.append(band.ComputeRasterMinMax())
405    
406          self.UnsetModified()          self.UnsetModified()
407    
408      def BoundingBox(self):      def BoundingBox(self):
# Line 395  class RasterLayer(BaseLayer): Line 457  class RasterLayer(BaseLayer):
457    
458          return bbox          return bbox
459    
460        def Type(self):
461            return "Image"
462    
463      def GetImageFilename(self):      def GetImageFilename(self):
464          return self.filename          return self.filename
465    
466        def MaskType(self):
467            """Return True if the mask should be used when rendering the layer."""
468            return self.mask_type
469    
470        def SetMaskType(self, type):
471            """Set the type of mask to use.
472    
473            type can be one of MASK_NONE, MASK_BIT, MASK_ALPHA
474    
475            If the state changes, a LAYER_CHANGED message is sent.
476            """
477            if type not in (self.MASK_NONE, self.MASK_BIT, self.MASK_ALPHA):
478                raise ValueError("type is invalid")
479    
480            if type != self.mask_type:
481                self.mask_type = type
482                self.changed(LAYER_CHANGED, self)
483    
484        def Opacity(self):
485            """Return the level of opacity used in alpha blending.
486            """
487            return self.opacity
488    
489        def SetOpacity(self, op):
490            """Set the level of alpha opacity.
491    
492            0 <= op <= 1.
493    
494            The layer is fully opaque when op = 1.
495            """
496            if not (0 <= op <= 1):
497                raise ValueError("op out of range")
498    
499            if op != self.opacity:
500                self.opacity = op
501                self.changed(LAYER_CHANGED, self)
502    
503        def ImageInfo(self):
504            return self.image_info
505    
506      def TreeInfo(self):      def TreeInfo(self):
507          items = []          items = []
508    
Legend:
Removed from v.2228  
changed lines
  Added in v.2587
[email protected]
 ViewVC Help
Powered by ViewVC 1.1.26