110 lines
2.5 KiB
C
110 lines
2.5 KiB
C
// lasp_fft.h
|
|
//
|
|
// Author: J.A. de Jong - ASCEE
|
|
//
|
|
// Description:
|
|
// Interface to the FFT library, multiple channel FFT's
|
|
//////////////////////////////////////////////////////////////////////
|
|
#pragma once
|
|
#ifndef LASP_FFT_H
|
|
#define LASP_FFT_H
|
|
#include "lasp_types.h"
|
|
#include "lasp_mat.h"
|
|
#ifdef __cplusplus
|
|
extern "C" {
|
|
#endif
|
|
|
|
/**
|
|
* Load wisdom data from a NULL-terminated character string. Note that at this
|
|
* point in time this is only used by the FFTW fft backend.
|
|
*
|
|
* @param[in] wisdom: Wisdom string
|
|
*/
|
|
void load_fft_wisdom(const char* wisdom);
|
|
|
|
/**
|
|
* Creates a NULL-terminated string containing FFTW wisdom, or state knowledge
|
|
* from other libraries.
|
|
*
|
|
* @returns A NULL-terminated string. *Should be deallocated after being used*
|
|
*/
|
|
char* store_fft_wisdom(void);
|
|
|
|
|
|
/**
|
|
* Perform forward FFT's on real time data.
|
|
*
|
|
*/
|
|
typedef struct Fft_s Fft;
|
|
|
|
/**
|
|
* Construct an Fft object
|
|
*
|
|
* @param nfft Nfft size
|
|
*
|
|
* @return Pointer to Fft handle, NULL on error
|
|
*/
|
|
Fft* Fft_create(const us nfft);
|
|
/**
|
|
* Returns the nfft for this Fft instance
|
|
*
|
|
* @return nfft
|
|
*/
|
|
us Fft_nfft(const Fft*);
|
|
|
|
/**
|
|
* Compute the fft of a single channel.
|
|
*
|
|
* @param[in] fft Fft handle.
|
|
*
|
|
* @param[in] timedata Input time data pointer, should have size nfft
|
|
* @param[out] result Pointer to result vector should have size
|
|
* nfft/2+1
|
|
*/
|
|
void Fft_fft_single(const Fft* fft,const vd* timedata,vc* result);
|
|
|
|
/**
|
|
* Compute the fft of the data matrix, first axis is assumed to be
|
|
* the time axis.
|
|
*
|
|
* @param[in] fft Fft handle.
|
|
|
|
* @param[in] timedata Input time data. First axis is assumed to be
|
|
* the time, second the channel number.
|
|
*
|
|
* @param[out] result: Fft't data, should have size (nfft/2+1) *
|
|
* nchannels
|
|
*/
|
|
void Fft_fft(const Fft* fft,const dmat* timedata,cmat* result);
|
|
|
|
/**
|
|
* Perform inverse fft on a single channel.
|
|
*
|
|
* @param[in] fft Fft handle.
|
|
* @param[in] freqdata Frequency domain input data, to be iFft'th.
|
|
* @param[out] timedata: iFft't data, should have size (nfft).
|
|
*/
|
|
void Fft_ifft_single(const Fft* fft,const vc* freqdata,vd* timedata);
|
|
|
|
/**
|
|
* Perform inverse FFT
|
|
*
|
|
* @param[in] fft Fft handle
|
|
* @param[in] freqdata Frequency domain data
|
|
* @param[out] timedata Time domain result
|
|
*/
|
|
void Fft_ifft(const Fft* fft,const cmat* freqdata,dmat* timedata);
|
|
|
|
/**
|
|
* Free up resources of Fft handle.
|
|
*
|
|
* @param fft Fft handle.
|
|
*/
|
|
void Fft_free(Fft* fft);
|
|
|
|
#ifdef __cplusplus
|
|
}
|
|
#endif
|
|
#endif // LASP_FFT_H
|
|
//////////////////////////////////////////////////////////////////////
|