# Modifications Copyright (C) 2022-2023 Advanced Micro Devices, Inc. All rights reserved. #
# Notified per clause 4(b) of the license

# ~~~
# Copyright (c) 2018 Valve Corporation
# Copyright (c) 2018 LunarG, Inc.
#
# Licensed under the Apache License, Version 2.0 (the "License");
# you may not use this file except in compliance with the License.
# You may obtain a copy of the License at
#
#     http://www.apache.org/licenses/LICENSE-2.0
#
# Unless required by applicable law or agreed to in writing, software
# distributed under the License is distributed on an "AS IS" BASIS,
# WITHOUT WARRANTIES OR CONDITIONS OF ANY KIND, either express or implied.
# See the License for the specific language governing permissions and
# limitations under the License.
# ~~~

cmake_minimum_required(VERSION 3.14)

project(Vulkan-Tools)

add_compile_options(-Wno-unused-result)
find_package(PythonInterp 3 QUIET)

set(CMAKE_MODULE_PATH ${CMAKE_MODULE_PATH} "${CMAKE_CURRENT_SOURCE_DIR}/cmake")
set(SCRIPTS_DIR "${CMAKE_CURRENT_SOURCE_DIR}/scripts")

find_package(Vulkan REQUIRED)
find_package(VulkanHeaders REQUIRED)

set(CUBE_INCLUDE_DIRS ${CMAKE_CURRENT_BINARY_DIR})

if (NOT GLSLANG_INSTALL_DIR AND DEFINED ENV{GLSLANG_INSTALL_DIR})
    set(GLSLANG_INSTALL_DIR $ENV{GLSLANG_INSTALL_DIR})
endif()

if(GLSLANG_INSTALL_DIR)
    message(STATUS "Using GLSLANG_INSTALL_DIR to look for glslangValidator")
    find_program(GLSLANG_VALIDATOR names glslangValidator HINTS "${GLSLANG_INSTALL_DIR}/bin")
else()
    set(GLSLANG_VALIDATOR_NAME "glslangValidator")
    message(STATUS "Using cmake find_program to look for glslangValidator")
    execute_process(COMMAND ${PYTHON_EXECUTABLE} ${SCRIPTS_DIR}/fetch_glslangvalidator.py glslang-master-linux-Release.zip)
    set(PLATFORM_DIR "${PROJECT_SOURCE_DIR}/glslang/linux/bin")
    find_program(GLSLANG_VALIDATOR NAMES ${GLSLANG_VALIDATOR_NAME} HINTS ${PLATFORM_DIR})
endif()

include(FindPkgConfig)

option(VK_RENDER_1080P "Build vkcubepp 1080p support" OFF)

if(VK_RENDER_1080P)
    add_definitions(-DVK_RENDER_1080P)
endif()

pkg_check_modules(XCODER REQUIRED xcoder)
include_directories(${XCODER_INCLUDE_DIRS})
link_libraries(${XCODER_LINK_LIBRARIES})
link_directories(${XCODER_LIBRARY_DIRS})

link_libraries(${API_LOWERCASE} m)

set(CMAKE_C_FLAGS "${CMAKE_C_FLAGS}")
set(CMAKE_CXX_FLAGS "${CMAKE_CXX_FLAGS}")

add_custom_command(COMMENT "Compiling cube vertex shader"
                   OUTPUT cube.vert.inc
                   COMMAND ${GLSLANG_VALIDATOR} -V -x -o ${CMAKE_CURRENT_BINARY_DIR}/cube.vert.inc
                            ${CMAKE_CURRENT_SOURCE_DIR}/shaders/cube.vert
                    MAIN_DEPENDENCY ${CMAKE_CURRENT_SOURCE_DIR}/shaders/cube.vert
                    DEPENDS ${CMAKE_CURRENT_SOURCE_DIR}/shaders/cube.vert ${GLSLANG_VALIDATOR})
add_custom_command(COMMENT "Compiling cube fragment shader"
                   OUTPUT cube.frag.inc
                   COMMAND ${GLSLANG_VALIDATOR} -V -x -o ${CMAKE_CURRENT_BINARY_DIR}/cube.frag.inc
                            ${CMAKE_CURRENT_SOURCE_DIR}/shaders/cube.frag
                    MAIN_DEPENDENCY ${CMAKE_CURRENT_SOURCE_DIR}/shaders/cube.frag
                    DEPENDS ${CMAKE_CURRENT_SOURCE_DIR}/shaders/cube.frag ${GLSLANG_VALIDATOR})

add_custom_command(COMMENT "Compiling show texture vertex shader"
                   OUTPUT show_texture.vert.inc
                   COMMAND ${GLSLANG_VALIDATOR} -V -x -o ${CMAKE_CURRENT_BINARY_DIR}/show_texture.vert.inc
                            ${CMAKE_CURRENT_SOURCE_DIR}/shaders/show_texture.vert
                    MAIN_DEPENDENCY ${CMAKE_CURRENT_SOURCE_DIR}/shaders/show_texture.vert
                    DEPENDS ${CMAKE_CURRENT_SOURCE_DIR}/shaders/show_texture.vert ${GLSLANG_VALIDATOR})
add_custom_command(COMMENT "Compiling show texture fragment shader"
                   OUTPUT show_texture.frag.inc
                   COMMAND ${GLSLANG_VALIDATOR} -V -x -o ${CMAKE_CURRENT_BINARY_DIR}/show_texture.frag.inc
                            ${CMAKE_CURRENT_SOURCE_DIR}/shaders/show_texture.frag
                    MAIN_DEPENDENCY ${CMAKE_CURRENT_SOURCE_DIR}/shaders/show_texture.frag
                    DEPENDS ${CMAKE_CURRENT_SOURCE_DIR}/shaders/show_texture.frag ${GLSLANG_VALIDATOR})

add_custom_command(COMMENT "Compiling rbg2nv12 color conversion compute shader"
                   OUTPUT rgb2nv12.comp.inc
                   COMMAND ${GLSLANG_VALIDATOR} -V -x -o ${CMAKE_CURRENT_BINARY_DIR}/rgb2nv12.comp.inc
                            ${CMAKE_CURRENT_SOURCE_DIR}/shaders/rgb2nv12.comp
                    MAIN_DEPENDENCY ${CMAKE_CURRENT_SOURCE_DIR}/shaders/rgb2nv12.comp
                    DEPENDS ${CMAKE_CURRENT_SOURCE_DIR}/shaders/rgb2nv12.comp ${GLSLANG_VALIDATOR})

add_custom_command(COMMENT "Compiling rgb2yuv420p color conversion compute shader"
                   OUTPUT rgb2yuv420p_netint.comp.inc
                   COMMAND ${GLSLANG_VALIDATOR} -V -x -o ${CMAKE_CURRENT_BINARY_DIR}/rgb2yuv420p_netint.comp.inc
                            ${CMAKE_CURRENT_SOURCE_DIR}/shaders/rgb2yuv420p_netint.comp
                    MAIN_DEPENDENCY ${CMAKE_CURRENT_SOURCE_DIR}/shaders/rgb2yuv420p_netint.comp
                    DEPENDS ${CMAKE_CURRENT_SOURCE_DIR}/shaders/rgb2yuv420p_netint.comp ${GLSLANG_VALIDATOR})

add_custom_command(COMMENT "Compiling rgb2yuv420p 1080p color conversion compute shader"
                   OUTPUT rgb2yuv420p_netint_1080p.comp.inc
                   COMMAND ${GLSLANG_VALIDATOR} -V -x -o ${CMAKE_CURRENT_BINARY_DIR}/rgb2yuv420p_netint_1080p.comp.inc
                            ${CMAKE_CURRENT_SOURCE_DIR}/shaders/rgb2yuv420p_netint_1080p.comp
                    MAIN_DEPENDENCY ${CMAKE_CURRENT_SOURCE_DIR}/shaders/rgb2yuv420p_netint_1080p.comp
                    DEPENDS ${CMAKE_CURRENT_SOURCE_DIR}/shaders/rgb2yuv420p_netint_1080p.comp ${GLSLANG_VALIDATOR})


include_directories(${CUBE_INCLUDE_DIRS} ${CMAKE_CURRENT_SOURCE_DIR})

# ----------------------------------------------------------------------------
# vkcubepp

find_package(Threads REQUIRED)
set(CMAKE_THREAD_PREFER_PTHREAD TRUE)
set(THREADS_PREFER_PTHREAD_FLAG TRUE)
add_executable(vkcubepp
                main.cpp
                cube.cpp
                cube.h
                surface_render_context.h
                options.h
                options.cpp
                vk_common.h
                vk_common.cpp
                gettime.h
                timestamp_record.cpp
                timestamp_record.h
                encoding_device.h
                netint_encoding.h
                netint_encoding.cpp
                cube.vert.inc
                cube.frag.inc
                show_texture.vert.inc
                show_texture.frag.inc
                rgb2nv12.comp.inc
                rgb2yuv420p_netint.comp.inc
                rgb2yuv420p_netint_1080p.comp.inc)
target_link_libraries(vkcubepp Vulkan::Vulkan Threads::Threads)

install(TARGETS vkcubepp RUNTIME DESTINATION ${CMAKE_INSTALL_BINDIR})

install(SCRIPT ${CMAKE_SOURCE_DIR}/scripts/run_multiple_encoding.py ${CMAKE_INSTALL_BINDIR})
