source: sasview/guiframe/data_loader.py @ 00561739

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 00561739 was c7bc3e7, checked in by Mathieu Doucet <doucetm@…>, 16 years ago

updated guiframe to use extension wildcards

  • Property mode set to 100644
File size: 2.6 KB
Line 
1import os, sys
2import wx
3from DataLoader.loader import Loader
4
5def choose_data_file(parent, location=None):
6    path = None
7    if location==None:
8        location = os.getcwd()
9   
10    l = Loader()
11    cards = l.get_wildcards()
12    wlist = '|'.join(cards)
13   
14    dlg = wx.FileDialog(parent, "Choose a file", location, "", wlist, wx.OPEN)
15    if dlg.ShowModal() == wx.ID_OK:
16        path = dlg.GetPath()
17        mypath = os.path.basename(path)
18    dlg.Destroy()
19   
20    return path
21
22
23def load_ascii_1D(path):
24    """
25        Load a 1D ascii file, with errors
26    """
27    import numpy
28    if path and os.path.isfile(path):
29   
30        file_x = numpy.zeros(0)
31        file_y = numpy.zeros(0)
32        file_dy = numpy.zeros(0)
33       
34        input_f = open(path,'r')
35        buff = input_f.read()
36        lines = buff.split('\n')
37       
38        has_dy = False
39       
40        for line in lines:
41            try:
42                toks = line.split()
43                x = float(toks[0])
44                y = float(toks[1])
45                if len(toks)==3:
46                    has_dy = True
47                    err = float(toks[2])
48                else:
49                    err = 0.0
50                file_x  = numpy.append(file_x, x)
51                file_y  = numpy.append(file_y, y)
52                file_dy = numpy.append(file_dy, err)
53            except:
54                print "READ ERROR", line
55   
56        if has_dy==False:
57            file_dy = None
58           
59        return file_x, file_y, file_dy
60    return None, None, None
61
62def plot_data(parent, path, name="Loaded Data"):
63    from sans.guicomm.events import NewPlotEvent, StatusEvent
64    from sans.guitools.plottables import Data1D, Theory1D
65    from DataLoader.loader import  Loader
66    import numpy
67    #Instantiate a loader
68    L=Loader()
69   
70    #Recieves data
71    try:
72        output=L.load(path)
73    except:
74        wx.PostEvent(parent, StatusEvent(status="Problem loading file: %s" % sys.exc_value))
75        return
76       
77    if output.dy==None:
78        new_plot = Theory1D(output.x, output.y)
79    else:
80        new_plot = Data1D(output.x, output.y, dy=output.dy)
81       
82    filename = os.path.basename(path)
83       
84    new_plot.name = name
85    new_plot.interactive = True
86   
87    # If the data file does not tell us what the axes are, just assume...
88    new_plot.xaxis("\\rm{Q}", 'A^{-1}')
89    new_plot.yaxis("\\rm{Intensity} ","cm^{-1}")
90    new_plot.group_id = filename
91       
92    wx.PostEvent(parent, NewPlotEvent(plot=new_plot, title=filename))
Note: See TracBrowser for help on using the repository browser.