LSAOPass.h

#pragma once
 
#include <RendererCore/../../../Data/Base/Shaders/Pipeline/LSAOConstants.h>
#include <RendererCore/Declarations.h>
#include <RendererCore/Pipeline/RenderPipelinePass.h>
#include <RendererCore/Shader/ConstantBufferStorage.h>
#include <RendererCore/Shader/ShaderResource.h>
#include <RendererFoundation/RendererFoundationDLL.h>
 
struct PL_RENDERERCORE_DLL plLSAODepthCompareFunction
{
  using StorageType = plUInt8;
 
  enum Enum
  {
    Depth,                   
    Normal,                  
    NormalAndSampleDistance, 
    Default = NormalAndSampleDistance
  };
};
 
PL_DECLARE_REFLECTABLE_TYPE(PL_RENDERERCORE_DLL, plLSAODepthCompareFunction);
 
class PL_RENDERERCORE_DLL plLSAOPass : public plRenderPipelinePass
{
  PL_ADD_DYNAMIC_REFLECTION(plLSAOPass, plRenderPipelinePass);
 
public:
  plLSAOPass();
  ~plLSAOPass();
 
  virtual bool GetRenderTargetDescriptions(const plView& view, const plArrayPtr<plGALTextureCreationDescription* const> inputs, plArrayPtr<plGALTextureCreationDescription> outputs) override;
  virtual void InitRenderPipelinePass(const plArrayPtr<plRenderPipelinePassConnection* const> inputs, const plArrayPtr<plRenderPipelinePassConnection* const> outputs) override;
 
  virtual void Execute(const plRenderViewContext& renderViewContext, const plArrayPtr<plRenderPipelinePassConnection* const> inputs, const plArrayPtr<plRenderPipelinePassConnection* const> outputs) override;
  virtual void ExecuteInactive(const plRenderViewContext& renderViewContext, const plArrayPtr<plRenderPipelinePassConnection* const> inputs, const plArrayPtr<plRenderPipelinePassConnection* const> outputs) override;
  virtual plResult Serialize(plStreamWriter& inout_stream) const override;
  virtual plResult Deserialize(plStreamReader& inout_stream) override;
 
  plUInt32 GetLineToLinePixelOffset() const { return m_iLineToLinePixelOffset; }
  void SetLineToLinePixelOffset(plUInt32 uiPixelOffset);
  plUInt32 GetLineSamplePixelOffset() const { return m_iLineSamplePixelOffsetFactor; }
  void SetLineSamplePixelOffset(plUInt32 uiPixelOffset);
 
  // Factor used for depth cutoffs (determines when a depth difference is too large to be considered)
  float GetDepthCutoffDistance() const;
  void SetDepthCutoffDistance(float fDepthCutoffDistance);
 
  // Determines how quickly the occlusion falls of.
  float GetOcclusionFalloff() const;
  void SetOcclusionFalloff(float fFalloff);
 
 
protected:
  void DestroyLineSweepData();
  void SetupLineSweepData(const plVec3I32& imageResolution);
 
 
  void AddLinesForDirection(const plVec3I32& imageResolution, const plVec2I32& sampleDir, plUInt32 lineIndex, plDynamicArray<LineInstruction>& outinLineInstructions, plUInt32& outinTotalNumberOfSamples);
 
  plRenderPipelineNodeInputPin m_PinDepthInput;
  plRenderPipelineNodeOutputPin m_PinOutput;
 
  plConstantBufferStorageHandle m_hLineSweepCB;
 
  bool m_bSweepDataDirty = true;
  bool m_bConstantsDirty = true;
 
  plGALBufferHandle m_hLineSweepOutputBuffer;
  plGALBufferUnorderedAccessViewHandle m_hLineSweepOutputUAV;
  plGALBufferResourceViewHandle m_hLineSweepOutputSRV;
 
  plGALBufferHandle m_hLineInfoBuffer;
  plGALBufferResourceViewHandle m_hLineSweepInfoSRV;
 
  plUInt32 m_uiNumSweepLines = 0;
 
  plInt32 m_iLineToLinePixelOffset = 2;
  plInt32 m_iLineSamplePixelOffsetFactor = 1;
  float m_fOcclusionFalloff = 0.2f;
  float m_fDepthCutoffDistance = 4.0f;
 
  plEnum<plLSAODepthCompareFunction> m_DepthCompareFunction;
  bool m_bDistributedGathering = true;
 
  plShaderResourceHandle m_hShaderLineSweep;
  plShaderResourceHandle m_hShaderGather;
  plShaderResourceHandle m_hShaderAverage;
};