From ea5cc8e4e0e9bd1d1a1e5e7140c0022a8bd2a59d Mon Sep 17 00:00:00 2001 From: Anthony LaTorre Date: Wed, 10 Aug 2011 00:20:23 -0400 Subject: add the ability to cache a geometry along with its bounding volume hierarchy. cached files are stored in $HOME/.chroma. fixed the timeit() decorator so that the decorated function is still able to pass back a return value. --- color/colormap.py | 11 +++++++++-- 1 file changed, 9 insertions(+), 2 deletions(-) (limited to 'color') diff --git a/color/colormap.py b/color/colormap.py index e5bd73e..6f3e056 100644 --- a/color/colormap.py +++ b/color/colormap.py @@ -1,13 +1,20 @@ import numpy as np import matplotlib.cm as cm -def map_to_color(a, range=None, map=cm.hsv): +def map_to_color(a, range=None, map=cm.hsv, weights=None): a = np.asarray(a) if range is None: range = (a.min(), a.max()) ax = (a - range[0])/(range[1]-range[0]) - rgba = (map(ax)*255).astype(np.uint32) + frgba = map(ax) + + if weights is not None: + frgba[:,0] *= weights + frgba[:,1] *= weights + frgba[:,2] *= weights + + rgba = (frgba*255).astype(np.uint32) return rgba[:,0] << 16 | rgba[:,1] << 8 | rgba[:,2] -- cgit