buffer.c

#include <kiba/gpu/buffer.h>
 
#include <kiba/gpu/vulkan/allocator.h>
#include <kiba/gpu/vulkan/conv.h>
#include <kiba/gpu/vulkan/device.h>
#include <kiba/gpu/vulkan/util.h>
#include "kiba/gpu/enums.h"
 
b8 gpu_backend_buffer_create(struct gpu_backend_buffer   *buffer,
                             struct gpu_backend_device   *device,
                             struct gpu_buffer_descriptor desc) {
    VkBufferCreateInfo create_info = {
        .sType       = VK_STRUCTURE_TYPE_BUFFER_CREATE_INFO,
        .size        = desc.size,
        .usage       = vk_convert_buffer_usage(desc.usage),
        .sharingMode = VK_SHARING_MODE_EXCLUSIVE,
    };
    VK_CALL_B8(vkCreateBuffer(device->logical, &create_info, &vk_alloc.vulkan_callbacks, &buffer->raw));
 
    VkMemoryRequirements memory_requirements;
    vkGetBufferMemoryRequirements(device->logical, buffer->raw, &memory_requirements);
 
    VkMemoryPropertyFlagBits properties = 0;
    if (KB_FLAGS_ANY_SET(desc.usage, GPU_BUFFER_USAGE_MAP_READ | GPU_BUFFER_USAGE_MAP_WRITE)) {
        properties |= VK_MEMORY_PROPERTY_HOST_VISIBLE_BIT | VK_MEMORY_PROPERTY_HOST_COHERENT_BIT;
    } else {
        properties |= VK_MEMORY_PROPERTY_DEVICE_LOCAL_BIT;
    }
 
    if (!vk_device_allocate_memory(*device, &buffer->memory, memory_requirements, properties)) {
        KB_ERROR("could not allocate memory for buffer");
        return false;
    }
    VK_CALL_B8(vkBindBufferMemory(device->logical, buffer->raw, buffer->memory, 0));
    buffer->mapped = KB_NULL;
 
    vk_device_set_object_name(device->logical, VK_OBJECT_TYPE_BUFFER, buffer->raw, desc.label);
    return true;
}
 
void *gpu_backend_buffer_mapped_memory(struct gpu_backend_buffer *buffer, struct gpu_backend_device *device) {
    if (!buffer->mapped) {
        VkResult res = vkMapMemory(device->logical, buffer->memory, 0, VK_WHOLE_SIZE, 0, &buffer->mapped);
        if (res != VK_SUCCESS) {
            return KB_NULL;
        }
    }
    return buffer->mapped;
}
 
void gpu_backend_buffer_destroy(struct gpu_backend_buffer *buffer, struct gpu_backend_device *device) {
    if (buffer->mapped) {
        vkUnmapMemory(device->logical, buffer->memory);
    }
    vkDestroyBuffer(device->logical, buffer->raw, &vk_alloc.vulkan_callbacks);
    vkFreeMemory(device->logical, buffer->memory, &vk_alloc.vulkan_callbacks);
}