Fix output

This commit is contained in:
linarphy 2023-06-16 17:42:47 +02:00
parent 0f33fb8072
commit f6970dce17
No known key found for this signature in database
GPG key ID: 3D4AAAC3AD16E79C

10
ETA.py
View file

@ -616,7 +616,13 @@ for x in range( size_x ):
] ), ] ),
fall[i] 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: if verbose:
print( 'ETA flattened' ) print( 'ETA flattened' )
@ -628,7 +634,7 @@ else:
output_file = pathlib.Path( output ) output_file = pathlib.Path( output )
with shelve.open( str( output_file ) ) as output: with shelve.open( str( output_file ) ) as output:
output[ 'data' ] = data output[ 'data' ] = stairs
output[ 'wavelength' ] = wavelength output[ 'wavelength' ] = wavelength
if verbose: if verbose: