| var w=new URL("./",self.location.href).href,k=512,_=150,$=960,z=33,f,s,S,v=null,y=0,F=0,M=new Map;function T(a,e=[]){let t=++F;return new Promise((n,r)=>{M.set(t,d=>d.error?r(new Error(d.error)):n(d)),S.postMessage({...a,id:t},e)})}var i=(a,e=[])=>self.postMessage(a,e),x=a=>new Promise(e=>setTimeout(e,a)),p=()=>performance.now();function J(a){let e=a.toJs();return a.destroy?.(),e}function b(a=!0,e=!1){let t=p(),n=J(s.main_frame(k)),r=a?s.strip_frames(_).toJs():null,d=JSON.parse(s.info()),c=p()-t,o=[n.buffer];r&&o.push(...r.map(l=>l.buffer)),i({type:"frame",main:n,strip:r,size:k,stripSize:_,info:d,ms:c,camera:e},o)}async function U(a,e="auto"){let t=p();S=a,S.onmessage=o=>{M.get(o.data.id)?.(o.data),M.delete(o.data.id)};let n=T({type:"load",provider:e}),{loadPyodide:r}=await import(`${w}pyodide/pyodide.mjs`);f=await r({indexURL:`${w}pyodide/`}),await f.loadPackage(["numpy","pillow"]);let d=await(await fetch(`${w}py.zip`)).arrayBuffer();f.unpackArchive(d,"zip",{extractDir:"/home/pyodide/amp"}),f.runPython("import sys; sys.path.insert(0, '/home/pyodide/amp')"),s=f.pyimport("bridge");let c=p()-t;i({type:"python-ready",ms:c}),b();try{v=(await n).ok}catch(o){i({type:"error",message:`AMP Tadpole could not load: ${o.message}`})}i({type:"ready",capabilities:["teacher","hints","voice_out",...v?["tadpole_live"]:[]],timings:{python_ms:c,total_ms:p()-t,model:v}})}function N(){let a=p(),e=s.model_inputs().toJs(),t=e[0],n=e[1],r=e[2],d=new Float32Array(n.buffer.slice(n.byteOffset,n.byteOffset+28)),c=p()-a;return T({type:"chunk",images:t,state:d,clause:r},[t.buffer]).then(o=>({chunk:o.ok,ms:o.ms,obsMs:c}))}async function R(a,e){let t=++y,n=JSON.parse(s.check(a,e));if(n.runs===null){i({type:"refused",message:n.message});return}n.missing.length&&(i({type:"skipped",message:n.message}),a=n.runs);let r;try{r=s.view.begin(a,e).toJs()}catch(u){i({type:"error",message:String(u.message||u).split(` |
| `).filter(Boolean).pop()});return}i({type:"started",clauses:r,backend:e});let d=JSON.parse(s.info()).logLength,c=0,o=[],l=null;for(;t===y;){e!=="Teacher"&&s.view.waiting()&&!l&&(l=N());let u=s.view.frames;if(c<u){let m=p();s.view.idx=c,b(),c++,await x(Math.max(0,z-(p()-m)));continue}if(l){let m;try{m=await l}catch(g){i({type:"error",message:g.message}),s.view.stop();return}if(l=null,t!==y)break;o.push(m.ms),s.give(m.chunk);continue}if(s.view.waiting()){s.view.give_teacher();continue}s.view.finish(),b(),i({type:"done",stepMs:o,log:JSON.parse(s.log_since(d))});return}}async function E(a){let e={main:[],strip:[],observation:[],tadpole:[]};s.view.begin("pick up the "+JSON.parse(s.hints("")).objects[0].name,"AMP Tadpole");for(let t=0;t<a;t++){let n=p();s.main_frame(k),e.main.push(p()-n),n=p(),s.strip_frames(_),e.strip.push(p()-n);let r=await N();e.observation.push(r.obsMs),e.tadpole.push(r.ms)}s.view.stop(),i({type:"bench",t:e})}async function I(a){let e=++y,t=[];for(let n of a){let r=s.eval_scene(n).toJs();for(let d of r){for(s.eval_begin(d);s.view.waiting();){let c=await N();if(e!==y)return;s.give(c.chunk)}t.push({seed:n,...JSON.parse(s.eval_end())})}i({type:"eval-progress",seed:n,rows:t.length})}i({type:"evaluate",rows:t})}var O=new Set;async function j(a){let e=`/home/pyodide/traces/${a}`;if(O.has(a))return e;f.FS.mkdirTree(e);for(let t of["trace.json","data.npz"]){let n=await fetch(`${w}traces/${a}/${t}`);if(!n.ok||!n.body)throw new Error(`trace ${a}/${t}: HTTP ${n.status}`);let r=Number(n.headers.get("Content-Length"))||0,d=[],c=0,o=0,l=n.body.getReader();for(;;){let{done:g,value:P}=await l.read();if(g)break;d.push(P),c+=P.length,t==="data.npz"&&p()-o>100&&(o=p(),i({type:"viewer-progress",got:c,total:r}))}let u=new Uint8Array(c),m=0;for(let g of d)u.set(g,m),m+=g.length;f.FS.writeFile(`${e}/${t}`,u)}return O.add(a),e}function A(a,e,t){if(!a){i({type:"viewer-status",status:t});return}let n=p(),r=J(s.viewer_picture(a==="draft")),[d,c]=s.viewer.size.toJs(),o=Math.max(c,$),l=a==="full"?J(s.viewer_side(e,o)):null,u=[r.buffer];l&&u.push(l.buffer),i({type:"viewer-frame",img:r,w:d,h:c,side:l,sideW:e,sideH:o,draft:a==="draft",status:t,ms:p()-n},u)}var h=0;async function W(a){let e=++h;for(;e===h;){let t=p(),n=JSON.parse(s.viewer_op("step",1.2));if(A(n.draw,a,n),!n.playing)break;await x(Math.max(10,60-(p()-t)))}}self.onmessage=async a=>{let e=a.data;try{switch(e.type){case"init":await U(e.port,e.provider);break;case"reset":y++,s.view.reset(e.seed??void 0),b();break;case"play":R(e.text,e.backend);break;case"stop":y++,s.view.stop(),b();break;case"camera":{let t=s.view;e.az!==void 0&&(t.az=e.az),e.el!==void 0&&(t.el=e.el),e.mode!==void 0&&(t.view_mode=e.mode),e.grid!==void 0&&(t.show_grid=e.grid),e.trail!==void 0&&(t.show_trail=e.trail),b(!1,!0);break}case"hints":i({type:"hints",id:e.id,hints:JSON.parse(s.hints(e.text,e.backend))});break;case"compose":i({type:"composed",id:e.id,text:s.view.compose(e.text,e.key)});break;case"bench":await E(e.n);break;case"digest":i({type:"digest",digest:JSON.parse(s.digest(JSON.stringify(e.seeds)))});break;case"evaluate":await I(e.seeds);break;case"teacher":i({type:"teacher",traj:JSON.parse(s.teacher_traj(e.seed,e.text))});break;case"viewer-load":{h++;let t=await j(e.scene),n=JSON.parse(s.viewer_load(t,e.w,e.h));i({type:"viewer-loaded",scene:e.scene,info:n}),A("full",e.sideW,n);break}case"viewer":{e.op!=="step"&&h++;let t=JSON.parse(s.viewer_op(e.op,e.a??null,e.b??null));A(t.draw,e.sideW,t),t.playing&&W(e.sideW);break}}}catch(t){i({type:"error",message:String(t?.message||t)})}}; |
