@meimu/mutracer 0.1.5 → 0.3.0

package/lib/renderer/WebGLFullQuadPass.js.map

@@ -1 +1 @@
-{"version":3,"file":"WebGLFullQuadPass.js","sourceRoot":"","sources":["../../src/renderer/WebGLFullQuadPass.ts"],"names":[],"mappings":"AAAA,OAAO,EAAC,oBAAoB,EAAC,MAAM,sBAAsB,CAAC;AAC1D,OAAO,EAAC,gBAAgB,EAAE,aAAa,EAAC,MAAM,oBAAoB,CAAC;AACnE,OAAO,EAAE,cAAc,EAAqB,MAAM,aAAa,CAAC;AAChE,OAAO,EACL,WAAW,EAEX,SAAS,EAET,oBAAoB,EACrB,MAAM,YAAY,CAAC;AAWpB;;;GAGG;AACH,MAAM,OAAO,iBAAiB;IACX,SAAS,CAAc;IAChC,SAAS,CAAuB;IACvB,aAAa,CAAiB;IAC9B,QAAQ,CAAM;IACd,YAAY,GAAG,IAAI,WAAW,EAAE,CAAC;IAC1C,eAAe,GAAG,EAA0B,CAAC;IACpC,cAAc,GAAG,IAAI,GAAG,EAAgC,CAAC;IAC1E,SAAS,CAAC,MAAc;QACtB,IAAI,OAAO,GAAG,IAAI,CAAC,cAAc,CAAC,GAAG,CAAC,MAAM,CAAC,CAAC;QAC9C,IAAI,CAAC,OAAO;YAAE,IAAI,CAAC,cAAc,CAAC,GAAG,CAAC,MAAM,EAAE,OAAO,GAAG,EAA0B,CAAC,CAAC;QACpF,IAAI,CAAC,eAAe,GAAG,OAAO,CAAC;IACjC,CAAC;IAEO,aAAa,GAAG,EAAwC,CAAC;IACzD,gBAAgB,GAAG,KAAK,CAAC;IACzB,cAAc,GAAG,CAAC,CAAC;IACnB,eAAe,GAAG,CAAC,CAAC;IACpB,MAAM,GAAG,CAAC,CAAC;IACX,OAAO,GAAG,CAAC,CAAC;IACZ,eAAe,GAAG,KAAK,CAAC;IAEhC,YACE,MAAc,EACd,QAAqB,EACrB,OAIC;QAED,IAAI,CAAC,SAAS,GAAG,QAAQ,CAAC;QAC1B,IAAI,CAAC,aAAa,GAAG,OAAO,CAAC,MAAM,CAAC;QACpC,IAAI,CAAC,SAAS,GAAG,IAAI,oBAAoB,CAAC,IAAI,CAAC,eAAe,CAAC,OAAO,CAAC,MAAM,CAAC,CAAC,CAAC;QAChF,IAAI,CAAC,QAAQ,GAAG,OAAO,CAAC,OAAO,IAAK,OAAO,CAAC,MAAM,CAAC,OAAe,CAAC;QACnE,IAAI,CAAC,eAAe,CAClB,MAAM,CAAC,WAAW,CAChB,IAAI,CAAC,QAAQ,CAAC,GAAG,CAAC,CAAC,IAAI,EAAE,EAAE,CAAC;YAC1B,IAAI;YACJ,OAAO,CAAC,YAAY,EAAE,CAAC,IAAI,CAAC,IAAI,EAAE;SACnC,CAAC,CACmC,CACxC,CAAC;IACJ,CAAC;IAED,eAAe;QACb,IAAI,IAAI,CAAC,gBAAgB;YAAE,OAAO;QAClC,IAAI,CAAC,gBAAgB,GAAG,IAAI,CAAC;QAC7B,IAAI,CAAC,SAAS,GAAG,IAAI,oBAAoB,CAAC,IAAI,CAAC,eAAe,CAC5D,IAAI,CAAC,aAAa,EAAE,IAAI,EAAE,IAAI,CAAC,SAAS,CAAC,QAAQ,EAAE,IAAI,CAAC,SAAS,CAAC,OAAO,CAC1E,CAAC,CAAC;IACL,CAAC;IAEO,eAAe,CACrB,MAAsB,EACtB,OAAO,GAAG,KAAK,EACf,QAAQ,GAAG,MAAM,CAAC,QAAQ,EAC1B,OAAO,GAAG,MAAM,CAAC,OAAO;QAExB,MAAM,QAAQ,GAAG,MAAM,CAAC,IAAI,CAAC,MAAM,CAAC,QAAQ,CAAC,CAAC,MAAM,CAAC,IAAI,CAAC,EAAE,CAAC,MAAM,CAAC,QAAQ,CAAC,IAAI,CAAC,CAAC,IAAI,KAAK,WAAW,CAAC,CAAC;QACzG,QAAQ,GAAG,EAAC,GAAG,QAAQ,EAAC,CAAC;QACzB,IAAI,OAAO;YAAE,KAAK,MAAM,IAAI,IAAI,QAAQ;gBAAE,QAAQ,CAAC,GAAG,IAAI,YAAY,CAAC,GAAG,EAAC,IAAI,EAAE,OAAO,EAAE,KAAK,EAAE,CAAC,CAAC,EAAE,CAAC,CAAC,EAAC,CAAC;QACzG,MAAM,OAAO,GAAG,IAAI,cAAc,CAAC,EAAC,IAAI,EAAE,MAAM,CAAC,IAAI,EAAE,OAAO,EAAE,MAAM,CAAC,OAAO,EAAE,OAAO,EAAE,QAAQ;YAC/F,QAAQ,EAAE,CAAC,EAAC,OAAO,EAAE,EAAE,EAAE,QAAQ,EAAE,EAAE,EAAE,IAAI,EAAE,EAAE,EAAE,SAAS,EAAE,CAAC,GAAG,EAAE,CAAC,oBAAoB,CAAC,QAAQ,EAAE,OAAO,CAAC,EAAE,GAAG,MAAM,CAAC,SAAS,CAAC,EAAC,CAAC;YAClI,IAAI,EAAE,MAAM,CAAC,IAAI,EAAC,CAAC,CAAC;QACtB,OAAO,CAAC,eAAe,GAAG,MAAM,CAAC,eAAe,CAAC;QACjD,OAAO,OAAO,CAAC;IACjB,CAAC;IAED,OAAO,CAAC,KAAa,EAAE,MAAc;QACnC,IAAI,IAAI,CAAC,MAAM,KAAK,KAAK,IAAI,IAAI,CAAC,OAAO,KAAK,MAAM;YAAE,OAAO;QAC7D,IAAI,CAAC,MAAM,GAAG,IAAI,CAAC,GAAG,CAAC,CAAC,EAAE,KAAK,CAAC,CAAC;QACjC,IAAI,CAAC,OAAO,GAAG,IAAI,CAAC,GAAG,CAAC,CAAC,EAAE,MAAM,CAAC,CAAC;QACnC,IAAI,CAAC,cAAc,GAAG,IAAI,CAAC,GAAG,CAAC,IAAI,CAAC,cAAc,EAAE,IAAI,CAAC,MAAM,CAAC,CAAC;QACjE,IAAI,CAAC,eAAe,GAAG,IAAI,CAAC,GAAG,CAAC,IAAI,CAAC,eAAe,EAAE,IAAI,CAAC,OAAO,CAAC,CAAC;QACpE,IAAI,CAAC,UAAU,CAAC,cAAc,EAAE,CAAC,IAAI,CAAC,MAAM,EAAE,IAAI,CAAC,OAAO,CAAC,CAAC,CAAC;IAC/D,CAAC;IAED,UAAU,CAAC,OAAqB;QAC9B,KAAK,MAAM,IAAI,IAAI,IAAI,CAAC,QAAQ;YAAE,IAAI,CAAC,OAAO,CAAC,QAAQ,CAAC,IAAI,CAAC,EAAE,CAAC;gBAC9D,IAAI,CAAC,eAAe,CAAC,IAAI,CAAC,EAAE,OAAO,CAAC,IAAI,CAAC,SAAS,CAAC,EAAE,CAAC,CAAC;gBAAC,OAAO,IAAI,CAAC,eAAe,CAAC,IAAI,CAAC,CAAC;YAC5F,CAAC;QACD,IAAI,CAAC,QAAQ,CAAC,MAAM,CAAC,CAAC,EAAE,IAAI,CAAC,QAAQ,CAAC,MAAM,EAAE,GAAG,OAAO,CAAC,CAAC;IAC5D,CAAC;IACD,eAAe,CAAC,MAA0C;QACxD,IAAI,CAAC,aAAa,GAAG,MAAM,CAAC;IAC9B,CAAC;IAED,iBAAiB,CAAC,cAAc,GAAG,IAAI;QACrC,IAAI,CAAC,eAAe,GAAG,cAAc,CAAC;IACxC,CAAC;IAED,UAAU,CAAC,KAAa,EAAE,IAAa;QACrC,OAAO,IAAI,CAAC,SAAS,CAAC,GAAG,CAAC,KAAK,EAAE,IAAI,CAAC,CAAC;IACzC,CAAC;IAED,WAAW,CAAC,MAA+B;QACzC,IAAI,CAAC,SAAS,CAAC,WAAW,CAAC,MAAM,CAAC,CAAC;IACrC,CAAC;IAED,UAAU,CAAC,KAAa;QACtB,OAAO,IAAI,CAAC,SAAS,CAAC,GAAG,CAAC,KAAK,CAAC,CAAC;IACnC,CAAC;IAED,UAAU,CACR,OAGC;QAED,IAAI,OAAO,GAAG,KAAK,CAAC;QACpB,KAAK,MAAM,CAAC,IAAI,EAAE,KAAK,CAAC,IAAI,MAAM,CAAC,OAAO,CAAC,OAAO,CAAC,EAAE,CAAC;YACpD,MAAM,QAAQ,GAAG,IAAI,CAAC,SAAS,CAAC,OAAO,CAAC,IAAI,CAAC,CAAC;YAC9C,IAAI,CAAC,KAAK,EAAE,CAAC;gBACX,IAAI,IAAI,IAAI,IAAI,CAAC,SAAS,CAAC,OAAO,EAAE,CAAC;oBACnC,IAAI,CAAC,SAAS,CAAC,YAAY,CAAC,IAAI,CAAC,CAAC;oBAClC,OAAO,GAAG,IAAI,CAAC;gBACjB,CAAC;YACH,CAAC;iBAAM,IAAI,QAAQ,KAAK,KAAK,EAAE,CAAC;gBAC9B,IAAI,CAAC,SAAS,CAAC,SAAS,CAAC,IAAI,EAAE,KAAK,CAAC,CAAC;gBACtC,OAAO,GAAG,IAAI,CAAC;YACjB,CAAC;QACH,CAAC;QACD,OAAO,OAAO,CAAC;IACjB,CAAC;IAEgB,gBAAgB,GAAG,IAAI,GAAG,EAAa,CAAC;IACzD,YAAY,CAAC,OAAkB,IAAI,IAAI,MAAM,CAAC,MAAM,CAAC,IAAI,CAAC,eAAe,CAAC,CAAC,QAAQ,CAAC,OAAO,CAAC;QAAE,IAAI,CAAC,gBAAgB,CAAC,GAAG,CAAC,OAAO,CAAC,CAAC,CAAC,CAAC;IACnI,aAAa,CAAC,OAAkB;QAC9B,IAAI,CAAC,IAAI,CAAC,gBAAgB,CAAC,MAAM,CAAC,OAAO,CAAC;YAAE,OAAO;QACnD,IAAI,CAAC,MAAM,CAAC,MAAM,CAAC,IAAI,CAAC,eAAe,CAAC,CAAC,QAAQ,CAAC,OAAO,CAAC;YAAE,OAAO,CAAC,OAAO,CAAC,IAAI,CAAC,SAAS,CAAC,EAAE,CAAC,CAAC;IACjG,CAAC;IAED,SAAS,CAAC,IAAO;QACf,OAAO,IAAI,CAAC,eAAe,CAAC,IAAI,CAAC,CAAC;IACpC,CAAC;IAEO,eAAe,CAAC,eAA+C;QACrE,MAAM,EAAE,GAAG,IAAI,CAAC,SAAS,CAAC,EAAE,CAAC;QAC7B,KAAK,MAAM,IAAI,IAAI,IAAI,CAAC,QAAQ,EAAE,CAAC;YACjC,MAAM,QAAQ,GAAG,eAAe,EAAE,CAAC,IAAI,CAAC,CAAC;YACzC,MAAM,OAAO,GAAG,IAAI,CAAC,eAAe,CAAC,IAAI,CAAC,CAAC;YAC3C,MAAM,cAAc,GAAG,OAAO,IAAI,IAAI,CAAC,gBAAgB,CAAC,GAAG,CAAC,OAAO,CAAC;gBAClE,CAAC,OAAO,CAAC,KAAK,KAAK,IAAI,CAAC,MAAM,IAAI,OAAO,CAAC,MAAM,KAAK,IAAI,CAAC,OAAO,CAAC,CAAC;YACrE,MAAM,OAAO,GACX,QAAQ;gBACR,CAAC,cAAc,CAAC,CAAC,CAAC,IAAI,SAAS,CAAC,IAAI,CAAC,aAAa,CAAC,IAAI,CAAC,CAAC,CAAC,CAAC,CAAC,SAAS,CAAC;gBACtE,IAAI,CAAC,eAAe,CAAC,IAAI,CAAC;gBAC1B,IAAI,SAAS,CAAC,IAAI,CAAC,aAAa,CAAC,IAAI,CAAC,CAAC,CAAC;YAC1C,OAAO,CAAC,aAAa,CACnB,EAAE,EACF,IAAI,CAAC,gBAAgB,CAAC,CAAC,CAAC,IAAI,CAAC,cAAc,CAAC,CAAC,CAAC,IAAI,CAAC,MAAM,EACzD,IAAI,CAAC,gBAAgB,CAAC,CAAC,CAAC,IAAI,CAAC,eAAe,CAAC,CAAC,CAAC,IAAI,CAAC,OAAO,EAC3D,IAAI,CAAC,aAAa,CAAC,IAAI,CAAC,CACzB,CAAC;YACF,IAAI,CAAC,eAAe,CAAC,IAAI,CAAC,GAAG,OAAO,CAAC;YACrC,gBAAgB,CAAC,OAAO,EAAE,IAAI,CAAC,MAAM,EAAE,IAAI,CAAC,OAAO,CAAC,CAAC;QACvD,CAAC;IACH,CAAC;IAED,UAAU,CACR,MAA4B,EAC5B,eAA+C,EAC/C,UAA4B,EAAE;QAE9B,MAAM,QAAQ,GAAG,IAAI,CAAC,SAAS,CAAC;QAChC,MAAM,EAAE,GAAG,QAAQ,CAAC,EAAE,CAAC;QAEvB,IAAI,IAAI,CAAC,eAAe,EAAE,CAAC;YACzB,EAAE,CAAC,eAAe,CAAC,EAAE,CAAC,WAAW,EAAE,IAAI,CAAC,CAAC;QAC3C,CAAC;aAAM,CAAC;YACN,IAAI,CAAC,eAAe,CAAC,eAAe,CAAC,CAAC;YACtC,IAAI,CAAC,YAAY,CAAC,IAAI,CACpB,QAAQ,EACR,IAAI,CAAC,QAAQ,CAAC,GAAG,CAAC,CAAC,IAAI,EAAE,EAAE,CAAC,IAAI,CAAC,eAAe,CAAC,IAAI,CAAC,CAAC,CACxD,CAAC;QACJ,CAAC;QAED,MAAM,QAAQ,GAAG,OAAO,CAAC,QAAQ,IAAI;YACnC,CAAC,EAAE,CAAC;YACJ,CAAC,EAAE,CAAC;YACJ,KAAK,EAAE,IAAI,CAAC,MAAM;YAClB,MAAM,EAAE,IAAI,CAAC,OAAO;SACrB,CAAC;QACF,QAAQ,CAAC,WAAW,CAClB,QAAQ,CAAC,CAAC,EACV,QAAQ,CAAC,CAAC,EACV,QAAQ,CAAC,KAAK,EACd,QAAQ,CAAC,MAAM,CAChB,CAAC;QAEF,uEAAuE;QACvE,wEAAwE;QACxE,IAAI,OAAO,CAAC,KAAK,EAAE,CAAC;YAClB,EAAE,CAAC,OAAO,CAAC,EAAE,CAAC,YAAY,CAAC,CAAC;YAC5B,EAAE,CAAC,UAAU,CAAC,CAAC,EAAE,CAAC,EAAE,CAAC,EAAE,CAAC,CAAC,CAAC;YAC1B,EAAE,CAAC,KAAK,CAAC,EAAE,CAAC,gBAAgB,CAAC,CAAC;QAChC,CAAC;QAED,IAAI,OAAO,CAAC,OAAO,EAAE,CAAC;YACpB,EAAE,CAAC,MAAM,CAAC,EAAE,CAAC,YAAY,CAAC,CAAC;YAC3B,EAAE,CAAC,OAAO,CACR,OAAO,CAAC,OAAO,CAAC,CAAC,EACjB,OAAO,CAAC,OAAO,CAAC,CAAC,EACjB,OAAO,CAAC,OAAO,CAAC,KAAK,EACrB,OAAO,CAAC,OAAO,CAAC,MAAM,CACvB,CAAC;QACJ,CAAC;aAAM,CAAC;YACN,EAAE,CAAC,OAAO,CAAC,EAAE,CAAC,YAAY,CAAC,CAAC;QAC9B,CAAC;QAED,IAAI,IAAI,CAAC,gBAAgB;YAAE,KAAK,MAAM,CAAC,IAAI,EAAE,OAAO,CAAC,IAAI,MAAM,CAAC,OAAO,CAAC,MAAM,CAA0B,EAAE,CAAC;gBACzG,MAAM,MAAM,GAAG,aAAa,CAAC,OAAO,CAAC,CAAC;gBACtC,IAAI,CAAC,UAAU,CAAC,GAAG,IAAI,YAAY,EAAE,CAAC,MAAM,CAAC,KAAK,EAAE,MAAM,CAAC,MAAM,CAAC,CAAC,CAAC;YACtE,CAAC;QACD,IAAI,CAAC,SAAS,CAAC,WAAW,CAAC,MAAM,CAAC,CAAC;QACnC,IAAI,CAAC,SAAS,CAAC,IAAI,CAAC,QAAQ,CAAC,CAAC;QAC9B,EAAE,CAAC,OAAO,CAAC,EAAE,CAAC,UAAU,CAAC,CAAC;QAC1B,IAAI,OAAO,CAAC,KAAK,EAAE,CAAC;YAClB,EAAE,CAAC,MAAM,CAAC,EAAE,CAAC,KAAK,CAAC,CAAC;YACpB,EAAE,CAAC,aAAa,CAAC,EAAE,CAAC,QAAQ,CAAC,CAAC;YAC9B,EAAE,CAAC,SAAS,CAAC,EAAE,CAAC,GAAG,EAAE,EAAE,CAAC,GAAG,CAAC,CAAC;QAC/B,CAAC;aAAM,CAAC;YACN,EAAE,CAAC,OAAO,CAAC,EAAE,CAAC,KAAK,CAAC,CAAC;QACvB,CAAC;QACD,EAAE,CAAC,UAAU,CAAC,EAAE,CAAC,SAAS,EAAE,CAAC,EAAE,CAAC,CAAC,CAAC;QAClC,EAAE,CAAC,OAAO,CAAC,EAAE,CAAC,YAAY,CAAC,CAAC;IAC9B,CAAC;IAED,aAAa,CAAC,MAAc;QAC1B,MAAM,OAAO,GAAG,IAAI,CAAC,cAAc,CAAC,GAAG,CAAC,MAAM,CAAC,CAAC;QAChD,IAAI,CAAC,OAAO;YAAE,OAAO;QACrB,IAAI,CAAC,cAAc,CAAC,MAAM,CAAC,MAAM,CAAC,CAAC;QACnC,KAAK,MAAM,OAAO,IAAK,MAAM,CAAC,MAAM,CAAC,OAAO,CAAiB;YAAE,OAAO,CAAC,OAAO,CAAC,IAAI,CAAC,SAAS,CAAC,EAAE,CAAC,CAAC;QAClG,IAAI,OAAO,KAAK,IAAI,CAAC,eAAe;YAAE,IAAI,CAAC,eAAe,GAAG,EAA0B,CAAC;IAC1F,CAAC;IAED,cAAc,CAAC,MAAe;QAC5B,MAAM,OAAO,GAAG,MAAM,CAAC,CAAC,CAAC,IAAI,CAAC,cAAc,CAAC,GAAG,CAAC,MAAM,CAAC,IAAI,EAAE,CAAC,CAAC,CAAC,IAAI,CAAC,eAAe,CAAC;QACtF,OAAO,KAAK,CAAC,IAAI,CAAC,IAAI,GAAG,CAAC,CAAC,GAAG,MAAM,CAAC,MAAM,CAAC,OAAO,CAAgB,EAAE,GAAG,IAAI,CAAC,gBAAgB,CAAC,CAAC,CAAC;aAC7F,MAAM,CAAC,CAAC,GAAG,EAAE,OAAO,EAAE,EAAE,CAAC,GAAG,GAAG,OAAO,CAAC,cAAc,EAAE,EAAE,CAAC,CAAC,CAAC;IACjE,CAAC;IAED,OAAO;QACL,MAAM,EAAE,GAAG,IAAI,CAAC,SAAS,CAAC,EAAE,CAAC;QAC5B,KAAK,CAAC,IAAI,CAAC,IAAI,GAAG,CAAC,CAAC,IAAI,CAAC,eAAe,EAAE,GAAG,IAAI,CAAC,cAAc,CAAC,MAAM,EAAE,CAAC,CAAC,CAAC,CAAC,OAAO,CAAC,OAAO,CAAC,EAAE,CAAC,MAAM,CAAC,MAAM,CAAC,OAAO,CAAC,CAAiB,CAAC,OAAO,CAAC,CAAC,OAAO,EAAE,EAAE,CAC3J,OAAO,CAAC,OAAO,CAAC,EAAE,CAAC,CACpB,CAAC;QACF,IAAI,CAAC,gBAAgB,CAAC,OAAO,CAAC,OAAO,CAAC,EAAE,CAAC,OAAO,CAAC,OAAO,CAAC,EAAE,CAAC,CAAC,CAAC;QAC9D,IAAI,CAAC,gBAAgB,CAAC,KAAK,EAAE,CAAC;QAC9B,IAAI,CAAC,YAAY,CAAC,OAAO,CAAC,IAAI,CAAC,SAAS,CAAC,CAAC;IAC5C,CAAC;CACF","sourcesContent":["import {scratchGLSLFunctions} from './scratchSampling.js';\nimport {setTextureRegion, textureRegion} from './textureRegion.js';\nimport { FragmentShader, ShaderDefineValue } from './shader.js';\nimport {\n  FrameBuffer,\n  WebGLDevice,\n  Texture2D,\n  TextureOptions,\n  WebGLProgramBindings\n} from './webgl.js';\n\nexport interface WebGLFullQuadPassOutput extends TextureOptions {}\n\nexport interface WebGLPassOptions {\n  clear?: boolean;\n  blend?: boolean;\n  viewport?: { x: number; y: number; width: number; height: number };\n  scissor?: { x: number; y: number; width: number; height: number };\n}\n\n/**\n * Explicit WebGL 2.0 render pass matching the lifecycle of WGPUFullQuadPass.\n * It owns its output textures and framebuffer; there is no render graph.\n */\nexport class WebGLFullQuadPass<I extends string, O extends string> {\n  private readonly _renderer: WebGLDevice;\n  private _bindings: WebGLProgramBindings;\n  private readonly _sourceShader: FragmentShader;\n  private readonly _outputs: O[];\n  private readonly _frameBuffer = new FrameBuffer();\n  private _outputTextures = {} as Record<O, Texture2D>;\n  private readonly _targetOutputs = new Map<object, Record<O, Texture2D>>();\n  useTarget(target: object) {\n    let outputs = this._targetOutputs.get(target);\n    if (!outputs) this._targetOutputs.set(target, outputs = {} as Record<O, Texture2D>);\n    this._outputTextures = outputs;\n  }\n\n  private _outputParams = {} as Record<O, WebGLFullQuadPassOutput>;\n  private _reserveCapacity = false;\n  private _capacityWidth = 1;\n  private _capacityHeight = 1;\n  private _width = 1;\n  private _height = 1;\n  private _renderToScreen = false;\n\n  constructor(\n    _label: string,\n    renderer: WebGLDevice,\n    options: {\n      shader: FragmentShader;\n      outputs?: O[];\n      outputParams?: Partial<Record<O, WebGLFullQuadPassOutput>>;\n    }\n  ) {\n    this._renderer = renderer;\n    this._sourceShader = options.shader;\n    this._bindings = new WebGLProgramBindings(this._samplingShader(options.shader));\n    this._outputs = options.outputs || (options.shader.outputs as O[]);\n    this.setOutputParams(\n      Object.fromEntries(\n        this._outputs.map((name) => [\n          name,\n          options.outputParams?.[name] || {}\n        ])\n      ) as Record<O, WebGLFullQuadPassOutput>\n    );\n  }\n\n  reserveCapacity() {\n    if (this._reserveCapacity) return;\n    this._reserveCapacity = true;\n    this._bindings = new WebGLProgramBindings(this._samplingShader(\n      this._sourceShader, true, this._bindings.uniforms, this._bindings.defines\n    ));\n  }\n\n  private _samplingShader(\n    shader: FragmentShader,\n    scratch = false,\n    uniforms = shader.uniforms,\n    defines = shader.defines\n  ) {\n    const textures = Object.keys(shader.uniforms).filter(name => shader.uniforms[name].type === 'sampler2D');\n    uniforms = {...uniforms};\n    if (scratch) for (const name of textures) uniforms[`${name}RegionSize`] = {type: 'ivec2', value: [1, 1]};\n    const sampled = new FragmentShader({name: shader.name, outputs: shader.outputs, defines, uniforms,\n      includes: [{defines: {}, uniforms: {}, main: '', functions: [() => scratchGLSLFunctions(textures, scratch), ...shader.functions]}],\n      main: shader.main});\n    sampled.parallelCompile = shader.parallelCompile;\n    return sampled;\n  }\n\n  setSize(width: number, height: number) {\n    if (this._width === width && this._height === height) return;\n    this._width = Math.max(1, width);\n    this._height = Math.max(1, height);\n    this._capacityWidth = Math.max(this._capacityWidth, this._width);\n    this._capacityHeight = Math.max(this._capacityHeight, this._height);\n    this.setUniform('viewportSize', [this._width, this._height]);\n  }\n\n  setOutputs(outputs: readonly O[]) {\n    for (const name of this._outputs) if (!outputs.includes(name)) {\n      this._outputTextures[name]?.dispose(this._renderer.gl); delete this._outputTextures[name];\n    }\n    this._outputs.splice(0, this._outputs.length, ...outputs);\n  }\n  setOutputParams(params: Record<O, WebGLFullQuadPassOutput>) {\n    this._outputParams = params;\n  }\n\n  setRenderToScreen(renderToScreen = true) {\n    this._renderToScreen = renderToScreen;\n  }\n\n  setUniform(label: string, data: unknown) {\n    return this._bindings.set(label, data);\n  }\n\n  setUniforms(values: Record<string, unknown>) {\n    this._bindings.setUniforms(values);\n  }\n\n  getUniform(label: string) {\n    return this._bindings.get(label);\n  }\n\n  setDefines(\n    defines: Record<\n      string,\n      ShaderDefineValue | undefined\n    >\n  ) {\n    let changed = false;\n    for (const [name, value] of Object.entries(defines)) {\n      const oldValue = this._bindings.defines[name];\n      if (!value) {\n        if (name in this._bindings.defines) {\n          this._bindings.deleteDefine(name);\n          changed = true;\n        }\n      } else if (oldValue !== value) {\n        this._bindings.setDefine(name, value);\n        changed = true;\n      }\n    }\n    return changed;\n  }\n\n  private readonly _retainedOutputs = new Set<Texture2D>();\n  retainOutput(texture: Texture2D) { if (Object.values(this._outputTextures).includes(texture)) this._retainedOutputs.add(texture); }\n  releaseOutput(texture: Texture2D) {\n    if (!this._retainedOutputs.delete(texture)) return;\n    if (!Object.values(this._outputTextures).includes(texture)) texture.dispose(this._renderer.gl);\n  }\n\n  getOutput(name: O) {\n    return this._outputTextures[name];\n  }\n\n  private _prepareOutputs(explicitOutputs?: Partial<Record<O, Texture2D>>) {\n    const gl = this._renderer.gl;\n    for (const name of this._outputs) {\n      const explicit = explicitOutputs?.[name];\n      const current = this._outputTextures[name];\n      const retainedResize = current && this._retainedOutputs.has(current) &&\n        (current.width !== this._width || current.height !== this._height);\n      const texture =\n        explicit ||\n        (retainedResize ? new Texture2D(this._outputParams[name]) : undefined) ||\n        this._outputTextures[name] ||\n        new Texture2D(this._outputParams[name]);\n      texture.ensureStorage(\n        gl,\n        this._reserveCapacity ? this._capacityWidth : this._width,\n        this._reserveCapacity ? this._capacityHeight : this._height,\n        this._outputParams[name]\n      );\n      this._outputTextures[name] = texture;\n      setTextureRegion(texture, this._width, this._height);\n    }\n  }\n\n  createPass(\n    inputs: Record<I, Texture2D>,\n    explicitOutputs?: Partial<Record<O, Texture2D>>,\n    options: WebGLPassOptions = {}\n  ) {\n    const renderer = this._renderer;\n    const gl = renderer.gl;\n\n    if (this._renderToScreen) {\n      gl.bindFramebuffer(gl.FRAMEBUFFER, null);\n    } else {\n      this._prepareOutputs(explicitOutputs);\n      this._frameBuffer.bind(\n        renderer,\n        this._outputs.map((name) => this._outputTextures[name])\n      );\n    }\n\n    const viewport = options.viewport || {\n      x: 0,\n      y: 0,\n      width: this._width,\n      height: this._height\n    };\n    renderer.setViewport(\n      viewport.x,\n      viewport.y,\n      viewport.width,\n      viewport.height\n    );\n\n    // Clear the full accumulation target before enabling the tile scissor.\n    // Otherwise every tile outside the first one retains stale camera data.\n    if (options.clear) {\n      gl.disable(gl.SCISSOR_TEST);\n      gl.clearColor(0, 0, 0, 0);\n      gl.clear(gl.COLOR_BUFFER_BIT);\n    }\n\n    if (options.scissor) {\n      gl.enable(gl.SCISSOR_TEST);\n      gl.scissor(\n        options.scissor.x,\n        options.scissor.y,\n        options.scissor.width,\n        options.scissor.height\n      );\n    } else {\n      gl.disable(gl.SCISSOR_TEST);\n    }\n\n    if (this._reserveCapacity) for (const [name, texture] of Object.entries(inputs) as [string, Texture2D][]) {\n      const region = textureRegion(texture);\n      this.setUniform(`${name}RegionSize`, [region.width, region.height]);\n    }\n    this._bindings.setUniforms(inputs);\n    this._bindings.bind(renderer);\n    gl.disable(gl.DEPTH_TEST);\n    if (options.blend) {\n      gl.enable(gl.BLEND);\n      gl.blendEquation(gl.FUNC_ADD);\n      gl.blendFunc(gl.ONE, gl.ONE);\n    } else {\n      gl.disable(gl.BLEND);\n    }\n    gl.drawArrays(gl.TRIANGLES, 0, 3);\n    gl.disable(gl.SCISSOR_TEST);\n  }\n\n  releaseTarget(target: object) {\n    const outputs = this._targetOutputs.get(target);\n    if (!outputs) return;\n    this._targetOutputs.delete(target);\n    for (const texture of (Object.values(outputs) as Texture2D[])) texture.dispose(this._renderer.gl);\n    if (outputs === this._outputTextures) this._outputTextures = {} as Record<O, Texture2D>;\n  }\n\n  getMemoryBytes(target?: object): number {\n    const outputs = target ? this._targetOutputs.get(target) ?? {} : this._outputTextures;\n    return Array.from(new Set([...Object.values(outputs) as Texture2D[], ...this._retainedOutputs]))\n      .reduce((sum, texture) => sum + texture.getMemoryBytes(), 0);\n  }\n\n  dispose() {\n    const gl = this._renderer.gl;\n    (Array.from(new Set([this._outputTextures, ...this._targetOutputs.values()])).flatMap(outputs => Object.values(outputs)) as Texture2D[]).forEach((texture) =>\n      texture.dispose(gl)\n    );\n    this._retainedOutputs.forEach(texture => texture.dispose(gl));\n    this._retainedOutputs.clear();\n    this._frameBuffer.dispose(this._renderer);\n  }\n}\n"]}
+{"version":3,"file":"WebGLFullQuadPass.js","sourceRoot":"","sources":["../../src/renderer/WebGLFullQuadPass.ts"],"names":[],"mappings":"AAAA,OAAO,EAAC,oBAAoB,EAAC,MAAM,sBAAsB,CAAC;AAC1D,OAAO,EAAC,gBAAgB,EAAE,aAAa,EAAC,MAAM,oBAAoB,CAAC;AACnE,OAAO,EAAE,cAAc,EAA0B,MAAM,aAAa,CAAC;AACrE,OAAO,EACL,WAAW,EAEX,SAAS,EAET,oBAAoB,EACrB,MAAM,YAAY,CAAC;AAWpB;;;GAGG;AACH,MAAM,OAAO,iBAAiB;IACX,SAAS,CAAc;IAChC,SAAS,CAAuB;IAChC,aAAa,CAAiB;IACrB,QAAQ,CAAM;IACd,YAAY,GAAG,IAAI,WAAW,EAAE,CAAC;IAC1C,eAAe,GAAG,EAA0B,CAAC;IACpC,cAAc,GAAG,IAAI,GAAG,EAAgC,CAAC;IAC1E,SAAS,CAAC,MAAc;QACtB,IAAI,OAAO,GAAG,IAAI,CAAC,cAAc,CAAC,GAAG,CAAC,MAAM,CAAC,CAAC;QAC9C,IAAI,CAAC,OAAO;YAAE,IAAI,CAAC,cAAc,CAAC,GAAG,CAAC,MAAM,EAAE,OAAO,GAAG,EAA0B,CAAC,CAAC;QACpF,IAAI,CAAC,eAAe,GAAG,OAAO,CAAC;IACjC,CAAC;IAEO,aAAa,GAAG,EAAwC,CAAC;IACzD,gBAAgB,GAAG,KAAK,CAAC;IACzB,cAAc,GAAG,CAAC,CAAC;IACnB,eAAe,GAAG,CAAC,CAAC;IACpB,MAAM,GAAG,CAAC,CAAC;IACX,OAAO,GAAG,CAAC,CAAC;IACZ,eAAe,GAAG,KAAK,CAAC;IAEhC,YACE,MAAc,EACd,QAAqB,EACrB,OAIC;QAED,IAAI,CAAC,SAAS,GAAG,QAAQ,CAAC;QAC1B,IAAI,CAAC,aAAa,GAAG,OAAO,CAAC,MAAM,CAAC;QACpC,IAAI,CAAC,SAAS,GAAG,IAAI,oBAAoB,CAAC,IAAI,CAAC,eAAe,CAAC,OAAO,CAAC,MAAM,CAAC,CAAC,CAAC;QAChF,IAAI,CAAC,QAAQ,GAAG,OAAO,CAAC,OAAO,IAAK,OAAO,CAAC,MAAM,CAAC,OAAe,CAAC;QACnE,IAAI,CAAC,eAAe,CAClB,MAAM,CAAC,WAAW,CAChB,IAAI,CAAC,QAAQ,CAAC,GAAG,CAAC,CAAC,IAAI,EAAE,EAAE,CAAC;YAC1B,IAAI;YACJ,OAAO,CAAC,YAAY,EAAE,CAAC,IAAI,CAAC,IAAI,EAAE;SACnC,CAAC,CACmC,CACxC,CAAC;IACJ,CAAC;IAED,SAAS,CAAC,MAAsB;QAC9B,IAAI,CAAC,aAAa,GAAG,MAAM,CAAC;QAC5B,IAAI,CAAC,SAAS,GAAG,IAAI,oBAAoB,CAAC,IAAI,CAAC,eAAe,CAC5D,MAAM,EAAE,IAAI,CAAC,gBAAgB,EAAE,EAAC,GAAG,MAAM,CAAC,QAAQ,EAAE,GAAG,IAAI,CAAC,SAAS,CAAC,QAAQ,EAAC,EAAE,IAAI,CAAC,SAAS,CAAC,OAAO,CACxG,CAAC,CAAC;IACL,CAAC;IAED,eAAe;QACb,IAAI,IAAI,CAAC,gBAAgB;YAAE,OAAO;QAClC,IAAI,CAAC,gBAAgB,GAAG,IAAI,CAAC;QAC7B,IAAI,CAAC,SAAS,GAAG,IAAI,oBAAoB,CAAC,IAAI,CAAC,eAAe,CAC5D,IAAI,CAAC,aAAa,EAAE,IAAI,EAAE,IAAI,CAAC,SAAS,CAAC,QAAQ,EAAE,IAAI,CAAC,SAAS,CAAC,OAAO,CAC1E,CAAC,CAAC;IACL,CAAC;IAEO,eAAe,CACrB,MAAsB,EACtB,OAAO,GAAG,KAAK,EACf,QAAQ,GAAG,MAAM,CAAC,QAAQ,EAC1B,OAAO,GAAG,MAAM,CAAC,OAAO;QAExB,MAAM,QAAQ,GAAG,MAAM,CAAC,IAAI,CAAC,MAAM,CAAC,QAAQ,CAAC,CAAC,MAAM,CAAC,IAAI,CAAC,EAAE,CAAC,MAAM,CAAC,QAAQ,CAAC,IAAI,CAAC,CAAC,IAAI,KAAK,WAAW,CAAC,CAAC;QACzG,QAAQ,GAAG,EAAC,GAAG,QAAQ,EAAC,CAAC;QACzB,IAAI,OAAO;YAAE,KAAK,MAAM,IAAI,IAAI,QAAQ;gBAAE,QAAQ,CAAC,GAAG,IAAI,YAAY,CAAC,GAAG,EAAC,IAAI,EAAE,OAAO,EAAE,KAAK,EAAE,CAAC,CAAC,EAAE,CAAC,CAAC,EAAC,CAAC;QACzG,MAAM,OAAO,GAAG,IAAI,cAAc,CAAC,EAAC,IAAI,EAAE,MAAM,CAAC,IAAI,EAAE,OAAO,EAAE,MAAM,CAAC,OAAO,EAAE,WAAW,EAAE,MAAM,CAAC,WAAW,EAAE,OAAO,EAAE,QAAQ;YAChI,QAAQ,EAAE,CAAC,EAAC,OAAO,EAAE,EAAE,EAAE,QAAQ,EAAE,EAAE,EAAE,IAAI,EAAE,EAAE,EAAE,SAAS,EAAE,CAAC,GAAG,EAAE,CAAC,oBAAoB,CAAC,QAAQ,EAAE,OAAO,CAAC,EAAE,GAAG,MAAM,CAAC,SAAS,CAAC,EAAC,CAAC;YAClI,IAAI,EAAE,MAAM,CAAC,IAAI,EAAC,CAAC,CAAC;QACtB,OAAO,CAAC,eAAe,GAAG,MAAM,CAAC,eAAe,CAAC;QACjD,OAAO,OAAO,CAAC;IACjB,CAAC;IAED,OAAO,CAAC,KAAa,EAAE,MAAc;QACnC,IAAI,IAAI,CAAC,MAAM,KAAK,KAAK,IAAI,IAAI,CAAC,OAAO,KAAK,MAAM;YAAE,OAAO;QAC7D,IAAI,CAAC,MAAM,GAAG,IAAI,CAAC,GAAG,CAAC,CAAC,EAAE,KAAK,CAAC,CAAC;QACjC,IAAI,CAAC,OAAO,GAAG,IAAI,CAAC,GAAG,CAAC,CAAC,EAAE,MAAM,CAAC,CAAC;QACnC,IAAI,CAAC,cAAc,GAAG,IAAI,CAAC,GAAG,CAAC,IAAI,CAAC,cAAc,EAAE,IAAI,CAAC,MAAM,CAAC,CAAC;QACjE,IAAI,CAAC,eAAe,GAAG,IAAI,CAAC,GAAG,CAAC,IAAI,CAAC,eAAe,EAAE,IAAI,CAAC,OAAO,CAAC,CAAC;QACpE,IAAI,CAAC,UAAU,CAAC,cAAc,EAAE,CAAC,IAAI,CAAC,MAAM,EAAE,IAAI,CAAC,OAAO,CAAC,CAAC,CAAC;IAC/D,CAAC;IAED,UAAU,CAAC,OAAqB;QAC9B,KAAK,MAAM,IAAI,IAAI,IAAI,CAAC,QAAQ;YAAE,IAAI,CAAC,OAAO,CAAC,QAAQ,CAAC,IAAI,CAAC,EAAE,CAAC;gBAC9D,IAAI,CAAC,eAAe,CAAC,IAAI,CAAC,EAAE,OAAO,CAAC,IAAI,CAAC,SAAS,CAAC,EAAE,CAAC,CAAC;gBAAC,OAAO,IAAI,CAAC,eAAe,CAAC,IAAI,CAAC,CAAC;YAC5F,CAAC;QACD,IAAI,CAAC,QAAQ,CAAC,MAAM,CAAC,CAAC,EAAE,IAAI,CAAC,QAAQ,CAAC,MAAM,EAAE,GAAG,OAAO,CAAC,CAAC;IAC5D,CAAC;IACD,eAAe,CAAC,MAA0C;QACxD,IAAI,CAAC,aAAa,GAAG,MAAM,CAAC;IAC9B,CAAC;IAED,iBAAiB,CAAC,cAAc,GAAG,IAAI;QACrC,IAAI,CAAC,eAAe,GAAG,cAAc,CAAC;IACxC,CAAC;IAED,UAAU,CAAC,KAAa,EAAE,IAAa;QACrC,OAAO,IAAI,CAAC,SAAS,CAAC,GAAG,CAAC,KAAK,EAAE,IAAI,CAAC,CAAC;IACzC,CAAC;IAED,WAAW,CAAC,MAA+B;QACzC,IAAI,CAAC,SAAS,CAAC,WAAW,CAAC,MAAM,CAAC,CAAC;IACrC,CAAC;IAED,UAAU,CAAC,KAAa;QACtB,OAAO,IAAI,CAAC,SAAS,CAAC,GAAG,CAAC,KAAK,CAAC,CAAC;IACnC,CAAC;IAED,UAAU,CACR,OAGC;QAED,IAAI,OAAO,GAAG,KAAK,CAAC;QACpB,KAAK,MAAM,CAAC,IAAI,EAAE,KAAK,CAAC,IAAI,MAAM,CAAC,OAAO,CAAC,OAAO,CAAC,EAAE,CAAC;YACpD,MAAM,QAAQ,GAAG,IAAI,CAAC,SAAS,CAAC,OAAO,CAAC,IAAI,CAAC,CAAC;YAC9C,IAAI,CAAC,KAAK,EAAE,CAAC;gBACX,IAAI,IAAI,IAAI,IAAI,CAAC,SAAS,CAAC,OAAO,EAAE,CAAC;oBACnC,IAAI,CAAC,SAAS,CAAC,YAAY,CAAC,IAAI,CAAC,CAAC;oBAClC,OAAO,GAAG,IAAI,CAAC;gBACjB,CAAC;YACH,CAAC;iBAAM,IAAI,QAAQ,KAAK,KAAK,EAAE,CAAC;gBAC9B,IAAI,CAAC,SAAS,CAAC,SAAS,CAAC,IAAI,EAAE,KAAK,CAAC,CAAC;gBACtC,OAAO,GAAG,IAAI,CAAC;YACjB,CAAC;QACH,CAAC;QACD,OAAO,OAAO,CAAC;IACjB,CAAC;IAEgB,gBAAgB,GAAG,IAAI,GAAG,EAAa,CAAC;IACzD,YAAY,CAAC,OAAkB,IAAI,IAAI,MAAM,CAAC,MAAM,CAAC,IAAI,CAAC,eAAe,CAAC,CAAC,QAAQ,CAAC,OAAO,CAAC;QAAE,IAAI,CAAC,gBAAgB,CAAC,GAAG,CAAC,OAAO,CAAC,CAAC,CAAC,CAAC;IACnI,aAAa,CAAC,OAAkB;QAC9B,IAAI,CAAC,IAAI,CAAC,gBAAgB,CAAC,MAAM,CAAC,OAAO,CAAC;YAAE,OAAO;QACnD,IAAI,CAAC,MAAM,CAAC,MAAM,CAAC,IAAI,CAAC,eAAe,CAAC,CAAC,QAAQ,CAAC,OAAO,CAAC;YAAE,OAAO,CAAC,OAAO,CAAC,IAAI,CAAC,SAAS,CAAC,EAAE,CAAC,CAAC;IACjG,CAAC;IAED,iFAAiF;IACjF,qBAAqB,CAAC,OAAsC;QAC1D,KAAK,MAAM,IAAI,IAAI,MAAM,CAAC,IAAI,CAAC,OAAO,CAAQ,EAAE,CAAC;YAC/C,IAAI,CAAC,eAAe,CAAC,IAAI,CAAC,EAAE,OAAO,CAAC,IAAI,CAAC,SAAS,CAAC,EAAE,CAAC,CAAC;YACvD,IAAI,CAAC,eAAe,CAAC,IAAI,CAAC,GAAG,OAAO,CAAC,IAAI,CAAE,CAAC;QAC9C,CAAC;IACH,CAAC;IAED,SAAS,CAAC,IAAO;QACf,OAAO,IAAI,CAAC,eAAe,CAAC,IAAI,CAAC,CAAC;IACpC,CAAC;IAEO,eAAe,CAAC,eAA+C;QACrE,MAAM,EAAE,GAAG,IAAI,CAAC,SAAS,CAAC,EAAE,CAAC;QAC7B,KAAK,MAAM,IAAI,IAAI,IAAI,CAAC,QAAQ,EAAE,CAAC;YACjC,MAAM,QAAQ,GAAG,eAAe,EAAE,CAAC,IAAI,CAAC,CAAC;YACzC,MAAM,OAAO,GAAG,IAAI,CAAC,eAAe,CAAC,IAAI,CAAC,CAAC;YAC3C,MAAM,cAAc,GAAG,OAAO,IAAI,IAAI,CAAC,gBAAgB,CAAC,GAAG,CAAC,OAAO,CAAC;gBAClE,CAAC,OAAO,CAAC,KAAK,KAAK,IAAI,CAAC,MAAM,IAAI,OAAO,CAAC,MAAM,KAAK,IAAI,CAAC,OAAO,CAAC,CAAC;YACrE,MAAM,OAAO,GACX,QAAQ;gBACR,CAAC,cAAc,CAAC,CAAC,CAAC,IAAI,SAAS,CAAC,IAAI,CAAC,aAAa,CAAC,IAAI,CAAC,CAAC,CAAC,CAAC,CAAC,SAAS,CAAC;gBACtE,IAAI,CAAC,eAAe,CAAC,IAAI,CAAC;gBAC1B,IAAI,SAAS,CAAC,IAAI,CAAC,aAAa,CAAC,IAAI,CAAC,CAAC,CAAC;YAC1C,OAAO,CAAC,aAAa,CACnB,EAAE,EACF,IAAI,CAAC,gBAAgB,CAAC,CAAC,CAAC,IAAI,CAAC,cAAc,CAAC,CAAC,CAAC,IAAI,CAAC,MAAM,EACzD,IAAI,CAAC,gBAAgB,CAAC,CAAC,CAAC,IAAI,CAAC,eAAe,CAAC,CAAC,CAAC,IAAI,CAAC,OAAO,EAC3D,IAAI,CAAC,aAAa,CAAC,IAAI,CAAC,CACzB,CAAC;YACF,IAAI,CAAC,eAAe,CAAC,IAAI,CAAC,GAAG,OAAO,CAAC;YACrC,gBAAgB,CAAC,OAAO,EAAE,IAAI,CAAC,MAAM,EAAE,IAAI,CAAC,OAAO,CAAC,CAAC;QACvD,CAAC;IACH,CAAC;IAED,cAAc,CAAC,MAA6B;QAC1C,IAAI,MAAM;YAAE,IAAI,CAAC,SAAS,CAAC,WAAW,CAAC,MAAM,CAAC,CAAC;QAC/C,OAAO,IAAI,CAAC,SAAS,CAAC,iBAAiB,CAAC,IAAI,CAAC,SAAS,CAAC,UAAU,CAAC,IAAI,CAAC,SAAS,CAAC,CAAC,CAAC;IACrF,CAAC;IAED,UAAU,CACR,MAA4B,EAC5B,eAA+C,EAC/C,UAA4B,EAAE;QAE9B,MAAM,QAAQ,GAAG,IAAI,CAAC,SAAS,CAAC;QAChC,MAAM,EAAE,GAAG,QAAQ,CAAC,EAAE,CAAC;QAEvB,IAAI,IAAI,CAAC,eAAe,EAAE,CAAC;YACzB,EAAE,CAAC,eAAe,CAAC,EAAE,CAAC,WAAW,EAAE,IAAI,CAAC,CAAC;QAC3C,CAAC;aAAM,CAAC;YACN,IAAI,CAAC,eAAe,CAAC,eAAe,CAAC,CAAC;YACtC,IAAI,CAAC,YAAY,CAAC,IAAI,CACpB,QAAQ,EACR,IAAI,CAAC,QAAQ,CAAC,GAAG,CAAC,CAAC,IAAI,EAAE,EAAE,CAAC,IAAI,CAAC,eAAe,CAAC,IAAI,CAAC,CAAC,CACxD,CAAC;QACJ,CAAC;QAED,MAAM,QAAQ,GAAG,OAAO,CAAC,QAAQ,IAAI;YACnC,CAAC,EAAE,CAAC;YACJ,CAAC,EAAE,CAAC;YACJ,KAAK,EAAE,IAAI,CAAC,MAAM;YAClB,MAAM,EAAE,IAAI,CAAC,OAAO;SACrB,CAAC;QACF,QAAQ,CAAC,WAAW,CAClB,QAAQ,CAAC,CAAC,EACV,QAAQ,CAAC,CAAC,EACV,QAAQ,CAAC,KAAK,EACd,QAAQ,CAAC,MAAM,CAChB,CAAC;QAEF,uEAAuE;QACvE,wEAAwE;QACxE,IAAI,OAAO,CAAC,KAAK,EAAE,CAAC;YAClB,EAAE,CAAC,OAAO,CAAC,EAAE,CAAC,YAAY,CAAC,CAAC;YAC5B,EAAE,CAAC,UAAU,CAAC,CAAC,EAAE,CAAC,EAAE,CAAC,EAAE,CAAC,CAAC,CAAC;YAC1B,EAAE,CAAC,KAAK,CAAC,EAAE,CAAC,gBAAgB,CAAC,CAAC;QAChC,CAAC;QAED,IAAI,OAAO,CAAC,OAAO,EAAE,CAAC;YACpB,EAAE,CAAC,MAAM,CAAC,EAAE,CAAC,YAAY,CAAC,CAAC;YAC3B,EAAE,CAAC,OAAO,CACR,OAAO,CAAC,OAAO,CAAC,CAAC,EACjB,OAAO,CAAC,OAAO,CAAC,CAAC,EACjB,OAAO,CAAC,OAAO,CAAC,KAAK,EACrB,OAAO,CAAC,OAAO,CAAC,MAAM,CACvB,CAAC;QACJ,CAAC;aAAM,CAAC;YACN,EAAE,CAAC,OAAO,CAAC,EAAE,CAAC,YAAY,CAAC,CAAC;QAC9B,CAAC;QAED,IAAI,IAAI,CAAC,gBAAgB;YAAE,KAAK,MAAM,CAAC,IAAI,EAAE,OAAO,CAAC,IAAI,MAAM,CAAC,OAAO,CAAC,MAAM,CAA0B,EAAE,CAAC;gBACzG,MAAM,MAAM,GAAG,aAAa,CAAC,OAAO,CAAC,CAAC;gBACtC,IAAI,CAAC,UAAU,CAAC,GAAG,IAAI,YAAY,EAAE,CAAC,MAAM,CAAC,KAAK,EAAE,MAAM,CAAC,MAAM,CAAC,CAAC,CAAC;YACtE,CAAC;QACD,IAAI,CAAC,SAAS,CAAC,WAAW,CAAC,MAAM,CAAC,CAAC;QACnC,IAAI,CAAC,IAAI,CAAC,SAAS,CAAC,IAAI,CAAC,QAAQ,CAAC,EAAE,CAAC;YACnC,EAAE,CAAC,OAAO,CAAC,EAAE,CAAC,YAAY,CAAC,CAAC;YAC5B,OAAO,KAAK,CAAC;QACf,CAAC;QACD,EAAE,CAAC,OAAO,CAAC,EAAE,CAAC,UAAU,CAAC,CAAC;QAC1B,IAAI,OAAO,CAAC,KAAK,EAAE,CAAC;YAClB,EAAE,CAAC,MAAM,CAAC,EAAE,CAAC,KAAK,CAAC,CAAC;YACpB,EAAE,CAAC,aAAa,CAAC,EAAE,CAAC,QAAQ,CAAC,CAAC;YAC9B,EAAE,CAAC,SAAS,CAAC,EAAE,CAAC,GAAG,EAAE,EAAE,CAAC,GAAG,CAAC,CAAC;QAC/B,CAAC;aAAM,CAAC;YACN,EAAE,CAAC,OAAO,CAAC,EAAE,CAAC,KAAK,CAAC,CAAC;QACvB,CAAC;QACD,EAAE,CAAC,UAAU,CAAC,EAAE,CAAC,SAAS,EAAE,CAAC,EAAE,CAAC,CAAC,CAAC;QAClC,EAAE,CAAC,OAAO,CAAC,EAAE,CAAC,YAAY,CAAC,CAAC;QAC5B,OAAO,IAAI,CAAC;IACd,CAAC;IAED,aAAa,CAAC,MAAc;QAC1B,MAAM,OAAO,GAAG,IAAI,CAAC,cAAc,CAAC,GAAG,CAAC,MAAM,CAAC,CAAC;QAChD,IAAI,CAAC,OAAO;YAAE,OAAO;QACrB,IAAI,CAAC,cAAc,CAAC,MAAM,CAAC,MAAM,CAAC,CAAC;QACnC,KAAK,MAAM,OAAO,IAAK,MAAM,CAAC,MAAM,CAAC,OAAO,CAAiB;YAAE,OAAO,CAAC,OAAO,CAAC,IAAI,CAAC,SAAS,CAAC,EAAE,CAAC,CAAC;QAClG,IAAI,OAAO,KAAK,IAAI,CAAC,eAAe;YAAE,IAAI,CAAC,eAAe,GAAG,EAA0B,CAAC;IAC1F,CAAC;IAED,cAAc,CAAC,MAAe;QAC5B,MAAM,OAAO,GAAG,MAAM,CAAC,CAAC,CAAC,IAAI,CAAC,cAAc,CAAC,GAAG,CAAC,MAAM,CAAC,IAAI,EAAE,CAAC,CAAC,CAAC,IAAI,CAAC,eAAe,CAAC;QACtF,OAAO,KAAK,CAAC,IAAI,CAAC,IAAI,GAAG,CAAC,CAAC,GAAG,MAAM,CAAC,MAAM,CAAC,OAAO,CAAgB,EAAE,GAAG,IAAI,CAAC,gBAAgB,CAAC,CAAC,CAAC;aAC7F,MAAM,CAAC,CAAC,GAAG,EAAE,OAAO,EAAE,EAAE,CAAC,GAAG,GAAG,OAAO,CAAC,cAAc,EAAE,EAAE,CAAC,CAAC,CAAC;IACjE,CAAC;IAED,OAAO;QACL,MAAM,EAAE,GAAG,IAAI,CAAC,SAAS,CAAC,EAAE,CAAC;QAC5B,KAAK,CAAC,IAAI,CAAC,IAAI,GAAG,CAAC,CAAC,IAAI,CAAC,eAAe,EAAE,GAAG,IAAI,CAAC,cAAc,CAAC,MAAM,EAAE,CAAC,CAAC,CAAC,CAAC,OAAO,CAAC,OAAO,CAAC,EAAE,CAAC,MAAM,CAAC,MAAM,CAAC,OAAO,CAAC,CAAiB,CAAC,OAAO,CAAC,CAAC,OAAO,EAAE,EAAE,CAC3J,OAAO,CAAC,OAAO,CAAC,EAAE,CAAC,CACpB,CAAC;QACF,IAAI,CAAC,gBAAgB,CAAC,OAAO,CAAC,OAAO,CAAC,EAAE,CAAC,OAAO,CAAC,OAAO,CAAC,EAAE,CAAC,CAAC,CAAC;QAC9D,IAAI,CAAC,gBAAgB,CAAC,KAAK,EAAE,CAAC;QAC9B,IAAI,CAAC,YAAY,CAAC,OAAO,CAAC,IAAI,CAAC,SAAS,CAAC,CAAC;IAC5C,CAAC;CACF","sourcesContent":["import {scratchGLSLFunctions} from './scratchSampling.js';\nimport {setTextureRegion, textureRegion} from './textureRegion.js';\nimport { FragmentShader, type ShaderDefineValue } from './shader.js';\nimport {\n  FrameBuffer,\n  WebGLDevice,\n  Texture2D,\n  type TextureOptions,\n  WebGLProgramBindings\n} from './webgl.js';\n\nexport interface WebGLFullQuadPassOutput extends TextureOptions {}\n\nexport interface WebGLPassOptions {\n  clear?: boolean;\n  blend?: boolean;\n  viewport?: { x: number; y: number; width: number; height: number };\n  scissor?: { x: number; y: number; width: number; height: number };\n}\n\n/**\n * Explicit WebGL 2.0 render pass matching the lifecycle of WGPUFullQuadPass.\n * It owns its output textures and framebuffer; there is no render graph.\n */\nexport class WebGLFullQuadPass<I extends string, O extends string> {\n  private readonly _renderer: WebGLDevice;\n  private _bindings: WebGLProgramBindings;\n  private _sourceShader: FragmentShader;\n  private readonly _outputs: O[];\n  private readonly _frameBuffer = new FrameBuffer();\n  private _outputTextures = {} as Record<O, Texture2D>;\n  private readonly _targetOutputs = new Map<object, Record<O, Texture2D>>();\n  useTarget(target: object) {\n    let outputs = this._targetOutputs.get(target);\n    if (!outputs) this._targetOutputs.set(target, outputs = {} as Record<O, Texture2D>);\n    this._outputTextures = outputs;\n  }\n\n  private _outputParams = {} as Record<O, WebGLFullQuadPassOutput>;\n  private _reserveCapacity = false;\n  private _capacityWidth = 1;\n  private _capacityHeight = 1;\n  private _width = 1;\n  private _height = 1;\n  private _renderToScreen = false;\n\n  constructor(\n    _label: string,\n    renderer: WebGLDevice,\n    options: {\n      shader: FragmentShader;\n      outputs?: O[];\n      outputParams?: Partial<Record<O, WebGLFullQuadPassOutput>>;\n    }\n  ) {\n    this._renderer = renderer;\n    this._sourceShader = options.shader;\n    this._bindings = new WebGLProgramBindings(this._samplingShader(options.shader));\n    this._outputs = options.outputs || (options.shader.outputs as O[]);\n    this.setOutputParams(\n      Object.fromEntries(\n        this._outputs.map((name) => [\n          name,\n          options.outputParams?.[name] || {}\n        ])\n      ) as Record<O, WebGLFullQuadPassOutput>\n    );\n  }\n\n  setShader(shader: FragmentShader) {\n    this._sourceShader = shader;\n    this._bindings = new WebGLProgramBindings(this._samplingShader(\n      shader, this._reserveCapacity, {...shader.uniforms, ...this._bindings.uniforms}, this._bindings.defines\n    ));\n  }\n\n  reserveCapacity() {\n    if (this._reserveCapacity) return;\n    this._reserveCapacity = true;\n    this._bindings = new WebGLProgramBindings(this._samplingShader(\n      this._sourceShader, true, this._bindings.uniforms, this._bindings.defines\n    ));\n  }\n\n  private _samplingShader(\n    shader: FragmentShader,\n    scratch = false,\n    uniforms = shader.uniforms,\n    defines = shader.defines\n  ) {\n    const textures = Object.keys(shader.uniforms).filter(name => shader.uniforms[name].type === 'sampler2D');\n    uniforms = {...uniforms};\n    if (scratch) for (const name of textures) uniforms[`${name}RegionSize`] = {type: 'ivec2', value: [1, 1]};\n    const sampled = new FragmentShader({name: shader.name, outputs: shader.outputs, outputTypes: shader.outputTypes, defines, uniforms,\n      includes: [{defines: {}, uniforms: {}, main: '', functions: [() => scratchGLSLFunctions(textures, scratch), ...shader.functions]}],\n      main: shader.main});\n    sampled.parallelCompile = shader.parallelCompile;\n    return sampled;\n  }\n\n  setSize(width: number, height: number) {\n    if (this._width === width && this._height === height) return;\n    this._width = Math.max(1, width);\n    this._height = Math.max(1, height);\n    this._capacityWidth = Math.max(this._capacityWidth, this._width);\n    this._capacityHeight = Math.max(this._capacityHeight, this._height);\n    this.setUniform('viewportSize', [this._width, this._height]);\n  }\n\n  setOutputs(outputs: readonly O[]) {\n    for (const name of this._outputs) if (!outputs.includes(name)) {\n      this._outputTextures[name]?.dispose(this._renderer.gl); delete this._outputTextures[name];\n    }\n    this._outputs.splice(0, this._outputs.length, ...outputs);\n  }\n  setOutputParams(params: Record<O, WebGLFullQuadPassOutput>) {\n    this._outputParams = params;\n  }\n\n  setRenderToScreen(renderToScreen = true) {\n    this._renderToScreen = renderToScreen;\n  }\n\n  setUniform(label: string, data: unknown) {\n    return this._bindings.set(label, data);\n  }\n\n  setUniforms(values: Record<string, unknown>) {\n    this._bindings.setUniforms(values);\n  }\n\n  getUniform(label: string) {\n    return this._bindings.get(label);\n  }\n\n  setDefines(\n    defines: Record<\n      string,\n      ShaderDefineValue | undefined\n    >\n  ) {\n    let changed = false;\n    for (const [name, value] of Object.entries(defines)) {\n      const oldValue = this._bindings.defines[name];\n      if (!value) {\n        if (name in this._bindings.defines) {\n          this._bindings.deleteDefine(name);\n          changed = true;\n        }\n      } else if (oldValue !== value) {\n        this._bindings.setDefine(name, value);\n        changed = true;\n      }\n    }\n    return changed;\n  }\n\n  private readonly _retainedOutputs = new Set<Texture2D>();\n  retainOutput(texture: Texture2D) { if (Object.values(this._outputTextures).includes(texture)) this._retainedOutputs.add(texture); }\n  releaseOutput(texture: Texture2D) {\n    if (!this._retainedOutputs.delete(texture)) return;\n    if (!Object.values(this._outputTextures).includes(texture)) texture.dispose(this._renderer.gl);\n  }\n\n  // Internal fixed-accumulation loader; transfer ownership to the selected target.\n  loadDiagnosticOutputs(outputs: Partial<Record<O, Texture2D>>) {\n    for (const name of Object.keys(outputs) as O[]) {\n      this._outputTextures[name]?.dispose(this._renderer.gl);\n      this._outputTextures[name] = outputs[name]!;\n    }\n  }\n\n  getOutput(name: O) {\n    return this._outputTextures[name];\n  }\n\n  private _prepareOutputs(explicitOutputs?: Partial<Record<O, Texture2D>>) {\n    const gl = this._renderer.gl;\n    for (const name of this._outputs) {\n      const explicit = explicitOutputs?.[name];\n      const current = this._outputTextures[name];\n      const retainedResize = current && this._retainedOutputs.has(current) &&\n        (current.width !== this._width || current.height !== this._height);\n      const texture =\n        explicit ||\n        (retainedResize ? new Texture2D(this._outputParams[name]) : undefined) ||\n        this._outputTextures[name] ||\n        new Texture2D(this._outputParams[name]);\n      texture.ensureStorage(\n        gl,\n        this._reserveCapacity ? this._capacityWidth : this._width,\n        this._reserveCapacity ? this._capacityHeight : this._height,\n        this._outputParams[name]\n      );\n      this._outputTextures[name] = texture;\n      setTextureRegion(texture, this._width, this._height);\n    }\n  }\n\n  isProgramReady(inputs?: Record<I, Texture2D>) {\n    if (inputs) this._bindings.setUniforms(inputs);\n    return this._renderer.isProgramCompiled(this._bindings.getProgram(this._renderer));\n  }\n\n  createPass(\n    inputs: Record<I, Texture2D>,\n    explicitOutputs?: Partial<Record<O, Texture2D>>,\n    options: WebGLPassOptions = {}\n  ) {\n    const renderer = this._renderer;\n    const gl = renderer.gl;\n\n    if (this._renderToScreen) {\n      gl.bindFramebuffer(gl.FRAMEBUFFER, null);\n    } else {\n      this._prepareOutputs(explicitOutputs);\n      this._frameBuffer.bind(\n        renderer,\n        this._outputs.map((name) => this._outputTextures[name])\n      );\n    }\n\n    const viewport = options.viewport || {\n      x: 0,\n      y: 0,\n      width: this._width,\n      height: this._height\n    };\n    renderer.setViewport(\n      viewport.x,\n      viewport.y,\n      viewport.width,\n      viewport.height\n    );\n\n    // Clear the full accumulation target before enabling the tile scissor.\n    // Otherwise every tile outside the first one retains stale camera data.\n    if (options.clear) {\n      gl.disable(gl.SCISSOR_TEST);\n      gl.clearColor(0, 0, 0, 0);\n      gl.clear(gl.COLOR_BUFFER_BIT);\n    }\n\n    if (options.scissor) {\n      gl.enable(gl.SCISSOR_TEST);\n      gl.scissor(\n        options.scissor.x,\n        options.scissor.y,\n        options.scissor.width,\n        options.scissor.height\n      );\n    } else {\n      gl.disable(gl.SCISSOR_TEST);\n    }\n\n    if (this._reserveCapacity) for (const [name, texture] of Object.entries(inputs) as [string, Texture2D][]) {\n      const region = textureRegion(texture);\n      this.setUniform(`${name}RegionSize`, [region.width, region.height]);\n    }\n    this._bindings.setUniforms(inputs);\n    if (!this._bindings.bind(renderer)) {\n      gl.disable(gl.SCISSOR_TEST);\n      return false;\n    }\n    gl.disable(gl.DEPTH_TEST);\n    if (options.blend) {\n      gl.enable(gl.BLEND);\n      gl.blendEquation(gl.FUNC_ADD);\n      gl.blendFunc(gl.ONE, gl.ONE);\n    } else {\n      gl.disable(gl.BLEND);\n    }\n    gl.drawArrays(gl.TRIANGLES, 0, 3);\n    gl.disable(gl.SCISSOR_TEST);\n    return true;\n  }\n\n  releaseTarget(target: object) {\n    const outputs = this._targetOutputs.get(target);\n    if (!outputs) return;\n    this._targetOutputs.delete(target);\n    for (const texture of (Object.values(outputs) as Texture2D[])) texture.dispose(this._renderer.gl);\n    if (outputs === this._outputTextures) this._outputTextures = {} as Record<O, Texture2D>;\n  }\n\n  getMemoryBytes(target?: object): number {\n    const outputs = target ? this._targetOutputs.get(target) ?? {} : this._outputTextures;\n    return Array.from(new Set([...Object.values(outputs) as Texture2D[], ...this._retainedOutputs]))\n      .reduce((sum, texture) => sum + texture.getMemoryBytes(), 0);\n  }\n\n  dispose() {\n    const gl = this._renderer.gl;\n    (Array.from(new Set([this._outputTextures, ...this._targetOutputs.values()])).flatMap(outputs => Object.values(outputs)) as Texture2D[]).forEach((texture) =>\n      texture.dispose(gl)\n    );\n    this._retainedOutputs.forEach(texture => texture.dispose(gl));\n    this._retainedOutputs.clear();\n    this._frameBuffer.dispose(this._renderer);\n  }\n}\n"]}

package/lib/renderer/WebGLJNLMDenoiser.d.ts

@@ -23,7 +23,7 @@ export declare class WebGLJNLMDenoisePass {
     private _filterPasses;
     constructor(renderer: WebGLDevice);
     /** Runs the denoiser and returns the display-encoded result. */
-    render(inputs: WebGLJNLMInputs): Texture2D;
+    render(inputs: WebGLJNLMInputs, trace?: string[]): Texture2D;
     reserveCapacity(): void;
     retainOutput(texture: Texture2D): void;
     releaseOutput(texture: Texture2D): void;

package/lib/renderer/WebGLJNLMDenoiser.js

@@ -45,27 +45,27 @@ void main() {
 ivec2 p = ivec2(gl_FragCoord.xy);
 ivec2 size = rawGuideTexRegion();
 vec2 center = load_rawGuideTex(p, 0).xy;
-float lu = 0.0;
-float jc = 0.0;
-float mY = 0.0;
-float iX = 0.0;
+float lM = 0.0;
+float je = 0.0;
+float nM = 0.0;
+float iV = 0.0;
 for (int dy = -1; dy <= 1; dy++) {
 for (int dx = -1; dx <= 1; dx++) {
 vec2 value = load_rawGuideTex(clamp(p + ivec2(dx, dy), ivec2(0), size - 1), 0).xy;
-mY += value.x;
-iX += value.x * value.x;
+nM += value.x;
+iV += value.x * value.x;
 if (value.y >= 0.0) {
-lu += value.y;
-jc += 1.0;
+lM += value.y;
+je += 1.0;
 }
 }
 }
 float variance;
 if (center.y >= 0.0) {
-variance = lu / jc;
+variance = lM / je;
 } else {
-float lt = mY / 9.0;
-variance = max(iX / 9.0 - lt * lt, 0.0) * 9.0 / 8.0;
+float lL = nM / 9.0;
+variance = max(iV / 9.0 - lL * lL, 0.0) * 9.0 / 8.0;
 }
 out_guide = vec4(center.x, variance, 0.0, 1.0);
 }`
@@ -105,68 +105,70 @@ const filterFragment = new FragmentShader({
 
 ivec2 imageSize;
 
-vec2 nH(ivec2 p) {
+vec2 loadGuide(ivec2 p) {
 return load_guideTex(clamp(p, ivec2(0), imageSize - 1), 0).xy;
 }
 
 
-vec4 rL(ivec2 p) {
+vec4 sI(ivec2 p) {
 vec4 c = load_colorTex(p, 0);
-if (decodeInput) c.rgb = pow(max(c.rgb, 0.0), vec3(2.2));
+if (decodeInput) c.rgb = mix(pow((max(c.rgb, 0.0) + 0.055) / 1.055, vec3(2.4)),
+c.rgb / 12.92, lessThanEqual(c.rgb, vec3(0.04045)));
 return vec4(c.rgb * c.a, c.a);
 }
 
-float lw(ivec2 q, vec4 bs, vec3 gT) {
+float lP(ivec2 q, vec4 bo, vec3 gO) {
 vec4 albedo = load_albedoTex(q, 0);
 
 
-vec3 pf = max((albedo.rgb + bs.rgb) * 0.5, vec3(${JNLM_ALBEDO_FLOOR}));
-vec3 ld = (albedo.rgb - bs.rgb) / pf;
-float iU = albedo.a - bs.a;
-vec3 lp = load_normalTex(q, 0).rgb - gT;
+vec3 pY = max((albedo.rgb + bo.rgb) * 0.5, vec3(${JNLM_ALBEDO_FLOOR}));
+vec3 lu = (albedo.rgb - bo.rgb) / pY;
+float iQ = albedo.a - bo.a;
+vec3 lG = load_normalTex(q, 0).rgb - gO;
 return max(
-dot(ld, ld) * INV_ALBEDO_BANDWIDTH2
-+ iU * iU * INV_COVERAGE_BANDWIDTH2,
-dot(lp, lp) * INV_NORMAL_BANDWIDTH2
+dot(lu, lu) * INV_ALBEDO_BANDWIDTH2
++ iQ * iQ * INV_COVERAGE_BANDWIDTH2,
+dot(lG, lG) * INV_NORMAL_BANDWIDTH2
 );
 }
 
 void main() {
 imageSize = guideTexRegion();
 ivec2 p = ivec2(gl_FragCoord.xy);
-vec4 bs = load_albedoTex(p, 0);
-vec4 fD = load_normalTex(p, 0);
-vec3 gT = fD.rgb;
+vec4 bo = load_albedoTex(p, 0);
+vec4 fC = load_normalTex(p, 0);
+vec3 gO = fC.rgb;
 
 
-if (fD.a <= 0.0 && load_shadowCatcherTex(p, 0).r <= 0.0) {
-vec4 c = rL(p);
+if (fC.a <= 0.0 && load_shadowCatcherTex(p, 0).r <= 0.0) {
+vec4 c = sI(p);
 vec3 rgb = c.rgb / max(c.a, 1e-6);
-if (encodeOutput) rgb = pow(max(rgb, 0.0), vec3(1.0 / 2.2));
+if (encodeOutput) rgb = mix(1.055 * pow(max(rgb, 0.0), vec3(1.0 / 2.4)) - 0.055,
+rgb * 12.92, lessThanEqual(rgb, vec3(0.0031308)));
 out_color = vec4(rgb, c.a);
-out_guide = vec4(nH(p), 0.0, 1.0);
+out_guide = vec4(loadGuide(p), 0.0, 1.0);
 return;
 }
 
-vec2 lf[PATCH_SIZE];
+vec2 lA[PATCH_SIZE];
 int k = 0;
 for (int dy = -PATCH_RADIUS; dy <= PATCH_RADIUS; dy++) {
 for (int dx = -PATCH_RADIUS; dx <= PATCH_RADIUS; dx++) {
-lf[k] = nH(p + ivec2(dx, dy) * dilation);
+lA[k] = loadGuide(p + ivec2(dx, dy) * dilation);
 k++;
 }
 }
 float k2 = colorSensitivity * colorSensitivity;
 
 vec4 sum = vec4(0.0);
-vec3 gW = max(bs.rgb, vec3(${JNLM_ALBEDO_FLOOR}));
-vec3 mp = vec3(0.0);
-vec3 gy = vec3(0.0);
-vec3 hs = vec3(0.0);
-float ps = 0.0;
-float gA = 0.0;
-float nU = 0.0;
-float kM = 0.0;
+vec3 gR = max(bo.rgb, vec3(${JNLM_ALBEDO_FLOOR}));
+vec3 mQ = vec3(0.0);
+vec3 gv = vec3(0.0);
+vec3 hm = vec3(0.0);
+float qf = 0.0;
+float gx = 0.0;
+float oE = 0.0;
+float kX = 0.0;
 for (int sy = -SEARCH_RADIUS; sy <= SEARCH_RADIUS; sy++) {
 for (int sx = -SEARCH_RADIUS; sx <= SEARCH_RADIUS; sx++) {
 ivec2 q = p + ivec2(sx, sy) * dilation;
@@ -174,68 +176,69 @@ if (
 (sx == 0 && sy == 0) ||
 q.x < 0 || q.y < 0 || q.x >= imageSize.x || q.y >= imageSize.y
 ) continue;
-float features = lw(q, bs, gT);
+float features = lP(q, bo, gO);
 if (features > 16.0) continue;
 
-float fY = 0.0;
+float fW = 0.0;
 k = 0;
 for (int dy = -PATCH_RADIUS; dy <= PATCH_RADIUS; dy++) {
 for (int dx = -PATCH_RADIUS; dx <= PATCH_RADIUS; dx++) {
-vec2 a = lf[k];
-vec2 b = nH(q + ivec2(dx, dy) * dilation);
-float qU = a.x - b.x;
-fY += (qU * qU
+vec2 a = lA[k];
+vec2 b = loadGuide(q + ivec2(dx, dy) * dilation);
+float rV = a.x - b.x;
+fW += (rV * rV
 - VARIANCE_CANCELLATION * (a.y + min(a.y, b.y)))
 / (1e-10 + k2 * (a.y + b.y));
 k++;
 }
 }
-fY /= float(PATCH_SIZE);
+fW /= float(PATCH_SIZE);
 
-float weight = exp(-max(max(fY, 0.0), features));
-vec2 g = nH(q);
+float weight = exp(-max(max(fW, 0.0), features));
+vec2 g = loadGuide(q);
 
-vec3 a = (load_albedoTex(q, 0).rgb - bs.rgb) / gW;
-vec4 value = rL(q);
+vec3 a = (load_albedoTex(q, 0).rgb - bo.rgb) / gR;
+vec4 value = sI(q);
 sum += value * weight;
-mp += a * weight;
-gy += a * a * weight;
-hs += a * value.rgb * weight;
-ps += g.x * weight;
-gA += g.y * weight * weight;
-nU += weight;
-kM = max(kM, weight);
+mQ += a * weight;
+gv += a * a * weight;
+hm += a * value.rgb * weight;
+qf += g.x * weight;
+gx += g.y * weight * weight;
+oE += weight;
+kX = max(kX, weight);
 }
 }
 
 
 
-vec4 center = rL(p);
-vec2 le = nH(p);
-float dG = max(kM, CENTER_WEIGHT_MIN);
-float totalWeight = nU + dG;
-vec4 mean = (sum + center * dG) / totalWeight;
+vec4 center = sI(p);
+vec2 lw = loadGuide(p);
+float dE = max(kX, CENTER_WEIGHT_MIN);
+float totalWeight = oE + dE;
+vec4 mean = (sum + center * dE) / totalWeight;
 
 
-vec3 ez = mp / totalWeight;
-vec3 lA = max(gy / totalWeight - ez * ez, vec3(0.0));
-vec3 qu = hs / totalWeight - ez * mean.rgb;
+vec3 er = mQ / totalWeight;
+vec3 lQ = max(gv / totalWeight - er * er, vec3(0.0));
+vec3 rq = hm / totalWeight - er * mean.rgb;
 
 
 
-vec3 qE = bs.rgb + ez * gW;
-vec3 oz = mean.rgb * gW / max(qE, vec3(${JNLM_ALBEDO_FLOOR}));
-vec3 vB = clamp(qu / (lA + vec3(${JNLM_ALBEDO_REGULARIZATION})), vec3(0.0), oz);
-vec3 rD = mean.rgb - vB * ez;
-vec4 filtered = vec4(max(rD, vec3(0.0)), max(mean.a, 0.0));
+vec3 rD = bo.rgb + er * gR;
+vec3 pd = mean.rgb * gR / max(rD, vec3(${JNLM_ALBEDO_FLOOR}));
+vec3 xI = clamp(rq / (lQ + vec3(${JNLM_ALBEDO_REGULARIZATION})), vec3(0.0), pd);
+vec3 sz = mean.rgb - xI * er;
+vec4 filtered = vec4(max(sz, vec3(0.0)), max(mean.a, 0.0));
 vec3 rgb = filtered.a > 1e-6
 ? filtered.rgb / max(filtered.a, 1e-6)
 : center.rgb / max(center.a, 1e-6);
-if (encodeOutput) rgb = pow(max(rgb, 0.0), vec3(1.0 / 2.2));
+if (encodeOutput) rgb = mix(1.055 * pow(max(rgb, 0.0), vec3(1.0 / 2.4)) - 0.055,
+rgb * 12.92, lessThanEqual(rgb, vec3(0.0031308)));
 out_color = vec4(rgb, filtered.a);
 out_guide = vec4(
-(ps + le.x * dG) / totalWeight,
-(gA + le.y * dG * dG)
+(qf + lw.x * dE) / totalWeight,
+(gx + lw.y * dE * dE)
 / (totalWeight * totalWeight),
 0.0,
 1.0
@@ -272,7 +275,7 @@ export class WebGLJNLMDenoisePass {
         }));
     }
     /** Runs the denoiser and returns the display-encoded result. */
-    render(inputs) {
+    render(inputs, trace) {
         const { width, height, shape } = inputs;
         const passes = [this._guidePass, this._variancePass, ...this._filterPasses];
         passes.forEach((pass) => pass.setSize(width, height));
@@ -300,6 +303,12 @@ export class WebGLJNLMDenoisePass {
                 decodeInput: i === 0,
                 encodeOutput: i === iterations - 1
             });
+            if (trace) {
+                if (i === 0)
+                    trace.push('Decode:DisplayLinear', 'Denoise:DisplayLinear');
+                if (i === iterations - 1)
+                    trace.push('Encode:DisplayEncoded');
+            }
             pass.createPass({
                 guideTex: guide,
                 colorTex: color,

package/lib/renderer/WebGLJNLMDenoiser.js.map

@@ -1 +1 @@
-{"version":3,"file":"WebGLJNLMDenoiser.js","sourceRoot":"","sources":["../../src/renderer/WebGLJNLMDenoiser.ts"],"names":[],"mappings":"AAAA,OAAO,EAAE,cAAc,EAAE,IAAI,EAAE,MAAM,EAAE,MAAM,aAAa,CAAC;AAC3D,OAAO,EAAE,SAAS,EAA0B,MAAM,YAAY,CAAC;AAC/D,OAAO,EAAE,iBAAiB,EAAE,MAAM,wBAAwB,CAAC;AAC3D,OAAO,EACL,8BAA8B,EAC9B,iBAAiB,EACjB,0BAA0B,EAC1B,kBAAkB,EAClB,uBAAuB,EACvB,mBAAmB,EACnB,gBAAgB,EAEjB,MAAM,WAAW,CAAC;AAEnB;;;GAGG;AAEH;;;GAGG;AACH,MAAM,aAAa,GAAG,IAAI,cAAc,CAAC;IACvC,IAAI,EAAE,WAAW;IACjB,OAAO,EAAE,CAAC,OAAO,CAAC;IAClB,QAAQ,EAAE;QACR,UAAU,EAAE,MAAM,CAAC,OAAO,CAAC,WAAW,CAAC;QACvC,iBAAiB,EAAE,MAAM,CAAC,OAAO,CAAC,WAAW,CAAC;KAC/C;IACD,IAAI,EAAE,IAAI,CAAA;;;;;;;;;;;EAWV;CACD,CAAC,CAAC;AAEH;;;GAGG;AACH,MAAM,gBAAgB,GAAG,IAAI,cAAc,CAAC;IAC1C,IAAI,EAAE,cAAc;IACpB,OAAO,EAAE,CAAC,OAAO,CAAC;IAClB,QAAQ,EAAE;QACR,WAAW,EAAE,MAAM,CAAC,OAAO,CAAC,WAAW,CAAC;KACzC;IACD,IAAI,EAAE,IAAI,CAAA;;;;;;;;;;;;;;;;;;;;;;;;;;;;EA4BV;CACD,CAAC,CAAC;AAEH;;;GAGG;AACH,MAAM,cAAc,GAAG,IAAI,cAAc,CAAC;IACxC,IAAI,EAAE,YAAY;IAClB,OAAO,EAAE,CAAC,OAAO,EAAE,OAAO,CAAC;IAC3B,OAAO,EAAE;QACP,aAAa,EAAE,kBAAkB,CAAC,YAAY;QAC9C,YAAY,EAAE,kBAAkB,CAAC,WAAW;QAC5C,qBAAqB,EAAE,CAAC,CAAC,GAAG,kBAAkB,CAAC,eAAe,IAAI,CAAC,CAAC,CAAC,OAAO,CAAC,CAAC,CAAC;QAC/E,qBAAqB,EAAE,CAAC,CAAC,GAAG,kBAAkB,CAAC,eAAe,IAAI,CAAC,CAAC,CAAC,OAAO,CAAC,CAAC,CAAC;KAChF;IACD,QAAQ,EAAE;QACR,QAAQ,EAAE,MAAM,CAAC,OAAO,CAAC,WAAW,CAAC;QACrC,QAAQ,EAAE,MAAM,CAAC,OAAO,CAAC,WAAW,CAAC;QACrC,SAAS,EAAE,MAAM,CAAC,OAAO,CAAC,WAAW,CAAC;QACtC,SAAS,EAAE,MAAM,CAAC,OAAO,CAAC,WAAW,CAAC;QACtC,gBAAgB,EAAE,MAAM,CAAC,OAAO,CAAC,WAAW,CAAC;QAC7C,gBAAgB,EAAE,MAAM,CAAC,OAAO,CAAC,OAAO,EAAE,8BAA8B,CAAC;QACzE,QAAQ,EAAE,MAAM,CAAC,OAAO,CAAC,KAAK,EAAE,CAAC,CAAC;QAClC,WAAW,EAAE,MAAM,CAAC,OAAO,CAAC,MAAM,EAAE,CAAC,CAAC;QACtC,YAAY,EAAE,MAAM,CAAC,OAAO,CAAC,MAAM,EAAE,CAAC,CAAC;KACxC;IACD,IAAI,EAAE,IAAI,CAAA;;;;;;;kCAOsB,CAAC,CAAC,GAAG,uBAAuB,IAAI,CAAC,CAAC,CAAC,OAAO,CAAC,CAAC,CAAC;;;;;;;;;;;;;;;;;;;kDAmBR,iBAAiB;;;;;;;;;;;;;;;;;;;;;;;;;;;;;;;;;;;;;;;6BAuCrC,iBAAiB;;;;;;;;;;;;;;;;;;;;;;;;;;;;;;;;;;;;;;;;;;;;;;;;;;;;;;;;;;;;;;;;;yCAiEG,iBAAiB;kCAC5B,0BAA0B;;;;;;;;;;;;;;;EAepF;CACD,CAAC,CAAC;AAmBH,MAAM,WAAW,GAAG;IAClB,IAAI,EAAE,SAAS,CAAC,KAAK;IACrB,SAAS,EAAE,SAAS,CAAC,OAAO;IAC5B,SAAS,EAAE,SAAS,CAAC,OAAO;CAC7B,CAAC;AAEF,MAAM,OAAO,oBAAoB;IACvB,SAAS,CAAc;IACvB,UAAU,CAAiE;IAC3E,aAAa,CAA4C;IACjE,uEAAuE;IAC/D,aAAa,CAGjB;IAEJ,YAAY,QAAqB;QAC/B,IAAI,CAAC,SAAS,GAAG,QAAQ,CAAC;QAC1B,IAAI,CAAC,UAAU,GAAG,IAAI,iBAAiB,CAAC,WAAW,EAAE,QAAQ,EAAE;YAC7D,MAAM,EAAE,aAAa;YACrB,YAAY,EAAE,EAAE,KAAK,EAAE,WAAW,EAAE;SACrC,CAAC,CAAC;QACH,IAAI,CAAC,aAAa,GAAG,IAAI,iBAAiB,CAAC,cAAc,EAAE,QAAQ,EAAE;YACnE,MAAM,EAAE,gBAAgB;YACxB,YAAY,EAAE,EAAE,KAAK,EAAE,WAAW,EAAE;SACrC,CAAC,CAAC;QACH,IAAI,CAAC,aAAa,GAAG,CAAC,CAAC,EAAE,CAAC,CAAC,CAAC,GAAG,CAAC,GAAG,EAAE,CACnC,IAAI,iBAAiB,CAAC,YAAY,EAAE,QAAQ,EAAE;YAC5C,MAAM,EAAE,cAAc;YACtB,YAAY,EAAE;gBACZ,KAAK,EAAE,EAAE,IAAI,EAAE,SAAS,CAAC,UAAU,EAAE;gBACrC,KAAK,EAAE,WAAW;aACnB;SACF,CAAC,CACH,CAAC;IACJ,CAAC;IAED,gEAAgE;IAChE,MAAM,CAAC,MAAuB;QAC5B,MAAM,EAAE,KAAK,EAAE,MAAM,EAAE,KAAK,EAAE,GAAG,MAAM,CAAC;QACxC,MAAM,MAAM,GAAG,CAAC,IAAI,CAAC,UAAU,EAAE,IAAI,CAAC,aAAa,EAAE,GAAG,IAAI,CAAC,aAAa,CAAC,CAAC;QAC5E,MAAM,CAAC,OAAO,CAAC,CAAC,IAAI,EAAE,EAAE,CAAC,IAAI,CAAC,OAAO,CAAC,KAAK,EAAE,MAAM,CAAC,CAAC,CAAC;QACtD,IAAI,CAAC,UAAU,CAAC,UAAU,CAAC;YACzB,UAAU,EAAE,MAAM,CAAC,OAAO;YAC1B,iBAAiB,EAAE,MAAM,CAAC,cAAc,IAAI,IAAI,CAAC,SAAS,CAAC,YAAY;SACxE,CAAC,CAAC;QACH,IAAI,CAAC,aAAa,CAAC,UAAU,CAAC,EAAE,WAAW,EAAE,IAAI,CAAC,UAAU,CAAC,SAAS,CAAC,OAAO,CAAC,EAAE,CAAC,CAAC;QAEnF,MAAM,WAAW,GAAG,8BAA8B,GAAG,IAAI,CAAC,GAAG,CAAC,MAAM,CAAC,QAAQ,EAAE,IAAI,CAAC,CAAC;QACrF,MAAM,UAAU,GAAG,IAAI,CAAC,GAAG,CAAC,mBAAmB,EAAE,IAAI,CAAC,GAAG,CAAC,CAAC,EAAE,IAAI,CAAC,KAAK,CAAC,KAAK,CAAC,UAAU,CAAC,CAAC,CAAC,CAAC;QAC5F,IAAI,KAAK,GAAG,IAAI,CAAC,aAAa,CAAC,SAAS,CAAC,OAAO,CAAC,CAAC;QAClD,IAAI,KAAK,GAAG,MAAM,CAAC,KAAK,CAAC;QACzB,KAAK,IAAI,CAAC,GAAG,CAAC,EAAE,CAAC,GAAG,UAAU,EAAE,CAAC,EAAE,EAAE,CAAC;YACpC,MAAM,IAAI,GAAG,IAAI,CAAC,aAAa,CAAC,CAAC,GAAG,CAAC,CAAC,CAAC;YACvC,IAAI,CAAC,UAAU,CAAC;gBACd,aAAa,EAAE,KAAK,CAAC,YAAY;gBACjC,6DAA6D;gBAC7D,YAAY,EAAE,MAAM,CAAC,KAAK,CAAC,WAAW,CAAC;gBACvC,qBAAqB,EAAE,CAAC,CAAC,GAAG,KAAK,CAAC,eAAe,IAAI,CAAC,CAAC,CAAC,OAAO,CAAC,CAAC,CAAC;gBAClE,qBAAqB,EAAE,CAAC,CAAC,GAAG,KAAK,CAAC,eAAe,IAAI,CAAC,CAAC,CAAC,OAAO,CAAC,CAAC,CAAC;aACnE,CAAC,CAAC;YACH,IAAI,CAAC,WAAW,CAAC;gBACf,gBAAgB,EAAE,WAAW;gBAC7B,QAAQ,EAAE,gBAAgB,CAAC,CAAC,CAAC;gBAC7B,WAAW,EAAE,CAAC,KAAK,CAAC;gBACpB,YAAY,EAAE,CAAC,KAAK,UAAU,GAAG,CAAC;aACnC,CAAC,CAAC;YACH,IAAI,CAAC,UAAU,CAAC;gBACd,QAAQ,EAAE,KAAK;gBACf,QAAQ,EAAE,KAAK;gBACf,SAAS,EAAE,MAAM,CAAC,MAAM;gBACxB,SAAS,EAAE,MAAM,CAAC,MAAM;gBACxB,gBAAgB,EAAE,MAAM,CAAC,aAAa;aACvC,CAAC,CAAC;YACH,KAAK,GAAG,IAAI,CAAC,SAAS,CAAC,OAAO,CAAC,CAAC;YAChC,KAAK,GAAG,IAAI,CAAC,SAAS,CAAC,OAAO,CAAC,CAAC;QAClC,CAAC;QACD,OAAO,KAAK,CAAC;IACf,CAAC;IAED,eAAe;QACb,CAAC,IAAI,CAAC,UAAU,EAAE,IAAI,CAAC,aAAa,EAAE,GAAG,IAAI,CAAC,aAAa,CAAC,CAAC,OAAO,CAAC,IAAI,CAAC,EAAE,CAAC,IAAI,CAAC,eAAe,EAAE,CAAC,CAAC;IACvG,CAAC;IAED,YAAY,CAAC,OAAkB,IAAI,IAAI,CAAC,aAAa,CAAC,OAAO,CAAC,IAAI,CAAC,EAAE,CAAC,IAAI,CAAC,YAAY,CAAC,OAAO,CAAC,CAAC,CAAC,CAAC,CAAC;IACpG,aAAa,CAAC,OAAkB,IAAI,IAAI,CAAC,aAAa,CAAC,OAAO,CAAC,IAAI,CAAC,EAAE,CAAC,IAAI,CAAC,aAAa,CAAC,OAAO,CAAC,CAAC,CAAC,CAAC,CAAC;IAEtG,cAAc;QACZ,OAAO,CAAC,IAAI,CAAC,UAAU,EAAE,IAAI,CAAC,aAAa,EAAE,GAAG,IAAI,CAAC,aAAa,CAAC;aAChE,MAAM,CAAC,CAAC,GAAG,EAAE,IAAI,EAAE,EAAE,CAAC,GAAG,GAAG,IAAI,CAAC,cAAc,EAAE,EAAE,CAAC,CAAC,CAAC;IAC3D,CAAC;IAED,OAAO;QACL,CAAC,IAAI,CAAC,UAAU,EAAE,IAAI,CAAC,aAAa,EAAE,GAAG,IAAI,CAAC,aAAa,CAAC,CAAC,OAAO,CAAC,CAAC,IAAI,EAAE,EAAE,CAC5E,IAAI,CAAC,OAAO,EAAE,CACf,CAAC;IACJ,CAAC;CACF","sourcesContent":["import { FragmentShader, glsl, Shader } from './shader.js';\nimport { constants, WebGLDevice, Texture2D } from './webgl.js';\nimport { WebGLFullQuadPass } from './WebGLFullQuadPass.js';\nimport {\n  JNLM_DEFAULT_COLOR_SENSITIVITY,\n  JNLM_ALBEDO_FLOOR,\n  JNLM_ALBEDO_REGULARIZATION,\n  JNLM_DEFAULT_SHAPE,\n  JNLM_COVERAGE_BANDWIDTH,\n  JNLM_MAX_ITERATIONS,\n  jnlmPassDilation,\n  type JNLMFilterShape\n} from './jnlm.js';\n\n/**\n * WebGL 2 passes of the JNLM denoiser described in `jnlm.ts`, matching\n * WGPUJNLMDenoiser. Textures replace its storage buffers.\n */\n\n/**\n * The guide mean and the variance of that mean. The main and catcher trace\n * passes each accumulate (sum, sum of squares, count) for their own samples.\n */\nconst guideFragment = new FragmentShader({\n  name: 'jnlmGuide',\n  outputs: ['guide'],\n  uniforms: {\n    momentsTex: Shader.uniform('sampler2D'),\n    catcherMomentsTex: Shader.uniform('sampler2D')\n  },\n  main: glsl`\nvoid main() {\n  ivec2 p = ivec2(gl_FragCoord.xy);\n  vec4 moments = load_momentsTex(p, 0) + load_catcherMomentsTex(p, 0);\n  float n = moments.z;\n  float mean = moments.x / max(n, 1.0);\n  // Variance of the mean, or -1 when one sample cannot estimate it.\n  float variance = n >= 2.0\n    ? max(moments.y - n * mean * mean, 0.0) / (n * (n - 1.0))\n    : -1.0;\n  out_guide = vec4(mean, variance, 0.0, 1.0);\n}`\n});\n\n/**\n * Smooths the noisy per-pixel variance over 3 x 3 pixels. Pixels with a\n * single sample take the spatial variance of their neighborhood instead.\n */\nconst varianceFragment = new FragmentShader({\n  name: 'jnlmVariance',\n  outputs: ['guide'],\n  uniforms: {\n    rawGuideTex: Shader.uniform('sampler2D')\n  },\n  main: glsl`\nvoid main() {\n  ivec2 p = ivec2(gl_FragCoord.xy);\n  ivec2 size = rawGuideTexRegion();\n  vec2 center = load_rawGuideTex(p, 0).xy;\n  float varianceSum = 0.0;\n  float varianceCount = 0.0;\n  float meanSum = 0.0;\n  float meanSquareSum = 0.0;\n  for (int dy = -1; dy <= 1; dy++) {\n    for (int dx = -1; dx <= 1; dx++) {\n      vec2 value = load_rawGuideTex(clamp(p + ivec2(dx, dy), ivec2(0), size - 1), 0).xy;\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  float variance;\n  if (center.y >= 0.0) {\n    variance = varianceSum / varianceCount;\n  } else {\n    float spatialMean = meanSum / 9.0;\n    variance = max(meanSquareSum / 9.0 - spatialMean * spatialMean, 0.0) * 9.0 / 8.0;\n  }\n  out_guide = vec4(center.x, variance, 0.0, 1.0);\n}`\n});\n\n/**\n * One joint NL-means pass; see createJNLMFilterWGSL. The first pass decodes\n * the display-encoded composition to linear light, the last encodes it again.\n */\nconst filterFragment = new FragmentShader({\n  name: 'jnlmFilter',\n  outputs: ['color', 'guide'],\n  defines: {\n    SEARCH_RADIUS: JNLM_DEFAULT_SHAPE.searchRadius,\n    PATCH_RADIUS: JNLM_DEFAULT_SHAPE.patchRadius,\n    INV_ALBEDO_BANDWIDTH2: (1 / JNLM_DEFAULT_SHAPE.albedoBandwidth ** 2).toFixed(6),\n    INV_NORMAL_BANDWIDTH2: (1 / JNLM_DEFAULT_SHAPE.normalBandwidth ** 2).toFixed(6)\n  },\n  uniforms: {\n    guideTex: Shader.uniform('sampler2D'),\n    colorTex: Shader.uniform('sampler2D'),\n    albedoTex: Shader.uniform('sampler2D'),\n    normalTex: Shader.uniform('sampler2D'),\n    shadowCatcherTex: Shader.uniform('sampler2D'),\n    colorSensitivity: Shader.uniform('float', JNLM_DEFAULT_COLOR_SENSITIVITY),\n    dilation: Shader.uniform('int', 1),\n    decodeInput: Shader.uniform('bool', 0),\n    encodeOutput: Shader.uniform('bool', 0)\n  },\n  main: glsl`\n#define PATCH_SIZE ((2 * PATCH_RADIUS + 1) * (2 * PATCH_RADIUS + 1))\n// Variance cancellation strength alpha of the paper.\n#define VARIANCE_CANCELLATION 1.0\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.\n#define CENTER_WEIGHT_MIN 0.01\n#define INV_COVERAGE_BANDWIDTH2 ${(1 / JNLM_COVERAGE_BANDWIDTH ** 2).toFixed(6)}\n\nivec2 imageSize;\n\nvec2 loadGuide(ivec2 p) {\n  return load_guideTex(clamp(p, ivec2(0), imageSize - 1), 0).xy;\n}\n\n// Premultiplied linear color.\nvec4 loadColor(ivec2 p) {\n  vec4 c = load_colorTex(p, 0);\n  if (decodeInput) c.rgb = pow(max(c.rgb, 0.0), vec3(2.2));\n  return vec4(c.rgb * c.a, c.a);\n}\n\nfloat featureDistance(ivec2 q, vec4 centerAlbedo, vec3 centerNormal) {\n  vec4 albedo = load_albedoTex(q, 0);\n  // Match WebGPU: protect relative RGB contrast, including dark textures,\n  // while keeping coverage differences in absolute units.\n  vec3 albedoScale = max((albedo.rgb + centerAlbedo.rgb) * 0.5, vec3(${JNLM_ALBEDO_FLOOR}));\n  vec3 albedoDelta = (albedo.rgb - centerAlbedo.rgb) / albedoScale;\n  float coverageDelta = albedo.a - centerAlbedo.a;\n  vec3 normalDelta = load_normalTex(q, 0).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\nvoid main() {\n  imageSize = guideTexRegion();\n  ivec2 p = ivec2(gl_FragCoord.xy);\n  vec4 centerAlbedo = load_albedoTex(p, 0);\n  vec4 centerNormalTexel = load_normalTex(p, 0);\n  vec3 centerNormal = centerNormalTexel.rgb;\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 && load_shadowCatcherTex(p, 0).r <= 0.0) {\n    vec4 c = loadColor(p);\n    vec3 rgb = c.rgb / max(c.a, 1e-6);\n    if (encodeOutput) rgb = pow(max(rgb, 0.0), vec3(1.0 / 2.2));\n    out_color = vec4(rgb, c.a);\n    out_guide = vec4(loadGuide(p), 0.0, 1.0);\n    return;\n  }\n\n  vec2 centerPatch[PATCH_SIZE];\n  int k = 0;\n  for (int dy = -PATCH_RADIUS; dy <= PATCH_RADIUS; dy++) {\n    for (int dx = -PATCH_RADIUS; dx <= PATCH_RADIUS; dx++) {\n      centerPatch[k] = loadGuide(p + ivec2(dx, dy) * dilation);\n      k++;\n    }\n  }\n  float k2 = colorSensitivity * colorSensitivity;\n\n  vec4 sum = vec4(0.0);\n  vec3 featureScale = max(centerAlbedo.rgb, vec3(${JNLM_ALBEDO_FLOOR}));\n  vec3 featureSum = vec3(0.0);\n  vec3 featureSquareSum = vec3(0.0);\n  vec3 featureColorSum = vec3(0.0);\n  float guideSum = 0.0;\n  float guideVarianceSum = 0.0;\n  float weightSum = 0.0;\n  float maxWeight = 0.0;\n  for (int sy = -SEARCH_RADIUS; sy <= SEARCH_RADIUS; sy++) {\n    for (int sx = -SEARCH_RADIUS; sx <= SEARCH_RADIUS; sx++) {\n      ivec2 q = p + ivec2(sx, sy) * dilation;\n      if (\n        (sx == 0 && sy == 0) ||\n        q.x < 0 || q.y < 0 || q.x >= imageSize.x || q.y >= imageSize.y\n      ) continue;\n      float features = featureDistance(q, centerAlbedo, centerNormal);\n      if (features > 16.0) continue;\n\n      float colorDistance = 0.0;\n      k = 0;\n      for (int dy = -PATCH_RADIUS; dy <= PATCH_RADIUS; dy++) {\n        for (int dx = -PATCH_RADIUS; dx <= PATCH_RADIUS; dx++) {\n          vec2 a = centerPatch[k];\n          vec2 b = loadGuide(q + ivec2(dx, dy) * dilation);\n          float 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 /= float(PATCH_SIZE);\n\n      float weight = exp(-max(max(colorDistance, 0.0), features));\n      vec2 g = loadGuide(q);\n      // Filter premultiplied color so coverage and color stay consistent.\n      vec3 a = (load_albedoTex(q, 0).rgb - centerAlbedo.rgb) / featureScale;\n      vec4 value = loadColor(q);\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  vec4 center = loadColor(p);\n  vec2 centerGuide = loadGuide(p);\n  float centerWeight = max(maxWeight, CENTER_WEIGHT_MIN);\n  float totalWeight = weightSum + centerWeight;\n  vec4 mean = (sum + center * 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  vec3 meanFeature = featureSum / totalWeight;\n  vec3 featureVariance = max(featureSquareSum / totalWeight - meanFeature * meanFeature, vec3(0.0));\n  vec3 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  vec3 meanAlbedo = centerAlbedo.rgb + meanFeature * featureScale;\n  vec3 maximumSlope = mean.rgb * featureScale / max(meanAlbedo, vec3(${JNLM_ALBEDO_FLOOR}));\n  vec3 slope = clamp(covariance / (featureVariance + vec3(${JNLM_ALBEDO_REGULARIZATION})), vec3(0.0), maximumSlope);\n  vec3 corrected = mean.rgb - slope * meanFeature;\n  vec4 filtered = vec4(max(corrected, vec3(0.0)), max(mean.a, 0.0));\n  vec3 rgb = filtered.a > 1e-6\n    ? filtered.rgb / max(filtered.a, 1e-6)\n    : center.rgb / max(center.a, 1e-6);\n  if (encodeOutput) rgb = pow(max(rgb, 0.0), vec3(1.0 / 2.2));\n  out_color = vec4(rgb, filtered.a);\n  out_guide = vec4(\n    (guideSum + centerGuide.x * centerWeight) / totalWeight,\n    (guideVarianceSum + centerGuide.y * centerWeight * centerWeight)\n      / (totalWeight * totalWeight),\n    0.0,\n    1.0\n  );\n}`\n});\n\nexport interface WebGLJNLMInputs {\n  width: number;\n  height: number;\n  /** Main trace pass moments: guide sum, sum of squares and count. */\n  moments: Texture2D;\n  /** Catcher trace pass moments, when the scene has a shadow catcher. */\n  catcherMoments?: Texture2D;\n  /** Display-encoded composition to filter. */\n  color: Texture2D;\n  albedo: Texture2D;\n  normal: Texture2D;\n  /** Normalized shadow-catcher mask; red is the catcher coverage. */\n  shadowCatcher: Texture2D;\n  strength: number;\n  shape: JNLMFilterShape;\n}\n\nconst floatOutput = {\n  type: constants.FLOAT,\n  minFilter: constants.NEAREST,\n  magFilter: constants.NEAREST\n};\n\nexport class WebGLJNLMDenoisePass {\n  private _renderer: WebGLDevice;\n  private _guidePass: WebGLFullQuadPass<'momentsTex' | 'catcherMomentsTex', 'guide'>;\n  private _variancePass: WebGLFullQuadPass<'rawGuideTex', 'guide'>;\n  // Two instances so consecutive passes ping-pong between their outputs.\n  private _filterPasses: WebGLFullQuadPass<\n    'guideTex' | 'colorTex' | 'albedoTex' | 'normalTex' | 'shadowCatcherTex',\n    'color' | 'guide'\n  >[];\n\n  constructor(renderer: WebGLDevice) {\n    this._renderer = renderer;\n    this._guidePass = new WebGLFullQuadPass('jnlmGuide', renderer, {\n      shader: guideFragment,\n      outputParams: { guide: floatOutput }\n    });\n    this._variancePass = new WebGLFullQuadPass('jnlmVariance', renderer, {\n      shader: varianceFragment,\n      outputParams: { guide: floatOutput }\n    });\n    this._filterPasses = [0, 1].map(() =>\n      new WebGLFullQuadPass('jnlmFilter', renderer, {\n        shader: filterFragment,\n        outputParams: {\n          color: { type: constants.HALF_FLOAT },\n          guide: floatOutput\n        }\n      })\n    );\n  }\n\n  /** Runs the denoiser and returns the display-encoded result. */\n  render(inputs: WebGLJNLMInputs): Texture2D {\n    const { width, height, shape } = inputs;\n    const passes = [this._guidePass, this._variancePass, ...this._filterPasses];\n    passes.forEach((pass) => pass.setSize(width, height));\n    this._guidePass.createPass({\n      momentsTex: inputs.moments,\n      catcherMomentsTex: inputs.catcherMoments ?? this._renderer.emptyTexture\n    });\n    this._variancePass.createPass({ rawGuideTex: this._guidePass.getOutput('guide') });\n\n    const sensitivity = JNLM_DEFAULT_COLOR_SENSITIVITY * Math.max(inputs.strength, 1e-3);\n    const iterations = Math.min(JNLM_MAX_ITERATIONS, Math.max(1, Math.round(shape.iterations)));\n    let guide = this._variancePass.getOutput('guide');\n    let color = inputs.color;\n    for (let i = 0; i < iterations; i++) {\n      const pass = this._filterPasses[i % 2];\n      pass.setDefines({\n        SEARCH_RADIUS: shape.searchRadius,\n        // Numeric zero removes a define in the WebGL shader builder.\n        PATCH_RADIUS: String(shape.patchRadius),\n        INV_ALBEDO_BANDWIDTH2: (1 / shape.albedoBandwidth ** 2).toFixed(6),\n        INV_NORMAL_BANDWIDTH2: (1 / shape.normalBandwidth ** 2).toFixed(6)\n      });\n      pass.setUniforms({\n        colorSensitivity: sensitivity,\n        dilation: jnlmPassDilation(i),\n        decodeInput: i === 0,\n        encodeOutput: i === iterations - 1\n      });\n      pass.createPass({\n        guideTex: guide,\n        colorTex: color,\n        albedoTex: inputs.albedo,\n        normalTex: inputs.normal,\n        shadowCatcherTex: inputs.shadowCatcher\n      });\n      guide = pass.getOutput('guide');\n      color = pass.getOutput('color');\n    }\n    return color;\n  }\n\n  reserveCapacity() {\n    [this._guidePass, this._variancePass, ...this._filterPasses].forEach(pass => pass.reserveCapacity());\n  }\n\n  retainOutput(texture: Texture2D) { this._filterPasses.forEach(pass => pass.retainOutput(texture)); }\n  releaseOutput(texture: Texture2D) { this._filterPasses.forEach(pass => pass.releaseOutput(texture)); }\n\n  getMemoryBytes() {\n    return [this._guidePass, this._variancePass, ...this._filterPasses]\n      .reduce((sum, pass) => sum + pass.getMemoryBytes(), 0);\n  }\n\n  dispose() {\n    [this._guidePass, this._variancePass, ...this._filterPasses].forEach((pass) =>\n      pass.dispose()\n    );\n  }\n}\n"]}
+{"version":3,"file":"WebGLJNLMDenoiser.js","sourceRoot":"","sources":["../../src/renderer/WebGLJNLMDenoiser.ts"],"names":[],"mappings":"AAAA,OAAO,EAAE,cAAc,EAAE,IAAI,EAAE,MAAM,EAAE,MAAM,aAAa,CAAC;AAC3D,OAAO,EAAE,SAAS,EAA0B,MAAM,YAAY,CAAC;AAC/D,OAAO,EAAE,iBAAiB,EAAE,MAAM,wBAAwB,CAAC;AAC3D,OAAO,EACL,8BAA8B,EAC9B,iBAAiB,EACjB,0BAA0B,EAC1B,kBAAkB,EAClB,uBAAuB,EACvB,mBAAmB,EACnB,gBAAgB,EAEjB,MAAM,WAAW,CAAC;AAEnB;;;GAGG;AAEH;;;GAGG;AACH,MAAM,aAAa,GAAG,IAAI,cAAc,CAAC;IACvC,IAAI,EAAE,WAAW;IACjB,OAAO,EAAE,CAAC,OAAO,CAAC;IAClB,QAAQ,EAAE;QACR,UAAU,EAAE,MAAM,CAAC,OAAO,CAAC,WAAW,CAAC;QACvC,iBAAiB,EAAE,MAAM,CAAC,OAAO,CAAC,WAAW,CAAC;KAC/C;IACD,IAAI,EAAE,IAAI,CAAA;;;;;;;;;;;EAWV;CACD,CAAC,CAAC;AAEH;;;GAGG;AACH,MAAM,gBAAgB,GAAG,IAAI,cAAc,CAAC;IAC1C,IAAI,EAAE,cAAc;IACpB,OAAO,EAAE,CAAC,OAAO,CAAC;IAClB,QAAQ,EAAE;QACR,WAAW,EAAE,MAAM,CAAC,OAAO,CAAC,WAAW,CAAC;KACzC;IACD,IAAI,EAAE,IAAI,CAAA;;;;;;;;;;;;;;;;;;;;;;;;;;;;EA4BV;CACD,CAAC,CAAC;AAEH;;;GAGG;AACH,MAAM,cAAc,GAAG,IAAI,cAAc,CAAC;IACxC,IAAI,EAAE,YAAY;IAClB,OAAO,EAAE,CAAC,OAAO,EAAE,OAAO,CAAC;IAC3B,OAAO,EAAE;QACP,aAAa,EAAE,kBAAkB,CAAC,YAAY;QAC9C,YAAY,EAAE,kBAAkB,CAAC,WAAW;QAC5C,qBAAqB,EAAE,CAAC,CAAC,GAAG,kBAAkB,CAAC,eAAe,IAAI,CAAC,CAAC,CAAC,OAAO,CAAC,CAAC,CAAC;QAC/E,qBAAqB,EAAE,CAAC,CAAC,GAAG,kBAAkB,CAAC,eAAe,IAAI,CAAC,CAAC,CAAC,OAAO,CAAC,CAAC,CAAC;KAChF;IACD,QAAQ,EAAE;QACR,QAAQ,EAAE,MAAM,CAAC,OAAO,CAAC,WAAW,CAAC;QACrC,QAAQ,EAAE,MAAM,CAAC,OAAO,CAAC,WAAW,CAAC;QACrC,SAAS,EAAE,MAAM,CAAC,OAAO,CAAC,WAAW,CAAC;QACtC,SAAS,EAAE,MAAM,CAAC,OAAO,CAAC,WAAW,CAAC;QACtC,gBAAgB,EAAE,MAAM,CAAC,OAAO,CAAC,WAAW,CAAC;QAC7C,gBAAgB,EAAE,MAAM,CAAC,OAAO,CAAC,OAAO,EAAE,8BAA8B,CAAC;QACzE,QAAQ,EAAE,MAAM,CAAC,OAAO,CAAC,KAAK,EAAE,CAAC,CAAC;QAClC,WAAW,EAAE,MAAM,CAAC,OAAO,CAAC,MAAM,EAAE,CAAC,CAAC;QACtC,YAAY,EAAE,MAAM,CAAC,OAAO,CAAC,MAAM,EAAE,CAAC,CAAC;KACxC;IACD,IAAI,EAAE,IAAI,CAAA;;;;;;;kCAOsB,CAAC,CAAC,GAAG,uBAAuB,IAAI,CAAC,CAAC,CAAC,OAAO,CAAC,CAAC,CAAC;;;;;;;;;;;;;;;;;;;;kDAoBR,iBAAiB;;;;;;;;;;;;;;;;;;;;;;;;;;;;;;;;;;;;;;;;6BAwCrC,iBAAiB;;;;;;;;;;;;;;;;;;;;;;;;;;;;;;;;;;;;;;;;;;;;;;;;;;;;;;;;;;;;;;;;;yCAiEG,iBAAiB;kCAC5B,0BAA0B;;;;;;;;;;;;;;;;EAgBpF;CACD,CAAC,CAAC;AAmBH,MAAM,WAAW,GAAG;IAClB,IAAI,EAAE,SAAS,CAAC,KAAK;IACrB,SAAS,EAAE,SAAS,CAAC,OAAO;IAC5B,SAAS,EAAE,SAAS,CAAC,OAAO;CAC7B,CAAC;AAEF,MAAM,OAAO,oBAAoB;IACvB,SAAS,CAAc;IACvB,UAAU,CAAiE;IAC3E,aAAa,CAA4C;IACjE,uEAAuE;IAC/D,aAAa,CAGjB;IAEJ,YAAY,QAAqB;QAC/B,IAAI,CAAC,SAAS,GAAG,QAAQ,CAAC;QAC1B,IAAI,CAAC,UAAU,GAAG,IAAI,iBAAiB,CAAC,WAAW,EAAE,QAAQ,EAAE;YAC7D,MAAM,EAAE,aAAa;YACrB,YAAY,EAAE,EAAE,KAAK,EAAE,WAAW,EAAE;SACrC,CAAC,CAAC;QACH,IAAI,CAAC,aAAa,GAAG,IAAI,iBAAiB,CAAC,cAAc,EAAE,QAAQ,EAAE;YACnE,MAAM,EAAE,gBAAgB;YACxB,YAAY,EAAE,EAAE,KAAK,EAAE,WAAW,EAAE;SACrC,CAAC,CAAC;QACH,IAAI,CAAC,aAAa,GAAG,CAAC,CAAC,EAAE,CAAC,CAAC,CAAC,GAAG,CAAC,GAAG,EAAE,CACnC,IAAI,iBAAiB,CAAC,YAAY,EAAE,QAAQ,EAAE;YAC5C,MAAM,EAAE,cAAc;YACtB,YAAY,EAAE;gBACZ,KAAK,EAAE,EAAE,IAAI,EAAE,SAAS,CAAC,UAAU,EAAE;gBACrC,KAAK,EAAE,WAAW;aACnB;SACF,CAAC,CACH,CAAC;IACJ,CAAC;IAED,gEAAgE;IAChE,MAAM,CAAC,MAAuB,EAAE,KAAgB;QAC9C,MAAM,EAAE,KAAK,EAAE,MAAM,EAAE,KAAK,EAAE,GAAG,MAAM,CAAC;QACxC,MAAM,MAAM,GAAG,CAAC,IAAI,CAAC,UAAU,EAAE,IAAI,CAAC,aAAa,EAAE,GAAG,IAAI,CAAC,aAAa,CAAC,CAAC;QAC5E,MAAM,CAAC,OAAO,CAAC,CAAC,IAAI,EAAE,EAAE,CAAC,IAAI,CAAC,OAAO,CAAC,KAAK,EAAE,MAAM,CAAC,CAAC,CAAC;QACtD,IAAI,CAAC,UAAU,CAAC,UAAU,CAAC;YACzB,UAAU,EAAE,MAAM,CAAC,OAAO;YAC1B,iBAAiB,EAAE,MAAM,CAAC,cAAc,IAAI,IAAI,CAAC,SAAS,CAAC,YAAY;SACxE,CAAC,CAAC;QACH,IAAI,CAAC,aAAa,CAAC,UAAU,CAAC,EAAE,WAAW,EAAE,IAAI,CAAC,UAAU,CAAC,SAAS,CAAC,OAAO,CAAC,EAAE,CAAC,CAAC;QAEnF,MAAM,WAAW,GAAG,8BAA8B,GAAG,IAAI,CAAC,GAAG,CAAC,MAAM,CAAC,QAAQ,EAAE,IAAI,CAAC,CAAC;QACrF,MAAM,UAAU,GAAG,IAAI,CAAC,GAAG,CAAC,mBAAmB,EAAE,IAAI,CAAC,GAAG,CAAC,CAAC,EAAE,IAAI,CAAC,KAAK,CAAC,KAAK,CAAC,UAAU,CAAC,CAAC,CAAC,CAAC;QAC5F,IAAI,KAAK,GAAG,IAAI,CAAC,aAAa,CAAC,SAAS,CAAC,OAAO,CAAC,CAAC;QAClD,IAAI,KAAK,GAAG,MAAM,CAAC,KAAK,CAAC;QACzB,KAAK,IAAI,CAAC,GAAG,CAAC,EAAE,CAAC,GAAG,UAAU,EAAE,CAAC,EAAE,EAAE,CAAC;YACpC,MAAM,IAAI,GAAG,IAAI,CAAC,aAAa,CAAC,CAAC,GAAG,CAAC,CAAC,CAAC;YACvC,IAAI,CAAC,UAAU,CAAC;gBACd,aAAa,EAAE,KAAK,CAAC,YAAY;gBACjC,6DAA6D;gBAC7D,YAAY,EAAE,MAAM,CAAC,KAAK,CAAC,WAAW,CAAC;gBACvC,qBAAqB,EAAE,CAAC,CAAC,GAAG,KAAK,CAAC,eAAe,IAAI,CAAC,CAAC,CAAC,OAAO,CAAC,CAAC,CAAC;gBAClE,qBAAqB,EAAE,CAAC,CAAC,GAAG,KAAK,CAAC,eAAe,IAAI,CAAC,CAAC,CAAC,OAAO,CAAC,CAAC,CAAC;aACnE,CAAC,CAAC;YACH,IAAI,CAAC,WAAW,CAAC;gBACf,gBAAgB,EAAE,WAAW;gBAC7B,QAAQ,EAAE,gBAAgB,CAAC,CAAC,CAAC;gBAC7B,WAAW,EAAE,CAAC,KAAK,CAAC;gBACpB,YAAY,EAAE,CAAC,KAAK,UAAU,GAAG,CAAC;aACnC,CAAC,CAAC;YACH,IAAI,KAAK,EAAE,CAAC;gBACV,IAAI,CAAC,KAAK,CAAC;oBAAE,KAAK,CAAC,IAAI,CAAC,sBAAsB,EAAE,uBAAuB,CAAC,CAAC;gBACzE,IAAI,CAAC,KAAK,UAAU,GAAG,CAAC;oBAAE,KAAK,CAAC,IAAI,CAAC,uBAAuB,CAAC,CAAC;YAChE,CAAC;YACD,IAAI,CAAC,UAAU,CAAC;gBACd,QAAQ,EAAE,KAAK;gBACf,QAAQ,EAAE,KAAK;gBACf,SAAS,EAAE,MAAM,CAAC,MAAM;gBACxB,SAAS,EAAE,MAAM,CAAC,MAAM;gBACxB,gBAAgB,EAAE,MAAM,CAAC,aAAa;aACvC,CAAC,CAAC;YACH,KAAK,GAAG,IAAI,CAAC,SAAS,CAAC,OAAO,CAAC,CAAC;YAChC,KAAK,GAAG,IAAI,CAAC,SAAS,CAAC,OAAO,CAAC,CAAC;QAClC,CAAC;QACD,OAAO,KAAK,CAAC;IACf,CAAC;IAED,eAAe;QACb,CAAC,IAAI,CAAC,UAAU,EAAE,IAAI,CAAC,aAAa,EAAE,GAAG,IAAI,CAAC,aAAa,CAAC,CAAC,OAAO,CAAC,IAAI,CAAC,EAAE,CAAC,IAAI,CAAC,eAAe,EAAE,CAAC,CAAC;IACvG,CAAC;IAED,YAAY,CAAC,OAAkB,IAAI,IAAI,CAAC,aAAa,CAAC,OAAO,CAAC,IAAI,CAAC,EAAE,CAAC,IAAI,CAAC,YAAY,CAAC,OAAO,CAAC,CAAC,CAAC,CAAC,CAAC;IACpG,aAAa,CAAC,OAAkB,IAAI,IAAI,CAAC,aAAa,CAAC,OAAO,CAAC,IAAI,CAAC,EAAE,CAAC,IAAI,CAAC,aAAa,CAAC,OAAO,CAAC,CAAC,CAAC,CAAC,CAAC;IAEtG,cAAc;QACZ,OAAO,CAAC,IAAI,CAAC,UAAU,EAAE,IAAI,CAAC,aAAa,EAAE,GAAG,IAAI,CAAC,aAAa,CAAC;aAChE,MAAM,CAAC,CAAC,GAAG,EAAE,IAAI,EAAE,EAAE,CAAC,GAAG,GAAG,IAAI,CAAC,cAAc,EAAE,EAAE,CAAC,CAAC,CAAC;IAC3D,CAAC;IAED,OAAO;QACL,CAAC,IAAI,CAAC,UAAU,EAAE,IAAI,CAAC,aAAa,EAAE,GAAG,IAAI,CAAC,aAAa,CAAC,CAAC,OAAO,CAAC,CAAC,IAAI,EAAE,EAAE,CAC5E,IAAI,CAAC,OAAO,EAAE,CACf,CAAC;IACJ,CAAC;CACF","sourcesContent":["import { FragmentShader, glsl, Shader } from './shader.js';\nimport { constants, WebGLDevice, Texture2D } from './webgl.js';\nimport { WebGLFullQuadPass } from './WebGLFullQuadPass.js';\nimport {\n  JNLM_DEFAULT_COLOR_SENSITIVITY,\n  JNLM_ALBEDO_FLOOR,\n  JNLM_ALBEDO_REGULARIZATION,\n  JNLM_DEFAULT_SHAPE,\n  JNLM_COVERAGE_BANDWIDTH,\n  JNLM_MAX_ITERATIONS,\n  jnlmPassDilation,\n  type JNLMFilterShape\n} from './jnlm.js';\n\n/**\n * WebGL 2 passes of the JNLM denoiser described in `jnlm.ts`, matching\n * WGPUJNLMDenoiser. Textures replace its storage buffers.\n */\n\n/**\n * The guide mean and the variance of that mean. The main and catcher trace\n * passes each accumulate (sum, sum of squares, count) for their own samples.\n */\nconst guideFragment = new FragmentShader({\n  name: 'jnlmGuide',\n  outputs: ['guide'],\n  uniforms: {\n    momentsTex: Shader.uniform('sampler2D'),\n    catcherMomentsTex: Shader.uniform('sampler2D')\n  },\n  main: glsl`\nvoid main() {\n  ivec2 p = ivec2(gl_FragCoord.xy);\n  vec4 moments = load_momentsTex(p, 0) + load_catcherMomentsTex(p, 0);\n  float n = moments.z;\n  float mean = moments.x / max(n, 1.0);\n  // Variance of the mean, or -1 when one sample cannot estimate it.\n  float variance = n >= 2.0\n    ? max(moments.y - n * mean * mean, 0.0) / (n * (n - 1.0))\n    : -1.0;\n  out_guide = vec4(mean, variance, 0.0, 1.0);\n}`\n});\n\n/**\n * Smooths the noisy per-pixel variance over 3 x 3 pixels. Pixels with a\n * single sample take the spatial variance of their neighborhood instead.\n */\nconst varianceFragment = new FragmentShader({\n  name: 'jnlmVariance',\n  outputs: ['guide'],\n  uniforms: {\n    rawGuideTex: Shader.uniform('sampler2D')\n  },\n  main: glsl`\nvoid main() {\n  ivec2 p = ivec2(gl_FragCoord.xy);\n  ivec2 size = rawGuideTexRegion();\n  vec2 center = load_rawGuideTex(p, 0).xy;\n  float varianceSum = 0.0;\n  float varianceCount = 0.0;\n  float meanSum = 0.0;\n  float meanSquareSum = 0.0;\n  for (int dy = -1; dy <= 1; dy++) {\n    for (int dx = -1; dx <= 1; dx++) {\n      vec2 value = load_rawGuideTex(clamp(p + ivec2(dx, dy), ivec2(0), size - 1), 0).xy;\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  float variance;\n  if (center.y >= 0.0) {\n    variance = varianceSum / varianceCount;\n  } else {\n    float spatialMean = meanSum / 9.0;\n    variance = max(meanSquareSum / 9.0 - spatialMean * spatialMean, 0.0) * 9.0 / 8.0;\n  }\n  out_guide = vec4(center.x, variance, 0.0, 1.0);\n}`\n});\n\n/**\n * One joint NL-means pass; see createJNLMFilterWGSL. The first pass decodes\n * the display-encoded composition to linear light, the last encodes it again.\n */\nconst filterFragment = new FragmentShader({\n  name: 'jnlmFilter',\n  outputs: ['color', 'guide'],\n  defines: {\n    SEARCH_RADIUS: JNLM_DEFAULT_SHAPE.searchRadius,\n    PATCH_RADIUS: JNLM_DEFAULT_SHAPE.patchRadius,\n    INV_ALBEDO_BANDWIDTH2: (1 / JNLM_DEFAULT_SHAPE.albedoBandwidth ** 2).toFixed(6),\n    INV_NORMAL_BANDWIDTH2: (1 / JNLM_DEFAULT_SHAPE.normalBandwidth ** 2).toFixed(6)\n  },\n  uniforms: {\n    guideTex: Shader.uniform('sampler2D'),\n    colorTex: Shader.uniform('sampler2D'),\n    albedoTex: Shader.uniform('sampler2D'),\n    normalTex: Shader.uniform('sampler2D'),\n    shadowCatcherTex: Shader.uniform('sampler2D'),\n    colorSensitivity: Shader.uniform('float', JNLM_DEFAULT_COLOR_SENSITIVITY),\n    dilation: Shader.uniform('int', 1),\n    decodeInput: Shader.uniform('bool', 0),\n    encodeOutput: Shader.uniform('bool', 0)\n  },\n  main: glsl`\n#define PATCH_SIZE ((2 * PATCH_RADIUS + 1) * (2 * PATCH_RADIUS + 1))\n// Variance cancellation strength alpha of the paper.\n#define VARIANCE_CANCELLATION 1.0\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.\n#define CENTER_WEIGHT_MIN 0.01\n#define INV_COVERAGE_BANDWIDTH2 ${(1 / JNLM_COVERAGE_BANDWIDTH ** 2).toFixed(6)}\n\nivec2 imageSize;\n\nvec2 loadGuide(ivec2 p) {\n  return load_guideTex(clamp(p, ivec2(0), imageSize - 1), 0).xy;\n}\n\n// Premultiplied linear color.\nvec4 loadColor(ivec2 p) {\n  vec4 c = load_colorTex(p, 0);\n  if (decodeInput) c.rgb = mix(pow((max(c.rgb, 0.0) + 0.055) / 1.055, vec3(2.4)),\n    c.rgb / 12.92, lessThanEqual(c.rgb, vec3(0.04045)));\n  return vec4(c.rgb * c.a, c.a);\n}\n\nfloat featureDistance(ivec2 q, vec4 centerAlbedo, vec3 centerNormal) {\n  vec4 albedo = load_albedoTex(q, 0);\n  // Match WebGPU: protect relative RGB contrast, including dark textures,\n  // while keeping coverage differences in absolute units.\n  vec3 albedoScale = max((albedo.rgb + centerAlbedo.rgb) * 0.5, vec3(${JNLM_ALBEDO_FLOOR}));\n  vec3 albedoDelta = (albedo.rgb - centerAlbedo.rgb) / albedoScale;\n  float coverageDelta = albedo.a - centerAlbedo.a;\n  vec3 normalDelta = load_normalTex(q, 0).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\nvoid main() {\n  imageSize = guideTexRegion();\n  ivec2 p = ivec2(gl_FragCoord.xy);\n  vec4 centerAlbedo = load_albedoTex(p, 0);\n  vec4 centerNormalTexel = load_normalTex(p, 0);\n  vec3 centerNormal = centerNormalTexel.rgb;\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 && load_shadowCatcherTex(p, 0).r <= 0.0) {\n    vec4 c = loadColor(p);\n    vec3 rgb = c.rgb / max(c.a, 1e-6);\n    if (encodeOutput) rgb = mix(1.055 * pow(max(rgb, 0.0), vec3(1.0 / 2.4)) - 0.055,\n      rgb * 12.92, lessThanEqual(rgb, vec3(0.0031308)));\n    out_color = vec4(rgb, c.a);\n    out_guide = vec4(loadGuide(p), 0.0, 1.0);\n    return;\n  }\n\n  vec2 centerPatch[PATCH_SIZE];\n  int k = 0;\n  for (int dy = -PATCH_RADIUS; dy <= PATCH_RADIUS; dy++) {\n    for (int dx = -PATCH_RADIUS; dx <= PATCH_RADIUS; dx++) {\n      centerPatch[k] = loadGuide(p + ivec2(dx, dy) * dilation);\n      k++;\n    }\n  }\n  float k2 = colorSensitivity * colorSensitivity;\n\n  vec4 sum = vec4(0.0);\n  vec3 featureScale = max(centerAlbedo.rgb, vec3(${JNLM_ALBEDO_FLOOR}));\n  vec3 featureSum = vec3(0.0);\n  vec3 featureSquareSum = vec3(0.0);\n  vec3 featureColorSum = vec3(0.0);\n  float guideSum = 0.0;\n  float guideVarianceSum = 0.0;\n  float weightSum = 0.0;\n  float maxWeight = 0.0;\n  for (int sy = -SEARCH_RADIUS; sy <= SEARCH_RADIUS; sy++) {\n    for (int sx = -SEARCH_RADIUS; sx <= SEARCH_RADIUS; sx++) {\n      ivec2 q = p + ivec2(sx, sy) * dilation;\n      if (\n        (sx == 0 && sy == 0) ||\n        q.x < 0 || q.y < 0 || q.x >= imageSize.x || q.y >= imageSize.y\n      ) continue;\n      float features = featureDistance(q, centerAlbedo, centerNormal);\n      if (features > 16.0) continue;\n\n      float colorDistance = 0.0;\n      k = 0;\n      for (int dy = -PATCH_RADIUS; dy <= PATCH_RADIUS; dy++) {\n        for (int dx = -PATCH_RADIUS; dx <= PATCH_RADIUS; dx++) {\n          vec2 a = centerPatch[k];\n          vec2 b = loadGuide(q + ivec2(dx, dy) * dilation);\n          float 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 /= float(PATCH_SIZE);\n\n      float weight = exp(-max(max(colorDistance, 0.0), features));\n      vec2 g = loadGuide(q);\n      // Filter premultiplied color so coverage and color stay consistent.\n      vec3 a = (load_albedoTex(q, 0).rgb - centerAlbedo.rgb) / featureScale;\n      vec4 value = loadColor(q);\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  vec4 center = loadColor(p);\n  vec2 centerGuide = loadGuide(p);\n  float centerWeight = max(maxWeight, CENTER_WEIGHT_MIN);\n  float totalWeight = weightSum + centerWeight;\n  vec4 mean = (sum + center * 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  vec3 meanFeature = featureSum / totalWeight;\n  vec3 featureVariance = max(featureSquareSum / totalWeight - meanFeature * meanFeature, vec3(0.0));\n  vec3 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  vec3 meanAlbedo = centerAlbedo.rgb + meanFeature * featureScale;\n  vec3 maximumSlope = mean.rgb * featureScale / max(meanAlbedo, vec3(${JNLM_ALBEDO_FLOOR}));\n  vec3 slope = clamp(covariance / (featureVariance + vec3(${JNLM_ALBEDO_REGULARIZATION})), vec3(0.0), maximumSlope);\n  vec3 corrected = mean.rgb - slope * meanFeature;\n  vec4 filtered = vec4(max(corrected, vec3(0.0)), max(mean.a, 0.0));\n  vec3 rgb = filtered.a > 1e-6\n    ? filtered.rgb / max(filtered.a, 1e-6)\n    : center.rgb / max(center.a, 1e-6);\n  if (encodeOutput) rgb = mix(1.055 * pow(max(rgb, 0.0), vec3(1.0 / 2.4)) - 0.055,\n      rgb * 12.92, lessThanEqual(rgb, vec3(0.0031308)));\n  out_color = vec4(rgb, filtered.a);\n  out_guide = vec4(\n    (guideSum + centerGuide.x * centerWeight) / totalWeight,\n    (guideVarianceSum + centerGuide.y * centerWeight * centerWeight)\n      / (totalWeight * totalWeight),\n    0.0,\n    1.0\n  );\n}`\n});\n\nexport interface WebGLJNLMInputs {\n  width: number;\n  height: number;\n  /** Main trace pass moments: guide sum, sum of squares and count. */\n  moments: Texture2D;\n  /** Catcher trace pass moments, when the scene has a shadow catcher. */\n  catcherMoments?: Texture2D;\n  /** Display-encoded composition to filter. */\n  color: Texture2D;\n  albedo: Texture2D;\n  normal: Texture2D;\n  /** Normalized shadow-catcher mask; red is the catcher coverage. */\n  shadowCatcher: Texture2D;\n  strength: number;\n  shape: JNLMFilterShape;\n}\n\nconst floatOutput = {\n  type: constants.FLOAT,\n  minFilter: constants.NEAREST,\n  magFilter: constants.NEAREST\n};\n\nexport class WebGLJNLMDenoisePass {\n  private _renderer: WebGLDevice;\n  private _guidePass: WebGLFullQuadPass<'momentsTex' | 'catcherMomentsTex', 'guide'>;\n  private _variancePass: WebGLFullQuadPass<'rawGuideTex', 'guide'>;\n  // Two instances so consecutive passes ping-pong between their outputs.\n  private _filterPasses: WebGLFullQuadPass<\n    'guideTex' | 'colorTex' | 'albedoTex' | 'normalTex' | 'shadowCatcherTex',\n    'color' | 'guide'\n  >[];\n\n  constructor(renderer: WebGLDevice) {\n    this._renderer = renderer;\n    this._guidePass = new WebGLFullQuadPass('jnlmGuide', renderer, {\n      shader: guideFragment,\n      outputParams: { guide: floatOutput }\n    });\n    this._variancePass = new WebGLFullQuadPass('jnlmVariance', renderer, {\n      shader: varianceFragment,\n      outputParams: { guide: floatOutput }\n    });\n    this._filterPasses = [0, 1].map(() =>\n      new WebGLFullQuadPass('jnlmFilter', renderer, {\n        shader: filterFragment,\n        outputParams: {\n          color: { type: constants.HALF_FLOAT },\n          guide: floatOutput\n        }\n      })\n    );\n  }\n\n  /** Runs the denoiser and returns the display-encoded result. */\n  render(inputs: WebGLJNLMInputs, trace?: string[]): Texture2D {\n    const { width, height, shape } = inputs;\n    const passes = [this._guidePass, this._variancePass, ...this._filterPasses];\n    passes.forEach((pass) => pass.setSize(width, height));\n    this._guidePass.createPass({\n      momentsTex: inputs.moments,\n      catcherMomentsTex: inputs.catcherMoments ?? this._renderer.emptyTexture\n    });\n    this._variancePass.createPass({ rawGuideTex: this._guidePass.getOutput('guide') });\n\n    const sensitivity = JNLM_DEFAULT_COLOR_SENSITIVITY * Math.max(inputs.strength, 1e-3);\n    const iterations = Math.min(JNLM_MAX_ITERATIONS, Math.max(1, Math.round(shape.iterations)));\n    let guide = this._variancePass.getOutput('guide');\n    let color = inputs.color;\n    for (let i = 0; i < iterations; i++) {\n      const pass = this._filterPasses[i % 2];\n      pass.setDefines({\n        SEARCH_RADIUS: shape.searchRadius,\n        // Numeric zero removes a define in the WebGL shader builder.\n        PATCH_RADIUS: String(shape.patchRadius),\n        INV_ALBEDO_BANDWIDTH2: (1 / shape.albedoBandwidth ** 2).toFixed(6),\n        INV_NORMAL_BANDWIDTH2: (1 / shape.normalBandwidth ** 2).toFixed(6)\n      });\n      pass.setUniforms({\n        colorSensitivity: sensitivity,\n        dilation: jnlmPassDilation(i),\n        decodeInput: i === 0,\n        encodeOutput: i === iterations - 1\n      });\n      if (trace) {\n        if (i === 0) trace.push('Decode:DisplayLinear', 'Denoise:DisplayLinear');\n        if (i === iterations - 1) trace.push('Encode:DisplayEncoded');\n      }\n      pass.createPass({\n        guideTex: guide,\n        colorTex: color,\n        albedoTex: inputs.albedo,\n        normalTex: inputs.normal,\n        shadowCatcherTex: inputs.shadowCatcher\n      });\n      guide = pass.getOutput('guide');\n      color = pass.getOutput('color');\n    }\n    return color;\n  }\n\n  reserveCapacity() {\n    [this._guidePass, this._variancePass, ...this._filterPasses].forEach(pass => pass.reserveCapacity());\n  }\n\n  retainOutput(texture: Texture2D) { this._filterPasses.forEach(pass => pass.retainOutput(texture)); }\n  releaseOutput(texture: Texture2D) { this._filterPasses.forEach(pass => pass.releaseOutput(texture)); }\n\n  getMemoryBytes() {\n    return [this._guidePass, this._variancePass, ...this._filterPasses]\n      .reduce((sum, pass) => sum + pass.getMemoryBytes(), 0);\n  }\n\n  dispose() {\n    [this._guidePass, this._variancePass, ...this._filterPasses].forEach((pass) =>\n      pass.dispose()\n    );\n  }\n}\n"]}

package/lib/renderer/WebGLRayTraceOutputPass.js

@@ -42,11 +42,11 @@ vec4 mask = sample_shadowCatcherTex(v_Texcoord);
 vec4 background = sample_catcherBackgroundTex(v_Texcoord);
 float hit = sample_albedoTex(v_Texcoord).a;
 
-float qi = max(sample_normalTex(v_Texcoord).a, 1.0) / sampleCount;
-float iV = max(hit - mask.b * qi, 0.0);
+float qZ = max(sample_normalTex(v_Texcoord).a, 1.0) / sampleCount;
+float iT = max(hit - mask.b * qZ, 0.0);
 out_foreground = vec4(
-(light.rgb - background.rgb * qi)
-/ (iV > 0.0 ? iV : sampleCount),
+(light.rgb - background.rgb * qZ)
+/ (iT > 0.0 ? iT : sampleCount),
 clamp((light.a - background.a) / sampleCount, 0.0, 1.0)
 );
 out_catcherBackground = background / sampleCount;
@@ -75,6 +75,9 @@ out_catcherBackground = background / sampleCount;
             normalTex: rayTracePass.getOutput('normal'),
             shadowCatcherTex: rayTracePass.hasShadowCatcher()
                 ? rayTracePass.getOutput('shadowCatcher')
+                : this._renderer.emptyTexture,
+            catcherMomentsTex: rayTracePass.hasShadowCatcher()
+                ? rayTracePass.getOutput('catcherMoments')
                 : this._renderer.emptyTexture
         }, undefined, { clear: true });
         if (rayTracePass.hasShadowCatcher()) {

package/lib/renderer/WebGLRayTraceOutputPass.js.map

@@ -1 +1 @@
-{"version":3,"file":"WebGLRayTraceOutputPass.js","sourceRoot":"","sources":["../../src/renderer/WebGLRayTraceOutputPass.ts"],"names":[],"mappings":"AAAA,OAAO,EAAE,cAAc,EAAE,MAAM,EAAE,IAAI,EAAE,MAAM,aAAa,CAAC;AAC3D,OAAO,EAAE,SAAS,EAA0B,MAAM,YAAY,CAAC;AAC/D,OAAO,uBAAuB,MAAM,wCAAwC,CAAC;AAC7E,OAAO,EAAE,iBAAiB,EAAE,MAAM,wBAAwB,CAAC;AAW3D,+EAA+E;AAC/E,MAAM,OAAO,uBAAuB;IACjB,SAAS,CAAc;IACvB,KAAK,CAAmD;IAExD,WAAW,CAE1B;IAEF,YAAY,QAAqB;QAC/B,IAAI,CAAC,SAAS,GAAG,QAAQ,CAAC;QAC1B,IAAI,CAAC,KAAK,GAAG,IAAI,iBAAiB,CAAC,gBAAgB,EAAE,QAAQ,EAAE;YAC7D,MAAM,EAAE,uBAAuB;YAC/B,OAAO,EAAE,CAAC,OAAO,EAAE,QAAQ,EAAE,QAAQ,EAAE,eAAe,CAAC;YACvD,YAAY,EAAE;gBACZ,KAAK,EAAE;oBACL,IAAI,EAAE,SAAS,CAAC,UAAU;oBAC1B,SAAS,EAAE,SAAS,CAAC,MAAM;oBAC3B,SAAS,EAAE,SAAS,CAAC,MAAM;iBAC5B;gBACD,MAAM,EAAE,EAAE,IAAI,EAAE,SAAS,CAAC,aAAa,EAAE;gBACzC,MAAM,EAAE,EAAE,IAAI,EAAE,SAAS,CAAC,aAAa,EAAE;gBACzC,aAAa,EAAE,EAAE,IAAI,EAAE,SAAS,CAAC,UAAU,EAAE;aAC9C;SACF,CAAC,CAAC;QACH,IAAI,CAAC,WAAW,GAAG,IAAI,iBAAiB,CAAC,gBAAgB,EAAE,QAAQ,EAAE;YACnE,MAAM,EAAE,IAAI,cAAc,CAAC;gBACzB,IAAI,EAAE,yBAAyB;gBAC/B,OAAO,EAAE,CAAC,YAAY,EAAE,mBAAmB,CAAC;gBAC5C,QAAQ,EAAE;oBACR,QAAQ,EAAE,MAAM,CAAC,OAAO,CAAC,WAAW,CAAC;oBACrC,SAAS,EAAE,MAAM,CAAC,OAAO,CAAC,WAAW,CAAC;oBACtC,SAAS,EAAE,MAAM,CAAC,OAAO,CAAC,WAAW,CAAC;oBACtC,gBAAgB,EAAE,MAAM,CAAC,OAAO,CAAC,WAAW,CAAC;oBAC7C,oBAAoB,EAAE,MAAM,CAAC,OAAO,CAAC,WAAW,CAAC;oBACjD,WAAW,EAAE,MAAM,CAAC,OAAO,CAAC,OAAO,EAAE,CAAC,CAAC;iBACxC;gBACD,IAAI,EAAE,IAAI,CAAA;;;;;;;;;;;;;;;EAehB;aACK,CAAC;YACF,OAAO,EAAE,CAAC,YAAY,EAAE,mBAAmB,CAAC;YAC5C,YAAY,EAAE;gBACZ,UAAU,EAAE,EAAE,IAAI,EAAE,SAAS,CAAC,UAAU,EAAE;gBAC1C,iBAAiB,EAAE,EAAE,IAAI,EAAE,SAAS,CAAC,UAAU,EAAE;aAClD;SACF,CAAC,CAAC;IACL,CAAC;IAED,OAAO,CAAC,KAAa,EAAE,MAAc;QACnC,IAAI,CAAC,KAAK,CAAC,OAAO,CAAC,KAAK,EAAE,MAAM,CAAC,CAAC;QAClC,IAAI,CAAC,WAAW,CAAC,OAAO,CAAC,KAAK,EAAE,MAAM,CAAC,CAAC;IAC1C,CAAC;IAED,cAAc,CAAC,WAAmB;QAChC,IAAI,CAAC,KAAK,CAAC,UAAU,CAAC,aAAa,EAAE,IAAI,CAAC,GAAG,CAAC,WAAW,EAAE,CAAC,CAAC,CAAC,CAAC;QAC/D,IAAI,CAAC,WAAW,CAAC,UAAU,CAAC,aAAa,EAAE,IAAI,CAAC,GAAG,CAAC,WAAW,EAAE,CAAC,CAAC,CAAC,CAAC;IACvE,CAAC;IAED,UAAU,CAAC,YAA+B;QACxC,IAAI,CAAC,KAAK,CAAC,UAAU,CAAC,YAAY,CAAC,gBAAgB,EAAE,CAAC,CAAC,CAAC,CAAC,OAAO,EAAE,QAAQ,EAAE,QAAQ,EAAE,eAAe,CAAC,CAAC,CAAC,CAAC,CAAC,OAAO,EAAE,QAAQ,EAAE,QAAQ,CAAC,CAAC,CAAC;QACxI,IAAI,CAAC,KAAK,CAAC,UAAU,CACnB;YACE,QAAQ,EAAE,YAAY,CAAC,SAAS,CAAC,OAAO,CAAC;YACzC,SAAS,EAAE,YAAY,CAAC,SAAS,CAAC,QAAQ,CAAC;YAC3C,SAAS,EAAE,YAAY,CAAC,SAAS,CAAC,QAAQ,CAAC;YAC3C,gBAAgB,EAAE,YAAY,CAAC,gBAAgB,EAAE;gBAC/C,CAAC,CAAC,YAAY,CAAC,SAAS,CAAC,eAAe,CAAC;gBACzC,CAAC,CAAC,IAAI,CAAC,SAAS,CAAC,YAAY;SAChC,EACD,SAAS,EACT,EAAE,KAAK,EAAE,IAAI,EAAE,CAChB,CAAC;QACF,IAAI,YAAY,CAAC,gBAAgB,EAAE,EAAE,CAAC;YACpC,IAAI,CAAC,WAAW,CAAC,UAAU,CAAC,CAAC,YAAY,EAAE,mBAAmB,CAAC,CAAC,CAAC;YACjE,IAAI,CAAC,WAAW,CAAC,UAAU,CAAC;gBAC1B,QAAQ,EAAE,YAAY,CAAC,SAAS,CAAC,OAAO,CAAC;gBACzC,SAAS,EAAE,YAAY,CAAC,SAAS,CAAC,QAAQ,CAAC;gBAC3C,SAAS,EAAE,YAAY,CAAC,SAAS,CAAC,QAAQ,CAAC;gBAC3C,gBAAgB,EAAE,YAAY,CAAC,SAAS,CAAC,eAAe,CAAC;gBACzD,oBAAoB,EAAE,YAAY,CAAC,SAAS,CAAC,mBAAmB,CAAC;aAClE,EAAE,SAAS,EAAE,EAAE,KAAK,EAAE,IAAI,EAAE,CAAC,CAAC;QACjC,CAAC;aAAM,CAAC;YACN,IAAI,CAAC,WAAW,CAAC,UAAU,CAAC,EAAE,CAAC,CAAC;QAClC,CAAC;IACH,CAAC;IAED,SAAS,CAAC,KAAmC;QAC3C,OAAO,KAAK,KAAK,YAAY,IAAI,KAAK,KAAK,mBAAmB;YAC5D,CAAC,CAAC,IAAI,CAAC,WAAW,CAAC,SAAS,CAAC,KAAK,CAAC,IAAI,IAAI,CAAC,SAAS,CAAC,YAAY,CAAC,CAAC,CAAC,IAAI,CAAC,KAAK,CAAC,SAAS,CAAC,KAAK,CAAC,IAAI,IAAI,CAAC,SAAS,CAAC,YAAY,CAAC;IACpI,CAAC;IAED,YAAY,CAAC,OAAkB,IAAI,IAAI,CAAC,KAAK,CAAC,YAAY,CAAC,OAAO,CAAC,CAAC,CAAC,IAAI,CAAC,WAAW,CAAC,YAAY,CAAC,OAAO,CAAC,CAAC,CAAC,CAAC;IAC9G,aAAa,CAAC,OAAkB,IAAI,IAAI,CAAC,KAAK,CAAC,aAAa,CAAC,OAAO,CAAC,CAAC,CAAC,IAAI,CAAC,WAAW,CAAC,aAAa,CAAC,OAAO,CAAC,CAAC,CAAC,CAAC;IAEjH,cAAc,KAAK,OAAO,IAAI,CAAC,KAAK,CAAC,cAAc,EAAE,GAAG,IAAI,CAAC,WAAW,CAAC,cAAc,EAAE,CAAC,CAAC,CAAC;IAE5F,OAAO;QACL,IAAI,CAAC,KAAK,CAAC,OAAO,EAAE,CAAC;QACrB,IAAI,CAAC,WAAW,CAAC,OAAO,EAAE,CAAC;IAC7B,CAAC;CACF","sourcesContent":["import { FragmentShader, Shader, glsl } from './shader.js';\nimport { constants, WebGLDevice, Texture2D } from './webgl.js';\nimport normalizeRayTraceOutput from './glsl/normalizeRayTraceOutput.frag.js';\nimport { WebGLFullQuadPass } from './WebGLFullQuadPass.js';\nimport WebGLRayTracePass from './WebGLRayTracePass.js';\n\ntype RayTraceInput =\n  | 'colorTex'\n  | 'albedoTex'\n  | 'normalTex'\n  | 'shadowCatcherTex';\ntype LayerOutput = 'foreground' | 'catcherBackground';\ntype RayTraceOutput = 'color' | 'albedo' | 'normal' | 'shadowCatcher';\n\n/** Converts the additive ray tracing buffers into display/denoise textures. */\nexport class WebGLRayTraceOutputPass {\n  private readonly _renderer: WebGLDevice;\n  private readonly _pass: WebGLFullQuadPass<RayTraceInput, RayTraceOutput>;\n\n  private readonly _layersPass: WebGLFullQuadPass<\n    RayTraceInput | 'catcherBackgroundTex', LayerOutput\n  >;\n\n  constructor(renderer: WebGLDevice) {\n    this._renderer = renderer;\n    this._pass = new WebGLFullQuadPass('rayTraceOutput', renderer, {\n      shader: normalizeRayTraceOutput,\n      outputs: ['color', 'albedo', 'normal', 'shadowCatcher'],\n      outputParams: {\n        color: {\n          type: constants.HALF_FLOAT,\n          minFilter: constants.LINEAR,\n          magFilter: constants.LINEAR\n        },\n        albedo: { type: constants.UNSIGNED_BYTE },\n        normal: { type: constants.UNSIGNED_BYTE },\n        shadowCatcher: { type: constants.HALF_FLOAT }\n      }\n    });\n    this._layersPass = new WebGLFullQuadPass('rayTraceLayers', renderer, {\n      shader: new FragmentShader({\n        name: 'normalizeRayTraceLayers',\n        outputs: ['foreground', 'catcherBackground'],\n        uniforms: {\n          colorTex: Shader.uniform('sampler2D'),\n          albedoTex: Shader.uniform('sampler2D'),\n          normalTex: Shader.uniform('sampler2D'),\n          shadowCatcherTex: Shader.uniform('sampler2D'),\n          catcherBackgroundTex: Shader.uniform('sampler2D'),\n          sampleCount: Shader.uniform('float', 1)\n        },\n        main: glsl`\nvoid main() {\n  vec4 light = sample_colorTex(v_Texcoord);\n  vec4 mask = sample_shadowCatcherTex(v_Texcoord);\n  vec4 background = sample_catcherBackgroundTex(v_Texcoord);\n  float hit = sample_albedoTex(v_Texcoord).a;\n  // Full spectral beauty has three wavelength contributions per shadow sample.\n  float contributions = max(sample_normalTex(v_Texcoord).a, 1.0) / sampleCount;\n  float foregroundHit = max(hit - mask.b * contributions, 0.0);\n  out_foreground = vec4(\n    (light.rgb - background.rgb * contributions)\n      / (foregroundHit > 0.0 ? foregroundHit : sampleCount),\n    clamp((light.a - background.a) / sampleCount, 0.0, 1.0)\n  );\n  out_catcherBackground = background / sampleCount;\n}`\n      }),\n      outputs: ['foreground', 'catcherBackground'],\n      outputParams: {\n        foreground: { type: constants.HALF_FLOAT },\n        catcherBackground: { type: constants.HALF_FLOAT }\n      }\n    });\n  }\n\n  setSize(width: number, height: number) {\n    this._pass.setSize(width, height);\n    this._layersPass.setSize(width, height);\n  }\n\n  setSampleCount(sampleCount: number) {\n    this._pass.setUniform('sampleCount', Math.max(sampleCount, 1));\n    this._layersPass.setUniform('sampleCount', Math.max(sampleCount, 1));\n  }\n\n  createPass(rayTracePass: WebGLRayTracePass) {\n    this._pass.setOutputs(rayTracePass.hasShadowCatcher() ? ['color', 'albedo', 'normal', 'shadowCatcher'] : ['color', 'albedo', 'normal']);\n    this._pass.createPass(\n      {\n        colorTex: rayTracePass.getOutput('color'),\n        albedoTex: rayTracePass.getOutput('albedo'),\n        normalTex: rayTracePass.getOutput('normal'),\n        shadowCatcherTex: rayTracePass.hasShadowCatcher()\n          ? rayTracePass.getOutput('shadowCatcher')\n          : this._renderer.emptyTexture\n      },\n      undefined,\n      { clear: true }\n    );\n    if (rayTracePass.hasShadowCatcher()) {\n      this._layersPass.setOutputs(['foreground', 'catcherBackground']);\n      this._layersPass.createPass({\n        colorTex: rayTracePass.getOutput('color'),\n        albedoTex: rayTracePass.getOutput('albedo'),\n        normalTex: rayTracePass.getOutput('normal'),\n        shadowCatcherTex: rayTracePass.getOutput('shadowCatcher'),\n        catcherBackgroundTex: rayTracePass.getOutput('catcherBackground')\n      }, undefined, { clear: true });\n    } else {\n      this._layersPass.setOutputs([]);\n    }\n  }\n\n  getOutput(label: RayTraceOutput | LayerOutput): Texture2D {\n    return label === 'foreground' || label === 'catcherBackground'\n      ? this._layersPass.getOutput(label) ?? this._renderer.emptyTexture : this._pass.getOutput(label) ?? this._renderer.emptyTexture;\n  }\n\n  retainOutput(texture: Texture2D) { this._pass.retainOutput(texture); this._layersPass.retainOutput(texture); }\n  releaseOutput(texture: Texture2D) { this._pass.releaseOutput(texture); this._layersPass.releaseOutput(texture); }\n\n  getMemoryBytes() { return this._pass.getMemoryBytes() + this._layersPass.getMemoryBytes(); }\n\n  dispose() {\n    this._pass.dispose();\n    this._layersPass.dispose();\n  }\n}\n\n"]}
+{"version":3,"file":"WebGLRayTraceOutputPass.js","sourceRoot":"","sources":["../../src/renderer/WebGLRayTraceOutputPass.ts"],"names":[],"mappings":"AAAA,OAAO,EAAE,cAAc,EAAE,MAAM,EAAE,IAAI,EAAE,MAAM,aAAa,CAAC;AAC3D,OAAO,EAAE,SAAS,EAA0B,MAAM,YAAY,CAAC;AAC/D,OAAO,uBAAuB,MAAM,wCAAwC,CAAC;AAC7E,OAAO,EAAE,iBAAiB,EAAE,MAAM,wBAAwB,CAAC;AAW3D,+EAA+E;AAC/E,MAAM,OAAO,uBAAuB;IACjB,SAAS,CAAc;IACvB,KAAK,CAAyE;IAE9E,WAAW,CAE1B;IAEF,YAAY,QAAqB;QAC/B,IAAI,CAAC,SAAS,GAAG,QAAQ,CAAC;QAC1B,IAAI,CAAC,KAAK,GAAG,IAAI,iBAAiB,CAAC,gBAAgB,EAAE,QAAQ,EAAE;YAC7D,MAAM,EAAE,uBAAuB;YAC/B,OAAO,EAAE,CAAC,OAAO,EAAE,QAAQ,EAAE,QAAQ,EAAE,eAAe,CAAC;YACvD,YAAY,EAAE;gBACZ,KAAK,EAAE;oBACL,IAAI,EAAE,SAAS,CAAC,UAAU;oBAC1B,SAAS,EAAE,SAAS,CAAC,MAAM;oBAC3B,SAAS,EAAE,SAAS,CAAC,MAAM;iBAC5B;gBACD,MAAM,EAAE,EAAE,IAAI,EAAE,SAAS,CAAC,aAAa,EAAE;gBACzC,MAAM,EAAE,EAAE,IAAI,EAAE,SAAS,CAAC,aAAa,EAAE;gBACzC,aAAa,EAAE,EAAE,IAAI,EAAE,SAAS,CAAC,UAAU,EAAE;aAC9C;SACF,CAAC,CAAC;QACH,IAAI,CAAC,WAAW,GAAG,IAAI,iBAAiB,CAAC,gBAAgB,EAAE,QAAQ,EAAE;YACnE,MAAM,EAAE,IAAI,cAAc,CAAC;gBACzB,IAAI,EAAE,yBAAyB;gBAC/B,OAAO,EAAE,CAAC,YAAY,EAAE,mBAAmB,CAAC;gBAC5C,QAAQ,EAAE;oBACR,QAAQ,EAAE,MAAM,CAAC,OAAO,CAAC,WAAW,CAAC;oBACrC,SAAS,EAAE,MAAM,CAAC,OAAO,CAAC,WAAW,CAAC;oBACtC,SAAS,EAAE,MAAM,CAAC,OAAO,CAAC,WAAW,CAAC;oBACtC,gBAAgB,EAAE,MAAM,CAAC,OAAO,CAAC,WAAW,CAAC;oBAC7C,oBAAoB,EAAE,MAAM,CAAC,OAAO,CAAC,WAAW,CAAC;oBACjD,WAAW,EAAE,MAAM,CAAC,OAAO,CAAC,OAAO,EAAE,CAAC,CAAC;iBACxC;gBACD,IAAI,EAAE,IAAI,CAAA;;;;;;;;;;;;;;;EAehB;aACK,CAAC;YACF,OAAO,EAAE,CAAC,YAAY,EAAE,mBAAmB,CAAC;YAC5C,YAAY,EAAE;gBACZ,UAAU,EAAE,EAAE,IAAI,EAAE,SAAS,CAAC,UAAU,EAAE;gBAC1C,iBAAiB,EAAE,EAAE,IAAI,EAAE,SAAS,CAAC,UAAU,EAAE;aAClD;SACF,CAAC,CAAC;IACL,CAAC;IAED,OAAO,CAAC,KAAa,EAAE,MAAc;QACnC,IAAI,CAAC,KAAK,CAAC,OAAO,CAAC,KAAK,EAAE,MAAM,CAAC,CAAC;QAClC,IAAI,CAAC,WAAW,CAAC,OAAO,CAAC,KAAK,EAAE,MAAM,CAAC,CAAC;IAC1C,CAAC;IAED,cAAc,CAAC,WAAmB;QAChC,IAAI,CAAC,KAAK,CAAC,UAAU,CAAC,aAAa,EAAE,IAAI,CAAC,GAAG,CAAC,WAAW,EAAE,CAAC,CAAC,CAAC,CAAC;QAC/D,IAAI,CAAC,WAAW,CAAC,UAAU,CAAC,aAAa,EAAE,IAAI,CAAC,GAAG,CAAC,WAAW,EAAE,CAAC,CAAC,CAAC,CAAC;IACvE,CAAC;IAED,UAAU,CAAC,YAA+B;QACxC,IAAI,CAAC,KAAK,CAAC,UAAU,CAAC,YAAY,CAAC,gBAAgB,EAAE,CAAC,CAAC,CAAC,CAAC,OAAO,EAAE,QAAQ,EAAE,QAAQ,EAAE,eAAe,CAAC,CAAC,CAAC,CAAC,CAAC,OAAO,EAAE,QAAQ,EAAE,QAAQ,CAAC,CAAC,CAAC;QACxI,IAAI,CAAC,KAAK,CAAC,UAAU,CACnB;YACE,QAAQ,EAAE,YAAY,CAAC,SAAS,CAAC,OAAO,CAAC;YACzC,SAAS,EAAE,YAAY,CAAC,SAAS,CAAC,QAAQ,CAAC;YAC3C,SAAS,EAAE,YAAY,CAAC,SAAS,CAAC,QAAQ,CAAC;YAC3C,gBAAgB,EAAE,YAAY,CAAC,gBAAgB,EAAE;gBAC/C,CAAC,CAAC,YAAY,CAAC,SAAS,CAAC,eAAe,CAAC;gBACzC,CAAC,CAAC,IAAI,CAAC,SAAS,CAAC,YAAY;YAC/B,iBAAiB,EAAE,YAAY,CAAC,gBAAgB,EAAE;gBAChD,CAAC,CAAC,YAAY,CAAC,SAAS,CAAC,gBAAgB,CAAC;gBAC1C,CAAC,CAAC,IAAI,CAAC,SAAS,CAAC,YAAY;SAChC,EACD,SAAS,EACT,EAAE,KAAK,EAAE,IAAI,EAAE,CAChB,CAAC;QACF,IAAI,YAAY,CAAC,gBAAgB,EAAE,EAAE,CAAC;YACpC,IAAI,CAAC,WAAW,CAAC,UAAU,CAAC,CAAC,YAAY,EAAE,mBAAmB,CAAC,CAAC,CAAC;YACjE,IAAI,CAAC,WAAW,CAAC,UAAU,CAAC;gBAC1B,QAAQ,EAAE,YAAY,CAAC,SAAS,CAAC,OAAO,CAAC;gBACzC,SAAS,EAAE,YAAY,CAAC,SAAS,CAAC,QAAQ,CAAC;gBAC3C,SAAS,EAAE,YAAY,CAAC,SAAS,CAAC,QAAQ,CAAC;gBAC3C,gBAAgB,EAAE,YAAY,CAAC,SAAS,CAAC,eAAe,CAAC;gBACzD,oBAAoB,EAAE,YAAY,CAAC,SAAS,CAAC,mBAAmB,CAAC;aAClE,EAAE,SAAS,EAAE,EAAE,KAAK,EAAE,IAAI,EAAE,CAAC,CAAC;QACjC,CAAC;aAAM,CAAC;YACN,IAAI,CAAC,WAAW,CAAC,UAAU,CAAC,EAAE,CAAC,CAAC;QAClC,CAAC;IACH,CAAC;IAED,SAAS,CAAC,KAAmC;QAC3C,OAAO,KAAK,KAAK,YAAY,IAAI,KAAK,KAAK,mBAAmB;YAC5D,CAAC,CAAC,IAAI,CAAC,WAAW,CAAC,SAAS,CAAC,KAAK,CAAC,IAAI,IAAI,CAAC,SAAS,CAAC,YAAY,CAAC,CAAC,CAAC,IAAI,CAAC,KAAK,CAAC,SAAS,CAAC,KAAK,CAAC,IAAI,IAAI,CAAC,SAAS,CAAC,YAAY,CAAC;IACpI,CAAC;IAED,YAAY,CAAC,OAAkB,IAAI,IAAI,CAAC,KAAK,CAAC,YAAY,CAAC,OAAO,CAAC,CAAC,CAAC,IAAI,CAAC,WAAW,CAAC,YAAY,CAAC,OAAO,CAAC,CAAC,CAAC,CAAC;IAC9G,aAAa,CAAC,OAAkB,IAAI,IAAI,CAAC,KAAK,CAAC,aAAa,CAAC,OAAO,CAAC,CAAC,CAAC,IAAI,CAAC,WAAW,CAAC,aAAa,CAAC,OAAO,CAAC,CAAC,CAAC,CAAC;IAEjH,cAAc,KAAK,OAAO,IAAI,CAAC,KAAK,CAAC,cAAc,EAAE,GAAG,IAAI,CAAC,WAAW,CAAC,cAAc,EAAE,CAAC,CAAC,CAAC;IAE5F,OAAO;QACL,IAAI,CAAC,KAAK,CAAC,OAAO,EAAE,CAAC;QACrB,IAAI,CAAC,WAAW,CAAC,OAAO,EAAE,CAAC;IAC7B,CAAC;CACF","sourcesContent":["import { FragmentShader, Shader, glsl } from './shader.js';\nimport { constants, WebGLDevice, Texture2D } from './webgl.js';\nimport normalizeRayTraceOutput from './glsl/normalizeRayTraceOutput.frag.js';\nimport { WebGLFullQuadPass } from './WebGLFullQuadPass.js';\nimport WebGLRayTracePass from './WebGLRayTracePass.js';\n\ntype RayTraceInput =\n  | 'colorTex'\n  | 'albedoTex'\n  | 'normalTex'\n  | 'shadowCatcherTex';\ntype LayerOutput = 'foreground' | 'catcherBackground';\ntype RayTraceOutput = 'color' | 'albedo' | 'normal' | 'shadowCatcher';\n\n/** Converts the additive ray tracing buffers into display/denoise textures. */\nexport class WebGLRayTraceOutputPass {\n  private readonly _renderer: WebGLDevice;\n  private readonly _pass: WebGLFullQuadPass<RayTraceInput | 'catcherMomentsTex', RayTraceOutput>;\n\n  private readonly _layersPass: WebGLFullQuadPass<\n    RayTraceInput | 'catcherBackgroundTex', LayerOutput\n  >;\n\n  constructor(renderer: WebGLDevice) {\n    this._renderer = renderer;\n    this._pass = new WebGLFullQuadPass('rayTraceOutput', renderer, {\n      shader: normalizeRayTraceOutput,\n      outputs: ['color', 'albedo', 'normal', 'shadowCatcher'],\n      outputParams: {\n        color: {\n          type: constants.HALF_FLOAT,\n          minFilter: constants.LINEAR,\n          magFilter: constants.LINEAR\n        },\n        albedo: { type: constants.UNSIGNED_BYTE },\n        normal: { type: constants.UNSIGNED_BYTE },\n        shadowCatcher: { type: constants.HALF_FLOAT }\n      }\n    });\n    this._layersPass = new WebGLFullQuadPass('rayTraceLayers', renderer, {\n      shader: new FragmentShader({\n        name: 'normalizeRayTraceLayers',\n        outputs: ['foreground', 'catcherBackground'],\n        uniforms: {\n          colorTex: Shader.uniform('sampler2D'),\n          albedoTex: Shader.uniform('sampler2D'),\n          normalTex: Shader.uniform('sampler2D'),\n          shadowCatcherTex: Shader.uniform('sampler2D'),\n          catcherBackgroundTex: Shader.uniform('sampler2D'),\n          sampleCount: Shader.uniform('float', 1)\n        },\n        main: glsl`\nvoid main() {\n  vec4 light = sample_colorTex(v_Texcoord);\n  vec4 mask = sample_shadowCatcherTex(v_Texcoord);\n  vec4 background = sample_catcherBackgroundTex(v_Texcoord);\n  float hit = sample_albedoTex(v_Texcoord).a;\n  // Full spectral beauty has three wavelength contributions per shadow sample.\n  float contributions = max(sample_normalTex(v_Texcoord).a, 1.0) / sampleCount;\n  float foregroundHit = max(hit - mask.b * contributions, 0.0);\n  out_foreground = vec4(\n    (light.rgb - background.rgb * contributions)\n      / (foregroundHit > 0.0 ? foregroundHit : sampleCount),\n    clamp((light.a - background.a) / sampleCount, 0.0, 1.0)\n  );\n  out_catcherBackground = background / sampleCount;\n}`\n      }),\n      outputs: ['foreground', 'catcherBackground'],\n      outputParams: {\n        foreground: { type: constants.HALF_FLOAT },\n        catcherBackground: { type: constants.HALF_FLOAT }\n      }\n    });\n  }\n\n  setSize(width: number, height: number) {\n    this._pass.setSize(width, height);\n    this._layersPass.setSize(width, height);\n  }\n\n  setSampleCount(sampleCount: number) {\n    this._pass.setUniform('sampleCount', Math.max(sampleCount, 1));\n    this._layersPass.setUniform('sampleCount', Math.max(sampleCount, 1));\n  }\n\n  createPass(rayTracePass: WebGLRayTracePass) {\n    this._pass.setOutputs(rayTracePass.hasShadowCatcher() ? ['color', 'albedo', 'normal', 'shadowCatcher'] : ['color', 'albedo', 'normal']);\n    this._pass.createPass(\n      {\n        colorTex: rayTracePass.getOutput('color'),\n        albedoTex: rayTracePass.getOutput('albedo'),\n        normalTex: rayTracePass.getOutput('normal'),\n        shadowCatcherTex: rayTracePass.hasShadowCatcher()\n          ? rayTracePass.getOutput('shadowCatcher')\n          : this._renderer.emptyTexture,\n        catcherMomentsTex: rayTracePass.hasShadowCatcher()\n          ? rayTracePass.getOutput('catcherMoments')\n          : this._renderer.emptyTexture\n      },\n      undefined,\n      { clear: true }\n    );\n    if (rayTracePass.hasShadowCatcher()) {\n      this._layersPass.setOutputs(['foreground', 'catcherBackground']);\n      this._layersPass.createPass({\n        colorTex: rayTracePass.getOutput('color'),\n        albedoTex: rayTracePass.getOutput('albedo'),\n        normalTex: rayTracePass.getOutput('normal'),\n        shadowCatcherTex: rayTracePass.getOutput('shadowCatcher'),\n        catcherBackgroundTex: rayTracePass.getOutput('catcherBackground')\n      }, undefined, { clear: true });\n    } else {\n      this._layersPass.setOutputs([]);\n    }\n  }\n\n  getOutput(label: RayTraceOutput | LayerOutput): Texture2D {\n    return label === 'foreground' || label === 'catcherBackground'\n      ? this._layersPass.getOutput(label) ?? this._renderer.emptyTexture : this._pass.getOutput(label) ?? this._renderer.emptyTexture;\n  }\n\n  retainOutput(texture: Texture2D) { this._pass.retainOutput(texture); this._layersPass.retainOutput(texture); }\n  releaseOutput(texture: Texture2D) { this._pass.releaseOutput(texture); this._layersPass.releaseOutput(texture); }\n\n  getMemoryBytes() { return this._pass.getMemoryBytes() + this._layersPass.getMemoryBytes(); }\n\n  dispose() {\n    this._pass.dispose();\n    this._layersPass.dispose();\n  }\n}\n\n"]}

package/lib/renderer/WebGLRayTracePass.d.ts

@@ -1,14 +1,21 @@
+import { WebGLAov } from './WebGLAov.js';
+import { type AovKind } from './aov.js';
 import { type SamplingPattern } from './sampling.js';
 import { type DepthSampleArea } from './depthQuery.js';
 import { WebGLDevice as WebGLRenderer, Texture2D as WebGLTexture2D } from './webgl.js';
 import Mesh from '../scene/Mesh.js';
 import OpenPBRMaterial from '../scene/OpenPBRMaterial.js';
-import { SceneBackground, SceneLight } from '../scene/Scene.js';
+import type { SceneBackground, SceneLight } from '../scene/Scene.js';
 import Camera from '../scene/Camera.js';
 import Geometry from '../scene/Geometry.js';
 declare class WebGLRayTracePass {
+    private _aovPrimitiveCache;
     private readonly _renderer;
     private _meshes;
+    private _emissiveActive;
+    private _directLightLegacy?;
+    private _emissiveCache;
+    private _emissiveTex;
     private _sceneEnvironment;
     private _rectLightTextures;
     private _analyticLights;
@@ -44,12 +51,14 @@ declare class WebGLRayTracePass {
     private _envMap;
     private _envMapDistribution;
     private _envImageSource?;
+    private readonly _skySource;
     private _backgroundTex;
     private _hasEnvMap;
     private _hasBackground;
     private _groundedBackgroundParams;
     private _groundedBackgroundCenter;
     private _backgroundMapTransform;
+    private _backgroundMapRotation;
     private _bounces;
     private _bevelSamples;
     private _clamping?;
@@ -74,6 +83,8 @@ declare class WebGLRayTracePass {
     private _viewProjInv;
     private _apertureRadius;
     private _focusDistance;
+    private _aovPass;
+    private _aovCenter;
     private _tracePasses;
     private readonly _depthPass;
     private readonly _depthAbort;
@@ -81,6 +92,7 @@ declare class WebGLRayTracePass {
     private _hasShadowCatcher;
     private _spectral;
     constructor(renderer: WebGLRenderer);
+    preparePrograms(): boolean;
     createPass(clear: boolean, tile: {
         x: number;
         y: number;
@@ -105,7 +117,7 @@ declare class WebGLRayTracePass {
         buffer: ArrayBuffer;
     };
     getDiagnostics(): {
-        textureAtlas: import("./MaterialTable.js").TextureAtlasStats | undefined;
+        textureAtlas: import("./textureAtlas.js").TextureAtlasStats | undefined;
     };
     private _updateMeshData;
     private _updateMaterialConstants;
@@ -128,8 +140,9 @@ declare class WebGLRayTracePass {
         bevel: boolean;
         dispersionWavelengths: 1 | 3;
         directLightCandidates: number;
+        caustics?: import('../Render.js').CausticSettings;
     }): void;
-    setWavelength(wavelength: number): void;
+    setSpectralChannel(channel: number): void;
     setSampleCount(sampleCount: number): void;
     isInvalid(): boolean;
     hasShadowCatcher(): boolean;
@@ -137,6 +150,8 @@ declare class WebGLRayTracePass {
     prepare(): void;
     setSamplingPattern(pattern: SamplingPattern, maxSamples: number, sequenceSeed?: number): boolean;
     setSampleSequenceOffset(_offset: number): void;
+    prepareAovs(state: WebGLAov, camera: Camera, width: number, height: number, kinds: readonly AovKind[]): boolean;
+    renderAovs(state: WebGLAov, camera: Camera, width: number, height: number, samples: number, cap: number, kinds: readonly AovKind[], sequenceOffset?: number): boolean;
     readDepthSamples(area: DepthSampleArea, aspect?: number): Promise<Float32Array | undefined>;
     dispose(): void;
 }