00001
00002
00003
00004
00005
00006
00007
00008
00009
00010
00011
00012
00013
00014
00015
00016
00017
00018
00019
00020
00021
00022
00023
00024
00025
00026
00027
00028
00029 #ifndef EPETRAEXT_RESTRICTEDMULTIVECTORWRAPPER_H
00030 #define EPETRAEXT_RESTRICTEDMULTIVECTORWRAPPER_H
00031 #include <EpetraExt_ConfigDefs.h>
00032 #include <Teuchos_RCP.hpp>
00033 #include <mpi.h>
00034
00035 #ifdef HAVE_MPI
00036 class Epetra_Comm;
00037 class Epetra_MpiComm;
00038 class Epetra_BlockMap;
00039 class Epetra_MultiVector;
00040
00041 namespace EpetraExt {
00042 class RestrictedMultiVectorWrapper{
00043
00044 public:
00045 RestrictedMultiVectorWrapper();
00046
00047 ~RestrictedMultiVectorWrapper();
00048
00050
00054 int SetMPISubComm(MPI_Comm MPI_SubComm);
00055
00057 MPI_Comm GetMPISubComm(){return MPI_SubComm_;}
00058
00060 const Epetra_MpiComm & RestrictedComm(){return *RestrictedComm_;}
00061
00063 bool RestrictedProcIsActive(){return proc_is_active;}
00064
00066 Teuchos::RCP<Epetra_MultiVector> InputMultiVector(){return input_mv_;}
00067
00069 Teuchos::RCP<Epetra_MultiVector> RestrictedMultiVector(){return restricted_mv_;}
00070
00072
00081 int restrict_comm(Teuchos::RCP<Epetra_MultiVector> input_mv);
00082
00083 private:
00084 bool proc_is_active;
00085 bool subcomm_is_set;
00086
00087 MPI_Comm MPI_SubComm_;
00088 Epetra_MpiComm *RestrictedComm_;
00089 Epetra_BlockMap *ResMap_;
00090
00091 Teuchos::RCP<Epetra_MultiVector> input_mv_;
00092 Teuchos::RCP<Epetra_MultiVector> restricted_mv_;
00093
00094 };
00095
00096
00097 }
00098
00099
00100 #endif
00101 #endif