From b04d0578c59299598c298f3ca83af3278b13f08c Mon Sep 17 00:00:00 2001
From: hachem <im@hachem.wtf>
Date: Sat, 12 Oct 2024 17:31:16 +0200
Subject: Refactor 2: Complex

---
 libpsi-core/src/maths/numeric.rs | 24 +++++++++++++++++++++---
 1 file changed, 21 insertions(+), 3 deletions(-)

(limited to 'libpsi-core/src/maths/numeric.rs')

diff --git a/libpsi-core/src/maths/numeric.rs b/libpsi-core/src/maths/numeric.rs
index e6f3247..0fd4380 100644
--- a/libpsi-core/src/maths/numeric.rs
+++ b/libpsi-core/src/maths/numeric.rs
@@ -17,6 +17,22 @@ macro_rules! impl_numeric {
     };
 }
 
+macro_rules! impl_cnumeric {
+    ($($t:ty),*) => {
+        $(
+            impl Numeric for Complex<$t> {
+                fn zero() -> Self {
+                    Complex::new(0.0, 0.0)
+                }
+
+                fn one() -> Self {
+                    Complex::new(1.0, 0.0)
+                }
+            }
+        )*
+    };
+}
+
 macro_rules! impl_float {
     ($($t:ty, $sqrt_fn:path, $atan2_fn:path),*) => {
         $(
@@ -33,7 +49,7 @@ macro_rules! impl_float {
     };
 }
 
-macro_rules! impl_complex {
+macro_rules! impl_cfloat {
     ($($t:ty, $sqrt_fn:path, $atan2_fn:path, $cos_fn:path, $sin_fn:path),*) => {
         $(
             impl Float for Complex<$t> {
@@ -79,10 +95,12 @@ pub trait Numeric:
 }
 
 impl_numeric!(i32, i64, f32, f64);
+impl_cnumeric!(f32, f64);
+
 impl_float!(f32, libm::sqrtf, libm::atan2f);
 impl_float!(f64, libm::sqrt, libm::atan2);
-impl_complex!(f32, libm::sqrtf, libm::atan2f, libm::cosf, libm::sinf);
-impl_complex!(f64, libm::sqrt, libm::atan2, libm::cos, libm::sin);
+impl_cfloat!(f32, libm::sqrtf, libm::atan2f, libm::cosf, libm::sinf);
+impl_cfloat!(f64, libm::sqrt, libm::atan2, libm::cos, libm::sin);
 
 pub trait Integer: Numeric {}
 pub trait Float: Numeric {
-- 
cgit v1.3