source: sasview/sansview/plugins/testmodel.py @ 97dea6c

"""
Test plug-in model
"""
from sans.models.pluginmodel import Model1DPlugin

import math


# Your model HAS to be called Model
class Model(Model1DPlugin):
    """
    Class that evaluates a cos(x) model. 
    """
    
    ## Name of the model
    name = "A+Bcos(2x)+Csin(2x)"
    
    def __init__(self):
        """
        Initialization
        """
        Model1DPlugin.__init__(self, name= self.name)
        
        ## Parameters definition and defaults
        self.params = {}
        self.params['A'] = 1.0
        self.params['B'] = 1.0
        self.params['C'] = 10.0

        ## Parameter details [units, min, max]
        self.details = {}
        self.details['A'] = ['', -1e16, 1e16]
        self.details['B'] = ['', -1e16, 1e16]
        self.details['C'] = ['', -1e16, 1e16]
        self.description = "F(x)=A+Bcos(2x)+Csin(2x) "
   
    def function(self, x = 0.0):
        """
        Evaluate the model
        
        :param x: input x
        
        :return: function value
        
        """
        return self.params['A']+self.params['B']*math.cos(2.0*x)+self.params['C']*math.sin(2.0*x)