diff options
Diffstat (limited to 'src/codec2_math_arm.c')
| -rw-r--r-- | src/codec2_math_arm.c | 57 |
1 files changed, 57 insertions, 0 deletions
diff --git a/src/codec2_math_arm.c b/src/codec2_math_arm.c new file mode 100644 index 0000000..b14ad3a --- /dev/null +++ b/src/codec2_math_arm.c @@ -0,0 +1,57 @@ +//========================================================================== +// Name: codec2_math_arm.c +// +// Purpose: A wrapper around architecture specific math libraries +// used on ARM embedded devices to improve Codec2 performance. +// Created: May 15, 2022 +// Authors: Mooneer Salem +// +// License: +// +// This program is free software; you can redistribute it and/or modify +// it under the terms of the GNU Lesser General Public License version 2.1, +// as published by the Free Software Foundation. This program is +// distributed in the hope that it will be useful, but WITHOUT ANY +// WARRANTY; without even the implied warranty of MERCHANTABILITY or +// FITNESS FOR A PARTICULAR PURPOSE. See the GNU General Public +// License for more details. +// +// You should have received a copy of the GNU Lesser General Public License +// along with this program; if not, see <http://www.gnu.org/licenses/>. +// +//========================================================================== + +#if defined(__EMBEDDED__) && defined(__ARM_ARCH) +#include "codec2_math.h" +#include "arm_math.h" + +//========================================================================== +/// Calculates the dot product of two real-valued float vectors. +/// +/// @param leftHandSideRealVector A pointer to the first vector to use for the dot product. +/// @param rightHandSideRealVector A pointer to the second vector to use for the dot product. +/// @param vectorLength The length of the vector. Both vectors should be at least this long. +/// @param resultReal A pointer to the variable in which to store the scalar result. +//========================================================================== +void codec2_dot_product_f32(float* leftHandSideRealVector, float* rightHandSideRealVector, size_t vectorLength, float* resultReal) +{ + arm_dot_prod_f32(leftHandSideRealVector, rightHandSideRealVector, vectorLength, resultReal); +} + +//========================================================================== +/// Calculates the dot product of two complex-valued float vectors. +/// +/// @param leftHandSideComplexVector A pointer to the first vector to use for the dot product. +/// @param rightHandSideComplexVector A pointer to the second vector to use for the dot product. +/// @param vectorLength The length of the vector. Both vectors should be at least this long. +/// @param resultReal A pointer to the variable in which to store the real component of the result. +/// @param resultImag A pointer to the variable in which to store the imaginary component of the result. +/// +/// @note Each array of floats is organized with even elements being real and odd elements imaginary. +//========================================================================== +void codec2_complex_dot_product_f32(COMP* leftHandSideComplexVector, COMP* rightHandSideComplexVector, size_t vectorLength, float* resultReal, float* resultImag) +{ + arm_cmplx_dot_prod_f32((float*)leftHandSideComplexVector, (float*)rightHandSideComplexVector, vectorLength, resultReal, resultImag); +} + +#endif // defined(__EMBEDDED__) && defined(__ARM_ARCH) |