Joep D / CMSIS_DSP_5

The CMSIS DSP 5 library

Dependents:   Nucleo-Heart-Rate ejercicioVrms2 PROYECTOFINAL ejercicioVrms ... more

functions/TransformFunctions/arm_cfft_radix2_f32.c@3:4098b9d3d571, 2018-06-21 (annotated)

Committer:
xorjoep
Date:
Thu Jun 21 11:56:27 2018 +0000
Revision:
3:4098b9d3d571
Parent:
1:24714b45cd1b 
headers is a folder not a library

Who changed what in which revision?

UserRevisionLine numberNew contents of line
xorjoep 1:24714b45cd1b 1 /* ----------------------------------------------------------------------
xorjoep 1:24714b45cd1b 2 * Project: CMSIS DSP Library
xorjoep 1:24714b45cd1b 3 * Title: arm_cfft_radix2_f32.c
xorjoep 1:24714b45cd1b 4 * Description: Radix-2 Decimation in Frequency CFFT & CIFFT Floating point processing function
xorjoep 1:24714b45cd1b 5 *
xorjoep 1:24714b45cd1b 6 * $Date: 27. January 2017
xorjoep 1:24714b45cd1b 7 * $Revision: V.1.5.1
xorjoep 1:24714b45cd1b 8 *
xorjoep 1:24714b45cd1b 9 * Target Processor: Cortex-M cores
xorjoep 1:24714b45cd1b 10 * -------------------------------------------------------------------- */
xorjoep 1:24714b45cd1b 11 /*
xorjoep 1:24714b45cd1b 12 * Copyright (C) 2010-2017 ARM Limited or its affiliates. All rights reserved.
xorjoep 1:24714b45cd1b 13 *
xorjoep 1:24714b45cd1b 14 * SPDX-License-Identifier: Apache-2.0
xorjoep 1:24714b45cd1b 15 *
xorjoep 1:24714b45cd1b 16 * Licensed under the Apache License, Version 2.0 (the License); you may
xorjoep 1:24714b45cd1b 17 * not use this file except in compliance with the License.
xorjoep 1:24714b45cd1b 18 * You may obtain a copy of the License at
xorjoep 1:24714b45cd1b 19 *
xorjoep 1:24714b45cd1b 20 * www.apache.org/licenses/LICENSE-2.0
xorjoep 1:24714b45cd1b 21 *
xorjoep 1:24714b45cd1b 22 * Unless required by applicable law or agreed to in writing, software
xorjoep 1:24714b45cd1b 23 * distributed under the License is distributed on an AS IS BASIS, WITHOUT
xorjoep 1:24714b45cd1b 24 * WARRANTIES OR CONDITIONS OF ANY KIND, either express or implied.
xorjoep 1:24714b45cd1b 25 * See the License for the specific language governing permissions and
xorjoep 1:24714b45cd1b 26 * limitations under the License.
xorjoep 1:24714b45cd1b 27 */
xorjoep 1:24714b45cd1b 28
xorjoep 1:24714b45cd1b 29 #include "arm_math.h"
xorjoep 1:24714b45cd1b 30
xorjoep 1:24714b45cd1b 31 void arm_radix2_butterfly_f32(
xorjoep 1:24714b45cd1b 32 float32_t * pSrc,
xorjoep 1:24714b45cd1b 33 uint32_t fftLen,
xorjoep 1:24714b45cd1b 34 float32_t * pCoef,
xorjoep 1:24714b45cd1b 35 uint16_t twidCoefModifier);
xorjoep 1:24714b45cd1b 36
xorjoep 1:24714b45cd1b 37 void arm_radix2_butterfly_inverse_f32(
xorjoep 1:24714b45cd1b 38 float32_t * pSrc,
xorjoep 1:24714b45cd1b 39 uint32_t fftLen,
xorjoep 1:24714b45cd1b 40 float32_t * pCoef,
xorjoep 1:24714b45cd1b 41 uint16_t twidCoefModifier,
xorjoep 1:24714b45cd1b 42 float32_t onebyfftLen);
xorjoep 1:24714b45cd1b 43
xorjoep 1:24714b45cd1b 44 extern void arm_bitreversal_f32(
xorjoep 1:24714b45cd1b 45 float32_t * pSrc,
xorjoep 1:24714b45cd1b 46 uint16_t fftSize,
xorjoep 1:24714b45cd1b 47 uint16_t bitRevFactor,
xorjoep 1:24714b45cd1b 48 uint16_t * pBitRevTab);
xorjoep 1:24714b45cd1b 49
xorjoep 1:24714b45cd1b 50 /**
xorjoep 1:24714b45cd1b 51 * @ingroup groupTransforms
xorjoep 1:24714b45cd1b 52 */
xorjoep 1:24714b45cd1b 53
xorjoep 1:24714b45cd1b 54 /**
xorjoep 1:24714b45cd1b 55 * @addtogroup ComplexFFT
xorjoep 1:24714b45cd1b 56 * @{
xorjoep 1:24714b45cd1b 57 */
xorjoep 1:24714b45cd1b 58
xorjoep 1:24714b45cd1b 59 /**
xorjoep 1:24714b45cd1b 60 * @details
xorjoep 1:24714b45cd1b 61 * @brief Radix-2 CFFT/CIFFT.
xorjoep 1:24714b45cd1b 62 * @deprecated Do not use this function. It has been superseded by \ref arm_cfft_f32 and will be removed
xorjoep 1:24714b45cd1b 63 * in the future.
xorjoep 1:24714b45cd1b 64 * @param[in] *S points to an instance of the floating-point Radix-2 CFFT/CIFFT structure.
xorjoep 1:24714b45cd1b 65 * @param[in, out] *pSrc points to the complex data buffer of size <code>2*fftLen</code>. Processing occurs in-place.
xorjoep 1:24714b45cd1b 66 * @return none.
xorjoep 1:24714b45cd1b 67 */
xorjoep 1:24714b45cd1b 68
xorjoep 1:24714b45cd1b 69 void arm_cfft_radix2_f32(
xorjoep 1:24714b45cd1b 70 const arm_cfft_radix2_instance_f32 * S,
xorjoep 1:24714b45cd1b 71 float32_t * pSrc)
xorjoep 1:24714b45cd1b 72 {
xorjoep 1:24714b45cd1b 73
xorjoep 1:24714b45cd1b 74 if (S->ifftFlag == 1U)
xorjoep 1:24714b45cd1b 75 {
xorjoep 1:24714b45cd1b 76 /* Complex IFFT radix-2 */
xorjoep 1:24714b45cd1b 77 arm_radix2_butterfly_inverse_f32(pSrc, S->fftLen, S->pTwiddle,
xorjoep 1:24714b45cd1b 78 S->twidCoefModifier, S->onebyfftLen);
xorjoep 1:24714b45cd1b 79 }
xorjoep 1:24714b45cd1b 80 else
xorjoep 1:24714b45cd1b 81 {
xorjoep 1:24714b45cd1b 82 /* Complex FFT radix-2 */
xorjoep 1:24714b45cd1b 83 arm_radix2_butterfly_f32(pSrc, S->fftLen, S->pTwiddle,
xorjoep 1:24714b45cd1b 84 S->twidCoefModifier);
xorjoep 1:24714b45cd1b 85 }
xorjoep 1:24714b45cd1b 86
xorjoep 1:24714b45cd1b 87 if (S->bitReverseFlag == 1U)
xorjoep 1:24714b45cd1b 88 {
xorjoep 1:24714b45cd1b 89 /* Bit Reversal */
xorjoep 1:24714b45cd1b 90 arm_bitreversal_f32(pSrc, S->fftLen, S->bitRevFactor, S->pBitRevTable);
xorjoep 1:24714b45cd1b 91 }
xorjoep 1:24714b45cd1b 92
xorjoep 1:24714b45cd1b 93 }
xorjoep 1:24714b45cd1b 94
xorjoep 1:24714b45cd1b 95
xorjoep 1:24714b45cd1b 96 /**
xorjoep 1:24714b45cd1b 97 * @} end of ComplexFFT group
xorjoep 1:24714b45cd1b 98 */
xorjoep 1:24714b45cd1b 99
xorjoep 1:24714b45cd1b 100
xorjoep 1:24714b45cd1b 101
xorjoep 1:24714b45cd1b 102 /* ----------------------------------------------------------------------
xorjoep 1:24714b45cd1b 103 ** Internal helper function used by the FFTs
xorjoep 1:24714b45cd1b 104 ** ------------------------------------------------------------------- */
xorjoep 1:24714b45cd1b 105
xorjoep 1:24714b45cd1b 106 /*
xorjoep 1:24714b45cd1b 107 * @brief Core function for the floating-point CFFT butterfly process.
xorjoep 1:24714b45cd1b 108 * @param[in, out] *pSrc points to the in-place buffer of floating-point data type.
xorjoep 1:24714b45cd1b 109 * @param[in] fftLen length of the FFT.
xorjoep 1:24714b45cd1b 110 * @param[in] *pCoef points to the twiddle coefficient buffer.
xorjoep 1:24714b45cd1b 111 * @param[in] twidCoefModifier twiddle coefficient modifier that supports different size FFTs with the same twiddle factor table.
xorjoep 1:24714b45cd1b 112 * @return none.
xorjoep 1:24714b45cd1b 113 */
xorjoep 1:24714b45cd1b 114
xorjoep 1:24714b45cd1b 115 void arm_radix2_butterfly_f32(
xorjoep 1:24714b45cd1b 116 float32_t * pSrc,
xorjoep 1:24714b45cd1b 117 uint32_t fftLen,
xorjoep 1:24714b45cd1b 118 float32_t * pCoef,
xorjoep 1:24714b45cd1b 119 uint16_t twidCoefModifier)
xorjoep 1:24714b45cd1b 120 {
xorjoep 1:24714b45cd1b 121
xorjoep 1:24714b45cd1b 122 uint32_t i, j, k, l;
xorjoep 1:24714b45cd1b 123 uint32_t n1, n2, ia;
xorjoep 1:24714b45cd1b 124 float32_t xt, yt, cosVal, sinVal;
xorjoep 1:24714b45cd1b 125 float32_t p0, p1, p2, p3;
xorjoep 1:24714b45cd1b 126 float32_t a0, a1;
xorjoep 1:24714b45cd1b 127
xorjoep 1:24714b45cd1b 128 #if defined (ARM_MATH_DSP)
xorjoep 1:24714b45cd1b 129
xorjoep 1:24714b45cd1b 130 /* Initializations for the first stage */
xorjoep 1:24714b45cd1b 131 n2 = fftLen >> 1;
xorjoep 1:24714b45cd1b 132 ia = 0;
xorjoep 1:24714b45cd1b 133 i = 0;
xorjoep 1:24714b45cd1b 134
xorjoep 1:24714b45cd1b 135 // loop for groups
xorjoep 1:24714b45cd1b 136 for (k = n2; k > 0; k--)
xorjoep 1:24714b45cd1b 137 {
xorjoep 1:24714b45cd1b 138 cosVal = pCoef[ia * 2];
xorjoep 1:24714b45cd1b 139 sinVal = pCoef[(ia * 2) + 1];
xorjoep 1:24714b45cd1b 140
xorjoep 1:24714b45cd1b 141 /* Twiddle coefficients index modifier */
xorjoep 1:24714b45cd1b 142 ia += twidCoefModifier;
xorjoep 1:24714b45cd1b 143
xorjoep 1:24714b45cd1b 144 /* index calculation for the input as, */
xorjoep 1:24714b45cd1b 145 /* pSrc[i + 0], pSrc[i + fftLen/1] */
xorjoep 1:24714b45cd1b 146 l = i + n2;
xorjoep 1:24714b45cd1b 147
xorjoep 1:24714b45cd1b 148 /* Butterfly implementation */
xorjoep 1:24714b45cd1b 149 a0 = pSrc[2 * i] + pSrc[2 * l];
xorjoep 1:24714b45cd1b 150 xt = pSrc[2 * i] - pSrc[2 * l];
xorjoep 1:24714b45cd1b 151
xorjoep 1:24714b45cd1b 152 yt = pSrc[2 * i + 1] - pSrc[2 * l + 1];
xorjoep 1:24714b45cd1b 153 a1 = pSrc[2 * l + 1] + pSrc[2 * i + 1];
xorjoep 1:24714b45cd1b 154
xorjoep 1:24714b45cd1b 155 p0 = xt * cosVal;
xorjoep 1:24714b45cd1b 156 p1 = yt * sinVal;
xorjoep 1:24714b45cd1b 157 p2 = yt * cosVal;
xorjoep 1:24714b45cd1b 158 p3 = xt * sinVal;
xorjoep 1:24714b45cd1b 159
xorjoep 1:24714b45cd1b 160 pSrc[2 * i] = a0;
xorjoep 1:24714b45cd1b 161 pSrc[2 * i + 1] = a1;
xorjoep 1:24714b45cd1b 162
xorjoep 1:24714b45cd1b 163 pSrc[2 * l] = p0 + p1;
xorjoep 1:24714b45cd1b 164 pSrc[2 * l + 1] = p2 - p3;
xorjoep 1:24714b45cd1b 165
xorjoep 1:24714b45cd1b 166 i++;
xorjoep 1:24714b45cd1b 167 } // groups loop end
xorjoep 1:24714b45cd1b 168
xorjoep 1:24714b45cd1b 169 twidCoefModifier <<= 1U;
xorjoep 1:24714b45cd1b 170
xorjoep 1:24714b45cd1b 171 // loop for stage
xorjoep 1:24714b45cd1b 172 for (k = n2; k > 2; k = k >> 1)
xorjoep 1:24714b45cd1b 173 {
xorjoep 1:24714b45cd1b 174 n1 = n2;
xorjoep 1:24714b45cd1b 175 n2 = n2 >> 1;
xorjoep 1:24714b45cd1b 176 ia = 0;
xorjoep 1:24714b45cd1b 177
xorjoep 1:24714b45cd1b 178 // loop for groups
xorjoep 1:24714b45cd1b 179 j = 0;
xorjoep 1:24714b45cd1b 180 do
xorjoep 1:24714b45cd1b 181 {
xorjoep 1:24714b45cd1b 182 cosVal = pCoef[ia * 2];
xorjoep 1:24714b45cd1b 183 sinVal = pCoef[(ia * 2) + 1];
xorjoep 1:24714b45cd1b 184 ia += twidCoefModifier;
xorjoep 1:24714b45cd1b 185
xorjoep 1:24714b45cd1b 186 // loop for butterfly
xorjoep 1:24714b45cd1b 187 i = j;
xorjoep 1:24714b45cd1b 188 do
xorjoep 1:24714b45cd1b 189 {
xorjoep 1:24714b45cd1b 190 l = i + n2;
xorjoep 1:24714b45cd1b 191 a0 = pSrc[2 * i] + pSrc[2 * l];
xorjoep 1:24714b45cd1b 192 xt = pSrc[2 * i] - pSrc[2 * l];
xorjoep 1:24714b45cd1b 193
xorjoep 1:24714b45cd1b 194 yt = pSrc[2 * i + 1] - pSrc[2 * l + 1];
xorjoep 1:24714b45cd1b 195 a1 = pSrc[2 * l + 1] + pSrc[2 * i + 1];
xorjoep 1:24714b45cd1b 196
xorjoep 1:24714b45cd1b 197 p0 = xt * cosVal;
xorjoep 1:24714b45cd1b 198 p1 = yt * sinVal;
xorjoep 1:24714b45cd1b 199 p2 = yt * cosVal;
xorjoep 1:24714b45cd1b 200 p3 = xt * sinVal;
xorjoep 1:24714b45cd1b 201
xorjoep 1:24714b45cd1b 202 pSrc[2 * i] = a0;
xorjoep 1:24714b45cd1b 203 pSrc[2 * i + 1] = a1;
xorjoep 1:24714b45cd1b 204
xorjoep 1:24714b45cd1b 205 pSrc[2 * l] = p0 + p1;
xorjoep 1:24714b45cd1b 206 pSrc[2 * l + 1] = p2 - p3;
xorjoep 1:24714b45cd1b 207
xorjoep 1:24714b45cd1b 208 i += n1;
xorjoep 1:24714b45cd1b 209 } while ( i < fftLen ); // butterfly loop end
xorjoep 1:24714b45cd1b 210 j++;
xorjoep 1:24714b45cd1b 211 } while ( j < n2); // groups loop end
xorjoep 1:24714b45cd1b 212 twidCoefModifier <<= 1U;
xorjoep 1:24714b45cd1b 213 } // stages loop end
xorjoep 1:24714b45cd1b 214
xorjoep 1:24714b45cd1b 215 // loop for butterfly
xorjoep 1:24714b45cd1b 216 for (i = 0; i < fftLen; i += 2)
xorjoep 1:24714b45cd1b 217 {
xorjoep 1:24714b45cd1b 218 a0 = pSrc[2 * i] + pSrc[2 * i + 2];
xorjoep 1:24714b45cd1b 219 xt = pSrc[2 * i] - pSrc[2 * i + 2];
xorjoep 1:24714b45cd1b 220
xorjoep 1:24714b45cd1b 221 yt = pSrc[2 * i + 1] - pSrc[2 * i + 3];
xorjoep 1:24714b45cd1b 222 a1 = pSrc[2 * i + 3] + pSrc[2 * i + 1];
xorjoep 1:24714b45cd1b 223
xorjoep 1:24714b45cd1b 224 pSrc[2 * i] = a0;
xorjoep 1:24714b45cd1b 225 pSrc[2 * i + 1] = a1;
xorjoep 1:24714b45cd1b 226 pSrc[2 * i + 2] = xt;
xorjoep 1:24714b45cd1b 227 pSrc[2 * i + 3] = yt;
xorjoep 1:24714b45cd1b 228 } // groups loop end
xorjoep 1:24714b45cd1b 229
xorjoep 1:24714b45cd1b 230 #else
xorjoep 1:24714b45cd1b 231
xorjoep 1:24714b45cd1b 232 n2 = fftLen;
xorjoep 1:24714b45cd1b 233
xorjoep 1:24714b45cd1b 234 // loop for stage
xorjoep 1:24714b45cd1b 235 for (k = fftLen; k > 1; k = k >> 1)
xorjoep 1:24714b45cd1b 236 {
xorjoep 1:24714b45cd1b 237 n1 = n2;
xorjoep 1:24714b45cd1b 238 n2 = n2 >> 1;
xorjoep 1:24714b45cd1b 239 ia = 0;
xorjoep 1:24714b45cd1b 240
xorjoep 1:24714b45cd1b 241 // loop for groups
xorjoep 1:24714b45cd1b 242 j = 0;
xorjoep 1:24714b45cd1b 243 do
xorjoep 1:24714b45cd1b 244 {
xorjoep 1:24714b45cd1b 245 cosVal = pCoef[ia * 2];
xorjoep 1:24714b45cd1b 246 sinVal = pCoef[(ia * 2) + 1];
xorjoep 1:24714b45cd1b 247 ia += twidCoefModifier;
xorjoep 1:24714b45cd1b 248
xorjoep 1:24714b45cd1b 249 // loop for butterfly
xorjoep 1:24714b45cd1b 250 i = j;
xorjoep 1:24714b45cd1b 251 do
xorjoep 1:24714b45cd1b 252 {
xorjoep 1:24714b45cd1b 253 l = i + n2;
xorjoep 1:24714b45cd1b 254 a0 = pSrc[2 * i] + pSrc[2 * l];
xorjoep 1:24714b45cd1b 255 xt = pSrc[2 * i] - pSrc[2 * l];
xorjoep 1:24714b45cd1b 256
xorjoep 1:24714b45cd1b 257 yt = pSrc[2 * i + 1] - pSrc[2 * l + 1];
xorjoep 1:24714b45cd1b 258 a1 = pSrc[2 * l + 1] + pSrc[2 * i + 1];
xorjoep 1:24714b45cd1b 259
xorjoep 1:24714b45cd1b 260 p0 = xt * cosVal;
xorjoep 1:24714b45cd1b 261 p1 = yt * sinVal;
xorjoep 1:24714b45cd1b 262 p2 = yt * cosVal;
xorjoep 1:24714b45cd1b 263 p3 = xt * sinVal;
xorjoep 1:24714b45cd1b 264
xorjoep 1:24714b45cd1b 265 pSrc[2 * i] = a0;
xorjoep 1:24714b45cd1b 266 pSrc[2 * i + 1] = a1;
xorjoep 1:24714b45cd1b 267
xorjoep 1:24714b45cd1b 268 pSrc[2 * l] = p0 + p1;
xorjoep 1:24714b45cd1b 269 pSrc[2 * l + 1] = p2 - p3;
xorjoep 1:24714b45cd1b 270
xorjoep 1:24714b45cd1b 271 i += n1;
xorjoep 1:24714b45cd1b 272 } while (i < fftLen);
xorjoep 1:24714b45cd1b 273 j++;
xorjoep 1:24714b45cd1b 274 } while (j < n2);
xorjoep 1:24714b45cd1b 275 twidCoefModifier <<= 1U;
xorjoep 1:24714b45cd1b 276 }
xorjoep 1:24714b45cd1b 277
xorjoep 1:24714b45cd1b 278 #endif // #if defined (ARM_MATH_DSP)
xorjoep 1:24714b45cd1b 279
xorjoep 1:24714b45cd1b 280 }
xorjoep 1:24714b45cd1b 281
xorjoep 1:24714b45cd1b 282
xorjoep 1:24714b45cd1b 283 void arm_radix2_butterfly_inverse_f32(
xorjoep 1:24714b45cd1b 284 float32_t * pSrc,
xorjoep 1:24714b45cd1b 285 uint32_t fftLen,
xorjoep 1:24714b45cd1b 286 float32_t * pCoef,
xorjoep 1:24714b45cd1b 287 uint16_t twidCoefModifier,
xorjoep 1:24714b45cd1b 288 float32_t onebyfftLen)
xorjoep 1:24714b45cd1b 289 {
xorjoep 1:24714b45cd1b 290
xorjoep 1:24714b45cd1b 291 uint32_t i, j, k, l;
xorjoep 1:24714b45cd1b 292 uint32_t n1, n2, ia;
xorjoep 1:24714b45cd1b 293 float32_t xt, yt, cosVal, sinVal;
xorjoep 1:24714b45cd1b 294 float32_t p0, p1, p2, p3;
xorjoep 1:24714b45cd1b 295 float32_t a0, a1;
xorjoep 1:24714b45cd1b 296
xorjoep 1:24714b45cd1b 297 #if defined (ARM_MATH_DSP)
xorjoep 1:24714b45cd1b 298
xorjoep 1:24714b45cd1b 299 n2 = fftLen >> 1;
xorjoep 1:24714b45cd1b 300 ia = 0;
xorjoep 1:24714b45cd1b 301
xorjoep 1:24714b45cd1b 302 // loop for groups
xorjoep 1:24714b45cd1b 303 for (i = 0; i < n2; i++)
xorjoep 1:24714b45cd1b 304 {
xorjoep 1:24714b45cd1b 305 cosVal = pCoef[ia * 2];
xorjoep 1:24714b45cd1b 306 sinVal = pCoef[(ia * 2) + 1];
xorjoep 1:24714b45cd1b 307 ia += twidCoefModifier;
xorjoep 1:24714b45cd1b 308
xorjoep 1:24714b45cd1b 309 l = i + n2;
xorjoep 1:24714b45cd1b 310 a0 = pSrc[2 * i] + pSrc[2 * l];
xorjoep 1:24714b45cd1b 311 xt = pSrc[2 * i] - pSrc[2 * l];
xorjoep 1:24714b45cd1b 312
xorjoep 1:24714b45cd1b 313 yt = pSrc[2 * i + 1] - pSrc[2 * l + 1];
xorjoep 1:24714b45cd1b 314 a1 = pSrc[2 * l + 1] + pSrc[2 * i + 1];
xorjoep 1:24714b45cd1b 315
xorjoep 1:24714b45cd1b 316 p0 = xt * cosVal;
xorjoep 1:24714b45cd1b 317 p1 = yt * sinVal;
xorjoep 1:24714b45cd1b 318 p2 = yt * cosVal;
xorjoep 1:24714b45cd1b 319 p3 = xt * sinVal;
xorjoep 1:24714b45cd1b 320
xorjoep 1:24714b45cd1b 321 pSrc[2 * i] = a0;
xorjoep 1:24714b45cd1b 322 pSrc[2 * i + 1] = a1;
xorjoep 1:24714b45cd1b 323
xorjoep 1:24714b45cd1b 324 pSrc[2 * l] = p0 - p1;
xorjoep 1:24714b45cd1b 325 pSrc[2 * l + 1] = p2 + p3;
xorjoep 1:24714b45cd1b 326 } // groups loop end
xorjoep 1:24714b45cd1b 327
xorjoep 1:24714b45cd1b 328 twidCoefModifier <<= 1U;
xorjoep 1:24714b45cd1b 329
xorjoep 1:24714b45cd1b 330 // loop for stage
xorjoep 1:24714b45cd1b 331 for (k = fftLen / 2; k > 2; k = k >> 1)
xorjoep 1:24714b45cd1b 332 {
xorjoep 1:24714b45cd1b 333 n1 = n2;
xorjoep 1:24714b45cd1b 334 n2 = n2 >> 1;
xorjoep 1:24714b45cd1b 335 ia = 0;
xorjoep 1:24714b45cd1b 336
xorjoep 1:24714b45cd1b 337 // loop for groups
xorjoep 1:24714b45cd1b 338 j = 0;
xorjoep 1:24714b45cd1b 339 do
xorjoep 1:24714b45cd1b 340 {
xorjoep 1:24714b45cd1b 341 cosVal = pCoef[ia * 2];
xorjoep 1:24714b45cd1b 342 sinVal = pCoef[(ia * 2) + 1];
xorjoep 1:24714b45cd1b 343 ia += twidCoefModifier;
xorjoep 1:24714b45cd1b 344
xorjoep 1:24714b45cd1b 345 // loop for butterfly
xorjoep 1:24714b45cd1b 346 i = j;
xorjoep 1:24714b45cd1b 347 do
xorjoep 1:24714b45cd1b 348 {
xorjoep 1:24714b45cd1b 349 l = i + n2;
xorjoep 1:24714b45cd1b 350 a0 = pSrc[2 * i] + pSrc[2 * l];
xorjoep 1:24714b45cd1b 351 xt = pSrc[2 * i] - pSrc[2 * l];
xorjoep 1:24714b45cd1b 352
xorjoep 1:24714b45cd1b 353 yt = pSrc[2 * i + 1] - pSrc[2 * l + 1];
xorjoep 1:24714b45cd1b 354 a1 = pSrc[2 * l + 1] + pSrc[2 * i + 1];
xorjoep 1:24714b45cd1b 355
xorjoep 1:24714b45cd1b 356 p0 = xt * cosVal;
xorjoep 1:24714b45cd1b 357 p1 = yt * sinVal;
xorjoep 1:24714b45cd1b 358 p2 = yt * cosVal;
xorjoep 1:24714b45cd1b 359 p3 = xt * sinVal;
xorjoep 1:24714b45cd1b 360
xorjoep 1:24714b45cd1b 361 pSrc[2 * i] = a0;
xorjoep 1:24714b45cd1b 362 pSrc[2 * i + 1] = a1;
xorjoep 1:24714b45cd1b 363
xorjoep 1:24714b45cd1b 364 pSrc[2 * l] = p0 - p1;
xorjoep 1:24714b45cd1b 365 pSrc[2 * l + 1] = p2 + p3;
xorjoep 1:24714b45cd1b 366
xorjoep 1:24714b45cd1b 367 i += n1;
xorjoep 1:24714b45cd1b 368 } while ( i < fftLen ); // butterfly loop end
xorjoep 1:24714b45cd1b 369 j++;
xorjoep 1:24714b45cd1b 370 } while (j < n2); // groups loop end
xorjoep 1:24714b45cd1b 371
xorjoep 1:24714b45cd1b 372 twidCoefModifier <<= 1U;
xorjoep 1:24714b45cd1b 373 } // stages loop end
xorjoep 1:24714b45cd1b 374
xorjoep 1:24714b45cd1b 375 // loop for butterfly
xorjoep 1:24714b45cd1b 376 for (i = 0; i < fftLen; i += 2)
xorjoep 1:24714b45cd1b 377 {
xorjoep 1:24714b45cd1b 378 a0 = pSrc[2 * i] + pSrc[2 * i + 2];
xorjoep 1:24714b45cd1b 379 xt = pSrc[2 * i] - pSrc[2 * i + 2];
xorjoep 1:24714b45cd1b 380
xorjoep 1:24714b45cd1b 381 a1 = pSrc[2 * i + 3] + pSrc[2 * i + 1];
xorjoep 1:24714b45cd1b 382 yt = pSrc[2 * i + 1] - pSrc[2 * i + 3];
xorjoep 1:24714b45cd1b 383
xorjoep 1:24714b45cd1b 384 p0 = a0 * onebyfftLen;
xorjoep 1:24714b45cd1b 385 p2 = xt * onebyfftLen;
xorjoep 1:24714b45cd1b 386 p1 = a1 * onebyfftLen;
xorjoep 1:24714b45cd1b 387 p3 = yt * onebyfftLen;
xorjoep 1:24714b45cd1b 388
xorjoep 1:24714b45cd1b 389 pSrc[2 * i] = p0;
xorjoep 1:24714b45cd1b 390 pSrc[2 * i + 1] = p1;
xorjoep 1:24714b45cd1b 391 pSrc[2 * i + 2] = p2;
xorjoep 1:24714b45cd1b 392 pSrc[2 * i + 3] = p3;
xorjoep 1:24714b45cd1b 393 } // butterfly loop end
xorjoep 1:24714b45cd1b 394
xorjoep 1:24714b45cd1b 395 #else
xorjoep 1:24714b45cd1b 396
xorjoep 1:24714b45cd1b 397 n2 = fftLen;
xorjoep 1:24714b45cd1b 398
xorjoep 1:24714b45cd1b 399 // loop for stage
xorjoep 1:24714b45cd1b 400 for (k = fftLen; k > 2; k = k >> 1)
xorjoep 1:24714b45cd1b 401 {
xorjoep 1:24714b45cd1b 402 n1 = n2;
xorjoep 1:24714b45cd1b 403 n2 = n2 >> 1;
xorjoep 1:24714b45cd1b 404 ia = 0;
xorjoep 1:24714b45cd1b 405
xorjoep 1:24714b45cd1b 406 // loop for groups
xorjoep 1:24714b45cd1b 407 j = 0;
xorjoep 1:24714b45cd1b 408 do
xorjoep 1:24714b45cd1b 409 {
xorjoep 1:24714b45cd1b 410 cosVal = pCoef[ia * 2];
xorjoep 1:24714b45cd1b 411 sinVal = pCoef[(ia * 2) + 1];
xorjoep 1:24714b45cd1b 412 ia = ia + twidCoefModifier;
xorjoep 1:24714b45cd1b 413
xorjoep 1:24714b45cd1b 414 // loop for butterfly
xorjoep 1:24714b45cd1b 415 i = j;
xorjoep 1:24714b45cd1b 416 do
xorjoep 1:24714b45cd1b 417 {
xorjoep 1:24714b45cd1b 418 l = i + n2;
xorjoep 1:24714b45cd1b 419 a0 = pSrc[2 * i] + pSrc[2 * l];
xorjoep 1:24714b45cd1b 420 xt = pSrc[2 * i] - pSrc[2 * l];
xorjoep 1:24714b45cd1b 421
xorjoep 1:24714b45cd1b 422 yt = pSrc[2 * i + 1] - pSrc[2 * l + 1];
xorjoep 1:24714b45cd1b 423 a1 = pSrc[2 * l + 1] + pSrc[2 * i + 1];
xorjoep 1:24714b45cd1b 424
xorjoep 1:24714b45cd1b 425 p0 = xt * cosVal;
xorjoep 1:24714b45cd1b 426 p1 = yt * sinVal;
xorjoep 1:24714b45cd1b 427 p2 = yt * cosVal;
xorjoep 1:24714b45cd1b 428 p3 = xt * sinVal;
xorjoep 1:24714b45cd1b 429
xorjoep 1:24714b45cd1b 430 pSrc[2 * i] = a0;
xorjoep 1:24714b45cd1b 431 pSrc[2 * i + 1] = a1;
xorjoep 1:24714b45cd1b 432
xorjoep 1:24714b45cd1b 433 pSrc[2 * l] = p0 - p1;
xorjoep 1:24714b45cd1b 434 pSrc[2 * l + 1] = p2 + p3;
xorjoep 1:24714b45cd1b 435
xorjoep 1:24714b45cd1b 436 i += n1;
xorjoep 1:24714b45cd1b 437 } while ( i < fftLen ); // butterfly loop end
xorjoep 1:24714b45cd1b 438 j++;
xorjoep 1:24714b45cd1b 439 } while ( j < n2 ); // groups loop end
xorjoep 1:24714b45cd1b 440
xorjoep 1:24714b45cd1b 441 twidCoefModifier = twidCoefModifier << 1U;
xorjoep 1:24714b45cd1b 442 } // stages loop end
xorjoep 1:24714b45cd1b 443
xorjoep 1:24714b45cd1b 444 n1 = n2;
xorjoep 1:24714b45cd1b 445 n2 = n2 >> 1;
xorjoep 1:24714b45cd1b 446
xorjoep 1:24714b45cd1b 447 // loop for butterfly
xorjoep 1:24714b45cd1b 448 for (i = 0; i < fftLen; i += n1)
xorjoep 1:24714b45cd1b 449 {
xorjoep 1:24714b45cd1b 450 l = i + n2;
xorjoep 1:24714b45cd1b 451
xorjoep 1:24714b45cd1b 452 a0 = pSrc[2 * i] + pSrc[2 * l];
xorjoep 1:24714b45cd1b 453 xt = pSrc[2 * i] - pSrc[2 * l];
xorjoep 1:24714b45cd1b 454
xorjoep 1:24714b45cd1b 455 a1 = pSrc[2 * l + 1] + pSrc[2 * i + 1];
xorjoep 1:24714b45cd1b 456 yt = pSrc[2 * i + 1] - pSrc[2 * l + 1];
xorjoep 1:24714b45cd1b 457
xorjoep 1:24714b45cd1b 458 p0 = a0 * onebyfftLen;
xorjoep 1:24714b45cd1b 459 p2 = xt * onebyfftLen;
xorjoep 1:24714b45cd1b 460 p1 = a1 * onebyfftLen;
xorjoep 1:24714b45cd1b 461 p3 = yt * onebyfftLen;
xorjoep 1:24714b45cd1b 462
xorjoep 1:24714b45cd1b 463 pSrc[2 * i] = p0;
xorjoep 1:24714b45cd1b 464 pSrc[2U * l] = p2;
xorjoep 1:24714b45cd1b 465
xorjoep 1:24714b45cd1b 466 pSrc[2 * i + 1] = p1;
xorjoep 1:24714b45cd1b 467 pSrc[2U * l + 1U] = p3;
xorjoep 1:24714b45cd1b 468 } // butterfly loop end
xorjoep 1:24714b45cd1b 469
xorjoep 1:24714b45cd1b 470 #endif // #if defined (ARM_MATH_DSP)
xorjoep 1:24714b45cd1b 471
xorjoep 1:24714b45cd1b 472 }