riscv_fft_bin_example_f32.c

/* ----------------------------------------------------------------------
* Copyright (C) 2010-2012 ARM Limited. All rights reserved.
* Copyright (c) 2019 Nuclei Limited. All rights reserved.
*
* $Date:         17. January 2013
* $Revision:     V1.4.0
*
* Project:       NMSIS DSP Library
* Title:             riscv_fft_bin_example_f32.c
*
* Description:   Example code demonstrating calculation of Max energy bin of
*                frequency domain of input signal.
*
* Target Processor: RISC-V Cores
*
* Redistribution and use in source and binary forms, with or without
* modification, are permitted provided that the following conditions
* are met:
*   - Redistributions of source code must retain the above copyright
*     notice, this list of conditions and the following disclaimer.
*   - Redistributions in binary form must reproduce the above copyright
*     notice, this list of conditions and the following disclaimer in
*     the documentation and/or other materials provided with the
*     distribution.
*   - Neither the name of ARM LIMITED nor the names of its contributors
*     may be used to endorse or promote products derived from this
*     software without specific prior written permission.
*
* THIS SOFTWARE IS PROVIDED BY THE COPYRIGHT HOLDERS AND CONTRIBUTORS
* "AS IS" AND ANY EXPRESS OR IMPLIED WARRANTIES, INCLUDING, BUT NOT
* LIMITED TO, THE IMPLIED WARRANTIES OF MERCHANTABILITY AND FITNESS
* FOR A PARTICULAR PURPOSE ARE DISCLAIMED. IN NO EVENT SHALL THE
* COPYRIGHT OWNER OR CONTRIBUTORS BE LIABLE FOR ANY DIRECT, INDIRECT,
* INCIDENTAL, SPECIAL, EXEMPLARY, OR CONSEQUENTIAL DAMAGES (INCLUDING,
* BUT NOT LIMITED TO, PROCUREMENT OF SUBSTITUTE GOODS OR SERVICES;
* LOSS OF USE, DATA, OR PROFITS; OR BUSINESS INTERRUPTION) HOWEVER
* CAUSED AND ON ANY THEORY OF LIABILITY, WHETHER IN CONTRACT, STRICT
* LIABILITY, OR TORT (INCLUDING NEGLIGENCE OR OTHERWISE) ARISING IN
* ANY WAY OUT OF THE USE OF THIS SOFTWARE, EVEN IF ADVISED OF THE
* POSSIBILITY OF SUCH DAMAGE.
 * -------------------------------------------------------------------- */
 
#include <stdlib.h>
#include <string.h>
 
#include "riscv_math.h"
#include "riscv_const_structs.h"
#include <stdio.h>
 
#define TEST_LENGTH_SAMPLES 2048
 
/* -------------------------------------------------------------------
* External Input and Output buffer Declarations for FFT Bin Example
* ------------------------------------------------------------------- */
extern float32_t testInput_f32_10khz[TEST_LENGTH_SAMPLES];
static float32_t testOutput[TEST_LENGTH_SAMPLES/2];
 
/* ------------------------------------------------------------------
* Global variables for FFT Bin Example
* ------------------------------------------------------------------- */
uint32_t fftSize = 1024;
uint32_t ifftFlag = 0;
uint32_t doBitReverse = 1;
 
/* Reference index at which max energy of bin ocuurs */
uint32_t refIndex = 213, testIndex = 0;
 
/* ----------------------------------------------------------------------
* Max magnitude FFT Bin test
* ------------------------------------------------------------------- */
 
int32_t main(void)
{
/*  Enable Vector  */
#if (defined(__riscv_vector))
  __RV_CSR_SET(CSR_MSTATUS, 0x200);
#endif
 
  riscv_status status;
  float32_t maxValue;
 
  status = RISCV_MATH_SUCCESS;
 
#if !defined(RISCV_MATH_VECTOR)
  /* Process the data through the CFFT/CIFFT module */
  riscv_cfft_f32(&riscv_cfft_sR_f32_len1024, testInput_f32_10khz, ifftFlag, doBitReverse);
 
  /* Process the data through the Complex Magnitude Module for
  calculating the magnitude at each bin */
  riscv_cmplx_mag_f32(testInput_f32_10khz, testOutput, fftSize);
#else
  riscv_cfft_instance_f32 S;
  riscv_cfft_init_f32(&S, fftSize);
  int32_t tmp_buf_size = riscv_cfft_tmp_buffer_size(RISCV_MATH_F32, fftSize);
  int32_t output_buf_size = riscv_cfft_output_buffer_size(RISCV_MATH_F32, fftSize);
  float32_t *tmp_buf = (float32_t *)malloc(tmp_buf_size * sizeof(float32_t));
  float32_t *output_buf = (float32_t *)malloc(output_buf_size * sizeof(float32_t));
  if (tmp_buf != NULL && output_buf != NULL) {
    memset(output_buf, 0, output_buf_size * sizeof(float32_t));
    riscv_cfft_f32(&S, testInput_f32_10khz, output_buf, tmp_buf, ifftFlag);
    riscv_cmplx_mag_f32(output_buf, testOutput, fftSize);
    free(tmp_buf);
    free(output_buf);
  } else {
    status = RISCV_MATH_TEST_FAILURE;
  }
#endif
 
  /* Calculates maxValue and returns corresponding BIN value */
  riscv_max_f32(testOutput, fftSize, &maxValue, &testIndex);
 
  if (testIndex !=  refIndex)
  {
    status = RISCV_MATH_TEST_FAILURE;
  }
 
  /* ----------------------------------------------------------------------
  ** Loop here if the signals fail the PASS check.
  ** This denotes a test failure
  ** ------------------------------------------------------------------- */
 
  if ( status != RISCV_MATH_SUCCESS)
  {
    printf("failed\n");
    return 1;
  }
 
  printf("passed\n");
 
  return 0;
}