// Copyright Epic Games, Inc. All Rights Reserved.

using UnrealBuildTool;
using System;
using System.IO;

public class Vulkan : ModuleRules
{
    public Vulkan(ReadOnlyTargetRules Target) : base(Target)
    {
        Type = ModuleType.External;

        string RootPath = Target.UEThirdPartySourceDirectory + "Vulkan";

        PublicSystemIncludePaths.Add(RootPath + "/Include");
        PublicSystemIncludePaths.Add(RootPath + "/Include/vulkan");

        if (Target.Platform == UnrealTargetPlatform.Win64)
        {
            // Let's always delay load the vulkan dll as not everyone has it installed
            PublicDelayLoadDLLs.Add("vulkan-1.dll");
        }

		// Always add the vulkan profile include folder from third party
		PublicSystemIncludePaths.Add(Target.UEThirdPartySourceDirectory + "/Vulkan/profiles/include");
	}

	internal static int GetVersionFromString(string Text)
    {
        string Token = "#define VK_HEADER_VERSION ";
        Int32 FoundIndex = Text.IndexOf(Token);
        if (FoundIndex > 0)
        {
            string Version = Text.Substring(FoundIndex + Token.Length, 5);
            int Index = 0;
            while (Version[Index] >= '0' && Version[Index] <= '9')
            {
                ++Index;
            }

            Version = Version.Substring(0, Index);

            int VersionNumber = Convert.ToInt32(Version);
            return VersionNumber;
        }

        return -1;
    }

    internal static int GetThirdPartyVersion()
    {
        try
        {
            // Extract current version on ThirdParty
            string Text = File.ReadAllText("ThirdParty/Vulkan/Include/vulkan/vulkan_core.h");
            return GetVersionFromString(Text);
        }
        catch (Exception)
        {
        }

        return -1;
    }

    internal static int GetSDKVersion(string VulkanSDKIncludePath)
    {
        try
        {
            // Extract current version on the SDK folder. Newer SDKs store the version in vulkan_core.h
            string Header = Path.Combine(VulkanSDKIncludePath, "vulkan/vulkan_core.h");
            if (!File.Exists(Header))
            {
                Header = Path.Combine(VulkanSDKIncludePath, "vulkan/vulkan.h");
            }
            string Text = File.ReadAllText(Header);
            return GetVersionFromString(Text);
        }
        catch (Exception)
        {
        }

        return -1;
    }

	internal static string GetSDKIncludePath(string VulkanSDKPath)
	{
		try
		{
			// For Linux, support both versions of include path with/without a capital I
			string IncludePath = Path.Combine(VulkanSDKPath, "Include");
			if (!Directory.Exists(IncludePath))
			{
				IncludePath = Path.Combine(VulkanSDKPath, "include");
			}
			return IncludePath;
		}
		catch (Exception)
		{
		}

		return "";
	}
}