527 lines
14 KiB
C++
527 lines
14 KiB
C++
|
//~~~~~~~~~~~~~~~~~~~~~~~~~~~~~~~~~~~~~~~~~~~~~~~~~~~~~~~~~~~~~~~~~~~~~~~~~~~~~~~
|
||
|
// Copyright 2012 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 <Python.h>
|
||
|
#include "estimateoffsetsmodule.h"
|
||
|
#include <cmath>
|
||
|
#include <sstream>
|
||
|
#include <iostream>
|
||
|
#include <string>
|
||
|
#include <stdint.h>
|
||
|
#include <vector>
|
||
|
using namespace std;
|
||
|
|
||
|
static const char * const __doc__ = "Python extension for estimateoffsets.F";
|
||
|
|
||
|
PyModuleDef moduledef = {
|
||
|
// header
|
||
|
PyModuleDef_HEAD_INIT,
|
||
|
// name of the module
|
||
|
"estimateoffsets",
|
||
|
// module documentation string
|
||
|
__doc__,
|
||
|
// size of the per-interpreter state of the module;
|
||
|
// -1 if this state is global
|
||
|
-1,
|
||
|
estimateoffsets_methods,
|
||
|
};
|
||
|
|
||
|
// initialization function for the module
|
||
|
// *must* be called PyInit_formslc
|
||
|
PyMODINIT_FUNC
|
||
|
PyInit_estimateoffsets()
|
||
|
{
|
||
|
// 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 * allocate_locationAcross_C(PyObject* self, PyObject* args)
|
||
|
{
|
||
|
int dim1 = 0;
|
||
|
if(!PyArg_ParseTuple(args, "i", &dim1))
|
||
|
{
|
||
|
return NULL;
|
||
|
}
|
||
|
allocate_locationAcross_f(&dim1);
|
||
|
return Py_BuildValue("i", 0);
|
||
|
}
|
||
|
|
||
|
PyObject * deallocate_locationAcross_C(PyObject* self, PyObject* args)
|
||
|
{
|
||
|
deallocate_locationAcross_f();
|
||
|
return Py_BuildValue("i", 0);
|
||
|
}
|
||
|
|
||
|
PyObject * allocate_locationAcrossOffset_C(PyObject* self, PyObject* args)
|
||
|
{
|
||
|
int dim1 = 0;
|
||
|
if(!PyArg_ParseTuple(args, "i", &dim1))
|
||
|
{
|
||
|
return NULL;
|
||
|
}
|
||
|
allocate_locationAcrossOffset_f(&dim1);
|
||
|
return Py_BuildValue("i", 0);
|
||
|
}
|
||
|
|
||
|
PyObject * deallocate_locationAcrossOffset_C(PyObject* self, PyObject* args)
|
||
|
{
|
||
|
deallocate_locationAcrossOffset_f();
|
||
|
return Py_BuildValue("i", 0);
|
||
|
}
|
||
|
|
||
|
PyObject * allocate_locationDown_C(PyObject* self, PyObject* args)
|
||
|
{
|
||
|
int dim1 = 0;
|
||
|
if(!PyArg_ParseTuple(args, "i", &dim1))
|
||
|
{
|
||
|
return NULL;
|
||
|
}
|
||
|
allocate_locationDown_f(&dim1);
|
||
|
return Py_BuildValue("i", 0);
|
||
|
}
|
||
|
|
||
|
PyObject * deallocate_locationDown_C(PyObject* self, PyObject* args)
|
||
|
{
|
||
|
deallocate_locationDown_f();
|
||
|
return Py_BuildValue("i", 0);
|
||
|
}
|
||
|
|
||
|
PyObject * allocate_locationDownOffset_C(PyObject* self, PyObject* args)
|
||
|
{
|
||
|
int dim1 = 0;
|
||
|
if(!PyArg_ParseTuple(args, "i", &dim1))
|
||
|
{
|
||
|
return NULL;
|
||
|
}
|
||
|
allocate_locationDownOffset_f(&dim1);
|
||
|
return Py_BuildValue("i", 0);
|
||
|
}
|
||
|
|
||
|
PyObject * deallocate_locationDownOffset_C(PyObject* self, PyObject* args)
|
||
|
{
|
||
|
deallocate_locationDownOffset_f();
|
||
|
return Py_BuildValue("i", 0);
|
||
|
}
|
||
|
|
||
|
PyObject * allocate_snrRet_C(PyObject* self, PyObject* args)
|
||
|
{
|
||
|
int dim1 = 0;
|
||
|
if(!PyArg_ParseTuple(args, "i", &dim1))
|
||
|
{
|
||
|
return NULL;
|
||
|
}
|
||
|
allocate_snrRet_f(&dim1);
|
||
|
return Py_BuildValue("i", 0);
|
||
|
}
|
||
|
|
||
|
PyObject * deallocate_snrRet_C(PyObject* self, PyObject* args)
|
||
|
{
|
||
|
deallocate_snrRet_f();
|
||
|
return Py_BuildValue("i", 0);
|
||
|
}
|
||
|
|
||
|
PyObject * estimateoffsets_C(PyObject* self, PyObject* args)
|
||
|
{
|
||
|
uint64_t var0;
|
||
|
uint64_t var1;
|
||
|
if(!PyArg_ParseTuple(args, "KK",&var0,&var1))
|
||
|
{
|
||
|
return NULL;
|
||
|
}
|
||
|
estimateoffsets_f(&var0,&var1);
|
||
|
return Py_BuildValue("i", 0);
|
||
|
}
|
||
|
PyObject * getLocationAcross_C(PyObject* self, PyObject* args)
|
||
|
{
|
||
|
int dim1 = 0;
|
||
|
if(!PyArg_ParseTuple(args, "i", &dim1))
|
||
|
{
|
||
|
return NULL;
|
||
|
}
|
||
|
PyObject * list = PyList_New(dim1);
|
||
|
int * vectorV = new int[dim1];
|
||
|
getLocationAcross_f(vectorV, &dim1);
|
||
|
for(int i = 0; i < dim1; ++i)
|
||
|
{
|
||
|
PyObject * listEl = PyLong_FromLong((long int) vectorV[i]);
|
||
|
if(listEl == NULL)
|
||
|
{
|
||
|
cout << "Error in file " << __FILE__ << " at line " << __LINE__ << ". Cannot set list element" << endl;
|
||
|
exit(1);
|
||
|
}
|
||
|
PyList_SetItem(list,i, listEl);
|
||
|
}
|
||
|
delete [] vectorV;
|
||
|
return Py_BuildValue("N",list);
|
||
|
}
|
||
|
|
||
|
PyObject * getLocationAcrossOffset_C(PyObject* self, PyObject* args)
|
||
|
{
|
||
|
int dim1 = 0;
|
||
|
if(!PyArg_ParseTuple(args, "i", &dim1))
|
||
|
{
|
||
|
return NULL;
|
||
|
}
|
||
|
PyObject * list = PyList_New(dim1);
|
||
|
float * vectorV = new float[dim1];
|
||
|
getLocationAcrossOffset_f(vectorV, &dim1);
|
||
|
for(int i = 0; i < dim1; ++i)
|
||
|
{
|
||
|
PyObject * listEl = PyFloat_FromDouble((double) vectorV[i]);
|
||
|
if(listEl == NULL)
|
||
|
{
|
||
|
cout << "Error in file " << __FILE__ << " at line " << __LINE__ << ". Cannot set list element" << endl;
|
||
|
exit(1);
|
||
|
}
|
||
|
PyList_SetItem(list,i, listEl);
|
||
|
}
|
||
|
delete [] vectorV;
|
||
|
return Py_BuildValue("N",list);
|
||
|
}
|
||
|
|
||
|
PyObject * getLocationDown_C(PyObject* self, PyObject* args)
|
||
|
{
|
||
|
int dim1 = 0;
|
||
|
if(!PyArg_ParseTuple(args, "i", &dim1))
|
||
|
{
|
||
|
return NULL;
|
||
|
}
|
||
|
PyObject * list = PyList_New(dim1);
|
||
|
int * vectorV = new int[dim1];
|
||
|
getLocationDown_f(vectorV, &dim1);
|
||
|
for(int i = 0; i < dim1; ++i)
|
||
|
{
|
||
|
PyObject * listEl = PyLong_FromLong((long int) vectorV[i]);
|
||
|
if(listEl == NULL)
|
||
|
{
|
||
|
cout << "Error in file " << __FILE__ << " at line " << __LINE__ << ". Cannot set list element" << endl;
|
||
|
exit(1);
|
||
|
}
|
||
|
PyList_SetItem(list,i, listEl);
|
||
|
}
|
||
|
delete [] vectorV;
|
||
|
return Py_BuildValue("N",list);
|
||
|
}
|
||
|
|
||
|
PyObject * getLocationDownOffset_C(PyObject* self, PyObject* args)
|
||
|
{
|
||
|
int dim1 = 0;
|
||
|
if(!PyArg_ParseTuple(args, "i", &dim1))
|
||
|
{
|
||
|
return NULL;
|
||
|
}
|
||
|
PyObject * list = PyList_New(dim1);
|
||
|
float * vectorV = new float[dim1];
|
||
|
getLocationDownOffset_f(vectorV, &dim1);
|
||
|
for(int i = 0; i < dim1; ++i)
|
||
|
{
|
||
|
PyObject * listEl = PyFloat_FromDouble((double) vectorV[i]);
|
||
|
if(listEl == NULL)
|
||
|
{
|
||
|
cout << "Error in file " << __FILE__ << " at line " << __LINE__ << ". Cannot set list element" << endl;
|
||
|
exit(1);
|
||
|
}
|
||
|
PyList_SetItem(list,i, listEl);
|
||
|
}
|
||
|
delete [] vectorV;
|
||
|
return Py_BuildValue("N",list);
|
||
|
}
|
||
|
|
||
|
PyObject * getSNR_C(PyObject* self, PyObject* args)
|
||
|
{
|
||
|
int dim1 = 0;
|
||
|
if(!PyArg_ParseTuple(args, "i", &dim1))
|
||
|
{
|
||
|
return NULL;
|
||
|
}
|
||
|
PyObject * list = PyList_New(dim1);
|
||
|
float * vectorV = new float[dim1];
|
||
|
getSNR_f(vectorV, &dim1);
|
||
|
for(int i = 0; i < dim1; ++i)
|
||
|
{
|
||
|
PyObject * listEl = PyFloat_FromDouble((double) vectorV[i]);
|
||
|
if(listEl == NULL)
|
||
|
{
|
||
|
cout << "Error in file " << __FILE__ << " at line " << __LINE__ << ". Cannot set list element" << endl;
|
||
|
exit(1);
|
||
|
}
|
||
|
PyList_SetItem(list,i, listEl);
|
||
|
}
|
||
|
delete [] vectorV;
|
||
|
return Py_BuildValue("N",list);
|
||
|
}
|
||
|
|
||
|
PyObject * setLineLength1_C(PyObject* self, PyObject* args)
|
||
|
{
|
||
|
int var;
|
||
|
if(!PyArg_ParseTuple(args, "i", &var))
|
||
|
{
|
||
|
return NULL;
|
||
|
}
|
||
|
setLineLength1_f(&var);
|
||
|
return Py_BuildValue("i", 0);
|
||
|
}
|
||
|
PyObject * setLineLength2_C(PyObject* self, PyObject* args)
|
||
|
{
|
||
|
int var;
|
||
|
if(!PyArg_ParseTuple(args, "i", &var))
|
||
|
{
|
||
|
return NULL;
|
||
|
}
|
||
|
setLineLength2_f(&var);
|
||
|
return Py_BuildValue("i", 0);
|
||
|
}
|
||
|
|
||
|
PyObject * setFileLength1_C(PyObject* self, PyObject* args)
|
||
|
{
|
||
|
int var;
|
||
|
if(!PyArg_ParseTuple(args, "i", &var))
|
||
|
{
|
||
|
return NULL;
|
||
|
}
|
||
|
setFileLength1_f(&var);
|
||
|
return Py_BuildValue("i", 0);
|
||
|
}
|
||
|
PyObject * setFileLength2_C(PyObject* self, PyObject* args)
|
||
|
{
|
||
|
int var;
|
||
|
if(!PyArg_ParseTuple(args, "i", &var))
|
||
|
{
|
||
|
return NULL;
|
||
|
}
|
||
|
setFileLength2_f(&var);
|
||
|
return Py_BuildValue("i", 0);
|
||
|
}
|
||
|
|
||
|
|
||
|
PyObject * setFirstSampleAcross_C(PyObject* self, PyObject* args)
|
||
|
{
|
||
|
int var;
|
||
|
if(!PyArg_ParseTuple(args, "i", &var))
|
||
|
{
|
||
|
return NULL;
|
||
|
}
|
||
|
setFirstSampleAcross_f(&var);
|
||
|
return Py_BuildValue("i", 0);
|
||
|
}
|
||
|
PyObject * setLastSampleAcross_C(PyObject* self, PyObject* args)
|
||
|
{
|
||
|
int var;
|
||
|
if(!PyArg_ParseTuple(args, "i", &var))
|
||
|
{
|
||
|
return NULL;
|
||
|
}
|
||
|
setLastSampleAcross_f(&var);
|
||
|
return Py_BuildValue("i", 0);
|
||
|
}
|
||
|
PyObject * setNumberLocationAcross_C(PyObject* self, PyObject* args)
|
||
|
{
|
||
|
int var;
|
||
|
if(!PyArg_ParseTuple(args, "i", &var))
|
||
|
{
|
||
|
return NULL;
|
||
|
}
|
||
|
setNumberLocationAcross_f(&var);
|
||
|
return Py_BuildValue("i", 0);
|
||
|
}
|
||
|
PyObject * setFirstSampleDown_C(PyObject* self, PyObject* args)
|
||
|
{
|
||
|
int var;
|
||
|
if(!PyArg_ParseTuple(args, "i", &var))
|
||
|
{
|
||
|
return NULL;
|
||
|
}
|
||
|
setFirstSampleDown_f(&var);
|
||
|
return Py_BuildValue("i", 0);
|
||
|
}
|
||
|
PyObject * setLastSampleDown_C(PyObject* self, PyObject* args)
|
||
|
{
|
||
|
int var;
|
||
|
if(!PyArg_ParseTuple(args, "i", &var))
|
||
|
{
|
||
|
return NULL;
|
||
|
}
|
||
|
setLastSampleDown_f(&var);
|
||
|
return Py_BuildValue("i", 0);
|
||
|
}
|
||
|
PyObject * setNumberLocationDown_C(PyObject* self, PyObject* args)
|
||
|
{
|
||
|
int var;
|
||
|
if(!PyArg_ParseTuple(args, "i", &var))
|
||
|
{
|
||
|
return NULL;
|
||
|
}
|
||
|
setNumberLocationDown_f(&var);
|
||
|
return Py_BuildValue("i", 0);
|
||
|
}
|
||
|
PyObject * setAcrossGrossOffset_C(PyObject* self, PyObject* args)
|
||
|
{
|
||
|
int var;
|
||
|
if(!PyArg_ParseTuple(args, "i", &var))
|
||
|
{
|
||
|
return NULL;
|
||
|
}
|
||
|
setAcrossGrossOffset_f(&var);
|
||
|
return Py_BuildValue("i", 0);
|
||
|
}
|
||
|
PyObject * setDownGrossOffset_C(PyObject* self, PyObject* args)
|
||
|
{
|
||
|
int var;
|
||
|
if(!PyArg_ParseTuple(args, "i", &var))
|
||
|
{
|
||
|
return NULL;
|
||
|
}
|
||
|
setDownGrossOffset_f(&var);
|
||
|
return Py_BuildValue("i", 0);
|
||
|
}
|
||
|
PyObject * setFirstPRF_C(PyObject* self, PyObject* args)
|
||
|
{
|
||
|
float var;
|
||
|
if(!PyArg_ParseTuple(args, "f", &var))
|
||
|
{
|
||
|
return NULL;
|
||
|
}
|
||
|
setFirstPRF_f(&var);
|
||
|
return Py_BuildValue("i", 0);
|
||
|
}
|
||
|
PyObject * setSecondPRF_C(PyObject* self, PyObject* args)
|
||
|
{
|
||
|
float var;
|
||
|
if(!PyArg_ParseTuple(args, "f", &var))
|
||
|
{
|
||
|
return NULL;
|
||
|
}
|
||
|
setSecondPRF_f(&var);
|
||
|
return Py_BuildValue("i", 0);
|
||
|
}
|
||
|
PyObject * setDebugFlag_C(PyObject* self, PyObject* args)
|
||
|
{
|
||
|
char * var;
|
||
|
int varInt;
|
||
|
if(!PyArg_ParseTuple(args, "s#", &var ,&varInt))
|
||
|
{
|
||
|
return NULL;
|
||
|
}
|
||
|
setDebugFlag_f(var,&varInt);
|
||
|
return Py_BuildValue("i", 0);
|
||
|
}
|
||
|
|
||
|
PyObject * setWindowSize_C(PyObject* self, PyObject* args)
|
||
|
{
|
||
|
int var;
|
||
|
if(!PyArg_ParseTuple(args, "i", &var))
|
||
|
{
|
||
|
return NULL;
|
||
|
}
|
||
|
setWindowSize_f(&var);
|
||
|
return Py_BuildValue("i", 0);
|
||
|
}
|
||
|
PyObject * setSearchWindowSize_C(PyObject* self, PyObject* args)
|
||
|
{
|
||
|
int var;
|
||
|
if(!PyArg_ParseTuple(args, "i", &var))
|
||
|
{
|
||
|
return NULL;
|
||
|
}
|
||
|
setSearchWindowSize_f(&var);
|
||
|
return Py_BuildValue("i", 0);
|
||
|
}
|
||
|
PyObject * setZoomWindowSize_C(PyObject* self, PyObject* args)
|
||
|
{
|
||
|
int var;
|
||
|
if(!PyArg_ParseTuple(args, "i", &var))
|
||
|
{
|
||
|
return NULL;
|
||
|
}
|
||
|
setZoomWindowSize_f(&var);
|
||
|
return Py_BuildValue("i", 0);
|
||
|
}
|
||
|
PyObject * setOversamplingFactor_C(PyObject* self, PyObject* args)
|
||
|
{
|
||
|
int var;
|
||
|
if(!PyArg_ParseTuple(args, "i", &var))
|
||
|
{
|
||
|
return NULL;
|
||
|
}
|
||
|
setOversamplingFactor_f(&var);
|
||
|
return Py_BuildValue("i", 0);
|
||
|
}
|
||
|
PyObject * setIsComplex1_C(PyObject* self, PyObject* args)
|
||
|
{
|
||
|
int var;
|
||
|
if(!PyArg_ParseTuple(args,"i",&var))
|
||
|
{
|
||
|
return NULL;
|
||
|
}
|
||
|
setIsComplex1_f(&var);
|
||
|
return Py_BuildValue("i",0);
|
||
|
}
|
||
|
PyObject * setIsComplex2_C(PyObject* self, PyObject* args)
|
||
|
{
|
||
|
int var;
|
||
|
if(!PyArg_ParseTuple(args,"i",&var))
|
||
|
{
|
||
|
return NULL;
|
||
|
}
|
||
|
setIsComplex2_f(&var);
|
||
|
return Py_BuildValue("i",0);
|
||
|
}
|
||
|
PyObject * setBand1_C(PyObject* self, PyObject* args)
|
||
|
{
|
||
|
int var;
|
||
|
if(!PyArg_ParseTuple(args,"i",&var))
|
||
|
{
|
||
|
return NULL;
|
||
|
}
|
||
|
var = var+1;
|
||
|
setBand1_f(&var);
|
||
|
return Py_BuildValue("i",0);
|
||
|
}
|
||
|
PyObject * setBand2_C(PyObject* self, PyObject* args)
|
||
|
{
|
||
|
int var;
|
||
|
if(!PyArg_ParseTuple(args,"i",&var))
|
||
|
{
|
||
|
return NULL;
|
||
|
}
|
||
|
var = var+1;
|
||
|
setBand2_f(&var);
|
||
|
return Py_BuildValue("i",0);
|
||
|
}
|