propagated the N parameter for choosing number of levels into more functions

Author: kthyng

cmocean/tools.py

@@ -6,7 +6,7 @@
 import matplotlib as mpl
 
 
-def print_colormaps(cmaps, returnrgb=True, savefiles=False):
+def print_colormaps(cmaps, N=256, returnrgb=True, savefiles=False):
     '''Print colormaps in 256 RGB colors to text files.
 
     :param returnrgb=False: Whether or not to return the rgb array. Only makes sense to do if print one colormaps' rgb.
@@ -17,7 +17,7 @@ def print_colormaps(cmaps, returnrgb=True, savefiles=False):
 
     for cmap in cmaps:
 
-        rgbtemp = cmap(np.linspace(0, 1, 256))[np.newaxis, :, :3][0]
+        rgbtemp = cmap(np.linspace(0, 1, N))[np.newaxis, :, :3][0]
         if savefiles:
             np.savetxt(cmap.name + '-rgb.txt', rgbtemp)
         rgb.append(rgbtemp)