source: sasview/sansmodels/src/sans/models/EllipticalCylinderModel.py @ c7a7e1b

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

working on model pickle

  • Property mode set to 100644
File size: 5.8 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\elliptical_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 CEllipticalCylinderModel
29import copy   
[96656e3]30
31def create_EllipticalCylinderModel():
32    obj = EllipticalCylinderModel()
33    #CEllipticalCylinderModel.__init__(obj) is called by EllipticalCylinderModel constructor
34    return obj
35
[ae3ce4e]36class EllipticalCylinderModel(CEllipticalCylinderModel, BaseComponent):
[79ac6f8]37    """
38    Class that evaluates a EllipticalCylinderModel model.
39    This file was auto-generated from ..\c_extensions\elliptical_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         r_minor         = 20.0 [A]
[0824909]45         r_ratio         = 1.5
[1ed3834]46         length          = 400.0 [A]
[f10063e]47         sldCyl          = 4e-006 [1/A^(2)]
48         sldSolv         = 1e-006 [1/A^(2)]
[0824909]49         background      = 0.0 [1/cm]
[4628e31]50         cyl_theta       = 90.0 [deg]
51         cyl_phi         = 0.0 [deg]
52         cyl_psi         = 0.0 [deg]
[ae3ce4e]53
54    """
55       
56    def __init__(self):
57        """ Initialization """
58       
59        # Initialize BaseComponent first, then sphere
60        BaseComponent.__init__(self)
[96656e3]61        #apply(CEllipticalCylinderModel.__init__, (self,))
[ae3ce4e]62        CEllipticalCylinderModel.__init__(self)
63       
64        ## Name of the model
65        self.name = "EllipticalCylinderModel"
[836fe6e]66        ## Model description
[7ad9887]67        self.description =""" Model parameters: r_minor = the radius of minor axis of the cross section
68                r_ratio = the ratio of (r_major /r_minor >= 1)
69                length = the length of the cylinder
[f10063e]70                sldCyl = SLD of the cylinder
71                sldSolv = SLD of solvent -
[7ad9887]72                background = incoherent background"""
[836fe6e]73       
[fe9c19b4]74        ## Parameter details [units, min, max]
[ae3ce4e]75        self.details = {}
76        self.details['scale'] = ['', None, None]
[1ed3834]77        self.details['r_minor'] = ['[A]', None, None]
[0824909]78        self.details['r_ratio'] = ['', None, None]
[1ed3834]79        self.details['length'] = ['[A]', None, None]
[f10063e]80        self.details['sldCyl'] = ['[1/A^(2)]', None, None]
81        self.details['sldSolv'] = ['[1/A^(2)]', None, None]
[0824909]82        self.details['background'] = ['[1/cm]', None, None]
[4628e31]83        self.details['cyl_theta'] = ['[deg]', None, None]
84        self.details['cyl_phi'] = ['[deg]', None, None]
85        self.details['cyl_psi'] = ['[deg]', None, None]
[836fe6e]86
[fe9c19b4]87        ## fittable parameters
[975ec8e]88        self.fixed=['cyl_phi.width', 'cyl_theta.width', 'cyl_psi.width', 'length.width', 'r_minor.width', 'r_ratio.width']
[25a608f5]89       
[35aface]90        ## non-fittable parameters
[96656e3]91        self.non_fittable = []
[35aface]92       
[25a608f5]93        ## parameters with orientation
[96656e3]94        self.orientation_params = ['cyl_phi', 'cyl_theta', 'cyl_psi', 'cyl_phi.width', 'cyl_theta.width', 'cyl_psi.width']
[c7a7e1b]95
96    def __setstate__(self, state):
97        """
98        restore the state of a model from pickle
99        """
100        self.__dict__, self.params, self.dispersion = state
101       
[96656e3]102    def __reduce_ex__(self, proto):
[79ac6f8]103        """
[96656e3]104        Overwrite the __reduce_ex__ of PyTypeObject *type call in the init of
105        c model.
[79ac6f8]106        """
[c7a7e1b]107        state = (self.__dict__, self.params, self.dispersion)
108        return (create_EllipticalCylinderModel,tuple(), state, None, None)
[fe9c19b4]109       
[96656e3]110    def clone(self):
111        """ Return a identical copy of self """
112        return self._clone(EllipticalCylinderModel())   
[fe9c19b4]113       
[ae3ce4e]114   
[79ac6f8]115    def run(self, x=0.0):
116        """
117        Evaluate the model
118       
119        :param x: input q, or [q,phi]
120       
121        :return: scattering function P(q)
122       
[ae3ce4e]123        """
124       
125        return CEllipticalCylinderModel.run(self, x)
126   
[79ac6f8]127    def runXY(self, x=0.0):
128        """
129        Evaluate the model in cartesian coordinates
130       
131        :param x: input q, or [qx, qy]
132       
133        :return: scattering function P(q)
134       
[ae3ce4e]135        """
136       
137        return CEllipticalCylinderModel.runXY(self, x)
[95986b5]138       
[79ac6f8]139    def evalDistribution(self, x=[]):
140        """
141        Evaluate the model in cartesian coordinates
142       
143        :param x: input q[], or [qx[], qy[]]
144       
145        :return: scattering function P(q[])
146       
[9bd69098]147        """
[f9a1279]148        return CEllipticalCylinderModel.evalDistribution(self, x)
[9bd69098]149       
[5eb9154]150    def calculate_ER(self):
[79ac6f8]151        """
152        Calculate the effective radius for P(q)*S(q)
153       
154        :return: the value of the effective radius
155       
[5eb9154]156        """       
157        return CEllipticalCylinderModel.calculate_ER(self)
158       
[95986b5]159    def set_dispersion(self, parameter, dispersion):
160        """
[79ac6f8]161        Set the dispersion object for a model parameter
162       
163        :param parameter: name of the parameter [string]
164        :param dispersion: dispersion object of type DispersionModel
165       
[95986b5]166        """
167        return CEllipticalCylinderModel.set_dispersion(self, parameter, dispersion.cdisp)
168       
[ae3ce4e]169   
170# End of file
Note: See TracBrowser for help on using the repository browser.