basic_dsp_vector 0.10.2

Digital signal processing based on real or complex vectors in time or frequency domain.
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

#[cfg(feature = "use_gpu")]
mod ocl;

#[cfg(feature = "use_gpu")]
pub use self::ocl::*;

#[cfg(not(feature = "use_gpu"))]
mod fallback;

#[cfg(not(feature = "use_gpu"))]
pub use self::fallback::*;

use crate::numbers::*;
use rustfft::FftDirection;
use std::ops::Range;

/// Trait which adds GPU support to types like `f32` and `f64`.
pub trait GpuSupport<T: RealNumber> {
    /// Indicates whether or not GPU support is available for this type. All other
    /// methods will panic if there is no GPU support so better check this one first.
    fn has_gpu_support() -> bool;

    /// Convolve a vector on the GPU.
    fn gpu_convolve_vector(
        is_complex: bool,
        source: &[T],
        target: &mut [T],
        imp_resp: &[T],
    ) -> Option<Range<usize>>;

    /// Indicates whether or not the parameters are supported by the FFT implementation.
    fn is_supported_fft_len(is_complex: bool, len: usize) -> bool;

    /// FFT on the GPU.
    fn fft(is_complex: bool, signal: &mut [T], direction: FftDirection);

    /// Applys a frequence response to a time domain signal.
    fn overlap_discard(
        x_time: &mut [T],
        tmp: &mut [T],
        x_freq: &mut [T],
        h_freq: &[T],
        imp_len: usize,
        step_size: usize,
    ) -> usize;
}