| 1 | import os, sys |
|---|
| 2 | import wx |
|---|
| 3 | from dataFitting import Data1D, Theory1D |
|---|
| 4 | from danse.common.plottools.plottables import Data2D |
|---|
| 5 | |
|---|
| 6 | from DataLoader.loader import Loader |
|---|
| 7 | |
|---|
| 8 | def 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 | |
|---|
| 26 | def 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 | |
|---|
| 65 | def 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 1 \ |
|---|
| 94 | created for : %s"% str(temp[temp==0]))) |
|---|
| 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 | #print "went here theory1D" |
|---|
| 105 | new_plot = Theory1D(output.x,output.y, dxl, dxw) |
|---|
| 106 | else: |
|---|
| 107 | new_plot = Data1D(x=output.x,y=output.y,dy=output.dy, dxl=dxl, dxw=dxw) |
|---|
| 108 | #print "dataloader",output[0],output[1] |
|---|
| 109 | |
|---|
| 110 | new_plot.source=output.source |
|---|
| 111 | new_plot.name = output.filename |
|---|
| 112 | new_plot.interactive = True |
|---|
| 113 | new_plot.id = True |
|---|
| 114 | new_plot.info= output |
|---|
| 115 | if hasattr(output, "dxl"): |
|---|
| 116 | new_plot.dxl = output.dxl |
|---|
| 117 | if hasattr(output, "dxw"): |
|---|
| 118 | new_plot.dxw = output.dxw |
|---|
| 119 | #print "loader output.detector",output.source |
|---|
| 120 | new_plot.detector =output.detector |
|---|
| 121 | |
|---|
| 122 | # If the data file does not tell us what the axes are, just assume... |
|---|
| 123 | new_plot.xaxis(output._xaxis,output._xunit) |
|---|
| 124 | new_plot.yaxis(output._yaxis,output._yunit) |
|---|
| 125 | new_plot.group_id = output.filename |
|---|
| 126 | new_plot.id = output.filename |
|---|
| 127 | wx.PostEvent(parent, NewPlotEvent(plot=new_plot, title=filename)) |
|---|
| 128 | else: |
|---|
| 129 | i=1 |
|---|
| 130 | for item in output: |
|---|
| 131 | |
|---|
| 132 | try: |
|---|
| 133 | dxl=item.dxl |
|---|
| 134 | dxw=item.dxw |
|---|
| 135 | except: |
|---|
| 136 | dxl=None |
|---|
| 137 | dxw=None |
|---|
| 138 | if item.dy ==None: |
|---|
| 139 | new_plot = Theory1D(item.x,item.y,dxl,dxw) |
|---|
| 140 | else: |
|---|
| 141 | new_plot = Data1D(x=item.x,y=item.y,dy=item.dy,dxl=dxl,dxw=dxw) |
|---|
| 142 | |
|---|
| 143 | new_plot.source=item.source |
|---|
| 144 | #new_plot.info=output |
|---|
| 145 | new_plot.name = str(item.run[0]) |
|---|
| 146 | new_plot.interactive = True |
|---|
| 147 | |
|---|
| 148 | #print "loader output.detector",output.source |
|---|
| 149 | new_plot.detector =item.detector |
|---|
| 150 | # If the data file does not tell us what the axes are, just assume... |
|---|
| 151 | new_plot.xaxis(item._xaxis,item._xunit) |
|---|
| 152 | new_plot.yaxis(item._yaxis,item._yunit) |
|---|
| 153 | new_plot.group_id = str(item.run[0]) |
|---|
| 154 | new_plot.id = str(item.run[0]) |
|---|
| 155 | |
|---|
| 156 | wx.PostEvent(parent, NewPlotEvent(plot=new_plot, title=filename)) |
|---|
| 157 | i+=1 |
|---|
| 158 | |
|---|
| 159 | |
|---|
