src/ParaMEDMEM/MPIProcessorGroup.hxx

   1 //  Copyright (C) 2007-2008  CEA/DEN, EDF R&D
   2 //
   3 //  This library is free software; you can redistribute it and/or
   4 //  modify it under the terms of the GNU Lesser General Public
   5 //  License as published by the Free Software Foundation; either
   6 //  version 2.1 of the License.
   7 //
   8 //  This library is distributed in the hope that it will be useful,
   9 //  but WITHOUT ANY WARRANTY; without even the implied warranty of
  10 //  MERCHANTABILITY or FITNESS FOR A PARTICULAR PURPOSE.  See the GNU
  11 //  Lesser General Public License for more details.
  12 //
  13 //  You should have received a copy of the GNU Lesser General Public
  14 //  License along with this library; if not, write to the Free Software
  15 //  Foundation, Inc., 59 Temple Place, Suite 330, Boston, MA  02111-1307 USA
  16 //
  17 //  See http://www.salome-platform.org/ or email : webmaster.salome@opencascade.com
  18 //
  19 #ifndef __MPIPROCESSORGROUP_HXX__
  20 #define __MPIPROCESSORGROUP_HXX__
  21
  22 #include <set>
  23 #include <mpi.h>
  24
  25 namespace ParaMEDMEM
  26 {
  27   class ProcessorGroup;
  28   class CommInterface;
  29
  30   class MPIProcessorGroup : public ProcessorGroup
  31   {
  32   public:
  33     MPIProcessorGroup(const CommInterface& interface);
  34     MPIProcessorGroup(const CommInterface& interface, std::set<int> proc_ids, const MPI_Comm& world_comm=MPI_COMM_WORLD);
  35     MPIProcessorGroup (const ProcessorGroup& proc_group, std::set<int> proc_ids);
  36     MPIProcessorGroup(const CommInterface& interface,int pstart, int pend);
  37     virtual ~MPIProcessorGroup();
  38     virtual ProcessorGroup* fuse (const ProcessorGroup&) const;
  39     void intersect (ProcessorGroup&) { }
  40     int myRank() const { int rank; MPI_Comm_rank(_comm,&rank); return rank; }
  41     bool containsMyRank() const { int rank; MPI_Group_rank(_group, &rank); return (rank!=MPI_UNDEFINED); }
  42     int translateRank(const ProcessorGroup* group, int rank) const;
  43     const MPI_Comm* getComm() const { return &_comm; }
  44     ProcessorGroup* createComplementProcGroup() const;
  45     ProcessorGroup* createProcGroup() const;
  46     MPI_Comm getWorldComm() { return _world_comm; }
  47   private:
  48     const MPI_Comm _world_comm;
  49     MPI_Group _group;
  50     MPI_Comm _comm;
  51   };
  52 }
  53
  54 #endif