01_vdsp_fft/
01_vdsp_fft.rs

1use apple_accelerate::{
2    add_f32, blackman_window, dot_f32, fft_direction, fft_radix, hamming_window, FftSetup,
3};
4
5fn main() {
6    let added = add_f32(&[1.0, 2.0, 3.0], &[4.0, 5.0, 6.0]).expect("vector add failed");
7    assert_eq!(added, vec![5.0, 7.0, 9.0]);
8
9    let dot = dot_f32(&[1.0, 2.0, 3.0], &[4.0, 5.0, 6.0]).expect("dot product failed");
10    assert!((dot - 32.0).abs() < 1.0e-6);
11
12    let hamm = hamming_window(8, 0);
13    let blk = blackman_window(8, 0);
14    assert_eq!(hamm.len(), 8);
15    assert_eq!(blk.len(), 8);
16
17    let setup = FftSetup::new(2, fft_radix::RADIX2).expect("failed to create FFT setup");
18    let mut real = vec![1.0_f32, 0.0, 0.0, 0.0];
19    let mut imag = vec![0.0_f32; 4];
20    setup
21        .fft_zip(&mut real, &mut imag, 2, fft_direction::FORWARD)
22        .expect("fft failed");
23
24    for value in &real {
25        assert!(
26            (*value - 1.0).abs() < 1.0e-5,
27            "unexpected FFT real output: {real:?}"
28        );
29    }
30    for value in &imag {
31        assert!(value.abs() < 1.0e-5, "unexpected FFT imag output: {imag:?}");
32    }
33
34    println!("vdsp smoke passed: added={added:?} dot={dot} fft_real={real:?}");
35}
01_vdsp_fft/01_vdsp_fft.rs

01_vdsp_fft/
01_vdsp_fft.rs
