playcanvas
Version:
Open-source WebGL/WebGPU 3D engine for the web
3 lines (2 loc) • 1.94 kB
TypeScript
declare const _default: "\nfn writeSplat(center: vec3f, rotation: vec4f, scale: vec3f, color: vec4f) {\n // Pack RGB as 11+11+10 bits into R32U, range [0, 4]\n let rgb = clamp(color.rgb, vec3f(0.0), vec3f(4.0));\n let rBits = u32(rgb.r * (2047.0 / 4.0) + 0.5);\n let gBits = u32(rgb.g * (2047.0 / 4.0) + 0.5);\n let bBits = u32(rgb.b * (1023.0 / 4.0) + 0.5);\n writeDataColor(vec4u(rBits | (gBits << 11u) | (bBits << 22u), 0u, 0u, 0u));\n\n #ifndef GSPLAT_COLOR_ONLY\n // Half-angle quaternion projection: rotation is (x,y,z,w) with w >= 0\n var q = rotation;\n if (q.w < 0.0) { q = -q; }\n let p = q.xyz * inverseSqrt(1.0 + q.w);\n\n // quantize from [-1, 1] to 11+11+10 bits\n let aBitsQ = u32(clamp(p.x * 0.5 + 0.5, 0.0, 1.0) * 2047.0 + 0.5);\n let bBitsQ = u32(clamp(p.y * 0.5 + 0.5, 0.0, 1.0) * 2047.0 + 0.5);\n let cBitsQ = u32(clamp(p.z * 0.5 + 0.5, 0.0, 1.0) * 1023.0 + 0.5);\n let packedQuat = aBitsQ | (bBitsQ << 11u) | (cBitsQ << 22u);\n\n // Log-encode scale (3x8 bits) + alpha (8 bits) into dataTransformA.w\n let invLogRange = 255.0 / 21.0;\n let logMin = -12.0;\n let sxBits = select(u32(clamp((log(scale.x) - logMin) * invLogRange + 0.5, 1.0, 255.0)), 0u, scale.x < 1e-10);\n let syBits = select(u32(clamp((log(scale.y) - logMin) * invLogRange + 0.5, 1.0, 255.0)), 0u, scale.y < 1e-10);\n let szBits = select(u32(clamp((log(scale.z) - logMin) * invLogRange + 0.5, 1.0, 255.0)), 0u, scale.z < 1e-10);\n let alphaBits = u32(clamp(color.a, 0.0, 1.0) * 255.0 + 0.5);\n let packedScaleAlpha = sxBits | (syBits << 8u) | (szBits << 16u) | (alphaBits << 24u);\n\n writeDataTransformA(vec4u(bitcast<u32>(center.x), bitcast<u32>(center.y), bitcast<u32>(center.z), packedScaleAlpha));\n\n writeDataTransformB(vec4u(packedQuat, 0u, 0u, 0u));\n #endif\n}\n";
export default _default;