playcanvas
Version:
Open-source WebGL/WebGPU 3D engine for the web
86 lines (85 loc) • 2.12 kB
JavaScript
const computeGsplatProjectCommonSource = `
struct ProjectedSplatCommon {
valid: bool,
splatId: u32,
center: vec3f,
opacity: f32,
proj: SplatCov2D
}
fn invalidProjectedSplatCommon() -> ProjectedSplatCommon {
var cov: SplatCov2D;
cov.screen = vec2f(0.0);
cov.a = 0.0;
cov.b = 0.0;
cov.c = 0.0;
cov.viewDepth = 0.0;
cov.valid = false;
return ProjectedSplatCommon(false, 0u, vec3f(0.0), 0.0, cov);
}
fn projectSplatCommon(
threadIdx: u32,
numVisible: u32,
alphaClip: f32,
minPixelSize: f32,
minContribution: f32,
foveationStrength: f32,
foveationCenter: f32,
viewMatrix: mat4x4f,
viewProj: mat4x4f,
focal: f32,
viewportWidth: f32,
viewportHeight: f32,
nearClip: f32,
farClip: f32,
isOrtho: u32,
#ifdef GSPLAT_FISHEYE
fisheye_k: f32,
fisheye_inv_k: f32,
fisheye_projMat00: f32,
fisheye_projMat11: f32,
#endif
#ifdef GSPLAT_XR
viewProj1: mat4x4f,
#endif
) -> ProjectedSplatCommon {
if (threadIdx >= numVisible) {
return invalidProjectedSplatCommon();
}
let splatId = compactedSplatIds[threadIdx];
setSplat(splatId);
let originalCenter = getCenter();
var center = originalCenter;
modifySplatCenter(¢er);
let opacity = getOpacity();
if (opacity <= alphaClip) {
return invalidProjectedSplatCommon();
}
var rotation: vec4f = getRotation().yzwx;
var scale: vec3f = getScale();
modifySplatRotationScale(originalCenter, center, &rotation, &scale);
let proj = computeSplatCov(
center, half4(rotation.wxyz), half3(scale),
viewMatrix, viewProj,
focal, viewportWidth, viewportHeight,
nearClip, farClip, opacity, minPixelSize,
isOrtho, alphaClip, minContribution,
foveationStrength, foveationCenter,
#ifdef GSPLAT_FISHEYE
fisheye_k, fisheye_inv_k,
fisheye_projMat00, fisheye_projMat11,
#endif
#ifdef GSPLAT_XR
viewProj1,
#endif
);
if (!proj.valid) {
return invalidProjectedSplatCommon();
}
return ProjectedSplatCommon(true, splatId, center, opacity, proj);
}
`;
var compute_gsplat_project_common_default = computeGsplatProjectCommonSource;
export {
computeGsplatProjectCommonSource,
compute_gsplat_project_common_default as default
};