CommandBuffer.hpp

#pragma once
 
#include "PixelBullet/Graphics/Commands/CommandPool.hpp"
 
#include <memory>
 
namespace PixelBullet
{
    class LogicalDevice;
 
    class CommandBuffer
    {
    public:
        explicit CommandBuffer(const LogicalDevice* logicalDevice, bool begin = true,
                               VkQueueFlagBits queueType = VK_QUEUE_GRAPHICS_BIT,
                               VkCommandBufferLevel bufferLevel = VK_COMMAND_BUFFER_LEVEL_PRIMARY);
 
        ~CommandBuffer();
 
        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 m_CommandBuffer;
        }
 
        const VkCommandBuffer& GetCommandBuffer() const
        {
            return m_CommandBuffer;
        }
        bool IsRunning() const
        {
            return m_Running;
        }
 
    private:
        VkQueue GetQueue() const;
 
    private:
        std::shared_ptr<CommandPool> m_CommandPool;
        const LogicalDevice* m_LogicalDevice;
 
        VkQueueFlagBits m_QueueType;
        VkCommandBuffer m_CommandBuffer = VK_NULL_HANDLE;
        bool m_Running = false;
    };
} // namespace PixelBullet