added MGH and MGZ to supported file-formats

Author: tclose

pydra/fileformats-mrtrix3/fileformats/mrtrix3/images.py

@@ -67,7 +67,7 @@ def dims(self):
         return self.metadata["dim"]
 
 
-class MrtrixImage(BaseMrtrixImage):
+class ImageFormat(BaseMrtrixImage):
 
     ext = ".mif"
 
@@ -83,14 +83,14 @@ def data_file(self):
         return self
 
 
-class MrtrixImageHeader(BaseMrtrixImage):
+class ImageHeader(BaseMrtrixImage):
 
     ext = ".mih"
 
     @mark.required
     @property
     def data_file(self):
-        return MrtrixDataFile(self.data_fspath)
+        return ImageDataFile(self.data_fspath)
 
     def __attrs_post_init__(self):
         if len(self.fspaths) == 1:
@@ -99,7 +99,7 @@ def __attrs_post_init__(self):
         super().__attrs_post_init__()
 
 
-class MrtrixDataFile(File):
+class ImageDataFile(File):
 
     ext = ".dat"
 
@@ -111,8 +111,10 @@ class MrtrixDataFile(File):
     fileformats.medimage.NiftiX,
     fileformats.medimage.Nifti1,
     fileformats.medimage.Nifti2,
-    MrtrixImage,
-    MrtrixImageHeader,
+    ImageFormat,
+    ImageHeader,
+    fileformats.medimage.MGH,
+    fileformats.medimage.MGHZip,
     fileformats.medimage.Analyze,
 ]
 
@@ -122,7 +124,9 @@ class MrtrixDataFile(File):
     fileformats.medimage.NiftiX,
     fileformats.medimage.Nifti1,
     fileformats.medimage.Nifti2,
-    MrtrixImage,
-    MrtrixImageHeader,
+    ImageFormat,
+    ImageHeader,
+    fileformats.medimage.MGH,
+    fileformats.medimage.MGHZip,
     fileformats.medimage.Analyze,
 ]