lasprs 0.14.0

Library for Acoustic Signal Processing (Rust edition, with optional Python bindings via pyo3)
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

//!
//! Provides code to estimate (cross)[PowerSpectra], averaged power spectra
//! [AvPowerSpectra] using
//! Welch' method, and windows for time-windowing the data with non-rectangular
//! windows (also known as 'tapers').
//!
mod aps;
mod apsmode;
mod apssettings;
mod cpssettings;
mod error;
mod fft;
mod freqsmooth;
mod ps;
mod spectrogram;

use crate::*;

pub use crate::common::Window;
pub use aps::AvPowerSpectra;
pub use apsmode::ApsMode;
pub use apssettings::{ApsSettings, ApsSettingsBuilder};
pub use cpssettings::{CPSSettings, CPSSettingsBuilder};
pub use error::*;
#[cfg(feature = "python-bindings")]
pub(crate) use freqsmooth::smoothSpectralData_py;
pub use freqsmooth::{SmoothingType, SmoothingWidth, smoothSpectralData};
pub use ps::{CPSResult, PowerSpectra};
pub use spectrogram::{SpectroGramEngine, SpectrogramResult};

/// Returns the frequency vector for a given sample rate and FFT size.
///
/// # Arguments
///
/// * `samplerate` - The sample rate of the signal.
/// * `nfft` - The FFT size.
///
/// # Panics
///
/// Panics if `nfft` is 0
///
/// # Returns
///
/// A vector of frequencies, starting at 0 Hz, ending at the Nyquist frequency.
pub fn getFreq(samplerate: StrictlyPositive, nfft: usize) -> Array1<Flt> {
    if nfft == 0 {
        panic!("nfft must be > 0");
    }
    let df = *samplerate / nfft as Flt;
    let K = nfft / 2 + 1;
    Array1::linspace(0., (K as Flt - 1.) * df, K)
}

#[cfg(feature = "python-bindings")]
#[gen_stub_pyfunction]
#[pyfunction(name = "getFreq")]
/// Wrapper function for `getFreq` that returns a Python array. Applies
/// pre-check on inputs to prevent GIGO. Raises an error if `nfft` is 0, of if
/// Fs <= 0.
pub(crate) fn getFreq_py(
    py: Python<'_>,
    fs: StrictlyPositive,
    nfft: usize,
) -> PyResult<Bound<PyArray1<Flt>, '_>> {
    if nfft == 0 {
        Err(PyValueError::new_err("nfft must be > 0"))
    } else {
        let freq = getFreq(fs, nfft);
        Ok(freq.into_pyarray(py))
    }
}