source: sasview/guiframe/data_loader.py @ 4da35bc

ESS_GUIESS_GUI_DocsESS_GUI_batch_fittingESS_GUI_bumps_abstractionESS_GUI_iss1116ESS_GUI_iss879ESS_GUI_iss959ESS_GUI_openclESS_GUI_orderingESS_GUI_sync_sascalccostrafo411magnetic_scattrelease-4.1.1release-4.1.2release-4.2.2release_4.0.1ticket-1009ticket-1094-headlessticket-1242-2d-resolutionticket-1243ticket-1249ticket885unittest-saveload
Last change on this file since 4da35bc was 4da35bc, checked in by Jae Cho <jhjcho@…>, 14 years ago

improved theoryplot display from (inv file)

  • Property mode set to 100644
File size: 11.3 KB
Line 
1
2import os, sys,numpy
3import wx
4import re
5
6from dataFitting import Data1D
7from dataFitting import Data2D
8from DataLoader.loader import Loader
9from load_thread import DataReader
10from sans.perspectives.invariant import InvStateUpdateEvent
11from perspectives.fitting import FitStateUpdateEvent
12
13from sans.guicomm.events import NewPlotEvent, StatusEvent
14
15def enable_add_data(existing_panel, new_plot):
16    """
17    Enable append data on a plot panel
18    """
19    is_theory = len(existing_panel.plots)<= 1 and \
20        existing_panel.plots.values()[0].__class__.__name__=="Theory1D"
21       
22    is_data2d = hasattr(new_plot, 'data')
23    is_data1d = existing_panel.__class__.__name__ == "ModelPanel1D"\
24        and existing_panel.group_id is not None
25    has_meta_data = hasattr(new_plot, 'meta_data')
26   
27    #disable_add_data if the data is being recovered from  a saved state file.
28    is_state_data = False
29    if has_meta_data:
30        if 'invstate' in new_plot.meta_data: is_state_data = True
31        if  'prstate' in new_plot.meta_data: is_state_data = True
32        if  'fitstate' in new_plot.meta_data: is_state_data = True
33
34    return is_data1d and not is_data2d and not is_theory and not is_state_data
35
36def parse_name(name, expression):
37    """
38    remove "_" in front of a name
39    """
40    if re.match(expression, name) is not None:
41        word = re.split(expression, name, 1)
42        for item in word:           
43            if item.lstrip().rstrip() != '':
44                return item
45    else:
46        return name
47   
48def choose_data_file(parent, location=None):
49    """
50    """
51    path = None
52    if location == None:
53        location = os.getcwd()
54   
55    l = Loader()
56    cards = l.get_wildcards()
57    wlist = '|'.join(cards)
58   
59    dlg = wx.FileDialog(parent, "Choose a file", location, "", wlist, wx.OPEN)
60    if dlg.ShowModal() == wx.ID_OK:
61        path = dlg.GetPath()
62        mypath = os.path.basename(path)
63    dlg.Destroy()
64   
65    return path
66
67def open_dialog_append_data(panel_name, data_name):
68    """
69    Pop up an error message.
70   
71    :param panel_name: the name of the current panel
72    :param data_name: the name of the current data
73   
74    """
75    message = " Do you want to append %s data\n in "%(str(data_name))
76    message += " %s panel?\n\n"%(str(panel_name))
77    dial = wx.MessageDialog(None, message, 'Question',
78                       wx.YES_NO|wx.NO_DEFAULT|wx.ICON_QUESTION)
79    if dial.ShowModal() == wx.ID_YES:
80        return True
81    else:
82        return False
83   
84
85def load_ascii_1D(path):
86    """
87    Load a 1D ascii file, with errors
88    """
89    if path and os.path.isfile(path):
90   
91        file_x = numpy.zeros(0)
92        file_y = numpy.zeros(0)
93        file_dy = numpy.zeros(0)
94        file_dx = numpy.zeros(0)
95       
96        input_f = open(path,'r')
97        buff = input_f.read()
98        lines = buff.split('\n')
99       
100        has_dy = False
101        has_dx = False
102       
103        for line in lines:
104            try:
105                toks = line.split()
106                x = float(toks[0])
107                y = float(toks[1])
108                if len(toks)==3:
109                    has_dy = True
110                    errdy = float(toks[2])
111                else:
112                    errdy = 0.0
113                if len(toks) == 4:
114                    has_dx = True
115                    errdx = float(toks[3])
116                else:
117                    errdx = 0.0
118                file_x  = numpy.append(file_x, x)
119                file_y  = numpy.append(file_y, y)
120                file_dy = numpy.append(file_dy, dyerr)
121                file_dx = numpy.append(file_dx, dxerr)
122            except:
123                print "READ ERROR", line
124   
125        if has_dy == False:
126            file_dy = None
127        if has_dx == False:
128            file_dx = None
129           
130        return file_x, file_y, file_dy, file_dx
131    return None, None, None, None
132
133def load_error(error=None):
134    """
135    Pop up an error message.
136   
137    :param error: details error message to be displayed
138    """
139    message = "The data file you selected could not be loaded.\n"
140    message += "Make sure the content of your file is properly formatted.\n\n"
141   
142    if error is not None:
143        message += "When contacting the DANSE team, mention the following:\n%s" % str(error)
144   
145    dial = wx.MessageDialog(None, message, 'Error Loading File', wx.OK | wx.ICON_EXCLAMATION)
146    dial.ShowModal()   
147
148def on_load_error(parent):
149    """
150    """
151    wx.PostEvent(parent, StatusEvent(status="Load cancel..", info="warning",
152                                                type="stop"))
153   
154def plot_data(parent, path, format=None):
155    """
156    Use the DataLoader loader to created data to plot.
157   
158    :param path: the path of the data to load
159    : param format: file format (as file extension)
160    """
161    from sans.guicomm.events import NewPlotEvent, StatusEvent
162    from DataLoader.loader import  Loader
163   
164    # Instantiate a loader
165    L = Loader()
166   
167    # Load data
168    try:
169        output = L.load(path, format)
170    except:
171        load_error(sys.exc_value)
172        return
173   
174    # Notify user if the loader completed the load but no data came out
175    if output == None:
176        basename  = os.path.basename(path)
177        if  not basename.endswith('.svs'):
178            load_error("The data file appears to be empty.")
179        return
180 
181     
182    filename = os.path.basename(path)
183   
184    if not  output.__class__.__name__ == "list":
185        ## Creating a Data2D with output
186        if hasattr(output,'data'):
187            msg = "Loading 2D data: %s"%output.filename
188            wx.PostEvent(parent, StatusEvent(status=msg, info="info", type="stop"))
189            new_plot = Data2D(image=None, err_image=None)
190     
191        else:
192            msg = "Loading 1D data: %s"%output.filename
193            wx.PostEvent(parent, StatusEvent(status=msg, info="info", type="stop"))
194            new_plot = Data1D(x=[], y=[], dx=None, dy=None)
195           
196        new_plot.copy_from_datainfo(output) 
197        output.clone_without_data(clone=new_plot)     
198     
199        ## data 's name
200        if output.filename is None or output.filename == "":
201            output.filename = str(filename)
202        ## name of the data allow to differentiate data when plotted
203        name = parse_name(name=output.filename, expression="_")
204        #if not name in parent.indice_load_data.keys():
205        #    parent.indice_load_data[name] = 0
206        #else:
207            ## create a copy of the loaded data
208        #    parent.indice_load_data[name] += 1
209        #    name = name +"[%i]"%parent.indice_load_data[name]
210       
211        new_plot.name = name
212        ## allow to highlight data when plotted
213        new_plot.interactive = True
214        ## when 2 data have the same id override the 1 st plotted
215        new_plot.id = name
216        ##group_id specify on which panel to plot this data
217        new_plot.group_id = name
218        new_plot.is_data = True
219        ##post data to plot
220        title = output.filename
221        if hasattr(new_plot,"title"):
222            title = str(new_plot.title.lstrip().rstrip())
223            if title == "":
224                title = str(name)
225        else:
226            title = str(name)
227        if hasattr(parent, "panel_on_focus") and not(parent.panel_on_focus is None):
228                existing_panel  = parent.panel_on_focus
229                panel_name = existing_panel.window_caption
230                data_name = new_plot.name
231                if enable_add_data(existing_panel, new_plot):
232                    if open_dialog_append_data(panel_name, data_name):
233                        #add this plot the an existing panel
234                        new_plot.group_id = existing_panel.group_id
235        wx.PostEvent(parent, NewPlotEvent(plot=new_plot, title=title))
236        wx.PostEvent(parent,InvStateUpdateEvent())
237        wx.PostEvent(parent,FitStateUpdateEvent())
238    ## the output of the loader is a list , some xml files contain more than one data
239    else:
240       
241        i=1
242        for item in output:
243            try:
244                ## Creating a Data2D with output
245                if hasattr(item,'data'):
246                    msg = "Loading 2D data: %s"%item.filename
247                    wx.PostEvent(parent, StatusEvent(status=msg, info="info", type="stop"))
248                    new_plot = Data2D(image=None, err_image=None)
249             
250                else:
251                    msg = "Loading 1D data: %s"%str(item.run[0])
252                    wx.PostEvent(parent, StatusEvent(status=msg, info="info", type="stop"))
253                    try:
254                        dx = item.dx
255                        dxl = item.dxl
256                        dxw = item.dxw
257                    except:
258                        dx = None
259                        dxl = None
260                        dxw = None
261       
262                    new_plot = Data1D(x=item.x,y=item.y,dx=dx,dy=item.dy)
263                   
264                    new_plot.dxl = dxl
265                    new_plot.dxw = dxw
266                   
267                item.clone_without_data(clone=new_plot)   
268                new_plot.copy_from_datainfo(item)
269                name = parse_name(name=str(item.run[0]), expression="_")
270                #if not name in parent.indice_load_data.keys():
271                #    parent.indice_load_data[name] = 0
272                #else:
273                    ## create a copy of the loaded data
274                   
275                    #TODO: this is a very annoying feature. We should make this
276                    # an option. Excel doesn't do this. Why should we?
277                    # What is the requirement for this feature, and are the
278                    # counter arguments stronger? Is this feature developed
279                    # to please at least 80% of the users or a special few?
280                    #parent.indice_load_data[name] += 1
281                    #name = name + "(copy %i)"%parent.indice_load_data[name]
282                   
283                new_plot.name = name
284                new_plot.interactive = True
285                new_plot.group_id = name
286                new_plot.id = name
287                new_plot.is_data = True
288           
289                if hasattr(item,"title"):
290                    title = item.title.lstrip().rstrip()
291                    if title == "":
292                        title = str(name)
293                else:
294                    title = name
295                if hasattr(parent, "panel_on_focus") and not(parent.panel_on_focus is None):
296                    existing_panel  = parent.panel_on_focus
297                    panel_name = existing_panel.window_caption
298                    data_name = new_plot.name
299                    if enable_add_data(existing_panel, new_plot):
300                        if open_dialog_append_data(panel_name, data_name):
301                            #add this plot the an existing panel
302                            new_plot.group_id = existing_panel.group_id
303                wx.PostEvent(parent, NewPlotEvent(plot=new_plot, title=str(title)))
304                wx.PostEvent(parent,InvStateUpdateEvent())
305                wx.PostEvent(parent,FitStateUpdateEvent())
306                i+=1
307            except:
308                pass
Note: See TracBrowser for help on using the repository browser.