LinkedCellContainer.h

#ifndef _MOLECULARDYNAMICS_COUPLING_DATASTRUCTURES_COUPLINGCELLSWITHLINKEDCELLS_H_
#define _MOLECULARDYNAMICS_COUPLING_DATASTRUCTURES_COUPLINGCELLSWITHLINKEDCELLS_H_
 
#include "CellContainer.h"
#include "coupling/datastructures/CouplingCellWithLinkedCells.h"
#include "coupling/interface/MDSolverInterface.h"
 
namespace coupling {
namespace datastructures {
template <class LinkedCell, unsigned int dim> class LinkedCellContainer;
} // namespace datastructures
} // namespace coupling

template <class LinkedCell, unsigned int dim>
class coupling::datastructures::LinkedCellContainer : public coupling::datastructures::CellContainer<I02, dim> {
public:
  LinkedCellContainer(tarch::la::Vector<dim, unsigned int> numberLinkedCellsPerCouplingCell,
                               coupling::interface::MDSolverInterface<LinkedCell, dim>* mdSolverInterface);
  ~LinkedCellContainer();

  coupling::datastructures::CouplingCellWithLinkedCells<LinkedCell, dim>* getLinkedCellContainer();

  template <class A> void applyToLocalNonGhostCouplingCellsWithLinkedCells(A& a);
  template <class A> void applyToLocalGhostCouplingCellsWithLinkedCells(A& a);
  template <class A> void applyToAllLocalCouplingCellsWithLinkedCells(A& a);
  template <class A> void applyToFirstLayerOfGlobalNonGhostCellsWithLinkedCells(A& a);

  template <class A> void applyXLayersOfGlobalNonGhostCellsWithLinkedCells(A& a, unsigned int layers2Use);
 
private:
  coupling::datastructures::CouplingCellWithLinkedCells<LinkedCell, dim>*
  initLinkedCellContainer(tarch::la::Vector<dim, unsigned int> numberLinkedCellsPerCouplingCell,
                                   coupling::interface::MDSolverInterface<LinkedCell, dim>* mdSolverInterface) const;

  std::vector<coupling::datastructures::CouplingCell<dim>*> initCouplingCells() const;
  coupling::datastructures::CouplingCellWithLinkedCells<LinkedCell, dim>* _couplingCellsWithLinkedCells;
};
 
#include "LinkedCellContainer.cpph"
 
#endif // _MOLECULARDYNAMICS_COUPLING_DATASTRUCTURES_COUPLINGCELLSWITHLINKEDCELLS_H_