import { simulate } from './solver';
import { simulateGPU } from './gpu';
import type { BeadSystem, Excitation, SimOptions, SimResult } from './types';

export interface WorkerRequest {
  sys: BeadSystem;
  exc: Excitation;
  opt: SimOptions;
  gpu: boolean;
}

self.onmessage = async (ev: MessageEvent<WorkerRequest>) => {
  const { sys, exc, opt, gpu } = ev.data;
  const progress = (f: number) => self.postMessage({ type: 'progress', fraction: f });
  try {
    let result: SimResult | undefined;
    let fallback: string | undefined;
    if (gpu) {
      try {
        const r = await simulateGPU(sys, exc, opt, progress);
        result = { ...r, backend: `WebGPU (${r.device})` };
      } catch (e) {
        fallback = (e as Error).message ?? String(e);
      }
    }
    result ??= { ...simulate(sys, exc, opt, progress), backend: 'CPU', fallback };
    const transfer = [result.audio.buffer, result.frames.buffer, result.frameTimes.buffer, result.energy.buffer, result.balance.buffer];
    self.postMessage({ type: 'done', result }, { transfer: transfer as ArrayBuffer[] });
  } catch (e) {
    self.postMessage({ type: 'error', message: String(e) });
  }
};