turanszkij
55 lines
1.4 KiB
C
55 lines
1.4 KiB
C
#ifndef _FFT_GENERATOR_H_
|
|
#define _FFT_GENERATOR_H_
|
|
|
|
#include "CommonInclude.h"
|
|
#include "wiGraphicsDevice.h"
|
|
|
|
|
|
//Memory access coherency (in threads)
|
|
#define COHERENCY_GRANULARITY 128
|
|
|
|
|
|
///////////////////////////////////////////////////////////////////////////////
|
|
// Common types
|
|
///////////////////////////////////////////////////////////////////////////////
|
|
|
|
typedef struct CSFFT_512x512_Data_t
|
|
{
|
|
// More than one array can be transformed at same time
|
|
UINT slices;
|
|
|
|
// For 512x512 config, we need 6 constant buffers
|
|
wiGraphicsTypes::GPUBuffer* pRadix008A_CB[6];
|
|
|
|
// Temporary buffers
|
|
wiGraphicsTypes::GPUBuffer* pBuffer_Tmp;
|
|
wiGraphicsTypes::GPUResource* pUAV_Tmp;
|
|
wiGraphicsTypes::GPUResource* pSRV_Tmp;
|
|
|
|
static void LoadShaders();
|
|
} CSFFT512x512_Plan;
|
|
|
|
////////////////////////////////////////////////////////////////////////////////
|
|
// Common constants
|
|
////////////////////////////////////////////////////////////////////////////////
|
|
#define TWO_PI 6.283185307179586476925286766559
|
|
|
|
#define FFT_DIMENSIONS 3U
|
|
#define FFT_PLAN_SIZE_LIMIT (1U << 27)
|
|
|
|
#define FFT_FORWARD -1
|
|
#define FFT_INVERSE 1
|
|
|
|
|
|
void fft512x512_create_plan(CSFFT512x512_Plan* plan, UINT slices);
|
|
void fft512x512_destroy_plan(CSFFT512x512_Plan* plan);
|
|
|
|
void fft_512x512_c2c(CSFFT512x512_Plan* fft_plan,
|
|
wiGraphicsTypes::GPUResource* pUAV_Dst,
|
|
wiGraphicsTypes::GPUResource* pSRV_Dst,
|
|
wiGraphicsTypes::GPUResource* pSRV_Src,
|
|
GRAPHICSTHREAD threadID);
|
|
|
|
|
|
#endif // _FFT_GENERATOR_H_
|