# 
# The MIT License (MIT)
# 
# Copyright (c) 2024 Advanced Micro Devices, Inc.,
# Fatalist Development AB (Avalanche Studio Group),
# and Miguel Petersen.
# 
# All Rights Reserved.
# 
# Permission is hereby granted, free of charge, to any person obtaining a copy 
# of this software and associated documentation files (the "Software"), to deal 
# in the Software without restriction, including without limitation the rights 
# to use, copy, modify, merge, publish, distribute, sublicense, and/or sell copies 
# of the Software, and to permit persons to whom the Software is furnished to do so, 
# subject to the following conditions:
# 
# The above copyright notice and this permission notice shall be included in all 
# copies or substantial portions of the Software.
# 
# THE SOFTWARE IS PROVIDED "AS IS", WITHOUT WARRANTY OF ANY KIND, EXPRESS OR IMPLIED, 
# INCLUDING BUT NOT LIMITED TO THE WARRANTIES OF MERCHANTABILITY, FITNESS FOR A PARTICULAR 
# PURPOSE AND NONINFRINGEMENT. IN NO EVENT SHALL THE AUTHORS OR COPYRIGHT HOLDERS BE LIABLE 
# FOR ANY CLAIM, DAMAGES OR OTHER LIABILITY, WHETHER IN AN ACTION OF CONTRACT, TORT OR OTHERWISE, 
# ARISING FROM, OUT OF OR IN CONNECTION WITH THE SOFTWARE OR THE USE OR OTHER DEALINGS IN THE SOFTWARE.
# 


function(Project_AddHLSL OUT_GENERATED PROFILE ARGS HLSL HEADER VAR)
    # Compiler path
    if (WIN32)
        set(CompilerPath ${CMAKE_SOURCE_DIR}/ThirdParty/DXC/bin/Win64/dxc.exe)
    else()
        set(CompilerPath ${CMAKE_SOURCE_DIR}/ThirdParty/DXC/bin/Linux/dxc)
    endif()

    # Hlsl path
    if (NOT IS_ABSOLUTE ${HLSL})
        set(Hlsl ${CMAKE_CURRENT_SOURCE_DIR}/${HLSL})
    else()
        set(Hlsl ${HLSL})
    endif()

    # Parse args
    separate_arguments(Args WINDOWS_COMMAND ${ARGS})

    # Generated dependencies
    set(InternalGenerated "")

    # Default arguments
    set(SupressArguments -Wno-unknown-attributes -Wno-ignored-attributes)

    # Backend specific arguments
    cmake_parse_arguments(
        ARGS
        "" # Options
        "D3D12;VULKAN" # One Value
        "" # Multi Value
        ${ARGN}
    )

    # D3D12 defaults
    if ("${ARGS_D3D12}" STREQUAL "")
        set(ARGS_D3D12 "-Zi -Qembed_debug")
    endif ()

    # Vulkan defaults
    if ("${ARGS_VULKAN}" STREQUAL "")
        set(ARGS_VULKAN "-Zi -Qembed_debug -fspv-target-env=vulkan1.1")
    endif ()

    # Parse additional args
    separate_arguments(ArgsD3D12 WINDOWS_COMMAND ${ARGS_D3D12})
    separate_arguments(ArgsVulkan WINDOWS_COMMAND ${ARGS_VULKAN})

    # Generate vulkan
    if (${ENABLE_BACKEND_VULKAN})
        add_custom_command(
            OUTPUT ${HEADER}Vulkan.h
            DEPENDS
                ${CompilerPath}
                ${Hlsl}
            COMMAND ${CompilerPath}
                -spirv
                -Zi
                -Qembed_debug
                -T${PROFILE}
                ${Hlsl}
                -Fh ${HEADER}Vulkan.h
                -Vn ${VAR}Vulkan
                ${SupressArguments}
                ${Args} ${ArgsVulkan}
        )

        # Add output
        list(APPEND InternalGenerated ${HEADER}Vulkan.h)
    endif()

    # Generate d3d12
    if (${ENABLE_BACKEND_DX12})
        add_custom_command(
            OUTPUT ${HEADER}D3D12.h
            DEPENDS
                ${CompilerPath}
                ${Hlsl}
            COMMAND ${CompilerPath}
                -T${PROFILE}
                ${Hlsl}
                -Fh ${HEADER}D3D12.h
                -Vn ${VAR}D3D12
                ${SupressArguments}
                ${Args} ${ArgsD3D12}
        )

        # Add output
        list(APPEND InternalGenerated ${HEADER}D3D12.h)

        if (${ENABLE_DXIL_DUMP})
            add_custom_command(
                OUTPUT ${HEADER}D3D12.dxil.txt
                DEPENDS
                    ${CompilerPath}
                    ${Hlsl}
                COMMAND ${CompilerPath}
                    -T${PROFILE}
                    ${Hlsl}
                    -Fc ${HEADER}D3D12.dxil.txt
                    ${SupressArguments}
                    ${Args} ${ArgsD3D12}
            )

            # Add output
            list(APPEND InternalGenerated ${HEADER}D3D12.dxil.txt)
        endif()
    endif()

    # Set output
    set(${OUT_GENERATED} "${${OUT_GENERATED}};${InternalGenerated}" PARENT_SCOPE)
endfunction()

# Get all includes
file(GLOB Sources DXC/include/*)

# Configure all includes
foreach (File ${Sources})
    get_filename_component(BaseName "${File}" NAME)
    configure_file(${File} ${CMAKE_BINARY_DIR}/External/include/DXC/${BaseName})
endforeach ()

# Copy binaries
if (WIN32)
    ConfigureOutput(DXC/bin/win64/dxil.dll Dependencies/DXC/GRS.dxil.dll)
    ConfigureOutput(DXC/bin/win64/dxilconv.dll Dependencies/DXC/GRS.dxilconv.dll)
    ConfigureOutput(DXC/bin/win64/dxcompiler.dll Dependencies/DXC/GRS.dxcompiler.dll)
endif()