ENH: Use nireports report generation

Author: mgxd

nibabies/cli/run.py

@@ -139,8 +139,6 @@ def main():
                 config.execution.unique_labels,
                 config.execution.nibabies_dir,
                 config.execution.run_uuid,
-                config=pkgrf("nibabies", "data/reports-spec.yml"),
-                packagename="nibabies",
             )
             write_derivative_description(config.execution.bids_dir, config.execution.nibabies_dir)
             write_bidsignore(config.execution.nibabies_dir)

nibabies/cli/workflow.py

@@ -55,8 +55,6 @@ def build_workflow(config_file):
             config.execution.unique_labels,
             nibabies_dir,
             config.execution.run_uuid,
-            config=pkgrf("nibabies", "data/reports-spec.yml"),
-            packagename="nibabies",
         )
         return retval
 

nibabies/reports/core.py

@@ -1,74 +1,16 @@
-from itertools import product
 from pathlib import Path
 
-from niworkflows.reports.core import Report as _Report
-from pkg_resources import resource_filename as pkgrf
+from nireports.assembler.report import Report
 
-
-class Report(_Report):
-    # niworkflows patch to preserve `out_filename` even if subject_id is present
-    def __init__(
-        self,
-        out_dir,
-        run_uuid,
-        config=None,
-        out_filename=None,
-        packagename=None,
-        reportlets_dir=None,
-        subject_id=None,
-    ):
-        self.root = Path(reportlets_dir or out_dir)
-
-        # Initialize structuring elements
-        self.sections = []
-        self.errors = []
-        self.out_dir = Path(out_dir)
-        self.run_uuid = run_uuid
-        self.packagename = packagename
-        self.subject_id = subject_id
-        if subject_id is not None:
-            self.subject_id = subject_id[4:] if subject_id.startswith("sub-") else subject_id
-            # ensure set output filename is preserved
-            if not out_filename:
-                out_filename = f"sub-{self.subject_id}.html"
-
-        self.out_filename = out_filename or "report.html"
-
-        # Default template from niworkflows
-        self.template_path = Path(pkgrf("niworkflows", "reports/report.tpl"))
-        self._load_config(Path(config or pkgrf("niworkflows", "reports/default.yml")))
-        assert self.template_path.exists()
-
-    # TODO: Upstream ``Report._load_config`` to niworkflows
-    def _load_config(self, config):
-        from yaml import safe_load as load
-
-        settings = load(config.read_text())
-        self.packagename = self.packagename or settings.get("package", None)
-
-        # Removed from here: Appending self.packagename to self.root and self.out_dir
-        # In this version, pass reportlets_dir and out_dir with nibabies in the path.
-
-        if self.subject_id is not None:
-            self.root = self.root / "sub-{}".format(self.subject_id)
-
-        if "template_path" in settings:
-            self.template_path = config.parent / settings["template_path"]
-
-        self.index(settings["sections"])
-
-
-#
-# The following are the interface used directly by NiBabies
-#
+from nibabies.data import load_resource
 
 
 def run_reports(
     out_dir,
-    subject_label,
+    subject,
     run_uuid,
-    config=None,
-    out_filename='report.html',
+    session=None,
+    out_filename=None,
     reportlets_dir=None,
     packagename=None,
 ):
@@ -78,10 +20,9 @@ def run_reports(
     return Report(
         out_dir,
         run_uuid,
-        config=config,
-        out_filename=out_filename,
-        subject_id=subject_label,
-        packagename=packagename,
+        subject=subject,
+        session=session,
+        bootstrap_file=load_resource('reports-spec.yml'),
         reportlets_dir=reportlets_dir,
     ).generate_report()
 
@@ -90,7 +31,6 @@ def generate_reports(
     sub_ses_list,
     output_dir,
     run_uuid,
-    config=None,
     work_dir=None,
     packagename=None,
 ):
@@ -100,17 +40,13 @@ def generate_reports(
         reportlets_dir = Path(work_dir) / "reportlets"
 
     report_errors = []
-    for subject_label, session in sub_ses_list:
-        html_report = ''.join(
-            [f"sub-{subject_label}", f"_ses-{session}" if session else "", ".html"]
-        )
+    for subject, session in sub_ses_list:
         report_errors.append(
             run_reports(
                 output_dir,
-                subject_label,
+                subject,
                 run_uuid,
-                config=config,
-                out_filename=html_report,
+                session=session,
                 packagename=packagename,
                 reportlets_dir=reportlets_dir,
             )

pyproject.toml

@@ -22,6 +22,7 @@ dependencies = [
     'importlib_resources; python_version < "3.9"',
     "nibabel >= 5.0.0",
     "nipype >= 1.8.5",
+    "nireports @ git+https://github.com/mgxd/nireports.git@enh/report-session",
     "nitime",
     "nitransforms >= 23.0.1",
     "niworkflows >= 1.9.0",