command_buffer.h

#pragma once
 
#include <volk.h>
 
#include <memory>
 
namespace pixelbullet
{
class CommandPool;
class LogicalDevice;
class RenderDevice;
 
class CommandBuffer
{
public:
    explicit CommandBuffer(const RenderDevice& render_device, bool begin = true, VkQueueFlagBits queueType = VK_QUEUE_GRAPHICS_BIT,
                           VkCommandBufferLevel bufferLevel = VK_COMMAND_BUFFER_LEVEL_PRIMARY);
 
    ~CommandBuffer();
 
    void Reset(VkCommandBufferResetFlags flags = 0);
 
    void Begin(VkCommandBufferUsageFlags usage = VK_COMMAND_BUFFER_USAGE_ONE_TIME_SUBMIT_BIT);
 
    void End();
 
    void SubmitIdle();
 
    void Submit(const VkSemaphore& waitSemaphore = VK_NULL_HANDLE, const VkSemaphore& signalSemaphore = VK_NULL_HANDLE,
                VkFence fence = VK_NULL_HANDLE);
 
    operator const VkCommandBuffer&() const
    {
        return command_buffer_;
    }
 
    const VkCommandBuffer& GetCommandBuffer() const
    {
        return command_buffer_;
    }
    bool IsRunning() const
    {
        return running_;
    }
 
private:
    VkQueue GetQueue() const;
 
private:
    std::shared_ptr<CommandPool> command_pool_;
    const LogicalDevice* logical_device_;
 
    VkQueueFlagBits queue_type_;
    VkCommandBuffer command_buffer_ = VK_NULL_HANDLE;
    bool running_ = false;
};
} // namespace pixelbullet