playcanvas
Version:
PlayCanvas WebGL game engine
105 lines (80 loc) • 2.39 kB
JavaScript
var gsplatVS = /* glsl */ `
varying mediump vec2 gaussianUV;
varying mediump vec4 gaussianColor;
varying float id;
mediump vec4 discardVec = vec4(0.0, 0.0, 2.0, 1.0);
varying float vLinearDepth;
uniform sampler2D colorRamp;
uniform float colorRampIntensity;
void main(void) {
// read gaussian details
SplatSource source;
if (!initSource(source)) {
gl_Position = discardVec;
return;
}
loadSplatTextures(source);
vec3 modelCenter = readCenter(source);
SplatCenter center;
center.modelCenterOriginal = modelCenter;
modifyCenter(modelCenter);
center.modelCenterModified = modelCenter;
if (!initCenter(modelCenter, center)) {
gl_Position = discardVec;
return;
}
// project center to screen space
SplatCorner corner;
if (!initCorner(source, center, corner)) {
gl_Position = discardVec;
return;
}
// read color
vec4 clr = readColor(source);
// apply AA compensation
clr.a *= corner.aaFactor;
// evaluate spherical harmonics
// calculate the model-space view direction
vec3 dir = normalize(center.view * mat3(center.modelView));
// read sh coefficients
vec3 sh[SH_COEFFS];
float scale;
readSHData(source, sh, scale);
// evaluate
clr.xyz += evalSH(sh, dir) * scale;
modifyColor(modelCenter, clr);
clipCorner(corner, clr.w);
// write output
gl_Position = center.proj + vec4(corner.offset, 0, 0);
gaussianUV = corner.uv;
// Overdraw visualization mode: color by elevation
float t = clamp(modelCenter.y / 20.0, 0.0, 1.0);
vec3 rampColor = textureLod(colorRamp, vec2(t, 0.5), 0.0).rgb;
clr.a *= (1.0 / 32.0) * colorRampIntensity;
gaussianColor = vec4(rampColor, clr.a);
gaussianColor = vec4(prepareOutputFromGamma(max(clr.xyz, 0.0)), clr.w);
id = float(source.id);
vLinearDepth = -center.view.z;
}
`;
export { gsplatVS as default };