xref: /petsc/src/ksp/pc/impls/bddc/bddc.c (revision efd4aadf157bf1ba2d80c2be092fcf4247860003)
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     }
126*efd4aadfSBarry Smith     ierr = PetscViewerASCIIPrintf(viewer,"  Use verbose output: %D\n",pcbddc->dbg_flag);CHKERRQ(ierr);
127*efd4aadfSBarry Smith     ierr = PetscViewerASCIIPrintf(viewer,"  Use user-defined CSR: %D\n",!!pcbddc->mat_graph->nvtxs_csr);CHKERRQ(ierr);
128*efd4aadfSBarry 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) {
130*efd4aadfSBarry Smith       ierr = PetscViewerASCIIPrintf(viewer,"  Connectivity graph topological dimension: 2\n");CHKERRQ(ierr);
131e9627c49SStefano Zampini     } else {
132*efd4aadfSBarry Smith       ierr = PetscViewerASCIIPrintf(viewer,"  Connectivity graph topological dimension: 3\n");CHKERRQ(ierr);
133e9627c49SStefano Zampini     }
134*efd4aadfSBarry Smith     ierr = PetscViewerASCIIPrintf(viewer,"  Graph max count: %D\n",pcbddc->graphmaxcount);CHKERRQ(ierr);
135*efd4aadfSBarry Smith     ierr = PetscViewerASCIIPrintf(viewer,"  Use vertices: %d (vertex size %d)\n",pcbddc->use_vertices,pcbddc->vertex_size);CHKERRQ(ierr);
136*efd4aadfSBarry Smith     ierr = PetscViewerASCIIPrintf(viewer,"  Use edges: %d\n",pcbddc->use_edges);CHKERRQ(ierr);
137*efd4aadfSBarry Smith     ierr = PetscViewerASCIIPrintf(viewer,"  Use faces: %d\n",pcbddc->use_faces);CHKERRQ(ierr);
138*efd4aadfSBarry Smith     ierr = PetscViewerASCIIPrintf(viewer,"  Use true near null space: %d\n",pcbddc->use_nnsp_true);CHKERRQ(ierr);
139*efd4aadfSBarry Smith     ierr = PetscViewerASCIIPrintf(viewer,"  Use QR for single constraints on cc: %d\n",pcbddc->use_qr_single);CHKERRQ(ierr);
140*efd4aadfSBarry Smith     ierr = PetscViewerASCIIPrintf(viewer,"  Use change of basis on local edge nodes: %d\n",pcbddc->use_change_of_basis);CHKERRQ(ierr);
141*efd4aadfSBarry Smith     ierr = PetscViewerASCIIPrintf(viewer,"  Use change of basis on local face nodes: %d\n",pcbddc->use_change_on_faces);CHKERRQ(ierr);
142*efd4aadfSBarry Smith     ierr = PetscViewerASCIIPrintf(viewer,"  User defined change of basis matrix: %d\n",!!pcbddc->user_ChangeOfBasisMatrix);CHKERRQ(ierr);
143*efd4aadfSBarry Smith     ierr = PetscViewerASCIIPrintf(viewer,"  Has change of basis matrix: %d\n",!!pcbddc->ChangeOfBasisMatrix);CHKERRQ(ierr);
144*efd4aadfSBarry Smith     ierr = PetscViewerASCIIPrintf(viewer,"  Eliminate dirichlet boundary dofs: %d\n",pcbddc->eliminate_dirdofs);CHKERRQ(ierr);
145*efd4aadfSBarry Smith     ierr = PetscViewerASCIIPrintf(viewer,"  Switch on static condensation ops around the interface preconditioner: %d\n",pcbddc->switch_static);CHKERRQ(ierr);
146*efd4aadfSBarry Smith     ierr = PetscViewerASCIIPrintf(viewer,"  Use exact dirichlet trick: %d\n",pcbddc->use_exact_dirichlet_trick);CHKERRQ(ierr);
147*efd4aadfSBarry Smith     ierr = PetscViewerASCIIPrintf(viewer,"  Multilevel max levels: %d\n",pcbddc->max_levels);CHKERRQ(ierr);
148*efd4aadfSBarry Smith     ierr = PetscViewerASCIIPrintf(viewer,"  Multilevel coarsening ratio: %d\n",pcbddc->coarsening_ratio);CHKERRQ(ierr);
149*efd4aadfSBarry Smith     ierr = PetscViewerASCIIPrintf(viewer,"  Use estimated eigs for coarse problem: %d\n",pcbddc->use_coarse_estimates);CHKERRQ(ierr);
150*efd4aadfSBarry Smith     ierr = PetscViewerASCIIPrintf(viewer,"  Use deluxe scaling: %d\n",pcbddc->use_deluxe_scaling);CHKERRQ(ierr);
151*efd4aadfSBarry Smith     ierr = PetscViewerASCIIPrintf(viewer,"  Use deluxe zerorows: %d\n",pcbddc->deluxe_zerorows);CHKERRQ(ierr);
152*efd4aadfSBarry Smith     ierr = PetscViewerASCIIPrintf(viewer,"  Use deluxe singlemat: %d\n",pcbddc->deluxe_singlemat);CHKERRQ(ierr);
153*efd4aadfSBarry Smith     ierr = PetscViewerASCIIPrintf(viewer,"  Rebuild interface graph for Schur principal minors: %d\n",pcbddc->sub_schurs_rebuild);CHKERRQ(ierr);
154*efd4aadfSBarry Smith     ierr = PetscViewerASCIIPrintf(viewer,"  Number of dofs' layers for the computation of principal minors: %d\n",pcbddc->sub_schurs_layers);CHKERRQ(ierr);
155*efd4aadfSBarry Smith     ierr = PetscViewerASCIIPrintf(viewer,"  Use user CSR graph to compute successive layers: %d\n",pcbddc->sub_schurs_use_useradj);CHKERRQ(ierr);
156*efd4aadfSBarry 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);
157*efd4aadfSBarry Smith     ierr = PetscViewerASCIIPrintf(viewer,"  Min constraints / connected component: %d\n",pcbddc->adaptive_nmin);CHKERRQ(ierr);
158*efd4aadfSBarry Smith     ierr = PetscViewerASCIIPrintf(viewer,"  Max constraints / connected component: %d\n",pcbddc->adaptive_nmax);CHKERRQ(ierr);
159*efd4aadfSBarry Smith     ierr = PetscViewerASCIIPrintf(viewer,"  Invert exact Schur complement for adaptive selection: %d\n",pcbddc->sub_schurs_exact_schur);CHKERRQ(ierr);
160*efd4aadfSBarry Smith     ierr = PetscViewerASCIIPrintf(viewer,"  Symmetric computation of primal basis functions: %d\n",pcbddc->symmetric_primal);CHKERRQ(ierr);
161*efd4aadfSBarry Smith     ierr = PetscViewerASCIIPrintf(viewer,"  Num. Procs. to map coarse adjacency list: %d\n",pcbddc->coarse_adj_red);CHKERRQ(ierr);
162*efd4aadfSBarry Smith     ierr = PetscViewerASCIIPrintf(viewer,"  Coarse eqs per proc (significant at the coarsest level): %d\n",pcbddc->coarse_eqs_per_proc);CHKERRQ(ierr);
163*efd4aadfSBarry Smith     ierr = PetscViewerASCIIPrintf(viewer,"  Detect disconnected: %d\n",pcbddc->detect_disconnected);CHKERRQ(ierr);
164*efd4aadfSBarry Smith     ierr = PetscViewerASCIIPrintf(viewer,"  Benign subspace trick: %d (change explicit %d)\n",pcbddc->benign_saddle_point,pcbddc->benign_change_explicit);CHKERRQ(ierr);
165*efd4aadfSBarry Smith     ierr = PetscViewerASCIIPrintf(viewer,"  Benign subspace trick is active: %d\n",pcbddc->benign_have_null);CHKERRQ(ierr);
166*efd4aadfSBarry Smith     ierr = PetscViewerASCIIPrintf(viewer,"  Algebraic computation of no-net-flux %d\n",pcbddc->compute_nonetflux);CHKERRQ(ierr);
167b74ba07aSstefano_zampini     if (!pc->setupcalled) PetscFunctionReturn(0);
1686b78500eSPatrick Sanan 
169fbad9177SStefano Zampini     /* compute interface size */
170e9627c49SStefano Zampini     ierr = VecSet(pcis->vec1_B,1.0);CHKERRQ(ierr);
171e9627c49SStefano Zampini     ierr = MatCreateVecs(pc->pmat,&counter,0);CHKERRQ(ierr);
172e9627c49SStefano Zampini     ierr = VecSet(counter,0.0);CHKERRQ(ierr);
173e9627c49SStefano Zampini     ierr = VecScatterBegin(pcis->global_to_B,pcis->vec1_B,counter,INSERT_VALUES,SCATTER_REVERSE);CHKERRQ(ierr);
174e9627c49SStefano Zampini     ierr = VecScatterEnd(pcis->global_to_B,pcis->vec1_B,counter,INSERT_VALUES,SCATTER_REVERSE);CHKERRQ(ierr);
175e9627c49SStefano Zampini     ierr = VecSum(counter,&interface_size);CHKERRQ(ierr);
176e9627c49SStefano Zampini     ierr = VecDestroy(&counter);CHKERRQ(ierr);
177fbad9177SStefano Zampini 
178fbad9177SStefano Zampini     /* compute some statistics on the domain decomposition */
179e9627c49SStefano Zampini     gsum[0] = 1;
180fbad9177SStefano Zampini     gsum[1] = gsum[2] = gsum[3] = gsum[4] = gsum[5] = 0;
181e9627c49SStefano Zampini     loc[0]  = !!pcis->n;
182e9627c49SStefano Zampini     loc[1]  = pcis->n - pcis->n_B;
183e9627c49SStefano Zampini     loc[2]  = pcis->n_B;
184e9627c49SStefano Zampini     loc[3]  = pcbddc->local_primal_size;
185345ecf6cSStefano Zampini     loc[4]  = pcis->n;
186fbad9177SStefano Zampini     loc[5]  = pcbddc->n_local_subs > 0 ? pcbddc->n_local_subs : (pcis->n ? 1 : 0);
187fbad9177SStefano Zampini     loc[6]  = pcbddc->benign_n;
188fbad9177SStefano Zampini     ierr = MPI_Reduce(loc,gsum,6,MPIU_INT64,MPI_SUM,0,PetscObjectComm((PetscObject)pc));CHKERRQ(ierr);
189fbad9177SStefano Zampini     if (!loc[0]) loc[1] = loc[2] = loc[3] = loc[4] = loc[5] = -1;
190fbad9177SStefano Zampini     ierr = MPI_Reduce(loc,gmax,6,MPIU_INT64,MPI_MAX,0,PetscObjectComm((PetscObject)pc));CHKERRQ(ierr);
191fbad9177SStefano Zampini     if (!loc[0]) loc[1] = loc[2] = loc[3] = loc[4] = loc[5] = PETSC_MAX_INT;
192fbad9177SStefano Zampini     ierr = MPI_Reduce(loc,gmin,6,MPIU_INT64,MPI_MIN,0,PetscObjectComm((PetscObject)pc));CHKERRQ(ierr);
193fbad9177SStefano Zampini     ierr = MPI_Reduce(&loc[6],&totbenign,1,MPIU_INT64,MPI_SUM,0,PetscObjectComm((PetscObject)pc));CHKERRQ(ierr);
194e9627c49SStefano Zampini     if (pcbddc->coarse_size) {
195e9627c49SStefano Zampini       ratio1 = pc->pmat->rmap->N/(1.*pcbddc->coarse_size);
196e9627c49SStefano Zampini       ratio2 = PetscRealPart(interface_size)/pcbddc->coarse_size;
197e9627c49SStefano Zampini     }
198*efd4aadfSBarry Smith     ierr = PetscViewerASCIIPrintf(viewer,"  ********************************** STATISTICS AT LEVEL %d **********************************\n",pcbddc->current_level);CHKERRQ(ierr);
199*efd4aadfSBarry 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);
200*efd4aadfSBarry Smith     ierr = PetscViewerASCIIPrintf(viewer,"  Coarsening ratios: all/coarse %d interface/coarse %d\n",(PetscInt)ratio1,(PetscInt)ratio2);CHKERRQ(ierr);
201*efd4aadfSBarry Smith     ierr = PetscViewerASCIIPrintf(viewer,"  Active processes : %d\n",(PetscInt)gsum[0]);CHKERRQ(ierr);
202*efd4aadfSBarry Smith     ierr = PetscViewerASCIIPrintf(viewer,"  Total subdomains : %d\n",(PetscInt)gsum[5]);CHKERRQ(ierr);
203345ecf6cSStefano Zampini     if (pcbddc->benign_have_null) {
204*efd4aadfSBarry Smith       ierr = PetscViewerASCIIPrintf(viewer,"  Benign subs      : %d\n",(PetscInt)totbenign);CHKERRQ(ierr);
205345ecf6cSStefano Zampini     }
206*efd4aadfSBarry Smith     ierr = PetscViewerASCIIPrintf(viewer,"  Dofs type        :\tMIN\tMAX\tMEAN\n",(PetscInt)gmin[1],(PetscInt)gmax[1],(PetscInt)(gsum[1]/gsum[0]));CHKERRQ(ierr);
207*efd4aadfSBarry 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);
208*efd4aadfSBarry 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);
209*efd4aadfSBarry 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);
210*efd4aadfSBarry 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);
211*efd4aadfSBarry Smith     ierr = PetscViewerASCIIPrintf(viewer,"  Local     subs   :\t%d\t%d\n",(PetscInt)gmin[5],(PetscInt)gmax[5]);CHKERRQ(ierr);
212*efd4aadfSBarry Smith     ierr = PetscViewerASCIIPrintf(viewer,"  ********************************** COARSE PROBLEM DETAILS *********************************\n",pcbddc->current_level);CHKERRQ(ierr);
21327b6a85dSStefano Zampini     ierr = PetscViewerFlush(viewer);CHKERRQ(ierr);
214e9627c49SStefano Zampini 
215fbad9177SStefano Zampini     /* the coarse problem can be handled by a different communicator */
216e9627c49SStefano Zampini     if (pcbddc->coarse_ksp) color = 1;
217e9627c49SStefano Zampini     else color = 0;
218e9627c49SStefano Zampini     ierr = MPI_Comm_rank(PetscObjectComm((PetscObject)pc),&rank);CHKERRQ(ierr);
2194b2aedd3SStefano Zampini     ierr = MPI_Comm_size(PetscObjectComm((PetscObject)pc),&size);CHKERRQ(ierr);
220e9627c49SStefano Zampini     ierr = PetscSubcommCreate(PetscObjectComm((PetscObject)pc),&subcomm);CHKERRQ(ierr);
2214b2aedd3SStefano Zampini     ierr = PetscSubcommSetNumber(subcomm,PetscMin(size,2));CHKERRQ(ierr);
222e9627c49SStefano Zampini     ierr = PetscSubcommSetTypeGeneral(subcomm,color,rank);CHKERRQ(ierr);
223e9627c49SStefano Zampini     ierr = PetscViewerGetSubViewer(viewer,PetscSubcommChild(subcomm),&subviewer);CHKERRQ(ierr);
224e9627c49SStefano Zampini     if (color == 1) {
225e9627c49SStefano Zampini       ierr = KSPView(pcbddc->coarse_ksp,subviewer);CHKERRQ(ierr);
226e9627c49SStefano Zampini       ierr = PetscViewerFlush(subviewer);CHKERRQ(ierr);
227e9627c49SStefano Zampini     }
228e9627c49SStefano Zampini     ierr = PetscViewerRestoreSubViewer(viewer,PetscSubcommChild(subcomm),&subviewer);CHKERRQ(ierr);
229e9627c49SStefano Zampini     ierr = PetscSubcommDestroy(&subcomm);CHKERRQ(ierr);
230e9627c49SStefano Zampini     ierr = PetscViewerFlush(viewer);CHKERRQ(ierr);
231e9627c49SStefano Zampini   }
2326b78500eSPatrick Sanan   PetscFunctionReturn(0);
2336b78500eSPatrick Sanan }
234a13144ffSStefano Zampini 
2351e0482f5SStefano Zampini static PetscErrorCode PCBDDCSetDiscreteGradient_BDDC(PC pc, Mat G, PetscInt order, PetscInt field, PetscBool global, PetscBool conforming)
236a13144ffSStefano Zampini {
237a13144ffSStefano Zampini   PC_BDDC        *pcbddc = (PC_BDDC*)pc->data;
238a13144ffSStefano Zampini   PetscErrorCode ierr;
239a13144ffSStefano Zampini 
240a13144ffSStefano Zampini   PetscFunctionBegin;
241a13144ffSStefano Zampini   ierr = PetscObjectReference((PetscObject)G);CHKERRQ(ierr);
242a13144ffSStefano Zampini   ierr = MatDestroy(&pcbddc->discretegradient);CHKERRQ(ierr);
243a13144ffSStefano Zampini   pcbddc->discretegradient = G;
244a13144ffSStefano Zampini   pcbddc->nedorder         = order > 0 ? order : -order;
245495a2a07SStefano Zampini   pcbddc->nedfield         = field;
2461e0482f5SStefano Zampini   pcbddc->nedglobal        = global;
2471e0482f5SStefano Zampini   pcbddc->conforming       = conforming;
248a13144ffSStefano Zampini   PetscFunctionReturn(0);
249a13144ffSStefano Zampini }
250a13144ffSStefano Zampini 
251a13144ffSStefano Zampini /*@
252a13144ffSStefano Zampini  PCBDDCSetDiscreteGradient - Sets the discrete gradient
253a13144ffSStefano Zampini 
254a13144ffSStefano Zampini    Collective on PC
255a13144ffSStefano Zampini 
256a13144ffSStefano Zampini    Input Parameters:
257a13144ffSStefano Zampini +  pc         - the preconditioning context
258a13144ffSStefano Zampini .  G          - the discrete gradient matrix (should be in AIJ format)
259a13144ffSStefano Zampini .  order      - the order of the Nedelec space (1 for the lowest order)
260495a2a07SStefano Zampini .  field      - the field id of the Nedelec dofs (not used if the fields have not been specified)
2611e0482f5SStefano Zampini .  global     - the type of global ordering for the rows of G
262a13144ffSStefano Zampini -  conforming - whether the mesh is conforming or not
263a13144ffSStefano Zampini 
264a13144ffSStefano Zampini    Level: advanced
265a13144ffSStefano Zampini 
266a13144ffSStefano Zampini    Notes: The discrete gradient matrix G is used to analyze the subdomain edges, and it should not contain any zero entry.
267495a2a07SStefano Zampini           For variable order spaces, the order should be set to zero.
2681e0482f5SStefano Zampini           If global is true, the rows of G should be given in global ordering for the whole dofs;
2691e0482f5SStefano Zampini           if false, the ordering should be global for the Nedelec field.
2701e0482f5SStefano 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
2711e0482f5SStefano Zampini           and geid the one for the Nedelec field.
272a13144ffSStefano Zampini 
273495a2a07SStefano Zampini .seealso: PCBDDC,PCBDDCSetDofsSplitting(),PCBDDCSetDofsSplittingLocal()
274a13144ffSStefano Zampini @*/
2751e0482f5SStefano Zampini PetscErrorCode PCBDDCSetDiscreteGradient(PC pc, Mat G, PetscInt order, PetscInt field, PetscBool global, PetscBool conforming)
276a13144ffSStefano Zampini {
277a13144ffSStefano Zampini   PetscErrorCode ierr;
278a13144ffSStefano Zampini 
279a13144ffSStefano Zampini   PetscFunctionBegin;
280a13144ffSStefano Zampini   PetscValidHeaderSpecific(pc,PC_CLASSID,1);
281a13144ffSStefano Zampini   PetscValidHeaderSpecific(G,MAT_CLASSID,2);
282a13144ffSStefano Zampini   PetscValidLogicalCollectiveInt(pc,order,3);
2831e0482f5SStefano Zampini   PetscValidLogicalCollectiveInt(pc,field,4);
2841e0482f5SStefano Zampini   PetscValidLogicalCollectiveBool(pc,global,5);
2851e0482f5SStefano Zampini   PetscValidLogicalCollectiveBool(pc,conforming,6);
2861e0482f5SStefano Zampini   PetscCheckSameComm(pc,1,G,2);
2871e0482f5SStefano Zampini   ierr = PetscTryMethod(pc,"PCBDDCSetDiscreteGradient_C",(PC,Mat,PetscInt,PetscInt,PetscBool,PetscBool),(pc,G,order,field,global,conforming));CHKERRQ(ierr);
288a13144ffSStefano Zampini   PetscFunctionReturn(0);
289a13144ffSStefano Zampini }
290a13144ffSStefano Zampini 
2918ae0ca82SStefano Zampini static PetscErrorCode PCBDDCSetDivergenceMat_BDDC(PC pc, Mat divudotp, PetscBool trans, IS vl2l)
292a198735bSStefano Zampini {
293a198735bSStefano Zampini   PC_BDDC        *pcbddc = (PC_BDDC*)pc->data;
294a198735bSStefano Zampini   PetscErrorCode ierr;
2956b78500eSPatrick Sanan 
296a198735bSStefano Zampini   PetscFunctionBegin;
297a198735bSStefano Zampini   ierr = PetscObjectReference((PetscObject)divudotp);CHKERRQ(ierr);
298a198735bSStefano Zampini   ierr = MatDestroy(&pcbddc->divudotp);CHKERRQ(ierr);
299a198735bSStefano Zampini   pcbddc->divudotp = divudotp;
3008ae0ca82SStefano Zampini   pcbddc->divudotp_trans = trans;
301a198735bSStefano Zampini   pcbddc->compute_nonetflux = PETSC_TRUE;
302a198735bSStefano Zampini   if (vl2l) {
303a198735bSStefano Zampini     ierr = PetscObjectReference((PetscObject)vl2l);CHKERRQ(ierr);
304fa23a32eSStefano Zampini     ierr = ISDestroy(&pcbddc->divudotp_vl2l);CHKERRQ(ierr);
305a198735bSStefano Zampini     pcbddc->divudotp_vl2l = vl2l;
306a198735bSStefano Zampini   }
307a198735bSStefano Zampini   PetscFunctionReturn(0);
308a198735bSStefano Zampini }
3093d996552SStefano Zampini 
310a198735bSStefano Zampini /*@
311a198735bSStefano Zampini  PCBDDCSetDivergenceMat - Sets the linear operator representing \int_\Omega \div {\bf u} \cdot p dx
312a198735bSStefano Zampini 
313a198735bSStefano Zampini    Collective on PC
314a198735bSStefano Zampini 
315a198735bSStefano Zampini    Input Parameters:
316a198735bSStefano Zampini +  pc - the preconditioning context
317a198735bSStefano Zampini .  divudotp - the matrix (must be of type MATIS)
3188ae0ca82SStefano Zampini .  trans - if trans if false (resp. true), then pressures are in the test (trial) space and velocities are in the trial (test) space.
319bb05f991SStefano 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
320a198735bSStefano Zampini 
321a198735bSStefano Zampini    Level: advanced
322a198735bSStefano Zampini 
3238ae0ca82SStefano Zampini    Notes: This auxiliary matrix is used to compute quadrature weights representing the net-flux across subdomain boundaries
324a198735bSStefano Zampini 
325a198735bSStefano Zampini .seealso: PCBDDC
326a198735bSStefano Zampini @*/
3278ae0ca82SStefano Zampini PetscErrorCode PCBDDCSetDivergenceMat(PC pc, Mat divudotp, PetscBool trans, IS vl2l)
328a198735bSStefano Zampini {
329a198735bSStefano Zampini   PetscBool      ismatis;
330a198735bSStefano Zampini   PetscErrorCode ierr;
331a198735bSStefano Zampini 
332a198735bSStefano Zampini   PetscFunctionBegin;
333a198735bSStefano Zampini   PetscValidHeaderSpecific(pc,PC_CLASSID,1);
334a198735bSStefano Zampini   PetscValidHeaderSpecific(divudotp,MAT_CLASSID,2);
335a198735bSStefano Zampini   PetscCheckSameComm(pc,1,divudotp,2);
3368ae0ca82SStefano Zampini   PetscValidLogicalCollectiveBool(pc,trans,3);
3378ae0ca82SStefano Zampini   if (vl2l) PetscValidHeaderSpecific(divudotp,IS_CLASSID,4);
338a198735bSStefano Zampini   ierr = PetscObjectTypeCompare((PetscObject)divudotp,MATIS,&ismatis);CHKERRQ(ierr);
339a198735bSStefano Zampini   if (!ismatis) SETERRQ(PetscObjectComm((PetscObject)pc),PETSC_ERR_ARG_WRONG,"Divergence matrix needs to be of type MATIS");
3408ae0ca82SStefano Zampini   ierr = PetscTryMethod(pc,"PCBDDCSetDivergenceMat_C",(PC,Mat,PetscBool,IS),(pc,divudotp,trans,vl2l));CHKERRQ(ierr);
341a198735bSStefano Zampini   PetscFunctionReturn(0);
342a198735bSStefano Zampini }
3432d505d7fSStefano Zampini 
3441dd7afcfSStefano Zampini static PetscErrorCode PCBDDCSetChangeOfBasisMat_BDDC(PC pc, Mat change, PetscBool interior)
345b9b85e73SStefano Zampini {
346b9b85e73SStefano Zampini   PC_BDDC        *pcbddc = (PC_BDDC*)pc->data;
347b9b85e73SStefano Zampini   PetscErrorCode ierr;
348b9b85e73SStefano Zampini 
349b9b85e73SStefano Zampini   PetscFunctionBegin;
350b9b85e73SStefano Zampini   ierr = PetscObjectReference((PetscObject)change);CHKERRQ(ierr);
35156282151SStefano Zampini   ierr = MatDestroy(&pcbddc->user_ChangeOfBasisMatrix);CHKERRQ(ierr);
352b9b85e73SStefano Zampini   pcbddc->user_ChangeOfBasisMatrix = change;
3531dd7afcfSStefano Zampini   pcbddc->change_interior = interior;
354b9b85e73SStefano Zampini   PetscFunctionReturn(0);
355b9b85e73SStefano Zampini }
356b9b85e73SStefano Zampini /*@
357906d46d4SStefano Zampini  PCBDDCSetChangeOfBasisMat - Set user defined change of basis for dofs
358b9b85e73SStefano Zampini 
359b9b85e73SStefano Zampini    Collective on PC
360b9b85e73SStefano Zampini 
361b9b85e73SStefano Zampini    Input Parameters:
362b9b85e73SStefano Zampini +  pc - the preconditioning context
3631dd7afcfSStefano Zampini .  change - the change of basis matrix
3641dd7afcfSStefano Zampini -  interior - whether or not the change of basis modifies interior dofs
365b9b85e73SStefano Zampini 
366b9b85e73SStefano Zampini    Level: intermediate
367b9b85e73SStefano Zampini 
368b9b85e73SStefano Zampini    Notes:
369b9b85e73SStefano Zampini 
370b9b85e73SStefano Zampini .seealso: PCBDDC
371b9b85e73SStefano Zampini @*/
3721dd7afcfSStefano Zampini PetscErrorCode PCBDDCSetChangeOfBasisMat(PC pc, Mat change, PetscBool interior)
373b9b85e73SStefano Zampini {
374b9b85e73SStefano Zampini   PetscErrorCode ierr;
375b9b85e73SStefano Zampini 
376b9b85e73SStefano Zampini   PetscFunctionBegin;
377b9b85e73SStefano Zampini   PetscValidHeaderSpecific(pc,PC_CLASSID,1);
378b9b85e73SStefano Zampini   PetscValidHeaderSpecific(change,MAT_CLASSID,2);
379906d46d4SStefano Zampini   PetscCheckSameComm(pc,1,change,2);
380906d46d4SStefano Zampini   if (pc->mat) {
381906d46d4SStefano Zampini     PetscInt rows_c,cols_c,rows,cols;
382906d46d4SStefano Zampini     ierr = MatGetSize(pc->mat,&rows,&cols);CHKERRQ(ierr);
383906d46d4SStefano Zampini     ierr = MatGetSize(change,&rows_c,&cols_c);CHKERRQ(ierr);
384e0fe2d75SToby 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);
385e0fe2d75SToby 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);
386906d46d4SStefano Zampini     ierr = MatGetLocalSize(pc->mat,&rows,&cols);CHKERRQ(ierr);
387906d46d4SStefano Zampini     ierr = MatGetLocalSize(change,&rows_c,&cols_c);CHKERRQ(ierr);
388e0fe2d75SToby 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);
389e0fe2d75SToby 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);
390906d46d4SStefano Zampini   }
3911dd7afcfSStefano Zampini   ierr = PetscTryMethod(pc,"PCBDDCSetChangeOfBasisMat_C",(PC,Mat,PetscBool),(pc,change,interior));CHKERRQ(ierr);
392b9b85e73SStefano Zampini   PetscFunctionReturn(0);
393b9b85e73SStefano Zampini }
3942d505d7fSStefano Zampini 
39530368db7SStefano Zampini static PetscErrorCode PCBDDCSetPrimalVerticesIS_BDDC(PC pc, IS PrimalVertices)
39630368db7SStefano Zampini {
39730368db7SStefano Zampini   PC_BDDC        *pcbddc = (PC_BDDC*)pc->data;
39856282151SStefano Zampini   PetscBool      isequal = PETSC_FALSE;
39930368db7SStefano Zampini   PetscErrorCode ierr;
40030368db7SStefano Zampini 
40130368db7SStefano Zampini   PetscFunctionBegin;
40256282151SStefano Zampini   ierr = PetscObjectReference((PetscObject)PrimalVertices);CHKERRQ(ierr);
40356282151SStefano Zampini   if (pcbddc->user_primal_vertices) {
40456282151SStefano Zampini     ierr = ISEqual(PrimalVertices,pcbddc->user_primal_vertices,&isequal);CHKERRQ(ierr);
40556282151SStefano Zampini   }
40630368db7SStefano Zampini   ierr = ISDestroy(&pcbddc->user_primal_vertices);CHKERRQ(ierr);
40730368db7SStefano Zampini   ierr = ISDestroy(&pcbddc->user_primal_vertices_local);CHKERRQ(ierr);
40830368db7SStefano Zampini   pcbddc->user_primal_vertices = PrimalVertices;
40956282151SStefano Zampini   if (!isequal) pcbddc->recompute_topography = PETSC_TRUE;
41030368db7SStefano Zampini   PetscFunctionReturn(0);
41130368db7SStefano Zampini }
41230368db7SStefano Zampini /*@
41330368db7SStefano Zampini  PCBDDCSetPrimalVerticesIS - Set additional user defined primal vertices in PCBDDC
41430368db7SStefano Zampini 
41530368db7SStefano Zampini    Collective
41630368db7SStefano Zampini 
41730368db7SStefano Zampini    Input Parameters:
41830368db7SStefano Zampini +  pc - the preconditioning context
41930368db7SStefano Zampini -  PrimalVertices - index set of primal vertices in global numbering (can be empty)
42030368db7SStefano Zampini 
42130368db7SStefano Zampini    Level: intermediate
42230368db7SStefano Zampini 
42330368db7SStefano Zampini    Notes:
42430368db7SStefano Zampini      Any process can list any global node
42530368db7SStefano Zampini 
42630368db7SStefano Zampini .seealso: PCBDDC
42730368db7SStefano Zampini @*/
42830368db7SStefano Zampini PetscErrorCode PCBDDCSetPrimalVerticesIS(PC pc, IS PrimalVertices)
42930368db7SStefano Zampini {
43030368db7SStefano Zampini   PetscErrorCode ierr;
43130368db7SStefano Zampini 
43230368db7SStefano Zampini   PetscFunctionBegin;
43330368db7SStefano Zampini   PetscValidHeaderSpecific(pc,PC_CLASSID,1);
43430368db7SStefano Zampini   PetscValidHeaderSpecific(PrimalVertices,IS_CLASSID,2);
43530368db7SStefano Zampini   PetscCheckSameComm(pc,1,PrimalVertices,2);
43630368db7SStefano Zampini   ierr = PetscTryMethod(pc,"PCBDDCSetPrimalVerticesIS_C",(PC,IS),(pc,PrimalVertices));CHKERRQ(ierr);
43730368db7SStefano Zampini   PetscFunctionReturn(0);
43830368db7SStefano Zampini }
4392d505d7fSStefano Zampini 
440674ae819SStefano Zampini static PetscErrorCode PCBDDCSetPrimalVerticesLocalIS_BDDC(PC pc, IS PrimalVertices)
441674ae819SStefano Zampini {
442674ae819SStefano Zampini   PC_BDDC        *pcbddc = (PC_BDDC*)pc->data;
44356282151SStefano Zampini   PetscBool      isequal = PETSC_FALSE;
444674ae819SStefano Zampini   PetscErrorCode ierr;
4451e6b0712SBarry Smith 
446674ae819SStefano Zampini   PetscFunctionBegin;
44756282151SStefano Zampini   ierr = PetscObjectReference((PetscObject)PrimalVertices);CHKERRQ(ierr);
44856282151SStefano Zampini   if (pcbddc->user_primal_vertices_local) {
44956282151SStefano Zampini     ierr = ISEqual(PrimalVertices,pcbddc->user_primal_vertices_local,&isequal);CHKERRQ(ierr);
45056282151SStefano Zampini   }
451674ae819SStefano Zampini   ierr = ISDestroy(&pcbddc->user_primal_vertices);CHKERRQ(ierr);
45230368db7SStefano Zampini   ierr = ISDestroy(&pcbddc->user_primal_vertices_local);CHKERRQ(ierr);
45330368db7SStefano Zampini   pcbddc->user_primal_vertices_local = PrimalVertices;
45456282151SStefano Zampini   if (!isequal) pcbddc->recompute_topography = PETSC_TRUE;
455674ae819SStefano Zampini   PetscFunctionReturn(0);
456674ae819SStefano Zampini }
457674ae819SStefano Zampini /*@
45828509bceSStefano Zampini  PCBDDCSetPrimalVerticesLocalIS - Set additional user defined primal vertices in PCBDDC
459674ae819SStefano Zampini 
46017eb1463SStefano Zampini    Collective
461674ae819SStefano Zampini 
462674ae819SStefano Zampini    Input Parameters:
463674ae819SStefano Zampini +  pc - the preconditioning context
46417eb1463SStefano Zampini -  PrimalVertices - index set of primal vertices in local numbering (can be empty)
465674ae819SStefano Zampini 
466674ae819SStefano Zampini    Level: intermediate
467674ae819SStefano Zampini 
468674ae819SStefano Zampini    Notes:
469674ae819SStefano Zampini 
470674ae819SStefano Zampini .seealso: PCBDDC
471674ae819SStefano Zampini @*/
472674ae819SStefano Zampini PetscErrorCode PCBDDCSetPrimalVerticesLocalIS(PC pc, IS PrimalVertices)
473674ae819SStefano Zampini {
474674ae819SStefano Zampini   PetscErrorCode ierr;
475674ae819SStefano Zampini 
476674ae819SStefano Zampini   PetscFunctionBegin;
477674ae819SStefano Zampini   PetscValidHeaderSpecific(pc,PC_CLASSID,1);
478674ae819SStefano Zampini   PetscValidHeaderSpecific(PrimalVertices,IS_CLASSID,2);
47917eb1463SStefano Zampini   PetscCheckSameComm(pc,1,PrimalVertices,2);
480674ae819SStefano Zampini   ierr = PetscTryMethod(pc,"PCBDDCSetPrimalVerticesLocalIS_C",(PC,IS),(pc,PrimalVertices));CHKERRQ(ierr);
481674ae819SStefano Zampini   PetscFunctionReturn(0);
482674ae819SStefano Zampini }
4832d505d7fSStefano Zampini 
4844fad6a16SStefano Zampini static PetscErrorCode PCBDDCSetCoarseningRatio_BDDC(PC pc,PetscInt k)
4854fad6a16SStefano Zampini {
4864fad6a16SStefano Zampini   PC_BDDC  *pcbddc = (PC_BDDC*)pc->data;
4874fad6a16SStefano Zampini 
4884fad6a16SStefano Zampini   PetscFunctionBegin;
4894fad6a16SStefano Zampini   pcbddc->coarsening_ratio = k;
4904fad6a16SStefano Zampini   PetscFunctionReturn(0);
4914fad6a16SStefano Zampini }
4921e6b0712SBarry Smith 
4934fad6a16SStefano Zampini /*@
49428509bceSStefano Zampini  PCBDDCSetCoarseningRatio - Set coarsening ratio used in multilevel
4954fad6a16SStefano Zampini 
4964fad6a16SStefano Zampini    Logically collective on PC
4974fad6a16SStefano Zampini 
4984fad6a16SStefano Zampini    Input Parameters:
4994fad6a16SStefano Zampini +  pc - the preconditioning context
50028509bceSStefano Zampini -  k - coarsening ratio (H/h at the coarser level)
5014fad6a16SStefano Zampini 
5020f202f7eSStefano Zampini    Options Database Keys:
5030f202f7eSStefano Zampini .    -pc_bddc_coarsening_ratio
5044fad6a16SStefano Zampini 
5054fad6a16SStefano Zampini    Level: intermediate
5064fad6a16SStefano Zampini 
5074fad6a16SStefano Zampini    Notes:
5080f202f7eSStefano Zampini      Approximatively k subdomains at the finer level will be aggregated into a single subdomain at the coarser level
5094fad6a16SStefano Zampini 
5100f202f7eSStefano Zampini .seealso: PCBDDC, PCBDDCSetLevels()
5114fad6a16SStefano Zampini @*/
5124fad6a16SStefano Zampini PetscErrorCode PCBDDCSetCoarseningRatio(PC pc,PetscInt k)
5134fad6a16SStefano Zampini {
5144fad6a16SStefano Zampini   PetscErrorCode ierr;
5154fad6a16SStefano Zampini 
5164fad6a16SStefano Zampini   PetscFunctionBegin;
5174fad6a16SStefano Zampini   PetscValidHeaderSpecific(pc,PC_CLASSID,1);
5182b510759SStefano Zampini   PetscValidLogicalCollectiveInt(pc,k,2);
5194fad6a16SStefano Zampini   ierr = PetscTryMethod(pc,"PCBDDCSetCoarseningRatio_C",(PC,PetscInt),(pc,k));CHKERRQ(ierr);
5204fad6a16SStefano Zampini   PetscFunctionReturn(0);
5214fad6a16SStefano Zampini }
5222b510759SStefano Zampini 
523b8ffe317SStefano Zampini /* The following functions (PCBDDCSetUseExactDirichlet PCBDDCSetLevel) are not public */
524b8ffe317SStefano Zampini static PetscErrorCode PCBDDCSetUseExactDirichlet_BDDC(PC pc,PetscBool flg)
525b8ffe317SStefano Zampini {
526b8ffe317SStefano Zampini   PC_BDDC  *pcbddc = (PC_BDDC*)pc->data;
527b8ffe317SStefano Zampini 
528b8ffe317SStefano Zampini   PetscFunctionBegin;
52985c4d303SStefano Zampini   pcbddc->use_exact_dirichlet_trick = flg;
530b8ffe317SStefano Zampini   PetscFunctionReturn(0);
531b8ffe317SStefano Zampini }
532b8ffe317SStefano Zampini 
533b8ffe317SStefano Zampini PetscErrorCode PCBDDCSetUseExactDirichlet(PC pc,PetscBool flg)
5342b510759SStefano Zampini {
5352b510759SStefano Zampini   PetscErrorCode ierr;
5362b510759SStefano Zampini 
5372b510759SStefano Zampini   PetscFunctionBegin;
5382b510759SStefano Zampini   PetscValidHeaderSpecific(pc,PC_CLASSID,1);
539b8ffe317SStefano Zampini   PetscValidLogicalCollectiveBool(pc,flg,2);
540b8ffe317SStefano Zampini   ierr = PetscTryMethod(pc,"PCBDDCSetUseExactDirichlet_C",(PC,PetscBool),(pc,flg));CHKERRQ(ierr);
5412b510759SStefano Zampini   PetscFunctionReturn(0);
5422b510759SStefano Zampini }
5431e6b0712SBarry Smith 
5442b510759SStefano Zampini static PetscErrorCode PCBDDCSetLevel_BDDC(PC pc,PetscInt level)
5454fad6a16SStefano Zampini {
5464fad6a16SStefano Zampini   PC_BDDC  *pcbddc = (PC_BDDC*)pc->data;
5474fad6a16SStefano Zampini 
5484fad6a16SStefano Zampini   PetscFunctionBegin;
5492b510759SStefano Zampini   pcbddc->current_level = level;
5504fad6a16SStefano Zampini   PetscFunctionReturn(0);
5514fad6a16SStefano Zampini }
5521e6b0712SBarry Smith 
553b8ffe317SStefano Zampini PetscErrorCode PCBDDCSetLevel(PC pc,PetscInt level)
554b8ffe317SStefano Zampini {
555b8ffe317SStefano Zampini   PetscErrorCode ierr;
556b8ffe317SStefano Zampini 
557b8ffe317SStefano Zampini   PetscFunctionBegin;
558b8ffe317SStefano Zampini   PetscValidHeaderSpecific(pc,PC_CLASSID,1);
559b8ffe317SStefano Zampini   PetscValidLogicalCollectiveInt(pc,level,2);
560b8ffe317SStefano Zampini   ierr = PetscTryMethod(pc,"PCBDDCSetLevel_C",(PC,PetscInt),(pc,level));CHKERRQ(ierr);
561b8ffe317SStefano Zampini   PetscFunctionReturn(0);
562b8ffe317SStefano Zampini }
563b8ffe317SStefano Zampini 
5642b510759SStefano Zampini static PetscErrorCode PCBDDCSetLevels_BDDC(PC pc,PetscInt levels)
5652b510759SStefano Zampini {
5662b510759SStefano Zampini   PC_BDDC  *pcbddc = (PC_BDDC*)pc->data;
5672b510759SStefano Zampini 
5682b510759SStefano Zampini   PetscFunctionBegin;
5692b510759SStefano Zampini   pcbddc->max_levels = levels;
5702b510759SStefano Zampini   PetscFunctionReturn(0);
5712b510759SStefano Zampini }
5722b510759SStefano Zampini 
5734fad6a16SStefano Zampini /*@
57428509bceSStefano Zampini  PCBDDCSetLevels - Sets the maximum number of levels for multilevel
5754fad6a16SStefano Zampini 
5764fad6a16SStefano Zampini    Logically collective on PC
5774fad6a16SStefano Zampini 
5784fad6a16SStefano Zampini    Input Parameters:
5794fad6a16SStefano Zampini +  pc - the preconditioning context
58028509bceSStefano Zampini -  levels - the maximum number of levels (max 9)
5814fad6a16SStefano Zampini 
5820f202f7eSStefano Zampini    Options Database Keys:
5830f202f7eSStefano Zampini .    -pc_bddc_levels
5844fad6a16SStefano Zampini 
5854fad6a16SStefano Zampini    Level: intermediate
5864fad6a16SStefano Zampini 
5874fad6a16SStefano Zampini    Notes:
5880f202f7eSStefano Zampini      Default value is 0, i.e. traditional one-level BDDC
5894fad6a16SStefano Zampini 
5900f202f7eSStefano Zampini .seealso: PCBDDC, PCBDDCSetCoarseningRatio()
5914fad6a16SStefano Zampini @*/
5922b510759SStefano Zampini PetscErrorCode PCBDDCSetLevels(PC pc,PetscInt levels)
5934fad6a16SStefano Zampini {
5944fad6a16SStefano Zampini   PetscErrorCode ierr;
5954fad6a16SStefano Zampini 
5964fad6a16SStefano Zampini   PetscFunctionBegin;
5974fad6a16SStefano Zampini   PetscValidHeaderSpecific(pc,PC_CLASSID,1);
5982b510759SStefano Zampini   PetscValidLogicalCollectiveInt(pc,levels,2);
599312be037SStefano Zampini   if (levels > 99) SETERRQ(PetscObjectComm((PetscObject)pc),PETSC_ERR_SUP,"Maximum number of levels for bddc is 99\n");
6002b510759SStefano Zampini   ierr = PetscTryMethod(pc,"PCBDDCSetLevels_C",(PC,PetscInt),(pc,levels));CHKERRQ(ierr);
6014fad6a16SStefano Zampini   PetscFunctionReturn(0);
6024fad6a16SStefano Zampini }
6031e6b0712SBarry Smith 
6043b03a366Sstefano_zampini static PetscErrorCode PCBDDCSetDirichletBoundaries_BDDC(PC pc,IS DirichletBoundaries)
6053b03a366Sstefano_zampini {
6063b03a366Sstefano_zampini   PC_BDDC        *pcbddc = (PC_BDDC*)pc->data;
60756282151SStefano Zampini   PetscBool      isequal = PETSC_FALSE;
6083b03a366Sstefano_zampini   PetscErrorCode ierr;
6093b03a366Sstefano_zampini 
6103b03a366Sstefano_zampini   PetscFunctionBegin;
61156282151SStefano Zampini   ierr = PetscObjectReference((PetscObject)DirichletBoundaries);CHKERRQ(ierr);
61256282151SStefano Zampini   if (pcbddc->DirichletBoundaries) {
61356282151SStefano Zampini     ierr = ISEqual(DirichletBoundaries,pcbddc->DirichletBoundaries,&isequal);CHKERRQ(ierr);
61456282151SStefano Zampini   }
615785d1243SStefano Zampini   /* last user setting takes precendence -> destroy any other customization */
616785d1243SStefano Zampini   ierr = ISDestroy(&pcbddc->DirichletBoundariesLocal);CHKERRQ(ierr);
6173b03a366Sstefano_zampini   ierr = ISDestroy(&pcbddc->DirichletBoundaries);CHKERRQ(ierr);
61836e030ebSStefano Zampini   pcbddc->DirichletBoundaries = DirichletBoundaries;
61956282151SStefano Zampini   if (!isequal) pcbddc->recompute_topography = PETSC_TRUE;
6203b03a366Sstefano_zampini   PetscFunctionReturn(0);
6213b03a366Sstefano_zampini }
6221e6b0712SBarry Smith 
6233b03a366Sstefano_zampini /*@
62428509bceSStefano Zampini  PCBDDCSetDirichletBoundaries - Set IS defining Dirichlet boundaries for the global problem.
6253b03a366Sstefano_zampini 
626785d1243SStefano Zampini    Collective
6273b03a366Sstefano_zampini 
6283b03a366Sstefano_zampini    Input Parameters:
6293b03a366Sstefano_zampini +  pc - the preconditioning context
630785d1243SStefano Zampini -  DirichletBoundaries - parallel IS defining the Dirichlet boundaries
6313b03a366Sstefano_zampini 
6323b03a366Sstefano_zampini    Level: intermediate
6333b03a366Sstefano_zampini 
6340f202f7eSStefano Zampini    Notes:
6350f202f7eSStefano Zampini      Provide the information if you used MatZeroRows/Columns routines. Any process can list any global node
6363b03a366Sstefano_zampini 
6370f202f7eSStefano Zampini .seealso: PCBDDC, PCBDDCSetDirichletBoundariesLocal(), MatZeroRows(), MatZeroRowsColumns()
6383b03a366Sstefano_zampini @*/
6393b03a366Sstefano_zampini PetscErrorCode PCBDDCSetDirichletBoundaries(PC pc,IS DirichletBoundaries)
6403b03a366Sstefano_zampini {
6413b03a366Sstefano_zampini   PetscErrorCode ierr;
6423b03a366Sstefano_zampini 
6433b03a366Sstefano_zampini   PetscFunctionBegin;
6443b03a366Sstefano_zampini   PetscValidHeaderSpecific(pc,PC_CLASSID,1);
645674ae819SStefano Zampini   PetscValidHeaderSpecific(DirichletBoundaries,IS_CLASSID,2);
646785d1243SStefano Zampini   PetscCheckSameComm(pc,1,DirichletBoundaries,2);
6473b03a366Sstefano_zampini   ierr = PetscTryMethod(pc,"PCBDDCSetDirichletBoundaries_C",(PC,IS),(pc,DirichletBoundaries));CHKERRQ(ierr);
6483b03a366Sstefano_zampini   PetscFunctionReturn(0);
6493b03a366Sstefano_zampini }
6501e6b0712SBarry Smith 
65182ba6b80SStefano Zampini static PetscErrorCode PCBDDCSetDirichletBoundariesLocal_BDDC(PC pc,IS DirichletBoundaries)
6523b03a366Sstefano_zampini {
6533b03a366Sstefano_zampini   PC_BDDC        *pcbddc = (PC_BDDC*)pc->data;
65456282151SStefano Zampini   PetscBool      isequal = PETSC_FALSE;
6553b03a366Sstefano_zampini   PetscErrorCode ierr;
6563b03a366Sstefano_zampini 
6573b03a366Sstefano_zampini   PetscFunctionBegin;
65856282151SStefano Zampini   ierr = PetscObjectReference((PetscObject)DirichletBoundaries);CHKERRQ(ierr);
65956282151SStefano Zampini   if (pcbddc->DirichletBoundariesLocal) {
66056282151SStefano Zampini     ierr = ISEqual(DirichletBoundaries,pcbddc->DirichletBoundariesLocal,&isequal);CHKERRQ(ierr);
66156282151SStefano Zampini   }
662785d1243SStefano Zampini   /* last user setting takes precendence -> destroy any other customization */
663785d1243SStefano Zampini   ierr = ISDestroy(&pcbddc->DirichletBoundariesLocal);CHKERRQ(ierr);
6643b03a366Sstefano_zampini   ierr = ISDestroy(&pcbddc->DirichletBoundaries);CHKERRQ(ierr);
665785d1243SStefano Zampini   pcbddc->DirichletBoundariesLocal = DirichletBoundaries;
66656282151SStefano Zampini   if (!isequal) pcbddc->recompute_topography = PETSC_TRUE;
6673b03a366Sstefano_zampini   PetscFunctionReturn(0);
6683b03a366Sstefano_zampini }
6693b03a366Sstefano_zampini 
6703b03a366Sstefano_zampini /*@
67182ba6b80SStefano Zampini  PCBDDCSetDirichletBoundariesLocal - Set IS defining Dirichlet boundaries for the global problem in local ordering.
6723b03a366Sstefano_zampini 
673785d1243SStefano Zampini    Collective
6743b03a366Sstefano_zampini 
6753b03a366Sstefano_zampini    Input Parameters:
6763b03a366Sstefano_zampini +  pc - the preconditioning context
67782ba6b80SStefano Zampini -  DirichletBoundaries - parallel IS defining the Dirichlet boundaries (in local ordering)
6783b03a366Sstefano_zampini 
6793b03a366Sstefano_zampini    Level: intermediate
6803b03a366Sstefano_zampini 
6813b03a366Sstefano_zampini    Notes:
6823b03a366Sstefano_zampini 
6830f202f7eSStefano Zampini .seealso: PCBDDC, PCBDDCSetDirichletBoundaries(), MatZeroRows(), MatZeroRowsColumns()
6843b03a366Sstefano_zampini @*/
68582ba6b80SStefano Zampini PetscErrorCode PCBDDCSetDirichletBoundariesLocal(PC pc,IS DirichletBoundaries)
6863b03a366Sstefano_zampini {
6873b03a366Sstefano_zampini   PetscErrorCode ierr;
6883b03a366Sstefano_zampini 
6893b03a366Sstefano_zampini   PetscFunctionBegin;
6903b03a366Sstefano_zampini   PetscValidHeaderSpecific(pc,PC_CLASSID,1);
6913b03a366Sstefano_zampini   PetscValidHeaderSpecific(DirichletBoundaries,IS_CLASSID,2);
69282ba6b80SStefano Zampini   PetscCheckSameComm(pc,1,DirichletBoundaries,2);
69382ba6b80SStefano Zampini   ierr = PetscTryMethod(pc,"PCBDDCSetDirichletBoundariesLocal_C",(PC,IS),(pc,DirichletBoundaries));CHKERRQ(ierr);
6943b03a366Sstefano_zampini   PetscFunctionReturn(0);
6953b03a366Sstefano_zampini }
6963b03a366Sstefano_zampini 
69753cdbc3dSStefano Zampini static PetscErrorCode PCBDDCSetNeumannBoundaries_BDDC(PC pc,IS NeumannBoundaries)
6980c7d97c5SJed Brown {
6990c7d97c5SJed Brown   PC_BDDC        *pcbddc = (PC_BDDC*)pc->data;
70056282151SStefano Zampini   PetscBool      isequal = PETSC_FALSE;
70153cdbc3dSStefano Zampini   PetscErrorCode ierr;
7020c7d97c5SJed Brown 
7030c7d97c5SJed Brown   PetscFunctionBegin;
70456282151SStefano Zampini   ierr = PetscObjectReference((PetscObject)NeumannBoundaries);CHKERRQ(ierr);
70556282151SStefano Zampini   if (pcbddc->NeumannBoundaries) {
70656282151SStefano Zampini     ierr = ISEqual(NeumannBoundaries,pcbddc->NeumannBoundaries,&isequal);CHKERRQ(ierr);
70756282151SStefano Zampini   }
708785d1243SStefano Zampini   /* last user setting takes precendence -> destroy any other customization */
709785d1243SStefano Zampini   ierr = ISDestroy(&pcbddc->NeumannBoundariesLocal);CHKERRQ(ierr);
71053cdbc3dSStefano Zampini   ierr = ISDestroy(&pcbddc->NeumannBoundaries);CHKERRQ(ierr);
71136e030ebSStefano Zampini   pcbddc->NeumannBoundaries = NeumannBoundaries;
71256282151SStefano Zampini   if (!isequal) pcbddc->recompute_topography = PETSC_TRUE;
7130c7d97c5SJed Brown   PetscFunctionReturn(0);
7140c7d97c5SJed Brown }
7151e6b0712SBarry Smith 
71657527edcSJed Brown /*@
71728509bceSStefano Zampini  PCBDDCSetNeumannBoundaries - Set IS defining Neumann boundaries for the global problem.
71857527edcSJed Brown 
719785d1243SStefano Zampini    Collective
72057527edcSJed Brown 
72157527edcSJed Brown    Input Parameters:
72257527edcSJed Brown +  pc - the preconditioning context
723785d1243SStefano Zampini -  NeumannBoundaries - parallel IS defining the Neumann boundaries
72457527edcSJed Brown 
72557527edcSJed Brown    Level: intermediate
72657527edcSJed Brown 
7270f202f7eSStefano Zampini    Notes:
7280f202f7eSStefano Zampini      Any process can list any global node
72957527edcSJed Brown 
7300f202f7eSStefano Zampini .seealso: PCBDDC, PCBDDCSetNeumannBoundariesLocal()
73157527edcSJed Brown @*/
73253cdbc3dSStefano Zampini PetscErrorCode PCBDDCSetNeumannBoundaries(PC pc,IS NeumannBoundaries)
7330c7d97c5SJed Brown {
7340c7d97c5SJed Brown   PetscErrorCode ierr;
7350c7d97c5SJed Brown 
7360c7d97c5SJed Brown   PetscFunctionBegin;
7370c7d97c5SJed Brown   PetscValidHeaderSpecific(pc,PC_CLASSID,1);
738674ae819SStefano Zampini   PetscValidHeaderSpecific(NeumannBoundaries,IS_CLASSID,2);
739785d1243SStefano Zampini   PetscCheckSameComm(pc,1,NeumannBoundaries,2);
74053cdbc3dSStefano Zampini   ierr = PetscTryMethod(pc,"PCBDDCSetNeumannBoundaries_C",(PC,IS),(pc,NeumannBoundaries));CHKERRQ(ierr);
74153cdbc3dSStefano Zampini   PetscFunctionReturn(0);
74253cdbc3dSStefano Zampini }
7431e6b0712SBarry Smith 
74482ba6b80SStefano Zampini static PetscErrorCode PCBDDCSetNeumannBoundariesLocal_BDDC(PC pc,IS NeumannBoundaries)
7450c7d97c5SJed Brown {
7460c7d97c5SJed Brown   PC_BDDC        *pcbddc = (PC_BDDC*)pc->data;
74756282151SStefano Zampini   PetscBool      isequal = PETSC_FALSE;
7480c7d97c5SJed Brown   PetscErrorCode ierr;
7490c7d97c5SJed Brown 
7500c7d97c5SJed Brown   PetscFunctionBegin;
75156282151SStefano Zampini   ierr = PetscObjectReference((PetscObject)NeumannBoundaries);CHKERRQ(ierr);
75256282151SStefano Zampini   if (pcbddc->NeumannBoundariesLocal) {
75356282151SStefano Zampini     ierr = ISEqual(NeumannBoundaries,pcbddc->NeumannBoundariesLocal,&isequal);CHKERRQ(ierr);
75456282151SStefano Zampini   }
755785d1243SStefano Zampini   /* last user setting takes precendence -> destroy any other customization */
756785d1243SStefano Zampini   ierr = ISDestroy(&pcbddc->NeumannBoundariesLocal);CHKERRQ(ierr);
7570c7d97c5SJed Brown   ierr = ISDestroy(&pcbddc->NeumannBoundaries);CHKERRQ(ierr);
758785d1243SStefano Zampini   pcbddc->NeumannBoundariesLocal = NeumannBoundaries;
75956282151SStefano Zampini   if (!isequal) pcbddc->recompute_topography = PETSC_TRUE;
7600c7d97c5SJed Brown   PetscFunctionReturn(0);
7610c7d97c5SJed Brown }
7620c7d97c5SJed Brown 
7630c7d97c5SJed Brown /*@
76482ba6b80SStefano Zampini  PCBDDCSetNeumannBoundariesLocal - Set IS defining Neumann boundaries for the global problem in local ordering.
7650c7d97c5SJed Brown 
766785d1243SStefano Zampini    Collective
7670c7d97c5SJed Brown 
7680c7d97c5SJed Brown    Input Parameters:
7690c7d97c5SJed Brown +  pc - the preconditioning context
77082ba6b80SStefano Zampini -  NeumannBoundaries - parallel IS defining the subdomain part of Neumann boundaries (in local ordering)
7710c7d97c5SJed Brown 
7720c7d97c5SJed Brown    Level: intermediate
7730c7d97c5SJed Brown 
7740c7d97c5SJed Brown    Notes:
7750c7d97c5SJed Brown 
7760f202f7eSStefano Zampini .seealso: PCBDDC, PCBDDCSetNeumannBoundaries()
7770c7d97c5SJed Brown @*/
77882ba6b80SStefano Zampini PetscErrorCode PCBDDCSetNeumannBoundariesLocal(PC pc,IS NeumannBoundaries)
7790c7d97c5SJed Brown {
7800c7d97c5SJed Brown   PetscErrorCode ierr;
7810c7d97c5SJed Brown 
7820c7d97c5SJed Brown   PetscFunctionBegin;
7830c7d97c5SJed Brown   PetscValidHeaderSpecific(pc,PC_CLASSID,1);
7840c7d97c5SJed Brown   PetscValidHeaderSpecific(NeumannBoundaries,IS_CLASSID,2);
78582ba6b80SStefano Zampini   PetscCheckSameComm(pc,1,NeumannBoundaries,2);
78682ba6b80SStefano Zampini   ierr = PetscTryMethod(pc,"PCBDDCSetNeumannBoundariesLocal_C",(PC,IS),(pc,NeumannBoundaries));CHKERRQ(ierr);
78753cdbc3dSStefano Zampini   PetscFunctionReturn(0);
78853cdbc3dSStefano Zampini }
78953cdbc3dSStefano Zampini 
790da1bb401SStefano Zampini static PetscErrorCode PCBDDCGetDirichletBoundaries_BDDC(PC pc,IS *DirichletBoundaries)
791da1bb401SStefano Zampini {
792da1bb401SStefano Zampini   PC_BDDC  *pcbddc = (PC_BDDC*)pc->data;
793da1bb401SStefano Zampini 
794da1bb401SStefano Zampini   PetscFunctionBegin;
795da1bb401SStefano Zampini   *DirichletBoundaries = pcbddc->DirichletBoundaries;
796da1bb401SStefano Zampini   PetscFunctionReturn(0);
797da1bb401SStefano Zampini }
7981e6b0712SBarry Smith 
799da1bb401SStefano Zampini /*@
800785d1243SStefano Zampini  PCBDDCGetDirichletBoundaries - Get parallel IS for Dirichlet boundaries
801da1bb401SStefano Zampini 
802785d1243SStefano Zampini    Collective
803785d1243SStefano Zampini 
804785d1243SStefano Zampini    Input Parameters:
805785d1243SStefano Zampini .  pc - the preconditioning context
806785d1243SStefano Zampini 
807785d1243SStefano Zampini    Output Parameters:
808785d1243SStefano Zampini .  DirichletBoundaries - index set defining the Dirichlet boundaries
809785d1243SStefano Zampini 
810785d1243SStefano Zampini    Level: intermediate
811785d1243SStefano Zampini 
8120f202f7eSStefano Zampini    Notes:
8130f202f7eSStefano Zampini      The IS returned (if any) is the same passed in earlier by the user with PCBDDCSetDirichletBoundaries
814785d1243SStefano Zampini 
815785d1243SStefano Zampini .seealso: PCBDDC
816785d1243SStefano Zampini @*/
817785d1243SStefano Zampini PetscErrorCode PCBDDCGetDirichletBoundaries(PC pc,IS *DirichletBoundaries)
818785d1243SStefano Zampini {
819785d1243SStefano Zampini   PetscErrorCode ierr;
820785d1243SStefano Zampini 
821785d1243SStefano Zampini   PetscFunctionBegin;
822785d1243SStefano Zampini   PetscValidHeaderSpecific(pc,PC_CLASSID,1);
823785d1243SStefano Zampini   ierr = PetscUseMethod(pc,"PCBDDCGetDirichletBoundaries_C",(PC,IS*),(pc,DirichletBoundaries));CHKERRQ(ierr);
824785d1243SStefano Zampini   PetscFunctionReturn(0);
825785d1243SStefano Zampini }
826785d1243SStefano Zampini 
827785d1243SStefano Zampini static PetscErrorCode PCBDDCGetDirichletBoundariesLocal_BDDC(PC pc,IS *DirichletBoundaries)
828785d1243SStefano Zampini {
829785d1243SStefano Zampini   PC_BDDC  *pcbddc = (PC_BDDC*)pc->data;
830785d1243SStefano Zampini 
831785d1243SStefano Zampini   PetscFunctionBegin;
832785d1243SStefano Zampini   *DirichletBoundaries = pcbddc->DirichletBoundariesLocal;
833785d1243SStefano Zampini   PetscFunctionReturn(0);
834785d1243SStefano Zampini }
835785d1243SStefano Zampini 
836da1bb401SStefano Zampini /*@
83782ba6b80SStefano Zampini  PCBDDCGetDirichletBoundariesLocal - Get parallel IS for Dirichlet boundaries (in local ordering)
838da1bb401SStefano Zampini 
839785d1243SStefano Zampini    Collective
840da1bb401SStefano Zampini 
841da1bb401SStefano Zampini    Input Parameters:
84228509bceSStefano Zampini .  pc - the preconditioning context
843da1bb401SStefano Zampini 
844da1bb401SStefano Zampini    Output Parameters:
84528509bceSStefano Zampini .  DirichletBoundaries - index set defining the subdomain part of Dirichlet boundaries
846da1bb401SStefano Zampini 
847da1bb401SStefano Zampini    Level: intermediate
848da1bb401SStefano Zampini 
849da1bb401SStefano Zampini    Notes:
8500f202f7eSStefano 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).
8510f202f7eSStefano Zampini           In the latter case, the IS will be available after PCSetUp.
852da1bb401SStefano Zampini 
853da1bb401SStefano Zampini .seealso: PCBDDC
854da1bb401SStefano Zampini @*/
85582ba6b80SStefano Zampini PetscErrorCode PCBDDCGetDirichletBoundariesLocal(PC pc,IS *DirichletBoundaries)
856da1bb401SStefano Zampini {
857da1bb401SStefano Zampini   PetscErrorCode ierr;
858da1bb401SStefano Zampini 
859da1bb401SStefano Zampini   PetscFunctionBegin;
860da1bb401SStefano Zampini   PetscValidHeaderSpecific(pc,PC_CLASSID,1);
86182ba6b80SStefano Zampini   ierr = PetscUseMethod(pc,"PCBDDCGetDirichletBoundariesLocal_C",(PC,IS*),(pc,DirichletBoundaries));CHKERRQ(ierr);
862da1bb401SStefano Zampini   PetscFunctionReturn(0);
863da1bb401SStefano Zampini }
8641e6b0712SBarry Smith 
86553cdbc3dSStefano Zampini static PetscErrorCode PCBDDCGetNeumannBoundaries_BDDC(PC pc,IS *NeumannBoundaries)
86653cdbc3dSStefano Zampini {
86753cdbc3dSStefano Zampini   PC_BDDC  *pcbddc = (PC_BDDC*)pc->data;
86853cdbc3dSStefano Zampini 
86953cdbc3dSStefano Zampini   PetscFunctionBegin;
87053cdbc3dSStefano Zampini   *NeumannBoundaries = pcbddc->NeumannBoundaries;
87153cdbc3dSStefano Zampini   PetscFunctionReturn(0);
87253cdbc3dSStefano Zampini }
8731e6b0712SBarry Smith 
87453cdbc3dSStefano Zampini /*@
875785d1243SStefano Zampini  PCBDDCGetNeumannBoundaries - Get parallel IS for Neumann boundaries
87653cdbc3dSStefano Zampini 
877785d1243SStefano Zampini    Collective
878785d1243SStefano Zampini 
879785d1243SStefano Zampini    Input Parameters:
880785d1243SStefano Zampini .  pc - the preconditioning context
881785d1243SStefano Zampini 
882785d1243SStefano Zampini    Output Parameters:
883785d1243SStefano Zampini .  NeumannBoundaries - index set defining the Neumann boundaries
884785d1243SStefano Zampini 
885785d1243SStefano Zampini    Level: intermediate
886785d1243SStefano Zampini 
8870f202f7eSStefano Zampini    Notes:
8880f202f7eSStefano Zampini      The IS returned (if any) is the same passed in earlier by the user with PCBDDCSetNeumannBoundaries
889785d1243SStefano Zampini 
890785d1243SStefano Zampini .seealso: PCBDDC
891785d1243SStefano Zampini @*/
892785d1243SStefano Zampini PetscErrorCode PCBDDCGetNeumannBoundaries(PC pc,IS *NeumannBoundaries)
893785d1243SStefano Zampini {
894785d1243SStefano Zampini   PetscErrorCode ierr;
895785d1243SStefano Zampini 
896785d1243SStefano Zampini   PetscFunctionBegin;
897785d1243SStefano Zampini   PetscValidHeaderSpecific(pc,PC_CLASSID,1);
898785d1243SStefano Zampini   ierr = PetscUseMethod(pc,"PCBDDCGetNeumannBoundaries_C",(PC,IS*),(pc,NeumannBoundaries));CHKERRQ(ierr);
899785d1243SStefano Zampini   PetscFunctionReturn(0);
900785d1243SStefano Zampini }
901785d1243SStefano Zampini 
902785d1243SStefano Zampini static PetscErrorCode PCBDDCGetNeumannBoundariesLocal_BDDC(PC pc,IS *NeumannBoundaries)
903785d1243SStefano Zampini {
904785d1243SStefano Zampini   PC_BDDC  *pcbddc = (PC_BDDC*)pc->data;
905785d1243SStefano Zampini 
906785d1243SStefano Zampini   PetscFunctionBegin;
907785d1243SStefano Zampini   *NeumannBoundaries = pcbddc->NeumannBoundariesLocal;
908785d1243SStefano Zampini   PetscFunctionReturn(0);
909785d1243SStefano Zampini }
910785d1243SStefano Zampini 
91153cdbc3dSStefano Zampini /*@
91282ba6b80SStefano Zampini  PCBDDCGetNeumannBoundariesLocal - Get parallel IS for Neumann boundaries (in local ordering)
91353cdbc3dSStefano Zampini 
914785d1243SStefano Zampini    Collective
91553cdbc3dSStefano Zampini 
91653cdbc3dSStefano Zampini    Input Parameters:
91728509bceSStefano Zampini .  pc - the preconditioning context
91853cdbc3dSStefano Zampini 
91953cdbc3dSStefano Zampini    Output Parameters:
92028509bceSStefano Zampini .  NeumannBoundaries - index set defining the subdomain part of Neumann boundaries
92153cdbc3dSStefano Zampini 
92253cdbc3dSStefano Zampini    Level: intermediate
92353cdbc3dSStefano Zampini 
92453cdbc3dSStefano Zampini    Notes:
9250f202f7eSStefano 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).
9260f202f7eSStefano Zampini           In the latter case, the IS will be available after PCSetUp.
92753cdbc3dSStefano Zampini 
92853cdbc3dSStefano Zampini .seealso: PCBDDC
92953cdbc3dSStefano Zampini @*/
93082ba6b80SStefano Zampini PetscErrorCode PCBDDCGetNeumannBoundariesLocal(PC pc,IS *NeumannBoundaries)
93153cdbc3dSStefano Zampini {
93253cdbc3dSStefano Zampini   PetscErrorCode ierr;
93353cdbc3dSStefano Zampini 
93453cdbc3dSStefano Zampini   PetscFunctionBegin;
93553cdbc3dSStefano Zampini   PetscValidHeaderSpecific(pc,PC_CLASSID,1);
93682ba6b80SStefano Zampini   ierr = PetscUseMethod(pc,"PCBDDCGetNeumannBoundariesLocal_C",(PC,IS*),(pc,NeumannBoundaries));CHKERRQ(ierr);
9370c7d97c5SJed Brown   PetscFunctionReturn(0);
9380c7d97c5SJed Brown }
9391e6b0712SBarry Smith 
9401a83f524SJed Brown static PetscErrorCode PCBDDCSetLocalAdjacencyGraph_BDDC(PC pc, PetscInt nvtxs,const PetscInt xadj[],const PetscInt adjncy[], PetscCopyMode copymode)
94136e030ebSStefano Zampini {
94236e030ebSStefano Zampini   PC_BDDC        *pcbddc = (PC_BDDC*)pc->data;
943da1bb401SStefano Zampini   PCBDDCGraph    mat_graph = pcbddc->mat_graph;
94456282151SStefano Zampini   PetscBool      same_data = PETSC_FALSE;
945da1bb401SStefano Zampini   PetscErrorCode ierr;
94636e030ebSStefano Zampini 
94736e030ebSStefano Zampini   PetscFunctionBegin;
9488687889aSStefano Zampini   if (!nvtxs) {
94904194a47SStefano Zampini     if (copymode == PETSC_OWN_POINTER) {
95004194a47SStefano Zampini       ierr = PetscFree(xadj);CHKERRQ(ierr);
95104194a47SStefano Zampini       ierr = PetscFree(adjncy);CHKERRQ(ierr);
95204194a47SStefano Zampini     }
9538687889aSStefano Zampini     ierr = PCBDDCGraphResetCSR(mat_graph);CHKERRQ(ierr);
9548687889aSStefano Zampini     PetscFunctionReturn(0);
9558687889aSStefano Zampini   }
95666da6bd7Sstefano_zampini   if (mat_graph->nvtxs == nvtxs && mat_graph->freecsr) { /* we own the data */
95756282151SStefano Zampini     if (mat_graph->xadj == xadj && mat_graph->adjncy == adjncy) same_data = PETSC_TRUE;
95856282151SStefano Zampini     if (!same_data && mat_graph->xadj[nvtxs] == xadj[nvtxs]) {
9592d505d7fSStefano Zampini       ierr = PetscMemcmp(xadj,mat_graph->xadj,(nvtxs+1)*sizeof(PetscInt),&same_data);CHKERRQ(ierr);
9602d505d7fSStefano Zampini       if (same_data) {
9612d505d7fSStefano Zampini         ierr = PetscMemcmp(adjncy,mat_graph->adjncy,xadj[nvtxs]*sizeof(PetscInt),&same_data);CHKERRQ(ierr);
9622d505d7fSStefano Zampini       }
96356282151SStefano Zampini     }
96456282151SStefano Zampini   }
96556282151SStefano Zampini   if (!same_data) {
966674ae819SStefano Zampini     /* free old CSR */
967674ae819SStefano Zampini     ierr = PCBDDCGraphResetCSR(mat_graph);CHKERRQ(ierr);
968674ae819SStefano Zampini     /* get CSR into graph structure */
969da1bb401SStefano Zampini     if (copymode == PETSC_COPY_VALUES) {
970854ce69bSBarry Smith       ierr = PetscMalloc1(nvtxs+1,&mat_graph->xadj);CHKERRQ(ierr);
971785e854fSJed Brown       ierr = PetscMalloc1(xadj[nvtxs],&mat_graph->adjncy);CHKERRQ(ierr);
972674ae819SStefano Zampini       ierr = PetscMemcpy(mat_graph->xadj,xadj,(nvtxs+1)*sizeof(PetscInt));CHKERRQ(ierr);
973674ae819SStefano Zampini       ierr = PetscMemcpy(mat_graph->adjncy,adjncy,xadj[nvtxs]*sizeof(PetscInt));CHKERRQ(ierr);
974a1dbd327SStefano Zampini       mat_graph->freecsr = PETSC_TRUE;
975da1bb401SStefano Zampini     } else if (copymode == PETSC_OWN_POINTER) {
9761a83f524SJed Brown       mat_graph->xadj    = (PetscInt*)xadj;
9771a83f524SJed Brown       mat_graph->adjncy  = (PetscInt*)adjncy;
978a1dbd327SStefano Zampini       mat_graph->freecsr = PETSC_TRUE;
979a1dbd327SStefano Zampini     } else if (copymode == PETSC_USE_POINTER) {
980a1dbd327SStefano Zampini       mat_graph->xadj    = (PetscInt*)xadj;
981a1dbd327SStefano Zampini       mat_graph->adjncy  = (PetscInt*)adjncy;
982a1dbd327SStefano Zampini       mat_graph->freecsr = PETSC_FALSE;
983e0fe2d75SToby Isaac     } else SETERRQ1(PETSC_COMM_SELF,PETSC_ERR_SUP,"Unsupported copy mode %D",copymode);
984575ad6abSStefano Zampini     mat_graph->nvtxs_csr = nvtxs;
98556282151SStefano Zampini     pcbddc->recompute_topography = PETSC_TRUE;
98656282151SStefano Zampini   }
98736e030ebSStefano Zampini   PetscFunctionReturn(0);
98836e030ebSStefano Zampini }
9891e6b0712SBarry Smith 
99036e030ebSStefano Zampini /*@
99154fffbccSStefano Zampini  PCBDDCSetLocalAdjacencyGraph - Set adjacency structure (CSR graph) of the local degrees of freedom.
99236e030ebSStefano Zampini 
99336e030ebSStefano Zampini    Not collective
99436e030ebSStefano Zampini 
99536e030ebSStefano Zampini    Input Parameters:
99654fffbccSStefano Zampini +  pc - the preconditioning context.
99754fffbccSStefano Zampini .  nvtxs - number of local vertices of the graph (i.e., the number of local dofs).
99854fffbccSStefano Zampini .  xadj, adjncy - the connectivity of the dofs in CSR format.
99954fffbccSStefano Zampini -  copymode - supported modes are PETSC_COPY_VALUES, PETSC_USE_POINTER or PETSC_OWN_POINTER.
100036e030ebSStefano Zampini 
100136e030ebSStefano Zampini    Level: intermediate
100236e030ebSStefano Zampini 
100354fffbccSStefano Zampini    Notes: A dof is considered connected with all local dofs if xadj[dof+1]-xadj[dof] == 1 and adjncy[xadj[dof]] is negative.
100436e030ebSStefano Zampini 
100528509bceSStefano Zampini .seealso: PCBDDC,PetscCopyMode
100636e030ebSStefano Zampini @*/
10071a83f524SJed Brown PetscErrorCode PCBDDCSetLocalAdjacencyGraph(PC pc,PetscInt nvtxs,const PetscInt xadj[],const PetscInt adjncy[], PetscCopyMode copymode)
100836e030ebSStefano Zampini {
1009575ad6abSStefano Zampini   void (*f)(void) = 0;
101036e030ebSStefano Zampini   PetscErrorCode ierr;
101136e030ebSStefano Zampini 
101236e030ebSStefano Zampini   PetscFunctionBegin;
101336e030ebSStefano Zampini   PetscValidHeaderSpecific(pc,PC_CLASSID,1);
10148687889aSStefano Zampini   if (nvtxs) {
1015674ae819SStefano Zampini     PetscValidIntPointer(xadj,3);
10161633d1f0SStefano Zampini     if (xadj[nvtxs]) PetscValidIntPointer(adjncy,4);
10178687889aSStefano Zampini   }
10181a83f524SJed Brown   ierr = PetscTryMethod(pc,"PCBDDCSetLocalAdjacencyGraph_C",(PC,PetscInt,const PetscInt[],const PetscInt[],PetscCopyMode),(pc,nvtxs,xadj,adjncy,copymode));CHKERRQ(ierr);
1019575ad6abSStefano Zampini   /* free arrays if PCBDDC is not the PC type */
1020575ad6abSStefano Zampini   ierr = PetscObjectQueryFunction((PetscObject)pc,"PCBDDCSetLocalAdjacencyGraph_C",&f);CHKERRQ(ierr);
1021575ad6abSStefano Zampini   if (!f && copymode == PETSC_OWN_POINTER) {
1022575ad6abSStefano Zampini     ierr = PetscFree(xadj);CHKERRQ(ierr);
1023575ad6abSStefano Zampini     ierr = PetscFree(adjncy);CHKERRQ(ierr);
1024da1bb401SStefano Zampini   }
102536e030ebSStefano Zampini   PetscFunctionReturn(0);
102636e030ebSStefano Zampini }
10271e6b0712SBarry Smith 
102863602bcaSStefano Zampini static PetscErrorCode PCBDDCSetDofsSplittingLocal_BDDC(PC pc,PetscInt n_is, IS ISForDofs[])
102963602bcaSStefano Zampini {
103063602bcaSStefano Zampini   PC_BDDC        *pcbddc = (PC_BDDC*)pc->data;
103163602bcaSStefano Zampini   PetscInt       i;
103256282151SStefano Zampini   PetscBool      isequal = PETSC_FALSE;
103363602bcaSStefano Zampini   PetscErrorCode ierr;
103463602bcaSStefano Zampini 
103563602bcaSStefano Zampini   PetscFunctionBegin;
103656282151SStefano Zampini   if (pcbddc->n_ISForDofsLocal == n_is) {
103756282151SStefano Zampini     for (i=0;i<n_is;i++) {
103856282151SStefano Zampini       PetscBool isequalt;
103956282151SStefano Zampini       ierr = ISEqual(ISForDofs[i],pcbddc->ISForDofsLocal[i],&isequalt);CHKERRQ(ierr);
104056282151SStefano Zampini       if (!isequalt) break;
104156282151SStefano Zampini     }
104256282151SStefano Zampini     if (i == n_is) isequal = PETSC_TRUE;
104356282151SStefano Zampini   }
104456282151SStefano Zampini   for (i=0;i<n_is;i++) {
104556282151SStefano Zampini     ierr = PetscObjectReference((PetscObject)ISForDofs[i]);CHKERRQ(ierr);
104656282151SStefano Zampini   }
104763602bcaSStefano Zampini   /* Destroy ISes if they were already set */
104863602bcaSStefano Zampini   for (i=0;i<pcbddc->n_ISForDofsLocal;i++) {
104963602bcaSStefano Zampini     ierr = ISDestroy(&pcbddc->ISForDofsLocal[i]);CHKERRQ(ierr);
105063602bcaSStefano Zampini   }
105163602bcaSStefano Zampini   ierr = PetscFree(pcbddc->ISForDofsLocal);CHKERRQ(ierr);
105263602bcaSStefano Zampini   /* last user setting takes precendence -> destroy any other customization */
105363602bcaSStefano Zampini   for (i=0;i<pcbddc->n_ISForDofs;i++) {
105463602bcaSStefano Zampini     ierr = ISDestroy(&pcbddc->ISForDofs[i]);CHKERRQ(ierr);
105563602bcaSStefano Zampini   }
105663602bcaSStefano Zampini   ierr = PetscFree(pcbddc->ISForDofs);CHKERRQ(ierr);
105763602bcaSStefano Zampini   pcbddc->n_ISForDofs = 0;
105863602bcaSStefano Zampini   /* allocate space then set */
1059d02579f5SStefano Zampini   if (n_is) {
1060d02579f5SStefano Zampini     ierr = PetscMalloc1(n_is,&pcbddc->ISForDofsLocal);CHKERRQ(ierr);
1061d02579f5SStefano Zampini   }
106263602bcaSStefano Zampini   for (i=0;i<n_is;i++) {
106363602bcaSStefano Zampini     pcbddc->ISForDofsLocal[i] = ISForDofs[i];
106463602bcaSStefano Zampini   }
106563602bcaSStefano Zampini   pcbddc->n_ISForDofsLocal = n_is;
106663602bcaSStefano Zampini   if (n_is) pcbddc->user_provided_isfordofs = PETSC_TRUE;
106756282151SStefano Zampini   if (!isequal) pcbddc->recompute_topography = PETSC_TRUE;
106863602bcaSStefano Zampini   PetscFunctionReturn(0);
106963602bcaSStefano Zampini }
107063602bcaSStefano Zampini 
107163602bcaSStefano Zampini /*@
107263602bcaSStefano Zampini  PCBDDCSetDofsSplittingLocal - Set index sets defining fields of the local subdomain matrix
107363602bcaSStefano Zampini 
107463602bcaSStefano Zampini    Collective
107563602bcaSStefano Zampini 
107663602bcaSStefano Zampini    Input Parameters:
107763602bcaSStefano Zampini +  pc - the preconditioning context
10780f202f7eSStefano Zampini .  n_is - number of index sets defining the fields
10790f202f7eSStefano Zampini -  ISForDofs - array of IS describing the fields in local ordering
108063602bcaSStefano Zampini 
108163602bcaSStefano Zampini    Level: intermediate
108263602bcaSStefano Zampini 
10830f202f7eSStefano Zampini    Notes:
10840f202f7eSStefano Zampini      n_is should be the same among processes. Not all nodes need to be listed: unlisted nodes will belong to the complement field.
108563602bcaSStefano Zampini 
108663602bcaSStefano Zampini .seealso: PCBDDC
108763602bcaSStefano Zampini @*/
108863602bcaSStefano Zampini PetscErrorCode PCBDDCSetDofsSplittingLocal(PC pc,PetscInt n_is, IS ISForDofs[])
108963602bcaSStefano Zampini {
109063602bcaSStefano Zampini   PetscInt       i;
109163602bcaSStefano Zampini   PetscErrorCode ierr;
109263602bcaSStefano Zampini 
109363602bcaSStefano Zampini   PetscFunctionBegin;
109463602bcaSStefano Zampini   PetscValidHeaderSpecific(pc,PC_CLASSID,1);
109563602bcaSStefano Zampini   PetscValidLogicalCollectiveInt(pc,n_is,2);
109663602bcaSStefano Zampini   for (i=0;i<n_is;i++) {
109763602bcaSStefano Zampini     PetscCheckSameComm(pc,1,ISForDofs[i],3);
109863602bcaSStefano Zampini     PetscValidHeaderSpecific(ISForDofs[i],IS_CLASSID,3);
109963602bcaSStefano Zampini   }
1100e71e7a71SStefano Zampini   ierr = PetscTryMethod(pc,"PCBDDCSetDofsSplittingLocal_C",(PC,PetscInt,IS[]),(pc,n_is,ISForDofs));CHKERRQ(ierr);
110163602bcaSStefano Zampini   PetscFunctionReturn(0);
110263602bcaSStefano Zampini }
110363602bcaSStefano Zampini 
11049c0446d6SStefano Zampini static PetscErrorCode PCBDDCSetDofsSplitting_BDDC(PC pc,PetscInt n_is, IS ISForDofs[])
11059c0446d6SStefano Zampini {
11069c0446d6SStefano Zampini   PC_BDDC        *pcbddc = (PC_BDDC*)pc->data;
11079c0446d6SStefano Zampini   PetscInt       i;
110856282151SStefano Zampini   PetscBool      isequal = PETSC_FALSE;
11099c0446d6SStefano Zampini   PetscErrorCode ierr;
11109c0446d6SStefano Zampini 
11119c0446d6SStefano Zampini   PetscFunctionBegin;
111256282151SStefano Zampini   if (pcbddc->n_ISForDofs == n_is) {
111356282151SStefano Zampini     for (i=0;i<n_is;i++) {
111456282151SStefano Zampini       PetscBool isequalt;
111556282151SStefano Zampini       ierr = ISEqual(ISForDofs[i],pcbddc->ISForDofs[i],&isequalt);CHKERRQ(ierr);
111656282151SStefano Zampini       if (!isequalt) break;
111756282151SStefano Zampini     }
111856282151SStefano Zampini     if (i == n_is) isequal = PETSC_TRUE;
111956282151SStefano Zampini   }
112056282151SStefano Zampini   for (i=0;i<n_is;i++) {
112156282151SStefano Zampini     ierr = PetscObjectReference((PetscObject)ISForDofs[i]);CHKERRQ(ierr);
112256282151SStefano Zampini   }
1123da1bb401SStefano Zampini   /* Destroy ISes if they were already set */
11249c0446d6SStefano Zampini   for (i=0;i<pcbddc->n_ISForDofs;i++) {
11259c0446d6SStefano Zampini     ierr = ISDestroy(&pcbddc->ISForDofs[i]);CHKERRQ(ierr);
11269c0446d6SStefano Zampini   }
1127d11ae9bbSstefano_zampini   ierr = PetscFree(pcbddc->ISForDofs);CHKERRQ(ierr);
112863602bcaSStefano Zampini   /* last user setting takes precendence -> destroy any other customization */
112963602bcaSStefano Zampini   for (i=0;i<pcbddc->n_ISForDofsLocal;i++) {
113063602bcaSStefano Zampini     ierr = ISDestroy(&pcbddc->ISForDofsLocal[i]);CHKERRQ(ierr);
113163602bcaSStefano Zampini   }
113263602bcaSStefano Zampini   ierr = PetscFree(pcbddc->ISForDofsLocal);CHKERRQ(ierr);
113363602bcaSStefano Zampini   pcbddc->n_ISForDofsLocal = 0;
1134da1bb401SStefano Zampini   /* allocate space then set */
1135d02579f5SStefano Zampini   if (n_is) {
1136785e854fSJed Brown     ierr = PetscMalloc1(n_is,&pcbddc->ISForDofs);CHKERRQ(ierr);
1137d02579f5SStefano Zampini   }
11389c0446d6SStefano Zampini   for (i=0;i<n_is;i++) {
1139da1bb401SStefano Zampini     pcbddc->ISForDofs[i] = ISForDofs[i];
11409c0446d6SStefano Zampini   }
11419c0446d6SStefano Zampini   pcbddc->n_ISForDofs = n_is;
114263602bcaSStefano Zampini   if (n_is) pcbddc->user_provided_isfordofs = PETSC_TRUE;
114356282151SStefano Zampini   if (!isequal) pcbddc->recompute_topography = PETSC_TRUE;
11449c0446d6SStefano Zampini   PetscFunctionReturn(0);
11459c0446d6SStefano Zampini }
11461e6b0712SBarry Smith 
11479c0446d6SStefano Zampini /*@
114863602bcaSStefano Zampini  PCBDDCSetDofsSplitting - Set index sets defining fields of the global matrix
11499c0446d6SStefano Zampini 
115063602bcaSStefano Zampini    Collective
11519c0446d6SStefano Zampini 
11529c0446d6SStefano Zampini    Input Parameters:
11539c0446d6SStefano Zampini +  pc - the preconditioning context
11540f202f7eSStefano Zampini .  n_is - number of index sets defining the fields
11550f202f7eSStefano Zampini -  ISForDofs - array of IS describing the fields in global ordering
11569c0446d6SStefano Zampini 
11579c0446d6SStefano Zampini    Level: intermediate
11589c0446d6SStefano Zampini 
11590f202f7eSStefano Zampini    Notes:
11600f202f7eSStefano Zampini      Any process can list any global node. Not all nodes need to be listed: unlisted nodes will belong to the complement field.
11619c0446d6SStefano Zampini 
11629c0446d6SStefano Zampini .seealso: PCBDDC
11639c0446d6SStefano Zampini @*/
11649c0446d6SStefano Zampini PetscErrorCode PCBDDCSetDofsSplitting(PC pc,PetscInt n_is, IS ISForDofs[])
11659c0446d6SStefano Zampini {
11662b510759SStefano Zampini   PetscInt       i;
11679c0446d6SStefano Zampini   PetscErrorCode ierr;
11689c0446d6SStefano Zampini 
11699c0446d6SStefano Zampini   PetscFunctionBegin;
11709c0446d6SStefano Zampini   PetscValidHeaderSpecific(pc,PC_CLASSID,1);
117163602bcaSStefano Zampini   PetscValidLogicalCollectiveInt(pc,n_is,2);
11722b510759SStefano Zampini   for (i=0;i<n_is;i++) {
117363602bcaSStefano Zampini     PetscValidHeaderSpecific(ISForDofs[i],IS_CLASSID,3);
1174a011d5a7Sstefano_zampini     PetscCheckSameComm(pc,1,ISForDofs[i],3);
11752b510759SStefano Zampini   }
11769c0446d6SStefano Zampini   ierr = PetscTryMethod(pc,"PCBDDCSetDofsSplitting_C",(PC,PetscInt,IS[]),(pc,n_is,ISForDofs));CHKERRQ(ierr);
11779c0446d6SStefano Zampini   PetscFunctionReturn(0);
11789c0446d6SStefano Zampini }
1179906d46d4SStefano Zampini 
1180534831adSStefano Zampini /*
1181534831adSStefano Zampini    PCPreSolve_BDDC - Changes the right hand side and (if necessary) the initial
1182534831adSStefano Zampini                      guess if a transformation of basis approach has been selected.
11839c0446d6SStefano Zampini 
1184534831adSStefano Zampini    Input Parameter:
1185534831adSStefano Zampini +  pc - the preconditioner contex
1186534831adSStefano Zampini 
1187534831adSStefano Zampini    Application Interface Routine: PCPreSolve()
1188534831adSStefano Zampini 
1189534831adSStefano Zampini    Notes:
1190534831adSStefano Zampini      The interface routine PCPreSolve() is not usually called directly by
1191534831adSStefano Zampini    the user, but instead is called by KSPSolve().
1192534831adSStefano Zampini */
1193534831adSStefano Zampini static PetscErrorCode PCPreSolve_BDDC(PC pc, KSP ksp, Vec rhs, Vec x)
1194534831adSStefano Zampini {
1195534831adSStefano Zampini   PetscErrorCode ierr;
1196534831adSStefano Zampini   PC_BDDC        *pcbddc = (PC_BDDC*)pc->data;
1197534831adSStefano Zampini   PC_IS          *pcis = (PC_IS*)(pc->data);
11983972b0daSStefano Zampini   Vec            used_vec;
119927b6a85dSStefano Zampini   PetscBool      save_rhs = PETSC_TRUE, benign_correction_computed;
1200534831adSStefano Zampini 
1201534831adSStefano Zampini   PetscFunctionBegin;
12021f4df5f7SStefano Zampini   /* if we are working with CG, one dirichlet solve can be avoided during Krylov iterations */
120385c4d303SStefano Zampini   if (ksp) {
12041f4df5f7SStefano Zampini     PetscBool iscg, isgroppcg, ispipecg, ispipecgrr;
120585c4d303SStefano Zampini     ierr = PetscObjectTypeCompare((PetscObject)ksp,KSPCG,&iscg);CHKERRQ(ierr);
120627b6a85dSStefano Zampini     ierr = PetscObjectTypeCompare((PetscObject)ksp,KSPGROPPCG,&isgroppcg);CHKERRQ(ierr);
120727b6a85dSStefano Zampini     ierr = PetscObjectTypeCompare((PetscObject)ksp,KSPPIPECG,&ispipecg);CHKERRQ(ierr);
1208f94e96cbSStefano Zampini     ierr = PetscObjectTypeCompare((PetscObject)ksp,KSPPIPECGRR,&ispipecgrr);CHKERRQ(ierr);
12091f4df5f7SStefano Zampini     if (pcbddc->benign_apply_coarse_only || pcbddc->switch_static || (!iscg && !isgroppcg && !ispipecg && !ispipecgrr)) {
121085c4d303SStefano Zampini       ierr = PCBDDCSetUseExactDirichlet(pc,PETSC_FALSE);CHKERRQ(ierr);
121185c4d303SStefano Zampini     }
121285c4d303SStefano Zampini   }
1213fc17d649SStefano Zampini   if (pcbddc->benign_apply_coarse_only || pcbddc->switch_static) {
1214fc17d649SStefano Zampini     ierr = PCBDDCSetUseExactDirichlet(pc,PETSC_FALSE);CHKERRQ(ierr);
1215fc17d649SStefano Zampini   }
12161f4df5f7SStefano Zampini 
121785c4d303SStefano Zampini   /* Creates parallel work vectors used in presolve */
121862a6ff1dSStefano Zampini   if (!pcbddc->original_rhs) {
121962a6ff1dSStefano Zampini     ierr = VecDuplicate(pcis->vec1_global,&pcbddc->original_rhs);CHKERRQ(ierr);
122062a6ff1dSStefano Zampini   }
122162a6ff1dSStefano Zampini   if (!pcbddc->temp_solution) {
122262a6ff1dSStefano Zampini     ierr = VecDuplicate(pcis->vec1_global,&pcbddc->temp_solution);CHKERRQ(ierr);
122362a6ff1dSStefano Zampini   }
12248d00608fSStefano Zampini 
122527b6a85dSStefano Zampini   pcbddc->temp_solution_used = PETSC_FALSE;
12263972b0daSStefano Zampini   if (x) {
12273972b0daSStefano Zampini     ierr = PetscObjectReference((PetscObject)x);CHKERRQ(ierr);
12283972b0daSStefano Zampini     used_vec = x;
12298d00608fSStefano Zampini   } else { /* it can only happen when calling PCBDDCMatFETIDPGetRHS */
12303972b0daSStefano Zampini     ierr = PetscObjectReference((PetscObject)pcbddc->temp_solution);CHKERRQ(ierr);
12313972b0daSStefano Zampini     used_vec = pcbddc->temp_solution;
12323972b0daSStefano Zampini     ierr = VecSet(used_vec,0.0);CHKERRQ(ierr);
123327b6a85dSStefano Zampini     pcbddc->temp_solution_used = PETSC_TRUE;
1234266e20e9SStefano Zampini     ierr = VecCopy(rhs,pcbddc->original_rhs);CHKERRQ(ierr);
1235266e20e9SStefano Zampini     save_rhs = PETSC_FALSE;
1236266e20e9SStefano Zampini     pcbddc->eliminate_dirdofs = PETSC_TRUE;
12373972b0daSStefano Zampini   }
12388efcfb23SStefano Zampini 
12398efcfb23SStefano Zampini   /* hack into ksp data structure since PCPreSolve comes earlier than setting to zero the guess in src/ksp/ksp/interface/itfunc.c */
12403972b0daSStefano Zampini   if (ksp) {
1241a0cb1b98SStefano Zampini     /* store the flag for the initial guess since it will be restored back during PCPostSolve_BDDC */
12428efcfb23SStefano Zampini     ierr = KSPGetInitialGuessNonzero(ksp,&pcbddc->ksp_guess_nonzero);CHKERRQ(ierr);
12438efcfb23SStefano Zampini     if (!pcbddc->ksp_guess_nonzero) {
12443972b0daSStefano Zampini       ierr = VecSet(used_vec,0.0);CHKERRQ(ierr);
12453972b0daSStefano Zampini     }
12463972b0daSStefano Zampini   }
12473308cffdSStefano Zampini 
12488d00608fSStefano Zampini   pcbddc->rhs_change = PETSC_FALSE;
12493972b0daSStefano Zampini   /* Take into account zeroed rows -> change rhs and store solution removed */
125070c64980SStefano Zampini   if (rhs && pcbddc->eliminate_dirdofs) {
12513975b054SStefano Zampini     IS dirIS = NULL;
12523975b054SStefano Zampini 
1253a07ea27aSStefano Zampini     /* DirichletBoundariesLocal may not be consistent among neighbours; gets a dirichlet dofs IS from graph (may be cached) */
12543975b054SStefano Zampini     ierr = PCBDDCGraphGetDirichletDofs(pcbddc->mat_graph,&dirIS);CHKERRQ(ierr);
12553975b054SStefano Zampini     if (dirIS) {
1256906d46d4SStefano Zampini       Mat_IS            *matis = (Mat_IS*)pc->pmat->data;
1257785d1243SStefano Zampini       PetscInt          dirsize,i,*is_indices;
12582b095fd8SStefano Zampini       PetscScalar       *array_x;
12592b095fd8SStefano Zampini       const PetscScalar *array_diagonal;
1260785d1243SStefano Zampini 
12613972b0daSStefano Zampini       ierr = MatGetDiagonal(pc->pmat,pcis->vec1_global);CHKERRQ(ierr);
12623972b0daSStefano Zampini       ierr = VecPointwiseDivide(pcis->vec1_global,rhs,pcis->vec1_global);CHKERRQ(ierr);
1263e176bc59SStefano Zampini       ierr = VecScatterBegin(matis->rctx,pcis->vec1_global,pcis->vec2_N,INSERT_VALUES,SCATTER_FORWARD);CHKERRQ(ierr);
1264e176bc59SStefano Zampini       ierr = VecScatterEnd(matis->rctx,pcis->vec1_global,pcis->vec2_N,INSERT_VALUES,SCATTER_FORWARD);CHKERRQ(ierr);
1265e176bc59SStefano Zampini       ierr = VecScatterBegin(matis->rctx,used_vec,pcis->vec1_N,INSERT_VALUES,SCATTER_FORWARD);CHKERRQ(ierr);
1266e176bc59SStefano Zampini       ierr = VecScatterEnd(matis->rctx,used_vec,pcis->vec1_N,INSERT_VALUES,SCATTER_FORWARD);CHKERRQ(ierr);
126782ba6b80SStefano Zampini       ierr = ISGetLocalSize(dirIS,&dirsize);CHKERRQ(ierr);
12683972b0daSStefano Zampini       ierr = VecGetArray(pcis->vec1_N,&array_x);CHKERRQ(ierr);
12692b095fd8SStefano Zampini       ierr = VecGetArrayRead(pcis->vec2_N,&array_diagonal);CHKERRQ(ierr);
12703972b0daSStefano Zampini       ierr = ISGetIndices(dirIS,(const PetscInt**)&is_indices);CHKERRQ(ierr);
12712fa5cd67SKarl Rupp       for (i=0; i<dirsize; i++) array_x[is_indices[i]] = array_diagonal[is_indices[i]];
12723972b0daSStefano Zampini       ierr = ISRestoreIndices(dirIS,(const PetscInt**)&is_indices);CHKERRQ(ierr);
12732b095fd8SStefano Zampini       ierr = VecRestoreArrayRead(pcis->vec2_N,&array_diagonal);CHKERRQ(ierr);
12743972b0daSStefano Zampini       ierr = VecRestoreArray(pcis->vec1_N,&array_x);CHKERRQ(ierr);
1275e176bc59SStefano Zampini       ierr = VecScatterBegin(matis->rctx,pcis->vec1_N,used_vec,INSERT_VALUES,SCATTER_REVERSE);CHKERRQ(ierr);
1276e176bc59SStefano Zampini       ierr = VecScatterEnd(matis->rctx,pcis->vec1_N,used_vec,INSERT_VALUES,SCATTER_REVERSE);CHKERRQ(ierr);
12778d00608fSStefano Zampini       pcbddc->rhs_change = PETSC_TRUE;
12781b968477SStefano Zampini       ierr = ISDestroy(&dirIS);CHKERRQ(ierr);
12798efcfb23SStefano Zampini     }
1280a07ea27aSStefano Zampini   }
1281b76ba322SStefano Zampini 
12828efcfb23SStefano Zampini   /* remove the computed solution or the initial guess from the rhs */
12838d00608fSStefano Zampini   if (pcbddc->rhs_change || (ksp && pcbddc->ksp_guess_nonzero) ) {
128427b6a85dSStefano Zampini     /* save the original rhs */
128527b6a85dSStefano Zampini     if (save_rhs) {
128627b6a85dSStefano Zampini       ierr = VecSwap(rhs,pcbddc->original_rhs);CHKERRQ(ierr);
128727b6a85dSStefano Zampini       save_rhs = PETSC_FALSE;
12888d00608fSStefano Zampini     }
12898d00608fSStefano Zampini     pcbddc->rhs_change = PETSC_TRUE;
12903972b0daSStefano Zampini     ierr = VecScale(used_vec,-1.0);CHKERRQ(ierr);
129127b6a85dSStefano Zampini     ierr = MatMultAdd(pc->mat,used_vec,pcbddc->original_rhs,rhs);CHKERRQ(ierr);
12923972b0daSStefano Zampini     ierr = VecScale(used_vec,-1.0);CHKERRQ(ierr);
12938efcfb23SStefano Zampini     ierr = VecCopy(used_vec,pcbddc->temp_solution);CHKERRQ(ierr);
129427b6a85dSStefano Zampini     pcbddc->temp_solution_used = PETSC_TRUE;
12957acc28cbSStefano Zampini     if (ksp) {
12967acc28cbSStefano Zampini       ierr = KSPSetInitialGuessNonzero(ksp,PETSC_FALSE);CHKERRQ(ierr);
12977acc28cbSStefano Zampini     }
12983308cffdSStefano Zampini   }
12998efcfb23SStefano Zampini   ierr = VecDestroy(&used_vec);CHKERRQ(ierr);
1300b76ba322SStefano Zampini 
1301fc17d649SStefano Zampini   /* compute initial vector in benign space if needed
130227b6a85dSStefano Zampini      and remove non-benign solution from the rhs */
130327b6a85dSStefano Zampini   benign_correction_computed = PETSC_FALSE;
130427b6a85dSStefano Zampini   if (rhs && pcbddc->benign_compute_correction && pcbddc->benign_have_null) {
13051f4df5f7SStefano Zampini     /* compute u^*_h using ideas similar to those in Xuemin Tu's PhD thesis (see Section 4.8.1)
13061f4df5f7SStefano Zampini        Recursively apply BDDC in the multilevel case */
13070369aaf7SStefano Zampini     if (!pcbddc->benign_vec) {
13080369aaf7SStefano Zampini       ierr = VecDuplicate(rhs,&pcbddc->benign_vec);CHKERRQ(ierr);
13090369aaf7SStefano Zampini     }
13104fee134fSStefano Zampini     pcbddc->benign_apply_coarse_only = PETSC_TRUE;
131127b6a85dSStefano Zampini     if (!pcbddc->benign_skip_correction) {
13121dd7afcfSStefano Zampini       ierr = PCApply_BDDC(pc,rhs,pcbddc->benign_vec);CHKERRQ(ierr);
13133bca92a6SStefano Zampini       benign_correction_computed = PETSC_TRUE;
13141f4df5f7SStefano Zampini       if (pcbddc->temp_solution_used) {
13151f4df5f7SStefano Zampini         ierr = VecAXPY(pcbddc->temp_solution,1.0,pcbddc->benign_vec);CHKERRQ(ierr);
13161f4df5f7SStefano Zampini       }
13171f4df5f7SStefano Zampini       ierr = VecScale(pcbddc->benign_vec,-1.0);CHKERRQ(ierr);
131827b6a85dSStefano Zampini       /* store the original rhs if not done earlier */
131927b6a85dSStefano Zampini       if (save_rhs) {
132027b6a85dSStefano Zampini         ierr = VecSwap(rhs,pcbddc->original_rhs);CHKERRQ(ierr);
132192e3dcfbSStefano Zampini       }
132227b6a85dSStefano Zampini       if (pcbddc->rhs_change) {
13230369aaf7SStefano Zampini         ierr = MatMultAdd(pc->mat,pcbddc->benign_vec,rhs,rhs);CHKERRQ(ierr);
132427b6a85dSStefano Zampini       } else {
132527b6a85dSStefano Zampini         ierr = MatMultAdd(pc->mat,pcbddc->benign_vec,pcbddc->original_rhs,rhs);CHKERRQ(ierr);
132627b6a85dSStefano Zampini       }
13270369aaf7SStefano Zampini       pcbddc->rhs_change = PETSC_TRUE;
132827b6a85dSStefano Zampini     }
132927b6a85dSStefano Zampini     pcbddc->benign_apply_coarse_only = PETSC_FALSE;
13300369aaf7SStefano Zampini   }
13312d4c4fecSStefano Zampini 
13322d4c4fecSStefano Zampini   /* dbg output */
1333a198735bSStefano Zampini   if (pcbddc->dbg_flag && benign_correction_computed) {
13341f4df5f7SStefano Zampini     Vec v;
13351f4df5f7SStefano Zampini     ierr = VecDuplicate(pcis->vec1_global,&v);CHKERRQ(ierr);
13361f4df5f7SStefano Zampini     ierr = MatMultTranspose(pcbddc->ChangeOfBasisMatrix,rhs,v);CHKERRQ(ierr);
13371f4df5f7SStefano Zampini     ierr = PCBDDCBenignGetOrSetP0(pc,v,PETSC_TRUE);CHKERRQ(ierr);
1338e0fe2d75SToby Isaac     ierr = PetscViewerASCIIPrintf(pcbddc->dbg_viewer,"LEVEL %D: is the correction benign?\n",pcbddc->current_level);CHKERRQ(ierr);
13392d4c4fecSStefano Zampini     ierr = PetscScalarView(pcbddc->benign_n,pcbddc->benign_p0,PETSC_VIEWER_STDOUT_(PetscObjectComm((PetscObject)pc)));CHKERRQ(ierr);
13401f4df5f7SStefano Zampini     ierr = VecDestroy(&v);CHKERRQ(ierr);
13411f4df5f7SStefano Zampini   }
13420369aaf7SStefano Zampini 
13430369aaf7SStefano Zampini   /* set initial guess if using PCG */
13448ae0ca82SStefano Zampini   pcbddc->exact_dirichlet_trick_app = PETSC_FALSE;
13450369aaf7SStefano Zampini   if (x && pcbddc->use_exact_dirichlet_trick) {
13460369aaf7SStefano Zampini     ierr = VecSet(x,0.0);CHKERRQ(ierr);
13471dd7afcfSStefano Zampini     if (pcbddc->ChangeOfBasisMatrix && pcbddc->change_interior) {
134827b6a85dSStefano Zampini       if (benign_correction_computed) { /* we have already saved the changed rhs */
13491dd7afcfSStefano Zampini         ierr = VecLockPop(pcis->vec1_global);CHKERRQ(ierr);
13501dd7afcfSStefano Zampini       } else {
13511dd7afcfSStefano Zampini         ierr = MatMultTranspose(pcbddc->ChangeOfBasisMatrix,rhs,pcis->vec1_global);CHKERRQ(ierr);
13521dd7afcfSStefano Zampini       }
13531dd7afcfSStefano Zampini       ierr = VecScatterBegin(pcis->global_to_D,pcis->vec1_global,pcis->vec1_D,INSERT_VALUES,SCATTER_FORWARD);CHKERRQ(ierr);
13541dd7afcfSStefano Zampini       ierr = VecScatterEnd(pcis->global_to_D,pcis->vec1_global,pcis->vec1_D,INSERT_VALUES,SCATTER_FORWARD);CHKERRQ(ierr);
13551dd7afcfSStefano Zampini     } else {
13560369aaf7SStefano Zampini       ierr = VecScatterBegin(pcis->global_to_D,rhs,pcis->vec1_D,INSERT_VALUES,SCATTER_FORWARD);CHKERRQ(ierr);
13570369aaf7SStefano Zampini       ierr = VecScatterEnd(pcis->global_to_D,rhs,pcis->vec1_D,INSERT_VALUES,SCATTER_FORWARD);CHKERRQ(ierr);
13581dd7afcfSStefano Zampini     }
13590369aaf7SStefano Zampini     ierr = KSPSolve(pcbddc->ksp_D,pcis->vec1_D,pcis->vec2_D);CHKERRQ(ierr);
13601dd7afcfSStefano Zampini     if (pcbddc->ChangeOfBasisMatrix && pcbddc->change_interior) {
13611dd7afcfSStefano Zampini       ierr = VecSet(pcis->vec1_global,0.);CHKERRQ(ierr);
13621dd7afcfSStefano Zampini       ierr = VecScatterBegin(pcis->global_to_D,pcis->vec2_D,pcis->vec1_global,INSERT_VALUES,SCATTER_REVERSE);CHKERRQ(ierr);
13631dd7afcfSStefano Zampini       ierr = VecScatterEnd(pcis->global_to_D,pcis->vec2_D,pcis->vec1_global,INSERT_VALUES,SCATTER_REVERSE);CHKERRQ(ierr);
13641dd7afcfSStefano Zampini       ierr = MatMult(pcbddc->ChangeOfBasisMatrix,pcis->vec1_global,x);CHKERRQ(ierr);
13651dd7afcfSStefano Zampini     } else {
13660369aaf7SStefano Zampini       ierr = VecScatterBegin(pcis->global_to_D,pcis->vec2_D,x,INSERT_VALUES,SCATTER_REVERSE);CHKERRQ(ierr);
13670369aaf7SStefano Zampini       ierr = VecScatterEnd(pcis->global_to_D,pcis->vec2_D,x,INSERT_VALUES,SCATTER_REVERSE);CHKERRQ(ierr);
13681dd7afcfSStefano Zampini     }
13690369aaf7SStefano Zampini     if (ksp) {
13700369aaf7SStefano Zampini       ierr = KSPSetInitialGuessNonzero(ksp,PETSC_TRUE);CHKERRQ(ierr);
13710369aaf7SStefano Zampini     }
13728ae0ca82SStefano Zampini     pcbddc->exact_dirichlet_trick_app = PETSC_TRUE;
1373266e20e9SStefano Zampini   } else if (pcbddc->ChangeOfBasisMatrix && pcbddc->change_interior && benign_correction_computed && pcbddc->use_exact_dirichlet_trick) {
1374266e20e9SStefano Zampini     ierr = VecLockPop(pcis->vec1_global);CHKERRQ(ierr);
13750369aaf7SStefano Zampini   }
1376534831adSStefano Zampini   PetscFunctionReturn(0);
1377534831adSStefano Zampini }
1378906d46d4SStefano Zampini 
1379534831adSStefano Zampini /*
1380534831adSStefano Zampini    PCPostSolve_BDDC - Changes the computed solution if a transformation of basis
1381534831adSStefano Zampini                      approach has been selected. Also, restores rhs to its original state.
1382534831adSStefano Zampini 
1383534831adSStefano Zampini    Input Parameter:
1384534831adSStefano Zampini +  pc - the preconditioner contex
1385534831adSStefano Zampini 
1386534831adSStefano Zampini    Application Interface Routine: PCPostSolve()
1387534831adSStefano Zampini 
1388534831adSStefano Zampini    Notes:
1389534831adSStefano Zampini      The interface routine PCPostSolve() is not usually called directly by
1390534831adSStefano Zampini      the user, but instead is called by KSPSolve().
1391534831adSStefano Zampini */
1392534831adSStefano Zampini static PetscErrorCode PCPostSolve_BDDC(PC pc, KSP ksp, Vec rhs, Vec x)
1393534831adSStefano Zampini {
1394534831adSStefano Zampini   PetscErrorCode ierr;
1395534831adSStefano Zampini   PC_BDDC        *pcbddc = (PC_BDDC*)pc->data;
1396534831adSStefano Zampini 
1397534831adSStefano Zampini   PetscFunctionBegin;
13983972b0daSStefano Zampini   /* add solution removed in presolve */
13996bcfc461SStefano Zampini   if (x && pcbddc->rhs_change) {
140027b6a85dSStefano Zampini     if (pcbddc->temp_solution_used) {
14013425bc38SStefano Zampini       ierr = VecAXPY(x,1.0,pcbddc->temp_solution);CHKERRQ(ierr);
1402af140850Sstefano_zampini     } else if (pcbddc->benign_compute_correction && pcbddc->benign_vec) {
140327b6a85dSStefano Zampini       ierr = VecAXPY(x,-1.0,pcbddc->benign_vec);CHKERRQ(ierr);
14043425bc38SStefano Zampini     }
1405af140850Sstefano_zampini     /* restore to original state (not for FETI-DP) */
1406af140850Sstefano_zampini     if (ksp) pcbddc->temp_solution_used = PETSC_FALSE;
140727b6a85dSStefano Zampini   }
140827b6a85dSStefano Zampini 
1409266e20e9SStefano Zampini   /* restore rhs to its original state (not needed for FETI-DP) */
14108d00608fSStefano Zampini   if (rhs && pcbddc->rhs_change) {
141127b6a85dSStefano Zampini     ierr = VecSwap(rhs,pcbddc->original_rhs);CHKERRQ(ierr);
14128d00608fSStefano Zampini     pcbddc->rhs_change = PETSC_FALSE;
1413af140850Sstefano_zampini   }
14148efcfb23SStefano Zampini   /* restore ksp guess state */
14158efcfb23SStefano Zampini   if (ksp) {
14168efcfb23SStefano Zampini     ierr = KSPSetInitialGuessNonzero(ksp,pcbddc->ksp_guess_nonzero);CHKERRQ(ierr);
14178ae0ca82SStefano Zampini     /* reset flag for exact dirichlet trick */
14188ae0ca82SStefano Zampini     pcbddc->exact_dirichlet_trick_app = PETSC_FALSE;
1419af140850Sstefano_zampini   }
1420534831adSStefano Zampini   PetscFunctionReturn(0);
1421534831adSStefano Zampini }
1422af140850Sstefano_zampini 
14230c7d97c5SJed Brown /*
14240c7d97c5SJed Brown    PCSetUp_BDDC - Prepares for the use of the BDDC preconditioner
14250c7d97c5SJed Brown                   by setting data structures and options.
14260c7d97c5SJed Brown 
14270c7d97c5SJed Brown    Input Parameter:
142853cdbc3dSStefano Zampini +  pc - the preconditioner context
14290c7d97c5SJed Brown 
14300c7d97c5SJed Brown    Application Interface Routine: PCSetUp()
14310c7d97c5SJed Brown 
14320c7d97c5SJed Brown    Notes:
14330c7d97c5SJed Brown      The interface routine PCSetUp() is not usually called directly by
14340c7d97c5SJed Brown      the user, but instead is called by PCApply() if necessary.
14350c7d97c5SJed Brown */
143653cdbc3dSStefano Zampini PetscErrorCode PCSetUp_BDDC(PC pc)
14370c7d97c5SJed Brown {
14380c7d97c5SJed Brown   PC_BDDC*        pcbddc = (PC_BDDC*)pc->data;
1439c703fcc7SStefano Zampini   PCBDDCSubSchurs sub_schurs;
14405e8657edSStefano Zampini   Mat_IS*         matis;
144108122e43SStefano Zampini   MatNullSpace    nearnullspace;
144235509ce9Sstefano_zampini   Mat             lA;
144335509ce9Sstefano_zampini   IS              lP,zerodiag = NULL;
144491e8d312SStefano Zampini   PetscInt        nrows,ncols;
1445c703fcc7SStefano Zampini   PetscBool       computesubschurs;
14468de1fae6SStefano Zampini   PetscBool       computeconstraintsmatrix;
14475e8657edSStefano Zampini   PetscBool       new_nearnullspace_provided,ismatis;
1448c703fcc7SStefano Zampini   PetscErrorCode  ierr;
14490c7d97c5SJed Brown 
14500c7d97c5SJed Brown   PetscFunctionBegin;
14515e8657edSStefano Zampini   ierr = PetscObjectTypeCompare((PetscObject)pc->pmat,MATIS,&ismatis);CHKERRQ(ierr);
14526c4ed002SBarry Smith   if (!ismatis) SETERRQ(PetscObjectComm((PetscObject)pc),PETSC_ERR_ARG_WRONG,"PCBDDC preconditioner requires matrix of type MATIS");
145391e8d312SStefano Zampini   ierr = MatGetSize(pc->pmat,&nrows,&ncols);CHKERRQ(ierr);
14546c4ed002SBarry Smith   if (nrows != ncols) SETERRQ(PetscObjectComm((PetscObject)pc),PETSC_ERR_SUP,"PCBDDC preconditioner requires a square preconditioning matrix");
14555e8657edSStefano Zampini   matis = (Mat_IS*)pc->pmat->data;
1456f4ddd8eeSStefano Zampini   /* the following lines of code should be replaced by a better logic between PCIS, PCNN, PCBDDC and other future nonoverlapping preconditioners */
14573b03a366Sstefano_zampini   /* For BDDC we need to define a local "Neumann" problem different to that defined in PCISSetup
145871582508SStefano Zampini      Also, BDDC builds its own KSP for the Dirichlet problem */
1459a485753aSstefano_zampini   if (!pc->setupcalled || pc->flag == DIFFERENT_NONZERO_PATTERN) pcbddc->recompute_topography = PETSC_TRUE;
1460c83e1ba7SStefano Zampini   if (pcbddc->recompute_topography) {
1461c83e1ba7SStefano Zampini     pcbddc->graphanalyzed    = PETSC_FALSE;
1462c83e1ba7SStefano Zampini     computeconstraintsmatrix = PETSC_TRUE;
1463c83e1ba7SStefano Zampini   } else {
14648de1fae6SStefano Zampini     computeconstraintsmatrix = PETSC_FALSE;
1465c83e1ba7SStefano Zampini   }
1466b087196eSStefano Zampini 
1467b087196eSStefano Zampini   /* check parameters' compatibility */
1468b7ab4a40SStefano Zampini   if (!pcbddc->use_deluxe_scaling) pcbddc->deluxe_zerorows = PETSC_FALSE;
14699d54b7f4SStefano Zampini   pcbddc->adaptive_selection = (PetscBool)(pcbddc->adaptive_threshold > 0.0);
1470bf3a8328SStefano Zampini   pcbddc->adaptive_userdefined = (PetscBool)(pcbddc->adaptive_selection && pcbddc->adaptive_userdefined);
1471862806e4SStefano Zampini   if (pcbddc->adaptive_selection) pcbddc->use_faces = PETSC_TRUE;
1472862806e4SStefano Zampini 
14735a95e1ceSStefano Zampini   computesubschurs = (PetscBool)(pcbddc->adaptive_selection || pcbddc->use_deluxe_scaling);
147416909a7fSStefano Zampini   if (pcbddc->switch_static) {
147516909a7fSStefano Zampini     PetscBool ismatis;
147616909a7fSStefano Zampini     ierr = PetscObjectTypeCompare((PetscObject)pc->mat,MATIS,&ismatis);CHKERRQ(ierr);
147716909a7fSStefano Zampini     if (!ismatis) SETERRQ(PetscObjectComm((PetscObject)pc),PETSC_ERR_SUP,"When the static switch is one, the iteration matrix should be of type MATIS");
147816909a7fSStefano Zampini   }
147916909a7fSStefano Zampini 
148071582508SStefano Zampini   /* activate all connected components if the netflux has been requested */
1481bb05f991SStefano Zampini   if (pcbddc->compute_nonetflux) {
1482bb05f991SStefano Zampini     pcbddc->use_vertices = PETSC_TRUE;
1483bb05f991SStefano Zampini     pcbddc->use_edges = PETSC_TRUE;
1484bb05f991SStefano Zampini     pcbddc->use_faces = PETSC_TRUE;
1485bb05f991SStefano Zampini   }
1486bb05f991SStefano Zampini 
1487f4ddd8eeSStefano Zampini   /* Get stdout for dbg */
148870cf5478SStefano Zampini   if (pcbddc->dbg_flag) {
148970cf5478SStefano Zampini     if (!pcbddc->dbg_viewer) {
149058a03d70SStefano Zampini       pcbddc->dbg_viewer = PETSC_VIEWER_STDOUT_(PetscObjectComm((PetscObject)pc));
14911575c14dSBarry Smith       ierr = PetscViewerASCIIPushSynchronized(pcbddc->dbg_viewer);CHKERRQ(ierr);
1492f4ddd8eeSStefano Zampini     }
149358a03d70SStefano Zampini     ierr = PetscViewerASCIIAddTab(pcbddc->dbg_viewer,2*pcbddc->current_level);CHKERRQ(ierr);
1494f4ddd8eeSStefano Zampini   }
1495f4ddd8eeSStefano Zampini 
1496c703fcc7SStefano Zampini   /* process topology information */
149771582508SStefano Zampini   if (pcbddc->recompute_topography) {
149871582508SStefano Zampini     ierr = PCBDDCComputeLocalTopologyInfo(pc);CHKERRQ(ierr);
1499c83e1ba7SStefano Zampini     /* detect local disconnected subdomains if requested (use matis->A) */
1500c83e1ba7SStefano Zampini     if (pcbddc->detect_disconnected) {
1501c80a6c00SStefano Zampini       IS       primalv;
1502c83e1ba7SStefano Zampini       PetscInt i;
1503c80a6c00SStefano Zampini 
1504c83e1ba7SStefano Zampini       for (i=0;i<pcbddc->n_local_subs;i++) {
1505c83e1ba7SStefano Zampini         ierr = ISDestroy(&pcbddc->local_subs[i]);CHKERRQ(ierr);
1506c83e1ba7SStefano Zampini       }
1507c83e1ba7SStefano Zampini       ierr = PetscFree(pcbddc->local_subs);CHKERRQ(ierr);
1508c80a6c00SStefano Zampini       ierr = PCBDDCDetectDisconnectedComponents(pc,&pcbddc->n_local_subs,&pcbddc->local_subs,&primalv);CHKERRQ(ierr);
1509c80a6c00SStefano Zampini       if (primalv) {
1510c80a6c00SStefano Zampini         if (pcbddc->user_primal_vertices_local) {
1511c80a6c00SStefano Zampini           IS list[2] = {primalv, pcbddc->user_primal_vertices_local}, newp;
1512c80a6c00SStefano Zampini 
1513c80a6c00SStefano Zampini           ierr = ISConcatenate(PetscObjectComm((PetscObject)pc),2,list,&newp);CHKERRQ(ierr);
1514c80a6c00SStefano Zampini           ierr = ISDestroy(&list[0]);CHKERRQ(ierr);
1515c80a6c00SStefano Zampini           ierr = ISDestroy(&list[1]);CHKERRQ(ierr);
1516c80a6c00SStefano Zampini           pcbddc->user_primal_vertices_local = newp;
1517c80a6c00SStefano Zampini         } else {
1518c80a6c00SStefano Zampini           pcbddc->user_primal_vertices_local = primalv;
1519c80a6c00SStefano Zampini         }
1520c80a6c00SStefano Zampini       }
1521c83e1ba7SStefano Zampini     }
1522c703fcc7SStefano Zampini     if (pcbddc->discretegradient) {
1523a13144ffSStefano Zampini       ierr = PCBDDCNedelecSupport(pc);CHKERRQ(ierr);
1524a13144ffSStefano Zampini     }
1525c703fcc7SStefano Zampini   }
1526a13144ffSStefano Zampini 
1527c703fcc7SStefano Zampini   /* change basis if requested by the user */
15285e8657edSStefano Zampini   if (pcbddc->user_ChangeOfBasisMatrix) {
15295e8657edSStefano Zampini     /* use_change_of_basis flag is used to automatically compute a change of basis from constraints */
15305e8657edSStefano Zampini     pcbddc->use_change_of_basis = PETSC_FALSE;
15315e8657edSStefano Zampini     ierr = PCBDDCComputeLocalMatrix(pc,pcbddc->user_ChangeOfBasisMatrix);CHKERRQ(ierr);
15325e8657edSStefano Zampini   } else {
1533b96c3477SStefano Zampini     ierr = MatDestroy(&pcbddc->local_mat);CHKERRQ(ierr);
15345e8657edSStefano Zampini     ierr = PetscObjectReference((PetscObject)matis->A);CHKERRQ(ierr);
15355e8657edSStefano Zampini     pcbddc->local_mat = matis->A;
1536d16cbb6bSStefano Zampini   }
1537d16cbb6bSStefano Zampini 
15384f1b2e48SStefano Zampini   /*
1539c703fcc7SStefano Zampini      Compute change of basis on local pressures (aka zerodiag dofs) with the benign trick
15404f1b2e48SStefano Zampini      This should come earlier then PCISSetUp for extracting the correct subdomain matrices
15414f1b2e48SStefano Zampini   */
15421dd7afcfSStefano Zampini   ierr = PCBDDCBenignShellMat(pc,PETSC_TRUE);CHKERRQ(ierr);
1543d16cbb6bSStefano Zampini   if (pcbddc->benign_saddle_point) {
15449f47a83aSStefano Zampini     PC_IS* pcis = (PC_IS*)pc->data;
15459f47a83aSStefano Zampini 
154605b28244SStefano Zampini     if (pcbddc->user_ChangeOfBasisMatrix || pcbddc->use_change_of_basis || !computesubschurs) pcbddc->benign_change_explicit = PETSC_TRUE;
1547669cc0f4SStefano Zampini     /* detect local saddle point and change the basis in pcbddc->local_mat (TODO: reuse case) */
1548339f8db1SStefano Zampini     ierr = PCBDDCBenignDetectSaddlePoint(pc,&zerodiag);CHKERRQ(ierr);
1549a3df083aSStefano Zampini     /* pop B0 mat from local mat */
1550c263805aSStefano Zampini     ierr = PCBDDCBenignPopOrPushB0(pc,PETSC_TRUE);CHKERRQ(ierr);
15511dd7afcfSStefano Zampini     /* give pcis a hint to not reuse submatrices during PCISCreate */
15521dd7afcfSStefano Zampini     if (pc->flag == SAME_NONZERO_PATTERN && pcis->reusesubmatrices == PETSC_TRUE) {
15531dd7afcfSStefano Zampini       if (pcbddc->benign_n && (pcbddc->benign_change_explicit || pcbddc->dbg_flag)) {
15541dd7afcfSStefano Zampini         pcis->reusesubmatrices = PETSC_FALSE;
15551dd7afcfSStefano Zampini       } else {
1556a3df083aSStefano Zampini         pcis->reusesubmatrices = PETSC_TRUE;
15571dd7afcfSStefano Zampini       }
1558a3df083aSStefano Zampini     } else {
15599f47a83aSStefano Zampini       pcis->reusesubmatrices = PETSC_FALSE;
1560674ae819SStefano Zampini     }
1561a3df083aSStefano Zampini   }
156227b6a85dSStefano Zampini 
1563c703fcc7SStefano Zampini   /* propagate relevant information -> TODO remove*/
15644f1b2e48SStefano Zampini #if !defined(PETSC_USE_COMPLEX) /* workaround for reals */
15653301b35fSStefano Zampini   if (matis->A->symmetric_set) {
15663301b35fSStefano Zampini     ierr = MatSetOption(pcbddc->local_mat,MAT_HERMITIAN,matis->A->symmetric);CHKERRQ(ierr);
15673301b35fSStefano Zampini   }
1568e496cd5dSStefano Zampini #endif
156906a4e24aSStefano Zampini   if (matis->A->symmetric_set) {
157006a4e24aSStefano Zampini     ierr = MatSetOption(pcbddc->local_mat,MAT_SYMMETRIC,matis->A->symmetric);CHKERRQ(ierr);
157106a4e24aSStefano Zampini   }
157206a4e24aSStefano Zampini   if (matis->A->spd_set) {
157306a4e24aSStefano Zampini     ierr = MatSetOption(pcbddc->local_mat,MAT_SPD,matis->A->spd);CHKERRQ(ierr);
157406a4e24aSStefano Zampini   }
1575e496cd5dSStefano Zampini 
15765e8657edSStefano Zampini   /* Set up all the "iterative substructuring" common block without computing solvers */
15775e8657edSStefano Zampini   {
15785e8657edSStefano Zampini     Mat temp_mat;
15795e8657edSStefano Zampini 
15805e8657edSStefano Zampini     temp_mat = matis->A;
15815e8657edSStefano Zampini     matis->A = pcbddc->local_mat;
15825e8657edSStefano Zampini     ierr = PCISSetUp(pc,PETSC_FALSE);CHKERRQ(ierr);
15835e8657edSStefano Zampini     pcbddc->local_mat = matis->A;
15845e8657edSStefano Zampini     matis->A = temp_mat;
15855e8657edSStefano Zampini   }
1586684f6988SStefano Zampini 
158781d14e9dSStefano Zampini   /* Analyze interface */
158864ac59b8SStefano Zampini   if (!pcbddc->graphanalyzed) {
1589674ae819SStefano Zampini     ierr = PCBDDCAnalyzeInterface(pc);CHKERRQ(ierr);
15908de1fae6SStefano Zampini     computeconstraintsmatrix = PETSC_TRUE;
1591345ecf6cSStefano Zampini     if (pcbddc->adaptive_selection && !pcbddc->use_deluxe_scaling && !pcbddc->mat_graph->twodim) {
15924247aa23Sstefano_zampini       SETERRQ(PetscObjectComm((PetscObject)pc),PETSC_ERR_SUP,"Cannot compute the adaptive primal space for a problem with 3D edges without deluxe scaling");
1593345ecf6cSStefano Zampini     }
1594a198735bSStefano Zampini     if (pcbddc->compute_nonetflux) {
1595669cc0f4SStefano Zampini       MatNullSpace nnfnnsp;
1596669cc0f4SStefano Zampini 
159721ef3d20SStefano Zampini       if (!pcbddc->divudotp) SETERRQ(PetscObjectComm((PetscObject)pc),PETSC_ERR_SUP,"Missing divudotp operator");
15988ae0ca82SStefano Zampini       ierr = PCBDDCComputeNoNetFlux(pc->pmat,pcbddc->divudotp,pcbddc->divudotp_trans,pcbddc->divudotp_vl2l,pcbddc->mat_graph,&nnfnnsp);CHKERRQ(ierr);
159971582508SStefano Zampini       /* TODO what if a nearnullspace is already attached? */
1600669cc0f4SStefano Zampini       ierr = MatSetNearNullSpace(pc->pmat,nnfnnsp);CHKERRQ(ierr);
1601669cc0f4SStefano Zampini       ierr = MatNullSpaceDestroy(&nnfnnsp);CHKERRQ(ierr);
1602669cc0f4SStefano Zampini     }
1603674ae819SStefano Zampini   }
1604fb8d54d4SStefano Zampini 
16055408967cSStefano Zampini   /* check existence of a divergence free extension, i.e.
16065408967cSStefano Zampini      b(v_I,p_0) = 0 for all v_I (raise error if not).
16075408967cSStefano Zampini      Also, check that PCBDDCBenignGetOrSetP0 works */
1608ff1f7e73Sstefano_zampini   if (pcbddc->benign_saddle_point && pcbddc->dbg_flag > 1) {
16095408967cSStefano Zampini     ierr = PCBDDCBenignCheck(pc,zerodiag);CHKERRQ(ierr);
161009f581a4SStefano Zampini   }
16114f1b2e48SStefano Zampini   ierr = ISDestroy(&zerodiag);CHKERRQ(ierr);
161206f24817SStefano Zampini 
1613b96c3477SStefano Zampini   /* Setup local dirichlet solver ksp_D and sub_schurs solvers */
1614c703fcc7SStefano Zampini   if (computesubschurs && pcbddc->recompute_topography) {
161508122e43SStefano Zampini     ierr = PCBDDCInitSubSchurs(pc);CHKERRQ(ierr);
1616b1b3d7a2SStefano Zampini   }
16179d54b7f4SStefano Zampini   /* SetUp Scaling operator (scaling matrices could be needed in SubSchursSetUp)*/
16189d54b7f4SStefano Zampini   if (!pcbddc->use_deluxe_scaling) {
16199d54b7f4SStefano Zampini     ierr = PCBDDCScalingSetUp(pc);CHKERRQ(ierr);
16209d54b7f4SStefano Zampini   }
1621c703fcc7SStefano Zampini 
1622c703fcc7SStefano Zampini   /* finish setup solvers and do adaptive selection of constraints */
1623b334f244SStefano Zampini   sub_schurs = pcbddc->sub_schurs;
1624b334f244SStefano Zampini   if (sub_schurs && sub_schurs->schur_explicit) {
16252070dbb6SStefano Zampini     if (computesubschurs) {
162608122e43SStefano Zampini       ierr = PCBDDCSetUpSubSchurs(pc);CHKERRQ(ierr);
16272070dbb6SStefano Zampini     }
1628d5574798SStefano Zampini     ierr = PCBDDCSetUpLocalSolvers(pc,PETSC_TRUE,PETSC_FALSE);CHKERRQ(ierr);
1629d5574798SStefano Zampini   } else {
1630d5574798SStefano Zampini     ierr = PCBDDCSetUpLocalSolvers(pc,PETSC_TRUE,PETSC_FALSE);CHKERRQ(ierr);
16312070dbb6SStefano Zampini     if (computesubschurs) {
1632d5574798SStefano Zampini       ierr = PCBDDCSetUpSubSchurs(pc);CHKERRQ(ierr);
1633d5574798SStefano Zampini     }
16342070dbb6SStefano Zampini   }
163508122e43SStefano Zampini   if (pcbddc->adaptive_selection) {
163608122e43SStefano Zampini     ierr = PCBDDCAdaptiveSelection(pc);CHKERRQ(ierr);
16378de1fae6SStefano Zampini     computeconstraintsmatrix = PETSC_TRUE;
1638b7eb3628SStefano Zampini   }
1639684f6988SStefano Zampini 
1640f4ddd8eeSStefano Zampini   /* infer if NullSpace object attached to Mat via MatSetNearNullSpace has changed */
1641fb8d54d4SStefano Zampini   new_nearnullspace_provided = PETSC_FALSE;
1642f4ddd8eeSStefano Zampini   ierr = MatGetNearNullSpace(pc->pmat,&nearnullspace);CHKERRQ(ierr);
1643f4ddd8eeSStefano Zampini   if (pcbddc->onearnullspace) { /* already used nearnullspace */
1644f4ddd8eeSStefano Zampini     if (!nearnullspace) { /* near null space attached to mat has been destroyed */
1645f4ddd8eeSStefano Zampini       new_nearnullspace_provided = PETSC_TRUE;
1646f4ddd8eeSStefano Zampini     } else {
1647f4ddd8eeSStefano Zampini       /* determine if the two nullspaces are different (should be lightweight) */
1648f4ddd8eeSStefano Zampini       if (nearnullspace != pcbddc->onearnullspace) {
1649f4ddd8eeSStefano Zampini         new_nearnullspace_provided = PETSC_TRUE;
1650165b64e2SStefano Zampini       } else { /* maybe the user has changed the content of the nearnullspace so check vectors ObjectStateId */
1651f4ddd8eeSStefano Zampini         PetscInt         i;
1652165b64e2SStefano Zampini         const Vec        *nearnullvecs;
1653165b64e2SStefano Zampini         PetscObjectState state;
1654165b64e2SStefano Zampini         PetscInt         nnsp_size;
1655165b64e2SStefano Zampini         ierr = MatNullSpaceGetVecs(nearnullspace,NULL,&nnsp_size,&nearnullvecs);CHKERRQ(ierr);
1656f4ddd8eeSStefano Zampini         for (i=0;i<nnsp_size;i++) {
1657f4ddd8eeSStefano Zampini           ierr = PetscObjectStateGet((PetscObject)nearnullvecs[i],&state);CHKERRQ(ierr);
1658165b64e2SStefano Zampini           if (pcbddc->onearnullvecs_state[i] != state) {
1659f4ddd8eeSStefano Zampini             new_nearnullspace_provided = PETSC_TRUE;
1660f4ddd8eeSStefano Zampini             break;
1661f4ddd8eeSStefano Zampini           }
1662f4ddd8eeSStefano Zampini         }
1663f4ddd8eeSStefano Zampini       }
1664f4ddd8eeSStefano Zampini     }
1665f4ddd8eeSStefano Zampini   } else {
1666f4ddd8eeSStefano Zampini     if (!nearnullspace) { /* both nearnullspaces are null */
1667f4ddd8eeSStefano Zampini       new_nearnullspace_provided = PETSC_FALSE;
1668f4ddd8eeSStefano Zampini     } else { /* nearnullspace attached later */
1669f4ddd8eeSStefano Zampini       new_nearnullspace_provided = PETSC_TRUE;
1670f4ddd8eeSStefano Zampini     }
1671f4ddd8eeSStefano Zampini   }
1672f4ddd8eeSStefano Zampini 
1673f4ddd8eeSStefano Zampini   /* Setup constraints and related work vectors */
1674727cdba6SStefano Zampini   /* reset primal space flags */
1675f4ddd8eeSStefano Zampini   pcbddc->new_primal_space = PETSC_FALSE;
1676727cdba6SStefano Zampini   pcbddc->new_primal_space_local = PETSC_FALSE;
16778de1fae6SStefano Zampini   if (computeconstraintsmatrix || new_nearnullspace_provided) {
1678727cdba6SStefano Zampini     /* It also sets the primal space flags */
1679674ae819SStefano Zampini     ierr = PCBDDCConstraintsSetUp(pc);CHKERRQ(ierr);
16809543d0ffSStefano Zampini   }
1681e7b262bdSStefano Zampini   /* Allocate needed local vectors (which depends on quantities defined during ConstraintsSetUp) */
1682f4ddd8eeSStefano Zampini   ierr = PCBDDCSetUpLocalWorkVectors(pc);CHKERRQ(ierr);
16835e8657edSStefano Zampini 
16845e8657edSStefano Zampini   if (pcbddc->use_change_of_basis) {
16855e8657edSStefano Zampini     PC_IS *pcis = (PC_IS*)(pc->data);
16865e8657edSStefano Zampini 
16875e8657edSStefano Zampini     ierr = PCBDDCComputeLocalMatrix(pc,pcbddc->ChangeOfBasisMatrix);CHKERRQ(ierr);
16884f1b2e48SStefano Zampini     if (pcbddc->benign_change) {
16891dd7afcfSStefano Zampini       ierr = MatDestroy(&pcbddc->benign_B0);CHKERRQ(ierr);
1690c263805aSStefano Zampini       /* pop B0 from pcbddc->local_mat */
1691c263805aSStefano Zampini       ierr = PCBDDCBenignPopOrPushB0(pc,PETSC_TRUE);CHKERRQ(ierr);
1692c263805aSStefano Zampini     }
16935e8657edSStefano Zampini     /* get submatrices */
16945e8657edSStefano Zampini     ierr = MatDestroy(&pcis->A_IB);CHKERRQ(ierr);
16955e8657edSStefano Zampini     ierr = MatDestroy(&pcis->A_BI);CHKERRQ(ierr);
16965e8657edSStefano Zampini     ierr = MatDestroy(&pcis->A_BB);CHKERRQ(ierr);
16977dae84e0SHong Zhang     ierr = MatCreateSubMatrix(pcbddc->local_mat,pcis->is_B_local,pcis->is_B_local,MAT_INITIAL_MATRIX,&pcis->A_BB);CHKERRQ(ierr);
16987dae84e0SHong Zhang     ierr = MatCreateSubMatrix(pcbddc->local_mat,pcis->is_I_local,pcis->is_B_local,MAT_INITIAL_MATRIX,&pcis->A_IB);CHKERRQ(ierr);
16997dae84e0SHong Zhang     ierr = MatCreateSubMatrix(pcbddc->local_mat,pcis->is_B_local,pcis->is_I_local,MAT_INITIAL_MATRIX,&pcis->A_BI);CHKERRQ(ierr);
17003975b054SStefano Zampini     /* set flag in pcis to not reuse submatrices during PCISCreate */
17013975b054SStefano Zampini     pcis->reusesubmatrices = PETSC_FALSE;
17029c6a02ceSStefano Zampini   } else if (!pcbddc->user_ChangeOfBasisMatrix && !pcbddc->benign_change) {
1703b96c3477SStefano Zampini     ierr = MatDestroy(&pcbddc->local_mat);CHKERRQ(ierr);
17045e8657edSStefano Zampini     ierr = PetscObjectReference((PetscObject)matis->A);CHKERRQ(ierr);
17055e8657edSStefano Zampini     pcbddc->local_mat = matis->A;
17065e8657edSStefano Zampini   }
170735509ce9Sstefano_zampini 
170835509ce9Sstefano_zampini   /* interface pressure block row for B_C */
170935509ce9Sstefano_zampini   ierr = PetscObjectQuery((PetscObject)pc,"__KSPFETIDP_lP" ,(PetscObject*)&lP);CHKERRQ(ierr);
171035509ce9Sstefano_zampini   ierr = PetscObjectQuery((PetscObject)pc,"__KSPFETIDP_lA" ,(PetscObject*)&lA);CHKERRQ(ierr);
171135509ce9Sstefano_zampini   if (lA && lP) {
171235509ce9Sstefano_zampini     PC_IS*    pcis = (PC_IS*)pc->data;
171335509ce9Sstefano_zampini     Mat       B_BI,B_BB,Bt_BI,Bt_BB;
171435509ce9Sstefano_zampini     PetscBool issym;
171535509ce9Sstefano_zampini     ierr = MatIsSymmetric(lA,PETSC_SMALL,&issym);CHKERRQ(ierr);
17166cc1294bSstefano_zampini     if (issym) {
17177dae84e0SHong Zhang       ierr = MatCreateSubMatrix(lA,lP,pcis->is_I_local,MAT_INITIAL_MATRIX,&B_BI);CHKERRQ(ierr);
17187dae84e0SHong Zhang       ierr = MatCreateSubMatrix(lA,lP,pcis->is_B_local,MAT_INITIAL_MATRIX,&B_BB);CHKERRQ(ierr);
171935509ce9Sstefano_zampini       ierr = MatCreateTranspose(B_BI,&Bt_BI);CHKERRQ(ierr);
172035509ce9Sstefano_zampini       ierr = MatCreateTranspose(B_BB,&Bt_BB);CHKERRQ(ierr);
172135509ce9Sstefano_zampini     } else {
17227dae84e0SHong Zhang       ierr = MatCreateSubMatrix(lA,lP,pcis->is_I_local,MAT_INITIAL_MATRIX,&B_BI);CHKERRQ(ierr);
17237dae84e0SHong Zhang       ierr = MatCreateSubMatrix(lA,lP,pcis->is_B_local,MAT_INITIAL_MATRIX,&B_BB);CHKERRQ(ierr);
17247dae84e0SHong Zhang       ierr = MatCreateSubMatrix(lA,pcis->is_I_local,lP,MAT_INITIAL_MATRIX,&Bt_BI);CHKERRQ(ierr);
17257dae84e0SHong Zhang       ierr = MatCreateSubMatrix(lA,pcis->is_B_local,lP,MAT_INITIAL_MATRIX,&Bt_BB);CHKERRQ(ierr);
172635509ce9Sstefano_zampini     }
172735509ce9Sstefano_zampini     ierr = PetscObjectCompose((PetscObject)pc,"__KSPFETIDP_B_BI",(PetscObject)B_BI);CHKERRQ(ierr);
172835509ce9Sstefano_zampini     ierr = PetscObjectCompose((PetscObject)pc,"__KSPFETIDP_B_BB",(PetscObject)B_BB);CHKERRQ(ierr);
172935509ce9Sstefano_zampini     ierr = PetscObjectCompose((PetscObject)pc,"__KSPFETIDP_Bt_BI",(PetscObject)Bt_BI);CHKERRQ(ierr);
173035509ce9Sstefano_zampini     ierr = PetscObjectCompose((PetscObject)pc,"__KSPFETIDP_Bt_BB",(PetscObject)Bt_BB);CHKERRQ(ierr);
173135509ce9Sstefano_zampini     ierr = MatDestroy(&B_BI);CHKERRQ(ierr);
173235509ce9Sstefano_zampini     ierr = MatDestroy(&B_BB);CHKERRQ(ierr);
173335509ce9Sstefano_zampini     ierr = MatDestroy(&Bt_BI);CHKERRQ(ierr);
173435509ce9Sstefano_zampini     ierr = MatDestroy(&Bt_BB);CHKERRQ(ierr);
173535509ce9Sstefano_zampini   }
173635509ce9Sstefano_zampini 
1737b96c3477SStefano Zampini   /* SetUp coarse and local Neumann solvers */
173899cc7994SStefano Zampini   ierr = PCBDDCSetUpSolvers(pc);CHKERRQ(ierr);
1739b96c3477SStefano Zampini   /* SetUp Scaling operator */
17409d54b7f4SStefano Zampini   if (pcbddc->use_deluxe_scaling) {
1741674ae819SStefano Zampini     ierr = PCBDDCScalingSetUp(pc);CHKERRQ(ierr);
17420c7d97c5SJed Brown   }
1743c703fcc7SStefano Zampini 
17441dd7afcfSStefano Zampini   /* mark topography as done */
174556282151SStefano Zampini   pcbddc->recompute_topography = PETSC_FALSE;
17460369aaf7SStefano Zampini 
17471dd7afcfSStefano Zampini   /* wrap pcis->A_IB and pcis->A_BI if we did not change explicitly the variables on the pressures */
17481dd7afcfSStefano Zampini   ierr = PCBDDCBenignShellMat(pc,PETSC_FALSE);CHKERRQ(ierr);
17491dd7afcfSStefano Zampini 
175058a03d70SStefano Zampini   if (pcbddc->dbg_flag) {
175158a03d70SStefano Zampini     ierr = PetscViewerASCIISubtractTab(pcbddc->dbg_viewer,2*pcbddc->current_level);CHKERRQ(ierr);
17522b510759SStefano Zampini   }
17530c7d97c5SJed Brown   PetscFunctionReturn(0);
17540c7d97c5SJed Brown }
17550c7d97c5SJed Brown 
17560c7d97c5SJed Brown /*
175750efa1b5SStefano Zampini    PCApply_BDDC - Applies the BDDC operator to a vector.
17580c7d97c5SJed Brown 
17590c7d97c5SJed Brown    Input Parameters:
17600f202f7eSStefano Zampini +  pc - the preconditioner context
17610f202f7eSStefano Zampini -  r - input vector (global)
17620c7d97c5SJed Brown 
17630c7d97c5SJed Brown    Output Parameter:
17640c7d97c5SJed Brown .  z - output vector (global)
17650c7d97c5SJed Brown 
17660c7d97c5SJed Brown    Application Interface Routine: PCApply()
17670c7d97c5SJed Brown  */
176853cdbc3dSStefano Zampini PetscErrorCode PCApply_BDDC(PC pc,Vec r,Vec z)
17690c7d97c5SJed Brown {
17700c7d97c5SJed Brown   PC_IS             *pcis = (PC_IS*)(pc->data);
17710c7d97c5SJed Brown   PC_BDDC           *pcbddc = (PC_BDDC*)(pc->data);
1772b097fa66SStefano Zampini   PetscInt          n_B = pcis->n_B, n_D = pcis->n - n_B;
17730c7d97c5SJed Brown   PetscErrorCode    ierr;
17743b03a366Sstefano_zampini   const PetscScalar one = 1.0;
17753b03a366Sstefano_zampini   const PetscScalar m_one = -1.0;
17762617d88aSStefano Zampini   const PetscScalar zero = 0.0;
17770c7d97c5SJed Brown 
17780c7d97c5SJed Brown /* This code is similar to that provided in nn.c for PCNN
17790c7d97c5SJed Brown    NN interface preconditioner changed to BDDC
1780b097fa66SStefano Zampini    Added support for M_3 preconditioner in the reference article (code is active if pcbddc->switch_static == PETSC_TRUE) */
17810c7d97c5SJed Brown 
17820c7d97c5SJed Brown   PetscFunctionBegin;
1783f3d41395Sstefano_zampini   ierr = PetscCitationsRegister(citation,&cited);CHKERRQ(ierr);
17841dd7afcfSStefano Zampini   if (pcbddc->ChangeOfBasisMatrix) {
17851dd7afcfSStefano Zampini     Vec swap;
178627b6a85dSStefano Zampini 
178727b6a85dSStefano Zampini     ierr = MatMultTranspose(pcbddc->ChangeOfBasisMatrix,r,pcbddc->work_change);CHKERRQ(ierr);
17881dd7afcfSStefano Zampini     swap = pcbddc->work_change;
17891dd7afcfSStefano Zampini     pcbddc->work_change = r;
17901dd7afcfSStefano Zampini     r = swap;
17911dd7afcfSStefano Zampini     /* save rhs so that we don't need to apply the change of basis for the exact dirichlet trick in PreSolve */
17929cc2a9b1Sstefano_zampini     if (pcbddc->benign_apply_coarse_only && pcbddc->use_exact_dirichlet_trick && pcbddc->change_interior) {
17931dd7afcfSStefano Zampini       ierr = VecCopy(r,pcis->vec1_global);CHKERRQ(ierr);
17941dd7afcfSStefano Zampini       ierr = VecLockPush(pcis->vec1_global);CHKERRQ(ierr);
17951dd7afcfSStefano Zampini     }
17961dd7afcfSStefano Zampini   }
179727b6a85dSStefano Zampini   if (pcbddc->benign_have_null) { /* get p0 from r */
1798015636ebSStefano Zampini     ierr = PCBDDCBenignGetOrSetP0(pc,r,PETSC_TRUE);CHKERRQ(ierr);
1799efc2fbd9SStefano Zampini   }
18008ae0ca82SStefano Zampini   if (!pcbddc->exact_dirichlet_trick_app && !pcbddc->benign_apply_coarse_only) {
1801b097fa66SStefano Zampini     ierr = VecCopy(r,z);CHKERRQ(ierr);
18020c7d97c5SJed Brown     /* First Dirichlet solve */
18030c7d97c5SJed Brown     ierr = VecScatterBegin(pcis->global_to_D,r,pcis->vec1_D,INSERT_VALUES,SCATTER_FORWARD);CHKERRQ(ierr);
18040c7d97c5SJed Brown     ierr = VecScatterEnd(pcis->global_to_D,r,pcis->vec1_D,INSERT_VALUES,SCATTER_FORWARD);CHKERRQ(ierr);
18050c7d97c5SJed Brown     /*
18060c7d97c5SJed Brown       Assembling right hand side for BDDC operator
1807b097fa66SStefano Zampini       - pcis->vec1_D for the Dirichlet part (if needed, i.e. pcbddc->switch_static == PETSC_TRUE)
1808674ae819SStefano Zampini       - pcis->vec1_B the interface part of the global vector z
18090c7d97c5SJed Brown     */
1810b097fa66SStefano Zampini     if (n_D) {
1811b097fa66SStefano Zampini       ierr = KSPSolve(pcbddc->ksp_D,pcis->vec1_D,pcis->vec2_D);CHKERRQ(ierr);
18120c7d97c5SJed Brown       ierr = VecScale(pcis->vec2_D,m_one);CHKERRQ(ierr);
181316909a7fSStefano Zampini       if (pcbddc->switch_static) {
181416909a7fSStefano Zampini         Mat_IS *matis = (Mat_IS*)(pc->mat->data);
181516909a7fSStefano Zampini 
181616909a7fSStefano Zampini         ierr = VecSet(pcis->vec1_N,0.);CHKERRQ(ierr);
181716909a7fSStefano Zampini         ierr = VecScatterBegin(pcis->N_to_D,pcis->vec2_D,pcis->vec1_N,INSERT_VALUES,SCATTER_REVERSE);CHKERRQ(ierr);
181816909a7fSStefano Zampini         ierr = VecScatterEnd(pcis->N_to_D,pcis->vec2_D,pcis->vec1_N,INSERT_VALUES,SCATTER_REVERSE);CHKERRQ(ierr);
181916909a7fSStefano Zampini         if (!pcbddc->switch_static_change) {
182016909a7fSStefano Zampini           ierr = MatMult(matis->A,pcis->vec1_N,pcis->vec2_N);CHKERRQ(ierr);
182116909a7fSStefano Zampini         } else {
182216909a7fSStefano Zampini           ierr = MatMult(pcbddc->switch_static_change,pcis->vec1_N,pcis->vec2_N);CHKERRQ(ierr);
182316909a7fSStefano Zampini           ierr = MatMult(matis->A,pcis->vec2_N,pcis->vec1_N);CHKERRQ(ierr);
182416909a7fSStefano Zampini           ierr = MatMultTranspose(pcbddc->switch_static_change,pcis->vec1_N,pcis->vec2_N);CHKERRQ(ierr);
182516909a7fSStefano Zampini         }
182616909a7fSStefano Zampini         ierr = VecScatterBegin(pcis->N_to_D,pcis->vec2_N,pcis->vec1_D,ADD_VALUES,SCATTER_FORWARD);CHKERRQ(ierr);
182716909a7fSStefano Zampini         ierr = VecScatterEnd(pcis->N_to_D,pcis->vec2_N,pcis->vec1_D,ADD_VALUES,SCATTER_FORWARD);CHKERRQ(ierr);
182816909a7fSStefano Zampini         ierr = VecScatterBegin(pcis->N_to_B,pcis->vec2_N,pcis->vec1_B,INSERT_VALUES,SCATTER_FORWARD);CHKERRQ(ierr);
182916909a7fSStefano Zampini         ierr = VecScatterEnd(pcis->N_to_B,pcis->vec2_N,pcis->vec1_B,INSERT_VALUES,SCATTER_FORWARD);CHKERRQ(ierr);
183016909a7fSStefano Zampini       } else {
1831b097fa66SStefano Zampini         ierr = MatMult(pcis->A_BI,pcis->vec2_D,pcis->vec1_B);CHKERRQ(ierr);
183216909a7fSStefano Zampini       }
1833b097fa66SStefano Zampini     } else {
1834b097fa66SStefano Zampini       ierr = VecSet(pcis->vec1_B,zero);CHKERRQ(ierr);
1835b097fa66SStefano Zampini     }
18360c7d97c5SJed Brown     ierr = VecScatterBegin(pcis->global_to_B,pcis->vec1_B,z,ADD_VALUES,SCATTER_REVERSE);CHKERRQ(ierr);
18370c7d97c5SJed Brown     ierr = VecScatterEnd(pcis->global_to_B,pcis->vec1_B,z,ADD_VALUES,SCATTER_REVERSE);CHKERRQ(ierr);
1838674ae819SStefano Zampini     ierr = PCBDDCScalingRestriction(pc,z,pcis->vec1_B);CHKERRQ(ierr);
1839b76ba322SStefano Zampini   } else {
18404fee134fSStefano Zampini     if (!pcbddc->benign_apply_coarse_only) {
1841674ae819SStefano Zampini       ierr = PCBDDCScalingRestriction(pc,r,pcis->vec1_B);CHKERRQ(ierr);
1842b76ba322SStefano Zampini     }
18434fee134fSStefano Zampini   }
1844b76ba322SStefano Zampini 
18452617d88aSStefano Zampini   /* Apply interface preconditioner
18462617d88aSStefano Zampini      input/output vecs: pcis->vec1_B and pcis->vec1_D */
1847dc359a40SStefano Zampini   ierr = PCBDDCApplyInterfacePreconditioner(pc,PETSC_FALSE);CHKERRQ(ierr);
18482617d88aSStefano Zampini 
1849674ae819SStefano Zampini   /* Apply transpose of partition of unity operator */
1850674ae819SStefano Zampini   ierr = PCBDDCScalingExtension(pc,pcis->vec1_B,z);CHKERRQ(ierr);
18510c7d97c5SJed Brown 
18523b03a366Sstefano_zampini   /* Second Dirichlet solve and assembling of output */
18530c7d97c5SJed Brown   ierr = VecScatterBegin(pcis->global_to_B,z,pcis->vec1_B,INSERT_VALUES,SCATTER_FORWARD);CHKERRQ(ierr);
18540c7d97c5SJed Brown   ierr = VecScatterEnd(pcis->global_to_B,z,pcis->vec1_B,INSERT_VALUES,SCATTER_FORWARD);CHKERRQ(ierr);
1855b097fa66SStefano Zampini   if (n_B) {
185616909a7fSStefano Zampini     if (pcbddc->switch_static) {
185716909a7fSStefano Zampini       Mat_IS *matis = (Mat_IS*)(pc->mat->data);
185816909a7fSStefano Zampini 
185916909a7fSStefano Zampini       ierr = VecScatterBegin(pcis->N_to_D,pcis->vec1_D,pcis->vec1_N,INSERT_VALUES,SCATTER_REVERSE);CHKERRQ(ierr);
186016909a7fSStefano Zampini       ierr = VecScatterEnd(pcis->N_to_D,pcis->vec1_D,pcis->vec1_N,INSERT_VALUES,SCATTER_REVERSE);CHKERRQ(ierr);
186116909a7fSStefano Zampini       ierr = VecScatterBegin(pcis->N_to_B,pcis->vec1_B,pcis->vec1_N,INSERT_VALUES,SCATTER_REVERSE);CHKERRQ(ierr);
186216909a7fSStefano Zampini       ierr = VecScatterEnd(pcis->N_to_B,pcis->vec1_B,pcis->vec1_N,INSERT_VALUES,SCATTER_REVERSE);CHKERRQ(ierr);
186316909a7fSStefano Zampini       if (!pcbddc->switch_static_change) {
186416909a7fSStefano Zampini         ierr = MatMult(matis->A,pcis->vec1_N,pcis->vec2_N);CHKERRQ(ierr);
186516909a7fSStefano Zampini       } else {
186616909a7fSStefano Zampini         ierr = MatMult(pcbddc->switch_static_change,pcis->vec1_N,pcis->vec2_N);CHKERRQ(ierr);
186716909a7fSStefano Zampini         ierr = MatMult(matis->A,pcis->vec2_N,pcis->vec1_N);CHKERRQ(ierr);
186816909a7fSStefano Zampini         ierr = MatMultTranspose(pcbddc->switch_static_change,pcis->vec1_N,pcis->vec2_N);CHKERRQ(ierr);
186916909a7fSStefano Zampini       }
187016909a7fSStefano Zampini       ierr = VecScatterBegin(pcis->N_to_D,pcis->vec2_N,pcis->vec3_D,INSERT_VALUES,SCATTER_FORWARD);CHKERRQ(ierr);
187116909a7fSStefano Zampini       ierr = VecScatterEnd(pcis->N_to_D,pcis->vec2_N,pcis->vec3_D,INSERT_VALUES,SCATTER_FORWARD);CHKERRQ(ierr);
187216909a7fSStefano Zampini     } else {
18730c7d97c5SJed Brown       ierr = MatMult(pcis->A_IB,pcis->vec1_B,pcis->vec3_D);CHKERRQ(ierr);
187416909a7fSStefano Zampini     }
187516909a7fSStefano Zampini   } else if (pcbddc->switch_static) { /* n_B is zero */
187616909a7fSStefano Zampini     Mat_IS *matis = (Mat_IS*)(pc->mat->data);
187716909a7fSStefano Zampini 
187816909a7fSStefano Zampini     if (!pcbddc->switch_static_change) {
187916909a7fSStefano Zampini       ierr = MatMult(matis->A,pcis->vec1_D,pcis->vec3_D);CHKERRQ(ierr);
188016909a7fSStefano Zampini     } else {
188116909a7fSStefano Zampini       ierr = MatMult(pcbddc->switch_static_change,pcis->vec1_D,pcis->vec1_N);CHKERRQ(ierr);
188216909a7fSStefano Zampini       ierr = MatMult(matis->A,pcis->vec1_N,pcis->vec2_N);CHKERRQ(ierr);
188316909a7fSStefano Zampini       ierr = MatMultTranspose(pcbddc->switch_static_change,pcis->vec2_N,pcis->vec3_D);CHKERRQ(ierr);
188416909a7fSStefano Zampini     }
1885b097fa66SStefano Zampini   }
1886df187020SStefano Zampini   ierr = KSPSolve(pcbddc->ksp_D,pcis->vec3_D,pcis->vec4_D);CHKERRQ(ierr);
1887efc2fbd9SStefano Zampini 
18888ae0ca82SStefano Zampini   if (!pcbddc->exact_dirichlet_trick_app && !pcbddc->benign_apply_coarse_only) {
1889b097fa66SStefano Zampini     if (pcbddc->switch_static) {
1890b097fa66SStefano Zampini       ierr = VecAXPBYPCZ(pcis->vec2_D,m_one,one,m_one,pcis->vec4_D,pcis->vec1_D);CHKERRQ(ierr);
1891b097fa66SStefano Zampini     } else {
1892b097fa66SStefano Zampini       ierr = VecAXPBY(pcis->vec2_D,m_one,m_one,pcis->vec4_D);CHKERRQ(ierr);
1893b097fa66SStefano Zampini     }
18940c7d97c5SJed Brown     ierr = VecScatterBegin(pcis->global_to_D,pcis->vec2_D,z,INSERT_VALUES,SCATTER_REVERSE);CHKERRQ(ierr);
18950c7d97c5SJed Brown     ierr = VecScatterEnd(pcis->global_to_D,pcis->vec2_D,z,INSERT_VALUES,SCATTER_REVERSE);CHKERRQ(ierr);
1896b097fa66SStefano Zampini   } else {
1897b097fa66SStefano Zampini     if (pcbddc->switch_static) {
1898b097fa66SStefano Zampini       ierr = VecAXPBY(pcis->vec4_D,one,m_one,pcis->vec1_D);CHKERRQ(ierr);
1899b097fa66SStefano Zampini     } else {
1900b097fa66SStefano Zampini       ierr = VecScale(pcis->vec4_D,m_one);CHKERRQ(ierr);
1901b097fa66SStefano Zampini     }
1902b097fa66SStefano Zampini     ierr = VecScatterBegin(pcis->global_to_D,pcis->vec4_D,z,INSERT_VALUES,SCATTER_REVERSE);CHKERRQ(ierr);
1903b097fa66SStefano Zampini     ierr = VecScatterEnd(pcis->global_to_D,pcis->vec4_D,z,INSERT_VALUES,SCATTER_REVERSE);CHKERRQ(ierr);
1904b097fa66SStefano Zampini   }
190527b6a85dSStefano Zampini   if (pcbddc->benign_have_null) { /* set p0 (computed in PCBDDCApplyInterface) */
19061dd7afcfSStefano Zampini     if (pcbddc->benign_apply_coarse_only) {
19071dd7afcfSStefano Zampini       ierr = PetscMemzero(pcbddc->benign_p0,pcbddc->benign_n*sizeof(PetscScalar));CHKERRQ(ierr);
19081dd7afcfSStefano Zampini     }
1909015636ebSStefano Zampini     ierr = PCBDDCBenignGetOrSetP0(pc,z,PETSC_FALSE);CHKERRQ(ierr);
1910efc2fbd9SStefano Zampini   }
19111f4df5f7SStefano Zampini 
19121dd7afcfSStefano Zampini   if (pcbddc->ChangeOfBasisMatrix) {
1913f913dca9SStefano Zampini     pcbddc->work_change = r;
19141dd7afcfSStefano Zampini     ierr = VecCopy(z,pcbddc->work_change);CHKERRQ(ierr);
19151dd7afcfSStefano Zampini     ierr = MatMult(pcbddc->ChangeOfBasisMatrix,pcbddc->work_change,z);CHKERRQ(ierr);
19161dd7afcfSStefano Zampini   }
19170c7d97c5SJed Brown   PetscFunctionReturn(0);
19180c7d97c5SJed Brown }
191950efa1b5SStefano Zampini 
192050efa1b5SStefano Zampini /*
192150efa1b5SStefano Zampini    PCApplyTranspose_BDDC - Applies the transpose of the BDDC operator to a vector.
192250efa1b5SStefano Zampini 
192350efa1b5SStefano Zampini    Input Parameters:
19240f202f7eSStefano Zampini +  pc - the preconditioner context
19250f202f7eSStefano Zampini -  r - input vector (global)
192650efa1b5SStefano Zampini 
192750efa1b5SStefano Zampini    Output Parameter:
192850efa1b5SStefano Zampini .  z - output vector (global)
192950efa1b5SStefano Zampini 
193050efa1b5SStefano Zampini    Application Interface Routine: PCApplyTranspose()
193150efa1b5SStefano Zampini  */
193250efa1b5SStefano Zampini PetscErrorCode PCApplyTranspose_BDDC(PC pc,Vec r,Vec z)
193350efa1b5SStefano Zampini {
193450efa1b5SStefano Zampini   PC_IS             *pcis = (PC_IS*)(pc->data);
193550efa1b5SStefano Zampini   PC_BDDC           *pcbddc = (PC_BDDC*)(pc->data);
1936b097fa66SStefano Zampini   PetscInt          n_B = pcis->n_B, n_D = pcis->n - n_B;
193750efa1b5SStefano Zampini   PetscErrorCode    ierr;
193850efa1b5SStefano Zampini   const PetscScalar one = 1.0;
193950efa1b5SStefano Zampini   const PetscScalar m_one = -1.0;
194050efa1b5SStefano Zampini   const PetscScalar zero = 0.0;
194150efa1b5SStefano Zampini 
194250efa1b5SStefano Zampini   PetscFunctionBegin;
1943f3d41395Sstefano_zampini   ierr = PetscCitationsRegister(citation,&cited);CHKERRQ(ierr);
19441dd7afcfSStefano Zampini   if (pcbddc->ChangeOfBasisMatrix) {
19451dd7afcfSStefano Zampini     Vec swap;
194627b6a85dSStefano Zampini 
194727b6a85dSStefano Zampini     ierr = MatMultTranspose(pcbddc->ChangeOfBasisMatrix,r,pcbddc->work_change);CHKERRQ(ierr);
19481dd7afcfSStefano Zampini     swap = pcbddc->work_change;
19491dd7afcfSStefano Zampini     pcbddc->work_change = r;
19501dd7afcfSStefano Zampini     r = swap;
195127b6a85dSStefano Zampini     /* save rhs so that we don't need to apply the change of basis for the exact dirichlet trick in PreSolve */
19528ae0ca82SStefano Zampini     if (pcbddc->benign_apply_coarse_only && pcbddc->exact_dirichlet_trick_app && pcbddc->change_interior) {
195327b6a85dSStefano Zampini       ierr = VecCopy(r,pcis->vec1_global);CHKERRQ(ierr);
195427b6a85dSStefano Zampini       ierr = VecLockPush(pcis->vec1_global);CHKERRQ(ierr);
19551dd7afcfSStefano Zampini     }
195627b6a85dSStefano Zampini   }
195727b6a85dSStefano Zampini   if (pcbddc->benign_have_null) { /* get p0 from r */
1958537c1cdfSStefano Zampini     ierr = PCBDDCBenignGetOrSetP0(pc,r,PETSC_TRUE);CHKERRQ(ierr);
1959537c1cdfSStefano Zampini   }
19608ae0ca82SStefano Zampini   if (!pcbddc->exact_dirichlet_trick_app && !pcbddc->benign_apply_coarse_only) {
1961b097fa66SStefano Zampini     ierr = VecCopy(r,z);CHKERRQ(ierr);
196250efa1b5SStefano Zampini     /* First Dirichlet solve */
196350efa1b5SStefano Zampini     ierr = VecScatterBegin(pcis->global_to_D,r,pcis->vec1_D,INSERT_VALUES,SCATTER_FORWARD);CHKERRQ(ierr);
196450efa1b5SStefano Zampini     ierr = VecScatterEnd(pcis->global_to_D,r,pcis->vec1_D,INSERT_VALUES,SCATTER_FORWARD);CHKERRQ(ierr);
196550efa1b5SStefano Zampini     /*
196650efa1b5SStefano Zampini       Assembling right hand side for BDDC operator
1967b097fa66SStefano Zampini       - pcis->vec1_D for the Dirichlet part (if needed, i.e. pcbddc->switch_static == PETSC_TRUE)
196850efa1b5SStefano Zampini       - pcis->vec1_B the interface part of the global vector z
196950efa1b5SStefano Zampini     */
1970b097fa66SStefano Zampini     if (n_D) {
1971b097fa66SStefano Zampini       ierr = KSPSolveTranspose(pcbddc->ksp_D,pcis->vec1_D,pcis->vec2_D);CHKERRQ(ierr);
197250efa1b5SStefano Zampini       ierr = VecScale(pcis->vec2_D,m_one);CHKERRQ(ierr);
197316909a7fSStefano Zampini       if (pcbddc->switch_static) {
197416909a7fSStefano Zampini         Mat_IS *matis = (Mat_IS*)(pc->mat->data);
197516909a7fSStefano Zampini 
197616909a7fSStefano Zampini         ierr = VecSet(pcis->vec1_N,0.);CHKERRQ(ierr);
197716909a7fSStefano Zampini         ierr = VecScatterBegin(pcis->N_to_D,pcis->vec2_D,pcis->vec1_N,INSERT_VALUES,SCATTER_REVERSE);CHKERRQ(ierr);
197816909a7fSStefano Zampini         ierr = VecScatterEnd(pcis->N_to_D,pcis->vec2_D,pcis->vec1_N,INSERT_VALUES,SCATTER_REVERSE);CHKERRQ(ierr);
197916909a7fSStefano Zampini         if (!pcbddc->switch_static_change) {
198016909a7fSStefano Zampini           ierr = MatMultTranspose(matis->A,pcis->vec1_N,pcis->vec2_N);CHKERRQ(ierr);
198116909a7fSStefano Zampini         } else {
198216909a7fSStefano Zampini           ierr = MatMult(pcbddc->switch_static_change,pcis->vec1_N,pcis->vec2_N);CHKERRQ(ierr);
198316909a7fSStefano Zampini           ierr = MatMultTranspose(matis->A,pcis->vec2_N,pcis->vec1_N);CHKERRQ(ierr);
198416909a7fSStefano Zampini           ierr = MatMultTranspose(pcbddc->switch_static_change,pcis->vec1_N,pcis->vec2_N);CHKERRQ(ierr);
198516909a7fSStefano Zampini         }
198616909a7fSStefano Zampini         ierr = VecScatterBegin(pcis->N_to_D,pcis->vec2_N,pcis->vec1_D,ADD_VALUES,SCATTER_FORWARD);CHKERRQ(ierr);
198716909a7fSStefano Zampini         ierr = VecScatterEnd(pcis->N_to_D,pcis->vec2_N,pcis->vec1_D,ADD_VALUES,SCATTER_FORWARD);CHKERRQ(ierr);
198816909a7fSStefano Zampini         ierr = VecScatterBegin(pcis->N_to_B,pcis->vec2_N,pcis->vec1_B,INSERT_VALUES,SCATTER_FORWARD);CHKERRQ(ierr);
198916909a7fSStefano Zampini         ierr = VecScatterEnd(pcis->N_to_B,pcis->vec2_N,pcis->vec1_B,INSERT_VALUES,SCATTER_FORWARD);CHKERRQ(ierr);
199016909a7fSStefano Zampini       } else {
1991b097fa66SStefano Zampini         ierr = MatMultTranspose(pcis->A_IB,pcis->vec2_D,pcis->vec1_B);CHKERRQ(ierr);
199216909a7fSStefano Zampini       }
1993b097fa66SStefano Zampini     } else {
1994b097fa66SStefano Zampini       ierr = VecSet(pcis->vec1_B,zero);CHKERRQ(ierr);
1995b097fa66SStefano Zampini     }
199650efa1b5SStefano Zampini     ierr = VecScatterBegin(pcis->global_to_B,pcis->vec1_B,z,ADD_VALUES,SCATTER_REVERSE);CHKERRQ(ierr);
199750efa1b5SStefano Zampini     ierr = VecScatterEnd(pcis->global_to_B,pcis->vec1_B,z,ADD_VALUES,SCATTER_REVERSE);CHKERRQ(ierr);
199850efa1b5SStefano Zampini     ierr = PCBDDCScalingRestriction(pc,z,pcis->vec1_B);CHKERRQ(ierr);
199950efa1b5SStefano Zampini   } else {
200050efa1b5SStefano Zampini     ierr = PCBDDCScalingRestriction(pc,r,pcis->vec1_B);CHKERRQ(ierr);
200150efa1b5SStefano Zampini   }
200250efa1b5SStefano Zampini 
200350efa1b5SStefano Zampini   /* Apply interface preconditioner
200450efa1b5SStefano Zampini      input/output vecs: pcis->vec1_B and pcis->vec1_D */
2005dc359a40SStefano Zampini   ierr = PCBDDCApplyInterfacePreconditioner(pc,PETSC_TRUE);CHKERRQ(ierr);
200650efa1b5SStefano Zampini 
200750efa1b5SStefano Zampini   /* Apply transpose of partition of unity operator */
200850efa1b5SStefano Zampini   ierr = PCBDDCScalingExtension(pc,pcis->vec1_B,z);CHKERRQ(ierr);
200950efa1b5SStefano Zampini 
201050efa1b5SStefano Zampini   /* Second Dirichlet solve and assembling of output */
201150efa1b5SStefano Zampini   ierr = VecScatterBegin(pcis->global_to_B,z,pcis->vec1_B,INSERT_VALUES,SCATTER_FORWARD);CHKERRQ(ierr);
201250efa1b5SStefano Zampini   ierr = VecScatterEnd(pcis->global_to_B,z,pcis->vec1_B,INSERT_VALUES,SCATTER_FORWARD);CHKERRQ(ierr);
2013b097fa66SStefano Zampini   if (n_B) {
201416909a7fSStefano Zampini     if (pcbddc->switch_static) {
201516909a7fSStefano Zampini       Mat_IS *matis = (Mat_IS*)(pc->mat->data);
201616909a7fSStefano Zampini 
201716909a7fSStefano Zampini       ierr = VecScatterBegin(pcis->N_to_D,pcis->vec1_D,pcis->vec1_N,INSERT_VALUES,SCATTER_REVERSE);CHKERRQ(ierr);
201816909a7fSStefano Zampini       ierr = VecScatterEnd(pcis->N_to_D,pcis->vec1_D,pcis->vec1_N,INSERT_VALUES,SCATTER_REVERSE);CHKERRQ(ierr);
201916909a7fSStefano Zampini       ierr = VecScatterBegin(pcis->N_to_B,pcis->vec1_B,pcis->vec1_N,INSERT_VALUES,SCATTER_REVERSE);CHKERRQ(ierr);
202016909a7fSStefano Zampini       ierr = VecScatterEnd(pcis->N_to_B,pcis->vec1_B,pcis->vec1_N,INSERT_VALUES,SCATTER_REVERSE);CHKERRQ(ierr);
202116909a7fSStefano Zampini       if (!pcbddc->switch_static_change) {
202216909a7fSStefano Zampini         ierr = MatMultTranspose(matis->A,pcis->vec1_N,pcis->vec2_N);CHKERRQ(ierr);
202316909a7fSStefano Zampini       } else {
202416909a7fSStefano Zampini         ierr = MatMult(pcbddc->switch_static_change,pcis->vec1_N,pcis->vec2_N);CHKERRQ(ierr);
202516909a7fSStefano Zampini         ierr = MatMultTranspose(matis->A,pcis->vec2_N,pcis->vec1_N);CHKERRQ(ierr);
202616909a7fSStefano Zampini         ierr = MatMultTranspose(pcbddc->switch_static_change,pcis->vec1_N,pcis->vec2_N);CHKERRQ(ierr);
202716909a7fSStefano Zampini       }
202816909a7fSStefano Zampini       ierr = VecScatterBegin(pcis->N_to_D,pcis->vec2_N,pcis->vec3_D,INSERT_VALUES,SCATTER_FORWARD);CHKERRQ(ierr);
202916909a7fSStefano Zampini       ierr = VecScatterEnd(pcis->N_to_D,pcis->vec2_N,pcis->vec3_D,INSERT_VALUES,SCATTER_FORWARD);CHKERRQ(ierr);
203016909a7fSStefano Zampini     } else {
203150efa1b5SStefano Zampini       ierr = MatMultTranspose(pcis->A_BI,pcis->vec1_B,pcis->vec3_D);CHKERRQ(ierr);
203216909a7fSStefano Zampini     }
203316909a7fSStefano Zampini   } else if (pcbddc->switch_static) { /* n_B is zero */
203416909a7fSStefano Zampini     Mat_IS *matis = (Mat_IS*)(pc->mat->data);
203516909a7fSStefano Zampini 
203616909a7fSStefano Zampini     if (!pcbddc->switch_static_change) {
203716909a7fSStefano Zampini       ierr = MatMultTranspose(matis->A,pcis->vec1_D,pcis->vec3_D);CHKERRQ(ierr);
203816909a7fSStefano Zampini     } else {
203916909a7fSStefano Zampini       ierr = MatMult(pcbddc->switch_static_change,pcis->vec1_D,pcis->vec1_N);CHKERRQ(ierr);
204016909a7fSStefano Zampini       ierr = MatMultTranspose(matis->A,pcis->vec1_N,pcis->vec2_N);CHKERRQ(ierr);
204116909a7fSStefano Zampini       ierr = MatMultTranspose(pcbddc->switch_static_change,pcis->vec2_N,pcis->vec3_D);CHKERRQ(ierr);
204216909a7fSStefano Zampini     }
2043b097fa66SStefano Zampini   }
2044b0147a47SStefano Zampini   ierr = KSPSolveTranspose(pcbddc->ksp_D,pcis->vec3_D,pcis->vec4_D);CHKERRQ(ierr);
20458ae0ca82SStefano Zampini   if (!pcbddc->exact_dirichlet_trick_app && !pcbddc->benign_apply_coarse_only) {
2046b097fa66SStefano Zampini     if (pcbddc->switch_static) {
2047b097fa66SStefano Zampini       ierr = VecAXPBYPCZ(pcis->vec2_D,m_one,one,m_one,pcis->vec4_D,pcis->vec1_D);CHKERRQ(ierr);
2048b097fa66SStefano Zampini     } else {
2049b097fa66SStefano Zampini       ierr = VecAXPBY(pcis->vec2_D,m_one,m_one,pcis->vec4_D);CHKERRQ(ierr);
2050b097fa66SStefano Zampini     }
205150efa1b5SStefano Zampini     ierr = VecScatterBegin(pcis->global_to_D,pcis->vec2_D,z,INSERT_VALUES,SCATTER_REVERSE);CHKERRQ(ierr);
205250efa1b5SStefano Zampini     ierr = VecScatterEnd(pcis->global_to_D,pcis->vec2_D,z,INSERT_VALUES,SCATTER_REVERSE);CHKERRQ(ierr);
2053b097fa66SStefano Zampini   } else {
2054b097fa66SStefano Zampini     if (pcbddc->switch_static) {
2055b097fa66SStefano Zampini       ierr = VecAXPBY(pcis->vec4_D,one,m_one,pcis->vec1_D);CHKERRQ(ierr);
2056b097fa66SStefano Zampini     } else {
2057b097fa66SStefano Zampini       ierr = VecScale(pcis->vec4_D,m_one);CHKERRQ(ierr);
2058b097fa66SStefano Zampini     }
2059b097fa66SStefano Zampini     ierr = VecScatterBegin(pcis->global_to_D,pcis->vec4_D,z,INSERT_VALUES,SCATTER_REVERSE);CHKERRQ(ierr);
2060b097fa66SStefano Zampini     ierr = VecScatterEnd(pcis->global_to_D,pcis->vec4_D,z,INSERT_VALUES,SCATTER_REVERSE);CHKERRQ(ierr);
2061b097fa66SStefano Zampini   }
206227b6a85dSStefano Zampini   if (pcbddc->benign_have_null) { /* set p0 (computed in PCBDDCApplyInterface) */
2063537c1cdfSStefano Zampini     ierr = PCBDDCBenignGetOrSetP0(pc,z,PETSC_FALSE);CHKERRQ(ierr);
2064537c1cdfSStefano Zampini   }
20651dd7afcfSStefano Zampini   if (pcbddc->ChangeOfBasisMatrix) {
2066f913dca9SStefano Zampini     pcbddc->work_change = r;
20671dd7afcfSStefano Zampini     ierr = VecCopy(z,pcbddc->work_change);CHKERRQ(ierr);
20681dd7afcfSStefano Zampini     ierr = MatMult(pcbddc->ChangeOfBasisMatrix,pcbddc->work_change,z);CHKERRQ(ierr);
20691dd7afcfSStefano Zampini   }
207050efa1b5SStefano Zampini   PetscFunctionReturn(0);
207150efa1b5SStefano Zampini }
2072674ae819SStefano Zampini 
20739326c5c6Sstefano_zampini PetscErrorCode PCReset_BDDC(PC pc)
2074da1bb401SStefano Zampini {
2075da1bb401SStefano Zampini   PC_BDDC        *pcbddc = (PC_BDDC*)pc->data;
20769326c5c6Sstefano_zampini   PC_IS          *pcis = (PC_IS*)pc->data;
20779326c5c6Sstefano_zampini   KSP            kspD,kspR,kspC;
2078da1bb401SStefano Zampini   PetscErrorCode ierr;
2079da1bb401SStefano Zampini 
2080da1bb401SStefano Zampini   PetscFunctionBegin;
2081674ae819SStefano Zampini   /* free BDDC custom data  */
2082674ae819SStefano Zampini   ierr = PCBDDCResetCustomization(pc);CHKERRQ(ierr);
2083674ae819SStefano Zampini   /* destroy objects related to topography */
2084674ae819SStefano Zampini   ierr = PCBDDCResetTopography(pc);CHKERRQ(ierr);
208534a97f8cSStefano Zampini   /* destroy objects for scaling operator */
2086674ae819SStefano Zampini   ierr = PCBDDCScalingDestroy(pc);CHKERRQ(ierr);
2087674ae819SStefano Zampini   /* free solvers stuff */
2088674ae819SStefano Zampini   ierr = PCBDDCResetSolvers(pc);CHKERRQ(ierr);
208962a6ff1dSStefano Zampini   /* free global vectors needed in presolve */
209062a6ff1dSStefano Zampini   ierr = VecDestroy(&pcbddc->temp_solution);CHKERRQ(ierr);
209162a6ff1dSStefano Zampini   ierr = VecDestroy(&pcbddc->original_rhs);CHKERRQ(ierr);
20921dd7afcfSStefano Zampini   /* free data created by PCIS */
20931dd7afcfSStefano Zampini   ierr = PCISDestroy(pc);CHKERRQ(ierr);
20949326c5c6Sstefano_zampini 
20959326c5c6Sstefano_zampini   /* restore defaults */
20969326c5c6Sstefano_zampini   kspD = pcbddc->ksp_D;
20979326c5c6Sstefano_zampini   kspR = pcbddc->ksp_R;
20989326c5c6Sstefano_zampini   kspC = pcbddc->coarse_ksp;
20999326c5c6Sstefano_zampini   ierr = PetscMemzero(pc->data,sizeof(*pcbddc));CHKERRQ(ierr);
21009326c5c6Sstefano_zampini   pcis->n_neigh                     = -1;
21019326c5c6Sstefano_zampini   pcis->scaling_factor              = 1.0;
21029326c5c6Sstefano_zampini   pcis->reusesubmatrices            = PETSC_TRUE;
21039326c5c6Sstefano_zampini   pcbddc->use_local_adj             = PETSC_TRUE;
21049326c5c6Sstefano_zampini   pcbddc->use_vertices              = PETSC_TRUE;
21059326c5c6Sstefano_zampini   pcbddc->use_edges                 = PETSC_TRUE;
21069326c5c6Sstefano_zampini   pcbddc->symmetric_primal          = PETSC_TRUE;
21079326c5c6Sstefano_zampini   pcbddc->vertex_size               = 1;
21089326c5c6Sstefano_zampini   pcbddc->recompute_topography      = PETSC_TRUE;
21099326c5c6Sstefano_zampini   pcbddc->coarse_size               = -1;
21109326c5c6Sstefano_zampini   pcbddc->use_exact_dirichlet_trick = PETSC_TRUE;
21119326c5c6Sstefano_zampini   pcbddc->coarsening_ratio          = 8;
21129326c5c6Sstefano_zampini   pcbddc->coarse_eqs_per_proc       = 1;
21139326c5c6Sstefano_zampini   pcbddc->benign_compute_correction = PETSC_TRUE;
21149326c5c6Sstefano_zampini   pcbddc->nedfield                  = -1;
21159326c5c6Sstefano_zampini   pcbddc->nedglobal                 = PETSC_TRUE;
21169326c5c6Sstefano_zampini   pcbddc->graphmaxcount             = PETSC_MAX_INT;
21179326c5c6Sstefano_zampini   pcbddc->sub_schurs_layers         = -1;
21189326c5c6Sstefano_zampini   pcbddc->ksp_D                     = kspD;
21199326c5c6Sstefano_zampini   pcbddc->ksp_R                     = kspR;
21209326c5c6Sstefano_zampini   pcbddc->coarse_ksp                = kspC;
21219326c5c6Sstefano_zampini   PetscFunctionReturn(0);
21229326c5c6Sstefano_zampini }
21239326c5c6Sstefano_zampini 
21249326c5c6Sstefano_zampini PetscErrorCode PCDestroy_BDDC(PC pc)
21259326c5c6Sstefano_zampini {
21269326c5c6Sstefano_zampini   PC_BDDC        *pcbddc = (PC_BDDC*)pc->data;
21279326c5c6Sstefano_zampini   PetscErrorCode ierr;
21289326c5c6Sstefano_zampini 
21299326c5c6Sstefano_zampini   PetscFunctionBegin;
21309326c5c6Sstefano_zampini   ierr = PCReset_BDDC(pc);CHKERRQ(ierr);
21319326c5c6Sstefano_zampini   ierr = KSPDestroy(&pcbddc->ksp_D);CHKERRQ(ierr);
21329326c5c6Sstefano_zampini   ierr = KSPDestroy(&pcbddc->ksp_R);CHKERRQ(ierr);
21339326c5c6Sstefano_zampini   ierr = KSPDestroy(&pcbddc->coarse_ksp);CHKERRQ(ierr);
2134a13144ffSStefano Zampini   ierr = PetscObjectComposeFunction((PetscObject)pc,"PCBDDCSetDiscreteGradient_C",NULL);CHKERRQ(ierr);
2135a198735bSStefano Zampini   ierr = PetscObjectComposeFunction((PetscObject)pc,"PCBDDCSetDivergenceMat_C",NULL);CHKERRQ(ierr);
2136906d46d4SStefano Zampini   ierr = PetscObjectComposeFunction((PetscObject)pc,"PCBDDCSetChangeOfBasisMat_C",NULL);CHKERRQ(ierr);
2137674ae819SStefano Zampini   ierr = PetscObjectComposeFunction((PetscObject)pc,"PCBDDCSetPrimalVerticesLocalIS_C",NULL);CHKERRQ(ierr);
213830368db7SStefano Zampini   ierr = PetscObjectComposeFunction((PetscObject)pc,"PCBDDCSetPrimalVerticesIS_C",NULL);CHKERRQ(ierr);
2139bdf89e91SBarry Smith   ierr = PetscObjectComposeFunction((PetscObject)pc,"PCBDDCSetCoarseningRatio_C",NULL);CHKERRQ(ierr);
21402b510759SStefano Zampini   ierr = PetscObjectComposeFunction((PetscObject)pc,"PCBDDCSetLevel_C",NULL);CHKERRQ(ierr);
2141b8ffe317SStefano Zampini   ierr = PetscObjectComposeFunction((PetscObject)pc,"PCBDDCSetUseExactDirichlet_C",NULL);CHKERRQ(ierr);
21422b510759SStefano Zampini   ierr = PetscObjectComposeFunction((PetscObject)pc,"PCBDDCSetLevels_C",NULL);CHKERRQ(ierr);
2143bdf89e91SBarry Smith   ierr = PetscObjectComposeFunction((PetscObject)pc,"PCBDDCSetDirichletBoundaries_C",NULL);CHKERRQ(ierr);
214482ba6b80SStefano Zampini   ierr = PetscObjectComposeFunction((PetscObject)pc,"PCBDDCSetDirichletBoundariesLocal_C",NULL);CHKERRQ(ierr);
2145bdf89e91SBarry Smith   ierr = PetscObjectComposeFunction((PetscObject)pc,"PCBDDCSetNeumannBoundaries_C",NULL);CHKERRQ(ierr);
214682ba6b80SStefano Zampini   ierr = PetscObjectComposeFunction((PetscObject)pc,"PCBDDCSetNeumannBoundariesLocal_C",NULL);CHKERRQ(ierr);
2147bdf89e91SBarry Smith   ierr = PetscObjectComposeFunction((PetscObject)pc,"PCBDDCGetDirichletBoundaries_C",NULL);CHKERRQ(ierr);
214882ba6b80SStefano Zampini   ierr = PetscObjectComposeFunction((PetscObject)pc,"PCBDDCGetDirichletBoundariesLocal_C",NULL);CHKERRQ(ierr);
2149bdf89e91SBarry Smith   ierr = PetscObjectComposeFunction((PetscObject)pc,"PCBDDCGetNeumannBoundaries_C",NULL);CHKERRQ(ierr);
2150785d1243SStefano Zampini   ierr = PetscObjectComposeFunction((PetscObject)pc,"PCBDDCGetNeumannBoundariesLocal_C",NULL);CHKERRQ(ierr);
2151bdf89e91SBarry Smith   ierr = PetscObjectComposeFunction((PetscObject)pc,"PCBDDCSetDofsSplitting_C",NULL);CHKERRQ(ierr);
215263602bcaSStefano Zampini   ierr = PetscObjectComposeFunction((PetscObject)pc,"PCBDDCSetDofsSplittingLocal_C",NULL);CHKERRQ(ierr);
2153bdf89e91SBarry Smith   ierr = PetscObjectComposeFunction((PetscObject)pc,"PCBDDCSetLocalAdjacencyGraph_C",NULL);CHKERRQ(ierr);
2154bdf89e91SBarry Smith   ierr = PetscObjectComposeFunction((PetscObject)pc,"PCBDDCCreateFETIDPOperators_C",NULL);CHKERRQ(ierr);
2155bdf89e91SBarry Smith   ierr = PetscObjectComposeFunction((PetscObject)pc,"PCBDDCMatFETIDPGetRHS_C",NULL);CHKERRQ(ierr);
2156bdf89e91SBarry Smith   ierr = PetscObjectComposeFunction((PetscObject)pc,"PCBDDCMatFETIDPGetSolution_C",NULL);CHKERRQ(ierr);
2157a06fd7f2SStefano Zampini   ierr = PetscObjectComposeFunction((PetscObject)pc,"PCPreSolveChangeRHS_C",NULL);CHKERRQ(ierr);
2158674ae819SStefano Zampini   ierr = PetscFree(pc->data);CHKERRQ(ierr);
2159da1bb401SStefano Zampini   PetscFunctionReturn(0);
2160da1bb401SStefano Zampini }
21611e6b0712SBarry Smith 
2162a06fd7f2SStefano Zampini static PetscErrorCode PCPreSolveChangeRHS_BDDC(PC pc, PetscBool* change)
2163a06fd7f2SStefano Zampini {
2164a06fd7f2SStefano Zampini   PetscFunctionBegin;
2165a06fd7f2SStefano Zampini   *change = PETSC_TRUE;
2166a06fd7f2SStefano Zampini   PetscFunctionReturn(0);
2167a06fd7f2SStefano Zampini }
2168a06fd7f2SStefano Zampini 
21693425bc38SStefano Zampini static PetscErrorCode PCBDDCMatFETIDPGetRHS_BDDC(Mat fetidp_mat, Vec standard_rhs, Vec fetidp_flux_rhs)
21703425bc38SStefano Zampini {
2171674ae819SStefano Zampini   FETIDPMat_ctx  mat_ctx;
2172266e20e9SStefano Zampini   Vec            work;
21733425bc38SStefano Zampini   PC_IS*         pcis;
21743425bc38SStefano Zampini   PC_BDDC*       pcbddc;
21753425bc38SStefano Zampini   PetscErrorCode ierr;
21760c7d97c5SJed Brown 
21773425bc38SStefano Zampini   PetscFunctionBegin;
21783425bc38SStefano Zampini   ierr = MatShellGetContext(fetidp_mat,&mat_ctx);CHKERRQ(ierr);
21793425bc38SStefano Zampini   pcis = (PC_IS*)mat_ctx->pc->data;
21803425bc38SStefano Zampini   pcbddc = (PC_BDDC*)mat_ctx->pc->data;
21813425bc38SStefano Zampini 
2182229984c5Sstefano_zampini   ierr = VecSet(fetidp_flux_rhs,0.0);CHKERRQ(ierr);
2183229984c5Sstefano_zampini   /* copy rhs since we may change it during PCPreSolve_BDDC */
2184229984c5Sstefano_zampini   if (!pcbddc->original_rhs) {
2185229984c5Sstefano_zampini     ierr = VecDuplicate(pcis->vec1_global,&pcbddc->original_rhs);CHKERRQ(ierr);
2186229984c5Sstefano_zampini   }
21876cc1294bSstefano_zampini   if (mat_ctx->rhs_flip) {
21886cc1294bSstefano_zampini     ierr = VecPointwiseMult(pcbddc->original_rhs,standard_rhs,mat_ctx->rhs_flip);CHKERRQ(ierr);
21896cc1294bSstefano_zampini   } else {
2190229984c5Sstefano_zampini     ierr = VecCopy(standard_rhs,pcbddc->original_rhs);CHKERRQ(ierr);
21916cc1294bSstefano_zampini   }
2192af140850Sstefano_zampini   if (mat_ctx->g2g_p) {
2193229984c5Sstefano_zampini     /* interface pressure rhs */
2194022d8d2bSstefano_zampini     ierr = VecScatterBegin(mat_ctx->g2g_p,fetidp_flux_rhs,pcbddc->original_rhs,INSERT_VALUES,SCATTER_REVERSE);CHKERRQ(ierr);
2195022d8d2bSstefano_zampini     ierr = VecScatterEnd(mat_ctx->g2g_p,fetidp_flux_rhs,pcbddc->original_rhs,INSERT_VALUES,SCATTER_REVERSE);CHKERRQ(ierr);
2196229984c5Sstefano_zampini     ierr = VecScatterBegin(mat_ctx->g2g_p,standard_rhs,fetidp_flux_rhs,INSERT_VALUES,SCATTER_FORWARD);CHKERRQ(ierr);
2197229984c5Sstefano_zampini     ierr = VecScatterEnd(mat_ctx->g2g_p,standard_rhs,fetidp_flux_rhs,INSERT_VALUES,SCATTER_FORWARD);CHKERRQ(ierr);
21986cc1294bSstefano_zampini     if (!mat_ctx->rhs_flip) {
2199229984c5Sstefano_zampini       ierr = VecScale(fetidp_flux_rhs,-1.);CHKERRQ(ierr);
2200229984c5Sstefano_zampini     }
22016cc1294bSstefano_zampini   }
2202c08af4c6SStefano Zampini   /*
2203c08af4c6SStefano Zampini      change of basis for physical rhs if needed
2204c08af4c6SStefano Zampini      It also changes the rhs in case of dirichlet boundaries
2205c08af4c6SStefano Zampini   */
22063738a8e6SStefano Zampini   ierr = PCPreSolve_BDDC(mat_ctx->pc,NULL,pcbddc->original_rhs,NULL);CHKERRQ(ierr);
2207fc17d649SStefano Zampini   if (pcbddc->ChangeOfBasisMatrix) {
22083738a8e6SStefano Zampini     ierr = MatMultTranspose(pcbddc->ChangeOfBasisMatrix,pcbddc->original_rhs,pcbddc->work_change);CHKERRQ(ierr);
22093738a8e6SStefano Zampini     work = pcbddc->work_change;
2210fc17d649SStefano Zampini    } else {
22113738a8e6SStefano Zampini     work = pcbddc->original_rhs;
2212fc17d649SStefano Zampini   }
22133425bc38SStefano Zampini   /* store vectors for computation of fetidp final solution */
2214266e20e9SStefano Zampini   ierr = VecScatterBegin(pcis->global_to_D,work,mat_ctx->temp_solution_D,INSERT_VALUES,SCATTER_FORWARD);CHKERRQ(ierr);
2215266e20e9SStefano Zampini   ierr = VecScatterEnd(pcis->global_to_D,work,mat_ctx->temp_solution_D,INSERT_VALUES,SCATTER_FORWARD);CHKERRQ(ierr);
2216fb223d50SStefano Zampini   /* scale rhs since it should be unassembled */
2217fb223d50SStefano Zampini   /* TODO use counter scaling? (also below) */
2218266e20e9SStefano Zampini   ierr = VecScatterBegin(pcis->global_to_B,work,mat_ctx->temp_solution_B,INSERT_VALUES,SCATTER_FORWARD);CHKERRQ(ierr);
2219266e20e9SStefano Zampini   ierr = VecScatterEnd(pcis->global_to_B,work,mat_ctx->temp_solution_B,INSERT_VALUES,SCATTER_FORWARD);CHKERRQ(ierr);
2220674ae819SStefano Zampini   /* Apply partition of unity */
22213425bc38SStefano Zampini   ierr = VecPointwiseMult(mat_ctx->temp_solution_B,pcis->D,mat_ctx->temp_solution_B);CHKERRQ(ierr);
2222266e20e9SStefano Zampini   /* ierr = PCBDDCScalingRestriction(mat_ctx->pc,work,mat_ctx->temp_solution_B);CHKERRQ(ierr); */
22238eeda7d8SStefano Zampini   if (!pcbddc->switch_static) {
22243425bc38SStefano Zampini     /* compute partially subassembled Schur complement right-hand side */
22253425bc38SStefano Zampini     ierr = KSPSolve(pcbddc->ksp_D,mat_ctx->temp_solution_D,pcis->vec1_D);CHKERRQ(ierr);
22263425bc38SStefano Zampini     ierr = MatMult(pcis->A_BI,pcis->vec1_D,pcis->vec1_B);CHKERRQ(ierr);
22273425bc38SStefano Zampini     ierr = VecAXPY(mat_ctx->temp_solution_B,-1.0,pcis->vec1_B);CHKERRQ(ierr);
2228266e20e9SStefano Zampini     ierr = VecSet(work,0.0);CHKERRQ(ierr);
2229266e20e9SStefano Zampini     ierr = VecScatterBegin(pcis->global_to_B,mat_ctx->temp_solution_B,work,ADD_VALUES,SCATTER_REVERSE);CHKERRQ(ierr);
2230266e20e9SStefano Zampini     ierr = VecScatterEnd(pcis->global_to_B,mat_ctx->temp_solution_B,work,ADD_VALUES,SCATTER_REVERSE);CHKERRQ(ierr);
2231266e20e9SStefano Zampini     /* ierr = PCBDDCScalingRestriction(mat_ctx->pc,work,mat_ctx->temp_solution_B);CHKERRQ(ierr); */
2232266e20e9SStefano Zampini     ierr = VecScatterBegin(pcis->global_to_B,work,mat_ctx->temp_solution_B,INSERT_VALUES,SCATTER_FORWARD);CHKERRQ(ierr);
2233266e20e9SStefano Zampini     ierr = VecScatterEnd(pcis->global_to_B,work,mat_ctx->temp_solution_B,INSERT_VALUES,SCATTER_FORWARD);CHKERRQ(ierr);
22343425bc38SStefano Zampini     ierr = VecPointwiseMult(mat_ctx->temp_solution_B,pcis->D,mat_ctx->temp_solution_B);CHKERRQ(ierr);
22353425bc38SStefano Zampini   }
22363425bc38SStefano Zampini   /* BDDC rhs */
22373425bc38SStefano Zampini   ierr = VecCopy(mat_ctx->temp_solution_B,pcis->vec1_B);CHKERRQ(ierr);
22388eeda7d8SStefano Zampini   if (pcbddc->switch_static) {
22393425bc38SStefano Zampini     ierr = VecCopy(mat_ctx->temp_solution_D,pcis->vec1_D);CHKERRQ(ierr);
22403425bc38SStefano Zampini   }
22413425bc38SStefano Zampini   /* apply BDDC */
2242229984c5Sstefano_zampini   ierr = PetscMemzero(pcbddc->benign_p0,pcbddc->benign_n*sizeof(PetscScalar));CHKERRQ(ierr);
2243dc359a40SStefano Zampini   ierr = PCBDDCApplyInterfacePreconditioner(mat_ctx->pc,PETSC_FALSE);CHKERRQ(ierr);
2244266e20e9SStefano Zampini   ierr = PetscMemzero(pcbddc->benign_p0,pcbddc->benign_n*sizeof(PetscScalar));CHKERRQ(ierr);
2245229984c5Sstefano_zampini 
22463425bc38SStefano Zampini   /* Application of B_delta and assembling of rhs for fetidp fluxes */
22473425bc38SStefano Zampini   ierr = MatMult(mat_ctx->B_delta,pcis->vec1_B,mat_ctx->lambda_local);CHKERRQ(ierr);
22483425bc38SStefano Zampini   ierr = VecScatterBegin(mat_ctx->l2g_lambda,mat_ctx->lambda_local,fetidp_flux_rhs,ADD_VALUES,SCATTER_FORWARD);CHKERRQ(ierr);
22493425bc38SStefano Zampini   ierr = VecScatterEnd(mat_ctx->l2g_lambda,mat_ctx->lambda_local,fetidp_flux_rhs,ADD_VALUES,SCATTER_FORWARD);CHKERRQ(ierr);
2250229984c5Sstefano_zampini   /* Add contribution to interface pressures */
2251229984c5Sstefano_zampini   if (mat_ctx->l2g_p) {
2252229984c5Sstefano_zampini     ierr = MatMult(mat_ctx->B_BB,pcis->vec1_B,mat_ctx->vP);CHKERRQ(ierr);
2253229984c5Sstefano_zampini     if (pcbddc->switch_static) {
2254229984c5Sstefano_zampini       ierr = MatMultAdd(mat_ctx->B_BI,pcis->vec1_D,mat_ctx->vP,mat_ctx->vP);CHKERRQ(ierr);
2255229984c5Sstefano_zampini     }
2256229984c5Sstefano_zampini     ierr = VecScatterBegin(mat_ctx->l2g_p,mat_ctx->vP,fetidp_flux_rhs,ADD_VALUES,SCATTER_FORWARD);CHKERRQ(ierr);
2257229984c5Sstefano_zampini     ierr = VecScatterEnd(mat_ctx->l2g_p,mat_ctx->vP,fetidp_flux_rhs,ADD_VALUES,SCATTER_FORWARD);CHKERRQ(ierr);
2258229984c5Sstefano_zampini   }
22593425bc38SStefano Zampini   PetscFunctionReturn(0);
22603425bc38SStefano Zampini }
22611e6b0712SBarry Smith 
22623425bc38SStefano Zampini /*@
22630f202f7eSStefano Zampini  PCBDDCMatFETIDPGetRHS - Compute the right-hand side for FETI-DP linear system using the physical right-hand side
22643425bc38SStefano Zampini 
22653425bc38SStefano Zampini    Collective
22663425bc38SStefano Zampini 
22673425bc38SStefano Zampini    Input Parameters:
22680f202f7eSStefano Zampini +  fetidp_mat      - the FETI-DP matrix object obtained by a call to PCBDDCCreateFETIDPOperators
22690f202f7eSStefano Zampini -  standard_rhs    - the right-hand side of the original linear system
22703425bc38SStefano Zampini 
22713425bc38SStefano Zampini    Output Parameters:
22720f202f7eSStefano Zampini .  fetidp_flux_rhs - the right-hand side for the FETI-DP linear system
22733425bc38SStefano Zampini 
22743425bc38SStefano Zampini    Level: developer
22753425bc38SStefano Zampini 
22763425bc38SStefano Zampini    Notes:
22773425bc38SStefano Zampini 
22780f202f7eSStefano Zampini .seealso: PCBDDC, PCBDDCCreateFETIDPOperators, PCBDDCMatFETIDPGetSolution
22793425bc38SStefano Zampini @*/
22803425bc38SStefano Zampini PetscErrorCode PCBDDCMatFETIDPGetRHS(Mat fetidp_mat, Vec standard_rhs, Vec fetidp_flux_rhs)
22813425bc38SStefano Zampini {
2282674ae819SStefano Zampini   FETIDPMat_ctx  mat_ctx;
22833425bc38SStefano Zampini   PetscErrorCode ierr;
22843425bc38SStefano Zampini 
22853425bc38SStefano Zampini   PetscFunctionBegin;
2286266e20e9SStefano Zampini   PetscValidHeaderSpecific(fetidp_mat,MAT_CLASSID,1);
2287266e20e9SStefano Zampini   PetscValidHeaderSpecific(standard_rhs,VEC_CLASSID,2);
2288266e20e9SStefano Zampini   PetscValidHeaderSpecific(fetidp_flux_rhs,VEC_CLASSID,3);
22893425bc38SStefano Zampini   ierr = MatShellGetContext(fetidp_mat,&mat_ctx);CHKERRQ(ierr);
2290163d334eSBarry Smith   ierr = PetscUseMethod(mat_ctx->pc,"PCBDDCMatFETIDPGetRHS_C",(Mat,Vec,Vec),(fetidp_mat,standard_rhs,fetidp_flux_rhs));CHKERRQ(ierr);
22913425bc38SStefano Zampini   PetscFunctionReturn(0);
22923425bc38SStefano Zampini }
22931e6b0712SBarry Smith 
22943425bc38SStefano Zampini static PetscErrorCode PCBDDCMatFETIDPGetSolution_BDDC(Mat fetidp_mat, Vec fetidp_flux_sol, Vec standard_sol)
22953425bc38SStefano Zampini {
2296674ae819SStefano Zampini   FETIDPMat_ctx  mat_ctx;
22973425bc38SStefano Zampini   PC_IS*         pcis;
22983425bc38SStefano Zampini   PC_BDDC*       pcbddc;
22993425bc38SStefano Zampini   PetscErrorCode ierr;
2300229984c5Sstefano_zampini   Vec            work;
23013425bc38SStefano Zampini 
23023425bc38SStefano Zampini   PetscFunctionBegin;
23033425bc38SStefano Zampini   ierr = MatShellGetContext(fetidp_mat,&mat_ctx);CHKERRQ(ierr);
23043425bc38SStefano Zampini   pcis = (PC_IS*)mat_ctx->pc->data;
23053425bc38SStefano Zampini   pcbddc = (PC_BDDC*)mat_ctx->pc->data;
23063425bc38SStefano Zampini 
23073425bc38SStefano Zampini   /* apply B_delta^T */
2308af140850Sstefano_zampini   ierr = VecSet(pcis->vec1_B,0.);CHKERRQ(ierr);
23093425bc38SStefano Zampini   ierr = VecScatterBegin(mat_ctx->l2g_lambda,fetidp_flux_sol,mat_ctx->lambda_local,INSERT_VALUES,SCATTER_REVERSE);CHKERRQ(ierr);
23103425bc38SStefano Zampini   ierr = VecScatterEnd(mat_ctx->l2g_lambda,fetidp_flux_sol,mat_ctx->lambda_local,INSERT_VALUES,SCATTER_REVERSE);CHKERRQ(ierr);
23113425bc38SStefano Zampini   ierr = MatMultTranspose(mat_ctx->B_delta,mat_ctx->lambda_local,pcis->vec1_B);CHKERRQ(ierr);
2312229984c5Sstefano_zampini   if (mat_ctx->l2g_p) {
2313229984c5Sstefano_zampini     ierr = VecScatterBegin(mat_ctx->l2g_p,fetidp_flux_sol,mat_ctx->vP,INSERT_VALUES,SCATTER_REVERSE);CHKERRQ(ierr);
2314229984c5Sstefano_zampini     ierr = VecScatterEnd(mat_ctx->l2g_p,fetidp_flux_sol,mat_ctx->vP,INSERT_VALUES,SCATTER_REVERSE);CHKERRQ(ierr);
2315229984c5Sstefano_zampini     ierr = MatMultAdd(mat_ctx->Bt_BB,mat_ctx->vP,pcis->vec1_B,pcis->vec1_B);CHKERRQ(ierr);
2316229984c5Sstefano_zampini   }
2317229984c5Sstefano_zampini 
23183425bc38SStefano Zampini   /* compute rhs for BDDC application */
23193425bc38SStefano Zampini   ierr = VecAYPX(pcis->vec1_B,-1.0,mat_ctx->temp_solution_B);CHKERRQ(ierr);
23208eeda7d8SStefano Zampini   if (pcbddc->switch_static) {
23213425bc38SStefano Zampini     ierr = VecCopy(mat_ctx->temp_solution_D,pcis->vec1_D);CHKERRQ(ierr);
2322229984c5Sstefano_zampini     if (mat_ctx->l2g_p) {
2323229984c5Sstefano_zampini       ierr = VecScale(mat_ctx->vP,-1.);CHKERRQ(ierr);
2324229984c5Sstefano_zampini       ierr = MatMultAdd(mat_ctx->Bt_BI,mat_ctx->vP,pcis->vec1_D,pcis->vec1_D);CHKERRQ(ierr);
23253425bc38SStefano Zampini     }
2326229984c5Sstefano_zampini   }
2327229984c5Sstefano_zampini 
23283425bc38SStefano Zampini   /* apply BDDC */
2329229984c5Sstefano_zampini   ierr = PetscMemzero(pcbddc->benign_p0,pcbddc->benign_n*sizeof(PetscScalar));CHKERRQ(ierr);
2330dc359a40SStefano Zampini   ierr = PCBDDCApplyInterfacePreconditioner(mat_ctx->pc,PETSC_FALSE);CHKERRQ(ierr);
2331229984c5Sstefano_zampini 
2332229984c5Sstefano_zampini   /* put values into global vector */
2333af140850Sstefano_zampini   if (pcbddc->ChangeOfBasisMatrix) work = pcbddc->work_change;
2334af140850Sstefano_zampini   else work = standard_sol;
2335229984c5Sstefano_zampini   ierr = VecScatterBegin(pcis->global_to_B,pcis->vec1_B,work,INSERT_VALUES,SCATTER_REVERSE);CHKERRQ(ierr);
2336229984c5Sstefano_zampini   ierr = VecScatterEnd(pcis->global_to_B,pcis->vec1_B,work,INSERT_VALUES,SCATTER_REVERSE);CHKERRQ(ierr);
23378eeda7d8SStefano Zampini   if (!pcbddc->switch_static) {
23383425bc38SStefano Zampini     /* compute values into the interior if solved for the partially subassembled Schur complement */
23393425bc38SStefano Zampini     ierr = MatMult(pcis->A_IB,pcis->vec1_B,pcis->vec1_D);CHKERRQ(ierr);
234000f6b531SStefano Zampini     ierr = VecAYPX(pcis->vec1_D,-1.0,mat_ctx->temp_solution_D);CHKERRQ(ierr);
234100f6b531SStefano Zampini     ierr = KSPSolve(pcbddc->ksp_D,pcis->vec1_D,pcis->vec1_D);CHKERRQ(ierr);
23423425bc38SStefano Zampini   }
2343229984c5Sstefano_zampini 
2344229984c5Sstefano_zampini   ierr = VecScatterBegin(pcis->global_to_D,pcis->vec1_D,work,INSERT_VALUES,SCATTER_REVERSE);CHKERRQ(ierr);
2345229984c5Sstefano_zampini   ierr = VecScatterEnd(pcis->global_to_D,pcis->vec1_D,work,INSERT_VALUES,SCATTER_REVERSE);CHKERRQ(ierr);
2346266e20e9SStefano Zampini   /* add p0 solution to final solution */
2347229984c5Sstefano_zampini   ierr = PCBDDCBenignGetOrSetP0(mat_ctx->pc,work,PETSC_FALSE);CHKERRQ(ierr);
2348fc17d649SStefano Zampini   if (pcbddc->ChangeOfBasisMatrix) {
2349af140850Sstefano_zampini     ierr = MatMult(pcbddc->ChangeOfBasisMatrix,work,standard_sol);CHKERRQ(ierr);
2350fc17d649SStefano Zampini   }
2351af140850Sstefano_zampini   ierr = PCPostSolve_BDDC(mat_ctx->pc,NULL,NULL,standard_sol);CHKERRQ(ierr);
2352af140850Sstefano_zampini   if (mat_ctx->g2g_p) {
2353229984c5Sstefano_zampini     ierr = VecScatterBegin(mat_ctx->g2g_p,fetidp_flux_sol,standard_sol,INSERT_VALUES,SCATTER_REVERSE);CHKERRQ(ierr);
2354229984c5Sstefano_zampini     ierr = VecScatterEnd(mat_ctx->g2g_p,fetidp_flux_sol,standard_sol,INSERT_VALUES,SCATTER_REVERSE);CHKERRQ(ierr);
2355229984c5Sstefano_zampini   }
23563425bc38SStefano Zampini   PetscFunctionReturn(0);
23573425bc38SStefano Zampini }
23581e6b0712SBarry Smith 
23593425bc38SStefano Zampini /*@
23600f202f7eSStefano Zampini  PCBDDCMatFETIDPGetSolution - Compute the physical solution using the solution of the FETI-DP linear system
23613425bc38SStefano Zampini 
23623425bc38SStefano Zampini    Collective
23633425bc38SStefano Zampini 
23643425bc38SStefano Zampini    Input Parameters:
23650f202f7eSStefano Zampini +  fetidp_mat      - the FETI-DP matrix obtained by a call to PCBDDCCreateFETIDPOperators
23660f202f7eSStefano Zampini -  fetidp_flux_sol - the solution of the FETI-DP linear system
23673425bc38SStefano Zampini 
23683425bc38SStefano Zampini    Output Parameters:
23690f202f7eSStefano Zampini .  standard_sol    - the solution defined on the physical domain
23703425bc38SStefano Zampini 
23713425bc38SStefano Zampini    Level: developer
23723425bc38SStefano Zampini 
23733425bc38SStefano Zampini    Notes:
23743425bc38SStefano Zampini 
23750f202f7eSStefano Zampini .seealso: PCBDDC, PCBDDCCreateFETIDPOperators, PCBDDCMatFETIDPGetRHS
23763425bc38SStefano Zampini @*/
23773425bc38SStefano Zampini PetscErrorCode PCBDDCMatFETIDPGetSolution(Mat fetidp_mat, Vec fetidp_flux_sol, Vec standard_sol)
23783425bc38SStefano Zampini {
2379674ae819SStefano Zampini   FETIDPMat_ctx  mat_ctx;
23803425bc38SStefano Zampini   PetscErrorCode ierr;
23813425bc38SStefano Zampini 
23823425bc38SStefano Zampini   PetscFunctionBegin;
2383266e20e9SStefano Zampini   PetscValidHeaderSpecific(fetidp_mat,MAT_CLASSID,1);
2384266e20e9SStefano Zampini   PetscValidHeaderSpecific(fetidp_flux_sol,VEC_CLASSID,2);
2385266e20e9SStefano Zampini   PetscValidHeaderSpecific(standard_sol,VEC_CLASSID,3);
23863425bc38SStefano Zampini   ierr = MatShellGetContext(fetidp_mat,&mat_ctx);CHKERRQ(ierr);
2387163d334eSBarry Smith   ierr = PetscUseMethod(mat_ctx->pc,"PCBDDCMatFETIDPGetSolution_C",(Mat,Vec,Vec),(fetidp_mat,fetidp_flux_sol,standard_sol));CHKERRQ(ierr);
23883425bc38SStefano Zampini   PetscFunctionReturn(0);
23893425bc38SStefano Zampini }
23901e6b0712SBarry Smith 
2391547c9a8eSstefano_zampini static PetscErrorCode PCBDDCCreateFETIDPOperators_BDDC(PC pc, PetscBool fully_redundant, const char* prefix, Mat *fetidp_mat, PC *fetidp_pc)
23923425bc38SStefano Zampini {
2393674ae819SStefano Zampini 
2394674ae819SStefano Zampini   FETIDPMat_ctx  fetidpmat_ctx;
23953425bc38SStefano Zampini   Mat            newmat;
2396674ae819SStefano Zampini   FETIDPPC_ctx   fetidppc_ctx;
23973425bc38SStefano Zampini   PC             newpc;
2398ce94432eSBarry Smith   MPI_Comm       comm;
23993425bc38SStefano Zampini   PetscErrorCode ierr;
24003425bc38SStefano Zampini 
24013425bc38SStefano Zampini   PetscFunctionBegin;
2402ce94432eSBarry Smith   ierr = PetscObjectGetComm((PetscObject)pc,&comm);CHKERRQ(ierr);
24033425bc38SStefano Zampini   /* FETIDP linear matrix */
24043425bc38SStefano Zampini   ierr = PCBDDCCreateFETIDPMatContext(pc,&fetidpmat_ctx);CHKERRQ(ierr);
24051720468bSStefano Zampini   fetidpmat_ctx->fully_redundant = fully_redundant;
24063425bc38SStefano Zampini   ierr = PCBDDCSetupFETIDPMatContext(fetidpmat_ctx);CHKERRQ(ierr);
2407a5bb87b3Sstefano_zampini   ierr = MatCreateShell(comm,fetidpmat_ctx->n,fetidpmat_ctx->n,fetidpmat_ctx->N,fetidpmat_ctx->N,fetidpmat_ctx,&newmat);CHKERRQ(ierr);
24083425bc38SStefano Zampini   ierr = MatShellSetOperation(newmat,MATOP_MULT,(void (*)(void))FETIDPMatMult);CHKERRQ(ierr);
2409edf7251bSStefano Zampini   ierr = MatShellSetOperation(newmat,MATOP_MULT_TRANSPOSE,(void (*)(void))FETIDPMatMultTranspose);CHKERRQ(ierr);
24103425bc38SStefano Zampini   ierr = MatShellSetOperation(newmat,MATOP_DESTROY,(void (*)(void))PCBDDCDestroyFETIDPMat);CHKERRQ(ierr);
2411547c9a8eSstefano_zampini   ierr = MatSetOptionsPrefix(newmat,prefix);CHKERRQ(ierr);
2412547c9a8eSstefano_zampini   ierr = MatAppendOptionsPrefix(newmat,"fetidp_");CHKERRQ(ierr);
24133425bc38SStefano Zampini   ierr = MatSetUp(newmat);CHKERRQ(ierr);
24143425bc38SStefano Zampini   /* FETIDP preconditioner */
24153425bc38SStefano Zampini   ierr = PCBDDCCreateFETIDPPCContext(pc,&fetidppc_ctx);CHKERRQ(ierr);
24163425bc38SStefano Zampini   ierr = PCBDDCSetupFETIDPPCContext(newmat,fetidppc_ctx);CHKERRQ(ierr);
24173425bc38SStefano Zampini   ierr = PCCreate(comm,&newpc);CHKERRQ(ierr);
2418e1214c54Sstefano_zampini   ierr = PCSetOperators(newpc,newmat,newmat);CHKERRQ(ierr);
2419e1214c54Sstefano_zampini   ierr = PCSetOptionsPrefix(newpc,prefix);CHKERRQ(ierr);
2420e1214c54Sstefano_zampini   ierr = PCAppendOptionsPrefix(newpc,"fetidp_");CHKERRQ(ierr);
2421e1214c54Sstefano_zampini   if (!fetidpmat_ctx->l2g_lambda_only) {
24223425bc38SStefano Zampini     ierr = PCSetType(newpc,PCSHELL);CHKERRQ(ierr);
24233425bc38SStefano Zampini     ierr = PCShellSetContext(newpc,fetidppc_ctx);CHKERRQ(ierr);
24243425bc38SStefano Zampini     ierr = PCShellSetApply(newpc,FETIDPPCApply);CHKERRQ(ierr);
2425edf7251bSStefano Zampini     ierr = PCShellSetApplyTranspose(newpc,FETIDPPCApplyTranspose);CHKERRQ(ierr);
2426c45b8d2dSstefano_zampini     ierr = PCShellSetView(newpc,FETIDPPCView);CHKERRQ(ierr);
24273425bc38SStefano Zampini     ierr = PCShellSetDestroy(newpc,PCBDDCDestroyFETIDPPC);CHKERRQ(ierr);
2428e1214c54Sstefano_zampini   } else {
2429e1214c54Sstefano_zampini     KSP      *ksps;
2430e1214c54Sstefano_zampini     PC       lagpc;
2431e1214c54Sstefano_zampini     Mat      M,AM,PM;
2432e1214c54Sstefano_zampini     PetscInt nn;
2433e1214c54Sstefano_zampini 
2434e1214c54Sstefano_zampini     ierr = PetscObjectQuery((PetscObject)pc,"__KSPFETIDP_PPmat",(PetscObject*)&M);CHKERRQ(ierr);
2435e1214c54Sstefano_zampini     ierr = PCSetType(newpc,PCFIELDSPLIT);CHKERRQ(ierr);
2436e1214c54Sstefano_zampini     ierr = PCFieldSplitSetIS(newpc,"lag",fetidpmat_ctx->lagrange);CHKERRQ(ierr);
2437e1214c54Sstefano_zampini     ierr = PCFieldSplitSetIS(newpc,"p",fetidpmat_ctx->pressure);CHKERRQ(ierr);
2438e1214c54Sstefano_zampini     ierr = PCFieldSplitSetType(newpc,PC_COMPOSITE_SCHUR);CHKERRQ(ierr);
243940c75d76SStefano Zampini     ierr = PCFieldSplitSetSchurFactType(newpc,PC_FIELDSPLIT_SCHUR_FACT_DIAG);CHKERRQ(ierr);
2440e1214c54Sstefano_zampini     ierr = PCFieldSplitSetSchurPre(newpc,PC_FIELDSPLIT_SCHUR_PRE_USER,M);CHKERRQ(ierr);
2441c096484dSStefano Zampini     ierr = PCFieldSplitSetSchurScale(newpc,1.0);CHKERRQ(ierr);
2442e1214c54Sstefano_zampini     ierr = PCSetFromOptions(newpc);CHKERRQ(ierr);
2443e1214c54Sstefano_zampini     ierr = PCSetUp(newpc);CHKERRQ(ierr);
2444e1214c54Sstefano_zampini 
2445e1214c54Sstefano_zampini     /* set the solver for the (0,0) block */
244640c75d76SStefano Zampini     ierr = PCFieldSplitGetSubKSP(newpc,&nn,&ksps);CHKERRQ(ierr);
2447e1214c54Sstefano_zampini     ierr = PCCreate(comm,&lagpc);CHKERRQ(ierr);
2448e1214c54Sstefano_zampini     ierr = PCSetType(lagpc,PCSHELL);CHKERRQ(ierr);
2449e1214c54Sstefano_zampini     ierr = KSPGetOperators(ksps[0],&AM,&PM);CHKERRQ(ierr);
2450e1214c54Sstefano_zampini     ierr = PCSetOperators(lagpc,AM,PM);CHKERRQ(ierr);
2451e1214c54Sstefano_zampini     ierr = PCShellSetContext(lagpc,fetidppc_ctx);CHKERRQ(ierr);
2452e1214c54Sstefano_zampini     ierr = PCShellSetApply(lagpc,FETIDPPCApply);CHKERRQ(ierr);
2453e1214c54Sstefano_zampini     ierr = PCShellSetApplyTranspose(lagpc,FETIDPPCApplyTranspose);CHKERRQ(ierr);
2454e1214c54Sstefano_zampini     ierr = PCShellSetView(lagpc,FETIDPPCView);CHKERRQ(ierr);
2455e1214c54Sstefano_zampini     ierr = PCShellSetDestroy(lagpc,PCBDDCDestroyFETIDPPC);CHKERRQ(ierr);
2456e1214c54Sstefano_zampini     ierr = KSPSetPC(ksps[0],lagpc);CHKERRQ(ierr);
2457e1214c54Sstefano_zampini     ierr = PCDestroy(&lagpc);CHKERRQ(ierr);
2458e1214c54Sstefano_zampini     ierr = PetscFree(ksps);CHKERRQ(ierr);
2459e1214c54Sstefano_zampini   }
24603425bc38SStefano Zampini   /* return pointers for objects created */
24613425bc38SStefano Zampini   *fetidp_mat = newmat;
24623425bc38SStefano Zampini   *fetidp_pc = newpc;
24633425bc38SStefano Zampini   PetscFunctionReturn(0);
24643425bc38SStefano Zampini }
24651e6b0712SBarry Smith 
246694ef8ddeSSatish Balay /*@C
24670f202f7eSStefano Zampini  PCBDDCCreateFETIDPOperators - Create FETI-DP operators
24683425bc38SStefano Zampini 
24693425bc38SStefano Zampini    Collective
24703425bc38SStefano Zampini 
24713425bc38SStefano Zampini    Input Parameters:
24721720468bSStefano Zampini +  pc - the BDDC preconditioning context (setup should have been called before)
2473547c9a8eSstefano_zampini .  fully_redundant - true for a fully redundant set of Lagrange multipliers
2474547c9a8eSstefano_zampini -  prefix - optional options database prefix for the objects to be created (can be NULL)
247528509bceSStefano Zampini 
247628509bceSStefano Zampini    Output Parameters:
24770f202f7eSStefano Zampini +  fetidp_mat - shell FETI-DP matrix object
24780f202f7eSStefano Zampini -  fetidp_pc  - shell Dirichlet preconditioner for FETI-DP matrix
247928509bceSStefano Zampini 
24803425bc38SStefano Zampini    Level: developer
24813425bc38SStefano Zampini 
24823425bc38SStefano Zampini    Notes:
24830f202f7eSStefano Zampini      Currently the only operations provided for FETI-DP matrix are MatMult and MatMultTranspose
24843425bc38SStefano Zampini 
24850f202f7eSStefano Zampini .seealso: PCBDDC, PCBDDCMatFETIDPGetRHS, PCBDDCMatFETIDPGetSolution
24863425bc38SStefano Zampini @*/
2487547c9a8eSstefano_zampini PetscErrorCode PCBDDCCreateFETIDPOperators(PC pc, PetscBool fully_redundant, const char *prefix, Mat *fetidp_mat, PC *fetidp_pc)
24883425bc38SStefano Zampini {
24893425bc38SStefano Zampini   PetscErrorCode ierr;
24903425bc38SStefano Zampini 
24913425bc38SStefano Zampini   PetscFunctionBegin;
24923425bc38SStefano Zampini   PetscValidHeaderSpecific(pc,PC_CLASSID,1);
24933425bc38SStefano Zampini   if (pc->setupcalled) {
2494547c9a8eSstefano_zampini     ierr = PetscUseMethod(pc,"PCBDDCCreateFETIDPOperators_C",(PC,PetscBool,const char*,Mat*,PC*),(pc,fully_redundant,prefix,fetidp_mat,fetidp_pc));CHKERRQ(ierr);
24954247aa23Sstefano_zampini   } else SETERRQ(PetscObjectComm((PetscObject)pc),PETSC_ERR_SUP,"You must call PCSetup_BDDC() first \n");
24963425bc38SStefano Zampini   PetscFunctionReturn(0);
24973425bc38SStefano Zampini }
24980c7d97c5SJed Brown /* -------------------------------------------------------------------------- */
2499da1bb401SStefano Zampini /*MC
2500da1bb401SStefano Zampini    PCBDDC - Balancing Domain Decomposition by Constraints.
25010c7d97c5SJed Brown 
250228509bceSStefano Zampini    An implementation of the BDDC preconditioner based on
250328509bceSStefano Zampini 
250428509bceSStefano Zampini .vb
250528509bceSStefano Zampini    [1] C. R. Dohrmann. "An approximate BDDC preconditioner", Numerical Linear Algebra with Applications Volume 14, Issue 2, pages 149-168, March 2007
250628509bceSStefano 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
250728509bceSStefano Zampini    [3] J. Mandel, B. Sousedik, C. R. Dohrmann. "Multispace and Multilevel BDDC", http://arxiv.org/abs/0712.3977
25080f202f7eSStefano 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
250928509bceSStefano Zampini .ve
251028509bceSStefano Zampini 
251128509bceSStefano Zampini    The matrix to be preconditioned (Pmat) must be of type MATIS.
251228509bceSStefano Zampini 
25130f202f7eSStefano Zampini    Currently works with MATIS matrices with local matrices of type MATSEQAIJ, MATSEQBAIJ or MATSEQSBAIJ, either with real or complex numbers.
251428509bceSStefano Zampini 
251528509bceSStefano Zampini    It also works with unsymmetric and indefinite problems.
251628509bceSStefano Zampini 
2517b6fdb6dfSStefano 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.
2518b6fdb6dfSStefano Zampini 
2519c7017625SStefano 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).
252028509bceSStefano Zampini 
25210f202f7eSStefano 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()
252230368db7SStefano Zampini    Additional information on dofs can be provided by using PCBDDCSetDofsSplitting(), PCBDDCSetDirichletBoundaries(), PCBDDCSetNeumannBoundaries(), and PCBDDCSetPrimalVerticesIS() and their local counterparts.
252328509bceSStefano Zampini 
25240f202f7eSStefano Zampini    Constraints can be customized by attaching a MatNullSpace object to the MATIS matrix via MatSetNearNullSpace(). Non-singular modes are retained via SVD.
252528509bceSStefano Zampini 
25260f202f7eSStefano 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.
25270f202f7eSStefano Zampini    User defined change of basis can be passed to PCBDDC by using PCBDDCSetChangeOfBasisMat()
252828509bceSStefano Zampini 
25290f202f7eSStefano Zampini    The PETSc implementation also supports multilevel BDDC [3]. Coarse grids are partitioned using a MatPartitioning object.
253028509bceSStefano Zampini 
2531df4d28bfSStefano 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.
253228509bceSStefano Zampini 
25330f202f7eSStefano 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.
25340f202f7eSStefano Zampini    Deluxe scaling is not supported yet for FETI-DP.
25350f202f7eSStefano Zampini 
25360f202f7eSStefano Zampini    Options Database Keys (some of them, run with -h for a complete list):
25370f202f7eSStefano Zampini 
25380f202f7eSStefano Zampini .    -pc_bddc_use_vertices <true> - use or not vertices in primal space
25390f202f7eSStefano Zampini .    -pc_bddc_use_edges <true> - use or not edges in primal space
25400f202f7eSStefano Zampini .    -pc_bddc_use_faces <false> - use or not faces in primal space
25410f202f7eSStefano Zampini .    -pc_bddc_symmetric <true> - symmetric computation of primal basis functions. Specify false for unsymmetric problems
25420f202f7eSStefano Zampini .    -pc_bddc_use_change_of_basis <false> - use change of basis approach (on edges only)
25430f202f7eSStefano Zampini .    -pc_bddc_use_change_on_faces <false> - use change of basis approach on faces if change of basis has been requested
25440f202f7eSStefano Zampini .    -pc_bddc_switch_static <false> - switches from M_2 (default) to M_3 operator (see reference article [1])
254528509bceSStefano Zampini .    -pc_bddc_levels <0> - maximum number of levels for multilevel
25460f202f7eSStefano 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)
25470f202f7eSStefano 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)
25480f202f7eSStefano Zampini .    -pc_bddc_use_deluxe_scaling <false> - use deluxe scaling
25490f202f7eSStefano 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)
2550df4d28bfSStefano 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)
255128509bceSStefano Zampini -    -pc_bddc_check_level <0> - set verbosity level of debugging output
255228509bceSStefano Zampini 
255328509bceSStefano Zampini    Options for Dirichlet, Neumann or coarse solver can be set with
255428509bceSStefano Zampini .vb
255528509bceSStefano Zampini       -pc_bddc_dirichlet_
255628509bceSStefano Zampini       -pc_bddc_neumann_
255728509bceSStefano Zampini       -pc_bddc_coarse_
255828509bceSStefano Zampini .ve
25590f202f7eSStefano Zampini    e.g -pc_bddc_dirichlet_ksp_type richardson -pc_bddc_dirichlet_pc_type gamg. PCBDDC uses by default KPSPREONLY and PCLU.
256028509bceSStefano Zampini 
25610f202f7eSStefano Zampini    When using a multilevel approach, solvers' options at the N-th level (N > 1) can be specified as
256228509bceSStefano Zampini .vb
2563312be037SStefano Zampini       -pc_bddc_dirichlet_lN_
2564312be037SStefano Zampini       -pc_bddc_neumann_lN_
2565312be037SStefano Zampini       -pc_bddc_coarse_lN_
256628509bceSStefano Zampini .ve
25670f202f7eSStefano Zampini    Note that level number ranges from the finest (0) to the coarsest (N).
25680f202f7eSStefano 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.
25690f202f7eSStefano Zampini .vb
25700f202f7eSStefano Zampini      -pc_bddc_coarse_pc_bddc_adaptive_threshold 5 -pc_bddc_coarse_l1_pc_bddc_redistribute 3
25710f202f7eSStefano Zampini .ve
25720f202f7eSStefano 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
2573da1bb401SStefano Zampini 
2574da1bb401SStefano Zampini    Level: intermediate
2575da1bb401SStefano Zampini 
2576b6fdb6dfSStefano Zampini    Developer notes:
2577da1bb401SStefano Zampini 
2578da1bb401SStefano Zampini    Contributed by Stefano Zampini
2579da1bb401SStefano Zampini 
2580da1bb401SStefano Zampini .seealso:  PCCreate(), PCSetType(), PCType (for list of available types), PC,  MATIS
2581da1bb401SStefano Zampini M*/
2582b2573a8aSBarry Smith 
25838cc058d9SJed Brown PETSC_EXTERN PetscErrorCode PCCreate_BDDC(PC pc)
2584da1bb401SStefano Zampini {
2585da1bb401SStefano Zampini   PetscErrorCode      ierr;
2586da1bb401SStefano Zampini   PC_BDDC             *pcbddc;
2587da1bb401SStefano Zampini 
2588da1bb401SStefano Zampini   PetscFunctionBegin;
2589b00a9115SJed Brown   ierr     = PetscNewLog(pc,&pcbddc);CHKERRQ(ierr);
2590da1bb401SStefano Zampini   pc->data = (void*)pcbddc;
2591da1bb401SStefano Zampini 
2592da1bb401SStefano Zampini   /* create PCIS data structure */
2593da1bb401SStefano Zampini   ierr = PCISCreate(pc);CHKERRQ(ierr);
2594da1bb401SStefano Zampini 
25959326c5c6Sstefano_zampini   /* create local graph structure */
25969326c5c6Sstefano_zampini   ierr = PCBDDCGraphCreate(&pcbddc->mat_graph);CHKERRQ(ierr);
25979326c5c6Sstefano_zampini 
25989326c5c6Sstefano_zampini   /* BDDC nonzero defaults */
259908a5cf49SStefano Zampini   pcbddc->use_local_adj             = PETSC_TRUE;
260047d04d0dSStefano Zampini   pcbddc->use_vertices              = PETSC_TRUE;
260147d04d0dSStefano Zampini   pcbddc->use_edges                 = PETSC_TRUE;
26023301b35fSStefano Zampini   pcbddc->symmetric_primal          = PETSC_TRUE;
260314f95afaSStefano Zampini   pcbddc->vertex_size               = 1;
2604c703fcc7SStefano Zampini   pcbddc->recompute_topography      = PETSC_TRUE;
260568457ee5SStefano Zampini   pcbddc->coarse_size               = -1;
260685c4d303SStefano Zampini   pcbddc->use_exact_dirichlet_trick = PETSC_TRUE;
260747d04d0dSStefano Zampini   pcbddc->coarsening_ratio          = 8;
260857de7509SStefano Zampini   pcbddc->coarse_eqs_per_proc       = 1;
260927b6a85dSStefano Zampini   pcbddc->benign_compute_correction = PETSC_TRUE;
26101e0482f5SStefano Zampini   pcbddc->nedfield                  = -1;
26111e0482f5SStefano Zampini   pcbddc->nedglobal                 = PETSC_TRUE;
2612be12c134Sstefano_zampini   pcbddc->graphmaxcount             = PETSC_MAX_INT;
2613b96c3477SStefano Zampini   pcbddc->sub_schurs_layers         = -1;
2614b7eb3628SStefano Zampini 
2615da1bb401SStefano Zampini   /* function pointers */
2616da1bb401SStefano Zampini   pc->ops->apply               = PCApply_BDDC;
261793bd9ae7SStefano Zampini   pc->ops->applytranspose      = PCApplyTranspose_BDDC;
2618da1bb401SStefano Zampini   pc->ops->setup               = PCSetUp_BDDC;
2619da1bb401SStefano Zampini   pc->ops->destroy             = PCDestroy_BDDC;
2620da1bb401SStefano Zampini   pc->ops->setfromoptions      = PCSetFromOptions_BDDC;
26216b78500eSPatrick Sanan   pc->ops->view                = PCView_BDDC;
2622da1bb401SStefano Zampini   pc->ops->applyrichardson     = 0;
2623da1bb401SStefano Zampini   pc->ops->applysymmetricleft  = 0;
2624da1bb401SStefano Zampini   pc->ops->applysymmetricright = 0;
2625534831adSStefano Zampini   pc->ops->presolve            = PCPreSolve_BDDC;
2626534831adSStefano Zampini   pc->ops->postsolve           = PCPostSolve_BDDC;
26279326c5c6Sstefano_zampini   pc->ops->reset               = PCReset_BDDC;
2628da1bb401SStefano Zampini 
2629da1bb401SStefano Zampini   /* composing function */
2630a13144ffSStefano Zampini   ierr = PetscObjectComposeFunction((PetscObject)pc,"PCBDDCSetDiscreteGradient_C",PCBDDCSetDiscreteGradient_BDDC);CHKERRQ(ierr);
2631a198735bSStefano Zampini   ierr = PetscObjectComposeFunction((PetscObject)pc,"PCBDDCSetDivergenceMat_C",PCBDDCSetDivergenceMat_BDDC);CHKERRQ(ierr);
2632906d46d4SStefano Zampini   ierr = PetscObjectComposeFunction((PetscObject)pc,"PCBDDCSetChangeOfBasisMat_C",PCBDDCSetChangeOfBasisMat_BDDC);CHKERRQ(ierr);
2633674ae819SStefano Zampini   ierr = PetscObjectComposeFunction((PetscObject)pc,"PCBDDCSetPrimalVerticesLocalIS_C",PCBDDCSetPrimalVerticesLocalIS_BDDC);CHKERRQ(ierr);
263430368db7SStefano Zampini   ierr = PetscObjectComposeFunction((PetscObject)pc,"PCBDDCSetPrimalVerticesIS_C",PCBDDCSetPrimalVerticesIS_BDDC);CHKERRQ(ierr);
2635bdf89e91SBarry Smith   ierr = PetscObjectComposeFunction((PetscObject)pc,"PCBDDCSetCoarseningRatio_C",PCBDDCSetCoarseningRatio_BDDC);CHKERRQ(ierr);
26362b510759SStefano Zampini   ierr = PetscObjectComposeFunction((PetscObject)pc,"PCBDDCSetLevel_C",PCBDDCSetLevel_BDDC);CHKERRQ(ierr);
2637b8ffe317SStefano Zampini   ierr = PetscObjectComposeFunction((PetscObject)pc,"PCBDDCSetUseExactDirichlet_C",PCBDDCSetUseExactDirichlet_BDDC);CHKERRQ(ierr);
26382b510759SStefano Zampini   ierr = PetscObjectComposeFunction((PetscObject)pc,"PCBDDCSetLevels_C",PCBDDCSetLevels_BDDC);CHKERRQ(ierr);
2639bdf89e91SBarry Smith   ierr = PetscObjectComposeFunction((PetscObject)pc,"PCBDDCSetDirichletBoundaries_C",PCBDDCSetDirichletBoundaries_BDDC);CHKERRQ(ierr);
264082ba6b80SStefano Zampini   ierr = PetscObjectComposeFunction((PetscObject)pc,"PCBDDCSetDirichletBoundariesLocal_C",PCBDDCSetDirichletBoundariesLocal_BDDC);CHKERRQ(ierr);
2641bdf89e91SBarry Smith   ierr = PetscObjectComposeFunction((PetscObject)pc,"PCBDDCSetNeumannBoundaries_C",PCBDDCSetNeumannBoundaries_BDDC);CHKERRQ(ierr);
264282ba6b80SStefano Zampini   ierr = PetscObjectComposeFunction((PetscObject)pc,"PCBDDCSetNeumannBoundariesLocal_C",PCBDDCSetNeumannBoundariesLocal_BDDC);CHKERRQ(ierr);
2643bdf89e91SBarry Smith   ierr = PetscObjectComposeFunction((PetscObject)pc,"PCBDDCGetDirichletBoundaries_C",PCBDDCGetDirichletBoundaries_BDDC);CHKERRQ(ierr);
264482ba6b80SStefano Zampini   ierr = PetscObjectComposeFunction((PetscObject)pc,"PCBDDCGetDirichletBoundariesLocal_C",PCBDDCGetDirichletBoundariesLocal_BDDC);CHKERRQ(ierr);
2645bdf89e91SBarry Smith   ierr = PetscObjectComposeFunction((PetscObject)pc,"PCBDDCGetNeumannBoundaries_C",PCBDDCGetNeumannBoundaries_BDDC);CHKERRQ(ierr);
264682ba6b80SStefano Zampini   ierr = PetscObjectComposeFunction((PetscObject)pc,"PCBDDCGetNeumannBoundariesLocal_C",PCBDDCGetNeumannBoundariesLocal_BDDC);CHKERRQ(ierr);
2647bdf89e91SBarry Smith   ierr = PetscObjectComposeFunction((PetscObject)pc,"PCBDDCSetDofsSplitting_C",PCBDDCSetDofsSplitting_BDDC);CHKERRQ(ierr);
264863602bcaSStefano Zampini   ierr = PetscObjectComposeFunction((PetscObject)pc,"PCBDDCSetDofsSplittingLocal_C",PCBDDCSetDofsSplittingLocal_BDDC);CHKERRQ(ierr);
2649bdf89e91SBarry Smith   ierr = PetscObjectComposeFunction((PetscObject)pc,"PCBDDCSetLocalAdjacencyGraph_C",PCBDDCSetLocalAdjacencyGraph_BDDC);CHKERRQ(ierr);
2650bdf89e91SBarry Smith   ierr = PetscObjectComposeFunction((PetscObject)pc,"PCBDDCCreateFETIDPOperators_C",PCBDDCCreateFETIDPOperators_BDDC);CHKERRQ(ierr);
2651bdf89e91SBarry Smith   ierr = PetscObjectComposeFunction((PetscObject)pc,"PCBDDCMatFETIDPGetRHS_C",PCBDDCMatFETIDPGetRHS_BDDC);CHKERRQ(ierr);
2652bdf89e91SBarry Smith   ierr = PetscObjectComposeFunction((PetscObject)pc,"PCBDDCMatFETIDPGetSolution_C",PCBDDCMatFETIDPGetSolution_BDDC);CHKERRQ(ierr);
2653a06fd7f2SStefano Zampini   ierr = PetscObjectComposeFunction((PetscObject)pc,"PCPreSolveChangeRHS_C",PCPreSolveChangeRHS_BDDC);CHKERRQ(ierr);
2654da1bb401SStefano Zampini   PetscFunctionReturn(0);
2655da1bb401SStefano Zampini }
26563425bc38SStefano Zampini 
2657