513 lines
14 KiB
C++
513 lines
14 KiB
C++
#include <Python.h>
|
|
#include "fitoffmodule.h"
|
|
#include <cmath>
|
|
#include <sstream>
|
|
#include <iostream>
|
|
#include <string>
|
|
#include <vector>
|
|
using namespace std;
|
|
|
|
static const char * const __doc__ = "Python extension for fitoff.F";
|
|
|
|
PyModuleDef moduledef = {
|
|
// header
|
|
PyModuleDef_HEAD_INIT,
|
|
// name of the module
|
|
"fitoff",
|
|
// module documentation string
|
|
__doc__,
|
|
// size of the per-interpreter state of the module;
|
|
// -1 if this state is global
|
|
-1,
|
|
fitoff_methods,
|
|
};
|
|
|
|
// initialization function for the module
|
|
// *must* be called PyInit_formslc
|
|
PyMODINIT_FUNC
|
|
PyInit_fitoff()
|
|
{
|
|
// create the module using moduledef struct defined above
|
|
PyObject * module = PyModule_Create(&moduledef);
|
|
// check whether module creation succeeded and raise an exception if not
|
|
if (!module) {
|
|
return module;
|
|
}
|
|
// otherwise, we have an initialized module
|
|
// and return the newly created module
|
|
return module;
|
|
}
|
|
|
|
|
|
PyObject * fitoff_C(PyObject * self, PyObject * args)
|
|
{
|
|
fitoff_f();
|
|
return Py_BuildValue("i", 0);
|
|
|
|
}
|
|
PyObject * setStdWriter_C(PyObject* self, PyObject* args)
|
|
{
|
|
uint64_t var;
|
|
if(!PyArg_ParseTuple(args, "K", &var))
|
|
{
|
|
return NULL;
|
|
}
|
|
setStdWriter_f(&var);
|
|
return Py_BuildValue("i",0);
|
|
}
|
|
PyObject * setMinPoint_C(PyObject* self, PyObject* args)
|
|
{
|
|
int varInt;
|
|
if(!PyArg_ParseTuple(args, "i", &varInt))
|
|
{
|
|
return NULL;
|
|
}
|
|
setMinPoint_f(&varInt);
|
|
return Py_BuildValue("i", 0);
|
|
}
|
|
PyObject * setNSig_C(PyObject* self, PyObject* args)
|
|
{
|
|
double varDouble;
|
|
if(!PyArg_ParseTuple(args, "d", &varDouble))
|
|
{
|
|
return NULL;
|
|
}
|
|
setNSig_f(&varDouble);
|
|
return Py_BuildValue("i", 0);
|
|
}
|
|
PyObject * setMaxRms_C(PyObject* self, PyObject* args)
|
|
{
|
|
double varDouble;
|
|
if(!PyArg_ParseTuple(args, "d", &varDouble))
|
|
{
|
|
return NULL;
|
|
}
|
|
setMaxRms_f(&varDouble);
|
|
return Py_BuildValue("i", 0);
|
|
}
|
|
PyObject * setNumberLines_C(PyObject* self, PyObject* args)
|
|
{
|
|
int varInt;
|
|
if(!PyArg_ParseTuple(args, "i", &varInt))
|
|
{
|
|
return NULL;
|
|
}
|
|
setNumberLines_f(&varInt);
|
|
return Py_BuildValue("i", 0);
|
|
}
|
|
PyObject * setMaxIter_C(PyObject* self, PyObject* args)
|
|
{
|
|
int varInt;
|
|
if(!PyArg_ParseTuple(args, "i", &varInt))
|
|
{
|
|
return NULL;
|
|
}
|
|
setMaxIter_f(&varInt);
|
|
return Py_BuildValue("i",0);
|
|
}
|
|
PyObject * setMinIter_C(PyObject* self, PyObject* args)
|
|
{
|
|
int varInt;
|
|
if(!PyArg_ParseTuple(args, "i", &varInt))
|
|
{
|
|
return NULL;
|
|
}
|
|
setMinIter_f(&varInt);
|
|
return Py_BuildValue("i",0);
|
|
}
|
|
|
|
PyObject * setL1normFlag_C(PyObject* self, PyObject *args)
|
|
{
|
|
int varInt;
|
|
if(!PyArg_ParseTuple(args, "i", &varInt))
|
|
{
|
|
return NULL;
|
|
}
|
|
setL1normFlag_f(&varInt);
|
|
return Py_BuildValue("i",0);
|
|
}
|
|
PyObject * setLocationAcross_C(PyObject* self, PyObject* args)
|
|
{
|
|
int dim1=0;
|
|
PyObject * list;
|
|
if(!PyArg_ParseTuple(args,"Oi", &list, &dim1))
|
|
{
|
|
return NULL;
|
|
}
|
|
if(!PyList_Check(list))
|
|
{
|
|
cout << "Error in file " << __FILE__ << " at line " << __LINE__ << ". Expecting a list type object" << endl;
|
|
exit(1);
|
|
}
|
|
double *vectorV = new double[dim1];
|
|
for(int i=0; i<dim1; ++i)
|
|
{
|
|
PyObject * listEl = PyList_GetItem(list,i);
|
|
if(listEl == NULL)
|
|
{
|
|
cout << "Error in file " << __FILE__ << " at line " << __LINE__ << ". Cannot retrieve list element" << endl;
|
|
exit(1);
|
|
}
|
|
vectorV[i] = (double) PyFloat_AsDouble(listEl);
|
|
if(PyErr_Occurred() != NULL)
|
|
{
|
|
cout << "Error in file " << __FILE__ << " at line "<< __LINE__ << ". Cannot convert Py Object to C " << endl;
|
|
exit(1);
|
|
}
|
|
}
|
|
setLocationAcross_f(vectorV, &dim1);
|
|
delete [] vectorV;
|
|
return Py_BuildValue("i", 0);
|
|
}
|
|
PyObject * setLocationDown_C(PyObject* self, PyObject* args)
|
|
{
|
|
int dim1=0;
|
|
PyObject * list;
|
|
if(!PyArg_ParseTuple(args,"Oi", &list, &dim1))
|
|
{
|
|
return NULL;
|
|
}
|
|
if(!PyList_Check(list))
|
|
{
|
|
cout << "Error in file " << __FILE__ << " at line " << __LINE__ << ". Expecting a list type object" << endl;
|
|
exit(1);
|
|
}
|
|
double *vectorV = new double[dim1];
|
|
for(int i=0; i<dim1; ++i)
|
|
{
|
|
PyObject * listEl = PyList_GetItem(list,i);
|
|
if(listEl == NULL)
|
|
{
|
|
cout << "Error in file " << __FILE__ << " at line " << __LINE__ << ". Cannot retrieve list element" << endl;
|
|
exit(1);
|
|
}
|
|
vectorV[i] = (double) PyFloat_AsDouble(listEl);
|
|
if(PyErr_Occurred() != NULL)
|
|
{
|
|
cout << "Error in file " << __FILE__ << " at line "<< __LINE__ << ". Cannot convert Py Object to C " << endl;
|
|
exit(1);
|
|
}
|
|
}
|
|
setLocationDown_f(vectorV, &dim1);
|
|
delete [] vectorV;
|
|
return Py_BuildValue("i", 0);
|
|
}
|
|
PyObject * setLocationDownOffset_C(PyObject* self, PyObject* args)
|
|
{
|
|
int dim1=0;
|
|
PyObject * list;
|
|
if(!PyArg_ParseTuple(args,"Oi", &list, &dim1))
|
|
{
|
|
return NULL;
|
|
}
|
|
if(!PyList_Check(list))
|
|
{
|
|
cout << "Error in file " << __FILE__ << " at line " << __LINE__ << ". Expecting a list type object" << endl;
|
|
exit(1);
|
|
}
|
|
double *vectorV = new double[dim1];
|
|
for(int i=0; i<dim1; ++i)
|
|
{
|
|
PyObject * listEl = PyList_GetItem(list,i);
|
|
if(listEl == NULL)
|
|
{
|
|
cout << "Error in file " << __FILE__ << " at line " << __LINE__ << ". Cannot retrieve list element" << endl;
|
|
exit(1);
|
|
}
|
|
vectorV[i] = (double) PyFloat_AsDouble(listEl);
|
|
if(PyErr_Occurred() != NULL)
|
|
{
|
|
cout << "Error in file " << __FILE__ << " at line "<< __LINE__ << ". Cannot convert Py Object to C " << endl;
|
|
exit(1);
|
|
}
|
|
}
|
|
setLocationDownOffset_f(vectorV, &dim1);
|
|
delete [] vectorV;
|
|
return Py_BuildValue("i", 0);
|
|
}
|
|
PyObject * setLocationAcrossOffset_C(PyObject* self, PyObject* args)
|
|
{
|
|
int dim1=0;
|
|
PyObject * list;
|
|
if(!PyArg_ParseTuple(args,"Oi", &list, &dim1))
|
|
{
|
|
return NULL;
|
|
}
|
|
if(!PyList_Check(list))
|
|
{
|
|
cout << "Error in file " << __FILE__ << " at line " << __LINE__ << ". Expecting a list type object" << endl;
|
|
exit(1);
|
|
}
|
|
double *vectorV = new double[dim1];
|
|
for(int i=0; i<dim1; ++i)
|
|
{
|
|
PyObject * listEl = PyList_GetItem(list,i);
|
|
if(listEl == NULL)
|
|
{
|
|
cout << "Error in file " << __FILE__ << " at line " << __LINE__ << ". Cannot retrieve list element" << endl;
|
|
exit(1);
|
|
}
|
|
vectorV[i] = (double) PyFloat_AsDouble(listEl);
|
|
if(PyErr_Occurred() != NULL)
|
|
{
|
|
cout << "Error in file " << __FILE__ << " at line "<< __LINE__ << ". Cannot convert Py Object to C " << endl;
|
|
exit(1);
|
|
}
|
|
}
|
|
setLocationAcrossOffset_f(vectorV, &dim1);
|
|
delete [] vectorV;
|
|
return Py_BuildValue("i", 0);
|
|
}
|
|
PyObject * setSNR_C(PyObject* self, PyObject* args)
|
|
{
|
|
int dim1=0;
|
|
PyObject * list;
|
|
if(!PyArg_ParseTuple(args,"Oi", &list, &dim1))
|
|
{
|
|
return NULL;
|
|
}
|
|
if(!PyList_Check(list))
|
|
{
|
|
cout << "Error in file " << __FILE__ << " at line " << __LINE__ << ". Expecting a list type object" << endl;
|
|
exit(1);
|
|
}
|
|
double *vectorV = new double[dim1];
|
|
for(int i=0; i<dim1; ++i)
|
|
{
|
|
PyObject * listEl = PyList_GetItem(list,i);
|
|
if(listEl == NULL)
|
|
{
|
|
cout << "Error in file " << __FILE__ << " at line " << __LINE__ << ". Cannot retrieve list element" << endl;
|
|
exit(1);
|
|
}
|
|
vectorV[i] = (double) PyFloat_AsDouble(listEl);
|
|
if(PyErr_Occurred() != NULL)
|
|
{
|
|
cout << "Error in file " << __FILE__ << " at line "<< __LINE__ << ". Cannot convert Py Object to C " << endl;
|
|
exit(1);
|
|
}
|
|
}
|
|
setSNR_f(vectorV, &dim1);
|
|
delete [] vectorV;
|
|
return Py_BuildValue("i", 0);
|
|
}
|
|
PyObject * setCovDown_C(PyObject* self, PyObject* args)
|
|
{
|
|
int dim1=0;
|
|
PyObject * list;
|
|
if(!PyArg_ParseTuple(args,"Oi", &list, &dim1))
|
|
{
|
|
return NULL;
|
|
}
|
|
if(!PyList_Check(list))
|
|
{
|
|
cout << "Error in file " << __FILE__ << " at line " << __LINE__ << ". Expecting a list type object" << endl;
|
|
exit(1);
|
|
}
|
|
double *vectorV = new double[dim1];
|
|
for(int i=0; i<dim1; ++i)
|
|
{
|
|
PyObject * listEl = PyList_GetItem(list,i);
|
|
if(listEl == NULL)
|
|
{
|
|
cout << "Error in file " << __FILE__ << " at line " << __LINE__ << ". Cannot retrieve list element" << endl;
|
|
exit(1);
|
|
}
|
|
vectorV[i] = (double) PyFloat_AsDouble(listEl);
|
|
if(PyErr_Occurred() != NULL)
|
|
{
|
|
cout << "Error in file " << __FILE__ << " at line "<< __LINE__ << ". Cannot convert Py Object to C " << endl;
|
|
exit(1);
|
|
}
|
|
}
|
|
setCovDown_f(vectorV, &dim1);
|
|
delete [] vectorV;
|
|
return Py_BuildValue("i", 0);
|
|
}
|
|
|
|
PyObject * setCovAcross_C(PyObject* self, PyObject* args)
|
|
{
|
|
int dim1=0;
|
|
PyObject * list;
|
|
if(!PyArg_ParseTuple(args,"Oi", &list, &dim1))
|
|
{
|
|
return NULL;
|
|
}
|
|
if(!PyList_Check(list))
|
|
{
|
|
cout << "Error in file " << __FILE__ << " at line " << __LINE__ << ". Expecting a list type object" << endl;
|
|
exit(1);
|
|
}
|
|
double *vectorV = new double[dim1];
|
|
for(int i=0; i<dim1; ++i)
|
|
{
|
|
PyObject * listEl = PyList_GetItem(list,i);
|
|
if(listEl == NULL)
|
|
{
|
|
cout << "Error in file " << __FILE__ << " at line " << __LINE__ << ". Cannot retrieve list element" << endl;
|
|
exit(1);
|
|
}
|
|
vectorV[i] = (double) PyFloat_AsDouble(listEl);
|
|
if(PyErr_Occurred() != NULL)
|
|
{
|
|
cout << "Error in file " << __FILE__ << " at line "<< __LINE__ << ". Cannot convert Py Object to C " << endl;
|
|
exit(1);
|
|
}
|
|
}
|
|
setCovAcross_f(vectorV, &dim1);
|
|
delete [] vectorV;
|
|
return Py_BuildValue("i", 0);
|
|
}
|
|
|
|
PyObject * setCovCross_C(PyObject* self, PyObject* args)
|
|
{
|
|
int dim1=0;
|
|
PyObject * list;
|
|
if(!PyArg_ParseTuple(args,"Oi", &list, &dim1))
|
|
{
|
|
return NULL;
|
|
}
|
|
if(!PyList_Check(list))
|
|
{
|
|
cout << "Error in file " << __FILE__ << " at line " << __LINE__ << ". Expecting a list type object" << endl;
|
|
exit(1);
|
|
}
|
|
double *vectorV = new double[dim1];
|
|
for(int i=0; i<dim1; ++i)
|
|
{
|
|
PyObject * listEl = PyList_GetItem(list,i);
|
|
if(listEl == NULL)
|
|
{
|
|
cout << "Error in file " << __FILE__ << " at line " << __LINE__ << ". Cannot retrieve list element" << endl;
|
|
exit(1);
|
|
}
|
|
vectorV[i] = (double) PyFloat_AsDouble(listEl);
|
|
if(PyErr_Occurred() != NULL)
|
|
{
|
|
cout << "Error in file " << __FILE__ << " at line "<< __LINE__ << ". Cannot convert Py Object to C " << endl;
|
|
exit(1);
|
|
}
|
|
}
|
|
setCovCross_f(vectorV, &dim1);
|
|
delete [] vectorV;
|
|
return Py_BuildValue("i", 0);
|
|
|
|
}
|
|
|
|
PyObject * getAffineVector_C(PyObject* self, PyObject* args)
|
|
{
|
|
int numElements = 6;
|
|
vector<double> affineVec(numElements,0);
|
|
|
|
getAffineVector_f(&affineVec[0]);
|
|
PyObject * pyList = PyList_New(numElements);
|
|
if(!pyList)
|
|
{
|
|
cout << "Error at line " << __LINE__ << " in file " << __FILE__ ". Exiting ..."<< endl;
|
|
exit(1);
|
|
}
|
|
for(int i = 0; i < numElements; ++i)
|
|
{
|
|
PyList_SetItem(pyList,i, PyFloat_FromDouble(affineVec[i]));
|
|
}
|
|
return Py_BuildValue("O", pyList);
|
|
|
|
}
|
|
|
|
PyObject *getNumberOfRefinedOffsets_C(PyObject* self, PyObject* args)
|
|
{
|
|
int numElements = 0;
|
|
getNumberOfRefinedOffsets_f(&numElements);
|
|
return Py_BuildValue("i", numElements);
|
|
}
|
|
|
|
PyObject *getRefinedOffsetField_C(PyObject* self, PyObject* args)
|
|
{
|
|
int numElements = 0;
|
|
int nValues = 8;
|
|
|
|
if(!PyArg_ParseTuple(args, "i", &numElements))
|
|
{
|
|
return NULL;
|
|
}
|
|
|
|
|
|
double *acLoc = new double[numElements];
|
|
double *dnLoc = new double[numElements];
|
|
double *acOff = new double[numElements];
|
|
double *dnOff = new double[numElements];
|
|
double *snr = new double[numElements];
|
|
double *covAc = new double[numElements];
|
|
double *covDn = new double[numElements];
|
|
double *covX = new double[numElements];
|
|
|
|
getRefinedLocationAcross_f(acLoc);
|
|
getRefinedLocationDown_f(dnLoc);
|
|
getRefinedLocationAcrossOffset_f(acOff);
|
|
getRefinedLocationDownOffset_f(dnOff);
|
|
getRefinedSNR_f(snr);
|
|
getRefinedCovAcross_f(covAc);
|
|
getRefinedCovDown_f(covDn);
|
|
getRefinedCovCross_f(covX);
|
|
|
|
PyObject *pyList = PyList_New(numElements);
|
|
if(!pyList)
|
|
{
|
|
cout << "Error at line " << __LINE__ << "in file " << __FILE__ ". Exiting ..." << endl;
|
|
exit(1);
|
|
}
|
|
for(int i=0; i<numElements; i++)
|
|
{
|
|
PyObject *nList = PyList_New(nValues);
|
|
if(!nList)
|
|
{
|
|
cout << "Error at line " << __LINE__ << " in file " << __FILE__ ". Exiting ..." << endl;
|
|
exit(1);
|
|
}
|
|
|
|
PyList_SetItem(nList, 0, PyFloat_FromDouble(acLoc[i]));
|
|
PyList_SetItem(nList, 1, PyFloat_FromDouble(dnLoc[i]));
|
|
PyList_SetItem(nList, 2, PyFloat_FromDouble(acOff[i]));
|
|
PyList_SetItem(nList, 3, PyFloat_FromDouble(dnOff[i]));
|
|
PyList_SetItem(nList, 4, PyFloat_FromDouble(snr[i]));
|
|
PyList_SetItem(nList, 5, PyFloat_FromDouble(covAc[i]));
|
|
PyList_SetItem(nList, 6, PyFloat_FromDouble(covDn[i]));
|
|
PyList_SetItem(nList, 7, PyFloat_FromDouble(covX[i]));
|
|
|
|
PyList_SetItem(pyList, i , nList);
|
|
}
|
|
|
|
delete [] acLoc;
|
|
delete [] dnLoc;
|
|
delete [] acOff;
|
|
delete [] dnOff;
|
|
delete [] snr;
|
|
delete [] covAc;
|
|
delete [] covDn;
|
|
delete [] covX;
|
|
|
|
return Py_BuildValue("O", pyList);
|
|
}
|
|
|
|
PyObject * allocate_Arrays_C(PyObject* self, PyObject *args)
|
|
{
|
|
allocate_LocationAcross_f();
|
|
allocate_LocationDown_f();
|
|
allocate_LocationAcrossOffset_f();
|
|
allocate_LocationDownOffset_f();
|
|
allocate_SNR_f();
|
|
allocate_Covariance_f();
|
|
return Py_BuildValue("i", 0);
|
|
}
|
|
|
|
PyObject * deallocate_Arrays_C(PyObject *self, PyObject *args)
|
|
{
|
|
deallocate_LocationAcross_f();
|
|
deallocate_LocationDown_f();
|
|
deallocate_LocationAcrossOffset_f();
|
|
deallocate_LocationDownOffset_f();
|
|
deallocate_SNR_f();
|
|
deallocate_Covariance_f();
|
|
return Py_BuildValue("i", 0);
|
|
}
|