SonicForgeStudio/app/static/js/services/soundfontPlayer.js

(function () {
    'use strict';

    const activeOscillators = {};

    let __gainNode = null;
    const getCtx = () => {
        if (typeof getAudioContext === 'function') {
            const ctx = getAudioContext();
            if (!__gainNode) {
                __gainNode = ctx.createGain();
                __gainNode.gain.value = 0.3;
                __gainNode.connect(ctx.destination);
            }
            return ctx;
        }
        if (!window.__sharedAudioCtx) {
            window.__sharedAudioCtx = new (window.AudioContext || window.webkitAudioContext)();
        }
        if (window.__sharedAudioCtx.state === 'suspended') {
            window.__sharedAudioCtx.resume();
        }
        if (!__gainNode) {
            __gainNode = window.__sharedAudioCtx.createGain();
            __gainNode.gain.value = 0.3;
            __gainNode.connect(window.__sharedAudioCtx.destination);
        }
        return window.__sharedAudioCtx;
    };

    const _channels = Array.from({ length: 16 }, () => ({ bank: 0, program: 0, isPercussion: false }));
    let _nextMelodicChannel = 0;

    let _synthInstance = null;
    let _initialized = false;
    let _initInProgress = false;
    let _currentSfId = null;

    const SonicSF = {
        loadedFonts: {},

        init: async function (audioContext) {
            if (_initialized && _synthInstance) return;
            if (_initInProgress) return;
            _initInProgress = true;
            if (!window.SpessaSynthClass || !window.__SpessaSynthCDN) {
                console.warn("[SonicSF] SpessaSynth CDN not loaded. Using oscillator fallback.");
                _initInProgress = false;
                return;
            }
            try {
                const procUrl = window.__SpessaSynthCDN + "spessasynth_processor.min.js";
                await audioContext.audioWorklet.addModule(procUrl);
                _synthInstance = new window.SpessaSynthClass(audioContext);
                await _synthInstance.isReady;
                // Try loading default SGM bank to verify SpessaSynth works
                try {
                    const resp = await fetch("/api/v1/plugins/soundfonts/download/sgm_v2.01?t=" + Date.now());
                    if (resp.ok) {
                        const buf = await resp.arrayBuffer();
                        await _synthInstance.soundBankManager.addSoundBank(buf.slice(0), "sgm_v2.01");
                        await _synthInstance.isReady;
                        const bankCount = _synthInstance.soundBankManager?.soundBankList?.length ?? 0;
                        if (bankCount > 0) {
                            _initialized = true;
                            _initInProgress = false;
                            console.log("[SonicSF] SpessaSynth ready with", bankCount, "banks.");
                            return;
                        }
                    }
                } catch (bankErr) {
                    console.warn("[SonicSF] Default bank load failed:", bankErr);
                }
                console.log("[SonicSF] SpessaSynth init OK but bank load failed. Using oscillator.");
            } catch (e) {
                console.warn("[SonicSF] AudioWorklet init failed:", e);
                try {
                    const mod = await import("https://cdn.jsdelivr.net/npm/spessasynth_lib@4.3.1/dist/index.js");
                    _synthInstance = new mod.WorkerSynthesizer(audioContext);
                    await _synthInstance.isReady;
                    console.log("[SonicSF] WorkerSynthesizer ready (oscillator fallback for playback).");
                } catch (e2) {
                    console.error("[SonicSF] All SpessaSynth paths failed:", e2);
                }
            }
            // Always reset initialized — playNote falls through to oscillator
            _initialized = false;
            _initInProgress = false;
        },

        selectInstrument: async function (channel, bank, program, sfId) {
            if (!_initialized || !_synthInstance) return;
            if (sfId) {
                const ok = await this.loadSoundFont(sfId);
                if (!ok) {
                    console.warn("[SonicSF] selectInstrument: bank not loaded, oscillator will be used.");
                    return;
                }
            }
            try { _synthInstance.controllerChange(channel, 0, bank); } catch (e) {}
            try { _synthInstance.controllerChange(channel, 32, 0); } catch (e) {}
            try { _synthInstance.programChange(channel, program); } catch (e) {}
            this.controllerChange(channel, 0, bank);
            this.programChange(channel, program);
            console.log("[SonicSF] selectInstrument ch", channel, "bank", bank, "prog", program, "sf", sfId);
        },

        loadSoundFont: async function (sfId) {
            if (!_initialized || !_synthInstance) return;
            if (_currentSfId === sfId) return;
            console.log("[SonicSF] Loading SoundFont:", sfId);

            let buffer = null;
            if (window.SonicSFStorage) {
                buffer = await window.SonicSFStorage.getBuffer(sfId);
            }
            if (!buffer) {
                try {
                    const resp = await fetch("/api/v1/plugins/soundfonts/download/" + encodeURIComponent(sfId) + "?t=" + Date.now());
                    if (!resp.ok) throw new Error("Download failed: " + resp.status);
                    buffer = await resp.arrayBuffer();
                    if (window.SonicSFStorage) {
                        await window.SonicSFStorage.saveBuffer(sfId, buffer);
                    }
                } catch (e) {
                    console.error("[SonicSF] Failed to download:", sfId, e);
                    return false;
                }
            }
            try {
                const bufCopy = buffer.slice(0);
                await _synthInstance.soundBankManager.addSoundBank(bufCopy, sfId);
                await _synthInstance.isReady;
                _currentSfId = sfId;
                console.log("[SonicSF] SoundFont loaded:", sfId);
                return true;
            } catch (e) {
                console.error("[SonicSF] Error loading SF in SpessaSynth:", sfId, e);
                if (window.SonicSFStorage) {
                    try {
                        const db = await window.SonicSFStorage.openDB();
                        const tx = db.transaction(window.SonicSFStorage.storeName, "readwrite");
                        tx.objectStore(window.SonicSFStorage.storeName).delete(sfId);
                    } catch (ce) {}
                }
                _currentSfId = null;
                return false;
            }
        },

        controllerChange: function (channel, controller, value) {
            if (channel < 0 || channel > 15) return;
            if (_initialized && _synthInstance) {
                try { _synthInstance.controllerChange(channel, controller, value); } catch (e) {}
            }
            if (controller === 0) {
                _channels[channel].bank = value;
                _channels[channel].isPercussion = (value === 128);
            }
        },

        programChange: function (channel, program) {
            if (channel < 0 || channel > 15) return;
            if (_initialized && _synthInstance) {
                try { _synthInstance.programChange(channel, program); } catch (e) {}
            }
            _channels[channel].program = program;
        },

        allocateChannel: function (bank) {
            if (bank === 128) return 9;
            const ch = _nextMelodicChannel % 9;
            _nextMelodicChannel = (_nextMelodicChannel + 1) % 9;
            return ch;
        },

        applyAITrackInstrument: function (bank, program, synthEngine) {
            if (synthEngine) {
                bank = bank !== undefined ? bank : (synthEngine.soundfont_bank || 0);
                program = program !== undefined ? program : (synthEngine.soundfont_program || 0);
            }
            const channel = this.allocateChannel(bank);
            this.selectInstrument(channel, bank, program, synthEngine?.soundfont_id);
            return channel;
        },

        getChannelState: function (channel) {
            if (channel < 0 || channel > 15) return null;
            return { ..._channels[channel] };
        },

        playNote: function (note, velocity, durationMs, startTime, program, destinationNode, channel, synthEngine) {
            if (_initialized && _synthInstance && _currentSfId) {
                return this._playNoteSpessa(note, velocity, durationMs, startTime, program, channel, synthEngine);
            } else {
                if (_initialized && _synthInstance && !_currentSfId) {
                    console.warn("[SonicSF] No sound bank loaded, using oscillator fallback");
                }
                return this._playNoteOsc(note, velocity, durationMs, startTime, program, destinationNode, channel, synthEngine);
            }
        },

        _playNoteSpessa: function (note, velocity, durationMs, startTime, program, channel, synthEngine) {
            const ctx = getCtx();
            const midiPitch = Math.min(127, Math.max(0, parseInt(note) || 60));
            const midiVel = Math.min(127, Math.max(1, Math.floor((typeof velocity === 'number' ? (velocity > 1 ? velocity : velocity * 127) : 100))));

            if (synthEngine) {
                const ch = channel !== undefined ? channel : (synthEngine.soundfont_bank === 128 ? 9 : 0);
                try { _synthInstance.controllerChange(ch, 0, synthEngine.soundfont_bank || 0); } catch (e) {}
                try { _synthInstance.controllerChange(ch, 32, 0); } catch (e) {}
                try { _synthInstance.programChange(ch, synthEngine.soundfont_program || 0); } catch (e) {}
                if (channel === undefined) channel = ch;
            }
            if (channel === undefined) channel = 0;

            const durSec = durationMs / 1000;
            const now = ctx.currentTime;
            const scheduledTime = (typeof startTime === 'number' && startTime > now) ? (startTime - now) : 0;

            const doNoteOn = () => {
                try {
                    _synthInstance.noteOn(channel, midiPitch, midiVel);
                    setTimeout(() => { try { _synthInstance.noteOff(channel, midiPitch); } catch (e) {} }, durSec * 1000);
                } catch (e) {
                    console.warn("[SonicSF] SpessaSynth noteOn failed, falling back to oscillator:", e);
                    this._playNoteOsc(note, velocity, durationMs, startTime, program, null, channel, synthEngine);
                }
            };
            if (scheduledTime > 0) setTimeout(doNoteOn, scheduledTime * 1000); else doNoteOn();
        },

        _playNoteOsc: function (note, velocity, durationMs, startTime, program, destinationNode, channel, synthEngine) {
            const ctx = getCtx();
            const freq = 440 * Math.pow(2, (note - 69) / 12);
            if (freq <= 0 || isNaN(freq)) return null;

            if (synthEngine) {
                const ch = channel !== undefined ? channel : (synthEngine.soundfont_bank === 128 ? 9 : 0);
                this.controllerChange(ch, 0, synthEngine.soundfont_bank || 0);
                this.programChange(ch, synthEngine.soundfont_program || 0);
                if (channel === undefined) channel = ch;
                if (program === undefined) program = synthEngine.soundfont_program;
            }

            const osc = ctx.createOscillator();
            const noteGain = ctx.createGain();

            let oscType = 'triangle';
            let attackTime = 0.03;
            let decayTime = 0.1;
            let sustainLevel = 0.5;
            let releaseTime = 0.2;
            let volFactor = 0.25;

            let prog = program !== undefined ? parseInt(program) : 0;
            if (channel !== undefined && channel >= 0 && channel < 16) {
                const chState = _channels[channel];
                prog = chState.program || prog;
            }
            if (prog >= 0 && prog <= 7) {
                oscType = 'sine';
                decayTime = 0.3;
                sustainLevel = 0.1;
                releaseTime = 0.2;
            } else if (prog >= 8 && prog <= 15) {
                oscType = 'sine';
                decayTime = 0.1;
                sustainLevel = 0.0;
                releaseTime = 0.1;
            } else if (prog >= 16 && prog <= 23) {
                oscType = 'sine';
                attackTime = 0.05;
                sustainLevel = 0.8;
                releaseTime = 0.1;
            } else if (prog >= 24 && prog <= 31) {
                oscType = 'triangle';
                decayTime = 0.4;
                sustainLevel = 0.2;
                releaseTime = 0.3;
            } else if (prog >= 32 && prog <= 39) {
                oscType = 'triangle';
                attackTime = 0.02;
                decayTime = 0.2;
                sustainLevel = 0.6;
                releaseTime = 0.2;
            } else if (prog >= 40 && prog <= 47) {
                oscType = 'sawtooth';
                attackTime = 0.15;
                sustainLevel = 0.8;
                releaseTime = 0.5;
                volFactor = 0.15;
            } else if (prog >= 48 && prog <= 55) {
                oscType = 'sawtooth';
                attackTime = 0.2;
                sustainLevel = 0.8;
                releaseTime = 0.6;
                volFactor = 0.12;
            } else if (prog >= 56 && prog <= 63) {
                oscType = 'sawtooth';
                attackTime = 0.08;
                sustainLevel = 0.7;
                releaseTime = 0.3;
                volFactor = 0.15;
            } else if (prog >= 64 && prog <= 71) {
                oscType = 'square';
                attackTime = 0.05;
                sustainLevel = 0.6;
                releaseTime = 0.2;
                volFactor = 0.15;
            } else if (prog >= 72 && prog <= 79) {
                oscType = 'sine';
                attackTime = 0.1;
                sustainLevel = 0.7;
                releaseTime = 0.3;
                volFactor = 0.2;
            } else if (prog >= 80 && prog <= 119) {
                oscType = 'sawtooth';
                attackTime = 0.05;
                sustainLevel = 0.6;
                releaseTime = 0.4;
                volFactor = 0.15;
            }

            osc.type = oscType;
            osc.frequency.setValueAtTime(freq, 0);

            const startAt = startTime !== undefined ? startTime : ctx.currentTime;
            const durSec = durationMs / 1000;

            const vel = typeof velocity === 'number' ? (velocity > 1 ? velocity / 127 : velocity) : 0.8;
            const targetGain = vel * volFactor;

            noteGain.gain.setValueAtTime(0, startAt);
            noteGain.gain.linearRampToValueAtTime(targetGain, startAt + attackTime);
            noteGain.gain.linearRampToValueAtTime(targetGain * sustainLevel, startAt + attackTime + decayTime);

            const releaseStart = startAt + Math.max(attackTime + decayTime, durSec);
            noteGain.gain.linearRampToValueAtTime(targetGain * sustainLevel, releaseStart);
            noteGain.gain.linearRampToValueAtTime(0, releaseStart + releaseTime);

            osc.connect(noteGain);

            const dest = destinationNode || __gainNode || ctx.destination;
            noteGain.connect(dest);

            osc.start(startAt);

            const stopAt = releaseStart + releaseTime + 0.02;
            osc.stop(stopAt);

            const oscId = `${note}_${Date.now()}_${Math.random()}`;
            activeOscillators[oscId] = { osc, gain: noteGain };

            setTimeout(() => {
                delete activeOscillators[oscId];
            }, (stopAt - ctx.currentTime) * 1000 + 100);

            return osc;
        },

        stopAll: function () {
            if (_initialized && _synthInstance) {
                try {
                    for (let ch = 0; ch < 16; ch++) _synthInstance.allNotesOff(ch);
                } catch (e) {}
            }
            const ctx = getCtx();
            const now = ctx.currentTime;
            Object.values(activeOscillators).forEach(entry => {
                try {
                    if (entry.gain) {
                        entry.gain.gain.cancelScheduledValues(now);
                        entry.gain.gain.setValueAtTime(0, now);
                    }
                    if (entry.osc) try { entry.osc.stop(now); } catch (e) { }
                } catch (e) { }
            });
            Object.keys(activeOscillators).forEach(k => delete activeOscillators[k]);
        },

        saveToIndexedDB: async function (name, arrayBuffer) {
            if (window.SonicStorage && window.SonicStorage.saveToIndexedDB) {
                await window.SonicStorage.saveToIndexedDB('soundfont_' + name, arrayBuffer);
            }
        },

        loadFromIndexedDB: async function (name) {
            if (window.SonicStorage && window.SonicStorage.loadFromIndexedDB) {
                return await window.SonicStorage.loadFromIndexedDB('soundfont_' + name);
            }
            return null;
        }
    };

    window.SonicSF = SonicSF;
})();