ReflectionProbeUpdater.h

#pragma once
 
#include <RendererCore/RendererCorePCH.h>
 
#include <Core/Graphics/Camera.h>
#include <RendererCore/Lights/Implementation/ReflectionProbeData.h>
#include <RendererCore/Pipeline/Declarations.h>
#include <RendererFoundation/RendererFoundationDLL.h>
 
PL_DECLARE_FLAGS(plUInt8, plReflectionProbeUpdaterFlags, SkyLight, HasCustomCubeMap);
 
class plReflectionProbeUpdater
{
public:
  struct TargetSlot
  {
    plGALTextureHandle m_hSpecularOutputTexture;   
    plGALTextureHandle m_hIrradianceOutputTexture; 
    plInt32 m_iSpecularOutputIndex = -1;           
    plInt32 m_iIrradianceOutputIndex = -1;         
  };
 
public:
  plReflectionProbeUpdater();
  ~plReflectionProbeUpdater();
 
  plUInt32 GetFreeUpdateSlots(plDynamicArray<plReflectionProbeRef>& out_updatesFinished);
 
  plResult StartDynamicUpdate(const plReflectionProbeRef& probe, const plReflectionProbeDesc& desc, const plTransform& globalTransform, const TargetSlot& target);
 
  plResult StartFilterUpdate(const plReflectionProbeRef& probe, const plReflectionProbeDesc& desc, plTextureCubeResourceHandle hSourceTexture, const TargetSlot& target);
 
  void CancelUpdate(const plReflectionProbeRef& probe);
 
  void GenerateUpdateSteps();
 
  void ScheduleUpdateSteps();
 
private:
  struct ReflectionView
  {
    plViewHandle m_hView;
    plCamera m_Camera;
  };
 
  struct UpdateStep
  {
    using StorageType = plUInt8;
 
    enum Enum
    {
      RenderFace0,
      RenderFace1,
      RenderFace2,
      RenderFace3,
      RenderFace4,
      RenderFace5,
      Filter,
 
      ENUM_COUNT,
 
      Default = Filter
    };
 
    static bool IsRenderStep(Enum value) { return value >= UpdateStep::RenderFace0 && value <= UpdateStep::RenderFace5; }
    static Enum NextStep(Enum value) { return static_cast<UpdateStep::Enum>((value + 1) % UpdateStep::ENUM_COUNT); }
  };
 
  struct ProbeUpdateInfo
  {
    ProbeUpdateInfo();
    ~ProbeUpdateInfo();
 
    plBitflags<plReflectionProbeUpdaterFlags> m_flags;
    plReflectionProbeRef m_probe;
    plReflectionProbeDesc m_desc;
    plTransform m_globalTransform;
    plTextureCubeResourceHandle m_sourceTexture;
    TargetSlot m_TargetSlot;
 
    struct Step
    {
      PL_DECLARE_POD_TYPE();
 
      plUInt8 m_uiViewIndex;
      plEnum<UpdateStep> m_UpdateStep;
    };
 
    bool m_bInUse = false;
    plEnum<UpdateStep> m_LastUpdateStep;
 
    plHybridArray<Step, 8> m_UpdateSteps;
 
    plGALTextureHandle m_hCubemap;
    plGALTextureHandle m_hCubemapProxies[6];
  };
 
private:
  static void CreateViews(
    plDynamicArray<ReflectionView>& views, plUInt32 uiMaxRenderViews, const char* szNameSuffix, const char* szRenderPipelineResource);
  void CreateReflectionViewsAndResources();
 
  void ResetProbeUpdateInfo(plUInt32 uiInfo);
  void AddViewToRender(const ProbeUpdateInfo::Step& step, ProbeUpdateInfo& updateInfo);
 
  bool m_bUpdateStepsFlushed = true;
 
  plDynamicArray<ReflectionView> m_RenderViews;
  plDynamicArray<ReflectionView> m_FilterViews;
 
  // Active Dynamic Updates
  plDynamicArray<plUniquePtr<ProbeUpdateInfo>> m_DynamicUpdates;
  plHybridArray<plReflectionProbeRef, 4> m_FinishedLastFrame;
};