1 /****************************************************************************** 2 * @file complex_math_functions_f16.h 3 * @brief Public header file for CMSIS DSP Library 4 * @version V1.9.0 5 * @date 23 April 2021 6 * Target Processor: Cortex-M and Cortex-A cores 7 ******************************************************************************/ 8 /* 9 * Copyright (c) 2010-2020 Arm Limited or its affiliates. All rights reserved. 10 * 11 * SPDX-License-Identifier: Apache-2.0 12 * 13 * Licensed under the Apache License, Version 2.0 (the License); you may 14 * not use this file except in compliance with the License. 15 * You may obtain a copy of the License at 16 * 17 * www.apache.org/licenses/LICENSE-2.0 18 * 19 * Unless required by applicable law or agreed to in writing, software 20 * distributed under the License is distributed on an AS IS BASIS, WITHOUT 21 * WARRANTIES OR CONDITIONS OF ANY KIND, either express or implied. 22 * See the License for the specific language governing permissions and 23 * limitations under the License. 24 */ 25 26 27 #ifndef _COMPLEX_MATH_FUNCTIONS_F16_H_ 28 #define _COMPLEX_MATH_FUNCTIONS_F16_H_ 29 30 #include "arm_math_types_f16.h" 31 #include "arm_math_memory.h" 32 33 #include "dsp/none.h" 34 #include "dsp/utils.h" 35 #include "dsp/fast_math_functions_f16.h" 36 37 #ifdef __cplusplus 38 extern "C" 39 { 40 #endif 41 42 #if defined(ARM_FLOAT16_SUPPORTED) 43 44 /** 45 * @brief Floating-point complex conjugate. 46 * @param[in] pSrc points to the input vector 47 * @param[out] pDst points to the output vector 48 * @param[in] numSamples number of complex samples in each vector 49 */ 50 void arm_cmplx_conj_f16( 51 const float16_t * pSrc, 52 float16_t * pDst, 53 uint32_t numSamples); 54 55 /** 56 * @brief Floating-point complex magnitude squared 57 * @param[in] pSrc points to the complex input vector 58 * @param[out] pDst points to the real output vector 59 * @param[in] numSamples number of complex samples in the input vector 60 */ 61 void arm_cmplx_mag_squared_f16( 62 const float16_t * pSrc, 63 float16_t * pDst, 64 uint32_t numSamples); 65 66 /** 67 * @brief Floating-point complex magnitude 68 * @param[in] pSrc points to the complex input vector 69 * @param[out] pDst points to the real output vector 70 * @param[in] numSamples number of complex samples in the input vector 71 */ 72 void arm_cmplx_mag_f16( 73 const float16_t * pSrc, 74 float16_t * pDst, 75 uint32_t numSamples); 76 77 /** 78 * @brief Floating-point complex dot product 79 * @param[in] pSrcA points to the first input vector 80 * @param[in] pSrcB points to the second input vector 81 * @param[in] numSamples number of complex samples in each vector 82 * @param[out] realResult real part of the result returned here 83 * @param[out] imagResult imaginary part of the result returned here 84 */ 85 void arm_cmplx_dot_prod_f16( 86 const float16_t * pSrcA, 87 const float16_t * pSrcB, 88 uint32_t numSamples, 89 float16_t * realResult, 90 float16_t * imagResult); 91 92 /** 93 * @brief Floating-point complex-by-real multiplication 94 * @param[in] pSrcCmplx points to the complex input vector 95 * @param[in] pSrcReal points to the real input vector 96 * @param[out] pCmplxDst points to the complex output vector 97 * @param[in] numSamples number of samples in each vector 98 */ 99 void arm_cmplx_mult_real_f16( 100 const float16_t * pSrcCmplx, 101 const float16_t * pSrcReal, 102 float16_t * pCmplxDst, 103 uint32_t numSamples); 104 105 /** 106 * @brief Floating-point complex-by-complex multiplication 107 * @param[in] pSrcA points to the first input vector 108 * @param[in] pSrcB points to the second input vector 109 * @param[out] pDst points to the output vector 110 * @param[in] numSamples number of complex samples in each vector 111 */ 112 void arm_cmplx_mult_cmplx_f16( 113 const float16_t * pSrcA, 114 const float16_t * pSrcB, 115 float16_t * pDst, 116 uint32_t numSamples); 117 118 #endif /*defined(ARM_FLOAT16_SUPPORTED)*/ 119 #ifdef __cplusplus 120 } 121 #endif 122 123 #endif /* ifndef _COMPLEX_MATH_FUNCTIONS_F16_H_ */ 124