|
12 | 12 |
|
13 | 13 | import numpy as np
|
14 | 14 | import os
|
| 15 | +from matplotlib import colors, cm |
15 | 16 |
|
16 | 17 | from . import tools
|
17 | 18 |
|
18 |
| -# If matplotlib is available, load it |
19 |
| -try: |
20 |
| - import matplotlib.colors as mplc |
21 |
| - import matplotlib.pyplot as plt |
22 |
| - do_registry = True |
23 |
| -except: |
24 |
| - do_registry = False |
25 |
| - |
26 | 19 | # Location of rgb files
|
27 | 20 | datadir = os.path.join(os.path.split(__file__)[0], 'rgb')
|
28 | 21 |
|
|
41 | 34 | cmap_d[cmapname].name = cmapname
|
42 | 35 | cmap_d[cmapname + '_r'] = tools.cmap(rgb[::-1, :], N=256)
|
43 | 36 | cmap_d[cmapname + '_r'].name = cmapname + '_r'
|
44 |
| - |
45 |
| - # Register the cmap with matplotlib, if available |
46 |
| - if do_registry: |
47 |
| - rgb_with_alpha = np.zeros((rgb.shape[0],4)) |
48 |
| - rgb_with_alpha[:,:3] = rgb |
49 |
| - rgb_with_alpha[:,3] = 1. #set alpha channel to 1 |
50 |
| - reg_map = mplc.ListedColormap(rgb_with_alpha, 'cmocean_' + cmapname, rgb.shape[0]) |
51 |
| - plt.register_cmap(cmap = reg_map) |
52 |
| - |
53 |
| - # Register the reversed map |
54 |
| - reg_map_r = mplc.ListedColormap(rgb_with_alpha[::-1,:], 'cmocean_' + cmapname + '_r', rgb.shape[0]) |
55 |
| - plt.register_cmap(cmap = reg_map_r) |
| 37 | + |
| 38 | + # Register the cmap with matplotlib |
| 39 | + rgb_with_alpha = np.zeros((rgb.shape[0],4)) |
| 40 | + rgb_with_alpha[:,:3] = rgb |
| 41 | + rgb_with_alpha[:,3] = 1. #set alpha channel to 1 |
| 42 | + reg_map = colors.ListedColormap(rgb_with_alpha, 'cmo.' + cmapname, rgb.shape[0]) |
| 43 | + cm.register_cmap(cmap = reg_map) |
| 44 | + |
| 45 | + # Register the reversed map |
| 46 | + reg_map_r = colors.ListedColormap(rgb_with_alpha[::-1,:], 'cmo.' + cmapname + '_r', rgb.shape[0]) |
| 47 | + cm.register_cmap(cmap = reg_map_r) |
56 | 48 |
|
57 | 49 | # make colormaps available to call
|
58 | 50 | locals().update(cmap_d)
|
0 commit comments