wsclean.py: Add argument --save-weights (WSClean -saveweights)

Also set temporary directory to "/tmp"

1 files changed, 7 insertions, 0 deletions

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: