MomentumController.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_MOMENTUMCONTROLLER_H_
#define _MOLECULARDYNAMICS_COUPLING_MOMENTUMCONTROLLER_H_
 
#include "coupling/cell-mappings/ComputeMassMapping.h"
#include "coupling/cell-mappings/ComputeMomentumMapping.h"
#include "coupling/cell-mappings/SetMomentumMapping.h"
#include "coupling/datastructures/CouplingCell.h"
#include "tarch/la/Vector.h"
 
namespace coupling {
template <class LinkedCell, unsigned int dim> class MomentumController;
}

template <class LinkedCell, unsigned int dim> class coupling::MomentumController {
public:
  MomentumController(coupling::interface::MDSolverInterface<LinkedCell, dim>* const mdSolverInterface)
      : _computeMassMapping(mdSolverInterface), _computeMomentumMapping(mdSolverInterface), _mdSolverInterface(mdSolverInterface) {}

  ~MomentumController() {}

  void computeMomentumAndMeanVelocity(coupling::datastructures::CouplingCellWithLinkedCells<LinkedCell, dim>& cell, tarch::la::Vector<dim, double>& momentum,
                                      tarch::la::Vector<dim, double>& meanVelocity) {
    cell.iterateConstCells(_computeMomentumMapping);
    momentum = _computeMomentumMapping.getMomentum();
    meanVelocity = _computeMomentumMapping.getMeanVelocity();
  }

  void computeMomentum(coupling::datastructures::CouplingCellWithLinkedCells<LinkedCell, dim>& cell, tarch::la::Vector<dim, double>& momentum) {
    cell.iterateConstCells(_computeMomentumMapping);
    momentum = _computeMomentumMapping.getMomentum();
  }

  void computeMeanVelocity(coupling::datastructures::CouplingCellWithLinkedCells<LinkedCell, dim>& cell, tarch::la::Vector<dim, double>& meanVelocity) {
    cell.iterateConstCells(_computeMomentumMapping);
    meanVelocity = _computeMomentumMapping.getMeanVelocity();
  }

  void setMomentum(coupling::datastructures::CouplingCellWithLinkedCells<LinkedCell, dim>& cell, const tarch::la::Vector<dim, double>& newMomentum) {
    tarch::la::Vector<dim, double> currentMomentum(0.0);
    computeMomentum(cell, currentMomentum);
    cell.iterateConstCells(_computeMassMapping);
    unsigned int numberParticles = _computeMassMapping.getNumberOfParticles();
    coupling::cellmappings::SetMomentumMapping<LinkedCell, dim> setMomentumMapping(currentMomentum, newMomentum, numberParticles, _mdSolverInterface);
    cell.iterateCells(setMomentumMapping);
  }
 
private:
  coupling::cellmappings::ComputeMassMapping<LinkedCell, dim> _computeMassMapping;
  coupling::cellmappings::ComputeMomentumMapping<LinkedCell, dim> _computeMomentumMapping;
  coupling::interface::MDSolverInterface<LinkedCell, dim>* const _mdSolverInterface;
};
#endif // _MOLECULARDYNAMICS_COUPLING_MOMENTUMCONTROLLER_H_