xref: /petsc/src/ksp/pc/impls/bddc/bddcprivate.c (revision 88ebb7492cd0215be3d18dc7a81e485fff46f8ca)
1674ae819SStefano Zampini #include "bddc.h"
2674ae819SStefano Zampini #include "bddcprivate.h"
3674ae819SStefano Zampini #include <petscblaslapack.h>
4674ae819SStefano Zampini 
5674ae819SStefano Zampini #undef __FUNCT__
6674ae819SStefano Zampini #define __FUNCT__ "PCBDDCResetCustomization"
7674ae819SStefano Zampini PetscErrorCode PCBDDCResetCustomization(PC pc)
8674ae819SStefano Zampini {
9674ae819SStefano Zampini   PC_BDDC        *pcbddc = (PC_BDDC*)pc->data;
10674ae819SStefano Zampini   PetscInt       i;
11674ae819SStefano Zampini   PetscErrorCode ierr;
12674ae819SStefano Zampini 
13674ae819SStefano Zampini   PetscFunctionBegin;
14674ae819SStefano Zampini   ierr = PCBDDCGraphResetCSR(pcbddc->mat_graph);CHKERRQ(ierr);
15674ae819SStefano Zampini   ierr = ISDestroy(&pcbddc->user_primal_vertices);CHKERRQ(ierr);
16674ae819SStefano Zampini   ierr = MatNullSpaceDestroy(&pcbddc->NullSpace);CHKERRQ(ierr);
17674ae819SStefano Zampini   ierr = ISDestroy(&pcbddc->NeumannBoundaries);CHKERRQ(ierr);
18674ae819SStefano Zampini   ierr = ISDestroy(&pcbddc->DirichletBoundaries);CHKERRQ(ierr);
19674ae819SStefano Zampini   for (i=0;i<pcbddc->n_ISForDofs;i++) {
20674ae819SStefano Zampini     ierr = ISDestroy(&pcbddc->ISForDofs[i]);CHKERRQ(ierr);
21674ae819SStefano Zampini   }
22674ae819SStefano Zampini   ierr = PetscFree(pcbddc->ISForDofs);CHKERRQ(ierr);
23674ae819SStefano Zampini   PetscFunctionReturn(0);
24674ae819SStefano Zampini }
25674ae819SStefano Zampini 
26674ae819SStefano Zampini #undef __FUNCT__
27674ae819SStefano Zampini #define __FUNCT__ "PCBDDCResetTopography"
28674ae819SStefano Zampini PetscErrorCode PCBDDCResetTopography(PC pc)
29674ae819SStefano Zampini {
30674ae819SStefano Zampini   PC_BDDC        *pcbddc = (PC_BDDC*)pc->data;
31674ae819SStefano Zampini   PetscErrorCode ierr;
32674ae819SStefano Zampini 
33674ae819SStefano Zampini   PetscFunctionBegin;
34674ae819SStefano Zampini   ierr = MatDestroy(&pcbddc->ChangeOfBasisMatrix);CHKERRQ(ierr);
35674ae819SStefano Zampini   ierr = MatDestroy(&pcbddc->ConstraintMatrix);CHKERRQ(ierr);
36674ae819SStefano Zampini   ierr = PCBDDCGraphReset(pcbddc->mat_graph);CHKERRQ(ierr);
37674ae819SStefano Zampini   PetscFunctionReturn(0);
38674ae819SStefano Zampini }
39674ae819SStefano Zampini 
40674ae819SStefano Zampini #undef __FUNCT__
41674ae819SStefano Zampini #define __FUNCT__ "PCBDDCResetSolvers"
42674ae819SStefano Zampini PetscErrorCode PCBDDCResetSolvers(PC pc)
43674ae819SStefano Zampini {
44674ae819SStefano Zampini   PC_BDDC        *pcbddc = (PC_BDDC*)pc->data;
45674ae819SStefano Zampini   PetscErrorCode ierr;
46674ae819SStefano Zampini 
47674ae819SStefano Zampini   PetscFunctionBegin;
48674ae819SStefano Zampini   ierr = VecDestroy(&pcbddc->coarse_vec);CHKERRQ(ierr);
49674ae819SStefano Zampini   ierr = VecDestroy(&pcbddc->coarse_rhs);CHKERRQ(ierr);
50674ae819SStefano Zampini   ierr = KSPDestroy(&pcbddc->coarse_ksp);CHKERRQ(ierr);
51674ae819SStefano Zampini   ierr = MatDestroy(&pcbddc->coarse_mat);CHKERRQ(ierr);
52674ae819SStefano Zampini   ierr = MatDestroy(&pcbddc->coarse_phi_B);CHKERRQ(ierr);
53674ae819SStefano Zampini   ierr = MatDestroy(&pcbddc->coarse_phi_D);CHKERRQ(ierr);
5415aaf578SStefano Zampini   ierr = MatDestroy(&pcbddc->coarse_psi_B);CHKERRQ(ierr);
5515aaf578SStefano Zampini   ierr = MatDestroy(&pcbddc->coarse_psi_D);CHKERRQ(ierr);
56674ae819SStefano Zampini   ierr = VecDestroy(&pcbddc->vec1_P);CHKERRQ(ierr);
57674ae819SStefano Zampini   ierr = VecDestroy(&pcbddc->vec1_C);CHKERRQ(ierr);
58674ae819SStefano Zampini   ierr = MatDestroy(&pcbddc->local_auxmat1);CHKERRQ(ierr);
59674ae819SStefano Zampini   ierr = MatDestroy(&pcbddc->local_auxmat2);CHKERRQ(ierr);
60674ae819SStefano Zampini   ierr = VecDestroy(&pcbddc->vec1_R);CHKERRQ(ierr);
61674ae819SStefano Zampini   ierr = VecDestroy(&pcbddc->vec2_R);CHKERRQ(ierr);
62674ae819SStefano Zampini   ierr = VecDestroy(&pcbddc->vec4_D);CHKERRQ(ierr);
63674ae819SStefano Zampini   ierr = VecScatterDestroy(&pcbddc->R_to_B);CHKERRQ(ierr);
64674ae819SStefano Zampini   ierr = VecScatterDestroy(&pcbddc->R_to_D);CHKERRQ(ierr);
65674ae819SStefano Zampini   ierr = VecScatterDestroy(&pcbddc->coarse_loc_to_glob);CHKERRQ(ierr);
66674ae819SStefano Zampini   ierr = PetscFree(pcbddc->local_primal_indices);CHKERRQ(ierr);
67674ae819SStefano Zampini   ierr = PetscFree(pcbddc->replicated_local_primal_indices);CHKERRQ(ierr);
68674ae819SStefano Zampini   ierr = PetscFree(pcbddc->replicated_local_primal_values);CHKERRQ(ierr);
69674ae819SStefano Zampini   ierr = PetscFree(pcbddc->local_primal_displacements);CHKERRQ(ierr);
70674ae819SStefano Zampini   ierr = PetscFree(pcbddc->local_primal_sizes);CHKERRQ(ierr);
71674ae819SStefano Zampini   PetscFunctionReturn(0);
72674ae819SStefano Zampini }
73674ae819SStefano Zampini 
74674ae819SStefano Zampini #undef __FUNCT__
756bfb1811SStefano Zampini #define __FUNCT__ "PCBDDCCreateWorkVectors"
766bfb1811SStefano Zampini PetscErrorCode PCBDDCCreateWorkVectors(PC pc)
776bfb1811SStefano Zampini {
786bfb1811SStefano Zampini   PC_BDDC        *pcbddc = (PC_BDDC*)pc->data;
796bfb1811SStefano Zampini   PC_IS          *pcis = (PC_IS*)pc->data;
806bfb1811SStefano Zampini   VecType        impVecType;
816bfb1811SStefano Zampini   PetscInt       n_vertices,n_constraints,local_primal_size,n_R;
826bfb1811SStefano Zampini   PetscErrorCode ierr;
836bfb1811SStefano Zampini 
846bfb1811SStefano Zampini   PetscFunctionBegin;
856bfb1811SStefano Zampini   ierr = PCBDDCGetPrimalVerticesLocalIdx(pc,&n_vertices,NULL);CHKERRQ(ierr);
866bfb1811SStefano Zampini   ierr = PCBDDCGetPrimalConstraintsLocalIdx(pc,&n_constraints,NULL);CHKERRQ(ierr);
876bfb1811SStefano Zampini   local_primal_size = n_constraints+n_vertices;
886bfb1811SStefano Zampini   n_R = pcis->n-n_vertices;
896bfb1811SStefano Zampini   /* local work vectors */
906bfb1811SStefano Zampini   ierr = VecGetType(pcis->vec1_N,&impVecType);CHKERRQ(ierr);
916bfb1811SStefano Zampini   ierr = VecDuplicate(pcis->vec1_D,&pcbddc->vec4_D);CHKERRQ(ierr);
926bfb1811SStefano Zampini   ierr = VecCreate(PetscObjectComm((PetscObject)pcis->vec1_N),&pcbddc->vec1_R);CHKERRQ(ierr);
936bfb1811SStefano Zampini   ierr = VecSetSizes(pcbddc->vec1_R,PETSC_DECIDE,n_R);CHKERRQ(ierr);
946bfb1811SStefano Zampini   ierr = VecSetType(pcbddc->vec1_R,impVecType);CHKERRQ(ierr);
956bfb1811SStefano Zampini   ierr = VecDuplicate(pcbddc->vec1_R,&pcbddc->vec2_R);CHKERRQ(ierr);
9683b7ccabSStefano Zampini   ierr = VecCreate(PetscObjectComm((PetscObject)pcis->vec1_N),&pcbddc->vec1_P);CHKERRQ(ierr);
976bfb1811SStefano Zampini   ierr = VecSetSizes(pcbddc->vec1_P,PETSC_DECIDE,local_primal_size);CHKERRQ(ierr);
986bfb1811SStefano Zampini   ierr = VecSetType(pcbddc->vec1_P,impVecType);CHKERRQ(ierr);
9983b7ccabSStefano Zampini   if (n_constraints) {
10083b7ccabSStefano Zampini     ierr = VecCreate(PetscObjectComm((PetscObject)pcis->vec1_N),&pcbddc->vec1_C);CHKERRQ(ierr);
10183b7ccabSStefano Zampini     ierr = VecSetSizes(pcbddc->vec1_C,PETSC_DECIDE,n_constraints);CHKERRQ(ierr);
10283b7ccabSStefano Zampini     ierr = VecSetType(pcbddc->vec1_C,impVecType);CHKERRQ(ierr);
10383b7ccabSStefano Zampini   }
1046bfb1811SStefano Zampini   PetscFunctionReturn(0);
1056bfb1811SStefano Zampini }
1066bfb1811SStefano Zampini 
1076bfb1811SStefano Zampini #undef __FUNCT__
108*88ebb749SStefano Zampini #define __FUNCT__ "PCBDDCSetUpCorrectionAndBasis"
109*88ebb749SStefano Zampini PetscErrorCode PCBDDCSetUpCorrectionAndBasis(PC pc, IS is_R_local)
110*88ebb749SStefano Zampini {
111*88ebb749SStefano Zampini   PC_IS        *pcis = (PC_IS*)pc->data;
112*88ebb749SStefano Zampini   PC_BDDC        *pcbddc = (PC_BDDC*)pc->data;
113*88ebb749SStefano Zampini   PetscErrorCode ierr;
114*88ebb749SStefano Zampini   Mat          A_RV,A_VR,A_VV;
115*88ebb749SStefano Zampini   Mat          M1;
116*88ebb749SStefano Zampini   Mat          C_CR;
117*88ebb749SStefano Zampini   Mat          AUXMAT;
118*88ebb749SStefano Zampini   Vec          vec1_C;
119*88ebb749SStefano Zampini   Vec          vec2_C;
120*88ebb749SStefano Zampini   Vec          vec1_V;
121*88ebb749SStefano Zampini   Vec          vec2_V;
122*88ebb749SStefano Zampini   IS           is_C_local,is_V_local,is_aux1;
123*88ebb749SStefano Zampini   ISLocalToGlobalMapping BtoNmap;
124*88ebb749SStefano Zampini   PetscInt     *nnz;
125*88ebb749SStefano Zampini   PetscInt     *idx_V_B;
126*88ebb749SStefano Zampini   PetscInt     *auxindices;
127*88ebb749SStefano Zampini   PetscInt     index;
128*88ebb749SStefano Zampini   PetscScalar* array2;
129*88ebb749SStefano Zampini   MatFactorInfo matinfo;
130*88ebb749SStefano Zampini   PetscBool    setsym=PETSC_FALSE,issym=PETSC_FALSE;
131*88ebb749SStefano Zampini   /* old variables */
132*88ebb749SStefano Zampini   VecType           impVecType;
133*88ebb749SStefano Zampini   MatType           impMatType;
134*88ebb749SStefano Zampini   PetscInt          n_R=0;
135*88ebb749SStefano Zampini   PetscInt          n_D=0;
136*88ebb749SStefano Zampini   PetscInt          n_B=0;
137*88ebb749SStefano Zampini   PetscScalar       zero=0.0;
138*88ebb749SStefano Zampini   PetscScalar       one=1.0;
139*88ebb749SStefano Zampini   PetscScalar       m_one=-1.0;
140*88ebb749SStefano Zampini   PetscScalar*      array;
141*88ebb749SStefano Zampini   PetscScalar       *coarse_submat_vals;
142*88ebb749SStefano Zampini   PetscInt          *idx_R_local;
143*88ebb749SStefano Zampini   PetscReal         *coarsefunctions_errors,*constraints_errors;
144*88ebb749SStefano Zampini   /* auxiliary indices */
145*88ebb749SStefano Zampini   PetscInt          i,j;
146*88ebb749SStefano Zampini   /* for verbose output of bddc */
147*88ebb749SStefano Zampini   PetscViewer       viewer=pcbddc->dbg_viewer;
148*88ebb749SStefano Zampini   PetscInt          dbg_flag=pcbddc->dbg_flag;
149*88ebb749SStefano Zampini   /* for counting coarse dofs */
150*88ebb749SStefano Zampini   PetscInt          n_vertices,n_constraints;
151*88ebb749SStefano Zampini   PetscInt          size_of_constraint;
152*88ebb749SStefano Zampini   PetscInt          *row_cmat_indices;
153*88ebb749SStefano Zampini   PetscScalar       *row_cmat_values;
154*88ebb749SStefano Zampini   PetscInt          *vertices;
155*88ebb749SStefano Zampini 
156*88ebb749SStefano Zampini   PetscFunctionBegin;
157*88ebb749SStefano Zampini   /* get number of vertices */
158*88ebb749SStefano Zampini   ierr = PCBDDCGetPrimalVerticesLocalIdx(pc,&n_vertices,NULL);CHKERRQ(ierr);
159*88ebb749SStefano Zampini   n_constraints = pcbddc->local_primal_size-n_vertices;
160*88ebb749SStefano Zampini   /* Set Non-overlapping dimensions */
161*88ebb749SStefano Zampini   n_B = pcis->n_B; n_D = pcis->n - n_B;
162*88ebb749SStefano Zampini   n_R = pcis->n-n_vertices;
163*88ebb749SStefano Zampini 
164*88ebb749SStefano Zampini   /* Set types for local objects needed by BDDC precondtioner */
165*88ebb749SStefano Zampini   impMatType = MATSEQDENSE;
166*88ebb749SStefano Zampini   impVecType = VECSEQ;
167*88ebb749SStefano Zampini 
168*88ebb749SStefano Zampini   /* Allocating some extra storage just to be safe */
169*88ebb749SStefano Zampini   ierr = PetscMalloc (pcis->n*sizeof(PetscInt),&nnz);CHKERRQ(ierr);
170*88ebb749SStefano Zampini   ierr = PetscMalloc (pcis->n*sizeof(PetscInt),&auxindices);CHKERRQ(ierr);
171*88ebb749SStefano Zampini   for (i=0;i<pcis->n;i++) auxindices[i]=i;
172*88ebb749SStefano Zampini 
173*88ebb749SStefano Zampini   ierr = PCBDDCGetPrimalVerticesLocalIdx(pc,&n_vertices,&vertices);CHKERRQ(ierr);
174*88ebb749SStefano Zampini   /* vertices in boundary numbering */
175*88ebb749SStefano Zampini   ierr = PetscMalloc(n_vertices*sizeof(PetscInt),&idx_V_B);CHKERRQ(ierr);
176*88ebb749SStefano Zampini   ierr = ISLocalToGlobalMappingCreateIS(pcis->is_B_local,&BtoNmap);CHKERRQ(ierr);
177*88ebb749SStefano Zampini   ierr = ISGlobalToLocalMappingApply(BtoNmap,IS_GTOLM_DROP,n_vertices,vertices,&i,idx_V_B);CHKERRQ(ierr);
178*88ebb749SStefano Zampini   ierr = ISLocalToGlobalMappingDestroy(&BtoNmap);CHKERRQ(ierr);
179*88ebb749SStefano Zampini   if (i != n_vertices) {
180*88ebb749SStefano Zampini     SETERRQ2(PETSC_COMM_SELF,PETSC_ERR_SUP,"Error in boundary numbering for BDDC vertices! %d != %d\n",n_vertices,i);
181*88ebb749SStefano Zampini   }
182*88ebb749SStefano Zampini 
183*88ebb749SStefano Zampini   /* some work vectors on vertices and/or constraints */
184*88ebb749SStefano Zampini   if (n_vertices) {
185*88ebb749SStefano Zampini     ierr = VecCreate(PETSC_COMM_SELF,&vec1_V);CHKERRQ(ierr);
186*88ebb749SStefano Zampini     ierr = VecSetSizes(vec1_V,n_vertices,n_vertices);CHKERRQ(ierr);
187*88ebb749SStefano Zampini     ierr = VecSetType(vec1_V,impVecType);CHKERRQ(ierr);
188*88ebb749SStefano Zampini     ierr = VecDuplicate(vec1_V,&vec2_V);CHKERRQ(ierr);
189*88ebb749SStefano Zampini   }
190*88ebb749SStefano Zampini   if (n_constraints) {
191*88ebb749SStefano Zampini     ierr = VecDuplicate(pcbddc->vec1_C,&vec1_C);CHKERRQ(ierr);
192*88ebb749SStefano Zampini     ierr = VecDuplicate(pcbddc->vec1_C,&vec2_C);CHKERRQ(ierr);
193*88ebb749SStefano Zampini   }
194*88ebb749SStefano Zampini   /* Precompute stuffs needed for preprocessing and application of BDDC*/
195*88ebb749SStefano Zampini   if (n_constraints) {
196*88ebb749SStefano Zampini     ierr = MatCreate(PETSC_COMM_SELF,&pcbddc->local_auxmat2);CHKERRQ(ierr);
197*88ebb749SStefano Zampini     ierr = MatSetSizes(pcbddc->local_auxmat2,n_R,n_constraints,n_R,n_constraints);CHKERRQ(ierr);
198*88ebb749SStefano Zampini     ierr = MatSetType(pcbddc->local_auxmat2,impMatType);CHKERRQ(ierr);
199*88ebb749SStefano Zampini     ierr = MatSeqDenseSetPreallocation(pcbddc->local_auxmat2,NULL);CHKERRQ(ierr);
200*88ebb749SStefano Zampini 
201*88ebb749SStefano Zampini     /* Create Constraint matrix on R nodes: C_{CR}  */
202*88ebb749SStefano Zampini     ierr = ISCreateStride(PETSC_COMM_SELF,n_constraints,n_vertices,1,&is_C_local);CHKERRQ(ierr);
203*88ebb749SStefano Zampini     ierr = MatGetSubMatrix(pcbddc->ConstraintMatrix,is_C_local,is_R_local,MAT_INITIAL_MATRIX,&C_CR);CHKERRQ(ierr);
204*88ebb749SStefano Zampini     ierr = ISDestroy(&is_C_local);CHKERRQ(ierr);
205*88ebb749SStefano Zampini 
206*88ebb749SStefano Zampini     /* Assemble local_auxmat2 = - A_{RR}^{-1} C^T_{CR} needed by BDDC application */
207*88ebb749SStefano Zampini     for (i=0;i<n_constraints;i++) {
208*88ebb749SStefano Zampini       ierr = VecSet(pcbddc->vec1_R,zero);CHKERRQ(ierr);
209*88ebb749SStefano Zampini       /* Get row of constraint matrix in R numbering */
210*88ebb749SStefano Zampini       ierr = VecGetArray(pcbddc->vec1_R,&array);CHKERRQ(ierr);
211*88ebb749SStefano Zampini       ierr = MatGetRow(C_CR,i,&size_of_constraint,(const PetscInt**)&row_cmat_indices,(const PetscScalar**)&row_cmat_values);CHKERRQ(ierr);
212*88ebb749SStefano Zampini       for (j=0;j<size_of_constraint;j++) array[row_cmat_indices[j]] = -row_cmat_values[j];
213*88ebb749SStefano Zampini       ierr = MatRestoreRow(C_CR,i,&size_of_constraint,(const PetscInt**)&row_cmat_indices,(const PetscScalar**)&row_cmat_values);CHKERRQ(ierr);
214*88ebb749SStefano Zampini       ierr = VecRestoreArray(pcbddc->vec1_R,&array);CHKERRQ(ierr);
215*88ebb749SStefano Zampini 
216*88ebb749SStefano Zampini       /* Solve for row of constraint matrix in R numbering */
217*88ebb749SStefano Zampini       ierr = KSPSolve(pcbddc->ksp_R,pcbddc->vec1_R,pcbddc->vec2_R);CHKERRQ(ierr);
218*88ebb749SStefano Zampini 
219*88ebb749SStefano Zampini       /* Set values */
220*88ebb749SStefano Zampini       ierr = VecGetArray(pcbddc->vec2_R,&array);CHKERRQ(ierr);
221*88ebb749SStefano Zampini       ierr = MatSetValues(pcbddc->local_auxmat2,n_R,auxindices,1,&i,array,INSERT_VALUES);CHKERRQ(ierr);
222*88ebb749SStefano Zampini       ierr = VecRestoreArray(pcbddc->vec2_R,&array);CHKERRQ(ierr);
223*88ebb749SStefano Zampini     }
224*88ebb749SStefano Zampini     ierr = MatAssemblyBegin(pcbddc->local_auxmat2,MAT_FINAL_ASSEMBLY);CHKERRQ(ierr);
225*88ebb749SStefano Zampini     ierr = MatAssemblyEnd(pcbddc->local_auxmat2,MAT_FINAL_ASSEMBLY);CHKERRQ(ierr);
226*88ebb749SStefano Zampini 
227*88ebb749SStefano Zampini     /* Assemble AUXMAT = ( LUFactor )( -C_{CR} A_{RR}^{-1} C^T_{CR} )^{-1} */
228*88ebb749SStefano Zampini     ierr = MatMatMult(C_CR,pcbddc->local_auxmat2,MAT_INITIAL_MATRIX,PETSC_DEFAULT,&AUXMAT);CHKERRQ(ierr);
229*88ebb749SStefano Zampini     ierr = MatFactorInfoInitialize(&matinfo);CHKERRQ(ierr);
230*88ebb749SStefano Zampini     ierr = ISCreateStride(PETSC_COMM_SELF,n_constraints,0,1,&is_aux1);CHKERRQ(ierr);
231*88ebb749SStefano Zampini     ierr = MatLUFactor(AUXMAT,is_aux1,is_aux1,&matinfo);CHKERRQ(ierr);
232*88ebb749SStefano Zampini     ierr = ISDestroy(&is_aux1);CHKERRQ(ierr);
233*88ebb749SStefano Zampini 
234*88ebb749SStefano Zampini     /* Assemble explicitly M1 = ( C_{CR} A_{RR}^{-1} C^T_{CR} )^{-1} needed in preproc  */
235*88ebb749SStefano Zampini     ierr = MatCreate(PETSC_COMM_SELF,&M1);CHKERRQ(ierr);
236*88ebb749SStefano Zampini     ierr = MatSetSizes(M1,n_constraints,n_constraints,n_constraints,n_constraints);CHKERRQ(ierr);
237*88ebb749SStefano Zampini     ierr = MatSetType(M1,impMatType);CHKERRQ(ierr);
238*88ebb749SStefano Zampini     ierr = MatSeqDenseSetPreallocation(M1,NULL);CHKERRQ(ierr);
239*88ebb749SStefano Zampini     for (i=0;i<n_constraints;i++) {
240*88ebb749SStefano Zampini       ierr = VecSet(vec1_C,zero);CHKERRQ(ierr);
241*88ebb749SStefano Zampini       ierr = VecSetValue(vec1_C,i,one,INSERT_VALUES);CHKERRQ(ierr);
242*88ebb749SStefano Zampini       ierr = VecAssemblyBegin(vec1_C);CHKERRQ(ierr);
243*88ebb749SStefano Zampini       ierr = VecAssemblyEnd(vec1_C);CHKERRQ(ierr);
244*88ebb749SStefano Zampini       ierr = MatSolve(AUXMAT,vec1_C,vec2_C);CHKERRQ(ierr);
245*88ebb749SStefano Zampini       ierr = VecScale(vec2_C,m_one);CHKERRQ(ierr);
246*88ebb749SStefano Zampini       ierr = VecGetArray(vec2_C,&array);CHKERRQ(ierr);
247*88ebb749SStefano Zampini       ierr = MatSetValues(M1,n_constraints,auxindices,1,&i,array,INSERT_VALUES);CHKERRQ(ierr);
248*88ebb749SStefano Zampini       ierr = VecRestoreArray(vec2_C,&array);CHKERRQ(ierr);
249*88ebb749SStefano Zampini     }
250*88ebb749SStefano Zampini     ierr = MatAssemblyBegin(M1,MAT_FINAL_ASSEMBLY);CHKERRQ(ierr);
251*88ebb749SStefano Zampini     ierr = MatAssemblyEnd(M1,MAT_FINAL_ASSEMBLY);CHKERRQ(ierr);
252*88ebb749SStefano Zampini     ierr = MatDestroy(&AUXMAT);CHKERRQ(ierr);
253*88ebb749SStefano Zampini     /* Assemble local_auxmat1 = M1*C_{CR} needed by BDDC application in KSP and in preproc */
254*88ebb749SStefano Zampini     ierr = MatMatMult(M1,C_CR,MAT_INITIAL_MATRIX,PETSC_DEFAULT,&pcbddc->local_auxmat1);CHKERRQ(ierr);
255*88ebb749SStefano Zampini 
256*88ebb749SStefano Zampini   }
257*88ebb749SStefano Zampini 
258*88ebb749SStefano Zampini   /* Get submatrices from subdomain matrix */
259*88ebb749SStefano Zampini   if (n_vertices) {
260*88ebb749SStefano Zampini     ierr = ISCreateGeneral(PETSC_COMM_SELF,n_vertices,vertices,PETSC_COPY_VALUES,&is_V_local);CHKERRQ(ierr);
261*88ebb749SStefano Zampini     ierr = MatGetSubMatrix(pcbddc->local_mat,is_R_local,is_V_local,MAT_INITIAL_MATRIX,&A_RV);CHKERRQ(ierr);
262*88ebb749SStefano Zampini     ierr = MatGetSubMatrix(pcbddc->local_mat,is_V_local,is_R_local,MAT_INITIAL_MATRIX,&A_VR);CHKERRQ(ierr);
263*88ebb749SStefano Zampini     ierr = MatGetSubMatrix(pcbddc->local_mat,is_V_local,is_V_local,MAT_INITIAL_MATRIX,&A_VV);CHKERRQ(ierr);
264*88ebb749SStefano Zampini     ierr = ISDestroy(&is_V_local);CHKERRQ(ierr);
265*88ebb749SStefano Zampini   }
266*88ebb749SStefano Zampini 
267*88ebb749SStefano Zampini   /* Matrix of coarse basis functions (local) */
268*88ebb749SStefano Zampini   ierr = MatCreate(PETSC_COMM_SELF,&pcbddc->coarse_phi_B);CHKERRQ(ierr);
269*88ebb749SStefano Zampini   ierr = MatSetSizes(pcbddc->coarse_phi_B,n_B,pcbddc->local_primal_size,n_B,pcbddc->local_primal_size);CHKERRQ(ierr);
270*88ebb749SStefano Zampini   ierr = MatSetType(pcbddc->coarse_phi_B,impMatType);CHKERRQ(ierr);
271*88ebb749SStefano Zampini   ierr = MatSeqDenseSetPreallocation(pcbddc->coarse_phi_B,NULL);CHKERRQ(ierr);
272*88ebb749SStefano Zampini   if (pcbddc->inexact_prec_type || dbg_flag ) {
273*88ebb749SStefano Zampini     ierr = MatCreate(PETSC_COMM_SELF,&pcbddc->coarse_phi_D);CHKERRQ(ierr);
274*88ebb749SStefano Zampini     ierr = MatSetSizes(pcbddc->coarse_phi_D,n_D,pcbddc->local_primal_size,n_D,pcbddc->local_primal_size);CHKERRQ(ierr);
275*88ebb749SStefano Zampini     ierr = MatSetType(pcbddc->coarse_phi_D,impMatType);CHKERRQ(ierr);
276*88ebb749SStefano Zampini     ierr = MatSeqDenseSetPreallocation(pcbddc->coarse_phi_D,NULL);CHKERRQ(ierr);
277*88ebb749SStefano Zampini   }
278*88ebb749SStefano Zampini 
279*88ebb749SStefano Zampini   if (dbg_flag) {
280*88ebb749SStefano Zampini     ierr = ISGetIndices(is_R_local,(const PetscInt**)&idx_R_local);CHKERRQ(ierr);
281*88ebb749SStefano Zampini     ierr = PetscMalloc(2*pcbddc->local_primal_size*sizeof(*coarsefunctions_errors),&coarsefunctions_errors);CHKERRQ(ierr);
282*88ebb749SStefano Zampini     ierr = PetscMalloc(2*pcbddc->local_primal_size*sizeof(*constraints_errors),&constraints_errors);CHKERRQ(ierr);
283*88ebb749SStefano Zampini   }
284*88ebb749SStefano Zampini   /* Subdomain contribution (Non-overlapping) to coarse matrix  */
285*88ebb749SStefano Zampini   ierr = PetscMalloc ((pcbddc->local_primal_size)*(pcbddc->local_primal_size)*sizeof(PetscScalar),&coarse_submat_vals);CHKERRQ(ierr);
286*88ebb749SStefano Zampini 
287*88ebb749SStefano Zampini   /* We are now ready to evaluate coarse basis functions and subdomain contribution to coarse problem */
288*88ebb749SStefano Zampini   for (i=0;i<n_vertices;i++){
289*88ebb749SStefano Zampini     ierr = VecSet(vec1_V,zero);CHKERRQ(ierr);
290*88ebb749SStefano Zampini     ierr = VecSetValue(vec1_V,i,one,INSERT_VALUES);CHKERRQ(ierr);
291*88ebb749SStefano Zampini     ierr = VecAssemblyBegin(vec1_V);CHKERRQ(ierr);
292*88ebb749SStefano Zampini     ierr = VecAssemblyEnd(vec1_V);CHKERRQ(ierr);
293*88ebb749SStefano Zampini     /* solution of saddle point problem */
294*88ebb749SStefano Zampini     ierr = MatMult(A_RV,vec1_V,pcbddc->vec1_R);CHKERRQ(ierr);
295*88ebb749SStefano Zampini     ierr = KSPSolve(pcbddc->ksp_R,pcbddc->vec1_R,pcbddc->vec1_R);CHKERRQ(ierr);
296*88ebb749SStefano Zampini     ierr = VecScale(pcbddc->vec1_R,m_one);CHKERRQ(ierr);
297*88ebb749SStefano Zampini     if (n_constraints) {
298*88ebb749SStefano Zampini       ierr = MatMult(pcbddc->local_auxmat1,pcbddc->vec1_R,vec1_C);CHKERRQ(ierr);
299*88ebb749SStefano Zampini       ierr = MatMultAdd(pcbddc->local_auxmat2,vec1_C,pcbddc->vec1_R,pcbddc->vec1_R);CHKERRQ(ierr);
300*88ebb749SStefano Zampini       ierr = VecScale(vec1_C,m_one);CHKERRQ(ierr);
301*88ebb749SStefano Zampini     }
302*88ebb749SStefano Zampini     ierr = MatMult(A_VR,pcbddc->vec1_R,vec2_V);CHKERRQ(ierr);
303*88ebb749SStefano Zampini     ierr = MatMultAdd(A_VV,vec1_V,vec2_V,vec2_V);CHKERRQ(ierr);
304*88ebb749SStefano Zampini 
305*88ebb749SStefano Zampini     /* Set values in coarse basis function and subdomain part of coarse_mat */
306*88ebb749SStefano Zampini     /* coarse basis functions */
307*88ebb749SStefano Zampini     ierr = VecSet(pcis->vec1_B,zero);CHKERRQ(ierr);
308*88ebb749SStefano Zampini     ierr = VecScatterBegin(pcbddc->R_to_B,pcbddc->vec1_R,pcis->vec1_B,INSERT_VALUES,SCATTER_FORWARD);CHKERRQ(ierr);
309*88ebb749SStefano Zampini     ierr = VecScatterEnd  (pcbddc->R_to_B,pcbddc->vec1_R,pcis->vec1_B,INSERT_VALUES,SCATTER_FORWARD);CHKERRQ(ierr);
310*88ebb749SStefano Zampini     ierr = VecGetArray(pcis->vec1_B,&array);CHKERRQ(ierr);
311*88ebb749SStefano Zampini     ierr = MatSetValues(pcbddc->coarse_phi_B,n_B,auxindices,1,&i,array,INSERT_VALUES);CHKERRQ(ierr);
312*88ebb749SStefano Zampini     ierr = VecRestoreArray(pcis->vec1_B,&array);CHKERRQ(ierr);
313*88ebb749SStefano Zampini     ierr = MatSetValue(pcbddc->coarse_phi_B,idx_V_B[i],i,one,INSERT_VALUES);CHKERRQ(ierr);
314*88ebb749SStefano Zampini     if ( pcbddc->inexact_prec_type || dbg_flag  ) {
315*88ebb749SStefano Zampini       ierr = VecScatterBegin(pcbddc->R_to_D,pcbddc->vec1_R,pcis->vec1_D,INSERT_VALUES,SCATTER_FORWARD);CHKERRQ(ierr);
316*88ebb749SStefano Zampini       ierr = VecScatterEnd(pcbddc->R_to_D,pcbddc->vec1_R,pcis->vec1_D,INSERT_VALUES,SCATTER_FORWARD);CHKERRQ(ierr);
317*88ebb749SStefano Zampini       ierr = VecGetArray(pcis->vec1_D,&array);CHKERRQ(ierr);
318*88ebb749SStefano Zampini       ierr = MatSetValues(pcbddc->coarse_phi_D,n_D,auxindices,1,&i,array,INSERT_VALUES);CHKERRQ(ierr);
319*88ebb749SStefano Zampini       ierr = VecRestoreArray(pcis->vec1_D,&array);CHKERRQ(ierr);
320*88ebb749SStefano Zampini     }
321*88ebb749SStefano Zampini     /* subdomain contribution to coarse matrix */
322*88ebb749SStefano Zampini     ierr = VecGetArray(vec2_V,&array);CHKERRQ(ierr);
323*88ebb749SStefano Zampini     for (j=0; j<n_vertices; j++) coarse_submat_vals[i*pcbddc->local_primal_size+j] = array[j];   /* WARNING -> column major ordering */
324*88ebb749SStefano Zampini     ierr = VecRestoreArray(vec2_V,&array);CHKERRQ(ierr);
325*88ebb749SStefano Zampini     if (n_constraints) {
326*88ebb749SStefano Zampini       ierr = VecGetArray(vec1_C,&array);CHKERRQ(ierr);
327*88ebb749SStefano Zampini       for (j=0; j<n_constraints; j++) coarse_submat_vals[i*pcbddc->local_primal_size+j+n_vertices] = array[j];   /* WARNING -> column major ordering */
328*88ebb749SStefano Zampini       ierr = VecRestoreArray(vec1_C,&array);CHKERRQ(ierr);
329*88ebb749SStefano Zampini     }
330*88ebb749SStefano Zampini 
331*88ebb749SStefano Zampini     if ( dbg_flag ) {
332*88ebb749SStefano Zampini       /* assemble subdomain vector on nodes */
333*88ebb749SStefano Zampini       ierr = VecSet(pcis->vec1_N,zero);CHKERRQ(ierr);
334*88ebb749SStefano Zampini       ierr = VecGetArray(pcis->vec1_N,&array);CHKERRQ(ierr);
335*88ebb749SStefano Zampini       ierr = VecGetArray(pcbddc->vec1_R,&array2);CHKERRQ(ierr);
336*88ebb749SStefano Zampini       for (j=0;j<n_R;j++) array[idx_R_local[j]] = array2[j];
337*88ebb749SStefano Zampini       array[ vertices[i] ] = one;
338*88ebb749SStefano Zampini       ierr = VecRestoreArray(pcbddc->vec1_R,&array2);CHKERRQ(ierr);
339*88ebb749SStefano Zampini       ierr = VecRestoreArray(pcis->vec1_N,&array);CHKERRQ(ierr);
340*88ebb749SStefano Zampini       /* assemble subdomain vector of lagrange multipliers (i.e. primal nodes) */
341*88ebb749SStefano Zampini       ierr = VecSet(pcbddc->vec1_P,zero);CHKERRQ(ierr);
342*88ebb749SStefano Zampini       ierr = VecGetArray(pcbddc->vec1_P,&array2);CHKERRQ(ierr);
343*88ebb749SStefano Zampini       ierr = VecGetArray(vec2_V,&array);CHKERRQ(ierr);
344*88ebb749SStefano Zampini       for (j=0;j<n_vertices;j++) array2[j]=array[j];
345*88ebb749SStefano Zampini       ierr = VecRestoreArray(vec2_V,&array);CHKERRQ(ierr);
346*88ebb749SStefano Zampini       if (n_constraints) {
347*88ebb749SStefano Zampini         ierr = VecGetArray(vec1_C,&array);CHKERRQ(ierr);
348*88ebb749SStefano Zampini         for (j=0;j<n_constraints;j++) array2[j+n_vertices]=array[j];
349*88ebb749SStefano Zampini         ierr = VecRestoreArray(vec1_C,&array);CHKERRQ(ierr);
350*88ebb749SStefano Zampini       }
351*88ebb749SStefano Zampini       ierr = VecRestoreArray(pcbddc->vec1_P,&array2);CHKERRQ(ierr);
352*88ebb749SStefano Zampini       ierr = VecScale(pcbddc->vec1_P,m_one);CHKERRQ(ierr);
353*88ebb749SStefano Zampini       /* check saddle point solution */
354*88ebb749SStefano Zampini       ierr = MatMult(pcbddc->local_mat,pcis->vec1_N,pcis->vec2_N);CHKERRQ(ierr);
355*88ebb749SStefano Zampini       ierr = MatMultTransposeAdd(pcbddc->ConstraintMatrix,pcbddc->vec1_P,pcis->vec2_N,pcis->vec2_N);CHKERRQ(ierr);
356*88ebb749SStefano Zampini       ierr = VecNorm(pcis->vec2_N,NORM_INFINITY,&coarsefunctions_errors[i]);CHKERRQ(ierr);
357*88ebb749SStefano Zampini       ierr = MatMult(pcbddc->ConstraintMatrix,pcis->vec1_N,pcbddc->vec1_P);CHKERRQ(ierr);
358*88ebb749SStefano Zampini       ierr = VecGetArray(pcbddc->vec1_P,&array);CHKERRQ(ierr);
359*88ebb749SStefano Zampini       array[i]=array[i]+m_one;  /* shift by the identity matrix */
360*88ebb749SStefano Zampini       ierr = VecRestoreArray(pcbddc->vec1_P,&array);CHKERRQ(ierr);
361*88ebb749SStefano Zampini       ierr = VecNorm(pcbddc->vec1_P,NORM_INFINITY,&constraints_errors[i]);CHKERRQ(ierr);
362*88ebb749SStefano Zampini     }
363*88ebb749SStefano Zampini   }
364*88ebb749SStefano Zampini 
365*88ebb749SStefano Zampini   for (i=0;i<n_constraints;i++){
366*88ebb749SStefano Zampini     ierr = VecSet(vec2_C,zero);CHKERRQ(ierr);
367*88ebb749SStefano Zampini     ierr = VecSetValue(vec2_C,i,m_one,INSERT_VALUES);CHKERRQ(ierr);
368*88ebb749SStefano Zampini     ierr = VecAssemblyBegin(vec2_C);CHKERRQ(ierr);
369*88ebb749SStefano Zampini     ierr = VecAssemblyEnd(vec2_C);CHKERRQ(ierr);
370*88ebb749SStefano Zampini     /* solution of saddle point problem */
371*88ebb749SStefano Zampini     ierr = MatMult(M1,vec2_C,vec1_C);CHKERRQ(ierr);
372*88ebb749SStefano Zampini     ierr = MatMult(pcbddc->local_auxmat2,vec1_C,pcbddc->vec1_R);CHKERRQ(ierr);
373*88ebb749SStefano Zampini     ierr = VecScale(vec1_C,m_one);CHKERRQ(ierr);
374*88ebb749SStefano Zampini     if (n_vertices) { ierr = MatMult(A_VR,pcbddc->vec1_R,vec2_V);CHKERRQ(ierr); }
375*88ebb749SStefano Zampini     /* Set values in coarse basis function and subdomain part of coarse_mat */
376*88ebb749SStefano Zampini     /* coarse basis functions */
377*88ebb749SStefano Zampini     index=i+n_vertices;
378*88ebb749SStefano Zampini     ierr = VecSet(pcis->vec1_B,zero);CHKERRQ(ierr);
379*88ebb749SStefano Zampini     ierr = VecScatterBegin(pcbddc->R_to_B,pcbddc->vec1_R,pcis->vec1_B,INSERT_VALUES,SCATTER_FORWARD);CHKERRQ(ierr);
380*88ebb749SStefano Zampini     ierr = VecScatterEnd(pcbddc->R_to_B,pcbddc->vec1_R,pcis->vec1_B,INSERT_VALUES,SCATTER_FORWARD);CHKERRQ(ierr);
381*88ebb749SStefano Zampini     ierr = VecGetArray(pcis->vec1_B,&array);CHKERRQ(ierr);
382*88ebb749SStefano Zampini     ierr = MatSetValues(pcbddc->coarse_phi_B,n_B,auxindices,1,&index,array,INSERT_VALUES);CHKERRQ(ierr);
383*88ebb749SStefano Zampini     ierr = VecRestoreArray(pcis->vec1_B,&array);CHKERRQ(ierr);
384*88ebb749SStefano Zampini     if ( pcbddc->inexact_prec_type || dbg_flag ) {
385*88ebb749SStefano Zampini       ierr = VecScatterBegin(pcbddc->R_to_D,pcbddc->vec1_R,pcis->vec1_D,INSERT_VALUES,SCATTER_FORWARD);CHKERRQ(ierr);
386*88ebb749SStefano Zampini       ierr = VecScatterEnd(pcbddc->R_to_D,pcbddc->vec1_R,pcis->vec1_D,INSERT_VALUES,SCATTER_FORWARD);CHKERRQ(ierr);
387*88ebb749SStefano Zampini       ierr = VecGetArray(pcis->vec1_D,&array);CHKERRQ(ierr);
388*88ebb749SStefano Zampini       ierr = MatSetValues(pcbddc->coarse_phi_D,n_D,auxindices,1,&index,array,INSERT_VALUES);CHKERRQ(ierr);
389*88ebb749SStefano Zampini       ierr = VecRestoreArray(pcis->vec1_D,&array);CHKERRQ(ierr);
390*88ebb749SStefano Zampini     }
391*88ebb749SStefano Zampini     /* subdomain contribution to coarse matrix */
392*88ebb749SStefano Zampini     if (n_vertices) {
393*88ebb749SStefano Zampini       ierr = VecGetArray(vec2_V,&array);CHKERRQ(ierr);
394*88ebb749SStefano Zampini       for (j=0; j<n_vertices; j++) coarse_submat_vals[index*pcbddc->local_primal_size+j]=array[j]; /* WARNING -> column major ordering */
395*88ebb749SStefano Zampini       ierr = VecRestoreArray(vec2_V,&array);CHKERRQ(ierr);
396*88ebb749SStefano Zampini     }
397*88ebb749SStefano Zampini     ierr = VecGetArray(vec1_C,&array);CHKERRQ(ierr);
398*88ebb749SStefano Zampini     for (j=0; j<n_constraints; j++) coarse_submat_vals[index*pcbddc->local_primal_size+j+n_vertices]=array[j]; /* WARNING -> column major ordering */
399*88ebb749SStefano Zampini     ierr = VecRestoreArray(vec1_C,&array);CHKERRQ(ierr);
400*88ebb749SStefano Zampini 
401*88ebb749SStefano Zampini     if ( dbg_flag ) {
402*88ebb749SStefano Zampini       /* assemble subdomain vector on nodes */
403*88ebb749SStefano Zampini       ierr = VecSet(pcis->vec1_N,zero);CHKERRQ(ierr);
404*88ebb749SStefano Zampini       ierr = VecGetArray(pcis->vec1_N,&array);CHKERRQ(ierr);
405*88ebb749SStefano Zampini       ierr = VecGetArray(pcbddc->vec1_R,&array2);CHKERRQ(ierr);
406*88ebb749SStefano Zampini       for (j=0;j<n_R;j++) array[idx_R_local[j]] = array2[j];
407*88ebb749SStefano Zampini       ierr = VecRestoreArray(pcbddc->vec1_R,&array2);CHKERRQ(ierr);
408*88ebb749SStefano Zampini       ierr = VecRestoreArray(pcis->vec1_N,&array);CHKERRQ(ierr);
409*88ebb749SStefano Zampini       /* assemble subdomain vector of lagrange multipliers */
410*88ebb749SStefano Zampini       ierr = VecSet(pcbddc->vec1_P,zero);CHKERRQ(ierr);
411*88ebb749SStefano Zampini       ierr = VecGetArray(pcbddc->vec1_P,&array2);CHKERRQ(ierr);
412*88ebb749SStefano Zampini       if ( n_vertices) {
413*88ebb749SStefano Zampini         ierr = VecGetArray(vec2_V,&array);CHKERRQ(ierr);
414*88ebb749SStefano Zampini         for (j=0;j<n_vertices;j++) array2[j]=-array[j];
415*88ebb749SStefano Zampini         ierr = VecRestoreArray(vec2_V,&array);CHKERRQ(ierr);
416*88ebb749SStefano Zampini       }
417*88ebb749SStefano Zampini       ierr = VecGetArray(vec1_C,&array);CHKERRQ(ierr);
418*88ebb749SStefano Zampini       for (j=0;j<n_constraints;j++) {array2[j+n_vertices]=-array[j];}
419*88ebb749SStefano Zampini       ierr = VecRestoreArray(vec1_C,&array);CHKERRQ(ierr);
420*88ebb749SStefano Zampini       ierr = VecRestoreArray(pcbddc->vec1_P,&array2);CHKERRQ(ierr);
421*88ebb749SStefano Zampini       /* check saddle point solution */
422*88ebb749SStefano Zampini       ierr = MatMult(pcbddc->local_mat,pcis->vec1_N,pcis->vec2_N);CHKERRQ(ierr);
423*88ebb749SStefano Zampini       ierr = MatMultTransposeAdd(pcbddc->ConstraintMatrix,pcbddc->vec1_P,pcis->vec2_N,pcis->vec2_N);CHKERRQ(ierr);
424*88ebb749SStefano Zampini       ierr = VecNorm(pcis->vec2_N,NORM_INFINITY,&coarsefunctions_errors[index]);CHKERRQ(ierr);
425*88ebb749SStefano Zampini       ierr = MatMult(pcbddc->ConstraintMatrix,pcis->vec1_N,pcbddc->vec1_P);CHKERRQ(ierr);
426*88ebb749SStefano Zampini       ierr = VecGetArray(pcbddc->vec1_P,&array);CHKERRQ(ierr);
427*88ebb749SStefano Zampini       array[index]=array[index]+m_one; /* shift by the identity matrix */
428*88ebb749SStefano Zampini       ierr = VecRestoreArray(pcbddc->vec1_P,&array);CHKERRQ(ierr);
429*88ebb749SStefano Zampini       ierr = VecNorm(pcbddc->vec1_P,NORM_INFINITY,&constraints_errors[index]);CHKERRQ(ierr);
430*88ebb749SStefano Zampini     }
431*88ebb749SStefano Zampini   }
432*88ebb749SStefano Zampini   ierr = MatAssemblyBegin(pcbddc->coarse_phi_B,MAT_FINAL_ASSEMBLY);CHKERRQ(ierr);
433*88ebb749SStefano Zampini   ierr = MatAssemblyEnd(pcbddc->coarse_phi_B,MAT_FINAL_ASSEMBLY);CHKERRQ(ierr);
434*88ebb749SStefano Zampini   if ( pcbddc->inexact_prec_type || dbg_flag ) {
435*88ebb749SStefano Zampini     ierr = MatAssemblyBegin(pcbddc->coarse_phi_D,MAT_FINAL_ASSEMBLY);CHKERRQ(ierr);
436*88ebb749SStefano Zampini     ierr = MatAssemblyEnd(pcbddc->coarse_phi_D,MAT_FINAL_ASSEMBLY);CHKERRQ(ierr);
437*88ebb749SStefano Zampini   }
438*88ebb749SStefano Zampini   /* compute other basis functions for non-symmetric problems */
439*88ebb749SStefano Zampini   ierr = MatIsSymmetricKnown(pc->pmat,&setsym,&issym);CHKERRQ(ierr);
440*88ebb749SStefano Zampini   if ( !setsym || (setsym && !issym) ) {
441*88ebb749SStefano Zampini     ierr = MatCreate(PETSC_COMM_SELF,&pcbddc->coarse_psi_B);CHKERRQ(ierr);
442*88ebb749SStefano Zampini     ierr = MatSetSizes(pcbddc->coarse_psi_B,n_B,pcbddc->local_primal_size,n_B,pcbddc->local_primal_size);CHKERRQ(ierr);
443*88ebb749SStefano Zampini     ierr = MatSetType(pcbddc->coarse_psi_B,impMatType);CHKERRQ(ierr);
444*88ebb749SStefano Zampini     ierr = MatSeqDenseSetPreallocation(pcbddc->coarse_psi_B,NULL);CHKERRQ(ierr);
445*88ebb749SStefano Zampini     if (pcbddc->inexact_prec_type || dbg_flag ) {
446*88ebb749SStefano Zampini       ierr = MatCreate(PETSC_COMM_SELF,&pcbddc->coarse_psi_D);CHKERRQ(ierr);
447*88ebb749SStefano Zampini       ierr = MatSetSizes(pcbddc->coarse_psi_D,n_D,pcbddc->local_primal_size,n_D,pcbddc->local_primal_size);CHKERRQ(ierr);
448*88ebb749SStefano Zampini       ierr = MatSetType(pcbddc->coarse_psi_D,impMatType);CHKERRQ(ierr);
449*88ebb749SStefano Zampini       ierr = MatSeqDenseSetPreallocation(pcbddc->coarse_psi_D,NULL);CHKERRQ(ierr);
450*88ebb749SStefano Zampini     }
451*88ebb749SStefano Zampini     for (i=0;i<pcbddc->local_primal_size;i++) {
452*88ebb749SStefano Zampini       if (n_constraints) {
453*88ebb749SStefano Zampini         ierr = VecSet(vec1_C,zero);CHKERRQ(ierr);
454*88ebb749SStefano Zampini         ierr = VecGetArray(vec1_C,&array);CHKERRQ(ierr);
455*88ebb749SStefano Zampini         for (j=0;j<n_constraints;j++) {
456*88ebb749SStefano Zampini           array[j]=coarse_submat_vals[(j+n_vertices)*pcbddc->local_primal_size+i];
457*88ebb749SStefano Zampini         }
458*88ebb749SStefano Zampini         ierr = VecRestoreArray(vec1_C,&array);CHKERRQ(ierr);
459*88ebb749SStefano Zampini       }
460*88ebb749SStefano Zampini       if (i<n_vertices) {
461*88ebb749SStefano Zampini         ierr = VecSet(vec1_V,zero);CHKERRQ(ierr);
462*88ebb749SStefano Zampini         ierr = VecSetValue(vec1_V,i,m_one,INSERT_VALUES);CHKERRQ(ierr);
463*88ebb749SStefano Zampini         ierr = VecAssemblyBegin(vec1_V);CHKERRQ(ierr);
464*88ebb749SStefano Zampini         ierr = VecAssemblyEnd(vec1_V);CHKERRQ(ierr);
465*88ebb749SStefano Zampini         ierr = MatMultTranspose(A_VR,vec1_V,pcbddc->vec1_R);CHKERRQ(ierr);
466*88ebb749SStefano Zampini         if (n_constraints) {
467*88ebb749SStefano Zampini           ierr = MatMultTransposeAdd(C_CR,vec1_C,pcbddc->vec1_R,pcbddc->vec1_R);CHKERRQ(ierr);
468*88ebb749SStefano Zampini         }
469*88ebb749SStefano Zampini       } else {
470*88ebb749SStefano Zampini         ierr = MatMultTranspose(C_CR,vec1_C,pcbddc->vec1_R);CHKERRQ(ierr);
471*88ebb749SStefano Zampini       }
472*88ebb749SStefano Zampini       ierr = KSPSolveTranspose(pcbddc->ksp_R,pcbddc->vec1_R,pcbddc->vec1_R);CHKERRQ(ierr);
473*88ebb749SStefano Zampini       ierr = VecSet(pcis->vec1_B,zero);CHKERRQ(ierr);
474*88ebb749SStefano Zampini       ierr = VecScatterBegin(pcbddc->R_to_B,pcbddc->vec1_R,pcis->vec1_B,INSERT_VALUES,SCATTER_FORWARD);CHKERRQ(ierr);
475*88ebb749SStefano Zampini       ierr = VecScatterEnd(pcbddc->R_to_B,pcbddc->vec1_R,pcis->vec1_B,INSERT_VALUES,SCATTER_FORWARD);CHKERRQ(ierr);
476*88ebb749SStefano Zampini       ierr = VecGetArray(pcis->vec1_B,&array);CHKERRQ(ierr);
477*88ebb749SStefano Zampini       ierr = MatSetValues(pcbddc->coarse_psi_B,n_B,auxindices,1,&i,array,INSERT_VALUES);CHKERRQ(ierr);
478*88ebb749SStefano Zampini       ierr = VecRestoreArray(pcis->vec1_B,&array);CHKERRQ(ierr);
479*88ebb749SStefano Zampini       if (i<n_vertices) {
480*88ebb749SStefano Zampini         ierr = MatSetValue(pcbddc->coarse_psi_B,idx_V_B[i],i,one,INSERT_VALUES);CHKERRQ(ierr);
481*88ebb749SStefano Zampini       }
482*88ebb749SStefano Zampini       if ( pcbddc->inexact_prec_type || dbg_flag  ) {
483*88ebb749SStefano Zampini         ierr = VecScatterBegin(pcbddc->R_to_D,pcbddc->vec1_R,pcis->vec1_D,INSERT_VALUES,SCATTER_FORWARD);CHKERRQ(ierr);
484*88ebb749SStefano Zampini         ierr = VecScatterEnd(pcbddc->R_to_D,pcbddc->vec1_R,pcis->vec1_D,INSERT_VALUES,SCATTER_FORWARD);CHKERRQ(ierr);
485*88ebb749SStefano Zampini         ierr = VecGetArray(pcis->vec1_D,&array);CHKERRQ(ierr);
486*88ebb749SStefano Zampini         ierr = MatSetValues(pcbddc->coarse_psi_D,n_D,auxindices,1,&i,array,INSERT_VALUES);CHKERRQ(ierr);
487*88ebb749SStefano Zampini         ierr = VecRestoreArray(pcis->vec1_D,&array);CHKERRQ(ierr);
488*88ebb749SStefano Zampini       }
489*88ebb749SStefano Zampini 
490*88ebb749SStefano Zampini       if ( dbg_flag ) {
491*88ebb749SStefano Zampini         /* assemble subdomain vector on nodes */
492*88ebb749SStefano Zampini         ierr = VecSet(pcis->vec1_N,zero);CHKERRQ(ierr);
493*88ebb749SStefano Zampini         ierr = VecGetArray(pcis->vec1_N,&array);CHKERRQ(ierr);
494*88ebb749SStefano Zampini         ierr = VecGetArray(pcbddc->vec1_R,&array2);CHKERRQ(ierr);
495*88ebb749SStefano Zampini         for (j=0;j<n_R;j++) array[idx_R_local[j]] = array2[j];
496*88ebb749SStefano Zampini         if (i<n_vertices) array[vertices[i]] = one;
497*88ebb749SStefano Zampini         ierr = VecRestoreArray(pcbddc->vec1_R,&array2);CHKERRQ(ierr);
498*88ebb749SStefano Zampini         ierr = VecRestoreArray(pcis->vec1_N,&array);CHKERRQ(ierr);
499*88ebb749SStefano Zampini         /* assemble subdomain vector of lagrange multipliers */
500*88ebb749SStefano Zampini         ierr = VecGetArray(pcbddc->vec1_P,&array);CHKERRQ(ierr);
501*88ebb749SStefano Zampini         for (j=0;j<pcbddc->local_primal_size;j++) {
502*88ebb749SStefano Zampini           array[j]=-coarse_submat_vals[j*pcbddc->local_primal_size+i];
503*88ebb749SStefano Zampini         }
504*88ebb749SStefano Zampini         ierr = VecRestoreArray(pcbddc->vec1_P,&array);CHKERRQ(ierr);
505*88ebb749SStefano Zampini         /* check saddle point solution */
506*88ebb749SStefano Zampini         ierr = MatMultTranspose(pcbddc->local_mat,pcis->vec1_N,pcis->vec2_N);CHKERRQ(ierr);
507*88ebb749SStefano Zampini         ierr = MatMultTransposeAdd(pcbddc->ConstraintMatrix,pcbddc->vec1_P,pcis->vec2_N,pcis->vec2_N);CHKERRQ(ierr);
508*88ebb749SStefano Zampini         ierr = VecNorm(pcis->vec2_N,NORM_INFINITY,&coarsefunctions_errors[i+pcbddc->local_primal_size]);CHKERRQ(ierr);
509*88ebb749SStefano Zampini         ierr = MatMult(pcbddc->ConstraintMatrix,pcis->vec1_N,pcbddc->vec1_P);CHKERRQ(ierr);
510*88ebb749SStefano Zampini         ierr = VecGetArray(pcbddc->vec1_P,&array);CHKERRQ(ierr);
511*88ebb749SStefano Zampini         array[i]=array[i]+m_one; /* shift by the identity matrix */
512*88ebb749SStefano Zampini         ierr = VecRestoreArray(pcbddc->vec1_P,&array);CHKERRQ(ierr);
513*88ebb749SStefano Zampini         ierr = VecNorm(pcbddc->vec1_P,NORM_INFINITY,&constraints_errors[i+pcbddc->local_primal_size]);CHKERRQ(ierr);
514*88ebb749SStefano Zampini       }
515*88ebb749SStefano Zampini     }
516*88ebb749SStefano Zampini     ierr = MatAssemblyBegin(pcbddc->coarse_psi_B,MAT_FINAL_ASSEMBLY);CHKERRQ(ierr);
517*88ebb749SStefano Zampini     ierr = MatAssemblyEnd(pcbddc->coarse_psi_B,MAT_FINAL_ASSEMBLY);CHKERRQ(ierr);
518*88ebb749SStefano Zampini     if ( pcbddc->inexact_prec_type || dbg_flag ) {
519*88ebb749SStefano Zampini       ierr = MatAssemblyBegin(pcbddc->coarse_psi_D,MAT_FINAL_ASSEMBLY);CHKERRQ(ierr);
520*88ebb749SStefano Zampini       ierr = MatAssemblyEnd(pcbddc->coarse_psi_D,MAT_FINAL_ASSEMBLY);CHKERRQ(ierr);
521*88ebb749SStefano Zampini     }
522*88ebb749SStefano Zampini   }
523*88ebb749SStefano Zampini   ierr = PetscFree(idx_V_B);CHKERRQ(ierr);
524*88ebb749SStefano Zampini   /* Checking coarse_sub_mat and coarse basis functios */
525*88ebb749SStefano Zampini   /* Symmetric case     : It should be \Phi^{(j)^T} A^{(j)} \Phi^{(j)}=coarse_sub_mat */
526*88ebb749SStefano Zampini   /* Non-symmetric case : It should be \Psi^{(j)^T} A^{(j)} \Phi^{(j)}=coarse_sub_mat */
527*88ebb749SStefano Zampini   if (dbg_flag) {
528*88ebb749SStefano Zampini     Mat         coarse_sub_mat;
529*88ebb749SStefano Zampini     Mat         TM1,TM2,TM3,TM4;
530*88ebb749SStefano Zampini     Mat         coarse_phi_D,coarse_phi_B;
531*88ebb749SStefano Zampini     Mat         coarse_psi_D,coarse_psi_B;
532*88ebb749SStefano Zampini     Mat         A_II,A_BB,A_IB,A_BI;
533*88ebb749SStefano Zampini     MatType     checkmattype=MATSEQAIJ;
534*88ebb749SStefano Zampini     PetscReal   real_value;
535*88ebb749SStefano Zampini 
536*88ebb749SStefano Zampini     ierr = MatConvert(pcis->A_II,checkmattype,MAT_INITIAL_MATRIX,&A_II);CHKERRQ(ierr);
537*88ebb749SStefano Zampini     ierr = MatConvert(pcis->A_IB,checkmattype,MAT_INITIAL_MATRIX,&A_IB);CHKERRQ(ierr);
538*88ebb749SStefano Zampini     ierr = MatConvert(pcis->A_BI,checkmattype,MAT_INITIAL_MATRIX,&A_BI);CHKERRQ(ierr);
539*88ebb749SStefano Zampini     ierr = MatConvert(pcis->A_BB,checkmattype,MAT_INITIAL_MATRIX,&A_BB);CHKERRQ(ierr);
540*88ebb749SStefano Zampini     ierr = MatConvert(pcbddc->coarse_phi_D,checkmattype,MAT_INITIAL_MATRIX,&coarse_phi_D);CHKERRQ(ierr);
541*88ebb749SStefano Zampini     ierr = MatConvert(pcbddc->coarse_phi_B,checkmattype,MAT_INITIAL_MATRIX,&coarse_phi_B);CHKERRQ(ierr);
542*88ebb749SStefano Zampini     if (pcbddc->coarse_psi_B) {
543*88ebb749SStefano Zampini       ierr = MatConvert(pcbddc->coarse_psi_D,checkmattype,MAT_INITIAL_MATRIX,&coarse_psi_D);CHKERRQ(ierr);
544*88ebb749SStefano Zampini       ierr = MatConvert(pcbddc->coarse_psi_B,checkmattype,MAT_INITIAL_MATRIX,&coarse_psi_B);CHKERRQ(ierr);
545*88ebb749SStefano Zampini     }
546*88ebb749SStefano Zampini     ierr = MatCreateSeqDense(PETSC_COMM_SELF,pcbddc->local_primal_size,pcbddc->local_primal_size,coarse_submat_vals,&coarse_sub_mat);CHKERRQ(ierr);
547*88ebb749SStefano Zampini 
548*88ebb749SStefano Zampini     ierr = PetscViewerASCIIPrintf(viewer,"--------------------------------------------------\n");CHKERRQ(ierr);
549*88ebb749SStefano Zampini     ierr = PetscViewerASCIIPrintf(viewer,"Check coarse sub mat and local basis functions\n");CHKERRQ(ierr);
550*88ebb749SStefano Zampini     ierr = PetscViewerFlush(viewer);CHKERRQ(ierr);
551*88ebb749SStefano Zampini     if (pcbddc->coarse_psi_B) {
552*88ebb749SStefano Zampini       ierr = MatMatMult(A_II,coarse_phi_D,MAT_INITIAL_MATRIX,1.0,&AUXMAT);CHKERRQ(ierr);
553*88ebb749SStefano Zampini       ierr = MatTransposeMatMult(coarse_psi_D,AUXMAT,MAT_INITIAL_MATRIX,1.0,&TM1);CHKERRQ(ierr);
554*88ebb749SStefano Zampini       ierr = MatDestroy(&AUXMAT);CHKERRQ(ierr);
555*88ebb749SStefano Zampini       ierr = MatMatMult(A_BB,coarse_phi_B,MAT_INITIAL_MATRIX,1.0,&AUXMAT);CHKERRQ(ierr);
556*88ebb749SStefano Zampini       ierr = MatTransposeMatMult(coarse_psi_B,AUXMAT,MAT_INITIAL_MATRIX,1.0,&TM2);CHKERRQ(ierr);
557*88ebb749SStefano Zampini       ierr = MatDestroy(&AUXMAT);CHKERRQ(ierr);
558*88ebb749SStefano Zampini       ierr = MatMatMult(A_IB,coarse_phi_B,MAT_INITIAL_MATRIX,1.0,&AUXMAT);CHKERRQ(ierr);
559*88ebb749SStefano Zampini       ierr = MatTransposeMatMult(coarse_psi_D,AUXMAT,MAT_INITIAL_MATRIX,1.0,&TM3);CHKERRQ(ierr);
560*88ebb749SStefano Zampini       ierr = MatDestroy(&AUXMAT);CHKERRQ(ierr);
561*88ebb749SStefano Zampini       ierr = MatMatMult(A_BI,coarse_phi_D,MAT_INITIAL_MATRIX,1.0,&AUXMAT);CHKERRQ(ierr);
562*88ebb749SStefano Zampini       ierr = MatTransposeMatMult(coarse_psi_B,AUXMAT,MAT_INITIAL_MATRIX,1.0,&TM4);CHKERRQ(ierr);
563*88ebb749SStefano Zampini       ierr = MatDestroy(&AUXMAT);CHKERRQ(ierr);
564*88ebb749SStefano Zampini     } else {
565*88ebb749SStefano Zampini       ierr = MatPtAP(A_II,coarse_phi_D,MAT_INITIAL_MATRIX,1.0,&TM1);CHKERRQ(ierr);
566*88ebb749SStefano Zampini       ierr = MatPtAP(A_BB,coarse_phi_B,MAT_INITIAL_MATRIX,1.0,&TM2);CHKERRQ(ierr);
567*88ebb749SStefano Zampini       ierr = MatMatMult(A_IB,coarse_phi_B,MAT_INITIAL_MATRIX,1.0,&AUXMAT);CHKERRQ(ierr);
568*88ebb749SStefano Zampini       ierr = MatTransposeMatMult(coarse_phi_D,AUXMAT,MAT_INITIAL_MATRIX,1.0,&TM3);CHKERRQ(ierr);
569*88ebb749SStefano Zampini       ierr = MatDestroy(&AUXMAT);CHKERRQ(ierr);
570*88ebb749SStefano Zampini       ierr = MatMatMult(A_BI,coarse_phi_D,MAT_INITIAL_MATRIX,1.0,&AUXMAT);CHKERRQ(ierr);
571*88ebb749SStefano Zampini       ierr = MatTransposeMatMult(coarse_phi_B,AUXMAT,MAT_INITIAL_MATRIX,1.0,&TM4);CHKERRQ(ierr);
572*88ebb749SStefano Zampini       ierr = MatDestroy(&AUXMAT);CHKERRQ(ierr);
573*88ebb749SStefano Zampini     }
574*88ebb749SStefano Zampini     ierr = MatAXPY(TM1,one,TM2,DIFFERENT_NONZERO_PATTERN);CHKERRQ(ierr);
575*88ebb749SStefano Zampini     ierr = MatAXPY(TM1,one,TM3,DIFFERENT_NONZERO_PATTERN);CHKERRQ(ierr);
576*88ebb749SStefano Zampini     ierr = MatAXPY(TM1,one,TM4,DIFFERENT_NONZERO_PATTERN);CHKERRQ(ierr);
577*88ebb749SStefano Zampini     ierr = MatConvert(TM1,MATSEQDENSE,MAT_REUSE_MATRIX,&TM1);CHKERRQ(ierr);
578*88ebb749SStefano Zampini     ierr = MatAXPY(TM1,m_one,coarse_sub_mat,SAME_NONZERO_PATTERN);CHKERRQ(ierr);
579*88ebb749SStefano Zampini     ierr = MatNorm(TM1,NORM_INFINITY,&real_value);CHKERRQ(ierr);
580*88ebb749SStefano Zampini     ierr = PetscViewerASCIISynchronizedPrintf(viewer,"----------------------------------\n");CHKERRQ(ierr);
581*88ebb749SStefano Zampini     ierr = PetscViewerASCIISynchronizedPrintf(viewer,"Subdomain %04d \n",PetscGlobalRank);CHKERRQ(ierr);
582*88ebb749SStefano Zampini     ierr = PetscViewerASCIISynchronizedPrintf(viewer,"matrix error = % 1.14e\n",real_value);CHKERRQ(ierr);
583*88ebb749SStefano Zampini     ierr = PetscViewerASCIISynchronizedPrintf(viewer,"coarse functions (phi) errors\n");CHKERRQ(ierr);
584*88ebb749SStefano Zampini     for (i=0;i<pcbddc->local_primal_size;i++) {
585*88ebb749SStefano Zampini       ierr = PetscViewerASCIISynchronizedPrintf(viewer,"local %02d-th function error = % 1.14e\n",i,coarsefunctions_errors[i]);CHKERRQ(ierr);
586*88ebb749SStefano Zampini     }
587*88ebb749SStefano Zampini     ierr = PetscViewerASCIISynchronizedPrintf(viewer,"constraints (phi) errors\n");CHKERRQ(ierr);
588*88ebb749SStefano Zampini     for (i=0;i<pcbddc->local_primal_size;i++) {
589*88ebb749SStefano Zampini       ierr = PetscViewerASCIISynchronizedPrintf(viewer,"local %02d-th function error = % 1.14e\n",i,constraints_errors[i]);CHKERRQ(ierr);
590*88ebb749SStefano Zampini     }
591*88ebb749SStefano Zampini     if (pcbddc->coarse_psi_B) {
592*88ebb749SStefano Zampini       ierr = PetscViewerASCIISynchronizedPrintf(viewer,"coarse functions (psi) errors\n");CHKERRQ(ierr);
593*88ebb749SStefano Zampini       for (i=pcbddc->local_primal_size;i<2*pcbddc->local_primal_size;i++) {
594*88ebb749SStefano Zampini         ierr = PetscViewerASCIISynchronizedPrintf(viewer,"local %02d-th function error = % 1.14e\n",i-pcbddc->local_primal_size,coarsefunctions_errors[i]);CHKERRQ(ierr);
595*88ebb749SStefano Zampini       }
596*88ebb749SStefano Zampini       ierr = PetscViewerASCIISynchronizedPrintf(viewer,"constraints (psi) errors\n");CHKERRQ(ierr);
597*88ebb749SStefano Zampini       for (i=pcbddc->local_primal_size;i<2*pcbddc->local_primal_size;i++) {
598*88ebb749SStefano Zampini         ierr = PetscViewerASCIISynchronizedPrintf(viewer,"local %02d-th function error = % 1.14e\n",i-pcbddc->local_primal_size,constraints_errors[i]);CHKERRQ(ierr);
599*88ebb749SStefano Zampini       }
600*88ebb749SStefano Zampini     }
601*88ebb749SStefano Zampini     ierr = PetscViewerFlush(viewer);CHKERRQ(ierr);
602*88ebb749SStefano Zampini     ierr = MatDestroy(&A_II);CHKERRQ(ierr);
603*88ebb749SStefano Zampini     ierr = MatDestroy(&A_BB);CHKERRQ(ierr);
604*88ebb749SStefano Zampini     ierr = MatDestroy(&A_IB);CHKERRQ(ierr);
605*88ebb749SStefano Zampini     ierr = MatDestroy(&A_BI);CHKERRQ(ierr);
606*88ebb749SStefano Zampini     ierr = MatDestroy(&TM1);CHKERRQ(ierr);
607*88ebb749SStefano Zampini     ierr = MatDestroy(&TM2);CHKERRQ(ierr);
608*88ebb749SStefano Zampini     ierr = MatDestroy(&TM3);CHKERRQ(ierr);
609*88ebb749SStefano Zampini     ierr = MatDestroy(&TM4);CHKERRQ(ierr);
610*88ebb749SStefano Zampini     ierr = MatDestroy(&coarse_phi_D);CHKERRQ(ierr);
611*88ebb749SStefano Zampini     ierr = MatDestroy(&coarse_phi_B);CHKERRQ(ierr);
612*88ebb749SStefano Zampini     if (pcbddc->coarse_psi_B) {
613*88ebb749SStefano Zampini       ierr = MatDestroy(&coarse_psi_D);CHKERRQ(ierr);
614*88ebb749SStefano Zampini       ierr = MatDestroy(&coarse_psi_B);CHKERRQ(ierr);
615*88ebb749SStefano Zampini     }
616*88ebb749SStefano Zampini     ierr = MatDestroy(&coarse_sub_mat);CHKERRQ(ierr);
617*88ebb749SStefano Zampini     ierr = ISRestoreIndices(is_R_local,(const PetscInt**)&idx_R_local);CHKERRQ(ierr);
618*88ebb749SStefano Zampini     ierr = PetscFree(coarsefunctions_errors);CHKERRQ(ierr);
619*88ebb749SStefano Zampini     ierr = PetscFree(constraints_errors);CHKERRQ(ierr);
620*88ebb749SStefano Zampini   }
621*88ebb749SStefano Zampini   /* free memory */
622*88ebb749SStefano Zampini   if (n_vertices) {
623*88ebb749SStefano Zampini     ierr = PetscFree(vertices);CHKERRQ(ierr);
624*88ebb749SStefano Zampini     ierr = VecDestroy(&vec1_V);CHKERRQ(ierr);
625*88ebb749SStefano Zampini     ierr = VecDestroy(&vec2_V);CHKERRQ(ierr);
626*88ebb749SStefano Zampini     ierr = MatDestroy(&A_RV);CHKERRQ(ierr);
627*88ebb749SStefano Zampini     ierr = MatDestroy(&A_VR);CHKERRQ(ierr);
628*88ebb749SStefano Zampini     ierr = MatDestroy(&A_VV);CHKERRQ(ierr);
629*88ebb749SStefano Zampini   }
630*88ebb749SStefano Zampini   if (n_constraints) {
631*88ebb749SStefano Zampini     ierr = VecDestroy(&vec1_C);CHKERRQ(ierr);
632*88ebb749SStefano Zampini     ierr = VecDestroy(&vec2_C);CHKERRQ(ierr);
633*88ebb749SStefano Zampini     ierr = MatDestroy(&M1);CHKERRQ(ierr);
634*88ebb749SStefano Zampini     ierr = MatDestroy(&C_CR);CHKERRQ(ierr);
635*88ebb749SStefano Zampini   }
636*88ebb749SStefano Zampini   ierr = PetscFree(auxindices);CHKERRQ(ierr);
637*88ebb749SStefano Zampini   ierr = PetscFree(nnz);CHKERRQ(ierr);
638*88ebb749SStefano Zampini   /* create coarse matrix and data structures for message passing associated actual choice of coarse problem type */
639*88ebb749SStefano Zampini   ierr = PCBDDCSetUpCoarseEnvironment(pc,coarse_submat_vals);CHKERRQ(ierr);
640*88ebb749SStefano Zampini   ierr = PetscFree(coarse_submat_vals);CHKERRQ(ierr);
641*88ebb749SStefano Zampini   PetscFunctionReturn(0);
642*88ebb749SStefano Zampini }
643*88ebb749SStefano Zampini 
644*88ebb749SStefano Zampini #undef __FUNCT__
645aa0d41d4SStefano Zampini #define __FUNCT__ "PCBDDCSetUpLocalMatrices"
646aa0d41d4SStefano Zampini PetscErrorCode PCBDDCSetUpLocalMatrices(PC pc)
647aa0d41d4SStefano Zampini {
648aa0d41d4SStefano Zampini   PC_IS*            pcis = (PC_IS*)(pc->data);
649aa0d41d4SStefano Zampini   PC_BDDC*          pcbddc = (PC_BDDC*)pc->data;
650aa0d41d4SStefano Zampini   Mat_IS*           matis = (Mat_IS*)pc->pmat->data;
651aa0d41d4SStefano Zampini   /* manage repeated solves */
652aa0d41d4SStefano Zampini   MatReuse          reuse;
653aa0d41d4SStefano Zampini   MatStructure      matstruct;
654aa0d41d4SStefano Zampini   PetscErrorCode    ierr;
655aa0d41d4SStefano Zampini 
656aa0d41d4SStefano Zampini   PetscFunctionBegin;
657aa0d41d4SStefano Zampini   /* get mat flags */
658aa0d41d4SStefano Zampini   ierr = PCGetOperators(pc,NULL,NULL,&matstruct);CHKERRQ(ierr);
659aa0d41d4SStefano Zampini   reuse = MAT_INITIAL_MATRIX;
660aa0d41d4SStefano Zampini   if (pc->setupcalled) {
661aa0d41d4SStefano Zampini     /* when matstruct is SAME_PRECONDITIONER, we shouldn't be here */
662aa0d41d4SStefano Zampini     if (matstruct == SAME_PRECONDITIONER) SETERRQ(PetscObjectComm((PetscObject)pc),PETSC_ERR_PLIB,"This should not happen");
663aa0d41d4SStefano Zampini     if (matstruct == SAME_NONZERO_PATTERN) {
664aa0d41d4SStefano Zampini       reuse = MAT_REUSE_MATRIX;
665aa0d41d4SStefano Zampini     } else {
666aa0d41d4SStefano Zampini       reuse = MAT_INITIAL_MATRIX;
667aa0d41d4SStefano Zampini     }
668aa0d41d4SStefano Zampini   }
669aa0d41d4SStefano Zampini   if (reuse == MAT_INITIAL_MATRIX) {
670aa0d41d4SStefano Zampini     ierr = MatDestroy(&pcis->A_II);CHKERRQ(ierr);
671aa0d41d4SStefano Zampini     ierr = MatDestroy(&pcis->A_IB);CHKERRQ(ierr);
672aa0d41d4SStefano Zampini     ierr = MatDestroy(&pcis->A_BI);CHKERRQ(ierr);
673aa0d41d4SStefano Zampini     ierr = MatDestroy(&pcis->A_BB);CHKERRQ(ierr);
674aa0d41d4SStefano Zampini     ierr = MatDestroy(&pcbddc->local_mat);CHKERRQ(ierr);
675aa0d41d4SStefano Zampini   }
676aa0d41d4SStefano Zampini 
677aa0d41d4SStefano Zampini   /* transform local matrices if needed */
678aa0d41d4SStefano Zampini   if (pcbddc->use_change_of_basis) {
679aa0d41d4SStefano Zampini     Mat         change_mat_all;
680aa0d41d4SStefano Zampini     PetscScalar *row_cmat_values;
681aa0d41d4SStefano Zampini     PetscInt    *row_cmat_indices;
682aa0d41d4SStefano Zampini     PetscInt    *nnz,*is_indices,*temp_indices;
683aa0d41d4SStefano Zampini     PetscInt    i,j,k,n_D,n_B;
684aa0d41d4SStefano Zampini 
685aa0d41d4SStefano Zampini     /* Get Non-overlapping dimensions */
686aa0d41d4SStefano Zampini     n_B = pcis->n_B;
687aa0d41d4SStefano Zampini     n_D = pcis->n-n_B;
688aa0d41d4SStefano Zampini 
689aa0d41d4SStefano Zampini     /* compute nonzero structure of change of basis on all local nodes */
690aa0d41d4SStefano Zampini     ierr = PetscMalloc(pcis->n*sizeof(PetscInt),&nnz);CHKERRQ(ierr);
691aa0d41d4SStefano Zampini     ierr = ISGetIndices(pcis->is_I_local,(const PetscInt**)&is_indices);CHKERRQ(ierr);
692aa0d41d4SStefano Zampini     for (i=0;i<n_D;i++) nnz[is_indices[i]] = 1;
693aa0d41d4SStefano Zampini     ierr = ISRestoreIndices(pcis->is_I_local,(const PetscInt**)&is_indices);CHKERRQ(ierr);
694aa0d41d4SStefano Zampini     ierr = ISGetIndices(pcis->is_B_local,(const PetscInt**)&is_indices);CHKERRQ(ierr);
695aa0d41d4SStefano Zampini     k=1;
696aa0d41d4SStefano Zampini     for (i=0;i<n_B;i++) {
697aa0d41d4SStefano Zampini       ierr = MatGetRow(pcbddc->ChangeOfBasisMatrix,i,&j,NULL,NULL);CHKERRQ(ierr);
698aa0d41d4SStefano Zampini       nnz[is_indices[i]]=j;
699aa0d41d4SStefano Zampini       if (k < j) k = j;
700aa0d41d4SStefano Zampini       ierr = MatRestoreRow(pcbddc->ChangeOfBasisMatrix,i,&j,NULL,NULL);CHKERRQ(ierr);
701aa0d41d4SStefano Zampini     }
702aa0d41d4SStefano Zampini     ierr = ISRestoreIndices(pcis->is_B_local,(const PetscInt**)&is_indices);CHKERRQ(ierr);
703aa0d41d4SStefano Zampini     /* assemble change of basis matrix on the whole set of local dofs */
704aa0d41d4SStefano Zampini     ierr = PetscMalloc(k*sizeof(PetscInt),&temp_indices);CHKERRQ(ierr);
705aa0d41d4SStefano Zampini     ierr = MatCreate(PETSC_COMM_SELF,&change_mat_all);CHKERRQ(ierr);
706aa0d41d4SStefano Zampini     ierr = MatSetSizes(change_mat_all,pcis->n,pcis->n,pcis->n,pcis->n);CHKERRQ(ierr);
707aa0d41d4SStefano Zampini     ierr = MatSetType(change_mat_all,MATSEQAIJ);CHKERRQ(ierr);
708aa0d41d4SStefano Zampini     ierr = MatSeqAIJSetPreallocation(change_mat_all,0,nnz);CHKERRQ(ierr);
709aa0d41d4SStefano Zampini     ierr = ISGetIndices(pcis->is_I_local,(const PetscInt**)&is_indices);CHKERRQ(ierr);
710aa0d41d4SStefano Zampini     for (i=0;i<n_D;i++) {
711aa0d41d4SStefano Zampini       ierr = MatSetValue(change_mat_all,is_indices[i],is_indices[i],1.0,INSERT_VALUES);CHKERRQ(ierr);
712aa0d41d4SStefano Zampini     }
713aa0d41d4SStefano Zampini     ierr = ISRestoreIndices(pcis->is_I_local,(const PetscInt**)&is_indices);CHKERRQ(ierr);
714aa0d41d4SStefano Zampini     ierr = ISGetIndices(pcis->is_B_local,(const PetscInt**)&is_indices);CHKERRQ(ierr);
715aa0d41d4SStefano Zampini     for (i=0;i<n_B;i++) {
716aa0d41d4SStefano Zampini       ierr = MatGetRow(pcbddc->ChangeOfBasisMatrix,i,&j,(const PetscInt**)&row_cmat_indices,(const PetscScalar**)&row_cmat_values);CHKERRQ(ierr);
717aa0d41d4SStefano Zampini       for (k=0; k<j; k++) temp_indices[k]=is_indices[row_cmat_indices[k]];
718aa0d41d4SStefano Zampini       ierr = MatSetValues(change_mat_all,1,&is_indices[i],j,temp_indices,row_cmat_values,INSERT_VALUES);CHKERRQ(ierr);
719aa0d41d4SStefano Zampini       ierr = MatRestoreRow(pcbddc->ChangeOfBasisMatrix,i,&j,(const PetscInt**)&row_cmat_indices,(const PetscScalar**)&row_cmat_values);CHKERRQ(ierr);
720aa0d41d4SStefano Zampini     }
721aa0d41d4SStefano Zampini     ierr = MatAssemblyBegin(change_mat_all,MAT_FINAL_ASSEMBLY);CHKERRQ(ierr);
722aa0d41d4SStefano Zampini     ierr = MatAssemblyEnd(change_mat_all,MAT_FINAL_ASSEMBLY);CHKERRQ(ierr);
723aa0d41d4SStefano Zampini     /* TODO: HOW TO WORK WITH BAIJ? PtAP not provided */
724aa0d41d4SStefano Zampini     ierr = MatGetBlockSize(matis->A,&i);CHKERRQ(ierr);
725aa0d41d4SStefano Zampini     if (i==1) {
726aa0d41d4SStefano Zampini       ierr = MatPtAP(matis->A,change_mat_all,reuse,2.0,&pcbddc->local_mat);CHKERRQ(ierr);
727aa0d41d4SStefano Zampini     } else {
728aa0d41d4SStefano Zampini       Mat work_mat;
729aa0d41d4SStefano Zampini       ierr = MatConvert(matis->A,MATSEQAIJ,MAT_INITIAL_MATRIX,&work_mat);CHKERRQ(ierr);
730aa0d41d4SStefano Zampini       ierr = MatPtAP(work_mat,change_mat_all,reuse,2.0,&pcbddc->local_mat);CHKERRQ(ierr);
731aa0d41d4SStefano Zampini       ierr = MatDestroy(&work_mat);CHKERRQ(ierr);
732aa0d41d4SStefano Zampini     }
733aa0d41d4SStefano Zampini     ierr = MatDestroy(&change_mat_all);CHKERRQ(ierr);
734aa0d41d4SStefano Zampini     ierr = PetscFree(nnz);CHKERRQ(ierr);
735aa0d41d4SStefano Zampini     ierr = PetscFree(temp_indices);CHKERRQ(ierr);
736aa0d41d4SStefano Zampini   } else {
737aa0d41d4SStefano Zampini     /* without change of basis, the local matrix is unchanged */
738aa0d41d4SStefano Zampini     if (!pcbddc->local_mat) {
739aa0d41d4SStefano Zampini       ierr = PetscObjectReference((PetscObject)matis->A);CHKERRQ(ierr);
740aa0d41d4SStefano Zampini       pcbddc->local_mat = matis->A;
741aa0d41d4SStefano Zampini     }
742aa0d41d4SStefano Zampini   }
743aa0d41d4SStefano Zampini 
744aa0d41d4SStefano Zampini   /* get submatrices */
745aa0d41d4SStefano Zampini   ierr = MatGetSubMatrix(pcbddc->local_mat,pcis->is_I_local,pcis->is_I_local,reuse,&pcis->A_II);CHKERRQ(ierr);
746aa0d41d4SStefano Zampini   ierr = MatGetSubMatrix(pcbddc->local_mat,pcis->is_I_local,pcis->is_B_local,reuse,&pcis->A_IB);CHKERRQ(ierr);
747aa0d41d4SStefano Zampini   ierr = MatGetSubMatrix(pcbddc->local_mat,pcis->is_B_local,pcis->is_I_local,reuse,&pcis->A_BI);CHKERRQ(ierr);
748aa0d41d4SStefano Zampini   ierr = MatGetSubMatrix(pcbddc->local_mat,pcis->is_B_local,pcis->is_B_local,reuse,&pcis->A_BB);CHKERRQ(ierr);
749aa0d41d4SStefano Zampini   PetscFunctionReturn(0);
750aa0d41d4SStefano Zampini }
751aa0d41d4SStefano Zampini 
752aa0d41d4SStefano Zampini #undef __FUNCT__
753a64d13efSStefano Zampini #define __FUNCT__ "PCBDDCSetUpLocalScatters"
754a64d13efSStefano Zampini PetscErrorCode PCBDDCSetUpLocalScatters(PC pc,IS* is_R_local_n)
755a64d13efSStefano Zampini {
756a64d13efSStefano Zampini   PC_IS*         pcis = (PC_IS*)(pc->data);
757a64d13efSStefano Zampini   PC_BDDC*       pcbddc = (PC_BDDC*)pc->data;
758a64d13efSStefano Zampini   IS             is_R_local,is_aux1,is_aux2;
759a64d13efSStefano Zampini   PetscInt       *vertices,*aux_array1,*aux_array2,*is_indices,*idx_R_local;
760a64d13efSStefano Zampini   PetscInt       n_vertices,n_constraints,i,j,n_R,n_D,n_B;
761a64d13efSStefano Zampini   PetscBool      *array_bool;
762a64d13efSStefano Zampini   PetscErrorCode ierr;
763a64d13efSStefano Zampini 
764a64d13efSStefano Zampini   PetscFunctionBegin;
765a64d13efSStefano Zampini   /* Set Non-overlapping dimensions */
766a64d13efSStefano Zampini   n_B = pcis->n_B; n_D = pcis->n - n_B;
767a64d13efSStefano Zampini   /* get vertex indices from constraint matrix */
768a64d13efSStefano Zampini   ierr = PCBDDCGetPrimalVerticesLocalIdx(pc,&n_vertices,&vertices);CHKERRQ(ierr);
769a64d13efSStefano Zampini   /* Set number of constraints */
770a64d13efSStefano Zampini   n_constraints = pcbddc->local_primal_size-n_vertices;
771a64d13efSStefano Zampini   /* Dohrmann's notation: dofs splitted in R (Remaining: all dofs but the vertices) and V (Vertices) */
772a64d13efSStefano Zampini   ierr = PetscMalloc(pcis->n*sizeof(PetscInt),&array_bool);CHKERRQ(ierr);
773a64d13efSStefano Zampini   for (i=0;i<pcis->n;i++) array_bool[i] = PETSC_TRUE;
774a64d13efSStefano Zampini   for (i=0;i<n_vertices;i++) array_bool[vertices[i]] = PETSC_FALSE;
775a64d13efSStefano Zampini   ierr = PetscMalloc((pcis->n-n_vertices)*sizeof(PetscInt),&idx_R_local);CHKERRQ(ierr);
776a64d13efSStefano Zampini   for (i=0, n_R=0; i<pcis->n; i++) {
777a64d13efSStefano Zampini     if (array_bool[i]) {
778a64d13efSStefano Zampini       idx_R_local[n_R] = i;
779a64d13efSStefano Zampini       n_R++;
780a64d13efSStefano Zampini     }
781a64d13efSStefano Zampini   }
782a64d13efSStefano Zampini   ierr = PetscFree(vertices);CHKERRQ(ierr);
783a64d13efSStefano Zampini   ierr = ISCreateGeneral(PETSC_COMM_SELF,n_R,idx_R_local,PETSC_OWN_POINTER,&is_R_local);CHKERRQ(ierr);
784a64d13efSStefano Zampini 
785a64d13efSStefano Zampini   /* print some info if requested */
786a64d13efSStefano Zampini   if (pcbddc->dbg_flag) {
787a64d13efSStefano Zampini     ierr = PetscViewerASCIIPrintf(pcbddc->dbg_viewer,"--------------------------------------------------\n");CHKERRQ(ierr);
788a64d13efSStefano Zampini     ierr = PetscViewerFlush(pcbddc->dbg_viewer);CHKERRQ(ierr);
789a64d13efSStefano Zampini     ierr = PetscViewerASCIISynchronizedPrintf(pcbddc->dbg_viewer,"Subdomain %04d local dimensions\n",PetscGlobalRank);CHKERRQ(ierr);
790a64d13efSStefano Zampini     ierr = PetscViewerASCIISynchronizedPrintf(pcbddc->dbg_viewer,"local_size = %d, dirichlet_size = %d, boundary_size = %d\n",pcis->n,n_D,n_B);CHKERRQ(ierr);
791a64d13efSStefano Zampini     ierr = PetscViewerASCIISynchronizedPrintf(pcbddc->dbg_viewer,"r_size = %d, v_size = %d, constraints = %d, local_primal_size = %d\n",n_R,n_vertices,n_constraints,pcbddc->local_primal_size);CHKERRQ(ierr);
792a64d13efSStefano Zampini     ierr = PetscViewerASCIISynchronizedPrintf(pcbddc->dbg_viewer,"pcbddc->n_vertices = %d, pcbddc->n_constraints = %d\n",pcbddc->n_vertices,pcbddc->n_constraints);CHKERRQ(ierr);
793a64d13efSStefano Zampini     ierr = PetscViewerFlush(pcbddc->dbg_viewer);CHKERRQ(ierr);
794a64d13efSStefano Zampini   }
795a64d13efSStefano Zampini 
796a64d13efSStefano Zampini   /* VecScatters pcbddc->R_to_B and (optionally) pcbddc->R_to_D */
797a64d13efSStefano Zampini   ierr = PetscMalloc((pcis->n_B-n_vertices)*sizeof(PetscInt),&aux_array1);CHKERRQ(ierr);
798a64d13efSStefano Zampini   ierr = PetscMalloc((pcis->n_B-n_vertices)*sizeof(PetscInt),&aux_array2);CHKERRQ(ierr);
799a64d13efSStefano Zampini   ierr = ISGetIndices(pcis->is_I_local,(const PetscInt**)&is_indices);CHKERRQ(ierr);
800a64d13efSStefano Zampini   for (i=0; i<n_D; i++) array_bool[is_indices[i]] = PETSC_FALSE;
801a64d13efSStefano Zampini   ierr = ISRestoreIndices(pcis->is_I_local,(const PetscInt**)&is_indices);CHKERRQ(ierr);
802a64d13efSStefano Zampini   for (i=0, j=0; i<n_R; i++) {
803a64d13efSStefano Zampini     if (array_bool[idx_R_local[i]]) {
804a64d13efSStefano Zampini       aux_array1[j] = i;
805a64d13efSStefano Zampini       j++;
806a64d13efSStefano Zampini     }
807a64d13efSStefano Zampini   }
808a64d13efSStefano Zampini   ierr = ISCreateGeneral(PETSC_COMM_SELF,j,aux_array1,PETSC_OWN_POINTER,&is_aux1);CHKERRQ(ierr);
809a64d13efSStefano Zampini   ierr = ISGetIndices(pcis->is_B_local,(const PetscInt**)&is_indices);CHKERRQ(ierr);
810a64d13efSStefano Zampini   for (i=0, j=0; i<n_B; i++) {
811a64d13efSStefano Zampini     if (array_bool[is_indices[i]]) {
812a64d13efSStefano Zampini       aux_array2[j] = i; j++;
813a64d13efSStefano Zampini     }
814a64d13efSStefano Zampini   }
815a64d13efSStefano Zampini   ierr = ISRestoreIndices(pcis->is_B_local,(const PetscInt**)&is_indices);CHKERRQ(ierr);
816a64d13efSStefano Zampini   ierr = ISCreateGeneral(PETSC_COMM_SELF,j,aux_array2,PETSC_OWN_POINTER,&is_aux2);CHKERRQ(ierr);
817a64d13efSStefano Zampini   ierr = VecScatterCreate(pcbddc->vec1_R,is_aux1,pcis->vec1_B,is_aux2,&pcbddc->R_to_B);CHKERRQ(ierr);
818a64d13efSStefano Zampini   ierr = ISDestroy(&is_aux1);CHKERRQ(ierr);
819a64d13efSStefano Zampini   ierr = ISDestroy(&is_aux2);CHKERRQ(ierr);
820a64d13efSStefano Zampini 
821a64d13efSStefano Zampini   if (pcbddc->inexact_prec_type || pcbddc->dbg_flag ) {
822a64d13efSStefano Zampini     ierr = PetscMalloc(n_D*sizeof(PetscInt),&aux_array1);CHKERRQ(ierr);
823a64d13efSStefano Zampini     for (i=0, j=0; i<n_R; i++) {
824a64d13efSStefano Zampini       if (!array_bool[idx_R_local[i]]) {
825a64d13efSStefano Zampini         aux_array1[j] = i;
826a64d13efSStefano Zampini         j++;
827a64d13efSStefano Zampini       }
828a64d13efSStefano Zampini     }
829a64d13efSStefano Zampini     ierr = ISCreateGeneral(PETSC_COMM_SELF,j,aux_array1,PETSC_OWN_POINTER,&is_aux1);CHKERRQ(ierr);
830a64d13efSStefano Zampini     ierr = VecScatterCreate(pcbddc->vec1_R,is_aux1,pcis->vec1_D,(IS)0,&pcbddc->R_to_D);CHKERRQ(ierr);
831a64d13efSStefano Zampini     ierr = ISDestroy(&is_aux1);CHKERRQ(ierr);
832a64d13efSStefano Zampini   }
833a64d13efSStefano Zampini   ierr = PetscFree(array_bool);CHKERRQ(ierr);
834a64d13efSStefano Zampini   *is_R_local_n = is_R_local;
835a64d13efSStefano Zampini   PetscFunctionReturn(0);
836a64d13efSStefano Zampini }
837a64d13efSStefano Zampini 
838a64d13efSStefano Zampini #undef __FUNCT__
839304d26faSStefano Zampini #define __FUNCT__ "PCBDDCSetUseExactDirichlet"
840304d26faSStefano Zampini PetscErrorCode PCBDDCSetUseExactDirichlet(PC pc,PetscBool use)
841304d26faSStefano Zampini {
842304d26faSStefano Zampini   PC_BDDC  *pcbddc = (PC_BDDC*)pc->data;
843304d26faSStefano Zampini 
844304d26faSStefano Zampini   PetscFunctionBegin;
845304d26faSStefano Zampini   pcbddc->use_exact_dirichlet=use;
846304d26faSStefano Zampini   PetscFunctionReturn(0);
847304d26faSStefano Zampini }
848304d26faSStefano Zampini 
849304d26faSStefano Zampini #undef __FUNCT__
850304d26faSStefano Zampini #define __FUNCT__ "PCBDDCSetUpLocalSolvers"
851304d26faSStefano Zampini PetscErrorCode PCBDDCSetUpLocalSolvers(PC pc, IS is_I_local, IS is_R_local)
852304d26faSStefano Zampini {
853304d26faSStefano Zampini   PC_BDDC        *pcbddc = (PC_BDDC*)pc->data;
854304d26faSStefano Zampini   PC_IS          *pcis = (PC_IS*)pc->data;
855304d26faSStefano Zampini   PC             pc_temp;
856304d26faSStefano Zampini   Mat            A_RR;
857304d26faSStefano Zampini   Vec            vec1,vec2,vec3;
858304d26faSStefano Zampini   MatStructure   matstruct;
859304d26faSStefano Zampini   PetscScalar    m_one = -1.0;
860304d26faSStefano Zampini   PetscReal      value;
861304d26faSStefano Zampini   PetscInt       n_D,n_R,use_exact,use_exact_reduced;
862304d26faSStefano Zampini   PetscErrorCode ierr;
863304d26faSStefano Zampini 
864304d26faSStefano Zampini   PetscFunctionBegin;
865304d26faSStefano Zampini   /* Creating PC contexts for local Dirichlet and Neumann problems */
866304d26faSStefano Zampini   ierr = PCGetOperators(pc,NULL,NULL,&matstruct);CHKERRQ(ierr);
867304d26faSStefano Zampini 
868304d26faSStefano Zampini   /* DIRICHLET PROBLEM */
869ac78edfcSStefano Zampini   /* Matrix for Dirichlet problem is pcis->A_II */
870304d26faSStefano Zampini   ierr = ISGetSize(is_I_local,&n_D);CHKERRQ(ierr);
871304d26faSStefano Zampini   if (!pcbddc->ksp_D) { /* create object if not yet build */
872304d26faSStefano Zampini     ierr = KSPCreate(PETSC_COMM_SELF,&pcbddc->ksp_D);CHKERRQ(ierr);
873304d26faSStefano Zampini     ierr = PetscObjectIncrementTabLevel((PetscObject)pcbddc->ksp_D,(PetscObject)pc,1);CHKERRQ(ierr);
874304d26faSStefano Zampini     /* default */
875304d26faSStefano Zampini     ierr = KSPSetType(pcbddc->ksp_D,KSPPREONLY);CHKERRQ(ierr);
876304d26faSStefano Zampini     ierr = KSPSetOptionsPrefix(pcbddc->ksp_D,"dirichlet_");CHKERRQ(ierr);
877304d26faSStefano Zampini     ierr = KSPGetPC(pcbddc->ksp_D,&pc_temp);CHKERRQ(ierr);
878304d26faSStefano Zampini     ierr = PCSetType(pc_temp,PCLU);CHKERRQ(ierr);
879304d26faSStefano Zampini     ierr = PCFactorSetReuseFill(pc_temp,PETSC_TRUE);CHKERRQ(ierr);
880304d26faSStefano Zampini   }
881304d26faSStefano Zampini   ierr = KSPSetOperators(pcbddc->ksp_D,pcis->A_II,pcis->A_II,matstruct);CHKERRQ(ierr);
882304d26faSStefano Zampini   /* Allow user's customization */
883304d26faSStefano Zampini   ierr = KSPSetFromOptions(pcbddc->ksp_D);CHKERRQ(ierr);
884304d26faSStefano Zampini   /* umfpack interface has a bug when matrix dimension is zero. TODO solve from umfpack interface */
885304d26faSStefano Zampini   if (!n_D) {
886304d26faSStefano Zampini     ierr = KSPGetPC(pcbddc->ksp_D,&pc_temp);CHKERRQ(ierr);
887304d26faSStefano Zampini     ierr = PCSetType(pc_temp,PCNONE);CHKERRQ(ierr);
888304d26faSStefano Zampini   }
889304d26faSStefano Zampini   /* Set Up KSP for Dirichlet problem of BDDC */
890304d26faSStefano Zampini   ierr = KSPSetUp(pcbddc->ksp_D);CHKERRQ(ierr);
891304d26faSStefano Zampini   /* set ksp_D into pcis data */
892304d26faSStefano Zampini   ierr = KSPDestroy(&pcis->ksp_D);CHKERRQ(ierr);
893304d26faSStefano Zampini   ierr = PetscObjectReference((PetscObject)pcbddc->ksp_D);CHKERRQ(ierr);
894304d26faSStefano Zampini   pcis->ksp_D = pcbddc->ksp_D;
895304d26faSStefano Zampini 
896304d26faSStefano Zampini   /* NEUMANN PROBLEM */
897304d26faSStefano Zampini   /* Matrix for Neumann problem is A_RR -> we need to create it */
898304d26faSStefano Zampini   ierr = ISGetSize(is_R_local,&n_R);CHKERRQ(ierr);
899304d26faSStefano Zampini   ierr = MatGetSubMatrix(pcbddc->local_mat,is_R_local,is_R_local,MAT_INITIAL_MATRIX,&A_RR);CHKERRQ(ierr);
900304d26faSStefano Zampini   if (!pcbddc->ksp_R) { /* create object if not yet build */
901304d26faSStefano Zampini     ierr = KSPCreate(PETSC_COMM_SELF,&pcbddc->ksp_R);CHKERRQ(ierr);
902304d26faSStefano Zampini     ierr = PetscObjectIncrementTabLevel((PetscObject)pcbddc->ksp_R,(PetscObject)pc,1);CHKERRQ(ierr);
903304d26faSStefano Zampini     /* default */
904304d26faSStefano Zampini     ierr = KSPSetType(pcbddc->ksp_R,KSPPREONLY);CHKERRQ(ierr);
905304d26faSStefano Zampini     ierr = KSPSetOptionsPrefix(pcbddc->ksp_R,"neumann_");CHKERRQ(ierr);
906304d26faSStefano Zampini     ierr = KSPGetPC(pcbddc->ksp_R,&pc_temp);CHKERRQ(ierr);
907304d26faSStefano Zampini     ierr = PCSetType(pc_temp,PCLU);CHKERRQ(ierr);
908304d26faSStefano Zampini     ierr = PCFactorSetReuseFill(pc_temp,PETSC_TRUE);CHKERRQ(ierr);
909304d26faSStefano Zampini   }
910304d26faSStefano Zampini   ierr = KSPSetOperators(pcbddc->ksp_R,A_RR,A_RR,matstruct);CHKERRQ(ierr);
911304d26faSStefano Zampini   /* Allow user's customization */
912304d26faSStefano Zampini   ierr = KSPSetFromOptions(pcbddc->ksp_R);CHKERRQ(ierr);
913304d26faSStefano Zampini   /* umfpack interface has a bug when matrix dimension is zero. TODO solve from umfpack interface */
914304d26faSStefano Zampini   if (!n_R) {
915304d26faSStefano Zampini     ierr = KSPGetPC(pcbddc->ksp_R,&pc_temp);CHKERRQ(ierr);
916304d26faSStefano Zampini     ierr = PCSetType(pc_temp,PCNONE);CHKERRQ(ierr);
917304d26faSStefano Zampini   }
918304d26faSStefano Zampini   /* Set Up KSP for Neumann problem of BDDC */
919304d26faSStefano Zampini   ierr = KSPSetUp(pcbddc->ksp_R);CHKERRQ(ierr);
920304d26faSStefano Zampini 
921304d26faSStefano Zampini   /* check Dirichlet and Neumann solvers and adapt them if a nullspace correction is needed */
922304d26faSStefano Zampini 
923304d26faSStefano Zampini   /* Dirichlet */
924304d26faSStefano Zampini   ierr = MatGetVecs(pcis->A_II,&vec1,&vec2);CHKERRQ(ierr);
925304d26faSStefano Zampini   ierr = VecDuplicate(vec1,&vec3);CHKERRQ(ierr);
926304d26faSStefano Zampini   ierr = VecSetRandom(vec1,NULL);CHKERRQ(ierr);
927304d26faSStefano Zampini   ierr = MatMult(pcis->A_II,vec1,vec2);CHKERRQ(ierr);
928304d26faSStefano Zampini   ierr = KSPSolve(pcbddc->ksp_D,vec2,vec3);CHKERRQ(ierr);
929304d26faSStefano Zampini   ierr = VecAXPY(vec3,m_one,vec1);CHKERRQ(ierr);
930304d26faSStefano Zampini   ierr = VecNorm(vec3,NORM_INFINITY,&value);CHKERRQ(ierr);
931304d26faSStefano Zampini   ierr = VecDestroy(&vec1);CHKERRQ(ierr);
932304d26faSStefano Zampini   ierr = VecDestroy(&vec2);CHKERRQ(ierr);
933304d26faSStefano Zampini   ierr = VecDestroy(&vec3);CHKERRQ(ierr);
934304d26faSStefano Zampini   /* need to be adapted? */
935304d26faSStefano Zampini   use_exact = (PetscAbsReal(value) > 1.e-4 ? 0 : 1);
936304d26faSStefano Zampini   ierr = MPI_Allreduce(&use_exact,&use_exact_reduced,1,MPIU_INT,MPI_LAND,PetscObjectComm((PetscObject)pc));CHKERRQ(ierr);
937304d26faSStefano Zampini   ierr = PCBDDCSetUseExactDirichlet(pc,(PetscBool)use_exact_reduced);CHKERRQ(ierr);
938304d26faSStefano Zampini   /* print info */
939304d26faSStefano Zampini   if (pcbddc->dbg_flag) {
940304d26faSStefano Zampini     ierr = PetscViewerFlush(pcbddc->dbg_viewer);CHKERRQ(ierr);
941304d26faSStefano Zampini     ierr = PetscViewerASCIIPrintf(pcbddc->dbg_viewer,"--------------------------------------------------\n");CHKERRQ(ierr);
942304d26faSStefano Zampini     ierr = PetscViewerASCIIPrintf(pcbddc->dbg_viewer,"Checking solution of Dirichlet and Neumann problems\n");CHKERRQ(ierr);
943304d26faSStefano Zampini     ierr = PetscViewerASCIISynchronizedPrintf(pcbddc->dbg_viewer,"Subdomain %04d infinity error for Dirichlet solve = % 1.14e \n",PetscGlobalRank,value);CHKERRQ(ierr);
944304d26faSStefano Zampini     ierr = PetscViewerFlush(pcbddc->dbg_viewer);CHKERRQ(ierr);
945304d26faSStefano Zampini   }
946304d26faSStefano Zampini   if (n_D && pcbddc->NullSpace && !use_exact_reduced && !pcbddc->inexact_prec_type) {
947304d26faSStefano Zampini     ierr = PCBDDCNullSpaceAssembleCorrection(pc,is_I_local);CHKERRQ(ierr);
948304d26faSStefano Zampini   }
949304d26faSStefano Zampini 
950304d26faSStefano Zampini   /* Neumann */
951304d26faSStefano Zampini   ierr = MatGetVecs(A_RR,&vec1,&vec2);CHKERRQ(ierr);
952304d26faSStefano Zampini   ierr = VecDuplicate(vec1,&vec3);CHKERRQ(ierr);
953304d26faSStefano Zampini   ierr = VecSetRandom(vec1,NULL);CHKERRQ(ierr);
954304d26faSStefano Zampini   ierr = MatMult(A_RR,vec1,vec2);CHKERRQ(ierr);
955304d26faSStefano Zampini   ierr = KSPSolve(pcbddc->ksp_R,vec2,vec3);CHKERRQ(ierr);
956304d26faSStefano Zampini   ierr = VecAXPY(vec3,m_one,vec1);CHKERRQ(ierr);
957304d26faSStefano Zampini   ierr = VecNorm(vec3,NORM_INFINITY,&value);CHKERRQ(ierr);
958304d26faSStefano Zampini   ierr = VecDestroy(&vec1);CHKERRQ(ierr);
959304d26faSStefano Zampini   ierr = VecDestroy(&vec2);CHKERRQ(ierr);
960304d26faSStefano Zampini   ierr = VecDestroy(&vec3);CHKERRQ(ierr);
961304d26faSStefano Zampini   /* need to be adapted? */
962304d26faSStefano Zampini   use_exact = (PetscAbsReal(value) > 1.e-4 ? 0 : 1);
963304d26faSStefano Zampini   if (PetscAbsReal(value) > 1.e-4) use_exact = 0;
964304d26faSStefano Zampini   ierr = MPI_Allreduce(&use_exact,&use_exact_reduced,1,MPIU_INT,MPI_LAND,PetscObjectComm((PetscObject)pc));CHKERRQ(ierr);
965304d26faSStefano Zampini   /* print info */
966304d26faSStefano Zampini   if (pcbddc->dbg_flag) {
967304d26faSStefano Zampini     ierr = PetscViewerASCIISynchronizedPrintf(pcbddc->dbg_viewer,"Subdomain %04d infinity error for  Neumann  solve = % 1.14e \n",PetscGlobalRank,value);CHKERRQ(ierr);
968304d26faSStefano Zampini     ierr = PetscViewerFlush(pcbddc->dbg_viewer);CHKERRQ(ierr);
969304d26faSStefano Zampini   }
970304d26faSStefano Zampini   if (n_R && pcbddc->NullSpace && !use_exact_reduced) { /* is it the right logic? */
971304d26faSStefano Zampini     ierr = PCBDDCNullSpaceAssembleCorrection(pc,is_R_local);CHKERRQ(ierr);
972304d26faSStefano Zampini   }
973304d26faSStefano Zampini 
974304d26faSStefano Zampini   /* free Neumann problem's matrix */
975304d26faSStefano Zampini   ierr = MatDestroy(&A_RR);CHKERRQ(ierr);
976304d26faSStefano Zampini   PetscFunctionReturn(0);
977304d26faSStefano Zampini }
978304d26faSStefano Zampini 
979304d26faSStefano Zampini #undef __FUNCT__
980674ae819SStefano Zampini #define __FUNCT__ "PCBDDCSolveSaddlePoint"
981674ae819SStefano Zampini static PetscErrorCode  PCBDDCSolveSaddlePoint(PC pc)
982674ae819SStefano Zampini {
983674ae819SStefano Zampini   PetscErrorCode ierr;
984674ae819SStefano Zampini   PC_BDDC*       pcbddc = (PC_BDDC*)(pc->data);
985674ae819SStefano Zampini 
986674ae819SStefano Zampini   PetscFunctionBegin;
987674ae819SStefano Zampini   ierr = KSPSolve(pcbddc->ksp_R,pcbddc->vec1_R,pcbddc->vec2_R);CHKERRQ(ierr);
988674ae819SStefano Zampini   if (pcbddc->local_auxmat1) {
989674ae819SStefano Zampini     ierr = MatMult(pcbddc->local_auxmat1,pcbddc->vec2_R,pcbddc->vec1_C);CHKERRQ(ierr);
990674ae819SStefano Zampini     ierr = MatMultAdd(pcbddc->local_auxmat2,pcbddc->vec1_C,pcbddc->vec2_R,pcbddc->vec2_R);CHKERRQ(ierr);
991674ae819SStefano Zampini   }
992674ae819SStefano Zampini   PetscFunctionReturn(0);
993674ae819SStefano Zampini }
994674ae819SStefano Zampini 
995674ae819SStefano Zampini #undef __FUNCT__
996674ae819SStefano Zampini #define __FUNCT__ "PCBDDCApplyInterfacePreconditioner"
997674ae819SStefano Zampini PetscErrorCode  PCBDDCApplyInterfacePreconditioner(PC pc)
998674ae819SStefano Zampini {
999674ae819SStefano Zampini   PetscErrorCode ierr;
1000674ae819SStefano Zampini   PC_BDDC*        pcbddc = (PC_BDDC*)(pc->data);
1001674ae819SStefano Zampini   PC_IS*            pcis = (PC_IS*)  (pc->data);
1002674ae819SStefano Zampini   const PetscScalar zero = 0.0;
1003674ae819SStefano Zampini 
1004674ae819SStefano Zampini   PetscFunctionBegin;
100515aaf578SStefano Zampini   /* Application of PHI^T (or PSI^T)  */
100615aaf578SStefano Zampini   if (pcbddc->coarse_psi_B) {
100715aaf578SStefano Zampini     ierr = MatMultTranspose(pcbddc->coarse_psi_B,pcis->vec1_B,pcbddc->vec1_P);CHKERRQ(ierr);
100815aaf578SStefano Zampini     if (pcbddc->inexact_prec_type) { ierr = MatMultTransposeAdd(pcbddc->coarse_psi_D,pcis->vec1_D,pcbddc->vec1_P,pcbddc->vec1_P);CHKERRQ(ierr); }
100915aaf578SStefano Zampini   } else {
1010674ae819SStefano Zampini     ierr = MatMultTranspose(pcbddc->coarse_phi_B,pcis->vec1_B,pcbddc->vec1_P);CHKERRQ(ierr);
1011674ae819SStefano Zampini     if (pcbddc->inexact_prec_type) { ierr = MatMultTransposeAdd(pcbddc->coarse_phi_D,pcis->vec1_D,pcbddc->vec1_P,pcbddc->vec1_P);CHKERRQ(ierr); }
101215aaf578SStefano Zampini   }
1013674ae819SStefano Zampini   /* Scatter data of coarse_rhs */
1014674ae819SStefano Zampini   if (pcbddc->coarse_rhs) { ierr = VecSet(pcbddc->coarse_rhs,zero);CHKERRQ(ierr); }
1015674ae819SStefano Zampini   ierr = PCBDDCScatterCoarseDataBegin(pc,pcbddc->vec1_P,pcbddc->coarse_rhs,ADD_VALUES,SCATTER_FORWARD);CHKERRQ(ierr);
1016674ae819SStefano Zampini 
1017674ae819SStefano Zampini   /* Local solution on R nodes */
1018674ae819SStefano Zampini   ierr = VecSet(pcbddc->vec1_R,zero);CHKERRQ(ierr);
1019674ae819SStefano Zampini   ierr = VecScatterBegin(pcbddc->R_to_B,pcis->vec1_B,pcbddc->vec1_R,INSERT_VALUES,SCATTER_REVERSE);CHKERRQ(ierr);
1020674ae819SStefano Zampini   ierr = VecScatterEnd  (pcbddc->R_to_B,pcis->vec1_B,pcbddc->vec1_R,INSERT_VALUES,SCATTER_REVERSE);CHKERRQ(ierr);
1021674ae819SStefano Zampini   if (pcbddc->inexact_prec_type) {
1022674ae819SStefano Zampini     ierr = VecScatterBegin(pcbddc->R_to_D,pcis->vec1_D,pcbddc->vec1_R,INSERT_VALUES,SCATTER_REVERSE);CHKERRQ(ierr);
1023674ae819SStefano Zampini     ierr = VecScatterEnd  (pcbddc->R_to_D,pcis->vec1_D,pcbddc->vec1_R,INSERT_VALUES,SCATTER_REVERSE);CHKERRQ(ierr);
1024674ae819SStefano Zampini   }
1025674ae819SStefano Zampini   ierr = PCBDDCSolveSaddlePoint(pc);CHKERRQ(ierr);
1026674ae819SStefano Zampini   ierr = VecSet(pcis->vec1_B,zero);CHKERRQ(ierr);
1027674ae819SStefano Zampini   ierr = VecScatterBegin(pcbddc->R_to_B,pcbddc->vec2_R,pcis->vec1_B,INSERT_VALUES,SCATTER_FORWARD);CHKERRQ(ierr);
1028674ae819SStefano Zampini   ierr = VecScatterEnd  (pcbddc->R_to_B,pcbddc->vec2_R,pcis->vec1_B,INSERT_VALUES,SCATTER_FORWARD);CHKERRQ(ierr);
1029674ae819SStefano Zampini   if (pcbddc->inexact_prec_type) {
1030674ae819SStefano Zampini     ierr = VecScatterBegin(pcbddc->R_to_D,pcbddc->vec2_R,pcis->vec1_D,INSERT_VALUES,SCATTER_FORWARD);CHKERRQ(ierr);
1031674ae819SStefano Zampini     ierr = VecScatterEnd  (pcbddc->R_to_D,pcbddc->vec2_R,pcis->vec1_D,INSERT_VALUES,SCATTER_FORWARD);CHKERRQ(ierr);
1032674ae819SStefano Zampini   }
1033674ae819SStefano Zampini 
1034674ae819SStefano Zampini   /* Coarse solution */
1035674ae819SStefano Zampini   ierr = PCBDDCScatterCoarseDataEnd(pc,pcbddc->vec1_P,pcbddc->coarse_rhs,ADD_VALUES,SCATTER_FORWARD);CHKERRQ(ierr);
1036674ae819SStefano Zampini   if (pcbddc->coarse_rhs) { /* TODO remove null space when doing multilevel */
1037674ae819SStefano Zampini     ierr = KSPSolve(pcbddc->coarse_ksp,pcbddc->coarse_rhs,pcbddc->coarse_vec);CHKERRQ(ierr);
1038674ae819SStefano Zampini   }
1039674ae819SStefano Zampini   ierr = PCBDDCScatterCoarseDataBegin(pc,pcbddc->coarse_vec,pcbddc->vec1_P,INSERT_VALUES,SCATTER_REVERSE);CHKERRQ(ierr);
1040674ae819SStefano Zampini   ierr = PCBDDCScatterCoarseDataEnd  (pc,pcbddc->coarse_vec,pcbddc->vec1_P,INSERT_VALUES,SCATTER_REVERSE);CHKERRQ(ierr);
1041674ae819SStefano Zampini 
1042674ae819SStefano Zampini   /* Sum contributions from two levels */
1043674ae819SStefano Zampini   ierr = MatMultAdd(pcbddc->coarse_phi_B,pcbddc->vec1_P,pcis->vec1_B,pcis->vec1_B);CHKERRQ(ierr);
1044674ae819SStefano Zampini   if (pcbddc->inexact_prec_type) { ierr = MatMultAdd(pcbddc->coarse_phi_D,pcbddc->vec1_P,pcis->vec1_D,pcis->vec1_D);CHKERRQ(ierr); }
1045674ae819SStefano Zampini   PetscFunctionReturn(0);
1046674ae819SStefano Zampini }
1047674ae819SStefano Zampini 
1048674ae819SStefano Zampini #undef __FUNCT__
1049674ae819SStefano Zampini #define __FUNCT__ "PCBDDCScatterCoarseDataBegin"
1050674ae819SStefano Zampini PetscErrorCode PCBDDCScatterCoarseDataBegin(PC pc,Vec vec_from, Vec vec_to, InsertMode imode, ScatterMode smode)
1051674ae819SStefano Zampini {
1052674ae819SStefano Zampini   PetscErrorCode ierr;
1053674ae819SStefano Zampini   PC_BDDC*       pcbddc = (PC_BDDC*)(pc->data);
1054674ae819SStefano Zampini 
1055674ae819SStefano Zampini   PetscFunctionBegin;
1056674ae819SStefano Zampini   switch (pcbddc->coarse_communications_type) {
1057674ae819SStefano Zampini     case SCATTERS_BDDC:
1058674ae819SStefano Zampini       ierr = VecScatterBegin(pcbddc->coarse_loc_to_glob,vec_from,vec_to,imode,smode);CHKERRQ(ierr);
1059674ae819SStefano Zampini       break;
1060674ae819SStefano Zampini     case GATHERS_BDDC:
1061674ae819SStefano Zampini       break;
1062674ae819SStefano Zampini   }
1063674ae819SStefano Zampini   PetscFunctionReturn(0);
1064674ae819SStefano Zampini }
1065674ae819SStefano Zampini 
1066674ae819SStefano Zampini #undef __FUNCT__
1067674ae819SStefano Zampini #define __FUNCT__ "PCBDDCScatterCoarseDataEnd"
1068674ae819SStefano Zampini PetscErrorCode PCBDDCScatterCoarseDataEnd(PC pc,Vec vec_from, Vec vec_to, InsertMode imode, ScatterMode smode)
1069674ae819SStefano Zampini {
1070674ae819SStefano Zampini   PetscErrorCode ierr;
1071674ae819SStefano Zampini   PC_BDDC*       pcbddc = (PC_BDDC*)(pc->data);
1072674ae819SStefano Zampini   PetscScalar*   array_to;
1073674ae819SStefano Zampini   PetscScalar*   array_from;
1074674ae819SStefano Zampini   MPI_Comm       comm;
1075674ae819SStefano Zampini   PetscInt       i;
1076674ae819SStefano Zampini 
1077674ae819SStefano Zampini   PetscFunctionBegin;
1078674ae819SStefano Zampini   ierr = PetscObjectGetComm((PetscObject)pc,&comm);CHKERRQ(ierr);
1079674ae819SStefano Zampini   switch (pcbddc->coarse_communications_type) {
1080674ae819SStefano Zampini     case SCATTERS_BDDC:
1081674ae819SStefano Zampini       ierr = VecScatterEnd(pcbddc->coarse_loc_to_glob,vec_from,vec_to,imode,smode);CHKERRQ(ierr);
1082674ae819SStefano Zampini       break;
1083674ae819SStefano Zampini     case GATHERS_BDDC:
1084674ae819SStefano Zampini       if (vec_from) {
1085674ae819SStefano Zampini         ierr = VecGetArray(vec_from,&array_from);CHKERRQ(ierr);
1086674ae819SStefano Zampini       }
1087674ae819SStefano Zampini       if (vec_to) {
1088674ae819SStefano Zampini         ierr = VecGetArray(vec_to,&array_to);CHKERRQ(ierr);
1089674ae819SStefano Zampini       }
1090674ae819SStefano Zampini       switch(pcbddc->coarse_problem_type){
1091674ae819SStefano Zampini         case SEQUENTIAL_BDDC:
1092674ae819SStefano Zampini           if (smode == SCATTER_FORWARD) {
1093674ae819SStefano Zampini             ierr = MPI_Gatherv(&array_from[0],pcbddc->local_primal_size,MPIU_SCALAR,&pcbddc->replicated_local_primal_values[0],pcbddc->local_primal_sizes,pcbddc->local_primal_displacements,MPIU_SCALAR,0,comm);CHKERRQ(ierr);
1094674ae819SStefano Zampini             if (vec_to) {
1095674ae819SStefano Zampini               if (imode == ADD_VALUES) {
1096674ae819SStefano Zampini                 for (i=0;i<pcbddc->replicated_primal_size;i++) {
1097674ae819SStefano Zampini                   array_to[pcbddc->replicated_local_primal_indices[i]]+=pcbddc->replicated_local_primal_values[i];
1098674ae819SStefano Zampini                 }
1099674ae819SStefano Zampini               } else {
1100674ae819SStefano Zampini                 for (i=0;i<pcbddc->replicated_primal_size;i++) {
1101674ae819SStefano Zampini                   array_to[pcbddc->replicated_local_primal_indices[i]]=pcbddc->replicated_local_primal_values[i];
1102674ae819SStefano Zampini                 }
1103674ae819SStefano Zampini               }
1104674ae819SStefano Zampini             }
1105674ae819SStefano Zampini           } else {
1106674ae819SStefano Zampini             if (vec_from) {
1107674ae819SStefano Zampini               if (imode == ADD_VALUES) {
1108674ae819SStefano Zampini                 MPI_Comm vec_from_comm;
1109674ae819SStefano Zampini                 ierr = PetscObjectGetComm((PetscObject)(vec_from),&vec_from_comm);CHKERRQ(ierr);
1110674ae819SStefano Zampini                 SETERRQ2(vec_from_comm,PETSC_ERR_SUP,"Unsupported insert mode ADD_VALUES for SCATTER_REVERSE in %s for case %d\n",__FUNCT__,pcbddc->coarse_problem_type);
1111674ae819SStefano Zampini               }
1112674ae819SStefano Zampini               for (i=0;i<pcbddc->replicated_primal_size;i++) {
1113674ae819SStefano Zampini                 pcbddc->replicated_local_primal_values[i]=array_from[pcbddc->replicated_local_primal_indices[i]];
1114674ae819SStefano Zampini               }
1115674ae819SStefano Zampini             }
1116674ae819SStefano Zampini             ierr = MPI_Scatterv(&pcbddc->replicated_local_primal_values[0],pcbddc->local_primal_sizes,pcbddc->local_primal_displacements,MPIU_SCALAR,&array_to[0],pcbddc->local_primal_size,MPIU_SCALAR,0,comm);CHKERRQ(ierr);
1117674ae819SStefano Zampini           }
1118674ae819SStefano Zampini           break;
1119674ae819SStefano Zampini         case REPLICATED_BDDC:
1120674ae819SStefano Zampini           if (smode == SCATTER_FORWARD) {
1121674ae819SStefano Zampini             ierr = MPI_Allgatherv(&array_from[0],pcbddc->local_primal_size,MPIU_SCALAR,&pcbddc->replicated_local_primal_values[0],pcbddc->local_primal_sizes,pcbddc->local_primal_displacements,MPIU_SCALAR,comm);CHKERRQ(ierr);
1122674ae819SStefano Zampini             if (imode == ADD_VALUES) {
1123674ae819SStefano Zampini               for (i=0;i<pcbddc->replicated_primal_size;i++) {
1124674ae819SStefano Zampini                 array_to[pcbddc->replicated_local_primal_indices[i]]+=pcbddc->replicated_local_primal_values[i];
1125674ae819SStefano Zampini               }
1126674ae819SStefano Zampini             } else {
1127674ae819SStefano Zampini               for (i=0;i<pcbddc->replicated_primal_size;i++) {
1128674ae819SStefano Zampini                 array_to[pcbddc->replicated_local_primal_indices[i]]=pcbddc->replicated_local_primal_values[i];
1129674ae819SStefano Zampini               }
1130674ae819SStefano Zampini             }
1131674ae819SStefano Zampini           } else { /* no communications needed for SCATTER_REVERSE since needed data is already present */
1132674ae819SStefano Zampini             if (imode == ADD_VALUES) {
1133674ae819SStefano Zampini               for (i=0;i<pcbddc->local_primal_size;i++) {
1134674ae819SStefano Zampini                 array_to[i]+=array_from[pcbddc->local_primal_indices[i]];
1135674ae819SStefano Zampini               }
1136674ae819SStefano Zampini             } else {
1137674ae819SStefano Zampini               for (i=0;i<pcbddc->local_primal_size;i++) {
1138674ae819SStefano Zampini                 array_to[i]=array_from[pcbddc->local_primal_indices[i]];
1139674ae819SStefano Zampini               }
1140674ae819SStefano Zampini             }
1141674ae819SStefano Zampini           }
1142674ae819SStefano Zampini           break;
1143674ae819SStefano Zampini         case MULTILEVEL_BDDC:
1144674ae819SStefano Zampini           break;
1145674ae819SStefano Zampini         case PARALLEL_BDDC:
1146674ae819SStefano Zampini           break;
1147674ae819SStefano Zampini       }
1148674ae819SStefano Zampini       if (vec_from) {
1149674ae819SStefano Zampini         ierr = VecRestoreArray(vec_from,&array_from);CHKERRQ(ierr);
1150674ae819SStefano Zampini       }
1151674ae819SStefano Zampini       if (vec_to) {
1152674ae819SStefano Zampini         ierr = VecRestoreArray(vec_to,&array_to);CHKERRQ(ierr);
1153674ae819SStefano Zampini       }
1154674ae819SStefano Zampini       break;
1155674ae819SStefano Zampini   }
1156674ae819SStefano Zampini   PetscFunctionReturn(0);
1157674ae819SStefano Zampini }
1158674ae819SStefano Zampini 
1159984c4197SStefano Zampini /* uncomment for testing purposes */
1160984c4197SStefano Zampini /* #define PETSC_MISSING_LAPACK_GESVD 1 */
1161674ae819SStefano Zampini #undef __FUNCT__
1162674ae819SStefano Zampini #define __FUNCT__ "PCBDDCConstraintsSetUp"
1163674ae819SStefano Zampini PetscErrorCode PCBDDCConstraintsSetUp(PC pc)
1164674ae819SStefano Zampini {
1165674ae819SStefano Zampini   PetscErrorCode    ierr;
1166674ae819SStefano Zampini   PC_IS*            pcis = (PC_IS*)(pc->data);
1167674ae819SStefano Zampini   PC_BDDC*          pcbddc = (PC_BDDC*)pc->data;
1168674ae819SStefano Zampini   Mat_IS*           matis = (Mat_IS*)pc->pmat->data;
1169984c4197SStefano Zampini   /* constraint and (optionally) change of basis matrix implemented as SeqAIJ */
1170674ae819SStefano Zampini   MatType           impMatType=MATSEQAIJ;
1171984c4197SStefano Zampini   /* one and zero */
1172984c4197SStefano Zampini   PetscScalar       one=1.0,zero=0.0;
1173984c4197SStefano Zampini   /* space to store constraints and their local indices */
1174984c4197SStefano Zampini   PetscScalar       *temp_quadrature_constraint;
1175984c4197SStefano Zampini   PetscInt          *temp_indices,*temp_indices_to_constraint,*temp_indices_to_constraint_B;
1176984c4197SStefano Zampini   /* iterators */
1177984c4197SStefano Zampini   PetscInt          i,j,k,total_counts,temp_start_ptr;
1178984c4197SStefano Zampini   /* stuff to store connected components stored in pcbddc->mat_graph */
1179984c4197SStefano Zampini   IS                ISForVertices,*ISForFaces,*ISForEdges,*used_IS;
1180984c4197SStefano Zampini   PetscInt          n_ISForFaces,n_ISForEdges;
1181984c4197SStefano Zampini   PetscBool         get_faces,get_edges,get_vertices;
1182984c4197SStefano Zampini   /* near null space stuff */
1183674ae819SStefano Zampini   MatNullSpace      nearnullsp;
1184674ae819SStefano Zampini   const Vec         *nearnullvecs;
1185674ae819SStefano Zampini   Vec               *localnearnullsp;
1186984c4197SStefano Zampini   PetscBool         nnsp_has_cnst;
1187984c4197SStefano Zampini   PetscInt          nnsp_size;
1188984c4197SStefano Zampini   PetscScalar       *array;
1189984c4197SStefano Zampini   /* BLAS integers */
1190e310c8b4SStefano Zampini   PetscBLASInt      lwork,lierr;
1191e310c8b4SStefano Zampini   PetscBLASInt      Blas_N,Blas_M,Blas_K,Blas_one=1;
1192c4303822SStefano Zampini   PetscBLASInt      Blas_LDA,Blas_LDB,Blas_LDC;
1193984c4197SStefano Zampini   /* LAPACK working arrays for SVD or POD */
1194242a89d7SStefano Zampini   PetscBool         skip_lapack;
1195984c4197SStefano Zampini   PetscScalar       *work;
1196984c4197SStefano Zampini   PetscReal         *singular_vals;
1197984c4197SStefano Zampini #if defined(PETSC_USE_COMPLEX)
1198984c4197SStefano Zampini   PetscReal         *rwork;
1199674ae819SStefano Zampini #endif
1200984c4197SStefano Zampini #if defined(PETSC_MISSING_LAPACK_GESVD)
1201e310c8b4SStefano Zampini   PetscBLASInt      Blas_one_2=1;
1202984c4197SStefano Zampini   PetscScalar       *temp_basis,*correlation_mat;
1203b7d8b9f8SStefano Zampini #else
1204b7d8b9f8SStefano Zampini   PetscBLASInt      dummy_int_1=1,dummy_int_2=1;
1205b7d8b9f8SStefano Zampini   PetscScalar       dummy_scalar_1=0.0,dummy_scalar_2=0.0;
1206984c4197SStefano Zampini #endif
1207984c4197SStefano Zampini   /* change of basis */
1208984c4197SStefano Zampini   PetscInt          *aux_primal_numbering,*aux_primal_minloc,*global_indices;
1209984c4197SStefano Zampini   PetscBool         boolforchange,*change_basis,*touched;
1210984c4197SStefano Zampini   /* auxiliary stuff */
1211984c4197SStefano Zampini   PetscInt          *nnz,*is_indices,*local_to_B;
1212984c4197SStefano Zampini   /* some quantities */
1213984c4197SStefano Zampini   PetscInt          n_vertices,total_primal_vertices;
1214984c4197SStefano Zampini   PetscInt          size_of_constraint,max_size_of_constraint,max_constraints,temp_constraints;
1215984c4197SStefano Zampini 
1216674ae819SStefano Zampini 
1217674ae819SStefano Zampini   PetscFunctionBegin;
1218674ae819SStefano Zampini   /* Get index sets for faces, edges and vertices from graph */
1219674ae819SStefano Zampini   get_faces = PETSC_TRUE;
1220674ae819SStefano Zampini   get_edges = PETSC_TRUE;
1221674ae819SStefano Zampini   get_vertices = PETSC_TRUE;
1222674ae819SStefano Zampini   if (pcbddc->vertices_flag) {
1223674ae819SStefano Zampini     get_faces = PETSC_FALSE;
1224674ae819SStefano Zampini     get_edges = PETSC_FALSE;
1225674ae819SStefano Zampini   }
1226674ae819SStefano Zampini   if (pcbddc->constraints_flag) {
1227674ae819SStefano Zampini     get_vertices = PETSC_FALSE;
1228674ae819SStefano Zampini   }
1229674ae819SStefano Zampini   if (pcbddc->faces_flag) {
1230674ae819SStefano Zampini     get_edges = PETSC_FALSE;
1231674ae819SStefano Zampini   }
1232674ae819SStefano Zampini   if (pcbddc->edges_flag) {
1233674ae819SStefano Zampini     get_faces = PETSC_FALSE;
1234674ae819SStefano Zampini   }
1235674ae819SStefano Zampini   /* default */
1236674ae819SStefano Zampini   if (!get_faces && !get_edges && !get_vertices) {
1237674ae819SStefano Zampini     get_vertices = PETSC_TRUE;
1238674ae819SStefano Zampini   }
1239674ae819SStefano Zampini   ierr = PCBDDCGraphGetCandidatesIS(pcbddc->mat_graph,get_faces,get_edges,get_vertices,&n_ISForFaces,&ISForFaces,&n_ISForEdges,&ISForEdges,&ISForVertices);
1240984c4197SStefano Zampini   /* print some info */
1241674ae819SStefano Zampini   if (pcbddc->dbg_flag) {
1242674ae819SStefano Zampini     ierr = PetscViewerASCIISynchronizedPrintf(pcbddc->dbg_viewer,"--------------------------------------------------------------\n");CHKERRQ(ierr);
1243674ae819SStefano Zampini     i = 0;
1244674ae819SStefano Zampini     if (ISForVertices) {
1245674ae819SStefano Zampini       ierr = ISGetSize(ISForVertices,&i);CHKERRQ(ierr);
1246674ae819SStefano Zampini     }
1247674ae819SStefano Zampini     ierr = PetscViewerASCIISynchronizedPrintf(pcbddc->dbg_viewer,"Subdomain %04d got %02d local candidate vertices\n",PetscGlobalRank,i);CHKERRQ(ierr);
1248674ae819SStefano Zampini     ierr = PetscViewerASCIISynchronizedPrintf(pcbddc->dbg_viewer,"Subdomain %04d got %02d local candidate edges\n",PetscGlobalRank,n_ISForEdges);CHKERRQ(ierr);
124915aaf578SStefano Zampini     ierr = PetscViewerASCIISynchronizedPrintf(pcbddc->dbg_viewer,"Subdomain %04d got %02d local candidate faces\n",PetscGlobalRank,n_ISForFaces);CHKERRQ(ierr);
1250674ae819SStefano Zampini     ierr = PetscViewerFlush(pcbddc->dbg_viewer);CHKERRQ(ierr);
1251674ae819SStefano Zampini   }
1252674ae819SStefano Zampini   /* check if near null space is attached to global mat */
1253674ae819SStefano Zampini   ierr = MatGetNearNullSpace(pc->pmat,&nearnullsp);CHKERRQ(ierr);
1254674ae819SStefano Zampini   if (nearnullsp) {
1255674ae819SStefano Zampini     ierr = MatNullSpaceGetVecs(nearnullsp,&nnsp_has_cnst,&nnsp_size,&nearnullvecs);CHKERRQ(ierr);
1256984c4197SStefano Zampini   } else { /* if near null space is not provided BDDC uses constants by default */
1257984c4197SStefano Zampini     nnsp_size = 0;
1258674ae819SStefano Zampini     nnsp_has_cnst = PETSC_TRUE;
1259674ae819SStefano Zampini   }
1260984c4197SStefano Zampini   /* get max number of constraints on a single cc */
1261984c4197SStefano Zampini   max_constraints = nnsp_size;
1262984c4197SStefano Zampini   if (nnsp_has_cnst) max_constraints++;
1263984c4197SStefano Zampini 
1264674ae819SStefano Zampini   /*
1265674ae819SStefano Zampini        Evaluate maximum storage size needed by the procedure
1266674ae819SStefano Zampini        - temp_indices will contain start index of each constraint stored as follows
1267674ae819SStefano Zampini        - temp_indices_to_constraint  [temp_indices[i],...,temp[indices[i+1]-1] will contain the indices (in local numbering) on which the constraint acts
1268674ae819SStefano Zampini        - temp_indices_to_constraint_B[temp_indices[i],...,temp[indices[i+1]-1] will contain the indices (in boundary numbering) on which the constraint acts
1269674ae819SStefano Zampini        - temp_quadrature_constraint  [temp_indices[i],...,temp[indices[i+1]-1] will contain the scalars representing the constraint itself
1270674ae819SStefano Zampini                                                                                                                                                          */
1271674ae819SStefano Zampini   total_counts = n_ISForFaces+n_ISForEdges;
1272984c4197SStefano Zampini   total_counts *= max_constraints;
1273674ae819SStefano Zampini   n_vertices = 0;
1274674ae819SStefano Zampini   if (ISForVertices) {
1275674ae819SStefano Zampini     ierr = ISGetSize(ISForVertices,&n_vertices);CHKERRQ(ierr);
1276674ae819SStefano Zampini   }
1277674ae819SStefano Zampini   total_counts += n_vertices;
1278674ae819SStefano Zampini   ierr = PetscMalloc((total_counts+1)*sizeof(PetscInt),&temp_indices);CHKERRQ(ierr);
1279674ae819SStefano Zampini   ierr = PetscMalloc((total_counts+1)*sizeof(PetscBool),&change_basis);CHKERRQ(ierr);
1280674ae819SStefano Zampini   total_counts = 0;
1281674ae819SStefano Zampini   max_size_of_constraint = 0;
1282674ae819SStefano Zampini   for (i=0;i<n_ISForEdges+n_ISForFaces;i++) {
1283674ae819SStefano Zampini     if (i<n_ISForEdges) {
1284674ae819SStefano Zampini       used_IS = &ISForEdges[i];
1285674ae819SStefano Zampini     } else {
1286674ae819SStefano Zampini       used_IS = &ISForFaces[i-n_ISForEdges];
1287674ae819SStefano Zampini     }
1288674ae819SStefano Zampini     ierr = ISGetSize(*used_IS,&j);CHKERRQ(ierr);
1289674ae819SStefano Zampini     total_counts += j;
1290674ae819SStefano Zampini     max_size_of_constraint = PetscMax(j,max_size_of_constraint);
1291674ae819SStefano Zampini   }
1292984c4197SStefano Zampini   total_counts *= max_constraints;
1293674ae819SStefano Zampini   total_counts += n_vertices;
1294674ae819SStefano Zampini   ierr = PetscMalloc(total_counts*sizeof(PetscScalar),&temp_quadrature_constraint);CHKERRQ(ierr);
1295674ae819SStefano Zampini   ierr = PetscMalloc(total_counts*sizeof(PetscInt),&temp_indices_to_constraint);CHKERRQ(ierr);
1296674ae819SStefano Zampini   ierr = PetscMalloc(total_counts*sizeof(PetscInt),&temp_indices_to_constraint_B);CHKERRQ(ierr);
1297984c4197SStefano Zampini   /* local to boundary numbering */
1298674ae819SStefano Zampini   ierr = PetscMalloc(pcis->n*sizeof(PetscInt),&local_to_B);CHKERRQ(ierr);
1299674ae819SStefano Zampini   ierr = ISGetIndices(pcis->is_B_local,(const PetscInt**)&is_indices);CHKERRQ(ierr);
1300984c4197SStefano Zampini   for (i=0;i<pcis->n;i++) local_to_B[i]=-1;
1301984c4197SStefano Zampini   for (i=0;i<pcis->n_B;i++) local_to_B[is_indices[i]]=i;
1302674ae819SStefano Zampini   ierr = ISRestoreIndices(pcis->is_B_local,(const PetscInt**)&is_indices);CHKERRQ(ierr);
1303984c4197SStefano Zampini   /* get local part of global near null space vectors */
1304984c4197SStefano Zampini   ierr = PetscMalloc(nnsp_size*sizeof(Vec),&localnearnullsp);CHKERRQ(ierr);
1305984c4197SStefano Zampini   for (k=0;k<nnsp_size;k++) {
1306984c4197SStefano Zampini     ierr = VecDuplicate(pcis->vec1_N,&localnearnullsp[k]);CHKERRQ(ierr);
1307984c4197SStefano Zampini     ierr = VecScatterBegin(matis->ctx,nearnullvecs[k],localnearnullsp[k],INSERT_VALUES,SCATTER_FORWARD);CHKERRQ(ierr);
1308984c4197SStefano Zampini     ierr = VecScatterEnd(matis->ctx,nearnullvecs[k],localnearnullsp[k],INSERT_VALUES,SCATTER_FORWARD);CHKERRQ(ierr);
1309984c4197SStefano Zampini   }
1310674ae819SStefano Zampini 
1311242a89d7SStefano Zampini   /* whether or not to skip lapack calls */
1312242a89d7SStefano Zampini   skip_lapack = PETSC_TRUE;
1313242a89d7SStefano Zampini   if (n_ISForFaces+n_ISForEdges) skip_lapack = PETSC_FALSE;
1314242a89d7SStefano Zampini 
1315984c4197SStefano Zampini   /* First we issue queries to allocate optimal workspace for LAPACKgesvd (or LAPACKsyev if SVD is missing) */
1316242a89d7SStefano Zampini   if (!pcbddc->use_nnsp_true && !skip_lapack) {
1317674ae819SStefano Zampini     PetscScalar temp_work;
1318674ae819SStefano Zampini #if defined(PETSC_MISSING_LAPACK_GESVD)
1319984c4197SStefano Zampini     /* Proper Orthogonal Decomposition (POD) using the snapshot method */
1320984c4197SStefano Zampini     ierr = PetscMalloc(max_constraints*max_constraints*sizeof(PetscScalar),&correlation_mat);CHKERRQ(ierr);
1321984c4197SStefano Zampini     ierr = PetscMalloc(max_constraints*sizeof(PetscReal),&singular_vals);CHKERRQ(ierr);
1322984c4197SStefano Zampini     ierr = PetscMalloc(max_size_of_constraint*max_constraints*sizeof(PetscScalar),&temp_basis);CHKERRQ(ierr);
1323674ae819SStefano Zampini #if defined(PETSC_USE_COMPLEX)
1324984c4197SStefano Zampini     ierr = PetscMalloc(3*max_constraints*sizeof(PetscReal),&rwork);CHKERRQ(ierr);
1325674ae819SStefano Zampini #endif
1326674ae819SStefano Zampini     /* now we evaluate the optimal workspace using query with lwork=-1 */
1327c8244a33SStefano Zampini     ierr = PetscBLASIntCast(max_constraints,&Blas_N);CHKERRQ(ierr);
1328c8244a33SStefano Zampini     ierr = PetscBLASIntCast(max_constraints,&Blas_LDA);CHKERRQ(ierr);
1329674ae819SStefano Zampini     lwork = -1;
1330674ae819SStefano Zampini     ierr = PetscFPTrapPush(PETSC_FP_TRAP_OFF);CHKERRQ(ierr);
1331674ae819SStefano Zampini #if !defined(PETSC_USE_COMPLEX)
1332c8244a33SStefano Zampini     PetscStackCallBLAS("LAPACKsyev",LAPACKsyev_("V","U",&Blas_N,correlation_mat,&Blas_LDA,singular_vals,&temp_work,&lwork,&lierr));
1333674ae819SStefano Zampini #else
1334c8244a33SStefano Zampini     PetscStackCallBLAS("LAPACKsyev",LAPACKsyev_("V","U",&Blas_N,correlation_mat,&Blas_LDA,singular_vals,&temp_work,&lwork,rwork,&lierr));
1335674ae819SStefano Zampini #endif
1336674ae819SStefano Zampini     ierr = PetscFPTrapPop();CHKERRQ(ierr);
1337984c4197SStefano Zampini     if (lierr) SETERRQ1(PETSC_COMM_SELF,PETSC_ERR_LIB,"Error in query to SYEV Lapack routine %d",(int)lierr);
1338674ae819SStefano Zampini #else /* on missing GESVD */
1339674ae819SStefano Zampini     /* SVD */
1340674ae819SStefano Zampini     PetscInt max_n,min_n;
1341674ae819SStefano Zampini     max_n = max_size_of_constraint;
1342984c4197SStefano Zampini     min_n = max_constraints;
1343984c4197SStefano Zampini     if (max_size_of_constraint < max_constraints) {
1344674ae819SStefano Zampini       min_n = max_size_of_constraint;
1345984c4197SStefano Zampini       max_n = max_constraints;
1346674ae819SStefano Zampini     }
1347674ae819SStefano Zampini     ierr = PetscMalloc(min_n*sizeof(PetscReal),&singular_vals);CHKERRQ(ierr);
1348674ae819SStefano Zampini #if defined(PETSC_USE_COMPLEX)
1349674ae819SStefano Zampini     ierr = PetscMalloc(5*min_n*sizeof(PetscReal),&rwork);CHKERRQ(ierr);
1350674ae819SStefano Zampini #endif
1351674ae819SStefano Zampini     /* now we evaluate the optimal workspace using query with lwork=-1 */
1352674ae819SStefano Zampini     lwork = -1;
1353e310c8b4SStefano Zampini     ierr = PetscBLASIntCast(max_n,&Blas_M);CHKERRQ(ierr);
1354e310c8b4SStefano Zampini     ierr = PetscBLASIntCast(min_n,&Blas_N);CHKERRQ(ierr);
1355b7d8b9f8SStefano Zampini     ierr = PetscBLASIntCast(max_n,&Blas_LDA);CHKERRQ(ierr);
1356674ae819SStefano Zampini     ierr = PetscFPTrapPush(PETSC_FP_TRAP_OFF);CHKERRQ(ierr);
1357674ae819SStefano Zampini #if !defined(PETSC_USE_COMPLEX)
1358e310c8b4SStefano Zampini     PetscStackCallBLAS("LAPACKgesvd",LAPACKgesvd_("O","N",&Blas_M,&Blas_N,&temp_quadrature_constraint[0],&Blas_LDA,singular_vals,&dummy_scalar_1,&dummy_int_1,&dummy_scalar_2,&dummy_int_2,&temp_work,&lwork,&lierr));
1359674ae819SStefano Zampini #else
1360e310c8b4SStefano Zampini     PetscStackCallBLAS("LAPACKgesvd",LAPACKgesvd_("O","N",&Blas_M,&Blas_N,&temp_quadrature_constraint[0],&Blas_LDA,singular_vals,&dummy_scalar_1,&dummy_int_1,&dummy_scalar_2,&dummy_int_2,&temp_work,&lwork,rwork,&lierr));
1361674ae819SStefano Zampini #endif
1362674ae819SStefano Zampini     ierr = PetscFPTrapPop();CHKERRQ(ierr);
1363984c4197SStefano Zampini     if (lierr) SETERRQ1(PETSC_COMM_SELF,PETSC_ERR_LIB,"Error in query to GESVD Lapack routine %d",(int)lierr);
1364984c4197SStefano Zampini #endif /* on missing GESVD */
1365674ae819SStefano Zampini     /* Allocate optimal workspace */
1366674ae819SStefano Zampini     ierr = PetscBLASIntCast((PetscInt)PetscRealPart(temp_work),&lwork);CHKERRQ(ierr);
1367984c4197SStefano Zampini     ierr = PetscMalloc((PetscInt)lwork*sizeof(PetscScalar),&work);CHKERRQ(ierr);
1368674ae819SStefano Zampini   }
1369674ae819SStefano Zampini   /* Now we can loop on constraining sets */
1370674ae819SStefano Zampini   total_counts = 0;
1371674ae819SStefano Zampini   temp_indices[0] = 0;
1372674ae819SStefano Zampini   /* vertices */
1373674ae819SStefano Zampini   if (ISForVertices) {
1374674ae819SStefano Zampini     ierr = ISGetIndices(ISForVertices,(const PetscInt**)&is_indices);CHKERRQ(ierr);
1375674ae819SStefano Zampini     if (nnsp_has_cnst) { /* consider all vertices */
1376674ae819SStefano Zampini       for (i=0;i<n_vertices;i++) {
1377674ae819SStefano Zampini         temp_indices_to_constraint[temp_indices[total_counts]]=is_indices[i];
1378674ae819SStefano Zampini         temp_indices_to_constraint_B[temp_indices[total_counts]]=local_to_B[is_indices[i]];
1379674ae819SStefano Zampini         temp_quadrature_constraint[temp_indices[total_counts]]=1.0;
1380674ae819SStefano Zampini         temp_indices[total_counts+1]=temp_indices[total_counts]+1;
1381674ae819SStefano Zampini         change_basis[total_counts]=PETSC_FALSE;
1382674ae819SStefano Zampini         total_counts++;
1383674ae819SStefano Zampini       }
1384674ae819SStefano Zampini     } else { /* consider vertices for which exist at least a localnearnullsp which is not null there */
1385984c4197SStefano Zampini       PetscBool used_vertex;
1386674ae819SStefano Zampini       for (i=0;i<n_vertices;i++) {
1387674ae819SStefano Zampini         used_vertex = PETSC_FALSE;
1388674ae819SStefano Zampini         k = 0;
1389674ae819SStefano Zampini         while (!used_vertex && k<nnsp_size) {
1390984c4197SStefano Zampini           ierr = VecGetArrayRead(localnearnullsp[k],(const PetscScalar**)&array);CHKERRQ(ierr);
1391984c4197SStefano Zampini           if (PetscAbsScalar(array[is_indices[i]])>0.0) {
1392674ae819SStefano Zampini             temp_indices_to_constraint[temp_indices[total_counts]]=is_indices[i];
1393674ae819SStefano Zampini             temp_indices_to_constraint_B[temp_indices[total_counts]]=local_to_B[is_indices[i]];
1394674ae819SStefano Zampini             temp_quadrature_constraint[temp_indices[total_counts]]=1.0;
1395674ae819SStefano Zampini             temp_indices[total_counts+1]=temp_indices[total_counts]+1;
1396674ae819SStefano Zampini             change_basis[total_counts]=PETSC_FALSE;
1397674ae819SStefano Zampini             total_counts++;
1398674ae819SStefano Zampini             used_vertex = PETSC_TRUE;
1399674ae819SStefano Zampini           }
1400984c4197SStefano Zampini           ierr = VecRestoreArrayRead(localnearnullsp[k],(const PetscScalar**)&array);CHKERRQ(ierr);
1401674ae819SStefano Zampini           k++;
1402674ae819SStefano Zampini         }
1403674ae819SStefano Zampini       }
1404674ae819SStefano Zampini     }
1405674ae819SStefano Zampini     ierr = ISRestoreIndices(ISForVertices,(const PetscInt**)&is_indices);CHKERRQ(ierr);
1406674ae819SStefano Zampini     n_vertices = total_counts;
1407674ae819SStefano Zampini   }
1408984c4197SStefano Zampini 
1409674ae819SStefano Zampini   /* edges and faces */
1410674ae819SStefano Zampini   for (i=0;i<n_ISForEdges+n_ISForFaces;i++) {
1411674ae819SStefano Zampini     if (i<n_ISForEdges) {
1412674ae819SStefano Zampini       used_IS = &ISForEdges[i];
1413984c4197SStefano Zampini       boolforchange = pcbddc->use_change_of_basis; /* change or not the basis on the edge */
1414674ae819SStefano Zampini     } else {
1415674ae819SStefano Zampini       used_IS = &ISForFaces[i-n_ISForEdges];
1416984c4197SStefano Zampini       boolforchange = (PetscBool)(pcbddc->use_change_of_basis && pcbddc->use_change_on_faces); /* change or not the basis on the face */
1417674ae819SStefano Zampini     }
1418674ae819SStefano Zampini     temp_constraints = 0;          /* zero the number of constraints I have on this conn comp */
1419674ae819SStefano Zampini     temp_start_ptr = total_counts; /* need to know the starting index of constraints stored */
1420674ae819SStefano Zampini     ierr = ISGetSize(*used_IS,&size_of_constraint);CHKERRQ(ierr);
1421674ae819SStefano Zampini     ierr = ISGetIndices(*used_IS,(const PetscInt**)&is_indices);CHKERRQ(ierr);
1422984c4197SStefano Zampini     /* change of basis should not be performed on local periodic nodes */
1423984c4197SStefano Zampini     if (pcbddc->mat_graph->mirrors && pcbddc->mat_graph->mirrors[is_indices[0]]) boolforchange = PETSC_FALSE;
1424674ae819SStefano Zampini     if (nnsp_has_cnst) {
14255b08dc53SStefano Zampini       PetscScalar quad_value;
1426674ae819SStefano Zampini       temp_constraints++;
1427674ae819SStefano Zampini       quad_value = (PetscScalar)(1.0/PetscSqrtReal((PetscReal)size_of_constraint));
1428674ae819SStefano Zampini       for (j=0;j<size_of_constraint;j++) {
1429674ae819SStefano Zampini         temp_indices_to_constraint[temp_indices[total_counts]+j]=is_indices[j];
1430674ae819SStefano Zampini         temp_indices_to_constraint_B[temp_indices[total_counts]+j]=local_to_B[is_indices[j]];
1431674ae819SStefano Zampini         temp_quadrature_constraint[temp_indices[total_counts]+j]=quad_value;
1432674ae819SStefano Zampini       }
1433674ae819SStefano Zampini       temp_indices[total_counts+1]=temp_indices[total_counts]+size_of_constraint;  /* store new starting point */
143451b0f6cfSStefano Zampini       change_basis[total_counts]=boolforchange;
1435674ae819SStefano Zampini       total_counts++;
1436674ae819SStefano Zampini     }
1437674ae819SStefano Zampini     for (k=0;k<nnsp_size;k++) {
1438984c4197SStefano Zampini       PetscReal real_value;
1439984c4197SStefano Zampini       ierr = VecGetArrayRead(localnearnullsp[k],(const PetscScalar**)&array);CHKERRQ(ierr);
1440674ae819SStefano Zampini       for (j=0;j<size_of_constraint;j++) {
1441674ae819SStefano Zampini         temp_indices_to_constraint[temp_indices[total_counts]+j]=is_indices[j];
1442674ae819SStefano Zampini         temp_indices_to_constraint_B[temp_indices[total_counts]+j]=local_to_B[is_indices[j]];
1443984c4197SStefano Zampini         temp_quadrature_constraint[temp_indices[total_counts]+j]=array[is_indices[j]];
1444674ae819SStefano Zampini       }
1445984c4197SStefano Zampini       ierr = VecRestoreArrayRead(localnearnullsp[k],(const PetscScalar**)&array);CHKERRQ(ierr);
1446984c4197SStefano Zampini       /* check if array is null on the connected component */
1447e310c8b4SStefano Zampini       ierr = PetscBLASIntCast(size_of_constraint,&Blas_N);CHKERRQ(ierr);
1448e310c8b4SStefano Zampini       PetscStackCallBLAS("BLASasum",real_value = BLASasum_(&Blas_N,&temp_quadrature_constraint[temp_indices[total_counts]],&Blas_one));
14495b08dc53SStefano Zampini       if (real_value > 0.0) { /* keep indices and values */
1450674ae819SStefano Zampini         temp_constraints++;
1451674ae819SStefano Zampini         temp_indices[total_counts+1]=temp_indices[total_counts]+size_of_constraint;  /* store new starting point */
145251b0f6cfSStefano Zampini         change_basis[total_counts]=boolforchange;
1453674ae819SStefano Zampini         total_counts++;
1454674ae819SStefano Zampini       }
1455674ae819SStefano Zampini     }
1456674ae819SStefano Zampini     ierr = ISRestoreIndices(*used_IS,(const PetscInt**)&is_indices);CHKERRQ(ierr);
1457984c4197SStefano Zampini     /* perform SVD on the constraints if use_nnsp_true has not be requested by the user */
1458984c4197SStefano Zampini     if (!pcbddc->use_nnsp_true) {
1459984c4197SStefano Zampini       PetscReal tol = 1.0e-8; /* tolerance for retaining eigenmodes */
1460674ae819SStefano Zampini 
1461674ae819SStefano Zampini #if defined(PETSC_MISSING_LAPACK_GESVD)
1462984c4197SStefano Zampini       /* SVD: Y = U*S*V^H                -> U (eigenvectors of Y*Y^H) = Y*V*(S)^\dag
1463984c4197SStefano Zampini          POD: Y^H*Y = V*D*V^H, D = S^H*S -> U = Y*V*D^(-1/2)
1464984c4197SStefano Zampini          -> When PETSC_USE_COMPLEX and PETSC_MISSING_LAPACK_GESVD are defined
1465984c4197SStefano Zampini             the constraints basis will differ (by a complex factor with absolute value equal to 1)
1466984c4197SStefano Zampini             from that computed using LAPACKgesvd
1467984c4197SStefano Zampini          -> This is due to a different computation of eigenvectors in LAPACKheev
1468984c4197SStefano Zampini          -> The quality of the POD-computed basis will be the same */
1469984c4197SStefano Zampini       ierr = PetscMemzero(correlation_mat,temp_constraints*temp_constraints*sizeof(PetscScalar));CHKERRQ(ierr);
1470674ae819SStefano Zampini       /* Store upper triangular part of correlation matrix */
1471e310c8b4SStefano Zampini       ierr = PetscBLASIntCast(size_of_constraint,&Blas_N);CHKERRQ(ierr);
1472984c4197SStefano Zampini       ierr = PetscFPTrapPush(PETSC_FP_TRAP_OFF);CHKERRQ(ierr);
1473674ae819SStefano Zampini       for (j=0;j<temp_constraints;j++) {
1474674ae819SStefano Zampini         for (k=0;k<j+1;k++) {
1475e310c8b4SStefano Zampini           PetscStackCallBLAS("BLASdot",correlation_mat[j*temp_constraints+k]=BLASdot_(&Blas_N,&temp_quadrature_constraint[temp_indices[temp_start_ptr+k]],&Blas_one,&temp_quadrature_constraint[temp_indices[temp_start_ptr+j]],&Blas_one_2));
1476674ae819SStefano Zampini         }
1477674ae819SStefano Zampini       }
1478e310c8b4SStefano Zampini       /* compute eigenvalues and eigenvectors of correlation matrix */
1479e310c8b4SStefano Zampini       ierr = PetscBLASIntCast(temp_constraints,&Blas_N);CHKERRQ(ierr);
1480e310c8b4SStefano Zampini       ierr = PetscBLASIntCast(temp_constraints,&Blas_LDA);CHKERRQ(ierr);
1481674ae819SStefano Zampini #if !defined(PETSC_USE_COMPLEX)
1482c8244a33SStefano Zampini       PetscStackCallBLAS("LAPACKsyev",LAPACKsyev_("V","U",&Blas_N,correlation_mat,&Blas_LDA,singular_vals,work,&lwork,&lierr));
1483674ae819SStefano Zampini #else
1484c8244a33SStefano Zampini       PetscStackCallBLAS("LAPACKsyev",LAPACKsyev_("V","U",&Blas_N,correlation_mat,&Blas_LDA,singular_vals,work,&lwork,rwork,&lierr));
1485674ae819SStefano Zampini #endif
1486674ae819SStefano Zampini       ierr = PetscFPTrapPop();CHKERRQ(ierr);
1487984c4197SStefano Zampini       if (lierr) SETERRQ1(PETSC_COMM_SELF,PETSC_ERR_LIB,"Error in SYEV Lapack routine %d",(int)lierr);
1488984c4197SStefano Zampini       /* retain eigenvalues greater than tol: note that LAPACKsyev gives eigs in ascending order */
1489674ae819SStefano Zampini       j=0;
1490984c4197SStefano Zampini       while (j < temp_constraints && singular_vals[j] < tol) j++;
1491674ae819SStefano Zampini       total_counts=total_counts-j;
1492e310c8b4SStefano Zampini       /* scale and copy POD basis into used quadrature memory */
1493c4303822SStefano Zampini       ierr = PetscBLASIntCast(size_of_constraint,&Blas_M);CHKERRQ(ierr);
1494c4303822SStefano Zampini       ierr = PetscBLASIntCast(temp_constraints,&Blas_N);CHKERRQ(ierr);
1495c4303822SStefano Zampini       ierr = PetscBLASIntCast(temp_constraints,&Blas_K);CHKERRQ(ierr);
1496c4303822SStefano Zampini       ierr = PetscBLASIntCast(size_of_constraint,&Blas_LDA);CHKERRQ(ierr);
1497c4303822SStefano Zampini       ierr = PetscBLASIntCast(temp_constraints,&Blas_LDB);CHKERRQ(ierr);
1498c4303822SStefano Zampini       ierr = PetscBLASIntCast(size_of_constraint,&Blas_LDC);CHKERRQ(ierr);
1499674ae819SStefano Zampini       if (j<temp_constraints) {
1500984c4197SStefano Zampini         PetscInt ii;
1501984c4197SStefano Zampini         for (k=j;k<temp_constraints;k++) singular_vals[k]=1.0/PetscSqrtReal(singular_vals[k]);
1502674ae819SStefano Zampini         ierr = PetscFPTrapPush(PETSC_FP_TRAP_OFF);CHKERRQ(ierr);
1503c4303822SStefano Zampini         PetscStackCallBLAS("BLASgemm",BLASgemm_("N","N",&Blas_M,&Blas_N,&Blas_K,&one,&temp_quadrature_constraint[temp_indices[temp_start_ptr]],&Blas_LDA,correlation_mat,&Blas_LDB,&zero,temp_basis,&Blas_LDC));
1504674ae819SStefano Zampini         ierr = PetscFPTrapPop();CHKERRQ(ierr);
1505984c4197SStefano Zampini         for (k=0;k<temp_constraints-j;k++) {
1506674ae819SStefano Zampini           for (ii=0;ii<size_of_constraint;ii++) {
1507984c4197SStefano Zampini             temp_quadrature_constraint[temp_indices[temp_start_ptr+k]+ii]=singular_vals[temp_constraints-1-k]*temp_basis[(temp_constraints-1-k)*size_of_constraint+ii];
1508674ae819SStefano Zampini           }
1509674ae819SStefano Zampini         }
1510674ae819SStefano Zampini       }
1511674ae819SStefano Zampini #else  /* on missing GESVD */
1512e310c8b4SStefano Zampini       ierr = PetscBLASIntCast(size_of_constraint,&Blas_M);CHKERRQ(ierr);
1513e310c8b4SStefano Zampini       ierr = PetscBLASIntCast(temp_constraints,&Blas_N);CHKERRQ(ierr);
1514b7d8b9f8SStefano Zampini       ierr = PetscBLASIntCast(size_of_constraint,&Blas_LDA);CHKERRQ(ierr);
1515674ae819SStefano Zampini       ierr = PetscFPTrapPush(PETSC_FP_TRAP_OFF);CHKERRQ(ierr);
1516674ae819SStefano Zampini #if !defined(PETSC_USE_COMPLEX)
1517e310c8b4SStefano Zampini       PetscStackCallBLAS("LAPACKgesvd",LAPACKgesvd_("O","N",&Blas_M,&Blas_N,&temp_quadrature_constraint[temp_indices[temp_start_ptr]],&Blas_LDA,singular_vals,&dummy_scalar_1,&dummy_int_1,&dummy_scalar_2,&dummy_int_2,work,&lwork,&lierr));
1518674ae819SStefano Zampini #else
1519e310c8b4SStefano Zampini       PetscStackCallBLAS("LAPACKgesvd",LAPACKgesvd_("O","N",&Blas_M,&Blas_N,&temp_quadrature_constraint[temp_indices[temp_start_ptr]],&Blas_LDA,singular_vals,&dummy_scalar_1,&dummy_int_1,&dummy_scalar_2,&dummy_int_2,work,&lwork,rwork,&lierr));
1520674ae819SStefano Zampini #endif
1521984c4197SStefano Zampini       if (lierr) SETERRQ1(PETSC_COMM_SELF,PETSC_ERR_LIB,"Error in GESVD Lapack routine %d",(int)lierr);
1522674ae819SStefano Zampini       ierr = PetscFPTrapPop();CHKERRQ(ierr);
1523984c4197SStefano Zampini       /* retain eigenvalues greater than tol: note that LAPACKgesvd gives eigs in descending order */
1524e310c8b4SStefano Zampini       k = temp_constraints;
1525e310c8b4SStefano Zampini       if (k > size_of_constraint) k = size_of_constraint;
1526674ae819SStefano Zampini       j = 0;
1527e310c8b4SStefano Zampini       while (j < k && singular_vals[k-j-1] < tol) j++;
1528e310c8b4SStefano Zampini       total_counts = total_counts-temp_constraints+k-j;
1529984c4197SStefano Zampini #endif /* on missing GESVD */
1530674ae819SStefano Zampini     }
1531674ae819SStefano Zampini   }
1532674ae819SStefano Zampini   /* free index sets of faces, edges and vertices */
1533674ae819SStefano Zampini   for (i=0;i<n_ISForFaces;i++) {
1534674ae819SStefano Zampini     ierr = ISDestroy(&ISForFaces[i]);CHKERRQ(ierr);
1535674ae819SStefano Zampini   }
1536674ae819SStefano Zampini   ierr = PetscFree(ISForFaces);CHKERRQ(ierr);
1537674ae819SStefano Zampini   for (i=0;i<n_ISForEdges;i++) {
1538674ae819SStefano Zampini     ierr = ISDestroy(&ISForEdges[i]);CHKERRQ(ierr);
1539674ae819SStefano Zampini   }
1540674ae819SStefano Zampini   ierr = PetscFree(ISForEdges);CHKERRQ(ierr);
1541674ae819SStefano Zampini   ierr = ISDestroy(&ISForVertices);CHKERRQ(ierr);
1542674ae819SStefano Zampini 
1543984c4197SStefano Zampini   /* free workspace */
1544242a89d7SStefano Zampini   if (!pcbddc->use_nnsp_true && !skip_lapack) {
1545984c4197SStefano Zampini     ierr = PetscFree(work);CHKERRQ(ierr);
1546984c4197SStefano Zampini #if defined(PETSC_USE_COMPLEX)
1547984c4197SStefano Zampini     ierr = PetscFree(rwork);CHKERRQ(ierr);
1548984c4197SStefano Zampini #endif
1549984c4197SStefano Zampini     ierr = PetscFree(singular_vals);CHKERRQ(ierr);
1550984c4197SStefano Zampini #if defined(PETSC_MISSING_LAPACK_GESVD)
1551984c4197SStefano Zampini     ierr = PetscFree(correlation_mat);CHKERRQ(ierr);
1552984c4197SStefano Zampini     ierr = PetscFree(temp_basis);CHKERRQ(ierr);
1553984c4197SStefano Zampini #endif
1554984c4197SStefano Zampini   }
1555984c4197SStefano Zampini   for (k=0;k<nnsp_size;k++) {
1556984c4197SStefano Zampini     ierr = VecDestroy(&localnearnullsp[k]);CHKERRQ(ierr);
1557984c4197SStefano Zampini   }
1558984c4197SStefano Zampini   ierr = PetscFree(localnearnullsp);CHKERRQ(ierr);
1559984c4197SStefano Zampini 
1560674ae819SStefano Zampini   /* set quantities in pcbddc data structure */
1561984c4197SStefano Zampini   /* n_vertices defines the number of subdomain corners in the primal space */
1562674ae819SStefano Zampini   /* n_constraints defines the number of averages (they can be point primal dofs if change of basis is requested) */
1563984c4197SStefano Zampini   pcbddc->local_primal_size = total_counts;
1564674ae819SStefano Zampini   pcbddc->n_vertices = n_vertices;
1565984c4197SStefano Zampini   pcbddc->n_constraints = pcbddc->local_primal_size-pcbddc->n_vertices;
1566674ae819SStefano Zampini 
1567674ae819SStefano Zampini   /* Create constraint matrix */
1568674ae819SStefano Zampini   /* The constraint matrix is used to compute the l2g map of primal dofs */
1569674ae819SStefano Zampini   /* so we need to set it up properly either with or without change of basis */
1570674ae819SStefano Zampini   ierr = MatCreate(PETSC_COMM_SELF,&pcbddc->ConstraintMatrix);CHKERRQ(ierr);
1571674ae819SStefano Zampini   ierr = MatSetType(pcbddc->ConstraintMatrix,impMatType);CHKERRQ(ierr);
1572984c4197SStefano Zampini   ierr = MatSetSizes(pcbddc->ConstraintMatrix,pcbddc->local_primal_size,pcis->n,pcbddc->local_primal_size,pcis->n);CHKERRQ(ierr);
1573984c4197SStefano Zampini   /* array to compute a local numbering of constraints : vertices first then constraints */
1574984c4197SStefano Zampini   ierr = PetscMalloc(pcbddc->local_primal_size*sizeof(PetscInt),&aux_primal_numbering);CHKERRQ(ierr);
1575984c4197SStefano Zampini   /* array to select the proper local node (of minimum index with respect to global ordering) when changing the basis */
1576984c4197SStefano Zampini   /* note: it should not be needed since IS for faces and edges are already sorted by global ordering when analyzing the graph but... just in case */
1577984c4197SStefano Zampini   ierr = PetscMalloc(pcbddc->local_primal_size*sizeof(PetscInt),&aux_primal_minloc);CHKERRQ(ierr);
1578984c4197SStefano Zampini   /* auxiliary stuff for basis change */
1579984c4197SStefano Zampini   ierr = PetscMalloc(max_size_of_constraint*sizeof(PetscInt),&global_indices);CHKERRQ(ierr);
1580984c4197SStefano Zampini   ierr = PetscMalloc(pcis->n_B*sizeof(PetscBool),&touched);CHKERRQ(ierr);
1581984c4197SStefano Zampini   ierr = PetscMemzero(touched,pcis->n_B*sizeof(PetscBool));CHKERRQ(ierr);
1582984c4197SStefano Zampini 
1583984c4197SStefano Zampini   /* find primal_dofs: subdomain corners plus dofs selected as primal after change of basis */
1584984c4197SStefano Zampini   total_primal_vertices=0;
1585984c4197SStefano Zampini   for (i=0;i<pcbddc->local_primal_size;i++) {
1586674ae819SStefano Zampini     size_of_constraint=temp_indices[i+1]-temp_indices[i];
1587984c4197SStefano Zampini     if (size_of_constraint == 1) {
1588984c4197SStefano Zampini       touched[temp_indices_to_constraint_B[temp_indices[i]]]=PETSC_TRUE;
1589984c4197SStefano Zampini       aux_primal_numbering[total_primal_vertices]=temp_indices_to_constraint[temp_indices[i]];
1590984c4197SStefano Zampini       aux_primal_minloc[total_primal_vertices]=0;
1591984c4197SStefano Zampini       total_primal_vertices++;
1592984c4197SStefano Zampini     } else if (change_basis[i]) { /* Same procedure used in PCBDDCGetPrimalConstraintsLocalIdx */
1593984c4197SStefano Zampini       PetscInt min_loc,min_index;
1594984c4197SStefano Zampini       ierr = ISLocalToGlobalMappingApply(pcbddc->mat_graph->l2gmap,size_of_constraint,&temp_indices_to_constraint[temp_indices[i]],global_indices);CHKERRQ(ierr);
1595984c4197SStefano Zampini       /* find first untouched local node */
1596674ae819SStefano Zampini       k = 0;
1597984c4197SStefano Zampini       while (touched[temp_indices_to_constraint_B[temp_indices[i]+k]]) k++;
1598984c4197SStefano Zampini       min_index = global_indices[k];
1599984c4197SStefano Zampini       min_loc = k;
1600984c4197SStefano Zampini       /* search the minimum among global nodes already untouched on the cc */
1601984c4197SStefano Zampini       for (k=1;k<size_of_constraint;k++) {
1602984c4197SStefano Zampini         /* there can be more than one constraint on a single connected component */
1603984c4197SStefano Zampini         if (min_index > global_indices[k] && !touched[temp_indices_to_constraint_B[temp_indices[i]+k]]) {
1604984c4197SStefano Zampini           min_index = global_indices[k];
1605984c4197SStefano Zampini           min_loc = k;
1606674ae819SStefano Zampini         }
1607674ae819SStefano Zampini       }
1608984c4197SStefano Zampini       touched[temp_indices_to_constraint_B[temp_indices[i]+min_loc]] = PETSC_TRUE;
1609984c4197SStefano Zampini       aux_primal_numbering[total_primal_vertices]=temp_indices_to_constraint[temp_indices[i]+min_loc];
1610984c4197SStefano Zampini       aux_primal_minloc[total_primal_vertices]=min_loc;
1611984c4197SStefano Zampini       total_primal_vertices++;
1612984c4197SStefano Zampini     }
1613984c4197SStefano Zampini   }
1614984c4197SStefano Zampini   /* free workspace */
1615984c4197SStefano Zampini   ierr = PetscFree(global_indices);CHKERRQ(ierr);
1616984c4197SStefano Zampini   ierr = PetscFree(touched);CHKERRQ(ierr);
1617674ae819SStefano Zampini   /* permute indices in order to have a sorted set of vertices */
1618984c4197SStefano Zampini   ierr = PetscSortInt(total_primal_vertices,aux_primal_numbering);
1619984c4197SStefano Zampini 
1620984c4197SStefano Zampini   /* nonzero structure of constraint matrix */
1621984c4197SStefano Zampini   ierr = PetscMalloc(pcbddc->local_primal_size*sizeof(PetscInt),&nnz);CHKERRQ(ierr);
1622984c4197SStefano Zampini   for (i=0;i<total_primal_vertices;i++) nnz[i]=1;
1623984c4197SStefano Zampini   j=total_primal_vertices;
1624984c4197SStefano Zampini   for (i=pcbddc->n_vertices;i<pcbddc->local_primal_size;i++) {
1625674ae819SStefano Zampini     if (!change_basis[i]) {
1626674ae819SStefano Zampini       nnz[j]=temp_indices[i+1]-temp_indices[i];
1627674ae819SStefano Zampini       j++;
1628674ae819SStefano Zampini     }
1629674ae819SStefano Zampini   }
1630674ae819SStefano Zampini   ierr = MatSeqAIJSetPreallocation(pcbddc->ConstraintMatrix,0,nnz);CHKERRQ(ierr);
1631674ae819SStefano Zampini   ierr = PetscFree(nnz);CHKERRQ(ierr);
1632674ae819SStefano Zampini   /* set values in constraint matrix */
1633984c4197SStefano Zampini   for (i=0;i<total_primal_vertices;i++) {
1634984c4197SStefano Zampini     ierr = MatSetValue(pcbddc->ConstraintMatrix,i,aux_primal_numbering[i],1.0,INSERT_VALUES);CHKERRQ(ierr);
1635674ae819SStefano Zampini   }
1636984c4197SStefano Zampini   total_counts = total_primal_vertices;
1637984c4197SStefano Zampini   for (i=pcbddc->n_vertices;i<pcbddc->local_primal_size;i++) {
1638674ae819SStefano Zampini     if (!change_basis[i]) {
1639674ae819SStefano Zampini       size_of_constraint=temp_indices[i+1]-temp_indices[i];
1640674ae819SStefano Zampini       ierr = MatSetValues(pcbddc->ConstraintMatrix,1,&total_counts,size_of_constraint,&temp_indices_to_constraint[temp_indices[i]],&temp_quadrature_constraint[temp_indices[i]],INSERT_VALUES);CHKERRQ(ierr);
1641674ae819SStefano Zampini       total_counts++;
1642674ae819SStefano Zampini     }
1643674ae819SStefano Zampini   }
1644674ae819SStefano Zampini   /* assembling */
1645674ae819SStefano Zampini   ierr = MatAssemblyBegin(pcbddc->ConstraintMatrix,MAT_FINAL_ASSEMBLY);CHKERRQ(ierr);
1646674ae819SStefano Zampini   ierr = MatAssemblyEnd(pcbddc->ConstraintMatrix,MAT_FINAL_ASSEMBLY);CHKERRQ(ierr);
1647984c4197SStefano Zampini   /*
1648984c4197SStefano Zampini   ierr = MatView(pcbddc->ConstraintMatrix,(PetscViewer)0);CHKERRQ(ierr);
1649984c4197SStefano Zampini   */
1650674ae819SStefano Zampini   /* Create matrix for change of basis. We don't need it in case pcbddc->use_change_of_basis is FALSE */
1651674ae819SStefano Zampini   if (pcbddc->use_change_of_basis) {
1652984c4197SStefano Zampini     PetscBool qr_needed = PETSC_FALSE;
1653984c4197SStefano Zampini     /* change of basis acts on local interfaces -> dimension is n_B x n_B */
1654674ae819SStefano Zampini     ierr = MatCreate(PETSC_COMM_SELF,&pcbddc->ChangeOfBasisMatrix);CHKERRQ(ierr);
1655674ae819SStefano Zampini     ierr = MatSetType(pcbddc->ChangeOfBasisMatrix,impMatType);CHKERRQ(ierr);
1656674ae819SStefano Zampini     ierr = MatSetSizes(pcbddc->ChangeOfBasisMatrix,pcis->n_B,pcis->n_B,pcis->n_B,pcis->n_B);CHKERRQ(ierr);
1657674ae819SStefano Zampini     /* work arrays */
1658674ae819SStefano Zampini     ierr = PetscMalloc(pcis->n_B*sizeof(PetscInt),&nnz);CHKERRQ(ierr);
1659984c4197SStefano Zampini     for (i=0;i<pcis->n_B;i++) nnz[i]=1;
1660984c4197SStefano Zampini     /* nonzeros per row */
1661984c4197SStefano Zampini     for (i=pcbddc->n_vertices;i<pcbddc->local_primal_size;i++) {
1662674ae819SStefano Zampini       if (change_basis[i]) {
1663984c4197SStefano Zampini         qr_needed = PETSC_TRUE;
1664674ae819SStefano Zampini         size_of_constraint = temp_indices[i+1]-temp_indices[i];
1665984c4197SStefano Zampini         for (j=0;j<size_of_constraint;j++) nnz[temp_indices_to_constraint_B[temp_indices[i]+j]] = size_of_constraint;
1666674ae819SStefano Zampini       }
1667674ae819SStefano Zampini     }
1668674ae819SStefano Zampini     ierr = MatSeqAIJSetPreallocation(pcbddc->ChangeOfBasisMatrix,0,nnz);CHKERRQ(ierr);
1669674ae819SStefano Zampini     ierr = PetscFree(nnz);CHKERRQ(ierr);
1670674ae819SStefano Zampini     /* Set initial identity in the matrix */
1671674ae819SStefano Zampini     for (i=0;i<pcis->n_B;i++) {
1672674ae819SStefano Zampini       ierr = MatSetValue(pcbddc->ChangeOfBasisMatrix,i,i,1.0,INSERT_VALUES);CHKERRQ(ierr);
1673674ae819SStefano Zampini     }
1674984c4197SStefano Zampini 
1675674ae819SStefano Zampini     /* Now we loop on the constraints which need a change of basis */
1676674ae819SStefano Zampini     /* Change of basis matrix is evaluated as the FIRST APPROACH in */
1677984c4197SStefano Zampini     /* Klawonn and Widlund, Dual-primal FETI-DP methods for linear elasticity, (see Sect 6.2.1) */
1678984c4197SStefano Zampini     /* Change of basis matrix T computed via QR decomposition of constraints */
1679984c4197SStefano Zampini     if (qr_needed) {
1680026de310SStefano Zampini       /* dual and primal dofs on a single cc */
1681984c4197SStefano Zampini       PetscInt     dual_dofs,primal_dofs;
1682026de310SStefano Zampini       /* iterator on aux_primal_minloc (ordered as read from nearnullspace: vertices, edges and then constraints) */
1683026de310SStefano Zampini       PetscInt     primal_counter;
1684984c4197SStefano Zampini       /* working stuff for GEQRF */
1685984c4197SStefano Zampini       PetscScalar  *qr_basis,*qr_tau,*qr_work,lqr_work_t;
1686984c4197SStefano Zampini       PetscBLASInt lqr_work;
1687984c4197SStefano Zampini       /* working stuff for UNGQR */
1688984c4197SStefano Zampini       PetscScalar  *gqr_work,lgqr_work_t;
1689984c4197SStefano Zampini       PetscBLASInt lgqr_work;
1690984c4197SStefano Zampini       /* working stuff for TRTRS */
1691984c4197SStefano Zampini       PetscScalar  *trs_rhs;
16923f08241aSStefano Zampini       PetscBLASInt Blas_NRHS;
1693984c4197SStefano Zampini       /* pointers for values insertion into change of basis matrix */
1694984c4197SStefano Zampini       PetscInt     *start_rows,*start_cols;
1695984c4197SStefano Zampini       PetscScalar  *start_vals;
1696984c4197SStefano Zampini       /* working stuff for values insertion */
1697984c4197SStefano Zampini       PetscBool    *is_primal;
1698984c4197SStefano Zampini 
1699984c4197SStefano Zampini       /* space to store Q */
1700984c4197SStefano Zampini       ierr = PetscMalloc((max_size_of_constraint)*(max_size_of_constraint)*sizeof(PetscScalar),&qr_basis);CHKERRQ(ierr);
1701984c4197SStefano Zampini       /* first we issue queries for optimal work */
17023f08241aSStefano Zampini       ierr = PetscBLASIntCast(max_size_of_constraint,&Blas_M);CHKERRQ(ierr);
17033f08241aSStefano Zampini       ierr = PetscBLASIntCast(max_constraints,&Blas_N);CHKERRQ(ierr);
17043f08241aSStefano Zampini       ierr = PetscBLASIntCast(max_size_of_constraint,&Blas_LDA);CHKERRQ(ierr);
1705984c4197SStefano Zampini       lqr_work = -1;
17063f08241aSStefano Zampini       PetscStackCallBLAS("LAPACKgeqrf",LAPACKgeqrf_(&Blas_M,&Blas_N,qr_basis,&Blas_LDA,qr_tau,&lqr_work_t,&lqr_work,&lierr));
1707984c4197SStefano Zampini       if (lierr) SETERRQ1(PETSC_COMM_SELF,PETSC_ERR_LIB,"Error in query to GEQRF Lapack routine %d",(int)lierr);
1708984c4197SStefano Zampini       ierr = PetscBLASIntCast((PetscInt)PetscRealPart(lqr_work_t),&lqr_work);CHKERRQ(ierr);
1709984c4197SStefano Zampini       ierr = PetscMalloc((PetscInt)PetscRealPart(lqr_work_t)*sizeof(*qr_work),&qr_work);CHKERRQ(ierr);
1710984c4197SStefano Zampini       lgqr_work = -1;
17113f08241aSStefano Zampini       ierr = PetscBLASIntCast(max_size_of_constraint,&Blas_M);CHKERRQ(ierr);
17123f08241aSStefano Zampini       ierr = PetscBLASIntCast(max_size_of_constraint,&Blas_N);CHKERRQ(ierr);
17133f08241aSStefano Zampini       ierr = PetscBLASIntCast(max_constraints,&Blas_K);CHKERRQ(ierr);
17143f08241aSStefano Zampini       ierr = PetscBLASIntCast(max_size_of_constraint,&Blas_LDA);CHKERRQ(ierr);
17153f08241aSStefano Zampini       if (Blas_K>Blas_M) Blas_K=Blas_M; /* adjust just for computing optimal work */
17163f08241aSStefano Zampini       PetscStackCallBLAS("LAPACKungqr",LAPACKungqr_(&Blas_M,&Blas_N,&Blas_K,qr_basis,&Blas_LDA,qr_tau,&lgqr_work_t,&lgqr_work,&lierr));
1717984c4197SStefano Zampini       if (lierr) SETERRQ1(PETSC_COMM_SELF,PETSC_ERR_LIB,"Error in query to UNGQR Lapack routine %d",(int)lierr);
1718984c4197SStefano Zampini       ierr = PetscBLASIntCast((PetscInt)PetscRealPart(lgqr_work_t),&lgqr_work);CHKERRQ(ierr);
1719984c4197SStefano Zampini       ierr = PetscMalloc((PetscInt)PetscRealPart(lgqr_work_t)*sizeof(*gqr_work),&gqr_work);CHKERRQ(ierr);
1720984c4197SStefano Zampini       /* array to store scaling factors for reflectors */
1721984c4197SStefano Zampini       ierr = PetscMalloc(max_constraints*sizeof(*qr_tau),&qr_tau);CHKERRQ(ierr);
1722984c4197SStefano Zampini       /* array to store rhs and solution of triangular solver */
1723984c4197SStefano Zampini       ierr = PetscMalloc(max_constraints*max_constraints*sizeof(*trs_rhs),&trs_rhs);CHKERRQ(ierr);
1724984c4197SStefano Zampini       /* array to store whether a node is primal or not */
1725984c4197SStefano Zampini       ierr = PetscMalloc(pcis->n_B*sizeof(*is_primal),&is_primal);CHKERRQ(ierr);
1726984c4197SStefano Zampini       ierr = PetscMemzero(is_primal,pcis->n_B*sizeof(*is_primal));CHKERRQ(ierr);
1727984c4197SStefano Zampini       for (i=0;i<total_primal_vertices;i++) is_primal[local_to_B[aux_primal_numbering[i]]] = PETSC_TRUE;
1728984c4197SStefano Zampini 
1729984c4197SStefano Zampini       /* allocating workspace for check */
1730984c4197SStefano Zampini       if (pcbddc->dbg_flag) {
1731984c4197SStefano Zampini         ierr = PetscViewerASCIISynchronizedPrintf(pcbddc->dbg_viewer,"--------------------------------------------------------------\n");CHKERRQ(ierr);
1732984c4197SStefano Zampini         ierr = PetscViewerASCIISynchronizedPrintf(pcbddc->dbg_viewer,"Checking change of basis computation for subdomain %04d\n",PetscGlobalRank);CHKERRQ(ierr);
1733984c4197SStefano Zampini         ierr = PetscMalloc(max_size_of_constraint*(max_constraints+max_size_of_constraint)*sizeof(*work),&work);CHKERRQ(ierr);
1734674ae819SStefano Zampini       }
1735984c4197SStefano Zampini 
1736026de310SStefano Zampini       /* loop on constraints and see whether or not they need a change of basis */
1737026de310SStefano Zampini       /* -> using implicit ordering contained in temp_indices data */
1738026de310SStefano Zampini       total_counts = pcbddc->n_vertices;
1739026de310SStefano Zampini       primal_counter = total_counts;
1740026de310SStefano Zampini       while (total_counts<pcbddc->local_primal_size) {
1741026de310SStefano Zampini         primal_dofs = 1;
1742026de310SStefano Zampini         if (change_basis[total_counts]) {
1743026de310SStefano Zampini           /* get all constraints with same support: if more then one constraint is present on the cc then surely indices are stored contiguosly */
1744026de310SStefano Zampini           while (total_counts+primal_dofs < pcbddc->local_primal_size && temp_indices_to_constraint_B[temp_indices[total_counts]] == temp_indices_to_constraint_B[temp_indices[total_counts+primal_dofs]]) {
1745026de310SStefano Zampini             primal_dofs++;
1746674ae819SStefano Zampini           }
1747984c4197SStefano Zampini           /* get constraint info */
1748026de310SStefano Zampini           size_of_constraint = temp_indices[total_counts+1]-temp_indices[total_counts];
1749984c4197SStefano Zampini           dual_dofs = size_of_constraint-primal_dofs;
1750984c4197SStefano Zampini 
1751984c4197SStefano Zampini           /* copy quadrature constraints for change of basis check */
1752984c4197SStefano Zampini           if (pcbddc->dbg_flag) {
1753026de310SStefano Zampini             ierr = PetscViewerASCIISynchronizedPrintf(pcbddc->dbg_viewer,"Constraint %d to %d need a change of basis (size %d)\n",total_counts,total_counts+primal_dofs,size_of_constraint);CHKERRQ(ierr);
1754026de310SStefano Zampini             ierr = PetscMemcpy(work,&temp_quadrature_constraint[temp_indices[total_counts]],size_of_constraint*primal_dofs*sizeof(PetscScalar));CHKERRQ(ierr);
1755674ae819SStefano Zampini           }
1756984c4197SStefano Zampini 
1757984c4197SStefano Zampini           /* copy temporary constraints into larger work vector (in order to store all columns of Q) */
1758026de310SStefano Zampini           ierr = PetscMemcpy(qr_basis,&temp_quadrature_constraint[temp_indices[total_counts]],size_of_constraint*primal_dofs*sizeof(PetscScalar));CHKERRQ(ierr);
1759984c4197SStefano Zampini 
1760984c4197SStefano Zampini           /* compute QR decomposition of constraints */
17613f08241aSStefano Zampini           ierr = PetscBLASIntCast(size_of_constraint,&Blas_M);CHKERRQ(ierr);
17623f08241aSStefano Zampini           ierr = PetscBLASIntCast(primal_dofs,&Blas_N);CHKERRQ(ierr);
17633f08241aSStefano Zampini           ierr = PetscBLASIntCast(size_of_constraint,&Blas_LDA);CHKERRQ(ierr);
1764674ae819SStefano Zampini           ierr = PetscFPTrapPush(PETSC_FP_TRAP_OFF);CHKERRQ(ierr);
17653f08241aSStefano Zampini           PetscStackCallBLAS("LAPACKgeqrf",LAPACKgeqrf_(&Blas_M,&Blas_N,qr_basis,&Blas_LDA,qr_tau,qr_work,&lqr_work,&lierr));
1766984c4197SStefano Zampini           if (lierr) SETERRQ1(PETSC_COMM_SELF,PETSC_ERR_LIB,"Error in GEQRF Lapack routine %d",(int)lierr);
1767674ae819SStefano Zampini           ierr = PetscFPTrapPop();CHKERRQ(ierr);
1768984c4197SStefano Zampini 
1769984c4197SStefano Zampini           /* explictly compute R^-T */
1770984c4197SStefano Zampini           ierr = PetscMemzero(trs_rhs,primal_dofs*primal_dofs*sizeof(*trs_rhs));CHKERRQ(ierr);
1771984c4197SStefano Zampini           for (j=0;j<primal_dofs;j++) trs_rhs[j*(primal_dofs+1)] = 1.0;
17723f08241aSStefano Zampini           ierr = PetscBLASIntCast(primal_dofs,&Blas_N);CHKERRQ(ierr);
17733f08241aSStefano Zampini           ierr = PetscBLASIntCast(primal_dofs,&Blas_NRHS);CHKERRQ(ierr);
17743f08241aSStefano Zampini           ierr = PetscBLASIntCast(size_of_constraint,&Blas_LDA);CHKERRQ(ierr);
17753f08241aSStefano Zampini           ierr = PetscBLASIntCast(primal_dofs,&Blas_LDB);CHKERRQ(ierr);
1776984c4197SStefano Zampini           ierr = PetscFPTrapPush(PETSC_FP_TRAP_OFF);CHKERRQ(ierr);
17773f08241aSStefano Zampini           PetscStackCallBLAS("LAPACKtrtrs",LAPACKtrtrs_("U","T","N",&Blas_N,&Blas_NRHS,qr_basis,&Blas_LDA,trs_rhs,&Blas_LDB,&lierr));
1778984c4197SStefano Zampini           if (lierr) SETERRQ1(PETSC_COMM_SELF,PETSC_ERR_LIB,"Error in TRTRS Lapack routine %d",(int)lierr);
1779984c4197SStefano Zampini           ierr = PetscFPTrapPop();CHKERRQ(ierr);
1780984c4197SStefano Zampini 
1781984c4197SStefano Zampini           /* explcitly compute all columns of Q (Q = [Q1 | Q2] ) overwriting QR factorization in qr_basis */
17823f08241aSStefano Zampini           ierr = PetscBLASIntCast(size_of_constraint,&Blas_M);CHKERRQ(ierr);
17833f08241aSStefano Zampini           ierr = PetscBLASIntCast(size_of_constraint,&Blas_N);CHKERRQ(ierr);
17843f08241aSStefano Zampini           ierr = PetscBLASIntCast(primal_dofs,&Blas_K);CHKERRQ(ierr);
17853f08241aSStefano Zampini           ierr = PetscBLASIntCast(size_of_constraint,&Blas_LDA);CHKERRQ(ierr);
1786984c4197SStefano Zampini           ierr = PetscFPTrapPush(PETSC_FP_TRAP_OFF);CHKERRQ(ierr);
17873f08241aSStefano Zampini           PetscStackCallBLAS("LAPACKungqr",LAPACKungqr_(&Blas_M,&Blas_N,&Blas_K,qr_basis,&Blas_LDA,qr_tau,gqr_work,&lgqr_work,&lierr));
1788984c4197SStefano Zampini           if (lierr) SETERRQ1(PETSC_COMM_SELF,PETSC_ERR_LIB,"Error in UNGQR Lapack routine %d",(int)lierr);
1789984c4197SStefano Zampini           ierr = PetscFPTrapPop();CHKERRQ(ierr);
1790984c4197SStefano Zampini 
1791984c4197SStefano Zampini           /* first primal_dofs columns of Q need to be re-scaled in order to be unitary w.r.t constraints
1792984c4197SStefano Zampini              i.e. C_{pxn}*Q_{nxn} should be equal to [I_pxp | 0_pxd] (see check below)
1793984c4197SStefano Zampini              where n=size_of_constraint, p=primal_dofs, d=dual_dofs (n=p+d), I and 0 identity and null matrix resp. */
17943f08241aSStefano Zampini           ierr = PetscBLASIntCast(size_of_constraint,&Blas_M);CHKERRQ(ierr);
17953f08241aSStefano Zampini           ierr = PetscBLASIntCast(primal_dofs,&Blas_N);CHKERRQ(ierr);
17963f08241aSStefano Zampini           ierr = PetscBLASIntCast(primal_dofs,&Blas_K);CHKERRQ(ierr);
17973f08241aSStefano Zampini           ierr = PetscBLASIntCast(size_of_constraint,&Blas_LDA);CHKERRQ(ierr);
17983f08241aSStefano Zampini           ierr = PetscBLASIntCast(primal_dofs,&Blas_LDB);CHKERRQ(ierr);
17993f08241aSStefano Zampini           ierr = PetscBLASIntCast(size_of_constraint,&Blas_LDC);CHKERRQ(ierr);
1800984c4197SStefano Zampini           ierr = PetscFPTrapPush(PETSC_FP_TRAP_OFF);CHKERRQ(ierr);
1801c4303822SStefano Zampini           PetscStackCallBLAS("BLASgemm",BLASgemm_("N","N",&Blas_M,&Blas_N,&Blas_K,&one,qr_basis,&Blas_LDA,trs_rhs,&Blas_LDB,&zero,&temp_quadrature_constraint[temp_indices[total_counts]],&Blas_LDC));
1802984c4197SStefano Zampini           ierr = PetscFPTrapPop();CHKERRQ(ierr);
1803026de310SStefano Zampini           ierr = PetscMemcpy(qr_basis,&temp_quadrature_constraint[temp_indices[total_counts]],size_of_constraint*primal_dofs*sizeof(PetscScalar));CHKERRQ(ierr);
1804984c4197SStefano Zampini 
1805984c4197SStefano Zampini           /* insert values in change of basis matrix respecting global ordering of new primal dofs */
1806026de310SStefano Zampini           start_rows = &temp_indices_to_constraint_B[temp_indices[total_counts]];
1807984c4197SStefano Zampini           /* insert cols for primal dofs */
1808984c4197SStefano Zampini           for (j=0;j<primal_dofs;j++) {
1809984c4197SStefano Zampini             start_vals = &qr_basis[j*size_of_constraint];
1810026de310SStefano Zampini             start_cols = &temp_indices_to_constraint_B[temp_indices[total_counts]+aux_primal_minloc[primal_counter+j]];
1811984c4197SStefano Zampini             ierr = MatSetValues(pcbddc->ChangeOfBasisMatrix,size_of_constraint,start_rows,1,start_cols,start_vals,INSERT_VALUES);CHKERRQ(ierr);
1812984c4197SStefano Zampini           }
1813984c4197SStefano Zampini           /* insert cols for dual dofs */
1814984c4197SStefano Zampini           for (j=0,k=0;j<dual_dofs;k++) {
1815026de310SStefano Zampini             if (!is_primal[temp_indices_to_constraint_B[temp_indices[total_counts]+k]]) {
1816984c4197SStefano Zampini               start_vals = &qr_basis[(primal_dofs+j)*size_of_constraint];
1817026de310SStefano Zampini               start_cols = &temp_indices_to_constraint_B[temp_indices[total_counts]+k];
1818984c4197SStefano Zampini               ierr = MatSetValues(pcbddc->ChangeOfBasisMatrix,size_of_constraint,start_rows,1,start_cols,start_vals,INSERT_VALUES);CHKERRQ(ierr);
1819984c4197SStefano Zampini               j++;
1820674ae819SStefano Zampini             }
1821674ae819SStefano Zampini           }
1822984c4197SStefano Zampini 
1823984c4197SStefano Zampini           /* check change of basis */
1824984c4197SStefano Zampini           if (pcbddc->dbg_flag) {
1825984c4197SStefano Zampini             PetscInt   ii,jj;
1826984c4197SStefano Zampini             PetscBool valid_qr=PETSC_TRUE;
1827c4303822SStefano Zampini             ierr = PetscBLASIntCast(primal_dofs,&Blas_M);CHKERRQ(ierr);
1828c4303822SStefano Zampini             ierr = PetscBLASIntCast(size_of_constraint,&Blas_N);CHKERRQ(ierr);
1829c4303822SStefano Zampini             ierr = PetscBLASIntCast(size_of_constraint,&Blas_K);CHKERRQ(ierr);
1830c4303822SStefano Zampini             ierr = PetscBLASIntCast(size_of_constraint,&Blas_LDA);CHKERRQ(ierr);
1831c4303822SStefano Zampini             ierr = PetscBLASIntCast(size_of_constraint,&Blas_LDB);CHKERRQ(ierr);
1832c4303822SStefano Zampini             ierr = PetscBLASIntCast(primal_dofs,&Blas_LDC);CHKERRQ(ierr);
1833984c4197SStefano Zampini             ierr = PetscFPTrapPush(PETSC_FP_TRAP_OFF);CHKERRQ(ierr);
1834c4303822SStefano Zampini             PetscStackCallBLAS("BLASgemm",BLASgemm_("T","N",&Blas_M,&Blas_N,&Blas_K,&one,work,&Blas_LDA,qr_basis,&Blas_LDB,&zero,&work[size_of_constraint*primal_dofs],&Blas_LDC));
1835984c4197SStefano Zampini             ierr = PetscFPTrapPop();CHKERRQ(ierr);
1836984c4197SStefano Zampini             for (jj=0;jj<size_of_constraint;jj++) {
1837984c4197SStefano Zampini               for (ii=0;ii<primal_dofs;ii++) {
1838984c4197SStefano Zampini                 if (ii != jj && PetscAbsScalar(work[size_of_constraint*primal_dofs+jj*primal_dofs+ii]) > 1.e-12) valid_qr = PETSC_FALSE;
1839984c4197SStefano Zampini                 if (ii == jj && PetscAbsScalar(work[size_of_constraint*primal_dofs+jj*primal_dofs+ii]-1.0) > 1.e-12) valid_qr = PETSC_FALSE;
1840674ae819SStefano Zampini               }
1841674ae819SStefano Zampini             }
1842984c4197SStefano Zampini             if (!valid_qr) {
1843984c4197SStefano Zampini               ierr = PetscViewerASCIISynchronizedPrintf(pcbddc->dbg_viewer,"\t-> wrong change of basis!\n",PetscGlobalRank);CHKERRQ(ierr);
1844984c4197SStefano Zampini               for (jj=0;jj<size_of_constraint;jj++) {
1845984c4197SStefano Zampini                 for (ii=0;ii<primal_dofs;ii++) {
1846984c4197SStefano Zampini                   if (ii != jj && PetscAbsScalar(work[size_of_constraint*primal_dofs+jj*primal_dofs+ii]) > 1.e-12) {
1847984c4197SStefano Zampini                     PetscViewerASCIISynchronizedPrintf(pcbddc->dbg_viewer,"\tQr basis function %d is not orthogonal to constraint %d (%1.14e)!\n",jj,ii,PetscAbsScalar(work[size_of_constraint*primal_dofs+jj*primal_dofs+ii]));
1848674ae819SStefano Zampini                   }
1849984c4197SStefano Zampini                   if (ii == jj && PetscAbsScalar(work[size_of_constraint*primal_dofs+jj*primal_dofs+ii]-1.0) > 1.e-12) {
1850984c4197SStefano Zampini                     PetscViewerASCIISynchronizedPrintf(pcbddc->dbg_viewer,"\tQr basis function %d is not unitary w.r.t constraint %d (%1.14e)!\n",jj,ii,PetscAbsScalar(work[size_of_constraint*primal_dofs+jj*primal_dofs+ii]));
1851984c4197SStefano Zampini                   }
1852984c4197SStefano Zampini                 }
1853984c4197SStefano Zampini               }
1854674ae819SStefano Zampini             } else {
1855984c4197SStefano Zampini               ierr = PetscViewerASCIISynchronizedPrintf(pcbddc->dbg_viewer,"\t-> right change of basis!\n",PetscGlobalRank);CHKERRQ(ierr);
1856674ae819SStefano Zampini             }
1857674ae819SStefano Zampini           }
1858026de310SStefano Zampini           /* increment primal counter */
1859026de310SStefano Zampini           primal_counter += primal_dofs;
1860984c4197SStefano Zampini         } else {
1861984c4197SStefano Zampini           if (pcbddc->dbg_flag) {
1862026de310SStefano Zampini             ierr = PetscViewerASCIISynchronizedPrintf(pcbddc->dbg_viewer,"Constraint %d does not need a change of basis (size %d)\n",total_counts,temp_indices[total_counts+1]-temp_indices[total_counts]);CHKERRQ(ierr);
1863674ae819SStefano Zampini           }
1864674ae819SStefano Zampini         }
1865026de310SStefano Zampini         /* increment constraint counter total_counts */
1866026de310SStefano Zampini         total_counts += primal_dofs;
1867674ae819SStefano Zampini       }
1868984c4197SStefano Zampini       if (pcbddc->dbg_flag) {
1869984c4197SStefano Zampini         ierr = PetscViewerFlush(pcbddc->dbg_viewer);CHKERRQ(ierr);
1870984c4197SStefano Zampini         ierr = PetscFree(work);CHKERRQ(ierr);
1871984c4197SStefano Zampini       }
1872984c4197SStefano Zampini       /* free workspace */
1873984c4197SStefano Zampini       ierr = PetscFree(trs_rhs);CHKERRQ(ierr);
1874984c4197SStefano Zampini       ierr = PetscFree(qr_tau);CHKERRQ(ierr);
1875984c4197SStefano Zampini       ierr = PetscFree(qr_work);CHKERRQ(ierr);
1876984c4197SStefano Zampini       ierr = PetscFree(gqr_work);CHKERRQ(ierr);
1877984c4197SStefano Zampini       ierr = PetscFree(is_primal);CHKERRQ(ierr);
1878984c4197SStefano Zampini       ierr = PetscFree(qr_basis);CHKERRQ(ierr);
1879674ae819SStefano Zampini     }
1880674ae819SStefano Zampini     /* assembling */
1881674ae819SStefano Zampini     ierr = MatAssemblyBegin(pcbddc->ChangeOfBasisMatrix,MAT_FINAL_ASSEMBLY);CHKERRQ(ierr);
1882674ae819SStefano Zampini     ierr = MatAssemblyEnd(pcbddc->ChangeOfBasisMatrix,MAT_FINAL_ASSEMBLY);CHKERRQ(ierr);
1883984c4197SStefano Zampini     /*
1884984c4197SStefano Zampini     ierr = MatView(pcbddc->ChangeOfBasisMatrix,(PetscViewer)0);CHKERRQ(ierr);
1885984c4197SStefano Zampini     */
1886674ae819SStefano Zampini   }
1887e310c8b4SStefano Zampini   /* free workspace */
1888984c4197SStefano Zampini   ierr = PetscFree(aux_primal_numbering);CHKERRQ(ierr);
1889984c4197SStefano Zampini   ierr = PetscFree(aux_primal_minloc);CHKERRQ(ierr);
1890674ae819SStefano Zampini   ierr = PetscFree(temp_indices);CHKERRQ(ierr);
1891674ae819SStefano Zampini   ierr = PetscFree(change_basis);CHKERRQ(ierr);
1892674ae819SStefano Zampini   ierr = PetscFree(temp_indices_to_constraint);CHKERRQ(ierr);
1893674ae819SStefano Zampini   ierr = PetscFree(temp_indices_to_constraint_B);CHKERRQ(ierr);
1894674ae819SStefano Zampini   ierr = PetscFree(local_to_B);CHKERRQ(ierr);
1895674ae819SStefano Zampini   ierr = PetscFree(temp_quadrature_constraint);CHKERRQ(ierr);
1896674ae819SStefano Zampini   PetscFunctionReturn(0);
1897674ae819SStefano Zampini }
1898674ae819SStefano Zampini 
1899674ae819SStefano Zampini #undef __FUNCT__
1900674ae819SStefano Zampini #define __FUNCT__ "PCBDDCAnalyzeInterface"
1901674ae819SStefano Zampini PetscErrorCode PCBDDCAnalyzeInterface(PC pc)
1902674ae819SStefano Zampini {
1903674ae819SStefano Zampini   PC_BDDC     *pcbddc = (PC_BDDC*)pc->data;
1904674ae819SStefano Zampini   PC_IS       *pcis = (PC_IS*)pc->data;
1905674ae819SStefano Zampini   Mat_IS      *matis  = (Mat_IS*)pc->pmat->data;
1906674ae819SStefano Zampini   PetscInt    bs,ierr,i,vertex_size;
1907674ae819SStefano Zampini   PetscViewer viewer=pcbddc->dbg_viewer;
1908674ae819SStefano Zampini 
1909674ae819SStefano Zampini   PetscFunctionBegin;
1910674ae819SStefano Zampini   /* Init local Graph struct */
1911674ae819SStefano Zampini   ierr = PCBDDCGraphInit(pcbddc->mat_graph,matis->mapping);CHKERRQ(ierr);
1912674ae819SStefano Zampini 
1913575ad6abSStefano Zampini   /* Check validity of the csr graph passed in by the user */
1914575ad6abSStefano Zampini   if (pcbddc->mat_graph->nvtxs_csr != pcbddc->mat_graph->nvtxs) {
1915575ad6abSStefano Zampini     ierr = PCBDDCGraphResetCSR(pcbddc->mat_graph);CHKERRQ(ierr);
1916575ad6abSStefano Zampini   }
1917674ae819SStefano Zampini   /* Set default CSR adjacency of local dofs if not provided by the user with PCBDDCSetLocalAdjacencyGraph */
1918674ae819SStefano Zampini   if (!pcbddc->mat_graph->xadj || !pcbddc->mat_graph->adjncy) {
1919674ae819SStefano Zampini     Mat mat_adj;
1920674ae819SStefano Zampini     const PetscInt *xadj,*adjncy;
1921674ae819SStefano Zampini     PetscBool flg_row=PETSC_TRUE;
1922674ae819SStefano Zampini 
1923674ae819SStefano Zampini     ierr = MatConvert(matis->A,MATMPIADJ,MAT_INITIAL_MATRIX,&mat_adj);CHKERRQ(ierr);
1924674ae819SStefano Zampini     ierr = MatGetRowIJ(mat_adj,0,PETSC_TRUE,PETSC_FALSE,&i,&xadj,&adjncy,&flg_row);CHKERRQ(ierr);
1925674ae819SStefano Zampini     if (!flg_row) {
1926674ae819SStefano Zampini       SETERRQ1(PETSC_COMM_SELF,PETSC_ERR_PLIB,"Error in MatGetRowIJ called in %s\n",__FUNCT__);
1927674ae819SStefano Zampini     }
1928674ae819SStefano Zampini     ierr = PCBDDCSetLocalAdjacencyGraph(pc,i,xadj,adjncy,PETSC_COPY_VALUES);CHKERRQ(ierr);
1929674ae819SStefano Zampini     ierr = MatRestoreRowIJ(mat_adj,0,PETSC_TRUE,PETSC_FALSE,&i,&xadj,&adjncy,&flg_row);CHKERRQ(ierr);
1930674ae819SStefano Zampini     if (!flg_row) {
1931674ae819SStefano Zampini       SETERRQ1(PETSC_COMM_SELF,PETSC_ERR_PLIB,"Error in MatRestoreRowIJ called in %s\n",__FUNCT__);
1932674ae819SStefano Zampini     }
1933674ae819SStefano Zampini     ierr = MatDestroy(&mat_adj);CHKERRQ(ierr);
1934674ae819SStefano Zampini   }
1935674ae819SStefano Zampini 
1936674ae819SStefano Zampini   /* Set default dofs' splitting if no information has been provided by the user with PCBDDCSetDofsSplitting */
1937674ae819SStefano Zampini   vertex_size = 1;
1938674ae819SStefano Zampini   if (!pcbddc->n_ISForDofs) {
1939674ae819SStefano Zampini     IS *custom_ISForDofs;
1940674ae819SStefano Zampini 
1941674ae819SStefano Zampini     ierr = MatGetBlockSize(matis->A,&bs);CHKERRQ(ierr);
1942674ae819SStefano Zampini     ierr = PetscMalloc(bs*sizeof(IS),&custom_ISForDofs);CHKERRQ(ierr);
1943674ae819SStefano Zampini     for (i=0;i<bs;i++) {
1944674ae819SStefano Zampini       ierr = ISCreateStride(PETSC_COMM_SELF,pcis->n/bs,i,bs,&custom_ISForDofs[i]);CHKERRQ(ierr);
1945674ae819SStefano Zampini     }
1946674ae819SStefano Zampini     ierr = PCBDDCSetDofsSplitting(pc,bs,custom_ISForDofs);CHKERRQ(ierr);
1947674ae819SStefano Zampini     /* remove my references to IS objects */
1948674ae819SStefano Zampini     for (i=0;i<bs;i++) {
1949674ae819SStefano Zampini       ierr = ISDestroy(&custom_ISForDofs[i]);CHKERRQ(ierr);
1950674ae819SStefano Zampini     }
1951674ae819SStefano Zampini     ierr = PetscFree(custom_ISForDofs);CHKERRQ(ierr);
1952674ae819SStefano Zampini   } else { /* mat block size as vertex size (used for elasticity) */
1953674ae819SStefano Zampini     ierr = MatGetBlockSize(matis->A,&vertex_size);CHKERRQ(ierr);
1954674ae819SStefano Zampini   }
1955674ae819SStefano Zampini 
1956674ae819SStefano Zampini   /* Setup of Graph */
1957674ae819SStefano Zampini   ierr = PCBDDCGraphSetUp(pcbddc->mat_graph,vertex_size,pcbddc->NeumannBoundaries,pcbddc->DirichletBoundaries,pcbddc->n_ISForDofs,pcbddc->ISForDofs,pcbddc->user_primal_vertices);
1958674ae819SStefano Zampini 
1959674ae819SStefano Zampini   /* Graph's connected components analysis */
1960674ae819SStefano Zampini   ierr = PCBDDCGraphComputeConnectedComponents(pcbddc->mat_graph);CHKERRQ(ierr);
1961674ae819SStefano Zampini 
1962674ae819SStefano Zampini   /* print some info to stdout */
1963674ae819SStefano Zampini   if (pcbddc->dbg_flag) {
1964e49050b4SStefano Zampini     ierr = PCBDDCGraphASCIIView(pcbddc->mat_graph,pcbddc->dbg_flag,viewer);
1965674ae819SStefano Zampini   }
1966674ae819SStefano Zampini   PetscFunctionReturn(0);
1967674ae819SStefano Zampini }
1968674ae819SStefano Zampini 
1969674ae819SStefano Zampini #undef __FUNCT__
1970674ae819SStefano Zampini #define __FUNCT__ "PCBDDCGetPrimalVerticesLocalIdx"
1971674ae819SStefano Zampini PetscErrorCode  PCBDDCGetPrimalVerticesLocalIdx(PC pc, PetscInt *n_vertices, PetscInt *vertices_idx[])
1972674ae819SStefano Zampini {
1973674ae819SStefano Zampini   PC_BDDC        *pcbddc = (PC_BDDC*)(pc->data);
1974674ae819SStefano Zampini   PetscInt       *vertices,*row_cmat_indices,n,i,size_of_constraint,local_primal_size;
1975674ae819SStefano Zampini   PetscErrorCode ierr;
1976674ae819SStefano Zampini 
1977674ae819SStefano Zampini   PetscFunctionBegin;
1978674ae819SStefano Zampini   n = 0;
1979674ae819SStefano Zampini   vertices = 0;
1980674ae819SStefano Zampini   if (pcbddc->ConstraintMatrix) {
1981674ae819SStefano Zampini     ierr = MatGetSize(pcbddc->ConstraintMatrix,&local_primal_size,&i);CHKERRQ(ierr);
1982b120a5c6SStefano Zampini     for (i=0;i<local_primal_size;i++) {
1983b120a5c6SStefano Zampini       ierr = MatGetRow(pcbddc->ConstraintMatrix,i,&size_of_constraint,NULL,NULL);CHKERRQ(ierr);
1984b120a5c6SStefano Zampini       if (size_of_constraint == 1) n++;
1985b120a5c6SStefano Zampini       ierr = MatRestoreRow(pcbddc->ConstraintMatrix,i,&size_of_constraint,NULL,NULL);CHKERRQ(ierr);
1986b120a5c6SStefano Zampini     }
1987811e8ca2SStefano Zampini     if (vertices_idx) {
1988b120a5c6SStefano Zampini       ierr = PetscMalloc(n*sizeof(PetscInt),&vertices);CHKERRQ(ierr);
1989b120a5c6SStefano Zampini       n = 0;
1990674ae819SStefano Zampini       for (i=0;i<local_primal_size;i++) {
1991674ae819SStefano Zampini         ierr = MatGetRow(pcbddc->ConstraintMatrix,i,&size_of_constraint,(const PetscInt**)&row_cmat_indices,NULL);CHKERRQ(ierr);
1992674ae819SStefano Zampini         if (size_of_constraint == 1) {
1993674ae819SStefano Zampini           vertices[n++]=row_cmat_indices[0];
1994674ae819SStefano Zampini         }
1995674ae819SStefano Zampini         ierr = MatRestoreRow(pcbddc->ConstraintMatrix,i,&size_of_constraint,(const PetscInt**)&row_cmat_indices,NULL);CHKERRQ(ierr);
1996674ae819SStefano Zampini       }
1997674ae819SStefano Zampini     }
1998811e8ca2SStefano Zampini   }
1999674ae819SStefano Zampini   *n_vertices = n;
2000811e8ca2SStefano Zampini   if (vertices_idx) *vertices_idx = vertices;
2001674ae819SStefano Zampini   PetscFunctionReturn(0);
2002674ae819SStefano Zampini }
2003674ae819SStefano Zampini 
2004674ae819SStefano Zampini #undef __FUNCT__
2005674ae819SStefano Zampini #define __FUNCT__ "PCBDDCGetPrimalConstraintsLocalIdx"
2006674ae819SStefano Zampini PetscErrorCode  PCBDDCGetPrimalConstraintsLocalIdx(PC pc, PetscInt *n_constraints, PetscInt *constraints_idx[])
2007674ae819SStefano Zampini {
2008674ae819SStefano Zampini   PC_BDDC        *pcbddc = (PC_BDDC*)(pc->data);
2009674ae819SStefano Zampini   PetscInt       *constraints_index,*row_cmat_indices,*row_cmat_global_indices;
2010674ae819SStefano Zampini   PetscInt       n,i,j,size_of_constraint,local_primal_size,local_size,max_size_of_constraint,min_index,min_loc;
2011674ae819SStefano Zampini   PetscBool      *touched;
2012674ae819SStefano Zampini   PetscErrorCode ierr;
2013674ae819SStefano Zampini 
2014674ae819SStefano Zampini   PetscFunctionBegin;
2015674ae819SStefano Zampini   n = 0;
2016674ae819SStefano Zampini   constraints_index = 0;
2017674ae819SStefano Zampini   if (pcbddc->ConstraintMatrix) {
2018674ae819SStefano Zampini     ierr = MatGetSize(pcbddc->ConstraintMatrix,&local_primal_size,&local_size);CHKERRQ(ierr);
2019674ae819SStefano Zampini     max_size_of_constraint = 0;
2020674ae819SStefano Zampini     for (i=0;i<local_primal_size;i++) {
2021674ae819SStefano Zampini       ierr = MatGetRow(pcbddc->ConstraintMatrix,i,&size_of_constraint,NULL,NULL);CHKERRQ(ierr);
2022674ae819SStefano Zampini       if (size_of_constraint > 1) {
2023674ae819SStefano Zampini         n++;
2024674ae819SStefano Zampini       }
2025674ae819SStefano Zampini       max_size_of_constraint = PetscMax(size_of_constraint,max_size_of_constraint);
2026674ae819SStefano Zampini       ierr = MatRestoreRow(pcbddc->ConstraintMatrix,i,&size_of_constraint,NULL,NULL);CHKERRQ(ierr);
2027674ae819SStefano Zampini     }
2028811e8ca2SStefano Zampini     if (constraints_idx) {
2029674ae819SStefano Zampini       ierr = PetscMalloc(n*sizeof(PetscInt),&constraints_index);CHKERRQ(ierr);
2030674ae819SStefano Zampini       ierr = PetscMalloc(max_size_of_constraint*sizeof(PetscInt),&row_cmat_global_indices);CHKERRQ(ierr);
2031674ae819SStefano Zampini       ierr = PetscMalloc(local_size*sizeof(PetscBool),&touched);CHKERRQ(ierr);
2032674ae819SStefano Zampini       ierr = PetscMemzero(touched,local_size*sizeof(PetscBool));CHKERRQ(ierr);
2033674ae819SStefano Zampini       n = 0;
2034674ae819SStefano Zampini       for (i=0;i<local_primal_size;i++) {
2035674ae819SStefano Zampini         ierr = MatGetRow(pcbddc->ConstraintMatrix,i,&size_of_constraint,(const PetscInt**)&row_cmat_indices,NULL);CHKERRQ(ierr);
2036674ae819SStefano Zampini         if (size_of_constraint > 1) {
2037674ae819SStefano Zampini           ierr = ISLocalToGlobalMappingApply(pcbddc->mat_graph->l2gmap,size_of_constraint,row_cmat_indices,row_cmat_global_indices);CHKERRQ(ierr);
203882d3d8afSStefano Zampini           /* find first untouched local node */
203982d3d8afSStefano Zampini           j = 0;
204082d3d8afSStefano Zampini           while(touched[row_cmat_indices[j]]) j++;
204182d3d8afSStefano Zampini           min_index = row_cmat_global_indices[j];
204282d3d8afSStefano Zampini           min_loc = j;
204382d3d8afSStefano Zampini           /* search the minimum among nodes not yet touched on the connected component
204482d3d8afSStefano Zampini              since there can be more than one constraint on a single cc */
2045674ae819SStefano Zampini           for (j=1;j<size_of_constraint;j++) {
2046674ae819SStefano Zampini             if (min_index > row_cmat_global_indices[j] && !touched[row_cmat_indices[j]]) {
2047674ae819SStefano Zampini               min_index = row_cmat_global_indices[j];
2048674ae819SStefano Zampini               min_loc = j;
2049674ae819SStefano Zampini             }
2050674ae819SStefano Zampini           }
2051674ae819SStefano Zampini           touched[row_cmat_indices[min_loc]] = PETSC_TRUE;
2052674ae819SStefano Zampini           constraints_index[n++] = row_cmat_indices[min_loc];
2053674ae819SStefano Zampini         }
2054674ae819SStefano Zampini         ierr = MatRestoreRow(pcbddc->ConstraintMatrix,i,&size_of_constraint,(const PetscInt**)&row_cmat_indices,NULL);CHKERRQ(ierr);
2055674ae819SStefano Zampini       }
2056674ae819SStefano Zampini       ierr = PetscFree(touched);CHKERRQ(ierr);
2057674ae819SStefano Zampini       ierr = PetscFree(row_cmat_global_indices);CHKERRQ(ierr);
2058811e8ca2SStefano Zampini     }
2059811e8ca2SStefano Zampini   }
2060674ae819SStefano Zampini   *n_constraints = n;
2061811e8ca2SStefano Zampini   if (constraints_idx) *constraints_idx = constraints_index;
2062674ae819SStefano Zampini   PetscFunctionReturn(0);
2063674ae819SStefano Zampini }
2064674ae819SStefano Zampini 
2065674ae819SStefano Zampini /* the next two functions has been adapted from pcis.c */
2066674ae819SStefano Zampini #undef __FUNCT__
2067674ae819SStefano Zampini #define __FUNCT__ "PCBDDCApplySchur"
2068674ae819SStefano Zampini PetscErrorCode  PCBDDCApplySchur(PC pc, Vec v, Vec vec1_B, Vec vec2_B, Vec vec1_D, Vec vec2_D)
2069674ae819SStefano Zampini {
2070674ae819SStefano Zampini   PetscErrorCode ierr;
2071674ae819SStefano Zampini   PC_IS          *pcis = (PC_IS*)(pc->data);
2072674ae819SStefano Zampini 
2073674ae819SStefano Zampini   PetscFunctionBegin;
2074674ae819SStefano Zampini   if (!vec2_B) { vec2_B = v; }
2075674ae819SStefano Zampini   ierr = MatMult(pcis->A_BB,v,vec1_B);CHKERRQ(ierr);
2076674ae819SStefano Zampini   ierr = MatMult(pcis->A_IB,v,vec1_D);CHKERRQ(ierr);
2077674ae819SStefano Zampini   ierr = KSPSolve(pcis->ksp_D,vec1_D,vec2_D);CHKERRQ(ierr);
2078674ae819SStefano Zampini   ierr = MatMult(pcis->A_BI,vec2_D,vec2_B);CHKERRQ(ierr);
2079674ae819SStefano Zampini   ierr = VecAXPY(vec1_B,-1.0,vec2_B);CHKERRQ(ierr);
2080674ae819SStefano Zampini   PetscFunctionReturn(0);
2081674ae819SStefano Zampini }
2082674ae819SStefano Zampini 
2083674ae819SStefano Zampini #undef __FUNCT__
2084674ae819SStefano Zampini #define __FUNCT__ "PCBDDCApplySchurTranspose"
2085674ae819SStefano Zampini PetscErrorCode  PCBDDCApplySchurTranspose(PC pc, Vec v, Vec vec1_B, Vec vec2_B, Vec vec1_D, Vec vec2_D)
2086674ae819SStefano Zampini {
2087674ae819SStefano Zampini   PetscErrorCode ierr;
2088674ae819SStefano Zampini   PC_IS          *pcis = (PC_IS*)(pc->data);
2089674ae819SStefano Zampini 
2090674ae819SStefano Zampini   PetscFunctionBegin;
2091674ae819SStefano Zampini   if (!vec2_B) { vec2_B = v; }
2092674ae819SStefano Zampini   ierr = MatMultTranspose(pcis->A_BB,v,vec1_B);CHKERRQ(ierr);
2093674ae819SStefano Zampini   ierr = MatMultTranspose(pcis->A_BI,v,vec1_D);CHKERRQ(ierr);
2094674ae819SStefano Zampini   ierr = KSPSolveTranspose(pcis->ksp_D,vec1_D,vec2_D);CHKERRQ(ierr);
2095674ae819SStefano Zampini   ierr = MatMultTranspose(pcis->A_IB,vec2_D,vec2_B);CHKERRQ(ierr);
2096674ae819SStefano Zampini   ierr = VecAXPY(vec1_B,-1.0,vec2_B);CHKERRQ(ierr);
2097674ae819SStefano Zampini   PetscFunctionReturn(0);
2098674ae819SStefano Zampini }
2099674ae819SStefano Zampini 
2100674ae819SStefano Zampini #undef __FUNCT__
2101674ae819SStefano Zampini #define __FUNCT__ "PCBDDCSubsetNumbering"
2102674ae819SStefano Zampini PetscErrorCode PCBDDCSubsetNumbering(MPI_Comm comm,ISLocalToGlobalMapping l2gmap, PetscInt n_local_dofs, PetscInt local_dofs[], PetscInt local_dofs_mult[], PetscInt* n_global_subset, PetscInt* global_numbering_subset[])
2103674ae819SStefano Zampini {
2104674ae819SStefano Zampini   Vec            local_vec,global_vec;
2105674ae819SStefano Zampini   IS             seqis,paris;
2106674ae819SStefano Zampini   VecScatter     scatter_ctx;
2107674ae819SStefano Zampini   PetscScalar    *array;
2108674ae819SStefano Zampini   PetscInt       *temp_global_dofs;
2109674ae819SStefano Zampini   PetscScalar    globalsum;
2110674ae819SStefano Zampini   PetscInt       i,j,s;
2111674ae819SStefano Zampini   PetscInt       nlocals,first_index,old_index,max_local;
2112674ae819SStefano Zampini   PetscMPIInt    rank_prec_comm,size_prec_comm,max_global;
2113674ae819SStefano Zampini   PetscMPIInt    *dof_sizes,*dof_displs;
2114674ae819SStefano Zampini   PetscBool      first_found;
2115674ae819SStefano Zampini   PetscErrorCode ierr;
2116674ae819SStefano Zampini 
2117674ae819SStefano Zampini   PetscFunctionBegin;
2118674ae819SStefano Zampini   /* mpi buffers */
2119674ae819SStefano Zampini   MPI_Comm_size(comm,&size_prec_comm);
2120674ae819SStefano Zampini   MPI_Comm_rank(comm,&rank_prec_comm);
2121674ae819SStefano Zampini   j = ( !rank_prec_comm ? size_prec_comm : 0);
2122674ae819SStefano Zampini   ierr = PetscMalloc(j*sizeof(*dof_sizes),&dof_sizes);CHKERRQ(ierr);
2123674ae819SStefano Zampini   ierr = PetscMalloc(j*sizeof(*dof_displs),&dof_displs);CHKERRQ(ierr);
2124674ae819SStefano Zampini   /* get maximum size of subset */
2125674ae819SStefano Zampini   ierr = PetscMalloc(n_local_dofs*sizeof(PetscInt),&temp_global_dofs);CHKERRQ(ierr);
2126674ae819SStefano Zampini   ierr = ISLocalToGlobalMappingApply(l2gmap,n_local_dofs,local_dofs,temp_global_dofs);CHKERRQ(ierr);
2127674ae819SStefano Zampini   max_local = 0;
2128674ae819SStefano Zampini   if (n_local_dofs) {
2129674ae819SStefano Zampini     max_local = temp_global_dofs[0];
2130674ae819SStefano Zampini     for (i=1;i<n_local_dofs;i++) {
2131674ae819SStefano Zampini       if (max_local < temp_global_dofs[i] ) {
2132674ae819SStefano Zampini         max_local = temp_global_dofs[i];
2133674ae819SStefano Zampini       }
2134674ae819SStefano Zampini     }
2135674ae819SStefano Zampini   }
2136674ae819SStefano Zampini   ierr = MPI_Allreduce(&max_local,&max_global,1,MPIU_INT,MPI_MAX,comm);
2137674ae819SStefano Zampini   max_global++;
2138674ae819SStefano Zampini   max_local = 0;
2139674ae819SStefano Zampini   if (n_local_dofs) {
2140674ae819SStefano Zampini     max_local = local_dofs[0];
2141674ae819SStefano Zampini     for (i=1;i<n_local_dofs;i++) {
2142674ae819SStefano Zampini       if (max_local < local_dofs[i] ) {
2143674ae819SStefano Zampini         max_local = local_dofs[i];
2144674ae819SStefano Zampini       }
2145674ae819SStefano Zampini     }
2146674ae819SStefano Zampini   }
2147674ae819SStefano Zampini   max_local++;
2148674ae819SStefano Zampini   /* allocate workspace */
2149674ae819SStefano Zampini   ierr = VecCreate(PETSC_COMM_SELF,&local_vec);CHKERRQ(ierr);
2150674ae819SStefano Zampini   ierr = VecSetSizes(local_vec,PETSC_DECIDE,max_local);CHKERRQ(ierr);
2151674ae819SStefano Zampini   ierr = VecSetType(local_vec,VECSEQ);CHKERRQ(ierr);
2152674ae819SStefano Zampini   ierr = VecCreate(comm,&global_vec);CHKERRQ(ierr);
2153674ae819SStefano Zampini   ierr = VecSetSizes(global_vec,PETSC_DECIDE,max_global);CHKERRQ(ierr);
2154674ae819SStefano Zampini   ierr = VecSetType(global_vec,VECMPI);CHKERRQ(ierr);
2155674ae819SStefano Zampini   /* create scatter */
2156674ae819SStefano Zampini   ierr = ISCreateGeneral(PETSC_COMM_SELF,n_local_dofs,local_dofs,PETSC_COPY_VALUES,&seqis);CHKERRQ(ierr);
2157674ae819SStefano Zampini   ierr = ISCreateGeneral(comm,n_local_dofs,temp_global_dofs,PETSC_COPY_VALUES,&paris);CHKERRQ(ierr);
2158674ae819SStefano Zampini   ierr = VecScatterCreate(local_vec,seqis,global_vec,paris,&scatter_ctx);CHKERRQ(ierr);
2159674ae819SStefano Zampini   ierr = ISDestroy(&seqis);CHKERRQ(ierr);
2160674ae819SStefano Zampini   ierr = ISDestroy(&paris);CHKERRQ(ierr);
2161674ae819SStefano Zampini   /* init array */
2162674ae819SStefano Zampini   ierr = VecSet(global_vec,0.0);CHKERRQ(ierr);
2163674ae819SStefano Zampini   ierr = VecSet(local_vec,0.0);CHKERRQ(ierr);
2164674ae819SStefano Zampini   ierr = VecGetArray(local_vec,&array);CHKERRQ(ierr);
2165674ae819SStefano Zampini   if (local_dofs_mult) {
2166674ae819SStefano Zampini     for (i=0;i<n_local_dofs;i++) {
2167674ae819SStefano Zampini       array[local_dofs[i]]=(PetscScalar)local_dofs_mult[i];
2168674ae819SStefano Zampini     }
2169674ae819SStefano Zampini   } else {
2170674ae819SStefano Zampini     for (i=0;i<n_local_dofs;i++) {
2171674ae819SStefano Zampini       array[local_dofs[i]]=1.0;
2172674ae819SStefano Zampini     }
2173674ae819SStefano Zampini   }
2174674ae819SStefano Zampini   ierr = VecRestoreArray(local_vec,&array);CHKERRQ(ierr);
2175674ae819SStefano Zampini   /* scatter into global vec and get total number of global dofs */
2176674ae819SStefano Zampini   ierr = VecScatterBegin(scatter_ctx,local_vec,global_vec,INSERT_VALUES,SCATTER_FORWARD);CHKERRQ(ierr);
2177674ae819SStefano Zampini   ierr = VecScatterEnd(scatter_ctx,local_vec,global_vec,INSERT_VALUES,SCATTER_FORWARD);CHKERRQ(ierr);
2178674ae819SStefano Zampini   ierr = VecSum(global_vec,&globalsum);CHKERRQ(ierr);
21795b08dc53SStefano Zampini   *n_global_subset = (PetscInt)PetscRealPart(globalsum);
2180674ae819SStefano Zampini   /* Fill global_vec with cumulative function for global numbering */
2181674ae819SStefano Zampini   ierr = VecGetArray(global_vec,&array);CHKERRQ(ierr);
2182674ae819SStefano Zampini   ierr = VecGetLocalSize(global_vec,&s);CHKERRQ(ierr);
2183674ae819SStefano Zampini   nlocals = 0;
2184674ae819SStefano Zampini   first_index = -1;
2185674ae819SStefano Zampini   first_found = PETSC_FALSE;
2186674ae819SStefano Zampini   for (i=0;i<s;i++) {
21875b08dc53SStefano Zampini     if (!first_found && PetscRealPart(array[i]) > 0.0) {
2188674ae819SStefano Zampini       first_found = PETSC_TRUE;
2189674ae819SStefano Zampini       first_index = i;
2190674ae819SStefano Zampini     }
21915b08dc53SStefano Zampini     nlocals += (PetscInt)PetscRealPart(array[i]);
2192674ae819SStefano Zampini   }
2193674ae819SStefano Zampini   ierr = MPI_Gather(&nlocals,1,MPIU_INT,dof_sizes,1,MPIU_INT,0,comm);CHKERRQ(ierr);
2194674ae819SStefano Zampini   if (!rank_prec_comm) {
2195674ae819SStefano Zampini     dof_displs[0]=0;
2196674ae819SStefano Zampini     for (i=1;i<size_prec_comm;i++) {
2197674ae819SStefano Zampini       dof_displs[i] = dof_displs[i-1]+dof_sizes[i-1];
2198674ae819SStefano Zampini     }
2199674ae819SStefano Zampini   }
2200674ae819SStefano Zampini   ierr = MPI_Scatter(dof_displs,1,MPIU_INT,&nlocals,1,MPIU_INT,0,comm);CHKERRQ(ierr);
2201674ae819SStefano Zampini   if (first_found) {
2202674ae819SStefano Zampini     array[first_index] += (PetscScalar)nlocals;
2203674ae819SStefano Zampini     old_index = first_index;
2204674ae819SStefano Zampini     for (i=first_index+1;i<s;i++) {
22055b08dc53SStefano Zampini       if (PetscRealPart(array[i]) > 0.0) {
2206674ae819SStefano Zampini         array[i] += array[old_index];
2207674ae819SStefano Zampini         old_index = i;
2208674ae819SStefano Zampini       }
2209674ae819SStefano Zampini     }
2210674ae819SStefano Zampini   }
2211674ae819SStefano Zampini   ierr = VecRestoreArray(global_vec,&array);CHKERRQ(ierr);
2212674ae819SStefano Zampini   ierr = VecSet(local_vec,0.0);CHKERRQ(ierr);
2213674ae819SStefano Zampini   ierr = VecScatterBegin(scatter_ctx,global_vec,local_vec,INSERT_VALUES,SCATTER_REVERSE);CHKERRQ(ierr);
2214674ae819SStefano Zampini   ierr = VecScatterEnd  (scatter_ctx,global_vec,local_vec,INSERT_VALUES,SCATTER_REVERSE);CHKERRQ(ierr);
2215674ae819SStefano Zampini   /* get global ordering of local dofs */
2216674ae819SStefano Zampini   ierr = VecGetArray(local_vec,&array);CHKERRQ(ierr);
2217674ae819SStefano Zampini   if (local_dofs_mult) {
2218674ae819SStefano Zampini     for (i=0;i<n_local_dofs;i++) {
22195b08dc53SStefano Zampini       temp_global_dofs[i] = (PetscInt)PetscRealPart(array[local_dofs[i]])-local_dofs_mult[i];
2220674ae819SStefano Zampini     }
2221674ae819SStefano Zampini   } else {
2222674ae819SStefano Zampini     for (i=0;i<n_local_dofs;i++) {
22235b08dc53SStefano Zampini       temp_global_dofs[i] = (PetscInt)PetscRealPart(array[local_dofs[i]])-1;
2224674ae819SStefano Zampini     }
2225674ae819SStefano Zampini   }
2226674ae819SStefano Zampini   ierr = VecRestoreArray(local_vec,&array);CHKERRQ(ierr);
2227674ae819SStefano Zampini   /* free workspace */
2228674ae819SStefano Zampini   ierr = VecScatterDestroy(&scatter_ctx);CHKERRQ(ierr);
2229674ae819SStefano Zampini   ierr = VecDestroy(&local_vec);CHKERRQ(ierr);
2230674ae819SStefano Zampini   ierr = VecDestroy(&global_vec);CHKERRQ(ierr);
2231674ae819SStefano Zampini   ierr = PetscFree(dof_sizes);CHKERRQ(ierr);
2232674ae819SStefano Zampini   ierr = PetscFree(dof_displs);CHKERRQ(ierr);
2233674ae819SStefano Zampini   /* return pointer to global ordering of local dofs */
2234674ae819SStefano Zampini   *global_numbering_subset = temp_global_dofs;
2235674ae819SStefano Zampini   PetscFunctionReturn(0);
2236674ae819SStefano Zampini }
22379a7d3425SStefano Zampini 
22389a7d3425SStefano Zampini #undef __FUNCT__
22399a7d3425SStefano Zampini #define __FUNCT__ "PCBDDCOrthonormalizeVecs"
22409a7d3425SStefano Zampini PetscErrorCode PCBDDCOrthonormalizeVecs(PetscInt n, Vec vecs[])
22419a7d3425SStefano Zampini {
22429a7d3425SStefano Zampini   PetscInt       i,j;
22439a7d3425SStefano Zampini   PetscScalar    *alphas;
22449a7d3425SStefano Zampini   PetscErrorCode ierr;
22459a7d3425SStefano Zampini 
22469a7d3425SStefano Zampini   PetscFunctionBegin;
22479a7d3425SStefano Zampini   /* this implements stabilized Gram-Schmidt */
22489a7d3425SStefano Zampini   ierr = PetscMalloc(n*sizeof(PetscScalar),&alphas);CHKERRQ(ierr);
22499a7d3425SStefano Zampini   for (i=0;i<n;i++) {
22509a7d3425SStefano Zampini     ierr = VecNormalize(vecs[i],NULL);CHKERRQ(ierr);
22519a7d3425SStefano Zampini     if (i<n) { ierr = VecMDot(vecs[i],n-i-1,&vecs[i+1],&alphas[i+1]);CHKERRQ(ierr); }
22529a7d3425SStefano Zampini     for (j=i+1;j<n;j++) { ierr = VecAXPY(vecs[j],PetscConj(-alphas[j]),vecs[i]);CHKERRQ(ierr); }
22539a7d3425SStefano Zampini   }
22549a7d3425SStefano Zampini   ierr = PetscFree(alphas);CHKERRQ(ierr);
22559a7d3425SStefano Zampini   PetscFunctionReturn(0);
22569a7d3425SStefano Zampini }
22579a7d3425SStefano Zampini 
22589a7d3425SStefano Zampini 
2259