1 static char help[] = "Tests various DMComposite routines.\n\n"; 2 3 #include <petscdm.h> 4 #include <petscdmda.h> 5 #include <petscdmcomposite.h> 6 7 int main(int argc, char **argv) 8 { 9 PetscMPIInt rank; 10 DM da1, da2, packer; 11 Vec local, global, globals[2], buffer; 12 PetscScalar value, *array; 13 PetscInt n; 14 PetscViewer viewer; 15 16 PetscFunctionBeginUser; 17 PetscCall(PetscInitialize(&argc, &argv, (char *)0, help)); 18 19 PetscCall(DMCompositeCreate(PETSC_COMM_WORLD, &packer)); 20 PetscCall(DMDACreate1d(PETSC_COMM_WORLD, DM_BOUNDARY_NONE, 8, 1, 1, NULL, &da1)); 21 PetscCall(DMSetFromOptions(da1)); 22 PetscCall(DMSetUp(da1)); 23 PetscCall(DMCompositeAddDM(packer, da1)); 24 PetscCall(DMDACreate1d(PETSC_COMM_WORLD, DM_BOUNDARY_NONE, 6, 1, 1, NULL, &da2)); 25 PetscCall(DMSetFromOptions(da2)); 26 PetscCall(DMSetUp(da2)); 27 PetscCall(DMCompositeAddDM(packer, da2)); 28 29 PetscCall(DMCreateGlobalVector(packer, &global)); 30 PetscCall(DMCreateLocalVector(packer, &local)); 31 PetscCall(DMCreateLocalVector(packer, &buffer)); 32 33 PetscCall(DMCompositeGetAccessArray(packer, global, 2, NULL, globals)); 34 value = 1; 35 PetscCall(VecSet(globals[0], value)); 36 value = -1; 37 PetscCall(VecSet(globals[1], value)); 38 PetscCallMPI(MPI_Comm_rank(PETSC_COMM_WORLD, &rank)); 39 value = rank + 1; 40 PetscCall(VecGetLocalSize(globals[0], &n)); 41 PetscCall(VecGetArray(globals[0], &array)); 42 for (PetscInt i = 0; i < n; i++) array[i] *= value; 43 PetscCall(VecRestoreArray(globals[0], &array)); 44 PetscCall(VecGetLocalSize(globals[1], &n)); 45 PetscCall(VecGetArray(globals[1], &array)); 46 for (PetscInt i = 0; i < n; i++) array[i] *= value; 47 PetscCall(VecRestoreArray(globals[1], &array)); 48 PetscCall(DMCompositeRestoreAccessArray(packer, global, 2, NULL, globals)); 49 50 /* Test GlobalToLocal in insert mode */ 51 PetscCall(DMGlobalToLocalBegin(packer, global, INSERT_VALUES, local)); 52 PetscCall(DMGlobalToLocalEnd(packer, global, INSERT_VALUES, local)); 53 54 PetscCall(PetscViewerASCIIPushSynchronized(PETSC_VIEWER_STDOUT_WORLD)); 55 PetscCall(PetscViewerGetSubViewer(PETSC_VIEWER_STDOUT_WORLD, PETSC_COMM_SELF, &viewer)); 56 PetscCall(PetscViewerASCIIPrintf(viewer, "\nLocal Vector: processor %d\n", rank)); 57 PetscCall(VecView(local, viewer)); 58 PetscCall(PetscViewerRestoreSubViewer(PETSC_VIEWER_STDOUT_WORLD, PETSC_COMM_SELF, &viewer)); 59 PetscCall(PetscViewerASCIIPopSynchronized(PETSC_VIEWER_STDOUT_WORLD)); 60 61 /* Test LocalToGlobal in insert mode */ 62 PetscCall(DMLocalToGlobalBegin(packer, local, INSERT_VALUES, global)); 63 PetscCall(DMLocalToGlobalEnd(packer, local, INSERT_VALUES, global)); 64 65 PetscCall(VecView(global, PETSC_VIEWER_STDOUT_WORLD)); 66 67 /* Test LocalToLocal in insert mode */ 68 PetscCall(DMLocalToLocalBegin(packer, local, INSERT_VALUES, buffer)); 69 PetscCall(DMLocalToLocalEnd(packer, local, INSERT_VALUES, buffer)); 70 71 PetscCall(PetscViewerASCIIPushSynchronized(PETSC_VIEWER_STDOUT_WORLD)); 72 PetscCall(PetscViewerGetSubViewer(PETSC_VIEWER_STDOUT_WORLD, PETSC_COMM_SELF, &viewer)); 73 PetscCall(PetscViewerASCIIPrintf(viewer, "\nLocal Vector: processor %d\n", rank)); 74 PetscCall(VecView(buffer, viewer)); 75 PetscCall(PetscViewerRestoreSubViewer(PETSC_VIEWER_STDOUT_WORLD, PETSC_COMM_SELF, &viewer)); 76 PetscCall(PetscViewerASCIIPopSynchronized(PETSC_VIEWER_STDOUT_WORLD)); 77 78 PetscCall(VecDestroy(&buffer)); 79 PetscCall(VecDestroy(&local)); 80 PetscCall(VecDestroy(&global)); 81 PetscCall(DMDestroy(&packer)); 82 PetscCall(DMDestroy(&da2)); 83 PetscCall(DMDestroy(&da1)); 84 85 PetscCall(PetscFinalize()); 86 return 0; 87 } 88 89 /*TEST 90 91 test: 92 nsize: 3 93 94 TEST*/ 95