mech_matrix/
dot.rs

1use crate::*;
2use mech_core::*;
3#[cfg(feature = "matrix")]
4use mech_core::matrix::Matrix;
5
6// MatMul ---------------------------------------------------------------------
7
8macro_rules! mul_op {
9  ($lhs:expr, $rhs:expr, $out:expr) => {
10    unsafe { *$out = *$lhs * *$rhs; }
11  };}
12
13macro_rules! dot_op {
14  ($lhs:expr, $rhs:expr, $out:expr) => {
15    unsafe { *$out = (*$lhs).dot(&*$rhs); }
16  };}
17
18macro_rules! impl_dot {
19  ($name:ident, $type1:ty, $type2:ty, $out_type:ty) => {
20    impl_binop!($name, $type1, $type2, $out_type, dot_op, FeatureFlag::Builtin(FeatureKind::Dot));
21    register_fxn_descriptor!($name, u8, "u8", u16, "u16", u32, "u32", u64, "u64", u128, "u128", i8, "i8", i16, "i16", i32, "i32", i64, "i64", i128, "i128", f32, "f32", f64, "f64");
22  };
23}
24
25impl_binop!(DotScalar, T, T, T, mul_op, FeatureFlag::Builtin(FeatureKind::Dot));
26register_fxn_descriptor!(DotScalar, u8, "u8", u16, "u16", u32, "u32", u64, "u64", u128, "u128", i8, "i8", i16, "i16", i32, "i32", i64, "i64", i128, "i128", f32, "f32", f64, "f64");
27
28#[cfg(all(feature = "row_vector2", feature = "row_vector2"))]
29impl_dot!(DotR2R2, RowVector2<T>, RowVector2<T>, T);
30#[cfg(all(feature = "vector2", feature = "vector2"))]
31impl_dot!(DotV2V2, Vector2<T>, Vector2<T>, T);
32
33#[cfg(all(feature = "row_vector3", feature = "row_vector3"))]
34impl_dot!(DotR3R3, RowVector3<T>, RowVector3<T>, T);
35#[cfg(all(feature = "vector3", feature = "vector3"))]
36impl_dot!(DotV3V3, Vector3<T>, Vector3<T>, T);
37
38#[cfg(all(feature = "row_vector4", feature = "row_vector4"))]
39impl_dot!(DotR4R4, RowVector4<T>, RowVector4<T>, T);
40#[cfg(all(feature = "vector4", feature = "vector4"))]
41impl_dot!(DotV4V4, Vector4<T>, Vector4<T>, T);
42
43#[cfg(all(feature = "matrix1", feature = "matrix1"))]
44impl_dot!(DotM1M1, Matrix2<T>, Matrix2<T>, T);
45#[cfg(all(feature = "matrix2", feature = "matrix2"))]
46impl_dot!(DotM2M2, Matrix2<T>, Matrix2<T>, T);
47#[cfg(all(feature = "matrix3", feature = "matrix3"))]
48impl_dot!(DotM3M3, Matrix3<T>, Matrix3<T>, T);
49#[cfg(all(feature = "matrix4", feature = "matrix4"))]
50impl_dot!(DotM4M4, Matrix4<T>, Matrix4<T>, T);
51
52#[cfg(all(feature = "matrixd", feature = "matrixd"))]
53impl_dot!(DotMDMD, DMatrix<T>, DMatrix<T>, T);
54#[cfg(all(feature = "vectord", feature = "vectord"))]
55impl_dot!(DotVDVD, DVector<T>, DVector<T>, T);
56#[cfg(all(feature = "row_vectord", feature = "row_vectord"))]
57impl_dot!(DotRDRD, RowDVector<T>, RowDVector<T>, T);
58
59macro_rules! impl_dot_match_arms {
60  ($arg:expr, $($lhs_type:tt, $($matrix_kind:tt, $target_type:tt, $value_string:tt),+);+ $(;)?) => {
61    match $arg {
62      $(
63        $(
64          #[cfg(feature = $value_string)]
65          (Value::$lhs_type(lhs), Value::$lhs_type(rhs)) => Ok(Box::new(DotScalar { lhs: lhs.clone(), rhs: rhs.clone(), out: Ref::new($target_type::default()) })),
66
67          #[cfg(all(feature = $value_string, feature = "vector2", feature = "vector2"))]
68          (Value::$matrix_kind(Matrix::Vector2(lhs)), Value::$matrix_kind(Matrix::Vector2(rhs))) => Ok(Box::new(DotV2V2 { lhs: lhs.clone(), rhs: rhs.clone(), out: Ref::new($target_type::default()) })),
69          #[cfg(all(feature = $value_string, feature = "row_vector2", feature = "row_vector2"))]
70          (Value::$matrix_kind(Matrix::RowVector2(lhs)), Value::$matrix_kind(Matrix::RowVector2(rhs))) => Ok(Box::new(DotR2R2 { lhs: lhs.clone(), rhs: rhs.clone(), out: Ref::new($target_type::default()) })),
71
72          #[cfg(all(feature = $value_string, feature = "vector3", feature = "vector3"))]
73          (Value::$matrix_kind(Matrix::Vector3(lhs)), Value::$matrix_kind(Matrix::Vector3(rhs))) => Ok(Box::new(DotV3V3 { lhs: lhs.clone(), rhs: rhs.clone(), out: Ref::new($target_type::default()) })),
74          #[cfg(all(feature = $value_string, feature = "row_vector3", feature = "row_vector3"))]
75          (Value::$matrix_kind(Matrix::RowVector3(lhs)), Value::$matrix_kind(Matrix::RowVector3(rhs))) => Ok(Box::new(DotR3R3 { lhs: lhs.clone(), rhs: rhs.clone(), out: Ref::new($target_type::default()) })),
76
77          #[cfg(all(feature = $value_string, feature = "vector4", feature = "vector4"))]
78          (Value::$matrix_kind(Matrix::Vector4(lhs)), Value::$matrix_kind(Matrix::Vector4(rhs))) => Ok(Box::new(DotV4V4 { lhs: lhs.clone(), rhs: rhs.clone(), out: Ref::new($target_type::default()) })),
79          #[cfg(all(feature = $value_string, feature = "row_vector4", feature = "row_vector4"))]
80          (Value::$matrix_kind(Matrix::RowVector4(lhs)), Value::$matrix_kind(Matrix::RowVector4(rhs))) => Ok(Box::new(DotR4R4 { lhs: lhs.clone(), rhs: rhs.clone(), out: Ref::new($target_type::default()) })),
81
82          #[cfg(all(feature = $value_string, feature = "matrix1", feature = "matrix1"))]
83          (Value::$matrix_kind(Matrix::Matrix2(lhs)), Value::$matrix_kind(Matrix::Matrix2(rhs))) => Ok(Box::new(DotM1M1 { lhs: lhs.clone(), rhs: rhs.clone(), out: Ref::new($target_type::default()) })),
84          #[cfg(all(feature = $value_string, feature = "matrix2", feature = "matrix2"))]
85          (Value::$matrix_kind(Matrix::Matrix2(lhs)), Value::$matrix_kind(Matrix::Matrix2(rhs))) => Ok(Box::new(DotM2M2 { lhs: lhs.clone(), rhs: rhs.clone(), out: Ref::new($target_type::default()) })),
86          #[cfg(all(feature = $value_string, feature = "matrix3", feature = "matrix3"))]
87          (Value::$matrix_kind(Matrix::Matrix3(lhs)), Value::$matrix_kind(Matrix::Matrix3(rhs))) => Ok(Box::new(DotM3M3 { lhs: lhs.clone(), rhs: rhs.clone(), out: Ref::new($target_type::default()) })),
88          #[cfg(all(feature = $value_string, feature = "matrix4", feature = "matrix4"))]
89          (Value::$matrix_kind(Matrix::Matrix4(lhs)), Value::$matrix_kind(Matrix::Matrix4(rhs))) => Ok(Box::new(DotM4M4 { lhs: lhs.clone(), rhs: rhs.clone(), out: Ref::new($target_type::default()) })),
90          #[cfg(all(feature = $value_string, feature = "matrixd", feature = "matrixd"))]
91          (Value::$matrix_kind(Matrix::DMatrix(lhs)), Value::$matrix_kind(Matrix::DMatrix(rhs))) => {
92            let (lhs_rows,lhs_cols) = {lhs.borrow().shape()};
93            let (rhs_rows,rhs_cols) = {rhs.borrow().shape()};
94            if lhs_rows != rhs_rows || lhs_cols != rhs_cols {
95              return Err(
96                MechError2::new(
97                  DimensionMismatch { dims: vec![lhs_rows, lhs_cols, rhs_rows, rhs_cols] },
98                  None
99                ).with_compiler_loc()
100              );
101            }
102            Ok(Box::new(DotMDMD { lhs: lhs.clone(), rhs: rhs.clone(), out: Ref::new($target_type::default()) }))
103          },
104          #[cfg(all(feature = $value_string, feature = "vectord", feature = "vectord"))]
105          (Value::$matrix_kind(Matrix::DVector(lhs)), Value::$matrix_kind(Matrix::DVector(rhs))) => {
106            let lhs_len = {lhs.borrow().len()};
107            let rhs_len = {rhs.borrow().len()};
108            if lhs_len != rhs_len {
109              return Err(MechError2::new(
110                DimensionMismatch { dims: vec![lhs_len, rhs_len] },
111                None
112              ).with_compiler_loc());
113            }
114            Ok(Box::new(DotVDVD { lhs: lhs.clone(), rhs: rhs.clone(), out: Ref::new($target_type::default()) }))
115          },
116          #[cfg(all(feature = $value_string, feature = "row_vectord", feature = "row_vectord"))]
117          (Value::$matrix_kind(Matrix::RowDVector(lhs)), Value::$matrix_kind(Matrix::RowDVector(rhs))) => {
118            let lhs_len = {lhs.borrow().len()};
119            let rhs_len = {rhs.borrow().len()};
120            if lhs_len != rhs_len {
121              return Err(MechError2::new(
122                DimensionMismatch { dims: vec![lhs_len, rhs_len] },
123                None
124              ).with_compiler_loc());
125            }
126            Ok(Box::new(DotRDRD { lhs: lhs.clone(), rhs: rhs.clone(), out: Ref::new($target_type::default()) }))
127          },
128        )+
129      )+
130      (arg1,arg2) => Err(MechError2::new(
131          UnhandledFunctionArgumentKind2 { arg: (arg1.kind(),arg2.kind()), fxn_name: stringify!($fxn).to_string() },
132          None
133        ).with_compiler_loc()
134      ),
135    }
136  }
137}
138
139fn impl_dot_fxn(lhs_value: Value, rhs_value: Value) -> MResult<Box<dyn MechFunction>> {
140  impl_dot_match_arms!(
141    (lhs_value, rhs_value),
142    I8,   MatrixI8,   i8,   "i8";
143    I16,  MatrixI16,  i16,  "i16";
144    I32,  MatrixI32,  i32,  "i32";
145    I64,  MatrixI64,  i64,  "i64";
146    I128, MatrixI128, i128, "i128";
147    U8,   MatrixU8,   u8,   "u8";
148    U16,  MatrixU16,  u16,  "u16";
149    U32,  MatrixU32,  u32,  "u32";
150    U64,  MatrixU64,  u64,  "u64";
151    U128, MatrixU128, u128, "u128";
152    F32,  MatrixF32,  f32,  "f32";
153    F64,  MatrixF64,  f64,  "f64";
154    R64, MatrixR64, R64, "rational";
155    C64, MatrixC64, C64, "complex";
156  )
157}
158
159impl_mech_binop_fxn!(MatrixDot,impl_dot_fxn,"matrix/dot");
mech_matrix/dot.rs

mech_matrix/
dot.rs
