pleascach/Renderer/CommandBuffer.cpp
2024-02-24 00:18:55 -05:00

109 lines
3.2 KiB
C++

#include <Renderer/CommandBuffer.hpp>
#include <Renderer/Pipeline.hpp>
#include <Renderer/VertexBuffer.hpp>
#include <Scene/Terrain.hpp>
#include <Model/Model.hpp>
#include <Memory/Buffer.hpp>
#include <Memory/Image.hpp>
#include <util/log.hpp>
CommandBuffer::CommandBuffer(vk::Device dev, u32 queue_family) {
Log::debug("Command Buffer created\n");
/* (For now) allow command buffers to be individually recycled */
auto pool_info = vk::CommandPoolCreateInfo {
.flags = vk::CommandPoolCreateFlagBits::eResetCommandBuffer
| vk::CommandPoolCreateFlagBits::eTransient,
.queueFamilyIndex = queue_family,
};
command_pool = dev.createCommandPool(pool_info);
/* for now, only using primary level command buffers */
auto alloc_info = vk::CommandBufferAllocateInfo{
.commandPool = command_pool,
.level = vk::CommandBufferLevel::ePrimary,
.commandBufferCount = 1,
};
command_buffer = dev.allocateCommandBuffers(alloc_info)[0];
}
void CommandBuffer::begin() {
auto begin_info = vk::CommandBufferBeginInfo{
.flags = static_cast<vk::CommandBufferUsageFlags>(0),
};
command_buffer.begin(begin_info);
}
void CommandBuffer::copy(vk::Buffer src, vk::Buffer dst, vk::ArrayProxy<const vk::BufferCopy> regions) {
command_buffer.copyBuffer(src, dst, regions);
}
void CommandBuffer::copy(Buffer& src, Image& dst, vk::ImageLayout layout) {
auto region = vk::BufferImageCopy{
.bufferOffset = 0,
/* RowLength and ImageHeight are automatically set to imageExtent dimensions if set to 0 */
.bufferRowLength = 0,
.bufferImageHeight = 0,
.imageSubresource = {
.aspectMask = vk::ImageAspectFlagBits::eColor,
.mipLevel = 0,
.baseArrayLayer = 0,
.layerCount = 1,
},
.imageOffset = { 0, 0, 0 },
.imageExtent = dst.extent,
};
command_buffer.copyBufferToImage(src, dst.image, layout, region);
}
void CommandBuffer::bind(const GraphicsPipeline& pipeline) {
command_buffer.bindPipeline(vk::PipelineBindPoint::eGraphics, pipeline.pipeline);
command_buffer.bindDescriptorSets(vk::PipelineBindPoint::eGraphics, pipeline.layout, 0, pipeline.desc_set, nullptr);
}
void CommandBuffer::bind(vk::PipelineLayout layout, vk::ArrayProxy<vk::DescriptorSet> desc_sets) {
command_buffer.bindDescriptorSets(vk::PipelineBindPoint::eGraphics, layout, 0, desc_sets, nullptr);
}
void CommandBuffer::bind(std::shared_ptr<Model> model) {
bind(*model->vertex_buffer);
command_buffer.bindIndexBuffer(*model->index_buffer, 0, vk::IndexType::eUint16);
}
void CommandBuffer::bind(Terrain* terrain) {
bind(*terrain->vertex_buffer);
command_buffer.bindIndexBuffer(*terrain->index_buffer, 0, vk::IndexType::eUint32);
}
void CommandBuffer::bind(HLBSP::BSP* bsp) {
bind(*bsp->pipeline);
bind(*bsp->vertex_buffer);
bind(bsp->pipeline->layout, bsp->pipeline->desc_set);
}
void CommandBuffer::draw(uint32_t vertex_count, uint32_t instance_count, uint32_t first_vertex, uint32_t first_instance) {
command_buffer.draw(vertex_count, instance_count, first_vertex, first_instance);
}
void CommandBuffer::end() {
command_buffer.end();
}
void CommandBuffer::recycle() {
command_buffer.reset();
}
void CommandBuffer::cleanup(vk::Device dev) {
dev.freeCommandBuffers(command_pool, command_buffer);
dev.destroyCommandPool(command_pool);
}