impls/plex/plexcoarsen.c

#include <petsc/private/dmpleximpl.h> /*I      "petscdmplex.h"   I*/

PetscErrorCode DMCoarsen_Plex(DM dm, MPI_Comm comm, DM *dmCoarsened) {
  PetscFunctionBegin;
  if (!dm->coarseMesh) PetscCall(DMPlexCoarsen_Internal(dm, NULL, NULL, NULL, &dm->coarseMesh));
  PetscCall(PetscObjectReference((PetscObject)dm->coarseMesh));
  *dmCoarsened = dm->coarseMesh;
  PetscFunctionReturn(0);
}

PetscErrorCode DMCoarsenHierarchy_Plex(DM dm, PetscInt nlevels, DM dmCoarsened[]) {
  DM        rdm = dm;
  PetscInt  c;
  PetscBool localized;

  PetscFunctionBegin;
  PetscCall(DMGetCoordinatesLocalized(dm, &localized));
  for (c = nlevels - 1; c >= 0; --c) {
    PetscCall(DMCoarsen(rdm, PetscObjectComm((PetscObject)dm), &dmCoarsened[c]));
    PetscCall(DMCopyDisc(rdm, dmCoarsened[c]));
    if (localized) PetscCall(DMLocalizeCoordinates(dmCoarsened[c]));
    PetscCall(DMSetCoarseDM(rdm, dmCoarsened[c]));
    rdm = dmCoarsened[c];
  }
  PetscFunctionReturn(0);
}