From f6970dce179f612761c00ae858f3362d57d4ff2b Mon Sep 17 00:00:00 2001
From: linarphy <linarphy@linarphy.net>
Date: Fri, 16 Jun 2023 17:42:47 +0200
Subject: [PATCH] Fix output

---
 ETA.py | 10 ++++++++--
 1 file changed, 8 insertions(+), 2 deletions(-)

diff --git a/ETA.py b/ETA.py
index cd239c9..d62478b 100644
--- a/ETA.py
+++ b/ETA.py
@@ -616,7 +616,13 @@ for x in range( size_x ):
             ] ),
             fall[i]
         ]
-    stairs[ -1 , x ] = [ 0 , fall[ - 1 ] ]
+    stairs[ -1 , x ] = [
+        np.mean( data[
+            calibrations[ 'top' ][ 'y' ][ 'max' ] + fall[ -1 ] : calibrations[ 'down' ][ 'y' ][ 'min' ],
+            border[ 'x' ][ 'min' ] + x
+        ] ),
+        fall[ - 1 ]
+    ]
 
 if verbose:
     print( 'ETA flattened' )
@@ -628,7 +634,7 @@ else:
 
     output_file = pathlib.Path( output )
     with shelve.open( str( output_file ) ) as output:
-        output[ 'data'   ]     = data
+        output[ 'data'   ]     = stairs
         output[ 'wavelength' ] = wavelength
 
 if verbose: