CouplingCellWithLinkedCells.h

#ifndef _MOLECULARDYNAMICS_COUPLING_DATASTRUCTURES_COUPLINGCELLWITHLINKEDCELLS_H_
#define _MOLECULARDYNAMICS_COUPLING_DATASTRUCTURES_COUPLINGCELLWITHLINKEDCELLS_H_
 
#include "CouplingCell.h"
 
namespace coupling {
namespace datastructures {
template <class LinkedCell, unsigned int dim> class CouplingCellWithLinkedCells;
} // namespace datastructures
} // namespace coupling

template <class LinkedCell, unsigned int dim> class coupling::datastructures::CouplingCellWithLinkedCells : public coupling::datastructures::CouplingCell<dim> {
public:
  CouplingCellWithLinkedCells(tarch::la::Vector<dim, unsigned int> blockSize)
      : coupling::datastructures::CouplingCell<dim>(), _numberCells(getNumberCells(blockSize)), _linkedCells(NULL) {
 
    _linkedCells = new LinkedCell*[_numberCells];
    if (_linkedCells == NULL) {
      std::cout << "ERROR coupling::datastructures::CouplingCellWithLinkedCells: "
                   "_linkedCells == NULL"
                << std::endl;
      exit(EXIT_FAILURE);
    }
    // set each pointer to a NULL pointer
    for (unsigned int i = 0; i < _numberCells; i++) {
      _linkedCells[i] = NULL;
    }
  }

  virtual ~CouplingCellWithLinkedCells() {
    if (_linkedCells != NULL) {
      for (unsigned int i = 0; i < _numberCells; i++) {
        _linkedCells[i] = NULL;
      }
      delete[] _linkedCells;
      _linkedCells = NULL;
    }
  }

  void addLinkedCell(LinkedCell& cell, const unsigned int& index) { _linkedCells[index] = &cell; }

  template <class A> void iterateCells(A& a) {
    a.beginCellIteration();
    for (unsigned int i = 0; i < _numberCells; i++) {
      a.handleCell(*(_linkedCells[i]));
    }
    a.endCellIteration();
  }

  template <class A> void iterateConstCells(A& a) const {
    a.beginCellIteration();
    for (unsigned int i = 0; i < _numberCells; i++) {
      a.handleCell(*(_linkedCells[i]));
    }
    a.endCellIteration();
  }
 
private:
  unsigned int getNumberCells(tarch::la::Vector<dim, unsigned int> blockSize) const {
    unsigned int num = 1;
    for (unsigned int d = 0; d < dim; d++) {
      num = num * blockSize[d];
    }
    return num;
  }

  const unsigned int _numberCells;

  LinkedCell** _linkedCells;
};
 
#endif // _MOLECULARDYNAMICS_COUPLING_DATASTRUCTURES_COUPLINGCELLWITHLINKEDCELLS_H_