source: sasview/sansmodels/src/sans/models/EllipsoidModel.py @ 0d86fecb

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 0d86fecb was c7a7e1b, checked in by Gervaise Alina <gervyh@…>, 14 years ago

working on model pickle

  • Property mode set to 100644
File size: 5.7 KB
Line 
1#!/usr/bin/env python
2
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##############################################################################
16
17
18"""
19Provide functionality for a C extension model
20
21:WARNING: THIS FILE WAS GENERATED BY WRAPPERGENERATOR.PY
22         DO NOT MODIFY THIS FILE, MODIFY ..\c_extensions\ellipsoid.h
23         AND RE-RUN THE GENERATOR SCRIPT
24
25"""
26
27from sans.models.BaseComponent import BaseComponent
28from sans_extension.c_models import CEllipsoidModel
29import copy   
30
31def create_EllipsoidModel():
32    obj = EllipsoidModel()
33    #CEllipsoidModel.__init__(obj) is called by EllipsoidModel constructor
34    return obj
35
36class EllipsoidModel(CEllipsoidModel, BaseComponent):
37    """
38    Class that evaluates a EllipsoidModel model.
39    This file was auto-generated from ..\c_extensions\ellipsoid.h.
40    Refer to that file and the structure it contains
41    for details of the model.
42    List of default parameters:
43         scale           = 1.0
44         radius_a        = 20.0 [A]
45         radius_b        = 400.0 [A]
46         sldEll          = 4e-006 [1/A^(2)]
47         sldSolv         = 1e-006 [1/A^(2)]
48         background      = 0.0 [1/cm]
49         axis_theta      = 90.0 [deg]
50         axis_phi        = 0.0 [deg]
51
52    """
53       
54    def __init__(self):
55        """ Initialization """
56       
57        # Initialize BaseComponent first, then sphere
58        BaseComponent.__init__(self)
59        #apply(CEllipsoidModel.__init__, (self,))
60        CEllipsoidModel.__init__(self)
61       
62        ## Name of the model
63        self.name = "EllipsoidModel"
64        ## Model description
65        self.description =""""P(q.alpha)= scale*f(q)^(2)+ bkg, where f(q)= 3*(sld_ell
66                - sld_solvent)*V*[sin(q*r(Ra,Rb,alpha))
67                -q*r*cos(qr(Ra,Rb,alpha))]
68                /[qr(Ra,Rb,alpha)]^(3)"
69               
70                r(Ra,Rb,alpha)= [Rb^(2)*(sin(alpha))^(2)
71                + Ra^(2)*(cos(alpha))^(2)]^(1/2)
72               
73                scatter_sld: SLD of the scatter
74                solvent_sld: SLD of the solvent
75                sldEll: SLD of ellipsoid
76                sldSolv: SLD of solvent
77                V: volune of the Eliipsoid
78                Ra: radius along the rotation axis
79                of the Ellipsoid
80                Rb: radius perpendicular to the
81                rotation axis of the ellipsoid"""
82       
83        ## Parameter details [units, min, max]
84        self.details = {}
85        self.details['scale'] = ['', None, None]
86        self.details['radius_a'] = ['[A]', None, None]
87        self.details['radius_b'] = ['[A]', None, None]
88        self.details['sldEll'] = ['[1/A^(2)]', None, None]
89        self.details['sldSolv'] = ['[1/A^(2)]', None, None]
90        self.details['background'] = ['[1/cm]', None, None]
91        self.details['axis_theta'] = ['[deg]', None, None]
92        self.details['axis_phi'] = ['[deg]', None, None]
93
94        ## fittable parameters
95        self.fixed=['axis_phi.width', 'axis_theta.width', 'radius_a.width', 'radius_b.width', 'length.width', 'r_minor.width']
96       
97        ## non-fittable parameters
98        self.non_fittable = []
99       
100        ## parameters with orientation
101        self.orientation_params = ['axis_phi.width', 'axis_theta.width', 'axis_phi', 'axis_theta']
102
103    def __setstate__(self, state):
104        """
105        restore the state of a model from pickle
106        """
107        self.__dict__, self.params, self.dispersion = state
108       
109    def __reduce_ex__(self, proto):
110        """
111        Overwrite the __reduce_ex__ of PyTypeObject *type call in the init of
112        c model.
113        """
114        state = (self.__dict__, self.params, self.dispersion)
115        return (create_EllipsoidModel,tuple(), state, None, None)
116       
117    def clone(self):
118        """ Return a identical copy of self """
119        return self._clone(EllipsoidModel())   
120       
121   
122    def run(self, x=0.0):
123        """
124        Evaluate the model
125       
126        :param x: input q, or [q,phi]
127       
128        :return: scattering function P(q)
129       
130        """
131       
132        return CEllipsoidModel.run(self, x)
133   
134    def runXY(self, x=0.0):
135        """
136        Evaluate the model in cartesian coordinates
137       
138        :param x: input q, or [qx, qy]
139       
140        :return: scattering function P(q)
141       
142        """
143       
144        return CEllipsoidModel.runXY(self, x)
145       
146    def evalDistribution(self, x=[]):
147        """
148        Evaluate the model in cartesian coordinates
149       
150        :param x: input q[], or [qx[], qy[]]
151       
152        :return: scattering function P(q[])
153       
154        """
155        return CEllipsoidModel.evalDistribution(self, x)
156       
157    def calculate_ER(self):
158        """
159        Calculate the effective radius for P(q)*S(q)
160       
161        :return: the value of the effective radius
162       
163        """       
164        return CEllipsoidModel.calculate_ER(self)
165       
166    def set_dispersion(self, parameter, dispersion):
167        """
168        Set the dispersion object for a model parameter
169       
170        :param parameter: name of the parameter [string]
171        :param dispersion: dispersion object of type DispersionModel
172       
173        """
174        return CEllipsoidModel.set_dispersion(self, parameter, dispersion.cdisp)
175       
176   
177# End of file
Note: See TracBrowser for help on using the repository browser.