mirror of
https://github.com/Kbz-8/42_vox.git
synced 2026-01-11 22:53:35 +00:00
95 lines
3.1 KiB
C++
95 lines
3.1 KiB
C++
#include <Renderer/Memory/Chunk.h>
|
|
#include <Renderer/RenderCore.h>
|
|
#include <Core/Logs.h>
|
|
|
|
#include <algorithm>
|
|
|
|
namespace Scop
|
|
{
|
|
MemoryChunk::MemoryChunk(VkDevice device, VkPhysicalDevice physical, VkDeviceSize size, std::int32_t memory_type_index, bool is_dedicated)
|
|
: m_device(device), m_physical(physical), m_size(size), m_memory_type_index(memory_type_index), m_is_dedicated(is_dedicated)
|
|
{
|
|
Verify(device != VK_NULL_HANDLE, "Memory Chunk : invalid device");
|
|
VkMemoryAllocateInfo alloc_info{};
|
|
alloc_info.sType = VK_STRUCTURE_TYPE_MEMORY_ALLOCATE_INFO;
|
|
alloc_info.allocationSize = size;
|
|
alloc_info.memoryTypeIndex = m_memory_type_index;
|
|
if(RenderCore::Get().vkAllocateMemory(m_device, &alloc_info, nullptr, &m_memory) != VK_SUCCESS)
|
|
FatalError("Vulkan: failed to allocate memory for a chunk");
|
|
|
|
VkPhysicalDeviceMemoryProperties properties;
|
|
RenderCore::Get().vkGetPhysicalDeviceMemoryProperties(m_physical, &properties);
|
|
if((properties.memoryTypes[m_memory_type_index].propertyFlags & VK_MEMORY_PROPERTY_HOST_VISIBLE_BIT) == VK_MEMORY_PROPERTY_HOST_VISIBLE_BIT)
|
|
{
|
|
if(RenderCore::Get().vkMapMemory(m_device, m_memory, 0, VK_WHOLE_SIZE, 0, &p_map) != VK_SUCCESS)
|
|
FatalError("Vulkan: failed to map a host visible chunk");
|
|
}
|
|
MemoryBlock& block = m_blocks.emplace_back();
|
|
block.memory = m_memory;
|
|
block.offset = 0;
|
|
block.size = size;
|
|
block.free = true;
|
|
}
|
|
|
|
[[nodiscard]] std::optional<MemoryBlock> MemoryChunk::Allocate(VkDeviceSize size, VkDeviceSize alignment)
|
|
{
|
|
for(std::size_t i = 0; i < m_blocks.size(); i++)
|
|
{
|
|
if(!m_blocks[i].free || m_blocks[i].size < size)
|
|
continue;
|
|
VkDeviceSize offset_displacement = (m_blocks[i].offset % alignment > 0) ? alignment - m_blocks[i].offset % alignment : 0;
|
|
VkDeviceSize old_size_available = m_blocks[i].size - offset_displacement;
|
|
|
|
if(size + offset_displacement <= m_blocks[i].size)
|
|
{
|
|
m_blocks[i].offset += offset_displacement;
|
|
m_blocks[i].size = size;
|
|
m_blocks[i].free = false;
|
|
|
|
if(p_map != nullptr)
|
|
m_blocks[i].map = reinterpret_cast<void*>(reinterpret_cast<std::uintptr_t>(p_map) + m_blocks[i].offset);
|
|
|
|
MemoryBlock new_block;
|
|
new_block.memory = m_memory;
|
|
new_block.offset = m_blocks[i].offset + size;
|
|
new_block.size = old_size_available - size;
|
|
new_block.free = true;
|
|
|
|
if(new_block.size > 0)
|
|
m_blocks.emplace(m_blocks.begin() + i + 1, new_block);
|
|
return m_blocks[i];
|
|
}
|
|
}
|
|
return std::nullopt;
|
|
}
|
|
|
|
void MemoryChunk::Deallocate(const MemoryBlock& block)
|
|
{
|
|
auto it = std::find(m_blocks.begin(), m_blocks.end(), block);
|
|
if(it == m_blocks.end())
|
|
FatalError("Memory Chunk : cannot deallocate a block that is owned by another chunk");
|
|
it->free = true;
|
|
|
|
bool end = false;
|
|
while(!end)
|
|
{
|
|
end = true;
|
|
for(auto it = m_blocks.begin(); it != m_blocks.end(); ++it)
|
|
{
|
|
if(it->free && (it + 1)->free)
|
|
{
|
|
it->size = (it + 1)->offset + (it + 1)->size - it->offset;
|
|
m_blocks.erase(it + 1);
|
|
end = false;
|
|
break;
|
|
}
|
|
}
|
|
}
|
|
}
|
|
|
|
MemoryChunk::~MemoryChunk()
|
|
{
|
|
RenderCore::Get().vkFreeMemory(m_device, m_memory, nullptr);
|
|
}
|
|
}
|