From 8b3cb2cb164565ec698f5e565d76c3b94c2bd083 Mon Sep 17 00:00:00 2001
From: Michael Wilensky <dc-wile1@login8b.pri.cosma7.alces.network>
Date: Tue, 28 May 2024 21:32:10 +0100
Subject: [PATCH] fix 0 vis bug

---
 beam_example.py | 4 ++--
 1 file changed, 2 insertions(+), 2 deletions(-)

diff --git a/beam_example.py b/beam_example.py
index 71c28ea..d7670a5 100644
--- a/beam_example.py
+++ b/beam_example.py
@@ -211,7 +211,7 @@
     # Generate fluxes
     beta_ptsrc = -2.7
     ptsrc_amps = 10.**np.random.uniform(low=-1., high=2., size=args.Nptsrc)
-    fluxes = get_flux_from_ptsrc_amp(ptsrc_amps, freqs, beta_ptsrc)
+    fluxes = get_flux_from_ptsrc_amp(ptsrc_amps, freqs * 1e-6, beta_ptsrc) # Have to put this in MHz...
     print("pstrc amps (input):", ptsrc_amps[:5])
     np.save(os.path.join(output_dir, "ptsrc_amps0"), ptsrc_amps)
     np.save(os.path.join(output_dir, "ptsrc_coords0"), np.column_stack((ra, dec)).T)
@@ -276,7 +276,7 @@
     noise_var = autos[:, :, None] * autos[:, :, :, None] / (args.integration_depth * args.ch_wid)
 
     noise = (np.random.normal(scale=np.sqrt(noise_var)) + 1.j * np.random.normal(scale=np.sqrt(noise_var))) / np.sqrt(2)
-    data = _sim_vis + noise
+    data = _sim_vis + _sim_vis.swapaxes(-1,-2).conj() + noise # fix some zeros
     del _sim_vis # Save some memory
     del noise