ndarray_cg 0.4.0

High-performance computer graphics mathematics library based on ndarray with vectors, matrices, and transformations
Documentation 
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
39
40
41
42
43
44
45
46
47
48
49
50
51
52
53
54
55
56
57
58
59
60
61
62
63
64
65
66
67
68
69
70
71
72
73
74
75
76
77
78
79
80
81

use crate::*;

/// Adds two matrices.
#[ inline ]
pub fn add< E, A, B, R >( r : &mut R, a : &A, b : &B )
where
  E : MatEl,
  E : nd::NdFloat,
  R : Indexable< Index = Ix2 > + IndexingMut< Scalar = E >,
  A : Indexable< Index = Ix2 > + IndexingRef< Scalar = E >,
  B : Indexable< Index = Ix2 > + IndexingRef< Scalar = E >,
{
  let rdim = r.dim();
  let adim = a.dim();
  let bdim = b.dim();

  // Check if dimensions are compatible for addition
  #[ cfg( debug_assertions ) ]
  if adim != bdim || rdim != adim
  {
    panic!
    (
      "Incompatible dimensions for matrix addition: a: {:?}, b: {:?}, r: {:?}",
      adim, bdim, rdim
    );
  }

  // println!( "a: {:?}, b: {:?}, r: {:?}", adim, bdim, rdim );
  for ( ( r_val, a_val ), b_val ) in r.iter_lsfirst_mut().zip( a.iter_lsfirst() ).zip( b.iter_lsfirst() )
  {
    *r_val = *a_val + *b_val;
  }
}

// This macro is workaround and undesirable.
// This macro is required because of Rut compiler compilation issue.
// ```
// overflow evaluating the requirement `d2::mat::private::Mat<_, _, _, _>: derive_tools::Add`
// ``

// impl_operator!( mat::DescriptorOrderRowMajor, Add, add );
// impl_operator!( mat::DescriptorOrderColumnMajor, Add, add );

impl< E, const ROWS : usize, const COLS : usize, Descriptor : mat::Descriptor > Add
for Mat< ROWS, COLS, E, Descriptor >
where
  E : MatEl,
  E : nd::NdFloat,
  Descriptor : mat::Descriptor,
  Mat< ROWS, COLS, E, Descriptor > : Indexable< Index = Ix2 > + IndexingMut< Scalar = E >,
{
  type Output = Self;

  #[ inline ]
  fn add( self, rhs : Self ) -> Self::Output
  {
    let mut result = Self::Output::default();
    add( &mut result, &self, &rhs );
    result
  }

}

impl< E, const ROWS : usize, const COLS : usize, Descriptor > Add< &Mat< ROWS, COLS, E, Descriptor > >
for &Mat< ROWS, COLS, E, Descriptor >
where
  Descriptor : mat::Descriptor,
  E : MatEl,
  E : nd::NdFloat,
  // Self : IndexingRef,
  Mat< ROWS, COLS, E, Descriptor > : Indexable< Index = Ix2 > + IndexingMut< Scalar = E >,
{
  type Output = Mat< ROWS, COLS, E, Descriptor >;

  fn add( self, rhs : &Mat< ROWS, COLS, E, Descriptor > ) -> Self::Output
  {
    let mut result = Self::Output::default();
    add( &mut result, self, rhs );
    result
  }
}