ENH: Resolve BIDS-URIs

Author: effigies

sdcflows/utils/tests/test_wrangler.py

@@ -46,13 +46,13 @@
                     "EchoTime": 1.2,
                     "PhaseEncodingDirection": "j-",
                     "TotalReadoutTime": 0.8,
-                    "IntendedFor": "ses-02/func/sub-01_ses-02_task-rest_bold.nii.gz"
+                    "IntendedFor": "bids::sub-01/ses-02/func/sub-01_ses-02_task-rest_bold.nii.gz"
                 }},
                 {"suffix": "epi", "dir": "PA", "metadata": {
                     "EchoTime": 1.2,
                     "PhaseEncodingDirection": "j",
                     "TotalReadoutTime": 0.8,
-                    "IntendedFor": "ses-02/func/sub-01_ses-02_task-rest_bold.nii.gz"
+                    "IntendedFor": "bids::sub-01/ses-02/func/sub-01_ses-02_task-rest_bold.nii.gz"
                 }}
             ],
             "func": [
@@ -75,13 +75,13 @@
                     "EchoTime": 1.2,
                     "PhaseEncodingDirection": "j-",
                     "TotalReadoutTime": 0.8,
-                    "IntendedFor": "ses-03/func/sub-01_ses-03_task-rest_bold.nii.gz"
+                    "IntendedFor": "bids::sub-01/ses-03/func/sub-01_ses-03_task-rest_bold.nii.gz"
                 }},
                 {"suffix": "epi", "dir": "PA", "metadata": {
                     "EchoTime": 1.2,
                     "PhaseEncodingDirection": "j",
                     "TotalReadoutTime": 0.8,
-                    "IntendedFor": "ses-03/func/sub-01_ses-03_task-rest_bold.nii.gz"
+                    "IntendedFor": "bids::sub-01/ses-03/func/sub-01_ses-03_task-rest_bold.nii.gz"
                 }}
             ],
             "func": [
@@ -138,7 +138,7 @@
                     "metadata": {
                         "EchoTime1": 1.2,
                         "EchoTime2": 1.4,
-                        "IntendedFor": "ses-02/func/sub-01_ses-02_task-rest_bold.nii.gz"
+                        "IntendedFor": "bids::sub-01/ses-02/func/sub-01_ses-02_task-rest_bold.nii.gz"
                     }
                 },
                 {"suffix": "magnitude1", "metadata": {"EchoTime": 1.2}},
@@ -165,7 +165,7 @@
                     "metadata": {
                         "EchoTime1": 1.2,
                         "EchoTime2": 1.4,
-                        "IntendedFor": "ses-03/func/sub-01_ses-03_task-rest_bold.nii.gz"
+                        "IntendedFor": "bids::sub-01/ses-03/func/sub-01_ses-03_task-rest_bold.nii.gz"
                     }
                 },
                 {"suffix": "magnitude1", "metadata": {"EchoTime": 1.2}},
@@ -216,3 +216,16 @@ def test_wrangler_filter(tmpdir, name, skeleton, estimations):
     est = find_estimators(layout=layout, subject='01', bids_filters=filters['fmap'])
     assert len(est) == estimations
     clear_registry()
+
+
+@pytest.mark.parametrize('name,skeleton,estimations', [
+    ('pepolar', pepolar, 3),
+    ('phasediff', phasediff, 3),
+])
+def test_wrangler_URIs(tmpdir, name, skeleton, estimations):
+    bids_dir = str(tmpdir / name)
+    generate_bids_skeleton(bids_dir, skeleton)
+    layout = gen_layout(bids_dir)
+    est = find_estimators(layout=layout, subject='01')
+    assert len(est) == estimations
+    clear_registry()

sdcflows/utils/wrangler.py

@@ -33,6 +33,19 @@
 from .. import fieldmaps as fm
 
 
+def _resolve_intent(
+    intent: str,
+    layout: BIDSLayout,
+    subject: str
+) -> str | None:
+    root = Path(layout.root)
+    if intent.startswith("bids::"):
+        return str(root / intent[6:])
+    if not intent.startswith("bids:"):
+        return str(root / f"sub-{subject}" / intent)
+    return intent
+
+
 def find_estimators(
     *,
     layout: BIDSLayout,
@@ -404,7 +417,7 @@ def find_estimators(
             # Find existing IntendedFor targets and warn if missing
             all_targets = []
             for intent in listify(epi_base_md["IntendedFor"]):
-                target = layout.get_file(str(subject_root / intent))
+                target = layout.get_file(_resolve_intent(intent, layout, subject))
                 if target is None:
                     logger.debug("Single PE target %s not found", intent)
                     continue