Merge pull request #12 from tyler-a-cox/fix_get_reds

fix get_pos_red to restrict to one side of the uv-plane
tyler-a-cox · Apr 17, 2024 · aeeea83 · aeeea83
2 parents c61ffbf + 47b4b2f
commit aeeea83
Showing 1 changed file with 11 additions and 1 deletion.
diff --git a/fftvis/utils.py b/fftvis/utils.py
@@ -54,4 +54,14 @@ def get_pos_reds(antpos, decimals=3, include_autos=True):
 
                 ci += 1
 
-    return [reds[k] for k in reds]
+    reds_list = []
+    for k in reds:
+        red = reds[k]
+        ant1, ant2 = red[0]
+        _, bly, _ = antpos[ant2] - antpos[ant1]
+        if bly < 0:
+            reds_list.append([(bl[1], bl[0]) for bl in red])
+        else:
+            reds_list.append(red)
+
+    return reds_list