XYZTopology.h

// Copyright (C) 2015 Technische Universitaet Muenchen
// This file is part of the Mamico project. For conditions of distribution
// and use, please see the copyright notice in Mamico's main folder, or at
// www5.in.tum.de/mamico
#ifndef _MOLECULARDYNAMICS_COUPLING_PARALLELTOPOLOGY_XYZTOPOLOGY_H_
#define _MOLECULARDYNAMICS_COUPLING_PARALLELTOPOLOGY_XYZTOPOLOGY_H_
 
#include "coupling/CouplingMDDefinitions.h"
#include "coupling/paralleltopology/ParallelTopology.h"
 
namespace coupling {
namespace paralleltopology {
template <unsigned int dim> class XYZTopology;
}
} // namespace coupling

template <unsigned int dim> class coupling::paralleltopology::XYZTopology : public coupling::paralleltopology::ParallelTopology<dim> {
public:
  XYZTopology(tarch::la::Vector<dim, unsigned int> numberProcesses)
      : coupling::paralleltopology::ParallelTopology<dim>(), _numberProcesses(numberProcesses),
        _divisionFactor4NumberProcesses(coupling::initDivisionFactor<dim>(numberProcesses)) {}

  virtual ~XYZTopology() {}
 
  tarch::la::Vector<dim, unsigned int> getProcessCoordinates(unsigned int rank, unsigned int topologyOffset) const {
#if (COUPLING_MD_DEBUG == COUPLING_MD_YES)
    std::cout << "Rank=" << rank
              << " corresponds to process coordinates=" << coupling::getVectorCellIndex<dim>(rank - topologyOffset, _divisionFactor4NumberProcesses)
              << std::endl;
#endif
    return coupling::getVectorCellIndex<dim>(rank - topologyOffset, _divisionFactor4NumberProcesses);
  }
 
  unsigned int getRank(tarch::la::Vector<dim, unsigned int> processCoordinates, unsigned int topologyOffset) const {
    unsigned int index = processCoordinates[dim - 1];
    for (int d = dim - 2; d > -1; d--) {
      index = _numberProcesses[d] * index + processCoordinates[d];
    }
    return index + topologyOffset;
  }
 
private:
  /* number of processes */
  const tarch::la::Vector<dim, unsigned int> _numberProcesses;
  /* division factor for number of processes */
  const tarch::la::Vector<dim, unsigned int> _divisionFactor4NumberProcesses;
};
 
#endif // _MOLECULARDYNAMICS_COUPLING_PARALLELTOPOLOGY_XYZTOPOLOGY_H_