source: sasview/src/sas/sasgui/guiframe/local_perspectives/data_loader/load_thread.py @ 2d9526d

"""
    Loading thread
"""
import time
import sys
import os

from sas.sascalc.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)
            _, 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)
                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)