diff --git a/src/resources.cpp b/src/resources.cpp index cb7cec0..e9d4ebf 100644 --- a/src/resources.cpp +++ b/src/resources.cpp @@ -53,6 +53,25 @@ void pipelineBarrier(VkCommandBuffer commandBuffer, VkDependencyFlags dependency vkCmdPipelineBarrier2(commandBuffer, &dependencyInfo); } +void stageBarrier(VkCommandBuffer commandBuffer, VkPipelineStageFlags2 stageMask) +{ + stageBarrier(commandBuffer, stageMask, stageMask); +} + +void stageBarrier(VkCommandBuffer commandBuffer, VkPipelineStageFlags2 srcStageMask, VkPipelineStageFlags2 dstStageMask) +{ + VkMemoryBarrier2 barrier = { VK_STRUCTURE_TYPE_MEMORY_BARRIER_2 }; + barrier.srcStageMask = srcStageMask; + barrier.dstStageMask = dstStageMask; + barrier.srcAccessMask = barrier.dstAccessMask = VK_ACCESS_MEMORY_READ_BIT | VK_ACCESS_MEMORY_WRITE_BIT; + + VkDependencyInfo dependencyInfo = { VK_STRUCTURE_TYPE_DEPENDENCY_INFO }; + dependencyInfo.memoryBarrierCount = 1; + dependencyInfo.pMemoryBarriers = &barrier; + + vkCmdPipelineBarrier2(commandBuffer, &dependencyInfo); +} + static uint32_t selectMemoryType(const VkPhysicalDeviceMemoryProperties& memoryProperties, uint32_t memoryTypeBits, VkMemoryPropertyFlags flags) { for (uint32_t i = 0; i < memoryProperties.memoryTypeCount; ++i) diff --git a/src/resources.h b/src/resources.h index 04bf7b6..09cbc46 100644 --- a/src/resources.h +++ b/src/resources.h @@ -20,6 +20,9 @@ VkBufferMemoryBarrier2 bufferBarrier(VkBuffer buffer, VkPipelineStageFlags2 srcS void pipelineBarrier(VkCommandBuffer commandBuffer, VkDependencyFlags dependencyFlags, size_t bufferBarrierCount, const VkBufferMemoryBarrier2* bufferBarriers, size_t imageBarrierCount, const VkImageMemoryBarrier2* imageBarriers); +void stageBarrier(VkCommandBuffer commandBuffer, VkPipelineStageFlags2 stageMask); +void stageBarrier(VkCommandBuffer commandBuffer, VkPipelineStageFlags2 srcStageMask, VkPipelineStageFlags2 dstStageMask); + void createBuffer(Buffer& result, VkDevice device, const VkPhysicalDeviceMemoryProperties& memoryProperties, size_t size, VkBufferUsageFlags usage, VkMemoryPropertyFlags memoryFlags); void uploadBuffer(VkDevice device, VkCommandPool commandPool, VkCommandBuffer commandBuffer, VkQueue queue, const Buffer& buffer, const Buffer& scratch, const void* data, size_t size); void destroyBuffer(const Buffer& buffer, VkDevice device);