Merge pull request #67 from bastorer/master

Update cm.py to register maps with matplotlib

Author: kthyng

cmocean/cm.py

@@ -12,6 +12,7 @@
 
 import numpy as np
 import os
+from matplotlib import colors, cm
 
 from . import tools
 
@@ -34,5 +35,16 @@
     cmap_d[cmapname + '_r'] = tools.cmap(rgb[::-1, :], N=256)
     cmap_d[cmapname + '_r'].name = cmapname + '_r'
 
+    # Register the cmap with matplotlib
+    rgb_with_alpha = np.zeros((rgb.shape[0],4))
+    rgb_with_alpha[:,:3] = rgb
+    rgb_with_alpha[:,3]  = 1.  #set alpha channel to 1
+    reg_map = colors.ListedColormap(rgb_with_alpha, 'cmo.' + cmapname, rgb.shape[0])
+    cm.register_cmap(cmap = reg_map)
+
+    # Register the reversed map
+    reg_map_r = colors.ListedColormap(rgb_with_alpha[::-1,:], 'cmo.' + cmapname + '_r', rgb.shape[0])
+    cm.register_cmap(cmap = reg_map_r)
+
 # make colormaps available to call
 locals().update(cmap_d)