source: src/comm/mpi/datatypes_local.hpp@ 716da7

#ifndef DATATYPES_LOCAL_HPP_
#define DATATYPES_LOCAL_HPP_

#include <vector>

#include "comm/mpi/datatype.hpp"

namespace VMG
{

class Grid;

namespace Particle
{
class LinkedCellList;
}

namespace MPI
{

class DatatypesLocal
{
public:
  DatatypesLocal(const Grid& grid, const MPI_Comm& comm)
  {
    InitDatatypesLocal(grid, comm, true);
  }

  DatatypesLocal(const VMG::Particle::LinkedCellList& grid, const MPI_Comm& comm)
  {
    InitDatatypesLocal(grid, comm, false);
  }

  ~DatatypesLocal() {}

  std::vector<Datatype>& Halo() {return _halo;}
  std::vector<Datatype>& NB() {return _nb;}

  const std::vector<Datatype>& Halo() const {return _halo;}
  const std::vector<Datatype>& NB() const {return _nb;}

  const std::vector<Index>& Offset() const {return _offset;}

private:
  template <class T>
  void InitDatatypesLocal(const T& grid, const MPI_Comm& comm, const bool& alloc_buffer);

  std::vector<Datatype> _halo, _nb;
  std::vector<Index> _offset;
};

}

}

#endif /* DATATYPES_LOCAL_HPP_ */