grid.hpp

Go to the documentation of this file.

#ifndef EXPLICIT_MAPMAKING_GRID_HPP_
#define EXPLICIT_MAPMAKING_GRID_HPP_

#include <mpi.h>
#include <vector>
#include <utility>

namespace mapmaking {
class Grid {
 public:
  Grid(const MPI_Comm comm);
  Grid(int n_row, int n_col, std::vector<int> id_proc_involved, const MPI_Comm comm);
  ~Grid();

  int context() const { return context_; }  // BLACS context identifying the created process grid
  int my_row() const { return my_row_; }  // Negative means not in the grid
  int my_col() const { return my_col_; }
  std::pair<int,int> my_coor() const { return std::pair<int,int>(my_row_, my_col_); }
  int my_rank() const { return my_rank_; }
  int n_row() const { return n_row_; }  
  int n_col() const { return n_col_; }
  int n_proc_grid() const { return n_proc_grid_; }
  int n_proc_comm() const { return n_proc_comm_; }
  const MPI_Comm& comm() const { return comm_; }
  bool i_am_in() const { return i_am_in_; }
  const std::vector<int>& row_indexes() const { return row_indexes_; }  // {0, 1, ..., n_col} TODO remove
  const std::vector<int>& col_indexes() const { return col_indexes_; }
  const std::vector<std::pair<int,int>>& coordinates() const { return coordinates_; }  // {(0,0), (0,1), ..., (n_row_-1, n_col_-1)} TODO remove
  std::pair<int,int> rank2coor(int rank) const;
  int coor2rank(std::pair<int,int> coor) const;

 private:
  int bhandle_ = 0;  
  int context_ = 0;  
  int my_row_ = -1;
  int my_col_ = -1;
  int my_rank_ = -1;
  int n_row_ = -1;
  int n_col_ = -1;
  int n_proc_grid_ = 0;
  int n_proc_comm_ = 0;
  const MPI_Comm comm_;
  bool i_am_in_ = false; // The processor is in the grid
  std::vector<int> row_indexes_;
  std::vector<int> col_indexes_;
  std::vector<int> rank_procs;
  std::vector<std::pair<int,int>> coordinates_;
};
}

#endif  // EXPLICIT_MAPMAKING_GRID_HPP_