binning

Module containing functions for binning visibility data.

ska_sdp_distributed_self_cal_prototype.processing_tasks.binning.bin_visibilities(dataset: xarray.Dataset, subgrid_config_list: list[ska_sdp_exec_swiftly.api.SubgridConfig], binning_info: dict) → list[source]

Bin visibility data.

Takes a subgrid configuration and a dataset and extracts visibilities, uvw and start and end channels.

Parameters:

  • dataset – xarray dataset containing visibilities and uvw

  • subgrid_config_list – list of configurations for subgrids

  • binning_info – dictionary containing global parameters for subgridding

Returns:

list of dask tasks to generate VisibilityBin instances populated

with binned visibility data

Return type:

visibility_bins_delayed

ska_sdp_distributed_self_cal_prototype.processing_tasks.binning.get_uvw(dataset: xarray.Dataset, flatten: bool = False) → numpy.ndarray[source]

Get UVW data from visibility partition.

Parameters:

  • dataset – Xarray dataset with uvw data.

  • flatten – If True flatten times and baselines into one dimension, default = False.

Returns:

Numpy array containing uvw values.

Return type:

uvw_values

ska_sdp_distributed_self_cal_prototype.processing_tasks.binning.get_visibilities(dataset: xarray.Dataset, vis_name: str, stokes_i: bool = False, flatten: bool = False) → dask.array.Array[source]

Get visibility data from visibility partition.

Parameters:

  • dataset – Xarray dataset with visibility data.

  • vis_name – Name of column containing visibility data.

  • stokes_i – If True return stokes i instead of raw visibilities.

  • flatten – If True flatten times and baselines.

Returns:

Dask array containing visibility values.

Return type:

visibilities

ska_sdp_distributed_self_cal_prototype.processing_tasks.binning.get_visibility_bin_data(subgrid_config: list[ska_sdp_exec_swiftly.api.SubgridConfig], visibilities: dask.array.Array, uvw: dask.array.Array, binning_info: dict) → VisibilityBin[source]

Extract visibility data for a subgrid configuration.

Takes a subgrid configuration and a dataset and extracts visibilities, uvw and start and end channels. Returns a VisibilityBin instance containing binned data.

Parameters:

  • subgrid_config – Configuration for subgrid.

  • visibilities – Unrolled visibility data to mask.

  • uvw – Unrolled uvw data to mask.

  • binning_info – Dictionary containing global parameters for subgridding.

Returns:

VisibilityBin instance populated with binned visibility data.

Return type:

visibility_bin