PatchIfaceInfo.h

Go to the documentation of this file.

/***************************************************************************
 *  ThunderEgg, a library for solvers on adaptively refined block-structured
 *  Cartesian grids.
 *
 *  Copyright (c) 2020-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_SCHUR_PATCHIFACEINFO_H
#define THUNDEREGG_SCHUR_PATCHIFACEINFO_H
 
#include <ThunderEgg/RuntimeError.h>
#include <ThunderEgg/Schur/CoarseIfaceInfo.h>
#include <ThunderEgg/Schur/FineIfaceInfo.h>
#include <ThunderEgg/Schur/NormalIfaceInfo.h>
namespace ThunderEgg {
namespace Schur {
template<int D>
class PatchIfaceInfo : public Serializable
{
public:
  PatchInfo<D> pinfo;
  std::array<std::shared_ptr<IfaceInfo<D>>, Side<D>::number_of> iface_info;
  PatchIfaceInfo() = default;
  explicit PatchIfaceInfo(const PatchInfo<D>& pinfo)
    : pinfo(pinfo)
  {
    // create iface objects
    for (Side<D> s : Side<D>::getValues()) {
      if (pinfo.hasNbr(s)) {
        switch (pinfo.getNbrType(s)) {
          case NbrType::Normal:
            setIfaceInfo(s, std::make_shared<NormalIfaceInfo<D>>(pinfo, s));
            break;
          case NbrType::Fine:
            setIfaceInfo(s, std::make_shared<FineIfaceInfo<D>>(pinfo, s));
            break;
          case NbrType::Coarse:
            setIfaceInfo(s, std::make_shared<CoarseIfaceInfo<D>>(pinfo, s));
            break;
          default:
            throw RuntimeError("Unsupported NbrType");
        }
      }
    }
  }
  void setIfaceInfo(Side<D> s, std::shared_ptr<IfaceInfo<D>> info)
  {
    iface_info[s.getIndex()] = info;
  }
  std::shared_ptr<IfaceInfo<D>> getIfaceInfo(Side<D> s) { return iface_info[s.getIndex()]; }
  std::shared_ptr<const IfaceInfo<D>> getIfaceInfo(Side<D> s) const
  {
    return iface_info[s.getIndex()];
  }
  std::shared_ptr<NormalIfaceInfo<D>> getNormalIfaceInfo(Side<D> s)
  {
    return std::dynamic_pointer_cast<NormalIfaceInfo<D>>(iface_info[s.getIndex()]);
  }
  std::shared_ptr<const NormalIfaceInfo<D>> getNormalIfaceInfo(Side<D> s) const
  {
    return std::dynamic_pointer_cast<const NormalIfaceInfo<D>>(iface_info[s.getIndex()]);
  }
  std::shared_ptr<CoarseIfaceInfo<D>> getCoarseIfaceInfo(Side<D> s)
  {
    return std::dynamic_pointer_cast<CoarseIfaceInfo<D>>(iface_info[s.getIndex()]);
  }
  std::shared_ptr<const CoarseIfaceInfo<D>> getCoarseIfaceInfo(Side<D> s) const
  {
    return std::dynamic_pointer_cast<const CoarseIfaceInfo<D>>(iface_info[s.getIndex()]);
  }
  std::shared_ptr<FineIfaceInfo<D>> getFineIfaceInfo(Side<D> s)
  {
    return std::dynamic_pointer_cast<FineIfaceInfo<D>>(iface_info[s.getIndex()]);
  }
  std::shared_ptr<const FineIfaceInfo<D>> getFineIfaceInfo(Side<D> s) const
  {
    return std::dynamic_pointer_cast<const FineIfaceInfo<D>>(iface_info[s.getIndex()]);
  }
  int serialize(char* buffer) const
  {
    BufferWriter writer(buffer);
    writer << pinfo;
    return writer.getPos();
  }
  int deserialize(char* buffer)
  {
    BufferReader reader(buffer);
    auto pinfo_in = std::make_shared<PatchInfo<D>>();
    reader >> pinfo;
    // create iface objects
    for (Side<D> s : Side<D>::getValues()) {
      if (pinfo.hasNbr(s)) {
        switch (pinfo.getNbrType(s)) {
          case NbrType::Normal:
            setIfaceInfo(s, std::make_shared<NormalIfaceInfo<D>>(pinfo, s));
            break;
          case NbrType::Fine:
            setIfaceInfo(s, std::make_shared<FineIfaceInfo<D>>(pinfo, s));
            break;
          case NbrType::Coarse:
            setIfaceInfo(s, std::make_shared<CoarseIfaceInfo<D>>(pinfo, s));
            break;
          default:
            throw RuntimeError("Unsupported NbrType");
        }
      }
    }
    return reader.getPos();
  }
};
extern template class PatchIfaceInfo<2>;
extern template class PatchIfaceInfo<3>;
} // namespace Schur
} // namespace ThunderEgg
#endif