detect_stage_from_extension()

std::optional< vk::ShaderStageFlagBits > MayaFlux::Core::VKShaderModule::detect_stage_from_extension ( const std::string &  filepath )

static

Auto-detect shader stage from file extension.

Parameters

filepath

Path to shader file

Returns

Detected stage, or nullopt if unknown extension

Definition at line 552 of file VKShaderModule.cpp.

{
    std::filesystem::path path(filepath);
    std::string ext = path.extension().string();
 
    std::transform(ext.begin(), ext.end(), ext.begin(), ::tolower);
 
    static const std::unordered_map<std::string, vk::ShaderStageFlagBits> extension_map = {
        { ".comp", vk::ShaderStageFlagBits::eCompute },
        { ".vert", vk::ShaderStageFlagBits::eVertex },
        { ".frag", vk::ShaderStageFlagBits::eFragment },
        { ".geom", vk::ShaderStageFlagBits::eGeometry },
        { ".tesc", vk::ShaderStageFlagBits::eTessellationControl },
        { ".tese", vk::ShaderStageFlagBits::eTessellationEvaluation },
        { ".rgen", vk::ShaderStageFlagBits::eRaygenKHR },
        { ".rint", vk::ShaderStageFlagBits::eIntersectionKHR },
        { ".rahit", vk::ShaderStageFlagBits::eAnyHitKHR },
        { ".rchit", vk::ShaderStageFlagBits::eClosestHitKHR },
        { ".rmiss", vk::ShaderStageFlagBits::eMissKHR },
        { ".rcall", vk::ShaderStageFlagBits::eCallableKHR },
        { ".mesh", vk::ShaderStageFlagBits::eMeshEXT },
        { ".task", vk::ShaderStageFlagBits::eTaskEXT }
    };
 
    auto it = extension_map.find(ext);
    if (it != extension_map.end()) {
        return it->second;
    }
 
    return std::nullopt;
}

Referenced by create_from_glsl_file(), and MayaFlux::Portal::Graphics::ShaderFoundry::detect_stage_from_extension().

Here is the caller graph for this function: