FourierTransform.h

#pragma once
 
#include "Rendering/Common.h"
 
namespace Cogs
{
  class GPGPUQuadRenderer;
  class IGraphicsDevice;
  struct RenderContext;
 
  class FourierTransform2D
  {
  public:
    FourierTransform2D() :device(nullptr), log2N(0) {}
 
    void initialize(IGraphicsDevice* device, GPGPUQuadRenderer& gpgpuQuadRenderer);
 
    void setSize(int log2N);
 
    void inverseFourierTransform(RenderContext& context,
                                 GPGPUQuadRenderer& gpgpuQuadRenderer,
                                 RenderTargetHandle& target,
                                 TextureHandle& source,
                                 bool doubleData = false);
 
 
  protected:
    IGraphicsDevice* device;
    int log2N;
 
    EffectHandle passRadix2x2;
    InputLayoutHandle passRadix2x2InputLayout;
 
    EffectHandle passRadix2x2Double;
    InputLayoutHandle passRadix2x2DoubleInputLayout;
 
    BufferHandle constantBuffer;
 
    TextureHandle tableTex;
 
    TextureHandle scratchTex[2];
    RenderTargetHandle scratchTarget[2];
 
    TextureHandle scratchDoubleTex[2];
    RenderTargetHandle scratchDoubleTarget[2];
 
    static TextureHandle createBakedTable(IGraphicsDevice* device, const int log2N);
  };
 
}