Code/Engine/RendererVulkan/Pools/Implementation/CommandBufferPoolVulkan.cpp - dev - redgpu/ezEngine

Code/Engine/RendererVulkan/Pools/Implementation/CommandBufferPoolVulkan.cpp

71 строка2 KB

Sanakan8472

Copy / Readback fixes (#1963)

11 июн 2026, 18:07

11 июн 2026, 18:07b50d6f1

#include <RendererVulkan/RendererVulkanPCH.h>

#include <RendererVulkan/Pools/CommandBufferPoolVulkan.h>

ezCommandBufferPoolVulkan::ezCommandBufferPoolVulkan(ezAllocator* pAllocator)
  : m_CommandBuffers(pAllocator)
{
}

ezCommandBufferPoolVulkan::~ezCommandBufferPoolVulkan()
{
  EZ_ASSERT_DEBUG(m_CommandBuffers.IsEmpty(), "Either DeInitialize was not called or ReclaimCommandBuffer was called after DeInitialize");
  EZ_ASSERT_DEBUG(!m_Device, "DeInitialize was not called");
}

void ezCommandBufferPoolVulkan::Initialize(vk::Device device, ezUInt32 uiGraphicsFamilyIndex)
{
  m_Device = device;

  // Command buffer
  vk::CommandPoolCreateInfo commandPoolCreateInfo = {};
  commandPoolCreateInfo.flags = vk::CommandPoolCreateFlagBits::eResetCommandBuffer;
  commandPoolCreateInfo.queueFamilyIndex = uiGraphicsFamilyIndex;

  m_CommandPool = m_Device.createCommandPool(commandPoolCreateInfo);
}

void ezCommandBufferPoolVulkan::DeInitialize()
{
  for (vk::CommandBuffer& commandBuffer : m_CommandBuffers)
  {
    m_Device.freeCommandBuffers(m_CommandPool, 1, &commandBuffer);
  }
  m_CommandBuffers.Clear();
  m_CommandBuffers.Compact();

  m_Device.destroyCommandPool(m_CommandPool);
  m_CommandPool = nullptr;

  m_Device = nullptr;
}

vk::CommandBuffer ezCommandBufferPoolVulkan::RequestCommandBuffer()
{
  EZ_ASSERT_DEBUG(m_Device, "ezCommandBufferPoolVulkan::Initialize not called");
  if (!m_CommandBuffers.IsEmpty())
  {
    vk::CommandBuffer CommandBuffer = m_CommandBuffers.PeekBack();
    m_CommandBuffers.PopBack();
    return CommandBuffer;
  }
  else
  {
    vk::CommandBuffer commandBuffer;

    vk::CommandBufferAllocateInfo commandBufferAllocateInfo = {};
    commandBufferAllocateInfo.commandBufferCount = 1;
    commandBufferAllocateInfo.commandPool = m_CommandPool;
    commandBufferAllocateInfo.level = vk::CommandBufferLevel::ePrimary;

    VK_ASSERT_DEV(m_Device.allocateCommandBuffers(&commandBufferAllocateInfo, &commandBuffer));
    return commandBuffer;
  }
}

void ezCommandBufferPoolVulkan::ReclaimCommandBuffer(vk::CommandBuffer& ref_commandBuffer)
{
  EZ_ASSERT_DEBUG(m_Device, "ezCommandBufferPoolVulkan::Initialize not called");
  ref_commandBuffer.reset(vk::CommandBufferResetFlagBits::eReleaseResources);
  m_CommandBuffers.PushBack(ref_commandBuffer);
}