1 #include <msp/core/algorithm.h>
2 #include <msp/core/maputils.h>
4 #include "transferqueue.h"
12 TransferQueue::TransferQueue(Device &d):
16 void TransferQueue::allocate_staging(PendingTransfer &transfer, size_t size)
18 auto i = find_if(buffers, [size](const StagingBuffer &b){ return b.used+size<=b.size; });
21 buffers.emplace_back(device, max(default_buffer_size, size));
22 i = prev(buffers.end());
25 transfer.buffer_index = distance(buffers.begin(), i);
26 transfer.offset = i->used;
28 transfer.staging_address = static_cast<char *>(i->mapped_address)+transfer.offset;
33 TransferQueue::PendingTransfer &TransferQueue::prepare_transfer(const void *object, bool ordered, size_t size)
35 PendingTransfer transfer;
36 transfer.object = object;
37 transfer.order = ordered;
41 allocate_staging(transfer, size);
42 auto i = lower_bound_member(async_transfers, transfer.staging_address, &PendingTransfer::staging_address);
43 i = async_transfers.emplace(i, move(transfer));
47 return insert_transfer(move(transfer));
50 void TransferQueue::finalize_transfer(void *staging)
52 auto i = lower_bound_member(async_transfers, staging, &PendingTransfer::staging_address);
53 if(i==async_transfers.end() || i->staging_address!=staging)
54 throw key_error(staging);
56 insert_transfer(move(*i));
57 async_transfers.erase(i);
60 TransferQueue::PendingTransfer &TransferQueue::insert_transfer(PendingTransfer &&pt)
62 bool ordered = pt.order;
64 unsigned &order = next_orders[pt.object];
66 order += (order&1)|ordered;
68 auto j = upper_bound_member(transfers, order, &PendingTransfer::order);
69 j = transfers.emplace(j, move(pt));
77 void TransferQueue::dispatch_transfers(VkCommandBuffer command_buffer)
82 for(auto i=transfers.begin(); i!=transfers.end(); )
85 for(; (j!=transfers.end() && j->order==i->order); ++j)
88 device.get_synchronizer().barrier(command_buffer);
92 VkBuffer buffer = (i->buffer_index>=0 ? buffers[i->buffer_index].buffer : 0);
93 i->transfer(command_buffer, buffer, i->offset);
102 TransferQueue::StagingBuffer::StagingBuffer(Device &d, size_t s):
106 const VulkanFunctions &vk = device.get_functions();
108 VkBufferCreateInfo buffer_info = { };
109 buffer_info.sType = VK_STRUCTURE_TYPE_BUFFER_CREATE_INFO;
110 buffer_info.size = size;
111 buffer_info.usage = VK_BUFFER_USAGE_TRANSFER_SRC_BIT;
112 buffer_info.sharingMode = VK_SHARING_MODE_EXCLUSIVE;
114 vk.CreateBuffer(buffer_info, buffer);
116 MemoryAllocator &allocator = device.get_allocator();
117 memory_id = allocator.allocate(buffer, STAGING_MEMORY);
118 mapped_address = allocator.map(memory_id, 0, size);
121 TransferQueue::StagingBuffer::StagingBuffer(StagingBuffer &&other):
122 device(other.device),
123 buffer(other.buffer),
124 memory_id(other.memory_id),
127 mapped_address(other.mapped_address)
131 other.mapped_address = 0;
134 TransferQueue::StagingBuffer::~StagingBuffer()
136 const VulkanFunctions &vk = device.get_functions();
137 MemoryAllocator &allocator = device.get_allocator();
141 allocator.unmap(mapped_address);
142 allocator.release(memory_id);
145 vk.DestroyBuffer(buffer);