QueryPoolVulkan.h

#pragma once
 
#include <RendererVulkan/RendererVulkanDLL.h>
 
#include <vulkan/vulkan.hpp>
 
class plGALDeviceVulkan;
 
class PL_RENDERERVULKAN_DLL plQueryPoolVulkan
{
public:
  plQueryPoolVulkan(plGALDeviceVulkan* pDevice);
 
  void Initialize(plUInt32 uiValidBits);
  void DeInitialize();
 
  void BeginFrame(vk::CommandBuffer commandBuffer);
 
  plGALTimestampHandle InsertTimestamp(vk::CommandBuffer commandBuffer, vk::PipelineStageFlagBits pipelineStage = vk::PipelineStageFlagBits::eBottomOfPipe);
 
  plEnum<plGALAsyncResult> GetTimestampResult(plGALTimestampHandle hTimestamp, plTime& out_result, bool bForce = false);
 
  plGALPoolHandle BeginOcclusionQuery(vk::CommandBuffer commandBuffer, plEnum<plGALQueryType> type);
  void EndOcclusionQuery(vk::CommandBuffer commandBuffer, plGALPoolHandle hPool);
  plEnum<plGALAsyncResult> GetOcclusionQueryResult(plGALPoolHandle hPool, plUInt64& out_uiQueryResult, bool bForce = false);
 
private:
  void Calibrate();
 
  static constexpr plUInt32 s_uiRetainFrames = 4;
 
  struct Query
  {
    vk::QueryPool m_pool;
    plUInt32 uiQueryIndex;
  };
 
  struct QueryPool
  {
    vk::QueryPool m_pool;
    bool m_bReady = false;
    plDynamicArray<uint64_t> m_queryResults;
  };
 
  struct FramePool
  {
    plUInt64 m_uiNextIndex = 0;    // Next query index in this frame. Use % and / to find the pool / element index.
    plUInt64 m_uiFrameCounter = 0; // plGALDevice::GetCurrentFrame
    plUInt8 m_uiReadyFrames = 0; 
    plHybridArray<QueryPool*, 2> m_pools;
  };
 
  struct Pool
  {
    Pool(plAllocator* pAllocator);
 
    void Initialize(vk::Device device, plUInt32 uiPoolSize, vk::QueryType queryType);
    void DeInitialize();
 
    QueryPool* GetFreePool();
    void BeginFrame(vk::CommandBuffer commandBuffer, plUInt64 uiCurrentFrame, plUInt64 uiSafeFrame);
    plGALPoolHandle CreateQuery(vk::CommandBuffer commandBuffer);
    Query GetQuery(plGALPoolHandle hPool);
    plEnum<plGALAsyncResult> GetResult(plGALPoolHandle hPool, plUInt64& out_uiResult, bool bForce);
 
    // Current active data.
    FramePool* m_pCurrentFrame = nullptr;
    plUInt64 m_uiFirstFrameIndex = 0;
    plDeque<FramePool> m_pendingFrames;
 
    // Pools.
    plHybridArray<vk::QueryPool, 8> m_resetPools;
    plHybridArray<QueryPool*, 8> m_freePools;
 
    vk::Device m_device;
    plUInt32 m_uiPoolSize = 0;
    vk::QueryType m_QueryType;
  };
 
  plGALDeviceVulkan* m_pDevice = nullptr;
 
  // Timestamp conversion and calibration data.
  double m_fNanoSecondsPerTick = 0;
  plUInt64 m_uiValidBitsMask = 0;
  plTime m_gpuToCpuDelta;
 
  // Pools
  Pool m_TimestampPool;
  Pool m_OcclusionPool;
};