@meimu/mutracer 0.1.5 → 0.3.0

package/lib/renderer/WGPUComputePass.js.map

@@ -1 +1 @@
-{"version":3,"file":"WGPUComputePass.js","sourceRoot":"","sources":["../../src/renderer/WGPUComputePass.ts"],"names":[],"mappings":"AAAA,OAAO,EAAE,oBAAoB,EAAE,MAAM,sBAAsB,CAAC;AAC5D,gCAAgC;AAChC,OAAO,EAAE,gBAAgB,EAAW,MAAM,uBAAuB,CAAC;AAIlE,SAAS,oBAAoB,CAC3B,MAA6B,EAC7B,KAA4B;IAE5B,OAAO,CACL,MAAM,CAAC,QAAQ,KAAK,KAAK,CAAC,QAAQ;QAClC,MAAM,CAAC,IAAI,KAAK,KAAK,CAAC,IAAI;QAC1B,MAAM,CAAC,KAAK,KAAK,KAAK,CAAC,KAAK;QAC5B,MAAM,CAAC,MAAM,KAAK,KAAK,CAAC,MAAM,CAC/B,CAAC;AACJ,CAAC;AAED,MAAM,CAAC,MAAM,cAAc,GAAG,CAAC,CAAC;AAiBhC,MAAM,OAAO,eAAe;IAClB,MAAM,CAAC;IAEP,OAAO,CAAC;IACR,cAAc,GAA8B,EAAE,CAAC;IAC/C,aAAa,GAA0C,EAAE,CAAC;IAE1D,SAAS,CAAsB;IAC/B,WAAW,GAAmB,EAAE,CAAC;IACjC,oBAAoB,GAAG,IAAI,CAAC;IAE5B,OAAO,GAAa,EAAE,CAAC;IACvB,QAAQ,GAAa,EAAE,CAAC;IACxB,SAAS,GAAc,EAAE,CAAC;IAC1B,YAAY,GAA4B,EAAE,CAAC;IAC3C,eAAe,GAA8B,EAAE,CAAC;IAEhD,cAAc,GAAG,CAAC,CAAC;IACnB,eAAe,GAAG,CAAC,CAAC;IACpB,gBAAgB,GAAG,KAAK,CAAC;IACzB,MAAM,GAAG,EAAE,CAAC;IACZ,OAAO,GAAG,EAAE,CAAC;IAEb,UAAU,CAAU;IACpB,WAAW,CAAU;IAErB,cAAc,GAAG,CAAC,cAAc,EAAE,cAAc,CAAC,CAAC;IAElD,OAAO,GAAG,EAAE,CAAC;IACb,OAAO,CAAC;IACR,SAAS,CAAC;IAEV,aAAa,GAAG;QACtB,MAAM,EAAE,CAAC;QACT,QAAQ,EAAE,CAAC;QACX,OAAO,EAAE,CAAC;KACX,CAAC;IAEF,YACE,KAAa,EACb,MAAiB,EACjB,IAMC;QAED,IAAI,CAAC,MAAM,GAAG,KAAK,CAAC;QACpB,IAAI,CAAC,OAAO,GAAG,MAAM,CAAC;QACtB,IAAI,CAAC,OAAO,GAAG,IAAI,CAAC,MAAM,CAAC;QAC3B,IAAI,CAAC,SAAS,GAAG,IAAI,CAAC,QAAQ,CAAC;QAC/B,IAAI,CAAC,OAAO,GAAG,IAAI,CAAC,MAAM,CAAC;QAC3B,IAAI,CAAC,QAAQ,GAAG,IAAI,CAAC,OAAO,CAAC;QAC7B,MAAM,QAAQ,GAAG,IAAI,CAAC,QAAQ,IAAI,EAAE,CAAC;QACrC,IAAI,CAAC,SAAS,GAAG,QAAQ,CAAC;QAE1B,IAAI,CAAC,YAAY,GAAG,IAAI,CAAC,SAAS,CAAC,MAAM,CAAC,CAAC,GAAG,EAAE,OAAO,EAAE,EAAE;YACzD,GAAG,CAAC,OAAO,CAAC,KAAK,CAAC,GAAG,OAAO,CAAC;YAC7B,OAAO,GAAG,CAAC;QACb,CAAC,EAAE,EAA6B,CAAC,CAAC;QAClC,QAAQ,CAAC,OAAO,CAAC,CAAC,OAAO,EAAE,EAAE;YAC3B,MAAM,IAAI,GAAG,gBAAgB,CAAC,OAAO,CAAC,IAAI,EAAE,OAAO,CAAC,IAAI,CAAC,CAAC;YAC1D,IAAI,CAAC,eAAe,CAAC,OAAO,CAAC,KAAK,CAAC,GAAG,MAAM,CAAC,YAAY,CAAC;gBACxD,KAAK,EAAE,IAAI,CAAC,MAAM,GAAG,GAAG,GAAG,OAAO,CAAC,KAAK;gBACxC,IAAI,EAAE,IAAI,CAAC,UAAU;gBACrB,KAAK,EAAE,cAAc,CAAC,OAAO,GAAG,cAAc,CAAC,QAAQ;aACxD,CAAC,CAAC;YACH,IAAI,CAAC,OAAO,CAAC,KAAK,CAAC,WAAW,CAC5B,IAAI,CAAC,eAAe,CAAC,OAAO,CAAC,KAAK,CAAC,EACnC,CAAC,EACD,IAAI,CACL,CAAC;QACJ,CAAC,CAAC,CAAC;IACL,CAAC;IAED,SAAS,CAAC,EAAE,QAAQ,EAAE,MAAM,EAAwC;QAClE,IAAI,CAAC,SAAS,GAAG,QAAQ,CAAC;QAC1B,IAAI,CAAC,OAAO,GAAG,MAAM,CAAC;QACtB,IAAI,CAAC,oBAAoB,GAAG,IAAI,CAAC;IACnC,CAAC;IAED,eAAe,KAAK,IAAI,CAAC,gBAAgB,GAAG,IAAI,CAAC,CAAC,IAAI,CAAC,cAAc,GAAG,IAAI,CAAC,MAAM,CAAC,CAAC,IAAI,CAAC,eAAe,GAAG,IAAI,CAAC,OAAO,CAAC,CAAC,CAAC;IAC3H,OAAO,CAAC,KAAa,EAAE,MAAc;QACnC,KAAK,GAAG,IAAI,CAAC,GAAG,CAAC,CAAC,EAAE,IAAI,CAAC,IAAI,CAAC,KAAK,CAAC,CAAC,CAAC;QACtC,MAAM,GAAG,IAAI,CAAC,GAAG,CAAC,CAAC,EAAE,IAAI,CAAC,IAAI,CAAC,MAAM,CAAC,CAAC,CAAC;QACxC,MAAM,WAAW,GAAG,KAAK,KAAK,IAAI,CAAC,MAAM,IAAI,MAAM,KAAK,IAAI,CAAC,OAAO,CAAC;QACrE,IAAI,CAAC,MAAM,GAAG,KAAK,CAAC;QACpB,IAAI,CAAC,OAAO,GAAG,MAAM,CAAC;QACtB,IAAI,WAAW,EAAE,CAAC;YAChB,IAAI,CAAC,oBAAoB,GAAG,IAAI,CAAC;YACjC,MAAM,SAAS,GAAG,IAAI,CAAC,GAAG,CAAC,IAAI,CAAC,cAAc,EAAE,KAAK,CAAC,EAAE,UAAU,GAAG,IAAI,CAAC,GAAG,CAAC,IAAI,CAAC,eAAe,EAAE,MAAM,CAAC,CAAC;YAC5G,MAAM,eAAe,GAAG,SAAS,KAAK,IAAI,CAAC,cAAc,IAAI,UAAU,KAAK,IAAI,CAAC,eAAe,CAAC;YACjG,IAAI,CAAC,cAAc,GAAG,SAAS,CAAC;YAAC,IAAI,CAAC,eAAe,GAAG,UAAU,CAAC;YACnE,IAAI,IAAI,CAAC,gBAAgB,IAAI,CAAC,eAAe;gBAAE,OAAO;YACtD,yBAAyB;YACzB,KAAK,MAAM,GAAG,IAAI,IAAI,CAAC,cAAc,EAAE,CAAC;gBACtC,IAAI,CAAC,cAAc,CAAC,GAAG,CAAC,CAAC,OAAO,EAAE,CAAC;YACrC,CAAC;YACD,IAAI,CAAC,cAAc,GAAG,EAAE,CAAC;QAC3B,CAAC;IACH,CAAC;IAED,cAAc,CAAC,KAAa,EAAE,MAAc;QAC1C,KAAK,GAAG,IAAI,CAAC,GAAG,CAAC,CAAC,EAAE,IAAI,CAAC,IAAI,CAAC,KAAK,CAAC,CAAC,CAAC;QACtC,MAAM,GAAG,IAAI,CAAC,GAAG,CAAC,CAAC,EAAE,IAAI,CAAC,IAAI,CAAC,MAAM,CAAC,CAAC,CAAC;QACxC,IAAI,CAAC,UAAU,GAAG,KAAK,CAAC;QACxB,IAAI,CAAC,WAAW,GAAG,MAAM,CAAC;IAC5B,CAAC;IAED,eAAe,CAAC,YAA8C;QAC5D,2BAA2B;QAC3B,IAAI,CAAC,+BAA+B,CAAC,YAAY,CAAC,CAAC;QACnD,IAAI,CAAC,aAAa,GAAG,YAAY,CAAC;QAClC,IAAI,CAAC,oBAAoB,GAAG,IAAI,CAAC;IACnC,CAAC;IAED,UAAU,CAAC,KAAa,EAAE,IAAqB;QAC7C,MAAM,MAAM,GAAG,IAAI,CAAC,eAAe,CAAC,KAAK,CAAC,CAAC;QAC3C,IAAI,CAAC,OAAO,CAAC,KAAK,CAAC,WAAW,CAC5B,MAAM,EACN,CAAC,EACD,gBAAgB,CAAC,IAAI,EAAE,IAAI,CAAC,YAAY,CAAC,KAAK,CAAC,CAAC,IAAI,CAAC,CACtD,CAAC;IACJ,CAAC;IAED,SAAS,CAAC,IAAO;QACf,MAAM,aAAa,GAAG,IAAI,CAAC,cAAc,CAAC;QAC1C,IAAI,CAAC,aAAa,CAAC,IAAI,CAAC,EAAE,CAAC;YACzB,aAAa,CAAC,IAAI,CAAC,GAAG,IAAI,CAAC,aAAa,CACtC,IAAI,EACJ,IAAI,CAAC,aAAa,CAAC,IAAI,CAAC,IAAI,EAAE,CAC/B,CAAC;QACJ,CAAC;QACD,OAAO,aAAa,CAAC,IAAI,CAAC,CAAC;IAC7B,CAAC;IAED,cAAc;QACZ,OAAO,CAAC,GAAG,MAAM,CAAC,MAAM,CAAC,IAAI,CAAC,eAAe,CAAC,EAAE,GAAG,MAAM,CAAC,MAAM,CAAC,IAAI,CAAC,cAAc,CAAC,CAAC;aACnF,MAAM,CAAC,CAAC,GAAG,EAAE,MAAM,EAAE,EAAE,CAAC,GAAG,GAAG,MAAM,CAAC,IAAI,EAAE,CAAC,CAAC,CAAC;IACnD,CAAC;IAED,OAAO;QACL,MAAM,CAAC,IAAI,CAAC,IAAI,CAAC,eAAe,CAAC,CAAC,OAAO,CAAC,CAAC,GAAG,EAAE,EAAE;YAC/C,IAAI,CAAC,eAAuB,CAAC,GAAG,CAAC,CAAC,OAAO,EAAE,CAAC;QAC/C,CAAC,CAAC,CAAC;QACH,MAAM,CAAC,IAAI,CAAC,IAAI,CAAC,cAAc,CAAC,CAAC,OAAO,CAAC,CAAC,GAAG,EAAE,EAAE;YAC9C,IAAI,CAAC,cAAsB,CAAC,GAAG,CAAC,CAAC,OAAO,EAAE,CAAC;QAC9C,CAAC,CAAC,CAAC;IACL,CAAC;IAED,gBAAgB,CAAC,KAAa,EAAE,MAAc;QAC5C,IAAI,CAAC,oBAAoB;YACvB,KAAK,KAAK,IAAI,CAAC,cAAc,CAAC,CAAC,CAAC,IAAI,MAAM,KAAK,IAAI,CAAC,cAAc,CAAC,CAAC,CAAC,CAAC;QACxE,IAAI,CAAC,cAAc,GAAG,CAAC,KAAK,EAAE,MAAM,CAAC,CAAC;IACxC,CAAC;IAEO,uBAAuB;QAC7B,OAAO;YACL,IAAI,CAAC,GAAG,CAAC,IAAI,CAAC,cAAc,CAAC,CAAC,CAAC,EAAE,IAAI,CAAC,UAAU,IAAI,IAAI,CAAC,MAAM,CAAC;YAChE,IAAI,CAAC,GAAG,CAAC,IAAI,CAAC,cAAc,CAAC,CAAC,CAAC,EAAE,IAAI,CAAC,WAAW,IAAI,IAAI,CAAC,OAAO,CAAC;SACnE,CAAC;IACJ,CAAC;IAEO,aAAa,CAAC,KAAa,EAAE,MAA6B;QAChE,MAAM,QAAQ,GAAG,MAAM,CAAC,QAAQ,IAAI,CAAC,CAAC;QACtC,MAAM,UAAU,GACd,CAAC,IAAI,CAAC,gBAAgB,CAAC,CAAC,CAAC,IAAI,CAAC,cAAc,GAAG,IAAI,CAAC,eAAe,CAAC,CAAC,CAAC,IAAI,CAAC,MAAM,GAAG,IAAI,CAAC,OAAO,CAAC,GAAG,QAAQ,GAAG,CAAC,MAAM,CAAC,KAAK,IAAI,CAAC,CAAC,GAAG,CAAC,CAAC;QACzI,OAAO,IAAI,CAAC,OAAO,CAAC,YAAY,CAAC;YAC/B,KAAK,EAAE,IAAI,CAAC,MAAM,GAAG,GAAG,GAAG,KAAK;YAChC,yCAAyC;YACzC,IAAI,EAAE,IAAI,CAAC,GAAG,CAAC,UAAU,EAAE,EAAE,CAAC;YAC9B,KAAK,EAAE,cAAc,CAAC,OAAO,GAAG,cAAc,CAAC,QAAQ;SACxD,CAAC,CAAC;IACL,CAAC;IAEO,+BAA+B,CAAC,SAA8B;QACpE,MAAM,aAAa,GAAG,IAAI,CAAC,cAAc,CAAC;QAC1C,MAAM,SAAS,GAAG,IAAI,CAAC,aAAa,CAAC;QACrC,KAAK,MAAM,GAAG,IAAI,aAAa,EAAE,CAAC;YAChC,IAAI,CAAC,oBAAoB,CAAC,SAAS,CAAC,GAAG,CAAC,IAAI,EAAE,EAAE,SAAS,CAAC,GAAG,CAAC,IAAI,EAAE,CAAC,EAAE,CAAC;gBACtE,aAAa,CAAC,GAAG,CAAC,CAAC,OAAO,EAAE,CAAC;gBAC7B,OAAO,aAAa,CAAC,GAAG,CAAC,CAAC;YAC5B,CAAC;QACH,CAAC;IACH,CAAC;IAEO,oBAAoB,CAAC,QAAmB;QAC9C,MAAM,aAAa,GAAG,IAAI,CAAC,cAAc,CAAC;QAC1C,MAAM,MAAM,GAAG,IAAI,CAAC,aAAa,CAAC;QAClC,KAAK,MAAM,GAAG,IAAI,MAAM,EAAE,CAAC;YACzB,IAAI,CAAC,aAAa,CAAC,GAAG,CAAC,IAAI,CAAC,QAAQ,EAAE,QAAQ,CAAC,GAAG,CAAC,EAAE,CAAC;gBACpD,aAAa,CAAC,GAAG,CAAC,GAAG,IAAI,CAAC,aAAa,CAAC,GAAG,EAAE,MAAM,CAAC,GAAG,CAAC,CAAC,CAAC;YAC5D,CAAC;QACH,CAAC;IACH,CAAC;IAEO,eAAe,CACrB,WAAiD,EACjD,UAAgD;QAEhD,IAAI,CAAC,IAAI,CAAC,oBAAoB,EAAE,CAAC;YAC/B,OAAO;QACT,CAAC;QACD,IAAI,CAAC,oBAAoB,GAAG,KAAK,CAAC;QAClC,MAAM,MAAM,GAAG,IAAI,CAAC,OAAO,CAAC;QAC5B,MAAM,MAAM,GAAG,IAAI,CAAC,UAAU,CAAC,WAAW,EAAE,UAAU,CAAC,CAAC;QACxD,IAAI,MAAM,KAAK,IAAI,CAAC,OAAO,EAAE,CAAC;YAC5B,OAAO;QACT,CAAC;QACD,IAAI,CAAC,OAAO,GAAG,MAAM,CAAC;QACtB,IAAI,CAAC,SAAS,GAAG,oBAAoB,CAAC,MAAM,CAAC,CAAC,qBAAqB,CAAC;YAClE,KAAK,EAAE,IAAI,CAAC,MAAM;YAClB,MAAM,EAAE,MAAM;YACd,OAAO,EAAE;gBACP,MAAM,EAAE,oBAAoB,CAAC,MAAM,CAAC,CAAC,kBAAkB,CAAC;oBACtD,KAAK,EAAE,IAAI,CAAC,MAAM;oBAClB,IAAI,EAAE,MAAM;iBACb,CAAC;gBACF,UAAU,EAAE,MAAM;aACnB;SACF,CAAC,CAAC;QACH,IAAI,CAAC,iBAAiB,EAAE,CAAC;IAC3B,CAAC;IAEO,UAAU,CAChB,WAAiD,EACjD,UAAgD;QAEhD,MAAM,MAAM,GAAG,IAAI,CAAC,OAAO,CAAC;QAC5B,MAAM,QAAQ,GAAG,IAAI,CAAC,SAAS,CAAC;QAChC,IAAI,MAAM,GAAG,CAAC,CAAC;QACf,MAAM,YAAY,GAAG,CAAC,IAAI,CAAC,aAAa,GAAG;YACzC,MAAM,EAAE,CAAC;YACT,QAAQ,EAAE,CAAC;YACX,OAAO,EAAE,CAAC;SACX,CAAC,CAAC;QACH,MAAM,aAAa,GAAG,IAAI,CAAC,uBAAuB,EAAE,CAAC;QAErD,IAAI,MAAM,CAAC,MAAM,GAAG,CAAC,EAAE,CAAC;YACtB,MAAM,EAAE,CAAC;QACX,CAAC;QACD,IAAI,QAAQ,CAAC,MAAM,GAAG,CAAC,EAAE,CAAC;YACxB,YAAY,CAAC,QAAQ,GAAG,MAAM,CAAC;YAC/B,MAAM,EAAE,CAAC;QACX,CAAC;QAED,SAAS,WAAW,CAAC,MAA6B;YAChD,MAAM,EAAE,IAAI,GAAG,KAAK,EAAE,QAAQ,EAAE,GAAG,MAAM,CAAC;YAC1C,IAAI,IAAI,GAAG,QAAQ,KAAK,CAAC,CAAC,CAAC,CAAC,IAAI,CAAC,CAAC,CAAC,MAAM,QAAQ,GAAG,IAAI,CAAC,CAAC,CAAC,EAAE,CAAC;YAC9D,IAAI,MAAM,CAAC,MAAM,EAAE,CAAC;gBAClB,IAAI,GAAG,UAAU,IAAI,GAAG,CAAC;YAC3B,CAAC;YACD,OAAO,IAAI,CAAC;QACd,CAAC;QACD,YAAY,CAAC,OAAO,GAAG,MAAM,CAAC;QAC9B,MAAM,EAAE,GAAG,UAAU,CAAC;EACxB,MAAM;aACL,IAAI,EAAE;aACN,GAAG,CAAC,CAAC,SAAS,EAAE,GAAG,EAAE,EAAE;YACtB,MAAM,aAAa,GAAG,UAAU,YAAY,CAAC,MAAM,cAAc,GAAG,IAAI,CAAC;YACzE,MAAM,OAAO,GAAG,MAAM,SAAS,EAAE,CAAC;YAClC,6BAA6B;YAC7B,OAAO,UAAU,CAAC,SAAS,CAAC,KAAK,SAAS;gBACxC,CAAC,CAAC,GAAG,aAAa,QAAQ,OAAO,oBAAoB;gBACrD,CAAC,CAAC,GAAG,aAAa,uBAAuB,OAAO,WAAW,WAAW,CAClE,WAAW,CAAC,SAAS,CAAC,CACvB,IAAI,CAAC;QACZ,CAAC,CAAC;aACD,IAAI,CAAC,IAAI,CAAC;EACX,IAAI,CAAC,SAAS;aACb,GAAG,CACF,CAAC,OAAO,EAAE,GAAG,EAAE,EAAE,CACf,UAAU,YAAY,CAAC,QAAQ,cAAc,GAAG,kBAAkB,OAAO,CAAC,KAAK,KAAK,OAAO,CAAC,IAAI,GAAG,CACtG;aACA,IAAI,CAAC,IAAI,CAAC;;EAEX,IAAI,CAAC,QAAQ;aACZ,GAAG,CAAC,CAAC,IAAI,EAAE,GAAG,EAAE,EAAE;YACjB,OAAO,UACL,YAAY,CAAC,OACf,cAAc,GAAG,kCAAkC,IAAI,WAAW,WAAW,CAC3E,IAAI,CAAC,aAAa,CAAC,IAAI,CAAC,IAAI,EAAE,CAC/B,IAAI,CAAC;QACR,CAAC,CAAC;aACD,IAAI,CAAC,IAAI,CAAC;EACX,IAAI,CAAC,SAAS,IAAI,EAAE;2BACK,aAAa,CAAC,CAAC,CAAC,KAAK,aAAa,CAAC,CAAC,CAAC;;EAE9D,IAAI,CAAC,OAAO;;CAEb,CAAC;QAEE,OAAO,EAAE,CAAC;IACZ,CAAC;IAEO,iBAAiB;QACvB,MAAM,UAAU,GAAmB,EAAE,CAAC;QACtC,MAAM,MAAM,GAAG,IAAI,CAAC,OAAO,CAAC;QAC5B,MAAM,YAAY,GAAG,IAAI,CAAC,aAAa,CAAC;QAExC,IAAI,IAAI,CAAC,SAAS,CAAC,MAAM,GAAG,CAAC,EAAE,CAAC;YAC9B,UAAU,CAAC,YAAY,CAAC,QAAQ,CAAC,GAAG,MAAM,CAAC,eAAe,CAAC;gBACzD,KAAK,EAAE,IAAI,CAAC,MAAM,GAAG,WAAW;gBAChC,MAAM,EAAE,IAAI,CAAC,SAAS,CAAC,kBAAkB,CAAC,YAAY,CAAC,QAAQ,CAAC;gBAChE,OAAO,EAAE,IAAI,CAAC,SAAS,CAAC,GAAG,CACzB,CAAC,OAAO,EAAE,GAAG,EAAE,EAAE,CACf,CAAC;oBACC,OAAO,EAAE,GAAG;oBACZ,QAAQ,EAAE;wBACR,MAAM,EAAE,IAAI,CAAC,eAAe,CAAC,OAAO,CAAC,KAAK,CAAC;qBAC5C;iBACoB,CAAA,CAC1B;aACF,CAAC,CAAC;QACL,CAAC;QAED,IAAI,CAAC,WAAW,GAAG,UAAU,CAAC;IAChC,CAAC;IAED,UAAU,CACR,cAAiC,EACjC,MAAuC;IAEvC;;OAEG;IACH,OAA8B;QAE9B,IAAI,CAAC,oBAAoB,CAAC,MAAM,CAAC,IAAI,CAAC,OAAO,IAAI,EAAE,CAAC,CAAC,CAAC;QACtD,MAAM,aAAa,GAAG;YACpB,GAAG,IAAI,CAAC,cAAc;YACtB,GAAG,OAAO;SACkB,CAAC;QAE/B,MAAM,UAAU,GAAG,IAAI,CAAC,OAAO,CAAC,MAAM,CAAC,CAAC,GAAG,EAAE,SAAS,EAAE,EAAE;YACxD,GAAG,CAAC,SAAS,CAAC,GAAG,MAAM,CAAC,SAAc,CAAC,CAAC,MAAM,CAAC,CAAC,CAAC,QAAQ,CAAC,CAAC,CAAC,SAAS,CAAC;YACtE,OAAO,GAAG,CAAC;QACb,CAAC,EAAE,EAA0C,CAAC,CAAC;QAE/C,IAAI,CAAC,eAAe,CAAC,MAAM,EAAE,UAAU,CAAC,CAAC;QAEzC,MAAM,YAAY,GAAG,IAAI,CAAC,aAAa,CAAC;QACxC,IAAI,IAAI,CAAC,OAAO,CAAC,MAAM,GAAG,CAAC,EAAE,CAAC;YAC5B,IAAI,CAAC,WAAW,CAAC,YAAY,CAAC,MAAM,CAAC,GAAG,IAAI,CAAC,OAAO,CAAC,eAAe,CAAC;gBACnE,KAAK,EAAE,IAAI,CAAC,MAAM,GAAG,SAAS;gBAC9B,MAAM,EAAE,IAAI,CAAC,SAAS,CAAC,kBAAkB,CAAC,YAAY,CAAC,MAAM,CAAC;gBAC9D,OAAO,EAAE,IAAI,CAAC,OAAO,CAAC,GAAG,CAAC,CAAC,SAAS,EAAE,GAAG,EAAE,EAAE,CAAC,CAAC;oBAC7C,OAAO,EAAE,GAAG;oBACZ,QAAQ,EAAE,MAAM,CAAC,SAAc,CAAC,CAAC,MAAM;wBACrC,CAAC,CAAC;4BACE,MAAM,EAAE,MAAM,CAAC,SAAc,CAAC,CAAC,MAAO;yBACvC;wBACH,CAAC,CAAC,MAAM,CAAC,SAAc,CAAC,CAAC,OAAQ,CAAC,UAAU,EAAE;iBACjD,CAAC,CAAC;aACJ,CAAC,CAAC;QACL,CAAC;QAED,UAAU;QACV,IAAI,CAAC,WAAW,CAAC,YAAY,CAAC,OAAO,CAAC,GAAG,IAAI,CAAC,OAAO,CAAC,eAAe,CAAC;YACpE,KAAK,EAAE,IAAI,CAAC,MAAM,GAAG,UAAU;YAC/B,MAAM,EAAE,IAAI,CAAC,SAAS,CAAC,kBAAkB,CAAC,YAAY,CAAC,OAAO,CAAC;YAC/D,OAAO,EAAE,IAAI,CAAC,QAAQ,CAAC,GAAG,CAAC,CAAC,UAAU,EAAE,GAAG,EAAE,EAAE,CAAC,CAAC;gBAC/C,OAAO,EAAE,GAAG;gBACZ,QAAQ,EAAE;oBACR,MAAM,EAAE,aAAa,CAAC,UAAU,CAAC;iBAClC;aACF,CAAC,CAAC;SACJ,CAAC,CAAC;QAEH,wBAAwB;QACxB,MAAM,WAAW,GAAG,cAAc,CAAC,gBAAgB,EAAE,CAAC;QAEtD,mBAAmB;QACnB,WAAW,CAAC,WAAW,CAAC,IAAI,CAAC,SAAS,CAAC,CAAC;QACxC,cAAc;QACd,IAAI,CAAC,WAAW,CAAC,OAAO,CAAC,CAAC,SAAS,EAAE,GAAG,EAAE,EAAE;YAC1C,WAAW,CAAC,YAAY,CAAC,GAAG,EAAE,SAAS,CAAC,CAAC;QAC3C,CAAC,CAAC,CAAC;QAEH,MAAM,aAAa,GAAG,IAAI,CAAC,uBAAuB,EAAE,CAAC;QACrD,WAAW,CAAC,kBAAkB,CAC5B,IAAI,CAAC,IAAI,CAAC,CAAC,IAAI,CAAC,UAAU,IAAI,IAAI,CAAC,MAAM,CAAC,GAAG,aAAa,CAAC,CAAC,CAAC,CAAC,EAC9D,IAAI,CAAC,IAAI,CAAC,CAAC,IAAI,CAAC,WAAW,IAAI,IAAI,CAAC,OAAO,CAAC,GAAG,aAAa,CAAC,CAAC,CAAC,CAAC,EAChE,CAAC,CACF,CAAC;QACF,sBAAsB;QACtB,WAAW,CAAC,GAAG,EAAE,CAAC;IACpB,CAAC;CACF","sourcesContent":["import { getTargetShaderCache } from './WGPUShaderCache.js';\n// TODO output read write buffer\nimport { getUniformBuffer, Uniform } from './WGPUFullQuadPass.js';\n\nexport type { Uniform };\n\nfunction isStorageParamsEqual(\n  params: WGPUComputePassOutput,\n  other: WGPUComputePassOutput\n) {\n  return (\n    params.channels === other.channels &&\n    params.type === other.type &&\n    params.count === other.count &&\n    params.atomic === other.atomic\n  );\n}\n\nexport const WORKGROUP_SIZE = 8;\n\nexport interface WGPUComputePassInput {\n  buffer?: GPUBuffer;\n  texture?: GPUTexture;\n  channels: number;\n  type?: 'u32' | 'i32' | 'f32';\n}\n\nexport interface WGPUComputePassOutput {\n  channels?: number;\n  // Default: 1. Count per element.\n  count?: number;\n  atomic?: boolean;\n  type?: 'u32' | 'i32' | 'f32';\n}\n\nexport class WGPUComputePass<I extends string, O extends string> {\n  private _label;\n\n  private _device;\n  private _outputBuffers: Record<string, GPUBuffer> = {};\n  private _outputParams: Record<string, WGPUComputePassOutput> = {};\n\n  private _pipeline!: GPUComputePipeline;\n  private _bindGroups: GPUBindGroup[] = [];\n  private _needsUpdatePipeline = true;\n\n  private _inputs: string[] = [];\n  private _outputs: string[] = [];\n  private _uniforms: Uniform[] = [];\n  private _uniformsMap: Record<string, Uniform> = {};\n  private _uniformBuffers: Record<string, GPUBuffer> = {};\n\n  private _capacityWidth = 0;\n  private _capacityHeight = 0;\n  private _reserveCapacity = false;\n  private _width = 10;\n  private _height = 10;\n\n  private _execWidth?: number;\n  private _execHeight?: number;\n\n  private _workGroupSize = [WORKGROUP_SIZE, WORKGROUP_SIZE];\n\n  private _csCode = '';\n  private _csMain;\n  private _csDefine;\n\n  private _groupOffsets = {\n    inputs: 0,\n    uniforms: 1,\n    outputs: 2\n  };\n\n  constructor(\n    label: string,\n    device: GPUDevice,\n    opts: {\n      inputs: I[];\n      outputs: O[];\n      csMain: string;\n      csDefine?: string;\n      uniforms?: Uniform[];\n    }\n  ) {\n    this._label = label;\n    this._device = device;\n    this._csMain = opts.csMain;\n    this._csDefine = opts.csDefine;\n    this._inputs = opts.inputs;\n    this._outputs = opts.outputs;\n    const uniforms = opts.uniforms || [];\n    this._uniforms = uniforms;\n\n    this._uniformsMap = this._uniforms.reduce((map, uniform) => {\n      map[uniform.label] = uniform;\n      return map;\n    }, {} as Record<string, Uniform>);\n    uniforms.forEach((uniform) => {\n      const data = getUniformBuffer(uniform.data, uniform.type);\n      this._uniformBuffers[uniform.label] = device.createBuffer({\n        label: this._label + '.' + uniform.label,\n        size: data.byteLength,\n        usage: GPUBufferUsage.UNIFORM | GPUBufferUsage.COPY_DST\n      });\n      this._device.queue.writeBuffer(\n        this._uniformBuffers[uniform.label],\n        0,\n        data\n      );\n    });\n  }\n\n  setCSCode({ csDefine, csMain }: { csDefine: string; csMain: string }) {\n    this._csDefine = csDefine;\n    this._csMain = csMain;\n    this._needsUpdatePipeline = true;\n  }\n\n  reserveCapacity() { this._reserveCapacity = true; this._capacityWidth = this._width; this._capacityHeight = this._height; }\n  setSize(width: number, height: number) {\n    width = Math.max(1, Math.ceil(width));\n    height = Math.max(1, Math.ceil(height));\n    const sizeChanged = width !== this._width || height !== this._height;\n    this._width = width;\n    this._height = height;\n    if (sizeChanged) {\n      this._needsUpdatePipeline = true;\n      const nextWidth = Math.max(this._capacityWidth, width), nextHeight = Math.max(this._capacityHeight, height);\n      const capacityChanged = nextWidth !== this._capacityWidth || nextHeight !== this._capacityHeight;\n      this._capacityWidth = nextWidth; this._capacityHeight = nextHeight;\n      if (this._reserveCapacity && !capacityChanged) return;\n      // Invalidate all buffers\n      for (const key in this._outputBuffers) {\n        this._outputBuffers[key].destroy();\n      }\n      this._outputBuffers = {};\n    }\n  }\n\n  setExecuteSize(width: number, height: number) {\n    width = Math.max(1, Math.ceil(width));\n    height = Math.max(1, Math.ceil(height));\n    this._execWidth = width;\n    this._execHeight = height;\n  }\n\n  setOutputParams(outputParams: Record<O, WGPUComputePassOutput>) {\n    // TODO lazy update buffers\n    this._invalidateChangedOutputBuffers(outputParams);\n    this._outputParams = outputParams;\n    this._needsUpdatePipeline = true;\n  }\n\n  setUniform(label: string, data: Uniform['data']) {\n    const buffer = this._uniformBuffers[label];\n    this._device.queue.writeBuffer(\n      buffer,\n      0,\n      getUniformBuffer(data, this._uniformsMap[label].type)\n    );\n  }\n\n  getOutput(name: O) {\n    const outputBuffers = this._outputBuffers;\n    if (!outputBuffers[name]) {\n      outputBuffers[name] = this._createBuffer(\n        name,\n        this._outputParams[name] || {}\n      );\n    }\n    return outputBuffers[name];\n  }\n\n  getMemoryBytes() {\n    return [...Object.values(this._uniformBuffers), ...Object.values(this._outputBuffers)]\n      .reduce((sum, buffer) => sum + buffer.size, 0);\n  }\n\n  dispose() {\n    Object.keys(this._uniformBuffers).forEach((key) => {\n      (this._uniformBuffers as any)[key].destroy();\n    });\n    Object.keys(this._outputBuffers).forEach((key) => {\n      (this._outputBuffers as any)[key].destroy();\n    });\n  }\n\n  setWorkGroupSize(width: number, height: number) {\n    this._needsUpdatePipeline =\n      width !== this._workGroupSize[0] || height !== this._workGroupSize[1];\n    this._workGroupSize = [width, height];\n  }\n\n  private _getActualWorkgroupSize() {\n    return [\n      Math.min(this._workGroupSize[0], this._execWidth ?? this._width),\n      Math.min(this._workGroupSize[1], this._execHeight ?? this._height)\n    ];\n  }\n\n  private _createBuffer(label: string, params: WGPUComputePassOutput) {\n    const channels = params.channels ?? 4;\n    const byteLength =\n      (this._reserveCapacity ? this._capacityWidth * this._capacityHeight : this._width * this._height) * channels * (params.count ?? 1) * 4;\n    return this._device.createBuffer({\n      label: this._label + '.' + label,\n      // webgpu needs buffer at least 80 bytes.\n      size: Math.max(byteLength, 80),\n      usage: GPUBufferUsage.STORAGE | GPUBufferUsage.COPY_SRC\n    });\n  }\n\n  private _invalidateChangedOutputBuffers(newParams: Record<string, any>) {\n    const outputBuffers = this._outputBuffers;\n    const oldParams = this._outputParams;\n    for (const key in outputBuffers) {\n      if (!isStorageParamsEqual(newParams[key] || {}, oldParams[key] || {})) {\n        outputBuffers[key].destroy();\n        delete outputBuffers[key];\n      }\n    }\n  }\n\n  private _updateOutputBuffers(excludes?: string[]) {\n    const outputBuffers = this._outputBuffers;\n    const params = this._outputParams;\n    for (const key in params) {\n      if (!outputBuffers[key] && !excludes?.includes(key)) {\n        outputBuffers[key] = this._createBuffer(key, params[key]);\n      }\n    }\n  }\n\n  private _updatePipeline(\n    inputParams: Record<string, WGPUComputePassInput>,\n    inputTypes: Record<string, 'texture' | 'buffer'>\n  ) {\n    if (!this._needsUpdatePipeline) {\n      return;\n    }\n    this._needsUpdatePipeline = false;\n    const device = this._device;\n    const csCode = this._getFullCs(inputParams, inputTypes);\n    if (csCode === this._csCode) {\n      return;\n    }\n    this._csCode = csCode;\n    this._pipeline = getTargetShaderCache(device).createComputePipeline({\n      label: this._label,\n      layout: 'auto',\n      compute: {\n        module: getTargetShaderCache(device).createShaderModule({\n          label: this._label,\n          code: csCode\n        }),\n        entryPoint: 'main'\n      }\n    });\n    this._updateBindGroups();\n  }\n\n  private _getFullCs(\n    inputParams: Record<string, WGPUComputePassInput>,\n    inputTypes: Record<string, 'texture' | 'buffer'>\n  ) {\n    const inputs = this._inputs;\n    const uniforms = this._uniforms;\n    let offset = 0;\n    const groupOffsets = (this._groupOffsets = {\n      inputs: 0,\n      uniforms: 0,\n      outputs: 0\n    });\n    const workgroupSize = this._getActualWorkgroupSize();\n\n    if (inputs.length > 0) {\n      offset++;\n    }\n    if (uniforms.length > 0) {\n      groupOffsets.uniforms = offset;\n      offset++;\n    }\n\n    function getTypeCode(params: WGPUComputePassOutput) {\n      const { type = 'f32', channels } = params;\n      let code = channels === 1 ? type : `vec${channels}${type[0]}`;\n      if (params.atomic) {\n        code = `atomic<${code}>`;\n      }\n      return code;\n    }\n    groupOffsets.outputs = offset;\n    const cs = /* wgsl */ `\n${inputs\n  .sort()\n  .map((inputName, idx) => {\n    const bindingPrefix = `@group(${groupOffsets.inputs}) @binding(${idx}) `;\n    const varName = `in_${inputName}`;\n    // TODO more channels option.\n    return inputTypes[inputName] === 'texture'\n      ? `${bindingPrefix} var ${varName}: texture_2d<f32>;`\n      : `${bindingPrefix} var<storage, read> ${varName}: array<${getTypeCode(\n          inputParams[inputName]\n        )}>;`;\n  })\n  .join('\\n')}\n${this._uniforms\n  .map(\n    (uniform, idx) =>\n      `@group(${groupOffsets.uniforms}) @binding(${idx}) var<uniform> ${uniform.label}: ${uniform.type};`\n  )\n  .join('\\n')}\n\n${this._outputs\n  .map((name, idx) => {\n    return `@group(${\n      groupOffsets.outputs\n    }) @binding(${idx}) var<storage, read_write> out_${name}: array<${getTypeCode(\n      this._outputParams[name] || {}\n    )}>;`;\n  })\n  .join('\\n')}\n${this._csDefine ?? ''}\n@compute @workgroup_size(${workgroupSize[0]}, ${workgroupSize[1]}, 1)\nfn main(@builtin(global_invocation_id) globalId: vec3u) {\n${this._csMain}\n}\n`;\n\n    return cs;\n  }\n\n  private _updateBindGroups() {\n    const bindGroups: GPUBindGroup[] = [];\n    const device = this._device;\n    const groupOffsets = this._groupOffsets;\n\n    if (this._uniforms.length > 0) {\n      bindGroups[groupOffsets.uniforms] = device.createBindGroup({\n        label: this._label + '.uniforms',\n        layout: this._pipeline.getBindGroupLayout(groupOffsets.uniforms),\n        entries: this._uniforms.map(\n          (uniform, idx) =>\n            ({\n              binding: idx,\n              resource: {\n                buffer: this._uniformBuffers[uniform.label]\n              }\n            } as GPUBindGroupEntry)\n        )\n      });\n    }\n\n    this._bindGroups = bindGroups;\n  }\n\n  createPass(\n    commandEncoder: GPUCommandEncoder,\n    inputs: Record<I, WGPUComputePassInput>,\n\n    /**\n     * Explict outputs. Used for buffers reuse.\n     */\n    outputs?: Record<O, GPUBuffer>\n  ) {\n    this._updateOutputBuffers(Object.keys(outputs || {}));\n    const outputBuffers = {\n      ...this._outputBuffers,\n      ...outputs\n    } as Record<string, GPUBuffer>;\n\n    const inputTypes = this._inputs.reduce((obj, inputName) => {\n      obj[inputName] = inputs[inputName as I].buffer ? 'buffer' : 'texture';\n      return obj;\n    }, {} as Record<string, 'texture' | 'buffer'>);\n\n    this._updatePipeline(inputs, inputTypes);\n\n    const groupOffsets = this._groupOffsets;\n    if (this._inputs.length > 0) {\n      this._bindGroups[groupOffsets.inputs] = this._device.createBindGroup({\n        label: this._label + '.inputs',\n        layout: this._pipeline.getBindGroupLayout(groupOffsets.inputs),\n        entries: this._inputs.map((inputName, idx) => ({\n          binding: idx,\n          resource: inputs[inputName as I].buffer\n            ? {\n                buffer: inputs[inputName as I].buffer!\n              }\n            : inputs[inputName as I].texture!.createView()\n        }))\n      });\n    }\n\n    // Outputs\n    this._bindGroups[groupOffsets.outputs] = this._device.createBindGroup({\n      label: this._label + '.outputs',\n      layout: this._pipeline.getBindGroupLayout(groupOffsets.outputs),\n      entries: this._outputs.map((bufferName, idx) => ({\n        binding: idx,\n        resource: {\n          buffer: outputBuffers[bufferName]\n        }\n      }))\n    });\n\n    // Begin the render pass\n    const computePass = commandEncoder.beginComputePass();\n\n    // Draw a full quad\n    computePass.setPipeline(this._pipeline);\n    // Bind groups\n    this._bindGroups.forEach((bindGroup, idx) => {\n      computePass.setBindGroup(idx, bindGroup);\n    });\n\n    const workgroupSize = this._getActualWorkgroupSize();\n    computePass.dispatchWorkgroups(\n      Math.ceil((this._execWidth ?? this._width) / workgroupSize[0]),\n      Math.ceil((this._execHeight ?? this._height) / workgroupSize[1]),\n      1\n    );\n    // End the render pass\n    computePass.end();\n  }\n}\n"]}
+{"version":3,"file":"WGPUComputePass.js","sourceRoot":"","sources":["../../src/renderer/WGPUComputePass.ts"],"names":[],"mappings":"AAAA,OAAO,EAAE,oBAAoB,EAAE,MAAM,sBAAsB,CAAC;AAC5D,gCAAgC;AAChC,OAAO,EAAE,gBAAgB,EAAgB,MAAM,uBAAuB,CAAC;AAIvE,SAAS,oBAAoB,CAC3B,MAA6B,EAC7B,KAA4B;IAE5B,OAAO,CACL,MAAM,CAAC,QAAQ,KAAK,KAAK,CAAC,QAAQ;QAClC,MAAM,CAAC,IAAI,KAAK,KAAK,CAAC,IAAI;QAC1B,MAAM,CAAC,KAAK,KAAK,KAAK,CAAC,KAAK;QAC5B,MAAM,CAAC,MAAM,KAAK,KAAK,CAAC,MAAM,CAC/B,CAAC;AACJ,CAAC;AAED,MAAM,CAAC,MAAM,cAAc,GAAG,CAAC,CAAC;AAiBhC,MAAM,OAAO,eAAe;IAClB,MAAM,CAAC;IAEP,OAAO,CAAC;IACR,cAAc,GAA8B,EAAE,CAAC;IAC/C,aAAa,GAA0C,EAAE,CAAC;IAE1D,SAAS,CAAsB;IAC/B,WAAW,GAAmB,EAAE,CAAC;IACjC,oBAAoB,GAAG,IAAI,CAAC;IAE5B,OAAO,GAAa,EAAE,CAAC;IACvB,QAAQ,GAAa,EAAE,CAAC;IACxB,SAAS,GAAc,EAAE,CAAC;IAC1B,YAAY,GAA4B,EAAE,CAAC;IAC3C,eAAe,GAA8B,EAAE,CAAC;IAEhD,cAAc,GAAG,CAAC,CAAC;IACnB,eAAe,GAAG,CAAC,CAAC;IACpB,gBAAgB,GAAG,KAAK,CAAC;IACzB,MAAM,GAAG,EAAE,CAAC;IACZ,OAAO,GAAG,EAAE,CAAC;IAEb,UAAU,CAAU;IACpB,WAAW,CAAU;IAErB,cAAc,GAAG,CAAC,cAAc,EAAE,cAAc,CAAC,CAAC;IAElD,OAAO,GAAG,EAAE,CAAC;IACb,OAAO,CAAC;IACR,SAAS,CAAC;IAEV,aAAa,GAAG;QACtB,MAAM,EAAE,CAAC;QACT,QAAQ,EAAE,CAAC;QACX,OAAO,EAAE,CAAC;KACX,CAAC;IAEF,YACE,KAAa,EACb,MAAiB,EACjB,IAMC;QAED,IAAI,CAAC,MAAM,GAAG,KAAK,CAAC;QACpB,IAAI,CAAC,OAAO,GAAG,MAAM,CAAC;QACtB,IAAI,CAAC,OAAO,GAAG,IAAI,CAAC,MAAM,CAAC;QAC3B,IAAI,CAAC,SAAS,GAAG,IAAI,CAAC,QAAQ,CAAC;QAC/B,IAAI,CAAC,OAAO,GAAG,IAAI,CAAC,MAAM,CAAC;QAC3B,IAAI,CAAC,QAAQ,GAAG,IAAI,CAAC,OAAO,CAAC;QAC7B,MAAM,QAAQ,GAAG,IAAI,CAAC,QAAQ,IAAI,EAAE,CAAC;QACrC,IAAI,CAAC,SAAS,GAAG,QAAQ,CAAC;QAE1B,IAAI,CAAC,YAAY,GAAG,IAAI,CAAC,SAAS,CAAC,MAAM,CAAC,CAAC,GAAG,EAAE,OAAO,EAAE,EAAE;YACzD,GAAG,CAAC,OAAO,CAAC,KAAK,CAAC,GAAG,OAAO,CAAC;YAC7B,OAAO,GAAG,CAAC;QACb,CAAC,EAAE,EAA6B,CAAC,CAAC;QAClC,QAAQ,CAAC,OAAO,CAAC,CAAC,OAAO,EAAE,EAAE;YAC3B,MAAM,IAAI,GAAG,gBAAgB,CAAC,OAAO,CAAC,IAAI,EAAE,OAAO,CAAC,IAAI,CAAC,CAAC;YAC1D,IAAI,CAAC,eAAe,CAAC,OAAO,CAAC,KAAK,CAAC,GAAG,MAAM,CAAC,YAAY,CAAC;gBACxD,KAAK,EAAE,IAAI,CAAC,MAAM,GAAG,GAAG,GAAG,OAAO,CAAC,KAAK;gBACxC,IAAI,EAAE,IAAI,CAAC,UAAU;gBACrB,KAAK,EAAE,cAAc,CAAC,OAAO,GAAG,cAAc,CAAC,QAAQ;aACxD,CAAC,CAAC;YACH,IAAI,CAAC,OAAO,CAAC,KAAK,CAAC,WAAW,CAC5B,IAAI,CAAC,eAAe,CAAC,OAAO,CAAC,KAAK,CAAC,EACnC,CAAC,EACD,IAAI,CACL,CAAC;QACJ,CAAC,CAAC,CAAC;IACL,CAAC;IAED,SAAS,CAAC,EAAE,QAAQ,EAAE,MAAM,EAAwC;QAClE,IAAI,CAAC,SAAS,GAAG,QAAQ,CAAC;QAC1B,IAAI,CAAC,OAAO,GAAG,MAAM,CAAC;QACtB,IAAI,CAAC,oBAAoB,GAAG,IAAI,CAAC;IACnC,CAAC;IAED,eAAe,KAAK,IAAI,CAAC,gBAAgB,GAAG,IAAI,CAAC,CAAC,IAAI,CAAC,cAAc,GAAG,IAAI,CAAC,MAAM,CAAC,CAAC,IAAI,CAAC,eAAe,GAAG,IAAI,CAAC,OAAO,CAAC,CAAC,CAAC;IAC3H,OAAO,CAAC,KAAa,EAAE,MAAc;QACnC,KAAK,GAAG,IAAI,CAAC,GAAG,CAAC,CAAC,EAAE,IAAI,CAAC,IAAI,CAAC,KAAK,CAAC,CAAC,CAAC;QACtC,MAAM,GAAG,IAAI,CAAC,GAAG,CAAC,CAAC,EAAE,IAAI,CAAC,IAAI,CAAC,MAAM,CAAC,CAAC,CAAC;QACxC,MAAM,WAAW,GAAG,KAAK,KAAK,IAAI,CAAC,MAAM,IAAI,MAAM,KAAK,IAAI,CAAC,OAAO,CAAC;QACrE,IAAI,CAAC,MAAM,GAAG,KAAK,CAAC;QACpB,IAAI,CAAC,OAAO,GAAG,MAAM,CAAC;QACtB,IAAI,WAAW,EAAE,CAAC;YAChB,IAAI,CAAC,oBAAoB,GAAG,IAAI,CAAC;YACjC,MAAM,SAAS,GAAG,IAAI,CAAC,GAAG,CAAC,IAAI,CAAC,cAAc,EAAE,KAAK,CAAC,EAAE,UAAU,GAAG,IAAI,CAAC,GAAG,CAAC,IAAI,CAAC,eAAe,EAAE,MAAM,CAAC,CAAC;YAC5G,MAAM,eAAe,GAAG,SAAS,KAAK,IAAI,CAAC,cAAc,IAAI,UAAU,KAAK,IAAI,CAAC,eAAe,CAAC;YACjG,IAAI,CAAC,cAAc,GAAG,SAAS,CAAC;YAAC,IAAI,CAAC,eAAe,GAAG,UAAU,CAAC;YACnE,IAAI,IAAI,CAAC,gBAAgB,IAAI,CAAC,eAAe;gBAAE,OAAO;YACtD,yBAAyB;YACzB,KAAK,MAAM,GAAG,IAAI,IAAI,CAAC,cAAc,EAAE,CAAC;gBACtC,IAAI,CAAC,cAAc,CAAC,GAAG,CAAC,CAAC,OAAO,EAAE,CAAC;YACrC,CAAC;YACD,IAAI,CAAC,cAAc,GAAG,EAAE,CAAC;QAC3B,CAAC;IACH,CAAC;IAED,cAAc,CAAC,KAAa,EAAE,MAAc;QAC1C,KAAK,GAAG,IAAI,CAAC,GAAG,CAAC,CAAC,EAAE,IAAI,CAAC,IAAI,CAAC,KAAK,CAAC,CAAC,CAAC;QACtC,MAAM,GAAG,IAAI,CAAC,GAAG,CAAC,CAAC,EAAE,IAAI,CAAC,IAAI,CAAC,MAAM,CAAC,CAAC,CAAC;QACxC,IAAI,CAAC,UAAU,GAAG,KAAK,CAAC;QACxB,IAAI,CAAC,WAAW,GAAG,MAAM,CAAC;IAC5B,CAAC;IAED,eAAe,CAAC,YAA8C;QAC5D,2BAA2B;QAC3B,IAAI,CAAC,+BAA+B,CAAC,YAAY,CAAC,CAAC;QACnD,IAAI,CAAC,aAAa,GAAG,YAAY,CAAC;QAClC,IAAI,CAAC,oBAAoB,GAAG,IAAI,CAAC;IACnC,CAAC;IAED,UAAU,CAAC,KAAa,EAAE,IAAqB;QAC7C,MAAM,MAAM,GAAG,IAAI,CAAC,eAAe,CAAC,KAAK,CAAC,CAAC;QAC3C,IAAI,CAAC,OAAO,CAAC,KAAK,CAAC,WAAW,CAC5B,MAAM,EACN,CAAC,EACD,gBAAgB,CAAC,IAAI,EAAE,IAAI,CAAC,YAAY,CAAC,KAAK,CAAC,CAAC,IAAI,CAAC,CACtD,CAAC;IACJ,CAAC;IAED,SAAS,CAAC,IAAO;QACf,MAAM,aAAa,GAAG,IAAI,CAAC,cAAc,CAAC;QAC1C,IAAI,CAAC,aAAa,CAAC,IAAI,CAAC,EAAE,CAAC;YACzB,aAAa,CAAC,IAAI,CAAC,GAAG,IAAI,CAAC,aAAa,CACtC,IAAI,EACJ,IAAI,CAAC,aAAa,CAAC,IAAI,CAAC,IAAI,EAAE,CAC/B,CAAC;QACJ,CAAC;QACD,OAAO,aAAa,CAAC,IAAI,CAAC,CAAC;IAC7B,CAAC;IAED,cAAc;QACZ,OAAO,CAAC,GAAG,MAAM,CAAC,MAAM,CAAC,IAAI,CAAC,eAAe,CAAC,EAAE,GAAG,MAAM,CAAC,MAAM,CAAC,IAAI,CAAC,cAAc,CAAC,CAAC;aACnF,MAAM,CAAC,CAAC,GAAG,EAAE,MAAM,EAAE,EAAE,CAAC,GAAG,GAAG,MAAM,CAAC,IAAI,EAAE,CAAC,CAAC,CAAC;IACnD,CAAC;IAED,OAAO;QACL,MAAM,CAAC,IAAI,CAAC,IAAI,CAAC,eAAe,CAAC,CAAC,OAAO,CAAC,CAAC,GAAG,EAAE,EAAE;YAC/C,IAAI,CAAC,eAAuB,CAAC,GAAG,CAAC,CAAC,OAAO,EAAE,CAAC;QAC/C,CAAC,CAAC,CAAC;QACH,MAAM,CAAC,IAAI,CAAC,IAAI,CAAC,cAAc,CAAC,CAAC,OAAO,CAAC,CAAC,GAAG,EAAE,EAAE;YAC9C,IAAI,CAAC,cAAsB,CAAC,GAAG,CAAC,CAAC,OAAO,EAAE,CAAC;QAC9C,CAAC,CAAC,CAAC;IACL,CAAC;IAED,gBAAgB,CAAC,KAAa,EAAE,MAAc;QAC5C,IAAI,CAAC,oBAAoB;YACvB,KAAK,KAAK,IAAI,CAAC,cAAc,CAAC,CAAC,CAAC,IAAI,MAAM,KAAK,IAAI,CAAC,cAAc,CAAC,CAAC,CAAC,CAAC;QACxE,IAAI,CAAC,cAAc,GAAG,CAAC,KAAK,EAAE,MAAM,CAAC,CAAC;IACxC,CAAC;IAEO,uBAAuB;QAC7B,OAAO;YACL,IAAI,CAAC,GAAG,CAAC,IAAI,CAAC,cAAc,CAAC,CAAC,CAAC,EAAE,IAAI,CAAC,UAAU,IAAI,IAAI,CAAC,MAAM,CAAC;YAChE,IAAI,CAAC,GAAG,CAAC,IAAI,CAAC,cAAc,CAAC,CAAC,CAAC,EAAE,IAAI,CAAC,WAAW,IAAI,IAAI,CAAC,OAAO,CAAC;SACnE,CAAC;IACJ,CAAC;IAEO,aAAa,CAAC,KAAa,EAAE,MAA6B;QAChE,MAAM,QAAQ,GAAG,MAAM,CAAC,QAAQ,IAAI,CAAC,CAAC;QACtC,MAAM,UAAU,GACd,CAAC,IAAI,CAAC,gBAAgB,CAAC,CAAC,CAAC,IAAI,CAAC,cAAc,GAAG,IAAI,CAAC,eAAe,CAAC,CAAC,CAAC,IAAI,CAAC,MAAM,GAAG,IAAI,CAAC,OAAO,CAAC,GAAG,QAAQ,GAAG,CAAC,MAAM,CAAC,KAAK,IAAI,CAAC,CAAC,GAAG,CAAC,CAAC;QACzI,OAAO,IAAI,CAAC,OAAO,CAAC,YAAY,CAAC;YAC/B,KAAK,EAAE,IAAI,CAAC,MAAM,GAAG,GAAG,GAAG,KAAK;YAChC,yCAAyC;YACzC,IAAI,EAAE,IAAI,CAAC,GAAG,CAAC,UAAU,EAAE,EAAE,CAAC;YAC9B,KAAK,EAAE,cAAc,CAAC,OAAO,GAAG,cAAc,CAAC,QAAQ;SACxD,CAAC,CAAC;IACL,CAAC;IAEO,+BAA+B,CAAC,SAA8B;QACpE,MAAM,aAAa,GAAG,IAAI,CAAC,cAAc,CAAC;QAC1C,MAAM,SAAS,GAAG,IAAI,CAAC,aAAa,CAAC;QACrC,KAAK,MAAM,GAAG,IAAI,aAAa,EAAE,CAAC;YAChC,IAAI,CAAC,oBAAoB,CAAC,SAAS,CAAC,GAAG,CAAC,IAAI,EAAE,EAAE,SAAS,CAAC,GAAG,CAAC,IAAI,EAAE,CAAC,EAAE,CAAC;gBACtE,aAAa,CAAC,GAAG,CAAC,CAAC,OAAO,EAAE,CAAC;gBAC7B,OAAO,aAAa,CAAC,GAAG,CAAC,CAAC;YAC5B,CAAC;QACH,CAAC;IACH,CAAC;IAEO,oBAAoB,CAAC,QAAmB;QAC9C,MAAM,aAAa,GAAG,IAAI,CAAC,cAAc,CAAC;QAC1C,MAAM,MAAM,GAAG,IAAI,CAAC,aAAa,CAAC;QAClC,KAAK,MAAM,GAAG,IAAI,MAAM,EAAE,CAAC;YACzB,IAAI,CAAC,aAAa,CAAC,GAAG,CAAC,IAAI,CAAC,QAAQ,EAAE,QAAQ,CAAC,GAAG,CAAC,EAAE,CAAC;gBACpD,aAAa,CAAC,GAAG,CAAC,GAAG,IAAI,CAAC,aAAa,CAAC,GAAG,EAAE,MAAM,CAAC,GAAG,CAAC,CAAC,CAAC;YAC5D,CAAC;QACH,CAAC;IACH,CAAC;IAEO,eAAe,CACrB,WAAiD,EACjD,UAAgD;QAEhD,IAAI,CAAC,IAAI,CAAC,oBAAoB,EAAE,CAAC;YAC/B,OAAO;QACT,CAAC;QACD,IAAI,CAAC,oBAAoB,GAAG,KAAK,CAAC;QAClC,MAAM,MAAM,GAAG,IAAI,CAAC,OAAO,CAAC;QAC5B,MAAM,MAAM,GAAG,IAAI,CAAC,UAAU,CAAC,WAAW,EAAE,UAAU,CAAC,CAAC;QACxD,IAAI,MAAM,KAAK,IAAI,CAAC,OAAO,EAAE,CAAC;YAC5B,OAAO;QACT,CAAC;QACD,IAAI,CAAC,OAAO,GAAG,MAAM,CAAC;QACtB,IAAI,CAAC,SAAS,GAAG,oBAAoB,CAAC,MAAM,CAAC,CAAC,qBAAqB,CAAC;YAClE,KAAK,EAAE,IAAI,CAAC,MAAM;YAClB,MAAM,EAAE,MAAM;YACd,OAAO,EAAE;gBACP,MAAM,EAAE,oBAAoB,CAAC,MAAM,CAAC,CAAC,kBAAkB,CAAC;oBACtD,KAAK,EAAE,IAAI,CAAC,MAAM;oBAClB,IAAI,EAAE,MAAM;iBACb,CAAC;gBACF,UAAU,EAAE,MAAM;aACnB;SACF,CAAC,CAAC;QACH,IAAI,CAAC,iBAAiB,EAAE,CAAC;IAC3B,CAAC;IAEO,UAAU,CAChB,WAAiD,EACjD,UAAgD;QAEhD,MAAM,MAAM,GAAG,IAAI,CAAC,OAAO,CAAC;QAC5B,MAAM,QAAQ,GAAG,IAAI,CAAC,SAAS,CAAC;QAChC,IAAI,MAAM,GAAG,CAAC,CAAC;QACf,MAAM,YAAY,GAAG,CAAC,IAAI,CAAC,aAAa,GAAG;YACzC,MAAM,EAAE,CAAC;YACT,QAAQ,EAAE,CAAC;YACX,OAAO,EAAE,CAAC;SACX,CAAC,CAAC;QACH,MAAM,aAAa,GAAG,IAAI,CAAC,uBAAuB,EAAE,CAAC;QAErD,IAAI,MAAM,CAAC,MAAM,GAAG,CAAC,EAAE,CAAC;YACtB,MAAM,EAAE,CAAC;QACX,CAAC;QACD,IAAI,QAAQ,CAAC,MAAM,GAAG,CAAC,EAAE,CAAC;YACxB,YAAY,CAAC,QAAQ,GAAG,MAAM,CAAC;YAC/B,MAAM,EAAE,CAAC;QACX,CAAC;QAED,SAAS,WAAW,CAAC,MAA6B;YAChD,MAAM,EAAE,IAAI,GAAG,KAAK,EAAE,QAAQ,EAAE,GAAG,MAAM,CAAC;YAC1C,IAAI,IAAI,GAAG,QAAQ,KAAK,CAAC,CAAC,CAAC,CAAC,IAAI,CAAC,CAAC,CAAC,MAAM,QAAQ,GAAG,IAAI,CAAC,CAAC,CAAC,EAAE,CAAC;YAC9D,IAAI,MAAM,CAAC,MAAM,EAAE,CAAC;gBAClB,IAAI,GAAG,UAAU,IAAI,GAAG,CAAC;YAC3B,CAAC;YACD,OAAO,IAAI,CAAC;QACd,CAAC;QACD,YAAY,CAAC,OAAO,GAAG,MAAM,CAAC;QAC9B,MAAM,EAAE,GAAG,UAAU,CAAC;EACxB,MAAM;aACL,IAAI,EAAE;aACN,GAAG,CAAC,CAAC,SAAS,EAAE,GAAG,EAAE,EAAE;YACtB,MAAM,aAAa,GAAG,UAAU,YAAY,CAAC,MAAM,cAAc,GAAG,IAAI,CAAC;YACzE,MAAM,OAAO,GAAG,MAAM,SAAS,EAAE,CAAC;YAClC,6BAA6B;YAC7B,OAAO,UAAU,CAAC,SAAS,CAAC,KAAK,SAAS;gBACxC,CAAC,CAAC,GAAG,aAAa,QAAQ,OAAO,oBAAoB;gBACrD,CAAC,CAAC,GAAG,aAAa,uBAAuB,OAAO,WAAW,WAAW,CAClE,WAAW,CAAC,SAAS,CAAC,CACvB,IAAI,CAAC;QACZ,CAAC,CAAC;aACD,IAAI,CAAC,IAAI,CAAC;EACX,IAAI,CAAC,SAAS;aACb,GAAG,CACF,CAAC,OAAO,EAAE,GAAG,EAAE,EAAE,CACf,UAAU,YAAY,CAAC,QAAQ,cAAc,GAAG,kBAAkB,OAAO,CAAC,KAAK,KAAK,OAAO,CAAC,IAAI,GAAG,CACtG;aACA,IAAI,CAAC,IAAI,CAAC;;EAEX,IAAI,CAAC,QAAQ;aACZ,GAAG,CAAC,CAAC,IAAI,EAAE,GAAG,EAAE,EAAE;YACjB,OAAO,UACL,YAAY,CAAC,OACf,cAAc,GAAG,kCAAkC,IAAI,WAAW,WAAW,CAC3E,IAAI,CAAC,aAAa,CAAC,IAAI,CAAC,IAAI,EAAE,CAC/B,IAAI,CAAC;QACR,CAAC,CAAC;aACD,IAAI,CAAC,IAAI,CAAC;EACX,IAAI,CAAC,SAAS,IAAI,EAAE;2BACK,aAAa,CAAC,CAAC,CAAC,KAAK,aAAa,CAAC,CAAC,CAAC;;EAE9D,IAAI,CAAC,OAAO;;CAEb,CAAC;QAEE,OAAO,EAAE,CAAC;IACZ,CAAC;IAEO,iBAAiB;QACvB,MAAM,UAAU,GAAmB,EAAE,CAAC;QACtC,MAAM,MAAM,GAAG,IAAI,CAAC,OAAO,CAAC;QAC5B,MAAM,YAAY,GAAG,IAAI,CAAC,aAAa,CAAC;QAExC,IAAI,IAAI,CAAC,SAAS,CAAC,MAAM,GAAG,CAAC,EAAE,CAAC;YAC9B,UAAU,CAAC,YAAY,CAAC,QAAQ,CAAC,GAAG,MAAM,CAAC,eAAe,CAAC;gBACzD,KAAK,EAAE,IAAI,CAAC,MAAM,GAAG,WAAW;gBAChC,MAAM,EAAE,IAAI,CAAC,SAAS,CAAC,kBAAkB,CAAC,YAAY,CAAC,QAAQ,CAAC;gBAChE,OAAO,EAAE,IAAI,CAAC,SAAS,CAAC,GAAG,CACzB,CAAC,OAAO,EAAE,GAAG,EAAE,EAAE,CACf,CAAC;oBACC,OAAO,EAAE,GAAG;oBACZ,QAAQ,EAAE;wBACR,MAAM,EAAE,IAAI,CAAC,eAAe,CAAC,OAAO,CAAC,KAAK,CAAC;qBAC5C;iBACoB,CAAA,CAC1B;aACF,CAAC,CAAC;QACL,CAAC;QAED,IAAI,CAAC,WAAW,GAAG,UAAU,CAAC;IAChC,CAAC;IAED,UAAU,CACR,cAAiC,EACjC,MAAuC;IAEvC;;OAEG;IACH,OAA8B;QAE9B,IAAI,CAAC,oBAAoB,CAAC,MAAM,CAAC,IAAI,CAAC,OAAO,IAAI,EAAE,CAAC,CAAC,CAAC;QACtD,MAAM,aAAa,GAAG;YACpB,GAAG,IAAI,CAAC,cAAc;YACtB,GAAG,OAAO;SACkB,CAAC;QAE/B,MAAM,UAAU,GAAG,IAAI,CAAC,OAAO,CAAC,MAAM,CAAC,CAAC,GAAG,EAAE,SAAS,EAAE,EAAE;YACxD,GAAG,CAAC,SAAS,CAAC,GAAG,MAAM,CAAC,SAAc,CAAC,CAAC,MAAM,CAAC,CAAC,CAAC,QAAQ,CAAC,CAAC,CAAC,SAAS,CAAC;YACtE,OAAO,GAAG,CAAC;QACb,CAAC,EAAE,EAA0C,CAAC,CAAC;QAE/C,IAAI,CAAC,eAAe,CAAC,MAAM,EAAE,UAAU,CAAC,CAAC;QAEzC,MAAM,YAAY,GAAG,IAAI,CAAC,aAAa,CAAC;QACxC,IAAI,IAAI,CAAC,OAAO,CAAC,MAAM,GAAG,CAAC,EAAE,CAAC;YAC5B,IAAI,CAAC,WAAW,CAAC,YAAY,CAAC,MAAM,CAAC,GAAG,IAAI,CAAC,OAAO,CAAC,eAAe,CAAC;gBACnE,KAAK,EAAE,IAAI,CAAC,MAAM,GAAG,SAAS;gBAC9B,MAAM,EAAE,IAAI,CAAC,SAAS,CAAC,kBAAkB,CAAC,YAAY,CAAC,MAAM,CAAC;gBAC9D,OAAO,EAAE,IAAI,CAAC,OAAO,CAAC,GAAG,CAAC,CAAC,SAAS,EAAE,GAAG,EAAE,EAAE,CAAC,CAAC;oBAC7C,OAAO,EAAE,GAAG;oBACZ,QAAQ,EAAE,MAAM,CAAC,SAAc,CAAC,CAAC,MAAM;wBACrC,CAAC,CAAC;4BACE,MAAM,EAAE,MAAM,CAAC,SAAc,CAAC,CAAC,MAAO;yBACvC;wBACH,CAAC,CAAC,MAAM,CAAC,SAAc,CAAC,CAAC,OAAQ,CAAC,UAAU,EAAE;iBACjD,CAAC,CAAC;aACJ,CAAC,CAAC;QACL,CAAC;QAED,UAAU;QACV,IAAI,CAAC,WAAW,CAAC,YAAY,CAAC,OAAO,CAAC,GAAG,IAAI,CAAC,OAAO,CAAC,eAAe,CAAC;YACpE,KAAK,EAAE,IAAI,CAAC,MAAM,GAAG,UAAU;YAC/B,MAAM,EAAE,IAAI,CAAC,SAAS,CAAC,kBAAkB,CAAC,YAAY,CAAC,OAAO,CAAC;YAC/D,OAAO,EAAE,IAAI,CAAC,QAAQ,CAAC,GAAG,CAAC,CAAC,UAAU,EAAE,GAAG,EAAE,EAAE,CAAC,CAAC;gBAC/C,OAAO,EAAE,GAAG;gBACZ,QAAQ,EAAE;oBACR,MAAM,EAAE,aAAa,CAAC,UAAU,CAAC;iBAClC;aACF,CAAC,CAAC;SACJ,CAAC,CAAC;QAEH,wBAAwB;QACxB,MAAM,WAAW,GAAG,cAAc,CAAC,gBAAgB,EAAE,CAAC;QAEtD,mBAAmB;QACnB,WAAW,CAAC,WAAW,CAAC,IAAI,CAAC,SAAS,CAAC,CAAC;QACxC,cAAc;QACd,IAAI,CAAC,WAAW,CAAC,OAAO,CAAC,CAAC,SAAS,EAAE,GAAG,EAAE,EAAE;YAC1C,WAAW,CAAC,YAAY,CAAC,GAAG,EAAE,SAAS,CAAC,CAAC;QAC3C,CAAC,CAAC,CAAC;QAEH,MAAM,aAAa,GAAG,IAAI,CAAC,uBAAuB,EAAE,CAAC;QACrD,WAAW,CAAC,kBAAkB,CAC5B,IAAI,CAAC,IAAI,CAAC,CAAC,IAAI,CAAC,UAAU,IAAI,IAAI,CAAC,MAAM,CAAC,GAAG,aAAa,CAAC,CAAC,CAAC,CAAC,EAC9D,IAAI,CAAC,IAAI,CAAC,CAAC,IAAI,CAAC,WAAW,IAAI,IAAI,CAAC,OAAO,CAAC,GAAG,aAAa,CAAC,CAAC,CAAC,CAAC,EAChE,CAAC,CACF,CAAC;QACF,sBAAsB;QACtB,WAAW,CAAC,GAAG,EAAE,CAAC;IACpB,CAAC;CACF","sourcesContent":["import { getTargetShaderCache } from './WGPUShaderCache.js';\n// TODO output read write buffer\nimport { getUniformBuffer, type Uniform } from './WGPUFullQuadPass.js';\n\nexport type { Uniform };\n\nfunction isStorageParamsEqual(\n  params: WGPUComputePassOutput,\n  other: WGPUComputePassOutput\n) {\n  return (\n    params.channels === other.channels &&\n    params.type === other.type &&\n    params.count === other.count &&\n    params.atomic === other.atomic\n  );\n}\n\nexport const WORKGROUP_SIZE = 8;\n\nexport interface WGPUComputePassInput {\n  buffer?: GPUBuffer;\n  texture?: GPUTexture;\n  channels: number;\n  type?: 'u32' | 'i32' | 'f32';\n}\n\nexport interface WGPUComputePassOutput {\n  channels?: number;\n  // Default: 1. Count per element.\n  count?: number;\n  atomic?: boolean;\n  type?: 'u32' | 'i32' | 'f32';\n}\n\nexport class WGPUComputePass<I extends string, O extends string> {\n  private _label;\n\n  private _device;\n  private _outputBuffers: Record<string, GPUBuffer> = {};\n  private _outputParams: Record<string, WGPUComputePassOutput> = {};\n\n  private _pipeline!: GPUComputePipeline;\n  private _bindGroups: GPUBindGroup[] = [];\n  private _needsUpdatePipeline = true;\n\n  private _inputs: string[] = [];\n  private _outputs: string[] = [];\n  private _uniforms: Uniform[] = [];\n  private _uniformsMap: Record<string, Uniform> = {};\n  private _uniformBuffers: Record<string, GPUBuffer> = {};\n\n  private _capacityWidth = 0;\n  private _capacityHeight = 0;\n  private _reserveCapacity = false;\n  private _width = 10;\n  private _height = 10;\n\n  private _execWidth?: number;\n  private _execHeight?: number;\n\n  private _workGroupSize = [WORKGROUP_SIZE, WORKGROUP_SIZE];\n\n  private _csCode = '';\n  private _csMain;\n  private _csDefine;\n\n  private _groupOffsets = {\n    inputs: 0,\n    uniforms: 1,\n    outputs: 2\n  };\n\n  constructor(\n    label: string,\n    device: GPUDevice,\n    opts: {\n      inputs: I[];\n      outputs: O[];\n      csMain: string;\n      csDefine?: string;\n      uniforms?: Uniform[];\n    }\n  ) {\n    this._label = label;\n    this._device = device;\n    this._csMain = opts.csMain;\n    this._csDefine = opts.csDefine;\n    this._inputs = opts.inputs;\n    this._outputs = opts.outputs;\n    const uniforms = opts.uniforms || [];\n    this._uniforms = uniforms;\n\n    this._uniformsMap = this._uniforms.reduce((map, uniform) => {\n      map[uniform.label] = uniform;\n      return map;\n    }, {} as Record<string, Uniform>);\n    uniforms.forEach((uniform) => {\n      const data = getUniformBuffer(uniform.data, uniform.type);\n      this._uniformBuffers[uniform.label] = device.createBuffer({\n        label: this._label + '.' + uniform.label,\n        size: data.byteLength,\n        usage: GPUBufferUsage.UNIFORM | GPUBufferUsage.COPY_DST\n      });\n      this._device.queue.writeBuffer(\n        this._uniformBuffers[uniform.label],\n        0,\n        data\n      );\n    });\n  }\n\n  setCSCode({ csDefine, csMain }: { csDefine: string; csMain: string }) {\n    this._csDefine = csDefine;\n    this._csMain = csMain;\n    this._needsUpdatePipeline = true;\n  }\n\n  reserveCapacity() { this._reserveCapacity = true; this._capacityWidth = this._width; this._capacityHeight = this._height; }\n  setSize(width: number, height: number) {\n    width = Math.max(1, Math.ceil(width));\n    height = Math.max(1, Math.ceil(height));\n    const sizeChanged = width !== this._width || height !== this._height;\n    this._width = width;\n    this._height = height;\n    if (sizeChanged) {\n      this._needsUpdatePipeline = true;\n      const nextWidth = Math.max(this._capacityWidth, width), nextHeight = Math.max(this._capacityHeight, height);\n      const capacityChanged = nextWidth !== this._capacityWidth || nextHeight !== this._capacityHeight;\n      this._capacityWidth = nextWidth; this._capacityHeight = nextHeight;\n      if (this._reserveCapacity && !capacityChanged) return;\n      // Invalidate all buffers\n      for (const key in this._outputBuffers) {\n        this._outputBuffers[key].destroy();\n      }\n      this._outputBuffers = {};\n    }\n  }\n\n  setExecuteSize(width: number, height: number) {\n    width = Math.max(1, Math.ceil(width));\n    height = Math.max(1, Math.ceil(height));\n    this._execWidth = width;\n    this._execHeight = height;\n  }\n\n  setOutputParams(outputParams: Record<O, WGPUComputePassOutput>) {\n    // TODO lazy update buffers\n    this._invalidateChangedOutputBuffers(outputParams);\n    this._outputParams = outputParams;\n    this._needsUpdatePipeline = true;\n  }\n\n  setUniform(label: string, data: Uniform['data']) {\n    const buffer = this._uniformBuffers[label];\n    this._device.queue.writeBuffer(\n      buffer,\n      0,\n      getUniformBuffer(data, this._uniformsMap[label].type)\n    );\n  }\n\n  getOutput(name: O) {\n    const outputBuffers = this._outputBuffers;\n    if (!outputBuffers[name]) {\n      outputBuffers[name] = this._createBuffer(\n        name,\n        this._outputParams[name] || {}\n      );\n    }\n    return outputBuffers[name];\n  }\n\n  getMemoryBytes() {\n    return [...Object.values(this._uniformBuffers), ...Object.values(this._outputBuffers)]\n      .reduce((sum, buffer) => sum + buffer.size, 0);\n  }\n\n  dispose() {\n    Object.keys(this._uniformBuffers).forEach((key) => {\n      (this._uniformBuffers as any)[key].destroy();\n    });\n    Object.keys(this._outputBuffers).forEach((key) => {\n      (this._outputBuffers as any)[key].destroy();\n    });\n  }\n\n  setWorkGroupSize(width: number, height: number) {\n    this._needsUpdatePipeline =\n      width !== this._workGroupSize[0] || height !== this._workGroupSize[1];\n    this._workGroupSize = [width, height];\n  }\n\n  private _getActualWorkgroupSize() {\n    return [\n      Math.min(this._workGroupSize[0], this._execWidth ?? this._width),\n      Math.min(this._workGroupSize[1], this._execHeight ?? this._height)\n    ];\n  }\n\n  private _createBuffer(label: string, params: WGPUComputePassOutput) {\n    const channels = params.channels ?? 4;\n    const byteLength =\n      (this._reserveCapacity ? this._capacityWidth * this._capacityHeight : this._width * this._height) * channels * (params.count ?? 1) * 4;\n    return this._device.createBuffer({\n      label: this._label + '.' + label,\n      // webgpu needs buffer at least 80 bytes.\n      size: Math.max(byteLength, 80),\n      usage: GPUBufferUsage.STORAGE | GPUBufferUsage.COPY_SRC\n    });\n  }\n\n  private _invalidateChangedOutputBuffers(newParams: Record<string, any>) {\n    const outputBuffers = this._outputBuffers;\n    const oldParams = this._outputParams;\n    for (const key in outputBuffers) {\n      if (!isStorageParamsEqual(newParams[key] || {}, oldParams[key] || {})) {\n        outputBuffers[key].destroy();\n        delete outputBuffers[key];\n      }\n    }\n  }\n\n  private _updateOutputBuffers(excludes?: string[]) {\n    const outputBuffers = this._outputBuffers;\n    const params = this._outputParams;\n    for (const key in params) {\n      if (!outputBuffers[key] && !excludes?.includes(key)) {\n        outputBuffers[key] = this._createBuffer(key, params[key]);\n      }\n    }\n  }\n\n  private _updatePipeline(\n    inputParams: Record<string, WGPUComputePassInput>,\n    inputTypes: Record<string, 'texture' | 'buffer'>\n  ) {\n    if (!this._needsUpdatePipeline) {\n      return;\n    }\n    this._needsUpdatePipeline = false;\n    const device = this._device;\n    const csCode = this._getFullCs(inputParams, inputTypes);\n    if (csCode === this._csCode) {\n      return;\n    }\n    this._csCode = csCode;\n    this._pipeline = getTargetShaderCache(device).createComputePipeline({\n      label: this._label,\n      layout: 'auto',\n      compute: {\n        module: getTargetShaderCache(device).createShaderModule({\n          label: this._label,\n          code: csCode\n        }),\n        entryPoint: 'main'\n      }\n    });\n    this._updateBindGroups();\n  }\n\n  private _getFullCs(\n    inputParams: Record<string, WGPUComputePassInput>,\n    inputTypes: Record<string, 'texture' | 'buffer'>\n  ) {\n    const inputs = this._inputs;\n    const uniforms = this._uniforms;\n    let offset = 0;\n    const groupOffsets = (this._groupOffsets = {\n      inputs: 0,\n      uniforms: 0,\n      outputs: 0\n    });\n    const workgroupSize = this._getActualWorkgroupSize();\n\n    if (inputs.length > 0) {\n      offset++;\n    }\n    if (uniforms.length > 0) {\n      groupOffsets.uniforms = offset;\n      offset++;\n    }\n\n    function getTypeCode(params: WGPUComputePassOutput) {\n      const { type = 'f32', channels } = params;\n      let code = channels === 1 ? type : `vec${channels}${type[0]}`;\n      if (params.atomic) {\n        code = `atomic<${code}>`;\n      }\n      return code;\n    }\n    groupOffsets.outputs = offset;\n    const cs = /* wgsl */ `\n${inputs\n  .sort()\n  .map((inputName, idx) => {\n    const bindingPrefix = `@group(${groupOffsets.inputs}) @binding(${idx}) `;\n    const varName = `in_${inputName}`;\n    // TODO more channels option.\n    return inputTypes[inputName] === 'texture'\n      ? `${bindingPrefix} var ${varName}: texture_2d<f32>;`\n      : `${bindingPrefix} var<storage, read> ${varName}: array<${getTypeCode(\n          inputParams[inputName]\n        )}>;`;\n  })\n  .join('\\n')}\n${this._uniforms\n  .map(\n    (uniform, idx) =>\n      `@group(${groupOffsets.uniforms}) @binding(${idx}) var<uniform> ${uniform.label}: ${uniform.type};`\n  )\n  .join('\\n')}\n\n${this._outputs\n  .map((name, idx) => {\n    return `@group(${\n      groupOffsets.outputs\n    }) @binding(${idx}) var<storage, read_write> out_${name}: array<${getTypeCode(\n      this._outputParams[name] || {}\n    )}>;`;\n  })\n  .join('\\n')}\n${this._csDefine ?? ''}\n@compute @workgroup_size(${workgroupSize[0]}, ${workgroupSize[1]}, 1)\nfn main(@builtin(global_invocation_id) globalId: vec3u) {\n${this._csMain}\n}\n`;\n\n    return cs;\n  }\n\n  private _updateBindGroups() {\n    const bindGroups: GPUBindGroup[] = [];\n    const device = this._device;\n    const groupOffsets = this._groupOffsets;\n\n    if (this._uniforms.length > 0) {\n      bindGroups[groupOffsets.uniforms] = device.createBindGroup({\n        label: this._label + '.uniforms',\n        layout: this._pipeline.getBindGroupLayout(groupOffsets.uniforms),\n        entries: this._uniforms.map(\n          (uniform, idx) =>\n            ({\n              binding: idx,\n              resource: {\n                buffer: this._uniformBuffers[uniform.label]\n              }\n            } as GPUBindGroupEntry)\n        )\n      });\n    }\n\n    this._bindGroups = bindGroups;\n  }\n\n  createPass(\n    commandEncoder: GPUCommandEncoder,\n    inputs: Record<I, WGPUComputePassInput>,\n\n    /**\n     * Explict outputs. Used for buffers reuse.\n     */\n    outputs?: Record<O, GPUBuffer>\n  ) {\n    this._updateOutputBuffers(Object.keys(outputs || {}));\n    const outputBuffers = {\n      ...this._outputBuffers,\n      ...outputs\n    } as Record<string, GPUBuffer>;\n\n    const inputTypes = this._inputs.reduce((obj, inputName) => {\n      obj[inputName] = inputs[inputName as I].buffer ? 'buffer' : 'texture';\n      return obj;\n    }, {} as Record<string, 'texture' | 'buffer'>);\n\n    this._updatePipeline(inputs, inputTypes);\n\n    const groupOffsets = this._groupOffsets;\n    if (this._inputs.length > 0) {\n      this._bindGroups[groupOffsets.inputs] = this._device.createBindGroup({\n        label: this._label + '.inputs',\n        layout: this._pipeline.getBindGroupLayout(groupOffsets.inputs),\n        entries: this._inputs.map((inputName, idx) => ({\n          binding: idx,\n          resource: inputs[inputName as I].buffer\n            ? {\n                buffer: inputs[inputName as I].buffer!\n              }\n            : inputs[inputName as I].texture!.createView()\n        }))\n      });\n    }\n\n    // Outputs\n    this._bindGroups[groupOffsets.outputs] = this._device.createBindGroup({\n      label: this._label + '.outputs',\n      layout: this._pipeline.getBindGroupLayout(groupOffsets.outputs),\n      entries: this._outputs.map((bufferName, idx) => ({\n        binding: idx,\n        resource: {\n          buffer: outputBuffers[bufferName]\n        }\n      }))\n    });\n\n    // Begin the render pass\n    const computePass = commandEncoder.beginComputePass();\n\n    // Draw a full quad\n    computePass.setPipeline(this._pipeline);\n    // Bind groups\n    this._bindGroups.forEach((bindGroup, idx) => {\n      computePass.setBindGroup(idx, bindGroup);\n    });\n\n    const workgroupSize = this._getActualWorkgroupSize();\n    computePass.dispatchWorkgroups(\n      Math.ceil((this._execWidth ?? this._width) / workgroupSize[0]),\n      Math.ceil((this._execHeight ?? this._height) / workgroupSize[1]),\n      1\n    );\n    // End the render pass\n    computePass.end();\n  }\n}\n"]}

package/lib/renderer/WGPUEnvMapDistribution.d.ts

@@ -0,0 +1,12 @@
+/** Builds the existing CDF/PDF texture layout without CPU readback or upload. */
+export declare class WGPUEnvMapDistribution {
+    private readonly device;
+    private readonly layout;
+    private readonly pipelines;
+    private pixels?;
+    private rows?;
+    static supported(device: GPUDevice, width: number, height: number): boolean;
+    constructor(device: GPUDevice);
+    build(input: GPUTexture, output: GPUTexture): void;
+    dispose(): void;
+}

package/lib/renderer/WGPUEnvMapDistribution.js

@@ -0,0 +1,150 @@
+const distributionWGSL = /* wgsl */ `
+@group(0) @binding(0) var radiance: texture_2d<f32>;
+
+@group(0) @binding(1) var<storage, read_write> pixels: array<vec2f>;
+@group(0) @binding(2) var<storage, read_write> rows: array<vec2f>;
+@group(0) @binding(3) var output: texture_storage_2d<rg32float, write>;
+var<workgroup> scan: array<f32, 256>;
+var<workgroup> tw: f32;
+
+fn iU(pT: u32) {
+workgroupBarrier();
+for (var offset = 1u; offset < 256u; offset *= 2u) {
+var previous = 0.0;
+if (pT >= offset) { previous = scan[pT - offset]; }
+workgroupBarrier();
+scan[pT] += previous;
+workgroupBarrier();
+}
+}
+
+@compute @workgroup_size(256)
+fn scanRows(@builtin(workgroup_id) group: vec3u, @builtin(local_invocation_index) pT: u32) {
+let size = textureDimensions(radiance);
+let y = group.x;
+if (pT == 0u) { tw = 0.0; }
+workgroupBarrier();
+for (var start = 0u; start < size.x; start += 256u) {
+let x = start + pT;
+var weight = 0.0;
+if (x < size.x) {
+
+var maximum = 0.0;
+for (var dy = -1; dy <= 1; dy++) {
+for (var dx = -1; dx <= 1; dx++) {
+let p = clamp(vec2i(i32(x) + dx, i32(y) + dy), vec2i(0), vec2i(size) - 1);
+maximum = max(maximum, dot(textureLoad(radiance, p, 0).rgb, vec3f(0.2126, 0.7152, 0.0722)));
+}
+}
+weight = max(maximum, 1e-12) * sin(3.141592653589793 * (f32(y) + 0.5) / f32(size.y));
+}
+scan[pT] = weight / f32(size.x);
+iU(pT);
+if (x < size.x) { pixels[y * size.x + x] = vec2f(tw + scan[pT], weight); }
+workgroupBarrier();
+if (pT == 0u) { tw += scan[255]; }
+workgroupBarrier();
+}
+if (pT == 0u) { rows[y] = vec2f(0.0, tw); }
+}
+
+@compute @workgroup_size(256)
+fn scanMarginal(@builtin(local_invocation_index) pT: u32) {
+let height = textureDimensions(radiance).y;
+if (pT == 0u) { tw = 0.0; }
+workgroupBarrier();
+for (var start = 0u; start < height; start += 256u) {
+let y = start + pT;
+var integral = 0.0;
+if (y < height) { integral = rows[y].y; }
+scan[pT] = integral / f32(height);
+iU(pT);
+if (y < height) { rows[y].x = tw + scan[pT]; }
+workgroupBarrier();
+if (pT == 0u) { tw += scan[255]; }
+workgroupBarrier();
+}
+}
+
+@compute @workgroup_size(16, 16)
+fn normalize(@builtin(global_invocation_id) id: vec3u) {
+let size = textureDimensions(radiance);
+let x = id.x;
+let y = id.y;
+if (x >= size.x + 2u || y >= size.y + 1u) { return; }
+var value = vec2f(0.0);
+if (x == 0u) {
+let total = rows[size.y - 1u].x;
+if (y > 0u) { value.x = rows[y - 1u].x / total; }
+if (y < size.y) { value.y = rows[y].y / total; }
+} else if (y < size.y) {
+let integral = rows[y].y;
+if (x > 1u) { value.x = pixels[y * size.x + x - 2u].x / integral; }
+if (x <= size.x) { value.y = pixels[y * size.x + x - 1u].y / integral; }
+}
+textureStore(output, vec2i(id.xy), vec4f(value, 0.0, 0.0));
+}
+`;
+/** Builds the existing CDF/PDF texture layout without CPU readback or upload. */
+export class WGPUEnvMapDistribution {
+    device;
+    layout;
+    pipelines;
+    pixels;
+    rows;
+    static supported(device, width, height) {
+        // Large HDRIs can exceed a single storage binding; retain the CPU path there.
+        return width * height * 8 <= Math.min(device.limits.maxStorageBufferBindingSize, device.limits.maxBufferSize);
+    }
+    constructor(device) {
+        this.device = device;
+        this.layout = device.createBindGroupLayout({
+            label: 'Environment distribution',
+            entries: [
+                { binding: 0, visibility: GPUShaderStage.COMPUTE, texture: { sampleType: 'unfilterable-float' } },
+                { binding: 1, visibility: GPUShaderStage.COMPUTE, buffer: { type: 'storage' } },
+                { binding: 2, visibility: GPUShaderStage.COMPUTE, buffer: { type: 'storage' } },
+                { binding: 3, visibility: GPUShaderStage.COMPUTE, storageTexture: { access: 'write-only', format: 'rg32float' } }
+            ]
+        });
+        const module = device.createShaderModule({ label: 'Environment distribution', code: distributionWGSL });
+        const layout = device.createPipelineLayout({ bindGroupLayouts: [this.layout] });
+        this.pipelines = ['scanRows', 'scanMarginal', 'normalize'].map(entryPoint => device.createComputePipeline({
+            label: `Environment distribution ${entryPoint}`, layout, compute: { module, entryPoint }
+        }));
+    }
+    build(input, output) {
+        const { width, height } = input;
+        if (this.pixels?.size !== width * height * 8 || this.rows?.size !== height * 8) {
+            this.dispose();
+            this.pixels = this.device.createBuffer({ label: 'Environment row scans', size: width * height * 8, usage: GPUBufferUsage.STORAGE });
+            this.rows = this.device.createBuffer({ label: 'Environment marginal scan', size: height * 8, usage: GPUBufferUsage.STORAGE });
+        }
+        const group = this.device.createBindGroup({
+            layout: this.layout,
+            entries: [
+                { binding: 0, resource: input.createView() },
+                { binding: 1, resource: { buffer: this.pixels } },
+                { binding: 2, resource: { buffer: this.rows } },
+                { binding: 3, resource: output.createView() }
+            ]
+        });
+        const encoder = this.device.createCommandEncoder({ label: 'Environment distribution' });
+        const workgroups = [[height, 1], [1, 1], [Math.ceil((width + 2) / 16), Math.ceil((height + 1) / 16)]];
+        this.pipelines.forEach((pipeline, index) => {
+            const pass = encoder.beginComputePass();
+            pass.setPipeline(pipeline);
+            pass.setBindGroup(0, group);
+            pass.dispatchWorkgroups(...workgroups[index]);
+            pass.end();
+        });
+        this.device.queue.submit([encoder.finish()]);
+    }
+    dispose() {
+        this.pixels?.destroy();
+        this.rows?.destroy();
+        this.pixels = undefined;
+        this.rows = undefined;
+    }
+}
+//# sourceMappingURL=WGPUEnvMapDistribution.js.map

package/lib/renderer/WGPUEnvMapDistribution.js.map

@@ -0,0 +1 @@
+{"version":3,"file":"WGPUEnvMapDistribution.js","sourceRoot":"","sources":["../../src/renderer/WGPUEnvMapDistribution.ts"],"names":[],"mappings":"AAAA,MAAM,gBAAgB,GAAG,UAAU,CAAC;;;;;;;;;;;;;;;;;;;;;;;;;;;;;;;;;;;;;;;;;;;;;;;;;;;;;;;;;;;;;;;;;;;;;;;;;;;;;;;;;;;;;;CAsFnC,CAAC;AAEF,iFAAiF;AACjF,MAAM,OAAO,sBAAsB;IAWJ;IAVZ,MAAM,CAAqB;IAC3B,SAAS,CAAuB;IACzC,MAAM,CAAa;IACnB,IAAI,CAAa;IAEzB,MAAM,CAAC,SAAS,CAAC,MAAiB,EAAE,KAAa,EAAE,MAAc;QAC/D,8EAA8E;QAC9E,OAAO,KAAK,GAAG,MAAM,GAAG,CAAC,IAAI,IAAI,CAAC,GAAG,CAAC,MAAM,CAAC,MAAM,CAAC,2BAA2B,EAAE,MAAM,CAAC,MAAM,CAAC,aAAa,CAAC,CAAC;IAChH,CAAC;IAED,YAA6B,MAAiB;QAAjB,WAAM,GAAN,MAAM,CAAW;QAC5C,IAAI,CAAC,MAAM,GAAG,MAAM,CAAC,qBAAqB,CAAC;YACzC,KAAK,EAAE,0BAA0B;YACjC,OAAO,EAAE;gBACP,EAAE,OAAO,EAAE,CAAC,EAAE,UAAU,EAAE,cAAc,CAAC,OAAO,EAAE,OAAO,EAAE,EAAE,UAAU,EAAE,oBAAoB,EAAE,EAAE;gBACjG,EAAE,OAAO,EAAE,CAAC,EAAE,UAAU,EAAE,cAAc,CAAC,OAAO,EAAE,MAAM,EAAE,EAAE,IAAI,EAAE,SAAS,EAAE,EAAE;gBAC/E,EAAE,OAAO,EAAE,CAAC,EAAE,UAAU,EAAE,cAAc,CAAC,OAAO,EAAE,MAAM,EAAE,EAAE,IAAI,EAAE,SAAS,EAAE,EAAE;gBAC/E,EAAE,OAAO,EAAE,CAAC,EAAE,UAAU,EAAE,cAAc,CAAC,OAAO,EAAE,cAAc,EAAE,EAAE,MAAM,EAAE,YAAY,EAAE,MAAM,EAAE,WAAW,EAAE,EAAE;aAClH;SACF,CAAC,CAAC;QACH,MAAM,MAAM,GAAG,MAAM,CAAC,kBAAkB,CAAC,EAAE,KAAK,EAAE,0BAA0B,EAAE,IAAI,EAAE,gBAAgB,EAAE,CAAC,CAAC;QACxG,MAAM,MAAM,GAAG,MAAM,CAAC,oBAAoB,CAAC,EAAE,gBAAgB,EAAE,CAAC,IAAI,CAAC,MAAM,CAAC,EAAE,CAAC,CAAC;QAChF,IAAI,CAAC,SAAS,GAAG,CAAC,UAAU,EAAE,cAAc,EAAE,WAAW,CAAC,CAAC,GAAG,CAAC,UAAU,CAAC,EAAE,CAAC,MAAM,CAAC,qBAAqB,CAAC;YACxG,KAAK,EAAE,4BAA4B,UAAU,EAAE,EAAE,MAAM,EAAE,OAAO,EAAE,EAAE,MAAM,EAAE,UAAU,EAAE;SACzF,CAAC,CAAC,CAAC;IACN,CAAC;IAED,KAAK,CAAC,KAAiB,EAAE,MAAkB;QACzC,MAAM,EAAE,KAAK,EAAE,MAAM,EAAE,GAAG,KAAK,CAAC;QAChC,IAAI,IAAI,CAAC,MAAM,EAAE,IAAI,KAAK,KAAK,GAAG,MAAM,GAAG,CAAC,IAAI,IAAI,CAAC,IAAI,EAAE,IAAI,KAAK,MAAM,GAAG,CAAC,EAAE,CAAC;YAC/E,IAAI,CAAC,OAAO,EAAE,CAAC;YACf,IAAI,CAAC,MAAM,GAAG,IAAI,CAAC,MAAM,CAAC,YAAY,CAAC,EAAE,KAAK,EAAE,uBAAuB,EAAE,IAAI,EAAE,KAAK,GAAG,MAAM,GAAG,CAAC,EAAE,KAAK,EAAE,cAAc,CAAC,OAAO,EAAE,CAAC,CAAC;YACpI,IAAI,CAAC,IAAI,GAAG,IAAI,CAAC,MAAM,CAAC,YAAY,CAAC,EAAE,KAAK,EAAE,2BAA2B,EAAE,IAAI,EAAE,MAAM,GAAG,CAAC,EAAE,KAAK,EAAE,cAAc,CAAC,OAAO,EAAE,CAAC,CAAC;QAChI,CAAC;QACD,MAAM,KAAK,GAAG,IAAI,CAAC,MAAM,CAAC,eAAe,CAAC;YACxC,MAAM,EAAE,IAAI,CAAC,MAAM;YACnB,OAAO,EAAE;gBACP,EAAE,OAAO,EAAE,CAAC,EAAE,QAAQ,EAAE,KAAK,CAAC,UAAU,EAAE,EAAE;gBAC5C,EAAE,OAAO,EAAE,CAAC,EAAE,QAAQ,EAAE,EAAE,MAAM,EAAE,IAAI,CAAC,MAAO,EAAE,EAAE;gBAClD,EAAE,OAAO,EAAE,CAAC,EAAE,QAAQ,EAAE,EAAE,MAAM,EAAE,IAAI,CAAC,IAAK,EAAE,EAAE;gBAChD,EAAE,OAAO,EAAE,CAAC,EAAE,QAAQ,EAAE,MAAM,CAAC,UAAU,EAAE,EAAE;aAC9C;SACF,CAAC,CAAC;QACH,MAAM,OAAO,GAAG,IAAI,CAAC,MAAM,CAAC,oBAAoB,CAAC,EAAE,KAAK,EAAE,0BAA0B,EAAE,CAAC,CAAC;QACxF,MAAM,UAAU,GAAG,CAAC,CAAC,MAAM,EAAE,CAAC,CAAC,EAAE,CAAC,CAAC,EAAE,CAAC,CAAC,EAAE,CAAC,IAAI,CAAC,IAAI,CAAC,CAAC,KAAK,GAAG,CAAC,CAAC,GAAG,EAAE,CAAC,EAAE,IAAI,CAAC,IAAI,CAAC,CAAC,MAAM,GAAG,CAAC,CAAC,GAAG,EAAE,CAAC,CAAC,CAAC,CAAC;QACtG,IAAI,CAAC,SAAS,CAAC,OAAO,CAAC,CAAC,QAAQ,EAAE,KAAK,EAAE,EAAE;YACzC,MAAM,IAAI,GAAG,OAAO,CAAC,gBAAgB,EAAE,CAAC;YACxC,IAAI,CAAC,WAAW,CAAC,QAAQ,CAAC,CAAC;YAC3B,IAAI,CAAC,YAAY,CAAC,CAAC,EAAE,KAAK,CAAC,CAAC;YAC5B,IAAI,CAAC,kBAAkB,CAAC,GAAG,UAAU,CAAC,KAAK,CAAqB,CAAC,CAAC;YAClE,IAAI,CAAC,GAAG,EAAE,CAAC;QACb,CAAC,CAAC,CAAC;QACH,IAAI,CAAC,MAAM,CAAC,KAAK,CAAC,MAAM,CAAC,CAAC,OAAO,CAAC,MAAM,EAAE,CAAC,CAAC,CAAC;IAC/C,CAAC;IAED,OAAO;QACL,IAAI,CAAC,MAAM,EAAE,OAAO,EAAE,CAAC;QACvB,IAAI,CAAC,IAAI,EAAE,OAAO,EAAE,CAAC;QACrB,IAAI,CAAC,MAAM,GAAG,SAAS,CAAC;QACxB,IAAI,CAAC,IAAI,GAAG,SAAS,CAAC;IACxB,CAAC;CACF","sourcesContent":["const distributionWGSL = /* wgsl */ `\n@group(0) @binding(0) var radiance: texture_2d<f32>;\n// Per-texel inclusive row integral and unnormalized PDF, then per-row CDF/PDF.\n@group(0) @binding(1) var<storage, read_write> pixels: array<vec2f>;\n@group(0) @binding(2) var<storage, read_write> rows: array<vec2f>;\n@group(0) @binding(3) var output: texture_storage_2d<rg32float, write>;\nvar<workgroup> scan: array<f32, 256>;\nvar<workgroup> carry: f32;\n\nfn inclusiveScan(lane: u32) {\n  workgroupBarrier();\n  for (var offset = 1u; offset < 256u; offset *= 2u) {\n    var previous = 0.0;\n    if (lane >= offset) { previous = scan[lane - offset]; }\n    workgroupBarrier();\n    scan[lane] += previous;\n    workgroupBarrier();\n  }\n}\n\n@compute @workgroup_size(256)\nfn scanRows(@builtin(workgroup_id) group: vec3u, @builtin(local_invocation_index) lane: u32) {\n  let size = textureDimensions(radiance);\n  let y = group.x;\n  if (lane == 0u) { carry = 0.0; }\n  workgroupBarrier();\n  for (var start = 0u; start < size.x; start += 256u) {\n    let x = start + lane;\n    var weight = 0.0;\n    if (x < size.x) {\n      // Match the CPU proposal's clamped 3×3 footprint around filtered HDR radiance.\n      var maximum = 0.0;\n      for (var dy = -1; dy <= 1; dy++) {\n        for (var dx = -1; dx <= 1; dx++) {\n          let p = clamp(vec2i(i32(x) + dx, i32(y) + dy), vec2i(0), vec2i(size) - 1);\n          maximum = max(maximum, dot(textureLoad(radiance, p, 0).rgb, vec3f(0.2126, 0.7152, 0.0722)));\n        }\n      }\n      weight = max(maximum, 1e-12) * sin(3.141592653589793 * (f32(y) + 0.5) / f32(size.y));\n    }\n    scan[lane] = weight / f32(size.x);\n    inclusiveScan(lane);\n    if (x < size.x) { pixels[y * size.x + x] = vec2f(carry + scan[lane], weight); }\n    workgroupBarrier();\n    if (lane == 0u) { carry += scan[255]; }\n    workgroupBarrier();\n  }\n  if (lane == 0u) { rows[y] = vec2f(0.0, carry); }\n}\n\n@compute @workgroup_size(256)\nfn scanMarginal(@builtin(local_invocation_index) lane: u32) {\n  let height = textureDimensions(radiance).y;\n  if (lane == 0u) { carry = 0.0; }\n  workgroupBarrier();\n  for (var start = 0u; start < height; start += 256u) {\n    let y = start + lane;\n    var integral = 0.0;\n    if (y < height) { integral = rows[y].y; }\n    scan[lane] = integral / f32(height);\n    inclusiveScan(lane);\n    if (y < height) { rows[y].x = carry + scan[lane]; }\n    workgroupBarrier();\n    if (lane == 0u) { carry += scan[255]; }\n    workgroupBarrier();\n  }\n}\n\n@compute @workgroup_size(16, 16)\nfn normalize(@builtin(global_invocation_id) id: vec3u) {\n  let size = textureDimensions(radiance);\n  let x = id.x;\n  let y = id.y;\n  if (x >= size.x + 2u || y >= size.y + 1u) { return; }\n  var value = vec2f(0.0);\n  if (x == 0u) {\n    let total = rows[size.y - 1u].x;\n    if (y > 0u) { value.x = rows[y - 1u].x / total; }\n    if (y < size.y) { value.y = rows[y].y / total; }\n  } else if (y < size.y) {\n    let integral = rows[y].y;\n    if (x > 1u) { value.x = pixels[y * size.x + x - 2u].x / integral; }\n    if (x <= size.x) { value.y = pixels[y * size.x + x - 1u].y / integral; }\n  }\n  textureStore(output, vec2i(id.xy), vec4f(value, 0.0, 0.0));\n}\n`;\n\n/** Builds the existing CDF/PDF texture layout without CPU readback or upload. */\nexport class WGPUEnvMapDistribution {\n  private readonly layout: GPUBindGroupLayout;\n  private readonly pipelines: GPUComputePipeline[];\n  private pixels?: GPUBuffer;\n  private rows?: GPUBuffer;\n\n  static supported(device: GPUDevice, width: number, height: number) {\n    // Large HDRIs can exceed a single storage binding; retain the CPU path there.\n    return width * height * 8 <= Math.min(device.limits.maxStorageBufferBindingSize, device.limits.maxBufferSize);\n  }\n\n  constructor(private readonly device: GPUDevice) {\n    this.layout = device.createBindGroupLayout({\n      label: 'Environment distribution',\n      entries: [\n        { binding: 0, visibility: GPUShaderStage.COMPUTE, texture: { sampleType: 'unfilterable-float' } },\n        { binding: 1, visibility: GPUShaderStage.COMPUTE, buffer: { type: 'storage' } },\n        { binding: 2, visibility: GPUShaderStage.COMPUTE, buffer: { type: 'storage' } },\n        { binding: 3, visibility: GPUShaderStage.COMPUTE, storageTexture: { access: 'write-only', format: 'rg32float' } }\n      ]\n    });\n    const module = device.createShaderModule({ label: 'Environment distribution', code: distributionWGSL });\n    const layout = device.createPipelineLayout({ bindGroupLayouts: [this.layout] });\n    this.pipelines = ['scanRows', 'scanMarginal', 'normalize'].map(entryPoint => device.createComputePipeline({\n      label: `Environment distribution ${entryPoint}`, layout, compute: { module, entryPoint }\n    }));\n  }\n\n  build(input: GPUTexture, output: GPUTexture) {\n    const { width, height } = input;\n    if (this.pixels?.size !== width * height * 8 || this.rows?.size !== height * 8) {\n      this.dispose();\n      this.pixels = this.device.createBuffer({ label: 'Environment row scans', size: width * height * 8, usage: GPUBufferUsage.STORAGE });\n      this.rows = this.device.createBuffer({ label: 'Environment marginal scan', size: height * 8, usage: GPUBufferUsage.STORAGE });\n    }\n    const group = this.device.createBindGroup({\n      layout: this.layout,\n      entries: [\n        { binding: 0, resource: input.createView() },\n        { binding: 1, resource: { buffer: this.pixels! } },\n        { binding: 2, resource: { buffer: this.rows! } },\n        { binding: 3, resource: output.createView() }\n      ]\n    });\n    const encoder = this.device.createCommandEncoder({ label: 'Environment distribution' });\n    const workgroups = [[height, 1], [1, 1], [Math.ceil((width + 2) / 16), Math.ceil((height + 1) / 16)]];\n    this.pipelines.forEach((pipeline, index) => {\n      const pass = encoder.beginComputePass();\n      pass.setPipeline(pipeline);\n      pass.setBindGroup(0, group);\n      pass.dispatchWorkgroups(...workgroups[index] as [number, number]);\n      pass.end();\n    });\n    this.device.queue.submit([encoder.finish()]);\n  }\n\n  dispose() {\n    this.pixels?.destroy();\n    this.rows?.destroy();\n    this.pixels = undefined;\n    this.rows = undefined;\n  }\n}\n"]}

package/lib/renderer/WGPUFullQuadPass.js

@@ -131,16 +131,17 @@ export class WGPUFullQuadPass {
         }
     }
     setOutputParams(outputParams) {
+        const wasScreen = !!this._renderToScreen;
         this._renderToScreen = undefined;
         const outputsChanged = Object.keys(outputParams).some((key) => !isTextureParamsEqual(outputParams[key], this._outputParams[key] || {}));
         this._needsUpdateOutputTextures ||= outputsChanged;
-        this._needsUpdatePipeline ||= outputsChanged;
-        if (outputsChanged)
+        this._needsUpdatePipeline ||= outputsChanged || wasScreen;
+        if (outputsChanged || wasScreen)
             this._fsCode = '';
         this._outputParams = outputParams;
     }
     setRenderToScreen(screenTexture, presentationFormat) {
-        if (this._renderToScreen &&
+        if (!this._renderToScreen ||
             this._renderToScreen.presentationFormat !== presentationFormat) {
             this._needsUpdatePipeline = true;
             this._fsCode = '';
@@ -273,7 +274,7 @@ ${hasInputs
 ${this._uniforms
             .map((uniform, idx) => `@group(${hasInputs ? 1 : 0}) @binding(${idx}) var<uniform> ${uniform.label}: ${uniform.type};`)
             .join('\n')}
-struct pd {
+struct pV {
 ${this._outputs
             .map((name, idx) => `@location(${idx}) ${name}: vec4f,`)
             .join('\n')}
@@ -282,11 +283,11 @@ ${scratchWGSLFunctions(this._reserveCapacity ? this._regionNames : inputs.filter
 ${this._fsDefine ?? ''}
 @fragment
 fn main(
-@builtin(position) fH: vec4f
-) -> pd {
-let uv = vec2i(floor(fH.xy));
-let uv2 = fH.xy / viewportSize;
-var output: pd;
+@builtin(position) coord: vec4f
+) -> pV {
+let uv = vec2i(floor(coord.xy));
+let uv2 = coord.xy / viewportSize;
+var output: pV;
 ${this._fsMain}
 return output;
 }

package/lib/renderer/WGPUFullQuadPass.js.map

@@ -1 +1 @@
-{"version":3,"file":"WGPUFullQuadPass.js","sourceRoot":"","sources":["../../src/renderer/WGPUFullQuadPass.ts"],"names":[],"mappings":"AAAA,OAAO,EAAC,oBAAoB,EAAC,MAAM,sBAAsB,CAAC;AAC1D,OAAO,EAAC,gBAAgB,EAAE,aAAa,EAAC,MAAM,oBAAoB,CAAC;AACnE,OAAO,EAAE,oBAAoB,EAAE,MAAM,sBAAsB,CAAC;AAC5D,OAAO,EAAE,YAAY,EAAE,MAAM,gBAAgB,CAAC;AAC9C,MAAM,CAAC,MAAM,8BAA8B,GAAG,SAAS,CAAC;;;;;;;;;;;;CAYvD,CAAC;AAEF,SAAS,oBAAoB,CAC3B,MAA8B,EAC9B,KAA6B;IAE7B,OAAO,MAAM,CAAC,MAAM,KAAK,KAAK,CAAC,MAAM,CAAC;AACxC,CAAC;AAED,MAAM,UAAU,gBAAgB,CAC9B,KAAU,EACV,IAAqB;IAErB,IACE,KAAK,YAAY,YAAY;QAC7B,KAAK,YAAY,UAAU;QAC3B,KAAK,YAAY,WAAW,EAC5B,CAAC;QACD,OAAO,KAAK,CAAC;IACf,CAAC;IAED,IAAI,OAAO,KAAK,KAAK,QAAQ,EAAE,CAAC;QAC9B,KAAK,GAAG,CAAC,KAAK,CAAC,CAAC;IAClB,CAAC;IAED,IAAI,IAAI,KAAK,KAAK,IAAI,IAAI,CAAC,QAAQ,CAAC,GAAG,CAAC,EAAE,CAAC;QACzC,OAAO,IAAI,UAAU,CAAC,KAAK,CAAC,CAAC;IAC/B,CAAC;SAAM,IAAI,IAAI,KAAK,KAAK,IAAI,IAAI,CAAC,QAAQ,CAAC,GAAG,CAAC,EAAE,CAAC;QAChD,OAAO,IAAI,WAAW,CAAC,KAAK,CAAC,CAAC;IAChC,CAAC;SAAM,CAAC;QACN,OAAO,IAAI,YAAY,CAAC,KAAK,CAAC,CAAC;IACjC,CAAC;AACH,CAAC;AAwBD,MAAM,OAAO,gBAAgB;IACnB,MAAM,CAAC;IAEP,OAAO,CAAC;IACR,eAAe,GAA+B,EAAE,CAAC;IACjD,aAAa,GAA2C,EAAE,CAAC;IAE3D,SAAS,CAAqB;IAC9B,WAAW,GAAmB,EAAE,CAAC;IACjC,oBAAoB,GAAG,IAAI,CAAC;IAC5B,0BAA0B,GAAG,IAAI,CAAC;IAC1C;;OAEG;IACK,eAAe,CAGrB;IACM,OAAO,GAAa,EAAE,CAAC;IACvB,oBAAoB,GAExB,EAAE,CAAC;IACC,QAAQ,GAAa,EAAE,CAAC;IACxB,SAAS,GAAc,EAAE,CAAC;IAC1B,YAAY,GAA4B,EAAE,CAAC;IAC3C,eAAe,GAA8B,EAAE,CAAC;IAEhD,SAAS,GAA+B,EAAE,CAAC;IAE3C,cAAc,GAAG,CAAC,CAAC;IACnB,eAAe,GAAG,CAAC,CAAC;IACpB,gBAAgB,GAAG,KAAK,CAAC;IACzB,YAAY,GAAa,EAAE,CAAC;IAC5B,MAAM,GAAG,EAAE,CAAC;IACZ,OAAO,GAAG,EAAE,CAAC;IAEb,OAAO,GAAG,EAAE,CAAC;IACJ,OAAO,CAAC;IACR,SAAS,CAAC;IAEnB,kBAAkB,CAAC;IAE3B,YACE,KAAa,EACb,MAAiB,EACjB,IASC;QAED,IAAI,CAAC,MAAM,GAAG,KAAK,CAAC;QACpB,IAAI,CAAC,OAAO,GAAG,MAAM,CAAC;QACtB,IAAI,CAAC,OAAO,GAAG,IAAI,CAAC,MAAM,CAAC;QAC3B,IAAI,CAAC,SAAS,GAAG,IAAI,CAAC,QAAQ,CAAC;QAC/B,IAAI,CAAC,OAAO,GAAG,IAAI,CAAC,MAAM,CAAC;QAC3B,IAAI,CAAC,oBAAoB,GAAG,IAAI,CAAC,mBAAmB,IAAI,EAAE,CAAC;QAC3D,IAAI,CAAC,QAAQ,GAAG,IAAI,CAAC,OAAO,CAAC;QAC7B,IAAI,CAAC,SAAS,GAAG,CAAC,IAAI,CAAC,QAAQ,IAAI,EAAE,CAAC,CAAC,MAAM,CAAC;YAC5C,KAAK,EAAE,cAAc;YACrB,IAAI,EAAE,OAAO;YACb,IAAI,EAAE,CAAC,CAAC,EAAE,CAAC,CAAC;SACb,CAAC,CAAC;QACH,IAAI,CAAC,SAAS,GAAG,IAAI,CAAC,QAAQ,IAAI,EAAE,CAAC;QACrC,IAAI,CAAC,YAAY,GAAG,IAAI,CAAC,SAAS,CAAC,MAAM,CAAC,CAAC,GAAG,EAAE,OAAO,EAAE,EAAE;YACzD,GAAG,CAAC,OAAO,CAAC,KAAK,CAAC,GAAG,OAAO,CAAC;YAC7B,OAAO,GAAG,CAAC;QACb,CAAC,EAAE,EAA6B,CAAC,CAAC;QAClC,IAAI,CAAC,kBAAkB,GAAG,IAAI,CAAC,iBAAiB,IAAI,KAAK,CAAC;QAE1D,IAAI,CAAC,SAAS,CAAC,OAAO,CAAC,CAAC,OAAO,EAAE,EAAE;YACjC,MAAM,IAAI,GAAG,gBAAgB,CAAC,OAAO,CAAC,IAAI,EAAE,OAAO,CAAC,IAAI,CAAC,CAAC;YAC1D,IAAI,CAAC,eAAe,CAAC,OAAO,CAAC,KAAK,CAAC,GAAG,MAAM,CAAC,YAAY,CAAC;gBACxD,KAAK,EAAE,IAAI,CAAC,MAAM;gBAClB,IAAI,EAAE,IAAI,CAAC,UAAU;gBACrB,KAAK,EAAE,cAAc,CAAC,OAAO,GAAG,cAAc,CAAC,QAAQ;aACxD,CAAC,CAAC;YACH,IAAI,CAAC,OAAO,CAAC,KAAK,CAAC,WAAW,CAC5B,IAAI,CAAC,eAAe,CAAC,OAAO,CAAC,KAAK,CAAC,EACnC,CAAC,EACD,IAAI,CACL,CAAC;QACJ,CAAC,CAAC,CAAC;IACL,CAAC;IAED,8FAA8F;IAC9F,eAAe;QACb,IAAI,IAAI,CAAC,gBAAgB;YAAE,OAAO;QAClC,IAAI,CAAC,gBAAgB,GAAG,IAAI,CAAC;QAC7B,IAAI,CAAC,cAAc,GAAG,IAAI,CAAC,MAAM,CAAC;QAAC,IAAI,CAAC,eAAe,GAAG,IAAI,CAAC,OAAO,CAAC;QACvE,MAAM,QAAQ,GAAG,IAAI,CAAC,OAAO,CAAC,MAAM,CAAC,IAAI,CAAC,EAAE,CAAC,CAAC,IAAI,CAAC,oBAAoB,CAAC,IAAS,CAAC,IAAI,IAAI,CAAC,oBAAoB,CAAC,IAAS,CAAC,KAAK,IAAI,CAAC,CAAC;QACrI,IAAI,CAAC,YAAY,GAAG,QAAQ,CAAC;QAC7B,IAAI,QAAQ,CAAC,MAAM,EAAE,CAAC;YACpB,MAAM,OAAO,GAAY,EAAC,KAAK,EAAE,cAAc,EAAE,IAAI,EAAE,gBAAgB,QAAQ,CAAC,MAAM,GAAG,EAAE,IAAI,EAAE,IAAI,WAAW,CAAC,QAAQ,CAAC,MAAM,GAAG,CAAC,CAAC,EAAC,CAAC;YACvI,IAAI,CAAC,SAAS,CAAC,IAAI,CAAC,OAAO,CAAC,CAAC;YAAC,IAAI,CAAC,YAAY,CAAC,OAAO,CAAC,KAAK,CAAC,GAAG,OAAO,CAAC;YACzE,IAAI,CAAC,eAAe,CAAC,OAAO,CAAC,KAAK,CAAC,GAAG,IAAI,CAAC,OAAO,CAAC,YAAY,CAAC,EAAC,KAAK,EAAE,GAAG,IAAI,CAAC,MAAM,IAAI,OAAO,CAAC,KAAK,EAAE,EAAE,IAAI,EAAE,QAAQ,CAAC,MAAM,GAAG,EAAE,EAAE,KAAK,EAAE,cAAc,CAAC,OAAO,GAAG,cAAc,CAAC,QAAQ,EAAC,CAAC,CAAC;QACnM,CAAC;QACD,IAAI,CAAC,oBAAoB,GAAG,IAAI,CAAC;QAAC,IAAI,CAAC,OAAO,GAAG,EAAE,CAAC;IACtD,CAAC;IACD,OAAO,CAAC,KAAa,EAAE,MAAc;QACnC,KAAK,GAAG,IAAI,CAAC,GAAG,CAAC,CAAC,EAAE,IAAI,CAAC,IAAI,CAAC,KAAK,CAAC,CAAC,CAAC;QACtC,MAAM,GAAG,IAAI,CAAC,GAAG,CAAC,CAAC,EAAE,IAAI,CAAC,IAAI,CAAC,MAAM,CAAC,CAAC,CAAC;QACxC,MAAM,WAAW,GAAG,KAAK,KAAK,IAAI,CAAC,MAAM,IAAI,MAAM,KAAK,IAAI,CAAC,OAAO,CAAC;QACrE,IAAI,CAAC,MAAM,GAAG,KAAK,CAAC;QACpB,IAAI,CAAC,OAAO,GAAG,MAAM,CAAC;QACtB,IAAI,WAAW,EAAE,CAAC;YAChB,MAAM,SAAS,GAAG,IAAI,CAAC,GAAG,CAAC,IAAI,CAAC,cAAc,EAAE,KAAK,CAAC,CAAC;YACvD,MAAM,UAAU,GAAG,IAAI,CAAC,GAAG,CAAC,IAAI,CAAC,eAAe,EAAE,MAAM,CAAC,CAAC;YAC1D,IAAI,CAAC,0BAA0B,KAAK,CAAC,IAAI,CAAC,gBAAgB,IAAI,SAAS,KAAK,IAAI,CAAC,cAAc,IAAI,UAAU,KAAK,IAAI,CAAC,eAAe,CAAC;YACvI,IAAI,CAAC,cAAc,GAAG,SAAS,CAAC;YAAC,IAAI,CAAC,eAAe,GAAG,UAAU,CAAC;YACnE,IAAI,CAAC,oBAAoB,GAAG,IAAI,CAAC;YAEjC,IAAI,CAAC,UAAU,CAAC,cAAc,EAAE,CAAC,KAAK,EAAE,MAAM,CAAC,CAAC,CAAC;QACnD,CAAC;IACH,CAAC;IAED,eAAe,CAAC,YAA+C;QAC7D,IAAI,CAAC,eAAe,GAAG,SAAS,CAAC;QACjC,MAAM,cAAc,GAAG,MAAM,CAAC,IAAI,CAAC,YAAY,CAAC,CAAC,IAAI,CACnD,CAAC,GAAG,EAAE,EAAE,CACN,CAAC,oBAAoB,CAClB,YAAoB,CAAC,GAAG,CAAC,EAC1B,IAAI,CAAC,aAAa,CAAC,GAAG,CAAC,IAAK,EAA6B,CAC1D,CACJ,CAAC;QACF,IAAI,CAAC,0BAA0B,KAAK,cAAc,CAAC;QACnD,IAAI,CAAC,oBAAoB,KAAK,cAAc,CAAC;QAC7C,IAAI,cAAc;YAAE,IAAI,CAAC,OAAO,GAAG,EAAE,CAAC;QACtC,IAAI,CAAC,aAAa,GAAG,YAAY,CAAC;IACpC,CAAC;IAED,iBAAiB,CACf,aAAyB,EACzB,kBAAoC;QAEpC,IACE,IAAI,CAAC,eAAe;YACpB,IAAI,CAAC,eAAe,CAAC,kBAAkB,KAAK,kBAAkB,EAC9D,CAAC;YACD,IAAI,CAAC,oBAAoB,GAAG,IAAI,CAAC;YACjC,IAAI,CAAC,OAAO,GAAG,EAAE,CAAC;QACpB,CAAC;QACD,IAAI,CAAC,eAAe,GAAG;YACrB,aAAa;YACb,kBAAkB;SACnB,CAAC;IACJ,CAAC;IAED,UAAU,CAAC,KAAa,EAAE,IAAqB;QAC7C,MAAM,MAAM,GAAG,IAAI,CAAC,eAAe,CAAC,KAAK,CAAC,CAAC;QAC3C,IAAI,CAAC,OAAO,CAAC,KAAK,CAAC,WAAW,CAC5B,MAAM,EACN,CAAC,EACD,gBAAgB,CAAC,IAAI,EAAE,IAAI,CAAC,YAAY,CAAC,KAAK,CAAC,CAAC,IAAI,CAAC,CACtD,CAAC;IACJ,CAAC;IAEgB,gBAAgB,GAAG,IAAI,GAAG,EAAc,CAAC;IAC1D,YAAY,CAAC,OAAmB,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;IACpI,aAAa,CAAC,OAAmB;QAC/B,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,EAAE,CAAC;YAC3D,KAAK,IAAI,CAAC,OAAO,CAAC,KAAK,CAAC,mBAAmB,EAAE,CAAC,KAAK,CAAC,GAAG,EAAE,CAAC,SAAS,CAAC,CAAC,IAAI,CAAC,GAAG,EAAE,CAAC,OAAO,CAAC,OAAO,EAAE,CAAC,CAAC;QACrG,CAAC;IACH,CAAC;IAED,SAAS,CAAC,IAAO;QACf,OAAO,IAAI,CAAC,eAAe,CAAC,IAAI,CAAC,CAAC;IACpC,CAAC;IAED,cAAc;QACZ,OAAO,MAAM,CAAC,MAAM,CAAC,IAAI,CAAC,eAAe,CAAC,CAAC,MAAM,CAAC,CAAC,GAAG,EAAE,MAAM,EAAE,EAAE,CAAC,GAAG,GAAG,MAAM,CAAC,IAAI,EAAE,CAAC,CAAC;cACpF,KAAK,CAAC,IAAI,CAAC,IAAI,GAAG,CAAC,CAAC,GAAG,MAAM,CAAC,MAAM,CAAC,IAAI,CAAC,eAAe,CAAC,EAAE,GAAG,IAAI,CAAC,gBAAgB,CAAC,CAAC,CAAC;iBACtF,MAAM,CAAC,CAAC,GAAG,EAAE,OAAO,EAAE,EAAE,CAAC,GAAG,GAAG,YAAY,CAAC,OAAO,CAAC,EAAE,CAAC,CAAC,CAAC;IAChE,CAAC;IAED,OAAO;QACL,MAAM,CAAC,IAAI,CAAC,IAAI,CAAC,eAAe,CAAC,CAAC,OAAO,CAAC,CAAC,GAAG,EAAE,EAAE;YAC/C,IAAI,CAAC,eAAuB,CAAC,GAAG,CAAC,CAAC,OAAO,EAAE,CAAC;QAC/C,CAAC,CAAC,CAAC;QACH,IAAI,GAAG,CAAC,CAAC,GAAG,MAAM,CAAC,MAAM,CAAC,IAAI,CAAC,eAAe,CAAC,EAAE,GAAG,IAAI,CAAC,gBAAgB,CAAC,CAAC,CAAC,OAAO,CAAC,OAAO,CAAC,EAAE,CAAC,OAAO,CAAC,OAAO,EAAE,CAAC,CAAC;QAClH,IAAI,CAAC,gBAAgB,CAAC,KAAK,EAAE,CAAC;IAChC,CAAC;IAEO,cAAc,CAAC,MAA8B;QACnD,OAAO,IAAI,CAAC,OAAO,CAAC,aAAa,CAAC;YAChC,KAAK,EAAE,IAAI,CAAC,MAAM;YAClB,IAAI,EAAE;gBACJ,KAAK,EAAE,IAAI,CAAC,gBAAgB,CAAC,CAAC,CAAC,IAAI,CAAC,cAAc,CAAC,CAAC,CAAC,IAAI,CAAC,MAAM;gBAChE,MAAM,EAAE,IAAI,CAAC,gBAAgB,CAAC,CAAC,CAAC,IAAI,CAAC,eAAe,CAAC,CAAC,CAAC,IAAI,CAAC,OAAO;gBACnE,kBAAkB,EAAE,CAAC;aACtB;YACD,MAAM,EAAE,MAAM,CAAC,MAAM;YACrB,KAAK,EACH,eAAe,CAAC,iBAAiB;gBACjC,eAAe,CAAC,eAAe;gBAC/B,CAAC,IAAI,CAAC,kBAAkB,CAAC,CAAC,CAAC,eAAe,CAAC,QAAQ,CAAC,CAAC,CAAC,CAAC,CAAC;SAC3D,CAAC,CAAC;IACL,CAAC;IAEO,qBAAqB,CAAC,QAAmB;QAC/C,MAAM,YAAY,GAAG,IAAI,CAAC,aAAa,CAAC;QACxC,KAAK,MAAM,GAAG,IAAI,YAAY,EAAE,CAAC;YAC/B,IAAI,QAAQ,EAAE,QAAQ,CAAC,GAAG,CAAC,EAAE,CAAC;gBAC5B,SAAS;YACX,CAAC;YACD,MAAM,MAAM,GAAG,YAAY,CAAC,GAAG,CAAC,CAAC;YACjC,IAAI,CAAC,IAAI,CAAC,gBAAgB,CAAC,GAAG,CAAC,IAAI,CAAC,eAAe,CAAC,GAAG,CAAC,CAAC;gBAAE,IAAI,CAAC,eAAe,CAAC,GAAG,CAAC,EAAE,OAAO,EAAE,CAAC;YAChG,IAAI,CAAC,eAAe,CAAC,GAAG,CAAC,GAAG,IAAI,CAAC,cAAc,CAAC,MAAM,CAAC,CAAC;QAC1D,CAAC;IACH,CAAC;IAEO,eAAe;QACrB,IAAI,CAAC,IAAI,CAAC,oBAAoB,EAAE,CAAC;YAC/B,OAAO;QACT,CAAC;QACD,IAAI,CAAC,oBAAoB,GAAG,KAAK,CAAC;QAElC,MAAM,YAAY,GAAG,IAAI,CAAC,aAAa,CAAC;QACxC,MAAM,MAAM,GAAG,IAAI,CAAC,OAAO,CAAC;QAC5B,MAAM,MAAM,GAAG,IAAI,CAAC,UAAU,EAAE,CAAC;QACjC,IAAI,MAAM,KAAK,IAAI,CAAC,OAAO,EAAE,CAAC;YAC5B,OAAO;QACT,CAAC;QACD,MAAM,EAAE,aAAa,EAAE,kBAAkB,EAAE,GAAG,IAAI,CAAC,eAAe,IAAI,EAAE,CAAC;QACzE,IAAI,CAAC,OAAO,GAAG,MAAM,CAAC;QACtB,IAAI,CAAC,SAAS,GAAG,oBAAoB,CAAC,MAAM,CAAC,CAAC,oBAAoB,CAAC;YACjE,KAAK,EAAE,IAAI,CAAC,MAAM;YAClB,MAAM,EAAE,MAAM;YACd,MAAM,EAAE;gBACN,MAAM,EAAE,oBAAoB,CAAC,MAAM,CAAC,CAAC,kBAAkB,CAAC;oBACtD,KAAK,EAAE,IAAI,CAAC,MAAM;oBAClB,IAAI,EAAE,8BAA8B;iBACrC,CAAC;gBACF,UAAU,EAAE,MAAM;aACnB;YACD,QAAQ,EAAE;gBACR,MAAM,EAAE,oBAAoB,CAAC,MAAM,CAAC,CAAC,kBAAkB,CAAC;oBACtD,KAAK,EAAE,IAAI,CAAC,MAAM;oBAClB,IAAI,EAAE,MAAM;iBACb,CAAC;gBACF,UAAU,EAAE,MAAM;gBAClB,OAAO,EAAE,aAAa;oBACpB,CAAC,CAAC;wBACE;4BACE,MAAM,EAAE,kBAAmB;yBAC5B;qBACF;oBACH,CAAC,CAAC,IAAI,CAAC,QAAQ,CAAC,GAAG,CAAC,CAAC,GAAG,EAAE,EAAE,CAAC,CAAC;wBAC1B,MAAM,EAAG,YAAoB,CAAC,GAAG,CAAC,CAAC,MAAM;qBAC1C,CAAC,CAAC;aACR;YACD,SAAS,EAAE;gBACT,QAAQ,EAAE,eAAe;aAC1B;SACF,CAAC,CAAC;QACH,IAAI,CAAC,iBAAiB,EAAE,CAAC;IAC3B,CAAC;IAEO,UAAU;QAChB,MAAM,MAAM,GAAG,IAAI,CAAC,OAAO,CAAC;QAC5B,MAAM,SAAS,GAAG,MAAM,CAAC,MAAM,CAAC;QAChC,MAAM,SAAS,GAAG,SAAS,GAAG,CAAC,CAAC;QAChC,MAAM,EAAE,GAAG,UAAU,CAAC;EACxB,MAAM;aACL,IAAI,EAAE;aACN,GAAG,CACF,CAAC,WAAW,EAAE,GAAG,EAAE,EAAE;YACnB,MAAM,aAAa,GAAG,IAAI,CAAC,oBAAoB,CAAC,WAAgB,CAAC,CAAC;YAClE,MAAM,WAAW,GAAG,aAAa,KAAK,IAAI,CAAC,CAAC,CAAC,YAAY,CAAC,CAAC,CAAC,YAAY,CAAC;YACzE,OAAO,sBAAsB,GAAG,SAAS,WAAW,KAAK,WAAW,QAAQ,CAAC;QAC/E,CAAC,CACF;aACA,IAAI,CAAC,IAAI,CAAC;;EAGX,SAAS;YACP,CAAC,CAAC,MAAM,CAAC,IAAI,CAAC,IAAI,CAAC,SAAS,CAAC;iBACxB,IAAI,EAAE;iBACN,GAAG,CACF,CAAC,WAAW,EAAE,GAAG,EAAE,EAAE,CACnB,sBACE,GAAG,GAAG,SACR,SAAS,WAAW,YAAY,CACnC;iBACA,IAAI,CAAC,IAAI,CAAC;YACf,CAAC,CAAC,EACN;EACE,IAAI,CAAC,SAAS;aACb,GAAG,CACF,CAAC,OAAO,EAAE,GAAG,EAAE,EAAE,CACf,UAAU,SAAS,CAAC,CAAC,CAAC,CAAC,CAAC,CAAC,CAAC,CAAC,cAAc,GAAG,kBAC1C,OAAO,CAAC,KACV,KAAK,OAAO,CAAC,IAAI,GAAG,CACvB;aACA,IAAI,CAAC,IAAI,CAAC;;EAEX,IAAI,CAAC,QAAQ;aACZ,GAAG,CAAC,CAAC,IAAI,EAAE,GAAG,EAAE,EAAE,CAAC,aAAa,GAAG,KAAK,IAAI,UAAU,CAAC;aACvD,IAAI,CAAC,IAAI,CAAC;;EAEX,oBAAoB,CACpB,IAAI,CAAC,gBAAgB,CAAC,CAAC,CAAC,IAAI,CAAC,YAAY,CAAC,CAAC,CAAC,MAAM,CAAC,MAAM,CAAC,IAAI,CAAC,EAAE,CAC/D,CAAC,IAAI,CAAC,oBAAoB,CAAC,IAAS,CAAC,IAAI,IAAI,CAAC,oBAAoB,CAAC,IAAS,CAAC,KAAK,IAAI,CAAC,EACzF,MAAM,CAAC,IAAI,CAAC,IAAI,CAAC,SAAS,CAAC,EAAE,IAAI,CAAC,gBAAgB,CACnD;EACC,IAAI,CAAC,SAAS,IAAI,EAAE;;;;;;;;EAQpB,IAAI,CAAC,OAAO;;;CAGb,CAAC;QAEE,OAAO,EAAE,CAAC;IACZ,CAAC;IAEO,iBAAiB;QACvB,MAAM,UAAU,GAAmB,EAAE,CAAC;QACtC,MAAM,MAAM,GAAG,IAAI,CAAC,OAAO,CAAC;QAC5B,MAAM,qBAAqB,GAAG,IAAI,CAAC,OAAO,CAAC,MAAM,GAAG,CAAC,CAAC,CAAC,CAAC,CAAC,CAAC,CAAC,CAAC,CAAC,CAAC;QAC9D,IAAI,IAAI,CAAC,SAAS,CAAC,MAAM,GAAG,CAAC,EAAE,CAAC;YAC9B,UAAU,CAAC,qBAAqB,CAAC,GAAG,MAAM,CAAC,eAAe,CAAC;gBACzD,KAAK,EAAE,IAAI,CAAC,MAAM;gBAClB,MAAM,EAAE,IAAI,CAAC,SAAS,CAAC,kBAAkB,CAAC,qBAAqB,CAAC;gBAChE,OAAO,EAAE,IAAI,CAAC,SAAS,CAAC,GAAG,CACzB,CAAC,OAAO,EAAE,GAAG,EAAE,EAAE,CACf,CAAC;oBACC,OAAO,EAAE,GAAG;oBACZ,QAAQ,EAAE;wBACR,MAAM,EAAE,IAAI,CAAC,eAAe,CAAC,OAAO,CAAC,KAAK,CAAC;qBAC5C;iBAC2B,CAAA,CACjC;aACF,CAAC,CAAC;QACL,CAAC;QAED,IAAI,CAAC,WAAW,GAAG,UAAU,CAAC;IAChC,CAAC;IAED,UAAU,CACR,cAAiC,EACjC,MAA6B;IAE7B;;OAEG;IACH,OAA+B;QAE/B,IAAI,IAAI,CAAC,0BAA0B,IAAI,CAAC,IAAI,CAAC,eAAe,EAAE,CAAC;YAC7D,IAAI,CAAC,qBAAqB,CAAC,MAAM,CAAC,IAAI,CAAC,OAAO,IAAI,EAAE,CAAC,CAAC,CAAC;YACvD,IAAI,CAAC,0BAA0B,GAAG,KAAK,CAAC;QAC1C,CAAC;QAED,IAAI,IAAI,CAAC,YAAY,CAAC,MAAM,EAAE,CAAC;YAC7B,MAAM,MAAM,GAAG,IAAI,WAAW,CAAC,IAAI,CAAC,YAAY,CAAC,MAAM,GAAG,CAAC,CAAC,CAAC;YAC7D,KAAK,MAAM,CAAC,KAAK,EAAE,IAAI,CAAC,IAAI,IAAI,CAAC,YAAY,CAAC,OAAO,EAAE,EAAE,CAAC;gBACxD,MAAM,MAAM,GAAG,aAAa,CAAC,MAAM,CAAC,IAAS,CAAC,CAAC,CAAC;gBAChD,MAAM,CAAC,GAAG,CAAC,CAAC,MAAM,CAAC,KAAK,EAAE,MAAM,CAAC,MAAM,CAAC,EAAE,KAAK,GAAG,CAAC,CAAC,CAAC;YACvD,CAAC;YACD,IAAI,CAAC,UAAU,CAAC,cAAc,EAAE,MAAM,CAAC,CAAC;QAC1C,CAAC;QACD,KAAK,MAAM,OAAO,IAAI,MAAM,CAAC,MAAM,CAAC,IAAI,CAAC,eAAe,CAAC;YAAE,gBAAgB,CAAC,OAAO,EAAE,IAAI,CAAC,MAAM,EAAE,IAAI,CAAC,OAAO,CAAC,CAAC;QAChH,IAAI,CAAC,eAAe,EAAE,CAAC;QAEvB,MAAM,SAAS,GAAG,IAAI,CAAC,OAAO,CAAC,MAAM,CAAC;QACtC,IAAI,SAAS,GAAG,CAAC,EAAE,CAAC;YAClB,IAAI,CAAC,WAAW,CAAC,CAAC,CAAC,GAAG,IAAI,CAAC,OAAO,CAAC,eAAe,CAAC;gBACjD,KAAK,EAAE,IAAI,CAAC,MAAM;gBAClB,MAAM,EAAE,IAAI,CAAC,SAAS,CAAC,kBAAkB,CAAC,CAAC,CAAC;gBAC5C,OAAO,EAAE;oBACP,GAAG,IAAI,CAAC,OAAO,CAAC,GAAG,CAAC,CAAC,WAAW,EAAE,GAAG,EAAE,EAAE,CAAC,CAAC;wBACzC,OAAO,EAAE,GAAG;wBACZ,QAAQ,EAAE,MAAM,CAAC,WAAgB,CAAC,CAAC,UAAU,CAC3C,IAAI,CAAC,oBAAoB,CAAC,WAAgB,CAAC;4BACzC,CAAC,CAAC,EAAE,SAAS,EAAE,IAAI,CAAC,oBAAoB,CAAC,WAAgB,CAAC,EAAE;4BAC5D,CAAC,CAAC,SAAS,CACd;qBACF,CAAC,CAAC;oBACH,GAAG,MAAM,CAAC,IAAI,CAAC,IAAI,CAAC,SAAS,CAAC,CAAC,GAAG,CAAC,CAAC,WAAW,EAAE,GAAG,EAAE,EAAE,CAAC,CAAC;wBACxD,OAAO,EAAE,GAAG,GAAG,SAAS;wBACxB,QAAQ,EAAE,IAAI,CAAC,SAAS,CAAC,WAAW,CAAC;qBACtC,CAAC,CAAC;iBACJ;aACF,CAAC,CAAC;QACL,CAAC;QAED,MAAM,cAAc,GAAG;YACrB,8CAA8C;YAC9C,GAAG,IAAI,CAAC,eAAe;YACvB,GAAG,OAAO;SACmB,CAAC;QAChC,wBAAwB;QACxB,MAAM,UAAU,GAAG,cAAc,CAAC,eAAe,CAAC;YAChD,gBAAgB,EAAE,IAAI,CAAC,eAAe;gBACpC,CAAC,CAAC;oBACE;wBACE,IAAI,EAAE,IAAI,CAAC,eAAe,CAAC,aAAa,CAAC,UAAU,EAAE;wBACrD,UAAU,EAAE,EAAE,CAAC,EAAE,CAAC,EAAE,CAAC,EAAE,CAAC,EAAE,CAAC,EAAE,CAAC,EAAE,CAAC,EAAE,CAAC,EAAE;wBACtC,OAAO,EAAE,OAAO;wBAChB,MAAM,EAAE,OAAO;qBAChB;iBACF;gBACH,CAAC,CAAC,IAAI,CAAC,QAAQ,CAAC,GAAG,CACf,CAAC,WAAW,EAAE,EAAE,CACd,CAAC;oBACC,IAAI,EAAE,cAAc,CAAC,WAAW,CAAC,CAAC,UAAU,EAAE;oBAC9C,UAAU,EAAE,EAAE,CAAC,EAAE,CAAC,EAAE,CAAC,EAAE,CAAC,EAAE,CAAC,EAAE,CAAC,EAAE,CAAC,EAAE,CAAC,EAAE;oBACtC,MAAM,EAAE,OAAO;oBACf,OAAO,EAAE,OAAO;iBACuB,CAAA,CAC5C;SACN,CAAC,CAAC;QAEH,mBAAmB;QACnB,UAAU,CAAC,WAAW,CAAC,CAAC,EAAE,CAAC,EAAE,IAAI,CAAC,MAAM,EAAE,IAAI,CAAC,OAAO,EAAE,CAAC,EAAE,CAAC,CAAC,CAAC;QAC9D,UAAU,CAAC,cAAc,CAAC,CAAC,EAAE,CAAC,EAAE,IAAI,CAAC,MAAM,EAAE,IAAI,CAAC,OAAO,CAAC,CAAC;QAC3D,UAAU,CAAC,WAAW,CAAC,IAAI,CAAC,SAAS,CAAC,CAAC;QACvC,cAAc;QACd,IAAI,CAAC,WAAW,CAAC,OAAO,CAAC,CAAC,SAAS,EAAE,GAAG,EAAE,EAAE;YAC1C,UAAU,CAAC,YAAY,CAAC,GAAG,EAAE,SAAS,CAAC,CAAC;QAC1C,CAAC,CAAC,CAAC;QACH,UAAU,CAAC,IAAI,CAAC,CAAC,EAAE,CAAC,EAAE,CAAC,EAAE,CAAC,CAAC,CAAC;QAC5B,sBAAsB;QACtB,UAAU,CAAC,GAAG,EAAE,CAAC;IACnB,CAAC;CACF","sourcesContent":["import {scratchWGSLFunctions} from './scratchSampling.js';\nimport {setTextureRegion, textureRegion} from './textureRegion.js';\nimport { getTargetShaderCache } from './WGPUShaderCache.js';\nimport { textureBytes } from './gpuMemory.js';\nexport const fullScreenQuadVertexShaderWGSL = /*wgsl */ `\n@vertex\nfn main(\n  @builtin(vertex_index) vertexIndex: u32\n) -> @builtin(position) vec4f {\n  const pos = array(\n    vec2(-1.0, -1.0), vec2(1.0, -1.0), vec2(-1.0, 1.0),\n    vec2(-1.0, 1.0), vec2(1.0, -1.0), vec2(1.0, 1.0),\n  );\n\n  return vec4<f32>(pos[vertexIndex], 0.0, 1.0);\n}\n`;\n\nfunction isTextureParamsEqual(\n  params: WGPUFullQuadPassOutput,\n  other: WGPUFullQuadPassOutput\n) {\n  return params.format === other.format;\n}\n\nexport function getUniformBuffer(\n  value: any,\n  type: Uniform['type']\n): Float32Array | Int32Array | Uint32Array {\n  if (\n    value instanceof Float32Array ||\n    value instanceof Int32Array ||\n    value instanceof Uint32Array\n  ) {\n    return value;\n  }\n\n  if (typeof value === 'number') {\n    value = [value];\n  }\n\n  if (type === 'i32' || type.endsWith('i')) {\n    return new Int32Array(value);\n  } else if (type === 'u32' || type.endsWith('u')) {\n    return new Uint32Array(value);\n  } else {\n    return new Float32Array(value);\n  }\n}\n\nexport interface Uniform {\n  label: string;\n  type:\n    | `array<vec4u, ${number}>`\n    | 'f32'\n    | 'i32'\n    | 'u32'\n    | 'vec3f'\n    | 'vec4f'\n    | 'vec2f'\n    | 'vec2i'\n    | 'vec3i'\n    | 'vec4i'\n    | 'vec2u'\n    | 'vec3u'\n    | 'vec4u';\n  data: Float32Array | Int32Array | Uint32Array | number | number[];\n}\n\nexport interface WGPUFullQuadPassOutput {\n  format: GPUTextureFormat;\n}\nexport class WGPUFullQuadPass<I extends string, O extends string> {\n  private _label;\n\n  private _device;\n  private _outputTextures: Record<string, GPUTexture> = {};\n  private _outputParams: Record<string, WGPUFullQuadPassOutput> = {};\n\n  private _pipeline!: GPURenderPipeline;\n  private _bindGroups: GPUBindGroup[] = [];\n  private _needsUpdatePipeline = true;\n  private _needsUpdateOutputTextures = true;\n  /**\n   * When render to canvas\n   */\n  private _renderToScreen?: {\n    screenTexture: GPUTexture;\n    presentationFormat: GPUTextureFormat;\n  };\n  private _inputs: string[] = [];\n  private _inputViewDimensions: Partial<\n    Record<I, GPUTextureViewDimension>\n  > = {};\n  private _outputs: string[] = [];\n  private _uniforms: Uniform[] = [];\n  private _uniformsMap: Record<string, Uniform> = {};\n  private _uniformBuffers: Record<string, GPUBuffer> = {};\n\n  private _samplers: Record<string, GPUSampler> = {};\n\n  private _capacityWidth = 0;\n  private _capacityHeight = 0;\n  private _reserveCapacity = false;\n  private _regionNames: string[] = [];\n  private _width = 10;\n  private _height = 10;\n\n  private _fsCode = '';\n  private readonly _fsMain;\n  private readonly _fsDefine;\n\n  private _textureWillBeRead;\n\n  constructor(\n    label: string,\n    device: GPUDevice,\n    opts: {\n      inputs: I[];\n      inputViewDimensions?: Partial<Record<I, GPUTextureViewDimension>>;\n      outputs: O[];\n      samplers?: Record<string, GPUSampler>;\n      fsMain: string;\n      fsDefine?: string;\n      uniforms?: Uniform[];\n      textureWillBeRead?: boolean;\n    }\n  ) {\n    this._label = label;\n    this._device = device;\n    this._fsMain = opts.fsMain;\n    this._fsDefine = opts.fsDefine;\n    this._inputs = opts.inputs;\n    this._inputViewDimensions = opts.inputViewDimensions || {};\n    this._outputs = opts.outputs;\n    this._uniforms = (opts.uniforms || []).concat({\n      label: 'viewportSize',\n      type: 'vec2f',\n      data: [0, 0]\n    });\n    this._samplers = opts.samplers || {};\n    this._uniformsMap = this._uniforms.reduce((map, uniform) => {\n      map[uniform.label] = uniform;\n      return map;\n    }, {} as Record<string, Uniform>);\n    this._textureWillBeRead = opts.textureWillBeRead || false;\n\n    this._uniforms.forEach((uniform) => {\n      const data = getUniformBuffer(uniform.data, uniform.type);\n      this._uniformBuffers[uniform.label] = device.createBuffer({\n        label: this._label,\n        size: data.byteLength,\n        usage: GPUBufferUsage.UNIFORM | GPUBufferUsage.COPY_DST\n      });\n      this._device.queue.writeBuffer(\n        this._uniformBuffers[uniform.label],\n        0,\n        data\n      );\n    });\n  }\n\n  /** Keep physical outputs large enough for every render while varying the logical viewport. */\n  reserveCapacity() {\n    if (this._reserveCapacity) return;\n    this._reserveCapacity = true;\n    this._capacityWidth = this._width; this._capacityHeight = this._height;\n    const textures = this._inputs.filter(name => !this._inputViewDimensions[name as I] || this._inputViewDimensions[name as I] === '2d');\n    this._regionNames = textures;\n    if (textures.length) {\n      const uniform: Uniform = {label: 'imageRegions', type: `array<vec4u, ${textures.length}>`, data: new Uint32Array(textures.length * 4)};\n      this._uniforms.push(uniform); this._uniformsMap[uniform.label] = uniform;\n      this._uniformBuffers[uniform.label] = this._device.createBuffer({label: `${this._label}.${uniform.label}`, size: textures.length * 16, usage: GPUBufferUsage.UNIFORM | GPUBufferUsage.COPY_DST});\n    }\n    this._needsUpdatePipeline = true; this._fsCode = '';\n  }\n  setSize(width: number, height: number) {\n    width = Math.max(1, Math.ceil(width));\n    height = Math.max(1, Math.ceil(height));\n    const sizeChanged = width !== this._width || height !== this._height;\n    this._width = width;\n    this._height = height;\n    if (sizeChanged) {\n      const nextWidth = Math.max(this._capacityWidth, width);\n      const nextHeight = Math.max(this._capacityHeight, height);\n      this._needsUpdateOutputTextures ||= !this._reserveCapacity || nextWidth !== this._capacityWidth || nextHeight !== this._capacityHeight;\n      this._capacityWidth = nextWidth; this._capacityHeight = nextHeight;\n      this._needsUpdatePipeline = true;\n\n      this.setUniform('viewportSize', [width, height]);\n    }\n  }\n\n  setOutputParams(outputParams: Record<O, WGPUFullQuadPassOutput>) {\n    this._renderToScreen = undefined;\n    const outputsChanged = Object.keys(outputParams).some(\n      (key) =>\n        !isTextureParamsEqual(\n          (outputParams as any)[key],\n          this._outputParams[key] || ({} as WGPUFullQuadPassOutput)\n        )\n    );\n    this._needsUpdateOutputTextures ||= outputsChanged;\n    this._needsUpdatePipeline ||= outputsChanged;\n    if (outputsChanged) this._fsCode = '';\n    this._outputParams = outputParams;\n  }\n\n  setRenderToScreen(\n    screenTexture: GPUTexture,\n    presentationFormat: GPUTextureFormat\n  ) {\n    if (\n      this._renderToScreen &&\n      this._renderToScreen.presentationFormat !== presentationFormat\n    ) {\n      this._needsUpdatePipeline = true;\n      this._fsCode = '';\n    }\n    this._renderToScreen = {\n      screenTexture,\n      presentationFormat\n    };\n  }\n\n  setUniform(label: string, data: Uniform['data']) {\n    const buffer = this._uniformBuffers[label];\n    this._device.queue.writeBuffer(\n      buffer,\n      0,\n      getUniformBuffer(data, this._uniformsMap[label].type)\n    );\n  }\n\n  private readonly _retainedOutputs = new Set<GPUTexture>();\n  retainOutput(texture: GPUTexture) { if (Object.values(this._outputTextures).includes(texture)) this._retainedOutputs.add(texture); }\n  releaseOutput(texture: GPUTexture) {\n    if (!this._retainedOutputs.delete(texture)) return;\n    if (!Object.values(this._outputTextures).includes(texture)) {\n      void this._device.queue.onSubmittedWorkDone().catch(() => undefined).then(() => texture.destroy());\n    }\n  }\n\n  getOutput(name: O) {\n    return this._outputTextures[name];\n  }\n\n  getMemoryBytes() {\n    return Object.values(this._uniformBuffers).reduce((sum, buffer) => sum + buffer.size, 0)\n      + Array.from(new Set([...Object.values(this._outputTextures), ...this._retainedOutputs]))\n        .reduce((sum, texture) => sum + textureBytes(texture), 0);\n  }\n\n  dispose() {\n    Object.keys(this._uniformBuffers).forEach((key) => {\n      (this._uniformBuffers as any)[key].destroy();\n    });\n    new Set([...Object.values(this._outputTextures), ...this._retainedOutputs]).forEach(texture => texture.destroy());\n    this._retainedOutputs.clear();\n  }\n\n  private _createTexture(params: WGPUFullQuadPassOutput) {\n    return this._device.createTexture({\n      label: this._label,\n      size: {\n        width: this._reserveCapacity ? this._capacityWidth : this._width,\n        height: this._reserveCapacity ? this._capacityHeight : this._height,\n        depthOrArrayLayers: 1\n      },\n      format: params.format,\n      usage:\n        GPUTextureUsage.RENDER_ATTACHMENT |\n        GPUTextureUsage.TEXTURE_BINDING |\n        (this._textureWillBeRead ? GPUTextureUsage.COPY_SRC : 0)\n    });\n  }\n\n  private _updateOutputTextures(excludes?: string[]) {\n    const outputParams = this._outputParams;\n    for (const key in outputParams) {\n      if (excludes?.includes(key)) {\n        continue;\n      }\n      const params = outputParams[key];\n      if (!this._retainedOutputs.has(this._outputTextures[key])) this._outputTextures[key]?.destroy();\n      this._outputTextures[key] = this._createTexture(params);\n    }\n  }\n\n  private _updatePipeline() {\n    if (!this._needsUpdatePipeline) {\n      return;\n    }\n    this._needsUpdatePipeline = false;\n\n    const outputParams = this._outputParams;\n    const device = this._device;\n    const fsCode = this._getFullFs();\n    if (fsCode === this._fsCode) {\n      return;\n    }\n    const { screenTexture, presentationFormat } = this._renderToScreen || {};\n    this._fsCode = fsCode;\n    this._pipeline = getTargetShaderCache(device).createRenderPipeline({\n      label: this._label,\n      layout: 'auto',\n      vertex: {\n        module: getTargetShaderCache(device).createShaderModule({\n          label: this._label,\n          code: fullScreenQuadVertexShaderWGSL\n        }),\n        entryPoint: 'main'\n      },\n      fragment: {\n        module: getTargetShaderCache(device).createShaderModule({\n          label: this._label,\n          code: fsCode\n        }),\n        entryPoint: 'main',\n        targets: screenTexture\n          ? [\n              {\n                format: presentationFormat!\n              }\n            ]\n          : this._outputs.map((key) => ({\n              format: (outputParams as any)[key].format\n            }))\n      },\n      primitive: {\n        topology: 'triangle-list'\n      }\n    });\n    this._updateBindGroups();\n  }\n\n  private _getFullFs() {\n    const inputs = this._inputs;\n    const inputsLen = inputs.length;\n    const hasInputs = inputsLen > 0;\n    const fs = /* wgsl */ `\n${inputs\n  .sort()\n  .map(\n    (textureName, idx) => {\n      const viewDimension = this._inputViewDimensions[textureName as I];\n      const textureType = viewDimension === '3d' ? 'texture_3d' : 'texture_2d';\n      return `@group(0) @binding(${idx}) var ${textureName}: ${textureType}<f32>;`;\n    }\n  )\n  .join('\\n')}\n\n${\n  hasInputs\n    ? Object.keys(this._samplers)\n        .sort()\n        .map(\n          (samplerName, idx) =>\n            `@group(0) @binding(${\n              idx + inputsLen\n            }) var ${samplerName}: sampler;`\n        )\n        .join('\\n')\n    : ''\n}\n${this._uniforms\n  .map(\n    (uniform, idx) =>\n      `@group(${hasInputs ? 1 : 0}) @binding(${idx}) var<uniform> ${\n        uniform.label\n      }: ${uniform.type};`\n  )\n  .join('\\n')}\nstruct FSOutput {\n${this._outputs\n  .map((name, idx) => `@location(${idx}) ${name}: vec4f,`)\n  .join('\\n')}\n}\n${scratchWGSLFunctions(\n  this._reserveCapacity ? this._regionNames : inputs.filter(name =>\n    !this._inputViewDimensions[name as I] || this._inputViewDimensions[name as I] === '2d'),\n  Object.keys(this._samplers), this._reserveCapacity\n)}\n${this._fsDefine ?? ''}\n@fragment\nfn main(\n  @builtin(position) coord: vec4f\n) -> FSOutput {\n  let uv = vec2i(floor(coord.xy));\n  let uv2 = coord.xy / viewportSize;\n  var output: FSOutput;\n${this._fsMain}\n  return output;\n}\n`;\n\n    return fs;\n  }\n\n  private _updateBindGroups() {\n    const bindGroups: GPUBindGroup[] = [];\n    const device = this._device;\n    const uniformBindGroupIndex = this._inputs.length > 0 ? 1 : 0;\n    if (this._uniforms.length > 0) {\n      bindGroups[uniformBindGroupIndex] = device.createBindGroup({\n        label: this._label,\n        layout: this._pipeline.getBindGroupLayout(uniformBindGroupIndex),\n        entries: this._uniforms.map(\n          (uniform, idx) =>\n            ({\n              binding: idx,\n              resource: {\n                buffer: this._uniformBuffers[uniform.label]\n              }\n            } satisfies GPUBindGroupEntry)\n        )\n      });\n    }\n\n    this._bindGroups = bindGroups;\n  }\n\n  createPass(\n    commandEncoder: GPUCommandEncoder,\n    inputs: Record<I, GPUTexture>,\n\n    /**\n     * Explict outputs. Used for buffers reuse.\n     */\n    outputs?: Record<O, GPUTexture>\n  ) {\n    if (this._needsUpdateOutputTextures && !this._renderToScreen) {\n      this._updateOutputTextures(Object.keys(outputs || {}));\n      this._needsUpdateOutputTextures = false;\n    }\n\n    if (this._regionNames.length) {\n      const values = new Uint32Array(this._regionNames.length * 4);\n      for (const [index, name] of this._regionNames.entries()) {\n        const region = textureRegion(inputs[name as I]);\n        values.set([region.width, region.height], index * 4);\n      }\n      this.setUniform('imageRegions', values);\n    }\n    for (const texture of Object.values(this._outputTextures)) setTextureRegion(texture, this._width, this._height);\n    this._updatePipeline();\n\n    const inputsLen = this._inputs.length;\n    if (inputsLen > 0) {\n      this._bindGroups[0] = this._device.createBindGroup({\n        label: this._label,\n        layout: this._pipeline.getBindGroupLayout(0),\n        entries: [\n          ...this._inputs.map((textureName, idx) => ({\n            binding: idx,\n            resource: inputs[textureName as I].createView(\n              this._inputViewDimensions[textureName as I]\n                ? { dimension: this._inputViewDimensions[textureName as I] }\n                : undefined\n            )\n          })),\n          ...Object.keys(this._samplers).map((samplerName, idx) => ({\n            binding: idx + inputsLen,\n            resource: this._samplers[samplerName]\n          }))\n        ]\n      });\n    }\n\n    const outputTextures = {\n      // Pick the texture from outputTextures first.\n      ...this._outputTextures,\n      ...outputs\n    } as Record<string, GPUTexture>;\n    // Begin the render pass\n    const renderPass = commandEncoder.beginRenderPass({\n      colorAttachments: this._renderToScreen\n        ? [\n            {\n              view: this._renderToScreen.screenTexture.createView(),\n              clearValue: { r: 0, g: 0, b: 0, a: 0 },\n              storeOp: 'store',\n              loadOp: 'clear'\n            }\n          ]\n        : this._outputs.map(\n            (textureName) =>\n              ({\n                view: outputTextures[textureName].createView(),\n                clearValue: { r: 0, g: 0, b: 0, a: 0 },\n                loadOp: 'clear',\n                storeOp: 'store'\n              } satisfies GPURenderPassColorAttachment)\n          )\n    });\n\n    // Draw a full quad\n    renderPass.setViewport(0, 0, this._width, this._height, 0, 1);\n    renderPass.setScissorRect(0, 0, this._width, this._height);\n    renderPass.setPipeline(this._pipeline);\n    // Bind groups\n    this._bindGroups.forEach((bindGroup, idx) => {\n      renderPass.setBindGroup(idx, bindGroup);\n    });\n    renderPass.draw(6, 1, 0, 0);\n    // End the render pass\n    renderPass.end();\n  }\n}\n"]}
+{"version":3,"file":"WGPUFullQuadPass.js","sourceRoot":"","sources":["../../src/renderer/WGPUFullQuadPass.ts"],"names":[],"mappings":"AAAA,OAAO,EAAC,oBAAoB,EAAC,MAAM,sBAAsB,CAAC;AAC1D,OAAO,EAAC,gBAAgB,EAAE,aAAa,EAAC,MAAM,oBAAoB,CAAC;AACnE,OAAO,EAAE,oBAAoB,EAAE,MAAM,sBAAsB,CAAC;AAC5D,OAAO,EAAE,YAAY,EAAE,MAAM,gBAAgB,CAAC;AAC9C,MAAM,CAAC,MAAM,8BAA8B,GAAG,SAAS,CAAC;;;;;;;;;;;;CAYvD,CAAC;AAEF,SAAS,oBAAoB,CAC3B,MAA8B,EAC9B,KAA6B;IAE7B,OAAO,MAAM,CAAC,MAAM,KAAK,KAAK,CAAC,MAAM,CAAC;AACxC,CAAC;AAED,MAAM,UAAU,gBAAgB,CAC9B,KAAU,EACV,IAAqB;IAErB,IACE,KAAK,YAAY,YAAY;QAC7B,KAAK,YAAY,UAAU;QAC3B,KAAK,YAAY,WAAW,EAC5B,CAAC;QACD,OAAO,KAAK,CAAC;IACf,CAAC;IAED,IAAI,OAAO,KAAK,KAAK,QAAQ,EAAE,CAAC;QAC9B,KAAK,GAAG,CAAC,KAAK,CAAC,CAAC;IAClB,CAAC;IAED,IAAI,IAAI,KAAK,KAAK,IAAI,IAAI,CAAC,QAAQ,CAAC,GAAG,CAAC,EAAE,CAAC;QACzC,OAAO,IAAI,UAAU,CAAC,KAAK,CAAC,CAAC;IAC/B,CAAC;SAAM,IAAI,IAAI,KAAK,KAAK,IAAI,IAAI,CAAC,QAAQ,CAAC,GAAG,CAAC,EAAE,CAAC;QAChD,OAAO,IAAI,WAAW,CAAC,KAAK,CAAC,CAAC;IAChC,CAAC;SAAM,CAAC;QACN,OAAO,IAAI,YAAY,CAAC,KAAK,CAAC,CAAC;IACjC,CAAC;AACH,CAAC;AAwBD,MAAM,OAAO,gBAAgB;IACnB,MAAM,CAAC;IAEP,OAAO,CAAC;IACR,eAAe,GAA+B,EAAE,CAAC;IACjD,aAAa,GAA2C,EAAE,CAAC;IAE3D,SAAS,CAAqB;IAC9B,WAAW,GAAmB,EAAE,CAAC;IACjC,oBAAoB,GAAG,IAAI,CAAC;IAC5B,0BAA0B,GAAG,IAAI,CAAC;IAC1C;;OAEG;IACK,eAAe,CAGrB;IACM,OAAO,GAAa,EAAE,CAAC;IACvB,oBAAoB,GAExB,EAAE,CAAC;IACC,QAAQ,GAAa,EAAE,CAAC;IACxB,SAAS,GAAc,EAAE,CAAC;IAC1B,YAAY,GAA4B,EAAE,CAAC;IAC3C,eAAe,GAA8B,EAAE,CAAC;IAEhD,SAAS,GAA+B,EAAE,CAAC;IAE3C,cAAc,GAAG,CAAC,CAAC;IACnB,eAAe,GAAG,CAAC,CAAC;IACpB,gBAAgB,GAAG,KAAK,CAAC;IACzB,YAAY,GAAa,EAAE,CAAC;IAC5B,MAAM,GAAG,EAAE,CAAC;IACZ,OAAO,GAAG,EAAE,CAAC;IAEb,OAAO,GAAG,EAAE,CAAC;IACJ,OAAO,CAAC;IACR,SAAS,CAAC;IAEnB,kBAAkB,CAAC;IAE3B,YACE,KAAa,EACb,MAAiB,EACjB,IASC;QAED,IAAI,CAAC,MAAM,GAAG,KAAK,CAAC;QACpB,IAAI,CAAC,OAAO,GAAG,MAAM,CAAC;QACtB,IAAI,CAAC,OAAO,GAAG,IAAI,CAAC,MAAM,CAAC;QAC3B,IAAI,CAAC,SAAS,GAAG,IAAI,CAAC,QAAQ,CAAC;QAC/B,IAAI,CAAC,OAAO,GAAG,IAAI,CAAC,MAAM,CAAC;QAC3B,IAAI,CAAC,oBAAoB,GAAG,IAAI,CAAC,mBAAmB,IAAI,EAAE,CAAC;QAC3D,IAAI,CAAC,QAAQ,GAAG,IAAI,CAAC,OAAO,CAAC;QAC7B,IAAI,CAAC,SAAS,GAAG,CAAC,IAAI,CAAC,QAAQ,IAAI,EAAE,CAAC,CAAC,MAAM,CAAC;YAC5C,KAAK,EAAE,cAAc;YACrB,IAAI,EAAE,OAAO;YACb,IAAI,EAAE,CAAC,CAAC,EAAE,CAAC,CAAC;SACb,CAAC,CAAC;QACH,IAAI,CAAC,SAAS,GAAG,IAAI,CAAC,QAAQ,IAAI,EAAE,CAAC;QACrC,IAAI,CAAC,YAAY,GAAG,IAAI,CAAC,SAAS,CAAC,MAAM,CAAC,CAAC,GAAG,EAAE,OAAO,EAAE,EAAE;YACzD,GAAG,CAAC,OAAO,CAAC,KAAK,CAAC,GAAG,OAAO,CAAC;YAC7B,OAAO,GAAG,CAAC;QACb,CAAC,EAAE,EAA6B,CAAC,CAAC;QAClC,IAAI,CAAC,kBAAkB,GAAG,IAAI,CAAC,iBAAiB,IAAI,KAAK,CAAC;QAE1D,IAAI,CAAC,SAAS,CAAC,OAAO,CAAC,CAAC,OAAO,EAAE,EAAE;YACjC,MAAM,IAAI,GAAG,gBAAgB,CAAC,OAAO,CAAC,IAAI,EAAE,OAAO,CAAC,IAAI,CAAC,CAAC;YAC1D,IAAI,CAAC,eAAe,CAAC,OAAO,CAAC,KAAK,CAAC,GAAG,MAAM,CAAC,YAAY,CAAC;gBACxD,KAAK,EAAE,IAAI,CAAC,MAAM;gBAClB,IAAI,EAAE,IAAI,CAAC,UAAU;gBACrB,KAAK,EAAE,cAAc,CAAC,OAAO,GAAG,cAAc,CAAC,QAAQ;aACxD,CAAC,CAAC;YACH,IAAI,CAAC,OAAO,CAAC,KAAK,CAAC,WAAW,CAC5B,IAAI,CAAC,eAAe,CAAC,OAAO,CAAC,KAAK,CAAC,EACnC,CAAC,EACD,IAAI,CACL,CAAC;QACJ,CAAC,CAAC,CAAC;IACL,CAAC;IAED,8FAA8F;IAC9F,eAAe;QACb,IAAI,IAAI,CAAC,gBAAgB;YAAE,OAAO;QAClC,IAAI,CAAC,gBAAgB,GAAG,IAAI,CAAC;QAC7B,IAAI,CAAC,cAAc,GAAG,IAAI,CAAC,MAAM,CAAC;QAAC,IAAI,CAAC,eAAe,GAAG,IAAI,CAAC,OAAO,CAAC;QACvE,MAAM,QAAQ,GAAG,IAAI,CAAC,OAAO,CAAC,MAAM,CAAC,IAAI,CAAC,EAAE,CAAC,CAAC,IAAI,CAAC,oBAAoB,CAAC,IAAS,CAAC,IAAI,IAAI,CAAC,oBAAoB,CAAC,IAAS,CAAC,KAAK,IAAI,CAAC,CAAC;QACrI,IAAI,CAAC,YAAY,GAAG,QAAQ,CAAC;QAC7B,IAAI,QAAQ,CAAC,MAAM,EAAE,CAAC;YACpB,MAAM,OAAO,GAAY,EAAC,KAAK,EAAE,cAAc,EAAE,IAAI,EAAE,gBAAgB,QAAQ,CAAC,MAAM,GAAG,EAAE,IAAI,EAAE,IAAI,WAAW,CAAC,QAAQ,CAAC,MAAM,GAAG,CAAC,CAAC,EAAC,CAAC;YACvI,IAAI,CAAC,SAAS,CAAC,IAAI,CAAC,OAAO,CAAC,CAAC;YAAC,IAAI,CAAC,YAAY,CAAC,OAAO,CAAC,KAAK,CAAC,GAAG,OAAO,CAAC;YACzE,IAAI,CAAC,eAAe,CAAC,OAAO,CAAC,KAAK,CAAC,GAAG,IAAI,CAAC,OAAO,CAAC,YAAY,CAAC,EAAC,KAAK,EAAE,GAAG,IAAI,CAAC,MAAM,IAAI,OAAO,CAAC,KAAK,EAAE,EAAE,IAAI,EAAE,QAAQ,CAAC,MAAM,GAAG,EAAE,EAAE,KAAK,EAAE,cAAc,CAAC,OAAO,GAAG,cAAc,CAAC,QAAQ,EAAC,CAAC,CAAC;QACnM,CAAC;QACD,IAAI,CAAC,oBAAoB,GAAG,IAAI,CAAC;QAAC,IAAI,CAAC,OAAO,GAAG,EAAE,CAAC;IACtD,CAAC;IACD,OAAO,CAAC,KAAa,EAAE,MAAc;QACnC,KAAK,GAAG,IAAI,CAAC,GAAG,CAAC,CAAC,EAAE,IAAI,CAAC,IAAI,CAAC,KAAK,CAAC,CAAC,CAAC;QACtC,MAAM,GAAG,IAAI,CAAC,GAAG,CAAC,CAAC,EAAE,IAAI,CAAC,IAAI,CAAC,MAAM,CAAC,CAAC,CAAC;QACxC,MAAM,WAAW,GAAG,KAAK,KAAK,IAAI,CAAC,MAAM,IAAI,MAAM,KAAK,IAAI,CAAC,OAAO,CAAC;QACrE,IAAI,CAAC,MAAM,GAAG,KAAK,CAAC;QACpB,IAAI,CAAC,OAAO,GAAG,MAAM,CAAC;QACtB,IAAI,WAAW,EAAE,CAAC;YAChB,MAAM,SAAS,GAAG,IAAI,CAAC,GAAG,CAAC,IAAI,CAAC,cAAc,EAAE,KAAK,CAAC,CAAC;YACvD,MAAM,UAAU,GAAG,IAAI,CAAC,GAAG,CAAC,IAAI,CAAC,eAAe,EAAE,MAAM,CAAC,CAAC;YAC1D,IAAI,CAAC,0BAA0B,KAAK,CAAC,IAAI,CAAC,gBAAgB,IAAI,SAAS,KAAK,IAAI,CAAC,cAAc,IAAI,UAAU,KAAK,IAAI,CAAC,eAAe,CAAC;YACvI,IAAI,CAAC,cAAc,GAAG,SAAS,CAAC;YAAC,IAAI,CAAC,eAAe,GAAG,UAAU,CAAC;YACnE,IAAI,CAAC,oBAAoB,GAAG,IAAI,CAAC;YAEjC,IAAI,CAAC,UAAU,CAAC,cAAc,EAAE,CAAC,KAAK,EAAE,MAAM,CAAC,CAAC,CAAC;QACnD,CAAC;IACH,CAAC;IAED,eAAe,CAAC,YAA+C;QAC7D,MAAM,SAAS,GAAG,CAAC,CAAC,IAAI,CAAC,eAAe,CAAC;QACzC,IAAI,CAAC,eAAe,GAAG,SAAS,CAAC;QACjC,MAAM,cAAc,GAAG,MAAM,CAAC,IAAI,CAAC,YAAY,CAAC,CAAC,IAAI,CACnD,CAAC,GAAG,EAAE,EAAE,CACN,CAAC,oBAAoB,CAClB,YAAoB,CAAC,GAAG,CAAC,EAC1B,IAAI,CAAC,aAAa,CAAC,GAAG,CAAC,IAAK,EAA6B,CAC1D,CACJ,CAAC;QACF,IAAI,CAAC,0BAA0B,KAAK,cAAc,CAAC;QACnD,IAAI,CAAC,oBAAoB,KAAK,cAAc,IAAI,SAAS,CAAC;QAC1D,IAAI,cAAc,IAAI,SAAS;YAAE,IAAI,CAAC,OAAO,GAAG,EAAE,CAAC;QACnD,IAAI,CAAC,aAAa,GAAG,YAAY,CAAC;IACpC,CAAC;IAED,iBAAiB,CACf,aAAyB,EACzB,kBAAoC;QAEpC,IACE,CAAC,IAAI,CAAC,eAAe;YACrB,IAAI,CAAC,eAAe,CAAC,kBAAkB,KAAK,kBAAkB,EAC9D,CAAC;YACD,IAAI,CAAC,oBAAoB,GAAG,IAAI,CAAC;YACjC,IAAI,CAAC,OAAO,GAAG,EAAE,CAAC;QACpB,CAAC;QACD,IAAI,CAAC,eAAe,GAAG;YACrB,aAAa;YACb,kBAAkB;SACnB,CAAC;IACJ,CAAC;IAED,UAAU,CAAC,KAAa,EAAE,IAAqB;QAC7C,MAAM,MAAM,GAAG,IAAI,CAAC,eAAe,CAAC,KAAK,CAAC,CAAC;QAC3C,IAAI,CAAC,OAAO,CAAC,KAAK,CAAC,WAAW,CAC5B,MAAM,EACN,CAAC,EACD,gBAAgB,CAAC,IAAI,EAAE,IAAI,CAAC,YAAY,CAAC,KAAK,CAAC,CAAC,IAAI,CAAC,CACtD,CAAC;IACJ,CAAC;IAEgB,gBAAgB,GAAG,IAAI,GAAG,EAAc,CAAC;IAC1D,YAAY,CAAC,OAAmB,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;IACpI,aAAa,CAAC,OAAmB;QAC/B,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,EAAE,CAAC;YAC3D,KAAK,IAAI,CAAC,OAAO,CAAC,KAAK,CAAC,mBAAmB,EAAE,CAAC,KAAK,CAAC,GAAG,EAAE,CAAC,SAAS,CAAC,CAAC,IAAI,CAAC,GAAG,EAAE,CAAC,OAAO,CAAC,OAAO,EAAE,CAAC,CAAC;QACrG,CAAC;IACH,CAAC;IAED,SAAS,CAAC,IAAO;QACf,OAAO,IAAI,CAAC,eAAe,CAAC,IAAI,CAAC,CAAC;IACpC,CAAC;IAED,cAAc;QACZ,OAAO,MAAM,CAAC,MAAM,CAAC,IAAI,CAAC,eAAe,CAAC,CAAC,MAAM,CAAC,CAAC,GAAG,EAAE,MAAM,EAAE,EAAE,CAAC,GAAG,GAAG,MAAM,CAAC,IAAI,EAAE,CAAC,CAAC;cACpF,KAAK,CAAC,IAAI,CAAC,IAAI,GAAG,CAAC,CAAC,GAAG,MAAM,CAAC,MAAM,CAAC,IAAI,CAAC,eAAe,CAAC,EAAE,GAAG,IAAI,CAAC,gBAAgB,CAAC,CAAC,CAAC;iBACtF,MAAM,CAAC,CAAC,GAAG,EAAE,OAAO,EAAE,EAAE,CAAC,GAAG,GAAG,YAAY,CAAC,OAAO,CAAC,EAAE,CAAC,CAAC,CAAC;IAChE,CAAC;IAED,OAAO;QACL,MAAM,CAAC,IAAI,CAAC,IAAI,CAAC,eAAe,CAAC,CAAC,OAAO,CAAC,CAAC,GAAG,EAAE,EAAE;YAC/C,IAAI,CAAC,eAAuB,CAAC,GAAG,CAAC,CAAC,OAAO,EAAE,CAAC;QAC/C,CAAC,CAAC,CAAC;QACH,IAAI,GAAG,CAAC,CAAC,GAAG,MAAM,CAAC,MAAM,CAAC,IAAI,CAAC,eAAe,CAAC,EAAE,GAAG,IAAI,CAAC,gBAAgB,CAAC,CAAC,CAAC,OAAO,CAAC,OAAO,CAAC,EAAE,CAAC,OAAO,CAAC,OAAO,EAAE,CAAC,CAAC;QAClH,IAAI,CAAC,gBAAgB,CAAC,KAAK,EAAE,CAAC;IAChC,CAAC;IAEO,cAAc,CAAC,MAA8B;QACnD,OAAO,IAAI,CAAC,OAAO,CAAC,aAAa,CAAC;YAChC,KAAK,EAAE,IAAI,CAAC,MAAM;YAClB,IAAI,EAAE;gBACJ,KAAK,EAAE,IAAI,CAAC,gBAAgB,CAAC,CAAC,CAAC,IAAI,CAAC,cAAc,CAAC,CAAC,CAAC,IAAI,CAAC,MAAM;gBAChE,MAAM,EAAE,IAAI,CAAC,gBAAgB,CAAC,CAAC,CAAC,IAAI,CAAC,eAAe,CAAC,CAAC,CAAC,IAAI,CAAC,OAAO;gBACnE,kBAAkB,EAAE,CAAC;aACtB;YACD,MAAM,EAAE,MAAM,CAAC,MAAM;YACrB,KAAK,EACH,eAAe,CAAC,iBAAiB;gBACjC,eAAe,CAAC,eAAe;gBAC/B,CAAC,IAAI,CAAC,kBAAkB,CAAC,CAAC,CAAC,eAAe,CAAC,QAAQ,CAAC,CAAC,CAAC,CAAC,CAAC;SAC3D,CAAC,CAAC;IACL,CAAC;IAEO,qBAAqB,CAAC,QAAmB;QAC/C,MAAM,YAAY,GAAG,IAAI,CAAC,aAAa,CAAC;QACxC,KAAK,MAAM,GAAG,IAAI,YAAY,EAAE,CAAC;YAC/B,IAAI,QAAQ,EAAE,QAAQ,CAAC,GAAG,CAAC,EAAE,CAAC;gBAC5B,SAAS;YACX,CAAC;YACD,MAAM,MAAM,GAAG,YAAY,CAAC,GAAG,CAAC,CAAC;YACjC,IAAI,CAAC,IAAI,CAAC,gBAAgB,CAAC,GAAG,CAAC,IAAI,CAAC,eAAe,CAAC,GAAG,CAAC,CAAC;gBAAE,IAAI,CAAC,eAAe,CAAC,GAAG,CAAC,EAAE,OAAO,EAAE,CAAC;YAChG,IAAI,CAAC,eAAe,CAAC,GAAG,CAAC,GAAG,IAAI,CAAC,cAAc,CAAC,MAAM,CAAC,CAAC;QAC1D,CAAC;IACH,CAAC;IAEO,eAAe;QACrB,IAAI,CAAC,IAAI,CAAC,oBAAoB,EAAE,CAAC;YAC/B,OAAO;QACT,CAAC;QACD,IAAI,CAAC,oBAAoB,GAAG,KAAK,CAAC;QAElC,MAAM,YAAY,GAAG,IAAI,CAAC,aAAa,CAAC;QACxC,MAAM,MAAM,GAAG,IAAI,CAAC,OAAO,CAAC;QAC5B,MAAM,MAAM,GAAG,IAAI,CAAC,UAAU,EAAE,CAAC;QACjC,IAAI,MAAM,KAAK,IAAI,CAAC,OAAO,EAAE,CAAC;YAC5B,OAAO;QACT,CAAC;QACD,MAAM,EAAE,aAAa,EAAE,kBAAkB,EAAE,GAAG,IAAI,CAAC,eAAe,IAAI,EAAE,CAAC;QACzE,IAAI,CAAC,OAAO,GAAG,MAAM,CAAC;QACtB,IAAI,CAAC,SAAS,GAAG,oBAAoB,CAAC,MAAM,CAAC,CAAC,oBAAoB,CAAC;YACjE,KAAK,EAAE,IAAI,CAAC,MAAM;YAClB,MAAM,EAAE,MAAM;YACd,MAAM,EAAE;gBACN,MAAM,EAAE,oBAAoB,CAAC,MAAM,CAAC,CAAC,kBAAkB,CAAC;oBACtD,KAAK,EAAE,IAAI,CAAC,MAAM;oBAClB,IAAI,EAAE,8BAA8B;iBACrC,CAAC;gBACF,UAAU,EAAE,MAAM;aACnB;YACD,QAAQ,EAAE;gBACR,MAAM,EAAE,oBAAoB,CAAC,MAAM,CAAC,CAAC,kBAAkB,CAAC;oBACtD,KAAK,EAAE,IAAI,CAAC,MAAM;oBAClB,IAAI,EAAE,MAAM;iBACb,CAAC;gBACF,UAAU,EAAE,MAAM;gBAClB,OAAO,EAAE,aAAa;oBACpB,CAAC,CAAC;wBACE;4BACE,MAAM,EAAE,kBAAmB;yBAC5B;qBACF;oBACH,CAAC,CAAC,IAAI,CAAC,QAAQ,CAAC,GAAG,CAAC,CAAC,GAAG,EAAE,EAAE,CAAC,CAAC;wBAC1B,MAAM,EAAG,YAAoB,CAAC,GAAG,CAAC,CAAC,MAAM;qBAC1C,CAAC,CAAC;aACR;YACD,SAAS,EAAE;gBACT,QAAQ,EAAE,eAAe;aAC1B;SACF,CAAC,CAAC;QACH,IAAI,CAAC,iBAAiB,EAAE,CAAC;IAC3B,CAAC;IAEO,UAAU;QAChB,MAAM,MAAM,GAAG,IAAI,CAAC,OAAO,CAAC;QAC5B,MAAM,SAAS,GAAG,MAAM,CAAC,MAAM,CAAC;QAChC,MAAM,SAAS,GAAG,SAAS,GAAG,CAAC,CAAC;QAChC,MAAM,EAAE,GAAG,UAAU,CAAC;EACxB,MAAM;aACL,IAAI,EAAE;aACN,GAAG,CACF,CAAC,WAAW,EAAE,GAAG,EAAE,EAAE;YACnB,MAAM,aAAa,GAAG,IAAI,CAAC,oBAAoB,CAAC,WAAgB,CAAC,CAAC;YAClE,MAAM,WAAW,GAAG,aAAa,KAAK,IAAI,CAAC,CAAC,CAAC,YAAY,CAAC,CAAC,CAAC,YAAY,CAAC;YACzE,OAAO,sBAAsB,GAAG,SAAS,WAAW,KAAK,WAAW,QAAQ,CAAC;QAC/E,CAAC,CACF;aACA,IAAI,CAAC,IAAI,CAAC;;EAGX,SAAS;YACP,CAAC,CAAC,MAAM,CAAC,IAAI,CAAC,IAAI,CAAC,SAAS,CAAC;iBACxB,IAAI,EAAE;iBACN,GAAG,CACF,CAAC,WAAW,EAAE,GAAG,EAAE,EAAE,CACnB,sBACE,GAAG,GAAG,SACR,SAAS,WAAW,YAAY,CACnC;iBACA,IAAI,CAAC,IAAI,CAAC;YACf,CAAC,CAAC,EACN;EACE,IAAI,CAAC,SAAS;aACb,GAAG,CACF,CAAC,OAAO,EAAE,GAAG,EAAE,EAAE,CACf,UAAU,SAAS,CAAC,CAAC,CAAC,CAAC,CAAC,CAAC,CAAC,CAAC,cAAc,GAAG,kBAC1C,OAAO,CAAC,KACV,KAAK,OAAO,CAAC,IAAI,GAAG,CACvB;aACA,IAAI,CAAC,IAAI,CAAC;;EAEX,IAAI,CAAC,QAAQ;aACZ,GAAG,CAAC,CAAC,IAAI,EAAE,GAAG,EAAE,EAAE,CAAC,aAAa,GAAG,KAAK,IAAI,UAAU,CAAC;aACvD,IAAI,CAAC,IAAI,CAAC;;EAEX,oBAAoB,CACpB,IAAI,CAAC,gBAAgB,CAAC,CAAC,CAAC,IAAI,CAAC,YAAY,CAAC,CAAC,CAAC,MAAM,CAAC,MAAM,CAAC,IAAI,CAAC,EAAE,CAC/D,CAAC,IAAI,CAAC,oBAAoB,CAAC,IAAS,CAAC,IAAI,IAAI,CAAC,oBAAoB,CAAC,IAAS,CAAC,KAAK,IAAI,CAAC,EACzF,MAAM,CAAC,IAAI,CAAC,IAAI,CAAC,SAAS,CAAC,EAAE,IAAI,CAAC,gBAAgB,CACnD;EACC,IAAI,CAAC,SAAS,IAAI,EAAE;;;;;;;;EAQpB,IAAI,CAAC,OAAO;;;CAGb,CAAC;QAEE,OAAO,EAAE,CAAC;IACZ,CAAC;IAEO,iBAAiB;QACvB,MAAM,UAAU,GAAmB,EAAE,CAAC;QACtC,MAAM,MAAM,GAAG,IAAI,CAAC,OAAO,CAAC;QAC5B,MAAM,qBAAqB,GAAG,IAAI,CAAC,OAAO,CAAC,MAAM,GAAG,CAAC,CAAC,CAAC,CAAC,CAAC,CAAC,CAAC,CAAC,CAAC,CAAC;QAC9D,IAAI,IAAI,CAAC,SAAS,CAAC,MAAM,GAAG,CAAC,EAAE,CAAC;YAC9B,UAAU,CAAC,qBAAqB,CAAC,GAAG,MAAM,CAAC,eAAe,CAAC;gBACzD,KAAK,EAAE,IAAI,CAAC,MAAM;gBAClB,MAAM,EAAE,IAAI,CAAC,SAAS,CAAC,kBAAkB,CAAC,qBAAqB,CAAC;gBAChE,OAAO,EAAE,IAAI,CAAC,SAAS,CAAC,GAAG,CACzB,CAAC,OAAO,EAAE,GAAG,EAAE,EAAE,CACf,CAAC;oBACC,OAAO,EAAE,GAAG;oBACZ,QAAQ,EAAE;wBACR,MAAM,EAAE,IAAI,CAAC,eAAe,CAAC,OAAO,CAAC,KAAK,CAAC;qBAC5C;iBAC2B,CAAA,CACjC;aACF,CAAC,CAAC;QACL,CAAC;QAED,IAAI,CAAC,WAAW,GAAG,UAAU,CAAC;IAChC,CAAC;IAED,UAAU,CACR,cAAiC,EACjC,MAA6B;IAE7B;;OAEG;IACH,OAA+B;QAE/B,IAAI,IAAI,CAAC,0BAA0B,IAAI,CAAC,IAAI,CAAC,eAAe,EAAE,CAAC;YAC7D,IAAI,CAAC,qBAAqB,CAAC,MAAM,CAAC,IAAI,CAAC,OAAO,IAAI,EAAE,CAAC,CAAC,CAAC;YACvD,IAAI,CAAC,0BAA0B,GAAG,KAAK,CAAC;QAC1C,CAAC;QAED,IAAI,IAAI,CAAC,YAAY,CAAC,MAAM,EAAE,CAAC;YAC7B,MAAM,MAAM,GAAG,IAAI,WAAW,CAAC,IAAI,CAAC,YAAY,CAAC,MAAM,GAAG,CAAC,CAAC,CAAC;YAC7D,KAAK,MAAM,CAAC,KAAK,EAAE,IAAI,CAAC,IAAI,IAAI,CAAC,YAAY,CAAC,OAAO,EAAE,EAAE,CAAC;gBACxD,MAAM,MAAM,GAAG,aAAa,CAAC,MAAM,CAAC,IAAS,CAAC,CAAC,CAAC;gBAChD,MAAM,CAAC,GAAG,CAAC,CAAC,MAAM,CAAC,KAAK,EAAE,MAAM,CAAC,MAAM,CAAC,EAAE,KAAK,GAAG,CAAC,CAAC,CAAC;YACvD,CAAC;YACD,IAAI,CAAC,UAAU,CAAC,cAAc,EAAE,MAAM,CAAC,CAAC;QAC1C,CAAC;QACD,KAAK,MAAM,OAAO,IAAI,MAAM,CAAC,MAAM,CAAC,IAAI,CAAC,eAAe,CAAC;YAAE,gBAAgB,CAAC,OAAO,EAAE,IAAI,CAAC,MAAM,EAAE,IAAI,CAAC,OAAO,CAAC,CAAC;QAChH,IAAI,CAAC,eAAe,EAAE,CAAC;QAEvB,MAAM,SAAS,GAAG,IAAI,CAAC,OAAO,CAAC,MAAM,CAAC;QACtC,IAAI,SAAS,GAAG,CAAC,EAAE,CAAC;YAClB,IAAI,CAAC,WAAW,CAAC,CAAC,CAAC,GAAG,IAAI,CAAC,OAAO,CAAC,eAAe,CAAC;gBACjD,KAAK,EAAE,IAAI,CAAC,MAAM;gBAClB,MAAM,EAAE,IAAI,CAAC,SAAS,CAAC,kBAAkB,CAAC,CAAC,CAAC;gBAC5C,OAAO,EAAE;oBACP,GAAG,IAAI,CAAC,OAAO,CAAC,GAAG,CAAC,CAAC,WAAW,EAAE,GAAG,EAAE,EAAE,CAAC,CAAC;wBACzC,OAAO,EAAE,GAAG;wBACZ,QAAQ,EAAE,MAAM,CAAC,WAAgB,CAAC,CAAC,UAAU,CAC3C,IAAI,CAAC,oBAAoB,CAAC,WAAgB,CAAC;4BACzC,CAAC,CAAC,EAAE,SAAS,EAAE,IAAI,CAAC,oBAAoB,CAAC,WAAgB,CAAC,EAAE;4BAC5D,CAAC,CAAC,SAAS,CACd;qBACF,CAAC,CAAC;oBACH,GAAG,MAAM,CAAC,IAAI,CAAC,IAAI,CAAC,SAAS,CAAC,CAAC,GAAG,CAAC,CAAC,WAAW,EAAE,GAAG,EAAE,EAAE,CAAC,CAAC;wBACxD,OAAO,EAAE,GAAG,GAAG,SAAS;wBACxB,QAAQ,EAAE,IAAI,CAAC,SAAS,CAAC,WAAW,CAAC;qBACtC,CAAC,CAAC;iBACJ;aACF,CAAC,CAAC;QACL,CAAC;QAED,MAAM,cAAc,GAAG;YACrB,8CAA8C;YAC9C,GAAG,IAAI,CAAC,eAAe;YACvB,GAAG,OAAO;SACmB,CAAC;QAChC,wBAAwB;QACxB,MAAM,UAAU,GAAG,cAAc,CAAC,eAAe,CAAC;YAChD,gBAAgB,EAAE,IAAI,CAAC,eAAe;gBACpC,CAAC,CAAC;oBACE;wBACE,IAAI,EAAE,IAAI,CAAC,eAAe,CAAC,aAAa,CAAC,UAAU,EAAE;wBACrD,UAAU,EAAE,EAAE,CAAC,EAAE,CAAC,EAAE,CAAC,EAAE,CAAC,EAAE,CAAC,EAAE,CAAC,EAAE,CAAC,EAAE,CAAC,EAAE;wBACtC,OAAO,EAAE,OAAO;wBAChB,MAAM,EAAE,OAAO;qBAChB;iBACF;gBACH,CAAC,CAAC,IAAI,CAAC,QAAQ,CAAC,GAAG,CACf,CAAC,WAAW,EAAE,EAAE,CACd,CAAC;oBACC,IAAI,EAAE,cAAc,CAAC,WAAW,CAAC,CAAC,UAAU,EAAE;oBAC9C,UAAU,EAAE,EAAE,CAAC,EAAE,CAAC,EAAE,CAAC,EAAE,CAAC,EAAE,CAAC,EAAE,CAAC,EAAE,CAAC,EAAE,CAAC,EAAE;oBACtC,MAAM,EAAE,OAAO;oBACf,OAAO,EAAE,OAAO;iBACuB,CAAA,CAC5C;SACN,CAAC,CAAC;QAEH,mBAAmB;QACnB,UAAU,CAAC,WAAW,CAAC,CAAC,EAAE,CAAC,EAAE,IAAI,CAAC,MAAM,EAAE,IAAI,CAAC,OAAO,EAAE,CAAC,EAAE,CAAC,CAAC,CAAC;QAC9D,UAAU,CAAC,cAAc,CAAC,CAAC,EAAE,CAAC,EAAE,IAAI,CAAC,MAAM,EAAE,IAAI,CAAC,OAAO,CAAC,CAAC;QAC3D,UAAU,CAAC,WAAW,CAAC,IAAI,CAAC,SAAS,CAAC,CAAC;QACvC,cAAc;QACd,IAAI,CAAC,WAAW,CAAC,OAAO,CAAC,CAAC,SAAS,EAAE,GAAG,EAAE,EAAE;YAC1C,UAAU,CAAC,YAAY,CAAC,GAAG,EAAE,SAAS,CAAC,CAAC;QAC1C,CAAC,CAAC,CAAC;QACH,UAAU,CAAC,IAAI,CAAC,CAAC,EAAE,CAAC,EAAE,CAAC,EAAE,CAAC,CAAC,CAAC;QAC5B,sBAAsB;QACtB,UAAU,CAAC,GAAG,EAAE,CAAC;IACnB,CAAC;CACF","sourcesContent":["import {scratchWGSLFunctions} from './scratchSampling.js';\nimport {setTextureRegion, textureRegion} from './textureRegion.js';\nimport { getTargetShaderCache } from './WGPUShaderCache.js';\nimport { textureBytes } from './gpuMemory.js';\nexport const fullScreenQuadVertexShaderWGSL = /*wgsl */ `\n@vertex\nfn main(\n  @builtin(vertex_index) vertexIndex: u32\n) -> @builtin(position) vec4f {\n  const pos = array(\n    vec2(-1.0, -1.0), vec2(1.0, -1.0), vec2(-1.0, 1.0),\n    vec2(-1.0, 1.0), vec2(1.0, -1.0), vec2(1.0, 1.0),\n  );\n\n  return vec4<f32>(pos[vertexIndex], 0.0, 1.0);\n}\n`;\n\nfunction isTextureParamsEqual(\n  params: WGPUFullQuadPassOutput,\n  other: WGPUFullQuadPassOutput\n) {\n  return params.format === other.format;\n}\n\nexport function getUniformBuffer(\n  value: any,\n  type: Uniform['type']\n): Float32Array | Int32Array | Uint32Array {\n  if (\n    value instanceof Float32Array ||\n    value instanceof Int32Array ||\n    value instanceof Uint32Array\n  ) {\n    return value;\n  }\n\n  if (typeof value === 'number') {\n    value = [value];\n  }\n\n  if (type === 'i32' || type.endsWith('i')) {\n    return new Int32Array(value);\n  } else if (type === 'u32' || type.endsWith('u')) {\n    return new Uint32Array(value);\n  } else {\n    return new Float32Array(value);\n  }\n}\n\nexport interface Uniform {\n  label: string;\n  type:\n    | `array<vec4u, ${number}>`\n    | 'f32'\n    | 'i32'\n    | 'u32'\n    | 'vec3f'\n    | 'vec4f'\n    | 'vec2f'\n    | 'vec2i'\n    | 'vec3i'\n    | 'vec4i'\n    | 'vec2u'\n    | 'vec3u'\n    | 'vec4u';\n  data: Float32Array | Int32Array | Uint32Array | number | number[];\n}\n\nexport interface WGPUFullQuadPassOutput {\n  format: GPUTextureFormat;\n}\nexport class WGPUFullQuadPass<I extends string, O extends string> {\n  private _label;\n\n  private _device;\n  private _outputTextures: Record<string, GPUTexture> = {};\n  private _outputParams: Record<string, WGPUFullQuadPassOutput> = {};\n\n  private _pipeline!: GPURenderPipeline;\n  private _bindGroups: GPUBindGroup[] = [];\n  private _needsUpdatePipeline = true;\n  private _needsUpdateOutputTextures = true;\n  /**\n   * When render to canvas\n   */\n  private _renderToScreen?: {\n    screenTexture: GPUTexture;\n    presentationFormat: GPUTextureFormat;\n  };\n  private _inputs: string[] = [];\n  private _inputViewDimensions: Partial<\n    Record<I, GPUTextureViewDimension>\n  > = {};\n  private _outputs: string[] = [];\n  private _uniforms: Uniform[] = [];\n  private _uniformsMap: Record<string, Uniform> = {};\n  private _uniformBuffers: Record<string, GPUBuffer> = {};\n\n  private _samplers: Record<string, GPUSampler> = {};\n\n  private _capacityWidth = 0;\n  private _capacityHeight = 0;\n  private _reserveCapacity = false;\n  private _regionNames: string[] = [];\n  private _width = 10;\n  private _height = 10;\n\n  private _fsCode = '';\n  private readonly _fsMain;\n  private readonly _fsDefine;\n\n  private _textureWillBeRead;\n\n  constructor(\n    label: string,\n    device: GPUDevice,\n    opts: {\n      inputs: I[];\n      inputViewDimensions?: Partial<Record<I, GPUTextureViewDimension>>;\n      outputs: O[];\n      samplers?: Record<string, GPUSampler>;\n      fsMain: string;\n      fsDefine?: string;\n      uniforms?: Uniform[];\n      textureWillBeRead?: boolean;\n    }\n  ) {\n    this._label = label;\n    this._device = device;\n    this._fsMain = opts.fsMain;\n    this._fsDefine = opts.fsDefine;\n    this._inputs = opts.inputs;\n    this._inputViewDimensions = opts.inputViewDimensions || {};\n    this._outputs = opts.outputs;\n    this._uniforms = (opts.uniforms || []).concat({\n      label: 'viewportSize',\n      type: 'vec2f',\n      data: [0, 0]\n    });\n    this._samplers = opts.samplers || {};\n    this._uniformsMap = this._uniforms.reduce((map, uniform) => {\n      map[uniform.label] = uniform;\n      return map;\n    }, {} as Record<string, Uniform>);\n    this._textureWillBeRead = opts.textureWillBeRead || false;\n\n    this._uniforms.forEach((uniform) => {\n      const data = getUniformBuffer(uniform.data, uniform.type);\n      this._uniformBuffers[uniform.label] = device.createBuffer({\n        label: this._label,\n        size: data.byteLength,\n        usage: GPUBufferUsage.UNIFORM | GPUBufferUsage.COPY_DST\n      });\n      this._device.queue.writeBuffer(\n        this._uniformBuffers[uniform.label],\n        0,\n        data\n      );\n    });\n  }\n\n  /** Keep physical outputs large enough for every render while varying the logical viewport. */\n  reserveCapacity() {\n    if (this._reserveCapacity) return;\n    this._reserveCapacity = true;\n    this._capacityWidth = this._width; this._capacityHeight = this._height;\n    const textures = this._inputs.filter(name => !this._inputViewDimensions[name as I] || this._inputViewDimensions[name as I] === '2d');\n    this._regionNames = textures;\n    if (textures.length) {\n      const uniform: Uniform = {label: 'imageRegions', type: `array<vec4u, ${textures.length}>`, data: new Uint32Array(textures.length * 4)};\n      this._uniforms.push(uniform); this._uniformsMap[uniform.label] = uniform;\n      this._uniformBuffers[uniform.label] = this._device.createBuffer({label: `${this._label}.${uniform.label}`, size: textures.length * 16, usage: GPUBufferUsage.UNIFORM | GPUBufferUsage.COPY_DST});\n    }\n    this._needsUpdatePipeline = true; this._fsCode = '';\n  }\n  setSize(width: number, height: number) {\n    width = Math.max(1, Math.ceil(width));\n    height = Math.max(1, Math.ceil(height));\n    const sizeChanged = width !== this._width || height !== this._height;\n    this._width = width;\n    this._height = height;\n    if (sizeChanged) {\n      const nextWidth = Math.max(this._capacityWidth, width);\n      const nextHeight = Math.max(this._capacityHeight, height);\n      this._needsUpdateOutputTextures ||= !this._reserveCapacity || nextWidth !== this._capacityWidth || nextHeight !== this._capacityHeight;\n      this._capacityWidth = nextWidth; this._capacityHeight = nextHeight;\n      this._needsUpdatePipeline = true;\n\n      this.setUniform('viewportSize', [width, height]);\n    }\n  }\n\n  setOutputParams(outputParams: Record<O, WGPUFullQuadPassOutput>) {\n    const wasScreen = !!this._renderToScreen;\n    this._renderToScreen = undefined;\n    const outputsChanged = Object.keys(outputParams).some(\n      (key) =>\n        !isTextureParamsEqual(\n          (outputParams as any)[key],\n          this._outputParams[key] || ({} as WGPUFullQuadPassOutput)\n        )\n    );\n    this._needsUpdateOutputTextures ||= outputsChanged;\n    this._needsUpdatePipeline ||= outputsChanged || wasScreen;\n    if (outputsChanged || wasScreen) this._fsCode = '';\n    this._outputParams = outputParams;\n  }\n\n  setRenderToScreen(\n    screenTexture: GPUTexture,\n    presentationFormat: GPUTextureFormat\n  ) {\n    if (\n      !this._renderToScreen ||\n      this._renderToScreen.presentationFormat !== presentationFormat\n    ) {\n      this._needsUpdatePipeline = true;\n      this._fsCode = '';\n    }\n    this._renderToScreen = {\n      screenTexture,\n      presentationFormat\n    };\n  }\n\n  setUniform(label: string, data: Uniform['data']) {\n    const buffer = this._uniformBuffers[label];\n    this._device.queue.writeBuffer(\n      buffer,\n      0,\n      getUniformBuffer(data, this._uniformsMap[label].type)\n    );\n  }\n\n  private readonly _retainedOutputs = new Set<GPUTexture>();\n  retainOutput(texture: GPUTexture) { if (Object.values(this._outputTextures).includes(texture)) this._retainedOutputs.add(texture); }\n  releaseOutput(texture: GPUTexture) {\n    if (!this._retainedOutputs.delete(texture)) return;\n    if (!Object.values(this._outputTextures).includes(texture)) {\n      void this._device.queue.onSubmittedWorkDone().catch(() => undefined).then(() => texture.destroy());\n    }\n  }\n\n  getOutput(name: O) {\n    return this._outputTextures[name];\n  }\n\n  getMemoryBytes() {\n    return Object.values(this._uniformBuffers).reduce((sum, buffer) => sum + buffer.size, 0)\n      + Array.from(new Set([...Object.values(this._outputTextures), ...this._retainedOutputs]))\n        .reduce((sum, texture) => sum + textureBytes(texture), 0);\n  }\n\n  dispose() {\n    Object.keys(this._uniformBuffers).forEach((key) => {\n      (this._uniformBuffers as any)[key].destroy();\n    });\n    new Set([...Object.values(this._outputTextures), ...this._retainedOutputs]).forEach(texture => texture.destroy());\n    this._retainedOutputs.clear();\n  }\n\n  private _createTexture(params: WGPUFullQuadPassOutput) {\n    return this._device.createTexture({\n      label: this._label,\n      size: {\n        width: this._reserveCapacity ? this._capacityWidth : this._width,\n        height: this._reserveCapacity ? this._capacityHeight : this._height,\n        depthOrArrayLayers: 1\n      },\n      format: params.format,\n      usage:\n        GPUTextureUsage.RENDER_ATTACHMENT |\n        GPUTextureUsage.TEXTURE_BINDING |\n        (this._textureWillBeRead ? GPUTextureUsage.COPY_SRC : 0)\n    });\n  }\n\n  private _updateOutputTextures(excludes?: string[]) {\n    const outputParams = this._outputParams;\n    for (const key in outputParams) {\n      if (excludes?.includes(key)) {\n        continue;\n      }\n      const params = outputParams[key];\n      if (!this._retainedOutputs.has(this._outputTextures[key])) this._outputTextures[key]?.destroy();\n      this._outputTextures[key] = this._createTexture(params);\n    }\n  }\n\n  private _updatePipeline() {\n    if (!this._needsUpdatePipeline) {\n      return;\n    }\n    this._needsUpdatePipeline = false;\n\n    const outputParams = this._outputParams;\n    const device = this._device;\n    const fsCode = this._getFullFs();\n    if (fsCode === this._fsCode) {\n      return;\n    }\n    const { screenTexture, presentationFormat } = this._renderToScreen || {};\n    this._fsCode = fsCode;\n    this._pipeline = getTargetShaderCache(device).createRenderPipeline({\n      label: this._label,\n      layout: 'auto',\n      vertex: {\n        module: getTargetShaderCache(device).createShaderModule({\n          label: this._label,\n          code: fullScreenQuadVertexShaderWGSL\n        }),\n        entryPoint: 'main'\n      },\n      fragment: {\n        module: getTargetShaderCache(device).createShaderModule({\n          label: this._label,\n          code: fsCode\n        }),\n        entryPoint: 'main',\n        targets: screenTexture\n          ? [\n              {\n                format: presentationFormat!\n              }\n            ]\n          : this._outputs.map((key) => ({\n              format: (outputParams as any)[key].format\n            }))\n      },\n      primitive: {\n        topology: 'triangle-list'\n      }\n    });\n    this._updateBindGroups();\n  }\n\n  private _getFullFs() {\n    const inputs = this._inputs;\n    const inputsLen = inputs.length;\n    const hasInputs = inputsLen > 0;\n    const fs = /* wgsl */ `\n${inputs\n  .sort()\n  .map(\n    (textureName, idx) => {\n      const viewDimension = this._inputViewDimensions[textureName as I];\n      const textureType = viewDimension === '3d' ? 'texture_3d' : 'texture_2d';\n      return `@group(0) @binding(${idx}) var ${textureName}: ${textureType}<f32>;`;\n    }\n  )\n  .join('\\n')}\n\n${\n  hasInputs\n    ? Object.keys(this._samplers)\n        .sort()\n        .map(\n          (samplerName, idx) =>\n            `@group(0) @binding(${\n              idx + inputsLen\n            }) var ${samplerName}: sampler;`\n        )\n        .join('\\n')\n    : ''\n}\n${this._uniforms\n  .map(\n    (uniform, idx) =>\n      `@group(${hasInputs ? 1 : 0}) @binding(${idx}) var<uniform> ${\n        uniform.label\n      }: ${uniform.type};`\n  )\n  .join('\\n')}\nstruct FSOutput {\n${this._outputs\n  .map((name, idx) => `@location(${idx}) ${name}: vec4f,`)\n  .join('\\n')}\n}\n${scratchWGSLFunctions(\n  this._reserveCapacity ? this._regionNames : inputs.filter(name =>\n    !this._inputViewDimensions[name as I] || this._inputViewDimensions[name as I] === '2d'),\n  Object.keys(this._samplers), this._reserveCapacity\n)}\n${this._fsDefine ?? ''}\n@fragment\nfn main(\n  @builtin(position) coord: vec4f\n) -> FSOutput {\n  let uv = vec2i(floor(coord.xy));\n  let uv2 = coord.xy / viewportSize;\n  var output: FSOutput;\n${this._fsMain}\n  return output;\n}\n`;\n\n    return fs;\n  }\n\n  private _updateBindGroups() {\n    const bindGroups: GPUBindGroup[] = [];\n    const device = this._device;\n    const uniformBindGroupIndex = this._inputs.length > 0 ? 1 : 0;\n    if (this._uniforms.length > 0) {\n      bindGroups[uniformBindGroupIndex] = device.createBindGroup({\n        label: this._label,\n        layout: this._pipeline.getBindGroupLayout(uniformBindGroupIndex),\n        entries: this._uniforms.map(\n          (uniform, idx) =>\n            ({\n              binding: idx,\n              resource: {\n                buffer: this._uniformBuffers[uniform.label]\n              }\n            } satisfies GPUBindGroupEntry)\n        )\n      });\n    }\n\n    this._bindGroups = bindGroups;\n  }\n\n  createPass(\n    commandEncoder: GPUCommandEncoder,\n    inputs: Record<I, GPUTexture>,\n\n    /**\n     * Explict outputs. Used for buffers reuse.\n     */\n    outputs?: Record<O, GPUTexture>\n  ) {\n    if (this._needsUpdateOutputTextures && !this._renderToScreen) {\n      this._updateOutputTextures(Object.keys(outputs || {}));\n      this._needsUpdateOutputTextures = false;\n    }\n\n    if (this._regionNames.length) {\n      const values = new Uint32Array(this._regionNames.length * 4);\n      for (const [index, name] of this._regionNames.entries()) {\n        const region = textureRegion(inputs[name as I]);\n        values.set([region.width, region.height], index * 4);\n      }\n      this.setUniform('imageRegions', values);\n    }\n    for (const texture of Object.values(this._outputTextures)) setTextureRegion(texture, this._width, this._height);\n    this._updatePipeline();\n\n    const inputsLen = this._inputs.length;\n    if (inputsLen > 0) {\n      this._bindGroups[0] = this._device.createBindGroup({\n        label: this._label,\n        layout: this._pipeline.getBindGroupLayout(0),\n        entries: [\n          ...this._inputs.map((textureName, idx) => ({\n            binding: idx,\n            resource: inputs[textureName as I].createView(\n              this._inputViewDimensions[textureName as I]\n                ? { dimension: this._inputViewDimensions[textureName as I] }\n                : undefined\n            )\n          })),\n          ...Object.keys(this._samplers).map((samplerName, idx) => ({\n            binding: idx + inputsLen,\n            resource: this._samplers[samplerName]\n          }))\n        ]\n      });\n    }\n\n    const outputTextures = {\n      // Pick the texture from outputTextures first.\n      ...this._outputTextures,\n      ...outputs\n    } as Record<string, GPUTexture>;\n    // Begin the render pass\n    const renderPass = commandEncoder.beginRenderPass({\n      colorAttachments: this._renderToScreen\n        ? [\n            {\n              view: this._renderToScreen.screenTexture.createView(),\n              clearValue: { r: 0, g: 0, b: 0, a: 0 },\n              storeOp: 'store',\n              loadOp: 'clear'\n            }\n          ]\n        : this._outputs.map(\n            (textureName) =>\n              ({\n                view: outputTextures[textureName].createView(),\n                clearValue: { r: 0, g: 0, b: 0, a: 0 },\n                loadOp: 'clear',\n                storeOp: 'store'\n              } satisfies GPURenderPassColorAttachment)\n          )\n    });\n\n    // Draw a full quad\n    renderPass.setViewport(0, 0, this._width, this._height, 0, 1);\n    renderPass.setScissorRect(0, 0, this._width, this._height);\n    renderPass.setPipeline(this._pipeline);\n    // Bind groups\n    this._bindGroups.forEach((bindGroup, idx) => {\n      renderPass.setBindGroup(idx, bindGroup);\n    });\n    renderPass.draw(6, 1, 0, 0);\n    // End the render pass\n    renderPass.end();\n  }\n}\n"]}

package/lib/renderer/WGPUJNLMDenoiser.d.ts

@@ -1,25 +1,6 @@
 import { type JNLMFilterShape } from './jnlm.js';
+export { createJNLMFilterWGSL, jnlmGuideWGSL, jnlmVarianceWGSL, jnlmLoadColorWGSL, JNLM_WORKGROUP_SIZE } from './wgsl/jnlm.compute.js';
 /** WebGPU passes of the JNLM denoiser described in `jnlm.ts`. */
-export declare const JNLM_WORKGROUP_SIZE = 8;
-/**
- * Pass 1: the guide mean and the variance of that mean from the accumulated
- * moments. Ray buffers are stored bottom-up; guides use texture rows.
- */
-export declare const jnlmGuideWGSL: string;
-/**
- * Pass 2: smooth the noisy per-pixel variance over 3 x 3 pixels. Pixels with
- * a single sample take the spatial variance of their neighborhood instead.
- */
-export declare const jnlmVarianceWGSL = "\n\nstruct JNLMParams {\n  size: vec2u,\n  colorSensitivity: f32,\n  // Pixel spacing of the search window and patches in this pass.\n  dilation: f32,\n}\n@group(0) @binding(0) var<uniform> params: JNLMParams;\n\n@group(0) @binding(1) var<storage, read> rawGuide: array<vec2f>;\n@group(0) @binding(2) var<storage, read_write> guide: array<vec2f>;\n\n@compute @workgroup_size(8, 8, 1)\nfn main(@builtin(global_invocation_id) id: vec3u) {\n  let size = params.size;\n  if (id.x >= size.x || id.y >= size.y) { return; }\n  let center = rawGuide[id.x + id.y * size.x];\n  var varianceSum = 0.0;\n  var varianceCount = 0.0;\n  var meanSum = 0.0;\n  var meanSquareSum = 0.0;\n  for (var dy = -1; dy <= 1; dy++) {\n    for (var dx = -1; dx <= 1; dx++) {\n      let q = clamp(vec2i(id.xy) + vec2i(dx, dy), vec2i(0), vec2i(size) - 1);\n      let value = rawGuide[u32(q.x) + u32(q.y) * size.x];\n      meanSum += value.x;\n      meanSquareSum += value.x * value.x;\n      if (value.y >= 0.0) {\n        varianceSum += value.y;\n        varianceCount += 1.0;\n      }\n    }\n  }\n  var variance: f32;\n  if (center.y >= 0.0) {\n    variance = varianceSum / varianceCount;\n  } else {\n    let spatialMean = meanSum / 9.0;\n    variance = max(meanSquareSum / 9.0 - spatialMean * spatialMean, 0.0) * 9.0 / 8.0;\n  }\n  guide[id.x + id.y * size.x] = vec2f(center.x, variance);\n}\n";
-/** Copies the composed image into the filter's color buffer. */
-export declare const jnlmLoadColorWGSL = "\n\nstruct JNLMParams {\n  size: vec2u,\n  colorSensitivity: f32,\n  // Pixel spacing of the search window and patches in this pass.\n  dilation: f32,\n}\n@group(0) @binding(0) var<uniform> params: JNLMParams;\n\n@group(0) @binding(1) var colorTex: texture_2d<f32>;\n@group(0) @binding(2) var<storage, read_write> outColor: array<vec4f>;\n\n@compute @workgroup_size(8, 8, 1)\nfn main(@builtin(global_invocation_id) id: vec3u) {\n  if (id.x >= params.size.x || id.y >= params.size.y) { return; }\n  outColor[id.x + id.y * params.size.x] = textureLoad(colorTex, vec2i(id.xy), 0);\n}\n";
-/**
- * Pass 3: the joint NL-means filter of the composed image. It also filters
- * the guide with the same weights, with the variance of the weighted mean,
- * for a following pass. The albedo regression corrects color only, keeping
- * uncertain fitted texture responses out of the matching guide.
- */
-export declare function createJNLMFilterWGSL({ searchRadius, patchRadius, albedoBandwidth, normalBandwidth }?: JNLMFilterShape): string;
 export interface JNLMInputs {
     width: number;
     height: number;