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
use number_traits::Num; #[inline] pub fn add<'a, 'b, T: Copy + Num>(out: &'a mut [T; 4], a: &'b [T; 4], b: &'b [T; 4]) -> &'a mut [T; 4] { out[0] = a[0] + b[0]; out[1] = a[1] + b[1]; out[2] = a[2] + b[2]; out[3] = a[3] + b[3]; out } #[test] fn test_add() { let mut v = [0, 0, 0, 0]; add(&mut v, &[1, 1, 1, 1], &[1, 1, 1, 1]); assert!(v[0] == 2); assert!(v[1] == 2); assert!(v[2] == 2); assert!(v[3] == 2); } #[inline] pub fn sadd<'a, 'b, T: Copy + Num>(out: &'a mut [T; 4], a: &'b [T; 4], s: T) -> &'a mut [T; 4] { out[0] = a[0] + s; out[1] = a[1] + s; out[2] = a[2] + s; out[3] = a[3] + s; out } #[test] fn test_sadd() { let mut v = [0, 0, 0, 0]; sadd(&mut v, &[0, 0, 0, 0], 1); assert!(v[0] == 1); assert!(v[1] == 1); assert!(v[2] == 1); assert!(v[3] == 1); }