diff options
author | Aaron LI <aaronly.me@outlook.com> | 2015-06-16 16:46:23 +0800 |
---|---|---|
committer | Aaron LI <aaronly.me@outlook.com> | 2015-06-16 16:46:23 +0800 |
commit | 9c7782686fd607fd4e38038ed44f0307da0cd8c7 (patch) | |
tree | f4c3d7167f9f86974f0b3ff19ce910a4feb88cbb /astro/marx | |
parent | 4b96527999e0b63917efa68ddfdffa57122c22cf (diff) | |
download | atoolbox-9c7782686fd607fd4e38038ed44f0307da0cd8c7.tar.bz2 |
Updated astro tools.
* fit_betasbp_cut.py
- fixed 'dof' calculation with 'n-p-1'
- fixed 'par_eps' calculation/configuration
- added 'fit_model_bounds' using 'scipy.optimize.minimize' to
perform function minimization with bounds
- split the data cut section to function 'cut_data'
- added argument 'options' to 'fit_model_bounds'
* marx_pntsrc.py
Run MARX simulation on a given list of point sources, merge the
output simulation results, and finally convert into FITS image.
* blanksky_add_time.py
Add a time column for the chandra blanksky event file.
The time data are generated with a uniform distribution
between TSTART and TSTOP.
Diffstat (limited to 'astro/marx')
-rwxr-xr-x | astro/marx/marx_pntsrc.py | 95 |
1 files changed, 95 insertions, 0 deletions
diff --git a/astro/marx/marx_pntsrc.py b/astro/marx/marx_pntsrc.py new file mode 100755 index 0000000..31ffe49 --- /dev/null +++ b/astro/marx/marx_pntsrc.py @@ -0,0 +1,95 @@ +#!/usr/bin/env python3 +# -*- coding: utf-8 -*- +# +# Aaron LI +# 2015/06/16 + +""" +Run MARX simulation on a given list of point sources, merge the +output simulation results, and finally convert into FITS image. +""" + +__version__ = "0.1.0" +__date__ = "2015/06/16" + +import sys +import argparse +import subprocess +import re + + +def marx_pntsrc(pfile, ra, dec, flux, outdir): + """ + Run MARX simulation for the provided point source. + """ + cmd = "marx @@%(pfile)s SourceRA=%(ra)s " % {"pfile": pfile, "ra": ra} + \ + "SourceDEC=%(dec)s SourceFlux=%(flux)s OutputDir=%(outdir)s" % \ + {"dec": dec, "flux": flux, "outdir": outdir} + print("CMD: %s" % cmd, file=sys.stderr) + subprocess.call(cmd, shell=True) + + +def marxcat(indirs, outdir): + """ + Concatenate a list of MARX simulation results. + """ + if isinstance(indirs, list): + marxdirs = " ".join(indirs) + elif isinstance(indirs, str): + marxdirs = indirs + else: + raise ValueError("invalid indirs type: %s" % indirs) + cmd = "marxcat %(marxdirs)s %(outdir)s" % \ + {"marxdirs": marxdirs, "outdir": outdir} + print("CMD: %s" % cmd, file=sys.stderr) + subprocess.call(cmd, shell=True) + + +def marx2fits(indir, outfile, params=""): + """ + Convert the results of MARX simulation into FITS image. + """ + cmd = "marx2fits %(params)s %(indir)s %(outfile)s" % \ + {"params": params, "indir": indir, "outfile": outfile} + print("CMD: %s" % cmd, file=sys.stderr) + subprocess.call(cmd, shell=True) + + +def main(): + parser = argparse.ArgumentParser( + description="Run MARX on a given list of point sources") + parser.add_argument("-V", "--version", action="version", + version="%(prog)s " + "%s (%s)" % (__version__, __date__)) + parser.add_argument("pfile", help="marx paramter file") + parser.add_argument("srclist", help="point source list file") + parser.add_argument("outprefix", help="prefix of output directories") + args = parser.parse_args() + + outdirs = [] + i = 0 + for line in open(args.srclist, "r"): + if re.match(r"^\s*$", line): + # skip blank line + continue + elif re.match(r"^\s*#", line): + # skip comment line + continue + i += 1 + ra, dec, flux = map(float, line.split()) + print("INFO: ra = %g, dec = %g, flux = %g" % (ra, dec, flux), + file=sys.stderr) + outdir = "%sp%02d" % (args.outprefix, i) + print("INFO: outdir = %s" % outdir, file=sys.stderr) + outdirs.append(outdir) + marx_pntsrc(args.pfile, ra, dec, flux, outdir) + # merge results + merged = args.outprefix + "merged" + marxcat(outdirs, merged) + # convert to FITS image + merged_fits = merged + ".fits" + marx2fits(merged, merged_fits) + + +if __name__ == "__main__": + main() + |