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

xradio/image/_util/_casacore/xds_from_casacore.py

@@ -599,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,
@@ -834,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.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.54.dist-info → xradio-0.0.55.dist-info}/METADATA

@@ -1,6 +1,6 @@
 Metadata-Version: 2.4
 Name: xradio
-Version: 0.0.54
+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.54.dist-info → xradio-0.0.55.dist-info}/RECORD

@@ -8,16 +8,16 @@ 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=ZLgOsQ9n-2sjho5xWeYk1I59QLBH0pqNrxAhF07HxSY,43172
+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
@@ -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.54.dist-info/licenses/LICENSE.txt,sha256=9CYIJt7riOXo9AD0eXBZviLxo_HebD-2JJI8oiWtzfg,1807
-xradio-0.0.54.dist-info/METADATA,sha256=wGZFgvOO0xPUPXAUlVlrxyrYVXsUmD6fP7hOFn9mNOk,5574
-xradio-0.0.54.dist-info/WHEEL,sha256=0CuiUZ_p9E4cD6NyLD6UG80LBXYyiSYZOKDm5lp32xk,91
-xradio-0.0.54.dist-info/top_level.txt,sha256=dQu27fGBZJ2Yk-gW5XeD-dZ76Xa4Xcvk60Vz-dwXp7k,7
-xradio-0.0.54.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,,