DataExchangeFromMacro2MD.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_DATAEXCHANGEFROMMACRO2MD_H_
#define _MOLECULARDYNAMICS_COUPLING_SENDRECV_DATAEXCHANGEFROMMACRO2MD_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 DataExchangeFromMacro2MD;
}
} // namespace coupling

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

  virtual ~DataExchangeFromMacro2MD() {}

  std::vector<unsigned int> getTargetRanks(I01 idx) override {
    // if we need information on MD side, return the respective ranks via
    if (I08::contains(idx) && !I12::contains(idx)) {
      return IDXS.getRanksForGlobalIndex(idx, _topologyOffset);
      // otherwise return empty vector
    }
    return std::vector<unsigned int>();
  }

  std::vector<unsigned int> getSourceRanks(I01 idx) override {
    if (I08::contains(idx) && !I12::contains(idx)) { // Global, no ghost, no MD2Macro
      return _msi->getSourceRanks(idx);
    }
    return std::vector<unsigned int>();
  }

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

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

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