[rustc.git] / vendor / packed_simd / src / api / ops / scalar_arithmetic.rs

//! Vertical (lane-wise) vector-scalar / scalar-vector arithmetic operations.

macro_rules! impl_ops_scalar_arithmetic {
    ([$elem_ty:ident; $elem_count:expr]: $id:ident | $test_tt:tt) => {
        impl crate::ops::Add<$elem_ty> for $id {
            type Output = Self;
            #[inline]
            fn add(self, other: $elem_ty) -> Self {
                self + $id::splat(other)
            }
        }
        impl crate::ops::Add<$id> for $elem_ty {
            type Output = $id;
            #[inline]
            fn add(self, other: $id) -> $id {
                $id::splat(self) + other
            }
        }

        impl crate::ops::Sub<$elem_ty> for $id {
            type Output = Self;
            #[inline]
            fn sub(self, other: $elem_ty) -> Self {
                self - $id::splat(other)
            }
        }
        impl crate::ops::Sub<$id> for $elem_ty {
            type Output = $id;
            #[inline]
            fn sub(self, other: $id) -> $id {
                $id::splat(self) - other
            }
        }

        impl crate::ops::Mul<$elem_ty> for $id {
            type Output = Self;
            #[inline]
            fn mul(self, other: $elem_ty) -> Self {
                self * $id::splat(other)
            }
        }
        impl crate::ops::Mul<$id> for $elem_ty {
            type Output = $id;
            #[inline]
            fn mul(self, other: $id) -> $id {
                $id::splat(self) * other
            }
        }

        impl crate::ops::Div<$elem_ty> for $id {
            type Output = Self;
            #[inline]
            fn div(self, other: $elem_ty) -> Self {
                self / $id::splat(other)
            }
        }
        impl crate::ops::Div<$id> for $elem_ty {
            type Output = $id;
            #[inline]
            fn div(self, other: $id) -> $id {
                $id::splat(self) / other
            }
        }

        impl crate::ops::Rem<$elem_ty> for $id {
            type Output = Self;
            #[inline]
            fn rem(self, other: $elem_ty) -> Self {
                self % $id::splat(other)
            }
        }
        impl crate::ops::Rem<$id> for $elem_ty {
            type Output = $id;
            #[inline]
            fn rem(self, other: $id) -> $id {
                $id::splat(self) % other
            }
        }

        impl crate::ops::AddAssign<$elem_ty> for $id {
            #[inline]
            fn add_assign(&mut self, other: $elem_ty) {
                *self = *self + other;
            }
        }

        impl crate::ops::SubAssign<$elem_ty> for $id {
            #[inline]
            fn sub_assign(&mut self, other: $elem_ty) {
                *self = *self - other;
            }
        }

        impl crate::ops::MulAssign<$elem_ty> for $id {
            #[inline]
            fn mul_assign(&mut self, other: $elem_ty) {
                *self = *self * other;
            }
        }

        impl crate::ops::DivAssign<$elem_ty> for $id {
            #[inline]
            fn div_assign(&mut self, other: $elem_ty) {
                *self = *self / other;
            }
        }

        impl crate::ops::RemAssign<$elem_ty> for $id {
            #[inline]
            fn rem_assign(&mut self, other: $elem_ty) {
                *self = *self % other;
            }
        }

        test_if!{
            $test_tt:
            paste::item! {
                pub mod [<$id _ops_scalar_arith>] {
                    use super::*;
                    #[cfg_attr(not(target_arch = "wasm32"), test)] #[cfg_attr(target_arch = "wasm32", wasm_bindgen_test)]
                    fn ops_scalar_arithmetic() {
                        let zi = 0 as $elem_ty;
                        let oi = 1 as $elem_ty;
                        let ti = 2 as $elem_ty;
                        let fi = 4 as $elem_ty;
                        let z = $id::splat(zi);
                        let o = $id::splat(oi);
                        let t = $id::splat(ti);
                        let f = $id::splat(fi);

                        // add
                        assert_eq!(zi + z, z);
                        assert_eq!(z + zi, z);
                        assert_eq!(oi + z, o);
                        assert_eq!(o + zi, o);
                        assert_eq!(ti + z, t);
                        assert_eq!(t + zi, t);
                        assert_eq!(ti + t, f);
                        assert_eq!(t + ti, f);
                        // sub
                        assert_eq!(zi - z, z);
                        assert_eq!(z - zi, z);
                        assert_eq!(oi - z, o);
                        assert_eq!(o - zi, o);
                        assert_eq!(ti - z, t);
                        assert_eq!(t - zi, t);
                        assert_eq!(fi - t, t);
                        assert_eq!(f - ti, t);
                        assert_eq!(f - o - o, t);
                        assert_eq!(f - oi - oi, t);
                        // mul
                        assert_eq!(zi * z, z);
                        assert_eq!(z * zi, z);
                        assert_eq!(zi * o, z);
                        assert_eq!(z * oi, z);
                        assert_eq!(zi * t, z);
                        assert_eq!(z * ti, z);
                        assert_eq!(oi * t, t);
                        assert_eq!(o * ti, t);
                        assert_eq!(ti * t, f);
                        assert_eq!(t * ti, f);
                        // div
                        assert_eq!(zi / o, z);
                        assert_eq!(z / oi, z);
                        assert_eq!(ti / o, t);
                        assert_eq!(t / oi, t);
                        assert_eq!(fi / o, f);
                        assert_eq!(f / oi, f);
                        assert_eq!(ti / t, o);
                        assert_eq!(t / ti, o);
                        assert_eq!(fi / t, t);
                        assert_eq!(f / ti, t);
                        // rem
                        assert_eq!(oi % o, z);
                        assert_eq!(o % oi, z);
                        assert_eq!(fi % t, z);
                        assert_eq!(f % ti, z);

                        {
                            let mut v = z;
                            assert_eq!(v, z);
                            v += oi; // add_assign
                            assert_eq!(v, o);
                            v -= oi; // sub_assign
                            assert_eq!(v, z);
                            v = t;
                            v *= oi; // mul_assign
                            assert_eq!(v, t);
                            v *= ti;
                            assert_eq!(v, f);
                            v /= oi; // div_assign
                            assert_eq!(v, f);
                            v /= ti;
                            assert_eq!(v, t);
                            v %= ti; // rem_assign
                            assert_eq!(v, z);
                        }
                    }
                }
            }
        }
    };
}
Commit	Line	Data
f20569fa XL	1	//! Vertical (lane-wise) vector-scalar / scalar-vector arithmetic operations.
	2
	3	macro_rules! impl_ops_scalar_arithmetic {
	4	([$elem_ty:ident; $elem_count:expr]: $id:ident \| $test_tt:tt) => {
	5	impl crate::ops::Add<$elem_ty> for $id {
	6	type Output = Self;
	7	#[inline]
	8	fn add(self, other: $elem_ty) -> Self {
	9	self + $id::splat(other)
	10	}
	11	}
	12	impl crate::ops::Add<$id> for $elem_ty {
	13	type Output = $id;
	14	#[inline]
	15	fn add(self, other: $id) -> $id {
	16	$id::splat(self) + other
	17	}
	18	}
	19
	20	impl crate::ops::Sub<$elem_ty> for $id {
	21	type Output = Self;
	22	#[inline]
	23	fn sub(self, other: $elem_ty) -> Self {
	24	self - $id::splat(other)
	25	}
	26	}
	27	impl crate::ops::Sub<$id> for $elem_ty {
	28	type Output = $id;
	29	#[inline]
	30	fn sub(self, other: $id) -> $id {
	31	$id::splat(self) - other
	32	}
	33	}
	34
	35	impl crate::ops::Mul<$elem_ty> for $id {
	36	type Output = Self;
	37	#[inline]
	38	fn mul(self, other: $elem_ty) -> Self {
	39	self * $id::splat(other)
	40	}
	41	}
	42	impl crate::ops::Mul<$id> for $elem_ty {
	43	type Output = $id;
	44	#[inline]
	45	fn mul(self, other: $id) -> $id {
	46	$id::splat(self) * other
	47	}
	48	}
	49
	50	impl crate::ops::Div<$elem_ty> for $id {
	51	type Output = Self;
	52	#[inline]
	53	fn div(self, other: $elem_ty) -> Self {
	54	self / $id::splat(other)
	55	}
	56	}
	57	impl crate::ops::Div<$id> for $elem_ty {
	58	type Output = $id;
	59	#[inline]
	60	fn div(self, other: $id) -> $id {
	61	$id::splat(self) / other
	62	}
	63	}
	64
65	impl crate::ops::Rem<$elem_ty> for $id {
66	type Output = Self;
67	#[inline]
68	fn rem(self, other: $elem_ty) -> Self {
69	self % $id::splat(other)
70	}
71	}
72	impl crate::ops::Rem<$id> for $elem_ty {
73	type Output = $id;
74	#[inline]
75	fn rem(self, other: $id) -> $id {
76	$id::splat(self) % other
77	}
78	}
79
80	impl crate::ops::AddAssign<$elem_ty> for $id {
81	#[inline]
82	fn add_assign(&mut self, other: $elem_ty) {
83	self = self + other;
84	}
85	}
86
87	impl crate::ops::SubAssign<$elem_ty> for $id {
88	#[inline]
89	fn sub_assign(&mut self, other: $elem_ty) {
90	self = self - other;
91	}
92	}
93
94	impl crate::ops::MulAssign<$elem_ty> for $id {
95	#[inline]
96	fn mul_assign(&mut self, other: $elem_ty) {
97	self = self * other;
98	}
99	}
100
101	impl crate::ops::DivAssign<$elem_ty> for $id {
102	#[inline]
103	fn div_assign(&mut self, other: $elem_ty) {
104	self = self / other;
105	}
106	}
107
108	impl crate::ops::RemAssign<$elem_ty> for $id {
109	#[inline]
110	fn rem_assign(&mut self, other: $elem_ty) {
111	self = self % other;
112	}
113	}
114
115	test_if!{
116	$test_tt:
117	paste::item! {
118	pub mod [<$id _ops_scalar_arith>] {
119	use super::*;
120	#[cfg_attr(not(target_arch = "wasm32"), test)] #[cfg_attr(target_arch = "wasm32", wasm_bindgen_test)]
121	fn ops_scalar_arithmetic() {
122	let zi = 0 as $elem_ty;
123	let oi = 1 as $elem_ty;
124	let ti = 2 as $elem_ty;
125	let fi = 4 as $elem_ty;
126	let z = $id::splat(zi);
127	let o = $id::splat(oi);
128	let t = $id::splat(ti);
129	let f = $id::splat(fi);
130
131	// add
132	assert_eq!(zi + z, z);
133	assert_eq!(z + zi, z);
134	assert_eq!(oi + z, o);
135	assert_eq!(o + zi, o);
136	assert_eq!(ti + z, t);
137	assert_eq!(t + zi, t);
138	assert_eq!(ti + t, f);
139	assert_eq!(t + ti, f);
140	// sub
141	assert_eq!(zi - z, z);
142	assert_eq!(z - zi, z);
143	assert_eq!(oi - z, o);
144	assert_eq!(o - zi, o);
145	assert_eq!(ti - z, t);
146	assert_eq!(t - zi, t);
147	assert_eq!(fi - t, t);
148	assert_eq!(f - ti, t);
149	assert_eq!(f - o - o, t);
150	assert_eq!(f - oi - oi, t);
151	// mul
152	assert_eq!(zi * z, z);
153	assert_eq!(z * zi, z);
154	assert_eq!(zi * o, z);
155	assert_eq!(z * oi, z);
156	assert_eq!(zi * t, z);
157	assert_eq!(z * ti, z);
158	assert_eq!(oi * t, t);
159	assert_eq!(o * ti, t);
160	assert_eq!(ti * t, f);
161	assert_eq!(t * ti, f);
162	// div
163	assert_eq!(zi / o, z);
164	assert_eq!(z / oi, z);
165	assert_eq!(ti / o, t);
166	assert_eq!(t / oi, t);
167	assert_eq!(fi / o, f);
168	assert_eq!(f / oi, f);
169	assert_eq!(ti / t, o);
170	assert_eq!(t / ti, o);
171	assert_eq!(fi / t, t);
172	assert_eq!(f / ti, t);
173	// rem
174	assert_eq!(oi % o, z);
175	assert_eq!(o % oi, z);
176	assert_eq!(fi % t, z);
177	assert_eq!(f % ti, z);
178
179	{
180	let mut v = z;
181	assert_eq!(v, z);
182	v += oi; // add_assign
183	assert_eq!(v, o);
184	v -= oi; // sub_assign
185	assert_eq!(v, z);
186	v = t;
187	v *= oi; // mul_assign
188	assert_eq!(v, t);
189	v *= ti;
190	assert_eq!(v, f);
191	v /= oi; // div_assign
192	assert_eq!(v, f);
193	v /= ti;
194	assert_eq!(v, t);
195	v %= ti; // rem_assign
196	assert_eq!(v, z);
197	}
198	}
199	}
200	}
201	}
202	};
203	}