geo_nd/
farray.rs

1use super::vector;
2use super::{Float, Vector};
3
4//a Macros
5//a FArray
6//tp FArray
7/// The [FArray] is a wrapper around a `D` sized array of [Float]s.
8///
9/// It provides implementations of the traits required for a [Vector]
10/// trait, hence it can be used for a [Vector] of any size `D`.
11///
12/// Default must be provided separately as it is only implemented
13/// for \[F; D\] up to D=32
14#[derive(Clone, Copy, Debug, PartialEq)]
15#[repr(transparent)]
16pub struct FArray<F: Float, const D: usize> {
17    data: [F; D],
18}
19
20//a Macros to implement the traits
21//mi farray_basic_traits!
22macro_rules! farray_basic_traits {
23    { $f:ty,  $e:expr, $ty:ty } => {
24
25        //ip Default for FArray
26        impl std::default::Default for $ty {
27            fn default() -> Self {
28                Self {
29                    data: vector::zero(),
30                }
31            }
32        }
33
34        //ip Display for FArray
35        impl std::fmt::Display for $ty {
36            fn fmt(&self, f: &mut std::fmt::Formatter) -> std::fmt::Result {
37                vector::fmt(f, &self.data)
38            }
39        }
40    }
41}
42
43//mi farray_mul_div_traits
44macro_rules! farray_mul_div_traits {
45    { $f:ty, $e:expr, $ty:ty,
46      $trait_op:ident, $fn_op:ident, $op:tt,
47      $trait_assign_op:ident, $fn_assign_op:ident, $op_assign:tt  } => {
48
49        // Mul/Div by  $f  for Self
50        impl std::ops::$trait_op<$f> for $ty {
51            type Output = $ty;
52            fn $fn_op(mut self, other: $f) -> $ty {
53                for i in 0..$e {
54                    self.data[i] $op_assign other;
55                }
56                self
57            }
58        }
59
60        impl std::ops::$trait_assign_op<$f> for $ty {
61            fn $fn_assign_op(&mut self, other: $f) {
62                for i in 0..$e {self.data[i] $op_assign other;}
63            }
64        }
65
66        // Mul/Div by  & $f  for Self
67        impl <'a> std::ops::$trait_op<&'a $f> for $ty {
68            type Output = $ty;
69            fn $fn_op(mut self, other: &'a $f) -> $ty {
70                for i in 0..$e {
71                    self.data[i] $op_assign other;
72                }
73                self
74            }
75        }
76
77        impl <'a> std::ops::$trait_assign_op<&'a $f> for $ty {
78            fn $fn_assign_op(&mut self, other: &'a $f) {
79                for i in 0..$e {self.data[i] $op_assign other;}
80            }
81        }
82    }
83}
84
85//mi farray_vector_trait
86macro_rules! farray_vector_trait {
87    { $f:ty,  $e:expr, $ty:ty } => {
88
89        impl Vector<$f, $e> for $ty {
90        }
91
92    }
93}
94
95//mi farray_traits
96macro_rules! farray_traits {
97    { $f:ty,  $e:expr, $ty:ty } => {
98
99        farray_basic_traits!{$f, $e, $ty}
100        crate::ref_traits!{$f, $e, $ty}
101        crate::convert_traits!{$f, $e, $ty}
102        crate::unary_traits!{$f, $e, $ty}
103        crate::elementwise_traits!{$f, $e, $ty, Add, add, +, AddAssign, add_assign, +=}
104        crate::elementwise_traits!{$f, $e, $ty, Sub, sub, -, SubAssign, sub_assign, -=}
105        crate::serialize_traits!{$f, $e, $ty}
106        farray_mul_div_traits!{$f, $e, $ty, Mul, mul, *, MulAssign, mul_assign, *=}
107        farray_mul_div_traits!{$f, $e, $ty, Div, div, /, DivAssign, div_assign, /=}
108
109        farray_vector_trait!{$f, $e, $ty}
110    }
111}
112
113//mi Invoke farray_traits!
114farray_traits! { f32, 2, FArray<f32,2> }
115farray_traits! { f32, 3, FArray<f32,3> }
116farray_traits! { f32, 4, FArray<f32,4> }
117farray_traits! { f32, 5, FArray<f32,5> }
118farray_traits! { f32, 6, FArray<f32,6> }
119farray_traits! { f32, 7, FArray<f32,7> }
120farray_traits! { f32, 8, FArray<f32,8> }
121farray_traits! { f32, 9, FArray<f32,9> }
122farray_traits! { f32, 10, FArray<f32,10> }
123farray_traits! { f32, 11, FArray<f32,11> }
124farray_traits! { f32, 12, FArray<f32,12> }
125farray_traits! { f32, 13, FArray<f32,13> }
126farray_traits! { f32, 14, FArray<f32,14> }
127farray_traits! { f32, 15, FArray<f32,15> }
128farray_traits! { f32, 16, FArray<f32,16> }
129
130farray_traits! { f64, 2, FArray<f64,2> }
131farray_traits! { f64, 3, FArray<f64,3> }
132farray_traits! { f64, 4, FArray<f64,4> }
133farray_traits! { f64, 5, FArray<f64,5> }
134farray_traits! { f64, 6, FArray<f64,6> }
135farray_traits! { f64, 7, FArray<f64,7> }
136farray_traits! { f64, 8, FArray<f64,8> }
137farray_traits! { f64, 9, FArray<f64,9> }
138farray_traits! { f64, 10, FArray<f64,10> }
139farray_traits! { f64, 11, FArray<f64,11> }
140farray_traits! { f64, 12, FArray<f64,12> }
141farray_traits! { f64, 13, FArray<f64,13> }
142farray_traits! { f64, 14, FArray<f64,14> }
143farray_traits! { f64, 15, FArray<f64,15> }
144farray_traits! { f64, 16, FArray<f64,16> }
geo_nd/farray.rs

geo_nd/
farray.rs
