onnxruntime/cmake/winml_sdk_helpers.cmake
2019-11-07 11:51:44 -08:00

101 lines
No EOL
3.6 KiB
CMake

cmake_minimum_required(VERSION 3.0)
# utility
function(convert_forward_slashes_to_back input output)
string(REGEX REPLACE "/" "\\\\" backwards ${input})
set(${output} ${backwards} PARENT_SCOPE)
endfunction()
# get window 10 install path from registry
function(get_installed_sdk
sdk_folder # the current sdk folder
output_sdk_version # the current sdk version
)
# return the kit path
get_filename_component(win10_sdk_root "[HKEY_LOCAL_MACHINE\\SOFTWARE\\Microsoft\\Windows Kits\\Installed Roots;KitsRoot10]" ABSOLUTE CACHE)
set(${sdk_folder} ${win10_sdk_root} PARENT_SCOPE)
# return the sdk version
set(${output_sdk_version} ${CMAKE_VS_WINDOWS_TARGET_PLATFORM_VERSION} PARENT_SCOPE)
endfunction()
# current sdk binary directory
function(get_sdk_binary_directory
sdk_folder # the kit path
sdk_version # the sdk version
binary_dir # the output folder variable
)
set(${binary_dir} "${sdk_folder}/bin/${sdk_version}" PARENT_SCOPE)
endfunction()
# current sdk include directory
function(get_sdk_include_folder
sdk_folder # the kit path
sdk_version # the sdk version
include_dir # the output folder variable
)
set(${include_dir} "${sdk_folder}/include/${sdk_version}" PARENT_SCOPE)
endfunction()
# current sdk metadata directory
function(get_sdk_metadata_folder
sdk_folder # the kit path
sdk_version # the sdk version
metadata_dir # the output folder variable
)
set(${metadata_dir} "${sdk_folder}/UnionMetadata/${sdk_version}" PARENT_SCOPE)
endfunction()
# current sdk midl exe path
function(get_sdk_midl_exe
sdk_folder # the kit path
sdk_version # the sdk version
midl_exe_path # the output exe path
)
get_sdk_binary_directory(${sdk_folder} ${sdk_version} bin_dir)
set(${midl_exe_path} "${bin_dir}/x64/midlrt.exe" PARENT_SCOPE)
endfunction()
# current cppwinrt cppwinrt exe path
function(get_installed_sdk_cppwinrt_exe
sdk_folder # the kit path
sdk_version # the sdk version
cppwinrt_exe_path # the output exe path
)
get_sdk_binary_directory(${sdk_folder} ${sdk_version} bin_dir)
set(${cppwinrt_exe_path} "${bin_dir}/x64/cppwinrt.exe" PARENT_SCOPE)
endfunction()
# current cppwinrt cppwinrt exe path
function(get_sdk_cppwinrt_exe
sdk_folder # the kit path
sdk_version # the sdk version
output_cppwinrt_exe_path # the output exe path
)
if (NOT DEFINED winml_CPPWINRT_EXE_PATH_OVERRIDE)
get_installed_sdk_cppwinrt_exe(${sdk_folder} ${sdk_version} cppwinrt_exe_path)
set(${output_cppwinrt_exe_path} ${cppwinrt_exe_path} PARENT_SCOPE)
else ()
set(${output_cppwinrt_exe_path} ${winml_CPPWINRT_EXE_PATH_OVERRIDE} PARENT_SCOPE)
endif()
endfunction()
function(get_sdk
output_sdk_folder # the path to the current sdk kit folder
output_sdk_version # the current sdk version
)
if ((NOT DEFINED winml_WINDOWS_SDK_DIR_OVERRIDE) AND
(NOT DEFINED winml_WINDOWS_SDK_VERSION_OVERRIDE))
get_installed_sdk(sdk_folder sdk_version)
set(${output_sdk_folder} ${sdk_folder} PARENT_SCOPE)
set(${output_sdk_version} ${sdk_version} PARENT_SCOPE)
elseif ((DEFINED winml_WINDOWS_SDK_DIR_OVERRIDE) AND
(DEFINED winml_WINDOWS_SDK_VERSION_OVERRIDE))
set(${output_sdk_folder} ${winml_WINDOWS_SDK_DIR_OVERRIDE} PARENT_SCOPE)
set(${output_sdk_version} ${winml_WINDOWS_SDK_VERSION_OVERRIDE} PARENT_SCOPE)
else()
message(
FATAL_ERROR
"Options winml_WINDOWS_SDK_DIR_OVERRIDE and winml_WINDOWS_SDK_VERSION_OVERRIDE must be defined together, or not at all.")
endif()
endfunction()