xradio 0.0.53__py3-none-any.whl → 0.0.55__py3-none-any.whl

xradio/image/_util/_casacore/xds_from_casacore.py

@@ -54,30 +54,6 @@ def _add_lin_attrs(xds, coord_dict, dir_axes):
     return xds
 
 
-def _add_freq_attrs(xds, coord_dict):
-    freq_coord = xds["frequency"]
-    meta = {}
-    for k in coord_dict:
-        if k.startswith("spectral"):
-            sd = coord_dict[k]
-            meta["rest_frequency"] = make_quantity(sd["restfreq"], "Hz")
-            meta["type"] = "frequency"
-            # meta["units"] = sd["unit"]
-            # meta["frame"] = sd["system"]
-            meta["wave_unit"] = sd["waveUnit"]
-            # meta["crval"] = sd["wcs"]["crval"]
-            # meta["cdelt"] = sd["wcs"]["cdelt"]
-            meta["reference_value"] = make_frequency_reference_dict(
-                value=sd["wcs"]["crval"],
-                units=sd["unit"],
-                observer=sd["system"],
-            )
-    if not meta:
-        # this is the default frequency information CASA creates
-        meta = _default_freq_info()
-    freq_coord.attrs = meta
-
-
 def _add_mask(
     xds: xr.Dataset, name: str, ary: Union[np.ndarray, da.array], dimorder: list
 ) -> xr.Dataset:
@@ -553,7 +529,7 @@ def _get_freq_values_attrs(
                 # attrs["type"] = "frequency"
                 # attrs["units"] = sd["unit"]
                 # attrs["frame"] = sd["system"]
-                attrs["wave_unit"] = sd["waveUnit"]
+                attrs["wave_unit"] = [sd["waveUnit"]]
                 # attrs["crval"] = sd["wcs"]["crval"]
                 # attrs["cdelt"] = sd["wcs"]["cdelt"]
 
@@ -623,36 +599,6 @@ def _get_mask_names(infile: str) -> list:
     return mymasks
 
 
-def _get_beam(imageinfo: dict, nchan: int, npol: int) -> Union[np.ndarray, None]:
-    """Returns None if the image has no beam(s)"""
-    x = ["perplanebeams", "restoringbeam"]
-    r = None
-    for z in x:
-        if z in imageinfo:
-            r = z
-            break
-    if r is None:
-        return None
-    beam = imageinfo[r]
-    beam_array = np.zeros([1, nchan, npol, 3])
-    if r == "perplanebeams":
-        for c in range(nchan):
-            for p in range(npol):
-                k = nchan * p + c
-                b = _casacore_q_to_xradio_q(beam["*" + str(k)])
-                beam_dict = _convert_beam_to_rad(b)
-                beam_array[0][c][p][0] = beam_dict["major"]["data"]
-                beam_array[0][c][p][1] = beam_dict["minor"]["data"]
-                beam_array[0][c][p][2] = beam_dict["pa"]["data"]
-    elif r == "restoringbeam":
-        b = _casacore_q_to_xradio_q(beam)
-        beam_dict = _convert_beam_to_rad(b)
-        beam_array[0, :, :, 0] = beam_dict["major"]["data"]
-        beam_array[0, :, :, 1] = beam_dict["minor"]["data"]
-        beam_array[0, :, :, 2] = beam_dict["pa"]["data"]
-    return beam_array
-
-
 def _get_persistent_block(
     infile: str,
     shapes: tuple,
@@ -858,25 +804,46 @@ def _get_velocity_values_attrs(
     )
 
 
-def _multibeam_array(
-    xds: xr.Dataset, img_full_path: str, as_dask_array: bool
+def _get_beam(
+    img_full_path: str, nchan: int, npol: int, as_dask_array: bool
 ) -> Union[xr.DataArray, None]:
     # the image may have multiple beams
     with _open_image_ro(img_full_path) as casa_image:
         imageinfo = casa_image.info()["imageinfo"]
-    mb = _get_beam(
-        imageinfo, nchan=xds.sizes["frequency"], npol=xds.sizes["polarization"]
-    )
-    if mb is not None:
-        if as_dask_array:
-            mb = da.array(mb)
-        xdb = xr.DataArray(mb, dims=["time", "frequency", "polarization", "beam_param"])
-        xdb = xdb.rename("BEAM")
-        xdb = xdb.assign_coords(beam_param=["major", "minor", "pa"])
-        xdb.attrs["units"] = "rad"
-        return xdb
-    else:
+    x = ["perplanebeams", "restoringbeam"]
+    r = None
+    for z in x:
+        if z in imageinfo:
+            r = z
+            break
+    if r is None:
         return None
+    beam = imageinfo[r]
+    beam_array = np.zeros([1, nchan, npol, 3])
+    if r == "perplanebeams":
+        for c in range(nchan):
+            for p in range(npol):
+                k = nchan * p + c
+                b = _casacore_q_to_xradio_q(beam["*" + str(k)])
+                beam_dict = _convert_beam_to_rad(b)
+                beam_array[0][c][p][0] = beam_dict["major"]["data"]
+                beam_array[0][c][p][1] = beam_dict["minor"]["data"]
+                beam_array[0][c][p][2] = beam_dict["pa"]["data"]
+    elif r == "restoringbeam":
+        b = _casacore_q_to_xradio_q(beam)
+        beam_dict = _convert_beam_to_rad(b)
+        beam_array[0, :, :, 0] = beam_dict["major"]["data"]
+        beam_array[0, :, :, 1] = beam_dict["minor"]["data"]
+        beam_array[0, :, :, 2] = beam_dict["pa"]["data"]
+    if as_dask_array:
+        beam_array = da.array(beam_array)
+    xdb = xr.DataArray(
+        beam_array, dims=["time", "frequency", "polarization", "beam_param"]
+    )
+    xdb = xdb.rename("BEAM")
+    xdb = xdb.assign_coords(beam_param=["major", "minor", "pa"])
+    xdb.attrs["units"] = "rad"
+    return xdb
 
 
 ###########################################################################

xradio/image/_util/_casacore/xds_to_casacore.py

@@ -88,9 +88,11 @@ def _compute_spectral_dict(xds: xr.Dataset) -> dict:
     spec["unit"] = u if isinstance(u, str) else u[0]
     spec["velType"] = _doppler_types.index(xds.velocity.attrs["doppler_type"])
     u = xds.velocity.attrs["units"]
-    spec["velUnit"] = u if isinstance(u, str) else u[0]
     spec["version"] = 2
-    spec["waveUnit"] = xds.frequency.attrs["wave_unit"]
+    # vel unit is a list[str] in the xds but needs to be a str in the casa image
+    spec["velUnit"] = xds.velocity.attrs["units"][0]
+    # wave unit is a list[str] in the xds but needs to be a str in the casa image
+    spec["waveUnit"] = xds.frequency.attrs["wave_unit"][0]
     wcs = {}
     wcs["ctype"] = "FREQ"
     wcs["pc"] = 1.0

xradio/image/_util/_fits/xds_from_fits.py

@@ -83,7 +83,7 @@ def _add_freq_attrs(xds: xr.Dataset, helpers: dict) -> xr.Dataset:
         meta["rest_frequency"] = make_quantity(helpers["restfreq"], "Hz")
         meta["rest_frequencies"] = [meta["rest_frequency"]]
         meta["type"] = "frequency"
-        meta["wave_unit"] = "mm"
+        meta["wave_unit"] = ["mm"]
         freq_axis = helpers["freq_axis"]
         meta["reference_value"] = make_frequency_reference_dict(
             helpers["crval"][freq_axis], ["Hz"], helpers["specsys"]

xradio/image/_util/casacore.py

@@ -21,7 +21,7 @@ from ._casacore.xds_from_casacore import (
     _get_persistent_block,
     _get_starts_shapes_slices,
     _get_transpose_list,
-    _multibeam_array,
+    _get_beam,
     _read_image_array,
 )
 from ._casacore.xds_to_casacore import (
@@ -42,6 +42,8 @@ def _load_casa_image_block(infile: str, block_des: dict, do_sky_coords) -> xr.Da
         cshape = casa_image.shape()
     ret = _casa_image_to_xds_coords(image_full_path, False, do_sky_coords)
     xds = ret["xds"].isel(block_des)
+    nchan = ret["xds"].dims["frequency"]
+    npol = ret["xds"].dims["polarization"]
     starts, shapes, slices = _get_starts_shapes_slices(block_des, coords, cshape)
     dimorder = _get_xds_dim_order(ret["sphr_dims"])
     transpose_list, new_axes = _get_transpose_list(coords)
@@ -60,13 +62,14 @@ def _load_casa_image_block(infile: str, block_des: dict, do_sky_coords) -> xr.Da
         # data vars are all caps by convention
         xds = _add_mask(xds, m.upper(), block, dimorder)
     xds.attrs = _casa_image_to_xds_attrs(image_full_path)
-    mb = _multibeam_array(xds, image_full_path, False)
-    if mb is not None:
-        selectors = {}
-        for k in ("time", "frequency", "polarization"):
-            if k in block_des:
-                selectors[k] = block_des[k]
-        xds["BEAM"] = mb.isel(selectors)
+    beam = _get_beam(image_full_path, nchan, npol, False)
+    if beam is not None:
+        selectors = {
+            k: block_des[k]
+            for k in ("time", "frequency", "polarization")
+            if k in block_des
+        }
+        xds["BEAM"] = beam.isel(selectors)
     return xds
 
 
@@ -97,9 +100,11 @@ def _read_casa_image(
             # data var names are all caps by convention
             xds = _add_mask(xds, m.upper(), ary, dimorder)
     xds.attrs = _casa_image_to_xds_attrs(img_full_path)
-    mb = _multibeam_array(xds, img_full_path, True)
-    if mb is not None:
-        xds["BEAM"] = mb
+    beam = _get_beam(
+        img_full_path, xds.dims["frequency"], xds.dims["polarization"], True
+    )
+    if beam is not None:
+        xds["BEAM"] = beam
     # xds = _add_coord_attrs(xds, ret["icoords"], ret["dir_axes"])
     xds = _dask_arrayize_dv(xds)
     return xds

xradio/image/image.py

@@ -159,7 +159,7 @@ def load_image(infile: str, block_des: dict = {}, do_sky_coords=True) -> xr.Data
         do_casa = False
     if do_casa:
         # comment next line when done debugging
-        # return _load_casa_image_block(infile, bd, do_sky_coords)
+        # return _load_casa_image_block(infile, selection, do_sky_coords)
         try:
             return _load_casa_image_block(infile, selection, do_sky_coords)
         except Exception as e:

{xradio-0.0.53.dist-info → xradio-0.0.55.dist-info}/METADATA

@@ -1,6 +1,6 @@
 Metadata-Version: 2.4
 Name: xradio
-Version: 0.0.53
+Version: 0.0.55
 Summary:  Xarray Radio Astronomy Data IO
 Author-email: Jan-Willem Steeb <jsteeb@nrao.edu>, Federico Montesino Pouzols <pouzols@eso.edu>, Dave Mehringer <dmehring@nrao.edu>, Peter Wortmann <peter.wortmann@skao.int>
 License: BSD 3-Clause License

{xradio-0.0.53.dist-info → xradio-0.0.55.dist-info}/RECORD

@@ -8,18 +8,18 @@ xradio/_utils/_casacore/tables.py,sha256=aq6E_4RRAHdTBCwMKrVil1cWhFU2O980DNH9IlR
 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=Zk1OE-ixqE2epKmKtlyERG8byB41K5sOZLi0JD0bJis,14572
+xradio/image/image.py,sha256=bPgjJoQaur10XEnK7nw66YG68tRKFmdXFlER7GjWh64,14579
 xradio/image/_util/__init__.py,sha256=47DEQpj8HBSa-_TImW-5JCeuQeRkm5NMpJWZG3hSuFU,0
-xradio/image/_util/casacore.py,sha256=eaiMcr546p7k9SU2s3pQDgpFyS94xW3Z1ZR3yXxVEfc,4375
+xradio/image/_util/casacore.py,sha256=vhmPz5tCUXTN46oQLuTxkdHTxI_9qR_RfHUEeqPn9SU,4511
 xradio/image/_util/common.py,sha256=y2QJXTHowvjqwNPG5a-cOzl0WneH7c8c9jAVSKQeK2w,8429
 xradio/image/_util/fits.py,sha256=gyGm06fuCKqVGK7uv-ObvQNfFawUDsIOa_nQyklM3Aw,329
 xradio/image/_util/image_factory.py,sha256=Mm0ZDraD0WoNpGqy98EneFr1PxgfyNZNQwquIH2t0nc,8610
 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=Z7Jl3AU7jVcgjNtCnvL7CCXJQAxXeEtowXBmSShuAv4,1329
-xradio/image/_util/_casacore/xds_from_casacore.py,sha256=KOtXG7fwHyTTRyHprlbQOVcu1ZnrSv47zcP8Mu4SGSc,44036
-xradio/image/_util/_casacore/xds_to_casacore.py,sha256=0tSmRHZxUcoSBWD3IMH5ymBEK2RDVPQ1-My42LF9bxw,16922
-xradio/image/_util/_fits/xds_from_fits.py,sha256=kx5CNkzPn7t2wDkEK60O2VwrrUre3P9QchvsapudDHo,29189
+xradio/image/_util/_casacore/xds_from_casacore.py,sha256=uc2jQRAjKIOMh6OO8qIJs7vrNC_J6jI3bObRdL_4fP4,42865
+xradio/image/_util/_casacore/xds_to_casacore.py,sha256=snHCcbZQZtv329KNMd-RjYXG3pfNQeJQbOWKduY1uQc,17085
+xradio/image/_util/_fits/xds_from_fits.py,sha256=n2tAIOK5GoIKzYBqfwCjZULFYSz2x74aeVkCo0vmOL4,29191
 xradio/image/_util/_zarr/common.py,sha256=ltlj3uFa-uv8lXlDtV79QnfNmfm0tyhXN5FDAjZtjzg,308
 xradio/image/_util/_zarr/xds_from_zarr.py,sha256=KMsfaSSm9kyVoztS6pUzGNxMZzQnCxkk0kDv2GxW5Kw,4451
 xradio/image/_util/_zarr/xds_to_zarr.py,sha256=nsDvDD-kuMuMF2dDlj0jTxSW4mdR-jjIsvXHi5uIERU,2373
@@ -70,8 +70,8 @@ xradio/schema/metamodel.py,sha256=WjtW7pAVzcjLRWifRH3sQoOiN6TV810hARpOIz1M_gw,38
 xradio/schema/typing.py,sha256=8-o6fZd99kJ4FVdgBYRTIRJ-wDqpcUNXzCTfJvl3TIw,10439
 xradio/sphinx/__init__.py,sha256=VGY-7Ty3q67qpnBee0-znbiJ-Iy0F93UO--IpjEdHXc,380
 xradio/sphinx/schema_table.py,sha256=uq33habbAbReqnEG6ASKSd4UOMZGUzA3qoTX45rq84U,12373
-xradio-0.0.53.dist-info/licenses/LICENSE.txt,sha256=9CYIJt7riOXo9AD0eXBZviLxo_HebD-2JJI8oiWtzfg,1807
-xradio-0.0.53.dist-info/METADATA,sha256=kKmHQUumboRASOFp-1VamR5TW9NRXzOsLlAPJX4fNxQ,5574
-xradio-0.0.53.dist-info/WHEEL,sha256=wXxTzcEDnjrTwFYjLPcsW_7_XihufBwmpiBeiXNBGEA,91
-xradio-0.0.53.dist-info/top_level.txt,sha256=dQu27fGBZJ2Yk-gW5XeD-dZ76Xa4Xcvk60Vz-dwXp7k,7
-xradio-0.0.53.dist-info/RECORD,,
+xradio-0.0.55.dist-info/licenses/LICENSE.txt,sha256=9CYIJt7riOXo9AD0eXBZviLxo_HebD-2JJI8oiWtzfg,1807
+xradio-0.0.55.dist-info/METADATA,sha256=iF5gDBadJWPQKJFcBEJpMUev97YRCCi1nArYtLDgWHw,5574
+xradio-0.0.55.dist-info/WHEEL,sha256=0CuiUZ_p9E4cD6NyLD6UG80LBXYyiSYZOKDm5lp32xk,91
+xradio-0.0.55.dist-info/top_level.txt,sha256=dQu27fGBZJ2Yk-gW5XeD-dZ76Xa4Xcvk60Vz-dwXp7k,7
+xradio-0.0.55.dist-info/RECORD,,

{xradio-0.0.53.dist-info → xradio-0.0.55.dist-info}/WHEEL

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