CellContainer.h

// This file is part of the Mamico project. For conditions of distribution
// and use, please see the copyright notice in Mamico's main folder
#pragma once
 
#include "coupling/CouplingMDDefinitions.h"
#include "coupling/datastructures/CouplingCell.h"
#include "coupling/indexing/IndexingService.h"
 
namespace coupling {
namespace datastructures {
template <class CellIndexT, unsigned int dim> class CellContainer;
} // namespace datastructures
} // namespace coupling

template <class CellIndexT, unsigned int dim> class coupling::datastructures::CellContainer {
 
public:
  CellContainer() { _couplingCells.reserve(CellIndexT::linearNumberCellsInDomain); }
  CellContainer(std::vector<coupling::datastructures::CouplingCell<dim>*> couplingCells) {
    _couplingCells.reserve(CellIndexT::linearNumberCellsInDomain);
    for (auto cell : couplingCells) {
      *this << cell;
    }
  }

  coupling::datastructures::CouplingCell<dim>* operator[](CellIndexT index) const {
#if (COUPLING_MD_ERROR == COUPLING_MD_YES)
    if (_couplingCells.size() < CellIndexT::linearNumberCellsInDomain) {
      std::cout << "CellContainer<" << CellIndexT::TNAME << "," << dim << "> accessed but not full " << std::endl;
      std::exit(EXIT_FAILURE);
    }
#endif
    return _couplingCells[indexing::convertToScalar(index)];
  }

  void operator<<(coupling::datastructures::CouplingCell<dim>* couplingCell) {
#if (COUPLING_MD_ERROR == COUPLING_MD_YES)
    if (_couplingCells.size() >= CellIndexT::linearNumberCellsInDomain) {
      std::cout << "CellContainer<" << CellIndexT::TNAME << "," << dim << "> can only hold " << CellIndexT::linearNumberCellsInDomain << " coupling cells!"
                << std::endl;
      std::exit(EXIT_FAILURE);
    }
#endif
    _couplingCells.push_back(couplingCell);
  }

  unsigned int size() const { return _couplingCells.size(); }

  class Iterator {
  public:
    using CouplingCellIterator = typename std::vector<coupling::datastructures::CouplingCell<dim>*>::const_iterator;
 
    Iterator(CouplingCellIterator itCouplingCells, typename CellIndexT::IndexIterator itIdx) : _itCouplingCells(itCouplingCells), _itIdx(itIdx) {}

    const std::pair<coupling::datastructures::CouplingCell<dim>*, CellIndexT> operator*() const { return std::make_pair(*_itCouplingCells, *_itIdx); }
 
    Iterator& operator++() {
      ++_itCouplingCells;
      ++_itIdx;
      return *this;
    }
 
    Iterator operator++(int) {
      Iterator tmp = *this;
      ++(*this);
      return tmp;
    }
 
    friend bool operator==(const Iterator& a, const Iterator& b) { return a._itCouplingCells == b._itCouplingCells; }
 
    friend bool operator!=(const Iterator& a, const Iterator& b) { return !(a == b); }
 
  private:
    CouplingCellIterator _itCouplingCells;

    typename CellIndexT::IndexIterator _itIdx;
  };

  Iterator begin() const { return Iterator(_couplingCells.begin(), CellIndexT::begin()); }

  Iterator end() const { return Iterator(_couplingCells.end(), CellIndexT::end()); }
 
protected:
  std::vector<coupling::datastructures::CouplingCell<dim>*> _couplingCells;
};