/[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 386 by jonathan, Mon Feb 3 11:44:27 2003 UTC revision 610 by jonathan, Fri Apr 4 13:56:59 2003 UTC
# Line 1  Line 1 
1  # Copyright (c) 2001, 2002 by Intevation GmbH  # Copyright (c) 2001, 2002 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 15  from Thuban import _ Line 16  from Thuban import _
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_LEGEND_CHANGED, \
19       LAYER_VISIBILITY_CHANGED       LAYER_VISIBILITY_CHANGED, LAYER_CHANGED
20    
21  from color import Color  from color import Color
 # Some predefined colors for internal use  
 _black = Color(0, 0, 0)  
22    
23  from classification import Classification  import classification
24    
25  from table import Table  from table import Table
26    
# Line 108  class Layer(BaseLayer): Line 107  class Layer(BaseLayer):
107      """      """
108    
109      def __init__(self, title, filename, projection = None,      def __init__(self, title, filename, projection = None,
110                   fill = None, stroke = _black, stroke_width = 1, visible = 1):                   fill = Color.Transparent,
111                     stroke = Color.Black,
112                     lineWidth = 1,
113                     visible = 1):
114          """Initialize the layer.          """Initialize the layer.
115    
116          title -- the title          title -- the title
# Line 116  class Layer(BaseLayer): Line 118  class Layer(BaseLayer):
118          projection -- the projection object. Its Inverse method is          projection -- the projection object. Its Inverse method is
119                 assumed to map the layer's coordinates to lat/long                 assumed to map the layer's coordinates to lat/long
120                 coordinates                 coordinates
121          fill -- the fill color or None if the shapes are not filled          fill -- the fill color or Color.Transparent if the shapes are
122          stroke -- the stroke color or None if the shapes are not stroked                  not filled
123            stroke -- the stroke color or Color.Transparent if the shapes
124                    are not stroked
125          visible -- boolean. If true the layer is visible.          visible -- boolean. If true the layer is visible.
126    
127          colors are expected to be instances of Color class          colors are expected to be instances of Color class
# Line 131  class Layer(BaseLayer): Line 135  class Layer(BaseLayer):
135          self.filename = os.path.abspath(filename)          self.filename = os.path.abspath(filename)
136    
137          self.projection = projection          self.projection = projection
         self.fill = fill  
         self.stroke = stroke  
         self.stroke_width = stroke_width  
138          self.shapefile = None          self.shapefile = None
139          self.shapetree = None          self.shapetree = None
140          self.open_shapefile()          self.open_shapefile()
# Line 143  class Layer(BaseLayer): Line 144  class Layer(BaseLayer):
144          self.shapetable = Table(filename)          self.shapetable = Table(filename)
145          self.table = self.shapetable          self.table = self.shapetable
146    
147          self.classification = Classification()          #
148          self.classification.setNull(          # this is really important so that when the classification class
149              {'stroke':stroke, 'stroke_width':stroke_width, 'fill':fill})          # tries to set its parent layer the variable will exist
150            #
151            self.__classification = None
152            self.__setClassLock = False
153    
154    
155            self.SetClassification(None)
156    
157            self.__classification.SetDefaultLineColor(stroke)
158            self.__classification.SetDefaultLineWidth(lineWidth)
159            self.__classification.SetDefaultFill(fill)
160            self.__classification.SetLayer(self)
161    
162            self.UnsetModified()
163    
164      def open_shapefile(self):      def open_shapefile(self):
165          if self.shapefile is None:          if self.shapefile is None:
# Line 179  class Layer(BaseLayer): Line 193  class Layer(BaseLayer):
193              self.shapefile.close()              self.shapefile.close()
194              self.shapefile = None              self.shapefile = None
195              self.shapetree = None              self.shapetree = None
196            self.SetClassification(None)
197          self.table.Destroy()          self.table.Destroy()
198    
199      def BoundingBox(self):      def BoundingBox(self):
# Line 206  class Layer(BaseLayer): Line 221  class Layer(BaseLayer):
221          else:          else:
222              return None              return None
223    
224        def GetFieldType(self, fieldName):
225            self.open_shapefile()
226            info = self.table.field_info_by_name(fieldName)
227            if info is not None:
228                return info[0]
229            else:
230                return None
231    
232      def NumShapes(self):      def NumShapes(self):
233          """Return the number of shapes in the layer"""          """Return the number of shapes in the layer"""
234          self.open_shapefile()          self.open_shapefile()
# Line 248  class Layer(BaseLayer): Line 271  class Layer(BaseLayer):
271          self.projection = projection          self.projection = projection
272          self.changed(LAYER_PROJECTION_CHANGED, self)          self.changed(LAYER_PROJECTION_CHANGED, self)
273    
274      def SetFill(self, fill):      def GetClassification(self):
275          """Set the layer's fill color. None means the shapes are not filled"""          return self.__classification
276          null = self.classification.getNull()  
277          null['fill'] = fill      def SetClassification(self, clazz):
278          self.classification.setNull(null)          """Set the classification to 'clazz'
279          self.changed(LAYER_LEGEND_CHANGED, self)  
280            If 'clazz' is None a default classification is created
281      def SetStroke(self, stroke):          """
282          """Set the layer's stroke color. None means the shapes are not  
283          stroked."""          # prevent infinite recursion when calling SetLayer()
284          null = self.classification.getNull()          if self.__setClassLock: return
         null['stroke'] = stroke  
         self.classification.setNull(null)  
         self.changed(LAYER_LEGEND_CHANGED, self)  
   
     def SetStrokeWidth(self, width):  
         """Set the layer's stroke width."""  
         null = self.classification.getNull()  
         null['stroke_width'] = width  
         self.classification.setNull(null)  
         self.changed(LAYER_LEGEND_CHANGED, self)  
285    
286            self.__setClassLock = True
287    
288            if clazz is None:
289                if self.__classification is not None:
290                    self.__classification.SetLayer(None)
291                self.__classification = classification.Classification()
292            else:
293                self.__classification = clazz
294                try:
295                    self.__classification.SetLayer(self)
296                except ValueError:
297                    self.__setClassLock = False
298                    raise ValueError
299    
300            self.changed(LAYER_CHANGED, self)
301    
302            self.__setClassLock = False
303    
304        def ClassChanged(self):
305            """Called from the classification object when it has changed."""
306            self.changed(LAYER_CHANGED, self)
307    
308      def TreeInfo(self):      def TreeInfo(self):
309          items = []          items = []
310    
# Line 286  class Layer(BaseLayer): Line 321  class Layer(BaseLayer):
321              items.append(_("Extent (lat-lon):"))              items.append(_("Extent (lat-lon):"))
322          items.append(_("Shapetype: %s") % shapetype_names[self.ShapeType()])          items.append(_("Shapetype: %s") % shapetype_names[self.ShapeType()])
323    
324          def color_string(color):          items.append(self.__classification)
             if color is None:  
                 return "None"  
             return "(%.3f, %.3f, %.3f)" % (color.red, color.green, color.blue)  
   
         # layers will always have a classification with at least a NULL data set  
   
         #items.append((_("Fill: %s") % color_string(self.fill), self.fill))  
         #items.append((_("Outline: %s") % color_string(self.stroke), self.stroke))  
   
         items.append(self.classification)  
325    
326          return (_("Layer '%s'") % self.Title(), items)          return (_("Layer '%s'") % self.Title(), items)
327    
Legend:
Removed from v.386  
changed lines
  Added in v.610
[email protected]
 ViewVC Help
Powered by ViewVC 1.1.26