xref: /petsc/src/dm/impls/plex/plexgenerate.c (revision 1ed6e3ff8437baa411029a28c2b08f047df9ad9a)

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

/*@
  DMPlexInvertCell - Flips cell orientations since `DMPLEX` stores some of them internally with outward normals.

  Input Parameters:
+ cellType - The cell type
- cone     - The incoming cone

  Output Parameter:
. cone - The inverted cone (in-place)

  Level: developer

.seealso: [](ch_unstructured), `DM`, `DMPLEX`, `DMPolytopeType`, `DMPlexGenerate()`
@*/
PetscErrorCode DMPlexInvertCell(DMPolytopeType cellType, PetscInt cone[])
{
#define SWAPCONE(cone, i, j) \
  do { \
    PetscInt _cone_tmp; \
    _cone_tmp = (cone)[i]; \
    (cone)[i] = (cone)[j]; \
    (cone)[j] = _cone_tmp; \
  } while (0)

  PetscFunctionBegin;
  switch (cellType) {
  case DM_POLYTOPE_POINT:
    break;
  case DM_POLYTOPE_SEGMENT:
    break;
  case DM_POLYTOPE_POINT_PRISM_TENSOR:
    break;
  case DM_POLYTOPE_TRIANGLE:
    break;
  case DM_POLYTOPE_QUADRILATERAL:
    break;
  case DM_POLYTOPE_SEG_PRISM_TENSOR:
    SWAPCONE(cone, 2, 3);
    break;
  case DM_POLYTOPE_TETRAHEDRON:
    SWAPCONE(cone, 0, 1);
    break;
  case DM_POLYTOPE_HEXAHEDRON:
    SWAPCONE(cone, 1, 3);
    break;
  case DM_POLYTOPE_TRI_PRISM:
    SWAPCONE(cone, 1, 2);
    break;
  case DM_POLYTOPE_TRI_PRISM_TENSOR:
    break;
  case DM_POLYTOPE_QUAD_PRISM_TENSOR:
    break;
  case DM_POLYTOPE_PYRAMID:
    SWAPCONE(cone, 1, 3);
    break;
  default:
    break;
  }
  PetscFunctionReturn(PETSC_SUCCESS);
#undef SWAPCONE
}

/*@
  DMPlexReorderCell - Flips cell orientations since `DMPLEX` stores some of them internally with outward normals.

  Input Parameters:
+ dm   - The `DMPLEX` object
. cell - The cell
- cone - The incoming cone

  Output Parameter:
. cone - The reordered cone (in-place)

  Level: developer

.seealso: [](ch_unstructured), `DM`, `DMPLEX`, `DMPolytopeType`, `DMPlexGenerate()`
@*/
PetscErrorCode DMPlexReorderCell(DM dm, PetscInt cell, PetscInt cone[])
{
  DMPolytopeType cellType;

  PetscFunctionBegin;
  PetscCall(DMPlexGetCellType(dm, cell, &cellType));
  PetscCall(DMPlexInvertCell(cellType, cone));
  PetscFunctionReturn(PETSC_SUCCESS);
}

/*@C
  DMPlexTriangleSetOptions - Set the options used for the Triangle mesh generator

  Not Collective

  Input Parameters:
+ dm   - The `DMPLEX` object
- opts - The command line options

  Level: developer

.seealso: [](ch_unstructured), `DM`, `DMPLEX`, `DMPlexTetgenSetOptions()`, `DMPlexGenerate()`
@*/
PetscErrorCode DMPlexTriangleSetOptions(DM dm, const char *opts)
{
  DM_Plex *mesh = (DM_Plex *)dm->data;

  PetscFunctionBegin;
  PetscValidHeaderSpecific(dm, DM_CLASSID, 1);
  PetscAssertPointer(opts, 2);
  PetscCall(PetscFree(mesh->triangleOpts));
  PetscCall(PetscStrallocpy(opts, &mesh->triangleOpts));
  PetscFunctionReturn(PETSC_SUCCESS);
}

/*@C
  DMPlexTetgenSetOptions - Set the options used for the Tetgen mesh generator

  Not Collective

  Input Parameters:
+ dm   - The `DMPLEX` object
- opts - The command line options

  Level: developer

.seealso: [](ch_unstructured), `DM`, `DMPLEX`, `DMPlexTriangleSetOptions()`, `DMPlexGenerate()`
@*/
PetscErrorCode DMPlexTetgenSetOptions(DM dm, const char *opts)
{
  DM_Plex *mesh = (DM_Plex *)dm->data;

  PetscFunctionBegin;
  PetscValidHeaderSpecific(dm, DM_CLASSID, 1);
  PetscAssertPointer(opts, 2);
  PetscCall(PetscFree(mesh->tetgenOpts));
  PetscCall(PetscStrallocpy(opts, &mesh->tetgenOpts));
  PetscFunctionReturn(PETSC_SUCCESS);
}

/*@
  DMPlexGenerate - Generates a mesh.

  Not Collective

  Input Parameters:
+ boundary    - The `DMPLEX` boundary object
. name        - The mesh generation package name
- interpolate - Flag to create intermediate mesh elements

  Output Parameter:
. mesh - The `DMPLEX` object

  Options Database Keys:
+ -dm_plex_generate <name> - package to generate mesh, for example, triangle, ctetgen or tetgen
- -dm_generator <name>     - package to generate mesh, for example, triangle, ctetgen or tetgen

  Level: intermediate

.seealso: [](ch_unstructured), `DM`, `DMPLEX`, `DMPlexCreate()`, `DMRefine()`
@*/
PetscErrorCode DMPlexGenerate(DM boundary, const char name[], PetscBool interpolate, DM *mesh)
{
  DMGeneratorFunctionList fl;
  char                    genname[PETSC_MAX_PATH_LEN];
  const char             *suggestions;
  PetscInt                dim;
  PetscBool               flg;

  PetscFunctionBegin;
  PetscValidHeaderSpecific(boundary, DM_CLASSID, 1);
  PetscValidLogicalCollectiveBool(boundary, interpolate, 3);
  PetscCall(DMGetDimension(boundary, &dim));
  PetscCall(PetscOptionsGetString(((PetscObject)boundary)->options, ((PetscObject)boundary)->prefix, "-dm_generator", genname, sizeof(genname), &flg));
  if (flg) name = genname;
  else {
    PetscCall(PetscOptionsGetString(((PetscObject)boundary)->options, ((PetscObject)boundary)->prefix, "-dm_plex_generate", genname, sizeof(genname), &flg));
    if (flg) name = genname;
  }

  fl = DMGenerateList;
  if (name) {
    while (fl) {
      PetscCall(PetscStrcmp(fl->name, name, &flg));
      if (flg) {
        PetscCall((*fl->generate)(boundary, interpolate, mesh));
        PetscFunctionReturn(PETSC_SUCCESS);
      }
      fl = fl->next;
    }
    SETERRQ(PETSC_COMM_SELF, PETSC_ERR_ARG_OUTOFRANGE, "Grid generator %s not registered; you may need to add --download-%s to your ./configure options", name, name);
  } else {
    while (fl) {
      if (boundary->dim == fl->dim) {
        PetscCall((*fl->generate)(boundary, interpolate, mesh));
        PetscFunctionReturn(PETSC_SUCCESS);
      }
      fl = fl->next;
    }
    suggestions = "";
    if (boundary->dim + 1 == 2) suggestions = " You may need to add --download-triangle to your ./configure options";
    else if (boundary->dim + 1 == 3) suggestions = " You may need to add --download-ctetgen or --download-tetgen in your ./configure options";
    SETERRQ(PETSC_COMM_SELF, PETSC_ERR_ARG_OUTOFRANGE, "No grid generator of dimension %" PetscInt_FMT " registered%s", boundary->dim + 1, suggestions);
  }
}