source: sasview/guiframe/data_loader.py @ 0881f51

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 0881f51 was 0881f51, checked in by Gervaise Alina <gervyh@…>, 14 years ago

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