source: sasview/sansmodels/src/sans/models/DisperseModel.py @ 6b8399b

#!/usr/bin/env python
""" 
        Wrapper for the Disperser class extension

    @author: Mathieu Doucet / UTK
    @contact: mathieu.doucet@nist.gov
"""

from sans.models.BaseComponent import BaseComponent
from sans_extension.c_models import Disperser
    
class DisperseModel(Disperser, BaseComponent):
    """ 
                Wrapper class for the Disperser extension
                Python class that takes a model and averages its output
                for a distribution of its parameters.
                  
                The parameters to be varied are specified at instantiation time.
                The distributions are Gaussian, with std deviations specified for
                each parameter at instantiation time.
                
                Example:
                        cyl = ModelFactory().getModel("CylinderModel")
                        disp = DisperseModel(cyl, ['cyl_phi'], [0.3])
                        disp.run([0.01, 1.57])
    """
        
    def __init__(self, model, paramList, sigmaList):
        """ Initialization 
                @param model: Model to disperse [BaseComponent]
                @param paramList: list of parameters to disperse [List of strings]
                @param sigmaList: list of std deviations for Gaussian dispersion [List of floats]
        """
        
        # Initialize BaseComponent first, then sphere
        BaseComponent.__init__(self)
        Disperser.__init__(self, model, paramList, sigmaList)
        ## Name of the model
        self.name = model.name
        ## Keep track of the underlying model
        self.model = model
   
    def clone(self):
        """ Return a identical copy of self """
        obj = DisperseModel(self.model, self.params['paramList'], self.params['sigmaList'])
        return obj   
   
    def setParam(self, name, value):
        if name.lower() in self.params:
                BaseComponent.setParam(self, name, value)
        else:
                self.model.setParam(name, value)
   
    def getParam(self, name):
        if name.lower() in self.params:
                return BaseComponent.getParam(self, name)
        else:
                return self.model.getParam(name)
   
    def run(self, x = 0.0):
        """ Evaluate the model
            @param x: input q, or [q,phi]
            @return: scattering function P(q)
        """
        return Disperser.run(self, x)
           
    def runXY(self, x = 0.0):
        """ Evaluate the model
            @param x: input q, or [q,phi]
            @return: scattering function P(q)
        """
        return Disperser.runXY(self, x)
   
# End of file