1c4762a1bSJed Brown /* 2c4762a1bSJed Brown Test DMCreateMatrix() for structure_only 3c4762a1bSJed Brown */ 4c4762a1bSJed Brown 5c4762a1bSJed Brown #include <petscdmda.h> 6c4762a1bSJed Brown 7c4762a1bSJed Brown int main(int argc, char *argv[]) 8c4762a1bSJed Brown { 9c4762a1bSJed Brown PetscErrorCode ierr; 10c4762a1bSJed Brown PetscInt nx=6,ny=6,nz=6,dim=1,dof=2; 11c4762a1bSJed Brown DM da; 12c4762a1bSJed Brown Mat A; 13c4762a1bSJed Brown PetscBool struct_only=PETSC_TRUE; 14c4762a1bSJed Brown 15*9566063dSJacob Faibussowitsch PetscCall(PetscInitialize(&argc,&argv,NULL,NULL)); 16*9566063dSJacob Faibussowitsch PetscCall(PetscOptionsGetInt(NULL,NULL,"-dim",&dim,NULL)); 17*9566063dSJacob Faibussowitsch PetscCall(PetscOptionsGetInt(NULL,NULL,"-dof",&dof,NULL)); 18c4762a1bSJed Brown switch (dim) { 19c4762a1bSJed Brown case 1: 20*9566063dSJacob Faibussowitsch PetscCall(DMDACreate1d(PETSC_COMM_WORLD,DM_BOUNDARY_NONE,nx,dof,1,NULL,&da)); 21c4762a1bSJed Brown break; 22c4762a1bSJed Brown case 2: 23*9566063dSJacob 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)); 24c4762a1bSJed Brown break; 25c4762a1bSJed Brown default: 26c4762a1bSJed Brown ierr = DMDACreate3d(PETSC_COMM_WORLD,DM_BOUNDARY_NONE,DM_BOUNDARY_NONE,DM_BOUNDARY_NONE,DMDA_STENCIL_BOX,nx,ny,nz, 27*9566063dSJacob Faibussowitsch PETSC_DECIDE,PETSC_DECIDE,PETSC_DECIDE,dof,2,NULL,NULL,NULL,&da);PetscCall(ierr); 28c4762a1bSJed Brown } 29c4762a1bSJed Brown 30*9566063dSJacob Faibussowitsch PetscCall(DMSetFromOptions(da)); 31*9566063dSJacob Faibussowitsch PetscCall(DMSetUp(da)); 32*9566063dSJacob Faibussowitsch PetscCall(DMView(da,PETSC_VIEWER_STDOUT_WORLD)); 33c4762a1bSJed Brown 34*9566063dSJacob Faibussowitsch PetscCall(PetscOptionsGetBool(NULL,NULL,"-struct_only",&struct_only,NULL)); 35*9566063dSJacob Faibussowitsch PetscCall(DMSetMatrixStructureOnly(da,struct_only)); 36*9566063dSJacob Faibussowitsch PetscCall(DMCreateMatrix(da,&A)); 37c4762a1bSJed Brown /* Set da->structure_only to default PETSC_FALSE in case da is being used to create new matrices */ 38*9566063dSJacob Faibussowitsch PetscCall(DMSetMatrixStructureOnly(da,PETSC_FALSE)); 39c4762a1bSJed Brown 40*9566063dSJacob Faibussowitsch PetscCall(MatView(A,PETSC_VIEWER_STDOUT_WORLD)); 41*9566063dSJacob Faibussowitsch PetscCall(MatDestroy(&A)); 42*9566063dSJacob Faibussowitsch PetscCall(DMDestroy(&da)); 43*9566063dSJacob Faibussowitsch PetscCall(PetscFinalize()); 44b122ec5aSJacob Faibussowitsch return 0; 45c4762a1bSJed Brown } 46c4762a1bSJed Brown 47c4762a1bSJed Brown /*TEST 48c4762a1bSJed Brown 49c4762a1bSJed Brown test: 50c4762a1bSJed Brown 51c4762a1bSJed Brown test: 52c4762a1bSJed Brown suffix: 2 53c4762a1bSJed Brown args: -dm_mat_type baij -dim 2 54c4762a1bSJed Brown 55c4762a1bSJed Brown TEST*/ 56