def _nifty_grid_streams(baselines, grid_config, indices, vis, weights, grid_in=None): """ Wrapper function for creating a grid of visibilities per row chunk """ return ng.ms2grid_c(baselines, grid_config, indices, vis, grid_in, weights)
def _nifty_grid(baselines, grid_config, indices, vis, weights): """ Wrapper function for creating a grid of visibilities per row chunk """ assert len(vis) == 1 and type(vis) == list return ng.ms2grid_c(baselines, grid_config, indices, vis[0], None, weights[0])[None, :, :]