import os, sys,numpy import wx from dataFitting import Data1D from dataFitting import Data2D from DataLoader.loader import Loader def choose_data_file(parent, location=None): path = None if location==None: location = os.getcwd() l = Loader() cards = l.get_wildcards() wlist = '|'.join(cards) dlg = wx.FileDialog(parent, "Choose a file", location, "", wlist, wx.OPEN) if dlg.ShowModal() == wx.ID_OK: path = dlg.GetPath() mypath = os.path.basename(path) dlg.Destroy() return path def load_ascii_1D(path): """ Load a 1D ascii file, with errors """ if path and os.path.isfile(path): file_x = numpy.zeros(0) file_y = numpy.zeros(0) file_dy = numpy.zeros(0) file_dx = numpy.zeros(0) input_f = open(path,'r') buff = input_f.read() lines = buff.split('\n') has_dy = False has_dx = False for line in lines: try: toks = line.split() x = float(toks[0]) y = float(toks[1]) if len(toks)==3: has_dy = True errdy = float(toks[2]) else: errdy = 0.0 if len(toks)==4: has_dx = True errdx = float(toks[3]) else: errdx = 0.0 file_x = numpy.append(file_x, x) file_y = numpy.append(file_y, y) file_dy = numpy.append(file_dy, dyerr) file_dx = numpy.append(file_dx, dxerr) except: print "READ ERROR", line if has_dy==False: file_dy = None if has_dx==False: file_dx = None return file_x, file_y, file_dy, file_dx return None, None, None, None def plot_data(parent, path): """ Use the DataLoader loader to created data to plot. @param path: the path of the data to load """ from sans.guicomm.events import NewPlotEvent, StatusEvent from DataLoader.loader import Loader #Instantiate a loader L = Loader() #Recieves data try: output=L.load(path) if output==None: msg="Could not open this page" wx.PostEvent(parent, StatusEvent(status=msg)) return except: wx.PostEvent(parent, StatusEvent(status="Problem loading file: %s" % sys.exc_value)) return filename = os.path.basename(path) if not output.__class__.__name__=="list": ## Creating a Data2D with output if hasattr(output,'data'): new_plot = Data2D(image=None,err_image=None) else: msg="Loading 1D data: " wx.PostEvent(parent, StatusEvent(status= "%s %s"%(msg, output.filename))) new_plot = Data1D(x=[], y=[], dx=None,dy= None) new_plot.copy_from_datainfo(output) output.clone_without_data(clone=new_plot) ## data 's name if output.filename==None or output.filename=="": output.filename = str(filename) ## name of the data allow to differentiate data when plotted name= output.filename if not name in parent.indice_load_data.keys(): parent.indice_load_data[name]=0 else: ## create a copy of the loaded data parent.indice_load_data[name]+=1 name = name +"[%i]"%parent.indice_load_data[name] new_plot.name = name ## allow to highlight data when plotted new_plot.interactive = True ## when 2 data have the same id override the 1 st plotted new_plot.id = name ## info is a reference to output of dataloader that can be used ## to save data 1D as cansas xml file new_plot.info= output ##group_id specify on which panel to plot this data new_plot.group_id = name new_plot.is_data =True ##post data to plot if hasattr(new_plot,"title"): title= str(new_plot.title) if title =="": title=str(name) else: title = str(name) wx.PostEvent(parent, NewPlotEvent(plot=new_plot, title= title )) ## the output of the loader is a list , some xml files contain more than one data else: i=1 for item in output: try: dx=item.dx dxl=item.dxl dxw=item.dxw except: dx=None dxl=None dxw=None new_plot = Data1D(x=item.x,y=item.y,dx=dx,dy=item.dy) new_plot.copy_from_datainfo(item) item.clone_without_data(clone=new_plot) new_plot.dxl = dxl new_plot.dxw = dxw name= str(item.run[0]) if not name in parent.indice_load_data.keys(): parent.indice_load_data[name]=0 else: ## create a copy of the loaded data #TODO: this is a very annoying feature. We should make this # an option. Excel doesn't do this. Why should we? # What is the requirement for this feature, and are the # counter arguments stronger? Is this feature developed # to please at least 80% of the users or a special few? parent.indice_load_data[name]+=1 name = name +"(copy %i)"%parent.indice_load_data[name] new_plot.name = name new_plot.interactive = True new_plot.group_id = name new_plot.id = name new_plot.info= item new_plot.is_data =True if hasattr(item,"title"): title= item.title if title=="": title=str(name) else: title= name wx.PostEvent(parent, NewPlotEvent(plot=new_plot, title=str(title))) i+=1