diff --git a/spectrum.py b/spectrum.py index 96936b1..73561d1 100755 --- a/spectrum.py +++ b/spectrum.py @@ -688,7 +688,9 @@ if intensity != None: true_intensity_value = ( intensity_stairs.shape[0] - np.polyval( index_polyval , intensity_value ) ) * step final_intensity[index] = np.exp( true_intensity_value ) + if final_intensity[ index ] > np.exp( 2.2 ): + final_intensity[ index ] = np.exp( 2.2 ) if verbose: print( 'intensity calibration finished' ) @@ -701,8 +703,6 @@ if output == None: else: if verbose: print( 'storing result in ' + output ) - plt.plot( final_intensity[ 1 : - 1 ] ) - plt.show() main_hdu = fits.PrimaryHDU( final_intensity[1:-1] ) # remove -1 main_hdu.header[ 'CRVAL1' ] = wavelengths[0] main_hdu.header[ 'CDELT1' ] = wavelengths[1] - wavelengths[0]