1234567891011121314151617181920212223242526272829303132333435363738394041424344454647484950515253545556575859606162636465666768697071727374757677787980818283848586878889909192939495969798991001011021031041051061071081091101111121131141151161171181191201211221231241251261271281291301311321331341351361371381391401411421431441451461471481491501511521531541551561571581591601611621631641651661671681691701711721731741751761771781791801811821831841851861871881891901911921931941951961971981992002012022032042052062072082092102112122132142152162172182192202212222232242252262272282292302312322332342352362372382392402412422432442452462472482492502512522532542552562572582592602612622632642652662672682692702712722732742752762772782792802812822832842852862872882892902912922932942952962972982993003013023033043053063073083093103113123133143153163173183193203213223233243253263273283293303313323333343353363373383393403413423433443453463473483493503513523533543553563573583593603613623633643653663673683693703713723733743753763773783793803813823833843853863873883893903913923933943953963973983994004014024034044054064074084094104114124134144154164174184194204214224234244254264274284294304314324334344354364374384394404414424434444454464474484494504514524534544554564574584594604614624634644654664674684694704714724734744754764774784794804814824834844854864874884894904914924934944954964974984995005015025035045055065075085095105115125135145155165175185195205215225235245255265275285295305315325335345355365375385395405415425435445455465475485495505515525535545555565575585595605615625635645655665675685695705715725735745755765775785795805815825835845855865875885895905915925935945955965975985996006016026036046056066076086096106116126136146156166176186196206216226236246256266276286296306316326336346356366376386396406416426436446456466476486496506516526536546556566576586596606616626636646656666676686696706716726736746756766776786796806816826836846856866876886896906916926936946956966976986997007017027037047057067077087097107117127137147157167177187197207217227237247257267277287297307317327337347357367377387397407417427437447457467477487497507517527537547557567577587597607617627637647657667677687697707717727737747757767777787797807817827837847857867877887897907917927937947957967977987998008018028038048058068078088098108118128138148158168178188198208218228238248258268278288298308318328338348358368378388398408418428438448458468478488498508518528538548558568578588598608618628638648658668678688698708718728738748758768778788798808818828838848858868878888898908918928938948958968978988999009019029039049059069079089099109119129139149159169179189199209219229239249259269279289299309319329339349359369379389399409419429439449459469479489499509519529539549559569579589599609619629639649659669679689699709719729739749759769779789799809819829839849859869879889899909919929939949959969979989991000100110021003100410051006100710081009101010111012101310141015101610171018101910201021102210231024102510261027102810291030103110321033103410351036103710381039104010411042104310441045104610471048104910501051105210531054105510561057105810591060106110621063106410651066106710681069107010711072107310741075107610771078107910801081108210831084108510861087108810891090109110921093109410951096109710981099110011011102110311041105110611071108110911101111111211131114111511161117111811191120112111221123112411251126112711281129113011311132113311341135113611371138113911401141114211431144114511461147114811491150115111521153115411551156115711581159116011611162116311641165116611671168116911701171117211731174117511761177117811791180118111821183118411851186118711881189119011911192119311941195119611971198119912001201120212031204120512061207120812091210121112121213121412151216121712181219122012211222122312241225122612271228122912301231123212331234123512361237123812391240124112421243124412451246124712481249125012511252125312541255125612571258125912601261126212631264126512661267126812691270127112721273127412751276127712781279128012811282128312841285128612871288128912901291129212931294129512961297129812991300130113021303130413051306130713081309131013111312131313141315131613171318131913201321132213231324132513261327132813291330133113321333133413351336133713381339134013411342134313441345134613471348134913501351135213531354135513561357135813591360136113621363136413651366136713681369137013711372137313741375137613771378137913801381138213831384138513861387138813891390139113921393139413951396139713981399140014011402140314041405140614071408140914101411141214131414141514161417141814191420142114221423142414251426142714281429143014311432143314341435143614371438143914401441144214431444144514461447144814491450145114521453145414551456145714581459146014611462146314641465146614671468146914701471147214731474147514761477147814791480148114821483148414851486148714881489149014911492149314941495149614971498149915001501150215031504150515061507150815091510151115121513151415151516151715181519152015211522152315241525152615271528152915301531153215331534153515361537153815391540154115421543154415451546154715481549155015511552155315541555155615571558155915601561156215631564156515661567156815691570157115721573157415751576157715781579158015811582158315841585158615871588158915901591159215931594159515961597159815991600160116021603160416051606160716081609161016111612161316141615161616171618161916201621162216231624162516261627162816291630163116321633163416351636163716381639164016411642164316441645164616471648164916501651165216531654165516561657165816591660166116621663166416651666166716681669167016711672167316741675167616771678167916801681168216831684168516861687168816891690169116921693169416951696169716981699170017011702170317041705170617071708170917101711171217131714171517161717171817191720172117221723172417251726172717281729173017311732173317341735173617371738173917401741174217431744174517461747174817491750175117521753175417551756175717581759176017611762176317641765176617671768176917701771177217731774177517761777177817791780178117821783178417851786178717881789179017911792179317941795179617971798179918001801180218031804180518061807180818091810 |
- /* ----------------------------------------------------------------------
- * Project: CMSIS DSP Library
- * Title: arm_cfft_radix4_q15.c
- * Description: This file has function definition of Radix-4 FFT & IFFT function and
- * In-place bit reversal using bit reversal table
- *
- * $Date: 18. March 2019
- * $Revision: V1.6.0
- *
- * Target Processor: Cortex-M cores
- * -------------------------------------------------------------------- */
- /*
- * Copyright (C) 2010-2019 ARM Limited or its affiliates. All rights reserved.
- *
- * SPDX-License-Identifier: Apache-2.0
- *
- * Licensed under the Apache License, Version 2.0 (the License); you may
- * not use this file except in compliance with the License.
- * You may obtain a copy of the License at
- *
- * www.apache.org/licenses/LICENSE-2.0
- *
- * Unless required by applicable law or agreed to in writing, software
- * distributed under the License is distributed on an AS IS BASIS, WITHOUT
- * WARRANTIES OR CONDITIONS OF ANY KIND, either express or implied.
- * See the License for the specific language governing permissions and
- * limitations under the License.
- */
- #include "arm_math.h"
- void arm_radix4_butterfly_q15(
- q15_t * pSrc16,
- uint32_t fftLen,
- const q15_t * pCoef16,
- uint32_t twidCoefModifier);
- void arm_radix4_butterfly_inverse_q15(
- q15_t * pSrc16,
- uint32_t fftLen,
- const q15_t * pCoef16,
- uint32_t twidCoefModifier);
- void arm_bitreversal_q15(
- q15_t * pSrc,
- uint32_t fftLen,
- uint16_t bitRevFactor,
- const uint16_t * pBitRevTab);
- /**
- @ingroup groupTransforms
- */
- /**
- @addtogroup ComplexFFT
- @{
- */
- /**
- @brief Processing function for the Q15 CFFT/CIFFT.
- @deprecated Do not use this function. It has been superseded by \ref arm_cfft_q15 and will be removed in the future.
- @param[in] S points to an instance of the Q15 CFFT/CIFFT structure.
- @param[in,out] pSrc points to the complex data buffer. Processing occurs in-place.
- @return none
-
- @par Input and output formats:
- Internally input is downscaled by 2 for every stage to avoid saturations inside CFFT/CIFFT process.
- Hence the output format is different for different FFT sizes.
- The input and output formats for different FFT sizes and number of bits to upscale are mentioned in the tables below for CFFT and CIFFT:
- @par
- \image html CFFTQ15.gif "Input and Output Formats for Q15 CFFT"
- \image html CIFFTQ15.gif "Input and Output Formats for Q15 CIFFT"
- */
- void arm_cfft_radix4_q15(
- const arm_cfft_radix4_instance_q15 * S,
- q15_t * pSrc)
- {
- if (S->ifftFlag == 1U)
- {
- /* Complex IFFT radix-4 */
- arm_radix4_butterfly_inverse_q15(pSrc, S->fftLen, S->pTwiddle, S->twidCoefModifier);
- }
- else
- {
- /* Complex FFT radix-4 */
- arm_radix4_butterfly_q15(pSrc, S->fftLen, S->pTwiddle, S->twidCoefModifier);
- }
- if (S->bitReverseFlag == 1U)
- {
- /* Bit Reversal */
- arm_bitreversal_q15(pSrc, S->fftLen, S->bitRevFactor, S->pBitRevTable);
- }
- }
- /**
- @} end of ComplexFFT group
- */
- /*
- * Radix-4 FFT algorithm used is :
- *
- * Input real and imaginary data:
- * x(n) = xa + j * ya
- * x(n+N/4 ) = xb + j * yb
- * x(n+N/2 ) = xc + j * yc
- * x(n+3N 4) = xd + j * yd
- *
- *
- * Output real and imaginary data:
- * x(4r) = xa'+ j * ya'
- * x(4r+1) = xb'+ j * yb'
- * x(4r+2) = xc'+ j * yc'
- * x(4r+3) = xd'+ j * yd'
- *
- *
- * Twiddle factors for radix-4 FFT:
- * Wn = co1 + j * (- si1)
- * W2n = co2 + j * (- si2)
- * W3n = co3 + j * (- si3)
-
- * The real and imaginary output values for the radix-4 butterfly are
- * xa' = xa + xb + xc + xd
- * ya' = ya + yb + yc + yd
- * xb' = (xa+yb-xc-yd)* co1 + (ya-xb-yc+xd)* (si1)
- * yb' = (ya-xb-yc+xd)* co1 - (xa+yb-xc-yd)* (si1)
- * xc' = (xa-xb+xc-xd)* co2 + (ya-yb+yc-yd)* (si2)
- * yc' = (ya-yb+yc-yd)* co2 - (xa-xb+xc-xd)* (si2)
- * xd' = (xa-yb-xc+yd)* co3 + (ya+xb-yc-xd)* (si3)
- * yd' = (ya+xb-yc-xd)* co3 - (xa-yb-xc+yd)* (si3)
- *
- */
- /**
- @brief Core function for the Q15 CFFT butterfly process.
- @param[in,out] pSrc16 points to the in-place buffer of Q15 data type
- @param[in] fftLen length of the FFT
- @param[in] pCoef16 points to twiddle coefficient buffer
- @param[in] twidCoefModifier twiddle coefficient modifier that supports different size FFTs with the same twiddle factor table
- @return none
- */
- void arm_radix4_butterfly_q15(
- q15_t * pSrc16,
- uint32_t fftLen,
- const q15_t * pCoef16,
- uint32_t twidCoefModifier)
- {
- #if defined (ARM_MATH_DSP)
- q31_t R, S, T, U;
- q31_t C1, C2, C3, out1, out2;
- uint32_t n1, n2, ic, i0, j, k;
- q15_t *ptr1;
- q15_t *pSi0;
- q15_t *pSi1;
- q15_t *pSi2;
- q15_t *pSi3;
- q31_t xaya, xbyb, xcyc, xdyd;
- /* Total process is divided into three stages */
- /* process first stage, middle stages, & last stage */
- /* Initializations for the first stage */
- n2 = fftLen;
- n1 = n2;
- /* n2 = fftLen/4 */
- n2 >>= 2U;
- /* Index for twiddle coefficient */
- ic = 0U;
- /* Index for input read and output write */
- j = n2;
- pSi0 = pSrc16;
- pSi1 = pSi0 + 2 * n2;
- pSi2 = pSi1 + 2 * n2;
- pSi3 = pSi2 + 2 * n2;
- /* Input is in 1.15(q15) format */
- /* start of first stage process */
- do
- {
- /* Butterfly implementation */
- /* Reading i0, i0+fftLen/2 inputs */
- /* Read ya (real), xa(imag) input */
- T = read_q15x2 (pSi0);
- T = __SHADD16(T, 0); /* this is just a SIMD arithmetic shift right by 1 */
- T = __SHADD16(T, 0); /* it turns out doing this twice is 2 cycles, the alternative takes 3 cycles */
- /*
- in = ((int16_t) (T & 0xFFFF)) >> 2; // alternative code that takes 3 cycles
- T = ((T >> 2) & 0xFFFF0000) | (in & 0xFFFF);
- */
- /* Read yc (real), xc(imag) input */
- S = read_q15x2 (pSi2);
- S = __SHADD16(S, 0);
- S = __SHADD16(S, 0);
- /* R = packed((ya + yc), (xa + xc) ) */
- R = __QADD16(T, S);
- /* S = packed((ya - yc), (xa - xc) ) */
- S = __QSUB16(T, S);
- /* Reading i0+fftLen/4 , i0+3fftLen/4 inputs */
- /* Read yb (real), xb(imag) input */
- T = read_q15x2 (pSi1);
- T = __SHADD16(T, 0);
- T = __SHADD16(T, 0);
- /* Read yd (real), xd(imag) input */
- U = read_q15x2 (pSi3);
- U = __SHADD16(U, 0);
- U = __SHADD16(U, 0);
- /* T = packed((yb + yd), (xb + xd) ) */
- T = __QADD16(T, U);
- /* writing the butterfly processed i0 sample */
- /* xa' = xa + xb + xc + xd */
- /* ya' = ya + yb + yc + yd */
- write_q15x2_ia (&pSi0, __SHADD16(R, T));
- /* R = packed((ya + yc) - (yb + yd), (xa + xc)- (xb + xd)) */
- R = __QSUB16(R, T);
- /* co2 & si2 are read from SIMD Coefficient pointer */
- C2 = read_q15x2 ((q15_t *) pCoef16 + (4U * ic));
- #ifndef ARM_MATH_BIG_ENDIAN
- /* xc' = (xa-xb+xc-xd)* co2 + (ya-yb+yc-yd)* (si2) */
- out1 = __SMUAD(C2, R) >> 16U;
- /* yc' = (ya-yb+yc-yd)* co2 - (xa-xb+xc-xd)* (si2) */
- out2 = __SMUSDX(C2, R);
- #else
- /* xc' = (ya-yb+yc-yd)* co2 - (xa-xb+xc-xd)* (si2) */
- out1 = __SMUSDX(R, C2) >> 16U;
- /* yc' = (xa-xb+xc-xd)* co2 + (ya-yb+yc-yd)* (si2) */
- out2 = __SMUAD(C2, R);
- #endif /* #ifndef ARM_MATH_BIG_ENDIAN */
- /* Reading i0+fftLen/4 */
- /* T = packed(yb, xb) */
- T = read_q15x2 (pSi1);
- T = __SHADD16(T, 0);
- T = __SHADD16(T, 0);
- /* writing the butterfly processed i0 + fftLen/4 sample */
- /* writing output(xc', yc') in little endian format */
- write_q15x2_ia (&pSi1, (q31_t) ((out2) & 0xFFFF0000) | (out1 & 0x0000FFFF));
- /* Butterfly calculations */
- /* U = packed(yd, xd) */
- U = read_q15x2 (pSi3);
- U = __SHADD16(U, 0);
- U = __SHADD16(U, 0);
- /* T = packed(yb-yd, xb-xd) */
- T = __QSUB16(T, U);
- #ifndef ARM_MATH_BIG_ENDIAN
- /* R = packed((ya-yc) + (xb- xd) , (xa-xc) - (yb-yd)) */
- R = __QASX(S, T);
- /* S = packed((ya-yc) - (xb- xd), (xa-xc) + (yb-yd)) */
- S = __QSAX(S, T);
- #else
- /* R = packed((ya-yc) + (xb- xd) , (xa-xc) - (yb-yd)) */
- R = __QSAX(S, T);
- /* S = packed((ya-yc) - (xb- xd), (xa-xc) + (yb-yd)) */
- S = __QASX(S, T);
- #endif /* #ifndef ARM_MATH_BIG_ENDIAN */
- /* co1 & si1 are read from SIMD Coefficient pointer */
- C1 = read_q15x2 ((q15_t *) pCoef16 + (2U * ic));
- /* Butterfly process for the i0+fftLen/2 sample */
- #ifndef ARM_MATH_BIG_ENDIAN
- /* xb' = (xa+yb-xc-yd)* co1 + (ya-xb-yc+xd)* (si1) */
- out1 = __SMUAD(C1, S) >> 16U;
- /* yb' = (ya-xb-yc+xd)* co1 - (xa+yb-xc-yd)* (si1) */
- out2 = __SMUSDX(C1, S);
- #else
- /* xb' = (ya-xb-yc+xd)* co1 - (xa+yb-xc-yd)* (si1) */
- out1 = __SMUSDX(S, C1) >> 16U;
- /* yb' = (xa+yb-xc-yd)* co1 + (ya-xb-yc+xd)* (si1) */
- out2 = __SMUAD(C1, S);
- #endif /* #ifndef ARM_MATH_BIG_ENDIAN */
- /* writing output(xb', yb') in little endian format */
- write_q15x2_ia (&pSi2, ((out2) & 0xFFFF0000) | ((out1) & 0x0000FFFF));
- /* co3 & si3 are read from SIMD Coefficient pointer */
- C3 = read_q15x2 ((q15_t *) pCoef16 + (6U * ic));
- /* Butterfly process for the i0+3fftLen/4 sample */
- #ifndef ARM_MATH_BIG_ENDIAN
- /* xd' = (xa-yb-xc+yd)* co3 + (ya+xb-yc-xd)* (si3) */
- out1 = __SMUAD(C3, R) >> 16U;
- /* yd' = (ya+xb-yc-xd)* co3 - (xa-yb-xc+yd)* (si3) */
- out2 = __SMUSDX(C3, R);
- #else
- /* xd' = (ya+xb-yc-xd)* co3 - (xa-yb-xc+yd)* (si3) */
- out1 = __SMUSDX(R, C3) >> 16U;
- /* yd' = (xa-yb-xc+yd)* co3 + (ya+xb-yc-xd)* (si3) */
- out2 = __SMUAD(C3, R);
- #endif /* #ifndef ARM_MATH_BIG_ENDIAN */
- /* writing output(xd', yd') in little endian format */
- write_q15x2_ia (&pSi3, ((out2) & 0xFFFF0000) | (out1 & 0x0000FFFF));
- /* Twiddle coefficients index modifier */
- ic = ic + twidCoefModifier;
- } while (--j);
- /* data is in 4.11(q11) format */
- /* end of first stage process */
- /* start of middle stage process */
- /* Twiddle coefficients index modifier */
- twidCoefModifier <<= 2U;
- /* Calculation of Middle stage */
- for (k = fftLen / 4U; k > 4U; k >>= 2U)
- {
- /* Initializations for the middle stage */
- n1 = n2;
- n2 >>= 2U;
- ic = 0U;
- for (j = 0U; j <= (n2 - 1U); j++)
- {
- /* index calculation for the coefficients */
- C1 = read_q15x2 ((q15_t *) pCoef16 + (2U * ic));
- C2 = read_q15x2 ((q15_t *) pCoef16 + (4U * ic));
- C3 = read_q15x2 ((q15_t *) pCoef16 + (6U * ic));
- /* Twiddle coefficients index modifier */
- ic = ic + twidCoefModifier;
- pSi0 = pSrc16 + 2 * j;
- pSi1 = pSi0 + 2 * n2;
- pSi2 = pSi1 + 2 * n2;
- pSi3 = pSi2 + 2 * n2;
- /* Butterfly implementation */
- for (i0 = j; i0 < fftLen; i0 += n1)
- {
- /* Reading i0, i0+fftLen/2 inputs */
- /* Read ya (real), xa(imag) input */
- T = read_q15x2 (pSi0);
- /* Read yc (real), xc(imag) input */
- S = read_q15x2 (pSi2);
- /* R = packed( (ya + yc), (xa + xc)) */
- R = __QADD16(T, S);
- /* S = packed((ya - yc), (xa - xc)) */
- S = __QSUB16(T, S);
- /* Reading i0+fftLen/4 , i0+3fftLen/4 inputs */
- /* Read yb (real), xb(imag) input */
- T = read_q15x2 (pSi1);
- /* Read yd (real), xd(imag) input */
- U = read_q15x2 (pSi3);
- /* T = packed( (yb + yd), (xb + xd)) */
- T = __QADD16(T, U);
- /* writing the butterfly processed i0 sample */
- /* xa' = xa + xb + xc + xd */
- /* ya' = ya + yb + yc + yd */
- out1 = __SHADD16(R, T);
- out1 = __SHADD16(out1, 0);
- write_q15x2 (pSi0, out1);
- pSi0 += 2 * n1;
- /* R = packed( (ya + yc) - (yb + yd), (xa + xc) - (xb + xd)) */
- R = __SHSUB16(R, T);
- #ifndef ARM_MATH_BIG_ENDIAN
- /* (ya-yb+yc-yd)* (si2) + (xa-xb+xc-xd)* co2 */
- out1 = __SMUAD(C2, R) >> 16U;
- /* (ya-yb+yc-yd)* co2 - (xa-xb+xc-xd)* (si2) */
- out2 = __SMUSDX(C2, R);
- #else
- /* (ya-yb+yc-yd)* co2 - (xa-xb+xc-xd)* (si2) */
- out1 = __SMUSDX(R, C2) >> 16U;
- /* (ya-yb+yc-yd)* (si2) + (xa-xb+xc-xd)* co2 */
- out2 = __SMUAD(C2, R);
- #endif /* #ifndef ARM_MATH_BIG_ENDIAN */
- /* Reading i0+3fftLen/4 */
- /* Read yb (real), xb(imag) input */
- T = read_q15x2 (pSi1);
- /* writing the butterfly processed i0 + fftLen/4 sample */
- /* xc' = (xa-xb+xc-xd)* co2 + (ya-yb+yc-yd)* (si2) */
- /* yc' = (ya-yb+yc-yd)* co2 - (xa-xb+xc-xd)* (si2) */
- write_q15x2 (pSi1, ((out2) & 0xFFFF0000) | (out1 & 0x0000FFFF));
- pSi1 += 2 * n1;
- /* Butterfly calculations */
- /* Read yd (real), xd(imag) input */
- U = read_q15x2 (pSi3);
- /* T = packed(yb-yd, xb-xd) */
- T = __QSUB16(T, U);
- #ifndef ARM_MATH_BIG_ENDIAN
- /* R = packed((ya-yc) + (xb- xd) , (xa-xc) - (yb-yd)) */
- R = __SHASX(S, T);
- /* S = packed((ya-yc) - (xb- xd), (xa-xc) + (yb-yd)) */
- S = __SHSAX(S, T);
- /* Butterfly process for the i0+fftLen/2 sample */
- out1 = __SMUAD(C1, S) >> 16U;
- out2 = __SMUSDX(C1, S);
- #else
- /* R = packed((ya-yc) + (xb- xd) , (xa-xc) - (yb-yd)) */
- R = __SHSAX(S, T);
- /* S = packed((ya-yc) - (xb- xd), (xa-xc) + (yb-yd)) */
- S = __SHASX(S, T);
- /* Butterfly process for the i0+fftLen/2 sample */
- out1 = __SMUSDX(S, C1) >> 16U;
- out2 = __SMUAD(C1, S);
- #endif /* #ifndef ARM_MATH_BIG_ENDIAN */
- /* xb' = (xa+yb-xc-yd)* co1 + (ya-xb-yc+xd)* (si1) */
- /* yb' = (ya-xb-yc+xd)* co1 - (xa+yb-xc-yd)* (si1) */
- write_q15x2 (pSi2, ((out2) & 0xFFFF0000) | (out1 & 0x0000FFFF));
- pSi2 += 2 * n1;
- /* Butterfly process for the i0+3fftLen/4 sample */
- #ifndef ARM_MATH_BIG_ENDIAN
- out1 = __SMUAD(C3, R) >> 16U;
- out2 = __SMUSDX(C3, R);
- #else
- out1 = __SMUSDX(R, C3) >> 16U;
- out2 = __SMUAD(C3, R);
- #endif /* #ifndef ARM_MATH_BIG_ENDIAN */
- /* xd' = (xa-yb-xc+yd)* co3 + (ya+xb-yc-xd)* (si3) */
- /* yd' = (ya+xb-yc-xd)* co3 - (xa-yb-xc+yd)* (si3) */
- write_q15x2 (pSi3, ((out2) & 0xFFFF0000) | (out1 & 0x0000FFFF));
- pSi3 += 2 * n1;
- }
- }
- /* Twiddle coefficients index modifier */
- twidCoefModifier <<= 2U;
- }
- /* end of middle stage process */
- /* data is in 10.6(q6) format for the 1024 point */
- /* data is in 8.8(q8) format for the 256 point */
- /* data is in 6.10(q10) format for the 64 point */
- /* data is in 4.12(q12) format for the 16 point */
- /* Initializations for the last stage */
- j = fftLen >> 2;
- ptr1 = &pSrc16[0];
- /* start of last stage process */
- /* Butterfly implementation */
- do
- {
- /* Read xa (real), ya(imag) input */
- xaya = read_q15x2_ia ((q15_t **) &ptr1);
- /* Read xb (real), yb(imag) input */
- xbyb = read_q15x2_ia ((q15_t **) &ptr1);
- /* Read xc (real), yc(imag) input */
- xcyc = read_q15x2_ia ((q15_t **) &ptr1);
- /* Read xd (real), yd(imag) input */
- xdyd = read_q15x2_ia ((q15_t **) &ptr1);
- /* R = packed((ya + yc), (xa + xc)) */
- R = __QADD16(xaya, xcyc);
- /* T = packed((yb + yd), (xb + xd)) */
- T = __QADD16(xbyb, xdyd);
- /* pointer updation for writing */
- ptr1 = ptr1 - 8U;
- /* xa' = xa + xb + xc + xd */
- /* ya' = ya + yb + yc + yd */
- write_q15x2_ia (&ptr1, __SHADD16(R, T));
- /* T = packed((yb + yd), (xb + xd)) */
- T = __QADD16(xbyb, xdyd);
- /* xc' = (xa-xb+xc-xd) */
- /* yc' = (ya-yb+yc-yd) */
- write_q15x2_ia (&ptr1, __SHSUB16(R, T));
- /* S = packed((ya - yc), (xa - xc)) */
- S = __QSUB16(xaya, xcyc);
- /* Read yd (real), xd(imag) input */
- /* T = packed( (yb - yd), (xb - xd)) */
- U = __QSUB16(xbyb, xdyd);
- #ifndef ARM_MATH_BIG_ENDIAN
- /* xb' = (xa+yb-xc-yd) */
- /* yb' = (ya-xb-yc+xd) */
- write_q15x2_ia (&ptr1, __SHSAX(S, U));
- /* xd' = (xa-yb-xc+yd) */
- /* yd' = (ya+xb-yc-xd) */
- write_q15x2_ia (&ptr1, __SHASX(S, U));
- #else
- /* xb' = (xa+yb-xc-yd) */
- /* yb' = (ya-xb-yc+xd) */
- write_q15x2_ia (&ptr1, __SHASX(S, U));
- /* xd' = (xa-yb-xc+yd) */
- /* yd' = (ya+xb-yc-xd) */
- write_q15x2_ia (&ptr1, __SHSAX(S, U));
- #endif /* #ifndef ARM_MATH_BIG_ENDIAN */
- } while (--j);
- /* end of last stage process */
- /* output is in 11.5(q5) format for the 1024 point */
- /* output is in 9.7(q7) format for the 256 point */
- /* output is in 7.9(q9) format for the 64 point */
- /* output is in 5.11(q11) format for the 16 point */
- #else /* #if defined (ARM_MATH_DSP) */
- q15_t R0, R1, S0, S1, T0, T1, U0, U1;
- q15_t Co1, Si1, Co2, Si2, Co3, Si3, out1, out2;
- uint32_t n1, n2, ic, i0, i1, i2, i3, j, k;
- /* Total process is divided into three stages */
- /* process first stage, middle stages, & last stage */
- /* Initializations for the first stage */
- n2 = fftLen;
- n1 = n2;
- /* n2 = fftLen/4 */
- n2 >>= 2U;
- /* Index for twiddle coefficient */
- ic = 0U;
- /* Index for input read and output write */
- i0 = 0U;
- j = n2;
- /* Input is in 1.15(q15) format */
- /* start of first stage process */
- do
- {
- /* Butterfly implementation */
- /* index calculation for the input as, */
- /* pSrc16[i0 + 0], pSrc16[i0 + fftLen/4], pSrc16[i0 + fftLen/2], pSrc16[i0 + 3fftLen/4] */
- i1 = i0 + n2;
- i2 = i1 + n2;
- i3 = i2 + n2;
- /* Reading i0, i0+fftLen/2 inputs */
- /* input is down scale by 4 to avoid overflow */
- /* Read ya (real), xa(imag) input */
- T0 = pSrc16[i0 * 2U] >> 2U;
- T1 = pSrc16[(i0 * 2U) + 1U] >> 2U;
- /* input is down scale by 4 to avoid overflow */
- /* Read yc (real), xc(imag) input */
- S0 = pSrc16[i2 * 2U] >> 2U;
- S1 = pSrc16[(i2 * 2U) + 1U] >> 2U;
- /* R0 = (ya + yc) */
- R0 = __SSAT(T0 + S0, 16U);
- /* R1 = (xa + xc) */
- R1 = __SSAT(T1 + S1, 16U);
- /* S0 = (ya - yc) */
- S0 = __SSAT(T0 - S0, 16);
- /* S1 = (xa - xc) */
- S1 = __SSAT(T1 - S1, 16);
- /* Reading i0+fftLen/4 , i0+3fftLen/4 inputs */
- /* input is down scale by 4 to avoid overflow */
- /* Read yb (real), xb(imag) input */
- T0 = pSrc16[i1 * 2U] >> 2U;
- T1 = pSrc16[(i1 * 2U) + 1U] >> 2U;
- /* input is down scale by 4 to avoid overflow */
- /* Read yd (real), xd(imag) input */
- U0 = pSrc16[i3 * 2U] >> 2U;
- U1 = pSrc16[(i3 * 2U) + 1] >> 2U;
- /* T0 = (yb + yd) */
- T0 = __SSAT(T0 + U0, 16U);
- /* T1 = (xb + xd) */
- T1 = __SSAT(T1 + U1, 16U);
- /* writing the butterfly processed i0 sample */
- /* ya' = ya + yb + yc + yd */
- /* xa' = xa + xb + xc + xd */
- pSrc16[i0 * 2U] = (R0 >> 1U) + (T0 >> 1U);
- pSrc16[(i0 * 2U) + 1U] = (R1 >> 1U) + (T1 >> 1U);
- /* R0 = (ya + yc) - (yb + yd) */
- /* R1 = (xa + xc) - (xb + xd) */
- R0 = __SSAT(R0 - T0, 16U);
- R1 = __SSAT(R1 - T1, 16U);
- /* co2 & si2 are read from Coefficient pointer */
- Co2 = pCoef16[2U * ic * 2U];
- Si2 = pCoef16[(2U * ic * 2U) + 1];
- /* xc' = (xa-xb+xc-xd)* co2 + (ya-yb+yc-yd)* (si2) */
- out1 = (q15_t) ((Co2 * R0 + Si2 * R1) >> 16U);
- /* yc' = (ya-yb+yc-yd)* co2 - (xa-xb+xc-xd)* (si2) */
- out2 = (q15_t) ((-Si2 * R0 + Co2 * R1) >> 16U);
- /* Reading i0+fftLen/4 */
- /* input is down scale by 4 to avoid overflow */
- /* T0 = yb, T1 = xb */
- T0 = pSrc16[i1 * 2U] >> 2;
- T1 = pSrc16[(i1 * 2U) + 1] >> 2;
- /* writing the butterfly processed i0 + fftLen/4 sample */
- /* writing output(xc', yc') in little endian format */
- pSrc16[i1 * 2U] = out1;
- pSrc16[(i1 * 2U) + 1] = out2;
- /* Butterfly calculations */
- /* input is down scale by 4 to avoid overflow */
- /* U0 = yd, U1 = xd */
- U0 = pSrc16[i3 * 2U] >> 2;
- U1 = pSrc16[(i3 * 2U) + 1] >> 2;
- /* T0 = yb-yd */
- T0 = __SSAT(T0 - U0, 16);
- /* T1 = xb-xd */
- T1 = __SSAT(T1 - U1, 16);
- /* R1 = (ya-yc) + (xb- xd), R0 = (xa-xc) - (yb-yd)) */
- R0 = (q15_t) __SSAT((q31_t) (S0 - T1), 16);
- R1 = (q15_t) __SSAT((q31_t) (S1 + T0), 16);
- /* S1 = (ya-yc) - (xb- xd), S0 = (xa-xc) + (yb-yd)) */
- S0 = (q15_t) __SSAT(((q31_t) S0 + T1), 16U);
- S1 = (q15_t) __SSAT(((q31_t) S1 - T0), 16U);
- /* co1 & si1 are read from Coefficient pointer */
- Co1 = pCoef16[ic * 2U];
- Si1 = pCoef16[(ic * 2U) + 1];
- /* Butterfly process for the i0+fftLen/2 sample */
- /* xb' = (xa+yb-xc-yd)* co1 + (ya-xb-yc+xd)* (si1) */
- out1 = (q15_t) ((Si1 * S1 + Co1 * S0) >> 16);
- /* yb' = (ya-xb-yc+xd)* co1 - (xa+yb-xc-yd)* (si1) */
- out2 = (q15_t) ((-Si1 * S0 + Co1 * S1) >> 16);
- /* writing output(xb', yb') in little endian format */
- pSrc16[i2 * 2U] = out1;
- pSrc16[(i2 * 2U) + 1] = out2;
- /* Co3 & si3 are read from Coefficient pointer */
- Co3 = pCoef16[3U * (ic * 2U)];
- Si3 = pCoef16[(3U * (ic * 2U)) + 1];
- /* Butterfly process for the i0+3fftLen/4 sample */
- /* xd' = (xa-yb-xc+yd)* Co3 + (ya+xb-yc-xd)* (si3) */
- out1 = (q15_t) ((Si3 * R1 + Co3 * R0) >> 16U);
- /* yd' = (ya+xb-yc-xd)* Co3 - (xa-yb-xc+yd)* (si3) */
- out2 = (q15_t) ((-Si3 * R0 + Co3 * R1) >> 16U);
- /* writing output(xd', yd') in little endian format */
- pSrc16[i3 * 2U] = out1;
- pSrc16[(i3 * 2U) + 1] = out2;
- /* Twiddle coefficients index modifier */
- ic = ic + twidCoefModifier;
- /* Updating input index */
- i0 = i0 + 1U;
- } while (--j);
- /* data is in 4.11(q11) format */
- /* end of first stage process */
- /* start of middle stage process */
- /* Twiddle coefficients index modifier */
- twidCoefModifier <<= 2U;
- /* Calculation of Middle stage */
- for (k = fftLen / 4U; k > 4U; k >>= 2U)
- {
- /* Initializations for the middle stage */
- n1 = n2;
- n2 >>= 2U;
- ic = 0U;
- for (j = 0U; j <= (n2 - 1U); j++)
- {
- /* index calculation for the coefficients */
- Co1 = pCoef16[ic * 2U];
- Si1 = pCoef16[(ic * 2U) + 1U];
- Co2 = pCoef16[2U * (ic * 2U)];
- Si2 = pCoef16[(2U * (ic * 2U)) + 1U];
- Co3 = pCoef16[3U * (ic * 2U)];
- Si3 = pCoef16[(3U * (ic * 2U)) + 1U];
- /* Twiddle coefficients index modifier */
- ic = ic + twidCoefModifier;
- /* Butterfly implementation */
- for (i0 = j; i0 < fftLen; i0 += n1)
- {
- /* index calculation for the input as, */
- /* pSrc16[i0 + 0], pSrc16[i0 + fftLen/4], pSrc16[i0 + fftLen/2], pSrc16[i0 + 3fftLen/4] */
- i1 = i0 + n2;
- i2 = i1 + n2;
- i3 = i2 + n2;
- /* Reading i0, i0+fftLen/2 inputs */
- /* Read ya (real), xa(imag) input */
- T0 = pSrc16[i0 * 2U];
- T1 = pSrc16[(i0 * 2U) + 1U];
- /* Read yc (real), xc(imag) input */
- S0 = pSrc16[i2 * 2U];
- S1 = pSrc16[(i2 * 2U) + 1U];
- /* R0 = (ya + yc), R1 = (xa + xc) */
- R0 = __SSAT(T0 + S0, 16);
- R1 = __SSAT(T1 + S1, 16);
- /* S0 = (ya - yc), S1 =(xa - xc) */
- S0 = __SSAT(T0 - S0, 16);
- S1 = __SSAT(T1 - S1, 16);
- /* Reading i0+fftLen/4 , i0+3fftLen/4 inputs */
- /* Read yb (real), xb(imag) input */
- T0 = pSrc16[i1 * 2U];
- T1 = pSrc16[(i1 * 2U) + 1U];
- /* Read yd (real), xd(imag) input */
- U0 = pSrc16[i3 * 2U];
- U1 = pSrc16[(i3 * 2U) + 1U];
- /* T0 = (yb + yd), T1 = (xb + xd) */
- T0 = __SSAT(T0 + U0, 16);
- T1 = __SSAT(T1 + U1, 16);
- /* writing the butterfly processed i0 sample */
- /* xa' = xa + xb + xc + xd */
- /* ya' = ya + yb + yc + yd */
- out1 = ((R0 >> 1U) + (T0 >> 1U)) >> 1U;
- out2 = ((R1 >> 1U) + (T1 >> 1U)) >> 1U;
- pSrc16[i0 * 2U] = out1;
- pSrc16[(2U * i0) + 1U] = out2;
- /* R0 = (ya + yc) - (yb + yd), R1 = (xa + xc) - (xb + xd) */
- R0 = (R0 >> 1U) - (T0 >> 1U);
- R1 = (R1 >> 1U) - (T1 >> 1U);
- /* (ya-yb+yc-yd)* (si2) + (xa-xb+xc-xd)* co2 */
- out1 = (q15_t) ((Co2 * R0 + Si2 * R1) >> 16U);
- /* (ya-yb+yc-yd)* co2 - (xa-xb+xc-xd)* (si2) */
- out2 = (q15_t) ((-Si2 * R0 + Co2 * R1) >> 16U);
- /* Reading i0+3fftLen/4 */
- /* Read yb (real), xb(imag) input */
- T0 = pSrc16[i1 * 2U];
- T1 = pSrc16[(i1 * 2U) + 1U];
- /* writing the butterfly processed i0 + fftLen/4 sample */
- /* xc' = (xa-xb+xc-xd)* co2 + (ya-yb+yc-yd)* (si2) */
- /* yc' = (ya-yb+yc-yd)* co2 - (xa-xb+xc-xd)* (si2) */
- pSrc16[i1 * 2U] = out1;
- pSrc16[(i1 * 2U) + 1U] = out2;
- /* Butterfly calculations */
- /* Read yd (real), xd(imag) input */
- U0 = pSrc16[i3 * 2U];
- U1 = pSrc16[(i3 * 2U) + 1U];
- /* T0 = yb-yd, T1 = xb-xd */
- T0 = __SSAT(T0 - U0, 16);
- T1 = __SSAT(T1 - U1, 16);
- /* R0 = (ya-yc) + (xb- xd), R1 = (xa-xc) - (yb-yd)) */
- R0 = (S0 >> 1U) - (T1 >> 1U);
- R1 = (S1 >> 1U) + (T0 >> 1U);
- /* S0 = (ya-yc) - (xb- xd), S1 = (xa-xc) + (yb-yd)) */
- S0 = (S0 >> 1U) + (T1 >> 1U);
- S1 = (S1 >> 1U) - (T0 >> 1U);
- /* Butterfly process for the i0+fftLen/2 sample */
- out1 = (q15_t) ((Co1 * S0 + Si1 * S1) >> 16U);
- out2 = (q15_t) ((-Si1 * S0 + Co1 * S1) >> 16U);
- /* xb' = (xa+yb-xc-yd)* co1 + (ya-xb-yc+xd)* (si1) */
- /* yb' = (ya-xb-yc+xd)* co1 - (xa+yb-xc-yd)* (si1) */
- pSrc16[i2 * 2U] = out1;
- pSrc16[(i2 * 2U) + 1U] = out2;
- /* Butterfly process for the i0+3fftLen/4 sample */
- out1 = (q15_t) ((Si3 * R1 + Co3 * R0) >> 16U);
- out2 = (q15_t) ((-Si3 * R0 + Co3 * R1) >> 16U);
- /* xd' = (xa-yb-xc+yd)* Co3 + (ya+xb-yc-xd)* (si3) */
- /* yd' = (ya+xb-yc-xd)* Co3 - (xa-yb-xc+yd)* (si3) */
- pSrc16[i3 * 2U] = out1;
- pSrc16[(i3 * 2U) + 1U] = out2;
- }
- }
- /* Twiddle coefficients index modifier */
- twidCoefModifier <<= 2U;
- }
- /* end of middle stage process */
- /* data is in 10.6(q6) format for the 1024 point */
- /* data is in 8.8(q8) format for the 256 point */
- /* data is in 6.10(q10) format for the 64 point */
- /* data is in 4.12(q12) format for the 16 point */
- /* Initializations for the last stage */
- n1 = n2;
- n2 >>= 2U;
- /* start of last stage process */
- /* Butterfly implementation */
- for (i0 = 0U; i0 <= (fftLen - n1); i0 += n1)
- {
- /* index calculation for the input as, */
- /* pSrc16[i0 + 0], pSrc16[i0 + fftLen/4], pSrc16[i0 + fftLen/2], pSrc16[i0 + 3fftLen/4] */
- i1 = i0 + n2;
- i2 = i1 + n2;
- i3 = i2 + n2;
- /* Reading i0, i0+fftLen/2 inputs */
- /* Read ya (real), xa(imag) input */
- T0 = pSrc16[i0 * 2U];
- T1 = pSrc16[(i0 * 2U) + 1U];
- /* Read yc (real), xc(imag) input */
- S0 = pSrc16[i2 * 2U];
- S1 = pSrc16[(i2 * 2U) + 1U];
- /* R0 = (ya + yc), R1 = (xa + xc) */
- R0 = __SSAT(T0 + S0, 16U);
- R1 = __SSAT(T1 + S1, 16U);
- /* S0 = (ya - yc), S1 = (xa - xc) */
- S0 = __SSAT(T0 - S0, 16U);
- S1 = __SSAT(T1 - S1, 16U);
- /* Reading i0+fftLen/4 , i0+3fftLen/4 inputs */
- /* Read yb (real), xb(imag) input */
- T0 = pSrc16[i1 * 2U];
- T1 = pSrc16[(i1 * 2U) + 1U];
- /* Read yd (real), xd(imag) input */
- U0 = pSrc16[i3 * 2U];
- U1 = pSrc16[(i3 * 2U) + 1U];
- /* T0 = (yb + yd), T1 = (xb + xd)) */
- T0 = __SSAT(T0 + U0, 16U);
- T1 = __SSAT(T1 + U1, 16U);
- /* writing the butterfly processed i0 sample */
- /* xa' = xa + xb + xc + xd */
- /* ya' = ya + yb + yc + yd */
- pSrc16[i0 * 2U] = (R0 >> 1U) + (T0 >> 1U);
- pSrc16[(i0 * 2U) + 1U] = (R1 >> 1U) + (T1 >> 1U);
- /* R0 = (ya + yc) - (yb + yd), R1 = (xa + xc) - (xb + xd) */
- R0 = (R0 >> 1U) - (T0 >> 1U);
- R1 = (R1 >> 1U) - (T1 >> 1U);
- /* Read yb (real), xb(imag) input */
- T0 = pSrc16[i1 * 2U];
- T1 = pSrc16[(i1 * 2U) + 1U];
- /* writing the butterfly processed i0 + fftLen/4 sample */
- /* xc' = (xa-xb+xc-xd) */
- /* yc' = (ya-yb+yc-yd) */
- pSrc16[i1 * 2U] = R0;
- pSrc16[(i1 * 2U) + 1U] = R1;
- /* Read yd (real), xd(imag) input */
- U0 = pSrc16[i3 * 2U];
- U1 = pSrc16[(i3 * 2U) + 1U];
- /* T0 = (yb - yd), T1 = (xb - xd) */
- T0 = __SSAT(T0 - U0, 16U);
- T1 = __SSAT(T1 - U1, 16U);
- /* writing the butterfly processed i0 + fftLen/2 sample */
- /* xb' = (xa+yb-xc-yd) */
- /* yb' = (ya-xb-yc+xd) */
- pSrc16[i2 * 2U] = (S0 >> 1U) + (T1 >> 1U);
- pSrc16[(i2 * 2U) + 1U] = (S1 >> 1U) - (T0 >> 1U);
- /* writing the butterfly processed i0 + 3fftLen/4 sample */
- /* xd' = (xa-yb-xc+yd) */
- /* yd' = (ya+xb-yc-xd) */
- pSrc16[i3 * 2U] = (S0 >> 1U) - (T1 >> 1U);
- pSrc16[(i3 * 2U) + 1U] = (S1 >> 1U) + (T0 >> 1U);
- }
- /* end of last stage process */
- /* output is in 11.5(q5) format for the 1024 point */
- /* output is in 9.7(q7) format for the 256 point */
- /* output is in 7.9(q9) format for the 64 point */
- /* output is in 5.11(q11) format for the 16 point */
- #endif /* #if defined (ARM_MATH_DSP) */
- }
- /**
- @brief Core function for the Q15 CIFFT butterfly process.
- @param[in,out] pSrc16 points to the in-place buffer of Q15 data type
- @param[in] fftLen length of the FFT
- @param[in] pCoef16 points to twiddle coefficient buffer
- @param[in] twidCoefModifier twiddle coefficient modifier that supports different size FFTs with the same twiddle factor table.
- @return none
- */
- /*
- * Radix-4 IFFT algorithm used is :
- *
- * CIFFT uses same twiddle coefficients as CFFT function
- * x[k] = x[n] + (j)k * x[n + fftLen/4] + (-1)k * x[n+fftLen/2] + (-j)k * x[n+3*fftLen/4]
- *
- *
- * IFFT is implemented with following changes in equations from FFT
- *
- * Input real and imaginary data:
- * x(n) = xa + j * ya
- * x(n+N/4 ) = xb + j * yb
- * x(n+N/2 ) = xc + j * yc
- * x(n+3N 4) = xd + j * yd
- *
- *
- * Output real and imaginary data:
- * x(4r) = xa'+ j * ya'
- * x(4r+1) = xb'+ j * yb'
- * x(4r+2) = xc'+ j * yc'
- * x(4r+3) = xd'+ j * yd'
- *
- *
- * Twiddle factors for radix-4 IFFT:
- * Wn = co1 + j * (si1)
- * W2n = co2 + j * (si2)
- * W3n = co3 + j * (si3)
-
- * The real and imaginary output values for the radix-4 butterfly are
- * xa' = xa + xb + xc + xd
- * ya' = ya + yb + yc + yd
- * xb' = (xa-yb-xc+yd)* co1 - (ya+xb-yc-xd)* (si1)
- * yb' = (ya+xb-yc-xd)* co1 + (xa-yb-xc+yd)* (si1)
- * xc' = (xa-xb+xc-xd)* co2 - (ya-yb+yc-yd)* (si2)
- * yc' = (ya-yb+yc-yd)* co2 + (xa-xb+xc-xd)* (si2)
- * xd' = (xa+yb-xc-yd)* co3 - (ya-xb-yc+xd)* (si3)
- * yd' = (ya-xb-yc+xd)* co3 + (xa+yb-xc-yd)* (si3)
- *
- */
- void arm_radix4_butterfly_inverse_q15(
- q15_t * pSrc16,
- uint32_t fftLen,
- const q15_t * pCoef16,
- uint32_t twidCoefModifier)
- {
- #if defined (ARM_MATH_DSP)
- q31_t R, S, T, U;
- q31_t C1, C2, C3, out1, out2;
- uint32_t n1, n2, ic, i0, j, k;
-
- q15_t *ptr1;
- q15_t *pSi0;
- q15_t *pSi1;
- q15_t *pSi2;
- q15_t *pSi3;
-
- q31_t xaya, xbyb, xcyc, xdyd;
- /* Total process is divided into three stages */
- /* process first stage, middle stages, & last stage */
- /* Initializations for the first stage */
- n2 = fftLen;
- n1 = n2;
- /* n2 = fftLen/4 */
- n2 >>= 2U;
- /* Index for twiddle coefficient */
- ic = 0U;
- /* Index for input read and output write */
- j = n2;
- pSi0 = pSrc16;
- pSi1 = pSi0 + 2 * n2;
- pSi2 = pSi1 + 2 * n2;
- pSi3 = pSi2 + 2 * n2;
- /* Input is in 1.15(q15) format */
- /* start of first stage process */
- do
- {
- /* Butterfly implementation */
- /* Reading i0, i0+fftLen/2 inputs */
- /* Read ya (real), xa(imag) input */
- T = read_q15x2 (pSi0);
- T = __SHADD16(T, 0);
- T = __SHADD16(T, 0);
- /* Read yc (real), xc(imag) input */
- S = read_q15x2 (pSi2);
- S = __SHADD16(S, 0);
- S = __SHADD16(S, 0);
- /* R = packed((ya + yc), (xa + xc) ) */
- R = __QADD16(T, S);
- /* S = packed((ya - yc), (xa - xc) ) */
- S = __QSUB16(T, S);
- /* Reading i0+fftLen/4 , i0+3fftLen/4 inputs */
- /* Read yb (real), xb(imag) input */
- T = read_q15x2 (pSi1);
- T = __SHADD16(T, 0);
- T = __SHADD16(T, 0);
- /* Read yd (real), xd(imag) input */
- U = read_q15x2 (pSi3);
- U = __SHADD16(U, 0);
- U = __SHADD16(U, 0);
- /* T = packed((yb + yd), (xb + xd) ) */
- T = __QADD16(T, U);
- /* writing the butterfly processed i0 sample */
- /* xa' = xa + xb + xc + xd */
- /* ya' = ya + yb + yc + yd */
- write_q15x2_ia (&pSi0, __SHADD16(R, T));
- /* R = packed((ya + yc) - (yb + yd), (xa + xc)- (xb + xd)) */
- R = __QSUB16(R, T);
- /* co2 & si2 are read from SIMD Coefficient pointer */
- C2 = read_q15x2 ((q15_t *) pCoef16 + (4U * ic));
- #ifndef ARM_MATH_BIG_ENDIAN
- /* xc' = (xa-xb+xc-xd)* co2 + (ya-yb+yc-yd)* (si2) */
- out1 = __SMUSD(C2, R) >> 16U;
- /* yc' = (ya-yb+yc-yd)* co2 - (xa-xb+xc-xd)* (si2) */
- out2 = __SMUADX(C2, R);
- #else
- /* xc' = (ya-yb+yc-yd)* co2 - (xa-xb+xc-xd)* (si2) */
- out1 = __SMUADX(C2, R) >> 16U;
- /* yc' = (xa-xb+xc-xd)* co2 + (ya-yb+yc-yd)* (si2) */
- out2 = __SMUSD(__QSUB16(0, C2), R);
- #endif /* #ifndef ARM_MATH_BIG_ENDIAN */
- /* Reading i0+fftLen/4 */
- /* T = packed(yb, xb) */
- T = read_q15x2 (pSi1);
- T = __SHADD16(T, 0);
- T = __SHADD16(T, 0);
- /* writing the butterfly processed i0 + fftLen/4 sample */
- /* writing output(xc', yc') in little endian format */
- write_q15x2_ia (&pSi1, (q31_t) ((out2) & 0xFFFF0000) | (out1 & 0x0000FFFF));
- /* Butterfly calculations */
- /* U = packed(yd, xd) */
- U = read_q15x2 (pSi3);
- U = __SHADD16(U, 0);
- U = __SHADD16(U, 0);
- /* T = packed(yb-yd, xb-xd) */
- T = __QSUB16(T, U);
- #ifndef ARM_MATH_BIG_ENDIAN
- /* R = packed((ya-yc) + (xb- xd) , (xa-xc) - (yb-yd)) */
- R = __QSAX(S, T);
- /* S = packed((ya-yc) + (xb- xd), (xa-xc) - (yb-yd)) */
- S = __QASX(S, T);
- #else
- /* R = packed((ya-yc) + (xb- xd) , (xa-xc) - (yb-yd)) */
- R = __QASX(S, T);
- /* S = packed((ya-yc) - (xb- xd), (xa-xc) + (yb-yd)) */
- S = __QSAX(S, T);
- #endif /* #ifndef ARM_MATH_BIG_ENDIAN */
- /* co1 & si1 are read from SIMD Coefficient pointer */
- C1 = read_q15x2 ((q15_t *) pCoef16 + (2U * ic));
- /* Butterfly process for the i0+fftLen/2 sample */
- #ifndef ARM_MATH_BIG_ENDIAN
- /* xb' = (xa+yb-xc-yd)* co1 + (ya-xb-yc+xd)* (si1) */
- out1 = __SMUSD(C1, S) >> 16U;
- /* yb' = (ya-xb-yc+xd)* co1 - (xa+yb-xc-yd)* (si1) */
- out2 = __SMUADX(C1, S);
- #else
- /* xb' = (ya-xb-yc+xd)* co1 - (xa+yb-xc-yd)* (si1) */
- out1 = __SMUADX(C1, S) >> 16U;
- /* yb' = (xa+yb-xc-yd)* co1 + (ya-xb-yc+xd)* (si1) */
- out2 = __SMUSD(__QSUB16(0, C1), S);
- #endif /* #ifndef ARM_MATH_BIG_ENDIAN */
- /* writing output(xb', yb') in little endian format */
- write_q15x2_ia (&pSi2, ((out2) & 0xFFFF0000) | ((out1) & 0x0000FFFF));
- /* co3 & si3 are read from SIMD Coefficient pointer */
- C3 = read_q15x2 ((q15_t *) pCoef16 + (6U * ic));
- /* Butterfly process for the i0+3fftLen/4 sample */
- #ifndef ARM_MATH_BIG_ENDIAN
- /* xd' = (xa-yb-xc+yd)* co3 + (ya+xb-yc-xd)* (si3) */
- out1 = __SMUSD(C3, R) >> 16U;
- /* yd' = (ya+xb-yc-xd)* co3 - (xa-yb-xc+yd)* (si3) */
- out2 = __SMUADX(C3, R);
- #else
- /* xd' = (ya+xb-yc-xd)* co3 - (xa-yb-xc+yd)* (si3) */
- out1 = __SMUADX(C3, R) >> 16U;
- /* yd' = (xa-yb-xc+yd)* co3 + (ya+xb-yc-xd)* (si3) */
- out2 = __SMUSD(__QSUB16(0, C3), R);
- #endif /* #ifndef ARM_MATH_BIG_ENDIAN */
- /* writing output(xd', yd') in little endian format */
- write_q15x2_ia (&pSi3, ((out2) & 0xFFFF0000) | (out1 & 0x0000FFFF));
- /* Twiddle coefficients index modifier */
- ic = ic + twidCoefModifier;
- } while (--j);
- /* data is in 4.11(q11) format */
- /* end of first stage process */
- /* start of middle stage process */
- /* Twiddle coefficients index modifier */
- twidCoefModifier <<= 2U;
- /* Calculation of Middle stage */
- for (k = fftLen / 4U; k > 4U; k >>= 2U)
- {
- /* Initializations for the middle stage */
- n1 = n2;
- n2 >>= 2U;
- ic = 0U;
- for (j = 0U; j <= (n2 - 1U); j++)
- {
- /* index calculation for the coefficients */
- C1 = read_q15x2 ((q15_t *) pCoef16 + (2U * ic));
- C2 = read_q15x2 ((q15_t *) pCoef16 + (4U * ic));
- C3 = read_q15x2 ((q15_t *) pCoef16 + (6U * ic));
- /* Twiddle coefficients index modifier */
- ic = ic + twidCoefModifier;
- pSi0 = pSrc16 + 2 * j;
- pSi1 = pSi0 + 2 * n2;
- pSi2 = pSi1 + 2 * n2;
- pSi3 = pSi2 + 2 * n2;
- /* Butterfly implementation */
- for (i0 = j; i0 < fftLen; i0 += n1)
- {
- /* Reading i0, i0+fftLen/2 inputs */
- /* Read ya (real), xa(imag) input */
- T = read_q15x2 (pSi0);
- /* Read yc (real), xc(imag) input */
- S = read_q15x2 (pSi2);
- /* R = packed( (ya + yc), (xa + xc)) */
- R = __QADD16(T, S);
- /* S = packed((ya - yc), (xa - xc)) */
- S = __QSUB16(T, S);
- /* Reading i0+fftLen/4 , i0+3fftLen/4 inputs */
- /* Read yb (real), xb(imag) input */
- T = read_q15x2 (pSi1);
- /* Read yd (real), xd(imag) input */
- U = read_q15x2 (pSi3);
- /* T = packed( (yb + yd), (xb + xd)) */
- T = __QADD16(T, U);
- /* writing the butterfly processed i0 sample */
- /* xa' = xa + xb + xc + xd */
- /* ya' = ya + yb + yc + yd */
- out1 = __SHADD16(R, T);
- out1 = __SHADD16(out1, 0);
- write_q15x2 (pSi0, out1);
- pSi0 += 2 * n1;
- /* R = packed( (ya + yc) - (yb + yd), (xa + xc) - (xb + xd)) */
- R = __SHSUB16(R, T);
- #ifndef ARM_MATH_BIG_ENDIAN
- /* (ya-yb+yc-yd)* (si2) + (xa-xb+xc-xd)* co2 */
- out1 = __SMUSD(C2, R) >> 16U;
- /* (ya-yb+yc-yd)* co2 - (xa-xb+xc-xd)* (si2) */
- out2 = __SMUADX(C2, R);
- #else
- /* (ya-yb+yc-yd)* co2 - (xa-xb+xc-xd)* (si2) */
- out1 = __SMUADX(R, C2) >> 16U;
- /* (ya-yb+yc-yd)* (si2) + (xa-xb+xc-xd)* co2 */
- out2 = __SMUSD(__QSUB16(0, C2), R);
- #endif /* #ifndef ARM_MATH_BIG_ENDIAN */
- /* Reading i0+3fftLen/4 */
- /* Read yb (real), xb(imag) input */
- T = read_q15x2 (pSi1);
- /* writing the butterfly processed i0 + fftLen/4 sample */
- /* xc' = (xa-xb+xc-xd)* co2 + (ya-yb+yc-yd)* (si2) */
- /* yc' = (ya-yb+yc-yd)* co2 - (xa-xb+xc-xd)* (si2) */
- write_q15x2 (pSi1, ((out2) & 0xFFFF0000) | (out1 & 0x0000FFFF));
- pSi1 += 2 * n1;
- /* Butterfly calculations */
- /* Read yd (real), xd(imag) input */
- U = read_q15x2 (pSi3);
- /* T = packed(yb-yd, xb-xd) */
- T = __QSUB16(T, U);
- #ifndef ARM_MATH_BIG_ENDIAN
- /* R = packed((ya-yc) + (xb- xd) , (xa-xc) - (yb-yd)) */
- R = __SHSAX(S, T);
- /* S = packed((ya-yc) - (xb- xd), (xa-xc) + (yb-yd)) */
- S = __SHASX(S, T);
- /* Butterfly process for the i0+fftLen/2 sample */
- out1 = __SMUSD(C1, S) >> 16U;
- out2 = __SMUADX(C1, S);
- #else
- /* R = packed((ya-yc) + (xb- xd) , (xa-xc) - (yb-yd)) */
- R = __SHASX(S, T);
- /* S = packed((ya-yc) - (xb- xd), (xa-xc) + (yb-yd)) */
- S = __SHSAX(S, T);
- /* Butterfly process for the i0+fftLen/2 sample */
- out1 = __SMUADX(S, C1) >> 16U;
- out2 = __SMUSD(__QSUB16(0, C1), S);
- #endif /* #ifndef ARM_MATH_BIG_ENDIAN */
- /* xb' = (xa+yb-xc-yd)* co1 + (ya-xb-yc+xd)* (si1) */
- /* yb' = (ya-xb-yc+xd)* co1 - (xa+yb-xc-yd)* (si1) */
- write_q15x2 (pSi2, ((out2) & 0xFFFF0000) | (out1 & 0x0000FFFF));
- pSi2 += 2 * n1;
- /* Butterfly process for the i0+3fftLen/4 sample */
- #ifndef ARM_MATH_BIG_ENDIAN
- out1 = __SMUSD(C3, R) >> 16U;
- out2 = __SMUADX(C3, R);
- #else
- out1 = __SMUADX(C3, R) >> 16U;
- out2 = __SMUSD(__QSUB16(0, C3), R);
- #endif /* #ifndef ARM_MATH_BIG_ENDIAN */
- /* xd' = (xa-yb-xc+yd)* co3 + (ya+xb-yc-xd)* (si3) */
- /* yd' = (ya+xb-yc-xd)* co3 - (xa-yb-xc+yd)* (si3) */
- write_q15x2 (pSi3, ((out2) & 0xFFFF0000) | (out1 & 0x0000FFFF));
- pSi3 += 2 * n1;
- }
- }
- /* Twiddle coefficients index modifier */
- twidCoefModifier <<= 2U;
- }
- /* end of middle stage process */
- /* data is in 10.6(q6) format for the 1024 point */
- /* data is in 8.8(q8) format for the 256 point */
- /* data is in 6.10(q10) format for the 64 point */
- /* data is in 4.12(q12) format for the 16 point */
- /* Initializations for the last stage */
- j = fftLen >> 2;
- ptr1 = &pSrc16[0];
- /* start of last stage process */
- /* Butterfly implementation */
- do
- {
- /* Read xa (real), ya(imag) input */
- xaya = read_q15x2_ia ((q15_t **) &ptr1);
- /* Read xb (real), yb(imag) input */
- xbyb = read_q15x2_ia ((q15_t **) &ptr1);
- /* Read xc (real), yc(imag) input */
- xcyc = read_q15x2_ia ((q15_t **) &ptr1);
- /* Read xd (real), yd(imag) input */
- xdyd = read_q15x2_ia ((q15_t **) &ptr1);
- /* R = packed((ya + yc), (xa + xc)) */
- R = __QADD16(xaya, xcyc);
- /* T = packed((yb + yd), (xb + xd)) */
- T = __QADD16(xbyb, xdyd);
- /* pointer updation for writing */
- ptr1 = ptr1 - 8U;
- /* xa' = xa + xb + xc + xd */
- /* ya' = ya + yb + yc + yd */
- write_q15x2_ia (&ptr1, __SHADD16(R, T));
- /* T = packed((yb + yd), (xb + xd)) */
- T = __QADD16(xbyb, xdyd);
- /* xc' = (xa-xb+xc-xd) */
- /* yc' = (ya-yb+yc-yd) */
- write_q15x2_ia (&ptr1, __SHSUB16(R, T));
- /* S = packed((ya - yc), (xa - xc)) */
- S = __QSUB16(xaya, xcyc);
- /* Read yd (real), xd(imag) input */
- /* T = packed( (yb - yd), (xb - xd)) */
- U = __QSUB16(xbyb, xdyd);
- #ifndef ARM_MATH_BIG_ENDIAN
- /* xb' = (xa+yb-xc-yd) */
- /* yb' = (ya-xb-yc+xd) */
- write_q15x2_ia (&ptr1, __SHASX(S, U));
- /* xd' = (xa-yb-xc+yd) */
- /* yd' = (ya+xb-yc-xd) */
- write_q15x2_ia (&ptr1, __SHSAX(S, U));
- #else
- /* xb' = (xa+yb-xc-yd) */
- /* yb' = (ya-xb-yc+xd) */
- write_q15x2_ia (&ptr1, __SHSAX(S, U));
- /* xd' = (xa-yb-xc+yd) */
- /* yd' = (ya+xb-yc-xd) */
- write_q15x2_ia (&ptr1, __SHASX(S, U));
- #endif /* #ifndef ARM_MATH_BIG_ENDIAN */
- } while (--j);
- /* end of last stage process */
- /* output is in 11.5(q5) format for the 1024 point */
- /* output is in 9.7(q7) format for the 256 point */
- /* output is in 7.9(q9) format for the 64 point */
- /* output is in 5.11(q11) format for the 16 point */
- #else /* arm_radix4_butterfly_inverse_q15 */
- q15_t R0, R1, S0, S1, T0, T1, U0, U1;
- q15_t Co1, Si1, Co2, Si2, Co3, Si3, out1, out2;
- uint32_t n1, n2, ic, i0, i1, i2, i3, j, k;
- /* Total process is divided into three stages */
- /* process first stage, middle stages, & last stage */
- /* Initializations for the first stage */
- n2 = fftLen;
- n1 = n2;
- /* n2 = fftLen/4 */
- n2 >>= 2U;
- /* Index for twiddle coefficient */
- ic = 0U;
- /* Index for input read and output write */
- i0 = 0U;
- j = n2;
- /* Input is in 1.15(q15) format */
- /* Start of first stage process */
- do
- {
- /* Butterfly implementation */
- /* index calculation for the input as, */
- /* pSrc16[i0 + 0], pSrc16[i0 + fftLen/4], pSrc16[i0 + fftLen/2], pSrc16[i0 + 3fftLen/4] */
- i1 = i0 + n2;
- i2 = i1 + n2;
- i3 = i2 + n2;
- /* Reading i0, i0+fftLen/2 inputs */
- /* input is down scale by 4 to avoid overflow */
- /* Read ya (real), xa(imag) input */
- T0 = pSrc16[i0 * 2U] >> 2U;
- T1 = pSrc16[(i0 * 2U) + 1U] >> 2U;
- /* input is down scale by 4 to avoid overflow */
- /* Read yc (real), xc(imag) input */
- S0 = pSrc16[i2 * 2U] >> 2U;
- S1 = pSrc16[(i2 * 2U) + 1U] >> 2U;
- /* R0 = (ya + yc), R1 = (xa + xc) */
- R0 = __SSAT(T0 + S0, 16U);
- R1 = __SSAT(T1 + S1, 16U);
- /* S0 = (ya - yc), S1 = (xa - xc) */
- S0 = __SSAT(T0 - S0, 16U);
- S1 = __SSAT(T1 - S1, 16U);
- /* Reading i0+fftLen/4 , i0+3fftLen/4 inputs */
- /* input is down scale by 4 to avoid overflow */
- /* Read yb (real), xb(imag) input */
- T0 = pSrc16[i1 * 2U] >> 2U;
- T1 = pSrc16[(i1 * 2U) + 1U] >> 2U;
- /* Read yd (real), xd(imag) input */
- /* input is down scale by 4 to avoid overflow */
- U0 = pSrc16[i3 * 2U] >> 2U;
- U1 = pSrc16[(i3 * 2U) + 1U] >> 2U;
- /* T0 = (yb + yd), T1 = (xb + xd) */
- T0 = __SSAT(T0 + U0, 16U);
- T1 = __SSAT(T1 + U1, 16U);
- /* writing the butterfly processed i0 sample */
- /* xa' = xa + xb + xc + xd */
- /* ya' = ya + yb + yc + yd */
- pSrc16[i0 * 2U] = (R0 >> 1U) + (T0 >> 1U);
- pSrc16[(i0 * 2U) + 1U] = (R1 >> 1U) + (T1 >> 1U);
- /* R0 = (ya + yc) - (yb + yd), R1 = (xa + xc)- (xb + xd) */
- R0 = __SSAT(R0 - T0, 16U);
- R1 = __SSAT(R1 - T1, 16U);
- /* co2 & si2 are read from Coefficient pointer */
- Co2 = pCoef16[2U * ic * 2U];
- Si2 = pCoef16[(2U * ic * 2U) + 1U];
- /* xc' = (xa-xb+xc-xd)* co2 - (ya-yb+yc-yd)* (si2) */
- out1 = (q15_t) ((Co2 * R0 - Si2 * R1) >> 16U);
- /* yc' = (ya-yb+yc-yd)* co2 + (xa-xb+xc-xd)* (si2) */
- out2 = (q15_t) ((Si2 * R0 + Co2 * R1) >> 16U);
- /* Reading i0+fftLen/4 */
- /* input is down scale by 4 to avoid overflow */
- /* T0 = yb, T1 = xb */
- T0 = pSrc16[i1 * 2U] >> 2U;
- T1 = pSrc16[(i1 * 2U) + 1U] >> 2U;
- /* writing the butterfly processed i0 + fftLen/4 sample */
- /* writing output(xc', yc') in little endian format */
- pSrc16[i1 * 2U] = out1;
- pSrc16[(i1 * 2U) + 1U] = out2;
- /* Butterfly calculations */
- /* input is down scale by 4 to avoid overflow */
- /* U0 = yd, U1 = xd) */
- U0 = pSrc16[i3 * 2U] >> 2U;
- U1 = pSrc16[(i3 * 2U) + 1U] >> 2U;
- /* T0 = yb-yd, T1 = xb-xd) */
- T0 = __SSAT(T0 - U0, 16U);
- T1 = __SSAT(T1 - U1, 16U);
- /* R0 = (ya-yc) - (xb- xd) , R1 = (xa-xc) + (yb-yd) */
- R0 = (q15_t) __SSAT((q31_t) (S0 + T1), 16);
- R1 = (q15_t) __SSAT((q31_t) (S1 - T0), 16);
- /* S = (ya-yc) + (xb- xd), S1 = (xa-xc) - (yb-yd) */
- S0 = (q15_t) __SSAT((q31_t) (S0 - T1), 16);
- S1 = (q15_t) __SSAT((q31_t) (S1 + T0), 16);
- /* co1 & si1 are read from Coefficient pointer */
- Co1 = pCoef16[ic * 2U];
- Si1 = pCoef16[(ic * 2U) + 1U];
- /* Butterfly process for the i0+fftLen/2 sample */
- /* xb' = (xa-yb-xc+yd)* co1 - (ya+xb-yc-xd)* (si1) */
- out1 = (q15_t) ((Co1 * S0 - Si1 * S1) >> 16U);
- /* yb' = (ya+xb-yc-xd)* co1 + (xa-yb-xc+yd)* (si1) */
- out2 = (q15_t) ((Si1 * S0 + Co1 * S1) >> 16U);
- /* writing output(xb', yb') in little endian format */
- pSrc16[i2 * 2U] = out1;
- pSrc16[(i2 * 2U) + 1U] = out2;
- /* Co3 & si3 are read from Coefficient pointer */
- Co3 = pCoef16[3U * ic * 2U];
- Si3 = pCoef16[(3U * ic * 2U) + 1U];
- /* Butterfly process for the i0+3fftLen/4 sample */
- /* xd' = (xa+yb-xc-yd)* Co3 - (ya-xb-yc+xd)* (si3) */
- out1 = (q15_t) ((Co3 * R0 - Si3 * R1) >> 16U);
- /* yd' = (ya-xb-yc+xd)* Co3 + (xa+yb-xc-yd)* (si3) */
- out2 = (q15_t) ((Si3 * R0 + Co3 * R1) >> 16U);
- /* writing output(xd', yd') in little endian format */
- pSrc16[i3 * 2U] = out1;
- pSrc16[(i3 * 2U) + 1U] = out2;
- /* Twiddle coefficients index modifier */
- ic = ic + twidCoefModifier;
- /* Updating input index */
- i0 = i0 + 1U;
- } while (--j);
- /* End of first stage process */
- /* data is in 4.11(q11) format */
- /* Start of Middle stage process */
- /* Twiddle coefficients index modifier */
- twidCoefModifier <<= 2U;
- /* Calculation of Middle stage */
- for (k = fftLen / 4U; k > 4U; k >>= 2U)
- {
- /* Initializations for the middle stage */
- n1 = n2;
- n2 >>= 2U;
- ic = 0U;
- for (j = 0U; j <= (n2 - 1U); j++)
- {
- /* index calculation for the coefficients */
- Co1 = pCoef16[ic * 2U];
- Si1 = pCoef16[(ic * 2U) + 1U];
- Co2 = pCoef16[2U * ic * 2U];
- Si2 = pCoef16[2U * ic * 2U + 1U];
- Co3 = pCoef16[3U * ic * 2U];
- Si3 = pCoef16[(3U * ic * 2U) + 1U];
- /* Twiddle coefficients index modifier */
- ic = ic + twidCoefModifier;
- /* Butterfly implementation */
- for (i0 = j; i0 < fftLen; i0 += n1)
- {
- /* index calculation for the input as, */
- /* pSrc16[i0 + 0], pSrc16[i0 + fftLen/4], pSrc16[i0 + fftLen/2], pSrc16[i0 + 3fftLen/4] */
- i1 = i0 + n2;
- i2 = i1 + n2;
- i3 = i2 + n2;
- /* Reading i0, i0+fftLen/2 inputs */
- /* Read ya (real), xa(imag) input */
- T0 = pSrc16[i0 * 2U];
- T1 = pSrc16[(i0 * 2U) + 1U];
- /* Read yc (real), xc(imag) input */
- S0 = pSrc16[i2 * 2U];
- S1 = pSrc16[(i2 * 2U) + 1U];
- /* R0 = (ya + yc), R1 = (xa + xc) */
- R0 = __SSAT(T0 + S0, 16U);
- R1 = __SSAT(T1 + S1, 16U);
- /* S0 = (ya - yc), S1 = (xa - xc) */
- S0 = __SSAT(T0 - S0, 16U);
- S1 = __SSAT(T1 - S1, 16U);
- /* Reading i0+fftLen/4 , i0+3fftLen/4 inputs */
- /* Read yb (real), xb(imag) input */
- T0 = pSrc16[i1 * 2U];
- T1 = pSrc16[(i1 * 2U) + 1U];
- /* Read yd (real), xd(imag) input */
- U0 = pSrc16[i3 * 2U];
- U1 = pSrc16[(i3 * 2U) + 1U];
- /* T0 = (yb + yd), T1 = (xb + xd) */
- T0 = __SSAT(T0 + U0, 16U);
- T1 = __SSAT(T1 + U1, 16U);
- /* writing the butterfly processed i0 sample */
- /* xa' = xa + xb + xc + xd */
- /* ya' = ya + yb + yc + yd */
- pSrc16[i0 * 2U] = ((R0 >> 1U) + (T0 >> 1U)) >> 1U;
- pSrc16[(i0 * 2U) + 1U] = ((R1 >> 1U) + (T1 >> 1U)) >> 1U;
- /* R0 = (ya + yc) - (yb + yd), R1 = (xa + xc) - (xb + xd) */
- R0 = (R0 >> 1U) - (T0 >> 1U);
- R1 = (R1 >> 1U) - (T1 >> 1U);
- /* (ya-yb+yc-yd)* (si2) - (xa-xb+xc-xd)* co2 */
- out1 = (q15_t) ((Co2 * R0 - Si2 * R1) >> 16);
- /* (ya-yb+yc-yd)* co2 + (xa-xb+xc-xd)* (si2) */
- out2 = (q15_t) ((Si2 * R0 + Co2 * R1) >> 16);
- /* Reading i0+3fftLen/4 */
- /* Read yb (real), xb(imag) input */
- T0 = pSrc16[i1 * 2U];
- T1 = pSrc16[(i1 * 2U) + 1U];
- /* writing the butterfly processed i0 + fftLen/4 sample */
- /* xc' = (xa-xb+xc-xd)* co2 - (ya-yb+yc-yd)* (si2) */
- /* yc' = (ya-yb+yc-yd)* co2 + (xa-xb+xc-xd)* (si2) */
- pSrc16[i1 * 2U] = out1;
- pSrc16[(i1 * 2U) + 1U] = out2;
- /* Butterfly calculations */
- /* Read yd (real), xd(imag) input */
- U0 = pSrc16[i3 * 2U];
- U1 = pSrc16[(i3 * 2U) + 1U];
- /* T0 = yb-yd, T1 = xb-xd) */
- T0 = __SSAT(T0 - U0, 16U);
- T1 = __SSAT(T1 - U1, 16U);
- /* R0 = (ya-yc) - (xb- xd) , R1 = (xa-xc) + (yb-yd) */
- R0 = (S0 >> 1U) + (T1 >> 1U);
- R1 = (S1 >> 1U) - (T0 >> 1U);
- /* S1 = (ya-yc) + (xb- xd), S1 = (xa-xc) - (yb-yd) */
- S0 = (S0 >> 1U) - (T1 >> 1U);
- S1 = (S1 >> 1U) + (T0 >> 1U);
- /* Butterfly process for the i0+fftLen/2 sample */
- out1 = (q15_t) ((Co1 * S0 - Si1 * S1) >> 16U);
- out2 = (q15_t) ((Si1 * S0 + Co1 * S1) >> 16U);
- /* xb' = (xa-yb-xc+yd)* co1 - (ya+xb-yc-xd)* (si1) */
- /* yb' = (ya+xb-yc-xd)* co1 + (xa-yb-xc+yd)* (si1) */
- pSrc16[i2 * 2U] = out1;
- pSrc16[(i2 * 2U) + 1U] = out2;
- /* Butterfly process for the i0+3fftLen/4 sample */
- out1 = (q15_t) ((Co3 * R0 - Si3 * R1) >> 16U);
- out2 = (q15_t) ((Si3 * R0 + Co3 * R1) >> 16U);
- /* xd' = (xa+yb-xc-yd)* Co3 - (ya-xb-yc+xd)* (si3) */
- /* yd' = (ya-xb-yc+xd)* Co3 + (xa+yb-xc-yd)* (si3) */
- pSrc16[i3 * 2U] = out1;
- pSrc16[(i3 * 2U) + 1U] = out2;
- }
- }
- /* Twiddle coefficients index modifier */
- twidCoefModifier <<= 2U;
- }
- /* End of Middle stages process */
- /* data is in 10.6(q6) format for the 1024 point */
- /* data is in 8.8(q8) format for the 256 point */
- /* data is in 6.10(q10) format for the 64 point */
- /* data is in 4.12(q12) format for the 16 point */
- /* start of last stage process */
- /* Initializations for the last stage */
- n1 = n2;
- n2 >>= 2U;
- /* Butterfly implementation */
- for (i0 = 0U; i0 <= (fftLen - n1); i0 += n1)
- {
- /* index calculation for the input as, */
- /* pSrc16[i0 + 0], pSrc16[i0 + fftLen/4], pSrc16[i0 + fftLen/2], pSrc16[i0 + 3fftLen/4] */
- i1 = i0 + n2;
- i2 = i1 + n2;
- i3 = i2 + n2;
- /* Reading i0, i0+fftLen/2 inputs */
- /* Read ya (real), xa(imag) input */
- T0 = pSrc16[i0 * 2U];
- T1 = pSrc16[(i0 * 2U) + 1U];
- /* Read yc (real), xc(imag) input */
- S0 = pSrc16[i2 * 2U];
- S1 = pSrc16[(i2 * 2U) + 1U];
- /* R0 = (ya + yc), R1 = (xa + xc) */
- R0 = __SSAT(T0 + S0, 16U);
- R1 = __SSAT(T1 + S1, 16U);
- /* S0 = (ya - yc), S1 = (xa - xc) */
- S0 = __SSAT(T0 - S0, 16U);
- S1 = __SSAT(T1 - S1, 16U);
- /* Reading i0+fftLen/4 , i0+3fftLen/4 inputs */
- /* Read yb (real), xb(imag) input */
- T0 = pSrc16[i1 * 2U];
- T1 = pSrc16[(i1 * 2U) + 1U];
- /* Read yd (real), xd(imag) input */
- U0 = pSrc16[i3 * 2U];
- U1 = pSrc16[(i3 * 2U) + 1U];
- /* T0 = (yb + yd), T1 = (xb + xd) */
- T0 = __SSAT(T0 + U0, 16U);
- T1 = __SSAT(T1 + U1, 16U);
- /* writing the butterfly processed i0 sample */
- /* xa' = xa + xb + xc + xd */
- /* ya' = ya + yb + yc + yd */
- pSrc16[i0 * 2U] = (R0 >> 1U) + (T0 >> 1U);
- pSrc16[(i0 * 2U) + 1U] = (R1 >> 1U) + (T1 >> 1U);
- /* R0 = (ya + yc) - (yb + yd), R1 = (xa + xc) - (xb + xd) */
- R0 = (R0 >> 1U) - (T0 >> 1U);
- R1 = (R1 >> 1U) - (T1 >> 1U);
- /* Read yb (real), xb(imag) input */
- T0 = pSrc16[i1 * 2U];
- T1 = pSrc16[(i1 * 2U) + 1U];
- /* writing the butterfly processed i0 + fftLen/4 sample */
- /* xc' = (xa-xb+xc-xd) */
- /* yc' = (ya-yb+yc-yd) */
- pSrc16[i1 * 2U] = R0;
- pSrc16[(i1 * 2U) + 1U] = R1;
- /* Read yd (real), xd(imag) input */
- U0 = pSrc16[i3 * 2U];
- U1 = pSrc16[(i3 * 2U) + 1U];
- /* T0 = (yb - yd), T1 = (xb - xd) */
- T0 = __SSAT(T0 - U0, 16U);
- T1 = __SSAT(T1 - U1, 16U);
- /* writing the butterfly processed i0 + fftLen/2 sample */
- /* xb' = (xa-yb-xc+yd) */
- /* yb' = (ya+xb-yc-xd) */
- pSrc16[i2 * 2U] = (S0 >> 1U) - (T1 >> 1U);
- pSrc16[(i2 * 2U) + 1U] = (S1 >> 1U) + (T0 >> 1U);
- /* writing the butterfly processed i0 + 3fftLen/4 sample */
- /* xd' = (xa+yb-xc-yd) */
- /* yd' = (ya-xb-yc+xd) */
- pSrc16[i3 * 2U] = (S0 >> 1U) + (T1 >> 1U);
- pSrc16[(i3 * 2U) + 1U] = (S1 >> 1U) - (T0 >> 1U);
- }
- /* end of last stage process */
- /* output is in 11.5(q5) format for the 1024 point */
- /* output is in 9.7(q7) format for the 256 point */
- /* output is in 7.9(q9) format for the 64 point */
- /* output is in 5.11(q11) format for the 16 point */
- #endif /* #if defined (ARM_MATH_DSP) */
- }
|