85 lines
7.1 KiB
JavaScript
85 lines
7.1 KiB
JavaScript
// SPDX-License-Identifier: GPL-3.0-or-later
|
||
function assetURL(path, fallback) { return globalThis.Lazy?.url?.(path) || new URL(fallback, import.meta.url); }
|
||
const [parameterAPI, irUtils, irStore] = await Promise.all([
|
||
import(assetURL('/vendor/jamesdsp/params.mjs', './params.mjs')),
|
||
import(assetURL('/vendor/jamesdsp/ir-utils.mjs', './ir-utils.mjs')),
|
||
import(assetURL('/vendor/jamesdsp/ir-store.mjs', './ir-store.mjs')),
|
||
]);
|
||
const {DEFAULT_PARAMS, validateParam} = parameterAPI;
|
||
const {prepareIR,resampleIR} = irUtils;
|
||
const {getIR} = irStore;
|
||
const moduleLoads = new WeakMap();
|
||
export class JamesDSPNode extends EventTarget {
|
||
static async create(context, {wasmURL=assetURL('/vendor/jamesdsp/dsp/jamesdsp.wasm','./dsp/jamesdsp.wasm'), workletURL=assetURL('/vendor/jamesdsp/worklet.mjs','./worklet.mjs')}={}) {
|
||
if (!Number.isFinite(context.sampleRate)||context.sampleRate<8000||context.sampleRate>96000) throw new RangeError('AudioContext rate must be 8–96 kHz');
|
||
if (!moduleLoads.has(context)) moduleLoads.set(context,new Map());
|
||
const cache=moduleLoads.get(context), key=String(workletURL);
|
||
if (!cache.has(key)) cache.set(key,context.audioWorklet.addModule(workletURL).catch(e=>{cache.delete(key);throw e;}));
|
||
const response=await fetch(wasmURL);
|
||
if (!response.ok) throw new Error(`WASM fetch failed: ${response.status}`);
|
||
const wasmBytes=await response.arrayBuffer();
|
||
await cache.get(key);
|
||
const wrapper=new JamesDSPNode(context,wasmBytes);
|
||
try { await wrapper.ready; return wrapper; } catch(e) { wrapper.destroy(); throw e; }
|
||
}
|
||
constructor(context,wasmBytes) {
|
||
super(); this.context=context; this.defaultParams=Object.fromEntries(Object.entries(DEFAULT_PARAMS).filter(([name])=>!/^(?:liveprog|ddc|arbitrary)\./.test(name))); this.params={...this.defaultParams}; this.pending=new Map(); this.serial=0; this.destroyed=false; this.source=null; this.irRef=null;
|
||
this.node=new AudioWorkletNode(context,'jamesdsp',{numberOfInputs:1,numberOfOutputs:1,outputChannelCount:[2],channelCount:2,channelCountMode:'explicit',processorOptions:{wasmBytes}});
|
||
this.ready=new Promise((resolve,reject)=> {
|
||
const timer=setTimeout(()=>reject(new Error('DSP initialization timed out')),15000);
|
||
this.node.onprocessorerror=()=>{const error=new Error('DSP processor stopped');clearTimeout(timer);reject(error);this.fail(error);};
|
||
this.node.port.onmessage=({data})=> {
|
||
if(data.type==='ready') {clearTimeout(timer);resolve();}
|
||
else if(data.type==='reply') { const p=this.pending.get(data.id); if(p) {this.pending.delete(data.id);clearTimeout(p.timer);data.error?p.reject(new Error(data.error)):p.resolve(data.result);} }
|
||
else if(data.type==='error') {const error=new Error(data.error); clearTimeout(timer); reject(error); this.fail(error);}
|
||
else this.dispatchEvent(new CustomEvent(data.type,{detail:data}));
|
||
};
|
||
});
|
||
}
|
||
fail(error) { for(const p of this.pending.values()) {clearTimeout(p.timer);p.reject(error);} this.pending.clear();this.dispatchEvent(new CustomEvent('error',{detail:error})); }
|
||
request(type,data={},transfer=[]) {
|
||
if(this.destroyed) return Promise.reject(new Error('DSP has been destroyed'));
|
||
const id=++this.serial;
|
||
return new Promise((resolve,reject)=> {
|
||
const timer=setTimeout(()=>{this.pending.delete(id);reject(new Error(`DSP ${type} timed out`));},15000);
|
||
this.pending.set(id,{resolve,reject,timer});this.node.port.postMessage({type,id,...data},transfer);
|
||
});
|
||
}
|
||
// One source → this DSP → destination. For arbitrary graphs use .node directly.
|
||
connect(source,destination=this.context.destination) {
|
||
if(this.destroyed) throw new Error('DSP has been destroyed');
|
||
if(this.source) this.source.disconnect(this.node);
|
||
this.node.disconnect();this.source=source;source.connect(this.node);this.node.connect(destination);return destination;
|
||
}
|
||
async setParam(name,value) {if(/^(?:liveprog|ddc|arbitrary)\./.test(name))throw new Error('Text effects are disabled in this player version');validateParam(name,value);await this.request('param',{name,value});this.params[name]=value;}
|
||
setBypass(enabled) {return this.setParam('bypass',Number(Boolean(enabled)));}
|
||
async setText(effect,text) {
|
||
throw new Error('Text effects are disabled in this player version');
|
||
}
|
||
async loadIR(buffer,reference=null) {
|
||
buffer=prepareIR(this.context,buffer,{normalize:false});
|
||
const internalRate=this.context.sampleRate===44100?44100:48000;
|
||
buffer=resampleIR(buffer,internalRate);
|
||
if(![1,2,4].includes(buffer.numberOfChannels) || buffer.length>65536 || buffer.length<1) throw new RangeError('IR must have 1, 2, or 4 channels and at most 65536 frames');
|
||
const pcm=new Float32Array(buffer.length*buffer.numberOfChannels);
|
||
for(let ch=0;ch<buffer.numberOfChannels;ch++) {const data=buffer.getChannelData(ch);for(let i=0;i<buffer.length;i++) pcm[i*buffer.numberOfChannels+ch]=data[i];}
|
||
await this.request('ir',{pcm,frames:buffer.length,channels:buffer.numberOfChannels,sourceRate:internalRate},[pcm.buffer]);this.params['convolver.enabled']=0;this.irRef=reference?{name:reference.name,hash:reference.hash}:null;
|
||
}
|
||
async getInfo() {const {bytes,heapBytes}=await this.request('info');const view=new DataView(bytes);return {abiVersion:view.getUint32(0,true),structBytes:view.getUint32(4,true),contextRate:view.getFloat32(8,true),internalRate:view.getFloat32(12,true),srcBackend:view.getUint32(16,true),latencyFrames:view.getUint32(20,true),irChannels:view.getUint32(24,true),irFrames:view.getUint32(28,true),irCapFrames:view.getUint32(32,true),headFrames:view.getUint32(36,true),tailFrames:view.getUint32(40,true),liveBytes:view.getUint32(44,true),peakBytes:view.getUint32(48,true),faults:view.getUint32(52,true),underruns:view.getUint32(56,true),overflows:view.getUint32(60,true),heapBytes};}
|
||
async applyPreset(preset) {
|
||
// All keys validated before mutation; applying a preset resets unspecified params.
|
||
if(preset?.texts && Object.keys(preset.texts).length) throw new Error('Text effects are disabled in this player version');
|
||
const params={...this.defaultParams,...(preset.params||preset)};
|
||
for(const [name,value] of Object.entries(params)) {if(/^(?:liveprog|ddc|arbitrary)\./.test(name))throw new Error('Text effects are disabled in this player version');validateParam(name,value);}
|
||
if(preset.irRef) {const saved=await getIR(preset.irRef.name);if(!saved||saved.hash!==preset.irRef.hash)throw new Error(`Preset IR is missing or changed: ${preset.irRef.name}`);await this.loadIR(saved.buffer,preset.irRef);}
|
||
// Configure before enabling; the referenced IR is loaded first.
|
||
for(const [name,value] of Object.entries(params).sort(([a],[b])=>Number(a.endsWith('.enabled'))-Number(b.endsWith('.enabled')))) await this.setParam(name,value);
|
||
}
|
||
toJSON() {return {version:2,params:{...this.params},...(this.irRef?{irRef:{...this.irRef}}:{})};}
|
||
destroy() {
|
||
if(this.destroyed) return;
|
||
if(this.source) {try {this.source.disconnect(this.node);} catch {}} this.node.disconnect();
|
||
this.node.port.postMessage({type:'destroy'});this.destroyed=true;this.fail(new Error('DSP destroyed'));this.node.port.close();
|
||
}
|
||
}
|