arm_cmplx_mag_f64.c 2.0 KB

1234567891011121314151617181920212223242526272829303132333435363738394041424344454647484950515253545556575859606162636465666768697071727374757677
  1. /* ----------------------------------------------------------------------
  2. * Project: CMSIS DSP Library
  3. * Title: arm_cmplx_mag_f64.c
  4. * Description: Floating-point complex magnitude
  5. *
  6. * $Date: 13 September 2021
  7. * $Revision: V1.10.0
  8. *
  9. * Target Processor: Cortex-M and Cortex-A cores
  10. * -------------------------------------------------------------------- */
  11. /*
  12. * Copyright (C) 2010-2021 ARM Limited or its affiliates. All rights reserved.
  13. *
  14. * SPDX-License-Identifier: Apache-2.0
  15. *
  16. * Licensed under the Apache License, Version 2.0 (the License); you may
  17. * not use this file except in compliance with the License.
  18. * You may obtain a copy of the License at
  19. *
  20. * www.apache.org/licenses/LICENSE-2.0
  21. *
  22. * Unless required by applicable law or agreed to in writing, software
  23. * distributed under the License is distributed on an AS IS BASIS, WITHOUT
  24. * WARRANTIES OR CONDITIONS OF ANY KIND, either express or implied.
  25. * See the License for the specific language governing permissions and
  26. * limitations under the License.
  27. */
  28. #include "dsp/complex_math_functions.h"
  29. /**
  30. @ingroup groupCmplxMath
  31. */
  32. /**
  33. @addtogroup cmplx_mag
  34. @{
  35. */
  36. /**
  37. @brief Floating-point complex magnitude.
  38. @param[in] pSrc points to input vector
  39. @param[out] pDst points to output vector
  40. @param[in] numSamples number of samples in each vector
  41. */
  42. void arm_cmplx_mag_f64(
  43. const float64_t * pSrc,
  44. float64_t * pDst,
  45. uint32_t numSamples)
  46. {
  47. uint32_t blkCnt; /* loop counter */
  48. float64_t real, imag; /* Temporary variables to hold input values */
  49. /* Initialize blkCnt with number of samples */
  50. blkCnt = numSamples;
  51. while (blkCnt > 0U)
  52. {
  53. /* C[0] = sqrt(A[0] * A[0] + A[1] * A[1]) */
  54. real = *pSrc++;
  55. imag = *pSrc++;
  56. /* store result in destination buffer. */
  57. *pDst++ = sqrt((real * real) + (imag * imag));
  58. /* Decrement loop counter */
  59. blkCnt--;
  60. }
  61. }
  62. /**
  63. @} end of cmplx_mag group
  64. */