fft_plan.h

Go to the documentation of this file.

 
#ifndef OAKFIELD_FFT_PLAN_H
#define OAKFIELD_FFT_PLAN_H
 
#include "oakfield/field.h" /* for SimResult */
#include <complex.h>
#include <stdbool.h>
#include <stddef.h>
#if defined(SIM_HAVE_VDSP)
#include <Accelerate/Accelerate.h>
#endif
 
#ifdef __cplusplus
extern "C" {
#endif
 
typedef struct FFTPow2Plan {
    size_t n;                       
    size_t levels;                  
    size_t *bit_reverse;            
    double complex *stage_twiddles; 
#if defined(SIM_HAVE_VDSP)
    FFTSetupD setup;  
    double *tmp_real; 
    double *tmp_imag; 
#endif
} FFTPow2Plan;
 
typedef struct FFTPlan {
    size_t n;                   
    bool use_bluestein;         
    FFTPow2Plan pow2_plan;      
    FFTPow2Plan bluestein_plan; 
    size_t bluestein_size;      
    double complex *chirp;      
    double complex *chirp_conj; 
    double complex *b_fft;      
    double complex *a_buffer;   
    double complex *scratch_n;  
} FFTPlan;
 
typedef struct FFTPlan2D {
    size_t rows;                 
    size_t cols;                 
    size_t row_stride;           
    size_t col_stride;           
    size_t scratch_len;          
    FFTPlan row_plan;            
    FFTPlan col_plan;            
    double complex *scratch_in;  
    double complex *scratch_out; 
} FFTPlan2D;
 
void fft_pow2_plan_reset(FFTPow2Plan *plan);
 
void fft_pow2_plan_destroy(FFTPow2Plan *plan);
 
SimResult fft_pow2_plan_init(FFTPow2Plan *plan, size_t n);
 
void fft_pow2_execute(const FFTPow2Plan *plan, double complex *data, bool inverse);
 
void fft_plan_reset(FFTPlan *plan);
 
void fft_plan_destroy(FFTPlan *plan);
 
SimResult fft_plan_init(FFTPlan *plan, size_t n);
 
SimResult fft_plan_forward(const FFTPlan *plan, const double complex *input,
                           double complex *output);
 
SimResult fft_plan_inverse(const FFTPlan *plan, const double complex *input,
                           double complex *output);
 
void fft_plan2d_reset(FFTPlan2D *plan);
 
void fft_plan2d_destroy(FFTPlan2D *plan);
 
SimResult fft_plan2d_init(FFTPlan2D *plan, size_t rows, size_t cols, size_t row_stride,
                          size_t col_stride);
 
SimResult fft_plan2d_forward(const FFTPlan2D *plan, const double complex *input,
                             double complex *output);
 
SimResult fft_plan2d_inverse(const FFTPlan2D *plan, const double complex *input,
                             double complex *output);
 
#ifdef __cplusplus
}
#endif
 
#endif /* OAKFIELD_FFT_PLAN_H */