def process_object(filename, arc, suffix="linearized"):
"""
Helper recipe to reduce the object file.
Returns
-------
AstroData
Processed arc.
"""
from astrodata.testing import download_from_archive
from geminidr.gmos.primitives_gmos_longslit import GMOSLongslit
processed_filename, ext = os.path.splitext(filename)
processed_filename += "_{:s}{:s}".format(suffix, ext)
if os.path.exists(processed_filename):
ad = astrodata.open(processed_filename)
else:
if os.path.exists(filename):
ad = astrodata.open(filename)
else:
ad = astrodata.open(
download_from_archive(filename,
path='',
env_var='DRAGONS_TEST'))
p = GMOSLongslit([ad])
p.prepare()
p.addDQ(static_bpm=None)
p.addVAR(read_noise=True)
p.overscanCorrect()
# p.biasCorrect()
p.ADUToElectrons()
p.addVAR(poisson_noise=True)
# p.flatCorrect()
# p.applyQECorrection()
p.distortionCorrect(arc=arc)
p.findSourceApertures(max_apertures=1)
p.skyCorrectFromSlit()
p.traceApertures()
p.extract1DSpectra()
p.linearizeSpectra() # TODO: needed?
p.calculateSensitivity()
ad = p.streams['main'][0]
ad.write(overwrite=True)
return ad
def test_correlation_and_trim(adinputs, caplog):
add_fake_offset(adinputs, offset=10)
p = GMOSLongslit(adinputs)
adout = p.adjustWCSToReference()
assert adout[1].phu['SLITOFF'] == -10
assert adout[2].phu['SLITOFF'] == -20
p.resampleToCommonFrame(dw=0.15, trim_data=True)
_check_params(caplog.records, 'w1=508.198 w2=978.802 dw=0.150 npix=3139')
ad = p.stackFrames()[0]
assert ad[0].shape == (512, 3139)
caplog.clear()
ad = p.findSourceApertures(max_apertures=1)[0]
assert len(ad[0].APERTURE) == 1
np.testing.assert_allclose(ad[0].APERTURE['c0'], 260.8, atol=0.25)
ad = p.extract1DSpectra()[0]
assert ad[0].shape == (3139, )
def test_correlation(adinputs, caplog):
add_fake_offset(adinputs, offset=10)
p = GMOSLongslit(adinputs)
adout = p.adjustWCSToReference()
assert adout[1].phu['SLITOFF'] == -10
assert adout[2].phu['SLITOFF'] == -20
p.resampleToCommonFrame(dw=0.15)
_check_params(caplog.records, 'w1=508.198 w2=1088.323 dw=0.150 npix=3869')
ad = p.stackFrames()[0]
assert ad[0].shape == (512, 3869)
caplog.clear()
ad = p.findSourceApertures(max_apertures=1)[0]
assert len(ad[0].APERTURE) == 1
#assert caplog.records[3].message == 'Found sources at rows: 260.8'
np.testing.assert_allclose(ad[0].APERTURE['c0'], 260.8, atol=0.25)
ad = p.extract1DSpectra()[0]
assert ad[0].shape == (3869, )
def create_inputs_recipe():
"""
Creates input data for tests using pre-processed standard star and its
calibration files.
The raw files will be downloaded and saved inside the path stored in the
`$DRAGONS_TEST/raw_inputs` directory. Processed files will be stored inside
a new folder called "dragons_test_inputs". The sub-directory structure
should reflect the one returned by the `path_to_inputs` fixture.
"""
import os
from astrodata.testing import download_from_archive
from recipe_system.reduction.coreReduce import Reduce
from gempy.utils import logutils
associated_calibrations = {
"S20190808S0048.fits": {
"arcs": ["S20190808S0167.fits"]
},
"S20190808S0049.fits": {
"arcs": ["S20190808S0168.fits"]
},
# "S20190808S0052.fits": {"arcs": ["S20190808S0165.fits"]}, # Can't find aperture
"S20190808S0053.fits": {
"arcs": ["S20190808S0169.fits"]
},
}
root_path = os.path.join("./dragons_test_inputs/")
module_path = "geminidr/gmos/spect/{}".format(__file__.split('.')[0])
path = os.path.join(root_path, module_path)
os.makedirs(path, exist_ok=True)
os.chdir(path)
os.makedirs("./inputs", exist_ok=True)
print('Current working directory:\n {:s}'.format(os.getcwd()))
for filename, cals in associated_calibrations.items():
print('Downloading files...')
sci_path = download_from_archive(filename)
arc_path = [download_from_archive(f) for f in cals['arcs']]
sci_ad = astrodata.open(sci_path)
data_label = sci_ad.data_label()
print('Reducing ARC for {:s}'.format(data_label))
logutils.config(file_name='log_arc_{}.txt'.format(data_label))
arc_reduce = Reduce()
arc_reduce.files.extend(arc_path)
arc_reduce.runr()
arc = arc_reduce.output_filenames.pop()
print('Reducing pre-processed data:')
logutils.config(file_name='log_{}.txt'.format(data_label))
p = GMOSLongslit([sci_ad])
p.prepare()
p.addDQ(static_bpm=None)
p.addVAR(read_noise=True)
p.overscanCorrect()
p.ADUToElectrons()
p.addVAR(poisson_noise=True)
p.mosaicDetectors()
p.distortionCorrect(arc=arc)
p.findSourceApertures(max_apertures=1)
p.skyCorrectFromSlit()
p.traceApertures()
p.extract1DSpectra()
os.chdir("inputs/")
_ = p.writeOutputs().pop()
os.chdir("../")
def create_inputs():
"""
Create inputs for `test_plot_spectra_for_qa_single_frame`.
The raw files will be downloaded and saved inside the path stored in the
`$DRAGONS_TEST/raw_inputs` directory. Processed files will be stored inside
a new folder called "dragons_test_inputs". The sub-directory structure
should reflect the one returned by the `path_to_inputs` fixture.
"""
import glob
import os
from geminidr.gmos.primitives_gmos_longslit import GMOSLongslit
from gempy.utils import logutils
from recipe_system.reduction.coreReduce import Reduce
from recipe_system.utils.reduce_utils import normalize_ucals
cwd = os.getcwd()
path = f"./dragons_test_inputs/geminidr/core/{__file__.split('.')[0]}/"
os.makedirs(path, exist_ok=True)
os.chdir(path)
os.makedirs("inputs/", exist_ok=True)
for raw_list, bias_list, quartz_list, arc_list in single_aperture_data:
if all([
os.path.exists(f"inputs/{s.split('.')[0]}_extracted.fits")
for s in raw_list
]):
print("Skipping already created input.")
continue
raw_paths = [download_from_archive(f) for f in raw_list]
bias_paths = [download_from_archive(f) for f in bias_list]
quartz_paths = [download_from_archive(f) for f in quartz_list]
arc_paths = [download_from_archive(f) for f in arc_list]
cals = []
raw_ads = [astrodata.open(p) for p in raw_paths]
data_label = raw_ads[0].data_label()
print('Current working directory:\n {:s}'.format(os.getcwd()))
if len(bias_paths):
logutils.config(file_name='log_bias_{}.txt'.format(data_label))
r = Reduce()
r.files.extend(bias_paths)
r.runr()
master_bias = r.output_filenames.pop()
cals.append(f"processed_bias:{master_bias}")
del r
else:
master_bias = None
if len(quartz_paths):
logutils.config(file_name='log_quartz_{}.txt'.format(data_label))
r = Reduce()
r.files.extend(quartz_paths)
r.ucals = normalize_ucals(r.files, cals)
r.runr()
master_quartz = r.output_filenames.pop()
cals.append(f"processed_flat:{master_quartz}")
del r
else:
master_quartz = None
logutils.config(file_name='log_arc_{}.txt'.format(data_label))
r = Reduce()
r.files.extend(arc_paths)
r.ucals = normalize_ucals(r.files, cals)
r.runr()
master_arc = r.output_filenames.pop()
logutils.config(file_name='log_{}.txt'.format(data_label))
p = GMOSLongslit(raw_ads)
p.prepare()
p.addDQ(static_bpm=None)
p.addVAR(read_noise=True)
p.overscanCorrect()
p.biasCorrect(do_bias=master_bias is not None, bias=master_bias)
p.ADUToElectrons()
p.addVAR(poisson_noise=True)
p.flatCorrect(do_flat=master_quartz is not None, flat=master_quartz)
p.QECorrect(arc=master_arc)
p.distortionCorrect(arc=master_arc)
p.findSourceApertures(max_apertures=3)
p.skyCorrectFromSlit()
p.traceApertures()
p.extract1DSpectra()
p.linearizeSpectra()
[os.remove(s) for s in glob.glob("*_arc.fits")]
[os.remove(s) for s in glob.glob("*_bias.fits")]
[os.remove(s) for s in glob.glob("*_flat.fits")]
[os.remove(s) for s in glob.glob("*_mosaic.fits")]
os.chdir("inputs/")
print("\n\n Writing processed files for tests into:\n"
" {:s}\n\n".format(os.getcwd()))
_ = p.writeOutputs()
os.chdir("../")
os.chdir(cwd)