dnsmatrix_mpi.h

// -*-C++-*-
// ==========================================================================
// General dense matrix class
// Distributed (MPI) version
// Each MPI process operates on a block of matrix rows (r0 <= r < r1)
// ==========================================================================

#ifndef __DNSMATRIXMPI_H
#define __DNSMATRIXMPI_H

#include "mathlib.h"

// ==========================================================================
// Nonmember declarations
// ==========================================================================

// ==========================================================================
// class TDenseMatrixMPI
// ==========================================================================
template<class MT> class TDenseMatrixMPI: public TMatrix<MT> {

public:
    TDenseMatrixMPI ();

    TDenseMatrixMPI (int r, int c);

    ~TDenseMatrixMPI ();

    inline MatrixStorage StorageType () const { return MATRIX_DENSE; }
    // Matrix element storage method

    void New (int r, int c);

    void MPIRange (int *_r0, int *_r1) const { *_r0 = r0; *_r1 = r1; }

    MT Get (int r, int c) const;

    MT &operator() (int r, int c);

    TVector<MT> Row (int r) const;

    TVector<MT> Col (int c) const;

    int SparseRow (int r, int *ci, MT *rv) const;

    void SetRow (int r, const TVector<MT> &rval);

    void ColScale (const TVector<MT> &scale);

    void RowScale (const TVector<MT> &scale);

    TVector<MT> RowSum() const;

    TVector<MT> RowSumSq() const;

    TVector<MT> ColSum() const;

    TVector<MT> ColSumSq() const;

    void Ax (const TVector<MT> &x, TVector<MT> &b) const;

    void ATx (const TVector<MT> &x, TVector<MT> &b) const;

    MPI_Datatype MPIType () const;

private:
    void Unlink ();
    void Alloc (int r, int c);

    MT *val;    // data array (only contains relevant sub-block)
    int r0, r1; // row index range
    int nr;     // number of rows for process
    int nr_max; // max number of rows for any process
    int rnk, sze; // MPI rank of process, MPI size
};


// ==========================================================================
// typedefs for specific instances of `TDenseMatrixMPI'

typedef TDenseMatrixMPI<double>   RDenseMatrixMPI;  // 'double real'
typedef TDenseMatrixMPI<float>    FDenseMatrixMPI;    // 'single real'
typedef TDenseMatrixMPI<toast::complex>  CDenseMatrixMPI;   // 'complex'
typedef TDenseMatrixMPI<scomplex> SCDenseMatrixMPI;   // 'single complex'
typedef TDenseMatrixMPI<int>      IDenseMatrixMPI;    // 'integer'

#ifndef MATHLIB_IMPLEMENTATION
extern template class MATHLIB TDenseMatrixMPI<double>;
extern template class MATHLIB TDenseMatrixMPI<float>;
extern template class MATHLIB TDenseMatrixMPI<toast::complex>;
extern template class MATHLIB TDenseMatrixMPI<scomplex>;
extern template class MATHLIB TDenseMatrixMPI<int>;
#endif // MATHLIB_IMPLEMENTATION

#endif // !__DNSMATRIXMPI_H