|
| 1 | +include_guard(GLOBAL) |
| 2 | + |
| 3 | +function(_webvulkan_resolve_node OUT_VAR) |
| 4 | + if(DEFINED WEBVULKAN_NODE_BIN AND NOT WEBVULKAN_NODE_BIN STREQUAL "") |
| 5 | + set(_node "${WEBVULKAN_NODE_BIN}") |
| 6 | + else() |
| 7 | + find_program(_node NAMES node node.exe) |
| 8 | + endif() |
| 9 | + if(NOT _node OR NOT EXISTS "${_node}") |
| 10 | + message(FATAL_ERROR "Node.js executable not found. Set WEBVULKAN_NODE_BIN to continue.") |
| 11 | + endif() |
| 12 | + set(${OUT_VAR} "${_node}" PARENT_SCOPE) |
| 13 | +endfunction() |
| 14 | + |
| 15 | +function(_webvulkan_resolve_wasmer OUT_VAR) |
| 16 | + if(DEFINED WEBVULKAN_WASMER_BIN AND NOT WEBVULKAN_WASMER_BIN STREQUAL "") |
| 17 | + set(_wasmer "${WEBVULKAN_WASMER_BIN}") |
| 18 | + else() |
| 19 | + find_program(_wasmer NAMES wasmer wasmer.exe) |
| 20 | + endif() |
| 21 | + if(NOT _wasmer OR NOT EXISTS "${_wasmer}") |
| 22 | + message(FATAL_ERROR "Wasmer executable not found. Set WEBVULKAN_WASMER_BIN to continue.") |
| 23 | + endif() |
| 24 | + set(${OUT_VAR} "${_wasmer}" PARENT_SCOPE) |
| 25 | +endfunction() |
| 26 | + |
| 27 | +function(webvulkan_compile_opencl_to_spirv) |
| 28 | + set(options) |
| 29 | + set(oneValueArgs SOURCE OUTPUT NODE_BIN WASMER_BIN CLANG_PACKAGE SPIRV_PACKAGE SPIRV_ENTRYPOINT) |
| 30 | + set(multiValueArgs DEPENDS) |
| 31 | + cmake_parse_arguments(WEBVULKAN_SHADER "${options}" "${oneValueArgs}" "${multiValueArgs}" ${ARGN}) |
| 32 | + |
| 33 | + if(NOT WEBVULKAN_SHADER_SOURCE) |
| 34 | + message(FATAL_ERROR "webvulkan_compile_opencl_to_spirv requires SOURCE") |
| 35 | + endif() |
| 36 | + if(NOT WEBVULKAN_SHADER_OUTPUT) |
| 37 | + message(FATAL_ERROR "webvulkan_compile_opencl_to_spirv requires OUTPUT") |
| 38 | + endif() |
| 39 | + |
| 40 | + if(NOT DEFINED WEBVULKAN_SHADER_COMPILER_SCRIPT OR WEBVULKAN_SHADER_COMPILER_SCRIPT STREQUAL "") |
| 41 | + set(_default_script "${CMAKE_CURRENT_LIST_DIR}/../tools/webvulkan_compile_spirv.mjs") |
| 42 | + if(EXISTS "${_default_script}") |
| 43 | + set(WEBVULKAN_SHADER_COMPILER_SCRIPT "${_default_script}") |
| 44 | + endif() |
| 45 | + endif() |
| 46 | + if(NOT EXISTS "${WEBVULKAN_SHADER_COMPILER_SCRIPT}") |
| 47 | + message(FATAL_ERROR "WEBVULKAN_SHADER_COMPILER_SCRIPT does not exist: ${WEBVULKAN_SHADER_COMPILER_SCRIPT}") |
| 48 | + endif() |
| 49 | + |
| 50 | + if(WEBVULKAN_SHADER_NODE_BIN) |
| 51 | + set(_node_bin "${WEBVULKAN_SHADER_NODE_BIN}") |
| 52 | + else() |
| 53 | + _webvulkan_resolve_node(_node_bin) |
| 54 | + endif() |
| 55 | + if(NOT EXISTS "${_node_bin}") |
| 56 | + message(FATAL_ERROR "Node.js executable does not exist: ${_node_bin}") |
| 57 | + endif() |
| 58 | + |
| 59 | + if(WEBVULKAN_SHADER_WASMER_BIN) |
| 60 | + set(_wasmer_bin "${WEBVULKAN_SHADER_WASMER_BIN}") |
| 61 | + else() |
| 62 | + _webvulkan_resolve_wasmer(_wasmer_bin) |
| 63 | + endif() |
| 64 | + if(NOT EXISTS "${_wasmer_bin}") |
| 65 | + message(FATAL_ERROR "Wasmer executable does not exist: ${_wasmer_bin}") |
| 66 | + endif() |
| 67 | + |
| 68 | + if(WEBVULKAN_SHADER_CLANG_PACKAGE) |
| 69 | + set(_clang_package "${WEBVULKAN_SHADER_CLANG_PACKAGE}") |
| 70 | + elseif(DEFINED WEBVULKAN_CLANG_WASM_PACKAGE AND NOT WEBVULKAN_CLANG_WASM_PACKAGE STREQUAL "") |
| 71 | + set(_clang_package "${WEBVULKAN_CLANG_WASM_PACKAGE}") |
| 72 | + else() |
| 73 | + set(_clang_package "clang/clang") |
| 74 | + endif() |
| 75 | + |
| 76 | + if(WEBVULKAN_SHADER_SPIRV_PACKAGE) |
| 77 | + set(_spirv_package "${WEBVULKAN_SHADER_SPIRV_PACKAGE}") |
| 78 | + elseif(DEFINED WEBVULKAN_SPIRV_WASM_PACKAGE AND NOT WEBVULKAN_SPIRV_WASM_PACKAGE STREQUAL "") |
| 79 | + set(_spirv_package "${WEBVULKAN_SPIRV_WASM_PACKAGE}") |
| 80 | + else() |
| 81 | + set(_spirv_package "lights0123/llvm-spir") |
| 82 | + endif() |
| 83 | + |
| 84 | + if(WEBVULKAN_SHADER_SPIRV_ENTRYPOINT) |
| 85 | + set(_spirv_entrypoint "${WEBVULKAN_SHADER_SPIRV_ENTRYPOINT}") |
| 86 | + elseif(DEFINED WEBVULKAN_SPIRV_WASM_ENTRYPOINT AND NOT WEBVULKAN_SPIRV_WASM_ENTRYPOINT STREQUAL "") |
| 87 | + set(_spirv_entrypoint "${WEBVULKAN_SPIRV_WASM_ENTRYPOINT}") |
| 88 | + else() |
| 89 | + set(_spirv_entrypoint "clspv") |
| 90 | + endif() |
| 91 | + |
| 92 | + get_filename_component(_output_dir "${WEBVULKAN_SHADER_OUTPUT}" DIRECTORY) |
| 93 | + if(_output_dir) |
| 94 | + set(_make_dir_command COMMAND "${CMAKE_COMMAND}" -E make_directory "${_output_dir}") |
| 95 | + else() |
| 96 | + set(_make_dir_command) |
| 97 | + endif() |
| 98 | + |
| 99 | + add_custom_command( |
| 100 | + OUTPUT "${WEBVULKAN_SHADER_OUTPUT}" |
| 101 | + ${_make_dir_command} |
| 102 | + COMMAND |
| 103 | + "${_node_bin}" "${WEBVULKAN_SHADER_COMPILER_SCRIPT}" |
| 104 | + --input "${WEBVULKAN_SHADER_SOURCE}" |
| 105 | + --output "${WEBVULKAN_SHADER_OUTPUT}" |
| 106 | + --wasmer "${_wasmer_bin}" |
| 107 | + --clang-package "${_clang_package}" |
| 108 | + --spirv-package "${_spirv_package}" |
| 109 | + --spirv-entrypoint "${_spirv_entrypoint}" |
| 110 | + DEPENDS |
| 111 | + "${WEBVULKAN_SHADER_SOURCE}" |
| 112 | + "${WEBVULKAN_SHADER_COMPILER_SCRIPT}" |
| 113 | + ${WEBVULKAN_SHADER_DEPENDS} |
| 114 | + VERBATIM |
| 115 | + ) |
| 116 | +endfunction() |
0 commit comments