Mercurial > hg > orthanc-stone

comparison Applications/Resources/Colormaps/GenerateColormaps.py @ 1586:b5417e377636

Find changesets by keywords (author, files, the commit message), revision number or hash, or revset expression.
reorganization
author Sebastien Jodogne <s.jodogne@gmail.com>
date Thu, 22 Oct 2020 16:17:10 +0200
parents OrthancStone/Resources/Colormaps/GenerateColormaps.py@244ad1e4e76a
children 8c5f9864545f
comparison
equal deleted inserted replaced
1585:94edbfa64c97 1586:b5417e377636
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))