added Manuel's contour linestyle patch

jdh2358 · jdh2358 · commit 51087e9504b3 · 2007-11-02T13:13:40.000Z
svn path=/trunk/matplotlib/; revision=4095
diff --git a/lib/matplotlib/contour.py b/lib/matplotlib/contour.py
@@ -390,7 +390,8 @@ def __init__(self, ax, *args, **kwargs):
         self.levels = kwargs.get('levels', None)
         self.filled = kwargs.get('filled', False)
         self.linewidths = kwargs.get('linewidths', None)
-
+        self.linestyles = kwargs.get('linestyles', 'solid')
+        
         self.alpha = kwargs.get('alpha', 1.0)
         self.origin = kwargs.get('origin', None)
         self.extent = kwargs.get('extent', None)
@@ -457,11 +458,13 @@ def __init__(self, ax, *args, **kwargs):
         else:
             tlinewidths = self._process_linewidths()
             self.tlinewidths = tlinewidths
+            tlinestyles = self._process_linestyles()
             C = _cntr.Cntr(x, y, z.filled(), _mask)
-            for level, width in zip(self.levels, tlinewidths):
+            for level, width, lstyle in zip(self.levels, tlinewidths, tlinestyles):
                 nlist = C.trace(level, points = 0)
                 col = collections.LineCollection(nlist,
-                                     linewidths = width)
+                                     linewidths = width,
+                                     linestyle = lstyle)
 
                 if level < 0.0 and self.monochrome:
                     ls = mpl.rcParams['contour.negative_linestyle']
@@ -696,6 +699,18 @@ def _process_linewidths(self):
                 linewidths = [linewidths] * Nlev
             tlinewidths = [(w,) for w in linewidths]
         return tlinewidths
+    
+    def _process_linestyles(self):
+        linestyles = self.linestyles
+        Nlev = len(self.levels)
+        if linestyles is None:
+            tlinestyles = ['solid'] * Nlev
+        else:
+            if cbook.is_string_like(linestyles):
+                tlinestyles = [linestyles] * Nlev
+            elif cbook.iterable(linestyles) and len(linestyles) < Nlev:
+                tlinestyles = list(linestyles) * int(npy.ceil(Nlev/len(linestyles)))
+        return tlinestyles
 
     def get_alpha(self):
         '''For compatibility with artists, return self.alpha'''
diff --git a/lib/matplotlib/mlab.py b/lib/matplotlib/mlab.py
@@ -811,7 +811,7 @@ def prctile(x, p = (0.0, 25.0, 50.0, 75.0, 100.0)):
     """
 
 
-    x = npy.ravel(x)
+    x = npy.array(x).ravel()  # we need a copy
     x.sort()
     Nx = len(x)
 
@@ -1449,7 +1449,7 @@ def rec2csv(r, fname, delimiter=','):
     files is automatic, if the filename ends in .gz
     """
     fh = cbook.to_filehandle(fname, 'w')
-    writer = csv.writer(fh, delimiter=delimiter)
+    writer = csv.writer(fh, delimiter=delimiter, quoting=csv.QUOTE_NONNUMERIC)
     header = r.dtype.names
     writer.writerow(header)
     for row in r:
