133 lines
3.1 KiB
C++
133 lines
3.1 KiB
C++
/*
|
|
* Created on: Oct 3, 2019
|
|
|
|
Copyright 2019 flyinghead
|
|
|
|
This file is part of Flycast.
|
|
|
|
Flycast is free software: you can redistribute it and/or modify
|
|
it under the terms of the GNU General Public License as published by
|
|
the Free Software Foundation, either version 2 of the License, or
|
|
(at your option) any later version.
|
|
|
|
Flycast is distributed in the hope that it will be useful,
|
|
but WITHOUT ANY WARRANTY; without even the implied warranty of
|
|
MERCHANTABILITY or FITNESS FOR A PARTICULAR PURPOSE. See the
|
|
GNU General Public License for more details.
|
|
|
|
You should have received a copy of the GNU General Public License
|
|
along with Flycast. If not, see <https://www.gnu.org/licenses/>.
|
|
*/
|
|
#pragma once
|
|
#include "vulkan.h"
|
|
#include "vmallocator.h"
|
|
#include "utils.h"
|
|
|
|
struct BufferData
|
|
{
|
|
BufferData(vk::DeviceSize size, vk::BufferUsageFlags usage,
|
|
vk::MemoryPropertyFlags propertyFlags =
|
|
vk::MemoryPropertyFlagBits::eHostVisible | vk::MemoryPropertyFlagBits::eHostCoherent);
|
|
~BufferData()
|
|
{
|
|
buffer.reset();
|
|
}
|
|
|
|
void upload(u32 size, const void *data, u32 bufOffset = 0) const
|
|
{
|
|
verify(bufOffset + size <= bufferSize);
|
|
|
|
void* dataPtr = (u8 *)allocation.MapMemory() + bufOffset;
|
|
memcpy(dataPtr, data, size);
|
|
allocation.UnmapMemory();
|
|
}
|
|
|
|
void upload(size_t count, const u32 *sizes, const void * const *data, u32 bufOffset = 0) const
|
|
{
|
|
u32 totalSize = 0;
|
|
for (size_t i = 0; i < count; i++)
|
|
totalSize += sizes[i];
|
|
verify(bufOffset + totalSize <= bufferSize);
|
|
void* dataPtr = (u8 *)allocation.MapMemory() + bufOffset;
|
|
for (size_t i = 0; i < count; i++)
|
|
{
|
|
if (data[i] != nullptr)
|
|
memcpy(dataPtr, data[i], sizes[i]);
|
|
dataPtr = (u8 *)dataPtr + sizes[i];
|
|
}
|
|
allocation.UnmapMemory();
|
|
}
|
|
|
|
void download(u32 size, void *data, u32 bufOffset = 0) const
|
|
{
|
|
verify(bufOffset + size <= bufferSize);
|
|
|
|
void* dataPtr = (u8 *)allocation.MapMemory() + bufOffset;
|
|
memcpy(data, dataPtr, size);
|
|
allocation.UnmapMemory();
|
|
}
|
|
|
|
void *MapMemory()
|
|
{
|
|
return allocation.MapMemory();
|
|
}
|
|
void UnmapMemory()
|
|
{
|
|
allocation.UnmapMemory();
|
|
}
|
|
|
|
vk::UniqueBuffer buffer;
|
|
vk::DeviceSize bufferSize;
|
|
Allocation allocation;
|
|
|
|
private:
|
|
vk::BufferUsageFlags m_usage;
|
|
};
|
|
|
|
class BufferPacker
|
|
{
|
|
public:
|
|
BufferPacker();
|
|
|
|
vk::DeviceSize addUniform(const void *p, size_t size) {
|
|
return add(p, size, uniformAlignment);
|
|
}
|
|
|
|
vk::DeviceSize addStorage(const void *p, size_t size) {
|
|
return add(p, size, storageAlignment);
|
|
}
|
|
|
|
vk::DeviceSize add(const void *p, size_t size, u32 alignment = 4)
|
|
{
|
|
u32 padding = align(offset, std::max(4u, alignment));
|
|
if (padding != 0)
|
|
{
|
|
chunks.push_back(nullptr);
|
|
chunkSizes.push_back(padding);
|
|
offset += padding;
|
|
}
|
|
vk::DeviceSize start = offset;
|
|
chunks.push_back(p);
|
|
chunkSizes.push_back(size);
|
|
offset += size;
|
|
|
|
return start;
|
|
}
|
|
|
|
void upload(BufferData& bufferData, u32 bufOffset = 0)
|
|
{
|
|
bufferData.upload(chunks.size(), &chunkSizes[0], &chunks[0], bufOffset);
|
|
}
|
|
|
|
vk::DeviceSize size() const {
|
|
return offset;
|
|
}
|
|
|
|
private:
|
|
std::vector<const void *> chunks;
|
|
std::vector<u32> chunkSizes;
|
|
vk::DeviceSize offset = 0;
|
|
vk::DeviceSize uniformAlignment;
|
|
vk::DeviceSize storageAlignment;
|
|
};
|