xref: /petsc/src/ksp/pc/impls/bddc/bddc.c (revision aefa1729f5398e0ca5a29e96912d0cf2a1d42d9d)
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 
26f3d41395Sstefano_zampini static PetscBool  cited = PETSC_FALSE;
27f3d41395Sstefano_zampini static const char citation[] =
28f3d41395Sstefano_zampini "@article{ZampiniPCBDDC,\n"
29f3d41395Sstefano_zampini "author = {Stefano Zampini},\n"
30f3d41395Sstefano_zampini "title = {{PCBDDC}: A Class of Robust Dual-Primal Methods in {PETS}c},\n"
31f3d41395Sstefano_zampini "journal = {SIAM Journal on Scientific Computing},\n"
32f3d41395Sstefano_zampini "volume = {38},\n"
33f3d41395Sstefano_zampini "number = {5},\n"
34f3d41395Sstefano_zampini "pages = {S282-S306},\n"
35f3d41395Sstefano_zampini "year = {2016},\n"
36f3d41395Sstefano_zampini "doi = {10.1137/15M1025785},\n"
37f3d41395Sstefano_zampini "URL = {http://dx.doi.org/10.1137/15M1025785},\n"
38f3d41395Sstefano_zampini "eprint = {http://dx.doi.org/10.1137/15M1025785}\n"
39f3d41395Sstefano_zampini "}\n";
40f3d41395Sstefano_zampini 
410369aaf7SStefano Zampini /* temporarily declare it */
420369aaf7SStefano Zampini PetscErrorCode PCApply_BDDC(PC,Vec,Vec);
430369aaf7SStefano Zampini 
444416b707SBarry Smith PetscErrorCode PCSetFromOptions_BDDC(PetscOptionItems *PetscOptionsObject,PC pc)
450c7d97c5SJed Brown {
460c7d97c5SJed Brown   PC_BDDC        *pcbddc = (PC_BDDC*)pc->data;
470c7d97c5SJed Brown   PetscErrorCode ierr;
480c7d97c5SJed Brown 
490c7d97c5SJed Brown   PetscFunctionBegin;
50e55864a3SBarry Smith   ierr = PetscOptionsHead(PetscOptionsObject,"BDDC options");CHKERRQ(ierr);
518eeda7d8SStefano Zampini   /* Verbose debugging */
52a13144ffSStefano Zampini   ierr = PetscOptionsInt("-pc_bddc_check_level","Verbose output for PCBDDC (intended for debug)","none",pcbddc->dbg_flag,&pcbddc->dbg_flag,NULL);CHKERRQ(ierr);
53a13144ffSStefano Zampini   /* Approximate solvers */
54c7017625SStefano 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);
55c7017625SStefano 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);
56c7017625SStefano 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);
57c7017625SStefano 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);
586b78500eSPatrick Sanan   /* Primal space customization */
5908a5cf49SStefano 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);
60be12c134Sstefano_zampini   ierr = PetscOptionsInt("-pc_bddc_graph_maxcount","Maximum number of shared subdomains for a connected component","none",pcbddc->graphmaxcount,&pcbddc->graphmaxcount,NULL);CHKERRQ(ierr);
618eeda7d8SStefano 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);
628eeda7d8SStefano 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);
638eeda7d8SStefano 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);
6414f95afaSStefano Zampini   ierr = PetscOptionsInt("-pc_bddc_vertex_size","Connected components smaller or equal to vertex size will be considered as primal vertices","none",pcbddc->vertex_size,&pcbddc->vertex_size,NULL);CHKERRQ(ierr);
656661aa1dSStefano 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);
6614f95afaSStefano Zampini   ierr = PetscOptionsBool("-pc_bddc_use_qr_single","Use QR factorization for single constraints on cc (QR is always used when multiple constraints are present)","none",pcbddc->use_qr_single,&pcbddc->use_qr_single,NULL);CHKERRQ(ierr);
678eeda7d8SStefano Zampini   /* Change of basis */
68b9b85e73SStefano 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);
69b9b85e73SStefano 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);
70674ae819SStefano Zampini   if (!pcbddc->use_change_of_basis) {
71674ae819SStefano Zampini     pcbddc->use_change_on_faces = PETSC_FALSE;
72674ae819SStefano Zampini   }
738eeda7d8SStefano Zampini   /* Switch between M_2 (default) and M_3 preconditioners (as defined by C. Dohrmann in the ref. article) */
748eeda7d8SStefano 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);
7557de7509SStefano Zampini   ierr = PetscOptionsInt("-pc_bddc_coarse_eqs_per_proc","Number of equations per process for coarse problem redistribution (significant only at the coarsest level)","none",pcbddc->coarse_eqs_per_proc,&pcbddc->coarse_eqs_per_proc,NULL);CHKERRQ(ierr);
760298fd71SBarry Smith   ierr = PetscOptionsInt("-pc_bddc_coarsening_ratio","Set coarsening ratio used in multilevel coarsening","none",pcbddc->coarsening_ratio,&pcbddc->coarsening_ratio,NULL);CHKERRQ(ierr);
772b510759SStefano Zampini   ierr = PetscOptionsInt("-pc_bddc_levels","Set maximum number of levels for multilevel","none",pcbddc->max_levels,&pcbddc->max_levels,NULL);CHKERRQ(ierr);
78323d395dSStefano 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);
79674ae819SStefano 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);
80b96c3477SStefano 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);
81b96c3477SStefano 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);
82b96c3477SStefano 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);
83683d3df6SStefano Zampini   ierr = PetscOptionsBool("-pc_bddc_schur_exact","Whether or not to use the exact Schur complement instead of the reduced one (which excludes size 1 cc)","none",pcbddc->sub_schurs_exact_schur,&pcbddc->sub_schurs_exact_schur,NULL);CHKERRQ(ierr);
84bf3a8328SStefano Zampini   ierr = PetscOptionsBool("-pc_bddc_deluxe_zerorows","Zero rows and columns of deluxe operators associated with primal dofs","none",pcbddc->deluxe_zerorows,&pcbddc->deluxe_zerorows,NULL);CHKERRQ(ierr);
85839e9adbSstefano_zampini   ierr = PetscOptionsBool("-pc_bddc_deluxe_singlemat","Collapse deluxe operators","none",pcbddc->deluxe_singlemat,&pcbddc->deluxe_singlemat,NULL);CHKERRQ(ierr);
86bf3a8328SStefano Zampini   ierr = PetscOptionsBool("-pc_bddc_adaptive_userdefined","Use user-defined constraints (should be attached via MatSetNearNullSpace to pmat) in addition to those adaptively generated","none",pcbddc->adaptive_userdefined,&pcbddc->adaptive_userdefined,NULL);CHKERRQ(ierr);
874c6709b3SStefano 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);
8808122e43SStefano Zampini   ierr = PetscOptionsInt("-pc_bddc_adaptive_nmin","Minimum number of constraints per connected components","none",pcbddc->adaptive_nmin,&pcbddc->adaptive_nmin,NULL);CHKERRQ(ierr);
8908122e43SStefano Zampini   ierr = PetscOptionsInt("-pc_bddc_adaptive_nmax","Maximum number of constraints per connected components","none",pcbddc->adaptive_nmax,&pcbddc->adaptive_nmax,NULL);CHKERRQ(ierr);
903301b35fSStefano Zampini   ierr = PetscOptionsBool("-pc_bddc_symmetric","Symmetric computation of primal basis functions","none",pcbddc->symmetric_primal,&pcbddc->symmetric_primal,NULL);CHKERRQ(ierr);
91b0c7d250SStefano 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);
92b3afcdbeSStefano Zampini   ierr = PetscOptionsBool("-pc_bddc_benign_trick","Apply the benign subspace trick to saddle point problems with discontinuous pressures","none",pcbddc->benign_saddle_point,&pcbddc->benign_saddle_point,NULL);CHKERRQ(ierr);
93e9627c49SStefano Zampini   ierr = PetscOptionsBool("-pc_bddc_benign_change","Compute the pressure change of basis explicitly","none",pcbddc->benign_change_explicit,&pcbddc->benign_change_explicit,NULL);CHKERRQ(ierr);
9427b6a85dSStefano Zampini   ierr = PetscOptionsBool("-pc_bddc_benign_compute_correction","Compute the benign correction during PreSolve","none",pcbddc->benign_compute_correction,&pcbddc->benign_compute_correction,NULL);CHKERRQ(ierr);
95a198735bSStefano Zampini   ierr = PetscOptionsBool("-pc_bddc_nonetflux","Automatic computation of no-net-flux quadrature weights","none",pcbddc->compute_nonetflux,&pcbddc->compute_nonetflux,NULL);CHKERRQ(ierr);
964f1b2e48SStefano Zampini   ierr = PetscOptionsBool("-pc_bddc_detect_disconnected","Detects disconnected subdomains","none",pcbddc->detect_disconnected,&pcbddc->detect_disconnected,NULL);CHKERRQ(ierr);
9770c64980SStefano Zampini   ierr = PetscOptionsBool("-pc_bddc_eliminate_dirichlet","Whether or not we want to eliminate dirichlet dofs during presolve","none",pcbddc->eliminate_dirdofs,&pcbddc->eliminate_dirdofs,NULL);CHKERRQ(ierr);
980c7d97c5SJed Brown   ierr = PetscOptionsTail();CHKERRQ(ierr);
990c7d97c5SJed Brown   PetscFunctionReturn(0);
1000c7d97c5SJed Brown }
1016b78500eSPatrick Sanan 
1026b78500eSPatrick Sanan static PetscErrorCode PCView_BDDC(PC pc,PetscViewer viewer)
1036b78500eSPatrick Sanan {
1046b78500eSPatrick Sanan   PC_BDDC              *pcbddc = (PC_BDDC*)pc->data;
105e9627c49SStefano Zampini   PC_IS                *pcis = (PC_IS*)pc->data;
1066b78500eSPatrick Sanan   PetscErrorCode       ierr;
1076b78500eSPatrick Sanan   PetscBool            isascii,isstring;
108e9627c49SStefano Zampini   PetscSubcomm         subcomm;
109e9627c49SStefano Zampini   PetscViewer          subviewer;
1106b78500eSPatrick Sanan 
1116b78500eSPatrick Sanan   PetscFunctionBegin;
1126b78500eSPatrick Sanan   ierr = PetscObjectTypeCompare((PetscObject)viewer,PETSCVIEWERASCII,&isascii);CHKERRQ(ierr);
1136b78500eSPatrick Sanan   ierr = PetscObjectTypeCompare((PetscObject)viewer,PETSCVIEWERSTRING,&isstring);CHKERRQ(ierr);
1146b78500eSPatrick Sanan   /* Nothing printed for the String viewer */
1156b78500eSPatrick Sanan   /* ASCII viewer */
1166b78500eSPatrick Sanan   if (isascii) {
1174b2aedd3SStefano Zampini     PetscMPIInt   color,rank,size;
118fbad9177SStefano Zampini     PetscInt64    loc[7],gsum[6],gmax[6],gmin[6],totbenign;
119e9627c49SStefano Zampini     PetscScalar   interface_size;
120e9627c49SStefano Zampini     PetscReal     ratio1=0.,ratio2=0.;
121e9627c49SStefano Zampini     Vec           counter;
1226b78500eSPatrick Sanan 
123b74ba07aSstefano_zampini     if (!pc->setupcalled) {
124b74ba07aSstefano_zampini       ierr = PetscViewerASCIIPrintf(viewer,"  Partial information available: preconditioner has not been setup yet\n");CHKERRQ(ierr);
125b74ba07aSstefano_zampini     }
126efd4aadfSBarry Smith     ierr = PetscViewerASCIIPrintf(viewer,"  Use verbose output: %D\n",pcbddc->dbg_flag);CHKERRQ(ierr);
127efd4aadfSBarry Smith     ierr = PetscViewerASCIIPrintf(viewer,"  Use user-defined CSR: %D\n",!!pcbddc->mat_graph->nvtxs_csr);CHKERRQ(ierr);
128efd4aadfSBarry Smith     ierr = PetscViewerASCIIPrintf(viewer,"  Use local mat graph: %D\n",pcbddc->use_local_adj && !pcbddc->mat_graph->nvtxs_csr);CHKERRQ(ierr);
129e9627c49SStefano Zampini     if (pcbddc->mat_graph->twodim) {
130efd4aadfSBarry Smith       ierr = PetscViewerASCIIPrintf(viewer,"  Connectivity graph topological dimension: 2\n");CHKERRQ(ierr);
131e9627c49SStefano Zampini     } else {
132efd4aadfSBarry Smith       ierr = PetscViewerASCIIPrintf(viewer,"  Connectivity graph topological dimension: 3\n");CHKERRQ(ierr);
133e9627c49SStefano Zampini     }
134*aefa1729SStefano Zampini     if (pcbddc->graphmaxcount != PETSC_MAX_INT) {
135efd4aadfSBarry Smith       ierr = PetscViewerASCIIPrintf(viewer,"  Graph max count: %D\n",pcbddc->graphmaxcount);CHKERRQ(ierr);
136*aefa1729SStefano Zampini     }
137efd4aadfSBarry Smith     ierr = PetscViewerASCIIPrintf(viewer,"  Use vertices: %d (vertex size %d)\n",pcbddc->use_vertices,pcbddc->vertex_size);CHKERRQ(ierr);
138efd4aadfSBarry Smith     ierr = PetscViewerASCIIPrintf(viewer,"  Use edges: %d\n",pcbddc->use_edges);CHKERRQ(ierr);
139efd4aadfSBarry Smith     ierr = PetscViewerASCIIPrintf(viewer,"  Use faces: %d\n",pcbddc->use_faces);CHKERRQ(ierr);
140efd4aadfSBarry Smith     ierr = PetscViewerASCIIPrintf(viewer,"  Use true near null space: %d\n",pcbddc->use_nnsp_true);CHKERRQ(ierr);
141efd4aadfSBarry Smith     ierr = PetscViewerASCIIPrintf(viewer,"  Use QR for single constraints on cc: %d\n",pcbddc->use_qr_single);CHKERRQ(ierr);
142efd4aadfSBarry Smith     ierr = PetscViewerASCIIPrintf(viewer,"  Use change of basis on local edge nodes: %d\n",pcbddc->use_change_of_basis);CHKERRQ(ierr);
143efd4aadfSBarry Smith     ierr = PetscViewerASCIIPrintf(viewer,"  Use change of basis on local face nodes: %d\n",pcbddc->use_change_on_faces);CHKERRQ(ierr);
144efd4aadfSBarry Smith     ierr = PetscViewerASCIIPrintf(viewer,"  User defined change of basis matrix: %d\n",!!pcbddc->user_ChangeOfBasisMatrix);CHKERRQ(ierr);
145efd4aadfSBarry Smith     ierr = PetscViewerASCIIPrintf(viewer,"  Has change of basis matrix: %d\n",!!pcbddc->ChangeOfBasisMatrix);CHKERRQ(ierr);
146efd4aadfSBarry Smith     ierr = PetscViewerASCIIPrintf(viewer,"  Eliminate dirichlet boundary dofs: %d\n",pcbddc->eliminate_dirdofs);CHKERRQ(ierr);
147efd4aadfSBarry Smith     ierr = PetscViewerASCIIPrintf(viewer,"  Switch on static condensation ops around the interface preconditioner: %d\n",pcbddc->switch_static);CHKERRQ(ierr);
148efd4aadfSBarry Smith     ierr = PetscViewerASCIIPrintf(viewer,"  Use exact dirichlet trick: %d\n",pcbddc->use_exact_dirichlet_trick);CHKERRQ(ierr);
149efd4aadfSBarry Smith     ierr = PetscViewerASCIIPrintf(viewer,"  Multilevel max levels: %d\n",pcbddc->max_levels);CHKERRQ(ierr);
150efd4aadfSBarry Smith     ierr = PetscViewerASCIIPrintf(viewer,"  Multilevel coarsening ratio: %d\n",pcbddc->coarsening_ratio);CHKERRQ(ierr);
151efd4aadfSBarry Smith     ierr = PetscViewerASCIIPrintf(viewer,"  Use estimated eigs for coarse problem: %d\n",pcbddc->use_coarse_estimates);CHKERRQ(ierr);
152efd4aadfSBarry Smith     ierr = PetscViewerASCIIPrintf(viewer,"  Use deluxe scaling: %d\n",pcbddc->use_deluxe_scaling);CHKERRQ(ierr);
153efd4aadfSBarry Smith     ierr = PetscViewerASCIIPrintf(viewer,"  Use deluxe zerorows: %d\n",pcbddc->deluxe_zerorows);CHKERRQ(ierr);
154efd4aadfSBarry Smith     ierr = PetscViewerASCIIPrintf(viewer,"  Use deluxe singlemat: %d\n",pcbddc->deluxe_singlemat);CHKERRQ(ierr);
155efd4aadfSBarry Smith     ierr = PetscViewerASCIIPrintf(viewer,"  Rebuild interface graph for Schur principal minors: %d\n",pcbddc->sub_schurs_rebuild);CHKERRQ(ierr);
156efd4aadfSBarry Smith     ierr = PetscViewerASCIIPrintf(viewer,"  Number of dofs' layers for the computation of principal minors: %d\n",pcbddc->sub_schurs_layers);CHKERRQ(ierr);
157efd4aadfSBarry Smith     ierr = PetscViewerASCIIPrintf(viewer,"  Use user CSR graph to compute successive layers: %d\n",pcbddc->sub_schurs_use_useradj);CHKERRQ(ierr);
158efd4aadfSBarry Smith     ierr = PetscViewerASCIIPrintf(viewer,"  Adaptive constraint selection threshold (active %d, userdefined %d): %g\n",pcbddc->adaptive_threshold,pcbddc->adaptive_selection,pcbddc->adaptive_userdefined);CHKERRQ(ierr);
159efd4aadfSBarry Smith     ierr = PetscViewerASCIIPrintf(viewer,"  Min constraints / connected component: %d\n",pcbddc->adaptive_nmin);CHKERRQ(ierr);
160efd4aadfSBarry Smith     ierr = PetscViewerASCIIPrintf(viewer,"  Max constraints / connected component: %d\n",pcbddc->adaptive_nmax);CHKERRQ(ierr);
161efd4aadfSBarry Smith     ierr = PetscViewerASCIIPrintf(viewer,"  Invert exact Schur complement for adaptive selection: %d\n",pcbddc->sub_schurs_exact_schur);CHKERRQ(ierr);
162efd4aadfSBarry Smith     ierr = PetscViewerASCIIPrintf(viewer,"  Symmetric computation of primal basis functions: %d\n",pcbddc->symmetric_primal);CHKERRQ(ierr);
163efd4aadfSBarry Smith     ierr = PetscViewerASCIIPrintf(viewer,"  Num. Procs. to map coarse adjacency list: %d\n",pcbddc->coarse_adj_red);CHKERRQ(ierr);
164efd4aadfSBarry Smith     ierr = PetscViewerASCIIPrintf(viewer,"  Coarse eqs per proc (significant at the coarsest level): %d\n",pcbddc->coarse_eqs_per_proc);CHKERRQ(ierr);
165efd4aadfSBarry Smith     ierr = PetscViewerASCIIPrintf(viewer,"  Detect disconnected: %d\n",pcbddc->detect_disconnected);CHKERRQ(ierr);
166efd4aadfSBarry Smith     ierr = PetscViewerASCIIPrintf(viewer,"  Benign subspace trick: %d (change explicit %d)\n",pcbddc->benign_saddle_point,pcbddc->benign_change_explicit);CHKERRQ(ierr);
167efd4aadfSBarry Smith     ierr = PetscViewerASCIIPrintf(viewer,"  Benign subspace trick is active: %d\n",pcbddc->benign_have_null);CHKERRQ(ierr);
168efd4aadfSBarry Smith     ierr = PetscViewerASCIIPrintf(viewer,"  Algebraic computation of no-net-flux %d\n",pcbddc->compute_nonetflux);CHKERRQ(ierr);
169b74ba07aSstefano_zampini     if (!pc->setupcalled) PetscFunctionReturn(0);
1706b78500eSPatrick Sanan 
171fbad9177SStefano Zampini     /* compute interface size */
172e9627c49SStefano Zampini     ierr = VecSet(pcis->vec1_B,1.0);CHKERRQ(ierr);
173e9627c49SStefano Zampini     ierr = MatCreateVecs(pc->pmat,&counter,0);CHKERRQ(ierr);
174e9627c49SStefano Zampini     ierr = VecSet(counter,0.0);CHKERRQ(ierr);
175e9627c49SStefano Zampini     ierr = VecScatterBegin(pcis->global_to_B,pcis->vec1_B,counter,INSERT_VALUES,SCATTER_REVERSE);CHKERRQ(ierr);
176e9627c49SStefano Zampini     ierr = VecScatterEnd(pcis->global_to_B,pcis->vec1_B,counter,INSERT_VALUES,SCATTER_REVERSE);CHKERRQ(ierr);
177e9627c49SStefano Zampini     ierr = VecSum(counter,&interface_size);CHKERRQ(ierr);
178e9627c49SStefano Zampini     ierr = VecDestroy(&counter);CHKERRQ(ierr);
179fbad9177SStefano Zampini 
180fbad9177SStefano Zampini     /* compute some statistics on the domain decomposition */
181e9627c49SStefano Zampini     gsum[0] = 1;
182fbad9177SStefano Zampini     gsum[1] = gsum[2] = gsum[3] = gsum[4] = gsum[5] = 0;
183e9627c49SStefano Zampini     loc[0]  = !!pcis->n;
184e9627c49SStefano Zampini     loc[1]  = pcis->n - pcis->n_B;
185e9627c49SStefano Zampini     loc[2]  = pcis->n_B;
186e9627c49SStefano Zampini     loc[3]  = pcbddc->local_primal_size;
187345ecf6cSStefano Zampini     loc[4]  = pcis->n;
188fbad9177SStefano Zampini     loc[5]  = pcbddc->n_local_subs > 0 ? pcbddc->n_local_subs : (pcis->n ? 1 : 0);
189fbad9177SStefano Zampini     loc[6]  = pcbddc->benign_n;
190fbad9177SStefano Zampini     ierr = MPI_Reduce(loc,gsum,6,MPIU_INT64,MPI_SUM,0,PetscObjectComm((PetscObject)pc));CHKERRQ(ierr);
191fbad9177SStefano Zampini     if (!loc[0]) loc[1] = loc[2] = loc[3] = loc[4] = loc[5] = -1;
192fbad9177SStefano Zampini     ierr = MPI_Reduce(loc,gmax,6,MPIU_INT64,MPI_MAX,0,PetscObjectComm((PetscObject)pc));CHKERRQ(ierr);
193fbad9177SStefano Zampini     if (!loc[0]) loc[1] = loc[2] = loc[3] = loc[4] = loc[5] = PETSC_MAX_INT;
194fbad9177SStefano Zampini     ierr = MPI_Reduce(loc,gmin,6,MPIU_INT64,MPI_MIN,0,PetscObjectComm((PetscObject)pc));CHKERRQ(ierr);
195fbad9177SStefano Zampini     ierr = MPI_Reduce(&loc[6],&totbenign,1,MPIU_INT64,MPI_SUM,0,PetscObjectComm((PetscObject)pc));CHKERRQ(ierr);
196e9627c49SStefano Zampini     if (pcbddc->coarse_size) {
197e9627c49SStefano Zampini       ratio1 = pc->pmat->rmap->N/(1.*pcbddc->coarse_size);
198e9627c49SStefano Zampini       ratio2 = PetscRealPart(interface_size)/pcbddc->coarse_size;
199e9627c49SStefano Zampini     }
200efd4aadfSBarry Smith     ierr = PetscViewerASCIIPrintf(viewer,"  ********************************** STATISTICS AT LEVEL %d **********************************\n",pcbddc->current_level);CHKERRQ(ierr);
201efd4aadfSBarry Smith     ierr = PetscViewerASCIIPrintf(viewer,"  Global dofs sizes: all %d interface %d coarse %d\n",pc->pmat->rmap->N,(PetscInt)PetscRealPart(interface_size),pcbddc->coarse_size);CHKERRQ(ierr);
202efd4aadfSBarry Smith     ierr = PetscViewerASCIIPrintf(viewer,"  Coarsening ratios: all/coarse %d interface/coarse %d\n",(PetscInt)ratio1,(PetscInt)ratio2);CHKERRQ(ierr);
203efd4aadfSBarry Smith     ierr = PetscViewerASCIIPrintf(viewer,"  Active processes : %d\n",(PetscInt)gsum[0]);CHKERRQ(ierr);
204efd4aadfSBarry Smith     ierr = PetscViewerASCIIPrintf(viewer,"  Total subdomains : %d\n",(PetscInt)gsum[5]);CHKERRQ(ierr);
205345ecf6cSStefano Zampini     if (pcbddc->benign_have_null) {
206efd4aadfSBarry Smith       ierr = PetscViewerASCIIPrintf(viewer,"  Benign subs      : %d\n",(PetscInt)totbenign);CHKERRQ(ierr);
207345ecf6cSStefano Zampini     }
208efd4aadfSBarry Smith     ierr = PetscViewerASCIIPrintf(viewer,"  Dofs type        :\tMIN\tMAX\tMEAN\n",(PetscInt)gmin[1],(PetscInt)gmax[1],(PetscInt)(gsum[1]/gsum[0]));CHKERRQ(ierr);
209efd4aadfSBarry Smith     ierr = PetscViewerASCIIPrintf(viewer,"  Interior  dofs   :\t%d\t%d\t%d\n",(PetscInt)gmin[1],(PetscInt)gmax[1],(PetscInt)(gsum[1]/gsum[0]));CHKERRQ(ierr);
210efd4aadfSBarry Smith     ierr = PetscViewerASCIIPrintf(viewer,"  Interface dofs   :\t%d\t%d\t%d\n",(PetscInt)gmin[2],(PetscInt)gmax[2],(PetscInt)(gsum[2]/gsum[0]));CHKERRQ(ierr);
211efd4aadfSBarry Smith     ierr = PetscViewerASCIIPrintf(viewer,"  Primal    dofs   :\t%d\t%d\t%d\n",(PetscInt)gmin[3],(PetscInt)gmax[3],(PetscInt)(gsum[3]/gsum[0]));CHKERRQ(ierr);
212efd4aadfSBarry Smith     ierr = PetscViewerASCIIPrintf(viewer,"  Local     dofs   :\t%d\t%d\t%d\n",(PetscInt)gmin[4],(PetscInt)gmax[4],(PetscInt)(gsum[4]/gsum[0]));CHKERRQ(ierr);
213efd4aadfSBarry Smith     ierr = PetscViewerASCIIPrintf(viewer,"  Local     subs   :\t%d\t%d\n",(PetscInt)gmin[5],(PetscInt)gmax[5]);CHKERRQ(ierr);
214efd4aadfSBarry Smith     ierr = PetscViewerASCIIPrintf(viewer,"  ********************************** COARSE PROBLEM DETAILS *********************************\n",pcbddc->current_level);CHKERRQ(ierr);
21527b6a85dSStefano Zampini     ierr = PetscViewerFlush(viewer);CHKERRQ(ierr);
216e9627c49SStefano Zampini 
217fbad9177SStefano Zampini     /* the coarse problem can be handled by a different communicator */
218e9627c49SStefano Zampini     if (pcbddc->coarse_ksp) color = 1;
219e9627c49SStefano Zampini     else color = 0;
220e9627c49SStefano Zampini     ierr = MPI_Comm_rank(PetscObjectComm((PetscObject)pc),&rank);CHKERRQ(ierr);
2214b2aedd3SStefano Zampini     ierr = MPI_Comm_size(PetscObjectComm((PetscObject)pc),&size);CHKERRQ(ierr);
222e9627c49SStefano Zampini     ierr = PetscSubcommCreate(PetscObjectComm((PetscObject)pc),&subcomm);CHKERRQ(ierr);
2234b2aedd3SStefano Zampini     ierr = PetscSubcommSetNumber(subcomm,PetscMin(size,2));CHKERRQ(ierr);
224e9627c49SStefano Zampini     ierr = PetscSubcommSetTypeGeneral(subcomm,color,rank);CHKERRQ(ierr);
225e9627c49SStefano Zampini     ierr = PetscViewerGetSubViewer(viewer,PetscSubcommChild(subcomm),&subviewer);CHKERRQ(ierr);
226e9627c49SStefano Zampini     if (color == 1) {
227e9627c49SStefano Zampini       ierr = KSPView(pcbddc->coarse_ksp,subviewer);CHKERRQ(ierr);
228e9627c49SStefano Zampini       ierr = PetscViewerFlush(subviewer);CHKERRQ(ierr);
229e9627c49SStefano Zampini     }
230e9627c49SStefano Zampini     ierr = PetscViewerRestoreSubViewer(viewer,PetscSubcommChild(subcomm),&subviewer);CHKERRQ(ierr);
231e9627c49SStefano Zampini     ierr = PetscSubcommDestroy(&subcomm);CHKERRQ(ierr);
232e9627c49SStefano Zampini     ierr = PetscViewerFlush(viewer);CHKERRQ(ierr);
233e9627c49SStefano Zampini   }
2346b78500eSPatrick Sanan   PetscFunctionReturn(0);
2356b78500eSPatrick Sanan }
236a13144ffSStefano Zampini 
2371e0482f5SStefano Zampini static PetscErrorCode PCBDDCSetDiscreteGradient_BDDC(PC pc, Mat G, PetscInt order, PetscInt field, PetscBool global, PetscBool conforming)
238a13144ffSStefano Zampini {
239a13144ffSStefano Zampini   PC_BDDC        *pcbddc = (PC_BDDC*)pc->data;
240a13144ffSStefano Zampini   PetscErrorCode ierr;
241a13144ffSStefano Zampini 
242a13144ffSStefano Zampini   PetscFunctionBegin;
243a13144ffSStefano Zampini   ierr = PetscObjectReference((PetscObject)G);CHKERRQ(ierr);
244a13144ffSStefano Zampini   ierr = MatDestroy(&pcbddc->discretegradient);CHKERRQ(ierr);
245a13144ffSStefano Zampini   pcbddc->discretegradient = G;
246a13144ffSStefano Zampini   pcbddc->nedorder         = order > 0 ? order : -order;
247495a2a07SStefano Zampini   pcbddc->nedfield         = field;
2481e0482f5SStefano Zampini   pcbddc->nedglobal        = global;
2491e0482f5SStefano Zampini   pcbddc->conforming       = conforming;
250a13144ffSStefano Zampini   PetscFunctionReturn(0);
251a13144ffSStefano Zampini }
252a13144ffSStefano Zampini 
253a13144ffSStefano Zampini /*@
254a13144ffSStefano Zampini  PCBDDCSetDiscreteGradient - Sets the discrete gradient
255a13144ffSStefano Zampini 
256a13144ffSStefano Zampini    Collective on PC
257a13144ffSStefano Zampini 
258a13144ffSStefano Zampini    Input Parameters:
259a13144ffSStefano Zampini +  pc         - the preconditioning context
260a13144ffSStefano Zampini .  G          - the discrete gradient matrix (should be in AIJ format)
261a13144ffSStefano Zampini .  order      - the order of the Nedelec space (1 for the lowest order)
262495a2a07SStefano Zampini .  field      - the field id of the Nedelec dofs (not used if the fields have not been specified)
2631e0482f5SStefano Zampini .  global     - the type of global ordering for the rows of G
264a13144ffSStefano Zampini -  conforming - whether the mesh is conforming or not
265a13144ffSStefano Zampini 
266a13144ffSStefano Zampini    Level: advanced
267a13144ffSStefano Zampini 
268a13144ffSStefano Zampini    Notes: The discrete gradient matrix G is used to analyze the subdomain edges, and it should not contain any zero entry.
269495a2a07SStefano Zampini           For variable order spaces, the order should be set to zero.
2701e0482f5SStefano Zampini           If global is true, the rows of G should be given in global ordering for the whole dofs;
2711e0482f5SStefano Zampini           if false, the ordering should be global for the Nedelec field.
2721e0482f5SStefano Zampini           In the latter case, it should hold gid[i] < gid[j] iff geid[i] < geid[j], with gid the global orderding for all the dofs
2731e0482f5SStefano Zampini           and geid the one for the Nedelec field.
274a13144ffSStefano Zampini 
275495a2a07SStefano Zampini .seealso: PCBDDC,PCBDDCSetDofsSplitting(),PCBDDCSetDofsSplittingLocal()
276a13144ffSStefano Zampini @*/
2771e0482f5SStefano Zampini PetscErrorCode PCBDDCSetDiscreteGradient(PC pc, Mat G, PetscInt order, PetscInt field, PetscBool global, PetscBool conforming)
278a13144ffSStefano Zampini {
279a13144ffSStefano Zampini   PetscErrorCode ierr;
280a13144ffSStefano Zampini 
281a13144ffSStefano Zampini   PetscFunctionBegin;
282a13144ffSStefano Zampini   PetscValidHeaderSpecific(pc,PC_CLASSID,1);
283a13144ffSStefano Zampini   PetscValidHeaderSpecific(G,MAT_CLASSID,2);
284a13144ffSStefano Zampini   PetscValidLogicalCollectiveInt(pc,order,3);
2851e0482f5SStefano Zampini   PetscValidLogicalCollectiveInt(pc,field,4);
2861e0482f5SStefano Zampini   PetscValidLogicalCollectiveBool(pc,global,5);
2871e0482f5SStefano Zampini   PetscValidLogicalCollectiveBool(pc,conforming,6);
2881e0482f5SStefano Zampini   PetscCheckSameComm(pc,1,G,2);
2891e0482f5SStefano Zampini   ierr = PetscTryMethod(pc,"PCBDDCSetDiscreteGradient_C",(PC,Mat,PetscInt,PetscInt,PetscBool,PetscBool),(pc,G,order,field,global,conforming));CHKERRQ(ierr);
290a13144ffSStefano Zampini   PetscFunctionReturn(0);
291a13144ffSStefano Zampini }
292a13144ffSStefano Zampini 
2938ae0ca82SStefano Zampini static PetscErrorCode PCBDDCSetDivergenceMat_BDDC(PC pc, Mat divudotp, PetscBool trans, IS vl2l)
294a198735bSStefano Zampini {
295a198735bSStefano Zampini   PC_BDDC        *pcbddc = (PC_BDDC*)pc->data;
296a198735bSStefano Zampini   PetscErrorCode ierr;
2976b78500eSPatrick Sanan 
298a198735bSStefano Zampini   PetscFunctionBegin;
299a198735bSStefano Zampini   ierr = PetscObjectReference((PetscObject)divudotp);CHKERRQ(ierr);
300a198735bSStefano Zampini   ierr = MatDestroy(&pcbddc->divudotp);CHKERRQ(ierr);
301a198735bSStefano Zampini   pcbddc->divudotp = divudotp;
3028ae0ca82SStefano Zampini   pcbddc->divudotp_trans = trans;
303a198735bSStefano Zampini   pcbddc->compute_nonetflux = PETSC_TRUE;
304a198735bSStefano Zampini   if (vl2l) {
305a198735bSStefano Zampini     ierr = PetscObjectReference((PetscObject)vl2l);CHKERRQ(ierr);
306fa23a32eSStefano Zampini     ierr = ISDestroy(&pcbddc->divudotp_vl2l);CHKERRQ(ierr);
307a198735bSStefano Zampini     pcbddc->divudotp_vl2l = vl2l;
308a198735bSStefano Zampini   }
309a198735bSStefano Zampini   PetscFunctionReturn(0);
310a198735bSStefano Zampini }
3113d996552SStefano Zampini 
312a198735bSStefano Zampini /*@
313a198735bSStefano Zampini  PCBDDCSetDivergenceMat - Sets the linear operator representing \int_\Omega \div {\bf u} \cdot p dx
314a198735bSStefano Zampini 
315a198735bSStefano Zampini    Collective on PC
316a198735bSStefano Zampini 
317a198735bSStefano Zampini    Input Parameters:
318a198735bSStefano Zampini +  pc - the preconditioning context
319a198735bSStefano Zampini .  divudotp - the matrix (must be of type MATIS)
3208ae0ca82SStefano Zampini .  trans - if trans if false (resp. true), then pressures are in the test (trial) space and velocities are in the trial (test) space.
321bb05f991SStefano Zampini -  vl2l - optional IS describing the local (wrt the local mat in divudotp) to local (wrt the local mat in pc->pmat) map for the velocities
322a198735bSStefano Zampini 
323a198735bSStefano Zampini    Level: advanced
324a198735bSStefano Zampini 
3258ae0ca82SStefano Zampini    Notes: This auxiliary matrix is used to compute quadrature weights representing the net-flux across subdomain boundaries
326a198735bSStefano Zampini 
327a198735bSStefano Zampini .seealso: PCBDDC
328a198735bSStefano Zampini @*/
3298ae0ca82SStefano Zampini PetscErrorCode PCBDDCSetDivergenceMat(PC pc, Mat divudotp, PetscBool trans, IS vl2l)
330a198735bSStefano Zampini {
331a198735bSStefano Zampini   PetscBool      ismatis;
332a198735bSStefano Zampini   PetscErrorCode ierr;
333a198735bSStefano Zampini 
334a198735bSStefano Zampini   PetscFunctionBegin;
335a198735bSStefano Zampini   PetscValidHeaderSpecific(pc,PC_CLASSID,1);
336a198735bSStefano Zampini   PetscValidHeaderSpecific(divudotp,MAT_CLASSID,2);
337a198735bSStefano Zampini   PetscCheckSameComm(pc,1,divudotp,2);
3388ae0ca82SStefano Zampini   PetscValidLogicalCollectiveBool(pc,trans,3);
3398ae0ca82SStefano Zampini   if (vl2l) PetscValidHeaderSpecific(divudotp,IS_CLASSID,4);
340a198735bSStefano Zampini   ierr = PetscObjectTypeCompare((PetscObject)divudotp,MATIS,&ismatis);CHKERRQ(ierr);
341a198735bSStefano Zampini   if (!ismatis) SETERRQ(PetscObjectComm((PetscObject)pc),PETSC_ERR_ARG_WRONG,"Divergence matrix needs to be of type MATIS");
3428ae0ca82SStefano Zampini   ierr = PetscTryMethod(pc,"PCBDDCSetDivergenceMat_C",(PC,Mat,PetscBool,IS),(pc,divudotp,trans,vl2l));CHKERRQ(ierr);
343a198735bSStefano Zampini   PetscFunctionReturn(0);
344a198735bSStefano Zampini }
3452d505d7fSStefano Zampini 
3461dd7afcfSStefano Zampini static PetscErrorCode PCBDDCSetChangeOfBasisMat_BDDC(PC pc, Mat change, PetscBool interior)
347b9b85e73SStefano Zampini {
348b9b85e73SStefano Zampini   PC_BDDC        *pcbddc = (PC_BDDC*)pc->data;
349b9b85e73SStefano Zampini   PetscErrorCode ierr;
350b9b85e73SStefano Zampini 
351b9b85e73SStefano Zampini   PetscFunctionBegin;
352b9b85e73SStefano Zampini   ierr = PetscObjectReference((PetscObject)change);CHKERRQ(ierr);
35356282151SStefano Zampini   ierr = MatDestroy(&pcbddc->user_ChangeOfBasisMatrix);CHKERRQ(ierr);
354b9b85e73SStefano Zampini   pcbddc->user_ChangeOfBasisMatrix = change;
3551dd7afcfSStefano Zampini   pcbddc->change_interior = interior;
356b9b85e73SStefano Zampini   PetscFunctionReturn(0);
357b9b85e73SStefano Zampini }
358b9b85e73SStefano Zampini /*@
359906d46d4SStefano Zampini  PCBDDCSetChangeOfBasisMat - Set user defined change of basis for dofs
360b9b85e73SStefano Zampini 
361b9b85e73SStefano Zampini    Collective on PC
362b9b85e73SStefano Zampini 
363b9b85e73SStefano Zampini    Input Parameters:
364b9b85e73SStefano Zampini +  pc - the preconditioning context
3651dd7afcfSStefano Zampini .  change - the change of basis matrix
3661dd7afcfSStefano Zampini -  interior - whether or not the change of basis modifies interior dofs
367b9b85e73SStefano Zampini 
368b9b85e73SStefano Zampini    Level: intermediate
369b9b85e73SStefano Zampini 
370b9b85e73SStefano Zampini    Notes:
371b9b85e73SStefano Zampini 
372b9b85e73SStefano Zampini .seealso: PCBDDC
373b9b85e73SStefano Zampini @*/
3741dd7afcfSStefano Zampini PetscErrorCode PCBDDCSetChangeOfBasisMat(PC pc, Mat change, PetscBool interior)
375b9b85e73SStefano Zampini {
376b9b85e73SStefano Zampini   PetscErrorCode ierr;
377b9b85e73SStefano Zampini 
378b9b85e73SStefano Zampini   PetscFunctionBegin;
379b9b85e73SStefano Zampini   PetscValidHeaderSpecific(pc,PC_CLASSID,1);
380b9b85e73SStefano Zampini   PetscValidHeaderSpecific(change,MAT_CLASSID,2);
381906d46d4SStefano Zampini   PetscCheckSameComm(pc,1,change,2);
382906d46d4SStefano Zampini   if (pc->mat) {
383906d46d4SStefano Zampini     PetscInt rows_c,cols_c,rows,cols;
384906d46d4SStefano Zampini     ierr = MatGetSize(pc->mat,&rows,&cols);CHKERRQ(ierr);
385906d46d4SStefano Zampini     ierr = MatGetSize(change,&rows_c,&cols_c);CHKERRQ(ierr);
386e0fe2d75SToby Isaac     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);
387e0fe2d75SToby Isaac     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);
388906d46d4SStefano Zampini     ierr = MatGetLocalSize(pc->mat,&rows,&cols);CHKERRQ(ierr);
389906d46d4SStefano Zampini     ierr = MatGetLocalSize(change,&rows_c,&cols_c);CHKERRQ(ierr);
390e0fe2d75SToby Isaac     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);
391e0fe2d75SToby Isaac     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);
392906d46d4SStefano Zampini   }
3931dd7afcfSStefano Zampini   ierr = PetscTryMethod(pc,"PCBDDCSetChangeOfBasisMat_C",(PC,Mat,PetscBool),(pc,change,interior));CHKERRQ(ierr);
394b9b85e73SStefano Zampini   PetscFunctionReturn(0);
395b9b85e73SStefano Zampini }
3962d505d7fSStefano Zampini 
39730368db7SStefano Zampini static PetscErrorCode PCBDDCSetPrimalVerticesIS_BDDC(PC pc, IS PrimalVertices)
39830368db7SStefano Zampini {
39930368db7SStefano Zampini   PC_BDDC        *pcbddc = (PC_BDDC*)pc->data;
40056282151SStefano Zampini   PetscBool      isequal = PETSC_FALSE;
40130368db7SStefano Zampini   PetscErrorCode ierr;
40230368db7SStefano Zampini 
40330368db7SStefano Zampini   PetscFunctionBegin;
40456282151SStefano Zampini   ierr = PetscObjectReference((PetscObject)PrimalVertices);CHKERRQ(ierr);
40556282151SStefano Zampini   if (pcbddc->user_primal_vertices) {
40656282151SStefano Zampini     ierr = ISEqual(PrimalVertices,pcbddc->user_primal_vertices,&isequal);CHKERRQ(ierr);
40756282151SStefano Zampini   }
40830368db7SStefano Zampini   ierr = ISDestroy(&pcbddc->user_primal_vertices);CHKERRQ(ierr);
40930368db7SStefano Zampini   ierr = ISDestroy(&pcbddc->user_primal_vertices_local);CHKERRQ(ierr);
41030368db7SStefano Zampini   pcbddc->user_primal_vertices = PrimalVertices;
41156282151SStefano Zampini   if (!isequal) pcbddc->recompute_topography = PETSC_TRUE;
41230368db7SStefano Zampini   PetscFunctionReturn(0);
41330368db7SStefano Zampini }
41430368db7SStefano Zampini /*@
41530368db7SStefano Zampini  PCBDDCSetPrimalVerticesIS - Set additional user defined primal vertices in PCBDDC
41630368db7SStefano Zampini 
41730368db7SStefano Zampini    Collective
41830368db7SStefano Zampini 
41930368db7SStefano Zampini    Input Parameters:
42030368db7SStefano Zampini +  pc - the preconditioning context
42130368db7SStefano Zampini -  PrimalVertices - index set of primal vertices in global numbering (can be empty)
42230368db7SStefano Zampini 
42330368db7SStefano Zampini    Level: intermediate
42430368db7SStefano Zampini 
42530368db7SStefano Zampini    Notes:
42630368db7SStefano Zampini      Any process can list any global node
42730368db7SStefano Zampini 
42830368db7SStefano Zampini .seealso: PCBDDC
42930368db7SStefano Zampini @*/
43030368db7SStefano Zampini PetscErrorCode PCBDDCSetPrimalVerticesIS(PC pc, IS PrimalVertices)
43130368db7SStefano Zampini {
43230368db7SStefano Zampini   PetscErrorCode ierr;
43330368db7SStefano Zampini 
43430368db7SStefano Zampini   PetscFunctionBegin;
43530368db7SStefano Zampini   PetscValidHeaderSpecific(pc,PC_CLASSID,1);
43630368db7SStefano Zampini   PetscValidHeaderSpecific(PrimalVertices,IS_CLASSID,2);
43730368db7SStefano Zampini   PetscCheckSameComm(pc,1,PrimalVertices,2);
43830368db7SStefano Zampini   ierr = PetscTryMethod(pc,"PCBDDCSetPrimalVerticesIS_C",(PC,IS),(pc,PrimalVertices));CHKERRQ(ierr);
43930368db7SStefano Zampini   PetscFunctionReturn(0);
44030368db7SStefano Zampini }
4412d505d7fSStefano Zampini 
442674ae819SStefano Zampini static PetscErrorCode PCBDDCSetPrimalVerticesLocalIS_BDDC(PC pc, IS PrimalVertices)
443674ae819SStefano Zampini {
444674ae819SStefano Zampini   PC_BDDC        *pcbddc = (PC_BDDC*)pc->data;
44556282151SStefano Zampini   PetscBool      isequal = PETSC_FALSE;
446674ae819SStefano Zampini   PetscErrorCode ierr;
4471e6b0712SBarry Smith 
448674ae819SStefano Zampini   PetscFunctionBegin;
44956282151SStefano Zampini   ierr = PetscObjectReference((PetscObject)PrimalVertices);CHKERRQ(ierr);
45056282151SStefano Zampini   if (pcbddc->user_primal_vertices_local) {
45156282151SStefano Zampini     ierr = ISEqual(PrimalVertices,pcbddc->user_primal_vertices_local,&isequal);CHKERRQ(ierr);
45256282151SStefano Zampini   }
453674ae819SStefano Zampini   ierr = ISDestroy(&pcbddc->user_primal_vertices);CHKERRQ(ierr);
45430368db7SStefano Zampini   ierr = ISDestroy(&pcbddc->user_primal_vertices_local);CHKERRQ(ierr);
45530368db7SStefano Zampini   pcbddc->user_primal_vertices_local = PrimalVertices;
45656282151SStefano Zampini   if (!isequal) pcbddc->recompute_topography = PETSC_TRUE;
457674ae819SStefano Zampini   PetscFunctionReturn(0);
458674ae819SStefano Zampini }
459674ae819SStefano Zampini /*@
46028509bceSStefano Zampini  PCBDDCSetPrimalVerticesLocalIS - Set additional user defined primal vertices in PCBDDC
461674ae819SStefano Zampini 
46217eb1463SStefano Zampini    Collective
463674ae819SStefano Zampini 
464674ae819SStefano Zampini    Input Parameters:
465674ae819SStefano Zampini +  pc - the preconditioning context
46617eb1463SStefano Zampini -  PrimalVertices - index set of primal vertices in local numbering (can be empty)
467674ae819SStefano Zampini 
468674ae819SStefano Zampini    Level: intermediate
469674ae819SStefano Zampini 
470674ae819SStefano Zampini    Notes:
471674ae819SStefano Zampini 
472674ae819SStefano Zampini .seealso: PCBDDC
473674ae819SStefano Zampini @*/
474674ae819SStefano Zampini PetscErrorCode PCBDDCSetPrimalVerticesLocalIS(PC pc, IS PrimalVertices)
475674ae819SStefano Zampini {
476674ae819SStefano Zampini   PetscErrorCode ierr;
477674ae819SStefano Zampini 
478674ae819SStefano Zampini   PetscFunctionBegin;
479674ae819SStefano Zampini   PetscValidHeaderSpecific(pc,PC_CLASSID,1);
480674ae819SStefano Zampini   PetscValidHeaderSpecific(PrimalVertices,IS_CLASSID,2);
48117eb1463SStefano Zampini   PetscCheckSameComm(pc,1,PrimalVertices,2);
482674ae819SStefano Zampini   ierr = PetscTryMethod(pc,"PCBDDCSetPrimalVerticesLocalIS_C",(PC,IS),(pc,PrimalVertices));CHKERRQ(ierr);
483674ae819SStefano Zampini   PetscFunctionReturn(0);
484674ae819SStefano Zampini }
4852d505d7fSStefano Zampini 
4864fad6a16SStefano Zampini static PetscErrorCode PCBDDCSetCoarseningRatio_BDDC(PC pc,PetscInt k)
4874fad6a16SStefano Zampini {
4884fad6a16SStefano Zampini   PC_BDDC  *pcbddc = (PC_BDDC*)pc->data;
4894fad6a16SStefano Zampini 
4904fad6a16SStefano Zampini   PetscFunctionBegin;
4914fad6a16SStefano Zampini   pcbddc->coarsening_ratio = k;
4924fad6a16SStefano Zampini   PetscFunctionReturn(0);
4934fad6a16SStefano Zampini }
4941e6b0712SBarry Smith 
4954fad6a16SStefano Zampini /*@
49628509bceSStefano Zampini  PCBDDCSetCoarseningRatio - Set coarsening ratio used in multilevel
4974fad6a16SStefano Zampini 
4984fad6a16SStefano Zampini    Logically collective on PC
4994fad6a16SStefano Zampini 
5004fad6a16SStefano Zampini    Input Parameters:
5014fad6a16SStefano Zampini +  pc - the preconditioning context
50228509bceSStefano Zampini -  k - coarsening ratio (H/h at the coarser level)
5034fad6a16SStefano Zampini 
5040f202f7eSStefano Zampini    Options Database Keys:
5050f202f7eSStefano Zampini .    -pc_bddc_coarsening_ratio
5064fad6a16SStefano Zampini 
5074fad6a16SStefano Zampini    Level: intermediate
5084fad6a16SStefano Zampini 
5094fad6a16SStefano Zampini    Notes:
5100f202f7eSStefano Zampini      Approximatively k subdomains at the finer level will be aggregated into a single subdomain at the coarser level
5114fad6a16SStefano Zampini 
5120f202f7eSStefano Zampini .seealso: PCBDDC, PCBDDCSetLevels()
5134fad6a16SStefano Zampini @*/
5144fad6a16SStefano Zampini PetscErrorCode PCBDDCSetCoarseningRatio(PC pc,PetscInt k)
5154fad6a16SStefano Zampini {
5164fad6a16SStefano Zampini   PetscErrorCode ierr;
5174fad6a16SStefano Zampini 
5184fad6a16SStefano Zampini   PetscFunctionBegin;
5194fad6a16SStefano Zampini   PetscValidHeaderSpecific(pc,PC_CLASSID,1);
5202b510759SStefano Zampini   PetscValidLogicalCollectiveInt(pc,k,2);
5214fad6a16SStefano Zampini   ierr = PetscTryMethod(pc,"PCBDDCSetCoarseningRatio_C",(PC,PetscInt),(pc,k));CHKERRQ(ierr);
5224fad6a16SStefano Zampini   PetscFunctionReturn(0);
5234fad6a16SStefano Zampini }
5242b510759SStefano Zampini 
525b8ffe317SStefano Zampini /* The following functions (PCBDDCSetUseExactDirichlet PCBDDCSetLevel) are not public */
526b8ffe317SStefano Zampini static PetscErrorCode PCBDDCSetUseExactDirichlet_BDDC(PC pc,PetscBool flg)
527b8ffe317SStefano Zampini {
528b8ffe317SStefano Zampini   PC_BDDC  *pcbddc = (PC_BDDC*)pc->data;
529b8ffe317SStefano Zampini 
530b8ffe317SStefano Zampini   PetscFunctionBegin;
53185c4d303SStefano Zampini   pcbddc->use_exact_dirichlet_trick = flg;
532b8ffe317SStefano Zampini   PetscFunctionReturn(0);
533b8ffe317SStefano Zampini }
534b8ffe317SStefano Zampini 
535b8ffe317SStefano Zampini PetscErrorCode PCBDDCSetUseExactDirichlet(PC pc,PetscBool flg)
5362b510759SStefano Zampini {
5372b510759SStefano Zampini   PetscErrorCode ierr;
5382b510759SStefano Zampini 
5392b510759SStefano Zampini   PetscFunctionBegin;
5402b510759SStefano Zampini   PetscValidHeaderSpecific(pc,PC_CLASSID,1);
541b8ffe317SStefano Zampini   PetscValidLogicalCollectiveBool(pc,flg,2);
542b8ffe317SStefano Zampini   ierr = PetscTryMethod(pc,"PCBDDCSetUseExactDirichlet_C",(PC,PetscBool),(pc,flg));CHKERRQ(ierr);
5432b510759SStefano Zampini   PetscFunctionReturn(0);
5442b510759SStefano Zampini }
5451e6b0712SBarry Smith 
5462b510759SStefano Zampini static PetscErrorCode PCBDDCSetLevel_BDDC(PC pc,PetscInt level)
5474fad6a16SStefano Zampini {
5484fad6a16SStefano Zampini   PC_BDDC  *pcbddc = (PC_BDDC*)pc->data;
5494fad6a16SStefano Zampini 
5504fad6a16SStefano Zampini   PetscFunctionBegin;
5512b510759SStefano Zampini   pcbddc->current_level = level;
5524fad6a16SStefano Zampini   PetscFunctionReturn(0);
5534fad6a16SStefano Zampini }
5541e6b0712SBarry Smith 
555b8ffe317SStefano Zampini PetscErrorCode PCBDDCSetLevel(PC pc,PetscInt level)
556b8ffe317SStefano Zampini {
557b8ffe317SStefano Zampini   PetscErrorCode ierr;
558b8ffe317SStefano Zampini 
559b8ffe317SStefano Zampini   PetscFunctionBegin;
560b8ffe317SStefano Zampini   PetscValidHeaderSpecific(pc,PC_CLASSID,1);
561b8ffe317SStefano Zampini   PetscValidLogicalCollectiveInt(pc,level,2);
562b8ffe317SStefano Zampini   ierr = PetscTryMethod(pc,"PCBDDCSetLevel_C",(PC,PetscInt),(pc,level));CHKERRQ(ierr);
563b8ffe317SStefano Zampini   PetscFunctionReturn(0);
564b8ffe317SStefano Zampini }
565b8ffe317SStefano Zampini 
5662b510759SStefano Zampini static PetscErrorCode PCBDDCSetLevels_BDDC(PC pc,PetscInt levels)
5672b510759SStefano Zampini {
5682b510759SStefano Zampini   PC_BDDC  *pcbddc = (PC_BDDC*)pc->data;
5692b510759SStefano Zampini 
5702b510759SStefano Zampini   PetscFunctionBegin;
5712b510759SStefano Zampini   pcbddc->max_levels = levels;
5722b510759SStefano Zampini   PetscFunctionReturn(0);
5732b510759SStefano Zampini }
5742b510759SStefano Zampini 
5754fad6a16SStefano Zampini /*@
57628509bceSStefano Zampini  PCBDDCSetLevels - Sets the maximum number of levels for multilevel
5774fad6a16SStefano Zampini 
5784fad6a16SStefano Zampini    Logically collective on PC
5794fad6a16SStefano Zampini 
5804fad6a16SStefano Zampini    Input Parameters:
5814fad6a16SStefano Zampini +  pc - the preconditioning context
58228509bceSStefano Zampini -  levels - the maximum number of levels (max 9)
5834fad6a16SStefano Zampini 
5840f202f7eSStefano Zampini    Options Database Keys:
5850f202f7eSStefano Zampini .    -pc_bddc_levels
5864fad6a16SStefano Zampini 
5874fad6a16SStefano Zampini    Level: intermediate
5884fad6a16SStefano Zampini 
5894fad6a16SStefano Zampini    Notes:
5900f202f7eSStefano Zampini      Default value is 0, i.e. traditional one-level BDDC
5914fad6a16SStefano Zampini 
5920f202f7eSStefano Zampini .seealso: PCBDDC, PCBDDCSetCoarseningRatio()
5934fad6a16SStefano Zampini @*/
5942b510759SStefano Zampini PetscErrorCode PCBDDCSetLevels(PC pc,PetscInt levels)
5954fad6a16SStefano Zampini {
5964fad6a16SStefano Zampini   PetscErrorCode ierr;
5974fad6a16SStefano Zampini 
5984fad6a16SStefano Zampini   PetscFunctionBegin;
5994fad6a16SStefano Zampini   PetscValidHeaderSpecific(pc,PC_CLASSID,1);
6002b510759SStefano Zampini   PetscValidLogicalCollectiveInt(pc,levels,2);
601312be037SStefano Zampini   if (levels > 99) SETERRQ(PetscObjectComm((PetscObject)pc),PETSC_ERR_SUP,"Maximum number of levels for bddc is 99\n");
6022b510759SStefano Zampini   ierr = PetscTryMethod(pc,"PCBDDCSetLevels_C",(PC,PetscInt),(pc,levels));CHKERRQ(ierr);
6034fad6a16SStefano Zampini   PetscFunctionReturn(0);
6044fad6a16SStefano Zampini }
6051e6b0712SBarry Smith 
6063b03a366Sstefano_zampini static PetscErrorCode PCBDDCSetDirichletBoundaries_BDDC(PC pc,IS DirichletBoundaries)
6073b03a366Sstefano_zampini {
6083b03a366Sstefano_zampini   PC_BDDC        *pcbddc = (PC_BDDC*)pc->data;
60956282151SStefano Zampini   PetscBool      isequal = PETSC_FALSE;
6103b03a366Sstefano_zampini   PetscErrorCode ierr;
6113b03a366Sstefano_zampini 
6123b03a366Sstefano_zampini   PetscFunctionBegin;
61356282151SStefano Zampini   ierr = PetscObjectReference((PetscObject)DirichletBoundaries);CHKERRQ(ierr);
61456282151SStefano Zampini   if (pcbddc->DirichletBoundaries) {
61556282151SStefano Zampini     ierr = ISEqual(DirichletBoundaries,pcbddc->DirichletBoundaries,&isequal);CHKERRQ(ierr);
61656282151SStefano Zampini   }
617785d1243SStefano Zampini   /* last user setting takes precendence -> destroy any other customization */
618785d1243SStefano Zampini   ierr = ISDestroy(&pcbddc->DirichletBoundariesLocal);CHKERRQ(ierr);
6193b03a366Sstefano_zampini   ierr = ISDestroy(&pcbddc->DirichletBoundaries);CHKERRQ(ierr);
62036e030ebSStefano Zampini   pcbddc->DirichletBoundaries = DirichletBoundaries;
62156282151SStefano Zampini   if (!isequal) pcbddc->recompute_topography = PETSC_TRUE;
6223b03a366Sstefano_zampini   PetscFunctionReturn(0);
6233b03a366Sstefano_zampini }
6241e6b0712SBarry Smith 
6253b03a366Sstefano_zampini /*@
62628509bceSStefano Zampini  PCBDDCSetDirichletBoundaries - Set IS defining Dirichlet boundaries for the global problem.
6273b03a366Sstefano_zampini 
628785d1243SStefano Zampini    Collective
6293b03a366Sstefano_zampini 
6303b03a366Sstefano_zampini    Input Parameters:
6313b03a366Sstefano_zampini +  pc - the preconditioning context
632785d1243SStefano Zampini -  DirichletBoundaries - parallel IS defining the Dirichlet boundaries
6333b03a366Sstefano_zampini 
6343b03a366Sstefano_zampini    Level: intermediate
6353b03a366Sstefano_zampini 
6360f202f7eSStefano Zampini    Notes:
6370f202f7eSStefano Zampini      Provide the information if you used MatZeroRows/Columns routines. Any process can list any global node
6383b03a366Sstefano_zampini 
6390f202f7eSStefano Zampini .seealso: PCBDDC, PCBDDCSetDirichletBoundariesLocal(), MatZeroRows(), MatZeroRowsColumns()
6403b03a366Sstefano_zampini @*/
6413b03a366Sstefano_zampini PetscErrorCode PCBDDCSetDirichletBoundaries(PC pc,IS DirichletBoundaries)
6423b03a366Sstefano_zampini {
6433b03a366Sstefano_zampini   PetscErrorCode ierr;
6443b03a366Sstefano_zampini 
6453b03a366Sstefano_zampini   PetscFunctionBegin;
6463b03a366Sstefano_zampini   PetscValidHeaderSpecific(pc,PC_CLASSID,1);
647674ae819SStefano Zampini   PetscValidHeaderSpecific(DirichletBoundaries,IS_CLASSID,2);
648785d1243SStefano Zampini   PetscCheckSameComm(pc,1,DirichletBoundaries,2);
6493b03a366Sstefano_zampini   ierr = PetscTryMethod(pc,"PCBDDCSetDirichletBoundaries_C",(PC,IS),(pc,DirichletBoundaries));CHKERRQ(ierr);
6503b03a366Sstefano_zampini   PetscFunctionReturn(0);
6513b03a366Sstefano_zampini }
6521e6b0712SBarry Smith 
65382ba6b80SStefano Zampini static PetscErrorCode PCBDDCSetDirichletBoundariesLocal_BDDC(PC pc,IS DirichletBoundaries)
6543b03a366Sstefano_zampini {
6553b03a366Sstefano_zampini   PC_BDDC        *pcbddc = (PC_BDDC*)pc->data;
65656282151SStefano Zampini   PetscBool      isequal = PETSC_FALSE;
6573b03a366Sstefano_zampini   PetscErrorCode ierr;
6583b03a366Sstefano_zampini 
6593b03a366Sstefano_zampini   PetscFunctionBegin;
66056282151SStefano Zampini   ierr = PetscObjectReference((PetscObject)DirichletBoundaries);CHKERRQ(ierr);
66156282151SStefano Zampini   if (pcbddc->DirichletBoundariesLocal) {
66256282151SStefano Zampini     ierr = ISEqual(DirichletBoundaries,pcbddc->DirichletBoundariesLocal,&isequal);CHKERRQ(ierr);
66356282151SStefano Zampini   }
664785d1243SStefano Zampini   /* last user setting takes precendence -> destroy any other customization */
665785d1243SStefano Zampini   ierr = ISDestroy(&pcbddc->DirichletBoundariesLocal);CHKERRQ(ierr);
6663b03a366Sstefano_zampini   ierr = ISDestroy(&pcbddc->DirichletBoundaries);CHKERRQ(ierr);
667785d1243SStefano Zampini   pcbddc->DirichletBoundariesLocal = DirichletBoundaries;
66856282151SStefano Zampini   if (!isequal) pcbddc->recompute_topography = PETSC_TRUE;
6693b03a366Sstefano_zampini   PetscFunctionReturn(0);
6703b03a366Sstefano_zampini }
6713b03a366Sstefano_zampini 
6723b03a366Sstefano_zampini /*@
67382ba6b80SStefano Zampini  PCBDDCSetDirichletBoundariesLocal - Set IS defining Dirichlet boundaries for the global problem in local ordering.
6743b03a366Sstefano_zampini 
675785d1243SStefano Zampini    Collective
6763b03a366Sstefano_zampini 
6773b03a366Sstefano_zampini    Input Parameters:
6783b03a366Sstefano_zampini +  pc - the preconditioning context
67982ba6b80SStefano Zampini -  DirichletBoundaries - parallel IS defining the Dirichlet boundaries (in local ordering)
6803b03a366Sstefano_zampini 
6813b03a366Sstefano_zampini    Level: intermediate
6823b03a366Sstefano_zampini 
6833b03a366Sstefano_zampini    Notes:
6843b03a366Sstefano_zampini 
6850f202f7eSStefano Zampini .seealso: PCBDDC, PCBDDCSetDirichletBoundaries(), MatZeroRows(), MatZeroRowsColumns()
6863b03a366Sstefano_zampini @*/
68782ba6b80SStefano Zampini PetscErrorCode PCBDDCSetDirichletBoundariesLocal(PC pc,IS DirichletBoundaries)
6883b03a366Sstefano_zampini {
6893b03a366Sstefano_zampini   PetscErrorCode ierr;
6903b03a366Sstefano_zampini 
6913b03a366Sstefano_zampini   PetscFunctionBegin;
6923b03a366Sstefano_zampini   PetscValidHeaderSpecific(pc,PC_CLASSID,1);
6933b03a366Sstefano_zampini   PetscValidHeaderSpecific(DirichletBoundaries,IS_CLASSID,2);
69482ba6b80SStefano Zampini   PetscCheckSameComm(pc,1,DirichletBoundaries,2);
69582ba6b80SStefano Zampini   ierr = PetscTryMethod(pc,"PCBDDCSetDirichletBoundariesLocal_C",(PC,IS),(pc,DirichletBoundaries));CHKERRQ(ierr);
6963b03a366Sstefano_zampini   PetscFunctionReturn(0);
6973b03a366Sstefano_zampini }
6983b03a366Sstefano_zampini 
69953cdbc3dSStefano Zampini static PetscErrorCode PCBDDCSetNeumannBoundaries_BDDC(PC pc,IS NeumannBoundaries)
7000c7d97c5SJed Brown {
7010c7d97c5SJed Brown   PC_BDDC        *pcbddc = (PC_BDDC*)pc->data;
70256282151SStefano Zampini   PetscBool      isequal = PETSC_FALSE;
70353cdbc3dSStefano Zampini   PetscErrorCode ierr;
7040c7d97c5SJed Brown 
7050c7d97c5SJed Brown   PetscFunctionBegin;
70656282151SStefano Zampini   ierr = PetscObjectReference((PetscObject)NeumannBoundaries);CHKERRQ(ierr);
70756282151SStefano Zampini   if (pcbddc->NeumannBoundaries) {
70856282151SStefano Zampini     ierr = ISEqual(NeumannBoundaries,pcbddc->NeumannBoundaries,&isequal);CHKERRQ(ierr);
70956282151SStefano Zampini   }
710785d1243SStefano Zampini   /* last user setting takes precendence -> destroy any other customization */
711785d1243SStefano Zampini   ierr = ISDestroy(&pcbddc->NeumannBoundariesLocal);CHKERRQ(ierr);
71253cdbc3dSStefano Zampini   ierr = ISDestroy(&pcbddc->NeumannBoundaries);CHKERRQ(ierr);
71336e030ebSStefano Zampini   pcbddc->NeumannBoundaries = NeumannBoundaries;
71456282151SStefano Zampini   if (!isequal) pcbddc->recompute_topography = PETSC_TRUE;
7150c7d97c5SJed Brown   PetscFunctionReturn(0);
7160c7d97c5SJed Brown }
7171e6b0712SBarry Smith 
71857527edcSJed Brown /*@
71928509bceSStefano Zampini  PCBDDCSetNeumannBoundaries - Set IS defining Neumann boundaries for the global problem.
72057527edcSJed Brown 
721785d1243SStefano Zampini    Collective
72257527edcSJed Brown 
72357527edcSJed Brown    Input Parameters:
72457527edcSJed Brown +  pc - the preconditioning context
725785d1243SStefano Zampini -  NeumannBoundaries - parallel IS defining the Neumann boundaries
72657527edcSJed Brown 
72757527edcSJed Brown    Level: intermediate
72857527edcSJed Brown 
7290f202f7eSStefano Zampini    Notes:
7300f202f7eSStefano Zampini      Any process can list any global node
73157527edcSJed Brown 
7320f202f7eSStefano Zampini .seealso: PCBDDC, PCBDDCSetNeumannBoundariesLocal()
73357527edcSJed Brown @*/
73453cdbc3dSStefano Zampini PetscErrorCode PCBDDCSetNeumannBoundaries(PC pc,IS NeumannBoundaries)
7350c7d97c5SJed Brown {
7360c7d97c5SJed Brown   PetscErrorCode ierr;
7370c7d97c5SJed Brown 
7380c7d97c5SJed Brown   PetscFunctionBegin;
7390c7d97c5SJed Brown   PetscValidHeaderSpecific(pc,PC_CLASSID,1);
740674ae819SStefano Zampini   PetscValidHeaderSpecific(NeumannBoundaries,IS_CLASSID,2);
741785d1243SStefano Zampini   PetscCheckSameComm(pc,1,NeumannBoundaries,2);
74253cdbc3dSStefano Zampini   ierr = PetscTryMethod(pc,"PCBDDCSetNeumannBoundaries_C",(PC,IS),(pc,NeumannBoundaries));CHKERRQ(ierr);
74353cdbc3dSStefano Zampini   PetscFunctionReturn(0);
74453cdbc3dSStefano Zampini }
7451e6b0712SBarry Smith 
74682ba6b80SStefano Zampini static PetscErrorCode PCBDDCSetNeumannBoundariesLocal_BDDC(PC pc,IS NeumannBoundaries)
7470c7d97c5SJed Brown {
7480c7d97c5SJed Brown   PC_BDDC        *pcbddc = (PC_BDDC*)pc->data;
74956282151SStefano Zampini   PetscBool      isequal = PETSC_FALSE;
7500c7d97c5SJed Brown   PetscErrorCode ierr;
7510c7d97c5SJed Brown 
7520c7d97c5SJed Brown   PetscFunctionBegin;
75356282151SStefano Zampini   ierr = PetscObjectReference((PetscObject)NeumannBoundaries);CHKERRQ(ierr);
75456282151SStefano Zampini   if (pcbddc->NeumannBoundariesLocal) {
75556282151SStefano Zampini     ierr = ISEqual(NeumannBoundaries,pcbddc->NeumannBoundariesLocal,&isequal);CHKERRQ(ierr);
75656282151SStefano Zampini   }
757785d1243SStefano Zampini   /* last user setting takes precendence -> destroy any other customization */
758785d1243SStefano Zampini   ierr = ISDestroy(&pcbddc->NeumannBoundariesLocal);CHKERRQ(ierr);
7590c7d97c5SJed Brown   ierr = ISDestroy(&pcbddc->NeumannBoundaries);CHKERRQ(ierr);
760785d1243SStefano Zampini   pcbddc->NeumannBoundariesLocal = NeumannBoundaries;
76156282151SStefano Zampini   if (!isequal) pcbddc->recompute_topography = PETSC_TRUE;
7620c7d97c5SJed Brown   PetscFunctionReturn(0);
7630c7d97c5SJed Brown }
7640c7d97c5SJed Brown 
7650c7d97c5SJed Brown /*@
76682ba6b80SStefano Zampini  PCBDDCSetNeumannBoundariesLocal - Set IS defining Neumann boundaries for the global problem in local ordering.
7670c7d97c5SJed Brown 
768785d1243SStefano Zampini    Collective
7690c7d97c5SJed Brown 
7700c7d97c5SJed Brown    Input Parameters:
7710c7d97c5SJed Brown +  pc - the preconditioning context
77282ba6b80SStefano Zampini -  NeumannBoundaries - parallel IS defining the subdomain part of Neumann boundaries (in local ordering)
7730c7d97c5SJed Brown 
7740c7d97c5SJed Brown    Level: intermediate
7750c7d97c5SJed Brown 
7760c7d97c5SJed Brown    Notes:
7770c7d97c5SJed Brown 
7780f202f7eSStefano Zampini .seealso: PCBDDC, PCBDDCSetNeumannBoundaries()
7790c7d97c5SJed Brown @*/
78082ba6b80SStefano Zampini PetscErrorCode PCBDDCSetNeumannBoundariesLocal(PC pc,IS NeumannBoundaries)
7810c7d97c5SJed Brown {
7820c7d97c5SJed Brown   PetscErrorCode ierr;
7830c7d97c5SJed Brown 
7840c7d97c5SJed Brown   PetscFunctionBegin;
7850c7d97c5SJed Brown   PetscValidHeaderSpecific(pc,PC_CLASSID,1);
7860c7d97c5SJed Brown   PetscValidHeaderSpecific(NeumannBoundaries,IS_CLASSID,2);
78782ba6b80SStefano Zampini   PetscCheckSameComm(pc,1,NeumannBoundaries,2);
78882ba6b80SStefano Zampini   ierr = PetscTryMethod(pc,"PCBDDCSetNeumannBoundariesLocal_C",(PC,IS),(pc,NeumannBoundaries));CHKERRQ(ierr);
78953cdbc3dSStefano Zampini   PetscFunctionReturn(0);
79053cdbc3dSStefano Zampini }
79153cdbc3dSStefano Zampini 
792da1bb401SStefano Zampini static PetscErrorCode PCBDDCGetDirichletBoundaries_BDDC(PC pc,IS *DirichletBoundaries)
793da1bb401SStefano Zampini {
794da1bb401SStefano Zampini   PC_BDDC  *pcbddc = (PC_BDDC*)pc->data;
795da1bb401SStefano Zampini 
796da1bb401SStefano Zampini   PetscFunctionBegin;
797da1bb401SStefano Zampini   *DirichletBoundaries = pcbddc->DirichletBoundaries;
798da1bb401SStefano Zampini   PetscFunctionReturn(0);
799da1bb401SStefano Zampini }
8001e6b0712SBarry Smith 
801da1bb401SStefano Zampini /*@
802785d1243SStefano Zampini  PCBDDCGetDirichletBoundaries - Get parallel IS for Dirichlet boundaries
803da1bb401SStefano Zampini 
804785d1243SStefano Zampini    Collective
805785d1243SStefano Zampini 
806785d1243SStefano Zampini    Input Parameters:
807785d1243SStefano Zampini .  pc - the preconditioning context
808785d1243SStefano Zampini 
809785d1243SStefano Zampini    Output Parameters:
810785d1243SStefano Zampini .  DirichletBoundaries - index set defining the Dirichlet boundaries
811785d1243SStefano Zampini 
812785d1243SStefano Zampini    Level: intermediate
813785d1243SStefano Zampini 
8140f202f7eSStefano Zampini    Notes:
8150f202f7eSStefano Zampini      The IS returned (if any) is the same passed in earlier by the user with PCBDDCSetDirichletBoundaries
816785d1243SStefano Zampini 
817785d1243SStefano Zampini .seealso: PCBDDC
818785d1243SStefano Zampini @*/
819785d1243SStefano Zampini PetscErrorCode PCBDDCGetDirichletBoundaries(PC pc,IS *DirichletBoundaries)
820785d1243SStefano Zampini {
821785d1243SStefano Zampini   PetscErrorCode ierr;
822785d1243SStefano Zampini 
823785d1243SStefano Zampini   PetscFunctionBegin;
824785d1243SStefano Zampini   PetscValidHeaderSpecific(pc,PC_CLASSID,1);
825785d1243SStefano Zampini   ierr = PetscUseMethod(pc,"PCBDDCGetDirichletBoundaries_C",(PC,IS*),(pc,DirichletBoundaries));CHKERRQ(ierr);
826785d1243SStefano Zampini   PetscFunctionReturn(0);
827785d1243SStefano Zampini }
828785d1243SStefano Zampini 
829785d1243SStefano Zampini static PetscErrorCode PCBDDCGetDirichletBoundariesLocal_BDDC(PC pc,IS *DirichletBoundaries)
830785d1243SStefano Zampini {
831785d1243SStefano Zampini   PC_BDDC  *pcbddc = (PC_BDDC*)pc->data;
832785d1243SStefano Zampini 
833785d1243SStefano Zampini   PetscFunctionBegin;
834785d1243SStefano Zampini   *DirichletBoundaries = pcbddc->DirichletBoundariesLocal;
835785d1243SStefano Zampini   PetscFunctionReturn(0);
836785d1243SStefano Zampini }
837785d1243SStefano Zampini 
838da1bb401SStefano Zampini /*@
83982ba6b80SStefano Zampini  PCBDDCGetDirichletBoundariesLocal - Get parallel IS for Dirichlet boundaries (in local ordering)
840da1bb401SStefano Zampini 
841785d1243SStefano Zampini    Collective
842da1bb401SStefano Zampini 
843da1bb401SStefano Zampini    Input Parameters:
84428509bceSStefano Zampini .  pc - the preconditioning context
845da1bb401SStefano Zampini 
846da1bb401SStefano Zampini    Output Parameters:
84728509bceSStefano Zampini .  DirichletBoundaries - index set defining the subdomain part of Dirichlet boundaries
848da1bb401SStefano Zampini 
849da1bb401SStefano Zampini    Level: intermediate
850da1bb401SStefano Zampini 
851da1bb401SStefano Zampini    Notes:
8520f202f7eSStefano 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).
8530f202f7eSStefano Zampini           In the latter case, the IS will be available after PCSetUp.
854da1bb401SStefano Zampini 
855da1bb401SStefano Zampini .seealso: PCBDDC
856da1bb401SStefano Zampini @*/
85782ba6b80SStefano Zampini PetscErrorCode PCBDDCGetDirichletBoundariesLocal(PC pc,IS *DirichletBoundaries)
858da1bb401SStefano Zampini {
859da1bb401SStefano Zampini   PetscErrorCode ierr;
860da1bb401SStefano Zampini 
861da1bb401SStefano Zampini   PetscFunctionBegin;
862da1bb401SStefano Zampini   PetscValidHeaderSpecific(pc,PC_CLASSID,1);
86382ba6b80SStefano Zampini   ierr = PetscUseMethod(pc,"PCBDDCGetDirichletBoundariesLocal_C",(PC,IS*),(pc,DirichletBoundaries));CHKERRQ(ierr);
864da1bb401SStefano Zampini   PetscFunctionReturn(0);
865da1bb401SStefano Zampini }
8661e6b0712SBarry Smith 
86753cdbc3dSStefano Zampini static PetscErrorCode PCBDDCGetNeumannBoundaries_BDDC(PC pc,IS *NeumannBoundaries)
86853cdbc3dSStefano Zampini {
86953cdbc3dSStefano Zampini   PC_BDDC  *pcbddc = (PC_BDDC*)pc->data;
87053cdbc3dSStefano Zampini 
87153cdbc3dSStefano Zampini   PetscFunctionBegin;
87253cdbc3dSStefano Zampini   *NeumannBoundaries = pcbddc->NeumannBoundaries;
87353cdbc3dSStefano Zampini   PetscFunctionReturn(0);
87453cdbc3dSStefano Zampini }
8751e6b0712SBarry Smith 
87653cdbc3dSStefano Zampini /*@
877785d1243SStefano Zampini  PCBDDCGetNeumannBoundaries - Get parallel IS for Neumann boundaries
87853cdbc3dSStefano Zampini 
879785d1243SStefano Zampini    Collective
880785d1243SStefano Zampini 
881785d1243SStefano Zampini    Input Parameters:
882785d1243SStefano Zampini .  pc - the preconditioning context
883785d1243SStefano Zampini 
884785d1243SStefano Zampini    Output Parameters:
885785d1243SStefano Zampini .  NeumannBoundaries - index set defining the Neumann boundaries
886785d1243SStefano Zampini 
887785d1243SStefano Zampini    Level: intermediate
888785d1243SStefano Zampini 
8890f202f7eSStefano Zampini    Notes:
8900f202f7eSStefano Zampini      The IS returned (if any) is the same passed in earlier by the user with PCBDDCSetNeumannBoundaries
891785d1243SStefano Zampini 
892785d1243SStefano Zampini .seealso: PCBDDC
893785d1243SStefano Zampini @*/
894785d1243SStefano Zampini PetscErrorCode PCBDDCGetNeumannBoundaries(PC pc,IS *NeumannBoundaries)
895785d1243SStefano Zampini {
896785d1243SStefano Zampini   PetscErrorCode ierr;
897785d1243SStefano Zampini 
898785d1243SStefano Zampini   PetscFunctionBegin;
899785d1243SStefano Zampini   PetscValidHeaderSpecific(pc,PC_CLASSID,1);
900785d1243SStefano Zampini   ierr = PetscUseMethod(pc,"PCBDDCGetNeumannBoundaries_C",(PC,IS*),(pc,NeumannBoundaries));CHKERRQ(ierr);
901785d1243SStefano Zampini   PetscFunctionReturn(0);
902785d1243SStefano Zampini }
903785d1243SStefano Zampini 
904785d1243SStefano Zampini static PetscErrorCode PCBDDCGetNeumannBoundariesLocal_BDDC(PC pc,IS *NeumannBoundaries)
905785d1243SStefano Zampini {
906785d1243SStefano Zampini   PC_BDDC  *pcbddc = (PC_BDDC*)pc->data;
907785d1243SStefano Zampini 
908785d1243SStefano Zampini   PetscFunctionBegin;
909785d1243SStefano Zampini   *NeumannBoundaries = pcbddc->NeumannBoundariesLocal;
910785d1243SStefano Zampini   PetscFunctionReturn(0);
911785d1243SStefano Zampini }
912785d1243SStefano Zampini 
91353cdbc3dSStefano Zampini /*@
91482ba6b80SStefano Zampini  PCBDDCGetNeumannBoundariesLocal - Get parallel IS for Neumann boundaries (in local ordering)
91553cdbc3dSStefano Zampini 
916785d1243SStefano Zampini    Collective
91753cdbc3dSStefano Zampini 
91853cdbc3dSStefano Zampini    Input Parameters:
91928509bceSStefano Zampini .  pc - the preconditioning context
92053cdbc3dSStefano Zampini 
92153cdbc3dSStefano Zampini    Output Parameters:
92228509bceSStefano Zampini .  NeumannBoundaries - index set defining the subdomain part of Neumann boundaries
92353cdbc3dSStefano Zampini 
92453cdbc3dSStefano Zampini    Level: intermediate
92553cdbc3dSStefano Zampini 
92653cdbc3dSStefano Zampini    Notes:
9270f202f7eSStefano 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).
9280f202f7eSStefano Zampini           In the latter case, the IS will be available after PCSetUp.
92953cdbc3dSStefano Zampini 
93053cdbc3dSStefano Zampini .seealso: PCBDDC
93153cdbc3dSStefano Zampini @*/
93282ba6b80SStefano Zampini PetscErrorCode PCBDDCGetNeumannBoundariesLocal(PC pc,IS *NeumannBoundaries)
93353cdbc3dSStefano Zampini {
93453cdbc3dSStefano Zampini   PetscErrorCode ierr;
93553cdbc3dSStefano Zampini 
93653cdbc3dSStefano Zampini   PetscFunctionBegin;
93753cdbc3dSStefano Zampini   PetscValidHeaderSpecific(pc,PC_CLASSID,1);
93882ba6b80SStefano Zampini   ierr = PetscUseMethod(pc,"PCBDDCGetNeumannBoundariesLocal_C",(PC,IS*),(pc,NeumannBoundaries));CHKERRQ(ierr);
9390c7d97c5SJed Brown   PetscFunctionReturn(0);
9400c7d97c5SJed Brown }
9411e6b0712SBarry Smith 
9421a83f524SJed Brown static PetscErrorCode PCBDDCSetLocalAdjacencyGraph_BDDC(PC pc, PetscInt nvtxs,const PetscInt xadj[],const PetscInt adjncy[], PetscCopyMode copymode)
94336e030ebSStefano Zampini {
94436e030ebSStefano Zampini   PC_BDDC        *pcbddc = (PC_BDDC*)pc->data;
945da1bb401SStefano Zampini   PCBDDCGraph    mat_graph = pcbddc->mat_graph;
94656282151SStefano Zampini   PetscBool      same_data = PETSC_FALSE;
947da1bb401SStefano Zampini   PetscErrorCode ierr;
94836e030ebSStefano Zampini 
94936e030ebSStefano Zampini   PetscFunctionBegin;
9508687889aSStefano Zampini   if (!nvtxs) {
95104194a47SStefano Zampini     if (copymode == PETSC_OWN_POINTER) {
95204194a47SStefano Zampini       ierr = PetscFree(xadj);CHKERRQ(ierr);
95304194a47SStefano Zampini       ierr = PetscFree(adjncy);CHKERRQ(ierr);
95404194a47SStefano Zampini     }
9558687889aSStefano Zampini     ierr = PCBDDCGraphResetCSR(mat_graph);CHKERRQ(ierr);
9568687889aSStefano Zampini     PetscFunctionReturn(0);
9578687889aSStefano Zampini   }
95866da6bd7Sstefano_zampini   if (mat_graph->nvtxs == nvtxs && mat_graph->freecsr) { /* we own the data */
95956282151SStefano Zampini     if (mat_graph->xadj == xadj && mat_graph->adjncy == adjncy) same_data = PETSC_TRUE;
96056282151SStefano Zampini     if (!same_data && mat_graph->xadj[nvtxs] == xadj[nvtxs]) {
9612d505d7fSStefano Zampini       ierr = PetscMemcmp(xadj,mat_graph->xadj,(nvtxs+1)*sizeof(PetscInt),&same_data);CHKERRQ(ierr);
9622d505d7fSStefano Zampini       if (same_data) {
9632d505d7fSStefano Zampini         ierr = PetscMemcmp(adjncy,mat_graph->adjncy,xadj[nvtxs]*sizeof(PetscInt),&same_data);CHKERRQ(ierr);
9642d505d7fSStefano Zampini       }
96556282151SStefano Zampini     }
96656282151SStefano Zampini   }
96756282151SStefano Zampini   if (!same_data) {
968674ae819SStefano Zampini     /* free old CSR */
969674ae819SStefano Zampini     ierr = PCBDDCGraphResetCSR(mat_graph);CHKERRQ(ierr);
970674ae819SStefano Zampini     /* get CSR into graph structure */
971da1bb401SStefano Zampini     if (copymode == PETSC_COPY_VALUES) {
972854ce69bSBarry Smith       ierr = PetscMalloc1(nvtxs+1,&mat_graph->xadj);CHKERRQ(ierr);
973785e854fSJed Brown       ierr = PetscMalloc1(xadj[nvtxs],&mat_graph->adjncy);CHKERRQ(ierr);
974674ae819SStefano Zampini       ierr = PetscMemcpy(mat_graph->xadj,xadj,(nvtxs+1)*sizeof(PetscInt));CHKERRQ(ierr);
975674ae819SStefano Zampini       ierr = PetscMemcpy(mat_graph->adjncy,adjncy,xadj[nvtxs]*sizeof(PetscInt));CHKERRQ(ierr);
976a1dbd327SStefano Zampini       mat_graph->freecsr = PETSC_TRUE;
977da1bb401SStefano Zampini     } else if (copymode == PETSC_OWN_POINTER) {
9781a83f524SJed Brown       mat_graph->xadj    = (PetscInt*)xadj;
9791a83f524SJed Brown       mat_graph->adjncy  = (PetscInt*)adjncy;
980a1dbd327SStefano Zampini       mat_graph->freecsr = PETSC_TRUE;
981a1dbd327SStefano Zampini     } else if (copymode == PETSC_USE_POINTER) {
982a1dbd327SStefano Zampini       mat_graph->xadj    = (PetscInt*)xadj;
983a1dbd327SStefano Zampini       mat_graph->adjncy  = (PetscInt*)adjncy;
984a1dbd327SStefano Zampini       mat_graph->freecsr = PETSC_FALSE;
985e0fe2d75SToby Isaac     } else SETERRQ1(PETSC_COMM_SELF,PETSC_ERR_SUP,"Unsupported copy mode %D",copymode);
986575ad6abSStefano Zampini     mat_graph->nvtxs_csr = nvtxs;
98756282151SStefano Zampini     pcbddc->recompute_topography = PETSC_TRUE;
98856282151SStefano Zampini   }
98936e030ebSStefano Zampini   PetscFunctionReturn(0);
99036e030ebSStefano Zampini }
9911e6b0712SBarry Smith 
99236e030ebSStefano Zampini /*@
99354fffbccSStefano Zampini  PCBDDCSetLocalAdjacencyGraph - Set adjacency structure (CSR graph) of the local degrees of freedom.
99436e030ebSStefano Zampini 
99536e030ebSStefano Zampini    Not collective
99636e030ebSStefano Zampini 
99736e030ebSStefano Zampini    Input Parameters:
99854fffbccSStefano Zampini +  pc - the preconditioning context.
99954fffbccSStefano Zampini .  nvtxs - number of local vertices of the graph (i.e., the number of local dofs).
100054fffbccSStefano Zampini .  xadj, adjncy - the connectivity of the dofs in CSR format.
100154fffbccSStefano Zampini -  copymode - supported modes are PETSC_COPY_VALUES, PETSC_USE_POINTER or PETSC_OWN_POINTER.
100236e030ebSStefano Zampini 
100336e030ebSStefano Zampini    Level: intermediate
100436e030ebSStefano Zampini 
100554fffbccSStefano Zampini    Notes: A dof is considered connected with all local dofs if xadj[dof+1]-xadj[dof] == 1 and adjncy[xadj[dof]] is negative.
100636e030ebSStefano Zampini 
100728509bceSStefano Zampini .seealso: PCBDDC,PetscCopyMode
100836e030ebSStefano Zampini @*/
10091a83f524SJed Brown PetscErrorCode PCBDDCSetLocalAdjacencyGraph(PC pc,PetscInt nvtxs,const PetscInt xadj[],const PetscInt adjncy[], PetscCopyMode copymode)
101036e030ebSStefano Zampini {
1011575ad6abSStefano Zampini   void (*f)(void) = 0;
101236e030ebSStefano Zampini   PetscErrorCode ierr;
101336e030ebSStefano Zampini 
101436e030ebSStefano Zampini   PetscFunctionBegin;
101536e030ebSStefano Zampini   PetscValidHeaderSpecific(pc,PC_CLASSID,1);
10168687889aSStefano Zampini   if (nvtxs) {
1017674ae819SStefano Zampini     PetscValidIntPointer(xadj,3);
10181633d1f0SStefano Zampini     if (xadj[nvtxs]) PetscValidIntPointer(adjncy,4);
10198687889aSStefano Zampini   }
10201a83f524SJed Brown   ierr = PetscTryMethod(pc,"PCBDDCSetLocalAdjacencyGraph_C",(PC,PetscInt,const PetscInt[],const PetscInt[],PetscCopyMode),(pc,nvtxs,xadj,adjncy,copymode));CHKERRQ(ierr);
1021575ad6abSStefano Zampini   /* free arrays if PCBDDC is not the PC type */
1022575ad6abSStefano Zampini   ierr = PetscObjectQueryFunction((PetscObject)pc,"PCBDDCSetLocalAdjacencyGraph_C",&f);CHKERRQ(ierr);
1023575ad6abSStefano Zampini   if (!f && copymode == PETSC_OWN_POINTER) {
1024575ad6abSStefano Zampini     ierr = PetscFree(xadj);CHKERRQ(ierr);
1025575ad6abSStefano Zampini     ierr = PetscFree(adjncy);CHKERRQ(ierr);
1026da1bb401SStefano Zampini   }
102736e030ebSStefano Zampini   PetscFunctionReturn(0);
102836e030ebSStefano Zampini }
10291e6b0712SBarry Smith 
103063602bcaSStefano Zampini static PetscErrorCode PCBDDCSetDofsSplittingLocal_BDDC(PC pc,PetscInt n_is, IS ISForDofs[])
103163602bcaSStefano Zampini {
103263602bcaSStefano Zampini   PC_BDDC        *pcbddc = (PC_BDDC*)pc->data;
103363602bcaSStefano Zampini   PetscInt       i;
103456282151SStefano Zampini   PetscBool      isequal = PETSC_FALSE;
103563602bcaSStefano Zampini   PetscErrorCode ierr;
103663602bcaSStefano Zampini 
103763602bcaSStefano Zampini   PetscFunctionBegin;
103856282151SStefano Zampini   if (pcbddc->n_ISForDofsLocal == n_is) {
103956282151SStefano Zampini     for (i=0;i<n_is;i++) {
104056282151SStefano Zampini       PetscBool isequalt;
104156282151SStefano Zampini       ierr = ISEqual(ISForDofs[i],pcbddc->ISForDofsLocal[i],&isequalt);CHKERRQ(ierr);
104256282151SStefano Zampini       if (!isequalt) break;
104356282151SStefano Zampini     }
104456282151SStefano Zampini     if (i == n_is) isequal = PETSC_TRUE;
104556282151SStefano Zampini   }
104656282151SStefano Zampini   for (i=0;i<n_is;i++) {
104756282151SStefano Zampini     ierr = PetscObjectReference((PetscObject)ISForDofs[i]);CHKERRQ(ierr);
104856282151SStefano Zampini   }
104963602bcaSStefano Zampini   /* Destroy ISes if they were already set */
105063602bcaSStefano Zampini   for (i=0;i<pcbddc->n_ISForDofsLocal;i++) {
105163602bcaSStefano Zampini     ierr = ISDestroy(&pcbddc->ISForDofsLocal[i]);CHKERRQ(ierr);
105263602bcaSStefano Zampini   }
105363602bcaSStefano Zampini   ierr = PetscFree(pcbddc->ISForDofsLocal);CHKERRQ(ierr);
105463602bcaSStefano Zampini   /* last user setting takes precendence -> destroy any other customization */
105563602bcaSStefano Zampini   for (i=0;i<pcbddc->n_ISForDofs;i++) {
105663602bcaSStefano Zampini     ierr = ISDestroy(&pcbddc->ISForDofs[i]);CHKERRQ(ierr);
105763602bcaSStefano Zampini   }
105863602bcaSStefano Zampini   ierr = PetscFree(pcbddc->ISForDofs);CHKERRQ(ierr);
105963602bcaSStefano Zampini   pcbddc->n_ISForDofs = 0;
106063602bcaSStefano Zampini   /* allocate space then set */
1061d02579f5SStefano Zampini   if (n_is) {
1062d02579f5SStefano Zampini     ierr = PetscMalloc1(n_is,&pcbddc->ISForDofsLocal);CHKERRQ(ierr);
1063d02579f5SStefano Zampini   }
106463602bcaSStefano Zampini   for (i=0;i<n_is;i++) {
106563602bcaSStefano Zampini     pcbddc->ISForDofsLocal[i] = ISForDofs[i];
106663602bcaSStefano Zampini   }
106763602bcaSStefano Zampini   pcbddc->n_ISForDofsLocal = n_is;
106863602bcaSStefano Zampini   if (n_is) pcbddc->user_provided_isfordofs = PETSC_TRUE;
106956282151SStefano Zampini   if (!isequal) pcbddc->recompute_topography = PETSC_TRUE;
107063602bcaSStefano Zampini   PetscFunctionReturn(0);
107163602bcaSStefano Zampini }
107263602bcaSStefano Zampini 
107363602bcaSStefano Zampini /*@
107463602bcaSStefano Zampini  PCBDDCSetDofsSplittingLocal - Set index sets defining fields of the local subdomain matrix
107563602bcaSStefano Zampini 
107663602bcaSStefano Zampini    Collective
107763602bcaSStefano Zampini 
107863602bcaSStefano Zampini    Input Parameters:
107963602bcaSStefano Zampini +  pc - the preconditioning context
10800f202f7eSStefano Zampini .  n_is - number of index sets defining the fields
10810f202f7eSStefano Zampini -  ISForDofs - array of IS describing the fields in local ordering
108263602bcaSStefano Zampini 
108363602bcaSStefano Zampini    Level: intermediate
108463602bcaSStefano Zampini 
10850f202f7eSStefano Zampini    Notes:
10860f202f7eSStefano Zampini      n_is should be the same among processes. Not all nodes need to be listed: unlisted nodes will belong to the complement field.
108763602bcaSStefano Zampini 
108863602bcaSStefano Zampini .seealso: PCBDDC
108963602bcaSStefano Zampini @*/
109063602bcaSStefano Zampini PetscErrorCode PCBDDCSetDofsSplittingLocal(PC pc,PetscInt n_is, IS ISForDofs[])
109163602bcaSStefano Zampini {
109263602bcaSStefano Zampini   PetscInt       i;
109363602bcaSStefano Zampini   PetscErrorCode ierr;
109463602bcaSStefano Zampini 
109563602bcaSStefano Zampini   PetscFunctionBegin;
109663602bcaSStefano Zampini   PetscValidHeaderSpecific(pc,PC_CLASSID,1);
109763602bcaSStefano Zampini   PetscValidLogicalCollectiveInt(pc,n_is,2);
109863602bcaSStefano Zampini   for (i=0;i<n_is;i++) {
109963602bcaSStefano Zampini     PetscCheckSameComm(pc,1,ISForDofs[i],3);
110063602bcaSStefano Zampini     PetscValidHeaderSpecific(ISForDofs[i],IS_CLASSID,3);
110163602bcaSStefano Zampini   }
1102e71e7a71SStefano Zampini   ierr = PetscTryMethod(pc,"PCBDDCSetDofsSplittingLocal_C",(PC,PetscInt,IS[]),(pc,n_is,ISForDofs));CHKERRQ(ierr);
110363602bcaSStefano Zampini   PetscFunctionReturn(0);
110463602bcaSStefano Zampini }
110563602bcaSStefano Zampini 
11069c0446d6SStefano Zampini static PetscErrorCode PCBDDCSetDofsSplitting_BDDC(PC pc,PetscInt n_is, IS ISForDofs[])
11079c0446d6SStefano Zampini {
11089c0446d6SStefano Zampini   PC_BDDC        *pcbddc = (PC_BDDC*)pc->data;
11099c0446d6SStefano Zampini   PetscInt       i;
111056282151SStefano Zampini   PetscBool      isequal = PETSC_FALSE;
11119c0446d6SStefano Zampini   PetscErrorCode ierr;
11129c0446d6SStefano Zampini 
11139c0446d6SStefano Zampini   PetscFunctionBegin;
111456282151SStefano Zampini   if (pcbddc->n_ISForDofs == n_is) {
111556282151SStefano Zampini     for (i=0;i<n_is;i++) {
111656282151SStefano Zampini       PetscBool isequalt;
111756282151SStefano Zampini       ierr = ISEqual(ISForDofs[i],pcbddc->ISForDofs[i],&isequalt);CHKERRQ(ierr);
111856282151SStefano Zampini       if (!isequalt) break;
111956282151SStefano Zampini     }
112056282151SStefano Zampini     if (i == n_is) isequal = PETSC_TRUE;
112156282151SStefano Zampini   }
112256282151SStefano Zampini   for (i=0;i<n_is;i++) {
112356282151SStefano Zampini     ierr = PetscObjectReference((PetscObject)ISForDofs[i]);CHKERRQ(ierr);
112456282151SStefano Zampini   }
1125da1bb401SStefano Zampini   /* Destroy ISes if they were already set */
11269c0446d6SStefano Zampini   for (i=0;i<pcbddc->n_ISForDofs;i++) {
11279c0446d6SStefano Zampini     ierr = ISDestroy(&pcbddc->ISForDofs[i]);CHKERRQ(ierr);
11289c0446d6SStefano Zampini   }
1129d11ae9bbSstefano_zampini   ierr = PetscFree(pcbddc->ISForDofs);CHKERRQ(ierr);
113063602bcaSStefano Zampini   /* last user setting takes precendence -> destroy any other customization */
113163602bcaSStefano Zampini   for (i=0;i<pcbddc->n_ISForDofsLocal;i++) {
113263602bcaSStefano Zampini     ierr = ISDestroy(&pcbddc->ISForDofsLocal[i]);CHKERRQ(ierr);
113363602bcaSStefano Zampini   }
113463602bcaSStefano Zampini   ierr = PetscFree(pcbddc->ISForDofsLocal);CHKERRQ(ierr);
113563602bcaSStefano Zampini   pcbddc->n_ISForDofsLocal = 0;
1136da1bb401SStefano Zampini   /* allocate space then set */
1137d02579f5SStefano Zampini   if (n_is) {
1138785e854fSJed Brown     ierr = PetscMalloc1(n_is,&pcbddc->ISForDofs);CHKERRQ(ierr);
1139d02579f5SStefano Zampini   }
11409c0446d6SStefano Zampini   for (i=0;i<n_is;i++) {
1141da1bb401SStefano Zampini     pcbddc->ISForDofs[i] = ISForDofs[i];
11429c0446d6SStefano Zampini   }
11439c0446d6SStefano Zampini   pcbddc->n_ISForDofs = n_is;
114463602bcaSStefano Zampini   if (n_is) pcbddc->user_provided_isfordofs = PETSC_TRUE;
114556282151SStefano Zampini   if (!isequal) pcbddc->recompute_topography = PETSC_TRUE;
11469c0446d6SStefano Zampini   PetscFunctionReturn(0);
11479c0446d6SStefano Zampini }
11481e6b0712SBarry Smith 
11499c0446d6SStefano Zampini /*@
115063602bcaSStefano Zampini  PCBDDCSetDofsSplitting - Set index sets defining fields of the global matrix
11519c0446d6SStefano Zampini 
115263602bcaSStefano Zampini    Collective
11539c0446d6SStefano Zampini 
11549c0446d6SStefano Zampini    Input Parameters:
11559c0446d6SStefano Zampini +  pc - the preconditioning context
11560f202f7eSStefano Zampini .  n_is - number of index sets defining the fields
11570f202f7eSStefano Zampini -  ISForDofs - array of IS describing the fields in global ordering
11589c0446d6SStefano Zampini 
11599c0446d6SStefano Zampini    Level: intermediate
11609c0446d6SStefano Zampini 
11610f202f7eSStefano Zampini    Notes:
11620f202f7eSStefano Zampini      Any process can list any global node. Not all nodes need to be listed: unlisted nodes will belong to the complement field.
11639c0446d6SStefano Zampini 
11649c0446d6SStefano Zampini .seealso: PCBDDC
11659c0446d6SStefano Zampini @*/
11669c0446d6SStefano Zampini PetscErrorCode PCBDDCSetDofsSplitting(PC pc,PetscInt n_is, IS ISForDofs[])
11679c0446d6SStefano Zampini {
11682b510759SStefano Zampini   PetscInt       i;
11699c0446d6SStefano Zampini   PetscErrorCode ierr;
11709c0446d6SStefano Zampini 
11719c0446d6SStefano Zampini   PetscFunctionBegin;
11729c0446d6SStefano Zampini   PetscValidHeaderSpecific(pc,PC_CLASSID,1);
117363602bcaSStefano Zampini   PetscValidLogicalCollectiveInt(pc,n_is,2);
11742b510759SStefano Zampini   for (i=0;i<n_is;i++) {
117563602bcaSStefano Zampini     PetscValidHeaderSpecific(ISForDofs[i],IS_CLASSID,3);
1176a011d5a7Sstefano_zampini     PetscCheckSameComm(pc,1,ISForDofs[i],3);
11772b510759SStefano Zampini   }
11789c0446d6SStefano Zampini   ierr = PetscTryMethod(pc,"PCBDDCSetDofsSplitting_C",(PC,PetscInt,IS[]),(pc,n_is,ISForDofs));CHKERRQ(ierr);
11799c0446d6SStefano Zampini   PetscFunctionReturn(0);
11809c0446d6SStefano Zampini }
1181906d46d4SStefano Zampini 
1182534831adSStefano Zampini /*
1183534831adSStefano Zampini    PCPreSolve_BDDC - Changes the right hand side and (if necessary) the initial
1184534831adSStefano Zampini                      guess if a transformation of basis approach has been selected.
11859c0446d6SStefano Zampini 
1186534831adSStefano Zampini    Input Parameter:
1187534831adSStefano Zampini +  pc - the preconditioner contex
1188534831adSStefano Zampini 
1189534831adSStefano Zampini    Application Interface Routine: PCPreSolve()
1190534831adSStefano Zampini 
1191534831adSStefano Zampini    Notes:
1192534831adSStefano Zampini      The interface routine PCPreSolve() is not usually called directly by
1193534831adSStefano Zampini    the user, but instead is called by KSPSolve().
1194534831adSStefano Zampini */
1195534831adSStefano Zampini static PetscErrorCode PCPreSolve_BDDC(PC pc, KSP ksp, Vec rhs, Vec x)
1196534831adSStefano Zampini {
1197534831adSStefano Zampini   PetscErrorCode ierr;
1198534831adSStefano Zampini   PC_BDDC        *pcbddc = (PC_BDDC*)pc->data;
1199534831adSStefano Zampini   PC_IS          *pcis = (PC_IS*)(pc->data);
12003972b0daSStefano Zampini   Vec            used_vec;
120127b6a85dSStefano Zampini   PetscBool      save_rhs = PETSC_TRUE, benign_correction_computed;
1202534831adSStefano Zampini 
1203534831adSStefano Zampini   PetscFunctionBegin;
12041f4df5f7SStefano Zampini   /* if we are working with CG, one dirichlet solve can be avoided during Krylov iterations */
120585c4d303SStefano Zampini   if (ksp) {
12061f4df5f7SStefano Zampini     PetscBool iscg, isgroppcg, ispipecg, ispipecgrr;
120785c4d303SStefano Zampini     ierr = PetscObjectTypeCompare((PetscObject)ksp,KSPCG,&iscg);CHKERRQ(ierr);
120827b6a85dSStefano Zampini     ierr = PetscObjectTypeCompare((PetscObject)ksp,KSPGROPPCG,&isgroppcg);CHKERRQ(ierr);
120927b6a85dSStefano Zampini     ierr = PetscObjectTypeCompare((PetscObject)ksp,KSPPIPECG,&ispipecg);CHKERRQ(ierr);
1210f94e96cbSStefano Zampini     ierr = PetscObjectTypeCompare((PetscObject)ksp,KSPPIPECGRR,&ispipecgrr);CHKERRQ(ierr);
12111f4df5f7SStefano Zampini     if (pcbddc->benign_apply_coarse_only || pcbddc->switch_static || (!iscg && !isgroppcg && !ispipecg && !ispipecgrr)) {
121285c4d303SStefano Zampini       ierr = PCBDDCSetUseExactDirichlet(pc,PETSC_FALSE);CHKERRQ(ierr);
121385c4d303SStefano Zampini     }
121485c4d303SStefano Zampini   }
1215fc17d649SStefano Zampini   if (pcbddc->benign_apply_coarse_only || pcbddc->switch_static) {
1216fc17d649SStefano Zampini     ierr = PCBDDCSetUseExactDirichlet(pc,PETSC_FALSE);CHKERRQ(ierr);
1217fc17d649SStefano Zampini   }
12181f4df5f7SStefano Zampini 
121985c4d303SStefano Zampini   /* Creates parallel work vectors used in presolve */
122062a6ff1dSStefano Zampini   if (!pcbddc->original_rhs) {
122162a6ff1dSStefano Zampini     ierr = VecDuplicate(pcis->vec1_global,&pcbddc->original_rhs);CHKERRQ(ierr);
122262a6ff1dSStefano Zampini   }
122362a6ff1dSStefano Zampini   if (!pcbddc->temp_solution) {
122462a6ff1dSStefano Zampini     ierr = VecDuplicate(pcis->vec1_global,&pcbddc->temp_solution);CHKERRQ(ierr);
122562a6ff1dSStefano Zampini   }
12268d00608fSStefano Zampini 
122727b6a85dSStefano Zampini   pcbddc->temp_solution_used = PETSC_FALSE;
12283972b0daSStefano Zampini   if (x) {
12293972b0daSStefano Zampini     ierr = PetscObjectReference((PetscObject)x);CHKERRQ(ierr);
12303972b0daSStefano Zampini     used_vec = x;
12318d00608fSStefano Zampini   } else { /* it can only happen when calling PCBDDCMatFETIDPGetRHS */
12323972b0daSStefano Zampini     ierr = PetscObjectReference((PetscObject)pcbddc->temp_solution);CHKERRQ(ierr);
12333972b0daSStefano Zampini     used_vec = pcbddc->temp_solution;
12343972b0daSStefano Zampini     ierr = VecSet(used_vec,0.0);CHKERRQ(ierr);
123527b6a85dSStefano Zampini     pcbddc->temp_solution_used = PETSC_TRUE;
1236266e20e9SStefano Zampini     ierr = VecCopy(rhs,pcbddc->original_rhs);CHKERRQ(ierr);
1237266e20e9SStefano Zampini     save_rhs = PETSC_FALSE;
1238266e20e9SStefano Zampini     pcbddc->eliminate_dirdofs = PETSC_TRUE;
12393972b0daSStefano Zampini   }
12408efcfb23SStefano Zampini 
12418efcfb23SStefano Zampini   /* hack into ksp data structure since PCPreSolve comes earlier than setting to zero the guess in src/ksp/ksp/interface/itfunc.c */
12423972b0daSStefano Zampini   if (ksp) {
1243a0cb1b98SStefano Zampini     /* store the flag for the initial guess since it will be restored back during PCPostSolve_BDDC */
12448efcfb23SStefano Zampini     ierr = KSPGetInitialGuessNonzero(ksp,&pcbddc->ksp_guess_nonzero);CHKERRQ(ierr);
12458efcfb23SStefano Zampini     if (!pcbddc->ksp_guess_nonzero) {
12463972b0daSStefano Zampini       ierr = VecSet(used_vec,0.0);CHKERRQ(ierr);
12473972b0daSStefano Zampini     }
12483972b0daSStefano Zampini   }
12493308cffdSStefano Zampini 
12508d00608fSStefano Zampini   pcbddc->rhs_change = PETSC_FALSE;
12513972b0daSStefano Zampini   /* Take into account zeroed rows -> change rhs and store solution removed */
125270c64980SStefano Zampini   if (rhs && pcbddc->eliminate_dirdofs) {
12533975b054SStefano Zampini     IS dirIS = NULL;
12543975b054SStefano Zampini 
1255a07ea27aSStefano Zampini     /* DirichletBoundariesLocal may not be consistent among neighbours; gets a dirichlet dofs IS from graph (may be cached) */
12563975b054SStefano Zampini     ierr = PCBDDCGraphGetDirichletDofs(pcbddc->mat_graph,&dirIS);CHKERRQ(ierr);
12573975b054SStefano Zampini     if (dirIS) {
1258906d46d4SStefano Zampini       Mat_IS            *matis = (Mat_IS*)pc->pmat->data;
1259785d1243SStefano Zampini       PetscInt          dirsize,i,*is_indices;
12602b095fd8SStefano Zampini       PetscScalar       *array_x;
12612b095fd8SStefano Zampini       const PetscScalar *array_diagonal;
1262785d1243SStefano Zampini 
12633972b0daSStefano Zampini       ierr = MatGetDiagonal(pc->pmat,pcis->vec1_global);CHKERRQ(ierr);
12643972b0daSStefano Zampini       ierr = VecPointwiseDivide(pcis->vec1_global,rhs,pcis->vec1_global);CHKERRQ(ierr);
1265e176bc59SStefano Zampini       ierr = VecScatterBegin(matis->rctx,pcis->vec1_global,pcis->vec2_N,INSERT_VALUES,SCATTER_FORWARD);CHKERRQ(ierr);
1266e176bc59SStefano Zampini       ierr = VecScatterEnd(matis->rctx,pcis->vec1_global,pcis->vec2_N,INSERT_VALUES,SCATTER_FORWARD);CHKERRQ(ierr);
1267e176bc59SStefano Zampini       ierr = VecScatterBegin(matis->rctx,used_vec,pcis->vec1_N,INSERT_VALUES,SCATTER_FORWARD);CHKERRQ(ierr);
1268e176bc59SStefano Zampini       ierr = VecScatterEnd(matis->rctx,used_vec,pcis->vec1_N,INSERT_VALUES,SCATTER_FORWARD);CHKERRQ(ierr);
126982ba6b80SStefano Zampini       ierr = ISGetLocalSize(dirIS,&dirsize);CHKERRQ(ierr);
12703972b0daSStefano Zampini       ierr = VecGetArray(pcis->vec1_N,&array_x);CHKERRQ(ierr);
12712b095fd8SStefano Zampini       ierr = VecGetArrayRead(pcis->vec2_N,&array_diagonal);CHKERRQ(ierr);
12723972b0daSStefano Zampini       ierr = ISGetIndices(dirIS,(const PetscInt**)&is_indices);CHKERRQ(ierr);
12732fa5cd67SKarl Rupp       for (i=0; i<dirsize; i++) array_x[is_indices[i]] = array_diagonal[is_indices[i]];
12743972b0daSStefano Zampini       ierr = ISRestoreIndices(dirIS,(const PetscInt**)&is_indices);CHKERRQ(ierr);
12752b095fd8SStefano Zampini       ierr = VecRestoreArrayRead(pcis->vec2_N,&array_diagonal);CHKERRQ(ierr);
12763972b0daSStefano Zampini       ierr = VecRestoreArray(pcis->vec1_N,&array_x);CHKERRQ(ierr);
1277e176bc59SStefano Zampini       ierr = VecScatterBegin(matis->rctx,pcis->vec1_N,used_vec,INSERT_VALUES,SCATTER_REVERSE);CHKERRQ(ierr);
1278e176bc59SStefano Zampini       ierr = VecScatterEnd(matis->rctx,pcis->vec1_N,used_vec,INSERT_VALUES,SCATTER_REVERSE);CHKERRQ(ierr);
12798d00608fSStefano Zampini       pcbddc->rhs_change = PETSC_TRUE;
12801b968477SStefano Zampini       ierr = ISDestroy(&dirIS);CHKERRQ(ierr);
12818efcfb23SStefano Zampini     }
1282a07ea27aSStefano Zampini   }
1283b76ba322SStefano Zampini 
12848efcfb23SStefano Zampini   /* remove the computed solution or the initial guess from the rhs */
12858d00608fSStefano Zampini   if (pcbddc->rhs_change || (ksp && pcbddc->ksp_guess_nonzero) ) {
128627b6a85dSStefano Zampini     /* save the original rhs */
128727b6a85dSStefano Zampini     if (save_rhs) {
128827b6a85dSStefano Zampini       ierr = VecSwap(rhs,pcbddc->original_rhs);CHKERRQ(ierr);
128927b6a85dSStefano Zampini       save_rhs = PETSC_FALSE;
12908d00608fSStefano Zampini     }
12918d00608fSStefano Zampini     pcbddc->rhs_change = PETSC_TRUE;
12923972b0daSStefano Zampini     ierr = VecScale(used_vec,-1.0);CHKERRQ(ierr);
129327b6a85dSStefano Zampini     ierr = MatMultAdd(pc->mat,used_vec,pcbddc->original_rhs,rhs);CHKERRQ(ierr);
12943972b0daSStefano Zampini     ierr = VecScale(used_vec,-1.0);CHKERRQ(ierr);
12958efcfb23SStefano Zampini     ierr = VecCopy(used_vec,pcbddc->temp_solution);CHKERRQ(ierr);
129627b6a85dSStefano Zampini     pcbddc->temp_solution_used = PETSC_TRUE;
12977acc28cbSStefano Zampini     if (ksp) {
12987acc28cbSStefano Zampini       ierr = KSPSetInitialGuessNonzero(ksp,PETSC_FALSE);CHKERRQ(ierr);
12997acc28cbSStefano Zampini     }
13003308cffdSStefano Zampini   }
13018efcfb23SStefano Zampini   ierr = VecDestroy(&used_vec);CHKERRQ(ierr);
1302b76ba322SStefano Zampini 
1303fc17d649SStefano Zampini   /* compute initial vector in benign space if needed
130427b6a85dSStefano Zampini      and remove non-benign solution from the rhs */
130527b6a85dSStefano Zampini   benign_correction_computed = PETSC_FALSE;
130627b6a85dSStefano Zampini   if (rhs && pcbddc->benign_compute_correction && pcbddc->benign_have_null) {
13071f4df5f7SStefano Zampini     /* compute u^*_h using ideas similar to those in Xuemin Tu's PhD thesis (see Section 4.8.1)
13081f4df5f7SStefano Zampini        Recursively apply BDDC in the multilevel case */
13090369aaf7SStefano Zampini     if (!pcbddc->benign_vec) {
13100369aaf7SStefano Zampini       ierr = VecDuplicate(rhs,&pcbddc->benign_vec);CHKERRQ(ierr);
13110369aaf7SStefano Zampini     }
13124fee134fSStefano Zampini     pcbddc->benign_apply_coarse_only = PETSC_TRUE;
131327b6a85dSStefano Zampini     if (!pcbddc->benign_skip_correction) {
13141dd7afcfSStefano Zampini       ierr = PCApply_BDDC(pc,rhs,pcbddc->benign_vec);CHKERRQ(ierr);
13153bca92a6SStefano Zampini       benign_correction_computed = PETSC_TRUE;
13161f4df5f7SStefano Zampini       if (pcbddc->temp_solution_used) {
13171f4df5f7SStefano Zampini         ierr = VecAXPY(pcbddc->temp_solution,1.0,pcbddc->benign_vec);CHKERRQ(ierr);
13181f4df5f7SStefano Zampini       }
13191f4df5f7SStefano Zampini       ierr = VecScale(pcbddc->benign_vec,-1.0);CHKERRQ(ierr);
132027b6a85dSStefano Zampini       /* store the original rhs if not done earlier */
132127b6a85dSStefano Zampini       if (save_rhs) {
132227b6a85dSStefano Zampini         ierr = VecSwap(rhs,pcbddc->original_rhs);CHKERRQ(ierr);
132392e3dcfbSStefano Zampini       }
132427b6a85dSStefano Zampini       if (pcbddc->rhs_change) {
13250369aaf7SStefano Zampini         ierr = MatMultAdd(pc->mat,pcbddc->benign_vec,rhs,rhs);CHKERRQ(ierr);
132627b6a85dSStefano Zampini       } else {
132727b6a85dSStefano Zampini         ierr = MatMultAdd(pc->mat,pcbddc->benign_vec,pcbddc->original_rhs,rhs);CHKERRQ(ierr);
132827b6a85dSStefano Zampini       }
13290369aaf7SStefano Zampini       pcbddc->rhs_change = PETSC_TRUE;
133027b6a85dSStefano Zampini     }
133127b6a85dSStefano Zampini     pcbddc->benign_apply_coarse_only = PETSC_FALSE;
13320369aaf7SStefano Zampini   }
13332d4c4fecSStefano Zampini 
13342d4c4fecSStefano Zampini   /* dbg output */
1335a198735bSStefano Zampini   if (pcbddc->dbg_flag && benign_correction_computed) {
13361f4df5f7SStefano Zampini     Vec v;
13371f4df5f7SStefano Zampini     ierr = VecDuplicate(pcis->vec1_global,&v);CHKERRQ(ierr);
13381f4df5f7SStefano Zampini     ierr = MatMultTranspose(pcbddc->ChangeOfBasisMatrix,rhs,v);CHKERRQ(ierr);
13391f4df5f7SStefano Zampini     ierr = PCBDDCBenignGetOrSetP0(pc,v,PETSC_TRUE);CHKERRQ(ierr);
1340e0fe2d75SToby Isaac     ierr = PetscViewerASCIIPrintf(pcbddc->dbg_viewer,"LEVEL %D: is the correction benign?\n",pcbddc->current_level);CHKERRQ(ierr);
13412d4c4fecSStefano Zampini     ierr = PetscScalarView(pcbddc->benign_n,pcbddc->benign_p0,PETSC_VIEWER_STDOUT_(PetscObjectComm((PetscObject)pc)));CHKERRQ(ierr);
13421f4df5f7SStefano Zampini     ierr = VecDestroy(&v);CHKERRQ(ierr);
13431f4df5f7SStefano Zampini   }
13440369aaf7SStefano Zampini 
13450369aaf7SStefano Zampini   /* set initial guess if using PCG */
13468ae0ca82SStefano Zampini   pcbddc->exact_dirichlet_trick_app = PETSC_FALSE;
13470369aaf7SStefano Zampini   if (x && pcbddc->use_exact_dirichlet_trick) {
13480369aaf7SStefano Zampini     ierr = VecSet(x,0.0);CHKERRQ(ierr);
13491dd7afcfSStefano Zampini     if (pcbddc->ChangeOfBasisMatrix && pcbddc->change_interior) {
135027b6a85dSStefano Zampini       if (benign_correction_computed) { /* we have already saved the changed rhs */
13511dd7afcfSStefano Zampini         ierr = VecLockPop(pcis->vec1_global);CHKERRQ(ierr);
13521dd7afcfSStefano Zampini       } else {
13531dd7afcfSStefano Zampini         ierr = MatMultTranspose(pcbddc->ChangeOfBasisMatrix,rhs,pcis->vec1_global);CHKERRQ(ierr);
13541dd7afcfSStefano Zampini       }
13551dd7afcfSStefano Zampini       ierr = VecScatterBegin(pcis->global_to_D,pcis->vec1_global,pcis->vec1_D,INSERT_VALUES,SCATTER_FORWARD);CHKERRQ(ierr);
13561dd7afcfSStefano Zampini       ierr = VecScatterEnd(pcis->global_to_D,pcis->vec1_global,pcis->vec1_D,INSERT_VALUES,SCATTER_FORWARD);CHKERRQ(ierr);
13571dd7afcfSStefano Zampini     } else {
13580369aaf7SStefano Zampini       ierr = VecScatterBegin(pcis->global_to_D,rhs,pcis->vec1_D,INSERT_VALUES,SCATTER_FORWARD);CHKERRQ(ierr);
13590369aaf7SStefano Zampini       ierr = VecScatterEnd(pcis->global_to_D,rhs,pcis->vec1_D,INSERT_VALUES,SCATTER_FORWARD);CHKERRQ(ierr);
13601dd7afcfSStefano Zampini     }
13610369aaf7SStefano Zampini     ierr = KSPSolve(pcbddc->ksp_D,pcis->vec1_D,pcis->vec2_D);CHKERRQ(ierr);
13621dd7afcfSStefano Zampini     if (pcbddc->ChangeOfBasisMatrix && pcbddc->change_interior) {
13631dd7afcfSStefano Zampini       ierr = VecSet(pcis->vec1_global,0.);CHKERRQ(ierr);
13641dd7afcfSStefano Zampini       ierr = VecScatterBegin(pcis->global_to_D,pcis->vec2_D,pcis->vec1_global,INSERT_VALUES,SCATTER_REVERSE);CHKERRQ(ierr);
13651dd7afcfSStefano Zampini       ierr = VecScatterEnd(pcis->global_to_D,pcis->vec2_D,pcis->vec1_global,INSERT_VALUES,SCATTER_REVERSE);CHKERRQ(ierr);
13661dd7afcfSStefano Zampini       ierr = MatMult(pcbddc->ChangeOfBasisMatrix,pcis->vec1_global,x);CHKERRQ(ierr);
13671dd7afcfSStefano Zampini     } else {
13680369aaf7SStefano Zampini       ierr = VecScatterBegin(pcis->global_to_D,pcis->vec2_D,x,INSERT_VALUES,SCATTER_REVERSE);CHKERRQ(ierr);
13690369aaf7SStefano Zampini       ierr = VecScatterEnd(pcis->global_to_D,pcis->vec2_D,x,INSERT_VALUES,SCATTER_REVERSE);CHKERRQ(ierr);
13701dd7afcfSStefano Zampini     }
13710369aaf7SStefano Zampini     if (ksp) {
13720369aaf7SStefano Zampini       ierr = KSPSetInitialGuessNonzero(ksp,PETSC_TRUE);CHKERRQ(ierr);
13730369aaf7SStefano Zampini     }
13748ae0ca82SStefano Zampini     pcbddc->exact_dirichlet_trick_app = PETSC_TRUE;
1375266e20e9SStefano Zampini   } else if (pcbddc->ChangeOfBasisMatrix && pcbddc->change_interior && benign_correction_computed && pcbddc->use_exact_dirichlet_trick) {
1376266e20e9SStefano Zampini     ierr = VecLockPop(pcis->vec1_global);CHKERRQ(ierr);
13770369aaf7SStefano Zampini   }
1378534831adSStefano Zampini   PetscFunctionReturn(0);
1379534831adSStefano Zampini }
1380906d46d4SStefano Zampini 
1381534831adSStefano Zampini /*
1382534831adSStefano Zampini    PCPostSolve_BDDC - Changes the computed solution if a transformation of basis
1383534831adSStefano Zampini                      approach has been selected. Also, restores rhs to its original state.
1384534831adSStefano Zampini 
1385534831adSStefano Zampini    Input Parameter:
1386534831adSStefano Zampini +  pc - the preconditioner contex
1387534831adSStefano Zampini 
1388534831adSStefano Zampini    Application Interface Routine: PCPostSolve()
1389534831adSStefano Zampini 
1390534831adSStefano Zampini    Notes:
1391534831adSStefano Zampini      The interface routine PCPostSolve() is not usually called directly by
1392534831adSStefano Zampini      the user, but instead is called by KSPSolve().
1393534831adSStefano Zampini */
1394534831adSStefano Zampini static PetscErrorCode PCPostSolve_BDDC(PC pc, KSP ksp, Vec rhs, Vec x)
1395534831adSStefano Zampini {
1396534831adSStefano Zampini   PetscErrorCode ierr;
1397534831adSStefano Zampini   PC_BDDC        *pcbddc = (PC_BDDC*)pc->data;
1398534831adSStefano Zampini 
1399534831adSStefano Zampini   PetscFunctionBegin;
14003972b0daSStefano Zampini   /* add solution removed in presolve */
14016bcfc461SStefano Zampini   if (x && pcbddc->rhs_change) {
140227b6a85dSStefano Zampini     if (pcbddc->temp_solution_used) {
14033425bc38SStefano Zampini       ierr = VecAXPY(x,1.0,pcbddc->temp_solution);CHKERRQ(ierr);
1404af140850Sstefano_zampini     } else if (pcbddc->benign_compute_correction && pcbddc->benign_vec) {
140527b6a85dSStefano Zampini       ierr = VecAXPY(x,-1.0,pcbddc->benign_vec);CHKERRQ(ierr);
14063425bc38SStefano Zampini     }
1407af140850Sstefano_zampini     /* restore to original state (not for FETI-DP) */
1408af140850Sstefano_zampini     if (ksp) pcbddc->temp_solution_used = PETSC_FALSE;
140927b6a85dSStefano Zampini   }
141027b6a85dSStefano Zampini 
1411266e20e9SStefano Zampini   /* restore rhs to its original state (not needed for FETI-DP) */
14128d00608fSStefano Zampini   if (rhs && pcbddc->rhs_change) {
141327b6a85dSStefano Zampini     ierr = VecSwap(rhs,pcbddc->original_rhs);CHKERRQ(ierr);
14148d00608fSStefano Zampini     pcbddc->rhs_change = PETSC_FALSE;
1415af140850Sstefano_zampini   }
14168efcfb23SStefano Zampini   /* restore ksp guess state */
14178efcfb23SStefano Zampini   if (ksp) {
14188efcfb23SStefano Zampini     ierr = KSPSetInitialGuessNonzero(ksp,pcbddc->ksp_guess_nonzero);CHKERRQ(ierr);
14198ae0ca82SStefano Zampini     /* reset flag for exact dirichlet trick */
14208ae0ca82SStefano Zampini     pcbddc->exact_dirichlet_trick_app = PETSC_FALSE;
1421af140850Sstefano_zampini   }
1422534831adSStefano Zampini   PetscFunctionReturn(0);
1423534831adSStefano Zampini }
1424af140850Sstefano_zampini 
14250c7d97c5SJed Brown /*
14260c7d97c5SJed Brown    PCSetUp_BDDC - Prepares for the use of the BDDC preconditioner
14270c7d97c5SJed Brown                   by setting data structures and options.
14280c7d97c5SJed Brown 
14290c7d97c5SJed Brown    Input Parameter:
143053cdbc3dSStefano Zampini +  pc - the preconditioner context
14310c7d97c5SJed Brown 
14320c7d97c5SJed Brown    Application Interface Routine: PCSetUp()
14330c7d97c5SJed Brown 
14340c7d97c5SJed Brown    Notes:
14350c7d97c5SJed Brown      The interface routine PCSetUp() is not usually called directly by
14360c7d97c5SJed Brown      the user, but instead is called by PCApply() if necessary.
14370c7d97c5SJed Brown */
143853cdbc3dSStefano Zampini PetscErrorCode PCSetUp_BDDC(PC pc)
14390c7d97c5SJed Brown {
14400c7d97c5SJed Brown   PC_BDDC*        pcbddc = (PC_BDDC*)pc->data;
1441c703fcc7SStefano Zampini   PCBDDCSubSchurs sub_schurs;
14425e8657edSStefano Zampini   Mat_IS*         matis;
144308122e43SStefano Zampini   MatNullSpace    nearnullspace;
144435509ce9Sstefano_zampini   Mat             lA;
144535509ce9Sstefano_zampini   IS              lP,zerodiag = NULL;
144691e8d312SStefano Zampini   PetscInt        nrows,ncols;
1447c703fcc7SStefano Zampini   PetscBool       computesubschurs;
14488de1fae6SStefano Zampini   PetscBool       computeconstraintsmatrix;
14495e8657edSStefano Zampini   PetscBool       new_nearnullspace_provided,ismatis;
1450c703fcc7SStefano Zampini   PetscErrorCode  ierr;
14510c7d97c5SJed Brown 
14520c7d97c5SJed Brown   PetscFunctionBegin;
14535e8657edSStefano Zampini   ierr = PetscObjectTypeCompare((PetscObject)pc->pmat,MATIS,&ismatis);CHKERRQ(ierr);
14546c4ed002SBarry Smith   if (!ismatis) SETERRQ(PetscObjectComm((PetscObject)pc),PETSC_ERR_ARG_WRONG,"PCBDDC preconditioner requires matrix of type MATIS");
145591e8d312SStefano Zampini   ierr = MatGetSize(pc->pmat,&nrows,&ncols);CHKERRQ(ierr);
14566c4ed002SBarry Smith   if (nrows != ncols) SETERRQ(PetscObjectComm((PetscObject)pc),PETSC_ERR_SUP,"PCBDDC preconditioner requires a square preconditioning matrix");
14575e8657edSStefano Zampini   matis = (Mat_IS*)pc->pmat->data;
1458f4ddd8eeSStefano Zampini   /* the following lines of code should be replaced by a better logic between PCIS, PCNN, PCBDDC and other future nonoverlapping preconditioners */
14593b03a366Sstefano_zampini   /* For BDDC we need to define a local "Neumann" problem different to that defined in PCISSetup
146071582508SStefano Zampini      Also, BDDC builds its own KSP for the Dirichlet problem */
1461a485753aSstefano_zampini   if (!pc->setupcalled || pc->flag == DIFFERENT_NONZERO_PATTERN) pcbddc->recompute_topography = PETSC_TRUE;
1462c83e1ba7SStefano Zampini   if (pcbddc->recompute_topography) {
1463c83e1ba7SStefano Zampini     pcbddc->graphanalyzed    = PETSC_FALSE;
1464c83e1ba7SStefano Zampini     computeconstraintsmatrix = PETSC_TRUE;
1465c83e1ba7SStefano Zampini   } else {
14668de1fae6SStefano Zampini     computeconstraintsmatrix = PETSC_FALSE;
1467c83e1ba7SStefano Zampini   }
1468b087196eSStefano Zampini 
1469b087196eSStefano Zampini   /* check parameters' compatibility */
1470b7ab4a40SStefano Zampini   if (!pcbddc->use_deluxe_scaling) pcbddc->deluxe_zerorows = PETSC_FALSE;
14719d54b7f4SStefano Zampini   pcbddc->adaptive_selection = (PetscBool)(pcbddc->adaptive_threshold > 0.0);
1472bf3a8328SStefano Zampini   pcbddc->adaptive_userdefined = (PetscBool)(pcbddc->adaptive_selection && pcbddc->adaptive_userdefined);
1473862806e4SStefano Zampini   if (pcbddc->adaptive_selection) pcbddc->use_faces = PETSC_TRUE;
1474862806e4SStefano Zampini 
14755a95e1ceSStefano Zampini   computesubschurs = (PetscBool)(pcbddc->adaptive_selection || pcbddc->use_deluxe_scaling);
147616909a7fSStefano Zampini   if (pcbddc->switch_static) {
147716909a7fSStefano Zampini     PetscBool ismatis;
147816909a7fSStefano Zampini     ierr = PetscObjectTypeCompare((PetscObject)pc->mat,MATIS,&ismatis);CHKERRQ(ierr);
147916909a7fSStefano Zampini     if (!ismatis) SETERRQ(PetscObjectComm((PetscObject)pc),PETSC_ERR_SUP,"When the static switch is one, the iteration matrix should be of type MATIS");
148016909a7fSStefano Zampini   }
148116909a7fSStefano Zampini 
148271582508SStefano Zampini   /* activate all connected components if the netflux has been requested */
1483bb05f991SStefano Zampini   if (pcbddc->compute_nonetflux) {
1484bb05f991SStefano Zampini     pcbddc->use_vertices = PETSC_TRUE;
1485bb05f991SStefano Zampini     pcbddc->use_edges = PETSC_TRUE;
1486bb05f991SStefano Zampini     pcbddc->use_faces = PETSC_TRUE;
1487bb05f991SStefano Zampini   }
1488bb05f991SStefano Zampini 
1489f4ddd8eeSStefano Zampini   /* Get stdout for dbg */
149070cf5478SStefano Zampini   if (pcbddc->dbg_flag) {
149170cf5478SStefano Zampini     if (!pcbddc->dbg_viewer) {
149258a03d70SStefano Zampini       pcbddc->dbg_viewer = PETSC_VIEWER_STDOUT_(PetscObjectComm((PetscObject)pc));
14931575c14dSBarry Smith       ierr = PetscViewerASCIIPushSynchronized(pcbddc->dbg_viewer);CHKERRQ(ierr);
1494f4ddd8eeSStefano Zampini     }
149558a03d70SStefano Zampini     ierr = PetscViewerASCIIAddTab(pcbddc->dbg_viewer,2*pcbddc->current_level);CHKERRQ(ierr);
1496f4ddd8eeSStefano Zampini   }
1497f4ddd8eeSStefano Zampini 
1498c703fcc7SStefano Zampini   /* process topology information */
149971582508SStefano Zampini   if (pcbddc->recompute_topography) {
150071582508SStefano Zampini     ierr = PCBDDCComputeLocalTopologyInfo(pc);CHKERRQ(ierr);
1501c83e1ba7SStefano Zampini     /* detect local disconnected subdomains if requested (use matis->A) */
1502c83e1ba7SStefano Zampini     if (pcbddc->detect_disconnected) {
1503c80a6c00SStefano Zampini       IS       primalv;
1504c83e1ba7SStefano Zampini       PetscInt i;
1505c80a6c00SStefano Zampini 
1506c83e1ba7SStefano Zampini       for (i=0;i<pcbddc->n_local_subs;i++) {
1507c83e1ba7SStefano Zampini         ierr = ISDestroy(&pcbddc->local_subs[i]);CHKERRQ(ierr);
1508c83e1ba7SStefano Zampini       }
1509c83e1ba7SStefano Zampini       ierr = PetscFree(pcbddc->local_subs);CHKERRQ(ierr);
1510c80a6c00SStefano Zampini       ierr = PCBDDCDetectDisconnectedComponents(pc,&pcbddc->n_local_subs,&pcbddc->local_subs,&primalv);CHKERRQ(ierr);
1511c80a6c00SStefano Zampini       if (primalv) {
1512c80a6c00SStefano Zampini         if (pcbddc->user_primal_vertices_local) {
1513956e2312SStefano Zampini           IS list[2], newp;
1514c80a6c00SStefano Zampini 
1515956e2312SStefano Zampini           list[0] = primalv;
1516956e2312SStefano Zampini           list[1] = pcbddc->user_primal_vertices_local;
1517c80a6c00SStefano Zampini           ierr = ISConcatenate(PetscObjectComm((PetscObject)pc),2,list,&newp);CHKERRQ(ierr);
1518c80a6c00SStefano Zampini           ierr = ISDestroy(&list[0]);CHKERRQ(ierr);
1519c80a6c00SStefano Zampini           ierr = ISDestroy(&list[1]);CHKERRQ(ierr);
1520c80a6c00SStefano Zampini           pcbddc->user_primal_vertices_local = newp;
1521c80a6c00SStefano Zampini         } else {
1522c80a6c00SStefano Zampini           pcbddc->user_primal_vertices_local = primalv;
1523c80a6c00SStefano Zampini         }
1524c80a6c00SStefano Zampini       }
1525c83e1ba7SStefano Zampini     }
1526c703fcc7SStefano Zampini     if (pcbddc->discretegradient) {
1527a13144ffSStefano Zampini       ierr = PCBDDCNedelecSupport(pc);CHKERRQ(ierr);
1528a13144ffSStefano Zampini     }
1529c703fcc7SStefano Zampini   }
1530a13144ffSStefano Zampini 
1531c703fcc7SStefano Zampini   /* change basis if requested by the user */
15325e8657edSStefano Zampini   if (pcbddc->user_ChangeOfBasisMatrix) {
15335e8657edSStefano Zampini     /* use_change_of_basis flag is used to automatically compute a change of basis from constraints */
15345e8657edSStefano Zampini     pcbddc->use_change_of_basis = PETSC_FALSE;
15355e8657edSStefano Zampini     ierr = PCBDDCComputeLocalMatrix(pc,pcbddc->user_ChangeOfBasisMatrix);CHKERRQ(ierr);
15365e8657edSStefano Zampini   } else {
1537b96c3477SStefano Zampini     ierr = MatDestroy(&pcbddc->local_mat);CHKERRQ(ierr);
15385e8657edSStefano Zampini     ierr = PetscObjectReference((PetscObject)matis->A);CHKERRQ(ierr);
15395e8657edSStefano Zampini     pcbddc->local_mat = matis->A;
1540d16cbb6bSStefano Zampini   }
1541d16cbb6bSStefano Zampini 
15424f1b2e48SStefano Zampini   /*
1543c703fcc7SStefano Zampini      Compute change of basis on local pressures (aka zerodiag dofs) with the benign trick
15444f1b2e48SStefano Zampini      This should come earlier then PCISSetUp for extracting the correct subdomain matrices
15454f1b2e48SStefano Zampini   */
15461dd7afcfSStefano Zampini   ierr = PCBDDCBenignShellMat(pc,PETSC_TRUE);CHKERRQ(ierr);
1547d16cbb6bSStefano Zampini   if (pcbddc->benign_saddle_point) {
15489f47a83aSStefano Zampini     PC_IS* pcis = (PC_IS*)pc->data;
15499f47a83aSStefano Zampini 
155005b28244SStefano Zampini     if (pcbddc->user_ChangeOfBasisMatrix || pcbddc->use_change_of_basis || !computesubschurs) pcbddc->benign_change_explicit = PETSC_TRUE;
1551669cc0f4SStefano Zampini     /* detect local saddle point and change the basis in pcbddc->local_mat (TODO: reuse case) */
1552339f8db1SStefano Zampini     ierr = PCBDDCBenignDetectSaddlePoint(pc,&zerodiag);CHKERRQ(ierr);
1553a3df083aSStefano Zampini     /* pop B0 mat from local mat */
1554c263805aSStefano Zampini     ierr = PCBDDCBenignPopOrPushB0(pc,PETSC_TRUE);CHKERRQ(ierr);
15551dd7afcfSStefano Zampini     /* give pcis a hint to not reuse submatrices during PCISCreate */
15561dd7afcfSStefano Zampini     if (pc->flag == SAME_NONZERO_PATTERN && pcis->reusesubmatrices == PETSC_TRUE) {
15571dd7afcfSStefano Zampini       if (pcbddc->benign_n && (pcbddc->benign_change_explicit || pcbddc->dbg_flag)) {
15581dd7afcfSStefano Zampini         pcis->reusesubmatrices = PETSC_FALSE;
15591dd7afcfSStefano Zampini       } else {
1560a3df083aSStefano Zampini         pcis->reusesubmatrices = PETSC_TRUE;
15611dd7afcfSStefano Zampini       }
1562a3df083aSStefano Zampini     } else {
15639f47a83aSStefano Zampini       pcis->reusesubmatrices = PETSC_FALSE;
1564674ae819SStefano Zampini     }
1565a3df083aSStefano Zampini   }
156627b6a85dSStefano Zampini 
1567c703fcc7SStefano Zampini   /* propagate relevant information -> TODO remove*/
15684f1b2e48SStefano Zampini #if !defined(PETSC_USE_COMPLEX) /* workaround for reals */
15693301b35fSStefano Zampini   if (matis->A->symmetric_set) {
15703301b35fSStefano Zampini     ierr = MatSetOption(pcbddc->local_mat,MAT_HERMITIAN,matis->A->symmetric);CHKERRQ(ierr);
15713301b35fSStefano Zampini   }
1572e496cd5dSStefano Zampini #endif
157306a4e24aSStefano Zampini   if (matis->A->symmetric_set) {
157406a4e24aSStefano Zampini     ierr = MatSetOption(pcbddc->local_mat,MAT_SYMMETRIC,matis->A->symmetric);CHKERRQ(ierr);
157506a4e24aSStefano Zampini   }
157606a4e24aSStefano Zampini   if (matis->A->spd_set) {
157706a4e24aSStefano Zampini     ierr = MatSetOption(pcbddc->local_mat,MAT_SPD,matis->A->spd);CHKERRQ(ierr);
157806a4e24aSStefano Zampini   }
1579e496cd5dSStefano Zampini 
15805e8657edSStefano Zampini   /* Set up all the "iterative substructuring" common block without computing solvers */
15815e8657edSStefano Zampini   {
15825e8657edSStefano Zampini     Mat temp_mat;
15835e8657edSStefano Zampini 
15845e8657edSStefano Zampini     temp_mat = matis->A;
15855e8657edSStefano Zampini     matis->A = pcbddc->local_mat;
15865e8657edSStefano Zampini     ierr = PCISSetUp(pc,PETSC_FALSE);CHKERRQ(ierr);
15875e8657edSStefano Zampini     pcbddc->local_mat = matis->A;
15885e8657edSStefano Zampini     matis->A = temp_mat;
15895e8657edSStefano Zampini   }
1590684f6988SStefano Zampini 
159181d14e9dSStefano Zampini   /* Analyze interface */
159264ac59b8SStefano Zampini   if (!pcbddc->graphanalyzed) {
1593674ae819SStefano Zampini     ierr = PCBDDCAnalyzeInterface(pc);CHKERRQ(ierr);
15948de1fae6SStefano Zampini     computeconstraintsmatrix = PETSC_TRUE;
1595345ecf6cSStefano Zampini     if (pcbddc->adaptive_selection && !pcbddc->use_deluxe_scaling && !pcbddc->mat_graph->twodim) {
15964247aa23Sstefano_zampini       SETERRQ(PetscObjectComm((PetscObject)pc),PETSC_ERR_SUP,"Cannot compute the adaptive primal space for a problem with 3D edges without deluxe scaling");
1597345ecf6cSStefano Zampini     }
1598a198735bSStefano Zampini     if (pcbddc->compute_nonetflux) {
1599669cc0f4SStefano Zampini       MatNullSpace nnfnnsp;
1600669cc0f4SStefano Zampini 
160121ef3d20SStefano Zampini       if (!pcbddc->divudotp) SETERRQ(PetscObjectComm((PetscObject)pc),PETSC_ERR_SUP,"Missing divudotp operator");
16028ae0ca82SStefano Zampini       ierr = PCBDDCComputeNoNetFlux(pc->pmat,pcbddc->divudotp,pcbddc->divudotp_trans,pcbddc->divudotp_vl2l,pcbddc->mat_graph,&nnfnnsp);CHKERRQ(ierr);
160371582508SStefano Zampini       /* TODO what if a nearnullspace is already attached? */
1604669cc0f4SStefano Zampini       ierr = MatSetNearNullSpace(pc->pmat,nnfnnsp);CHKERRQ(ierr);
1605669cc0f4SStefano Zampini       ierr = MatNullSpaceDestroy(&nnfnnsp);CHKERRQ(ierr);
1606669cc0f4SStefano Zampini     }
1607674ae819SStefano Zampini   }
1608fb8d54d4SStefano Zampini 
16095408967cSStefano Zampini   /* check existence of a divergence free extension, i.e.
16105408967cSStefano Zampini      b(v_I,p_0) = 0 for all v_I (raise error if not).
16115408967cSStefano Zampini      Also, check that PCBDDCBenignGetOrSetP0 works */
1612ff1f7e73Sstefano_zampini   if (pcbddc->benign_saddle_point && pcbddc->dbg_flag > 1) {
16135408967cSStefano Zampini     ierr = PCBDDCBenignCheck(pc,zerodiag);CHKERRQ(ierr);
161409f581a4SStefano Zampini   }
16154f1b2e48SStefano Zampini   ierr = ISDestroy(&zerodiag);CHKERRQ(ierr);
161606f24817SStefano Zampini 
1617b96c3477SStefano Zampini   /* Setup local dirichlet solver ksp_D and sub_schurs solvers */
1618c703fcc7SStefano Zampini   if (computesubschurs && pcbddc->recompute_topography) {
161908122e43SStefano Zampini     ierr = PCBDDCInitSubSchurs(pc);CHKERRQ(ierr);
1620b1b3d7a2SStefano Zampini   }
16219d54b7f4SStefano Zampini   /* SetUp Scaling operator (scaling matrices could be needed in SubSchursSetUp)*/
16229d54b7f4SStefano Zampini   if (!pcbddc->use_deluxe_scaling) {
16239d54b7f4SStefano Zampini     ierr = PCBDDCScalingSetUp(pc);CHKERRQ(ierr);
16249d54b7f4SStefano Zampini   }
1625c703fcc7SStefano Zampini 
1626c703fcc7SStefano Zampini   /* finish setup solvers and do adaptive selection of constraints */
1627b334f244SStefano Zampini   sub_schurs = pcbddc->sub_schurs;
1628b334f244SStefano Zampini   if (sub_schurs && sub_schurs->schur_explicit) {
16292070dbb6SStefano Zampini     if (computesubschurs) {
163008122e43SStefano Zampini       ierr = PCBDDCSetUpSubSchurs(pc);CHKERRQ(ierr);
16312070dbb6SStefano Zampini     }
1632d5574798SStefano Zampini     ierr = PCBDDCSetUpLocalSolvers(pc,PETSC_TRUE,PETSC_FALSE);CHKERRQ(ierr);
1633d5574798SStefano Zampini   } else {
1634d5574798SStefano Zampini     ierr = PCBDDCSetUpLocalSolvers(pc,PETSC_TRUE,PETSC_FALSE);CHKERRQ(ierr);
16352070dbb6SStefano Zampini     if (computesubschurs) {
1636d5574798SStefano Zampini       ierr = PCBDDCSetUpSubSchurs(pc);CHKERRQ(ierr);
1637d5574798SStefano Zampini     }
16382070dbb6SStefano Zampini   }
163908122e43SStefano Zampini   if (pcbddc->adaptive_selection) {
164008122e43SStefano Zampini     ierr = PCBDDCAdaptiveSelection(pc);CHKERRQ(ierr);
16418de1fae6SStefano Zampini     computeconstraintsmatrix = PETSC_TRUE;
1642b7eb3628SStefano Zampini   }
1643684f6988SStefano Zampini 
1644f4ddd8eeSStefano Zampini   /* infer if NullSpace object attached to Mat via MatSetNearNullSpace has changed */
1645fb8d54d4SStefano Zampini   new_nearnullspace_provided = PETSC_FALSE;
1646f4ddd8eeSStefano Zampini   ierr = MatGetNearNullSpace(pc->pmat,&nearnullspace);CHKERRQ(ierr);
1647f4ddd8eeSStefano Zampini   if (pcbddc->onearnullspace) { /* already used nearnullspace */
1648f4ddd8eeSStefano Zampini     if (!nearnullspace) { /* near null space attached to mat has been destroyed */
1649f4ddd8eeSStefano Zampini       new_nearnullspace_provided = PETSC_TRUE;
1650f4ddd8eeSStefano Zampini     } else {
1651f4ddd8eeSStefano Zampini       /* determine if the two nullspaces are different (should be lightweight) */
1652f4ddd8eeSStefano Zampini       if (nearnullspace != pcbddc->onearnullspace) {
1653f4ddd8eeSStefano Zampini         new_nearnullspace_provided = PETSC_TRUE;
1654165b64e2SStefano Zampini       } else { /* maybe the user has changed the content of the nearnullspace so check vectors ObjectStateId */
1655f4ddd8eeSStefano Zampini         PetscInt         i;
1656165b64e2SStefano Zampini         const Vec        *nearnullvecs;
1657165b64e2SStefano Zampini         PetscObjectState state;
1658165b64e2SStefano Zampini         PetscInt         nnsp_size;
1659165b64e2SStefano Zampini         ierr = MatNullSpaceGetVecs(nearnullspace,NULL,&nnsp_size,&nearnullvecs);CHKERRQ(ierr);
1660f4ddd8eeSStefano Zampini         for (i=0;i<nnsp_size;i++) {
1661f4ddd8eeSStefano Zampini           ierr = PetscObjectStateGet((PetscObject)nearnullvecs[i],&state);CHKERRQ(ierr);
1662165b64e2SStefano Zampini           if (pcbddc->onearnullvecs_state[i] != state) {
1663f4ddd8eeSStefano Zampini             new_nearnullspace_provided = PETSC_TRUE;
1664f4ddd8eeSStefano Zampini             break;
1665f4ddd8eeSStefano Zampini           }
1666f4ddd8eeSStefano Zampini         }
1667f4ddd8eeSStefano Zampini       }
1668f4ddd8eeSStefano Zampini     }
1669f4ddd8eeSStefano Zampini   } else {
1670f4ddd8eeSStefano Zampini     if (!nearnullspace) { /* both nearnullspaces are null */
1671f4ddd8eeSStefano Zampini       new_nearnullspace_provided = PETSC_FALSE;
1672f4ddd8eeSStefano Zampini     } else { /* nearnullspace attached later */
1673f4ddd8eeSStefano Zampini       new_nearnullspace_provided = PETSC_TRUE;
1674f4ddd8eeSStefano Zampini     }
1675f4ddd8eeSStefano Zampini   }
1676f4ddd8eeSStefano Zampini 
1677f4ddd8eeSStefano Zampini   /* Setup constraints and related work vectors */
1678727cdba6SStefano Zampini   /* reset primal space flags */
1679f4ddd8eeSStefano Zampini   pcbddc->new_primal_space = PETSC_FALSE;
1680727cdba6SStefano Zampini   pcbddc->new_primal_space_local = PETSC_FALSE;
16818de1fae6SStefano Zampini   if (computeconstraintsmatrix || new_nearnullspace_provided) {
1682727cdba6SStefano Zampini     /* It also sets the primal space flags */
1683674ae819SStefano Zampini     ierr = PCBDDCConstraintsSetUp(pc);CHKERRQ(ierr);
16849543d0ffSStefano Zampini   }
1685e7b262bdSStefano Zampini   /* Allocate needed local vectors (which depends on quantities defined during ConstraintsSetUp) */
1686f4ddd8eeSStefano Zampini   ierr = PCBDDCSetUpLocalWorkVectors(pc);CHKERRQ(ierr);
16875e8657edSStefano Zampini 
16885e8657edSStefano Zampini   if (pcbddc->use_change_of_basis) {
16895e8657edSStefano Zampini     PC_IS *pcis = (PC_IS*)(pc->data);
16905e8657edSStefano Zampini 
16915e8657edSStefano Zampini     ierr = PCBDDCComputeLocalMatrix(pc,pcbddc->ChangeOfBasisMatrix);CHKERRQ(ierr);
16924f1b2e48SStefano Zampini     if (pcbddc->benign_change) {
16931dd7afcfSStefano Zampini       ierr = MatDestroy(&pcbddc->benign_B0);CHKERRQ(ierr);
1694c263805aSStefano Zampini       /* pop B0 from pcbddc->local_mat */
1695c263805aSStefano Zampini       ierr = PCBDDCBenignPopOrPushB0(pc,PETSC_TRUE);CHKERRQ(ierr);
1696c263805aSStefano Zampini     }
16975e8657edSStefano Zampini     /* get submatrices */
16985e8657edSStefano Zampini     ierr = MatDestroy(&pcis->A_IB);CHKERRQ(ierr);
16995e8657edSStefano Zampini     ierr = MatDestroy(&pcis->A_BI);CHKERRQ(ierr);
17005e8657edSStefano Zampini     ierr = MatDestroy(&pcis->A_BB);CHKERRQ(ierr);
17017dae84e0SHong Zhang     ierr = MatCreateSubMatrix(pcbddc->local_mat,pcis->is_B_local,pcis->is_B_local,MAT_INITIAL_MATRIX,&pcis->A_BB);CHKERRQ(ierr);
17027dae84e0SHong Zhang     ierr = MatCreateSubMatrix(pcbddc->local_mat,pcis->is_I_local,pcis->is_B_local,MAT_INITIAL_MATRIX,&pcis->A_IB);CHKERRQ(ierr);
17037dae84e0SHong Zhang     ierr = MatCreateSubMatrix(pcbddc->local_mat,pcis->is_B_local,pcis->is_I_local,MAT_INITIAL_MATRIX,&pcis->A_BI);CHKERRQ(ierr);
17043975b054SStefano Zampini     /* set flag in pcis to not reuse submatrices during PCISCreate */
17053975b054SStefano Zampini     pcis->reusesubmatrices = PETSC_FALSE;
17069c6a02ceSStefano Zampini   } else if (!pcbddc->user_ChangeOfBasisMatrix && !pcbddc->benign_change) {
1707b96c3477SStefano Zampini     ierr = MatDestroy(&pcbddc->local_mat);CHKERRQ(ierr);
17085e8657edSStefano Zampini     ierr = PetscObjectReference((PetscObject)matis->A);CHKERRQ(ierr);
17095e8657edSStefano Zampini     pcbddc->local_mat = matis->A;
17105e8657edSStefano Zampini   }
171135509ce9Sstefano_zampini 
171235509ce9Sstefano_zampini   /* interface pressure block row for B_C */
171335509ce9Sstefano_zampini   ierr = PetscObjectQuery((PetscObject)pc,"__KSPFETIDP_lP" ,(PetscObject*)&lP);CHKERRQ(ierr);
171435509ce9Sstefano_zampini   ierr = PetscObjectQuery((PetscObject)pc,"__KSPFETIDP_lA" ,(PetscObject*)&lA);CHKERRQ(ierr);
171535509ce9Sstefano_zampini   if (lA && lP) {
171635509ce9Sstefano_zampini     PC_IS*    pcis = (PC_IS*)pc->data;
171735509ce9Sstefano_zampini     Mat       B_BI,B_BB,Bt_BI,Bt_BB;
171835509ce9Sstefano_zampini     PetscBool issym;
171935509ce9Sstefano_zampini     ierr = MatIsSymmetric(lA,PETSC_SMALL,&issym);CHKERRQ(ierr);
17206cc1294bSstefano_zampini     if (issym) {
17217dae84e0SHong Zhang       ierr = MatCreateSubMatrix(lA,lP,pcis->is_I_local,MAT_INITIAL_MATRIX,&B_BI);CHKERRQ(ierr);
17227dae84e0SHong Zhang       ierr = MatCreateSubMatrix(lA,lP,pcis->is_B_local,MAT_INITIAL_MATRIX,&B_BB);CHKERRQ(ierr);
172335509ce9Sstefano_zampini       ierr = MatCreateTranspose(B_BI,&Bt_BI);CHKERRQ(ierr);
172435509ce9Sstefano_zampini       ierr = MatCreateTranspose(B_BB,&Bt_BB);CHKERRQ(ierr);
172535509ce9Sstefano_zampini     } else {
17267dae84e0SHong Zhang       ierr = MatCreateSubMatrix(lA,lP,pcis->is_I_local,MAT_INITIAL_MATRIX,&B_BI);CHKERRQ(ierr);
17277dae84e0SHong Zhang       ierr = MatCreateSubMatrix(lA,lP,pcis->is_B_local,MAT_INITIAL_MATRIX,&B_BB);CHKERRQ(ierr);
17287dae84e0SHong Zhang       ierr = MatCreateSubMatrix(lA,pcis->is_I_local,lP,MAT_INITIAL_MATRIX,&Bt_BI);CHKERRQ(ierr);
17297dae84e0SHong Zhang       ierr = MatCreateSubMatrix(lA,pcis->is_B_local,lP,MAT_INITIAL_MATRIX,&Bt_BB);CHKERRQ(ierr);
173035509ce9Sstefano_zampini     }
173135509ce9Sstefano_zampini     ierr = PetscObjectCompose((PetscObject)pc,"__KSPFETIDP_B_BI",(PetscObject)B_BI);CHKERRQ(ierr);
173235509ce9Sstefano_zampini     ierr = PetscObjectCompose((PetscObject)pc,"__KSPFETIDP_B_BB",(PetscObject)B_BB);CHKERRQ(ierr);
173335509ce9Sstefano_zampini     ierr = PetscObjectCompose((PetscObject)pc,"__KSPFETIDP_Bt_BI",(PetscObject)Bt_BI);CHKERRQ(ierr);
173435509ce9Sstefano_zampini     ierr = PetscObjectCompose((PetscObject)pc,"__KSPFETIDP_Bt_BB",(PetscObject)Bt_BB);CHKERRQ(ierr);
173535509ce9Sstefano_zampini     ierr = MatDestroy(&B_BI);CHKERRQ(ierr);
173635509ce9Sstefano_zampini     ierr = MatDestroy(&B_BB);CHKERRQ(ierr);
173735509ce9Sstefano_zampini     ierr = MatDestroy(&Bt_BI);CHKERRQ(ierr);
173835509ce9Sstefano_zampini     ierr = MatDestroy(&Bt_BB);CHKERRQ(ierr);
173935509ce9Sstefano_zampini   }
174035509ce9Sstefano_zampini 
1741b96c3477SStefano Zampini   /* SetUp coarse and local Neumann solvers */
174299cc7994SStefano Zampini   ierr = PCBDDCSetUpSolvers(pc);CHKERRQ(ierr);
1743b96c3477SStefano Zampini   /* SetUp Scaling operator */
17449d54b7f4SStefano Zampini   if (pcbddc->use_deluxe_scaling) {
1745674ae819SStefano Zampini     ierr = PCBDDCScalingSetUp(pc);CHKERRQ(ierr);
17460c7d97c5SJed Brown   }
1747c703fcc7SStefano Zampini 
17481dd7afcfSStefano Zampini   /* mark topography as done */
174956282151SStefano Zampini   pcbddc->recompute_topography = PETSC_FALSE;
17500369aaf7SStefano Zampini 
17511dd7afcfSStefano Zampini   /* wrap pcis->A_IB and pcis->A_BI if we did not change explicitly the variables on the pressures */
17521dd7afcfSStefano Zampini   ierr = PCBDDCBenignShellMat(pc,PETSC_FALSE);CHKERRQ(ierr);
17531dd7afcfSStefano Zampini 
175458a03d70SStefano Zampini   if (pcbddc->dbg_flag) {
175558a03d70SStefano Zampini     ierr = PetscViewerASCIISubtractTab(pcbddc->dbg_viewer,2*pcbddc->current_level);CHKERRQ(ierr);
17562b510759SStefano Zampini   }
17570c7d97c5SJed Brown   PetscFunctionReturn(0);
17580c7d97c5SJed Brown }
17590c7d97c5SJed Brown 
17600c7d97c5SJed Brown /*
176150efa1b5SStefano Zampini    PCApply_BDDC - Applies the BDDC operator to a vector.
17620c7d97c5SJed Brown 
17630c7d97c5SJed Brown    Input Parameters:
17640f202f7eSStefano Zampini +  pc - the preconditioner context
17650f202f7eSStefano Zampini -  r - input vector (global)
17660c7d97c5SJed Brown 
17670c7d97c5SJed Brown    Output Parameter:
17680c7d97c5SJed Brown .  z - output vector (global)
17690c7d97c5SJed Brown 
17700c7d97c5SJed Brown    Application Interface Routine: PCApply()
17710c7d97c5SJed Brown  */
177253cdbc3dSStefano Zampini PetscErrorCode PCApply_BDDC(PC pc,Vec r,Vec z)
17730c7d97c5SJed Brown {
17740c7d97c5SJed Brown   PC_IS             *pcis = (PC_IS*)(pc->data);
17750c7d97c5SJed Brown   PC_BDDC           *pcbddc = (PC_BDDC*)(pc->data);
1776b097fa66SStefano Zampini   PetscInt          n_B = pcis->n_B, n_D = pcis->n - n_B;
17770c7d97c5SJed Brown   PetscErrorCode    ierr;
17783b03a366Sstefano_zampini   const PetscScalar one = 1.0;
17793b03a366Sstefano_zampini   const PetscScalar m_one = -1.0;
17802617d88aSStefano Zampini   const PetscScalar zero = 0.0;
17810c7d97c5SJed Brown 
17820c7d97c5SJed Brown /* This code is similar to that provided in nn.c for PCNN
17830c7d97c5SJed Brown    NN interface preconditioner changed to BDDC
1784b097fa66SStefano Zampini    Added support for M_3 preconditioner in the reference article (code is active if pcbddc->switch_static == PETSC_TRUE) */
17850c7d97c5SJed Brown 
17860c7d97c5SJed Brown   PetscFunctionBegin;
1787f3d41395Sstefano_zampini   ierr = PetscCitationsRegister(citation,&cited);CHKERRQ(ierr);
17881dd7afcfSStefano Zampini   if (pcbddc->ChangeOfBasisMatrix) {
17891dd7afcfSStefano Zampini     Vec swap;
179027b6a85dSStefano Zampini 
179127b6a85dSStefano Zampini     ierr = MatMultTranspose(pcbddc->ChangeOfBasisMatrix,r,pcbddc->work_change);CHKERRQ(ierr);
17921dd7afcfSStefano Zampini     swap = pcbddc->work_change;
17931dd7afcfSStefano Zampini     pcbddc->work_change = r;
17941dd7afcfSStefano Zampini     r = swap;
17951dd7afcfSStefano Zampini     /* save rhs so that we don't need to apply the change of basis for the exact dirichlet trick in PreSolve */
17969cc2a9b1Sstefano_zampini     if (pcbddc->benign_apply_coarse_only && pcbddc->use_exact_dirichlet_trick && pcbddc->change_interior) {
17971dd7afcfSStefano Zampini       ierr = VecCopy(r,pcis->vec1_global);CHKERRQ(ierr);
17981dd7afcfSStefano Zampini       ierr = VecLockPush(pcis->vec1_global);CHKERRQ(ierr);
17991dd7afcfSStefano Zampini     }
18001dd7afcfSStefano Zampini   }
180127b6a85dSStefano Zampini   if (pcbddc->benign_have_null) { /* get p0 from r */
1802015636ebSStefano Zampini     ierr = PCBDDCBenignGetOrSetP0(pc,r,PETSC_TRUE);CHKERRQ(ierr);
1803efc2fbd9SStefano Zampini   }
18048ae0ca82SStefano Zampini   if (!pcbddc->exact_dirichlet_trick_app && !pcbddc->benign_apply_coarse_only) {
1805b097fa66SStefano Zampini     ierr = VecCopy(r,z);CHKERRQ(ierr);
18060c7d97c5SJed Brown     /* First Dirichlet solve */
18070c7d97c5SJed Brown     ierr = VecScatterBegin(pcis->global_to_D,r,pcis->vec1_D,INSERT_VALUES,SCATTER_FORWARD);CHKERRQ(ierr);
18080c7d97c5SJed Brown     ierr = VecScatterEnd(pcis->global_to_D,r,pcis->vec1_D,INSERT_VALUES,SCATTER_FORWARD);CHKERRQ(ierr);
18090c7d97c5SJed Brown     /*
18100c7d97c5SJed Brown       Assembling right hand side for BDDC operator
1811b097fa66SStefano Zampini       - pcis->vec1_D for the Dirichlet part (if needed, i.e. pcbddc->switch_static == PETSC_TRUE)
1812674ae819SStefano Zampini       - pcis->vec1_B the interface part of the global vector z
18130c7d97c5SJed Brown     */
1814b097fa66SStefano Zampini     if (n_D) {
1815b097fa66SStefano Zampini       ierr = KSPSolve(pcbddc->ksp_D,pcis->vec1_D,pcis->vec2_D);CHKERRQ(ierr);
18160c7d97c5SJed Brown       ierr = VecScale(pcis->vec2_D,m_one);CHKERRQ(ierr);
181716909a7fSStefano Zampini       if (pcbddc->switch_static) {
181816909a7fSStefano Zampini         Mat_IS *matis = (Mat_IS*)(pc->mat->data);
181916909a7fSStefano Zampini 
182016909a7fSStefano Zampini         ierr = VecSet(pcis->vec1_N,0.);CHKERRQ(ierr);
182116909a7fSStefano Zampini         ierr = VecScatterBegin(pcis->N_to_D,pcis->vec2_D,pcis->vec1_N,INSERT_VALUES,SCATTER_REVERSE);CHKERRQ(ierr);
182216909a7fSStefano Zampini         ierr = VecScatterEnd(pcis->N_to_D,pcis->vec2_D,pcis->vec1_N,INSERT_VALUES,SCATTER_REVERSE);CHKERRQ(ierr);
182316909a7fSStefano Zampini         if (!pcbddc->switch_static_change) {
182416909a7fSStefano Zampini           ierr = MatMult(matis->A,pcis->vec1_N,pcis->vec2_N);CHKERRQ(ierr);
182516909a7fSStefano Zampini         } else {
182616909a7fSStefano Zampini           ierr = MatMult(pcbddc->switch_static_change,pcis->vec1_N,pcis->vec2_N);CHKERRQ(ierr);
182716909a7fSStefano Zampini           ierr = MatMult(matis->A,pcis->vec2_N,pcis->vec1_N);CHKERRQ(ierr);
182816909a7fSStefano Zampini           ierr = MatMultTranspose(pcbddc->switch_static_change,pcis->vec1_N,pcis->vec2_N);CHKERRQ(ierr);
182916909a7fSStefano Zampini         }
183016909a7fSStefano Zampini         ierr = VecScatterBegin(pcis->N_to_D,pcis->vec2_N,pcis->vec1_D,ADD_VALUES,SCATTER_FORWARD);CHKERRQ(ierr);
183116909a7fSStefano Zampini         ierr = VecScatterEnd(pcis->N_to_D,pcis->vec2_N,pcis->vec1_D,ADD_VALUES,SCATTER_FORWARD);CHKERRQ(ierr);
183216909a7fSStefano Zampini         ierr = VecScatterBegin(pcis->N_to_B,pcis->vec2_N,pcis->vec1_B,INSERT_VALUES,SCATTER_FORWARD);CHKERRQ(ierr);
183316909a7fSStefano Zampini         ierr = VecScatterEnd(pcis->N_to_B,pcis->vec2_N,pcis->vec1_B,INSERT_VALUES,SCATTER_FORWARD);CHKERRQ(ierr);
183416909a7fSStefano Zampini       } else {
1835b097fa66SStefano Zampini         ierr = MatMult(pcis->A_BI,pcis->vec2_D,pcis->vec1_B);CHKERRQ(ierr);
183616909a7fSStefano Zampini       }
1837b097fa66SStefano Zampini     } else {
1838b097fa66SStefano Zampini       ierr = VecSet(pcis->vec1_B,zero);CHKERRQ(ierr);
1839b097fa66SStefano Zampini     }
18400c7d97c5SJed Brown     ierr = VecScatterBegin(pcis->global_to_B,pcis->vec1_B,z,ADD_VALUES,SCATTER_REVERSE);CHKERRQ(ierr);
18410c7d97c5SJed Brown     ierr = VecScatterEnd(pcis->global_to_B,pcis->vec1_B,z,ADD_VALUES,SCATTER_REVERSE);CHKERRQ(ierr);
1842674ae819SStefano Zampini     ierr = PCBDDCScalingRestriction(pc,z,pcis->vec1_B);CHKERRQ(ierr);
1843b76ba322SStefano Zampini   } else {
18444fee134fSStefano Zampini     if (!pcbddc->benign_apply_coarse_only) {
1845674ae819SStefano Zampini       ierr = PCBDDCScalingRestriction(pc,r,pcis->vec1_B);CHKERRQ(ierr);
1846b76ba322SStefano Zampini     }
18474fee134fSStefano Zampini   }
1848b76ba322SStefano Zampini 
18492617d88aSStefano Zampini   /* Apply interface preconditioner
18502617d88aSStefano Zampini      input/output vecs: pcis->vec1_B and pcis->vec1_D */
1851dc359a40SStefano Zampini   ierr = PCBDDCApplyInterfacePreconditioner(pc,PETSC_FALSE);CHKERRQ(ierr);
18522617d88aSStefano Zampini 
1853674ae819SStefano Zampini   /* Apply transpose of partition of unity operator */
1854674ae819SStefano Zampini   ierr = PCBDDCScalingExtension(pc,pcis->vec1_B,z);CHKERRQ(ierr);
18550c7d97c5SJed Brown 
18563b03a366Sstefano_zampini   /* Second Dirichlet solve and assembling of output */
18570c7d97c5SJed Brown   ierr = VecScatterBegin(pcis->global_to_B,z,pcis->vec1_B,INSERT_VALUES,SCATTER_FORWARD);CHKERRQ(ierr);
18580c7d97c5SJed Brown   ierr = VecScatterEnd(pcis->global_to_B,z,pcis->vec1_B,INSERT_VALUES,SCATTER_FORWARD);CHKERRQ(ierr);
1859b097fa66SStefano Zampini   if (n_B) {
186016909a7fSStefano Zampini     if (pcbddc->switch_static) {
186116909a7fSStefano Zampini       Mat_IS *matis = (Mat_IS*)(pc->mat->data);
186216909a7fSStefano Zampini 
186316909a7fSStefano Zampini       ierr = VecScatterBegin(pcis->N_to_D,pcis->vec1_D,pcis->vec1_N,INSERT_VALUES,SCATTER_REVERSE);CHKERRQ(ierr);
186416909a7fSStefano Zampini       ierr = VecScatterEnd(pcis->N_to_D,pcis->vec1_D,pcis->vec1_N,INSERT_VALUES,SCATTER_REVERSE);CHKERRQ(ierr);
186516909a7fSStefano Zampini       ierr = VecScatterBegin(pcis->N_to_B,pcis->vec1_B,pcis->vec1_N,INSERT_VALUES,SCATTER_REVERSE);CHKERRQ(ierr);
186616909a7fSStefano Zampini       ierr = VecScatterEnd(pcis->N_to_B,pcis->vec1_B,pcis->vec1_N,INSERT_VALUES,SCATTER_REVERSE);CHKERRQ(ierr);
186716909a7fSStefano Zampini       if (!pcbddc->switch_static_change) {
186816909a7fSStefano Zampini         ierr = MatMult(matis->A,pcis->vec1_N,pcis->vec2_N);CHKERRQ(ierr);
186916909a7fSStefano Zampini       } else {
187016909a7fSStefano Zampini         ierr = MatMult(pcbddc->switch_static_change,pcis->vec1_N,pcis->vec2_N);CHKERRQ(ierr);
187116909a7fSStefano Zampini         ierr = MatMult(matis->A,pcis->vec2_N,pcis->vec1_N);CHKERRQ(ierr);
187216909a7fSStefano Zampini         ierr = MatMultTranspose(pcbddc->switch_static_change,pcis->vec1_N,pcis->vec2_N);CHKERRQ(ierr);
187316909a7fSStefano Zampini       }
187416909a7fSStefano Zampini       ierr = VecScatterBegin(pcis->N_to_D,pcis->vec2_N,pcis->vec3_D,INSERT_VALUES,SCATTER_FORWARD);CHKERRQ(ierr);
187516909a7fSStefano Zampini       ierr = VecScatterEnd(pcis->N_to_D,pcis->vec2_N,pcis->vec3_D,INSERT_VALUES,SCATTER_FORWARD);CHKERRQ(ierr);
187616909a7fSStefano Zampini     } else {
18770c7d97c5SJed Brown       ierr = MatMult(pcis->A_IB,pcis->vec1_B,pcis->vec3_D);CHKERRQ(ierr);
187816909a7fSStefano Zampini     }
187916909a7fSStefano Zampini   } else if (pcbddc->switch_static) { /* n_B is zero */
188016909a7fSStefano Zampini     Mat_IS *matis = (Mat_IS*)(pc->mat->data);
188116909a7fSStefano Zampini 
188216909a7fSStefano Zampini     if (!pcbddc->switch_static_change) {
188316909a7fSStefano Zampini       ierr = MatMult(matis->A,pcis->vec1_D,pcis->vec3_D);CHKERRQ(ierr);
188416909a7fSStefano Zampini     } else {
188516909a7fSStefano Zampini       ierr = MatMult(pcbddc->switch_static_change,pcis->vec1_D,pcis->vec1_N);CHKERRQ(ierr);
188616909a7fSStefano Zampini       ierr = MatMult(matis->A,pcis->vec1_N,pcis->vec2_N);CHKERRQ(ierr);
188716909a7fSStefano Zampini       ierr = MatMultTranspose(pcbddc->switch_static_change,pcis->vec2_N,pcis->vec3_D);CHKERRQ(ierr);
188816909a7fSStefano Zampini     }
1889b097fa66SStefano Zampini   }
1890df187020SStefano Zampini   ierr = KSPSolve(pcbddc->ksp_D,pcis->vec3_D,pcis->vec4_D);CHKERRQ(ierr);
1891efc2fbd9SStefano Zampini 
18928ae0ca82SStefano Zampini   if (!pcbddc->exact_dirichlet_trick_app && !pcbddc->benign_apply_coarse_only) {
1893b097fa66SStefano Zampini     if (pcbddc->switch_static) {
1894b097fa66SStefano Zampini       ierr = VecAXPBYPCZ(pcis->vec2_D,m_one,one,m_one,pcis->vec4_D,pcis->vec1_D);CHKERRQ(ierr);
1895b097fa66SStefano Zampini     } else {
1896b097fa66SStefano Zampini       ierr = VecAXPBY(pcis->vec2_D,m_one,m_one,pcis->vec4_D);CHKERRQ(ierr);
1897b097fa66SStefano Zampini     }
18980c7d97c5SJed Brown     ierr = VecScatterBegin(pcis->global_to_D,pcis->vec2_D,z,INSERT_VALUES,SCATTER_REVERSE);CHKERRQ(ierr);
18990c7d97c5SJed Brown     ierr = VecScatterEnd(pcis->global_to_D,pcis->vec2_D,z,INSERT_VALUES,SCATTER_REVERSE);CHKERRQ(ierr);
1900b097fa66SStefano Zampini   } else {
1901b097fa66SStefano Zampini     if (pcbddc->switch_static) {
1902b097fa66SStefano Zampini       ierr = VecAXPBY(pcis->vec4_D,one,m_one,pcis->vec1_D);CHKERRQ(ierr);
1903b097fa66SStefano Zampini     } else {
1904b097fa66SStefano Zampini       ierr = VecScale(pcis->vec4_D,m_one);CHKERRQ(ierr);
1905b097fa66SStefano Zampini     }
1906b097fa66SStefano Zampini     ierr = VecScatterBegin(pcis->global_to_D,pcis->vec4_D,z,INSERT_VALUES,SCATTER_REVERSE);CHKERRQ(ierr);
1907b097fa66SStefano Zampini     ierr = VecScatterEnd(pcis->global_to_D,pcis->vec4_D,z,INSERT_VALUES,SCATTER_REVERSE);CHKERRQ(ierr);
1908b097fa66SStefano Zampini   }
190927b6a85dSStefano Zampini   if (pcbddc->benign_have_null) { /* set p0 (computed in PCBDDCApplyInterface) */
19101dd7afcfSStefano Zampini     if (pcbddc->benign_apply_coarse_only) {
19111dd7afcfSStefano Zampini       ierr = PetscMemzero(pcbddc->benign_p0,pcbddc->benign_n*sizeof(PetscScalar));CHKERRQ(ierr);
19121dd7afcfSStefano Zampini     }
1913015636ebSStefano Zampini     ierr = PCBDDCBenignGetOrSetP0(pc,z,PETSC_FALSE);CHKERRQ(ierr);
1914efc2fbd9SStefano Zampini   }
19151f4df5f7SStefano Zampini 
19161dd7afcfSStefano Zampini   if (pcbddc->ChangeOfBasisMatrix) {
1917f913dca9SStefano Zampini     pcbddc->work_change = r;
19181dd7afcfSStefano Zampini     ierr = VecCopy(z,pcbddc->work_change);CHKERRQ(ierr);
19191dd7afcfSStefano Zampini     ierr = MatMult(pcbddc->ChangeOfBasisMatrix,pcbddc->work_change,z);CHKERRQ(ierr);
19201dd7afcfSStefano Zampini   }
19210c7d97c5SJed Brown   PetscFunctionReturn(0);
19220c7d97c5SJed Brown }
192350efa1b5SStefano Zampini 
192450efa1b5SStefano Zampini /*
192550efa1b5SStefano Zampini    PCApplyTranspose_BDDC - Applies the transpose of the BDDC operator to a vector.
192650efa1b5SStefano Zampini 
192750efa1b5SStefano Zampini    Input Parameters:
19280f202f7eSStefano Zampini +  pc - the preconditioner context
19290f202f7eSStefano Zampini -  r - input vector (global)
193050efa1b5SStefano Zampini 
193150efa1b5SStefano Zampini    Output Parameter:
193250efa1b5SStefano Zampini .  z - output vector (global)
193350efa1b5SStefano Zampini 
193450efa1b5SStefano Zampini    Application Interface Routine: PCApplyTranspose()
193550efa1b5SStefano Zampini  */
193650efa1b5SStefano Zampini PetscErrorCode PCApplyTranspose_BDDC(PC pc,Vec r,Vec z)
193750efa1b5SStefano Zampini {
193850efa1b5SStefano Zampini   PC_IS             *pcis = (PC_IS*)(pc->data);
193950efa1b5SStefano Zampini   PC_BDDC           *pcbddc = (PC_BDDC*)(pc->data);
1940b097fa66SStefano Zampini   PetscInt          n_B = pcis->n_B, n_D = pcis->n - n_B;
194150efa1b5SStefano Zampini   PetscErrorCode    ierr;
194250efa1b5SStefano Zampini   const PetscScalar one = 1.0;
194350efa1b5SStefano Zampini   const PetscScalar m_one = -1.0;
194450efa1b5SStefano Zampini   const PetscScalar zero = 0.0;
194550efa1b5SStefano Zampini 
194650efa1b5SStefano Zampini   PetscFunctionBegin;
1947f3d41395Sstefano_zampini   ierr = PetscCitationsRegister(citation,&cited);CHKERRQ(ierr);
19481dd7afcfSStefano Zampini   if (pcbddc->ChangeOfBasisMatrix) {
19491dd7afcfSStefano Zampini     Vec swap;
195027b6a85dSStefano Zampini 
195127b6a85dSStefano Zampini     ierr = MatMultTranspose(pcbddc->ChangeOfBasisMatrix,r,pcbddc->work_change);CHKERRQ(ierr);
19521dd7afcfSStefano Zampini     swap = pcbddc->work_change;
19531dd7afcfSStefano Zampini     pcbddc->work_change = r;
19541dd7afcfSStefano Zampini     r = swap;
195527b6a85dSStefano Zampini     /* save rhs so that we don't need to apply the change of basis for the exact dirichlet trick in PreSolve */
19568ae0ca82SStefano Zampini     if (pcbddc->benign_apply_coarse_only && pcbddc->exact_dirichlet_trick_app && pcbddc->change_interior) {
195727b6a85dSStefano Zampini       ierr = VecCopy(r,pcis->vec1_global);CHKERRQ(ierr);
195827b6a85dSStefano Zampini       ierr = VecLockPush(pcis->vec1_global);CHKERRQ(ierr);
19591dd7afcfSStefano Zampini     }
196027b6a85dSStefano Zampini   }
196127b6a85dSStefano Zampini   if (pcbddc->benign_have_null) { /* get p0 from r */
1962537c1cdfSStefano Zampini     ierr = PCBDDCBenignGetOrSetP0(pc,r,PETSC_TRUE);CHKERRQ(ierr);
1963537c1cdfSStefano Zampini   }
19648ae0ca82SStefano Zampini   if (!pcbddc->exact_dirichlet_trick_app && !pcbddc->benign_apply_coarse_only) {
1965b097fa66SStefano Zampini     ierr = VecCopy(r,z);CHKERRQ(ierr);
196650efa1b5SStefano Zampini     /* First Dirichlet solve */
196750efa1b5SStefano Zampini     ierr = VecScatterBegin(pcis->global_to_D,r,pcis->vec1_D,INSERT_VALUES,SCATTER_FORWARD);CHKERRQ(ierr);
196850efa1b5SStefano Zampini     ierr = VecScatterEnd(pcis->global_to_D,r,pcis->vec1_D,INSERT_VALUES,SCATTER_FORWARD);CHKERRQ(ierr);
196950efa1b5SStefano Zampini     /*
197050efa1b5SStefano Zampini       Assembling right hand side for BDDC operator
1971b097fa66SStefano Zampini       - pcis->vec1_D for the Dirichlet part (if needed, i.e. pcbddc->switch_static == PETSC_TRUE)
197250efa1b5SStefano Zampini       - pcis->vec1_B the interface part of the global vector z
197350efa1b5SStefano Zampini     */
1974b097fa66SStefano Zampini     if (n_D) {
1975b097fa66SStefano Zampini       ierr = KSPSolveTranspose(pcbddc->ksp_D,pcis->vec1_D,pcis->vec2_D);CHKERRQ(ierr);
197650efa1b5SStefano Zampini       ierr = VecScale(pcis->vec2_D,m_one);CHKERRQ(ierr);
197716909a7fSStefano Zampini       if (pcbddc->switch_static) {
197816909a7fSStefano Zampini         Mat_IS *matis = (Mat_IS*)(pc->mat->data);
197916909a7fSStefano Zampini 
198016909a7fSStefano Zampini         ierr = VecSet(pcis->vec1_N,0.);CHKERRQ(ierr);
198116909a7fSStefano Zampini         ierr = VecScatterBegin(pcis->N_to_D,pcis->vec2_D,pcis->vec1_N,INSERT_VALUES,SCATTER_REVERSE);CHKERRQ(ierr);
198216909a7fSStefano Zampini         ierr = VecScatterEnd(pcis->N_to_D,pcis->vec2_D,pcis->vec1_N,INSERT_VALUES,SCATTER_REVERSE);CHKERRQ(ierr);
198316909a7fSStefano Zampini         if (!pcbddc->switch_static_change) {
198416909a7fSStefano Zampini           ierr = MatMultTranspose(matis->A,pcis->vec1_N,pcis->vec2_N);CHKERRQ(ierr);
198516909a7fSStefano Zampini         } else {
198616909a7fSStefano Zampini           ierr = MatMult(pcbddc->switch_static_change,pcis->vec1_N,pcis->vec2_N);CHKERRQ(ierr);
198716909a7fSStefano Zampini           ierr = MatMultTranspose(matis->A,pcis->vec2_N,pcis->vec1_N);CHKERRQ(ierr);
198816909a7fSStefano Zampini           ierr = MatMultTranspose(pcbddc->switch_static_change,pcis->vec1_N,pcis->vec2_N);CHKERRQ(ierr);
198916909a7fSStefano Zampini         }
199016909a7fSStefano Zampini         ierr = VecScatterBegin(pcis->N_to_D,pcis->vec2_N,pcis->vec1_D,ADD_VALUES,SCATTER_FORWARD);CHKERRQ(ierr);
199116909a7fSStefano Zampini         ierr = VecScatterEnd(pcis->N_to_D,pcis->vec2_N,pcis->vec1_D,ADD_VALUES,SCATTER_FORWARD);CHKERRQ(ierr);
199216909a7fSStefano Zampini         ierr = VecScatterBegin(pcis->N_to_B,pcis->vec2_N,pcis->vec1_B,INSERT_VALUES,SCATTER_FORWARD);CHKERRQ(ierr);
199316909a7fSStefano Zampini         ierr = VecScatterEnd(pcis->N_to_B,pcis->vec2_N,pcis->vec1_B,INSERT_VALUES,SCATTER_FORWARD);CHKERRQ(ierr);
199416909a7fSStefano Zampini       } else {
1995b097fa66SStefano Zampini         ierr = MatMultTranspose(pcis->A_IB,pcis->vec2_D,pcis->vec1_B);CHKERRQ(ierr);
199616909a7fSStefano Zampini       }
1997b097fa66SStefano Zampini     } else {
1998b097fa66SStefano Zampini       ierr = VecSet(pcis->vec1_B,zero);CHKERRQ(ierr);
1999b097fa66SStefano Zampini     }
200050efa1b5SStefano Zampini     ierr = VecScatterBegin(pcis->global_to_B,pcis->vec1_B,z,ADD_VALUES,SCATTER_REVERSE);CHKERRQ(ierr);
200150efa1b5SStefano Zampini     ierr = VecScatterEnd(pcis->global_to_B,pcis->vec1_B,z,ADD_VALUES,SCATTER_REVERSE);CHKERRQ(ierr);
200250efa1b5SStefano Zampini     ierr = PCBDDCScalingRestriction(pc,z,pcis->vec1_B);CHKERRQ(ierr);
200350efa1b5SStefano Zampini   } else {
200450efa1b5SStefano Zampini     ierr = PCBDDCScalingRestriction(pc,r,pcis->vec1_B);CHKERRQ(ierr);
200550efa1b5SStefano Zampini   }
200650efa1b5SStefano Zampini 
200750efa1b5SStefano Zampini   /* Apply interface preconditioner
200850efa1b5SStefano Zampini      input/output vecs: pcis->vec1_B and pcis->vec1_D */
2009dc359a40SStefano Zampini   ierr = PCBDDCApplyInterfacePreconditioner(pc,PETSC_TRUE);CHKERRQ(ierr);
201050efa1b5SStefano Zampini 
201150efa1b5SStefano Zampini   /* Apply transpose of partition of unity operator */
201250efa1b5SStefano Zampini   ierr = PCBDDCScalingExtension(pc,pcis->vec1_B,z);CHKERRQ(ierr);
201350efa1b5SStefano Zampini 
201450efa1b5SStefano Zampini   /* Second Dirichlet solve and assembling of output */
201550efa1b5SStefano Zampini   ierr = VecScatterBegin(pcis->global_to_B,z,pcis->vec1_B,INSERT_VALUES,SCATTER_FORWARD);CHKERRQ(ierr);
201650efa1b5SStefano Zampini   ierr = VecScatterEnd(pcis->global_to_B,z,pcis->vec1_B,INSERT_VALUES,SCATTER_FORWARD);CHKERRQ(ierr);
2017b097fa66SStefano Zampini   if (n_B) {
201816909a7fSStefano Zampini     if (pcbddc->switch_static) {
201916909a7fSStefano Zampini       Mat_IS *matis = (Mat_IS*)(pc->mat->data);
202016909a7fSStefano Zampini 
202116909a7fSStefano Zampini       ierr = VecScatterBegin(pcis->N_to_D,pcis->vec1_D,pcis->vec1_N,INSERT_VALUES,SCATTER_REVERSE);CHKERRQ(ierr);
202216909a7fSStefano Zampini       ierr = VecScatterEnd(pcis->N_to_D,pcis->vec1_D,pcis->vec1_N,INSERT_VALUES,SCATTER_REVERSE);CHKERRQ(ierr);
202316909a7fSStefano Zampini       ierr = VecScatterBegin(pcis->N_to_B,pcis->vec1_B,pcis->vec1_N,INSERT_VALUES,SCATTER_REVERSE);CHKERRQ(ierr);
202416909a7fSStefano Zampini       ierr = VecScatterEnd(pcis->N_to_B,pcis->vec1_B,pcis->vec1_N,INSERT_VALUES,SCATTER_REVERSE);CHKERRQ(ierr);
202516909a7fSStefano Zampini       if (!pcbddc->switch_static_change) {
202616909a7fSStefano Zampini         ierr = MatMultTranspose(matis->A,pcis->vec1_N,pcis->vec2_N);CHKERRQ(ierr);
202716909a7fSStefano Zampini       } else {
202816909a7fSStefano Zampini         ierr = MatMult(pcbddc->switch_static_change,pcis->vec1_N,pcis->vec2_N);CHKERRQ(ierr);
202916909a7fSStefano Zampini         ierr = MatMultTranspose(matis->A,pcis->vec2_N,pcis->vec1_N);CHKERRQ(ierr);
203016909a7fSStefano Zampini         ierr = MatMultTranspose(pcbddc->switch_static_change,pcis->vec1_N,pcis->vec2_N);CHKERRQ(ierr);
203116909a7fSStefano Zampini       }
203216909a7fSStefano Zampini       ierr = VecScatterBegin(pcis->N_to_D,pcis->vec2_N,pcis->vec3_D,INSERT_VALUES,SCATTER_FORWARD);CHKERRQ(ierr);
203316909a7fSStefano Zampini       ierr = VecScatterEnd(pcis->N_to_D,pcis->vec2_N,pcis->vec3_D,INSERT_VALUES,SCATTER_FORWARD);CHKERRQ(ierr);
203416909a7fSStefano Zampini     } else {
203550efa1b5SStefano Zampini       ierr = MatMultTranspose(pcis->A_BI,pcis->vec1_B,pcis->vec3_D);CHKERRQ(ierr);
203616909a7fSStefano Zampini     }
203716909a7fSStefano Zampini   } else if (pcbddc->switch_static) { /* n_B is zero */
203816909a7fSStefano Zampini     Mat_IS *matis = (Mat_IS*)(pc->mat->data);
203916909a7fSStefano Zampini 
204016909a7fSStefano Zampini     if (!pcbddc->switch_static_change) {
204116909a7fSStefano Zampini       ierr = MatMultTranspose(matis->A,pcis->vec1_D,pcis->vec3_D);CHKERRQ(ierr);
204216909a7fSStefano Zampini     } else {
204316909a7fSStefano Zampini       ierr = MatMult(pcbddc->switch_static_change,pcis->vec1_D,pcis->vec1_N);CHKERRQ(ierr);
204416909a7fSStefano Zampini       ierr = MatMultTranspose(matis->A,pcis->vec1_N,pcis->vec2_N);CHKERRQ(ierr);
204516909a7fSStefano Zampini       ierr = MatMultTranspose(pcbddc->switch_static_change,pcis->vec2_N,pcis->vec3_D);CHKERRQ(ierr);
204616909a7fSStefano Zampini     }
2047b097fa66SStefano Zampini   }
2048b0147a47SStefano Zampini   ierr = KSPSolveTranspose(pcbddc->ksp_D,pcis->vec3_D,pcis->vec4_D);CHKERRQ(ierr);
20498ae0ca82SStefano Zampini   if (!pcbddc->exact_dirichlet_trick_app && !pcbddc->benign_apply_coarse_only) {
2050b097fa66SStefano Zampini     if (pcbddc->switch_static) {
2051b097fa66SStefano Zampini       ierr = VecAXPBYPCZ(pcis->vec2_D,m_one,one,m_one,pcis->vec4_D,pcis->vec1_D);CHKERRQ(ierr);
2052b097fa66SStefano Zampini     } else {
2053b097fa66SStefano Zampini       ierr = VecAXPBY(pcis->vec2_D,m_one,m_one,pcis->vec4_D);CHKERRQ(ierr);
2054b097fa66SStefano Zampini     }
205550efa1b5SStefano Zampini     ierr = VecScatterBegin(pcis->global_to_D,pcis->vec2_D,z,INSERT_VALUES,SCATTER_REVERSE);CHKERRQ(ierr);
205650efa1b5SStefano Zampini     ierr = VecScatterEnd(pcis->global_to_D,pcis->vec2_D,z,INSERT_VALUES,SCATTER_REVERSE);CHKERRQ(ierr);
2057b097fa66SStefano Zampini   } else {
2058b097fa66SStefano Zampini     if (pcbddc->switch_static) {
2059b097fa66SStefano Zampini       ierr = VecAXPBY(pcis->vec4_D,one,m_one,pcis->vec1_D);CHKERRQ(ierr);
2060b097fa66SStefano Zampini     } else {
2061b097fa66SStefano Zampini       ierr = VecScale(pcis->vec4_D,m_one);CHKERRQ(ierr);
2062b097fa66SStefano Zampini     }
2063b097fa66SStefano Zampini     ierr = VecScatterBegin(pcis->global_to_D,pcis->vec4_D,z,INSERT_VALUES,SCATTER_REVERSE);CHKERRQ(ierr);
2064b097fa66SStefano Zampini     ierr = VecScatterEnd(pcis->global_to_D,pcis->vec4_D,z,INSERT_VALUES,SCATTER_REVERSE);CHKERRQ(ierr);
2065b097fa66SStefano Zampini   }
206627b6a85dSStefano Zampini   if (pcbddc->benign_have_null) { /* set p0 (computed in PCBDDCApplyInterface) */
2067537c1cdfSStefano Zampini     ierr = PCBDDCBenignGetOrSetP0(pc,z,PETSC_FALSE);CHKERRQ(ierr);
2068537c1cdfSStefano Zampini   }
20691dd7afcfSStefano Zampini   if (pcbddc->ChangeOfBasisMatrix) {
2070f913dca9SStefano Zampini     pcbddc->work_change = r;
20711dd7afcfSStefano Zampini     ierr = VecCopy(z,pcbddc->work_change);CHKERRQ(ierr);
20721dd7afcfSStefano Zampini     ierr = MatMult(pcbddc->ChangeOfBasisMatrix,pcbddc->work_change,z);CHKERRQ(ierr);
20731dd7afcfSStefano Zampini   }
207450efa1b5SStefano Zampini   PetscFunctionReturn(0);
207550efa1b5SStefano Zampini }
2076674ae819SStefano Zampini 
20779326c5c6Sstefano_zampini PetscErrorCode PCReset_BDDC(PC pc)
2078da1bb401SStefano Zampini {
2079da1bb401SStefano Zampini   PC_BDDC        *pcbddc = (PC_BDDC*)pc->data;
20809326c5c6Sstefano_zampini   PC_IS          *pcis = (PC_IS*)pc->data;
20819326c5c6Sstefano_zampini   KSP            kspD,kspR,kspC;
2082da1bb401SStefano Zampini   PetscErrorCode ierr;
2083da1bb401SStefano Zampini 
2084da1bb401SStefano Zampini   PetscFunctionBegin;
2085674ae819SStefano Zampini   /* free BDDC custom data  */
2086674ae819SStefano Zampini   ierr = PCBDDCResetCustomization(pc);CHKERRQ(ierr);
2087674ae819SStefano Zampini   /* destroy objects related to topography */
2088674ae819SStefano Zampini   ierr = PCBDDCResetTopography(pc);CHKERRQ(ierr);
208934a97f8cSStefano Zampini   /* destroy objects for scaling operator */
2090674ae819SStefano Zampini   ierr = PCBDDCScalingDestroy(pc);CHKERRQ(ierr);
2091674ae819SStefano Zampini   /* free solvers stuff */
2092674ae819SStefano Zampini   ierr = PCBDDCResetSolvers(pc);CHKERRQ(ierr);
209362a6ff1dSStefano Zampini   /* free global vectors needed in presolve */
209462a6ff1dSStefano Zampini   ierr = VecDestroy(&pcbddc->temp_solution);CHKERRQ(ierr);
209562a6ff1dSStefano Zampini   ierr = VecDestroy(&pcbddc->original_rhs);CHKERRQ(ierr);
20961dd7afcfSStefano Zampini   /* free data created by PCIS */
20971dd7afcfSStefano Zampini   ierr = PCISDestroy(pc);CHKERRQ(ierr);
20989326c5c6Sstefano_zampini 
20999326c5c6Sstefano_zampini   /* restore defaults */
21009326c5c6Sstefano_zampini   kspD = pcbddc->ksp_D;
21019326c5c6Sstefano_zampini   kspR = pcbddc->ksp_R;
21029326c5c6Sstefano_zampini   kspC = pcbddc->coarse_ksp;
21039326c5c6Sstefano_zampini   ierr = PetscMemzero(pc->data,sizeof(*pcbddc));CHKERRQ(ierr);
21049326c5c6Sstefano_zampini   pcis->n_neigh                     = -1;
21059326c5c6Sstefano_zampini   pcis->scaling_factor              = 1.0;
21069326c5c6Sstefano_zampini   pcis->reusesubmatrices            = PETSC_TRUE;
21079326c5c6Sstefano_zampini   pcbddc->use_local_adj             = PETSC_TRUE;
21089326c5c6Sstefano_zampini   pcbddc->use_vertices              = PETSC_TRUE;
21099326c5c6Sstefano_zampini   pcbddc->use_edges                 = PETSC_TRUE;
21109326c5c6Sstefano_zampini   pcbddc->symmetric_primal          = PETSC_TRUE;
21119326c5c6Sstefano_zampini   pcbddc->vertex_size               = 1;
21129326c5c6Sstefano_zampini   pcbddc->recompute_topography      = PETSC_TRUE;
21139326c5c6Sstefano_zampini   pcbddc->coarse_size               = -1;
21149326c5c6Sstefano_zampini   pcbddc->use_exact_dirichlet_trick = PETSC_TRUE;
21159326c5c6Sstefano_zampini   pcbddc->coarsening_ratio          = 8;
21169326c5c6Sstefano_zampini   pcbddc->coarse_eqs_per_proc       = 1;
21179326c5c6Sstefano_zampini   pcbddc->benign_compute_correction = PETSC_TRUE;
21189326c5c6Sstefano_zampini   pcbddc->nedfield                  = -1;
21199326c5c6Sstefano_zampini   pcbddc->nedglobal                 = PETSC_TRUE;
21209326c5c6Sstefano_zampini   pcbddc->graphmaxcount             = PETSC_MAX_INT;
21219326c5c6Sstefano_zampini   pcbddc->sub_schurs_layers         = -1;
21229326c5c6Sstefano_zampini   pcbddc->ksp_D                     = kspD;
21239326c5c6Sstefano_zampini   pcbddc->ksp_R                     = kspR;
21249326c5c6Sstefano_zampini   pcbddc->coarse_ksp                = kspC;
21259326c5c6Sstefano_zampini   PetscFunctionReturn(0);
21269326c5c6Sstefano_zampini }
21279326c5c6Sstefano_zampini 
21289326c5c6Sstefano_zampini PetscErrorCode PCDestroy_BDDC(PC pc)
21299326c5c6Sstefano_zampini {
21309326c5c6Sstefano_zampini   PC_BDDC        *pcbddc = (PC_BDDC*)pc->data;
21319326c5c6Sstefano_zampini   PetscErrorCode ierr;
21329326c5c6Sstefano_zampini 
21339326c5c6Sstefano_zampini   PetscFunctionBegin;
21349326c5c6Sstefano_zampini   ierr = PCReset_BDDC(pc);CHKERRQ(ierr);
21359326c5c6Sstefano_zampini   ierr = KSPDestroy(&pcbddc->ksp_D);CHKERRQ(ierr);
21369326c5c6Sstefano_zampini   ierr = KSPDestroy(&pcbddc->ksp_R);CHKERRQ(ierr);
21379326c5c6Sstefano_zampini   ierr = KSPDestroy(&pcbddc->coarse_ksp);CHKERRQ(ierr);
2138a13144ffSStefano Zampini   ierr = PetscObjectComposeFunction((PetscObject)pc,"PCBDDCSetDiscreteGradient_C",NULL);CHKERRQ(ierr);
2139a198735bSStefano Zampini   ierr = PetscObjectComposeFunction((PetscObject)pc,"PCBDDCSetDivergenceMat_C",NULL);CHKERRQ(ierr);
2140906d46d4SStefano Zampini   ierr = PetscObjectComposeFunction((PetscObject)pc,"PCBDDCSetChangeOfBasisMat_C",NULL);CHKERRQ(ierr);
2141674ae819SStefano Zampini   ierr = PetscObjectComposeFunction((PetscObject)pc,"PCBDDCSetPrimalVerticesLocalIS_C",NULL);CHKERRQ(ierr);
214230368db7SStefano Zampini   ierr = PetscObjectComposeFunction((PetscObject)pc,"PCBDDCSetPrimalVerticesIS_C",NULL);CHKERRQ(ierr);
2143bdf89e91SBarry Smith   ierr = PetscObjectComposeFunction((PetscObject)pc,"PCBDDCSetCoarseningRatio_C",NULL);CHKERRQ(ierr);
21442b510759SStefano Zampini   ierr = PetscObjectComposeFunction((PetscObject)pc,"PCBDDCSetLevel_C",NULL);CHKERRQ(ierr);
2145b8ffe317SStefano Zampini   ierr = PetscObjectComposeFunction((PetscObject)pc,"PCBDDCSetUseExactDirichlet_C",NULL);CHKERRQ(ierr);
21462b510759SStefano Zampini   ierr = PetscObjectComposeFunction((PetscObject)pc,"PCBDDCSetLevels_C",NULL);CHKERRQ(ierr);
2147bdf89e91SBarry Smith   ierr = PetscObjectComposeFunction((PetscObject)pc,"PCBDDCSetDirichletBoundaries_C",NULL);CHKERRQ(ierr);
214882ba6b80SStefano Zampini   ierr = PetscObjectComposeFunction((PetscObject)pc,"PCBDDCSetDirichletBoundariesLocal_C",NULL);CHKERRQ(ierr);
2149bdf89e91SBarry Smith   ierr = PetscObjectComposeFunction((PetscObject)pc,"PCBDDCSetNeumannBoundaries_C",NULL);CHKERRQ(ierr);
215082ba6b80SStefano Zampini   ierr = PetscObjectComposeFunction((PetscObject)pc,"PCBDDCSetNeumannBoundariesLocal_C",NULL);CHKERRQ(ierr);
2151bdf89e91SBarry Smith   ierr = PetscObjectComposeFunction((PetscObject)pc,"PCBDDCGetDirichletBoundaries_C",NULL);CHKERRQ(ierr);
215282ba6b80SStefano Zampini   ierr = PetscObjectComposeFunction((PetscObject)pc,"PCBDDCGetDirichletBoundariesLocal_C",NULL);CHKERRQ(ierr);
2153bdf89e91SBarry Smith   ierr = PetscObjectComposeFunction((PetscObject)pc,"PCBDDCGetNeumannBoundaries_C",NULL);CHKERRQ(ierr);
2154785d1243SStefano Zampini   ierr = PetscObjectComposeFunction((PetscObject)pc,"PCBDDCGetNeumannBoundariesLocal_C",NULL);CHKERRQ(ierr);
2155bdf89e91SBarry Smith   ierr = PetscObjectComposeFunction((PetscObject)pc,"PCBDDCSetDofsSplitting_C",NULL);CHKERRQ(ierr);
215663602bcaSStefano Zampini   ierr = PetscObjectComposeFunction((PetscObject)pc,"PCBDDCSetDofsSplittingLocal_C",NULL);CHKERRQ(ierr);
2157bdf89e91SBarry Smith   ierr = PetscObjectComposeFunction((PetscObject)pc,"PCBDDCSetLocalAdjacencyGraph_C",NULL);CHKERRQ(ierr);
2158bdf89e91SBarry Smith   ierr = PetscObjectComposeFunction((PetscObject)pc,"PCBDDCCreateFETIDPOperators_C",NULL);CHKERRQ(ierr);
2159bdf89e91SBarry Smith   ierr = PetscObjectComposeFunction((PetscObject)pc,"PCBDDCMatFETIDPGetRHS_C",NULL);CHKERRQ(ierr);
2160bdf89e91SBarry Smith   ierr = PetscObjectComposeFunction((PetscObject)pc,"PCBDDCMatFETIDPGetSolution_C",NULL);CHKERRQ(ierr);
2161a06fd7f2SStefano Zampini   ierr = PetscObjectComposeFunction((PetscObject)pc,"PCPreSolveChangeRHS_C",NULL);CHKERRQ(ierr);
2162674ae819SStefano Zampini   ierr = PetscFree(pc->data);CHKERRQ(ierr);
2163da1bb401SStefano Zampini   PetscFunctionReturn(0);
2164da1bb401SStefano Zampini }
21651e6b0712SBarry Smith 
2166a06fd7f2SStefano Zampini static PetscErrorCode PCPreSolveChangeRHS_BDDC(PC pc, PetscBool* change)
2167a06fd7f2SStefano Zampini {
2168a06fd7f2SStefano Zampini   PetscFunctionBegin;
2169a06fd7f2SStefano Zampini   *change = PETSC_TRUE;
2170a06fd7f2SStefano Zampini   PetscFunctionReturn(0);
2171a06fd7f2SStefano Zampini }
2172a06fd7f2SStefano Zampini 
21733425bc38SStefano Zampini static PetscErrorCode PCBDDCMatFETIDPGetRHS_BDDC(Mat fetidp_mat, Vec standard_rhs, Vec fetidp_flux_rhs)
21743425bc38SStefano Zampini {
2175674ae819SStefano Zampini   FETIDPMat_ctx  mat_ctx;
2176266e20e9SStefano Zampini   Vec            work;
21773425bc38SStefano Zampini   PC_IS*         pcis;
21783425bc38SStefano Zampini   PC_BDDC*       pcbddc;
21793425bc38SStefano Zampini   PetscErrorCode ierr;
21800c7d97c5SJed Brown 
21813425bc38SStefano Zampini   PetscFunctionBegin;
21823425bc38SStefano Zampini   ierr = MatShellGetContext(fetidp_mat,&mat_ctx);CHKERRQ(ierr);
21833425bc38SStefano Zampini   pcis = (PC_IS*)mat_ctx->pc->data;
21843425bc38SStefano Zampini   pcbddc = (PC_BDDC*)mat_ctx->pc->data;
21853425bc38SStefano Zampini 
2186229984c5Sstefano_zampini   ierr = VecSet(fetidp_flux_rhs,0.0);CHKERRQ(ierr);
2187229984c5Sstefano_zampini   /* copy rhs since we may change it during PCPreSolve_BDDC */
2188229984c5Sstefano_zampini   if (!pcbddc->original_rhs) {
2189229984c5Sstefano_zampini     ierr = VecDuplicate(pcis->vec1_global,&pcbddc->original_rhs);CHKERRQ(ierr);
2190229984c5Sstefano_zampini   }
21916cc1294bSstefano_zampini   if (mat_ctx->rhs_flip) {
21926cc1294bSstefano_zampini     ierr = VecPointwiseMult(pcbddc->original_rhs,standard_rhs,mat_ctx->rhs_flip);CHKERRQ(ierr);
21936cc1294bSstefano_zampini   } else {
2194229984c5Sstefano_zampini     ierr = VecCopy(standard_rhs,pcbddc->original_rhs);CHKERRQ(ierr);
21956cc1294bSstefano_zampini   }
2196af140850Sstefano_zampini   if (mat_ctx->g2g_p) {
2197229984c5Sstefano_zampini     /* interface pressure rhs */
2198022d8d2bSstefano_zampini     ierr = VecScatterBegin(mat_ctx->g2g_p,fetidp_flux_rhs,pcbddc->original_rhs,INSERT_VALUES,SCATTER_REVERSE);CHKERRQ(ierr);
2199022d8d2bSstefano_zampini     ierr = VecScatterEnd(mat_ctx->g2g_p,fetidp_flux_rhs,pcbddc->original_rhs,INSERT_VALUES,SCATTER_REVERSE);CHKERRQ(ierr);
2200229984c5Sstefano_zampini     ierr = VecScatterBegin(mat_ctx->g2g_p,standard_rhs,fetidp_flux_rhs,INSERT_VALUES,SCATTER_FORWARD);CHKERRQ(ierr);
2201229984c5Sstefano_zampini     ierr = VecScatterEnd(mat_ctx->g2g_p,standard_rhs,fetidp_flux_rhs,INSERT_VALUES,SCATTER_FORWARD);CHKERRQ(ierr);
22026cc1294bSstefano_zampini     if (!mat_ctx->rhs_flip) {
2203229984c5Sstefano_zampini       ierr = VecScale(fetidp_flux_rhs,-1.);CHKERRQ(ierr);
2204229984c5Sstefano_zampini     }
22056cc1294bSstefano_zampini   }
2206c08af4c6SStefano Zampini   /*
2207c08af4c6SStefano Zampini      change of basis for physical rhs if needed
2208c08af4c6SStefano Zampini      It also changes the rhs in case of dirichlet boundaries
2209c08af4c6SStefano Zampini   */
22103738a8e6SStefano Zampini   ierr = PCPreSolve_BDDC(mat_ctx->pc,NULL,pcbddc->original_rhs,NULL);CHKERRQ(ierr);
2211fc17d649SStefano Zampini   if (pcbddc->ChangeOfBasisMatrix) {
22123738a8e6SStefano Zampini     ierr = MatMultTranspose(pcbddc->ChangeOfBasisMatrix,pcbddc->original_rhs,pcbddc->work_change);CHKERRQ(ierr);
22133738a8e6SStefano Zampini     work = pcbddc->work_change;
2214fc17d649SStefano Zampini    } else {
22153738a8e6SStefano Zampini     work = pcbddc->original_rhs;
2216fc17d649SStefano Zampini   }
22173425bc38SStefano Zampini   /* store vectors for computation of fetidp final solution */
2218266e20e9SStefano Zampini   ierr = VecScatterBegin(pcis->global_to_D,work,mat_ctx->temp_solution_D,INSERT_VALUES,SCATTER_FORWARD);CHKERRQ(ierr);
2219266e20e9SStefano Zampini   ierr = VecScatterEnd(pcis->global_to_D,work,mat_ctx->temp_solution_D,INSERT_VALUES,SCATTER_FORWARD);CHKERRQ(ierr);
2220fb223d50SStefano Zampini   /* scale rhs since it should be unassembled */
2221fb223d50SStefano Zampini   /* TODO use counter scaling? (also below) */
2222266e20e9SStefano Zampini   ierr = VecScatterBegin(pcis->global_to_B,work,mat_ctx->temp_solution_B,INSERT_VALUES,SCATTER_FORWARD);CHKERRQ(ierr);
2223266e20e9SStefano Zampini   ierr = VecScatterEnd(pcis->global_to_B,work,mat_ctx->temp_solution_B,INSERT_VALUES,SCATTER_FORWARD);CHKERRQ(ierr);
2224674ae819SStefano Zampini   /* Apply partition of unity */
22253425bc38SStefano Zampini   ierr = VecPointwiseMult(mat_ctx->temp_solution_B,pcis->D,mat_ctx->temp_solution_B);CHKERRQ(ierr);
2226266e20e9SStefano Zampini   /* ierr = PCBDDCScalingRestriction(mat_ctx->pc,work,mat_ctx->temp_solution_B);CHKERRQ(ierr); */
22278eeda7d8SStefano Zampini   if (!pcbddc->switch_static) {
22283425bc38SStefano Zampini     /* compute partially subassembled Schur complement right-hand side */
22293425bc38SStefano Zampini     ierr = KSPSolve(pcbddc->ksp_D,mat_ctx->temp_solution_D,pcis->vec1_D);CHKERRQ(ierr);
22303425bc38SStefano Zampini     ierr = MatMult(pcis->A_BI,pcis->vec1_D,pcis->vec1_B);CHKERRQ(ierr);
22313425bc38SStefano Zampini     ierr = VecAXPY(mat_ctx->temp_solution_B,-1.0,pcis->vec1_B);CHKERRQ(ierr);
2232266e20e9SStefano Zampini     ierr = VecSet(work,0.0);CHKERRQ(ierr);
2233266e20e9SStefano Zampini     ierr = VecScatterBegin(pcis->global_to_B,mat_ctx->temp_solution_B,work,ADD_VALUES,SCATTER_REVERSE);CHKERRQ(ierr);
2234266e20e9SStefano Zampini     ierr = VecScatterEnd(pcis->global_to_B,mat_ctx->temp_solution_B,work,ADD_VALUES,SCATTER_REVERSE);CHKERRQ(ierr);
2235266e20e9SStefano Zampini     /* ierr = PCBDDCScalingRestriction(mat_ctx->pc,work,mat_ctx->temp_solution_B);CHKERRQ(ierr); */
2236266e20e9SStefano Zampini     ierr = VecScatterBegin(pcis->global_to_B,work,mat_ctx->temp_solution_B,INSERT_VALUES,SCATTER_FORWARD);CHKERRQ(ierr);
2237266e20e9SStefano Zampini     ierr = VecScatterEnd(pcis->global_to_B,work,mat_ctx->temp_solution_B,INSERT_VALUES,SCATTER_FORWARD);CHKERRQ(ierr);
22383425bc38SStefano Zampini     ierr = VecPointwiseMult(mat_ctx->temp_solution_B,pcis->D,mat_ctx->temp_solution_B);CHKERRQ(ierr);
22393425bc38SStefano Zampini   }
22403425bc38SStefano Zampini   /* BDDC rhs */
22413425bc38SStefano Zampini   ierr = VecCopy(mat_ctx->temp_solution_B,pcis->vec1_B);CHKERRQ(ierr);
22428eeda7d8SStefano Zampini   if (pcbddc->switch_static) {
22433425bc38SStefano Zampini     ierr = VecCopy(mat_ctx->temp_solution_D,pcis->vec1_D);CHKERRQ(ierr);
22443425bc38SStefano Zampini   }
22453425bc38SStefano Zampini   /* apply BDDC */
2246229984c5Sstefano_zampini   ierr = PetscMemzero(pcbddc->benign_p0,pcbddc->benign_n*sizeof(PetscScalar));CHKERRQ(ierr);
2247dc359a40SStefano Zampini   ierr = PCBDDCApplyInterfacePreconditioner(mat_ctx->pc,PETSC_FALSE);CHKERRQ(ierr);
2248266e20e9SStefano Zampini   ierr = PetscMemzero(pcbddc->benign_p0,pcbddc->benign_n*sizeof(PetscScalar));CHKERRQ(ierr);
2249229984c5Sstefano_zampini 
22503425bc38SStefano Zampini   /* Application of B_delta and assembling of rhs for fetidp fluxes */
22513425bc38SStefano Zampini   ierr = MatMult(mat_ctx->B_delta,pcis->vec1_B,mat_ctx->lambda_local);CHKERRQ(ierr);
22523425bc38SStefano Zampini   ierr = VecScatterBegin(mat_ctx->l2g_lambda,mat_ctx->lambda_local,fetidp_flux_rhs,ADD_VALUES,SCATTER_FORWARD);CHKERRQ(ierr);
22533425bc38SStefano Zampini   ierr = VecScatterEnd(mat_ctx->l2g_lambda,mat_ctx->lambda_local,fetidp_flux_rhs,ADD_VALUES,SCATTER_FORWARD);CHKERRQ(ierr);
2254229984c5Sstefano_zampini   /* Add contribution to interface pressures */
2255229984c5Sstefano_zampini   if (mat_ctx->l2g_p) {
2256229984c5Sstefano_zampini     ierr = MatMult(mat_ctx->B_BB,pcis->vec1_B,mat_ctx->vP);CHKERRQ(ierr);
2257229984c5Sstefano_zampini     if (pcbddc->switch_static) {
2258229984c5Sstefano_zampini       ierr = MatMultAdd(mat_ctx->B_BI,pcis->vec1_D,mat_ctx->vP,mat_ctx->vP);CHKERRQ(ierr);
2259229984c5Sstefano_zampini     }
2260229984c5Sstefano_zampini     ierr = VecScatterBegin(mat_ctx->l2g_p,mat_ctx->vP,fetidp_flux_rhs,ADD_VALUES,SCATTER_FORWARD);CHKERRQ(ierr);
2261229984c5Sstefano_zampini     ierr = VecScatterEnd(mat_ctx->l2g_p,mat_ctx->vP,fetidp_flux_rhs,ADD_VALUES,SCATTER_FORWARD);CHKERRQ(ierr);
2262229984c5Sstefano_zampini   }
22633425bc38SStefano Zampini   PetscFunctionReturn(0);
22643425bc38SStefano Zampini }
22651e6b0712SBarry Smith 
22663425bc38SStefano Zampini /*@
22670f202f7eSStefano Zampini  PCBDDCMatFETIDPGetRHS - Compute the right-hand side for FETI-DP linear system using the physical right-hand side
22683425bc38SStefano Zampini 
22693425bc38SStefano Zampini    Collective
22703425bc38SStefano Zampini 
22713425bc38SStefano Zampini    Input Parameters:
22720f202f7eSStefano Zampini +  fetidp_mat      - the FETI-DP matrix object obtained by a call to PCBDDCCreateFETIDPOperators
22730f202f7eSStefano Zampini -  standard_rhs    - the right-hand side of the original linear system
22743425bc38SStefano Zampini 
22753425bc38SStefano Zampini    Output Parameters:
22760f202f7eSStefano Zampini .  fetidp_flux_rhs - the right-hand side for the FETI-DP linear system
22773425bc38SStefano Zampini 
22783425bc38SStefano Zampini    Level: developer
22793425bc38SStefano Zampini 
22803425bc38SStefano Zampini    Notes:
22813425bc38SStefano Zampini 
22820f202f7eSStefano Zampini .seealso: PCBDDC, PCBDDCCreateFETIDPOperators, PCBDDCMatFETIDPGetSolution
22833425bc38SStefano Zampini @*/
22843425bc38SStefano Zampini PetscErrorCode PCBDDCMatFETIDPGetRHS(Mat fetidp_mat, Vec standard_rhs, Vec fetidp_flux_rhs)
22853425bc38SStefano Zampini {
2286674ae819SStefano Zampini   FETIDPMat_ctx  mat_ctx;
22873425bc38SStefano Zampini   PetscErrorCode ierr;
22883425bc38SStefano Zampini 
22893425bc38SStefano Zampini   PetscFunctionBegin;
2290266e20e9SStefano Zampini   PetscValidHeaderSpecific(fetidp_mat,MAT_CLASSID,1);
2291266e20e9SStefano Zampini   PetscValidHeaderSpecific(standard_rhs,VEC_CLASSID,2);
2292266e20e9SStefano Zampini   PetscValidHeaderSpecific(fetidp_flux_rhs,VEC_CLASSID,3);
22933425bc38SStefano Zampini   ierr = MatShellGetContext(fetidp_mat,&mat_ctx);CHKERRQ(ierr);
2294163d334eSBarry Smith   ierr = PetscUseMethod(mat_ctx->pc,"PCBDDCMatFETIDPGetRHS_C",(Mat,Vec,Vec),(fetidp_mat,standard_rhs,fetidp_flux_rhs));CHKERRQ(ierr);
22953425bc38SStefano Zampini   PetscFunctionReturn(0);
22963425bc38SStefano Zampini }
22971e6b0712SBarry Smith 
22983425bc38SStefano Zampini static PetscErrorCode PCBDDCMatFETIDPGetSolution_BDDC(Mat fetidp_mat, Vec fetidp_flux_sol, Vec standard_sol)
22993425bc38SStefano Zampini {
2300674ae819SStefano Zampini   FETIDPMat_ctx  mat_ctx;
23013425bc38SStefano Zampini   PC_IS*         pcis;
23023425bc38SStefano Zampini   PC_BDDC*       pcbddc;
23033425bc38SStefano Zampini   PetscErrorCode ierr;
2304229984c5Sstefano_zampini   Vec            work;
23053425bc38SStefano Zampini 
23063425bc38SStefano Zampini   PetscFunctionBegin;
23073425bc38SStefano Zampini   ierr = MatShellGetContext(fetidp_mat,&mat_ctx);CHKERRQ(ierr);
23083425bc38SStefano Zampini   pcis = (PC_IS*)mat_ctx->pc->data;
23093425bc38SStefano Zampini   pcbddc = (PC_BDDC*)mat_ctx->pc->data;
23103425bc38SStefano Zampini 
23113425bc38SStefano Zampini   /* apply B_delta^T */
2312af140850Sstefano_zampini   ierr = VecSet(pcis->vec1_B,0.);CHKERRQ(ierr);
23133425bc38SStefano Zampini   ierr = VecScatterBegin(mat_ctx->l2g_lambda,fetidp_flux_sol,mat_ctx->lambda_local,INSERT_VALUES,SCATTER_REVERSE);CHKERRQ(ierr);
23143425bc38SStefano Zampini   ierr = VecScatterEnd(mat_ctx->l2g_lambda,fetidp_flux_sol,mat_ctx->lambda_local,INSERT_VALUES,SCATTER_REVERSE);CHKERRQ(ierr);
23153425bc38SStefano Zampini   ierr = MatMultTranspose(mat_ctx->B_delta,mat_ctx->lambda_local,pcis->vec1_B);CHKERRQ(ierr);
2316229984c5Sstefano_zampini   if (mat_ctx->l2g_p) {
2317229984c5Sstefano_zampini     ierr = VecScatterBegin(mat_ctx->l2g_p,fetidp_flux_sol,mat_ctx->vP,INSERT_VALUES,SCATTER_REVERSE);CHKERRQ(ierr);
2318229984c5Sstefano_zampini     ierr = VecScatterEnd(mat_ctx->l2g_p,fetidp_flux_sol,mat_ctx->vP,INSERT_VALUES,SCATTER_REVERSE);CHKERRQ(ierr);
2319229984c5Sstefano_zampini     ierr = MatMultAdd(mat_ctx->Bt_BB,mat_ctx->vP,pcis->vec1_B,pcis->vec1_B);CHKERRQ(ierr);
2320229984c5Sstefano_zampini   }
2321229984c5Sstefano_zampini 
23223425bc38SStefano Zampini   /* compute rhs for BDDC application */
23233425bc38SStefano Zampini   ierr = VecAYPX(pcis->vec1_B,-1.0,mat_ctx->temp_solution_B);CHKERRQ(ierr);
23248eeda7d8SStefano Zampini   if (pcbddc->switch_static) {
23253425bc38SStefano Zampini     ierr = VecCopy(mat_ctx->temp_solution_D,pcis->vec1_D);CHKERRQ(ierr);
2326229984c5Sstefano_zampini     if (mat_ctx->l2g_p) {
2327229984c5Sstefano_zampini       ierr = VecScale(mat_ctx->vP,-1.);CHKERRQ(ierr);
2328229984c5Sstefano_zampini       ierr = MatMultAdd(mat_ctx->Bt_BI,mat_ctx->vP,pcis->vec1_D,pcis->vec1_D);CHKERRQ(ierr);
23293425bc38SStefano Zampini     }
2330229984c5Sstefano_zampini   }
2331229984c5Sstefano_zampini 
23323425bc38SStefano Zampini   /* apply BDDC */
2333229984c5Sstefano_zampini   ierr = PetscMemzero(pcbddc->benign_p0,pcbddc->benign_n*sizeof(PetscScalar));CHKERRQ(ierr);
2334dc359a40SStefano Zampini   ierr = PCBDDCApplyInterfacePreconditioner(mat_ctx->pc,PETSC_FALSE);CHKERRQ(ierr);
2335229984c5Sstefano_zampini 
2336229984c5Sstefano_zampini   /* put values into global vector */
2337af140850Sstefano_zampini   if (pcbddc->ChangeOfBasisMatrix) work = pcbddc->work_change;
2338af140850Sstefano_zampini   else work = standard_sol;
2339229984c5Sstefano_zampini   ierr = VecScatterBegin(pcis->global_to_B,pcis->vec1_B,work,INSERT_VALUES,SCATTER_REVERSE);CHKERRQ(ierr);
2340229984c5Sstefano_zampini   ierr = VecScatterEnd(pcis->global_to_B,pcis->vec1_B,work,INSERT_VALUES,SCATTER_REVERSE);CHKERRQ(ierr);
23418eeda7d8SStefano Zampini   if (!pcbddc->switch_static) {
23423425bc38SStefano Zampini     /* compute values into the interior if solved for the partially subassembled Schur complement */
23433425bc38SStefano Zampini     ierr = MatMult(pcis->A_IB,pcis->vec1_B,pcis->vec1_D);CHKERRQ(ierr);
234400f6b531SStefano Zampini     ierr = VecAYPX(pcis->vec1_D,-1.0,mat_ctx->temp_solution_D);CHKERRQ(ierr);
234500f6b531SStefano Zampini     ierr = KSPSolve(pcbddc->ksp_D,pcis->vec1_D,pcis->vec1_D);CHKERRQ(ierr);
23463425bc38SStefano Zampini   }
2347229984c5Sstefano_zampini 
2348229984c5Sstefano_zampini   ierr = VecScatterBegin(pcis->global_to_D,pcis->vec1_D,work,INSERT_VALUES,SCATTER_REVERSE);CHKERRQ(ierr);
2349229984c5Sstefano_zampini   ierr = VecScatterEnd(pcis->global_to_D,pcis->vec1_D,work,INSERT_VALUES,SCATTER_REVERSE);CHKERRQ(ierr);
2350266e20e9SStefano Zampini   /* add p0 solution to final solution */
2351229984c5Sstefano_zampini   ierr = PCBDDCBenignGetOrSetP0(mat_ctx->pc,work,PETSC_FALSE);CHKERRQ(ierr);
2352fc17d649SStefano Zampini   if (pcbddc->ChangeOfBasisMatrix) {
2353af140850Sstefano_zampini     ierr = MatMult(pcbddc->ChangeOfBasisMatrix,work,standard_sol);CHKERRQ(ierr);
2354fc17d649SStefano Zampini   }
2355af140850Sstefano_zampini   ierr = PCPostSolve_BDDC(mat_ctx->pc,NULL,NULL,standard_sol);CHKERRQ(ierr);
2356af140850Sstefano_zampini   if (mat_ctx->g2g_p) {
2357229984c5Sstefano_zampini     ierr = VecScatterBegin(mat_ctx->g2g_p,fetidp_flux_sol,standard_sol,INSERT_VALUES,SCATTER_REVERSE);CHKERRQ(ierr);
2358229984c5Sstefano_zampini     ierr = VecScatterEnd(mat_ctx->g2g_p,fetidp_flux_sol,standard_sol,INSERT_VALUES,SCATTER_REVERSE);CHKERRQ(ierr);
2359229984c5Sstefano_zampini   }
23603425bc38SStefano Zampini   PetscFunctionReturn(0);
23613425bc38SStefano Zampini }
23621e6b0712SBarry Smith 
23633425bc38SStefano Zampini /*@
23640f202f7eSStefano Zampini  PCBDDCMatFETIDPGetSolution - Compute the physical solution using the solution of the FETI-DP linear system
23653425bc38SStefano Zampini 
23663425bc38SStefano Zampini    Collective
23673425bc38SStefano Zampini 
23683425bc38SStefano Zampini    Input Parameters:
23690f202f7eSStefano Zampini +  fetidp_mat      - the FETI-DP matrix obtained by a call to PCBDDCCreateFETIDPOperators
23700f202f7eSStefano Zampini -  fetidp_flux_sol - the solution of the FETI-DP linear system
23713425bc38SStefano Zampini 
23723425bc38SStefano Zampini    Output Parameters:
23730f202f7eSStefano Zampini .  standard_sol    - the solution defined on the physical domain
23743425bc38SStefano Zampini 
23753425bc38SStefano Zampini    Level: developer
23763425bc38SStefano Zampini 
23773425bc38SStefano Zampini    Notes:
23783425bc38SStefano Zampini 
23790f202f7eSStefano Zampini .seealso: PCBDDC, PCBDDCCreateFETIDPOperators, PCBDDCMatFETIDPGetRHS
23803425bc38SStefano Zampini @*/
23813425bc38SStefano Zampini PetscErrorCode PCBDDCMatFETIDPGetSolution(Mat fetidp_mat, Vec fetidp_flux_sol, Vec standard_sol)
23823425bc38SStefano Zampini {
2383674ae819SStefano Zampini   FETIDPMat_ctx  mat_ctx;
23843425bc38SStefano Zampini   PetscErrorCode ierr;
23853425bc38SStefano Zampini 
23863425bc38SStefano Zampini   PetscFunctionBegin;
2387266e20e9SStefano Zampini   PetscValidHeaderSpecific(fetidp_mat,MAT_CLASSID,1);
2388266e20e9SStefano Zampini   PetscValidHeaderSpecific(fetidp_flux_sol,VEC_CLASSID,2);
2389266e20e9SStefano Zampini   PetscValidHeaderSpecific(standard_sol,VEC_CLASSID,3);
23903425bc38SStefano Zampini   ierr = MatShellGetContext(fetidp_mat,&mat_ctx);CHKERRQ(ierr);
2391163d334eSBarry Smith   ierr = PetscUseMethod(mat_ctx->pc,"PCBDDCMatFETIDPGetSolution_C",(Mat,Vec,Vec),(fetidp_mat,fetidp_flux_sol,standard_sol));CHKERRQ(ierr);
23923425bc38SStefano Zampini   PetscFunctionReturn(0);
23933425bc38SStefano Zampini }
23941e6b0712SBarry Smith 
2395547c9a8eSstefano_zampini static PetscErrorCode PCBDDCCreateFETIDPOperators_BDDC(PC pc, PetscBool fully_redundant, const char* prefix, Mat *fetidp_mat, PC *fetidp_pc)
23963425bc38SStefano Zampini {
2397674ae819SStefano Zampini 
2398674ae819SStefano Zampini   FETIDPMat_ctx  fetidpmat_ctx;
23993425bc38SStefano Zampini   Mat            newmat;
2400674ae819SStefano Zampini   FETIDPPC_ctx   fetidppc_ctx;
24013425bc38SStefano Zampini   PC             newpc;
2402ce94432eSBarry Smith   MPI_Comm       comm;
24033425bc38SStefano Zampini   PetscErrorCode ierr;
24043425bc38SStefano Zampini 
24053425bc38SStefano Zampini   PetscFunctionBegin;
2406ce94432eSBarry Smith   ierr = PetscObjectGetComm((PetscObject)pc,&comm);CHKERRQ(ierr);
24073425bc38SStefano Zampini   /* FETIDP linear matrix */
24083425bc38SStefano Zampini   ierr = PCBDDCCreateFETIDPMatContext(pc,&fetidpmat_ctx);CHKERRQ(ierr);
24091720468bSStefano Zampini   fetidpmat_ctx->fully_redundant = fully_redundant;
24103425bc38SStefano Zampini   ierr = PCBDDCSetupFETIDPMatContext(fetidpmat_ctx);CHKERRQ(ierr);
2411a5bb87b3Sstefano_zampini   ierr = MatCreateShell(comm,fetidpmat_ctx->n,fetidpmat_ctx->n,fetidpmat_ctx->N,fetidpmat_ctx->N,fetidpmat_ctx,&newmat);CHKERRQ(ierr);
24123425bc38SStefano Zampini   ierr = MatShellSetOperation(newmat,MATOP_MULT,(void (*)(void))FETIDPMatMult);CHKERRQ(ierr);
2413edf7251bSStefano Zampini   ierr = MatShellSetOperation(newmat,MATOP_MULT_TRANSPOSE,(void (*)(void))FETIDPMatMultTranspose);CHKERRQ(ierr);
24143425bc38SStefano Zampini   ierr = MatShellSetOperation(newmat,MATOP_DESTROY,(void (*)(void))PCBDDCDestroyFETIDPMat);CHKERRQ(ierr);
2415547c9a8eSstefano_zampini   ierr = MatSetOptionsPrefix(newmat,prefix);CHKERRQ(ierr);
2416547c9a8eSstefano_zampini   ierr = MatAppendOptionsPrefix(newmat,"fetidp_");CHKERRQ(ierr);
24173425bc38SStefano Zampini   ierr = MatSetUp(newmat);CHKERRQ(ierr);
24183425bc38SStefano Zampini   /* FETIDP preconditioner */
24193425bc38SStefano Zampini   ierr = PCBDDCCreateFETIDPPCContext(pc,&fetidppc_ctx);CHKERRQ(ierr);
24203425bc38SStefano Zampini   ierr = PCBDDCSetupFETIDPPCContext(newmat,fetidppc_ctx);CHKERRQ(ierr);
24213425bc38SStefano Zampini   ierr = PCCreate(comm,&newpc);CHKERRQ(ierr);
2422e1214c54Sstefano_zampini   ierr = PCSetOperators(newpc,newmat,newmat);CHKERRQ(ierr);
2423e1214c54Sstefano_zampini   ierr = PCSetOptionsPrefix(newpc,prefix);CHKERRQ(ierr);
2424e1214c54Sstefano_zampini   ierr = PCAppendOptionsPrefix(newpc,"fetidp_");CHKERRQ(ierr);
2425e1214c54Sstefano_zampini   if (!fetidpmat_ctx->l2g_lambda_only) {
24263425bc38SStefano Zampini     ierr = PCSetType(newpc,PCSHELL);CHKERRQ(ierr);
24273425bc38SStefano Zampini     ierr = PCShellSetContext(newpc,fetidppc_ctx);CHKERRQ(ierr);
24283425bc38SStefano Zampini     ierr = PCShellSetApply(newpc,FETIDPPCApply);CHKERRQ(ierr);
2429edf7251bSStefano Zampini     ierr = PCShellSetApplyTranspose(newpc,FETIDPPCApplyTranspose);CHKERRQ(ierr);
2430c45b8d2dSstefano_zampini     ierr = PCShellSetView(newpc,FETIDPPCView);CHKERRQ(ierr);
24313425bc38SStefano Zampini     ierr = PCShellSetDestroy(newpc,PCBDDCDestroyFETIDPPC);CHKERRQ(ierr);
2432e1214c54Sstefano_zampini   } else {
2433e1214c54Sstefano_zampini     KSP      *ksps;
2434e1214c54Sstefano_zampini     PC       lagpc;
2435e1214c54Sstefano_zampini     Mat      M,AM,PM;
2436e1214c54Sstefano_zampini     PetscInt nn;
2437e1214c54Sstefano_zampini 
2438e1214c54Sstefano_zampini     ierr = PetscObjectQuery((PetscObject)pc,"__KSPFETIDP_PPmat",(PetscObject*)&M);CHKERRQ(ierr);
2439e1214c54Sstefano_zampini     ierr = PCSetType(newpc,PCFIELDSPLIT);CHKERRQ(ierr);
2440e1214c54Sstefano_zampini     ierr = PCFieldSplitSetIS(newpc,"lag",fetidpmat_ctx->lagrange);CHKERRQ(ierr);
2441e1214c54Sstefano_zampini     ierr = PCFieldSplitSetIS(newpc,"p",fetidpmat_ctx->pressure);CHKERRQ(ierr);
2442e1214c54Sstefano_zampini     ierr = PCFieldSplitSetType(newpc,PC_COMPOSITE_SCHUR);CHKERRQ(ierr);
244340c75d76SStefano Zampini     ierr = PCFieldSplitSetSchurFactType(newpc,PC_FIELDSPLIT_SCHUR_FACT_DIAG);CHKERRQ(ierr);
2444e1214c54Sstefano_zampini     ierr = PCFieldSplitSetSchurPre(newpc,PC_FIELDSPLIT_SCHUR_PRE_USER,M);CHKERRQ(ierr);
2445c096484dSStefano Zampini     ierr = PCFieldSplitSetSchurScale(newpc,1.0);CHKERRQ(ierr);
2446e1214c54Sstefano_zampini     ierr = PCSetFromOptions(newpc);CHKERRQ(ierr);
2447e1214c54Sstefano_zampini     ierr = PCSetUp(newpc);CHKERRQ(ierr);
2448e1214c54Sstefano_zampini 
2449e1214c54Sstefano_zampini     /* set the solver for the (0,0) block */
245040c75d76SStefano Zampini     ierr = PCFieldSplitGetSubKSP(newpc,&nn,&ksps);CHKERRQ(ierr);
2451e1214c54Sstefano_zampini     ierr = PCCreate(comm,&lagpc);CHKERRQ(ierr);
2452e1214c54Sstefano_zampini     ierr = PCSetType(lagpc,PCSHELL);CHKERRQ(ierr);
2453e1214c54Sstefano_zampini     ierr = KSPGetOperators(ksps[0],&AM,&PM);CHKERRQ(ierr);
2454e1214c54Sstefano_zampini     ierr = PCSetOperators(lagpc,AM,PM);CHKERRQ(ierr);
2455e1214c54Sstefano_zampini     ierr = PCShellSetContext(lagpc,fetidppc_ctx);CHKERRQ(ierr);
2456e1214c54Sstefano_zampini     ierr = PCShellSetApply(lagpc,FETIDPPCApply);CHKERRQ(ierr);
2457e1214c54Sstefano_zampini     ierr = PCShellSetApplyTranspose(lagpc,FETIDPPCApplyTranspose);CHKERRQ(ierr);
2458e1214c54Sstefano_zampini     ierr = PCShellSetView(lagpc,FETIDPPCView);CHKERRQ(ierr);
2459e1214c54Sstefano_zampini     ierr = PCShellSetDestroy(lagpc,PCBDDCDestroyFETIDPPC);CHKERRQ(ierr);
2460e1214c54Sstefano_zampini     ierr = KSPSetPC(ksps[0],lagpc);CHKERRQ(ierr);
2461e1214c54Sstefano_zampini     ierr = PCDestroy(&lagpc);CHKERRQ(ierr);
2462e1214c54Sstefano_zampini     ierr = PetscFree(ksps);CHKERRQ(ierr);
2463e1214c54Sstefano_zampini   }
24643425bc38SStefano Zampini   /* return pointers for objects created */
24653425bc38SStefano Zampini   *fetidp_mat = newmat;
24663425bc38SStefano Zampini   *fetidp_pc = newpc;
24673425bc38SStefano Zampini   PetscFunctionReturn(0);
24683425bc38SStefano Zampini }
24691e6b0712SBarry Smith 
247094ef8ddeSSatish Balay /*@C
24710f202f7eSStefano Zampini  PCBDDCCreateFETIDPOperators - Create FETI-DP operators
24723425bc38SStefano Zampini 
24733425bc38SStefano Zampini    Collective
24743425bc38SStefano Zampini 
24753425bc38SStefano Zampini    Input Parameters:
24761720468bSStefano Zampini +  pc - the BDDC preconditioning context (setup should have been called before)
2477547c9a8eSstefano_zampini .  fully_redundant - true for a fully redundant set of Lagrange multipliers
2478547c9a8eSstefano_zampini -  prefix - optional options database prefix for the objects to be created (can be NULL)
247928509bceSStefano Zampini 
248028509bceSStefano Zampini    Output Parameters:
24810f202f7eSStefano Zampini +  fetidp_mat - shell FETI-DP matrix object
24820f202f7eSStefano Zampini -  fetidp_pc  - shell Dirichlet preconditioner for FETI-DP matrix
248328509bceSStefano Zampini 
24843425bc38SStefano Zampini    Level: developer
24853425bc38SStefano Zampini 
24863425bc38SStefano Zampini    Notes:
24870f202f7eSStefano Zampini      Currently the only operations provided for FETI-DP matrix are MatMult and MatMultTranspose
24883425bc38SStefano Zampini 
24890f202f7eSStefano Zampini .seealso: PCBDDC, PCBDDCMatFETIDPGetRHS, PCBDDCMatFETIDPGetSolution
24903425bc38SStefano Zampini @*/
2491547c9a8eSstefano_zampini PetscErrorCode PCBDDCCreateFETIDPOperators(PC pc, PetscBool fully_redundant, const char *prefix, Mat *fetidp_mat, PC *fetidp_pc)
24923425bc38SStefano Zampini {
24933425bc38SStefano Zampini   PetscErrorCode ierr;
24943425bc38SStefano Zampini 
24953425bc38SStefano Zampini   PetscFunctionBegin;
24963425bc38SStefano Zampini   PetscValidHeaderSpecific(pc,PC_CLASSID,1);
24973425bc38SStefano Zampini   if (pc->setupcalled) {
2498547c9a8eSstefano_zampini     ierr = PetscUseMethod(pc,"PCBDDCCreateFETIDPOperators_C",(PC,PetscBool,const char*,Mat*,PC*),(pc,fully_redundant,prefix,fetidp_mat,fetidp_pc));CHKERRQ(ierr);
24994247aa23Sstefano_zampini   } else SETERRQ(PetscObjectComm((PetscObject)pc),PETSC_ERR_SUP,"You must call PCSetup_BDDC() first \n");
25003425bc38SStefano Zampini   PetscFunctionReturn(0);
25013425bc38SStefano Zampini }
25020c7d97c5SJed Brown /* -------------------------------------------------------------------------- */
2503da1bb401SStefano Zampini /*MC
2504da1bb401SStefano Zampini    PCBDDC - Balancing Domain Decomposition by Constraints.
25050c7d97c5SJed Brown 
250628509bceSStefano Zampini    An implementation of the BDDC preconditioner based on
250728509bceSStefano Zampini 
250828509bceSStefano Zampini .vb
250928509bceSStefano Zampini    [1] C. R. Dohrmann. "An approximate BDDC preconditioner", Numerical Linear Algebra with Applications Volume 14, Issue 2, pages 149-168, March 2007
251028509bceSStefano 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
251128509bceSStefano Zampini    [3] J. Mandel, B. Sousedik, C. R. Dohrmann. "Multispace and Multilevel BDDC", http://arxiv.org/abs/0712.3977
25120f202f7eSStefano 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
251328509bceSStefano Zampini .ve
251428509bceSStefano Zampini 
251528509bceSStefano Zampini    The matrix to be preconditioned (Pmat) must be of type MATIS.
251628509bceSStefano Zampini 
25170f202f7eSStefano Zampini    Currently works with MATIS matrices with local matrices of type MATSEQAIJ, MATSEQBAIJ or MATSEQSBAIJ, either with real or complex numbers.
251828509bceSStefano Zampini 
251928509bceSStefano Zampini    It also works with unsymmetric and indefinite problems.
252028509bceSStefano Zampini 
2521b6fdb6dfSStefano 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.
2522b6fdb6dfSStefano Zampini 
2523c7017625SStefano 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).
252428509bceSStefano Zampini 
25250f202f7eSStefano 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()
252630368db7SStefano Zampini    Additional information on dofs can be provided by using PCBDDCSetDofsSplitting(), PCBDDCSetDirichletBoundaries(), PCBDDCSetNeumannBoundaries(), and PCBDDCSetPrimalVerticesIS() and their local counterparts.
252728509bceSStefano Zampini 
25280f202f7eSStefano Zampini    Constraints can be customized by attaching a MatNullSpace object to the MATIS matrix via MatSetNearNullSpace(). Non-singular modes are retained via SVD.
252928509bceSStefano Zampini 
25300f202f7eSStefano 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.
25310f202f7eSStefano Zampini    User defined change of basis can be passed to PCBDDC by using PCBDDCSetChangeOfBasisMat()
253228509bceSStefano Zampini 
25330f202f7eSStefano Zampini    The PETSc implementation also supports multilevel BDDC [3]. Coarse grids are partitioned using a MatPartitioning object.
253428509bceSStefano Zampini 
2535df4d28bfSStefano Zampini    Adaptive selection of primal constraints [4] is supported for SPD systems with high-contrast in the coefficients if MUMPS or MKL_PARDISO are present. Future versions of the code will also consider using PASTIX.
253628509bceSStefano Zampini 
25370f202f7eSStefano 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.
25380f202f7eSStefano Zampini    Deluxe scaling is not supported yet for FETI-DP.
25390f202f7eSStefano Zampini 
25400f202f7eSStefano Zampini    Options Database Keys (some of them, run with -h for a complete list):
25410f202f7eSStefano Zampini 
25420f202f7eSStefano Zampini .    -pc_bddc_use_vertices <true> - use or not vertices in primal space
25430f202f7eSStefano Zampini .    -pc_bddc_use_edges <true> - use or not edges in primal space
25440f202f7eSStefano Zampini .    -pc_bddc_use_faces <false> - use or not faces in primal space
25450f202f7eSStefano Zampini .    -pc_bddc_symmetric <true> - symmetric computation of primal basis functions. Specify false for unsymmetric problems
25460f202f7eSStefano Zampini .    -pc_bddc_use_change_of_basis <false> - use change of basis approach (on edges only)
25470f202f7eSStefano Zampini .    -pc_bddc_use_change_on_faces <false> - use change of basis approach on faces if change of basis has been requested
25480f202f7eSStefano Zampini .    -pc_bddc_switch_static <false> - switches from M_2 (default) to M_3 operator (see reference article [1])
254928509bceSStefano Zampini .    -pc_bddc_levels <0> - maximum number of levels for multilevel
25500f202f7eSStefano 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)
25510f202f7eSStefano 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)
25520f202f7eSStefano Zampini .    -pc_bddc_use_deluxe_scaling <false> - use deluxe scaling
25530f202f7eSStefano 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)
2554df4d28bfSStefano 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 or MKL_PARDISO installed)
255528509bceSStefano Zampini -    -pc_bddc_check_level <0> - set verbosity level of debugging output
255628509bceSStefano Zampini 
255728509bceSStefano Zampini    Options for Dirichlet, Neumann or coarse solver can be set with
255828509bceSStefano Zampini .vb
255928509bceSStefano Zampini       -pc_bddc_dirichlet_
256028509bceSStefano Zampini       -pc_bddc_neumann_
256128509bceSStefano Zampini       -pc_bddc_coarse_
256228509bceSStefano Zampini .ve
25630f202f7eSStefano Zampini    e.g -pc_bddc_dirichlet_ksp_type richardson -pc_bddc_dirichlet_pc_type gamg. PCBDDC uses by default KPSPREONLY and PCLU.
256428509bceSStefano Zampini 
25650f202f7eSStefano Zampini    When using a multilevel approach, solvers' options at the N-th level (N > 1) can be specified as
256628509bceSStefano Zampini .vb
2567312be037SStefano Zampini       -pc_bddc_dirichlet_lN_
2568312be037SStefano Zampini       -pc_bddc_neumann_lN_
2569312be037SStefano Zampini       -pc_bddc_coarse_lN_
257028509bceSStefano Zampini .ve
25710f202f7eSStefano Zampini    Note that level number ranges from the finest (0) to the coarsest (N).
25720f202f7eSStefano 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.
25730f202f7eSStefano Zampini .vb
25740f202f7eSStefano Zampini      -pc_bddc_coarse_pc_bddc_adaptive_threshold 5 -pc_bddc_coarse_l1_pc_bddc_redistribute 3
25750f202f7eSStefano Zampini .ve
25760f202f7eSStefano 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
2577da1bb401SStefano Zampini 
2578da1bb401SStefano Zampini    Level: intermediate
2579da1bb401SStefano Zampini 
2580b6fdb6dfSStefano Zampini    Developer notes:
2581da1bb401SStefano Zampini 
2582da1bb401SStefano Zampini    Contributed by Stefano Zampini
2583da1bb401SStefano Zampini 
2584da1bb401SStefano Zampini .seealso:  PCCreate(), PCSetType(), PCType (for list of available types), PC,  MATIS
2585da1bb401SStefano Zampini M*/
2586b2573a8aSBarry Smith 
25878cc058d9SJed Brown PETSC_EXTERN PetscErrorCode PCCreate_BDDC(PC pc)
2588da1bb401SStefano Zampini {
2589da1bb401SStefano Zampini   PetscErrorCode      ierr;
2590da1bb401SStefano Zampini   PC_BDDC             *pcbddc;
2591da1bb401SStefano Zampini 
2592da1bb401SStefano Zampini   PetscFunctionBegin;
2593b00a9115SJed Brown   ierr     = PetscNewLog(pc,&pcbddc);CHKERRQ(ierr);
2594da1bb401SStefano Zampini   pc->data = (void*)pcbddc;
2595da1bb401SStefano Zampini 
2596da1bb401SStefano Zampini   /* create PCIS data structure */
2597da1bb401SStefano Zampini   ierr = PCISCreate(pc);CHKERRQ(ierr);
2598da1bb401SStefano Zampini 
25999326c5c6Sstefano_zampini   /* create local graph structure */
26009326c5c6Sstefano_zampini   ierr = PCBDDCGraphCreate(&pcbddc->mat_graph);CHKERRQ(ierr);
26019326c5c6Sstefano_zampini 
26029326c5c6Sstefano_zampini   /* BDDC nonzero defaults */
260308a5cf49SStefano Zampini   pcbddc->use_local_adj             = PETSC_TRUE;
260447d04d0dSStefano Zampini   pcbddc->use_vertices              = PETSC_TRUE;
260547d04d0dSStefano Zampini   pcbddc->use_edges                 = PETSC_TRUE;
26063301b35fSStefano Zampini   pcbddc->symmetric_primal          = PETSC_TRUE;
260714f95afaSStefano Zampini   pcbddc->vertex_size               = 1;
2608c703fcc7SStefano Zampini   pcbddc->recompute_topography      = PETSC_TRUE;
260968457ee5SStefano Zampini   pcbddc->coarse_size               = -1;
261085c4d303SStefano Zampini   pcbddc->use_exact_dirichlet_trick = PETSC_TRUE;
261147d04d0dSStefano Zampini   pcbddc->coarsening_ratio          = 8;
261257de7509SStefano Zampini   pcbddc->coarse_eqs_per_proc       = 1;
261327b6a85dSStefano Zampini   pcbddc->benign_compute_correction = PETSC_TRUE;
26141e0482f5SStefano Zampini   pcbddc->nedfield                  = -1;
26151e0482f5SStefano Zampini   pcbddc->nedglobal                 = PETSC_TRUE;
2616be12c134Sstefano_zampini   pcbddc->graphmaxcount             = PETSC_MAX_INT;
2617b96c3477SStefano Zampini   pcbddc->sub_schurs_layers         = -1;
2618b7eb3628SStefano Zampini 
2619da1bb401SStefano Zampini   /* function pointers */
2620da1bb401SStefano Zampini   pc->ops->apply               = PCApply_BDDC;
262193bd9ae7SStefano Zampini   pc->ops->applytranspose      = PCApplyTranspose_BDDC;
2622da1bb401SStefano Zampini   pc->ops->setup               = PCSetUp_BDDC;
2623da1bb401SStefano Zampini   pc->ops->destroy             = PCDestroy_BDDC;
2624da1bb401SStefano Zampini   pc->ops->setfromoptions      = PCSetFromOptions_BDDC;
26256b78500eSPatrick Sanan   pc->ops->view                = PCView_BDDC;
2626da1bb401SStefano Zampini   pc->ops->applyrichardson     = 0;
2627da1bb401SStefano Zampini   pc->ops->applysymmetricleft  = 0;
2628da1bb401SStefano Zampini   pc->ops->applysymmetricright = 0;
2629534831adSStefano Zampini   pc->ops->presolve            = PCPreSolve_BDDC;
2630534831adSStefano Zampini   pc->ops->postsolve           = PCPostSolve_BDDC;
26319326c5c6Sstefano_zampini   pc->ops->reset               = PCReset_BDDC;
2632da1bb401SStefano Zampini 
2633da1bb401SStefano Zampini   /* composing function */
2634a13144ffSStefano Zampini   ierr = PetscObjectComposeFunction((PetscObject)pc,"PCBDDCSetDiscreteGradient_C",PCBDDCSetDiscreteGradient_BDDC);CHKERRQ(ierr);
2635a198735bSStefano Zampini   ierr = PetscObjectComposeFunction((PetscObject)pc,"PCBDDCSetDivergenceMat_C",PCBDDCSetDivergenceMat_BDDC);CHKERRQ(ierr);
2636906d46d4SStefano Zampini   ierr = PetscObjectComposeFunction((PetscObject)pc,"PCBDDCSetChangeOfBasisMat_C",PCBDDCSetChangeOfBasisMat_BDDC);CHKERRQ(ierr);
2637674ae819SStefano Zampini   ierr = PetscObjectComposeFunction((PetscObject)pc,"PCBDDCSetPrimalVerticesLocalIS_C",PCBDDCSetPrimalVerticesLocalIS_BDDC);CHKERRQ(ierr);
263830368db7SStefano Zampini   ierr = PetscObjectComposeFunction((PetscObject)pc,"PCBDDCSetPrimalVerticesIS_C",PCBDDCSetPrimalVerticesIS_BDDC);CHKERRQ(ierr);
2639bdf89e91SBarry Smith   ierr = PetscObjectComposeFunction((PetscObject)pc,"PCBDDCSetCoarseningRatio_C",PCBDDCSetCoarseningRatio_BDDC);CHKERRQ(ierr);
26402b510759SStefano Zampini   ierr = PetscObjectComposeFunction((PetscObject)pc,"PCBDDCSetLevel_C",PCBDDCSetLevel_BDDC);CHKERRQ(ierr);
2641b8ffe317SStefano Zampini   ierr = PetscObjectComposeFunction((PetscObject)pc,"PCBDDCSetUseExactDirichlet_C",PCBDDCSetUseExactDirichlet_BDDC);CHKERRQ(ierr);
26422b510759SStefano Zampini   ierr = PetscObjectComposeFunction((PetscObject)pc,"PCBDDCSetLevels_C",PCBDDCSetLevels_BDDC);CHKERRQ(ierr);
2643bdf89e91SBarry Smith   ierr = PetscObjectComposeFunction((PetscObject)pc,"PCBDDCSetDirichletBoundaries_C",PCBDDCSetDirichletBoundaries_BDDC);CHKERRQ(ierr);
264482ba6b80SStefano Zampini   ierr = PetscObjectComposeFunction((PetscObject)pc,"PCBDDCSetDirichletBoundariesLocal_C",PCBDDCSetDirichletBoundariesLocal_BDDC);CHKERRQ(ierr);
2645bdf89e91SBarry Smith   ierr = PetscObjectComposeFunction((PetscObject)pc,"PCBDDCSetNeumannBoundaries_C",PCBDDCSetNeumannBoundaries_BDDC);CHKERRQ(ierr);
264682ba6b80SStefano Zampini   ierr = PetscObjectComposeFunction((PetscObject)pc,"PCBDDCSetNeumannBoundariesLocal_C",PCBDDCSetNeumannBoundariesLocal_BDDC);CHKERRQ(ierr);
2647bdf89e91SBarry Smith   ierr = PetscObjectComposeFunction((PetscObject)pc,"PCBDDCGetDirichletBoundaries_C",PCBDDCGetDirichletBoundaries_BDDC);CHKERRQ(ierr);
264882ba6b80SStefano Zampini   ierr = PetscObjectComposeFunction((PetscObject)pc,"PCBDDCGetDirichletBoundariesLocal_C",PCBDDCGetDirichletBoundariesLocal_BDDC);CHKERRQ(ierr);
2649bdf89e91SBarry Smith   ierr = PetscObjectComposeFunction((PetscObject)pc,"PCBDDCGetNeumannBoundaries_C",PCBDDCGetNeumannBoundaries_BDDC);CHKERRQ(ierr);
265082ba6b80SStefano Zampini   ierr = PetscObjectComposeFunction((PetscObject)pc,"PCBDDCGetNeumannBoundariesLocal_C",PCBDDCGetNeumannBoundariesLocal_BDDC);CHKERRQ(ierr);
2651bdf89e91SBarry Smith   ierr = PetscObjectComposeFunction((PetscObject)pc,"PCBDDCSetDofsSplitting_C",PCBDDCSetDofsSplitting_BDDC);CHKERRQ(ierr);
265263602bcaSStefano Zampini   ierr = PetscObjectComposeFunction((PetscObject)pc,"PCBDDCSetDofsSplittingLocal_C",PCBDDCSetDofsSplittingLocal_BDDC);CHKERRQ(ierr);
2653bdf89e91SBarry Smith   ierr = PetscObjectComposeFunction((PetscObject)pc,"PCBDDCSetLocalAdjacencyGraph_C",PCBDDCSetLocalAdjacencyGraph_BDDC);CHKERRQ(ierr);
2654bdf89e91SBarry Smith   ierr = PetscObjectComposeFunction((PetscObject)pc,"PCBDDCCreateFETIDPOperators_C",PCBDDCCreateFETIDPOperators_BDDC);CHKERRQ(ierr);
2655bdf89e91SBarry Smith   ierr = PetscObjectComposeFunction((PetscObject)pc,"PCBDDCMatFETIDPGetRHS_C",PCBDDCMatFETIDPGetRHS_BDDC);CHKERRQ(ierr);
2656bdf89e91SBarry Smith   ierr = PetscObjectComposeFunction((PetscObject)pc,"PCBDDCMatFETIDPGetSolution_C",PCBDDCMatFETIDPGetSolution_BDDC);CHKERRQ(ierr);
2657a06fd7f2SStefano Zampini   ierr = PetscObjectComposeFunction((PetscObject)pc,"PCPreSolveChangeRHS_C",PCPreSolveChangeRHS_BDDC);CHKERRQ(ierr);
2658da1bb401SStefano Zampini   PetscFunctionReturn(0);
2659da1bb401SStefano Zampini }
26603425bc38SStefano Zampini 
2661