zune-image 0.4.1

/*
 * Copyright (c) 2023.
 *
 * This software is free software;
 *
 * You can redistribute it or modify it under terms of the MIT, Apache License or Zlib license
 */

use crate::deinterleave::scalar;

pub fn de_interleave_three_channels_u8(source: &[u8], c1: &mut [u8], c2: &mut [u8], c3: &mut [u8]) {
    #[cfg(any(target_arch = "x86", target_arch = "x86_64"))]
    {
        #[cfg(feature = "simd")]
        {
            use crate::deinterleave::avx2::de_interleave_three_channels_avx2;

            if is_x86_feature_detected!("avx2") {
                unsafe {
                    return de_interleave_three_channels_avx2(source, c1, c2, c3);
                }
            }
        }
        #[cfg(feature = "simd")]
        {
            use crate::deinterleave::sse41::de_interleave_three_channels_sse3_u8;

            if is_x86_feature_detected!("sse3") {
                unsafe {
                    return de_interleave_three_channels_sse3_u8(source, c1, c2, c3);
                }
            }
        }
        #[cfg(feature = "simd")]
        {
            use crate::deinterleave::sse2::de_interleave_three_channels_sse2;

            if is_x86_feature_detected!("sse2") {
                unsafe {
                    return de_interleave_three_channels_sse2(source, c1, c2, c3);
                }
            }
        }
    }
    crate::deinterleave::scalar::de_interleave_three_channels_scalar(source, c1, c2, c3);
}

pub fn de_interleave_three_channels_u16(
    source: &[u16], c1: &mut [u16], c2: &mut [u16], c3: &mut [u16]
) {
    #[cfg(any(target_arch = "x86", target_arch = "x86_64"))]
    {
        #[cfg(feature = "simd")]
        {
            use crate::deinterleave::avx2::de_interleave_three_channels_avx2;

            if is_x86_feature_detected!("avx2") {
                unsafe {
                    return de_interleave_three_channels_avx2(source, c1, c2, c3);
                }
            }
        }
    }
    scalar::de_interleave_three_channels_scalar(source, c1, c2, c3);
}

pub fn de_interleave_four_channels_u16(
    source: &[u16], c1: &mut [u16], c2: &mut [u16], c3: &mut [u16], c4: &mut [u16]
) {
    scalar::de_interleave_four_channels_scalar(source, c1, c2, c3, c4);
}

pub fn deinterleave_four_channels_u8(
    source: &[u8], c1: &mut [u8], c2: &mut [u8], c3: &mut [u8], c4: &mut [u8]
) {
    #[cfg(any(target_arch = "x86", target_arch = "x86_64"))]
    {
        #[cfg(feature = "simd")]
        {
            use crate::deinterleave::avx2::de_interleave_four_channels_avx2;
            if is_x86_feature_detected!("sse4.1") {
                unsafe {
                    return de_interleave_four_channels_avx2(source, c1, c2, c3, c4);
                }
            }
        }
        #[cfg(feature = "simd")]
        {
            use crate::deinterleave::sse41::de_interleave_four_channels_sse41;
            if is_x86_feature_detected!("sse4.1") {
                unsafe {
                    return de_interleave_four_channels_sse41(source, c1, c2, c3, c4);
                }
            }
        }
    }

    scalar::de_interleave_four_channels_scalar(source, c1, c2, c3, c4);
}

/// De interleave interleaved images in float to separate color components
/// and store them in their respective arrays
///
/// # Arguments
///
/// * `source`: Source containing interleaved RGB/three component data
/// * `c1`: Output for first component
/// * `c2`: Output for second component
/// * `c3`: Output for third component
///
pub fn de_interleave_three_channels_f32(
    source: &[f32], c1: &mut [f32], c2: &mut [f32], c3: &mut [f32]
) {
    #[cfg(any(target_arch = "x86", target_arch = "x86_64"))]
    {
        #[cfg(feature = "avx2")]
        {
            use crate::deinterleave::avx2::de_interleave_three_channels_avx2;

            if is_x86_feature_detected!("avx2") {
                unsafe {
                    return de_interleave_three_channels_avx2(source, c1, c2, c3);
                }
            }
        }
    }
    scalar::de_interleave_three_channels_scalar(source, c1, c2, c3);
}

pub fn de_interleave_four_channels_f32(
    source: &[f32], c1: &mut [f32], c2: &mut [f32], c3: &mut [f32], c4: &mut [f32]
) {
    #[cfg(any(target_arch = "x86", target_arch = "x86_64"))]
    {
        #[cfg(feature = "avx2")]
        {
            use crate::deinterleave::avx2::de_interleave_four_channels_avx2;

            if is_x86_feature_detected!("avx2") {
                unsafe {
                    return de_interleave_four_channels_avx2(source, c1, c2, c3, c4);
                }
            }
        }
    }
    scalar::de_interleave_four_channels_scalar(source, c1, c2, c3, c4);
}

#[cfg(feature = "benchmarks")]
#[cfg(test)]
mod benchmarks {
    extern crate test;

    #[cfg(feature = "sse2")]
    #[bench]
    fn de_interleave_3_channels_sse2_bench(b: &mut test::Bencher) {
        use crate::deinterleave::sse2::de_interleave_three_channels_sse2;
        let width = 800;
        let height = 800;
        let dimensions = width * height;

        let mut c1 = vec![0; dimensions];
        let mut c2 = vec![0; dimensions];
        let mut c3 = vec![0; dimensions];

        let c4 = vec![255; dimensions * 3];
        b.iter(|| {
            unsafe {
                de_interleave_three_channels_sse2(&c4, &mut c1, &mut c2, &mut c3);
            };
        });
    }

    #[bench]
    fn de_interleave_3_channels_scalar_bench(b: &mut test::Bencher) {
        use crate::deinterleave::scalar::de_interleave_three_channels_scalar;
        let width = 800;
        let height = 800;
        let dimensions = width * height;

        let mut c1 = vec![0; dimensions];
        let mut c2 = vec![0; dimensions];
        let mut c3 = vec![0; dimensions];

        let c4 = vec![255; dimensions * 3];
        b.iter(|| {
            de_interleave_three_channels_scalar(&c4, &mut c1, &mut c2, &mut c3);
        });
    }

    #[cfg(feature = "sse41")]
    #[bench]
    fn de_interleave_3_channels_sse41_bench(b: &mut test::Bencher) {
        use crate::deinterleave::sse41::de_interleave_three_channels_sse3_u8;
        let width = 800;
        let height = 800;
        let dimensions = width * height;

        let mut c1 = vec![0; dimensions];
        let mut c2 = vec![0; dimensions];
        let mut c3 = vec![0; dimensions];

        let c4 = vec![255; dimensions * 3];
        b.iter(|| {
            unsafe {
                de_interleave_three_channels_sse3_u8(&c4, &mut c1, &mut c2, &mut c3);
            };
        });
    }

    #[cfg(feature = "avx2")]
    #[bench]
    fn de_interleave_3_channels_avx2_bench(b: &mut test::Bencher) {
        use crate::deinterleave::avx2::de_interleave_three_channels_avx2;
        let width = 800;
        let height = 800;
        let dimensions = width * height;

        let mut c1 = vec![0; dimensions];
        let mut c2 = vec![0; dimensions];
        let mut c3 = vec![0; dimensions];

        let c4 = vec![255; dimensions * 3];
        b.iter(|| {
            unsafe {
                de_interleave_three_channels_avx2(&c4, &mut c1, &mut c2, &mut c3);
            };
        });
    }

    #[cfg(feature = "sse41")]
    #[bench]
    fn de_interleave_4_channels_sse41_bench(b: &mut test::Bencher) {
        use crate::deinterleave::sse41::de_interleave_four_channels_sse41;
        let width = 800;
        let height = 800;
        let dimensions = width * height;

        let mut c1 = vec![0; dimensions];
        let mut c2 = vec![0; dimensions];
        let mut c3 = vec![0; dimensions];
        let mut c4 = vec![0; dimensions];

        let c5 = vec![255; dimensions * 4];
        b.iter(|| {
            unsafe {
                de_interleave_four_channels_sse41(&c5, &mut c1, &mut c2, &mut c3, &mut c4);
            };
        });
    }

    #[bench]
    fn de_interleave_4_channels_scalar_bench(b: &mut test::Bencher) {
        use crate::deinterleave::scalar::de_interleave_four_channels_scalar;
        let width = 800;
        let height = 800;
        let dimensions = width * height;

        let mut c1 = vec![0; dimensions];
        let mut c2 = vec![0; dimensions];
        let mut c3 = vec![0; dimensions];
        let mut c4 = vec![0; dimensions];

        let c5 = vec![255_u8; dimensions * 4];
        b.iter(|| {
            de_interleave_four_channels_scalar(&c5, &mut c1, &mut c2, &mut c3, &mut c4);
        });
    }

    #[bench]
    fn de_interleave_4_channels_avx2_bench(b: &mut test::Bencher) {
        use crate::deinterleave::avx2::de_interleave_four_channels_avx2;
        let width = 800;
        let height = 800;
        let dimensions = width * height;

        let mut c1 = vec![0; dimensions];
        let mut c2 = vec![0; dimensions];
        let mut c3 = vec![0; dimensions];
        let mut c4 = vec![0; dimensions];

        let c5 = vec![255_u8; dimensions * 4];
        b.iter(|| unsafe {
            de_interleave_four_channels_avx2(&c5, &mut c1, &mut c2, &mut c3, &mut c4);
        });
    }
}