def test_save_bandpass():
fil_file = os.path.join(_install_dir, "data/28.fil")
y = Your(fil_file)
bandpass = y.bandpass()
mask = np.zeros_like(bandpass, dtype=np.bool)
mask[:10] = True
save_bandpass(y, bandpass, outname="28_bp.png")
assert os.path.isfile("28_bp.png")
os.remove("28_bp.png")
save_bandpass(y, bandpass, mask=mask)
assert os.path.isfile("28_bandpass.png")
os.remove("28_bandpass.png")
def bandpass(f, time=0, nspectra=8192, outname=None):
"""
Plots and saves the bandpass
Args:
f: File to read
time: Time in seconds to read for bandpass
nspectra: Number of spectra to read for bandpass
"""
y = Your(f)
if time > 0:
ns = time // y.your_header.native_tsamp
else:
ns = nspectra
bandpass = y.bandpass(nspectra=ns)
save_bandpass(y,
bandpass,
chan_nos=None,
mask=None,
outdir=None,
outname=outname)
else:
if values.verbose:
logging.basicConfig(
level=logging.DEBUG,
format=logging_format,
handlers=[RichHandler(rich_tracebacks=True)],
)
else:
logging.basicConfig(
level=logging.INFO,
format=logging_format,
handlers=[RichHandler(rich_tracebacks=True)],
)
your_object = Your(file=values.files)
bandpass = your_object.bandpass(values.nspectra)
mask = sk_sg_filter(
data=your_object.get_data(0, values.nspectra),
your_object=your_object,
spectral_kurtosis_sigma=values.spectral_kurtosis_sigma,
savgol_frequency_window=values.savgol_frequency_window,
savgol_sigma=values.savgol_sigma,
)
basename = f"{values.output_dir}/{your_object.your_header.basename}_your_rfi_mask"
chan_nos = np.array(range(your_object.your_header.nchans), dtype=int)
save_bandpass(
your_object,
bandpass,
chan_nos=chan_nos,
def test_save_bandpass():
fil_file = os.path.join(_install_dir, 'data/28.fil')
y = Your(fil_file)
save_bandpass(y, y.bandpass(), outname='28_bp.png')
assert os.path.isfile('28_bp.png')
os.remove('28_bp.png')
def test_save_bandpass():
fil_file = os.path.join(_install_dir, "data/28.fil")
y = Your(fil_file)
save_bandpass(y, y.bandpass(), outname="28_bp.png")
assert os.path.isfile("28_bp.png")
os.remove("28_bp.png")
default=[6],
type=float,
nargs='+')
parser.add_argument('-o',
'--output_dir',
help='Output dir for heimdall candidates',
type=str,
required=False,
default='.')
args = parser.parse_args()
your_object = Your(file=args.files)
if args.apply_savgol:
bandpass = your_object.bandpass(nspectra=8192)
chan_nos = np.arange(0, bandpass.shape[0], dtype=np.int)
for fw, sig in zip(args.filter_window, args.sigma):
mask = savgol_filter(bandpass,
your_object.your_header.foff,
fw=fw,
sig=sig)
basename = f'{args.output_dir}/{your_object.your_header.basename}_w{fw}_sig{sig}'
save_bandpass(your_object,
bandpass,
chan_nos=chan_nos,
mask=mask,
outdir=args.output_dir + '/',
outname=f'{basename}_bandpass.png')
kill_mask_file = f'{basename}.bad_chans'
with open(kill_mask_file, 'w') as f:
