emit_glsl_kernel()

std::string MayaFlux::Portal::Graphics::detail::emit_glsl_kernel

(

const ShaderSpec &

spec

)

Emit a complete GLSL compute shader from spec metadata and a KernelSource body.

Binding names are used directly as GLSL identifiers. The body text from KernelSource::parse() is injected verbatim inside main() after uint i and optional ivec2 coord are declared. Push constant fields are accessible as pc.<name>. SSBO arrays are accessible as <name>[i].

Definition at line 1638 of file AsmGenerator.cpp.

{
    const auto& ws = spec.workgroup_size;
    const auto& ks = *spec.kernel;
 
    bool has_image = false;
    for (const auto& b : spec.bindings) {
        if (b.modality == Kakshya::DataModality::IMAGE_2D)
            has_image = true;
    }
 
    std::string o;
    o += "#version 460\n";
    o += "layout(local_size_x = " + std::to_string(ws[0])
        + ", local_size_y = " + std::to_string(ws[1])
        + ", local_size_z = " + std::to_string(ws[2]) + ") in;\n\n";
 
    for (const auto& b : spec.bindings) {
        if (b.modality == Kakshya::DataModality::IMAGE_2D) {
            const std::string qual = (b.direction == BindingDirection::Input)
                ? "readonly"
                : "writeonly";
            o += "layout(set = 0, binding = " + std::to_string(b.binding_index)
                + ", rgba32f) " + qual + " uniform image2D " + b.name + ";\n";
            continue;
        }
        if (b.modality == Kakshya::DataModality::TEXTURE_2D) {
            o += "layout(set = 0, binding = " + std::to_string(b.binding_index)
                + ") uniform sampler2D " + b.name + ";\n";
            continue;
        }
        const auto t = std::string(glsl_type(b.format));
        o += "layout(set = 0, binding = " + std::to_string(b.binding_index)
            + ", std430) buffer Block_" + b.name
            + " { " + t + " " + b.name + "[]; };\n";
    }
 
    if (!spec.pc_fields.empty()) {
        o += "\nlayout(push_constant) uniform PC {\n";
        for (const auto& f : spec.pc_fields)
            o += "    " + std::string(glsl_type(f.format)) + " " + f.name + ";\n";
        o += "} pc;\n";
    }
 
    for (const auto& f : spec.functions) {
        o += "\n" + f.return_type + " " + f.name + "(" + f.params + ") {\n";
        o += f.body;
        o += "\n}\n";
    }
 
    o += "\nvoid main() {\n";
    o += "    uint i = gl_GlobalInvocationID.x;\n";
    if (has_image)
        o += "    ivec2 coord = ivec2(gl_GlobalInvocationID.xy);\n";
 
    for (const auto& f : spec.pc_fields) {
        const auto t = std::string(glsl_type(f.format));
        o += "    " + t + " " + f.name + " = pc." + f.name + ";\n";
    }
 
    o += ks.body;
    o += "\n}\n";
    return o;
}

References b, MayaFlux::Portal::Graphics::ShaderSpec::bindings, MayaFlux::Portal::Graphics::ShaderSpec::functions, MayaFlux::Kakshya::IMAGE_2D, MayaFlux::Portal::Graphics::Input, MayaFlux::Portal::Graphics::ShaderSpec::kernel, MayaFlux::Portal::Graphics::ShaderSpec::pc_fields, MayaFlux::Kakshya::TEXTURE_2D, and MayaFlux::Portal::Graphics::ShaderSpec::workgroup_size.

Referenced by MayaFlux::Portal::Graphics::ShaderFoundry::load_shader().

Here is the caller graph for this function: