P8estDomainGenerator.h

Go to the documentation of this file.

/***************************************************************************
 *  ThunderEgg, a library for solvers on adaptively refined block-structured
 *  Cartesian grids.
 *
 *  Copyright (c) 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_P8ESTDOMAINGENERATOR_H
#define THUNDEREGG_P8ESTDOMAINGENERATOR_H
 
#include <ThunderEgg/DomainGenerator.h>
#include <functional>
#include <p8est.h>
namespace ThunderEgg {
class P8estDomainGenerator : public DomainGenerator<3>
{
public:
  using BlockMapFunc = std::function<void(int block_no,
                                          double unit_x,
                                          double unit_y,
                                          double unit_z,
                                          double& x,
                                          double& y,
                                          double& z)>;
 
private:
  p8est_t* my_p8est;
  std::list<std::vector<PatchInfo<3>>> domain_patches;
 
  std::array<int, 3> ns;
  int num_ghost_cells;
  int curr_level;
  BlockMapFunc bmf;
 
  int id = 0;
 
  void extractLevel();
 
  void coarsenTree();
 
  void createPatchInfos();
 
  void linkNeighbors();
 
  void updateParentRanksOfPreviousDomain();
 
public:
  P8estDomainGenerator(p8est_t* p8est,
                       const std::array<int, 3>& ns,
                       int num_ghost_cells,
                       const BlockMapFunc& bmf);
  P8estDomainGenerator(P8estDomainGenerator&);
  P8estDomainGenerator(P8estDomainGenerator&&) = default;
  P8estDomainGenerator& operator=(const P8estDomainGenerator&);
  P8estDomainGenerator& operator=(P8estDomainGenerator&&) = default;
  ~P8estDomainGenerator();
  Domain<3> getFinestDomain();
  bool hasCoarserDomain();
  Domain<3> getCoarserDomain();
};
} // namespace ThunderEgg
#endif