#include "TransformQ31.h" #include "Error.h" #include "arm_const_structs.h" #define CFFT_INIT(typ, suffix, S, fftLen) \ \ { \ \ /* Initialise the default arm status */ \ arm_status status = ARM_MATH_SUCCESS; \ \ /* Initialise the FFT length */ \ S->fftLen = fftLen; \ \ /* Initialise the Twiddle coefficient pointer */ \ S->pTwiddle = (typ *)twiddleCoef_4096_##suffix; \ \ \ /* Initializations of Instance structure depending on the FFT length */\ switch (S->fftLen) { \ \ /* Initializations of structure parameters for 2048 point FFT */ \ case 2048U: \ /* Initialise the bit reversal table modifier */ \ S->bitRevLength = ARMBITREVINDEXTABLE_FIXED_2048_TABLE_LENGTH; \ S->pBitRevTable = (uint16_t *)armBitRevIndexTable_fixed_2048; \ S->pTwiddle = (typ *)twiddleCoef_2048_##suffix; \ break; \ \ /* Initializations of structure parameters for 1024 point FFT */ \ case 1024U: \ /* Initialise the bit reversal table modifier */ \ S->bitRevLength = ARMBITREVINDEXTABLE_FIXED_1024_TABLE_LENGTH; \ S->pBitRevTable = (uint16_t *)armBitRevIndexTable_fixed_1024; \ S->pTwiddle = (typ *)twiddleCoef_1024_##suffix; \ break; \ \ /* Initializations of structure parameters for 512 point FFT */ \ case 512U: \ /* Initialise the bit reversal table modifier */ \ S->bitRevLength = ARMBITREVINDEXTABLE_FIXED_512_TABLE_LENGTH; \ S->pBitRevTable = (uint16_t *)armBitRevIndexTable_fixed_512; \ S->pTwiddle = (typ *)twiddleCoef_512_##suffix; \ break; \ \ case 256U: \ S->bitRevLength = ARMBITREVINDEXTABLE_FIXED_256_TABLE_LENGTH; \ S->pBitRevTable = (uint16_t *)armBitRevIndexTable_fixed_256; \ S->pTwiddle = (typ *)twiddleCoef_256_##suffix; \ \ break; \ \ case 128U: \ S->bitRevLength = ARMBITREVINDEXTABLE_FIXED_128_TABLE_LENGTH; \ S->pBitRevTable = (uint16_t *)armBitRevIndexTable_fixed_128; \ S->pTwiddle = (typ *)twiddleCoef_128_##suffix; \ \ break; \ \ case 64U: \ S->bitRevLength = ARMBITREVINDEXTABLE_FIXED_64_TABLE_LENGTH; \ S->pBitRevTable = (uint16_t *)armBitRevIndexTable_fixed_64; \ S->pTwiddle = (typ *)twiddleCoef_64_##suffix; \ break; \ \ case 32U: \ S->bitRevLength = ARMBITREVINDEXTABLE_FIXED_32_TABLE_LENGTH; \ S->pBitRevTable = (uint16_t *)armBitRevIndexTable_fixed_32; \ S->pTwiddle = (typ *)twiddleCoef_32_##suffix; \ break; \ \ case 16U: \ /* Initializations of structure parameters for 16 point FFT */ \ S->bitRevLength = ARMBITREVINDEXTABLE_FIXED_16_TABLE_LENGTH; \ S->pBitRevTable = (uint16_t *)armBitRevIndexTable_fixed_16; \ S->pTwiddle = (typ *)twiddleCoef_16_##suffix; \ break; \ \ \ default: \ /* Reporting argument error if fftSize is not valid value */ \ status = ARM_MATH_ARGUMENT_ERROR; \ break; \ } \ \ \ return (status); \ \ } arm_status arm_cfft_init_q31(arm_cfft_instance_q31 *S, uint16_t fftLen) { CFFT_INIT(q31_t, q31, S, fftLen); } void TransformQ31::test_cfft_q31() { arm_cfft_q31(&this->cfftInstance, this->pDst, this->ifft,this->bitRev); } void TransformQ31::test_rfft_q31() { arm_rfft_q31(&this->rfftInstance, this->pSrc, this->pDst); } void TransformQ31::test_dct4_q31() { arm_dct4_q31( &this->dct4Instance, this->pState, this->pDst); } void TransformQ31::test_cfft_radix4_q31() { arm_cfft_radix4_q31(&this->cfftRadix4Instance,this->pDst); } void TransformQ31::test_cfft_radix2_q31() { arm_cfft_radix2_q31(&this->cfftRadix2Instance,this->pDst); } void TransformQ31::setUp(Testing::testID_t id,std::vector& params,Client::PatternMgr *mgr) { float32_t normalize; std::vector::iterator it = params.begin(); this->nbSamples = *it++; this->ifft = *it++; this->bitRev = *it; switch(id) { case TEST_CFFT_Q31_1: samples.reload(TransformQ31::INPUTC_Q31_ID,mgr,2*this->nbSamples); output.create(2*this->nbSamples,TransformQ31::OUT_Q31_ID,mgr); this->pSrc=samples.ptr(); this->pDst=output.ptr(); arm_cfft_init_q31(&this->cfftInstance,this->nbSamples); memcpy(this->pDst,this->pSrc,2*sizeof(q31_t)*this->nbSamples); break; case TEST_RFFT_Q31_2: samples.reload(TransformQ31::INPUTR_Q31_ID,mgr,this->nbSamples); output.create(this->nbSamples,TransformQ31::OUT_Q31_ID,mgr); this->pSrc=samples.ptr(); this->pDst=output.ptr(); arm_rfft_init_q31(&this->rfftInstance, this->nbSamples,this->ifft,this->bitRev); break; case TEST_DCT4_Q31_3: samples.reload(TransformQ31::INPUTR_Q31_ID,mgr,this->nbSamples); output.create(this->nbSamples,TransformQ31::OUT_Q31_ID,mgr); state.create(2*this->nbSamples,TransformQ31::STATE_Q31_ID,mgr); this->pSrc=samples.ptr(); this->pDst=output.ptr(); this->pState=state.ptr(); normalize = sqrt((2.0f/(float32_t)this->nbSamples)); memcpy(this->pDst,this->pSrc,sizeof(q31_t)*this->nbSamples); arm_dct4_init_q31( &this->dct4Instance, &this->rfftInstance, &this->cfftRadix4Instance, this->nbSamples, this->nbSamples/2, normalize); break; case TEST_CFFT_RADIX4_Q31_4: samples.reload(TransformQ31::INPUTC_Q31_ID,mgr,2*this->nbSamples); output.create(2*this->nbSamples,TransformQ31::OUT_Q31_ID,mgr); this->pSrc=samples.ptr(); this->pDst=output.ptr(); memcpy(this->pDst,this->pSrc,2*sizeof(q31_t)*this->nbSamples); arm_cfft_radix4_init_q31(&this->cfftRadix4Instance, this->nbSamples, this->ifft, this->bitRev); break; case TEST_CFFT_RADIX2_Q31_5: samples.reload(TransformQ31::INPUTC_Q31_ID,mgr,2*this->nbSamples); output.create(2*this->nbSamples,TransformQ31::OUT_Q31_ID,mgr); this->pSrc=samples.ptr(); this->pDst=output.ptr(); memcpy(this->pDst,this->pSrc,2*sizeof(q31_t)*this->nbSamples); arm_cfft_radix2_init_q31(&this->cfftRadix2Instance, this->nbSamples, this->ifft, this->bitRev); break; } } void TransformQ31::tearDown(Testing::testID_t id,Client::PatternMgr *mgr) { }