diff options
| author | Anthony LaTorre <tlatorre9@gmail.com> | 2011-08-08 14:43:19 -0400 |
|---|---|---|
| committer | Anthony LaTorre <tlatorre9@gmail.com> | 2011-08-08 14:43:19 -0400 |
| commit | ed642493fccbcf13efef5491f73241c6a9434ad8 (patch) | |
| tree | 654e86f40cb8e722aa4eb1de636dcd585ac60da6 /color/colormap.py | |
| parent | 8f3f0a9ca7d7fcaffc53a64798d2427342662115 (diff) | |
| download | chroma-ed642493fccbcf13efef5491f73241c6a9434ad8.tar.gz chroma-ed642493fccbcf13efef5491f73241c6a9434ad8.tar.bz2 chroma-ed642493fccbcf13efef5491f73241c6a9434ad8.zip | |
toggle fullscreen mode with f11 key. use matplotlib colors maps defined in matplotlib.cm.
Diffstat (limited to 'color/colormap.py')
| -rw-r--r-- | color/colormap.py | 24 |
1 files changed, 5 insertions, 19 deletions
diff --git a/color/colormap.py b/color/colormap.py index b3761ae..e5bd73e 100644 --- a/color/colormap.py +++ b/color/colormap.py @@ -1,27 +1,13 @@ import numpy as np +import matplotlib.cm as cm -import matplotlib.pyplot as plt - -def map_to_color(a, range=None): +def map_to_color(a, range=None, map=cm.hsv): a = np.asarray(a) if range is None: range = (a.min(), a.max()) - x = np.linspace(0, np.pi, 100) - - yr = np.cos(x)**2 - yr[x > np.pi/2] = 0.0 - yg = np.sin(x)**2 - yb = np.cos(x)**2 - yb[x < np.pi/2] = 0.0 - - #plt.plot(x, yr, 'r-', x, yb, 'b-', x, yg, 'g-') - #plt.show() - - ax = np.pi*(a - range[0])/(range[1]-range[0]) + ax = (a - range[0])/(range[1]-range[0]) - r = (np.interp(ax, x, yr)*255).astype(np.uint32) - g = (np.interp(ax, x, yg)*255).astype(np.uint32) - b = (np.interp(ax, x, yb)*255).astype(np.uint32) + rgba = (map(ax)*255).astype(np.uint32) - return r << 16 | g << 8 | b + return rgba[:,0] << 16 | rgba[:,1] << 8 | rgba[:,2] |