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);
        }
}