Level.h

Go to the documentation of this file.

/***************************************************************************
 *  ThunderEgg, a library for solvers on adaptively refined block-structured
 *  Cartesian grids.
 *
 *  Copyright (c) 2018-2021 Scott Aiton
 *
 *  This program is free software: you can redistribute it and/or modify
 *  it under the terms of the GNU General Public License as published by
 *  the Free Software Foundation, either version 3 of the License, or
 *  (at your option) any later version.
 *
 *  This program is distributed in the hope that it will be useful,
 *  but WITHOUT ANY WARRANTY; without even the implied warranty of
 *  MERCHANTABILITY or FITNESS FOR A PARTICULAR PURPOSE.  See the
 *  GNU General Public License for more details.
 *
 *  You should have received a copy of the GNU General Public License
 *  along with this program.  If not, see <https://www.gnu.org/licenses/>.
 ***************************************************************************/
 
#ifndef THUNDEREGG_GMG_LEVEL_H
#define THUNDEREGG_GMG_LEVEL_H
 
#include <ThunderEgg/GMG/Interpolator.h>
#include <ThunderEgg/GMG/Restrictor.h>
#include <ThunderEgg/GMG/Smoother.h>
#include <ThunderEgg/Operator.h>
#include <memory>
namespace ThunderEgg::GMG {
template<int D>
class Level
{
private:
  std::shared_ptr<const Operator<D>> op;
  std::shared_ptr<const Restrictor<D>> restrictor;
  std::shared_ptr<const Interpolator<D>> interpolator;
  std::shared_ptr<const Smoother<D>> smoother;
  std::shared_ptr<const Level> coarser;
 
public:
  Level() {}
  void setRestrictor(const Restrictor<D>& restrictor)
  {
    this->restrictor.reset(restrictor.clone());
  }
  const Restrictor<D>& getRestrictor() const
  {
    if (restrictor == nullptr) {
      throw RuntimeError("This level does not have a restrictor");
    }
    return *restrictor;
  }
  void setInterpolator(const Interpolator<D>& interpolator)
  {
    this->interpolator.reset(interpolator.clone());
  }
  const Interpolator<D>& getInterpolator() const
  {
    if (interpolator == nullptr) {
      throw RuntimeError("This level does not have an interpolator");
    }
    return *interpolator;
  }
  void setOperator(const Operator<D>& op) { this->op.reset(op.clone()); }
  const Operator<D>& getOperator() const
  {
    if (op == nullptr) {
      throw RuntimeError("This level does not have an Operator");
    }
    return *op;
  }
  void setSmoother(const Smoother<D>& smoother) { this->smoother.reset(smoother.clone()); }
  const Smoother<D>& getSmoother() const
  {
    if (smoother == nullptr) {
      throw RuntimeError("This level does not have a smoother");
    }
    return *smoother;
  }
  void setCoarser(std::shared_ptr<const Level> coarser) { this->coarser = coarser; }
  const Level& getCoarser() const
  {
    if (coarser == nullptr) {
      throw RuntimeError("This level does not have a coarser level.");
    }
    return *coarser;
  }
  bool finest() const { return interpolator == nullptr; }
  bool coarsest() const { return coarser == nullptr; }
};
extern template class Level<2>;
extern template class Level<3>;
} // namespace ThunderEgg::GMG
#endif