enh: add labels

Author: oesteban

sdcflows/interfaces/reportlets.py

@@ -21,6 +21,10 @@ class _FieldmapReportletInputSpec(reporting.ReportCapableInputSpec):
                       desc='filename for the visual report')
     show = traits.Enum(1, 0, 'both', usedefault=True,
                        desc='where the fieldmap should be shown')
+    reference_label = traits.Str('Reference', usedefault=True,
+                                 desc='a label name for the reference mosaic')
+    moving_label = traits.Str('Fieldmap (Hz)', usedefault=True,
+                              desc='a label name for the reference mosaic')
 
 
 class FieldmapReportlet(reporting.ReportCapableInterface):
@@ -42,9 +46,12 @@ def _generate_report(self):
         """Generate a reportlet."""
         NIWORKFLOWS_LOG.info('Generating visual report')
 
-        refnii = load_img(self.inputs.reference)
+        movnii = refnii = load_img(self.inputs.reference)
         fmapnii = load_img(self.inputs.fieldmap)
 
+        if isdefined(self.inputs.moving):
+            movnii = load_img(self.inputs.moving)
+
         contour_nii = mask_nii = None
         if isdefined(self.inputs.mask):
             contour_nii = load_img(self.inputs.mask)
@@ -71,17 +78,17 @@ def _generate_report(self):
 
         # Call composer
         compose_view(
-            plot_registration(refnii, 'moving-image',
+            plot_registration(movnii, 'moving-image',
                               estimate_brightness=True,
                               cuts=cuts,
-                              label='fieldmap (Hz)',
+                              label=self.inputs.moving_label,
                               contour=contour_nii,
                               compress=False,
                               **fmap_overlay[1]),
             plot_registration(refnii, 'fixed-image',
                               estimate_brightness=True,
                               cuts=cuts,
-                              label='reference',
+                              label=self.inputs.reference_label,
                               contour=contour_nii,
                               compress=False,
                               **fmap_overlay[0]),

sdcflows/workflows/tests/test_pepolar.py

@@ -157,7 +157,7 @@ def test_pepolar_wf1(bids_layouts, output_path, dataset, workdir):
             wf.inputs.inputnode.in_reference_brain = boldref.path
             wf.inputs.inputnode.in_reference = boldref.path
 
-        rep = pe.Node(FieldmapReportlet(), 'simple_report')
+        rep = pe.Node(FieldmapReportlet(reference_label='EPI Reference'), 'simple_report')
         rep.interface._always_run = True
         dsink = pe.Node(DerivativesDataSink(
             base_directory=str(output_path), keep_dtype=True,

sdcflows/workflows/tests/test_phdiff.py

@@ -31,7 +31,7 @@ def test_workflow(bids_layouts, tmpdir, output_path, dataset, workdir):
 
     if output_path:
         from ...interfaces.reportlets import FieldmapReportlet
-        rep = pe.Node(FieldmapReportlet(), 'simple_report')
+        rep = pe.Node(FieldmapReportlet(reference_label='Magnitude'), 'simple_report')
         rep.interface._always_run = True
 
         dsink = pe.Node(DerivativesDataSink(