xref: /petsc/src/ksp/pc/impls/bddc/bddc.c (revision 00f6b5310dc197cd33ccfb73113c74376b0f1eb6)
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    MATIS related operations contained in BDDC code
18eb97c9d2SStefano Zampini    - Provide general case for subassembling
19eb97c9d2SStefano Zampini 
2053cdbc3dSStefano Zampini */
210c7d97c5SJed Brown 
22ab5c6b0cSJed Brown #include <../src/ksp/pc/impls/bddc/bddc.h> /*I "petscpc.h" I*/  /* includes for fortran wrappers */
23ab5c6b0cSJed Brown #include <../src/ksp/pc/impls/bddc/bddcprivate.h>
243b03a366Sstefano_zampini #include <petscblaslapack.h>
25674ae819SStefano Zampini 
260c7d97c5SJed Brown /* -------------------------------------------------------------------------- */
270c7d97c5SJed Brown #undef __FUNCT__
280c7d97c5SJed Brown #define __FUNCT__ "PCSetFromOptions_BDDC"
294416b707SBarry Smith PetscErrorCode PCSetFromOptions_BDDC(PetscOptionItems *PetscOptionsObject,PC pc)
300c7d97c5SJed Brown {
310c7d97c5SJed Brown   PC_BDDC        *pcbddc = (PC_BDDC*)pc->data;
320c7d97c5SJed Brown   PetscErrorCode ierr;
330c7d97c5SJed Brown 
340c7d97c5SJed Brown   PetscFunctionBegin;
35e55864a3SBarry Smith   ierr = PetscOptionsHead(PetscOptionsObject,"BDDC options");CHKERRQ(ierr);
368eeda7d8SStefano Zampini   /* Verbose debugging */
37c7017625SStefano Zampini   ierr = PetscOptionsBool("-pc_bddc_dirichlet_approximate","Inform PCBDDC that we are using approximate Dirichlet solvers","none",pcbddc->NullSpace_corr[0],&pcbddc->NullSpace_corr[0],NULL);CHKERRQ(ierr);
38c7017625SStefano Zampini   ierr = PetscOptionsBool("-pc_bddc_dirichlet_approximate_scale","Inform PCBDDC that we need to scale the Dirichlet solve","none",pcbddc->NullSpace_corr[1],&pcbddc->NullSpace_corr[1],NULL);CHKERRQ(ierr);
39c7017625SStefano Zampini   ierr = PetscOptionsBool("-pc_bddc_neumann_approximate","Inform PCBDDC that we are using approximate Neumann solvers","none",pcbddc->NullSpace_corr[2],&pcbddc->NullSpace_corr[2],NULL);CHKERRQ(ierr);
40c7017625SStefano Zampini   ierr = PetscOptionsBool("-pc_bddc_neumann_approximate_scale","Inform PCBDDC that we need to scale the Neumann solve","none",pcbddc->NullSpace_corr[3],&pcbddc->NullSpace_corr[3],NULL);CHKERRQ(ierr);
418eeda7d8SStefano Zampini   ierr = PetscOptionsInt("-pc_bddc_check_level","Verbose output for PCBDDC (intended for debug)","none",pcbddc->dbg_flag,&pcbddc->dbg_flag,NULL);CHKERRQ(ierr);
426b78500eSPatrick Sanan   /* Primal space customization */
4308a5cf49SStefano 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);
448eeda7d8SStefano 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);
458eeda7d8SStefano 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);
468eeda7d8SStefano 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);
476661aa1dSStefano 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);
48ac632422SStefano 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);
498eeda7d8SStefano Zampini   /* Change of basis */
50b9b85e73SStefano 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);
51b9b85e73SStefano 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);
52674ae819SStefano Zampini   if (!pcbddc->use_change_of_basis) {
53674ae819SStefano Zampini     pcbddc->use_change_on_faces = PETSC_FALSE;
54674ae819SStefano Zampini   }
558eeda7d8SStefano Zampini   /* Switch between M_2 (default) and M_3 preconditioners (as defined by C. Dohrmann in the ref. article) */
568eeda7d8SStefano 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);
5774e2c79eSStefano 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);
580298fd71SBarry Smith   ierr = PetscOptionsInt("-pc_bddc_coarsening_ratio","Set coarsening ratio used in multilevel coarsening","none",pcbddc->coarsening_ratio,&pcbddc->coarsening_ratio,NULL);CHKERRQ(ierr);
592b510759SStefano Zampini   ierr = PetscOptionsInt("-pc_bddc_levels","Set maximum number of levels for multilevel","none",pcbddc->max_levels,&pcbddc->max_levels,NULL);CHKERRQ(ierr);
60323d395dSStefano 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);
61674ae819SStefano 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);
62b96c3477SStefano 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);
63b96c3477SStefano 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);
64b96c3477SStefano 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);
65ac632422SStefano 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);
664c6709b3SStefano 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);
6708122e43SStefano Zampini   ierr = PetscOptionsInt("-pc_bddc_adaptive_nmin","Minimum number of constraints per connected components","none",pcbddc->adaptive_nmin,&pcbddc->adaptive_nmin,NULL);CHKERRQ(ierr);
6808122e43SStefano Zampini   ierr = PetscOptionsInt("-pc_bddc_adaptive_nmax","Maximum number of constraints per connected components","none",pcbddc->adaptive_nmax,&pcbddc->adaptive_nmax,NULL);CHKERRQ(ierr);
693301b35fSStefano Zampini   ierr = PetscOptionsBool("-pc_bddc_symmetric","Symmetric computation of primal basis functions","none",pcbddc->symmetric_primal,&pcbddc->symmetric_primal,NULL);CHKERRQ(ierr);
70b0c7d250SStefano 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);
710c7d97c5SJed Brown   ierr = PetscOptionsTail();CHKERRQ(ierr);
720c7d97c5SJed Brown   PetscFunctionReturn(0);
730c7d97c5SJed Brown }
746b78500eSPatrick Sanan 
756b78500eSPatrick Sanan /* -------------------------------------------------------------------------- */
766b78500eSPatrick Sanan #undef __FUNCT__
776b78500eSPatrick Sanan #define __FUNCT__ "PCView_BDDC"
786b78500eSPatrick Sanan static PetscErrorCode PCView_BDDC(PC pc,PetscViewer viewer)
796b78500eSPatrick Sanan {
806b78500eSPatrick Sanan   PC_BDDC              *pcbddc = (PC_BDDC*)pc->data;
816b78500eSPatrick Sanan   PetscErrorCode       ierr;
826b78500eSPatrick Sanan   PetscBool            isascii,isstring;
836b78500eSPatrick Sanan 
846b78500eSPatrick Sanan   PetscFunctionBegin;
856b78500eSPatrick Sanan   ierr = PetscObjectTypeCompare((PetscObject)viewer,PETSCVIEWERASCII,&isascii);CHKERRQ(ierr);
866b78500eSPatrick Sanan   ierr = PetscObjectTypeCompare((PetscObject)viewer,PETSCVIEWERSTRING,&isstring);CHKERRQ(ierr);
876b78500eSPatrick Sanan 
886b78500eSPatrick Sanan   /* In a braindead way, print out anything which the user can control from the command line,
896b78500eSPatrick Sanan      cribbing from PCSetFromOptions_BDDC */
906b78500eSPatrick Sanan 
916b78500eSPatrick Sanan   /* Nothing printed for the String viewer */
926b78500eSPatrick Sanan 
936b78500eSPatrick Sanan   /* ASCII viewer */
946b78500eSPatrick Sanan   if (isascii) {
956b78500eSPatrick Sanan     /* Verbose debugging */
966b78500eSPatrick Sanan     ierr = PetscViewerASCIIPrintf(viewer,    "  BDDC: Use verbose output: %d\n",pcbddc->dbg_flag);CHKERRQ(ierr);
976b78500eSPatrick Sanan 
986b78500eSPatrick Sanan     /* Primal space customization */
996b78500eSPatrick Sanan     ierr = PetscViewerASCIIPrintf(viewer,    "  BDDC: Use local mat graph: %d\n",pcbddc->use_local_adj);CHKERRQ(ierr);
1006b78500eSPatrick Sanan     ierr = PetscViewerASCIIPrintf(viewer,    "  BDDC: Use vertices: %d\n",pcbddc->use_vertices);CHKERRQ(ierr);
1016b78500eSPatrick Sanan     ierr = PetscViewerASCIIPrintf(viewer,    "  BDDC: Use edges: %d\n",pcbddc->use_edges);CHKERRQ(ierr);
1026b78500eSPatrick Sanan     ierr = PetscViewerASCIIPrintf(viewer,    "  BDDC: Use faces: %d\n",pcbddc->use_faces);CHKERRQ(ierr);
1036b78500eSPatrick Sanan     ierr = PetscViewerASCIIPrintf(viewer,    "  BDDC: Use true near null space: %d\n",pcbddc->use_nnsp_true);CHKERRQ(ierr);
1046b78500eSPatrick Sanan     ierr = PetscViewerASCIIPrintf(viewer,    "  BDDC: Use QR for single constraints on cc: %d\n",pcbddc->use_qr_single);CHKERRQ(ierr);
1056b78500eSPatrick Sanan 
1066b78500eSPatrick Sanan     /* Change of basis */
1076b78500eSPatrick Sanan     ierr = PetscViewerASCIIPrintf(viewer,    "  BDDC: Use change of basis on local edge nodes: %d\n",pcbddc->use_change_of_basis);CHKERRQ(ierr);
1086b78500eSPatrick Sanan     ierr = PetscViewerASCIIPrintf(viewer,    "  BDDC: Use change of basis on local face nodes: %d\n",pcbddc->use_change_on_faces);CHKERRQ(ierr);
1096b78500eSPatrick Sanan 
1106b78500eSPatrick Sanan     /* Switch between M_2 (default) and M_3 preconditioners (as defined by C. Dohrmann in the ref. article) */
1116b78500eSPatrick Sanan     ierr = PetscViewerASCIIPrintf(viewer,    "  BDDC: Switch on static condensation ops around the interface preconditioner: %d\n",pcbddc->switch_static);CHKERRQ(ierr);
1126b78500eSPatrick Sanan     ierr = PetscViewerASCIIPrintf(viewer,    "  BDDC: Coarse problem restribute procs: %d\n",pcbddc->redistribute_coarse);CHKERRQ(ierr);
1136b78500eSPatrick Sanan     ierr = PetscViewerASCIIPrintf(viewer,    "  BDDC: Multilevel coarsening ratio: %d\n",pcbddc->coarsening_ratio);CHKERRQ(ierr);
1146b78500eSPatrick Sanan     ierr = PetscViewerASCIIPrintf(viewer,    "  BDDC: Multilevel max levels: %d\n",pcbddc->max_levels);CHKERRQ(ierr);
1156b78500eSPatrick Sanan     ierr = PetscViewerASCIIPrintf(viewer,    "  BDDC: Use estimated eigs for coarse problem: %d\n",pcbddc->use_coarse_estimates);CHKERRQ(ierr);
1166b78500eSPatrick Sanan     ierr = PetscViewerASCIIPrintf(viewer,    "  BDDC: Use deluxe scaling: %d\n",pcbddc->use_deluxe_scaling);CHKERRQ(ierr);
1176b78500eSPatrick Sanan     ierr = PetscViewerASCIIPrintf(viewer,    "  BDDC: Rebuild interface graph for Schur principal minors: %d\n",pcbddc->sub_schurs_rebuild);CHKERRQ(ierr);
1186b78500eSPatrick Sanan     ierr = PetscViewerASCIIPrintf(viewer,    "  BDDC: Number of dofs' layers for the computation of principal minors: %d\n",pcbddc->sub_schurs_layers);CHKERRQ(ierr);
1196b78500eSPatrick Sanan     ierr = PetscViewerASCIIPrintf(viewer,    "  BDDC: Use user CSR graph to compute successive layers: %d\n",pcbddc->sub_schurs_use_useradj);CHKERRQ(ierr);
1206b78500eSPatrick Sanan     ierr = PetscViewerASCIIPrintf(viewer,    "  BDDC: Fast deluxe scaling: %d\n",pcbddc->faster_deluxe);CHKERRQ(ierr);
1216b78500eSPatrick Sanan     ierr = PetscViewerASCIIPrintf(viewer,    "  BDDC: Adaptive constraint selection threshold: %g\n",pcbddc->adaptive_threshold);CHKERRQ(ierr);
1226b78500eSPatrick Sanan     ierr = PetscViewerASCIIPrintf(viewer,    "  BDDC: Min constraints / connected component: %d\n",pcbddc->adaptive_nmin);CHKERRQ(ierr);
1236b78500eSPatrick Sanan     ierr = PetscViewerASCIIPrintf(viewer,    "  BDDC: Max constraints / connected component: %d\n",pcbddc->adaptive_nmax);CHKERRQ(ierr);
1246b78500eSPatrick Sanan     ierr = PetscViewerASCIIPrintf(viewer,    "  BDDC: Symmetric computation of primal basis functions: %d\n",pcbddc->symmetric_primal);CHKERRQ(ierr);
1256b78500eSPatrick Sanan     ierr = PetscViewerASCIIPrintf(viewer,    "  BDDC: Num. Procs. to map coarse adjacency list: %d\n",pcbddc->coarse_adj_red);CHKERRQ(ierr);
1266b78500eSPatrick Sanan   }
1276b78500eSPatrick Sanan   PetscFunctionReturn(0);
1286b78500eSPatrick Sanan }
1296b78500eSPatrick Sanan 
1300c7d97c5SJed Brown /* -------------------------------------------------------------------------- */
131674ae819SStefano Zampini #undef __FUNCT__
132906d46d4SStefano Zampini #define __FUNCT__ "PCBDDCSetChangeOfBasisMat_BDDC"
133906d46d4SStefano Zampini static PetscErrorCode PCBDDCSetChangeOfBasisMat_BDDC(PC pc, Mat change)
134b9b85e73SStefano Zampini {
135b9b85e73SStefano Zampini   PC_BDDC        *pcbddc = (PC_BDDC*)pc->data;
136b9b85e73SStefano Zampini   PetscErrorCode ierr;
137b9b85e73SStefano Zampini 
138b9b85e73SStefano Zampini   PetscFunctionBegin;
139b9b85e73SStefano Zampini   ierr = MatDestroy(&pcbddc->user_ChangeOfBasisMatrix);CHKERRQ(ierr);
140b9b85e73SStefano Zampini   ierr = PetscObjectReference((PetscObject)change);CHKERRQ(ierr);
141b9b85e73SStefano Zampini   pcbddc->user_ChangeOfBasisMatrix = change;
142b9b85e73SStefano Zampini   PetscFunctionReturn(0);
143b9b85e73SStefano Zampini }
144b9b85e73SStefano Zampini #undef __FUNCT__
145906d46d4SStefano Zampini #define __FUNCT__ "PCBDDCSetChangeOfBasisMat"
146b9b85e73SStefano Zampini /*@
147906d46d4SStefano Zampini  PCBDDCSetChangeOfBasisMat - Set user defined change of basis for dofs
148b9b85e73SStefano Zampini 
149b9b85e73SStefano Zampini    Collective on PC
150b9b85e73SStefano Zampini 
151b9b85e73SStefano Zampini    Input Parameters:
152b9b85e73SStefano Zampini +  pc - the preconditioning context
153906d46d4SStefano Zampini -  change - the change of basis matrix
154b9b85e73SStefano Zampini 
155b9b85e73SStefano Zampini    Level: intermediate
156b9b85e73SStefano Zampini 
157b9b85e73SStefano Zampini    Notes:
158b9b85e73SStefano Zampini 
159b9b85e73SStefano Zampini .seealso: PCBDDC
160b9b85e73SStefano Zampini @*/
161906d46d4SStefano Zampini PetscErrorCode PCBDDCSetChangeOfBasisMat(PC pc, Mat change)
162b9b85e73SStefano Zampini {
163b9b85e73SStefano Zampini   PetscErrorCode ierr;
164b9b85e73SStefano Zampini 
165b9b85e73SStefano Zampini   PetscFunctionBegin;
166b9b85e73SStefano Zampini   PetscValidHeaderSpecific(pc,PC_CLASSID,1);
167b9b85e73SStefano Zampini   PetscValidHeaderSpecific(change,MAT_CLASSID,2);
168906d46d4SStefano Zampini   PetscCheckSameComm(pc,1,change,2);
169906d46d4SStefano Zampini   if (pc->mat) {
170906d46d4SStefano Zampini     PetscInt rows_c,cols_c,rows,cols;
171906d46d4SStefano Zampini     ierr = MatGetSize(pc->mat,&rows,&cols);CHKERRQ(ierr);
172906d46d4SStefano Zampini     ierr = MatGetSize(change,&rows_c,&cols_c);CHKERRQ(ierr);
1736c4ed002SBarry Smith     if (rows_c != rows) SETERRQ2(PetscObjectComm((PetscObject)pc),PETSC_ERR_SUP,"Invalid number of rows for change of basis matrix! %d != %d",rows_c,rows);
1746c4ed002SBarry Smith     if (cols_c != cols) SETERRQ2(PetscObjectComm((PetscObject)pc),PETSC_ERR_SUP,"Invalid number of columns for change of basis matrix! %d != %d",cols_c,cols);
175906d46d4SStefano Zampini     ierr = MatGetLocalSize(pc->mat,&rows,&cols);CHKERRQ(ierr);
176906d46d4SStefano Zampini     ierr = MatGetLocalSize(change,&rows_c,&cols_c);CHKERRQ(ierr);
1776c4ed002SBarry Smith     if (rows_c != rows) SETERRQ2(PetscObjectComm((PetscObject)pc),PETSC_ERR_SUP,"Invalid number of local rows for change of basis matrix! %d != %d",rows_c,rows);
1786c4ed002SBarry Smith     if (cols_c != cols) SETERRQ2(PetscObjectComm((PetscObject)pc),PETSC_ERR_SUP,"Invalid number of local columns for change of basis matrix! %d != %d",cols_c,cols);
179906d46d4SStefano Zampini   }
180906d46d4SStefano Zampini   ierr = PetscTryMethod(pc,"PCBDDCSetChangeOfBasisMat_C",(PC,Mat),(pc,change));CHKERRQ(ierr);
181b9b85e73SStefano Zampini   PetscFunctionReturn(0);
182b9b85e73SStefano Zampini }
183b9b85e73SStefano Zampini /* -------------------------------------------------------------------------- */
184b9b85e73SStefano Zampini #undef __FUNCT__
185674ae819SStefano Zampini #define __FUNCT__ "PCBDDCSetPrimalVerticesLocalIS_BDDC"
186674ae819SStefano Zampini static PetscErrorCode PCBDDCSetPrimalVerticesLocalIS_BDDC(PC pc, IS PrimalVertices)
187674ae819SStefano Zampini {
188674ae819SStefano Zampini   PC_BDDC        *pcbddc = (PC_BDDC*)pc->data;
189674ae819SStefano Zampini   PetscErrorCode ierr;
1901e6b0712SBarry Smith 
191674ae819SStefano Zampini   PetscFunctionBegin;
192674ae819SStefano Zampini   ierr = ISDestroy(&pcbddc->user_primal_vertices);CHKERRQ(ierr);
193674ae819SStefano Zampini   ierr = PetscObjectReference((PetscObject)PrimalVertices);CHKERRQ(ierr);
194674ae819SStefano Zampini   pcbddc->user_primal_vertices = PrimalVertices;
195674ae819SStefano Zampini   PetscFunctionReturn(0);
196674ae819SStefano Zampini }
197674ae819SStefano Zampini #undef __FUNCT__
198674ae819SStefano Zampini #define __FUNCT__ "PCBDDCSetPrimalVerticesLocalIS"
199674ae819SStefano Zampini /*@
20028509bceSStefano Zampini  PCBDDCSetPrimalVerticesLocalIS - Set additional user defined primal vertices in PCBDDC
201674ae819SStefano Zampini 
20217eb1463SStefano Zampini    Collective
203674ae819SStefano Zampini 
204674ae819SStefano Zampini    Input Parameters:
205674ae819SStefano Zampini +  pc - the preconditioning context
20617eb1463SStefano Zampini -  PrimalVertices - index set of primal vertices in local numbering (can be empty)
207674ae819SStefano Zampini 
208674ae819SStefano Zampini    Level: intermediate
209674ae819SStefano Zampini 
210674ae819SStefano Zampini    Notes:
211674ae819SStefano Zampini 
212674ae819SStefano Zampini .seealso: PCBDDC
213674ae819SStefano Zampini @*/
214674ae819SStefano Zampini PetscErrorCode PCBDDCSetPrimalVerticesLocalIS(PC pc, IS PrimalVertices)
215674ae819SStefano Zampini {
216674ae819SStefano Zampini   PetscErrorCode ierr;
217674ae819SStefano Zampini 
218674ae819SStefano Zampini   PetscFunctionBegin;
219674ae819SStefano Zampini   PetscValidHeaderSpecific(pc,PC_CLASSID,1);
220674ae819SStefano Zampini   PetscValidHeaderSpecific(PrimalVertices,IS_CLASSID,2);
22117eb1463SStefano Zampini   PetscCheckSameComm(pc,1,PrimalVertices,2);
222674ae819SStefano Zampini   ierr = PetscTryMethod(pc,"PCBDDCSetPrimalVerticesLocalIS_C",(PC,IS),(pc,PrimalVertices));CHKERRQ(ierr);
223674ae819SStefano Zampini   PetscFunctionReturn(0);
224674ae819SStefano Zampini }
225674ae819SStefano Zampini /* -------------------------------------------------------------------------- */
2260c7d97c5SJed Brown #undef __FUNCT__
2274fad6a16SStefano Zampini #define __FUNCT__ "PCBDDCSetCoarseningRatio_BDDC"
2284fad6a16SStefano Zampini static PetscErrorCode PCBDDCSetCoarseningRatio_BDDC(PC pc,PetscInt k)
2294fad6a16SStefano Zampini {
2304fad6a16SStefano Zampini   PC_BDDC  *pcbddc = (PC_BDDC*)pc->data;
2314fad6a16SStefano Zampini 
2324fad6a16SStefano Zampini   PetscFunctionBegin;
2334fad6a16SStefano Zampini   pcbddc->coarsening_ratio = k;
2344fad6a16SStefano Zampini   PetscFunctionReturn(0);
2354fad6a16SStefano Zampini }
2361e6b0712SBarry Smith 
2374fad6a16SStefano Zampini #undef __FUNCT__
2384fad6a16SStefano Zampini #define __FUNCT__ "PCBDDCSetCoarseningRatio"
2394fad6a16SStefano Zampini /*@
24028509bceSStefano Zampini  PCBDDCSetCoarseningRatio - Set coarsening ratio used in multilevel
2414fad6a16SStefano Zampini 
2424fad6a16SStefano Zampini    Logically collective on PC
2434fad6a16SStefano Zampini 
2444fad6a16SStefano Zampini    Input Parameters:
2454fad6a16SStefano Zampini +  pc - the preconditioning context
24628509bceSStefano Zampini -  k - coarsening ratio (H/h at the coarser level)
2474fad6a16SStefano Zampini 
2480f202f7eSStefano Zampini    Options Database Keys:
2490f202f7eSStefano Zampini .    -pc_bddc_coarsening_ratio
2504fad6a16SStefano Zampini 
2514fad6a16SStefano Zampini    Level: intermediate
2524fad6a16SStefano Zampini 
2534fad6a16SStefano Zampini    Notes:
2540f202f7eSStefano Zampini      Approximatively k subdomains at the finer level will be aggregated into a single subdomain at the coarser level
2554fad6a16SStefano Zampini 
2560f202f7eSStefano Zampini .seealso: PCBDDC, PCBDDCSetLevels()
2574fad6a16SStefano Zampini @*/
2584fad6a16SStefano Zampini PetscErrorCode PCBDDCSetCoarseningRatio(PC pc,PetscInt k)
2594fad6a16SStefano Zampini {
2604fad6a16SStefano Zampini   PetscErrorCode ierr;
2614fad6a16SStefano Zampini 
2624fad6a16SStefano Zampini   PetscFunctionBegin;
2634fad6a16SStefano Zampini   PetscValidHeaderSpecific(pc,PC_CLASSID,1);
2642b510759SStefano Zampini   PetscValidLogicalCollectiveInt(pc,k,2);
2654fad6a16SStefano Zampini   ierr = PetscTryMethod(pc,"PCBDDCSetCoarseningRatio_C",(PC,PetscInt),(pc,k));CHKERRQ(ierr);
2664fad6a16SStefano Zampini   PetscFunctionReturn(0);
2674fad6a16SStefano Zampini }
2682b510759SStefano Zampini 
269b8ffe317SStefano Zampini /* The following functions (PCBDDCSetUseExactDirichlet PCBDDCSetLevel) are not public */
2702b510759SStefano Zampini #undef __FUNCT__
271b8ffe317SStefano Zampini #define __FUNCT__ "PCBDDCSetUseExactDirichlet_BDDC"
272b8ffe317SStefano Zampini static PetscErrorCode PCBDDCSetUseExactDirichlet_BDDC(PC pc,PetscBool flg)
273b8ffe317SStefano Zampini {
274b8ffe317SStefano Zampini   PC_BDDC  *pcbddc = (PC_BDDC*)pc->data;
275b8ffe317SStefano Zampini 
276b8ffe317SStefano Zampini   PetscFunctionBegin;
27785c4d303SStefano Zampini   pcbddc->use_exact_dirichlet_trick = flg;
278b8ffe317SStefano Zampini   PetscFunctionReturn(0);
279b8ffe317SStefano Zampini }
280b8ffe317SStefano Zampini 
281b8ffe317SStefano Zampini #undef __FUNCT__
282b8ffe317SStefano Zampini #define __FUNCT__ "PCBDDCSetUseExactDirichlet"
283b8ffe317SStefano Zampini PetscErrorCode PCBDDCSetUseExactDirichlet(PC pc,PetscBool flg)
2842b510759SStefano Zampini {
2852b510759SStefano Zampini   PetscErrorCode ierr;
2862b510759SStefano Zampini 
2872b510759SStefano Zampini   PetscFunctionBegin;
2882b510759SStefano Zampini   PetscValidHeaderSpecific(pc,PC_CLASSID,1);
289b8ffe317SStefano Zampini   PetscValidLogicalCollectiveBool(pc,flg,2);
290b8ffe317SStefano Zampini   ierr = PetscTryMethod(pc,"PCBDDCSetUseExactDirichlet_C",(PC,PetscBool),(pc,flg));CHKERRQ(ierr);
2912b510759SStefano Zampini   PetscFunctionReturn(0);
2922b510759SStefano Zampini }
2931e6b0712SBarry Smith 
2944fad6a16SStefano Zampini #undef __FUNCT__
2952b510759SStefano Zampini #define __FUNCT__ "PCBDDCSetLevel_BDDC"
2962b510759SStefano Zampini static PetscErrorCode PCBDDCSetLevel_BDDC(PC pc,PetscInt level)
2974fad6a16SStefano Zampini {
2984fad6a16SStefano Zampini   PC_BDDC  *pcbddc = (PC_BDDC*)pc->data;
2994fad6a16SStefano Zampini 
3004fad6a16SStefano Zampini   PetscFunctionBegin;
3012b510759SStefano Zampini   pcbddc->current_level = level;
3024fad6a16SStefano Zampini   PetscFunctionReturn(0);
3034fad6a16SStefano Zampini }
3041e6b0712SBarry Smith 
3054fad6a16SStefano Zampini #undef __FUNCT__
306b8ffe317SStefano Zampini #define __FUNCT__ "PCBDDCSetLevel"
307b8ffe317SStefano Zampini PetscErrorCode PCBDDCSetLevel(PC pc,PetscInt level)
308b8ffe317SStefano Zampini {
309b8ffe317SStefano Zampini   PetscErrorCode ierr;
310b8ffe317SStefano Zampini 
311b8ffe317SStefano Zampini   PetscFunctionBegin;
312b8ffe317SStefano Zampini   PetscValidHeaderSpecific(pc,PC_CLASSID,1);
313b8ffe317SStefano Zampini   PetscValidLogicalCollectiveInt(pc,level,2);
314b8ffe317SStefano Zampini   ierr = PetscTryMethod(pc,"PCBDDCSetLevel_C",(PC,PetscInt),(pc,level));CHKERRQ(ierr);
315b8ffe317SStefano Zampini   PetscFunctionReturn(0);
316b8ffe317SStefano Zampini }
317b8ffe317SStefano Zampini 
318b8ffe317SStefano Zampini #undef __FUNCT__
3192b510759SStefano Zampini #define __FUNCT__ "PCBDDCSetLevels_BDDC"
3202b510759SStefano Zampini static PetscErrorCode PCBDDCSetLevels_BDDC(PC pc,PetscInt levels)
3212b510759SStefano Zampini {
3222b510759SStefano Zampini   PC_BDDC  *pcbddc = (PC_BDDC*)pc->data;
3232b510759SStefano Zampini 
3242b510759SStefano Zampini   PetscFunctionBegin;
3252b510759SStefano Zampini   pcbddc->max_levels = levels;
3262b510759SStefano Zampini   PetscFunctionReturn(0);
3272b510759SStefano Zampini }
3282b510759SStefano Zampini 
3292b510759SStefano Zampini #undef __FUNCT__
3302b510759SStefano Zampini #define __FUNCT__ "PCBDDCSetLevels"
3314fad6a16SStefano Zampini /*@
33228509bceSStefano Zampini  PCBDDCSetLevels - Sets the maximum number of levels for multilevel
3334fad6a16SStefano Zampini 
3344fad6a16SStefano Zampini    Logically collective on PC
3354fad6a16SStefano Zampini 
3364fad6a16SStefano Zampini    Input Parameters:
3374fad6a16SStefano Zampini +  pc - the preconditioning context
33828509bceSStefano Zampini -  levels - the maximum number of levels (max 9)
3394fad6a16SStefano Zampini 
3400f202f7eSStefano Zampini    Options Database Keys:
3410f202f7eSStefano Zampini .    -pc_bddc_levels
3424fad6a16SStefano Zampini 
3434fad6a16SStefano Zampini    Level: intermediate
3444fad6a16SStefano Zampini 
3454fad6a16SStefano Zampini    Notes:
3460f202f7eSStefano Zampini      Default value is 0, i.e. traditional one-level BDDC
3474fad6a16SStefano Zampini 
3480f202f7eSStefano Zampini .seealso: PCBDDC, PCBDDCSetCoarseningRatio()
3494fad6a16SStefano Zampini @*/
3502b510759SStefano Zampini PetscErrorCode PCBDDCSetLevels(PC pc,PetscInt levels)
3514fad6a16SStefano Zampini {
3524fad6a16SStefano Zampini   PetscErrorCode ierr;
3534fad6a16SStefano Zampini 
3544fad6a16SStefano Zampini   PetscFunctionBegin;
3554fad6a16SStefano Zampini   PetscValidHeaderSpecific(pc,PC_CLASSID,1);
3562b510759SStefano Zampini   PetscValidLogicalCollectiveInt(pc,levels,2);
357312be037SStefano Zampini   if (levels > 99) SETERRQ(PetscObjectComm((PetscObject)pc),PETSC_ERR_SUP,"Maximum number of levels for bddc is 99\n");
3582b510759SStefano Zampini   ierr = PetscTryMethod(pc,"PCBDDCSetLevels_C",(PC,PetscInt),(pc,levels));CHKERRQ(ierr);
3594fad6a16SStefano Zampini   PetscFunctionReturn(0);
3604fad6a16SStefano Zampini }
3614fad6a16SStefano Zampini /* -------------------------------------------------------------------------- */
3621e6b0712SBarry Smith 
3634fad6a16SStefano Zampini #undef __FUNCT__
3643b03a366Sstefano_zampini #define __FUNCT__ "PCBDDCSetDirichletBoundaries_BDDC"
3653b03a366Sstefano_zampini static PetscErrorCode PCBDDCSetDirichletBoundaries_BDDC(PC pc,IS DirichletBoundaries)
3663b03a366Sstefano_zampini {
3673b03a366Sstefano_zampini   PC_BDDC  *pcbddc = (PC_BDDC*)pc->data;
3683b03a366Sstefano_zampini   PetscErrorCode ierr;
3693b03a366Sstefano_zampini 
3703b03a366Sstefano_zampini   PetscFunctionBegin;
371785d1243SStefano Zampini   /* last user setting takes precendence -> destroy any other customization */
372785d1243SStefano Zampini   ierr = ISDestroy(&pcbddc->DirichletBoundariesLocal);CHKERRQ(ierr);
3733b03a366Sstefano_zampini   ierr = ISDestroy(&pcbddc->DirichletBoundaries);CHKERRQ(ierr);
37436e030ebSStefano Zampini   ierr = PetscObjectReference((PetscObject)DirichletBoundaries);CHKERRQ(ierr);
37536e030ebSStefano Zampini   pcbddc->DirichletBoundaries = DirichletBoundaries;
376fb180af4SStefano Zampini   pcbddc->recompute_topography = PETSC_TRUE;
3773b03a366Sstefano_zampini   PetscFunctionReturn(0);
3783b03a366Sstefano_zampini }
3791e6b0712SBarry Smith 
3803b03a366Sstefano_zampini #undef __FUNCT__
3813b03a366Sstefano_zampini #define __FUNCT__ "PCBDDCSetDirichletBoundaries"
3823b03a366Sstefano_zampini /*@
38328509bceSStefano Zampini  PCBDDCSetDirichletBoundaries - Set IS defining Dirichlet boundaries for the global problem.
3843b03a366Sstefano_zampini 
385785d1243SStefano Zampini    Collective
3863b03a366Sstefano_zampini 
3873b03a366Sstefano_zampini    Input Parameters:
3883b03a366Sstefano_zampini +  pc - the preconditioning context
389785d1243SStefano Zampini -  DirichletBoundaries - parallel IS defining the Dirichlet boundaries
3903b03a366Sstefano_zampini 
3913b03a366Sstefano_zampini    Level: intermediate
3923b03a366Sstefano_zampini 
3930f202f7eSStefano Zampini    Notes:
3940f202f7eSStefano Zampini      Provide the information if you used MatZeroRows/Columns routines. Any process can list any global node
3953b03a366Sstefano_zampini 
3960f202f7eSStefano Zampini .seealso: PCBDDC, PCBDDCSetDirichletBoundariesLocal(), MatZeroRows(), MatZeroRowsColumns()
3973b03a366Sstefano_zampini @*/
3983b03a366Sstefano_zampini PetscErrorCode PCBDDCSetDirichletBoundaries(PC pc,IS DirichletBoundaries)
3993b03a366Sstefano_zampini {
4003b03a366Sstefano_zampini   PetscErrorCode ierr;
4013b03a366Sstefano_zampini 
4023b03a366Sstefano_zampini   PetscFunctionBegin;
4033b03a366Sstefano_zampini   PetscValidHeaderSpecific(pc,PC_CLASSID,1);
404674ae819SStefano Zampini   PetscValidHeaderSpecific(DirichletBoundaries,IS_CLASSID,2);
405785d1243SStefano Zampini   PetscCheckSameComm(pc,1,DirichletBoundaries,2);
4063b03a366Sstefano_zampini   ierr = PetscTryMethod(pc,"PCBDDCSetDirichletBoundaries_C",(PC,IS),(pc,DirichletBoundaries));CHKERRQ(ierr);
4073b03a366Sstefano_zampini   PetscFunctionReturn(0);
4083b03a366Sstefano_zampini }
4093b03a366Sstefano_zampini /* -------------------------------------------------------------------------- */
4101e6b0712SBarry Smith 
4113b03a366Sstefano_zampini #undef __FUNCT__
41282ba6b80SStefano Zampini #define __FUNCT__ "PCBDDCSetDirichletBoundariesLocal_BDDC"
41382ba6b80SStefano Zampini static PetscErrorCode PCBDDCSetDirichletBoundariesLocal_BDDC(PC pc,IS DirichletBoundaries)
4140c7d97c5SJed Brown {
4150c7d97c5SJed Brown   PC_BDDC  *pcbddc = (PC_BDDC*)pc->data;
4160c7d97c5SJed Brown   PetscErrorCode ierr;
4170c7d97c5SJed Brown 
4180c7d97c5SJed Brown   PetscFunctionBegin;
419785d1243SStefano Zampini   /* last user setting takes precendence -> destroy any other customization */
420785d1243SStefano Zampini   ierr = ISDestroy(&pcbddc->DirichletBoundariesLocal);CHKERRQ(ierr);
4210c7d97c5SJed Brown   ierr = ISDestroy(&pcbddc->DirichletBoundaries);CHKERRQ(ierr);
4220c7d97c5SJed Brown   ierr = PetscObjectReference((PetscObject)DirichletBoundaries);CHKERRQ(ierr);
423785d1243SStefano Zampini   pcbddc->DirichletBoundariesLocal = DirichletBoundaries;
4247d24d155SStefano Zampini   pcbddc->recompute_topography = PETSC_TRUE;
4250c7d97c5SJed Brown   PetscFunctionReturn(0);
4260c7d97c5SJed Brown }
4270c7d97c5SJed Brown 
4280c7d97c5SJed Brown #undef __FUNCT__
42982ba6b80SStefano Zampini #define __FUNCT__ "PCBDDCSetDirichletBoundariesLocal"
4309c0446d6SStefano Zampini /*@
43182ba6b80SStefano Zampini  PCBDDCSetDirichletBoundariesLocal - Set IS defining Dirichlet boundaries for the global problem in local ordering.
4329c0446d6SStefano Zampini 
433785d1243SStefano Zampini    Collective
43453cdbc3dSStefano Zampini 
43553cdbc3dSStefano Zampini    Input Parameters:
43653cdbc3dSStefano Zampini +  pc - the preconditioning context
43782ba6b80SStefano Zampini -  DirichletBoundaries - parallel IS defining the Dirichlet boundaries (in local ordering)
43853cdbc3dSStefano Zampini 
43953cdbc3dSStefano Zampini    Level: intermediate
44053cdbc3dSStefano Zampini 
4419c0446d6SStefano Zampini    Notes:
44253cdbc3dSStefano Zampini 
4430f202f7eSStefano Zampini .seealso: PCBDDC, PCBDDCSetDirichletBoundaries(), MatZeroRows(), MatZeroRowsColumns()
44453cdbc3dSStefano Zampini @*/
44582ba6b80SStefano Zampini PetscErrorCode PCBDDCSetDirichletBoundariesLocal(PC pc,IS DirichletBoundaries)
4460c7d97c5SJed Brown {
4470c7d97c5SJed Brown   PetscErrorCode ierr;
4480c7d97c5SJed Brown 
4490c7d97c5SJed Brown   PetscFunctionBegin;
4500c7d97c5SJed Brown   PetscValidHeaderSpecific(pc,PC_CLASSID,1);
4510c7d97c5SJed Brown   PetscValidHeaderSpecific(DirichletBoundaries,IS_CLASSID,2);
45282ba6b80SStefano Zampini   PetscCheckSameComm(pc,1,DirichletBoundaries,2);
45382ba6b80SStefano Zampini   ierr = PetscTryMethod(pc,"PCBDDCSetDirichletBoundariesLocal_C",(PC,IS),(pc,DirichletBoundaries));CHKERRQ(ierr);
4540c7d97c5SJed Brown   PetscFunctionReturn(0);
4550c7d97c5SJed Brown }
4560c7d97c5SJed Brown /* -------------------------------------------------------------------------- */
4570c7d97c5SJed Brown 
4580c7d97c5SJed Brown #undef __FUNCT__
4590c7d97c5SJed Brown #define __FUNCT__ "PCBDDCSetNeumannBoundaries_BDDC"
46053cdbc3dSStefano Zampini static PetscErrorCode PCBDDCSetNeumannBoundaries_BDDC(PC pc,IS NeumannBoundaries)
4610c7d97c5SJed Brown {
4620c7d97c5SJed Brown   PC_BDDC  *pcbddc = (PC_BDDC*)pc->data;
46353cdbc3dSStefano Zampini   PetscErrorCode ierr;
4640c7d97c5SJed Brown 
4650c7d97c5SJed Brown   PetscFunctionBegin;
466785d1243SStefano Zampini   /* last user setting takes precendence -> destroy any other customization */
467785d1243SStefano Zampini   ierr = ISDestroy(&pcbddc->NeumannBoundariesLocal);CHKERRQ(ierr);
46853cdbc3dSStefano Zampini   ierr = ISDestroy(&pcbddc->NeumannBoundaries);CHKERRQ(ierr);
46936e030ebSStefano Zampini   ierr = PetscObjectReference((PetscObject)NeumannBoundaries);CHKERRQ(ierr);
47036e030ebSStefano Zampini   pcbddc->NeumannBoundaries = NeumannBoundaries;
471fb180af4SStefano Zampini   pcbddc->recompute_topography = PETSC_TRUE;
4720c7d97c5SJed Brown   PetscFunctionReturn(0);
4730c7d97c5SJed Brown }
4741e6b0712SBarry Smith 
4750c7d97c5SJed Brown #undef __FUNCT__
4760c7d97c5SJed Brown #define __FUNCT__ "PCBDDCSetNeumannBoundaries"
47757527edcSJed Brown /*@
47828509bceSStefano Zampini  PCBDDCSetNeumannBoundaries - Set IS defining Neumann boundaries for the global problem.
47957527edcSJed Brown 
480785d1243SStefano Zampini    Collective
48157527edcSJed Brown 
48257527edcSJed Brown    Input Parameters:
48357527edcSJed Brown +  pc - the preconditioning context
484785d1243SStefano Zampini -  NeumannBoundaries - parallel IS defining the Neumann boundaries
48557527edcSJed Brown 
48657527edcSJed Brown    Level: intermediate
48757527edcSJed Brown 
4880f202f7eSStefano Zampini    Notes:
4890f202f7eSStefano Zampini      Any process can list any global node
49057527edcSJed Brown 
4910f202f7eSStefano Zampini .seealso: PCBDDC, PCBDDCSetNeumannBoundariesLocal()
49257527edcSJed Brown @*/
49353cdbc3dSStefano Zampini PetscErrorCode PCBDDCSetNeumannBoundaries(PC pc,IS NeumannBoundaries)
4940c7d97c5SJed Brown {
4950c7d97c5SJed Brown   PetscErrorCode ierr;
4960c7d97c5SJed Brown 
4970c7d97c5SJed Brown   PetscFunctionBegin;
4980c7d97c5SJed Brown   PetscValidHeaderSpecific(pc,PC_CLASSID,1);
499674ae819SStefano Zampini   PetscValidHeaderSpecific(NeumannBoundaries,IS_CLASSID,2);
500785d1243SStefano Zampini   PetscCheckSameComm(pc,1,NeumannBoundaries,2);
50153cdbc3dSStefano Zampini   ierr = PetscTryMethod(pc,"PCBDDCSetNeumannBoundaries_C",(PC,IS),(pc,NeumannBoundaries));CHKERRQ(ierr);
50253cdbc3dSStefano Zampini   PetscFunctionReturn(0);
50353cdbc3dSStefano Zampini }
50453cdbc3dSStefano Zampini /* -------------------------------------------------------------------------- */
5051e6b0712SBarry Smith 
50653cdbc3dSStefano Zampini #undef __FUNCT__
50782ba6b80SStefano Zampini #define __FUNCT__ "PCBDDCSetNeumannBoundariesLocal_BDDC"
50882ba6b80SStefano Zampini static PetscErrorCode PCBDDCSetNeumannBoundariesLocal_BDDC(PC pc,IS NeumannBoundaries)
5090c7d97c5SJed Brown {
5100c7d97c5SJed Brown   PC_BDDC  *pcbddc = (PC_BDDC*)pc->data;
5110c7d97c5SJed Brown   PetscErrorCode ierr;
5120c7d97c5SJed Brown 
5130c7d97c5SJed Brown   PetscFunctionBegin;
514785d1243SStefano Zampini   /* last user setting takes precendence -> destroy any other customization */
515785d1243SStefano Zampini   ierr = ISDestroy(&pcbddc->NeumannBoundariesLocal);CHKERRQ(ierr);
5160c7d97c5SJed Brown   ierr = ISDestroy(&pcbddc->NeumannBoundaries);CHKERRQ(ierr);
5170c7d97c5SJed Brown   ierr = PetscObjectReference((PetscObject)NeumannBoundaries);CHKERRQ(ierr);
518785d1243SStefano Zampini   pcbddc->NeumannBoundariesLocal = NeumannBoundaries;
5197d24d155SStefano Zampini   pcbddc->recompute_topography = PETSC_TRUE;
5200c7d97c5SJed Brown   PetscFunctionReturn(0);
5210c7d97c5SJed Brown }
5220c7d97c5SJed Brown 
5230c7d97c5SJed Brown #undef __FUNCT__
52482ba6b80SStefano Zampini #define __FUNCT__ "PCBDDCSetNeumannBoundariesLocal"
5250c7d97c5SJed Brown /*@
52682ba6b80SStefano Zampini  PCBDDCSetNeumannBoundariesLocal - Set IS defining Neumann boundaries for the global problem in local ordering.
5270c7d97c5SJed Brown 
528785d1243SStefano Zampini    Collective
5290c7d97c5SJed Brown 
5300c7d97c5SJed Brown    Input Parameters:
5310c7d97c5SJed Brown +  pc - the preconditioning context
53282ba6b80SStefano Zampini -  NeumannBoundaries - parallel IS defining the subdomain part of Neumann boundaries (in local ordering)
5330c7d97c5SJed Brown 
5340c7d97c5SJed Brown    Level: intermediate
5350c7d97c5SJed Brown 
5360c7d97c5SJed Brown    Notes:
5370c7d97c5SJed Brown 
5380f202f7eSStefano Zampini .seealso: PCBDDC, PCBDDCSetNeumannBoundaries()
5390c7d97c5SJed Brown @*/
54082ba6b80SStefano Zampini PetscErrorCode PCBDDCSetNeumannBoundariesLocal(PC pc,IS NeumannBoundaries)
5410c7d97c5SJed Brown {
5420c7d97c5SJed Brown   PetscErrorCode ierr;
5430c7d97c5SJed Brown 
5440c7d97c5SJed Brown   PetscFunctionBegin;
5450c7d97c5SJed Brown   PetscValidHeaderSpecific(pc,PC_CLASSID,1);
5460c7d97c5SJed Brown   PetscValidHeaderSpecific(NeumannBoundaries,IS_CLASSID,2);
54782ba6b80SStefano Zampini   PetscCheckSameComm(pc,1,NeumannBoundaries,2);
54882ba6b80SStefano Zampini   ierr = PetscTryMethod(pc,"PCBDDCSetNeumannBoundariesLocal_C",(PC,IS),(pc,NeumannBoundaries));CHKERRQ(ierr);
54953cdbc3dSStefano Zampini   PetscFunctionReturn(0);
55053cdbc3dSStefano Zampini }
55153cdbc3dSStefano Zampini /* -------------------------------------------------------------------------- */
55253cdbc3dSStefano Zampini 
55353cdbc3dSStefano Zampini #undef __FUNCT__
554da1bb401SStefano Zampini #define __FUNCT__ "PCBDDCGetDirichletBoundaries_BDDC"
555da1bb401SStefano Zampini static PetscErrorCode PCBDDCGetDirichletBoundaries_BDDC(PC pc,IS *DirichletBoundaries)
556da1bb401SStefano Zampini {
557da1bb401SStefano Zampini   PC_BDDC  *pcbddc = (PC_BDDC*)pc->data;
558da1bb401SStefano Zampini 
559da1bb401SStefano Zampini   PetscFunctionBegin;
560da1bb401SStefano Zampini   *DirichletBoundaries = pcbddc->DirichletBoundaries;
561da1bb401SStefano Zampini   PetscFunctionReturn(0);
562da1bb401SStefano Zampini }
5631e6b0712SBarry Smith 
564da1bb401SStefano Zampini #undef __FUNCT__
565da1bb401SStefano Zampini #define __FUNCT__ "PCBDDCGetDirichletBoundaries"
566da1bb401SStefano Zampini /*@
567785d1243SStefano Zampini  PCBDDCGetDirichletBoundaries - Get parallel IS for Dirichlet boundaries
568da1bb401SStefano Zampini 
569785d1243SStefano Zampini    Collective
570785d1243SStefano Zampini 
571785d1243SStefano Zampini    Input Parameters:
572785d1243SStefano Zampini .  pc - the preconditioning context
573785d1243SStefano Zampini 
574785d1243SStefano Zampini    Output Parameters:
575785d1243SStefano Zampini .  DirichletBoundaries - index set defining the Dirichlet boundaries
576785d1243SStefano Zampini 
577785d1243SStefano Zampini    Level: intermediate
578785d1243SStefano Zampini 
5790f202f7eSStefano Zampini    Notes:
5800f202f7eSStefano Zampini      The IS returned (if any) is the same passed in earlier by the user with PCBDDCSetDirichletBoundaries
581785d1243SStefano Zampini 
582785d1243SStefano Zampini .seealso: PCBDDC
583785d1243SStefano Zampini @*/
584785d1243SStefano Zampini PetscErrorCode PCBDDCGetDirichletBoundaries(PC pc,IS *DirichletBoundaries)
585785d1243SStefano Zampini {
586785d1243SStefano Zampini   PetscErrorCode ierr;
587785d1243SStefano Zampini 
588785d1243SStefano Zampini   PetscFunctionBegin;
589785d1243SStefano Zampini   PetscValidHeaderSpecific(pc,PC_CLASSID,1);
590785d1243SStefano Zampini   ierr = PetscUseMethod(pc,"PCBDDCGetDirichletBoundaries_C",(PC,IS*),(pc,DirichletBoundaries));CHKERRQ(ierr);
591785d1243SStefano Zampini   PetscFunctionReturn(0);
592785d1243SStefano Zampini }
593785d1243SStefano Zampini /* -------------------------------------------------------------------------- */
594785d1243SStefano Zampini 
595785d1243SStefano Zampini #undef __FUNCT__
596785d1243SStefano Zampini #define __FUNCT__ "PCBDDCGetDirichletBoundariesLocal_BDDC"
597785d1243SStefano Zampini static PetscErrorCode PCBDDCGetDirichletBoundariesLocal_BDDC(PC pc,IS *DirichletBoundaries)
598785d1243SStefano Zampini {
599785d1243SStefano Zampini   PC_BDDC  *pcbddc = (PC_BDDC*)pc->data;
600785d1243SStefano Zampini 
601785d1243SStefano Zampini   PetscFunctionBegin;
602785d1243SStefano Zampini   *DirichletBoundaries = pcbddc->DirichletBoundariesLocal;
603785d1243SStefano Zampini   PetscFunctionReturn(0);
604785d1243SStefano Zampini }
605785d1243SStefano Zampini 
606785d1243SStefano Zampini #undef __FUNCT__
60782ba6b80SStefano Zampini #define __FUNCT__ "PCBDDCGetDirichletBoundariesLocal"
608da1bb401SStefano Zampini /*@
60982ba6b80SStefano Zampini  PCBDDCGetDirichletBoundariesLocal - Get parallel IS for Dirichlet boundaries (in local ordering)
610da1bb401SStefano Zampini 
611785d1243SStefano Zampini    Collective
612da1bb401SStefano Zampini 
613da1bb401SStefano Zampini    Input Parameters:
61428509bceSStefano Zampini .  pc - the preconditioning context
615da1bb401SStefano Zampini 
616da1bb401SStefano Zampini    Output Parameters:
61728509bceSStefano Zampini .  DirichletBoundaries - index set defining the subdomain part of Dirichlet boundaries
618da1bb401SStefano Zampini 
619da1bb401SStefano Zampini    Level: intermediate
620da1bb401SStefano Zampini 
621da1bb401SStefano Zampini    Notes:
6220f202f7eSStefano 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).
6230f202f7eSStefano Zampini           In the latter case, the IS will be available after PCSetUp.
624da1bb401SStefano Zampini 
625da1bb401SStefano Zampini .seealso: PCBDDC
626da1bb401SStefano Zampini @*/
62782ba6b80SStefano Zampini PetscErrorCode PCBDDCGetDirichletBoundariesLocal(PC pc,IS *DirichletBoundaries)
628da1bb401SStefano Zampini {
629da1bb401SStefano Zampini   PetscErrorCode ierr;
630da1bb401SStefano Zampini 
631da1bb401SStefano Zampini   PetscFunctionBegin;
632da1bb401SStefano Zampini   PetscValidHeaderSpecific(pc,PC_CLASSID,1);
63382ba6b80SStefano Zampini   ierr = PetscUseMethod(pc,"PCBDDCGetDirichletBoundariesLocal_C",(PC,IS*),(pc,DirichletBoundaries));CHKERRQ(ierr);
634da1bb401SStefano Zampini   PetscFunctionReturn(0);
635da1bb401SStefano Zampini }
636da1bb401SStefano Zampini /* -------------------------------------------------------------------------- */
6371e6b0712SBarry Smith 
638da1bb401SStefano Zampini #undef __FUNCT__
63953cdbc3dSStefano Zampini #define __FUNCT__ "PCBDDCGetNeumannBoundaries_BDDC"
64053cdbc3dSStefano Zampini static PetscErrorCode PCBDDCGetNeumannBoundaries_BDDC(PC pc,IS *NeumannBoundaries)
64153cdbc3dSStefano Zampini {
64253cdbc3dSStefano Zampini   PC_BDDC  *pcbddc = (PC_BDDC*)pc->data;
64353cdbc3dSStefano Zampini 
64453cdbc3dSStefano Zampini   PetscFunctionBegin;
64553cdbc3dSStefano Zampini   *NeumannBoundaries = pcbddc->NeumannBoundaries;
64653cdbc3dSStefano Zampini   PetscFunctionReturn(0);
64753cdbc3dSStefano Zampini }
6481e6b0712SBarry Smith 
64953cdbc3dSStefano Zampini #undef __FUNCT__
65053cdbc3dSStefano Zampini #define __FUNCT__ "PCBDDCGetNeumannBoundaries"
65153cdbc3dSStefano Zampini /*@
652785d1243SStefano Zampini  PCBDDCGetNeumannBoundaries - Get parallel IS for Neumann boundaries
65353cdbc3dSStefano Zampini 
654785d1243SStefano Zampini    Collective
655785d1243SStefano Zampini 
656785d1243SStefano Zampini    Input Parameters:
657785d1243SStefano Zampini .  pc - the preconditioning context
658785d1243SStefano Zampini 
659785d1243SStefano Zampini    Output Parameters:
660785d1243SStefano Zampini .  NeumannBoundaries - index set defining the Neumann boundaries
661785d1243SStefano Zampini 
662785d1243SStefano Zampini    Level: intermediate
663785d1243SStefano Zampini 
6640f202f7eSStefano Zampini    Notes:
6650f202f7eSStefano Zampini      The IS returned (if any) is the same passed in earlier by the user with PCBDDCSetNeumannBoundaries
666785d1243SStefano Zampini 
667785d1243SStefano Zampini .seealso: PCBDDC
668785d1243SStefano Zampini @*/
669785d1243SStefano Zampini PetscErrorCode PCBDDCGetNeumannBoundaries(PC pc,IS *NeumannBoundaries)
670785d1243SStefano Zampini {
671785d1243SStefano Zampini   PetscErrorCode ierr;
672785d1243SStefano Zampini 
673785d1243SStefano Zampini   PetscFunctionBegin;
674785d1243SStefano Zampini   PetscValidHeaderSpecific(pc,PC_CLASSID,1);
675785d1243SStefano Zampini   ierr = PetscUseMethod(pc,"PCBDDCGetNeumannBoundaries_C",(PC,IS*),(pc,NeumannBoundaries));CHKERRQ(ierr);
676785d1243SStefano Zampini   PetscFunctionReturn(0);
677785d1243SStefano Zampini }
678785d1243SStefano Zampini /* -------------------------------------------------------------------------- */
679785d1243SStefano Zampini 
680785d1243SStefano Zampini #undef __FUNCT__
681785d1243SStefano Zampini #define __FUNCT__ "PCBDDCGetNeumannBoundariesLocal_BDDC"
682785d1243SStefano Zampini static PetscErrorCode PCBDDCGetNeumannBoundariesLocal_BDDC(PC pc,IS *NeumannBoundaries)
683785d1243SStefano Zampini {
684785d1243SStefano Zampini   PC_BDDC  *pcbddc = (PC_BDDC*)pc->data;
685785d1243SStefano Zampini 
686785d1243SStefano Zampini   PetscFunctionBegin;
687785d1243SStefano Zampini   *NeumannBoundaries = pcbddc->NeumannBoundariesLocal;
688785d1243SStefano Zampini   PetscFunctionReturn(0);
689785d1243SStefano Zampini }
690785d1243SStefano Zampini 
691785d1243SStefano Zampini #undef __FUNCT__
69282ba6b80SStefano Zampini #define __FUNCT__ "PCBDDCGetNeumannBoundariesLocal"
69353cdbc3dSStefano Zampini /*@
69482ba6b80SStefano Zampini  PCBDDCGetNeumannBoundariesLocal - Get parallel IS for Neumann boundaries (in local ordering)
69553cdbc3dSStefano Zampini 
696785d1243SStefano Zampini    Collective
69753cdbc3dSStefano Zampini 
69853cdbc3dSStefano Zampini    Input Parameters:
69928509bceSStefano Zampini .  pc - the preconditioning context
70053cdbc3dSStefano Zampini 
70153cdbc3dSStefano Zampini    Output Parameters:
70228509bceSStefano Zampini .  NeumannBoundaries - index set defining the subdomain part of Neumann boundaries
70353cdbc3dSStefano Zampini 
70453cdbc3dSStefano Zampini    Level: intermediate
70553cdbc3dSStefano Zampini 
70653cdbc3dSStefano Zampini    Notes:
7070f202f7eSStefano 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).
7080f202f7eSStefano Zampini           In the latter case, the IS will be available after PCSetUp.
70953cdbc3dSStefano Zampini 
71053cdbc3dSStefano Zampini .seealso: PCBDDC
71153cdbc3dSStefano Zampini @*/
71282ba6b80SStefano Zampini PetscErrorCode PCBDDCGetNeumannBoundariesLocal(PC pc,IS *NeumannBoundaries)
71353cdbc3dSStefano Zampini {
71453cdbc3dSStefano Zampini   PetscErrorCode ierr;
71553cdbc3dSStefano Zampini 
71653cdbc3dSStefano Zampini   PetscFunctionBegin;
71753cdbc3dSStefano Zampini   PetscValidHeaderSpecific(pc,PC_CLASSID,1);
71882ba6b80SStefano Zampini   ierr = PetscUseMethod(pc,"PCBDDCGetNeumannBoundariesLocal_C",(PC,IS*),(pc,NeumannBoundaries));CHKERRQ(ierr);
7190c7d97c5SJed Brown   PetscFunctionReturn(0);
7200c7d97c5SJed Brown }
72136e030ebSStefano Zampini /* -------------------------------------------------------------------------- */
7221e6b0712SBarry Smith 
72336e030ebSStefano Zampini #undef __FUNCT__
724da1bb401SStefano Zampini #define __FUNCT__ "PCBDDCSetLocalAdjacencyGraph_BDDC"
7251a83f524SJed Brown static PetscErrorCode PCBDDCSetLocalAdjacencyGraph_BDDC(PC pc, PetscInt nvtxs,const PetscInt xadj[],const PetscInt adjncy[], PetscCopyMode copymode)
72636e030ebSStefano Zampini {
72736e030ebSStefano Zampini   PC_BDDC        *pcbddc = (PC_BDDC*)pc->data;
728da1bb401SStefano Zampini   PCBDDCGraph    mat_graph = pcbddc->mat_graph;
729da1bb401SStefano Zampini   PetscErrorCode ierr;
73036e030ebSStefano Zampini 
73136e030ebSStefano Zampini   PetscFunctionBegin;
732674ae819SStefano Zampini   /* free old CSR */
733674ae819SStefano Zampini   ierr = PCBDDCGraphResetCSR(mat_graph);CHKERRQ(ierr);
734fb223d50SStefano Zampini   /* TODO: PCBDDCGraphSetAdjacency */
735674ae819SStefano Zampini   /* get CSR into graph structure */
736da1bb401SStefano Zampini   if (copymode == PETSC_COPY_VALUES) {
737854ce69bSBarry Smith     ierr = PetscMalloc1(nvtxs+1,&mat_graph->xadj);CHKERRQ(ierr);
738785e854fSJed Brown     ierr = PetscMalloc1(xadj[nvtxs],&mat_graph->adjncy);CHKERRQ(ierr);
739674ae819SStefano Zampini     ierr = PetscMemcpy(mat_graph->xadj,xadj,(nvtxs+1)*sizeof(PetscInt));CHKERRQ(ierr);
740674ae819SStefano Zampini     ierr = PetscMemcpy(mat_graph->adjncy,adjncy,xadj[nvtxs]*sizeof(PetscInt));CHKERRQ(ierr);
741da1bb401SStefano Zampini   } else if (copymode == PETSC_OWN_POINTER) {
7421a83f524SJed Brown     mat_graph->xadj = (PetscInt*)xadj;
7431a83f524SJed Brown     mat_graph->adjncy = (PetscInt*)adjncy;
744674ae819SStefano Zampini   }
745575ad6abSStefano Zampini   mat_graph->nvtxs_csr = nvtxs;
74636e030ebSStefano Zampini   PetscFunctionReturn(0);
74736e030ebSStefano Zampini }
7481e6b0712SBarry Smith 
74936e030ebSStefano Zampini #undef __FUNCT__
750da1bb401SStefano Zampini #define __FUNCT__ "PCBDDCSetLocalAdjacencyGraph"
75136e030ebSStefano Zampini /*@
7520f202f7eSStefano Zampini  PCBDDCSetLocalAdjacencyGraph - Set adjacency structure (CSR graph) of the local matrix
75336e030ebSStefano Zampini 
75436e030ebSStefano Zampini    Not collective
75536e030ebSStefano Zampini 
75636e030ebSStefano Zampini    Input Parameters:
75736e030ebSStefano Zampini +  pc - the preconditioning context
7580f202f7eSStefano Zampini .  nvtxs - number of local vertices of the graph (i.e., the size of the local problem)
75928509bceSStefano Zampini .  xadj, adjncy - the CSR graph
76028509bceSStefano Zampini -  copymode - either PETSC_COPY_VALUES or PETSC_OWN_POINTER.
76136e030ebSStefano Zampini 
76236e030ebSStefano Zampini    Level: intermediate
76336e030ebSStefano Zampini 
76436e030ebSStefano Zampini    Notes:
76536e030ebSStefano Zampini 
76628509bceSStefano Zampini .seealso: PCBDDC,PetscCopyMode
76736e030ebSStefano Zampini @*/
7681a83f524SJed Brown PetscErrorCode PCBDDCSetLocalAdjacencyGraph(PC pc,PetscInt nvtxs,const PetscInt xadj[],const PetscInt adjncy[], PetscCopyMode copymode)
76936e030ebSStefano Zampini {
770575ad6abSStefano Zampini   void (*f)(void) = 0;
77136e030ebSStefano Zampini   PetscErrorCode ierr;
77236e030ebSStefano Zampini 
77336e030ebSStefano Zampini   PetscFunctionBegin;
77436e030ebSStefano Zampini   PetscValidHeaderSpecific(pc,PC_CLASSID,1);
775674ae819SStefano Zampini   PetscValidIntPointer(xadj,3);
776d7de1dd9SStefano Zampini   PetscValidIntPointer(adjncy,4);
7776c4ed002SBarry Smith   if (copymode != PETSC_COPY_VALUES && copymode != PETSC_OWN_POINTER)  SETERRQ1(PETSC_COMM_SELF,PETSC_ERR_SUP,"Unsupported copy mode %d",copymode);
7781a83f524SJed Brown   ierr = PetscTryMethod(pc,"PCBDDCSetLocalAdjacencyGraph_C",(PC,PetscInt,const PetscInt[],const PetscInt[],PetscCopyMode),(pc,nvtxs,xadj,adjncy,copymode));CHKERRQ(ierr);
779575ad6abSStefano Zampini   /* free arrays if PCBDDC is not the PC type */
780575ad6abSStefano Zampini   ierr = PetscObjectQueryFunction((PetscObject)pc,"PCBDDCSetLocalAdjacencyGraph_C",&f);CHKERRQ(ierr);
781575ad6abSStefano Zampini   if (!f && copymode == PETSC_OWN_POINTER) {
782575ad6abSStefano Zampini     ierr = PetscFree(xadj);CHKERRQ(ierr);
783575ad6abSStefano Zampini     ierr = PetscFree(adjncy);CHKERRQ(ierr);
784da1bb401SStefano Zampini   }
78536e030ebSStefano Zampini   PetscFunctionReturn(0);
78636e030ebSStefano Zampini }
7879c0446d6SStefano Zampini /* -------------------------------------------------------------------------- */
7881e6b0712SBarry Smith 
7899c0446d6SStefano Zampini #undef __FUNCT__
79063602bcaSStefano Zampini #define __FUNCT__ "PCBDDCSetDofsSplittingLocal_BDDC"
79163602bcaSStefano Zampini static PetscErrorCode PCBDDCSetDofsSplittingLocal_BDDC(PC pc,PetscInt n_is, IS ISForDofs[])
79263602bcaSStefano Zampini {
79363602bcaSStefano Zampini   PC_BDDC  *pcbddc = (PC_BDDC*)pc->data;
79463602bcaSStefano Zampini   PetscInt i;
79563602bcaSStefano Zampini   PetscErrorCode ierr;
79663602bcaSStefano Zampini 
79763602bcaSStefano Zampini   PetscFunctionBegin;
79863602bcaSStefano Zampini   /* Destroy ISes if they were already set */
79963602bcaSStefano Zampini   for (i=0;i<pcbddc->n_ISForDofsLocal;i++) {
80063602bcaSStefano Zampini     ierr = ISDestroy(&pcbddc->ISForDofsLocal[i]);CHKERRQ(ierr);
80163602bcaSStefano Zampini   }
80263602bcaSStefano Zampini   ierr = PetscFree(pcbddc->ISForDofsLocal);CHKERRQ(ierr);
80363602bcaSStefano Zampini   /* last user setting takes precendence -> destroy any other customization */
80463602bcaSStefano Zampini   for (i=0;i<pcbddc->n_ISForDofs;i++) {
80563602bcaSStefano Zampini     ierr = ISDestroy(&pcbddc->ISForDofs[i]);CHKERRQ(ierr);
80663602bcaSStefano Zampini   }
80763602bcaSStefano Zampini   ierr = PetscFree(pcbddc->ISForDofs);CHKERRQ(ierr);
80863602bcaSStefano Zampini   pcbddc->n_ISForDofs = 0;
80963602bcaSStefano Zampini   /* allocate space then set */
810d02579f5SStefano Zampini   if (n_is) {
811d02579f5SStefano Zampini     ierr = PetscMalloc1(n_is,&pcbddc->ISForDofsLocal);CHKERRQ(ierr);
812d02579f5SStefano Zampini   }
81363602bcaSStefano Zampini   for (i=0;i<n_is;i++) {
81463602bcaSStefano Zampini     ierr = PetscObjectReference((PetscObject)ISForDofs[i]);CHKERRQ(ierr);
81563602bcaSStefano Zampini     pcbddc->ISForDofsLocal[i]=ISForDofs[i];
81663602bcaSStefano Zampini   }
81763602bcaSStefano Zampini   pcbddc->n_ISForDofsLocal=n_is;
81863602bcaSStefano Zampini   if (n_is) pcbddc->user_provided_isfordofs = PETSC_TRUE;
8191a8a16d1SStefano Zampini   pcbddc->recompute_topography = PETSC_TRUE;
82063602bcaSStefano Zampini   PetscFunctionReturn(0);
82163602bcaSStefano Zampini }
82263602bcaSStefano Zampini 
82363602bcaSStefano Zampini #undef __FUNCT__
82463602bcaSStefano Zampini #define __FUNCT__ "PCBDDCSetDofsSplittingLocal"
82563602bcaSStefano Zampini /*@
82663602bcaSStefano Zampini  PCBDDCSetDofsSplittingLocal - Set index sets defining fields of the local subdomain matrix
82763602bcaSStefano Zampini 
82863602bcaSStefano Zampini    Collective
82963602bcaSStefano Zampini 
83063602bcaSStefano Zampini    Input Parameters:
83163602bcaSStefano Zampini +  pc - the preconditioning context
8320f202f7eSStefano Zampini .  n_is - number of index sets defining the fields
8330f202f7eSStefano Zampini -  ISForDofs - array of IS describing the fields in local ordering
83463602bcaSStefano Zampini 
83563602bcaSStefano Zampini    Level: intermediate
83663602bcaSStefano Zampini 
8370f202f7eSStefano Zampini    Notes:
8380f202f7eSStefano Zampini      n_is should be the same among processes. Not all nodes need to be listed: unlisted nodes will belong to the complement field.
83963602bcaSStefano Zampini 
84063602bcaSStefano Zampini .seealso: PCBDDC
84163602bcaSStefano Zampini @*/
84263602bcaSStefano Zampini PetscErrorCode PCBDDCSetDofsSplittingLocal(PC pc,PetscInt n_is, IS ISForDofs[])
84363602bcaSStefano Zampini {
84463602bcaSStefano Zampini   PetscInt       i;
84563602bcaSStefano Zampini   PetscErrorCode ierr;
84663602bcaSStefano Zampini 
84763602bcaSStefano Zampini   PetscFunctionBegin;
84863602bcaSStefano Zampini   PetscValidHeaderSpecific(pc,PC_CLASSID,1);
84963602bcaSStefano Zampini   PetscValidLogicalCollectiveInt(pc,n_is,2);
85063602bcaSStefano Zampini   for (i=0;i<n_is;i++) {
85163602bcaSStefano Zampini     PetscCheckSameComm(pc,1,ISForDofs[i],3);
85263602bcaSStefano Zampini     PetscValidHeaderSpecific(ISForDofs[i],IS_CLASSID,3);
85363602bcaSStefano Zampini   }
854e71e7a71SStefano Zampini   ierr = PetscTryMethod(pc,"PCBDDCSetDofsSplittingLocal_C",(PC,PetscInt,IS[]),(pc,n_is,ISForDofs));CHKERRQ(ierr);
85563602bcaSStefano Zampini   PetscFunctionReturn(0);
85663602bcaSStefano Zampini }
85763602bcaSStefano Zampini /* -------------------------------------------------------------------------- */
85863602bcaSStefano Zampini 
85963602bcaSStefano Zampini #undef __FUNCT__
8609c0446d6SStefano Zampini #define __FUNCT__ "PCBDDCSetDofsSplitting_BDDC"
8619c0446d6SStefano Zampini static PetscErrorCode PCBDDCSetDofsSplitting_BDDC(PC pc,PetscInt n_is, IS ISForDofs[])
8629c0446d6SStefano Zampini {
8639c0446d6SStefano Zampini   PC_BDDC  *pcbddc = (PC_BDDC*)pc->data;
8649c0446d6SStefano Zampini   PetscInt i;
8659c0446d6SStefano Zampini   PetscErrorCode ierr;
8669c0446d6SStefano Zampini 
8679c0446d6SStefano Zampini   PetscFunctionBegin;
868da1bb401SStefano Zampini   /* Destroy ISes if they were already set */
8699c0446d6SStefano Zampini   for (i=0;i<pcbddc->n_ISForDofs;i++) {
8709c0446d6SStefano Zampini     ierr = ISDestroy(&pcbddc->ISForDofs[i]);CHKERRQ(ierr);
8719c0446d6SStefano Zampini   }
872d11ae9bbSstefano_zampini   ierr = PetscFree(pcbddc->ISForDofs);CHKERRQ(ierr);
87363602bcaSStefano Zampini   /* last user setting takes precendence -> destroy any other customization */
87463602bcaSStefano Zampini   for (i=0;i<pcbddc->n_ISForDofsLocal;i++) {
87563602bcaSStefano Zampini     ierr = ISDestroy(&pcbddc->ISForDofsLocal[i]);CHKERRQ(ierr);
87663602bcaSStefano Zampini   }
87763602bcaSStefano Zampini   ierr = PetscFree(pcbddc->ISForDofsLocal);CHKERRQ(ierr);
87863602bcaSStefano Zampini   pcbddc->n_ISForDofsLocal = 0;
879da1bb401SStefano Zampini   /* allocate space then set */
880d02579f5SStefano Zampini   if (n_is) {
881785e854fSJed Brown     ierr = PetscMalloc1(n_is,&pcbddc->ISForDofs);CHKERRQ(ierr);
882d02579f5SStefano Zampini   }
8839c0446d6SStefano Zampini   for (i=0;i<n_is;i++) {
884da1bb401SStefano Zampini     ierr = PetscObjectReference((PetscObject)ISForDofs[i]);CHKERRQ(ierr);
885da1bb401SStefano Zampini     pcbddc->ISForDofs[i]=ISForDofs[i];
8869c0446d6SStefano Zampini   }
8879c0446d6SStefano Zampini   pcbddc->n_ISForDofs=n_is;
88863602bcaSStefano Zampini   if (n_is) pcbddc->user_provided_isfordofs = PETSC_TRUE;
8891a8a16d1SStefano Zampini   pcbddc->recompute_topography = PETSC_TRUE;
8909c0446d6SStefano Zampini   PetscFunctionReturn(0);
8919c0446d6SStefano Zampini }
8921e6b0712SBarry Smith 
8939c0446d6SStefano Zampini #undef __FUNCT__
8949c0446d6SStefano Zampini #define __FUNCT__ "PCBDDCSetDofsSplitting"
8959c0446d6SStefano Zampini /*@
89663602bcaSStefano Zampini  PCBDDCSetDofsSplitting - Set index sets defining fields of the global matrix
8979c0446d6SStefano Zampini 
89863602bcaSStefano Zampini    Collective
8999c0446d6SStefano Zampini 
9009c0446d6SStefano Zampini    Input Parameters:
9019c0446d6SStefano Zampini +  pc - the preconditioning context
9020f202f7eSStefano Zampini .  n_is - number of index sets defining the fields
9030f202f7eSStefano Zampini -  ISForDofs - array of IS describing the fields in global ordering
9049c0446d6SStefano Zampini 
9059c0446d6SStefano Zampini    Level: intermediate
9069c0446d6SStefano Zampini 
9070f202f7eSStefano Zampini    Notes:
9080f202f7eSStefano Zampini      Any process can list any global node. Not all nodes need to be listed: unlisted nodes will belong to the complement field.
9099c0446d6SStefano Zampini 
9109c0446d6SStefano Zampini .seealso: PCBDDC
9119c0446d6SStefano Zampini @*/
9129c0446d6SStefano Zampini PetscErrorCode PCBDDCSetDofsSplitting(PC pc,PetscInt n_is, IS ISForDofs[])
9139c0446d6SStefano Zampini {
9142b510759SStefano Zampini   PetscInt       i;
9159c0446d6SStefano Zampini   PetscErrorCode ierr;
9169c0446d6SStefano Zampini 
9179c0446d6SStefano Zampini   PetscFunctionBegin;
9189c0446d6SStefano Zampini   PetscValidHeaderSpecific(pc,PC_CLASSID,1);
91963602bcaSStefano Zampini   PetscValidLogicalCollectiveInt(pc,n_is,2);
9202b510759SStefano Zampini   for (i=0;i<n_is;i++) {
92163602bcaSStefano Zampini     PetscCheckSameComm(pc,1,ISForDofs[i],3);
92263602bcaSStefano Zampini     PetscValidHeaderSpecific(ISForDofs[i],IS_CLASSID,3);
9232b510759SStefano Zampini   }
9249c0446d6SStefano Zampini   ierr = PetscTryMethod(pc,"PCBDDCSetDofsSplitting_C",(PC,PetscInt,IS[]),(pc,n_is,ISForDofs));CHKERRQ(ierr);
9259c0446d6SStefano Zampini   PetscFunctionReturn(0);
9269c0446d6SStefano Zampini }
927906d46d4SStefano Zampini 
928da1bb401SStefano Zampini /* -------------------------------------------------------------------------- */
929534831adSStefano Zampini #undef __FUNCT__
930534831adSStefano Zampini #define __FUNCT__ "PCPreSolve_BDDC"
931534831adSStefano Zampini /* -------------------------------------------------------------------------- */
932534831adSStefano Zampini /*
933534831adSStefano Zampini    PCPreSolve_BDDC - Changes the right hand side and (if necessary) the initial
934534831adSStefano Zampini                      guess if a transformation of basis approach has been selected.
9359c0446d6SStefano Zampini 
936534831adSStefano Zampini    Input Parameter:
937534831adSStefano Zampini +  pc - the preconditioner contex
938534831adSStefano Zampini 
939534831adSStefano Zampini    Application Interface Routine: PCPreSolve()
940534831adSStefano Zampini 
941534831adSStefano Zampini    Notes:
942534831adSStefano Zampini      The interface routine PCPreSolve() is not usually called directly by
943534831adSStefano Zampini    the user, but instead is called by KSPSolve().
944534831adSStefano Zampini */
945534831adSStefano Zampini static PetscErrorCode PCPreSolve_BDDC(PC pc, KSP ksp, Vec rhs, Vec x)
946534831adSStefano Zampini {
947534831adSStefano Zampini   PetscErrorCode ierr;
948534831adSStefano Zampini   PC_BDDC        *pcbddc = (PC_BDDC*)pc->data;
949534831adSStefano Zampini   PC_IS          *pcis = (PC_IS*)(pc->data);
9503972b0daSStefano Zampini   Vec            used_vec;
9518d00608fSStefano Zampini   PetscBool      copy_rhs = PETSC_TRUE;
952534831adSStefano Zampini 
953534831adSStefano Zampini   PetscFunctionBegin;
95485c4d303SStefano Zampini   /* if we are working with cg, one dirichlet solve can be avoided during Krylov iterations */
95585c4d303SStefano Zampini   if (ksp) {
95685c4d303SStefano Zampini     PetscBool iscg;
95785c4d303SStefano Zampini     ierr = PetscObjectTypeCompare((PetscObject)ksp,KSPCG,&iscg);CHKERRQ(ierr);
95885c4d303SStefano Zampini     if (!iscg) {
95985c4d303SStefano Zampini       ierr = PCBDDCSetUseExactDirichlet(pc,PETSC_FALSE);CHKERRQ(ierr);
96085c4d303SStefano Zampini     }
96185c4d303SStefano Zampini   }
962c7017625SStefano 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 
9913972b0daSStefano Zampini   /* Take into account zeroed rows -> change rhs and store solution removed */
992a07ea27aSStefano Zampini   if (rhs) {
9933975b054SStefano Zampini     IS dirIS = NULL;
9943975b054SStefano Zampini 
995a07ea27aSStefano Zampini     /* DirichletBoundariesLocal may not be consistent among neighbours; gets a dirichlet dofs IS from graph (may be cached) */
9963975b054SStefano Zampini     ierr = PCBDDCGraphGetDirichletDofs(pcbddc->mat_graph,&dirIS);CHKERRQ(ierr);
9973975b054SStefano Zampini     if (dirIS) {
998906d46d4SStefano Zampini       Mat_IS            *matis = (Mat_IS*)pc->pmat->data;
999785d1243SStefano Zampini       PetscInt          dirsize,i,*is_indices;
10002b095fd8SStefano Zampini       PetscScalar       *array_x;
10012b095fd8SStefano Zampini       const PetscScalar *array_diagonal;
1002785d1243SStefano Zampini 
10033972b0daSStefano Zampini       ierr = MatGetDiagonal(pc->pmat,pcis->vec1_global);CHKERRQ(ierr);
10043972b0daSStefano Zampini       ierr = VecPointwiseDivide(pcis->vec1_global,rhs,pcis->vec1_global);CHKERRQ(ierr);
1005e176bc59SStefano Zampini       ierr = VecScatterBegin(matis->rctx,pcis->vec1_global,pcis->vec2_N,INSERT_VALUES,SCATTER_FORWARD);CHKERRQ(ierr);
1006e176bc59SStefano Zampini       ierr = VecScatterEnd(matis->rctx,pcis->vec1_global,pcis->vec2_N,INSERT_VALUES,SCATTER_FORWARD);CHKERRQ(ierr);
1007e176bc59SStefano Zampini       ierr = VecScatterBegin(matis->rctx,used_vec,pcis->vec1_N,INSERT_VALUES,SCATTER_FORWARD);CHKERRQ(ierr);
1008e176bc59SStefano Zampini       ierr = VecScatterEnd(matis->rctx,used_vec,pcis->vec1_N,INSERT_VALUES,SCATTER_FORWARD);CHKERRQ(ierr);
100982ba6b80SStefano Zampini       ierr = ISGetLocalSize(dirIS,&dirsize);CHKERRQ(ierr);
10103972b0daSStefano Zampini       ierr = VecGetArray(pcis->vec1_N,&array_x);CHKERRQ(ierr);
10112b095fd8SStefano Zampini       ierr = VecGetArrayRead(pcis->vec2_N,&array_diagonal);CHKERRQ(ierr);
10123972b0daSStefano Zampini       ierr = ISGetIndices(dirIS,(const PetscInt**)&is_indices);CHKERRQ(ierr);
10132fa5cd67SKarl Rupp       for (i=0; i<dirsize; i++) array_x[is_indices[i]] = array_diagonal[is_indices[i]];
10143972b0daSStefano Zampini       ierr = ISRestoreIndices(dirIS,(const PetscInt**)&is_indices);CHKERRQ(ierr);
10152b095fd8SStefano Zampini       ierr = VecRestoreArrayRead(pcis->vec2_N,&array_diagonal);CHKERRQ(ierr);
10163972b0daSStefano Zampini       ierr = VecRestoreArray(pcis->vec1_N,&array_x);CHKERRQ(ierr);
1017e176bc59SStefano Zampini       ierr = VecScatterBegin(matis->rctx,pcis->vec1_N,used_vec,INSERT_VALUES,SCATTER_REVERSE);CHKERRQ(ierr);
1018e176bc59SStefano Zampini       ierr = VecScatterEnd(matis->rctx,pcis->vec1_N,used_vec,INSERT_VALUES,SCATTER_REVERSE);CHKERRQ(ierr);
10198d00608fSStefano Zampini       pcbddc->rhs_change = PETSC_TRUE;
10201b968477SStefano Zampini       ierr = ISDestroy(&dirIS);CHKERRQ(ierr);
10218efcfb23SStefano Zampini     }
1022a07ea27aSStefano Zampini   }
1023b76ba322SStefano Zampini 
10248efcfb23SStefano Zampini   /* remove the computed solution or the initial guess from the rhs */
10258d00608fSStefano Zampini   if (pcbddc->rhs_change || (ksp && pcbddc->ksp_guess_nonzero) ) {
10268d00608fSStefano Zampini     /* store the original rhs */
10278d00608fSStefano Zampini     if (copy_rhs) {
10288d00608fSStefano Zampini       ierr = VecCopy(rhs,pcbddc->original_rhs);CHKERRQ(ierr);
10298d00608fSStefano Zampini       copy_rhs = PETSC_FALSE;
10308d00608fSStefano Zampini     }
10318d00608fSStefano Zampini     pcbddc->rhs_change = PETSC_TRUE;
10323972b0daSStefano Zampini     ierr = VecScale(used_vec,-1.0);CHKERRQ(ierr);
10333972b0daSStefano Zampini     ierr = MatMultAdd(pc->pmat,used_vec,rhs,rhs);CHKERRQ(ierr);
10343972b0daSStefano Zampini     ierr = VecScale(used_vec,-1.0);CHKERRQ(ierr);
10358efcfb23SStefano Zampini     ierr = VecCopy(used_vec,pcbddc->temp_solution);CHKERRQ(ierr);
10367acc28cbSStefano Zampini     if (ksp) {
10377acc28cbSStefano Zampini       ierr = KSPSetInitialGuessNonzero(ksp,PETSC_FALSE);CHKERRQ(ierr);
10387acc28cbSStefano Zampini     }
10393308cffdSStefano Zampini   }
10408efcfb23SStefano Zampini   ierr = VecDestroy(&used_vec);CHKERRQ(ierr);
1041b76ba322SStefano Zampini 
1042b76ba322SStefano Zampini   /* store partially computed solution and set initial guess */
1043b097fa66SStefano Zampini   if (x && pcbddc->use_exact_dirichlet_trick) {
10448efcfb23SStefano Zampini     ierr = VecSet(x,0.0);CHKERRQ(ierr);
1045b76ba322SStefano Zampini     ierr = VecScatterBegin(pcis->global_to_D,rhs,pcis->vec1_D,INSERT_VALUES,SCATTER_FORWARD);CHKERRQ(ierr);
1046b76ba322SStefano Zampini     ierr = VecScatterEnd(pcis->global_to_D,rhs,pcis->vec1_D,INSERT_VALUES,SCATTER_FORWARD);CHKERRQ(ierr);
1047b76ba322SStefano Zampini     ierr = KSPSolve(pcbddc->ksp_D,pcis->vec1_D,pcis->vec2_D);CHKERRQ(ierr);
10488efcfb23SStefano Zampini     ierr = VecScatterBegin(pcis->global_to_D,pcis->vec2_D,x,INSERT_VALUES,SCATTER_REVERSE);CHKERRQ(ierr);
10498efcfb23SStefano Zampini     ierr = VecScatterEnd(pcis->global_to_D,pcis->vec2_D,x,INSERT_VALUES,SCATTER_REVERSE);CHKERRQ(ierr);
1050a0cb1b98SStefano Zampini     if (ksp) {
1051b76ba322SStefano Zampini       ierr = KSPSetInitialGuessNonzero(ksp,PETSC_TRUE);CHKERRQ(ierr);
1052b76ba322SStefano Zampini     }
1053b76ba322SStefano Zampini   }
1054b76ba322SStefano Zampini 
1055b9b85e73SStefano Zampini   if (pcbddc->ChangeOfBasisMatrix) {
1056906d46d4SStefano Zampini     PCBDDCChange_ctx change_ctx;
1057906d46d4SStefano Zampini 
1058906d46d4SStefano Zampini     /* get change ctx */
1059906d46d4SStefano Zampini     ierr = MatShellGetContext(pcbddc->new_global_mat,&change_ctx);CHKERRQ(ierr);
1060906d46d4SStefano Zampini 
1061906d46d4SStefano Zampini     /* set current iteration matrix inside change context (change of basis has been already set into the ctx during PCSetUp) */
1062906d46d4SStefano Zampini     ierr = MatDestroy(&change_ctx->original_mat);CHKERRQ(ierr);
1063906d46d4SStefano Zampini     ierr = PetscObjectReference((PetscObject)pc->mat);CHKERRQ(ierr);
1064906d46d4SStefano Zampini     change_ctx->original_mat = pc->mat;
1065906d46d4SStefano Zampini 
1066906d46d4SStefano Zampini     /* change iteration matrix */
1067906d46d4SStefano Zampini     ierr = MatDestroy(&pc->mat);CHKERRQ(ierr);
1068906d46d4SStefano Zampini     ierr = PetscObjectReference((PetscObject)pcbddc->new_global_mat);CHKERRQ(ierr);
1069906d46d4SStefano Zampini     pc->mat = pcbddc->new_global_mat;
1070906d46d4SStefano Zampini 
10718d00608fSStefano Zampini     /* store the original rhs */
10728d00608fSStefano Zampini     if (copy_rhs) {
10738d00608fSStefano Zampini       ierr = VecCopy(rhs,pcbddc->original_rhs);CHKERRQ(ierr);
10748d00608fSStefano Zampini       copy_rhs = PETSC_FALSE;
10758d00608fSStefano Zampini     }
10768d00608fSStefano Zampini 
1077906d46d4SStefano Zampini     /* change rhs */
1078906d46d4SStefano Zampini     ierr = MatMultTranspose(change_ctx->global_change,rhs,pcis->vec1_global);CHKERRQ(ierr);
1079906d46d4SStefano Zampini     ierr = VecCopy(pcis->vec1_global,rhs);CHKERRQ(ierr);
10808d00608fSStefano Zampini     pcbddc->rhs_change = PETSC_TRUE;
108192e3dcfbSStefano Zampini   }
1082534831adSStefano Zampini   PetscFunctionReturn(0);
1083534831adSStefano Zampini }
1084906d46d4SStefano Zampini 
1085534831adSStefano Zampini /* -------------------------------------------------------------------------- */
1086534831adSStefano Zampini #undef __FUNCT__
1087534831adSStefano Zampini #define __FUNCT__ "PCPostSolve_BDDC"
1088534831adSStefano Zampini /* -------------------------------------------------------------------------- */
1089534831adSStefano Zampini /*
1090534831adSStefano Zampini    PCPostSolve_BDDC - Changes the computed solution if a transformation of basis
1091534831adSStefano Zampini                      approach has been selected. Also, restores rhs to its original state.
1092534831adSStefano Zampini 
1093534831adSStefano Zampini    Input Parameter:
1094534831adSStefano Zampini +  pc - the preconditioner contex
1095534831adSStefano Zampini 
1096534831adSStefano Zampini    Application Interface Routine: PCPostSolve()
1097534831adSStefano Zampini 
1098534831adSStefano Zampini    Notes:
1099534831adSStefano Zampini      The interface routine PCPostSolve() is not usually called directly by
1100534831adSStefano Zampini      the user, but instead is called by KSPSolve().
1101534831adSStefano Zampini */
1102534831adSStefano Zampini static PetscErrorCode PCPostSolve_BDDC(PC pc, KSP ksp, Vec rhs, Vec x)
1103534831adSStefano Zampini {
1104534831adSStefano Zampini   PetscErrorCode ierr;
1105534831adSStefano Zampini   PC_BDDC        *pcbddc = (PC_BDDC*)pc->data;
1106534831adSStefano Zampini 
1107534831adSStefano Zampini   PetscFunctionBegin;
1108b9b85e73SStefano Zampini   if (pcbddc->ChangeOfBasisMatrix) {
1109906d46d4SStefano Zampini     PCBDDCChange_ctx change_ctx;
1110906d46d4SStefano Zampini 
1111906d46d4SStefano Zampini     /* get change ctx */
1112906d46d4SStefano Zampini     ierr = MatShellGetContext(pcbddc->new_global_mat,&change_ctx);CHKERRQ(ierr);
1113906d46d4SStefano Zampini 
1114906d46d4SStefano Zampini     /* restore iteration matrix */
1115906d46d4SStefano Zampini     ierr = MatDestroy(&pc->mat);CHKERRQ(ierr);
1116906d46d4SStefano Zampini     ierr = PetscObjectReference((PetscObject)change_ctx->original_mat);CHKERRQ(ierr);
1117906d46d4SStefano Zampini     pc->mat = change_ctx->original_mat;
1118906d46d4SStefano Zampini 
1119906d46d4SStefano Zampini     /* get solution in original basis */
1120906d46d4SStefano Zampini     if (x) {
1121906d46d4SStefano Zampini       PC_IS *pcis = (PC_IS*)(pc->data);
1122906d46d4SStefano Zampini       ierr = MatMult(change_ctx->global_change,x,pcis->vec1_global);CHKERRQ(ierr);
1123906d46d4SStefano Zampini       ierr = VecCopy(pcis->vec1_global,x);CHKERRQ(ierr);
11243425bc38SStefano Zampini     }
1125534831adSStefano Zampini   }
1126906d46d4SStefano Zampini 
11273972b0daSStefano Zampini   /* add solution removed in presolve */
11286bcfc461SStefano Zampini   if (x && pcbddc->rhs_change) {
11293425bc38SStefano Zampini     ierr = VecAXPY(x,1.0,pcbddc->temp_solution);CHKERRQ(ierr);
11303425bc38SStefano Zampini   }
1131906d46d4SStefano Zampini 
1132fb223d50SStefano Zampini   /* restore rhs to its original state */
11338d00608fSStefano Zampini   if (rhs && pcbddc->rhs_change) {
1134fb223d50SStefano Zampini     ierr = VecCopy(pcbddc->original_rhs,rhs);CHKERRQ(ierr);
1135fb223d50SStefano Zampini   }
11368d00608fSStefano Zampini   pcbddc->rhs_change = PETSC_FALSE;
11378efcfb23SStefano Zampini 
11388efcfb23SStefano Zampini   /* restore ksp guess state */
11398efcfb23SStefano Zampini   if (ksp) {
11408efcfb23SStefano Zampini     ierr = KSPSetInitialGuessNonzero(ksp,pcbddc->ksp_guess_nonzero);CHKERRQ(ierr);
11418efcfb23SStefano Zampini   }
1142534831adSStefano Zampini   PetscFunctionReturn(0);
1143534831adSStefano Zampini }
1144534831adSStefano Zampini /* -------------------------------------------------------------------------- */
114553cdbc3dSStefano Zampini #undef __FUNCT__
114653cdbc3dSStefano Zampini #define __FUNCT__ "PCSetUp_BDDC"
11470c7d97c5SJed Brown /* -------------------------------------------------------------------------- */
11480c7d97c5SJed Brown /*
11490c7d97c5SJed Brown    PCSetUp_BDDC - Prepares for the use of the BDDC preconditioner
11500c7d97c5SJed Brown                   by setting data structures and options.
11510c7d97c5SJed Brown 
11520c7d97c5SJed Brown    Input Parameter:
115353cdbc3dSStefano Zampini +  pc - the preconditioner context
11540c7d97c5SJed Brown 
11550c7d97c5SJed Brown    Application Interface Routine: PCSetUp()
11560c7d97c5SJed Brown 
11570c7d97c5SJed Brown    Notes:
11580c7d97c5SJed Brown      The interface routine PCSetUp() is not usually called directly by
11590c7d97c5SJed Brown      the user, but instead is called by PCApply() if necessary.
11600c7d97c5SJed Brown */
116153cdbc3dSStefano Zampini PetscErrorCode PCSetUp_BDDC(PC pc)
11620c7d97c5SJed Brown {
11630c7d97c5SJed Brown   PetscErrorCode ierr;
11640c7d97c5SJed Brown   PC_BDDC*       pcbddc = (PC_BDDC*)pc->data;
11655e8657edSStefano Zampini   Mat_IS*        matis;
116608122e43SStefano Zampini   MatNullSpace   nearnullspace;
116791e8d312SStefano Zampini   PetscInt       nrows,ncols;
116808122e43SStefano Zampini   PetscBool      computetopography,computesolvers,computesubschurs;
11698de1fae6SStefano Zampini   PetscBool      computeconstraintsmatrix;
11705e8657edSStefano Zampini   PetscBool      new_nearnullspace_provided,ismatis;
11710c7d97c5SJed Brown 
11720c7d97c5SJed Brown   PetscFunctionBegin;
11735e8657edSStefano Zampini   ierr = PetscObjectTypeCompare((PetscObject)pc->pmat,MATIS,&ismatis);CHKERRQ(ierr);
11746c4ed002SBarry Smith   if (!ismatis) SETERRQ(PetscObjectComm((PetscObject)pc),PETSC_ERR_ARG_WRONG,"PCBDDC preconditioner requires matrix of type MATIS");
117591e8d312SStefano Zampini   ierr = MatGetSize(pc->pmat,&nrows,&ncols);CHKERRQ(ierr);
11766c4ed002SBarry Smith   if (nrows != ncols) SETERRQ(PetscObjectComm((PetscObject)pc),PETSC_ERR_SUP,"PCBDDC preconditioner requires a square preconditioning matrix");
11775e8657edSStefano Zampini   matis = (Mat_IS*)pc->pmat->data;
1178f4ddd8eeSStefano Zampini   /* the following lines of code should be replaced by a better logic between PCIS, PCNN, PCBDDC and other future nonoverlapping preconditioners */
11793b03a366Sstefano_zampini   /* For BDDC we need to define a local "Neumann" problem different to that defined in PCISSetup
1180fcd409f5SStefano Zampini      Also, BDDC directly build the Dirichlet problem */
1181f4ddd8eeSStefano Zampini   /* split work */
1182674ae819SStefano Zampini   if (pc->setupcalled) {
1183d1e9a80fSBarry Smith     if (pc->flag == SAME_NONZERO_PATTERN) {
1184674ae819SStefano Zampini       computetopography = PETSC_FALSE;
1185674ae819SStefano Zampini       computesolvers = PETSC_TRUE;
1186674ae819SStefano Zampini     } else { /* DIFFERENT_NONZERO_PATTERN */
1187674ae819SStefano Zampini       computetopography = PETSC_TRUE;
1188674ae819SStefano Zampini       computesolvers = PETSC_TRUE;
1189674ae819SStefano Zampini     }
1190674ae819SStefano Zampini   } else {
1191674ae819SStefano Zampini     computetopography = PETSC_TRUE;
1192674ae819SStefano Zampini     computesolvers = PETSC_TRUE;
1193674ae819SStefano Zampini   }
1194fb180af4SStefano Zampini   if (pcbddc->recompute_topography) {
1195fb180af4SStefano Zampini     computetopography = PETSC_TRUE;
1196fb180af4SStefano Zampini   }
11978de1fae6SStefano Zampini   computeconstraintsmatrix = PETSC_FALSE;
11986c4ed002SBarry Smith   if (pcbddc->adaptive_threshold > 0.0 && !pcbddc->use_deluxe_scaling) SETERRQ(PetscObjectComm((PetscObject)pc),PETSC_ERR_SUP,"Cannot compute adaptive constraints without deluxe scaling. Rerun with -pc_bddc_use_deluxe_scaling");
11995a95e1ceSStefano Zampini   pcbddc->adaptive_selection = (PetscBool)(pcbddc->adaptive_threshold > 0.0 && pcbddc->use_deluxe_scaling);
1200862806e4SStefano Zampini   if (pcbddc->adaptive_selection) pcbddc->use_faces = PETSC_TRUE;
1201862806e4SStefano Zampini 
12025a95e1ceSStefano Zampini   computesubschurs = (PetscBool)(pcbddc->adaptive_selection || pcbddc->use_deluxe_scaling);
12036c4ed002SBarry Smith   if (pcbddc->faster_deluxe && pcbddc->adaptive_selection && pcbddc->use_change_of_basis) 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");
12046c4ed002SBarry Smith 
1205f4ddd8eeSStefano Zampini   /* Get stdout for dbg */
120670cf5478SStefano Zampini   if (pcbddc->dbg_flag) {
120770cf5478SStefano Zampini     if (!pcbddc->dbg_viewer) {
120858a03d70SStefano Zampini       pcbddc->dbg_viewer = PETSC_VIEWER_STDOUT_(PetscObjectComm((PetscObject)pc));
12091575c14dSBarry Smith       ierr = PetscViewerASCIIPushSynchronized(pcbddc->dbg_viewer);CHKERRQ(ierr);
1210f4ddd8eeSStefano Zampini     }
121158a03d70SStefano Zampini     ierr = PetscViewerASCIIAddTab(pcbddc->dbg_viewer,2*pcbddc->current_level);CHKERRQ(ierr);
1212f4ddd8eeSStefano Zampini   }
1213f4ddd8eeSStefano Zampini 
12145e8657edSStefano Zampini   if (pcbddc->user_ChangeOfBasisMatrix) {
12155e8657edSStefano Zampini     /* use_change_of_basis flag is used to automatically compute a change of basis from constraints */
12165e8657edSStefano Zampini     pcbddc->use_change_of_basis = PETSC_FALSE;
12175e8657edSStefano Zampini     ierr = PCBDDCComputeLocalMatrix(pc,pcbddc->user_ChangeOfBasisMatrix);CHKERRQ(ierr);
12185e8657edSStefano Zampini   } else {
1219b96c3477SStefano Zampini     ierr = MatDestroy(&pcbddc->local_mat);CHKERRQ(ierr);
12205e8657edSStefano Zampini     ierr = PetscObjectReference((PetscObject)matis->A);CHKERRQ(ierr);
12215e8657edSStefano Zampini     pcbddc->local_mat = matis->A;
1222674ae819SStefano Zampini   }
1223f4ddd8eeSStefano Zampini 
1224e496cd5dSStefano Zampini   /* workaround for reals */
1225e496cd5dSStefano Zampini #if !defined(PETSC_USE_COMPLEX)
12263301b35fSStefano Zampini   if (matis->A->symmetric_set) {
12273301b35fSStefano Zampini     ierr = MatSetOption(pcbddc->local_mat,MAT_HERMITIAN,matis->A->symmetric);CHKERRQ(ierr);
12283301b35fSStefano Zampini   }
1229e496cd5dSStefano Zampini #endif
1230e496cd5dSStefano Zampini 
12315e8657edSStefano Zampini   /* Set up all the "iterative substructuring" common block without computing solvers */
12325e8657edSStefano Zampini   {
12335e8657edSStefano Zampini     Mat temp_mat;
12345e8657edSStefano Zampini 
12355e8657edSStefano Zampini     temp_mat = matis->A;
12365e8657edSStefano Zampini     matis->A = pcbddc->local_mat;
12375e8657edSStefano Zampini     ierr = PCISSetUp(pc,PETSC_FALSE);CHKERRQ(ierr);
12385e8657edSStefano Zampini     pcbddc->local_mat = matis->A;
12395e8657edSStefano Zampini     matis->A = temp_mat;
12405e8657edSStefano Zampini   }
1241684f6988SStefano Zampini 
1242b96c3477SStefano Zampini   /* Analyze interface and setup sub_schurs data */
1243674ae819SStefano Zampini   if (computetopography) {
1244674ae819SStefano Zampini     ierr = PCBDDCAnalyzeInterface(pc);CHKERRQ(ierr);
12458de1fae6SStefano Zampini     computeconstraintsmatrix = PETSC_TRUE;
1246674ae819SStefano Zampini   }
1247fb8d54d4SStefano Zampini 
1248b96c3477SStefano Zampini   /* Setup local dirichlet solver ksp_D and sub_schurs solvers */
1249684f6988SStefano Zampini   if (computesolvers) {
1250d5574798SStefano Zampini     PCBDDCSubSchurs sub_schurs=pcbddc->sub_schurs;
1251d5574798SStefano Zampini 
1252d5574798SStefano Zampini     if (computesubschurs && computetopography) {
125308122e43SStefano Zampini       ierr = PCBDDCInitSubSchurs(pc);CHKERRQ(ierr);
1254b1b3d7a2SStefano Zampini     }
1255d5574798SStefano Zampini     if (sub_schurs->use_mumps) {
12562070dbb6SStefano Zampini       if (computesubschurs) {
125708122e43SStefano Zampini         ierr = PCBDDCSetUpSubSchurs(pc);CHKERRQ(ierr);
12582070dbb6SStefano Zampini       }
1259d5574798SStefano Zampini       ierr = PCBDDCSetUpLocalSolvers(pc,PETSC_TRUE,PETSC_FALSE);CHKERRQ(ierr);
1260d5574798SStefano Zampini     } else {
1261d5574798SStefano Zampini       ierr = PCBDDCSetUpLocalSolvers(pc,PETSC_TRUE,PETSC_FALSE);CHKERRQ(ierr);
12622070dbb6SStefano Zampini       if (computesubschurs) {
1263d5574798SStefano Zampini         ierr = PCBDDCSetUpSubSchurs(pc);CHKERRQ(ierr);
1264d5574798SStefano Zampini       }
12652070dbb6SStefano Zampini     }
126608122e43SStefano Zampini     if (pcbddc->adaptive_selection) {
126708122e43SStefano Zampini       ierr = PCBDDCAdaptiveSelection(pc);CHKERRQ(ierr);
12688de1fae6SStefano Zampini       computeconstraintsmatrix = PETSC_TRUE;
1269b7eb3628SStefano Zampini     }
1270b1b3d7a2SStefano Zampini   }
1271684f6988SStefano Zampini 
1272f4ddd8eeSStefano Zampini   /* infer if NullSpace object attached to Mat via MatSetNearNullSpace has changed */
1273fb8d54d4SStefano Zampini   new_nearnullspace_provided = PETSC_FALSE;
1274f4ddd8eeSStefano Zampini   ierr = MatGetNearNullSpace(pc->pmat,&nearnullspace);CHKERRQ(ierr);
1275f4ddd8eeSStefano Zampini   if (pcbddc->onearnullspace) { /* already used nearnullspace */
1276f4ddd8eeSStefano Zampini     if (!nearnullspace) { /* near null space attached to mat has been destroyed */
1277f4ddd8eeSStefano Zampini       new_nearnullspace_provided = PETSC_TRUE;
1278f4ddd8eeSStefano Zampini     } else {
1279f4ddd8eeSStefano Zampini       /* determine if the two nullspaces are different (should be lightweight) */
1280f4ddd8eeSStefano Zampini       if (nearnullspace != pcbddc->onearnullspace) {
1281f4ddd8eeSStefano Zampini         new_nearnullspace_provided = PETSC_TRUE;
1282165b64e2SStefano Zampini       } else { /* maybe the user has changed the content of the nearnullspace so check vectors ObjectStateId */
1283f4ddd8eeSStefano Zampini         PetscInt         i;
1284165b64e2SStefano Zampini         const Vec        *nearnullvecs;
1285165b64e2SStefano Zampini         PetscObjectState state;
1286165b64e2SStefano Zampini         PetscInt         nnsp_size;
1287165b64e2SStefano Zampini         ierr = MatNullSpaceGetVecs(nearnullspace,NULL,&nnsp_size,&nearnullvecs);CHKERRQ(ierr);
1288f4ddd8eeSStefano Zampini         for (i=0;i<nnsp_size;i++) {
1289f4ddd8eeSStefano Zampini           ierr = PetscObjectStateGet((PetscObject)nearnullvecs[i],&state);CHKERRQ(ierr);
1290165b64e2SStefano Zampini           if (pcbddc->onearnullvecs_state[i] != state) {
1291f4ddd8eeSStefano Zampini             new_nearnullspace_provided = PETSC_TRUE;
1292f4ddd8eeSStefano Zampini             break;
1293f4ddd8eeSStefano Zampini           }
1294f4ddd8eeSStefano Zampini         }
1295f4ddd8eeSStefano Zampini       }
1296f4ddd8eeSStefano Zampini     }
1297f4ddd8eeSStefano Zampini   } else {
1298f4ddd8eeSStefano Zampini     if (!nearnullspace) { /* both nearnullspaces are null */
1299f4ddd8eeSStefano Zampini       new_nearnullspace_provided = PETSC_FALSE;
1300f4ddd8eeSStefano Zampini     } else { /* nearnullspace attached later */
1301f4ddd8eeSStefano Zampini       new_nearnullspace_provided = PETSC_TRUE;
1302f4ddd8eeSStefano Zampini     }
1303f4ddd8eeSStefano Zampini   }
1304f4ddd8eeSStefano Zampini 
1305f4ddd8eeSStefano Zampini   /* Setup constraints and related work vectors */
1306727cdba6SStefano Zampini   /* reset primal space flags */
1307f4ddd8eeSStefano Zampini   pcbddc->new_primal_space = PETSC_FALSE;
1308727cdba6SStefano Zampini   pcbddc->new_primal_space_local = PETSC_FALSE;
13098de1fae6SStefano Zampini   if (computeconstraintsmatrix || new_nearnullspace_provided) {
1310727cdba6SStefano Zampini     /* It also sets the primal space flags */
1311674ae819SStefano Zampini     ierr = PCBDDCConstraintsSetUp(pc);CHKERRQ(ierr);
1312e7b262bdSStefano Zampini     /* Allocate needed local vectors (which depends on quantities defined during ConstraintsSetUp) */
1313f4ddd8eeSStefano Zampini     ierr = PCBDDCSetUpLocalWorkVectors(pc);CHKERRQ(ierr);
13143975b054SStefano Zampini   }
13155e8657edSStefano Zampini 
13163975b054SStefano Zampini   if (computesolvers || pcbddc->new_primal_space) {
13175e8657edSStefano Zampini     if (pcbddc->use_change_of_basis) {
13185e8657edSStefano Zampini       PC_IS *pcis = (PC_IS*)(pc->data);
13195e8657edSStefano Zampini 
13205e8657edSStefano Zampini       ierr = PCBDDCComputeLocalMatrix(pc,pcbddc->ChangeOfBasisMatrix);CHKERRQ(ierr);
13215e8657edSStefano Zampini       /* get submatrices */
13225e8657edSStefano Zampini       ierr = MatDestroy(&pcis->A_IB);CHKERRQ(ierr);
13235e8657edSStefano Zampini       ierr = MatDestroy(&pcis->A_BI);CHKERRQ(ierr);
13245e8657edSStefano Zampini       ierr = MatDestroy(&pcis->A_BB);CHKERRQ(ierr);
13255e8657edSStefano Zampini       ierr = MatGetSubMatrix(pcbddc->local_mat,pcis->is_B_local,pcis->is_B_local,MAT_INITIAL_MATRIX,&pcis->A_BB);CHKERRQ(ierr);
13265e8657edSStefano Zampini       ierr = MatGetSubMatrix(pcbddc->local_mat,pcis->is_I_local,pcis->is_B_local,MAT_INITIAL_MATRIX,&pcis->A_IB);CHKERRQ(ierr);
13275e8657edSStefano Zampini       ierr = MatGetSubMatrix(pcbddc->local_mat,pcis->is_B_local,pcis->is_I_local,MAT_INITIAL_MATRIX,&pcis->A_BI);CHKERRQ(ierr);
13283975b054SStefano Zampini       /* set flag in pcis to not reuse submatrices during PCISCreate */
13293975b054SStefano Zampini       pcis->reusesubmatrices = PETSC_FALSE;
13305e8657edSStefano Zampini     } else if (!pcbddc->user_ChangeOfBasisMatrix) {
1331b96c3477SStefano Zampini       ierr = MatDestroy(&pcbddc->local_mat);CHKERRQ(ierr);
13325e8657edSStefano Zampini       ierr = PetscObjectReference((PetscObject)matis->A);CHKERRQ(ierr);
13335e8657edSStefano Zampini       pcbddc->local_mat = matis->A;
13345e8657edSStefano Zampini     }
1335b96c3477SStefano Zampini     /* SetUp coarse and local Neumann solvers */
133699cc7994SStefano Zampini     ierr = PCBDDCSetUpSolvers(pc);CHKERRQ(ierr);
1337b96c3477SStefano Zampini     /* SetUp Scaling operator */
1338674ae819SStefano Zampini     ierr = PCBDDCScalingSetUp(pc);CHKERRQ(ierr);
13390c7d97c5SJed Brown   }
134070cf5478SStefano Zampini 
134158a03d70SStefano Zampini   if (pcbddc->dbg_flag) {
134258a03d70SStefano Zampini     ierr = PetscViewerASCIISubtractTab(pcbddc->dbg_viewer,2*pcbddc->current_level);CHKERRQ(ierr);
13432b510759SStefano Zampini   }
13440c7d97c5SJed Brown   PetscFunctionReturn(0);
13450c7d97c5SJed Brown }
13460c7d97c5SJed Brown 
13470c7d97c5SJed Brown /* -------------------------------------------------------------------------- */
13480c7d97c5SJed Brown /*
134950efa1b5SStefano Zampini    PCApply_BDDC - Applies the BDDC operator to a vector.
13500c7d97c5SJed Brown 
13510c7d97c5SJed Brown    Input Parameters:
13520f202f7eSStefano Zampini +  pc - the preconditioner context
13530f202f7eSStefano Zampini -  r - input vector (global)
13540c7d97c5SJed Brown 
13550c7d97c5SJed Brown    Output Parameter:
13560c7d97c5SJed Brown .  z - output vector (global)
13570c7d97c5SJed Brown 
13580c7d97c5SJed Brown    Application Interface Routine: PCApply()
13590c7d97c5SJed Brown  */
13600c7d97c5SJed Brown #undef __FUNCT__
13610c7d97c5SJed Brown #define __FUNCT__ "PCApply_BDDC"
136253cdbc3dSStefano Zampini PetscErrorCode PCApply_BDDC(PC pc,Vec r,Vec z)
13630c7d97c5SJed Brown {
13640c7d97c5SJed Brown   PC_IS             *pcis = (PC_IS*)(pc->data);
13650c7d97c5SJed Brown   PC_BDDC           *pcbddc = (PC_BDDC*)(pc->data);
1366b097fa66SStefano Zampini   PetscInt          n_B = pcis->n_B, n_D = pcis->n - n_B;
13670c7d97c5SJed Brown   PetscErrorCode    ierr;
13683b03a366Sstefano_zampini   const PetscScalar one = 1.0;
13693b03a366Sstefano_zampini   const PetscScalar m_one = -1.0;
13702617d88aSStefano Zampini   const PetscScalar zero = 0.0;
13710c7d97c5SJed Brown 
13720c7d97c5SJed Brown /* This code is similar to that provided in nn.c for PCNN
13730c7d97c5SJed Brown    NN interface preconditioner changed to BDDC
1374b097fa66SStefano Zampini    Added support for M_3 preconditioner in the reference article (code is active if pcbddc->switch_static == PETSC_TRUE) */
13750c7d97c5SJed Brown 
13760c7d97c5SJed Brown   PetscFunctionBegin;
137785c4d303SStefano Zampini   if (!pcbddc->use_exact_dirichlet_trick) {
1378b097fa66SStefano Zampini     ierr = VecCopy(r,z);CHKERRQ(ierr);
13790c7d97c5SJed Brown     /* First Dirichlet solve */
13800c7d97c5SJed Brown     ierr = VecScatterBegin(pcis->global_to_D,r,pcis->vec1_D,INSERT_VALUES,SCATTER_FORWARD);CHKERRQ(ierr);
13810c7d97c5SJed Brown     ierr = VecScatterEnd(pcis->global_to_D,r,pcis->vec1_D,INSERT_VALUES,SCATTER_FORWARD);CHKERRQ(ierr);
13820c7d97c5SJed Brown     /*
13830c7d97c5SJed Brown       Assembling right hand side for BDDC operator
1384b097fa66SStefano Zampini       - pcis->vec1_D for the Dirichlet part (if needed, i.e. pcbddc->switch_static == PETSC_TRUE)
1385674ae819SStefano Zampini       - pcis->vec1_B the interface part of the global vector z
13860c7d97c5SJed Brown     */
1387b097fa66SStefano Zampini     if (n_D) {
1388b097fa66SStefano Zampini       ierr = KSPSolve(pcbddc->ksp_D,pcis->vec1_D,pcis->vec2_D);CHKERRQ(ierr);
13890c7d97c5SJed Brown       ierr = VecScale(pcis->vec2_D,m_one);CHKERRQ(ierr);
13908eeda7d8SStefano Zampini       if (pcbddc->switch_static) { ierr = MatMultAdd(pcis->A_II,pcis->vec2_D,pcis->vec1_D,pcis->vec1_D);CHKERRQ(ierr); }
1391b097fa66SStefano Zampini       ierr = MatMult(pcis->A_BI,pcis->vec2_D,pcis->vec1_B);CHKERRQ(ierr);
1392b097fa66SStefano Zampini     } else {
1393b097fa66SStefano Zampini       ierr = VecSet(pcis->vec1_B,zero);CHKERRQ(ierr);
1394b097fa66SStefano Zampini     }
13950c7d97c5SJed Brown     ierr = VecScatterBegin(pcis->global_to_B,pcis->vec1_B,z,ADD_VALUES,SCATTER_REVERSE);CHKERRQ(ierr);
13960c7d97c5SJed Brown     ierr = VecScatterEnd(pcis->global_to_B,pcis->vec1_B,z,ADD_VALUES,SCATTER_REVERSE);CHKERRQ(ierr);
1397674ae819SStefano Zampini     ierr = PCBDDCScalingRestriction(pc,z,pcis->vec1_B);CHKERRQ(ierr);
1398b76ba322SStefano Zampini   } else {
1399b097fa66SStefano Zampini     if (pcbddc->switch_static) {
14000bdf917eSStefano Zampini       ierr = VecSet(pcis->vec1_D,zero);CHKERRQ(ierr);
1401b097fa66SStefano Zampini     }
1402674ae819SStefano Zampini     ierr = PCBDDCScalingRestriction(pc,r,pcis->vec1_B);CHKERRQ(ierr);
1403b76ba322SStefano Zampini   }
1404b76ba322SStefano Zampini 
14052617d88aSStefano Zampini   /* Apply interface preconditioner
14062617d88aSStefano Zampini      input/output vecs: pcis->vec1_B and pcis->vec1_D */
1407dc359a40SStefano Zampini   ierr = PCBDDCApplyInterfacePreconditioner(pc,PETSC_FALSE);CHKERRQ(ierr);
14082617d88aSStefano Zampini 
1409674ae819SStefano Zampini   /* Apply transpose of partition of unity operator */
1410674ae819SStefano Zampini   ierr = PCBDDCScalingExtension(pc,pcis->vec1_B,z);CHKERRQ(ierr);
14110c7d97c5SJed Brown 
14123b03a366Sstefano_zampini   /* Second Dirichlet solve and assembling of output */
14130c7d97c5SJed Brown   ierr = VecScatterBegin(pcis->global_to_B,z,pcis->vec1_B,INSERT_VALUES,SCATTER_FORWARD);CHKERRQ(ierr);
14140c7d97c5SJed Brown   ierr = VecScatterEnd(pcis->global_to_B,z,pcis->vec1_B,INSERT_VALUES,SCATTER_FORWARD);CHKERRQ(ierr);
1415b097fa66SStefano Zampini   if (n_B) {
14160c7d97c5SJed Brown     ierr = MatMult(pcis->A_IB,pcis->vec1_B,pcis->vec3_D);CHKERRQ(ierr);
14178eeda7d8SStefano Zampini     if (pcbddc->switch_static) { ierr = MatMultAdd(pcis->A_II,pcis->vec1_D,pcis->vec3_D,pcis->vec3_D);CHKERRQ(ierr); }
1418b097fa66SStefano Zampini   } else if (pcbddc->switch_static) {
1419b097fa66SStefano Zampini     ierr = MatMult(pcis->A_II,pcis->vec1_D,pcis->vec3_D);CHKERRQ(ierr);
1420b097fa66SStefano Zampini   }
1421df187020SStefano Zampini   ierr = KSPSolve(pcbddc->ksp_D,pcis->vec3_D,pcis->vec4_D);CHKERRQ(ierr);
1422b097fa66SStefano Zampini   if (!pcbddc->use_exact_dirichlet_trick) {
1423b097fa66SStefano Zampini     if (pcbddc->switch_static) {
1424b097fa66SStefano Zampini       ierr = VecAXPBYPCZ(pcis->vec2_D,m_one,one,m_one,pcis->vec4_D,pcis->vec1_D);CHKERRQ(ierr);
1425b097fa66SStefano Zampini     } else {
1426b097fa66SStefano Zampini       ierr = VecAXPBY(pcis->vec2_D,m_one,m_one,pcis->vec4_D);CHKERRQ(ierr);
1427b097fa66SStefano Zampini     }
14280c7d97c5SJed Brown     ierr = VecScatterBegin(pcis->global_to_D,pcis->vec2_D,z,INSERT_VALUES,SCATTER_REVERSE);CHKERRQ(ierr);
14290c7d97c5SJed Brown     ierr = VecScatterEnd(pcis->global_to_D,pcis->vec2_D,z,INSERT_VALUES,SCATTER_REVERSE);CHKERRQ(ierr);
1430b097fa66SStefano Zampini   } else {
1431b097fa66SStefano Zampini     if (pcbddc->switch_static) {
1432b097fa66SStefano Zampini       ierr = VecAXPBY(pcis->vec4_D,one,m_one,pcis->vec1_D);CHKERRQ(ierr);
1433b097fa66SStefano Zampini     } else {
1434b097fa66SStefano Zampini       ierr = VecScale(pcis->vec4_D,m_one);CHKERRQ(ierr);
1435b097fa66SStefano Zampini     }
1436b097fa66SStefano Zampini     ierr = VecScatterBegin(pcis->global_to_D,pcis->vec4_D,z,INSERT_VALUES,SCATTER_REVERSE);CHKERRQ(ierr);
1437b097fa66SStefano Zampini     ierr = VecScatterEnd(pcis->global_to_D,pcis->vec4_D,z,INSERT_VALUES,SCATTER_REVERSE);CHKERRQ(ierr);
1438b097fa66SStefano Zampini   }
14390c7d97c5SJed Brown   PetscFunctionReturn(0);
14400c7d97c5SJed Brown }
144150efa1b5SStefano Zampini 
144250efa1b5SStefano Zampini /* -------------------------------------------------------------------------- */
144350efa1b5SStefano Zampini /*
144450efa1b5SStefano Zampini    PCApplyTranspose_BDDC - Applies the transpose of the BDDC operator to a vector.
144550efa1b5SStefano Zampini 
144650efa1b5SStefano Zampini    Input Parameters:
14470f202f7eSStefano Zampini +  pc - the preconditioner context
14480f202f7eSStefano Zampini -  r - input vector (global)
144950efa1b5SStefano Zampini 
145050efa1b5SStefano Zampini    Output Parameter:
145150efa1b5SStefano Zampini .  z - output vector (global)
145250efa1b5SStefano Zampini 
145350efa1b5SStefano Zampini    Application Interface Routine: PCApplyTranspose()
145450efa1b5SStefano Zampini  */
145550efa1b5SStefano Zampini #undef __FUNCT__
145650efa1b5SStefano Zampini #define __FUNCT__ "PCApplyTranspose_BDDC"
145750efa1b5SStefano Zampini PetscErrorCode PCApplyTranspose_BDDC(PC pc,Vec r,Vec z)
145850efa1b5SStefano Zampini {
145950efa1b5SStefano Zampini   PC_IS             *pcis = (PC_IS*)(pc->data);
146050efa1b5SStefano Zampini   PC_BDDC           *pcbddc = (PC_BDDC*)(pc->data);
1461b097fa66SStefano Zampini   PetscInt          n_B = pcis->n_B, n_D = pcis->n - n_B;
146250efa1b5SStefano Zampini   PetscErrorCode    ierr;
146350efa1b5SStefano Zampini   const PetscScalar one = 1.0;
146450efa1b5SStefano Zampini   const PetscScalar m_one = -1.0;
146550efa1b5SStefano Zampini   const PetscScalar zero = 0.0;
146650efa1b5SStefano Zampini 
146750efa1b5SStefano Zampini   PetscFunctionBegin;
146850efa1b5SStefano Zampini   if (!pcbddc->use_exact_dirichlet_trick) {
1469b097fa66SStefano Zampini     ierr = VecCopy(r,z);CHKERRQ(ierr);
147050efa1b5SStefano Zampini     /* First Dirichlet solve */
147150efa1b5SStefano Zampini     ierr = VecScatterBegin(pcis->global_to_D,r,pcis->vec1_D,INSERT_VALUES,SCATTER_FORWARD);CHKERRQ(ierr);
147250efa1b5SStefano Zampini     ierr = VecScatterEnd(pcis->global_to_D,r,pcis->vec1_D,INSERT_VALUES,SCATTER_FORWARD);CHKERRQ(ierr);
147350efa1b5SStefano Zampini     /*
147450efa1b5SStefano Zampini       Assembling right hand side for BDDC operator
1475b097fa66SStefano Zampini       - pcis->vec1_D for the Dirichlet part (if needed, i.e. pcbddc->switch_static == PETSC_TRUE)
147650efa1b5SStefano Zampini       - pcis->vec1_B the interface part of the global vector z
147750efa1b5SStefano Zampini     */
1478b097fa66SStefano Zampini     if (n_D) {
1479b097fa66SStefano Zampini       ierr = KSPSolveTranspose(pcbddc->ksp_D,pcis->vec1_D,pcis->vec2_D);CHKERRQ(ierr);
148050efa1b5SStefano Zampini       ierr = VecScale(pcis->vec2_D,m_one);CHKERRQ(ierr);
148150efa1b5SStefano Zampini       if (pcbddc->switch_static) { ierr = MatMultTransposeAdd(pcis->A_II,pcis->vec2_D,pcis->vec1_D,pcis->vec1_D);CHKERRQ(ierr); }
1482b097fa66SStefano Zampini       ierr = MatMultTranspose(pcis->A_IB,pcis->vec2_D,pcis->vec1_B);CHKERRQ(ierr);
1483b097fa66SStefano Zampini     } else {
1484b097fa66SStefano Zampini       ierr = VecSet(pcis->vec1_B,zero);CHKERRQ(ierr);
1485b097fa66SStefano Zampini     }
148650efa1b5SStefano Zampini     ierr = VecScatterBegin(pcis->global_to_B,pcis->vec1_B,z,ADD_VALUES,SCATTER_REVERSE);CHKERRQ(ierr);
148750efa1b5SStefano Zampini     ierr = VecScatterEnd(pcis->global_to_B,pcis->vec1_B,z,ADD_VALUES,SCATTER_REVERSE);CHKERRQ(ierr);
148850efa1b5SStefano Zampini     ierr = PCBDDCScalingRestriction(pc,z,pcis->vec1_B);CHKERRQ(ierr);
148950efa1b5SStefano Zampini   } else {
1490b097fa66SStefano Zampini     if (pcbddc->switch_static) {
149150efa1b5SStefano Zampini       ierr = VecSet(pcis->vec1_D,zero);CHKERRQ(ierr);
1492b097fa66SStefano Zampini     }
149350efa1b5SStefano Zampini     ierr = PCBDDCScalingRestriction(pc,r,pcis->vec1_B);CHKERRQ(ierr);
149450efa1b5SStefano Zampini   }
149550efa1b5SStefano Zampini 
149650efa1b5SStefano Zampini   /* Apply interface preconditioner
149750efa1b5SStefano Zampini      input/output vecs: pcis->vec1_B and pcis->vec1_D */
1498dc359a40SStefano Zampini   ierr = PCBDDCApplyInterfacePreconditioner(pc,PETSC_TRUE);CHKERRQ(ierr);
149950efa1b5SStefano Zampini 
150050efa1b5SStefano Zampini   /* Apply transpose of partition of unity operator */
150150efa1b5SStefano Zampini   ierr = PCBDDCScalingExtension(pc,pcis->vec1_B,z);CHKERRQ(ierr);
150250efa1b5SStefano Zampini 
150350efa1b5SStefano Zampini   /* Second Dirichlet solve and assembling of output */
150450efa1b5SStefano Zampini   ierr = VecScatterBegin(pcis->global_to_B,z,pcis->vec1_B,INSERT_VALUES,SCATTER_FORWARD);CHKERRQ(ierr);
150550efa1b5SStefano Zampini   ierr = VecScatterEnd(pcis->global_to_B,z,pcis->vec1_B,INSERT_VALUES,SCATTER_FORWARD);CHKERRQ(ierr);
1506b097fa66SStefano Zampini   if (n_B) {
150750efa1b5SStefano Zampini     ierr = MatMultTranspose(pcis->A_BI,pcis->vec1_B,pcis->vec3_D);CHKERRQ(ierr);
150850efa1b5SStefano Zampini     if (pcbddc->switch_static) { ierr = MatMultTransposeAdd(pcis->A_II,pcis->vec1_D,pcis->vec3_D,pcis->vec3_D);CHKERRQ(ierr); }
1509b097fa66SStefano Zampini   } else if (pcbddc->switch_static) {
1510b097fa66SStefano Zampini     ierr = MatMultTranspose(pcis->A_II,pcis->vec1_D,pcis->vec3_D);CHKERRQ(ierr);
1511b097fa66SStefano Zampini   }
1512b0147a47SStefano Zampini   ierr = KSPSolveTranspose(pcbddc->ksp_D,pcis->vec3_D,pcis->vec4_D);CHKERRQ(ierr);
1513b097fa66SStefano Zampini   if (!pcbddc->use_exact_dirichlet_trick) {
1514b097fa66SStefano Zampini     if (pcbddc->switch_static) {
1515b097fa66SStefano Zampini       ierr = VecAXPBYPCZ(pcis->vec2_D,m_one,one,m_one,pcis->vec4_D,pcis->vec1_D);CHKERRQ(ierr);
1516b097fa66SStefano Zampini     } else {
1517b097fa66SStefano Zampini       ierr = VecAXPBY(pcis->vec2_D,m_one,m_one,pcis->vec4_D);CHKERRQ(ierr);
1518b097fa66SStefano Zampini     }
151950efa1b5SStefano Zampini     ierr = VecScatterBegin(pcis->global_to_D,pcis->vec2_D,z,INSERT_VALUES,SCATTER_REVERSE);CHKERRQ(ierr);
152050efa1b5SStefano Zampini     ierr = VecScatterEnd(pcis->global_to_D,pcis->vec2_D,z,INSERT_VALUES,SCATTER_REVERSE);CHKERRQ(ierr);
1521b097fa66SStefano Zampini   } else {
1522b097fa66SStefano Zampini     if (pcbddc->switch_static) {
1523b097fa66SStefano Zampini       ierr = VecAXPBY(pcis->vec4_D,one,m_one,pcis->vec1_D);CHKERRQ(ierr);
1524b097fa66SStefano Zampini     } else {
1525b097fa66SStefano Zampini       ierr = VecScale(pcis->vec4_D,m_one);CHKERRQ(ierr);
1526b097fa66SStefano Zampini     }
1527b097fa66SStefano Zampini     ierr = VecScatterBegin(pcis->global_to_D,pcis->vec4_D,z,INSERT_VALUES,SCATTER_REVERSE);CHKERRQ(ierr);
1528b097fa66SStefano Zampini     ierr = VecScatterEnd(pcis->global_to_D,pcis->vec4_D,z,INSERT_VALUES,SCATTER_REVERSE);CHKERRQ(ierr);
1529b097fa66SStefano Zampini   }
153050efa1b5SStefano Zampini   PetscFunctionReturn(0);
153150efa1b5SStefano Zampini }
1532da1bb401SStefano Zampini /* -------------------------------------------------------------------------- */
1533674ae819SStefano Zampini 
1534da1bb401SStefano Zampini #undef __FUNCT__
1535da1bb401SStefano Zampini #define __FUNCT__ "PCDestroy_BDDC"
1536da1bb401SStefano Zampini PetscErrorCode PCDestroy_BDDC(PC pc)
1537da1bb401SStefano Zampini {
1538da1bb401SStefano Zampini   PC_BDDC        *pcbddc = (PC_BDDC*)pc->data;
1539da1bb401SStefano Zampini   PetscErrorCode ierr;
1540da1bb401SStefano Zampini 
1541da1bb401SStefano Zampini   PetscFunctionBegin;
1542da1bb401SStefano Zampini   /* free data created by PCIS */
1543da1bb401SStefano Zampini   ierr = PCISDestroy(pc);CHKERRQ(ierr);
1544674ae819SStefano Zampini   /* free BDDC custom data  */
1545674ae819SStefano Zampini   ierr = PCBDDCResetCustomization(pc);CHKERRQ(ierr);
1546674ae819SStefano Zampini   /* destroy objects related to topography */
1547674ae819SStefano Zampini   ierr = PCBDDCResetTopography(pc);CHKERRQ(ierr);
1548674ae819SStefano Zampini   /* free allocated graph structure */
1549da1bb401SStefano Zampini   ierr = PetscFree(pcbddc->mat_graph);CHKERRQ(ierr);
1550b96c3477SStefano Zampini   /* free allocated sub schurs structure */
1551b96c3477SStefano Zampini   ierr = PetscFree(pcbddc->sub_schurs);CHKERRQ(ierr);
155234a97f8cSStefano Zampini   /* destroy objects for scaling operator */
1553674ae819SStefano Zampini   ierr = PCBDDCScalingDestroy(pc);CHKERRQ(ierr);
155434a97f8cSStefano Zampini   ierr = PetscFree(pcbddc->deluxe_ctx);CHKERRQ(ierr);
1555674ae819SStefano Zampini   /* free solvers stuff */
1556674ae819SStefano Zampini   ierr = PCBDDCResetSolvers(pc);CHKERRQ(ierr);
155762a6ff1dSStefano Zampini   /* free global vectors needed in presolve */
155862a6ff1dSStefano Zampini   ierr = VecDestroy(&pcbddc->temp_solution);CHKERRQ(ierr);
155962a6ff1dSStefano Zampini   ierr = VecDestroy(&pcbddc->original_rhs);CHKERRQ(ierr);
1560906d46d4SStefano Zampini   /* free stuff for change of basis hooks */
1561906d46d4SStefano Zampini   if (pcbddc->new_global_mat) {
1562906d46d4SStefano Zampini     PCBDDCChange_ctx change_ctx;
1563906d46d4SStefano Zampini     ierr = MatShellGetContext(pcbddc->new_global_mat,&change_ctx);CHKERRQ(ierr);
1564906d46d4SStefano Zampini     ierr = MatDestroy(&change_ctx->original_mat);CHKERRQ(ierr);
1565906d46d4SStefano Zampini     ierr = MatDestroy(&change_ctx->global_change);CHKERRQ(ierr);
1566906d46d4SStefano Zampini     ierr = VecDestroyVecs(2,&change_ctx->work);CHKERRQ(ierr);
1567906d46d4SStefano Zampini     ierr = PetscFree(change_ctx);CHKERRQ(ierr);
1568906d46d4SStefano Zampini   }
1569906d46d4SStefano Zampini   ierr = MatDestroy(&pcbddc->new_global_mat);CHKERRQ(ierr);
15703425bc38SStefano Zampini   /* remove functions */
1571906d46d4SStefano Zampini   ierr = PetscObjectComposeFunction((PetscObject)pc,"PCBDDCSetChangeOfBasisMat_C",NULL);CHKERRQ(ierr);
1572674ae819SStefano Zampini   ierr = PetscObjectComposeFunction((PetscObject)pc,"PCBDDCSetPrimalVerticesLocalIS_C",NULL);CHKERRQ(ierr);
1573bdf89e91SBarry Smith   ierr = PetscObjectComposeFunction((PetscObject)pc,"PCBDDCSetCoarseningRatio_C",NULL);CHKERRQ(ierr);
15742b510759SStefano Zampini   ierr = PetscObjectComposeFunction((PetscObject)pc,"PCBDDCSetLevel_C",NULL);CHKERRQ(ierr);
1575b8ffe317SStefano Zampini   ierr = PetscObjectComposeFunction((PetscObject)pc,"PCBDDCSetUseExactDirichlet_C",NULL);CHKERRQ(ierr);
15762b510759SStefano Zampini   ierr = PetscObjectComposeFunction((PetscObject)pc,"PCBDDCSetLevels_C",NULL);CHKERRQ(ierr);
1577bdf89e91SBarry Smith   ierr = PetscObjectComposeFunction((PetscObject)pc,"PCBDDCSetDirichletBoundaries_C",NULL);CHKERRQ(ierr);
157882ba6b80SStefano Zampini   ierr = PetscObjectComposeFunction((PetscObject)pc,"PCBDDCSetDirichletBoundariesLocal_C",NULL);CHKERRQ(ierr);
1579bdf89e91SBarry Smith   ierr = PetscObjectComposeFunction((PetscObject)pc,"PCBDDCSetNeumannBoundaries_C",NULL);CHKERRQ(ierr);
158082ba6b80SStefano Zampini   ierr = PetscObjectComposeFunction((PetscObject)pc,"PCBDDCSetNeumannBoundariesLocal_C",NULL);CHKERRQ(ierr);
1581bdf89e91SBarry Smith   ierr = PetscObjectComposeFunction((PetscObject)pc,"PCBDDCGetDirichletBoundaries_C",NULL);CHKERRQ(ierr);
158282ba6b80SStefano Zampini   ierr = PetscObjectComposeFunction((PetscObject)pc,"PCBDDCGetDirichletBoundariesLocal_C",NULL);CHKERRQ(ierr);
1583bdf89e91SBarry Smith   ierr = PetscObjectComposeFunction((PetscObject)pc,"PCBDDCGetNeumannBoundaries_C",NULL);CHKERRQ(ierr);
1584785d1243SStefano Zampini   ierr = PetscObjectComposeFunction((PetscObject)pc,"PCBDDCGetNeumannBoundariesLocal_C",NULL);CHKERRQ(ierr);
1585bdf89e91SBarry Smith   ierr = PetscObjectComposeFunction((PetscObject)pc,"PCBDDCSetDofsSplitting_C",NULL);CHKERRQ(ierr);
158663602bcaSStefano Zampini   ierr = PetscObjectComposeFunction((PetscObject)pc,"PCBDDCSetDofsSplittingLocal_C",NULL);CHKERRQ(ierr);
1587bdf89e91SBarry Smith   ierr = PetscObjectComposeFunction((PetscObject)pc,"PCBDDCSetLocalAdjacencyGraph_C",NULL);CHKERRQ(ierr);
1588bdf89e91SBarry Smith   ierr = PetscObjectComposeFunction((PetscObject)pc,"PCBDDCCreateFETIDPOperators_C",NULL);CHKERRQ(ierr);
1589bdf89e91SBarry Smith   ierr = PetscObjectComposeFunction((PetscObject)pc,"PCBDDCMatFETIDPGetRHS_C",NULL);CHKERRQ(ierr);
1590bdf89e91SBarry Smith   ierr = PetscObjectComposeFunction((PetscObject)pc,"PCBDDCMatFETIDPGetSolution_C",NULL);CHKERRQ(ierr);
1591a06fd7f2SStefano Zampini   ierr = PetscObjectComposeFunction((PetscObject)pc,"PCPreSolveChangeRHS_C",NULL);CHKERRQ(ierr);
1592674ae819SStefano Zampini   /* Free the private data structure */
1593674ae819SStefano Zampini   ierr = PetscFree(pc->data);CHKERRQ(ierr);
1594da1bb401SStefano Zampini   PetscFunctionReturn(0);
1595da1bb401SStefano Zampini }
15963425bc38SStefano Zampini /* -------------------------------------------------------------------------- */
15971e6b0712SBarry Smith 
15983425bc38SStefano Zampini #undef __FUNCT__
1599a06fd7f2SStefano Zampini #define __FUNCT__ "PCPreSolveChangeRHS_BDDC"
1600a06fd7f2SStefano Zampini static PetscErrorCode PCPreSolveChangeRHS_BDDC(PC pc, PetscBool* change)
1601a06fd7f2SStefano Zampini {
1602a06fd7f2SStefano Zampini   PetscFunctionBegin;
1603a06fd7f2SStefano Zampini   *change = PETSC_TRUE;
1604a06fd7f2SStefano Zampini   PetscFunctionReturn(0);
1605a06fd7f2SStefano Zampini }
1606a06fd7f2SStefano Zampini 
1607a06fd7f2SStefano Zampini #undef __FUNCT__
16083425bc38SStefano Zampini #define __FUNCT__ "PCBDDCMatFETIDPGetRHS_BDDC"
16093425bc38SStefano Zampini static PetscErrorCode PCBDDCMatFETIDPGetRHS_BDDC(Mat fetidp_mat, Vec standard_rhs, Vec fetidp_flux_rhs)
16103425bc38SStefano Zampini {
1611674ae819SStefano Zampini   FETIDPMat_ctx  mat_ctx;
1612c08af4c6SStefano Zampini   Vec            copy_standard_rhs;
16133425bc38SStefano Zampini   PC_IS*         pcis;
16143425bc38SStefano Zampini   PC_BDDC*       pcbddc;
16153425bc38SStefano Zampini   PetscErrorCode ierr;
16160c7d97c5SJed Brown 
16173425bc38SStefano Zampini   PetscFunctionBegin;
16183425bc38SStefano Zampini   ierr = MatShellGetContext(fetidp_mat,&mat_ctx);CHKERRQ(ierr);
16193425bc38SStefano Zampini   pcis = (PC_IS*)mat_ctx->pc->data;
16203425bc38SStefano Zampini   pcbddc = (PC_BDDC*)mat_ctx->pc->data;
16213425bc38SStefano Zampini 
1622c08af4c6SStefano Zampini   /*
1623c08af4c6SStefano Zampini      change of basis for physical rhs if needed
1624c08af4c6SStefano Zampini      It also changes the rhs in case of dirichlet boundaries
1625c08af4c6SStefano Zampini      TODO: better management when FETIDP will have its own class
1626c08af4c6SStefano Zampini   */
1627c08af4c6SStefano Zampini   ierr = VecDuplicate(standard_rhs,&copy_standard_rhs);CHKERRQ(ierr);
1628c08af4c6SStefano Zampini   ierr = VecCopy(standard_rhs,copy_standard_rhs);CHKERRQ(ierr);
1629c08af4c6SStefano Zampini   ierr = PCPreSolve_BDDC(mat_ctx->pc,NULL,copy_standard_rhs,NULL);CHKERRQ(ierr);
16303425bc38SStefano Zampini   /* store vectors for computation of fetidp final solution */
1631c08af4c6SStefano Zampini   ierr = VecScatterBegin(pcis->global_to_D,copy_standard_rhs,mat_ctx->temp_solution_D,INSERT_VALUES,SCATTER_FORWARD);CHKERRQ(ierr);
1632c08af4c6SStefano Zampini   ierr = VecScatterEnd(pcis->global_to_D,copy_standard_rhs,mat_ctx->temp_solution_D,INSERT_VALUES,SCATTER_FORWARD);CHKERRQ(ierr);
1633fb223d50SStefano Zampini   /* scale rhs since it should be unassembled */
1634fb223d50SStefano Zampini   /* TODO use counter scaling? (also below) */
1635c08af4c6SStefano Zampini   ierr = VecScatterBegin(pcis->global_to_B,copy_standard_rhs,mat_ctx->temp_solution_B,INSERT_VALUES,SCATTER_FORWARD);CHKERRQ(ierr);
1636c08af4c6SStefano Zampini   ierr = VecScatterEnd(pcis->global_to_B,copy_standard_rhs,mat_ctx->temp_solution_B,INSERT_VALUES,SCATTER_FORWARD);CHKERRQ(ierr);
1637674ae819SStefano Zampini   /* Apply partition of unity */
16383425bc38SStefano Zampini   ierr = VecPointwiseMult(mat_ctx->temp_solution_B,pcis->D,mat_ctx->temp_solution_B);CHKERRQ(ierr);
1639c08af4c6SStefano Zampini   /* ierr = PCBDDCScalingRestriction(mat_ctx->pc,copy_standard_rhs,mat_ctx->temp_solution_B);CHKERRQ(ierr); */
16408eeda7d8SStefano Zampini   if (!pcbddc->switch_static) {
16413425bc38SStefano Zampini     /* compute partially subassembled Schur complement right-hand side */
16423425bc38SStefano Zampini     ierr = KSPSolve(pcbddc->ksp_D,mat_ctx->temp_solution_D,pcis->vec1_D);CHKERRQ(ierr);
16433425bc38SStefano Zampini     ierr = MatMult(pcis->A_BI,pcis->vec1_D,pcis->vec1_B);CHKERRQ(ierr);
16443425bc38SStefano Zampini     ierr = VecAXPY(mat_ctx->temp_solution_B,-1.0,pcis->vec1_B);CHKERRQ(ierr);
1645c08af4c6SStefano Zampini     ierr = VecSet(copy_standard_rhs,0.0);CHKERRQ(ierr);
1646c08af4c6SStefano Zampini     ierr = VecScatterBegin(pcis->global_to_B,mat_ctx->temp_solution_B,copy_standard_rhs,ADD_VALUES,SCATTER_REVERSE);CHKERRQ(ierr);
1647c08af4c6SStefano Zampini     ierr = VecScatterEnd(pcis->global_to_B,mat_ctx->temp_solution_B,copy_standard_rhs,ADD_VALUES,SCATTER_REVERSE);CHKERRQ(ierr);
1648c08af4c6SStefano Zampini     /* ierr = PCBDDCScalingRestriction(mat_ctx->pc,copy_standard_rhs,mat_ctx->temp_solution_B);CHKERRQ(ierr); */
1649c08af4c6SStefano Zampini     ierr = VecScatterBegin(pcis->global_to_B,copy_standard_rhs,mat_ctx->temp_solution_B,INSERT_VALUES,SCATTER_FORWARD);CHKERRQ(ierr);
1650c08af4c6SStefano Zampini     ierr = VecScatterEnd(pcis->global_to_B,copy_standard_rhs,mat_ctx->temp_solution_B,INSERT_VALUES,SCATTER_FORWARD);CHKERRQ(ierr);
16513425bc38SStefano Zampini     ierr = VecPointwiseMult(mat_ctx->temp_solution_B,pcis->D,mat_ctx->temp_solution_B);CHKERRQ(ierr);
16523425bc38SStefano Zampini   }
1653c08af4c6SStefano Zampini   ierr = VecDestroy(&copy_standard_rhs);CHKERRQ(ierr);
16543425bc38SStefano Zampini   /* BDDC rhs */
16553425bc38SStefano Zampini   ierr = VecCopy(mat_ctx->temp_solution_B,pcis->vec1_B);CHKERRQ(ierr);
16568eeda7d8SStefano Zampini   if (pcbddc->switch_static) {
16573425bc38SStefano Zampini     ierr = VecCopy(mat_ctx->temp_solution_D,pcis->vec1_D);CHKERRQ(ierr);
16583425bc38SStefano Zampini   }
16593425bc38SStefano Zampini   /* apply BDDC */
1660dc359a40SStefano Zampini   ierr = PCBDDCApplyInterfacePreconditioner(mat_ctx->pc,PETSC_FALSE);CHKERRQ(ierr);
16613425bc38SStefano Zampini   /* Application of B_delta and assembling of rhs for fetidp fluxes */
16623425bc38SStefano Zampini   ierr = VecSet(fetidp_flux_rhs,0.0);CHKERRQ(ierr);
16633425bc38SStefano Zampini   ierr = MatMult(mat_ctx->B_delta,pcis->vec1_B,mat_ctx->lambda_local);CHKERRQ(ierr);
16643425bc38SStefano Zampini   ierr = VecScatterBegin(mat_ctx->l2g_lambda,mat_ctx->lambda_local,fetidp_flux_rhs,ADD_VALUES,SCATTER_FORWARD);CHKERRQ(ierr);
16653425bc38SStefano Zampini   ierr = VecScatterEnd(mat_ctx->l2g_lambda,mat_ctx->lambda_local,fetidp_flux_rhs,ADD_VALUES,SCATTER_FORWARD);CHKERRQ(ierr);
16663425bc38SStefano Zampini   PetscFunctionReturn(0);
16673425bc38SStefano Zampini }
16681e6b0712SBarry Smith 
16693425bc38SStefano Zampini #undef __FUNCT__
16703425bc38SStefano Zampini #define __FUNCT__ "PCBDDCMatFETIDPGetRHS"
16713425bc38SStefano Zampini /*@
16720f202f7eSStefano Zampini  PCBDDCMatFETIDPGetRHS - Compute the right-hand side for FETI-DP linear system using the physical right-hand side
16733425bc38SStefano Zampini 
16743425bc38SStefano Zampini    Collective
16753425bc38SStefano Zampini 
16763425bc38SStefano Zampini    Input Parameters:
16770f202f7eSStefano Zampini +  fetidp_mat      - the FETI-DP matrix object obtained by a call to PCBDDCCreateFETIDPOperators
16780f202f7eSStefano Zampini -  standard_rhs    - the right-hand side of the original linear system
16793425bc38SStefano Zampini 
16803425bc38SStefano Zampini    Output Parameters:
16810f202f7eSStefano Zampini .  fetidp_flux_rhs - the right-hand side for the FETI-DP linear system
16823425bc38SStefano Zampini 
16833425bc38SStefano Zampini    Level: developer
16843425bc38SStefano Zampini 
16853425bc38SStefano Zampini    Notes:
16863425bc38SStefano Zampini 
16870f202f7eSStefano Zampini .seealso: PCBDDC, PCBDDCCreateFETIDPOperators, PCBDDCMatFETIDPGetSolution
16883425bc38SStefano Zampini @*/
16893425bc38SStefano Zampini PetscErrorCode PCBDDCMatFETIDPGetRHS(Mat fetidp_mat, Vec standard_rhs, Vec fetidp_flux_rhs)
16903425bc38SStefano Zampini {
1691674ae819SStefano Zampini   FETIDPMat_ctx  mat_ctx;
16923425bc38SStefano Zampini   PetscErrorCode ierr;
16933425bc38SStefano Zampini 
16943425bc38SStefano Zampini   PetscFunctionBegin;
16953425bc38SStefano Zampini   ierr = MatShellGetContext(fetidp_mat,&mat_ctx);CHKERRQ(ierr);
1696163d334eSBarry Smith   ierr = PetscUseMethod(mat_ctx->pc,"PCBDDCMatFETIDPGetRHS_C",(Mat,Vec,Vec),(fetidp_mat,standard_rhs,fetidp_flux_rhs));CHKERRQ(ierr);
16973425bc38SStefano Zampini   PetscFunctionReturn(0);
16983425bc38SStefano Zampini }
16993425bc38SStefano Zampini /* -------------------------------------------------------------------------- */
17001e6b0712SBarry Smith 
17013425bc38SStefano Zampini #undef __FUNCT__
17023425bc38SStefano Zampini #define __FUNCT__ "PCBDDCMatFETIDPGetSolution_BDDC"
17033425bc38SStefano Zampini static PetscErrorCode PCBDDCMatFETIDPGetSolution_BDDC(Mat fetidp_mat, Vec fetidp_flux_sol, Vec standard_sol)
17043425bc38SStefano Zampini {
1705674ae819SStefano Zampini   FETIDPMat_ctx  mat_ctx;
17063425bc38SStefano Zampini   PC_IS*         pcis;
17073425bc38SStefano Zampini   PC_BDDC*       pcbddc;
17083425bc38SStefano Zampini   PetscErrorCode ierr;
17093425bc38SStefano Zampini 
17103425bc38SStefano Zampini   PetscFunctionBegin;
17113425bc38SStefano Zampini   ierr = MatShellGetContext(fetidp_mat,&mat_ctx);CHKERRQ(ierr);
17123425bc38SStefano Zampini   pcis = (PC_IS*)mat_ctx->pc->data;
17133425bc38SStefano Zampini   pcbddc = (PC_BDDC*)mat_ctx->pc->data;
17143425bc38SStefano Zampini 
17153425bc38SStefano Zampini   /* apply B_delta^T */
17163425bc38SStefano Zampini   ierr = VecScatterBegin(mat_ctx->l2g_lambda,fetidp_flux_sol,mat_ctx->lambda_local,INSERT_VALUES,SCATTER_REVERSE);CHKERRQ(ierr);
17173425bc38SStefano Zampini   ierr = VecScatterEnd(mat_ctx->l2g_lambda,fetidp_flux_sol,mat_ctx->lambda_local,INSERT_VALUES,SCATTER_REVERSE);CHKERRQ(ierr);
17183425bc38SStefano Zampini   ierr = MatMultTranspose(mat_ctx->B_delta,mat_ctx->lambda_local,pcis->vec1_B);CHKERRQ(ierr);
17193425bc38SStefano Zampini   /* compute rhs for BDDC application */
17203425bc38SStefano Zampini   ierr = VecAYPX(pcis->vec1_B,-1.0,mat_ctx->temp_solution_B);CHKERRQ(ierr);
17218eeda7d8SStefano Zampini   if (pcbddc->switch_static) {
17223425bc38SStefano Zampini     ierr = VecCopy(mat_ctx->temp_solution_D,pcis->vec1_D);CHKERRQ(ierr);
17233425bc38SStefano Zampini   }
17243425bc38SStefano Zampini   /* apply BDDC */
1725dc359a40SStefano Zampini   ierr = PCBDDCApplyInterfacePreconditioner(mat_ctx->pc,PETSC_FALSE);CHKERRQ(ierr);
17263425bc38SStefano Zampini   /* put values into standard global vector */
17273425bc38SStefano Zampini   ierr = VecScatterBegin(pcis->global_to_B,pcis->vec1_B,standard_sol,INSERT_VALUES,SCATTER_REVERSE);CHKERRQ(ierr);
17283425bc38SStefano Zampini   ierr = VecScatterEnd  (pcis->global_to_B,pcis->vec1_B,standard_sol,INSERT_VALUES,SCATTER_REVERSE);CHKERRQ(ierr);
17298eeda7d8SStefano Zampini   if (!pcbddc->switch_static) {
17303425bc38SStefano Zampini     /* compute values into the interior if solved for the partially subassembled Schur complement */
17313425bc38SStefano Zampini     ierr = MatMult(pcis->A_IB,pcis->vec1_B,pcis->vec1_D);CHKERRQ(ierr);
1732*00f6b531SStefano Zampini     ierr = VecAYPX(pcis->vec1_D,-1.0,mat_ctx->temp_solution_D);CHKERRQ(ierr);
1733*00f6b531SStefano Zampini     ierr = KSPSolve(pcbddc->ksp_D,pcis->vec1_D,pcis->vec1_D);CHKERRQ(ierr);
17343425bc38SStefano Zampini   }
17353425bc38SStefano Zampini   ierr = VecScatterBegin(pcis->global_to_D,pcis->vec1_D,standard_sol,INSERT_VALUES,SCATTER_REVERSE);CHKERRQ(ierr);
17363425bc38SStefano Zampini   ierr = VecScatterEnd  (pcis->global_to_D,pcis->vec1_D,standard_sol,INSERT_VALUES,SCATTER_REVERSE);CHKERRQ(ierr);
17373425bc38SStefano Zampini   /* final change of basis if needed
17383425bc38SStefano Zampini      Is also sums the dirichlet part removed during RHS assembling */
17393308cffdSStefano Zampini   ierr = PCPostSolve_BDDC(mat_ctx->pc,NULL,NULL,standard_sol);CHKERRQ(ierr);
17403425bc38SStefano Zampini   PetscFunctionReturn(0);
17413425bc38SStefano Zampini }
17421e6b0712SBarry Smith 
17433425bc38SStefano Zampini #undef __FUNCT__
17443425bc38SStefano Zampini #define __FUNCT__ "PCBDDCMatFETIDPGetSolution"
17453425bc38SStefano Zampini /*@
17460f202f7eSStefano Zampini  PCBDDCMatFETIDPGetSolution - Compute the physical solution using the solution of the FETI-DP linear system
17473425bc38SStefano Zampini 
17483425bc38SStefano Zampini    Collective
17493425bc38SStefano Zampini 
17503425bc38SStefano Zampini    Input Parameters:
17510f202f7eSStefano Zampini +  fetidp_mat      - the FETI-DP matrix obtained by a call to PCBDDCCreateFETIDPOperators
17520f202f7eSStefano Zampini -  fetidp_flux_sol - the solution of the FETI-DP linear system
17533425bc38SStefano Zampini 
17543425bc38SStefano Zampini    Output Parameters:
17550f202f7eSStefano Zampini .  standard_sol    - the solution defined on the physical domain
17563425bc38SStefano Zampini 
17573425bc38SStefano Zampini    Level: developer
17583425bc38SStefano Zampini 
17593425bc38SStefano Zampini    Notes:
17603425bc38SStefano Zampini 
17610f202f7eSStefano Zampini .seealso: PCBDDC, PCBDDCCreateFETIDPOperators, PCBDDCMatFETIDPGetRHS
17623425bc38SStefano Zampini @*/
17633425bc38SStefano Zampini PetscErrorCode PCBDDCMatFETIDPGetSolution(Mat fetidp_mat, Vec fetidp_flux_sol, Vec standard_sol)
17643425bc38SStefano Zampini {
1765674ae819SStefano Zampini   FETIDPMat_ctx  mat_ctx;
17663425bc38SStefano Zampini   PetscErrorCode ierr;
17673425bc38SStefano Zampini 
17683425bc38SStefano Zampini   PetscFunctionBegin;
17693425bc38SStefano Zampini   ierr = MatShellGetContext(fetidp_mat,&mat_ctx);CHKERRQ(ierr);
1770163d334eSBarry Smith   ierr = PetscUseMethod(mat_ctx->pc,"PCBDDCMatFETIDPGetSolution_C",(Mat,Vec,Vec),(fetidp_mat,fetidp_flux_sol,standard_sol));CHKERRQ(ierr);
17713425bc38SStefano Zampini   PetscFunctionReturn(0);
17723425bc38SStefano Zampini }
17733425bc38SStefano Zampini /* -------------------------------------------------------------------------- */
17741e6b0712SBarry Smith 
1775f23aa3ddSBarry Smith extern PetscErrorCode FETIDPMatMult(Mat,Vec,Vec);
1776edf7251bSStefano Zampini extern PetscErrorCode FETIDPMatMultTranspose(Mat,Vec,Vec);
1777f23aa3ddSBarry Smith extern PetscErrorCode PCBDDCDestroyFETIDPMat(Mat);
1778f23aa3ddSBarry Smith extern PetscErrorCode FETIDPPCApply(PC,Vec,Vec);
1779edf7251bSStefano Zampini extern PetscErrorCode FETIDPPCApplyTranspose(PC,Vec,Vec);
1780f23aa3ddSBarry Smith extern PetscErrorCode PCBDDCDestroyFETIDPPC(PC);
1781674ae819SStefano Zampini 
17823425bc38SStefano Zampini #undef __FUNCT__
17833425bc38SStefano Zampini #define __FUNCT__ "PCBDDCCreateFETIDPOperators_BDDC"
17841720468bSStefano Zampini static PetscErrorCode PCBDDCCreateFETIDPOperators_BDDC(PC pc, PetscBool fully_redundant, Mat *fetidp_mat, PC *fetidp_pc)
17853425bc38SStefano Zampini {
1786674ae819SStefano Zampini 
1787674ae819SStefano Zampini   FETIDPMat_ctx  fetidpmat_ctx;
17883425bc38SStefano Zampini   Mat            newmat;
1789674ae819SStefano Zampini   FETIDPPC_ctx   fetidppc_ctx;
17903425bc38SStefano Zampini   PC             newpc;
1791ce94432eSBarry Smith   MPI_Comm       comm;
17923425bc38SStefano Zampini   PetscErrorCode ierr;
17933425bc38SStefano Zampini 
17943425bc38SStefano Zampini   PetscFunctionBegin;
1795ce94432eSBarry Smith   ierr = PetscObjectGetComm((PetscObject)pc,&comm);CHKERRQ(ierr);
17963425bc38SStefano Zampini   /* FETIDP linear matrix */
17973425bc38SStefano Zampini   ierr = PCBDDCCreateFETIDPMatContext(pc,&fetidpmat_ctx);CHKERRQ(ierr);
17981720468bSStefano Zampini   fetidpmat_ctx->fully_redundant = fully_redundant;
17993425bc38SStefano Zampini   ierr = PCBDDCSetupFETIDPMatContext(fetidpmat_ctx);CHKERRQ(ierr);
18003425bc38SStefano Zampini   ierr = MatCreateShell(comm,PETSC_DECIDE,PETSC_DECIDE,fetidpmat_ctx->n_lambda,fetidpmat_ctx->n_lambda,fetidpmat_ctx,&newmat);CHKERRQ(ierr);
18013425bc38SStefano Zampini   ierr = MatShellSetOperation(newmat,MATOP_MULT,(void (*)(void))FETIDPMatMult);CHKERRQ(ierr);
1802edf7251bSStefano Zampini   ierr = MatShellSetOperation(newmat,MATOP_MULT_TRANSPOSE,(void (*)(void))FETIDPMatMultTranspose);CHKERRQ(ierr);
18033425bc38SStefano Zampini   ierr = MatShellSetOperation(newmat,MATOP_DESTROY,(void (*)(void))PCBDDCDestroyFETIDPMat);CHKERRQ(ierr);
18043425bc38SStefano Zampini   ierr = MatSetUp(newmat);CHKERRQ(ierr);
18053425bc38SStefano Zampini   /* FETIDP preconditioner */
18063425bc38SStefano Zampini   ierr = PCBDDCCreateFETIDPPCContext(pc,&fetidppc_ctx);CHKERRQ(ierr);
18073425bc38SStefano Zampini   ierr = PCBDDCSetupFETIDPPCContext(newmat,fetidppc_ctx);CHKERRQ(ierr);
18083425bc38SStefano Zampini   ierr = PCCreate(comm,&newpc);CHKERRQ(ierr);
18093425bc38SStefano Zampini   ierr = PCSetType(newpc,PCSHELL);CHKERRQ(ierr);
18103425bc38SStefano Zampini   ierr = PCShellSetContext(newpc,fetidppc_ctx);CHKERRQ(ierr);
18113425bc38SStefano Zampini   ierr = PCShellSetApply(newpc,FETIDPPCApply);CHKERRQ(ierr);
1812edf7251bSStefano Zampini   ierr = PCShellSetApplyTranspose(newpc,FETIDPPCApplyTranspose);CHKERRQ(ierr);
18133425bc38SStefano Zampini   ierr = PCShellSetDestroy(newpc,PCBDDCDestroyFETIDPPC);CHKERRQ(ierr);
181423ee1639SBarry Smith   ierr = PCSetOperators(newpc,newmat,newmat);CHKERRQ(ierr);
18153425bc38SStefano Zampini   ierr = PCSetUp(newpc);CHKERRQ(ierr);
18163425bc38SStefano Zampini   /* return pointers for objects created */
18173425bc38SStefano Zampini   *fetidp_mat=newmat;
18183425bc38SStefano Zampini   *fetidp_pc=newpc;
18193425bc38SStefano Zampini   PetscFunctionReturn(0);
18203425bc38SStefano Zampini }
18211e6b0712SBarry Smith 
18223425bc38SStefano Zampini #undef __FUNCT__
18233425bc38SStefano Zampini #define __FUNCT__ "PCBDDCCreateFETIDPOperators"
18243425bc38SStefano Zampini /*@
18250f202f7eSStefano Zampini  PCBDDCCreateFETIDPOperators - Create FETI-DP operators
18263425bc38SStefano Zampini 
18273425bc38SStefano Zampini    Collective
18283425bc38SStefano Zampini 
18293425bc38SStefano Zampini    Input Parameters:
18301720468bSStefano Zampini +  pc - the BDDC preconditioning context (setup should have been called before)
18311720468bSStefano Zampini -  fully_redundant - true for a fully redundant set of Lagrange multipliers
183228509bceSStefano Zampini 
183328509bceSStefano Zampini    Output Parameters:
18340f202f7eSStefano Zampini +  fetidp_mat - shell FETI-DP matrix object
18350f202f7eSStefano Zampini -  fetidp_pc  - shell Dirichlet preconditioner for FETI-DP matrix
183628509bceSStefano Zampini 
18373425bc38SStefano Zampini    Level: developer
18383425bc38SStefano Zampini 
18393425bc38SStefano Zampini    Notes:
18400f202f7eSStefano Zampini      Currently the only operations provided for FETI-DP matrix are MatMult and MatMultTranspose
18413425bc38SStefano Zampini 
18420f202f7eSStefano Zampini .seealso: PCBDDC, PCBDDCMatFETIDPGetRHS, PCBDDCMatFETIDPGetSolution
18433425bc38SStefano Zampini @*/
18441720468bSStefano Zampini PetscErrorCode PCBDDCCreateFETIDPOperators(PC pc, PetscBool fully_redundant, Mat *fetidp_mat, PC *fetidp_pc)
18453425bc38SStefano Zampini {
18463425bc38SStefano Zampini   PetscErrorCode ierr;
18473425bc38SStefano Zampini 
18483425bc38SStefano Zampini   PetscFunctionBegin;
18493425bc38SStefano Zampini   PetscValidHeaderSpecific(pc,PC_CLASSID,1);
18503425bc38SStefano Zampini   if (pc->setupcalled) {
18511720468bSStefano Zampini     ierr = PetscUseMethod(pc,"PCBDDCCreateFETIDPOperators_C",(PC,PetscBool,Mat*,PC*),(pc,fully_redundant,fetidp_mat,fetidp_pc));CHKERRQ(ierr);
1852f23aa3ddSBarry Smith   } else SETERRQ(PETSC_COMM_SELF,PETSC_ERR_SUP,"You must call PCSetup_BDDC() first \n");
18533425bc38SStefano Zampini   PetscFunctionReturn(0);
18543425bc38SStefano Zampini }
18550c7d97c5SJed Brown /* -------------------------------------------------------------------------- */
1856da1bb401SStefano Zampini /*MC
1857da1bb401SStefano Zampini    PCBDDC - Balancing Domain Decomposition by Constraints.
18580c7d97c5SJed Brown 
185928509bceSStefano Zampini    An implementation of the BDDC preconditioner based on
186028509bceSStefano Zampini 
186128509bceSStefano Zampini .vb
186228509bceSStefano Zampini    [1] C. R. Dohrmann. "An approximate BDDC preconditioner", Numerical Linear Algebra with Applications Volume 14, Issue 2, pages 149-168, March 2007
186328509bceSStefano 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
186428509bceSStefano Zampini    [3] J. Mandel, B. Sousedik, C. R. Dohrmann. "Multispace and Multilevel BDDC", http://arxiv.org/abs/0712.3977
18650f202f7eSStefano 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
186628509bceSStefano Zampini .ve
186728509bceSStefano Zampini 
186828509bceSStefano Zampini    The matrix to be preconditioned (Pmat) must be of type MATIS.
186928509bceSStefano Zampini 
18700f202f7eSStefano Zampini    Currently works with MATIS matrices with local matrices of type MATSEQAIJ, MATSEQBAIJ or MATSEQSBAIJ, either with real or complex numbers.
187128509bceSStefano Zampini 
187228509bceSStefano Zampini    It also works with unsymmetric and indefinite problems.
187328509bceSStefano Zampini 
1874b6fdb6dfSStefano 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.
1875b6fdb6dfSStefano Zampini 
1876c7017625SStefano Zampini    Approximate local solvers are automatically adapted (see [1]) if the user has attached a nullspace object to the subdomain matrices, and informed BDDC of using approximate solvers (via the command line).
187728509bceSStefano Zampini 
18780f202f7eSStefano 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()
18790f202f7eSStefano Zampini    Additional information on dofs can be provided by using PCBDDCSetDofsSplitting(), PCBDDCSetDirichletBoundaries(), PCBDDCSetNeumannBoundaries(), and PCBDDCSetPrimalVerticesLocalIS()
188028509bceSStefano Zampini 
18810f202f7eSStefano Zampini    Constraints can be customized by attaching a MatNullSpace object to the MATIS matrix via MatSetNearNullSpace(). Non-singular modes are retained via SVD.
188228509bceSStefano Zampini 
18830f202f7eSStefano 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.
18840f202f7eSStefano Zampini    User defined change of basis can be passed to PCBDDC by using PCBDDCSetChangeOfBasisMat()
188528509bceSStefano Zampini 
18860f202f7eSStefano Zampini    The PETSc implementation also supports multilevel BDDC [3]. Coarse grids are partitioned using a MatPartitioning object.
188728509bceSStefano Zampini 
18880f202f7eSStefano 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.
188928509bceSStefano Zampini 
18900f202f7eSStefano 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.
18910f202f7eSStefano Zampini    Deluxe scaling is not supported yet for FETI-DP.
18920f202f7eSStefano Zampini 
18930f202f7eSStefano Zampini    Options Database Keys (some of them, run with -h for a complete list):
18940f202f7eSStefano Zampini 
18950f202f7eSStefano Zampini .    -pc_bddc_use_vertices <true> - use or not vertices in primal space
18960f202f7eSStefano Zampini .    -pc_bddc_use_edges <true> - use or not edges in primal space
18970f202f7eSStefano Zampini .    -pc_bddc_use_faces <false> - use or not faces in primal space
18980f202f7eSStefano Zampini .    -pc_bddc_symmetric <true> - symmetric computation of primal basis functions. Specify false for unsymmetric problems
18990f202f7eSStefano Zampini .    -pc_bddc_use_change_of_basis <false> - use change of basis approach (on edges only)
19000f202f7eSStefano Zampini .    -pc_bddc_use_change_on_faces <false> - use change of basis approach on faces if change of basis has been requested
19010f202f7eSStefano Zampini .    -pc_bddc_switch_static <false> - switches from M_2 (default) to M_3 operator (see reference article [1])
190228509bceSStefano Zampini .    -pc_bddc_levels <0> - maximum number of levels for multilevel
19030f202f7eSStefano 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)
19040f202f7eSStefano 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)
19050f202f7eSStefano Zampini .    -pc_bddc_use_deluxe_scaling <false> - use deluxe scaling
19060f202f7eSStefano 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)
19070f202f7eSStefano 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)
190828509bceSStefano Zampini -    -pc_bddc_check_level <0> - set verbosity level of debugging output
190928509bceSStefano Zampini 
191028509bceSStefano Zampini    Options for Dirichlet, Neumann or coarse solver can be set with
191128509bceSStefano Zampini .vb
191228509bceSStefano Zampini       -pc_bddc_dirichlet_
191328509bceSStefano Zampini       -pc_bddc_neumann_
191428509bceSStefano Zampini       -pc_bddc_coarse_
191528509bceSStefano Zampini .ve
19160f202f7eSStefano Zampini    e.g -pc_bddc_dirichlet_ksp_type richardson -pc_bddc_dirichlet_pc_type gamg. PCBDDC uses by default KPSPREONLY and PCLU.
191728509bceSStefano Zampini 
19180f202f7eSStefano Zampini    When using a multilevel approach, solvers' options at the N-th level (N > 1) can be specified as
191928509bceSStefano Zampini .vb
1920312be037SStefano Zampini       -pc_bddc_dirichlet_lN_
1921312be037SStefano Zampini       -pc_bddc_neumann_lN_
1922312be037SStefano Zampini       -pc_bddc_coarse_lN_
192328509bceSStefano Zampini .ve
19240f202f7eSStefano Zampini    Note that level number ranges from the finest (0) to the coarsest (N).
19250f202f7eSStefano 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.
19260f202f7eSStefano Zampini .vb
19270f202f7eSStefano Zampini      -pc_bddc_coarse_pc_bddc_adaptive_threshold 5 -pc_bddc_coarse_l1_pc_bddc_redistribute 3
19280f202f7eSStefano Zampini .ve
19290f202f7eSStefano 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
1930da1bb401SStefano Zampini 
1931da1bb401SStefano Zampini    Level: intermediate
1932da1bb401SStefano Zampini 
1933b6fdb6dfSStefano Zampini    Developer notes:
1934da1bb401SStefano Zampini 
1935da1bb401SStefano Zampini    Contributed by Stefano Zampini
1936da1bb401SStefano Zampini 
1937da1bb401SStefano Zampini .seealso:  PCCreate(), PCSetType(), PCType (for list of available types), PC,  MATIS
1938da1bb401SStefano Zampini M*/
1939b2573a8aSBarry Smith 
1940da1bb401SStefano Zampini #undef __FUNCT__
1941da1bb401SStefano Zampini #define __FUNCT__ "PCCreate_BDDC"
19428cc058d9SJed Brown PETSC_EXTERN PetscErrorCode PCCreate_BDDC(PC pc)
1943da1bb401SStefano Zampini {
1944da1bb401SStefano Zampini   PetscErrorCode      ierr;
1945da1bb401SStefano Zampini   PC_BDDC             *pcbddc;
1946da1bb401SStefano Zampini 
1947da1bb401SStefano Zampini   PetscFunctionBegin;
1948da1bb401SStefano Zampini   /* Creates the private data structure for this preconditioner and attach it to the PC object. */
1949b00a9115SJed Brown   ierr      = PetscNewLog(pc,&pcbddc);CHKERRQ(ierr);
1950da1bb401SStefano Zampini   pc->data  = (void*)pcbddc;
1951da1bb401SStefano Zampini 
1952da1bb401SStefano Zampini   /* create PCIS data structure */
1953da1bb401SStefano Zampini   ierr = PCISCreate(pc);CHKERRQ(ierr);
1954da1bb401SStefano Zampini 
195547d04d0dSStefano Zampini   /* BDDC customization */
195608a5cf49SStefano Zampini   pcbddc->use_local_adj       = PETSC_TRUE;
195747d04d0dSStefano Zampini   pcbddc->use_vertices        = PETSC_TRUE;
195847d04d0dSStefano Zampini   pcbddc->use_edges           = PETSC_TRUE;
195947d04d0dSStefano Zampini   pcbddc->use_faces           = PETSC_FALSE;
196047d04d0dSStefano Zampini   pcbddc->use_change_of_basis = PETSC_FALSE;
196147d04d0dSStefano Zampini   pcbddc->use_change_on_faces = PETSC_FALSE;
196247d04d0dSStefano Zampini   pcbddc->switch_static       = PETSC_FALSE;
1963fa434743SStefano Zampini   pcbddc->use_nnsp_true       = PETSC_FALSE;
1964fa434743SStefano Zampini   pcbddc->use_qr_single       = PETSC_FALSE;
19653301b35fSStefano Zampini   pcbddc->symmetric_primal    = PETSC_TRUE;
196647d04d0dSStefano Zampini   pcbddc->dbg_flag            = 0;
1967b9d89cd5SStefano Zampini   /* private */
1968727cdba6SStefano Zampini   pcbddc->local_primal_size          = 0;
19690e6343abSStefano Zampini   pcbddc->local_primal_size_cc       = 0;
19700e6343abSStefano Zampini   pcbddc->local_primal_ref_node      = 0;
19710e6343abSStefano Zampini   pcbddc->local_primal_ref_mult      = 0;
1972e9189074SStefano Zampini   pcbddc->n_vertices                 = 0;
1973727cdba6SStefano Zampini   pcbddc->primal_indices_local_idxs  = 0;
1974fb180af4SStefano Zampini   pcbddc->recompute_topography       = PETSC_FALSE;
197568457ee5SStefano Zampini   pcbddc->coarse_size                = -1;
1976f4ddd8eeSStefano Zampini   pcbddc->new_primal_space           = PETSC_FALSE;
1977727cdba6SStefano Zampini   pcbddc->new_primal_space_local     = PETSC_FALSE;
1978f4ddd8eeSStefano Zampini   pcbddc->global_primal_indices      = 0;
1979f4ddd8eeSStefano Zampini   pcbddc->onearnullspace             = 0;
1980f4ddd8eeSStefano Zampini   pcbddc->onearnullvecs_state        = 0;
1981674ae819SStefano Zampini   pcbddc->user_primal_vertices       = 0;
19823972b0daSStefano Zampini   pcbddc->temp_solution              = 0;
1983534831adSStefano Zampini   pcbddc->original_rhs               = 0;
1984534831adSStefano Zampini   pcbddc->local_mat                  = 0;
1985534831adSStefano Zampini   pcbddc->ChangeOfBasisMatrix        = 0;
1986b9b85e73SStefano Zampini   pcbddc->user_ChangeOfBasisMatrix   = 0;
1987906d46d4SStefano Zampini   pcbddc->new_global_mat             = 0;
1988da1bb401SStefano Zampini   pcbddc->coarse_vec                 = 0;
1989da1bb401SStefano Zampini   pcbddc->coarse_ksp                 = 0;
1990da1bb401SStefano Zampini   pcbddc->coarse_phi_B               = 0;
1991da1bb401SStefano Zampini   pcbddc->coarse_phi_D               = 0;
199215aaf578SStefano Zampini   pcbddc->coarse_psi_B               = 0;
199315aaf578SStefano Zampini   pcbddc->coarse_psi_D               = 0;
1994da1bb401SStefano Zampini   pcbddc->vec1_P                     = 0;
1995da1bb401SStefano Zampini   pcbddc->vec1_R                     = 0;
1996da1bb401SStefano Zampini   pcbddc->vec2_R                     = 0;
1997da1bb401SStefano Zampini   pcbddc->local_auxmat1              = 0;
1998da1bb401SStefano Zampini   pcbddc->local_auxmat2              = 0;
1999da1bb401SStefano Zampini   pcbddc->R_to_B                     = 0;
2000da1bb401SStefano Zampini   pcbddc->R_to_D                     = 0;
2001da1bb401SStefano Zampini   pcbddc->ksp_D                      = 0;
2002da1bb401SStefano Zampini   pcbddc->ksp_R                      = 0;
2003da1bb401SStefano Zampini   pcbddc->NeumannBoundaries          = 0;
2004785d1243SStefano Zampini   pcbddc->NeumannBoundariesLocal     = 0;
2005785d1243SStefano Zampini   pcbddc->DirichletBoundaries        = 0;
2006785d1243SStefano Zampini   pcbddc->DirichletBoundariesLocal   = 0;
200760d44989SStefano Zampini   pcbddc->user_provided_isfordofs    = PETSC_FALSE;
200860d44989SStefano Zampini   pcbddc->n_ISForDofs                = 0;
200963602bcaSStefano Zampini   pcbddc->n_ISForDofsLocal           = 0;
2010da1bb401SStefano Zampini   pcbddc->ISForDofs                  = 0;
201163602bcaSStefano Zampini   pcbddc->ISForDofsLocal             = 0;
2012da1bb401SStefano Zampini   pcbddc->ConstraintMatrix           = 0;
201385c4d303SStefano Zampini   pcbddc->use_exact_dirichlet_trick  = PETSC_TRUE;
201447d04d0dSStefano Zampini   pcbddc->coarse_loc_to_glob         = 0;
201547d04d0dSStefano Zampini   pcbddc->coarsening_ratio           = 8;
2016b0c7d250SStefano Zampini   pcbddc->coarse_adj_red             = 0;
20174fad6a16SStefano Zampini   pcbddc->current_level              = 0;
20182b510759SStefano Zampini   pcbddc->max_levels                 = 0;
2019323d395dSStefano Zampini   pcbddc->use_coarse_estimates       = PETSC_FALSE;
202074e2c79eSStefano Zampini   pcbddc->redistribute_coarse        = 0;
2021f3bde8b3SStefano Zampini   pcbddc->coarse_subassembling       = 0;
2022323d395dSStefano Zampini   pcbddc->coarse_subassembling_init  = 0;
2023da1bb401SStefano Zampini 
2024674ae819SStefano Zampini   /* create local graph structure */
2025674ae819SStefano Zampini   ierr = PCBDDCGraphCreate(&pcbddc->mat_graph);CHKERRQ(ierr);
2026674ae819SStefano Zampini 
2027674ae819SStefano Zampini   /* scaling */
2028674ae819SStefano Zampini   pcbddc->work_scaling          = 0;
202934a97f8cSStefano Zampini   pcbddc->use_deluxe_scaling    = PETSC_FALSE;
2030ac632422SStefano Zampini   pcbddc->faster_deluxe         = PETSC_FALSE;
2031b96c3477SStefano Zampini 
2032b96c3477SStefano Zampini   /* create sub schurs structure */
2033b96c3477SStefano Zampini   ierr = PCBDDCSubSchursCreate(&pcbddc->sub_schurs);CHKERRQ(ierr);
2034b96c3477SStefano Zampini   pcbddc->sub_schurs_rebuild     = PETSC_FALSE;
2035b96c3477SStefano Zampini   pcbddc->sub_schurs_layers      = -1;
2036b96c3477SStefano Zampini   pcbddc->sub_schurs_use_useradj = PETSC_FALSE;
2037b96c3477SStefano Zampini 
2038b96c3477SStefano Zampini   pcbddc->computed_rowadj = PETSC_FALSE;
2039da1bb401SStefano Zampini 
2040b7eb3628SStefano Zampini   /* adaptivity */
2041f6f667cfSStefano Zampini   pcbddc->adaptive_threshold      = 0.0;
204208122e43SStefano Zampini   pcbddc->adaptive_nmax           = 0;
2043f6f667cfSStefano Zampini   pcbddc->adaptive_nmin           = 0;
2044b7eb3628SStefano Zampini 
2045da1bb401SStefano Zampini   /* function pointers */
2046da1bb401SStefano Zampini   pc->ops->apply               = PCApply_BDDC;
204793bd9ae7SStefano Zampini   pc->ops->applytranspose      = PCApplyTranspose_BDDC;
2048da1bb401SStefano Zampini   pc->ops->setup               = PCSetUp_BDDC;
2049da1bb401SStefano Zampini   pc->ops->destroy             = PCDestroy_BDDC;
2050da1bb401SStefano Zampini   pc->ops->setfromoptions      = PCSetFromOptions_BDDC;
20516b78500eSPatrick Sanan   pc->ops->view                = PCView_BDDC;
2052da1bb401SStefano Zampini   pc->ops->applyrichardson     = 0;
2053da1bb401SStefano Zampini   pc->ops->applysymmetricleft  = 0;
2054da1bb401SStefano Zampini   pc->ops->applysymmetricright = 0;
2055534831adSStefano Zampini   pc->ops->presolve            = PCPreSolve_BDDC;
2056534831adSStefano Zampini   pc->ops->postsolve           = PCPostSolve_BDDC;
2057da1bb401SStefano Zampini 
2058da1bb401SStefano Zampini   /* composing function */
2059906d46d4SStefano Zampini   ierr = PetscObjectComposeFunction((PetscObject)pc,"PCBDDCSetChangeOfBasisMat_C",PCBDDCSetChangeOfBasisMat_BDDC);CHKERRQ(ierr);
2060674ae819SStefano Zampini   ierr = PetscObjectComposeFunction((PetscObject)pc,"PCBDDCSetPrimalVerticesLocalIS_C",PCBDDCSetPrimalVerticesLocalIS_BDDC);CHKERRQ(ierr);
2061bdf89e91SBarry Smith   ierr = PetscObjectComposeFunction((PetscObject)pc,"PCBDDCSetCoarseningRatio_C",PCBDDCSetCoarseningRatio_BDDC);CHKERRQ(ierr);
20622b510759SStefano Zampini   ierr = PetscObjectComposeFunction((PetscObject)pc,"PCBDDCSetLevel_C",PCBDDCSetLevel_BDDC);CHKERRQ(ierr);
2063b8ffe317SStefano Zampini   ierr = PetscObjectComposeFunction((PetscObject)pc,"PCBDDCSetUseExactDirichlet_C",PCBDDCSetUseExactDirichlet_BDDC);CHKERRQ(ierr);
20642b510759SStefano Zampini   ierr = PetscObjectComposeFunction((PetscObject)pc,"PCBDDCSetLevels_C",PCBDDCSetLevels_BDDC);CHKERRQ(ierr);
2065bdf89e91SBarry Smith   ierr = PetscObjectComposeFunction((PetscObject)pc,"PCBDDCSetDirichletBoundaries_C",PCBDDCSetDirichletBoundaries_BDDC);CHKERRQ(ierr);
206682ba6b80SStefano Zampini   ierr = PetscObjectComposeFunction((PetscObject)pc,"PCBDDCSetDirichletBoundariesLocal_C",PCBDDCSetDirichletBoundariesLocal_BDDC);CHKERRQ(ierr);
2067bdf89e91SBarry Smith   ierr = PetscObjectComposeFunction((PetscObject)pc,"PCBDDCSetNeumannBoundaries_C",PCBDDCSetNeumannBoundaries_BDDC);CHKERRQ(ierr);
206882ba6b80SStefano Zampini   ierr = PetscObjectComposeFunction((PetscObject)pc,"PCBDDCSetNeumannBoundariesLocal_C",PCBDDCSetNeumannBoundariesLocal_BDDC);CHKERRQ(ierr);
2069bdf89e91SBarry Smith   ierr = PetscObjectComposeFunction((PetscObject)pc,"PCBDDCGetDirichletBoundaries_C",PCBDDCGetDirichletBoundaries_BDDC);CHKERRQ(ierr);
207082ba6b80SStefano Zampini   ierr = PetscObjectComposeFunction((PetscObject)pc,"PCBDDCGetDirichletBoundariesLocal_C",PCBDDCGetDirichletBoundariesLocal_BDDC);CHKERRQ(ierr);
2071bdf89e91SBarry Smith   ierr = PetscObjectComposeFunction((PetscObject)pc,"PCBDDCGetNeumannBoundaries_C",PCBDDCGetNeumannBoundaries_BDDC);CHKERRQ(ierr);
207282ba6b80SStefano Zampini   ierr = PetscObjectComposeFunction((PetscObject)pc,"PCBDDCGetNeumannBoundariesLocal_C",PCBDDCGetNeumannBoundariesLocal_BDDC);CHKERRQ(ierr);
2073bdf89e91SBarry Smith   ierr = PetscObjectComposeFunction((PetscObject)pc,"PCBDDCSetDofsSplitting_C",PCBDDCSetDofsSplitting_BDDC);CHKERRQ(ierr);
207463602bcaSStefano Zampini   ierr = PetscObjectComposeFunction((PetscObject)pc,"PCBDDCSetDofsSplittingLocal_C",PCBDDCSetDofsSplittingLocal_BDDC);CHKERRQ(ierr);
2075bdf89e91SBarry Smith   ierr = PetscObjectComposeFunction((PetscObject)pc,"PCBDDCSetLocalAdjacencyGraph_C",PCBDDCSetLocalAdjacencyGraph_BDDC);CHKERRQ(ierr);
2076bdf89e91SBarry Smith   ierr = PetscObjectComposeFunction((PetscObject)pc,"PCBDDCCreateFETIDPOperators_C",PCBDDCCreateFETIDPOperators_BDDC);CHKERRQ(ierr);
2077bdf89e91SBarry Smith   ierr = PetscObjectComposeFunction((PetscObject)pc,"PCBDDCMatFETIDPGetRHS_C",PCBDDCMatFETIDPGetRHS_BDDC);CHKERRQ(ierr);
2078bdf89e91SBarry Smith   ierr = PetscObjectComposeFunction((PetscObject)pc,"PCBDDCMatFETIDPGetSolution_C",PCBDDCMatFETIDPGetSolution_BDDC);CHKERRQ(ierr);
2079a06fd7f2SStefano Zampini   ierr = PetscObjectComposeFunction((PetscObject)pc,"PCPreSolveChangeRHS_C",PCPreSolveChangeRHS_BDDC);CHKERRQ(ierr);
2080da1bb401SStefano Zampini   PetscFunctionReturn(0);
2081da1bb401SStefano Zampini }
20823425bc38SStefano Zampini 
2083