1#version 450 core 2#define PRECISION ${PRECISION} 3#define FORMAT ${FORMAT} 4 5layout(std430) buffer; 6 7/* Qualifiers: layout - storage - precision - memory */ 8 9layout(set = 0, binding = 0, FORMAT) uniform PRECISION restrict writeonly image3D uOutput; 10layout(set = 0, binding = 1) uniform PRECISION sampler3D uInput; 11layout(set = 0, binding = 2) uniform PRECISION restrict Block { 12 ivec4 size; 13 float kBeta; /* M_SQRT2 * M_2_SQRTPI * 0.5 */ 14} uBlock; 15 16layout(local_size_x_id = 0, local_size_y_id = 1, local_size_z_id = 2) in; 17 18void main() { 19 const ivec3 pos = ivec3(gl_GlobalInvocationID); 20 21 if (all(lessThan(pos, uBlock.size.xyz))) { 22 const vec4 inval = texelFetch(uInput, pos, 0); 23 const vec4 invalcube = inval * inval * inval; 24 const vec4 inner = vec4(uBlock.kBeta) * (inval + vec4(0.044715) * invalcube); 25 const vec4 outval = vec4(0.5) * inval * (vec4(1.0) + tanh(inner)); 26 imageStore(uOutput, pos, outval); 27 } 28} 29