xradio 0.0.40__py3-none-any.whl → 0.0.41__py3-none-any.whl

xradio/correlated_data/test__processing_set.py

@@ -0,0 +1,74 @@
+import unittest
+from xradio.src.xradio.correlated_data.processing_set import processing_set
+
+
+class TestProcessingSet(unittest.TestCase):
+
+    def setUp(self):
+        # Create a sample processing set
+        self.ps = processing_set()
+        self.ps["ms1"] = {
+            "partition_info": {
+                "obs_mode": "OBSERVE_TARGET#ON_SOURCE",
+                "spectral_window_name": "spw1",
+            },
+            "polarization": ["RR", "LL"],
+            "visibility": "vis1",
+            "frequency": [1e9, 2e9],
+        }
+        self.ps["ms2"] = {
+            "partition_info": {
+                "obs_mode": "OBSERVE_TARGET#CALIBRATE_POLARIZATION",
+                "spectral_window_name": "spw2",
+            },
+            "polarization": ["RR"],
+            "spectrum": "spec1",
+            "frequency": [2e9, 3e9],
+        }
+        self.ps["ms3"] = {
+            "partition_info": {
+                "obs_mode": "OBSERVE_TARGET#ON_SOURCE",
+                "spectral_window_name": "spw1",
+            },
+            "polarization": ["LL"],
+            "visibility": "vis2",
+            "frequency": [3e9, 4e9],
+        }
+
+    def test_summary(self):
+        # Test the summary method
+        summary = self.ps.summary()
+        self.assertEqual(
+            len(summary), 3
+        )  # Check the number of rows in the summary table
+
+    def test_get_ps_max_dims(self):
+        # Test the get_ps_max_dims method
+        max_dims = self.ps.get_ps_max_dims()
+        self.assertEqual(max_dims, {"frequency": 2, "polarization": 2})
+
+    def test_get_ps_freq_axis(self):
+        # Test the get_ps_freq_axis method
+        freq_axis = self.ps.get_ps_freq_axis()
+        self.assertEqual(len(freq_axis), 4)  # Check the length of the frequency axis
+
+    def test_sel(self):
+        # Test the sel method
+        subset = self.ps.sel(
+            obs_mode="OBSERVE_TARGET#ON_SOURCE", polarization=["RR", "LL"]
+        )
+        self.assertEqual(len(subset), 2)  # Check the number of MSs in the subset
+
+    def test_ms_sel(self):
+        # Test the ms_sel method
+        subset = self.ps.ms_sel(obs_mode="OBSERVE_TARGET#ON_SOURCE", polarization="RR")
+        self.assertEqual(len(subset), 1)  # Check the number of MSs in the subset
+
+    def test_ms_isel(self):
+        # Test the ms_isel method
+        subset = self.ps.ms_isel(obs_mode="OBSERVE_TARGET#ON_SOURCE", polarization="LL")
+        self.assertEqual(len(subset), 1)  # Check the number of MSs in the subset
+
+
+if __name__ == "__main__":
+    unittest.main()

{xradio-0.0.40.dist-info → xradio-0.0.41.dist-info}/METADATA

@@ -1,6 +1,6 @@
 Metadata-Version: 2.1
 Name: xradio
-Version: 0.0.40
+Version: 0.0.41
 Summary: Xarray Radio Astronomy Data IO
 Author-email: Jan-Willem Steeb <jsteeb@nrao.edu>
 License: BSD 3-Clause License
@@ -43,18 +43,14 @@ License-File: LICENSE.txt
 Requires-Dist: astropy
 Requires-Dist: dask
 Requires-Dist: distributed
-Requires-Dist: graphviper
 Requires-Dist: toolviper
-Requires-Dist: matplotlib
 Requires-Dist: numba >=0.57.0
 Requires-Dist: numpy
-Requires-Dist: prettytable
 Requires-Dist: pytest
 Requires-Dist: pytest-cov
 Requires-Dist: pytest-html
 Requires-Dist: s3fs
 Requires-Dist: scipy
-Requires-Dist: tqdm
 Requires-Dist: xarray
 Requires-Dist: zarr
 Requires-Dist: pyarrow
@@ -62,11 +58,6 @@ Requires-Dist: typeguard
 Requires-Dist: typing-extensions ; python_version < "3.10"
 Requires-Dist: python-casacore >=3.6.1 ; sys_platform != "darwin"
 Provides-Extra: docs
-Requires-Dist: jupyterlab ; extra == 'docs'
-Requires-Dist: ipykernel ; extra == 'docs'
-Requires-Dist: ipympl ; extra == 'docs'
-Requires-Dist: ipython ; extra == 'docs'
-Requires-Dist: jupyter-client ; extra == 'docs'
 Requires-Dist: nbsphinx ; extra == 'docs'
 Requires-Dist: recommonmark ; extra == 'docs'
 Requires-Dist: scanpydoc ; extra == 'docs'
@@ -75,6 +66,14 @@ Requires-Dist: sphinx-autosummary-accessors ; extra == 'docs'
 Requires-Dist: sphinx-rtd-theme ; extra == 'docs'
 Requires-Dist: twine ; extra == 'docs'
 Requires-Dist: pandoc ; extra == 'docs'
+Provides-Extra: interactive
+Requires-Dist: matplotlib ; extra == 'interactive'
+Requires-Dist: prettytable ; extra == 'interactive'
+Requires-Dist: jupyterlab ; extra == 'interactive'
+Requires-Dist: ipykernel ; extra == 'interactive'
+Requires-Dist: ipympl ; extra == 'interactive'
+Requires-Dist: ipython ; extra == 'interactive'
+Requires-Dist: jupyter-client ; extra == 'interactive'
 
 # xradio
 Xarray Radio Astronomy Data IO is still in development.

xradio-0.0.41.dist-info/RECORD

@@ -0,0 +1,75 @@
+xradio/__init__.py,sha256=WHBhQWQie3YQqfIxQBL3LKiKuUcN7ZL7sPMEcdWOp5E,382
+xradio/_utils/__init__.py,sha256=47DEQpj8HBSa-_TImW-5JCeuQeRkm5NMpJWZG3hSuFU,0
+xradio/_utils/common.py,sha256=GY_fjet9wLhnx9XUINTq6ONBoceyLWRXHXlrNla3j3o,3726
+xradio/_utils/list_and_array.py,sha256=_wznOiHra1pHrWNdtQPXhk-TAXJ8qb9fTGE0czQNpIo,2802
+xradio/_utils/schema.py,sha256=tdGyvQTd-rQivtkzmH9-1f7KMIB9KeSf57EN2v7Mw_o,6855
+xradio/_utils/_casacore/tables.py,sha256=aq6E_4RRAHdTBCwMKrVil1cWhFU2O980DNH9IlRKXLw,1280
+xradio/_utils/zarr/__init__.py,sha256=47DEQpj8HBSa-_TImW-5JCeuQeRkm5NMpJWZG3hSuFU,0
+xradio/_utils/zarr/common.py,sha256=egj3Zma0BUK0msOBDozMa-62rHrcxrjCNE5XkkZUq70,5332
+xradio/correlated_data/__init__.py,sha256=2oVnsJL4afv17FYpuHLH_bxspcJNWCWiT3wYCOn5jLk,423
+xradio/correlated_data/convert_msv2_to_processing_set.py,sha256=Hae-NOHSyxydlMEssiH3DpkYSJorDHTslsp3Dl8Dq08,6422
+xradio/correlated_data/correlated_xds.py,sha256=Q_qlQyAYC-9jhxvvR3wgGjPLy2tBctu2bvaD212BtVY,256
+xradio/correlated_data/load_processing_set.py,sha256=Sn8WQ6kdpnzngsGaW9o2fd6xOqOn2wM7yIKj45SXobQ,5526
+xradio/correlated_data/open_processing_set.py,sha256=BcW03m7ZoKcrdZsugczK6sbyu1YmPgoh2WZY5FYsrz8,4002
+xradio/correlated_data/processing_set.py,sha256=DXh34lr_-15kwy_ZkL3Ee25ptqcEwpQ9gbfJwU14zPI,11951
+xradio/correlated_data/schema.py,sha256=pJuWi0qwQoiJkGnb541f3Iq-YjiB6EsjDbbfW8Tc85c,59247
+xradio/correlated_data/test__processing_set.py,sha256=zWyUfCGr_FwxUqYZInNi3bYAH-ctmE_xB9ORSRaT4fs,2527
+xradio/correlated_data/_utils/__init__.py,sha256=Scu6rKJ2SpO8aG7F-xdTZcYfyWx0viV8gFh8E8ur_gI,93
+xradio/correlated_data/_utils/ms.py,sha256=m7cYGZGbdCD3uSPe1BldgRurZ4Y-K3vJr6aePUVyGjE,4310
+xradio/correlated_data/_utils/zarr.py,sha256=ehXlu0Xh_UZ5Xm2RnHCxESsRZ26c3DQAO5rqMK5MwTk,3947
+xradio/correlated_data/_utils/_ms/chunks.py,sha256=JTPk3il6fk570BjWZMoOAtsbvnLmqPcBv9EPY6A2yOs,2964
+xradio/correlated_data/_utils/_ms/conversion.py,sha256=Gz5_7vu0rUMrLxZJEMyCaCJ9UC9dKvVOemabVM67JCw,40940
+xradio/correlated_data/_utils/_ms/create_antenna_xds.py,sha256=w5yjNV4Ow1mL0zev9avLn7jfRfcBP-wmkpS0rGBGE4Y,17686
+xradio/correlated_data/_utils/_ms/create_field_and_source_xds.py,sha256=IPYbLx4580UGd2ibe2KtTThCCY4Ckz86reIL6s_8u4I,30881
+xradio/correlated_data/_utils/_ms/descr.py,sha256=dYK8mhXxODIh-dfqaOm-YZb7kmoN1N2golX_RFncO94,5215
+xradio/correlated_data/_utils/_ms/msv2_msv3.py,sha256=9AKs2HWly7Ivv_Cjr11dIPGmm33_rtSBoGF9wN5ZwEQ,116
+xradio/correlated_data/_utils/_ms/msv2_to_msv4_meta.py,sha256=gk9gU7g2Lk7dmaiLW8qecOEt574pRtGsCHnUnHXM3D0,1614
+xradio/correlated_data/_utils/_ms/msv4_info_dicts.py,sha256=kQpKNvgIki_ZIpC-lbKRqCKah7aLBoy5Q_NH1Q2vz3g,6895
+xradio/correlated_data/_utils/_ms/msv4_sub_xdss.py,sha256=qZUq6edNe5F9rLPrIEMXAPYehkk60WnVIq42MTqVho0,18440
+xradio/correlated_data/_utils/_ms/optimised_functions.py,sha256=47DEQpj8HBSa-_TImW-5JCeuQeRkm5NMpJWZG3hSuFU,0
+xradio/correlated_data/_utils/_ms/partition_queries.py,sha256=hEe3VAzGqtEPg3X4GUhvHaxyesjS6WVy5et421qSdZg,14573
+xradio/correlated_data/_utils/_ms/partitions.py,sha256=_KhRq8bSx2QxuWp9K57fLoLxcU6kvJ35e6wvJ-THbwc,12979
+xradio/correlated_data/_utils/_ms/subtables.py,sha256=_mpOOtHexqhiqEKt7S4LVqImJoNMJKSY18vNVw83r_U,3945
+xradio/correlated_data/_utils/_ms/_tables/load.py,sha256=IR3fdKlq8rgH4bHmB1JTtB5gSGuITIvErJEVjUA8rWM,1799
+xradio/correlated_data/_utils/_ms/_tables/load_main_table.py,sha256=npy30aRt_qtd8vfgGWYQgHRj1YON-ZYjfQi7zzLtToM,14806
+xradio/correlated_data/_utils/_ms/_tables/read.py,sha256=JLhn_8W2spLJy_3MjZ1mdMPKe7GPTq3VwuLeY-DiCjk,42937
+xradio/correlated_data/_utils/_ms/_tables/read_main_table.py,sha256=8AbNt-AxrhPK3EPRa7xqJXffxzIgfVsv1BDfoVJEXLU,26056
+xradio/correlated_data/_utils/_ms/_tables/read_subtables.py,sha256=JM6pGUQtjQR881u9VqakmbJjppEFq-EVKnEZ14JqnAw,12438
+xradio/correlated_data/_utils/_ms/_tables/table_query.py,sha256=q8EGFf_zIwHcHnvFJOn8hPh8zFZQ3f7BGbXvL3bHad4,555
+xradio/correlated_data/_utils/_ms/_tables/write.py,sha256=43XQ-tHhbhex0eUTRknNpPEEOnNR-w1lGCox9WZ9NHE,9540
+xradio/correlated_data/_utils/_ms/_tables/write_exp_api.py,sha256=GDEll8nMwkQGc6vosu4UddFL5_ld7WurRgF9hYFTRmU,15511
+xradio/correlated_data/_utils/_utils/cds.py,sha256=OpvKowSheIthUbcPEv2AoKmxlEt3DqJZS5C1AYh5z10,1179
+xradio/correlated_data/_utils/_utils/partition_attrs.py,sha256=JaePHts_A0EbB4K-0a_uC98RZ2EmfjB9pDSEI11oAwk,3401
+xradio/correlated_data/_utils/_utils/stokes_types.py,sha256=DMa8TmmS7BQ99Xm8c7ZjcRapMtLbrKVxrt4f0qUIOvg,561
+xradio/correlated_data/_utils/_utils/xds_helper.py,sha256=jZWR55e574j9NJ4Yf5HdHD7J7QtKdXbXStbblQ66WdE,13218
+xradio/correlated_data/_utils/_zarr/encoding.py,sha256=GENIlThV6a9CUCL6gIGlu9c6NR3OFWNos6mpxZjMwDc,536
+xradio/correlated_data/_utils/_zarr/read.py,sha256=O9DiwD2Gn8WiatQ-Q6WGGSwjsXwFktG4f81lM-mgcSg,7596
+xradio/correlated_data/_utils/_zarr/write.py,sha256=k5IfqtI44Dm4KBDiKFGhL5hN7kwNOulvVHmeP5Mi7N4,10043
+xradio/image/__init__.py,sha256=HAD0GfopIbhdxOYckyW6S9US_dSWmZrwIl3FHUzZwrE,435
+xradio/image/image.py,sha256=QoJ_BTLoMfeXJzU1yvtidBIhaMmjNA5_-6C3FWJRUeI,15635
+xradio/image/_util/__init__.py,sha256=M9lxD1Gc7kv0ucDEDbjLRuIEuESev-IG8j9EaCKUAkA,77
+xradio/image/_util/casacore.py,sha256=DmBTHUQ6870N5ARuFnYSfjZSLniJYgsjrsICUlCREYM,4234
+xradio/image/_util/common.py,sha256=vQs7Du9FSkb_D64Syt3IkiWkb_GmG9xhDPdgbpgP9Ok,9031
+xradio/image/_util/fits.py,sha256=gyGm06fuCKqVGK7uv-ObvQNfFawUDsIOa_nQyklM3Aw,329
+xradio/image/_util/image_factory.py,sha256=6tPzs20FTm2wEshHc1xqtTV7D0TbKxGLUKAVtvOc68I,10506
+xradio/image/_util/zarr.py,sha256=lhQqVRC1GEWClG3zRbuDr2IlQBfXeDqaLUJIN-MVMxA,1652
+xradio/image/_util/_casacore/__init__.py,sha256=OlsiRE40o1jSbBI4khgQQzgfDYbAlOMKIhO4UFlbGhg,41
+xradio/image/_util/_casacore/common.py,sha256=ky999eTCWta8w-uIs-7P7rPhZRLuh9yTuQXAxPvaPm4,1579
+xradio/image/_util/_casacore/xds_from_casacore.py,sha256=4puiE2_1-7Y1-0yK7L8FYyDp5Ab8hEhDjLvoXiGMx9E,42561
+xradio/image/_util/_casacore/xds_to_casacore.py,sha256=P6c-yoOjuVQkm07ApA7FFKfje4aPwV-MsRFKaRaPq9I,15338
+xradio/image/_util/_fits/xds_from_fits.py,sha256=kpZ2gdbDQtoUD1DdQZymup5OVnB3e73D66LruLYESsw,28445
+xradio/image/_util/_zarr/common.py,sha256=apMX_bF4Hr3pFGjnDFpp36KgmhTYAPBZquNkjBHrsXk,307
+xradio/image/_util/_zarr/xds_from_zarr.py,sha256=4b6KHmAcnrhBbCi-Z7e3Lm6l6wziJL1zaNIohmPAYDk,3601
+xradio/image/_util/_zarr/xds_to_zarr.py,sha256=wogXbwX8n3Sl9PHoc3_Y_LBowQsQ-94HZQFZ5NcxUZA,1624
+xradio/image/_util/_zarr/zarr_low_level.py,sha256=_skL70DhtmVjmxRJsfZaospt3rndB9l-2QoHFF1kAKE,12786
+xradio/schema/__init__.py,sha256=EzEMnOtN8G_wdjo8QBRKfq5MrYgfr_nt1pfunlI6i6Q,733
+xradio/schema/bases.py,sha256=5BiE6gAq2xmaJEyiaGbpCSoNek83ly9f0R0Rv1rx9DM,17081
+xradio/schema/check.py,sha256=Quugw5pC-c3c6C_JujSOQSfU_XVFIwymmg8wM0IZtXY,18521
+xradio/schema/dataclass.py,sha256=Byegqga9tTkgFDMK6JXwLkDo78OX906v3CBElH592N4,12356
+xradio/schema/metamodel.py,sha256=WjtW7pAVzcjLRWifRH3sQoOiN6TV810hARpOIz1M_gw,3845
+xradio/schema/typing.py,sha256=8-o6fZd99kJ4FVdgBYRTIRJ-wDqpcUNXzCTfJvl3TIw,10439
+xradio-0.0.41.dist-info/LICENSE.txt,sha256=9CYIJt7riOXo9AD0eXBZviLxo_HebD-2JJI8oiWtzfg,1807
+xradio-0.0.41.dist-info/METADATA,sha256=vzs61B5tZwpiZDr-p1PQKm_EedSfE8JeA48D_3Kz43k,4510
+xradio-0.0.41.dist-info/WHEEL,sha256=GV9aMThwP_4oNCtvEC2ec3qUYutgWeAzklro_0m4WJQ,91
+xradio-0.0.41.dist-info/top_level.txt,sha256=dQu27fGBZJ2Yk-gW5XeD-dZ76Xa4Xcvk60Vz-dwXp7k,7
+xradio-0.0.41.dist-info/RECORD,,

{xradio-0.0.40.dist-info → xradio-0.0.41.dist-info}/WHEEL

@@ -1,5 +1,5 @@
 Wheel-Version: 1.0
-Generator: setuptools (74.1.2)
+Generator: setuptools (75.1.0)
 Root-Is-Purelib: true
 Tag: py3-none-any
 

xradio/vis/_vis_utils/_ms/msv4_sub_xdss.py

@@ -1,306 +0,0 @@
-import toolviper.utils.logger as logger
-import time
-from typing import Tuple, Union
-
-import numpy as np
-import xarray as xr
-
-from xradio._utils.schema import (
-    column_description_casacore_to_msv4_measure,
-    convert_generic_xds_to_xradio_schema,
-)
-from .subtables import subt_rename_ids
-from ._tables.read import make_taql_where_between_min_max, load_generic_table
-
-
-def interpolate_to_time(
-    xds: xr.Dataset,
-    interp_time: Union[xr.DataArray, None],
-    message_prefix: str,
-    time_name: str = "time",
-) -> xr.Dataset:
-    """
-    Interpolate the time coordinate of the input xarray dataset to the
-    a data array. This can be used for example to interpolate a pointing_xds
-    to the time coord of the (main) MSv4, or similarly the ephemeris
-    data variables of a field_and_source_xds.
-
-    Uses interpolation method "linear", unless the source number of points is
-    1 in which case "nearest" is used, to avoid divide-by-zero issues.
-
-    Parameters:
-    ----------
-    xds : xr.Dataset
-        Xarray dataset to interpolate (presumably a pointing_xds or an xds of
-        ephemeris variables)
-    interp_time : Union[xr.DataArray, None]
-        Time axis to interpolate the dataset to (usually main MSv4 time)
-    message_prefix: str
-        A prefix for info/debug/etc. messages
-
-    Returns:
-    -------
-    interpolated_xds : xr.Dataset
-        xarray dataset with time axis interpolated to interp_time.
-    """
-    if interp_time is not None:
-        points_before = xds[time_name].size
-        if points_before > 1:
-            method = "linear"
-        else:
-            method = "nearest"
-        xds = xds.interp({time_name: interp_time}, method=method, assume_sorted=True)
-        points_after = xds[time_name].size
-        logger.debug(
-            f"{message_prefix}: interpolating the time coordinate "
-            f"from {points_before} to {points_after} points"
-        )
-
-    return xds
-
-
-def create_weather_xds(in_file: str):
-    """
-    Creates a Weather Xarray Dataset from a MS v2 WEATHER table.
-
-    Parameters
-    ----------
-    in_file : str
-        Input MS name.
-
-    Returns
-    -------
-    xr.Dataset
-        Weather Xarray Dataset.
-    """
-
-    dims_station_time = ["station_id", "time"]
-    to_new_data_variables = {
-        "H20": ["H2O", dims_station_time],
-        "IONOS_ELECTRON": ["IONOS_ELECTRON", dims_station_time],
-        "PRESSURE": ["PRESSURE", dims_station_time],
-        "REL_HUMIDITY": ["REL_HUMIDITY", dims_station_time],
-        "TEMPERATURE": ["TEMPERATURE", dims_station_time],
-        "DEW_POINT": ["DEW_POINT", dims_station_time],
-        "WIND_DIRECTION": ["WIND_DIRECTION", dims_station_time],
-        "WIND_SPEED": ["WIND_SPEED", dims_station_time],
-    }
-
-    to_new_coords = {
-        "ANTENNA_ID": ["station_id", ["station_id"]],
-        "TIME": ["time", ["time"]],
-    }
-
-    # Read WEATHER table into a Xarray Dataset.
-    try:
-        generic_weather_xds = load_generic_table(
-            in_file,
-            "WEATHER",
-            rename_ids=subt_rename_ids["WEATHER"],
-        )
-    except ValueError as _exc:
-        return None
-
-    weather_xds = xr.Dataset(attrs={"type": "weather"})
-    weather_xds = convert_generic_xds_to_xradio_schema(
-        generic_weather_xds, weather_xds, to_new_data_variables, to_new_coords
-    )
-
-    # correct expected types
-    weather_xds["station_id"] = weather_xds["station_id"].astype(np.int64)
-
-    return weather_xds
-
-
-def correct_generic_pointing_xds(
-    generic_pointing_xds: xr.Dataset, to_new_data_variables: dict[str, list]
-) -> xr.Dataset:
-    """
-    Takes a (generic) pointing_xds as read from a POINTING subtable of an MSv2
-    and tries to correct several deviations from the MSv2 specs seen in
-    common test data.
-    The problems fixed here include wrong dimensions:
-    - for example transposed dimensions with respect to the MSv2 specs (output
-    from CASA simulator),
-    - missing/additional unexpected dimensions when some of the columns are
-    empty (in the sense of "empty casacore cells").
-
-    This function modifies the data arrays of the data vars affected by such
-    issues.
-
-    Parameters
-    ----------
-    generic_pointing_xds: xr.Dataset
-        The generic pointing dataset (loaded from MSv2) to be fixed
-    to_new_data_variables: dict
-        The dict used for convert_generic_xds_to_xradio_schema, which gives all
-        the data variables relevant for the final MSv4 dataset.
-
-    Returns:
-    --------
-    xr.Dataset
-        Corrected dataset with dimensions conforming to MSv2 specs.
-    """
-
-    correct_pointing_xds = generic_pointing_xds.copy()
-
-    for key in generic_pointing_xds:
-        if key in to_new_data_variables:
-            data_var_name = to_new_data_variables[key]
-            # Corrects dim sizes of "empty cell" variables, such as empty DIRECTION, TARGET, etc.
-            if (
-                "dim_2" in generic_pointing_xds.sizes
-                and generic_pointing_xds.sizes["dim_2"] == 0
-            ):
-                # When all direction variables are "empty"
-                data_var_data = xr.DataArray(
-                    [[[[np.nan, np.nan]]]],
-                    dims=generic_pointing_xds.dims,
-                ).isel(n_polynomial=0, drop=True)
-                correct_pointing_xds[data_var_name].data = data_var_data
-
-            elif (
-                "dir" in generic_pointing_xds.sizes
-                and generic_pointing_xds.sizes["dir"] == 0
-            ):
-                # When some direction variables are "empty" but some are populated properly
-                if "dim_2" in generic_pointing_xds[key].sizes:
-                    data_var_data = xr.DataArray(
-                        generic_pointing_xds[key].values,
-                        dims=generic_pointing_xds[key].dims,
-                    )
-                else:
-                    shape = tuple(
-                        generic_pointing_xds.sizes[dim]
-                        for dim in ["TIME", "ANTENNA_ID"]
-                    ) + (2,)
-                    data_var_data = xr.DataArray(
-                        np.full(shape, np.nan),
-                        dims=generic_pointing_xds[key].dims,
-                    )
-                correct_pointing_xds[data_var_name].data = data_var_data
-
-    return correct_pointing_xds
-
-
-def create_pointing_xds(
-    in_file: str,
-    ant_xds_name_ids: xr.DataArray,
-    time_min_max: Union[Tuple[np.float64, np.float64], None],
-    interp_time: Union[xr.DataArray, None] = None,
-) -> xr.Dataset:
-    """
-    Creates a Pointing Xarray Dataset from an MS v2 POINTING (sub)table.
-
-    WIP: details of a few direction variables (and possibly moving some to attributes) to be
-    settled (see MSv4 spreadsheet).
-
-    Parameters
-    ----------
-    in_file : str
-        Input MS name.
-    ant_xds_name_ids : xr.Dataset
-        antenna_name data array from antenna_xds, with name/id information
-    time_min_max : tuple
-        min / max times values to constrain loading (from the TIME column)
-    interp_time : Union[xr.DataArray, None] (Default value = None)
-        interpolate time to this (presumably main dataset time)
-
-    Returns
-    -------
-    xr.Dataset
-         Pointing Xarray dataset
-    """
-    start = time.time()
-
-    time_ant_dims = ["time", "antenna_name"]
-    time_ant_dir_dims = time_ant_dims + ["sky_dir_label"]
-    to_new_data_variables = {
-        "DIRECTION": ["BEAM_POINTING", time_ant_dir_dims],
-        "ENCODER": ["DISH_MEASURED_POINTING", time_ant_dir_dims],
-        # => attribute?
-        "TARGET": ["TARGET", time_ant_dir_dims],
-        "POINTING_OFFSET": ["POINTING_OFFSET", time_ant_dir_dims],
-        "SOURCE_OFFSET": ["SOURCE_OFFSET", time_ant_dir_dims],
-        "OVER_THE_TOP": ["OVER_THE_TOP", time_ant_dims],
-    }
-
-    to_new_coords = {
-        "TIME": ["time", ["time"]],
-        # "ANTENNA_ID": ["antenna_name", ["antenna_name"]],
-        "dim_2": ["sky_dir_label", ["sky_dir_label"]],
-    }
-
-    taql_time_range = make_taql_where_between_min_max(
-        time_min_max, in_file, "POINTING", "TIME"
-    )
-
-    if taql_time_range is None:
-        taql_where = f"WHERE (ANTENNA_ID IN [{','.join(map(str, ant_xds_name_ids.antenna_id.values))}])"
-    else:
-        taql_where = (
-            taql_time_range
-            + f" AND (ANTENNA_ID IN [{','.join(map(str, ant_xds_name_ids.antenna_id.values))}])"
-        )
-    # Read POINTING table into a Xarray Dataset.
-    generic_pointing_xds = load_generic_table(
-        in_file,
-        "POINTING",
-        rename_ids=subt_rename_ids["POINTING"],
-        taql_where=taql_where,
-    )
-
-    if not generic_pointing_xds.data_vars:
-        # apparently empty MS/POINTING table => produce empty xds
-        return xr.Dataset()
-
-    # Checking a simple way of using only the one single coefficient of the polynomials
-    if "n_polynomial" in generic_pointing_xds.sizes:
-        size = generic_pointing_xds.sizes["n_polynomial"]
-        if size == 1:
-            generic_pointing_xds = generic_pointing_xds.sel({"n_polynomial": 0})
-
-    generic_pointing_xds = correct_generic_pointing_xds(
-        generic_pointing_xds, to_new_data_variables
-    )
-
-    pointing_xds = xr.Dataset(attrs={"type": "pointing"})
-    coords = {
-        "antenna_name": ant_xds_name_ids.sel(
-            antenna_id=generic_pointing_xds["ANTENNA_ID"]
-        ).data,
-        "sky_dir_label": ["ra", "dec"],
-    }
-    pointing_xds = pointing_xds.assign_coords(coords)
-    pointing_xds = convert_generic_xds_to_xradio_schema(
-        generic_pointing_xds, pointing_xds, to_new_data_variables, to_new_coords
-    )
-
-    # Add attributes specific to pointing_xds
-    if "TRACKING" in generic_pointing_xds.data_vars:
-        pointing_xds.attrs["tracking"] = generic_pointing_xds.data_vars[
-            "TRACKING"
-        ].values[0, 0]
-
-    # Move target from data_vars to attributes?
-    move_target_as_attr = False
-    if move_target_as_attr:
-        pointing_column_descriptions = generic_pointing_xds.attrs["other"]["msv2"][
-            "ctds_attrs"
-        ]["column_descriptions"]
-
-        target = generic_pointing_xds.data_vars["TARGET"]
-        pointing_xds.attrs["target"] = {
-            "dims": ["sky_dir_label"],
-            "data": target.values[0, 0].tolist(),
-            "attrs": column_description_casacore_to_msv4_measure(
-                pointing_column_descriptions["TARGET"]
-            ),
-        }
-    # TODO: move also source_offset/pointing_offset from data_vars to attrs?
-
-    pointing_xds = interpolate_to_time(pointing_xds, interp_time, "pointing_xds")
-
-    logger.debug(f"create_pointing_xds() execution time {time.time() - start:0.2f} s")
-
-    return pointing_xds

xradio-0.0.40.dist-info/RECORD

@@ -1,73 +0,0 @@
-xradio/__init__.py,sha256=WHBhQWQie3YQqfIxQBL3LKiKuUcN7ZL7sPMEcdWOp5E,382
-xradio/_utils/__init__.py,sha256=47DEQpj8HBSa-_TImW-5JCeuQeRkm5NMpJWZG3hSuFU,0
-xradio/_utils/common.py,sha256=GY_fjet9wLhnx9XUINTq6ONBoceyLWRXHXlrNla3j3o,3726
-xradio/_utils/list_and_array.py,sha256=_wznOiHra1pHrWNdtQPXhk-TAXJ8qb9fTGE0czQNpIo,2802
-xradio/_utils/schema.py,sha256=tdGyvQTd-rQivtkzmH9-1f7KMIB9KeSf57EN2v7Mw_o,6855
-xradio/_utils/_casacore/tables.py,sha256=aq6E_4RRAHdTBCwMKrVil1cWhFU2O980DNH9IlRKXLw,1280
-xradio/_utils/zarr/__init__.py,sha256=47DEQpj8HBSa-_TImW-5JCeuQeRkm5NMpJWZG3hSuFU,0
-xradio/_utils/zarr/common.py,sha256=egj3Zma0BUK0msOBDozMa-62rHrcxrjCNE5XkkZUq70,5332
-xradio/image/__init__.py,sha256=HAD0GfopIbhdxOYckyW6S9US_dSWmZrwIl3FHUzZwrE,435
-xradio/image/image.py,sha256=QoJ_BTLoMfeXJzU1yvtidBIhaMmjNA5_-6C3FWJRUeI,15635
-xradio/image/_util/__init__.py,sha256=M9lxD1Gc7kv0ucDEDbjLRuIEuESev-IG8j9EaCKUAkA,77
-xradio/image/_util/casacore.py,sha256=DmBTHUQ6870N5ARuFnYSfjZSLniJYgsjrsICUlCREYM,4234
-xradio/image/_util/common.py,sha256=vQs7Du9FSkb_D64Syt3IkiWkb_GmG9xhDPdgbpgP9Ok,9031
-xradio/image/_util/fits.py,sha256=gyGm06fuCKqVGK7uv-ObvQNfFawUDsIOa_nQyklM3Aw,329
-xradio/image/_util/image_factory.py,sha256=6tPzs20FTm2wEshHc1xqtTV7D0TbKxGLUKAVtvOc68I,10506
-xradio/image/_util/zarr.py,sha256=lhQqVRC1GEWClG3zRbuDr2IlQBfXeDqaLUJIN-MVMxA,1652
-xradio/image/_util/_casacore/__init__.py,sha256=OlsiRE40o1jSbBI4khgQQzgfDYbAlOMKIhO4UFlbGhg,41
-xradio/image/_util/_casacore/common.py,sha256=ky999eTCWta8w-uIs-7P7rPhZRLuh9yTuQXAxPvaPm4,1579
-xradio/image/_util/_casacore/xds_from_casacore.py,sha256=4puiE2_1-7Y1-0yK7L8FYyDp5Ab8hEhDjLvoXiGMx9E,42561
-xradio/image/_util/_casacore/xds_to_casacore.py,sha256=P6c-yoOjuVQkm07ApA7FFKfje4aPwV-MsRFKaRaPq9I,15338
-xradio/image/_util/_fits/xds_from_fits.py,sha256=kpZ2gdbDQtoUD1DdQZymup5OVnB3e73D66LruLYESsw,28445
-xradio/image/_util/_zarr/common.py,sha256=apMX_bF4Hr3pFGjnDFpp36KgmhTYAPBZquNkjBHrsXk,307
-xradio/image/_util/_zarr/xds_from_zarr.py,sha256=4b6KHmAcnrhBbCi-Z7e3Lm6l6wziJL1zaNIohmPAYDk,3601
-xradio/image/_util/_zarr/xds_to_zarr.py,sha256=wogXbwX8n3Sl9PHoc3_Y_LBowQsQ-94HZQFZ5NcxUZA,1624
-xradio/image/_util/_zarr/zarr_low_level.py,sha256=_skL70DhtmVjmxRJsfZaospt3rndB9l-2QoHFF1kAKE,12786
-xradio/schema/__init__.py,sha256=EzEMnOtN8G_wdjo8QBRKfq5MrYgfr_nt1pfunlI6i6Q,733
-xradio/schema/bases.py,sha256=5BiE6gAq2xmaJEyiaGbpCSoNek83ly9f0R0Rv1rx9DM,17081
-xradio/schema/check.py,sha256=Quugw5pC-c3c6C_JujSOQSfU_XVFIwymmg8wM0IZtXY,18521
-xradio/schema/dataclass.py,sha256=Byegqga9tTkgFDMK6JXwLkDo78OX906v3CBElH592N4,12356
-xradio/schema/metamodel.py,sha256=WjtW7pAVzcjLRWifRH3sQoOiN6TV810hARpOIz1M_gw,3845
-xradio/schema/typing.py,sha256=8-o6fZd99kJ4FVdgBYRTIRJ-wDqpcUNXzCTfJvl3TIw,10439
-xradio/vis/__init__.py,sha256=oOaHEnHp799cu72Zwo4WUQdrgFE2JCy-O0XVhaz3C2w,381
-xradio/vis/_processing_set.py,sha256=rvPAU5ksfODDDaJQBzBlZaHQDSDseV6UqCxN2Qa5gxs,12097
-xradio/vis/convert_msv2_to_processing_set.py,sha256=x5P9P8BZiK243iFi5kjn3i4gMgaGbozSN1sh1bgyYNI,5906
-xradio/vis/load_processing_set.py,sha256=sQmKG04grVn9jqOrhGc8K6B2-tk65NR0GBW8I8W62cI,5755
-xradio/vis/read_processing_set.py,sha256=8h1UDXFZqwObs6J6p5ef0FSA9ILW3A6I_6HMbQrbRFc,4170
-xradio/vis/schema.py,sha256=pQelyv3f6uIW12UKT_sB8KETyXpMX8klPeUOpq4dyd8,44858
-xradio/vis/_vis_utils/__init__.py,sha256=Scu6rKJ2SpO8aG7F-xdTZcYfyWx0viV8gFh8E8ur_gI,93
-xradio/vis/_vis_utils/ms.py,sha256=m7cYGZGbdCD3uSPe1BldgRurZ4Y-K3vJr6aePUVyGjE,4310
-xradio/vis/_vis_utils/zarr.py,sha256=d7T7P2rLzHTvoH7jh4FsiqkFq-PHsxRHUhM4mxbZI9M,3947
-xradio/vis/_vis_utils/_ms/chunks.py,sha256=JTPk3il6fk570BjWZMoOAtsbvnLmqPcBv9EPY6A2yOs,2964
-xradio/vis/_vis_utils/_ms/conversion.py,sha256=KTpv5CaPYVsqu3WQI-j4yZiNL3NWn6sev905tAyzq8I,38889
-xradio/vis/_vis_utils/_ms/create_antenna_xds.py,sha256=WMX1Av328Et0o56pTcva5tYSOFlu-mBxhJJHp33rb3k,17181
-xradio/vis/_vis_utils/_ms/create_field_and_source_xds.py,sha256=d_WlX_xA33rpHS8Dhqct8HmczZfteA2W7CHC1EPuZEw,30897
-xradio/vis/_vis_utils/_ms/descr.py,sha256=dYK8mhXxODIh-dfqaOm-YZb7kmoN1N2golX_RFncO94,5215
-xradio/vis/_vis_utils/_ms/msv2_msv3.py,sha256=9AKs2HWly7Ivv_Cjr11dIPGmm33_rtSBoGF9wN5ZwEQ,116
-xradio/vis/_vis_utils/_ms/msv2_to_msv4_meta.py,sha256=gk9gU7g2Lk7dmaiLW8qecOEt574pRtGsCHnUnHXM3D0,1614
-xradio/vis/_vis_utils/_ms/msv4_infos.py,sha256=47DEQpj8HBSa-_TImW-5JCeuQeRkm5NMpJWZG3hSuFU,0
-xradio/vis/_vis_utils/_ms/msv4_sub_xdss.py,sha256=DaLIqC05AZbkdc-XiahesYwfRFyLXNlWW4DoBv72d5I,10649
-xradio/vis/_vis_utils/_ms/optimised_functions.py,sha256=47DEQpj8HBSa-_TImW-5JCeuQeRkm5NMpJWZG3hSuFU,0
-xradio/vis/_vis_utils/_ms/partition_queries.py,sha256=hEe3VAzGqtEPg3X4GUhvHaxyesjS6WVy5et421qSdZg,14573
-xradio/vis/_vis_utils/_ms/partitions.py,sha256=_KhRq8bSx2QxuWp9K57fLoLxcU6kvJ35e6wvJ-THbwc,12979
-xradio/vis/_vis_utils/_ms/subtables.py,sha256=6--dh6betXhlxGejzcJJd8bYdiNkl5ppzLoyorofZ6c,3943
-xradio/vis/_vis_utils/_ms/_tables/load.py,sha256=IR3fdKlq8rgH4bHmB1JTtB5gSGuITIvErJEVjUA8rWM,1799
-xradio/vis/_vis_utils/_ms/_tables/load_main_table.py,sha256=_Ov4cs5t06RqfmrAHZoYwpb8K27b7iqjIkCMsJ0TR6o,14798
-xradio/vis/_vis_utils/_ms/_tables/read.py,sha256=ZrDHvUoyiAJu6P_2LIpedZgrDuWBwMi99RdMHrj6wyY,42555
-xradio/vis/_vis_utils/_ms/_tables/read_main_table.py,sha256=8AbNt-AxrhPK3EPRa7xqJXffxzIgfVsv1BDfoVJEXLU,26056
-xradio/vis/_vis_utils/_ms/_tables/read_subtables.py,sha256=JM6pGUQtjQR881u9VqakmbJjppEFq-EVKnEZ14JqnAw,12438
-xradio/vis/_vis_utils/_ms/_tables/table_query.py,sha256=q8EGFf_zIwHcHnvFJOn8hPh8zFZQ3f7BGbXvL3bHad4,555
-xradio/vis/_vis_utils/_ms/_tables/write.py,sha256=43XQ-tHhbhex0eUTRknNpPEEOnNR-w1lGCox9WZ9NHE,9540
-xradio/vis/_vis_utils/_ms/_tables/write_exp_api.py,sha256=GDEll8nMwkQGc6vosu4UddFL5_ld7WurRgF9hYFTRmU,15511
-xradio/vis/_vis_utils/_utils/cds.py,sha256=OpvKowSheIthUbcPEv2AoKmxlEt3DqJZS5C1AYh5z10,1179
-xradio/vis/_vis_utils/_utils/partition_attrs.py,sha256=JaePHts_A0EbB4K-0a_uC98RZ2EmfjB9pDSEI11oAwk,3401
-xradio/vis/_vis_utils/_utils/stokes_types.py,sha256=DMa8TmmS7BQ99Xm8c7ZjcRapMtLbrKVxrt4f0qUIOvg,561
-xradio/vis/_vis_utils/_utils/xds_helper.py,sha256=jZWR55e574j9NJ4Yf5HdHD7J7QtKdXbXStbblQ66WdE,13218
-xradio/vis/_vis_utils/_zarr/encoding.py,sha256=GENIlThV6a9CUCL6gIGlu9c6NR3OFWNos6mpxZjMwDc,536
-xradio/vis/_vis_utils/_zarr/read.py,sha256=O9DiwD2Gn8WiatQ-Q6WGGSwjsXwFktG4f81lM-mgcSg,7596
-xradio/vis/_vis_utils/_zarr/write.py,sha256=k5IfqtI44Dm4KBDiKFGhL5hN7kwNOulvVHmeP5Mi7N4,10043
-xradio-0.0.40.dist-info/LICENSE.txt,sha256=9CYIJt7riOXo9AD0eXBZviLxo_HebD-2JJI8oiWtzfg,1807
-xradio-0.0.40.dist-info/METADATA,sha256=JSY0V_0Zij1K06oFq-4U_z8rRIZCDFsbBadezmVS-jM,4443
-xradio-0.0.40.dist-info/WHEEL,sha256=cVxcB9AmuTcXqmwrtPhNK88dr7IR_b6qagTj0UvIEbY,91
-xradio-0.0.40.dist-info/top_level.txt,sha256=dQu27fGBZJ2Yk-gW5XeD-dZ76Xa4Xcvk60Vz-dwXp7k,7
-xradio-0.0.40.dist-info/RECORD,,