ENH: Add set_norm_zooms

nipy · effigies · Oct 24, 2017 · Dec 18, 2017 · Dec 21, 2017 · Dec 21, 2017
commit de9323c42ff9e7cc6ef90abcee69eaebd8deb060
diff --git a/nibabel/analyze.py b/nibabel/analyze.py
@@ -710,6 +710,10 @@ def get_norm_zooms(self, raise_unknown=False):
         ''' Get zooms in mm/s units '''
         return self.get_zooms()
 
+    def set_norm_zooms(self, zooms):
+        ''' Set zooms in mm/s units '''
+        return self.set_zooms(zooms)
+
     def as_analyze_map(self):
         """ Return header as mapping for conversion to Analyze types
 

diff --git a/nibabel/freesurfer/mghformat.py b/nibabel/freesurfer/mghformat.py
@@ -294,6 +294,12 @@ def get_norm_zooms(self, raise_unknown=False):
 
         return zooms
 
+    def set_norm_zooms(self, zooms):
+        if len(zooms) == 4:
+            zooms = zooms[:3] + (zooms[3] * 1000,)
+
+        self.set_zooms(zooms)
+
     def get_data_shape(self):
         """ Get shape of data
         """

diff --git a/nibabel/nifti1.py b/nibabel/nifti1.py
@@ -1710,6 +1710,11 @@ def get_norm_zooms(self, raise_unknown=False):
 
         return xyz_zooms + t_zoom
 
+    def set_norm_zooms(self, zooms):
+        ''' Set zooms in mm/s units '''
+        self.set_zooms(zooms)
+        self.set_xyzt_units('mm', 'sec')
+
     def _clean_after_mapping(self):
         """ Set format-specific stuff after converting header from mapping
 

diff --git a/nibabel/spatialimages.py b/nibabel/spatialimages.py
@@ -242,6 +242,10 @@ def get_norm_zooms(self, raise_unknown=False):
         ''' Get zooms in mm/s units '''
         return self.get_zooms()
 
+    def set_norm_zooms(self, zooms):
+        ''' Get zooms in mm/s units '''
+        return self.set_zooms(zooms)
+
     def get_base_affine(self):
         shape = self.get_data_shape()
         zooms = self.get_zooms()