From 681a0fbc5df573583f17f508ee085cd49086b18a Mon Sep 17 00:00:00 2001
From: Aaron LI <aly@aaronly.me>
Date: Sun, 3 Sep 2017 20:52:12 +0800
Subject: wsclean.py: Add argument --save-weights (WSClean -saveweights)

Also set temporary directory to "/tmp"
---
 astro/oskar/wsclean.py | 7 +++++++
 1 file changed, 7 insertions(+)

diff --git a/astro/oskar/wsclean.py b/astro/oskar/wsclean.py
index da22e7f..68593f5 100755
--- a/astro/oskar/wsclean.py
+++ b/astro/oskar/wsclean.py
@@ -44,6 +44,9 @@ def main():
     parser.add_argument("--update-model", dest="update_model",
                         action="store_true",
                         help="update the MODEL_DATA column in MS")
+    parser.add_argument("--save-weights", dest="save_weights",
+                        action="store_true",
+                        help="save the gridded weights in <name>-weights.fits")
     parser.add_argument("--weight-briggs", dest="briggs",
                         type=float, default=0.0,
                         help="Briggs weight parameter")
@@ -86,6 +89,7 @@ def main():
         "-log-time",
         "-pol", "XX",  # OSKAR "Scalar" simulation only give "XX" component
         "-make-psf",  # always make the PSF, even no cleaning performed
+        "-tempdir", "/tmp",
     ]
 
     if args.dirty:
@@ -108,6 +112,9 @@ def main():
     else:
         cmdargs += ["-no-update-model-required"]
 
+    if args.save_weights:
+        cmdargs += ["-saveweights"]
+
     if args.threshold:
         cmdargs += ["-threshold", str(args.threshold)]
     else:
-- 
cgit v1.2.2