From b636bc9c6bfd28877aee957174d079efe348c39a Mon Sep 17 00:00:00 2001
From: AK <kutkin@gmail.com>
Date: Wed, 11 Aug 2021 13:07:44 +0000
Subject: [PATCH] add option to clip model after cleaning
---
imcal.py | 6 ++++--
1 file changed, 4 insertions(+), 2 deletions(-)
diff --git a/imcal.py b/imcal.py
index 613ebc9..2b56bae 100755
--- a/imcal.py
+++ b/imcal.py
@@ -44,7 +44,7 @@ def modify_filename(fname, string, ext=None):
def wsclean(msin, outname=None, pixelsize=3, imagesize=3072, mgain=0.8, multifreq=0, autothresh=0.3,
automask=3, niter=1000000, multiscale=False, save_source_list=True,
- clearfiles=True,
+ clearfiles=True, clip_model_level=None,
fitsmask=None, kwstring=''):
"""
wsclean
@@ -81,6 +81,8 @@ def wsclean(msin, outname=None, pixelsize=3, imagesize=3072, mgain=0.8, multifre
for f in todelete:
os.remove(f)
+ remove_model_components_below_level(f'{outname}-sources.txt', clip_model_level)
+
return 0
@@ -302,7 +304,7 @@ def remove_model_components_below_level(model, level=0.0, out=None):
if level is None:
return model
out = out or model
- logging.warning('Overwriting the model')
+ logging.warning('Clipping the model %s to level %f', model, level)
df = pd.read_csv(model, skipinitialspace=True)
new = df.query('I>@level')
new.to_csv(out, index=False)
--
GitLab