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

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

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

revision 479 by jonathan, Thu Mar 6 16:46:07 2003 UTC revision 609 by jonathan, Fri Apr 4 13:55:59 2003 UTC
# Line 23  on the mapping algorithm. Line 23  on the mapping algorithm.
23  # fix for people using python2.1  # fix for people using python2.1
24  from __future__ import nested_scopes  from __future__ import nested_scopes
25    
26    import copy
27    
28  from types import *  from types import *
29    
30  from messages import LAYER_PROJECTION_CHANGED, LAYER_LEGEND_CHANGED, \  from messages import LAYER_PROJECTION_CHANGED, LAYER_LEGEND_CHANGED, \
# Line 57  class Classification: Line 59  class Classification:
59          self.field = None          self.field = None
60          self.fieldType = None          self.fieldType = None
61          self.groups = []          self.groups = []
         self.__sendMessages = False  
62    
63          self.__ToggleMessages(False)          self.__setLayerLock = False
64    
65          self.SetDefaultGroup(ClassGroupDefault())          self.SetDefaultGroup(ClassGroupDefault())
66    
67          self.SetLayer(layer)          self.SetLayer(layer)
68          self.SetField(field)          self.SetField(field)
69    
         self.__ToggleMessages(True)  
   
70      def __iter__(self):      def __iter__(self):
71          return ClassIterator(self.groups)          return ClassIterator(self.groups)
72    
73      def __ToggleMessages(self, on):      def __SendNotification(self):
74          self.__sendMessages = on          """Notify the layer that this class has changed."""
75            if self.layer is not None:
76      def __SendMessage(self, message):              self.layer.ClassChanged()
         """Send the message 'message' to the parent layer."""  
         if self.__sendMessages and self.layer is not None:  
             self.layer.changed(message, self.layer)  
77            
78      def SetField(self, field = None):      def SetField(self, field):
79          """Set the name of the data table field to use.          """Set the name of the data table field to use.
80                    
81             If there is no layer then the field type is set to None,             If there is no layer then the field type is set to None,
# Line 90  class Classification: Line 88  class Classification:
88          if field == "":          if field == "":
89              field = None              field = None
90    
         self.field = field  
91    
92          if self.layer is not None:          if field is None:
93              fieldType = self.layer.GetFieldType(field)              if self.layer is not None:
94                    self.fieldType = None
95          else:          else:
96              fieldType = None              if self.layer is not None:
97                    fieldType = self.layer.GetFieldType(field)
98                    if fieldType is None:
99                        raise ValueError("'%s' was not found in the layer's table."
100                                         % self.field)
101    
102                    #
103                    # unfortunately we cannot call SetFieldType() because it
104                    # requires the layer to be None
105                    #
106                    self.fieldType = fieldType
107                    #self.SetFieldType(fieldType)
108    
109          self.SetFieldType(fieldType)          self.field = field
   
         # XXX: if fieldType comes back None then field isn't in the table!  
110    
111          self.__SendMessage(LAYER_LEGEND_CHANGED)          self.__SendNotification()
112    
113      def GetField(self):      def GetField(self):
114          """Return the name of the field."""          """Return the name of the field."""
# Line 112  class Classification: Line 119  class Classification:
119          return self.fieldType          return self.fieldType
120    
121      def SetFieldType(self, type):      def SetFieldType(self, type):
122          self.fieldType = type          """Set the type of the field used by this classification.
123    
124      def SetLayer(self, layer):          A ValueError is raised if the owning layer is not None and
125          """Set the owning Layer of this classification."""          'type' is different from the current field type.
126            """
127    
128          if __debug__:          if type != self.fieldType:
129              if layer is not None:              if self.layer is not None:
130                  assert(isinstance(layer, Thuban.Model.layer.Layer))                  raise ValueError()
131                else:
132                    self.fieldType = type
133                    self.__SendNotification()
134    
135        def SetLayer(self, layer):
136            """Set the owning Layer of this classification.
137    
138               A ValueError exception will be thrown either the field or
139               field type mismatch the information in the layer's table.
140            """
141    
142          # prevent infinite recursion when calling SetClassification()          # prevent infinite recursion when calling SetClassification()
143          if self.layer is not None and layer == self.layer:          if self.__setLayerLock: return
             return  
144    
145          self.layer = layer          self.__setLayerLock = True
         self.SetField(self.GetField()) # XXX: this sync's the fieldType  
146    
147          if self.layer is not None:          if layer is None:
148              self.layer.SetClassification(self)              if self.layer is not None:
149                    l = self.layer
150                    self.layer = None
151                    l.SetClassification(None)
152            else:
153                assert isinstance(layer, Thuban.Model.layer.Layer)
154    
155                old_layer = self.layer
156    
157          #self.__SendMessage(LAYER_LEGEND_CHANGED)              self.layer = layer
158    
159                try:
160                    self.SetField(self.GetField()) # this sync's the fieldType
161                except ValueError:
162                    self.layer = old_layer
163                    self.__setLayerLock = False
164                    raise ValueError
165                else:
166                    self.layer.SetClassification(self)
167    
168            self.__setLayerLock = False
169    
170      def GetLayer(self):      def GetLayer(self):
171          """Return the parent layer."""          """Return the parent layer."""
# Line 143  class Classification: Line 177  class Classification:
177             group -- group that the value maps to.             group -- group that the value maps to.
178          """          """
179    
180          assert(isinstance(group, ClassGroupDefault))          assert isinstance(group, ClassGroupDefault)
181          self.AddGroup(group)          self.AddGroup(group)
182    
183      def GetDefaultGroup(self):      def GetDefaultGroup(self):
# Line 161  class Classification: Line 195  class Classification:
195    
196          fill -- a Color object.          fill -- a Color object.
197          """          """
198          assert(isinstance(fill, Color))          assert isinstance(fill, Color)
199          self.GetDefaultGroup().GetProperties().SetFill(fill)          self.GetDefaultGroup().GetProperties().SetFill(fill)
200          self.__SendMessage(LAYER_LEGEND_CHANGED)          self.__SendNotification()
201                    
202      def GetDefaultFill(self):      def GetDefaultFill(self):
203          """Return the default fill color."""          """Return the default fill color."""
# Line 174  class Classification: Line 208  class Classification:
208    
209          color -- a Color object.          color -- a Color object.
210          """          """
211          assert(isinstance(color, Color))          assert isinstance(color, Color)
212          self.GetDefaultGroup().GetProperties().SetLineColor(color)          self.GetDefaultGroup().GetProperties().SetLineColor(color)
213          self.__SendMessage(LAYER_LEGEND_CHANGED)          self.__SendNotification()
214                    
215      def GetDefaultLineColor(self):      def GetDefaultLineColor(self):
216          """Return the default line color."""          """Return the default line color."""
# Line 187  class Classification: Line 221  class Classification:
221    
222          lineWidth -- an integer > 0.          lineWidth -- an integer > 0.
223          """          """
224          assert(isinstance(lineWidth, IntType))          assert isinstance(lineWidth, IntType)
225          self.GetDefaultGroup().GetProperties().SetLineWidth(lineWidth)          self.GetDefaultGroup().GetProperties().SetLineWidth(lineWidth)
226          self.__SendMessage(LAYER_LEGEND_CHANGED)          self.__SendNotification()
227                    
228      def GetDefaultLineWidth(self):      def GetDefaultLineWidth(self):
229          """Return the default line width."""          """Return the default line width."""
# Line 201  class Classification: Line 235  class Classification:
235          item -- this must be a valid ClassGroup object          item -- this must be a valid ClassGroup object
236          """          """
237    
238          assert(isinstance(item, ClassGroup))          assert isinstance(item, ClassGroup)
239    
240          if len(self.groups) > 0 and isinstance(item, ClassGroupDefault):          if len(self.groups) > 0 and isinstance(item, ClassGroupDefault):
241              self.groups[0] = item              self.groups[0] = item
             #self.SetDefaultGroup(item)  
242          else:          else:
243              self.groups.append(item)              self.groups.append(item)
244    
245          self.__SendMessage(LAYER_LEGEND_CHANGED)          self.__SendNotification()
246    
247      def GetGroup(self, value):      def GetGroup(self, value):
248          """Return the associated group, or the default group.          """Return the associated group, or the default group.
# Line 244  class Classification: Line 277  class Classification:
277          items = []          items = []
278    
279          def build_color_item(text, color):          def build_color_item(text, color):
280              if color is Color.None:              if color is Color.Transparent:
281                  return ("%s: %s" % (text, _("None")), None)                  return ("%s: %s" % (text, _("None")), None)
282    
283              return ("%s: (%.3f, %.3f, %.3f)" %              return ("%s: (%.3f, %.3f, %.3f)" %
# Line 340  class ClassGroupProperties: Line 373  class ClassGroupProperties:
373          props -- a ClassGroupProperties object. The class is copied if          props -- a ClassGroupProperties object. The class is copied if
374                   prop is not None. Otherwise, a default set of properties                   prop is not None. Otherwise, a default set of properties
375                   is created such that: line color = Color.Black, line width = 1,                   is created such that: line color = Color.Black, line width = 1,
376                   and fill color = Color.None                   and fill color = Color.Transparent
377          """          """
378    
379          self.stroke = None          self.stroke = None
# Line 350  class ClassGroupProperties: Line 383  class ClassGroupProperties:
383          if props is not None:          if props is not None:
384              self.SetProperties(props)              self.SetProperties(props)
385          else:          else:
386              self.SetLineColor(Color.None)              self.SetLineColor(Color.Black)
387              self.SetLineWidth(1)              self.SetLineWidth(1)
388              self.SetFill(Color.None)              self.SetFill(Color.Transparent)
389    
390      def SetProperties(self, props):      def SetProperties(self, props):
391          """Set this class's properties to those in class props."""          """Set this class's properties to those in class props."""
392    
393          assert(isinstance(props, ClassGroupProperties))          assert isinstance(props, ClassGroupProperties)
394          self.SetLineColor(props.GetLineColor())          self.SetLineColor(props.GetLineColor())
395          self.SetLineWidth(props.GetLineWidth())          self.SetLineWidth(props.GetLineWidth())
396          self.SetFill(props.GetFill())          self.SetFill(props.GetFill())
# Line 372  class ClassGroupProperties: Line 405  class ClassGroupProperties:
405          color -- the color of the line. This must be a Color object.          color -- the color of the line. This must be a Color object.
406          """          """
407    
408          assert(isinstance(color, Color))          assert isinstance(color, Color)
409          self.stroke = color          self.stroke = color
410    
411      def GetLineWidth(self):      def GetLineWidth(self):
# Line 384  class ClassGroupProperties: Line 417  class ClassGroupProperties:
417    
418          lineWidth -- the new line width. This must be > 0.          lineWidth -- the new line width. This must be > 0.
419          """          """
420          assert(isinstance(lineWidth, IntType))          assert isinstance(lineWidth, IntType)
421          if (lineWidth < 1):          if (lineWidth < 1):
422              raise ValueError(_("lineWidth < 1"))              raise ValueError(_("lineWidth < 1"))
423    
# Line 400  class ClassGroupProperties: Line 433  class ClassGroupProperties:
433          fill -- the color of the fill. This must be a Color object.          fill -- the color of the fill. This must be a Color object.
434          """          """
435    
436          assert(isinstance(fill, Color))          assert isinstance(fill, Color)
437          self.fill = fill          self.fill = fill
438    
439      def __eq__(self, other):      def __eq__(self, other):
# Line 414  class ClassGroupProperties: Line 447  class ClassGroupProperties:
447      def __ne__(self, other):      def __ne__(self, other):
448          return not self.__eq__(other)          return not self.__eq__(other)
449    
450        def __copy__(self):
451            return ClassGroupProperties(self)
452    
453        def __deepcopy__(self):
454            return ClassGroupProperties(self)
455    
456  class ClassGroup:  class ClassGroup:
457      """A base class for all Groups within a Classification"""      """A base class for all Groups within a Classification"""
458    
# Line 437  class ClassGroup: Line 476  class ClassGroup:
476          label -- a string representing the Group's label. This must          label -- a string representing the Group's label. This must
477                   not be None.                   not be None.
478          """          """
479          assert(isinstance(label, StringType))          assert isinstance(label, StringType)
480          self.label = label          self.label = label
481    
482        def GetDisplayText(self):
483            assert False, "GetDisplay must be overridden by subclass!"
484            return ""
485    
486      def Matches(self, value):      def Matches(self, value):
487          """Determines if this Group is associated with the given value.          """Determines if this Group is associated with the given value.
488    
489          Returns False. This needs to be overridden by all subclasses.          Returns False. This needs to be overridden by all subclasses.
490          """          """
491            assert False, "GetMatches must be overridden by subclass!"
492          return False          return False
493    
494      def GetProperties(self):      def GetProperties(self):
# Line 452  class ClassGroup: Line 496  class ClassGroup:
496    
497          Returns None. This needs to be overridden by all subclasses.          Returns None. This needs to be overridden by all subclasses.
498          """          """
499            assert False, "GetProperties must be overridden by subclass!"
500          return None          return None
501    
502            
# Line 481  class ClassGroupSingleton(ClassGroup): Line 526  class ClassGroupSingleton(ClassGroup):
526                                     self.GetProperties(),                                     self.GetProperties(),
527                                     self.GetLabel())                                     self.GetLabel())
528    
529        def __deepcopy__(self, memo):
530            return ClassGroupSingleton(copy.copy(self.GetValue()),
531                                       copy.copy(self.GetProperties()),
532                                       copy.copy(self.GetLabel()))
533    
534      def GetValue(self):      def GetValue(self):
535          """Return the associated value."""          """Return the associated value."""
536          return self.value          return self.value
# Line 509  class ClassGroupSingleton(ClassGroup): Line 559  class ClassGroupSingleton(ClassGroup):
559          """          """
560    
561          if prop is None: prop = ClassGroupProperties()          if prop is None: prop = ClassGroupProperties()
562          assert(isinstance(prop, ClassGroupProperties))          assert isinstance(prop, ClassGroupProperties)
563          self.prop = prop          self.prop = prop
564    
565        def GetDisplayText(self):
566            label = self.GetLabel()
567    
568            if label != "": return label
569    
570            return str(self.GetValue())
571    
572      def __eq__(self, other):      def __eq__(self, other):
573          return isinstance(other, ClassGroupSingleton) \          return isinstance(other, ClassGroupSingleton) \
574              and self.GetProperties() == other.GetProperties() \              and self.GetProperties() == other.GetProperties() \
# Line 540  class ClassGroupDefault(ClassGroup): Line 597  class ClassGroupDefault(ClassGroup):
597      def __copy__(self):      def __copy__(self):
598          return ClassGroupDefault(self.GetProperties(), self.GetLabel())          return ClassGroupDefault(self.GetProperties(), self.GetLabel())
599    
600        def __deepcopy__(self, memo):
601            return ClassGroupDefault(copy.copy(self.GetProperties()),
602                                     copy.copy(self.GetLabel()))
603    
604      def Matches(self, value):      def Matches(self, value):
605          return True          return True
606    
# Line 555  class ClassGroupDefault(ClassGroup): Line 616  class ClassGroupDefault(ClassGroup):
616          """          """
617    
618          if prop is None: prop = ClassGroupProperties()          if prop is None: prop = ClassGroupProperties()
619          assert(isinstance(prop, ClassGroupProperties))          assert isinstance(prop, ClassGroupProperties)
620          self.prop = prop          self.prop = prop
621    
622        def GetDisplayText(self):
623            label = self.GetLabel()
624    
625            if label != "": return label
626    
627            return "DEFAULT"
628    
629      def __eq__(self, other):      def __eq__(self, other):
630          return isinstance(other, ClassGroupDefault) \          return isinstance(other, ClassGroupDefault) \
631              and self.GetProperties() == other.GetProperties()              and self.GetProperties() == other.GetProperties()
# Line 598  class ClassGroupRange(ClassGroup): Line 666  class ClassGroupRange(ClassGroup):
666                                 self.GetProperties(),                                 self.GetProperties(),
667                                 self.GetLabel())                                 self.GetLabel())
668    
669        def __deepcopy__(self, memo):
670            return ClassGroupRange(copy.copy(self.GetMin()),
671                                   copy.copy(self.GetMax()),
672                                   copy.copy(self.GetProperties()),
673                                   copy.copy(self.GetLabel()))
674    
675      def GetMin(self):      def GetMin(self):
676          """Return the range's minimum value."""          """Return the range's minimum value."""
677          return self.min          return self.min
# Line 661  class ClassGroupRange(ClassGroup): Line 735  class ClassGroupRange(ClassGroup):
735                  a default set of properties is created.                  a default set of properties is created.
736          """          """
737          if prop is None: prop = ClassGroupProperties()          if prop is None: prop = ClassGroupProperties()
738          assert(isinstance(prop, ClassGroupProperties))          assert isinstance(prop, ClassGroupProperties)
739          self.prop = prop          self.prop = prop
740    
741        def GetDisplayText(self):
742            label = self.GetLabel()
743    
744            if label != "": return label
745    
746            return _("%s - %s") % (self.GetMin(), self.GetMax())
747    
748      def __eq__(self, other):      def __eq__(self, other):
749          return isinstance(other, ClassGroupRange) \          return isinstance(other, ClassGroupRange) \
750              and self.GetProperties() == other.GetProperties() \              and self.GetProperties() == other.GetProperties() \
# Line 695  class ClassGroupMap(ClassGroup): Line 776  class ClassGroupMap(ClassGroup):
776      def GetPropertiesFromValue(self, value):      def GetPropertiesFromValue(self, value):
777          pass          pass
778    
779        def GetDisplayText(self):
780            return "Map: " + self.map_type
781    
782      #      #
783      # built-in mappings      # built-in mappings
784      #      #
Legend:
Removed from v.479  
changed lines
  Added in v.609
[email protected]
 ViewVC Help
Powered by ViewVC 1.1.26