1
2
3
4
5
6
7
8
9
10
11
12
13
14
15
16
17
18
19
20
21
22
23
24
25
26
27
28
29
30
31
32
33
34
35
36
37
38
39
40
41
42
43
44
45
46
47
48
49
50
51
52
53
54
55
56
57
58
59
60
61
62
63
64
65
|
#!/usr/bin/env python
"""
ColorScalars
"""
"""
Copyright 2001 Pearu Peterson all rights reserved,
Pearu Peterson <pearu@ioc.ee>
Permission to use, modify, and distribute this software is given under the
terms of the LGPL. See http://www.fsf.org
NO WARRANTY IS EXPRESSED OR IMPLIED. USE AT YOUR OWN RISK.
$Revision: 1.2 $
$Date: 2001-05-31 17:48:54 $
Pearu Peterson
"""
__version__ = "$Id: ColorScalars.py,v 1.2 2001-05-31 17:48:54 pearu Exp $"
import common
import DataSetAttr
class ColorScalars(DataSetAttr.DataSetAttr):
"""Holds VTK color scalars.
Usage:
ColorScalars(<sequence of n-sequences> ,name = <string>)
Attributes:
scalars
name
Public methods:
get_size()
to_string(format = 'ascii')
"""
def __init__(self,scalars,name=None):
self.name = self._get_name(name)
self.scalars = self.get_n_seq_seq(scalars,self.default_value)
def to_string(self,format='ascii'):
ret = ['COLOR_SCALARS %s %s'%(self.name,len(self.scalars[0]))]
seq = self.scalars
if format=='binary':
if not common.is_int255(seq):
seq = self.float01_to_int255(seq)
ret.append(self.seq_to_string(seq,format,'unsigned char'))
else:
if not common.is_float01(seq):
seq = self.int255_to_float01(seq)
ret.append(self.seq_to_string(seq,format,'float'))
return '\n'.join(ret)
def get_size(self):
return len(self.scalars)
def color_scalars_fromfile(f,n,sl):
assert len(sl)==2
dataname = sl[0].strip()
nvals = eval(sl[1])
scalars = []
while len(scalars)<nvals*n:
scalars += map(eval,common._getline(f).split(' '))
assert len(scalars)==nvals*n
scalars2 = []
for i in range(0,len(scalars),nvals):
scalars2.append(scalars[i:i+nvals])
return ColorScalars(scalars2,dataname)
if __name__ == "__main__":
print ColorScalars([[3,3],[4,3],240,3,2]).to_string()
|
