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 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 14*327415f7SBarry 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) { 19c4762a1bSJed Brown case 1: 209566063dSJacob Faibussowitsch PetscCall(DMDACreate1d(PETSC_COMM_WORLD,DM_BOUNDARY_NONE,nx,dof,1,NULL,&da)); 21c4762a1bSJed Brown break; 22c4762a1bSJed Brown case 2: 239566063dSJacob 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: 26d0609cedSBarry Smith PetscCall(DMDACreate3d(PETSC_COMM_WORLD,DM_BOUNDARY_NONE,DM_BOUNDARY_NONE,DM_BOUNDARY_NONE,DMDA_STENCIL_BOX,nx,ny,nz, 27d0609cedSBarry Smith PETSC_DECIDE,PETSC_DECIDE,PETSC_DECIDE,dof,2,NULL,NULL,NULL,&da)); 28c4762a1bSJed Brown } 29c4762a1bSJed Brown 309566063dSJacob Faibussowitsch PetscCall(DMSetFromOptions(da)); 319566063dSJacob Faibussowitsch PetscCall(DMSetUp(da)); 329566063dSJacob Faibussowitsch PetscCall(DMView(da,PETSC_VIEWER_STDOUT_WORLD)); 33c4762a1bSJed Brown 349566063dSJacob Faibussowitsch PetscCall(PetscOptionsGetBool(NULL,NULL,"-struct_only",&struct_only,NULL)); 359566063dSJacob Faibussowitsch PetscCall(DMSetMatrixStructureOnly(da,struct_only)); 369566063dSJacob 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 */ 389566063dSJacob Faibussowitsch PetscCall(DMSetMatrixStructureOnly(da,PETSC_FALSE)); 39c4762a1bSJed Brown 409566063dSJacob Faibussowitsch PetscCall(MatView(A,PETSC_VIEWER_STDOUT_WORLD)); 419566063dSJacob Faibussowitsch PetscCall(MatDestroy(&A)); 429566063dSJacob Faibussowitsch PetscCall(DMDestroy(&da)); 439566063dSJacob 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