P4estDomainGenerator.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_P4ESTDOMAINGENERATOR_H
#define THUNDEREGG_P4ESTDOMAINGENERATOR_H
 
#include <ThunderEgg/DomainGenerator.h>
#include <functional>
#include <list>
#include <p4est_extended.h>
namespace ThunderEgg {
class P4estDomainGenerator : public DomainGenerator<2>
{
public:
  using BlockMapFunc =
    std::function<void(int block_no, double unit_x, double unit_y, double& x, double& y)>;
 
private:
  p4est_t* my_p4est;
  std::list<std::vector<PatchInfo<2>>> domain_patches;
 
  std::array<int, 2> ns;
  int num_ghost_cells;
  int curr_level;
  BlockMapFunc bmf;
 
  int id = 0;
 
  void extractLevel();
 
  void coarsenTree();
 
  void createPatchInfos();
 
  void linkNeighbors();
 
  void updateParentRanksOfPreviousDomain();
 
public:
  P4estDomainGenerator(p4est_t* p4est,
                       const std::array<int, 2>& ns,
                       int num_ghost_cells,
                       const BlockMapFunc& bmf);
  ~P4estDomainGenerator();
  P4estDomainGenerator(const P4estDomainGenerator&);
  P4estDomainGenerator(P4estDomainGenerator&&) = default;
  P4estDomainGenerator& operator=(const P4estDomainGenerator&);
  P4estDomainGenerator& operator=(P4estDomainGenerator&&) = default;
  Domain<2> getFinestDomain();
  bool hasCoarserDomain();
  Domain<2> getCoarserDomain();
};
} // namespace ThunderEgg
#endif