QueryPoolDX11.h

#pragma once
 
#include <RendererDX11/RendererDX11DLL.h>
#include <d3d11.h>
 
struct ID3D11Query;
class plGALDeviceDX11;
 
class PL_RENDERERDX11_DLL plQueryPoolDX11
{
public:
  plQueryPoolDX11(plGALDeviceDX11* pDevice);
 
  plResult Initialize();
  void DeInitialize();
 
  void BeginFrame();
  void EndFrame();
 
  plGALTimestampHandle InsertTimestamp();
 
  plEnum<plGALAsyncResult> GetTimestampResult(plGALTimestampHandle hTimestamp, plTime& out_result);
 
  plGALPoolHandle BeginOcclusionQuery(plEnum<plGALQueryType> type);
  void EndOcclusionQuery(plGALPoolHandle hPool);
  plEnum<plGALAsyncResult> GetOcclusionQueryResult(plGALPoolHandle hPool, plUInt64& out_uiQueryResult);
 
private:
  static constexpr plUInt32 s_uiRetainFrames = 4;
  static constexpr plUInt64 s_uiPredicateFlag = PL_BIT(19);
  static constexpr double s_fInvalid = -1.0;
 
  struct PerFrameData
  {
    plGALFenceHandle m_hFence;
    ID3D11Query* m_pDisjointTimerQuery = nullptr;
    double m_fInvTicksPerSecond = s_fInvalid;
    plUInt32 m_uiReadyFrames = 0; 
    plUInt64 m_uiFrameCounter = plUInt64(-1);
  };
 
private:
  PerFrameData GetFreeFrame();
 
private:
  plGALDeviceDX11* m_pDevice = nullptr;
 
  struct Pool
  {
    Pool(plAllocator* pAllocator);
 
    plResult Initialize(plGALDeviceDX11* pDevice, D3D11_QUERY queryType, plUInt32 uiCount);
    void DeInitialize();
 
    plGALPoolHandle CreateQuery();
    ID3D11Query* GetQuery(plGALPoolHandle hPool);
    template <typename T>
    plEnum<plGALAsyncResult> GetResult(plGALPoolHandle hPool, T& out_uiResult)
    {
      ID3D11Query* pQuery = GetQuery(hPool);
      HRESULT res = m_pDevice->GetDXImmediateContext()->GetData(pQuery, &out_uiResult, sizeof(out_uiResult), D3D11_ASYNC_GETDATA_DONOTFLUSH);
      if (res == S_FALSE)
      {
        return plGALAsyncResult::Pending;
      }
      else if (res == S_OK)
      {
        return plGALAsyncResult::Ready;
      }
      else
      {
        return plGALAsyncResult::Expired;
      }
    }
 
    // #TODO_DX11 Replace ring buffer with proper pool like in Vulkan to prevent buffer overrun.
    plDynamicArray<ID3D11Query*, plLocalAllocatorWrapper> m_Queries;
    plUInt32 m_uiNextTimestamp = 0;
    plGALDeviceDX11* m_pDevice = nullptr;
  };
 
  // Pools
  Pool m_TimestampPool;
  Pool m_OcclusionPool;
  Pool m_OcclusionPredicatePool;
 
  // Disjoint timer and frame meta data needed for timestamps
  plDeque<PerFrameData> m_PendingFrames;
  plDeque<PerFrameData> m_FreeFrames;
  plUInt64 m_uiFirstFrameIndex = 0;
 
  plTime m_SyncTimeDiff;
  bool m_bSyncTimeNeeded = true;
};