You cannot select more than 25 topics
Topics must start with a letter or number, can include dashes ('-') and can be up to 35 characters long.
453 lines
18 KiB
C
453 lines
18 KiB
C
/* ----------------------------------------------------------------------
|
|
* Project: CMSIS DSP Python Wrapper
|
|
* Title: cmsismodule.h
|
|
* Description: C code for the CMSIS-DSP Python wrapper
|
|
*
|
|
* $Date: 27 April 2021
|
|
* $Revision: V1.0
|
|
*
|
|
* Target Processor: Cortex-M cores
|
|
* -------------------------------------------------------------------- */
|
|
/*
|
|
* Copyright (C) 2010-2021 ARM Limited or its affiliates. All rights reserved.
|
|
*
|
|
* SPDX-License-Identifier: Apache-2.0
|
|
*
|
|
* 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
|
|
*
|
|
* 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.
|
|
*/
|
|
|
|
#define MODNAME "cmsisdsp_distance"
|
|
#define MODINITNAME cmsisdsp_distance
|
|
|
|
#include "cmsisdsp_module.h"
|
|
MATRIXFROMNUMPY(f32,float32_t,double,NPY_DOUBLE);
|
|
CREATEMATRIX(f32,float32_t);
|
|
NUMPYARRAYFROMMATRIX(f32,NPY_FLOAT);
|
|
|
|
MATRIXFROMNUMPY(q7,q7_t,int8_t,NPY_BYTE);
|
|
CREATEMATRIX(q7,q7_t);
|
|
NUMPYARRAYFROMMATRIX(q7,NPY_BYTE);
|
|
|
|
|
|
NUMPYVECTORFROMBUFFER(f32,float32_t,NPY_FLOAT);
|
|
|
|
|
|
void typeRegistration(PyObject *module) {
|
|
|
|
|
|
}
|
|
|
|
#define FLOATDIST(NAME) \
|
|
static PyObject * \
|
|
cmsis_arm_##NAME##_f32(PyObject *obj, PyObject *args) \
|
|
{ \
|
|
\
|
|
PyObject *pSrcA=NULL; \
|
|
float32_t *pSrcA_converted=NULL; \
|
|
PyObject *pSrcB=NULL; \
|
|
float32_t *pSrcB_converted=NULL; \
|
|
uint32_t blockSize; \
|
|
float32_t result; \
|
|
\
|
|
if (PyArg_ParseTuple(args,"OO",&pSrcA,&pSrcB)) \
|
|
{ \
|
|
\
|
|
GETARGUMENT(pSrcA,NPY_DOUBLE,double,float32_t); \
|
|
GETARGUMENT(pSrcB,NPY_DOUBLE,double,float32_t); \
|
|
blockSize = arraySizepSrcA ; \
|
|
\
|
|
\
|
|
\
|
|
result=arm_##NAME##_f32(pSrcA_converted,pSrcB_converted,blockSize);\
|
|
PyObject* resultOBJ=Py_BuildValue("f",result); \
|
|
\
|
|
PyObject *pythonResult = Py_BuildValue("O",resultOBJ); \
|
|
\
|
|
FREEARGUMENT(pSrcA_converted); \
|
|
FREEARGUMENT(pSrcB_converted); \
|
|
Py_DECREF(resultOBJ); \
|
|
return(pythonResult); \
|
|
\
|
|
} \
|
|
return(NULL); \
|
|
}
|
|
|
|
#define FLOAT64DIST(NAME) \
|
|
static PyObject * \
|
|
cmsis_arm_##NAME##_f64(PyObject *obj, PyObject *args) \
|
|
{ \
|
|
\
|
|
PyObject *pSrcA=NULL; \
|
|
float64_t *pSrcA_converted=NULL; \
|
|
PyObject *pSrcB=NULL; \
|
|
float64_t *pSrcB_converted=NULL; \
|
|
uint32_t blockSize; \
|
|
float64_t result; \
|
|
\
|
|
if (PyArg_ParseTuple(args,"OO",&pSrcA,&pSrcB)) \
|
|
{ \
|
|
\
|
|
GETARGUMENT(pSrcA,NPY_DOUBLE,double,float64_t); \
|
|
GETARGUMENT(pSrcB,NPY_DOUBLE,double,float64_t); \
|
|
blockSize = arraySizepSrcA ; \
|
|
\
|
|
\
|
|
\
|
|
result=arm_##NAME##_f64(pSrcA_converted,pSrcB_converted,blockSize);\
|
|
PyObject* resultOBJ=Py_BuildValue("d",result); \
|
|
\
|
|
PyObject *pythonResult = Py_BuildValue("O",resultOBJ); \
|
|
\
|
|
FREEARGUMENT(pSrcA_converted); \
|
|
FREEARGUMENT(pSrcB_converted); \
|
|
Py_DECREF(resultOBJ); \
|
|
return(pythonResult); \
|
|
\
|
|
} \
|
|
return(NULL); \
|
|
}
|
|
|
|
FLOAT64DIST(chebyshev_distance);
|
|
FLOAT64DIST(cityblock_distance);
|
|
FLOAT64DIST(cosine_distance);
|
|
FLOAT64DIST(euclidean_distance);
|
|
|
|
|
|
FLOATDIST(braycurtis_distance);
|
|
FLOATDIST(canberra_distance);
|
|
FLOATDIST(chebyshev_distance);
|
|
FLOATDIST(cityblock_distance);
|
|
FLOATDIST(correlation_distance);
|
|
FLOATDIST(cosine_distance);
|
|
FLOATDIST(euclidean_distance);
|
|
FLOATDIST(jensenshannon_distance);
|
|
|
|
static PyObject *
|
|
cmsis_arm_minkowski_distance_f32(PyObject *obj, PyObject *args)
|
|
{
|
|
|
|
PyObject *pSrcA=NULL;
|
|
float32_t *pSrcA_converted=NULL;
|
|
PyObject *pSrcB=NULL;
|
|
float32_t *pSrcB_converted=NULL;
|
|
int32_t w;
|
|
uint32_t blockSize;
|
|
float32_t result;
|
|
|
|
if (PyArg_ParseTuple(args,"OOi",&pSrcA,&pSrcB,&w))
|
|
{
|
|
|
|
GETARGUMENT(pSrcA,NPY_DOUBLE,double,float32_t);
|
|
GETARGUMENT(pSrcB,NPY_DOUBLE,double,float32_t);
|
|
|
|
blockSize = arraySizepSrcA ;
|
|
|
|
|
|
|
|
result=arm_minkowski_distance_f32(pSrcA_converted,pSrcB_converted,w,blockSize);
|
|
PyObject* resultOBJ=Py_BuildValue("f",result);
|
|
|
|
PyObject *pythonResult = Py_BuildValue("O",resultOBJ);
|
|
|
|
FREEARGUMENT(pSrcA_converted);
|
|
FREEARGUMENT(pSrcB_converted);
|
|
|
|
Py_DECREF(resultOBJ);
|
|
return(pythonResult);
|
|
|
|
}
|
|
return(NULL);
|
|
}
|
|
|
|
|
|
#define INTDIST(NAME) \
|
|
static PyObject * \
|
|
cmsis_arm_##NAME (PyObject *obj, PyObject *args) \
|
|
{ \
|
|
\
|
|
PyObject *pSrcA=NULL; \
|
|
uint32_t *pSrcA_converted=NULL; \
|
|
PyObject *pSrcB=NULL; \
|
|
uint32_t *pSrcB_converted=NULL; \
|
|
uint32_t blockSize; \
|
|
float32_t result; \
|
|
\
|
|
\
|
|
if (PyArg_ParseTuple(args,"OOi",&pSrcA,&pSrcB,&blockSize)) \
|
|
{ \
|
|
\
|
|
GETARGUMENT(pSrcA,NPY_UINT32,uint32_t,uint32_t); \
|
|
GETARGUMENT(pSrcB,NPY_UINT32,uint32_t,uint32_t); \
|
|
\
|
|
\
|
|
\
|
|
result=arm_##NAME (pSrcA_converted,pSrcB_converted,blockSize); \
|
|
PyObject* resultOBJ=Py_BuildValue("f",result); \
|
|
\
|
|
PyObject *pythonResult = Py_BuildValue("O",resultOBJ); \
|
|
\
|
|
FREEARGUMENT(pSrcA_converted); \
|
|
FREEARGUMENT(pSrcB_converted); \
|
|
Py_DECREF(resultOBJ); \
|
|
return(pythonResult); \
|
|
\
|
|
} \
|
|
return(NULL); \
|
|
}
|
|
|
|
|
|
INTDIST(dice_distance);
|
|
INTDIST(hamming_distance);
|
|
INTDIST(jaccard_distance);
|
|
INTDIST(kulsinski_distance);
|
|
INTDIST(rogerstanimoto_distance);
|
|
INTDIST(russellrao_distance);
|
|
INTDIST(sokalmichener_distance);
|
|
INTDIST(sokalsneath_distance);
|
|
INTDIST(yule_distance);
|
|
|
|
static PyObject *
|
|
cmsis_arm_dtw_init_window_q7(PyObject *obj,
|
|
PyObject *args)
|
|
{
|
|
|
|
PyObject *pSrc=NULL; // input
|
|
int32_t winType;
|
|
int32_t winSize;
|
|
arm_matrix_instance_q7 pSrc_converted; // input
|
|
|
|
|
|
if (PyArg_ParseTuple(args,"iiO",&winType,&winSize,&pSrc));
|
|
{
|
|
|
|
q7MatrixFromNumpy(&pSrc_converted,pSrc);
|
|
uint32_t row = pSrc_converted.numCols ;
|
|
uint32_t column = pSrc_converted.numRows ;
|
|
|
|
arm_status returnValue =
|
|
arm_dtw_init_window_q7(winType,
|
|
winSize,
|
|
&pSrc_converted
|
|
);
|
|
PyObject* theReturnOBJ=Py_BuildValue("i",returnValue);
|
|
PyObject* dstOBJ=NumpyArrayFromq7Matrix(&pSrc_converted);
|
|
|
|
PyObject *pythonResult = Py_BuildValue("OO",theReturnOBJ,dstOBJ);
|
|
|
|
Py_DECREF(theReturnOBJ);
|
|
Py_DECREF(dstOBJ);
|
|
return(pythonResult);
|
|
|
|
}
|
|
Py_RETURN_NONE;
|
|
}
|
|
|
|
static PyObject *
|
|
cmsis_arm_dtw_distance_f32(PyObject *obj,
|
|
PyObject *args)
|
|
{
|
|
|
|
PyObject *pDist=NULL; // input
|
|
arm_matrix_instance_f32 pDist_converted; // input
|
|
|
|
PyObject *pWin=NULL; // input
|
|
arm_matrix_instance_q7 pWin_converted; // input
|
|
arm_matrix_instance_q7 *pWinMatrix;
|
|
|
|
arm_matrix_instance_f32 dtw_converted;
|
|
|
|
|
|
if (PyArg_ParseTuple(args,"OO",&pDist,&pWin));
|
|
{
|
|
|
|
f32MatrixFromNumpy(&pDist_converted,pDist);
|
|
if (pWin != Py_None)
|
|
{
|
|
q7MatrixFromNumpy(&pWin_converted,pWin);
|
|
pWinMatrix = &pWin_converted;
|
|
}
|
|
else
|
|
{
|
|
pWinMatrix = NULL;
|
|
}
|
|
|
|
uint32_t column = pDist_converted.numCols ;
|
|
uint32_t row = pDist_converted.numRows ;
|
|
createf32Matrix(&dtw_converted,row,column);
|
|
float32_t distance;
|
|
|
|
arm_status returnValue =
|
|
arm_dtw_distance_f32(&pDist_converted,
|
|
pWinMatrix,
|
|
&dtw_converted,
|
|
&distance
|
|
);
|
|
|
|
|
|
PyObject* theReturnOBJ=Py_BuildValue("i",returnValue);
|
|
PyObject* distOBJ=Py_BuildValue("f",distance);
|
|
|
|
PyObject* dstOBJ=NumpyArrayFromf32Matrix(&dtw_converted);
|
|
|
|
|
|
PyObject *pythonResult = Py_BuildValue("OOO",theReturnOBJ,distOBJ,dstOBJ);
|
|
|
|
Py_DECREF(theReturnOBJ);
|
|
Py_DECREF(distOBJ);
|
|
|
|
FREEMATRIX(&pDist_converted);
|
|
if (pWinMatrix)
|
|
{
|
|
FREEMATRIX(pWinMatrix);
|
|
}
|
|
Py_DECREF(dstOBJ);
|
|
|
|
return(pythonResult);
|
|
|
|
}
|
|
Py_RETURN_NONE;
|
|
}
|
|
|
|
static PyObject *
|
|
cmsis_arm_dtw_path_f32(PyObject *obj,
|
|
PyObject *args)
|
|
{
|
|
PyObject *pCost=NULL; // input
|
|
arm_matrix_instance_f32 pCost_converted; // input
|
|
int16_t *pDst=NULL; // output
|
|
|
|
if (PyArg_ParseTuple(args,"O",&pCost))
|
|
{
|
|
f32MatrixFromNumpy(&pCost_converted,pCost);
|
|
|
|
uint32_t pathLength;
|
|
int32_t blockSize;
|
|
blockSize=2*(pCost_converted.numRows+pCost_converted.numCols);
|
|
pDst=PyMem_Malloc(sizeof(int16_t)*blockSize);
|
|
|
|
|
|
arm_dtw_path_f32(&pCost_converted,
|
|
pDst,
|
|
&pathLength);
|
|
|
|
INT16ARRAY1(pDstOBJ,2*pathLength,pDst);
|
|
|
|
PyObject *pythonResult = Py_BuildValue("O",pDstOBJ);
|
|
|
|
FREEMATRIX(&pCost_converted);
|
|
Py_DECREF(pDstOBJ);
|
|
return(pythonResult);
|
|
}
|
|
|
|
}
|
|
|
|
static PyMethodDef CMSISDSPMethods[] = {
|
|
|
|
{"arm_braycurtis_distance_f32", cmsis_arm_braycurtis_distance_f32, METH_VARARGS,""},
|
|
{"arm_canberra_distance_f32" , cmsis_arm_canberra_distance_f32, METH_VARARGS,""},
|
|
{"arm_chebyshev_distance_f32" , cmsis_arm_chebyshev_distance_f32, METH_VARARGS,""},
|
|
{"arm_chebyshev_distance_f64" , cmsis_arm_chebyshev_distance_f64, METH_VARARGS,""},
|
|
|
|
{"arm_cityblock_distance_f32", cmsis_arm_cityblock_distance_f32, METH_VARARGS,""},
|
|
{"arm_cityblock_distance_f64", cmsis_arm_cityblock_distance_f64, METH_VARARGS,""},
|
|
|
|
{"arm_correlation_distance_f32", cmsis_arm_correlation_distance_f32, METH_VARARGS,""},
|
|
|
|
{"arm_cosine_distance_f32", cmsis_arm_cosine_distance_f32, METH_VARARGS,""},
|
|
{"arm_cosine_distance_f64", cmsis_arm_cosine_distance_f64, METH_VARARGS,""},
|
|
|
|
{"arm_euclidean_distance_f32", cmsis_arm_euclidean_distance_f32, METH_VARARGS,""},
|
|
{"arm_euclidean_distance_f64", cmsis_arm_euclidean_distance_f64, METH_VARARGS,""},
|
|
|
|
{"arm_jensenshannon_distance_f32", cmsis_arm_jensenshannon_distance_f32, METH_VARARGS,""},
|
|
{"arm_minkowski_distance_f32", cmsis_arm_minkowski_distance_f32, METH_VARARGS,""},
|
|
|
|
{"arm_dice_distance",cmsis_arm_dice_distance, METH_VARARGS,""},
|
|
{"arm_hamming_distance",cmsis_arm_hamming_distance, METH_VARARGS,""},
|
|
{"arm_jaccard_distance",cmsis_arm_jaccard_distance, METH_VARARGS,""},
|
|
{"arm_kulsinski_distance",cmsis_arm_kulsinski_distance, METH_VARARGS,""},
|
|
{"arm_rogerstanimoto_distance",cmsis_arm_rogerstanimoto_distance, METH_VARARGS,""},
|
|
{"arm_russellrao_distance",cmsis_arm_russellrao_distance, METH_VARARGS,""},
|
|
{"arm_sokalmichener_distance",cmsis_arm_sokalmichener_distance, METH_VARARGS,""},
|
|
{"arm_sokalsneath_distance",cmsis_arm_sokalsneath_distance, METH_VARARGS,""},
|
|
{"arm_yule_distance",cmsis_arm_yule_distance, METH_VARARGS,""},
|
|
|
|
{"arm_dtw_init_window_q7", cmsis_arm_dtw_init_window_q7, METH_VARARGS,""},
|
|
{"arm_dtw_distance_f32", cmsis_arm_dtw_distance_f32, METH_VARARGS,""},
|
|
{"arm_dtw_path_f32", cmsis_arm_dtw_path_f32, METH_VARARGS,""},
|
|
|
|
{"error_out", (PyCFunction)error_out, METH_NOARGS, NULL},
|
|
{NULL, NULL, 0, NULL} /* Sentinel */
|
|
};
|
|
|
|
#ifdef IS_PY3K
|
|
static int cmsisdsp_traverse(PyObject *m, visitproc visit, void *arg) {
|
|
Py_VISIT(GETSTATE(m)->error);
|
|
return 0;
|
|
}
|
|
|
|
static int cmsisdsp_clear(PyObject *m) {
|
|
Py_CLEAR(GETSTATE(m)->error);
|
|
return 0;
|
|
}
|
|
|
|
|
|
static struct PyModuleDef moduledef = {
|
|
PyModuleDef_HEAD_INIT,
|
|
MODNAME,
|
|
NULL,
|
|
sizeof(struct module_state),
|
|
CMSISDSPMethods,
|
|
NULL,
|
|
cmsisdsp_traverse,
|
|
cmsisdsp_clear,
|
|
NULL
|
|
};
|
|
|
|
#define INITERROR return NULL
|
|
|
|
PyMODINIT_FUNC
|
|
CAT(PyInit_,MODINITNAME)(void)
|
|
|
|
|
|
#else
|
|
#define INITERROR return
|
|
|
|
void CAT(init,MODINITNAME)(void)
|
|
#endif
|
|
{
|
|
import_array();
|
|
|
|
#ifdef IS_PY3K
|
|
PyObject *module = PyModule_Create(&moduledef);
|
|
#else
|
|
PyObject *module = Py_InitModule(MODNAME, CMSISDSPMethods);
|
|
#endif
|
|
|
|
if (module == NULL)
|
|
INITERROR;
|
|
struct module_state *st = GETSTATE(module);
|
|
|
|
st->error = PyErr_NewException(MODNAME".Error", NULL, NULL);
|
|
if (st->error == NULL) {
|
|
Py_DECREF(module);
|
|
INITERROR;
|
|
}
|
|
|
|
|
|
typeRegistration(module);
|
|
|
|
#ifdef IS_PY3K
|
|
return module;
|
|
#endif
|
|
} |