source: sasview/guiframe/data_loader.py @ 1acbf0a

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 1acbf0a was 8bd764d, checked in by Gervaise Alina <gervyh@…>, 16 years ago

context menu improve for xml

  • Property mode set to 100644
File size: 5.6 KB
Line 
1import os, sys
2import wx
3from dataFitting import Data1D, Theory1D
4from danse.common.plottools.plottables import Data2D
5
6from DataLoader.loader import Loader
7
8def choose_data_file(parent, location=None):
9    path = None
10    if location==None:
11        location = os.getcwd()
12   
13    l = Loader()
14    cards = l.get_wildcards()
15    wlist = '|'.join(cards)
16   
17    dlg = wx.FileDialog(parent, "Choose a file", location, "", wlist, wx.OPEN)
18    if dlg.ShowModal() == wx.ID_OK:
19        path = dlg.GetPath()
20        mypath = os.path.basename(path)
21    dlg.Destroy()
22   
23    return path
24
25
26def load_ascii_1D(path):
27    """
28        Load a 1D ascii file, with errors
29    """
30    import numpy
31    if path and os.path.isfile(path):
32   
33        file_x = numpy.zeros(0)
34        file_y = numpy.zeros(0)
35        file_dy = numpy.zeros(0)
36       
37        input_f = open(path,'r')
38        buff = input_f.read()
39        lines = buff.split('\n')
40       
41        has_dy = False
42       
43        for line in lines:
44            try:
45                toks = line.split()
46                x = float(toks[0])
47                y = float(toks[1])
48                if len(toks)==3:
49                    has_dy = True
50                    err = float(toks[2])
51                else:
52                    err = 0.0
53                file_x  = numpy.append(file_x, x)
54                file_y  = numpy.append(file_y, y)
55                file_dy = numpy.append(file_dy, err)
56            except:
57                print "READ ERROR", line
58   
59        if has_dy==False:
60            file_dy = None
61           
62        return file_x, file_y, file_dy
63    return None, None, None
64
65def plot_data(parent, path, name="Loaded Data"):
66    from sans.guicomm.events import NewPlotEvent, StatusEvent
67   
68    from DataLoader.loader import  Loader
69    import numpy
70    #Instantiate a loader
71    L = Loader()
72   
73    #Recieves data
74    try:
75        output=L.load(path)
76       
77    except:
78        wx.PostEvent(parent, StatusEvent(status="Problem loading file: %s" % sys.exc_value))
79        return
80    filename = os.path.basename(path)
81   
82    if not  output.__class__.__name__=="list":
83        try:
84            dxl=output.dxl
85            dxw=output.dxw
86        except:
87            dxl=None
88            dxw=None
89        if hasattr(output,'data'):
90            temp = output.err_data
91            temp[temp==0]=1
92
93            wx.PostEvent(parent, StatusEvent(status="Loading 2D error bars of value 1 \
94            were added to : %s"% output.filename))
95            new_plot = Data2D(image=output.data,err_image=temp,
96                              xmin=output.xmin,xmax=output.xmax,
97                              ymin=output.ymin,ymax=output.ymax)
98            new_plot.x_bins=output.x_bins
99            new_plot.y_bins=output.y_bins
100            #print "data_loader",output
101        else:
102            #print "output.dx, output.dy",output.dx, output.dy
103            if output.dy ==None :
104                new_plot = Theory1D(output.x,output.y, dxl, dxw)
105            elif len(output.dy[output.dy==0])==len(output.dy):
106                output.dy[output.dy==0]=1 
107                wx.PostEvent(parent, StatusEvent(status="Loading 1D error bars of value 1 \
108                    were added to : %s"%output.filename))
109                new_plot = Theory1D(output.x,output.y,output.dy, dxl, dxw)
110            else:
111                   
112                new_plot = Data1D(x=output.x,y=output.y,dx=output.dx,dy=output.dy, dxl=dxl, dxw=dxw)
113       
114        new_plot.source=output.source
115        new_plot.name = output.filename
116        new_plot.interactive = True
117        new_plot.id = True
118        new_plot.info= output
119        if hasattr(output, "dxl"):
120            new_plot.dxl = output.dxl
121        if hasattr(output, "dxw"):
122            new_plot.dxw = output.dxw
123        #print "loader output.detector",output.source
124        new_plot.detector =output.detector
125       
126        # If the data file does not tell us what the axes are, just assume...
127        new_plot.xaxis(output._xaxis,output._xunit)
128        new_plot.yaxis(output._yaxis,output._yunit)
129        new_plot.group_id = output.filename
130        new_plot.id = output.filename
131        try:
132            title=output.filename
133        except:
134            title= filename
135       
136        wx.PostEvent(parent, NewPlotEvent(plot=new_plot, title=str(title)))
137    else:
138        i=1
139        for item in output:
140           
141            try:
142                dxl=item.dxl
143                dxw=item.dxw
144            except:
145                dxl=None
146                dxw=None
147            if item.dy ==None:
148                new_plot = Theory1D(item.x,item.y,dxl,dxw)
149            else:
150                new_plot = Data1D(x=item.x,y=item.y,dx=item.dx,dy=item.dy,dxl=dxl,dxw=dxw)
151           
152            new_plot.source=item.source
153            #new_plot.info=output
154            new_plot.name = str(item.run[0])
155            new_plot.interactive = True
156           
157            #print "loader output.detector",output.source
158            new_plot.detector =item.detector
159            # If the data file does not tell us what the axes are, just assume...
160            new_plot.xaxis(item._xaxis,item._xunit)
161            new_plot.yaxis(item._yaxis,item._yunit)
162            new_plot.group_id = str(item.run[0])
163            new_plot.id = str(item.run[0])
164            if hasattr(item,"title"):
165                title= item.title
166            else:
167                title= str(item.run[0])
168            wx.PostEvent(parent, NewPlotEvent(plot=new_plot, title=str(title)))
169            i+=1
170           
171           
Note: See TracBrowser for help on using the repository browser.