From da5b7d6b9ee35c19b237b6cdc1a073e12d022144 Mon Sep 17 00:00:00 2001 From: hyzboy Date: Sat, 22 Aug 2020 18:12:50 +0800 Subject: [PATCH] first commit --- CMakeLists.txt | 33 ++++++++ FindVulkan.cmake | 87 +++++++++++++++++++ glsl2spv.cpp | 216 +++++++++++++++++++++++++++++++++++++++++++++++ 3 files changed, 336 insertions(+) create mode 100644 CMakeLists.txt create mode 100644 FindVulkan.cmake create mode 100644 glsl2spv.cpp diff --git a/CMakeLists.txt b/CMakeLists.txt new file mode 100644 index 0000000..0bea0dd --- /dev/null +++ b/CMakeLists.txt @@ -0,0 +1,33 @@ +cmake_minimum_required(VERSION 3.0) +SET(CMAKE_ALLOW_LOOSE_LOOP_CONSTRUCTS TRUE) + +project(GLSLCompiler) + +if(MSVC) + SET(CMAKE_C_FLAGS_DEBUG "${CMAKE_C_FLAGS_DEBUG} /MD") + SET(CMAKE_CXX_FLAGS_DEBUG "${CMAKE_CXX_FLAGS_DEBUG} /MD") + + SET(CMAKE_C_FLAGS_RELEASE "${CMAKE_C_FLAGS_RELEASE} /MD") + SET(CMAKE_CXX_FLAGS_RELEASE "${CMAKE_CXX_FLAGS_RELEASE} /MD") + add_definitions(-D_CRT_SECURE_NO_WARNINGS) +endif() + +include("FindVulkan.cmake") + +include_directories(${Vulkan_INCLUDE_DIRS}) +link_directories(${Vulkan_LIBRARIES_DIR}) + +set(VULKAN_SPIRV_LIBS GenericCodeGen + glslang + HLSL + MachineIndependent + OGLCompiler + OSDependent + SPIRV + SPIRV-Tools + SPIRV-Tools-opt + spirv-cross-core) + +add_library(GLSLCompiler SHARED glsl2spv.cpp) + +target_link_libraries(GLSLCompiler PRIVATE ${VULKAN_SPIRV_LIBS} ${Vulkan_LIBRARY}) diff --git a/FindVulkan.cmake b/FindVulkan.cmake new file mode 100644 index 0000000..607d985 --- /dev/null +++ b/FindVulkan.cmake @@ -0,0 +1,87 @@ +# Distributed under the OSI-approved BSD 3-Clause License. See accompanying +# file Copyright.txt or https://cmake.org/licensing for details. + +#.rst: +# FindVulkan +# ---------- +# +# Try to find Vulkan +# +# IMPORTED Targets +# ^^^^^^^^^^^^^^^^ +# +# This module defines :prop_tgt:`IMPORTED` target ``Vulkan::Vulkan``, if +# Vulkan has been found. +# +# Result Variables +# ^^^^^^^^^^^^^^^^ +# +# This module defines the following variables:: +# +# Vulkan_FOUND - True if Vulkan was found +# Vulkan_INCLUDE_DIRS - include directories for Vulkan +# Vulkan_LIBRARIES - link against this library to use Vulkan +# +# The module will also define two cache variables:: +# +# Vulkan_INCLUDE_DIR - the Vulkan include directory +# Vulkan_LIBRARY - the path to the Vulkan library +# + +if(WIN32) + find_path(Vulkan_INCLUDE_DIR + NAMES vulkan/vulkan.h + PATHS + "$ENV{VULKAN_SDK}/Include" + ) + + if(CMAKE_SIZEOF_VOID_P EQUAL 8) + find_library(Vulkan_LIBRARY + NAMES vulkan-1 + PATHS + "$ENV{VULKAN_SDK}/Lib" + "$ENV{VULKAN_SDK}/Bin" + ) + + set(Vulkan_LIBRARIES_DIR $ENV{VULKAN_SDK}/Lib) + elseif(CMAKE_SIZEOF_VOID_P EQUAL 4) + find_library(Vulkan_LIBRARY + NAMES vulkan-1 + PATHS + "$ENV{VULKAN_SDK}/Lib32" + "$ENV{VULKAN_SDK}/Bin32" + NO_SYSTEM_ENVIRONMENT_PATH + ) + + set(Vulkan_LIBRARIES_DIR $ENV{VULKAN_SDK}/Lib32) + endif() +else() + find_path(Vulkan_INCLUDE_DIR + NAMES vulkan/vulkan.h + PATHS + "$ENV{VULKAN_SDK}/include") + find_library(Vulkan_LIBRARY + NAMES vulkan + PATHS + "$ENV{VULKAN_SDK}/lib") + + set(Vulkan_LIBRARIES_DIR $ENV{VULKAN_SDK}/Lib) +endif() + +set(Vulkan_LIBRARIES ${Vulkan_LIBRARY}) +set(Vulkan_INCLUDE_DIRS ${Vulkan_INCLUDE_DIR}) + +#include(${CMAKE_CURRENT_LIST_DIR}/FindPackageHandleStandardArgs.cmake) +include(FindPackageHandleStandardArgs) +find_package_handle_standard_args(Vulkan + DEFAULT_MSG + Vulkan_LIBRARY Vulkan_INCLUDE_DIR) + +mark_as_advanced(Vulkan_INCLUDE_DIR Vulkan_LIBRARY) + +if(Vulkan_FOUND AND NOT TARGET Vulkan::Vulkan) + add_library(Vulkan::Vulkan UNKNOWN IMPORTED) + set_target_properties(Vulkan::Vulkan PROPERTIES + IMPORTED_LOCATION "${Vulkan_LIBRARIES}" + INTERFACE_INCLUDE_DIRECTORIES "${Vulkan_INCLUDE_DIRS}") +endif() diff --git a/glsl2spv.cpp b/glsl2spv.cpp new file mode 100644 index 0000000..2787b7f --- /dev/null +++ b/glsl2spv.cpp @@ -0,0 +1,216 @@ +#include +#include +#include +#include +#include + +static TBuiltInResource default_build_in_resource; + +void init_default_build_in_resource() +{ + default_build_in_resource.maxLights = 32; + default_build_in_resource.maxClipPlanes = 6; + default_build_in_resource.maxTextureUnits = 32; + default_build_in_resource.maxTextureCoords = 32; + default_build_in_resource.maxVertexAttribs = 64; + default_build_in_resource.maxVertexUniformComponents = 4096; + default_build_in_resource.maxVaryingFloats = 64; + default_build_in_resource.maxVertexTextureImageUnits = 32; + default_build_in_resource.maxCombinedTextureImageUnits = 80; + default_build_in_resource.maxTextureImageUnits = 32; + default_build_in_resource.maxFragmentUniformComponents = 4096; + default_build_in_resource.maxDrawBuffers = 32; + default_build_in_resource.maxVertexUniformVectors = 128; + default_build_in_resource.maxVaryingVectors = 8; + default_build_in_resource.maxFragmentUniformVectors = 16; + default_build_in_resource.maxVertexOutputVectors = 16; + default_build_in_resource.maxFragmentInputVectors = 15; + default_build_in_resource.minProgramTexelOffset = -8; + default_build_in_resource.maxProgramTexelOffset = 7; + default_build_in_resource.maxClipDistances = 8; + default_build_in_resource.maxComputeWorkGroupCountX = 65535; + default_build_in_resource.maxComputeWorkGroupCountY = 65535; + default_build_in_resource.maxComputeWorkGroupCountZ = 65535; + default_build_in_resource.maxComputeWorkGroupSizeX = 1024; + default_build_in_resource.maxComputeWorkGroupSizeY = 1024; + default_build_in_resource.maxComputeWorkGroupSizeZ = 64; + default_build_in_resource.maxComputeUniformComponents = 1024; + default_build_in_resource.maxComputeTextureImageUnits = 16; + default_build_in_resource.maxComputeImageUniforms = 8; + default_build_in_resource.maxComputeAtomicCounters = 8; + default_build_in_resource.maxComputeAtomicCounterBuffers = 1; + default_build_in_resource.maxVaryingComponents = 60; + default_build_in_resource.maxVertexOutputComponents = 64; + default_build_in_resource.maxGeometryInputComponents = 64; + default_build_in_resource.maxGeometryOutputComponents = 128; + default_build_in_resource.maxFragmentInputComponents = 128; + default_build_in_resource.maxImageUnits = 8; + default_build_in_resource.maxCombinedImageUnitsAndFragmentOutputs = 8; + default_build_in_resource.maxCombinedShaderOutputResources = 8; + default_build_in_resource.maxImageSamples = 0; + default_build_in_resource.maxVertexImageUniforms = 0; + default_build_in_resource.maxTessControlImageUniforms = 0; + default_build_in_resource.maxTessEvaluationImageUniforms = 0; + default_build_in_resource.maxGeometryImageUniforms = 0; + default_build_in_resource.maxFragmentImageUniforms = 8; + default_build_in_resource.maxCombinedImageUniforms = 8; + default_build_in_resource.maxGeometryTextureImageUnits = 16; + default_build_in_resource.maxGeometryOutputVertices = 256; + default_build_in_resource.maxGeometryTotalOutputComponents = 1024; + default_build_in_resource.maxGeometryUniformComponents = 1024; + default_build_in_resource.maxGeometryVaryingComponents = 64; + default_build_in_resource.maxTessControlInputComponents = 128; + default_build_in_resource.maxTessControlOutputComponents = 128; + default_build_in_resource.maxTessControlTextureImageUnits = 16; + default_build_in_resource.maxTessControlUniformComponents = 1024; + default_build_in_resource.maxTessControlTotalOutputComponents = 4096; + default_build_in_resource.maxTessEvaluationInputComponents = 128; + default_build_in_resource.maxTessEvaluationOutputComponents = 128; + default_build_in_resource.maxTessEvaluationTextureImageUnits = 16; + default_build_in_resource.maxTessEvaluationUniformComponents = 1024; + default_build_in_resource.maxTessPatchComponents = 120; + default_build_in_resource.maxPatchVertices = 32; + default_build_in_resource.maxTessGenLevel = 64; + default_build_in_resource.maxViewports = 16; + default_build_in_resource.maxVertexAtomicCounters = 0; + default_build_in_resource.maxTessControlAtomicCounters = 0; + default_build_in_resource.maxTessEvaluationAtomicCounters = 0; + default_build_in_resource.maxGeometryAtomicCounters = 0; + default_build_in_resource.maxFragmentAtomicCounters = 8; + default_build_in_resource.maxCombinedAtomicCounters = 8; + default_build_in_resource.maxAtomicCounterBindings = 1; + default_build_in_resource.maxVertexAtomicCounterBuffers = 0; + default_build_in_resource.maxTessControlAtomicCounterBuffers = 0; + default_build_in_resource.maxTessEvaluationAtomicCounterBuffers = 0; + default_build_in_resource.maxGeometryAtomicCounterBuffers = 0; + default_build_in_resource.maxFragmentAtomicCounterBuffers = 1; + default_build_in_resource.maxCombinedAtomicCounterBuffers = 1; + default_build_in_resource.maxAtomicCounterBufferSize = 16384; + default_build_in_resource.maxTransformFeedbackBuffers = 4; + default_build_in_resource.maxTransformFeedbackInterleavedComponents = 64; + default_build_in_resource.maxCullDistances = 8; + default_build_in_resource.maxCombinedClipAndCullDistances = 8; + default_build_in_resource.maxSamples = 4; + default_build_in_resource.maxMeshOutputVerticesNV = 256; + default_build_in_resource.maxMeshOutputPrimitivesNV = 512; + default_build_in_resource.maxMeshWorkGroupSizeX_NV = 32; + default_build_in_resource.maxMeshWorkGroupSizeY_NV = 1; + default_build_in_resource.maxMeshWorkGroupSizeZ_NV = 1; + default_build_in_resource.maxTaskWorkGroupSizeX_NV = 32; + default_build_in_resource.maxTaskWorkGroupSizeY_NV = 1; + default_build_in_resource.maxTaskWorkGroupSizeZ_NV = 1; + default_build_in_resource.maxMeshViewCountNV = 4; + default_build_in_resource.limits.nonInductiveForLoops = 1; + default_build_in_resource.limits.whileLoops = 1; + default_build_in_resource.limits.doWhileLoops = 1; + default_build_in_resource.limits.generalUniformIndexing = 1; + default_build_in_resource.limits.generalAttributeMatrixVectorIndexing = 1; + default_build_in_resource.limits.generalVaryingIndexing = 1; + default_build_in_resource.limits.generalSamplerIndexing = 1; + default_build_in_resource.limits.generalVariableIndexing = 1; + default_build_in_resource.limits.generalConstantMatrixVectorIndexing = 1; +} + +EShLanguage FindLanguage(const VkShaderStageFlagBits shader_type) +{ + switch (shader_type) + { + case VK_SHADER_STAGE_VERTEX_BIT: + return EShLangVertex; + + case VK_SHADER_STAGE_TESSELLATION_CONTROL_BIT: + return EShLangTessControl; + + case VK_SHADER_STAGE_TESSELLATION_EVALUATION_BIT: + return EShLangTessEvaluation; + + case VK_SHADER_STAGE_GEOMETRY_BIT: + return EShLangGeometry; + + case VK_SHADER_STAGE_FRAGMENT_BIT: + return EShLangFragment; + + case VK_SHADER_STAGE_COMPUTE_BIT: + return EShLangCompute; + + case VK_SHADER_STAGE_TASK_BIT_NV: + return EShLangTaskNV; + + case VK_SHADER_STAGE_MESH_BIT_NV: + return EShLangMeshNV; + + default: + return EShLangVertex; + } +} + +extern "C" +{ + bool InitShaderCompiler() + { + init_default_build_in_resource(); + + return glslang::InitializeProcess(); + } + + void CloseShaderCompiler() + { + glslang::FinalizeProcess(); + } + + bool GLSL2SPV(const VkShaderStageFlagBits shader_type,const char *shader_source,std::vector &spirv) + { + EShLanguage stage = FindLanguage(shader_type); + + glslang::TShader shader(stage); + glslang::TProgram program; + + const char *shaderStrings[1]; + TBuiltInResource Resources=default_build_in_resource; + + // Enable SPIR-V and Vulkan rules when parsing GLSL + EShMessages messages = (EShMessages)(EShMsgSpvRules | EShMsgVulkanRules); + + shaderStrings[0] = shader_source; + shader.setStrings(shaderStrings, 1); + + if (!shader.parse(&Resources, 100, false, messages)) + { + puts(shader.getInfoLog()); + puts(shader.getInfoDebugLog()); + return false; // something didn't work + } + + program.addShader(&shader); + + // + // Program-level processing... + // + + if (!program.link(messages)) + { + puts(shader.getInfoLog()); + puts(shader.getInfoDebugLog()); + fflush(stdout); + return false; + } + + glslang::GlslangToSpv(*program.getIntermediate(stage),spirv); + return(true); + } + + VkShaderStageFlagBits GetShaderStageFlagByExtName(const char *ext_name) + { + if (stricmp(ext_name,"vert") == 0)return VK_SHADER_STAGE_VERTEX_BIT; else + if (stricmp(ext_name,"tesc") == 0)return VK_SHADER_STAGE_TESSELLATION_CONTROL_BIT; else + if (stricmp(ext_name,"tese") == 0)return VK_SHADER_STAGE_TESSELLATION_EVALUATION_BIT; else + if (stricmp(ext_name,"geom") == 0)return VK_SHADER_STAGE_GEOMETRY_BIT; else + if (stricmp(ext_name,"frag") == 0)return VK_SHADER_STAGE_FRAGMENT_BIT; else + if (stricmp(ext_name,"comp") == 0)return VK_SHADER_STAGE_COMPUTE_BIT; else + if (stricmp(ext_name,"task") == 0)return VK_SHADER_STAGE_TASK_BIT_NV; else + if (stricmp(ext_name,"mesh") == 0)return VK_SHADER_STAGE_MESH_BIT_NV; else + { + return (VkShaderStageFlagBits)0; + } + } +}//extern "C" \ No newline at end of file