forked from StrafesNET/strafe-project
matrix: macro mat mul
This commit is contained in:
parent
176eb762e3
commit
e5f95b97ce
@ -83,17 +83,94 @@ macro_rules! impl_wide_vector_operations {
|
|||||||
};
|
};
|
||||||
}
|
}
|
||||||
|
|
||||||
|
#[doc(hidden)]
|
||||||
|
#[macro_export(local_inner_macros)]
|
||||||
|
macro_rules! impl_matrix_wide_dot_transpose_helper {
|
||||||
|
(
|
||||||
|
$value:ident,
|
||||||
|
($struct: ident { $($field: ident), + }),
|
||||||
|
($from_struct: ident { $($from_field: ident), + }),
|
||||||
|
$static_field: ident
|
||||||
|
) => {
|
||||||
|
$struct {
|
||||||
|
$(
|
||||||
|
$field: $value.$from_field.$static_field
|
||||||
|
), +
|
||||||
|
}
|
||||||
|
}
|
||||||
|
}
|
||||||
|
#[doc(hidden)]
|
||||||
|
#[macro_export(local_inner_macros)]
|
||||||
|
macro_rules! impl_matrix_wide_dot_inner {
|
||||||
|
(
|
||||||
|
// MatY<VecX>.MatX<VecZ> = MatY<VecZ>
|
||||||
|
$lhs:ident, $lhs_field_outer:ident, $wide_dot:ident, $rhs:ident,
|
||||||
|
$struct_inner_thru: tt, //VecX
|
||||||
|
($struct_inner: ident { $($field_inner: ident), + }), //VecX
|
||||||
|
($rhs_struct_inner: ident { $($rhs_field_inner: ident), + }), //VecZ
|
||||||
|
$rhs_outer: tt //MatX
|
||||||
|
) => {
|
||||||
|
$rhs_struct_inner {
|
||||||
|
$(
|
||||||
|
$rhs_field_inner: $lhs.$lhs_field_outer.$wide_dot(
|
||||||
|
//construct a transposed vector with the same width as $struct_outer
|
||||||
|
$crate::impl_matrix_wide_dot_transpose_helper!{
|
||||||
|
$rhs,
|
||||||
|
$struct_inner_thru, //VecZ
|
||||||
|
$rhs_outer, //MatX
|
||||||
|
$rhs_field_inner
|
||||||
|
}
|
||||||
|
)
|
||||||
|
), +
|
||||||
|
}
|
||||||
|
}
|
||||||
|
}
|
||||||
|
#[doc(hidden)]
|
||||||
|
#[macro_export(local_inner_macros)]
|
||||||
|
macro_rules! impl_matrix_wide_dot_outer {
|
||||||
|
(
|
||||||
|
// MatY<VecX>.MatX<VecZ> = MatY<VecZ>
|
||||||
|
$lhs:ident, $wide_dot:ident, $rhs:ident,
|
||||||
|
//result matrix shape
|
||||||
|
($struct_outer: ident { $($field_outer: ident), + }),//MatY
|
||||||
|
$rhs_struct_inner: tt,//VecZ
|
||||||
|
//inner loop shape
|
||||||
|
$struct_inner: tt,//VecX
|
||||||
|
$rhs_matrix: tt//MatX
|
||||||
|
|
||||||
|
) => {
|
||||||
|
$struct_outer {
|
||||||
|
$(
|
||||||
|
$field_outer: $crate::impl_matrix_wide_dot_inner!{
|
||||||
|
$lhs, $field_outer, $wide_dot, $rhs,
|
||||||
|
$struct_inner, //VecX
|
||||||
|
$struct_inner, //VecX
|
||||||
|
$rhs_struct_inner, //VecZ
|
||||||
|
$rhs_matrix //MatX
|
||||||
|
}
|
||||||
|
), +
|
||||||
|
}
|
||||||
|
}
|
||||||
|
}
|
||||||
|
|
||||||
// Notes:
|
// Notes:
|
||||||
// Mat3<Vec2>.dot(Vec2) -> Vec3
|
// Mat3<Vec2>.dot(Vec2) -> Vec3
|
||||||
|
// lhs.dot(rhs) -> out
|
||||||
|
// lhs = Mat3<Vec4>
|
||||||
|
// rhs = Mat4<Vec2>
|
||||||
|
// out = Mat3<Vec2>
|
||||||
// Mat3<Vec4>.dot(Mat4<Vec2>) -> Mat3<Vec2>
|
// Mat3<Vec4>.dot(Mat4<Vec2>) -> Mat3<Vec2>
|
||||||
// mat.mat can be implemented off the back of mat.vec
|
// how to matrix multiply:
|
||||||
|
// RHS TRANSPOSE
|
||||||
|
// Mat4<Vec2> -> Mat2<Vec4>
|
||||||
|
// rhs_t = Mat2<Vec4>
|
||||||
|
// inner loop:
|
||||||
|
// out[y][x] = lhs[y].dot(rhs_t[x])
|
||||||
|
|
||||||
#[doc(hidden)]
|
#[doc(hidden)]
|
||||||
#[macro_export(local_inner_macros)]
|
#[macro_export(local_inner_macros)]
|
||||||
macro_rules! impl_matrix_wide_mul {
|
macro_rules! impl_matrix_wide_dot {
|
||||||
(
|
(
|
||||||
//TODO: Fixed point impls
|
|
||||||
($struct_outer: ident { $($field_outer: ident), + }, $vector_outer: ident { $($vector_field_outer: ident), + }, $size_outer: expr),
|
($struct_outer: ident { $($field_outer: ident), + }, $vector_outer: ident { $($vector_field_outer: ident), + }, $size_outer: expr),
|
||||||
($struct_inner: ident { $($field_inner: ident), + }, $matrix_inner: ident { $($matrix_field_inner: ident), + }, $size_inner: expr),
|
($struct_inner: ident { $($field_inner: ident), + }, $matrix_inner: ident { $($matrix_field_inner: ident), + }, $size_inner: expr),
|
||||||
($rhs_struct_inner: ident { $($rhs_field_inner: ident), + }, $rhs_matrix_inner: ident { $($rhs_matrix_field_inner: ident), + }, $rhs_size_inner: expr),
|
($rhs_struct_inner: ident { $($rhs_field_inner: ident), + }, $rhs_matrix_inner: ident { $($rhs_matrix_field_inner: ident), + }, $rhs_size_inner: expr),
|
||||||
@ -103,13 +180,15 @@ macro_rules! impl_matrix_wide_mul {
|
|||||||
paste::item!{
|
paste::item!{
|
||||||
#[inline]
|
#[inline]
|
||||||
pub fn [<wide_dot_ $size_outer x $size_inner _ $size_inner x $rhs_size_inner _ $lhs _ $rhs>](self,rhs:$matrix_inner<$rhs_struct_inner<fixed_wide::fixed::Fixed<{$rhs},{$rhs*32}>>>)->$struct_outer<$rhs_struct_inner<fixed_wide::fixed::Fixed<{$lhs+$rhs},{($lhs+$rhs)*32}>>>{
|
pub fn [<wide_dot_ $size_outer x $size_inner _ $size_inner x $rhs_size_inner _ $lhs _ $rhs>](self,rhs:$matrix_inner<$rhs_struct_inner<fixed_wide::fixed::Fixed<{$rhs},{$rhs*32}>>>)->$struct_outer<$rhs_struct_inner<fixed_wide::fixed::Fixed<{$lhs+$rhs},{($lhs+$rhs)*32}>>>{
|
||||||
//just made this up, don't trust it
|
$crate::impl_matrix_wide_dot_outer!(
|
||||||
let tr=rhs.transpose();
|
//constituent idents
|
||||||
//TODO: use a macro expansion instead of transpose and map
|
self,[<wide_dot_ $lhs _ $rhs>],rhs,
|
||||||
self.map(|axis|
|
//result matrix shape
|
||||||
tr.map(|trax|
|
($struct_outer { $($field_outer), + }),
|
||||||
axis.[<wide_dot_ $lhs _ $rhs>](trax)
|
($rhs_struct_inner { $($rhs_field_inner), + }),
|
||||||
).to_vector()
|
//inner loop shape
|
||||||
|
($struct_inner { $($field_inner), + }),
|
||||||
|
($matrix_inner { $($matrix_field_inner), + })
|
||||||
)
|
)
|
||||||
}
|
}
|
||||||
}
|
}
|
||||||
@ -119,32 +198,32 @@ macro_rules! impl_matrix_wide_mul {
|
|||||||
|
|
||||||
#[doc(hidden)]
|
#[doc(hidden)]
|
||||||
#[macro_export(local_inner_macros)]
|
#[macro_export(local_inner_macros)]
|
||||||
macro_rules! impl_matrix_wide_mul_shim {
|
macro_rules! impl_matrix_wide_dot_shim {
|
||||||
(
|
(
|
||||||
($outer_info:tt,$inner_info:tt,$rhs_info:tt),
|
($outer_info:tt,$inner_info:tt,$rhs_info:tt),
|
||||||
($lhs: expr, $rhs: expr)
|
($lhs: expr, $rhs: expr)
|
||||||
) => {
|
) => {
|
||||||
$crate::impl_matrix_wide_mul!($outer_info,$inner_info,$rhs_info,($lhs,$rhs));
|
$crate::impl_matrix_wide_dot!($outer_info,$inner_info,$rhs_info,($lhs,$rhs));
|
||||||
}
|
}
|
||||||
}
|
}
|
||||||
#[doc(hidden)]
|
#[doc(hidden)]
|
||||||
#[macro_export(local_inner_macros)]
|
#[macro_export(local_inner_macros)]
|
||||||
macro_rules! impl_matrix_wide_mul_8x8 {
|
macro_rules! impl_matrix_wide_dot_8x8 {
|
||||||
(
|
(
|
||||||
($outer_info:tt,$inner_info:tt),
|
($outer_info:tt,$inner_info:tt),
|
||||||
$rhs_info:tt
|
$rhs_info:tt
|
||||||
) => {
|
) => {
|
||||||
$crate::do_macro_8x8!(impl_matrix_wide_mul_shim,($outer_info,$inner_info,$rhs_info));
|
$crate::do_macro_8x8!(impl_matrix_wide_dot_shim,($outer_info,$inner_info,$rhs_info));
|
||||||
}
|
}
|
||||||
}
|
}
|
||||||
#[doc(hidden)]
|
#[doc(hidden)]
|
||||||
#[macro_export(local_inner_macros)]
|
#[macro_export(local_inner_macros)]
|
||||||
macro_rules! impl_matrix_wide_mul_repeat_rhs {
|
macro_rules! impl_matrix_wide_dot_repeat_rhs {
|
||||||
(
|
(
|
||||||
($outer_info:tt,($($rhs_info:tt),+)),
|
($outer_info:tt,($($rhs_info:tt),+)),
|
||||||
$inner_info:tt
|
$inner_info:tt
|
||||||
) => {
|
) => {
|
||||||
$crate::macro_repeated!(impl_matrix_wide_mul_8x8,($outer_info,$inner_info),$($rhs_info),+);
|
$crate::macro_repeated!(impl_matrix_wide_dot_8x8,($outer_info,$inner_info),$($rhs_info),+);
|
||||||
}
|
}
|
||||||
}
|
}
|
||||||
#[doc(hidden)]
|
#[doc(hidden)]
|
||||||
|
@ -48,7 +48,7 @@ macro_rules! impl_matrix_inner_shim {
|
|||||||
) => {
|
) => {
|
||||||
$crate::macro_repeated!(impl_matrix_inner,$matrix_info,$($vector_info),+);
|
$crate::macro_repeated!(impl_matrix_inner,$matrix_info,$($vector_info),+);
|
||||||
#[cfg(feature="fixed_wide")]
|
#[cfg(feature="fixed_wide")]
|
||||||
$crate::macro_repeated!(impl_matrix_wide_mul_repeat_rhs,($matrix_info,($($vector_info),+)),$($vector_info),+);
|
$crate::macro_repeated!(impl_matrix_wide_dot_repeat_rhs,($matrix_info,($($vector_info),+)),$($vector_info),+);
|
||||||
}
|
}
|
||||||
}
|
}
|
||||||
#[doc(hidden)]
|
#[doc(hidden)]
|
||||||
|
Loading…
Reference in New Issue
Block a user