From 999f93f5e2c518be8c5d8d70c9a23d89d43d0a5d Mon Sep 17 00:00:00 2001
From: Lizzy Fleckenstein <eliasfleckenstein@web.de>
Date: Tue, 9 May 2023 18:26:12 +0200
Subject: [PATCH] Add support for cgmath and collision

---
 lib/Cargo.toml |  6 +++++
 lib/src/lib.rs | 73 ++++++++++++++++++++++++++++++++++++++++++++++++++
 2 files changed, 79 insertions(+)

diff --git a/lib/Cargo.toml b/lib/Cargo.toml
index ddd300b..1513795 100644
--- a/lib/Cargo.toml
+++ b/lib/Cargo.toml
@@ -11,10 +11,16 @@ description = "Generate random data"
 readme = "README.md"
 keywords = ["random"]
 
+[features]
+cgmath = ["dep:cgmath"]
+collision = ["cgmath", "dep:collision"]
+
 [dependencies]
 rand = "0.8.5"
 generate-random-macro = { version = "0.1.0", path = "../derive-macro" }
 enumset = { git = "https://github.com/Lymia/enumset", optional = true }
+cgmath = { version = "0.17.0", optional = true }
+collision = { version = "0.20.1", optional = true }
 
 [dev-dependencies]
 rand_chacha = "0.3.1"
diff --git a/lib/src/lib.rs b/lib/src/lib.rs
index bd49e6b..71f68a8 100644
--- a/lib/src/lib.rs
+++ b/lib/src/lib.rs
@@ -226,6 +226,79 @@ macro_rules! impl_generate_random_tuple {
 
 impl_generate_random_tuple!(A B C D E F G H I J K L);
 
+#[cfg(feature = "cgmath")]
+mod impl_cgmath {
+    use super::*;
+    use cgmath::*;
+
+    macro_rules! impl_from {
+		($ident:ident, $from:ty, $bound:ident $(+ $others:ident )*) => {
+			impl<T: $bound $(+ $others)* + GenerateRandom> GenerateRandom for $ident<T> {
+				fn generate_random<R: rand::Rng + ?Sized>(rng: &mut R) -> Self {
+					<$from>::generate_random(rng).into()
+				}
+			}
+		};
+	}
+
+    macro_rules! impl_wrap {
+        ($ident:ident) => {
+            impl<T: GenerateRandom> GenerateRandom for $ident<T> {
+                fn generate_random<R: rand::Rng + ?Sized>(rng: &mut R) -> Self {
+                    Self(GenerateRandom::generate_random(rng))
+                }
+            }
+        };
+    }
+
+    impl_from!(Vector1, [T; 1], Clone);
+    impl_from!(Vector2, [T; 2], Clone);
+    impl_from!(Vector3, [T; 3], Clone);
+    impl_from!(Vector4, [T; 4], Clone);
+
+    impl_from!(Point1, [T; 1], Clone);
+    impl_from!(Point2, [T; 2], Clone);
+    impl_from!(Point3, [T; 3], Clone);
+
+    impl_from!(Matrix2, [[T; 2]; 2], Copy);
+    impl_from!(Matrix3, [[T; 3]; 3], Copy);
+    impl_from!(Matrix4, [[T; 4]; 4], Copy);
+
+    impl_wrap!(Deg);
+    impl_wrap!(Rad);
+
+    impl<A: GenerateRandom> GenerateRandom for Euler<A> {
+        fn generate_random<R: rand::Rng + ?Sized>(rng: &mut R) -> Self {
+            let x = GenerateRandom::generate_random(rng);
+            let y = GenerateRandom::generate_random(rng);
+            let z = GenerateRandom::generate_random(rng);
+            Self::new(x, y, z)
+        }
+    }
+}
+
+#[cfg(feature = "collision")]
+mod impl_collision {
+    use super::*;
+    use cgmath::*;
+    use collision::*;
+
+    macro_rules! impl_abbb {
+        ($ident:ident) => {
+            impl<S: GenerateRandom + BaseNum> GenerateRandom for $ident<S> {
+                fn generate_random<R: rand::Rng + ?Sized>(rng: &mut R) -> Self {
+                    let p1 = GenerateRandom::generate_random(rng);
+                    let p2 = GenerateRandom::generate_random(rng);
+                    Self::new(p1, p2)
+                }
+            }
+        };
+    }
+
+    impl_abbb!(Aabb2);
+    impl_abbb!(Aabb3);
+}
+
 #[cfg(test)]
 mod tests {
     use super::*;
-- 
2.44.0