source: sasview/sansmodels/src/sans/models/StackedDisksModel.py @ bdc25e2

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

working on model pickle

  • Property mode set to 100644
File size: 6.0 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\stacked_disks.h
23         AND RE-RUN THE GENERATOR SCRIPT
[5068697]24
25"""
26
27from sans.models.BaseComponent import BaseComponent
28from sans_extension.c_models import CStackedDisksModel
29import copy   
[96656e3]30
31def create_StackedDisksModel():
32    obj = StackedDisksModel()
33    #CStackedDisksModel.__init__(obj) is called by StackedDisksModel constructor
34    return obj
35
[5068697]36class StackedDisksModel(CStackedDisksModel, BaseComponent):
[79ac6f8]37    """
38    Class that evaluates a StackedDisksModel model.
39    This file was auto-generated from ..\c_extensions\stacked_disks.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           = 0.01
44         radius          = 3000.0 [A]
[975ec8e]45         core_thick      = 10.0 [A]
46         layer_thick     = 15.0 [A]
[27972c1d]47         core_sld        = 4e-006 [1/A^(2)]
48         layer_sld       = -4e-007 [1/A^(2)]
49         solvent_sld     = 5e-006 [1/A^(2)]
[975ec8e]50         n_stacking      = 1.0
51         sigma_d         = 0.0
[5068697]52         background      = 0.001 [1/cm]
[975ec8e]53         axis_theta      = 0.0 [rad]
54         axis_phi        = 0.0 [rad]
[5068697]55
56    """
57       
58    def __init__(self):
59        """ Initialization """
60       
61        # Initialize BaseComponent first, then sphere
62        BaseComponent.__init__(self)
[96656e3]63        #apply(CStackedDisksModel.__init__, (self,))
[5068697]64        CStackedDisksModel.__init__(self)
65       
66        ## Name of the model
67        self.name = "StackedDisksModel"
68        ## Model description
[7ad9887]69        self.description =""" One layer of disk consists of a core, a top layer, and a bottom layer.
70                radius =  the radius of the disk
71                core_thick = thickness of the core
72                layer_thick = thickness of a layer
73                core_sld = the SLD of the core
74                layer_sld = the SLD of the layers
75                n_stacking = the number of the disks
76                sigma_d =  Gaussian STD of d-spacing
77                solvent_sld = the SLD of the solvent"""
[5068697]78       
[fe9c19b4]79        ## Parameter details [units, min, max]
[5068697]80        self.details = {}
81        self.details['scale'] = ['', None, None]
82        self.details['radius'] = ['[A]', None, None]
[975ec8e]83        self.details['core_thick'] = ['[A]', None, None]
84        self.details['layer_thick'] = ['[A]', None, None]
[27972c1d]85        self.details['core_sld'] = ['[1/A^(2)]', None, None]
86        self.details['layer_sld'] = ['[1/A^(2)]', None, None]
87        self.details['solvent_sld'] = ['[1/A^(2)]', None, None]
[975ec8e]88        self.details['n_stacking'] = ['', None, None]
89        self.details['sigma_d'] = ['', None, None]
[5068697]90        self.details['background'] = ['[1/cm]', None, None]
91        self.details['axis_theta'] = ['[rad]', None, None]
92        self.details['axis_phi'] = ['[rad]', None, None]
93
[fe9c19b4]94        ## fittable parameters
[975ec8e]95        self.fixed=['core_thick.width', 'layer_thick.width', 'radius.width', 'axis_theta.width', 'axis_phi.width']
[5068697]96       
[35aface]97        ## non-fittable parameters
[96656e3]98        self.non_fittable = []
[35aface]99       
[5068697]100        ## parameters with orientation
[96656e3]101        self.orientation_params = ['axis_phi', 'axis_theta', 'axis_phi.width', 'axis_theta.width']
[c7a7e1b]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       
[96656e3]109    def __reduce_ex__(self, proto):
[79ac6f8]110        """
[96656e3]111        Overwrite the __reduce_ex__ of PyTypeObject *type call in the init of
112        c model.
[79ac6f8]113        """
[c7a7e1b]114        state = (self.__dict__, self.params, self.dispersion)
115        return (create_StackedDisksModel,tuple(), state, None, None)
[fe9c19b4]116       
[96656e3]117    def clone(self):
118        """ Return a identical copy of self """
119        return self._clone(StackedDisksModel())   
[fe9c19b4]120       
[5068697]121   
[79ac6f8]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       
[5068697]130        """
131       
132        return CStackedDisksModel.run(self, x)
133   
[79ac6f8]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       
[5068697]142        """
143       
144        return CStackedDisksModel.runXY(self, x)
145       
[79ac6f8]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       
[9bd69098]154        """
[f9a1279]155        return CStackedDisksModel.evalDistribution(self, x)
[9bd69098]156       
[5eb9154]157    def calculate_ER(self):
[79ac6f8]158        """
159        Calculate the effective radius for P(q)*S(q)
160       
161        :return: the value of the effective radius
162       
[5eb9154]163        """       
164        return CStackedDisksModel.calculate_ER(self)
165       
[5068697]166    def set_dispersion(self, parameter, dispersion):
167        """
[79ac6f8]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       
[5068697]173        """
174        return CStackedDisksModel.set_dispersion(self, parameter, dispersion.cdisp)
175       
176   
177# End of file
Note: See TracBrowser for help on using the repository browser.