1c4762a1bSJed Brown /* 2c4762a1bSJed Brown Test DMCreateMatrix() for structure_only 3c4762a1bSJed Brown */ 4c4762a1bSJed Brown 5c4762a1bSJed Brown #include <petscdmda.h> 6c4762a1bSJed Brown 7*d71ae5a4SJacob Faibussowitsch int main(int argc, char *argv[]) 8*d71ae5a4SJacob Faibussowitsch { 9c4762a1bSJed Brown PetscInt nx = 6, ny = 6, nz = 6, dim = 1, dof = 2; 10c4762a1bSJed Brown DM da; 11c4762a1bSJed Brown Mat A; 12c4762a1bSJed Brown PetscBool struct_only = PETSC_TRUE; 13c4762a1bSJed Brown 14327415f7SBarry Smith PetscFunctionBeginUser; 159566063dSJacob Faibussowitsch PetscCall(PetscInitialize(&argc, &argv, NULL, NULL)); 169566063dSJacob Faibussowitsch PetscCall(PetscOptionsGetInt(NULL, NULL, "-dim", &dim, NULL)); 179566063dSJacob Faibussowitsch PetscCall(PetscOptionsGetInt(NULL, NULL, "-dof", &dof, NULL)); 18c4762a1bSJed Brown switch (dim) { 19*d71ae5a4SJacob Faibussowitsch case 1: 20*d71ae5a4SJacob Faibussowitsch PetscCall(DMDACreate1d(PETSC_COMM_WORLD, DM_BOUNDARY_NONE, nx, dof, 1, NULL, &da)); 21*d71ae5a4SJacob Faibussowitsch break; 22*d71ae5a4SJacob Faibussowitsch case 2: 23*d71ae5a4SJacob Faibussowitsch PetscCall(DMDACreate2d(PETSC_COMM_WORLD, DM_BOUNDARY_NONE, DM_BOUNDARY_NONE, DMDA_STENCIL_STAR, nx, ny, PETSC_DECIDE, PETSC_DECIDE, dof, 1, NULL, NULL, &da)); 24*d71ae5a4SJacob Faibussowitsch break; 25*d71ae5a4SJacob Faibussowitsch default: 26*d71ae5a4SJacob Faibussowitsch PetscCall(DMDACreate3d(PETSC_COMM_WORLD, DM_BOUNDARY_NONE, DM_BOUNDARY_NONE, DM_BOUNDARY_NONE, DMDA_STENCIL_BOX, nx, ny, nz, PETSC_DECIDE, PETSC_DECIDE, PETSC_DECIDE, dof, 2, NULL, NULL, NULL, &da)); 27c4762a1bSJed Brown } 28c4762a1bSJed Brown 299566063dSJacob Faibussowitsch PetscCall(DMSetFromOptions(da)); 309566063dSJacob Faibussowitsch PetscCall(DMSetUp(da)); 319566063dSJacob Faibussowitsch PetscCall(DMView(da, PETSC_VIEWER_STDOUT_WORLD)); 32c4762a1bSJed Brown 339566063dSJacob Faibussowitsch PetscCall(PetscOptionsGetBool(NULL, NULL, "-struct_only", &struct_only, NULL)); 349566063dSJacob Faibussowitsch PetscCall(DMSetMatrixStructureOnly(da, struct_only)); 359566063dSJacob Faibussowitsch PetscCall(DMCreateMatrix(da, &A)); 36c4762a1bSJed Brown /* Set da->structure_only to default PETSC_FALSE in case da is being used to create new matrices */ 379566063dSJacob Faibussowitsch PetscCall(DMSetMatrixStructureOnly(da, PETSC_FALSE)); 38c4762a1bSJed Brown 399566063dSJacob Faibussowitsch PetscCall(MatView(A, PETSC_VIEWER_STDOUT_WORLD)); 409566063dSJacob Faibussowitsch PetscCall(MatDestroy(&A)); 419566063dSJacob Faibussowitsch PetscCall(DMDestroy(&da)); 429566063dSJacob Faibussowitsch PetscCall(PetscFinalize()); 43b122ec5aSJacob Faibussowitsch return 0; 44c4762a1bSJed Brown } 45c4762a1bSJed Brown 46c4762a1bSJed Brown /*TEST 47c4762a1bSJed Brown 48c4762a1bSJed Brown test: 49c4762a1bSJed Brown 50c4762a1bSJed Brown test: 51c4762a1bSJed Brown suffix: 2 52c4762a1bSJed Brown args: -dm_mat_type baij -dim 2 53c4762a1bSJed Brown 54c4762a1bSJed Brown TEST*/ 55