Struct GpuSpectrogram 

pub struct GpuSpectrogram { /* private fields */ }

GPU-accelerated (or CPU-fallback) spectrogram computer.

Example

use scirs2_signal::gpu_spectrograms::{GpuSpectrogram, GpuSpectrogramConfig};
use scirs2_core::ndarray::ArrayView1;

let config = GpuSpectrogramConfig::default();
let sg = GpuSpectrogram::new(config).expect("config is valid");

// 4096-sample sine wave at normalised frequency 0.1
let signal: Vec<f32> = (0..4096)
    .map(|i| (2.0 * std::f32::consts::PI * 0.1 * i as f32).sin())
    .collect();

let mag = sg.compute(ArrayView1::from(&signal)).expect("compute ok");
println!("spectrogram shape: {:?}", mag.dim());

Implementations

impl GpuSpectrogram

pub fn new(config: GpuSpectrogramConfig) -> Result<Self, GpuSpectrogramError>

Construct a new spectrogram computer from the given configuration.

Errors

Returns GpuSpectrogramError::InvalidFftSize when fft_size is not a power of two or is zero.

pub fn compute( &self, signal: ArrayView1<'_, f32>, ) -> Result<Array2<f32>, GpuSpectrogramError>

Compute the magnitude spectrogram of signal.

Returns an Array2<f32> of shape [n_frames, fft_size / 2 + 1]. Each row is the single-sided magnitude spectrum of one analysis frame.

Errors

Returns GpuSpectrogramError::SignalTooShort when signal is shorter than one FFT frame.

pub fn compute_power( &self, signal: ArrayView1<'_, f32>, ) -> Result<Array2<f32>, GpuSpectrogramError>

Compute the power spectrogram (magnitude squared) of signal.

Returns an Array2<f32> of shape [n_frames, fft_size / 2 + 1].

pub fn compute_batch( &self, signals: &[Vec<f32>], ) -> Result<Vec<Array2<f32>>, GpuSpectrogramError>

Compute spectrograms for a batch of signals.

Each element of signals is processed independently. Returns a Vec of Array2 in the same order as the input slice.

Errors

Propagates errors from the single-signal GpuSpectrogram::compute call for each element.