handle case where only reversed baseline is in FR_SPECTRA_FILE

HERA-Team · Feb 2, 2024 · b044d1c · b044d1c
1 parent 3703547
commit b044d1c
Showing 1 changed file with 6 additions and 2 deletions.
diff --git a/notebooks/single_baseline_postprocessing_and_pspec.ipynb b/notebooks/single_baseline_postprocessing_and_pspec.ipynb
@@ -496,8 +496,12 @@
     "    metadata = h5f[\"metadata\"]\n",
     "    bl_to_index_map = {tuple(ap): int(index) for index, antpairs in metadata[\"baseline_groups\"].items() for ap in antpairs}\n",
     "    spectrum_freqs = metadata[\"frequencies_MHz\"][()] * 1e6\n",
-    "    m_modes = metadata[\"erh_mode_integer_index\"][()]    \n",
-    "    mmode_spectrum = h5f[\"erh_mode_power_spectrum\"][:, :, bl_to_index_map[ANTPAIR]]"
+    "    m_modes = metadata[\"erh_mode_integer_index\"][()]\n",
+    "    if ANTPAIR in bl_to_index_map:\n",
+    "        mmode_spectrum = h5f[\"erh_mode_power_spectrum\"][:, :, bl_to_index_map[ANTPAIR]]\n",
+    "    else:\n",
+    "        # If ANTPAIR is not in the FR_SPECTRA_FILE, but the reverse is, also reverse the spectrum\n",
+    "        mmode_spectrum = np.flip(h5f[\"erh_mode_power_spectrum\"][:, :, bl_to_index_map[ANTPAIR[::-1]]], axis=0)"
    ]
   },
   {