[3c44c66] | 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 | This module manages all data loaded into the application. Data_manager makes |
---|
| 12 | available all data loaded for the current perspective. |
---|
| 13 | |
---|
[f444b20] | 14 | All modules "creating Data" posts their data to data_manager . |
---|
| 15 | Data_manager make these new data available for all other perspectives. |
---|
[3c44c66] | 16 | """ |
---|
[f444b20] | 17 | import logging |
---|
[c5e84fb] | 18 | from sans.guiframe.data_state import DataState |
---|
[75fbd17] | 19 | from sans.guiframe.utils import parse_name |
---|
[f444b20] | 20 | |
---|
[3c44c66] | 21 | class DataManager(object): |
---|
| 22 | """ |
---|
[f444b20] | 23 | Manage a list of data |
---|
[3c44c66] | 24 | """ |
---|
[f444b20] | 25 | def __init__(self): |
---|
[3c44c66] | 26 | """ |
---|
[aebc4cc] | 27 | Store opened path and data object created at the loading time |
---|
[f444b20] | 28 | :param auto_plot: if True the datamanager sends data to plotting |
---|
| 29 | plugin. |
---|
| 30 | :param auto_set_data: if True the datamanager sends to the current |
---|
| 31 | perspective |
---|
[4e9583c] | 32 | """ |
---|
[c5e84fb] | 33 | self._selected_data = {} |
---|
[f444b20] | 34 | self.stored_data = {} |
---|
[aebc4cc] | 35 | self.message = "" |
---|
[75fbd17] | 36 | self.data_name_dict = {} |
---|
[f444b20] | 37 | |
---|
[75fbd17] | 38 | def rename(self, name): |
---|
| 39 | """ |
---|
| 40 | rename data |
---|
| 41 | """ |
---|
| 42 | ## name of the data allow to differentiate data when plotted |
---|
| 43 | name = parse_name(name=name, expression="_") |
---|
| 44 | |
---|
| 45 | max_char = name.find("[") |
---|
| 46 | if max_char < 0: |
---|
| 47 | max_char = len(name) |
---|
| 48 | name = name[0:max_char] |
---|
| 49 | |
---|
| 50 | if name not in self.data_name_dict: |
---|
| 51 | self.data_name_dict[name] = 0 |
---|
| 52 | else: |
---|
| 53 | self.data_name_dict[name] += 1 |
---|
| 54 | name = name + " [" + str(self.data_name_dict[name]) + "]" |
---|
| 55 | return name |
---|
| 56 | |
---|
[f444b20] | 57 | def add_data(self, data_list): |
---|
| 58 | """ |
---|
| 59 | receive a list of |
---|
| 60 | """ |
---|
[584c4c4] | 61 | self._selected_data = {} |
---|
[f444b20] | 62 | for data in data_list: |
---|
| 63 | if data.id in self.stored_data: |
---|
| 64 | msg = "Data manager already stores %s" % str(data.name) |
---|
| 65 | msg += "" |
---|
| 66 | logging.info(msg) |
---|
[c5e84fb] | 67 | data_state = DataState(data) |
---|
[584c4c4] | 68 | self._selected_data[data.id] = data_state |
---|
| 69 | self.stored_data[data.id] = data_state |
---|
[213892bc] | 70 | |
---|
[f444b20] | 71 | def set_auto_plot(self, flag=False): |
---|
| 72 | """ |
---|
| 73 | When flag is true the data is plotted automatically |
---|
| 74 | """ |
---|
| 75 | self._auto_set_data = flag |
---|
| 76 | |
---|
| 77 | def set_auto_set_data(self, flag=False): |
---|
| 78 | """ |
---|
| 79 | When flag is true the data is send to the current perspective |
---|
| 80 | automatically |
---|
| 81 | """ |
---|
| 82 | self._auto_set_data = flag |
---|
[3c44c66] | 83 | |
---|
[aebc4cc] | 84 | def get_message(self): |
---|
[3c44c66] | 85 | """ |
---|
[aebc4cc] | 86 | return message |
---|
[3c44c66] | 87 | """ |
---|
[aebc4cc] | 88 | return self.message |
---|
[3c44c66] | 89 | |
---|
[f444b20] | 90 | def get_by_id(self, id_list=None): |
---|
| 91 | """ |
---|
| 92 | get a list of data given a list of id |
---|
| 93 | """ |
---|
[584c4c4] | 94 | self._selected_data = {} |
---|
[f444b20] | 95 | for id in id_list: |
---|
| 96 | if id in self.stored_data: |
---|
[584c4c4] | 97 | self._selected_data[id] = self.stored_data[id] |
---|
[f444b20] | 98 | return self._selected_data |
---|
| 99 | |
---|
[213892bc] | 100 | def append_theory(self, data_id, theory): |
---|
| 101 | """ |
---|
| 102 | """ |
---|
| 103 | print "append theory", self.stored_data, data_id |
---|
| 104 | if data_id in self.stored_data: |
---|
| 105 | data_state = self.stored_data[data_id] |
---|
| 106 | data_state.set_theory(theory) |
---|
| 107 | |
---|
| 108 | def delete_data(self, data_id, theory_id, delete_all): |
---|
| 109 | """ |
---|
| 110 | """ |
---|
[6db811e] | 111 | if data_id in self.stored_data: |
---|
| 112 | del self.stored_data[data_id] |
---|
| 113 | if data_id in self._selected_data: |
---|
| 114 | del self._selected_data |
---|
| 115 | |
---|
[f444b20] | 116 | def delete_by_id(self, id_list=None): |
---|
[3c44c66] | 117 | """ |
---|
[aebc4cc] | 118 | save data and path |
---|
[3c44c66] | 119 | """ |
---|
[f444b20] | 120 | for id in id_list: |
---|
| 121 | if id in self.stored_data: |
---|
| 122 | del self.stored_data[id] |
---|
[584c4c4] | 123 | if id in self._selected_data: |
---|
| 124 | del self._selected_data[id] |
---|
[aebc4cc] | 125 | |
---|
[f444b20] | 126 | def get_by_name(self, name_list=None): |
---|
[3c44c66] | 127 | """ |
---|
[f444b20] | 128 | return a list of data given a list of data names |
---|
[3c44c66] | 129 | """ |
---|
[584c4c4] | 130 | self._selected_data = {} |
---|
[f444b20] | 131 | for selected_name in name_list: |
---|
[584c4c4] | 132 | for id, data_state in self.stored_data.iteritems(): |
---|
| 133 | if data_state.data.name == selected_name: |
---|
| 134 | self._selected_data[id] = data_state |
---|
[f444b20] | 135 | return self._selected_data |
---|
| 136 | |
---|
| 137 | def delete_by_name(self, name_list=None): |
---|
| 138 | """ |
---|
| 139 | save data and path |
---|
| 140 | """ |
---|
| 141 | for selected_name in name_list: |
---|
[584c4c4] | 142 | for id, data_state in self.stored_data.iteritems(): |
---|
| 143 | if data_state.data.name == selected_name: |
---|
| 144 | del self._selected_data[id] |
---|
| 145 | del self.stored_data[data.id] |
---|
[f444b20] | 146 | |
---|
| 147 | def get_selected_data(self): |
---|
| 148 | """ |
---|
| 149 | Send list of selected data |
---|
| 150 | """ |
---|
| 151 | return self._selected_data |
---|
| 152 | |
---|
| 153 | def get_all_data(self): |
---|
| 154 | """ |
---|
| 155 | return list of all available data |
---|
| 156 | """ |
---|
[584c4c4] | 157 | return self.stored_data |
---|
[f444b20] | 158 | |
---|
| 159 | |
---|
[3c44c66] | 160 | |
---|