# ~~~
# Copyright (c) 2026-2026 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.
# ~~~

find_package(Python3 REQUIRED COMPONENTS Interpreter)

function(add_vulkan_python_test TARGET_NAME SCRIPT_FILENAME)
    set(REGISTRY_PATH "${VULKAN_HEADERS_INSTALL_DIR}/${CMAKE_INSTALL_DATADIR}/vulkan/registry")
    set(FULL_SCRIPT_PATH "${CMAKE_CURRENT_SOURCE_DIR}/${SCRIPT_FILENAME}")

    # Register the script as a CTest test case
    add_test(NAME ${TARGET_NAME}
        COMMAND ${VENV_PYTHON_EXECUTABLE} "${FULL_SCRIPT_PATH}" --registry "${REGISTRY_PATH}/vk.xml"
        WORKING_DIRECTORY "${CMAKE_CURRENT_SOURCE_DIR}"
    )

    # Cleanly inject environment variables into the test environment
    set_tests_properties(${TARGET_NAME} PROPERTIES
        ENVIRONMENT "PYTHONPATH=${REGISTRY_PATH}"
    )
endfunction()

if(NOT APPLE) # macOS doesn't handle dynamically loading Python jsonschema'
    add_vulkan_python_test(VkProfiles_TestVulkanProfilesUtil           test_util.py)
    add_vulkan_python_test(VkProfiles_TestVulkanObjectEmpty            test_init.py)
    add_vulkan_python_test(VkProfiles_TestVulkanVersion                test_vulkan_version.py)
    add_vulkan_python_test(VkProfiles_TestVulkanObjectUtils            test_vulkan_object_utils.py)
    add_vulkan_python_test(VkProfiles_TestVulkanObjectDepends          test_vulkan_object_expression_parsing.py)
    add_vulkan_python_test(VkProfiles_TestVulkanProfilesJson           test_profiles_json_utils.py)
    add_vulkan_python_test(VkProfiles_TestFormatFlagsConverter         test_format_flags.py)
    add_vulkan_python_test(VkProfiles_TestCombineDifference            test_combine_difference.py)
    add_vulkan_python_test(VkProfiles_TestCombineIntersection          test_combine_intersection.py)
    add_vulkan_python_test(VkProfiles_TestCombineUnion                 test_combine_union.py)
    add_vulkan_python_test(VkProfiles_TestCombineLimittype             test_combine_limittype.py)
    add_vulkan_python_test(VkProfiles_TestTransformPullExts            test_transform_pull_caps_exts.py)
    add_vulkan_python_test(VkProfiles_TestTransformPullCapsCore10      test_transform_pull_caps_core10.py)
    add_vulkan_python_test(VkProfiles_TestTransformPullCapsCore11      test_transform_pull_caps_core11.py)
    add_vulkan_python_test(VkProfiles_TestTransformPullCapsCore12      test_transform_pull_caps_core12.py)
    add_vulkan_python_test(VkProfiles_TestTransformPullCapsCore13      test_transform_pull_caps_core13.py)
    add_vulkan_python_test(VkProfiles_TestTransformPullCapsCore14      test_transform_pull_caps_core14.py)
    add_vulkan_python_test(VkProfiles_TestTransformPullCapsInhe        test_transform_pull_caps_inheritance.py)
    add_vulkan_python_test(VkProfiles_TestTransformPullCapsProf        test_transform_pull_caps_prof.py)
    add_vulkan_python_test(VkProfiles_TestTransformPullCapsOr          test_transform_pull_caps_or.py)
    add_vulkan_python_test(VkProfiles_TestTransformPullCapsDepends     test_transform_pull_caps_depends.py)
    add_vulkan_python_test(VkProfiles_TestTransformPullPromotedExts    test_transform_pull_promoted_exts.py)
    add_vulkan_python_test(VkProfiles_TestTransformPullAliases         test_transform_pull_aliases.py)
    add_vulkan_python_test(VkProfiles_TestTransformConsolidate         test_transform_consolidate.py)
    add_vulkan_python_test(VkProfiles_TestTransformStripHelper         test_transform_strip_helper.py)
    add_vulkan_python_test(VkProfiles_TestTransformStripDups           test_transform_strip_duplication.py)
    add_vulkan_python_test(VkProfiles_TestTransformStripPromotedExts   test_transform_strip_promoted_exts.py)
    add_vulkan_python_test(VkProfiles_TestTransformSort                test_transform_sort.py)
    add_vulkan_python_test(VkProfiles_TestExtract                      test_extract.py)
    add_vulkan_python_test(VkProfiles_TestMinApiVersion                test_min_api_version.py)
    add_vulkan_python_test(VkProfiles_TestValidate                     test_validate.py)
endif()

# Convert Android Profiles custom target
set(REGISTRY_PATH "${VULKAN_HEADERS_INSTALL_DIR}/${CMAKE_INSTALL_DATADIR}/vulkan/registry")
set(PROFILES_INPUT_DIR "${CMAKE_CURRENT_SOURCE_DIR}/../../profiles/Android")
set(PROFILES_OUTPUT_DIR "${CMAKE_CURRENT_BINARY_DIR}/profiles")

add_custom_target(transform_android_profiles ALL
    COMMAND ${CMAKE_COMMAND} -E make_directory "${PROFILES_OUTPUT_DIR}"
    COMMAND ${CMAKE_COMMAND} -E env "PYTHONPATH=${REGISTRY_PATH}"
        ${VENV_PYTHON_EXECUTABLE} "${CMAKE_CURRENT_SOURCE_DIR}/../vkprofiles.py" transform
        --registry "${REGISTRY_PATH}/vk.xml"
        --input "${PROFILES_INPUT_DIR}"
        --output "${PROFILES_OUTPUT_DIR}"
        --pull required-capabilities
               ignore-extension-versions
               aliases
        --strip duplication
        --format flatten
    WORKING_DIRECTORY "${CMAKE_CURRENT_SOURCE_DIR}"
    COMMENT "Testing vkprofiles with Android Vulkan profiles..."
    VERBATIM
)
add_dependencies(transform_android_profiles VpProfilesProcessor python_venv)