shader.c

#include <kiba/core/memory.h>
#include <kiba/gpu/shader.h>
#include <kiba/gpu/vulkan/allocator.h>
#include <kiba/gpu/vulkan/device.h>
#include <kiba/gpu/vulkan/util.h>
 
b8 gpu_backend_shader_module_create(struct gpu_backend_shader_module   *shader,
                                    struct gpu_backend_device          *device,
                                    struct gpu_shader_module_descriptor desc) {
    usize size_in_bytes = 0;
    if (desc.code == KB_NULL || (size_in_bytes = array_size(desc.code)) == 0) {
        KB_ERROR("provided invalid shader code");
        return false;
    }
    u32 code_size         = (size_in_bytes + 3u) & ~3u;  // ensure code size is multiple of 4 with enough space for code
    array_of(u32) vk_code = array_create(u32, code_size / sizeof(u32), &vk_alloc.kiba_alloc);
    if (vk_code == KB_NULL) {
        KB_ERROR("cannot reserve memory to transfer shader code");
        return false;
    }
    memory_copy(vk_code, desc.code, size_in_bytes);
 
    VkShaderModuleCreateInfo shader_info = {
        .sType    = VK_STRUCTURE_TYPE_SHADER_MODULE_CREATE_INFO,
        .codeSize = code_size,
        .pCode    = vk_code,
    };
    // TODO: this will leak in case its not successfull, could add cleanup option to macro
    VK_CALL_B8(vkCreateShaderModule(device->logical, &shader_info, &vk_alloc.vulkan_callbacks, &shader->raw));
    array_destroy(&vk_code);
    vk_device_set_object_name(device->logical, VK_OBJECT_TYPE_SHADER_MODULE, shader->raw, desc.label);
    return true;
}
 
void gpu_backend_shader_module_destroy(struct gpu_backend_shader_module *shader, struct gpu_backend_device *device) {
    if (shader->raw != VK_NULL_HANDLE) {
        vkDestroyShaderModule(device->logical, shader->raw, &vk_alloc.vulkan_callbacks);
        shader->raw = VK_NULL_HANDLE;
    }
}