CD_ItoIteratorImplem.H

Go to the documentation of this file.

/* chombo-discharge
 * Copyright © 2021 SINTEF Energy Research.
 * Please refer to Copyright.txt and LICENSE in the chombo-discharge root directory.
 */
 
#ifndef CD_ItoIteratorImplem_H
#define CD_ItoIteratorImplem_H
 
// Chombo includes
#include <CH_Timer.H>
 
// Our includes
#include <CD_ItoIterator.H>
#include <CD_NamespaceHeader.H>
 
template <class T>
ItoIterator<T>::ItoIterator(ItoLayout<T>& a_layout, const SpeciesIteration a_mode)
{
  CH_TIME("ItoIterator<T>::ItoIterator<T>");
 
  m_solvers    = a_layout.getSolvers();
  m_species    = a_layout.getSpecies();
  m_numSolvers = m_solvers.size();
  m_mode       = a_mode;
  m_isDefined  = true;
 
  this->reset();
}
 
template <class T>
ItoIterator<T>::~ItoIterator<T>()
{
  CH_TIME("ItoIterator<T>::~ItoIterator<T>");
}
 
template <class T>
int
ItoIterator<T>::getNumberOfSolvers()
{
  CH_assert(m_isDefined);
 
  return m_numSolvers;
}
 
template <class T>
int
ItoIterator<T>::index() const
{
  CH_assert(m_isDefined);
 
  return m_isolver;
}
 
template <class T>
void
ItoIterator<T>::reset()
{
  CH_assert(m_isDefined);
 
  m_isolver = 0;
}
 
template <class T>
bool
ItoIterator<T>::ok()
{
  return (m_isolver < m_numSolvers);
}
 
template <class T>
void
ItoIterator<T>::operator++()
{
  CH_assert(m_isDefined);
 
  if (m_mode == SpeciesIteration::All) {
    m_isolver++;
  }
  else {
    m_isolver++;
 
    for (int isolver = m_isolver; this->ok(); ++isolver) {
      const RefCountedPtr<ItoSpecies>& species = m_species[isolver];
 
      if (m_mode == SpeciesIteration::Charged) {
        if (species->getChargeNumber() != 0) {
          m_isolver = isolver;
 
          break;
        }
      }
      else if (m_mode == SpeciesIteration::Negative) {
        if (species->getChargeNumber() < 0) {
          m_isolver = isolver;
 
          break;
        }
      }
      else if (m_mode == SpeciesIteration::Positive) {
        if (species->getChargeNumber() > 0) {
          m_isolver = isolver;
 
          break;
        }
      }
      else if (m_mode == SpeciesIteration::Neutral) {
        if (species->getChargeNumber() == 0) {
          m_isolver = isolver;
 
          break;
        }
      }
    }
  }
}
 
template <class T>
RefCountedPtr<T>&
ItoIterator<T>::operator()()
{
  CH_assert(m_isDefined);
  CH_assert(m_isolver < m_numSolvers);
 
  return m_solvers[m_isolver];
}
 
template <class T>
RefCountedPtr<ItoSpecies>&
ItoIterator<T>::getSpecies()
{
  CH_assert(m_isDefined);
  CH_assert(m_isolver < m_numSolvers);
 
  return m_species[m_isolver];
}
 
#include <CD_NamespaceFooter.H>
 
#endif