xref: /petsc/src/dm/tutorials/ex19.c (revision b122ec5aa1bd4469eb4e0673542fb7de3f411254) !

/*
     Demonstrates creating domain decomposition DAs and how to shuffle around data between the two
 */

#include <petscdm.h>
#include <petscdmda.h>

static char help[] = "Test for DMDA with overlap.\n\n";

int main(int argc,char **argv)
{
  DM             da;

  CHKERRQ(PetscInitialize(&argc,&argv,(char*)0,help));
  /* Build of the DMDA -- 1D -- boundary_none */
  CHKERRQ(PetscPrintf(PETSC_COMM_WORLD,"1D -- DM_BOUNDARY_NONE\n"));
  CHKERRQ(DMDACreate(PETSC_COMM_WORLD, &da));
  CHKERRQ(DMSetDimension(da, 1));
  CHKERRQ(DMDASetSizes(da, 8, 1, 1));
  CHKERRQ(DMDASetBoundaryType(da, DM_BOUNDARY_NONE, DM_BOUNDARY_NONE, DM_BOUNDARY_NONE));
  CHKERRQ(DMDASetDof(da, 1));
  CHKERRQ(DMDASetStencilWidth(da, 1));
  CHKERRQ(DMDASetOverlap(da,1,1,1));
  CHKERRQ(DMSetFromOptions(da));
  CHKERRQ(DMSetOptionsPrefix(da,"n1d_"));
  CHKERRQ(DMSetFromOptions(da));
  CHKERRQ(DMSetUp(da));
  CHKERRQ(DMView(da,PETSC_VIEWER_STDOUT_WORLD));
  CHKERRQ(DMDestroy(&da));

  /* Build of the DMDA -- 1D -- boundary_ghosted */
  CHKERRQ(PetscPrintf(PETSC_COMM_WORLD,"1D -- DM_BOUNDARY_GHOSTED\n"));
  CHKERRQ(DMDACreate(PETSC_COMM_WORLD, &da));
  CHKERRQ(DMSetDimension(da, 1));
  CHKERRQ(DMDASetSizes(da, 8, 1, 1));
  CHKERRQ(DMDASetBoundaryType(da, DM_BOUNDARY_GHOSTED, DM_BOUNDARY_GHOSTED, DM_BOUNDARY_GHOSTED));
  CHKERRQ(DMDASetDof(da, 2));
  CHKERRQ(DMDASetStencilWidth(da, 1));
  CHKERRQ(DMDASetOverlap(da,1,1,1));
  CHKERRQ(DMSetFromOptions(da));
  CHKERRQ(DMSetOptionsPrefix(da,"g1d_"));
  CHKERRQ(DMSetFromOptions(da));
  CHKERRQ(DMSetUp(da));
  CHKERRQ(DMView(da,PETSC_VIEWER_STDOUT_WORLD));
  CHKERRQ(DMDestroy(&da));

  /* Build of the DMDA -- 1D -- boundary_periodic */
  CHKERRQ(PetscPrintf(PETSC_COMM_WORLD,"1D -- DM_BOUNDARY_PERIODIC\n"));
  CHKERRQ(DMDACreate(PETSC_COMM_WORLD, &da));
  CHKERRQ(DMSetDimension(da, 1));
  CHKERRQ(DMDASetSizes(da, 8, 1, 1));
  CHKERRQ(DMDASetBoundaryType(da, DM_BOUNDARY_PERIODIC, DM_BOUNDARY_PERIODIC, DM_BOUNDARY_PERIODIC));
  CHKERRQ(DMDASetDof(da, 2));
  CHKERRQ(DMDASetStencilWidth(da, 1));
  CHKERRQ(DMDASetOverlap(da,1,1,1));
  CHKERRQ(DMSetFromOptions(da));
  CHKERRQ(DMSetOptionsPrefix(da,"p1d_"));
  CHKERRQ(DMSetFromOptions(da));
  CHKERRQ(DMSetUp(da));
  CHKERRQ(DMView(da,PETSC_VIEWER_STDOUT_WORLD));
  CHKERRQ(DMDestroy(&da));

  /* Build of the DMDA -- 2D -- boundary_none */
  CHKERRQ(PetscPrintf(PETSC_COMM_WORLD,"2D -- DM_BOUNDARY_NONE\n"));
  CHKERRQ(DMDACreate(PETSC_COMM_WORLD, &da));
  CHKERRQ(DMSetDimension(da, 2));
  CHKERRQ(DMDASetSizes(da, 8, 8, 1));
  CHKERRQ(DMDASetBoundaryType(da, DM_BOUNDARY_NONE, DM_BOUNDARY_NONE, DM_BOUNDARY_NONE));
  CHKERRQ(DMDASetDof(da, 2));
  CHKERRQ(DMDASetStencilWidth(da, 1));
  CHKERRQ(DMDASetOverlap(da,1,1,1));
  CHKERRQ(DMSetFromOptions(da));
  CHKERRQ(DMSetOptionsPrefix(da,"n2d_"));
  CHKERRQ(DMSetFromOptions(da));
  CHKERRQ(DMSetUp(da));
  CHKERRQ(DMView(da,PETSC_VIEWER_STDOUT_WORLD));
  CHKERRQ(DMDestroy(&da));

  /* Build of the DMDA -- 2D -- boundary_ghosted */
  CHKERRQ(PetscPrintf(PETSC_COMM_WORLD,"2D -- DM_BOUNDARY_GHOSTED\n"));
  CHKERRQ(DMDACreate(PETSC_COMM_WORLD, &da));
  CHKERRQ(DMSetDimension(da, 2));
  CHKERRQ(DMDASetSizes(da, 8, 8, 1));
  CHKERRQ(DMDASetBoundaryType(da, DM_BOUNDARY_GHOSTED, DM_BOUNDARY_GHOSTED, DM_BOUNDARY_GHOSTED));
  CHKERRQ(DMDASetDof(da, 2));
  CHKERRQ(DMDASetStencilWidth(da, 1));
  CHKERRQ(DMDASetOverlap(da,1,1,1));
  CHKERRQ(DMSetFromOptions(da));
  CHKERRQ(DMSetOptionsPrefix(da,"g2d_"));
  CHKERRQ(DMSetFromOptions(da));
  CHKERRQ(DMSetUp(da));
  CHKERRQ(DMView(da,PETSC_VIEWER_STDOUT_WORLD));
  CHKERRQ(DMDestroy(&da));

  /* Build of the DMDA -- 2D -- boundary_periodic */
  CHKERRQ(PetscPrintf(PETSC_COMM_WORLD,"2D -- DM_BOUNDARY_PERIODIC\n"));
  CHKERRQ(DMDACreate(PETSC_COMM_WORLD, &da));
  CHKERRQ(DMSetDimension(da, 2));
  CHKERRQ(DMDASetSizes(da, 8, 8, 1));
  CHKERRQ(DMDASetBoundaryType(da, DM_BOUNDARY_PERIODIC, DM_BOUNDARY_PERIODIC, DM_BOUNDARY_PERIODIC));
  CHKERRQ(DMDASetDof(da, 2));
  CHKERRQ(DMDASetStencilWidth(da, 1));
  CHKERRQ(DMDASetOverlap(da,1,1,1));
  CHKERRQ(DMSetFromOptions(da));
  CHKERRQ(DMSetOptionsPrefix(da,"p2d_"));
  CHKERRQ(DMSetFromOptions(da));
  CHKERRQ(DMSetUp(da));
  CHKERRQ(DMView(da,PETSC_VIEWER_STDOUT_WORLD));
  CHKERRQ(DMDestroy(&da));

  /* Build of the DMDA -- 3D -- boundary_none */
  CHKERRQ(PetscPrintf(PETSC_COMM_WORLD,"3D -- DM_BOUNDARY_NONE\n"));
  CHKERRQ(DMDACreate(PETSC_COMM_WORLD, &da));
  CHKERRQ(DMSetDimension(da, 3));
  CHKERRQ(DMDASetSizes(da, 8, 8, 8));
  CHKERRQ(DMDASetBoundaryType(da, DM_BOUNDARY_NONE, DM_BOUNDARY_NONE, DM_BOUNDARY_NONE));
  CHKERRQ(DMDASetDof(da, 2));
  CHKERRQ(DMDASetStencilWidth(da, 1));
  CHKERRQ(DMDASetOverlap(da,1,1,1));
  CHKERRQ(DMSetFromOptions(da));
  CHKERRQ(DMSetOptionsPrefix(da,"n3d_"));
  CHKERRQ(DMSetFromOptions(da));
  CHKERRQ(DMSetUp(da));
  CHKERRQ(DMView(da,PETSC_VIEWER_STDOUT_WORLD));
  CHKERRQ(DMDestroy(&da));

  /* Build of the DMDA -- 3D -- boundary_ghosted */
  CHKERRQ(PetscPrintf(PETSC_COMM_WORLD,"3D -- DM_BOUNDARY_GHOSTED\n"));
  CHKERRQ(DMDACreate(PETSC_COMM_WORLD, &da));
  CHKERRQ(DMSetDimension(da, 3));
  CHKERRQ(DMDASetSizes(da, 8, 8, 8));
  CHKERRQ(DMDASetBoundaryType(da, DM_BOUNDARY_GHOSTED, DM_BOUNDARY_GHOSTED, DM_BOUNDARY_GHOSTED));
  CHKERRQ(DMDASetDof(da, 2));
  CHKERRQ(DMDASetStencilWidth(da, 1));
  CHKERRQ(DMDASetOverlap(da,1,1,1));
  CHKERRQ(DMSetFromOptions(da));
  CHKERRQ(DMSetOptionsPrefix(da,"g3d_"));
  CHKERRQ(DMSetFromOptions(da));
  CHKERRQ(DMSetUp(da));
  CHKERRQ(DMView(da,PETSC_VIEWER_STDOUT_WORLD));
  CHKERRQ(DMDestroy(&da));

  /* Build of the DMDA -- 3D -- boundary_periodic */
  CHKERRQ(PetscPrintf(PETSC_COMM_WORLD,"3D -- DM_BOUNDARY_PERIODIC\n"));
  CHKERRQ(DMDACreate(PETSC_COMM_WORLD, &da));
  CHKERRQ(DMSetDimension(da, 3));
  CHKERRQ(DMDASetSizes(da, 8, 8, 8));
  CHKERRQ(DMDASetBoundaryType(da, DM_BOUNDARY_PERIODIC, DM_BOUNDARY_PERIODIC, DM_BOUNDARY_PERIODIC));
  CHKERRQ(DMDASetDof(da, 2));
  CHKERRQ(DMDASetStencilWidth(da, 1));
  CHKERRQ(DMDASetOverlap(da,1,1,1));
  CHKERRQ(DMSetFromOptions(da));
  CHKERRQ(DMSetOptionsPrefix(da,"p3d_"));
  CHKERRQ(DMSetFromOptions(da));
  CHKERRQ(DMSetUp(da));
  CHKERRQ(DMView(da,PETSC_VIEWER_STDOUT_WORLD));
  CHKERRQ(DMDestroy(&da));

  /* test moving data in and out */
  CHKERRQ(PetscFinalize());
  return 0;
}

/*TEST

   test:

TEST*/