review refactors

This commit is contained in:
Metehan Tuncbilek
2024-07-16 18:31:43 +03:00
parent ac1c5f20ec
commit dbd7b2dad5
6 changed files with 79 additions and 71 deletions

View File

@@ -4,17 +4,12 @@ function(SetupVulkan TARGET)
target_link_libraries(${TARGET} PRIVATE ${MoltenVK_LIBRARIES}) target_link_libraries(${TARGET} PRIVATE ${MoltenVK_LIBRARIES})
else () else ()
find_package(Vulkan REQUIRED) find_package(Vulkan REQUIRED)
get_filename_component(Vulkan_SDK_DIR "${Vulkan_LIBRARIES}" DIRECTORY) # Get VulkanSDK/[version]/Bin
target_link_libraries(${TARGET} PRIVATE Vulkan::Vulkan) target_link_libraries(${TARGET} PRIVATE Vulkan::Vulkan)
if(CMAKE_BUILD_TYPE MATCHES "Debug") find_package(Vulkan OPTIONAL_COMPONENTS shaderc_combined)
target_link_libraries(${TARGET} PRIVATE "${Vulkan_SDK_DIR}/shaderc_combinedd.lib" if (Vulkan_shaderc_combined_FOUND)
"${Vulkan_SDK_DIR}/spirv-cross-cored.lib" "${Vulkan_SDK_DIR}/spirv-cross-glsld.lib" target_link_libraries(${TARGET} PRIVATE Vulkan::shaderc_combined)
"${Vulkan_SDK_DIR}/spirv-cross-hlsld.lib") target_compile_definitions(${TARGET} PRIVATE HAS_SHADERC)
else() endif ()
target_link_libraries(${TARGET} PRIVATE "${Vulkan_SDK_DIR}/shaderc_combined.lib"
"${Vulkan_SDK_DIR}/spirv-cross-core.lib" "${Vulkan_SDK_DIR}/spirv-cross-glsl.lib"
"${Vulkan_SDK_DIR}/spirv-cross-hlsl.lib")
endif()
endif () endif ()
target_include_directories(${TARGET} PUBLIC ${Vulkan_INCLUDE_DIR}) target_include_directories(${TARGET} PUBLIC ${Vulkan_INCLUDE_DIR})

View File

@@ -70,19 +70,10 @@ namespace OpenVulkano
throw std::runtime_error("Failed to open file '" + filePath + "'!"); throw std::runtime_error("Failed to open file '" + filePath + "'!");
} }
const size_t fileSize = static_cast<size_t>(file.tellg()); const size_t fileSize = static_cast<size_t>(file.tellg());
if (nullTerminateString) Array<char> data(fileSize + nullTerminateString);
{
Array<char> data(fileSize + 1);
file.seekg(0);
file.read(data.Data(), fileSize);
data[fileSize] = '\0';
file.close();
return data;
}
Array<char> data(fileSize);
file.seekg(0); file.seekg(0);
file.read(data.Data(), fileSize); file.read(data.Data(), fileSize);
if (nullTerminateString) data[fileSize] = '\0';
file.close(); file.close();
return data; return data;
} }

View File

@@ -2,41 +2,93 @@
#include "Base/Logger.hpp" #include "Base/Logger.hpp"
#include <map>
#include <filesystem> #include <filesystem>
namespace OpenVulkano namespace OpenVulkano
{ {
Unique<void*> ShaderCompiler::CompileGLSLToSpirv(const std::string& absPath, const std::string& incPath, #if defined(HAS_SHADERC)
const std::string& entryPoint, shaderc_shader_kind shaderStage, class ShaderIncluder : public shaderc::CompileOptions::IncluderInterface
bool bHaveIncludes) {
struct IncludeData
{
shaderc_include_result result = {};
std::string m_fullPath;
Array<char> m_content;
};
public:
ShaderIncluder(const std::string& path);
~ShaderIncluder() override = default;
shaderc_include_result* GetInclude(const char* requestedSource, shaderc_include_type type,
const char* requestingSource, uint64_t includeDepth) override;
void ReleaseInclude(shaderc_include_result* data) override;
private:
std::string ResolveInclude(const std::string& requestedSource) const;
private:
std::string m_includePath;
};
shaderc_shader_kind CheckStage(const std::string& extensionName)
{
std::map<std::string, shaderc_shader_kind> stageMap = {
{ "vert", shaderc_glsl_vertex_shader }, { "frag", shaderc_glsl_fragment_shader },
{ "comp", shaderc_glsl_compute_shader }, { "geom", shaderc_glsl_geometry_shader },
{ "tesc", shaderc_glsl_tess_control_shader }, { "tese", shaderc_glsl_tess_evaluation_shader },
{ "rgen", shaderc_glsl_raygen_shader }, { "rint", shaderc_glsl_intersection_shader },
{ "rahit", shaderc_glsl_anyhit_shader }, { "rchit", shaderc_glsl_closesthit_shader },
{ "rmiss", shaderc_glsl_miss_shader }, { "rcall", shaderc_glsl_callable_shader },
{ "task", shaderc_glsl_task_shader }, { "mesh", shaderc_glsl_mesh_shader }
};
auto it = stageMap.find(extensionName);
if (it != stageMap.end()) return it->second;
else throw std::runtime_error("Failed to find shader stage for extension '" + extensionName + "'!");
}
Array<uint32_t> ShaderCompiler::CompileGLSLToSpirv(const std::string& absPath, bool hasIncludes,
const std::string& incPath, const std::string& entryPoint)
{ {
Array<char> file = Utils::ReadFile(absPath, false, true); Array<char> file = Utils::ReadFile(absPath, false, true);
shaderc::Compiler shaderCompiler; shaderc::Compiler shaderCompiler;
shaderc::CompileOptions options; shaderc::CompileOptions options;
if (bHaveIncludes) options.SetIncluder(std::make_unique<ShaderIncluder>(incPath)); if (hasIncludes && incPath.size() > 0) { options.SetIncluder(std::make_unique<ShaderIncluder>(incPath)); }
options.SetSourceLanguage(shaderc_source_language_glsl); options.SetSourceLanguage(shaderc_source_language_glsl);
options.SetSuppressWarnings(); options.SetSuppressWarnings();
auto extensionSplit = Utils::SplitAtLastOccurrence(absPath, '.');
shaderc::PreprocessedSourceCompilationResult preResult = shaderc::PreprocessedSourceCompilationResult preResult =
shaderCompiler.PreprocessGlsl(file.Data(), shaderStage, entryPoint.c_str(), options); shaderCompiler.PreprocessGlsl(file.Data(), CheckStage(extensionSplit.second), entryPoint.c_str(), options);
#if defined(_DEBUG)
std::string test = static_cast<const char*>(preResult.begin(), preResult.end());
// printf("Preprocessed shader: %s\n\n\n", test.c_str()); // Works fine
// Logger::APP->info("Preprocessed shader: {0}", test); gives error
#endif
if (preResult.GetCompilationStatus() != shaderc_compilation_status_success) if (preResult.GetCompilationStatus() != shaderc_compilation_status_success)
{ {
throw std::runtime_error("Failed preprocessing shader. Reason: " + preResult.GetErrorMessage()); throw std::runtime_error("Failed preprocessing shader. Reason: " + preResult.GetErrorMessage());
} }
shaderc::CompilationResult result = shaderc::CompilationResult<uint32_t> result = shaderCompiler.CompileGlslToSpv(
shaderCompiler.CompileGlslToSpv(static_cast<const char*>(preResult.begin()), shaderStage, "", options); static_cast<const char*>(preResult.begin()), CheckStage(extensionSplit.second), "", options);
if (result.GetCompilationStatus() != shaderc_compilation_status_success) if (result.GetCompilationStatus() != shaderc_compilation_status_success)
{ {
throw std::runtime_error("Failed compiling shader. Reason: " + result.GetErrorMessage()); throw std::runtime_error("Failed compiling shader. Reason: " + result.GetErrorMessage());
} }
Unique<void*> spirv = std::make_unique<void*>((void*) result.begin()); Array<uint32_t> returnResult(result.cend() - result.cbegin());
return spirv; // Copy the data to the return array
std::copy(result.begin(), result.end(), returnResult.Data());
return returnResult;
} }
ShaderIncluder::ShaderIncluder(const std::string& path) : m_includePath(path) {} ShaderIncluder::ShaderIncluder(const std::string& path) : m_includePath(path) {}
@@ -46,7 +98,9 @@ namespace OpenVulkano
{ {
IncludeData* includeData = new IncludeData(); IncludeData* includeData = new IncludeData();
includeData->m_fullPath = ResolveInclude(requestedSource); includeData->m_fullPath = ResolveInclude(requestedSource);
includeData->m_content = Utils::ReadFile(includeData->m_fullPath, false, false); // using nullTerminate as true causes crash in here. Needed to use as false.
// Do not null terminate the string
includeData->m_content = Utils::ReadFile(includeData->m_fullPath, false, false);
shaderc_include_result* result = &includeData->result; shaderc_include_result* result = &includeData->result;
result->content = includeData->m_content.Data(); result->content = includeData->m_content.Data();
@@ -70,4 +124,5 @@ namespace OpenVulkano
if (std::filesystem::exists(path)) return path; if (std::filesystem::exists(path)) return path;
else throw std::runtime_error("Failed to resolve include '" + requestedSource + "'!"); else throw std::runtime_error("Failed to resolve include '" + requestedSource + "'!");
} }
#endif
} }

View File

@@ -3,35 +3,10 @@
#include "Base/Utils.hpp" #include "Base/Utils.hpp"
#include "Base/Wrapper.hpp" #include "Base/Wrapper.hpp"
#include <spirv_cross/spirv_hlsl.hpp>
#include <spirv_cross/spirv_glsl.hpp>
#include <shaderc/shaderc.hpp> #include <shaderc/shaderc.hpp>
namespace OpenVulkano namespace OpenVulkano
{ {
class ShaderIncluder : public shaderc::CompileOptions::IncluderInterface
{
struct IncludeData
{
shaderc_include_result result = {};
std::string m_fullPath;
Array<char> m_content;
};
public:
ShaderIncluder(const std::string& path);
~ShaderIncluder() override = default;
shaderc_include_result* GetInclude(const char* requestedSource, shaderc_include_type type,
const char* requestingSource, uint64_t includeDepth) override;
void ReleaseInclude(shaderc_include_result* data) override;
private:
std::string ResolveInclude(const std::string& requestedSource) const;
private:
std::string m_includePath;
};
class ShaderCompiler class ShaderCompiler
{ {
public: public:
@@ -40,10 +15,13 @@ namespace OpenVulkano
* @param incPath - include path that will be used to resolve includes * @param incPath - include path that will be used to resolve includes
* @param entryPoint - the name of the void function in the shader * @param entryPoint - the name of the void function in the shader
* @param shaderStage - type of the shader that needs to be compiled * @param shaderStage - type of the shader that needs to be compiled
* @param bHaveIncludes - if incPath is empty, make it false
*/ */
static Unique<void*> CompileGLSLToSpirv(const std::string& absPath, static Array<uint32_t> CompileGLSLToSpirv(const std::string& absPath, bool hasIncludes,
const std::string& incPath, const std::string& entryPoint, const std::string& incPath = std::string(), const std::string& entryPoint = "main")
shaderc_shader_kind shaderStage, bool bHaveIncludes); #if defined(HAS_SHADERC)
;
#else
{ throw std::runtime_error("Shader compiler is not available on this platform"); }
#endif
}; };
} }

View File

@@ -1,8 +0,0 @@
#version 460
#extension GL_KHR_vulkan_glsl : enable
#include "test1.glsl"
void main()
{
};

View File

@@ -1,3 +0,0 @@
void testfunction()
{
};