import time
tstart = time.time()

import os
os.environ['HDF5_USE_FILE_LOCKING'] = 'FALSE'
import h5py
import hdf5plugin  # REQUIRED to have the compression plugins available
import numpy as np
import glob
import matplotlib.pyplot as plt
import matplotlib
import copy
import warnings
from pyuvdata import UVFlag, UVCal
from hera_cal import utils
from hera_qm import xrfi
from hera_qm.time_series_metrics import true_stretches

from IPython.display import display, HTML
%matplotlib inline
display(HTML("<style>.container { width:100% !important; }</style>"))
_ = np.seterr(all='ignore')  # get rid of red warnings
%config InlineBackend.figure_format = 'retina'

# get input data file names
SUM_FILE = os.environ.get("SUM_FILE", None)
# SUM_FILE = '/lustre/aoc/projects/hera/h6c-analysis/IDR2/2459861/zen.2459861.25297.sum.uvh5'
SUM_SUFFIX = os.environ.get("SUM_SUFFIX", 'sum.uvh5')

# get input and output suffixes
SMOOTH_CAL_SUFFIX = os.environ.get("SMOOTH_CAL_SUFFIX", 'sum.smooth.calfits')
ZSCORE_SUFFIX =  os.environ.get("ZSCORE_SUFFIX", 'sum.red_avg_zscore.h5')
FLAG_WATERFALL2_SUFFIX = os.environ.get("FLAG_WATERFALL2_SUFFIX", 'sum.flag_waterfall_round_2.h5')
OUT_YAML_SUFFIX = os.environ.get("OUT_YAML_SUFFIX", '_aposteriori_flags.yaml')
OUT_YAML_DIR = os.environ.get("OUT_YAML_DIR", None)

# build globs
sum_glob = '.'.join(SUM_FILE.split('.')[:-3]) + '.*.' + SUM_SUFFIX
cal_files_glob = sum_glob.replace(SUM_SUFFIX, SMOOTH_CAL_SUFFIX)
zscore_glob = sum_glob.replace(SUM_SUFFIX, ZSCORE_SUFFIX)

# build out yaml file
if OUT_YAML_DIR is None:
    OUT_YAML_DIR = os.path.dirname(SUM_FILE)
out_yaml_file = os.path.join(OUT_YAML_DIR, SUM_FILE.split('.')[-4] + OUT_YAML_SUFFIX)    

# get flagging parameters
Z_THRESH = float(os.environ.get("Z_THRESH", 5))
WS_Z_THRESH = float(os.environ.get("WS_Z_THRESH", 4))
AVG_Z_THRESH = float(os.environ.get("AVG_Z_THRESH", 1))
MAX_FREQ_FLAG_FRAC = float(os.environ.get("MAX_FREQ_FLAG_FRAC", .25))
MAX_TIME_FLAG_FRAC = float(os.environ.get("MAX_TIME_FLAG_FRAC", .1))

for setting in ['Z_THRESH', 'WS_Z_THRESH', 'AVG_Z_THRESH', 'MAX_FREQ_FLAG_FRAC', 'MAX_TIME_FLAG_FRAC']:
    print(f'{setting} = {eval(setting)}')

Z_THRESH = 5.0
WS_Z_THRESH = 4.0
AVG_Z_THRESH = 1.0
MAX_FREQ_FLAG_FRAC = 0.25
MAX_TIME_FLAG_FRAC = 0.1

# load z-scores
zscore_files = sorted(glob.glob(zscore_glob))
print(f'Found {len(zscore_files)} *.{ZSCORE_SUFFIX} files starting with {zscore_files[0]}.')
uvf = UVFlag(zscore_files, use_future_array_shapes=True)

Found 1568 *.sum.red_avg_zscore.h5 files starting with /mnt/sn1/data1/2460451/zen.2460451.16897.sum.red_avg_zscore.h5.

# get calibration solution files
cal_files = sorted(glob.glob(cal_files_glob))
print(f'Found {len(cal_files)} *.{SMOOTH_CAL_SUFFIX} files starting with {cal_files[0]}.')

Found 1568 *.sum.smooth.calfits files starting with /mnt/sn1/data1/2460451/zen.2460451.16897.sum.smooth.calfits.

assert len(zscore_files) == len(cal_files)

# extract z-scores and correct by a single number per polarization to account for biases created by filtering
zscore = {pol: uvf.metric_array[:, :, np.argwhere(uvf.polarization_array == utils.polstr2num(pol, x_orientation=uvf.x_orientation))[0][0]] for pol in ['ee', 'nn']}
zscore = {pol: zscore[pol] - np.nanmedian(zscore[pol]) for pol in zscore}

freqs = uvf.freq_array
times = uvf.time_array

extent = [freqs[0] / 1e6, freqs[-1] / 1e6, times[-1] - int(times[0]), times[0] - int(times[0])]

def plot_max_z_score(zscore, flags=None):
    if flags is None:
        flags = np.any(~np.isfinite(list(zscore.values())), axis=0)
    plt.figure(figsize=(14,10), dpi=100)
    plt.imshow(np.where(flags, np.nan, np.nanmax([zscore['ee'], zscore['nn']], axis=0)), aspect='auto', 
               cmap='coolwarm', interpolation='none', vmin=-10, vmax=10, extent=extent)
    plt.colorbar(location='top', label='Max z-score of either polarization', extend='both', aspect=40, pad=.02)
    plt.xlabel('Frequency (MHz)')
    plt.ylabel(f'JD - {int(times[0])}')
    plt.tight_layout()

plot_max_z_score(zscore)

All-NaN axis encountered

def plot_histogram():
    plt.figure(figsize=(14,4), dpi=100)
    bins = np.arange(-50, 100, .1)
    hist_ee = plt.hist(np.ravel(zscore['ee']), bins=bins, density=True, label='ee-polarized z-scores', alpha=.5)
    hist_nn = plt.hist(np.ravel(zscore['nn']), bins=bins, density=True, label='nn-polarized z-scores', alpha=.5)
    plt.plot(bins, (2*np.pi)**-.5 * np.exp(-bins**2 / 2), 'k:', label='Gaussian approximate\nnoise-only distribution')
    plt.axvline(WS_Z_THRESH, c='r', ls='--', label='Watershed z-score')
    plt.axvline(Z_THRESH, c='r', ls='-', label='Threshold z-score')
    plt.yscale('log')
    all_densities = np.concatenate([hist_ee[0][hist_ee[0] > 0], hist_nn[0][hist_nn[0] > 0]]) 
    plt.ylim(np.min(all_densities) / 2, np.max(all_densities) * 2)
    plt.xlim([-50, 100])
    plt.legend()
    plt.xlabel('z-score')
    plt.ylabel('Density')
    plt.tight_layout()

plot_histogram()

def iteratively_flag_on_averaged_zscore(flags, zscore, avg_z_thresh=1.5, verbose=True):
    '''Flag whole integrations or channels based on average z-score. This is done
    iteratively to prevent bad times affecting channel averages or vice versa.'''
    flagged_chan_count = 0
    flagged_int_count = 0
    while True:
        zspec = np.nanmean(np.where(flags, np.nan, zscore), axis=0)
        ztseries = np.nanmean(np.where(flags, np.nan, zscore), axis=1)

        if (np.nanmax(zspec) < avg_z_thresh) and (np.nanmax(ztseries) < avg_z_thresh):
            break

        if np.nanmax(zspec) >= np.nanmax(ztseries):
            flagged_chan_count += np.sum((zspec >= np.nanmax(ztseries)) & (zspec >= avg_z_thresh))
            flags[:, (zspec >= np.nanmax(ztseries)) & (zspec >= avg_z_thresh)] = True
        else:
            flagged_int_count += np.sum((ztseries >= np.nanmax(zspec)) & (ztseries >= avg_z_thresh))
            flags[(ztseries >= np.nanmax(zspec)) & (ztseries >= avg_z_thresh), :] = True

    if verbose:
        print(f'\tFlagging an additional {flagged_int_count} integrations and {flagged_chan_count} channels.')

def impose_max_chan_flag_frac(flags, max_flag_frac=.25, verbose=True):
    '''Flag channels already flagged more than max_flag_frac (excluding completely flagged times).'''
    unflagged_times = ~np.all(flags, axis=1)
    frequently_flagged_chans =  np.mean(flags[unflagged_times, :], axis=0) >= max_flag_frac
    if verbose:
        print(f'\tFlagging {np.sum(frequently_flagged_chans) - np.sum(np.all(flags, axis=0))} channels previously flagged {max_flag_frac:.2%} or more.')        
    flags[:, frequently_flagged_chans] = True 
        
def impose_max_time_flag_frac(flags, max_flag_frac=.25, verbose=True):
    '''Flag times already flagged more than max_flag_frac (excluding completely flagged channels).'''
    unflagged_chans = ~np.all(flags, axis=0)
    frequently_flagged_times =  np.mean(flags[:, unflagged_chans], axis=1) >= max_flag_frac
    if verbose:
        print(f'\tFlagging {np.sum(frequently_flagged_times) - np.sum(np.all(flags, axis=1))} times previously flagged {max_flag_frac:.2%} or more.')
    flags[frequently_flagged_times, :] = True

flags = np.any(~np.isfinite(list(zscore.values())), axis=0)
print(f'{np.mean(flags):.3%} of waterfall flagged to start.')

# flag largest outliers
for pol in ['ee', 'nn']:
    flags |= (zscore[pol] > Z_THRESH) 
print(f'{np.mean(flags):.3%} of waterfall flagged after flagging z > {Z_THRESH} outliers.')
    
# watershed flagging
while True:
    nflags = np.sum(flags)
    for pol in ['ee', 'nn']:
        flags |= xrfi._ws_flag_waterfall(zscore[pol], flags, WS_Z_THRESH)
    if np.sum(flags) == nflags:
        break
print(f'{np.mean(flags):.3%} of waterfall flagged after watershed flagging on z > {WS_Z_THRESH} neightbors of prior flags.')
        
# flag whole integrations or channels
while True:
    nflags = np.sum(flags)
    for pol in ['ee', 'nn']:    
        iteratively_flag_on_averaged_zscore(flags, zscore[pol], avg_z_thresh=AVG_Z_THRESH, verbose=True)
        impose_max_chan_flag_frac(flags, max_flag_frac=MAX_FREQ_FLAG_FRAC, verbose=True)
        impose_max_time_flag_frac(flags, max_flag_frac=MAX_TIME_FLAG_FRAC, verbose=True)
    if np.sum(flags) == nflags:
        break  
print(f'{np.mean(flags):.3%} of waterfall flagged after flagging whole times and channels with average z > {AVG_Z_THRESH}.')

30.586% of waterfall flagged to start.
34.129% of waterfall flagged after flagging z > 5.0 outliers.

34.588% of waterfall flagged after watershed flagging on z > 4.0 neightbors of prior flags.

Mean of empty slice

Mean of empty slice

	Flagging an additional 0 integrations and 33 channels.
	Flagging 33 channels previously flagged 25.00% or more.
	Flagging 403 times previously flagged 10.00% or more.

	Flagging an additional 0 integrations and 0 channels.
	Flagging 0 channels previously flagged 25.00% or more.
	Flagging 0 times previously flagged 10.00% or more.

	Flagging an additional 0 integrations and 4 channels.
	Flagging 0 channels previously flagged 25.00% or more.
	Flagging 0 times previously flagged 10.00% or more.

	Flagging an additional 0 integrations and 0 channels.
	Flagging 0 channels previously flagged 25.00% or more.
	Flagging 0 times previously flagged 10.00% or more.

	Flagging an additional 0 integrations and 0 channels.
	Flagging 0 channels previously flagged 25.00% or more.
	Flagging 0 times previously flagged 10.00% or more.

	Flagging an additional 0 integrations and 0 channels.
	Flagging 0 channels previously flagged 25.00% or more.
	Flagging 0 times previously flagged 10.00% or more.
45.171% of waterfall flagged after flagging whole times and channels with average z > 1.0.

plot_max_z_score(zscore, flags=flags)

All-NaN axis encountered

def zscore_spectra():
    fig, axes = plt.subplots(2, 1, figsize=(14,6), dpi=100, sharex=True, sharey=True, gridspec_kw={'hspace': 0})
    for ax, pol in zip(axes, ['ee', 'nn']):

        ax.plot(freqs / 1e6, np.nanmean(zscore[pol], axis=0),'r', label=f'{pol}-Polarization Before Round 2 Flagging', lw=.5)
        ax.plot(freqs / 1e6, np.nanmean(np.where(flags, np.nan, zscore[pol]), axis=0), label=f'{pol}-Polarization After Round 2 Flagging')
        ax.legend(loc='lower right')
        ax.set_ylabel('Time-Averged Z-Score\n(Excluding Flags)')
        ax.set_ylim(-11, 11)
    axes[1].set_xlabel('Frequency (MHz)')
    plt.tight_layout()

zscore_spectra()

Mean of empty slice

Mean of empty slice

def summarize_flagging():
    plt.figure(figsize=(14,10), dpi=100)
    cmap = matplotlib.colors.ListedColormap(((0, 0, 0),) + matplotlib.cm.get_cmap("Set2").colors[0:2])
    plt.imshow(np.where(np.any(~np.isfinite(list(zscore.values())), axis=0), 1, np.where(flags, 2, 0)), 
               aspect='auto', cmap=cmap, interpolation='none', extent=extent)
    plt.clim([-.5, 2.5])
    cbar = plt.colorbar(location='top', aspect=40, pad=.02)
    cbar.set_ticks([0, 1, 2])
    cbar.set_ticklabels(['Unflagged', 'Previously Flagged', 'Flagged Here Using Delayed Filtered z-Scores'])
    plt.xlabel('Frequency (MHz)')
    plt.ylabel(f'JD - {int(times[0])}')
    plt.tight_layout()

summarize_flagging()

The get_cmap function was deprecated in Matplotlib 3.7 and will be removed two minor releases later. Use ``matplotlib.colormaps[name]`` or ``matplotlib.colormaps.get_cmap(obj)`` instead.

add_to_history = 'by full_day_rfi_round_2 notebook with the following environment:\n' + '=' * 65 + '\n' + os.popen('conda env export').read() + '=' * 65

tind = 0
always_flagged_ants = set()
ever_unflagged_ants = set()
for cal_file in cal_files:
    with warnings.catch_warnings():
        warnings.simplefilter("ignore")    
        
        # update cal_file
        uvc = UVCal()
        uvc.read(cal_file, use_future_array_shapes=True)
        uvc.flag_array |= (flags[tind:tind + len(uvc.time_array), :].T)[None, :, :, None]
        uvc.history += 'Modified ' + add_to_history
        uvc.write_calfits(cal_file, clobber=True)
        
        # keep track of flagged antennas
        for antnum in uvc.ant_array:
            for antpol in ['Jee', 'Jnn']:
                if np.all(uvc.get_flags(antnum, antpol)):
                    if (antnum, antpol) not in ever_unflagged_ants:
                        always_flagged_ants.add((antnum, antpol))
                else:
                    ever_unflagged_ants.add((antnum, antpol))
                    always_flagged_ants.discard((antnum, antpol))
                

        # Create new flag object
        uvf_out = UVFlag(uvc, waterfall=True, mode='flag')
        uvf_out.flag_array |= flags[tind:tind + len(uvc.time_array), :, None]
        uvf_out.history += 'Produced ' + add_to_history
        uvf_out.write(cal_file.replace(SMOOTH_CAL_SUFFIX, FLAG_WATERFALL2_SUFFIX), clobber=True)
        
        # increment time index
        tind += len(uvc.time_array)

print(f'Saved {len(cal_files)} *.{FLAG_WATERFALL2_SUFFIX} files starting with {cal_files[0].replace(SMOOTH_CAL_SUFFIX, FLAG_WATERFALL2_SUFFIX)}.')

File /mnt/sn1/data1/2460451/zen.2460451.16897.sum.flag_waterfall_round_2.h5 exists; clobbering

File /mnt/sn1/data1/2460451/zen.2460451.16919.sum.flag_waterfall_round_2.h5 exists; clobbering

File /mnt/sn1/data1/2460451/zen.2460451.16941.sum.flag_waterfall_round_2.h5 exists; clobbering

File /mnt/sn1/data1/2460451/zen.2460451.16964.sum.flag_waterfall_round_2.h5 exists; clobbering

File /mnt/sn1/data1/2460451/zen.2460451.16986.sum.flag_waterfall_round_2.h5 exists; clobbering

File /mnt/sn1/data1/2460451/zen.2460451.17009.sum.flag_waterfall_round_2.h5 exists; clobbering

File /mnt/sn1/data1/2460451/zen.2460451.17031.sum.flag_waterfall_round_2.h5 exists; clobbering

File /mnt/sn1/data1/2460451/zen.2460451.17053.sum.flag_waterfall_round_2.h5 exists; clobbering

File /mnt/sn1/data1/2460451/zen.2460451.17076.sum.flag_waterfall_round_2.h5 exists; clobbering

File /mnt/sn1/data1/2460451/zen.2460451.17098.sum.flag_waterfall_round_2.h5 exists; clobbering

# write summary of entirely flagged times/freqs/ants to yaml
all_flagged_times = np.all(flags, axis=1)
all_flagged_freqs = np.all(flags, axis=0)
all_flagged_ants = sorted(always_flagged_ants)

dt = np.median(np.diff(times))
out_yml_str = 'JD_flags: ' + str([[times[flag_stretch][0] - dt / 2, times[flag_stretch][-1] + dt / 2] 
                                  for flag_stretch in true_stretches(all_flagged_times)])
df = np.median(np.diff(freqs))
out_yml_str += '\n\nfreq_flags: ' + str([[freqs[flag_stretch][0] - df / 2, freqs[flag_stretch][-1] + df / 2] 
                                         for flag_stretch in true_stretches(all_flagged_freqs)])
out_yml_str += '\n\nex_ants: ' + str(all_flagged_ants).replace("'", "").replace('(', '[').replace(')', ']')

print(f'Writing the following to {out_yaml_file}\n' + '-' * (25 + len(out_yaml_file)))
print(out_yml_str)
with open(out_yaml_file, 'w') as outfile:
    outfile.writelines(out_yml_str)

Writing the following to /mnt/sn1/data1/2460451/2460451_aposteriori_flags.yaml
------------------------------------------------------------------------------
JD_flags: [[2460451.168855062, 2460451.1691906066], [2460451.1693024547, 2460451.169414303], [2460451.169526151, 2460451.1718749614], [2460451.1719868095, 2460451.1720986576], [2460451.172322354, 2460451.172434202], [2460451.17254605, 2460451.1727697463], [2460451.1728815944, 2460451.1731052906], [2460451.1735526826, 2460451.1736645307], [2460451.173776379, 2460451.174000075], [2460451.1751185562, 2460451.1752304044], [2460451.1753422525, 2460451.1761251893], [2460451.1771318223, 2460451.1773555186], [2460451.177579215, 2460451.177802911], [2460451.1794806323, 2460451.1795924804], [2460451.1799280248, 2460451.180039873], [2460451.180151721, 2460451.180263569], [2460451.1804872653, 2460451.1807109616], [2460451.180934658, 2460451.181046506], [2460451.181158354, 2460451.1813820503], [2460451.1814938984, 2460451.1818294427], [2460451.182276835, 2460451.1832834682], [2460451.1833953164, 2460451.1836190126], [2460451.1837308607, 2460451.183842709], [2460451.184066405, 2460451.184178253], [2460451.184401949, 2460451.184513797], [2460451.1848493414, 2460451.1849611895], [2460451.1859678226, 2460451.1860796707], [2460451.186303367, 2460451.1877573924], [2460451.1887640255, 2460451.1888758736], [2460451.1895469623, 2460451.1896588104], [2460451.1902180505, 2460451.1903298986], [2460451.1904417467, 2460451.190665443], [2460451.190777291, 2460451.1923431647], [2460451.192455013, 2460451.193461646], [2460451.193573494, 2460451.19379719], [2460451.1940208864, 2460451.1950275195], [2460451.196146, 2460451.1963696964], [2460451.196817089, 2460451.196928937], [2460451.1976000257, 2460451.197823722], [2460451.19793557, 2460451.1982711144], [2460451.198718507, 2460451.199054051], [2460451.200731773, 2460451.2012910135], [2460451.201402861, 2460451.20207395], [2460451.2025213423, 2460451.2026331904], [2460451.2028568867, 2460451.202968735], [2460451.2045346084, 2460451.2047583046], [2460451.2054293933, 2460451.2057649377], [2460451.2063241783, 2460451.2067715707], [2460451.207554507, 2460451.2077782033], [2460451.210462558, 2460451.210574406], [2460451.211357343, 2460451.2115810392], [2460451.212363976, 2460451.21269952], [2460451.212811368, 2460451.212923216], [2460451.2130350643, 2460451.2132587605], [2460451.2142653936, 2460451.2143772417], [2460451.2159431153, 2460451.2161668115], [2460451.2171734446, 2460451.217397141], [2460451.219074862, 2460451.2195222545], [2460451.2201933432, 2460451.2205288876], [2460451.220752584, 2460451.220864432], [2460451.2211999763, 2460451.2216473687], [2460451.2224303056, 2460451.222877698], [2460451.223660635, 2460451.2239961787], [2460451.2264568373, 2460451.2267923816], [2460451.227127926, 2460451.227910863], [2460451.228134559, 2460451.2283582552], [2460451.229029344, 2460451.229141192], [2460451.2300359765, 2460451.2302596727], [2460451.230595217, 2460451.230707065], [2460451.2310426096, 2460451.231266306], [2460451.2319373945, 2460451.2331677238], [2460451.2346217493, 2460451.2347335974], [2460451.2349572936, 2460451.2350691417], [2460451.235852078, 2460451.235963926], [2460451.2360757743, 2460451.240885243], [2460451.2413326353, 2460451.2416681796], [2460451.243793294, 2460451.2443525344], [2460451.2444643825, 2460451.2453591675], [2460451.245694712, 2460451.2462539524], [2460451.247372433, 2460451.247484281], [2460451.2479316737, 2460451.248267218], [2460451.2486027624, 2460451.2487146105], [2460451.2488264586, 2460451.2489383067], [2460451.249162003, 2460451.2499449397], [2460451.2505041803, 2460451.2507278766], [2460451.2516226615, 2460451.251958206], [2460451.2521819016, 2460451.252405598], [2460451.252629294, 2460451.2528529903], [2460451.25464256, 2460451.2549781045], [2460451.25643213, 2460451.2566558262], [2460451.2592283324, 2460451.259675725], [2460451.2604586617, 2460451.260794206], [2460451.2612415985, 2460451.261688991], [2460451.262583776, 2460451.262695624], [2460451.263925953, 2460451.2643733453], [2460451.2644851934, 2460451.2647088896], [2460451.2648207378, 2460451.265156282], [2460451.266162915, 2460451.2662747633], [2460451.2663866114, 2460451.2666103076], [2460451.266945852, 2460451.2670577], [2460451.2673932444, 2460451.267728789], [2460451.268064333, 2460451.2682880294], [2460451.26884727, 2460451.268959118], [2460451.269182814, 2460451.269294662], [2460451.2698539025, 2460451.2700775987], [2460451.2719790167, 2460451.272090865], [2460451.272202713, 2460451.272426409], [2460451.2727619535, 2460451.2728738016], [2460451.2736567385, 2460451.2738804347], [2460451.274104131, 2460451.274327827], [2460451.274551523, 2460451.274775219], [2460451.2767884852, 2460451.2769003334], [2460451.2770121815, 2460451.277459574], [2460451.2780188145, 2460451.2782425107], [2460451.279249144, 2460451.2802557764], [2460451.2815979538, 2460451.2822690425], [2460451.284617853, 2460451.2849533972], [2460451.2851770935, 2460451.2852889416], [2460451.2877495997, 2460451.287973296], [2460451.289203625, 2460451.2894273214], [2460451.290769499, 2460451.290881347], [2460451.2916642833, 2460451.2917761314], [2460451.2918879795, 2460451.2919998276], [2460451.2921116757, 2460451.292223524], [2460451.292335372, 2460451.2931183088], [2460451.293342005, 2460451.294460486], [2460451.2946841824, 2460451.2947960305], [2460451.2950197267, 2460451.295243423], [2460451.295467119, 2460451.2955789673], [2460451.2969211447, 2460451.297480385], [2460451.2979277773, 2460451.2981514735], [2460451.298487018, 2460451.298598866], [2460451.299717347, 2460451.2998291953], [2460451.300612132, 2460451.3008358283], [2460451.3026253977, 2460451.302737246], [2460451.3045268157, 2460451.304750512], [2460451.30486236, 2460451.304974208], [2460451.3063163855, 2460451.306987474], [2460451.3081059554, 2460451.3082178035], [2460451.3088888917, 2460451.30900074], [2460451.3094481323, 2460451.310119221], [2460451.3104547653, 2460451.3106784616], [2460451.3113495503, 2460451.3114613984], [2460451.3116850946, 2460451.311908791], [2460451.312020639, 2460451.312244335], [2460451.312803576, 2460451.312915424], [2460451.3134746645, 2460451.3135865126], [2460451.3150405376, 2460451.3151523857], [2460451.316270867, 2460451.3166064112], [2460451.3168301075, 2460451.3170538037], [2460451.31906707, 2460451.319178918], [2460451.3199618543, 2460451.3200737024], [2460451.3220869685, 2460451.3221988166], [2460451.3229817534, 2460451.3230936015], [2460451.323429146, 2460451.323652842], [2460451.3262253483, 2460451.3263371964], [2460451.3264490445, 2460451.3266727407], [2460451.3274556776, 2460451.3275675257], [2460451.329804488, 2460451.330028184], [2460451.3302518805, 2460451.330587425], [2460451.332376994, 2460451.3326006904], [2460451.3328243867, 2460451.333159931], [2460451.3345021084, 2460451.3346139565], [2460451.340206362, 2460451.340541906], [2460451.341548539, 2460451.3416603873], [2460451.3422196275, 2460451.342555172], [2460451.342778868, 2460451.342890716], [2460451.3434499567, 2460451.343897349], [2460451.34445659, 2460451.344568438], [2460451.346693552, 2460451.3468054], [2460451.3497134508, 2460451.349825299], [2460451.350048995, 2460451.350160843], [2460451.3548584636, 2460451.3549703117], [2460451.3627996794, 2460451.3629115275], [2460451.364253705, 2460451.364365553], [2460451.365595882, 2460451.365819578], [2460451.3679446923, 2460451.3680565404], [2460451.369734262, 2460451.3701816546], [2460451.3705171985, 2460451.370852743], [2460451.3725304645, 2460451.372754161], [2460451.373089705, 2460451.3732015532], [2460451.3831560346, 2460451.3838271233], [2460451.385392997, 2460451.385504845], [2460451.3861759338, 2460451.386287782], [2460451.386623326, 2460451.3868470225], [2460451.391544643, 2460451.391656491], [2460451.3921038834, 2460451.3922157316], [2460451.39288682, 2460451.3935579085], [2460451.3943408453, 2460451.3944526934], [2460451.3954593264, 2460451.3956830227], [2460451.3962422633, 2460451.3963541114], [2460451.396801504, 2460451.396913352], [2460451.397696289, 2460451.397808137], [2460451.4006043393, 2460451.4007161874], [2460451.401275428, 2460451.401387276], [2460451.4036242384, 2460451.403959783], [2460451.4040716304, 2460451.4041834786], [2460451.4050782635, 2460451.4051901116], [2460451.405749352, 2460451.4059730484], [2460451.4088810994, 2460451.4089929475], [2460451.40944034, 2460451.409552188], [2460451.409887732, 2460451.4101114282], [2460451.4121246943, 2460451.4122365424], [2460451.4134668717, 2460451.413690568], [2460451.4183881884, 2460451.4185000365], [2460451.418835581, 2460451.419059277], [2460451.424092442, 2460451.424316138], [2460451.427671581, 2460451.4280071254], [2460451.428678214, 2460451.428790062], [2460451.431474417, 2460451.431698113], [2460451.432704746, 2460451.432928442], [2460451.436954974, 2460451.4370668223], [2460451.4387445436, 2460451.439080088], [2460451.4413170503, 2460451.4414288984], [2460451.4416525946, 2460451.441876291], [2460451.4424355314, 2460451.4425473795], [2460451.4475805443, 2460451.4476923924], [2460451.450824139, 2460451.4509359873], [2460451.451830772, 2460451.4519426203], [2460451.4565283926, 2460451.4566402407], [2460451.4571994813, 2460451.457646874], [2460451.462344494, 2460451.4624563423], [2460451.4627918866, 2460451.4629037348], [2460451.4636866716, 2460451.4637985197], [2460451.463910368, 2460451.464022216], [2460451.464134064, 2460451.4643577603], [2460451.4658117853, 2460451.4720752793], [2460451.473641153, 2460451.473753001], [2460451.4739766973, 2460451.4752070266], [2460451.476437356, 2460451.476661052], [2460451.4767729, 2460451.4768847483], [2460451.477108444, 2460451.477220292], [2460451.477891381, 2460451.478003229], [2460451.4782269252, 2460451.480687584], [2460451.480799432, 2460451.484154875], [2460451.484266723, 2460451.4846022674], [2460451.4857207485, 2460451.4858325967], [2460451.4863918372, 2460451.4865036854], [2460451.4867273816, 2460451.487286622], [2460451.4873984703, 2460451.4875103184], [2460451.487957711, 2460451.488181407], [2460451.4909776095, 2460451.491425002], [2460451.4920960907, 2460451.492207939], [2460451.4942212044, 2460451.4943330525], [2460451.494556749, 2460451.494668597], [2460451.494892293, 2460451.49567523], [2460451.495898926, 2460451.496570015], [2460451.496793711, 2460451.4980240404], [2460451.4981358885, 2460451.4983595847], [2460451.498583281, 2460451.498806977], [2460451.4990306734, 2460451.499589914], [2460451.500037306, 2460451.500261002], [2460451.5007083947, 2460451.500820243], [2460451.501043939, 2460451.5014913315], [2460451.5016031796, 2460451.5017150277], [2460451.502050572, 2460451.502833509], [2460451.503057205, 2460451.5045112306], [2460451.504846775, 2460451.5052941674], [2460451.5059652557, 2460451.506077104], [2460451.506188952, 2460451.5065244962], [2460451.5066363444, 2460451.5068600406], [2460451.5069718887, 2460451.508202218], [2460451.5086496104, 2460451.509097003], [2460451.511557661, 2460451.511669509], [2460451.5120050535, 2460451.5121169016], [2460451.5127879903, 2460451.5130116865], [2460451.5137946233, 2460451.514242016], [2460451.514465712, 2460451.5200581173]]

freq_flags: [[46859741.2109375, 47836303.7109375], [47958374.0234375, 49667358.3984375], [49911499.0234375, 50033569.3359375], [50155639.6484375, 50521850.5859375], [50765991.2109375, 50888061.5234375], [51010131.8359375, 52474975.5859375], [62240600.5859375, 62850952.1484375], [69931030.2734375, 70053100.5859375], [87387084.9609375, 108261108.3984375], [112167358.3984375, 112411499.0234375], [112655639.6484375, 112777709.9609375], [113021850.5859375, 113754272.4609375], [116439819.3359375, 116561889.6484375], [116683959.9609375, 116806030.2734375], [121566772.4609375, 122055053.7109375], [124740600.5859375, 125228881.8359375], [127548217.7734375, 127670288.0859375], [129989624.0234375, 130111694.3359375], [136337280.2734375, 136459350.5859375], [136825561.5234375, 137313842.7734375], [137435913.0859375, 138046264.6484375], [141464233.3984375, 141586303.7109375], [141708374.0234375, 141830444.3359375], [142074584.9609375, 142318725.5859375], [142684936.5234375, 142807006.8359375], [143051147.4609375, 143173217.7734375], [143783569.3359375, 144271850.5859375], [145858764.6484375, 145980834.9609375], [147445678.7109375, 147567749.0234375], [149887084.9609375, 150009155.2734375], [154159545.8984375, 154403686.5234375], [169906616.2109375, 170150756.8359375], [170883178.7109375, 171005249.0234375], [171737670.8984375, 171859741.2109375], [175155639.6484375, 175399780.2734375], [181137084.9609375, 181381225.5859375], [183212280.2734375, 183334350.5859375], [187362670.8984375, 187606811.5234375], [189926147.4609375, 190048217.7734375], [191146850.5859375, 191513061.5234375], [197128295.8984375, 197372436.5234375], [197738647.4609375, 198226928.7109375], [198348999.0234375, 198715209.9609375], [199203491.2109375, 199325561.5234375], [201278686.5234375, 201644897.4609375], [201766967.7734375, 202377319.3359375], [204940795.8984375, 205062866.2109375], [207138061.5234375, 207260131.8359375], [207992553.7109375, 209213256.8359375], [209945678.7109375, 210067749.0234375], [212142944.3359375, 212265014.6484375], [215194702.1484375, 215316772.4609375], [220687866.2109375, 220809936.5234375], [222885131.8359375, 223617553.7109375], [227401733.3984375, 227523803.7109375], [227645874.0234375, 227767944.3359375], [229110717.7734375, 229354858.3984375], [231063842.7734375, 231185913.0859375]]

ex_ants: [[3, Jnn], [7, Jee], [7, Jnn], [8, Jnn], [9, Jee], [9, Jnn], [10, Jee], [10, Jnn], [18, Jee], [18, Jnn], [20, Jee], [22, Jnn], [27, Jee], [27, Jnn], [28, Jee], [28, Jnn], [29, Jnn], [31, Jnn], [32, Jnn], [34, Jee], [35, Jnn], [36, Jee], [37, Jnn], [40, Jnn], [45, Jee], [46, Jee], [47, Jee], [47, Jnn], [51, Jee], [54, Jnn], [55, Jee], [61, Jee], [61, Jnn], [62, Jee], [63, Jee], [63, Jnn], [64, Jee], [64, Jnn], [69, Jee], [73, Jee], [73, Jnn], [77, Jee], [77, Jnn], [78, Jee], [78, Jnn], [80, Jnn], [81, Jee], [81, Jnn], [83, Jee], [83, Jnn], [86, Jee], [86, Jnn], [87, Jee], [88, Jee], [88, Jnn], [90, Jee], [90, Jnn], [92, Jee], [93, Jnn], [95, Jee], [96, Jnn], [97, Jnn], [103, Jnn], [104, Jnn], [107, Jee], [107, Jnn], [108, Jnn], [109, Jnn], [115, Jnn], [116, Jnn], [117, Jnn], [119, Jee], [119, Jnn], [120, Jee], [120, Jnn], [127, Jee], [127, Jnn], [129, Jee], [130, Jee], [130, Jnn], [131, Jnn], [134, Jnn], [136, Jnn], [141, Jee], [154, Jnn], [161, Jnn], [170, Jee], [171, Jnn], [173, Jnn], [174, Jnn], [175, Jee], [175, Jnn], [176, Jee], [176, Jnn], [177, Jee], [177, Jnn], [178, Jee], [178, Jnn], [180, Jee], [180, Jnn], [183, Jee], [183, Jnn], [188, Jnn], [192, Jee], [195, Jee], [195, Jnn], [196, Jee], [196, Jnn], [199, Jee], [199, Jnn], [200, Jee], [200, Jnn], [202, Jnn], [204, Jnn], [206, Jee], [209, Jnn], [212, Jnn], [213, Jee], [213, Jnn], [214, Jee], [214, Jnn], [215, Jnn], [217, Jee], [218, Jnn], [229, Jee], [231, Jnn], [232, Jee], [241, Jee], [241, Jnn], [242, Jee], [242, Jnn], [243, Jee], [243, Jnn], [245, Jnn], [246, Jee], [246, Jnn], [250, Jee], [251, Jee], [252, Jnn], [253, Jnn], [255, Jee], [255, Jnn], [256, Jnn], [261, Jee], [261, Jnn], [262, Jee], [262, Jnn], [266, Jee], [267, Jnn], [268, Jnn], [269, Jnn], [272, Jee], [272, Jnn], [281, Jee], [320, Jee], [320, Jnn], [321, Jee], [321, Jnn], [323, Jee], [323, Jnn], [324, Jee], [324, Jnn], [325, Jee], [325, Jnn], [326, Jee], [326, Jnn], [327, Jee], [327, Jnn], [328, Jee], [328, Jnn], [329, Jee], [329, Jnn], [331, Jee], [331, Jnn], [332, Jee], [332, Jnn], [333, Jee], [333, Jnn], [336, Jee], [336, Jnn], [340, Jee], [340, Jnn]]

for repo in ['hera_cal', 'hera_qm', 'hera_filters', 'hera_notebook_templates', 'pyuvdata']:
    exec(f'from {repo} import __version__')
    print(f'{repo}: {__version__}')

hera_cal: 3.6.dev110+gcc0a13d
hera_qm: 2.1.5.dev6+g23b7cf7
hera_filters: 0.1.5

hera_notebook_templates: 0.1.dev734+g90f16f4
pyuvdata: 2.4.2

print(f'Finished execution in {(time.time() - tstart) / 60:.2f} minutes.')

Finished execution in 45.71 minutes.

Second Round of Full Day RFI Flagging¶

• Figure 1: Waterfall of Maximum z-Score of Either Polarization Before Round 2 Flagging ¶

• Figure 2: Histogram of z-scores ¶

• Figure 3: Waterfall of Maximum z-Score of Either Polarization After Round 2 Flagging ¶

• Figure 4: Spectra of Time-Averaged z-Scores ¶

• Figure 5: Summary of Flags Before and After Round 2 Flagging ¶

Load z-scores¶

Figure 1: Waterfall of Maximum z-Score of Either Polarization Before Round 2 Flagging¶

Figure 2: Histogram of z-scores¶

Perform flagging¶

Show results of flagging¶

Figure 3: Waterfall of Maximum z-Score of Either Polarization After Round 2 Flagging¶

Figure 4: Spectra of Time-Averaged z-Scores¶

Figure 5: Summary of Flags Before and After Round 2 Flagging¶

Save results¶

Metadata¶

Second Round of Full Day RFI Flagging¶

• Figure 1: Waterfall of Maximum z-Score of Either Polarization Before Round 2 Flagging¶

• Figure 2: Histogram of z-scores¶

• Figure 3: Waterfall of Maximum z-Score of Either Polarization After Round 2 Flagging¶

• Figure 4: Spectra of Time-Averaged z-Scores¶

• Figure 5: Summary of Flags Before and After Round 2 Flagging¶

Load z-scores¶

Figure 1: Waterfall of Maximum z-Score of Either Polarization Before Round 2 Flagging¶

Figure 2: Histogram of z-scores¶

Perform flagging¶

Show results of flagging¶

Figure 3: Waterfall of Maximum z-Score of Either Polarization After Round 2 Flagging¶

Figure 4: Spectra of Time-Averaged z-Scores¶

Figure 5: Summary of Flags Before and After Round 2 Flagging¶

Save results¶

Metadata¶

• Figure 1: Waterfall of Maximum z-Score of Either Polarization Before Round 2 Flagging ¶

• Figure 2: Histogram of z-scores ¶

• Figure 3: Waterfall of Maximum z-Score of Either Polarization After Round 2 Flagging ¶

• Figure 4: Spectra of Time-Averaged z-Scores ¶

• Figure 5: Summary of Flags Before and After Round 2 Flagging ¶