rusty-opus 0.9.1

pure Rust Opus codec - performance fork of opus-rs (AVX2 SILK/CELT kernels, frame-parallel encode)
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
76
77
78
79
80
81
82
83
84
85
86
87
88
89
90
91
92
93
94
95
96
97
98
99
100
101
102
103
104
105
106
107
108

#[test]
fn test_celt_mdct_passthrough() {
    use rusty_opus::modes::default_mode;

    let mode = default_mode();
    let frame_size = 960;
    let overlap = mode.overlap;
    // For long blocks: shift = mode.max_lm - lm. At full frame_size, lm = max_lm, so shift = 0.
    let shift = 0usize;
    let b = 1usize;
    // MDCT forward needs n + overlap = 1920 + 120 = 2040 samples
    let mdct_input_size = mode.mdct.n + overlap;
    let syn_mem_size = mdct_input_size + frame_size; // Enough for one frame of history

    // Create a continuous sine wave input (5 frames worth to have enough history)
    let freq_hz = 440.0 / 48000.0 * 2.0 * std::f32::consts::PI;
    let total_samples = 5 * frame_size;
    let mut all_in = vec![0.0f32; total_samples];
    for i in 0..total_samples {
        all_in[i] = (freq_hz * i as f32).sin();
    }

    // Simulate encoder syn_mem + decoder decode_mem
    let mut syn_mem = vec![0.0f32; syn_mem_size];
    let decode_buffer_size = mode.mdct.n + overlap;
    let mut decode_mem = vec![0.0f32; decode_buffer_size];

    let mut all_out = Vec::new();

    // Encode and decode frames
    for frame_idx in 0..3 {
        let frame_start = (frame_idx + 1) * frame_size; // Start from frame 1 for history

        // Shift encoder history left by frame_size, then insert new samples at end
        for i in 0..syn_mem_size - frame_size {
            syn_mem[i] = syn_mem[i + frame_size];
        }
        for i in 0..frame_size {
            syn_mem[syn_mem_size - frame_size + i] = all_in[frame_start + i];
        }

        // MDCT forward - use the last mdct_input_size samples
        let mut freq_buf = vec![0.0f32; frame_size];
        mode.mdct.forward(
            &syn_mem[syn_mem_size - mdct_input_size..],
            &mut freq_buf,
            mode.window,
            overlap,
            shift,
            b,
        );

        // MDCT backward (overlap-add)
        mode.mdct
            .backward(&freq_buf, &mut decode_mem, mode.window, overlap, shift, b);

        // Extract frame output (after overlap region)
        let mut frame_out = vec![0.0f32; frame_size];
        frame_out.copy_from_slice(&decode_mem[overlap..overlap + frame_size]);
        all_out.extend_from_slice(&frame_out);
    }

    // Compare frame 2 output against input, searching for the best delay alignment.
    let compare_frame = 2;
    let mut best_snr = -100.0f64;
    let mut best_delay = 0usize;

    for delay in 0..2 * frame_size {
        let mut signal_power = 0.0f64;
        let mut error_power = 0.0f64;
        let mut count = 0;
        for i in 0..frame_size {
            let out_idx = compare_frame * frame_size + i;
            if delay > (compare_frame + 1) * frame_size + i {
                continue;
            }
            let in_idx = (compare_frame + 1) * frame_size + i - delay;
            if in_idx < total_samples && out_idx < all_out.len() {
                let sig = all_in[in_idx] as f64;
                let out = all_out[out_idx] as f64;
                signal_power += sig * sig;
                error_power += (out - sig) * (out - sig);
                count += 1;
            }
        }
        if count > frame_size / 2 && signal_power > 1e-10 {
            let snr = 10.0 * (signal_power / error_power.max(1e-20)).log10();
            if snr > best_snr {
                best_snr = snr;
                best_delay = delay;
            }
        }
    }

    eprintln!(
        "CELT MDCT passthrough: best SNR = {:.2} dB at delay = {}",
        best_snr, best_delay
    );

    // TODO: Current MDCT implementation has quality issues
    // Target: >60 dB, Current: varies
    assert!(
        best_snr > 0.0,
        "MDCT passthrough SNR too low: {:.2} dB at delay {}",
        best_snr,
        best_delay
    );
}