import sys import time from data_util.calcthread import CalcThread def map_getattr(classInstance, classFunc, *args): """ Take an instance of a class and a function name as a string. Execute class.function and return result """ return getattr(classInstance,classFunc)(*args) def map_apply(arguments): return apply(arguments[0], arguments[1:]) class FitThread(CalcThread): """Thread performing the fit """ def __init__(self, fn, page_id, handler, batch_outputs, batch_inputs=None, pars=None, completefn = None, updatefn = None, yieldtime = 0.03, worktime = 0.03, ftol = None, reset_flag = False): CalcThread.__init__(self,completefn, updatefn, yieldtime, worktime) self.handler = handler self.fitter = fn self.pars = pars self.batch_inputs = batch_inputs self.batch_outputs = batch_outputs self.page_id = page_id self.starttime = time.time() self.updatefn = updatefn #Relative error desired in the sum of squares. self.ftol = ftol self.reset_flag = reset_flag def isquit(self): """ :raise KeyboardInterrupt: when the thread is interrupted """ try: CalcThread.isquit(self) except KeyboardInterrupt: msg = "Fitting: terminated by the user." raise KeyboardInterrupt, msg def compute(self): """ Perform a fit """ msg = "" try: import copy list_handler = [] list_curr_thread = [] list_ftol = [] list_reset_flag = [] list_map_get_attr = [] list_fit_function = [] list_q = [] for i in range(len(self.fitter)): list_handler.append(self.handler) list_q.append(None) list_curr_thread.append(self) list_ftol.append(self.ftol) list_reset_flag.append(self.reset_flag) list_fit_function.append('fit') list_map_get_attr.append(map_getattr) #from multiprocessing import Pool inputs = zip(list_map_get_attr,self.fitter, list_fit_function, list_q, list_q, list_handler,list_curr_thread,list_ftol, list_reset_flag) result = map(map_apply, inputs) self.complete(result=result, batch_inputs=self.batch_inputs, batch_outputs=self.batch_outputs, page_id=self.page_id, pars = self.pars, elapsed=time.time()-self.starttime) except KeyboardInterrupt, msg: # Thread was interrupted, just proceed and re-raise. # Real code should not print, but this is an example... #print "keyboard exception" #Stop on exception during fitting. Todo: need to put #some mssg and reset progress bar. if self.handler is not None: self.handler.stop(msg=msg) except: if self.handler is not None: self.handler.error(msg=str(sys.exc_value))