source: sasview/sansmodels/src/sans/models/TriaxialEllipsoidModel.py @ 0cbfaeb

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 0cbfaeb 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
RevLine 
[5068697]1#!/usr/bin/env python
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##############################################################################
[5068697]16
17
[79ac6f8]18"""
19Provide functionality for a C extension model
[5068697]20
[79ac6f8]21:WARNING: THIS FILE WAS GENERATED BY WRAPPERGENERATOR.PY
22         DO NOT MODIFY THIS FILE, MODIFY ..\c_extensions\triaxial_ellipsoid.h
23         AND RE-RUN THE GENERATOR SCRIPT
[5068697]24
25"""
26
27from sans.models.BaseComponent import BaseComponent
28from sans_extension.c_models import CTriaxialEllipsoidModel
29import copy   
[96656e3]30
31def create_TriaxialEllipsoidModel():
32    obj = TriaxialEllipsoidModel()
33    #CTriaxialEllipsoidModel.__init__(obj) is called by TriaxialEllipsoidModel constructor
34    return obj
35
[5068697]36class TriaxialEllipsoidModel(CTriaxialEllipsoidModel, BaseComponent):
[79ac6f8]37    """
38    Class that evaluates a TriaxialEllipsoidModel model.
39    This file was auto-generated from ..\c_extensions\triaxial_ellipsoid.h.
40    Refer to that file and the structure it contains
41    for details of the model.
42    List of default parameters:
[5068697]43         scale           = 1.0
[3c102d4]44         semi_axisA      = 35.0 [A]
45         semi_axisB      = 100.0 [A]
[5068697]46         semi_axisC      = 400.0 [A]
[13eb1c4]47         sldEll          = 1e-006 [1/A^(2)]
48         sldSolv         = 6.3e-006 [1/A^(2)]
[5068697]49         background      = 0.0 [1/cm]
[4628e31]50         axis_theta      = 57.325 [deg]
51         axis_phi        = 57.325 [deg]
52         axis_psi        = 0.0 [deg]
[5068697]53
54    """
55       
56    def __init__(self):
57        """ Initialization """
58       
59        # Initialize BaseComponent first, then sphere
60        BaseComponent.__init__(self)
[96656e3]61        #apply(CTriaxialEllipsoidModel.__init__, (self,))
[5068697]62        CTriaxialEllipsoidModel.__init__(self)
63       
64        ## Name of the model
65        self.name = "TriaxialEllipsoidModel"
66        ## Model description
[7ad9887]67        self.description ="""Note: During fitting ensure that the inequality A<B<C is not
68                violated. Otherwise the calculation will
69                not be correct."""
[5068697]70       
[fe9c19b4]71        ## Parameter details [units, min, max]
[5068697]72        self.details = {}
73        self.details['scale'] = ['', None, None]
74        self.details['semi_axisA'] = ['[A]', None, None]
[3c102d4]75        self.details['semi_axisB'] = ['[A]', None, None]
[5068697]76        self.details['semi_axisC'] = ['[A]', None, None]
[13eb1c4]77        self.details['sldEll'] = ['[1/A^(2)]', None, None]
78        self.details['sldSolv'] = ['[1/A^(2)]', None, None]
[5068697]79        self.details['background'] = ['[1/cm]', None, None]
[4628e31]80        self.details['axis_theta'] = ['[deg]', None, None]
81        self.details['axis_phi'] = ['[deg]', None, None]
82        self.details['axis_psi'] = ['[deg]', None, None]
[5068697]83
[fe9c19b4]84        ## fittable parameters
[975ec8e]85        self.fixed=['axis_psi.width', 'axis_phi.width', 'axis_theta.width', 'semi_axisA.width', 'semi_axisB.width', 'semi_axisC.width']
[5068697]86       
[35aface]87        ## non-fittable parameters
[96656e3]88        self.non_fittable = []
[35aface]89       
[5068697]90        ## parameters with orientation
[96656e3]91        self.orientation_params = ['axis_psi', 'axis_phi', 'axis_theta', 'axis_psi.width', 'axis_phi.width', 'axis_theta.width']
[c7a7e1b]92
93    def __setstate__(self, state):
94        """
95        restore the state of a model from pickle
96        """
97        self.__dict__, self.params, self.dispersion = state
98       
[96656e3]99    def __reduce_ex__(self, proto):
[79ac6f8]100        """
[96656e3]101        Overwrite the __reduce_ex__ of PyTypeObject *type call in the init of
102        c model.
[79ac6f8]103        """
[c7a7e1b]104        state = (self.__dict__, self.params, self.dispersion)
105        return (create_TriaxialEllipsoidModel,tuple(), state, None, None)
[fe9c19b4]106       
[96656e3]107    def clone(self):
108        """ Return a identical copy of self """
109        return self._clone(TriaxialEllipsoidModel())   
[fe9c19b4]110       
[5068697]111   
[79ac6f8]112    def run(self, x=0.0):
113        """
114        Evaluate the model
115       
116        :param x: input q, or [q,phi]
117       
118        :return: scattering function P(q)
119       
[5068697]120        """
121       
122        return CTriaxialEllipsoidModel.run(self, x)
123   
[79ac6f8]124    def runXY(self, x=0.0):
125        """
126        Evaluate the model in cartesian coordinates
127       
128        :param x: input q, or [qx, qy]
129       
130        :return: scattering function P(q)
131       
[5068697]132        """
133       
134        return CTriaxialEllipsoidModel.runXY(self, x)
135       
[79ac6f8]136    def evalDistribution(self, x=[]):
137        """
138        Evaluate the model in cartesian coordinates
139       
140        :param x: input q[], or [qx[], qy[]]
141       
142        :return: scattering function P(q[])
143       
[9bd69098]144        """
[f9a1279]145        return CTriaxialEllipsoidModel.evalDistribution(self, x)
[9bd69098]146       
[5eb9154]147    def calculate_ER(self):
[79ac6f8]148        """
149        Calculate the effective radius for P(q)*S(q)
150       
151        :return: the value of the effective radius
152       
[5eb9154]153        """       
154        return CTriaxialEllipsoidModel.calculate_ER(self)
155       
[5068697]156    def set_dispersion(self, parameter, dispersion):
157        """
[79ac6f8]158        Set the dispersion object for a model parameter
159       
160        :param parameter: name of the parameter [string]
161        :param dispersion: dispersion object of type DispersionModel
162       
[5068697]163        """
164        return CTriaxialEllipsoidModel.set_dispersion(self, parameter, dispersion.cdisp)
165       
166   
167# End of file
Note: See TracBrowser for help on using the repository browser.