diff options
Diffstat (limited to 'src/vk-renderer.cpp')
| -rw-r--r-- | src/vk-renderer.cpp | 284 |
1 files changed, 0 insertions, 284 deletions
diff --git a/src/vk-renderer.cpp b/src/vk-renderer.cpp deleted file mode 100644 index ddff268..0000000 --- a/src/vk-renderer.cpp +++ /dev/null @@ -1,284 +0,0 @@ -#include "vk-renderer.hpp" - -#include <cstring> -#include <cassert> - -#include <vk_mem_alloc.h> - -#include "vulkan-loader.hpp" - -#include "compute-renderpass.hpp" -#include "primitive-renderpass.hpp" - -#include "texture.hpp" - -#include "window.hpp" - -#ifdef _DEBUG -#define VKRESULT(result) assert(result == VK_SUCCESS); -#else -#define VKRESULT(result) result; -#endif - -auto vkrenderer::context = vkcontext(); -auto vkrenderer::api = vkapi(vkrenderer::context); -auto vkrenderer::upload_queue = std::vector<Texture*>(); - -vkrenderer::vkrenderer(window& wnd) { - platform_surface = api.create_surface(wnd); - swapchain = api.create_swapchain( - platform_surface, - min_swapchain_image_count, - VK_IMAGE_USAGE_COLOR_ATTACHMENT_BIT | - VK_IMAGE_USAGE_TRANSFER_DST_BIT | - VK_IMAGE_USAGE_STORAGE_BIT, - VK_NULL_HANDLE - ); - - swapchain_textures.resize(swapchain.image_count); - for (size_t index = 0; index < swapchain.image_count; index++) { - swapchain_textures[index] = new Texture(swapchain.images[index]); - } - - tonemapping_pipeline = api.create_compute_pipeline("tonemapping"); - - graphics_command_pool = api.create_command_pool(); - api.allocate_command_buffers(graphics_command_pool, graphics_command_buffers, virtual_frames_count); - - for(size_t index { 0 }; index < virtual_frames_count; index++) { - submission_fences[index] = api.create_fence(); - execution_semaphores[index] = api.create_semaphore(); - acquire_semaphores[index] = api.create_semaphore(); - - copy_command_pools[index] = api.create_command_pool(); - api.allocate_command_buffers(copy_command_pools[index], ©_command_buffers[index], 1); - - // staging_buffers[index] = create_staging_buffer(4096 * 4096 * sizeof(uint32_t)); - staging_buffers[index] = new RingBuffer(4096ULL * 4096ULL * sizeof(uint32_t)); - } -} - -vkrenderer::~vkrenderer() { - VKRESULT(vkWaitForFences(context.device, virtual_frames_count, submission_fences, VK_TRUE, UINT64_MAX)) - - for (auto& renderpass : renderpasses) { - delete renderpass; - } - - for(size_t index { 0 }; index < virtual_frames_count; index++) { - vkFreeCommandBuffers(context.device, copy_command_pools[index], 1, ©_command_buffers[index]); - vkDestroyCommandPool(context.device, copy_command_pools[index], nullptr); - } - - api.destroy_fences(submission_fences, virtual_frames_count); - api.destroy_semaphores(execution_semaphores, virtual_frames_count); - api.destroy_semaphores(acquire_semaphores, virtual_frames_count); - - vkFreeCommandBuffers(context.device, graphics_command_pool, virtual_frames_count, graphics_command_buffers); - vkDestroyCommandPool(context.device, graphics_command_pool, nullptr); - - - api.destroy_pipeline(tonemapping_pipeline); - - api.destroy_swapchain(swapchain); - api.destroy_surface(platform_surface); -} -void vkrenderer::recreate_swapchain() { - VKRESULT(vkWaitForFences(context.device, vkrenderer::virtual_frames_count, submission_fences, VK_TRUE, UINT64_MAX)) - - auto swapchain_image_usages = api.get_image(swapchain.images[0]).usages; - struct swapchain old_swapchain = swapchain; - swapchain = api.create_swapchain( - platform_surface, - min_swapchain_image_count, - swapchain_image_usages, - old_swapchain.handle - ); - - api.destroy_swapchain(old_swapchain); - - swapchain_textures.resize(swapchain.image_count); - for (size_t index = 0; index < swapchain.image_count; index++) { - swapchain_textures[index] = new Texture(swapchain.images[index]); - } -} - -void vkrenderer::queue_image_update(Texture* texture) { - upload_queue.push_back(texture); -} - -void vkrenderer::update_images() { - if(upload_queue.empty()) - return; - - auto* staging_buffer = staging_buffers[virtual_frame_index]; - staging_buffer->reset(); - - VkCommandBuffer command_buffer = copy_command_buffers[virtual_frame_index]; - vkrenderer::api.start_record(command_buffer); - - while(!upload_queue.empty()) { - auto* texture = upload_queue.back(); - auto texture_size = texture->size(); - auto offset = staging_buffer->alloc(texture_size); - - if (offset == RingBuffer::invalid_alloc) - break; - - upload_queue.pop_back(); - - staging_buffer->write(texture->data, offset, texture_size); - - vkrenderer::api.image_barrier(command_buffer, VK_IMAGE_LAYOUT_TRANSFER_DST_OPTIMAL, VK_PIPELINE_STAGE_TOP_OF_PIPE_BIT, 0, texture->device_image); - vkrenderer::api.copy_buffer(command_buffer, staging_buffer->device_buffer, texture->device_image, 0, offset); - vkrenderer::api.image_barrier(command_buffer, VK_IMAGE_LAYOUT_SHADER_READ_ONLY_OPTIMAL, VK_PIPELINE_STAGE_TOP_OF_PIPE_BIT, 0, texture->device_image); - } - - vkrenderer::api.end_record(command_buffer); - - recorded_command_buffers.push_back(command_buffer); -} - -Buffer* vkrenderer::create_buffer(size_t size) { - return new Buffer(size, VK_BUFFER_USAGE_STORAGE_BUFFER_BIT); -} - -Buffer* vkrenderer::create_index_buffer(size_t size) { - return new Buffer(size, VK_BUFFER_USAGE_INDEX_BUFFER_BIT); -} - -Buffer* vkrenderer::create_staging_buffer(size_t size) { - return new Buffer(size, VK_BUFFER_USAGE_TRANSFER_SRC_BIT); -} - -Texture* vkrenderer::create_2d_texture(size_t width, size_t height, VkFormat format, Sampler *sampler) { - return new Texture(width, height, 1, format, sampler); -} - -// TODO: Check if a similar sampler has been allocated -Sampler* vkrenderer::create_sampler(VkFilter filter, VkSamplerAddressMode address_mode) { - auto* sampler = new Sampler(); - - sampler->device_sampler = api.create_sampler(filter, address_mode); - - return sampler; -} - -ComputeRenderpass* vkrenderer::create_compute_renderpass() { - renderpasses.push_back(new ComputeRenderpass(api)); - - return (ComputeRenderpass*)renderpasses.back(); -} - -PrimitiveRenderpass* vkrenderer::create_primitive_renderpass() { - renderpasses.push_back(new PrimitiveRenderpass(api)); - - return (PrimitiveRenderpass*)renderpasses.back(); -} - -Primitive* vkrenderer::create_primitive(PrimitiveRenderpass& primitive_render_pass) { - return new Primitive(primitive_render_pass); -} - -void vkrenderer::render() { - swapchain_image_index = 0; - auto acquire_result = vkAcquireNextImageKHR(context.device, swapchain.handle, UINT64_MAX, acquire_semaphores[virtual_frame_index], VK_NULL_HANDLE, &swapchain_image_index); - handle_swapchain_result(acquire_result); - - api.start_record(graphics_command_buffers[virtual_frame_index]); - - auto* cmd_buf = graphics_command_buffers[virtual_frame_index]; - - // wait for transfer operations - - for (auto& renderpass: renderpasses) { - renderpass->execute(*this, cmd_buf); - - api.image_barrier(cmd_buf, VK_IMAGE_LAYOUT_TRANSFER_SRC_OPTIMAL, VK_PIPELINE_STAGE_TRANSFER_BIT, VK_ACCESS_TRANSFER_READ_BIT, ((ComputeRenderpass*)renderpass)->output_texture->device_image); - - api.image_barrier(cmd_buf, VK_IMAGE_LAYOUT_TRANSFER_DST_OPTIMAL, VK_PIPELINE_STAGE_TRANSFER_BIT, VK_ACCESS_TRANSFER_WRITE_BIT, swapchain.images[swapchain_image_index]); - - api.blit_full(cmd_buf, ((ComputeRenderpass*)renderpass)->output_texture->device_image, swapchain.images[swapchain_image_index]); - } - - - // api.image_barrier(cmd_buf, VK_IMAGE_LAYOUT_COLOR_ATTACHMENT_OPTIMAL, VK_PIPELINE_STAGE_COLOR_ATTACHMENT_OUTPUT_BIT, VK_ACCESS_COLOR_ATTACHMENT_WRITE_BIT, swapchain.images[swapchain_image_index]); - - // renderpasses[1]->execute(*this, cmd_buf); - - api.image_barrier(cmd_buf, VK_IMAGE_LAYOUT_PRESENT_SRC_KHR, VK_PIPELINE_STAGE_TOP_OF_PIPE_BIT, 0, swapchain.images[swapchain_image_index]); - - api.end_record(cmd_buf); - - recorded_command_buffers.push_back(cmd_buf); -} - -void vkrenderer::begin_frame() { - recorded_command_buffers.clear(); - - VKRESULT(vkWaitForFences(context.device, 1, &submission_fences[virtual_frame_index], VK_TRUE, UINT64_MAX)) - VKRESULT(vkResetFences(context.device, 1, &submission_fences[virtual_frame_index])) - - update_images(); -} - -void vkrenderer::finish_frame() { - api.submit(recorded_command_buffers.data(), recorded_command_buffers.size(), acquire_semaphores[virtual_frame_index], execution_semaphores[virtual_frame_index], submission_fences[virtual_frame_index]); - - auto present_result = api.present(swapchain, swapchain_image_index, execution_semaphores[virtual_frame_index]); - handle_swapchain_result(present_result); - - virtual_frame_index = ++virtual_frame_index % virtual_frames_count; -} - - - -// private functions -void vkrenderer::handle_swapchain_result(VkResult function_result) { - switch(function_result) { - case VK_ERROR_OUT_OF_DATE_KHR: - return recreate_swapchain(); - case VK_ERROR_SURFACE_LOST_KHR: - assert(true && "surface lost"); - default: - return; - } -} - - - -Buffer::Buffer(size_t size, VkBufferUsageFlagBits usage) - : buffer_size(size), usage(usage) { - device_buffer = vkrenderer::api.create_buffer(size, usage, VMA_MEMORY_USAGE_CPU_TO_GPU); -} - -Buffer::~Buffer() { - vkrenderer::api.destroy_buffer(device_buffer); -} - -void Buffer::write(void* data, off_t alloc_offset, size_t data_size) const { - auto api_buffer = vkrenderer::api.get_buffer(device_buffer); - - std::memcpy((uint8_t*)api_buffer.device_ptr + alloc_offset, data, data_size); -} - -void Buffer::resize(size_t new_size) { - auto old_device_buffer = device_buffer; - new_size = static_cast<uint32_t>(exp2(ceil(log2((double)new_size)))); - - device_buffer = vkrenderer::api.create_buffer(new_size, usage, VMA_MEMORY_USAGE_CPU_TO_GPU); - - auto old_buffer = vkrenderer::api.get_buffer(device_buffer); - auto new_buffer = vkrenderer::api.get_buffer(device_buffer); - std::memcpy(new_buffer.device_ptr, old_buffer.device_ptr, buffer_size); - - buffer_size = new_size; - - vkrenderer::api.destroy_buffer(old_device_buffer); -} - - -RingBuffer::RingBuffer(size_t size) - : buffer_size(size) { - device_buffer = vkrenderer::api.create_buffer(size, VK_BUFFER_USAGE_TRANSFER_SRC_BIT, VMA_MEMORY_USAGE_CPU_ONLY); - } |