@meimu/mutracer 0.0.0-stage → 0.1.0

package/lib/renderer/glsl/rayTrace.frag.js

@@ -0,0 +1,619 @@
+import { MAX_PATH_EVENTS } from '../pathLimits.js';
+import cameraRayGlsl from './chunks/cameraRay.glsl.js';
+// Portions adapted from ray-tracing-renderer@0.10.15
+// (https://github.com/hoverinc/ray-tracing-renderer).
+// Copyright (c) 2019 HOVER, MIT license; see LICENSES/RayTracingRenderer-MIT.txt.
+import commonGlsl from './chunks/common.glsl.js';
+import rayTraceCore from './chunks/rayTraceCore.glsl.js';
+import textureLinear from './chunks/textureLinear.glsl.js';
+import materialBufferGlsl from './chunks/materialBuffer.glsl.js';
+import intersect from './chunks/intersect.glsl.js';
+import randGlsl from './chunks/rand.glsl.js';
+import random from './chunks/random.glsl.js';
+import envMap from './chunks/envMap.glsl.js';
+import sampleMaterialGlsl from './chunks/sampleMaterial.glsl.js';
+import sampleLightGlsl from './chunks/sampleLight.glsl.js';
+import { FragmentShader, glsl, Shader } from '../shader.js';
+import openPbrBsdfGlsl from './chunks/openPbrBsdf.glsl.js';
+import samplingGlsl from './chunks/sampling.glsl.js';
+import { isIntelMac } from '../util.js';
+import uvProjectionGlsl from './chunks/uvProjection.glsl.js';
+const doubleTransmissionCheckCode = glsl `
+#if defined(HAS_TRANSMISSION) || defined(HAS_ALPHA)
+if (!si.isTransparency && BOUNCES < 16 && depth * 2 > BOUNCES) {
+aa.abort = true;
+return;
+}
+#endif
+`;
+// bounce fix on intel mac
+// The intel mac only have one bounce if the loop is not unrolled
+// But windows will have extremely slow shader compilation if the loop is unrolled
+// So here the code are injected separately.
+const bounceCodeOnIntelMac = glsl `
+
+for(int vH = 0; vH < BOUNCES; vH++) {{
+if (aa.abort) {
+return vec4(aa.li, aa.alpha);
+}
+bounce(aa, depth, si, mat, outAux);
+}}
+`;
+const bounceCodeOnOther = glsl `
+bool uR = !si.isTransparency;
+int qB = min(BOUNCES, bounceLimit);
+for(int idx = 0; idx < min(BOUNCES * 3, ${MAX_PATH_EVENTS}); idx++) {
+if (uR && idx > qB) {
+aa.abort = true;
+}
+if (aa.abort) {
+return vec4(aa.li, aa.alpha);
+}
+bounce(aa, depth, si, mat, outAux);
+}
+`;
+export default function createRayTraceFragmentShader() {
+    const intelMac = isIntelMac();
+    return new FragmentShader({
+        name: 'rayTraceFrag',
+        outputs: ['color', 'albedo', 'normal', 'moments'],
+        defines: {
+            BOUNCES: 2
+        },
+        uniforms: {
+            cameraApertureRadius: Shader.uniform('float', 0),
+            cameraFocusDistance: Shader.uniform('float', 10),
+            viewProjectionInv: Shader.uniform('mat4'),
+            viewInv: Shader.uniform('mat4'),
+            transmissiveOutputAlphaMin: Shader.uniform('float'),
+            openPbrEnergyLut: Shader.uniform('sampler2D'),
+            viewportSize: Shader.uniform('vec2'),
+            wavelength: Shader.uniform('float', 0),
+            bevelEnabled: Shader.uniform('bool', true),
+            dispersionWavelengths: Shader.uniform('int', 3),
+            bounceLimit: Shader.uniform('int', 8)
+        },
+        includes: [rayTraceCore, materialBufferGlsl, intersect, random, envMap],
+        main: glsl `
+${commonGlsl}
+
+${randGlsl}
+${random.main}
+
+${textureLinear}
+
+${rayTraceCore.main}
+
+${uvProjectionGlsl}
+
+${materialBufferGlsl.main}
+${intersect.main}
+${envMap.main}
+
+${samplingGlsl}
+${openPbrBsdfGlsl}
+
+float mh;
+float bf;
+
+${sampleLightGlsl}
+${sampleMaterialGlsl}
+
+${cameraRayGlsl}
+struct RayTraceAuxiliary {
+
+vec3 albedo;
+vec3 normal;
+bool isBlank;
+
+bool isShadowCatcher;
+};
+
+void bounce(inout Path aa, inout int depth, inout ad si, inout J mat, inout RayTraceAuxiliary outAux) {
+cE = uint(max(depth, 0));
+
+
+
+
+${intelMac ? doubleTransmissionCheckCode : ''}
+
+bool isShadowCatcher = mat.materialType == SHADOW_CATCHER;
+if (isShadowCatcher) {
+if (aa.fromEye) {
+si.hit = false;
+outAux.isShadowCatcher = true;
+}
+else {
+
+
+#ifdef USE_ENVMAP
+if (aa.isNextRefraction) {
+
+
+
+vec4 bg = cw(aa.ray.o, aa.ray.d);
+bg.rgb = ai(bg.rgb, bf);
+float cT = aa.scatteringIsDelta ? 1.0
+: aF(aa.scatteringPdf,
+iv(aS(aa.ray.d)) * envLightSelectionPmf);
+aa.li += cT * aa.beta * bg.rgb * bg.a;
+aa.alpha = max(aa.alpha, bg.a);
+}
+else {
+vec2 tE = aS(aa.ray.d);
+vec3 emission = textureLinear(envMap, tE).rgb;
+emission = ai(emission, bf);
+aa.li += aa.beta * emission *
+aF(
+aa.scatteringPdf,
+iv(tE) * envLightSelectionPmf
+);
+}
+#else
+vec4 bg = cw(aa.ray.o, aa.ray.d);
+bg.rgb = ai(bg.rgb, bf);
+aa.li += aa.beta * bg.rgb * bg.a;
+#endif
+aa.abort = true;
+return;
+}
+}
+
+if (!si.hit) {
+
+vec4 bg = cw(aa.ray.o, aa.ray.d);
+
+
+
+bg.rgb = ai(bg.rgb, bf);
+if (aa.fromEye) {
+aa.li = bg.rgb * bg.a;
+aa.alpha = bg.a;
+
+outAux.isBlank = bg.a < 0.01;
+}
+else {
+Ray cam = hz(v_Texcoord + rN());
+bool mz = dot(cam.d, aa.ray.d) > 0.0;
+if (transmissiveOutputAlphaMin < 1.0) {
+if (mz) {
+aa.alpha = max(aa.alpha, bg.a);
+} else {
+aa.alpha = 1.0;
+}
+}
+#if !defined(USE_ENVMAP)
+
+if (mz) {
+aa.li = aa.beta * aF(aa.scatteringPdf, 1.0) * bg.rgb;
+}
+#endif
+}
+aa.abort = true;
+return;
+}
+
+bool cG = false;
+if ((mat.materialType == STANDARD || mat.materialType == UNLIT)) {
+if (transmissiveOutputAlphaMin < 1.0) {
+
+float alpha = max(1.0 - mat.transmissionWeight, 1.0 - pow(1.0 - transmissiveOutputAlphaMin, float(depth)));
+aa.alpha = max(alpha, aa.alpha);
+}
+float alpha = mat.alpha;
+if (alpha < 1.0) {
+
+
+aa.alpha = max(aa.alpha, alpha);
+if (eo(3u) > alpha) {
+cG = true;
+}
+}
+if (!cG) {
+lO(si, mat, aa);
+}
+else {
+aa.ray.o = si.position + aa.ray.d * EPS;
+}
+}
+
+if (!si.albedoSampled && !cG) {
+
+if ((mat.materialType == STANDARD || mat.materialType == UNLIT) &&
+!(mat.transmissionWeight > 0.5 ||
+(mat.specularRoughness <= 0.02 && mat.baseMetalness > 0.98))) {
+
+
+
+outAux.albedo = mat.baseColor;
+outAux.normal = dot(si.normal, si.position - vec3(viewInv[3])) > 0.0
+? -si.normal
+: si.normal;
+si.albedoSampled = true;
+}
+if (depth > 4) {
+si.albedoSampled = true;
+}
+}
+
+
+
+#ifdef USE_RR
+if (depth >= RR_MAX && !cG) {
+
+float oj = max(max(aa.beta.x, aa.beta.y), aa.beta.z);
+float q = max(0.0, 1.0 - oj);
+if (dz(cv(RUSSIAN_ROULETTE)) < q) {
+
+
+aa.alpha = 1.0;
+aa.abort = true;
+}
+else {
+aa.beta /= 1.0 - q;
+}
+}
+#endif
+
+if (aa.abort) {
+return;
+}
+
+
+cY(aa.ray, si, true);
+eT(mat, si.materialIndex, si.uv, si.uv1, si.instanceIndex);
+si.isTransparency = mat.transmissionWeight > 0.001 || mat.subsurfaceWeight > 0.001 || mat.alpha < 0.999;
+
+
+
+
+if (dispersionWavelengths == 1 && si.hit && mat.transmissionWeight > 0.001 && mat.transmissionDispersionScale > 0.001) {
+bf = mh;
+}
+if (mat.transmissionWeight > 0.001 && bf > 0.0) {
+mat.specularIor = eK(
+bf,
+mat.specularIor,
+mat.transmissionDispersionScale,
+mat.dispersionAbbeNumber
+);
+}
+
+if (cG) {
+return;
+}
+
+bool hu = eo(3u) < mat.alpha;
+#ifdef NUM_LIGHTS
+
+LightSampleRec lightSample;
+for (int i = 0; i < NUM_LIGHTS; i++) {
+
+Light light = ql(i);
+float d = mk(aa.ray, light, lightSample);
+bool mp = si.hit && d > distance(si.position, aa.ray.o);
+if (d != INF && !(mp && hu)) {
+lightSample.pdf *= light.selectionPmf;
+float eW = aF(aa.scatteringPdf, lightSample.pdf);
+lightSample.emission = ai(lightSample.emission, bf);
+aa.li += eW * aa.beta * lightSample.emission;
+
+aa.alpha = 1.0;
+aa.abort = true;
+}
+}
+#endif
+
+#ifdef USE_ENVMAP
+if (!(si.hit && hu)) {
+if (aa.isNextRefraction) {
+
+
+vec4 bg = cw(aa.ray.o, aa.ray.d);
+bg.rgb = ai(bg.rgb, bf);
+float cT = aa.scatteringIsDelta ? 1.0
+: aF(aa.scatteringPdf,
+iv(aS(aa.ray.d)) * envLightSelectionPmf);
+aa.li += cT * aa.beta * bg.rgb * bg.a;
+aa.alpha = max(aa.alpha, bg.a);
+aa.abort = true;
+return;
+}
+vec2 uv = aS(aa.ray.d);
+vec3 emission = textureLinear(envMap, uv).rgb;
+float eW = aF(
+aa.scatteringPdf,
+iv(uv) * envLightSelectionPmf
+);
+emission = ai(emission, bf);
+aa.li += aa.beta * emission * eW;
+
+
+#if !defined(USE_BACKGROUND)
+aa.alpha = 1.0;
+#endif
+aa.abort = true;
+return;
+}
+#endif
+
+depth++;
+}
+
+
+vec2 kn(float radius, float xi) {
+float x = 0.25;
+for (int np = 0; np < 10; np++) {
+float x2 = x * x;
+float x3 = x2 * x;
+float kD = 1.0 - x;
+float value = 10.0 * x2 - 20.0 * x3 + 15.0 * x2 * x2 -
+4.0 * x2 * x3 - xi;
+float lV = 20.0 * x * kD * kD * kD;
+if (abs(value) < 1e-6 || lV == 0.0) break;
+x = clamp(x - value / lV, 0.0, 1.0);
+}
+float jh = x * radius;
+return vec2(
+jh,
+sqrt(max(radius * radius - jh * jh, 0.0))
+);
+}
+
+float iN(float radius, float gr) {
+if (gr >= radius) return 0.0;
+float nS = radius - gr;
+return nS * nS * nS;
+}
+
+
+vec3 mf(ad source, float radius) {
+vec3 geometricNormal = source.faceNormal;
+if (dot(geometricNormal, source.vertexNormal) < 0.0) geometricNormal = -geometricNormal;
+vec3 tangent;
+vec3 bitangent;
+aK(geometricNormal, tangent, bitangent);
+vec3 iB = vec3(0.0);
+
+for (int ib = 0; ib < BEVEL_SAMPLES; ib++) {
+vec2 oU = fk(CAMERA_BEVEL_BASE + uint(ib));
+float cz = oU.x;
+float ou = oU.y;
+vec3 di = geometricNormal;
+vec3 fE = tangent;
+vec3 ek = bitangent;
+float fZ = 0.5;
+float hV = 0.25;
+float gy = 0.25;
+if (cz < 0.5) {
+cz *= 2.0;
+}
+else if (cz < 0.75) {
+vec3 mx = di;
+di = fE;
+fE = mx;
+fZ = 0.25;
+hV = 0.5;
+cz = (cz - 0.5) * 4.0;
+}
+else {
+vec3 mx = di;
+di = ek;
+ek = mx;
+fZ = 0.25;
+gy = 0.5;
+cz = (cz - 0.75) * 4.0;
+}
+
+vec2 gO = kn(radius, ou);
+float phi = TWOPI * cz;
+vec3 qj = gO.x *
+(cos(phi) * fE + sin(phi) * ek);
+Ray ray;
+iw(
+ray,
+source.position + di * gO.y + qj,
+-di
+);
+ad hit;
+iG(
+ray,
+hit,
+source.instanceIndex,
+max(2.0 * gO.y, EPS)
+);
+if (!hit.hit) continue;
+
+float kE = fZ * abs(dot(di, hit.faceNormal));
+float mt = hV * abs(dot(fE, hit.faceNormal));
+float jS = gy * abs(dot(ek, hit.faceNormal));
+float weight = kE / max(
+kE * kE + mt * mt +
+jS * jS,
+1e-8
+);
+weight *= iN(radius, distance(hit.position, source.position)) /
+max(iN(radius, gO.x), 1e-8);
+iB += weight * hit.vertexNormal;
+}
+
+if (dot(iB, iB) <= 1e-12) return source.normal;
+vec3 jA = normalize(iB);
+vec3 result = source.normal + jA - source.vertexNormal;
+return dot(result, result) > 1e-12 ? normalize(result) : source.normal;
+}
+
+vec4 bw(ad si, Ray ray, J mat, inout RayTraceAuxiliary outAux) {
+Path aa;
+aa.ray = ray;
+aa.li = vec3(0);
+aa.alpha = transmissiveOutputAlphaMin < 1.0 ? 0.0 : 1.0;
+aa.beta = vec3(1.0);
+aa.fromEye = true;
+aa.scatteringPdf = 0.0;
+aa.scatteringIsDelta = false;
+aa.isNextRefraction = false;
+aa.abort = false;
+int depth = 0;
+${intelMac ? bounceCodeOnIntelMac : bounceCodeOnOther}
+return vec4(aa.li, aa.alpha);
+}
+
+vec4 iz(ad si, Ray ray, J mat, inout RayTraceAuxiliary outAux) {
+float coverage = clamp(mat.alpha, 0.0, 1.0);
+if (!si.hit || (mat.materialType != STANDARD && mat.materialType != UNLIT) || coverage >= 1.0) {
+return bw(si, ray, mat, outAux);
+}
+
+
+
+
+
+Ray nK = ray;
+nK.o = si.position + ray.d * (EPS * 0.02);
+ad dU = si;
+dU.albedoSampled = false;
+cY(nK, dU, false);
+J nJ;
+eT(nJ, dU.materialIndex, dU.uv, dU.uv1, dU.instanceIndex);
+RayTraceAuxiliary gg = outAux;
+if (dU.hit && (nJ.materialType == STANDARD || nJ.materialType == UNLIT)) {
+gg.albedo = nJ.transmissionWeight > 0.5
+? vec3(1.0)
+: nJ.baseColor;
+gg.normal = dot(
+dU.normal,
+dU.position - vec3(viewInv[3])
+) > 0.0
+? -dU.normal
+: dU.normal;
+dU.albedoSampled = nJ.transmissionWeight > 0.5;
+}
+vec4 back = bw(dU, nK, nJ, gg);
+if (coverage <= 0.0) {
+outAux = gg;
+return back;
+}
+
+ad te = si;
+J tF = mat;
+tF.alpha = 1.0;
+RayTraceAuxiliary iQ = outAux;
+vec4 qL = bw(te, ray, tF, iQ);
+outAux.albedo = mix(gg.albedo, iQ.albedo, coverage);
+vec3 fR = mix(gg.normal, iQ.normal, coverage);
+outAux.normal = dot(fR, fR) > 1e-12
+? normalize(fR)
+: vec3(0.0);
+outAux.isBlank = gg.isBlank && iQ.isBlank;
+return mix(back, qL, coverage);
+}
+
+
+
+vec4 mj(Ray ray, out RayTraceAuxiliary outAux) {
+
+ad si;
+J mat;
+
+si.albedoSampled = false;
+#ifdef HAS_MEDIUM
+si.mediumType = 0;
+si.mediumExtinction = vec3(0.0);
+si.mediumScattering = vec3(0.0);
+si.mediumAnisotropy = 0.0;
+#endif
+
+cY(ray, si, false);
+eT(mat, si.materialIndex, si.uv, si.uv1, si.instanceIndex);
+si.isTransparency = mat.transmissionWeight > 0.001 || mat.subsurfaceWeight > 0.001 || mat.alpha < 0.999;
+
+#ifdef HAS_BEVEL
+if (bevelEnabled && mat.bevel > 0.0) {
+si.normal = mf(si, mat.bevel);
+}
+#endif
+
+
+
+if (si.hit && (mat.materialType == STANDARD || mat.materialType == UNLIT)) {
+outAux.albedo = mat.transmissionWeight > 0.5 ? vec3(1.0) : mat.baseColor;
+outAux.normal = dot(si.normal, si.position - vec3(viewInv[3])) > 0.0
+? -si.normal
+: si.normal;
+
+
+si.albedoSampled = mat.transmissionWeight > 0.5;
+}
+
+if (mat.transmissionDispersionScale > 0.001) {
+mat.specularIor = eK(
+bf,
+mat.specularIor,
+mat.transmissionDispersionScale,
+mat.dispersionAbbeNumber
+);
+}
+return iz(si, ray, mat, outAux);
+}
+
+void main() {
+
+se();
+jY(v_Texcoord * viewportSize, int(uint(sampleCount) ^ eg(uint(sequenceSeed))));
+
+
+bf = dispersionWavelengths == 1 ? 0.0 : wavelength;
+mh = wavelength;
+if (dispersionWavelengths == 1 && wavelength > 0.0) {
+float channel = floor(dz(CAMERA_WAVELENGTH) * 3.0);
+mh = channel < 1.0 ? 650.0 : channel < 2.0 ? 510.0 : 475.0;
+}
+
+
+
+Ray cam = nv();
+
+RayTraceAuxiliary outAux;
+outAux.albedo = vec3(0.0);
+outAux.normal = vec3(0.0);
+outAux.isBlank = false;
+outAux.isShadowCatcher = false;
+
+vec4 aN = mj(cam, outAux);
+
+
+
+#ifdef CLAMPING
+float clamping = float(CLAMPING);
+aN.rgb = clamp(aN.rgb, -clamping, clamping);
+#endif
+
+
+if (bf > 0.0) {
+
+aN.rgb *= 3.0;
+if (dispersionWavelengths != 1) {
+
+
+aN.a /= 3.0;
+}
+}
+if (!(aN.x < INF && aN.x > -EPS)) {
+aN = vec4(0, 0, 0, 1);
+}
+
+out_color = aN;
+out_albedo = vec4(outAux.albedo, outAux.isBlank ? 0.0 : 1.0);
+out_normal = vec4(outAux.normal, 1.0);
+
+
+float jj = dot(aN.rgb, vec3(0.2126, 0.7152, 0.0722));
+out_moments = outAux.isShadowCatcher
+? vec4(0.0)
+: vec4(jj, jj * jj, 1.0, 0.0);
+
+}`
+    });
+}
+//# sourceMappingURL=rayTrace.frag.js.map

package/lib/renderer/glsl/rayTrace.frag.js.map

@@ -0,0 +1 @@
+{"version":3,"file":"rayTrace.frag.js","sourceRoot":"","sources":["../../../src/renderer/glsl/rayTrace.frag.ts"],"names":[],"mappings":"AAAA,OAAO,EAAE,eAAe,EAAE,MAAM,kBAAkB,CAAC;AACnD,OAAO,aAAa,MAAM,4BAA4B,CAAC;AACvD,qDAAqD;AACrD,sDAAsD;AACtD,kFAAkF;AAElF,OAAO,UAAU,MAAM,yBAAyB,CAAC;AACjD,OAAO,YAAY,MAAM,+BAA+B,CAAC;AACzD,OAAO,aAAa,MAAM,gCAAgC,CAAC;AAC3D,OAAO,kBAAkB,MAAM,iCAAiC,CAAC;AACjE,OAAO,SAAS,MAAM,4BAA4B,CAAC;AACnD,OAAO,QAAQ,MAAM,uBAAuB,CAAC;AAC7C,OAAO,MAAM,MAAM,yBAAyB,CAAC;AAC7C,OAAO,MAAM,MAAM,yBAAyB,CAAC;AAC7C,OAAO,kBAAkB,MAAM,iCAAiC,CAAC;AACjE,OAAO,eAAe,MAAM,8BAA8B,CAAC;AAC3D,OAAO,EAAE,cAAc,EAAE,IAAI,EAAE,MAAM,EAAE,MAAM,cAAc,CAAC;AAC5D,OAAO,eAAe,MAAM,8BAA8B,CAAC;AAC3D,OAAO,YAAY,MAAM,2BAA2B,CAAC;AACrD,OAAO,EAAE,UAAU,EAAE,MAAM,YAAY,CAAC;AACxC,OAAO,gBAAgB,MAAM,+BAA+B,CAAC;AAE7D,MAAM,2BAA2B,GAAG,IAAI,CAAA;;;;;;;CAOvC,CAAC;AAEF,0BAA0B;AAC1B,iEAAiE;AACjE,kFAAkF;AAClF,4CAA4C;AAC5C,MAAM,oBAAoB,GAAG,IAAI,CAAA;;;;;;;;CAQhC,CAAC;AACF,MAAM,iBAAiB,GAAG,IAAI,CAAA;;;0CAGY,eAAe;;;;;;;;;CASxD,CAAC;AAEF,MAAM,CAAC,OAAO,UAAU,4BAA4B;IAClD,MAAM,QAAQ,GAAG,UAAU,EAAE,CAAC;IAC9B,OAAO,IAAI,cAAc,CAAC;QAC1B,IAAI,EAAE,cAAc;QACpB,OAAO,EAAE,CAAC,OAAO,EAAE,QAAQ,EAAE,QAAQ,EAAE,SAAS,CAAC;QACjD,OAAO,EAAE;YACP,OAAO,EAAE,CAAC;SACX;QACD,QAAQ,EAAE;YACR,oBAAoB,EAAE,MAAM,CAAC,OAAO,CAAC,OAAO,EAAE,CAAC,CAAC;YAChD,mBAAmB,EAAE,MAAM,CAAC,OAAO,CAAC,OAAO,EAAE,EAAE,CAAC;YAEhD,iBAAiB,EAAE,MAAM,CAAC,OAAO,CAAC,MAAM,CAAC;YACzC,OAAO,EAAE,MAAM,CAAC,OAAO,CAAC,MAAM,CAAC;YAC/B,0BAA0B,EAAE,MAAM,CAAC,OAAO,CAAC,OAAO,CAAC;YACnD,gBAAgB,EAAE,MAAM,CAAC,OAAO,CAAC,WAAW,CAAC;YAE7C,YAAY,EAAE,MAAM,CAAC,OAAO,CAAC,MAAM,CAAC;YAEpC,UAAU,EAAE,MAAM,CAAC,OAAO,CAAC,OAAO,EAAE,CAAC,CAAC;YACtC,YAAY,EAAE,MAAM,CAAC,OAAO,CAAC,MAAM,EAAE,IAAI,CAAC;YAC1C,qBAAqB,EAAE,MAAM,CAAC,OAAO,CAAC,KAAK,EAAE,CAAC,CAAC;YAC/C,WAAW,EAAE,MAAM,CAAC,OAAO,CAAC,KAAK,EAAE,CAAC,CAAC;SACtC;QACD,QAAQ,EAAE,CAAC,YAAY,EAAE,kBAAkB,EAAE,SAAS,EAAE,MAAM,EAAE,MAAM,CAAC;QACvE,IAAI,EAAE,IAAI,CAAA;EACV,UAAU;;EAEV,QAAQ;EACR,MAAM,CAAC,IAAI;;EAEX,aAAa;;EAEb,YAAY,CAAC,IAAI;;EAEjB,gBAAgB;;EAEhB,kBAAkB,CAAC,IAAI;EACvB,SAAS,CAAC,IAAI;EACd,MAAM,CAAC,IAAI;;EAEX,YAAY;EACZ,eAAe;;;;;EAKf,eAAe;EACf,kBAAkB;;EAElB,aAAa;;;;;;;;;;;;;;;;EAgBb,QAAQ,CAAC,CAAC,CAAC,2BAA2B,CAAC,CAAC,CAAC,EAAE;;;;;;;;;;;;;;;;;;;;;;;;;;;;;;;;;;;;;;;;;;;;;;;;;;;;;;;;;;;;;;;;;;;;;;;;;;;;;;;;;;;;;;;;;;;;;;;;;;;;;;;;;;;;;;;;;;;;;;;;;;;;;;;;;;;;;;;;;;;;;;;;;;;;;;;;;;;;;;;;;;;;;;;;;;;;;;;;;;;;;;;;;;;;;;;;;;;;;;;;;;;;;;;;;;;;;;;;;;;;;;;;;;;;;;;;;;;;;;;;;;;;;;;;;;;;;;;;;;;;;;;;;;;;;;;;;;;;;;;;;;;;;;;;;;;;;;;;;;;;;;;;;;;;;;;;;;;;;;;;;;;;;;;;;;;;;;;;;;;;EAoV3C,QAAQ,CAAC,CAAC,CAAC,oBAAoB,CAAC,CAAC,CAAC,iBAAiB;;;;;;;;;;;;;;;;;;;;;;;;;;;;;;;;;;;;;;;;;;;;;;;;;;;;;;;;;;;;;;;;;;;;;;;;;;;;;;;;;;;;;;;;;;;;;;;;;;;;;;;;;;;;;;;;;;;;;;;;;;;;;;;;;;;;;;;;;;;;;;;;;;;;;;;;;;;;;;EA8JnD;KACC,CAAC,CAAC;AACL,CAAC","sourcesContent":["import { MAX_PATH_EVENTS } from '../pathLimits.js';\nimport cameraRayGlsl from './chunks/cameraRay.glsl.js';\n// Portions adapted from ray-tracing-renderer@0.10.15\n// (https://github.com/hoverinc/ray-tracing-renderer).\n// Copyright (c) 2019 HOVER, MIT license; see LICENSES/RayTracingRenderer-MIT.txt.\n\nimport commonGlsl from './chunks/common.glsl.js';\nimport rayTraceCore from './chunks/rayTraceCore.glsl.js';\nimport textureLinear from './chunks/textureLinear.glsl.js';\nimport materialBufferGlsl from './chunks/materialBuffer.glsl.js';\nimport intersect from './chunks/intersect.glsl.js';\nimport randGlsl from './chunks/rand.glsl.js';\nimport random from './chunks/random.glsl.js';\nimport envMap from './chunks/envMap.glsl.js';\nimport sampleMaterialGlsl from './chunks/sampleMaterial.glsl.js';\nimport sampleLightGlsl from './chunks/sampleLight.glsl.js';\nimport { FragmentShader, glsl, Shader } from '../shader.js';\nimport openPbrBsdfGlsl from './chunks/openPbrBsdf.glsl.js';\nimport samplingGlsl from './chunks/sampling.glsl.js';\nimport { isIntelMac } from '../util.js';\nimport uvProjectionGlsl from './chunks/uvProjection.glsl.js';\n\nconst doubleTransmissionCheckCode = glsl`\n#if defined(HAS_TRANSMISSION) || defined(HAS_ALPHA)\n  if (!si.isTransparency && BOUNCES < 16 && depth * 2 > BOUNCES) {\n    path.abort = true;\n    return;\n  }\n#endif\n`;\n\n// bounce fix on intel mac\n// The intel mac only have one bounce if the loop is not unrolled\n// But windows will have extremely slow shader compilation if the loop is unrolled\n// So here the code are injected separately.\nconst bounceCodeOnIntelMac = glsl`\n// Mark loop can be unrolled\nfor(int _idx_ = 0; _idx_ < BOUNCES; _idx_++) {{\n  if (path.abort) {\n    return vec4(path.li, path.alpha);\n  }\n  bounce(path, depth, si, mat, outAux);\n}}\n`;\nconst bounceCodeOnOther = glsl`\nbool isOpaque = !si.isTransparency;\nint activeBounceLimit = min(BOUNCES, bounceLimit);\nfor(int idx = 0; idx < min(BOUNCES * 3, ${MAX_PATH_EVENTS}); idx++) {\n  if (isOpaque && idx > activeBounceLimit) {\n    path.abort = true;\n  }\n  if (path.abort) {\n    return vec4(path.li, path.alpha);\n  }\n  bounce(path, depth, si, mat, outAux);\n}\n`;\n\nexport default function createRayTraceFragmentShader() {\n  const intelMac = isIntelMac();\n  return new FragmentShader({\n  name: 'rayTraceFrag',\n  outputs: ['color', 'albedo', 'normal', 'moments'],\n  defines: {\n    BOUNCES: 2\n  },\n  uniforms: {\n    cameraApertureRadius: Shader.uniform('float', 0),\n    cameraFocusDistance: Shader.uniform('float', 10),\n\n    viewProjectionInv: Shader.uniform('mat4'),\n    viewInv: Shader.uniform('mat4'),\n    transmissiveOutputAlphaMin: Shader.uniform('float'),\n    openPbrEnergyLut: Shader.uniform('sampler2D'),\n\n    viewportSize: Shader.uniform('vec2'),\n\n    wavelength: Shader.uniform('float', 0),\n    bevelEnabled: Shader.uniform('bool', true),\n    dispersionWavelengths: Shader.uniform('int', 3),\n    bounceLimit: Shader.uniform('int', 8)\n  },\n  includes: [rayTraceCore, materialBufferGlsl, intersect, random, envMap],\n  main: glsl`\n${commonGlsl}\n\n${randGlsl}\n${random.main}\n\n${textureLinear}\n\n${rayTraceCore.main}\n\n${uvProjectionGlsl}\n\n${materialBufferGlsl.main}\n${intersect.main}\n${envMap.main}\n\n${samplingGlsl}\n${openPbrBsdfGlsl}\n\nfloat heroWavelength;\nfloat pathWavelength;\n\n${sampleLightGlsl}\n${sampleMaterialGlsl}\n\n${cameraRayGlsl}\nstruct RayTraceAuxiliary {\n  // Denoising guides and auxiliary pixel classification.\n  vec3 albedo;\n  vec3 normal;\n  bool isBlank;\n  // The primary ray hit a shadow catcher, whose sample the catcher pass owns.\n  bool isShadowCatcher;\n};\n\nvoid bounce(inout Path path, inout int depth, inout SurfaceInteraction si, inout OpenPbrResolvedMaterial mat, inout RayTraceAuxiliary outAux) {\n  samplingBounce = uint(max(depth, 0));\n  // The bounce will be doubled on the transmission material.\n  // For other materials. We will quite early\n  // TODO It is only used on Mac with AMD card. If I enabled it on windows.\n  // It will cause shader compilation too slow.\n${intelMac ? doubleTransmissionCheckCode : ''}\n\n  bool isShadowCatcher = mat.materialType == SHADOW_CATCHER;\n  if (isShadowCatcher) {\n    if (path.fromEye) {\n      si.hit = false;\n      outAux.isShadowCatcher = true;\n    }\n    else {\n      // The catcher belongs only to the shadow-mask pass. Keep its default\n      // white material out of glossy and transmission reflections.\n#ifdef USE_ENVMAP\n      if (path.isNextRefraction) {\n        // The invisible catcher can sit between a transmitted camera ray and\n        // the grounded skybox. Use the displayed background here instead of\n        // terminating early with the raw lighting environment.\n        vec4 bg = sampleBackgroundFromRay(path.ray.o, path.ray.d);\n        bg.rgb = sampling_spectral_rgb_filter(bg.rgb, pathWavelength);\n        float backgroundMisWeight = path.scatteringIsDelta ? 1.0\n          : sampling_power_heuristic(path.scatteringPdf,\n              envMapPdf(cartesianToEquirect(path.ray.d)) * envLightSelectionPmf);\n        path.li += backgroundMisWeight * path.beta * bg.rgb * bg.a;\n        path.alpha = max(path.alpha, bg.a);\n      }\n      else {\n        vec2 envUv = cartesianToEquirect(path.ray.d);\n        vec3 emission = textureLinear(envMap, envUv).rgb;\n        emission = sampling_spectral_rgb_filter(emission, pathWavelength);\n        path.li += path.beta * emission *\n          sampling_power_heuristic(\n            path.scatteringPdf,\n            envMapPdf(envUv) * envLightSelectionPmf\n          );\n      }\n#else\n      vec4 bg = sampleBackgroundFromRay(path.ray.o, path.ray.d);\n      bg.rgb = sampling_spectral_rgb_filter(bg.rgb, pathWavelength);\n      path.li += path.beta * bg.rgb * bg.a;\n#endif\n      path.abort = true;\n      return;\n    }\n  }\n\n  if (!si.hit) {\n    // Sample from background if first ray not hit anything.\n    vec4 bg = sampleBackgroundFromRay(path.ray.o, path.ray.d);\n    // Dispersion renders one additive pass per RGB wavelength. Background\n    // radiance has to be split in the same way as lights and emissive\n    // surfaces, otherwise the full skybox is accumulated three times.\n    bg.rgb = sampling_spectral_rgb_filter(bg.rgb, pathWavelength);\n    if (path.fromEye) {\n      path.li = bg.rgb * bg.a;\n      path.alpha = bg.a;\n      // Only treat as blank pixels when the background is transparent\n      outAux.isBlank = bg.a < 0.01;\n    }\n    else {\n      Ray cam = initCameraRay(v_Texcoord + pixelJitter());\n      bool towardsCam = dot(cam.d, path.ray.d) > 0.0;\n      if (transmissiveOutputAlphaMin < 1.0) {\n        if (towardsCam) {\n          path.alpha = max(path.alpha, bg.a);\n        } else {\n          path.alpha = 1.0;\n        }\n      }\n#if !defined(USE_ENVMAP)\n      // TODO treat background as a light source\n      if (towardsCam) {\n        path.li = path.beta * sampling_power_heuristic(path.scatteringPdf, 1.0) * bg.rgb;\n      }\n#endif\n    }\n    path.abort = true;\n    return;\n  }\n\n  bool sampleIgnored = false;\n  if ((mat.materialType == STANDARD || mat.materialType == UNLIT)) {\n    if (transmissiveOutputAlphaMin < 1.0) {\n      // TODO increase the alpha when there are more bounces.\n      float alpha = max(1.0 - mat.transmissionWeight, 1.0 - pow(1.0 - transmissiveOutputAlphaMin, float(depth)));\n      path.alpha = max(alpha, path.alpha);\n    }\n    float alpha = mat.alpha;\n    if (alpha < 1.0) {\n      // Alpha is sampled as stochastic coverage. An accepted surface branch\n      // must use full radiance weight; multiplying by alpha again is biased.\n      path.alpha = max(path.alpha, alpha);\n      if (hashedSample(3u) > alpha) {\n        sampleIgnored = true;\n      }\n    }\n    if (!sampleIgnored) {\n      SampleMaterial(si, mat, path);\n    }\n    else {\n      path.ray.o = si.position + path.ray.d * EPS;\n    }\n  }\n\n  if (!si.albedoSampled && !sampleIgnored) {\n    // Select denoising guides.\n    if ((mat.materialType == STANDARD || mat.materialType == UNLIT) &&\n      !(mat.transmissionWeight > 0.5 ||\n        (mat.specularRoughness <= 0.02 && mat.baseMetalness > 0.98))) {\n      // Alpha-blended decals can be skipped stochastically. In that case keep\n      // looking for the first useful rough-surface guide behind the decal,\n      // even though it is reached after depth zero.\n      outAux.albedo = mat.baseColor;\n      outAux.normal = dot(si.normal, si.position - vec3(viewInv[3])) > 0.0\n        ? -si.normal\n        : si.normal;\n      si.albedoSampled = true;\n    }\n    if (depth > 4) {\n      si.albedoSampled = true;\n    }\n  }\n\n\n  // Russian Roulette sampling\n#ifdef USE_RR\n  if (depth >= RR_MAX && !sampleIgnored) {\n    // Max-component survival (pbrt-v4) avoids over-amplifying saturated paths.\n    float maxComponent = max(max(path.beta.x, path.beta.y), path.beta.z);\n    float q = max(0.0, 1.0 - maxComponent);\n    if (sample1D(bounceDim(RUSSIAN_ROULETTE)) < q) {\n      // TODO we force use alpha 1.0 here.\n      // To avoid the glass pixels is transparent even if it's on an opaque object.\n      path.alpha = 1.0;\n      path.abort = true;\n    }\n    else {\n      path.beta /= 1.0 - q;\n    }\n  }\n#endif\n\n  if (path.abort) {\n    return;\n  }\n\n  // Next ray hit\n  intersectScene(path.ray, si, true);\n  getMaterial(mat, si.materialIndex, si.uv, si.uv1, si.instanceIndex);\n  si.isTransparency = mat.transmissionWeight > 0.001 || mat.subsurfaceWeight > 0.001 || mat.alpha < 0.999;\n\n  // Dynamic preview only becomes spectral when the primary visible surface\n  // actually disperses transmitted light. Direct background and ordinary\n  // opaque materials stay full RGB and therefore remain noise-free.\n  if (dispersionWavelengths == 1 && si.hit && mat.transmissionWeight > 0.001 && mat.transmissionDispersionScale > 0.001) {\n    pathWavelength = heroWavelength;\n  }\n  if (mat.transmissionWeight > 0.001 && pathWavelength > 0.0) {\n    mat.specularIor = sampling_cauchy_ior(\n      pathWavelength,\n      mat.specularIor,\n      mat.transmissionDispersionScale,\n      mat.dispersionAbbeNumber\n    );\n  }\n\n  if (sampleIgnored) {\n    return;\n  }\n\n  bool isAlphaOccluded = hashedSample(3u) < mat.alpha;\n#ifdef NUM_LIGHTS\n  // Use light if hit the light\n  LightSampleRec lightSample;\n  for (int i = 0; i < NUM_LIGHTS; i++) {\n    // Fetch light Data\n    Light light = fetchLight(i);\n    float d = intersectLight(path.ray, light, lightSample);\n    bool occluded = si.hit && d > distance(si.position, path.ray.o);\n    if (d != INF && !(occluded && isAlphaOccluded)) {\n      lightSample.pdf *= light.selectionPmf;\n      float misWeight = sampling_power_heuristic(path.scatteringPdf, lightSample.pdf);\n      lightSample.emission = sampling_spectral_rgb_filter(lightSample.emission, pathWavelength);\n      path.li += misWeight * path.beta * lightSample.emission;\n      // TODO set alpha to 1.0 will cause some sharp edges\n      path.alpha = 1.0;\n      path.abort = true;\n    }\n  }\n#endif\n\n#ifdef USE_ENVMAP\n  if (!(si.hit && isAlphaOccluded)) {\n    if (path.isNextRefraction) {\n      // Camera-visible transmission must use the displayed background. The\n      // raw environment remains the lighting source for non-transmission rays.\n      vec4 bg = sampleBackgroundFromRay(path.ray.o, path.ray.d);\n      bg.rgb = sampling_spectral_rgb_filter(bg.rgb, pathWavelength);\n      float backgroundMisWeight = path.scatteringIsDelta ? 1.0\n          : sampling_power_heuristic(path.scatteringPdf,\n              envMapPdf(cartesianToEquirect(path.ray.d)) * envLightSelectionPmf);\n        path.li += backgroundMisWeight * path.beta * bg.rgb * bg.a;\n      path.alpha = max(path.alpha, bg.a);\n      path.abort = true;\n      return;\n    }\n    vec2 uv = cartesianToEquirect(path.ray.d);\n    vec3 emission = textureLinear(envMap, uv).rgb;\n    float misWeight = sampling_power_heuristic(\n      path.scatteringPdf,\n      envMapPdf(uv) * envLightSelectionPmf\n    );\n    emission = sampling_spectral_rgb_filter(emission, pathWavelength);\n    path.li += path.beta * emission * misWeight;\n\n  // TODO force the min alpha to be 1 when showing envmap as background\n  #if !defined(USE_BACKGROUND)\n    path.alpha = 1.0;\n  #endif\n    path.abort = true;\n    return;\n  }\n#endif\n\n  depth++;\n}\n\n\nvec2 bevelCubicSample(float radius, float xi) {\n  float x = 0.25;\n  for (int iteration = 0; iteration < 10; iteration++) {\n    float x2 = x * x;\n    float x3 = x2 * x;\n    float oneMinusX = 1.0 - x;\n    float value = 10.0 * x2 - 20.0 * x3 + 15.0 * x2 * x2 -\n      4.0 * x2 * x3 - xi;\n    float derivative = 20.0 * x * oneMinusX * oneMinusX * oneMinusX;\n    if (abs(value) < 1e-6 || derivative == 0.0) break;\n    x = clamp(x - value / derivative, 0.0, 1.0);\n  }\n  float diskRadius = x * radius;\n  return vec2(\n    diskRadius,\n    sqrt(max(radius * radius - diskRadius * diskRadius, 0.0))\n  );\n}\n\nfloat bevelCubicPdf(float radius, float distanceToSource) {\n  if (distanceToSource >= radius) return 0.0;\n  float falloff = radius - distanceToSource;\n  return falloff * falloff * falloff;\n}\n\n// Shader bevel. Usage: docs/materials.md (Renderer extensions)\nvec3 getBevelNormal(SurfaceInteraction source, float radius) {\n  vec3 geometricNormal = source.faceNormal;\n  if (dot(geometricNormal, source.vertexNormal) < 0.0) geometricNormal = -geometricNormal;\n  vec3 tangent;\n  vec3 bitangent;\n  sampling_orthonormal_basis(geometricNormal, tangent, bitangent);\n  vec3 normalSum = vec3(0.0);\n\n  for (int sampleIndex = 0; sampleIndex < BEVEL_SAMPLES; sampleIndex++) {\n    vec2 bevelRnd = sample2D(CAMERA_BEVEL_BASE + uint(sampleIndex));\n    float axisSample = bevelRnd.x;\n    float radialSample = bevelRnd.y;\n    vec3 diskNormal = geometricNormal;\n    vec3 diskTangent = tangent;\n    vec3 diskBitangent = bitangent;\n    float pickPdfNormal = 0.5;\n    float pickPdfTangent = 0.25;\n    float pickPdfBitangent = 0.25;\n    if (axisSample < 0.5) {\n      axisSample *= 2.0;\n    }\n    else if (axisSample < 0.75) {\n      vec3 swapAxis = diskNormal;\n      diskNormal = diskTangent;\n      diskTangent = swapAxis;\n      pickPdfNormal = 0.25;\n      pickPdfTangent = 0.5;\n      axisSample = (axisSample - 0.5) * 4.0;\n    }\n    else {\n      vec3 swapAxis = diskNormal;\n      diskNormal = diskBitangent;\n      diskBitangent = swapAxis;\n      pickPdfNormal = 0.25;\n      pickPdfBitangent = 0.5;\n      axisSample = (axisSample - 0.75) * 4.0;\n    }\n\n    vec2 diskSample = bevelCubicSample(radius, radialSample);\n    float phi = TWOPI * axisSample;\n    vec3 diskOffset = diskSample.x *\n      (cos(phi) * diskTangent + sin(phi) * diskBitangent);\n    Ray ray;\n    initRay(\n      ray,\n      source.position + diskNormal * diskSample.y + diskOffset,\n      -diskNormal\n    );\n    SurfaceInteraction hit;\n    intersectSceneBevel(\n      ray,\n      hit,\n      source.instanceIndex,\n      max(2.0 * diskSample.y, EPS)\n    );\n    if (!hit.hit) continue;\n\n    float pdfNormal = pickPdfNormal * abs(dot(diskNormal, hit.faceNormal));\n    float pdfTangent = pickPdfTangent * abs(dot(diskTangent, hit.faceNormal));\n    float pdfBitangent = pickPdfBitangent * abs(dot(diskBitangent, hit.faceNormal));\n    float weight = pdfNormal / max(\n      pdfNormal * pdfNormal + pdfTangent * pdfTangent +\n        pdfBitangent * pdfBitangent,\n      1e-8\n    );\n    weight *= bevelCubicPdf(radius, distance(hit.position, source.position)) /\n      max(bevelCubicPdf(radius, diskSample.x), 1e-8);\n    normalSum += weight * hit.vertexNormal;\n  }\n\n  if (dot(normalSum, normalSum) <= 1e-12) return source.normal;\n  vec3 bevelVertexNormal = normalize(normalSum);\n  vec3 result = source.normal + bevelVertexNormal - source.vertexNormal;\n  return dot(result, result) > 1e-12 ? normalize(result) : source.normal;\n}\n\nvec4 integrateOnePathSample(SurfaceInteraction si, Ray ray, OpenPbrResolvedMaterial mat, inout RayTraceAuxiliary outAux) {\n  Path path;\n  path.ray = ray;\n  path.li = vec3(0);\n  path.alpha = transmissiveOutputAlphaMin < 1.0 ? 0.0 : 1.0;\n  path.beta = vec3(1.0);\n  path.fromEye = true;\n  path.scatteringPdf = 0.0;\n  path.scatteringIsDelta = false;\n  path.isNextRefraction = false;\n  path.abort = false;\n  int depth = 0;\n${intelMac ? bounceCodeOnIntelMac : bounceCodeOnOther}\n  return vec4(path.li, path.alpha);\n}\n\nvec4 integrateOnePath(SurfaceInteraction si, Ray ray, OpenPbrResolvedMaterial mat, inout RayTraceAuxiliary outAux) {\n  float coverage = clamp(mat.alpha, 0.0, 1.0);\n  if (!si.hit || (mat.materialType != STANDARD && mat.materialType != UNLIT) || coverage >= 1.0) {\n    return integrateOnePathSample(si, ray, mat, outAux);\n  }\n\n  // glTF BLEND is source-over opacity, not a physical glass boundary. Split\n  // the primary ray so reflected radiance is attenuated by alpha every sample\n  // while the complementary branch continues straight to the scene behind it.\n  // Secondary alpha hits remain stochastic to avoid exponential path growth.\n  Ray backRay = ray;\n  backRay.o = si.position + ray.d * (EPS * 0.02);\n  SurfaceInteraction backSi = si;\n  backSi.albedoSampled = false;\n  intersectScene(backRay, backSi, false);\n  OpenPbrResolvedMaterial backMat;\n  getMaterial(backMat, backSi.materialIndex, backSi.uv, backSi.uv1, backSi.instanceIndex);\n  RayTraceAuxiliary backAux = outAux;\n  if (backSi.hit && (backMat.materialType == STANDARD || backMat.materialType == UNLIT)) {\n    backAux.albedo = backMat.transmissionWeight > 0.5\n      ? vec3(1.0)\n      : backMat.baseColor;\n    backAux.normal = dot(\n      backSi.normal,\n      backSi.position - vec3(viewInv[3])\n    ) > 0.0\n      ? -backSi.normal\n      : backSi.normal;\n    backSi.albedoSampled = backMat.transmissionWeight > 0.5;\n  }\n  vec4 back = integrateOnePathSample(backSi, backRay, backMat, backAux);\n  if (coverage <= 0.0) {\n    outAux = backAux;\n    return back;\n  }\n\n  SurfaceInteraction frontSi = si;\n  OpenPbrResolvedMaterial frontMat = mat;\n  frontMat.alpha = 1.0;\n  RayTraceAuxiliary frontAux = outAux;\n  vec4 front = integrateOnePathSample(frontSi, ray, frontMat, frontAux);\n  outAux.albedo = mix(backAux.albedo, frontAux.albedo, coverage);\n  vec3 blendedNormal = mix(backAux.normal, frontAux.normal, coverage);\n  outAux.normal = dot(blendedNormal, blendedNormal) > 1e-12\n    ? normalize(blendedNormal)\n    : vec3(0.0);\n  outAux.isBlank = backAux.isBlank && frontAux.isBlank;\n  return mix(back, front, coverage);\n}\n\n// Path tracing integrator as described in\n// http://www.pbr-book.org/3ed-2018/Light_Transport_I_Surface_Reflection/Path_Tracing.html#\nvec4 integrator(Ray ray, out RayTraceAuxiliary outAux) {\n\n  SurfaceInteraction si;\n  OpenPbrResolvedMaterial mat;\n\n  si.albedoSampled = false;\n#ifdef HAS_MEDIUM\n  si.mediumType = 0;\n  si.mediumExtinction = vec3(0.0);\n  si.mediumScattering = vec3(0.0);\n  si.mediumAnisotropy = 0.0;\n#endif\n\n  intersectScene(ray, si, false);\n  getMaterial(mat, si.materialIndex, si.uv, si.uv1, si.instanceIndex);\n  si.isTransparency = mat.transmissionWeight > 0.001 || mat.subsurfaceWeight > 0.001 || mat.alpha < 0.999;\n\n#ifdef HAS_BEVEL\n  if (bevelEnabled && mat.bevel > 0.0) {\n    si.normal = getBevelNormal(si, mat.bevel);\n  }\n#endif\n\n  // Keep a clean first-hit fallback for denoising. Glass retains this stable\n  // guide; other delta paths may replace it at their first non-delta hit.\n  if (si.hit && (mat.materialType == STANDARD || mat.materialType == UNLIT)) {\n    outAux.albedo = mat.transmissionWeight > 0.5 ? vec3(1.0) : mat.baseColor;\n    outAux.normal = dot(si.normal, si.position - vec3(viewInv[3])) > 0.0\n      ? -si.normal\n      : si.normal;\n    // A refracted/reflected continuation is stochastic and therefore not a\n    // clean OIDN feature. Keep the stable primary glass guide instead.\n    si.albedoSampled = mat.transmissionWeight > 0.5;\n  }\n\n  if (mat.transmissionDispersionScale > 0.001) {\n    mat.specularIor = sampling_cauchy_ior(\n      pathWavelength,\n      mat.specularIor,\n      mat.transmissionDispersionScale,\n      mat.dispersionAbbeNumber\n    );\n  }\n  return integrateOnePath(si, ray, mat, outAux);\n}\n\nvoid main() {\n\n  initRandom();\n  rng_initialize(v_Texcoord * viewportSize, int(uint(sampleCount) ^ sampleHash(uint(sequenceSeed))));\n\n\n  pathWavelength = dispersionWavelengths == 1 ? 0.0 : wavelength;\n  heroWavelength = wavelength;\n  if (dispersionWavelengths == 1 && wavelength > 0.0) {\n    float channel = floor(sample1D(CAMERA_WAVELENGTH) * 3.0);\n    heroWavelength = channel < 1.0 ? 650.0 : channel < 2.0 ? 510.0 : 475.0;\n  }\n\n  // Thin lens model with depth-of-field\n  // http://www.pbr-book.org/3ed-2018/Camera_Models/Projective_Camera_Models.html#TheThinLensModelandDepthofField\n  Ray cam = primaryCameraRay();\n\n  RayTraceAuxiliary outAux;\n  outAux.albedo = vec3(0.0);\n  outAux.normal = vec3(0.0);\n  outAux.isBlank = false;\n  outAux.isShadowCatcher = false;\n\n  vec4 liAndAlpha = integrator(cam, outAux);\n\n  // Clamp at Integration Level to reduce fireflies.\n  // TODO Path level configuration?\n#ifdef CLAMPING\n  float clamping = float(CLAMPING);\n  liAndAlpha.rgb = clamp(liAndAlpha.rgb, -clamping, clamping);\n#endif\n\n  // Each sample will be accumulated three times.\n  if (pathWavelength > 0.0) {\n    // RGB will be normalized using accumated hit in out_albedo.a\n    liAndAlpha.rgb *= 3.0;\n    if (dispersionWavelengths != 1) {\n      // Full spectral samples use three additive passes. Preview only submits\n      // one hero wavelength, so its alpha must not be divided.\n      liAndAlpha.a /= 3.0;\n    }\n  }\n  if (!(liAndAlpha.x < INF && liAndAlpha.x > -EPS)) {\n    liAndAlpha = vec4(0, 0, 0, 1);\n  }\n\n  out_color = liAndAlpha;\n  out_albedo = vec4(outAux.albedo, outAux.isBlank ? 0.0 : 1.0);\n  out_normal = vec4(outAux.normal, 1.0);\n  // Moments and count of the built-in denoiser's guide value. The catcher\n  // pass accumulates its own samples (see shadowCatcher.frag).\n  float guideValue = dot(liAndAlpha.rgb, vec3(0.2126, 0.7152, 0.0722));\n  out_moments = outAux.isShadowCatcher\n    ? vec4(0.0)\n    : vec4(guideValue, guideValue * guideValue, 1.0, 0.0);\n\n}`\n  });\n}\n"]}

package/lib/renderer/glsl/shadowCatcher.frag.d.ts

@@ -0,0 +1,3 @@
+import { FragmentShader } from '../shader.js';
+declare const _default: FragmentShader<Record<string, import("../shader.js").ShaderDefineValue>, Record<string, import("../shader.js").ShaderUniform<unknown>>>;
+export default _default;