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

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

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

revision 1352 by jonathan, Wed Jul 2 09:36:39 2003 UTC revision 1759 by bh, Fri Sep 26 18:36:01 2003 UTC
# Line 26  def generate_singletons(_list, ramp): Line 26  def generate_singletons(_list, ramp):
26      The resulting classification will consist of one group for each      The resulting classification will consist of one group for each
27      item in _list whose properties ramp between 'prop1' and 'prop2'.      item in _list whose properties ramp between 'prop1' and 'prop2'.
28    
29      _list -- any object that implements the iterator interface      _list -- a list of values for each singleton
30    
31      ramp -- an object which implements the CustomRamp interface      ramp -- an object which implements the CustomRamp interface
32      """      """
# Line 34  def generate_singletons(_list, ramp): Line 34  def generate_singletons(_list, ramp):
34      clazz = Classification()      clazz = Classification()
35    
36      i = 0      i = 0
37        maxValue = float(len(_list) - 1)
38        if maxValue < 1: maxValue = 1
39    
40      for value in _list:      for value in _list:
41          prop = ramp.GetProperties(float(i) / len(_list))          prop = ramp.GetProperties(i / maxValue)
42          clazz.AppendGroup(ClassGroupSingleton(value, prop))          clazz.AppendGroup(ClassGroupSingleton(value, prop))
43          i += 1          i += 1
44    
# Line 56  def generate_uniform_distribution(min, m Line 59  def generate_uniform_distribution(min, m
59      cur_min = min      cur_min = min
60    
61      end = "["      end = "["
62        maxValue = float(numGroups - 1)
63        if maxValue < 1: maxValue = 1
64    
65      for i in range(1, numGroups + 1):      for i in range(1, numGroups + 1):
66    
67          prop = ramp.GetProperties(float(i-1) / numGroups)          prop = ramp.GetProperties(float(i-1) / maxValue)
68    
69          if intStep:          if intStep:
70              cur_max = min + int(round((i * (max - min + 1)) / float(numGroups)))              cur_max = min + int(round((i * (max - min + 1)) / float(numGroups)))
# Line 122  def generate_quantiles(_list, percents, Line 128  def generate_quantiles(_list, percents,
128              i = 1              i = 1
129              end = "]"              end = "]"
130    
131                maxValue = float(numGroups - 1)
132                if maxValue < 1: maxValue = 1
133              for (q, p) in quantiles[3]:              for (q, p) in quantiles[3]:
134    
135                  prop = ramp.GetProperties(float(i-1) / numGroups)                  prop = ramp.GetProperties(float(i-1) / maxValue)
136    
137                  if i == numGroups:                  if i == numGroups:
138                      max = endMax                      max = endMax
# Line 144  def generate_quantiles(_list, percents, Line 152  def generate_quantiles(_list, percents,
152    
153      return (adjusted, clazz)      return (adjusted, clazz)
154    
 def GenQuantiles0(_list, percents, ramp, _range):  
     """Same as GenQuantiles, but the first class won't be added to  
     the classification.  
   
     Returns a tuple (adjusted, Classification, upper_class0) where  
     upper_class0 is the highest value inside the first class.  
   
     _list -- a sort list of values  
   
     percents -- a sorted list of floats in the range 0.0-1.0 which  
                 represent the upper bound of each quantile. the  
                 union of all percentiles should be the entire  
                 range from 0.0-1.0  
   
     ramp -- an object which implements the CustomRamp interface  
   
     _range -- a Range object  
   
     Raises a Value Error if 'percents' has fewer than two items, or  
     does not cover the entire range.  
     """  
   
     clazz = Classification()  
     quantiles = calculate_quantiles(_list, percents, _range)  
     adjusted = True  
   
     if quantiles is not None:  
   
         numGroups = len(quantiles[3]) - 1  
   
         if numGroups > 0:  
             adjusted = quantiles[0]  
   
             start, min, endMax, right = _range.GetRange()  
   
             class0 = quantiles[3][0]  
             min = _list[class0[0]]  
             oldp = class0[1]  
             i = 1  
             end = "]"  
   
             for (q, p) in quantiles[3][1:]:  
                 prop = ramp.GetProperties(float(i) / numGroups)  
   
                 if i == numGroups:  
                     max = endMax  
                     end = right  
                 else:  
                     max = _list[q]  
   
                 group = ClassGroupRange(Range((start, min, max, end)), prop)  
       
                 group.SetLabel("%s%% - %s%%" % (round(oldp*100, 2),  
                                                 round(p*100, 2)))  
                 oldp = p  
                 start = "]"  
                 min = max  
                 clazz.AppendGroup(group)  
                 i += 1  
   
     return (adjusted, clazz, _list[class0[0]])  
   
155    
156  def calculate_quantiles(_list, percents, _range):  def calculate_quantiles(_list, percents, _range):
157      """Calculate quantiles for the given _list of percents from the      """Calculate quantiles for the given _list of percents from the
# Line 367  def calculate_quantiles(_list, percents, Line 313  def calculate_quantiles(_list, percents,
313  class CustomRamp:  class CustomRamp:
314    
315      def __init__(self, prop1, prop2):      def __init__(self, prop1, prop2):
316            """Create a ramp between prop1 and prop2."""
317          self.prop1 = prop1          self.prop1 = prop1
318          self.prop2 = prop2          self.prop2 = prop2
319    
320      def GetRamp(self):      def GetRamp(self):
321            """Return this ramp."""
322          return self          return self
323    
324      def GetProperties(self, index):      def GetProperties(self, index):
# Line 386  class CustomRamp: Line 334  class CustomRamp:
334    
335          newProps = ClassGroupProperties()          newProps = ClassGroupProperties()
336    
337          color1 = self.prop1.GetLineColor()          self.__SetProperty(self.prop1.GetLineColor(),
338          color2 = self.prop2.GetLineColor()                             self.prop2.GetLineColor(),
339                               index, newProps.SetLineColor)
340          self.__SetProperty(color1, color2, index, newProps.SetLineColor)          self.__SetProperty(self.prop1.GetFill(), self.prop2.GetFill(),
341          self.__SetProperty(color1, color2, index, newProps.SetFill)                             index, newProps.SetFill)
342    
343          w = (self.prop2.GetLineWidth() - self.prop1.GetLineWidth()) \          w = (self.prop2.GetLineWidth() - self.prop1.GetLineWidth()) \
344              * index \              * index \
345              + self.prop1.GetLineWidth()              + self.prop1.GetLineWidth()
   
346          newProps.SetLineWidth(int(round(w)))          newProps.SetLineWidth(int(round(w)))
347    
348          return newProps          return newProps
349    
350      def __SetProperty(self, color1, color2, index, setf):      def __SetProperty(self, color1, color2, index, setf):
351            """Use setf to set the appropriate property for the point
352            index percent between color1 and color2. setf is a function
353            to call that accepts a Color object or Transparent.
354            """
355    
356          if color1 is Transparent and color2 is Transparent:          if color1 is Transparent and color2 is Transparent:
357              setf(Transparent)              setf(Transparent)
# Line 421  class CustomRamp: Line 372  class CustomRamp:
372                  (color2.blue  - color1.blue)  * index + color1.blue))                  (color2.blue  - color1.blue)  * index + color1.blue))
373    
374  class MonochromaticRamp(CustomRamp):  class MonochromaticRamp(CustomRamp):
375        """Helper class to make ramps between two colors."""
376    
377      def __init__(self, start, end):      def __init__(self, start, end):
378            """Create a Monochromatic Ramp.
379    
380            start -- starting Color
381    
382            end -- ending Color
383            """
384          sp = ClassGroupProperties()          sp = ClassGroupProperties()
385          sp.SetLineColor(start)          sp.SetLineColor(start)
386          sp.SetFill(start)          sp.SetFill(start)
# Line 432  class MonochromaticRamp(CustomRamp): Line 391  class MonochromaticRamp(CustomRamp):
391    
392          CustomRamp.__init__(self, sp, ep)          CustomRamp.__init__(self, sp, ep)
393    
394  GreyRamp       = MonochromaticRamp(Color(1, 1, 1),  Color(0, 0, 0))  grey_ramp         = MonochromaticRamp(Color(1, 1, 1),  Color(0, 0, 0))
395  RedRamp        = MonochromaticRamp(Color(1, 1, 1),  Color(.8, 0, 0))  red_ramp          = MonochromaticRamp(Color(1, 1, 1),  Color(.8, 0, 0))
396  GreenRamp      = MonochromaticRamp(Color(1, 1, 1),  Color(0, .8, 0))  green_ramp        = MonochromaticRamp(Color(1, 1, 1),  Color(0, .8, 0))
397  BlueRamp       = MonochromaticRamp(Color(1, 1, 1),  Color(0, 0, .8))  blue_ramp         = MonochromaticRamp(Color(1, 1, 1),  Color(0, 0, .8))
398  GreenToRedRamp = MonochromaticRamp(Color(1, .8, 1), Color(1, 0, 0))  green_to_red_ramp = MonochromaticRamp(Color(0, .8, 0), Color(1, 0, 0))
399    
400  class HotToColdRamp:  class HotToColdRamp:
401        """A ramp that generates properties with colors ranging from
402        'hot' colors (e.g. red, orange) to 'cold' colors (e.g. green, blue)
403        """
404    
405      def GetRamp(self):      def GetRamp(self):
406            """Return this ramp."""
407          return self          return self
408    
409      def GetProperties(self, index):      def GetProperties(self, index):
# Line 471  class HotToColdRamp: Line 434  class HotToColdRamp:
434    
435          return prop          return prop
436    
437    class FixedRamp:
438        """FixedRamp allows particular properties of a ramp to be
439        held constant over the ramp.
440        """
441    
442        def __init__(self, ramp, fixes):
443            """
444            ramp -- a source ramp to get the default properties
445    
446            fixes -- a tuple (lineColor, lineWidth, fillColor) such that
447                 if any item is not None, the appropriate property will
448                 be fixed to that item value.
449            """
450    
451            self.fixes = fixes
452            self.ramp = ramp
453    
454        def GetRamp(self):
455            """Return this ramp."""
456            return self
457    
458        def GetProperties(self, index):
459            """Return a ClassGroupProperties object whose properties
460            represent a point at 'index' between the properties in
461            the ramp that initialized this FixedRamp.
462    
463            index -- a value such that 0 <= index <= 1
464            """
465    
466            props = self.ramp.GetProperties(index)
467            if self.fixes[0] is not None: props.SetLineColor(self.fixes[0])
468            if self.fixes[1] is not None: props.SetLineWidth(self.fixes[1])
469            if self.fixes[2] is not None: props.SetFill(self.fixes[2])
470    
471            return props
Legend:
Removed from v.1352  
changed lines
  Added in v.1759
[email protected]
 ViewVC Help
Powered by ViewVC 1.1.26