add contrast outputs for EstimatedModel when factor_info used in Level1design

Author: manuegrx

nipype/interfaces/spm/model.py

@@ -250,6 +250,34 @@ class EstimateModelOutputSpec(TraitedSpec):
         ImageFileSPM(exists=True),
         desc="Images of the standard deviation of parameter posteriors",
     )
+    con_images = OutputMultiPath(
+        File(exists=True),
+        desc=(
+            "contrast images from a t-contrast "
+            "(created if factor_info used in Level1Design)"
+        ),
+    )
+    spmT_images = OutputMultiPath(
+        File(exists=True),
+        desc=(
+            "stat images from a t-contrast"
+            "(created if factor_info used in Level1Design)"
+        ),
+    )
+    ess_images = OutputMultiPath(
+        File(exists=True),
+        desc=(
+            "contrast images from an F-contrast"
+            "(created if factor_info used in Level1Design)"
+        ),
+    )
+    spmF_images = OutputMultiPath(
+        File(exists=True),
+        desc=(
+            "stat images from an F-contrast"
+            "(created if factor_info used in Level1Design)"
+        ),
+    )
 
 
 class EstimateModel(SPMCommand):
@@ -319,6 +347,29 @@ def _list_outputs(self):
                 outputs["residual_images"] = glob(os.path.join(pth, "Res_*"))
             if betas:
                 outputs["beta_images"] = [os.path.join(pth, beta) for beta in betas]
+            # When 'factor_info' is used in Level1Design
+            # spm automatically creates contrast
+            try:
+                contrast = [c.Vcon[0][0].fname[0] for c in spm["SPM"][0, 0].xCon[0]]
+                contrast_spm = [c.Vspm[0][0].fname[0] for c in spm["SPM"][0, 0].xCon[0]]
+            except Exception:
+                contrast = []
+                contrast_spm = []
+
+            if contrast:
+                outputs["con_images"] = [
+                    (os.path.join(pth, cont) for cont in contrast if 'con' in cont)
+                ]
+                outputs["ess_images"] = [
+                    (os.path.join(pth, cont) for cont in contrast if 'ess' in cont)
+                ]
+            if contrast_spm:
+                outputs["spmT_images"] = [
+                    (os.path.join(pth, cont) for cont in contrast_spm if 'spmT' in cont)
+                ]
+                outputs["spmF_images"] = [
+                    (os.path.join(pth, cont) for cont in contrast_spm if 'spmF' in cont)
+                ]
 
         outputs["mask_image"] = os.path.join(pth, "mask.{}".format(outtype))
         outputs["spm_mat_file"] = os.path.join(pth, "SPM.mat")

nipype/interfaces/spm/tests/test_auto_EstimateModel.py

@@ -46,6 +46,8 @@ def test_EstimateModel_outputs():
         SDbetas=dict(),
         SDerror=dict(),
         beta_images=dict(),
+        con_images=dict(),
+        ess_images=dict(),
         labels=dict(
             extensions=[".hdr", ".img", ".img.gz", ".nii"],
         ),
@@ -56,6 +58,8 @@ def test_EstimateModel_outputs():
             extensions=[".hdr", ".img", ".img.gz", ".nii"],
         ),
         residual_images=dict(),
+        spmF_images=dict(),
+        spmT_images=dict(),
         spm_mat_file=dict(
             extensions=None,
         ),