Skip to content
Snippets Groups Projects
Commit 600bb885 authored by Jakob Roth's avatar Jakob Roth
Browse files

convert_brightness_units: remove trailing spaces

parent 5b1c93fa
No related branches found
No related tags found
No related merge requests found
Pipeline #199220 passed
...@@ -12,11 +12,11 @@ def _convert_brightness_units(inputfile, outputfile, observations=[], mode='to_p ...@@ -12,11 +12,11 @@ def _convert_brightness_units(inputfile, outputfile, observations=[], mode='to_p
# keys modified already # keys modified already
used_keys = {'BITPIX','BUNIT','CDELT1','CDELT2','CDELT3','CDELT4','CRPIX1','CRPIX2','CRPIX3','CRPIX4','CRVAL1','CRVAL2','CRVAL3','CRVAL4','CTYPE1','CTYPE2','CTYPE3','CTYPE4','CUNIT1','CUNIT2','CUNIT3','CUNIT4','DATE-MAP','NAXIS','NAXIS1','NAXIS2','NAXIS3','NAXIS4','OBSDEC','OBSRA','ORIGIN','SIMPLE'} used_keys = {'BITPIX','BUNIT','CDELT1','CDELT2','CDELT3','CDELT4','CRPIX1','CRPIX2','CRPIX3','CRPIX4','CRVAL1','CRVAL2','CRVAL3','CRVAL4','CTYPE1','CTYPE2','CTYPE3','CTYPE4','CUNIT1','CUNIT2','CUNIT3','CUNIT4','DATE-MAP','NAXIS','NAXIS1','NAXIS2','NAXIS3','NAXIS4','OBSDEC','OBSRA','ORIGIN','SIMPLE'}
# compute beam arear # compute beam arear
theta_maj = fits_file[0].header['BMAJ'] # deg theta_maj = fits_file[0].header['BMAJ'] # deg
theta_min = fits_file[0].header['BMIN'] # deg theta_min = fits_file[0].header['BMIN'] # deg
Omega = np.pi*theta_maj*theta_min / (4 * np.log(2)) # deg**2 Omega = np.pi*theta_maj*theta_min / (4 * np.log(2)) # deg**2
Omega = ((np.pi/180)**2) * Omega # sr Omega = ((np.pi/180)**2) * Omega # sr
...@@ -26,15 +26,15 @@ def _convert_brightness_units(inputfile, outputfile, observations=[], mode='to_p ...@@ -26,15 +26,15 @@ def _convert_brightness_units(inputfile, outputfile, observations=[], mode='to_p
fits_data *= Omega fits_data *= Omega
else: else:
raise ValueError(f"Invalid mode encountered: {mode}") raise ValueError(f"Invalid mode encountered: {mode}")
delt = np.abs(fits_file[0].header['CDELT1']*np.pi/180) # smallest angular size delt = np.abs(fits_file[0].header['CDELT1']*np.pi/180) # smallest angular size
sky_space = ift.RGSpace([fits_data.shape[2], fits_data.shape[3]], distances=delt) sky_space = ift.RGSpace([fits_data.shape[2], fits_data.shape[3]], distances=delt)
sky_domain = rve.default_sky_domain(sdom=sky_space) sky_domain = rve.default_sky_domain(sdom=sky_space)
data = np.zeros(sky_domain.shape) data = np.zeros(sky_domain.shape)
data[0,0,0,:,:] = fits_data[0,0,:,:].T data[0,0,0,:,:] = fits_data[0,0,:,:].T
data = ift.makeField(sky_domain, data) data = ift.makeField(sky_domain, data)
header_override = fits_file[0].header header_override = fits_file[0].header
...@@ -43,9 +43,9 @@ def _convert_brightness_units(inputfile, outputfile, observations=[], mode='to_p ...@@ -43,9 +43,9 @@ def _convert_brightness_units(inputfile, outputfile, observations=[], mode='to_p
if mode == 'to_per_beam': if mode == 'to_per_beam':
header_override['BUNIT'] = 'Jy/beam' header_override['BUNIT'] = 'Jy/beam'
rve.ubik_tools.field2fits(data, outputfile, observations=observations, header_override=header_override) rve.ubik_tools.field2fits(data, outputfile, observations=observations, header_override=header_override)
if __name__ == "__main__": if __name__ == "__main__":
if len(sys.argv) < 3 or len(sys.argv) > 4: if len(sys.argv) < 3 or len(sys.argv) > 4:
...@@ -56,9 +56,9 @@ if __name__ == "__main__": ...@@ -56,9 +56,9 @@ if __name__ == "__main__":
mode = sys.argv[3] mode = sys.argv[3]
else: else:
mode = 'to_per_str' mode = 'to_per_str'
inputfile = sys.argv[1] inputfile = sys.argv[1]
outputfile = sys.argv[2] outputfile = sys.argv[2]
_convert_brightness_units(inputfile, outputfile, mode=mode) _convert_brightness_units(inputfile, outputfile, mode=mode)
0% Loading or .
You are about to add 0 people to the discussion. Proceed with caution.
Please register or to comment