import time
import sys
import numpy
import math
from sans.models.smearing_2d import Smearer2D
from data_util.calcthread import CalcThread

class Calc2D(CalcThread):
    """
    Compute 2D model
    This calculation assumes a 2-fold symmetry of the model
    where points are computed for one half of the detector
    and I(qx, qy) = I(-qx, -qy) is assumed.
    """
    def __init__(self, data, model, smearer, qmin, qmax,  page_id,
                 state=None,
                 toggle_mode_on=False,
                 completefn=None,
                 updatefn=None,
                 update_chisqr=True,
                 yieldtime=0.04,
                 worktime=0.04
                 ):
        CalcThread.__init__(self,completefn,
                 updatefn,
                 yieldtime,
                 worktime)
        self.qmin = qmin
        self.qmax = qmax
        #self.qstep = qstep
        self.toggle_mode_on = toggle_mode_on
        self.data = data
        self.page_id = page_id
        self.state = None
        # the model on to calculate
        self.model = model
        self.smearer = smearer
        self.starttime = 0  
        self.update_chisqr = update_chisqr
        
    def compute(self):
        """
        Compute the data given a model function
        """
        self.starttime = time.time()
        # Determine appropriate q range
        if self.qmin == None:
            self.qmin = 0
        if self.qmax == None:
            if self.data != None:
                newx = math.pow(max(math.fabs(self.data.xmax),
                                   math.fabs(self.data.xmin)), 2)
                newy = math.pow(max(math.fabs(self.data.ymax),
                                   math.fabs(self.data.ymin)), 2)
                self.qmax = math.sqrt(newx + newy)
        
        if self.data is None:
            msg = "Compute Calc2D receive data = %s.\n" % str(self.data)    
            raise ValueError, msg
            
        # Define matrix where data will be plotted 
        radius= numpy.sqrt((self.data.qx_data * self.data.qx_data) + \
                    (self.data.qy_data * self.data.qy_data))
        index_data = (self.qmin <= radius) & self.data.mask

        # For theory, qmax is based on 1d qmax 
        # so that must be mulitified by sqrt(2) to get actual max for 2d
        index_model = (self.qmin <= radius) & (radius <= self.qmax)
        index_model = index_model & self.data.mask
        index_model = index_model & numpy.isfinite(self.data.data)
      
        if self.smearer is not None:
            # Set smearer w/ data, model and index.
            fn = self.smearer
            fn.set_model(self.model)
            fn.set_index(index_model)
            # Get necessary data from self.data and set the data for smearing
            fn.get_data()
            # Calculate smeared Intensity 
            #(by Gaussian averaging): DataLoader/smearing2d/Smearer2D()
            value = fn.get_value()
        else:    
            # calculation w/o smearing
            value =  self.model.evalDistribution([self.data.qx_data[index_model],
                                                self.data.qy_data[index_model]])
        output = numpy.zeros(len(self.data.qx_data))
        # output default is None
        # This method is to distinguish between masked 
        #point(nan) and data point = 0.
        output = output/output
        # set value for self.mask==True, else still None to Plottools
        output[index_model] = value 
        elapsed = time.time()-self.starttime
        self.complete(image=output,
                       data=self.data, 
                       page_id=self.page_id,
                       model=self.model,
                       state=self.state,
                       toggle_mode_on=self.toggle_mode_on,
                       elapsed=elapsed,
                       index=index_model,
                       qmin=self.qmin,
                       qmax=self.qmax,
                       #qstep=self.qstep,
                       update_chisqr = self.update_chisqr)
        

class Calc1D(CalcThread):
    """
    Compute 1D data
    """
    def __init__(self, model,
                 page_id,
                 data,
                 qmin=None,
                 qmax=None,
                 smearer=None,
                 toggle_mode_on=False,
                 state=None,
                 completefn = None,
                 update_chisqr=True,
                 updatefn=None,
                 yieldtime=0.01,
                 worktime=0.01
                 ):
        """
        """
        CalcThread.__init__(self,completefn,
                 updatefn,
                 yieldtime,
                 worktime)
        self.data = data
        self.qmin = qmin
        self.qmax = qmax
        self.model = model
        self.toggle_mode_on = toggle_mode_on
        self.state = state
        self.page_id = page_id
        self.smearer = smearer
        self.starttime = 0
        self.update_chisqr = update_chisqr
        
    def compute(self):
        """
        Compute model 1d value given qmin , qmax , x value 
        """
        self.starttime = time.time()
        output = numpy.zeros((len(self.data.x)))
        index= (self.qmin <= self.data.x)& (self.data.x <= self.qmax)
     
        ##smearer the ouput of the plot    
        if self.smearer is not None:
            first_bin, last_bin = self.smearer.get_bin_range(self.qmin, 
                                                             self.qmax)
            mask = self.data.x[first_bin:last_bin]
            output[first_bin:last_bin] = self.model.evalDistribution(mask)
            output = self.smearer(output, first_bin, last_bin) 
        else:
            output[index] = self.model.evalDistribution(self.data.x[index])
         
        elapsed = time.time() - self.starttime
       
        self.complete(x=self.data.x[index], y=output[index], 
                      page_id=self.page_id,
                      state=self.state,
                      toggle_mode_on=self.toggle_mode_on,
                      elapsed=elapsed,index=index, model=self.model,
                      data=self.data, 
                      update_chisqr = self.update_chisqr)
        
    def results(self):
        """
        Send resuts of the computation
        """
        return [self.out, self.index]

"""
Example: ::
                     
    class CalcCommandline:
        def __init__(self, n=20000):
            #print thread.get_ident()
            from sans.models.CylinderModel import CylinderModel
            
            model = CylinderModel()
            
             
            print model.runXY([0.01, 0.02])
            
            qmax = 0.01
            qstep = 0.0001
            self.done = False
            
            x = numpy.arange(-qmax, qmax+qstep*0.01, qstep)
            y = numpy.arange(-qmax, qmax+qstep*0.01, qstep)
        
        
            calc_thread_2D = Calc2D(x, y, None, model.clone(),None,
                                    -qmax, qmax,qstep,
                                            completefn=self.complete,
                                            updatefn=self.update ,
                                            yieldtime=0.0)
         
            calc_thread_2D.queue()
            calc_thread_2D.ready(2.5)
            
            while not self.done:
                time.sleep(1)
    
        def update(self,output):
            print "update"
    
        def complete(self, image, data, model, elapsed, qmin, qmax,index, qstep ):
            print "complete"
            self.done = True
    
    if __name__ == "__main__":
        CalcCommandline()
"""