ogg_opus/
lib.rs

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
109
110
mod common;
mod decode;
mod encode;

use thiserror::Error;

pub use decode::decode;
pub use encode::encode;

use std::io::{Read, Seek, SeekFrom};
pub fn is_ogg_opus<T: Read + Seek>(mut d: T) -> bool {
    let mut buff = [0u8; 8];
    if let Ok(_) = d.seek(SeekFrom::Start(28)) {
        if let Ok(d) = d.read(&mut buff) {
            if d == 8 {
                return buff == decode::OPUS_MAGIC_HEADER;
            }
        }
    }
    // If anything fails
    false
}
#[derive(Debug, Error)]
pub enum Error {
    #[error("Input audio was malformed")]
    MalformedAudio,

    #[error("Encoding error")]
    OpusError(#[from] audiopus::Error),

    #[error("Failed to decode ogg")]
    OggReadError(#[from] ogg::OggReadError),

    #[error("Failed to write in OGG")]
    OggWriteError(#[from]std::io::Error),

    #[error("Invalid samples per second")]
    InvalidSps,
}

#[cfg(test)]
mod tests {
    use std::fs::{File};
    use std::io::{Cursor};

        fn read_file_i16(path: &str) -> Vec<i16> {
            let mut f = File::open(path).expect("no file found");
            let (_, b) = wav::read(&mut f).unwrap();
            b.try_into_sixteen().unwrap()
        }

        #[test]
        fn dec_enc_empty() {
            let audio = Vec::new();
            let opus = crate::encode::<16000, 1>(&audio).unwrap();
            let enc_fin_range = crate::encode::get_final_range();
            let (audio2, _) = crate::decode::<_,16000>(Cursor::new(opus)).unwrap();
            let dec_fin_range = crate::decode::get_final_range();
            assert_eq!(audio.len(), audio2.len()); // Should be the same, empty
            assert_eq!(enc_fin_range, dec_fin_range);
        }

        #[test]
        fn dec_enc_recording_big() {
            let audio = read_file_i16("test_assets/big.wav");
            let opus = crate::encode::<16000, 1>(&audio).unwrap();
            let enc_fin_range = crate::encode::get_final_range();
            let (a2,_) = crate::decode::<_,16000>(Cursor::new(opus)).unwrap();
            let dec_fin_range = crate::decode::get_final_range();
            assert_eq!(dec_fin_range, enc_fin_range);
            assert_eq!(audio.len(), a2.len());
        }

        #[test]
        fn dec_enc_recording_small() {
            // This file (when added the skip) decodes to exactly 63 20ms 
            // + a 2.5 ms packet
            let audio = read_file_i16("test_assets/small.wav");
            let opus = crate::encode::<16000, 1>(&audio).unwrap();
            let enc_fin_range = crate::encode::get_final_range();
            let (a2, _) = crate::decode::<_, 16000>(Cursor::new(opus)).unwrap();
            let dec_fin_range = crate::decode::get_final_range();
            assert_eq!(dec_fin_range, enc_fin_range);
            assert_eq!(audio.len(), a2.len());
        }

        #[test]
        // Record, encode, decode , encode and decode again, finally compare the
        // first and second decodes, to make sure nothing is lost (can't compare
        // raw audio as vorbis is lossy)
        fn dec_enc_recording_whole() {
            let audio = read_file_i16("test_assets/small.wav");
            
            let opus = crate::encode::<16000, 1>(&audio).unwrap();
            let enc_fr1 = crate::encode::get_final_range();

            let (audio2, _) = crate::decode::<_, 16000>(Cursor::new(opus)).unwrap();
            let dec_fr1 = crate::decode::get_final_range();

            let opus2 = crate::encode::<16000, 1>(&audio2).unwrap();
            let enc_fr2 = crate::encode::get_final_range();

            let (audio3, _) = crate::decode::<_, 16000>(Cursor::new(opus2)).unwrap();
            let dec_fr2 = crate::decode::get_final_range();

            assert_eq!(audio2.len(), audio3.len());
            assert_eq!(enc_fr1, dec_fr1);
            assert_eq!(enc_fr2, dec_fr2);
        }
}