ENH: Add set_norm_zooms

Author: effigies

nibabel/analyze.py

@@ -714,6 +714,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
 

nibabel/freesurfer/mghformat.py

@@ -296,6 +296,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
         '''

nibabel/nifti1.py

@@ -1671,6 +1671,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
 

nibabel/spatialimages.py

@@ -246,6 +246,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()