CMSIS DSP Library from CMSIS 2.0. See http://www.onarm.com/cmsis/ for full details

Dependents:   K22F_DSP_Matrix_least_square BNO055-ELEC3810 1BNO055 ECE4180Project--Slave2 ... more

Committer:
simon
Date:
Thu Mar 10 15:07:50 2011 +0000
Revision:
0:1014af42efd9

        

Who changed what in which revision?

UserRevisionLine numberNew contents of line
simon 0:1014af42efd9 1 /* ----------------------------------------------------------------------
simon 0:1014af42efd9 2 * Copyright (C) 2010 ARM Limited. All rights reserved.
simon 0:1014af42efd9 3 *
simon 0:1014af42efd9 4 * $Date: 29. November 2010
simon 0:1014af42efd9 5 * $Revision: V1.0.3
simon 0:1014af42efd9 6 *
simon 0:1014af42efd9 7 * Project: CMSIS DSP Library
simon 0:1014af42efd9 8 * Title: arm_cmplx_mag_q15.c
simon 0:1014af42efd9 9 *
simon 0:1014af42efd9 10 * Description: Q15 complex magnitude.
simon 0:1014af42efd9 11 *
simon 0:1014af42efd9 12 * Target Processor: Cortex-M4/Cortex-M3
simon 0:1014af42efd9 13 *
simon 0:1014af42efd9 14 * Version 1.0.3 2010/11/29
simon 0:1014af42efd9 15 * Re-organized the CMSIS folders and updated documentation.
simon 0:1014af42efd9 16 *
simon 0:1014af42efd9 17 * Version 1.0.2 2010/11/11
simon 0:1014af42efd9 18 * Documentation updated.
simon 0:1014af42efd9 19 *
simon 0:1014af42efd9 20 * Version 1.0.1 2010/10/05
simon 0:1014af42efd9 21 * Production release and review comments incorporated.
simon 0:1014af42efd9 22 *
simon 0:1014af42efd9 23 * Version 1.0.0 2010/09/20
simon 0:1014af42efd9 24 * Production release and review comments incorporated.
simon 0:1014af42efd9 25 * ---------------------------------------------------------------------------- */
simon 0:1014af42efd9 26
simon 0:1014af42efd9 27 #include "arm_math.h"
simon 0:1014af42efd9 28
simon 0:1014af42efd9 29 /**
simon 0:1014af42efd9 30 * @ingroup groupCmplxMath
simon 0:1014af42efd9 31 */
simon 0:1014af42efd9 32
simon 0:1014af42efd9 33 /**
simon 0:1014af42efd9 34 * @addtogroup cmplx_mag
simon 0:1014af42efd9 35 * @{
simon 0:1014af42efd9 36 */
simon 0:1014af42efd9 37
simon 0:1014af42efd9 38
simon 0:1014af42efd9 39 /**
simon 0:1014af42efd9 40 * @brief Q15 complex magnitude
simon 0:1014af42efd9 41 * @param *pSrc points to the complex input vector
simon 0:1014af42efd9 42 * @param *pDst points to the real output vector
simon 0:1014af42efd9 43 * @param numSamples number of complex samples in the input vector
simon 0:1014af42efd9 44 * @return none.
simon 0:1014af42efd9 45 *
simon 0:1014af42efd9 46 * <b>Scaling and Overflow Behavior:</b>
simon 0:1014af42efd9 47 * \par
simon 0:1014af42efd9 48 * The function implements 1.15 by 1.15 multiplications and finally output is converted into 2.14 format.
simon 0:1014af42efd9 49 */
simon 0:1014af42efd9 50
simon 0:1014af42efd9 51 void arm_cmplx_mag_q15(
simon 0:1014af42efd9 52 q15_t * pSrc,
simon 0:1014af42efd9 53 q15_t * pDst,
simon 0:1014af42efd9 54 uint32_t numSamples)
simon 0:1014af42efd9 55 {
simon 0:1014af42efd9 56 q15_t real, imag; /* Temporary variables to hold input values */
simon 0:1014af42efd9 57 q31_t acc0, acc1; /* Accumulators */
simon 0:1014af42efd9 58 uint32_t blkCnt; /* loop counter */
simon 0:1014af42efd9 59
simon 0:1014af42efd9 60
simon 0:1014af42efd9 61 /*loop Unrolling */
simon 0:1014af42efd9 62 blkCnt = numSamples >> 2u;
simon 0:1014af42efd9 63
simon 0:1014af42efd9 64 /* First part of the processing with loop unrolling. Compute 4 outputs at a time.
simon 0:1014af42efd9 65 ** a second loop below computes the remaining 1 to 3 samples. */
simon 0:1014af42efd9 66 while(blkCnt > 0u)
simon 0:1014af42efd9 67 {
simon 0:1014af42efd9 68
simon 0:1014af42efd9 69 /* C[0] = sqrt(A[0] * A[0] + A[1] * A[1]) */
simon 0:1014af42efd9 70 real = *pSrc++;
simon 0:1014af42efd9 71 imag = *pSrc++;
simon 0:1014af42efd9 72 acc0 = __SMUAD(real, real);
simon 0:1014af42efd9 73 acc1 = __SMUAD(imag, imag);
simon 0:1014af42efd9 74 /* store the result in 2.14 format in the destination buffer. */
simon 0:1014af42efd9 75 arm_sqrt_q15((q15_t) (((q63_t) acc0 + acc1) >> 17), pDst++);
simon 0:1014af42efd9 76
simon 0:1014af42efd9 77 real = *pSrc++;
simon 0:1014af42efd9 78 imag = *pSrc++;
simon 0:1014af42efd9 79 acc0 = __SMUAD(real, real);
simon 0:1014af42efd9 80 acc1 = __SMUAD(imag, imag);
simon 0:1014af42efd9 81 /* store the result in 2.14 format in the destination buffer. */
simon 0:1014af42efd9 82 arm_sqrt_q15((q15_t) (((q63_t) acc0 + acc1) >> 17), pDst++);
simon 0:1014af42efd9 83
simon 0:1014af42efd9 84 real = *pSrc++;
simon 0:1014af42efd9 85 imag = *pSrc++;
simon 0:1014af42efd9 86 acc0 = __SMUAD(real, real);
simon 0:1014af42efd9 87 acc1 = __SMUAD(imag, imag);
simon 0:1014af42efd9 88 /* store the result in 2.14 format in the destination buffer. */
simon 0:1014af42efd9 89 arm_sqrt_q15((q15_t) (((q63_t) acc0 + acc1) >> 17), pDst++);
simon 0:1014af42efd9 90
simon 0:1014af42efd9 91 real = *pSrc++;
simon 0:1014af42efd9 92 imag = *pSrc++;
simon 0:1014af42efd9 93 acc0 = __SMUAD(real, real);
simon 0:1014af42efd9 94 acc1 = __SMUAD(imag, imag);
simon 0:1014af42efd9 95 /* store the result in 2.14 format in the destination buffer. */
simon 0:1014af42efd9 96 arm_sqrt_q15((q15_t) (((q63_t) acc0 + acc1) >> 17), pDst++);
simon 0:1014af42efd9 97
simon 0:1014af42efd9 98 /* Decrement the loop counter */
simon 0:1014af42efd9 99 blkCnt--;
simon 0:1014af42efd9 100 }
simon 0:1014af42efd9 101
simon 0:1014af42efd9 102 /* If the numSamples is not a multiple of 4, compute any remaining output samples here.
simon 0:1014af42efd9 103 ** No loop unrolling is used. */
simon 0:1014af42efd9 104 blkCnt = numSamples % 0x4u;
simon 0:1014af42efd9 105
simon 0:1014af42efd9 106 while(blkCnt > 0u)
simon 0:1014af42efd9 107 {
simon 0:1014af42efd9 108 /* C[0] = sqrt(A[0] * A[0] + A[1] * A[1]) */
simon 0:1014af42efd9 109 real = *pSrc++;
simon 0:1014af42efd9 110 imag = *pSrc++;
simon 0:1014af42efd9 111 acc0 = __SMUAD(real, real);
simon 0:1014af42efd9 112 acc1 = __SMUAD(imag, imag);
simon 0:1014af42efd9 113 /* store the result in 2.14 format in the destination buffer. */
simon 0:1014af42efd9 114 arm_sqrt_q15((q15_t) (((q63_t) acc0 + acc1) >> 17), pDst++);
simon 0:1014af42efd9 115
simon 0:1014af42efd9 116 /* Decrement the loop counter */
simon 0:1014af42efd9 117 blkCnt--;
simon 0:1014af42efd9 118 }
simon 0:1014af42efd9 119 }
simon 0:1014af42efd9 120
simon 0:1014af42efd9 121 /**
simon 0:1014af42efd9 122 * @} end of cmplx_mag group
simon 0:1014af42efd9 123 */