source: sasview/guiframe/data_loader.py @ 675e0ab

import os, sys,numpy
import wx
from dataFitting import Data1D, Theory1D
from danse.common.plottools.plottables 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.
        
        TODO: this method needs a major cleanup. 
        It's a complete mess. It needs:
          - to be cleaned of code duplication
          - to be cleaned of data duplication: we should not
            need to reference all the Data1D data members.
            It should be sufficient to use the Data1D class
            defined in dataFitting (which btw needs a better name).
        
        @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)
        
    except:
        wx.PostEvent(parent, StatusEvent(status="Problem loading file: %s" % sys.exc_value))
        return
    if output==None:
        msg="Could not open this page"
        wx.PostEvent(parent, StatusEvent(status=msg))
        return
    filename = os.path.basename(path)
    
    if not  output.__class__.__name__=="list":
        ## smearing info
        try:
            dxl = output.dxl
            dxw = output.dxw
        except:
            dxl = None
            dxw = None
       
            
        ## Creating a Data2D with output
        if hasattr(output,'data'):
            
            new_plot = Data2D(image=output.data,err_image=output.err_data,
                              xmin=output.xmin,xmax=output.xmax,
                              ymin=output.ymin,ymax=output.ymax)
            new_plot.x_bins=output.x_bins
            new_plot.y_bins=output.y_bins
            
        ##Creating Data1D with output
        else:
            ##dy values checked
            dy= output.dy
            if dy ==None:
                dy= numpy.zeros(len(output.y))
            
            msg="Loading 1D data: "
            wx.PostEvent(parent, StatusEvent(status= "%s %s"%(msg, output.filename)))
            new_plot = Data1D(x=output.x, y=output.y, dx=output.dx,
                              dy= dy, dxl=dxl, dxw=dxw)
            # Copy all the data to the new object so that the 
            # new_plot object properly behaves as a data_info.Data1D.
            # This also means that we should not have to copy all
            # the data below...
            output.clone_without_data(clone=new_plot)
                
        ## source will request in dataLoader .manipulation module
        new_plot.source=output.source
        ## data 's name
        if output.filename==None:
            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 +"(copy %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
        ## detector used by dataLoader.manipulation module
        new_plot.detector =output.detector
        ## If the data file does not tell us what the axes are, just assume...
        new_plot.xaxis(output._xaxis,output._xunit)
        new_plot.yaxis(output._yaxis,output._yunit)
        ##group_id specify on which panel to plot this data
        new_plot.group_id = name
        new_plot.is_data =True
        ##post data to plot
        wx.PostEvent(parent, NewPlotEvent(plot=new_plot, title=str(name)))
        
    ## the output of the loader is a list , some xml files contain more than one data
    #TODO: refactor this so that there is no duplication of code.
    # There is no reason why the code above shouldn't be put in 
    # a private method to be used within the loop below.
    else:
        i=1
        for item in output:
            try:
                dx=item.dx
                dxl=item.dxl
                dxw=item.dxw
            except:
                dx=None
                dxl=None
                dxw=None
            dy = item.dy  
            if dy ==None:
                dy= numpy.zeros(len(item.y))
               
            new_plot = Data1D(x=item.x,y=item.y,dx=dx,dy=item.dy,dxl=dxl,dxw=dxw)
            # Copy all the data to the new object
            item.clone_without_data(clone=new_plot)
            new_plot.source=item.source
            
            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.detector =item.detector
            # If the data file does not tell us what the axes are, just assume...
            new_plot.xaxis(item._xaxis,item._xunit)
            new_plot.yaxis(item._yaxis,item._yunit)
            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
            else:
                title= name
            wx.PostEvent(parent, NewPlotEvent(plot=new_plot, title=str(title)))
            i+=1