#include "jtest.h"
|
#include "ref.h"
|
#include "arr_desc.h"
|
#include "transform_templates.h"
|
#include "transform_test_data.h"
|
|
#define CFFT_SNR_THRESHOLD 120
|
|
/*
|
CFFT function test template. Arguments are: inverse-transform flag, function
|
suffix (q7/q15/q31/f32) and the output type (q7_t, q15_t, q31_t, float32_t)
|
*/
|
#define CFFT_TEST_BODY(ifft_flag, suffix, output_type) \
|
do \
|
{ \
|
/* Go through all arm_cfft_instances */ \
|
TEMPLATE_DO_ARR_DESC( \
|
cfft_inst_idx, const arm_cfft_instance_##suffix *, cfft_inst_ptr, \
|
transform_cfft_##suffix##_structs \
|
, \
|
\
|
TRANSFORM_PREPARE_INPLACE_INPUTS( \
|
transform_fft_##suffix##_inputs, \
|
cfft_inst_ptr->fftLen * \
|
sizeof(output_type) * \
|
2 /*complex_inputs*/); \
|
\
|
/* Display parameter values */ \
|
JTEST_DUMP_STRF("Block Size: %d\n" \
|
"Inverse-transform flag: %d\n", \
|
(int)cfft_inst_ptr->fftLen, \
|
(int)ifft_flag); \
|
\
|
/* Display cycle count and run test */ \
|
JTEST_COUNT_CYCLES( \
|
arm_cfft_##suffix(cfft_inst_ptr, \
|
(void *) transform_fft_inplace_input_fut, \
|
ifft_flag, /* IFFT Flag */ \
|
1)); /* Bitreverse flag */ \
|
ref_cfft_##suffix(cfft_inst_ptr, \
|
(void *) transform_fft_inplace_input_ref, \
|
ifft_flag, /* IFFT Flag */ \
|
1); /* Bitreverse flag */ \
|
\
|
/* Test correctness */ \
|
TRANSFORM_SNR_COMPARE_CMPLX_INTERFACE( \
|
cfft_inst_ptr->fftLen, \
|
output_type)); \
|
\
|
return JTEST_TEST_PASSED; \
|
} while (0)
|
|
|
/*
|
CFFT function with downshift test template. Arguments are: inverse-transform flag,
|
function suffix (q7/q15/q31/f32) and the output type (q7_t, q15_t, q31_t, float32_t)
|
*/
|
#define CFFT_DOWNSHIFT_INPUT_TEST_BODY(ifft_flag, suffix, output_type) \
|
do \
|
{ \
|
/* Go through all arm_cfft_instances */ \
|
TEMPLATE_DO_ARR_DESC( \
|
cfft_inst_idx, const arm_cfft_instance_##suffix *, cfft_inst_ptr, \
|
transform_cfft_##suffix##_structs \
|
, \
|
\
|
TRANSFORM_PREPARE_INPLACE_INPUTS_DOWNSHIFT( \
|
transform_fft_##suffix##_inputs, \
|
cfft_inst_ptr->fftLen * \
|
sizeof(output_type) * \
|
2 /*complex_inputs*/, output_type); \
|
\
|
/* Display parameter values */ \
|
JTEST_DUMP_STRF("Block Size: %d\n" \
|
"Inverse-transform flag: %d\n", \
|
(int)cfft_inst_ptr->fftLen, \
|
(int)ifft_flag); \
|
\
|
/* Display cycle count and run test */ \
|
JTEST_COUNT_CYCLES( \
|
arm_cfft_##suffix(cfft_inst_ptr, \
|
(void *) transform_fft_inplace_input_fut, \
|
ifft_flag, /* IFFT Flag */ \
|
1)); /* Bitreverse flag */ \
|
ref_cfft_##suffix(cfft_inst_ptr, \
|
(void *) transform_fft_inplace_input_ref, \
|
ifft_flag, /* IFFT Flag */ \
|
1); /* Bitreverse flag */ \
|
\
|
/* Test correctness */ \
|
TRANSFORM_SNR_COMPARE_CMPLX_INTERFACE( \
|
cfft_inst_ptr->fftLen, \
|
output_type)); \
|
\
|
return JTEST_TEST_PASSED; \
|
} while (0)
|
|
|
/* Test declarations */
|
JTEST_DEFINE_TEST(cfft_f32_test, cfft_f32)
|
{
|
CFFT_TEST_BODY((uint8_t) 0, f32, float32_t);
|
}
|
|
JTEST_DEFINE_TEST(cfft_f32_ifft_test, cfft_f32)
|
{
|
CFFT_TEST_BODY((uint8_t) 1, f32, float32_t);
|
}
|
|
JTEST_DEFINE_TEST(cfft_q31_test, cfft_q31)
|
{
|
CFFT_TEST_BODY((uint8_t) 0, q31, q31_t);
|
}
|
|
JTEST_DEFINE_TEST(cfft_q31_ifft_test, cfft_q31)
|
{
|
CFFT_TEST_BODY((uint8_t) 1, q31, q31_t);
|
}
|
|
JTEST_DEFINE_TEST(cfft_q15_test, cfft_q15)
|
{
|
CFFT_TEST_BODY((uint8_t) 0, q15, q15_t);
|
}
|
|
JTEST_DEFINE_TEST(cfft_q15_ifft_test, cfft_q15)
|
{
|
CFFT_TEST_BODY((uint8_t) 1, q15, q15_t);
|
}
|
|
/*--------------------------------------------------------------------------------*/
|
/* Collect all tests in a group */
|
/*--------------------------------------------------------------------------------*/
|
|
JTEST_DEFINE_GROUP(cfft_tests)
|
{
|
JTEST_TEST_CALL(cfft_f32_test);
|
JTEST_TEST_CALL(cfft_f32_ifft_test);
|
|
JTEST_TEST_CALL(cfft_q31_test);
|
JTEST_TEST_CALL(cfft_q31_ifft_test);
|
|
JTEST_TEST_CALL(cfft_q15_test);
|
JTEST_TEST_CALL(cfft_q15_ifft_test);
|
}
|
