source: XIOS3/trunk/src/mpi_garbage_collector.hpp @ 2569

#ifndef __MPI_GARBAGE_COLLECTOR_HPP__
#define __MPI_GARBAGE_COLLECTOR_HPP__
#include "mpi.hpp"
namespace xios
{
  class CMpiGarbageCollector
  {
    
    private:

      struct SType
      {
        enum {COMM, WIN} type ;
        MPI_Comm comm ;
        MPI_Win win ;
        std::string str;
      } ;
      std::list<SType> stack_ ;
    
    public:

      void registerCommunicator(MPI_Comm& comm, std::string str="") { stack_.push_back(SType{SType::COMM, comm, MPI_WIN_NULL, str}) ;}
      void registerWindow(MPI_Win& win, std::string str="") { stack_.push_back(SType{SType::WIN, MPI_COMM_NULL, win, str}) ;}
      void release(void)
      {
        for( auto& it : stack_) 
          if (it.type==SType::COMM) MPI_Comm_free(&it.comm);
          else if (it.type==SType::WIN) MPI_Win_free(&it.win);
        stack_.clear();
      }
  } ;

}

#endif