enum iterator (#45)

bempp · Oct 28, 2024 · f724e04 · f724e04
1 parent 18fbc93
commit f724e04
Show file tree

Hide file tree

Showing 2 changed files with 4 additions and 1 deletion.
diff --git a/Cargo.toml b/Cargo.toml
@@ -28,6 +28,8 @@ mpi = { version = "0.8.*", optional = true }
 num = "0.4"
 rlst = { version = "0.2.0", default-features = false }
 serde = { version = "1", features = ["derive"], optional = true }
+strum = "0.26"
+strum_macros = "0.26"
 
 [dev-dependencies]
 paste = "1.*"

diff --git a/src/types.rs b/src/types.rs
@@ -2,6 +2,7 @@
 #[cfg(feature = "mpi")]
 use mpi::traits::Equivalence;
 use rlst::{Array, BaseArray, VectorContainer};
+use strum_macros::EnumIter;
 
 /// An N-dimensional array
 pub type ArrayND<const N: usize, T> = Array<T, BaseArray<T, VectorContainer<T>, N>, N>;
@@ -44,7 +45,7 @@ pub enum MapType {
 
 /// The type of a reference cell
 #[cfg_attr(feature = "serde", derive(serde::Serialize, serde::Deserialize))]
-#[derive(Debug, PartialEq, Eq, Clone, Copy, Hash)]
+#[derive(EnumIter, Debug, PartialEq, Eq, Clone, Copy, Hash)]
 #[repr(u8)]
 pub enum ReferenceCellType {
     /// A point