Mercurial > hg > orthanc-stone
comparison Resources/Colormaps/GenerateColormaps.py @ 0:351ab0da0150
initial commit
author | Sebastien Jodogne <s.jodogne@gmail.com> |
---|---|
date | Fri, 14 Oct 2016 15:34:11 +0200 |
parents | |
children |
comparison
equal
deleted
inserted
replaced
-1:000000000000 | 0:351ab0da0150 |
---|---|
1 #!/usr/bin/python | |
2 | |
3 import array | |
4 import matplotlib.pyplot as plt | |
5 | |
6 def GenerateColormap(name): | |
7 colormap = [] | |
8 | |
9 for gray in range(256): | |
10 if name == 'red': | |
11 color = (gray / 255.0, 0, 0) | |
12 elif name == 'green': | |
13 color = (0, gray / 255.0, 0) | |
14 elif name == 'blue': | |
15 color = (0, 0, gray / 255.0) | |
16 else: | |
17 color = plt.get_cmap(name) (gray) | |
18 | |
19 colormap += map(lambda k: int(round(color[k] * 255)), range(3)) | |
20 | |
21 colormap[0] = 0 | |
22 colormap[1] = 0 | |
23 colormap[2] = 0 | |
24 | |
25 return array.array('B', colormap).tostring() | |
26 | |
27 | |
28 for name in [ | |
29 'hot', | |
30 'jet', | |
31 'blue', | |
32 'green', | |
33 'red', | |
34 ]: | |
35 with open('%s.lut' % name, 'w') as f: | |
36 f.write(GenerateColormap(name)) |