parallel_2d.h

Go to the documentation of this file.

#ifndef _PARALLEL_2D_H_
#define _PARALLEL_2D_H_
 
#include <cstdint>
#include <vector>
 
#include "source_base/module_external/blacs_connector.h"
 
class Parallel_2D
{
  public:
    Parallel_2D() = default;
    ~Parallel_2D() = default;
 
    Parallel_2D& operator=(Parallel_2D&& rhs) = default;
    Parallel_2D(Parallel_2D&& rhs) = default;
 
    int get_row_size() const
    {
        return nrow;
    };
 
    int get_col_size() const
    {
        return ncol;
    };
 
    int get_global_row_size() const;
 
    int get_global_col_size() const;
 
    int64_t get_local_size() const
    {
        return nloc;
    };
 
    int global2local_row(const int igr) const
    {
        return global2local_row_[igr];
    }
 
    int global2local_col(const int igc) const
    {
        return global2local_col_[igc];
    }
 
    int local2global_row(const int ilr) const
    {
        return local2global_row_[ilr];
    }
 
    int local2global_col(const int ilc) const
    {
        return local2global_col_[ilc];
    }
 
    bool in_this_processor(const int iw1_all, const int iw2_all) const;
 
    int get_block_size() const
    {
        return nb;
    };
 
#ifdef __MPI
    int init(const int mg,
             const int ng,
             const int nb, // square block is assumed
             const MPI_Comm comm,
             bool mode = false);
 
    int set(const int mg,
            const int ng,
            const int nb, // square block is assumed
            const int blacs_ctxt);
 
    int blacs_ctxt = -1;
 
    int desc[9] = {};
 
    MPI_Comm comm() const;
#endif
 
    void set_serial(const int mg, const int ng);
 
    // FIXME the following variables should be private, but they are
    // widely used in the code currently. Public visibility is kept
    // for now, but might be changed in the future.
 
    int nrow = 0;
    int ncol = 0;
    int64_t nloc = 0;
    // NOTE: ScaLAPACK descriptors use int type for the number of rows and columns of
    // both the global and local matrices, so nrow & ncol have to be int type. Their
    // product, however, can exceed the range of int type.
 
    int nb = 1;
 
    int dim0 = 0;
    int dim1 = 0;
 
    int coord[2] = {-1, -1};
 
    bool is_serial = false;
 
  protected:
    std::vector<int> global2local_row_;
    std::vector<int> global2local_col_;
 
    std::vector<int> local2global_row_;
    std::vector<int> local2global_col_;
 
#ifdef __MPI
    void _init_proc_grid(const MPI_Comm comm, const bool mode);
    void _set_dist_info(const int mg, const int ng, const int nb);
#endif
};
#endif