DataExchangeFromMD2Macro.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_SENDRECV_DATAEXCHANGEFROMMD2MACRO_H_
#define _MOLECULARDYNAMICS_COUPLING_SENDRECV_DATAEXCHANGEFROMMD2MACRO_H_
 
#include "coupling/CouplingMDDefinitions.h"
#include "coupling/datastructures/CouplingCell.h"
#include "coupling/indexing/IndexingService.h"
#include "coupling/interface/MacroscopicSolverInterface.h"
#include "coupling/sendrecv/DataExchange.h"
 
namespace coupling {
namespace sendrecv {
template <unsigned int dim> class DataExchangeFromMD2Macro;
}
} // namespace coupling

template <unsigned int dim>
class coupling::sendrecv::DataExchangeFromMD2Macro : public coupling::sendrecv::DataExchange<coupling::datastructures::CouplingCell<dim>, dim> {
 
public:
  DataExchangeFromMD2Macro(coupling::interface::MacroscopicSolverInterface<dim>* interface, unsigned int topologyOffset, unsigned int tagoffset = 0)
      : coupling::sendrecv::DataExchange<coupling::datastructures::CouplingCell<dim>, dim>(TAG_FROM_MD2MACRO + tagoffset), _msi(interface),
        _topologyOffset(topologyOffset) {
#if (COUPLING_MD_DEBUG == COUPLING_MD_YES)
    std::cout << "DataExchangeFromMD2Macro initialised..." << std::endl;
#endif
  }

  virtual ~DataExchangeFromMD2Macro() {}

  std::vector<unsigned int> getTargetRanks(I01 idx) override {
    // if we need information on macroscopic solver side, return the respective
    // ranks via interface
    if (I12::contains(idx)) {
      return _msi->getTargetRanks(idx);
      // otherwise return empty vector
    } else {
      return std::vector<unsigned int>();
    }
  }

  std::vector<unsigned int> getSourceRanks(I01 idx) override {
    std::vector<unsigned int> sourceRanks;
    if (I12::contains(idx)) {
      sourceRanks.push_back(IDXS.getUniqueRankForCouplingCell(idx, _topologyOffset));
    }
    return sourceRanks;
  }

  void readFromCell(double* const buffer, const coupling::datastructures::CouplingCell<dim>& cell) override {
    buffer[0] = cell.getMacroscopicMass();
    for (unsigned int d = 0; d < dim; d++) {
      buffer[d + 1] = cell.getMacroscopicMomentum()[d];
    }
  }

  void writeToCell(const double* const buffer, coupling::datastructures::CouplingCell<dim>& cell) override {
    tarch::la::Vector<dim, double> macroscopicMomentum(0.0);
    for (unsigned int d = 0; d < dim; d++) {
      macroscopicMomentum[d] = buffer[1 + d];
    }
    cell.setMacroscopicMomentum(macroscopicMomentum);
    cell.setMacroscopicMass(buffer[0]);
  }

  unsigned int getDoublesPerCell() const override {
    // 1 double: macroscopic mass; dim doubles: macroscopic momentum
    return 1 + dim;
  }
 
private:
  coupling::interface::MacroscopicSolverInterface<dim>* _msi;
  unsigned int _topologyOffset;
};
#endif // _MOLECULARDYNAMICS_COUPLING_SENDRECV_DATAEXCHANGEFROMMD2MACRO_H_