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
 61
 62
 63
 64
 65
 66
 67
 68
 69
 70
 71
 72
 73
 74
 75
 76
 77
 78
 79
 80
 81
 82
 83
 84
 85
 86
 87
 88
 89
 90
 91
 92
 93
 94
 95
 96
 97
 98
 99
100
101
102
103
104
105
106
107
108
109
110
111
112
113
114

//! Vertical (lane-wise) vector comparisons returning vector masks.

macro_rules! impl_cmp_vertical {
    (
        [$elem_ty:ident; $elem_count:expr]:
        $id:ident,
        $mask_ty:ident,
        $is_mask:expr,($true:expr, $false:expr) | $test_tt:tt
    ) => {
        impl $id {
            /// Lane-wise equality comparison.
            #[inline]
            pub fn eq(self, other: Self) -> $mask_ty {
                use crate::llvm::simd_eq;
                Simd(unsafe { simd_eq(self.0, other.0) })
            }

            /// Lane-wise inequality comparison.
            #[inline]
            pub fn ne(self, other: Self) -> $mask_ty {
                use crate::llvm::simd_ne;
                Simd(unsafe { simd_ne(self.0, other.0) })
            }

            /// Lane-wise less-than comparison.
            #[inline]
            pub fn lt(self, other: Self) -> $mask_ty {
                use crate::llvm::{simd_gt, simd_lt};
                if $is_mask {
                    Simd(unsafe { simd_gt(self.0, other.0) })
                } else {
                    Simd(unsafe { simd_lt(self.0, other.0) })
                }
            }

            /// Lane-wise less-than-or-equals comparison.
            #[inline]
            pub fn le(self, other: Self) -> $mask_ty {
                use crate::llvm::{simd_ge, simd_le};
                if $is_mask {
                    Simd(unsafe { simd_ge(self.0, other.0) })
                } else {
                    Simd(unsafe { simd_le(self.0, other.0) })
                }
            }

            /// Lane-wise greater-than comparison.
            #[inline]
            pub fn gt(self, other: Self) -> $mask_ty {
                use crate::llvm::{simd_gt, simd_lt};
                if $is_mask {
                    Simd(unsafe { simd_lt(self.0, other.0) })
                } else {
                    Simd(unsafe { simd_gt(self.0, other.0) })
                }
            }

            /// Lane-wise greater-than-or-equals comparison.
            #[inline]
            pub fn ge(self, other: Self) -> $mask_ty {
                use crate::llvm::{simd_ge, simd_le};
                if $is_mask {
                    Simd(unsafe { simd_le(self.0, other.0) })
                } else {
                    Simd(unsafe { simd_ge(self.0, other.0) })
                }
            }
        }
        test_if!{
            $test_tt:
            paste::item! {
                pub mod [<$id _cmp_vertical>] {
                    use super::*;
                    #[cfg_attr(not(target_arch = "wasm32"), test)] #[cfg_attr(target_arch = "wasm32", wasm_bindgen_test)]
                    fn cmp() {
                        let a = $id::splat($false);
                        let b = $id::splat($true);

                        let r = a.lt(b);
                        let e = $mask_ty::splat(true);
                        assert!(r == e);
                        let r = a.le(b);
                        assert!(r == e);

                        let e = $mask_ty::splat(false);
                        let r = a.gt(b);
                        assert!(r == e);
                        let r = a.ge(b);
                        assert!(r == e);
                        let r = a.eq(b);
                        assert!(r == e);

                        let mut a = a;
                        let mut b = b;
                        let mut e = e;
                        for i in 0..$id::lanes() {
                            if i % 2 == 0 {
                                a = a.replace(i, $false);
                                b = b.replace(i, $true);
                                e = e.replace(i, true);
                            } else {
                                a = a.replace(i, $true);
                                b = b.replace(i, $false);
                                e = e.replace(i, false);
                            }
                        }
                        let r = a.lt(b);
                        assert!(r == e);
                    }
                }
            }
        }
    };
}