mpi_utilities.h

Go to the documentation of this file.

#ifndef PURIFY_MPI_UTILITIES_H
#define PURIFY_MPI_UTILITIES_H
 
#include "purify/config.h"
#include <vector>
#include "purify/uvw_utilities.h"
#include <sopt/linear_transform.h>
 
#ifdef PURIFY_MPI
#include <sopt/mpi/communicator.h>
#endif
#ifdef PURIFY_OPENMP
#include <omp.h>
#endif
namespace purify {
namespace utilities {
#ifdef PURIFY_MPI
void regroup(utilities::vis_params &params, std::vector<t_int> const &groups,
             const t_int max_groups);
void regroup(vis_params &uv_params, std::vector<t_int> &image_index,
             std::vector<t_int> const &groups_, const t_int max_groups);
vis_params regroup_and_scatter(vis_params const &params, std::vector<t_int> const &groups,
                               sopt::mpi::Communicator const &comm);
std::tuple<vis_params, std::vector<t_int>> regroup_and_all_to_all(
    vis_params const &params, const std::vector<t_int> &image_index,
    std::vector<t_int> const &groups, sopt::mpi::Communicator const &comm);
vis_params regroup_and_all_to_all(vis_params const &params, std::vector<t_int> const &groups,
                                  sopt::mpi::Communicator const &comm);
vis_params scatter_visibilities(vis_params const &params, std::vector<t_int> const &sizes,
                                sopt::mpi::Communicator const &comm);
 
vis_params scatter_visibilities(sopt::mpi::Communicator const &comm);
vis_params all_to_all_visibilities(vis_params const &params, std::vector<t_int> const &sizes,
                                   sopt::mpi::Communicator const &comm);
utilities::vis_params distribute_params(utilities::vis_params const &params,
                                        sopt::mpi::Communicator const &comm);
utilities::vis_params set_cell_size(const sopt::mpi::Communicator &comm,
                                    utilities::vis_params const &uv_vis, const t_real &cell_x,
                                    const t_real &cell_y);
utilities::vis_params w_stacking(utilities::vis_params const &params,
                                 sopt::mpi::Communicator const &comm, const t_int iters,
                                 const std::function<t_real(t_real)> &cost,
                                 const t_real k_means_rel_diff = 1e-5);
std::tuple<utilities::vis_params, std::vector<t_int>, std::vector<t_real>>
w_stacking_with_all_to_all(utilities::vis_params const &params, const t_real du,
                           const t_int min_support, const t_int max_support,
                           sopt::mpi::Communicator const &comm, const t_int iters,
                           const t_real fill_relaxation, const std::function<t_real(t_real)> &cost,
                           const t_real k_means_rel_diff = 1e-5);
#endif
template <class T>
t_real step_size(T const &vis, const std::shared_ptr<sopt::LinearTransform<T> const> &measurements,
                 const std::shared_ptr<sopt::LinearTransform<T> const> &wavelets,
                 const t_uint sara_size) {
  // measurement operator may use different number of nodes than wavelet operator
  // so needs to be done separately
  const T dimage = measurements->adjoint() * vis;
  return (sara_size > 0) ? (wavelets->adjoint() * dimage).cwiseAbs().maxCoeff() : 0.;
};
}  // namespace utilities
}  // namespace purify
#endif