source: sasview/fittingview/src/sans/perspectives/fitting/fit_thread.py @ 56acb5d

ESS_GUIESS_GUI_DocsESS_GUI_batch_fittingESS_GUI_bumps_abstractionESS_GUI_iss1116ESS_GUI_iss879ESS_GUI_iss959ESS_GUI_openclESS_GUI_orderingESS_GUI_sync_sascalccostrafo411magnetic_scattrelease-4.1.1release-4.1.2release-4.2.2release_4.0.1ticket-1009ticket-1094-headlessticket-1242-2d-resolutionticket-1243ticket-1249ticket885unittest-saveload
Last change on this file since 56acb5d was aff2913, checked in by Jae Cho <jhjcho@…>, 13 years ago

Add KeyInterrupt? error msg

  • Property mode set to 100644
File size: 3.6 KB
RevLine 
[e9b4cc4]1
[66ff250]2import sys
[2bb37c3]3import time
[e54d2c32]4from data_util.calcthread import CalcThread
[e9b4cc4]5
[cc31608]6def map_getattr(classInstance, classFunc, *args):
7    """
8    Take an instance of a class and a function name as a string.
9    Execute class.function and return result
10    """
11    return  getattr(classInstance,classFunc)(*args)
12
13def map_apply(arguments):
14    return apply(arguments[0], arguments[1:])
[66ff250]15
[e9b4cc4]16class FitThread(CalcThread):
[ed2ea6a]17    """Thread performing the fit """
[e9b4cc4]18   
[58e0c83]19    def __init__(self, 
[e54d2c32]20                  fn,
[66ff250]21                  page_id,
[e54d2c32]22                   handler,
[33dd2e5]23                    batch_outputs,
24                    batch_inputs=None,             
[e54d2c32]25                  pars=None,
[e9b4cc4]26                 completefn = None,
27                 updatefn   = None,
28                 yieldtime  = 0.01,
[2296316]29                 worktime   = 0.01,
[7db52f1]30                 ftol       = None,
31                 reset_flag = False):
[e9b4cc4]32        CalcThread.__init__(self,completefn,
33                 updatefn,
34                 yieldtime,
35                 worktime)
[e54d2c32]36        self.handler = handler
37        self.fitter = fn
[e9b4cc4]38        self.pars = pars
[33dd2e5]39        self.batch_inputs = batch_inputs
40        self.batch_outputs = batch_outputs
[66ff250]41        self.page_id = page_id
[2bb37c3]42        self.starttime = time.time()
[e54d2c32]43        self.updatefn = updatefn
[2296316]44        #Relative error desired in the sum of squares.
45        self.ftol = ftol
[7db52f1]46        self.reset_flag = reset_flag
[e54d2c32]47   
[e9b4cc4]48    def isquit(self):
[ed2ea6a]49        """
[5062bbf]50        :raise KeyboardInterrupt: when the thread is interrupted
51       
[ed2ea6a]52        """
[e9b4cc4]53        try:
54            CalcThread.isquit(self)
55        except KeyboardInterrupt:
[aff2913]56            msg = "Fitting: terminated by the user."
57            raise KeyboardInterrupt, msg
[66ff250]58       
[e9b4cc4]59    def compute(self):
[ed2ea6a]60        """
[5062bbf]61        Perform a fit
[ed2ea6a]62        """
[40953a9]63        msg = ""
[cc31608]64        try:
[33dd2e5]65            import copy
[cc31608]66            list_handler = []
67            list_curr_thread = [] 
68            list_ftol = []
[7db52f1]69            list_reset_flag = []
[cc31608]70            list_map_get_attr = []
71            list_fit_function = []
72            list_q = []
73            for i in range(len(self.fitter)):
[c935fac]74                list_handler.append(self.handler)
[cc31608]75                list_q.append(None)
[9c9a224]76                list_curr_thread.append(self)
[cc31608]77                list_ftol.append(self.ftol)
[7db52f1]78                list_reset_flag.append(self.reset_flag)
[cc31608]79                list_fit_function.append('fit')
80                list_map_get_attr.append(map_getattr)
[8d57bdb]81            #from multiprocessing import Pool
[67ae937]82            inputs = zip(list_map_get_attr,self.fitter, list_fit_function,
[c935fac]83                          list_q, list_q, list_handler,list_curr_thread,list_ftol,
[7db52f1]84                         list_reset_flag)
[8d57bdb]85            result =  map(map_apply, inputs)
[2bb37c3]86   
[33dd2e5]87            self.complete(result=result,
88                          batch_inputs=self.batch_inputs,
89                           batch_outputs=self.batch_outputs,
[66ff250]90                          page_id=self.page_id,
[2bb37c3]91                          pars = self.pars,
92                          elapsed=time.time()-self.starttime)
[785c8233]93           
[e54d2c32]94        except KeyboardInterrupt, msg:
[e9b4cc4]95            # Thread was interrupted, just proceed and re-raise.
96            # Real code should not print, but this is an example...
[ad6dd4c]97            #print "keyboard exception"
[67ae937]98            #Stop on exception during fitting. Todo: need to put
99            #some mssg and reset progress bar.
[2bb37c3]100           
101            if self.handler is not None:
102                self.handler.error(msg=msg)
[66ff250]103        except:
[f7f6886]104            if self.handler is not None:
105                self.handler.error(msg=str(sys.exc_value))
[66ff250]106           
107       
108   
Note: See TracBrowser for help on using the repository browser.