diff options
Diffstat (limited to 'color/colormap.py')
| -rw-r--r-- | color/colormap.py | 11 |
1 files changed, 9 insertions, 2 deletions
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] |