Merge pull request #35 from yhuang43/master

complex64 support
freesurfer · Jul 29, 2024 · ded5f1d · ded5f1d
2 parents c0f7824 + 281e451
commit ded5f1d
Showing 1 changed file with 2 additions and 0 deletions.
diff --git a/surfa/io/framed.py b/surfa/io/framed.py
@@ -230,6 +230,7 @@ def dtype_from_id(self, id):
             3:  '>f4',  # float
             4:  '>i2',  # short
             6:  '>f4',  # tensor
+            7:  '>c8',  # complex64
             10: '>u2',  # ushort
         }
         dtype = mgh_types.get(id)
@@ -398,6 +399,7 @@ def save(self, arr, filename, intent=FramedArrayIntents.mri):
                 np.floating: 3,
                 np.int16: 4,
                 np.uint16: 10,
+                np.complex64: 7,
             }
             dtype_id = next((i for dt, i in type_map.items() if np.issubdtype(arr.dtype, dt)), None)
             if dtype_id is None: