playcanvas
Version:
Open-source WebGL/WebGPU 3D engine for the web
68 lines (67 loc) • 2.15 kB
JavaScript
var gsplatCenter_default = `
uniform mat4 matrix_model;
uniform mat4 matrix_view;
uniform vec4 camera_params;
uniform mat4 matrix_projection;
uniform float fisheye_k;
uniform float fisheye_inv_k;
uniform float fisheye_projMat00;
uniform float fisheye_projMat11;
bool initCenter(vec3 modelCenter, inout SplatCenter center) {
mat4 modelView = matrix_view * matrix_model;
vec4 centerView = modelView * vec4(modelCenter, 1.0);
vec3 v = centerView.xyz;
float r_xy = length(v.xy);
float neg_z = -v.z;
float theta = atan(r_xy, neg_z);
float maxTheta = min(fisheye_k * 1.5707963, 3.13);
if (theta > maxTheta - 0.01 || dot(v, v) < 0.0001) {
return false;
}
float tk = theta * fisheye_inv_k;
float sin_tk = sin(tk);
float cos_tk = cos(tk);
float g_theta = fisheye_k * sin_tk / cos_tk;
float fisheye_s = (r_xy > 1e-4) ? g_theta / r_xy : (neg_z > 0.0 ? 1.0 / neg_z : 0.0);
vec2 ndc = vec2(fisheye_projMat00 * fisheye_s * v.x, fisheye_projMat11 * fisheye_s * v.y);
float near = camera_params.z;
float far = camera_params.y;
float linearDepth = neg_z;
float depthNdc = clamp((linearDepth - near) / (far - near), 0.0, 1.0);
float depthNdc = clamp(2.0 * (linearDepth - near) / (far - near) - 1.0, -1.0, 1.0);
center.proj = vec4(ndc, depthNdc, 1.0);
center.projMat00 = fisheye_projMat00;
center.fisheyeSinTK = sin_tk;
center.fisheyeCosTK = cos_tk;
center.fisheyeRxy = r_xy;
if (camera_params.w != 1.0 && centerView.z > 0.0) {
return false;
}
vec4 centerProj = matrix_projection * centerView;
centerProj.z = clamp(centerProj.z, 0, abs(centerProj.w));
centerProj.z = clamp(centerProj.z, -abs(centerProj.w), abs(centerProj.w));
center.proj = centerProj;
center.projMat00 = matrix_projection[0][0];
center.view = centerView.xyz / centerView.w;
center.modelView = modelView;
return true;
}
`;
export {
gsplatCenter_default as default
};