cathar 0.7.0

Audio toolkit in pure Rust — denoise, de-hum, de-click, de-clip, de-reverb, normalise, and more.
Documentation 
1
2
3
4
5
6
7
8
9
10
11
12
13
14
15
16
17
18
19
20
21
22
23
24
25
26
27
28
29
30
31
32
33
34
35
36
37
38
39
40
41
42
43
44
45
46
47
48
49
50
51
52
53
54
55
56
57
58
59
60
61
62
63
64
65
66
67
68
69
70
71
72
73
74
75

//! Dequantization: reduce grain from low-bit-depth sources.

/// Reduce quantization grain from a signal assumed to have been recorded at
/// `bits` bits per sample.
///
/// Uses neighbour linear prediction: each sample is nudged toward the local
/// midpoint of its neighbours while staying within one quantisation step of its
/// lattice position. `strength` in `[0, 1]` controls correction (0 = bypass).
pub fn dequantize(signal: &[f32], _sample_rate: u32, bits: u32, strength: f32) -> Vec<f32> {
    let strength = strength.clamp(0.0, 1.0);
    let n = signal.len();
    if strength <= 0.0 || n < 3 {
        return signal.to_vec();
    }
    let bits = bits.clamp(4, 24);
    let step = 2.0f32 / (1u32 << bits) as f32;
    let half = step * 0.5;

    let mut out = signal.to_vec();
    for i in 1..n - 1 {
        let q = (out[i] / step).round() * step;
        let pred = (out[i - 1] + out[i + 1]) * 0.5;
        let target = pred.clamp(q - half, q + half);
        out[i] += strength * (target - out[i]);
    }

    // Second pass with a wider window for low-frequency content.
    if strength > 0.3 && n >= 5 {
        let src = out.clone();
        for i in 2..n - 2 {
            let q = (src[i] / step).round() * step;
            let pred = (src[i - 2] + src[i - 1] + src[i + 1] + src[i + 2]) * 0.25;
            let target = pred.clamp(q - half, q + half);
            out[i] += strength * 0.5 * (target - out[i]);
        }
    }
    out
}

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

    fn quantize(signal: &[f32], bits: u32) -> Vec<f32> {
        let step = 2.0f32 / (1u32 << bits) as f32;
        signal.iter().map(|s| (s / step).round() * step).collect()
    }

    #[test]
    fn dequantize_reduces_quantization_error() {
        let fs = 48_000u32;
        let n = fs as usize * 2;
        let clean: Vec<f32> = (0..n)
            .map(|i| {
                0.5 * (2.0 * std::f32::consts::PI * 440.0 * i as f32 / fs as f32).sin()
                    + 0.3 * (2.0 * std::f32::consts::PI * 880.0 * i as f32 / fs as f32).sin()
            })
            .collect();
        let noisy = quantize(&clean, 6);
        let restored = dequantize(&noisy, fs, 6, 1.0);
        let err = |a: &[f32], b: &[f32]| {
            a.iter().zip(b).map(|(x, y)| (x - y).abs()).sum::<f32>() / a.len() as f32
        };
        assert!(
            err(&restored, &clean) < err(&noisy, &clean),
            "dequantize should move closer to the clean signal"
        );
    }

    #[test]
    fn dequantize_zero_strength_is_bypass() {
        let sig = vec![0.1, 0.2, 0.3, 0.4];
        assert_eq!(dequantize(&sig, 48_000, 16, 0.0), sig);
    }
}