import * as starry from '../../src/starry';
import OnnxBeadPicker from '../../src/utils/onnxBeadPicker';

const loadBeadPickers = (path: string, seqs: number[] = [32, 64, 128, 512]): Promise<starry.BeadPicker[]> => {
	return Promise.all(
		seqs.map(async (n_seq) => {
			let loading: any;

			const url = path.replace(/seq\d+/, `seq${n_seq}`);
			const picker = new OnnxBeadPicker(url, {
				n_seq,
				usePivotX: true,
				onLoad: (p) => (loading = p),
			});
			await loading;

			return picker;
		})
	);
};

export { loadBeadPickers };