02_blas_vimage/
02_blas_vimage.rs

1use apple_accelerate::{
2    contrast_stretch_planar8, rotate_argb8888, sgemm_row_major, vimage_flags, ImageBuffer,
3};
4
5fn main() {
6    let mut output = vec![0.0_f32; 4];
7    sgemm_row_major(
8        2,
9        2,
10        3,
11        1.0,
12        &[1.0, 2.0, 3.0, 4.0, 5.0, 6.0],
13        &[7.0, 8.0, 9.0, 10.0, 11.0, 12.0],
14        0.0,
15        &mut output,
16    )
17    .expect("sgemm failed");
18    let expected = [58.0_f32, 64.0, 139.0, 154.0];
19    for (actual, expected_value) in output.iter().zip(expected) {
20        assert!(
21            (actual - expected_value).abs() < 1.0e-5,
22            "unexpected sgemm output: {output:?}"
23        );
24    }
25
26    let source_pixels = vec![
27        255_u8, 10, 20, 30, 255, 40, 50, 60, 255, 70, 80, 90, 255, 100, 110, 120,
28    ];
29    let mut src_pixels = source_pixels.clone();
30    let mut rotated_pixels = vec![0_u8; source_pixels.len()];
31    let src = ImageBuffer::from_argb8888(&mut src_pixels, 2, 2).expect("src buffer");
32    let mut rotated =
33        ImageBuffer::from_argb8888(&mut rotated_pixels, 2, 2).expect("rotated buffer");
34    rotate_argb8888(
35        &src,
36        &mut rotated,
37        0.0,
38        [0, 0, 0, 0],
39        vimage_flags::NO_FLAGS,
40    )
41    .expect("rotate failed");
42    assert_eq!(rotated_pixels, source_pixels);
43
44    let mut planar_src = vec![10_u8, 20, 30, 40];
45    let mut planar_dst = vec![0_u8; 4];
46    let planar_src_buffer = ImageBuffer::from_planar8(&mut planar_src, 2, 2).expect("planar src");
47    let mut planar_dst_buffer =
48        ImageBuffer::from_planar8(&mut planar_dst, 2, 2).expect("planar dst");
49    contrast_stretch_planar8(
50        &planar_src_buffer,
51        &mut planar_dst_buffer,
52        vimage_flags::NO_FLAGS,
53    )
54    .expect("contrast stretch failed");
55    assert_eq!(planar_dst, vec![0, 85, 170, 255]);
56
57    println!("blas+vimage smoke passed: sgemm={output:?} contrast={planar_dst:?}");
58}
02_blas_vimage/02_blas_vimage.rs

02_blas_vimage/
02_blas_vimage.rs
