[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 |
---|
[0348245] | 18 | import os |
---|
[c5e84fb] | 19 | from sans.guiframe.data_state import DataState |
---|
[75fbd17] | 20 | from sans.guiframe.utils import parse_name |
---|
[0348245] | 21 | import DataLoader.data_info as DataInfo |
---|
| 22 | from sans.guiframe.dataFitting import Data1D |
---|
| 23 | from sans.guiframe.dataFitting import Data2D |
---|
[f444b20] | 24 | |
---|
[a07e72f] | 25 | import wx |
---|
| 26 | |
---|
[3c44c66] | 27 | class DataManager(object): |
---|
| 28 | """ |
---|
[f444b20] | 29 | Manage a list of data |
---|
[3c44c66] | 30 | """ |
---|
[f444b20] | 31 | def __init__(self): |
---|
[3c44c66] | 32 | """ |
---|
[aebc4cc] | 33 | Store opened path and data object created at the loading time |
---|
[f444b20] | 34 | :param auto_plot: if True the datamanager sends data to plotting |
---|
| 35 | plugin. |
---|
| 36 | :param auto_set_data: if True the datamanager sends to the current |
---|
| 37 | perspective |
---|
[4e9583c] | 38 | """ |
---|
[c5e84fb] | 39 | self._selected_data = {} |
---|
[f444b20] | 40 | self.stored_data = {} |
---|
[aebc4cc] | 41 | self.message = "" |
---|
[75fbd17] | 42 | self.data_name_dict = {} |
---|
[f444b20] | 43 | |
---|
[0348245] | 44 | def create_gui_data(self, data, path=None): |
---|
| 45 | """ |
---|
| 46 | Receive data from loader and create a data to use for guiframe |
---|
| 47 | """ |
---|
| 48 | |
---|
| 49 | if issubclass(DataInfo.Data2D, data.__class__): |
---|
| 50 | new_plot = Data2D(image=None, err_image=None) |
---|
| 51 | else: |
---|
| 52 | new_plot = Data1D(x=[], y=[], dx=None, dy=None) |
---|
| 53 | |
---|
| 54 | new_plot.copy_from_datainfo(data) |
---|
| 55 | data.clone_without_data(clone=new_plot) |
---|
| 56 | #creating a name for data |
---|
| 57 | name = "" |
---|
| 58 | title = "" |
---|
| 59 | file_name = "" |
---|
| 60 | if path is not None: |
---|
| 61 | file_name = os.path.basename(path) |
---|
| 62 | if data.run: |
---|
| 63 | name = data.run[0] |
---|
| 64 | if name == "": |
---|
| 65 | name = file_name |
---|
| 66 | name = self.rename(name) |
---|
| 67 | #find title |
---|
| 68 | if data.title.strip(): |
---|
| 69 | title = data.title |
---|
| 70 | if title.strip() == "": |
---|
| 71 | title = file_name |
---|
| 72 | |
---|
| 73 | if new_plot.filename.strip() == "": |
---|
| 74 | new_plot.filename = file_name |
---|
| 75 | |
---|
| 76 | new_plot.name = name |
---|
| 77 | new_plot.title = title |
---|
| 78 | ## allow to highlight data when plotted |
---|
| 79 | new_plot.interactive = True |
---|
| 80 | ## when 2 data have the same id override the 1 st plotted |
---|
[a07e72f] | 81 | new_plot.id = wx.NewId() |
---|
[0348245] | 82 | ##group_id specify on which panel to plot this data |
---|
[a07e72f] | 83 | new_plot.group_id = [wx.NewId()] |
---|
[0348245] | 84 | new_plot.is_data = True |
---|
| 85 | new_plot.path = path |
---|
| 86 | ##post data to plot |
---|
| 87 | # plot data |
---|
| 88 | return new_plot |
---|
| 89 | |
---|
[75fbd17] | 90 | def rename(self, name): |
---|
| 91 | """ |
---|
| 92 | rename data |
---|
| 93 | """ |
---|
| 94 | ## name of the data allow to differentiate data when plotted |
---|
| 95 | name = parse_name(name=name, expression="_") |
---|
| 96 | |
---|
| 97 | max_char = name.find("[") |
---|
| 98 | if max_char < 0: |
---|
| 99 | max_char = len(name) |
---|
| 100 | name = name[0:max_char] |
---|
| 101 | |
---|
| 102 | if name not in self.data_name_dict: |
---|
| 103 | self.data_name_dict[name] = 0 |
---|
| 104 | else: |
---|
| 105 | self.data_name_dict[name] += 1 |
---|
| 106 | name = name + " [" + str(self.data_name_dict[name]) + "]" |
---|
| 107 | return name |
---|
| 108 | |
---|
[f444b20] | 109 | def add_data(self, data_list): |
---|
| 110 | """ |
---|
| 111 | receive a list of |
---|
| 112 | """ |
---|
[584c4c4] | 113 | self._selected_data = {} |
---|
[f444b20] | 114 | for data in data_list: |
---|
| 115 | if data.id in self.stored_data: |
---|
| 116 | msg = "Data manager already stores %s" % str(data.name) |
---|
| 117 | msg += "" |
---|
| 118 | logging.info(msg) |
---|
[c5e84fb] | 119 | data_state = DataState(data) |
---|
[584c4c4] | 120 | self._selected_data[data.id] = data_state |
---|
| 121 | self.stored_data[data.id] = data_state |
---|
[213892bc] | 122 | |
---|
[f444b20] | 123 | def set_auto_plot(self, flag=False): |
---|
| 124 | """ |
---|
| 125 | When flag is true the data is plotted automatically |
---|
| 126 | """ |
---|
| 127 | self._auto_set_data = flag |
---|
| 128 | |
---|
| 129 | def set_auto_set_data(self, flag=False): |
---|
| 130 | """ |
---|
| 131 | When flag is true the data is send to the current perspective |
---|
| 132 | automatically |
---|
| 133 | """ |
---|
| 134 | self._auto_set_data = flag |
---|
[3c44c66] | 135 | |
---|
[aebc4cc] | 136 | def get_message(self): |
---|
[3c44c66] | 137 | """ |
---|
[aebc4cc] | 138 | return message |
---|
[3c44c66] | 139 | """ |
---|
[aebc4cc] | 140 | return self.message |
---|
[3c44c66] | 141 | |
---|
[f444b20] | 142 | def get_by_id(self, id_list=None): |
---|
| 143 | """ |
---|
| 144 | get a list of data given a list of id |
---|
| 145 | """ |
---|
[584c4c4] | 146 | self._selected_data = {} |
---|
[f444b20] | 147 | for id in id_list: |
---|
| 148 | if id in self.stored_data: |
---|
[584c4c4] | 149 | self._selected_data[id] = self.stored_data[id] |
---|
[f444b20] | 150 | return self._selected_data |
---|
| 151 | |
---|
[213892bc] | 152 | def append_theory(self, data_id, theory): |
---|
| 153 | """ |
---|
| 154 | """ |
---|
| 155 | if data_id in self.stored_data: |
---|
| 156 | data_state = self.stored_data[data_id] |
---|
| 157 | data_state.set_theory(theory) |
---|
| 158 | |
---|
| 159 | def delete_data(self, data_id, theory_id, delete_all): |
---|
| 160 | """ |
---|
| 161 | """ |
---|
[6db811e] | 162 | if data_id in self.stored_data: |
---|
| 163 | del self.stored_data[data_id] |
---|
| 164 | if data_id in self._selected_data: |
---|
| 165 | del self._selected_data |
---|
| 166 | |
---|
[f444b20] | 167 | def delete_by_id(self, id_list=None): |
---|
[3c44c66] | 168 | """ |
---|
[aebc4cc] | 169 | save data and path |
---|
[3c44c66] | 170 | """ |
---|
[f444b20] | 171 | for id in id_list: |
---|
| 172 | if id in self.stored_data: |
---|
| 173 | del self.stored_data[id] |
---|
[584c4c4] | 174 | if id in self._selected_data: |
---|
| 175 | del self._selected_data[id] |
---|
[aebc4cc] | 176 | |
---|
[f444b20] | 177 | def get_by_name(self, name_list=None): |
---|
[3c44c66] | 178 | """ |
---|
[f444b20] | 179 | return a list of data given a list of data names |
---|
[3c44c66] | 180 | """ |
---|
[584c4c4] | 181 | self._selected_data = {} |
---|
[f444b20] | 182 | for selected_name in name_list: |
---|
[584c4c4] | 183 | for id, data_state in self.stored_data.iteritems(): |
---|
| 184 | if data_state.data.name == selected_name: |
---|
| 185 | self._selected_data[id] = data_state |
---|
[f444b20] | 186 | return self._selected_data |
---|
| 187 | |
---|
| 188 | def delete_by_name(self, name_list=None): |
---|
| 189 | """ |
---|
| 190 | save data and path |
---|
| 191 | """ |
---|
| 192 | for selected_name in name_list: |
---|
[584c4c4] | 193 | for id, data_state in self.stored_data.iteritems(): |
---|
| 194 | if data_state.data.name == selected_name: |
---|
| 195 | del self._selected_data[id] |
---|
| 196 | del self.stored_data[data.id] |
---|
[f444b20] | 197 | |
---|
| 198 | def get_selected_data(self): |
---|
| 199 | """ |
---|
| 200 | Send list of selected data |
---|
| 201 | """ |
---|
| 202 | return self._selected_data |
---|
| 203 | |
---|
| 204 | def get_all_data(self): |
---|
| 205 | """ |
---|
| 206 | return list of all available data |
---|
| 207 | """ |
---|
[584c4c4] | 208 | return self.stored_data |
---|
[f444b20] | 209 | |
---|
| 210 | |
---|
[3c44c66] | 211 | |
---|