typegpu 0.5.7 → 0.5.9
This diff represents the content of publicly available package versions that have been released to one of the supported registries. The information contained in this diff is provided for informational purposes only and reflects changes between package versions as they appear in their respective public registries.
- package/chunk-734H4JWF.js +2 -0
- package/chunk-734H4JWF.js.map +1 -0
- package/chunk-BJSN5UVV.cjs +41 -0
- package/chunk-BJSN5UVV.cjs.map +1 -0
- package/chunk-BZVU4Q6A.cjs +2 -0
- package/chunk-BZVU4Q6A.cjs.map +1 -0
- package/chunk-CDUKTYES.cjs +2 -0
- package/chunk-CDUKTYES.cjs.map +1 -0
- package/chunk-CGVYLGBG.js +2 -0
- package/chunk-CGVYLGBG.js.map +1 -0
- package/chunk-SWJVYRFM.js +41 -0
- package/chunk-SWJVYRFM.js.map +1 -0
- package/data/index.cjs +1 -1
- package/data/index.cjs.map +1 -1
- package/data/index.d.cts +4 -77
- package/data/index.d.ts +4 -77
- package/data/index.js +1 -1
- package/index.cjs +21 -20
- package/index.cjs.map +1 -1
- package/index.d.cts +5 -3
- package/index.d.ts +5 -3
- package/index.js +21 -20
- package/index.js.map +1 -1
- package/matrix-BULDDGYa.d.ts +122 -0
- package/matrix-BnXitNJ7.d.cts +122 -0
- package/package.json +2 -2
- package/std/index.cjs +1 -1
- package/std/index.cjs.map +1 -1
- package/std/index.d.cts +104 -17
- package/std/index.d.ts +104 -17
- package/std/index.js +1 -1
- package/std/index.js.map +1 -1
- package/{tgpuComputeFn-DH8_PcIR.d.cts → tgpuComputeFn-BxPDI5hQ.d.cts} +178 -95
- package/{tgpuComputeFn-DH8_PcIR.d.ts → tgpuComputeFn-BxPDI5hQ.d.ts} +178 -95
- package/chunk-5RYM4COI.js +0 -4
- package/chunk-5RYM4COI.js.map +0 -1
- package/chunk-FHOQ6WZE.cjs +0 -2
- package/chunk-FHOQ6WZE.cjs.map +0 -1
- package/chunk-ODKBWWHU.cjs +0 -4
- package/chunk-ODKBWWHU.cjs.map +0 -1
- package/chunk-SMTSXYNG.js +0 -2
- package/chunk-SMTSXYNG.js.map +0 -1
package/index.js
CHANGED
@@ -1,30 +1,31 @@
|
|
1
|
-
import{
|
1
|
+
import{S as V,T as Q,V as S,X as st,_ as tr,a as L,aa as mt,b as Me,ba as Ue,ca as cn}from"./chunk-734H4JWF.js";import{c as fn}from"./chunk-CGVYLGBG.js";import{A as de,Aa as G,B as at,Ba as k,C as rn,Ca as er,D as nn,Da as dn,E as Y,F as an,G as on,H as sn,I as fe,K as be,L as z,M as A,Ma as ut,N as Ht,Na as pt,O as un,Oa as lt,U as ot,V as it,W as Se,X as ve,Y as q,Z as pn,a as d,aa as ee,b,ba as O,c as I,ca as $,d as v,da as h,e as p,ea as ln,f as w,fa as mn,g as j,ga as Jt,h as Jr,ha as De,i as Yr,ia as Yt,j as le,ja as E,k as H,ka as R,l as Nt,la as C,m as jt,ma as f,n as zt,na as ke,o as me,oa as W,p as Ee,pa as Le,q as Ge,qa as te,r as Xr,ra as re,s as Z,sa as Xt,t as qt,ta as ce,u as Qt,ua as Oe,v as J,va as Fe,w as Zr,wa as Ie,x as en,xa as Zt,y as Kt,ya as P,z as tn,za as We}from"./chunk-SWJVYRFM.js";function gn(e){if(typeof e?.[I]=="function")return e[I].bind(e)}function dt(e,t){throw new Error(`Failed to handle ${e} at ${t}`)}var ba={f:{1:f,2:W,3:ce,4:P},h:{1:ke,2:Le,3:Oe,4:We},i:{1:C,2:te,3:Fe,4:G},u:{1:R,2:re,3:Ie,4:k},b:{1:E,2:Xt,3:Zt,4:er}},Sa={vec2f:W,vec2h:Le,vec2i:te,vec2u:re,"vec2<bool>":Xt,vec3f:ce,vec3h:Oe,vec3i:Fe,vec3u:Ie,"vec3<bool>":Zt,vec4f:P,vec4h:We,vec4i:G,vec4u:k,"vec4<bool>":er,mat2x2f:ut,mat3x3f:pt,mat4x4f:lt},yn={vec2f:f,vec2h:ke,vec2i:C,vec2u:R,"vec2<bool>":E,vec3f:f,vec3h:ke,vec3i:C,vec3u:R,"vec3<bool>":E,vec4f:f,vec4h:ke,vec4i:C,vec4u:R,"vec4<bool>":E,mat2x2f:W,mat3x3f:ce,mat4x4f:P};function Ne(e,t){if(A(e)||q(e))return e.propTypes[t]??$;if(e===E||fn(e))return $;let r=t.length;if(Y(e)&&r>=1&&r<=4){let n=e.type.includes("bool")?"b":e.type[4],a=ba[n][r];if(a)return a}return $}function rr(e){return z(e)||ve(e)?e.elementType:e.type in yn?yn[e.type]:$}function nr(e){if(/^0x[0-9a-f]+$/i.test(e))return h(e,De);if(/^0b[01]+$/i.test(e))return h(`${Number.parseInt(e.slice(2),2)}`,De);if(/^-?(?:\d+\.\d*|\d*\.\d+)$/i.test(e))return h(e,Yt);if(/^-?\d+(?:\.\d+)?e-?\d+$/i.test(e))return h(e,Yt);if(/^-?\d+$/i.test(e))return h(e,De)}var ar={rank:Number.POSITIVE_INFINITY,action:"none"};function Ae(e){return e.type==="decorated"?e.inner:e}function Tn(e){return Y(e)?dn[e.type]:void 0}function ft(e,t){let r=Ae(e),n=Ae(t);if(r.type===n.type)return{rank:0,action:"none"};if(r.type==="abstractFloat"){if(n.type==="f32")return{rank:1,action:"none"};if(n.type==="f16")return{rank:2,action:"none"}}if(r.type==="abstractInt"){if(n.type==="i32")return{rank:3,action:"none"};if(n.type==="u32")return{rank:4,action:"none"};if(n.type==="abstractFloat")return{rank:5,action:"none"};if(n.type==="f32")return{rank:6,action:"none"};if(n.type==="f16")return{rank:7,action:"none"}}if(Y(r)&&Y(n)){let a=Tn(r),o=Tn(n);if(a&&o)return ft(a,o)}return fe(r)&&fe(n)?{rank:0,action:"none"}:ar}function va(e,t){let r=Ae(e),n=Ae(t);if(r.type==="ptr"&&ft(r.inner,n).rank<Number.POSITIVE_INFINITY)return{rank:0,action:"deref"};if(n.type==="ptr"&&ft(r,n.inner).rank<Number.POSITIVE_INFINITY)return{rank:1,action:"ref"};let a={f32:0,f16:1,i32:2,u32:3,bool:4};if(r.type in a&&n.type in a){let o=r.type,i=n.type;if(o!==i){let s=a[o];return{rank:a[i]<s?10:20,action:"cast",targetType:n}}}return ar}function Da(e,t,r){let n=ft(e,t);return n.rank<Number.POSITIVE_INFINITY?n:r?va(e,t):ar}function xn(e,t,r){let n,a=Number.POSITIVE_INFINITY,o=new Map;for(let u of t){let c=0,m=[],g=!0;for(let T of e){let x=Da(T,u,r);if(x.rank===Number.POSITIVE_INFINITY){g=!1;break}c+=x.rank,m.push(x)}g&&c<a&&(a=c,n=u,o.set(n,m))}if(!n)return;let s=o.get(n).map((u,c)=>({sourceIndex:c,action:u.action,...u.action==="cast"&&{targetType:u.targetType}})),l=s.some(u=>u.action==="cast");return{targetType:n,actions:s,hasImplicitConversions:l}}function or(e){return e.type==="abstractFloat"?f:e.type==="abstractInt"?C:e}function hn(e,t){if(e.length===0)return;let r=[...new Set(e.map(Ae))],n=t?[...new Set(t.map(Ae))]:r,a=xn(e,n,!1);if(a)return a;let o=xn(e,n,!0);if(o)return o.hasImplicitConversions=o.actions.some(i=>i.action==="cast"),o}function Fa(e,t,r,n){if(r.action==="none")return h(t.value,n);let a=e.resolve(t.value);switch(r.action){case"ref":return h(`&${a}`,n);case"deref":return h(`*${a}`,n);case"cast":return h(`${e.resolve(n)}(${a})`,n);default:dt(r.action,"applyActionToSnippet")}}function M(e,t,r){let n=t.map(o=>o.dataType);if(n.some(o=>o===$))return;let a=hn(n,r);if(a)return a.hasImplicitConversions&&console.warn(`Implicit conversions from [
|
2
2
|
${t.map(o=>` ${o.value}: ${o.dataType.type}`).join(`,
|
3
3
|
`)}
|
4
4
|
] to ${a.targetType.type} are supported, but not recommended.
|
5
|
-
Consider using explicit conversions instead.`),t.map((o,
|
6
|
-
|
5
|
+
Consider using explicit conversions instead.`),t.map((o,i)=>{let s=a.actions[i];return le(s,"Action should not be undefined"),Fa(e,o,s,a.targetType)})}function ir(e,t,r){return Object.keys(t.propTypes).map(a=>{let o=r[a];if(!o)throw new Error(`Missing property ${a}`);let i=t.propTypes[a];return M(e,[o],[i])?.[0]??o})}function ge(e){if(ln(e))return e;if(tn(e))return h(e,e[b]);if(at(e)||an(e))return h(e,Sa[e.kind]);if(Array.isArray(e)){let t=e.map(ge).filter(Boolean),r=Z();if(!r)throw new Error("Tried to coerce array without a context");let n=M(r,t),a=hn(t.map(o=>o.dataType))?.targetType;return!n||!a?h(e,$):h(n.map(o=>o.value).join(", "),st(or(a),e.length))}return typeof e=="string"||typeof e=="function"||typeof e=="object"||typeof e=="symbol"||typeof e>"u"||e===null?h(e,$):typeof e=="number"||typeof e=="bigint"?h(e,nr(String(e))?.dataType??$):typeof e=="boolean"?h(e,E):h(e,$)}var K={get(e,t){if(t in e)return Reflect.get(e,t);if(t!=="~providing")return t==="toString"||t===Symbol.toStringTag||t===Symbol.toPrimitive?()=>e.toString():new Proxy({"~resolve":r=>`${r.resolve(e)}.${String(t)}`,toString:()=>`.value(...).${String(t)}:${p(e)??"<unnamed>"}`,[b]:Ne(e[b],String(t))},K)}};function je(e,t){let r=t,n;for(;n=gn(r);)r=n(e);return r}function wn(e,t){return new sr(e,t)}var sr=class{constructor(t,r){this.dataType=t;this._value=r;this[d]={dataType:t}}[d];$name(t){return w(this,t),this}"~resolve"(t){let r=t.names.makeUnique(p(this)),n=t.resolveValue(this._value,this.dataType);return t.addDeclaration(`const ${r} = ${n};`),r}toString(){return`const:${p(this)??"<unnamed>"}`}[I](){return new Proxy({"~resolve":t=>t.resolve(this),toString:()=>`.value:${p(this)??"<unnamed>"}`,[b]:this.dataType},K)}get value(){return J()?this[I]():this._value}};function Pe(e){return e?.resourceType==="slot"}function Be(e){return e?.resourceType==="derived"}function ze(e){return e?.["~providing"]!==void 0}function qe(e){return e?.resourceType==="accessor"}function ur(e){return typeof e?.["~resolve"]=="function"}function ct(e){return typeof e=="number"||typeof e=="boolean"||typeof e=="string"||ur(e)||be(e)||Pe(e)||Be(e)||ze(e)}function bn(e){return!!e&&typeof e=="object"&&"getMappedRange"in e&&"mapAsync"in e}function Sn(e){return e?.resourceType==="buffer-usage"}function pr(e){return!!e?.[d]}function ye(e,t){for(let[r,n]of Object.entries(t))e[r]=n,j(n)&&p(n)===void 0&&n.$name(r)}function vn(e,t,r){let n=[...e.matchAll(/:\s*(?<arg>.*?)\s*[,)]/g)].map(a=>a?a[1]:void 0);r(Object.fromEntries(t.flatMap((a,o)=>{let i=n?n[o]:void 0;return A(a)&&i!==void 0?[[i,a]]:[]})))}function Re(e,t,r){let n=e.match(/->\s(?<output>[\w\d_]+)\s{/),a=n?n[1]?.trim():void 0;A(t)&&a&&!/\s/g.test(a)&&r({[a]:t})}function Ia(e){return new RegExp(`(?<![\\w\\$_.])${e.replaceAll(".","\\.").replaceAll("$","\\$")}(?![\\w\\$_])`,"g")}function Te(e,t,r){return Object.entries(t).reduce((n,[a,o])=>ct(o)||Se(o)?n.replaceAll(Ia(a),e.resolve(o)):o!==null&&typeof o=="object"?([...r.matchAll(new RegExp(`${a.replaceAll(".","\\.").replaceAll("$","\\$")}\\.(?<prop>.*?)(?![\\w\\$_])`,"g"))].map(s=>s[1])??[]).reduce((s,l)=>l&&l in o?Te(e,{[`${a}.${l}`]:o[l]},s):s,n):n,r)}function Dn(e){return new lr(e)}var lr=class{constructor(t){this.declaration=t}externalsToApply=[];$uses(t){return this.externalsToApply.push(t),this}"~resolve"(t){let r={};for(let a of this.externalsToApply)ye(r,a);let n=Te(t,r,this.declaration);return t.addDeclaration(n),""}toString(){return`declare: ${this.declaration}`}};function Fn(e,t,r){return Yr(e,{ast:t,externals:r}),e}function In(e){return()=>{throw new Error(`The function "${e??"<unnamed>"}" is invokable only on the GPU. If you want to use it on the CPU, mark it with the "kernel & js" directive.`)}}import{FuncParameterType as Rn}from"tinyest";function Pn(e){let{strippedCode:t,argRange:r}=Ua(e),n=new gt(t);n.consume("(");let a=[];for(;!n.isAt(")");){let i=[];for(;n.isAt("@");)n.parseUntil(Un,An),n.consume(")"),i.push(n.lastParsed);n.parseUntil(Pa);let s=n.lastParsed,l;n.isAt(":")&&(n.consume(":"),n.parseUntil(Ba,Ca),l=n.lastParsed),a.push({identifier:s,attributes:i,type:l}),n.isAt(",")&&n.consume(",")}n.consume(")");let o;if(n.isAt("->")){n.consume("->");let i=[];for(;n.isAt("@");)n.parseUntil(Un,An),n.consume(")"),i.push(n.lastParsed);o={type:n.str.slice(n.pos),attributes:i}}return{args:a,ret:o,range:{begin:r[0],end:r[1]}}}function Ua(e){let t=new gt(e),r="",n;for(;!t.isFinished();){if(t.isAt(Aa)){t.advanceBy(1);continue}if(t.isAt("//")){t.consume("//"),t.parseUntil(Bn),t.advanceBy(1);continue}if(t.isAt("/*")){t.parseUntil(Ra,Va),t.consume("*/");continue}if(t.isAt("{"))return{strippedCode:r,argRange:[n,t.pos]};t.isAt("(")&&n===void 0&&(n=t.pos),n!==void 0&&(r+=t.str[t.pos]),t.advanceBy(1)}throw new Error("Invalid wgsl code!")}var gt=class{constructor(t){this.str=t;this.#e=0}#t;#e;get pos(){return this.#e}get lastParsed(){if(this.#t===void 0)throw new Error("Parse was not called yet!");return this.str.slice(this.#t,this.pos)}isFinished(){return this.#e>=this.str.length}isAt(t){if(typeof t=="string"){for(let r=0;r<t.length;r++)if(this.str[this.#e+r]!==t[r])return!1;return!0}for(let r of t)if(this.isAt(r))return!0;return!1}parseUntil(t,r){this.#t=this.#e;let n=0;for(;this.#e<this.str.length;){if(r&&this.isAt(r[0])&&(n+=1),r&&this.isAt(r[1])&&(n-=1),n===0&&this.isAt(t))return this.#e;this.#e+=1}throw new Error("Reached the end of the string without finding a match!")}advanceBy(t){this.#e+=t}consume(t){if(!this.isAt(t))throw new Error(`Expected '${t}' at position ${this.#e}, but found '${this.str.slice(this.#e,this.#e+t.length)}'`);this.advanceBy(t.length)}},Bn=new Set([`
|
6
|
+
`,"\v","\f","\r","\x85","\u2028","\u2029"]),Aa=new Set([...Bn," "," ","\u200E","\u200F"]),Un=new Set([")"]),Pa=new Set([":",",",")"]),Ba=new Set([",",")"]),Ra=new Set(["*/"]),An=["(",")"],Ca=["<",">"],Va=["/*","*/"];function ne(e,t){let r=[];typeof t=="string"&&(e.isEntry?(A(e.argTypes[0])&&r.push({In:e.argTypes[0]}),A(e.returnType)&&r.push({Out:e.returnType})):(vn(t,e.argTypes,o=>r.push(o)),Re(t,e.returnType,o=>r.push(o))));let n={applyExternals(o){r.push(o)},resolve(o,i=""){let s={};for(let u of r)ye(s,u);let l=o.names.makeUnique(p(this));if(typeof t=="string"){let u=Te(o,s,t),c="",m="";if(e.isEntry){let g=A(e.argTypes[0])?`(in: ${o.resolve(e.argTypes[0])})`:"()",T=be(e.returnType)?Ue(e.returnType):"",x=e.returnType!==de?A(e.returnType)?`-> ${o.resolve(e.returnType)}`:`-> ${T!==""?T:"@location(0)"} ${o.resolve(e.returnType)}`:"";c=`${g} ${x} `,m=u}else{let g=Pn(u);if(g.args.length!==e.argTypes.length)throw new Error(`WGSL implementation has ${g.args.length} arguments, while the shell has ${e.argTypes.length} arguments.`);let T=g.args.map((y,F)=>`${y.identifier}: ${Cn(o,`parameter ${y.identifier}`,y.type,e.argTypes[F])}`).join(", "),x=e.returnType===de?"":`-> ${Cn(o,"return type",g.ret?.type,e.returnType)}`;c=`(${T}) ${x}`,m=u.slice(g.range.end)}o.addDeclaration(`${i}fn ${l}${c}${m}`)}else{let u=Jr(t);if(u?.externals){let x=Object.fromEntries(Object.entries(u.externals).filter(([y])=>!(y in s)));ye(s,x)}let c=u?.ast??o.transpileFn(String(t)),m=c.externalNames.filter(x=>!(x in s));if(m.length>0)throw new zt(p(this),m);let{head:g,body:T}=o.fnToWgsl({args:e.argTypes.map((x,y)=>h(c.params[y]?.type===Rn.identifier?c.params[y].name:`_arg_${y}`,x)),argAliases:Object.fromEntries(c.params.flatMap((x,y)=>x.type===Rn.destructuredObject?x.props.map(({name:F,alias:U})=>[U,h(`_arg_${y}.${F}`,e.argTypes[y].propTypes[F])]):[])),returnType:e.returnType,body:c.body,externalMap:s});o.addDeclaration(`${i}fn ${l}${o.resolve(g)}${o.resolve(T)}`)}return l}},a=p(t);return a!==void 0&&w(n,a),n}function Cn(e,t,r,n){let a=e.resolve(n).replace(/\s/g,"");if(!r)return a;let o=r.replace(/\s/g,"");if(o!==a)throw new Error(`Type mismatch between TGPU shell and WGSL code string: ${t}, JS type "${a}", WGSL type "${o}".`);return r}function _a(e){let t=0;return Object.fromEntries(Object.entries(e).map(([r,n])=>{if(mt(n))return[r,n];let a=ee(n);return a!==void 0?(t=a+1,[r,n]):[r,tr(t++,n)]}))}function ae(e){return O(e)?it(e)||ee(e)!==void 0?e:tr(0,e):cn(_a(e))}function oe(e,...t){return $a(e)?Ea(e,...t):e}function $a(e){return Array.isArray(e)&&"raw"in e&&Array.isArray(e.raw)&&e.raw.every(t=>typeof t=="string")}function Ea(e,...t){return e.slice(1).reduce((r,n,a)=>`${r}${t[a]}${n}`,e[0])}function Vn(e){let t={argTypes:e.in&&Object.keys(e.in).length!==0?[ae(e.in)]:[],returnType:de,workgroupSize:[e.workgroupSize[0]??1,e.workgroupSize[1]??1,e.workgroupSize[2]??1],isEntry:!0},r=(n,...a)=>Ga(t,e.workgroupSize,oe(n,...a));return Object.assign(Object.assign(r,t),{does:r})}function Ga(e,t,r){let n=ne(e,r),a=e.argTypes[0];return{shell:e,$uses(i){return n.applyExternals(i),this},[v]:n,$name(i){return w(n,i),j(a)&&a.$name(`${i}_Input`),this},"~resolve"(i){return n.resolve(i,`@compute @workgroup_size(${t.join(", ")}) `)},toString(){return`computeFn:${p(n)??"<unnamed>"}`}}}function _n(e,t){let r={[d]:!0,argTypes:e,returnType:t??de,isEntry:!1},n=(a,...o)=>La(r,oe(a,...o));return Object.assign(Object.assign(n,r),{does:n})}function mr(e){return!!e?.[d]&&e?.resourceType==="function"}function ka([e,t]){return`${p(e)??"<unnamed>"}=${t}`}function La(e,t){let r=ne(e,t),n={[d]:{implementation:t,argTypes:e.argTypes},shell:e,resourceType:"function",$uses(i){return r.applyExternals(i),this},[v]:r,$name(i){return w(r,i),this},with(i,s){return $n(o,[[qe(i)?i.slot:i,s]])},"~resolve"(i){if(typeof t=="string")return r.resolve(i);let s=i;if(s.callStack===void 0)throw new Error("Cannot resolve a TGSL function outside of a generation context");try{return s.callStack.push(e.returnType),r.resolve(i)}finally{s.callStack.pop()}}},a=Jt((...i)=>{if(typeof t=="string")throw new Error("Cannot execute on the CPU functions constructed with raw WGSL");return t(...i)},(...i)=>h(new yt(o,i.map(s=>s.value)),e.returnType??$),"tgpuFnCall",e.argTypes);a[d].implementation=t;let o=Object.assign(a,n);return Object.defineProperty(o,"toString",{value(){return`fn:${p(r)??"<unnamed>"}`}}),o}function $n(e,t){let r={[d]:{implementation:e[d].implementation,argTypes:e[d].argTypes},resourceType:"function",shell:e.shell,"~providing":{inner:e,pairs:t},$uses(o){return e.$uses(o),this},[v]:e,$name(o){return e.$name(o),this},with(o,i){return $n(a,[...t,[qe(o)?o.slot:o,i]])}},n=Jt((...o)=>e(...o),(...o)=>h(new yt(a,o.map(i=>i.value)),e.shell.returnType??$),"tgpuFnCall",e.shell.argTypes),a=Object.assign(n,r);return Object.defineProperty(a,"toString",{value(){return`fn:${p(e)??"<unnamed>"}[${t.map(ka).join(", ")}]`}}),a[d].implementation=e[d].implementation,a}var yt=class{constructor(t,r){this._fn=t;this._params=r;this[v]=t}[v];"~resolve"(t){return t.resolve(`${t.resolve(this._fn)}(${this._params.map(r=>t.resolve(r)).join(", ")})`)}toString(){return`call:${p(this)??"<unnamed>"}`}};function En(e){let t={argTypes:e.in&&Object.keys(e.in).length!==0?[ae(e.in)]:[],targets:e.out,returnType:ae(e.out),isEntry:!0},r=(n,...a)=>Oa(t,oe(n,...a));return Object.assign(Object.assign(r,t),{does:r})}function Oa(e,t){let r=ne(e,t),n=e.returnType,a=e.argTypes[0];return typeof t=="string"&&Re(t,n,i=>r.applyExternals(i)),{shell:e,outputType:n,$uses(i){return r.applyExternals(i),this},[v]:r,$name(i){return w(r,i),j(n)&&n.$name(`${i}_Output`),j(a)&&a.$name(`${i}_Input`),this},"~resolve"(i){if(typeof t=="string")return r.resolve(i,"@fragment ");let s=i;if(s.callStack===void 0)throw new Error("Cannot resolve a TGSL function outside of a generation context");try{return s.callStack.push(n),r.resolve(i,"@fragment ")}finally{s.callStack.pop()}},toString(){return`fragmentFn:${p(r)??"<unnamed>"}`}}}function Gn(e){if(Object.keys(e.out).length===0)throw new Error("A vertexFn output cannot be empty since it must include the 'position' builtin.");let t={attributes:[e.in??{}],returnType:ae(e.out),argTypes:e.in&&Object.keys(e.in).length!==0?[ae(e.in)]:[],isEntry:!0},r=(n,...a)=>Wa(t,oe(n,...a));return Object.assign(Object.assign(r,t),{does:r})}function Wa(e,t){let r=ne(e,t),n=e.returnType,a=e.argTypes[0];return typeof t=="string"&&Re(t,n,i=>r.applyExternals(i)),{shell:e,outputType:n,inputType:a,$uses(i){return r.applyExternals(i),this},[v]:r,$name(i){return w(r,i),j(n)&&n.$name(`${i}_Output`),j(a)&&a.$name(`${i}_Input`),this},"~resolve"(i){if(typeof t=="string")return r.resolve(i,"@vertex ");let s=i;if(s.callStack===void 0)throw new Error("Cannot resolve a TGSL function outside of a generation context");try{return s.callStack.push(n),r.resolve(i,"@vertex ")}finally{s.callStack.pop()}},toString(){return`vertexFn:${p(r)??"<unnamed>"}`}}}var ie=class{lastUniqueId=0;makeUnique(t){let r;return t?(r=t.replaceAll(/\s/g,"_"),r=r.replaceAll(/[^\w\d]/g,"")):r="item",`${r}_${this.lastUniqueId++}`}},se=class{_usedNames=new Set;makeUnique(t){if(t===void 0)throw new Error("Unnamed item found when using a strict name registry");let r=0,n=t;for(;this._usedNames.has(n);)r++,n=`${t}_${r}`;return this._usedNames.add(n),n}};function Qe(e){return typeof e?.format=="string"}function kn(e,t){let r=[];if(O(e)){if(!Qe(t))throw new Error("Shader expected a single attribute, not a record of attributes to be passed in.");return r.push(t._layout),{usedVertexLayouts:r,bufferDefinitions:[{arrayStride:t._layout.stride,stepMode:t._layout.stepMode,attributes:[{format:t.format,offset:t.offset,shaderLocation:ee(e)??0}]}]}}let n=[],a=new WeakMap,o=0;for(let[i,s]of Object.entries(e)){if(mt(s))continue;let l=t[i];if(!l)throw new Error(`An attribute by the name of '${i}' was not provided to the shader.`);let u=l._layout,c=a.get(u);c||(r.push(u),c=[],n.push({arrayStride:u.stride,stepMode:u.stepMode,attributes:c}),a.set(u,c)),o=ee(s)??o,c.push({format:l.format,offset:l.offset,shaderLocation:o++})}return{usedVertexLayouts:r,bufferDefinitions:n}}var Ma=["bool","f32","f16","i32","u32","vec2f","vec3f","vec4f","vec2h","vec3h","vec4h","vec2i","vec3i","vec4i","vec2u","vec3u","vec4u","vec2<bool>","vec3<bool>","vec4<bool>","mat2x2f","mat3x3f","mat4x4f"];function Na(e){return Ma.includes(e.type)}function dr(e,[t,r]){return` ${Ue(r)}${t}: ${e.resolve(r)},
|
7
|
+
`}function ja(e,t){let r=e.names.makeUnique(p(t));return e.addDeclaration(`
|
7
8
|
struct ${r} {
|
8
|
-
${Object.entries(t.propTypes).map(n=>
|
9
|
-
`),r}function
|
9
|
+
${Object.entries(t.propTypes).map(n=>dr(e,n)).join("")}}
|
10
|
+
`),r}function za(e,t){let r=e.names.makeUnique(p(t));return e.addDeclaration(`
|
10
11
|
struct ${r} {
|
11
|
-
${Object.entries(t.propTypes).map(n=>
|
12
|
+
${Object.entries(t.propTypes).map(n=>Qe(n[1])?dr(e,[n[0],Me[n[1].format]]):dr(e,n)).join("")}
|
12
13
|
}
|
13
|
-
`),r}function
|
14
|
-
`,
|
15
|
-
`,
|
16
|
-
`;return o}if(
|
17
|
-
`}return u}let n=
|
18
|
-
`}function Tt(e){if(pr.has(e))return pr.get(e);let t=ct(e,"offset","value"),r=new Function("output","offset","value","littleEndian=true",t);return pr.set(e,r),r}var Re={bool(){throw new Error("Booleans are not host-shareable")},f32(e,t,r){e.writeFloat32(r)},f16(e,t,r){e.writeFloat16(r)},i32(e,t,r){e.writeInt32(r)},u32(e,t,r){e.writeUint32(r)},vec2f(e,t,r){e.writeFloat32(r.x),e.writeFloat32(r.y)},vec2h(e,t,r){e.writeFloat16(r.x),e.writeFloat16(r.y)},vec2i(e,t,r){e.writeInt32(r.x),e.writeInt32(r.y)},vec2u(e,t,r){e.writeUint32(r.x),e.writeUint32(r.y)},"vec2<bool>"(){throw new Error("Booleans are not host-shareable")},vec3f(e,t,r){e.writeFloat32(r.x),e.writeFloat32(r.y),e.writeFloat32(r.z)},vec3h(e,t,r){e.writeFloat16(r.x),e.writeFloat16(r.y),e.writeFloat16(r.z)},vec3i(e,t,r){e.writeInt32(r.x),e.writeInt32(r.y),e.writeInt32(r.z)},vec3u(e,t,r){e.writeUint32(r.x),e.writeUint32(r.y),e.writeUint32(r.z)},"vec3<bool>"(){throw new Error("Booleans are not host-shareable")},vec4f(e,t,r){e.writeFloat32(r.x),e.writeFloat32(r.y),e.writeFloat32(r.z),e.writeFloat32(r.w)},vec4h(e,t,r){e.writeFloat16(r.x),e.writeFloat16(r.y),e.writeFloat16(r.z),e.writeFloat16(r.w)},vec4i(e,t,r){e.writeInt32(r.x),e.writeInt32(r.y),e.writeInt32(r.z),e.writeInt32(r.w)},vec4u(e,t,r){e.writeUint32(r.x),e.writeUint32(r.y),e.writeUint32(r.z),e.writeUint32(r.w)},"vec4<bool>"(){throw new Error("Booleans are not host-shareable")},mat2x2f(e,t,r){for(let n=0;n<r.length;++n)e.writeFloat32(r[n])},mat3x3f(e,t,r){for(let n=0;n<r.length;++n)e.writeFloat32(r[n])},mat4x4f(e,t,r){for(let n=0;n<r.length;++n)e.writeFloat32(r[n])},struct(e,t,r){let n=P(t);B(e,n);for(let[a,o]of Object.entries(t.propTypes))B(e,P(o)),ne(e,o,r[a]);B(e,n)},array(e,t,r){if(t.elementCount===0)throw new Error("Cannot write using a runtime-sized schema.");let n=P(t);B(e,n);let a=e.currentByteOffset;for(let o=0;o<Math.min(t.elementCount,r.length);o++)B(e,n),ne(e,t.elementType,r[o]);e.seekTo(a+S(t))},ptr(){throw new Error("Pointers are not host-shareable")},atomic(e,t,r){Re[t.inner.type]?.(e,t,r)},decorated(e,t,r){let n=K(t);B(e,n);let a=e.currentByteOffset;Re[t.inner?.type]?.(e,t.inner,r),e.seekTo(a+S(t))},uint8(e,t,r){e.writeUint8(r)},uint8x2(e,t,r){e.writeUint8(r.x),e.writeUint8(r.y)},uint8x4(e,t,r){e.writeUint8(r.x),e.writeUint8(r.y),e.writeUint8(r.z),e.writeUint8(r.w)},sint8(e,t,r){e.writeInt8(r)},sint8x2(e,t,r){e.writeInt8(r.x),e.writeInt8(r.y)},sint8x4(e,t,r){e.writeInt8(r.x),e.writeInt8(r.y),e.writeInt8(r.z),e.writeInt8(r.w)},unorm8(e,t,r){e.writeUint8(r*255)},unorm8x2(e,t,r){e.writeUint8(r.x*255),e.writeUint8(r.y*255)},unorm8x4(e,t,r){e.writeUint8(r.x*255),e.writeUint8(r.y*255),e.writeUint8(r.z*255),e.writeUint8(r.w*255)},snorm8(e,t,r){e.writeUint8(r*127+128)},snorm8x2(e,t,r){e.writeUint8(r.x*127+128),e.writeUint8(r.y*127+128)},snorm8x4(e,t,r){e.writeUint8(r.x*127+128),e.writeUint8(r.y*127+128),e.writeUint8(r.z*127+128),e.writeUint8(r.w*127+128)},uint16(e,t,r){e.writeUint16(r)},uint16x2(e,t,r){e.writeUint16(r.x),e.writeUint16(r.y)},uint16x4(e,t,r){e.writeUint16(r.x),e.writeUint16(r.y),e.writeUint16(r.z),e.writeUint16(r.w)},sint16(e,t,r){e.writeInt16(r)},sint16x2(e,t,r){e.writeInt16(r.x),e.writeInt16(r.y)},sint16x4(e,t,r){e.writeInt16(r.x),e.writeInt16(r.y),e.writeInt16(r.z),e.writeInt16(r.w)},unorm16(e,t,r){e.writeUint16(r*65535)},unorm16x2(e,t,r){e.writeUint16(r.x*65535),e.writeUint16(r.y*65535)},unorm16x4(e,t,r){e.writeUint16(r.x*65535),e.writeUint16(r.y*65535),e.writeUint16(r.z*65535),e.writeUint16(r.w*65535)},snorm16(e,t,r){e.writeUint16(r*32767+32768)},snorm16x2(e,t,r){e.writeUint16(r.x*32767+32768),e.writeUint16(r.y*32767+32768)},snorm16x4(e,t,r){e.writeUint16(r.x*32767+32768),e.writeUint16(r.y*32767+32768),e.writeUint16(r.z*32767+32768),e.writeUint16(r.w*32767+32768)},float16(e,t,r){e.writeFloat16(r)},float16x2(e,t,r){e.writeFloat16(r.x),e.writeFloat16(r.y)},float16x4(e,t,r){e.writeFloat16(r.x),e.writeFloat16(r.y),e.writeFloat16(r.z),e.writeFloat16(r.w)},float32(e,t,r){e.writeFloat32(r)},float32x2(e,t,r){e.writeFloat32(r.x),e.writeFloat32(r.y)},float32x3(e,t,r){e.writeFloat32(r.x),e.writeFloat32(r.y),e.writeFloat32(r.z)},float32x4(e,t,r){e.writeFloat32(r.x),e.writeFloat32(r.y),e.writeFloat32(r.z),e.writeFloat32(r.w)},uint32(e,t,r){e.writeUint32(r)},uint32x2(e,t,r){e.writeUint32(r.x),e.writeUint32(r.y)},uint32x3(e,t,r){e.writeUint32(r.x),e.writeUint32(r.y),e.writeUint32(r.z)},uint32x4(e,t,r){e.writeUint32(r.x),e.writeUint32(r.y),e.writeUint32(r.z),e.writeUint32(r.w)},sint32(e,t,r){e.writeInt32(r)},sint32x2(e,t,r){e.writeInt32(r.x),e.writeInt32(r.y)},sint32x3(e,t,r){e.writeInt32(r.x),e.writeInt32(r.y),e.writeInt32(r.z)},sint32x4(e,t,r){e.writeInt32(r.x),e.writeInt32(r.y),e.writeInt32(r.z),e.writeInt32(r.w)},"unorm10-10-10-2"(e,t,r){let n=0;n|=(r.x*1023&1023)<<22,n|=(r.x*1023&1023)<<12,n|=(r.y*1023&1023)<<2,n|=r.z*3&3,e.writeUint32(n)},"unorm8x4-bgra"(e,t,r){e.writeUint8(r.z*255),e.writeUint8(r.y*255),e.writeUint8(r.x*255),e.writeUint8(r.w*255)},disarray(e,t,r){let n=P(t);B(e,n);let a=e.currentByteOffset;for(let o=0;o<Math.min(t.elementCount,r.length);o++)B(e,n),Re[t.elementType?.type]?.(e,t.elementType,r[o]);e.seekTo(a+S(t))},unstruct(e,t,r){for(let[n,a]of Object.entries(t.propTypes))Re[a.type]?.(e,a,r[n])},"loose-decorated"(e,t,r){let n=K(t);B(e,n);let a=e.currentByteOffset,o=Re[t.inner?.type];return o?.(e,t.inner,r),e.seekTo(a+S(t)),r}};function ne(e,t,r){let n=Re[t.type];if(!n)throw new Error(`Cannot write data of type '${t.type}'.`);n(e,t,r)}var Te={bool(){throw new Error("Booleans are not host-shareable")},f32(e){return e.readFloat32()},f16(e){return e.readFloat16()},i32(e){return e.readInt32()},u32(e){return e.readUint32()},vec2f(e){return O(e.readFloat32(),e.readFloat32())},vec3f(e){return fe(e.readFloat32(),e.readFloat32(),e.readFloat32())},vec4f(e){return F(e.readFloat32(),e.readFloat32(),e.readFloat32(),e.readFloat32())},vec2h(e){return Oe(e.readFloat16(),e.readFloat16())},vec3h(e){return Ge(e.readFloat16(),e.readFloat16(),e.readFloat16())},vec4h(e){return We(e.readFloat16(),e.readFloat16(),e.readFloat16(),e.readFloat16())},vec2i(e){return te(e.readInt32(),e.readInt32())},vec3i(e){return Fe(e.readInt32(),e.readInt32(),e.readInt32())},vec4i(e){return k(e.readInt32(),e.readInt32(),e.readInt32(),e.readInt32())},vec2u(e){return re(e.readUint32(),e.readUint32())},vec3u(e){return Ae(e.readUint32(),e.readUint32(),e.readUint32())},vec4u(e){return E(e.readUint32(),e.readUint32(),e.readUint32(),e.readUint32())},"vec2<bool>"(){throw new Error("Booleans are not host-shareable")},"vec3<bool>"(){throw new Error("Booleans are not host-shareable")},"vec4<bool>"(){throw new Error("Booleans are not host-shareable")},mat2x2f(e){return ot(e.readFloat32(),e.readFloat32(),e.readFloat32(),e.readFloat32())},mat3x3f(e){let t=()=>{let r=e.readFloat32();return e.readFloat32(),r};return st(e.readFloat32(),e.readFloat32(),t(),e.readFloat32(),e.readFloat32(),t(),e.readFloat32(),e.readFloat32(),t())},mat4x4f(e){return it(e.readFloat32(),e.readFloat32(),e.readFloat32(),e.readFloat32(),e.readFloat32(),e.readFloat32(),e.readFloat32(),e.readFloat32(),e.readFloat32(),e.readFloat32(),e.readFloat32(),e.readFloat32(),e.readFloat32(),e.readFloat32(),e.readFloat32(),e.readFloat32())},struct(e,t){let r=P(t);B(e,r);let n={};for(let[a,o]of Object.entries(t.propTypes))B(e,P(o)),n[a]=M(e,o);return B(e,r),n},array(e,t){if(t.elementCount===0)throw new Error("Cannot read using a runtime-sized schema.");let r=P(t),n=[];for(let a=0;a<t.elementCount;a++){B(e,r);let o=t.elementType,s=M(e,o);n.push(s)}return B(e,r),n},ptr(){throw new Error("Pointers are not host-shareable")},atomic(e,t){return M(e,t.inner)},decorated(e,t){let r=K(t);B(e,r);let n=e.currentByteOffset,a=M(e,t.inner);return e.seekTo(n+S(t)),a},uint8:e=>e.readUint8(),uint8x2:e=>re(e.readUint8(),e.readUint8()),uint8x4:e=>E(e.readUint8(),e.readUint8(),e.readUint8(),e.readUint8()),sint8:e=>e.readInt8(),sint8x2:e=>te(e.readInt8(),e.readInt8()),sint8x4:e=>k(e.readInt8(),e.readInt8(),e.readInt8(),e.readInt8()),unorm8:e=>e.readUint8()/255,unorm8x2:e=>O(e.readUint8()/255,e.readUint8()/255),unorm8x4:e=>F(e.readUint8()/255,e.readUint8()/255,e.readUint8()/255,e.readUint8()/255),snorm8:e=>(e.readUint8()-128)/127,snorm8x2:e=>O((e.readUint8()-128)/127,(e.readUint8()-128)/127),snorm8x4:e=>F((e.readUint8()-128)/127,(e.readUint8()-128)/127,(e.readUint8()-128)/127,(e.readUint8()-128)/127),uint16:e=>e.readUint16(),uint16x2:e=>re(e.readUint16(),e.readUint16()),uint16x4:e=>E(e.readUint16(),e.readUint16(),e.readUint16(),e.readUint16()),sint16:e=>e.readInt16(),sint16x2:e=>te(e.readInt16(),e.readInt16()),sint16x4:e=>k(e.readInt16(),e.readInt16(),e.readInt16(),e.readInt16()),unorm16:e=>e.readUint16()/65535,unorm16x2:e=>O(e.readUint16()/65535,e.readUint16()/65535),unorm16x4:e=>F(e.readUint16()/65535,e.readUint16()/65535,e.readUint16()/65535,e.readUint16()/65535),snorm16:e=>(e.readUint16()-32768)/32767,snorm16x2:e=>O(Te.snorm16(e),Te.snorm16(e)),snorm16x4:e=>F(Te.snorm16(e),Te.snorm16(e),Te.snorm16(e),Te.snorm16(e)),float16(e){return e.readFloat16()},float16x2:e=>O(e.readFloat16(),e.readFloat16()),float16x4:e=>F(e.readFloat16(),e.readFloat16(),e.readFloat16(),e.readFloat16()),float32:e=>e.readFloat32(),float32x2:e=>O(e.readFloat32(),e.readFloat32()),float32x3:e=>fe(e.readFloat32(),e.readFloat32(),e.readFloat32()),float32x4:e=>F(e.readFloat32(),e.readFloat32(),e.readFloat32(),e.readFloat32()),uint32:e=>e.readUint32(),uint32x2:e=>re(e.readUint32(),e.readUint32()),uint32x3:e=>Ae(e.readUint32(),e.readUint32(),e.readUint32()),uint32x4:e=>E(e.readUint32(),e.readUint32(),e.readUint32(),e.readUint32()),sint32:e=>e.readInt32(),sint32x2:e=>te(e.readInt32(),e.readInt32()),sint32x3:e=>Fe(e.readInt32(),e.readInt32(),e.readInt32()),sint32x4:e=>k(e.readInt32(),e.readInt32(),e.readInt32(),e.readInt32()),"unorm10-10-10-2"(e){let t=e.readUint32(),r=(t>>22)/1023,n=(t>>12&1023)/1023,a=(t>>2&1023)/1023,o=(t&3)/3;return F(r,n,a,o)},"unorm8x4-bgra"(e){let t=e.readByte()/255,r=e.readByte()/255,n=e.readByte()/255,a=e.readByte()/255;return F(n,r,t,a)},unstruct(e,t){let r={};for(let[n,a]of Object.entries(t.propTypes))r[n]=M(e,a);return r},disarray(e,t){let r=P(t),n=[];for(let a=0;a<t.elementCount;a++)B(e,r),n.push(M(e,t.elementType));return B(e,r),n},"loose-decorated"(e,t){B(e,K(t));let r=e.currentByteOffset,n=M(e,t.inner);return e.seekTo(r+S(t)),n}};function M(e,t){let r=Te[t.type];if(!r)throw new Error(`Cannot read data of type '${t.type}'.`);return r(e,t)}import{BufferWriter as ba}from"typed-binary";function hn(e,t){let r=S(e);if(r===0||t===void 0||t===null)return[];let n=new ArrayBuffer(r),a=new ba(n),o=[];function s(u,d,g,c){if(d!=null){if(I(u)||Z(u)){let y=gt(u);for(let[x,h]of Object.entries(y)){let T=u.propTypes[x];if(!T)continue;let D=d[x];D!==void 0&&s(T,D,g+h.offset,h.padding??c)}return}if(z(u)||nt(u)){let y=u,x=_(S(y.elementType),P(y.elementType));if(!Array.isArray(d))throw new Error("Partial value for array must be an array");let h=d??[];h.sort((T,D)=>T.idx-D.idx);for(let{idx:T,value:D}of h)s(y.elementType,D,g+T*x,x-S(y.elementType))}else{let y=S(u);a.seekTo(g),ne(a,u,d),o.push({start:g,end:g+y,padding:c})}}}if(s(e,t,0),o.length===0)return[];let i=[],l=o[0];for(let u=1;u<o.length;u++){let d=o[u];if(!d||!l)throw new Error("Internal error: missing segment");d.start===l.end+(l.padding??0)?(l.end=d.end,l.padding=d.padding):(i.push({data:new Uint8Array(n,l.start,l.end-l.start)}),l=d)}if(!l)throw new Error("Internal error: missing segment");return i.push({data:new Uint8Array(n,l.start,l.end-l.start)}),i}function xe(e){return!!e?.usableAsStorage}var He=class e extends Error{constructor(t){super(`Resource '${p(t)??"<unnamed>"}' cannot be bound as 'storage'. Use .$usage('storage') to allow it.`),Object.setPrototypeOf(this,e.prototype)}};function xt(e){return!!e.usableAsUniform}var vn={uniform:"uniform",mutable:"storage, read_write",readonly:"storage, read"},qe=class{constructor(t,r){this.usage=t;this.buffer=r;this[m]={dataType:r.dataType},this[b]=r}resourceType="buffer-usage";[m];[b];$name(t){return this.buffer.$name(t),this}"~resolve"(t){let r=t.names.makeUnique(p(this)),{group:n,binding:a}=t.allocateFixedEntry(this.usage==="uniform"?{uniform:this.buffer.dataType}:{storage:this.buffer.dataType,access:this.usage},this.buffer),o=vn[this.usage];return t.addDeclaration(`@group(${n}) @binding(${a}) var<${o}> ${r}: ${t.resolve(this.buffer.dataType)};`),r}write(t){this.buffer.write(t)}toString(){return`${this.usage}:${p(this)??"<unnamed>"}`}get value(){if(!de())throw new Error("Cannot access buffer's value directly in JS.");return new Proxy({"~resolve":t=>t.resolve(this),toString:()=>`.value:${p(this)??"<unnamed>"}`,[m]:{dataType:this.buffer.dataType}},Y)}},Je=class{constructor(t,r,n){this.usage=t;this.dataType=r;this._membership=n;this[m]={dataType:r},w(this,n.key)}resourceType="buffer-usage";[m];"~resolve"(t){let r=t.names.makeUnique(p(this)),n=t.allocateLayoutEntry(this._membership.layout),a=vn[this.usage];return t.addDeclaration(`@group(${n}) @binding(${this._membership.idx}) var<${a}> ${r}: ${t.resolve(this.dataType)};`),r}toString(){return`${this.usage}:${p(this)??"<unnamed>"}`}get value(){if(!de())throw new Error("Cannot access buffer's value directly in JS.");return new Proxy({"~resolve":t=>t.resolve(this),toString:()=>`.value:${p(this)??"<unnamed>"}`,[m]:{dataType:this.dataType}},Y)}},wn=new WeakMap;function mr(e){if(!xe(e))throw new Error(`Cannot pass ${e} to asMutable, as it is not allowed to be used as storage. To allow it, call .$usage('storage') when creating the buffer.`);let t=wn.get(e);return t||(t=new qe("mutable",e),wn.set(e,t)),t}var bn=new WeakMap;function fr(e){if(!xe(e))throw new Error(`Cannot pass ${e} to asReadonly, as it is not allowed to be used as storage. To allow it, call .$usage('storage') when creating the buffer.`);let t=bn.get(e);return t||(t=new qe("readonly",e),bn.set(e,t)),t}var Sn=new WeakMap;function gr(e){if(!xt(e))throw new Error(`Cannot pass ${e} to asUniform, as it is not allowed to be used as a uniform. To allow it, call .$usage('uniform') when creating the buffer.`);let t=Sn.get(e);return t||(t=new qe("uniform",e),Sn.set(e,t)),t}var va={uniform:gr,mutable:mr,readonly:fr};function Fn(e,t,r){return j(t)?new ht(e,t,r):new ht(e,t,r,["storage","uniform"])}function ae(e){return e.resourceType==="buffer"}function Da(e){return!!e.usableAsVertex}var Dn=Sa(),ht=class{constructor(t,r,n,a){this._group=t;this.dataType=r;this.initialOrBuffer=n;this._disallowedUsages=a;nn(n)?(this._ownBuffer=!1,this._buffer=n):(this._ownBuffer=!0,this.initial=n)}resourceType="buffer";flags=GPUBufferUsage.COPY_DST|GPUBufferUsage.COPY_SRC;_buffer=null;_ownBuffer;_destroyed=!1;_hostBuffer;initial;usableAsUniform=!1;usableAsStorage=!1;usableAsVertex=!1;get buffer(){let t=this._group.device;if(this._destroyed)throw new Error("This buffer has been destroyed");if(!this._buffer&&(this._buffer=t.createBuffer({size:S(this.dataType),usage:this.flags,mappedAtCreation:!!this.initial,label:p(this)??"<unnamed>"}),this.initial)){let r=new yr(this._buffer.getMappedRange());ne(r,this.dataType,this.initial),this._buffer.unmap()}return this._buffer}get destroyed(){return this._destroyed}$name(t){return w(this,t),this._buffer&&(this._buffer.label=t),this}$usage(...t){for(let r of t){if(this._disallowedUsages?.includes(r))throw new Error(`Buffer of type ${this.dataType.type} cannot be used as ${r}`);this.flags|=r==="uniform"?GPUBufferUsage.UNIFORM:0,this.flags|=r==="storage"?GPUBufferUsage.STORAGE:0,this.flags|=r==="vertex"?GPUBufferUsage.VERTEX:0,this.usableAsUniform=this.usableAsUniform||r==="uniform",this.usableAsStorage=this.usableAsStorage||r==="storage",this.usableAsVertex=this.usableAsVertex||r==="vertex"}return this}$addFlags(t){if(!this._ownBuffer)throw new Error("Cannot add flags to a buffer that is not managed by TypeGPU.");return t&GPUBufferUsage.MAP_READ?(this.flags=GPUBufferUsage.COPY_DST|GPUBufferUsage.MAP_READ,this):t&GPUBufferUsage.MAP_WRITE?(this.flags=GPUBufferUsage.COPY_SRC|GPUBufferUsage.MAP_WRITE,this):(this.flags|=t,this)}compileWriter(){if(yt)Tt(this.dataType);else throw new Error("This environment does not allow eval")}write(t){let r=this.buffer,n=this._group.device;if(r.mapState==="mapped"){let o=r.getMappedRange();if(yt){Tt(this.dataType)(new DataView(o),0,t,Dn==="little");return}ne(new yr(o),this.dataType,t);return}let a=S(this.dataType);this._hostBuffer||(this._hostBuffer=new ArrayBuffer(a)),this._group.flush(),yt?Tt(this.dataType)(new DataView(this._hostBuffer),0,t,Dn==="little"):ne(new yr(this._hostBuffer),this.dataType,t),n.queue.writeBuffer(r,0,this._hostBuffer,0,a)}writePartial(t){let r=this.buffer,n=this._group.device,a=hn(this.dataType,t);if(r.mapState==="mapped"){let o=r.getMappedRange(),s=new Uint8Array(o);for(let i of a)s.set(i.data,i.data.byteOffset)}else for(let o of a)n.queue.writeBuffer(r,o.data.byteOffset,o.data,0,o.data.byteLength)}copyFrom(t){if(this.buffer.mapState==="mapped")throw new Error("Cannot copy to a mapped buffer.");let r=S(this.dataType);this._group.commandEncoder.copyBufferToBuffer(t.buffer,0,this.buffer,0,r)}async read(){this._group.flush();let t=this.buffer,r=this._group.device;if(t.mapState==="mapped"){let s=t.getMappedRange();return M(new cr(s),this.dataType)}if(t.usage&GPUBufferUsage.MAP_READ){await t.mapAsync(GPUMapMode.READ);let s=t.getMappedRange(),i=M(new cr(s),this.dataType);return t.unmap(),i}let n=r.createBuffer({size:S(this.dataType),usage:GPUBufferUsage.COPY_DST|GPUBufferUsage.MAP_READ}),a=r.createCommandEncoder();a.copyBufferToBuffer(t,0,n,0,S(this.dataType)),r.queue.submit([a.finish()]),await r.queue.onSubmittedWorkDone(),await n.mapAsync(GPUMapMode.READ,0,S(this.dataType));let o=M(new cr(n.getMappedRange()),this.dataType);return n.unmap(),n.destroy(),o}as(t){return va[t]?.(this)}destroy(){this._destroyed||(this._destroyed=!0,this._ownBuffer&&this._buffer?.destroy())}toString(){return`buffer:${p(this)??"<unnamed>"}`}};function An(e){return new Tr(e)}function In(e){return new xr(e)}function Ye(e){let t=e;return t?.resourceType==="sampler"&&!!t[m]}function Qe(e){let t=e;return t?.resourceType==="sampler-comparison"&&!!t[m]}var wt=class{constructor(t){this._membership=t;this[m]={},w(this,t.key)}[m];resourceType="sampler";"~resolve"(t){let r=t.names.makeUnique(p(this)),n=t.allocateLayoutEntry(this._membership.layout);return t.addDeclaration(`@group(${n}) @binding(${this._membership.idx}) var ${r}: sampler;`),r}toString(){return`${this.resourceType}:${p(this)??"<unnamed>"}`}},bt=class{constructor(t){this._membership=t;this[m]={},w(this,t.key)}[m];resourceType="sampler-comparison";"~resolve"(t){let r=t.names.makeUnique(p(this)),n=t.allocateLayoutEntry(this._membership.layout);return t.addDeclaration(`@group(${n}) @binding(${this._membership.idx}) var ${r}: sampler_comparison;`),r}toString(){return`${this.resourceType}:${p(this)??"<unnamed>"}`}},Tr=class{constructor(t){this._props=t;this[m]={unwrap:r=>(this._sampler||(this._sampler=r.device.createSampler({...this._props,label:p(this)??"<unnamed>"})),this._sampler)},this._filtering=t.minFilter==="linear"||t.magFilter==="linear"||t.mipmapFilter==="linear"}[m];resourceType="sampler";_filtering;_sampler=null;$name(t){return w(this,t),this}"~resolve"(t){let r=t.names.makeUnique(p(this)),{group:n,binding:a}=t.allocateFixedEntry({sampler:this._filtering?"filtering":"non-filtering"},this);return t.addDeclaration(`@group(${n}) @binding(${a}) var ${r}: sampler;`),r}toString(){return`${this.resourceType}:${p(this)??"<unnamed>"}`}},xr=class{constructor(t){this._props=t;this[m]={unwrap:r=>(this._sampler||(this._sampler=r.device.createSampler({...this._props,label:p(this)??"<unnamed>"})),this._sampler)}}[m];resourceType="sampler-comparison";_sampler=null;$name(t){return w(this,t),this}"~resolve"(t){let r=t.names.makeUnique(p(this)),{group:n,binding:a}=t.allocateFixedEntry({sampler:"comparison"},this);return t.addDeclaration(`@group(${n}) @binding(${a}) var ${r}: sampler_comparison;`),r}toString(){return`${this.resourceType}:${p(this)??"<unnamed>"}`}};var St=class{constructor(t){this._membership=t;w(this,t.key)}resourceType="external-texture";"~resolve"(t){let r=t.names.makeUnique(p(this)),n=t.allocateLayoutEntry(this._membership.layout);return t.addDeclaration(`@group(${n}) @binding(${this._membership.idx}) var ${r}: texture_external;`),r}toString(){return`${this.resourceType}:${p(this)??"<unnamed>"}`}};var Un={r8unorm:f,r8snorm:f,r8uint:R,r8sint:C,r16uint:R,r16sint:C,r16float:f,rg8unorm:f,rg8snorm:f,rg8uint:R,rg8sint:C,r32uint:R,r32sint:C,r32float:f,rg16uint:R,rg16sint:C,rg16float:f,rgba8unorm:f,"rgba8unorm-srgb":f,rgba8snorm:f,rgba8uint:R,rgba8sint:C,bgra8unorm:f,"bgra8unorm-srgb":f,rgb9e5ufloat:f,rgb10a2uint:R,rgb10a2unorm:f,rg11b10ufloat:f,rg32uint:R,rg32sint:C,rg32float:f,rgba16uint:R,rgba16sint:C,rgba16float:f,rgba32uint:R,rgba32sint:C,rgba32float:f,stencil8:f,depth16unorm:f,depth24plus:f,"depth24plus-stencil8":f,depth32float:f,"depth32float-stencil8":f,"bc1-rgba-unorm":f,"bc1-rgba-unorm-srgb":f,"bc2-rgba-unorm":f,"bc2-rgba-unorm-srgb":f,"bc3-rgba-unorm":f,"bc3-rgba-unorm-srgb":f,"bc4-r-unorm":f,"bc4-r-snorm":f,"bc5-rg-unorm":f,"bc5-rg-snorm":f,"bc6h-rgb-ufloat":f,"bc6h-rgb-float":f,"bc7-rgba-unorm":f,"bc7-rgba-unorm-srgb":f,"etc2-rgb8unorm":f,"etc2-rgb8unorm-srgb":f,"etc2-rgb8a1unorm":f,"etc2-rgb8a1unorm-srgb":f,"etc2-rgba8unorm":f,"etc2-rgba8unorm-srgb":f,"eac-r11unorm":f,"eac-r11snorm":f,"eac-rg11unorm":f,"eac-rg11snorm":f,"astc-4x4-unorm":f,"astc-4x4-unorm-srgb":f,"astc-5x4-unorm":f,"astc-5x4-unorm-srgb":f,"astc-5x5-unorm":f,"astc-5x5-unorm-srgb":f,"astc-6x5-unorm":f,"astc-6x5-unorm-srgb":f,"astc-6x6-unorm":f,"astc-6x6-unorm-srgb":f,"astc-8x5-unorm":f,"astc-8x5-unorm-srgb":f,"astc-8x6-unorm":f,"astc-8x6-unorm-srgb":f,"astc-8x8-unorm":f,"astc-8x8-unorm-srgb":f,"astc-10x5-unorm":f,"astc-10x5-unorm-srgb":f,"astc-10x6-unorm":f,"astc-10x6-unorm-srgb":f,"astc-10x8-unorm":f,"astc-10x8-unorm-srgb":f,"astc-10x10-unorm":f,"astc-10x10-unorm-srgb":f,"astc-12x10-unorm":f,"astc-12x10-unorm-srgb":f,"astc-12x12-unorm":f,"astc-12x12-unorm-srgb":f},Xe={rgba8unorm:F,rgba8snorm:F,rgba8uint:E,rgba8sint:k,rgba16uint:E,rgba16sint:k,rgba16float:F,r32uint:E,r32sint:k,r32float:F,rg32uint:E,rg32sint:k,rg32float:F,rgba32uint:E,rgba32sint:k,rgba32float:F,bgra8unorm:F},Bn={f32:"float",u32:"uint",i32:"sint"},Rn={float:f,"unfilterable-float":f,uint:R,sint:C,depth:f};function Cn(e,t){return new hr(e,t)}function X(e){return e?.resourceType==="texture"&&!!e[m]}function Ze(e){return e?.resourceType==="texture-storage-view"&&!!e[m]}function et(e){return e?.resourceType==="texture-sampled-view"&&!!e[m]}var Pn={mutable:"read_write",readonly:"read",writeonly:"write"},hr=class{constructor(t,r){this.props=t;this._branch=r;this[m]={unwrap:()=>{if(this._destroyed)throw new Error("This texture has been destroyed");return this._texture||(this._texture=this._branch.device.createTexture({label:p(this)??"<unnamed>",format:this.props.format,size:this.props.size,usage:this._flags,dimension:this.props.dimension??"2d",viewFormats:this.props.viewFormats??[],mipLevelCount:this.props.mipLevelCount??1,sampleCount:this.props.sampleCount??1})),this._texture}}}[m];resourceType="texture";usableAsSampled=!1;usableAsStorage=!1;usableAsRender=!1;_destroyed=!1;_flags=GPUTextureUsage.COPY_DST|GPUTextureUsage.COPY_SRC;_texture=null;$name(t){return w(this,t),this}$usage(...t){let r=t.includes("storage"),n=t.includes("sampled"),a=t.includes("render");return this._flags|=n?GPUTextureUsage.TEXTURE_BINDING:0,this._flags|=r?GPUTextureUsage.STORAGE_BINDING:0,this._flags|=a?GPUTextureUsage.RENDER_ATTACHMENT:0,this.usableAsStorage||=r,this.usableAsSampled||=n,this.usableAsRender||=a,this}createView(t,r){if(t==="sampled")return this._asSampled(r);let n=r;switch(t){case"mutable":return this._asMutable(n);case"readonly":return this._asReadonly(n);case"writeonly":return this._asWriteonly(n)}}_asStorage(t,r){if(!this.usableAsStorage)throw new Error("Unusable as storage");let n=t?.format??this.props.format,a=Xe[n];return pe(!!a,`Unsupported storage texture format: ${n}`),new wr(t??{},r,this)}_asReadonly(t){return this._asStorage(t,"readonly")}_asWriteonly(t){return this._asStorage(t,"writeonly")}_asMutable(t){return this._asStorage(t,"mutable")}_asSampled(t){if(!this.usableAsSampled)throw new Error("Unusable as sampled");let r=t?.format??this.props.format;if(!Xe[r])throw new Error(`Unsupported storage texture format: ${r}`);return new br(t,this)}destroy(){this._destroyed||(this._destroyed=!0,this._texture?.destroy())}},Ft={"1d":"1d","2d":"2d","2d-array":"2d_array",cube:"cube","cube-array":"cube_array","3d":"3d"},wr=class{constructor(t,r,n){this.access=r;this._texture=n;this[m]={unwrap:()=>(this._view||(this._view=this._texture[m].unwrap().createView({label:`${p(this)??"<unnamed>"} - View`,format:this._format,dimension:this.dimension})),this._view)},this[b]=n,this.dimension=t?.dimension??n.props.dimension??"2d",this._format=t?.format??n.props.format,this.texelDataType=Xe[this._format]}[m];[b];resourceType="texture-storage-view";texelDataType;dimension;_view;_format;$name(t){return this._texture.$name(t),this}"~resolve"(t){let r=t.names.makeUnique(p(this)),{group:n,binding:a}=t.allocateFixedEntry({storageTexture:this._format,access:this.access,viewDimension:this.dimension},this),o=`texture_storage_${Ft[this.dimension]}`;return t.addDeclaration(`@group(${n}) @binding(${a}) var ${r}: ${o}<${this._format}, ${Pn[this.access]}>;`),r}toString(){return`${this.resourceType}:${p(this)??"<unnamed>"}`}},vt=class{constructor(t,r,n,a){this._format=t;this.dimension=r;this.access=n;this._membership=a;this[m]={},this.texelDataType=Xe[this._format],w(this,a.key)}[m];resourceType="texture-storage-view";texelDataType;"~resolve"(t){let r=t.names.makeUnique(p(this)),n=t.allocateLayoutEntry(this._membership.layout),a=`texture_storage_${Ft[this.dimension]}`;return t.addDeclaration(`@group(${n}) @binding(${this._membership.idx}) var ${r}: ${a}<${this._format}, ${Pn[this.access]}>;`),r}toString(){return`${this.resourceType}:${p(this)??"<unnamed>"}`}},br=class{constructor(t,r){this._props=t;this._texture=r;this[m]={unwrap:()=>(this._view||(this._view=this._texture[m].unwrap().createView({label:`${p(this)??"<unnamed>"} - View`,...this._props})),this._view)},this[b]=r,this.dimension=t?.dimension??r.props.dimension??"2d",this._format=t?.format??r.props.format,this.channelDataType=Un[this._format]}[m];[b];resourceType="texture-sampled-view";channelDataType;dimension;_format;_view;$name(t){return this._texture.$name(t),this}"~resolve"(t){let r=t.names.makeUnique(p(this)),n=(this._texture.props.sampleCount??1)>1,{group:a,binding:o}=t.allocateFixedEntry({texture:Bn[this.channelDataType.type],viewDimension:this.dimension,multisampled:n},this),s=n?"texture_multisampled_2d":`texture_${Ft[this.dimension]}`;return t.addDeclaration(`@group(${a}) @binding(${o}) var ${r}: ${s}<${t.resolve(this.channelDataType)}>;`),r}toString(){return`${this.resourceType}:${p(this)??"<unnamed>"}`}},Dt=class{constructor(t,r,n,a){this.dimension=r;this._multisampled=n;this._membership=a;this[m]={},w(this,a.key),this.channelDataType=Rn[t]}[m];resourceType="texture-sampled-view";channelDataType;"~resolve"(t){let r=t.names.makeUnique(p(this)),n=t.allocateLayoutEntry(this._membership.layout),a=this._multisampled?"texture_multisampled_2d":`texture_${Ft[this.dimension]}`;return t.addDeclaration(`@group(${n}) @binding(${this._membership.idx}) var ${r}: ${a}<${t.resolve(this.channelDataType)}>;`),r}toString(){return`${this.resourceType}:${p(this)??"<unnamed>"}`}};function Sr(e){return!!e?.usableAsSampled}function Fa(e){return!!e?.usableAsRender}var At=class e extends Error{constructor(t){super(`Resource '${p(t)??"<unnamed>"}' cannot be bound as 'sampled'. Use .$usage('sampled') to allow it.`),Object.setPrototypeOf(this,e.prototype)}};function It(e){return new Dr(e)}function Ut(e){return!!e&&e.resourceType==="bind-group-layout"}function Fr(e){return!!e&&e.resourceType==="bind-group"}var vr=class e extends Error{constructor(t,r){super(`Bind group '${t??"<unnamed>"}' is missing a required binding '${r}'`),Object.setPrototypeOf(this,e.prototype)}},Vn=["compute"],Ce=["compute","vertex","fragment"],Dr=class{constructor(t){this.entries=t;let r=0;for(let[n,a]of Object.entries(t)){if(a===null){r++;continue}let o={layout:this,key:n,idx:r};if("uniform"in a&&(this.bound[n]=new Je("uniform",a.uniform,o)),"storage"in a){let s="type"in a.storage?a.storage:a.storage(0);this.bound[n]=new Je(a.access??"readonly",s,o)}"texture"in a&&(this.bound[n]=new Dt(a.texture,a.viewDimension??"2d",a.multisampled??!1,o)),"storageTexture"in a&&(this.bound[n]=new vt(a.storageTexture,a.viewDimension??"2d",a.access??"writeonly",o)),"externalTexture"in a&&(this.bound[n]=new St(o)),"sampler"in a&&(a.sampler==="comparison"?this.bound[n]=new bt(o):this.bound[n]=new wt(o)),"texture"in a||"storageTexture"in a||"externalTexture"in a||"sampler"in a?this.value[n]=this.bound[n]:Object.defineProperty(this.value,n,{get:()=>this.bound[n].value}),r++}}_index;resourceType="bind-group-layout";bound={};value={};$=this.value;toString(){return`bindGroupLayout:${p(this)??"<unnamed>"}`}get index(){return this._index}$name(t){return w(this,t),this}$idx(t){return this._index=t,this}unwrap(t){return t.device.createBindGroupLayout({label:p(this)??"<unnamed>",entries:Object.values(this.entries).map((n,a)=>{if(n===null)return null;let o=n.visibility,s={binding:a,visibility:0};if("uniform"in n)o=o??Ce,s.buffer={type:"uniform"};else if("storage"in n)o=o??(n.access==="mutable"?Vn:Ce),s.buffer={type:n.access==="mutable"?"storage":"read-only-storage"};else if("sampler"in n)o=o??Ce,s.sampler={type:n.sampler};else if("texture"in n)o=o??Ce,s.texture={sampleType:n.texture,viewDimension:n.viewDimension??"2d",multisampled:n.multisampled??!1};else if("storageTexture"in n){let i=n.access??"writeonly";o=o??(i==="readonly"?Ce:Vn),s.storageTexture={format:n.storageTexture,access:{mutable:"read-write",readonly:"read-only",writeonly:"write-only"}[i],viewDimension:n.viewDimension??"2d"}}else"externalTexture"in n&&(o=o??Ce,s.externalTexture={});return o?.includes("compute")&&(s.visibility|=GPUShaderStage.COMPUTE),o?.includes("vertex")&&(s.visibility|=GPUShaderStage.VERTEX),o?.includes("fragment")&&(s.visibility|=GPUShaderStage.FRAGMENT),s}).filter(n=>n!==null)})}},Pe=class{constructor(t,r){this.layout=t;this.entries=r;for(let n of Object.keys(t.entries))if(t.entries[n]!==null&&!(n in r))throw new vr(p(t),n)}resourceType="bind-group";unwrap(t){return t.device.createBindGroup({label:p(this.layout)??"<unnamed>",layout:t.unwrap(this.layout),entries:Object.entries(this.layout.entries).map(([n,a],o)=>{if(a===null)return null;let s=this.entries[n];if(s===void 0)throw new Error(`'${n}' is a resource required to populate bind group layout '${p(this.layout)??"<unnamed>"}'.`);if("uniform"in a){let i;if(ae(s)){if(!xt(s))throw new Ot(s);i={buffer:t.unwrap(s)}}else i={buffer:s};return{binding:o,resource:i}}if("storage"in a){let i;if(ae(s)){if(!xe(s))throw new He(s);i={buffer:t.unwrap(s)}}else i={buffer:s};return{binding:o,resource:i}}if("texture"in a){let i;if(X(s)){if(!Sr(s))throw new At(s);i=t.unwrap(s.createView("sampled"))}else et(s)?i=t.unwrap(s):i=s;return{binding:o,resource:i}}if("storageTexture"in a){let i;if(X(s)){if(!xe(s))throw new He(s);a.access==="readonly"?i=t.unwrap(s.createView("readonly")):a.access==="mutable"?i=t.unwrap(s.createView("mutable")):i=t.unwrap(s.createView("writeonly"))}else Ze(s)?i=t.unwrap(s):i=s;return{binding:o,resource:i}}if("sampler"in a)return Ye(s)||Qe(s)?{binding:o,resource:t.unwrap(s)}:{binding:o,resource:s};if("externalTexture"in a)return{binding:o,resource:s};throw new Error(`Malformed bind group entry: ${s} (${JSON.stringify(s)})`)}).filter(n=>n!==null)})}};import*as Aa from"tinyest";var{NodeTypeCatalog:v}=Aa,Ia=["==","!=","<","<=",">",">=","<<",">>","+","-","*","/","%","|","^","&","&&","||"],Ua=["&&","||","==","!=","<","<=",">",">="];function $n(e,t,r){return r?Ua.includes(t)?L:t==="="?r:e:t==="!"||t==="~"?L:e}function Ba(e){throw new Error(`'${JSON.stringify(e)}' was not handled by the WGSL generator.`)}function kn(e,t){return{value:t?"true":"false",dataType:L}}function En(e,[t,r]){e.pushBlockScope();try{return`${e.indent()}{
|
19
|
-
${r.map(n=>
|
14
|
+
`),r}function qa(e,t){let r=e.resolve(t.elementType);return t.elementCount===0?`array<${r}>`:`array<${r}, ${t.elementCount}>`}function Qa(e,t){let r=e.resolve(Qe(t.elementType)?Me[t.elementType.format]:t.elementType);return t.elementCount===0?`array<${r}>`:`array<${r}, ${t.elementCount}>`}function fr(e,t){if(Se(t))return t.type==="unstruct"?za(e,t):t.type==="disarray"?Qa(e,t):t.type==="loose-decorated"?e.resolve(Qe(t.inner)?Me[t.inner.format]:t.inner):e.resolve(Me[t.type]);if(Na(t))return t.type;if(t.type==="struct")return ja(e,t);if(t.type==="array")return qa(e,t);if(t.type==="atomic")return`atomic<${fr(e,t.inner)}>`;if(t.type==="decorated")return e.resolve(t.inner);if(t.type==="ptr")return t.addressSpace==="storage"?`ptr<storage, ${e.resolve(t.inner)}, ${t.access==="read-write"?"read_write":t.access}>`:`ptr<${t.addressSpace}, ${e.resolve(t.inner)}>`;if(t.type==="abstractInt"||t.type==="abstractFloat")throw new Error("Abstract types have no concrete representation in WGSL");if(t.type==="void")throw new Error("Void has no representation in WGSL");dt(t,"resolveData")}import{BufferReader as wr,BufferWriter as br,getSystemEndianness as Xa}from"typed-binary";import{Measurer as Ha}from"typed-binary";function Ka(e,t){let r="size"in e?e.size:e.currentByteOffset,n=t-1,a=r&n;"skipBytes"in e?e.skipBytes(t-a&n):e.add(t-a&n)}var B=Ka;var Ln=new WeakMap;function Tt(e){let t=Ln.get(e);if(t)return t;let r=new Ha,n={},a;for(let o in e.propTypes){let i=e.propTypes[o];if(i===void 0)throw new Error(`Property ${o} is undefined in struct`);let s=r.size;B(r,q(e)?Q(i):V(i)),a&&(a.padding=r.size-s);let l=S(i);n[o]={offset:r.size,size:l},a=n[o],r.add(l)}return a&&(a.padding=L(S(e),V(e))-r.size),Ln.set(e,n),n}var ht=(()=>{try{return new Function("return true"),!0}catch{return!1}})(),cr=new WeakMap,gr={u32:"u32",vec2u:"u32",vec3u:"u32",vec4u:"u32",i32:"i32",vec2i:"i32",vec3i:"i32",vec4i:"i32",f32:"f32",vec2f:"f32",vec3f:"f32",vec4f:"f32",vec2h:"f32",vec3h:"f32",vec4h:"f32",mat2x2f:"f32",mat3x3f:"f32",mat4x4f:"f32"},yr={u32:"setUint32",i32:"setInt32",f32:"setFloat32"};function xt(e,t,r){if(un(e)||ot(e))return xt(e.inner,t,r);if(A(e)||q(e)){let a=Tt(e),o=Object.entries(a).sort((s,l)=>s[1].offset-l[1].offset),i="";for(let[s,l]of o){let u=e.propTypes[s];u&&(i+=xt(u,`(${t} + ${l.offset})`,`${r}.${s}`))}return i}if(z(e)||ve(e)){let a=e,o=L(S(a.elementType),V(a.elementType)),i="";return i+=`for (let i = 0; i < ${a.elementCount}; i++) {
|
15
|
+
`,i+=xt(a.elementType,`(${t} + i * ${o})`,`${r}[i]`),i+=`}
|
16
|
+
`,i}if(Y(e)){let a=gr[e.type],o="",i=yr[a],s=["x","y","z","w"],l=rn(e)?2:nn(e)?3:4;for(let u=0;u<l;u++)o+=`output.${i}((${t} + ${u*4}), ${r}.${s[u]}, littleEndian);
|
17
|
+
`;return o}if(fe(e)){let a=gr[e.type],o=yr[a],i=on(e)?2:sn(e)?3:4,s=i*i,l=L(i*4,8),u="";for(let c=0;c<s;c++){let m=Math.floor(c/i),g=c%i,T=m*l+g*4;u+=`output.${o}((${t} + ${T}), ${r}.columns[${m}].${["x","y","z","w"][g]}, littleEndian);
|
18
|
+
`}return u}let n=gr[e.type];return`output.${yr[n]}(${t}, ${r}, littleEndian);
|
19
|
+
`}function wt(e){if(cr.has(e))return cr.get(e);let t=xt(e,"offset","value"),r=new Function("output","offset","value","littleEndian=true",t);return cr.set(e,r),r}var Ce={bool(){throw new Error("Booleans are not host-shareable")},f32(e,t,r){e.writeFloat32(r)},f16(e,t,r){e.writeFloat16(r)},i32(e,t,r){e.writeInt32(r)},u32(e,t,r){e.writeUint32(r)},vec2f(e,t,r){e.writeFloat32(r.x),e.writeFloat32(r.y)},vec2h(e,t,r){e.writeFloat16(r.x),e.writeFloat16(r.y)},vec2i(e,t,r){e.writeInt32(r.x),e.writeInt32(r.y)},vec2u(e,t,r){e.writeUint32(r.x),e.writeUint32(r.y)},"vec2<bool>"(){throw new Error("Booleans are not host-shareable")},vec3f(e,t,r){e.writeFloat32(r.x),e.writeFloat32(r.y),e.writeFloat32(r.z)},vec3h(e,t,r){e.writeFloat16(r.x),e.writeFloat16(r.y),e.writeFloat16(r.z)},vec3i(e,t,r){e.writeInt32(r.x),e.writeInt32(r.y),e.writeInt32(r.z)},vec3u(e,t,r){e.writeUint32(r.x),e.writeUint32(r.y),e.writeUint32(r.z)},"vec3<bool>"(){throw new Error("Booleans are not host-shareable")},vec4f(e,t,r){e.writeFloat32(r.x),e.writeFloat32(r.y),e.writeFloat32(r.z),e.writeFloat32(r.w)},vec4h(e,t,r){e.writeFloat16(r.x),e.writeFloat16(r.y),e.writeFloat16(r.z),e.writeFloat16(r.w)},vec4i(e,t,r){e.writeInt32(r.x),e.writeInt32(r.y),e.writeInt32(r.z),e.writeInt32(r.w)},vec4u(e,t,r){e.writeUint32(r.x),e.writeUint32(r.y),e.writeUint32(r.z),e.writeUint32(r.w)},"vec4<bool>"(){throw new Error("Booleans are not host-shareable")},mat2x2f(e,t,r){for(let n=0;n<r.length;++n)e.writeFloat32(r[n])},mat3x3f(e,t,r){for(let n=0;n<r.length;++n)e.writeFloat32(r[n])},mat4x4f(e,t,r){for(let n=0;n<r.length;++n)e.writeFloat32(r[n])},struct(e,t,r){let n=V(t);B(e,n);for(let[a,o]of Object.entries(t.propTypes))B(e,V(o)),ue(e,o,r[a]);B(e,n)},array(e,t,r){if(t.elementCount===0)throw new Error("Cannot write using a runtime-sized schema.");let n=V(t);B(e,n);let a=e.currentByteOffset;for(let o=0;o<Math.min(t.elementCount,r.length);o++)B(e,n),ue(e,t.elementType,r[o]);e.seekTo(a+S(t))},ptr(){throw new Error("Pointers are not host-shareable")},atomic(e,t,r){Ce[t.inner.type]?.(e,t,r)},decorated(e,t,r){let n=Q(t);B(e,n);let a=e.currentByteOffset;Ce[t.inner?.type]?.(e,t.inner,r),e.seekTo(a+S(t))},uint8(e,t,r){e.writeUint8(r)},uint8x2(e,t,r){e.writeUint8(r.x),e.writeUint8(r.y)},uint8x4(e,t,r){e.writeUint8(r.x),e.writeUint8(r.y),e.writeUint8(r.z),e.writeUint8(r.w)},sint8(e,t,r){e.writeInt8(r)},sint8x2(e,t,r){e.writeInt8(r.x),e.writeInt8(r.y)},sint8x4(e,t,r){e.writeInt8(r.x),e.writeInt8(r.y),e.writeInt8(r.z),e.writeInt8(r.w)},unorm8(e,t,r){e.writeUint8(r*255)},unorm8x2(e,t,r){e.writeUint8(r.x*255),e.writeUint8(r.y*255)},unorm8x4(e,t,r){e.writeUint8(r.x*255),e.writeUint8(r.y*255),e.writeUint8(r.z*255),e.writeUint8(r.w*255)},snorm8(e,t,r){e.writeUint8(r*127+128)},snorm8x2(e,t,r){e.writeUint8(r.x*127+128),e.writeUint8(r.y*127+128)},snorm8x4(e,t,r){e.writeUint8(r.x*127+128),e.writeUint8(r.y*127+128),e.writeUint8(r.z*127+128),e.writeUint8(r.w*127+128)},uint16(e,t,r){e.writeUint16(r)},uint16x2(e,t,r){e.writeUint16(r.x),e.writeUint16(r.y)},uint16x4(e,t,r){e.writeUint16(r.x),e.writeUint16(r.y),e.writeUint16(r.z),e.writeUint16(r.w)},sint16(e,t,r){e.writeInt16(r)},sint16x2(e,t,r){e.writeInt16(r.x),e.writeInt16(r.y)},sint16x4(e,t,r){e.writeInt16(r.x),e.writeInt16(r.y),e.writeInt16(r.z),e.writeInt16(r.w)},unorm16(e,t,r){e.writeUint16(r*65535)},unorm16x2(e,t,r){e.writeUint16(r.x*65535),e.writeUint16(r.y*65535)},unorm16x4(e,t,r){e.writeUint16(r.x*65535),e.writeUint16(r.y*65535),e.writeUint16(r.z*65535),e.writeUint16(r.w*65535)},snorm16(e,t,r){e.writeUint16(r*32767+32768)},snorm16x2(e,t,r){e.writeUint16(r.x*32767+32768),e.writeUint16(r.y*32767+32768)},snorm16x4(e,t,r){e.writeUint16(r.x*32767+32768),e.writeUint16(r.y*32767+32768),e.writeUint16(r.z*32767+32768),e.writeUint16(r.w*32767+32768)},float16(e,t,r){e.writeFloat16(r)},float16x2(e,t,r){e.writeFloat16(r.x),e.writeFloat16(r.y)},float16x4(e,t,r){e.writeFloat16(r.x),e.writeFloat16(r.y),e.writeFloat16(r.z),e.writeFloat16(r.w)},float32(e,t,r){e.writeFloat32(r)},float32x2(e,t,r){e.writeFloat32(r.x),e.writeFloat32(r.y)},float32x3(e,t,r){e.writeFloat32(r.x),e.writeFloat32(r.y),e.writeFloat32(r.z)},float32x4(e,t,r){e.writeFloat32(r.x),e.writeFloat32(r.y),e.writeFloat32(r.z),e.writeFloat32(r.w)},uint32(e,t,r){e.writeUint32(r)},uint32x2(e,t,r){e.writeUint32(r.x),e.writeUint32(r.y)},uint32x3(e,t,r){e.writeUint32(r.x),e.writeUint32(r.y),e.writeUint32(r.z)},uint32x4(e,t,r){e.writeUint32(r.x),e.writeUint32(r.y),e.writeUint32(r.z),e.writeUint32(r.w)},sint32(e,t,r){e.writeInt32(r)},sint32x2(e,t,r){e.writeInt32(r.x),e.writeInt32(r.y)},sint32x3(e,t,r){e.writeInt32(r.x),e.writeInt32(r.y),e.writeInt32(r.z)},sint32x4(e,t,r){e.writeInt32(r.x),e.writeInt32(r.y),e.writeInt32(r.z),e.writeInt32(r.w)},"unorm10-10-10-2"(e,t,r){let n=0;n|=(r.x*1023&1023)<<22,n|=(r.x*1023&1023)<<12,n|=(r.y*1023&1023)<<2,n|=r.z*3&3,e.writeUint32(n)},"unorm8x4-bgra"(e,t,r){e.writeUint8(r.z*255),e.writeUint8(r.y*255),e.writeUint8(r.x*255),e.writeUint8(r.w*255)},disarray(e,t,r){let n=V(t);B(e,n);let a=e.currentByteOffset;for(let o=0;o<Math.min(t.elementCount,r.length);o++)B(e,n),Ce[t.elementType?.type]?.(e,t.elementType,r[o]);e.seekTo(a+S(t))},unstruct(e,t,r){for(let[n,a]of Object.entries(t.propTypes))Ce[a.type]?.(e,a,r[n])},"loose-decorated"(e,t,r){let n=Q(t);B(e,n);let a=e.currentByteOffset,o=Ce[t.inner?.type];return o?.(e,t.inner,r),e.seekTo(a+S(t)),r}};function ue(e,t,r){let n=Ce[t.type];if(!n)throw new Error(`Cannot write data of type '${t.type}'.`);n(e,t,r)}var xe={bool(){throw new Error("Booleans are not host-shareable")},f32(e){return e.readFloat32()},f16(e){return e.readFloat16()},i32(e){return e.readInt32()},u32(e){return e.readUint32()},vec2f(e){return W(e.readFloat32(),e.readFloat32())},vec3f(e){return ce(e.readFloat32(),e.readFloat32(),e.readFloat32())},vec4f(e){return P(e.readFloat32(),e.readFloat32(),e.readFloat32(),e.readFloat32())},vec2h(e){return Le(e.readFloat16(),e.readFloat16())},vec3h(e){return Oe(e.readFloat16(),e.readFloat16(),e.readFloat16())},vec4h(e){return We(e.readFloat16(),e.readFloat16(),e.readFloat16(),e.readFloat16())},vec2i(e){return te(e.readInt32(),e.readInt32())},vec3i(e){return Fe(e.readInt32(),e.readInt32(),e.readInt32())},vec4i(e){return G(e.readInt32(),e.readInt32(),e.readInt32(),e.readInt32())},vec2u(e){return re(e.readUint32(),e.readUint32())},vec3u(e){return Ie(e.readUint32(),e.readUint32(),e.readUint32())},vec4u(e){return k(e.readUint32(),e.readUint32(),e.readUint32(),e.readUint32())},"vec2<bool>"(){throw new Error("Booleans are not host-shareable")},"vec3<bool>"(){throw new Error("Booleans are not host-shareable")},"vec4<bool>"(){throw new Error("Booleans are not host-shareable")},mat2x2f(e){return ut(e.readFloat32(),e.readFloat32(),e.readFloat32(),e.readFloat32())},mat3x3f(e){let t=()=>{let r=e.readFloat32();return e.readFloat32(),r};return pt(e.readFloat32(),e.readFloat32(),t(),e.readFloat32(),e.readFloat32(),t(),e.readFloat32(),e.readFloat32(),t())},mat4x4f(e){return lt(e.readFloat32(),e.readFloat32(),e.readFloat32(),e.readFloat32(),e.readFloat32(),e.readFloat32(),e.readFloat32(),e.readFloat32(),e.readFloat32(),e.readFloat32(),e.readFloat32(),e.readFloat32(),e.readFloat32(),e.readFloat32(),e.readFloat32(),e.readFloat32())},struct(e,t){let r=V(t);B(e,r);let n={};for(let[a,o]of Object.entries(t.propTypes))B(e,V(o)),n[a]=N(e,o);return B(e,r),n},array(e,t){if(t.elementCount===0)throw new Error("Cannot read using a runtime-sized schema.");let r=V(t),n=[];for(let a=0;a<t.elementCount;a++){B(e,r);let o=t.elementType,i=N(e,o);n.push(i)}return B(e,r),n},ptr(){throw new Error("Pointers are not host-shareable")},atomic(e,t){return N(e,t.inner)},decorated(e,t){let r=Q(t);B(e,r);let n=e.currentByteOffset,a=N(e,t.inner);return e.seekTo(n+S(t)),a},uint8:e=>e.readUint8(),uint8x2:e=>re(e.readUint8(),e.readUint8()),uint8x4:e=>k(e.readUint8(),e.readUint8(),e.readUint8(),e.readUint8()),sint8:e=>e.readInt8(),sint8x2:e=>te(e.readInt8(),e.readInt8()),sint8x4:e=>G(e.readInt8(),e.readInt8(),e.readInt8(),e.readInt8()),unorm8:e=>e.readUint8()/255,unorm8x2:e=>W(e.readUint8()/255,e.readUint8()/255),unorm8x4:e=>P(e.readUint8()/255,e.readUint8()/255,e.readUint8()/255,e.readUint8()/255),snorm8:e=>(e.readUint8()-128)/127,snorm8x2:e=>W((e.readUint8()-128)/127,(e.readUint8()-128)/127),snorm8x4:e=>P((e.readUint8()-128)/127,(e.readUint8()-128)/127,(e.readUint8()-128)/127,(e.readUint8()-128)/127),uint16:e=>e.readUint16(),uint16x2:e=>re(e.readUint16(),e.readUint16()),uint16x4:e=>k(e.readUint16(),e.readUint16(),e.readUint16(),e.readUint16()),sint16:e=>e.readInt16(),sint16x2:e=>te(e.readInt16(),e.readInt16()),sint16x4:e=>G(e.readInt16(),e.readInt16(),e.readInt16(),e.readInt16()),unorm16:e=>e.readUint16()/65535,unorm16x2:e=>W(e.readUint16()/65535,e.readUint16()/65535),unorm16x4:e=>P(e.readUint16()/65535,e.readUint16()/65535,e.readUint16()/65535,e.readUint16()/65535),snorm16:e=>(e.readUint16()-32768)/32767,snorm16x2:e=>W(xe.snorm16(e),xe.snorm16(e)),snorm16x4:e=>P(xe.snorm16(e),xe.snorm16(e),xe.snorm16(e),xe.snorm16(e)),float16(e){return e.readFloat16()},float16x2:e=>W(e.readFloat16(),e.readFloat16()),float16x4:e=>P(e.readFloat16(),e.readFloat16(),e.readFloat16(),e.readFloat16()),float32:e=>e.readFloat32(),float32x2:e=>W(e.readFloat32(),e.readFloat32()),float32x3:e=>ce(e.readFloat32(),e.readFloat32(),e.readFloat32()),float32x4:e=>P(e.readFloat32(),e.readFloat32(),e.readFloat32(),e.readFloat32()),uint32:e=>e.readUint32(),uint32x2:e=>re(e.readUint32(),e.readUint32()),uint32x3:e=>Ie(e.readUint32(),e.readUint32(),e.readUint32()),uint32x4:e=>k(e.readUint32(),e.readUint32(),e.readUint32(),e.readUint32()),sint32:e=>e.readInt32(),sint32x2:e=>te(e.readInt32(),e.readInt32()),sint32x3:e=>Fe(e.readInt32(),e.readInt32(),e.readInt32()),sint32x4:e=>G(e.readInt32(),e.readInt32(),e.readInt32(),e.readInt32()),"unorm10-10-10-2"(e){let t=e.readUint32(),r=(t>>22)/1023,n=(t>>12&1023)/1023,a=(t>>2&1023)/1023,o=(t&3)/3;return P(r,n,a,o)},"unorm8x4-bgra"(e){let t=e.readByte()/255,r=e.readByte()/255,n=e.readByte()/255,a=e.readByte()/255;return P(n,r,t,a)},unstruct(e,t){let r={};for(let[n,a]of Object.entries(t.propTypes))r[n]=N(e,a);return r},disarray(e,t){let r=V(t),n=[];for(let a=0;a<t.elementCount;a++)B(e,r),n.push(N(e,t.elementType));return B(e,r),n},"loose-decorated"(e,t){B(e,Q(t));let r=e.currentByteOffset,n=N(e,t.inner);return e.seekTo(r+S(t)),n}};function N(e,t){let r=xe[t.type];if(!r)throw new Error(`Cannot read data of type '${t.type}'.`);return r(e,t)}import{BufferWriter as Ya}from"typed-binary";function On(e,t){let r=S(e);if(r===0||t===void 0||t===null)return[];let n=new ArrayBuffer(r),a=new Ya(n),o=[];function i(u,c,m,g){if(c!=null){if(A(u)||q(u)){let T=Tt(u);for(let[x,y]of Object.entries(T)){let F=u.propTypes[x];if(!F)continue;let U=c[x];U!==void 0&&i(F,U,m+y.offset,y.padding??g)}return}if(z(u)||ve(u)){let T=u,x=L(S(T.elementType),V(T.elementType));if(!Array.isArray(c))throw new Error("Partial value for array must be an array");let y=c??[];y.sort((F,U)=>F.idx-U.idx);for(let{idx:F,value:U}of y)i(T.elementType,U,m+F*x,x-S(T.elementType))}else{let T=S(u);a.seekTo(m),ue(a,u,c),o.push({start:m,end:m+T,padding:g})}}}if(i(e,t,0),o.length===0)return[];let s=[],l=o[0];for(let u=1;u<o.length;u++){let c=o[u];if(!c||!l)throw new Error("Internal error: missing segment");c.start===l.end+(l.padding??0)?(l.end=c.end,l.padding=c.padding):(s.push({data:new Uint8Array(n,l.start,l.end-l.start)}),l=c)}if(!l)throw new Error("Internal error: missing segment");return s.push({data:new Uint8Array(n,l.start,l.end-l.start)}),s}function he(e){return!!e?.usableAsStorage}var Ke=class e extends Error{constructor(t){super(`Resource '${p(t)??"<unnamed>"}' cannot be bound as 'storage'. Use .$usage('storage') to allow it.`),Object.setPrototypeOf(this,e.prototype)}};function bt(e){return!!e.usableAsUniform}var jn={uniform:"uniform",mutable:"storage, read_write",readonly:"storage, read"},He=class{constructor(t,r){this.usage=t;this.buffer=r;this[d]={dataType:r.dataType},this[v]=r}resourceType="buffer-usage";[d];[v];$name(t){return this.buffer.$name(t),this}"~resolve"(t){let r=t.names.makeUnique(p(this)),{group:n,binding:a}=t.allocateFixedEntry(this.usage==="uniform"?{uniform:this.buffer.dataType}:{storage:this.buffer.dataType,access:this.usage},this.buffer),o=jn[this.usage];return t.addDeclaration(`@group(${n}) @binding(${a}) var<${o}> ${r}: ${t.resolve(this.buffer.dataType)};`),r}write(t){this.buffer.write(t)}toString(){return`${this.usage}:${p(this)??"<unnamed>"}`}[I](){return new Proxy({"~resolve":t=>t.resolve(this),toString:()=>`.value:${p(this)??"<unnamed>"}`,[b]:this.buffer.dataType},K)}get value(){if(!J())throw new Error("Cannot access buffer's value directly in JS.");return this[I]()}},Je=class{constructor(t,r,n){this.usage=t;this.dataType=r;this._membership=n;this[d]={dataType:r},w(this,n.key)}resourceType="buffer-usage";[d];"~resolve"(t){let r=t.names.makeUnique(p(this)),n=t.allocateLayoutEntry(this._membership.layout),a=jn[this.usage];return t.addDeclaration(`@group(${n}) @binding(${this._membership.idx}) var<${a}> ${r}: ${t.resolve(this.dataType)};`),r}toString(){return`${this.usage}:${p(this)??"<unnamed>"}`}[I](){return new Proxy({"~resolve":t=>t.resolve(this),toString:()=>`.value:${p(this)??"<unnamed>"}`,[b]:this.dataType},K)}get value(){if(!J())throw new Error("Cannot access buffer's value directly in JS.");return this[I]()}},Wn=new WeakMap;function Tr(e){if(!he(e))throw new Error(`Cannot pass ${e} to asMutable, as it is not allowed to be used as storage. To allow it, call .$usage('storage') when creating the buffer.`);let t=Wn.get(e);return t||(t=new He("mutable",e),Wn.set(e,t)),t}var Mn=new WeakMap;function xr(e){if(!he(e))throw new Error(`Cannot pass ${e} to asReadonly, as it is not allowed to be used as storage. To allow it, call .$usage('storage') when creating the buffer.`);let t=Mn.get(e);return t||(t=new He("readonly",e),Mn.set(e,t)),t}var Nn=new WeakMap;function hr(e){if(!bt(e))throw new Error(`Cannot pass ${e} to asUniform, as it is not allowed to be used as a uniform. To allow it, call .$usage('uniform') when creating the buffer.`);let t=Nn.get(e);return t||(t=new He("uniform",e),Nn.set(e,t)),t}var Za={uniform:hr,mutable:Tr,readonly:xr};function qn(e,t,r){return be(t)?new St(e,t,r):new St(e,t,r,["storage","uniform"])}function pe(e){return e.resourceType==="buffer"}function eo(e){return!!e.usableAsVertex}var zn=Xa(),St=class{constructor(t,r,n,a){this._group=t;this.dataType=r;this.initialOrBuffer=n;this._disallowedUsages=a;bn(n)?(this._ownBuffer=!1,this._buffer=n):(this._ownBuffer=!0,this.initial=n)}resourceType="buffer";flags=GPUBufferUsage.COPY_DST|GPUBufferUsage.COPY_SRC;_buffer=null;_ownBuffer;_destroyed=!1;_hostBuffer;initial;usableAsUniform=!1;usableAsStorage=!1;usableAsVertex=!1;get buffer(){let t=this._group.device;if(this._destroyed)throw new Error("This buffer has been destroyed");if(!this._buffer&&(this._buffer=t.createBuffer({size:S(this.dataType),usage:this.flags,mappedAtCreation:!!this.initial,label:p(this)??"<unnamed>"}),this.initial)){let r=new br(this._buffer.getMappedRange());ue(r,this.dataType,this.initial),this._buffer.unmap()}return this._buffer}get destroyed(){return this._destroyed}$name(t){return w(this,t),this._buffer&&(this._buffer.label=t),this}$usage(...t){for(let r of t){if(this._disallowedUsages?.includes(r))throw new Error(`Buffer of type ${this.dataType.type} cannot be used as ${r}`);this.flags|=r==="uniform"?GPUBufferUsage.UNIFORM:0,this.flags|=r==="storage"?GPUBufferUsage.STORAGE:0,this.flags|=r==="vertex"?GPUBufferUsage.VERTEX:0,this.usableAsUniform=this.usableAsUniform||r==="uniform",this.usableAsStorage=this.usableAsStorage||r==="storage",this.usableAsVertex=this.usableAsVertex||r==="vertex"}return this}$addFlags(t){if(!this._ownBuffer)throw new Error("Cannot add flags to a buffer that is not managed by TypeGPU.");return t&GPUBufferUsage.MAP_READ?(this.flags=GPUBufferUsage.COPY_DST|GPUBufferUsage.MAP_READ,this):t&GPUBufferUsage.MAP_WRITE?(this.flags=GPUBufferUsage.COPY_SRC|GPUBufferUsage.MAP_WRITE,this):(this.flags|=t,this)}compileWriter(){if(ht)wt(this.dataType);else throw new Error("This environment does not allow eval")}write(t){let r=this.buffer,n=this._group.device;if(r.mapState==="mapped"){let o=r.getMappedRange();if(ht){wt(this.dataType)(new DataView(o),0,t,zn==="little");return}ue(new br(o),this.dataType,t);return}let a=S(this.dataType);this._hostBuffer||(this._hostBuffer=new ArrayBuffer(a)),this._group.flush(),ht?wt(this.dataType)(new DataView(this._hostBuffer),0,t,zn==="little"):ue(new br(this._hostBuffer),this.dataType,t),n.queue.writeBuffer(r,0,this._hostBuffer,0,a)}writePartial(t){let r=this.buffer,n=this._group.device,a=On(this.dataType,t);if(r.mapState==="mapped"){let o=r.getMappedRange(),i=new Uint8Array(o);for(let s of a)i.set(s.data,s.data.byteOffset)}else for(let o of a)n.queue.writeBuffer(r,o.data.byteOffset,o.data,0,o.data.byteLength)}copyFrom(t){if(this.buffer.mapState==="mapped")throw new Error("Cannot copy to a mapped buffer.");let r=S(this.dataType);this._group.commandEncoder.copyBufferToBuffer(t.buffer,0,this.buffer,0,r)}async read(){this._group.flush();let t=this.buffer,r=this._group.device;if(t.mapState==="mapped"){let i=t.getMappedRange();return N(new wr(i),this.dataType)}if(t.usage&GPUBufferUsage.MAP_READ){await t.mapAsync(GPUMapMode.READ);let i=t.getMappedRange(),s=N(new wr(i),this.dataType);return t.unmap(),s}let n=r.createBuffer({size:S(this.dataType),usage:GPUBufferUsage.COPY_DST|GPUBufferUsage.MAP_READ}),a=r.createCommandEncoder();a.copyBufferToBuffer(t,0,n,0,S(this.dataType)),r.queue.submit([a.finish()]),await r.queue.onSubmittedWorkDone(),await n.mapAsync(GPUMapMode.READ,0,S(this.dataType));let o=N(new wr(n.getMappedRange()),this.dataType);return n.unmap(),n.destroy(),o}as(t){return Za[t]?.(this)}destroy(){this._destroyed||(this._destroyed=!0,this._ownBuffer&&this._buffer?.destroy())}toString(){return`buffer:${p(this)??"<unnamed>"}`}};function Qn(e){return new Sr(e)}function Kn(e){return new vr(e)}function Ye(e){let t=e;return t?.resourceType==="sampler"&&!!t[d]}function Xe(e){let t=e;return t?.resourceType==="sampler-comparison"&&!!t[d]}var vt=class{constructor(t){this._membership=t;this[b]=this,this[d]={},w(this,t.key)}[b];[d];resourceType="sampler";"~resolve"(t){let r=t.names.makeUnique(p(this)),n=t.allocateLayoutEntry(this._membership.layout);return t.addDeclaration(`@group(${n}) @binding(${this._membership.idx}) var ${r}: sampler;`),r}toString(){return`${this.resourceType}:${p(this)??"<unnamed>"}`}},Dt=class{constructor(t){this._membership=t;this[b]=this,this[d]={},w(this,t.key)}[b];[d];resourceType="sampler-comparison";"~resolve"(t){let r=t.names.makeUnique(p(this)),n=t.allocateLayoutEntry(this._membership.layout);return t.addDeclaration(`@group(${n}) @binding(${this._membership.idx}) var ${r}: sampler_comparison;`),r}toString(){return`${this.resourceType}:${p(this)??"<unnamed>"}`}},Sr=class{constructor(t){this._props=t;this[b]=this,this[d]={unwrap:r=>(this._sampler||(this._sampler=r.device.createSampler({...this._props,label:p(this)??"<unnamed>"})),this._sampler)},this._filtering=t.minFilter==="linear"||t.magFilter==="linear"||t.mipmapFilter==="linear"}[b];[d];resourceType="sampler";_filtering;_sampler=null;$name(t){return w(this,t),this}"~resolve"(t){let r=t.names.makeUnique(p(this)),{group:n,binding:a}=t.allocateFixedEntry({sampler:this._filtering?"filtering":"non-filtering"},this);return t.addDeclaration(`@group(${n}) @binding(${a}) var ${r}: sampler;`),r}toString(){return`${this.resourceType}:${p(this)??"<unnamed>"}`}},vr=class{constructor(t){this._props=t;this[b]=this,this[d]={unwrap:r=>(this._sampler||(this._sampler=r.device.createSampler({...this._props,label:p(this)??"<unnamed>"})),this._sampler)}}[b];[d];resourceType="sampler-comparison";_sampler=null;$name(t){return w(this,t),this}"~resolve"(t){let r=t.names.makeUnique(p(this)),{group:n,binding:a}=t.allocateFixedEntry({sampler:"comparison"},this);return t.addDeclaration(`@group(${n}) @binding(${a}) var ${r}: sampler_comparison;`),r}toString(){return`${this.resourceType}:${p(this)??"<unnamed>"}`}};var Ft=class{constructor(t){this._membership=t;w(this,t.key)}resourceType="external-texture";"~resolve"(t){let r=t.names.makeUnique(p(this)),n=t.allocateLayoutEntry(this._membership.layout);return t.addDeclaration(`@group(${n}) @binding(${this._membership.idx}) var ${r}: texture_external;`),r}toString(){return`${this.resourceType}:${p(this)??"<unnamed>"}`}};var Hn={r8unorm:f,r8snorm:f,r8uint:R,r8sint:C,r16uint:R,r16sint:C,r16float:f,rg8unorm:f,rg8snorm:f,rg8uint:R,rg8sint:C,r32uint:R,r32sint:C,r32float:f,rg16uint:R,rg16sint:C,rg16float:f,rgba8unorm:f,"rgba8unorm-srgb":f,rgba8snorm:f,rgba8uint:R,rgba8sint:C,bgra8unorm:f,"bgra8unorm-srgb":f,rgb9e5ufloat:f,rgb10a2uint:R,rgb10a2unorm:f,rg11b10ufloat:f,rg32uint:R,rg32sint:C,rg32float:f,rgba16uint:R,rgba16sint:C,rgba16float:f,rgba32uint:R,rgba32sint:C,rgba32float:f,stencil8:f,depth16unorm:f,depth24plus:f,"depth24plus-stencil8":f,depth32float:f,"depth32float-stencil8":f,"bc1-rgba-unorm":f,"bc1-rgba-unorm-srgb":f,"bc2-rgba-unorm":f,"bc2-rgba-unorm-srgb":f,"bc3-rgba-unorm":f,"bc3-rgba-unorm-srgb":f,"bc4-r-unorm":f,"bc4-r-snorm":f,"bc5-rg-unorm":f,"bc5-rg-snorm":f,"bc6h-rgb-ufloat":f,"bc6h-rgb-float":f,"bc7-rgba-unorm":f,"bc7-rgba-unorm-srgb":f,"etc2-rgb8unorm":f,"etc2-rgb8unorm-srgb":f,"etc2-rgb8a1unorm":f,"etc2-rgb8a1unorm-srgb":f,"etc2-rgba8unorm":f,"etc2-rgba8unorm-srgb":f,"eac-r11unorm":f,"eac-r11snorm":f,"eac-rg11unorm":f,"eac-rg11snorm":f,"astc-4x4-unorm":f,"astc-4x4-unorm-srgb":f,"astc-5x4-unorm":f,"astc-5x4-unorm-srgb":f,"astc-5x5-unorm":f,"astc-5x5-unorm-srgb":f,"astc-6x5-unorm":f,"astc-6x5-unorm-srgb":f,"astc-6x6-unorm":f,"astc-6x6-unorm-srgb":f,"astc-8x5-unorm":f,"astc-8x5-unorm-srgb":f,"astc-8x6-unorm":f,"astc-8x6-unorm-srgb":f,"astc-8x8-unorm":f,"astc-8x8-unorm-srgb":f,"astc-10x5-unorm":f,"astc-10x5-unorm-srgb":f,"astc-10x6-unorm":f,"astc-10x6-unorm-srgb":f,"astc-10x8-unorm":f,"astc-10x8-unorm-srgb":f,"astc-10x10-unorm":f,"astc-10x10-unorm-srgb":f,"astc-12x10-unorm":f,"astc-12x10-unorm-srgb":f,"astc-12x12-unorm":f,"astc-12x12-unorm-srgb":f},Ze={rgba8unorm:P,rgba8snorm:P,rgba8uint:k,rgba8sint:G,rgba16uint:k,rgba16sint:G,rgba16float:P,r32uint:k,r32sint:G,r32float:P,rg32uint:k,rg32sint:G,rg32float:P,rgba32uint:k,rgba32sint:G,rgba32float:P,bgra8unorm:P},Jn={f32:"float",u32:"uint",i32:"sint"},Yn={float:f,"unfilterable-float":f,uint:R,sint:C,depth:f};function Xn(e,t){return new Dr(e,t)}function X(e){return e?.resourceType==="texture"&&!!e[d]}function et(e){return e?.resourceType==="texture-storage-view"&&!!e[d]}function tt(e){return e?.resourceType==="texture-sampled-view"&&!!e[d]}var Zn={mutable:"read_write",readonly:"read",writeonly:"write"},Dr=class{constructor(t,r){this.props=t;this._branch=r;this[d]={unwrap:()=>{if(this._destroyed)throw new Error("This texture has been destroyed");return this._texture||(this._texture=this._branch.device.createTexture({label:p(this)??"<unnamed>",format:this.props.format,size:this.props.size,usage:this._flags,dimension:this.props.dimension??"2d",viewFormats:this.props.viewFormats??[],mipLevelCount:this.props.mipLevelCount??1,sampleCount:this.props.sampleCount??1})),this._texture}}}[d];resourceType="texture";usableAsSampled=!1;usableAsStorage=!1;usableAsRender=!1;_destroyed=!1;_flags=GPUTextureUsage.COPY_DST|GPUTextureUsage.COPY_SRC;_texture=null;$name(t){return w(this,t),this}$usage(...t){let r=t.includes("storage"),n=t.includes("sampled"),a=t.includes("render");return this._flags|=n?GPUTextureUsage.TEXTURE_BINDING:0,this._flags|=r?GPUTextureUsage.STORAGE_BINDING:0,this._flags|=a?GPUTextureUsage.RENDER_ATTACHMENT:0,this.usableAsStorage||=r,this.usableAsSampled||=n,this.usableAsRender||=a,this}createView(t,r){if(t==="sampled")return this._asSampled(r);let n=r;switch(t){case"mutable":return this._asMutable(n);case"readonly":return this._asReadonly(n);case"writeonly":return this._asWriteonly(n)}}_asStorage(t,r){if(!this.usableAsStorage)throw new Error("Unusable as storage");let n=t?.format??this.props.format,a=Ze[n];return le(!!a,`Unsupported storage texture format: ${n}`),new Fr(t??{},r,this)}_asReadonly(t){return this._asStorage(t,"readonly")}_asWriteonly(t){return this._asStorage(t,"writeonly")}_asMutable(t){return this._asStorage(t,"mutable")}_asSampled(t){if(!this.usableAsSampled)throw new Error("Unusable as sampled");let r=t?.format??this.props.format;if(!Ze[r])throw new Error(`Unsupported storage texture format: ${r}`);return new Ir(t,this)}destroy(){this._destroyed||(this._destroyed=!0,this._texture?.destroy())}},At={"1d":"1d","2d":"2d","2d-array":"2d_array",cube:"cube","cube-array":"cube_array","3d":"3d"},Fr=class{constructor(t,r,n){this.access=r;this._texture=n;this[b]=this,this[d]={unwrap:()=>(this._view||(this._view=this._texture[d].unwrap().createView({label:`${p(this)??"<unnamed>"} - View`,format:this._format,dimension:this.dimension})),this._view)},this[v]=n,this.dimension=t?.dimension??n.props.dimension??"2d",this._format=t?.format??n.props.format,this.texelDataType=Ze[this._format]}[b];[d];[v];resourceType="texture-storage-view";texelDataType;dimension;_view;_format;$name(t){return this._texture.$name(t),this}"~resolve"(t){let r=t.names.makeUnique(p(this)),{group:n,binding:a}=t.allocateFixedEntry({storageTexture:this._format,access:this.access,viewDimension:this.dimension},this),o=`texture_storage_${At[this.dimension]}`;return t.addDeclaration(`@group(${n}) @binding(${a}) var ${r}: ${o}<${this._format}, ${Zn[this.access]}>;`),r}toString(){return`${this.resourceType}:${p(this)??"<unnamed>"}`}},It=class{constructor(t,r,n,a){this._format=t;this.dimension=r;this.access=n;this._membership=a;this[b]=this,this[d]={},this.texelDataType=Ze[this._format],w(this,a.key)}[b];[d];resourceType="texture-storage-view";texelDataType;"~resolve"(t){let r=t.names.makeUnique(p(this)),n=t.allocateLayoutEntry(this._membership.layout),a=`texture_storage_${At[this.dimension]}`;return t.addDeclaration(`@group(${n}) @binding(${this._membership.idx}) var ${r}: ${a}<${this._format}, ${Zn[this.access]}>;`),r}toString(){return`${this.resourceType}:${p(this)??"<unnamed>"}`}},Ir=class{constructor(t,r){this._props=t;this._texture=r;this[b]=this,this[d]={unwrap:()=>(this._view||(this._view=this._texture[d].unwrap().createView({label:`${p(this)??"<unnamed>"} - View`,...this._props})),this._view)},this[v]=r,this.dimension=t?.dimension??r.props.dimension??"2d",this._format=t?.format??r.props.format,this.channelDataType=Hn[this._format]}[b];[d];[v];resourceType="texture-sampled-view";channelDataType;dimension;_format;_view;$name(t){return this._texture.$name(t),this}"~resolve"(t){let r=t.names.makeUnique(p(this)),n=(this._texture.props.sampleCount??1)>1,{group:a,binding:o}=t.allocateFixedEntry({texture:Jn[this.channelDataType.type],viewDimension:this.dimension,multisampled:n},this),i=n?"texture_multisampled_2d":`texture_${At[this.dimension]}`;return t.addDeclaration(`@group(${a}) @binding(${o}) var ${r}: ${i}<${t.resolve(this.channelDataType)}>;`),r}toString(){return`${this.resourceType}:${p(this)??"<unnamed>"}`}},Ut=class{constructor(t,r,n,a){this.dimension=r;this._multisampled=n;this._membership=a;this[b]=this,this[d]={},w(this,a.key),this.channelDataType=Yn[t]}[b];[d];resourceType="texture-sampled-view";channelDataType;"~resolve"(t){let r=t.names.makeUnique(p(this)),n=t.allocateLayoutEntry(this._membership.layout),a=this._multisampled?"texture_multisampled_2d":`texture_${At[this.dimension]}`;return t.addDeclaration(`@group(${n}) @binding(${this._membership.idx}) var ${r}: ${a}<${t.resolve(this.channelDataType)}>;`),r}toString(){return`${this.resourceType}:${p(this)??"<unnamed>"}`}};function Ur(e){return!!e?.usableAsSampled}function to(e){return!!e?.usableAsRender}var Pt=class e extends Error{constructor(t){super(`Resource '${p(t)??"<unnamed>"}' cannot be bound as 'sampled'. Use .$usage('sampled') to allow it.`),Object.setPrototypeOf(this,e.prototype)}};function Bt(e){return new Pr(e)}function Rt(e){return!!e&&e.resourceType==="bind-group-layout"}function Br(e){return!!e&&e.resourceType==="bind-group"}var Ar=class e extends Error{constructor(t,r){super(`Bind group '${t??"<unnamed>"}' is missing a required binding '${r}'`),Object.setPrototypeOf(this,e.prototype)}},ea=["compute"],Ve=["compute","vertex","fragment"],Pr=class{constructor(t){this.entries=t;let r=0;for(let[n,a]of Object.entries(t)){if(a===null){r++;continue}let o={layout:this,key:n,idx:r};if("uniform"in a&&(this.bound[n]=new Je("uniform",a.uniform,o)),"storage"in a){let i="type"in a.storage?a.storage:a.storage(0);this.bound[n]=new Je(a.access??"readonly",i,o)}"texture"in a&&(this.bound[n]=new Ut(a.texture,a.viewDimension??"2d",a.multisampled??!1,o)),"storageTexture"in a&&(this.bound[n]=new It(a.storageTexture,a.viewDimension??"2d",a.access??"writeonly",o)),"externalTexture"in a&&(this.bound[n]=new Ft(o)),"sampler"in a&&(a.sampler==="comparison"?this.bound[n]=new Dt(o):this.bound[n]=new vt(o)),"texture"in a||"storageTexture"in a||"externalTexture"in a||"sampler"in a?this.value[n]=this.bound[n]:Object.defineProperty(this.value,n,{get:()=>this.bound[n].value}),r++}}_index;resourceType="bind-group-layout";bound={};value={};$=this.value;toString(){return`bindGroupLayout:${p(this)??"<unnamed>"}`}get index(){return this._index}$name(t){return w(this,t),this}$idx(t){return this._index=t,this}unwrap(t){return t.device.createBindGroupLayout({label:p(this)??"<unnamed>",entries:Object.values(this.entries).map((n,a)=>{if(n===null)return null;let o=n.visibility,i={binding:a,visibility:0};if("uniform"in n)o=o??Ve,i.buffer={type:"uniform"};else if("storage"in n)o=o??(n.access==="mutable"?ea:Ve),i.buffer={type:n.access==="mutable"?"storage":"read-only-storage"};else if("sampler"in n)o=o??Ve,i.sampler={type:n.sampler};else if("texture"in n)o=o??Ve,i.texture={sampleType:n.texture,viewDimension:n.viewDimension??"2d",multisampled:n.multisampled??!1};else if("storageTexture"in n){let s=n.access??"writeonly";o=o??(s==="readonly"?Ve:ea),i.storageTexture={format:n.storageTexture,access:{mutable:"read-write",readonly:"read-only",writeonly:"write-only"}[s],viewDimension:n.viewDimension??"2d"}}else"externalTexture"in n&&(o=o??Ve,i.externalTexture={});return o?.includes("compute")&&(i.visibility|=GPUShaderStage.COMPUTE),o?.includes("vertex")&&(i.visibility|=GPUShaderStage.VERTEX),o?.includes("fragment")&&(i.visibility|=GPUShaderStage.FRAGMENT),i}).filter(n=>n!==null)})}},_e=class{constructor(t,r){this.layout=t;this.entries=r;for(let n of Object.keys(t.entries))if(t.entries[n]!==null&&!(n in r))throw new Ar(p(t),n)}resourceType="bind-group";unwrap(t){return t.device.createBindGroup({label:p(this.layout)??"<unnamed>",layout:t.unwrap(this.layout),entries:Object.entries(this.layout.entries).map(([n,a],o)=>{if(a===null)return null;let i=this.entries[n];if(i===void 0)throw new Error(`'${n}' is a resource required to populate bind group layout '${p(this.layout)??"<unnamed>"}'.`);if("uniform"in a){let s;if(pe(i)){if(!bt(i))throw new jt(i);s={buffer:t.unwrap(i)}}else s={buffer:i};return{binding:o,resource:s}}if("storage"in a){let s;if(pe(i)){if(!he(i))throw new Ke(i);s={buffer:t.unwrap(i)}}else s={buffer:i};return{binding:o,resource:s}}if("texture"in a){let s;if(X(i)){if(!Ur(i))throw new Pt(i);s=t.unwrap(i.createView("sampled"))}else tt(i)?s=t.unwrap(i):s=i;return{binding:o,resource:s}}if("storageTexture"in a){let s;if(X(i)){if(!he(i))throw new Ke(i);a.access==="readonly"?s=t.unwrap(i.createView("readonly")):a.access==="mutable"?s=t.unwrap(i.createView("mutable")):s=t.unwrap(i.createView("writeonly"))}else et(i)?s=t.unwrap(i):s=i;return{binding:o,resource:s}}if("sampler"in a)return Ye(i)||Xe(i)?{binding:o,resource:t.unwrap(i)}:{binding:o,resource:i};if("externalTexture"in a)return{binding:o,resource:i};throw new Error(`Malformed bind group entry: ${i} (${JSON.stringify(i)})`)}).filter(n=>n!==null)})}};import*as ro from"tinyest";var{NodeTypeCatalog:D}=ro,no=["==","!=","<","<=",">",">=","<<",">>","+","-","*","/","%","|","^","&","&&","||"],ao=["&&","||","==","!=","<","<=",">",">="];function ta(e,t,r){return r?ao.includes(t)?E:t==="="?r:e:t==="!"||t==="~"?E:e}function oo(e){throw new Error(`'${JSON.stringify(e)}' was not handled by the WGSL generator.`)}function ra(e,[t,r]){e.pushBlockScope();try{return`${e.indent()}{
|
20
|
+
${r.map(n=>we(e,n)).join(`
|
20
21
|
`)}
|
21
|
-
${e.dedent()}}`}finally{e.popBlockScope()}}function
|
22
|
-
${
|
22
|
+
${e.dedent()}}`}finally{e.popBlockScope()}}function io(e,t,r){return e.defineVariable(t,r)}function Rr(e,t){let r=e.getById(t);if(!r)throw new Error(`Identifier ${t} not found`);return r}function _(e,t){if(typeof t=="string")return Rr(e,t);if(typeof t=="boolean")return h(t?"true":"false",E);if(t[0]===D.logicalExpr||t[0]===D.binaryExpr||t[0]===D.assignmentExpr){let[r,n,a,o]=t,i=_(e,n),s=_(e,o),l=M(e,[i,s]),[u,c]=l||[i,s],m=e.resolve(u.value),g=e.resolve(c.value),T=ta(u.dataType,a,c.dataType);return h(no.includes(a)?`(${m} ${a} ${g})`:`${m} ${a} ${g}`,T)}if(t[0]===D.postUpdate){let[r,n,a]=t,o=_(e,a),i=e.resolve(o.value);return h(`${i}${n}`,o.dataType)}if(t[0]===D.unaryExpr){let[r,n,a]=t,o=_(e,a),i=e.resolve(o.value),s=ta(o.dataType,n);return h(`${n}${i}`,s)}if(t[0]===D.memberAccess){let[r,n,a]=t,o=_(e,n);if(o.dataType.type==="unknown"){let i=o.value[a];return ge(i)}return Ht(o.dataType)?h(`(*${e.resolve(o.value)}).${a}`,Ne(o.dataType.inner,a)):z(o.dataType)&&a==="length"?o.dataType.elementCount===0?h(`arrayLength(&${e.resolve(o.value)})`,R):h(String(o.dataType.elementCount),De):fe(o.dataType)&&a==="columns"?h(o.value,o.dataType):Y(o.dataType)&&at(o.value)?ge(o.value[a]):h(`${e.resolve(o.value)}.${a}`,Ne(o.dataType,a))}if(t[0]===D.indexAccess){let[r,n,a]=t,o=_(e,n),i=_(e,a),s=e.resolve(o.value),l=e.resolve(i.value);if(o.dataType.type==="unknown"){if(Array.isArray(a)&&a[0]===D.numericLiteral)return ge(o.value[a[1]]);throw new Error(`Cannot index value ${s} of unknown type with index ${l}`)}return Ht(o.dataType)?h(`(*${s})[${l}]`,rr(o.dataType.inner)):h(`${s}[${l}]`,O(o.dataType)?rr(o.dataType):$)}if(t[0]===D.numericLiteral){let r=nr(t[1]);if(!r)throw new Error(`Invalid numeric literal ${t[1]}`);return r}if(t[0]===D.call){let[r,n,a]=t,o=_(e,n);e.callStack.push(o.value);let s=a.map(m=>_(e,m)).map(m=>h(e.resolve(m.value),m.dataType)),l=s.map(m=>m.value);if(e.callStack.pop(),s.forEach((m,g)=>{if(m.dataType===$)throw new Error(`Tried to pass '${m.value}' of unknown type as argument #${g} to '${typeof o.value=="string"?o.value:p(o.value)??"<unnamed>"}()'`)}),typeof o.value=="string")return h(`${o.value}(${l.join(", ")})`,o.dataType);if(A(o.value)){let m=e.resolve(o.value);return h(`${m}(${l.join(", ")})`,o.value)}if(!pr(o.value))throw new Error(`Function ${String(o.value)} ${p(o.value)} has not been created using TypeGPU APIs. Did you mean to wrap the function with tgpu.fn(args, return)(...) ?`);let u=o.value[d]?.argTypes,c;try{!u||u==="keep"?c=s:u==="coerce"?c=M(e,s)??s:c=(Array.isArray(u)?u:u(...s)).map((T,x)=>[T,s[x]]).map(([T,x])=>{if(x.dataType.type==="unknown")return console.warn(`Internal error: unknown type when generating expression: ${t}`),x;let y=M(e,[x],[T])?.[0];if(!y)throw new H(`Cannot convert argument of type '${x.dataType.type}' to '${T.type}' for function ${p(o.value)}`,[{function:o.value,callStack:e.callStack,error:`Cannot convert argument of type '${x.dataType.type}' to '${T.type}'`,toString:()=>p(o.value)}]);return y});let m=o.value(...c);return h(e.resolve(m.value),m.dataType)}catch(m){throw new H(m,[{toString:()=>p(o.value)}])}}if(t[0]===D.objectExpr){let r=t[1],n=e.callStack[e.callStack.length-1];if(A(n)){let a=Object.keys(n.propTypes),o=Object.fromEntries(a.map(s=>{let l=r[s];if(l===void 0)throw new Error(`Missing property ${s} in object literal for struct ${n}`);return[s,_(e,l)]})),i=ir(e,n,o);return h(i.map(s=>e.resolve(s.value)).join(", "),n)}if(pr(n)){let a=n[d]?.argTypes;if(typeof a=="object"&&a!==null){let o=Object.keys(a),i={};for(let s of o){let l=r[s];if(l===void 0)throw new Error(`Missing property ${s} in object literal for function ${n}`);let u=_(e,l),c=a[s],m=M(e,[u],[c]);i[s]=m?.[0]??u}return h(i,$)}}throw new Error("Object expressions are only allowed as return values of functions or as arguments to structs.")}if(t[0]===D.arrayExpr){let[r,n]=t,a=n.map(m=>_(e,m));if(a.length===0)throw new Error("Cannot create empty array literal.");let o=M(e,a);if(!o)throw new Error("The given values cannot be automatically converted to a common type. Consider explicitly casting them.");let i=o[0]?.dataType,s=i.type==="abstractFloat"?f:i.type==="abstractInt"?C:i,u=`array<${e.resolve(s)}, ${a.length}>`,c=o.map(m=>e.resolve(m.value));return h(`${u}( ${c.join(", ")} )`,st(s,a.length))}if(t[0]===D.stringLiteral)throw new Error("Cannot use string literals in TGSL.");if(t[0]===D.preUpdate)throw new Error("Cannot use pre-updates in TGSL.");oo(t)}function Ct(e){return typeof e!="object"||e[0]!==D.block?[D.block,[e]]:e}function we(e,t){if(typeof t=="string")return`${e.pre}${e.resolve(Rr(e,t).value)};`;if(typeof t=="boolean")return`${e.pre}${t?"true":"false"};`;if(t[0]===D.return){let r=t[1],n=r!==void 0?e.resolve(_(e,r).value):void 0;if(A(e.callStack[e.callStack.length-1])&&typeof r=="object"&&r[0]===D.objectExpr){let a=e.resolve(e.callStack[e.callStack.length-1]);return`${e.pre}return ${a}(${n});`}return n?`${e.pre}return ${n};`:`${e.pre}return;`}if(t[0]===D.if){let[r,n,a,o]=t,i=_(e,n),s=i,l=M(e,[i],[E]);l?.[0]&&([s]=l);let u=e.resolve(s.value);e.indent();let c=we(e,Ct(a));e.dedent(),e.indent();let m=o?we(e,Ct(o)):void 0;return e.dedent(),m?`${e.pre}if (${u})
|
23
|
+
${c}
|
23
24
|
${e.pre}else
|
24
|
-
${
|
25
|
-
${
|
26
|
-
${x}`}if(t[0]===
|
27
|
-
${l}`}return t[0]===
|
25
|
+
${m}`:`${e.pre}if (${u})
|
26
|
+
${c}`}if(t[0]===D.let||t[0]===D.const){let[r,n,a]=t,o=a!==void 0?_(e,a):void 0;if(!o)throw new Error(`Cannot create variable '${n}' without an initial value.`);if(Se(o.dataType))throw new Error(`Cannot create variable '${n}' with loose data type.`);io(e,n,or(o.dataType));let i=e.resolve(Rr(e,n).value);if(typeof a=="object"&&a[0]===D.objectExpr&&A(e.callStack[e.callStack.length-1])){let s=e.callStack[e.callStack.length-1],l=a[1],u={};for(let[g,T]of Object.entries(l)){if(!T)throw new Error(`Missing property ${g} in object literal`);u[g]=_(e,T)}let c=ir(e,s,u),m=e.resolve(s);return`${e.pre}var ${i} = ${m}(${c.map(g=>e.resolve(g.value)).join(", ")});`}return`${e.pre}var ${i} = ${e.resolve(o.value)};`}if(t[0]===D.block)return ra(e,t);if(t[0]===D.for){let[r,n,a,o,i]=t,s=n?we(e,n):void 0,l=s?s.slice(0,-1):"",u=a?_(e,a):void 0,c=u;if(u){let y=M(e,[u],[E]);y?.[0]&&([c]=y)}let m=c?e.resolve(c.value):"",g=o?we(e,o):void 0,T=g?g.slice(0,-1):"";e.indent();let x=we(e,Ct(i));return e.dedent(),`${e.pre}for (${l}; ${m}; ${T})
|
27
|
+
${x}`}if(t[0]===D.while){let[r,n,a]=t,o=_(e,n),i=o;if(o){let u=M(e,[o],[E]);u?.[0]&&([i]=u)}let s=e.resolve(i.value);e.indent();let l=we(e,Ct(a));return e.dedent(),`${e.pre}while (${s})
|
28
|
+
${l}`}return t[0]===D.continue?`${e.pre}continue;`:t[0]===D.break?`${e.pre}break;`:`${e.pre}${e.resolve(_(e,t).value)};`}function na(e,t){return ra(e,t)}var aa="#CATCHALL#",Vr=class{_stack=[];_itemDepth=0;get itemDepth(){return this._itemDepth}get topItem(){let t=this._stack[this._stack.length-1];if(!t||t.type!=="item")throw new Error("Internal error, expected item layer to be on top.");return t}pushItem(){this._itemDepth++,this._stack.push({type:"item",usedSlots:new Set})}popItem(){this.pop("item")}pushSlotBindings(t){this._stack.push({type:"slotBinding",bindingMap:new WeakMap(t)})}popSlotBindings(){this.pop("slotBinding")}pushFunctionScope(t,r,n,a){this._stack.push({type:"functionScope",args:t,argAliases:r,returnType:n,externalMap:a})}popFunctionScope(){this.pop("functionScope")}pushBlockScope(){this._stack.push({type:"blockScope",declarations:new Map})}popBlockScope(){this.pop("blockScope")}pop(t){let r=this._stack[this._stack.length-1];if(!r||t&&r.type!==t)throw new Error(`Internal error, expected a ${t} layer to be on top.`);this._stack.pop(),t==="item"&&this._itemDepth--}readSlot(t){for(let r=this._stack.length-1;r>=0;--r){let n=this._stack[r];if(n?.type==="item")n.usedSlots.add(t);else if(n?.type==="slotBinding"){let a=n.bindingMap.get(t);if(a!==void 0)return a}else if(!(n?.type==="functionScope"||n?.type==="blockScope"))throw new Error("Unknown layer type.")}return t.defaultValue}getSnippetById(t){for(let r=this._stack.length-1;r>=0;--r){let n=this._stack[r];if(n?.type==="functionScope"){let a=n.args.find(i=>i.value===t);if(a!==void 0)return a;if(n.argAliases[t])return n.argAliases[t];let o=n.externalMap[t];return o!=null?ge(o):void 0}if(n?.type==="blockScope"){let a=n.declarations.get(t);if(a!==void 0)return h(t,a)}}}defineBlockVariable(t,r){if(r.type==="unknown")throw Error(`Tried to define variable '${t}' of unknown type`);for(let n=this._stack.length-1;n>=0;--n){let a=this._stack[n];if(a?.type==="blockScope")return a.declarations.set(t,r),h(t,r)}throw new Error("No block scope found to define a variable in.")}},Vt=[""," "," "," "," "," "," "," "," "],Cr=Vt.length-1,_r=class{identLevel=0;get pre(){return Vt[this.identLevel]??Vt[Cr].repeat(this.identLevel/Cr)+Vt[this.identLevel%Cr]}indent(){let t=this.pre;return this.identLevel++,t}dedent(){return this.identLevel--,this.pre}},$r=class{_memoizedResolves=new WeakMap;_memoizedDerived=new WeakMap;_indentController=new _r;_jitTranspiler;_itemStateStack=new Vr;_declarations=[];[d]={itemStateStack:this._itemStateStack};bindGroupLayoutsToPlaceholderMap=new Map;_nextFreeLayoutPlaceholderIdx=0;fixedBindings=[];callStack=[];names;constructor(t){this.names=t.names,this._jitTranspiler=t.jitTranspiler}get pre(){return this._indentController.pre}indent(){return this._indentController.indent()}dedent(){return this._indentController.dedent()}getById(t){let r=this._itemStateStack.getSnippetById(t);return r===void 0?null:r}defineVariable(t,r){return this._itemStateStack.defineBlockVariable(t,r)}pushBlockScope(){this._itemStateStack.pushBlockScope()}popBlockScope(){this._itemStateStack.popBlockScope()}transpileFn(t){if(!this._jitTranspiler)throw new Error("Tried to execute a tgpu.fn function without providing a JIT transpiler, or transpiling at build time.");return this._jitTranspiler.transpileFn(t)}fnToWgsl(t){this._itemStateStack.pushFunctionScope(t.args,t.argAliases,t.returnType,t.externalMap);try{return{head:so(this,t.args,t.returnType),body:na(this,t.body)}}finally{this._itemStateStack.popFunctionScope()}}addDeclaration(t){this._declarations.push(t)}allocateLayoutEntry(t){let r=this.bindGroupLayoutsToPlaceholderMap,n=r.get(t);return n||(n=`#BIND_GROUP_LAYOUT_${this._nextFreeLayoutPlaceholderIdx++}#`,r.set(t,n)),n}allocateFixedEntry(t,r){let n=this.fixedBindings.length;return this.fixedBindings.push({layoutEntry:t,resource:r}),{group:aa,binding:n}}readSlot(t){let r=this._itemStateStack.readSlot(t);if(r===void 0)throw new Nt(t);return r}withSlots(t,r){this._itemStateStack.pushSlotBindings(t);try{return r()}finally{this._itemStateStack.popSlotBindings()}}unwrap(t){if(ze(t))return this.withSlots(t["~providing"].pairs,()=>this.unwrap(t["~providing"].inner));let r=t;for(;;)if(Pe(r))r=this.readSlot(r);else if(Be(r))r=this._getOrCompute(r);else break;return r}_getOrCompute(t){let r=this._memoizedDerived.get(t)??[];this._itemStateStack.pushItem();try{for(let o of r)if([...o.slotToValueMap.entries()].every(([s,l])=>s.areEqual(this._itemStateStack.readSlot(s),l)))return o.result;qt(Ge.CPU);let n;try{n=t["~compute"]()}finally{Qt(Ge.CPU)}let a=new Map;for(let o of this._itemStateStack.topItem.usedSlots)a.set(o,this._itemStateStack.readSlot(o));return r.push({slotToValueMap:a,result:n}),this._memoizedDerived.set(t,r),n}catch(n){throw n instanceof H?n.appendToTrace(t):new H(n,[t])}finally{this._itemStateStack.popItem()}}_getOrInstantiate(t){let r=this._memoizedResolves.get(t)??[];this._itemStateStack.pushItem();try{for(let o of r)if([...o.slotToValueMap.entries()].every(([s,l])=>s.areEqual(this._itemStateStack.readSlot(s),l)))return o.result;let n;O(t)?n=fr(this,t):Be(t)||Pe(t)?n=this.resolve(this.unwrap(t)):ur(t)?n=t["~resolve"](this):n=this.resolveValue(t);let a=new Map;for(let o of this._itemStateStack.topItem.usedSlots)a.set(o,this._itemStateStack.readSlot(o));return r.push({slotToValueMap:a,result:n}),this._memoizedResolves.set(t,r),n}catch(n){throw n instanceof H?n.appendToTrace(t):new H(n,[t])}finally{this._itemStateStack.popItem()}}resolve(t){if(ze(t))return this.withSlots(t["~providing"].pairs,()=>this.resolve(t["~providing"].inner));if(t&&typeof t=="object"||typeof t=="function"){if(this._itemStateStack.itemDepth===0)try{qt(Ge.GPU);let r=Xr(this,()=>this._getOrInstantiate(t));return`${[...this._declarations].join(`
|
28
29
|
|
29
|
-
`)}${r}`}finally{Mt(Le.GPU)}return this._getOrInstantiate(t)}return String(t)}resolveValue(t,r){if(J(t))return this.resolve(t);if(r&&z(r))return`array(${t.map(n=>this.resolveValue(n,r.elementType))})`;if(Array.isArray(t))return`array(${t.map(n=>this.resolveValue(n))})`;if(r&&I(r))return`${this.resolve(r)}(${Object.entries(r.propTypes).map(([n,a])=>this.resolveValue(t[n],a))})`;throw new Error(`Value ${t} (as json: ${JSON.stringify(t)}) of schema ${r} is not resolvable to WGSL`)}};function Ve(e,t){let r=new Rr(t),n=r.resolve(e),a=r.bindGroupLayoutsToPlaceholderMap,o=[],s=new Set([...a.keys()].map(g=>g.index).filter(g=>g!==void 0)),i=tn(s),l=r.fixedBindings.map((g,c)=>[String(c),g.layoutEntry]),u=()=>{let g=i.next().value,c=It(Object.fromEntries(l));return o[g]=c,n=n.replaceAll(_n,String(g)),[g,new Pe(c,Object.fromEntries(r.fixedBindings.map((y,x)=>[String(x),y.resource])))]},d=l.length>0?u():null;for(let[g,c]of a.entries()){let y=g.index??i.next().value;o[y]=g,n=n.replaceAll(c,String(y))}return{code:n,bindGroupLayouts:o,catchall:d}}function Cr(e,t,r){let n=t.map(a=>`${a.value}: ${e.resolve(a.dataType)}`).join(", ");return r!==void 0?`(${n}) -> ${Ie(r)} ${e.resolve(r)}`:`(${n})`}function oe(e,t){let r=[];typeof t=="string"&&(e.isEntry?(Array.isArray(e.argTypes)&&I(e.argTypes[0])&&r.push({In:e.argTypes[0]}),I(e.returnType)&&r.push({Out:e.returnType})):Array.isArray(e.argTypes)&&(dn(t,Array.isArray(e.argTypes)?e.argTypes:Object.values(e.argTypes),o=>r.push(o)),Be(t,e.returnType,o=>r.push(o))));let n={applyExternals(o){r.push(o)},resolve(o,s=""){let i={};for(let u of r)Q(i,u);let l=o.names.makeUnique(p(this));if(typeof t=="string"){let u="";if(!e.isEntry)u=Array.isArray(e.argTypes)?"":Cr(o,Object.entries(e.argTypes).map(([g,c])=>({value:g,dataType:c})),e.returnType);else{let g=Array.isArray(e.argTypes)&&I(e.argTypes[0])?"(in: In)":"()",c=j(e.returnType)?Ie(e.returnType):"",y=e.returnType!==void 0?I(e.returnType)?"-> Out":`-> ${c!==""?c:"@location(0)"} ${o.resolve(e.returnType)}`:"";u=`${g} ${y} `}let d=ye(o,i,`${u}${t.trim()}`);o.addDeclaration(`${s}fn ${l}${d}`)}else{let u=gn(t);if(u?.externals){let h=Object.fromEntries(Object.entries(u.externals).filter(([T])=>!(T in i)));Q(i,h)}let d=u?.ast??o.transpileFn(String(t));d.argNames.type==="destructured-object"&&Q(i,Object.fromEntries(d.argNames.props.map(({prop:h,alias:T})=>[T,h]))),!Array.isArray(e.argTypes)&&d.argNames.type==="identifiers"&&d.argNames.names[0]!==void 0&&Q(i,{[d.argNames.names[0]]:Object.fromEntries(Object.keys(e.argTypes).map(h=>[h,h]))});let g=d.externalNames.filter(h=>!(h in i));if(g.length>0)throw new Gt(p(this),g);let c=Array.isArray(e.argTypes)?d.argNames.type==="identifiers"?e.argTypes.map((h,T)=>({value:(d.argNames.type==="identifiers"?d.argNames.names[T]:void 0)??`arg_${T}`,dataType:h})):[]:Object.entries(e.argTypes).map(([h,T])=>({value:h,dataType:T})),{head:y,body:x}=o.fnToWgsl({args:c,returnType:e.returnType,body:d.body,externalMap:i});o.addDeclaration(`${s}fn ${l}${o.resolve(y)}${o.resolve(x)}`)}return l}},a=p(t);return a!==void 0&&w(n,a),n}function On(e){let t=0;return Object.fromEntries(Object.entries(e).map(([r,n])=>{if(ut(n))return[r,n];let a=ee(n);return a!==void 0?(t=a+1,[r,n]):[r,Yt(t++,n)]}))}function Ct(e){return $(e)?rt(e)?void 0:ee(e)!==void 0?e:Yt(0,e):Qt(On(e))}function $e(e){return Qt(On(e))}function se(e,...t){return Ca(e)?Pa(e,...t):e}function Ca(e){return Array.isArray(e)&&"raw"in e&&Array.isArray(e.raw)&&e.raw.every(t=>typeof t=="string")}function Pa(e,...t){return e.slice(1).reduce((r,n,a)=>`${r}${t[a]}${n}`,e[0])}function Gn(e){let t={argTypes:e.in&&Object.keys(e.in).length!==0?[$e(e.in)]:[],returnType:void 0,workgroupSize:[e.workgroupSize[0]??1,e.workgroupSize[1]??1,e.workgroupSize[2]??1],isEntry:!0},r=(n,...a)=>Va(t,e.workgroupSize,se(n,...a));return Object.assign(Object.assign(r,t),{does:r})}function Va(e,t,r){let n=oe(e,r),a=e.argTypes[0];return{shell:e,$uses(o){return n.applyExternals(o),this},[b]:n,$name(o){return w(n,o),N(a)&&a.$name(`${o}_Input`),this},"~resolve"(o){return n.resolve(o,`@compute @workgroup_size(${t.join(", ")}) `)},toString(){return`computeFn:${p(n)??"<unnamed>"}`}}}function Wn(e,t){let r={[m]:!0,argTypes:e,returnType:t,isEntry:!1},n=(a,...o)=>ka(r,se(a,...o));return Object.assign(Object.assign(n,r),{does:n})}function Pr(e){return e?.resourceType==="function"}function $a([e,t]){return`${p(e)??"<unnamed>"}=${t}`}function ka(e,t){let r=oe(e,t),n={[m]:{implementation:t,argTypes:e.argTypes},shell:e,resourceType:"function",$uses(s){return r.applyExternals(s),this},[b]:r,$name(s){return w(r,s),this},with(s,i){return Mn(o,[[ge(s)?s.slot:s,i]])},"~resolve"(s){if(typeof t=="string")return r.resolve(s);let i=s;if(i.callStack===void 0)throw new Error("Cannot resolve a TGSL function outside of a generation context");try{return i.callStack.push(e.returnType),r.resolve(s)}finally{i.callStack.pop()}}},a=zt((...s)=>{if(typeof t=="string")throw new Error("Cannot execute on the CPU functions constructed with raw WGSL");return t(...s)},(...s)=>({value:new Pt(o,s.map(i=>i.value)),dataType:e.returnType??U}),e.argTypes);a[m].implementation=t;let o=Object.assign(a,n);return Object.defineProperty(o,"toString",{value(){return`fn:${p(r)??"<unnamed>"}`}}),o}function Mn(e,t){let r={[m]:{implementation:e[m].implementation,argTypes:e[m].argTypes},resourceType:"function",shell:e.shell,"~providing":{inner:e,pairs:t},$uses(o){return e.$uses(o),this},[b]:e,$name(o){return e.$name(o),this},with(o,s){return Mn(a,[...t,[ge(o)?o.slot:o,s]])}},n=zt((...o)=>e(...o),(...o)=>({value:new Pt(a,o.map(s=>s.value)),dataType:e.shell.returnType??U}),e.shell.argTypes),a=Object.assign(n,r);return Object.defineProperty(a,"toString",{value(){return`fn:${p(e)??"<unnamed>"}[${t.map($a).join(", ")}]`}}),a[m].implementation=e[m].implementation,a}var Pt=class{constructor(t,r){this._fn=t;this._params=r;this[b]=t}[b];"~resolve"(t){return t.resolve(`${t.resolve(this._fn)}(${this._params.map(r=>t.resolve(r)).join(", ")})`)}toString(){return`call:${p(this)??"<unnamed>"}`}};function Nn(e){let t={argTypes:e.in&&Object.keys(e.in).length!==0?[$e(e.in)]:[],targets:e.out,returnType:Ct(e.out),isEntry:!0},r=(n,...a)=>Ea(t,se(n,...a));return Object.assign(Object.assign(r,t),{does:r})}function Ea(e,t){let r=oe(e,t),n=e.returnType,a=e.argTypes[0];return typeof t=="string"&&Be(t,n,s=>r.applyExternals(s)),{shell:e,outputType:n,$uses(s){return r.applyExternals(s),this},[b]:r,$name(s){return w(r,s),N(n)&&n.$name(`${s}_Output`),N(a)&&a.$name(`${s}_Input`),this},"~resolve"(s){if(typeof t=="string")return r.resolve(s,"@fragment ");let i=s;if(i.callStack===void 0)throw new Error("Cannot resolve a TGSL function outside of a generation context");try{return i.callStack.push(n),r.resolve(s,"@fragment ")}finally{i.callStack.pop()}},toString(){return`fragmentFn:${p(r)??"<unnamed>"}`}}}function jn(e){let t={attributes:[e.in??{}],returnType:Object.keys(e.out).length!==0?Ct(e.out):void 0,argTypes:e.in&&Object.keys(e.in).length!==0?[$e(e.in)]:[],isEntry:!0},r=(n,...a)=>La(t,se(n,...a));return Object.assign(Object.assign(r,t),{does:r})}function La(e,t){let r=oe(e,t),n=e.returnType,a=e.argTypes[0];return typeof t=="string"&&Be(t,n,o=>r.applyExternals(o)),{shell:e,outputType:n,inputType:a,$uses(o){return r.applyExternals(o),this},[b]:r,$name(o){return w(r,o),N(n)&&n.$name(`${o}_Output`),N(a)&&a.$name(`${o}_Input`),this},"~resolve"(o){if(typeof t=="string")return r.resolve(o,"@vertex ");let s=o;if(s.callStack===void 0)throw new Error("Cannot resolve a TGSL function outside of a generation context");try{return s.callStack.push(n),r.resolve(o,"@vertex ")}finally{s.callStack.pop()}},toString(){return`vertexFn:${p(r)??"<unnamed>"}`}}}var ie=class{lastUniqueId=0;makeUnique(t){let r;return t?(r=t.replaceAll(/\s/g,"_"),r=r.replaceAll(/[^\w\d]/g,"")):r="item",`${r}_${this.lastUniqueId++}`}},ue=class{_usedNames=new Set;makeUnique(t){if(t===void 0)throw new Error("Unnamed item found when using a strict name registry");let r=0,n=t;for(;this._usedNames.has(n);)r++,n=`${t}_${r}`;return this._usedNames.add(n),n}};function zn(e){let{externals:t,template:r,names:n,unstable_jitTranspiler:a}=e,o={};Q(o,t??{});let s={"~resolve"(l){return ye(l,o,r??"")},toString:()=>"<root>"},{code:i}=Ve(s,{names:n==="strict"?new ue:new ie,jitTranspiler:a});return i}var tt=class{constructor(t){this._make=t}_map=new WeakMap;getOrMake(t,...r){if(this._map.has(t))return this._map.get(t);let n=this._make(t,...r);return this._map.set(t,n),n}};function Kn(e,t,r){return new Vr(new $r(e,t,r),{})}function Hn(e){let t=e;return t?.resourceType==="compute-pipeline"&&!!t[m]}var Vr=class e{constructor(t,r){this._core=t;this._priors=r;this[m]={get rawPipeline(){return t.unwrap().pipeline}},this[b]=t}[m];resourceType="compute-pipeline";[b];get rawPipeline(){return this._core.unwrap().pipeline}with(t,r){return new e(this._core,{bindGroupLayoutMap:new Map([...this._priors.bindGroupLayoutMap??[],[t,r]])})}dispatchWorkgroups(t,r,n){let a=this._core.unwrap(),{branch:o}=this._core,s=o.commandEncoder.beginComputePass({label:p(this._core)??"<unnamed>"});s.setPipeline(a.pipeline);let i=new Set(a.bindGroupLayouts);if(a.bindGroupLayouts.forEach((l,u)=>{if(a.catchall&&u===a.catchall[0])s.setBindGroup(u,o.unwrap(a.catchall[1])),i.delete(l);else{let d=this._priors.bindGroupLayoutMap?.get(l);d!==void 0&&(i.delete(l),s.setBindGroup(u,o.unwrap(d)))}}),i.size>0)throw new le(i);s.dispatchWorkgroups(t,r,n),s.end()}$name(t){return w(this._core,t),this}},$r=class{constructor(t,r,n){this.branch=t;this._slotBindings=r;this._entryFn=n}_memo;unwrap(){if(this._memo===void 0){let t=this.branch.device,{code:r,bindGroupLayouts:n,catchall:a}=Ve({"~resolve":o=>(o.withSlots(this._slotBindings,()=>{o.resolve(this._entryFn)}),""),toString:()=>`computePipeline:${p(this)??"<unnamed>"}`},{names:this.branch.nameRegistry,jitTranspiler:this.branch.jitTranspiler});a!==null&&n[a[0]]?.$name(`${p(this)??"<unnamed>"} - Automatic Bind Group & Layout`),this._memo={pipeline:t.createComputePipeline({label:p(this)??"<unnamed>",layout:t.createPipelineLayout({label:`${p(this)??"<unnamed>"} - Pipeline Layout`,bindGroupLayouts:n.map(o=>this.branch.unwrap(o))}),compute:{module:t.createShaderModule({label:`${p(this)??"<unnamed>"} - Shader`,code:r})}}),bindGroupLayouts:n,catchall:a}}return this._memo}};function Lr(e,t="vertex"){return new Er(e,t)}function $t(e){return e?.resourceType==="vertex-layout"}var kr=Symbol("defaultAttribEntry");function Vt(e,t,r,n,a){if(Se(t)||en(t)){let o=ee(t);return o!==void 0&&(n[a??kr]=o),Vt(e,t.inner,_(r,K(t)),n)}if(I(t)){let o=r;return Object.fromEntries(Object.entries(t.propTypes).map(([s,i])=>{o=_(o,P(i));let l=[s,Vt(e,i,o,n,s)];return o+=S(i),l}))}if(Z(t)){let o=r;return Object.fromEntries(Object.entries(t.propTypes).map(([s,i])=>{o=_(o,K(i));let l=[s,Vt(e,i,o,n,s)];return o+=S(i),l}))}if("type"in t&&typeof t.type=="string"){if(Hr.includes(t.type))return{_layout:e,format:t.type,offset:r};let o=qr[t.type];if(o)return{_layout:e,format:o,offset:r}}throw new Error(`Unsupported data used in vertex layout: ${String(t)}`)}var Er=class{constructor(t,r){this.schemaForCount=t;this.stepMode=r;let n=t(0);this.stride=_(S(n.elementType),P(n)),this.attrib=Vt(this,n.elementType,0,this._customLocationMap)}resourceType="vertex-layout";stride;attrib;_customLocationMap={};get vertexLayout(){if(this._customLocationMap[kr]!==void 0){if(typeof this.attrib.format!="string"||typeof this.attrib.offset!="number")throw new Error("Single attribute vertex layouts must have a format and offset.");return{arrayStride:this.stride,stepMode:this.stepMode,attributes:[{format:this.attrib.format,offset:this.attrib.offset,shaderLocation:this._customLocationMap[kr]}]}}if(!Object.keys(this.attrib).every(r=>this._customLocationMap[r]!==void 0))throw new Error("All attributes must have custom locations in order to unwrap a vertex layout.");return{arrayStride:this.stride,stepMode:this.stepMode,attributes:[...Object.entries(this.attrib).map(([r,n])=>({format:n.format,offset:n.offset,shaderLocation:this._customLocationMap[r]}))]}}$name(t){return w(this,t),this}};function _a(e){return typeof e?.loadOp=="string"}function qn(e,t){if($(e)){if(!_a(t))throw new Error("Expected a single color attachment, not a record.");return[t]}let r=[];for(let n of Object.keys(e)){let a=t[n];if(!a)throw new Error(`A color attachment by the name of '${n}' was not provided to the shader.`);r.push(a)}return r}function Oa(e){return typeof e?.format=="string"}function Jn(e,t){if($(e)){if(rt(e))return[];if(!Oa(t))throw new Error("Expected a single color target configuration, not a record.");return[t]}let r=[];for(let n of Object.keys(e)){let a=t[n];if(!a)throw new Error(`A color target by the name of '${n}' was not provided to the shader.`);r.push(a)}return r}function Yn(e){return new _r(new Or(e),{})}function Qn(e){let t=e;return t?.resourceType==="render-pipeline"&&!!t[m]}var _r=class e{[m];resourceType="render-pipeline";[b];constructor(t,r){this[m]={core:t,priors:r},this[b]=t}$name(t){return w(this[m].core,t),this}with(t,r){let n=this[m];if(Ut(t))return new e(n.core,{...n.priors,bindGroupLayoutMap:new Map([...n.priors.bindGroupLayoutMap??[],[t,r]])});if($t(t))return new e(n.core,{...n.priors,vertexLayoutMap:new Map([...n.priors.vertexLayoutMap??[],[t,r]])});throw new Error("Unsupported value passed into .with()")}withColorAttachment(t){let r=this[m];return new e(r.core,{...r.priors,colorAttachment:t})}withDepthStencilAttachment(t){let r=this[m];return new e(r.core,{...r.priors,depthStencilAttachment:t})}draw(t,r,n,a){let o=this[m],s=o.core.unwrap(),{branch:i,fragmentFn:l}=o.core.options,d={colorAttachments:qn(l.shell.targets,o.priors.colorAttachment??{}).map(T=>X(T.view)?{...T,view:i.unwrap(T.view).createView()}:T)},g=p(o.core);if(g!==void 0&&(d.label=g),o.priors.depthStencilAttachment!==void 0){let T=o.priors.depthStencilAttachment;X(T.view)?d.depthStencilAttachment={...T,view:i.unwrap(T.view).createView()}:d.depthStencilAttachment=T}let c=i.commandEncoder.beginRenderPass(d);c.setPipeline(s.pipeline);let y=new Set(s.bindGroupLayouts);s.bindGroupLayouts.forEach((T,D)=>{if(s.catchall&&D===s.catchall[0])c.setBindGroup(D,i.unwrap(s.catchall[1])),y.delete(T);else{let ke=o.priors.bindGroupLayoutMap?.get(T);ke!==void 0&&(y.delete(T),c.setBindGroup(D,i.unwrap(ke)))}});let x=new Set(o.core.usedVertexLayouts);if(o.core.usedVertexLayouts.forEach((T,D)=>{let ke=o.priors.vertexLayoutMap?.get(T);ke&&(x.delete(T),c.setVertexBuffer(D,i.unwrap(ke)))}),y.size>0)throw new le(y);if(x.size>0)throw new Ee(x);c.draw(t,r,n,a),c.end(),i.flush()}},Or=class{constructor(t){this.options=t;let r=Tn(t.vertexFn.shell.attributes[0],t.vertexAttribs);this._vertexBufferLayouts=r.bufferDefinitions,this.usedVertexLayouts=r.usedVertexLayouts,this._targets=Jn(t.fragmentFn.shell.targets,t.targets)}usedVertexLayouts;_memo;_vertexBufferLayouts;_targets;unwrap(){if(this._memo===void 0){let{branch:t,vertexFn:r,fragmentFn:n,slotBindings:a,primitiveState:o,depthStencilState:s,multisampleState:i}=this.options,{code:l,bindGroupLayouts:u,catchall:d}=Ve({"~resolve":h=>(h.withSlots(a,()=>{h.resolve(r),h.resolve(n)}),""),toString:()=>`renderPipeline:${p(this)??"<unnamed>"}`},{names:t.nameRegistry,jitTranspiler:t.jitTranspiler});d!==null&&u[d[0]]?.$name(`${p(this)??"<unnamed>"} - Automatic Bind Group & Layout`);let g=t.device,c=g.createShaderModule({label:`${p(this)??"<unnamed>"} - Shader`,code:l}),y={layout:g.createPipelineLayout({label:`${p(this)??"<unnamed>"} - Pipeline Layout`,bindGroupLayouts:u.map(h=>t.unwrap(h))}),vertex:{module:c,buffers:this._vertexBufferLayouts},fragment:{module:c,targets:this._targets}},x=p(this);x!==void 0&&(y.label=x),o&&(y.primitive=o),s&&(y.depthStencil=s),i&&(y.multisample=i),this._memo={pipeline:g.createRenderPipeline(y),bindGroupLayouts:u,catchall:d}}return this._memo}};var Gr=class e{constructor(t,r){this._getRoot=t;this._slotBindings=r}with(t,r){return new e(this._getRoot,[...this._slotBindings,[ge(t)?t.slot:t,r]])}withCompute(t){return new Wr(this._getRoot(),this._slotBindings,t)}withVertex(t,r){return new Mr({branch:this._getRoot(),primitiveState:void 0,depthStencilState:void 0,slotBindings:this._slotBindings,vertexFn:t,vertexAttribs:r,multisampleState:void 0})}},Wr=class{constructor(t,r,n){this._root=t;this._slotBindings=r;this._entryFn=n}createPipeline(){return Kn(this._root,this._slotBindings,this._entryFn)}},Mr=class{constructor(t){this._options=t}withFragment(t,r,n){return pe(typeof t!="string","Just type mismatch validation"),pe(typeof r!="string","Just type mismatch validation"),new Nr({...this._options,fragmentFn:t,targets:r})}},Nr=class e{constructor(t){this._options=t}withPrimitive(t){return new e({...this._options,primitiveState:t})}withDepthStencil(t){return new e({...this._options,depthStencilState:t})}withMultisample(t){return new e({...this._options,multisampleState:t})}createPipeline(){return Yn(this._options)}},kt=class extends Gr{constructor(r,n,a,o){super(()=>this,[]);this.device=r;this.nameRegistry=n;this.jitTranspiler=a;this._ownDevice=o;this["~unstable"]=this}"~unstable";_disposables=[];_unwrappedBindGroupLayouts=new tt(r=>r.unwrap(this));_unwrappedBindGroups=new tt(r=>r.unwrap(this));_commandEncoder=null;get commandEncoder(){return this._commandEncoder||(this._commandEncoder=this.device.createCommandEncoder()),this._commandEncoder}createBuffer(r,n){let a=Fn(this,r,n);return this._disposables.push(a),a}createUniform(r,n){return this.createBuffer(r,n).$usage("uniform").as("uniform")}createMutable(r,n){return this.createBuffer(r,n).$usage("storage").as("mutable")}createReadonly(r,n){return this.createBuffer(r,n).$usage("storage").as("readonly")}createBindGroup(r,n){return new Pe(r,n)}destroy(){for(let r of this._disposables)r.destroy();this._ownDevice&&this.device.destroy()}createTexture(r){let n=Cn(r,this);return this._disposables.push(n),n}unwrap(r){if(Hn(r))return r[m].rawPipeline;if(Qn(r))return r[m].core.unwrap().pipeline;if(Ut(r))return this._unwrappedBindGroupLayouts.getOrMake(r);if(Fr(r))return this._unwrappedBindGroups.getOrMake(r);if(ae(r))return r.buffer;if(X(r))return r[m].unwrap();if(Ze(r)){if(r[m].unwrap)return r[m].unwrap();throw new Error("Cannot unwrap laid-out texture view.")}if(et(r)){if(r[m].unwrap)return r[m].unwrap();throw new Error("Cannot unwrap laid-out texture view.")}if($t(r))return r.vertexLayout;if(Ye(r)){if(r[m].unwrap)return r[m].unwrap(this);throw new Error("Cannot unwrap laid-out sampler.")}if(Qe(r)){if(r[m].unwrap)return r[m].unwrap(this);throw new Error("Cannot unwrap laid-out comparison sampler.")}throw new Error(`Unknown resource type: ${r}`)}beginRenderPass(r,n){let a=this.commandEncoder.beginRenderPass(r),o=new Map,s=new Map,i,l=()=>{if(!i)throw new Error("Cannot draw without a call to pass.setPipeline");let{core:u,priors:d}=i[m],g=u.unwrap();a.setPipeline(g.pipeline);let c=new Set(g.bindGroupLayouts);g.bindGroupLayouts.forEach((x,h)=>{if(g.catchall&&h===g.catchall[0])a.setBindGroup(h,this.unwrap(g.catchall[1])),c.delete(x);else{let T=d.bindGroupLayoutMap?.get(x)??o.get(x);T!==void 0&&(c.delete(x),Fr(T)?a.setBindGroup(h,this.unwrap(T)):a.setBindGroup(h,T))}});let y=new Set;if(u.usedVertexLayouts.forEach((x,h)=>{let T=d.vertexLayoutMap?.get(x),D=T?{buffer:T,offset:void 0,size:void 0}:s.get(x);!D||!D.buffer?y.add(x):ae(D.buffer)?a.setVertexBuffer(h,this.unwrap(D.buffer),D.offset,D.size):a.setVertexBuffer(h,D.buffer,D.offset,D.size)}),c.size>0)throw new le(c);if(y.size>0)throw new Ee(y)};n({setViewport(...u){a.setViewport(...u)},setScissorRect(...u){a.setScissorRect(...u)},setBlendConstant(...u){a.setBlendConstant(...u)},setStencilReference(...u){a.setStencilReference(...u)},beginOcclusionQuery(...u){a.beginOcclusionQuery(...u)},endOcclusionQuery(...u){a.endOcclusionQuery(...u)},executeBundles(...u){a.executeBundles(...u)},setPipeline(u){i=u},setIndexBuffer:(u,d,g,c)=>{ae(u)?a.setIndexBuffer(this.unwrap(u),d,g,c):a.setIndexBuffer(u,d,g,c)},setVertexBuffer(u,d,g,c){s.set(u,{buffer:d,offset:g,size:c})},setBindGroup(u,d){o.set(u,d)},draw(u,d,g,c){l(),a.draw(u,d,g,c)},drawIndexed(...u){l(),a.drawIndexed(...u)},drawIndirect(...u){l(),a.drawIndirect(...u)},drawIndexedIndirect(...u){l(),a.drawIndexedIndirect(...u)}}),a.end()}flush(){this._commandEncoder&&(this.device.queue.submit([this._commandEncoder.finish()]),this._commandEncoder=null)}};async function Xn(e){let{adapter:t,device:r,unstable_names:n="random",unstable_jitTranspiler:a}=e??{};if(!navigator.gpu)throw new Error("WebGPU is not supported by this browser.");let o=await navigator.gpu.requestAdapter(t);if(!o)throw new Error("Could not find a compatible GPU");return new kt(await o.requestDevice(r),n==="random"?new ie:new ue,a,!0)}function Zn(e){let{device:t,unstable_names:r="random",unstable_jitTranspiler:n}=e??{};return new kt(t,r==="random"?new ie:new ue,n,!1)}function Et(e){return new jr(e)}var jr=class{constructor(t=void 0){this.defaultValue=t}resourceType="slot";$name(t){return w(this,t),this}areEqual(t,r){return Object.is(t,r)}toString(){return`slot:${p(this)??"<unnamed>"}`}get value(){let t=G();if(!t)throw new Error("Cannot access tgpu.slot's value outside of resolution.");return ze(t.unwrap(this))}};function ea(e,t){return new zr(e,t)}var zr=class{constructor(t,r=void 0){this.schema=t;this.defaultValue=r;this.slot=Et(r),this[b]=this.slot}resourceType="accessor";slot;[b];$name(t){return this.slot.$name(t),this}toString(){return`accessor:${p(this)??"<unnamed>"}`}get value(){if(!G())throw new Error("Cannot access tgpu.accessor's value outside of resolution.");return new Proxy({"~resolve":r=>r.resolve(this),toString:()=>`.value:${p(this)??"<unnamed>"}`,[m]:{dataType:this.schema}},Y)}"~resolve"(t){let r=t.unwrap(this.slot);return lt(r)?t.resolve(r):Pr(r)?`${t.resolve(r)}()`:t.resolveValue(r,this.schema)}};function ta(e){return Wa(e)}function Ga([e,t]){return`${p(e)??"<unnamed>"}=${t}`}function Wa(e){if(G())throw new Error("Cannot create tgpu.derived objects at the resolution stage.");return{resourceType:"derived","~compute":e,[Nt]:void 0,get value(){let r=G();if(!r)throw new Error("Cannot access tgpu.derived's value outside of resolution.");return ze(r.unwrap(this))},with(r,n){return ra(this,[[r,n]])},toString(){return"derived"}}}function ra(e,t){return{resourceType:"derived",[Nt]:void 0,"~compute"(){throw new Error("'~compute' should never be read on bound derived items.")},"~providing":{inner:e,pairs:t},get value(){let n=G();if(!n)throw new Error("Cannot access tgpu.derived's value outside of resolution.");return ze(n.unwrap(this))},with(n,a){return ra(e,[...t,[n,a]])},toString(){return`derived[${t.map(Ga).join(", ")}]`}}}function na(e,t){return new Lt("private",e,t)}function aa(e){return new Lt("workgroup",e)}var Lt=class{constructor(t,r,n){this.scope=t;this._dataType=r;this._initialValue=n}"~resolve"(t){let r=t.names.makeUnique(p(this));return this._initialValue?t.addDeclaration(`var<${this.scope}> ${r}: ${t.resolve(this._dataType)} = ${t.resolveValue(this._initialValue,this._dataType)};`):t.addDeclaration(`var<${this.scope}> ${r}: ${t.resolve(this._dataType)};`),r}$name(t){return w(this,t),this}toString(){return`var:${p(this)??"<unnamed>"}`}get value(){if(!de())throw new Error("Cannot access tgpu.var's value directly in JS.");return new Proxy({"~resolve":t=>t.resolve(this),toString:()=>`.value:${p(this)??"<unnamed>"}`,[m]:{dataType:this._dataType}},Y)}};var oa={bindGroupLayout:It,vertexLayout:Lr,init:Xn,initFromDevice:Zn,resolve:zn,"~unstable":{fn:Wn,fragmentFn:Nn,vertexFn:jn,computeFn:Gn,vertexLayout:Lr,derived:ta,slot:Et,accessor:ea,privateVar:na,workgroupVar:aa,const:ln,declare:mn,sampler:An,comparisonSampler:In}},jp=oa;Object.assign(oa,{__assignAst:cn,__removedJsImpl:yn});export{le as MissingBindGroupsError,Gt as MissingLinksError,_t as MissingSlotValueError,Ee as MissingVertexBuffersError,Ot as NotUniformError,ie as RandomNameRegistry,we as ResolutionError,ue as StrictNameRegistry,jp as default,ae as isBuffer,Qe as isComparisonSampler,q as isDerived,et as isSampledTextureView,Ye as isSampler,H as isSlot,Ze as isStorageTextureView,X as isTexture,Pr as isTgpuFn,Fa as isUsableAsRender,Sr as isUsableAsSampled,xe as isUsableAsStorage,xt as isUsableAsUniform,Da as isUsableAsVertex,oa as tgpu,mr as unstable_asMutable,fr as unstable_asReadonly,gr as unstable_asUniform};
|
30
|
+
`)}${r}`}finally{Qt(Ge.GPU)}return this._getOrInstantiate(t)}return String(t)}resolveValue(t,r){if(ct(t))return this.resolve(t);if(r&&z(r))return`array(${t.map(n=>this.resolveValue(n,r.elementType))})`;if(Array.isArray(t))return`array(${t.map(n=>this.resolveValue(n))})`;if(r&&A(r))return`${this.resolve(r)}(${Object.entries(r.propTypes).map(([n,a])=>this.resolveValue(t[n],a))})`;throw new Error(`Value ${t} (as json: ${JSON.stringify(t)}) of schema ${r} is not resolvable to WGSL`)}};function $e(e,t){let r=new $r(t),n=r.resolve(e),a=r.bindGroupLayoutsToPlaceholderMap,o=[],i=new Set([...a.keys()].map(m=>m.index).filter(m=>m!==void 0)),s=mn(i),l=r.fixedBindings.map((m,g)=>[String(g),m.layoutEntry]),u=()=>{let m=s.next().value,g=Bt(Object.fromEntries(l));return o[m]=g,n=n.replaceAll(aa,String(m)),[m,new _e(g,Object.fromEntries(r.fixedBindings.map((T,x)=>[String(x),T.resource])))]},c=l.length>0?u():null;for(let[m,g]of a.entries()){let T=m.index??s.next().value;o[T]=m,n=n.replaceAll(g,String(T))}return{code:n,bindGroupLayouts:o,catchall:c}}function so(e,t,r){let n=t.map(a=>`${a.value}: ${e.resolve(a.dataType)}`).join(", ");return r.type!=="void"?`(${n}) -> ${Ue(r)} ${e.resolve(r)}`:`(${n})`}function oa(e){let{externals:t,template:r,names:n,unstable_jitTranspiler:a}=e,o={};ye(o,t??{});let i={"~resolve"(l){return Te(l,o,r??"")},toString:()=>"<root>"},{code:s}=$e(i,{names:n==="strict"?new se:new ie,jitTranspiler:a});return s}function ia(e,t,r,n){return new Er(e,t,r,n)}function rt(e){let t=e;return t?.resourceType==="query-set"&&!!t[d]}var Er=class{constructor(t,r,n,a){this._group=t;this.type=r;this.count=n;this.rawQuerySet=a;this._ownQuerySet=!a,this._querySet=a||null}resourceType="query-set";_querySet=null;_ownQuerySet;_destroyed=!1;_available=!0;_readBuffer=null;_resolveBuffer=null;get querySet(){if(this._destroyed)throw new Error("This QuerySet has been destroyed.");return this.rawQuerySet?this.rawQuerySet:this._querySet?this._querySet:(this._querySet=this._group.device.createQuerySet({type:this.type,count:this.count}),this._querySet)}get destroyed(){return this._destroyed}get available(){return this._available}get[d](){let t=this;return{get readBuffer(){return t._readBuffer||(t._readBuffer=t._group.device.createBuffer({size:t.count*BigUint64Array.BYTES_PER_ELEMENT,usage:GPUBufferUsage.COPY_DST|GPUBufferUsage.MAP_READ})),t._readBuffer},get resolveBuffer(){return t._resolveBuffer||(t._resolveBuffer=t._group.device.createBuffer({size:t.count*BigUint64Array.BYTES_PER_ELEMENT,usage:GPUBufferUsage.QUERY_RESOLVE|GPUBufferUsage.COPY_SRC})),t._resolveBuffer}}}$name(t){return w(this,t),this._querySet&&(this._querySet.label=t),this}resolve(){if(this._destroyed)throw new Error("This QuerySet has been destroyed.");if(!this._available)throw new Error("This QuerySet is busy resolving or reading.");let t=this._group.device.createCommandEncoder();t.resolveQuerySet(this.querySet,0,this.count,this[d].resolveBuffer,0),this._group.device.queue.submit([t.finish()])}async read(){if(this._group.flush(),!this._resolveBuffer)throw new Error("QuerySet must be resolved before reading.");this._available=!1;try{let t=this._group.device.createCommandEncoder();t.copyBufferToBuffer(this[d].resolveBuffer,0,this[d].readBuffer,0,this.count*BigUint64Array.BYTES_PER_ELEMENT),this._group.device.queue.submit([t.finish()]),await this._group.device.queue.onSubmittedWorkDone();let r=this[d].readBuffer;await r.mapAsync(GPUMapMode.READ);let n=new BigUint64Array(r.getMappedRange().slice());return r.unmap(),Array.from(n)}finally{this._available=!0}}destroy(){this._destroyed||(this._destroyed=!0,this._querySet&&this._ownQuerySet&&this._querySet.destroy(),this._readBuffer?.destroy(),this._resolveBuffer?.destroy(),this._readBuffer=this._resolveBuffer=null)}};var nt=class{constructor(t){this._make=t}_map=new WeakMap;getOrMake(t,...r){if(this._map.has(t))return this._map.get(t);let n=this._make(t,...r);return this._map.set(t,n),n}};function _t(e,t,r){if(!r.enabledFeatures.has("timestamp-query"))throw new Error('Performance callback requires the "timestamp-query" feature to be enabled on GPU device.');return e.timestampWrites?{...e,performanceCallback:t}:{...e,performanceCallback:t,hasAutoQuerySet:!0,timestampWrites:{querySet:r.createQuerySet("timestamp",2),beginningOfPassWriteIndex:0,endOfPassWriteIndex:1}}}function $t(e,t,r){if(!r.enabledFeatures.has("timestamp-query"))throw new Error('Timestamp writes require the "timestamp-query" feature to be enabled on GPU device.');e.hasAutoQuerySet&&e.timestampWrites&&e.timestampWrites.querySet.destroy();let n={querySet:t.querySet};return t.beginningOfPassWriteIndex!==void 0&&(n.beginningOfPassWriteIndex=t.beginningOfPassWriteIndex),t.endOfPassWriteIndex!==void 0&&(n.endOfPassWriteIndex=t.endOfPassWriteIndex),{...e,hasAutoQuerySet:!1,timestampWrites:n}}function Et(e,t){if(!e.timestampWrites)return{};let{querySet:r,beginningOfPassWriteIndex:n,endOfPassWriteIndex:a}=e.timestampWrites,o={querySet:rt(r)?t.unwrap(r):r};return n!==void 0&&(o.beginningOfPassWriteIndex=n),a!==void 0&&(o.endOfPassWriteIndex=a),{timestampWrites:o}}function Gt({root:e,priors:t}){let r=t.timestampWrites?.querySet,n=t.performanceCallback;if(!r)throw new Error("Cannot dispatch workgroups with performance callback without a query set.");if(!rt(r))throw new Error("Performance callback with raw GPUQuerySet is not supported. Use TgpuQuerySet instead.");e.commandEncoder.resolveQuerySet(e.unwrap(r),0,r.count,r[d].resolveBuffer,0),e.flush(),e.device.queue.onSubmittedWorkDone().then(async()=>{if(!r.available)return;let a=await r.read(),o=a[t.timestampWrites?.beginningOfPassWriteIndex??0],i=a[t.timestampWrites?.endOfPassWriteIndex??1];if(o===void 0||i===void 0)throw new Error("QuerySet did not return valid timestamps.");await n(o,i)})}function sa(e,t,r){return new Gr(new kr(e,t,r),{})}function ua(e){let t=e;return t?.resourceType==="compute-pipeline"&&!!t[d]}var Gr=class e{constructor(t,r){this._core=t;this._priors=r;this[d]={get rawPipeline(){return t.unwrap().pipeline},get priors(){return r}},this[v]=t}[d];resourceType="compute-pipeline";[v];get rawPipeline(){return this._core.unwrap().pipeline}with(t,r){return new e(this._core,{...this._priors,bindGroupLayoutMap:new Map([...this._priors.bindGroupLayoutMap??[],[t,r]])})}withPerformanceCallback(t){let r=_t(this._priors,t,this._core.branch);return new e(this._core,r)}withTimestampWrites(t){let r=$t(this._priors,t,this._core.branch);return new e(this._core,r)}dispatchWorkgroups(t,r,n){let a=this._core.unwrap(),{branch:o}=this._core,i={label:p(this._core)??"<unnamed>",...Et(this._priors,o)},s=o.commandEncoder.beginComputePass(i);s.setPipeline(a.pipeline);let l=new Set(a.bindGroupLayouts);if(a.bindGroupLayouts.forEach((u,c)=>{if(a.catchall&&c===a.catchall[0])s.setBindGroup(c,o.unwrap(a.catchall[1])),l.delete(u);else{let m=this._priors.bindGroupLayoutMap?.get(u);m!==void 0&&(l.delete(u),s.setBindGroup(c,o.unwrap(m)))}}),l.size>0)throw new me(l);s.dispatchWorkgroups(t,r,n),s.end(),this._priors.performanceCallback&&Gt({root:o,priors:this._priors})}$name(t){return w(this._core,t),this}},kr=class{constructor(t,r,n){this.branch=t;this._slotBindings=r;this._entryFn=n}_memo;unwrap(){if(this._memo===void 0){let t=this.branch.device,{code:r,bindGroupLayouts:n,catchall:a}=$e({"~resolve":o=>(o.withSlots(this._slotBindings,()=>{o.resolve(this._entryFn)}),""),toString:()=>`computePipeline:${p(this)??"<unnamed>"}`},{names:this.branch.nameRegistry,jitTranspiler:this.branch.jitTranspiler});a!==null&&n[a[0]]?.$name(`${p(this)??"<unnamed>"} - Automatic Bind Group & Layout`),this._memo={pipeline:t.createComputePipeline({label:p(this)??"<unnamed>",layout:t.createPipelineLayout({label:`${p(this)??"<unnamed>"} - Pipeline Layout`,bindGroupLayouts:n.map(o=>this.branch.unwrap(o))}),compute:{module:t.createShaderModule({label:`${p(this)??"<unnamed>"} - Shader`,code:r})}}),bindGroupLayouts:n,catchall:a}}return this._memo}};function Wr(e,t="vertex"){return new Or(e,t)}function Lt(e){return e?.resourceType==="vertex-layout"}var Lr=Symbol("defaultAttribEntry");function kt(e,t,r,n,a){if(ot(t)||pn(t)){let o=ee(t);return o!==void 0&&(n[a??Lr]=o),kt(e,t.inner,L(r,Q(t)),n)}if(A(t)){let o=r;return Object.fromEntries(Object.entries(t.propTypes).map(([i,s])=>{o=L(o,V(s));let l=[i,kt(e,s,o,n,i)];return o+=S(s),l}))}if(q(t)){let o=r;return Object.fromEntries(Object.entries(t.propTypes).map(([i,s])=>{o=L(o,Q(s));let l=[i,kt(e,s,o,n,i)];return o+=S(s),l}))}if("type"in t&&typeof t.type=="string"){if(Zr.includes(t.type))return{_layout:e,format:t.type,offset:r};let o=en[t.type];if(o)return{_layout:e,format:o,offset:r}}throw new Error(`Unsupported data used in vertex layout: ${String(t)}`)}var Or=class{constructor(t,r){this.schemaForCount=t;this.stepMode=r;let n=t(0);this.stride=L(S(n.elementType),V(n)),this.attrib=kt(this,n.elementType,0,this._customLocationMap)}resourceType="vertex-layout";stride;attrib;_customLocationMap={};get vertexLayout(){if(this._customLocationMap[Lr]!==void 0){if(typeof this.attrib.format!="string"||typeof this.attrib.offset!="number")throw new Error("Single attribute vertex layouts must have a format and offset.");return{arrayStride:this.stride,stepMode:this.stepMode,attributes:[{format:this.attrib.format,offset:this.attrib.offset,shaderLocation:this._customLocationMap[Lr]}]}}if(!Object.keys(this.attrib).every(r=>this._customLocationMap[r]!==void 0))throw new Error("All attributes must have custom locations in order to unwrap a vertex layout.");return{arrayStride:this.stride,stepMode:this.stepMode,attributes:[...Object.entries(this.attrib).map(([r,n])=>({format:n.format,offset:n.offset,shaderLocation:this._customLocationMap[r]}))]}}$name(t){return w(this,t),this}};function uo(e){return typeof e?.loadOp=="string"}function pa(e,t){if(O(e)){if(!uo(t))throw new Error("Expected a single color attachment, not a record.");return[t]}let r=[];for(let n of Object.keys(e)){let a=t[n];if(!a)throw new Error(`A color attachment by the name of '${n}' was not provided to the shader.`);r.push(a)}return r}function po(e){return typeof e?.format=="string"}function la(e,t){if(O(e)){if(it(e))return[];if(!po(t))throw new Error("Expected a single color target configuration, not a record.");return[t]}let r=[];for(let n of Object.keys(e)){let a=t[n];if(!a)throw new Error(`A color target by the name of '${n}' was not provided to the shader.`);r.push(a)}return r}function ma(e){return new Mr(new Nr(e),{})}function da(e){let t=e;return t?.resourceType==="render-pipeline"&&!!t[d]}var Mr=class e{[d];resourceType="render-pipeline";[v];constructor(t,r){this[d]={core:t,priors:r},this[v]=t}$name(t){return w(this[d].core,t),this}with(t,r){let n=this[d];if(Rt(t))return new e(n.core,{...n.priors,bindGroupLayoutMap:new Map([...n.priors.bindGroupLayoutMap??[],[t,r]])});if(Lt(t))return new e(n.core,{...n.priors,vertexLayoutMap:new Map([...n.priors.vertexLayoutMap??[],[t,r]])});throw new Error("Unsupported value passed into .with()")}withColorAttachment(t){let r=this[d];return new e(r.core,{...r.priors,colorAttachment:t})}withDepthStencilAttachment(t){let r=this[d];return new e(r.core,{...r.priors,depthStencilAttachment:t})}withPerformanceCallback(t){let r=this[d],n=_t(r.priors,t,r.core.options.branch);return new e(r.core,n)}withTimestampWrites(t){let r=this[d],n=$t(r.priors,t,r.core.options.branch);return new e(r.core,n)}draw(t,r,n,a){let o=this[d],i=o.core.unwrap(),{branch:s,fragmentFn:l}=o.core.options,u=pa(l.shell.targets,o.priors.colorAttachment??{}).map(y=>X(y.view)?{...y,view:s.unwrap(y.view).createView()}:y),c={label:p(o.core)??"<unnamed>",colorAttachments:u,...Et(o.priors,s)};if(o.priors.depthStencilAttachment!==void 0){let y=o.priors.depthStencilAttachment;X(y.view)?c.depthStencilAttachment={...y,view:s.unwrap(y.view).createView()}:c.depthStencilAttachment=y}let m=s.commandEncoder.beginRenderPass(c);m.setPipeline(i.pipeline);let g=new Set(i.bindGroupLayouts);i.bindGroupLayouts.forEach((y,F)=>{if(i.catchall&&F===i.catchall[0])m.setBindGroup(F,s.unwrap(i.catchall[1])),g.delete(y);else{let U=o.priors.bindGroupLayoutMap?.get(y);U!==void 0&&(g.delete(y),m.setBindGroup(F,s.unwrap(U)))}});let T=new Set(o.core.usedVertexLayouts);if(o.core.usedVertexLayouts.forEach((y,F)=>{let U=o.priors.vertexLayoutMap?.get(y);U&&(T.delete(y),m.setVertexBuffer(F,s.unwrap(U)))}),g.size>0)throw new me(g);if(T.size>0)throw new Ee(T);m.draw(t,r,n,a),m.end(),o.priors.performanceCallback?Gt({root:s,priors:o.priors}):s.flush()}},Nr=class{constructor(t){this.options=t;let r=kn(t.vertexFn.shell.attributes[0],t.vertexAttribs);this._vertexBufferLayouts=r.bufferDefinitions,this.usedVertexLayouts=r.usedVertexLayouts,this._targets=la(t.fragmentFn.shell.targets,t.targets)}usedVertexLayouts;_memo;_vertexBufferLayouts;_targets;unwrap(){if(this._memo===void 0){let{branch:t,vertexFn:r,fragmentFn:n,slotBindings:a,primitiveState:o,depthStencilState:i,multisampleState:s}=this.options,{code:l,bindGroupLayouts:u,catchall:c}=$e({"~resolve":y=>(y.withSlots(a,()=>{y.resolve(r),y.resolve(n)}),""),toString:()=>`renderPipeline:${p(this)??"<unnamed>"}`},{names:t.nameRegistry,jitTranspiler:t.jitTranspiler});c!==null&&u[c[0]]?.$name(`${p(this)??"<unnamed>"} - Automatic Bind Group & Layout`);let m=t.device,g=m.createShaderModule({label:`${p(this)??"<unnamed>"} - Shader`,code:l}),T={layout:m.createPipelineLayout({label:`${p(this)??"<unnamed>"} - Pipeline Layout`,bindGroupLayouts:u.map(y=>t.unwrap(y))}),vertex:{module:g,buffers:this._vertexBufferLayouts},fragment:{module:g,targets:this._targets}},x=p(this);x!==void 0&&(T.label=x),o&&(T.primitive=o),i&&(T.depthStencil=i),s&&(T.multisample=s),this._memo={pipeline:m.createRenderPipeline(T),bindGroupLayouts:u,catchall:c}}return this._memo}};var jr=class e{constructor(t,r){this._getRoot=t;this._slotBindings=r}with(t,r){return new e(this._getRoot,[...this._slotBindings,[qe(t)?t.slot:t,r]])}withCompute(t){return new zr(this._getRoot(),this._slotBindings,t)}withVertex(t,r){return new qr({branch:this._getRoot(),primitiveState:void 0,depthStencilState:void 0,slotBindings:this._slotBindings,vertexFn:t,vertexAttribs:r,multisampleState:void 0})}},zr=class{constructor(t,r,n){this._root=t;this._slotBindings=r;this._entryFn=n}createPipeline(){return sa(this._root,this._slotBindings,this._entryFn)}},qr=class{constructor(t){this._options=t}withFragment(t,r,n){return le(typeof t!="string","Just type mismatch validation"),le(typeof r!="string","Just type mismatch validation"),new Qr({...this._options,fragmentFn:t,targets:r})}},Qr=class e{constructor(t){this._options=t}withPrimitive(t){return new e({...this._options,primitiveState:t})}withDepthStencil(t){return new e({...this._options,depthStencilState:t})}withMultisample(t){return new e({...this._options,multisampleState:t})}createPipeline(){return ma(this._options)}},Ot=class extends jr{constructor(r,n,a,o){super(()=>this,[]);this.device=r;this.nameRegistry=n;this.jitTranspiler=a;this._ownDevice=o;this["~unstable"]=this}"~unstable";_disposables=[];_unwrappedBindGroupLayouts=new nt(r=>r.unwrap(this));_unwrappedBindGroups=new nt(r=>r.unwrap(this));_commandEncoder=null;get commandEncoder(){return this._commandEncoder||(this._commandEncoder=this.device.createCommandEncoder()),this._commandEncoder}get enabledFeatures(){return new Set(this.device.features)}createBuffer(r,n){let a=qn(this,r,n);return this._disposables.push(a),a}createUniform(r,n){return this.createBuffer(r,n).$usage("uniform").as("uniform")}createMutable(r,n){return this.createBuffer(r,n).$usage("storage").as("mutable")}createReadonly(r,n){return this.createBuffer(r,n).$usage("storage").as("readonly")}createQuerySet(r,n,a){return ia(this,r,n,a)}createBindGroup(r,n){return new _e(r,n)}destroy(){for(let r of this._disposables)r.destroy();this._ownDevice&&this.device.destroy()}createTexture(r){let n=Xn(r,this);return this._disposables.push(n),n}unwrap(r){if(ua(r))return r[d].rawPipeline;if(da(r))return r[d].core.unwrap().pipeline;if(Rt(r))return this._unwrappedBindGroupLayouts.getOrMake(r);if(Br(r))return this._unwrappedBindGroups.getOrMake(r);if(pe(r))return r.buffer;if(X(r))return r[d].unwrap();if(et(r)){if(r[d].unwrap)return r[d].unwrap();throw new Error("Cannot unwrap laid-out texture view.")}if(tt(r)){if(r[d].unwrap)return r[d].unwrap();throw new Error("Cannot unwrap laid-out texture view.")}if(Lt(r))return r.vertexLayout;if(Ye(r)){if(r[d].unwrap)return r[d].unwrap(this);throw new Error("Cannot unwrap laid-out sampler.")}if(Xe(r)){if(r[d].unwrap)return r[d].unwrap(this);throw new Error("Cannot unwrap laid-out comparison sampler.")}if(rt(r))return r.querySet;throw new Error(`Unknown resource type: ${r}`)}beginRenderPass(r,n){let a=this.commandEncoder.beginRenderPass(r),o=new Map,i=new Map,s,l=()=>{if(!s)throw new Error("Cannot draw without a call to pass.setPipeline");let{core:u,priors:c}=s[d],m=u.unwrap();a.setPipeline(m.pipeline);let g=new Set(m.bindGroupLayouts);m.bindGroupLayouts.forEach((x,y)=>{if(m.catchall&&y===m.catchall[0])a.setBindGroup(y,this.unwrap(m.catchall[1])),g.delete(x);else{let F=c.bindGroupLayoutMap?.get(x)??o.get(x);F!==void 0&&(g.delete(x),Br(F)?a.setBindGroup(y,this.unwrap(F)):a.setBindGroup(y,F))}});let T=new Set;if(u.usedVertexLayouts.forEach((x,y)=>{let F=c.vertexLayoutMap?.get(x),U=F?{buffer:F,offset:void 0,size:void 0}:i.get(x);!U||!U.buffer?T.add(x):pe(U.buffer)?a.setVertexBuffer(y,this.unwrap(U.buffer),U.offset,U.size):a.setVertexBuffer(y,U.buffer,U.offset,U.size)}),g.size>0)throw new me(g);if(T.size>0)throw new Ee(T)};n({setViewport(...u){a.setViewport(...u)},setScissorRect(...u){a.setScissorRect(...u)},setBlendConstant(...u){a.setBlendConstant(...u)},setStencilReference(...u){a.setStencilReference(...u)},beginOcclusionQuery(...u){a.beginOcclusionQuery(...u)},endOcclusionQuery(...u){a.endOcclusionQuery(...u)},executeBundles(...u){a.executeBundles(...u)},setPipeline(u){s=u},setIndexBuffer:(u,c,m,g)=>{pe(u)?a.setIndexBuffer(this.unwrap(u),c,m,g):a.setIndexBuffer(u,c,m,g)},setVertexBuffer(u,c,m,g){i.set(u,{buffer:c,offset:m,size:g})},setBindGroup(u,c){o.set(u,c)},draw(u,c,m,g){l(),a.draw(u,c,m,g)},drawIndexed(...u){l(),a.drawIndexed(...u)},drawIndirect(...u){l(),a.drawIndirect(...u)},drawIndexedIndirect(...u){l(),a.drawIndexedIndirect(...u)}}),a.end()}flush(){this._commandEncoder&&(this.device.queue.submit([this._commandEncoder.finish()]),this._commandEncoder=null)}};async function fa(e){let{adapter:t,device:r,unstable_names:n="random",unstable_jitTranspiler:a}=e??{};if(!navigator.gpu)throw new Error("WebGPU is not supported by this browser.");let o=await navigator.gpu.requestAdapter(t);if(!o)throw new Error("Could not find a compatible GPU");let i=[];for(let s of r?.requiredFeatures??[]){if(!o.features.has(s))throw new Error(`Requested feature "${s}" is not supported by the adapter.`);i.push(s)}for(let s of r?.optionalFeatures??[])o.features.has(s)?i.push(s):console.warn(`Optional feature "${s}" is not supported by the adapter.`);return new Ot(await o.requestDevice({...r,requiredFeatures:i}),n==="random"?new ie:new se,a,!0)}function ca(e){let{device:t,unstable_names:r="random",unstable_jitTranspiler:n}=e??{};return new Ot(t,r==="random"?new ie:new se,n,!1)}function Wt(e){return new Kr(e)}var Kr=class{constructor(t=void 0){this.defaultValue=t}resourceType="slot";$name(t){return w(this,t),this}areEqual(t,r){return Object.is(t,r)}toString(){return`slot:${p(this)??"<unnamed>"}`}[I](t){return je(t,t.unwrap(this))}get value(){let t=Z();if(!t)throw new Error("Cannot access tgpu.slot's value outside of resolution.");return this[I](t)}};function ga(e,t){return new Hr(e,t)}var Hr=class{constructor(t,r=void 0){this.schema=t;this.defaultValue=r;this.slot=Wt(r),this[v]=this.slot}resourceType="accessor";slot;$name(t){return this.slot.$name(t),this}toString(){return`accessor:${p(this)??"<unnamed>"}`}[I](){return new Proxy({"~resolve":t=>t.resolve(this),toString:()=>`.value:${p(this)??"<unnamed>"}`,[b]:this.schema},K)}get value(){if(!J())throw new Error("`tgpu.accessor` values are only accessible on the GPU");return this[I]()}"~resolve"(t){let r=t.unwrap(this.slot);return Sn(r)?t.resolve(r):mr(r)?`${t.resolve(r)}()`:t.resolveValue(r,this.schema)}};function ya(e){return mo(e)}function lo([e,t]){return`${p(e)??"<unnamed>"}=${t}`}function mo(e){if(Z())throw new Error("Cannot create tgpu.derived objects at the resolution stage.");return{resourceType:"derived","~compute":e,[Kt]:void 0,"~gpuRepr":void 0,[I](r){return je(r,r.unwrap(this))},get value(){let r=Z();if(!r)throw new Error("Cannot access tgpu.derived's value outside of resolution.");return this[I](r)},with(r,n){return Ta(this,[[r,n]])},toString(){return"derived"}}}function Ta(e,t){return{resourceType:"derived",[Kt]:void 0,"~gpuRepr":void 0,"~compute"(){throw new Error("'~compute' should never be read on bound derived items.")},"~providing":{inner:e,pairs:t},[I](n){return je(n,n.unwrap(this))},get value(){let n=Z();if(!n)throw new Error("Cannot access tgpu.derived's value outside of resolution.");return this[I](n)},with(n,a){return Ta(e,[...t,[n,a]])},toString(){return`derived[${t.map(lo).join(", ")}]`}}}function xa(e,t){return new Mt("private",e,t)}function ha(e){return new Mt("workgroup",e)}var Mt=class{constructor(t,r,n){this.scope=t;this._dataType=r;this._initialValue=n;this[d]={scope:t}}"~resolve"(t){let r=t.names.makeUnique(p(this));return this._initialValue?t.addDeclaration(`var<${this.scope}> ${r}: ${t.resolve(this._dataType)} = ${t.resolveValue(this._initialValue,this._dataType)};`):t.addDeclaration(`var<${this.scope}> ${r}: ${t.resolve(this._dataType)};`),r}$name(t){return w(this,t),this}toString(){return`var:${p(this)??"<unnamed>"}`}[I](){return new Proxy({"~resolve":t=>t.resolve(this),toString:()=>`.value:${p(this)??"<unnamed>"}`,[b]:this._dataType},K)}get value(){if(!J())throw new Error("`tgpu.var` values are only accessible on the GPU");return this[I]()}};var wa={bindGroupLayout:Bt,vertexLayout:Wr,init:fa,initFromDevice:ca,resolve:oa,"~unstable":{fn:_n,fragmentFn:En,vertexFn:Gn,computeFn:Vn,vertexLayout:Wr,derived:ya,slot:Wt,accessor:ga,privateVar:xa,workgroupVar:ha,const:wn,declare:Dn,sampler:Qn,comparisonSampler:Kn}},Vl=wa;Object.assign(wa,{__assignAst:Fn,__removedJsImpl:In});export{me as MissingBindGroupsError,zt as MissingLinksError,Nt as MissingSlotValueError,Ee as MissingVertexBuffersError,jt as NotUniformError,ie as RandomNameRegistry,H as ResolutionError,se as StrictNameRegistry,Vl as default,pe as isBuffer,Xe as isComparisonSampler,Be as isDerived,tt as isSampledTextureView,Ye as isSampler,Pe as isSlot,et as isStorageTextureView,X as isTexture,mr as isTgpuFn,to as isUsableAsRender,Ur as isUsableAsSampled,he as isUsableAsStorage,bt as isUsableAsUniform,eo as isUsableAsVertex,wa as tgpu,Tr as unstable_asMutable,xr as unstable_asReadonly,hr as unstable_asUniform};
|
30
31
|
//# sourceMappingURL=index.js.map
|