mirror of
https://github.com/Kbz-8/Pulse.git
synced 2026-01-11 15:33:34 +00:00
111 lines
4.1 KiB
C
111 lines
4.1 KiB
C
// Copyright (C) 2025 kanel
|
|
// This file is part of "Pulse"
|
|
// For conditions of distribution and use, see copyright notice in LICENSE
|
|
|
|
#include "Vulkan.h"
|
|
#include "VulkanDevice.h"
|
|
#include "VulkanComputePass.h"
|
|
#include "VulkanComputePipeline.h"
|
|
|
|
PulseComputePass VulkanCreateComputePass(PulseDevice device, PulseCommandList cmd)
|
|
{
|
|
PulseComputePass pass = (PulseComputePass)calloc(1, sizeof(PulseComputePassHandler));
|
|
PULSE_CHECK_ALLOCATION_RETVAL(pass, PULSE_NULL_HANDLE);
|
|
|
|
VulkanComputePass* vulkan_pass = (VulkanComputePass*)calloc(1, sizeof(VulkanComputePass));
|
|
PULSE_CHECK_ALLOCATION_RETVAL(vulkan_pass, PULSE_NULL_HANDLE);
|
|
|
|
VulkanDevice* vulkan_device = VULKAN_RETRIEVE_DRIVER_DATA_AS(device, VulkanDevice*);
|
|
|
|
// vulkan_pass->read_only_descriptor_set = VulkanRequestDescriptorSetFromPool(
|
|
// VulkanGetAvailableDescriptorSetPool(&vulkan_device->descriptor_set_pool_manager),
|
|
// VulkanGetDescriptorSetLayout(&vulkan_device->descriptor_set_layout_manager, ));
|
|
|
|
pass->cmd = cmd;
|
|
pass->driver_data = vulkan_pass;
|
|
|
|
return pass;
|
|
}
|
|
|
|
void VulkanDestroyComputePass(PulseDevice device, PulseComputePass pass)
|
|
{
|
|
(void)device; // Maybe reserved for future use
|
|
free(pass->driver_data);
|
|
free(pass);
|
|
}
|
|
|
|
void VulkanBindStorageBuffers(PulseComputePass pass, uint32_t starting_slot, const PulseBuffer* buffers, uint32_t num_buffers)
|
|
{
|
|
PulseBufferUsageFlags usage = buffers[0]->usage;
|
|
PulseBuffer* array = ((usage & PULSE_BUFFER_USAGE_STORAGE_WRITE) == 1) ? pass->readwrite_storage_buffers : pass->readonly_storage_buffers;
|
|
VulkanComputePass* vulkan_pass = VULKAN_RETRIEVE_DRIVER_DATA_AS(pass, VulkanComputePass*);
|
|
|
|
for(uint32_t i = 0; i < num_buffers; i++)
|
|
{
|
|
if(array[starting_slot + i] == buffers[i])
|
|
continue;
|
|
array[starting_slot + i] = buffers[i];
|
|
|
|
if((usage & PULSE_BUFFER_USAGE_STORAGE_WRITE) == 1)
|
|
vulkan_pass->should_recreate_write_descriptor_sets = true;
|
|
else
|
|
vulkan_pass->should_recreate_read_only_descriptor_sets = true;
|
|
}
|
|
}
|
|
|
|
void VulkanBindUniformData(PulseComputePass pass, uint32_t slot, const void* data, uint32_t data_size)
|
|
{
|
|
}
|
|
|
|
void VulkanBindStorageImages(PulseComputePass pass, uint32_t starting_slot, const PulseImage* images, uint32_t num_images)
|
|
{
|
|
PulseImageUsageFlags usage = images[0]->usage;
|
|
PulseImage* array = ((usage & PULSE_IMAGE_USAGE_STORAGE_WRITE) == 1) ? pass->readwrite_images : pass->readonly_images;
|
|
VulkanComputePass* vulkan_pass = VULKAN_RETRIEVE_DRIVER_DATA_AS(pass, VulkanComputePass*);
|
|
|
|
for(uint32_t i = 0; i < num_images; i++)
|
|
{
|
|
if(array[starting_slot + i] == images[i])
|
|
continue;
|
|
array[starting_slot + i] = images[i];
|
|
|
|
if((usage & PULSE_IMAGE_USAGE_STORAGE_WRITE) == 1)
|
|
vulkan_pass->should_recreate_write_descriptor_sets = true;
|
|
else
|
|
vulkan_pass->should_recreate_read_only_descriptor_sets = true;
|
|
}
|
|
}
|
|
|
|
void VulkanBindComputePipeline(PulseComputePass pass, PulseComputePipeline pipeline)
|
|
{
|
|
VulkanComputePipeline* vulkan_pipeline = VULKAN_RETRIEVE_DRIVER_DATA_AS(pipeline, VulkanComputePipeline*);
|
|
VulkanDevice* vulkan_device = VULKAN_RETRIEVE_DRIVER_DATA_AS(pass->cmd->device, VulkanDevice*);
|
|
VulkanCommandList* vulkan_cmd = VULKAN_RETRIEVE_DRIVER_DATA_AS(pass->cmd, VulkanCommandList*);
|
|
VulkanComputePass* vulkan_pass = VULKAN_RETRIEVE_DRIVER_DATA_AS(pass, VulkanComputePass*);
|
|
|
|
vulkan_device->vkCmdBindPipeline(vulkan_cmd->cmd, VK_PIPELINE_BIND_POINT_COMPUTE, vulkan_pipeline->pipeline);
|
|
|
|
vulkan_pass->should_recreate_read_only_descriptor_sets = true;
|
|
vulkan_pass->should_recreate_write_descriptor_sets = true;
|
|
vulkan_pass->should_recreate_uniform_descriptor_sets = true;
|
|
}
|
|
|
|
void VulkanDispatchComputations(PulseComputePass pass, uint32_t groupcount_x, uint32_t groupcount_y, uint32_t groupcount_z)
|
|
{
|
|
VulkanDevice* vulkan_device = VULKAN_RETRIEVE_DRIVER_DATA_AS(pass->cmd->device, VulkanDevice*);
|
|
VulkanCommandList* vulkan_cmd = VULKAN_RETRIEVE_DRIVER_DATA_AS(pass->cmd, VulkanCommandList*);
|
|
|
|
VulkanBindDescriptorSets(pass);
|
|
|
|
vulkan_device->vkCmdDispatch(vulkan_cmd->cmd, groupcount_x, groupcount_y, groupcount_z);
|
|
}
|
|
|
|
PulseComputePass VulkanBeginComputePass(PulseCommandList cmd)
|
|
{
|
|
return cmd->pass;
|
|
}
|
|
|
|
void VulkanEndComputePass(PulseComputePass pass)
|
|
{
|
|
}
|