def test_schlafly_map():
"""Test the local query on the Schlafly map."""
red = reddening.Reddening(RA, DEC)
red.query_local_map(dustmap='schlafly')
def extinction(argv=None):
"""Get color excess E(B-V) and store it in the data table for further use."""
description = """Get color excess E(B-V) and store it in the data table for further use."""
prog = "clusters_extinction.py"
usage = """%s [options] config input""" % prog
parser = ArgumentParser(prog=prog,
usage=usage,
description=description,
formatter_class=ArgumentDefaultsHelpFormatter)
parser.add_argument('config', help='Configuration (yaml) file')
parser.add_argument(
'input',
help='Input data file: output of clusters_data.py, i.e, hdf5 file')
parser.add_argument("--output", help="Name of the output file (hdf5 file)")
parser.add_argument(
"--overwrite",
action="store_true",
default=False,
help="Overwrite the output files if they exist already")
parser.add_argument("--plot",
action='store_true',
default=False,
help="Make some plots")
args = parser.parse_args(argv)
config = cdata.load_config(args.config)
if args.output is None:
args.output = os.path.basename(args.input).replace(
'.hdf5', '_extinction.hdf5')
if not args.overwrite and os.path.exists(args.output):
raise IOError(
"Output already exists. Remove them or use --overwrite.")
print "INFO: Working on cluster %s (z=%.4f)" % (config['cluster'],
config['redshift'])
print "INFO: Working on filters", config['filter']
# Load the data
data = cdata.read_hdf5(args.input, path='deepCoadd_forced_src', dic=False)
# Query for E(b-v) and compute the extinction
red = reddening.Reddening(data['coord_ra_deg'].tolist(),
data['coord_dec_deg'].tolist())
ebmv = {'ebv_sfd': red.from_argonaut()}
albds = {}
for k in ebmv:
albd = cextinction.from_ebv_sfd_to_megacam_albd(ebmv[k])
albds.update(
{k.replace('ebv_', 'albd_%s_' % f): albd[f]
for f in albd})
# Create a new table and save it
new_tab = hstack([
data['objectId', 'coord_ra', 'coord_dec', 'filter'],
Table(ebmv),
Table(albds)
],
join_type='inner')
new_tab.write(args.output,
path='extinction',
compression=True,
serialize_meta=True,
overwrite=args.overwrite)
print "INFO: Milky Way dust extinction correction applied"
print "INFO: Data saved in", args.output
# Make some plots if asked
if args.plot:
print "INFO: Making some plots"
filt = new_tab['filter'] == config['filter'][0]
cextinction.plots(new_tab['coord_ra'][filt],
new_tab['coord_dec'][filt],
new_tab['ebv_sfd'],
albds['albd_g_sfd'][filt],
title='Dust extinction map, %s, %i sources, g_sfd' %
(config['cluster'], len(new_tab['coord_ra'][filt])),
figname=config['cluster'])
def test_sncosmo():
"""Test the sncosmo wrapper."""
red = reddening.Reddening(RA, DEC)
red.from_sncosmo()
def test_sfd_map():
"""Test the local query on the SFD map."""
red = reddening.Reddening(RA, DEC)
red.query_local_map(dustmap='sfd')
def test_snfactory():
"""Test the 'snfactory' query method."""
red = reddening.Reddening(RA, DEC)
red.from_snfactory()
def test_argonaut():
"""Test the query to the argonaut server."""
red = reddening.Reddening(RA, DEC)
red.from_argonaut()
def test_astroquery():
"""Test the astroquery wrapper."""
red = reddening.Reddening(RA, DEC)
red.from_astroquery(dustmap='SFD98')
red.from_astroquery(dustmap='SF11')
def test_init_mixt():
"""Initialize with one float and one list."""
reddening.Reddening(RA, [DEC])
def test_init_list():
"""Initialize with two list."""
reddening.Reddening([RA], [DEC])
def test_init_float():
"""Initialize with two floats."""
reddening.Reddening(RA, DEC)
def extinction(argv=None):
"""Get color excess E(B-V) and store it in the data table for further use."""
description = """Get color excess E(B-V) and store it in the data table for further use."""
prog = "clusters_extinction.py"
usage = """%s [options] config input""" % prog
parser = ArgumentParser(prog=prog,
usage=usage,
description=description,
formatter_class=ArgumentDefaultsHelpFormatter)
parser.add_argument('config', help='Configuration (yaml) file')
parser.add_argument(
'input',
help='Input data file: output of clusters_data.py, i.e, hdf5 file')
parser.add_argument("--output", help="Name of the output file (hdf5 file)")
parser.add_argument(
"--overwrite",
action="store_true",
default=False,
help="Overwrite the paths in the output file if they exist already")
parser.add_argument("--plot",
action='store_true',
default=False,
help="Make some plots")
parser.add_argument(
"--dustmap",
help="Dustmap to use. The seleted dust map has to be in the"
" following list: sfd, planck, schlafly, green. Comma separated for"
" multplie maps. Overwrites the 'dustmap' key of the config file if any"
"Default map will be SFD")
args = parser.parse_args(argv)
config = cutils.load_config(args.config)
if args.output is None:
args.output = args.input
print("INFO: Working on cluster %s (z=%.4f)" %
(config['cluster'], config['redshift']))
print("INFO: Working on filters", config['filter'])
# Load the data
print("INFO: Loading data from ", args.input)
data = cutils.read_hdf5(args.input, path='deepCoadd_meas', dic=False)
# Query for E(b-v) and compute the extinction
print("INFO: Loading the coordinates")
red = reddening.Reddening(data['coord_ra_deg'].tolist(),
data['coord_dec_deg'].tolist())
if args.dustmap is not None:
dustmap = args.dustmap.split(',')
elif "dustmap" in config:
dustmap = config['dustmap'] if isinstance(
config['dustmap'], list) else [config['dustmap']]
else:
dustmap = ['sfd']
ebmv = {}
print("INFO: Getting the dust maps and the corresponding color excesses")
for dustm in dustmap:
ebmv['ebv_%s' % dustm] = red.query_local_map(dustmap=dustm)
print(
"INFO: Computing the extinction using the loaded dust maps for all filters"
)
albds = {}
for k in ebmv:
albd = cextinction.from_ebv_sfd_to_megacam_albd(ebmv[k])
albds.update(
{k.replace('ebv_', 'albd_%s_' % f): albd[f]
for f in albd})
# Create a new table and save it
print("INFO: Stacking the data into a single table")
new_tab = hstack([
data['id', 'coord_ra', 'coord_dec', 'filter'],
Table(ebmv),
Table(albds)
],
join_type='inner')
cutils.overwrite_or_append(args.output,
'extinction',
new_tab,
overwrite=args.overwrite)
print("INFO: Milky Way dust extinction correction applied")
print("INFO: Data saved in", args.output)
# Make some plots if asked
if args.plot:
print("INFO: Making some plots")
filt = new_tab['filter'] == config['filter'][0]
cextinction.plots(new_tab['coord_ra'][filt],
new_tab['coord_dec'][filt],
new_tab['ebv_sfd'][filt],
albds['albd_g_sfd'][filt],
title='Dust extinction map, %s, %i sources, g_sfd' %
(config['cluster'], len(new_tab['coord_ra'][filt])),
figname=config['cluster'])