1c4762a1bSJed Brown static char help[] = "Tests DMDAVecGetArrayDOF()\n"; 2c4762a1bSJed Brown #include <petscdm.h> 3c4762a1bSJed Brown #include <petscdmda.h> 4c4762a1bSJed Brown 5c4762a1bSJed Brown int main(int argc, char *argv[]) 6c4762a1bSJed Brown { 7c4762a1bSJed Brown DM da, daX, daY; 8c4762a1bSJed Brown DMDALocalInfo info; 9c4762a1bSJed Brown MPI_Comm commX, commY; 10c4762a1bSJed Brown Vec basisX, basisY; 11c4762a1bSJed Brown PetscScalar **arrayX, **arrayY; 12c4762a1bSJed Brown const PetscInt *lx, *ly; 13c4762a1bSJed Brown PetscInt M = 3, N = 3; 14c4762a1bSJed Brown PetscInt p = 1; 15c4762a1bSJed Brown PetscInt numGP = 3; 16c4762a1bSJed Brown PetscInt dof = 2*(p+1)*numGP; 17c4762a1bSJed Brown PetscMPIInt rank, subsize, subrank; 18c4762a1bSJed Brown PetscErrorCode ierr; 19c4762a1bSJed Brown 20c4762a1bSJed Brown ierr = PetscInitialize(&argc,&argv,0,help);if (ierr) return ierr; 21*5f80ce2aSJacob Faibussowitsch CHKERRMPI(MPI_Comm_rank(PETSC_COMM_WORLD, &rank)); 22c4762a1bSJed Brown /* Create 2D DMDA */ 23*5f80ce2aSJacob Faibussowitsch CHKERRQ(DMDACreate2d(PETSC_COMM_WORLD,DM_BOUNDARY_NONE,DM_BOUNDARY_NONE,DMDA_STENCIL_STAR,M,N,PETSC_DECIDE,PETSC_DECIDE,1,1,NULL,NULL,&da)); 24*5f80ce2aSJacob Faibussowitsch CHKERRQ(DMSetFromOptions(da)); 25*5f80ce2aSJacob Faibussowitsch CHKERRQ(DMSetUp(da)); 26c4762a1bSJed Brown /* Create 1D DMDAs along two directions. */ 27*5f80ce2aSJacob Faibussowitsch CHKERRQ(DMDAGetOwnershipRanges(da, &lx, &ly, NULL)); 28*5f80ce2aSJacob Faibussowitsch CHKERRQ(DMDAGetLocalInfo(da, &info)); 29c4762a1bSJed Brown /* Partitioning in the X direction makes a subcomm extending in the Y direction and vice-versa. */ 30*5f80ce2aSJacob Faibussowitsch CHKERRQ(DMDAGetProcessorSubsets(da, DM_X, &commY)); 31*5f80ce2aSJacob Faibussowitsch CHKERRQ(DMDAGetProcessorSubsets(da, DM_Y, &commX)); 32*5f80ce2aSJacob Faibussowitsch CHKERRMPI(MPI_Comm_size(commX, &subsize)); 33*5f80ce2aSJacob Faibussowitsch CHKERRMPI(MPI_Comm_rank(commX, &subrank)); 34*5f80ce2aSJacob Faibussowitsch CHKERRQ(PetscSynchronizedPrintf(PETSC_COMM_WORLD, "[%d]X subrank: %d subsize: %d\n", rank, subrank, subsize)); 35*5f80ce2aSJacob Faibussowitsch CHKERRQ(PetscSynchronizedFlush(PETSC_COMM_WORLD,PETSC_STDOUT)); 36*5f80ce2aSJacob Faibussowitsch CHKERRMPI(MPI_Comm_size(commY, &subsize)); 37*5f80ce2aSJacob Faibussowitsch CHKERRMPI(MPI_Comm_rank(commY, &subrank)); 38*5f80ce2aSJacob Faibussowitsch CHKERRQ(PetscSynchronizedPrintf(PETSC_COMM_WORLD, "[%d]Y subrank: %d subsize: %d\n", rank, subrank, subsize)); 39*5f80ce2aSJacob Faibussowitsch CHKERRQ(PetscSynchronizedFlush(PETSC_COMM_WORLD,PETSC_STDOUT)); 40*5f80ce2aSJacob Faibussowitsch CHKERRQ(DMDACreate1d(commX, DM_BOUNDARY_NONE, info.mx, dof, 1, lx, &daX)); 41*5f80ce2aSJacob Faibussowitsch CHKERRQ(DMSetUp(daX)); 42*5f80ce2aSJacob Faibussowitsch CHKERRQ(DMDACreate1d(commY, DM_BOUNDARY_NONE, info.my, dof, 1, ly, &daY)); 43*5f80ce2aSJacob Faibussowitsch CHKERRQ(DMSetUp(daY)); 44c4762a1bSJed Brown /* Create 1D vectors for basis functions */ 45*5f80ce2aSJacob Faibussowitsch CHKERRQ(DMGetGlobalVector(daX, &basisX)); 46*5f80ce2aSJacob Faibussowitsch CHKERRQ(DMGetGlobalVector(daY, &basisY)); 47c4762a1bSJed Brown /* Extract basis functions */ 48*5f80ce2aSJacob Faibussowitsch CHKERRQ(DMDAVecGetArrayDOF(daX, basisX, &arrayX)); 49*5f80ce2aSJacob Faibussowitsch CHKERRQ(DMDAVecGetArrayDOF(daY, basisY, &arrayY)); 50c4762a1bSJed Brown /*arrayX[i][ndof]; */ 51c4762a1bSJed Brown /*arrayY[j][ndof]; */ 52*5f80ce2aSJacob Faibussowitsch CHKERRQ(DMDAVecRestoreArrayDOF(daX, basisX, &arrayX)); 53*5f80ce2aSJacob Faibussowitsch CHKERRQ(DMDAVecRestoreArrayDOF(daY, basisY, &arrayY)); 54c4762a1bSJed Brown /* Return basis vectors */ 55*5f80ce2aSJacob Faibussowitsch CHKERRQ(DMRestoreGlobalVector(daX, &basisX)); 56*5f80ce2aSJacob Faibussowitsch CHKERRQ(DMRestoreGlobalVector(daY, &basisY)); 57c4762a1bSJed Brown /* Cleanup */ 58*5f80ce2aSJacob Faibussowitsch CHKERRMPI(MPI_Comm_free(&commX)); 59*5f80ce2aSJacob Faibussowitsch CHKERRMPI(MPI_Comm_free(&commY)); 60*5f80ce2aSJacob Faibussowitsch CHKERRQ(DMDestroy(&daX)); 61*5f80ce2aSJacob Faibussowitsch CHKERRQ(DMDestroy(&daY)); 62*5f80ce2aSJacob Faibussowitsch CHKERRQ(DMDestroy(&da)); 63c4762a1bSJed Brown ierr = PetscFinalize(); 64c4762a1bSJed Brown return ierr; 65c4762a1bSJed Brown } 66c4762a1bSJed Brown 67c4762a1bSJed Brown /*TEST 68c4762a1bSJed Brown 69c4762a1bSJed Brown test: 70c4762a1bSJed Brown nsize: 2 71c4762a1bSJed Brown 72c4762a1bSJed Brown TEST*/ 73