xref: /petsc/src/dm/impls/plex/tests/ex66.c (revision e7c9d70e6fa1f2421fa559ca7614e5a911560555)

static const char help[] = "Test for non-manifold interpolation";

#include <petscdmplex.h>

/*
       3-------------7
      /|            /|
     / |           / |
    /  |          /  |
   1-------------5   |
   |   |         |   |
   |   |         |   |
   |   |         |   |
   |   |         |   |
   z   4---------|---8
   ^  /          |  /
   | y           | /
   |/            |/
   2--->-x-------6-------------9
*/
int main(int argc, char **argv)
{
  DM        dm, idm;
  DMLabel   ctLabel;
  PetscBool has_vtk = PETSC_FALSE;

  // 9 vertices
  // 1 edge
  // 0 faces
  // 1 volume
  PetscInt num_points[4] = {9, 1, 0, 1};

  // point 0 = hexahedron (defined by 8 vertices)
  // points 1-9 = vertices
  // point 10 = edged (defined by 2 vertices)
  PetscInt cone_size[11] = {8, 0, 0, 0, 0, 0, 0, 0, 0, 0, 2};

  // hexahedron defined by points
  PetscInt    cones[11]             = {3, 4, 2, 1, 7, 5, 6, 8, 6, 9};
  PetscInt    cone_orientations[11] = {0, 0, 0, 0, 0, 0, 0, 0, 0, 0, 0};
  PetscScalar vertex_coords[3 * 9]  = {0, 0, 1, 0, 0, 0, 0, 1, 1, 0, 1, 0, 1, 0, 1, 1, 0, 0, 1, 1, 1, 1, 1, 0, 2, 0, 0};

  PetscFunctionBeginUser;
  PetscCall(PetscInitialize(&argc, &argv, NULL, help));
  PetscOptionsBegin(PETSC_COMM_WORLD, NULL, "Output VTK?", "ex66");
  PetscCall(PetscOptionsGetBool(NULL, NULL, "-vtk", &has_vtk, NULL));
  PetscOptionsEnd();

  PetscCall(DMCreate(PETSC_COMM_WORLD, &dm));
  PetscCall(PetscObjectSetName((PetscObject)dm, "cubeline-fromdag"));
  PetscCall(DMSetType(dm, DMPLEX));
  PetscCall(DMSetDimension(dm, 3));

  PetscCall(DMPlexCreateFromDAG(dm, 3, num_points, cone_size, cones, cone_orientations, vertex_coords));
  PetscCall(DMViewFromOptions(dm, NULL, "-dm_view"));

  // TODO: make it work with a DM made from a msh file
  // PetscCall(DMPlexCreateGmshFromFile(PETSC_COMM_WORLD, "cube-line.msh", PETSC_FALSE, &dm));
  // PetscCall(PetscObjectSetName((PetscObject)dm, "msh"));

  // Must set cell types
  PetscCall(DMPlexGetCellTypeLabel(dm, &ctLabel));
  PetscCall(DMLabelSetValue(ctLabel, 0, DM_POLYTOPE_HEXAHEDRON));
  PetscCall(DMLabelSetValue(ctLabel, 1, DM_POLYTOPE_POINT));
  PetscCall(DMLabelSetValue(ctLabel, 2, DM_POLYTOPE_POINT));
  PetscCall(DMLabelSetValue(ctLabel, 3, DM_POLYTOPE_POINT));
  PetscCall(DMLabelSetValue(ctLabel, 4, DM_POLYTOPE_POINT));
  PetscCall(DMLabelSetValue(ctLabel, 5, DM_POLYTOPE_POINT));
  PetscCall(DMLabelSetValue(ctLabel, 6, DM_POLYTOPE_POINT));
  PetscCall(DMLabelSetValue(ctLabel, 7, DM_POLYTOPE_POINT));
  PetscCall(DMLabelSetValue(ctLabel, 8, DM_POLYTOPE_POINT));
  PetscCall(DMLabelSetValue(ctLabel, 9, DM_POLYTOPE_POINT));
  PetscCall(DMLabelSetValue(ctLabel, 10, DM_POLYTOPE_SEGMENT));

  // interpolate (make sure to use -interp_dm_plex_stratify_celltype)
  PetscCall(PetscObjectSetOptionsPrefix((PetscObject)dm, "interp_"));
  PetscCall(DMPlexInterpolate(dm, &idm));
  PetscCall(DMDestroy(&dm));
  dm = idm;

  PetscCall(DMSetFromOptions(dm));
  PetscCall(DMViewFromOptions(dm, NULL, "-dm_view"));

  if (has_vtk) {
    PetscViewer viewer;
    PetscCall(PetscViewerCreate(PETSC_COMM_WORLD, &viewer));
    PetscCall(PetscViewerSetType(viewer, PETSCVIEWERVTK));
    PetscCall(PetscViewerFileSetName(viewer, "ex66.vtk"));
    PetscCall(DMView(dm, viewer));
    PetscCall(PetscViewerDestroy(&viewer));
  }

  PetscCall(DMDestroy(&dm));
  PetscCall(PetscFinalize());
  return 0;
}

/*TEST
  test:
    suffix: 0
    args: -interp_dm_plex_stratify_celltype -dm_view ::ascii_info_detail -interp_dm_view ::ascii_info_detail

TEST*/