1 | ################################################################################ |
---|
2 | #This software was developed by the University of Tennessee as part of the |
---|
3 | #Distributed Data Analysis of Neutron Scattering Experiments (DANSE) |
---|
4 | #project funded by the US National Science Foundation. |
---|
5 | # |
---|
6 | #See the license text in license.txt |
---|
7 | # |
---|
8 | #copyright 2010, University of Tennessee |
---|
9 | ################################################################################ |
---|
10 | """ |
---|
11 | """ |
---|
12 | import copy |
---|
13 | |
---|
14 | |
---|
15 | class DataState(object): |
---|
16 | """ |
---|
17 | Store information about data |
---|
18 | """ |
---|
19 | def __init__(self, data=None, parent=None): |
---|
20 | """ |
---|
21 | |
---|
22 | """ |
---|
23 | self.parent = parent |
---|
24 | self.data = data |
---|
25 | self.name = "" |
---|
26 | self.path = None |
---|
27 | self.theory_list = {} |
---|
28 | self.message = "" |
---|
29 | self.id = None |
---|
30 | |
---|
31 | def __str__(self): |
---|
32 | _str = "" |
---|
33 | _str += "State with ID : %s \n" % str(self.id) |
---|
34 | if self.data is not None: |
---|
35 | _str += "Data name : %s \n" % str(self.data.name) |
---|
36 | _str += "Data ID : %s \n" % str(self.data.id) |
---|
37 | else: |
---|
38 | _str += "Theory Data: %s \n" % str(self.theory_data) |
---|
39 | if self.theory_data is not None: |
---|
40 | _str += "Data name : %s \n" % str(self.theory_data.name) |
---|
41 | _str += "Theory Data ID : %s \n" % str(self.theory_data.id) |
---|
42 | else: |
---|
43 | _str += "Theory Data: %s \n" % str(self.theory_data) |
---|
44 | |
---|
45 | _str += "Theories available: %s \n" % len(self.theory_list) |
---|
46 | if self.theory_list: |
---|
47 | for id, item in self.theory_list.iteritems(): |
---|
48 | theory_data, theory_state = item |
---|
49 | _str += "Theory name : %s \n" % str(theory_data.name) |
---|
50 | _str += "Theory ID : %s \n" % str(id) |
---|
51 | _str += "Theory info: \n" |
---|
52 | _str += str(theory_data) |
---|
53 | |
---|
54 | return _str |
---|
55 | |
---|
56 | def clone(self): |
---|
57 | obj = DataState(copy.deepcopy(self.data)) |
---|
58 | obj.parent = self.parent |
---|
59 | obj.name = self.name |
---|
60 | obj.path = self.path |
---|
61 | obj.message = self.message |
---|
62 | obj.id = self.id |
---|
63 | for id, item in self.theory_list.iteritems(): |
---|
64 | theory_data, theory_state = item |
---|
65 | state = None |
---|
66 | if theory_state is not None: |
---|
67 | state = theory_state.clone() |
---|
68 | obj.theory_list[id] = [copy.deepcopy(theory_data), |
---|
69 | state] |
---|
70 | return obj |
---|
71 | |
---|
72 | def set_name(self, name): |
---|
73 | self.name = name |
---|
74 | |
---|
75 | def get_name(self): |
---|
76 | return self.name |
---|
77 | |
---|
78 | def set_data(self, data): |
---|
79 | """ |
---|
80 | """ |
---|
81 | self.data = data |
---|
82 | |
---|
83 | |
---|
84 | def get_data(self): |
---|
85 | """ |
---|
86 | """ |
---|
87 | return self.data |
---|
88 | |
---|
89 | def set_path(self, path): |
---|
90 | """ |
---|
91 | Set the path of the loaded data |
---|
92 | """ |
---|
93 | self.path = path |
---|
94 | |
---|
95 | def get_path(self): |
---|
96 | """ |
---|
97 | return the path of the loaded data |
---|
98 | """ |
---|
99 | return self.path |
---|
100 | |
---|
101 | def set_theory(self, theory_data, theory_state=None): |
---|
102 | """ |
---|
103 | """ |
---|
104 | self.theory_list[theory_data.id] = [theory_data, theory_state] |
---|
105 | data, state = self.theory_list.values()[0] |
---|
106 | |
---|
107 | def get_theory(self): |
---|
108 | return self.theory_list |
---|
109 | |
---|
110 | def get_message(self): |
---|
111 | """ |
---|
112 | return message |
---|
113 | """ |
---|
114 | return self.message |
---|
115 | |
---|
116 | |
---|