[3feed3e] | 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 | """ |
---|
[c70eb7c] | 12 | import copy |
---|
| 13 | |
---|
[3feed3e] | 14 | |
---|
| 15 | class DataState(object): |
---|
| 16 | """ |
---|
| 17 | Store information about data |
---|
| 18 | """ |
---|
[ec611aad] | 19 | def __init__(self, data=None, parent=None): |
---|
[3feed3e] | 20 | """ |
---|
| 21 | |
---|
| 22 | """ |
---|
| 23 | self.parent = parent |
---|
[584c4c4] | 24 | self.data = data |
---|
| 25 | self.name = "" |
---|
| 26 | self.path = None |
---|
[5c4b674] | 27 | self.theory_list = {} |
---|
[584c4c4] | 28 | self.message = "" |
---|
[c70eb7c] | 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) |
---|
[df22224] | 37 | _str += "Theories available: %s \n" % len(self.theory_list) |
---|
[c70eb7c] | 38 | if self.theory_list: |
---|
| 39 | for id, item in self.theory_list.iteritems(): |
---|
| 40 | theory_data, theory_state = item |
---|
| 41 | _str += "Theory name : %s \n" % str(theory_data.name) |
---|
| 42 | _str += "Theory ID : %s \n" % str(id) |
---|
[df22224] | 43 | _str += "Theory info: \n" |
---|
[c70eb7c] | 44 | _str += str(theory_data) |
---|
[5080cda] | 45 | |
---|
[c70eb7c] | 46 | return _str |
---|
| 47 | |
---|
| 48 | def clone(self): |
---|
| 49 | obj = DataState(copy.deepcopy(self.data)) |
---|
| 50 | obj.parent = self.parent |
---|
| 51 | obj.name = self.name |
---|
| 52 | obj.path = self.path |
---|
| 53 | obj.message = self.message |
---|
| 54 | obj.id = self.id |
---|
| 55 | for id, item in self.theory_list.iteritems(): |
---|
| 56 | theory_data, theory_state = item |
---|
| 57 | state = None |
---|
| 58 | if theory_state is not None: |
---|
| 59 | state = theory_state.clone() |
---|
| 60 | obj.theory_list[id] = [copy.deepcopy(theory_data), |
---|
| 61 | state] |
---|
| 62 | return obj |
---|
[3feed3e] | 63 | |
---|
| 64 | def set_name(self, name): |
---|
[584c4c4] | 65 | self.name = name |
---|
[c70eb7c] | 66 | |
---|
[3feed3e] | 67 | def get_name(self): |
---|
[584c4c4] | 68 | return self.name |
---|
[c70eb7c] | 69 | |
---|
[3feed3e] | 70 | def set_data(self, data): |
---|
[a3c96f7a] | 71 | """ |
---|
| 72 | """ |
---|
[ec611aad] | 73 | self.data = data |
---|
| 74 | |
---|
| 75 | |
---|
| 76 | def get_data(self): |
---|
[a3c96f7a] | 77 | """ |
---|
| 78 | """ |
---|
[584c4c4] | 79 | return self.data |
---|
[3feed3e] | 80 | |
---|
| 81 | def set_path(self, path): |
---|
| 82 | """ |
---|
| 83 | Set the path of the loaded data |
---|
| 84 | """ |
---|
[584c4c4] | 85 | self.path = path |
---|
[3feed3e] | 86 | |
---|
| 87 | def get_path(self): |
---|
| 88 | """ |
---|
| 89 | return the path of the loaded data |
---|
| 90 | """ |
---|
[584c4c4] | 91 | return self.path |
---|
[3feed3e] | 92 | |
---|
[c70eb7c] | 93 | def set_theory(self, theory_data, theory_state=None): |
---|
[3feed3e] | 94 | """ |
---|
| 95 | """ |
---|
[c70eb7c] | 96 | self.theory_list[theory_data.id] = [theory_data, theory_state] |
---|
[df22224] | 97 | data, state = self.theory_list.values()[0] |
---|
[552b3d5] | 98 | |
---|
[3feed3e] | 99 | def get_theory(self): |
---|
[584c4c4] | 100 | return self.theory_list |
---|
[3feed3e] | 101 | |
---|
| 102 | def get_message(self): |
---|
| 103 | """ |
---|
| 104 | return message |
---|
| 105 | """ |
---|
[584c4c4] | 106 | return self.message |
---|
[3feed3e] | 107 | |
---|
| 108 | |
---|