//~~~~~~~~~~~~~~~~~~~~~~~~~~~~~~~~~~~~~~~~~~~~~~~~~~~~~~~~~~~~~~~~~~~~~~~~~~~~~~~ // Copyright 2010 California Institute of Technology. ALL RIGHTS RESERVED. // // Licensed under the Apache License, Version 2.0 (the "License"); // you may not use this file except in compliance with the License. // You may obtain a copy of the License at // // http://www.apache.org/licenses/LICENSE-2.0 // // Unless required by applicable law or agreed to in writing, software // distributed under the License is distributed on an "AS IS" BASIS, // WITHOUT WARRANTIES OR CONDITIONS OF ANY KIND, either express or implied. // See the License for the specific language governing permissions and // limitations under the License. // // United States Government Sponsorship acknowledged. This software is subject to // U.S. export control laws and regulations and has been classified as 'EAR99 NLR' // (No [Export] License Required except when exporting to an embargoed country, // end user, or in support of a prohibited end use). By downloading this software, // the user agrees to comply with all applicable U.S. export laws and regulations. // The user has the responsibility to obtain export licenses, or other export // authority as may be required before exporting this software to any 'EAR99' // embargoed foreign country or citizen of those countries. // // Author: Giangi Sacco //~~~~~~~~~~~~~~~~~~~~~~~~~~~~~~~~~~~~~~~~~~~~~~~~~~~~~~~~~~~~~~~~~~~~~~~~~~~~~~~ #include #include "mocompbaselinemodule.h" #include #include #include #include #include #include using namespace std; static const char * const __doc__ = "Python extension for mocompbaseline"; PyModuleDef moduledef = { // header PyModuleDef_HEAD_INIT, // name of the module "mocompbaseline", // module documentation string __doc__, // size of the per-interpreter state of the module; // -1 if this state is global -1, mocompbaseline_methods, }; // initialization function for the module // *must* be called PyInit_mocompbaseline PyMODINIT_FUNC PyInit_mocompbaseline() { // 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 * 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 * allocate_sch1_C(PyObject* self, PyObject* args) { int dim1 = 0; int dim2 = 0; if(!PyArg_ParseTuple(args, "ii", &dim1, &dim2)) { return NULL; } allocate_sch1_f(&dim1, &dim2); return Py_BuildValue("i", 0); } PyObject * deallocate_sch1_C(PyObject* self, PyObject* args) { deallocate_sch1_f(); return Py_BuildValue("i", 0); } PyObject * allocate_sch2_C(PyObject* self, PyObject* args) { int dim1 = 0; int dim2 = 0; if(!PyArg_ParseTuple(args, "ii", &dim1, &dim2)) { return NULL; } allocate_sch2_f(&dim1, &dim2); return Py_BuildValue("i", 0); } PyObject * deallocate_sch2_C(PyObject* self, PyObject* args) { deallocate_sch2_f(); return Py_BuildValue("i", 0); } PyObject * allocate_s1_C(PyObject* self, PyObject* args) { int dim1 = 0; if(!PyArg_ParseTuple(args, "i", &dim1)) { return NULL; } allocate_s1_f(&dim1); return Py_BuildValue("i", 0); } PyObject * deallocate_s1_C(PyObject* self, PyObject* args) { deallocate_s1_f(); return Py_BuildValue("i", 0); } PyObject * allocate_is1_C(PyObject* self, PyObject* args) { int dim1 = 0; if(!PyArg_ParseTuple(args, "i", &dim1)) { return NULL; } allocate_is1_f(&dim1); return Py_BuildValue("i", 0); } PyObject * deallocate_is1_C(PyObject* self, PyObject* args) { deallocate_is1_f(); return Py_BuildValue("i", 0); } PyObject * allocate_s2_C(PyObject* self, PyObject* args) { int dim1 = 0; if(!PyArg_ParseTuple(args, "i", &dim1)) { return NULL; } allocate_s2_f(&dim1); return Py_BuildValue("i", 0); } PyObject * deallocate_s2_C(PyObject* self, PyObject* args) { deallocate_s2_f(); return Py_BuildValue("i", 0); } PyObject * allocate_is2_C(PyObject* self, PyObject* args) { int dim1 = 0; if(!PyArg_ParseTuple(args, "i", &dim1)) { return NULL; } allocate_is2_f(&dim1); return Py_BuildValue("i", 0); } PyObject * deallocate_is2_C(PyObject* self, PyObject* args) { deallocate_is2_f(); return Py_BuildValue("i", 0); } PyObject * allocate_baselineArray_C(PyObject* self, PyObject* args) { int dim1 = 0; int dim2 = 0; if(!PyArg_ParseTuple(args, "ii", &dim1, &dim2)) { return NULL; } allocate_baselineArray_f(&dim1, &dim2); return Py_BuildValue("i", 0); } PyObject * deallocate_baselineArray_C(PyObject* self, PyObject* args) { deallocate_baselineArray_f(); return Py_BuildValue("i", 0); } PyObject * mocompbaseline_C(PyObject* self, PyObject* args) { mocompbaseline_f(); return Py_BuildValue("i", 0); } PyObject * setSchPosition1_C(PyObject* self, PyObject* args) { PyObject * list; int dim1 = 0; int dim2 = 0; if(!PyArg_ParseTuple(args, "Oii", &list, &dim1, &dim2)) { 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*dim2]; for(int i = 0; i < dim1; ++i) { PyObject * listEl = PyList_GetItem(list,i); if(!PyList_Check(listEl)) { cout << "Error in file " << __FILE__ << " at line " << __LINE__ << ". Expecting a list type object" << endl; exit(1); } for(int j = 0; j < dim2; ++j) { PyObject * listElEl = PyList_GetItem(listEl,j); if(listElEl == NULL) { cout << "Error in file " << __FILE__ << " at line " << __LINE__ << ". Cannot retrieve list element" << endl; exit(1); } vectorV[dim2*i + j] = (double) PyFloat_AsDouble(listElEl); if(PyErr_Occurred() != NULL) { cout << "Error in file " << __FILE__ << " at line " << __LINE__ << ". Cannot convert Py Object to C " << endl; exit(1); } } } setSchPosition1_f(vectorV, &dim1, &dim2); delete [] vectorV; return Py_BuildValue("i", 0); } PyObject * setSchPosition2_C(PyObject* self, PyObject* args) { PyObject * list; int dim1 = 0; int dim2 = 0; if(!PyArg_ParseTuple(args, "Oii", &list, &dim1, &dim2)) { 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*dim2]; for(int i = 0; i < dim1; ++i) { PyObject * listEl = PyList_GetItem(list,i); if(!PyList_Check(listEl)) { cout << "Error in file " << __FILE__ << " at line " << __LINE__ << ". Expecting a list type object" << endl; exit(1); } for(int j = 0; j < dim2; ++j) { PyObject * listElEl = PyList_GetItem(listEl,j); if(listElEl == NULL) { cout << "Error in file " << __FILE__ << " at line " << __LINE__ << ". Cannot retrieve list element" << endl; exit(1); } vectorV[dim2*i + j] = (double) PyFloat_AsDouble(listElEl); if(PyErr_Occurred() != NULL) { cout << "Error in file " << __FILE__ << " at line " << __LINE__ << ". Cannot convert Py Object to C " << endl; exit(1); } } } setSchPosition2_f(vectorV, &dim1, &dim2); delete [] vectorV; return Py_BuildValue("i", 0); } PyObject * setMocompPosition1_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); } } setMocompPosition1_f(vectorV, &dim1); delete [] vectorV; return Py_BuildValue("i", 0); } PyObject * setMocompPositionIndex1_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); } int * vectorV = new int[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] = (int) PyLong_AsLong(listEl); if(PyErr_Occurred() != NULL) { cout << "Error in file " << __FILE__ << " at line " << __LINE__ << ". Cannot convert Py Object to C " << endl; exit(1); } } setMocompPositionIndex1_f(vectorV, &dim1); delete [] vectorV; return Py_BuildValue("i", 0); } PyObject * setMocompPosition2_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); } } setMocompPosition2_f(vectorV, &dim1); delete [] vectorV; return Py_BuildValue("i", 0); } PyObject * setMocompPositionIndex2_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); } int * vectorV = new int[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] = (int) PyLong_AsLong(listEl); if(PyErr_Occurred() != NULL) { cout << "Error in file " << __FILE__ << " at line " << __LINE__ << ". Cannot convert Py Object to C " << endl; exit(1); } } setMocompPositionIndex2_f(vectorV, &dim1); delete [] vectorV; return Py_BuildValue("i", 0); } PyObject * setEllipsoidMajorSemiAxis_C(PyObject* self, PyObject* args) { double var; if(!PyArg_ParseTuple(args, "d", &var)) { return NULL; } setEllipsoidMajorSemiAxis_f(&var); return Py_BuildValue("i", 0); } PyObject * setEllipsoidEccentricitySquared_C(PyObject* self, PyObject* args) { double var; if(!PyArg_ParseTuple(args, "d", &var)) { return NULL; } setEllipsoidEccentricitySquared_f(&var); return Py_BuildValue("i", 0); } PyObject * setPlanetLocalRadius_C(PyObject* self, PyObject* args) { double var; if(!PyArg_ParseTuple(args, "d", &var)) { return NULL; } setPlanetLocalRadius_f(&var); return Py_BuildValue("i", 0); } PyObject * setPegLatitude_C(PyObject* self, PyObject* args) { double var; if(!PyArg_ParseTuple(args, "d", &var)) { return NULL; } setPegLatitude_f(&var); return Py_BuildValue("i", 0); } PyObject * setPegLongitude_C(PyObject* self, PyObject* args) { double var; if(!PyArg_ParseTuple(args, "d", &var)) { return NULL; } setPegLongitude_f(&var); return Py_BuildValue("i", 0); } PyObject * setPegHeading_C(PyObject* self, PyObject* args) { double var; if(!PyArg_ParseTuple(args, "d", &var)) { return NULL; } setPegHeading_f(&var); return Py_BuildValue("i", 0); } PyObject * setHeight_C(PyObject* self, PyObject* args) { double var; if(!PyArg_ParseTuple(args, "d", &var)) { return NULL; } setHeight_f(&var); return Py_BuildValue("i", 0); } PyObject * getBaseline_C(PyObject* self, PyObject* args) { int dim1 = 0; int dim2 = 0; if(!PyArg_ParseTuple(args, "ii", &dim1, &dim2)) { return NULL; } PyObject * list1 = PyList_New(dim1); double * vectorV = new double[dim1*dim2]; getBaseline_f(vectorV, &dim1, &dim2); for(int i = 0; i < dim1; ++i) { PyObject * list2 = PyList_New(dim2); for(int j = 0; j < dim2; ++j) { PyObject * listEl = PyFloat_FromDouble( (double) vectorV[i*dim2 + j]); if(listEl == NULL) { cout << "Error in file " << __FILE__ << " at line " << __LINE__ << ". Cannot set list element" << endl; exit(1); } PyList_SetItem(list2,j,listEl); } PyList_SetItem(list1,i,list2); } delete [] vectorV; return Py_BuildValue("N",list1); } PyObject * allocate_midPointArray_C(PyObject* self, PyObject* args) { int dim1 = 0; int dim2 = 0; if(!PyArg_ParseTuple(args, "ii", &dim1, &dim2)) { return NULL; } allocate_midPointArray_f(&dim1, &dim2); return Py_BuildValue("i", 0); } PyObject * deallocate_midPointArray_C(PyObject* self, PyObject* args) { deallocate_midPointArray_f(); return Py_BuildValue("i", 0); } PyObject * allocate_midPointArray1_C(PyObject* self, PyObject* args) { int dim1 = 0; int dim2 = 0; if(!PyArg_ParseTuple(args, "ii", &dim1, &dim2)) { return NULL; } allocate_midPointArray1_f(&dim1, &dim2); return Py_BuildValue("i", 0); } PyObject * deallocate_midPointArray1_C(PyObject* self, PyObject* args) { deallocate_midPointArray1_f(); return Py_BuildValue("i", 0); } PyObject * allocate_midPointArray2_C(PyObject* self, PyObject* args) { int dim1 = 0; int dim2 = 0; if(!PyArg_ParseTuple(args, "ii", &dim1, &dim2)) { return NULL; } allocate_midPointArray2_f(&dim1, &dim2); return Py_BuildValue("i", 0); } PyObject * deallocate_midPointArray2_C(PyObject* self, PyObject* args) { deallocate_midPointArray2_f(); return Py_BuildValue("i", 0); } PyObject * allocate_baselineArray1_C(PyObject* self, PyObject* args) { int dim1 = 0; int dim2 = 0; if(!PyArg_ParseTuple(args, "ii", &dim1, &dim2)) { return NULL; } allocate_baselineArray1_f(&dim1, &dim2); return Py_BuildValue("i", 0); } PyObject * deallocate_baselineArray1_C(PyObject* self, PyObject* args) { deallocate_baselineArray1_f(); return Py_BuildValue("i", 0); } PyObject * allocate_baselineArray2_C(PyObject* self, PyObject* args) { int dim1 = 0; int dim2 = 0; if(!PyArg_ParseTuple(args, "ii", &dim1, &dim2)) { return NULL; } allocate_baselineArray2_f(&dim1, &dim2); return Py_BuildValue("i", 0); } PyObject * deallocate_baselineArray2_C(PyObject* self, PyObject* args) { deallocate_baselineArray2_f(); return Py_BuildValue("i", 0); } PyObject * allocate_schArray_C(PyObject* self, PyObject* args) { int dim1 = 0; int dim2 = 0; if(!PyArg_ParseTuple(args, "ii", &dim1, &dim2)) { return NULL; } allocate_schArray_f(&dim1, &dim2); return Py_BuildValue("i", 0); } PyObject * deallocate_schArray_C(PyObject* self, PyObject* args) { deallocate_schArray_f(); return Py_BuildValue("i", 0); } PyObject * allocate_scArray_C(PyObject* self, PyObject* args) { int dim1 = 0; int dim2 = 0; if(!PyArg_ParseTuple(args, "ii", &dim1, &dim2)) { return NULL; } allocate_scArray_f(&dim1, &dim2); return Py_BuildValue("i", 0); } PyObject * deallocate_scArray_C(PyObject* self, PyObject* args) { deallocate_scArray_f(); return Py_BuildValue("i", 0); } PyObject * getMidpoint_C(PyObject* self, PyObject* args) { int dim1 = 0; int dim2 = 0; if(!PyArg_ParseTuple(args, "ii", &dim1, &dim2)) { return NULL; } PyObject * list1 = PyList_New(dim1); double * vectorV = new double[dim1*dim2]; getMidpoint_f(vectorV, &dim1, &dim2); for(int i = 0; i < dim1; ++i) { PyObject * list2 = PyList_New(dim2); for(int j = 0; j < dim2; ++j) { PyObject * listEl = PyFloat_FromDouble( (double) vectorV[i*dim2 + j]); if(listEl == NULL) { cout << "Error in file " << __FILE__ << " at line " << __LINE__ << ". Cannot set list element" << endl; exit(1); } PyList_SetItem(list2,j,listEl); } PyList_SetItem(list1,i,list2); } delete [] vectorV; return Py_BuildValue("N",list1); } PyObject * getMidpoint1_C(PyObject* self, PyObject* args) { int dim1 = 0; int dim2 = 0; if(!PyArg_ParseTuple(args, "ii", &dim1, &dim2)) { return NULL; } PyObject * list1 = PyList_New(dim1); double * vectorV = new double[dim1*dim2]; getMidpoint1_f(vectorV, &dim1, &dim2); for(int i = 0; i < dim1; ++i) { PyObject * list2 = PyList_New(dim2); for(int j = 0; j < dim2; ++j) { PyObject * listEl = PyFloat_FromDouble( (double) vectorV[i*dim2 + j]); if(listEl == NULL) { cout << "Error in file " << __FILE__ << " at line " << __LINE__ << ". Cannot set list element" << endl; exit(1); } PyList_SetItem(list2,j,listEl); } PyList_SetItem(list1,i,list2); } delete [] vectorV; return Py_BuildValue("N",list1); } PyObject * getMidpoint2_C(PyObject* self, PyObject* args) { int dim1 = 0; int dim2 = 0; if(!PyArg_ParseTuple(args, "ii", &dim1, &dim2)) { return NULL; } PyObject * list1 = PyList_New(dim1); double * vectorV = new double[dim1*dim2]; getMidpoint2_f(vectorV, &dim1, &dim2); for(int i = 0; i < dim1; ++i) { PyObject * list2 = PyList_New(dim2); for(int j = 0; j < dim2; ++j) { PyObject * listEl = PyFloat_FromDouble( (double) vectorV[i*dim2 + j]); if(listEl == NULL) { cout << "Error in file " << __FILE__ << " at line " << __LINE__ << ". Cannot set list element" << endl; exit(1); } PyList_SetItem(list2,j,listEl); } PyList_SetItem(list1,i,list2); } delete [] vectorV; return Py_BuildValue("N",list1); } PyObject * getBaseline1_C(PyObject* self, PyObject* args) { int dim1 = 0; int dim2 = 0; if(!PyArg_ParseTuple(args, "ii", &dim1, &dim2)) { return NULL; } PyObject * list1 = PyList_New(dim1); double * vectorV = new double[dim1*dim2]; getBaseline1_f(vectorV, &dim1, &dim2); for(int i = 0; i < dim1; ++i) { PyObject * list2 = PyList_New(dim2); for(int j = 0; j < dim2; ++j) { PyObject * listEl = PyFloat_FromDouble( (double) vectorV[i*dim2 + j]); if(listEl == NULL) { cout << "Error in file " << __FILE__ << " at line " << __LINE__ << ". Cannot set list element" << endl; exit(1); } PyList_SetItem(list2,j,listEl); } PyList_SetItem(list1,i,list2); } delete [] vectorV; return Py_BuildValue("N",list1); } PyObject * getBaseline2_C(PyObject* self, PyObject* args) { int dim1 = 0; int dim2 = 0; if(!PyArg_ParseTuple(args, "ii", &dim1, &dim2)) { return NULL; } PyObject * list1 = PyList_New(dim1); double * vectorV = new double[dim1*dim2]; getBaseline2_f(vectorV, &dim1, &dim2); for(int i = 0; i < dim1; ++i) { PyObject * list2 = PyList_New(dim2); for(int j = 0; j < dim2; ++j) { PyObject * listEl = PyFloat_FromDouble( (double) vectorV[i*dim2 + j]); if(listEl == NULL) { cout << "Error in file " << __FILE__ << " at line " << __LINE__ << ". Cannot set list element" << endl; exit(1); } PyList_SetItem(list2,j,listEl); } PyList_SetItem(list1,i,list2); } delete [] vectorV; return Py_BuildValue("N",list1); } PyObject * getSch_C(PyObject* self, PyObject* args) { int dim1 = 0; int dim2 = 0; if(!PyArg_ParseTuple(args, "ii", &dim1, &dim2)) { return NULL; } PyObject * list1 = PyList_New(dim1); double * vectorV = new double[dim1*dim2]; getSch_f(vectorV, &dim1, &dim2); for(int i = 0; i < dim1; ++i) { PyObject * list2 = PyList_New(dim2); for(int j = 0; j < dim2; ++j) { PyObject * listEl = PyFloat_FromDouble( (double) vectorV[i*dim2 + j]); if(listEl == NULL) { cout << "Error in file " << __FILE__ << " at line " << __LINE__ << ". Cannot set list element" << endl; exit(1); } PyList_SetItem(list2,j,listEl); } PyList_SetItem(list1,i,list2); } delete [] vectorV; return Py_BuildValue("N",list1); } PyObject * getSc_C(PyObject* self, PyObject* args) { int dim1 = 0; int dim2 = 0; if(!PyArg_ParseTuple(args, "ii", &dim1, &dim2)) { return NULL; } PyObject * list1 = PyList_New(dim1); double * vectorV = new double[dim1*dim2]; getSc_f(vectorV, &dim1, &dim2); for(int i = 0; i < dim1; ++i) { PyObject * list2 = PyList_New(dim2); for(int j = 0; j < dim2; ++j) { PyObject * listEl = PyFloat_FromDouble( (double) vectorV[i*dim2 + j]); if(listEl == NULL) { cout << "Error in file " << __FILE__ << " at line " << __LINE__ << ". Cannot set list element" << endl; exit(1); } PyList_SetItem(list2,j,listEl); } PyList_SetItem(list1,i,list2); } delete [] vectorV; return Py_BuildValue("N",list1); } PyObject * get_dim1_s1_C(PyObject* self, PyObject* args) { int dim1 = 0; get_dim1_s1_f(&dim1); return Py_BuildValue("i", dim1); } // end of file