ParFECommunicator3D.h

// =======================================================================
// @(#)ParFECommunicator3D.h
//
// Class:      TParFECommunicator3D
// Purpose:    Class containing communication routines
//
// Author:     Sashikumaar Ganesan (24.04.15)
//
// History:    Start of implementation 24.04.15 (Sashikumaar Ganesan)
//
// ======================================================================= 

#ifdef _MPI

#ifndef __PARFECOMMUNICATOR3D__
#define __PARFECOMMUNICATOR3D__

#include "mpi.h"

#include <ParFEMapper3D.h>
#include <FESpace3D.h>
#include <SquareStructure.h>
#include <SquareStructure3D.h>

class TParFECommunicator3D
{
  protected:
    TParFEMapper3D *Mapper;
    
    int N_Dim;
    
    int N_Dof;
    
    int N_SendDof, N_SendDofMS, N_SendDofH1, N_SendDofH2;
    
    double *Send_Info, *Send_InfoMS, *Send_InfoH1, *Send_InfoH2;
    
    double *Recv_Info, *Recv_InfoMS, *Recv_InfoH1, *Recv_InfoH2;
    
    int *N_DofSend, *N_DofSendMS, *N_DofSendH1, *N_DofSendH2; 
    
    int *N_DofRecv, *N_DofRecvMS, *N_DofRecvH1, *N_DofRecvH2;
    
    int *sdispl, *sdisplMS, *sdisplH1, *sdisplH2;
    
    int *rdispl, *rdisplMS, *rdisplH1, *rdisplH2; 
    
    int *DofSend, *DofSendMS, *DofSendH1, *DofSendH2;
    
    int *DofRecv, *DofRecvMS, *DofRecvH1, *DofRecvH2;
    
    int N_Slave, N_InterfaceS, N_Halo1, N_Halo2;
    
    MPI_Comm Comm;
    
    int *Master;
    
  public:
    TParFECommunicator3D(TParFEMapper3D *mapper);
    
    TParFECommunicator3D();
    
    ~TParFECommunicator3D();
    
    void CommUpdateMS(double *sol);
    
    void CommUpdateH1(double *sol);
    
    void CommUpdateH2(double *sol);
    
    void CommUpdate_M_H1(double *sol);
    
    void CommUpdate(double *sol);
    
    void CommUpdateReduce(double *rhs);
    
    void CommUpdate(double *sol, double *rhs);
    
    void GatherToRoot(double *&GlobalArray, int &GolabalSize, double *LocalArray, int LocalSize, int root);
    
    void ScatterFromRoot(double *GlobalArray, int &GlobalSize, double *LocalArray, int LocalSize, int root);
    
    int *GetMaster()
    {return Master;}
    
    int GetNDof()
    {return N_Dof;}
    
    int* Get_Local2Global()
    { return Mapper->Get_Local2Global();}
    
    int GetN_Master()
    { return Mapper->GetN_Master();}
    
};

#endif
#endif