From b34c1a68eb294ef1831d8b78fc0de9efa3c6f29c Mon Sep 17 00:00:00 2001
From: fgbossema <38940120+fgbossema@users.noreply.github.com>
Date: Mon, 26 Oct 2020 10:48:58 +0100
Subject: [PATCH] if rnorms are tracked using SIRT, return rnorms

---
 flextomo/projector.py | 3 ++-
 1 file changed, 2 insertions(+), 1 deletion(-)

diff --git a/flextomo/projector.py b/flextomo/projector.py
index 733ee8c..5761efa 100644
--- a/flextomo/projector.py
+++ b/flextomo/projector.py
@@ -120,6 +120,7 @@ def SIRT( projections, volume, geometry, iterations):
                      
     if rnorms:   
          display.plot2d(rnorms, semilogy = True, title = 'Resudual L2')   
+         return rnorms
          
 def PWLS(projections, volume, geometry, iterations):
     """
@@ -918,4 +919,4 @@ def _subset_generator_(projections, volume, geometry, copy = False):
             proj_geom = geometry[jj].astra_projection_geom(projections[jj].shape, index = sl)
             vol_geom = geometry[jj].astra_volume_geom(volume.shape)
      
-            yield subset, proj_geom, vol_geom 
\ No newline at end of file
+            yield subset, proj_geom, vol_geom