source: sasview/sansmodels/src/sans/models/c_models/c_models.cpp @ 4d09267

/** c_models
 *
 * Module containing all SANS model extensions
 *
 * @author   M.Doucet / UTK
 */
#include <Python.h>
#include <parameters.hh>

void addCCylinderModel(PyObject *module);
void addCSphereModel(PyObject *module);
void addCHardsphereStructure(PyObject *module);
void addCStickyHSStructure(PyObject *module);
void addCSquareWellStructure(PyObject *module);
void addCHayterMSAStructure(PyObject *module);
void addCDiamEllipFunc(PyObject *module);
void addCDiamCylFunc(PyObject *module);
void addCCoreShellModel(PyObject *module);
void addCCoreShellCylinderModel(PyObject *module);
void addCEllipsoidModel(PyObject *module);
void addCEllipticalCylinderModel(PyObject *module);

extern "C" {
        //void addCCoreShellCylinderModel(PyObject *module);
        //void addCCoreShellModel(PyObject *module);
        //void addCEllipsoidModel(PyObject *module);
        //void addCEllipticalCylinderModel(PyObject *module);
        void addDisperser(PyObject *module);
        void addCGaussian(PyObject *module);
        void addCLorentzian(PyObject *module);
}

/**
 * Delete a dispersion model object
 */
void del_dispersion_model(void *ptr){
        DispersionModel * disp = static_cast<DispersionModel *>(ptr);
        delete disp;
        return;
}

/**
 * Create a dispersion model as a python object
 */
PyObject * new_dispersion_model(PyObject *, PyObject *args) {
        DispersionModel *disp = new DispersionModel();
        return PyCObject_FromVoidPtr(disp, del_dispersion_model);
}


/**
 * Delete a gaussian dispersion model object
 */
void del_gaussian_dispersion(void *ptr){
        GaussianDispersion * disp = static_cast<GaussianDispersion *>(ptr);
        delete disp;
        return;
}

/**
 * Create a gaussian dispersion model as a python object
 */
PyObject * new_gaussian_dispersion(PyObject *, PyObject *args) {
        GaussianDispersion *disp = new GaussianDispersion();
        return PyCObject_FromVoidPtr(disp, del_gaussian_dispersion);
}

/**
 * Delete an array dispersion model object
 */
void del_array_dispersion(void *ptr){
        ArrayDispersion * disp = static_cast<ArrayDispersion *>(ptr);
        delete disp;
        return;
}

/**
 * Create an array dispersion model as a python object
 */
PyObject * new_array_dispersion(PyObject *, PyObject *args) {
        ArrayDispersion *disp = new ArrayDispersion();
        return PyCObject_FromVoidPtr(disp, del_array_dispersion);
}

#define INVECTOR(obj,buf,len)                                                                           \
    do { \
        int err = PyObject_AsReadBuffer(obj, (const void **)(&buf), &len); \
        if (err < 0) return NULL; \
        len /= sizeof(*buf); \
    } while (0)

/**
 * Sets weights from a numpy array
 */
PyObject * set_weights(PyObject *, PyObject *args) {
        PyObject *val_obj;
        PyObject *wei_obj;
        PyObject *disp;
        Py_ssize_t nval;
        Py_ssize_t nwei;
        double *values;
        double *weights;
    int i;

        if (!PyArg_ParseTuple(args, "OOO", &disp, &val_obj, &wei_obj)) return NULL;
        INVECTOR(val_obj, values, nval);
        INVECTOR(wei_obj, weights, nwei);

        // Sanity check
        if(nval!=nwei) return NULL;

        // Set the array pointers
        void *temp = PyCObject_AsVoidPtr(disp);
        DispersionModel * dispersion = static_cast<DispersionModel *>(temp);
        dispersion->set_weights(nval, values, weights);

        return Py_BuildValue("i",1);
}



/**
 * Define empty module
 */
static PyMethodDef module_methods[] = {
        {"new_dispersion_model", (PyCFunction)new_dispersion_model     , METH_VARARGS,
                  "Create a new DispersionModel object"},
        {"new_gaussian_model",   (PyCFunction)new_gaussian_dispersion, METH_VARARGS,
                  "Create a new GaussianDispersion object"},
        {"new_array_model",      (PyCFunction)new_array_dispersion  , METH_VARARGS,
                  "Create a new ArrayDispersion object"},
        {"set_dispersion_weights",(PyCFunction)set_weights  , METH_VARARGS,
                        "Create the dispersion weight arrays for an Array Dispersion object"},
    {NULL}
};


#ifndef PyMODINIT_FUNC  /* declarations for DLL import/export */
#define PyMODINIT_FUNC void
#endif
PyMODINIT_FUNC
initc_models(void)
{
    PyObject* m;

    m = Py_InitModule3("c_models", module_methods,
                       "C extension module for SANS scattering models.");

        addCCylinderModel(m);
        addCCoreShellCylinderModel(m);
        addCCoreShellModel(m);
        addCEllipsoidModel(m);
        addCSphereModel(m);
        addCHardsphereStructure(m);
        addCStickyHSStructure(m);
        addCSquareWellStructure(m);
        addCHayterMSAStructure(m);
        addCDiamEllipFunc(m);
        addCDiamCylFunc(m);
        addCEllipticalCylinderModel(m);
        addDisperser(m);
        addCGaussian(m);
        addCLorentzian(m);

}