262 lines
18 KiB
C
262 lines
18 KiB
C
|
/* ----------------------------------------------------------------------
|
||
|
|
* Project: CMSIS DSP Python Wrapper
|
||
|
|
* Title: fftinit.c
|
||
|
|
* Description: FFT init functions for the Python wrapper
|
||
|
|
*
|
||
|
|
* $Date: 25. March 2019
|
||
|
|
* $Revision: V0.0.1
|
||
|
|
*
|
||
|
|
* 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"
|
||
|
|
#include "arm_common_tables.h"
|
||
|
|
#include "arm_const_structs.h"
|
||
|
|
|
||
|
|
#define FFTINIT(SIZE) \
|
||
|
|
S->bitRevLength = arm_cfft_sR_f32_len##SIZE.bitRevLength; \
|
||
|
|
S->pBitRevTable = arm_cfft_sR_f32_len##SIZE.pBitRevTable; \
|
||
|
|
S->pTwiddle = arm_cfft_sR_f32_len##SIZE.pTwiddle;
|
||
|
|
|
||
|
|
#define FFTFXTINIT(EXT,SIZE) \
|
||
|
|
S->bitRevLength = arm_cfft_sR_##EXT##_len##SIZE.bitRevLength; \
|
||
|
|
S->pBitRevTable = arm_cfft_sR_##EXT##_len##SIZE.pBitRevTable; \
|
||
|
|
S->pTwiddle = arm_cfft_sR_##EXT##_len##SIZE.pTwiddle;
|
||
|
|
|
||
|
|
arm_status arm_cfft_init_f32(
|
||
|
|
arm_cfft_instance_f32 * S,
|
||
|
|
uint16_t 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 = (float32_t *)twiddleCoef_4096;
|
||
|
|
|
||
|
|
|
||
|
|
/* Initializations of Instance structure depending on the FFT length */
|
||
|
|
switch (S->fftLen) {
|
||
|
|
/* Initializations of structure parameters for 4096 point FFT */
|
||
|
|
case 4096U:
|
||
|
|
/* Initialise the bit reversal table modifier */
|
||
|
|
FFTINIT(4096);
|
||
|
|
break;
|
||
|
|
|
||
|
|
/* Initializations of structure parameters for 2048 point FFT */
|
||
|
|
case 2048U:
|
||
|
|
/* Initialise the bit reversal table modifier */
|
||
|
|
FFTINIT(2048);
|
||
|
|
|
||
|
|
break;
|
||
|
|
|
||
|
|
/* Initializations of structure parameters for 1024 point FFT */
|
||
|
|
case 1024U:
|
||
|
|
/* Initialise the bit reversal table modifier */
|
||
|
|
FFTINIT(1024);
|
||
|
|
|
||
|
|
break;
|
||
|
|
|
||
|
|
/* Initializations of structure parameters for 512 point FFT */
|
||
|
|
case 512U:
|
||
|
|
/* Initialise the bit reversal table modifier */
|
||
|
|
FFTINIT(512);
|
||
|
|
break;
|
||
|
|
|
||
|
|
case 256U:
|
||
|
|
FFTINIT(256);
|
||
|
|
break;
|
||
|
|
|
||
|
|
case 128U:
|
||
|
|
FFTINIT(128);
|
||
|
|
break;
|
||
|
|
|
||
|
|
case 64U:
|
||
|
|
FFTINIT(64);
|
||
|
|
break;
|
||
|
|
|
||
|
|
case 32U:
|
||
|
|
FFTINIT(32);
|
||
|
|
break;
|
||
|
|
|
||
|
|
case 16U:
|
||
|
|
/* Initializations of structure parameters for 16 point FFT */
|
||
|
|
FFTINIT(16);
|
||
|
|
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)
|
||
|
|
{
|
||
|
|
/* 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 = (float32_t *)twiddleCoef_4096;
|
||
|
|
|
||
|
|
|
||
|
|
/* Initializations of Instance structure depending on the FFT length */
|
||
|
|
switch (S->fftLen) {
|
||
|
|
/* Initializations of structure parameters for 4096 point FFT */
|
||
|
|
case 4096U:
|
||
|
|
/* Initialise the bit reversal table modifier */
|
||
|
|
FFTFXTINIT(q31,4096);
|
||
|
|
break;
|
||
|
|
|
||
|
|
/* Initializations of structure parameters for 2048 point FFT */
|
||
|
|
case 2048U:
|
||
|
|
/* Initialise the bit reversal table modifier */
|
||
|
|
FFTFXTINIT(q31,2048);
|
||
|
|
|
||
|
|
break;
|
||
|
|
|
||
|
|
/* Initializations of structure parameters for 1024 point FFT */
|
||
|
|
case 1024U:
|
||
|
|
/* Initialise the bit reversal table modifier */
|
||
|
|
FFTFXTINIT(q31,1024);
|
||
|
|
|
||
|
|
break;
|
||
|
|
|
||
|
|
/* Initializations of structure parameters for 512 point FFT */
|
||
|
|
case 512U:
|
||
|
|
/* Initialise the bit reversal table modifier */
|
||
|
|
FFTFXTINIT(q31,512);
|
||
|
|
break;
|
||
|
|
|
||
|
|
case 256U:
|
||
|
|
FFTFXTINIT(q31,256);
|
||
|
|
break;
|
||
|
|
|
||
|
|
case 128U:
|
||
|
|
FFTFXTINIT(q31,128);
|
||
|
|
break;
|
||
|
|
|
||
|
|
case 64U:
|
||
|
|
FFTFXTINIT(q31,64);
|
||
|
|
break;
|
||
|
|
|
||
|
|
case 32U:
|
||
|
|
FFTFXTINIT(q31,32);
|
||
|
|
break;
|
||
|
|
|
||
|
|
case 16U:
|
||
|
|
/* Initializations of structure parameters for 16 point FFT */
|
||
|
|
FFTFXTINIT(q31,16);
|
||
|
|
break;
|
||
|
|
|
||
|
|
|
||
|
|
default:
|
||
|
|
/* Reporting argument error if fftSize is not valid value */
|
||
|
|
status = ARM_MATH_ARGUMENT_ERROR;
|
||
|
|
break;
|
||
|
|
}
|
||
|
|
|
||
|
|
|
||
|
|
return (status);
|
||
|
|
}
|
||
|
|
|
||
|
|
arm_status arm_cfft_init_q15(
|
||
|
|
arm_cfft_instance_q15 * S,
|
||
|
|
uint16_t 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 = (float32_t *)twiddleCoef_4096;
|
||
|
|
|
||
|
|
|
||
|
|
/* Initializations of Instance structure depending on the FFT length */
|
||
|
|
switch (S->fftLen) {
|
||
|
|
/* Initializations of structure parameters for 4096 point FFT */
|
||
|
|
case 4096U:
|
||
|
|
/* Initialise the bit reversal table modifier */
|
||
|
|
FFTFXTINIT(q15,4096);
|
||
|
|
break;
|
||
|
|
|
||
|
|
/* Initializations of structure parameters for 2048 point FFT */
|
||
|
|
case 2048U:
|
||
|
|
/* Initialise the bit reversal table modifier */
|
||
|
|
FFTFXTINIT(q15,2048);
|
||
|
|
|
||
|
|
break;
|
||
|
|
|
||
|
|
/* Initializations of structure parameters for 1024 point FFT */
|
||
|
|
case 1024U:
|
||
|
|
/* Initialise the bit reversal table modifier */
|
||
|
|
FFTFXTINIT(q15,1024);
|
||
|
|
|
||
|
|
break;
|
||
|
|
|
||
|
|
/* Initializations of structure parameters for 512 point FFT */
|
||
|
|
case 512U:
|
||
|
|
/* Initialise the bit reversal table modifier */
|
||
|
|
FFTFXTINIT(q15,512);
|
||
|
|
break;
|
||
|
|
|
||
|
|
case 256U:
|
||
|
|
FFTFXTINIT(q15,256);
|
||
|
|
break;
|
||
|
|
|
||
|
|
case 128U:
|
||
|
|
FFTFXTINIT(q15,128);
|
||
|
|
break;
|
||
|
|
|
||
|
|
case 64U:
|
||
|
|
FFTFXTINIT(q15,64);
|
||
|
|
break;
|
||
|
|
|
||
|
|
case 32U:
|
||
|
|
FFTFXTINIT(q15,32);
|
||
|
|
break;
|
||
|
|
|
||
|
|
case 16U:
|
||
|
|
/* Initializations of structure parameters for 16 point FFT */
|
||
|
|
FFTFXTINIT(q15,16);
|
||
|
|
break;
|
||
|
|
|
||
|
|
|
||
|
|
default:
|
||
|
|
/* Reporting argument error if fftSize is not valid value */
|
||
|
|
status = ARM_MATH_ARGUMENT_ERROR;
|
||
|
|
break;
|
||
|
|
}
|
||
|
|
|
||
|
|
|
||
|
|
return (status);
|
||
|
|
}
|