123ce1328SBarry Smith 2aaa7dc30SBarry Smith #include <../src/ksp/pc/impls/is/pcis.h> /*I "petscpc.h" I*/ 3831a100dSStefano Zampini 4b965d45eSStefano Zampini static PetscErrorCode PCISSetUseStiffnessScaling_IS(PC pc, PetscBool use) 5b965d45eSStefano Zampini { 6b965d45eSStefano Zampini PC_IS *pcis = (PC_IS*)pc->data; 7b965d45eSStefano Zampini 8b965d45eSStefano Zampini PetscFunctionBegin; 9b965d45eSStefano Zampini pcis->use_stiffness_scaling = use; 10b965d45eSStefano Zampini PetscFunctionReturn(0); 11b965d45eSStefano Zampini } 12b965d45eSStefano Zampini 13b965d45eSStefano Zampini /*@ 14b965d45eSStefano Zampini PCISSetUseStiffnessScaling - Tells PCIS to construct partition of unity using 15b965d45eSStefano Zampini local matrices' diagonal. 16b965d45eSStefano Zampini 17b965d45eSStefano Zampini Not collective 18b965d45eSStefano Zampini 19b965d45eSStefano Zampini Input Parameters: 20b965d45eSStefano Zampini + pc - the preconditioning context 21b965d45eSStefano Zampini - use - whether or not pcis use matrix diagonal to build partition of unity. 22b965d45eSStefano Zampini 23b965d45eSStefano Zampini Level: intermediate 24b965d45eSStefano Zampini 25b965d45eSStefano Zampini Notes: 26b965d45eSStefano Zampini 27b965d45eSStefano Zampini .seealso: PCBDDC 28b965d45eSStefano Zampini @*/ 29b965d45eSStefano Zampini PetscErrorCode PCISSetUseStiffnessScaling(PC pc, PetscBool use) 30b965d45eSStefano Zampini { 31b965d45eSStefano Zampini PetscErrorCode ierr; 32b965d45eSStefano Zampini 33b965d45eSStefano Zampini PetscFunctionBegin; 34b965d45eSStefano Zampini PetscValidHeaderSpecific(pc,PC_CLASSID,1); 35b965d45eSStefano Zampini ierr = PetscTryMethod(pc,"PCISSetUseStiffnessScaling_C",(PC,PetscBool),(pc,use));CHKERRQ(ierr); 36b965d45eSStefano Zampini PetscFunctionReturn(0); 37b965d45eSStefano Zampini } 38b965d45eSStefano Zampini 39ba1573a8SStefano Zampini static PetscErrorCode PCISSetSubdomainDiagonalScaling_IS(PC pc, Vec scaling_factors) 40ba1573a8SStefano Zampini { 41ba1573a8SStefano Zampini PetscErrorCode ierr; 42ba1573a8SStefano Zampini PC_IS *pcis = (PC_IS*)pc->data; 43ba1573a8SStefano Zampini 44ba1573a8SStefano Zampini PetscFunctionBegin; 45ba1573a8SStefano Zampini ierr = PetscObjectReference((PetscObject)scaling_factors);CHKERRQ(ierr); 46bf83c2afSStefano Zampini ierr = VecDestroy(&pcis->D);CHKERRQ(ierr); 47ba1573a8SStefano Zampini pcis->D = scaling_factors; 48ba1573a8SStefano Zampini PetscFunctionReturn(0); 49ba1573a8SStefano Zampini } 50ba1573a8SStefano Zampini 51ba1573a8SStefano Zampini /*@ 52ba1573a8SStefano Zampini PCISSetSubdomainDiagonalScaling - Set diagonal scaling for PCIS. 53ba1573a8SStefano Zampini 54ba1573a8SStefano Zampini Not collective 55ba1573a8SStefano Zampini 56ba1573a8SStefano Zampini Input Parameters: 57ba1573a8SStefano Zampini + pc - the preconditioning context 58ba1573a8SStefano Zampini - scaling_factors - scaling factors for the subdomain 59ba1573a8SStefano Zampini 60ba1573a8SStefano Zampini Level: intermediate 61ba1573a8SStefano Zampini 62ba1573a8SStefano Zampini Notes: 63ba1573a8SStefano Zampini Intended to use with jumping coefficients cases. 64ba1573a8SStefano Zampini 65ba1573a8SStefano Zampini .seealso: PCBDDC 66ba1573a8SStefano Zampini @*/ 67ba1573a8SStefano Zampini PetscErrorCode PCISSetSubdomainDiagonalScaling(PC pc, Vec scaling_factors) 68ba1573a8SStefano Zampini { 69ba1573a8SStefano Zampini PetscErrorCode ierr; 70ba1573a8SStefano Zampini 71ba1573a8SStefano Zampini PetscFunctionBegin; 72ba1573a8SStefano Zampini PetscValidHeaderSpecific(pc,PC_CLASSID,1); 73ba1573a8SStefano Zampini ierr = PetscTryMethod(pc,"PCISSetSubdomainDiagonalScaling_C",(PC,Vec),(pc,scaling_factors));CHKERRQ(ierr); 74ba1573a8SStefano Zampini PetscFunctionReturn(0); 75ba1573a8SStefano Zampini } 76ba1573a8SStefano Zampini 77831a100dSStefano Zampini static PetscErrorCode PCISSetSubdomainScalingFactor_IS(PC pc, PetscScalar scal) 78831a100dSStefano Zampini { 79831a100dSStefano Zampini PC_IS *pcis = (PC_IS*)pc->data; 80831a100dSStefano Zampini 81831a100dSStefano Zampini PetscFunctionBegin; 82831a100dSStefano Zampini pcis->scaling_factor = scal; 83831a100dSStefano Zampini PetscFunctionReturn(0); 84831a100dSStefano Zampini } 85831a100dSStefano Zampini 86831a100dSStefano Zampini /*@ 87831a100dSStefano Zampini PCISSetSubdomainScalingFactor - Set scaling factor for PCIS. 88831a100dSStefano Zampini 89831a100dSStefano Zampini Not collective 90831a100dSStefano Zampini 91831a100dSStefano Zampini Input Parameters: 92831a100dSStefano Zampini + pc - the preconditioning context 93831a100dSStefano Zampini - scal - scaling factor for the subdomain 94831a100dSStefano Zampini 95831a100dSStefano Zampini Level: intermediate 96831a100dSStefano Zampini 97831a100dSStefano Zampini Notes: 98831a100dSStefano Zampini Intended to use with jumping coefficients cases. 99831a100dSStefano Zampini 100831a100dSStefano Zampini .seealso: PCBDDC 101831a100dSStefano Zampini @*/ 102831a100dSStefano Zampini PetscErrorCode PCISSetSubdomainScalingFactor(PC pc, PetscScalar scal) 103831a100dSStefano Zampini { 104831a100dSStefano Zampini PetscErrorCode ierr; 105831a100dSStefano Zampini 106831a100dSStefano Zampini PetscFunctionBegin; 107831a100dSStefano Zampini PetscValidHeaderSpecific(pc,PC_CLASSID,1); 108831a100dSStefano Zampini ierr = PetscTryMethod(pc,"PCISSetSubdomainScalingFactor_C",(PC,PetscScalar),(pc,scal));CHKERRQ(ierr); 109831a100dSStefano Zampini PetscFunctionReturn(0); 110831a100dSStefano Zampini } 111831a100dSStefano Zampini 112b4319ba4SBarry Smith 113b4319ba4SBarry Smith /* -------------------------------------------------------------------------- */ 114b4319ba4SBarry Smith /* 115b4319ba4SBarry Smith PCISSetUp - 116b4319ba4SBarry Smith */ 1175e8657edSStefano Zampini PetscErrorCode PCISSetUp(PC pc, PetscBool computesolvers) 118b4319ba4SBarry Smith { 119b4319ba4SBarry Smith PC_IS *pcis = (PC_IS*)(pc->data); 120bf327c11SStefano Zampini Mat_IS *matis; 1215e8657edSStefano Zampini MatReuse reuse; 1226849ba73SBarry Smith PetscErrorCode ierr; 12385c21eb1SStefano Zampini PetscBool flg,issbaij; 124831a100dSStefano Zampini Vec counter; 125b4319ba4SBarry Smith 126b4319ba4SBarry Smith PetscFunctionBegin; 12785c21eb1SStefano Zampini ierr = PetscObjectTypeCompare((PetscObject)pc->pmat,MATIS,&flg);CHKERRQ(ierr); 128ce94432eSBarry Smith if (!flg) SETERRQ(PetscObjectComm((PetscObject)pc),PETSC_ERR_ARG_WRONG,"Preconditioner type of Neumann Neumman requires matrix of type MATIS"); 12985c21eb1SStefano Zampini matis = (Mat_IS*)pc->pmat->data; 130b4319ba4SBarry Smith 1315e8657edSStefano Zampini /* first time creation, get info on substructuring */ 1325e8657edSStefano Zampini if (!pc->setupcalled) { 1335e8657edSStefano Zampini PetscInt n_I; 1345e8657edSStefano Zampini PetscInt *idx_I_local,*idx_B_local,*idx_I_global,*idx_B_global; 1356f516dd7SStefano Zampini PetscBT bt; 1365e8657edSStefano Zampini PetscInt i,j; 137b4319ba4SBarry Smith 138892d8026SStefano Zampini /* get info on mapping */ 1393bbff08aSStefano Zampini ierr = PetscObjectReference((PetscObject)pc->pmat->rmap->mapping);CHKERRQ(ierr); 1408d12c799SStefano Zampini ierr = ISLocalToGlobalMappingDestroy(&pcis->mapping);CHKERRQ(ierr); 1413bbff08aSStefano Zampini pcis->mapping = pc->pmat->rmap->mapping; 1428d12c799SStefano Zampini ierr = ISLocalToGlobalMappingGetSize(pcis->mapping,&pcis->n);CHKERRQ(ierr); 1438d12c799SStefano Zampini ierr = ISLocalToGlobalMappingGetInfo(pcis->mapping,&(pcis->n_neigh),&(pcis->neigh),&(pcis->n_shared),&(pcis->shared));CHKERRQ(ierr); 144892d8026SStefano Zampini 145b4319ba4SBarry Smith /* Identifying interior and interface nodes, in local numbering */ 1466f516dd7SStefano Zampini ierr = PetscBTCreate(pcis->n,&bt);CHKERRQ(ierr); 1474843afd6SStefano Zampini for (i=0;i<pcis->n_neigh;i++) 1486f516dd7SStefano Zampini for (j=0;j<pcis->n_shared[i];j++) { 1496f516dd7SStefano Zampini ierr = PetscBTSet(bt,pcis->shared[i][j]);CHKERRQ(ierr); 1506f516dd7SStefano Zampini } 151892d8026SStefano Zampini 1525e8657edSStefano Zampini /* Creating local and global index sets for interior and inteface nodes. */ 153785e854fSJed Brown ierr = PetscMalloc1(pcis->n,&idx_I_local);CHKERRQ(ierr); 154785e854fSJed Brown ierr = PetscMalloc1(pcis->n,&idx_B_local);CHKERRQ(ierr); 155b4319ba4SBarry Smith for (i=0, pcis->n_B=0, n_I=0; i<pcis->n; i++) { 1566f516dd7SStefano Zampini if (!PetscBTLookup(bt,i)) { 1572fa5cd67SKarl Rupp idx_I_local[n_I] = i; 1582fa5cd67SKarl Rupp n_I++; 1592fa5cd67SKarl Rupp } else { 1602fa5cd67SKarl Rupp idx_B_local[pcis->n_B] = i; 1612fa5cd67SKarl Rupp pcis->n_B++; 1622fa5cd67SKarl Rupp } 163b4319ba4SBarry Smith } 1646f516dd7SStefano Zampini 165b4319ba4SBarry Smith /* Getting the global numbering */ 166b4319ba4SBarry Smith idx_B_global = idx_I_local + n_I; /* Just avoiding allocating extra memory, since we have vacant space */ 167b4319ba4SBarry Smith idx_I_global = idx_B_local + pcis->n_B; 1688d12c799SStefano Zampini ierr = ISLocalToGlobalMappingApply(pcis->mapping,pcis->n_B,idx_B_local,idx_B_global);CHKERRQ(ierr); 1698d12c799SStefano Zampini ierr = ISLocalToGlobalMappingApply(pcis->mapping,n_I,idx_I_local,idx_I_global);CHKERRQ(ierr); 1702fa5cd67SKarl Rupp 1715e8657edSStefano Zampini /* Creating the index sets */ 172bb943df9SStefano Zampini ierr = ISCreateGeneral(PETSC_COMM_SELF,pcis->n_B,idx_B_local,PETSC_COPY_VALUES, &pcis->is_B_local);CHKERRQ(ierr); 173bb943df9SStefano Zampini ierr = ISCreateGeneral(PETSC_COMM_SELF,pcis->n_B,idx_B_global,PETSC_COPY_VALUES,&pcis->is_B_global);CHKERRQ(ierr); 174bb943df9SStefano Zampini ierr = ISCreateGeneral(PETSC_COMM_SELF,n_I,idx_I_local,PETSC_COPY_VALUES, &pcis->is_I_local);CHKERRQ(ierr); 175bb943df9SStefano Zampini ierr = ISCreateGeneral(PETSC_COMM_SELF,n_I,idx_I_global,PETSC_COPY_VALUES,&pcis->is_I_global);CHKERRQ(ierr); 1762fa5cd67SKarl Rupp 1775e8657edSStefano Zampini /* Freeing memory */ 178b4319ba4SBarry Smith ierr = PetscFree(idx_B_local);CHKERRQ(ierr); 179b4319ba4SBarry Smith ierr = PetscFree(idx_I_local);CHKERRQ(ierr); 1806f516dd7SStefano Zampini ierr = PetscBTDestroy(&bt);CHKERRQ(ierr); 181b4319ba4SBarry Smith 1825e8657edSStefano Zampini /* Creating work vectors and arrays */ 183892d8026SStefano Zampini ierr = VecDuplicate(matis->x,&pcis->vec1_N);CHKERRQ(ierr); 184b4319ba4SBarry Smith ierr = VecDuplicate(pcis->vec1_N,&pcis->vec2_N);CHKERRQ(ierr); 185b4319ba4SBarry Smith ierr = VecCreateSeq(PETSC_COMM_SELF,pcis->n-pcis->n_B,&pcis->vec1_D);CHKERRQ(ierr); 186b4319ba4SBarry Smith ierr = VecDuplicate(pcis->vec1_D,&pcis->vec2_D);CHKERRQ(ierr); 187b4319ba4SBarry Smith ierr = VecDuplicate(pcis->vec1_D,&pcis->vec3_D);CHKERRQ(ierr); 188df187020SStefano Zampini ierr = VecDuplicate(pcis->vec1_D,&pcis->vec4_D);CHKERRQ(ierr); 189b4319ba4SBarry Smith ierr = VecCreateSeq(PETSC_COMM_SELF,pcis->n_B,&pcis->vec1_B);CHKERRQ(ierr); 190b4319ba4SBarry Smith ierr = VecDuplicate(pcis->vec1_B,&pcis->vec2_B);CHKERRQ(ierr); 191b4319ba4SBarry Smith ierr = VecDuplicate(pcis->vec1_B,&pcis->vec3_B);CHKERRQ(ierr); 1922a7a6963SBarry Smith ierr = MatCreateVecs(pc->pmat,&pcis->vec1_global,0);CHKERRQ(ierr); 193854ce69bSBarry Smith ierr = PetscMalloc1(pcis->n,&pcis->work_N);CHKERRQ(ierr); 1945e8657edSStefano Zampini /* scaling vector */ 195bf83c2afSStefano Zampini if (!pcis->D) { /* it can happen that the user passed in a scaling vector via PCISSetSubdomainDiagonalScaling */ 1965e8657edSStefano Zampini ierr = VecDuplicate(pcis->vec1_B,&pcis->D);CHKERRQ(ierr); 197bf83c2afSStefano Zampini ierr = VecSet(pcis->D,pcis->scaling_factor);CHKERRQ(ierr); 198bf83c2afSStefano Zampini } 199b4319ba4SBarry Smith 200b4319ba4SBarry Smith /* Creating the scatter contexts */ 20116909a7fSStefano Zampini ierr = VecScatterCreate(pcis->vec1_N,pcis->is_I_local,pcis->vec1_D,(IS)0,&pcis->N_to_D);CHKERRQ(ierr); 20223ce1328SBarry Smith ierr = VecScatterCreate(pcis->vec1_global,pcis->is_I_global,pcis->vec1_D,(IS)0,&pcis->global_to_D);CHKERRQ(ierr); 203b4319ba4SBarry Smith ierr = VecScatterCreate(pcis->vec1_N,pcis->is_B_local,pcis->vec1_B,(IS)0,&pcis->N_to_B);CHKERRQ(ierr); 20423ce1328SBarry Smith ierr = VecScatterCreate(pcis->vec1_global,pcis->is_B_global,pcis->vec1_B,(IS)0,&pcis->global_to_B);CHKERRQ(ierr); 205b4319ba4SBarry Smith 2065e8657edSStefano Zampini /* map from boundary to local */ 2075e8657edSStefano Zampini ierr = ISLocalToGlobalMappingCreateIS(pcis->is_B_local,&pcis->BtoNmap);CHKERRQ(ierr); 2085e8657edSStefano Zampini } 2095e8657edSStefano Zampini 2105e8657edSStefano Zampini /* 2115e8657edSStefano Zampini Extracting the blocks A_II, A_BI, A_IB and A_BB from A. If the numbering 2125e8657edSStefano Zampini is such that interior nodes come first than the interface ones, we have 2135e8657edSStefano Zampini 2145e8657edSStefano Zampini [ A_II | A_IB ] 2155e8657edSStefano Zampini A = [------+------] 2165e8657edSStefano Zampini [ A_BI | A_BB ] 2175e8657edSStefano Zampini */ 2185e8657edSStefano Zampini reuse = MAT_INITIAL_MATRIX; 2193975b054SStefano Zampini if (pcis->reusesubmatrices && pc->setupcalled) { 2205e8657edSStefano Zampini if (pc->flag == SAME_NONZERO_PATTERN) { 2215e8657edSStefano Zampini reuse = MAT_REUSE_MATRIX; 2225e8657edSStefano Zampini } else { 2235e8657edSStefano Zampini reuse = MAT_INITIAL_MATRIX; 2245e8657edSStefano Zampini } 2255e8657edSStefano Zampini } 2265e8657edSStefano Zampini if (reuse == MAT_INITIAL_MATRIX) { 2275e8657edSStefano Zampini ierr = MatDestroy(&pcis->A_II);CHKERRQ(ierr); 2285e8657edSStefano Zampini ierr = MatDestroy(&pcis->A_IB);CHKERRQ(ierr); 2295e8657edSStefano Zampini ierr = MatDestroy(&pcis->A_BI);CHKERRQ(ierr); 2305e8657edSStefano Zampini ierr = MatDestroy(&pcis->A_BB);CHKERRQ(ierr); 2315e8657edSStefano Zampini } 2325e8657edSStefano Zampini 233*7dae84e0SHong Zhang ierr = MatCreateSubMatrix(matis->A,pcis->is_I_local,pcis->is_I_local,reuse,&pcis->A_II);CHKERRQ(ierr); 234*7dae84e0SHong Zhang ierr = MatCreateSubMatrix(matis->A,pcis->is_B_local,pcis->is_B_local,reuse,&pcis->A_BB);CHKERRQ(ierr); 2355e8657edSStefano Zampini ierr = PetscObjectTypeCompare((PetscObject)matis->A,MATSEQSBAIJ,&issbaij);CHKERRQ(ierr); 2365e8657edSStefano Zampini if (!issbaij) { 237*7dae84e0SHong Zhang ierr = MatCreateSubMatrix(matis->A,pcis->is_I_local,pcis->is_B_local,reuse,&pcis->A_IB);CHKERRQ(ierr); 238*7dae84e0SHong Zhang ierr = MatCreateSubMatrix(matis->A,pcis->is_B_local,pcis->is_I_local,reuse,&pcis->A_BI);CHKERRQ(ierr); 2395e8657edSStefano Zampini } else { 2405e8657edSStefano Zampini Mat newmat; 2415e8657edSStefano Zampini ierr = MatConvert(matis->A,MATSEQBAIJ,MAT_INITIAL_MATRIX,&newmat);CHKERRQ(ierr); 242*7dae84e0SHong Zhang ierr = MatCreateSubMatrix(newmat,pcis->is_I_local,pcis->is_B_local,reuse,&pcis->A_IB);CHKERRQ(ierr); 243*7dae84e0SHong Zhang ierr = MatCreateSubMatrix(newmat,pcis->is_B_local,pcis->is_I_local,reuse,&pcis->A_BI);CHKERRQ(ierr); 2445e8657edSStefano Zampini ierr = MatDestroy(&newmat);CHKERRQ(ierr); 2455e8657edSStefano Zampini } 2465e8657edSStefano Zampini 247bf83c2afSStefano Zampini /* Creating scaling vector D */ 248c5929fdfSBarry Smith ierr = PetscOptionsGetBool(((PetscObject)pc)->options,((PetscObject)pc)->prefix,"-pc_is_use_stiffness_scaling",&pcis->use_stiffness_scaling,NULL);CHKERRQ(ierr); 249bf83c2afSStefano Zampini if (pcis->use_stiffness_scaling) { 250bf83c2afSStefano Zampini ierr = MatGetDiagonal(pcis->A_BB,pcis->D);CHKERRQ(ierr); 251ba1573a8SStefano Zampini } 2522a7a6963SBarry Smith ierr = MatCreateVecs(pc->pmat,&counter,0);CHKERRQ(ierr); /* temporary auxiliar vector */ 253831a100dSStefano Zampini ierr = VecSet(counter,0.0);CHKERRQ(ierr); 254bf83c2afSStefano Zampini ierr = VecScatterBegin(pcis->global_to_B,pcis->D,counter,ADD_VALUES,SCATTER_REVERSE);CHKERRQ(ierr); 255bf83c2afSStefano Zampini ierr = VecScatterEnd(pcis->global_to_B,pcis->D,counter,ADD_VALUES,SCATTER_REVERSE);CHKERRQ(ierr); 256ba1573a8SStefano Zampini ierr = VecScatterBegin(pcis->global_to_B,counter,pcis->vec1_B,INSERT_VALUES,SCATTER_FORWARD);CHKERRQ(ierr); 257ba1573a8SStefano Zampini ierr = VecScatterEnd(pcis->global_to_B,counter,pcis->vec1_B,INSERT_VALUES,SCATTER_FORWARD);CHKERRQ(ierr); 258ba1573a8SStefano Zampini ierr = VecPointwiseDivide(pcis->D,pcis->D,pcis->vec1_B);CHKERRQ(ierr); 259892d8026SStefano Zampini ierr = VecDestroy(&counter);CHKERRQ(ierr); 260b4319ba4SBarry Smith 261b4319ba4SBarry Smith /* See historical note 01, at the bottom of this file. */ 262b4319ba4SBarry Smith 2635e8657edSStefano Zampini /* Creating the KSP contexts for the local Dirichlet and Neumann problems */ 2645e8657edSStefano Zampini if (computesolvers) { 265b4319ba4SBarry Smith PC pc_ctx; 2665e8657edSStefano Zampini 2675e8657edSStefano Zampini pcis->pure_neumann = matis->pure_neumann; 268b4319ba4SBarry Smith /* Dirichlet */ 269b4319ba4SBarry Smith ierr = KSPCreate(PETSC_COMM_SELF,&pcis->ksp_D);CHKERRQ(ierr); 270422a814eSBarry Smith ierr = KSPSetErrorIfNotConverged(pcis->ksp_D,pc->erroriffailure);CHKERRQ(ierr); 2711cee3971SBarry Smith ierr = PetscObjectIncrementTabLevel((PetscObject)pcis->ksp_D,(PetscObject)pc,1);CHKERRQ(ierr); 27223ee1639SBarry Smith ierr = KSPSetOperators(pcis->ksp_D,pcis->A_II,pcis->A_II);CHKERRQ(ierr); 273b4319ba4SBarry Smith ierr = KSPSetOptionsPrefix(pcis->ksp_D,"is_localD_");CHKERRQ(ierr); 274b4319ba4SBarry Smith ierr = KSPGetPC(pcis->ksp_D,&pc_ctx);CHKERRQ(ierr); 275b4319ba4SBarry Smith ierr = PCSetType(pc_ctx,PCLU);CHKERRQ(ierr); 276b4319ba4SBarry Smith ierr = KSPSetType(pcis->ksp_D,KSPPREONLY);CHKERRQ(ierr); 277b4319ba4SBarry Smith ierr = KSPSetFromOptions(pcis->ksp_D);CHKERRQ(ierr); 278b4319ba4SBarry Smith /* the vectors in the following line are dummy arguments, just telling the KSP the vector size. Values are not used */ 279b4319ba4SBarry Smith ierr = KSPSetUp(pcis->ksp_D);CHKERRQ(ierr); 280b4319ba4SBarry Smith /* Neumann */ 281b4319ba4SBarry Smith ierr = KSPCreate(PETSC_COMM_SELF,&pcis->ksp_N);CHKERRQ(ierr); 282422a814eSBarry Smith ierr = KSPSetErrorIfNotConverged(pcis->ksp_N,pc->erroriffailure);CHKERRQ(ierr); 2831cee3971SBarry Smith ierr = PetscObjectIncrementTabLevel((PetscObject)pcis->ksp_N,(PetscObject)pc,1);CHKERRQ(ierr); 28423ee1639SBarry Smith ierr = KSPSetOperators(pcis->ksp_N,matis->A,matis->A);CHKERRQ(ierr); 285b4319ba4SBarry Smith ierr = KSPSetOptionsPrefix(pcis->ksp_N,"is_localN_");CHKERRQ(ierr); 286b4319ba4SBarry Smith ierr = KSPGetPC(pcis->ksp_N,&pc_ctx);CHKERRQ(ierr); 287b4319ba4SBarry Smith ierr = PCSetType(pc_ctx,PCLU);CHKERRQ(ierr); 288b4319ba4SBarry Smith ierr = KSPSetType(pcis->ksp_N,KSPPREONLY);CHKERRQ(ierr); 289b4319ba4SBarry Smith ierr = KSPSetFromOptions(pcis->ksp_N);CHKERRQ(ierr); 290b4319ba4SBarry Smith { 291ace3abfcSBarry Smith PetscBool damp_fixed = PETSC_FALSE, 29290d69ab7SBarry Smith remove_nullspace_fixed = PETSC_FALSE, 29390d69ab7SBarry Smith set_damping_factor_floating = PETSC_FALSE, 29490d69ab7SBarry Smith not_damp_floating = PETSC_FALSE, 29590d69ab7SBarry Smith not_remove_nullspace_floating = PETSC_FALSE; 296b4319ba4SBarry Smith PetscReal fixed_factor, 297b4319ba4SBarry Smith floating_factor; 298b4319ba4SBarry Smith 299c5929fdfSBarry Smith ierr = PetscOptionsGetReal(((PetscObject)pc_ctx)->options,((PetscObject)pc_ctx)->prefix,"-pc_is_damp_fixed",&fixed_factor,&damp_fixed);CHKERRQ(ierr); 3002fa5cd67SKarl Rupp if (!damp_fixed) fixed_factor = 0.0; 301c5929fdfSBarry Smith ierr = PetscOptionsGetBool(((PetscObject)pc_ctx)->options,((PetscObject)pc_ctx)->prefix,"-pc_is_damp_fixed",&damp_fixed,NULL);CHKERRQ(ierr); 302b4319ba4SBarry Smith 303c5929fdfSBarry Smith ierr = PetscOptionsGetBool(((PetscObject)pc_ctx)->options,((PetscObject)pc_ctx)->prefix,"-pc_is_remove_nullspace_fixed",&remove_nullspace_fixed,NULL);CHKERRQ(ierr); 304b4319ba4SBarry Smith 305c5929fdfSBarry Smith ierr = PetscOptionsGetReal(((PetscObject)pc_ctx)->options,((PetscObject)pc_ctx)->prefix,"-pc_is_set_damping_factor_floating", 306b4319ba4SBarry Smith &floating_factor,&set_damping_factor_floating);CHKERRQ(ierr); 3072fa5cd67SKarl Rupp if (!set_damping_factor_floating) floating_factor = 0.0; 308c5929fdfSBarry Smith ierr = PetscOptionsGetBool(((PetscObject)pc_ctx)->options,((PetscObject)pc_ctx)->prefix,"-pc_is_set_damping_factor_floating",&set_damping_factor_floating,NULL);CHKERRQ(ierr); 3092fa5cd67SKarl Rupp if (!set_damping_factor_floating) floating_factor = 1.e-12; 310b4319ba4SBarry Smith 311c5929fdfSBarry Smith ierr = PetscOptionsGetBool(((PetscObject)pc_ctx)->options,((PetscObject)pc_ctx)->prefix,"-pc_is_not_damp_floating",¬_damp_floating,NULL);CHKERRQ(ierr); 312b4319ba4SBarry Smith 313c5929fdfSBarry Smith ierr = PetscOptionsGetBool(((PetscObject)pc_ctx)->options,((PetscObject)pc_ctx)->prefix,"-pc_is_not_remove_nullspace_floating",¬_remove_nullspace_floating,NULL);CHKERRQ(ierr); 314b4319ba4SBarry Smith 315b4319ba4SBarry Smith if (pcis->pure_neumann) { /* floating subdomain */ 316b4319ba4SBarry Smith if (!(not_damp_floating)) { 317c88783f4SHong Zhang ierr = PCFactorSetShiftType(pc_ctx,MAT_SHIFT_NONZERO);CHKERRQ(ierr); 318c88783f4SHong Zhang ierr = PCFactorSetShiftAmount(pc_ctx,floating_factor);CHKERRQ(ierr); 319b4319ba4SBarry Smith } 320b4319ba4SBarry Smith if (!(not_remove_nullspace_floating)) { 321b4319ba4SBarry Smith MatNullSpace nullsp; 3220298fd71SBarry Smith ierr = MatNullSpaceCreate(PETSC_COMM_SELF,PETSC_TRUE,0,NULL,&nullsp);CHKERRQ(ierr); 3235fa7ec2dSBarry Smith ierr = MatSetNullSpace(matis->A,nullsp);CHKERRQ(ierr); 324d34fcf5fSBarry Smith ierr = MatNullSpaceDestroy(&nullsp);CHKERRQ(ierr); 325b4319ba4SBarry Smith } 326b4319ba4SBarry Smith } else { /* fixed subdomain */ 327b4319ba4SBarry Smith if (damp_fixed) { 328c88783f4SHong Zhang ierr = PCFactorSetShiftType(pc_ctx,MAT_SHIFT_NONZERO);CHKERRQ(ierr); 329c88783f4SHong Zhang ierr = PCFactorSetShiftAmount(pc_ctx,floating_factor);CHKERRQ(ierr); 330b4319ba4SBarry Smith } 331b4319ba4SBarry Smith if (remove_nullspace_fixed) { 332b4319ba4SBarry Smith MatNullSpace nullsp; 3330298fd71SBarry Smith ierr = MatNullSpaceCreate(PETSC_COMM_SELF,PETSC_TRUE,0,NULL,&nullsp);CHKERRQ(ierr); 3345fa7ec2dSBarry Smith ierr = MatSetNullSpace(matis->A,nullsp);CHKERRQ(ierr); 335d34fcf5fSBarry Smith ierr = MatNullSpaceDestroy(&nullsp);CHKERRQ(ierr); 336b4319ba4SBarry Smith } 337b4319ba4SBarry Smith } 338b4319ba4SBarry Smith } 339b4319ba4SBarry Smith /* the vectors in the following line are dummy arguments, just telling the KSP the vector size. Values are not used */ 340b4319ba4SBarry Smith ierr = KSPSetUp(pcis->ksp_N);CHKERRQ(ierr); 341b4319ba4SBarry Smith } 342b4319ba4SBarry Smith 343b4319ba4SBarry Smith PetscFunctionReturn(0); 344b4319ba4SBarry Smith } 345b4319ba4SBarry Smith 346b4319ba4SBarry Smith /* -------------------------------------------------------------------------- */ 347b4319ba4SBarry Smith /* 348b4319ba4SBarry Smith PCISDestroy - 349b4319ba4SBarry Smith */ 3507087cfbeSBarry Smith PetscErrorCode PCISDestroy(PC pc) 351b4319ba4SBarry Smith { 352b4319ba4SBarry Smith PC_IS *pcis = (PC_IS*)(pc->data); 353dfbe8321SBarry Smith PetscErrorCode ierr; 354b4319ba4SBarry Smith 355b4319ba4SBarry Smith PetscFunctionBegin; 356fcfd50ebSBarry Smith ierr = ISDestroy(&pcis->is_B_local);CHKERRQ(ierr); 357fcfd50ebSBarry Smith ierr = ISDestroy(&pcis->is_I_local);CHKERRQ(ierr); 358fcfd50ebSBarry Smith ierr = ISDestroy(&pcis->is_B_global);CHKERRQ(ierr); 359fcfd50ebSBarry Smith ierr = ISDestroy(&pcis->is_I_global);CHKERRQ(ierr); 360fcfd50ebSBarry Smith ierr = MatDestroy(&pcis->A_II);CHKERRQ(ierr); 361fcfd50ebSBarry Smith ierr = MatDestroy(&pcis->A_IB);CHKERRQ(ierr); 362fcfd50ebSBarry Smith ierr = MatDestroy(&pcis->A_BI);CHKERRQ(ierr); 363fcfd50ebSBarry Smith ierr = MatDestroy(&pcis->A_BB);CHKERRQ(ierr); 364fcfd50ebSBarry Smith ierr = VecDestroy(&pcis->D);CHKERRQ(ierr); 365fcfd50ebSBarry Smith ierr = KSPDestroy(&pcis->ksp_N);CHKERRQ(ierr); 366fcfd50ebSBarry Smith ierr = KSPDestroy(&pcis->ksp_D);CHKERRQ(ierr); 367fcfd50ebSBarry Smith ierr = VecDestroy(&pcis->vec1_N);CHKERRQ(ierr); 368fcfd50ebSBarry Smith ierr = VecDestroy(&pcis->vec2_N);CHKERRQ(ierr); 369fcfd50ebSBarry Smith ierr = VecDestroy(&pcis->vec1_D);CHKERRQ(ierr); 370fcfd50ebSBarry Smith ierr = VecDestroy(&pcis->vec2_D);CHKERRQ(ierr); 371fcfd50ebSBarry Smith ierr = VecDestroy(&pcis->vec3_D);CHKERRQ(ierr); 372df187020SStefano Zampini ierr = VecDestroy(&pcis->vec4_D);CHKERRQ(ierr); 373fcfd50ebSBarry Smith ierr = VecDestroy(&pcis->vec1_B);CHKERRQ(ierr); 374fcfd50ebSBarry Smith ierr = VecDestroy(&pcis->vec2_B);CHKERRQ(ierr); 375fcfd50ebSBarry Smith ierr = VecDestroy(&pcis->vec3_B);CHKERRQ(ierr); 376fcfd50ebSBarry Smith ierr = VecDestroy(&pcis->vec1_global);CHKERRQ(ierr); 377fcfd50ebSBarry Smith ierr = VecScatterDestroy(&pcis->global_to_D);CHKERRQ(ierr); 378fcfd50ebSBarry Smith ierr = VecScatterDestroy(&pcis->N_to_B);CHKERRQ(ierr); 37916909a7fSStefano Zampini ierr = VecScatterDestroy(&pcis->N_to_D);CHKERRQ(ierr); 380fcfd50ebSBarry Smith ierr = VecScatterDestroy(&pcis->global_to_B);CHKERRQ(ierr); 38105b42c5fSBarry Smith ierr = PetscFree(pcis->work_N);CHKERRQ(ierr); 3827dbfca69SStefano Zampini if (pcis->n_neigh > -1) { 3838d12c799SStefano Zampini ierr = ISLocalToGlobalMappingRestoreInfo(pcis->mapping,&(pcis->n_neigh),&(pcis->neigh),&(pcis->n_shared),&(pcis->shared));CHKERRQ(ierr); 3847dbfca69SStefano Zampini } 3858d12c799SStefano Zampini ierr = ISLocalToGlobalMappingDestroy(&pcis->mapping);CHKERRQ(ierr); 3865e8657edSStefano Zampini ierr = ISLocalToGlobalMappingDestroy(&pcis->BtoNmap);CHKERRQ(ierr); 387bdf89e91SBarry Smith ierr = PetscObjectComposeFunction((PetscObject)pc,"PCISSetUseStiffnessScaling_C",NULL);CHKERRQ(ierr); 388bdf89e91SBarry Smith ierr = PetscObjectComposeFunction((PetscObject)pc,"PCISSetSubdomainScalingFactor_C",NULL);CHKERRQ(ierr); 389bdf89e91SBarry Smith ierr = PetscObjectComposeFunction((PetscObject)pc,"PCISSetSubdomainDiagonalScaling_C",NULL);CHKERRQ(ierr); 390b4319ba4SBarry Smith PetscFunctionReturn(0); 391b4319ba4SBarry Smith } 392b4319ba4SBarry Smith 393b4319ba4SBarry Smith /* -------------------------------------------------------------------------- */ 394b4319ba4SBarry Smith /* 395b4319ba4SBarry Smith PCISCreate - 396b4319ba4SBarry Smith */ 3977087cfbeSBarry Smith PetscErrorCode PCISCreate(PC pc) 398b4319ba4SBarry Smith { 399b4319ba4SBarry Smith PC_IS *pcis = (PC_IS*)(pc->data); 400831a100dSStefano Zampini PetscErrorCode ierr; 401b4319ba4SBarry Smith 402b4319ba4SBarry Smith PetscFunctionBegin; 403b4319ba4SBarry Smith pcis->is_B_local = 0; 404b4319ba4SBarry Smith pcis->is_I_local = 0; 405b4319ba4SBarry Smith pcis->is_B_global = 0; 406b4319ba4SBarry Smith pcis->is_I_global = 0; 407b4319ba4SBarry Smith pcis->A_II = 0; 408b4319ba4SBarry Smith pcis->A_IB = 0; 409b4319ba4SBarry Smith pcis->A_BI = 0; 410b4319ba4SBarry Smith pcis->A_BB = 0; 411b4319ba4SBarry Smith pcis->D = 0; 412b4319ba4SBarry Smith pcis->ksp_N = 0; 413b4319ba4SBarry Smith pcis->ksp_D = 0; 414b4319ba4SBarry Smith pcis->vec1_N = 0; 415b4319ba4SBarry Smith pcis->vec2_N = 0; 416b4319ba4SBarry Smith pcis->vec1_D = 0; 417b4319ba4SBarry Smith pcis->vec2_D = 0; 418b4319ba4SBarry Smith pcis->vec3_D = 0; 419b4319ba4SBarry Smith pcis->vec1_B = 0; 420b4319ba4SBarry Smith pcis->vec2_B = 0; 421b4319ba4SBarry Smith pcis->vec3_B = 0; 422b4319ba4SBarry Smith pcis->vec1_global = 0; 423b4319ba4SBarry Smith pcis->work_N = 0; 424b4319ba4SBarry Smith pcis->global_to_D = 0; 425b4319ba4SBarry Smith pcis->N_to_B = 0; 42616909a7fSStefano Zampini pcis->N_to_D = 0; 427b4319ba4SBarry Smith pcis->global_to_B = 0; 4288d12c799SStefano Zampini pcis->mapping = 0; 4295e8657edSStefano Zampini pcis->BtoNmap = 0; 4307dbfca69SStefano Zampini pcis->n_neigh = -1; 431831a100dSStefano Zampini pcis->scaling_factor = 1.0; 4323975b054SStefano Zampini pcis->reusesubmatrices = PETSC_TRUE; 433831a100dSStefano Zampini /* composing functions */ 434bdf89e91SBarry Smith ierr = PetscObjectComposeFunction((PetscObject)pc,"PCISSetUseStiffnessScaling_C",PCISSetUseStiffnessScaling_IS);CHKERRQ(ierr); 435bdf89e91SBarry Smith ierr = PetscObjectComposeFunction((PetscObject)pc,"PCISSetSubdomainScalingFactor_C",PCISSetSubdomainScalingFactor_IS);CHKERRQ(ierr); 436bdf89e91SBarry Smith ierr = PetscObjectComposeFunction((PetscObject)pc,"PCISSetSubdomainDiagonalScaling_C",PCISSetSubdomainDiagonalScaling_IS);CHKERRQ(ierr); 437b4319ba4SBarry Smith PetscFunctionReturn(0); 438b4319ba4SBarry Smith } 439b4319ba4SBarry Smith 440b4319ba4SBarry Smith /* -------------------------------------------------------------------------- */ 441b4319ba4SBarry Smith /* 442b4319ba4SBarry Smith PCISApplySchur - 443b4319ba4SBarry Smith 444b4319ba4SBarry Smith Input parameters: 445b4319ba4SBarry Smith . pc - preconditioner context 446b4319ba4SBarry Smith . v - vector to which the Schur complement is to be applied (it is NOT modified inside this function, UNLESS vec2_B is null) 447b4319ba4SBarry Smith 448b4319ba4SBarry Smith Output parameters: 449b4319ba4SBarry Smith . vec1_B - result of Schur complement applied to chunk 450b4319ba4SBarry Smith . vec2_B - garbage (used as work space), or null (and v is used as workspace) 451b4319ba4SBarry Smith . vec1_D - garbage (used as work space) 452b4319ba4SBarry Smith . vec2_D - garbage (used as work space) 453b4319ba4SBarry Smith 454b4319ba4SBarry Smith */ 4557087cfbeSBarry Smith PetscErrorCode PCISApplySchur(PC pc, Vec v, Vec vec1_B, Vec vec2_B, Vec vec1_D, Vec vec2_D) 456b4319ba4SBarry Smith { 457dfbe8321SBarry Smith PetscErrorCode ierr; 458b4319ba4SBarry Smith PC_IS *pcis = (PC_IS*)(pc->data); 459b4319ba4SBarry Smith 460b4319ba4SBarry Smith PetscFunctionBegin; 4612fa5cd67SKarl Rupp if (!vec2_B) vec2_B = v; 462b4319ba4SBarry Smith 463b4319ba4SBarry Smith ierr = MatMult(pcis->A_BB,v,vec1_B);CHKERRQ(ierr); 464b4319ba4SBarry Smith ierr = MatMult(pcis->A_IB,v,vec1_D);CHKERRQ(ierr); 46523ce1328SBarry Smith ierr = KSPSolve(pcis->ksp_D,vec1_D,vec2_D);CHKERRQ(ierr); 466b4319ba4SBarry Smith ierr = MatMult(pcis->A_BI,vec2_D,vec2_B);CHKERRQ(ierr); 467efb30889SBarry Smith ierr = VecAXPY(vec1_B,-1.0,vec2_B);CHKERRQ(ierr); 468b4319ba4SBarry Smith PetscFunctionReturn(0); 469b4319ba4SBarry Smith } 470b4319ba4SBarry Smith 471b4319ba4SBarry Smith /* -------------------------------------------------------------------------- */ 472b4319ba4SBarry Smith /* 473b4319ba4SBarry Smith PCISScatterArrayNToVecB - Scatters interface node values from a big array (of all local nodes, interior or interface, 474b4319ba4SBarry Smith including ghosts) into an interface vector, when in SCATTER_FORWARD mode, or vice-versa, when in SCATTER_REVERSE 475b4319ba4SBarry Smith mode. 476b4319ba4SBarry Smith 477b4319ba4SBarry Smith Input parameters: 478b4319ba4SBarry Smith . pc - preconditioner context 479b4319ba4SBarry Smith . array_N - [when in SCATTER_FORWARD mode] Array to be scattered into the vector 480b4319ba4SBarry Smith . v_B - [when in SCATTER_REVERSE mode] Vector to be scattered into the array 481b4319ba4SBarry Smith 482b4319ba4SBarry Smith Output parameter: 483b4319ba4SBarry Smith . array_N - [when in SCATTER_REVERSE mode] Array to receive the scattered vector 484b4319ba4SBarry Smith . v_B - [when in SCATTER_FORWARD mode] Vector to receive the scattered array 485b4319ba4SBarry Smith 486b4319ba4SBarry Smith Notes: 487b4319ba4SBarry Smith The entries in the array that do not correspond to interface nodes remain unaltered. 488b4319ba4SBarry Smith */ 4897087cfbeSBarry Smith PetscErrorCode PCISScatterArrayNToVecB(PetscScalar *array_N, Vec v_B, InsertMode imode, ScatterMode smode, PC pc) 490b4319ba4SBarry Smith { 4915d0c19d7SBarry Smith PetscInt i; 4925d0c19d7SBarry Smith const PetscInt *idex; 4936849ba73SBarry Smith PetscErrorCode ierr; 494b4319ba4SBarry Smith PetscScalar *array_B; 495b4319ba4SBarry Smith PC_IS *pcis = (PC_IS*)(pc->data); 496b4319ba4SBarry Smith 497b4319ba4SBarry Smith PetscFunctionBegin; 498b4319ba4SBarry Smith ierr = VecGetArray(v_B,&array_B);CHKERRQ(ierr); 499b4319ba4SBarry Smith ierr = ISGetIndices(pcis->is_B_local,&idex);CHKERRQ(ierr); 500b4319ba4SBarry Smith 501b4319ba4SBarry Smith if (smode == SCATTER_FORWARD) { 502b4319ba4SBarry Smith if (imode == INSERT_VALUES) { 5032fa5cd67SKarl Rupp for (i=0; i<pcis->n_B; i++) array_B[i] = array_N[idex[i]]; 504b4319ba4SBarry Smith } else { /* ADD_VALUES */ 5052fa5cd67SKarl Rupp for (i=0; i<pcis->n_B; i++) array_B[i] += array_N[idex[i]]; 506b4319ba4SBarry Smith } 507b4319ba4SBarry Smith } else { /* SCATTER_REVERSE */ 508b4319ba4SBarry Smith if (imode == INSERT_VALUES) { 5092fa5cd67SKarl Rupp for (i=0; i<pcis->n_B; i++) array_N[idex[i]] = array_B[i]; 510b4319ba4SBarry Smith } else { /* ADD_VALUES */ 5112fa5cd67SKarl Rupp for (i=0; i<pcis->n_B; i++) array_N[idex[i]] += array_B[i]; 512b4319ba4SBarry Smith } 513b4319ba4SBarry Smith } 514b4319ba4SBarry Smith ierr = ISRestoreIndices(pcis->is_B_local,&idex);CHKERRQ(ierr); 515b4319ba4SBarry Smith ierr = VecRestoreArray(v_B,&array_B);CHKERRQ(ierr); 516b4319ba4SBarry Smith PetscFunctionReturn(0); 517b4319ba4SBarry Smith } 518b4319ba4SBarry Smith 519b4319ba4SBarry Smith /* -------------------------------------------------------------------------- */ 520b4319ba4SBarry Smith /* 521b4319ba4SBarry Smith PCISApplyInvSchur - Solves the Neumann problem related to applying the inverse of the Schur complement. 522b4319ba4SBarry Smith More precisely, solves the problem: 523b4319ba4SBarry Smith [ A_II A_IB ] [ . ] [ 0 ] 524b4319ba4SBarry Smith [ ] [ ] = [ ] 525b4319ba4SBarry Smith [ A_BI A_BB ] [ x ] [ b ] 526b4319ba4SBarry Smith 527b4319ba4SBarry Smith Input parameters: 528b4319ba4SBarry Smith . pc - preconditioner context 529b4319ba4SBarry Smith . b - vector of local interface nodes (including ghosts) 530b4319ba4SBarry Smith 531b4319ba4SBarry Smith Output parameters: 532b4319ba4SBarry Smith . x - vector of local interface nodes (including ghosts); returns the application of the inverse of the Schur 533b4319ba4SBarry Smith complement to b 534b4319ba4SBarry Smith . vec1_N - vector of local nodes (interior and interface, including ghosts); returns garbage (used as work space) 535b4319ba4SBarry Smith . vec2_N - vector of local nodes (interior and interface, including ghosts); returns garbage (used as work space) 536b4319ba4SBarry Smith 537b4319ba4SBarry Smith */ 5387087cfbeSBarry Smith PetscErrorCode PCISApplyInvSchur(PC pc, Vec b, Vec x, Vec vec1_N, Vec vec2_N) 539b4319ba4SBarry Smith { 540dfbe8321SBarry Smith PetscErrorCode ierr; 541b4319ba4SBarry Smith PC_IS *pcis = (PC_IS*)(pc->data); 542b4319ba4SBarry Smith 543b4319ba4SBarry Smith PetscFunctionBegin; 544b4319ba4SBarry Smith /* 545b4319ba4SBarry Smith Neumann solvers. 546b4319ba4SBarry Smith Applying the inverse of the local Schur complement, i.e, solving a Neumann 547b4319ba4SBarry Smith Problem with zero at the interior nodes of the RHS and extracting the interface 548b4319ba4SBarry Smith part of the solution. inverse Schur complement is applied to b and the result 549b4319ba4SBarry Smith is stored in x. 550b4319ba4SBarry Smith */ 551b4319ba4SBarry Smith /* Setting the RHS vec1_N */ 552efb30889SBarry Smith ierr = VecSet(vec1_N,0.0);CHKERRQ(ierr); 553ca9f406cSSatish Balay ierr = VecScatterBegin(pcis->N_to_B,b,vec1_N,INSERT_VALUES,SCATTER_REVERSE);CHKERRQ(ierr); 554ca9f406cSSatish Balay ierr = VecScatterEnd (pcis->N_to_B,b,vec1_N,INSERT_VALUES,SCATTER_REVERSE);CHKERRQ(ierr); 555b4319ba4SBarry Smith /* Checking for consistency of the RHS */ 556b4319ba4SBarry Smith { 557ace3abfcSBarry Smith PetscBool flg = PETSC_FALSE; 558c5929fdfSBarry Smith ierr = PetscOptionsGetBool(NULL,NULL,"-pc_is_check_consistency",&flg,NULL);CHKERRQ(ierr); 559b4319ba4SBarry Smith if (flg) { 560b4319ba4SBarry Smith PetscScalar average; 5613050cee2SBarry Smith PetscViewer viewer; 562ce94432eSBarry Smith ierr = PetscViewerASCIIGetStdout(PetscObjectComm((PetscObject)pc),&viewer);CHKERRQ(ierr); 5633050cee2SBarry Smith 564b4319ba4SBarry Smith ierr = VecSum(vec1_N,&average);CHKERRQ(ierr); 565b4319ba4SBarry Smith average = average / ((PetscReal)pcis->n); 5661575c14dSBarry Smith ierr = PetscViewerASCIIPushSynchronized(viewer);CHKERRQ(ierr); 567b4319ba4SBarry Smith if (pcis->pure_neumann) { 5689f73f8ecSBarry Smith ierr = PetscViewerASCIISynchronizedPrintf(viewer,"Subdomain %04d is floating. Average = % 1.14e\n",PetscGlobalRank,PetscAbsScalar(average));CHKERRQ(ierr); 569b4319ba4SBarry Smith } else { 5709f73f8ecSBarry Smith ierr = PetscViewerASCIISynchronizedPrintf(viewer,"Subdomain %04d is fixed. Average = % 1.14e\n",PetscGlobalRank,PetscAbsScalar(average));CHKERRQ(ierr); 571b4319ba4SBarry Smith } 5723050cee2SBarry Smith ierr = PetscViewerFlush(viewer);CHKERRQ(ierr); 5731575c14dSBarry Smith ierr = PetscViewerASCIIPopSynchronized(viewer);CHKERRQ(ierr); 574b4319ba4SBarry Smith } 575b4319ba4SBarry Smith } 576b4319ba4SBarry Smith /* Solving the system for vec2_N */ 57723ce1328SBarry Smith ierr = KSPSolve(pcis->ksp_N,vec1_N,vec2_N);CHKERRQ(ierr); 578b4319ba4SBarry Smith /* Extracting the local interface vector out of the solution */ 579ca9f406cSSatish Balay ierr = VecScatterBegin(pcis->N_to_B,vec2_N,x,INSERT_VALUES,SCATTER_FORWARD);CHKERRQ(ierr); 580ca9f406cSSatish Balay ierr = VecScatterEnd (pcis->N_to_B,vec2_N,x,INSERT_VALUES,SCATTER_FORWARD);CHKERRQ(ierr); 581b4319ba4SBarry Smith PetscFunctionReturn(0); 582b4319ba4SBarry Smith } 583b4319ba4SBarry Smith 584b4319ba4SBarry Smith 585b4319ba4SBarry Smith 586b4319ba4SBarry Smith 587b4319ba4SBarry Smith 588b4319ba4SBarry Smith 589b4319ba4SBarry Smith 590b4319ba4SBarry Smith 591b4319ba4SBarry Smith 592