NazaraEngine/bin/resources/god_rays.nzsl

96 lines
1.7 KiB
Plaintext

[layout(std140)]
struct ViewerData
{
projectionMatrix: mat4<f32>,
invProjectionMatrix: mat4<f32>,
viewMatrix: mat4<f32>,
invViewMatrix: mat4<f32>,
viewProjMatrix: mat4<f32>,
invViewProjMatrix: mat4<f32>,
renderTargetSize: vec2<f32>,
invRenderTargetSize: vec2<f32>,
eyePosition: vec3<f32>
}
[layout(std140)]
struct Settings
{
exposure: f32,
decay: f32,
density: f32,
weight: f32,
lightPosition: vec2<f32>, //< TODO: Switch to world position
}
const SampleCount: i32 = 200;
external
{
[set(0), binding(0)] viewerData: uniform<ViewerData>,
[set(0), binding(1)] settings: uniform<Settings>,
[set(0), binding(2)] occluderTexture: sampler2D<f32>
}
struct FragIn
{
[builtin(fragcoord)] fragcoord: vec4<f32>,
[location(0)] uv: vec2<f32>
}
struct FragOut
{
[location(0)] color: vec4<f32>
}
struct VertIn
{
[location(0)] pos: vec2<f32>,
[location(1)] uv: vec2<f32>
}
struct VertOut
{
[builtin(position)] position: vec4<f32>,
[location(0)] uv: vec2<f32>
}
[entry(frag)]
fn main(input: FragIn) -> FragOut
{
let deltaUV = input.uv - settings.lightPosition;
deltaUV *= 1.0 / f32(SampleCount) * settings.density;
let illuminationDecay = 1.0;
let uv = input.uv;
let outputColor = vec4<f32>(0.0, 0.0, 0.0, 1.0);
let i = 0;
while (i < SampleCount)
{
uv -= deltaUV;
let sample = occluderTexture.Sample(uv);
sample *= illuminationDecay * settings.weight;
outputColor += sample;
illuminationDecay *= settings.decay;
i += 1;
}
let output: FragOut;
output.color = outputColor;
return output;
}
[entry(vert)]
fn main(input: VertIn) -> VertOut
{
let output: VertOut;
output.position = vec4<f32>(input.pos, 0.0, 1.0);
output.uv = input.uv;
return output;
}