~i-martividal/uvmultifit/trunk-1

# Copyright (c) Ivan Marti-Vidal 2024

#               Universitat de Valencia (Spain)

 

 

// compiler warning that we use a deprecated NumPy API

// #define NPY_NO_DEPRECATED_API NPY_1_7_API_VERSION

//#define NO_IMPORT_ARRAY

#if PY_MAJOR_VERSION >= 3

#define NPY_NO_DEPRECATED_API 0x0

#endif

#include <numpy/npy_common.h>

 

 

 

 

// cribbed from SWIG machinery

#if PY_MAJOR_VERSION >= 3

#define PyClass_Check(obj) PyObject_IsInstance(obj, (PyObject *)&PyType_Type)

#define PyInt_Check(x) PyLong_Check(x)

#define PyInt_AsLong(x) PyLong_AsLong(x)

#define PyInt_FromLong(x) PyLong_FromLong(x)

#define PyInt_FromSize_t(x) PyLong_FromSize_t(x)

#define PyString_Check(name) PyBytes_Check(name)

#define PyString_FromString(x) PyUnicode_FromString(x)

#define PyString_Format(fmt, args)  PyUnicode_Format(fmt, args)

//#define PyString_AsString(str) PyBytes_AsString(str)

#define PyString_Size(str) PyBytes_Size(str)

#define PyString_InternFromString(key) PyUnicode_InternFromString(key)

#define Py_TPFLAGS_HAVE_CLASS Py_TPFLAGS_BASETYPE

#define PyString_AS_STRING(x) PyUnicode_AS_STRING(x)

#define _PyLong_FromSsize_t(x) PyLong_FromSsize_t(x)

// For PyArray_FromDimsAndData -> PyArray_SimpleNewFromData

//#define INTEGER long

//#define INTEGERCAST  (const npy_intp*)

//#else

// For PyArray_FromDimsAndData -> PyArray_SimpleNewFromData

//#define INTEGER int

//#define INTEGERCAST (long int *)

#endif

// and after some hacking

#if PY_MAJOR_VERSION >= 3

#define PyString_AsString(obj) PyUnicode_AsUTF8(obj)

#endif

 

 

 

 

 

 

 

// normally abort() is called on problems, which breaks CASA.

// here we report and save the error condition which can be

// noticed for a cleaner exit.

//int gsl_death_by = GSL_SUCCESS;

//static void gsl_death(const char * reason, const char * file,

//    int line, int gsl_errno) {

//    // stderr does not end up synchronized with stdout

//    printf("GSL Death by '%s' in file %s at line %d: GSL Error %d\n",

//        reason, file, line, gsl_errno);

//    fflush(stdout); std::cout << std::flush;

//    gsl_death_by = gsl_errno;

//}

 

 

 

 

 

 

 

#if PY_MAJOR_VERSION >= 3

static struct PyModuleDef pc_module_def = {

    PyModuleDef_HEAD_INIT,

    "_uvmultimodel",         /* m_name */

    module_docstring,       /* m_doc */

    -1,                     /* m_size */

    module_methods,         /* m_methods */

    NULL,NULL,NULL,NULL     /* m_reload, m_traverse, m_clear, m_free */

};

PyMODINIT_FUNC PyInit__uvmultimodel(void)

{

    PyObject *m = PyModule_Create(&pc_module_def);

  //  import_array();

  //  (void)gsl_set_error_handler(gsl_death);

 

//////////////////////

// Initiate variables with dummy values:

    NCPU = 0; Nspw = 0; npar = -1; Nants = 0; ncomp = -1; Nbas = 0;

    master.t0 = new int[1];

    master.t1 = new int[1];

 

    return(m);

}

 

 

 

#else

 

#endif

 

 

 

 

/* Clears pointers to the data

WARNING: In principle, this shouldn't be done, since the

data in these pointers may still be used by Python!!

*/

/*

*/

/* WARNING! There are pointers here that may still

be used by Python! */

//      delete mod.vars;

//      delete mod.fixp;

//      delete mod.parAnt;

//      delete mod.nparAnt;

//      delete mod.models;

isModel = false;

 

 

    vis.ants[0][IF] = (int *)PyArray_DATA(ant1l);

    vis.ants[1][IF] = (int *)PyArray_DATA(ant2l);

 

 

    vis.dtArray[IF] = (double *)PyArray_DATA(tArr);

    vis.dtIndex[IF] = (int *)PyArray_DATA(tIdx);

 

    vis.uv[0][IF] = (double *)PyArray_DATA(pu);

    vis.uv[1][IF] = (double *)PyArray_DATA(pv);

    vis.uv[2][IF] = (double *)PyArray_DATA(pw);

 

    vis.wgt[0][IF] = (double *)PyArray_DATA(pwgt);

    vis.ObsVis[IF] = (cplx64 *)PyArray_DATA(preal);

 

    mod.ModVis[IF] = (cplx64 *)PyArray_DATA(poreal);

 

    vis.freqs[IF] = (double *)PyArray_DATA(pfreqs);

 

    vis.fittable[IF] = (int *)PyArray_DATA(pfittable);

    vis.wgt[1][IF] = (double *)PyArray_DATA(pwgtcorr);

 

    vis.isGain[IF] = (int *)PyArray_DATA(iG);

 

 

    vis.dt[IF] = (double *)PyArray_DATA(dtime);

 

    vis.RAshift[IF] = (double *)PyArray_DATA(RAoffset);

 

    vis.Decshift[IF] = (double *)PyArray_DATA(Decoffset);

 

    vis.Stretch[IF] = (double *)PyArray_DATA(Stretchoff);

    mod.models = (int *)PyArray_DATA(modArr);

    mod.Hessian = (double *)PyArray_DATA(HessArr);

    mod.Gradient = (double *)PyArray_DATA(GradArr);

    mod.dpar = (double *)PyArray_DATA(dparArr);

    mod.muRA = (double *)PyArray_DATA(propRA);

    mod.muDec = (double *)PyArray_DATA(propDec);

    vis.phaseCenter = (double *)PyArray_DATA(refpos);

    mod.triSpec = (int *)PyArray_DATA(triSpec);

      mod.parAnt[i] = (int *)PyArray_DATA(PyList_GetItem(parDep,i));

          mod.Gain[IF][j][i] = (cplx64 *)PyArray_DATA(tempArr);

      mod.vars[i] = (double *)PyArray_DATA(tempArr);

      mod.fixp[i] = (double *)PyArray_DATA(tempArr);

  PyObject *ret, *gainList;

    if (!PyArg_ParseTuple(args, "iiiiO",&IFFit, &refant, &doModel, &doGlobal, &gainList)){printf("FAILED QuinnFringe!\n"); fflush(stdout);  ret = Py_BuildValue("i",-1); return ret;};

//double *Rates = new double[Nants]; 

//double *Delays = new double[Nants];

//double *Phases = new double[Nants];

 

double *Delays = (double *)PyArray_DATA(PyList_GetItem(gainList,0));

double *Rates = (double *)PyArray_DATA(PyList_GetItem(gainList,1));

double *Phases = (double *)PyArray_DATA(PyList_GetItem(gainList,2));

//PyObject *PyRate, *PyDelay, *PyPhase;

npy_intp Dims[1];

//int *Dims = new int[1];

Dims[0] = Nants;

//PyRate = PyArray_SimpleNewFromData(1, Dims, NPY_FLOAT64, (void *) Rates);

//PyDelay = PyArray_SimpleNewFromData(1, Dims, NPY_FLOAT64, (void *) Delays);

//PyPhase = PyArray_SimpleNewFromData(1, Dims, NPY_FLOAT64, (void *) Phases);

//ret = Py_BuildValue("[O,O,O,d,d]",PyDelay,PyRate,PyPhase,Bins[0],Bins[1]);

ret = Py_BuildValue("[d,d]",Bins[0],Bins[1]);