source: sasview/sansmodels/src/sans/models/c_extensions/CFlexibleCylinderModel.c @ 5eb9154

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 5eb9154 was e0a8a3c, checked in by Gervaise Alina <gervyh@…>, 15 years ago

change the orientation of models

  • Property mode set to 100644
File size: 10.1 KB
Line 
1/** CFlexibleCylinderModel
2 *
3 * C extension
4 *
5 * WARNING: THIS FILE WAS GENERATED BY WRAPPERGENERATOR.PY
6 *          DO NOT MODIFY THIS FILE, MODIFY flexible_cylinder.h
7 *          AND RE-RUN THE GENERATOR SCRIPT
8 *
9 * @author   M.Doucet / UTK
10 */
11 
12#include <Python.h>
13#include "structmember.h"
14#include <stdio.h>
15#include <stdlib.h>
16#include <math.h>
17#include <time.h>
18
19#include "flexible_cylinder.h"
20
21/// Error object for raised exceptions
22static PyObject * CFlexibleCylinderModelError = NULL;
23
24
25// Class definition
26typedef struct {
27    PyObject_HEAD
28    /// Parameters
29    PyObject * params;
30    /// Log for unit testing
31    PyObject * log;
32    /// Model parameters
33        FlexibleCylinderParameters model_pars;
34} CFlexibleCylinderModel;
35
36
37static void
38CFlexibleCylinderModel_dealloc(CFlexibleCylinderModel* self)
39{
40    self->ob_type->tp_free((PyObject*)self);
41   
42
43}
44
45static PyObject *
46CFlexibleCylinderModel_new(PyTypeObject *type, PyObject *args, PyObject *kwds)
47{
48    CFlexibleCylinderModel *self;
49   
50    self = (CFlexibleCylinderModel *)type->tp_alloc(type, 0);
51   
52    return (PyObject *)self;
53}
54
55static int
56CFlexibleCylinderModel_init(CFlexibleCylinderModel *self, PyObject *args, PyObject *kwds)
57{
58    if (self != NULL) {
59       
60        // Create parameters
61        self->params = PyDict_New();
62       
63        // Initialize parameter dictionary
64        PyDict_SetItemString(self->params,"scale",Py_BuildValue("d",1.000000));
65        PyDict_SetItemString(self->params,"length",Py_BuildValue("d",1000.000000));
66        PyDict_SetItemString(self->params,"radius",Py_BuildValue("d",20.000000));
67        PyDict_SetItemString(self->params,"background",Py_BuildValue("d",0.000100));
68        PyDict_SetItemString(self->params,"kuhn_length",Py_BuildValue("d",100.000000));
69        PyDict_SetItemString(self->params,"contrast",Py_BuildValue("d",0.000005));
70
71         
72        // Create empty log
73        self->log = PyDict_New();
74       
75       
76
77    }
78    return 0;
79}
80
81static PyMemberDef CFlexibleCylinderModel_members[] = {
82    {"params", T_OBJECT, offsetof(CFlexibleCylinderModel, params), 0,
83     "Parameters"},
84    {"log", T_OBJECT, offsetof(CFlexibleCylinderModel, log), 0,
85     "Log"},
86    {NULL}  /* Sentinel */
87};
88
89/** Read double from PyObject
90    @param p PyObject
91    @return double
92*/
93double CFlexibleCylinderModel_readDouble(PyObject *p) {
94    if (PyFloat_Check(p)==1) {
95        return (double)(((PyFloatObject *)(p))->ob_fval);
96    } else if (PyInt_Check(p)==1) {
97        return (double)(((PyIntObject *)(p))->ob_ival);
98    } else if (PyLong_Check(p)==1) {
99        return (double)PyLong_AsLong(p);
100    } else {
101        return 0.0;
102    }
103}
104
105
106/**
107 * Function to call to evaluate model
108 * @param args: input q or [q,phi]
109 * @return: function value
110 */
111static PyObject * run(CFlexibleCylinderModel *self, PyObject *args) {
112        double q_value, phi_value;
113        PyObject* pars;
114        int npars;
115       
116        // Get parameters
117       
118        // Reader parameter dictionary
119    self->model_pars.scale = PyFloat_AsDouble( PyDict_GetItemString(self->params, "scale") );
120    self->model_pars.length = PyFloat_AsDouble( PyDict_GetItemString(self->params, "length") );
121    self->model_pars.radius = PyFloat_AsDouble( PyDict_GetItemString(self->params, "radius") );
122    self->model_pars.background = PyFloat_AsDouble( PyDict_GetItemString(self->params, "background") );
123    self->model_pars.kuhn_length = PyFloat_AsDouble( PyDict_GetItemString(self->params, "kuhn_length") );
124    self->model_pars.contrast = PyFloat_AsDouble( PyDict_GetItemString(self->params, "contrast") );
125
126       
127        // Get input and determine whether we have to supply a 1D or 2D return value.
128        if ( !PyArg_ParseTuple(args,"O",&pars) ) {
129            PyErr_SetString(CFlexibleCylinderModelError, 
130                "CFlexibleCylinderModel.run expects a q value.");
131                return NULL;
132        }
133         
134        // Check params
135        if( PyList_Check(pars)==1) {
136               
137                // Length of list should be 2 for I(q,phi)
138            npars = PyList_GET_SIZE(pars); 
139            if(npars!=2) {
140                PyErr_SetString(CFlexibleCylinderModelError, 
141                        "CFlexibleCylinderModel.run expects a double or a list of dimension 2.");
142                return NULL;
143            }
144            // We have a vector q, get the q and phi values at which
145            // to evaluate I(q,phi)
146            q_value = CFlexibleCylinderModel_readDouble(PyList_GET_ITEM(pars,0));
147            phi_value = CFlexibleCylinderModel_readDouble(PyList_GET_ITEM(pars,1));
148            // Skip zero
149            if (q_value==0) {
150                return Py_BuildValue("d",0.0);
151            }
152                return Py_BuildValue("d",flexible_cylinder_analytical_2D(&(self->model_pars),q_value,phi_value));
153
154        } else {
155
156                // We have a scalar q, we will evaluate I(q)
157                q_value = CFlexibleCylinderModel_readDouble(pars);             
158               
159                return Py_BuildValue("d",flexible_cylinder_analytical_1D(&(self->model_pars),q_value));
160        }       
161}
162
163/**
164 * Function to call to evaluate model in cartesian coordinates
165 * @param args: input q or [qx, qy]]
166 * @return: function value
167 */
168static PyObject * runXY(CFlexibleCylinderModel *self, PyObject *args) {
169        double qx_value, qy_value;
170        PyObject* pars;
171        int npars;
172       
173        // Get parameters
174       
175        // Reader parameter dictionary
176    self->model_pars.scale = PyFloat_AsDouble( PyDict_GetItemString(self->params, "scale") );
177    self->model_pars.length = PyFloat_AsDouble( PyDict_GetItemString(self->params, "length") );
178    self->model_pars.radius = PyFloat_AsDouble( PyDict_GetItemString(self->params, "radius") );
179    self->model_pars.background = PyFloat_AsDouble( PyDict_GetItemString(self->params, "background") );
180    self->model_pars.kuhn_length = PyFloat_AsDouble( PyDict_GetItemString(self->params, "kuhn_length") );
181    self->model_pars.contrast = PyFloat_AsDouble( PyDict_GetItemString(self->params, "contrast") );
182
183       
184        // Get input and determine whether we have to supply a 1D or 2D return value.
185        if ( !PyArg_ParseTuple(args,"O",&pars) ) {
186            PyErr_SetString(CFlexibleCylinderModelError, 
187                "CFlexibleCylinderModel.run expects a q value.");
188                return NULL;
189        }
190         
191        // Check params
192        if( PyList_Check(pars)==1) {
193               
194                // Length of list should be 2 for I(qx, qy))
195            npars = PyList_GET_SIZE(pars); 
196            if(npars!=2) {
197                PyErr_SetString(CFlexibleCylinderModelError, 
198                        "CFlexibleCylinderModel.run expects a double or a list of dimension 2.");
199                return NULL;
200            }
201            // We have a vector q, get the qx and qy values at which
202            // to evaluate I(qx,qy)
203            qx_value = CFlexibleCylinderModel_readDouble(PyList_GET_ITEM(pars,0));
204            qy_value = CFlexibleCylinderModel_readDouble(PyList_GET_ITEM(pars,1));
205                return Py_BuildValue("d",flexible_cylinder_analytical_2DXY(&(self->model_pars),qx_value,qy_value));
206
207        } else {
208
209                // We have a scalar q, we will evaluate I(q)
210                qx_value = CFlexibleCylinderModel_readDouble(pars);             
211               
212                return Py_BuildValue("d",flexible_cylinder_analytical_1D(&(self->model_pars),qx_value));
213        }       
214}
215
216static PyObject * reset(CFlexibleCylinderModel *self, PyObject *args) {
217   
218
219    return Py_BuildValue("d",0.0);
220}
221
222
223static PyMethodDef CFlexibleCylinderModel_methods[] = {
224    {"run",      (PyCFunction)run     , METH_VARARGS,
225      "Evaluate the model at a given Q or Q, phi"},
226    {"runXY",      (PyCFunction)runXY     , METH_VARARGS,
227      "Evaluate the model at a given Q or Qx, Qy"},
228    {"reset",    (PyCFunction)reset   , METH_VARARGS,
229      "Reset pair correlation"},
230    //{"numerical_1D",      (PyCFunction)numerical_1D     , METH_VARARGS,
231    //  "Evaluate the 1D model at a given Q"},
232   {NULL}
233};
234
235static PyTypeObject CFlexibleCylinderModelType = {
236    PyObject_HEAD_INIT(NULL)
237    0,                         /*ob_size*/
238    "CFlexibleCylinderModel",             /*tp_name*/
239    sizeof(CFlexibleCylinderModel),             /*tp_basicsize*/
240    0,                         /*tp_itemsize*/
241    (destructor)CFlexibleCylinderModel_dealloc, /*tp_dealloc*/
242    0,                         /*tp_print*/
243    0,                         /*tp_getattr*/
244    0,                         /*tp_setattr*/
245    0,                         /*tp_compare*/
246    0,                         /*tp_repr*/
247    0,                         /*tp_as_number*/
248    0,                         /*tp_as_sequence*/
249    0,                         /*tp_as_mapping*/
250    0,                         /*tp_hash */
251    0,                         /*tp_call*/
252    0,                         /*tp_str*/
253    0,                         /*tp_getattro*/
254    0,                         /*tp_setattro*/
255    0,                         /*tp_as_buffer*/
256    Py_TPFLAGS_DEFAULT | Py_TPFLAGS_BASETYPE, /*tp_flags*/
257    "CFlexibleCylinderModel objects",           /* tp_doc */
258    0,                         /* tp_traverse */
259    0,                         /* tp_clear */
260    0,                         /* tp_richcompare */
261    0,                         /* tp_weaklistoffset */
262    0,                         /* tp_iter */
263    0,                         /* tp_iternext */
264    CFlexibleCylinderModel_methods,             /* tp_methods */
265    CFlexibleCylinderModel_members,             /* tp_members */
266    0,                         /* tp_getset */
267    0,                         /* tp_base */
268    0,                         /* tp_dict */
269    0,                         /* tp_descr_get */
270    0,                         /* tp_descr_set */
271    0,                         /* tp_dictoffset */
272    (initproc)CFlexibleCylinderModel_init,      /* tp_init */
273    0,                         /* tp_alloc */
274    CFlexibleCylinderModel_new,                 /* tp_new */
275};
276
277
278static PyMethodDef module_methods[] = {
279    {NULL} 
280};
281
282/**
283 * Function used to add the model class to a module
284 * @param module: module to add the class to
285 */ 
286void addCFlexibleCylinderModel(PyObject *module) {
287        PyObject *d;
288       
289    if (PyType_Ready(&CFlexibleCylinderModelType) < 0)
290        return;
291
292    Py_INCREF(&CFlexibleCylinderModelType);
293    PyModule_AddObject(module, "CFlexibleCylinderModel", (PyObject *)&CFlexibleCylinderModelType);
294   
295    d = PyModule_GetDict(module);
296    CFlexibleCylinderModelError = PyErr_NewException("CFlexibleCylinderModel.error", NULL, NULL);
297    PyDict_SetItemString(d, "CFlexibleCylinderModelError", CFlexibleCylinderModelError);
298}
299
Note: See TracBrowser for help on using the repository browser.