feat: Add workflow arguments for metadata estimates and fallback TRT

Author: effigies

sdcflows/workflows/apply/correction.py

@@ -29,6 +29,8 @@
 def init_unwarp_wf(
     *,
     jacobian=True,
+    use_metadata_estimates=False,
+    fallback_total_readout_time=None,
     free_mem=None,
     omp_nthreads=1,
     debug=False,
@@ -117,7 +119,14 @@ def init_unwarp_wf(
         name="outputnode",
     )
 
-    rotime = pe.Node(GetReadoutTime(), name="rotime")
+    rotime = pe.Node(
+        GetReadoutTime(
+            use_estimate=use_metadata_estimates,
+            fallback=fallback_total_readout_time,
+        ),
+        name="rotime",
+        run_without_submitting=True,
+    )
     rotime.interface._always_run = debug
 
     # resample is memory-hungry; choose a smaller number of threads

sdcflows/workflows/fit/pepolar.py

@@ -36,6 +36,8 @@
 
 def init_topup_wf(
     grid_reference=0,
+    use_metadata_estimates=False,
+    fallback_total_readout_time=None,
     omp_nthreads=1,
     sloppy=False,
     debug=False,
@@ -122,7 +124,10 @@ def init_topup_wf(
 
     # Calculate the total readout time of each run
     readout_time = pe.MapNode(
-        GetReadoutTime(),
+        GetReadoutTime(
+            use_estimate=use_metadata_estimates,
+            fallback=fallback_total_readout_time,
+        ),
         name="readout_time",
         iterfield=["metadata", "in_file"],
         run_without_submitting=True,

sdcflows/workflows/fit/syn.py

@@ -42,6 +42,8 @@
 def init_syn_sdc_wf(
     *,
     atlas_threshold=3,
+    use_metadata_estimates=False,
+    fallback_total_readout_time=None,
     sloppy=False,
     debug=False,
     name="syn_sdc_wf",
@@ -160,7 +162,10 @@ def init_syn_sdc_wf(
     outputnode.inputs.method = 'FLB ("fieldmap-less", SyN-based)'
 
     readout_time = pe.Node(
-        GetReadoutTime(),
+        GetReadoutTime(
+            use_estimate=use_metadata_estimates,
+            fallback=fallback_total_readout_time,
+        ),
         name="readout_time",
         run_without_submitting=True,
     )