device.c

#include <vulkan/vulkan.h>
#include <stdint.h>
#include <stdio.h>
#include <string.h>
#include <stdlib.h>
#include "device.h"
#include "compute.h"
#include "instance.h"
#include "pipeline.h"
#include "memory.h"

uint32_t ComputeQueueFamilyIndex;

VkDevice LogicalDevice = VK_NULL_HANDLE;
VkQueue ComputingQueue = VK_NULL_HANDLE;
VkCommandPool ComputeCmdPool = VK_NULL_HANDLE;
VkDescriptorPool DescriptorPool = VK_NULL_HANDLE;

void CreateDeviceAndComuteQueue(void)
{
    uint32_t count = 0;
    vkGetPhysicalDeviceQueueFamilyProperties(PhysicalDevice, &count, NULL);
    VkQueueFamilyProperties families[count];
    vkGetPhysicalDeviceQueueFamilyProperties(PhysicalDevice, &count, families);

    printf("Found queue families count: %d\n", count);

    ComputeQueueFamilyIndex = 0;
    while ((ComputeQueueFamilyIndex < count) && ((families[ComputeQueueFamilyIndex].queueFlags & VK_QUEUE_COMPUTE_BIT) == 0))
    {
        ComputeQueueFamilyIndex++;
    }
    free(families);
    
    if(ComputeQueueFamilyIndex == count)
    {
        printf("Compute family not found\n");
        return;
    }

    float prio = 1.0f;

    VkDeviceCreateInfo deviceCreateInfo = {
        .sType = VK_STRUCTURE_TYPE_DEVICE_CREATE_INFO,
        .pQueueCreateInfos = &(VkDeviceQueueCreateInfo) {
            .sType = VK_STRUCTURE_TYPE_DEVICE_QUEUE_CREATE_INFO,
            .queueFamilyIndex = ComputeQueueFamilyIndex,
            .queueCount = 1,
            .pQueuePriorities = &prio,
        },
        .queueCreateInfoCount = 1,

    };

    if (vkCreateDevice(PhysicalDevice, &deviceCreateInfo, NULL, &LogicalDevice) != VK_SUCCESS)
    {
        printf("Could not create logical device\n");
        return;
    }

    vkGetDeviceQueue(LogicalDevice, ComputeQueueFamilyIndex, 0, &ComputingQueue);
}

void CreateCommandPool(void)
{
    VkCommandPoolCreateInfo poolCreateInfo = {
        .sType = VK_STRUCTURE_TYPE_COMMAND_POOL_CREATE_INFO,
        .queueFamilyIndex = ComputeQueueFamilyIndex,
    };

    if (vkCreateCommandPool(LogicalDevice, &poolCreateInfo, NULL, &ComputeCmdPool) != VK_SUCCESS)
    {
        printf("Failed to create command pool\n");
    }
}

void CreateDescriptorPool(void)
{
    VkDescriptorPoolCreateInfo descriptorPoolCreateInfo = {
        .sType = VK_STRUCTURE_TYPE_DESCRIPTOR_POOL_CREATE_INFO,
        .maxSets = 1,
        .pPoolSizes = &(VkDescriptorPoolSize){
            .type = VK_DESCRIPTOR_TYPE_STORAGE_BUFFER,
            .descriptorCount = 2,
        },
        .poolSizeCount = 1,
    };

    if (vkCreateDescriptorPool(LogicalDevice, &descriptorPoolCreateInfo, NULL, &DescriptorPool) != VK_SUCCESS)
    {
        printf("Failed to create descriptor pool\n");
    }
}

void DestroyCommandPoolAndLogicalDevice(void)
{
    if (ComputeCmdPool != VK_NULL_HANDLE)
    {
        vkDestroyCommandPool(LogicalDevice, ComputeCmdPool, NULL);
    }
    if (DescriptorPool != VK_NULL_HANDLE)
    {
        vkDestroyDescriptorPool(LogicalDevice, DescriptorPool, NULL);
    }

    DestroyBuffers();

    if(LogicalDevice != VK_NULL_HANDLE)
    {
        vkDestroyDevice(LogicalDevice,NULL);
    }
}