matrix_simba.rs - source
Description: Source of the Rust file `/Users/erlendbasso/.cargo/registry/src/github.com-1ecc6299db9ec823/nalgebra-0.32.1/src/base/matrix_simba.rs`.
Keywords: rust, rustlang, rust-lang

    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
    47
    48
    49
    50
    51
    52
    53
    54
    55
    56
    57
    58
    59
    60

    use simba::simd::SimdValue;

    use crate::base::allocator::Allocator;
    use crate::base::dimension::Dim;
    use crate::base::{DefaultAllocator, OMatrix, Scalar};

    /*
     *
     * Simd structures.
     *
     */
    impl<T, R, C> SimdValue for OMatrix<T, R, C>
    where
        T: Scalar + SimdValue,
        R: Dim,
        C: Dim,
        T::Element: Scalar,
        DefaultAllocator: Allocator<T, R, C> + Allocator<T::Element, R, C>,
    {
        type Element = OMatrix<T::Element, R, C>;
        type SimdBool = T::SimdBool;

        #[inline]
        fn lanes() -> usize {
            T::lanes()
        }

        #[inline]
        fn splat(val: Self::Element) -> Self {
            val.map(T::splat)
        }

        #[inline]
        fn extract(&self, i: usize) -> Self::Element {
            self.map(|e| e.extract(i))
        }

        #[inline]
        unsafe fn extract_unchecked(&self, i: usize) -> Self::Element {
            self.map(|e| e.extract_unchecked(i))
        }

        #[inline]
        fn replace(&mut self, i: usize, val: Self::Element) {
            self.zip_apply(&val, |a, b| {
                a.replace(i, b);
            })
        }

        #[inline]
        unsafe fn replace_unchecked(&mut self, i: usize, val: Self::Element) {
            self.zip_apply(&val, |a, b| {
                a.replace_unchecked(i, b);
            })
        }

        fn select(self, cond: Self::SimdBool, other: Self) -> Self {
            self.zip_map(&other, |a, b| a.select(cond, b))
        }
    }