diff options
| -rwxr-xr-x | astro/oskar/run_oskar.py | 89 |
1 files changed, 89 insertions, 0 deletions
diff --git a/astro/oskar/run_oskar.py b/astro/oskar/run_oskar.py new file mode 100755 index 0000000..34860c4 --- /dev/null +++ b/astro/oskar/run_oskar.py @@ -0,0 +1,89 @@ +#!/usr/bin/env python3 +# +# Copyright (c) Weitna LI <weitian@aaronly.me> +# MIT License +# +# Run OSKAR interferometer simulator, with corresponding settings +# applied to the common configurations. +# +# 2017-06-12 +# + +import os +import re +import argparse +import subprocess +import shutil +from time import time + + +def run_oskar(configfile, model, freq, vis_oskar, vis_ms, dryrun=False): + if vis_oskar is None and vis_ms is None: + raise ValueError("both 'vis_oskar' & 'vis_ms' are missing") + prog = "oskar_sim_interferometer" + print("Update settings ...") + for item, value in [ + ("sky/oskar_sky_model/file", model), + ("observation/start_frequency_hz", str(freq*1e6)), + ("interferometer/oskar_vis_filename", vis_oskar), + ("interferometer/ms_filename", vis_ms)]: + if value is not None: + subprocess.check_call([prog, "--set", configfile, item, value]) + print("Set '%s' -> '%s'" % (item, value)) + print("Simulating %s @ %.2f [MHz] ..." % (model, freq)) + cmd = [prog, configfile] + if dryrun: + print("Dry run mode!") + print("CMD: %s" % " ".join(cmd)) + else: + t1 = time() + subprocess.check_call(cmd) + t2 = time() + print("Elapsed time: %.1f [s]" % (t2-t1)) + + +def main(): + default_fconfig = "sim_interferometer.f{freq:06.2f}.ini" + parser = argparse.ArgumentParser() + parser.add_argument("-d", "--dryrun", dest="dryrun", action="store_true", + help="dry run") + parser.add_argument("-c", "--config", dest="config", required=True, + help="OSKAR base/common configuration file") + parser.add_argument("-C", "--fconfig", dest="fconfig", + default=default_fconfig, + help="filename pattern of the configuration files " + + "updated for OSKAR usage " + + "(default: %s)" % default_fconfig) + parser.add_argument("-l", "--list", dest="listfile", required=True, + help="List of frequencies [MHz] and input sky models") + parser.add_argument("-o", "--outdir", dest="outdir", + default="visibility", + help="simulated visibilities output directory " + + "(default: 'visibility/')") + args = parser.parse_args() + + if not os.path.exists(args.outdir): + os.mkdir(args.outdir) + print("Created output visibility directory: %s" % args.outdir) + + skymodels = [] + for line in open(args.listfile).readlines(): + if re.match(r"^(\s*$|\s*#)", line): + continue + freq, skyfile = line.strip().split() + skymodels.append((float(freq), skyfile)) + print("Got sky model: %s @ %s [MHz]" % (skyfile, freq)) + + for freq, skyfile in skymodels: + basename = os.path.splitext(os.path.basename(skyfile))[0] + vis_oskar = os.path.join(args.outdir, basename+".oskar") + vis_ms = os.path.join(args.outdir, basename+".ms") + configfile = args.fconfig.format(freq=freq) + shutil.copy(args.config, configfile) + print("Copied OSKAR configuration file as: %s" % configfile) + run_oskar(configfile=configfile, freq=freq, model=skyfile, + vis_oskar=vis_oskar, vis_ms=vis_ms, dryrun=args.dryrun) + + +if __name__ == "__main__": + main() |