Update equatorial bounds for negative k and kp

lincc-frameworks · Jun 21, 2024 · 47eb123 · 47eb123
1 parent f32aa97
commit 47eb123
Showing 1 changed file with 4 additions and 2 deletions.
diff --git a/docs/notebooks/boundaries.ipynb b/docs/notebooks/boundaries.ipynb
@@ -107,8 +107,10 @@
     "\n",
     "\n",
     "def eq(k, kp):\n",
-    "    z = 2 / 3 * (2 - (kp + k) / nside)\n",
-    "    phi = np.pi / 4 / nside * (nside - kp + k)\n",
+    "    z = 2/3 * (2 - (abs(kp) + abs(k)) / nside)\n",
+    "    phi = np.pi/4/nside * (nside - abs(kp) + abs(k))\n",
+    "    phi = np.where(k >= 0, phi, -phi)\n",
+    "    phi = np.where(kp >= 0, phi, np.pi - phi)\n",
     "    return phi, z\n",
     "\n",
     "\n",