Merge pull request #80 from SyneRBI/data-path

expose data.path
SyneRBI · Jul 31, 2024 · 730325b · 730325b
2 parents b1dd679 + e72d5ca
commit 730325b
Showing 1 changed file with 3 additions and 2 deletions.
diff --git a/petric.py b/petric.py
@@ -18,7 +18,7 @@
 import logging
 import os
 from dataclasses import dataclass
-from pathlib import Path
+from pathlib import Path, PurePath
 from time import time
 
 import numpy as np
@@ -198,6 +198,7 @@ class Dataset:
     whole_object_mask: STIR.ImageData | None
     background_mask: STIR.ImageData | None
     voi_masks: dict[str, STIR.ImageData]
+    path: PurePath
 
 
 def get_data(srcdir=".", outdir=OUTDIR, sirf_verbosity=0):
@@ -235,7 +236,7 @@ def get_image(fname):
         for voi in (srcdir / 'PETRIC').glob("VOI_*.hv") if voi.stem[4:] not in ('background', 'whole_object')}
 
     return Dataset(acquired_data, additive_term, mult_factors, OSEM_image, prior, kappa, reference_image,
-                   whole_object_mask, background_mask, voi_masks)
+                   whole_object_mask, background_mask, voi_masks, srcdir.resolve())
 
 
 if SRCDIR.is_dir():