jolly-roger 0.0.2__py3-none-any.whl → 0.1.0__py3-none-any.whl

jolly_roger/_version.py

@@ -17,5 +17,5 @@ __version__: str
 __version_tuple__: VERSION_TUPLE
 version_tuple: VERSION_TUPLE
 
-__version__ = version = '0.0.2'
-__version_tuple__ = version_tuple = (0, 0, 2)
+__version__ = version = '0.1.0'
+__version_tuple__ = version_tuple = (0, 1, 0)

jolly_roger/hour_angles.py

@@ -8,15 +8,14 @@ from typing import Literal
 
 import astropy.units as u
 import numpy as np
-from astropy.coordinates import EarthLocation, SkyCoord, get_sun
+from astropy.coordinates import AltAz, EarthLocation, SkyCoord, get_sun
 from astropy.time import Time
 from casacore.tables import table
 
 from jolly_roger.logging import logger
 
 # Default location with XYZ based on mean of antenna positions
-ASKAP_XYZ_m = np.array([-2556146.66356375, 5097426.58592797, -2848333.08164107]) * u.m
-ASKAP = EarthLocation(*ASKAP_XYZ_m)
+ASKAP = EarthLocation.of_site("ASKAP")
 
 
 @dataclass
@@ -75,6 +74,9 @@ def _process_position(
         if position == "sun":
             logger.info("Getting sky-position of the sun")
             position = get_sun(times)
+        else:
+            logger.info(f"Getting sky-position of {position=}")
+            position = SkyCoord.from_name(position)
 
     if position is None:
         if ms_path is None:
@@ -83,8 +85,8 @@ def _process_position(
 
         with table(str(ms_path / "FIELD")) as tab:
             logger.info(f"Getting the sky-position from PHASE_DIR of {ms_path=}")
-            field_positions = tab.getcol("PHASE_DIR")
-            position = SkyCoord(field_positions[0] * u.rad)
+            field_positions = tab.getcol("PHASE_DIR")[:]
+            position = SkyCoord(*(field_positions * u.rad).squeeze())
 
     if isinstance(position, SkyCoord):
         return position
@@ -116,7 +118,7 @@ def make_hour_angles_for_ms(
     logger.info(f"Computing hour angles for {ms_path=}")
     with table(str(ms_path), ack=False) as tab:
         logger.info("Extracting timesteps and constructing time mapping")
-        times_mjds = tab.getcol("TIME_CENTROID")
+        times_mjds = tab.getcol("TIME_CENTROID")[:] * u.s
 
         # get unique time steps and make sure they are in their first appeared order
         times_mjds, indices = np.unique(times_mjds, return_index=True)
@@ -127,33 +129,28 @@ def make_hour_angles_for_ms(
     if whole_day:
         logger.info(f"Assuming a full day from {times_mjds} MJD (seconds)")
         time_step = times_mjds[1] - times_mjds[0]
-        times_mjds = times_mjds[0] + time_step * np.arange(
-            int(60 * 60 * 24 / time_step)
+        times_mjds = np.arange(
+            start=times_mjds[0],
+            stop=times_mjds[0] + 24 * u.hour,
+            step=time_step,
         )
 
-    times = Time(times_mjds / 60 / 60 / 24, format="mjd")
+    times = Time(times_mjds, format="mjd", scale="utc")
 
-    sky_position: SkyCoord = _process_position(
-        position=position, times=times, ms_path=ms_path
-    )
+    sky_position = _process_position(position=position, times=times, ms_path=ms_path)
 
     lst = times.sidereal_time("apparent", longitude=location.lon)
-    hour_angle = lst - sky_position.ra
-    mask = hour_angle > 12 * u.hourangle
-    hour_angle[mask] -= 24 * u.hourangle
+    hour_angle = (lst - sky_position.ra).wrap_at(12 * u.hourangle)
 
     logger.info("Creatring elevation curve")
-    sin_alt = np.arcsin(
-        np.sin(location.lat) * np.sin(sky_position[0].dec.rad)
-        + np.cos(location.lat) * np.cos(sky_position.dec.rad) * np.cos(hour_angle)
-    ).to(u.rad)
+    altaz = sky_position.transform_to(AltAz(obstime=times, location=location))
 
     return PositionHourAngles(
         hour_angle=hour_angle,
         time_mjds=times_mjds,
         location=location,
         position=sky_position,
-        elevation=sin_alt,
+        elevation=altaz.alt.to(u.rad),
         time=times,
         time_map=time_map,
     )

jolly_roger/uvws.py

@@ -44,12 +44,11 @@ def xyz_to_uvw(
     """
     b_xyz = baselines.b_xyz
 
-    # Getting the units right is important, mate
-    ha = hour_angles.hour_angle
-    ha = ha.to(u.rad)
+    # Convert HA to geocentric hour angle (at Greenwich meridian)
+    # This is why we subtract the location's longitude
+    ha = hour_angles.hour_angle - hour_angles.location.lon
 
     declination = hour_angles.position.dec
-    declination = declination.to(u.rad)
 
     # This is necessary for broadcastung in the matrix to work.
     # Should the position be a solar object like the sub its position
@@ -71,21 +70,22 @@ def xyz_to_uvw(
             [sin_ha, cos_ha, zeros],
             [-sin_dec * cos_ha, sin_dec * sin_ha, cos_dec],
             [
-                np.cos(declination) * np.cos(ha),
-                np.cos(declination) * np.sin(ha),
-                np.sin(declination),
+                cos_dec * cos_ha,
+                -cos_dec * sin_ha,
+                sin_dec,
             ],
         ]
     )
 
     # Every time this confuses me and I need the first mate to look over.
-    # b_xyz shape: (baselines, coord) where coord is XYZ
+    # b_xyz shape: (baselines, 3) where coord is XYZ
     # mat shape: (3, 3, timesteps)
-    # uvw shape: (baseline, coord, timesteps) where coord is UVW
-    uvw = np.einsum("ijk,lj->lik", mat, b_xyz, optimize=True)  # codespell:ignore lik
+    # uvw shape: (3, baseline, timesteps) where coord is UVW
+    uvw = np.einsum("ijk,lj->ilk", mat, b_xyz, optimize=True)  # codespell:ignore ilk
+    # i,j,k -> (3, 3, time)
+    # l,j -> (baseline, 3)
+    # i,l,k -> (3, baseline, time)
 
-    # Make order (coord, baseline, timesteps)
-    uvw = np.swapaxes(uvw, 0, 1)
     logger.debug(f"{uvw.shape=}")
 
     return UVWs(uvws=uvw, hour_angles=hour_angles, baselines=baselines)

{jolly_roger-0.0.2.dist-info → jolly_roger-0.1.0.dist-info}/METADATA

@@ -1,6 +1,6 @@
 Metadata-Version: 2.4
 Name: jolly-roger
-Version: 0.0.2
+Version: 0.1.0
 Summary: The pirate flagger
 Project-URL: Homepage, https://github.com/flint-crew/jolly-roger
 Project-URL: Bug Tracker, https://github.com/flint-crew/jolly-roger/issues

{jolly_roger-0.0.2.dist-info → jolly_roger-0.1.0.dist-info}/RECORD

@@ -1,14 +1,14 @@
 jolly_roger/__init__.py,sha256=7xiZLdeY-7sgrYGQ1gNdCjgCfqnoPXK7AeaHncY_DGU,204
-jolly_roger/_version.py,sha256=wO7XWlZte1hxA4mMvRc6zhNdGm74Nhhn2bfWRAxaKbI,511
+jolly_roger/_version.py,sha256=-LyU5F1uZDjn6Q8_Z6-_FJt_8RE4Kq9zcKdg1abSSps,511
 jolly_roger/_version.pyi,sha256=j5kbzfm6lOn8BzASXWjGIA1yT0OlHTWqlbyZ8Si_o0E,118
 jolly_roger/baselines.py,sha256=C_vC3v_ciU2T_si31oS0hUmsMNTQA0USxrm4118vYvY,4615
 jolly_roger/flagger.py,sha256=tlC-M_MpLpqOvkF544zw2EvOUpbSpasO2zlMlXMcxSs,3034
-jolly_roger/hour_angles.py,sha256=ChWTy69dkRN0R2HWEknHagv6W3xTXuSJJn9sAqBABCc,6160
+jolly_roger/hour_angles.py,sha256=SUUN_DcT3xzYp9JZ1U9ZcJpMjsfYETJ4D0YghUjee7Y,6096
 jolly_roger/logging.py,sha256=04YVHnF_8tKDkXNtXQ-iMyJ2BLV-qowbPAqqMFDxYE4,1338
 jolly_roger/py.typed,sha256=47DEQpj8HBSa-_TImW-5JCeuQeRkm5NMpJWZG3hSuFU,0
-jolly_roger/uvws.py,sha256=8G2E7bq8y5EEc2wQW5nli1WKsebZbBEHN8fhPzlinWk,10558
-jolly_roger-0.0.2.dist-info/METADATA,sha256=BtHEC57mNeu12PI6_pnDSkD8i767av0kQfTr_vINWuI,4172
-jolly_roger-0.0.2.dist-info/WHEEL,sha256=qtCwoSJWgHk21S1Kb4ihdzI2rlJ1ZKaIurTj_ngOhyQ,87
-jolly_roger-0.0.2.dist-info/entry_points.txt,sha256=ZwEZAe4DBn5nznVI0tP0a1wUinYouwXxxcZP6p7Pkvk,58
-jolly_roger-0.0.2.dist-info/licenses/LICENSE,sha256=7G-TthaPSOehr-pdj4TJydXj3eIUmerMbCUSatMr8hc,1522
-jolly_roger-0.0.2.dist-info/RECORD,,
+jolly_roger/uvws.py,sha256=42FngtA425Fk8QSlEIbl-9tEBd_-In0FtV2LYcnk6-0,10555
+jolly_roger-0.1.0.dist-info/METADATA,sha256=d40uefWM1e6HJtaqxNoak1PFBNFFZaBaS_18wno_Au8,4172
+jolly_roger-0.1.0.dist-info/WHEEL,sha256=qtCwoSJWgHk21S1Kb4ihdzI2rlJ1ZKaIurTj_ngOhyQ,87
+jolly_roger-0.1.0.dist-info/entry_points.txt,sha256=ZwEZAe4DBn5nznVI0tP0a1wUinYouwXxxcZP6p7Pkvk,58
+jolly_roger-0.1.0.dist-info/licenses/LICENSE,sha256=7G-TthaPSOehr-pdj4TJydXj3eIUmerMbCUSatMr8hc,1522
+jolly_roger-0.1.0.dist-info/RECORD,,