creenCenterPos * simSize;\\n return simPos;\\n}\\n\\nfn simToScreen(pos: vec2f) -> vec2f {\\n let screenSize = vec2f(textureDimensions(screen));\\n let minScreenSize = min(screenSize.x, screenSize.y);\\n let simSize = vec2f(SIZE);\\n let screenCenterPos = pos / simSize;\\n let screenPos = minScreenSize * (screenCenterPos - 0.5) + 0.5 * screenSize;\\n return screenPos;\\n}\\n\\nfn linearIndex(id: vec3u) -> u32\\n{\\n return id.x + id.y*SIZE + id.z*SIZE*SIZE;\\n}\\n\\nfn linearIndexRepeat(id: vec3i) -> u32\\n{\\n let repeat = id;// - SIZE * (id / SIZE);\\n return linearIndex(vec3u(repeat));\\n}\\n\\nfn indexTo3D(index: u32) -> vec3u\\n{\\n return vec3u(index % SIZE, (index / SIZE) % SIZE, index / (SIZE * SIZE));\\n}\\n\\nfn indexTo2D(index: u32) -> vec2u\\n{\\n return vec2u(index % SIZE, index / SIZE);\\n}\\n\\nfn getAxisIndex(id: u32, group: u32, axis: u32) -> u32\\n{\\n let groupGrid = indexTo2D(group);\\n if(axis == 0) { return linearIndex(vec3u(id, groupGrid)); }\\n if(axis == 1) { return linearIndex(vec3u(groupGrid.x, id, groupGrid.y)); }\\n return linearIndex(vec3u(groupGrid, id));\\n}\\n\\nfn reverseLowestBits(num: u32, bits: u32) -> u32 {\\n let reversed = reverseBits(num);\\n let shifted_reversed = reversed >> (32u - bits);\\n let upper_bits = (num >> bits) << bits;\\n return shifted_reversed + upper_bits;\\n}\\n\\nfn getIndexPair(i: u32, it: u32) -> vec2 {\\n let k1 = reverseLowestBits(2u * i, it + 1u);\\n let k2 = k1 + (1u << it);\\n return vec2u(k1, k2);\\n}\\n\\nfn expi(angle: float) -> vec2f {\\n return vec2f(cos(angle), sin(angle));\\n}\\n\\nfn cmul(a: vec2f, b: vec2f) -> vec2f {\\n return vec2f(a.x * b.x - a.y * b.y, a.x * b.y + a.y * b.x);\\n}\\n\\nfn unityRoot(k: u32, N: u32, inverse: bool) -> vec2f {\\n let d = select(-1.0, 1.0, inverse);\\n return expi(2.0 * d * PI * float(k) / float(N));\\n}\\n\\nfn blobIC(pos: vec3f, bpos:vec3f, vel:vec3f, rad: float) -> vec2f\\n{\\n let thickness = 5.0;\\n let dx = pos - bpos;\\n let dist = length(dx);\\n let density = 1.0 - smoothstep(rad, rad + thickness, dist);\\n return 1.0 * density * expi(-dot(vel, dx));\\n}\\n\\nfn fftshift(index: u32) -> u32 \\n{\\n return (index + SIZE / 2) % SIZE;\\n}\\n\\nfn ifftshift(index: u32) -> u32 \\n{\\n return (index + (SIZE + 1) / 2) % SIZE;\\n}\\n\\nvar TEMP: array;\\n\\nfn fft(index: u32, group: u32, axis: u32, inverse: bool) {\\n for (var iteration = 0u; iteration <= u32(IT_NUM); iteration++)\\n {\\n var ids = getIndexPair(index, select(iteration, IT_NUM, iteration == 0u));\\n let v1 = select(TEMP[ids.x], sim[getAxisIndex(ids.x, group, axis)], iteration == 0u);\\n let v2 = select(TEMP[ids.y], sim[getAxisIndex(ids.y, group, axis)], iteration == 0u);\\n\\n ids = getIndexPair(index, iteration);\\n let rootIndex = (ids.x & ((1u << iteration) - 1u)) << (u32(IT_NUM) - iteration);\\n let V = cmul(unityRoot(rootIndex, SIZE, inverse), v2);\\n TEMP[ids.x] = v1 + V;\\n TEMP[ids.y] = v1 - V;\\n\\n workgroupBarrier();\\n }\\n\\n let M = SIZE / WG_SIZE;\\n for (var i = 0u; i < u32(M); i++) {\\n let rowIndex = index + i * WG_SIZE;\\n let idx = getAxisIndex(rowIndex, group, axis);\\n sim[idx] = TEMP[rowIndex] / select(1.0, SIZE, inverse);\\n }\\n\\n storageBarrier();\\n}\\n\\nfn gaussian(dx: vec3f, sigma: float) -> float {\\n return exp(-dot(dx, dx) / (2.0 * sigma * sigma));\\n}\\n\\nfn real_space_update(index: u32)\\n{\\n let id = indexTo3D(index);\\n let posf = vec3f(id);\\n if(time.frame < 5) \\n {\\n let uv = (vec3f(id) + 0.5) / SIZE;\\n let len = length(vec3f(1.6,1.0,1.0)*(uv-vec3f(0.45)));\\n var InitCond = blobIC(posf, vec3f(0.5, 0.5, 0.5)*SIZE, vec3f(0.0, 0.0, 0.0), 20.0);\\n sim[index] = InitCond;\\n } \\n else\\n {\\n //apply potential energy time evolution operator\\n var field = sim[index];\\n \\n\\n if(CUR_INDEX%2000 ==

@compute.toys