source: sasview/guiframe/data_loader.py @ e5a9e32

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 e5a9e32 was 768656e, checked in by Gervaise Alina <gervyh@…>, 16 years ago

fixe plotting more than2 data at the same time from xml file
add another interactor not connect to code yet
  • Property mode set to 100644
File size: 4.7 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    if not  output.__class__.__name__=="list":
82        try:
83            dxl=output.dxl
84            dxw=output.dxw
85        except:
86            dxl=None
87            dxw=None
88        if hasattr(output,'data'):
89            new_plot = Data2D(image=output.data,err_image=output.err_data,
90                              xmin=output.xmin,xmax=output.xmax,
91                              ymin=output.ymin,ymax=output.ymax)
92            new_plot.x_bins=output.x_bins
93            new_plot.y_bins=output.y_bins
94            #print "data_loader",output
95        else:
96            print "output.dx, output.dy",output.dx, output.dy
97            if output.dy ==None:
98                print "went here theory1D"
99                new_plot = Theory1D(output.x,output.y, dxl, dxw)
100            else:
101                new_plot = Data1D(x=output.x,y=output.y,dy=output.dy, dxl=dxl, dxw=dxw)
102        #print "dataloader",output[0],output[1]
103       
104        new_plot.source=output.source
105        new_plot.name = filename
106        new_plot.interactive = True
107        new_plot.info= output
108        if hasattr(output, "dxl"):
109            new_plot.dxl = output.dxl
110        if hasattr(output, "dxw"):
111            new_plot.dxw = output.dxw
112        #print "loader output.detector",output.source
113        new_plot.detector =output.detector
114       
115        # If the data file does not tell us what the axes are, just assume...
116        new_plot.xaxis(output._xaxis,output._xunit)
117        new_plot.yaxis(output._yaxis,output._yunit)
118        new_plot.group_id = filename
119        wx.PostEvent(parent, NewPlotEvent(plot=new_plot, title=filename))
120    else:
121        i=1
122        for item in output:
123            try:
124                dxl=item.dxl
125                dxw=item.dxw
126            except:
127                dxl=None
128                dxw=None
129            if item.dy ==None:
130                new_plot = Theory1D(item.x,item.y,dxl,dxw)
131            else:
132                new_plot = Data1D(x=item.x,y=item.y,dy=item.dy,dxl=dxl,dxw=dxw)
133           
134            new_plot.source=item.source
135            new_plot.info=output
136            new_plot.name = filename+" "+ str(i)
137            new_plot.interactive = True
138           
139            #print "loader output.detector",output.source
140            new_plot.detector =item.detector
141            # If the data file does not tell us what the axes are, just assume...
142            new_plot.xaxis(item._xaxis,item._xunit)
143            new_plot.yaxis(item._yaxis,item._yunit)
144            new_plot.group_id = filename
145            wx.PostEvent(parent, NewPlotEvent(plot=new_plot, title=filename))
146            i+=1
147           
148           
Note: See TracBrowser for help on using the repository browser.