Implement H5Type for complex

Author: mulimoen

hdf5-types/Cargo.toml

@@ -15,12 +15,14 @@ edition.workspace = true
 
 [features]
 h5-alloc = []
+complex = ["num-complex"]
 
 [dependencies]
 ascii = "1.1"
 cfg-if = { workspace = true }
 hdf5-sys = { workspace = true }
 libc = { workspace = true }
+num-complex = { version = "0.4", optional = true, default-features = false }
 
 [dev-dependencies]
 quickcheck = { version = "1.0", default-features = false }

hdf5-types/src/complex.rs

@@ -0,0 +1,30 @@
+use super::{CompoundField, CompoundType, H5Type, TypeDescriptor};
+
+use std::mem::size_of;
+
+use num_complex::Complex;
+
+unsafe impl<T: H5Type> H5Type for Complex<T> {
+    fn type_descriptor() -> TypeDescriptor {
+        TypeDescriptor::Compound(CompoundType {
+            fields: vec![
+                // Compatible with h5py definition of complex
+                CompoundField::typed::<T>("r", 0, 0),
+                CompoundField::typed::<T>("i", size_of::<T>(), 1),
+            ],
+            size: size_of::<T>() * 2,
+        })
+    }
+}
+
+#[cfg(test)]
+mod tests {
+    use super::*;
+    use num_complex::{Complex32, Complex64};
+
+    #[test]
+    fn complex() {
+        assert_eq!(Complex32::type_descriptor().size(), 8);
+        assert_eq!(Complex64::type_descriptor().size(), 16);
+    }
+}

hdf5-types/src/lib.rs

@@ -22,6 +22,9 @@ pub mod dyn_value;
 mod h5type;
 mod string;
 
+#[cfg(feature = "complex")]
+mod complex;
+
 pub use self::array::VarLenArray;
 pub use self::dyn_value::{DynValue, OwnedDynValue};
 pub use self::h5type::{