You cannot select more than 25 topics
Topics must start with a letter or number, can include dashes ('-') and can be up to 35 characters long.
135 lines
3.8 KiB
C++
135 lines
3.8 KiB
C++
#include "TransformF32.h"
|
|
#include "Error.h"
|
|
#include "arm_math.h"
|
|
#include "arm_const_structs.h"
|
|
|
|
|
|
void TransformF32::test_cfft_f32()
|
|
{
|
|
arm_cfft_f32(&(this->cfftInstance), this->pDst, this->ifft,this->bitRev);
|
|
}
|
|
|
|
void TransformF32::test_rfft_f32()
|
|
{
|
|
arm_rfft_fast_f32(&this->rfftFastInstance, this->pSrc, this->pDst, this->ifft);
|
|
}
|
|
|
|
void TransformF32::test_dct4_f32()
|
|
{
|
|
arm_dct4_f32(
|
|
&this->dct4Instance,
|
|
this->pState,
|
|
this->pDst);
|
|
}
|
|
|
|
void TransformF32::test_cfft_radix4_f32()
|
|
{
|
|
arm_cfft_radix4_f32(&this->cfftRadix4Instance,this->pDst);
|
|
}
|
|
|
|
void TransformF32::test_cfft_radix2_f32()
|
|
{
|
|
arm_cfft_radix2_f32(&this->cfftRadix2Instance,this->pDst);
|
|
}
|
|
|
|
|
|
void TransformF32::setUp(Testing::testID_t id,std::vector<Testing::param_t>& params,Client::PatternMgr *mgr)
|
|
{
|
|
|
|
float32_t normalize;
|
|
|
|
std::vector<Testing::param_t>::iterator it = params.begin();
|
|
this->nbSamples = *it++;
|
|
this->ifft = *it++;
|
|
this->bitRev = *it;
|
|
|
|
switch(id)
|
|
{
|
|
case TEST_CFFT_F32_1:
|
|
samples.reload(TransformF32::INPUTC_F32_ID,mgr,2*this->nbSamples);
|
|
output.create(2*this->nbSamples,TransformF32::OUT_F32_ID,mgr);
|
|
|
|
this->pSrc=samples.ptr();
|
|
this->pDst=output.ptr();
|
|
|
|
status=arm_cfft_init_f32(&cfftInstance,this->nbSamples);
|
|
memcpy(this->pDst,this->pSrc,2*sizeof(float32_t)*this->nbSamples);
|
|
break;
|
|
|
|
case TEST_RFFT_F32_2:
|
|
samples.reload(TransformF32::INPUTR_F32_ID,mgr,this->nbSamples);
|
|
output.create(this->nbSamples,TransformF32::OUT_F32_ID,mgr);
|
|
|
|
this->pSrc=samples.ptr();
|
|
this->pDst=output.ptr();
|
|
|
|
arm_rfft_fast_init_f32(&this->rfftFastInstance, this->nbSamples);
|
|
break;
|
|
|
|
case TEST_DCT4_F32_3:
|
|
samples.reload(TransformF32::INPUTR_F32_ID,mgr,this->nbSamples);
|
|
output.create(this->nbSamples,TransformF32::OUT_F32_ID,mgr);
|
|
state.create(2*this->nbSamples,TransformF32::STATE_F32_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(float32_t)*this->nbSamples);
|
|
|
|
arm_dct4_init_f32(
|
|
&this->dct4Instance,
|
|
&this->rfftInstance,
|
|
&this->cfftRadix4Instance,
|
|
this->nbSamples,
|
|
this->nbSamples/2,
|
|
normalize);
|
|
break;
|
|
|
|
case TEST_CFFT_RADIX4_F32_4:
|
|
samples.reload(TransformF32::INPUTC_F32_ID,mgr,2*this->nbSamples);
|
|
output.create(2*this->nbSamples,TransformF32::OUT_F32_ID,mgr);
|
|
|
|
this->pSrc=samples.ptr();
|
|
this->pDst=output.ptr();
|
|
|
|
|
|
memcpy(this->pDst,this->pSrc,2*sizeof(float32_t)*this->nbSamples);
|
|
|
|
arm_cfft_radix4_init_f32(&this->cfftRadix4Instance,
|
|
this->nbSamples,
|
|
this->ifft,
|
|
this->bitRev);
|
|
|
|
break;
|
|
|
|
case TEST_CFFT_RADIX2_F32_5:
|
|
samples.reload(TransformF32::INPUTC_F32_ID,mgr,2*this->nbSamples);
|
|
output.create(2*this->nbSamples,TransformF32::OUT_F32_ID,mgr);
|
|
|
|
this->pSrc=samples.ptr();
|
|
this->pDst=output.ptr();
|
|
|
|
|
|
memcpy(this->pDst,this->pSrc,2*sizeof(float32_t)*this->nbSamples);
|
|
|
|
arm_cfft_radix2_init_f32(&this->cfftRadix2Instance,
|
|
this->nbSamples,
|
|
this->ifft,
|
|
this->bitRev);
|
|
break;
|
|
|
|
}
|
|
|
|
|
|
|
|
|
|
}
|
|
|
|
void TransformF32::tearDown(Testing::testID_t id,Client::PatternMgr *mgr)
|
|
{
|
|
}
|