1*c4762a1bSJed Brown 2*c4762a1bSJed Brown static char help[] = "Tests MatSetValuesBlockedStencil() in 3d.\n\n"; 3*c4762a1bSJed Brown 4*c4762a1bSJed Brown #include <petscmat.h> 5*c4762a1bSJed Brown #include <petscdm.h> 6*c4762a1bSJed Brown #include <petscdmda.h> 7*c4762a1bSJed Brown 8*c4762a1bSJed Brown int main(int argc,char **argv) 9*c4762a1bSJed Brown { 10*c4762a1bSJed Brown PetscInt M = 3,N = 4,P = 2,s = 1,w = 2,i, m = PETSC_DECIDE,n = PETSC_DECIDE,p = PETSC_DECIDE; 11*c4762a1bSJed Brown PetscErrorCode ierr; 12*c4762a1bSJed Brown DM da; 13*c4762a1bSJed Brown Mat mat; 14*c4762a1bSJed Brown DMDAStencilType stencil_type = DMDA_STENCIL_BOX; 15*c4762a1bSJed Brown PetscBool flg = PETSC_FALSE; 16*c4762a1bSJed Brown MatStencil idx[2],idy[2]; 17*c4762a1bSJed Brown PetscScalar *values; 18*c4762a1bSJed Brown 19*c4762a1bSJed Brown ierr = PetscInitialize(&argc,&argv,(char*)0,help);if (ierr) return ierr; 20*c4762a1bSJed Brown ierr = PetscOptionsGetInt(NULL,NULL,"-M",&M,NULL);CHKERRQ(ierr); 21*c4762a1bSJed Brown ierr = PetscOptionsGetInt(NULL,NULL,"-N",&N,NULL);CHKERRQ(ierr); 22*c4762a1bSJed Brown ierr = PetscOptionsGetInt(NULL,NULL,"-P",&P,NULL);CHKERRQ(ierr); 23*c4762a1bSJed Brown ierr = PetscOptionsGetInt(NULL,NULL,"-m",&m,NULL);CHKERRQ(ierr); 24*c4762a1bSJed Brown ierr = PetscOptionsGetInt(NULL,NULL,"-n",&n,NULL);CHKERRQ(ierr); 25*c4762a1bSJed Brown ierr = PetscOptionsGetInt(NULL,NULL,"-p",&p,NULL);CHKERRQ(ierr); 26*c4762a1bSJed Brown ierr = PetscOptionsGetInt(NULL,NULL,"-s",&s,NULL);CHKERRQ(ierr); 27*c4762a1bSJed Brown ierr = PetscOptionsGetInt(NULL,NULL,"-w",&w,NULL);CHKERRQ(ierr); 28*c4762a1bSJed Brown ierr = PetscOptionsGetBool(NULL,NULL,"-star",&flg,NULL);CHKERRQ(ierr); 29*c4762a1bSJed Brown if (flg) stencil_type = DMDA_STENCIL_STAR; 30*c4762a1bSJed Brown 31*c4762a1bSJed Brown /* Create distributed array and get vectors */ 32*c4762a1bSJed Brown ierr = DMDACreate3d(PETSC_COMM_WORLD,DM_BOUNDARY_NONE,DM_BOUNDARY_NONE,DM_BOUNDARY_NONE,stencil_type,M,N,P,m,n,p,w,s,0,0,0,&da);CHKERRQ(ierr); 33*c4762a1bSJed Brown ierr = DMSetFromOptions(da);CHKERRQ(ierr); 34*c4762a1bSJed Brown ierr = DMSetUp(da);CHKERRQ(ierr); 35*c4762a1bSJed Brown ierr = DMSetMatType(da,MATMPIBAIJ);CHKERRQ(ierr); 36*c4762a1bSJed Brown ierr = DMCreateMatrix(da,&mat);CHKERRQ(ierr); 37*c4762a1bSJed Brown 38*c4762a1bSJed Brown idx[0].i = 1; idx[0].j = 1; idx[0].k = 0; 39*c4762a1bSJed Brown idx[1].i = 2; idx[1].j = 1; idx[1].k = 0; 40*c4762a1bSJed Brown idy[0].i = 1; idy[0].j = 2; idy[0].k = 0; 41*c4762a1bSJed Brown idy[1].i = 2; idy[1].j = 2; idy[1].k = 0; 42*c4762a1bSJed Brown ierr = PetscMalloc1(2*2*w*w,&values);CHKERRQ(ierr); 43*c4762a1bSJed Brown for (i=0; i<2*2*w*w; i++) values[i] = i; 44*c4762a1bSJed Brown ierr = MatSetValuesBlockedStencil(mat,2,idx,2,idy,values,INSERT_VALUES);CHKERRQ(ierr); 45*c4762a1bSJed Brown ierr = MatAssemblyBegin(mat,MAT_FINAL_ASSEMBLY);CHKERRQ(ierr); 46*c4762a1bSJed Brown ierr = MatAssemblyEnd(mat,MAT_FINAL_ASSEMBLY);CHKERRQ(ierr); 47*c4762a1bSJed Brown 48*c4762a1bSJed Brown /* Free memory */ 49*c4762a1bSJed Brown ierr = PetscFree(values);CHKERRQ(ierr); 50*c4762a1bSJed Brown ierr = MatDestroy(&mat);CHKERRQ(ierr); 51*c4762a1bSJed Brown ierr = DMDestroy(&da);CHKERRQ(ierr); 52*c4762a1bSJed Brown ierr = PetscFinalize(); 53*c4762a1bSJed Brown return ierr; 54*c4762a1bSJed Brown } 55*c4762a1bSJed Brown 56*c4762a1bSJed Brown 57*c4762a1bSJed Brown 58*c4762a1bSJed Brown 59*c4762a1bSJed Brown 60*c4762a1bSJed Brown 61*c4762a1bSJed Brown 62*c4762a1bSJed Brown 63*c4762a1bSJed Brown 64*c4762a1bSJed Brown 65*c4762a1bSJed Brown 66*c4762a1bSJed Brown 67*c4762a1bSJed Brown 68*c4762a1bSJed Brown 69*c4762a1bSJed Brown 70*c4762a1bSJed Brown 71*c4762a1bSJed Brown 72*c4762a1bSJed Brown 73*c4762a1bSJed Brown 74*c4762a1bSJed Brown 75*c4762a1bSJed Brown 76