source: src/comm/mpi/aux.hpp@ 138f86

/**
 * @file   comm_mpi_aux.hpp
 * @author Julian <julian@ubuntu-laptop>
 * @date   Wed May 11 17:29:46 2011
 *
 * @brief  Auxiliary classes for VMG::CommMPI and VMG::CommMPIInfo.
 *
 */

#ifndef COMM_MPI_AUX_HPP_
#define COMM_MPI_AUX_HPP_

#ifndef HAVE_MPI
#error MPI needed to compile VMG::CommMPI
#endif

#include <mpi.h>

#include "base/index.hpp"

namespace VMG
{

class Grid;

class DatatypeKey
{
public:
  DatatypeKey(const Index& start_, const Index& end_, const Index& gridsize_) :
    start(start_),
    subsize(end_ - start_),
    gridsize(gridsize_)
  {}

  bool operator==(const DatatypeKey& rhs) const
  {
    bool equal = true;

    for (int i=0; i<3; ++i) {
      equal &= this->start[i] == rhs.start[i];
      equal &= this->subsize[i] == rhs.subsize[i];
      equal &= this->gridsize[i] == rhs.gridsize[i];
    }

    return equal;
  }

  bool operator<(const DatatypeKey& rhs) const
  {
    for (int i=0; i<3; ++i) {

      if (this->start[i] < rhs.start[i])
        return true;
      else if (this->start[i] > rhs.start[i])
        return false;

      if (this->subsize[i] < rhs.subsize[i])
        return true;
      else if (this->subsize[i] > rhs.subsize[i])
        return false;

      if (this->gridsize[i] < rhs.gridsize[i])
        return true;
      else if (this->gridsize[i] > rhs.gridsize[i])
        return false;

    }

    return false;
  }

  Index& Start() {return start;}
  Index& SubSize() {return subsize;}
  Index& GridSize() {return gridsize;}

private:
  Index start;
  Index subsize;
  Index gridsize;
};

}

#endif /* COMM_MPI_AUX_HPP_ */