1 files changed, 447 insertions, 0 deletions
diff --git a/tools/mq_editor/app.js b/tools/mq_editor/app.js
new file mode 100644
index 0000000..059d700
--- /dev/null
+++ b/tools/mq_editor/app.js
@@ -0,0 +1,447 @@
+// MQ Editor — application glue (extracted from index.html)
+
+// LP: y[n] = k*x[n] + (1-k)*y[n-1]  => -3dB at cos(w) = (2-2k-k²)/(2(1-k))
+function k1ToHz(k, sr) {
+  if (k >= 1.0) return sr / 2;
+  const cosW = (2 - 2*k - k*k) / (2*(1 - k));
+  return Math.acos(Math.max(-1, Math.min(1, cosW))) * sr / (2 * Math.PI);
+}
+// HP: y[n] = k*(y[n-1]+x[n]-x[n-1])  => -3dB from peak at cos(w) = 2k/(1+k²)
+function k2ToHz(k, sr) {
+  if (k >= 1.0) return 0;
+  const cosW = 2*k / (1 + k*k);
+  return Math.acos(Math.max(-1, Math.min(1, cosW))) * sr / (2 * Math.PI);
+}
+function fmtHz(f) {
+  return f >= 1000 ? (f/1000).toFixed(1) + 'k' : Math.round(f) + 'Hz';
+}
+function getSR() { return (typeof audioBuffer !== 'undefined' && audioBuffer) ? audioBuffer.sampleRate : 44100; }
+
+// LP/HP slider live display
+document.getElementById('lpK1').addEventListener('input', function() {
+  const k = parseFloat(this.value);
+  const f = k1ToHz(k, getSR());
+  document.getElementById('lpK1Val').textContent = k >= 1.0 ? 'bypass' : fmtHz(f);
+});
+document.getElementById('hpK2').addEventListener('input', function() {
+  const k = parseFloat(this.value);
+  const f = k2ToHz(k, getSR());
+  document.getElementById('hpK2Val').textContent = k >= 1.0 ? 'bypass' : fmtHz(f);
+});
+
+// Show/hide global resonator params when forceResonator toggled
+document.getElementById('forceResonator').addEventListener('change', function() {
+  document.getElementById('globalResParams').style.display = this.checked ? '' : 'none';
+});
+document.getElementById('globalR').addEventListener('input', function() {
+  document.getElementById('globalRVal').textContent = parseFloat(this.value).toFixed(4);
+});
+document.getElementById('globalGain').addEventListener('input', function() {
+  document.getElementById('globalGainVal').textContent = parseFloat(this.value).toFixed(2);
+});
+let audioBuffer = null;
+let viewer = null;
+let audioContext = null;
+let currentSource = null;
+let extractedPartials = null;
+let stftCache = null;
+
+const wavFile = document.getElementById('wavFile');
+const chooseFileBtn = document.getElementById('chooseFileBtn');
+const extractBtn = document.getElementById('extractBtn');
+const autoSpreadAllBtn = document.getElementById('autoSpreadAllBtn');
+const playBtn = document.getElementById('playBtn');
+const stopBtn = document.getElementById('stopBtn');
+const canvas = document.getElementById('canvas');
+const status = document.getElementById('status');
+const fileLabel = document.getElementById('fileLabel');
+
+const hopSize = document.getElementById('hopSize');
+const threshold = document.getElementById('threshold');
+const prominence = document.getElementById('prominence');
+const freqWeightCb = document.getElementById('freqWeight');
+const birthPersistenceEl = document.getElementById('birthPersistence');
+const deathAgeEl = document.getElementById('deathAge');
+const phaseErrorWeightEl = document.getElementById('phaseErrorWeight');
+const minLengthEl = document.getElementById('minLength');
+const keepPct = document.getElementById('keepPct');
+const keepPctLabel = document.getElementById('keepPctLabel');
+const fftSize = 1024; // Fixed
+
+function getKeepCount() {
+  return Math.max(1, Math.ceil(extractedPartials.length * parseInt(keepPct.value) / 100));
+}
+
+keepPct.addEventListener('input', () => {
+  keepPctLabel.textContent = keepPct.value + '%';
+  if (viewer && extractedPartials) viewer.setKeepCount(getKeepCount());
+});
+
+// --- Editor ---
+const editor = new PartialEditor();
+editor.onPartialDeleted = () => {
+  if (viewer && extractedPartials)
+    viewer.setKeepCount(extractedPartials.length > 0 ? getKeepCount() : 0);
+};
+
+// Initialize audio context
+function initAudioContext() {
+  if (!audioContext) {
+    audioContext = new (window.AudioContext || window.webkitAudioContext)();
+  }
+}
+
+// Shared: initialize editor from an AudioBuffer
+function loadAudioBuffer(buffer, label) {
+  audioBuffer = buffer;
+  initAudioContext();
+  extractBtn.disabled = false;
+  playBtn.disabled = false;
+  setStatus('Computing STFT cache...', 'info');
+
+  // Reset partials from previous file
+  extractedPartials = null;
+  editor.setPartials(null);
+
+  setTimeout(() => {
+    const signal = audioBuffer.getChannelData(0);
+    stftCache = new STFTCache(signal, audioBuffer.sampleRate, fftSize, Math.max(64, parseInt(hopSize.value) || 64));
+    setStatus(`${label} — ${audioBuffer.duration.toFixed(2)}s, ${audioBuffer.sampleRate}Hz, ${audioBuffer.numberOfChannels}ch (${stftCache.getNumFrames()} frames cached)`, 'info');
+    viewer = new SpectrogramViewer(canvas, audioBuffer, stftCache);
+    editor.setViewer(viewer);
+    viewer.onPartialSelect = (i) => editor.onPartialSelect(i);
+    viewer.onRender        = ()  => editor.onRender();
+    if (label.startsWith('Test WAV')) validateTestWAVPeaks(stftCache);
+  }, 10);
+}
+
+// File chooser button
+chooseFileBtn.addEventListener('click', () => wavFile.click());
+
+// Load WAV file
+wavFile.addEventListener('change', async (e) => {
+  const file = e.target.files[0];
+  if (!file) return;
+
+  fileLabel.textContent = file.name;
+  setStatus('Loading WAV...', 'info');
+  try {
+    const arrayBuffer = await file.arrayBuffer();
+    const ctx = new AudioContext();
+    const buf = await ctx.decodeAudioData(arrayBuffer);
+    loadAudioBuffer(buf, `Loaded: ${file.name}`);
+  } catch (err) {
+    setStatus('Error loading WAV: ' + err.message, 'error');
+    console.error(err);
+  }
+});
+
+// Test WAV: generate synthetic signal (two sine waves) in-memory
+document.getElementById('testWavBtn').addEventListener('click', () => {
+  initAudioContext();
+  const SR = 32000;
+  const duration = 2.0;
+  const numSamples = SR * duration;
+
+  // Two sine waves: 440 Hz (A4) + 660 Hz (E5, perfect fifth), equal amplitude
+  const buf = audioContext.createBuffer(1, numSamples, SR);
+  const data = buf.getChannelData(0);
+  for (let i = 0; i < numSamples; ++i) {
+    data[i] = 0.5 * Math.sin(2 * Math.PI * 440 * i / SR)
+            + 0.5 * Math.sin(2 * Math.PI * 660 * i / SR);
+  }
+
+  fileLabel.textContent = 'test-440+660hz.wav';
+  loadAudioBuffer(buf, 'Test WAV: 440Hz + 660Hz (2s, 32kHz)');
+});
+
+// Update cache when hop size changes
+hopSize.addEventListener('change', () => {
+  const val = Math.max(64, parseInt(hopSize.value) || 64);
+  hopSize.value = val;
+  if (stftCache) {
+    setStatus('Updating STFT cache...', 'info');
+    setTimeout(() => {
+      stftCache.setHopSize(val);
+      setStatus(`Cache updated (${stftCache.getNumFrames()} frames)`, 'info');
+      if (viewer) viewer.render();
+    }, 10);
+  }
+});
+
+function runExtraction() {
+  if (!stftCache) return;
+
+  setStatus('Extracting partials...', 'info');
+  extractBtn.disabled = true;
+
+  setTimeout(() => {
+    try {
+      const params = {
+        fftSize: fftSize,
+        hopSize: parseInt(hopSize.value),
+        threshold: parseFloat(threshold.value),
+        prominence: parseFloat(prominence.value),
+        freqWeight: freqWeightCb.checked,
+        birthPersistence: parseInt(birthPersistenceEl.value),
+        deathAge: parseInt(deathAgeEl.value),
+        phaseErrorWeight: parseFloat(phaseErrorWeightEl.value),
+        minLength: parseInt(minLengthEl.value),
+        sampleRate: audioBuffer.sampleRate
+      };
+
+      const result = extractPartials(params, stftCache);
+
+      // Sort by decreasing peak amplitude
+      result.partials.sort((a, b) => {
+        const peakA = a.amps.reduce((m, v) => Math.max(m, v), 0);
+        const peakB = b.amps.reduce((m, v) => Math.max(m, v), 0);
+        return peakB - peakA;
+      });
+
+      extractedPartials = result.partials;
+      editor.setPartials(result.partials);
+      viewer.setFrames(result.frames);
+      setStatus(`Extracted ${result.partials.length} partials`, 'info');
+      viewer.setPartials(result.partials);
+      viewer.setKeepCount(getKeepCount());
+      viewer.selectPartial(-1);
+
+    } catch (err) {
+      setStatus('Extraction error: ' + err.message, 'error');
+      console.error(err);
+    }
+    extractBtn.disabled = false;
+    autoSpreadAllBtn.disabled = false;
+  }, 50);
+}
+
+extractBtn.addEventListener('click', () => {
+  if (!audioBuffer) return;
+  runExtraction();
+});
+
+autoSpreadAllBtn.addEventListener('click', () => {
+  if (!extractedPartials || !stftCache) return;
+  const fs = stftCache.fftSize;
+  const sr = audioBuffer.sampleRate;
+  const defaults = { decay_alpha: 0.1, jitter: 0.05, spread_above: 0.02, spread_below: 0.02 };
+  for (const p of extractedPartials) {
+    const {spread_above, spread_below} = autodetectSpread(p, stftCache, fs, sr);
+    if (!p.replicas) p.replicas = { ...defaults };
+    p.replicas.spread_above = spread_above;
+    p.replicas.spread_below = spread_below;
+  }
+  if (viewer) viewer.render();
+  const sel = viewer ? viewer.selectedPartial : -1;
+  if (sel >= 0) editor.onPartialSelect(sel);
+  setStatus(`Auto-spread applied to ${extractedPartials.length} partials`, 'info');
+});
+
+threshold.addEventListener('change', () => {
+  if (stftCache) runExtraction();
+});
+
+freqWeightCb.addEventListener('change', () => {
+  if (stftCache) runExtraction();
+});
+
+for (const el of [birthPersistenceEl, deathAgeEl, phaseErrorWeightEl, minLengthEl]) {
+  el.addEventListener('change', () => { if (stftCache) runExtraction(); });
+}
+
+function playAudioBuffer(buffer, statusMsg) {
+  const startTime = audioContext.currentTime;
+  currentSource = audioContext.createBufferSource();
+  currentSource.buffer = buffer;
+  currentSource.connect(audioContext.destination);
+  currentSource.start();
+  currentSource.onended = () => {
+    currentSource = null;
+    playBtn.disabled = false;
+    stopBtn.disabled = true;
+    viewer.setPlayheadTime(-1);
+    setStatus('Stopped', 'info');
+  };
+  playBtn.disabled = true;
+  stopBtn.disabled = false;
+  setStatus(statusMsg, 'info');
+  function tick() {
+    if (!currentSource) return;
+    viewer.setPlayheadTime(audioContext.currentTime - startTime);
+    requestAnimationFrame(tick);
+  }
+  tick();
+}
+
+function stopAudio() {
+  if (currentSource) {
+    try { currentSource.stop(); } catch (e) {}
+    currentSource = null;
+  }
+  if (viewer) viewer.setPlayheadTime(-1);
+  playBtn.disabled = false;
+  stopBtn.disabled = true;
+  setStatus('Stopped', 'info');
+}
+
+// Play audio
+playBtn.addEventListener('click', () => {
+  if (!audioBuffer || !audioContext) return;
+  stopAudio();
+  playAudioBuffer(audioBuffer, 'Playing...');
+});
+
+// Stop audio
+stopBtn.addEventListener('click', () => {
+  stopAudio();
+});
+
+function setStatus(msg, type = '') {
+  status.innerHTML = msg;
+  status.className = type;
+}
+
+// Play synthesized audio
+function playSynthesized() {
+  if (!extractedPartials || extractedPartials.length === 0) {
+    setStatus('No partials extracted yet', 'warn');
+    return;
+  }
+  if (!audioBuffer || !audioContext) return;
+
+  stopAudio();
+
+  setStatus('Synthesizing...', 'info');
+
+  const keepCount = getKeepCount();
+  const partialsToUse = extractedPartials.slice(0, keepCount).filter(p => !p.muted);
+  setStatus(`Synthesizing ${partialsToUse.length}/${extractedPartials.length} partials (${keepPct.value}%)...`, 'info');
+
+  const integratePhase  = document.getElementById('integratePhase').checked;
+  const disableJitter   = document.getElementById('disableJitter').checked;
+  const disableSpread   = document.getElementById('disableSpread').checked;
+  const forceResonator  = document.getElementById('forceResonator').checked;
+  const lpK1Raw = parseFloat(document.getElementById('lpK1').value);
+  const hpK2Raw = parseFloat(document.getElementById('hpK2').value);
+  const k1 = lpK1Raw < 1.0 ? lpK1Raw : null;
+  const k2 = hpK2Raw < 1.0 ? hpK2Raw : null;
+  const forceRGain   = forceResonator && document.getElementById('forceRGain').checked;
+  const globalR      = parseFloat(document.getElementById('globalR').value);
+  const globalGain   = parseFloat(document.getElementById('globalGain').value);
+  const pcm = synthesizeMQ(partialsToUse, audioBuffer.sampleRate, audioBuffer.duration,
+                           integratePhase, {disableJitter, disableSpread, forceResonator,
+                                            forceRGain, globalR, globalGain, k1, k2});
+
+  if (viewer) {
+    viewer.setSynthStftCache(new STFTCache(pcm, audioBuffer.sampleRate, fftSize, parseInt(hopSize.value)));
+  }
+
+  const synthBuffer = audioContext.createBuffer(1, pcm.length, audioBuffer.sampleRate);
+  synthBuffer.getChannelData(0).set(pcm);
+  playAudioBuffer(synthBuffer, `Playing synthesized (${partialsToUse.length}/${extractedPartials.length} partials, ${keepPct.value}%)...`);
+}
+
+// Keyboard shortcuts
+document.addEventListener('keydown', (e) => {
+  if (e.target.tagName === 'INPUT' || e.target.tagName === 'TEXTAREA') return;
+  if (e.code === 'Digit1') {
+    e.preventDefault();
+    playSynthesized();
+  } else if (e.code === 'Digit2') {
+    e.preventDefault();
+    if (!playBtn.disabled) {
+      playBtn.click();
+    }
+  } else if (e.code === 'KeyP') {
+    e.preventDefault();
+    if (viewer) viewer.togglePeaks();
+  } else if (e.code === 'KeyA') {
+    e.preventDefault();
+    if (viewer) {
+      viewer.showSynthFFT = !viewer.showSynthFFT;
+      viewer.renderSpectrum();
+    }
+  } else if (e.code === 'KeyE') {
+    e.preventDefault();
+    if (!extractBtn.disabled) extractBtn.click();
+  } else if (e.code === 'Escape') {
+    if (viewer) viewer.selectPartial(-1);
+  }
+});
+
+// Curve tab switching
+document.querySelectorAll('.tab-btn').forEach(btn => {
+  btn.addEventListener('click', () => {
+    document.querySelectorAll('.tab-btn').forEach(b => b.classList.remove('active'));
+    btn.classList.add('active');
+    document.querySelectorAll('.tab-pane').forEach(p => p.style.display = 'none');
+    document.getElementById('tab' + btn.dataset.tab).style.display = '';
+  });
+});
+
+// --- Test WAV peak validation ---
+function validateTestWAVPeaks(cache) {
+  const SR = cache.sampleRate;
+  const N = cache.fftSize;
+  const binWidth = SR / N;  // Hz per bin
+  const numBins = N / 2;
+  const numBars = 100;      // mini-spectrum bar count
+
+  // Use a mid-signal frame (avoid edge effects)
+  const midFrame = cache.frames[Math.floor(cache.frames.length / 2)];
+  if (!midFrame) { console.error('[TestWAV] No frames computed'); return; }
+  const sq = midFrame.squaredAmplitude;
+  const t = midFrame.time;
+
+  console.group('[TestWAV] Peak validation @ t=' + t.toFixed(3) + 's');
+
+  // Top 5 bins by magnitude
+  const ranked = Array.from(sq)
+    .map((v, i) => ({ bin: i, freq: i * binWidth, db: 10 * Math.log10(Math.max(v, 1e-20)) }))
+    .sort((a, b) => b.db - a.db);
+  console.log('Top 5 FFT bins:');
+  ranked.slice(0, 5).forEach(x =>
+    console.log(`  bin ${x.bin.toString().padStart(3)}: ${x.freq.toFixed(1).padStart(7)}Hz  ${x.db.toFixed(1)}dB`));
+
+  // Expected bins for 440/660 Hz
+  const bin440 = Math.round(440 / binWidth);
+  const bin660 = Math.round(660 / binWidth);
+  const db440 = 10 * Math.log10(Math.max(sq[bin440], 1e-20));
+  const db660 = 10 * Math.log10(Math.max(sq[bin660], 1e-20));
+  console.log(`440Hz → bin ${bin440} (${(bin440 * binWidth).toFixed(1)}Hz): ${db440.toFixed(1)}dB`);
+  console.log(`660Hz → bin ${bin660} (${(bin660 * binWidth).toFixed(1)}Hz): ${db660.toFixed(1)}dB`);
+
+  // Validate: 440/660 Hz must be in top-10
+  const top10Freqs = ranked.slice(0, 10).map(x => x.freq);
+  const pass440 = top10Freqs.some(f => Math.abs(f - 440) < binWidth * 2);
+  const pass660 = top10Freqs.some(f => Math.abs(f - 660) < binWidth * 2);
+  console.log('Peak check: 440Hz ' + (pass440 ? 'PASS ✓' : 'FAIL ✗') +
+              ',  660Hz ' + (pass660 ? 'PASS ✓' : 'FAIL ✗'));
+
+  // Mini-spectrum: which bar do these peaks land in?
+  const bar440 = Math.floor(bin440 * numBars / numBins);
+  const bar660 = Math.floor(bin660 * numBars / numBins);
+  const sampledBin440 = Math.floor(bar440 * numBins / numBars);
+  const sampledBin660 = Math.floor(bar660 * numBars / numBars);
+  console.log('Mini-spectrum (linear scale, 100 bars):');
+  console.log(`  440Hz (bin ${bin440}) → bar ${bar440}/100  [bar samples bin ${sampledBin440} = ${(sampledBin440 * binWidth).toFixed(1)}Hz]`);
+  console.log(`  660Hz (bin ${bin660}) → bar ${bar660}/100  [bar samples bin ${Math.floor(bar660 * numBins / numBars)} = ${(Math.floor(bar660 * numBins / numBars) * binWidth).toFixed(1)}Hz]`);
+  if (bar440 < 5 || bar660 < 5) {
+    console.warn('  ⚠ BUG: peaks fall in bars ' + bar440 + ' and ' + bar660 +
+                 ' (leftmost ~' + Math.max(bar440, bar660) * 2 + 'px of 200px canvas)' +
+                 ' — linear scale hides low-frequency peaks. Need log-scale bar mapping.');
+  }
+
+  // Main spectrogram: confirm bins are in draw range
+  const mainFreqStart = 20, mainFreqEnd = 16000;
+  const inRange440 = 440 >= mainFreqStart && 440 <= mainFreqEnd;
+  const inRange660 = 660 >= mainFreqStart && 660 <= mainFreqEnd;
+  const norm440 = (Math.log2(440) - Math.log2(mainFreqStart)) / (Math.log2(mainFreqEnd) - Math.log2(mainFreqStart));
+  const norm660 = (Math.log2(660) - Math.log2(mainFreqStart)) / (Math.log2(mainFreqEnd) - Math.log2(mainFreqStart));
+  console.log('Main spectrogram (log Y-axis, 600px):');
+  console.log(`  440Hz: in range=${inRange440}, y=${Math.round(600 * (1 - norm440))}px, db=${db440.toFixed(1)}dB → intensity=${Math.min(1, Math.pow(Math.max(0, (db440 + 80) / 80), 2)).toFixed(2)}`);
+  console.log(`  660Hz: in range=${inRange660}, y=${Math.round(600 * (1 - norm660))}px, db=${db660.toFixed(1)}dB → intensity=${Math.min(1, Math.pow(Math.max(0, (db660 + 80) / 80), 2)).toFixed(2)}`);
+
+  console.groupEnd();
+}