From e72d5ca7e3e323db65b3642908c8e57efc01d6d3 Mon Sep 17 00:00:00 2001
From: Casper da Costa-Luis <casper.dcl@physics.org>
Date: Wed, 31 Jul 2024 18:16:17 +0100
Subject: [PATCH] expose data.path

- fixes #77
---
 petric.py | 5 +++--
 1 file changed, 3 insertions(+), 2 deletions(-)

diff --git a/petric.py b/petric.py
index 1393a39..786d2e3 100755
--- 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
 from traceback import print_exc
 
@@ -199,6 +199,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):
@@ -236,7 +237,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():