PatchOperator.h

Go to the documentation of this file.

/***************************************************************************
 *  ThunderEgg, a library for solvers on adaptively refined block-structured
 *  Cartesian grids.
 *
 *  Copyright (c) 2019-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_PATCHOPERATOR_H
#define THUNDEREGG_PATCHOPERATOR_H
 
#include <ThunderEgg/Domain.h>
#include <ThunderEgg/GhostFiller.h>
#include <ThunderEgg/Operator.h>
#include <ThunderEgg/Vector.h>
namespace ThunderEgg {
template<int D>
class PatchOperator : public Operator<D>
{
private:
  Domain<D> domain;
  std::shared_ptr<const GhostFiller<D>> ghost_filler;
 
public:
  PatchOperator(const Domain<D>& domain, const GhostFiller<D>& ghost_filler)
    : domain(domain)
    , ghost_filler(ghost_filler.clone())
  {}
  virtual PatchOperator<D>* clone() const override = 0;
  virtual ~PatchOperator() {}
 
  virtual void applySinglePatch(const PatchInfo<D>& pinfo,
                                const PatchView<const double, D>& u_view,
                                const PatchView<double, D>& f_view) const = 0;
 
  virtual void modifyRHSForInternalBoundaryConditions(const PatchInfo<D>& pinfo,
                                                      const PatchView<const double, D>& u_view,
                                                      const PatchView<double, D>& f_view) const = 0;
 
  virtual void applySinglePatchWithInternalBoundaryConditions(
    const PatchInfo<D>& pinfo,
    const PatchView<const double, D>& u_view,
    const PatchView<double, D>& f_view) const = 0;
 
  void apply(const Vector<D>& u, Vector<D>& f) const override
  {
    if constexpr (ENABLE_DEBUG) {
      if (u.getNumLocalPatches() != domain.getNumLocalPatches()) {
        throw RuntimeError("u vector is incorrect length");
      }
      if (f.getNumLocalPatches() != domain.getNumLocalPatches()) {
        throw RuntimeError("f vector is incorrect length");
      }
    }
    f.setWithGhost(0);
    ghost_filler->fillGhost(u);
    for (const PatchInfo<D>& pinfo : domain.getPatchInfoVector()) {
      PatchView<const double, D> u_view = u.getPatchView(pinfo.local_index);
      PatchView<double, D> f_view = f.getPatchView(pinfo.local_index);
      applySinglePatch(pinfo, u_view, f_view);
    }
  }
  const Domain<D>& getDomain() const { return domain; }
  const GhostFiller<D>& getGhostFiller() const { return *ghost_filler; }
};
} // namespace ThunderEgg
extern template class ThunderEgg::PatchOperator<2>;
extern template class ThunderEgg::PatchOperator<3>;
#endif