Skip to content
Snippets Groups Projects
Commit b636bc9c authored by Alexander Kutkin's avatar Alexander Kutkin
Browse files

add option to clip model after cleaning

parent acc3a3d5
Branches
No related tags found
No related merge requests found
...@@ -44,7 +44,7 @@ def modify_filename(fname, string, ext=None): ...@@ -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, 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, automask=3, niter=1000000, multiscale=False, save_source_list=True,
clearfiles=True, clearfiles=True, clip_model_level=None,
fitsmask=None, kwstring=''): fitsmask=None, kwstring=''):
""" """
wsclean wsclean
...@@ -81,6 +81,8 @@ def wsclean(msin, outname=None, pixelsize=3, imagesize=3072, mgain=0.8, multifre ...@@ -81,6 +81,8 @@ def wsclean(msin, outname=None, pixelsize=3, imagesize=3072, mgain=0.8, multifre
for f in todelete: for f in todelete:
os.remove(f) os.remove(f)
remove_model_components_below_level(f'{outname}-sources.txt', clip_model_level)
return 0 return 0
...@@ -302,7 +304,7 @@ def remove_model_components_below_level(model, level=0.0, out=None): ...@@ -302,7 +304,7 @@ def remove_model_components_below_level(model, level=0.0, out=None):
if level is None: if level is None:
return model return model
out = out or 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) df = pd.read_csv(model, skipinitialspace=True)
new = df.query('I>@level') new = df.query('I>@level')
new.to_csv(out, index=False) new.to_csv(out, index=False)
......
0% Loading or .
You are about to add 0 people to the discussion. Proceed with caution.
Please register or to comment