source: sasview/sansguiframe/src/sans/guiframe/local_perspectives/data_loader/load_thread.py @ e9d9081

import time
import sys
import os

from data_util.calcthread import CalcThread

        
EXTENSIONS = ['.svs', '.prv', '.inv', '.fitv']

class DataReader(CalcThread):
    """
    Load a data given a filename
    """
    def __init__(self, path, loader,
                 flag=True,
                 transform_data=None,
                 completefn=None,
                 updatefn   = None,
                 yieldtime  = 0.01,
                 worktime   = 0.01
                 ):
        CalcThread.__init__(self, completefn,
                 updatefn,
                 yieldtime,
                 worktime)
        self.load_state_flag = flag
        self.transform_data = transform_data
        self.list_path = path
        #Instantiate a loader 
        self.loader = loader
        self.message = ""
        self.starttime = 0  
        self.updatefn = updatefn
        
    def isquit(self):
        """
        :raise KeyboardInterrupt: when the thread is interrupted
        """
        try:
            CalcThread.isquit(self)
        except KeyboardInterrupt:
            raise KeyboardInterrupt   
        
        
    def compute(self):
        """
        read some data
        """
        self.starttime = time.time()
        output = []
        error_message = ""
        for path in self.list_path:
            basename  = os.path.basename(path)
            root, extension = os.path.splitext(basename)
            if self.load_state_flag:
                if extension.lower() in EXTENSIONS:
                    pass
            else:
                if extension.lower() not in EXTENSIONS:
                    pass
            try:
                temp =  self.loader.load(path)
                elapsed = time.time() - self.starttime
                if temp.__class__.__name__ == "list":
                    for item in temp:
                        data = self.transform_data(item, path)
                        output.append(data)
                else:
                    data = self.transform_data(temp, path)
                    output.append(data)
                message = "Loading ..." + str(path) + "\n"
                if self.updatefn is not None:
                    self.updatefn(output=output, message=message)
            except:
                error_message = "Error while loading: %s\n" % str(path)
                error_message += str(sys.exc_value) + "\n"
                self.updatefn(output=output, message=error_message)
                
        message = "Loading Complete!"
        self.complete(output=output, error_message=error_message,
                       message=message, path=self.list_path)