From 4e558ed9bfe8e96d4e3b64d2ff0a2354d7cb5383 Mon Sep 17 00:00:00 2001 From: Gabriel Brammer Date: Tue, 8 Oct 2024 11:06:53 +0200 Subject: [PATCH] write var image if available --- grizli/aws/aws_drizzler.py | 45 ++++++++++++++++++++++++++++---------- 1 file changed, 34 insertions(+), 11 deletions(-) diff --git a/grizli/aws/aws_drizzler.py b/grizli/aws/aws_drizzler.py index 3479ea20..36d8ede4 100755 --- a/grizli/aws/aws_drizzler.py +++ b/grizli/aws/aws_drizzler.py @@ -531,7 +531,7 @@ def segmentation_figure(label, cat, segfile): th.close() -def drizzle_images(label='macs0647-jd1', ra=101.9822125, dec=70.24326667, pixscale=0.1, size=10, wcs=None, pixfrac=0.33, kernel='square', theta=0, half_optical_pixscale=True, filters=['f160w', 'f140w', 'f125w', 'f105w', 'f110w', 'f098m', 'f850lp', 'f814w', 'f775w', 'f606w', 'f475w', 'f555w', 'f600lp', 'f390w', 'f350lp'], skip=None, remove=True, rgb_params=RGB_PARAMS, master='grizli-v1-19.12.04', aws_bucket='s3://grizli/CutoutProducts/', scale_ab=21, thumb_height=2.0, sync_fits=True, subtract_median=True, include_saturated=True, include_ir_psf=False, oversample_psf=False, show_filters=['visb', 'visr', 'y', 'j', 'h'], combine_similar_filters=True, single_output=True, aws_prep_dir=None, make_segmentation_figure=False, get_dict=False, dryrun=False, thumbnail_ext='png', write_ctx=False, **kwargs): +def drizzle_images(label='macs0647-jd1', ra=101.9822125, dec=70.24326667, pixscale=0.1, size=10, wcs=None, pixfrac=0.33, kernel='square', theta=0, half_optical_pixscale=True, filters=['f160w', 'f140w', 'f125w', 'f105w', 'f110w', 'f098m', 'f850lp', 'f814w', 'f775w', 'f606w', 'f475w', 'f555w', 'f600lp', 'f390w', 'f350lp'], weight_type="jwst_var", skip=None, remove=True, rgb_params=RGB_PARAMS, master='grizli-v1-19.12.04', aws_bucket='s3://grizli/CutoutProducts/', scale_ab=21, thumb_height=2.0, sync_fits=True, subtract_median=True, include_saturated=True, include_ir_psf=False, oversample_psf=False, show_filters=['visb', 'visr', 'y', 'j', 'h'], combine_similar_filters=True, single_output=True, aws_prep_dir=None, make_segmentation_figure=False, get_dict=False, dryrun=False, thumbnail_ext='png', write_ctx=False, **kwargs): """ label='cp561356'; ra=150.208875; dec=1.850241667; size=40; filters=['f160w','f814w', 'f140w','f125w','f105w','f606w','f475w'] @@ -789,7 +789,7 @@ def drizzle_images(label='macs0647-jd1', ra=101.9822125, dec=70.24326667, pixsca kernel=kernel, clean=clean_i, include_saturated=include_saturated, - weight_type='median_err', + weight_type=weight_type, skip=skip, dryrun=dryrun ) @@ -818,17 +818,40 @@ def drizzle_images(label='macs0647-jd1', ra=101.9822125, dec=70.24326667, pixsca med = 0. outh['IMGMED'] = (0., 'Median subtracted from the image') - pyfits.writeto('{0}-{1}_drz_sci.fits'.format(label, filt), - data=sci, header=outh, overwrite=True, - output_verify='fix') + # Write files + pyfits.writeto( + '{0}-{1}_drz_sci.fits'.format(label, filt), + data=sci, + header=outh, + overwrite=True, + output_verify='fix' + ) + + pyfits.writeto( + '{0}-{1}_drz_wht.fits'.format(label, filt), + data=wht, + header=outh, + overwrite=True, + output_verify='fix' + ) + + if var is not None: + pyfits.writeto( + '{0}-{1}_drz_var.fits'.format(label, filt), + data=var, + header=outh, + overwrite=True, + output_verify='fix' + ) - pyfits.writeto('{0}-{1}_drz_wht.fits'.format(label, filt), - data=wht, header=outh, overwrite=True, - output_verify='fix') if write_ctx: - pyfits.writeto('{0}-{1}_drz_ctx.fits'.format(label, filt), - data=ctx, header=outh, overwrite=True, - output_verify='fix') + pyfits.writeto( + '{0}-{1}_drz_ctx.fits'.format(label, filt), + data=ctx, + header=outh, + overwrite=True, + output_verify='fix' + ) has_filts.append(filt)