4a8abcfString of beads: 3D bead-spring solver, control panel, waveform, spectrogram, wav, displacement viewerJeremy Magland 1import { simulate } from './solver';
a78afe0Add WebGPU solver with a toggle; faster CPU solverJeremy Magland 2import { simulateGPU } from './gpu';
3import type { BeadSystem, Excitation, SimOptions, SimResult } from './types';
4a8abcfString of beads: 3D bead-spring solver, control panel, waveform, spectrogram, wav, displacement viewerJeremy Magland 4
5export interface WorkerRequest {
6 sys: BeadSystem;
7 exc: Excitation;
8 opt: SimOptions;
4a8abcfString of beads: 3D bead-spring solver, control panel, waveform, spectrogram, wav, displacement viewerJeremy Magland 10}
a78afe0Add WebGPU solver with a toggle; faster CPU solverJeremy Magland 12self.onmessage = async (ev: MessageEvent<WorkerRequest>) => {
13 const { sys, exc, opt, gpu } = ev.data;
14 const progress = (f: number) => self.postMessage({ type: 'progress', fraction: f });
4a8abcfString of beads: 3D bead-spring solver, control panel, waveform, spectrogram, wav, displacement viewerJeremy Magland 15 try {
a78afe0Add WebGPU solver with a toggle; faster CPU solverJeremy Magland 16 let result: SimResult | undefined;
17 let fallback: string | undefined;
18 if (gpu) {
19 try {
20 const r = await simulateGPU(sys, exc, opt, progress);
21 result = { ...r, backend: `WebGPU (${r.device})` };
22 } catch (e) {
23 fallback = (e as Error).message ?? String(e);
24 }
25 }
26 result ??= { ...simulate(sys, exc, opt, progress), backend: 'CPU', fallback };
27 const transfer = [result.audio.buffer, result.frames.buffer, result.frameTimes.buffer, result.energy.buffer, result.balance.buffer];
28 self.postMessage({ type: 'done', result }, { transfer: transfer as ArrayBuffer[] });
4a8abcfString of beads: 3D bead-spring solver, control panel, waveform, spectrogram, wav, displacement viewerJeremy Magland 29 } catch (e) {
30 self.postMessage({ type: 'error', message: String(e) });
31 }
32};