xref: /petsc/src/ksp/pc/impls/bddc/bddc.c (revision f1a72664264a09e4d4566c61c72d046451d30565)
153cdbc3dSStefano Zampini /* TODOLIST
2eb97c9d2SStefano Zampini 
3eb97c9d2SStefano Zampini    Solvers
4a0d3c3abSStefano Zampini    - Add support for cholesky for coarse solver (similar to local solvers)
5eb97c9d2SStefano Zampini    - Propagate ksp prefixes for solvers to mat objects?
6eb97c9d2SStefano Zampini 
7eb97c9d2SStefano Zampini    User interface
80f202f7eSStefano Zampini    - ** DM attached to pc?
9eb97c9d2SStefano Zampini 
10eb97c9d2SStefano Zampini    Debugging output
11b9b85e73SStefano Zampini    - * Better management of verbosity levels of debugging output
12eb97c9d2SStefano Zampini 
13eb97c9d2SStefano Zampini    Extra
14b9b85e73SStefano Zampini    - *** Is it possible to work with PCBDDCGraph on boundary indices only (less memory consumed)?
15eb97c9d2SStefano Zampini    - BDDC with MG framework?
16eb97c9d2SStefano Zampini 
17eb97c9d2SStefano Zampini    FETIDP
18eb97c9d2SStefano Zampini    - Move FETIDP code to its own classes
19eb97c9d2SStefano Zampini 
20eb97c9d2SStefano Zampini    MATIS related operations contained in BDDC code
21eb97c9d2SStefano Zampini    - Provide general case for subassembling
22eb97c9d2SStefano Zampini 
2353cdbc3dSStefano Zampini */
240c7d97c5SJed Brown 
25ab5c6b0cSJed Brown #include <../src/ksp/pc/impls/bddc/bddc.h> /*I "petscpc.h" I*/  /* includes for fortran wrappers */
26ab5c6b0cSJed Brown #include <../src/ksp/pc/impls/bddc/bddcprivate.h>
273b03a366Sstefano_zampini #include <petscblaslapack.h>
28674ae819SStefano Zampini 
290369aaf7SStefano Zampini /* temporarily declare it */
300369aaf7SStefano Zampini PetscErrorCode PCApply_BDDC(PC,Vec,Vec);
310369aaf7SStefano Zampini 
320c7d97c5SJed Brown /* -------------------------------------------------------------------------- */
330c7d97c5SJed Brown #undef __FUNCT__
340c7d97c5SJed Brown #define __FUNCT__ "PCSetFromOptions_BDDC"
358c34d3f5SBarry Smith PetscErrorCode PCSetFromOptions_BDDC(PetscOptions *PetscOptionsObject,PC pc)
360c7d97c5SJed Brown {
370c7d97c5SJed Brown   PC_BDDC        *pcbddc = (PC_BDDC*)pc->data;
380c7d97c5SJed Brown   PetscErrorCode ierr;
390c7d97c5SJed Brown 
400c7d97c5SJed Brown   PetscFunctionBegin;
41e55864a3SBarry Smith   ierr = PetscOptionsHead(PetscOptionsObject,"BDDC options");CHKERRQ(ierr);
428eeda7d8SStefano Zampini   /* Verbose debugging */
438eeda7d8SStefano Zampini   ierr = PetscOptionsInt("-pc_bddc_check_level","Verbose output for PCBDDC (intended for debug)","none",pcbddc->dbg_flag,&pcbddc->dbg_flag,NULL);CHKERRQ(ierr);
448eeda7d8SStefano Zampini   /* Primal space cumstomization */
4508a5cf49SStefano Zampini   ierr = PetscOptionsBool("-pc_bddc_use_local_mat_graph","Use or not adjacency graph of local mat for interface analysis","none",pcbddc->use_local_adj,&pcbddc->use_local_adj,NULL);CHKERRQ(ierr);
468eeda7d8SStefano Zampini   ierr = PetscOptionsBool("-pc_bddc_use_vertices","Use or not corner dofs in coarse space","none",pcbddc->use_vertices,&pcbddc->use_vertices,NULL);CHKERRQ(ierr);
478eeda7d8SStefano Zampini   ierr = PetscOptionsBool("-pc_bddc_use_edges","Use or not edge constraints in coarse space","none",pcbddc->use_edges,&pcbddc->use_edges,NULL);CHKERRQ(ierr);
488eeda7d8SStefano Zampini   ierr = PetscOptionsBool("-pc_bddc_use_faces","Use or not face constraints in coarse space","none",pcbddc->use_faces,&pcbddc->use_faces,NULL);CHKERRQ(ierr);
496661aa1dSStefano Zampini   ierr = PetscOptionsBool("-pc_bddc_use_true_nnsp","Use near null space attached to the matrix without modifications","none",pcbddc->use_nnsp_true,&pcbddc->use_nnsp_true,NULL);CHKERRQ(ierr);
50ac632422SStefano Zampini   ierr = PetscOptionsBool("-pc_bddc_use_qr_single","Use QR factorization for single constraints on cc (QR is used when multiple constraints are present)","none",pcbddc->use_qr_single,&pcbddc->use_qr_single,NULL);CHKERRQ(ierr);
518eeda7d8SStefano Zampini   /* Change of basis */
52b9b85e73SStefano Zampini   ierr = PetscOptionsBool("-pc_bddc_use_change_of_basis","Use or not internal change of basis on local edge nodes","none",pcbddc->use_change_of_basis,&pcbddc->use_change_of_basis,NULL);CHKERRQ(ierr);
53b9b85e73SStefano Zampini   ierr = PetscOptionsBool("-pc_bddc_use_change_on_faces","Use or not internal change of basis on local face nodes","none",pcbddc->use_change_on_faces,&pcbddc->use_change_on_faces,NULL);CHKERRQ(ierr);
54674ae819SStefano Zampini   if (!pcbddc->use_change_of_basis) {
55674ae819SStefano Zampini     pcbddc->use_change_on_faces = PETSC_FALSE;
56674ae819SStefano Zampini   }
578eeda7d8SStefano Zampini   /* Switch between M_2 (default) and M_3 preconditioners (as defined by C. Dohrmann in the ref. article) */
588eeda7d8SStefano Zampini   ierr = PetscOptionsBool("-pc_bddc_switch_static","Switch on static condensation ops around the interface preconditioner","none",pcbddc->switch_static,&pcbddc->switch_static,NULL);CHKERRQ(ierr);
5974e2c79eSStefano Zampini   ierr = PetscOptionsInt("-pc_bddc_coarse_redistribute","Number of procs where to redistribute coarse problem","none",pcbddc->redistribute_coarse,&pcbddc->redistribute_coarse,NULL);CHKERRQ(ierr);
600298fd71SBarry Smith   ierr = PetscOptionsInt("-pc_bddc_coarsening_ratio","Set coarsening ratio used in multilevel coarsening","none",pcbddc->coarsening_ratio,&pcbddc->coarsening_ratio,NULL);CHKERRQ(ierr);
612b510759SStefano Zampini   ierr = PetscOptionsInt("-pc_bddc_levels","Set maximum number of levels for multilevel","none",pcbddc->max_levels,&pcbddc->max_levels,NULL);CHKERRQ(ierr);
62323d395dSStefano Zampini   ierr = PetscOptionsBool("-pc_bddc_use_coarse_estimates","Use estimated eigenvalues for coarse problem","none",pcbddc->use_coarse_estimates,&pcbddc->use_coarse_estimates,NULL);CHKERRQ(ierr);
63674ae819SStefano Zampini   ierr = PetscOptionsBool("-pc_bddc_use_deluxe_scaling","Use deluxe scaling for BDDC","none",pcbddc->use_deluxe_scaling,&pcbddc->use_deluxe_scaling,NULL);CHKERRQ(ierr);
64b96c3477SStefano Zampini   ierr = PetscOptionsBool("-pc_bddc_schur_rebuild","Whether or not the interface graph for Schur principal minors has to be rebuilt (i.e. define the interface without any adjacency)","none",pcbddc->sub_schurs_rebuild,&pcbddc->sub_schurs_rebuild,NULL);CHKERRQ(ierr);
65b96c3477SStefano Zampini   ierr = PetscOptionsInt("-pc_bddc_schur_layers","Number of dofs' layers for the computation of principal minors (i.e. -1 uses all dofs)","none",pcbddc->sub_schurs_layers,&pcbddc->sub_schurs_layers,NULL);CHKERRQ(ierr);
66b96c3477SStefano Zampini   ierr = PetscOptionsBool("-pc_bddc_schur_use_useradj","Whether or not the CSR graph specified by the user should be used for computing successive layers (default is to use adj of local mat)","none",pcbddc->sub_schurs_use_useradj,&pcbddc->sub_schurs_use_useradj,NULL);CHKERRQ(ierr);
67ac632422SStefano Zampini   ierr = PetscOptionsBool("-pc_bddc_deluxe_faster","Faster application of deluxe scaling (requires extra work during setup)","none",pcbddc->faster_deluxe,&pcbddc->faster_deluxe,NULL);CHKERRQ(ierr);
684c6709b3SStefano Zampini   ierr = PetscOptionsReal("-pc_bddc_adaptive_threshold","Threshold to be used for adaptive selection of constraints","none",pcbddc->adaptive_threshold,&pcbddc->adaptive_threshold,NULL);CHKERRQ(ierr);
6908122e43SStefano Zampini   ierr = PetscOptionsInt("-pc_bddc_adaptive_nmin","Minimum number of constraints per connected components","none",pcbddc->adaptive_nmin,&pcbddc->adaptive_nmin,NULL);CHKERRQ(ierr);
7008122e43SStefano Zampini   ierr = PetscOptionsInt("-pc_bddc_adaptive_nmax","Maximum number of constraints per connected components","none",pcbddc->adaptive_nmax,&pcbddc->adaptive_nmax,NULL);CHKERRQ(ierr);
713301b35fSStefano Zampini   ierr = PetscOptionsBool("-pc_bddc_symmetric","Symmetric computation of primal basis functions","none",pcbddc->symmetric_primal,&pcbddc->symmetric_primal,NULL);CHKERRQ(ierr);
72b0c7d250SStefano Zampini   ierr = PetscOptionsInt("-pc_bddc_coarse_adj","Number of processors where to map the coarse adjacency list","none",pcbddc->coarse_adj_red,&pcbddc->coarse_adj_red,NULL);CHKERRQ(ierr);
7306a4e24aSStefano Zampini   ierr = PetscOptionsBool("-pc_bddc_benign_trick","Apply the benign subspace trick to a class of saddle point problems","none",pcbddc->benign_saddle_point,&pcbddc->benign_saddle_point,NULL);CHKERRQ(ierr);
740c7d97c5SJed Brown   ierr = PetscOptionsTail();CHKERRQ(ierr);
750c7d97c5SJed Brown   PetscFunctionReturn(0);
760c7d97c5SJed Brown }
770c7d97c5SJed Brown /* -------------------------------------------------------------------------- */
78674ae819SStefano Zampini #undef __FUNCT__
79906d46d4SStefano Zampini #define __FUNCT__ "PCBDDCSetChangeOfBasisMat_BDDC"
80906d46d4SStefano Zampini static PetscErrorCode PCBDDCSetChangeOfBasisMat_BDDC(PC pc, Mat change)
81b9b85e73SStefano Zampini {
82b9b85e73SStefano Zampini   PC_BDDC        *pcbddc = (PC_BDDC*)pc->data;
83b9b85e73SStefano Zampini   PetscErrorCode ierr;
84b9b85e73SStefano Zampini 
85b9b85e73SStefano Zampini   PetscFunctionBegin;
86b9b85e73SStefano Zampini   ierr = MatDestroy(&pcbddc->user_ChangeOfBasisMatrix);CHKERRQ(ierr);
87b9b85e73SStefano Zampini   ierr = PetscObjectReference((PetscObject)change);CHKERRQ(ierr);
88b9b85e73SStefano Zampini   pcbddc->user_ChangeOfBasisMatrix = change;
89b9b85e73SStefano Zampini   PetscFunctionReturn(0);
90b9b85e73SStefano Zampini }
91b9b85e73SStefano Zampini #undef __FUNCT__
92906d46d4SStefano Zampini #define __FUNCT__ "PCBDDCSetChangeOfBasisMat"
93b9b85e73SStefano Zampini /*@
94906d46d4SStefano Zampini  PCBDDCSetChangeOfBasisMat - Set user defined change of basis for dofs
95b9b85e73SStefano Zampini 
96b9b85e73SStefano Zampini    Collective on PC
97b9b85e73SStefano Zampini 
98b9b85e73SStefano Zampini    Input Parameters:
99b9b85e73SStefano Zampini +  pc - the preconditioning context
100906d46d4SStefano Zampini -  change - the change of basis matrix
101b9b85e73SStefano Zampini 
102b9b85e73SStefano Zampini    Level: intermediate
103b9b85e73SStefano Zampini 
104b9b85e73SStefano Zampini    Notes:
105b9b85e73SStefano Zampini 
106b9b85e73SStefano Zampini .seealso: PCBDDC
107b9b85e73SStefano Zampini @*/
108906d46d4SStefano Zampini PetscErrorCode PCBDDCSetChangeOfBasisMat(PC pc, Mat change)
109b9b85e73SStefano Zampini {
110b9b85e73SStefano Zampini   PetscErrorCode ierr;
111b9b85e73SStefano Zampini 
112b9b85e73SStefano Zampini   PetscFunctionBegin;
113b9b85e73SStefano Zampini   PetscValidHeaderSpecific(pc,PC_CLASSID,1);
114b9b85e73SStefano Zampini   PetscValidHeaderSpecific(change,MAT_CLASSID,2);
115906d46d4SStefano Zampini   PetscCheckSameComm(pc,1,change,2);
116906d46d4SStefano Zampini   if (pc->mat) {
117906d46d4SStefano Zampini     PetscInt rows_c,cols_c,rows,cols;
118906d46d4SStefano Zampini     ierr = MatGetSize(pc->mat,&rows,&cols);CHKERRQ(ierr);
119906d46d4SStefano Zampini     ierr = MatGetSize(change,&rows_c,&cols_c);CHKERRQ(ierr);
120906d46d4SStefano Zampini     if (rows_c != rows) {
121906d46d4SStefano Zampini       SETERRQ2(PetscObjectComm((PetscObject)pc),PETSC_ERR_SUP,"Invalid number of rows for change of basis matrix! %d != %d",rows_c,rows);
122906d46d4SStefano Zampini     }
123906d46d4SStefano Zampini     if (cols_c != cols) {
124906d46d4SStefano Zampini       SETERRQ2(PetscObjectComm((PetscObject)pc),PETSC_ERR_SUP,"Invalid number of columns for change of basis matrix! %d != %d",cols_c,cols);
125906d46d4SStefano Zampini     }
126906d46d4SStefano Zampini     ierr = MatGetLocalSize(pc->mat,&rows,&cols);CHKERRQ(ierr);
127906d46d4SStefano Zampini     ierr = MatGetLocalSize(change,&rows_c,&cols_c);CHKERRQ(ierr);
128906d46d4SStefano Zampini     if (rows_c != rows) {
129906d46d4SStefano Zampini       SETERRQ2(PetscObjectComm((PetscObject)pc),PETSC_ERR_SUP,"Invalid number of local rows for change of basis matrix! %d != %d",rows_c,rows);
130906d46d4SStefano Zampini     }
131906d46d4SStefano Zampini     if (cols_c != cols) {
132906d46d4SStefano Zampini       SETERRQ2(PetscObjectComm((PetscObject)pc),PETSC_ERR_SUP,"Invalid number of local columns for change of basis matrix! %d != %d",cols_c,cols);
133906d46d4SStefano Zampini     }
134906d46d4SStefano Zampini   }
135906d46d4SStefano Zampini   ierr = PetscTryMethod(pc,"PCBDDCSetChangeOfBasisMat_C",(PC,Mat),(pc,change));CHKERRQ(ierr);
136b9b85e73SStefano Zampini   PetscFunctionReturn(0);
137b9b85e73SStefano Zampini }
138b9b85e73SStefano Zampini /* -------------------------------------------------------------------------- */
139b9b85e73SStefano Zampini #undef __FUNCT__
140674ae819SStefano Zampini #define __FUNCT__ "PCBDDCSetPrimalVerticesLocalIS_BDDC"
141674ae819SStefano Zampini static PetscErrorCode PCBDDCSetPrimalVerticesLocalIS_BDDC(PC pc, IS PrimalVertices)
142674ae819SStefano Zampini {
143674ae819SStefano Zampini   PC_BDDC        *pcbddc = (PC_BDDC*)pc->data;
144674ae819SStefano Zampini   PetscErrorCode ierr;
1451e6b0712SBarry Smith 
146674ae819SStefano Zampini   PetscFunctionBegin;
147674ae819SStefano Zampini   ierr = ISDestroy(&pcbddc->user_primal_vertices);CHKERRQ(ierr);
148674ae819SStefano Zampini   ierr = PetscObjectReference((PetscObject)PrimalVertices);CHKERRQ(ierr);
149674ae819SStefano Zampini   pcbddc->user_primal_vertices = PrimalVertices;
150674ae819SStefano Zampini   PetscFunctionReturn(0);
151674ae819SStefano Zampini }
152674ae819SStefano Zampini #undef __FUNCT__
153674ae819SStefano Zampini #define __FUNCT__ "PCBDDCSetPrimalVerticesLocalIS"
154674ae819SStefano Zampini /*@
15528509bceSStefano Zampini  PCBDDCSetPrimalVerticesLocalIS - Set additional user defined primal vertices in PCBDDC
156674ae819SStefano Zampini 
15717eb1463SStefano Zampini    Collective
158674ae819SStefano Zampini 
159674ae819SStefano Zampini    Input Parameters:
160674ae819SStefano Zampini +  pc - the preconditioning context
16117eb1463SStefano Zampini -  PrimalVertices - index set of primal vertices in local numbering (can be empty)
162674ae819SStefano Zampini 
163674ae819SStefano Zampini    Level: intermediate
164674ae819SStefano Zampini 
165674ae819SStefano Zampini    Notes:
166674ae819SStefano Zampini 
167674ae819SStefano Zampini .seealso: PCBDDC
168674ae819SStefano Zampini @*/
169674ae819SStefano Zampini PetscErrorCode PCBDDCSetPrimalVerticesLocalIS(PC pc, IS PrimalVertices)
170674ae819SStefano Zampini {
171674ae819SStefano Zampini   PetscErrorCode ierr;
172674ae819SStefano Zampini 
173674ae819SStefano Zampini   PetscFunctionBegin;
174674ae819SStefano Zampini   PetscValidHeaderSpecific(pc,PC_CLASSID,1);
175674ae819SStefano Zampini   PetscValidHeaderSpecific(PrimalVertices,IS_CLASSID,2);
17617eb1463SStefano Zampini   PetscCheckSameComm(pc,1,PrimalVertices,2);
177674ae819SStefano Zampini   ierr = PetscTryMethod(pc,"PCBDDCSetPrimalVerticesLocalIS_C",(PC,IS),(pc,PrimalVertices));CHKERRQ(ierr);
178674ae819SStefano Zampini   PetscFunctionReturn(0);
179674ae819SStefano Zampini }
180674ae819SStefano Zampini /* -------------------------------------------------------------------------- */
1810c7d97c5SJed Brown #undef __FUNCT__
1824fad6a16SStefano Zampini #define __FUNCT__ "PCBDDCSetCoarseningRatio_BDDC"
1834fad6a16SStefano Zampini static PetscErrorCode PCBDDCSetCoarseningRatio_BDDC(PC pc,PetscInt k)
1844fad6a16SStefano Zampini {
1854fad6a16SStefano Zampini   PC_BDDC  *pcbddc = (PC_BDDC*)pc->data;
1864fad6a16SStefano Zampini 
1874fad6a16SStefano Zampini   PetscFunctionBegin;
1884fad6a16SStefano Zampini   pcbddc->coarsening_ratio = k;
1894fad6a16SStefano Zampini   PetscFunctionReturn(0);
1904fad6a16SStefano Zampini }
1911e6b0712SBarry Smith 
1924fad6a16SStefano Zampini #undef __FUNCT__
1934fad6a16SStefano Zampini #define __FUNCT__ "PCBDDCSetCoarseningRatio"
1944fad6a16SStefano Zampini /*@
19528509bceSStefano Zampini  PCBDDCSetCoarseningRatio - Set coarsening ratio used in multilevel
1964fad6a16SStefano Zampini 
1974fad6a16SStefano Zampini    Logically collective on PC
1984fad6a16SStefano Zampini 
1994fad6a16SStefano Zampini    Input Parameters:
2004fad6a16SStefano Zampini +  pc - the preconditioning context
20128509bceSStefano Zampini -  k - coarsening ratio (H/h at the coarser level)
2024fad6a16SStefano Zampini 
2030f202f7eSStefano Zampini    Options Database Keys:
2040f202f7eSStefano Zampini .    -pc_bddc_coarsening_ratio
2054fad6a16SStefano Zampini 
2064fad6a16SStefano Zampini    Level: intermediate
2074fad6a16SStefano Zampini 
2084fad6a16SStefano Zampini    Notes:
2090f202f7eSStefano Zampini      Approximatively k subdomains at the finer level will be aggregated into a single subdomain at the coarser level
2104fad6a16SStefano Zampini 
2110f202f7eSStefano Zampini .seealso: PCBDDC, PCBDDCSetLevels()
2124fad6a16SStefano Zampini @*/
2134fad6a16SStefano Zampini PetscErrorCode PCBDDCSetCoarseningRatio(PC pc,PetscInt k)
2144fad6a16SStefano Zampini {
2154fad6a16SStefano Zampini   PetscErrorCode ierr;
2164fad6a16SStefano Zampini 
2174fad6a16SStefano Zampini   PetscFunctionBegin;
2184fad6a16SStefano Zampini   PetscValidHeaderSpecific(pc,PC_CLASSID,1);
2192b510759SStefano Zampini   PetscValidLogicalCollectiveInt(pc,k,2);
2204fad6a16SStefano Zampini   ierr = PetscTryMethod(pc,"PCBDDCSetCoarseningRatio_C",(PC,PetscInt),(pc,k));CHKERRQ(ierr);
2214fad6a16SStefano Zampini   PetscFunctionReturn(0);
2224fad6a16SStefano Zampini }
2232b510759SStefano Zampini 
224b8ffe317SStefano Zampini /* The following functions (PCBDDCSetUseExactDirichlet PCBDDCSetLevel) are not public */
2252b510759SStefano Zampini #undef __FUNCT__
226b8ffe317SStefano Zampini #define __FUNCT__ "PCBDDCSetUseExactDirichlet_BDDC"
227b8ffe317SStefano Zampini static PetscErrorCode PCBDDCSetUseExactDirichlet_BDDC(PC pc,PetscBool flg)
228b8ffe317SStefano Zampini {
229b8ffe317SStefano Zampini   PC_BDDC  *pcbddc = (PC_BDDC*)pc->data;
230b8ffe317SStefano Zampini 
231b8ffe317SStefano Zampini   PetscFunctionBegin;
23285c4d303SStefano Zampini   pcbddc->use_exact_dirichlet_trick = flg;
233b8ffe317SStefano Zampini   PetscFunctionReturn(0);
234b8ffe317SStefano Zampini }
235b8ffe317SStefano Zampini 
236b8ffe317SStefano Zampini #undef __FUNCT__
237b8ffe317SStefano Zampini #define __FUNCT__ "PCBDDCSetUseExactDirichlet"
238b8ffe317SStefano Zampini PetscErrorCode PCBDDCSetUseExactDirichlet(PC pc,PetscBool flg)
2392b510759SStefano Zampini {
2402b510759SStefano Zampini   PetscErrorCode ierr;
2412b510759SStefano Zampini 
2422b510759SStefano Zampini   PetscFunctionBegin;
2432b510759SStefano Zampini   PetscValidHeaderSpecific(pc,PC_CLASSID,1);
244b8ffe317SStefano Zampini   PetscValidLogicalCollectiveBool(pc,flg,2);
245b8ffe317SStefano Zampini   ierr = PetscTryMethod(pc,"PCBDDCSetUseExactDirichlet_C",(PC,PetscBool),(pc,flg));CHKERRQ(ierr);
2462b510759SStefano Zampini   PetscFunctionReturn(0);
2472b510759SStefano Zampini }
2481e6b0712SBarry Smith 
2494fad6a16SStefano Zampini #undef __FUNCT__
2502b510759SStefano Zampini #define __FUNCT__ "PCBDDCSetLevel_BDDC"
2512b510759SStefano Zampini static PetscErrorCode PCBDDCSetLevel_BDDC(PC pc,PetscInt level)
2524fad6a16SStefano Zampini {
2534fad6a16SStefano Zampini   PC_BDDC  *pcbddc = (PC_BDDC*)pc->data;
2544fad6a16SStefano Zampini 
2554fad6a16SStefano Zampini   PetscFunctionBegin;
2562b510759SStefano Zampini   pcbddc->current_level = level;
2574fad6a16SStefano Zampini   PetscFunctionReturn(0);
2584fad6a16SStefano Zampini }
2591e6b0712SBarry Smith 
2604fad6a16SStefano Zampini #undef __FUNCT__
261b8ffe317SStefano Zampini #define __FUNCT__ "PCBDDCSetLevel"
262b8ffe317SStefano Zampini PetscErrorCode PCBDDCSetLevel(PC pc,PetscInt level)
263b8ffe317SStefano Zampini {
264b8ffe317SStefano Zampini   PetscErrorCode ierr;
265b8ffe317SStefano Zampini 
266b8ffe317SStefano Zampini   PetscFunctionBegin;
267b8ffe317SStefano Zampini   PetscValidHeaderSpecific(pc,PC_CLASSID,1);
268b8ffe317SStefano Zampini   PetscValidLogicalCollectiveInt(pc,level,2);
269b8ffe317SStefano Zampini   ierr = PetscTryMethod(pc,"PCBDDCSetLevel_C",(PC,PetscInt),(pc,level));CHKERRQ(ierr);
270b8ffe317SStefano Zampini   PetscFunctionReturn(0);
271b8ffe317SStefano Zampini }
272b8ffe317SStefano Zampini 
273b8ffe317SStefano Zampini #undef __FUNCT__
2742b510759SStefano Zampini #define __FUNCT__ "PCBDDCSetLevels_BDDC"
2752b510759SStefano Zampini static PetscErrorCode PCBDDCSetLevels_BDDC(PC pc,PetscInt levels)
2762b510759SStefano Zampini {
2772b510759SStefano Zampini   PC_BDDC  *pcbddc = (PC_BDDC*)pc->data;
2782b510759SStefano Zampini 
2792b510759SStefano Zampini   PetscFunctionBegin;
2802b510759SStefano Zampini   pcbddc->max_levels = levels;
2812b510759SStefano Zampini   PetscFunctionReturn(0);
2822b510759SStefano Zampini }
2832b510759SStefano Zampini 
2842b510759SStefano Zampini #undef __FUNCT__
2852b510759SStefano Zampini #define __FUNCT__ "PCBDDCSetLevels"
2864fad6a16SStefano Zampini /*@
28728509bceSStefano Zampini  PCBDDCSetLevels - Sets the maximum number of levels for multilevel
2884fad6a16SStefano Zampini 
2894fad6a16SStefano Zampini    Logically collective on PC
2904fad6a16SStefano Zampini 
2914fad6a16SStefano Zampini    Input Parameters:
2924fad6a16SStefano Zampini +  pc - the preconditioning context
29328509bceSStefano Zampini -  levels - the maximum number of levels (max 9)
2944fad6a16SStefano Zampini 
2950f202f7eSStefano Zampini    Options Database Keys:
2960f202f7eSStefano Zampini .    -pc_bddc_levels
2974fad6a16SStefano Zampini 
2984fad6a16SStefano Zampini    Level: intermediate
2994fad6a16SStefano Zampini 
3004fad6a16SStefano Zampini    Notes:
3010f202f7eSStefano Zampini      Default value is 0, i.e. traditional one-level BDDC
3024fad6a16SStefano Zampini 
3030f202f7eSStefano Zampini .seealso: PCBDDC, PCBDDCSetCoarseningRatio()
3044fad6a16SStefano Zampini @*/
3052b510759SStefano Zampini PetscErrorCode PCBDDCSetLevels(PC pc,PetscInt levels)
3064fad6a16SStefano Zampini {
3074fad6a16SStefano Zampini   PetscErrorCode ierr;
3084fad6a16SStefano Zampini 
3094fad6a16SStefano Zampini   PetscFunctionBegin;
3104fad6a16SStefano Zampini   PetscValidHeaderSpecific(pc,PC_CLASSID,1);
3112b510759SStefano Zampini   PetscValidLogicalCollectiveInt(pc,levels,2);
312312be037SStefano Zampini   if (levels > 99) SETERRQ(PetscObjectComm((PetscObject)pc),PETSC_ERR_SUP,"Maximum number of levels for bddc is 99\n");
3132b510759SStefano Zampini   ierr = PetscTryMethod(pc,"PCBDDCSetLevels_C",(PC,PetscInt),(pc,levels));CHKERRQ(ierr);
3144fad6a16SStefano Zampini   PetscFunctionReturn(0);
3154fad6a16SStefano Zampini }
3164fad6a16SStefano Zampini /* -------------------------------------------------------------------------- */
3171e6b0712SBarry Smith 
3184fad6a16SStefano Zampini #undef __FUNCT__
3190bdf917eSStefano Zampini #define __FUNCT__ "PCBDDCSetNullSpace_BDDC"
3200bdf917eSStefano Zampini static PetscErrorCode PCBDDCSetNullSpace_BDDC(PC pc,MatNullSpace NullSpace)
3210bdf917eSStefano Zampini {
3220bdf917eSStefano Zampini   PC_BDDC  *pcbddc = (PC_BDDC*)pc->data;
3230bdf917eSStefano Zampini   PetscErrorCode ierr;
3240bdf917eSStefano Zampini 
3250bdf917eSStefano Zampini   PetscFunctionBegin;
3260bdf917eSStefano Zampini   ierr = PetscObjectReference((PetscObject)NullSpace);CHKERRQ(ierr);
3270bdf917eSStefano Zampini   ierr = MatNullSpaceDestroy(&pcbddc->NullSpace);CHKERRQ(ierr);
3280bdf917eSStefano Zampini   pcbddc->NullSpace = NullSpace;
3290bdf917eSStefano Zampini   PetscFunctionReturn(0);
3300bdf917eSStefano Zampini }
3311e6b0712SBarry Smith 
3320bdf917eSStefano Zampini #undef __FUNCT__
3330bdf917eSStefano Zampini #define __FUNCT__ "PCBDDCSetNullSpace"
3340bdf917eSStefano Zampini /*@
33528509bceSStefano Zampini  PCBDDCSetNullSpace - Set nullspace for BDDC operator
3360bdf917eSStefano Zampini 
3370bdf917eSStefano Zampini    Logically collective on PC and MatNullSpace
3380bdf917eSStefano Zampini 
3390bdf917eSStefano Zampini    Input Parameters:
3400bdf917eSStefano Zampini +  pc - the preconditioning context
34128509bceSStefano Zampini -  NullSpace - Null space of the linear operator to be preconditioned (Pmat)
3420bdf917eSStefano Zampini 
3430bdf917eSStefano Zampini    Level: intermediate
3440bdf917eSStefano Zampini 
3450bdf917eSStefano Zampini    Notes:
3460bdf917eSStefano Zampini 
3470bdf917eSStefano Zampini .seealso: PCBDDC
3480bdf917eSStefano Zampini @*/
3490bdf917eSStefano Zampini PetscErrorCode PCBDDCSetNullSpace(PC pc,MatNullSpace NullSpace)
3500bdf917eSStefano Zampini {
3510bdf917eSStefano Zampini   PetscErrorCode ierr;
3520bdf917eSStefano Zampini 
3530bdf917eSStefano Zampini   PetscFunctionBegin;
3540bdf917eSStefano Zampini   PetscValidHeaderSpecific(pc,PC_CLASSID,1);
355674ae819SStefano Zampini   PetscValidHeaderSpecific(NullSpace,MAT_NULLSPACE_CLASSID,2);
3562b510759SStefano Zampini   PetscCheckSameComm(pc,1,NullSpace,2);
3570bdf917eSStefano Zampini   ierr = PetscTryMethod(pc,"PCBDDCSetNullSpace_C",(PC,MatNullSpace),(pc,NullSpace));CHKERRQ(ierr);
3580bdf917eSStefano Zampini   PetscFunctionReturn(0);
3590bdf917eSStefano Zampini }
3600bdf917eSStefano Zampini /* -------------------------------------------------------------------------- */
3611e6b0712SBarry Smith 
3620bdf917eSStefano Zampini #undef __FUNCT__
3633b03a366Sstefano_zampini #define __FUNCT__ "PCBDDCSetDirichletBoundaries_BDDC"
3643b03a366Sstefano_zampini static PetscErrorCode PCBDDCSetDirichletBoundaries_BDDC(PC pc,IS DirichletBoundaries)
3653b03a366Sstefano_zampini {
3663b03a366Sstefano_zampini   PC_BDDC  *pcbddc = (PC_BDDC*)pc->data;
3673b03a366Sstefano_zampini   PetscErrorCode ierr;
3683b03a366Sstefano_zampini 
3693b03a366Sstefano_zampini   PetscFunctionBegin;
370785d1243SStefano Zampini   /* last user setting takes precendence -> destroy any other customization */
371785d1243SStefano Zampini   ierr = ISDestroy(&pcbddc->DirichletBoundariesLocal);CHKERRQ(ierr);
3723b03a366Sstefano_zampini   ierr = ISDestroy(&pcbddc->DirichletBoundaries);CHKERRQ(ierr);
37336e030ebSStefano Zampini   ierr = PetscObjectReference((PetscObject)DirichletBoundaries);CHKERRQ(ierr);
37436e030ebSStefano Zampini   pcbddc->DirichletBoundaries = DirichletBoundaries;
375fb180af4SStefano Zampini   pcbddc->recompute_topography = PETSC_TRUE;
3763b03a366Sstefano_zampini   PetscFunctionReturn(0);
3773b03a366Sstefano_zampini }
3781e6b0712SBarry Smith 
3793b03a366Sstefano_zampini #undef __FUNCT__
3803b03a366Sstefano_zampini #define __FUNCT__ "PCBDDCSetDirichletBoundaries"
3813b03a366Sstefano_zampini /*@
38228509bceSStefano Zampini  PCBDDCSetDirichletBoundaries - Set IS defining Dirichlet boundaries for the global problem.
3833b03a366Sstefano_zampini 
384785d1243SStefano Zampini    Collective
3853b03a366Sstefano_zampini 
3863b03a366Sstefano_zampini    Input Parameters:
3873b03a366Sstefano_zampini +  pc - the preconditioning context
388785d1243SStefano Zampini -  DirichletBoundaries - parallel IS defining the Dirichlet boundaries
3893b03a366Sstefano_zampini 
3903b03a366Sstefano_zampini    Level: intermediate
3913b03a366Sstefano_zampini 
3920f202f7eSStefano Zampini    Notes:
3930f202f7eSStefano Zampini      Provide the information if you used MatZeroRows/Columns routines. Any process can list any global node
3943b03a366Sstefano_zampini 
3950f202f7eSStefano Zampini .seealso: PCBDDC, PCBDDCSetDirichletBoundariesLocal(), MatZeroRows(), MatZeroRowsColumns()
3963b03a366Sstefano_zampini @*/
3973b03a366Sstefano_zampini PetscErrorCode PCBDDCSetDirichletBoundaries(PC pc,IS DirichletBoundaries)
3983b03a366Sstefano_zampini {
3993b03a366Sstefano_zampini   PetscErrorCode ierr;
4003b03a366Sstefano_zampini 
4013b03a366Sstefano_zampini   PetscFunctionBegin;
4023b03a366Sstefano_zampini   PetscValidHeaderSpecific(pc,PC_CLASSID,1);
403674ae819SStefano Zampini   PetscValidHeaderSpecific(DirichletBoundaries,IS_CLASSID,2);
404785d1243SStefano Zampini   PetscCheckSameComm(pc,1,DirichletBoundaries,2);
4053b03a366Sstefano_zampini   ierr = PetscTryMethod(pc,"PCBDDCSetDirichletBoundaries_C",(PC,IS),(pc,DirichletBoundaries));CHKERRQ(ierr);
4063b03a366Sstefano_zampini   PetscFunctionReturn(0);
4073b03a366Sstefano_zampini }
4083b03a366Sstefano_zampini /* -------------------------------------------------------------------------- */
4091e6b0712SBarry Smith 
4103b03a366Sstefano_zampini #undef __FUNCT__
41182ba6b80SStefano Zampini #define __FUNCT__ "PCBDDCSetDirichletBoundariesLocal_BDDC"
41282ba6b80SStefano Zampini static PetscErrorCode PCBDDCSetDirichletBoundariesLocal_BDDC(PC pc,IS DirichletBoundaries)
4130c7d97c5SJed Brown {
4140c7d97c5SJed Brown   PC_BDDC  *pcbddc = (PC_BDDC*)pc->data;
4150c7d97c5SJed Brown   PetscErrorCode ierr;
4160c7d97c5SJed Brown 
4170c7d97c5SJed Brown   PetscFunctionBegin;
418785d1243SStefano Zampini   /* last user setting takes precendence -> destroy any other customization */
419785d1243SStefano Zampini   ierr = ISDestroy(&pcbddc->DirichletBoundariesLocal);CHKERRQ(ierr);
4200c7d97c5SJed Brown   ierr = ISDestroy(&pcbddc->DirichletBoundaries);CHKERRQ(ierr);
4210c7d97c5SJed Brown   ierr = PetscObjectReference((PetscObject)DirichletBoundaries);CHKERRQ(ierr);
422785d1243SStefano Zampini   pcbddc->DirichletBoundariesLocal = DirichletBoundaries;
4237d24d155SStefano Zampini   pcbddc->recompute_topography = PETSC_TRUE;
4240c7d97c5SJed Brown   PetscFunctionReturn(0);
4250c7d97c5SJed Brown }
4260c7d97c5SJed Brown 
4270c7d97c5SJed Brown #undef __FUNCT__
42882ba6b80SStefano Zampini #define __FUNCT__ "PCBDDCSetDirichletBoundariesLocal"
4299c0446d6SStefano Zampini /*@
43082ba6b80SStefano Zampini  PCBDDCSetDirichletBoundariesLocal - Set IS defining Dirichlet boundaries for the global problem in local ordering.
4319c0446d6SStefano Zampini 
432785d1243SStefano Zampini    Collective
43353cdbc3dSStefano Zampini 
43453cdbc3dSStefano Zampini    Input Parameters:
43553cdbc3dSStefano Zampini +  pc - the preconditioning context
43682ba6b80SStefano Zampini -  DirichletBoundaries - parallel IS defining the Dirichlet boundaries (in local ordering)
43753cdbc3dSStefano Zampini 
43853cdbc3dSStefano Zampini    Level: intermediate
43953cdbc3dSStefano Zampini 
4409c0446d6SStefano Zampini    Notes:
44153cdbc3dSStefano Zampini 
4420f202f7eSStefano Zampini .seealso: PCBDDC, PCBDDCSetDirichletBoundaries(), MatZeroRows(), MatZeroRowsColumns()
44353cdbc3dSStefano Zampini @*/
44482ba6b80SStefano Zampini PetscErrorCode PCBDDCSetDirichletBoundariesLocal(PC pc,IS DirichletBoundaries)
4450c7d97c5SJed Brown {
4460c7d97c5SJed Brown   PetscErrorCode ierr;
4470c7d97c5SJed Brown 
4480c7d97c5SJed Brown   PetscFunctionBegin;
4490c7d97c5SJed Brown   PetscValidHeaderSpecific(pc,PC_CLASSID,1);
4500c7d97c5SJed Brown   PetscValidHeaderSpecific(DirichletBoundaries,IS_CLASSID,2);
45182ba6b80SStefano Zampini   PetscCheckSameComm(pc,1,DirichletBoundaries,2);
45282ba6b80SStefano Zampini   ierr = PetscTryMethod(pc,"PCBDDCSetDirichletBoundariesLocal_C",(PC,IS),(pc,DirichletBoundaries));CHKERRQ(ierr);
4530c7d97c5SJed Brown   PetscFunctionReturn(0);
4540c7d97c5SJed Brown }
4550c7d97c5SJed Brown /* -------------------------------------------------------------------------- */
4560c7d97c5SJed Brown 
4570c7d97c5SJed Brown #undef __FUNCT__
4580c7d97c5SJed Brown #define __FUNCT__ "PCBDDCSetNeumannBoundaries_BDDC"
45953cdbc3dSStefano Zampini static PetscErrorCode PCBDDCSetNeumannBoundaries_BDDC(PC pc,IS NeumannBoundaries)
4600c7d97c5SJed Brown {
4610c7d97c5SJed Brown   PC_BDDC  *pcbddc = (PC_BDDC*)pc->data;
46253cdbc3dSStefano Zampini   PetscErrorCode ierr;
4630c7d97c5SJed Brown 
4640c7d97c5SJed Brown   PetscFunctionBegin;
465785d1243SStefano Zampini   /* last user setting takes precendence -> destroy any other customization */
466785d1243SStefano Zampini   ierr = ISDestroy(&pcbddc->NeumannBoundariesLocal);CHKERRQ(ierr);
46753cdbc3dSStefano Zampini   ierr = ISDestroy(&pcbddc->NeumannBoundaries);CHKERRQ(ierr);
46836e030ebSStefano Zampini   ierr = PetscObjectReference((PetscObject)NeumannBoundaries);CHKERRQ(ierr);
46936e030ebSStefano Zampini   pcbddc->NeumannBoundaries = NeumannBoundaries;
470fb180af4SStefano Zampini   pcbddc->recompute_topography = PETSC_TRUE;
4710c7d97c5SJed Brown   PetscFunctionReturn(0);
4720c7d97c5SJed Brown }
4731e6b0712SBarry Smith 
4740c7d97c5SJed Brown #undef __FUNCT__
4750c7d97c5SJed Brown #define __FUNCT__ "PCBDDCSetNeumannBoundaries"
47657527edcSJed Brown /*@
47728509bceSStefano Zampini  PCBDDCSetNeumannBoundaries - Set IS defining Neumann boundaries for the global problem.
47857527edcSJed Brown 
479785d1243SStefano Zampini    Collective
48057527edcSJed Brown 
48157527edcSJed Brown    Input Parameters:
48257527edcSJed Brown +  pc - the preconditioning context
483785d1243SStefano Zampini -  NeumannBoundaries - parallel IS defining the Neumann boundaries
48457527edcSJed Brown 
48557527edcSJed Brown    Level: intermediate
48657527edcSJed Brown 
4870f202f7eSStefano Zampini    Notes:
4880f202f7eSStefano Zampini      Any process can list any global node
48957527edcSJed Brown 
4900f202f7eSStefano Zampini .seealso: PCBDDC, PCBDDCSetNeumannBoundariesLocal()
49157527edcSJed Brown @*/
49253cdbc3dSStefano Zampini PetscErrorCode PCBDDCSetNeumannBoundaries(PC pc,IS NeumannBoundaries)
4930c7d97c5SJed Brown {
4940c7d97c5SJed Brown   PetscErrorCode ierr;
4950c7d97c5SJed Brown 
4960c7d97c5SJed Brown   PetscFunctionBegin;
4970c7d97c5SJed Brown   PetscValidHeaderSpecific(pc,PC_CLASSID,1);
498674ae819SStefano Zampini   PetscValidHeaderSpecific(NeumannBoundaries,IS_CLASSID,2);
499785d1243SStefano Zampini   PetscCheckSameComm(pc,1,NeumannBoundaries,2);
50053cdbc3dSStefano Zampini   ierr = PetscTryMethod(pc,"PCBDDCSetNeumannBoundaries_C",(PC,IS),(pc,NeumannBoundaries));CHKERRQ(ierr);
50153cdbc3dSStefano Zampini   PetscFunctionReturn(0);
50253cdbc3dSStefano Zampini }
50353cdbc3dSStefano Zampini /* -------------------------------------------------------------------------- */
5041e6b0712SBarry Smith 
50553cdbc3dSStefano Zampini #undef __FUNCT__
50682ba6b80SStefano Zampini #define __FUNCT__ "PCBDDCSetNeumannBoundariesLocal_BDDC"
50782ba6b80SStefano Zampini static PetscErrorCode PCBDDCSetNeumannBoundariesLocal_BDDC(PC pc,IS NeumannBoundaries)
5080c7d97c5SJed Brown {
5090c7d97c5SJed Brown   PC_BDDC  *pcbddc = (PC_BDDC*)pc->data;
5100c7d97c5SJed Brown   PetscErrorCode ierr;
5110c7d97c5SJed Brown 
5120c7d97c5SJed Brown   PetscFunctionBegin;
513785d1243SStefano Zampini   /* last user setting takes precendence -> destroy any other customization */
514785d1243SStefano Zampini   ierr = ISDestroy(&pcbddc->NeumannBoundariesLocal);CHKERRQ(ierr);
5150c7d97c5SJed Brown   ierr = ISDestroy(&pcbddc->NeumannBoundaries);CHKERRQ(ierr);
5160c7d97c5SJed Brown   ierr = PetscObjectReference((PetscObject)NeumannBoundaries);CHKERRQ(ierr);
517785d1243SStefano Zampini   pcbddc->NeumannBoundariesLocal = NeumannBoundaries;
5187d24d155SStefano Zampini   pcbddc->recompute_topography = PETSC_TRUE;
5190c7d97c5SJed Brown   PetscFunctionReturn(0);
5200c7d97c5SJed Brown }
5210c7d97c5SJed Brown 
5220c7d97c5SJed Brown #undef __FUNCT__
52382ba6b80SStefano Zampini #define __FUNCT__ "PCBDDCSetNeumannBoundariesLocal"
5240c7d97c5SJed Brown /*@
52582ba6b80SStefano Zampini  PCBDDCSetNeumannBoundariesLocal - Set IS defining Neumann boundaries for the global problem in local ordering.
5260c7d97c5SJed Brown 
527785d1243SStefano Zampini    Collective
5280c7d97c5SJed Brown 
5290c7d97c5SJed Brown    Input Parameters:
5300c7d97c5SJed Brown +  pc - the preconditioning context
53182ba6b80SStefano Zampini -  NeumannBoundaries - parallel IS defining the subdomain part of Neumann boundaries (in local ordering)
5320c7d97c5SJed Brown 
5330c7d97c5SJed Brown    Level: intermediate
5340c7d97c5SJed Brown 
5350c7d97c5SJed Brown    Notes:
5360c7d97c5SJed Brown 
5370f202f7eSStefano Zampini .seealso: PCBDDC, PCBDDCSetNeumannBoundaries()
5380c7d97c5SJed Brown @*/
53982ba6b80SStefano Zampini PetscErrorCode PCBDDCSetNeumannBoundariesLocal(PC pc,IS NeumannBoundaries)
5400c7d97c5SJed Brown {
5410c7d97c5SJed Brown   PetscErrorCode ierr;
5420c7d97c5SJed Brown 
5430c7d97c5SJed Brown   PetscFunctionBegin;
5440c7d97c5SJed Brown   PetscValidHeaderSpecific(pc,PC_CLASSID,1);
5450c7d97c5SJed Brown   PetscValidHeaderSpecific(NeumannBoundaries,IS_CLASSID,2);
54682ba6b80SStefano Zampini   PetscCheckSameComm(pc,1,NeumannBoundaries,2);
54782ba6b80SStefano Zampini   ierr = PetscTryMethod(pc,"PCBDDCSetNeumannBoundariesLocal_C",(PC,IS),(pc,NeumannBoundaries));CHKERRQ(ierr);
54853cdbc3dSStefano Zampini   PetscFunctionReturn(0);
54953cdbc3dSStefano Zampini }
55053cdbc3dSStefano Zampini /* -------------------------------------------------------------------------- */
55153cdbc3dSStefano Zampini 
55253cdbc3dSStefano Zampini #undef __FUNCT__
553da1bb401SStefano Zampini #define __FUNCT__ "PCBDDCGetDirichletBoundaries_BDDC"
554da1bb401SStefano Zampini static PetscErrorCode PCBDDCGetDirichletBoundaries_BDDC(PC pc,IS *DirichletBoundaries)
555da1bb401SStefano Zampini {
556da1bb401SStefano Zampini   PC_BDDC  *pcbddc = (PC_BDDC*)pc->data;
557da1bb401SStefano Zampini 
558da1bb401SStefano Zampini   PetscFunctionBegin;
559da1bb401SStefano Zampini   *DirichletBoundaries = pcbddc->DirichletBoundaries;
560da1bb401SStefano Zampini   PetscFunctionReturn(0);
561da1bb401SStefano Zampini }
5621e6b0712SBarry Smith 
563da1bb401SStefano Zampini #undef __FUNCT__
564da1bb401SStefano Zampini #define __FUNCT__ "PCBDDCGetDirichletBoundaries"
565da1bb401SStefano Zampini /*@
566785d1243SStefano Zampini  PCBDDCGetDirichletBoundaries - Get parallel IS for Dirichlet boundaries
567da1bb401SStefano Zampini 
568785d1243SStefano Zampini    Collective
569785d1243SStefano Zampini 
570785d1243SStefano Zampini    Input Parameters:
571785d1243SStefano Zampini .  pc - the preconditioning context
572785d1243SStefano Zampini 
573785d1243SStefano Zampini    Output Parameters:
574785d1243SStefano Zampini .  DirichletBoundaries - index set defining the Dirichlet boundaries
575785d1243SStefano Zampini 
576785d1243SStefano Zampini    Level: intermediate
577785d1243SStefano Zampini 
5780f202f7eSStefano Zampini    Notes:
5790f202f7eSStefano Zampini      The IS returned (if any) is the same passed in earlier by the user with PCBDDCSetDirichletBoundaries
580785d1243SStefano Zampini 
581785d1243SStefano Zampini .seealso: PCBDDC
582785d1243SStefano Zampini @*/
583785d1243SStefano Zampini PetscErrorCode PCBDDCGetDirichletBoundaries(PC pc,IS *DirichletBoundaries)
584785d1243SStefano Zampini {
585785d1243SStefano Zampini   PetscErrorCode ierr;
586785d1243SStefano Zampini 
587785d1243SStefano Zampini   PetscFunctionBegin;
588785d1243SStefano Zampini   PetscValidHeaderSpecific(pc,PC_CLASSID,1);
589785d1243SStefano Zampini   ierr = PetscUseMethod(pc,"PCBDDCGetDirichletBoundaries_C",(PC,IS*),(pc,DirichletBoundaries));CHKERRQ(ierr);
590785d1243SStefano Zampini   PetscFunctionReturn(0);
591785d1243SStefano Zampini }
592785d1243SStefano Zampini /* -------------------------------------------------------------------------- */
593785d1243SStefano Zampini 
594785d1243SStefano Zampini #undef __FUNCT__
595785d1243SStefano Zampini #define __FUNCT__ "PCBDDCGetDirichletBoundariesLocal_BDDC"
596785d1243SStefano Zampini static PetscErrorCode PCBDDCGetDirichletBoundariesLocal_BDDC(PC pc,IS *DirichletBoundaries)
597785d1243SStefano Zampini {
598785d1243SStefano Zampini   PC_BDDC  *pcbddc = (PC_BDDC*)pc->data;
599785d1243SStefano Zampini 
600785d1243SStefano Zampini   PetscFunctionBegin;
601785d1243SStefano Zampini   *DirichletBoundaries = pcbddc->DirichletBoundariesLocal;
602785d1243SStefano Zampini   PetscFunctionReturn(0);
603785d1243SStefano Zampini }
604785d1243SStefano Zampini 
605785d1243SStefano Zampini #undef __FUNCT__
60682ba6b80SStefano Zampini #define __FUNCT__ "PCBDDCGetDirichletBoundariesLocal"
607da1bb401SStefano Zampini /*@
60882ba6b80SStefano Zampini  PCBDDCGetDirichletBoundariesLocal - Get parallel IS for Dirichlet boundaries (in local ordering)
609da1bb401SStefano Zampini 
610785d1243SStefano Zampini    Collective
611da1bb401SStefano Zampini 
612da1bb401SStefano Zampini    Input Parameters:
61328509bceSStefano Zampini .  pc - the preconditioning context
614da1bb401SStefano Zampini 
615da1bb401SStefano Zampini    Output Parameters:
61628509bceSStefano Zampini .  DirichletBoundaries - index set defining the subdomain part of Dirichlet boundaries
617da1bb401SStefano Zampini 
618da1bb401SStefano Zampini    Level: intermediate
619da1bb401SStefano Zampini 
620da1bb401SStefano Zampini    Notes:
6210f202f7eSStefano Zampini      The IS returned could be the same passed in earlier by the user (if provided with PCBDDCSetDirichletBoundariesLocal) or a global-to-local map of the global IS (if provided with PCBDDCSetDirichletBoundaries).
6220f202f7eSStefano Zampini           In the latter case, the IS will be available after PCSetUp.
623da1bb401SStefano Zampini 
624da1bb401SStefano Zampini .seealso: PCBDDC
625da1bb401SStefano Zampini @*/
62682ba6b80SStefano Zampini PetscErrorCode PCBDDCGetDirichletBoundariesLocal(PC pc,IS *DirichletBoundaries)
627da1bb401SStefano Zampini {
628da1bb401SStefano Zampini   PetscErrorCode ierr;
629da1bb401SStefano Zampini 
630da1bb401SStefano Zampini   PetscFunctionBegin;
631da1bb401SStefano Zampini   PetscValidHeaderSpecific(pc,PC_CLASSID,1);
63282ba6b80SStefano Zampini   ierr = PetscUseMethod(pc,"PCBDDCGetDirichletBoundariesLocal_C",(PC,IS*),(pc,DirichletBoundaries));CHKERRQ(ierr);
633da1bb401SStefano Zampini   PetscFunctionReturn(0);
634da1bb401SStefano Zampini }
635da1bb401SStefano Zampini /* -------------------------------------------------------------------------- */
6361e6b0712SBarry Smith 
637da1bb401SStefano Zampini #undef __FUNCT__
63853cdbc3dSStefano Zampini #define __FUNCT__ "PCBDDCGetNeumannBoundaries_BDDC"
63953cdbc3dSStefano Zampini static PetscErrorCode PCBDDCGetNeumannBoundaries_BDDC(PC pc,IS *NeumannBoundaries)
64053cdbc3dSStefano Zampini {
64153cdbc3dSStefano Zampini   PC_BDDC  *pcbddc = (PC_BDDC*)pc->data;
64253cdbc3dSStefano Zampini 
64353cdbc3dSStefano Zampini   PetscFunctionBegin;
64453cdbc3dSStefano Zampini   *NeumannBoundaries = pcbddc->NeumannBoundaries;
64553cdbc3dSStefano Zampini   PetscFunctionReturn(0);
64653cdbc3dSStefano Zampini }
6471e6b0712SBarry Smith 
64853cdbc3dSStefano Zampini #undef __FUNCT__
64953cdbc3dSStefano Zampini #define __FUNCT__ "PCBDDCGetNeumannBoundaries"
65053cdbc3dSStefano Zampini /*@
651785d1243SStefano Zampini  PCBDDCGetNeumannBoundaries - Get parallel IS for Neumann boundaries
65253cdbc3dSStefano Zampini 
653785d1243SStefano Zampini    Collective
654785d1243SStefano Zampini 
655785d1243SStefano Zampini    Input Parameters:
656785d1243SStefano Zampini .  pc - the preconditioning context
657785d1243SStefano Zampini 
658785d1243SStefano Zampini    Output Parameters:
659785d1243SStefano Zampini .  NeumannBoundaries - index set defining the Neumann boundaries
660785d1243SStefano Zampini 
661785d1243SStefano Zampini    Level: intermediate
662785d1243SStefano Zampini 
6630f202f7eSStefano Zampini    Notes:
6640f202f7eSStefano Zampini      The IS returned (if any) is the same passed in earlier by the user with PCBDDCSetNeumannBoundaries
665785d1243SStefano Zampini 
666785d1243SStefano Zampini .seealso: PCBDDC
667785d1243SStefano Zampini @*/
668785d1243SStefano Zampini PetscErrorCode PCBDDCGetNeumannBoundaries(PC pc,IS *NeumannBoundaries)
669785d1243SStefano Zampini {
670785d1243SStefano Zampini   PetscErrorCode ierr;
671785d1243SStefano Zampini 
672785d1243SStefano Zampini   PetscFunctionBegin;
673785d1243SStefano Zampini   PetscValidHeaderSpecific(pc,PC_CLASSID,1);
674785d1243SStefano Zampini   ierr = PetscUseMethod(pc,"PCBDDCGetNeumannBoundaries_C",(PC,IS*),(pc,NeumannBoundaries));CHKERRQ(ierr);
675785d1243SStefano Zampini   PetscFunctionReturn(0);
676785d1243SStefano Zampini }
677785d1243SStefano Zampini /* -------------------------------------------------------------------------- */
678785d1243SStefano Zampini 
679785d1243SStefano Zampini #undef __FUNCT__
680785d1243SStefano Zampini #define __FUNCT__ "PCBDDCGetNeumannBoundariesLocal_BDDC"
681785d1243SStefano Zampini static PetscErrorCode PCBDDCGetNeumannBoundariesLocal_BDDC(PC pc,IS *NeumannBoundaries)
682785d1243SStefano Zampini {
683785d1243SStefano Zampini   PC_BDDC  *pcbddc = (PC_BDDC*)pc->data;
684785d1243SStefano Zampini 
685785d1243SStefano Zampini   PetscFunctionBegin;
686785d1243SStefano Zampini   *NeumannBoundaries = pcbddc->NeumannBoundariesLocal;
687785d1243SStefano Zampini   PetscFunctionReturn(0);
688785d1243SStefano Zampini }
689785d1243SStefano Zampini 
690785d1243SStefano Zampini #undef __FUNCT__
69182ba6b80SStefano Zampini #define __FUNCT__ "PCBDDCGetNeumannBoundariesLocal"
69253cdbc3dSStefano Zampini /*@
69382ba6b80SStefano Zampini  PCBDDCGetNeumannBoundariesLocal - Get parallel IS for Neumann boundaries (in local ordering)
69453cdbc3dSStefano Zampini 
695785d1243SStefano Zampini    Collective
69653cdbc3dSStefano Zampini 
69753cdbc3dSStefano Zampini    Input Parameters:
69828509bceSStefano Zampini .  pc - the preconditioning context
69953cdbc3dSStefano Zampini 
70053cdbc3dSStefano Zampini    Output Parameters:
70128509bceSStefano Zampini .  NeumannBoundaries - index set defining the subdomain part of Neumann boundaries
70253cdbc3dSStefano Zampini 
70353cdbc3dSStefano Zampini    Level: intermediate
70453cdbc3dSStefano Zampini 
70553cdbc3dSStefano Zampini    Notes:
7060f202f7eSStefano Zampini      The IS returned could be the same passed in earlier by the user (if provided with PCBDDCSetNeumannBoundariesLocal) or a global-to-local map of the global IS (if provided with PCBDDCSetNeumannBoundaries).
7070f202f7eSStefano Zampini           In the latter case, the IS will be available after PCSetUp.
70853cdbc3dSStefano Zampini 
70953cdbc3dSStefano Zampini .seealso: PCBDDC
71053cdbc3dSStefano Zampini @*/
71182ba6b80SStefano Zampini PetscErrorCode PCBDDCGetNeumannBoundariesLocal(PC pc,IS *NeumannBoundaries)
71253cdbc3dSStefano Zampini {
71353cdbc3dSStefano Zampini   PetscErrorCode ierr;
71453cdbc3dSStefano Zampini 
71553cdbc3dSStefano Zampini   PetscFunctionBegin;
71653cdbc3dSStefano Zampini   PetscValidHeaderSpecific(pc,PC_CLASSID,1);
71782ba6b80SStefano Zampini   ierr = PetscUseMethod(pc,"PCBDDCGetNeumannBoundariesLocal_C",(PC,IS*),(pc,NeumannBoundaries));CHKERRQ(ierr);
7180c7d97c5SJed Brown   PetscFunctionReturn(0);
7190c7d97c5SJed Brown }
72036e030ebSStefano Zampini /* -------------------------------------------------------------------------- */
7211e6b0712SBarry Smith 
72236e030ebSStefano Zampini #undef __FUNCT__
723da1bb401SStefano Zampini #define __FUNCT__ "PCBDDCSetLocalAdjacencyGraph_BDDC"
7241a83f524SJed Brown static PetscErrorCode PCBDDCSetLocalAdjacencyGraph_BDDC(PC pc, PetscInt nvtxs,const PetscInt xadj[],const PetscInt adjncy[], PetscCopyMode copymode)
72536e030ebSStefano Zampini {
72636e030ebSStefano Zampini   PC_BDDC        *pcbddc = (PC_BDDC*)pc->data;
727da1bb401SStefano Zampini   PCBDDCGraph    mat_graph = pcbddc->mat_graph;
728da1bb401SStefano Zampini   PetscErrorCode ierr;
72936e030ebSStefano Zampini 
73036e030ebSStefano Zampini   PetscFunctionBegin;
731674ae819SStefano Zampini   /* free old CSR */
732674ae819SStefano Zampini   ierr = PCBDDCGraphResetCSR(mat_graph);CHKERRQ(ierr);
733fb223d50SStefano Zampini   /* TODO: PCBDDCGraphSetAdjacency */
734674ae819SStefano Zampini   /* get CSR into graph structure */
735da1bb401SStefano Zampini   if (copymode == PETSC_COPY_VALUES) {
736854ce69bSBarry Smith     ierr = PetscMalloc1(nvtxs+1,&mat_graph->xadj);CHKERRQ(ierr);
737785e854fSJed Brown     ierr = PetscMalloc1(xadj[nvtxs],&mat_graph->adjncy);CHKERRQ(ierr);
738674ae819SStefano Zampini     ierr = PetscMemcpy(mat_graph->xadj,xadj,(nvtxs+1)*sizeof(PetscInt));CHKERRQ(ierr);
739674ae819SStefano Zampini     ierr = PetscMemcpy(mat_graph->adjncy,adjncy,xadj[nvtxs]*sizeof(PetscInt));CHKERRQ(ierr);
740da1bb401SStefano Zampini   } else if (copymode == PETSC_OWN_POINTER) {
7411a83f524SJed Brown     mat_graph->xadj = (PetscInt*)xadj;
7421a83f524SJed Brown     mat_graph->adjncy = (PetscInt*)adjncy;
743674ae819SStefano Zampini   }
744575ad6abSStefano Zampini   mat_graph->nvtxs_csr = nvtxs;
74536e030ebSStefano Zampini   PetscFunctionReturn(0);
74636e030ebSStefano Zampini }
7471e6b0712SBarry Smith 
74836e030ebSStefano Zampini #undef __FUNCT__
749da1bb401SStefano Zampini #define __FUNCT__ "PCBDDCSetLocalAdjacencyGraph"
75036e030ebSStefano Zampini /*@
7510f202f7eSStefano Zampini  PCBDDCSetLocalAdjacencyGraph - Set adjacency structure (CSR graph) of the local matrix
75236e030ebSStefano Zampini 
75336e030ebSStefano Zampini    Not collective
75436e030ebSStefano Zampini 
75536e030ebSStefano Zampini    Input Parameters:
75636e030ebSStefano Zampini +  pc - the preconditioning context
7570f202f7eSStefano Zampini .  nvtxs - number of local vertices of the graph (i.e., the size of the local problem)
75828509bceSStefano Zampini .  xadj, adjncy - the CSR graph
75928509bceSStefano Zampini -  copymode - either PETSC_COPY_VALUES or PETSC_OWN_POINTER.
76036e030ebSStefano Zampini 
76136e030ebSStefano Zampini    Level: intermediate
76236e030ebSStefano Zampini 
76336e030ebSStefano Zampini    Notes:
76436e030ebSStefano Zampini 
76528509bceSStefano Zampini .seealso: PCBDDC,PetscCopyMode
76636e030ebSStefano Zampini @*/
7671a83f524SJed Brown PetscErrorCode PCBDDCSetLocalAdjacencyGraph(PC pc,PetscInt nvtxs,const PetscInt xadj[],const PetscInt adjncy[], PetscCopyMode copymode)
76836e030ebSStefano Zampini {
769575ad6abSStefano Zampini   void (*f)(void) = 0;
77036e030ebSStefano Zampini   PetscErrorCode ierr;
77136e030ebSStefano Zampini 
77236e030ebSStefano Zampini   PetscFunctionBegin;
77336e030ebSStefano Zampini   PetscValidHeaderSpecific(pc,PC_CLASSID,1);
774674ae819SStefano Zampini   PetscValidIntPointer(xadj,3);
775d7de1dd9SStefano Zampini   PetscValidIntPointer(adjncy,4);
776674ae819SStefano Zampini   if (copymode != PETSC_COPY_VALUES && copymode != PETSC_OWN_POINTER) {
777674ae819SStefano Zampini     SETERRQ2(PETSC_COMM_SELF,PETSC_ERR_SUP,"Unsupported copy mode %d in %s\n",copymode,__FUNCT__);
778da1bb401SStefano Zampini   }
77936e030ebSStefano Zampini   ierr = PetscTryMethod(pc,"PCBDDCSetLocalAdjacencyGraph_C",(PC,PetscInt,const PetscInt[],const PetscInt[],PetscCopyMode),(pc,nvtxs,xadj,adjncy,copymode));CHKERRQ(ierr);
780575ad6abSStefano Zampini   /* free arrays if PCBDDC is not the PC type */
781575ad6abSStefano Zampini   ierr = PetscObjectQueryFunction((PetscObject)pc,"PCBDDCSetLocalAdjacencyGraph_C",&f);CHKERRQ(ierr);
782575ad6abSStefano Zampini   if (!f && copymode == PETSC_OWN_POINTER) {
783575ad6abSStefano Zampini     ierr = PetscFree(xadj);CHKERRQ(ierr);
784575ad6abSStefano Zampini     ierr = PetscFree(adjncy);CHKERRQ(ierr);
78536e030ebSStefano Zampini   }
78636e030ebSStefano Zampini   PetscFunctionReturn(0);
78736e030ebSStefano Zampini }
7889c0446d6SStefano Zampini /* -------------------------------------------------------------------------- */
7891e6b0712SBarry Smith 
7909c0446d6SStefano Zampini #undef __FUNCT__
79163602bcaSStefano Zampini #define __FUNCT__ "PCBDDCSetDofsSplittingLocal_BDDC"
79263602bcaSStefano Zampini static PetscErrorCode PCBDDCSetDofsSplittingLocal_BDDC(PC pc,PetscInt n_is, IS ISForDofs[])
79363602bcaSStefano Zampini {
79463602bcaSStefano Zampini   PC_BDDC  *pcbddc = (PC_BDDC*)pc->data;
79563602bcaSStefano Zampini   PetscInt i;
79663602bcaSStefano Zampini   PetscErrorCode ierr;
79763602bcaSStefano Zampini 
79863602bcaSStefano Zampini   PetscFunctionBegin;
79963602bcaSStefano Zampini   /* Destroy ISes if they were already set */
80063602bcaSStefano Zampini   for (i=0;i<pcbddc->n_ISForDofsLocal;i++) {
80163602bcaSStefano Zampini     ierr = ISDestroy(&pcbddc->ISForDofsLocal[i]);CHKERRQ(ierr);
80263602bcaSStefano Zampini   }
80363602bcaSStefano Zampini   ierr = PetscFree(pcbddc->ISForDofsLocal);CHKERRQ(ierr);
80463602bcaSStefano Zampini   /* last user setting takes precendence -> destroy any other customization */
80563602bcaSStefano Zampini   for (i=0;i<pcbddc->n_ISForDofs;i++) {
80663602bcaSStefano Zampini     ierr = ISDestroy(&pcbddc->ISForDofs[i]);CHKERRQ(ierr);
80763602bcaSStefano Zampini   }
80863602bcaSStefano Zampini   ierr = PetscFree(pcbddc->ISForDofs);CHKERRQ(ierr);
80963602bcaSStefano Zampini   pcbddc->n_ISForDofs = 0;
81063602bcaSStefano Zampini   /* allocate space then set */
811d02579f5SStefano Zampini   if (n_is) {
812d02579f5SStefano Zampini     ierr = PetscMalloc1(n_is,&pcbddc->ISForDofsLocal);CHKERRQ(ierr);
813d02579f5SStefano Zampini   }
81463602bcaSStefano Zampini   for (i=0;i<n_is;i++) {
81563602bcaSStefano Zampini     ierr = PetscObjectReference((PetscObject)ISForDofs[i]);CHKERRQ(ierr);
81663602bcaSStefano Zampini     pcbddc->ISForDofsLocal[i]=ISForDofs[i];
81763602bcaSStefano Zampini   }
81863602bcaSStefano Zampini   pcbddc->n_ISForDofsLocal=n_is;
81963602bcaSStefano Zampini   if (n_is) pcbddc->user_provided_isfordofs = PETSC_TRUE;
8201a8a16d1SStefano Zampini   pcbddc->recompute_topography = PETSC_TRUE;
82163602bcaSStefano Zampini   PetscFunctionReturn(0);
82263602bcaSStefano Zampini }
82363602bcaSStefano Zampini 
82463602bcaSStefano Zampini #undef __FUNCT__
82563602bcaSStefano Zampini #define __FUNCT__ "PCBDDCSetDofsSplittingLocal"
82663602bcaSStefano Zampini /*@
82763602bcaSStefano Zampini  PCBDDCSetDofsSplittingLocal - Set index sets defining fields of the local subdomain matrix
82863602bcaSStefano Zampini 
82963602bcaSStefano Zampini    Collective
83063602bcaSStefano Zampini 
83163602bcaSStefano Zampini    Input Parameters:
83263602bcaSStefano Zampini +  pc - the preconditioning context
8330f202f7eSStefano Zampini .  n_is - number of index sets defining the fields
8340f202f7eSStefano Zampini -  ISForDofs - array of IS describing the fields in local ordering
83563602bcaSStefano Zampini 
83663602bcaSStefano Zampini    Level: intermediate
83763602bcaSStefano Zampini 
8380f202f7eSStefano Zampini    Notes:
8390f202f7eSStefano Zampini      n_is should be the same among processes. Not all nodes need to be listed: unlisted nodes will belong to the complement field.
84063602bcaSStefano Zampini 
84163602bcaSStefano Zampini .seealso: PCBDDC
84263602bcaSStefano Zampini @*/
84363602bcaSStefano Zampini PetscErrorCode PCBDDCSetDofsSplittingLocal(PC pc,PetscInt n_is, IS ISForDofs[])
84463602bcaSStefano Zampini {
84563602bcaSStefano Zampini   PetscInt       i;
84663602bcaSStefano Zampini   PetscErrorCode ierr;
84763602bcaSStefano Zampini 
84863602bcaSStefano Zampini   PetscFunctionBegin;
84963602bcaSStefano Zampini   PetscValidHeaderSpecific(pc,PC_CLASSID,1);
85063602bcaSStefano Zampini   PetscValidLogicalCollectiveInt(pc,n_is,2);
85163602bcaSStefano Zampini   for (i=0;i<n_is;i++) {
85263602bcaSStefano Zampini     PetscCheckSameComm(pc,1,ISForDofs[i],3);
85363602bcaSStefano Zampini     PetscValidHeaderSpecific(ISForDofs[i],IS_CLASSID,3);
85463602bcaSStefano Zampini   }
855e71e7a71SStefano Zampini   ierr = PetscTryMethod(pc,"PCBDDCSetDofsSplittingLocal_C",(PC,PetscInt,IS[]),(pc,n_is,ISForDofs));CHKERRQ(ierr);
85663602bcaSStefano Zampini   PetscFunctionReturn(0);
85763602bcaSStefano Zampini }
85863602bcaSStefano Zampini /* -------------------------------------------------------------------------- */
85963602bcaSStefano Zampini 
86063602bcaSStefano Zampini #undef __FUNCT__
8619c0446d6SStefano Zampini #define __FUNCT__ "PCBDDCSetDofsSplitting_BDDC"
8629c0446d6SStefano Zampini static PetscErrorCode PCBDDCSetDofsSplitting_BDDC(PC pc,PetscInt n_is, IS ISForDofs[])
8639c0446d6SStefano Zampini {
8649c0446d6SStefano Zampini   PC_BDDC  *pcbddc = (PC_BDDC*)pc->data;
8659c0446d6SStefano Zampini   PetscInt i;
8669c0446d6SStefano Zampini   PetscErrorCode ierr;
8679c0446d6SStefano Zampini 
8689c0446d6SStefano Zampini   PetscFunctionBegin;
869da1bb401SStefano Zampini   /* Destroy ISes if they were already set */
8709c0446d6SStefano Zampini   for (i=0;i<pcbddc->n_ISForDofs;i++) {
8719c0446d6SStefano Zampini     ierr = ISDestroy(&pcbddc->ISForDofs[i]);CHKERRQ(ierr);
8729c0446d6SStefano Zampini   }
873d11ae9bbSstefano_zampini   ierr = PetscFree(pcbddc->ISForDofs);CHKERRQ(ierr);
87463602bcaSStefano Zampini   /* last user setting takes precendence -> destroy any other customization */
87563602bcaSStefano Zampini   for (i=0;i<pcbddc->n_ISForDofsLocal;i++) {
87663602bcaSStefano Zampini     ierr = ISDestroy(&pcbddc->ISForDofsLocal[i]);CHKERRQ(ierr);
87763602bcaSStefano Zampini   }
87863602bcaSStefano Zampini   ierr = PetscFree(pcbddc->ISForDofsLocal);CHKERRQ(ierr);
87963602bcaSStefano Zampini   pcbddc->n_ISForDofsLocal = 0;
880da1bb401SStefano Zampini   /* allocate space then set */
881d02579f5SStefano Zampini   if (n_is) {
882785e854fSJed Brown     ierr = PetscMalloc1(n_is,&pcbddc->ISForDofs);CHKERRQ(ierr);
883d02579f5SStefano Zampini   }
8849c0446d6SStefano Zampini   for (i=0;i<n_is;i++) {
885da1bb401SStefano Zampini     ierr = PetscObjectReference((PetscObject)ISForDofs[i]);CHKERRQ(ierr);
886da1bb401SStefano Zampini     pcbddc->ISForDofs[i]=ISForDofs[i];
8879c0446d6SStefano Zampini   }
8889c0446d6SStefano Zampini   pcbddc->n_ISForDofs=n_is;
88963602bcaSStefano Zampini   if (n_is) pcbddc->user_provided_isfordofs = PETSC_TRUE;
8901a8a16d1SStefano Zampini   pcbddc->recompute_topography = PETSC_TRUE;
8919c0446d6SStefano Zampini   PetscFunctionReturn(0);
8929c0446d6SStefano Zampini }
8931e6b0712SBarry Smith 
8949c0446d6SStefano Zampini #undef __FUNCT__
8959c0446d6SStefano Zampini #define __FUNCT__ "PCBDDCSetDofsSplitting"
8969c0446d6SStefano Zampini /*@
89763602bcaSStefano Zampini  PCBDDCSetDofsSplitting - Set index sets defining fields of the global matrix
8989c0446d6SStefano Zampini 
89963602bcaSStefano Zampini    Collective
9009c0446d6SStefano Zampini 
9019c0446d6SStefano Zampini    Input Parameters:
9029c0446d6SStefano Zampini +  pc - the preconditioning context
9030f202f7eSStefano Zampini .  n_is - number of index sets defining the fields
9040f202f7eSStefano Zampini -  ISForDofs - array of IS describing the fields in global ordering
9059c0446d6SStefano Zampini 
9069c0446d6SStefano Zampini    Level: intermediate
9079c0446d6SStefano Zampini 
9080f202f7eSStefano Zampini    Notes:
9090f202f7eSStefano Zampini      Any process can list any global node. Not all nodes need to be listed: unlisted nodes will belong to the complement field.
9109c0446d6SStefano Zampini 
9119c0446d6SStefano Zampini .seealso: PCBDDC
9129c0446d6SStefano Zampini @*/
9139c0446d6SStefano Zampini PetscErrorCode PCBDDCSetDofsSplitting(PC pc,PetscInt n_is, IS ISForDofs[])
9149c0446d6SStefano Zampini {
9152b510759SStefano Zampini   PetscInt       i;
9169c0446d6SStefano Zampini   PetscErrorCode ierr;
9179c0446d6SStefano Zampini 
9189c0446d6SStefano Zampini   PetscFunctionBegin;
9199c0446d6SStefano Zampini   PetscValidHeaderSpecific(pc,PC_CLASSID,1);
92063602bcaSStefano Zampini   PetscValidLogicalCollectiveInt(pc,n_is,2);
9212b510759SStefano Zampini   for (i=0;i<n_is;i++) {
92263602bcaSStefano Zampini     PetscCheckSameComm(pc,1,ISForDofs[i],3);
92363602bcaSStefano Zampini     PetscValidHeaderSpecific(ISForDofs[i],IS_CLASSID,3);
9242b510759SStefano Zampini   }
9259c0446d6SStefano Zampini   ierr = PetscTryMethod(pc,"PCBDDCSetDofsSplitting_C",(PC,PetscInt,IS[]),(pc,n_is,ISForDofs));CHKERRQ(ierr);
9269c0446d6SStefano Zampini   PetscFunctionReturn(0);
9279c0446d6SStefano Zampini }
928906d46d4SStefano Zampini 
929da1bb401SStefano Zampini /* -------------------------------------------------------------------------- */
930534831adSStefano Zampini #undef __FUNCT__
931534831adSStefano Zampini #define __FUNCT__ "PCPreSolve_BDDC"
932534831adSStefano Zampini /* -------------------------------------------------------------------------- */
933534831adSStefano Zampini /*
934534831adSStefano Zampini    PCPreSolve_BDDC - Changes the right hand side and (if necessary) the initial
935534831adSStefano Zampini                      guess if a transformation of basis approach has been selected.
9369c0446d6SStefano Zampini 
937534831adSStefano Zampini    Input Parameter:
938534831adSStefano Zampini +  pc - the preconditioner contex
939534831adSStefano Zampini 
940534831adSStefano Zampini    Application Interface Routine: PCPreSolve()
941534831adSStefano Zampini 
942534831adSStefano Zampini    Notes:
943534831adSStefano Zampini      The interface routine PCPreSolve() is not usually called directly by
944534831adSStefano Zampini    the user, but instead is called by KSPSolve().
945534831adSStefano Zampini */
946534831adSStefano Zampini static PetscErrorCode PCPreSolve_BDDC(PC pc, KSP ksp, Vec rhs, Vec x)
947534831adSStefano Zampini {
948534831adSStefano Zampini   PetscErrorCode ierr;
949534831adSStefano Zampini   PC_BDDC        *pcbddc = (PC_BDDC*)pc->data;
950534831adSStefano Zampini   PC_IS          *pcis = (PC_IS*)(pc->data);
9513972b0daSStefano Zampini   Vec            used_vec;
9528d00608fSStefano Zampini   PetscBool      copy_rhs = PETSC_TRUE;
953534831adSStefano Zampini 
954534831adSStefano Zampini   PetscFunctionBegin;
95585c4d303SStefano Zampini   /* if we are working with cg, one dirichlet solve can be avoided during Krylov iterations */
95685c4d303SStefano Zampini   if (ksp) {
95785c4d303SStefano Zampini     PetscBool iscg;
95885c4d303SStefano Zampini     ierr = PetscObjectTypeCompare((PetscObject)ksp,KSPCG,&iscg);CHKERRQ(ierr);
95985c4d303SStefano Zampini     if (!iscg) {
96085c4d303SStefano Zampini       ierr = PCBDDCSetUseExactDirichlet(pc,PETSC_FALSE);CHKERRQ(ierr);
96185c4d303SStefano Zampini     }
96285c4d303SStefano Zampini   }
96385c4d303SStefano Zampini   /* Creates parallel work vectors used in presolve */
96462a6ff1dSStefano Zampini   if (!pcbddc->original_rhs) {
96562a6ff1dSStefano Zampini     ierr = VecDuplicate(pcis->vec1_global,&pcbddc->original_rhs);CHKERRQ(ierr);
96662a6ff1dSStefano Zampini   }
96762a6ff1dSStefano Zampini   if (!pcbddc->temp_solution) {
96862a6ff1dSStefano Zampini     ierr = VecDuplicate(pcis->vec1_global,&pcbddc->temp_solution);CHKERRQ(ierr);
96962a6ff1dSStefano Zampini   }
9708d00608fSStefano Zampini 
9713972b0daSStefano Zampini   if (x) {
9723972b0daSStefano Zampini     ierr = PetscObjectReference((PetscObject)x);CHKERRQ(ierr);
9733972b0daSStefano Zampini     used_vec = x;
9748d00608fSStefano Zampini   } else { /* it can only happen when calling PCBDDCMatFETIDPGetRHS */
9753972b0daSStefano Zampini     ierr = PetscObjectReference((PetscObject)pcbddc->temp_solution);CHKERRQ(ierr);
9763972b0daSStefano Zampini     used_vec = pcbddc->temp_solution;
9773972b0daSStefano Zampini     ierr = VecSet(used_vec,0.0);CHKERRQ(ierr);
9783972b0daSStefano Zampini   }
9798efcfb23SStefano Zampini 
9808efcfb23SStefano Zampini   /* hack into ksp data structure since PCPreSolve comes earlier than setting to zero the guess in src/ksp/ksp/interface/itfunc.c */
9813972b0daSStefano Zampini   if (ksp) {
982a0cb1b98SStefano Zampini     /* store the flag for the initial guess since it will be restored back during PCPostSolve_BDDC */
9838efcfb23SStefano Zampini     ierr = KSPGetInitialGuessNonzero(ksp,&pcbddc->ksp_guess_nonzero);CHKERRQ(ierr);
9848efcfb23SStefano Zampini     if (!pcbddc->ksp_guess_nonzero) {
9853972b0daSStefano Zampini       ierr = VecSet(used_vec,0.0);CHKERRQ(ierr);
9863972b0daSStefano Zampini     }
9873972b0daSStefano Zampini   }
9883308cffdSStefano Zampini 
9898d00608fSStefano Zampini   pcbddc->rhs_change = PETSC_FALSE;
9903972b0daSStefano Zampini   /* Take into account zeroed rows -> change rhs and store solution removed */
991a07ea27aSStefano Zampini   if (rhs) {
9923975b054SStefano Zampini     IS dirIS = NULL;
9933975b054SStefano Zampini 
994a07ea27aSStefano Zampini     /* DirichletBoundariesLocal may not be consistent among neighbours; gets a dirichlet dofs IS from graph (may be cached) */
9953975b054SStefano Zampini     ierr = PCBDDCGraphGetDirichletDofs(pcbddc->mat_graph,&dirIS);CHKERRQ(ierr);
9963975b054SStefano Zampini     if (dirIS) {
997906d46d4SStefano Zampini       Mat_IS            *matis = (Mat_IS*)pc->pmat->data;
998785d1243SStefano Zampini       PetscInt          dirsize,i,*is_indices;
9992b095fd8SStefano Zampini       PetscScalar       *array_x;
10002b095fd8SStefano Zampini       const PetscScalar *array_diagonal;
1001785d1243SStefano Zampini 
10023972b0daSStefano Zampini       ierr = MatGetDiagonal(pc->pmat,pcis->vec1_global);CHKERRQ(ierr);
10033972b0daSStefano Zampini       ierr = VecPointwiseDivide(pcis->vec1_global,rhs,pcis->vec1_global);CHKERRQ(ierr);
1004e176bc59SStefano Zampini       ierr = VecScatterBegin(matis->rctx,pcis->vec1_global,pcis->vec2_N,INSERT_VALUES,SCATTER_FORWARD);CHKERRQ(ierr);
1005e176bc59SStefano Zampini       ierr = VecScatterEnd(matis->rctx,pcis->vec1_global,pcis->vec2_N,INSERT_VALUES,SCATTER_FORWARD);CHKERRQ(ierr);
1006e176bc59SStefano Zampini       ierr = VecScatterBegin(matis->rctx,used_vec,pcis->vec1_N,INSERT_VALUES,SCATTER_FORWARD);CHKERRQ(ierr);
1007e176bc59SStefano Zampini       ierr = VecScatterEnd(matis->rctx,used_vec,pcis->vec1_N,INSERT_VALUES,SCATTER_FORWARD);CHKERRQ(ierr);
100882ba6b80SStefano Zampini       ierr = ISGetLocalSize(dirIS,&dirsize);CHKERRQ(ierr);
10093972b0daSStefano Zampini       ierr = VecGetArray(pcis->vec1_N,&array_x);CHKERRQ(ierr);
10102b095fd8SStefano Zampini       ierr = VecGetArrayRead(pcis->vec2_N,&array_diagonal);CHKERRQ(ierr);
10113972b0daSStefano Zampini       ierr = ISGetIndices(dirIS,(const PetscInt**)&is_indices);CHKERRQ(ierr);
10122fa5cd67SKarl Rupp       for (i=0; i<dirsize; i++) array_x[is_indices[i]] = array_diagonal[is_indices[i]];
10133972b0daSStefano Zampini       ierr = ISRestoreIndices(dirIS,(const PetscInt**)&is_indices);CHKERRQ(ierr);
10142b095fd8SStefano Zampini       ierr = VecRestoreArrayRead(pcis->vec2_N,&array_diagonal);CHKERRQ(ierr);
10153972b0daSStefano Zampini       ierr = VecRestoreArray(pcis->vec1_N,&array_x);CHKERRQ(ierr);
1016e176bc59SStefano Zampini       ierr = VecScatterBegin(matis->rctx,pcis->vec1_N,used_vec,INSERT_VALUES,SCATTER_REVERSE);CHKERRQ(ierr);
1017e176bc59SStefano Zampini       ierr = VecScatterEnd(matis->rctx,pcis->vec1_N,used_vec,INSERT_VALUES,SCATTER_REVERSE);CHKERRQ(ierr);
10188d00608fSStefano Zampini       pcbddc->rhs_change = PETSC_TRUE;
10191b968477SStefano Zampini       ierr = ISDestroy(&dirIS);CHKERRQ(ierr);
10208efcfb23SStefano Zampini     }
1021a07ea27aSStefano Zampini   }
1022b76ba322SStefano Zampini 
10238efcfb23SStefano Zampini   /* remove the computed solution or the initial guess from the rhs */
10248d00608fSStefano Zampini   if (pcbddc->rhs_change || (ksp && pcbddc->ksp_guess_nonzero) ) {
10258d00608fSStefano Zampini     /* store the original rhs */
10268d00608fSStefano Zampini     if (copy_rhs) {
10278d00608fSStefano Zampini       ierr = VecCopy(rhs,pcbddc->original_rhs);CHKERRQ(ierr);
10288d00608fSStefano Zampini       copy_rhs = PETSC_FALSE;
10298d00608fSStefano Zampini     }
10308d00608fSStefano Zampini     pcbddc->rhs_change = PETSC_TRUE;
10313972b0daSStefano Zampini     ierr = VecScale(used_vec,-1.0);CHKERRQ(ierr);
10320369aaf7SStefano Zampini     ierr = MatMultAdd(pc->mat,used_vec,rhs,rhs);CHKERRQ(ierr);
10333972b0daSStefano Zampini     ierr = VecScale(used_vec,-1.0);CHKERRQ(ierr);
10348efcfb23SStefano Zampini     ierr = VecCopy(used_vec,pcbddc->temp_solution);CHKERRQ(ierr);
10357acc28cbSStefano Zampini     if (ksp) {
10367acc28cbSStefano Zampini       ierr = KSPSetInitialGuessNonzero(ksp,PETSC_FALSE);CHKERRQ(ierr);
10377acc28cbSStefano Zampini     }
10383308cffdSStefano Zampini   }
10398efcfb23SStefano Zampini   ierr = VecDestroy(&used_vec);CHKERRQ(ierr);
1040b76ba322SStefano Zampini 
10410369aaf7SStefano Zampini   /* When using the benign trick: (TODO: what about FETI-DP?)
10420369aaf7SStefano Zampini      - change rhs on pressures (iteration matrix is surely of type MATIS)
10430369aaf7SStefano Zampini      - compute initial vector in benign space
10440369aaf7SStefano Zampini   */
10450369aaf7SStefano Zampini   if (rhs && pcbddc->benign_saddle_point) {
104606f24817SStefano Zampini     Mat_IS *matis = (Mat_IS*)(pc->mat->data);
104706f24817SStefano Zampini 
104806f24817SStefano Zampini     /* store the original rhs */
104906f24817SStefano Zampini     if (copy_rhs) {
105006f24817SStefano Zampini       ierr = VecCopy(rhs,pcbddc->original_rhs);CHKERRQ(ierr);
105106f24817SStefano Zampini       copy_rhs = PETSC_FALSE;
105206f24817SStefano Zampini     }
105306f24817SStefano Zampini 
105406f24817SStefano Zampini     /* now change (locally) the basis */
105506f24817SStefano Zampini     ierr = VecScatterBegin(matis->rctx,rhs,pcis->vec1_N,INSERT_VALUES,SCATTER_FORWARD);CHKERRQ(ierr);
105606f24817SStefano Zampini     ierr = VecScatterEnd(matis->rctx,rhs,pcis->vec1_N,INSERT_VALUES,SCATTER_FORWARD);CHKERRQ(ierr);
105706f24817SStefano Zampini     if (pcbddc->benign_change) {
105806f24817SStefano Zampini       ierr = MatMultTranspose(pcbddc->benign_change,pcis->vec1_N,pcis->vec2_N);CHKERRQ(ierr);
105906f24817SStefano Zampini       ierr = VecScatterBegin(matis->rctx,pcis->vec2_N,rhs,INSERT_VALUES,SCATTER_REVERSE);CHKERRQ(ierr);
106006f24817SStefano Zampini       ierr = VecScatterEnd(matis->rctx,pcis->vec2_N,rhs,INSERT_VALUES,SCATTER_REVERSE);CHKERRQ(ierr);
106106f24817SStefano Zampini       /* change local iteration matrix */
106206f24817SStefano Zampini       ierr = MatPtAP(matis->A,pcbddc->benign_change,MAT_REUSE_MATRIX,2.0,&pcbddc->local_mat);CHKERRQ(ierr);
106306f24817SStefano Zampini       ierr = MatDestroy(&pcbddc->benign_original_mat);CHKERRQ(ierr);
106406f24817SStefano Zampini       ierr = PetscObjectReference((PetscObject)matis->A);CHKERRQ(ierr);
106506f24817SStefano Zampini       pcbddc->benign_original_mat = matis->A;
106606f24817SStefano Zampini       ierr = MatDestroy(&matis->A);CHKERRQ(ierr);
106706f24817SStefano Zampini       ierr = PetscObjectReference((PetscObject)pcbddc->local_mat);CHKERRQ(ierr);
106806f24817SStefano Zampini       matis->A = pcbddc->local_mat;
106906f24817SStefano Zampini       ierr = MatDestroy(&pcbddc->local_mat);CHKERRQ(ierr);
107006f24817SStefano Zampini     } else {
107106f24817SStefano Zampini       ierr = VecScatterBegin(matis->rctx,pcis->vec1_N,rhs,INSERT_VALUES,SCATTER_REVERSE);CHKERRQ(ierr);
107206f24817SStefano Zampini       ierr = VecScatterEnd(matis->rctx,pcis->vec1_N,rhs,INSERT_VALUES,SCATTER_REVERSE);CHKERRQ(ierr);
107306f24817SStefano Zampini     }
107406f24817SStefano Zampini     pcbddc->rhs_change = PETSC_TRUE;
107506f24817SStefano Zampini 
10760369aaf7SStefano Zampini     /* compute u^*_h as in Xuemin Tu's thesis (see Section 4.8.1) */
10770369aaf7SStefano Zampini     /* TODO: what about Stokes? */
10780369aaf7SStefano Zampini     if (!pcbddc->benign_vec) {
10790369aaf7SStefano Zampini       ierr = VecDuplicate(rhs,&pcbddc->benign_vec);CHKERRQ(ierr);
10800369aaf7SStefano Zampini     }
10810369aaf7SStefano Zampini     ierr = VecSet(pcis->vec1_global,0.);CHKERRQ(ierr);
10820369aaf7SStefano Zampini     pcbddc->benign_p0 = 0.;
10830369aaf7SStefano Zampini     if (pcbddc->benign_p0_lidx >= 0) {
10840369aaf7SStefano Zampini       const PetscScalar *array;
10850369aaf7SStefano Zampini 
10860369aaf7SStefano Zampini       ierr = VecGetArrayRead(pcis->vec2_N,&array);CHKERRQ(ierr);
10870369aaf7SStefano Zampini       pcbddc->benign_p0 = array[pcbddc->benign_p0_lidx];
10880369aaf7SStefano Zampini       ierr = VecRestoreArrayRead(pcis->vec2_N,&array);CHKERRQ(ierr);
10890369aaf7SStefano Zampini     }
10908666afb4SStefano Zampini #if 0
10910369aaf7SStefano Zampini     ierr = PetscPrintf(PETSC_COMM_SELF,"[%d] benign u* should be %1.4e\n",PetscGlobalRank,pcbddc->benign_p0);CHKERRQ(ierr);
10928666afb4SStefano Zampini #endif
10930369aaf7SStefano Zampini     ierr = PCBDDCBenignPopOrPushP0(pc,pcis->vec1_global,PETSC_FALSE);CHKERRQ(ierr);
10940369aaf7SStefano Zampini     ierr = PCApply_BDDC(pc,pcis->vec1_global,pcbddc->benign_vec);CHKERRQ(ierr);
10950369aaf7SStefano Zampini     pcbddc->benign_p0 = 0.;
10960369aaf7SStefano Zampini     ierr = PCBDDCBenignPopOrPushP0(pc,pcbddc->benign_vec,PETSC_FALSE);CHKERRQ(ierr);
10970369aaf7SStefano Zampini     if (pcbddc->benign_saddle_point) {
10980369aaf7SStefano Zampini       Mat_IS* matis = (Mat_IS*)(pc->mat->data);
10990369aaf7SStefano Zampini       ierr = VecScatterBegin(matis->rctx,pcbddc->benign_vec,matis->x,INSERT_VALUES,SCATTER_FORWARD);CHKERRQ(ierr);
11000369aaf7SStefano Zampini       ierr = VecScatterEnd(matis->rctx,pcbddc->benign_vec,matis->x,INSERT_VALUES,SCATTER_FORWARD);CHKERRQ(ierr);
11010369aaf7SStefano Zampini 
11028666afb4SStefano Zampini #if 0
11030369aaf7SStefano Zampini       if (pcbddc->benign_p0_lidx >=0) {
11040369aaf7SStefano Zampini         Mat B0;
11050369aaf7SStefano Zampini         Vec dummy_vec;
11060369aaf7SStefano Zampini         PetscScalar *array;
11070369aaf7SStefano Zampini         PetscInt ii[2];
11080369aaf7SStefano Zampini 
11090369aaf7SStefano Zampini         ii[0] = 0;
11100369aaf7SStefano Zampini         ii[1] = pcbddc->B0_ncol;
11110369aaf7SStefano Zampini         ierr = MatCreateSeqAIJWithArrays(PETSC_COMM_SELF,1,pcis->n,ii,pcbddc->B0_cols,pcbddc->B0_vals,&B0);CHKERRQ(ierr);
11120369aaf7SStefano Zampini         ierr = MatCreateVecs(B0,NULL,&dummy_vec);CHKERRQ(ierr);
11130369aaf7SStefano Zampini         ierr = MatMult(B0,matis->x,dummy_vec);CHKERRQ(ierr);
11140369aaf7SStefano Zampini         ierr = VecGetArray(dummy_vec,&array);CHKERRQ(ierr);
11150369aaf7SStefano Zampini         ierr = PetscPrintf(PETSC_COMM_SELF,"[%d] benign u* is %1.4e\n",PetscGlobalRank,array[0]);CHKERRQ(ierr);
11160369aaf7SStefano Zampini         ierr = VecRestoreArray(dummy_vec,&array);CHKERRQ(ierr);
11170369aaf7SStefano Zampini         ierr = MatDestroy(&B0);CHKERRQ(ierr);
11180369aaf7SStefano Zampini         ierr = VecDestroy(&dummy_vec);CHKERRQ(ierr);
11190369aaf7SStefano Zampini       }
11208666afb4SStefano Zampini #endif
1121b76ba322SStefano Zampini     }
1122b76ba322SStefano Zampini   }
1123b76ba322SStefano Zampini 
11240369aaf7SStefano Zampini   /* change rhs and iteration matrix if using the change of basis */
1125b9b85e73SStefano Zampini   if (pcbddc->ChangeOfBasisMatrix) {
1126906d46d4SStefano Zampini     PCBDDCChange_ctx change_ctx;
1127906d46d4SStefano Zampini 
1128906d46d4SStefano Zampini     /* get change ctx */
1129906d46d4SStefano Zampini     ierr = MatShellGetContext(pcbddc->new_global_mat,&change_ctx);CHKERRQ(ierr);
1130906d46d4SStefano Zampini 
1131906d46d4SStefano Zampini     /* set current iteration matrix inside change context (change of basis has been already set into the ctx during PCSetUp) */
1132906d46d4SStefano Zampini     ierr = MatDestroy(&change_ctx->original_mat);CHKERRQ(ierr);
1133906d46d4SStefano Zampini     ierr = PetscObjectReference((PetscObject)pc->mat);CHKERRQ(ierr);
1134906d46d4SStefano Zampini     change_ctx->original_mat = pc->mat;
1135906d46d4SStefano Zampini 
1136906d46d4SStefano Zampini     /* change iteration matrix */
1137906d46d4SStefano Zampini     ierr = MatDestroy(&pc->mat);CHKERRQ(ierr);
1138906d46d4SStefano Zampini     ierr = PetscObjectReference((PetscObject)pcbddc->new_global_mat);CHKERRQ(ierr);
1139906d46d4SStefano Zampini     pc->mat = pcbddc->new_global_mat;
1140906d46d4SStefano Zampini 
11418d00608fSStefano Zampini     /* store the original rhs */
11428d00608fSStefano Zampini     if (copy_rhs) {
11438d00608fSStefano Zampini       ierr = VecCopy(rhs,pcbddc->original_rhs);CHKERRQ(ierr);
11448d00608fSStefano Zampini       copy_rhs = PETSC_FALSE;
11458d00608fSStefano Zampini     }
11468d00608fSStefano Zampini 
1147906d46d4SStefano Zampini     /* change rhs */
1148906d46d4SStefano Zampini     ierr = MatMultTranspose(change_ctx->global_change,rhs,pcis->vec1_global);CHKERRQ(ierr);
1149906d46d4SStefano Zampini     ierr = VecCopy(pcis->vec1_global,rhs);CHKERRQ(ierr);
11508d00608fSStefano Zampini     pcbddc->rhs_change = PETSC_TRUE;
115192e3dcfbSStefano Zampini   }
115292e3dcfbSStefano Zampini 
11530369aaf7SStefano Zampini   /* remove non-benign solution from the rhs */
11540369aaf7SStefano Zampini   if (pcbddc->benign_saddle_point) {
11550369aaf7SStefano Zampini     /* store the original rhs */
11560369aaf7SStefano Zampini     if (copy_rhs) {
11570369aaf7SStefano Zampini       ierr = VecCopy(rhs,pcbddc->original_rhs);CHKERRQ(ierr);
11580369aaf7SStefano Zampini       copy_rhs = PETSC_FALSE;
11590369aaf7SStefano Zampini     }
11600369aaf7SStefano Zampini     ierr = VecScale(pcbddc->benign_vec,-1.0);CHKERRQ(ierr);
11610369aaf7SStefano Zampini     ierr = MatMultAdd(pc->mat,pcbddc->benign_vec,rhs,rhs);CHKERRQ(ierr);
11620369aaf7SStefano Zampini     pcbddc->rhs_change = PETSC_TRUE;
11630369aaf7SStefano Zampini   }
11640369aaf7SStefano Zampini 
11650369aaf7SStefano Zampini   /* set initial guess if using PCG */
11660369aaf7SStefano Zampini   if (x && pcbddc->use_exact_dirichlet_trick) {
11670369aaf7SStefano Zampini     ierr = VecSet(x,0.0);CHKERRQ(ierr);
11680369aaf7SStefano Zampini     ierr = VecScatterBegin(pcis->global_to_D,rhs,pcis->vec1_D,INSERT_VALUES,SCATTER_FORWARD);CHKERRQ(ierr);
11690369aaf7SStefano Zampini     ierr = VecScatterEnd(pcis->global_to_D,rhs,pcis->vec1_D,INSERT_VALUES,SCATTER_FORWARD);CHKERRQ(ierr);
11700369aaf7SStefano Zampini     ierr = KSPSolve(pcbddc->ksp_D,pcis->vec1_D,pcis->vec2_D);CHKERRQ(ierr);
11710369aaf7SStefano Zampini     ierr = VecScatterBegin(pcis->global_to_D,pcis->vec2_D,x,INSERT_VALUES,SCATTER_REVERSE);CHKERRQ(ierr);
11720369aaf7SStefano Zampini     ierr = VecScatterEnd(pcis->global_to_D,pcis->vec2_D,x,INSERT_VALUES,SCATTER_REVERSE);CHKERRQ(ierr);
11730369aaf7SStefano Zampini     if (ksp) {
11740369aaf7SStefano Zampini       ierr = KSPSetInitialGuessNonzero(ksp,PETSC_TRUE);CHKERRQ(ierr);
11750369aaf7SStefano Zampini     }
11760369aaf7SStefano Zampini   }
11770369aaf7SStefano Zampini 
117892e3dcfbSStefano Zampini   /* remove nullspace if present */
11798efcfb23SStefano Zampini   if (ksp && x && pcbddc->NullSpace) {
11808efcfb23SStefano Zampini     ierr = MatNullSpaceRemove(pcbddc->NullSpace,x);CHKERRQ(ierr);
11818d00608fSStefano Zampini     /* store the original rhs */
11828d00608fSStefano Zampini     if (copy_rhs) {
11838d00608fSStefano Zampini       ierr = VecCopy(rhs,pcbddc->original_rhs);CHKERRQ(ierr);
11848d00608fSStefano Zampini       copy_rhs = PETSC_FALSE;
11858d00608fSStefano Zampini     }
11868d00608fSStefano Zampini     pcbddc->rhs_change = PETSC_TRUE;
1187d0195637SJed Brown     ierr = MatNullSpaceRemove(pcbddc->NullSpace,rhs);CHKERRQ(ierr);
1188b76ba322SStefano Zampini   }
1189534831adSStefano Zampini   PetscFunctionReturn(0);
1190534831adSStefano Zampini }
1191906d46d4SStefano Zampini 
1192534831adSStefano Zampini /* -------------------------------------------------------------------------- */
1193534831adSStefano Zampini #undef __FUNCT__
1194534831adSStefano Zampini #define __FUNCT__ "PCPostSolve_BDDC"
1195534831adSStefano Zampini /* -------------------------------------------------------------------------- */
1196534831adSStefano Zampini /*
1197534831adSStefano Zampini    PCPostSolve_BDDC - Changes the computed solution if a transformation of basis
1198534831adSStefano Zampini                      approach has been selected. Also, restores rhs to its original state.
1199534831adSStefano Zampini 
1200534831adSStefano Zampini    Input Parameter:
1201534831adSStefano Zampini +  pc - the preconditioner contex
1202534831adSStefano Zampini 
1203534831adSStefano Zampini    Application Interface Routine: PCPostSolve()
1204534831adSStefano Zampini 
1205534831adSStefano Zampini    Notes:
1206534831adSStefano Zampini      The interface routine PCPostSolve() is not usually called directly by
1207534831adSStefano Zampini      the user, but instead is called by KSPSolve().
1208534831adSStefano Zampini */
1209534831adSStefano Zampini static PetscErrorCode PCPostSolve_BDDC(PC pc, KSP ksp, Vec rhs, Vec x)
1210534831adSStefano Zampini {
1211534831adSStefano Zampini   PetscErrorCode ierr;
1212534831adSStefano Zampini   PC_BDDC        *pcbddc = (PC_BDDC*)pc->data;
1213534831adSStefano Zampini 
1214534831adSStefano Zampini   PetscFunctionBegin;
1215b9b85e73SStefano Zampini   if (pcbddc->ChangeOfBasisMatrix) {
1216906d46d4SStefano Zampini     PCBDDCChange_ctx change_ctx;
1217906d46d4SStefano Zampini 
1218906d46d4SStefano Zampini     /* get change ctx */
1219906d46d4SStefano Zampini     ierr = MatShellGetContext(pcbddc->new_global_mat,&change_ctx);CHKERRQ(ierr);
1220906d46d4SStefano Zampini 
1221906d46d4SStefano Zampini     /* restore iteration matrix */
1222906d46d4SStefano Zampini     ierr = MatDestroy(&pc->mat);CHKERRQ(ierr);
1223906d46d4SStefano Zampini     ierr = PetscObjectReference((PetscObject)change_ctx->original_mat);CHKERRQ(ierr);
1224906d46d4SStefano Zampini     pc->mat = change_ctx->original_mat;
1225906d46d4SStefano Zampini 
12268666afb4SStefano Zampini     /* need to restore the local matrices */
12278666afb4SStefano Zampini     if (pcbddc->benign_saddle_point && pcbddc->benign_change) {
12288666afb4SStefano Zampini       Mat_IS *matis = (Mat_IS*)pc->mat->data;
12298666afb4SStefano Zampini 
12308666afb4SStefano Zampini       ierr = MatDestroy(&matis->A);CHKERRQ(ierr);
12318666afb4SStefano Zampini       ierr = PetscObjectReference((PetscObject)pcbddc->benign_original_mat);CHKERRQ(ierr);
12328666afb4SStefano Zampini       matis->A = pcbddc->benign_original_mat;
12338666afb4SStefano Zampini       ierr = MatDestroy(&pcbddc->benign_original_mat);CHKERRQ(ierr);
12348666afb4SStefano Zampini     }
12358666afb4SStefano Zampini 
1236906d46d4SStefano Zampini     /* get solution in original basis */
1237906d46d4SStefano Zampini     if (x) {
1238906d46d4SStefano Zampini       PC_IS *pcis = (PC_IS*)(pc->data);
123906f24817SStefano Zampini 
12408666afb4SStefano Zampini 
12412d3346b4SStefano Zampini       /* restore solution on pressures */
124206a4e24aSStefano Zampini       if (pcbddc->benign_saddle_point) {
12432d3346b4SStefano Zampini         Mat_IS *matis = (Mat_IS*)pc->mat->data;
12442d3346b4SStefano Zampini 
12450369aaf7SStefano Zampini         /* add non-benign solution */
12468666afb4SStefano Zampini         ierr = VecAXPY(x,-1.0,pcbddc->benign_vec);CHKERRQ(ierr);
12470369aaf7SStefano Zampini 
12480369aaf7SStefano Zampini         /* change basis on pressures for x */
124906f24817SStefano Zampini         ierr = VecScatterBegin(matis->rctx,x,pcis->vec1_N,INSERT_VALUES,SCATTER_FORWARD);CHKERRQ(ierr);
125006f24817SStefano Zampini         ierr = VecScatterEnd(matis->rctx,x,pcis->vec1_N,INSERT_VALUES,SCATTER_FORWARD);CHKERRQ(ierr);
12512d3346b4SStefano Zampini         if (pcbddc->benign_change) {
12522d3346b4SStefano Zampini 
12532d3346b4SStefano Zampini           ierr = MatMult(pcbddc->benign_change,pcis->vec1_N,pcis->vec2_N);CHKERRQ(ierr);
125406f24817SStefano Zampini           ierr = VecScatterBegin(matis->rctx,pcis->vec2_N,x,INSERT_VALUES,SCATTER_REVERSE);CHKERRQ(ierr);
125506f24817SStefano Zampini           ierr = VecScatterEnd(matis->rctx,pcis->vec2_N,x,INSERT_VALUES,SCATTER_REVERSE);CHKERRQ(ierr);
12562d3346b4SStefano Zampini         } else {
125706f24817SStefano Zampini           ierr = VecScatterBegin(matis->rctx,pcis->vec1_N,x,INSERT_VALUES,SCATTER_REVERSE);CHKERRQ(ierr);
125806f24817SStefano Zampini           ierr = VecScatterEnd(matis->rctx,pcis->vec1_N,x,INSERT_VALUES,SCATTER_REVERSE);CHKERRQ(ierr);
12592d3346b4SStefano Zampini         }
12602d3346b4SStefano Zampini       }
12610369aaf7SStefano Zampini       /* change basis on x */
12620369aaf7SStefano Zampini       ierr = MatMult(change_ctx->global_change,x,pcis->vec1_global);CHKERRQ(ierr);
1263906d46d4SStefano Zampini       ierr = VecCopy(pcis->vec1_global,x);CHKERRQ(ierr);
12643425bc38SStefano Zampini     }
1265534831adSStefano Zampini   }
1266906d46d4SStefano Zampini 
12673972b0daSStefano Zampini   /* add solution removed in presolve */
12686bcfc461SStefano Zampini   if (x && pcbddc->rhs_change) {
12693425bc38SStefano Zampini     ierr = VecAXPY(x,1.0,pcbddc->temp_solution);CHKERRQ(ierr);
12703425bc38SStefano Zampini   }
1271906d46d4SStefano Zampini 
1272fb223d50SStefano Zampini   /* restore rhs to its original state */
12738d00608fSStefano Zampini   if (rhs && pcbddc->rhs_change) {
1274fb223d50SStefano Zampini     ierr = VecCopy(pcbddc->original_rhs,rhs);CHKERRQ(ierr);
1275fb223d50SStefano Zampini   }
12768d00608fSStefano Zampini   pcbddc->rhs_change = PETSC_FALSE;
12778efcfb23SStefano Zampini 
12788efcfb23SStefano Zampini   /* restore ksp guess state */
12798efcfb23SStefano Zampini   if (ksp) {
12808efcfb23SStefano Zampini     ierr = KSPSetInitialGuessNonzero(ksp,pcbddc->ksp_guess_nonzero);CHKERRQ(ierr);
12818efcfb23SStefano Zampini   }
1282534831adSStefano Zampini   PetscFunctionReturn(0);
1283534831adSStefano Zampini }
1284534831adSStefano Zampini /* -------------------------------------------------------------------------- */
128553cdbc3dSStefano Zampini #undef __FUNCT__
128653cdbc3dSStefano Zampini #define __FUNCT__ "PCSetUp_BDDC"
12870c7d97c5SJed Brown /* -------------------------------------------------------------------------- */
12880c7d97c5SJed Brown /*
12890c7d97c5SJed Brown    PCSetUp_BDDC - Prepares for the use of the BDDC preconditioner
12900c7d97c5SJed Brown                   by setting data structures and options.
12910c7d97c5SJed Brown 
12920c7d97c5SJed Brown    Input Parameter:
129353cdbc3dSStefano Zampini +  pc - the preconditioner context
12940c7d97c5SJed Brown 
12950c7d97c5SJed Brown    Application Interface Routine: PCSetUp()
12960c7d97c5SJed Brown 
12970c7d97c5SJed Brown    Notes:
12980c7d97c5SJed Brown      The interface routine PCSetUp() is not usually called directly by
12990c7d97c5SJed Brown      the user, but instead is called by PCApply() if necessary.
13000c7d97c5SJed Brown */
130153cdbc3dSStefano Zampini PetscErrorCode PCSetUp_BDDC(PC pc)
13020c7d97c5SJed Brown {
13030c7d97c5SJed Brown   PetscErrorCode ierr;
13040c7d97c5SJed Brown   PC_BDDC*       pcbddc = (PC_BDDC*)pc->data;
13055e8657edSStefano Zampini   Mat_IS*        matis;
130608122e43SStefano Zampini   MatNullSpace   nearnullspace;
1307c9ed8603SStefano Zampini   IS             zerodiag = NULL;
130891e8d312SStefano Zampini   PetscInt       nrows,ncols;
130908122e43SStefano Zampini   PetscBool      computetopography,computesolvers,computesubschurs;
13108de1fae6SStefano Zampini   PetscBool      computeconstraintsmatrix;
13115e8657edSStefano Zampini   PetscBool      new_nearnullspace_provided,ismatis;
13120c7d97c5SJed Brown 
13130c7d97c5SJed Brown   PetscFunctionBegin;
13145e8657edSStefano Zampini   ierr = PetscObjectTypeCompare((PetscObject)pc->pmat,MATIS,&ismatis);CHKERRQ(ierr);
13155e8657edSStefano Zampini   if (!ismatis) {
13165e8657edSStefano Zampini     SETERRQ(PetscObjectComm((PetscObject)pc),PETSC_ERR_ARG_WRONG,"PCBDDC preconditioner requires matrix of type MATIS");
13175e8657edSStefano Zampini   }
131891e8d312SStefano Zampini   ierr = MatGetSize(pc->pmat,&nrows,&ncols);CHKERRQ(ierr);
131991e8d312SStefano Zampini   if (nrows != ncols) {
132091e8d312SStefano Zampini     SETERRQ(PetscObjectComm((PetscObject)pc),PETSC_ERR_SUP,"PCBDDC preconditioner requires a square preconditioning matrix");
132191e8d312SStefano Zampini   }
13225e8657edSStefano Zampini   matis = (Mat_IS*)pc->pmat->data;
1323f4ddd8eeSStefano Zampini   /* the following lines of code should be replaced by a better logic between PCIS, PCNN, PCBDDC and other future nonoverlapping preconditioners */
13243b03a366Sstefano_zampini   /* For BDDC we need to define a local "Neumann" problem different to that defined in PCISSetup
1325fcd409f5SStefano Zampini      Also, BDDC directly build the Dirichlet problem */
1326f4ddd8eeSStefano Zampini   /* split work */
1327674ae819SStefano Zampini   if (pc->setupcalled) {
1328d1e9a80fSBarry Smith     if (pc->flag == SAME_NONZERO_PATTERN) {
1329674ae819SStefano Zampini       computetopography = PETSC_FALSE;
1330674ae819SStefano Zampini       computesolvers = PETSC_TRUE;
1331674ae819SStefano Zampini     } else { /* DIFFERENT_NONZERO_PATTERN */
1332674ae819SStefano Zampini       computetopography = PETSC_TRUE;
1333674ae819SStefano Zampini       computesolvers = PETSC_TRUE;
1334674ae819SStefano Zampini     }
1335674ae819SStefano Zampini   } else {
1336674ae819SStefano Zampini     computetopography = PETSC_TRUE;
1337674ae819SStefano Zampini     computesolvers = PETSC_TRUE;
1338674ae819SStefano Zampini   }
1339fb180af4SStefano Zampini   if (pcbddc->recompute_topography) {
1340fb180af4SStefano Zampini     computetopography = PETSC_TRUE;
1341fb180af4SStefano Zampini   }
13428de1fae6SStefano Zampini   computeconstraintsmatrix = PETSC_FALSE;
1343b087196eSStefano Zampini 
1344b087196eSStefano Zampini   /* check parameters' compatibility */
13455a95e1ceSStefano Zampini   if (pcbddc->adaptive_threshold > 0.0 && !pcbddc->use_deluxe_scaling) {
13465a95e1ceSStefano Zampini     SETERRQ(PetscObjectComm((PetscObject)pc),PETSC_ERR_SUP,"Cannot compute adaptive constraints without deluxe scaling. Rerun with -pc_bddc_use_deluxe_scaling");
13475a95e1ceSStefano Zampini   }
13485a95e1ceSStefano Zampini   pcbddc->adaptive_selection = (PetscBool)(pcbddc->adaptive_threshold > 0.0 && pcbddc->use_deluxe_scaling);
1349862806e4SStefano Zampini   if (pcbddc->adaptive_selection) pcbddc->use_faces = PETSC_TRUE;
1350862806e4SStefano Zampini 
13515a95e1ceSStefano Zampini   computesubschurs = (PetscBool)(pcbddc->adaptive_selection || pcbddc->use_deluxe_scaling);
13526816873aSStefano Zampini   if (pcbddc->faster_deluxe && pcbddc->adaptive_selection && pcbddc->use_change_of_basis) {
13536816873aSStefano Zampini     SETERRQ(PetscObjectComm((PetscObject)pc),PETSC_ERR_SUP,"Cannot compute faster deluxe if adaptivity and change of basis are both requested. Rerun with -pc_bddc_deluxe_faster false");
13546816873aSStefano Zampini   }
13552d3346b4SStefano Zampini 
13562d3346b4SStefano Zampini   /* check if the iteration matrix is of type MATIS in case the benign trick has been requested */
13572d3346b4SStefano Zampini   ierr = PetscObjectTypeCompare((PetscObject)pc->mat,MATIS,&ismatis);CHKERRQ(ierr);
135806a4e24aSStefano Zampini   if (pcbddc->benign_saddle_point && !ismatis) {
13592d3346b4SStefano Zampini     SETERRQ(PetscObjectComm((PetscObject)pc),PETSC_ERR_ARG_WRONG,"PCBDDC preconditioner with benign subspace trick requires the iteration matrix to be of type MATIS");
13602d3346b4SStefano Zampini   }
13612d3346b4SStefano Zampini 
1362ae9e6838SStefano Zampini   /* raise error if the user has provided the change of basis and the benign trick has been requested */
1363ae9e6838SStefano Zampini   if (pcbddc->benign_saddle_point && pcbddc->user_ChangeOfBasisMatrix) {
1364ae9e6838SStefano Zampini     SETERRQ(PetscObjectComm((PetscObject)pc),PETSC_ERR_SUP,"Cannot apply the benign trick with a user defined change of basis");
1365ae9e6838SStefano Zampini   }
1366ae9e6838SStefano Zampini 
1367f4ddd8eeSStefano Zampini   /* Get stdout for dbg */
136870cf5478SStefano Zampini   if (pcbddc->dbg_flag) {
136970cf5478SStefano Zampini     if (!pcbddc->dbg_viewer) {
137058a03d70SStefano Zampini       pcbddc->dbg_viewer = PETSC_VIEWER_STDOUT_(PetscObjectComm((PetscObject)pc));
1371f4ddd8eeSStefano Zampini       ierr = PetscViewerASCIISynchronizedAllow(pcbddc->dbg_viewer,PETSC_TRUE);CHKERRQ(ierr);
1372f4ddd8eeSStefano Zampini     }
137358a03d70SStefano Zampini     ierr = PetscViewerASCIIAddTab(pcbddc->dbg_viewer,2*pcbddc->current_level);CHKERRQ(ierr);
1374f4ddd8eeSStefano Zampini   }
1375f4ddd8eeSStefano Zampini 
13765e8657edSStefano Zampini   if (pcbddc->user_ChangeOfBasisMatrix) {
13775e8657edSStefano Zampini     /* use_change_of_basis flag is used to automatically compute a change of basis from constraints */
13785e8657edSStefano Zampini     pcbddc->use_change_of_basis = PETSC_FALSE;
13795e8657edSStefano Zampini     ierr = PCBDDCComputeLocalMatrix(pc,pcbddc->user_ChangeOfBasisMatrix);CHKERRQ(ierr);
13805e8657edSStefano Zampini   } else {
1381b96c3477SStefano Zampini     ierr = MatDestroy(&pcbddc->local_mat);CHKERRQ(ierr);
138206a4e24aSStefano Zampini     if (!pcbddc->benign_saddle_point) {
13835e8657edSStefano Zampini       ierr = PetscObjectReference((PetscObject)matis->A);CHKERRQ(ierr);
13845e8657edSStefano Zampini       pcbddc->local_mat = matis->A;
1385ae9e6838SStefano Zampini     } else {
138681d14e9dSStefano Zampini       PetscInt  nz;
138781d14e9dSStefano Zampini       PetscBool sorted;
1388f5fd1fbfSStefano Zampini 
1389c9ed8603SStefano Zampini       ierr = MatFindZeroDiagonals(matis->A,&zerodiag);CHKERRQ(ierr);
1390c9ed8603SStefano Zampini       ierr = ISSorted(zerodiag,&sorted);CHKERRQ(ierr);
139181d14e9dSStefano Zampini       if (!sorted) {
1392c9ed8603SStefano Zampini         ierr = ISSort(zerodiag);CHKERRQ(ierr);
1393f5fd1fbfSStefano Zampini       }
1394c9ed8603SStefano Zampini       ierr = ISGetLocalSize(zerodiag,&nz);CHKERRQ(ierr);
139581d14e9dSStefano Zampini       if (nz) {
139681d14e9dSStefano Zampini         IS                zerodiagc;
139781d14e9dSStefano Zampini         PetscScalar       *array;
1398c263805aSStefano Zampini         const PetscInt    *idxs,*idxsc;
1399c263805aSStefano Zampini         PetscInt          i,n,*nnz;
140081d14e9dSStefano Zampini 
140181d14e9dSStefano Zampini         /* TODO: add check for shared dofs */
140281d14e9dSStefano Zampini         pcbddc->use_change_of_basis = PETSC_TRUE;
1403c9ed8603SStefano Zampini         pcbddc->use_change_on_faces = PETSC_TRUE;
140481d14e9dSStefano Zampini         ierr = MatGetLocalSize(matis->A,&n,NULL);CHKERRQ(ierr);
1405c9ed8603SStefano Zampini         ierr = ISComplement(zerodiag,0,n,&zerodiagc);CHKERRQ(ierr);
1406c9ed8603SStefano Zampini         ierr = ISGetIndices(zerodiag,&idxs);CHKERRQ(ierr);
140781d14e9dSStefano Zampini         ierr = ISGetIndices(zerodiagc,&idxsc);CHKERRQ(ierr);
140881d14e9dSStefano Zampini         /* local change of basis for pressures */
140981d14e9dSStefano Zampini         ierr = MatDestroy(&pcbddc->benign_change);CHKERRQ(ierr);
141081d14e9dSStefano Zampini         ierr = MatCreate(PetscObjectComm((PetscObject)matis->A),&pcbddc->benign_change);CHKERRQ(ierr);
141181d14e9dSStefano Zampini         ierr = MatSetType(pcbddc->benign_change,MATAIJ);CHKERRQ(ierr);
141281d14e9dSStefano Zampini         ierr = MatSetSizes(pcbddc->benign_change,n,n,PETSC_DECIDE,PETSC_DECIDE);CHKERRQ(ierr);
141381d14e9dSStefano Zampini         ierr = PetscMalloc1(n,&nnz);CHKERRQ(ierr);
141481d14e9dSStefano Zampini         for (i=0;i<n-nz;i++) nnz[idxsc[i]] = 1; /* identity on velocities */
141581d14e9dSStefano Zampini         for (i=0;i<nz-1;i++) nnz[idxs[i]] = 2; /* change on pressures */
141681d14e9dSStefano Zampini         nnz[idxs[nz-1]] = nz; /* last local pressure dof: _0 set */
141781d14e9dSStefano Zampini         ierr = MatSeqAIJSetPreallocation(pcbddc->benign_change,0,nnz);CHKERRQ(ierr);
141881d14e9dSStefano Zampini         ierr = PetscFree(nnz);CHKERRQ(ierr);
141981d14e9dSStefano Zampini         /* set identity on velocities */
142081d14e9dSStefano Zampini         for (i=0;i<n-nz;i++) {
142181d14e9dSStefano Zampini           ierr = MatSetValue(pcbddc->benign_change,idxsc[i],idxsc[i],1.,INSERT_VALUES);CHKERRQ(ierr);
142281d14e9dSStefano Zampini         }
142381d14e9dSStefano Zampini         /* set change on pressures */
142481d14e9dSStefano Zampini         for (i=0;i<nz-1;i++) {
142581d14e9dSStefano Zampini           PetscScalar vals[2];
142681d14e9dSStefano Zampini           PetscInt    cols[2];
142781d14e9dSStefano Zampini 
142881d14e9dSStefano Zampini           /* TODO: add quadrature */
142981d14e9dSStefano Zampini           cols[0] = idxs[i];
143081d14e9dSStefano Zampini           cols[1] = idxs[nz-1];
143181d14e9dSStefano Zampini           vals[0] = 1.;
143281d14e9dSStefano Zampini           vals[1] = 1./nz;
143381d14e9dSStefano Zampini           ierr = MatSetValues(pcbddc->benign_change,1,idxs+i,2,cols,vals,INSERT_VALUES);CHKERRQ(ierr);
143481d14e9dSStefano Zampini         }
143581d14e9dSStefano Zampini         ierr = PetscMalloc1(nz,&array);CHKERRQ(ierr);
143681d14e9dSStefano Zampini         for (i=0;i<nz-1;i++) array[i] = -1.;
143781d14e9dSStefano Zampini         array[nz-1] = 1./nz;
143881d14e9dSStefano Zampini         ierr = MatSetValues(pcbddc->benign_change,1,idxs+nz-1,nz,idxs,array,INSERT_VALUES);CHKERRQ(ierr);
143981d14e9dSStefano Zampini         ierr = PetscFree(array);CHKERRQ(ierr);
144081d14e9dSStefano Zampini         ierr = MatAssemblyBegin(pcbddc->benign_change,MAT_FINAL_ASSEMBLY);CHKERRQ(ierr);
144181d14e9dSStefano Zampini         ierr = MatAssemblyEnd(pcbddc->benign_change,MAT_FINAL_ASSEMBLY);CHKERRQ(ierr);
144281d14e9dSStefano Zampini         /* TODO: need optimization? */
144381d14e9dSStefano Zampini         ierr = MatPtAP(matis->A,pcbddc->benign_change,MAT_INITIAL_MATRIX,2.0,&pcbddc->local_mat);CHKERRQ(ierr);
1444efc2fbd9SStefano Zampini         /* store local and global idxs for p0 */
1445efc2fbd9SStefano Zampini         pcbddc->benign_p0_lidx = idxs[nz-1];
1446efc2fbd9SStefano Zampini         ierr = ISLocalToGlobalMappingApply(pc->pmat->rmap->mapping,1,&idxs[nz-1],&pcbddc->benign_p0_gidx);CHKERRQ(ierr);
1447c9ed8603SStefano Zampini         ierr = ISRestoreIndices(zerodiag,&idxs);CHKERRQ(ierr);
1448c263805aSStefano Zampini         ierr = ISRestoreIndices(zerodiagc,&idxsc);CHKERRQ(ierr);
1449c263805aSStefano Zampini         ierr = ISDestroy(&zerodiagc);CHKERRQ(ierr);
1450c263805aSStefano Zampini         /* pop B0 mat from pcbddc->local_mat */
1451c263805aSStefano Zampini         ierr = PCBDDCBenignPopOrPushB0(pc,PETSC_TRUE);CHKERRQ(ierr);
1452019a44ceSStefano Zampini       } else { /* this is unlikely to happen but, just in case, destroy the empty IS */
1453c9ed8603SStefano Zampini         ierr = ISDestroy(&zerodiag);CHKERRQ(ierr);
145481d14e9dSStefano Zampini         ierr = PetscObjectReference((PetscObject)matis->A);CHKERRQ(ierr);
145581d14e9dSStefano Zampini         pcbddc->local_mat = matis->A;
145681d14e9dSStefano Zampini       }
1457f5fd1fbfSStefano Zampini     }
1458674ae819SStefano Zampini   }
1459f4ddd8eeSStefano Zampini 
1460*f1a72664SStefano Zampini   /* propagate relevant information */
1461e496cd5dSStefano Zampini   /* workaround for reals */
1462e496cd5dSStefano Zampini #if !defined(PETSC_USE_COMPLEX)
14633301b35fSStefano Zampini   if (matis->A->symmetric_set) {
14643301b35fSStefano Zampini     ierr = MatSetOption(pcbddc->local_mat,MAT_HERMITIAN,matis->A->symmetric);CHKERRQ(ierr);
14653301b35fSStefano Zampini   }
1466e496cd5dSStefano Zampini #endif
146706a4e24aSStefano Zampini   if (matis->A->symmetric_set) {
146806a4e24aSStefano Zampini     ierr = MatSetOption(pcbddc->local_mat,MAT_SYMMETRIC,matis->A->symmetric);CHKERRQ(ierr);
146906a4e24aSStefano Zampini   }
147006a4e24aSStefano Zampini   if (matis->A->spd_set) {
147106a4e24aSStefano Zampini     ierr = MatSetOption(pcbddc->local_mat,MAT_SPD,matis->A->spd);CHKERRQ(ierr);
147206a4e24aSStefano Zampini   }
1473e496cd5dSStefano Zampini 
14745e8657edSStefano Zampini   /* Set up all the "iterative substructuring" common block without computing solvers */
14755e8657edSStefano Zampini   {
14765e8657edSStefano Zampini     Mat temp_mat;
14775e8657edSStefano Zampini 
14785e8657edSStefano Zampini     temp_mat = matis->A;
14795e8657edSStefano Zampini     matis->A = pcbddc->local_mat;
14805e8657edSStefano Zampini     ierr = PCISSetUp(pc,PETSC_FALSE);CHKERRQ(ierr);
14815e8657edSStefano Zampini     pcbddc->local_mat = matis->A;
14825e8657edSStefano Zampini     matis->A = temp_mat;
14835e8657edSStefano Zampini   }
1484684f6988SStefano Zampini 
148581d14e9dSStefano Zampini   /* Analyze interface */
1486674ae819SStefano Zampini   if (computetopography) {
1487674ae819SStefano Zampini     ierr = PCBDDCAnalyzeInterface(pc);CHKERRQ(ierr);
14888de1fae6SStefano Zampini     computeconstraintsmatrix = PETSC_TRUE;
1489674ae819SStefano Zampini   }
1490fb8d54d4SStefano Zampini 
1491c9ed8603SStefano Zampini   /* check b(v_I,p_0) = 0 for all v_I */
1492c9ed8603SStefano Zampini   if (pcbddc->dbg_flag && zerodiag) {
149306f24817SStefano Zampini     PC_IS          *pcis = (PC_IS*)(pc->data);
1494*f1a72664SStefano Zampini     Mat            A;
1495*f1a72664SStefano Zampini     Vec            vec3_N;
149606f24817SStefano Zampini     IS             dirIS = NULL;
149706f24817SStefano Zampini     PetscScalar    *vals;
149806f24817SStefano Zampini     const PetscInt *idxs;
149906f24817SStefano Zampini     PetscInt       i,nz;
150006f24817SStefano Zampini 
150106f24817SStefano Zampini     /* p0 */
150206f24817SStefano Zampini     ierr = VecSet(pcis->vec1_N,0.);CHKERRQ(ierr);
150306f24817SStefano Zampini     ierr = PetscMalloc1(pcis->n,&vals);CHKERRQ(ierr);
1504c9ed8603SStefano Zampini     ierr = ISGetLocalSize(zerodiag,&nz);CHKERRQ(ierr);
1505c9ed8603SStefano Zampini     ierr = ISGetIndices(zerodiag,&idxs);CHKERRQ(ierr);
150606f24817SStefano Zampini     for (i=0;i<nz;i++) vals[i] = 1.;
150706f24817SStefano Zampini     ierr = VecSetValues(pcis->vec1_N,nz,idxs,vals,INSERT_VALUES);CHKERRQ(ierr);
150806f24817SStefano Zampini     ierr = VecAssemblyBegin(pcis->vec1_N);CHKERRQ(ierr);
150906f24817SStefano Zampini     ierr = VecAssemblyEnd(pcis->vec1_N);CHKERRQ(ierr);
151006f24817SStefano Zampini     /* v_I */
151106f24817SStefano Zampini     ierr = VecSetRandom(pcis->vec2_N,NULL);CHKERRQ(ierr);
151206f24817SStefano Zampini     for (i=0;i<nz;i++) vals[i] = 0.;
151306f24817SStefano Zampini     ierr = VecSetValues(pcis->vec2_N,nz,idxs,vals,INSERT_VALUES);CHKERRQ(ierr);
1514c9ed8603SStefano Zampini     ierr = ISRestoreIndices(zerodiag,&idxs);CHKERRQ(ierr);
151506f24817SStefano Zampini     ierr = ISGetIndices(pcis->is_B_local,&idxs);CHKERRQ(ierr);
151606f24817SStefano Zampini     for (i=0;i<pcis->n_B;i++) vals[i] = 0.;
151706f24817SStefano Zampini     ierr = VecSetValues(pcis->vec2_N,pcis->n_B,idxs,vals,INSERT_VALUES);CHKERRQ(ierr);
151806f24817SStefano Zampini     ierr = ISRestoreIndices(pcis->is_B_local,&idxs);CHKERRQ(ierr);
151906f24817SStefano Zampini     ierr = PCBDDCGraphGetDirichletDofs(pcbddc->mat_graph,&dirIS);CHKERRQ(ierr);
152006f24817SStefano Zampini     if (dirIS) {
152106f24817SStefano Zampini       PetscInt n;
152206f24817SStefano Zampini 
152306f24817SStefano Zampini       ierr = ISGetLocalSize(dirIS,&n);CHKERRQ(ierr);
152406f24817SStefano Zampini       ierr = ISGetIndices(dirIS,&idxs);CHKERRQ(ierr);
152506f24817SStefano Zampini       for (i=0;i<n;i++) vals[i] = 0.;
152606f24817SStefano Zampini       ierr = VecSetValues(pcis->vec2_N,n,idxs,vals,INSERT_VALUES);CHKERRQ(ierr);
152706f24817SStefano Zampini       ierr = ISRestoreIndices(dirIS,&idxs);CHKERRQ(ierr);
152806f24817SStefano Zampini     }
152906f24817SStefano Zampini     ierr = ISDestroy(&dirIS);CHKERRQ(ierr);
153006f24817SStefano Zampini     ierr = VecAssemblyBegin(pcis->vec2_N);CHKERRQ(ierr);
153106f24817SStefano Zampini     ierr = VecAssemblyEnd(pcis->vec2_N);CHKERRQ(ierr);
1532*f1a72664SStefano Zampini     ierr = VecDuplicate(pcis->vec1_N,&vec3_N);CHKERRQ(ierr);
1533*f1a72664SStefano Zampini     ierr = VecSet(vec3_N,0.);CHKERRQ(ierr);
1534*f1a72664SStefano Zampini     ierr = MatISGetLocalMat(pc->pmat,&A);CHKERRQ(ierr);
1535*f1a72664SStefano Zampini     ierr = MatMult(A,pcis->vec1_N,vec3_N);CHKERRQ(ierr);
1536*f1a72664SStefano Zampini     ierr = VecDot(vec3_N,pcis->vec2_N,&vals[0]);CHKERRQ(ierr);
153706f24817SStefano Zampini     ierr = PetscPrintf(PETSC_COMM_SELF,"[%d] check original mat: %1.4e\n",PetscGlobalRank,vals[0]);CHKERRQ(ierr);
153806f24817SStefano Zampini     ierr = PetscFree(vals);CHKERRQ(ierr);
1539*f1a72664SStefano Zampini     ierr = VecDestroy(&vec3_N);CHKERRQ(ierr);
1540c9ed8603SStefano Zampini   }
1541efc2fbd9SStefano Zampini 
1542efc2fbd9SStefano Zampini   /* check PCBDDCBenignPopOrPush */
1543c9ed8603SStefano Zampini   if (pcbddc->dbg_flag && pcbddc->benign_saddle_point) {
1544c9ed8603SStefano Zampini     PC_IS  *pcis = (PC_IS*)(pc->data);
1545c9ed8603SStefano Zampini 
1546efc2fbd9SStefano Zampini     ierr = VecSetRandom(pcis->vec1_global,NULL);CHKERRQ(ierr);
1547efc2fbd9SStefano Zampini     pcbddc->benign_p0 = -PetscGlobalRank;
1548efc2fbd9SStefano Zampini     ierr = PCBDDCBenignPopOrPushP0(pc,pcis->vec1_global,PETSC_FALSE);CHKERRQ(ierr);
1549efc2fbd9SStefano Zampini     pcbddc->benign_p0 = 1;
1550efc2fbd9SStefano Zampini     ierr = PCBDDCBenignPopOrPushP0(pc,pcis->vec1_global,PETSC_TRUE);CHKERRQ(ierr);
1551c9ed8603SStefano Zampini     if (pcbddc->benign_p0_gidx >=0 && pcbddc->benign_p0 != -PetscGlobalRank) {
1552efc2fbd9SStefano Zampini       SETERRQ2(PETSC_COMM_SELF,PETSC_ERR_PLIB,"Error testing PCBDDCBenignPopOrPushP0! Found %1.4e instead of %1.4e\n",pcbddc->benign_p0,-PetscGlobalRank);CHKERRQ(ierr);
1553efc2fbd9SStefano Zampini     }
155406f24817SStefano Zampini   }
155506f24817SStefano Zampini 
1556b96c3477SStefano Zampini   /* Setup local dirichlet solver ksp_D and sub_schurs solvers */
1557684f6988SStefano Zampini   if (computesolvers) {
1558d5574798SStefano Zampini     PCBDDCSubSchurs sub_schurs=pcbddc->sub_schurs;
1559d5574798SStefano Zampini 
1560d5574798SStefano Zampini     if (computesubschurs && computetopography) {
156108122e43SStefano Zampini       ierr = PCBDDCInitSubSchurs(pc);CHKERRQ(ierr);
1562b1b3d7a2SStefano Zampini     }
1563d5574798SStefano Zampini     if (sub_schurs->use_mumps) {
15642070dbb6SStefano Zampini       if (computesubschurs) {
156508122e43SStefano Zampini         ierr = PCBDDCSetUpSubSchurs(pc);CHKERRQ(ierr);
15662070dbb6SStefano Zampini       }
1567d5574798SStefano Zampini       ierr = PCBDDCSetUpLocalSolvers(pc,PETSC_TRUE,PETSC_FALSE);CHKERRQ(ierr);
1568d5574798SStefano Zampini     } else {
1569d5574798SStefano Zampini       ierr = PCBDDCSetUpLocalSolvers(pc,PETSC_TRUE,PETSC_FALSE);CHKERRQ(ierr);
15702070dbb6SStefano Zampini       if (computesubschurs) {
1571d5574798SStefano Zampini         ierr = PCBDDCSetUpSubSchurs(pc);CHKERRQ(ierr);
1572d5574798SStefano Zampini       }
15732070dbb6SStefano Zampini     }
157408122e43SStefano Zampini     if (pcbddc->adaptive_selection) {
157508122e43SStefano Zampini       ierr = PCBDDCAdaptiveSelection(pc);CHKERRQ(ierr);
15768de1fae6SStefano Zampini       computeconstraintsmatrix = PETSC_TRUE;
1577b7eb3628SStefano Zampini     }
1578b1b3d7a2SStefano Zampini   }
1579684f6988SStefano Zampini 
1580f4ddd8eeSStefano Zampini   /* infer if NullSpace object attached to Mat via MatSetNearNullSpace has changed */
1581fb8d54d4SStefano Zampini   new_nearnullspace_provided = PETSC_FALSE;
1582f4ddd8eeSStefano Zampini   ierr = MatGetNearNullSpace(pc->pmat,&nearnullspace);CHKERRQ(ierr);
1583f4ddd8eeSStefano Zampini   if (pcbddc->onearnullspace) { /* already used nearnullspace */
1584f4ddd8eeSStefano Zampini     if (!nearnullspace) { /* near null space attached to mat has been destroyed */
1585f4ddd8eeSStefano Zampini       new_nearnullspace_provided = PETSC_TRUE;
1586f4ddd8eeSStefano Zampini     } else {
1587f4ddd8eeSStefano Zampini       /* determine if the two nullspaces are different (should be lightweight) */
1588f4ddd8eeSStefano Zampini       if (nearnullspace != pcbddc->onearnullspace) {
1589f4ddd8eeSStefano Zampini         new_nearnullspace_provided = PETSC_TRUE;
1590165b64e2SStefano Zampini       } else { /* maybe the user has changed the content of the nearnullspace so check vectors ObjectStateId */
1591f4ddd8eeSStefano Zampini         PetscInt         i;
1592165b64e2SStefano Zampini         const Vec        *nearnullvecs;
1593165b64e2SStefano Zampini         PetscObjectState state;
1594165b64e2SStefano Zampini         PetscInt         nnsp_size;
1595165b64e2SStefano Zampini         ierr = MatNullSpaceGetVecs(nearnullspace,NULL,&nnsp_size,&nearnullvecs);CHKERRQ(ierr);
1596f4ddd8eeSStefano Zampini         for (i=0;i<nnsp_size;i++) {
1597f4ddd8eeSStefano Zampini           ierr = PetscObjectStateGet((PetscObject)nearnullvecs[i],&state);CHKERRQ(ierr);
1598165b64e2SStefano Zampini           if (pcbddc->onearnullvecs_state[i] != state) {
1599f4ddd8eeSStefano Zampini             new_nearnullspace_provided = PETSC_TRUE;
1600f4ddd8eeSStefano Zampini             break;
1601f4ddd8eeSStefano Zampini           }
1602f4ddd8eeSStefano Zampini         }
1603f4ddd8eeSStefano Zampini       }
1604f4ddd8eeSStefano Zampini     }
1605f4ddd8eeSStefano Zampini   } else {
1606f4ddd8eeSStefano Zampini     if (!nearnullspace) { /* both nearnullspaces are null */
1607f4ddd8eeSStefano Zampini       new_nearnullspace_provided = PETSC_FALSE;
1608f4ddd8eeSStefano Zampini     } else { /* nearnullspace attached later */
1609f4ddd8eeSStefano Zampini       new_nearnullspace_provided = PETSC_TRUE;
1610f4ddd8eeSStefano Zampini     }
1611f4ddd8eeSStefano Zampini   }
1612f4ddd8eeSStefano Zampini 
1613f4ddd8eeSStefano Zampini   /* Setup constraints and related work vectors */
1614727cdba6SStefano Zampini   /* reset primal space flags */
1615f4ddd8eeSStefano Zampini   pcbddc->new_primal_space = PETSC_FALSE;
1616727cdba6SStefano Zampini   pcbddc->new_primal_space_local = PETSC_FALSE;
16178de1fae6SStefano Zampini   if (computeconstraintsmatrix || new_nearnullspace_provided) {
1618727cdba6SStefano Zampini     /* It also sets the primal space flags */
1619674ae819SStefano Zampini     ierr = PCBDDCConstraintsSetUp(pc);CHKERRQ(ierr);
1620e7b262bdSStefano Zampini     /* Allocate needed local vectors (which depends on quantities defined during ConstraintsSetUp) */
1621f4ddd8eeSStefano Zampini     ierr = PCBDDCSetUpLocalWorkVectors(pc);CHKERRQ(ierr);
16223975b054SStefano Zampini   }
16235e8657edSStefano Zampini 
16243975b054SStefano Zampini   if (computesolvers || pcbddc->new_primal_space) {
16255e8657edSStefano Zampini     if (pcbddc->use_change_of_basis) {
16265e8657edSStefano Zampini       PC_IS *pcis = (PC_IS*)(pc->data);
1627c263805aSStefano Zampini       Mat   temp_mat = NULL;
16285e8657edSStefano Zampini 
1629c9ed8603SStefano Zampini       if (zerodiag) {
1630c263805aSStefano Zampini         /* insert B0 in pcbddc->local_mat */
1631c263805aSStefano Zampini         ierr = PCBDDCBenignPopOrPushB0(pc,PETSC_FALSE);CHKERRQ(ierr);
163206f24817SStefano Zampini         if (pcbddc->dbg_flag) {
163306f24817SStefano Zampini           PC_IS          *pcis = (PC_IS*)(pc->data);
1634*f1a72664SStefano Zampini           Vec            vec3_N;
163506f24817SStefano Zampini           IS             dirIS = NULL;
163606f24817SStefano Zampini           PetscScalar    *vals;
163706f24817SStefano Zampini           const PetscInt *idxs;
163806f24817SStefano Zampini           PetscInt       i,nz;
163906f24817SStefano Zampini 
164006f24817SStefano Zampini           /* p0 */
164106f24817SStefano Zampini           ierr = VecSet(pcis->vec1_N,0.);CHKERRQ(ierr);
164206f24817SStefano Zampini           ierr = PetscMalloc1(pcis->n,&vals);CHKERRQ(ierr);
1643c9ed8603SStefano Zampini           ierr = ISGetLocalSize(zerodiag,&nz);CHKERRQ(ierr);
1644c9ed8603SStefano Zampini           ierr = ISGetIndices(zerodiag,&idxs);CHKERRQ(ierr);
164506f24817SStefano Zampini           vals[0] = 1.;
164606f24817SStefano Zampini           ierr = VecSetValues(pcis->vec1_N,1,&idxs[nz-1],vals,INSERT_VALUES);CHKERRQ(ierr);
164706f24817SStefano Zampini           ierr = VecAssemblyBegin(pcis->vec1_N);CHKERRQ(ierr);
164806f24817SStefano Zampini           ierr = VecAssemblyEnd(pcis->vec1_N);CHKERRQ(ierr);
164906f24817SStefano Zampini           /* v_I */
165006f24817SStefano Zampini           ierr = VecSetRandom(pcis->vec2_N,NULL);CHKERRQ(ierr);
165106f24817SStefano Zampini           for (i=0;i<nz;i++) vals[i] = 0.;
165206f24817SStefano Zampini           ierr = VecSetValues(pcis->vec2_N,nz,idxs,vals,INSERT_VALUES);CHKERRQ(ierr);
1653c9ed8603SStefano Zampini           ierr = ISRestoreIndices(zerodiag,&idxs);CHKERRQ(ierr);
165406f24817SStefano Zampini           ierr = ISGetIndices(pcis->is_B_local,&idxs);CHKERRQ(ierr);
165506f24817SStefano Zampini           for (i=0;i<pcis->n_B;i++) vals[i] = 0.;
165606f24817SStefano Zampini           ierr = VecSetValues(pcis->vec2_N,pcis->n_B,idxs,vals,INSERT_VALUES);CHKERRQ(ierr);
165706f24817SStefano Zampini           ierr = ISRestoreIndices(pcis->is_B_local,&idxs);CHKERRQ(ierr);
165806f24817SStefano Zampini           ierr = PCBDDCGraphGetDirichletDofs(pcbddc->mat_graph,&dirIS);CHKERRQ(ierr);
165906f24817SStefano Zampini           if (dirIS) {
166006f24817SStefano Zampini             PetscInt n;
166106f24817SStefano Zampini 
166206f24817SStefano Zampini             ierr = ISGetLocalSize(dirIS,&n);CHKERRQ(ierr);
166306f24817SStefano Zampini             ierr = ISGetIndices(dirIS,&idxs);CHKERRQ(ierr);
166406f24817SStefano Zampini             for (i=0;i<n;i++) vals[i] = 0.;
166506f24817SStefano Zampini             ierr = VecSetValues(pcis->vec2_N,n,idxs,vals,INSERT_VALUES);CHKERRQ(ierr);
166606f24817SStefano Zampini             ierr = ISRestoreIndices(dirIS,&idxs);CHKERRQ(ierr);
166706f24817SStefano Zampini           }
166806f24817SStefano Zampini           ierr = ISDestroy(&dirIS);CHKERRQ(ierr);
166906f24817SStefano Zampini           ierr = VecAssemblyBegin(pcis->vec2_N);CHKERRQ(ierr);
167006f24817SStefano Zampini           ierr = VecAssemblyEnd(pcis->vec2_N);CHKERRQ(ierr);
1671*f1a72664SStefano Zampini           ierr = VecDuplicate(pcis->vec1_N,&vec3_N);CHKERRQ(ierr);
1672*f1a72664SStefano Zampini           ierr = VecSet(vec3_N,0.);CHKERRQ(ierr);
1673*f1a72664SStefano Zampini           ierr = MatMult(pcbddc->local_mat,pcis->vec1_N,vec3_N);CHKERRQ(ierr);
1674*f1a72664SStefano Zampini           ierr = VecDot(vec3_N,pcis->vec2_N,&vals[0]);CHKERRQ(ierr);
1675*f1a72664SStefano Zampini           ierr = VecDestroy(&vec3_N);CHKERRQ(ierr);
167606f24817SStefano Zampini           ierr = PetscPrintf(PETSC_COMM_SELF,"[%d] check new mat: %1.4e\n",PetscGlobalRank,vals[0]);CHKERRQ(ierr);
167706f24817SStefano Zampini           ierr = PetscFree(vals);CHKERRQ(ierr);
167806f24817SStefano Zampini         }
1679*f1a72664SStefano Zampini         /* swap local matrices */
1680*f1a72664SStefano Zampini         ierr = MatISGetLocalMat(pc->pmat,&temp_mat);CHKERRQ(ierr);
1681*f1a72664SStefano Zampini         ierr = PetscObjectReference((PetscObject)temp_mat);CHKERRQ(ierr);
1682*f1a72664SStefano Zampini         ierr = MatISSetLocalMat(pc->pmat,pcbddc->local_mat);CHKERRQ(ierr);
1683*f1a72664SStefano Zampini         ierr = MatDestroy(&pcbddc->local_mat);CHKERRQ(ierr);
1684c263805aSStefano Zampini       }
16855e8657edSStefano Zampini       ierr = PCBDDCComputeLocalMatrix(pc,pcbddc->ChangeOfBasisMatrix);CHKERRQ(ierr);
1686c9ed8603SStefano Zampini       if (zerodiag) {
1687c263805aSStefano Zampini         /* restore original matrix */
1688*f1a72664SStefano Zampini         ierr = MatISSetLocalMat(pc->pmat,temp_mat);CHKERRQ(ierr);
1689*f1a72664SStefano Zampini         ierr = PetscObjectDereference((PetscObject)temp_mat);CHKERRQ(ierr);
1690c263805aSStefano Zampini         /* pop B0 from pcbddc->local_mat */
1691c263805aSStefano Zampini         ierr = PCBDDCBenignPopOrPushB0(pc,PETSC_TRUE);CHKERRQ(ierr);
1692c263805aSStefano Zampini       }
16935e8657edSStefano Zampini       /* get submatrices */
16945e8657edSStefano Zampini       ierr = MatDestroy(&pcis->A_IB);CHKERRQ(ierr);
16955e8657edSStefano Zampini       ierr = MatDestroy(&pcis->A_BI);CHKERRQ(ierr);
16965e8657edSStefano Zampini       ierr = MatDestroy(&pcis->A_BB);CHKERRQ(ierr);
16975e8657edSStefano Zampini       ierr = MatGetSubMatrix(pcbddc->local_mat,pcis->is_B_local,pcis->is_B_local,MAT_INITIAL_MATRIX,&pcis->A_BB);CHKERRQ(ierr);
16985e8657edSStefano Zampini       ierr = MatGetSubMatrix(pcbddc->local_mat,pcis->is_I_local,pcis->is_B_local,MAT_INITIAL_MATRIX,&pcis->A_IB);CHKERRQ(ierr);
16995e8657edSStefano Zampini       ierr = MatGetSubMatrix(pcbddc->local_mat,pcis->is_B_local,pcis->is_I_local,MAT_INITIAL_MATRIX,&pcis->A_BI);CHKERRQ(ierr);
17003975b054SStefano Zampini       /* set flag in pcis to not reuse submatrices during PCISCreate */
17013975b054SStefano Zampini       pcis->reusesubmatrices = PETSC_FALSE;
17025e8657edSStefano Zampini     } else if (!pcbddc->user_ChangeOfBasisMatrix) {
1703b96c3477SStefano Zampini       ierr = MatDestroy(&pcbddc->local_mat);CHKERRQ(ierr);
17045e8657edSStefano Zampini       ierr = PetscObjectReference((PetscObject)matis->A);CHKERRQ(ierr);
17055e8657edSStefano Zampini       pcbddc->local_mat = matis->A;
17065e8657edSStefano Zampini     }
1707b96c3477SStefano Zampini     /* SetUp coarse and local Neumann solvers */
170899cc7994SStefano Zampini     ierr = PCBDDCSetUpSolvers(pc);CHKERRQ(ierr);
1709b96c3477SStefano Zampini     /* SetUp Scaling operator */
1710674ae819SStefano Zampini     ierr = PCBDDCScalingSetUp(pc);CHKERRQ(ierr);
17110c7d97c5SJed Brown   }
171270cf5478SStefano Zampini 
17130369aaf7SStefano Zampini   /* check BDDC operator */
17140369aaf7SStefano Zampini   if (pcbddc->dbg_flag && (
17150369aaf7SStefano Zampini       (pcbddc->n_vertices == pcbddc->local_primal_size) || pcbddc->benign_saddle_point) ) {
17160369aaf7SStefano Zampini     PC_IS          *pcis = (PC_IS*)(pc->data);
17170369aaf7SStefano Zampini     Mat            S_j,B0=NULL,B0_B=NULL;
17180369aaf7SStefano Zampini     Vec            dummy_vec=NULL,vec_check_B,vec_scale_P;
17190369aaf7SStefano Zampini     PetscScalar    norm,p0_check,*array,*array2;
17200369aaf7SStefano Zampini     PetscInt       i;
17210369aaf7SStefano Zampini 
17220369aaf7SStefano Zampini     /* B0 and B0_B */
17230369aaf7SStefano Zampini     if (zerodiag) {
17240369aaf7SStefano Zampini       IS       dummy;
17250369aaf7SStefano Zampini       PetscInt ii[2];
17260369aaf7SStefano Zampini 
17270369aaf7SStefano Zampini       ii[0] = 0;
17280369aaf7SStefano Zampini       ii[1] = pcbddc->B0_ncol;
17290369aaf7SStefano Zampini       ierr = MatCreateSeqAIJWithArrays(PETSC_COMM_SELF,1,pcis->n,ii,pcbddc->B0_cols,pcbddc->B0_vals,&B0);CHKERRQ(ierr);
17300369aaf7SStefano Zampini       ierr = ISCreateStride(PETSC_COMM_SELF,1,0,1,&dummy);CHKERRQ(ierr);
17310369aaf7SStefano Zampini       ierr = MatGetSubMatrix(B0,dummy,pcis->is_B_local,MAT_INITIAL_MATRIX,&B0_B);CHKERRQ(ierr);
17320369aaf7SStefano Zampini       ierr = MatCreateVecs(B0_B,NULL,&dummy_vec);CHKERRQ(ierr);
17330369aaf7SStefano Zampini       ierr = ISDestroy(&dummy);CHKERRQ(ierr);
17340369aaf7SStefano Zampini     }
17350369aaf7SStefano Zampini     /* I need a primal vector to scale primal nodes since BDDC sums contibutions */
17360369aaf7SStefano Zampini     ierr = VecDuplicate(pcbddc->vec1_P,&vec_scale_P);CHKERRQ(ierr);
17370369aaf7SStefano Zampini     ierr = VecSet(pcbddc->vec1_P,1.0);CHKERRQ(ierr);
17380369aaf7SStefano Zampini     ierr = VecScatterBegin(pcbddc->coarse_loc_to_glob,pcbddc->vec1_P,pcbddc->coarse_vec,ADD_VALUES,SCATTER_FORWARD);CHKERRQ(ierr);
17390369aaf7SStefano Zampini     ierr = VecScatterEnd(pcbddc->coarse_loc_to_glob,pcbddc->vec1_P,pcbddc->coarse_vec,ADD_VALUES,SCATTER_FORWARD);CHKERRQ(ierr);
17400369aaf7SStefano Zampini     ierr = VecScatterBegin(pcbddc->coarse_loc_to_glob,pcbddc->coarse_vec,vec_scale_P,INSERT_VALUES,SCATTER_REVERSE);CHKERRQ(ierr);
17410369aaf7SStefano Zampini     ierr = VecScatterEnd(pcbddc->coarse_loc_to_glob,pcbddc->coarse_vec,vec_scale_P,INSERT_VALUES,SCATTER_REVERSE);CHKERRQ(ierr);
17420369aaf7SStefano Zampini     ierr = VecReciprocal(vec_scale_P);CHKERRQ(ierr);
17430369aaf7SStefano Zampini     /* S_j */
17440369aaf7SStefano Zampini     ierr = MatCreateSchurComplement(pcis->A_II,pcis->A_II,pcis->A_IB,pcis->A_BI,pcis->A_BB,&S_j);CHKERRQ(ierr);
17450369aaf7SStefano Zampini     ierr = MatSchurComplementSetKSP(S_j,pcbddc->ksp_D);CHKERRQ(ierr);
17460369aaf7SStefano Zampini 
17470369aaf7SStefano Zampini     /* mimic vector in \widetilde{W}_\Gamma */
17480369aaf7SStefano Zampini     ierr = VecSetRandom(pcis->vec1_N,NULL);CHKERRQ(ierr);
17490369aaf7SStefano Zampini     /* continuous in primal space */
17500369aaf7SStefano Zampini     ierr = VecSetRandom(pcbddc->coarse_vec,NULL);CHKERRQ(ierr);
17510369aaf7SStefano Zampini     ierr = VecScatterBegin(pcbddc->coarse_loc_to_glob,pcbddc->coarse_vec,pcbddc->vec1_P,INSERT_VALUES,SCATTER_REVERSE);CHKERRQ(ierr);
17520369aaf7SStefano Zampini     ierr = VecScatterEnd(pcbddc->coarse_loc_to_glob,pcbddc->coarse_vec,pcbddc->vec1_P,INSERT_VALUES,SCATTER_REVERSE);CHKERRQ(ierr);
17530369aaf7SStefano Zampini     ierr = VecGetArray(pcbddc->vec1_P,&array);CHKERRQ(ierr);
17540369aaf7SStefano Zampini     if (zerodiag) {
17550369aaf7SStefano Zampini       p0_check = array[pcbddc->local_primal_size-1];
17560369aaf7SStefano Zampini     } else {
17570369aaf7SStefano Zampini       p0_check = 0;
17580369aaf7SStefano Zampini     }
17590369aaf7SStefano Zampini     ierr = VecSetValues(pcis->vec1_N,pcbddc->local_primal_size,pcbddc->local_primal_ref_node,array,INSERT_VALUES);CHKERRQ(ierr);
17600369aaf7SStefano Zampini     ierr = VecRestoreArray(pcbddc->vec1_P,&array);CHKERRQ(ierr);
17610369aaf7SStefano Zampini     ierr = VecAssemblyBegin(pcis->vec1_N);CHKERRQ(ierr);
17620369aaf7SStefano Zampini     ierr = VecAssemblyEnd(pcis->vec1_N);CHKERRQ(ierr);
17630369aaf7SStefano Zampini     ierr = VecScatterBegin(pcis->N_to_B,pcis->vec1_N,pcis->vec2_B,INSERT_VALUES,SCATTER_FORWARD);CHKERRQ(ierr);
17640369aaf7SStefano Zampini     ierr = VecScatterEnd(pcis->N_to_B,pcis->vec1_N,pcis->vec2_B,INSERT_VALUES,SCATTER_FORWARD);CHKERRQ(ierr);
17650369aaf7SStefano Zampini     ierr = VecDuplicate(pcis->vec2_B,&vec_check_B);CHKERRQ(ierr);
17660369aaf7SStefano Zampini     ierr = VecCopy(pcis->vec2_B,vec_check_B);CHKERRQ(ierr);
17670369aaf7SStefano Zampini 
17680369aaf7SStefano Zampini     /* assemble rhs for coarse problem */
17690369aaf7SStefano Zampini     /* widetilde{S}_\Gamma w_\Gamma + \widetilde{B0}^T_B p0 */
17700369aaf7SStefano Zampini     /* local with Schur */
17710369aaf7SStefano Zampini     ierr = MatMult(S_j,pcis->vec2_B,pcis->vec1_B);CHKERRQ(ierr);
17720369aaf7SStefano Zampini     if (zerodiag) {
17730369aaf7SStefano Zampini       ierr = VecGetArray(dummy_vec,&array);CHKERRQ(ierr);
17740369aaf7SStefano Zampini       array[0] = p0_check;
17750369aaf7SStefano Zampini       ierr = VecRestoreArray(dummy_vec,&array);CHKERRQ(ierr);
17760369aaf7SStefano Zampini       ierr = MatMultTransposeAdd(B0_B,dummy_vec,pcis->vec1_B,pcis->vec1_B);CHKERRQ(ierr);
17770369aaf7SStefano Zampini     }
17780369aaf7SStefano Zampini     /* sum on primal nodes the local contributions */
17790369aaf7SStefano Zampini     ierr = VecScatterBegin(pcis->N_to_B,pcis->vec1_B,pcis->vec1_N,INSERT_VALUES,SCATTER_REVERSE);CHKERRQ(ierr);
17800369aaf7SStefano Zampini     ierr = VecScatterEnd(pcis->N_to_B,pcis->vec1_B,pcis->vec1_N,INSERT_VALUES,SCATTER_REVERSE);CHKERRQ(ierr);
17810369aaf7SStefano Zampini     ierr = VecGetArray(pcis->vec1_N,&array);CHKERRQ(ierr);
17820369aaf7SStefano Zampini     ierr = VecGetArray(pcbddc->vec1_P,&array2);CHKERRQ(ierr);
17830369aaf7SStefano Zampini     for (i=0;i<pcbddc->local_primal_size;i++) array2[i] = array[pcbddc->local_primal_ref_node[i]];
17840369aaf7SStefano Zampini     ierr = VecRestoreArray(pcbddc->vec1_P,&array2);CHKERRQ(ierr);
17850369aaf7SStefano Zampini     ierr = VecRestoreArray(pcis->vec1_N,&array);CHKERRQ(ierr);
17860369aaf7SStefano Zampini     ierr = VecSet(pcbddc->coarse_vec,0.);CHKERRQ(ierr);
17870369aaf7SStefano Zampini     ierr = VecScatterBegin(pcbddc->coarse_loc_to_glob,pcbddc->vec1_P,pcbddc->coarse_vec,ADD_VALUES,SCATTER_FORWARD);CHKERRQ(ierr);
17880369aaf7SStefano Zampini     ierr = VecScatterEnd(pcbddc->coarse_loc_to_glob,pcbddc->vec1_P,pcbddc->coarse_vec,ADD_VALUES,SCATTER_FORWARD);CHKERRQ(ierr);
17890369aaf7SStefano Zampini     ierr = VecScatterBegin(pcbddc->coarse_loc_to_glob,pcbddc->coarse_vec,pcbddc->vec1_P,INSERT_VALUES,SCATTER_REVERSE);CHKERRQ(ierr);
17900369aaf7SStefano Zampini     ierr = VecScatterEnd(pcbddc->coarse_loc_to_glob,pcbddc->coarse_vec,pcbddc->vec1_P,INSERT_VALUES,SCATTER_REVERSE);CHKERRQ(ierr);
17910369aaf7SStefano Zampini     ierr = VecGetArray(pcbddc->vec1_P,&array);CHKERRQ(ierr);
17920369aaf7SStefano Zampini     /* scale primal nodes (BDDC sums contibutions) */
17930369aaf7SStefano Zampini     ierr = VecPointwiseMult(pcbddc->vec1_P,vec_scale_P,pcbddc->vec1_P);CHKERRQ(ierr);
17940369aaf7SStefano Zampini     ierr = VecSetValues(pcis->vec1_N,pcbddc->local_primal_size,pcbddc->local_primal_ref_node,array,INSERT_VALUES);CHKERRQ(ierr);
17950369aaf7SStefano Zampini     ierr = VecRestoreArray(pcbddc->vec1_P,&array);CHKERRQ(ierr);
17960369aaf7SStefano Zampini     ierr = VecAssemblyBegin(pcis->vec1_N);CHKERRQ(ierr);
17970369aaf7SStefano Zampini     ierr = VecAssemblyEnd(pcis->vec1_N);CHKERRQ(ierr);
17980369aaf7SStefano Zampini     ierr = VecScatterBegin(pcis->N_to_B,pcis->vec1_N,pcis->vec1_B,INSERT_VALUES,SCATTER_FORWARD);CHKERRQ(ierr);
17990369aaf7SStefano Zampini     ierr = VecScatterEnd(pcis->N_to_B,pcis->vec1_N,pcis->vec1_B,INSERT_VALUES,SCATTER_FORWARD);CHKERRQ(ierr);
18000369aaf7SStefano Zampini     /* global: \widetilde{B0}_B w_\Gamma */
18010369aaf7SStefano Zampini     if (zerodiag) {
18020369aaf7SStefano Zampini       ierr = MatMult(B0_B,pcis->vec2_B,dummy_vec);CHKERRQ(ierr);
18030369aaf7SStefano Zampini       ierr = VecGetArray(dummy_vec,&array);CHKERRQ(ierr);
18040369aaf7SStefano Zampini       pcbddc->benign_p0 = array[0];
18050369aaf7SStefano Zampini       ierr = VecRestoreArray(dummy_vec,&array);CHKERRQ(ierr);
18060369aaf7SStefano Zampini     } else {
18070369aaf7SStefano Zampini       pcbddc->benign_p0 = 0.;
18080369aaf7SStefano Zampini     }
18090369aaf7SStefano Zampini     /* BDDC */
18100369aaf7SStefano Zampini     ierr = VecSet(pcis->vec1_D,0.);CHKERRQ(ierr);
18110369aaf7SStefano Zampini     ierr = PCBDDCApplyInterfacePreconditioner(pc,PETSC_FALSE);CHKERRQ(ierr);
18120369aaf7SStefano Zampini 
18130369aaf7SStefano Zampini     ierr = VecCopy(pcis->vec1_B,pcis->vec2_B);CHKERRQ(ierr);
18140369aaf7SStefano Zampini     ierr = VecAXPY(pcis->vec1_B,-1.0,vec_check_B);CHKERRQ(ierr);
18150369aaf7SStefano Zampini     ierr = VecNorm(pcis->vec1_B,NORM_INFINITY,&norm);CHKERRQ(ierr);
18160369aaf7SStefano Zampini     PetscPrintf(PETSC_COMM_SELF,"[%d] BDDC local error is %1.4e\n",PetscGlobalRank,norm);
18170369aaf7SStefano Zampini     if (pcbddc->benign_p0_lidx >= 0) {
18180369aaf7SStefano Zampini       PetscPrintf(PETSC_COMM_SELF,"[%d] BDDC p0 error is %1.4e\n",PetscGlobalRank,PetscAbsScalar(pcbddc->benign_p0-p0_check));
18190369aaf7SStefano Zampini     }
18200369aaf7SStefano Zampini 
18210369aaf7SStefano Zampini     ierr = VecDestroy(&vec_scale_P);CHKERRQ(ierr);
18220369aaf7SStefano Zampini     ierr = VecDestroy(&vec_check_B);CHKERRQ(ierr);
18230369aaf7SStefano Zampini     ierr = VecDestroy(&dummy_vec);CHKERRQ(ierr);
18240369aaf7SStefano Zampini     ierr = MatDestroy(&S_j);CHKERRQ(ierr);
18250369aaf7SStefano Zampini     ierr = MatDestroy(&B0);CHKERRQ(ierr);
18260369aaf7SStefano Zampini     ierr = MatDestroy(&B0_B);CHKERRQ(ierr);
18270369aaf7SStefano Zampini   }
18280369aaf7SStefano Zampini   ierr = ISDestroy(&zerodiag);CHKERRQ(ierr);
18290369aaf7SStefano Zampini 
183058a03d70SStefano Zampini   if (pcbddc->dbg_flag) {
183158a03d70SStefano Zampini     ierr = PetscViewerASCIISubtractTab(pcbddc->dbg_viewer,2*pcbddc->current_level);CHKERRQ(ierr);
18322b510759SStefano Zampini   }
18330c7d97c5SJed Brown   PetscFunctionReturn(0);
18340c7d97c5SJed Brown }
18350c7d97c5SJed Brown 
18360c7d97c5SJed Brown /* -------------------------------------------------------------------------- */
18370c7d97c5SJed Brown /*
183850efa1b5SStefano Zampini    PCApply_BDDC - Applies the BDDC operator to a vector.
18390c7d97c5SJed Brown 
18400c7d97c5SJed Brown    Input Parameters:
18410f202f7eSStefano Zampini +  pc - the preconditioner context
18420f202f7eSStefano Zampini -  r - input vector (global)
18430c7d97c5SJed Brown 
18440c7d97c5SJed Brown    Output Parameter:
18450c7d97c5SJed Brown .  z - output vector (global)
18460c7d97c5SJed Brown 
18470c7d97c5SJed Brown    Application Interface Routine: PCApply()
18480c7d97c5SJed Brown  */
18490c7d97c5SJed Brown #undef __FUNCT__
18500c7d97c5SJed Brown #define __FUNCT__ "PCApply_BDDC"
185153cdbc3dSStefano Zampini PetscErrorCode PCApply_BDDC(PC pc,Vec r,Vec z)
18520c7d97c5SJed Brown {
18530c7d97c5SJed Brown   PC_IS             *pcis = (PC_IS*)(pc->data);
18540c7d97c5SJed Brown   PC_BDDC           *pcbddc = (PC_BDDC*)(pc->data);
1855b097fa66SStefano Zampini   PetscInt          n_B = pcis->n_B, n_D = pcis->n - n_B;
18560c7d97c5SJed Brown   PetscErrorCode    ierr;
18573b03a366Sstefano_zampini   const PetscScalar one = 1.0;
18583b03a366Sstefano_zampini   const PetscScalar m_one = -1.0;
18592617d88aSStefano Zampini   const PetscScalar zero = 0.0;
18600c7d97c5SJed Brown 
18610c7d97c5SJed Brown /* This code is similar to that provided in nn.c for PCNN
18620c7d97c5SJed Brown    NN interface preconditioner changed to BDDC
1863b097fa66SStefano Zampini    Added support for M_3 preconditioner in the reference article (code is active if pcbddc->switch_static == PETSC_TRUE) */
18640c7d97c5SJed Brown 
18650c7d97c5SJed Brown   PetscFunctionBegin;
1866efc2fbd9SStefano Zampini   if (pcbddc->benign_saddle_point) { /* extract p0 from r */
1867efc2fbd9SStefano Zampini     ierr = PCBDDCBenignPopOrPushP0(pc,r,PETSC_TRUE);CHKERRQ(ierr);
1868efc2fbd9SStefano Zampini   }
186985c4d303SStefano Zampini   if (!pcbddc->use_exact_dirichlet_trick) {
1870b097fa66SStefano Zampini     ierr = VecCopy(r,z);CHKERRQ(ierr);
18710c7d97c5SJed Brown     /* First Dirichlet solve */
18720c7d97c5SJed Brown     ierr = VecScatterBegin(pcis->global_to_D,r,pcis->vec1_D,INSERT_VALUES,SCATTER_FORWARD);CHKERRQ(ierr);
18730c7d97c5SJed Brown     ierr = VecScatterEnd(pcis->global_to_D,r,pcis->vec1_D,INSERT_VALUES,SCATTER_FORWARD);CHKERRQ(ierr);
18740c7d97c5SJed Brown     /*
18750c7d97c5SJed Brown       Assembling right hand side for BDDC operator
1876b097fa66SStefano Zampini       - pcis->vec1_D for the Dirichlet part (if needed, i.e. pcbddc->switch_static == PETSC_TRUE)
1877674ae819SStefano Zampini       - pcis->vec1_B the interface part of the global vector z
18780c7d97c5SJed Brown     */
1879b097fa66SStefano Zampini     if (n_D) {
1880b097fa66SStefano Zampini       ierr = KSPSolve(pcbddc->ksp_D,pcis->vec1_D,pcis->vec2_D);CHKERRQ(ierr);
18810c7d97c5SJed Brown       ierr = VecScale(pcis->vec2_D,m_one);CHKERRQ(ierr);
18828eeda7d8SStefano Zampini       if (pcbddc->switch_static) { ierr = MatMultAdd(pcis->A_II,pcis->vec2_D,pcis->vec1_D,pcis->vec1_D);CHKERRQ(ierr); }
1883b097fa66SStefano Zampini       ierr = MatMult(pcis->A_BI,pcis->vec2_D,pcis->vec1_B);CHKERRQ(ierr);
1884b097fa66SStefano Zampini     } else {
1885b097fa66SStefano Zampini       ierr = VecSet(pcis->vec1_B,zero);CHKERRQ(ierr);
1886b097fa66SStefano Zampini     }
18870c7d97c5SJed Brown     ierr = VecScatterBegin(pcis->global_to_B,pcis->vec1_B,z,ADD_VALUES,SCATTER_REVERSE);CHKERRQ(ierr);
18880c7d97c5SJed Brown     ierr = VecScatterEnd(pcis->global_to_B,pcis->vec1_B,z,ADD_VALUES,SCATTER_REVERSE);CHKERRQ(ierr);
1889674ae819SStefano Zampini     ierr = PCBDDCScalingRestriction(pc,z,pcis->vec1_B);CHKERRQ(ierr);
1890b76ba322SStefano Zampini   } else {
1891b097fa66SStefano Zampini     if (pcbddc->switch_static) {
18920bdf917eSStefano Zampini       ierr = VecSet(pcis->vec1_D,zero);CHKERRQ(ierr);
1893b097fa66SStefano Zampini     }
1894674ae819SStefano Zampini     ierr = PCBDDCScalingRestriction(pc,r,pcis->vec1_B);CHKERRQ(ierr);
1895b76ba322SStefano Zampini   }
1896b76ba322SStefano Zampini 
18972617d88aSStefano Zampini   /* Apply interface preconditioner
18982617d88aSStefano Zampini      input/output vecs: pcis->vec1_B and pcis->vec1_D */
1899dc359a40SStefano Zampini   ierr = PCBDDCApplyInterfacePreconditioner(pc,PETSC_FALSE);CHKERRQ(ierr);
19002617d88aSStefano Zampini 
1901674ae819SStefano Zampini   /* Apply transpose of partition of unity operator */
1902674ae819SStefano Zampini   ierr = PCBDDCScalingExtension(pc,pcis->vec1_B,z);CHKERRQ(ierr);
19030c7d97c5SJed Brown 
19043b03a366Sstefano_zampini   /* Second Dirichlet solve and assembling of output */
19050c7d97c5SJed Brown   ierr = VecScatterBegin(pcis->global_to_B,z,pcis->vec1_B,INSERT_VALUES,SCATTER_FORWARD);CHKERRQ(ierr);
19060c7d97c5SJed Brown   ierr = VecScatterEnd(pcis->global_to_B,z,pcis->vec1_B,INSERT_VALUES,SCATTER_FORWARD);CHKERRQ(ierr);
1907b097fa66SStefano Zampini   if (n_B) {
19080c7d97c5SJed Brown     ierr = MatMult(pcis->A_IB,pcis->vec1_B,pcis->vec3_D);CHKERRQ(ierr);
19098eeda7d8SStefano Zampini     if (pcbddc->switch_static) { ierr = MatMultAdd(pcis->A_II,pcis->vec1_D,pcis->vec3_D,pcis->vec3_D);CHKERRQ(ierr); }
1910b097fa66SStefano Zampini   } else if (pcbddc->switch_static) {
1911b097fa66SStefano Zampini     ierr = MatMult(pcis->A_II,pcis->vec1_D,pcis->vec3_D);CHKERRQ(ierr);
1912b097fa66SStefano Zampini   }
1913df187020SStefano Zampini   ierr = KSPSolve(pcbddc->ksp_D,pcis->vec3_D,pcis->vec4_D);CHKERRQ(ierr);
1914efc2fbd9SStefano Zampini 
1915b097fa66SStefano Zampini   if (!pcbddc->use_exact_dirichlet_trick) {
1916b097fa66SStefano Zampini     if (pcbddc->switch_static) {
1917b097fa66SStefano Zampini       ierr = VecAXPBYPCZ(pcis->vec2_D,m_one,one,m_one,pcis->vec4_D,pcis->vec1_D);CHKERRQ(ierr);
1918b097fa66SStefano Zampini     } else {
1919b097fa66SStefano Zampini       ierr = VecAXPBY(pcis->vec2_D,m_one,m_one,pcis->vec4_D);CHKERRQ(ierr);
1920b097fa66SStefano Zampini     }
19210c7d97c5SJed Brown     ierr = VecScatterBegin(pcis->global_to_D,pcis->vec2_D,z,INSERT_VALUES,SCATTER_REVERSE);CHKERRQ(ierr);
19220c7d97c5SJed Brown     ierr = VecScatterEnd(pcis->global_to_D,pcis->vec2_D,z,INSERT_VALUES,SCATTER_REVERSE);CHKERRQ(ierr);
1923b097fa66SStefano Zampini   } else {
1924b097fa66SStefano Zampini     if (pcbddc->switch_static) {
1925b097fa66SStefano Zampini       ierr = VecAXPBY(pcis->vec4_D,one,m_one,pcis->vec1_D);CHKERRQ(ierr);
1926b097fa66SStefano Zampini     } else {
1927b097fa66SStefano Zampini       ierr = VecScale(pcis->vec4_D,m_one);CHKERRQ(ierr);
1928b097fa66SStefano Zampini     }
1929b097fa66SStefano Zampini     ierr = VecScatterBegin(pcis->global_to_D,pcis->vec4_D,z,INSERT_VALUES,SCATTER_REVERSE);CHKERRQ(ierr);
1930b097fa66SStefano Zampini     ierr = VecScatterEnd(pcis->global_to_D,pcis->vec4_D,z,INSERT_VALUES,SCATTER_REVERSE);CHKERRQ(ierr);
1931b097fa66SStefano Zampini   }
1932efc2fbd9SStefano Zampini 
1933efc2fbd9SStefano Zampini   if (pcbddc->benign_saddle_point) { /* push p0 (computed in PCBDDCApplyInterface) */
1934efc2fbd9SStefano Zampini     ierr = PCBDDCBenignPopOrPushP0(pc,z,PETSC_FALSE);CHKERRQ(ierr);
1935efc2fbd9SStefano Zampini   }
19360c7d97c5SJed Brown   PetscFunctionReturn(0);
19370c7d97c5SJed Brown }
193850efa1b5SStefano Zampini 
193950efa1b5SStefano Zampini /* -------------------------------------------------------------------------- */
194050efa1b5SStefano Zampini /*
194150efa1b5SStefano Zampini    PCApplyTranspose_BDDC - Applies the transpose of the BDDC operator to a vector.
194250efa1b5SStefano Zampini 
194350efa1b5SStefano Zampini    Input Parameters:
19440f202f7eSStefano Zampini +  pc - the preconditioner context
19450f202f7eSStefano Zampini -  r - input vector (global)
194650efa1b5SStefano Zampini 
194750efa1b5SStefano Zampini    Output Parameter:
194850efa1b5SStefano Zampini .  z - output vector (global)
194950efa1b5SStefano Zampini 
195050efa1b5SStefano Zampini    Application Interface Routine: PCApplyTranspose()
195150efa1b5SStefano Zampini  */
195250efa1b5SStefano Zampini #undef __FUNCT__
195350efa1b5SStefano Zampini #define __FUNCT__ "PCApplyTranspose_BDDC"
195450efa1b5SStefano Zampini PetscErrorCode PCApplyTranspose_BDDC(PC pc,Vec r,Vec z)
195550efa1b5SStefano Zampini {
195650efa1b5SStefano Zampini   PC_IS             *pcis = (PC_IS*)(pc->data);
195750efa1b5SStefano Zampini   PC_BDDC           *pcbddc = (PC_BDDC*)(pc->data);
1958b097fa66SStefano Zampini   PetscInt          n_B = pcis->n_B, n_D = pcis->n - n_B;
195950efa1b5SStefano Zampini   PetscErrorCode    ierr;
196050efa1b5SStefano Zampini   const PetscScalar one = 1.0;
196150efa1b5SStefano Zampini   const PetscScalar m_one = -1.0;
196250efa1b5SStefano Zampini   const PetscScalar zero = 0.0;
196350efa1b5SStefano Zampini 
196450efa1b5SStefano Zampini   PetscFunctionBegin;
196550efa1b5SStefano Zampini   if (!pcbddc->use_exact_dirichlet_trick) {
1966b097fa66SStefano Zampini     ierr = VecCopy(r,z);CHKERRQ(ierr);
196750efa1b5SStefano Zampini     /* First Dirichlet solve */
196850efa1b5SStefano Zampini     ierr = VecScatterBegin(pcis->global_to_D,r,pcis->vec1_D,INSERT_VALUES,SCATTER_FORWARD);CHKERRQ(ierr);
196950efa1b5SStefano Zampini     ierr = VecScatterEnd(pcis->global_to_D,r,pcis->vec1_D,INSERT_VALUES,SCATTER_FORWARD);CHKERRQ(ierr);
197050efa1b5SStefano Zampini     /*
197150efa1b5SStefano Zampini       Assembling right hand side for BDDC operator
1972b097fa66SStefano Zampini       - pcis->vec1_D for the Dirichlet part (if needed, i.e. pcbddc->switch_static == PETSC_TRUE)
197350efa1b5SStefano Zampini       - pcis->vec1_B the interface part of the global vector z
197450efa1b5SStefano Zampini     */
1975b097fa66SStefano Zampini     if (n_D) {
1976b097fa66SStefano Zampini       ierr = KSPSolveTranspose(pcbddc->ksp_D,pcis->vec1_D,pcis->vec2_D);CHKERRQ(ierr);
197750efa1b5SStefano Zampini       ierr = VecScale(pcis->vec2_D,m_one);CHKERRQ(ierr);
197850efa1b5SStefano Zampini       if (pcbddc->switch_static) { ierr = MatMultTransposeAdd(pcis->A_II,pcis->vec2_D,pcis->vec1_D,pcis->vec1_D);CHKERRQ(ierr); }
1979b097fa66SStefano Zampini       ierr = MatMultTranspose(pcis->A_IB,pcis->vec2_D,pcis->vec1_B);CHKERRQ(ierr);
1980b097fa66SStefano Zampini     } else {
1981b097fa66SStefano Zampini       ierr = VecSet(pcis->vec1_B,zero);CHKERRQ(ierr);
1982b097fa66SStefano Zampini     }
198350efa1b5SStefano Zampini     ierr = VecScatterBegin(pcis->global_to_B,pcis->vec1_B,z,ADD_VALUES,SCATTER_REVERSE);CHKERRQ(ierr);
198450efa1b5SStefano Zampini     ierr = VecScatterEnd(pcis->global_to_B,pcis->vec1_B,z,ADD_VALUES,SCATTER_REVERSE);CHKERRQ(ierr);
198550efa1b5SStefano Zampini     ierr = PCBDDCScalingRestriction(pc,z,pcis->vec1_B);CHKERRQ(ierr);
198650efa1b5SStefano Zampini   } else {
1987b097fa66SStefano Zampini     if (pcbddc->switch_static) {
198850efa1b5SStefano Zampini       ierr = VecSet(pcis->vec1_D,zero);CHKERRQ(ierr);
1989b097fa66SStefano Zampini     }
199050efa1b5SStefano Zampini     ierr = PCBDDCScalingRestriction(pc,r,pcis->vec1_B);CHKERRQ(ierr);
199150efa1b5SStefano Zampini   }
199250efa1b5SStefano Zampini 
199350efa1b5SStefano Zampini   /* Apply interface preconditioner
199450efa1b5SStefano Zampini      input/output vecs: pcis->vec1_B and pcis->vec1_D */
1995dc359a40SStefano Zampini   ierr = PCBDDCApplyInterfacePreconditioner(pc,PETSC_TRUE);CHKERRQ(ierr);
199650efa1b5SStefano Zampini 
199750efa1b5SStefano Zampini   /* Apply transpose of partition of unity operator */
199850efa1b5SStefano Zampini   ierr = PCBDDCScalingExtension(pc,pcis->vec1_B,z);CHKERRQ(ierr);
199950efa1b5SStefano Zampini 
200050efa1b5SStefano Zampini   /* Second Dirichlet solve and assembling of output */
200150efa1b5SStefano Zampini   ierr = VecScatterBegin(pcis->global_to_B,z,pcis->vec1_B,INSERT_VALUES,SCATTER_FORWARD);CHKERRQ(ierr);
200250efa1b5SStefano Zampini   ierr = VecScatterEnd(pcis->global_to_B,z,pcis->vec1_B,INSERT_VALUES,SCATTER_FORWARD);CHKERRQ(ierr);
2003b097fa66SStefano Zampini   if (n_B) {
200450efa1b5SStefano Zampini     ierr = MatMultTranspose(pcis->A_BI,pcis->vec1_B,pcis->vec3_D);CHKERRQ(ierr);
200550efa1b5SStefano Zampini     if (pcbddc->switch_static) { ierr = MatMultTransposeAdd(pcis->A_II,pcis->vec1_D,pcis->vec3_D,pcis->vec3_D);CHKERRQ(ierr); }
2006b097fa66SStefano Zampini   } else if (pcbddc->switch_static) {
2007b097fa66SStefano Zampini     ierr = MatMultTranspose(pcis->A_II,pcis->vec1_D,pcis->vec3_D);CHKERRQ(ierr);
2008b097fa66SStefano Zampini   }
2009b0147a47SStefano Zampini   ierr = KSPSolveTranspose(pcbddc->ksp_D,pcis->vec3_D,pcis->vec4_D);CHKERRQ(ierr);
2010b097fa66SStefano Zampini   if (!pcbddc->use_exact_dirichlet_trick) {
2011b097fa66SStefano Zampini     if (pcbddc->switch_static) {
2012b097fa66SStefano Zampini       ierr = VecAXPBYPCZ(pcis->vec2_D,m_one,one,m_one,pcis->vec4_D,pcis->vec1_D);CHKERRQ(ierr);
2013b097fa66SStefano Zampini     } else {
2014b097fa66SStefano Zampini       ierr = VecAXPBY(pcis->vec2_D,m_one,m_one,pcis->vec4_D);CHKERRQ(ierr);
2015b097fa66SStefano Zampini     }
201650efa1b5SStefano Zampini     ierr = VecScatterBegin(pcis->global_to_D,pcis->vec2_D,z,INSERT_VALUES,SCATTER_REVERSE);CHKERRQ(ierr);
201750efa1b5SStefano Zampini     ierr = VecScatterEnd(pcis->global_to_D,pcis->vec2_D,z,INSERT_VALUES,SCATTER_REVERSE);CHKERRQ(ierr);
2018b097fa66SStefano Zampini   } else {
2019b097fa66SStefano Zampini     if (pcbddc->switch_static) {
2020b097fa66SStefano Zampini       ierr = VecAXPBY(pcis->vec4_D,one,m_one,pcis->vec1_D);CHKERRQ(ierr);
2021b097fa66SStefano Zampini     } else {
2022b097fa66SStefano Zampini       ierr = VecScale(pcis->vec4_D,m_one);CHKERRQ(ierr);
2023b097fa66SStefano Zampini     }
2024b097fa66SStefano Zampini     ierr = VecScatterBegin(pcis->global_to_D,pcis->vec4_D,z,INSERT_VALUES,SCATTER_REVERSE);CHKERRQ(ierr);
2025b097fa66SStefano Zampini     ierr = VecScatterEnd(pcis->global_to_D,pcis->vec4_D,z,INSERT_VALUES,SCATTER_REVERSE);CHKERRQ(ierr);
2026b097fa66SStefano Zampini   }
202750efa1b5SStefano Zampini   PetscFunctionReturn(0);
202850efa1b5SStefano Zampini }
2029da1bb401SStefano Zampini /* -------------------------------------------------------------------------- */
2030674ae819SStefano Zampini 
2031da1bb401SStefano Zampini #undef __FUNCT__
2032da1bb401SStefano Zampini #define __FUNCT__ "PCDestroy_BDDC"
2033da1bb401SStefano Zampini PetscErrorCode PCDestroy_BDDC(PC pc)
2034da1bb401SStefano Zampini {
2035da1bb401SStefano Zampini   PC_BDDC        *pcbddc = (PC_BDDC*)pc->data;
2036da1bb401SStefano Zampini   PetscErrorCode ierr;
2037da1bb401SStefano Zampini 
2038da1bb401SStefano Zampini   PetscFunctionBegin;
2039da1bb401SStefano Zampini   /* free data created by PCIS */
2040da1bb401SStefano Zampini   ierr = PCISDestroy(pc);CHKERRQ(ierr);
2041674ae819SStefano Zampini   /* free BDDC custom data  */
2042674ae819SStefano Zampini   ierr = PCBDDCResetCustomization(pc);CHKERRQ(ierr);
2043674ae819SStefano Zampini   /* destroy objects related to topography */
2044674ae819SStefano Zampini   ierr = PCBDDCResetTopography(pc);CHKERRQ(ierr);
2045674ae819SStefano Zampini   /* free allocated graph structure */
2046da1bb401SStefano Zampini   ierr = PetscFree(pcbddc->mat_graph);CHKERRQ(ierr);
2047b96c3477SStefano Zampini   /* free allocated sub schurs structure */
2048b96c3477SStefano Zampini   ierr = PetscFree(pcbddc->sub_schurs);CHKERRQ(ierr);
204934a97f8cSStefano Zampini   /* destroy objects for scaling operator */
2050674ae819SStefano Zampini   ierr = PCBDDCScalingDestroy(pc);CHKERRQ(ierr);
205134a97f8cSStefano Zampini   ierr = PetscFree(pcbddc->deluxe_ctx);CHKERRQ(ierr);
2052674ae819SStefano Zampini   /* free solvers stuff */
2053674ae819SStefano Zampini   ierr = PCBDDCResetSolvers(pc);CHKERRQ(ierr);
205462a6ff1dSStefano Zampini   /* free global vectors needed in presolve */
205562a6ff1dSStefano Zampini   ierr = VecDestroy(&pcbddc->temp_solution);CHKERRQ(ierr);
205662a6ff1dSStefano Zampini   ierr = VecDestroy(&pcbddc->original_rhs);CHKERRQ(ierr);
2057906d46d4SStefano Zampini   /* free stuff for change of basis hooks */
2058906d46d4SStefano Zampini   if (pcbddc->new_global_mat) {
2059906d46d4SStefano Zampini     PCBDDCChange_ctx change_ctx;
2060906d46d4SStefano Zampini     ierr = MatShellGetContext(pcbddc->new_global_mat,&change_ctx);CHKERRQ(ierr);
2061906d46d4SStefano Zampini     ierr = MatDestroy(&change_ctx->original_mat);CHKERRQ(ierr);
2062906d46d4SStefano Zampini     ierr = MatDestroy(&change_ctx->global_change);CHKERRQ(ierr);
2063906d46d4SStefano Zampini     ierr = VecDestroyVecs(2,&change_ctx->work);CHKERRQ(ierr);
2064906d46d4SStefano Zampini     ierr = PetscFree(change_ctx);CHKERRQ(ierr);
2065906d46d4SStefano Zampini   }
2066906d46d4SStefano Zampini   ierr = MatDestroy(&pcbddc->new_global_mat);CHKERRQ(ierr);
20673425bc38SStefano Zampini   /* remove functions */
2068906d46d4SStefano Zampini   ierr = PetscObjectComposeFunction((PetscObject)pc,"PCBDDCSetChangeOfBasisMat_C",NULL);CHKERRQ(ierr);
2069674ae819SStefano Zampini   ierr = PetscObjectComposeFunction((PetscObject)pc,"PCBDDCSetPrimalVerticesLocalIS_C",NULL);CHKERRQ(ierr);
2070bdf89e91SBarry Smith   ierr = PetscObjectComposeFunction((PetscObject)pc,"PCBDDCSetCoarseningRatio_C",NULL);CHKERRQ(ierr);
20712b510759SStefano Zampini   ierr = PetscObjectComposeFunction((PetscObject)pc,"PCBDDCSetLevel_C",NULL);CHKERRQ(ierr);
2072b8ffe317SStefano Zampini   ierr = PetscObjectComposeFunction((PetscObject)pc,"PCBDDCSetUseExactDirichlet_C",NULL);CHKERRQ(ierr);
20732b510759SStefano Zampini   ierr = PetscObjectComposeFunction((PetscObject)pc,"PCBDDCSetLevels_C",NULL);CHKERRQ(ierr);
2074bdf89e91SBarry Smith   ierr = PetscObjectComposeFunction((PetscObject)pc,"PCBDDCSetNullSpace_C",NULL);CHKERRQ(ierr);
2075bdf89e91SBarry Smith   ierr = PetscObjectComposeFunction((PetscObject)pc,"PCBDDCSetDirichletBoundaries_C",NULL);CHKERRQ(ierr);
207682ba6b80SStefano Zampini   ierr = PetscObjectComposeFunction((PetscObject)pc,"PCBDDCSetDirichletBoundariesLocal_C",NULL);CHKERRQ(ierr);
2077bdf89e91SBarry Smith   ierr = PetscObjectComposeFunction((PetscObject)pc,"PCBDDCSetNeumannBoundaries_C",NULL);CHKERRQ(ierr);
207882ba6b80SStefano Zampini   ierr = PetscObjectComposeFunction((PetscObject)pc,"PCBDDCSetNeumannBoundariesLocal_C",NULL);CHKERRQ(ierr);
2079bdf89e91SBarry Smith   ierr = PetscObjectComposeFunction((PetscObject)pc,"PCBDDCGetDirichletBoundaries_C",NULL);CHKERRQ(ierr);
208082ba6b80SStefano Zampini   ierr = PetscObjectComposeFunction((PetscObject)pc,"PCBDDCGetDirichletBoundariesLocal_C",NULL);CHKERRQ(ierr);
2081bdf89e91SBarry Smith   ierr = PetscObjectComposeFunction((PetscObject)pc,"PCBDDCGetNeumannBoundaries_C",NULL);CHKERRQ(ierr);
2082785d1243SStefano Zampini   ierr = PetscObjectComposeFunction((PetscObject)pc,"PCBDDCGetNeumannBoundariesLocal_C",NULL);CHKERRQ(ierr);
2083bdf89e91SBarry Smith   ierr = PetscObjectComposeFunction((PetscObject)pc,"PCBDDCSetDofsSplitting_C",NULL);CHKERRQ(ierr);
208463602bcaSStefano Zampini   ierr = PetscObjectComposeFunction((PetscObject)pc,"PCBDDCSetDofsSplittingLocal_C",NULL);CHKERRQ(ierr);
2085bdf89e91SBarry Smith   ierr = PetscObjectComposeFunction((PetscObject)pc,"PCBDDCSetLocalAdjacencyGraph_C",NULL);CHKERRQ(ierr);
2086bdf89e91SBarry Smith   ierr = PetscObjectComposeFunction((PetscObject)pc,"PCBDDCCreateFETIDPOperators_C",NULL);CHKERRQ(ierr);
2087bdf89e91SBarry Smith   ierr = PetscObjectComposeFunction((PetscObject)pc,"PCBDDCMatFETIDPGetRHS_C",NULL);CHKERRQ(ierr);
2088bdf89e91SBarry Smith   ierr = PetscObjectComposeFunction((PetscObject)pc,"PCBDDCMatFETIDPGetSolution_C",NULL);CHKERRQ(ierr);
2089674ae819SStefano Zampini   /* Free the private data structure */
2090674ae819SStefano Zampini   ierr = PetscFree(pc->data);CHKERRQ(ierr);
2091da1bb401SStefano Zampini   PetscFunctionReturn(0);
2092da1bb401SStefano Zampini }
20933425bc38SStefano Zampini /* -------------------------------------------------------------------------- */
20941e6b0712SBarry Smith 
20953425bc38SStefano Zampini #undef __FUNCT__
20963425bc38SStefano Zampini #define __FUNCT__ "PCBDDCMatFETIDPGetRHS_BDDC"
20973425bc38SStefano Zampini static PetscErrorCode PCBDDCMatFETIDPGetRHS_BDDC(Mat fetidp_mat, Vec standard_rhs, Vec fetidp_flux_rhs)
20983425bc38SStefano Zampini {
2099674ae819SStefano Zampini   FETIDPMat_ctx  mat_ctx;
2100c08af4c6SStefano Zampini   Vec            copy_standard_rhs;
21013425bc38SStefano Zampini   PC_IS*         pcis;
21023425bc38SStefano Zampini   PC_BDDC*       pcbddc;
21033425bc38SStefano Zampini   PetscErrorCode ierr;
21040c7d97c5SJed Brown 
21053425bc38SStefano Zampini   PetscFunctionBegin;
21063425bc38SStefano Zampini   ierr = MatShellGetContext(fetidp_mat,&mat_ctx);CHKERRQ(ierr);
21073425bc38SStefano Zampini   pcis = (PC_IS*)mat_ctx->pc->data;
21083425bc38SStefano Zampini   pcbddc = (PC_BDDC*)mat_ctx->pc->data;
21093425bc38SStefano Zampini 
2110c08af4c6SStefano Zampini   /*
2111c08af4c6SStefano Zampini      change of basis for physical rhs if needed
2112c08af4c6SStefano Zampini      It also changes the rhs in case of dirichlet boundaries
2113c08af4c6SStefano Zampini      TODO: better management when FETIDP will have its own class
2114c08af4c6SStefano Zampini   */
2115c08af4c6SStefano Zampini   ierr = VecDuplicate(standard_rhs,&copy_standard_rhs);CHKERRQ(ierr);
2116c08af4c6SStefano Zampini   ierr = VecCopy(standard_rhs,copy_standard_rhs);CHKERRQ(ierr);
2117c08af4c6SStefano Zampini   ierr = PCPreSolve_BDDC(mat_ctx->pc,NULL,copy_standard_rhs,NULL);CHKERRQ(ierr);
21183425bc38SStefano Zampini   /* store vectors for computation of fetidp final solution */
2119c08af4c6SStefano Zampini   ierr = VecScatterBegin(pcis->global_to_D,copy_standard_rhs,mat_ctx->temp_solution_D,INSERT_VALUES,SCATTER_FORWARD);CHKERRQ(ierr);
2120c08af4c6SStefano Zampini   ierr = VecScatterEnd(pcis->global_to_D,copy_standard_rhs,mat_ctx->temp_solution_D,INSERT_VALUES,SCATTER_FORWARD);CHKERRQ(ierr);
2121fb223d50SStefano Zampini   /* scale rhs since it should be unassembled */
2122fb223d50SStefano Zampini   /* TODO use counter scaling? (also below) */
2123c08af4c6SStefano Zampini   ierr = VecScatterBegin(pcis->global_to_B,copy_standard_rhs,mat_ctx->temp_solution_B,INSERT_VALUES,SCATTER_FORWARD);CHKERRQ(ierr);
2124c08af4c6SStefano Zampini   ierr = VecScatterEnd(pcis->global_to_B,copy_standard_rhs,mat_ctx->temp_solution_B,INSERT_VALUES,SCATTER_FORWARD);CHKERRQ(ierr);
2125674ae819SStefano Zampini   /* Apply partition of unity */
21263425bc38SStefano Zampini   ierr = VecPointwiseMult(mat_ctx->temp_solution_B,pcis->D,mat_ctx->temp_solution_B);CHKERRQ(ierr);
2127c08af4c6SStefano Zampini   /* ierr = PCBDDCScalingRestriction(mat_ctx->pc,copy_standard_rhs,mat_ctx->temp_solution_B);CHKERRQ(ierr); */
21288eeda7d8SStefano Zampini   if (!pcbddc->switch_static) {
21293425bc38SStefano Zampini     /* compute partially subassembled Schur complement right-hand side */
21303425bc38SStefano Zampini     ierr = KSPSolve(pcbddc->ksp_D,mat_ctx->temp_solution_D,pcis->vec1_D);CHKERRQ(ierr);
21313425bc38SStefano Zampini     ierr = MatMult(pcis->A_BI,pcis->vec1_D,pcis->vec1_B);CHKERRQ(ierr);
21323425bc38SStefano Zampini     ierr = VecAXPY(mat_ctx->temp_solution_B,-1.0,pcis->vec1_B);CHKERRQ(ierr);
2133c08af4c6SStefano Zampini     ierr = VecSet(copy_standard_rhs,0.0);CHKERRQ(ierr);
2134c08af4c6SStefano Zampini     ierr = VecScatterBegin(pcis->global_to_B,mat_ctx->temp_solution_B,copy_standard_rhs,ADD_VALUES,SCATTER_REVERSE);CHKERRQ(ierr);
2135c08af4c6SStefano Zampini     ierr = VecScatterEnd(pcis->global_to_B,mat_ctx->temp_solution_B,copy_standard_rhs,ADD_VALUES,SCATTER_REVERSE);CHKERRQ(ierr);
2136c08af4c6SStefano Zampini     /* ierr = PCBDDCScalingRestriction(mat_ctx->pc,copy_standard_rhs,mat_ctx->temp_solution_B);CHKERRQ(ierr); */
2137c08af4c6SStefano Zampini     ierr = VecScatterBegin(pcis->global_to_B,copy_standard_rhs,mat_ctx->temp_solution_B,INSERT_VALUES,SCATTER_FORWARD);CHKERRQ(ierr);
2138c08af4c6SStefano Zampini     ierr = VecScatterEnd(pcis->global_to_B,copy_standard_rhs,mat_ctx->temp_solution_B,INSERT_VALUES,SCATTER_FORWARD);CHKERRQ(ierr);
21393425bc38SStefano Zampini     ierr = VecPointwiseMult(mat_ctx->temp_solution_B,pcis->D,mat_ctx->temp_solution_B);CHKERRQ(ierr);
21403425bc38SStefano Zampini   }
2141c08af4c6SStefano Zampini   ierr = VecDestroy(&copy_standard_rhs);CHKERRQ(ierr);
21423425bc38SStefano Zampini   /* BDDC rhs */
21433425bc38SStefano Zampini   ierr = VecCopy(mat_ctx->temp_solution_B,pcis->vec1_B);CHKERRQ(ierr);
21448eeda7d8SStefano Zampini   if (pcbddc->switch_static) {
21453425bc38SStefano Zampini     ierr = VecCopy(mat_ctx->temp_solution_D,pcis->vec1_D);CHKERRQ(ierr);
21463425bc38SStefano Zampini   }
21473425bc38SStefano Zampini   /* apply BDDC */
2148dc359a40SStefano Zampini   ierr = PCBDDCApplyInterfacePreconditioner(mat_ctx->pc,PETSC_FALSE);CHKERRQ(ierr);
21493425bc38SStefano Zampini   /* Application of B_delta and assembling of rhs for fetidp fluxes */
21503425bc38SStefano Zampini   ierr = VecSet(fetidp_flux_rhs,0.0);CHKERRQ(ierr);
21513425bc38SStefano Zampini   ierr = MatMult(mat_ctx->B_delta,pcis->vec1_B,mat_ctx->lambda_local);CHKERRQ(ierr);
21523425bc38SStefano Zampini   ierr = VecScatterBegin(mat_ctx->l2g_lambda,mat_ctx->lambda_local,fetidp_flux_rhs,ADD_VALUES,SCATTER_FORWARD);CHKERRQ(ierr);
21533425bc38SStefano Zampini   ierr = VecScatterEnd(mat_ctx->l2g_lambda,mat_ctx->lambda_local,fetidp_flux_rhs,ADD_VALUES,SCATTER_FORWARD);CHKERRQ(ierr);
21543425bc38SStefano Zampini   PetscFunctionReturn(0);
21553425bc38SStefano Zampini }
21561e6b0712SBarry Smith 
21573425bc38SStefano Zampini #undef __FUNCT__
21583425bc38SStefano Zampini #define __FUNCT__ "PCBDDCMatFETIDPGetRHS"
21593425bc38SStefano Zampini /*@
21600f202f7eSStefano Zampini  PCBDDCMatFETIDPGetRHS - Compute the right-hand side for FETI-DP linear system using the physical right-hand side
21613425bc38SStefano Zampini 
21623425bc38SStefano Zampini    Collective
21633425bc38SStefano Zampini 
21643425bc38SStefano Zampini    Input Parameters:
21650f202f7eSStefano Zampini +  fetidp_mat      - the FETI-DP matrix object obtained by a call to PCBDDCCreateFETIDPOperators
21660f202f7eSStefano Zampini -  standard_rhs    - the right-hand side of the original linear system
21673425bc38SStefano Zampini 
21683425bc38SStefano Zampini    Output Parameters:
21690f202f7eSStefano Zampini .  fetidp_flux_rhs - the right-hand side for the FETI-DP linear system
21703425bc38SStefano Zampini 
21713425bc38SStefano Zampini    Level: developer
21723425bc38SStefano Zampini 
21733425bc38SStefano Zampini    Notes:
21743425bc38SStefano Zampini 
21750f202f7eSStefano Zampini .seealso: PCBDDC, PCBDDCCreateFETIDPOperators, PCBDDCMatFETIDPGetSolution
21763425bc38SStefano Zampini @*/
21773425bc38SStefano Zampini PetscErrorCode PCBDDCMatFETIDPGetRHS(Mat fetidp_mat, Vec standard_rhs, Vec fetidp_flux_rhs)
21783425bc38SStefano Zampini {
2179674ae819SStefano Zampini   FETIDPMat_ctx  mat_ctx;
21803425bc38SStefano Zampini   PetscErrorCode ierr;
21813425bc38SStefano Zampini 
21823425bc38SStefano Zampini   PetscFunctionBegin;
21833425bc38SStefano Zampini   ierr = MatShellGetContext(fetidp_mat,&mat_ctx);CHKERRQ(ierr);
21843425bc38SStefano Zampini   ierr = PetscTryMethod(mat_ctx->pc,"PCBDDCMatFETIDPGetRHS_C",(Mat,Vec,Vec),(fetidp_mat,standard_rhs,fetidp_flux_rhs));CHKERRQ(ierr);
21853425bc38SStefano Zampini   PetscFunctionReturn(0);
21863425bc38SStefano Zampini }
21873425bc38SStefano Zampini /* -------------------------------------------------------------------------- */
21881e6b0712SBarry Smith 
21893425bc38SStefano Zampini #undef __FUNCT__
21903425bc38SStefano Zampini #define __FUNCT__ "PCBDDCMatFETIDPGetSolution_BDDC"
21913425bc38SStefano Zampini static PetscErrorCode PCBDDCMatFETIDPGetSolution_BDDC(Mat fetidp_mat, Vec fetidp_flux_sol, Vec standard_sol)
21923425bc38SStefano Zampini {
2193674ae819SStefano Zampini   FETIDPMat_ctx  mat_ctx;
21943425bc38SStefano Zampini   PC_IS*         pcis;
21953425bc38SStefano Zampini   PC_BDDC*       pcbddc;
21963425bc38SStefano Zampini   PetscErrorCode ierr;
21973425bc38SStefano Zampini 
21983425bc38SStefano Zampini   PetscFunctionBegin;
21993425bc38SStefano Zampini   ierr = MatShellGetContext(fetidp_mat,&mat_ctx);CHKERRQ(ierr);
22003425bc38SStefano Zampini   pcis = (PC_IS*)mat_ctx->pc->data;
22013425bc38SStefano Zampini   pcbddc = (PC_BDDC*)mat_ctx->pc->data;
22023425bc38SStefano Zampini 
22033425bc38SStefano Zampini   /* apply B_delta^T */
22043425bc38SStefano Zampini   ierr = VecScatterBegin(mat_ctx->l2g_lambda,fetidp_flux_sol,mat_ctx->lambda_local,INSERT_VALUES,SCATTER_REVERSE);CHKERRQ(ierr);
22053425bc38SStefano Zampini   ierr = VecScatterEnd  (mat_ctx->l2g_lambda,fetidp_flux_sol,mat_ctx->lambda_local,INSERT_VALUES,SCATTER_REVERSE);CHKERRQ(ierr);
22063425bc38SStefano Zampini   ierr = MatMultTranspose(mat_ctx->B_delta,mat_ctx->lambda_local,pcis->vec1_B);CHKERRQ(ierr);
22073425bc38SStefano Zampini   /* compute rhs for BDDC application */
22083425bc38SStefano Zampini   ierr = VecAYPX(pcis->vec1_B,-1.0,mat_ctx->temp_solution_B);CHKERRQ(ierr);
22098eeda7d8SStefano Zampini   if (pcbddc->switch_static) {
22103425bc38SStefano Zampini     ierr = VecCopy(mat_ctx->temp_solution_D,pcis->vec1_D);CHKERRQ(ierr);
22113425bc38SStefano Zampini   }
22123425bc38SStefano Zampini   /* apply BDDC */
2213dc359a40SStefano Zampini   ierr = PCBDDCApplyInterfacePreconditioner(mat_ctx->pc,PETSC_FALSE);CHKERRQ(ierr);
22143425bc38SStefano Zampini   /* put values into standard global vector */
22153425bc38SStefano Zampini   ierr = VecScatterBegin(pcis->global_to_B,pcis->vec1_B,standard_sol,INSERT_VALUES,SCATTER_REVERSE);CHKERRQ(ierr);
22163425bc38SStefano Zampini   ierr = VecScatterEnd  (pcis->global_to_B,pcis->vec1_B,standard_sol,INSERT_VALUES,SCATTER_REVERSE);CHKERRQ(ierr);
22178eeda7d8SStefano Zampini   if (!pcbddc->switch_static) {
22183425bc38SStefano Zampini     /* compute values into the interior if solved for the partially subassembled Schur complement */
22193425bc38SStefano Zampini     ierr = MatMult(pcis->A_IB,pcis->vec1_B,pcis->vec1_D);CHKERRQ(ierr);
22203425bc38SStefano Zampini     ierr = VecAXPY(mat_ctx->temp_solution_D,-1.0,pcis->vec1_D);CHKERRQ(ierr);
22213425bc38SStefano Zampini     ierr = KSPSolve(pcbddc->ksp_D,mat_ctx->temp_solution_D,pcis->vec1_D);CHKERRQ(ierr);
22223425bc38SStefano Zampini   }
22233425bc38SStefano Zampini   ierr = VecScatterBegin(pcis->global_to_D,pcis->vec1_D,standard_sol,INSERT_VALUES,SCATTER_REVERSE);CHKERRQ(ierr);
22243425bc38SStefano Zampini   ierr = VecScatterEnd  (pcis->global_to_D,pcis->vec1_D,standard_sol,INSERT_VALUES,SCATTER_REVERSE);CHKERRQ(ierr);
22253425bc38SStefano Zampini   /* final change of basis if needed
22263425bc38SStefano Zampini      Is also sums the dirichlet part removed during RHS assembling */
22273308cffdSStefano Zampini   ierr = PCPostSolve_BDDC(mat_ctx->pc,NULL,NULL,standard_sol);CHKERRQ(ierr);
22283425bc38SStefano Zampini   PetscFunctionReturn(0);
22293425bc38SStefano Zampini }
22301e6b0712SBarry Smith 
22313425bc38SStefano Zampini #undef __FUNCT__
22323425bc38SStefano Zampini #define __FUNCT__ "PCBDDCMatFETIDPGetSolution"
22333425bc38SStefano Zampini /*@
22340f202f7eSStefano Zampini  PCBDDCMatFETIDPGetSolution - Compute the physical solution using the solution of the FETI-DP linear system
22353425bc38SStefano Zampini 
22363425bc38SStefano Zampini    Collective
22373425bc38SStefano Zampini 
22383425bc38SStefano Zampini    Input Parameters:
22390f202f7eSStefano Zampini +  fetidp_mat      - the FETI-DP matrix obtained by a call to PCBDDCCreateFETIDPOperators
22400f202f7eSStefano Zampini -  fetidp_flux_sol - the solution of the FETI-DP linear system
22413425bc38SStefano Zampini 
22423425bc38SStefano Zampini    Output Parameters:
22430f202f7eSStefano Zampini .  standard_sol    - the solution defined on the physical domain
22443425bc38SStefano Zampini 
22453425bc38SStefano Zampini    Level: developer
22463425bc38SStefano Zampini 
22473425bc38SStefano Zampini    Notes:
22483425bc38SStefano Zampini 
22490f202f7eSStefano Zampini .seealso: PCBDDC, PCBDDCCreateFETIDPOperators, PCBDDCMatFETIDPGetRHS
22503425bc38SStefano Zampini @*/
22513425bc38SStefano Zampini PetscErrorCode PCBDDCMatFETIDPGetSolution(Mat fetidp_mat, Vec fetidp_flux_sol, Vec standard_sol)
22523425bc38SStefano Zampini {
2253674ae819SStefano Zampini   FETIDPMat_ctx  mat_ctx;
22543425bc38SStefano Zampini   PetscErrorCode ierr;
22553425bc38SStefano Zampini 
22563425bc38SStefano Zampini   PetscFunctionBegin;
22573425bc38SStefano Zampini   ierr = MatShellGetContext(fetidp_mat,&mat_ctx);CHKERRQ(ierr);
22583425bc38SStefano Zampini   ierr = PetscTryMethod(mat_ctx->pc,"PCBDDCMatFETIDPGetSolution_C",(Mat,Vec,Vec),(fetidp_mat,fetidp_flux_sol,standard_sol));CHKERRQ(ierr);
22593425bc38SStefano Zampini   PetscFunctionReturn(0);
22603425bc38SStefano Zampini }
22613425bc38SStefano Zampini /* -------------------------------------------------------------------------- */
22621e6b0712SBarry Smith 
2263f23aa3ddSBarry Smith extern PetscErrorCode FETIDPMatMult(Mat,Vec,Vec);
2264edf7251bSStefano Zampini extern PetscErrorCode FETIDPMatMultTranspose(Mat,Vec,Vec);
2265f23aa3ddSBarry Smith extern PetscErrorCode PCBDDCDestroyFETIDPMat(Mat);
2266f23aa3ddSBarry Smith extern PetscErrorCode FETIDPPCApply(PC,Vec,Vec);
2267edf7251bSStefano Zampini extern PetscErrorCode FETIDPPCApplyTranspose(PC,Vec,Vec);
2268f23aa3ddSBarry Smith extern PetscErrorCode PCBDDCDestroyFETIDPPC(PC);
2269674ae819SStefano Zampini 
22703425bc38SStefano Zampini #undef __FUNCT__
22713425bc38SStefano Zampini #define __FUNCT__ "PCBDDCCreateFETIDPOperators_BDDC"
22723425bc38SStefano Zampini static PetscErrorCode PCBDDCCreateFETIDPOperators_BDDC(PC pc, Mat *fetidp_mat, PC *fetidp_pc)
22733425bc38SStefano Zampini {
2274674ae819SStefano Zampini 
2275674ae819SStefano Zampini   FETIDPMat_ctx  fetidpmat_ctx;
22763425bc38SStefano Zampini   Mat            newmat;
2277674ae819SStefano Zampini   FETIDPPC_ctx   fetidppc_ctx;
22783425bc38SStefano Zampini   PC             newpc;
2279ce94432eSBarry Smith   MPI_Comm       comm;
22803425bc38SStefano Zampini   PetscErrorCode ierr;
22813425bc38SStefano Zampini 
22823425bc38SStefano Zampini   PetscFunctionBegin;
2283ce94432eSBarry Smith   ierr = PetscObjectGetComm((PetscObject)pc,&comm);CHKERRQ(ierr);
22843425bc38SStefano Zampini   /* FETIDP linear matrix */
22853425bc38SStefano Zampini   ierr = PCBDDCCreateFETIDPMatContext(pc,&fetidpmat_ctx);CHKERRQ(ierr);
22863425bc38SStefano Zampini   ierr = PCBDDCSetupFETIDPMatContext(fetidpmat_ctx);CHKERRQ(ierr);
22873425bc38SStefano Zampini   ierr = MatCreateShell(comm,PETSC_DECIDE,PETSC_DECIDE,fetidpmat_ctx->n_lambda,fetidpmat_ctx->n_lambda,fetidpmat_ctx,&newmat);CHKERRQ(ierr);
22883425bc38SStefano Zampini   ierr = MatShellSetOperation(newmat,MATOP_MULT,(void (*)(void))FETIDPMatMult);CHKERRQ(ierr);
2289edf7251bSStefano Zampini   ierr = MatShellSetOperation(newmat,MATOP_MULT_TRANSPOSE,(void (*)(void))FETIDPMatMultTranspose);CHKERRQ(ierr);
22903425bc38SStefano Zampini   ierr = MatShellSetOperation(newmat,MATOP_DESTROY,(void (*)(void))PCBDDCDestroyFETIDPMat);CHKERRQ(ierr);
22913425bc38SStefano Zampini   ierr = MatSetUp(newmat);CHKERRQ(ierr);
22923425bc38SStefano Zampini   /* FETIDP preconditioner */
22933425bc38SStefano Zampini   ierr = PCBDDCCreateFETIDPPCContext(pc,&fetidppc_ctx);CHKERRQ(ierr);
22943425bc38SStefano Zampini   ierr = PCBDDCSetupFETIDPPCContext(newmat,fetidppc_ctx);CHKERRQ(ierr);
22953425bc38SStefano Zampini   ierr = PCCreate(comm,&newpc);CHKERRQ(ierr);
22963425bc38SStefano Zampini   ierr = PCSetType(newpc,PCSHELL);CHKERRQ(ierr);
22973425bc38SStefano Zampini   ierr = PCShellSetContext(newpc,fetidppc_ctx);CHKERRQ(ierr);
22983425bc38SStefano Zampini   ierr = PCShellSetApply(newpc,FETIDPPCApply);CHKERRQ(ierr);
2299edf7251bSStefano Zampini   ierr = PCShellSetApplyTranspose(newpc,FETIDPPCApplyTranspose);CHKERRQ(ierr);
23003425bc38SStefano Zampini   ierr = PCShellSetDestroy(newpc,PCBDDCDestroyFETIDPPC);CHKERRQ(ierr);
230123ee1639SBarry Smith   ierr = PCSetOperators(newpc,newmat,newmat);CHKERRQ(ierr);
23023425bc38SStefano Zampini   ierr = PCSetUp(newpc);CHKERRQ(ierr);
23033425bc38SStefano Zampini   /* return pointers for objects created */
23043425bc38SStefano Zampini   *fetidp_mat=newmat;
23053425bc38SStefano Zampini   *fetidp_pc=newpc;
23063425bc38SStefano Zampini   PetscFunctionReturn(0);
23073425bc38SStefano Zampini }
23081e6b0712SBarry Smith 
23093425bc38SStefano Zampini #undef __FUNCT__
23103425bc38SStefano Zampini #define __FUNCT__ "PCBDDCCreateFETIDPOperators"
23113425bc38SStefano Zampini /*@
23120f202f7eSStefano Zampini  PCBDDCCreateFETIDPOperators - Create FETI-DP operators
23133425bc38SStefano Zampini 
23143425bc38SStefano Zampini    Collective
23153425bc38SStefano Zampini 
23163425bc38SStefano Zampini    Input Parameters:
23170f202f7eSStefano Zampini .  pc - the BDDC preconditioning context (setup should have been called before)
231828509bceSStefano Zampini 
231928509bceSStefano Zampini    Output Parameters:
23200f202f7eSStefano Zampini +  fetidp_mat - shell FETI-DP matrix object
23210f202f7eSStefano Zampini -  fetidp_pc  - shell Dirichlet preconditioner for FETI-DP matrix
232228509bceSStefano Zampini 
232328509bceSStefano Zampini    Options Database Keys:
23240f202f7eSStefano Zampini .    -fetidp_fullyredundant <false> - use or not a fully redundant set of Lagrange multipliers
23253425bc38SStefano Zampini 
23263425bc38SStefano Zampini    Level: developer
23273425bc38SStefano Zampini 
23283425bc38SStefano Zampini    Notes:
23290f202f7eSStefano Zampini      Currently the only operations provided for FETI-DP matrix are MatMult and MatMultTranspose
23303425bc38SStefano Zampini 
23310f202f7eSStefano Zampini .seealso: PCBDDC, PCBDDCMatFETIDPGetRHS, PCBDDCMatFETIDPGetSolution
23323425bc38SStefano Zampini @*/
23333425bc38SStefano Zampini PetscErrorCode PCBDDCCreateFETIDPOperators(PC pc, Mat *fetidp_mat, PC *fetidp_pc)
23343425bc38SStefano Zampini {
23353425bc38SStefano Zampini   PetscErrorCode ierr;
23363425bc38SStefano Zampini 
23373425bc38SStefano Zampini   PetscFunctionBegin;
23383425bc38SStefano Zampini   PetscValidHeaderSpecific(pc,PC_CLASSID,1);
23393425bc38SStefano Zampini   if (pc->setupcalled) {
2340516d51a7SStefano Zampini     ierr = PetscUseMethod(pc,"PCBDDCCreateFETIDPOperators_C",(PC,Mat*,PC*),(pc,fetidp_mat,fetidp_pc));CHKERRQ(ierr);
2341f23aa3ddSBarry Smith   } else SETERRQ(PETSC_COMM_SELF,PETSC_ERR_SUP,"You must call PCSetup_BDDC() first \n");
23423425bc38SStefano Zampini   PetscFunctionReturn(0);
23433425bc38SStefano Zampini }
23440c7d97c5SJed Brown /* -------------------------------------------------------------------------- */
2345da1bb401SStefano Zampini /*MC
2346da1bb401SStefano Zampini    PCBDDC - Balancing Domain Decomposition by Constraints.
23470c7d97c5SJed Brown 
234828509bceSStefano Zampini    An implementation of the BDDC preconditioner based on
234928509bceSStefano Zampini 
235028509bceSStefano Zampini .vb
235128509bceSStefano Zampini    [1] C. R. Dohrmann. "An approximate BDDC preconditioner", Numerical Linear Algebra with Applications Volume 14, Issue 2, pages 149-168, March 2007
235228509bceSStefano Zampini    [2] A. Klawonn and O. B. Widlund. "Dual-Primal FETI Methods for Linear Elasticity", http://cs.nyu.edu/csweb/Research/TechReports/TR2004-855/TR2004-855.pdf
235328509bceSStefano Zampini    [3] J. Mandel, B. Sousedik, C. R. Dohrmann. "Multispace and Multilevel BDDC", http://arxiv.org/abs/0712.3977
23540f202f7eSStefano Zampini    [4] C. Pechstein and C. R. Dohrmann. "Modern domain decomposition methods BDDC, deluxe scaling, and an algebraic approach", Seminar talk, Linz, December 2013, http://people.ricam.oeaw.ac.at/c.pechstein/pechstein-bddc2013.pdf
235528509bceSStefano Zampini .ve
235628509bceSStefano Zampini 
235728509bceSStefano Zampini    The matrix to be preconditioned (Pmat) must be of type MATIS.
235828509bceSStefano Zampini 
23590f202f7eSStefano Zampini    Currently works with MATIS matrices with local matrices of type MATSEQAIJ, MATSEQBAIJ or MATSEQSBAIJ, either with real or complex numbers.
236028509bceSStefano Zampini 
236128509bceSStefano Zampini    It also works with unsymmetric and indefinite problems.
236228509bceSStefano Zampini 
2363b6fdb6dfSStefano Zampini    Unlike 'conventional' interface preconditioners, PCBDDC iterates over all degrees of freedom, not just those on the interface. This allows the use of approximate solvers on the subdomains.
2364b6fdb6dfSStefano Zampini 
23650f202f7eSStefano Zampini    Approximate local solvers are automatically adapted for singular linear problems (see [1]) if the user has provided the nullspace using PCBDDCSetNullSpace()
236628509bceSStefano Zampini 
23670f202f7eSStefano Zampini    Boundary nodes are split in vertices, edges and faces classes using information from the local to global mapping of dofs and the local connectivity graph of nodes. The latter can be customized by using PCBDDCSetLocalAdjacencyGraph()
23680f202f7eSStefano Zampini    Additional information on dofs can be provided by using PCBDDCSetDofsSplitting(), PCBDDCSetDirichletBoundaries(), PCBDDCSetNeumannBoundaries(), and PCBDDCSetPrimalVerticesLocalIS()
236928509bceSStefano Zampini 
23700f202f7eSStefano Zampini    Constraints can be customized by attaching a MatNullSpace object to the MATIS matrix via MatSetNearNullSpace(). Non-singular modes are retained via SVD.
237128509bceSStefano Zampini 
23720f202f7eSStefano Zampini    Change of basis is performed similarly to [2] when requested. When more than one constraint is present on a single connected component (i.e. an edge or a face), a robust method based on local QR factorizations is used.
23730f202f7eSStefano Zampini    User defined change of basis can be passed to PCBDDC by using PCBDDCSetChangeOfBasisMat()
237428509bceSStefano Zampini 
23750f202f7eSStefano Zampini    The PETSc implementation also supports multilevel BDDC [3]. Coarse grids are partitioned using a MatPartitioning object.
237628509bceSStefano Zampini 
23770f202f7eSStefano Zampini    Adaptive selection of primal constraints [4] is supported for SPD systems with high-contrast in the coefficients if MUMPS is present. Future versions of the code will also consider using MKL_PARDISO or PASTIX.
237828509bceSStefano Zampini 
23790f202f7eSStefano Zampini    An experimental interface to the FETI-DP method is available. FETI-DP operators could be created using PCBDDCCreateFETIDPOperators(). A stand-alone class for the FETI-DP method will be provided in the next releases.
23800f202f7eSStefano Zampini    Deluxe scaling is not supported yet for FETI-DP.
23810f202f7eSStefano Zampini 
23820f202f7eSStefano Zampini    Options Database Keys (some of them, run with -h for a complete list):
23830f202f7eSStefano Zampini 
23840f202f7eSStefano Zampini .    -pc_bddc_use_vertices <true> - use or not vertices in primal space
23850f202f7eSStefano Zampini .    -pc_bddc_use_edges <true> - use or not edges in primal space
23860f202f7eSStefano Zampini .    -pc_bddc_use_faces <false> - use or not faces in primal space
23870f202f7eSStefano Zampini .    -pc_bddc_symmetric <true> - symmetric computation of primal basis functions. Specify false for unsymmetric problems
23880f202f7eSStefano Zampini .    -pc_bddc_use_change_of_basis <false> - use change of basis approach (on edges only)
23890f202f7eSStefano Zampini .    -pc_bddc_use_change_on_faces <false> - use change of basis approach on faces if change of basis has been requested
23900f202f7eSStefano Zampini .    -pc_bddc_switch_static <false> - switches from M_2 (default) to M_3 operator (see reference article [1])
239128509bceSStefano Zampini .    -pc_bddc_levels <0> - maximum number of levels for multilevel
23920f202f7eSStefano Zampini .    -pc_bddc_coarsening_ratio <8> - number of subdomains which will be aggregated together at the coarser level (e.g. H/h ratio at the coarser level, significative only in the multilevel case)
23930f202f7eSStefano Zampini .    -pc_bddc_redistribute <0> - size of a subset of processors where the coarse problem will be remapped (the value is ignored if not at the coarsest level)
23940f202f7eSStefano Zampini .    -pc_bddc_use_deluxe_scaling <false> - use deluxe scaling
23950f202f7eSStefano Zampini .    -pc_bddc_schur_layers <-1> - select the economic version of deluxe scaling by specifying the number of layers (-1 corresponds to the original deluxe scaling)
23960f202f7eSStefano Zampini .    -pc_bddc_adaptive_threshold <0.0> - when a value greater than one is specified, adaptive selection of constraints is performed on edges and faces (requires deluxe scaling and MUMPS installed)
239728509bceSStefano Zampini -    -pc_bddc_check_level <0> - set verbosity level of debugging output
239828509bceSStefano Zampini 
239928509bceSStefano Zampini    Options for Dirichlet, Neumann or coarse solver can be set with
240028509bceSStefano Zampini .vb
240128509bceSStefano Zampini       -pc_bddc_dirichlet_
240228509bceSStefano Zampini       -pc_bddc_neumann_
240328509bceSStefano Zampini       -pc_bddc_coarse_
240428509bceSStefano Zampini .ve
24050f202f7eSStefano Zampini    e.g -pc_bddc_dirichlet_ksp_type richardson -pc_bddc_dirichlet_pc_type gamg. PCBDDC uses by default KPSPREONLY and PCLU.
240628509bceSStefano Zampini 
24070f202f7eSStefano Zampini    When using a multilevel approach, solvers' options at the N-th level (N > 1) can be specified as
240828509bceSStefano Zampini .vb
2409312be037SStefano Zampini       -pc_bddc_dirichlet_lN_
2410312be037SStefano Zampini       -pc_bddc_neumann_lN_
2411312be037SStefano Zampini       -pc_bddc_coarse_lN_
241228509bceSStefano Zampini .ve
24130f202f7eSStefano Zampini    Note that level number ranges from the finest (0) to the coarsest (N).
24140f202f7eSStefano Zampini    In order to specify options for the BDDC operators at the coarser levels (and not for the solvers), prepend -pc_bddc_coarse_ or -pc_bddc_coarse_l to the option, e.g.
24150f202f7eSStefano Zampini .vb
24160f202f7eSStefano Zampini      -pc_bddc_coarse_pc_bddc_adaptive_threshold 5 -pc_bddc_coarse_l1_pc_bddc_redistribute 3
24170f202f7eSStefano Zampini .ve
24180f202f7eSStefano Zampini    will use a threshold of 5 for constraints' selection at the first coarse level and will redistribute the coarse problem of the first coarse level on 3 processors
2419da1bb401SStefano Zampini 
2420da1bb401SStefano Zampini    Level: intermediate
2421da1bb401SStefano Zampini 
2422b6fdb6dfSStefano Zampini    Developer notes:
2423da1bb401SStefano Zampini 
2424da1bb401SStefano Zampini    Contributed by Stefano Zampini
2425da1bb401SStefano Zampini 
2426da1bb401SStefano Zampini .seealso:  PCCreate(), PCSetType(), PCType (for list of available types), PC,  MATIS
2427da1bb401SStefano Zampini M*/
2428b2573a8aSBarry Smith 
2429da1bb401SStefano Zampini #undef __FUNCT__
2430da1bb401SStefano Zampini #define __FUNCT__ "PCCreate_BDDC"
24318cc058d9SJed Brown PETSC_EXTERN PetscErrorCode PCCreate_BDDC(PC pc)
2432da1bb401SStefano Zampini {
2433da1bb401SStefano Zampini   PetscErrorCode      ierr;
2434da1bb401SStefano Zampini   PC_BDDC             *pcbddc;
2435da1bb401SStefano Zampini 
2436da1bb401SStefano Zampini   PetscFunctionBegin;
2437da1bb401SStefano Zampini   /* Creates the private data structure for this preconditioner and attach it to the PC object. */
2438b00a9115SJed Brown   ierr      = PetscNewLog(pc,&pcbddc);CHKERRQ(ierr);
2439da1bb401SStefano Zampini   pc->data  = (void*)pcbddc;
2440da1bb401SStefano Zampini 
2441da1bb401SStefano Zampini   /* create PCIS data structure */
2442da1bb401SStefano Zampini   ierr = PCISCreate(pc);CHKERRQ(ierr);
2443da1bb401SStefano Zampini 
244447d04d0dSStefano Zampini   /* BDDC customization */
244508a5cf49SStefano Zampini   pcbddc->use_local_adj       = PETSC_TRUE;
244647d04d0dSStefano Zampini   pcbddc->use_vertices        = PETSC_TRUE;
244747d04d0dSStefano Zampini   pcbddc->use_edges           = PETSC_TRUE;
244847d04d0dSStefano Zampini   pcbddc->use_faces           = PETSC_FALSE;
244947d04d0dSStefano Zampini   pcbddc->use_change_of_basis = PETSC_FALSE;
245047d04d0dSStefano Zampini   pcbddc->use_change_on_faces = PETSC_FALSE;
245147d04d0dSStefano Zampini   pcbddc->switch_static       = PETSC_FALSE;
2452fa434743SStefano Zampini   pcbddc->use_nnsp_true       = PETSC_FALSE;
2453fa434743SStefano Zampini   pcbddc->use_qr_single       = PETSC_FALSE;
24543301b35fSStefano Zampini   pcbddc->symmetric_primal    = PETSC_TRUE;
245506a4e24aSStefano Zampini   pcbddc->benign_saddle_point = PETSC_FALSE;
245647d04d0dSStefano Zampini   pcbddc->dbg_flag            = 0;
2457b9d89cd5SStefano Zampini   /* private */
2458727cdba6SStefano Zampini   pcbddc->local_primal_size          = 0;
24590e6343abSStefano Zampini   pcbddc->local_primal_size_cc       = 0;
24600e6343abSStefano Zampini   pcbddc->local_primal_ref_node      = 0;
24610e6343abSStefano Zampini   pcbddc->local_primal_ref_mult      = 0;
2462e9189074SStefano Zampini   pcbddc->n_vertices                 = 0;
2463727cdba6SStefano Zampini   pcbddc->primal_indices_local_idxs  = 0;
2464fb180af4SStefano Zampini   pcbddc->recompute_topography       = PETSC_FALSE;
246568457ee5SStefano Zampini   pcbddc->coarse_size                = -1;
2466f4ddd8eeSStefano Zampini   pcbddc->new_primal_space           = PETSC_FALSE;
2467727cdba6SStefano Zampini   pcbddc->new_primal_space_local     = PETSC_FALSE;
2468f4ddd8eeSStefano Zampini   pcbddc->global_primal_indices      = 0;
2469f4ddd8eeSStefano Zampini   pcbddc->onearnullspace             = 0;
2470f4ddd8eeSStefano Zampini   pcbddc->onearnullvecs_state        = 0;
2471674ae819SStefano Zampini   pcbddc->user_primal_vertices       = 0;
24720bdf917eSStefano Zampini   pcbddc->NullSpace                  = 0;
24733972b0daSStefano Zampini   pcbddc->temp_solution              = 0;
2474534831adSStefano Zampini   pcbddc->original_rhs               = 0;
2475534831adSStefano Zampini   pcbddc->local_mat                  = 0;
2476534831adSStefano Zampini   pcbddc->ChangeOfBasisMatrix        = 0;
2477b9b85e73SStefano Zampini   pcbddc->user_ChangeOfBasisMatrix   = 0;
2478906d46d4SStefano Zampini   pcbddc->new_global_mat             = 0;
2479da1bb401SStefano Zampini   pcbddc->coarse_vec                 = 0;
2480da1bb401SStefano Zampini   pcbddc->coarse_ksp                 = 0;
2481da1bb401SStefano Zampini   pcbddc->coarse_phi_B               = 0;
2482da1bb401SStefano Zampini   pcbddc->coarse_phi_D               = 0;
248315aaf578SStefano Zampini   pcbddc->coarse_psi_B               = 0;
248415aaf578SStefano Zampini   pcbddc->coarse_psi_D               = 0;
2485da1bb401SStefano Zampini   pcbddc->vec1_P                     = 0;
2486da1bb401SStefano Zampini   pcbddc->vec1_R                     = 0;
2487da1bb401SStefano Zampini   pcbddc->vec2_R                     = 0;
2488da1bb401SStefano Zampini   pcbddc->local_auxmat1              = 0;
2489da1bb401SStefano Zampini   pcbddc->local_auxmat2              = 0;
2490da1bb401SStefano Zampini   pcbddc->R_to_B                     = 0;
2491da1bb401SStefano Zampini   pcbddc->R_to_D                     = 0;
2492da1bb401SStefano Zampini   pcbddc->ksp_D                      = 0;
2493da1bb401SStefano Zampini   pcbddc->ksp_R                      = 0;
2494da1bb401SStefano Zampini   pcbddc->NeumannBoundaries          = 0;
2495785d1243SStefano Zampini   pcbddc->NeumannBoundariesLocal     = 0;
2496785d1243SStefano Zampini   pcbddc->DirichletBoundaries        = 0;
2497785d1243SStefano Zampini   pcbddc->DirichletBoundariesLocal   = 0;
249860d44989SStefano Zampini   pcbddc->user_provided_isfordofs    = PETSC_FALSE;
249960d44989SStefano Zampini   pcbddc->n_ISForDofs                = 0;
250063602bcaSStefano Zampini   pcbddc->n_ISForDofsLocal           = 0;
2501da1bb401SStefano Zampini   pcbddc->ISForDofs                  = 0;
250263602bcaSStefano Zampini   pcbddc->ISForDofsLocal             = 0;
2503da1bb401SStefano Zampini   pcbddc->ConstraintMatrix           = 0;
250485c4d303SStefano Zampini   pcbddc->use_exact_dirichlet_trick  = PETSC_TRUE;
250547d04d0dSStefano Zampini   pcbddc->coarse_loc_to_glob         = 0;
250647d04d0dSStefano Zampini   pcbddc->coarsening_ratio           = 8;
2507b0c7d250SStefano Zampini   pcbddc->coarse_adj_red             = 0;
25084fad6a16SStefano Zampini   pcbddc->current_level              = 0;
25092b510759SStefano Zampini   pcbddc->max_levels                 = 0;
2510323d395dSStefano Zampini   pcbddc->use_coarse_estimates       = PETSC_FALSE;
251174e2c79eSStefano Zampini   pcbddc->redistribute_coarse        = 0;
2512f3bde8b3SStefano Zampini   pcbddc->coarse_subassembling       = 0;
2513323d395dSStefano Zampini   pcbddc->coarse_subassembling_init  = 0;
251481d14e9dSStefano Zampini 
251581d14e9dSStefano Zampini   /* benign subspace trick */
251681d14e9dSStefano Zampini   pcbddc->B0_ncol                    = 0;
251781d14e9dSStefano Zampini   pcbddc->B0_cols                    = NULL;
251881d14e9dSStefano Zampini   pcbddc->B0_vals                    = NULL;
251981d14e9dSStefano Zampini   pcbddc->benign_change              = 0;
25200369aaf7SStefano Zampini   pcbddc->benign_vec                 = 0;
25210369aaf7SStefano Zampini   pcbddc->benign_original_mat        = 0;
25220369aaf7SStefano Zampini   pcbddc->benign_sf                  = 0;
25238082dc6fSStefano Zampini   pcbddc->benign_p0_lidx             = -1;
25248082dc6fSStefano Zampini   pcbddc->benign_p0_gidx             = -1;
252581d14e9dSStefano Zampini 
2526674ae819SStefano Zampini   /* create local graph structure */
2527674ae819SStefano Zampini   ierr = PCBDDCGraphCreate(&pcbddc->mat_graph);CHKERRQ(ierr);
2528674ae819SStefano Zampini 
2529674ae819SStefano Zampini   /* scaling */
2530674ae819SStefano Zampini   pcbddc->work_scaling          = 0;
253134a97f8cSStefano Zampini   pcbddc->use_deluxe_scaling    = PETSC_FALSE;
2532ac632422SStefano Zampini   pcbddc->faster_deluxe         = PETSC_FALSE;
2533b96c3477SStefano Zampini 
2534b96c3477SStefano Zampini   /* create sub schurs structure */
2535b96c3477SStefano Zampini   ierr = PCBDDCSubSchursCreate(&pcbddc->sub_schurs);CHKERRQ(ierr);
2536b96c3477SStefano Zampini   pcbddc->sub_schurs_rebuild     = PETSC_FALSE;
2537b96c3477SStefano Zampini   pcbddc->sub_schurs_layers      = -1;
2538b96c3477SStefano Zampini   pcbddc->sub_schurs_use_useradj = PETSC_FALSE;
2539b96c3477SStefano Zampini 
2540b96c3477SStefano Zampini   pcbddc->computed_rowadj = PETSC_FALSE;
2541da1bb401SStefano Zampini 
2542b7eb3628SStefano Zampini   /* adaptivity */
2543f6f667cfSStefano Zampini   pcbddc->adaptive_threshold      = 0.0;
254408122e43SStefano Zampini   pcbddc->adaptive_nmax           = 0;
2545f6f667cfSStefano Zampini   pcbddc->adaptive_nmin           = 0;
2546b7eb3628SStefano Zampini 
2547da1bb401SStefano Zampini   /* function pointers */
2548da1bb401SStefano Zampini   pc->ops->apply               = PCApply_BDDC;
254993bd9ae7SStefano Zampini   pc->ops->applytranspose      = PCApplyTranspose_BDDC;
2550da1bb401SStefano Zampini   pc->ops->setup               = PCSetUp_BDDC;
2551da1bb401SStefano Zampini   pc->ops->destroy             = PCDestroy_BDDC;
2552da1bb401SStefano Zampini   pc->ops->setfromoptions      = PCSetFromOptions_BDDC;
2553da1bb401SStefano Zampini   pc->ops->view                = 0;
2554da1bb401SStefano Zampini   pc->ops->applyrichardson     = 0;
2555da1bb401SStefano Zampini   pc->ops->applysymmetricleft  = 0;
2556da1bb401SStefano Zampini   pc->ops->applysymmetricright = 0;
2557534831adSStefano Zampini   pc->ops->presolve            = PCPreSolve_BDDC;
2558534831adSStefano Zampini   pc->ops->postsolve           = PCPostSolve_BDDC;
2559da1bb401SStefano Zampini 
2560da1bb401SStefano Zampini   /* composing function */
2561906d46d4SStefano Zampini   ierr = PetscObjectComposeFunction((PetscObject)pc,"PCBDDCSetChangeOfBasisMat_C",PCBDDCSetChangeOfBasisMat_BDDC);CHKERRQ(ierr);
2562674ae819SStefano Zampini   ierr = PetscObjectComposeFunction((PetscObject)pc,"PCBDDCSetPrimalVerticesLocalIS_C",PCBDDCSetPrimalVerticesLocalIS_BDDC);CHKERRQ(ierr);
2563bdf89e91SBarry Smith   ierr = PetscObjectComposeFunction((PetscObject)pc,"PCBDDCSetCoarseningRatio_C",PCBDDCSetCoarseningRatio_BDDC);CHKERRQ(ierr);
25642b510759SStefano Zampini   ierr = PetscObjectComposeFunction((PetscObject)pc,"PCBDDCSetLevel_C",PCBDDCSetLevel_BDDC);CHKERRQ(ierr);
2565b8ffe317SStefano Zampini   ierr = PetscObjectComposeFunction((PetscObject)pc,"PCBDDCSetUseExactDirichlet_C",PCBDDCSetUseExactDirichlet_BDDC);CHKERRQ(ierr);
25662b510759SStefano Zampini   ierr = PetscObjectComposeFunction((PetscObject)pc,"PCBDDCSetLevels_C",PCBDDCSetLevels_BDDC);CHKERRQ(ierr);
2567bdf89e91SBarry Smith   ierr = PetscObjectComposeFunction((PetscObject)pc,"PCBDDCSetNullSpace_C",PCBDDCSetNullSpace_BDDC);CHKERRQ(ierr);
2568bdf89e91SBarry Smith   ierr = PetscObjectComposeFunction((PetscObject)pc,"PCBDDCSetDirichletBoundaries_C",PCBDDCSetDirichletBoundaries_BDDC);CHKERRQ(ierr);
256982ba6b80SStefano Zampini   ierr = PetscObjectComposeFunction((PetscObject)pc,"PCBDDCSetDirichletBoundariesLocal_C",PCBDDCSetDirichletBoundariesLocal_BDDC);CHKERRQ(ierr);
2570bdf89e91SBarry Smith   ierr = PetscObjectComposeFunction((PetscObject)pc,"PCBDDCSetNeumannBoundaries_C",PCBDDCSetNeumannBoundaries_BDDC);CHKERRQ(ierr);
257182ba6b80SStefano Zampini   ierr = PetscObjectComposeFunction((PetscObject)pc,"PCBDDCSetNeumannBoundariesLocal_C",PCBDDCSetNeumannBoundariesLocal_BDDC);CHKERRQ(ierr);
2572bdf89e91SBarry Smith   ierr = PetscObjectComposeFunction((PetscObject)pc,"PCBDDCGetDirichletBoundaries_C",PCBDDCGetDirichletBoundaries_BDDC);CHKERRQ(ierr);
257382ba6b80SStefano Zampini   ierr = PetscObjectComposeFunction((PetscObject)pc,"PCBDDCGetDirichletBoundariesLocal_C",PCBDDCGetDirichletBoundariesLocal_BDDC);CHKERRQ(ierr);
2574bdf89e91SBarry Smith   ierr = PetscObjectComposeFunction((PetscObject)pc,"PCBDDCGetNeumannBoundaries_C",PCBDDCGetNeumannBoundaries_BDDC);CHKERRQ(ierr);
257582ba6b80SStefano Zampini   ierr = PetscObjectComposeFunction((PetscObject)pc,"PCBDDCGetNeumannBoundariesLocal_C",PCBDDCGetNeumannBoundariesLocal_BDDC);CHKERRQ(ierr);
2576bdf89e91SBarry Smith   ierr = PetscObjectComposeFunction((PetscObject)pc,"PCBDDCSetDofsSplitting_C",PCBDDCSetDofsSplitting_BDDC);CHKERRQ(ierr);
257763602bcaSStefano Zampini   ierr = PetscObjectComposeFunction((PetscObject)pc,"PCBDDCSetDofsSplittingLocal_C",PCBDDCSetDofsSplittingLocal_BDDC);CHKERRQ(ierr);
2578bdf89e91SBarry Smith   ierr = PetscObjectComposeFunction((PetscObject)pc,"PCBDDCSetLocalAdjacencyGraph_C",PCBDDCSetLocalAdjacencyGraph_BDDC);CHKERRQ(ierr);
2579bdf89e91SBarry Smith   ierr = PetscObjectComposeFunction((PetscObject)pc,"PCBDDCCreateFETIDPOperators_C",PCBDDCCreateFETIDPOperators_BDDC);CHKERRQ(ierr);
2580bdf89e91SBarry Smith   ierr = PetscObjectComposeFunction((PetscObject)pc,"PCBDDCMatFETIDPGetRHS_C",PCBDDCMatFETIDPGetRHS_BDDC);CHKERRQ(ierr);
2581bdf89e91SBarry Smith   ierr = PetscObjectComposeFunction((PetscObject)pc,"PCBDDCMatFETIDPGetSolution_C",PCBDDCMatFETIDPGetSolution_BDDC);CHKERRQ(ierr);
2582da1bb401SStefano Zampini   PetscFunctionReturn(0);
2583da1bb401SStefano Zampini }
25843425bc38SStefano Zampini 
2585