1c4762a1bSJed Brown 2c4762a1bSJed Brown static char help[] = "MatLoad test for loading matrices that are created by DMCreateMatrix() and\n\ 3c4762a1bSJed Brown stored in binary via MatView_MPI_DA.MatView_MPI_DA stores the matrix\n\ 4c4762a1bSJed Brown in natural ordering. Hence MatLoad() has to read the matrix first in\n\ 5c4762a1bSJed Brown natural ordering and then permute it back to the application ordering.This\n\ 6c4762a1bSJed Brown example is used for testing the subroutine MatLoad_MPI_DA\n\n"; 7c4762a1bSJed Brown 8c4762a1bSJed Brown #include <petscdm.h> 9c4762a1bSJed Brown #include <petscdmda.h> 10c4762a1bSJed Brown 11*d71ae5a4SJacob Faibussowitsch int main(int argc, char **argv) 12*d71ae5a4SJacob Faibussowitsch { 13c4762a1bSJed Brown PetscInt X = 10, Y = 8, Z = 8; 14c4762a1bSJed Brown DM da; 15c4762a1bSJed Brown PetscViewer viewer; 16c4762a1bSJed Brown Mat A; 17c4762a1bSJed Brown 18327415f7SBarry Smith PetscFunctionBeginUser; 199566063dSJacob Faibussowitsch PetscCall(PetscInitialize(&argc, &argv, (char *)0, help)); 209566063dSJacob Faibussowitsch PetscCall(PetscViewerBinaryOpen(PETSC_COMM_WORLD, "temp.dat", FILE_MODE_WRITE, &viewer)); 21c4762a1bSJed Brown 22c4762a1bSJed Brown /* Read options */ 239566063dSJacob Faibussowitsch PetscCall(PetscOptionsGetInt(NULL, NULL, "-X", &X, NULL)); 249566063dSJacob Faibussowitsch PetscCall(PetscOptionsGetInt(NULL, NULL, "-Y", &Y, NULL)); 259566063dSJacob Faibussowitsch PetscCall(PetscOptionsGetInt(NULL, NULL, "-Z", &Z, NULL)); 26c4762a1bSJed Brown 27c4762a1bSJed Brown /* Create distributed array and get vectors */ 289566063dSJacob Faibussowitsch PetscCall(DMDACreate3d(PETSC_COMM_WORLD, DM_BOUNDARY_NONE, DM_BOUNDARY_NONE, DM_BOUNDARY_NONE, DMDA_STENCIL_STAR, X, Y, Z, PETSC_DECIDE, PETSC_DECIDE, PETSC_DECIDE, 1, 1, NULL, NULL, NULL, &da)); 299566063dSJacob Faibussowitsch PetscCall(DMSetFromOptions(da)); 309566063dSJacob Faibussowitsch PetscCall(DMSetUp(da)); 319566063dSJacob Faibussowitsch PetscCall(DMSetMatType(da, MATMPIAIJ)); 329566063dSJacob Faibussowitsch PetscCall(DMCreateMatrix(da, &A)); 339566063dSJacob Faibussowitsch PetscCall(MatShift(A, X)); 349566063dSJacob Faibussowitsch PetscCall(MatView(A, viewer)); 359566063dSJacob Faibussowitsch PetscCall(MatDestroy(&A)); 369566063dSJacob Faibussowitsch PetscCall(PetscViewerDestroy(&viewer)); 37c4762a1bSJed Brown 389566063dSJacob Faibussowitsch PetscCall(PetscViewerBinaryOpen(PETSC_COMM_WORLD, "temp.dat", FILE_MODE_READ, &viewer)); 399566063dSJacob Faibussowitsch PetscCall(DMCreateMatrix(da, &A)); 409566063dSJacob Faibussowitsch PetscCall(MatLoad(A, viewer)); 41c4762a1bSJed Brown 42c4762a1bSJed Brown /* Free memory */ 439566063dSJacob Faibussowitsch PetscCall(MatDestroy(&A)); 449566063dSJacob Faibussowitsch PetscCall(PetscViewerDestroy(&viewer)); 459566063dSJacob Faibussowitsch PetscCall(DMDestroy(&da)); 469566063dSJacob Faibussowitsch PetscCall(PetscFinalize()); 47b122ec5aSJacob Faibussowitsch return 0; 48c4762a1bSJed Brown } 49