PyPI - tabsim - Versions diffs - 0.0.1__tar.gz → 0.0.3__tar.gz - Mend

tabsim 0.0.1tar.gz → 0.0.3tar.gz

This diff represents the content of publicly available package versions that have been released to one of the supported registries. The information contained in this diff is provided for informational purposes only and reflects changes between package versions as they appear in their respective public registries.

Files changed (50) hide show

{tabsim-0.0.1 → tabsim-0.0.3}/PKG-INFO RENAMED Viewed

@@ -1,6 +1,6 @@
 Metadata-Version: 2.4
 Name: tabsim
-Version: 0.0.1
+Version: 0.0.3
 Summary: Trajectory-based Radio Frequency Interference (RFI) simulations for radio interferometeric data.
 Author-email: Chris Finlay <christopher.finlay@epfl.ch>
 Maintainer-email: Chris Finlay <christopher.finlay@epfl.ch>
@@ -702,12 +702,18 @@ Requires-Dist: regions; extra == "sat"
 Provides-Extra: ipy
 Requires-Dist: ipython; extra == "ipy"
 Requires-Dist: ipykernel; extra == "ipy"
+Provides-Extra: test
+Requires-Dist: pytest; extra == "test"
+Provides-Extra: docs
+Requires-Dist: sphinx; extra == "docs"
+Requires-Dist: sphinx-rtd-theme; extra == "docs"
 Dynamic: license-file
 # tab-sim
 [![DOI:10.1093/mnras/stad1979](https://zenodo.org/badge/DOI/10.1093/mnras/stad1979.svg)](https://doi.org/10.1093/mnras/stad1979)
 [![DOI:10.48550/arXiv.2502.00106](https://img.shields.io/badge/arXiv-2502.00106-b31b1b.svg)](https://doi.org/10.48550/arXiv.2502.00106)
+[![Documentation Status](https://readthedocs.org/projects/tab-sim/badge/?version=latest)](https://tab-sim.readthedocs.io/en/latest/?badge=latest)
 **T**r**A**jectory **B**ased RFI **Sim**ulations of radio interferometry data.
 A source to visibility model for RFI sources including certain near-field effects.
@@ -715,6 +721,10 @@ A source to visibility model for RFI sources including certain near-field effect
 `tab-sim` is written in [JAX](https://jax.readthedocs.io/en/latest/notebooks/quickstart.html)
 and [Dask](https://www.dask.org) and can therefore use GPUs and/or CPUs to perform larger than memory computation.
+## Documentation
+[https://tab-sim.readthedocs.io/](https://tab-sim.readthedocs.io/)
 ## Installation
 The following instructions are expected to work on Linux machine. If you are running Windows it is recommended to use WSL. If you are running Mac then it is best to start with a [conda](https://docs.conda.io/en/latest/)/[mamba](https://mamba.readthedocs.io/en/latest/index.html) environment and install `python-casacore` before continuing with the `pip` install method. If all else fails there is the [Docker install](#docker).

{tabsim-0.0.1 → tabsim-0.0.3}/README.md RENAMED Viewed

@@ -2,6 +2,7 @@
 [![DOI:10.1093/mnras/stad1979](https://zenodo.org/badge/DOI/10.1093/mnras/stad1979.svg)](https://doi.org/10.1093/mnras/stad1979)
 [![DOI:10.48550/arXiv.2502.00106](https://img.shields.io/badge/arXiv-2502.00106-b31b1b.svg)](https://doi.org/10.48550/arXiv.2502.00106)
+[![Documentation Status](https://readthedocs.org/projects/tab-sim/badge/?version=latest)](https://tab-sim.readthedocs.io/en/latest/?badge=latest)
 **T**r**A**jectory **B**ased RFI **Sim**ulations of radio interferometry data.
 A source to visibility model for RFI sources including certain near-field effects.
@@ -9,6 +10,10 @@ A source to visibility model for RFI sources including certain near-field effect
 `tab-sim` is written in [JAX](https://jax.readthedocs.io/en/latest/notebooks/quickstart.html)
 and [Dask](https://www.dask.org) and can therefore use GPUs and/or CPUs to perform larger than memory computation.
+## Documentation
+[https://tab-sim.readthedocs.io/](https://tab-sim.readthedocs.io/)
 ## Installation
 The following instructions are expected to work on Linux machine. If you are running Windows it is recommended to use WSL. If you are running Mac then it is best to start with a [conda](https://docs.conda.io/en/latest/)/[mamba](https://mamba.readthedocs.io/en/latest/index.html) environment and install `python-casacore` before continuing with the `pip` install method. If all else fails there is the [Docker install](#docker).

tabsim-0.0.3/docs/conf.py ADDED Viewed

@@ -0,0 +1,14 @@
+import os
+import sys
+sys.path.insert(0, os.path.abspath('..'))
+project = 'tab‑sim'
+author = 'Chris Finlay'
+extensions = [
+    'sphinx.ext.autodoc',
+    'sphinx.ext.napoleon',
+    'sphinx.ext.viewcode',
+]
+templates_path = ['_templates']
+exclude_patterns = []
+html_theme = 'sphinx_rtd_theme'

{tabsim-0.0.1 → tabsim-0.0.3}/pyproject.toml RENAMED Viewed

@@ -5,7 +5,7 @@ build-backend = "setuptools.build_meta"
 [project]
 name = "tabsim"
-version = "0.0.1"
+version = "0.0.3"
 license = {file = "LICENSE"}
 readme = "README.md"
 # dynamic = ["version", "readme"]
@@ -31,6 +31,8 @@ sat = ["regions"]
 # all = ["jax[cuda12]", "regions", "bdsf", "TGEpy @ git+https://github.com/chrisfinlay/TGEpy.git"]
 # all_cpu = ["regions", "bdsf", "TGEpy @ git+https://github.com/chrisfinlay/TGEpy.git"]
 ipy = ["ipython", "ipykernel"]
+test = ["pytest"]
+docs = ["sphinx", "sphinx-rtd-theme"]
 [project.urls]
 Repository = "https://github.com/chrisfinlay/tab-sim.git"

tabsim-0.0.3/tabsim/__init__.py ADDED Viewed

@@ -0,0 +1,6 @@
+try:
+    from importlib.metadata import version
+except ImportError:
+    from importlib_metadata import version
+__version__ = version("tabsim")

{tabsim-0.0.1 → tabsim-0.0.3}/tabsim/config.py RENAMED Viewed

@@ -5,7 +5,8 @@ import sys
 import shutil
 import collections.abc
-from typing import Tuple
+from typing import Tuple, Union, Optional
 from datetime import datetime
 import dask.array as da
@@ -13,13 +14,14 @@ from dask.array.core import Array
 import xarray as xr
 import numpy as np
 import pandas as pd
+from astropy.time import Time
 from tabsim.dask.observation import Observation
 from tabsim.sky import generate_random_sky
 from tabsim.plot import plot_uv, plot_src_alt, plot_angular_seps
 from tabsim.write import write_ms, mk_obs_name, mk_obs_dir
-from tabsim.jax.coordinates import calculate_fringe_frequency
-from tabsim.tle import get_visible_satellite_tles
+from tabsim.jax.coordinates import calculate_fringe_frequency, jd_to_mjd
+from tabsim.tle import get_visible_satellite_tles, id_generator
 from daskms import xds_from_ms
@@ -188,7 +190,7 @@ def load_sky_model(file_path: str, freqs: Array, src_type: str) -> tuple:
         raise KeyError("'src_type' must be one of {'point', 'gauss', 'exp'}.")
-def sigma_value(value: str | float | int, obs: Observation):
+def sigma_value(value: Union[str, float, int], obs: Observation):
     try:
         if isinstance(value, str):
             sigma = (np.mean(obs.noise_std) / np.sqrt(obs.n_time * obs.n_bl)).compute()
@@ -212,12 +214,9 @@ def load_obs(obs_spec: dict) -> Observation:
         return x
     if obs_["start_time_jd"]:
-        from tabascal.jax.coordinates import jd_to_mjd
         start_time_mjd = jd_to_mjd(obs_["start_time_jd"])
     elif obs_["start_time_isot"]:
-        from astropy.time import Time
-        from tabascal.jax.coordinates import jd_to_mjd
         start_time_mjd = jd_to_mjd(
             Time(obs_["start_time_isot"], format="isot", scale="ut1").jd
@@ -837,11 +836,14 @@ def print_fringe_freq_tle_sat(obs: Observation):
 def run_sim_config(
-    obs_spec: dict = None, config_path: str = None, spacetrack_path: str = None
+    obs_spec: Optional[dict] = None,
+    config_path: Optional[str] = None,
+    spacetrack_path: Optional[str] = None,
 ) -> Tuple[Observation, str]:
     from tabsim.tle import id_generator
     log_path = f"log_sim_{id_generator()}.txt"
     log = open(log_path, "w")
     backup = sys.stdout
@@ -859,7 +861,7 @@ def run_sim_config(
     obs = load_obs(obs_spec)
     add_astro_sources(obs, obs_spec)
     add_satellite_sources(obs, obs_spec)
-    if obs_spec["rfi_sources"]["tle_satellite"]["max_n_sat"] != 0:
+    if obs_spec["rfi_sources"]["tle_satellite"]["max_n_sat"] != 0 and spacetrack_path:
         add_tle_satellite_sources(obs, obs_spec, spacetrack_path)
     add_stationary_sources(obs, obs_spec)
     add_gains(obs, obs_spec)

tabsim-0.0.3/tabsim/data/aux_data/tles/2023-02-21-J8HE0T.json ADDED Viewed

@@ -0,0 +1 @@

+ {"COMMENT":{"0":"GENERATED VIA SPACETRACK.ORG API","1":"GENERATED VIA SPACETRACK.ORG API","2":"GENERATED VIA SPACETRACK.ORG API","3":"GENERATED VIA SPACETRACK.ORG API","4":"GENERATED VIA SPACETRACK.ORG API","5":"GENERATED VIA SPACETRACK.ORG API","6":"GENERATED VIA SPACETRACK.ORG API","7":"GENERATED VIA SPACETRACK.ORG API"},"ORIGINATOR":{"0":"18 SPCS","1":"18 SPCS","2":"18 SPCS","3":"18 SPCS","4":"18 SPCS","5":"18 SPCS","6":"18 SPCS","7":"18 SPCS"},"NORAD_CAT_ID":{"0":"20452","1":"38833","2":"38833","3":"38833","4":"38833","5":"45854","6":"45854","7":"45854"},"OBJECT_NAME":{"0":"NAVSTAR 18 (USA 50)","1":"NAVSTAR 67 (USA 239)","2":"NAVSTAR 67 (USA 239)","3":"NAVSTAR 67 (USA 239)","4":"NAVSTAR 67 (USA 239)","5":"NAVSTAR 79 (USA 304)","6":"NAVSTAR 79 (USA 304)","7":"NAVSTAR 79 (USA 304)"},"OBJECT_TYPE":{"0":"PAYLOAD","1":"PAYLOAD","2":"PAYLOAD","3":"PAYLOAD","4":"PAYLOAD","5":"PAYLOAD","6":"PAYLOAD","7":"PAYLOAD"},"CLASSIFICATION_TYPE":{"0":"U","1":"U","2":"U","3":"U","4":"U","5":"U","6":"U","7":"U"},"INTLDES":{"0":"90008A","1":"12053A","2":"12053A","3":"12053A","4":"12053A","5":"20041A","6":"20041A","7":"20041A"},"EPOCH":{"0":"2023-02-21 06:38:50","1":"2023-02-21 06:37:41","2":"2023-02-21 06:37:41","3":"2023-02-21 06:37:41","4":"2023-02-22 06:33:35","5":"2023-02-21 08:40:26","6":"2023-02-22 10:18:13","7":"2023-02-22 10:18:13"},"EPOCH_MICROSECONDS":{"0":"12736","1":"186496","2":"186496","3":"189952","4":"104608","5":"76288","6":"882176","7":"882176"},"MEAN_MOTION":{"0":"1.89597679","1":"2.0056438","2":"2.0056438","3":"2.00564379","4":"2.00564407","5":"2.00566443","6":"2.00566487","7":"2.00566487"},"ECCENTRICITY":{"0":"0.0013992","1":"0.0135647","2":"0.0135647","3":"0.013565","4":"0.0135663","5":"0.0030268","6":"0.0030262","7":"0.0030262"},"INCLINATION":{"0":"54.7164","1":"53.518","2":"53.518","3":"53.518","4":"53.5179","5":"55.7203","6":"55.7213","7":"55.7213"},"RA_OF_ASC_NODE":{"0":"175.9594","1":"193.1658","2":"193.1658","3":"193.1658","4":"193.125","5":"76.4335","6":"76.391","7":"76.391"},"ARG_OF_PERICENTER":{"0":"264.8777","1":"50.1444","2":"50.1444","3":"50.1442","4":"50.1485","5":"186.8094","6":"186.8192","7":"186.8192"},"MEAN_ANOMALY":{"0":"93.0539","1":"310.9883","2":"310.9883","3":"310.9885","4":"310.9878","5":"299.724","6":"350.8122","7":"350.8122"},"EPHEMERIS_TYPE":{"0":"0","1":"0","2":"0","3":"0","4":"0","5":"0","6":"0","7":"0"},"ELEMENT_SET_NO":{"0":"999","1":"999","2":"999","3":"999","4":"999","5":"999","6":"999","7":"999"},"REV_AT_EPOCH":{"0":"23344","1":"7602","2":"7513","3":"7513","4":"7515","5":"1973","6":"1974","7":"1975"},"BSTAR":{"0":"0","1":"0","2":"0","3":"0","4":"0","5":"0","6":"0","7":"0"},"MEAN_MOTION_DOT":{"0":"0.00000009","1":"0.00000013","2":"0.00000013","3":"0.00000013","4":"0.0000002","5":"-0.00000013","6":"-0.00000024","7":"-0.00000024"},"MEAN_MOTION_DDOT":{"0":"0","1":"0","2":"0","3":"0","4":"0","5":"0","6":"0","7":"0"},"FILE":{"0":"3819803","1":"3818603","2":"3819803","3":"3821329","4":"3824148","5":"3821329","6":"3823581","7":"3824148"},"TLE_LINE0":{"0":"0 NAVSTAR 18 (USA 50)","1":"0 NAVSTAR 67 (USA 239)","2":"0 NAVSTAR 67 (USA 239)","3":"0 NAVSTAR 67 (USA 239)","4":"0 NAVSTAR 67 (USA 239)","5":"0 NAVSTAR 79 (USA 304)","6":"0 NAVSTAR 79 (USA 304)","7":"0 NAVSTAR 79 (USA 304)"},"TLE_LINE1":{"0":"1 20452U 90008A 23052.27696774 .00000009 00000-0 00000-0 0 9999","1":"1 38833U 12053A 23052.27617114 .00000013 00000-0 00000+0 0 9990","2":"1 38833U 12053A 23052.27617114 .00000013 00000-0 00000-0 0 9991","3":"1 38833U 12053A 23052.27617118 .00000013 00000-0 00000-0 0 9995","4":"1 38833U 12053A 23053.27332297 .00000020 00000-0 00000-0 0 9996","5":"1 45854U 20041A 23052.36141292 -.00000013 00000-0 00000-0 0 9998","6":"1 45854U 20041A 23053.42932734 -.00000024 00000-0 00000+0 0 9996","7":"1 45854U 20041A 23053.42932734 -.00000024 00000-0 00000-0 0 9997"},"TLE_LINE2":{"0":"2 20452 54.7164 175.9594 0013992 264.8777 93.0539 1.89597679233443","1":"2 38833 53.5180 193.1658 0135647 50.1444 310.9883 2.00564380 76021","2":"2 38833 53.5180 193.1658 0135647 50.1444 310.9883 2.00564380 75132","3":"2 38833 53.5180 193.1658 0135650 50.1442 310.9885 2.00564379 75134","4":"2 38833 53.5179 193.1250 0135663 50.1485 310.9878 2.00564407 75157","5":"2 45854 55.7203 76.4335 0030268 186.8094 299.7240 2.00566443 19736","6":"2 45854 55.7213 76.3910 0030262 186.8192 350.8122 2.00566487 19745","7":"2 45854 55.7213 76.3910 0030262 186.8192 350.8122 2.00566487 19756"},"OBJECT_ID":{"0":"1990-008A","1":"2012-053A","2":"2012-053A","3":"2012-053A","4":"2012-053A","5":"2020-041A","6":"2020-041A","7":"2020-041A"},"OBJECT_NUMBER":{"0":"20452","1":"38833","2":"38833","3":"38833","4":"38833","5":"45854","6":"45854","7":"45854"},"SEMIMAJOR_AXIS":{"0":"27574.850","1":"26560.279","2":"26560.279","3":"26560.279","4":"26560.277","5":"26560.097","6":"26560.093","7":"26560.093"},"PERIOD":{"0":"759.503","1":"717.974","2":"717.974","3":"717.974","4":"717.974","5":"717.967","6":"717.966","7":"717.966"},"APOGEE":{"0":"21235.298","1":"20542.427","2":"20542.427","3":"20542.435","4":"20542.467","5":"20262.354","6":"20262.334","7":"20262.334"},"PERIGEE":{"0":"21158.133","1":"19821.862","2":"19821.862","3":"19821.854","4":"19821.817","5":"20101.570","6":"20101.582","7":"20101.582"},"DECAYED":{"0":"0","1":"0","2":"0","3":"0","4":"0","5":"0","6":"0","7":"0"},"Fetch_Timestamp":{"0":"2025-06-12T20:52:43.524","1":"2025-06-12T20:52:43.524","2":"2025-06-12T20:52:43.524","3":"2025-06-12T20:52:43.524","4":"2025-06-12T20:52:43.524","5":"2025-06-12T20:52:43.524","6":"2025-06-12T20:52:43.524","7":"2025-06-12T20:52:43.524"}}

{tabsim-0.0.1 → tabsim-0.0.3}/tabsim/jax/coordinates.py RENAMED Viewed

@@ -1,6 +1,7 @@
 import jax.numpy as jnp
 from jax import jit, vmap, jacrev, config, Array
 from tabsim.jax.utils import jit_with_doc
+from jax.lax import scan
 config.update("jax_enable_x64", True)
@@ -690,7 +691,7 @@ def calculate_fringe_frequency(
     r_ecef = xyz_to_itrf(rfi_xyz, gsa)
     s_ecef = r_ecef - jnp.mean(ants_itrf, axis=0)
     s_hat_ecef = s_ecef / jnp.linalg.norm(s_ecef, axis=-1, keepdims=True)
-    s_hat_dot = jnp.gradient(s_hat_ecef, times, axis=0)
+    s_hat_dot = jnp.gradient(s_hat_ecef, jnp.diff(times[:2])[0], axis=0)
     a1, a2 = jnp.triu_indices(len(ants_itrf), 1)
     bl_ecef = ants_itrf[a1] - ants_itrf[a2]
@@ -1080,3 +1081,256 @@ def orbit_fisher(times: Array, orbit_params: Array, RIC_std: Array) -> Array:
     fisher = vmap(propagate, in_axes=(0, None))(J, F).sum(axis=0)
     return fisher
+# New Orbit Model with Keplerian Elements
+def R_z(angle):
+    c, s = jnp.cos(angle), jnp.sin(angle)
+    return jnp.array([[c, -s, 0], [s, c, 0], [0, 0, 1]])
+def R_x(angle):
+    c, s = jnp.cos(angle), jnp.sin(angle)
+    return jnp.array([[1, 0, 0], [0, c, -s], [0, s, c]])
+def solve_kepler(M, e, max_iter=100):
+    """Newton-Raphson solver for Kepler's equation."""
+    def newton_step(E, _):
+        f = E - e * jnp.sin(E) - M
+        f_prime = 1 - e * jnp.cos(E)
+        E_new = E - f / f_prime
+        return E_new, E_new
+    E0 = M
+    E_final, _ = scan(newton_step, E0, None, length=max_iter)
+    return E_final
+def kepler_orbit(times, t_epoch, params):
+    """
+    Compute orbit position over time for a single satellite.
+    times: (T,)
+    params: (6,) = [a, e, i, Ω, ω, M0]
+    Returns: (T, 3)
+    """
+    a, e, i_deg, Ω_deg, ω_deg, M0_deg = params
+    mu = 398600.4418  # km^3/s^2
+    n = jnp.sqrt(mu / jnp.abs(a) ** 3) * 86400
+    i = jnp.radians(i_deg)
+    Ω = jnp.radians(Ω_deg)
+    ω = jnp.radians(ω_deg)
+    M_0 = jnp.radians(M0_deg)
+    M = M_0 + n * (times - t_epoch)  # Mean anomaly
+    E = vmap(solve_kepler, (0, None))(jnp.atleast_1d(M), e)
+    nu = 2 * jnp.arctan2(
+        jnp.sqrt(1 + e) * jnp.sin(E / 2), jnp.sqrt(1 - e) * jnp.cos(E / 2)
+    )
+    r = a * (1 - e * jnp.cos(E))
+    x_orb = r * jnp.cos(nu)
+    y_orb = r * jnp.sin(nu)
+    z_orb = jnp.zeros_like(x_orb)
+    R = R_z(Ω) @ R_x(i) @ R_z(ω)
+    pos_orb = jnp.stack([x_orb, y_orb, z_orb], axis=1)  # (T, 3)
+    pos_inertial = pos_orb @ R.T  # (T, 3)
+    return pos_inertial * 1e3
+def kepler_orbit_many(
+    times_jd: Array,
+    epoch_jd: Array,
+    elements: Array,
+    # semi_major: Array,
+    # eccentricity: Array,
+    # inclination: Array,
+    # ra_of_asc_node: Array,
+    # arg_of_peri: Array,
+    # mean_anomaly: Array,
+):
+    """Propagate Keplerian orbits for a set of satellites using orbital elements as defined in a TLE.
+    Parameters
+    ----------
+    times_jd : Array (n_time,)
+        Propagation times in Julian Date.
+    epoch_jd : Array (n_sat,)
+        Epoch of the TLE in Julian Date.
+    semi_major : Array (n_sat,)
+        Semi major axis of the orbit in km.
+    eccentricity : Array (n_sat,)
+        Eccentricity of the orbit.
+    inclination : Array (n_sat,)
+        Inclination of the orbit in degrees.
+    ra_of_asc_node : Array (n_sat,)
+        Right ascension of the ascending node in degrees.
+    arg_of_peri : Array (n_sat,)
+        Argument of the pericentre in degrees.
+    mean_anomaly : Array (n_sat,)
+        Mean anomaly of the orbit at the epoch time in degrees.
+    """
+    # elements = [
+    #     jnp.atleast_1d(x)
+    #     for x in [
+    #         semi_major,
+    #         eccentricity,
+    #         inclination,
+    #         ra_of_asc_node,
+    #         arg_of_peri,
+    #         mean_anomaly,
+    #     ]
+    # ]
+    pos = vmap(kepler_orbit, (None, 0, 0))(times_jd, epoch_jd, elements)
+    return pos
+def kepler_orbit_velocity(times_jd: Array, epoch_jd: float, elements: Array) -> Array:
+    vel_vmap = vmap(jacrev(kepler_orbit, argnums=(0)), in_axes=(0, None, None))
+    velocity = vel_vmap(times_jd, epoch_jd, elements).reshape(len(times_jd), 3)
+    return velocity
+def R_uvw(
+    times_jd: Array,
+    epoch_jd: float,
+    elements: Array,
+) -> Array:
+    """
+    Calculate the rotation matrices at each time step to transform from an Earth
+    centric reference frame (ECI) to a satellite centric frame given the
+    parameters of its (circular) orbit.
+    Parameters
+    ----------
+    times: ndarray (n_time,)
+        Times at which to evaluate the rotation matrices.
+    elevation: float
+        Elevation/Altitude of the orbit in metres.
+    inclination: float
+        Inclination angle of the orbit relative to the equatorial plane.
+    lon_asc_node: float
+        Longitude of the ascending node of the orbit. This is the longitude of
+        when the orbit crosses the equator from the south to the north.
+    periapsis: float
+        Perisapsis of the orbit. This is the angular starting point of the orbit
+        at t = 0.
+    Returns
+    -------
+    R: ndarray (3, 3)
+        The rotation matrix to orient to a satellite centric (RIC) frame defined
+        by the Radial, In-track, Cross-track components.
+    """
+    times_jd = jnp.asarray(times_jd)
+    elements = jnp.asarray(elements)
+    position = kepler_orbit(times_jd, epoch_jd, elements)
+    velocity = kepler_orbit_velocity(times_jd, epoch_jd, elements)
+    vel_hat = velocity / jnp.linalg.norm(velocity, axis=-1, keepdims=True)
+    u_hat = position / jnp.linalg.norm(position, axis=-1, keepdims=True)
+    w_hat = jnp.cross(u_hat, vel_hat)
+    v_hat = jnp.cross(w_hat, u_hat)
+    R = jnp.stack([u_hat.T, v_hat.T, w_hat.T])
+    return R
+def kepler_RIC_dev(
+    times_jd: Array,
+    epoch_jd: float,
+    true_elements: Array,
+    estimated_elements: Array,
+) -> Array:
+    """
+    Calculate the Radial (R), In-track (I) and Cross-track (C) deviations
+    between two circular orbits given their orbit parameters at many time steps.
+    Parameters
+    ----------
+    times: array (n_time,)
+        Times at which to evaluate the RIC deviations.
+    true_orb_params: ndarray (4,)
+        Orbit parameters (elevation, inclination, lon_asc_node, periapsis) of
+        the object of most interest. The more accurate measurement when
+        comparing the same object.
+    est_orb_params: array (elevation, inclination, lon_asc_node, periapsis)
+        Estimated orbit parameters.
+    Returns
+    -------
+    RIC: ndarray (n_time, 3)
+        Radial, In-track and Cross-track coordinates, in metres, of the orbiting
+        body relative to the orbit defined by 'true_orbit_params'.
+    """
+    times_jd = jnp.asarray(times_jd)
+    true_elements = jnp.asarray(true_elements)
+    estimated_elements = jnp.asarray(estimated_elements)
+    true_xyz = kepler_orbit(times_jd, epoch_jd, true_elements)
+    est_xyz = kepler_orbit(times_jd, epoch_jd, estimated_elements)
+    R = R_uvw(times_jd, epoch_jd, true_elements)
+    true_uvw = vmap(lambda x, y: x @ y, in_axes=(-1, 0))(R, true_xyz)
+    est_uvw = vmap(lambda x, y: x @ y, in_axes=(-1, 0))(R, est_xyz)
+    return est_uvw - true_uvw
+def kepler_orbit_fisher(
+    times_jd: Array, epoch_jd: float, elements: Array, RIC_std: Array
+) -> Array:
+    """
+    Calculate the inverse covariance (Fisher) matrix in orbital elements
+    induced by errors in the RIC frame of an orbiting object. This is
+    essentially linear uncertainty propagation assuming Gaussian errors.
+    Parameters
+    ----------
+    times: ndarray (n_time,)
+        Times at which the RIC covariance/standard deviations is defined.
+    orbit_params: ndarray (4,)
+        Orbit parameters (elevation, inclination, lon_asc_node, periapsis) for
+        the orbit defining the origin of the RIC frame.
+    RIC_std: ndarray (3,)
+        The standard deviation in the Radial, In-track and Cross-track
+        directions.
+    Returns
+    -------
+    fisher: ndarray (6, 6)
+        The inverse covariance (Fisher) matrix for the orbit parameters induced
+        by the orbit uncertainties in the RIC frame.
+    """
+    times_jd = jnp.asarray(times_jd)
+    elements = jnp.asarray(elements)
+    RIC_std = jnp.asarray(RIC_std)
+    J = jacrev(kepler_RIC_dev, argnums=(3,))(times_jd, epoch_jd, elements, elements)[0]
+    F = jnp.diag(1.0 / RIC_std**2)
+    def propagate(J, F):
+        return J.T @ F @ J
+    fisher = vmap(propagate, in_axes=(0, None))(J, F).sum(axis=0)
+    return fisher

{tabsim-0.0.1 → tabsim-0.0.3}/tabsim/scripts/sim_vis.py RENAMED Viewed

@@ -1,10 +1,14 @@
 import argparse
 import os
+import sys
+from pathlib import Path
 from tabsim.config import load_config, run_sim_config
+from typing import Union
-def get_abs_path(rel_path: str | None, work_dir: str) -> str | None:
+def get_abs_path(rel_path: Union[str, None], work_dir: str) -> Union[str, None]:
     if rel_path:
         return os.path.abspath(os.path.join(work_dir, rel_path))
     else:
@@ -63,6 +67,11 @@ def main():
     args = parser.parse_args()
     rfi_amp = args.rfi_amp
     spacetrack_path = args.spacetrack
+    config_path = Path(args.config_path)
+    if not config_path.is_file():
+        print(f"Error: config file not found: {config_path}", file=sys.stderr)
+        sys.exit(1)  # Exit with error code
     config_path = os.path.abspath(args.config_path)
     work_dir = os.path.split(config_path)[0]

{tabsim-0.0.1 → tabsim-0.0.3}/tabsim/sky.py RENAMED Viewed

@@ -2,8 +2,10 @@ import numpy as np
 import jax.numpy as jnp
 import dask.array as da
+from typing import Optional
-def uniform_points_disk(radius: float, n_src: int, random_seed: int | None = None):
+def uniform_points_disk(radius: float, n_src: int, random_seed: Optional[int] = None):
     """
     Generate uniformly distributed random points on a disk.

{tabsim-0.0.1 → tabsim-0.0.3}/tabsim/tle.py RENAMED Viewed

@@ -21,8 +21,10 @@ from glob import glob
 from importlib.resources import files
+from typing import Optional
-def make_tle_dir(tle_dir: str | None):
+def make_tle_dir(tle_dir: Optional[str]):
     if tle_dir:
         tle_dir = os.path.abspath(tle_dir)
@@ -89,7 +91,7 @@ def get_tles_by_id(
     epoch_jd: float,
     window_days: float = 1.0,
     limit: int = 2000,
-    tle_dir: str | None = None,
+    tle_dir: Optional[str] = None,
 ) -> pd.DataFrame:
     tle_dir = make_tle_dir(tle_dir)
@@ -166,7 +168,7 @@ def get_tles_by_name(
     epoch_jd: float,
     window_days: float = 1.0,
     limit: int = 10000,
-    tle_dir: str | None = None,
+    tle_dir: Optional[str] = None,
 ) -> pd.DataFrame:
     tle_dir = make_tle_dir(tle_dir)
@@ -291,7 +293,7 @@ def get_visible_satellite_tles(
     min_elevation: float,
     names: ArrayLike = [],
     norad_ids: ArrayLike = [],
-    tle_dir: str | None = None,
+    tle_dir: Optional[str] = None,
 ) -> tuple:
     """Get the TLEs corresponding to satellites that satisfy the conditions given.

{tabsim-0.0.1 → tabsim-0.0.3}/tabsim/write.py RENAMED Viewed

@@ -1,6 +1,6 @@
 from __future__ import annotations
-from typing import TYPE_CHECKING
+from typing import TYPE_CHECKING, Optional
 if TYPE_CHECKING:
     from tabsim.dask.observation import Observation
@@ -12,9 +12,11 @@ import dask
 import dask.array as da
 import numpy as np
 import xarray as xr
-from daskms import Dataset, xds_to_table
+from daskms import Dataset, xds_to_table, xds_from_ms
 import numpy as np
+from dask.array.core import Array
 def rm_dir(path: str, overwrite: bool = True):
     """Remove a directory but check for existence and overwrite flag before.
@@ -73,7 +75,7 @@ def mk_obs_dir(output_path: str, obs_name: str, overwrite: bool = True) -> tuple
     return save_path, zarr_path, ms_path
-def mk_obs_name(prefix: str, obs: Observation, suffix: str = None) -> str:
+def mk_obs_name(prefix: str, obs: Observation, suffix: Optional[str] = None) -> str:
     """Construct an observation name based on the parameters of the observation and an additional prefix.
     Parameters
@@ -381,6 +383,40 @@ def get_stationary_rfi_data(obs: Observation):
     return rfi_stat
+def add_to_ms(ds: Dataset, ms_path: str, flags: Optional[Array] = None):
+    n_time = ds.attrs["n_time"]
+    n_freq = ds.attrs["n_freq"]
+    n_corr = 1
+    n_bl = ds.attrs["n_bl"]
+    n_row = n_time * n_bl
+    xds_ms = xds_from_ms(ms_path)[0]
+    dims = ["row", "chan", "corr"]
+    chunks = {k: v for k, v in xds_ms.chunks.items() if k in dims}
+    vis_rfi = ds.vis_rfi.data.reshape(n_row, n_freq, n_corr)
+    vis_rfi = xr.DataArray(
+        da.stack(
+            [vis_rfi, da.zeros_like(vis_rfi), da.zeros_like(vis_rfi), vis_rfi], axis=2
+        ),
+        dims=("row", "chan", "corr"),
+    )
+    xds_ms = xds_ms.assign(DATA=xds_ms.DATA + vis_rfi)
+    cols = "DATA"
+    if flags:
+        flags = xr.DataArray(
+            ds.flags.data.reshape(n_row, n_freq, n_corr), dims=("row", "chan", "corr")
+        )
+        xds_ms = xds_ms.assign(FLAG=flags)
+        cols.append("FLAG")
+    dask.compute(xds_to_table([xds_ms], ms_path, cols))
 def write_ms(
     ds: Dataset,
     ms_path: str,

{tabsim-0.0.1 → tabsim-0.0.3}/tabsim.egg-info/PKG-INFO RENAMED Viewed

@@ -1,6 +1,6 @@
 Metadata-Version: 2.4
 Name: tabsim
-Version: 0.0.1
+Version: 0.0.3
 Summary: Trajectory-based Radio Frequency Interference (RFI) simulations for radio interferometeric data.
 Author-email: Chris Finlay <christopher.finlay@epfl.ch>
 Maintainer-email: Chris Finlay <christopher.finlay@epfl.ch>
@@ -702,12 +702,18 @@ Requires-Dist: regions; extra == "sat"
 Provides-Extra: ipy
 Requires-Dist: ipython; extra == "ipy"
 Requires-Dist: ipykernel; extra == "ipy"
+Provides-Extra: test
+Requires-Dist: pytest; extra == "test"
+Provides-Extra: docs
+Requires-Dist: sphinx; extra == "docs"
+Requires-Dist: sphinx-rtd-theme; extra == "docs"
 Dynamic: license-file
 # tab-sim
 [![DOI:10.1093/mnras/stad1979](https://zenodo.org/badge/DOI/10.1093/mnras/stad1979.svg)](https://doi.org/10.1093/mnras/stad1979)
 [![DOI:10.48550/arXiv.2502.00106](https://img.shields.io/badge/arXiv-2502.00106-b31b1b.svg)](https://doi.org/10.48550/arXiv.2502.00106)
+[![Documentation Status](https://readthedocs.org/projects/tab-sim/badge/?version=latest)](https://tab-sim.readthedocs.io/en/latest/?badge=latest)
 **T**r**A**jectory **B**ased RFI **Sim**ulations of radio interferometry data.
 A source to visibility model for RFI sources including certain near-field effects.
@@ -715,6 +721,10 @@ A source to visibility model for RFI sources including certain near-field effect
 `tab-sim` is written in [JAX](https://jax.readthedocs.io/en/latest/notebooks/quickstart.html)
 and [Dask](https://www.dask.org) and can therefore use GPUs and/or CPUs to perform larger than memory computation.
+## Documentation
+[https://tab-sim.readthedocs.io/](https://tab-sim.readthedocs.io/)
 ## Installation
 The following instructions are expected to work on Linux machine. If you are running Windows it is recommended to use WSL. If you are running Mac then it is best to start with a [conda](https://docs.conda.io/en/latest/)/[mamba](https://mamba.readthedocs.io/en/latest/index.html) environment and install `python-casacore` before continuing with the `pip` install method. If all else fails there is the [Docker install](#docker).

{tabsim-0.0.1 → tabsim-0.0.3}/tabsim.egg-info/SOURCES.txt RENAMED Viewed

@@ -1,6 +1,7 @@
 LICENSE
 README.md
 pyproject.toml
+docs/conf.py
 tabsim/__init__.py
 tabsim/config.py
 tabsim/plot.py
@@ -29,6 +30,7 @@ tabsim/data/aux_data/satellite.ole
 tabsim/data/aux_data/satellite.rfimodel
 tabsim/data/aux_data/stationary.loc
 tabsim/data/aux_data/stationary.rfimodel
+tabsim/data/aux_data/tles/2023-02-21-J8HE0T.json
 tabsim/data/config_files/sim_config_base.yaml
 tabsim/data/config_files/tab_config_base.yaml
 tabsim/jax/__init__.py
@@ -39,4 +41,7 @@ tabsim/scripts/__init__.py
 tabsim/scripts/ast_region.py
 tabsim/scripts/flag_data.py
 tabsim/scripts/sim_vis.py
-tabsim/scripts/tle_sat_region.py
+tabsim/scripts/tle_sat_region.py
+tests/test_basic.py
+tests/test_sim-vis.py
+tests/test_version.py

{tabsim-0.0.1 → tabsim-0.0.3}/tabsim.egg-info/requires.txt RENAMED Viewed

@@ -11,6 +11,10 @@ spacetrack
 skyfield
 numpy<2.0.0
+[docs]
+sphinx
+sphinx-rtd-theme
 [gpu]
 jax[cuda12]
@@ -20,3 +24,6 @@ ipykernel
 [sat]
 regions
+[test]
+pytest

{tabsim-0.0.1 → tabsim-0.0.3}/tabsim.egg-info/top_level.txt RENAMED Viewed

@@ -1,3 +1,5 @@
 dist
+docs
 examples
 tabsim
+tests

tabsim-0.0.3/tests/test_basic.py ADDED Viewed

@@ -0,0 +1,5 @@
+def test_import():
+    import tabsim
+    assert hasattr(tabsim, "__version__")

tabsim-0.0.3/tests/test_sim-vis.py ADDED Viewed

@@ -0,0 +1,40 @@
+import sys
+import pytest
+from unittest.mock import patch
+from pathlib import Path
+from tabsim.scripts import sim_vis
+def test_sim_vis_help_exists(capsys):
+    test_args = ["sim-vis", "--help"]
+    with patch.object(sys, "argv", test_args):
+        with pytest.raises(SystemExit) as e:
+            sim_vis.main()
+    assert e.value.code == 0  # argparse exits with code 0 for --help
+    out = capsys.readouterr().out
+    assert "usage:" in out.lower()
+    assert "--help" in out
+def test_missing_config_file():
+    args = ["sim-vis", "--config", "nonexistent.yaml"]
+    with patch.object(sys, "argv", args), pytest.raises(SystemExit):
+        sim_vis.main()
+def test_simulation_runs_with_config(capsys):
+    config_path = (
+        Path(__file__).parent.parent / "examples" / "test" / "sim_test_16A.yaml"
+    )
+    assert config_path.is_file(), f"Missing config file: {config_path}"
+    args = ["sim-vis", "--config", str(config_path), "-o"]
+    with patch.object(sys, "argv", args):
+        sim_vis.main()
+    output = capsys.readouterr().out
+    assert "Total simulation time" in output

tabsim-0.0.3/tests/test_version.py ADDED Viewed

@@ -0,0 +1,6 @@
+import tabsim
+def test_version_format():
+    assert isinstance(tabsim.__version__, str)
+    assert len(tabsim.__version__) > 0
+    assert tabsim.__version__.count('.') >= 1  # basic semantic versioning check