|
10 | 10 |
|
11 | 11 | from __future__ import absolute_import |
12 | 12 |
|
13 | | -import matplotlib |
14 | 13 | import numpy as np |
15 | 14 | import os |
16 | 15 |
|
|
20 | 19 | datadir = os.path.join(os.path.split(__file__)[0], 'rgb') |
21 | 20 |
|
22 | 21 | # List of colormap names |
23 | | -cmaps = ['thermal', 'haline', 'solar', 'ice', 'gray', 'oxy', 'deep', 'dense', |
24 | | - 'algae', 'matter', 'turbid', 'speed', 'amp', 'tempo', 'phase', |
25 | | - 'balance', 'delta', 'curl'] |
| 22 | +cmapnames = ['thermal', 'haline', 'solar', 'ice', 'gray', 'oxy', 'deep', |
| 23 | + 'dense', 'algae', 'matter', 'turbid', 'speed', 'amp', 'tempo', |
| 24 | + 'phase', 'balance', 'delta', 'curl'] |
26 | 25 |
|
27 | 26 | # initialize dictionary to contain colormaps |
28 | 27 | cmap_d = dict() |
29 | 28 |
|
30 | 29 | # add colormaps and reversed to dictionary |
31 | | -for cmap in cmaps: |
32 | | - # rgb = np.loadtxt(cmap + '-rgb.txt') |
33 | | - rgb = np.loadtxt(os.path.join(datadir, cmap + '-rgb.txt')) |
34 | | - cmap_d[cmap] = tools.cmap(rgb, N=256) |
35 | | - cmap_d[cmap].name = cmap |
36 | | - cmap_d[cmap + '_r'] = tools.cmap(rgb[::-1, :], N=256) |
37 | | - cmap_d[cmap + '_r'].name = cmap + '_r' |
| 30 | +for cmapname in cmapnames: |
| 31 | + rgb = np.loadtxt(os.path.join(datadir, cmapname + '-rgb.txt')) |
| 32 | + cmap_d[cmapname] = tools.cmap(rgb, N=256) |
| 33 | + cmap_d[cmapname].name = cmapname |
| 34 | + cmap_d[cmapname + '_r'] = tools.cmap(rgb[::-1, :], N=256) |
| 35 | + cmap_d[cmapname + '_r'].name = cmapname + '_r' |
38 | 36 |
|
39 | 37 | # make colormaps available to call |
40 | 38 | locals().update(cmap_d) |
0 commit comments