forge-audio 0.1.0

//! DSP primitives: load, effects, write.

use std::collections::VecDeque;

/// Deterministic xorshift64 RNG. No heap, no deps.
struct SimpleRng(u64);
impl SimpleRng {
    fn new(seed: u64) -> Self { Self(seed.wrapping_add(1)) }
    fn next_u64(&mut self) -> u64 {
        self.0 ^= self.0 << 13;
        self.0 ^= self.0 >> 7;
        self.0 ^= self.0 << 17;
        self.0
    }
    fn next_f32(&mut self) -> f32 { (self.next_u64() >> 40) as f32 / (1u64 << 24) as f32 }
}

/// Mono or stereo audio buffer at a known sample rate.
#[derive(Debug, Clone)]
pub struct AudioBuffer {
    pub samples: Vec<Vec<f32>>,
    pub sample_rate: u32,
}

impl AudioBuffer {
    pub fn channels(&self) -> usize { self.samples.len() }
    pub fn len(&self) -> usize { self.samples.first().map(|c| c.len()).unwrap_or(0) }
    pub fn duration_secs(&self) -> f32 { self.len() as f32 / self.sample_rate as f32 }
    pub fn to_mono(&self) -> Vec<f32> {
        if self.channels() == 1 { return self.samples[0].clone(); }
        let n = self.len();
        let mut mono = vec![0.0; n];
        let scale = 1.0 / self.channels() as f32;
        for ch in &self.samples { for (i, &s) in ch.iter().enumerate() { mono[i] += s * scale; } }
        mono
    }
}

// --- I/O ---

pub fn load_wav(path: &str) -> Result<AudioBuffer, String> {
    let reader = hound::WavReader::open(path).map_err(|e| format!("WAV read: {}", e))?;
    let spec = reader.spec();
    let channels = spec.channels as usize;
    let raw: Vec<f32> = match spec.sample_format {
        hound::SampleFormat::Float => reader.into_samples::<f32>().map(|s| s.unwrap_or(0.0)).collect(),
        hound::SampleFormat::Int => {
            let max = (1i64 << (spec.bits_per_sample - 1)) as f32;
            reader.into_samples::<i32>().map(|s| s.unwrap_or(0) as f32 / max).collect()
        }
    };
    let mut samples = vec![Vec::new(); channels];
    for (i, &s) in raw.iter().enumerate() { samples[i % channels].push(s); }
    Ok(AudioBuffer { samples, sample_rate: spec.sample_rate })
}

pub fn write_wav(path: &str, buf: &AudioBuffer) -> Result<(), String> {
    let spec = hound::WavSpec { channels: buf.channels() as u16, sample_rate: buf.sample_rate, bits_per_sample: 32, sample_format: hound::SampleFormat::Float };
    let mut w = hound::WavWriter::create(path, spec).map_err(|e| format!("WAV write: {}", e))?;
    for i in 0..buf.len() { for ch in &buf.samples { w.write_sample(ch[i]).map_err(|e| format!("{}", e))?; } }
    w.finalize().map_err(|e| format!("{}", e))
}

/// Write game-ready audio: 16-bit WAV.
/// For OGG Vorbis export, use external tools (ffmpeg) or wait for a pure-Rust Vorbis encoder.
// TODO: Add OGG Vorbis export when a pure-Rust encoder crate becomes available.
pub fn write_game_audio(path: &str, buf: &AudioBuffer) -> Result<(), String> {
    let spec = hound::WavSpec {
        channels: buf.channels() as u16,
        sample_rate: buf.sample_rate,
        bits_per_sample: 16,
        sample_format: hound::SampleFormat::Int,
    };
    let mut w = hound::WavWriter::create(path, spec).map_err(|e| format!("WAV write: {}", e))?;
    for i in 0..buf.len() {
        for ch in &buf.samples {
            let sample = (ch[i].clamp(-1.0, 1.0) * 32767.0) as i16;
            w.write_sample(sample).map_err(|e| format!("{}", e))?;
        }
    }
    w.finalize().map_err(|e| format!("{}", e))
}

/// Universal audio loader using symphonia. Supports MP3, FLAC, OGG, WAV, and AIFF.
pub fn load_audio(path: &str) -> Result<AudioBuffer, String> {
    use symphonia::core::audio::SampleBuffer;
    use symphonia::core::codecs::DecoderOptions;
    use symphonia::core::formats::FormatOptions;
    use symphonia::core::io::MediaSourceStream;
    use symphonia::core::meta::MetadataOptions;
    use symphonia::core::probe::Hint;

    // 1. Open file and create MediaSourceStream
    let file = std::fs::File::open(path).map_err(|e| format!("open {}: {}", path, e))?;
    let mss = MediaSourceStream::new(Box::new(file), Default::default());

    // 2. Probe format using file extension hint
    let mut hint = Hint::new();
    if let Some(ext) = std::path::Path::new(path).extension().and_then(|e| e.to_str()) {
        hint.with_extension(ext);
    }

    let probed = symphonia::default::get_probe()
        .format(&hint, mss, &FormatOptions::default(), &MetadataOptions::default())
        .map_err(|e| format!("probe {}: {}", path, e))?;

    let mut format_reader = probed.format;

    // 3. Get default track, extract sample_rate and channel count
    let track = format_reader
        .default_track()
        .ok_or_else(|| "no default track".to_string())?;
    let sample_rate = track.codec_params.sample_rate.ok_or("no sample rate")?;
    let n_channels = track
        .codec_params
        .channels
        .map(|c| c.count())
        .unwrap_or(1);
    let track_id = track.id;

    let mut decoder = symphonia::default::get_codecs()
        .make(&track.codec_params, &DecoderOptions::default())
        .map_err(|e| format!("decoder: {}", e))?;

    // 4. Decode all packets into interleaved f32 samples
    let mut interleaved = Vec::<f32>::new();
    loop {
        let packet = match format_reader.next_packet() {
            Ok(p) => p,
            Err(symphonia::core::errors::Error::IoError(ref e))
                if e.kind() == std::io::ErrorKind::UnexpectedEof =>
            {
                break;
            }
            Err(_) => break,
        };
        if packet.track_id() != track_id {
            continue;
        }
        let decoded = match decoder.decode(&packet) {
            Ok(d) => d,
            Err(_) => continue,
        };
        let spec = *decoded.spec();
        let duration = decoded.capacity();
        let mut sample_buf = SampleBuffer::<f32>::new(duration as u64, spec);
        sample_buf.copy_interleaved_ref(decoded);
        interleaved.extend_from_slice(sample_buf.samples());
    }

    // 5. De-interleave into per-channel Vec<f32>
    let mut samples = vec![Vec::new(); n_channels];
    for (i, &s) in interleaved.iter().enumerate() {
        samples[i % n_channels].push(s);
    }

    // 6. Return AudioBuffer
    Ok(AudioBuffer {
        samples,
        sample_rate,
    })
}

// --- Original effects ---

pub fn time_stretch(buf: &AudioBuffer, factor: f64) -> Result<AudioBuffer, String> {
    let new_len = (buf.len() as f64 * factor) as usize;
    let samples = buf.samples.iter().map(|ch| {
        (0..new_len).map(|i| {
            let src = i as f64 / factor;
            let idx = src as usize;
            let frac = (src - idx as f64) as f32;
            let a = ch.get(idx).copied().unwrap_or(0.0);
            let b = ch.get(idx + 1).copied().unwrap_or(a);
            a + (b - a) * frac
        }).collect()
    }).collect();
    Ok(AudioBuffer { samples, sample_rate: buf.sample_rate })
}

pub fn pitch_shift(buf: &AudioBuffer, semitones: f32) -> Result<AudioBuffer, String> {
    let ratio = 2.0_f64.powf(semitones as f64 / 12.0);
    let mut stretched = time_stretch(buf, ratio)?;
    stretched.sample_rate = buf.sample_rate;
    Ok(stretched)
}

pub fn reverb(mut buf: AudioBuffer, room_size: f32, damping: f32, mix: f32) -> AudioBuffer {
    let sr = buf.sample_rate as f32;
    let comb_lengths = [(0.0297 * room_size * sr) as usize, (0.0371 * room_size * sr) as usize,
                        (0.0411 * room_size * sr) as usize, (0.0437 * room_size * sr) as usize];
    let ap_lengths = [(0.0050 * sr) as usize, (0.0017 * sr) as usize];
    for ch in &mut buf.samples {
        let mut comb_out = vec![0.0f32; ch.len()];
        for &len in &comb_lengths {
            let mut delay = VecDeque::from(vec![0.0f32; len.max(1)]);
            let mut fs = 0.0f32;
            for i in 0..ch.len() {
                let d = delay.pop_front().unwrap_or(0.0);
                fs = d * (1.0 - damping) + fs * damping;
                delay.push_back(ch[i] + fs * 0.7);
                comb_out[i] += d;
            }
        }
        for &len in &ap_lengths {
            let mut delay = VecDeque::from(vec![0.0f32; len.max(1)]);
            for i in 0..comb_out.len() {
                let d = delay.pop_front().unwrap_or(0.0);
                let inp = comb_out[i];
                comb_out[i] = d - 0.5 * inp;
                delay.push_back(inp + 0.5 * d);
            }
        }
        for i in 0..ch.len() { ch[i] = ch[i] * (1.0 - mix) + comb_out[i] * mix * 0.25; }
    }
    buf
}

pub fn delay(mut buf: AudioBuffer, time_ms: f32, feedback: f32, mix: f32) -> AudioBuffer {
    let dl = (time_ms * 0.001 * buf.sample_rate as f32) as usize;
    let dl = dl.max(1);
    for ch in &mut buf.samples {
        let mut ring = vec![0.0f32; dl];
        let mut pos = 0usize;
        for i in 0..ch.len() {
            let d = ring[pos % dl];
            ring[pos % dl] = ch[i] + d * feedback;
            ch[i] = ch[i] * (1.0 - mix) + d * mix;
            pos += 1;
        }
    }
    buf
}

pub fn lowpass(mut buf: AudioBuffer, cutoff_hz: f32) -> AudioBuffer {
    let alpha = {
        let rc = 1.0 / (2.0 * std::f32::consts::PI * cutoff_hz);
        let dt = 1.0 / buf.sample_rate as f32;
        dt / (rc + dt)
    };
    for ch in &mut buf.samples { let mut p = 0.0f32; for s in ch.iter_mut() { *s = p + alpha * (*s - p); p = *s; } }
    buf
}

pub fn highpass(mut buf: AudioBuffer, cutoff_hz: f32) -> AudioBuffer {
    let alpha = {
        let rc = 1.0 / (2.0 * std::f32::consts::PI * cutoff_hz);
        let dt = 1.0 / buf.sample_rate as f32;
        rc / (rc + dt)
    };
    for ch in &mut buf.samples {
        let (mut pi, mut po) = (0.0f32, 0.0f32);
        for s in ch.iter_mut() { let i = *s; *s = alpha * (po + i - pi); pi = i; po = *s; }
    }
    buf
}

pub fn granular(buf: &AudioBuffer, grain_ms: f32, density: f32, scatter: f32, seed: u64) -> AudioBuffer {
    let mut rng = SimpleRng::new(seed);
    let gl = (grain_ms * 0.001 * buf.sample_rate as f32) as usize;
    let n = buf.len();
    let mut out = AudioBuffer { samples: vec![vec![0.0; n]; buf.channels()], sample_rate: buf.sample_rate };
    let num = (density * n as f32 / gl.max(1) as f32) as usize;
    for _ in 0..num {
        let ss = (rng.next_f32() * n.saturating_sub(gl) as f32) as usize;
        let ds = ((rng.next_f32() * n as f32) as usize + (rng.next_f32() * scatter * n as f32) as usize) % n;
        for i in 0..gl.min(n - ds) {
            let env = hann(i, gl);
            for c in 0..buf.channels() { if ss + i < buf.samples[c].len() { out.samples[c][ds + i] += buf.samples[c][ss + i] * env; } }
        }
    }
    out
}

pub fn reverse(buf: &AudioBuffer) -> AudioBuffer {
    let mut out = buf.clone();
    for ch in &mut out.samples { ch.reverse(); }
    out
}

// --- 10 new effects ---

/// Phase vocoder time stretch (simplified overlap-add with windowed segments).
pub fn phase_vocoder(buf: &AudioBuffer, stretch_factor: f32) -> AudioBuffer {
    let win = 2048usize;
    let hop_in = win / 4;
    let hop_out = (hop_in as f32 * stretch_factor) as usize;
    let hop_out = hop_out.max(1);
    let new_len = (buf.len() as f32 * stretch_factor) as usize;
    let mut out_samples = vec![vec![0.0f32; new_len]; buf.channels()];
    for (ci, ch) in buf.samples.iter().enumerate() {
        let mut out_pos = 0usize;
        let mut in_pos = 0usize;
        while in_pos + win <= ch.len() && out_pos + win <= new_len {
            for j in 0..win {
                let env = hann(j, win);
                out_samples[ci][out_pos + j] += ch[in_pos + j] * env;
            }
            in_pos += hop_in;
            out_pos += hop_out;
        }
    }
    AudioBuffer { samples: out_samples, sample_rate: buf.sample_rate }
}

/// Notch (band-reject) filter.
pub fn notch_filter(buf: &AudioBuffer, freq_hz: f32, q: f32, _gain_db: f32) -> AudioBuffer {
    let mut out = buf.clone();
    let w0 = 2.0 * std::f32::consts::PI * freq_hz / buf.sample_rate as f32;
    let alpha = w0.sin() / (2.0 * q);
    let b0 = 1.0; let b1 = -2.0 * w0.cos(); let b2 = 1.0;
    let a0 = 1.0 + alpha; let a1 = -2.0 * w0.cos(); let a2 = 1.0 - alpha;
    for ch in &mut out.samples { biquad_stateless(ch, b0/a0, b1/a0, b2/a0, a1/a0, a2/a0); }
    out
}

/// LFO amplitude modulation.
pub fn lfo_modulate(buf: &AudioBuffer, rate_hz: f32, depth: f32) -> AudioBuffer {
    let mut out = buf.clone();
    let sr = buf.sample_rate as f32;
    for ch in &mut out.samples {
        for (i, s) in ch.iter_mut().enumerate() {
            let lfo = (2.0 * std::f32::consts::PI * rate_hz * i as f32 / sr).sin();
            *s *= 1.0 + lfo * depth;
        }
    }
    out
}

/// Reverse preswell: slice tail, reverse, prepend.
pub fn reverse_preswell(buf: &AudioBuffer, segment_ms: f32) -> AudioBuffer {
    let seg_len = (segment_ms * 0.001 * buf.sample_rate as f32) as usize;
    if seg_len == 0 || seg_len >= buf.len() { return buf.clone(); }
    let mut out = AudioBuffer { samples: Vec::new(), sample_rate: buf.sample_rate };
    for ch in &buf.samples {
        let tail_start = ch.len() - seg_len;
        let mut tail: Vec<f32> = ch[tail_start..].to_vec();
        tail.reverse();
        // Crossfade envelope on the reversed tail
        for (i, s) in tail.iter_mut().enumerate() { *s *= hann(i, seg_len); }
        let mut combined = tail;
        combined.extend_from_slice(ch);
        out.samples.push(combined);
    }
    out
}

/// Bitcrush: reduce bit depth and sample rate.
pub fn bitcrush(buf: &AudioBuffer, bit_depth: u32, target_rate: u32) -> AudioBuffer {
    let mut out = buf.clone();
    let levels = 2.0f32.powi(bit_depth.clamp(1, 32) as i32);
    let hold_every = (buf.sample_rate / target_rate.max(1)).max(1) as usize;
    for ch in &mut out.samples {
        let mut held = 0.0f32;
        for (i, s) in ch.iter_mut().enumerate() {
            if i % hold_every == 0 { held = (*s * levels).round() / levels; }
            *s = held;
        }
    }
    out
}

/// Bandpass filter (lowpass + highpass).
pub fn bandpass(buf: &AudioBuffer, low_hz: f32, high_hz: f32) -> AudioBuffer {
    highpass(lowpass(buf.clone(), high_hz), low_hz)
}

/// 3-band EQ using biquad shelving/peaking filters (no phase-splitting).
/// Low shelf at 200 Hz, mid peak at 1 kHz (Q=0.7), high shelf at 3 kHz.
/// Audio EQ Cookbook coefficients (Robert Bristow-Johnson).
///
/// `state` is `[low/mid/high][channel]` — persists across blocks for click-free filtering.
pub fn eq_3band(buf: &mut AudioBuffer, low_db: f32, mid_db: f32, high_db: f32, state: &mut [[BiquadState; 2]; 3]) {
    if low_db.abs() < 0.5 && mid_db.abs() < 0.5 && high_db.abs() < 0.5 {
        return;
    }

    let sr = buf.sample_rate as f32;

    // Low shelf at 200 Hz
    if low_db.abs() >= 0.5 {
        let c = biquad_low_shelf(200.0, low_db, 0.7, sr);
        // Coeffs are pre-normalized (divided by a0), so a0 = 1.0
        let coeffs = [c.0, c.1, c.2, 1.0, c.3, c.4];
        for (ch_idx, ch) in buf.samples.iter_mut().enumerate() {
            biquad(ch, &coeffs, &mut state[0][ch_idx.min(1)]);
        }
    }

    // Mid peak at 1 kHz
    if mid_db.abs() >= 0.5 {
        let c = biquad_peaking(1000.0, mid_db, 0.7, sr);
        let coeffs = [c.0, c.1, c.2, 1.0, c.3, c.4];
        for (ch_idx, ch) in buf.samples.iter_mut().enumerate() {
            biquad(ch, &coeffs, &mut state[1][ch_idx.min(1)]);
        }
    }

    // High shelf at 3 kHz
    if high_db.abs() >= 0.5 {
        let c = biquad_high_shelf(3000.0, high_db, 0.7, sr);
        let coeffs = [c.0, c.1, c.2, 1.0, c.3, c.4];
        for (ch_idx, ch) in buf.samples.iter_mut().enumerate() {
            biquad(ch, &coeffs, &mut state[2][ch_idx.min(1)]);
        }
    }
}

/// Offline 3-band EQ (stateless — for non-realtime use like file processing).
pub fn eq_3band_offline(buf: &AudioBuffer, low_db: f32, mid_db: f32, high_db: f32) -> AudioBuffer {
    let mut out = buf.clone();
    let mut state: [[BiquadState; 2]; 3] = Default::default();
    eq_3band(&mut out, low_db, mid_db, high_db, &mut state);
    out
}

// --- Helpers ---

/// Persistent biquad filter state — must survive across audio blocks.
#[derive(Default, Clone)]
pub struct BiquadState {
    pub x1: f32, pub x2: f32,
    pub y1: f32, pub y2: f32,
}

fn hann(i: usize, len: usize) -> f32 {
    if len <= 1 { return 1.0; }
    0.5 * (1.0 - (2.0 * std::f32::consts::PI * i as f32 / (len - 1) as f32).cos())
}

/// Biquad peaking EQ coefficients (Audio EQ Cookbook).
fn biquad_peaking(freq: f32, db_gain: f32, q: f32, sr: f32) -> (f32, f32, f32, f32, f32) {
    let a = 10.0_f32.powf(db_gain / 40.0);
    let w0 = 2.0 * std::f32::consts::PI * freq / sr;
    let alpha = w0.sin() / (2.0 * q);
    let a0 = 1.0 + alpha / a;
    ( (1.0 + alpha * a) / a0,
      (-2.0 * w0.cos()) / a0,
      (1.0 - alpha * a) / a0,
      (-2.0 * w0.cos()) / a0,
      (1.0 - alpha / a) / a0 )
}

/// Biquad low shelf coefficients (Audio EQ Cookbook).
fn biquad_low_shelf(freq: f32, db_gain: f32, q: f32, sr: f32) -> (f32, f32, f32, f32, f32) {
    let a = 10.0_f32.powf(db_gain / 40.0);
    let w0 = 2.0 * std::f32::consts::PI * freq / sr;
    let alpha = w0.sin() / (2.0 * q);
    let two_sqrt_a_alpha = 2.0 * a.sqrt() * alpha;
    let a0 = (a + 1.0) + (a - 1.0) * w0.cos() + two_sqrt_a_alpha;
    ( (a * ((a + 1.0) - (a - 1.0) * w0.cos() + two_sqrt_a_alpha)) / a0,
      (2.0 * a * ((a - 1.0) - (a + 1.0) * w0.cos())) / a0,
      (a * ((a + 1.0) - (a - 1.0) * w0.cos() - two_sqrt_a_alpha)) / a0,
      (-2.0 * ((a - 1.0) + (a + 1.0) * w0.cos())) / a0,
      ((a + 1.0) + (a - 1.0) * w0.cos() - two_sqrt_a_alpha) / a0 )
}

/// Biquad high shelf coefficients (Audio EQ Cookbook).
fn biquad_high_shelf(freq: f32, db_gain: f32, q: f32, sr: f32) -> (f32, f32, f32, f32, f32) {
    let a = 10.0_f32.powf(db_gain / 40.0);
    let w0 = 2.0 * std::f32::consts::PI * freq / sr;
    let alpha = w0.sin() / (2.0 * q);
    let two_sqrt_a_alpha = 2.0 * a.sqrt() * alpha;
    let a0 = (a + 1.0) - (a - 1.0) * w0.cos() + two_sqrt_a_alpha;
    ( (a * ((a + 1.0) + (a - 1.0) * w0.cos() + two_sqrt_a_alpha)) / a0,
      (-2.0 * a * ((a - 1.0) + (a + 1.0) * w0.cos())) / a0,
      (a * ((a + 1.0) + (a - 1.0) * w0.cos() - two_sqrt_a_alpha)) / a0,
      (2.0 * ((a - 1.0) - (a + 1.0) * w0.cos())) / a0,
      ((a + 1.0) - (a - 1.0) * w0.cos() - two_sqrt_a_alpha) / a0 )
}

/// Biquad filter with persistent state (for real-time block-by-block processing).
pub fn biquad(ch: &mut [f32], coeffs: &[f32; 6], state: &mut BiquadState) {
    let (b0, b1, b2, a0, a1, a2) = (coeffs[0], coeffs[1], coeffs[2], coeffs[3], coeffs[4], coeffs[5]);
    for s in ch.iter_mut() {
        let x0 = *s;
        let y0 = (b0 * x0 + b1 * state.x1 + b2 * state.x2 - a1 * state.y1 - a2 * state.y2) / a0;
        state.x2 = state.x1; state.x1 = x0;
        state.y2 = state.y1; state.y1 = y0;
        *s = y0;
    }
}

/// Resample an AudioBuffer to the target sample rate using linear interpolation.
/// Good enough for playback normalization — sinc resampler comes later.
pub fn resample(buf: &AudioBuffer, target_rate: u32) -> AudioBuffer {
    if buf.sample_rate == target_rate {
        return buf.clone();
    }
    let ratio = target_rate as f64 / buf.sample_rate as f64;
    let new_len = (buf.len() as f64 * ratio) as usize;
    let mut out_samples = Vec::with_capacity(buf.channels());

    for ch in &buf.samples {
        let mut resampled = Vec::with_capacity(new_len);
        for i in 0..new_len {
            let src_pos = i as f64 / ratio;
            let idx = src_pos as usize;
            let frac = (src_pos - idx as f64) as f32;
            let a = ch.get(idx).copied().unwrap_or(0.0);
            let b = ch.get(idx + 1).copied().unwrap_or(a);
            resampled.push(a + (b - a) * frac);
        }
        out_samples.push(resampled);
    }

    AudioBuffer {
        samples: out_samples,
        sample_rate: target_rate,
    }
}

/// Biquad filter with internal (zeroed) state — for offline/full-buffer processing.
fn biquad_stateless(ch: &mut [f32], b0: f32, b1: f32, b2: f32, a1: f32, a2: f32) {
    let (mut x1, mut x2, mut y1, mut y2) = (0.0f32, 0.0f32, 0.0f32, 0.0f32);
    for s in ch.iter_mut() {
        let x0 = *s;
        let y0 = b0 * x0 + b1 * x1 + b2 * x2 - a1 * y1 - a2 * y2;
        x2 = x1; x1 = x0; y2 = y1; y1 = y0;
        *s = y0;
    }
}

/// Compute 3-band waveform energy at `resolution`-sample windows.
/// Returns Vec<[low, mid, high]> with each value normalized 0.0–1.0.
/// Bands: low <200Hz, mid 200–4000Hz, high >4000Hz.
pub fn compute_waveform_bands(buf: &AudioBuffer, resolution: usize) -> Vec<[f32; 3]> {
    let mono = buf.to_mono();
    if mono.is_empty() || resolution == 0 { return vec![]; }
    let sr = buf.sample_rate as f32;
    let chunk = resolution.max(1);
    let n_windows = (mono.len() + chunk - 1) / chunk;

    // Biquad coefficients: lowpass at 200Hz, bandpass 200-4000Hz (via highpass+lowpass), highpass at 4000Hz
    // We use simple 2nd-order filters: LP@200, HP@200 then LP@4000 for mid, HP@4000 for high
    let lp200 = biquad_lp_coeffs(200.0, 0.707, sr);
    let hp200 = biquad_hp_coeffs(200.0, 0.707, sr);
    let lp4000 = biquad_lp_coeffs(4000.0, 0.707, sr);
    let hp4000 = biquad_hp_coeffs(4000.0, 0.707, sr);

    // Process full signal through filters
    let mut low_sig = mono.clone();
    let mut mid_sig = mono.clone();
    let mut high_sig = mono.clone();

    biquad_apply(&mut low_sig, &lp200);

    biquad_apply(&mut mid_sig, &hp200);
    biquad_apply(&mut mid_sig, &lp4000);

    biquad_apply(&mut high_sig, &hp4000);

    // Compute RMS per window, track global max
    let mut result = Vec::with_capacity(n_windows);
    let mut max_e = [0.0f32; 3];
    for w in 0..n_windows {
        let start = w * chunk;
        let end = (start + chunk).min(mono.len());
        let n = (end - start) as f32;
        if n <= 0.0 { result.push([0.0; 3]); continue; }
        let lo: f32 = low_sig[start..end].iter().map(|s| s * s).sum::<f32>() / n;
        let mi: f32 = mid_sig[start..end].iter().map(|s| s * s).sum::<f32>() / n;
        let hi: f32 = high_sig[start..end].iter().map(|s| s * s).sum::<f32>() / n;
        let e = [lo.sqrt(), mi.sqrt(), hi.sqrt()];
        for b in 0..3 { if e[b] > max_e[b] { max_e[b] = e[b]; } }
        result.push(e);
    }
    // Normalize to 0.0–1.0
    for r in &mut result {
        for b in 0..3 {
            if max_e[b] > 0.0 { r[b] /= max_e[b]; }
        }
    }
    result
}

/// Simple 2nd-order lowpass biquad coefficients.
fn biquad_lp_coeffs(freq: f32, q: f32, sr: f32) -> [f32; 5] {
    let w0 = 2.0 * std::f32::consts::PI * freq / sr;
    let alpha = w0.sin() / (2.0 * q);
    let cos_w0 = w0.cos();
    let a0 = 1.0 + alpha;
    let b0 = ((1.0 - cos_w0) / 2.0) / a0;
    let b1 = (1.0 - cos_w0) / a0;
    let b2 = b0;
    let a1 = (-2.0 * cos_w0) / a0;
    let a2 = (1.0 - alpha) / a0;
    [b0, b1, b2, a1, a2]
}

/// Simple 2nd-order highpass biquad coefficients.
fn biquad_hp_coeffs(freq: f32, q: f32, sr: f32) -> [f32; 5] {
    let w0 = 2.0 * std::f32::consts::PI * freq / sr;
    let alpha = w0.sin() / (2.0 * q);
    let cos_w0 = w0.cos();
    let a0 = 1.0 + alpha;
    let b0 = ((1.0 + cos_w0) / 2.0) / a0;
    let b1 = (-(1.0 + cos_w0)) / a0;
    let b2 = b0;
    let a1 = (-2.0 * cos_w0) / a0;
    let a2 = (1.0 - alpha) / a0;
    [b0, b1, b2, a1, a2]
}

/// Apply biquad filter in-place (stateless, single pass).
fn biquad_apply(samples: &mut [f32], coeffs: &[f32; 5]) {
    let (b0, b1, b2, a1, a2) = (coeffs[0], coeffs[1], coeffs[2], coeffs[3], coeffs[4]);
    let (mut x1, mut x2, mut y1, mut y2) = (0.0f32, 0.0f32, 0.0f32, 0.0f32);
    for s in samples.iter_mut() {
        let x0 = *s;
        let y0 = b0 * x0 + b1 * x1 + b2 * x2 - a1 * y1 - a2 * y2;
        x2 = x1; x1 = x0;
        y2 = y1; y1 = y0;
        *s = y0;
    }
}

#[cfg(test)]
mod tests {
    use super::*;

    fn sine(freq: f32, dur: f32, sr: u32) -> AudioBuffer {
        let n = (dur * sr as f32) as usize;
        let s = (0..n).map(|i| (2.0 * std::f32::consts::PI * freq * i as f32 / sr as f32).sin()).collect();
        AudioBuffer { samples: vec![s], sample_rate: sr }
    }

    #[test] fn reverb_len() { let b = sine(440.0, 0.5, 44100); let n = b.len(); assert_eq!(reverb(b, 1.0, 0.5, 0.3).len(), n); }
    #[test] fn lp_attenuates() { let b = sine(8000.0, 0.1, 44100); let b2 = sine(8000.0, 0.1, 44100); let o = lowpass(b, 200.0);
        let ei: f32 = b2.samples[0].iter().map(|s| s*s).sum(); let eo: f32 = o.samples[0].iter().map(|s| s*s).sum();
        assert!(eo < ei * 0.5); }
    #[test] fn rev_involution() { let b = sine(440.0, 0.1, 44100); assert_eq!(b.samples[0], reverse(&reverse(&b)).samples[0]); }
    #[test] fn notch_removes_freq() { let b = sine(1000.0, 0.2, 44100); let o = notch_filter(&b, 1000.0, 10.0, 0.0);
        let ei: f32 = b.samples[0].iter().map(|s| s*s).sum(); let eo: f32 = o.samples[0].iter().map(|s| s*s).sum();
        assert!(eo < ei * 0.3, "notch should remove target freq"); }
    #[test] fn bitcrush_quantizes() { let b = sine(440.0, 0.1, 44100); let o = bitcrush(&b, 4, 8000);
        let unique: std::collections::HashSet<u32> = o.samples[0].iter().map(|s| s.to_bits()).collect();
        assert!(unique.len() < b.len() / 2, "bitcrush should reduce unique values"); }
    #[test] fn bandpass_works() { let b = sine(440.0, 0.1, 44100); let o = bandpass(&b, 200.0, 800.0);
        assert_eq!(o.len(), b.len()); }
    #[test] fn phase_vocoder_stretches() { let b = sine(440.0, 0.5, 44100); let o = phase_vocoder(&b, 2.0);
        assert!(o.len() > b.len(), "vocoder should stretch"); }
    #[test] fn preswell_prepends() { let b = sine(440.0, 0.5, 44100); let o = reverse_preswell(&b, 100.0);
        assert!(o.len() > b.len(), "preswell should add samples"); }
}