linarphy/naroo_reader
1
0
Fork 0
Code Issues Pull requests Projects Releases Packages Wiki Activity Actions

Update to check if file exist and other fix

Browse source
This commit is contained in:
linarphy 2023-07-18 15:11:49 +02:00
parent 98efc87ee4
commit b855e870b7
No known key found for this signature in database
GPG key ID: 3D4AAAC3AD16E79C
1 changed files with 26 additions and 17 deletions
Show stats Download patch file Download diff file Expand all files Collapse all files

41
ETA.py
View file

@ -10,7 +10,8 @@ from scipy.signal import convolve as sp_convolve
from scipy.signal import find_peaks
from scipy.ndimage import rotate
cache , filename , output , calibration , verbose , no_cache = '' , None , None , None , False , False
cache , filename , output , calibration = None , None , None , None
verbose , no_cache = False , False
if len( sys.argv ) < 2:
raise Exception( 'ETA.py: type \'ETA.py -h\' for more information' )
@ -96,21 +97,33 @@ if verbose:
\n --verbose: True ( default to False)\
\n\
\n===========================================' )
# TODO: check in advance file to check if exists or writeable
files = {}
if calibration != None:
files[ 'wavelength' ] = pathlib.Path( calibration )
if cache != None:
files[ 'cache' ] = pathlib.Path( cache )
files[ 'filename' ] = pathlib.Path( filename )
for name , path in files.items():
if name in [
'wavelength',
] and not path.is_file():
raise Exception(
'ETA.py: could not open the ' + name + ' file'
)
data = utils.load( filename )
if verbose:
print( 'data loaded' )
cache_file = pathlib.Path( cache )
if cache_file.is_file() and not no_cache:
if 'cache' in files and files[ 'cache' ].is_file() and not no_cache:
if verbose:
print( 'using cache' )
with shelve.open( str( cache_file ) ) as cache:
with shelve.open( str( files[ 'cache' ] ) ) as cache:
for key in [ 'data' , 'border' , 'calibrations' ]:
if key not in cache:
raise Exception( 'ETA.py: missing data in cache_file' )
raise Exception( 'ETA.py: missing data in cache file' )
data = cache[ 'data' ]
border = cache[ 'border' ]
calibrations = cache[ 'calibrations' ]
@ -444,10 +457,10 @@ else:
if verbose:
print( 'main ETA curved' )
if not cache_file.exists() and not no_cache:
if 'cache' in files and not files[ 'cache' ].exists() and not no_cache:
if verbose:
print( 'writing result in cache' )
with shelve.open( str( cache_file ) ) as cache:
with shelve.open( str( files[ 'cache' ] ) ) as cache:
cache[ 'data' ] = data
cache[ 'border' ] = border
cache[ 'calibrations'] = calibrations
@ -588,8 +601,8 @@ if verbose:
bias = {
'top': np.mean(
data[
calibrations[ 'top' ][ 'y' ][ 'min' ] :
calibrations[ 'top' ][ 'y' ][ 'min' ] - 500,
calibrations[ 'top' ][ 'y' ][ 'min' ] - 100:
calibrations[ 'top' ][ 'y' ][ 'min' ] ,
calibrations[ 'top' ][ 'x' ][ 'min' ] :
calibrations[ 'top' ][ 'x' ][ 'max' ]
] ,
@ -609,21 +622,17 @@ bias = {
mean_bias = sp_convolve(
np.mean(
[
bias[ 'down' ] ,
bias[ 'top' ],
bias[ 'down' ],
] ,
axis = 0,
) ,
np.ones(
(
50,
) ,
) ,
'same' ,
) / 50
print( 'mean_bias ETA: ' + str( mean_bias ) )
data[
: ,
border[ 'x' ][ 'min' ] : border[ 'x' ][ 'max' ]