audio-lab 0.1.0

Toolkit for audio research and testing
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

use crate::signal::{FreqSignal, SignalError, TimeSignal};

#[derive(Debug, thiserror::Error)]
pub enum NpyOrSignalError {
    #[error(transparent)]
    Npy(#[from] ndarray_npy::ReadNpyError),
    #[error(transparent)]
    Signal(#[from] SignalError),
}

pub fn read_npy_time(
    path: impl AsRef<std::path::Path>,
    sample_rate: f64,
) -> Result<TimeSignal, NpyOrSignalError> {
    let y_data = ndarray_npy::read_npy(path).map_err(NpyOrSignalError::Npy)?;
    TimeSignal::new(y_data, sample_rate).map_err(NpyOrSignalError::Signal)
}

pub fn write_npy_time(
    signal: &TimeSignal,
    path: impl AsRef<std::path::Path>,
) -> Result<(), ndarray_npy::WriteNpyError> {
    ndarray_npy::write_npy(path, &signal.time_data().to_owned())
}

pub fn read_npy_freq(
    path: impl AsRef<std::path::Path>,
    sample_rate: f64,
    num_time_steps: Option<usize>,
) -> Result<FreqSignal, NpyOrSignalError> {
    let y_data = ndarray_npy::read_npy(path).map_err(NpyOrSignalError::Npy)?;
    FreqSignal::new(y_data, sample_rate, num_time_steps).map_err(NpyOrSignalError::Signal)
}

pub fn write_npy_freq(
    signal: &FreqSignal,
    path: impl AsRef<std::path::Path>,
) -> Result<(), ndarray_npy::WriteNpyError> {
    ndarray_npy::write_npy(path, &signal.freq_data().to_owned())
}

#[cfg(test)]
mod tests {
    use std::time::{SystemTime, UNIX_EPOCH};

    use ndarray::arr2;
    use num::complex::Complex64;

    use super::*;

    #[test]
    fn writes_and_reads_time_signal_npy() {
        let signal = TimeSignal::new(arr2(&[[1.0, 2.0, 3.0], [4.0, 5.0, 6.0]]), 44_100.0).unwrap();
        let path = std::env::temp_dir().join(format!(
            "audio-lab-{}-{}.npy",
            std::process::id(),
            SystemTime::now()
                .duration_since(UNIX_EPOCH)
                .unwrap()
                .as_nanos()
        ));

        write_npy_time(&signal, &path).unwrap();
        let loaded = read_npy_time(&path, 44_100.0).unwrap();
        std::fs::remove_file(&path).unwrap();

        assert_eq!(loaded, signal);
    }

    #[test]
    fn writes_and_reads_freq_signal_npy() {
        let signal = FreqSignal::new(
            arr2(&[
                [Complex64::new(1.0, 2.0), Complex64::new(3.0, 4.0)],
                [Complex64::new(5.0, 6.0), Complex64::new(7.0, 8.0)],
            ]),
            44_100.0,
            Some(3),
        )
        .unwrap();
        let path = std::env::temp_dir().join(format!(
            "audio-lab-freq-{}-{}.npy",
            std::process::id(),
            SystemTime::now()
                .duration_since(UNIX_EPOCH)
                .unwrap()
                .as_nanos()
        ));

        write_npy_freq(&signal, &path).unwrap();
        let loaded = read_npy_freq(&path, 44_100.0, Some(3)).unwrap();
        std::fs::remove_file(&path).unwrap();

        assert_eq!(loaded, signal);
    }
}