class POLLUX_IMAGE(SYOTool):
tool_prefix = "pollux"
save_models = [
"telescope", "camera", "spectrograph", "spectropolarimeter", "exposure"
]
save_params = {
"redshift": None, #slider value
"renorm_magnitude": None, #slider value
"exptime": None, #slider value
"grating": ("spectropolarimeter", "mode"), #drop-down selection
"aperture": ("telescope", "aperture"), #slider value
"spectrum_type": ("exposure", "sed_id"), #drop-down selection
"user_prefix": None
}
save_dir = os.path.join(os.environ['LUVOIR_SIMTOOLS_DIR'], 'saves')
#must include this to set defaults before the interface is constructed
tool_defaults = {
'redshift': pre_encode(0.25 * u.dimensionless_unscaled),
'renorm_magnitude': pre_encode(21.0 * u.mag('AB')),
'exptime': pre_encode(1.0 * u.hour),
'grating': "NUV_POL",
'aperture': pre_encode(15.0 * u.m),
'spectrum_type': 'qso'
}
def __init__(self):
print('Initializing the parameters ...')
self.grating = grating_pollux
self.aperture = ap_pollux
self.exptime = expt_pollux
self.redshift = pre_encode(red_pollux[1]['value'])
self.renorm_magnitude = mag_pollux
self.spectrum_type = spectrum_pollux
self.tool_preinit()
self.photon_count = photon_count
self.EM_gain = gain
self.show_plot = show_plot
self.file_name = file_name
if snr_pollux == 1:
self.make_image = False
self.update_exposure()
self.plot_snr()
print('Saving the file at', basedir, '/pollux_tool/files/')
np.savetxt(
basedir + '/pollux_tool/files/' + self.file_name + '_' +
self.grating + '_snr_data.txt',
np.array((self.background_wave, self._snr)).T)
if image_pollux == 1:
self.make_image = True
self.update_exposure()
self.plot_image()
print('Saving the file at', basedir, '/pollux_tool/files/')
np.savetxt(
basedir + '/pollux_tool/files/' + self.file_name + '_' +
self.grating + '_2dimage_data.txt', self._final_image)
np.savetxt(
basedir + '/pollux_tool/files/' + self.file_name + '_' +
self.grating + '_2dimage_wavelength.txt', self._wave_image)
def tool_preinit(self):
"""
Pre-initialize any required attributes for the interface.
"""
#initialize engine objects
self.telescope = Telescope(temperature=pre_encode(280.0 * u.K))
self.spectrograph = Spectropolarimeter()
self.exposure = Exposure()
self.telescope.add_spectrograph(self.spectrograph)
self.spectrograph.add_exposure(self.exposure)
#set interface variables
tool_postinit = None
def update_exposure(self):
"""
Update the exposure's parameters and recalculate everything.
"""
#We turn off calculation at the beginning so we can update everything
#at once without recalculating every time we change something
#Update all the parameters
self.telescope.aperture = self.aperture
self.spectrograph.mode = self.grating
self.exposure.exptime = pre_decode(self.exptime)
self.exposure.redshift = pre_decode(self.redshift)
self.exposure.sed_id = self.spectrum_type
self.exposure.renorm_sed(pre_decode(self.renorm_magnitude),
bandpass='******')
#Now we turn calculations back on and recalculate
self.exposure.enable(make_image=self.make_image,
photon_count=self.photon_count,
gain=self.EM_gain)
#Set the spectrum template
self.spectrum_template = pre_decode(self.exposure.sed)
def plot_snr(self):
"""
"""
plt.figure(1)
plt.subplot(211)
plt.plot(self.template_wave, self.template_flux, '-b', label='source')
plt.plot(self.background_wave,
self.background_flux,
'--k',
label='background')
plt.legend(loc='upper right', fontsize='large', frameon=False)
plt.xlim(900, 4000.)
#plt.xlabel('Wavelangth [A]', fontsize='x-large')
plt.ylabel('Flux [erg/s/cm^2/A]', fontsize='x-large')
plt.subplot(212)
plt.plot(self.background_wave, self._snr, '-b')
plt.xlim(900, 4000.)
plt.xlabel('Wavelangth [A]', fontsize='x-large')
plt.ylabel('S/N per resel', fontsize='x-large')
if show_plot == 1:
print('Saving the image at', basedir, '/pollux_tool/plots/')
plt.savefig(basedir + '/pollux_tool/plots/' + self.file_name +
'_' + self.grating + '_snr_plot.png',
dpi=300)
plt.show()
def plot_image(self):
"""
"""
#if self.grating[0] == 'F':
#plt.imshow(np.log10(self._final_image), origin='lower left', cmap='hot', interpolation='none',aspect='auto')
#else:
plt.imshow(self._final_image,
origin='lower left',
cmap='hot',
interpolation='none',
aspect='auto')
plt.colorbar()
if show_plot == 1:
print('Saving the image at', basedir, '/pollux_tool/plots/')
plt.savefig(basedir + '/pollux_tool/plots/' + self.file_name +
'_' + self.grating + '_2dimage_plot.png',
dpi=300)
plt.show()
@property
def template_wave(self):
"""
Easy SED wavelength access for the Bokeh widgets.
"""
return self.exposure.recover('sed').wave
@property
def template_flux(self):
"""
Easy SED flux access for the Bokeh widgets.
"""
sed = self.exposure.recover('sed')
wave = sed.wave * u.Unit(sed.waveunits.name)
if sed.fluxunits.name == "abmag":
funit = u.ABmag
elif sed.fluxunits.name == "photlam":
funit = u.ph / u.s / u.cm**2 / u.AA
else:
funit = u.Unit(sed.fluxunits.name)
flux = (sed.flux * funit).to(u.erg / u.s / u.cm**2 / u.AA,
equivalencies=u.spectral_density(wave))
return flux.value
@property
def background_wave(self):
"""
Easy instrument wavelength access for the Bokeh widgets.
"""
bwave = self.spectrograph.recover('wave').to(u.AA)
return bwave.value
@property
def background_flux(self):
"""
Easy instrument background flux access for the Bokeh widgets.
"""
bef = self.spectrograph.recover('bef').to(u.erg / u.s / u.cm**2 / u.AA)
return bef.value
@property
def _snr(self):
return np.nan_to_num(self.exposure.recover('snr').value)
@property
def _final_image(self):
return np.nan_to_num(self.exposure.recover('final_image'))
@property
def _wave_image(self):
return np.nan_to_num(self.exposure.recover('wave_image'))
class LUMOS_ETC(SYOTool):
tool_prefix = "lumos"
save_models = ["telescope", "camera", "spectrograph", "exposure"]
save_params = {"redshift": None, #slider value
"renorm_magnitude": None, #slider value
"exptime": None, #slider value
"grating": ("spectrograph", "mode"), #drop-down selection
"aperture": ("telescope", "aperture"), #slider value
"spectrum_type": ("exposure", "sed_id"), #drop-down selection
"user_prefix": None}
save_dir = os.path.join(os.environ['LUVOIR_SIMTOOLS_DIR'],'saves')
#must include this to set defaults before the interface is constructed
tool_defaults = {'redshift': pre_encode(0.0 * u.dimensionless_unscaled),
'renorm_magnitude': pre_encode(21.0 * u.mag('AB')),
'exptime': pre_encode(1.0 * u.hour),
'grating': "G120M",
'aperture': pre_encode(15.0 * u.m),
'spectrum_type': 'qso'}
def tool_preinit(self):
"""
Pre-initialize any required attributes for the interface.
"""
#initialize engine objects
self.telescope = Telescope(temperature=pre_encode(280.0*u.K))
self.spectrograph = Spectrograph()
self.exposure = Exposure()
self.telescope.add_spectrograph(self.spectrograph)
self.spectrograph.add_exposure(self.exposure)
#set interface variables
self.templates = ['flam', 'qso', 's99', 'o5v', 'g2v', 'g191b2b',
'gd71', 'gd153', 'ctts', 'mdwarf', 'orion', 'nodust',
'ebv6', 'hi1hei1', 'hi0hei1']
self.template_options = [SpectralLibrary[t] for t in self.templates]
self.help_text = help_text
self.gratings = self.spectrograph.modes
self.grating_options = [self.spectrograph.descriptions[g] for g in self.gratings]
self.dl_filename = ""
#set default input values
self.update_exposure()
#Formatting & interface stuff:
self.format_string = interface_format
self.interface_file = os.path.join(script_dir, "interface.yaml")
#For saving calculations
self.current_savefile = ""
self.overwrite_save = False
tool_postinit = None
def update_exposure(self):
"""
Update the exposure's parameters and recalculate everything.
"""
#We turn off calculation at the beginning so we can update everything
#at once without recalculating every time we change something
self.exposure.disable()
#Update all the parameters
self.telescope.aperture = self.aperture
self.spectrograph.mode = self.grating
self.exposure.exptime = pre_decode(self.exptime)
self.exposure.redshift = pre_decode(self.redshift)
self.exposure.sed_id = self.spectrum_type
self.exposure.renorm_sed(pre_decode(self.renorm_magnitude),
bandpass='******')
#Now we turn calculations back on and recalculate
self.exposure.enable()
#Set the spectrum template
self.spectrum_template = pre_decode(self.exposure.sed)
@property
def template_wave(self):
"""
Easy SED wavelength access for the Bokeh widgets.
"""
return self.exposure.recover('sed').wave
@property
def template_flux(self):
"""
Easy SED flux access for the Bokeh widgets.
"""
sed = self.exposure.recover('sed')
wave = sed.wave * u.Unit(sed.waveunits.name)
if sed.fluxunits.name == "abmag":
funit = u.ABmag
elif sed.fluxunits.name == "photlam":
funit = u.ph / u.s / u.cm**2 / u.AA
else:
funit = u.Unit(sed.fluxunits.name)
flux = (sed.flux * funit).to(u.erg / u.s / u.cm**2 / u.AA,
equivalencies=u.spectral_density(wave))
return flux.value
@property
def background_wave(self):
"""
Easy instrument wavelength access for the Bokeh widgets.
"""
bwave = self.spectrograph.recover('wave').to(u.AA)
return bwave.value
@property
def background_flux(self):
"""
Easy instrument background flux access for the Bokeh widgets.
"""
bef = self.spectrograph.recover('bef').to(u.erg / u.s / u.cm**2 / u.AA)
return bef.value
@property
def _snr(self):
return np.nan_to_num(self.exposure.recover('snr').value)
def controller(self, attr, old, new):
"""
Callback to recalculate everything for the figures whenever an input's
value is changed
"""
#Grab values from the inputs
self.exptime = pre_encode(self.refs["exp_slider"].value * u.hour)
self.renorm_magnitude = pre_encode(self.refs["mag_slider"].value * u.mag('AB'))
self.redshift = pre_encode(self.refs["red_slider"].value)
self.aperture = pre_encode(self.refs["ap_slider"].value * u.m)
temp = self.template_options.index(self.refs["template_select"].value)
self.spectrum_type = self.templates[temp]
grat = self.grating_options.index(self.refs["grating_select"].value)
self.grating = self.gratings[grat]
#update the exposure and grab all our relevant values.
self.update_exposure()
snr = self._snr #SNR at bwave
bwave, bflux = self.background_wave, self.background_flux
twave, tflux = self.template_wave, self.template_flux
self.refs["flux_yrange"].start = 0.
self.refs["flux_yrange"].end = 1.5 * tflux.max()
#self.refs["flux_xrange"].start = min(bwave.min(), twave.min())
#self.refs["flux_xrange"].end = max(bwave.max(), twave.max())
self.refs["snr_yrange"].start = 0.
self.refs["snr_yrange"].end = 1.5 * snr.max()
#self.refs["snr_xrange"].start = min(bwave.min(), twave.min())
#self.refs["snr_xrange"].end = max(bwave.max(), twave.max())
self.refs["snr_source"].data = {'y': snr, 'x': bwave}
self.refs["spectrum_template"].data = {'x': twave, 'y': tflux}
self.refs["instrument_background"].data = {'x':bwave, 'y':bflux}
self.refs["red_slider"].start = self.exposure.zmin
self.refs["red_slider"].end = self.exposure.zmax
def dl_change_filename(self, attr, old, new):
self.dl_filename = self.refs["dl_textinput"].value
self.refs["dl_format_button_group"].active = None
def dl_execute(self, *arg):
which_format = self.refs["dl_format_button_group"].active
ext = ['.txt', '.fits'][which_format]
fmt = ['ascii', 'fits'][which_format]
outfile = self.dl_filename + ext
self.refs["dl_linkbox"].text = "Please wait..."
twave = self.template_wave
swave = self.background_wave
snr = self._snr
out_snr = np.interp(twave, swave, snr, left=0., right=0.)
out_table = Table([twave, self.template_flux, out_snr],
names=('wave','flux','sn'))
out_table.write(outfile, format=fmt, overwrite=True)
os.system('gzip -f ' + outfile)
os.system('cp -rp ' + outfile + '.gz /home/jtastro/jt-astro.science/outputs')
out_msg = "Your file is <a href='http://jt-astro.science/outputs/{0}.gz'>{0}.gz</a>. "
self.refs["dl_linkbox"].text = out_msg.format(outfile)