tla-ts 0.0.3-dev.6979965 → 0.0.3-dev.79f7ed7

package/dist/parallel-worker.cjs

@@ -6,7 +6,7 @@
  * This source code is licensed under the MIT license found in the
  * LICENSE file in the root directory of this source tree.
  */
-const e=require("./transfer.cjs"),t=e=>{let t=e.reduce((e,t)=>e+t,0),n=t/e.length;if(Math.max(...e)-Math.min(...e)<=Math.max(64,n/4))return[];let r=e.map((r,i)=>Math.floor(n)+ +(i<t%e.length)),i=e.flatMap((e,t)=>e>r[t]?[{index:t,amount:e-r[t]}]:[]),a=e.flatMap((e,t)=>e<r[t]?[{index:t,amount:r[t]-e}]:[]),o=[];for(let e of i)for(let t of a){let n=Math.min(e.amount,t.amount);n!==0&&(o.push({from:e.index,to:t.index,count:n}),e.amount-=n,t.amount-=n)}return o},n=async()=>typeof globalThis.postMessage==`function`?globalThis:(await import(
+const e=require("./transfer.cjs"),t=async()=>typeof globalThis.postMessage==`function`?globalThis:(await import(
 /* @vite-ignore */
 /* webpackIgnore: true */
-`node:worker_threads`)).parentPort,r=()=>{let e=/* @__PURE__ */ new Map,t=/* @__PURE__ */ new Map,n=(e,t)=>`${e}:${t}`,r=n=>{let r=t.get(n),i=e.get(n)??[];!r||i.length<r.count||(t.delete(n),e.delete(n),r.resolve(i))};return{deliver:t=>{let i=n(t.phase,t.level);e.set(i,[...e.get(i)??[],t]),r(i)},receive:(e,i,a)=>new Promise(o=>{let s=n(e,i);t.set(s,{count:a,resolve:o}),r(s)})}},i=(n,i)=>{let a=new Map(n.modules),o=e=>a.get(e),s=new Map(n.classes);e.O(e._(n.source,n.config,o));let c=e.D(),l=e.y(n.config,{source:n.source,loadModule:o,loadClass:e=>s.get(e),print:()=>{},files:{readFile:void 0,writeFile:void 0}}).environment,u=e.u(n.config,l,()=>e.v(n.source,o)),d=u?.spec.next;if(!u||d===void 0)return i.postMessage({type:`abort`});let f=n.index,p=e.T(),m=e.t(l.variables.length),h={fingerprints:n.fingerprints},g=u.isSymmetric?e.r():void 0,_=/* @__PURE__ */ new Int32Array(2),v=r(),y=n.ports.flatMap((e,t)=>e?[{port:e,index:t}]:[]);y.forEach(e=>{e.port.addEventListener(`message`,e=>v.deliver(e.data)),e.port.start?.()});let b={states:m(n.seed).map(p)},x=(e,t,r)=>n.ports[e]?.postMessage(t,r),S=t=>{let r=b.states;for(;t.origin<r.length;t.origin++){let i=r[t.origin];if(!t.successors&&(t.successors=e.w(d,i,l,u.spec.scope),t.generated+=t.successors.length,t.successors.length===0&&n.config.checkDeadlock))return;let a=t.successors;for(;t.index<a.length;t.index++){if(e.s(h.fingerprints))return`paused`;let n=a[t.index];if(u.findViolatedActionProperty(i,n.state))return;if(!u.satisfiesActionConstraints(i,n.state)||g&&(e.c(n.state,_),!g.insert(_[0],_[1])))continue;p(n.state);let r=u.getFingerprintState(n.state);if(e.c(r,_),e.o(h.fingerprints,_[0],_[1])){if(u.findViolatedInvariant(n.state,i))return;u.satisfiesConstraints(n.state,i)&&(t.next.push(n.state),t.fingerprintStates.push(r),t.newStates[t.origin]++)}}t.successors=void 0,t.index=0}return`done`},C=async(n,r,i)=>{let a=t(i);if(a.length===0)return!0;let o=new Map(y.map(t=>[t.index,e.n()]));for(let e of a.filter(e=>e.from===f)){let t=o.get(e.to),n=r.splice(r.length-e.count,e.count);if(!t||!n.every(t.encode))return!1}return o.forEach((e,t)=>{let r=e.finish();x(t,{phase:`balance`,level:n,from:f,states:r},[r.words.buffer])}),(await v.receive(`balance`,n,y.length)).forEach(e=>r.push(...m(e.states).map(p))),!0},w={resume:()=>{},balance:e=>{}},T=e=>i.postMessage(e,e.type===`level`?[e.fingerprints.buffer]:[]),E=async t=>{let n={origin:0,successors:void 0,index:0,generated:0,newStates:new Int32Array(b.states.length),next:[],fingerprintStates:[]};for(let e=S(n);e!==`done`;e=S(n)){if(e===void 0)return;let t=new Promise(e=>w.resume=e);T({type:`paused`}),await t}if(e.D()!==c)return;let r=n.next,i=r.length,a=Array.from({length:32},()=>0);n.newStates.forEach(e=>a[Math.min(e,31)]++);let o=u.getStoredFingerprints(r,n.fingerprintStates),s=new Promise(e=>w.balance=e);if(T({type:`expanded`,from:f,added:i}),await C(t,r,await s))return b.states=r,{type:`level`,level:t,generated:n.generated,added:i,outDegrees:a,fingerprints:o}},D=e=>{E(e).then(e=>T(e??{type:`abort`}),()=>T({type:`abort`}))};i.addEventListener(`message`,e=>{let t=e.data;t.type===`level`&&D(t.level),t.type===`resize`&&(h.fingerprints=t.fingerprints,w.resume()),t.type===`balance`&&w.balance(t.sizes)}),T({type:`ready`})};n().then(e=>{let t=n=>{let r=n.data;if(r.type===`start`){e.removeEventListener?.(`message`,t);try{i(r,e)}catch{e.postMessage({type:`abort`})}}};e.addEventListener(`message`,t),e.start?.()});
+`node:worker_threads`)).parentPort,n=(t,n)=>{let r=new Map(t.modules),i=e=>r.get(e),a=new Map(t.classes);e.O(e._(t.source,t.config,i));let o=e.D(),s=e.y(t.config,{source:t.source,loadModule:i,loadClass:e=>a.get(e),print:()=>{},files:{readFile:void 0,writeFile:void 0}}).environment,c=e.u(t.config,s,()=>e.v(t.source,i)),l=c?.spec.next;if(!c||l===void 0)return n.postMessage({type:`abort`});let u=t.index,d=e.T(),f=e.t(s.variables.length),p={fingerprints:t.fingerprints},m=t.requests,h=t.remaining,g=c.isSymmetric?e.r():void 0,_=/* @__PURE__ */ new Int32Array(2),v={resume:()=>{},loot:e=>{}};t.ports.forEach(e=>{e?.addEventListener(`message`,e=>v.loot(e.data)),e?.start?.()});let y={states:f(t.seed).map(d)},b=(n,r)=>{let i=Atomics.load(m,u)-1,a=Math.floor((n.end-n.origin-1)/2),o=e.n(),s=n.states.slice(n.end-a,n.end),c=a>0&&s.every(o.encode);c&&(n.end-=a);let l=c?o.finish():e.n().finish();t.ports[i]?.postMessage(l,[l.words.buffer]),Atomics.store(m,u,r),Atomics.notify(m,u)},x=e=>{Atomics.store(h,u,0),Atomics.compareExchange(m,u,0,-1)!==0&&b(e,-1)},S=e=>{Atomics.store(h,u,e.end-e.origin),Atomics.store(m,u,0)},C=n=>{for(;n.origin<n.end;n.origin++){let r=n.states[n.origin];if(!n.successors&&(Atomics.load(m,u)>0&&b(n,0),Atomics.store(h,u,n.end-n.origin-1),n.successors=e.w(l,r,s,c.spec.scope),n.generated+=n.successors.length,n.successors.length===0&&t.config.checkDeadlock))return;let i=n.successors;for(;n.index<i.length;n.index++){if(e.s(p.fingerprints))return`paused`;let t=i[n.index];if(c.findViolatedActionProperty(r,t.state))return;if(!c.satisfiesActionConstraints(r,t.state)||g&&(e.c(t.state,_),!g.insert(_[0],_[1])))continue;d(t.state);let a=c.getFingerprintState(t.state);if(e.c(a,_),e.o(p.fingerprints,_[0],_[1])){if(c.findViolatedInvariant(t.state,r))return;c.satisfiesConstraints(t.state,r)&&(n.next.push(t.state),n.fingerprintStates.push(a),n.added++)}}n.outDegrees[Math.min(n.added,31)]++,n.successors=void 0,n.index=0,n.added=0}return`done`},w=async e=>{let n=/* @__PURE__ */ new Set;for(;;){let r=t.ports.reduce((e,t,r)=>!t||n.has(r)||Atomics.load(h,r)<2?e:e===void 0||Atomics.load(h,r)>Atomics.load(h,e)?r:e,void 0);if(r===void 0)return!1;let i=Atomics.compareExchange(m,r,0,u+1);if(i===-1&&n.add(r),i>0&&Atomics.wait(m,r,i),i!==0)continue;let a=await new Promise(e=>v.loot=e);if(a.count===0)n.add(r);else return e.states=f(a).map(d),e.origin=0,e.end=e.states.length,S(e),!0}},T=e=>n.postMessage(e,e.type===`level`?[e.fingerprints.buffer]:[]),E=async t=>{let n={states:y.states,origin:0,end:y.states.length,successors:void 0,index:0,added:0,generated:0,outDegrees:Array.from({length:32},()=>0),next:[],fingerprintStates:[]};for(;;){let e=C(n);if(e===void 0)return;if(x(n),e===`paused`){let e=new Promise(e=>v.resume=e);T({type:`paused`}),await e,S(n)}else if(!await w(n))break}if(e.D()===o)return y.states=n.next,Atomics.store(h,u,y.states.length),{type:`level`,level:t,generated:n.generated,added:n.next.length,outDegrees:n.outDegrees,fingerprints:c.getStoredFingerprints(n.next,n.fingerprintStates)}},D=e=>{E(e).then(e=>T(e??{type:`abort`}),()=>T({type:`abort`}))};n.addEventListener(`message`,e=>{let t=e.data;t.type===`level`&&D(t.level),t.type===`resize`&&(p.fingerprints=t.fingerprints,v.resume())}),Atomics.store(h,u,y.states.length),T({type:`ready`})};t().then(e=>{let t=r=>{let i=r.data;if(i.type===`start`){e.removeEventListener?.(`message`,t);try{n(i,e)}catch{e.postMessage({type:`abort`})}}};e.addEventListener(`message`,t),e.start?.()});

package/dist/parallel-worker.js

@@ -6,7 +6,7 @@
  * This source code is licensed under the MIT license found in the
  * LICENSE file in the root directory of this source tree.
  */
-import{D as e,O as t,T as n,_ as r,c as i,n as a,o,r as s,s as c,t as l,u,v as d,w as f,y as p}from"./transfer.js";const m=e=>{let t=e.reduce((e,t)=>e+t,0),n=t/e.length;if(Math.max(...e)-Math.min(...e)<=Math.max(64,n/4))return[];let r=e.map((r,i)=>Math.floor(n)+ +(i<t%e.length)),i=e.flatMap((e,t)=>e>r[t]?[{index:t,amount:e-r[t]}]:[]),a=e.flatMap((e,t)=>e<r[t]?[{index:t,amount:r[t]-e}]:[]),o=[];for(let e of i)for(let t of a){let n=Math.min(e.amount,t.amount);n!==0&&(o.push({from:e.index,to:t.index,count:n}),e.amount-=n,t.amount-=n)}return o},h=async()=>typeof globalThis.postMessage==`function`?globalThis:(await import(
+import{D as e,O as t,T as n,_ as r,c as i,n as a,o,r as s,s as c,t as l,u,v as d,w as f,y as p}from"./transfer.js";const m=async()=>typeof globalThis.postMessage==`function`?globalThis:(await import(
 /* @vite-ignore */
 /* webpackIgnore: true */
-`node:worker_threads`)).parentPort,g=()=>{let e=/* @__PURE__ */ new Map,t=/* @__PURE__ */ new Map,n=(e,t)=>`${e}:${t}`,r=n=>{let r=t.get(n),i=e.get(n)??[];!r||i.length<r.count||(t.delete(n),e.delete(n),r.resolve(i))};return{deliver:t=>{let i=n(t.phase,t.level);e.set(i,[...e.get(i)??[],t]),r(i)},receive:(e,i,a)=>new Promise(o=>{let s=n(e,i);t.set(s,{count:a,resolve:o}),r(s)})}},_=(h,_)=>{let v=new Map(h.modules),y=e=>v.get(e),b=new Map(h.classes);t(r(h.source,h.config,y));let x=e(),S=p(h.config,{source:h.source,loadModule:y,loadClass:e=>b.get(e),print:()=>{},files:{readFile:void 0,writeFile:void 0}}).environment,C=u(h.config,S,()=>d(h.source,y)),w=C?.spec.next;if(!C||w===void 0)return _.postMessage({type:`abort`});let T=h.index,E=n(),D=l(S.variables.length),O={fingerprints:h.fingerprints},k=C.isSymmetric?s():void 0,A=/* @__PURE__ */ new Int32Array(2),j=g(),M=h.ports.flatMap((e,t)=>e?[{port:e,index:t}]:[]);M.forEach(e=>{e.port.addEventListener(`message`,e=>j.deliver(e.data)),e.port.start?.()});let N={states:D(h.seed).map(E)},P=(e,t,n)=>h.ports[e]?.postMessage(t,n),F=e=>{let t=N.states;for(;e.origin<t.length;e.origin++){let n=t[e.origin];if(!e.successors&&(e.successors=f(w,n,S,C.spec.scope),e.generated+=e.successors.length,e.successors.length===0&&h.config.checkDeadlock))return;let r=e.successors;for(;e.index<r.length;e.index++){if(c(O.fingerprints))return`paused`;let t=r[e.index];if(C.findViolatedActionProperty(n,t.state))return;if(!C.satisfiesActionConstraints(n,t.state)||k&&(i(t.state,A),!k.insert(A[0],A[1])))continue;E(t.state);let a=C.getFingerprintState(t.state);if(i(a,A),o(O.fingerprints,A[0],A[1])){if(C.findViolatedInvariant(t.state,n))return;C.satisfiesConstraints(t.state,n)&&(e.next.push(t.state),e.fingerprintStates.push(a),e.newStates[e.origin]++)}}e.successors=void 0,e.index=0}return`done`},I=async(e,t,n)=>{let r=m(n);if(r.length===0)return!0;let i=new Map(M.map(e=>[e.index,a()]));for(let e of r.filter(e=>e.from===T)){let n=i.get(e.to),r=t.splice(t.length-e.count,e.count);if(!n||!r.every(n.encode))return!1}return i.forEach((t,n)=>{let r=t.finish();P(n,{phase:`balance`,level:e,from:T,states:r},[r.words.buffer])}),(await j.receive(`balance`,e,M.length)).forEach(e=>t.push(...D(e.states).map(E))),!0},L={resume:()=>{},balance:e=>{}},R=e=>_.postMessage(e,e.type===`level`?[e.fingerprints.buffer]:[]),z=async t=>{let n={origin:0,successors:void 0,index:0,generated:0,newStates:new Int32Array(N.states.length),next:[],fingerprintStates:[]};for(let e=F(n);e!==`done`;e=F(n)){if(e===void 0)return;let t=new Promise(e=>L.resume=e);R({type:`paused`}),await t}if(e()!==x)return;let r=n.next,i=r.length,a=Array.from({length:32},()=>0);n.newStates.forEach(e=>a[Math.min(e,31)]++);let o=C.getStoredFingerprints(r,n.fingerprintStates),s=new Promise(e=>L.balance=e);if(R({type:`expanded`,from:T,added:i}),await I(t,r,await s))return N.states=r,{type:`level`,level:t,generated:n.generated,added:i,outDegrees:a,fingerprints:o}},B=e=>{z(e).then(e=>R(e??{type:`abort`}),()=>R({type:`abort`}))};_.addEventListener(`message`,e=>{let t=e.data;t.type===`level`&&B(t.level),t.type===`resize`&&(O.fingerprints=t.fingerprints,L.resume()),t.type===`balance`&&L.balance(t.sizes)}),R({type:`ready`})};h().then(e=>{let t=n=>{let r=n.data;if(r.type===`start`){e.removeEventListener?.(`message`,t);try{_(r,e)}catch{e.postMessage({type:`abort`})}}};e.addEventListener(`message`,t),e.start?.()});
+`node:worker_threads`)).parentPort,h=(m,h)=>{let g=new Map(m.modules),_=e=>g.get(e),v=new Map(m.classes);t(r(m.source,m.config,_));let y=e(),b=p(m.config,{source:m.source,loadModule:_,loadClass:e=>v.get(e),print:()=>{},files:{readFile:void 0,writeFile:void 0}}).environment,x=u(m.config,b,()=>d(m.source,_)),S=x?.spec.next;if(!x||S===void 0)return h.postMessage({type:`abort`});let C=m.index,w=n(),T=l(b.variables.length),E={fingerprints:m.fingerprints},D=m.requests,O=m.remaining,k=x.isSymmetric?s():void 0,A=/* @__PURE__ */ new Int32Array(2),j={resume:()=>{},loot:e=>{}};m.ports.forEach(e=>{e?.addEventListener(`message`,e=>j.loot(e.data)),e?.start?.()});let M={states:T(m.seed).map(w)},N=(e,t)=>{let n=Atomics.load(D,C)-1,r=Math.floor((e.end-e.origin-1)/2),i=a(),o=e.states.slice(e.end-r,e.end),s=r>0&&o.every(i.encode);s&&(e.end-=r);let c=s?i.finish():a().finish();m.ports[n]?.postMessage(c,[c.words.buffer]),Atomics.store(D,C,t),Atomics.notify(D,C)},P=e=>{Atomics.store(O,C,0),Atomics.compareExchange(D,C,0,-1)!==0&&N(e,-1)},F=e=>{Atomics.store(O,C,e.end-e.origin),Atomics.store(D,C,0)},I=e=>{for(;e.origin<e.end;e.origin++){let t=e.states[e.origin];if(!e.successors&&(Atomics.load(D,C)>0&&N(e,0),Atomics.store(O,C,e.end-e.origin-1),e.successors=f(S,t,b,x.spec.scope),e.generated+=e.successors.length,e.successors.length===0&&m.config.checkDeadlock))return;let n=e.successors;for(;e.index<n.length;e.index++){if(c(E.fingerprints))return`paused`;let r=n[e.index];if(x.findViolatedActionProperty(t,r.state))return;if(!x.satisfiesActionConstraints(t,r.state)||k&&(i(r.state,A),!k.insert(A[0],A[1])))continue;w(r.state);let a=x.getFingerprintState(r.state);if(i(a,A),o(E.fingerprints,A[0],A[1])){if(x.findViolatedInvariant(r.state,t))return;x.satisfiesConstraints(r.state,t)&&(e.next.push(r.state),e.fingerprintStates.push(a),e.added++)}}e.outDegrees[Math.min(e.added,31)]++,e.successors=void 0,e.index=0,e.added=0}return`done`},L=async e=>{let t=/* @__PURE__ */ new Set;for(;;){let n=m.ports.reduce((e,n,r)=>!n||t.has(r)||Atomics.load(O,r)<2?e:e===void 0||Atomics.load(O,r)>Atomics.load(O,e)?r:e,void 0);if(n===void 0)return!1;let r=Atomics.compareExchange(D,n,0,C+1);if(r===-1&&t.add(n),r>0&&Atomics.wait(D,n,r),r!==0)continue;let i=await new Promise(e=>j.loot=e);if(i.count===0)t.add(n);else return e.states=T(i).map(w),e.origin=0,e.end=e.states.length,F(e),!0}},R=e=>h.postMessage(e,e.type===`level`?[e.fingerprints.buffer]:[]),z=async t=>{let n={states:M.states,origin:0,end:M.states.length,successors:void 0,index:0,added:0,generated:0,outDegrees:Array.from({length:32},()=>0),next:[],fingerprintStates:[]};for(;;){let e=I(n);if(e===void 0)return;if(P(n),e===`paused`){let e=new Promise(e=>j.resume=e);R({type:`paused`}),await e,F(n)}else if(!await L(n))break}if(e()===y)return M.states=n.next,Atomics.store(O,C,M.states.length),{type:`level`,level:t,generated:n.generated,added:n.next.length,outDegrees:n.outDegrees,fingerprints:x.getStoredFingerprints(n.next,n.fingerprintStates)}},B=e=>{z(e).then(e=>R(e??{type:`abort`}),()=>R({type:`abort`}))};h.addEventListener(`message`,e=>{let t=e.data;t.type===`level`&&B(t.level),t.type===`resize`&&(E.fingerprints=t.fingerprints,j.resume())}),Atomics.store(O,C,M.states.length),R({type:`ready`})};m().then(e=>{let t=n=>{let r=n.data;if(r.type===`start`){e.removeEventListener?.(`message`,t);try{h(r,e)}catch{e.postMessage({type:`abort`})}}};e.addEventListener(`message`,t),e.start?.()});

package/dist/parallel.cjs

@@ -9,4 +9,4 @@
 const e=require("./transfer.cjs"),t=async e=>{if(typeof Worker==`function`){let t=new Worker(new URL(`./parallel-worker.js`,require("url").pathToFileURL(__filename).href),{type:`module`});return t.addEventListener(`message`,t=>e(t.data)),t.addEventListener(`error`,()=>e({type:`abort`})),t}let t=new(await(import(
 /* @vite-ignore */
 /* webpackIgnore: true */
-`node:worker_threads`))).Worker(new URL(`./parallel-worker.js`,require("url").pathToFileURL(__filename).href));return t.on(`message`,e),t.on(`error`,()=>e({type:`abort`})),{postMessage:(e,n)=>t.postMessage(e,n),terminate:()=>void t.terminate()}},n=async n=>{let r=typeof n.config==`string`?e.A(n.config):n.config,i=()=>e.l({...n,config:r}),a=n.workers??1;if(a<2||typeof SharedArrayBuffer!=`function`||r.view!==void 0||r.postconditions.length>0||r.possible.length>0)return i();let o=/* @__PURE__ */ new Map,s=e=>{let t=n.loadModule?.(e);return o.set(e,t),t},c=/* @__PURE__ */ new Map,l=e=>{let t=n.loadClass?.(e);return c.set(e,t),t};e.O(e._(n.source,r,s));let u=e.D(),d=e.y(r,{source:n.source,loadModule:s,loadClass:l,print:n.onPrint??console.log,files:{readFile:n.readFile,writeFile:n.writeFile}}),f=d.environment,p=e.u(r,f,()=>e.v(n.source,s));if(e.d(d.assumptions,f)||!p||p.spec.init===void 0||p.spec.next===void 0||p.properties.liveness.length>0||e.S(f))return i();let m=e.T(),h={fingerprints:e.i()},g=/* @__PURE__ */ new Int32Array(2),_=Array.from({length:a},e.n),v={generatedStates:0,distinctStates:0,queuedStates:0,depth:0,outDegrees:Array.from({length:32},()=>0)},y=[],b=()=>({status:`success`,violation:void 0,failure:void 0,...v,fingerprintDistance:e.m(v.distinctStates,v.generatedStates,()=>{let e=new BigInt64Array(v.distinctStates);return y.reduce((t,n)=>(e.set(n,t),t+n.length),0),e})}),x=t=>{let n=[];for(let r of e.C(t,f,p.spec.scope))if(m(r),v.generatedStates++,e.c(p.getFingerprintState(r),g),e.s(h.fingerprints)&&(h.fingerprints=e.a(h.fingerprints)),e.o(h.fingerprints,g[0],g[1])){if(p.findViolatedInvariant(r)||p.findFailedInitialPredicate(r))return!1;if(p.satisfiesConstraints(r)){if(!_[n.length%a].encode(r))return!1;n.push(r),v.distinctStates++}}return y.push(BigInt64Array.from(n,e=>f.runtime.getFingerprint(e))),!0};try{if(!x(p.spec.init))return i()}catch(t){if(!(t instanceof e.P))throw t;return i()}if(e.D()!==u)return i();if(v.depth=v.generatedStates===0?0:1,v.distinctStates===0)return b();let S=[],C={count:a,resolve:e=>{}},w=e=>{S.push(e),(e.type===`abort`||S.length===C.count)&&C.resolve(S.splice(0))},T=e=>new Promise(t=>{C.count=e,C.resolve=t}),E=T(a),D=await Promise.all(Array.from({length:a},()=>t(w))),O=()=>D.forEach(e=>e.terminate()),k=/* @__PURE__ */ new Map,A=(e,t)=>{let n=e<t?`${e}:${t}`:`${t}:${e}`,r=k.get(n)??new MessageChannel;return k.set(n,r),e<t?r.port1:r.port2};D.forEach((e,t)=>{let i=Array.from({length:a},(e,n)=>n===t?void 0:A(t,n)),s=_[t].finish(),l={type:`start`,source:n.source,config:r,modules:[...o],classes:[...c],index:t,ports:i,seed:s,fingerprints:h.fingerprints};e.postMessage(l,[...i.filter(e=>e!==void 0),s.words.buffer])});let j=e=>e.some(e=>e.type===`abort`),M=e=>D.forEach(t=>t.postMessage(e)),N=async t=>{let n=t.filter(e=>e.type===`paused`).length;if(n===0||j(t))return t;h.fingerprints=e.a(h.fingerprints);let r=T(n);return M({type:`resize`,fingerprints:h.fingerprints}),[...t.filter(e=>e.type!==`paused`),...await N(await r)]};if(j(await E))return O(),i();for(let e=1;;e++){let t=T(a);M({type:`level`,level:e});let n=await N(await t);if(j(n))return O(),i();let r=Array.from({length:a},()=>0);n.forEach(e=>{e.type===`expanded`&&(r[e.from]=e.added)});let o=T(a);M({type:`balance`,sizes:r});let s=await o;if(j(s))return O(),i();let c=s.flatMap(e=>e.type===`level`?[e]:[]),l=c.reduce((e,t)=>e+t.added,0);if(v.generatedStates+=c.reduce((e,t)=>e+t.generated,0),v.distinctStates+=l,c.forEach(e=>{e.outDegrees.forEach((e,t)=>v.outDegrees[t]+=e),y.push(e.fingerprints)}),l===0)return O(),b();v.depth++}};Object.defineProperty(exports,"t",{enumerable:!0,get:function(){return n}});
+`node:worker_threads`))).Worker(new URL(`./parallel-worker.js`,require("url").pathToFileURL(__filename).href));return t.on(`message`,e),t.on(`error`,()=>e({type:`abort`})),{postMessage:(e,n)=>t.postMessage(e,n),terminate:()=>void t.terminate()}},n=async n=>{let r=typeof n.config==`string`?e.A(n.config):n.config,i=()=>e.l({...n,config:r}),a=n.workers??1;if(a<2||typeof SharedArrayBuffer!=`function`||r.view!==void 0||r.postconditions.length>0||r.possible.length>0)return i();let o=/* @__PURE__ */ new Map,s=e=>{let t=n.loadModule?.(e);return o.set(e,t),t},c=/* @__PURE__ */ new Map,l=e=>{let t=n.loadClass?.(e);return c.set(e,t),t};e.O(e._(n.source,r,s));let u=e.D(),d=e.y(r,{source:n.source,loadModule:s,loadClass:l,print:n.onPrint??console.log,files:{readFile:n.readFile,writeFile:n.writeFile}}),f=d.environment,p=e.u(r,f,()=>e.v(n.source,s));if(e.d(d.assumptions,f)||!p||p.spec.init===void 0||p.spec.next===void 0||p.properties.liveness.length>0||e.S(f))return i();let m=e.T(),h={fingerprints:e.i()},g=/* @__PURE__ */ new Int32Array(2),_=Array.from({length:a},e.n),v={generatedStates:0,distinctStates:0,queuedStates:0,depth:0,outDegrees:Array.from({length:32},()=>0)},y=[],b=()=>({status:`success`,violation:void 0,failure:void 0,...v,fingerprintDistance:e.m(v.distinctStates,v.generatedStates,()=>{let e=new BigInt64Array(v.distinctStates);return y.reduce((t,n)=>(e.set(n,t),t+n.length),0),e})}),x=t=>{let n=[];for(let r of e.C(t,f,p.spec.scope))if(m(r),v.generatedStates++,e.c(p.getFingerprintState(r),g),e.s(h.fingerprints)&&(h.fingerprints=e.a(h.fingerprints)),e.o(h.fingerprints,g[0],g[1])){if(p.findViolatedInvariant(r)||p.findFailedInitialPredicate(r))return!1;if(p.satisfiesConstraints(r)){if(!_[n.length%a].encode(r))return!1;n.push(r),v.distinctStates++}}return y.push(BigInt64Array.from(n,e=>f.runtime.getFingerprint(e))),!0};try{if(!x(p.spec.init))return i()}catch(t){if(!(t instanceof e.P))throw t;return i()}if(e.D()!==u)return i();if(v.depth=v.generatedStates===0?0:1,v.distinctStates===0)return b();let S=[],C={count:a,resolve:e=>{}},w=e=>{S.push(e),(e.type===`abort`||S.length===C.count)&&C.resolve(S.splice(0))},T=e=>new Promise(t=>{C.count=e,C.resolve=t}),E=T(a),D=new Int32Array(new SharedArrayBuffer(a*4)),O=new Int32Array(new SharedArrayBuffer(a*4)),k=await Promise.all(Array.from({length:a},()=>t(w))),A=()=>k.forEach(e=>e.terminate()),j=/* @__PURE__ */ new Map,M=(e,t)=>{let n=e<t?`${e}:${t}`:`${t}:${e}`,r=j.get(n)??new MessageChannel;return j.set(n,r),e<t?r.port1:r.port2};k.forEach((e,t)=>{let i=Array.from({length:a},(e,n)=>n===t?void 0:M(t,n)),s=_[t].finish(),l={type:`start`,source:n.source,config:r,modules:[...o],classes:[...c],index:t,ports:i,seed:s,fingerprints:h.fingerprints,requests:D,remaining:O};e.postMessage(l,[...i.filter(e=>e!==void 0),s.words.buffer])});let N=e=>e.some(e=>e.type===`abort`),P=e=>k.forEach(t=>t.postMessage(e)),F=async t=>{let n=t.filter(e=>e.type===`paused`).length;if(n===0||N(t))return t;h.fingerprints=e.a(h.fingerprints);let r=T(n);return P({type:`resize`,fingerprints:h.fingerprints}),[...t.filter(e=>e.type!==`paused`),...await F(await r)]};if(N(await E))return A(),i();for(let e=1;;e++){let t=T(a);D.forEach((e,t)=>Atomics.store(D,t,0)),P({type:`level`,level:e});let n=await F(await t);if(N(n))return A(),i();let r=n.flatMap(e=>e.type===`level`?[e]:[]),o=r.reduce((e,t)=>e+t.added,0);if(v.generatedStates+=r.reduce((e,t)=>e+t.generated,0),v.distinctStates+=o,r.forEach(e=>{e.outDegrees.forEach((e,t)=>v.outDegrees[t]+=e),y.push(e.fingerprints)}),o===0)return A(),b();v.depth++}};Object.defineProperty(exports,"t",{enumerable:!0,get:function(){return n}});

package/dist/parallel.js

@@ -6,7 +6,7 @@
  * This source code is licensed under the MIT license found in the
  * LICENSE file in the root directory of this source tree.
  */
-import{A as e,C as t,D as n,O as r,P as i,S as a,T as o,_ as s,a as c,c as l,d as u,i as d,l as f,m as p,n as m,o as h,s as g,u as _,v,y}from"./transfer.js";const b=async e=>{if(typeof Worker==`function`){let t=new Worker(new URL(`./parallel-worker.js`,import.meta.url),{type:`module`});return t.addEventListener(`message`,t=>e(t.data)),t.addEventListener(`error`,()=>e({type:`abort`})),t}let t=new(await(import(
+import{A as e,C as t,D as n,O as r,P as i,S as ee,T as a,_ as o,a as s,c,d as l,i as u,l as d,m as f,n as p,o as m,s as h,u as g,v as _,y as v}from"./transfer.js";const y=async e=>{if(typeof Worker==`function`){let t=new Worker(new URL(`./parallel-worker.js`,import.meta.url),{type:`module`});return t.addEventListener(`message`,t=>e(t.data)),t.addEventListener(`error`,()=>e({type:`abort`})),t}let t=new(await(import(
 /* @vite-ignore */
 /* webpackIgnore: true */
-`node:worker_threads`))).Worker(new URL(`./parallel-worker.js`,import.meta.url));return t.on(`message`,e),t.on(`error`,()=>e({type:`abort`})),{postMessage:(e,n)=>t.postMessage(e,n),terminate:()=>void t.terminate()}},x=async x=>{let S=typeof x.config==`string`?e(x.config):x.config,C=()=>f({...x,config:S}),w=x.workers??1;if(w<2||typeof SharedArrayBuffer!=`function`||S.view!==void 0||S.postconditions.length>0||S.possible.length>0)return C();let T=/* @__PURE__ */ new Map,E=e=>{let t=x.loadModule?.(e);return T.set(e,t),t},D=/* @__PURE__ */ new Map,O=e=>{let t=x.loadClass?.(e);return D.set(e,t),t};r(s(x.source,S,E));let k=n(),A=y(S,{source:x.source,loadModule:E,loadClass:O,print:x.onPrint??console.log,files:{readFile:x.readFile,writeFile:x.writeFile}}),j=A.environment,M=_(S,j,()=>v(x.source,E));if(u(A.assumptions,j)||!M||M.spec.init===void 0||M.spec.next===void 0||M.properties.liveness.length>0||a(j))return C();let N=o(),P={fingerprints:d()},F=/* @__PURE__ */ new Int32Array(2),I=Array.from({length:w},m),L={generatedStates:0,distinctStates:0,queuedStates:0,depth:0,outDegrees:Array.from({length:32},()=>0)},R=[],z=()=>({status:`success`,violation:void 0,failure:void 0,...L,fingerprintDistance:p(L.distinctStates,L.generatedStates,()=>{let e=new BigInt64Array(L.distinctStates);return R.reduce((t,n)=>(e.set(n,t),t+n.length),0),e})}),B=e=>{let n=[];for(let r of t(e,j,M.spec.scope))if(N(r),L.generatedStates++,l(M.getFingerprintState(r),F),g(P.fingerprints)&&(P.fingerprints=c(P.fingerprints)),h(P.fingerprints,F[0],F[1])){if(M.findViolatedInvariant(r)||M.findFailedInitialPredicate(r))return!1;if(M.satisfiesConstraints(r)){if(!I[n.length%w].encode(r))return!1;n.push(r),L.distinctStates++}}return R.push(BigInt64Array.from(n,e=>j.runtime.getFingerprint(e))),!0};try{if(!B(M.spec.init))return C()}catch(e){if(!(e instanceof i))throw e;return C()}if(n()!==k)return C();if(L.depth=L.generatedStates===0?0:1,L.distinctStates===0)return z();let V=[],H={count:w,resolve:e=>{}},U=e=>{V.push(e),(e.type===`abort`||V.length===H.count)&&H.resolve(V.splice(0))},W=e=>new Promise(t=>{H.count=e,H.resolve=t}),G=W(w),K=await Promise.all(Array.from({length:w},()=>b(U))),q=()=>K.forEach(e=>e.terminate()),J=/* @__PURE__ */ new Map,Y=(e,t)=>{let n=e<t?`${e}:${t}`:`${t}:${e}`,r=J.get(n)??new MessageChannel;return J.set(n,r),e<t?r.port1:r.port2};K.forEach((e,t)=>{let n=Array.from({length:w},(e,n)=>n===t?void 0:Y(t,n)),r=I[t].finish(),i={type:`start`,source:x.source,config:S,modules:[...T],classes:[...D],index:t,ports:n,seed:r,fingerprints:P.fingerprints};e.postMessage(i,[...n.filter(e=>e!==void 0),r.words.buffer])});let X=e=>e.some(e=>e.type===`abort`),Z=e=>K.forEach(t=>t.postMessage(e)),Q=async e=>{let t=e.filter(e=>e.type===`paused`).length;if(t===0||X(e))return e;P.fingerprints=c(P.fingerprints);let n=W(t);return Z({type:`resize`,fingerprints:P.fingerprints}),[...e.filter(e=>e.type!==`paused`),...await Q(await n)]};if(X(await G))return q(),C();for(let e=1;;e++){let t=W(w);Z({type:`level`,level:e});let n=await Q(await t);if(X(n))return q(),C();let r=Array.from({length:w},()=>0);n.forEach(e=>{e.type===`expanded`&&(r[e.from]=e.added)});let i=W(w);Z({type:`balance`,sizes:r});let a=await i;if(X(a))return q(),C();let o=a.flatMap(e=>e.type===`level`?[e]:[]),s=o.reduce((e,t)=>e+t.added,0);if(L.generatedStates+=o.reduce((e,t)=>e+t.generated,0),L.distinctStates+=s,o.forEach(e=>{e.outDegrees.forEach((e,t)=>L.outDegrees[t]+=e),R.push(e.fingerprints)}),s===0)return q(),z();L.depth++}};export{x as t};
+`node:worker_threads`))).Worker(new URL(`./parallel-worker.js`,import.meta.url));return t.on(`message`,e),t.on(`error`,()=>e({type:`abort`})),{postMessage:(e,n)=>t.postMessage(e,n),terminate:()=>void t.terminate()}},b=async b=>{let x=typeof b.config==`string`?e(b.config):b.config,S=()=>d({...b,config:x}),C=b.workers??1;if(C<2||typeof SharedArrayBuffer!=`function`||x.view!==void 0||x.postconditions.length>0||x.possible.length>0)return S();let w=/* @__PURE__ */ new Map,T=e=>{let t=b.loadModule?.(e);return w.set(e,t),t},E=/* @__PURE__ */ new Map,D=e=>{let t=b.loadClass?.(e);return E.set(e,t),t};r(o(b.source,x,T));let O=n(),k=v(x,{source:b.source,loadModule:T,loadClass:D,print:b.onPrint??console.log,files:{readFile:b.readFile,writeFile:b.writeFile}}),A=k.environment,j=g(x,A,()=>_(b.source,T));if(l(k.assumptions,A)||!j||j.spec.init===void 0||j.spec.next===void 0||j.properties.liveness.length>0||ee(A))return S();let M=a(),N={fingerprints:u()},P=/* @__PURE__ */ new Int32Array(2),F=Array.from({length:C},p),I={generatedStates:0,distinctStates:0,queuedStates:0,depth:0,outDegrees:Array.from({length:32},()=>0)},L=[],R=()=>({status:`success`,violation:void 0,failure:void 0,...I,fingerprintDistance:f(I.distinctStates,I.generatedStates,()=>{let e=new BigInt64Array(I.distinctStates);return L.reduce((t,n)=>(e.set(n,t),t+n.length),0),e})}),z=e=>{let n=[];for(let r of t(e,A,j.spec.scope))if(M(r),I.generatedStates++,c(j.getFingerprintState(r),P),h(N.fingerprints)&&(N.fingerprints=s(N.fingerprints)),m(N.fingerprints,P[0],P[1])){if(j.findViolatedInvariant(r)||j.findFailedInitialPredicate(r))return!1;if(j.satisfiesConstraints(r)){if(!F[n.length%C].encode(r))return!1;n.push(r),I.distinctStates++}}return L.push(BigInt64Array.from(n,e=>A.runtime.getFingerprint(e))),!0};try{if(!z(j.spec.init))return S()}catch(e){if(!(e instanceof i))throw e;return S()}if(n()!==O)return S();if(I.depth=I.generatedStates===0?0:1,I.distinctStates===0)return R();let B=[],V={count:C,resolve:e=>{}},H=e=>{B.push(e),(e.type===`abort`||B.length===V.count)&&V.resolve(B.splice(0))},U=e=>new Promise(t=>{V.count=e,V.resolve=t}),W=U(C),G=new Int32Array(new SharedArrayBuffer(C*4)),K=new Int32Array(new SharedArrayBuffer(C*4)),q=await Promise.all(Array.from({length:C},()=>y(H))),J=()=>q.forEach(e=>e.terminate()),Y=/* @__PURE__ */ new Map,X=(e,t)=>{let n=e<t?`${e}:${t}`:`${t}:${e}`,r=Y.get(n)??new MessageChannel;return Y.set(n,r),e<t?r.port1:r.port2};q.forEach((e,t)=>{let n=Array.from({length:C},(e,n)=>n===t?void 0:X(t,n)),r=F[t].finish(),i={type:`start`,source:b.source,config:x,modules:[...w],classes:[...E],index:t,ports:n,seed:r,fingerprints:N.fingerprints,requests:G,remaining:K};e.postMessage(i,[...n.filter(e=>e!==void 0),r.words.buffer])});let Z=e=>e.some(e=>e.type===`abort`),Q=e=>q.forEach(t=>t.postMessage(e)),$=async e=>{let t=e.filter(e=>e.type===`paused`).length;if(t===0||Z(e))return e;N.fingerprints=s(N.fingerprints);let n=U(t);return Q({type:`resize`,fingerprints:N.fingerprints}),[...e.filter(e=>e.type!==`paused`),...await $(await n)]};if(Z(await W))return J(),S();for(let e=1;;e++){let t=U(C);G.forEach((e,t)=>Atomics.store(G,t,0)),Q({type:`level`,level:e});let n=await $(await t);if(Z(n))return J(),S();let r=n.flatMap(e=>e.type===`level`?[e]:[]),i=r.reduce((e,t)=>e+t.added,0);if(I.generatedStates+=r.reduce((e,t)=>e+t.generated,0),I.distinctStates+=i,r.forEach(e=>{e.outDegrees.forEach((e,t)=>I.outDegrees[t]+=e),L.push(e.fingerprints)}),i===0)return J(),R();I.depth++}};export{b as t};

package/package.json

@@ -1,6 +1,6 @@
 {
   "name": "tla-ts",
-  "version": "0.0.3-dev.6979965",
+  "version": "0.0.3-dev.79f7ed7",
   "description": "A TLA+ interpreter and model checker in TypeScript, 2× faster than TLC",
   "keywords": [
     "formal verification",

package/src/parallel-worker.ts

@@ -21,56 +21,21 @@ import { createStateDecoder, createStateEncoder, type EncodedStates } from "./tr
 import { getRankedStringCount, setStringOrder, type Value } from "./values";
 
 const WORKER_THREADS_MODULE = "node:worker_threads";
-
-interface PeerMessage {
-  phase: "balance";
-  level: number;
-  from: number;
-  states: EncodedStates;
-}
+const LOCKED = -1;
 
 interface Expansion {
+  states: Value[][];
   origin: number;
+  end: number;
   successors: Successor[] | undefined;
   index: number;
+  added: number;
   generated: number;
-  newStates: Int32Array;
+  outDegrees: number[];
   next: Value[][];
   fingerprintStates: Value[][];
 }
 
-interface Transfer {
-  from: number;
-  to: number;
-  count: number;
-}
-
-const getBalancePlan = (sizes: number[]) => {
-  const total = sizes.reduce((sum, size) => sum + size, 0);
-  const average = total / sizes.length;
-  if (Math.max(...sizes) - Math.min(...sizes) <= Math.max(64, average / 4)) return [];
-  const targets = sizes.map(
-    (_, index) => Math.floor(average) + (index < total % sizes.length ? 1 : 0),
-  );
-  const donors = sizes.flatMap((size, index) =>
-    size > targets[index] ? [{ index, amount: size - targets[index] }] : [],
-  );
-  const receivers = sizes.flatMap((size, index) =>
-    size < targets[index] ? [{ index, amount: targets[index] - size }] : [],
-  );
-  const transfers: Transfer[] = [];
-  for (const donor of donors) {
-    for (const receiver of receivers) {
-      const moved = Math.min(donor.amount, receiver.amount);
-      if (moved === 0) continue;
-      transfers.push({ from: donor.index, to: receiver.index, count: moved });
-      donor.amount -= moved;
-      receiver.amount -= moved;
-    }
-  }
-  return transfers;
-};
-
 const getParentPort = async () => {
   if (typeof (globalThis as { postMessage?: unknown }).postMessage === "function") {
     return globalThis as unknown as Port;
@@ -81,33 +46,6 @@ const getParentPort = async () => {
   return threads.parentPort as unknown as Port;
 };
 
-const createInbox = () => {
-  const messages = new Map<string, PeerMessage[]>();
-  const waiters = new Map<string, { count: number; resolve: (messages: PeerMessage[]) => void }>();
-  const getKey = (phase: PeerMessage["phase"], level: number) => `${phase}:${level}`;
-  const settle = (key: string) => {
-    const waiter = waiters.get(key);
-    const received = messages.get(key) ?? [];
-    if (!waiter || received.length < waiter.count) return;
-    waiters.delete(key);
-    messages.delete(key);
-    waiter.resolve(received);
-  };
-  return {
-    deliver: (message: PeerMessage) => {
-      const key = getKey(message.phase, message.level);
-      messages.set(key, [...(messages.get(key) ?? []), message]);
-      settle(key);
-    },
-    receive: (phase: PeerMessage["phase"], level: number, count: number) =>
-      new Promise<PeerMessage[]>((resolve) => {
-        const key = getKey(phase, level);
-        waiters.set(key, { count, resolve });
-        settle(key);
-      }),
-  };
-};
-
 const run = (start: StartMessage, parent: Port) => {
   const modules = new Map(start.modules);
   const loadModule = (name: string) => modules.get(name);
@@ -132,26 +70,55 @@ const run = (start: StartMessage, parent: Port) => {
   const canonicalize = createCanonicalizer();
   const decode = createStateDecoder(environment.variables.length);
   const shared = { fingerprints: start.fingerprints };
+  const requests = start.requests;
+  const remaining = start.remaining;
   // Under symmetry, a successor the worker generated before has the representative it had then,
   // so the worker skips the search for one.
   const generatedStates = search.isSymmetric ? createFingerprintTable() : undefined;
   const fingerprint = new Int32Array(2);
-  const inbox = createInbox();
-  const peers = start.ports.flatMap((port, index) => (port ? [{ port, index }] : []));
-  peers.forEach((peer) => {
-    peer.port.addEventListener("message", (event) => inbox.deliver(event.data as PeerMessage));
-    peer.port.start?.();
+  const waiters = { resume: () => {}, loot: (_states: EncodedStates) => {} };
+  start.ports.forEach((port) => {
+    port?.addEventListener("message", (event) => waiters.loot(event.data as EncodedStates));
+    port?.start?.();
   });
   const frontier = { states: decode(start.seed).map(canonicalize) };
-  const send = (index: number, message: PeerMessage, transfer: unknown[]) =>
-    start.ports[index]?.postMessage(message, transfer);
+
+  // A worker that runs out of states asks a peer for some by writing its index + 1 to the peer's
+  // request word. The peer sends half of the states it hasn't started, keeps expanding, and wakes
+  // the workers waiting to ask it next.
+  const answer = (expansion: Expansion, request: number) => {
+    const thief = Atomics.load(requests, self) - 1;
+    const given = Math.floor((expansion.end - expansion.origin - 1) / 2);
+    const encoder = createStateEncoder();
+    const states = expansion.states.slice(expansion.end - given, expansion.end);
+    const isEncoded = given > 0 && states.every(encoder.encode);
+    if (isEncoded) expansion.end -= given;
+    const loot = isEncoded ? encoder.finish() : createStateEncoder().finish();
+    start.ports[thief]?.postMessage(loot, [loot.words.buffer]);
+    Atomics.store(requests, self, request);
+    Atomics.notify(requests, self);
+  };
+
+  // A worker answers any request before it pauses or runs out of states, and takes no more until
+  // it resumes.
+  const lock = (expansion: Expansion) => {
+    Atomics.store(remaining, self, 0);
+    if (Atomics.compareExchange(requests, self, 0, LOCKED) === 0) return;
+    answer(expansion, LOCKED);
+  };
+
+  const unlock = (expansion: Expansion) => {
+    Atomics.store(remaining, self, expansion.end - expansion.origin);
+    Atomics.store(requests, self, 0);
+  };
 
   // The worker pauses before a successor when the shared table needs to grow, and resumes there.
   const expand = (expansion: Expansion) => {
-    const states = frontier.states;
-    for (; expansion.origin < states.length; expansion.origin++) {
-      const state = states[expansion.origin];
+    for (; expansion.origin < expansion.end; expansion.origin++) {
+      const state = expansion.states[expansion.origin];
       if (!expansion.successors) {
+        if (Atomics.load(requests, self) > 0) answer(expansion, 0);
+        Atomics.store(remaining, self, expansion.end - expansion.origin - 1);
         expansion.successors = getSuccessors(nextAction, state, environment, search.spec.scope);
         expansion.generated += expansion.successors.length;
         if (expansion.successors.length === 0 && start.config.checkDeadlock) return undefined;
@@ -174,69 +141,85 @@ const run = (start: StartMessage, parent: Port) => {
         if (!search.satisfiesConstraints(successor.state, state)) continue;
         expansion.next.push(successor.state);
         expansion.fingerprintStates.push(fingerprintState);
-        expansion.newStates[expansion.origin]++;
+        expansion.added++;
       }
+      expansion.outDegrees[Math.min(expansion.added, OUT_DEGREE_BUCKETS - 1)]++;
       expansion.successors = undefined;
       expansion.index = 0;
+      expansion.added = 0;
     }
     return "done";
   };
 
-  const balance = async (level: number, next: Value[][], sizes: number[]) => {
-    const transfers = getBalancePlan(sizes);
-    if (transfers.length === 0) return true;
-    const outgoing = new Map(peers.map((peer) => [peer.index, createStateEncoder()]));
-    for (const transfer of transfers.filter((candidate) => candidate.from === self)) {
-      const encoder = outgoing.get(transfer.to);
-      const states = next.splice(next.length - transfer.count, transfer.count);
-      if (!encoder || !states.every(encoder.encode)) return false;
+  // Each attempt asks the peer with the most states left. While another worker's request to that
+  // peer is open, the worker waits for the peer to answer it. The worker stops once no peer that
+  // takes requests has two states left.
+  const steal = async (expansion: Expansion) => {
+    const skipped = new Set<number>();
+    for (;;) {
+      const victim = start.ports.reduce<number | undefined>((best, port, index) => {
+        if (!port || skipped.has(index) || Atomics.load(remaining, index) < 2) return best;
+        return best === undefined || Atomics.load(remaining, index) > Atomics.load(remaining, best)
+          ? index
+          : best;
+      }, undefined);
+      if (victim === undefined) return false;
+      const request = Atomics.compareExchange(requests, victim, 0, self + 1);
+      if (request === LOCKED) skipped.add(victim);
+      if (request > 0) Atomics.wait(requests, victim, request);
+      if (request !== 0) continue;
+      const loot = await new Promise<EncodedStates>((resolve) => (waiters.loot = resolve));
+      if (loot.count === 0) {
+        skipped.add(victim);
+        continue;
+      }
+      expansion.states = decode(loot).map(canonicalize);
+      expansion.origin = 0;
+      expansion.end = expansion.states.length;
+      unlock(expansion);
+      return true;
     }
-    outgoing.forEach((encoder, index) => {
-      const states = encoder.finish();
-      send(index, { phase: "balance", level, from: self, states }, [states.words.buffer]);
-    });
-    const incoming = await inbox.receive("balance", level, peers.length);
-    incoming.forEach((message) => next.push(...decode(message.states).map(canonicalize)));
-    return true;
   };
 
-  const waiters = { resume: () => {}, balance: (_sizes: number[]) => {} };
   const report = (message: WorkerMessage) =>
     parent.postMessage(message, message.type === "level" ? [message.fingerprints.buffer] : []);
 
   const runLevel = async (level: number): Promise<LevelMessage | undefined> => {
     const expansion: Expansion = {
+      states: frontier.states,
       origin: 0,
+      end: frontier.states.length,
       successors: undefined,
       index: 0,
+      added: 0,
       generated: 0,
-      newStates: new Int32Array(frontier.states.length),
+      outDegrees: Array.from({ length: OUT_DEGREE_BUCKETS }, () => 0),
       next: [],
       fingerprintStates: [],
     };
-    for (let status = expand(expansion); status !== "done"; status = expand(expansion)) {
+    for (;;) {
+      const status = expand(expansion);
       if (status === undefined) return undefined;
-      const resumed = new Promise<void>((resolve) => (waiters.resume = resolve));
-      report({ type: "paused" });
-      await resumed;
+      lock(expansion);
+      if (status === "paused") {
+        const resumed = new Promise<void>((resolve) => (waiters.resume = resolve));
+        report({ type: "paused" });
+        await resumed;
+        unlock(expansion);
+        continue;
+      }
+      if (!(await steal(expansion))) break;
     }
     if (getRankedStringCount() !== startupStrings) return undefined;
-    const next = expansion.next;
-    const added = next.length;
-    const outDegrees = Array.from({ length: OUT_DEGREE_BUCKETS }, () => 0);
-    expansion.newStates.forEach((degree) => outDegrees[Math.min(degree, OUT_DEGREE_BUCKETS - 1)]++);
-    const fingerprints = search.getStoredFingerprints(next, expansion.fingerprintStates);
-    const sizes = new Promise<number[]>((resolve) => (waiters.balance = resolve));
-    report({ type: "expanded", from: self, added });
-    if (!(await balance(level, next, await sizes))) return undefined;
-    frontier.states = next;
+    frontier.states = expansion.next;
+    Atomics.store(remaining, self, frontier.states.length);
     return {
       type: "level",
       level,
       generated: expansion.generated,
-      added,
-      outDegrees,
-      fingerprints,
+      added: expansion.next.length,
+      outDegrees: expansion.outDegrees,
+      fingerprints: search.getStoredFingerprints(expansion.next, expansion.fingerprintStates),
     };
   };
 
@@ -253,8 +236,8 @@ const run = (start: StartMessage, parent: Port) => {
       shared.fingerprints = message.fingerprints;
       waiters.resume();
     }
-    if (message.type === "balance") waiters.balance(message.sizes);
   });
+  Atomics.store(remaining, self, frontier.states.length);
   report({ type: "ready" });
 };
 

package/src/parallel.ts

@@ -44,6 +44,8 @@ export interface StartMessage {
   ports: (Port | undefined)[];
   seed: EncodedStates;
   fingerprints: SharedFingerprints;
+  requests: Int32Array;
+  remaining: Int32Array;
 }
 
 export interface LevelMessage {
@@ -59,13 +61,11 @@ export type WorkerMessage =
   | { type: "ready" }
   | { type: "abort" }
   | { type: "paused" }
-  | { type: "expanded"; from: number; added: number }
   | LevelMessage;
 
 export type CoordinatorMessage =
   | { type: "level"; level: number }
-  | { type: "resize"; fingerprints: SharedFingerprints }
-  | { type: "balance"; sizes: number[] };
+  | { type: "resize"; fingerprints: SharedFingerprints };
 
 interface WorkerHandle {
   postMessage: (message: unknown, transfer?: unknown[]) => void;
@@ -225,6 +225,8 @@ export const check = async (options: CheckOptions): Promise<CheckResult> => {
       pending.resolve = resolve;
     });
   const ready = waitForReports(count);
+  const requests = new Int32Array(new SharedArrayBuffer(count * 4));
+  const remaining = new Int32Array(new SharedArrayBuffer(count * 4));
   const workers = await Promise.all(Array.from({ length: count }, () => spawnWorker(onMessage)));
   const terminate = () => workers.forEach((worker) => worker.terminate());
   const channels = new Map<string, MessageChannel>();
@@ -249,6 +251,8 @@ export const check = async (options: CheckOptions): Promise<CheckResult> => {
       ports,
       seed,
       fingerprints: shared.fingerprints,
+      requests,
+      remaining,
     };
     worker.postMessage(message, [...ports.filter((port) => port !== undefined), seed.words.buffer]);
   });
@@ -257,34 +261,25 @@ export const check = async (options: CheckOptions): Promise<CheckResult> => {
   const send = (message: CoordinatorMessage) =>
     workers.forEach((worker) => worker.postMessage(message));
   // Workers pause when the shared table needs to grow, and resume once it has.
-  const collectExpansions = async (messages: WorkerMessage[]): Promise<WorkerMessage[]> => {
+  const collectLevels = async (messages: WorkerMessage[]): Promise<WorkerMessage[]> => {
     const paused = messages.filter((message) => message.type === "paused").length;
     if (paused === 0 || isAborted(messages)) return messages;
     shared.fingerprints = growSharedFingerprints(shared.fingerprints);
     const resumed = waitForReports(paused);
     send({ type: "resize", fingerprints: shared.fingerprints });
-    const expanded = messages.filter((message) => message.type !== "paused");
-    return [...expanded, ...(await collectExpansions(await resumed))];
+    const reported = messages.filter((message) => message.type !== "paused");
+    return [...reported, ...(await collectLevels(await resumed))];
   };
   if (isAborted(await ready)) {
     terminate();
     return fallBack();
   }
   for (let level = 1; ; level++) {
-    const expanding = waitForReports(count);
-    send({ type: "level", level });
-    const expansions = await collectExpansions(await expanding);
-    if (isAborted(expansions)) {
-      terminate();
-      return fallBack();
-    }
-    const sizes = Array.from({ length: count }, () => 0);
-    expansions.forEach((message) => {
-      if (message.type === "expanded") sizes[message.from] = message.added;
-    });
     const reported = waitForReports(count);
-    send({ type: "balance", sizes });
-    const messages = await reported;
+    // Every worker is idle, so peers can ask any of them for states once the level starts.
+    requests.forEach((_, index) => Atomics.store(requests, index, 0));
+    send({ type: "level", level });
+    const messages = await collectLevels(await reported);
     if (isAborted(messages)) {
       terminate();
       return fallBack();