source: sasview/sansmodels/src/sans/models/CylinderModel.py @ 876192b2

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 876192b2 was c7a7e1b, checked in by Gervaise Alina <gervyh@…>, 13 years ago

working on model pickle

  • Property mode set to 100644
File size: 5.5 KB
RevLine 
[ae3ce4e]1#!/usr/bin/env python
[95986b5]2
[79ac6f8]3##############################################################################
4#       This software was developed by the University of Tennessee as part of the
5#       Distributed Data Analysis of Neutron Scattering Experiments (DANSE)
6#       project funded by the US National Science Foundation.
7#
8#       If you use DANSE applications to do scientific research that leads to
9#       publication, we ask that you acknowledge the use of the software with the
10#       following sentence:
11#
12#       "This work benefited from DANSE software developed under NSF award DMR-0520547."
13#
14#       copyright 2008, University of Tennessee
15##############################################################################
[95986b5]16
17
[79ac6f8]18"""
19Provide functionality for a C extension model
[ae3ce4e]20
[79ac6f8]21:WARNING: THIS FILE WAS GENERATED BY WRAPPERGENERATOR.PY
22         DO NOT MODIFY THIS FILE, MODIFY ..\c_extensions\cylinder.h
23         AND RE-RUN THE GENERATOR SCRIPT
[ae3ce4e]24
25"""
26
27from sans.models.BaseComponent import BaseComponent
28from sans_extension.c_models import CCylinderModel
29import copy   
[96656e3]30
31def create_CylinderModel():
32    obj = CylinderModel()
33    #CCylinderModel.__init__(obj) is called by CylinderModel constructor
34    return obj
35
[ae3ce4e]36class CylinderModel(CCylinderModel, BaseComponent):
[79ac6f8]37    """
38    Class that evaluates a CylinderModel model.
39    This file was auto-generated from ..\c_extensions\cylinder.h.
40    Refer to that file and the structure it contains
41    for details of the model.
42    List of default parameters:
[ae3ce4e]43         scale           = 1.0
[1ed3834]44         radius          = 20.0 [A]
45         length          = 400.0 [A]
[f10063e]46         sldCyl          = 4e-006 [1/A^(2)]
47         sldSolv         = 1e-006 [1/A^(2)]
[0824909]48         background      = 0.0 [1/cm]
[4628e31]49         cyl_theta       = 60.0 [deg]
50         cyl_phi         = 60.0 [deg]
[ae3ce4e]51
52    """
53       
54    def __init__(self):
55        """ Initialization """
56       
57        # Initialize BaseComponent first, then sphere
58        BaseComponent.__init__(self)
[96656e3]59        #apply(CCylinderModel.__init__, (self,))
[ae3ce4e]60        CCylinderModel.__init__(self)
61       
62        ## Name of the model
63        self.name = "CylinderModel"
[836fe6e]64        ## Model description
[f10063e]65        self.description =""" f(q)= 2*(sldCyl - sldSolv)*V*sin(qLcos(alpha/2))
[1ed3834]66                /[qLcos(alpha/2)]*J1(qRsin(alpha/2))/[qRsin(alpha)]
67               
68                P(q,alpha)= scale/V*f(q)^(2)+bkg
[9316609]69                V: Volume of the cylinder
70                R: Radius of the cylinder
71                L: Length of the cylinder
72                J1: The bessel function
[1ed3834]73                alpha: angle betweenthe axis of the
74                cylinder and the q-vector for 1D
75                :the ouput is P(q)=scale/V*integral
76                from pi/2 to zero of...
[0824909]77                f(q)^(2)*sin(alpha)*dalpha+ bkg"""
[836fe6e]78       
[fe9c19b4]79        ## Parameter details [units, min, max]
[ae3ce4e]80        self.details = {}
81        self.details['scale'] = ['', None, None]
[1ed3834]82        self.details['radius'] = ['[A]', None, None]
83        self.details['length'] = ['[A]', None, None]
[f10063e]84        self.details['sldCyl'] = ['[1/A^(2)]', None, None]
85        self.details['sldSolv'] = ['[1/A^(2)]', None, None]
[0824909]86        self.details['background'] = ['[1/cm]', None, None]
[4628e31]87        self.details['cyl_theta'] = ['[deg]', None, None]
88        self.details['cyl_phi'] = ['[deg]', None, None]
[836fe6e]89
[fe9c19b4]90        ## fittable parameters
[836fe6e]91        self.fixed=['cyl_phi.width', 'cyl_theta.width', 'length.width', 'radius.width']
[25a608f5]92       
[35aface]93        ## non-fittable parameters
[96656e3]94        self.non_fittable = []
[35aface]95       
[25a608f5]96        ## parameters with orientation
[96656e3]97        self.orientation_params = ['cyl_phi', 'cyl_theta', 'cyl_phi.width', 'cyl_theta.width']
[c7a7e1b]98
99    def __setstate__(self, state):
100        """
101        restore the state of a model from pickle
102        """
103        self.__dict__, self.params, self.dispersion = state
104       
[96656e3]105    def __reduce_ex__(self, proto):
[79ac6f8]106        """
[96656e3]107        Overwrite the __reduce_ex__ of PyTypeObject *type call in the init of
108        c model.
[79ac6f8]109        """
[c7a7e1b]110        state = (self.__dict__, self.params, self.dispersion)
111        return (create_CylinderModel,tuple(), state, None, None)
[fe9c19b4]112       
[96656e3]113    def clone(self):
114        """ Return a identical copy of self """
115        return self._clone(CylinderModel())   
[fe9c19b4]116       
[ae3ce4e]117   
[79ac6f8]118    def run(self, x=0.0):
119        """
120        Evaluate the model
121       
122        :param x: input q, or [q,phi]
123       
124        :return: scattering function P(q)
125       
[ae3ce4e]126        """
127       
128        return CCylinderModel.run(self, x)
129   
[79ac6f8]130    def runXY(self, x=0.0):
131        """
132        Evaluate the model in cartesian coordinates
133       
134        :param x: input q, or [qx, qy]
135       
136        :return: scattering function P(q)
137       
[ae3ce4e]138        """
139       
140        return CCylinderModel.runXY(self, x)
[95986b5]141       
[79ac6f8]142    def evalDistribution(self, x=[]):
143        """
144        Evaluate the model in cartesian coordinates
145       
146        :param x: input q[], or [qx[], qy[]]
147       
148        :return: scattering function P(q[])
149       
[9bd69098]150        """
[f9a1279]151        return CCylinderModel.evalDistribution(self, x)
[9bd69098]152       
[5eb9154]153    def calculate_ER(self):
[79ac6f8]154        """
155        Calculate the effective radius for P(q)*S(q)
156       
157        :return: the value of the effective radius
158       
[5eb9154]159        """       
160        return CCylinderModel.calculate_ER(self)
161       
[95986b5]162    def set_dispersion(self, parameter, dispersion):
163        """
[79ac6f8]164        Set the dispersion object for a model parameter
165       
166        :param parameter: name of the parameter [string]
167        :param dispersion: dispersion object of type DispersionModel
168       
[95986b5]169        """
170        return CCylinderModel.set_dispersion(self, parameter, dispersion.cdisp)
171       
[ae3ce4e]172   
173# End of file
Note: See TracBrowser for help on using the repository browser.