TransferStrategy4NieCoupling.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_TRANSFERSTRATEGIES_TRANSFERSTRATEGY4NIECOUPLING_H_
#define _MOLECULARDYNAMICS_COUPLING_TRANSFERSTRATEGIES_TRANSFERSTRATEGY4NIECOUPLING_H_
 
#include "coupling/cell-mappings/ComputeMassMapping.h"
#include "coupling/cell-mappings/ComputeMomentumMapping.h"
#include "coupling/transferstrategies/TransferStrategy.h"
 
namespace coupling {
namespace transferstrategies {
template <class LinkedCell, unsigned int dim> class TransferStrategy4NieCoupling;
}
} // namespace coupling

template <class LinkedCell, unsigned int dim>
class coupling::transferstrategies::TransferStrategy4NieCoupling : public coupling::transferstrategies::TransferStrategy<LinkedCell, dim> {
public:
  TransferStrategy4NieCoupling(coupling::interface::MDSolverInterface<LinkedCell, dim>* const mdSolverInterface, unsigned int numberMDSteps,
                               double shiftTimestep, tarch::la::Vector<2 * dim, bool> massFluxBoundary);

  virtual ~TransferStrategy4NieCoupling();

  void beginProcessInnerCouplingCellsBeforeReceivingMacroscopicSolverData() override;

  void processInnerCouplingCellBeforeReceivingMacroscopicSolverData(coupling::datastructures::CouplingCellWithLinkedCells<LinkedCell, dim>& cell,
                                                                    I02 index) override;

  void processInnerCouplingCellAfterReceivingMacroscopicSolverData(coupling::datastructures::CouplingCellWithLinkedCells<LinkedCell, dim>& cell,
                                                                   I02 index) override;
 
  // virtual void beginProcessInnerCouplingCellsBeforeSendingMDSolverData();

  void processInnerCouplingCellBeforeSendingMDSolverData(coupling::datastructures::CouplingCellWithLinkedCells<LinkedCell, dim>& cell, I02 index) override;
 
  // virtual void endProcessInnerCouplingCellsBeforeSendingMDSolverData();

  void beginProcessInnerCouplingCellsAfterMDTimestep() override;

  void processInnerCouplingCellAfterMDTimestep(coupling::datastructures::CouplingCellWithLinkedCells<LinkedCell, dim>& cell, I02 index) override;
 
private:
  double computeMassFlux(const double& mass, const tarch::la::Vector<dim, double>& velocity, I01 index);
  coupling::cellmappings::ComputeMassMapping<LinkedCell, dim> _massMapping;
  coupling::cellmappings::ComputeMomentumMapping<LinkedCell, dim> _momentumMapping;
  tarch::la::Vector<dim, double>* _oldSolution;
  tarch::la::Vector<dim, double>* _newSolution;
  const unsigned int _numberMDSteps;
  const double _shiftTimestep;
  const unsigned int _numberLocalCells;
  unsigned int _timestepCounter;
  double* _excessMass;
  const tarch::la::Vector<2 * dim, bool> _massFluxBoundary;
};
#include "coupling/transferstrategies/TransferStrategy4NieCoupling.cpph"
 
#endif // _MOLECULARDYNAMICS_COUPLING_TRANSFERSTRATEGIES_TRANSFERSTRATEGY4NIECOUPLING_H_