1/**
2 * Persistence. Small structured state lives in localStorage: voices with
3 * their training-clip descriptions and cached features, per-instrument
4 * geometry overrides, the take's events, and UI knobs. Bulk audio and
5 * binary data live in IndexedDB: training-clip audio, the take's audio,
6 * and the bakes (the drum's is a couple of hundred MB, which is exactly
7 * why it is worth keeping). The stores can drift only in harmless
8 * directions: a clip whose audio is missing cannot be re-edited but keeps
9 * classifying, and a bake whose key no longer matches is ignored.
10 */
11import type { Binding } from './render/types.ts';
13export interface TrainHit { vec: number[]; peakDb: number; }
14export interface TrainClip {
15 id: string;
16 sr: number;
17 sens: number; // the sensitivity the onsets were detected with
18 onsets: number[]; // seconds
19 excluded: number[]; // indices into onsets
20 hits: TrainHit[]; // features of the included onsets, cached
21}
22export interface Voice {
23 id: string;
24 name: string;
25 color: string;
26 binding: Binding;
27 volume: number; // playback level, applied at mix time; never renders
28 clips: TrainClip[];
29 loose: TrainHit[]; // hits learned from takes; they belong to no clip
30}
31export interface TakeHit {
32 id: string; t: number; voiceId: string; gain: number;
33 peakDb: number; vec: number[] | null; // null for hand-added hits
34 s0: number; s1: number; // sample bounds in the take clip; -1 if hand-added
35}
37/** The pooled training set a voice contributes to classification. */
38export function trainPool(v: Voice): TrainHit[] {
39 return [...v.clips.flatMap((c) => c.hits), ...v.loose];
40}
42const KEY = 'instrawave-compose/v2';
43const KEY_V1 = 'instrawave-compose/v1';
44const KEY_IW = 'instrawave-compose/iw';
45const KEY_TAKE = 'instrawave-compose/take';
46const KEY_UI = 'instrawave-compose/ui';
48function read<T>(key: string): T | null {
49 try {
50 const raw = localStorage.getItem(key);
51 return raw ? (JSON.parse(raw) as T) : null;
52 } catch {
53 return null;
54 }
55}
57export function loadVoices(): Voice[] {
58 const doc = read<{ voices: Voice[] }>(KEY);
59 if (doc) {
60 for (const v of doc.voices) v.volume ??= 1; // pre-volume stores
61 return doc.voices;
62 }
63 // v1 stored a flat training list and a synth preset per voice.
64 const v1 = read<{ voices: { id: string; name: string; color: string; preset: string; train: TrainHit[] }[] }>(KEY_V1);
65 if (v1) {
66 const voices: Voice[] = v1.voices.map((v) => ({
67 id: v.id, name: v.name, color: v.color,
68 binding: { kind: 'synth', preset: v.preset as ('kick' | 'snare' | 'hat' | 'tom') },
69 volume: 1,
70 clips: [], loose: v.train,
71 }));
72 localStorage.setItem(KEY, JSON.stringify({ voices }));
73 localStorage.removeItem(KEY_V1);
74 return voices;
75 }
76 return [];
77}
79export function saveVoices(voices: Voice[]): void {
80 localStorage.setItem(KEY, JSON.stringify({ voices }));
81}
83/** Per-instrument geometry and accuracy overrides. */
84export function loadIwOverrides(): Record<string, Record<string, number>> {
85 return read(KEY_IW) ?? {};
86}
87export function saveIwOverrides(o: Record<string, Record<string, number>>): void {
88 localStorage.setItem(KEY_IW, JSON.stringify(o));
89}
91export function loadTakeHits(): TakeHit[] | null {
92 return read(KEY_TAKE);
93}
94export function saveTakeHits(hits: TakeHit[]): void {
95 localStorage.setItem(KEY_TAKE, JSON.stringify(hits));
96}
98export interface UiState { takeSens: number; captureSens: number; }
99export function loadUi(): UiState {
100 return read(KEY_UI) ?? { takeSens: 0.6, captureSens: 0.6 };
101}
102export function saveUi(ui: UiState): void {
103 localStorage.setItem(KEY_UI, JSON.stringify(ui));
104}
106/** How takes are classified: the shipped numbl algorithm (default, its
107 * source editable), a custom numbl algorithm, or the original TS
108 * feature-kNN. Null sources mean "the shipped default" / "the template". */
109export interface ClassifierState {
110 method: 'builtin' | 'custom' | 'legacy';
111 builtinSrc: string | null;
112 customSrc: string | null;
113}
114const KEY_CLF = 'instrawave-compose/classifier';
115export function loadClassifier(): ClassifierState {
116 return read(KEY_CLF) ?? { method: 'builtin', builtinSrc: null, customSrc: null };
117}
118export function saveClassifier(c: ClassifierState): void {
119 localStorage.setItem(KEY_CLF, JSON.stringify(c));
120}
122export function clearStore(): void {
123 for (const k of [KEY, KEY_IW, KEY_TAKE, KEY_UI, KEY_CLF]) localStorage.removeItem(k);
124}
126// ---- bulk data in IndexedDB ---------------------------------------------------
128const STORES = ['clips', 'takes', 'bakes', 'bakemeta', 'renders'] as const;
130let dbPromise: Promise<IDBDatabase> | null = null;
131function db(): Promise<IDBDatabase> {
132 return (dbPromise ??= new Promise((resolve, reject) => {
133 const req = indexedDB.open('instrawave-compose', 3);
134 req.onupgradeneeded = () => {
135 for (const s of STORES) {
136 if (!req.result.objectStoreNames.contains(s)) req.result.createObjectStore(s);
137 }
138 };
139 req.onsuccess = () => resolve(req.result);
140 req.onerror = () => reject(req.error);
141 }));
142}
144function op<T>(store: string, mode: IDBTransactionMode, fn: (s: IDBObjectStore) => IDBRequest<T>): Promise<T> {
145 return db().then((d) => new Promise<T>((resolve, reject) => {
146 const req = fn(d.transaction(store, mode).objectStore(store));
147 req.onsuccess = () => resolve(req.result);
148 req.onerror = () => reject(req.error);
149 }));
150}
152export function putClipAudio(id: string, samples: Float32Array, sr: number): Promise<unknown> {
153 return op('clips', 'readwrite', (s) => s.put({ samples, sr }, id));
154}
155export function getClipAudio(id: string): Promise<{ samples: Float32Array; sr: number } | null> {
156 return op<{ samples: Float32Array; sr: number } | undefined>('clips', 'readonly', (s) => s.get(id))
157 .then((r) => r ?? null);
158}
159export function deleteClipAudio(id: string): Promise<unknown> {
160 return op('clips', 'readwrite', (s) => s.delete(id));
161}
163export function putTakeAudio(samples: Float32Array, sr: number): Promise<unknown> {
164 return op('takes', 'readwrite', (s) => s.put({ samples, sr }, 'take'));
165}
166export function getTakeAudio(): Promise<{ samples: Float32Array; sr: number } | null> {
167 return op<{ samples: Float32Array; sr: number } | undefined>('takes', 'readonly', (s) => s.get('take'))
168 .then((r) => r ?? null);
169}
171/** Bake bytes and their key are separate stores so staleness can be
172 * checked without deserializing hundreds of MB. */
173export async function putBake(name: string, key: string, bytes: Uint8Array): Promise<void> {
174 await op('bakes', 'readwrite', (s) => s.put(bytes, name));
175 await op('bakemeta', 'readwrite', (s) => s.put(key, name));
176}
177export function getBakeKey(name: string): Promise<string | null> {
178 return op<string | undefined>('bakemeta', 'readonly', (s) => s.get(name)).then((r) => r ?? null);
179}
180export function getBakeBytes(name: string): Promise<Uint8Array | null> {
181 return op<Uint8Array | undefined>('bakes', 'readonly', (s) => s.get(name)).then((r) => r ?? null);
182}
184/** A rendered instrument track at the mix rate, keyed by the instrument
185 * and stamped with the exact events, tuning, and bake key it was rendered
186 * from; a matching signature means the GPU has nothing to redo. */
187export function putRender(name: string, sig: string, data: Float32Array): Promise<unknown> {
188 return op('renders', 'readwrite', (s) => s.put({ sig, data }, name));
189}
190export function getRender(name: string): Promise<{ sig: string; data: Float32Array } | null> {
191 return op<{ sig: string; data: Float32Array } | undefined>('renders', 'readonly', (s) => s.get(name))
192 .then((r) => r ?? null);
193}
194export function deleteRender(name: string): Promise<unknown> {
195 return op('renders', 'readwrite', (s) => s.delete(name));
196}
198export async function clearBulk(): Promise<void> {
199 for (const s of STORES) await op(s, 'readwrite', (st) => st.clear());
200}