argmin_math/primitives/
dot.rs

1// Copyright 2018-2024 argmin developers
2//
3// Licensed under the Apache License, Version 2.0 <LICENSE-APACHE or
4// http://apache.org/licenses/LICENSE-2.0> or the MIT license <LICENSE-MIT or
5// http://opensource.org/licenses/MIT>, at your option. This file may not be
6// copied, modified, or distributed except according to those terms.
7
8use crate::ArgminDot;
9use num_complex::Complex;
10
11macro_rules! make_dot_vec {
12    ($t:ty) => {
13        impl ArgminDot<$t, $t> for $t {
14            #[inline]
15            fn dot(&self, other: &$t) -> $t {
16                self * other
17            }
18        }
19    };
20}
21
22make_dot_vec!(f32);
23make_dot_vec!(f64);
24make_dot_vec!(i8);
25make_dot_vec!(i16);
26make_dot_vec!(i32);
27make_dot_vec!(i64);
28make_dot_vec!(u8);
29make_dot_vec!(u16);
30make_dot_vec!(u32);
31make_dot_vec!(u64);
32make_dot_vec!(Complex<f32>);
33make_dot_vec!(Complex<f64>);
34make_dot_vec!(Complex<i8>);
35make_dot_vec!(Complex<i16>);
36make_dot_vec!(Complex<i32>);
37make_dot_vec!(Complex<i64>);
38make_dot_vec!(Complex<u8>);
39make_dot_vec!(Complex<u16>);
40make_dot_vec!(Complex<u32>);
41make_dot_vec!(Complex<u64>);
42
43#[cfg(test)]
44mod tests {
45    use super::*;
46    use approx::assert_relative_eq;
47    use paste::item;
48
49    macro_rules! make_test {
50        ($t:ty) => {
51            item! {
52                #[test]
53                fn [<test_vec_vec_ $t>]() {
54                    let a = 21 as $t;
55                    let b = 2 as $t;
56                    let res = a.dot(&b);
57                    assert_relative_eq!(42 as f64, res as f64, epsilon = f64::EPSILON);
58                }
59            }
60        };
61    }
62
63    make_test!(i8);
64    make_test!(u8);
65    make_test!(i16);
66    make_test!(u16);
67    make_test!(i32);
68    make_test!(u32);
69    make_test!(i64);
70    make_test!(u64);
71    make_test!(f32);
72    make_test!(f64);
73}
argmin_math/primitives/dot.rs

argmin_math/primitives/
dot.rs
