ez], bitcast(c.vel.y));\\n atomicStore(&C[w+2*cez*cez], bitcast(c.mas ));\\n}\\nfn cellRead(r: i32) -> Cell\\n{\\n var velx = bitcast(atomicLoad(&C[r+0*cez*cez]));\\n var vely = bitcast(atomicLoad(&C[r+1*cez*cez]));\\n var mass = bitcast(atomicLoad(&C[r+2*cez*cez]));\\n return Cell(vec2f(velx,vely), mass);\\n}\\nfn mod2(a: vec2f, b: f32) -> vec2f\\n{\\n return fract(a/b)*b;\\n}\\n@compute @workgroup_size(8,8,1)\\nfn clearScreen(@builtin(global_invocation_id) id3: uint3)\\n{\\n var screen_size = textureDimensions(screen);\\n if(id3.x >= screen_size.x){return;}\\n if(id3.y >= screen_size.y){return;}\\n textureStore(screen, id3.xy, vec4f(0));\\n}\\n#workgroup_count cellClear 4 cez 1\\n@compute @workgroup_size(cez/4,1,1)\\nfn cellClear(@builtin(global_invocation_id) id3: uint3)\\n{\\n var id2 = vec2i(id3.xy);\\n var id1 = dot(id2,vec2i(1,cez));\\n var c = Cell(vec2f(0), 0f);\\n cellWrite(id1, c);\\n}\\n#dispatch_once particleIni\\n#workgroup_count particleIni 2 paz 1\\n@compute @workgroup_size(paz/2,1,1)\\nfn particleIni(@builtin(global_invocation_id) id3: uint3)\\n{\\n var id2 = vec2i(id3.xy);\\n var id1 = dot(id2,vec2i(1,paz));\\n var p = Particle(\\n vec2f(id3.xy)/f32(paz)*f32(cez)*.5f + f32(cez)*.4f,\\n vec2f(0),\\n mat2x2f(0,0,0,0),\\n 1f,\\n 1f,\\n mat2x2f(1,0,0,1)\\n );\\n P[id1] = p;\\n}\\n#workgroup_count particleToCell 2 paz 1\\n@compute @workgroup_size(paz/2,1,1)\\nfn particleToCell(@builtin(global_invocation_id) id3: uint3)\\n{\\n var id2 = vec2i(id3.xy);\\n var id1 = dot(id2,vec2i(1,paz));\\n var p = P[id1];\\n var F = p.dfr;\\n var J = determinant(F);\\n var F_T = transpose(F);\\n var Jinv = 1f; if(J!=0f){Jinv = 1f/J;}\\n var F_inv_T = mat2x2f(\\n F_T[1][1], -F_T[0][1],\\n -F_T[1][0], F_T[0][0]) * Jinv;\\n var F_minus_F_inv_T = F - F_inv_T;\\n var P_term_0 = elastic_mu * F_minus_F_inv_T;\\n var P_term_1 = elastic_la * log(J) * F_inv_T;\\n var P = P_term_0 + P_term_1;\\n var stress = Jinv * (P*F_T);\\n var volume = p.vol * J;\\n var eq_16_term_0 = -volume * 4f * stress * dt;\\n var cell_idx = floor(p.pos);\\n var cell_diff = fract(p.pos) - .5f;\\n var weights = array(vec2f(0),vec2f(0),vec2f(0));\\n weights[0] = .50f * pow(.5f - cell_diff, vec2f(2));\\n weights[1] = .75f - pow( cell_diff, vec2f(2));\\n weights[2] = .50f * pow(.5f + cell_diff, vec2f(2));\\n for(var gy = 0; gy < 3; gy++){\\n for(var gx = 0; gx < 3; gx++){\\n var cell_pos = floor(p.pos + vec2f(f32(gx),f32(gy))-1f);\\n if(cell_pos.x<0f || cell_pos.x>=f32(cez)){continue;}\\n if(cell_pos.y<0f || cell_pos.y>=f32(cez)){continue;}\\n var cell_dist = (cell_pos - p.pos) + .5f;\\n var cell_index = i32(dot(cell_pos,vec2f(1,cez)));\\n var weight = weights[gx].x * weights[gy].y;\\n var Q = p.mom * cell_dist;\\n var weighted_mass = weight * p.mas;\\n var cell = Cell(\\n weighted_mass * (p.vel + Q) + eq_16_term_0 * weight * cell_dist,\\n weighted_mass\\n );\\n cellAdd(cell_index, cell);\\n }}\\n}\\n#workgroup_count cellUpdate 4 cez 1\\n@compute @workgroup_size(cez/4,1,1)\\nfn cellUpdate(@builtin(global_invocation_id) id3: uint3)\\n{\\n var id2 = vec2i(id3.xy);\\n var id1 = dot(id2,vec2i(1,cez));\\n var cell = cellRead(id1);\\n if(cell.mas != 0f){cell.vel /= cell.mas;}\\n else {cell.vel = vec2f(0);}\\n cell.vel += dt * vec2f(0,gravity);// * f32(cell.mas != 0f);\\n if(id2.x < 2 || id2.x > cez - 3){ cell.vel *= vec2f(0,1); }\\n if(id2.y < 2 || id2.y > cez - 3){ cell.vel *= vec2f(1,0); }\\n cellWrite(id1, cell);\\n}\\n#workgroup_count cellToParticle 2 paz 1\\n@compute @workgroup_size(paz/2,1,1)\\nfn cellToParticle(@builtin(global_invocation_id) id3: uint3)\\n{\\n var id2 = vec2i(id3.xy);\\n var id1 = dot(id2,vec2i(1,paz));\\n var p = P[id1];\\n

@compute.toys