From ed642493fccbcf13efef5491f73241c6a9434ad8 Mon Sep 17 00:00:00 2001 From: Anthony LaTorre Date: Mon, 8 Aug 2011 14:43:19 -0400 Subject: toggle fullscreen mode with f11 key. use matplotlib colors maps defined in matplotlib.cm. --- color/colormap.py | 24 +++++------------------- 1 file changed, 5 insertions(+), 19 deletions(-) (limited to 'color/colormap.py') 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] -- cgit