FineNbrInfo.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_FINENBRINFO_H
#define THUNDEREGG_FINENBRINFO_H
 
#include <ThunderEgg/BufferReader.h>
#include <ThunderEgg/BufferWriter.h>
#include <ThunderEgg/NbrInfo.h>
#include <ThunderEgg/Orthant.h>
 
namespace ThunderEgg {
template<int D>
class FineNbrInfo : public NbrInfo<D>
{
public:
  std::array<int, Orthant<D>::num_orthants> ranks;
  std::array<int, Orthant<D>::num_orthants> ids;
  std::array<int, Orthant<D>::num_orthants> global_indexes;
  std::array<int, Orthant<D>::num_orthants> local_indexes;
  FineNbrInfo()
  {
    ranks.fill(0);
    ids.fill(0);
    local_indexes.fill(-1);
    global_indexes.fill(-1);
  }
  ~FineNbrInfo() = default;
  FineNbrInfo(std::array<int, Orthant<D>::num_orthants> ids)
    : ids(ids)
  {
    ranks.fill(0);
    local_indexes.fill(-1);
    global_indexes.fill(-1);
  }
  NbrType getNbrType() const override { return NbrType::Fine; }
  void getNbrIds(std::deque<int>& nbr_ids) const override
  {
    for (size_t i = 0; i < ids.size(); i++) {
      nbr_ids.push_back(ids[i]);
    }
  };
  void getNbrRanks(std::deque<int>& nbr_ranks) const override
  {
    for (size_t i = 0; i < ranks.size(); i++) {
      nbr_ranks.push_back(ranks[i]);
    }
  }
  void setGlobalIndexes(const std::map<int, int>& id_to_global_index_map) override
  {
    for (size_t i = 0; i < global_indexes.size(); i++) {
      global_indexes[i] = id_to_global_index_map.at(ids[i]);
    }
  }
  void setLocalIndexes(const std::map<int, int>& id_to_local_index_map) override
  {
    for (size_t i = 0; i < local_indexes.size(); i++) {
      auto iter = id_to_local_index_map.find(ids[i]);
      if (iter != id_to_local_index_map.end()) {
        local_indexes[i] = iter->second;
      }
    }
  }
  int serialize(char* buffer) const override
  {
    BufferWriter writer(buffer);
    writer << ranks;
    writer << ids;
    return writer.getPos();
  }
  int deserialize(char* buffer) override
  {
    BufferReader reader(buffer);
    reader >> ranks;
    reader >> ids;
    return reader.getPos();
  }
  std::unique_ptr<NbrInfoBase> clone() const override
  {
    return std::make_unique<FineNbrInfo<D>>(*this);
  }
};
template<int D>
void
to_json(tpl::nlohmann::json& j, const FineNbrInfo<D>& n)
{
  j["type"] = NbrType::Fine;
  j["ids"] = n.ids;
  j["ranks"] = n.ranks;
}
template<int D>
void
from_json(const tpl::nlohmann::json& j, FineNbrInfo<D>& n)
{
  n.ids = j["ids"].get<std::array<int, Orthant<D>::num_orthants>>();
  n.ranks = j["ranks"].get<std::array<int, Orthant<D>::num_orthants>>();
}
} // namespace ThunderEgg
#endif