Changeset 693 for branches/Release-2-fixes/python/asaplot.py

Timestamp:

09/29/05 16:35:59 (19 years ago)

Author:

mar637

Message:

added user customisable color and linestyles

File:

• branches/Release-2-fixes/python/asaplot.py (modified) (4 diffs)

branches/Release-2-fixes/python/asaplot.py

@@ -17 +17 @@
 from matplotlib.numerix import sqrt
 from matplotlib import rc, rcParams
+from asap import rcParams as asaprcParams
 
 # Force use of the newfangled toolbar.
@@ -58 +59 @@
             self.set_panels(rows, cols)
 
-
-        # Set matplotlib default colour sequence.
-        self.colormap = ['blue', 'green', 'red', 'cyan', 'magenta', 'yellow', 'black', 'purple', 'orange', 'pink']
+        # Set matplotlib default colour sequence.
+        self.colormap = "green red black cyan magenta orange blue purple yellow pink".split()
+        
+        c = asaprcParams['plotter.colours']
+        if isinstance(c,str) and len(c) > 0:
+            self.colormap = c.split()
+
+        self.lsalias = {"line":  [1,0],
+                        "dashdot": [4,2,1,2],
+                        "dashed" : [4,2,4,2],
+                        "dotted" : [1,2],
+                        "dashdotdot": [4,2,1,2,1,2],
+                        "dashdashdot": [4,2,4,2,1,2]
+                        }
+
+        styles = "line dashed dotted dashdot".split()
+        c = asaprcParams['plotter.linestyles']
+        if isinstance(c,str) and len(c) > 0:
+            styles = c.split()
+        s = []
+        for ls in styles:
+            if self.lsalias.has_key(ls):
+                s.append(self.lsalias.get(ls))
+            else:
+                s.append('-')
+        self.linestyles = s
+
         self.color = 0;
+        self.linestyle = 0;
         self.attributes = {}
         self.loc = 0
@@ -80 +106 @@
         self.axes.clear()
         self.color = 0
+        self.linestyle = 0
         self.lines = []
 
-
-    def palette(self, color, colormap=None):
+    def palette(self, color, colormap=None, linestyle=0, linestyles=None):
         if colormap:
-            self.colormap = colormap
+            if isinstance(colormap,list):
+                self.colormap = colormap
+            elif isinstance(colormap,str):
+                self.colormap = colormap.split()
         if 0 <= color < len(self.colormap):
             self.color = color
+        if linestyles:
+            self.linestyles = []
+            if isinstance(linestyles,list):
+                styles = linestyles
+            elif isinstance(linestyles,str):
+                styles = linestyles.split()
+            for ls in styles:
+                if self.lsalias.has_key(ls):
+                    self.linestyles.append(self.lsalias.get(ls))
+                else:
+                    self.linestyles.append(self.lsalias.get('line'))
+        if 0 <= linestyle < len(self.linestyles):
+            self.linestyle = linestyle
 
     def delete(self, numbers=None):
@@ -244 +286 @@
             for segment in self.lines[i]:
                 getattr(segment, "set_color")(self.colormap[self.color])
-
+                if len(self.colormap)  == 1:
+                    getattr(segment, "set_dashes")(self.linestyles[self.linestyle])
             self.color += 1
             if self.color >= len(self.colormap):
                 self.color = 0
+
+            if len(self.colormap) == 1:
+                self.linestyle += 1
+            if self.linestyle >= len(self.linestyles):
+                self.linestyle = 0               
 
         self.show()