// Copyright (C) 2020 Jérôme Leclercq
// This file is part of the "Nazara Engine - Vulkan Renderer"
// For conditions of distribution and use, see copyright notice in Config.hpp

#include <Nazara/VulkanRenderer/Wrapper/Device.hpp>
#include <Nazara/Core/Error.hpp>
#include <Nazara/VulkanRenderer/Utils.hpp>
#include <Nazara/VulkanRenderer/Wrapper/Instance.hpp>
#include <Nazara/VulkanRenderer/Debug.hpp>

namespace Nz
{
	namespace Vk
	{
		inline const std::vector<Device::QueueFamilyInfo>& Device::GetEnabledQueues() const
		{
			return m_enabledQueuesInfos;
		}

		inline const Device::QueueList& Device::GetEnabledQueues(UInt32 familyQueue) const
		{
			NazaraAssert(familyQueue < m_enabledQueuesInfos.size(), "Invalid family queue");

			return *m_queuesByFamily[familyQueue];
		}

		inline Instance& Device::GetInstance()
		{
			return m_instance;
		}

		inline const Instance& Device::GetInstance() const
		{
			return m_instance;
		}

		inline VkResult Device::GetLastErrorCode() const
		{
			return m_lastErrorCode;
		}

		inline VmaAllocator Device::GetMemoryAllocator() const
		{
			return m_memAllocator;
		}

		inline VkPhysicalDevice Device::GetPhysicalDevice() const
		{
			return m_physicalDevice->physDevice;
		}

		inline const Vk::PhysicalDevice& Device::GetPhysicalDeviceInfo() const
		{
			return *m_physicalDevice;
		}

		inline UInt32 Device::GetDefaultFamilyIndex(QueueType queueType) const
		{
			return m_defaultQueues[UnderlyingCast(queueType)];
		}

		inline bool Device::IsExtensionLoaded(const std::string& extensionName)
		{
			return m_loadedExtensions.count(extensionName) > 0;
		}

		inline bool Device::IsLayerLoaded(const std::string& layerName)
		{
			return m_loadedLayers.count(layerName) > 0;
		}

		inline bool Device::WaitForIdle()
		{
			m_lastErrorCode = vkDeviceWaitIdle(m_device);
			if (m_lastErrorCode != VkResult::VK_SUCCESS)
			{
				NazaraError("Failed to wait for device idle: " + TranslateVulkanError(m_lastErrorCode));
				return false;
			}

			return true;
		}

		inline Device::operator VkDevice()
		{
			return m_device;
		}

		inline PFN_vkVoidFunction Device::GetProcAddr(const char* name)
		{
			PFN_vkVoidFunction func = m_instance.GetDeviceProcAddr(m_device, name);
			if (!func)
				NazaraError("Failed to get " + std::string(name) + " address");
			
			return func;
		}
	}
}

#include <Nazara/VulkanRenderer/DebugOff.hpp>
