def test_remote_spec_vegafile():
sp1 = spparser.parse_spec('spec(crcalspec$alpha_lyr_stis_007.fits)')
_single_functioncall(sp1, SourceSpectrum, Empirical1D,
'spec(crcalspec$alpha_lyr_stis_007.fits)')
sp2 = SourceSpectrum.from_file(resolve_filename(
os.environ['PYSYN_CDBS'], 'calspec', 'alpha_lyr_stis_007.fits'))
_compare_spectra(sp1, sp2)
def test_remote_z_vega():
sp1 = spparser.parse_spec('z(crcalspec$alpha_lyr_stis_007.fits, 0.1)')
_single_functioncall(sp1, SourceSpectrum, None,
'z(crcalspec$alpha_lyr_stis_007.fits,0.1)', ans_z=0.1)
sp2 = SourceSpectrum.from_file(resolve_filename(
os.environ['PYSYN_CDBS'], 'calspec', 'alpha_lyr_stis_007.fits'))
sp2.z = 0.1
_compare_spectra(sp1, sp2)
def test_remote_rn_calspec_box():
sp1 = spparser.parse_spec(
'rn(crcalspec$gd71_mod_005.fits, box(5000, 10), 17, vegamag)')
_single_functioncall(
sp1, SourceSpectrum, None,
'rn(crcalspec$gd71_mod_005.fits,box(5000.0,10.0),17.0,vegamag)')
gd71 = SourceSpectrum.from_file(resolve_filename(
os.environ['PYSYN_CDBS'], 'calspec', 'gd71_mod_005.fits'))
box = SpectralElement(Box1D, amplitude=1, x_0=5000 * u.AA, width=10 * u.AA)
sp2 = gd71.normalize(17 * units.VEGAMAG, band=box, vegaspec=spectrum.Vega)
_compare_spectra(sp1, sp2)
def test_remote_rn_calspec_u():
sp1 = spparser.parse_spec(
'rn(crcalspec$bd_75d325_stis_002.fits, band(u), 9.5, vegamag) * '
'band(fos, blue, 4.3, g160l)')
# NOTE: No expr for this combo.
_single_functioncall(sp1, SourceSpectrum, None, '')
bd75 = SourceSpectrum.from_file(resolve_filename(
os.environ['PYSYN_CDBS'], 'calspec', 'bd_75d325_stis_002.fits'))
bp_u = SpectralElement.from_filter('johnson_u')
bd75_norm = bd75.normalize(
9.5 * units.VEGAMAG, band=bp_u, vegaspec=spectrum.Vega)
bp_fos = spectrum.band('fos, blue, 4.3, g160l')
sp2 = bd75_norm * bp_fos
_compare_spectra(sp1, sp2)
def _convertstr(value):
"""Convert given filename to source spectrum or passband.
This is used by the interpreter to do the conversion from
string to spectrum object.
"""
if not isinstance(value, str):
return value
value = irafconvert(value)
try:
sp = SourceSpectrum.from_file(value)
except KeyError:
sp = SpectralElement.from_file(value)
return sp
def _get_spectrum(parlist, catdir):
"""Get list of spectra for given parameter list and base name."""
name = parlist[3]
filename = name.split('[')[0]
column = name.split('[')[1][:-1]
filename = os.path.join(catdir, filename)
sp = SourceSpectrum.from_file(filename, flux_col=column)
result = [member for member in parlist]
result.pop()
result.append(sp)
return result
def _get_spectrum(parlist, catdir):
"""Get list of spectra for given parameter list and base name."""
name = parlist[3]
filename = name.split('[')[0]
column = name.split('[')[1][:-1]
filename = os.path.join(catdir, filename)
sp = SourceSpectrum.from_file(filename, flux_col=column)
totflux = sp.integrate()
try:
validate_totalflux(totflux)
except synexceptions.SynphotError:
raise exceptions.ParameterOutOfBounds(
"Parameter '{0}' has no valid data.".format(parlist))
result = [member for member in parlist]
result.pop()
result.append(sp)
return result
def p_functioncall(self, tree):
# Where all the real interpreter action is.
# Note that things that should only be done at the top level
# are performed in :func:`interpret` defined below.
""" V ::= function_call ( V LPAREN V RPAREN ) """
if not isinstance(tree[2].value, list):
args = [tree[2].value]
else:
args = tree[2].value
fname = tree[0].value
metadata = {'expr': '{0}{1}'.format(fname, tuple(args))}
if fname not in _SYFUNCTIONS:
log.error('Unknown function: {0}'.format(fname))
self.error(fname)
else:
# Constant spectrum
if fname == 'unit':
if args[1] not in _SYFORMS:
log.error('Unrecognized unit: {0}'.format(args[1]))
self.error(fname)
try:
fluxunit = units.validate_unit(args[1])
tree.value = SourceSpectrum(ConstFlux1D,
amplitude=args[0] * fluxunit,
meta=metadata)
except NotImplementedError as e:
log.error(str(e))
self.error(fname)
# Black body
elif fname == 'bb':
tree.value = SourceSpectrum(BlackBodyNorm1D,
temperature=args[0])
# Power law
elif fname == 'pl':
if args[2] not in _SYFORMS:
log.error('Unrecognized unit: {0}'.format(args[2]))
self.error(fname)
try:
fluxunit = units.validate_unit(args[2])
tree.value = SourceSpectrum(PowerLawFlux1D,
amplitude=1 * fluxunit,
x_0=args[0],
alpha=-args[1],
meta=metadata)
except (synexceptions.SynphotError, NotImplementedError) as e:
log.error(str(e))
self.error(fname)
# Box throughput
elif fname == 'box':
tree.value = SpectralElement(Box1D,
amplitude=1,
x_0=args[0],
width=args[1],
meta=metadata)
# Source spectrum from file
elif fname == 'spec':
tree.value = SourceSpectrum.from_file(irafconvert(args[0]))
tree.value.meta.update(metadata)
# Passband
elif fname == 'band':
tree.value = spectrum.band(tree[2].svalue)
tree.value.meta.update(metadata)
# Gaussian emission line
elif fname == 'em':
if args[3] not in _SYFORMS:
log.error('Unrecognized unit: {0}'.format(args[3]))
self.error(fname)
x0 = args[0]
fluxunit = units.validate_unit(args[3])
totflux = units.convert_flux(x0, args[2] * fluxunit,
units.PHOTLAM).value
tree.value = SourceSpectrum(GaussianFlux1D,
total_flux=totflux,
mean=x0,
fwhm=args[1])
# Catalog interpolation
elif fname == 'icat':
tree.value = grid_to_spec(*args)
# Renormalize source spectrum
elif fname == 'rn':
sp = args[0]
bp = args[1]
fluxunit = units.validate_unit(args[3])
rnval = args[2] * fluxunit
if not isinstance(sp, SourceSpectrum):
sp = SourceSpectrum.from_file(irafconvert(sp))
if not isinstance(bp, SpectralElement):
bp = SpectralElement.from_file(irafconvert(bp))
# Always force the renormalization to occur: prevent exceptions
# in case of partial overlap. Less robust but duplicates
# IRAF SYNPHOT. Force the renormalization in the case of
# partial overlap, but raise an exception if the spectrum and
# bandpass are entirely disjoint.
try:
tree.value = sp.normalize(rnval,
band=bp,
area=conf.area,
vegaspec=spectrum.Vega)
except synexceptions.PartialOverlap:
tree.value = sp.normalize(rnval,
band=bp,
area=conf.area,
vegaspec=spectrum.Vega,
force=True)
tree.value.warnings = {
'force_renorm': ('Renormalization exceeds the limit '
'of the specified passband.')
}
tree.value.meta.update(metadata)
# Redshift source spectrum (flat spectrum if fails)
elif fname == 'z':
sp = args[0]
# ETC generates junk (i.e., 'null') sometimes
if isinstance(sp, str) and sp != 'null':
sp = SourceSpectrum.from_file(irafconvert(sp))
if isinstance(sp, SourceSpectrum):
tree.value = sp
tree.value.z = args[1]
else:
tree.value = SourceSpectrum(ConstFlux1D, amplitude=1)
tree.value.meta.update(metadata)
# Extinction
elif fname == 'ebmvx':
try:
tree.value = spectrum.ebmvx(args[1], args[0])
except synexceptions.SynphotError as e:
log.error(str(e))
self.error(fname)
tree.value.meta.update(metadata)
# Default
else:
tree.value = ('would call {0} with the following args: '
'{1}'.format(fname, repr(args)))