mech_math/ops/
add.rs

1
2use crate::*;
3use num_traits::*;
4
5#[cfg(feature = "matrix")]
6use mech_core::matrix::Matrix;
7
8// Add ------------------------------------------------------------------------
9
10macro_rules! add_op {
11($lhs:expr, $rhs:expr, $out:expr) => {
12  unsafe { *$out = *$lhs + *$rhs; }
13  };}
14
15macro_rules! add_vec_op {
16  ($lhs:expr, $rhs:expr, $out:expr) => {
17    unsafe { (*$lhs).add_to(&*$rhs,&mut *$out) }
18  };}
19
20macro_rules! add_mat_vec_op {
21  ($lhs:expr, $rhs:expr, $out:expr) => {
22    unsafe {
23      let mut out_deref = &mut (*$out);
24      let lhs_deref = &(*$lhs);
25      let rhs_deref = &(*$rhs);
26      for (mut col, lhs_col) in out_deref.column_iter_mut().zip(lhs_deref.column_iter()) {
27        lhs_col.add_to(&rhs_deref,&mut col);
28      }
29    }
30  };}
31
32macro_rules! add_vec_mat_op {
33  ($lhs:expr, $rhs:expr, $out:expr) => {
34    unsafe {
35      let mut out_deref = &mut (*$out);
36      let lhs_deref = &(*$lhs);
37      let rhs_deref = &(*$rhs);
38      for (mut col, rhs_col) in out_deref.column_iter_mut().zip(rhs_deref.column_iter()) {
39        lhs_deref.add_to(&rhs_col,&mut col);
40      }
41    }
42  };}
43
44macro_rules! add_mat_row_op {
45  ($lhs:expr, $rhs:expr, $out:expr) => {
46    unsafe {
47      let mut out_deref = &mut (*$out);
48      let lhs_deref = &(*$lhs);
49      let rhs_deref = &(*$rhs);
50      for (mut row, lhs_row) in out_deref.row_iter_mut().zip(lhs_deref.row_iter()) {
51        lhs_row.add_to(&rhs_deref,&mut row);
52      }
53    }
54  };}
55
56macro_rules! add_row_mat_op {
57  ($lhs:expr, $rhs:expr, $out:expr) => {
58    unsafe {
59      let mut out_deref = &mut (*$out);
60      let lhs_deref = &(*$lhs);
61      let rhs_deref = &(*$rhs);
62      for (mut row, rhs_row) in out_deref.row_iter_mut().zip(rhs_deref.row_iter()) {
63        lhs_deref.add_to(&rhs_row,&mut row);
64      }
65    }
66  };}  
67
68macro_rules! add_scalar_lhs_op {
69  ($lhs:expr, $rhs:expr, $out:expr) => {
70    unsafe { *$out = (*$lhs).add_scalar(*$rhs); }
71  };}
72
73macro_rules! add_scalar_rhs_op {
74  ($lhs:expr, $rhs:expr, $out:expr) => {
75    unsafe { *$out = (*$rhs).add_scalar(*$lhs); }
76  };}
77
78impl_math_fxns!(Add);
79
80fn impl_add_fxn(lhs_value: Value, rhs_value: Value) -> MResult<Box<dyn MechFunction>> {
81  impl_binop_match_arms!(
82    Add,
83    register_fxn_descriptor_inner,
84    (lhs_value, rhs_value),
85    I8,   i8,   "i8";
86    I16,  i16,  "i16";
87    I32,  i32,  "i32";
88    I64,  i64,  "i64";
89    I128, i128, "i128";
90    U8,   u8,   "u8";
91    U16,  u16,  "u16";
92    U32,  u32,  "u32";
93    U64,  u64,  "u64";
94    U128, u128, "u128";
95    F32,  f32,  "f32";
96    F64,  f64,  "f64";
97    R64, R64, "rational";
98    C64, C64, "complex";
99  )
100}
101
102impl_mech_binop_fxn!(MathAdd,impl_add_fxn,"math/add");
mech_math/ops/add.rs

mech_math/ops/
add.rs
