From e1ba5dc6abf3d9e684b618e1680dbedcee898e96 Mon Sep 17 00:00:00 2001
From: Tamino Huxohl <thuxohl@techfak.uni-bielefeld.de>
Date: Wed, 4 Jan 2023 12:50:54 +0100
Subject: [PATCH] discarding mu_map slices can now be used to discard same kind
 of slices from reconstructions

---
 mu_map/data/review_mu_map.py | 12 +++++++++++-
 1 file changed, 11 insertions(+), 1 deletion(-)

diff --git a/mu_map/data/review_mu_map.py b/mu_map/data/review_mu_map.py
index 1bf3771..3f12762 100644
--- a/mu_map/data/review_mu_map.py
+++ b/mu_map/data/review_mu_map.py
@@ -1,3 +1,5 @@
+from typing import Optional
+
 import numpy as np
 import pandas as pd
 
@@ -6,22 +8,30 @@ HEADER_DISC_FIRST = "discard_first"
 HEADER_DISC_LAST = "discard_last"
 
 
-def discard_slices(row: pd.Series, μ_map: np.ndarray) -> np.ndarray:
+def discard_slices(row: pd.Series, μ_map: np.ndarray, recon: Optional[np.ndarray] = None) -> np.ndarray:
     """
     Discard slices based on the flags in the row of th according table.
     The row is expected to contain the flags 'discard_first' and 'discard_last'.
 
     :param row: the row of meta configuration file of a dataset
     :param μ_map: the μ_map
+    :param recon: optional reconstruction of which the same slice is discarded so that the alignment stays the same
     :return: the μ_map with according slices removed
     """
     _res = μ_map
 
     if row[HEADER_DISC_FIRST]:
         _res = _res[1:]
+        if recon is not None:
+            recon = recon[1:]
 
     if row[HEADER_DISC_LAST]:
         _res = _res[:-1]
+        if recon is not None:
+            recon = recon[:-1]
+
+    if recon is not None:
+        return _res, recon
 
     return _res
 
-- 
GitLab