1c4762a1bSJed Brown 2c4762a1bSJed Brown static char help[] = "Creates a matrix using 9 pt stencil, and uses it to test MatIncreaseOverlap (needed for additive Schwarz preconditioner). \n\ 3c4762a1bSJed Brown -m <size> : problem size\n\ 4c4762a1bSJed Brown -x1, -x2 <size> : no of subdomains in x and y directions\n\n"; 5c4762a1bSJed Brown 6c4762a1bSJed Brown #include <petscksp.h> 7c4762a1bSJed Brown 8c4762a1bSJed Brown PetscErrorCode FormElementStiffness(PetscReal H,PetscScalar *Ke) 9c4762a1bSJed Brown { 10c4762a1bSJed Brown Ke[0] = H/6.0; Ke[1] = -.125*H; Ke[2] = H/12.0; Ke[3] = -.125*H; 11c4762a1bSJed Brown Ke[4] = -.125*H; Ke[5] = H/6.0; Ke[6] = -.125*H; Ke[7] = H/12.0; 12c4762a1bSJed Brown Ke[8] = H/12.0; Ke[9] = -.125*H; Ke[10] = H/6.0; Ke[11] = -.125*H; 13c4762a1bSJed Brown Ke[12] = -.125*H; Ke[13] = H/12.0; Ke[14] = -.125*H; Ke[15] = H/6.0; 14c4762a1bSJed Brown return 0; 15c4762a1bSJed Brown } 16c4762a1bSJed Brown PetscErrorCode FormElementRhs(PetscReal x,PetscReal y,PetscReal H,PetscScalar *r) 17c4762a1bSJed Brown { 18c4762a1bSJed Brown r[0] = 0.; r[1] = 0.; r[2] = 0.; r[3] = 0.0; 19c4762a1bSJed Brown return 0; 20c4762a1bSJed Brown } 21c4762a1bSJed Brown 22c4762a1bSJed Brown int main(int argc,char **args) 23c4762a1bSJed Brown { 24c4762a1bSJed Brown Mat C; 25c4762a1bSJed Brown PetscInt i,m = 2,N,M,idx[4],Nsub1,Nsub2,ol=1,x1,x2; 26c4762a1bSJed Brown PetscScalar Ke[16]; 27c4762a1bSJed Brown PetscReal h; 28c4762a1bSJed Brown IS *is1,*is2,*islocal1,*islocal2; 29c4762a1bSJed Brown PetscBool flg; 30c4762a1bSJed Brown 319566063dSJacob Faibussowitsch PetscCall(PetscInitialize(&argc,&args,(char*)0,help)); 329566063dSJacob Faibussowitsch PetscCall(PetscOptionsGetInt(NULL,NULL,"-m",&m,NULL)); 33c4762a1bSJed Brown N = (m+1)*(m+1); /* dimension of matrix */ 34c4762a1bSJed Brown M = m*m; /* number of elements */ 35c4762a1bSJed Brown h = 1.0/m; /* mesh width */ 36c4762a1bSJed Brown x1 = (m+1)/2; 37c4762a1bSJed Brown x2 = x1; 389566063dSJacob Faibussowitsch PetscCall(PetscOptionsGetInt(NULL,NULL,"-x1",&x1,NULL)); 399566063dSJacob Faibussowitsch PetscCall(PetscOptionsGetInt(NULL,NULL,"-x2",&x2,NULL)); 40c4762a1bSJed Brown /* create stiffness matrix */ 419566063dSJacob Faibussowitsch PetscCall(MatCreateSeqAIJ(PETSC_COMM_SELF,N,N,9,NULL,&C)); 42c4762a1bSJed Brown 43c4762a1bSJed Brown /* forms the element stiffness for the Laplacian */ 449566063dSJacob Faibussowitsch PetscCall(FormElementStiffness(h*h,Ke)); 45c4762a1bSJed Brown for (i=0; i<M; i++) { 46c4762a1bSJed Brown /* node numbers for the four corners of element */ 47c4762a1bSJed Brown idx[0] = (m+1)*(i/m) + (i % m); 48c4762a1bSJed Brown idx[1] = idx[0]+1; idx[2] = idx[1] + m + 1; idx[3] = idx[2] - 1; 499566063dSJacob Faibussowitsch PetscCall(MatSetValues(C,4,idx,4,idx,Ke,ADD_VALUES)); 50c4762a1bSJed Brown } 519566063dSJacob Faibussowitsch PetscCall(MatAssemblyBegin(C,MAT_FINAL_ASSEMBLY)); 529566063dSJacob Faibussowitsch PetscCall(MatAssemblyEnd(C,MAT_FINAL_ASSEMBLY)); 53c4762a1bSJed Brown 54c4762a1bSJed Brown for (ol=0; ol<m+2; ++ol) { 55c4762a1bSJed Brown 569566063dSJacob Faibussowitsch PetscCall(PCASMCreateSubdomains2D(m+1,m+1,x1,x2,1,0,&Nsub1,&is1,&islocal1)); 579566063dSJacob Faibussowitsch PetscCall(MatIncreaseOverlap(C,Nsub1,is1,ol)); 589566063dSJacob Faibussowitsch PetscCall(PCASMCreateSubdomains2D(m+1,m+1,x1,x2,1,ol,&Nsub2,&is2,&islocal2)); 59c4762a1bSJed Brown 609566063dSJacob Faibussowitsch PetscCall(PetscPrintf(PETSC_COMM_SELF,"flg == 1 => both index sets are same\n")); 61c4762a1bSJed Brown if (Nsub1 != Nsub2) { 629566063dSJacob Faibussowitsch PetscCall(PetscPrintf(PETSC_COMM_SELF,"Error: No of indes sets don't match\n")); 63c4762a1bSJed Brown } 64c4762a1bSJed Brown 65c4762a1bSJed Brown for (i=0; i<Nsub1; ++i) { 669566063dSJacob Faibussowitsch PetscCall(ISEqual(is1[i],is2[i],&flg)); 67*63a3b9bcSJacob Faibussowitsch PetscCall(PetscPrintf(PETSC_COMM_SELF,"i = %" PetscInt_FMT ",flg = %d \n",i,(int)flg)); 68c4762a1bSJed Brown 69c4762a1bSJed Brown } 709566063dSJacob Faibussowitsch for (i=0; i<Nsub1; ++i) PetscCall(ISDestroy(&is1[i])); 719566063dSJacob Faibussowitsch for (i=0; i<Nsub2; ++i) PetscCall(ISDestroy(&is2[i])); 729566063dSJacob Faibussowitsch for (i=0; i<Nsub1; ++i) PetscCall(ISDestroy(&islocal1[i])); 739566063dSJacob Faibussowitsch for (i=0; i<Nsub2; ++i) PetscCall(ISDestroy(&islocal2[i])); 74c4762a1bSJed Brown 759566063dSJacob Faibussowitsch PetscCall(PetscFree(is1)); 769566063dSJacob Faibussowitsch PetscCall(PetscFree(is2)); 779566063dSJacob Faibussowitsch PetscCall(PetscFree(islocal1)); 789566063dSJacob Faibussowitsch PetscCall(PetscFree(islocal2)); 79c4762a1bSJed Brown } 809566063dSJacob Faibussowitsch PetscCall(MatDestroy(&C)); 819566063dSJacob Faibussowitsch PetscCall(PetscFinalize()); 82b122ec5aSJacob Faibussowitsch return 0; 83c4762a1bSJed Brown } 84c4762a1bSJed Brown 85c4762a1bSJed Brown /*TEST 86c4762a1bSJed Brown 87c4762a1bSJed Brown test: 88c4762a1bSJed Brown args: -m 7 89c4762a1bSJed Brown 90c4762a1bSJed Brown TEST*/ 91