CommunicationList.h

Go to the documentation of this file.

#ifndef included_AMP_CommunicationList_h
#define included_AMP_CommunicationList_h
 
#include "AMP/utils/AMP_MPI.h"
#include "AMP/utils/ParameterBase.h"
#include <memory>
 
#include <vector>
 
 
namespace AMP::LinearAlgebra {
 
class VectorData;
class VectorIndexer;
 
 
class CommunicationListParameters : public ParameterBase
{
public:
    AMP_MPI d_comm;
 
    size_t d_localsize;
 
    std::vector<size_t> d_remote_DOFs;
 
    CommunicationListParameters();
 
    CommunicationListParameters( const CommunicationListParameters & );
 
    CommunicationListParameters &operator=( const CommunicationListParameters & ) = delete;
};
 
 
class CommunicationList final
{
public:
    CommunicationList( std::shared_ptr<const CommunicationListParameters> params );
 
    CommunicationList( size_t local, const AMP_MPI &comm );
 
    CommunicationList( const AMP_MPI &comm, std::vector<size_t> local, std::vector<size_t> remote );
 
    CommunicationList( const std::vector<size_t> &d_partition, const AMP_MPI &comm );
 
    std::shared_ptr<CommunicationList> getNoCommunicationList();
 
    std::shared_ptr<CommunicationList> subset( std::shared_ptr<VectorIndexer> sub );
 
    size_t getVectorReceiveBufferSize() const;
 
    size_t getVectorSendBufferSize() const;
 
    const std::vector<size_t> &getGhostIDList() const;
 
    const std::vector<size_t> &getReplicatedIDList() const;
 
    const std::vector<int> &getReceiveSizes() const;
 
    const std::vector<int> &getSendSizes() const;
 
    const std::vector<int> &getReceiveDisp() const;
 
    const std::vector<int> &getSendDisp() const;
 
    const std::vector<size_t> &getPartition() const;
 
    size_t getStartGID() const;
 
    size_t getTotalSize() const;
 
    size_t numLocalRows() const;
 
    void clearBuffers();
 
    bool anyCommunication();
 
    size_t getLocalGhostID( size_t dof ) const;
 
    const AMP_MPI &getComm() const;
 
    uint64_t getID() const;
 
 
public:
    // Initialize the internal data
    void initialize() const;
 
protected:
    // Empty constructor
    CommunicationList();
 
private:
    AMP_MPI d_comm;                            // Communicator
    mutable bool d_initialized   = false;      // Have we initialized all of the data
    mutable bool d_anyRankRemote = false;      // Does any rank have remote data
    std::vector<size_t> d_ReceiveDOFList;      // Sorted DOF receive lists
    std::vector<size_t> d_partition;           // Partition info
    mutable std::vector<size_t> d_SendDOFList; // Sorted DOF send lists
    mutable std::vector<int> d_ReceiveSizes;   // Number of DOFs to receive from each rank
    mutable std::vector<int> d_ReceiveDisp;    // Displacement for each rank
    mutable std::vector<int> d_SendSizes;      // Number of DOFs to send from each rank
    mutable std::vector<int> d_SendDisp;       // Displacement for each rank
};
 
} // namespace AMP::LinearAlgebra
 
#endif