@meimu/mutracer 0.2.0 → 0.3.1

package/lib/renderer/wgsl/easuDetail.wgsl.js

@@ -0,0 +1,230 @@
+import { easuShader } from './easu.wgsl.js';
+import { composeDefine } from './compose.frag.wgsl.js';
+/** Scene-linear log2 range covered by the tone lookup table. */
+export const TONE_LUT_LOG2 = [-24, 24];
+export const TONE_LUT_SIZE = 1024;
+/** Detail compensation defaults and fixed regularizers shared by the Web and native passes. */
+export const EASU_DETAIL = {
+    ratioRange: [0.7, 1.43], depthThreshold: 0.02, coverageThreshold: 0.95,
+    // Albedo below this is treated as black; relative depth tolerance floor.
+    minimumAlbedo: 1e-3, depthTolerance: 1e-4,
+    // Regularizes the local albedo variance of the albedo-radiance fit, in squared albedo units.
+    guidedEpsilon: 1e-4,
+};
+// Tone table lookups and display encoding shared by the detail passes; tone.x is the maximum display value.
+const toneCodingDefine = /* wgsl */ `
+
+
+@group(0) @binding(8) var toneLut: texture_2d<f32>;
+@group(0) @binding(9) var<uniform> tone: vec4f;
+fn vZ(row: i32, u: vec3f) -> vec3f {
+let last=f32(textureDimensions(toneLut).x-1u);let p=clamp(u,vec3f(0),vec3f(1))*last;
+let i=vec3i(floor(p));let j=min(i+1,vec3i(i32(last)));let f=p-floor(p);
+var result=vec3f(0);
+for(var c=0;c<3;c++){result[c]=mix(textureLoad(toneLut,vec2i(i[c],row),0)[c],textureLoad(toneLut,vec2i(j[c],row),0)[c],f[c]);}
+return result;
+}
+fn uM(AK: vec3f) -> vec3f {return vZ(0,(AK-(${TONE_LUT_LOG2[0]}.0))/${TONE_LUT_LOG2[1] - TONE_LUT_LOG2[0]}.0);}
+fn uN(d: vec3f) -> vec3f {return vZ(1,sqrt(max(d,vec3f(0))/tone.x));}
+
+
+
+
+fn zk(d: vec3f, fw: vec3f) -> vec3f {
+let x=uN(d);
+let changed=max(d+uM(x+log2(max(fw,vec3f(1e-6))))-uM(x),vec3f(0));
+return select(changed,d,abs(fw-vec3f(1))<vec3f(1e-4));
+}
+fn pS(c: vec3f) -> vec3f {
+let a=max(c,vec3f(0));
+return select(pow((a+0.055)/1.055,vec3f(2.4)),a/12.92,a<=vec3f(0.04045));
+}
+fn tp(c: vec3f) -> vec3f {
+let a=max(c,vec3f(0));
+return select(1.055*pow(a,vec3f(1.0/2.4))-0.055,a*12.92,a<=vec3f(0.0031308));
+}
+`;
+/**
+ * guided: scale only the albedo-correlated part of local scene-linear radiance (default);
+ * separate: lobe-share approximation; silhouettes: depth-guided edge reconstruction.
+ * The last two are retained for comparisons.
+ */
+export function easuDetailShader(separate = false, silhouettes = false, guided = true) {
+    return easuShader.replace(`@vertex fn vertex`, /* wgsl */ `
+@group(0) @binding(2) var diffuseHi: texture_2d<f32>;
+@group(0) @binding(3) var specularHi: texture_2d<f32>;
+@group(0) @binding(4) var depthHi: texture_2d<f32>;
+@group(0) @binding(5) var diffuseLo: texture_2d<f32>;
+@group(0) @binding(6) var specularLo: texture_2d<f32>;
+@group(0) @binding(7) var<uniform> detail: vec4f;
+${silhouettes ? '@group(0) @binding(10) var depthLo: texture_2d<f32>;' : ''}
+${guided ? '@group(0) @binding(11) var slopeLo: texture_2d<f32>;\n@group(0) @binding(12) var offsetLo: texture_2d<f32>;' : ''}
+fn cY(image: texture_2d<f32>, p: vec2i) -> vec4f {
+return textureLoad(image,clamp(p,vec2i(0),vec2i(textureDimensions(image))-1),0);
+}
+${toneCodingDefine}
+@vertex fn vertex`).replace(`return clamp(result/total,lower,upper);`, /* wgsl */ `
+let color=clamp(result/total,lower,upper);
+var diffuse=vec4f(0);var specular=vec4f(0);
+for(var i=0u;i<12u;i++){
+
+let weight=sJ(vec2f(offsets[i])-phase,direction,length,lobe);
+diffuse+=cY(diffuseLo,base+offsets[i])*weight;
+specular+=cY(specularLo,base+offsets[i])*weight;
+}
+let dlo=min(min(cY(diffuseLo,base),cY(diffuseLo,base+vec2i(1,0))),min(cY(diffuseLo,base+vec2i(0,1)),cY(diffuseLo,base+vec2i(1,1))));
+let dhi=max(max(cY(diffuseLo,base),cY(diffuseLo,base+vec2i(1,0))),max(cY(diffuseLo,base+vec2i(0,1)),cY(diffuseLo,base+vec2i(1,1))));
+let slo=min(min(cY(specularLo,base),cY(specularLo,base+vec2i(1,0))),min(cY(specularLo,base+vec2i(0,1)),cY(specularLo,base+vec2i(1,1))));
+let shi=max(max(cY(specularLo,base),cY(specularLo,base+vec2i(1,0))),max(cY(specularLo,base+vec2i(0,1)),cY(specularLo,base+vec2i(1,1))));
+diffuse=clamp(diffuse/total,dlo,dhi);specular=clamp(specular/total,slo,shi);
+${guided ? `var slope=vec4f(0);var pn=vec4f(0);
+for(var i=0u;i<12u;i++){
+let weight=sJ(vec2f(offsets[i])-phase,direction,length,lobe);
+slope+=cY(slopeLo,base+offsets[i])*weight;pn+=cY(offsetLo,base+offsets[i])*weight;
+}
+var qk=cY(slopeLo,base);var te=qk;var kI=cY(offsetLo,base);var ow=kI;
+for(var k=1;k<4;k++){
+let q=base+vec2i(k&1,k>>1u);let a=cY(slopeLo,q);let b=cY(offsetLo,q);
+qk=min(qk,a);te=max(te,a);kI=min(kI,b);ow=max(ow,b);
+}
+slope=clamp(slope/total,qk,te);pn=clamp(pn/total,kI,ow);` : ''}
+let p=vec2i(position.xy);let size=vec2i(textureDimensions(depthHi));
+let d=textureLoad(diffuseHi,p,0);let s=textureLoad(specularHi,p,0);let z=textureLoad(depthHi,p,0).xy;
+let sf=max(detail.w,detail.z*abs(z.x));
+let radius=max(vec2i(1),vec2i(ceil(outputSize.xy/vec2f(textureDimensions(colorImage)))));
+
+
+var silhouette=false;let inverse=1.0/max(z.x,1e-6);
+let directions=array<vec2i,4>(vec2i(1,0),vec2i(0,1),vec2i(1,1),vec2i(1,-1));
+for(var k=0;k<4;k++){
+let a=textureLoad(depthHi,clamp(p+directions[k]*radius,vec2i(0),size-1),0).xy;
+let b=textureLoad(depthHi,clamp(p-directions[k]*radius,vec2i(0),size-1),0).xy;
+silhouette=silhouette || abs(a.y-z.y)>0.5 || abs(b.y-z.y)>0.5 ||
+(min(min(a.y,b.y),z.y)>0.0 && abs(1.0/max(a.x,1e-6)+1.0/max(b.x,1e-6)-2.0*inverse)>detail.z*inverse);
+}
+${silhouettes ? `  
+let guided=min(d.a,s.a)>=detail.y;
+if(silhouette && z.y>=0.5 && !guided){return color;}
+if(silhouette){
+
+
+
+let center=vec2i(round(point));var front=vec4f(0);var back=vec4f(0);var oK=0.0;var pO=0.0;
+for(var y=-1;y<=1;y++){for(var x=-1;x<=1;x++){
+let q=center+vec2i(x,y);let offset=vec2f(q)-point;
+let rB=cY(depthLo,q).xy;let yO=exp(-dot(offset,offset));
+let Ab=exp(-pow(abs(rB.x-z.x)/sf,2.0));
+let sample=ns(q);let value=vec4f(pS(sample.rgb),sample.a);
+let f=yO*rB.y*rB.y*select(1.0,Ab,z.y>0.0);let b=yO*(1.0-rB.y)*(1.0-rB.y);
+front+=value*f;oK+=f;back+=value*b;pO+=b;
+}}
+let ug=oK>1e-3;let xs=pO>1e-3;
+if(!ug && !xs){return color;}
+var zE=select(back/max(pO,1e-8),front/max(oK,1e-8),ug);
+if(ug && xs){zE=mix(back/pO,front/oK,z.y);}
+return vec4f(tp(zE.rgb),zE.a);
+}
+` : '  if(silhouette){return color;}\n'}  let valid=min(min(d.a,s.a),min(diffuse.a,specular.a))>=detail.y && z.y>=detail.y;
+if(!valid){return color;}
+let low=diffuse.rgb+specular.rgb;
+var fw=clamp((d.rgb+s.rgb)/max(low,vec3f(detail.x)),vec3f(outputSize.z),vec3f(outputSize.w));
+${separate ? `
+
+let share=diffuse.rgb/max(low,vec3f(detail.x));
+let rd=clamp(d.rgb/max(diffuse.rgb,vec3f(detail.x)),vec3f(outputSize.z),vec3f(outputSize.w));
+let rs=clamp(s.rgb/max(specular.rgb,vec3f(detail.x)),vec3f(outputSize.z),vec3f(outputSize.w));
+fw=share*rd+(vec3f(1)-share)*rs;` : ''}
+${guided ? `
+
+
+let yJ=slope.rgb*low+pn.rgb;
+fw=select(fw,clamp((slope.rgb*(d.rgb+s.rgb)+pn.rgb)/max(yJ,vec3f(1e-8)),vec3f(outputSize.z),vec3f(outputSize.w)),yJ>vec3f(1e-8));` : ''}
+fw=select(vec3f(1),fw,low>vec3f(detail.x));
+
+return vec4f(tp(zk(pS(color.rgb),fw)),color.a);`);
+}
+// Exact box overlap integrates noninteger footprints instead of point sampling.
+export const albedoDownsampleShader = (depthOutput = false) => /* wgsl */ `
+@group(0) @binding(0) var diffuse: texture_2d<f32>;
+@group(0) @binding(1) var specular: texture_2d<f32>;
+@group(0) @binding(2) var<uniform> outputSize: vec4f;
+@group(0) @binding(3) var depth: texture_2d<f32>;
+@vertex fn vertex(@builtin(vertex_index) index: u32) -> @builtin(position) vec4f {
+let p=vec2f(f32((index<<1u)&2u),f32(index&2u));return vec4f(p*2.0-1.0,0,1);
+}
+struct uS {@location(0) diffuse: vec4f, @location(1) specular: vec4f, ${depthOutput ? '@location(2) depth: vec4f,' : ''}}
+@fragment fn fragment(@builtin(position) p: vec4f) -> uS {
+let scale=vec2f(textureDimensions(diffuse))/outputSize.xy;
+let start=floor(p.xy)*scale;let end=start+scale;
+var d=vec4f(0);var s=vec4f(0);var nearest=3.4e38;var hits=0.0;
+for(var y=i32(floor(start.y));y<i32(ceil(end.y));y++){
+for(var x=i32(floor(start.x));x<i32(ceil(end.x));x++){
+let tK=max(vec2f(0),min(vec2f(vec2i(x+1,y+1)),end)-max(vec2f(vec2i(x,y)),start));let weight=tK.x*tK.y;
+let a=textureLoad(diffuse,vec2i(x,y),0);let b=textureLoad(specular,vec2i(x,y),0);let z=textureLoad(depth,vec2i(x,y),0).xy;
+d+=vec4f(a.rgb*a.a,a.a)*weight;s+=vec4f(b.rgb*b.a,b.a)*weight;
+hits+=z.y*weight;if(z.y>0.0){nearest=min(nearest,z.x);}
+}
+}
+var result: uS;
+${depthOutput ? 'result.depth=vec4f(select(0.0,nearest,hits>0.0),hits/(scale.x*scale.y),0,0);' : ''}
+result.diffuse=vec4f(d.rgb/max(d.a,1e-8),d.a/(scale.x*scale.y));
+result.specular=vec4f(s.rgb/max(s.a,1e-8),s.a/(scale.x*scale.y));return result;
+}
+`;
+// Builds the tone lookup rows for the current grading and tone mapping.
+// toneMappingParams: x mode, y maximum display value, z exposure, w 1 for extended output.
+export const toneLutShader = /* wgsl */ `
+@group(0) @binding(0) var<uniform> toneMappingParams: vec4f;
+@group(0) @binding(1) var<uniform> colorGradingParams: vec4f;
+${composeDefine}
+fn xP(x: f32) -> vec3f {
+var v=fm(vec3f(x)*toneMappingParams.z);
+if(toneMappingParams.w<0.5){v=bH(v);}
+return v;
+}
+@vertex fn vertex(@builtin(vertex_index) index: u32) -> @builtin(position) vec4f {
+let p=vec2f(f32((index<<1u)&2u),f32(index&2u));return vec4f(p*2.0-1.0,0,1);
+}
+@fragment fn fragment(@builtin(position) p: vec4f) -> @location(0) vec4f {
+let u=floor(p.x)/${TONE_LUT_SIZE - 1}.0;
+if(p.y<1.0){return vec4f(xP(exp2(mix(${TONE_LUT_LOG2[0]}.0,${TONE_LUT_LOG2[1]}.0,u))),1);}
+let d=u*u*toneMappingParams.y;var result=vec3f(0);
+for(var c=0;c<3;c++){
+var lo=${TONE_LUT_LOG2[0]}.0;var hi=${TONE_LUT_LOG2[1]}.0;
+for(var i=0;i<28;i++){let mid=0.5*(lo+hi);if(xP(exp2(mid))[c]<d){lo=mid;}else{hi=mid;}}
+result[c]=0.5*(lo+hi);
+}
+return vec4f(result,1);
+}
+`;
+// Per low-resolution pixel, fits scene-linear radiance x ~ slope * albedo + offset over a
+// 3x3 window of guided samples, per channel (a guided-filter regression). Flat albedo
+// gives slope ~0, so its radiance is not modulated; tone.y regularizes the variance.
+export const guidedCoefficientShader = /* wgsl */ `
+@group(0) @binding(0) var colorImage: texture_2d<f32>;
+@group(0) @binding(1) var diffuseLo: texture_2d<f32>;
+@group(0) @binding(2) var specularLo: texture_2d<f32>;
+${toneCodingDefine}
+fn yg(image: texture_2d<f32>, p: vec2i) -> vec4f {return textureLoad(image,clamp(p,vec2i(0),vec2i(textureDimensions(image))-1),0);}
+@vertex fn vertex(@builtin(vertex_index) index: u32) -> @builtin(position) vec4f {
+let p=vec2f(f32((index<<1u)&2u),f32(index&2u));return vec4f(p*2.0-1.0,0,1);
+}
+struct uS {@location(0) slope: vec4f, @location(1) offset: vec4f,}
+@fragment fn fragment(@builtin(position) position: vec4f) -> uS {
+let center=vec2i(position.xy);
+var weight=0.0;var gt=vec3f(0);var nj=vec3f(0);var rC=vec3f(0);var sS=vec3f(0);
+for(var y=-1;y<=1;y++){for(var x=-1;x<=1;x++){
+let q=center+vec2i(x,y);let d=yg(diffuseLo,q);let s=yg(specularLo,q);
+let w=step(0.5,min(d.a,s.a));let albedo=d.rgb+s.rgb;
+let radiance=exp2(uN(pS(yg(colorImage,q).rgb)));
+weight+=w;gt+=albedo*w;nj+=radiance*w;rC+=albedo*radiance*w;sS+=albedo*albedo*w;
+}}
+var result: uS;
+if(weight<1.0){return result;}
+gt/=weight;nj/=weight;rC/=weight;sS/=weight;
+let slope=max((rC-gt*nj)/(sS-gt*gt+vec3f(tone.y)),vec3f(0));
+result.slope=vec4f(slope,1);result.offset=vec4f(nj-slope*gt,1);
+return result;
+}
+`;
+//# sourceMappingURL=easuDetail.wgsl.js.map

package/lib/renderer/wgsl/easuDetail.wgsl.js.map

@@ -0,0 +1 @@
+{"version":3,"file":"easuDetail.wgsl.js","sourceRoot":"","sources":["../../../src/renderer/wgsl/easuDetail.wgsl.ts"],"names":[],"mappings":"AAAA,OAAO,EAAC,UAAU,EAAC,MAAM,gBAAgB,CAAC;AAC1C,OAAO,EAAC,aAAa,EAAC,MAAM,wBAAwB,CAAC;AAErD,gEAAgE;AAChE,MAAM,CAAC,MAAM,aAAa,GAAG,CAAC,CAAC,EAAE,EAAE,EAAE,CAAU,CAAC;AAChD,MAAM,CAAC,MAAM,aAAa,GAAG,IAAI,CAAC;AAClC,+FAA+F;AAC/F,MAAM,CAAC,MAAM,WAAW,GAAG;IACzB,UAAU,EAAE,CAAC,GAAG,EAAE,IAAI,CAAU,EAAE,cAAc,EAAE,IAAI,EAAE,iBAAiB,EAAE,IAAI;IAC/E,yEAAyE;IACzE,aAAa,EAAE,IAAI,EAAE,cAAc,EAAE,IAAI;IACzC,6FAA6F;IAC7F,aAAa,EAAE,IAAI;CACpB,CAAC;AAEF,4GAA4G;AAC5G,MAAM,gBAAgB,GAAG,UAAU,CAAC;;;;;;;;;;;;8CAYiC,aAAa,CAAC,CAAC,CAAC,QAAQ,aAAa,CAAC,CAAC,CAAC,GAAC,aAAa,CAAC,CAAC,CAAC;;;;;;;;;;;;;;;;;;;CAmB7H,CAAC;AAEF;;;;GAIG;AACH,MAAM,UAAU,gBAAgB,CAAC,QAAQ,GAAG,KAAK,EAAE,WAAW,GAAG,KAAK,EAAE,MAAM,GAAG,IAAI;IACnF,OAAO,UAAU,CAAC,OAAO,CAAC,mBAAmB,EAAE,UAAU,CAAC;;;;;;;EAO1D,WAAW,CAAC,CAAC,CAAC,sDAAsD,CAAC,CAAC,CAAC,EAAE;EACzE,MAAM,CAAC,CAAC,CAAC,6GAA6G,CAAC,CAAC,CAAC,EAAE;;;;EAI3H,gBAAgB;kBACA,CAAC,CAAC,OAAO,CAAC,yCAAyC,EAAE,UAAU,CAAC;;;;;;;;;;;;;;EAc9E,MAAM,CAAC,CAAC,CAAC;;;;;;;;;;yDAUoG,CAAC,CAAC,CAAC,EAAE;;;;;;;;;;;;;;;EAelH,WAAW,CAAC,CAAC,CAAC;;;;;;;;;;;;;;;;;;;;;;CAsBjB,CAAC,CAAC,CAAC,mCAAmC;;;;EAInC,QAAQ,CAAC,CAAC,CAAC;;;;;iCAKuB,CAAC,CAAC,CAAC,EAAE;EACvC,MAAM,CAAC,CAAC,CAAC;;;;kIAI8I,CAAC,CAAC,CAAC,EAAE;;;gDAGhF,CAAC,CAAC;AAClF,CAAC;AAED,gFAAgF;AAChF,MAAM,CAAC,MAAM,sBAAsB,GAAG,CAAC,WAAW,GAAG,KAAK,EAAE,EAAE,CAAC,UAAU,CAAC;;;;;;;;wEAQE,WAAW,CAAC,CAAC,CAAC,4BAA4B,CAAC,CAAC,CAAC,EAAE;;;;;;;;;;;;;;EAcvH,WAAW,CAAC,CAAC,CAAC,8EAA8E,CAAC,CAAC,CAAC,EAAE;;;;CAIpG,CAAC;AAEF,wEAAwE;AACxE,2FAA2F;AAC3F,MAAM,CAAC,MAAM,aAAa,GAAG,UAAU,CAAC;;;EAGtC,aAAa;;;;;;;;;;mBAUM,aAAa,GAAG,CAAC;uCACS,aAAa,CAAC,CAAC,CAAC,MAAM,aAAa,CAAC,CAAC,CAAC;;;SAGxE,aAAa,CAAC,CAAC,CAAC,aAAa,aAAa,CAAC,CAAC,CAAC;;;;;;CAMzD,CAAC;AAEF,0FAA0F;AAC1F,sFAAsF;AACtF,qFAAqF;AACrF,MAAM,CAAC,MAAM,uBAAuB,GAAG,UAAU,CAAC;;;;EAIhD,gBAAgB;;;;;;;;;;;;;;;;;;;;;;CAsBjB,CAAC","sourcesContent":["import {easuShader} from './easu.wgsl.js';\nimport {composeDefine} from './compose.frag.wgsl.js';\n\n/** Scene-linear log2 range covered by the tone lookup table. */\nexport const TONE_LUT_LOG2 = [-24, 24] as const;\nexport const TONE_LUT_SIZE = 1024;\n/** Detail compensation defaults and fixed regularizers shared by the Web and native passes. */\nexport const EASU_DETAIL = {\n  ratioRange: [0.7, 1.43] as const, depthThreshold: 0.02, coverageThreshold: 0.95,\n  // Albedo below this is treated as black; relative depth tolerance floor.\n  minimumAlbedo: 1e-3, depthTolerance: 1e-4,\n  // Regularizes the local albedo variance of the albedo-radiance fit, in squared albedo units.\n  guidedEpsilon: 1e-4,\n};\n\n// Tone table lookups and display encoding shared by the detail passes; tone.x is the maximum display value.\nconst toneCodingDefine = /* wgsl */ `\n// Row 0: display value of a gray scene-linear input at TONE_LUT_LOG2 stops, per channel.\n// Row 1: log2 scene-linear input per channel for display values (sqrt-spaced up to tone.x).\n@group(0) @binding(8) var toneLut: texture_2d<f32>;\n@group(0) @binding(9) var<uniform> tone: vec4f;\nfn toneLutRow(row: i32, u: vec3f) -> vec3f {\n  let last=f32(textureDimensions(toneLut).x-1u);let p=clamp(u,vec3f(0),vec3f(1))*last;\n  let i=vec3i(floor(p));let j=min(i+1,vec3i(i32(last)));let f=p-floor(p);\n  var result=vec3f(0);\n  for(var c=0;c<3;c++){result[c]=mix(textureLoad(toneLut,vec2i(i[c],row),0)[c],textureLoad(toneLut,vec2i(j[c],row),0)[c],f[c]);}\n  return result;\n}\nfn toneForward(log2x: vec3f) -> vec3f {return toneLutRow(0,(log2x-(${TONE_LUT_LOG2[0]}.0))/${TONE_LUT_LOG2[1]-TONE_LUT_LOG2[0]}.0);}\nfn toneInverse(d: vec3f) -> vec3f {return toneLutRow(1,sqrt(max(d,vec3f(0))/tone.x));}\n// Applies a scene-linear albedo ratio before the display transform: displayed\n// RGB is mapped back to scene-linear per channel, scaled and mapped forward, so\n// curve shoulders and toes respond as the tone mapper would. Only the change is\n// added to the displayed value, which keeps it exact where the ratio is 1.\nfn toneAware(d: vec3f, ratio: vec3f) -> vec3f {\n  let x=toneInverse(d);\n  let changed=max(d+toneForward(x+log2(max(ratio,vec3f(1e-6))))-toneForward(x),vec3f(0));\n  return select(changed,d,abs(ratio-vec3f(1))<vec3f(1e-4));\n}\nfn detailDecode(c: vec3f) -> vec3f {\n  let a=max(c,vec3f(0));\n  return select(pow((a+0.055)/1.055,vec3f(2.4)),a/12.92,a<=vec3f(0.04045));\n}\nfn detailEncode(c: vec3f) -> vec3f {\n  let a=max(c,vec3f(0));\n  return select(1.055*pow(a,vec3f(1.0/2.4))-0.055,a*12.92,a<=vec3f(0.0031308));\n}\n`;\n\n/**\n * guided: scale only the albedo-correlated part of local scene-linear radiance (default);\n * separate: lobe-share approximation; silhouettes: depth-guided edge reconstruction.\n * The last two are retained for comparisons.\n */\nexport function easuDetailShader(separate = false, silhouettes = false, guided = true) {\n  return easuShader.replace(`@vertex fn vertex`, /* wgsl */ `\n@group(0) @binding(2) var diffuseHi: texture_2d<f32>;\n@group(0) @binding(3) var specularHi: texture_2d<f32>;\n@group(0) @binding(4) var depthHi: texture_2d<f32>;\n@group(0) @binding(5) var diffuseLo: texture_2d<f32>;\n@group(0) @binding(6) var specularLo: texture_2d<f32>;\n@group(0) @binding(7) var<uniform> detail: vec4f;\n${silhouettes ? '@group(0) @binding(10) var depthLo: texture_2d<f32>;' : ''}\n${guided ? '@group(0) @binding(11) var slopeLo: texture_2d<f32>;\\n@group(0) @binding(12) var offsetLo: texture_2d<f32>;' : ''}\nfn loadLow(image: texture_2d<f32>, p: vec2i) -> vec4f {\n  return textureLoad(image,clamp(p,vec2i(0),vec2i(textureDimensions(image))-1),0);\n}\n${toneCodingDefine}\n@vertex fn vertex`).replace(`return clamp(result/total,lower,upper);`, /* wgsl */ `\n  let color=clamp(result/total,lower,upper);\n  var diffuse=vec4f(0);var specular=vec4f(0);\n  for(var i=0u;i<12u;i++){\n    // Identical direction, phase, support and weights as beauty.\n    let weight=easuWeight(vec2f(offsets[i])-phase,direction,length,lobe);\n    diffuse+=loadLow(diffuseLo,base+offsets[i])*weight;\n    specular+=loadLow(specularLo,base+offsets[i])*weight;\n  }\n  let dlo=min(min(loadLow(diffuseLo,base),loadLow(diffuseLo,base+vec2i(1,0))),min(loadLow(diffuseLo,base+vec2i(0,1)),loadLow(diffuseLo,base+vec2i(1,1))));\n  let dhi=max(max(loadLow(diffuseLo,base),loadLow(diffuseLo,base+vec2i(1,0))),max(loadLow(diffuseLo,base+vec2i(0,1)),loadLow(diffuseLo,base+vec2i(1,1))));\n  let slo=min(min(loadLow(specularLo,base),loadLow(specularLo,base+vec2i(1,0))),min(loadLow(specularLo,base+vec2i(0,1)),loadLow(specularLo,base+vec2i(1,1))));\n  let shi=max(max(loadLow(specularLo,base),loadLow(specularLo,base+vec2i(1,0))),max(loadLow(specularLo,base+vec2i(0,1)),loadLow(specularLo,base+vec2i(1,1))));\n  diffuse=clamp(diffuse/total,dlo,dhi);specular=clamp(specular/total,slo,shi);\n  ${guided ? `var slope=vec4f(0);var intercept=vec4f(0);\n  for(var i=0u;i<12u;i++){\n    let weight=easuWeight(vec2f(offsets[i])-phase,direction,length,lobe);\n    slope+=loadLow(slopeLo,base+offsets[i])*weight;intercept+=loadLow(offsetLo,base+offsets[i])*weight;\n  }\n  var lowerSlope=loadLow(slopeLo,base);var upperSlope=lowerSlope;var lowerIntercept=loadLow(offsetLo,base);var upperIntercept=lowerIntercept;\n  for(var k=1;k<4;k++){\n    let q=base+vec2i(k&1,k>>1u);let a=loadLow(slopeLo,q);let b=loadLow(offsetLo,q);\n    lowerSlope=min(lowerSlope,a);upperSlope=max(upperSlope,a);lowerIntercept=min(lowerIntercept,b);upperIntercept=max(upperIntercept,b);\n  }\n  slope=clamp(slope/total,lowerSlope,upperSlope);intercept=clamp(intercept/total,lowerIntercept,upperIntercept);` : ''}\n  let p=vec2i(position.xy);let size=vec2i(textureDimensions(depthHi));\n  let d=textureLoad(diffuseHi,p,0);let s=textureLoad(specularHi,p,0);let z=textureLoad(depthHi,p,0).xy;\n  let tolerance=max(detail.w,detail.z*abs(z.x));\n  let radius=max(vec2i(1),vec2i(ceil(outputSize.xy/vec2f(textureDimensions(colorImage)))));\n  // A silhouette changes hit coverage or breaks the planarity of inverse depth,\n  // which is linear in screen space on planes; sloped planes are not silhouettes.\n  var silhouette=false;let inverse=1.0/max(z.x,1e-6);\n  let directions=array<vec2i,4>(vec2i(1,0),vec2i(0,1),vec2i(1,1),vec2i(1,-1));\n  for(var k=0;k<4;k++){\n    let a=textureLoad(depthHi,clamp(p+directions[k]*radius,vec2i(0),size-1),0).xy;\n    let b=textureLoad(depthHi,clamp(p-directions[k]*radius,vec2i(0),size-1),0).xy;\n    silhouette=silhouette || abs(a.y-z.y)>0.5 || abs(b.y-z.y)>0.5 ||\n      (min(min(a.y,b.y),z.y)>0.0 && abs(1.0/max(a.x,1e-6)+1.0/max(b.x,1e-6)-2.0*inverse)>detail.z*inverse);\n  }\n  ${silhouettes ? `  // Transmissive and other unguided hits show surfaces behind their depth; keep EASU there.\n  let guided=min(d.a,s.a)>=detail.y;\n  if(silhouette && z.y>=0.5 && !guided){return color;}\n  if(silhouette){\n    // Silhouettes: estimate this surface's color from covered low-resolution neighbors\n    // at a similar nearest depth, the background from uncovered ones, and mix them by\n    // this pixel's full-resolution hit coverage, which keeps edges antialiased.\n    let center=vec2i(round(point));var front=vec4f(0);var back=vec4f(0);var frontWeight=0.0;var backWeight=0.0;\n    for(var y=-1;y<=1;y++){for(var x=-1;x<=1;x++){\n      let q=center+vec2i(x,y);let offset=vec2f(q)-point;\n      let lowDepth=loadLow(depthLo,q).xy;let spatial=exp(-dot(offset,offset));\n      let similar=exp(-pow(abs(lowDepth.x-z.x)/tolerance,2.0));\n      let sample=loadColor(q);let value=vec4f(detailDecode(sample.rgb),sample.a);\n      let f=spatial*lowDepth.y*lowDepth.y*select(1.0,similar,z.y>0.0);let b=spatial*(1.0-lowDepth.y)*(1.0-lowDepth.y);\n      front+=value*f;frontWeight+=f;back+=value*b;backWeight+=b;\n    }}\n    let haveFront=frontWeight>1e-3;let haveBack=backWeight>1e-3;\n    if(!haveFront && !haveBack){return color;}\n    var mixed=select(back/max(backWeight,1e-8),front/max(frontWeight,1e-8),haveFront);\n    if(haveFront && haveBack){mixed=mix(back/backWeight,front/frontWeight,z.y);}\n    return vec4f(detailEncode(mixed.rgb),mixed.a);\n  }\n` : '  if(silhouette){return color;}\\n'}  let valid=min(min(d.a,s.a),min(diffuse.a,specular.a))>=detail.y && z.y>=detail.y;\n  if(!valid){return color;}\n  let low=diffuse.rgb+specular.rgb;\n  var ratio=clamp((d.rgb+s.rgb)/max(low,vec3f(detail.x)),vec3f(outputSize.z),vec3f(outputSize.w));\n  ${separate ? `\n  // Estimate beauty lobe shares from low albedo; no separate beauty lobes are available.\n  let share=diffuse.rgb/max(low,vec3f(detail.x));\n  let rd=clamp(d.rgb/max(diffuse.rgb,vec3f(detail.x)),vec3f(outputSize.z),vec3f(outputSize.w));\n  let rs=clamp(s.rgb/max(specular.rgb,vec3f(detail.x)),vec3f(outputSize.z),vec3f(outputSize.w));\n  ratio=share*rd+(vec3f(1)-share)*rs;` : ''}\n  ${guided ? `// Local radiance is modeled as slope * albedo + intercept. Only the albedo-correlated\n  // part follows the full-resolution albedo; the intercept, such as specular highlights,\n  // does not. Pure diffuse (zero intercept) reduces to the albedo ratio.\n  let modeled=slope.rgb*low+intercept.rgb;\n  ratio=select(ratio,clamp((slope.rgb*(d.rgb+s.rgb)+intercept.rgb)/max(modeled,vec3f(1e-8)),vec3f(outputSize.z),vec3f(outputSize.w)),modeled>vec3f(1e-8));` : ''}\n  ratio=select(vec3f(1),ratio,low>vec3f(detail.x));\n  // Albedo ratios are scene-linear; the composed image is tone mapped and sRGB encoded.\n  return vec4f(detailEncode(toneAware(detailDecode(color.rgb),ratio)),color.a);`);\n}\n\n// Exact box overlap integrates noninteger footprints instead of point sampling.\nexport const albedoDownsampleShader = (depthOutput = false) => /* wgsl */ `\n@group(0) @binding(0) var diffuse: texture_2d<f32>;\n@group(0) @binding(1) var specular: texture_2d<f32>;\n@group(0) @binding(2) var<uniform> outputSize: vec4f;\n@group(0) @binding(3) var depth: texture_2d<f32>;\n@vertex fn vertex(@builtin(vertex_index) index: u32) -> @builtin(position) vec4f {\n  let p=vec2f(f32((index<<1u)&2u),f32(index&2u));return vec4f(p*2.0-1.0,0,1);\n}\nstruct Result {@location(0) diffuse: vec4f, @location(1) specular: vec4f, ${depthOutput ? '@location(2) depth: vec4f,' : ''}}\n@fragment fn fragment(@builtin(position) p: vec4f) -> Result {\n  let scale=vec2f(textureDimensions(diffuse))/outputSize.xy;\n  let start=floor(p.xy)*scale;let end=start+scale;\n  var d=vec4f(0);var s=vec4f(0);var nearest=3.4e38;var hits=0.0;\n  for(var y=i32(floor(start.y));y<i32(ceil(end.y));y++){\n    for(var x=i32(floor(start.x));x<i32(ceil(end.x));x++){\n      let overlap=max(vec2f(0),min(vec2f(vec2i(x+1,y+1)),end)-max(vec2f(vec2i(x,y)),start));let weight=overlap.x*overlap.y;\n      let a=textureLoad(diffuse,vec2i(x,y),0);let b=textureLoad(specular,vec2i(x,y),0);let z=textureLoad(depth,vec2i(x,y),0).xy;\n      d+=vec4f(a.rgb*a.a,a.a)*weight;s+=vec4f(b.rgb*b.a,b.a)*weight;\n      hits+=z.y*weight;if(z.y>0.0){nearest=min(nearest,z.x);}\n    }\n  }\n  var result: Result;\n  ${depthOutput ? 'result.depth=vec4f(select(0.0,nearest,hits>0.0),hits/(scale.x*scale.y),0,0);' : ''}\n  result.diffuse=vec4f(d.rgb/max(d.a,1e-8),d.a/(scale.x*scale.y));\n  result.specular=vec4f(s.rgb/max(s.a,1e-8),s.a/(scale.x*scale.y));return result;\n}\n`;\n\n// Builds the tone lookup rows for the current grading and tone mapping.\n// toneMappingParams: x mode, y maximum display value, z exposure, w 1 for extended output.\nexport const toneLutShader = /* wgsl */ `\n@group(0) @binding(0) var<uniform> toneMappingParams: vec4f;\n@group(0) @binding(1) var<uniform> colorGradingParams: vec4f;\n${composeDefine}\nfn toneGray(x: f32) -> vec3f {\n  var v=gradeColor(vec3f(x)*toneMappingParams.z);\n  if(toneMappingParams.w<0.5){v=applyToneMapping(v);}\n  return v;\n}\n@vertex fn vertex(@builtin(vertex_index) index: u32) -> @builtin(position) vec4f {\n  let p=vec2f(f32((index<<1u)&2u),f32(index&2u));return vec4f(p*2.0-1.0,0,1);\n}\n@fragment fn fragment(@builtin(position) p: vec4f) -> @location(0) vec4f {\n  let u=floor(p.x)/${TONE_LUT_SIZE - 1}.0;\n  if(p.y<1.0){return vec4f(toneGray(exp2(mix(${TONE_LUT_LOG2[0]}.0,${TONE_LUT_LOG2[1]}.0,u))),1);}\n  let d=u*u*toneMappingParams.y;var result=vec3f(0);\n  for(var c=0;c<3;c++){\n    var lo=${TONE_LUT_LOG2[0]}.0;var hi=${TONE_LUT_LOG2[1]}.0;\n    for(var i=0;i<28;i++){let mid=0.5*(lo+hi);if(toneGray(exp2(mid))[c]<d){lo=mid;}else{hi=mid;}}\n    result[c]=0.5*(lo+hi);\n  }\n  return vec4f(result,1);\n}\n`;\n\n// Per low-resolution pixel, fits scene-linear radiance x ~ slope * albedo + offset over a\n// 3x3 window of guided samples, per channel (a guided-filter regression). Flat albedo\n// gives slope ~0, so its radiance is not modulated; tone.y regularizes the variance.\nexport const guidedCoefficientShader = /* wgsl */ `\n@group(0) @binding(0) var colorImage: texture_2d<f32>;\n@group(0) @binding(1) var diffuseLo: texture_2d<f32>;\n@group(0) @binding(2) var specularLo: texture_2d<f32>;\n${toneCodingDefine}\nfn loadAt(image: texture_2d<f32>, p: vec2i) -> vec4f {return textureLoad(image,clamp(p,vec2i(0),vec2i(textureDimensions(image))-1),0);}\n@vertex fn vertex(@builtin(vertex_index) index: u32) -> @builtin(position) vec4f {\n  let p=vec2f(f32((index<<1u)&2u),f32(index&2u));return vec4f(p*2.0-1.0,0,1);\n}\nstruct Result {@location(0) slope: vec4f, @location(1) offset: vec4f,}\n@fragment fn fragment(@builtin(position) position: vec4f) -> Result {\n  let center=vec2i(position.xy);\n  var weight=0.0;var meanAlbedo=vec3f(0);var meanRadiance=vec3f(0);var meanProduct=vec3f(0);var meanSquare=vec3f(0);\n  for(var y=-1;y<=1;y++){for(var x=-1;x<=1;x++){\n    let q=center+vec2i(x,y);let d=loadAt(diffuseLo,q);let s=loadAt(specularLo,q);\n    let w=step(0.5,min(d.a,s.a));let albedo=d.rgb+s.rgb;\n    let radiance=exp2(toneInverse(detailDecode(loadAt(colorImage,q).rgb)));\n    weight+=w;meanAlbedo+=albedo*w;meanRadiance+=radiance*w;meanProduct+=albedo*radiance*w;meanSquare+=albedo*albedo*w;\n  }}\n  var result: Result;\n  if(weight<1.0){return result;}\n  meanAlbedo/=weight;meanRadiance/=weight;meanProduct/=weight;meanSquare/=weight;\n  let slope=max((meanProduct-meanAlbedo*meanRadiance)/(meanSquare-meanAlbedo*meanAlbedo+vec3f(tone.y)),vec3f(0));\n  result.slope=vec4f(slope,1);result.offset=vec4f(meanRadiance-slope*meanAlbedo,1);\n  return result;\n}\n`;\n"]}

package/lib/renderer/wgsl/jnlm.compute.js

@@ -7,7 +7,7 @@ function guideBinding(guides, binding, name) {
         : `@group(0) @binding(${binding}) var<storage, read> ${name}: array<vec4f>;`;
 }
 function loadGuide(guides, name, position) {
-    return guides === 'textures' ? `textureLoad(${name}, ${position}, 0)` : `${name}[gv(${position})]`;
+    return guides === 'textures' ? `textureLoad(${name}, ${position}, 0)` : `${name}[gx(${position})]`;
 }
 // Shared uniform packing also fixes sensitivity and dilation for native callers.
 export function packJNLMParameters(width, height, strength, pass) {
@@ -43,9 +43,9 @@ ${shadowCatcherAccumulationWGSL}
 fn main(@builtin(global_invocation_id) id: vec3u) {
 let size = params.size;
 if (id.x >= size.x || id.y >= size.y) { return; }
-let qE = id.x + (size.y - 1u - id.y) * size.x;
-let n = normalBuffer[qE].a;
-let moments = accumulation[qE].moments;
+let rJ = id.x + (size.y - 1u - id.y) * size.x;
+let n = normalBuffer[rJ].a;
+let moments = accumulation[rJ].moments;
 let mean = moments.x / max(n, 1.0);
 
 var variance = -1.0;
@@ -69,28 +69,28 @@ fn main(@builtin(global_invocation_id) id: vec3u) {
 let size = params.size;
 if (id.x >= size.x || id.y >= size.y) { return; }
 let center = rawGuide[id.x + id.y * size.x];
-var mf = 0.0;
-var jx = 0.0;
-var nW = 0.0;
-var jp = 0.0;
+var mF = 0.0;
+var jI = 0.0;
+var oL = 0.0;
+var jB = 0.0;
 for (var dy = -1; dy <= 1; dy++) {
 for (var dx = -1; dx <= 1; dx++) {
 let q = clamp(vec2i(id.xy) + vec2i(dx, dy), vec2i(0), vec2i(size) - 1);
 let value = rawGuide[u32(q.x) + u32(q.y) * size.x];
-nW += value.x;
-jp += value.x * value.x;
+oL += value.x;
+jB += value.x * value.x;
 if (value.y >= 0.0) {
-mf += value.y;
-jx += 1.0;
+mF += value.y;
+jI += 1.0;
 }
 }
 }
 var variance: f32;
 if (center.y >= 0.0) {
-variance = mf / jx;
+variance = mF / jI;
 } else {
-let me = nW / 9.0;
-variance = max(jp / 9.0 - me * me, 0.0) * 9.0 / 8.0;
+let mE = oL / 9.0;
+variance = max(jB / 9.0 - mE * mE, 0.0) * 9.0 / 8.0;
 }
 guide[id.x + id.y * size.x] = vec2f(center.x, variance);
 }
@@ -114,11 +114,11 @@ fn main(@builtin(global_invocation_id) id: vec3u) {
 if (id.x >= params.size.x || id.y >= params.size.y) { return; }
 outColor[id.x + id.y * params.size.x] = textureLoad(colorTex, vec2i(id.xy), 0);
 ${guides === 'buffers' ? `  let index = id.x + id.y * params.size.x;
-let qE = id.x + (params.size.y - 1u - id.y) * params.size.x;
-let n = normalAccumulation[qE];
+let rJ = id.x + (params.size.y - 1u - id.y) * params.size.x;
+let n = normalAccumulation[rJ];
 let samples = max(n.a, 1.0);
 normal[index] = vec4f(n.rgb / samples * 0.5 + 0.5, select(0.0, 1.0, dot(n.rgb, n.rgb) > 0.1));
-shadow[index] = shadowAccumulation[qE].mask / samples;
+shadow[index] = shadowAccumulation[rJ].mask / samples;
 albedo[index] = textureLoad(albedoTex, vec2i(id.xy), 0);` : ''}
 }
 `;
@@ -155,23 +155,23 @@ const INV_NORMAL_BANDWIDTH2 = ${(1 / normalBandwidth ** 2).toFixed(6)};
 
 const CENTER_WEIGHT_MIN = 0.01;
 
-fn gv(p: vec2i) -> u32 {
+fn gx(p: vec2i) -> u32 {
 let q = clamp(p, vec2i(0), vec2i(params.size) - 1);
 return u32(q.x) + u32(q.y) * params.size.x;
 }
 
-fn mi(q: vec2i, bC: vec4f, hg: vec3f) -> f32 {
+fn mJ(q: vec2i, bp: vec4f, hk: vec3f) -> f32 {
 let albedo = ${loadGuide(guides, 'albedoTex', 'q')};
 
 
-let qd = max((albedo.rgb + bC.rgb) * 0.5, vec3f(${JNLM_ALBEDO_FLOOR}));
-let lN = (albedo.rgb - bC.rgb) / qd;
-let jk = albedo.a - bC.a;
-let lX = ${loadGuide(guides, 'normalTex', 'q')}.rgb - hg;
+let rc = max((albedo.rgb + bp.rgb) * 0.5, vec3f(${JNLM_ALBEDO_FLOOR}));
+let mn = (albedo.rgb - bp.rgb) / rc;
+let jv = albedo.a - bp.a;
+let my = ${loadGuide(guides, 'normalTex', 'q')}.rgb - hk;
 return max(
-dot(lN, lN) * INV_ALBEDO_BANDWIDTH2
-+ jk * jk * INV_COVERAGE_BANDWIDTH2,
-dot(lX, lX) * INV_NORMAL_BANDWIDTH2
+dot(mn, mn) * INV_ALBEDO_BANDWIDTH2
++ jv * jv * INV_COVERAGE_BANDWIDTH2,
+dot(my, my) * INV_NORMAL_BANDWIDTH2
 );
 }
 
@@ -185,37 +185,37 @@ let size = vec2i(params.size);
 let p = vec2i(id.xy);
 if (p.x >= size.x || p.y >= size.y) { return; }
 let step = i32(params.dilation);
-let bC = ${loadGuide(guides, 'albedoTex', 'p')};
-let fR = ${loadGuide(guides, 'normalTex', 'p')};
-let hg = fR.rgb;
-let index = gv(p);
+let bp = ${loadGuide(guides, 'albedoTex', 'p')};
+let fO = ${loadGuide(guides, 'normalTex', 'p')};
+let hk = fO.rgb;
+let index = gx(p);
 
 
-if (fR.a <= 0.0 && ${loadGuide(guides, 'shadowCatcherTex', 'p')}.r <= 0.0) {
+if (fO.a <= 0.0 && ${loadGuide(guides, 'shadowCatcherTex', 'p')}.r <= 0.0) {
 outColor[index] = color[index];
 outGuide[index] = guide[index];
 return;
 }
 
-var lP: array<vec2f, PATCH_SIZE>;
+var mq: array<vec2f, PATCH_SIZE>;
 var k = 0;
 for (var dy = -PATCH_RADIUS; dy <= PATCH_RADIUS; dy++) {
 for (var dx = -PATCH_RADIUS; dx <= PATCH_RADIUS; dx++) {
-lP[k] = guide[gv(p + vec2i(dx, dy) * step)];
+mq[k] = guide[gx(p + vec2i(dx, dy) * step)];
 k++;
 }
 }
 let k2 = params.colorSensitivity * params.colorSensitivity;
 
 var sum = vec4f(0.0);
-let hk = max(bC.rgb, vec3f(${JNLM_ALBEDO_FLOOR}));
-var ng = vec3f(0.0);
-var gN = vec3f(0.0);
-var hG = vec3f(0.0);
-var qp = 0.0;
-var gP = 0.0;
-var oP = 0.0;
-var lt = 0.0;
+let hn = max(bp.rgb, vec3f(${JNLM_ALBEDO_FLOOR}));
+var nS = vec3f(0.0);
+var gP = vec3f(0.0);
+var hH = vec3f(0.0);
+var rn = 0.0;
+var gR = 0.0;
+var pK = 0.0;
+var lR = 0.0;
 for (var sy = -SEARCH_RADIUS; sy <= SEARCH_RADIUS; sy++) {
 for (var sx = -SEARCH_RADIUS; sx <= SEARCH_RADIUS; sx++) {
 let q = p + vec2i(sx, sy) * step;
@@ -224,63 +224,63 @@ if (
 q.x < 0 || q.y < 0 || q.x >= size.x || q.y >= size.y
 ) { continue; }
 
-let features = mi(q, bC, hg);
+let features = mJ(q, bp, hk);
 if (features > 16.0) { continue; }
 
-var gk = 0.0;
+var gn = 0.0;
 k = 0;
 for (var dy = -PATCH_RADIUS; dy <= PATCH_RADIUS; dy++) {
 for (var dx = -PATCH_RADIUS; dx <= PATCH_RADIUS; dx++) {
-let a = lP[k];
-let b = guide[gv(q + vec2i(dx, dy) * step)];
-let rZ = a.x - b.x;
-gk += (rZ * rZ
+let a = mq[k];
+let b = guide[gx(q + vec2i(dx, dy) * step)];
+let tn = a.x - b.x;
+gn += (tn * tn
 - VARIANCE_CANCELLATION * (a.y + min(a.y, b.y)))
 / (1e-10 + k2 * (a.y + b.y));
 k++;
 }
 }
-gk /= f32(PATCH_SIZE);
+gn /= f32(PATCH_SIZE);
 
-let weight = exp(-max(max(gk, 0.0), features));
-let index = gv(q);
+let weight = exp(-max(max(gn, 0.0), features));
+let index = gx(q);
 let c = color[index];
 let g = guide[index];
 
-let a = (${loadGuide(guides, 'albedoTex', 'q')}.rgb - bC.rgb) / hk;
+let a = (${loadGuide(guides, 'albedoTex', 'q')}.rgb - bp.rgb) / hn;
 let value = premultiplied(c);
 sum += value * weight;
-ng += a * weight;
-gN += a * a * weight;
-hG += a * value.rgb * weight;
-qp += g.x * weight;
-gP += g.y * weight * weight;
-oP += weight;
-lt = max(lt, weight);
+nS += a * weight;
+gP += a * a * weight;
+hH += a * value.rgb * weight;
+rn += g.x * weight;
+gR += g.y * weight * weight;
+pK += weight;
+lR = max(lR, weight);
 }
 }
 
 
 
 let center = color[index];
-let lO = guide[index];
-let dR = max(lt, CENTER_WEIGHT_MIN);
-let totalWeight = oP + dR;
-let vj = premultiplied(center);
-let mean = (sum + vj * dR) / totalWeight;
+let mp = guide[index];
+let dG = max(lR, CENTER_WEIGHT_MIN);
+let totalWeight = pK + dG;
+let xj = premultiplied(center);
+let mean = (sum + xj * dG) / totalWeight;
 
 
-let eI = ng / totalWeight;
-let mj = max(gN / totalWeight - eI * eI, vec3f(0.0));
-let rv = hG / totalWeight - eI * mean.rgb;
+let eA = nS / totalWeight;
+let mK = max(gP / totalWeight - eA * eA, vec3f(0.0));
+let sH = hH / totalWeight - eA * mean.rgb;
 
 
 
-let rI = bC.rgb + eI * hk;
-let po = mean.rgb * hk / max(rI, vec3f(${JNLM_ALBEDO_FLOOR}));
-let xg = clamp(rv / (mj + vec3f(${JNLM_ALBEDO_REGULARIZATION})), vec3f(0.0), po);
-let sD = mean.rgb - xg * eI;
-let filtered = vec4f(max(sD, vec3f(0.0)), max(mean.a, 0.0));
+let gt = bp.rgb + eA * hn;
+let qm = mean.rgb * hn / max(gt, vec3f(${JNLM_ALBEDO_FLOOR}));
+let slope = clamp(sH / (mK + vec3f(${JNLM_ALBEDO_REGULARIZATION})), vec3f(0.0), qm);
+let tY = mean.rgb - slope * eA;
+let filtered = vec4f(max(tY, vec3f(0.0)), max(mean.a, 0.0));
 let rgb = select(
 center.rgb,
 filtered.rgb / max(filtered.a, 1e-6),
@@ -288,8 +288,8 @@ filtered.a > 1e-6
 );
 outColor[index] = vec4f(rgb, filtered.a);
 outGuide[index] = vec2f(
-(qp + lO.x * dR) / totalWeight,
-(gP + lO.y * dR * dR)
+(rn + mp.x * dG) / totalWeight,
+(gR + mp.y * dG * dG)
 / (totalWeight * totalWeight)
 );
 }

package/lib/renderer/wgsl/jnlm.compute.js.map

@@ -1 +1 @@
-{"version":3,"file":"jnlm.compute.js","sourceRoot":"","sources":["../../../src/renderer/wgsl/jnlm.compute.ts"],"names":[],"mappings":"AAAA,OAAO,6BAA6B,MAAM,4CAA4C,CAAC;AACvF,OAAO,EAAC,kBAAkB,EAAC,MAAM,wBAAwB,CAAC;AAC1D,OAAO,EACL,8BAA8B,EAAE,kBAAkB,EAAE,iBAAiB,EACrE,0BAA0B,EAAE,uBAAuB,EAAE,gBAAgB,EAEtE,MAAM,YAAY,CAAC;AAIpB,SAAS,YAAY,CAAC,MAAyB,EAAE,OAAe,EAAE,IAAY;IAC5E,OAAO,MAAM,KAAK,UAAU;QAC1B,CAAC,CAAC,sBAAsB,OAAO,SAAS,IAAI,oBAAoB;QAChE,CAAC,CAAC,sBAAsB,OAAO,wBAAwB,IAAI,iBAAiB,CAAC;AACjF,CAAC;AACD,SAAS,SAAS,CAAC,MAAyB,EAAE,IAAY,EAAE,QAAgB;IAC1E,OAAO,MAAM,KAAK,UAAU,CAAC,CAAC,CAAC,eAAe,IAAI,KAAK,QAAQ,MAAM,CAAC,CAAC,CAAC,GAAG,IAAI,OAAe,QAAQ,IAAI,CAAC;AAC7G,CAAC;AAED,iFAAiF;AACjF,MAAM,UAAU,kBAAkB,CAAC,KAAa,EAAE,MAAc,EAAE,QAAgB,EAAE,IAAY;IAC9F,MAAM,KAAK,GAAG,IAAI,WAAW,CAAC,EAAE,CAAC,CAAC;IAClC,IAAI,WAAW,CAAC,KAAK,EAAE,CAAC,EAAE,CAAC,CAAC,CAAC,GAAG,CAAC,CAAC,KAAK,EAAE,MAAM,CAAC,CAAC,CAAC;IAClD,IAAI,YAAY,CAAC,KAAK,EAAE,CAAC,EAAE,CAAC,CAAC,CAAC,GAAG,CAAC;QAChC,8BAA8B,GAAG,IAAI,CAAC,GAAG,CAAC,QAAQ,EAAE,IAAI,CAAC,EAAE,gBAAgB,CAAC,IAAI,CAAC;KAClF,CAAC,CAAC;IACH,OAAO,KAAK,CAAC;AACf,CAAC;AAED,MAAM,CAAC,MAAM,mBAAmB,GAAG,CAAC,CAAC;AAErC,MAAM,UAAU,GAAG,UAAU,CAAC;;;;;;;;CAQ7B,CAAC;AAEF;;;GAGG;AACH,MAAM,CAAC,MAAM,aAAa,GAAG,UAAU,CAAC;EACtC,UAAU;EACV,6BAA6B;;;;;2BAKJ,mBAAmB,KAAK,mBAAmB;;;;;;;;;;;;;;;CAerE,CAAC;AAEF;;;GAGG;AACH,MAAM,CAAC,MAAM,gBAAgB,GAAG,UAAU,CAAC;EACzC,UAAU;;;;2BAIe,mBAAmB,KAAK,mBAAmB;;;;;;;;;;;;;;;;;;;;;;;;;;;;;;CA8BrE,CAAC;AAEF,gEAAgE;AAChE,MAAM,UAAU,uBAAuB,CAAC,SAA4B,UAAU;IAC5E,OAAO,UAAU,CAAC;EAClB,UAAU;;;EAGV,MAAM,KAAK,SAAS,CAAC,CAAC,CAAC,GAAG,6BAA6B;;;;;;qEAMY,CAAC,CAAC,CAAC,EAAE;;2BAE/C,mBAAmB,KAAK,mBAAmB;;;;EAIpE,MAAM,KAAK,SAAS,CAAC,CAAC,CAAC;;;;;;yDAMkC,CAAC,CAAC,CAAC,EAAE;;CAE/D,CAAC;AACF,CAAC;AACD,MAAM,CAAC,MAAM,iBAAiB,GAAG,uBAAuB,EAAE,CAAC;AAE3D;;;;;GAKG;AACH,MAAM,UAAU,oBAAoB,CAAC,EACnC,YAAY,EAAE,WAAW,EAAE,eAAe,EAAE,eAAe,KACxC,kBAAkB,EAAE,SAA4B,UAAU;IAC7E,MAAM,SAAS,GAAG,CAAC,CAAC,GAAG,WAAW,GAAG,CAAC,CAAC,IAAI,CAAC,CAAC;IAC7C,OAAO,UAAU,CAAC;EAClB,UAAU;;;EAGV,YAAY,CAAC,MAAM,EAAE,CAAC,EAAE,WAAW,CAAC;EACpC,YAAY,CAAC,MAAM,EAAE,CAAC,EAAE,WAAW,CAAC;;;EAGpC,YAAY,CAAC,MAAM,EAAE,CAAC,EAAE,kBAAkB,CAAC;;wBAErB,YAAY;uBACb,WAAW;qBACb,SAAS;;;;gCAIE,CAAC,CAAC,GAAG,eAAe,IAAI,CAAC,CAAC,CAAC,OAAO,CAAC,CAAC,CAAC;kCACnC,CAAC,CAAC,GAAG,uBAAuB,IAAI,CAAC,CAAC,CAAC,OAAO,CAAC,CAAC,CAAC;gCAC/C,CAAC,CAAC,GAAG,eAAe,IAAI,CAAC,CAAC,CAAC,OAAO,CAAC,CAAC,CAAC;;;;;;;;;;;eAWpD,SAAS,CAAC,MAAM,EAAE,WAAW,EAAE,GAAG,CAAC;;;kDAGmB,iBAAiB;;;WAGlE,SAAS,CAAC,MAAM,EAAE,WAAW,EAAE,GAAG,CAAC;;;;;;;;;;;;2BAY9B,mBAAmB,KAAK,mBAAmB;;;;;;WAM/C,SAAS,CAAC,MAAM,EAAE,WAAW,EAAE,GAAG,CAAC;WAC9B,SAAS,CAAC,MAAM,EAAE,WAAW,EAAE,GAAG,CAAC;;;;;qBAKzB,SAAS,CAAC,MAAM,EAAE,kBAAkB,EAAE,GAAG,CAAC;;;;;;;;;;;;;;;;;6BAiB7B,iBAAiB;;;;;;;;;;;;;;;;;;;;;;;;;;;;;;;;;;;;;;;WAuCnD,SAAS,CAAC,MAAM,EAAE,WAAW,EAAE,GAAG,CAAC;;;;;;;;;;;;;;;;;;;;;;;;;;;;;;yCA8BmB,iBAAiB;kCAC5B,0BAA0B;;;;;;;;;;;;;;;CAerF,CAAC;AACF,CAAC;AAED,4EAA4E;AAC5E,6DAA6D;AAC7D,MAAM,CAAC,MAAM,cAAc,GAAG,UAAU,CAAC;;;;;EAKvC,kBAAkB;2BACO,mBAAmB,KAAK,mBAAmB;;;;;;;;CAQrE,CAAC","sourcesContent":["import shadowCatcherAccumulationWGSL from './chunks/shadowCatcherAccumulation.wgsl.js';\nimport {linearToSrgbDefine} from './compose.frag.wgsl.js';\nimport {\n  JNLM_DEFAULT_COLOR_SENSITIVITY, JNLM_DEFAULT_SHAPE, JNLM_ALBEDO_FLOOR,\n  JNLM_ALBEDO_REGULARIZATION, JNLM_COVERAGE_BANDWIDTH, jnlmPassDilation,\n  type JNLMFilterShape\n} from '../jnlm.js';\n\n// Web textures and native normalized guide buffers use the same filter body.\nexport type JNLMGuideBindings = 'textures' | 'buffers';\nfunction guideBinding(guides: JNLMGuideBindings, binding: number, name: string): string {\n  return guides === 'textures'\n    ? `@group(0) @binding(${binding}) var ${name}: texture_2d<f32>;`\n    : `@group(0) @binding(${binding}) var<storage, read> ${name}: array<vec4f>;`;\n}\nfunction loadGuide(guides: JNLMGuideBindings, name: string, position: string): string {\n  return guides === 'textures' ? `textureLoad(${name}, ${position}, 0)` : `${name}[pixelIndex(${position})]`;\n}\n\n// Shared uniform packing also fixes sensitivity and dilation for native callers.\nexport function packJNLMParameters(width: number, height: number, strength: number, pass: number): ArrayBuffer {\n  const bytes = new ArrayBuffer(16);\n  new Uint32Array(bytes, 0, 2).set([width, height]);\n  new Float32Array(bytes, 8, 2).set([\n    JNLM_DEFAULT_COLOR_SENSITIVITY * Math.max(strength, 1e-3), jnlmPassDilation(pass)\n  ]);\n  return bytes;\n}\n\nexport const JNLM_WORKGROUP_SIZE = 8;\n\nconst paramsWGSL = /* wgsl */ `\nstruct JNLMParams {\n  size: vec2u,\n  colorSensitivity: f32,\n  // Pixel spacing of the search window and patches in this pass.\n  dilation: f32,\n}\n@group(0) @binding(0) var<uniform> params: JNLMParams;\n`;\n\n/**\n * Pass 1: the guide mean and the variance of that mean from the accumulated\n * moments. Ray buffers are stored bottom-up; guides use texture rows.\n */\nexport const jnlmGuideWGSL = /* wgsl */ `\n${paramsWGSL}\n${shadowCatcherAccumulationWGSL}\n@group(0) @binding(1) var<storage, read> accumulation: array<ShadowCatcherAccumulation>;\n@group(0) @binding(2) var<storage, read> normalBuffer: array<vec4f>;\n@group(0) @binding(3) var<storage, read_write> rawGuide: array<vec2f>;\n\n@compute @workgroup_size(${JNLM_WORKGROUP_SIZE}, ${JNLM_WORKGROUP_SIZE}, 1)\nfn main(@builtin(global_invocation_id) id: vec3u) {\n  let size = params.size;\n  if (id.x >= size.x || id.y >= size.y) { return; }\n  let rayIndex = id.x + (size.y - 1u - id.y) * size.x;\n  let n = normalBuffer[rayIndex].a;\n  let moments = accumulation[rayIndex].moments;\n  let mean = moments.x / max(n, 1.0);\n  // Variance of the mean, or -1 when one sample cannot estimate it.\n  var variance = -1.0;\n  if (n >= 2.0) {\n    variance = max(moments.y - n * mean * mean, 0.0) / (n * (n - 1.0));\n  }\n  rawGuide[id.x + id.y * size.x] = vec2f(mean, variance);\n}\n`;\n\n/**\n * Pass 2: smooth the noisy per-pixel variance over 3 x 3 pixels. Pixels with\n * a single sample take the spatial variance of their neighborhood instead.\n */\nexport const jnlmVarianceWGSL = /* wgsl */ `\n${paramsWGSL}\n@group(0) @binding(1) var<storage, read> rawGuide: array<vec2f>;\n@group(0) @binding(2) var<storage, read_write> guide: array<vec2f>;\n\n@compute @workgroup_size(${JNLM_WORKGROUP_SIZE}, ${JNLM_WORKGROUP_SIZE}, 1)\nfn main(@builtin(global_invocation_id) id: vec3u) {\n  let size = params.size;\n  if (id.x >= size.x || id.y >= size.y) { return; }\n  let center = rawGuide[id.x + id.y * size.x];\n  var varianceSum = 0.0;\n  var varianceCount = 0.0;\n  var meanSum = 0.0;\n  var meanSquareSum = 0.0;\n  for (var dy = -1; dy <= 1; dy++) {\n    for (var dx = -1; dx <= 1; dx++) {\n      let q = clamp(vec2i(id.xy) + vec2i(dx, dy), vec2i(0), vec2i(size) - 1);\n      let value = rawGuide[u32(q.x) + u32(q.y) * size.x];\n      meanSum += value.x;\n      meanSquareSum += value.x * value.x;\n      if (value.y >= 0.0) {\n        varianceSum += value.y;\n        varianceCount += 1.0;\n      }\n    }\n  }\n  var variance: f32;\n  if (center.y >= 0.0) {\n    variance = varianceSum / varianceCount;\n  } else {\n    let spatialMean = meanSum / 9.0;\n    variance = max(meanSquareSum / 9.0 - spatialMean * spatialMean, 0.0) * 9.0 / 8.0;\n  }\n  guide[id.x + id.y * size.x] = vec2f(center.x, variance);\n}\n`;\n\n/** Copies the composed image into the filter's color buffer. */\nexport function createJNLMLoadColorWGSL(guides: JNLMGuideBindings = 'textures'): string {\n  return /* wgsl */ `\n${paramsWGSL}\n@group(0) @binding(1) var colorTex: texture_2d<f32>;\n@group(0) @binding(2) var<storage, read_write> outColor: array<vec4f>;\n${guides === 'buffers' ? `${shadowCatcherAccumulationWGSL}\n@group(0) @binding(3) var<storage, read> normalAccumulation: array<vec4f>;\n@group(0) @binding(4) var<storage, read_write> normal: array<vec4f>;\n@group(0) @binding(5) var<storage, read> shadowAccumulation: array<ShadowCatcherAccumulation>;\n@group(0) @binding(6) var<storage, read_write> shadow: array<vec4f>;\n@group(0) @binding(7) var albedoTex: texture_2d<f32>;\n@group(0) @binding(8) var<storage, read_write> albedo: array<vec4f>;` : ''}\n\n@compute @workgroup_size(${JNLM_WORKGROUP_SIZE}, ${JNLM_WORKGROUP_SIZE}, 1)\nfn main(@builtin(global_invocation_id) id: vec3u) {\n  if (id.x >= params.size.x || id.y >= params.size.y) { return; }\n  outColor[id.x + id.y * params.size.x] = textureLoad(colorTex, vec2i(id.xy), 0);\n${guides === 'buffers' ? `  let index = id.x + id.y * params.size.x;\n  let rayIndex = id.x + (params.size.y - 1u - id.y) * params.size.x;\n  let n = normalAccumulation[rayIndex];\n  let samples = max(n.a, 1.0);\n  normal[index] = vec4f(n.rgb / samples * 0.5 + 0.5, select(0.0, 1.0, dot(n.rgb, n.rgb) > 0.1));\n  shadow[index] = shadowAccumulation[rayIndex].mask / samples;\n  albedo[index] = textureLoad(albedoTex, vec2i(id.xy), 0);` : ''}\n}\n`;\n}\nexport const jnlmLoadColorWGSL = createJNLMLoadColorWGSL();\n\n/**\n * Pass 3: the joint NL-means filter of the composed image. It also filters\n * the guide with the same weights, with the variance of the weighted mean,\n * for a following pass. The albedo regression corrects color only, keeping\n * uncertain fitted texture responses out of the matching guide.\n */\nexport function createJNLMFilterWGSL({\n  searchRadius, patchRadius, albedoBandwidth, normalBandwidth\n}: JNLMFilterShape = JNLM_DEFAULT_SHAPE, guides: JNLMGuideBindings = 'textures') {\n  const patchSize = (2 * patchRadius + 1) ** 2;\n  return /* wgsl */ `\n${paramsWGSL}\n@group(0) @binding(1) var<storage, read> guide: array<vec2f>;\n@group(0) @binding(2) var<storage, read> color: array<vec4f>;\n${guideBinding(guides, 3, 'albedoTex')}\n${guideBinding(guides, 4, 'normalTex')}\n@group(0) @binding(5) var<storage, read_write> outColor: array<vec4f>;\n@group(0) @binding(6) var<storage, read_write> outGuide: array<vec2f>;\n${guideBinding(guides, 7, 'shadowCatcherTex')}\n\nconst SEARCH_RADIUS = ${searchRadius};\nconst PATCH_RADIUS = ${patchRadius};\nconst PATCH_SIZE = ${patchSize};\n// Variance cancellation strength alpha of the paper.\nconst VARIANCE_CANCELLATION = 1.0;\n// Inverse squared feature bandwidths. Normals are encoded as n * 0.5 + 0.5.\nconst INV_ALBEDO_BANDWIDTH2 = ${(1 / albedoBandwidth ** 2).toFixed(6)};\nconst INV_COVERAGE_BANDWIDTH2 = ${(1 / JNLM_COVERAGE_BANDWIDTH ** 2).toFixed(6)};\nconst INV_NORMAL_BANDWIDTH2 = ${(1 / normalBandwidth ** 2).toFixed(6)};\n// Lower bound of the center weight, so a pixel unlike all of its neighbors\n// (a thin highlight) is kept rather than replaced by them.\nconst CENTER_WEIGHT_MIN = 0.01;\n\nfn pixelIndex(p: vec2i) -> u32 {\n  let q = clamp(p, vec2i(0), vec2i(params.size) - 1);\n  return u32(q.x) + u32(q.y) * params.size.x;\n}\n\nfn featureDistance(q: vec2i, centerAlbedo: vec4f, centerNormal: vec3f) -> f32 {\n  let albedo = ${loadGuide(guides, 'albedoTex', 'q')};\n  // Absolute differences in linear albedo erase low-contrast dark textures.\n  // Measure RGB contrast relative to both pixels; keep coverage absolute.\n  let albedoScale = max((albedo.rgb + centerAlbedo.rgb) * 0.5, vec3f(${JNLM_ALBEDO_FLOOR}));\n  let albedoDelta = (albedo.rgb - centerAlbedo.rgb) / albedoScale;\n  let coverageDelta = albedo.a - centerAlbedo.a;\n  let normalDelta = ${loadGuide(guides, 'normalTex', 'q')}.rgb - centerNormal;\n  return max(\n    dot(albedoDelta, albedoDelta) * INV_ALBEDO_BANDWIDTH2\n      + coverageDelta * coverageDelta * INV_COVERAGE_BANDWIDTH2,\n    dot(normalDelta, normalDelta) * INV_NORMAL_BANDWIDTH2\n  );\n}\n\nfn premultiplied(c: vec4f) -> vec4f {\n  return vec4f(c.rgb * c.a, c.a);\n}\n\n@compute @workgroup_size(${JNLM_WORKGROUP_SIZE}, ${JNLM_WORKGROUP_SIZE}, 1)\nfn main(@builtin(global_invocation_id) id: vec3u) {\n  let size = vec2i(params.size);\n  let p = vec2i(id.xy);\n  if (p.x >= size.x || p.y >= size.y) { return; }\n  let step = i32(params.dilation);\n  let centerAlbedo = ${loadGuide(guides, 'albedoTex', 'p')};\n  let centerNormalTexel = ${loadGuide(guides, 'normalTex', 'p')};\n  let centerNormal = centerNormalTexel.rgb;\n  let index = pixelIndex(p);\n  // Primary misses show the background directly. It has no features that\n  // protect its detail and no noise beyond antialiasing, so keep it.\n  if (centerNormalTexel.a <= 0.0 && ${loadGuide(guides, 'shadowCatcherTex', 'p')}.r <= 0.0) {\n    outColor[index] = color[index];\n    outGuide[index] = guide[index];\n    return;\n  }\n\n  var centerPatch: array<vec2f, PATCH_SIZE>;\n  var k = 0;\n  for (var dy = -PATCH_RADIUS; dy <= PATCH_RADIUS; dy++) {\n    for (var dx = -PATCH_RADIUS; dx <= PATCH_RADIUS; dx++) {\n      centerPatch[k] = guide[pixelIndex(p + vec2i(dx, dy) * step)];\n      k++;\n    }\n  }\n  let k2 = params.colorSensitivity * params.colorSensitivity;\n\n  var sum = vec4f(0.0);\n  let featureScale = max(centerAlbedo.rgb, vec3f(${JNLM_ALBEDO_FLOOR}));\n  var featureSum = vec3f(0.0);\n  var featureSquareSum = vec3f(0.0);\n  var featureColorSum = vec3f(0.0);\n  var guideSum = 0.0;\n  var guideVarianceSum = 0.0;\n  var weightSum = 0.0;\n  var maxWeight = 0.0;\n  for (var sy = -SEARCH_RADIUS; sy <= SEARCH_RADIUS; sy++) {\n    for (var sx = -SEARCH_RADIUS; sx <= SEARCH_RADIUS; sx++) {\n      let q = p + vec2i(sx, sy) * step;\n      if (\n        (sx == 0 && sy == 0) ||\n        q.x < 0 || q.y < 0 || q.x >= size.x || q.y >= size.y\n      ) { continue; }\n\n      let features = featureDistance(q, centerAlbedo, centerNormal);\n      if (features > 16.0) { continue; }\n\n      var colorDistance = 0.0;\n      k = 0;\n      for (var dy = -PATCH_RADIUS; dy <= PATCH_RADIUS; dy++) {\n        for (var dx = -PATCH_RADIUS; dx <= PATCH_RADIUS; dx++) {\n          let a = centerPatch[k];\n          let b = guide[pixelIndex(q + vec2i(dx, dy) * step)];\n          let delta = a.x - b.x;\n          colorDistance += (delta * delta\n              - VARIANCE_CANCELLATION * (a.y + min(a.y, b.y)))\n            / (1e-10 + k2 * (a.y + b.y));\n          k++;\n        }\n      }\n      colorDistance /= f32(PATCH_SIZE);\n\n      let weight = exp(-max(max(colorDistance, 0.0), features));\n      let index = pixelIndex(q);\n      let c = color[index];\n      let g = guide[index];\n      // Filter premultiplied color so coverage and color stay consistent.\n      let a = (${loadGuide(guides, 'albedoTex', 'q')}.rgb - centerAlbedo.rgb) / featureScale;\n      let value = premultiplied(c);\n      sum += value * weight;\n      featureSum += a * weight;\n      featureSquareSum += a * a * weight;\n      featureColorSum += a * value.rgb * weight;\n      guideSum += g.x * weight;\n      guideVarianceSum += g.y * weight * weight;\n      weightSum += weight;\n      maxWeight = max(maxWeight, weight);\n    }\n  }\n\n  // Weight the center like its most similar neighbor (Buades et al. 2005)\n  // instead of 1, so isolated outliers are not kept unfiltered.\n  let center = color[index];\n  let centerGuide = guide[index];\n  let centerWeight = max(maxWeight, CENTER_WEIGHT_MIN);\n  let totalWeight = weightSum + centerWeight;\n  let centerValue = premultiplied(center);\n  let mean = (sum + centerValue * centerWeight) / totalWeight;\n  // Fit each color channel against relative albedo with the same JNLM weights.\n  // Centering at this pixel avoids cancellation when the feature is constant.\n  let meanFeature = featureSum / totalWeight;\n  let featureVariance = max(featureSquareSum / totalWeight - meanFeature * meanFeature, vec3f(0.0));\n  let covariance = featureColorSum / totalWeight - meanFeature * mean.rgb;\n  // A 1% contrast ridge stabilizes flat guides. Bound the response between no\n  // texture dependence and multiplicative albedo modulation; specular color\n  // with an unrelated albedo falls back to the weighted mean.\n  let meanAlbedo = centerAlbedo.rgb + meanFeature * featureScale;\n  let maximumSlope = mean.rgb * featureScale / max(meanAlbedo, vec3f(${JNLM_ALBEDO_FLOOR}));\n  let slope = clamp(covariance / (featureVariance + vec3f(${JNLM_ALBEDO_REGULARIZATION})), vec3f(0.0), maximumSlope);\n  let corrected = mean.rgb - slope * meanFeature;\n  let filtered = vec4f(max(corrected, vec3f(0.0)), max(mean.a, 0.0));\n  let rgb = select(\n    center.rgb,\n    filtered.rgb / max(filtered.a, 1e-6),\n    filtered.a > 1e-6\n  );\n  outColor[index] = vec4f(rgb, filtered.a);\n  outGuide[index] = vec2f(\n    (guideSum + centerGuide.x * centerWeight) / totalWeight,\n    (guideVarianceSum + centerGuide.y * centerWeight * centerWeight)\n      / (totalWeight * totalWeight)\n  );\n}\n`;\n}\n\n// Compute presentation for hosts that consume the filtered buffer directly.\n// Use the same sRGB transfer as browser display composition.\nexport const jnlmEncodeWGSL = /* wgsl */ `\nstruct OutputParams { size: vec2u, premultiply: u32, reserved: u32 }\n@group(0) @binding(0) var<uniform> params: OutputParams;\n@group(0) @binding(1) var<storage, read> color: array<vec4f>;\n@group(0) @binding(2) var output: texture_storage_2d<rgba8unorm, write>;\n${linearToSrgbDefine}\n@compute @workgroup_size(${JNLM_WORKGROUP_SIZE}, ${JNLM_WORKGROUP_SIZE}, 1)\nfn main(@builtin(global_invocation_id) id: vec3u) {\n  if (any(id.xy >= params.size)) { return; }\n  let c = color[id.y * params.size.x + id.x];\n  var rgb = linearToSrgb(c.rgb);\n  if (params.premultiply != 0u) { rgb *= c.a; }\n  textureStore(output, id.xy, vec4f(rgb, c.a));\n}\n`;\n"]}
+{"version":3,"file":"jnlm.compute.js","sourceRoot":"","sources":["../../../src/renderer/wgsl/jnlm.compute.ts"],"names":[],"mappings":"AAAA,OAAO,6BAA6B,MAAM,4CAA4C,CAAC;AACvF,OAAO,EAAC,kBAAkB,EAAC,MAAM,wBAAwB,CAAC;AAC1D,OAAO,EACL,8BAA8B,EAAE,kBAAkB,EAAE,iBAAiB,EACrE,0BAA0B,EAAE,uBAAuB,EAAE,gBAAgB,EAEtE,MAAM,YAAY,CAAC;AAIpB,SAAS,YAAY,CAAC,MAAyB,EAAE,OAAe,EAAE,IAAY;IAC5E,OAAO,MAAM,KAAK,UAAU;QAC1B,CAAC,CAAC,sBAAsB,OAAO,SAAS,IAAI,oBAAoB;QAChE,CAAC,CAAC,sBAAsB,OAAO,wBAAwB,IAAI,iBAAiB,CAAC;AACjF,CAAC;AACD,SAAS,SAAS,CAAC,MAAyB,EAAE,IAAY,EAAE,QAAgB;IAC1E,OAAO,MAAM,KAAK,UAAU,CAAC,CAAC,CAAC,eAAe,IAAI,KAAK,QAAQ,MAAM,CAAC,CAAC,CAAC,GAAG,IAAI,OAAe,QAAQ,IAAI,CAAC;AAC7G,CAAC;AAED,iFAAiF;AACjF,MAAM,UAAU,kBAAkB,CAAC,KAAa,EAAE,MAAc,EAAE,QAAgB,EAAE,IAAY;IAC9F,MAAM,KAAK,GAAG,IAAI,WAAW,CAAC,EAAE,CAAC,CAAC;IAClC,IAAI,WAAW,CAAC,KAAK,EAAE,CAAC,EAAE,CAAC,CAAC,CAAC,GAAG,CAAC,CAAC,KAAK,EAAE,MAAM,CAAC,CAAC,CAAC;IAClD,IAAI,YAAY,CAAC,KAAK,EAAE,CAAC,EAAE,CAAC,CAAC,CAAC,GAAG,CAAC;QAChC,8BAA8B,GAAG,IAAI,CAAC,GAAG,CAAC,QAAQ,EAAE,IAAI,CAAC,EAAE,gBAAgB,CAAC,IAAI,CAAC;KAClF,CAAC,CAAC;IACH,OAAO,KAAK,CAAC;AACf,CAAC;AAED,MAAM,CAAC,MAAM,mBAAmB,GAAG,CAAC,CAAC;AAErC,MAAM,UAAU,GAAG,UAAU,CAAC;;;;;;;;CAQ7B,CAAC;AAEF;;;GAGG;AACH,MAAM,CAAC,MAAM,aAAa,GAAG,UAAU,CAAC;EACtC,UAAU;EACV,6BAA6B;;;;;2BAKJ,mBAAmB,KAAK,mBAAmB;;;;;;;;;;;;;;;CAerE,CAAC;AAEF;;;GAGG;AACH,MAAM,CAAC,MAAM,gBAAgB,GAAG,UAAU,CAAC;EACzC,UAAU;;;;2BAIe,mBAAmB,KAAK,mBAAmB;;;;;;;;;;;;;;;;;;;;;;;;;;;;;;CA8BrE,CAAC;AAEF,gEAAgE;AAChE,MAAM,UAAU,uBAAuB,CAAC,SAA4B,UAAU;IAC5E,OAAO,UAAU,CAAC;EAClB,UAAU;;;EAGV,MAAM,KAAK,SAAS,CAAC,CAAC,CAAC,GAAG,6BAA6B;;;;;;qEAMY,CAAC,CAAC,CAAC,EAAE;;2BAE/C,mBAAmB,KAAK,mBAAmB;;;;EAIpE,MAAM,KAAK,SAAS,CAAC,CAAC,CAAC;;;;;;yDAMkC,CAAC,CAAC,CAAC,EAAE;;CAE/D,CAAC;AACF,CAAC;AACD,MAAM,CAAC,MAAM,iBAAiB,GAAG,uBAAuB,EAAE,CAAC;AAE3D;;;;;GAKG;AACH,MAAM,UAAU,oBAAoB,CAAC,EACnC,YAAY,EAAE,WAAW,EAAE,eAAe,EAAE,eAAe,KACxC,kBAAkB,EAAE,SAA4B,UAAU;IAC7E,MAAM,SAAS,GAAG,CAAC,CAAC,GAAG,WAAW,GAAG,CAAC,CAAC,IAAI,CAAC,CAAC;IAC7C,OAAO,UAAU,CAAC;EAClB,UAAU;;;EAGV,YAAY,CAAC,MAAM,EAAE,CAAC,EAAE,WAAW,CAAC;EACpC,YAAY,CAAC,MAAM,EAAE,CAAC,EAAE,WAAW,CAAC;;;EAGpC,YAAY,CAAC,MAAM,EAAE,CAAC,EAAE,kBAAkB,CAAC;;wBAErB,YAAY;uBACb,WAAW;qBACb,SAAS;;;;gCAIE,CAAC,CAAC,GAAG,eAAe,IAAI,CAAC,CAAC,CAAC,OAAO,CAAC,CAAC,CAAC;kCACnC,CAAC,CAAC,GAAG,uBAAuB,IAAI,CAAC,CAAC,CAAC,OAAO,CAAC,CAAC,CAAC;gCAC/C,CAAC,CAAC,GAAG,eAAe,IAAI,CAAC,CAAC,CAAC,OAAO,CAAC,CAAC,CAAC;;;;;;;;;;;eAWpD,SAAS,CAAC,MAAM,EAAE,WAAW,EAAE,GAAG,CAAC;;;kDAGmB,iBAAiB;;;WAGlE,SAAS,CAAC,MAAM,EAAE,WAAW,EAAE,GAAG,CAAC;;;;;;;;;;;;2BAY9B,mBAAmB,KAAK,mBAAmB;;;;;;WAM/C,SAAS,CAAC,MAAM,EAAE,WAAW,EAAE,GAAG,CAAC;WAC9B,SAAS,CAAC,MAAM,EAAE,WAAW,EAAE,GAAG,CAAC;;;;;qBAKzB,SAAS,CAAC,MAAM,EAAE,kBAAkB,EAAE,GAAG,CAAC;;;;;;;;;;;;;;;;;6BAiB7B,iBAAiB;;;;;;;;;;;;;;;;;;;;;;;;;;;;;;;;;;;;;;;WAuCnD,SAAS,CAAC,MAAM,EAAE,WAAW,EAAE,GAAG,CAAC;;;;;;;;;;;;;;;;;;;;;;;;;;;;;;yCA8BmB,iBAAiB;qCAC5B,0BAA0B;;;;;;;;;;;;;;;CAerF,CAAC;AACF,CAAC;AAED,4EAA4E;AAC5E,6DAA6D;AAC7D,MAAM,CAAC,MAAM,cAAc,GAAG,UAAU,CAAC;;;;;EAKvC,kBAAkB;2BACO,mBAAmB,KAAK,mBAAmB;;;;;;;;CAQrE,CAAC","sourcesContent":["import shadowCatcherAccumulationWGSL from './chunks/shadowCatcherAccumulation.wgsl.js';\nimport {linearToSrgbDefine} from './compose.frag.wgsl.js';\nimport {\n  JNLM_DEFAULT_COLOR_SENSITIVITY, JNLM_DEFAULT_SHAPE, JNLM_ALBEDO_FLOOR,\n  JNLM_ALBEDO_REGULARIZATION, JNLM_COVERAGE_BANDWIDTH, jnlmPassDilation,\n  type JNLMFilterShape\n} from '../jnlm.js';\n\n// Web textures and native normalized guide buffers use the same filter body.\nexport type JNLMGuideBindings = 'textures' | 'buffers';\nfunction guideBinding(guides: JNLMGuideBindings, binding: number, name: string): string {\n  return guides === 'textures'\n    ? `@group(0) @binding(${binding}) var ${name}: texture_2d<f32>;`\n    : `@group(0) @binding(${binding}) var<storage, read> ${name}: array<vec4f>;`;\n}\nfunction loadGuide(guides: JNLMGuideBindings, name: string, position: string): string {\n  return guides === 'textures' ? `textureLoad(${name}, ${position}, 0)` : `${name}[pixelIndex(${position})]`;\n}\n\n// Shared uniform packing also fixes sensitivity and dilation for native callers.\nexport function packJNLMParameters(width: number, height: number, strength: number, pass: number): ArrayBuffer {\n  const bytes = new ArrayBuffer(16);\n  new Uint32Array(bytes, 0, 2).set([width, height]);\n  new Float32Array(bytes, 8, 2).set([\n    JNLM_DEFAULT_COLOR_SENSITIVITY * Math.max(strength, 1e-3), jnlmPassDilation(pass)\n  ]);\n  return bytes;\n}\n\nexport const JNLM_WORKGROUP_SIZE = 8;\n\nconst paramsWGSL = /* wgsl */ `\nstruct JNLMParams {\n  size: vec2u,\n  colorSensitivity: f32,\n  // Pixel spacing of the search window and patches in this pass.\n  dilation: f32,\n}\n@group(0) @binding(0) var<uniform> params: JNLMParams;\n`;\n\n/**\n * Pass 1: the guide mean and the variance of that mean from the accumulated\n * moments. Ray buffers are stored bottom-up; guides use texture rows.\n */\nexport const jnlmGuideWGSL = /* wgsl */ `\n${paramsWGSL}\n${shadowCatcherAccumulationWGSL}\n@group(0) @binding(1) var<storage, read> accumulation: array<ShadowCatcherAccumulation>;\n@group(0) @binding(2) var<storage, read> normalBuffer: array<vec4f>;\n@group(0) @binding(3) var<storage, read_write> rawGuide: array<vec2f>;\n\n@compute @workgroup_size(${JNLM_WORKGROUP_SIZE}, ${JNLM_WORKGROUP_SIZE}, 1)\nfn main(@builtin(global_invocation_id) id: vec3u) {\n  let size = params.size;\n  if (id.x >= size.x || id.y >= size.y) { return; }\n  let rayIndex = id.x + (size.y - 1u - id.y) * size.x;\n  let n = normalBuffer[rayIndex].a;\n  let moments = accumulation[rayIndex].moments;\n  let mean = moments.x / max(n, 1.0);\n  // Variance of the mean, or -1 when one sample cannot estimate it.\n  var variance = -1.0;\n  if (n >= 2.0) {\n    variance = max(moments.y - n * mean * mean, 0.0) / (n * (n - 1.0));\n  }\n  rawGuide[id.x + id.y * size.x] = vec2f(mean, variance);\n}\n`;\n\n/**\n * Pass 2: smooth the noisy per-pixel variance over 3 x 3 pixels. Pixels with\n * a single sample take the spatial variance of their neighborhood instead.\n */\nexport const jnlmVarianceWGSL = /* wgsl */ `\n${paramsWGSL}\n@group(0) @binding(1) var<storage, read> rawGuide: array<vec2f>;\n@group(0) @binding(2) var<storage, read_write> guide: array<vec2f>;\n\n@compute @workgroup_size(${JNLM_WORKGROUP_SIZE}, ${JNLM_WORKGROUP_SIZE}, 1)\nfn main(@builtin(global_invocation_id) id: vec3u) {\n  let size = params.size;\n  if (id.x >= size.x || id.y >= size.y) { return; }\n  let center = rawGuide[id.x + id.y * size.x];\n  var varianceSum = 0.0;\n  var varianceCount = 0.0;\n  var meanSum = 0.0;\n  var meanSquareSum = 0.0;\n  for (var dy = -1; dy <= 1; dy++) {\n    for (var dx = -1; dx <= 1; dx++) {\n      let q = clamp(vec2i(id.xy) + vec2i(dx, dy), vec2i(0), vec2i(size) - 1);\n      let value = rawGuide[u32(q.x) + u32(q.y) * size.x];\n      meanSum += value.x;\n      meanSquareSum += value.x * value.x;\n      if (value.y >= 0.0) {\n        varianceSum += value.y;\n        varianceCount += 1.0;\n      }\n    }\n  }\n  var variance: f32;\n  if (center.y >= 0.0) {\n    variance = varianceSum / varianceCount;\n  } else {\n    let spatialMean = meanSum / 9.0;\n    variance = max(meanSquareSum / 9.0 - spatialMean * spatialMean, 0.0) * 9.0 / 8.0;\n  }\n  guide[id.x + id.y * size.x] = vec2f(center.x, variance);\n}\n`;\n\n/** Copies the composed image into the filter's color buffer. */\nexport function createJNLMLoadColorWGSL(guides: JNLMGuideBindings = 'textures'): string {\n  return /* wgsl */ `\n${paramsWGSL}\n@group(0) @binding(1) var colorTex: texture_2d<f32>;\n@group(0) @binding(2) var<storage, read_write> outColor: array<vec4f>;\n${guides === 'buffers' ? `${shadowCatcherAccumulationWGSL}\n@group(0) @binding(3) var<storage, read> normalAccumulation: array<vec4f>;\n@group(0) @binding(4) var<storage, read_write> normal: array<vec4f>;\n@group(0) @binding(5) var<storage, read> shadowAccumulation: array<ShadowCatcherAccumulation>;\n@group(0) @binding(6) var<storage, read_write> shadow: array<vec4f>;\n@group(0) @binding(7) var albedoTex: texture_2d<f32>;\n@group(0) @binding(8) var<storage, read_write> albedo: array<vec4f>;` : ''}\n\n@compute @workgroup_size(${JNLM_WORKGROUP_SIZE}, ${JNLM_WORKGROUP_SIZE}, 1)\nfn main(@builtin(global_invocation_id) id: vec3u) {\n  if (id.x >= params.size.x || id.y >= params.size.y) { return; }\n  outColor[id.x + id.y * params.size.x] = textureLoad(colorTex, vec2i(id.xy), 0);\n${guides === 'buffers' ? `  let index = id.x + id.y * params.size.x;\n  let rayIndex = id.x + (params.size.y - 1u - id.y) * params.size.x;\n  let n = normalAccumulation[rayIndex];\n  let samples = max(n.a, 1.0);\n  normal[index] = vec4f(n.rgb / samples * 0.5 + 0.5, select(0.0, 1.0, dot(n.rgb, n.rgb) > 0.1));\n  shadow[index] = shadowAccumulation[rayIndex].mask / samples;\n  albedo[index] = textureLoad(albedoTex, vec2i(id.xy), 0);` : ''}\n}\n`;\n}\nexport const jnlmLoadColorWGSL = createJNLMLoadColorWGSL();\n\n/**\n * Pass 3: the joint NL-means filter of the composed image. It also filters\n * the guide with the same weights, with the variance of the weighted mean,\n * for a following pass. The albedo regression corrects color only, keeping\n * uncertain fitted texture responses out of the matching guide.\n */\nexport function createJNLMFilterWGSL({\n  searchRadius, patchRadius, albedoBandwidth, normalBandwidth\n}: JNLMFilterShape = JNLM_DEFAULT_SHAPE, guides: JNLMGuideBindings = 'textures') {\n  const patchSize = (2 * patchRadius + 1) ** 2;\n  return /* wgsl */ `\n${paramsWGSL}\n@group(0) @binding(1) var<storage, read> guide: array<vec2f>;\n@group(0) @binding(2) var<storage, read> color: array<vec4f>;\n${guideBinding(guides, 3, 'albedoTex')}\n${guideBinding(guides, 4, 'normalTex')}\n@group(0) @binding(5) var<storage, read_write> outColor: array<vec4f>;\n@group(0) @binding(6) var<storage, read_write> outGuide: array<vec2f>;\n${guideBinding(guides, 7, 'shadowCatcherTex')}\n\nconst SEARCH_RADIUS = ${searchRadius};\nconst PATCH_RADIUS = ${patchRadius};\nconst PATCH_SIZE = ${patchSize};\n// Variance cancellation strength alpha of the paper.\nconst VARIANCE_CANCELLATION = 1.0;\n// Inverse squared feature bandwidths. Normals are encoded as n * 0.5 + 0.5.\nconst INV_ALBEDO_BANDWIDTH2 = ${(1 / albedoBandwidth ** 2).toFixed(6)};\nconst INV_COVERAGE_BANDWIDTH2 = ${(1 / JNLM_COVERAGE_BANDWIDTH ** 2).toFixed(6)};\nconst INV_NORMAL_BANDWIDTH2 = ${(1 / normalBandwidth ** 2).toFixed(6)};\n// Lower bound of the center weight, so a pixel unlike all of its neighbors\n// (a thin highlight) is kept rather than replaced by them.\nconst CENTER_WEIGHT_MIN = 0.01;\n\nfn pixelIndex(p: vec2i) -> u32 {\n  let q = clamp(p, vec2i(0), vec2i(params.size) - 1);\n  return u32(q.x) + u32(q.y) * params.size.x;\n}\n\nfn featureDistance(q: vec2i, centerAlbedo: vec4f, centerNormal: vec3f) -> f32 {\n  let albedo = ${loadGuide(guides, 'albedoTex', 'q')};\n  // Absolute differences in linear albedo erase low-contrast dark textures.\n  // Measure RGB contrast relative to both pixels; keep coverage absolute.\n  let albedoScale = max((albedo.rgb + centerAlbedo.rgb) * 0.5, vec3f(${JNLM_ALBEDO_FLOOR}));\n  let albedoDelta = (albedo.rgb - centerAlbedo.rgb) / albedoScale;\n  let coverageDelta = albedo.a - centerAlbedo.a;\n  let normalDelta = ${loadGuide(guides, 'normalTex', 'q')}.rgb - centerNormal;\n  return max(\n    dot(albedoDelta, albedoDelta) * INV_ALBEDO_BANDWIDTH2\n      + coverageDelta * coverageDelta * INV_COVERAGE_BANDWIDTH2,\n    dot(normalDelta, normalDelta) * INV_NORMAL_BANDWIDTH2\n  );\n}\n\nfn premultiplied(c: vec4f) -> vec4f {\n  return vec4f(c.rgb * c.a, c.a);\n}\n\n@compute @workgroup_size(${JNLM_WORKGROUP_SIZE}, ${JNLM_WORKGROUP_SIZE}, 1)\nfn main(@builtin(global_invocation_id) id: vec3u) {\n  let size = vec2i(params.size);\n  let p = vec2i(id.xy);\n  if (p.x >= size.x || p.y >= size.y) { return; }\n  let step = i32(params.dilation);\n  let centerAlbedo = ${loadGuide(guides, 'albedoTex', 'p')};\n  let centerNormalTexel = ${loadGuide(guides, 'normalTex', 'p')};\n  let centerNormal = centerNormalTexel.rgb;\n  let index = pixelIndex(p);\n  // Primary misses show the background directly. It has no features that\n  // protect its detail and no noise beyond antialiasing, so keep it.\n  if (centerNormalTexel.a <= 0.0 && ${loadGuide(guides, 'shadowCatcherTex', 'p')}.r <= 0.0) {\n    outColor[index] = color[index];\n    outGuide[index] = guide[index];\n    return;\n  }\n\n  var centerPatch: array<vec2f, PATCH_SIZE>;\n  var k = 0;\n  for (var dy = -PATCH_RADIUS; dy <= PATCH_RADIUS; dy++) {\n    for (var dx = -PATCH_RADIUS; dx <= PATCH_RADIUS; dx++) {\n      centerPatch[k] = guide[pixelIndex(p + vec2i(dx, dy) * step)];\n      k++;\n    }\n  }\n  let k2 = params.colorSensitivity * params.colorSensitivity;\n\n  var sum = vec4f(0.0);\n  let featureScale = max(centerAlbedo.rgb, vec3f(${JNLM_ALBEDO_FLOOR}));\n  var featureSum = vec3f(0.0);\n  var featureSquareSum = vec3f(0.0);\n  var featureColorSum = vec3f(0.0);\n  var guideSum = 0.0;\n  var guideVarianceSum = 0.0;\n  var weightSum = 0.0;\n  var maxWeight = 0.0;\n  for (var sy = -SEARCH_RADIUS; sy <= SEARCH_RADIUS; sy++) {\n    for (var sx = -SEARCH_RADIUS; sx <= SEARCH_RADIUS; sx++) {\n      let q = p + vec2i(sx, sy) * step;\n      if (\n        (sx == 0 && sy == 0) ||\n        q.x < 0 || q.y < 0 || q.x >= size.x || q.y >= size.y\n      ) { continue; }\n\n      let features = featureDistance(q, centerAlbedo, centerNormal);\n      if (features > 16.0) { continue; }\n\n      var colorDistance = 0.0;\n      k = 0;\n      for (var dy = -PATCH_RADIUS; dy <= PATCH_RADIUS; dy++) {\n        for (var dx = -PATCH_RADIUS; dx <= PATCH_RADIUS; dx++) {\n          let a = centerPatch[k];\n          let b = guide[pixelIndex(q + vec2i(dx, dy) * step)];\n          let delta = a.x - b.x;\n          colorDistance += (delta * delta\n              - VARIANCE_CANCELLATION * (a.y + min(a.y, b.y)))\n            / (1e-10 + k2 * (a.y + b.y));\n          k++;\n        }\n      }\n      colorDistance /= f32(PATCH_SIZE);\n\n      let weight = exp(-max(max(colorDistance, 0.0), features));\n      let index = pixelIndex(q);\n      let c = color[index];\n      let g = guide[index];\n      // Filter premultiplied color so coverage and color stay consistent.\n      let a = (${loadGuide(guides, 'albedoTex', 'q')}.rgb - centerAlbedo.rgb) / featureScale;\n      let value = premultiplied(c);\n      sum += value * weight;\n      featureSum += a * weight;\n      featureSquareSum += a * a * weight;\n      featureColorSum += a * value.rgb * weight;\n      guideSum += g.x * weight;\n      guideVarianceSum += g.y * weight * weight;\n      weightSum += weight;\n      maxWeight = max(maxWeight, weight);\n    }\n  }\n\n  // Weight the center like its most similar neighbor (Buades et al. 2005)\n  // instead of 1, so isolated outliers are not kept unfiltered.\n  let center = color[index];\n  let centerGuide = guide[index];\n  let centerWeight = max(maxWeight, CENTER_WEIGHT_MIN);\n  let totalWeight = weightSum + centerWeight;\n  let centerValue = premultiplied(center);\n  let mean = (sum + centerValue * centerWeight) / totalWeight;\n  // Fit each color channel against relative albedo with the same JNLM weights.\n  // Centering at this pixel avoids cancellation when the feature is constant.\n  let meanFeature = featureSum / totalWeight;\n  let featureVariance = max(featureSquareSum / totalWeight - meanFeature * meanFeature, vec3f(0.0));\n  let covariance = featureColorSum / totalWeight - meanFeature * mean.rgb;\n  // A 1% contrast ridge stabilizes flat guides. Bound the response between no\n  // texture dependence and multiplicative albedo modulation; specular color\n  // with an unrelated albedo falls back to the weighted mean.\n  let meanAlbedo = centerAlbedo.rgb + meanFeature * featureScale;\n  let maximumSlope = mean.rgb * featureScale / max(meanAlbedo, vec3f(${JNLM_ALBEDO_FLOOR}));\n  let slope = clamp(covariance / (featureVariance + vec3f(${JNLM_ALBEDO_REGULARIZATION})), vec3f(0.0), maximumSlope);\n  let corrected = mean.rgb - slope * meanFeature;\n  let filtered = vec4f(max(corrected, vec3f(0.0)), max(mean.a, 0.0));\n  let rgb = select(\n    center.rgb,\n    filtered.rgb / max(filtered.a, 1e-6),\n    filtered.a > 1e-6\n  );\n  outColor[index] = vec4f(rgb, filtered.a);\n  outGuide[index] = vec2f(\n    (guideSum + centerGuide.x * centerWeight) / totalWeight,\n    (guideVarianceSum + centerGuide.y * centerWeight * centerWeight)\n      / (totalWeight * totalWeight)\n  );\n}\n`;\n}\n\n// Compute presentation for hosts that consume the filtered buffer directly.\n// Use the same sRGB transfer as browser display composition.\nexport const jnlmEncodeWGSL = /* wgsl */ `\nstruct OutputParams { size: vec2u, premultiply: u32, reserved: u32 }\n@group(0) @binding(0) var<uniform> params: OutputParams;\n@group(0) @binding(1) var<storage, read> color: array<vec4f>;\n@group(0) @binding(2) var output: texture_storage_2d<rgba8unorm, write>;\n${linearToSrgbDefine}\n@compute @workgroup_size(${JNLM_WORKGROUP_SIZE}, ${JNLM_WORKGROUP_SIZE}, 1)\nfn main(@builtin(global_invocation_id) id: vec3u) {\n  if (any(id.xy >= params.size)) { return; }\n  let c = color[id.y * params.size.x + id.x];\n  var rgb = linearToSrgb(c.rgb);\n  if (params.premultiply != 0u) { rgb *= c.a; }\n  textureStore(output, id.xy, vec4f(rgb, c.a));\n}\n`;\n"]}