CD_PetscGrid.H

Go to the documentation of this file.

/* chombo-discharge
 * Copyright © 2026 SINTEF Energy Research.
 * Please refer to Copyright.txt and LICENSE in the chombo-discharge root directory.
 */
 
#ifndef CD_PetscGrid_H
#define CD_PetscGrid_H
 
#ifdef CH_USE_PETSC
 
// Petsc includes
#include "petsc.h"
#include "petscis.h"
 
// Chombo includes
#include <DisjointBoxLayout.H>
#include <EBCellFAB.H>
#include <MFCellFAB.H>
#include <EBISLayout.H>
 
// Our includes
#include <CD_DischargeIO.H>
#include <CD_EBAMRData.H>
#include <CD_MFLevelGrid.H>
#include <CD_PetscDOF.H>
#include <CD_PetscAMRCell.H>
#include <CD_NamespaceHeader.H>
 
class PetscGrid
{
public:
  PetscGrid() noexcept;
 
  virtual ~PetscGrid() noexcept;
 
  virtual void
  define(const Vector<RefCountedPtr<MFLevelGrid>>&              a_levelGrids,
         const Vector<RefCountedPtr<MFLevelGrid>>&              a_levelGridsCoFi,
         const Vector<RefCountedPtr<MFLevelGrid>>&              a_levelGridsFiCo,
         const Vector<RefCountedPtr<LevelData<BaseFab<bool>>>>& a_validCells,
         const Vector<RefCountedPtr<LevelData<BaseFab<bool>>>>& a_coarHaloCF,
         const Vector<RefCountedPtr<LevelData<BaseFab<bool>>>>& a_fineHaloCF,
         const Vector<int>&                                     a_refinementRatios,
         const Vector<Real>&                                    a_dx,
         const int                                              a_finestLevel,
         const int                                              a_numGhost) noexcept;
 
  virtual void
  clear() noexcept;
 
  virtual void
  create(Vec& x) noexcept;
 
  virtual void
  destroy(Vec& x) noexcept;
 
  virtual void
  setValue(Vec& a_x, const PetscScalar a_value) const noexcept;
 
  virtual void
  putChomboInPetsc(Vec& a_x, const MFAMRCellData& a_y) const noexcept;
 
  virtual void
  putPetscInChombo(MFAMRCellData& a_y, const Vec& a_x) const noexcept;
 
  virtual void
  dumpPetscGrid(const std::string a_filename) const noexcept;
 
protected:
  bool m_isDefined;
 
  bool m_verbose;
 
  bool m_debug;
 
  bool m_profile;
 
  int m_finestLevel;
 
  int m_numGhost;
 
  int m_numPhases;
 
  PetscInt m_numLocalRows;
 
  PetscInt m_numGlobalRows;
 
  PetscInt m_localRowBegin;
 
  ISLocalToGlobalMapping m_localToGlobalIS;
 
  Vector<PetscInt> m_numRowsPerRank;
 
  Vector<RefCountedPtr<MFLevelGrid>> m_levelGrids;
 
  Vector<RefCountedPtr<MFLevelGrid>> m_levelGridsCoFi;
 
  Vector<RefCountedPtr<MFLevelGrid>> m_levelGridsFiCo;
 
  Vector<RefCountedPtr<LevelData<BaseFab<bool>>>> m_validCells;
 
  Vector<RefCountedPtr<LevelData<BaseFab<bool>>>> m_coarHaloCF;
 
  Vector<RefCountedPtr<LevelData<BaseFab<bool>>>> m_fineHaloCF;
 
  Vector<int> m_refRat;
 
  Vector<Real> m_dx;
 
  Vector<RefCountedPtr<LevelData<BaseFab<PetscAMRCell>>>> m_amrToPetsc;
 
  Vector<RefCountedPtr<LevelData<BaseFab<PetscAMRCell>>>> m_amrToPetscCoFi;
 
  Vector<RefCountedPtr<LevelData<BaseFab<PetscAMRCell>>>> m_amrToPetscFiCo;
 
  Vector<RefCountedPtr<LayoutData<Vector<PetscDOF>>>> m_petscToAMR[2];
 
  virtual void
  defineAMRCells() noexcept;
 
  virtual void
  definePetscRows() noexcept;
 
  virtual void
  defineCoFiBuffers() noexcept;
};
 
#include <CD_NamespaceFooter.H>
 
#endif
 
#endif