source: sasview/DataLoader/plugins/TXT2_Reader.py @ c45976b

# class Loader  to load any king of file
import wx
import string,numpy
import logging
class Reader:
    """
        This class is loading values from given file or value giving by the user
        should be able to read only 2 columns of data
    """
    ext=['.txt','.dat'] 
    def _init_(self,x=None,y=None,dx=None,dy=None):
        # variable to store loaded values
        self.x = x
        self.y = y
        self.dx = dx
        self.dy = dy
        
    def read(self,path, format=None):
        """ Store the values loaded from file in local variables """
        if not path == None:
            read_it = False
             
            for item in self.ext:
                if path.lower().find(item)>=0:
                    read_it = True
            #print "this is the flag",read_it, path.lower()
            if read_it==False:
                return None
            else:
                try:
                    input_f =  open(path,'r')
                except :
                    raise  RuntimeError,"TXT2_Reader cannot open %s"%(path)
                buff = input_f.read()
                lines = buff.split('\n')
                self.x=[]
                self.y=[]
                self.dx = [] 
                self.dy=[]
                for line in lines:
                    toks = line.split()
                    try:
                        x = float(toks[0])
                        self.x.append(x)
                    except:
                        pass
                        #print "READ ERROR", line
                    try:
                        y = float(toks[1])
                        self.y.append(y)
                    except:
                        pass
                        #print "READ ERROR", line
                    try:
                        dy=float(toks[2])
                        self.dy.append(dy)
    
                    except:
                        pass
                        #print "READ ERROR", line
    
                if(( self.x==[])or (self.y==[])) or (self.dy !=[]):
                    #print "went here"
                    #return value
                    raise ValueError, "TXT2_Reader can't read %s"%path
                else:
                    #msg="R1 reading: \n"+"this x :"+ str(self.x) +"\n"+"this y:"+str(self.y)+"\n"+"this dy :"+str(self.dy)+"\n"
                    #return msg
                    logging.info("TXT2_Reader reading  %s\n" %path)
                    return self.x,self.y,self.dy
        return None
   
       
    
if __name__ == "__main__": 
    read= Reader()
    read.load("testdata_line.txt")