153cdbc3dSStefano Zampini /* TODOLIST 2eb97c9d2SStefano Zampini 3eb97c9d2SStefano Zampini Solvers 4a0d3c3abSStefano Zampini - Add support for cholesky for coarse solver (similar to local solvers) 5eb97c9d2SStefano Zampini - Propagate ksp prefixes for solvers to mat objects? 6eb97c9d2SStefano Zampini 7eb97c9d2SStefano Zampini User interface 80f202f7eSStefano Zampini - ** DM attached to pc? 9eb97c9d2SStefano Zampini 10eb97c9d2SStefano Zampini Debugging output 11b9b85e73SStefano Zampini - * Better management of verbosity levels of debugging output 12eb97c9d2SStefano Zampini 13eb97c9d2SStefano Zampini Extra 14b9b85e73SStefano Zampini - *** Is it possible to work with PCBDDCGraph on boundary indices only (less memory consumed)? 15eb97c9d2SStefano Zampini - BDDC with MG framework? 16eb97c9d2SStefano Zampini 17eb97c9d2SStefano Zampini FETIDP 18eb97c9d2SStefano Zampini - Move FETIDP code to its own classes 19eb97c9d2SStefano Zampini 20eb97c9d2SStefano Zampini MATIS related operations contained in BDDC code 21eb97c9d2SStefano Zampini - Provide general case for subassembling 22eb97c9d2SStefano Zampini 2353cdbc3dSStefano Zampini */ 240c7d97c5SJed Brown 25ab5c6b0cSJed Brown #include <../src/ksp/pc/impls/bddc/bddc.h> /*I "petscpc.h" I*/ /* includes for fortran wrappers */ 26ab5c6b0cSJed Brown #include <../src/ksp/pc/impls/bddc/bddcprivate.h> 273b03a366Sstefano_zampini #include <petscblaslapack.h> 28674ae819SStefano Zampini 290c7d97c5SJed Brown /* -------------------------------------------------------------------------- */ 300c7d97c5SJed Brown #undef __FUNCT__ 310c7d97c5SJed Brown #define __FUNCT__ "PCSetFromOptions_BDDC" 328c34d3f5SBarry Smith PetscErrorCode PCSetFromOptions_BDDC(PetscOptions *PetscOptionsObject,PC pc) 330c7d97c5SJed Brown { 340c7d97c5SJed Brown PC_BDDC *pcbddc = (PC_BDDC*)pc->data; 350c7d97c5SJed Brown PetscErrorCode ierr; 360c7d97c5SJed Brown 370c7d97c5SJed Brown PetscFunctionBegin; 38e55864a3SBarry Smith ierr = PetscOptionsHead(PetscOptionsObject,"BDDC options");CHKERRQ(ierr); 398eeda7d8SStefano Zampini /* Verbose debugging */ 408eeda7d8SStefano Zampini ierr = PetscOptionsInt("-pc_bddc_check_level","Verbose output for PCBDDC (intended for debug)","none",pcbddc->dbg_flag,&pcbddc->dbg_flag,NULL);CHKERRQ(ierr); 418eeda7d8SStefano Zampini /* Primal space cumstomization */ 4208a5cf49SStefano 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); 438eeda7d8SStefano 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); 448eeda7d8SStefano 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); 458eeda7d8SStefano 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); 466661aa1dSStefano 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); 47ac632422SStefano Zampini ierr = PetscOptionsBool("-pc_bddc_use_qr_single","Use QR factorization for single constraints on cc (QR is used when multiple constraints are present)","none",pcbddc->use_qr_single,&pcbddc->use_qr_single,NULL);CHKERRQ(ierr); 488eeda7d8SStefano Zampini /* Change of basis */ 49b9b85e73SStefano 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); 50b9b85e73SStefano 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); 51674ae819SStefano Zampini if (!pcbddc->use_change_of_basis) { 52674ae819SStefano Zampini pcbddc->use_change_on_faces = PETSC_FALSE; 53674ae819SStefano Zampini } 548eeda7d8SStefano Zampini /* Switch between M_2 (default) and M_3 preconditioners (as defined by C. Dohrmann in the ref. article) */ 558eeda7d8SStefano 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); 5674e2c79eSStefano Zampini ierr = PetscOptionsInt("-pc_bddc_coarse_redistribute","Number of procs where to redistribute coarse problem","none",pcbddc->redistribute_coarse,&pcbddc->redistribute_coarse,NULL);CHKERRQ(ierr); 570298fd71SBarry Smith ierr = PetscOptionsInt("-pc_bddc_coarsening_ratio","Set coarsening ratio used in multilevel coarsening","none",pcbddc->coarsening_ratio,&pcbddc->coarsening_ratio,NULL);CHKERRQ(ierr); 582b510759SStefano Zampini ierr = PetscOptionsInt("-pc_bddc_levels","Set maximum number of levels for multilevel","none",pcbddc->max_levels,&pcbddc->max_levels,NULL);CHKERRQ(ierr); 59323d395dSStefano 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); 60674ae819SStefano 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); 61b96c3477SStefano 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); 62b96c3477SStefano 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); 63b96c3477SStefano 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); 64ac632422SStefano Zampini ierr = PetscOptionsBool("-pc_bddc_deluxe_faster","Faster application of deluxe scaling (requires extra work during setup)","none",pcbddc->faster_deluxe,&pcbddc->faster_deluxe,NULL);CHKERRQ(ierr); 654c6709b3SStefano 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); 6608122e43SStefano Zampini ierr = PetscOptionsInt("-pc_bddc_adaptive_nmin","Minimum number of constraints per connected components","none",pcbddc->adaptive_nmin,&pcbddc->adaptive_nmin,NULL);CHKERRQ(ierr); 6708122e43SStefano Zampini ierr = PetscOptionsInt("-pc_bddc_adaptive_nmax","Maximum number of constraints per connected components","none",pcbddc->adaptive_nmax,&pcbddc->adaptive_nmax,NULL);CHKERRQ(ierr); 683301b35fSStefano Zampini ierr = PetscOptionsBool("-pc_bddc_symmetric","Symmetric computation of primal basis functions","none",pcbddc->symmetric_primal,&pcbddc->symmetric_primal,NULL);CHKERRQ(ierr); 69b0c7d250SStefano 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); 700c7d97c5SJed Brown ierr = PetscOptionsTail();CHKERRQ(ierr); 710c7d97c5SJed Brown PetscFunctionReturn(0); 720c7d97c5SJed Brown } 730c7d97c5SJed Brown /* -------------------------------------------------------------------------- */ 74674ae819SStefano Zampini #undef __FUNCT__ 75906d46d4SStefano Zampini #define __FUNCT__ "PCBDDCSetChangeOfBasisMat_BDDC" 76906d46d4SStefano Zampini static PetscErrorCode PCBDDCSetChangeOfBasisMat_BDDC(PC pc, Mat change) 77b9b85e73SStefano Zampini { 78b9b85e73SStefano Zampini PC_BDDC *pcbddc = (PC_BDDC*)pc->data; 79b9b85e73SStefano Zampini PetscErrorCode ierr; 80b9b85e73SStefano Zampini 81b9b85e73SStefano Zampini PetscFunctionBegin; 82b9b85e73SStefano Zampini ierr = MatDestroy(&pcbddc->user_ChangeOfBasisMatrix);CHKERRQ(ierr); 83b9b85e73SStefano Zampini ierr = PetscObjectReference((PetscObject)change);CHKERRQ(ierr); 84b9b85e73SStefano Zampini pcbddc->user_ChangeOfBasisMatrix = change; 85b9b85e73SStefano Zampini PetscFunctionReturn(0); 86b9b85e73SStefano Zampini } 87b9b85e73SStefano Zampini #undef __FUNCT__ 88906d46d4SStefano Zampini #define __FUNCT__ "PCBDDCSetChangeOfBasisMat" 89b9b85e73SStefano Zampini /*@ 90906d46d4SStefano Zampini PCBDDCSetChangeOfBasisMat - Set user defined change of basis for dofs 91b9b85e73SStefano Zampini 92b9b85e73SStefano Zampini Collective on PC 93b9b85e73SStefano Zampini 94b9b85e73SStefano Zampini Input Parameters: 95b9b85e73SStefano Zampini + pc - the preconditioning context 96906d46d4SStefano Zampini - change - the change of basis matrix 97b9b85e73SStefano Zampini 98b9b85e73SStefano Zampini Level: intermediate 99b9b85e73SStefano Zampini 100b9b85e73SStefano Zampini Notes: 101b9b85e73SStefano Zampini 102b9b85e73SStefano Zampini .seealso: PCBDDC 103b9b85e73SStefano Zampini @*/ 104906d46d4SStefano Zampini PetscErrorCode PCBDDCSetChangeOfBasisMat(PC pc, Mat change) 105b9b85e73SStefano Zampini { 106b9b85e73SStefano Zampini PetscErrorCode ierr; 107b9b85e73SStefano Zampini 108b9b85e73SStefano Zampini PetscFunctionBegin; 109b9b85e73SStefano Zampini PetscValidHeaderSpecific(pc,PC_CLASSID,1); 110b9b85e73SStefano Zampini PetscValidHeaderSpecific(change,MAT_CLASSID,2); 111906d46d4SStefano Zampini PetscCheckSameComm(pc,1,change,2); 112906d46d4SStefano Zampini if (pc->mat) { 113906d46d4SStefano Zampini PetscInt rows_c,cols_c,rows,cols; 114906d46d4SStefano Zampini ierr = MatGetSize(pc->mat,&rows,&cols);CHKERRQ(ierr); 115906d46d4SStefano Zampini ierr = MatGetSize(change,&rows_c,&cols_c);CHKERRQ(ierr); 116*6c4ed002SBarry Smith if (rows_c != rows) SETERRQ2(PetscObjectComm((PetscObject)pc),PETSC_ERR_SUP,"Invalid number of rows for change of basis matrix! %d != %d",rows_c,rows); 117*6c4ed002SBarry Smith if (cols_c != cols) SETERRQ2(PetscObjectComm((PetscObject)pc),PETSC_ERR_SUP,"Invalid number of columns for change of basis matrix! %d != %d",cols_c,cols); 118906d46d4SStefano Zampini ierr = MatGetLocalSize(pc->mat,&rows,&cols);CHKERRQ(ierr); 119906d46d4SStefano Zampini ierr = MatGetLocalSize(change,&rows_c,&cols_c);CHKERRQ(ierr); 120*6c4ed002SBarry Smith if (rows_c != rows) SETERRQ2(PetscObjectComm((PetscObject)pc),PETSC_ERR_SUP,"Invalid number of local rows for change of basis matrix! %d != %d",rows_c,rows); 121*6c4ed002SBarry Smith if (cols_c != cols) SETERRQ2(PetscObjectComm((PetscObject)pc),PETSC_ERR_SUP,"Invalid number of local columns for change of basis matrix! %d != %d",cols_c,cols); 122906d46d4SStefano Zampini } 123906d46d4SStefano Zampini ierr = PetscTryMethod(pc,"PCBDDCSetChangeOfBasisMat_C",(PC,Mat),(pc,change));CHKERRQ(ierr); 124b9b85e73SStefano Zampini PetscFunctionReturn(0); 125b9b85e73SStefano Zampini } 126b9b85e73SStefano Zampini /* -------------------------------------------------------------------------- */ 127b9b85e73SStefano Zampini #undef __FUNCT__ 128674ae819SStefano Zampini #define __FUNCT__ "PCBDDCSetPrimalVerticesLocalIS_BDDC" 129674ae819SStefano Zampini static PetscErrorCode PCBDDCSetPrimalVerticesLocalIS_BDDC(PC pc, IS PrimalVertices) 130674ae819SStefano Zampini { 131674ae819SStefano Zampini PC_BDDC *pcbddc = (PC_BDDC*)pc->data; 132674ae819SStefano Zampini PetscErrorCode ierr; 1331e6b0712SBarry Smith 134674ae819SStefano Zampini PetscFunctionBegin; 135674ae819SStefano Zampini ierr = ISDestroy(&pcbddc->user_primal_vertices);CHKERRQ(ierr); 136674ae819SStefano Zampini ierr = PetscObjectReference((PetscObject)PrimalVertices);CHKERRQ(ierr); 137674ae819SStefano Zampini pcbddc->user_primal_vertices = PrimalVertices; 138674ae819SStefano Zampini PetscFunctionReturn(0); 139674ae819SStefano Zampini } 140674ae819SStefano Zampini #undef __FUNCT__ 141674ae819SStefano Zampini #define __FUNCT__ "PCBDDCSetPrimalVerticesLocalIS" 142674ae819SStefano Zampini /*@ 14328509bceSStefano Zampini PCBDDCSetPrimalVerticesLocalIS - Set additional user defined primal vertices in PCBDDC 144674ae819SStefano Zampini 14517eb1463SStefano Zampini Collective 146674ae819SStefano Zampini 147674ae819SStefano Zampini Input Parameters: 148674ae819SStefano Zampini + pc - the preconditioning context 14917eb1463SStefano Zampini - PrimalVertices - index set of primal vertices in local numbering (can be empty) 150674ae819SStefano Zampini 151674ae819SStefano Zampini Level: intermediate 152674ae819SStefano Zampini 153674ae819SStefano Zampini Notes: 154674ae819SStefano Zampini 155674ae819SStefano Zampini .seealso: PCBDDC 156674ae819SStefano Zampini @*/ 157674ae819SStefano Zampini PetscErrorCode PCBDDCSetPrimalVerticesLocalIS(PC pc, IS PrimalVertices) 158674ae819SStefano Zampini { 159674ae819SStefano Zampini PetscErrorCode ierr; 160674ae819SStefano Zampini 161674ae819SStefano Zampini PetscFunctionBegin; 162674ae819SStefano Zampini PetscValidHeaderSpecific(pc,PC_CLASSID,1); 163674ae819SStefano Zampini PetscValidHeaderSpecific(PrimalVertices,IS_CLASSID,2); 16417eb1463SStefano Zampini PetscCheckSameComm(pc,1,PrimalVertices,2); 165674ae819SStefano Zampini ierr = PetscTryMethod(pc,"PCBDDCSetPrimalVerticesLocalIS_C",(PC,IS),(pc,PrimalVertices));CHKERRQ(ierr); 166674ae819SStefano Zampini PetscFunctionReturn(0); 167674ae819SStefano Zampini } 168674ae819SStefano Zampini /* -------------------------------------------------------------------------- */ 1690c7d97c5SJed Brown #undef __FUNCT__ 1704fad6a16SStefano Zampini #define __FUNCT__ "PCBDDCSetCoarseningRatio_BDDC" 1714fad6a16SStefano Zampini static PetscErrorCode PCBDDCSetCoarseningRatio_BDDC(PC pc,PetscInt k) 1724fad6a16SStefano Zampini { 1734fad6a16SStefano Zampini PC_BDDC *pcbddc = (PC_BDDC*)pc->data; 1744fad6a16SStefano Zampini 1754fad6a16SStefano Zampini PetscFunctionBegin; 1764fad6a16SStefano Zampini pcbddc->coarsening_ratio = k; 1774fad6a16SStefano Zampini PetscFunctionReturn(0); 1784fad6a16SStefano Zampini } 1791e6b0712SBarry Smith 1804fad6a16SStefano Zampini #undef __FUNCT__ 1814fad6a16SStefano Zampini #define __FUNCT__ "PCBDDCSetCoarseningRatio" 1824fad6a16SStefano Zampini /*@ 18328509bceSStefano Zampini PCBDDCSetCoarseningRatio - Set coarsening ratio used in multilevel 1844fad6a16SStefano Zampini 1854fad6a16SStefano Zampini Logically collective on PC 1864fad6a16SStefano Zampini 1874fad6a16SStefano Zampini Input Parameters: 1884fad6a16SStefano Zampini + pc - the preconditioning context 18928509bceSStefano Zampini - k - coarsening ratio (H/h at the coarser level) 1904fad6a16SStefano Zampini 1910f202f7eSStefano Zampini Options Database Keys: 1920f202f7eSStefano Zampini . -pc_bddc_coarsening_ratio 1934fad6a16SStefano Zampini 1944fad6a16SStefano Zampini Level: intermediate 1954fad6a16SStefano Zampini 1964fad6a16SStefano Zampini Notes: 1970f202f7eSStefano Zampini Approximatively k subdomains at the finer level will be aggregated into a single subdomain at the coarser level 1984fad6a16SStefano Zampini 1990f202f7eSStefano Zampini .seealso: PCBDDC, PCBDDCSetLevels() 2004fad6a16SStefano Zampini @*/ 2014fad6a16SStefano Zampini PetscErrorCode PCBDDCSetCoarseningRatio(PC pc,PetscInt k) 2024fad6a16SStefano Zampini { 2034fad6a16SStefano Zampini PetscErrorCode ierr; 2044fad6a16SStefano Zampini 2054fad6a16SStefano Zampini PetscFunctionBegin; 2064fad6a16SStefano Zampini PetscValidHeaderSpecific(pc,PC_CLASSID,1); 2072b510759SStefano Zampini PetscValidLogicalCollectiveInt(pc,k,2); 2084fad6a16SStefano Zampini ierr = PetscTryMethod(pc,"PCBDDCSetCoarseningRatio_C",(PC,PetscInt),(pc,k));CHKERRQ(ierr); 2094fad6a16SStefano Zampini PetscFunctionReturn(0); 2104fad6a16SStefano Zampini } 2112b510759SStefano Zampini 212b8ffe317SStefano Zampini /* The following functions (PCBDDCSetUseExactDirichlet PCBDDCSetLevel) are not public */ 2132b510759SStefano Zampini #undef __FUNCT__ 214b8ffe317SStefano Zampini #define __FUNCT__ "PCBDDCSetUseExactDirichlet_BDDC" 215b8ffe317SStefano Zampini static PetscErrorCode PCBDDCSetUseExactDirichlet_BDDC(PC pc,PetscBool flg) 216b8ffe317SStefano Zampini { 217b8ffe317SStefano Zampini PC_BDDC *pcbddc = (PC_BDDC*)pc->data; 218b8ffe317SStefano Zampini 219b8ffe317SStefano Zampini PetscFunctionBegin; 22085c4d303SStefano Zampini pcbddc->use_exact_dirichlet_trick = flg; 221b8ffe317SStefano Zampini PetscFunctionReturn(0); 222b8ffe317SStefano Zampini } 223b8ffe317SStefano Zampini 224b8ffe317SStefano Zampini #undef __FUNCT__ 225b8ffe317SStefano Zampini #define __FUNCT__ "PCBDDCSetUseExactDirichlet" 226b8ffe317SStefano Zampini PetscErrorCode PCBDDCSetUseExactDirichlet(PC pc,PetscBool flg) 2272b510759SStefano Zampini { 2282b510759SStefano Zampini PetscErrorCode ierr; 2292b510759SStefano Zampini 2302b510759SStefano Zampini PetscFunctionBegin; 2312b510759SStefano Zampini PetscValidHeaderSpecific(pc,PC_CLASSID,1); 232b8ffe317SStefano Zampini PetscValidLogicalCollectiveBool(pc,flg,2); 233b8ffe317SStefano Zampini ierr = PetscTryMethod(pc,"PCBDDCSetUseExactDirichlet_C",(PC,PetscBool),(pc,flg));CHKERRQ(ierr); 2342b510759SStefano Zampini PetscFunctionReturn(0); 2352b510759SStefano Zampini } 2361e6b0712SBarry Smith 2374fad6a16SStefano Zampini #undef __FUNCT__ 2382b510759SStefano Zampini #define __FUNCT__ "PCBDDCSetLevel_BDDC" 2392b510759SStefano Zampini static PetscErrorCode PCBDDCSetLevel_BDDC(PC pc,PetscInt level) 2404fad6a16SStefano Zampini { 2414fad6a16SStefano Zampini PC_BDDC *pcbddc = (PC_BDDC*)pc->data; 2424fad6a16SStefano Zampini 2434fad6a16SStefano Zampini PetscFunctionBegin; 2442b510759SStefano Zampini pcbddc->current_level = level; 2454fad6a16SStefano Zampini PetscFunctionReturn(0); 2464fad6a16SStefano Zampini } 2471e6b0712SBarry Smith 2484fad6a16SStefano Zampini #undef __FUNCT__ 249b8ffe317SStefano Zampini #define __FUNCT__ "PCBDDCSetLevel" 250b8ffe317SStefano Zampini PetscErrorCode PCBDDCSetLevel(PC pc,PetscInt level) 251b8ffe317SStefano Zampini { 252b8ffe317SStefano Zampini PetscErrorCode ierr; 253b8ffe317SStefano Zampini 254b8ffe317SStefano Zampini PetscFunctionBegin; 255b8ffe317SStefano Zampini PetscValidHeaderSpecific(pc,PC_CLASSID,1); 256b8ffe317SStefano Zampini PetscValidLogicalCollectiveInt(pc,level,2); 257b8ffe317SStefano Zampini ierr = PetscTryMethod(pc,"PCBDDCSetLevel_C",(PC,PetscInt),(pc,level));CHKERRQ(ierr); 258b8ffe317SStefano Zampini PetscFunctionReturn(0); 259b8ffe317SStefano Zampini } 260b8ffe317SStefano Zampini 261b8ffe317SStefano Zampini #undef __FUNCT__ 2622b510759SStefano Zampini #define __FUNCT__ "PCBDDCSetLevels_BDDC" 2632b510759SStefano Zampini static PetscErrorCode PCBDDCSetLevels_BDDC(PC pc,PetscInt levels) 2642b510759SStefano Zampini { 2652b510759SStefano Zampini PC_BDDC *pcbddc = (PC_BDDC*)pc->data; 2662b510759SStefano Zampini 2672b510759SStefano Zampini PetscFunctionBegin; 2682b510759SStefano Zampini pcbddc->max_levels = levels; 2692b510759SStefano Zampini PetscFunctionReturn(0); 2702b510759SStefano Zampini } 2712b510759SStefano Zampini 2722b510759SStefano Zampini #undef __FUNCT__ 2732b510759SStefano Zampini #define __FUNCT__ "PCBDDCSetLevels" 2744fad6a16SStefano Zampini /*@ 27528509bceSStefano Zampini PCBDDCSetLevels - Sets the maximum number of levels for multilevel 2764fad6a16SStefano Zampini 2774fad6a16SStefano Zampini Logically collective on PC 2784fad6a16SStefano Zampini 2794fad6a16SStefano Zampini Input Parameters: 2804fad6a16SStefano Zampini + pc - the preconditioning context 28128509bceSStefano Zampini - levels - the maximum number of levels (max 9) 2824fad6a16SStefano Zampini 2830f202f7eSStefano Zampini Options Database Keys: 2840f202f7eSStefano Zampini . -pc_bddc_levels 2854fad6a16SStefano Zampini 2864fad6a16SStefano Zampini Level: intermediate 2874fad6a16SStefano Zampini 2884fad6a16SStefano Zampini Notes: 2890f202f7eSStefano Zampini Default value is 0, i.e. traditional one-level BDDC 2904fad6a16SStefano Zampini 2910f202f7eSStefano Zampini .seealso: PCBDDC, PCBDDCSetCoarseningRatio() 2924fad6a16SStefano Zampini @*/ 2932b510759SStefano Zampini PetscErrorCode PCBDDCSetLevels(PC pc,PetscInt levels) 2944fad6a16SStefano Zampini { 2954fad6a16SStefano Zampini PetscErrorCode ierr; 2964fad6a16SStefano Zampini 2974fad6a16SStefano Zampini PetscFunctionBegin; 2984fad6a16SStefano Zampini PetscValidHeaderSpecific(pc,PC_CLASSID,1); 2992b510759SStefano Zampini PetscValidLogicalCollectiveInt(pc,levels,2); 300312be037SStefano Zampini if (levels > 99) SETERRQ(PetscObjectComm((PetscObject)pc),PETSC_ERR_SUP,"Maximum number of levels for bddc is 99\n"); 3012b510759SStefano Zampini ierr = PetscTryMethod(pc,"PCBDDCSetLevels_C",(PC,PetscInt),(pc,levels));CHKERRQ(ierr); 3024fad6a16SStefano Zampini PetscFunctionReturn(0); 3034fad6a16SStefano Zampini } 3044fad6a16SStefano Zampini /* -------------------------------------------------------------------------- */ 3051e6b0712SBarry Smith 3064fad6a16SStefano Zampini #undef __FUNCT__ 3070bdf917eSStefano Zampini #define __FUNCT__ "PCBDDCSetNullSpace_BDDC" 3080bdf917eSStefano Zampini static PetscErrorCode PCBDDCSetNullSpace_BDDC(PC pc,MatNullSpace NullSpace) 3090bdf917eSStefano Zampini { 3100bdf917eSStefano Zampini PC_BDDC *pcbddc = (PC_BDDC*)pc->data; 3110bdf917eSStefano Zampini PetscErrorCode ierr; 3120bdf917eSStefano Zampini 3130bdf917eSStefano Zampini PetscFunctionBegin; 3140bdf917eSStefano Zampini ierr = PetscObjectReference((PetscObject)NullSpace);CHKERRQ(ierr); 3150bdf917eSStefano Zampini ierr = MatNullSpaceDestroy(&pcbddc->NullSpace);CHKERRQ(ierr); 3160bdf917eSStefano Zampini pcbddc->NullSpace = NullSpace; 3170bdf917eSStefano Zampini PetscFunctionReturn(0); 3180bdf917eSStefano Zampini } 3191e6b0712SBarry Smith 3200bdf917eSStefano Zampini #undef __FUNCT__ 3210bdf917eSStefano Zampini #define __FUNCT__ "PCBDDCSetNullSpace" 3220bdf917eSStefano Zampini /*@ 32328509bceSStefano Zampini PCBDDCSetNullSpace - Set nullspace for BDDC operator 3240bdf917eSStefano Zampini 3250bdf917eSStefano Zampini Logically collective on PC and MatNullSpace 3260bdf917eSStefano Zampini 3270bdf917eSStefano Zampini Input Parameters: 3280bdf917eSStefano Zampini + pc - the preconditioning context 32928509bceSStefano Zampini - NullSpace - Null space of the linear operator to be preconditioned (Pmat) 3300bdf917eSStefano Zampini 3310bdf917eSStefano Zampini Level: intermediate 3320bdf917eSStefano Zampini 3330bdf917eSStefano Zampini Notes: 3340bdf917eSStefano Zampini 3350bdf917eSStefano Zampini .seealso: PCBDDC 3360bdf917eSStefano Zampini @*/ 3370bdf917eSStefano Zampini PetscErrorCode PCBDDCSetNullSpace(PC pc,MatNullSpace NullSpace) 3380bdf917eSStefano Zampini { 3390bdf917eSStefano Zampini PetscErrorCode ierr; 3400bdf917eSStefano Zampini 3410bdf917eSStefano Zampini PetscFunctionBegin; 3420bdf917eSStefano Zampini PetscValidHeaderSpecific(pc,PC_CLASSID,1); 343674ae819SStefano Zampini PetscValidHeaderSpecific(NullSpace,MAT_NULLSPACE_CLASSID,2); 3442b510759SStefano Zampini PetscCheckSameComm(pc,1,NullSpace,2); 3450bdf917eSStefano Zampini ierr = PetscTryMethod(pc,"PCBDDCSetNullSpace_C",(PC,MatNullSpace),(pc,NullSpace));CHKERRQ(ierr); 3460bdf917eSStefano Zampini PetscFunctionReturn(0); 3470bdf917eSStefano Zampini } 3480bdf917eSStefano Zampini /* -------------------------------------------------------------------------- */ 3491e6b0712SBarry Smith 3500bdf917eSStefano Zampini #undef __FUNCT__ 3513b03a366Sstefano_zampini #define __FUNCT__ "PCBDDCSetDirichletBoundaries_BDDC" 3523b03a366Sstefano_zampini static PetscErrorCode PCBDDCSetDirichletBoundaries_BDDC(PC pc,IS DirichletBoundaries) 3533b03a366Sstefano_zampini { 3543b03a366Sstefano_zampini PC_BDDC *pcbddc = (PC_BDDC*)pc->data; 3553b03a366Sstefano_zampini PetscErrorCode ierr; 3563b03a366Sstefano_zampini 3573b03a366Sstefano_zampini PetscFunctionBegin; 358785d1243SStefano Zampini /* last user setting takes precendence -> destroy any other customization */ 359785d1243SStefano Zampini ierr = ISDestroy(&pcbddc->DirichletBoundariesLocal);CHKERRQ(ierr); 3603b03a366Sstefano_zampini ierr = ISDestroy(&pcbddc->DirichletBoundaries);CHKERRQ(ierr); 36136e030ebSStefano Zampini ierr = PetscObjectReference((PetscObject)DirichletBoundaries);CHKERRQ(ierr); 36236e030ebSStefano Zampini pcbddc->DirichletBoundaries = DirichletBoundaries; 363fb180af4SStefano Zampini pcbddc->recompute_topography = PETSC_TRUE; 3643b03a366Sstefano_zampini PetscFunctionReturn(0); 3653b03a366Sstefano_zampini } 3661e6b0712SBarry Smith 3673b03a366Sstefano_zampini #undef __FUNCT__ 3683b03a366Sstefano_zampini #define __FUNCT__ "PCBDDCSetDirichletBoundaries" 3693b03a366Sstefano_zampini /*@ 37028509bceSStefano Zampini PCBDDCSetDirichletBoundaries - Set IS defining Dirichlet boundaries for the global problem. 3713b03a366Sstefano_zampini 372785d1243SStefano Zampini Collective 3733b03a366Sstefano_zampini 3743b03a366Sstefano_zampini Input Parameters: 3753b03a366Sstefano_zampini + pc - the preconditioning context 376785d1243SStefano Zampini - DirichletBoundaries - parallel IS defining the Dirichlet boundaries 3773b03a366Sstefano_zampini 3783b03a366Sstefano_zampini Level: intermediate 3793b03a366Sstefano_zampini 3800f202f7eSStefano Zampini Notes: 3810f202f7eSStefano Zampini Provide the information if you used MatZeroRows/Columns routines. Any process can list any global node 3823b03a366Sstefano_zampini 3830f202f7eSStefano Zampini .seealso: PCBDDC, PCBDDCSetDirichletBoundariesLocal(), MatZeroRows(), MatZeroRowsColumns() 3843b03a366Sstefano_zampini @*/ 3853b03a366Sstefano_zampini PetscErrorCode PCBDDCSetDirichletBoundaries(PC pc,IS DirichletBoundaries) 3863b03a366Sstefano_zampini { 3873b03a366Sstefano_zampini PetscErrorCode ierr; 3883b03a366Sstefano_zampini 3893b03a366Sstefano_zampini PetscFunctionBegin; 3903b03a366Sstefano_zampini PetscValidHeaderSpecific(pc,PC_CLASSID,1); 391674ae819SStefano Zampini PetscValidHeaderSpecific(DirichletBoundaries,IS_CLASSID,2); 392785d1243SStefano Zampini PetscCheckSameComm(pc,1,DirichletBoundaries,2); 3933b03a366Sstefano_zampini ierr = PetscTryMethod(pc,"PCBDDCSetDirichletBoundaries_C",(PC,IS),(pc,DirichletBoundaries));CHKERRQ(ierr); 3943b03a366Sstefano_zampini PetscFunctionReturn(0); 3953b03a366Sstefano_zampini } 3963b03a366Sstefano_zampini /* -------------------------------------------------------------------------- */ 3971e6b0712SBarry Smith 3983b03a366Sstefano_zampini #undef __FUNCT__ 39982ba6b80SStefano Zampini #define __FUNCT__ "PCBDDCSetDirichletBoundariesLocal_BDDC" 40082ba6b80SStefano Zampini static PetscErrorCode PCBDDCSetDirichletBoundariesLocal_BDDC(PC pc,IS DirichletBoundaries) 4010c7d97c5SJed Brown { 4020c7d97c5SJed Brown PC_BDDC *pcbddc = (PC_BDDC*)pc->data; 4030c7d97c5SJed Brown PetscErrorCode ierr; 4040c7d97c5SJed Brown 4050c7d97c5SJed Brown PetscFunctionBegin; 406785d1243SStefano Zampini /* last user setting takes precendence -> destroy any other customization */ 407785d1243SStefano Zampini ierr = ISDestroy(&pcbddc->DirichletBoundariesLocal);CHKERRQ(ierr); 4080c7d97c5SJed Brown ierr = ISDestroy(&pcbddc->DirichletBoundaries);CHKERRQ(ierr); 4090c7d97c5SJed Brown ierr = PetscObjectReference((PetscObject)DirichletBoundaries);CHKERRQ(ierr); 410785d1243SStefano Zampini pcbddc->DirichletBoundariesLocal = DirichletBoundaries; 4117d24d155SStefano Zampini pcbddc->recompute_topography = PETSC_TRUE; 4120c7d97c5SJed Brown PetscFunctionReturn(0); 4130c7d97c5SJed Brown } 4140c7d97c5SJed Brown 4150c7d97c5SJed Brown #undef __FUNCT__ 41682ba6b80SStefano Zampini #define __FUNCT__ "PCBDDCSetDirichletBoundariesLocal" 4179c0446d6SStefano Zampini /*@ 41882ba6b80SStefano Zampini PCBDDCSetDirichletBoundariesLocal - Set IS defining Dirichlet boundaries for the global problem in local ordering. 4199c0446d6SStefano Zampini 420785d1243SStefano Zampini Collective 42153cdbc3dSStefano Zampini 42253cdbc3dSStefano Zampini Input Parameters: 42353cdbc3dSStefano Zampini + pc - the preconditioning context 42482ba6b80SStefano Zampini - DirichletBoundaries - parallel IS defining the Dirichlet boundaries (in local ordering) 42553cdbc3dSStefano Zampini 42653cdbc3dSStefano Zampini Level: intermediate 42753cdbc3dSStefano Zampini 4289c0446d6SStefano Zampini Notes: 42953cdbc3dSStefano Zampini 4300f202f7eSStefano Zampini .seealso: PCBDDC, PCBDDCSetDirichletBoundaries(), MatZeroRows(), MatZeroRowsColumns() 43153cdbc3dSStefano Zampini @*/ 43282ba6b80SStefano Zampini PetscErrorCode PCBDDCSetDirichletBoundariesLocal(PC pc,IS DirichletBoundaries) 4330c7d97c5SJed Brown { 4340c7d97c5SJed Brown PetscErrorCode ierr; 4350c7d97c5SJed Brown 4360c7d97c5SJed Brown PetscFunctionBegin; 4370c7d97c5SJed Brown PetscValidHeaderSpecific(pc,PC_CLASSID,1); 4380c7d97c5SJed Brown PetscValidHeaderSpecific(DirichletBoundaries,IS_CLASSID,2); 43982ba6b80SStefano Zampini PetscCheckSameComm(pc,1,DirichletBoundaries,2); 44082ba6b80SStefano Zampini ierr = PetscTryMethod(pc,"PCBDDCSetDirichletBoundariesLocal_C",(PC,IS),(pc,DirichletBoundaries));CHKERRQ(ierr); 4410c7d97c5SJed Brown PetscFunctionReturn(0); 4420c7d97c5SJed Brown } 4430c7d97c5SJed Brown /* -------------------------------------------------------------------------- */ 4440c7d97c5SJed Brown 4450c7d97c5SJed Brown #undef __FUNCT__ 4460c7d97c5SJed Brown #define __FUNCT__ "PCBDDCSetNeumannBoundaries_BDDC" 44753cdbc3dSStefano Zampini static PetscErrorCode PCBDDCSetNeumannBoundaries_BDDC(PC pc,IS NeumannBoundaries) 4480c7d97c5SJed Brown { 4490c7d97c5SJed Brown PC_BDDC *pcbddc = (PC_BDDC*)pc->data; 45053cdbc3dSStefano Zampini PetscErrorCode ierr; 4510c7d97c5SJed Brown 4520c7d97c5SJed Brown PetscFunctionBegin; 453785d1243SStefano Zampini /* last user setting takes precendence -> destroy any other customization */ 454785d1243SStefano Zampini ierr = ISDestroy(&pcbddc->NeumannBoundariesLocal);CHKERRQ(ierr); 45553cdbc3dSStefano Zampini ierr = ISDestroy(&pcbddc->NeumannBoundaries);CHKERRQ(ierr); 45636e030ebSStefano Zampini ierr = PetscObjectReference((PetscObject)NeumannBoundaries);CHKERRQ(ierr); 45736e030ebSStefano Zampini pcbddc->NeumannBoundaries = NeumannBoundaries; 458fb180af4SStefano Zampini pcbddc->recompute_topography = PETSC_TRUE; 4590c7d97c5SJed Brown PetscFunctionReturn(0); 4600c7d97c5SJed Brown } 4611e6b0712SBarry Smith 4620c7d97c5SJed Brown #undef __FUNCT__ 4630c7d97c5SJed Brown #define __FUNCT__ "PCBDDCSetNeumannBoundaries" 46457527edcSJed Brown /*@ 46528509bceSStefano Zampini PCBDDCSetNeumannBoundaries - Set IS defining Neumann boundaries for the global problem. 46657527edcSJed Brown 467785d1243SStefano Zampini Collective 46857527edcSJed Brown 46957527edcSJed Brown Input Parameters: 47057527edcSJed Brown + pc - the preconditioning context 471785d1243SStefano Zampini - NeumannBoundaries - parallel IS defining the Neumann boundaries 47257527edcSJed Brown 47357527edcSJed Brown Level: intermediate 47457527edcSJed Brown 4750f202f7eSStefano Zampini Notes: 4760f202f7eSStefano Zampini Any process can list any global node 47757527edcSJed Brown 4780f202f7eSStefano Zampini .seealso: PCBDDC, PCBDDCSetNeumannBoundariesLocal() 47957527edcSJed Brown @*/ 48053cdbc3dSStefano Zampini PetscErrorCode PCBDDCSetNeumannBoundaries(PC pc,IS NeumannBoundaries) 4810c7d97c5SJed Brown { 4820c7d97c5SJed Brown PetscErrorCode ierr; 4830c7d97c5SJed Brown 4840c7d97c5SJed Brown PetscFunctionBegin; 4850c7d97c5SJed Brown PetscValidHeaderSpecific(pc,PC_CLASSID,1); 486674ae819SStefano Zampini PetscValidHeaderSpecific(NeumannBoundaries,IS_CLASSID,2); 487785d1243SStefano Zampini PetscCheckSameComm(pc,1,NeumannBoundaries,2); 48853cdbc3dSStefano Zampini ierr = PetscTryMethod(pc,"PCBDDCSetNeumannBoundaries_C",(PC,IS),(pc,NeumannBoundaries));CHKERRQ(ierr); 48953cdbc3dSStefano Zampini PetscFunctionReturn(0); 49053cdbc3dSStefano Zampini } 49153cdbc3dSStefano Zampini /* -------------------------------------------------------------------------- */ 4921e6b0712SBarry Smith 49353cdbc3dSStefano Zampini #undef __FUNCT__ 49482ba6b80SStefano Zampini #define __FUNCT__ "PCBDDCSetNeumannBoundariesLocal_BDDC" 49582ba6b80SStefano Zampini static PetscErrorCode PCBDDCSetNeumannBoundariesLocal_BDDC(PC pc,IS NeumannBoundaries) 4960c7d97c5SJed Brown { 4970c7d97c5SJed Brown PC_BDDC *pcbddc = (PC_BDDC*)pc->data; 4980c7d97c5SJed Brown PetscErrorCode ierr; 4990c7d97c5SJed Brown 5000c7d97c5SJed Brown PetscFunctionBegin; 501785d1243SStefano Zampini /* last user setting takes precendence -> destroy any other customization */ 502785d1243SStefano Zampini ierr = ISDestroy(&pcbddc->NeumannBoundariesLocal);CHKERRQ(ierr); 5030c7d97c5SJed Brown ierr = ISDestroy(&pcbddc->NeumannBoundaries);CHKERRQ(ierr); 5040c7d97c5SJed Brown ierr = PetscObjectReference((PetscObject)NeumannBoundaries);CHKERRQ(ierr); 505785d1243SStefano Zampini pcbddc->NeumannBoundariesLocal = NeumannBoundaries; 5067d24d155SStefano Zampini pcbddc->recompute_topography = PETSC_TRUE; 5070c7d97c5SJed Brown PetscFunctionReturn(0); 5080c7d97c5SJed Brown } 5090c7d97c5SJed Brown 5100c7d97c5SJed Brown #undef __FUNCT__ 51182ba6b80SStefano Zampini #define __FUNCT__ "PCBDDCSetNeumannBoundariesLocal" 5120c7d97c5SJed Brown /*@ 51382ba6b80SStefano Zampini PCBDDCSetNeumannBoundariesLocal - Set IS defining Neumann boundaries for the global problem in local ordering. 5140c7d97c5SJed Brown 515785d1243SStefano Zampini Collective 5160c7d97c5SJed Brown 5170c7d97c5SJed Brown Input Parameters: 5180c7d97c5SJed Brown + pc - the preconditioning context 51982ba6b80SStefano Zampini - NeumannBoundaries - parallel IS defining the subdomain part of Neumann boundaries (in local ordering) 5200c7d97c5SJed Brown 5210c7d97c5SJed Brown Level: intermediate 5220c7d97c5SJed Brown 5230c7d97c5SJed Brown Notes: 5240c7d97c5SJed Brown 5250f202f7eSStefano Zampini .seealso: PCBDDC, PCBDDCSetNeumannBoundaries() 5260c7d97c5SJed Brown @*/ 52782ba6b80SStefano Zampini PetscErrorCode PCBDDCSetNeumannBoundariesLocal(PC pc,IS NeumannBoundaries) 5280c7d97c5SJed Brown { 5290c7d97c5SJed Brown PetscErrorCode ierr; 5300c7d97c5SJed Brown 5310c7d97c5SJed Brown PetscFunctionBegin; 5320c7d97c5SJed Brown PetscValidHeaderSpecific(pc,PC_CLASSID,1); 5330c7d97c5SJed Brown PetscValidHeaderSpecific(NeumannBoundaries,IS_CLASSID,2); 53482ba6b80SStefano Zampini PetscCheckSameComm(pc,1,NeumannBoundaries,2); 53582ba6b80SStefano Zampini ierr = PetscTryMethod(pc,"PCBDDCSetNeumannBoundariesLocal_C",(PC,IS),(pc,NeumannBoundaries));CHKERRQ(ierr); 53653cdbc3dSStefano Zampini PetscFunctionReturn(0); 53753cdbc3dSStefano Zampini } 53853cdbc3dSStefano Zampini /* -------------------------------------------------------------------------- */ 53953cdbc3dSStefano Zampini 54053cdbc3dSStefano Zampini #undef __FUNCT__ 541da1bb401SStefano Zampini #define __FUNCT__ "PCBDDCGetDirichletBoundaries_BDDC" 542da1bb401SStefano Zampini static PetscErrorCode PCBDDCGetDirichletBoundaries_BDDC(PC pc,IS *DirichletBoundaries) 543da1bb401SStefano Zampini { 544da1bb401SStefano Zampini PC_BDDC *pcbddc = (PC_BDDC*)pc->data; 545da1bb401SStefano Zampini 546da1bb401SStefano Zampini PetscFunctionBegin; 547da1bb401SStefano Zampini *DirichletBoundaries = pcbddc->DirichletBoundaries; 548da1bb401SStefano Zampini PetscFunctionReturn(0); 549da1bb401SStefano Zampini } 5501e6b0712SBarry Smith 551da1bb401SStefano Zampini #undef __FUNCT__ 552da1bb401SStefano Zampini #define __FUNCT__ "PCBDDCGetDirichletBoundaries" 553da1bb401SStefano Zampini /*@ 554785d1243SStefano Zampini PCBDDCGetDirichletBoundaries - Get parallel IS for Dirichlet boundaries 555da1bb401SStefano Zampini 556785d1243SStefano Zampini Collective 557785d1243SStefano Zampini 558785d1243SStefano Zampini Input Parameters: 559785d1243SStefano Zampini . pc - the preconditioning context 560785d1243SStefano Zampini 561785d1243SStefano Zampini Output Parameters: 562785d1243SStefano Zampini . DirichletBoundaries - index set defining the Dirichlet boundaries 563785d1243SStefano Zampini 564785d1243SStefano Zampini Level: intermediate 565785d1243SStefano Zampini 5660f202f7eSStefano Zampini Notes: 5670f202f7eSStefano Zampini The IS returned (if any) is the same passed in earlier by the user with PCBDDCSetDirichletBoundaries 568785d1243SStefano Zampini 569785d1243SStefano Zampini .seealso: PCBDDC 570785d1243SStefano Zampini @*/ 571785d1243SStefano Zampini PetscErrorCode PCBDDCGetDirichletBoundaries(PC pc,IS *DirichletBoundaries) 572785d1243SStefano Zampini { 573785d1243SStefano Zampini PetscErrorCode ierr; 574785d1243SStefano Zampini 575785d1243SStefano Zampini PetscFunctionBegin; 576785d1243SStefano Zampini PetscValidHeaderSpecific(pc,PC_CLASSID,1); 577785d1243SStefano Zampini ierr = PetscUseMethod(pc,"PCBDDCGetDirichletBoundaries_C",(PC,IS*),(pc,DirichletBoundaries));CHKERRQ(ierr); 578785d1243SStefano Zampini PetscFunctionReturn(0); 579785d1243SStefano Zampini } 580785d1243SStefano Zampini /* -------------------------------------------------------------------------- */ 581785d1243SStefano Zampini 582785d1243SStefano Zampini #undef __FUNCT__ 583785d1243SStefano Zampini #define __FUNCT__ "PCBDDCGetDirichletBoundariesLocal_BDDC" 584785d1243SStefano Zampini static PetscErrorCode PCBDDCGetDirichletBoundariesLocal_BDDC(PC pc,IS *DirichletBoundaries) 585785d1243SStefano Zampini { 586785d1243SStefano Zampini PC_BDDC *pcbddc = (PC_BDDC*)pc->data; 587785d1243SStefano Zampini 588785d1243SStefano Zampini PetscFunctionBegin; 589785d1243SStefano Zampini *DirichletBoundaries = pcbddc->DirichletBoundariesLocal; 590785d1243SStefano Zampini PetscFunctionReturn(0); 591785d1243SStefano Zampini } 592785d1243SStefano Zampini 593785d1243SStefano Zampini #undef __FUNCT__ 59482ba6b80SStefano Zampini #define __FUNCT__ "PCBDDCGetDirichletBoundariesLocal" 595da1bb401SStefano Zampini /*@ 59682ba6b80SStefano Zampini PCBDDCGetDirichletBoundariesLocal - Get parallel IS for Dirichlet boundaries (in local ordering) 597da1bb401SStefano Zampini 598785d1243SStefano Zampini Collective 599da1bb401SStefano Zampini 600da1bb401SStefano Zampini Input Parameters: 60128509bceSStefano Zampini . pc - the preconditioning context 602da1bb401SStefano Zampini 603da1bb401SStefano Zampini Output Parameters: 60428509bceSStefano Zampini . DirichletBoundaries - index set defining the subdomain part of Dirichlet boundaries 605da1bb401SStefano Zampini 606da1bb401SStefano Zampini Level: intermediate 607da1bb401SStefano Zampini 608da1bb401SStefano Zampini Notes: 6090f202f7eSStefano 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). 6100f202f7eSStefano Zampini In the latter case, the IS will be available after PCSetUp. 611da1bb401SStefano Zampini 612da1bb401SStefano Zampini .seealso: PCBDDC 613da1bb401SStefano Zampini @*/ 61482ba6b80SStefano Zampini PetscErrorCode PCBDDCGetDirichletBoundariesLocal(PC pc,IS *DirichletBoundaries) 615da1bb401SStefano Zampini { 616da1bb401SStefano Zampini PetscErrorCode ierr; 617da1bb401SStefano Zampini 618da1bb401SStefano Zampini PetscFunctionBegin; 619da1bb401SStefano Zampini PetscValidHeaderSpecific(pc,PC_CLASSID,1); 62082ba6b80SStefano Zampini ierr = PetscUseMethod(pc,"PCBDDCGetDirichletBoundariesLocal_C",(PC,IS*),(pc,DirichletBoundaries));CHKERRQ(ierr); 621da1bb401SStefano Zampini PetscFunctionReturn(0); 622da1bb401SStefano Zampini } 623da1bb401SStefano Zampini /* -------------------------------------------------------------------------- */ 6241e6b0712SBarry Smith 625da1bb401SStefano Zampini #undef __FUNCT__ 62653cdbc3dSStefano Zampini #define __FUNCT__ "PCBDDCGetNeumannBoundaries_BDDC" 62753cdbc3dSStefano Zampini static PetscErrorCode PCBDDCGetNeumannBoundaries_BDDC(PC pc,IS *NeumannBoundaries) 62853cdbc3dSStefano Zampini { 62953cdbc3dSStefano Zampini PC_BDDC *pcbddc = (PC_BDDC*)pc->data; 63053cdbc3dSStefano Zampini 63153cdbc3dSStefano Zampini PetscFunctionBegin; 63253cdbc3dSStefano Zampini *NeumannBoundaries = pcbddc->NeumannBoundaries; 63353cdbc3dSStefano Zampini PetscFunctionReturn(0); 63453cdbc3dSStefano Zampini } 6351e6b0712SBarry Smith 63653cdbc3dSStefano Zampini #undef __FUNCT__ 63753cdbc3dSStefano Zampini #define __FUNCT__ "PCBDDCGetNeumannBoundaries" 63853cdbc3dSStefano Zampini /*@ 639785d1243SStefano Zampini PCBDDCGetNeumannBoundaries - Get parallel IS for Neumann boundaries 64053cdbc3dSStefano Zampini 641785d1243SStefano Zampini Collective 642785d1243SStefano Zampini 643785d1243SStefano Zampini Input Parameters: 644785d1243SStefano Zampini . pc - the preconditioning context 645785d1243SStefano Zampini 646785d1243SStefano Zampini Output Parameters: 647785d1243SStefano Zampini . NeumannBoundaries - index set defining the Neumann boundaries 648785d1243SStefano Zampini 649785d1243SStefano Zampini Level: intermediate 650785d1243SStefano Zampini 6510f202f7eSStefano Zampini Notes: 6520f202f7eSStefano Zampini The IS returned (if any) is the same passed in earlier by the user with PCBDDCSetNeumannBoundaries 653785d1243SStefano Zampini 654785d1243SStefano Zampini .seealso: PCBDDC 655785d1243SStefano Zampini @*/ 656785d1243SStefano Zampini PetscErrorCode PCBDDCGetNeumannBoundaries(PC pc,IS *NeumannBoundaries) 657785d1243SStefano Zampini { 658785d1243SStefano Zampini PetscErrorCode ierr; 659785d1243SStefano Zampini 660785d1243SStefano Zampini PetscFunctionBegin; 661785d1243SStefano Zampini PetscValidHeaderSpecific(pc,PC_CLASSID,1); 662785d1243SStefano Zampini ierr = PetscUseMethod(pc,"PCBDDCGetNeumannBoundaries_C",(PC,IS*),(pc,NeumannBoundaries));CHKERRQ(ierr); 663785d1243SStefano Zampini PetscFunctionReturn(0); 664785d1243SStefano Zampini } 665785d1243SStefano Zampini /* -------------------------------------------------------------------------- */ 666785d1243SStefano Zampini 667785d1243SStefano Zampini #undef __FUNCT__ 668785d1243SStefano Zampini #define __FUNCT__ "PCBDDCGetNeumannBoundariesLocal_BDDC" 669785d1243SStefano Zampini static PetscErrorCode PCBDDCGetNeumannBoundariesLocal_BDDC(PC pc,IS *NeumannBoundaries) 670785d1243SStefano Zampini { 671785d1243SStefano Zampini PC_BDDC *pcbddc = (PC_BDDC*)pc->data; 672785d1243SStefano Zampini 673785d1243SStefano Zampini PetscFunctionBegin; 674785d1243SStefano Zampini *NeumannBoundaries = pcbddc->NeumannBoundariesLocal; 675785d1243SStefano Zampini PetscFunctionReturn(0); 676785d1243SStefano Zampini } 677785d1243SStefano Zampini 678785d1243SStefano Zampini #undef __FUNCT__ 67982ba6b80SStefano Zampini #define __FUNCT__ "PCBDDCGetNeumannBoundariesLocal" 68053cdbc3dSStefano Zampini /*@ 68182ba6b80SStefano Zampini PCBDDCGetNeumannBoundariesLocal - Get parallel IS for Neumann boundaries (in local ordering) 68253cdbc3dSStefano Zampini 683785d1243SStefano Zampini Collective 68453cdbc3dSStefano Zampini 68553cdbc3dSStefano Zampini Input Parameters: 68628509bceSStefano Zampini . pc - the preconditioning context 68753cdbc3dSStefano Zampini 68853cdbc3dSStefano Zampini Output Parameters: 68928509bceSStefano Zampini . NeumannBoundaries - index set defining the subdomain part of Neumann boundaries 69053cdbc3dSStefano Zampini 69153cdbc3dSStefano Zampini Level: intermediate 69253cdbc3dSStefano Zampini 69353cdbc3dSStefano Zampini Notes: 6940f202f7eSStefano 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). 6950f202f7eSStefano Zampini In the latter case, the IS will be available after PCSetUp. 69653cdbc3dSStefano Zampini 69753cdbc3dSStefano Zampini .seealso: PCBDDC 69853cdbc3dSStefano Zampini @*/ 69982ba6b80SStefano Zampini PetscErrorCode PCBDDCGetNeumannBoundariesLocal(PC pc,IS *NeumannBoundaries) 70053cdbc3dSStefano Zampini { 70153cdbc3dSStefano Zampini PetscErrorCode ierr; 70253cdbc3dSStefano Zampini 70353cdbc3dSStefano Zampini PetscFunctionBegin; 70453cdbc3dSStefano Zampini PetscValidHeaderSpecific(pc,PC_CLASSID,1); 70582ba6b80SStefano Zampini ierr = PetscUseMethod(pc,"PCBDDCGetNeumannBoundariesLocal_C",(PC,IS*),(pc,NeumannBoundaries));CHKERRQ(ierr); 7060c7d97c5SJed Brown PetscFunctionReturn(0); 7070c7d97c5SJed Brown } 70836e030ebSStefano Zampini /* -------------------------------------------------------------------------- */ 7091e6b0712SBarry Smith 71036e030ebSStefano Zampini #undef __FUNCT__ 711da1bb401SStefano Zampini #define __FUNCT__ "PCBDDCSetLocalAdjacencyGraph_BDDC" 7121a83f524SJed Brown static PetscErrorCode PCBDDCSetLocalAdjacencyGraph_BDDC(PC pc, PetscInt nvtxs,const PetscInt xadj[],const PetscInt adjncy[], PetscCopyMode copymode) 71336e030ebSStefano Zampini { 71436e030ebSStefano Zampini PC_BDDC *pcbddc = (PC_BDDC*)pc->data; 715da1bb401SStefano Zampini PCBDDCGraph mat_graph = pcbddc->mat_graph; 716da1bb401SStefano Zampini PetscErrorCode ierr; 71736e030ebSStefano Zampini 71836e030ebSStefano Zampini PetscFunctionBegin; 719674ae819SStefano Zampini /* free old CSR */ 720674ae819SStefano Zampini ierr = PCBDDCGraphResetCSR(mat_graph);CHKERRQ(ierr); 721fb223d50SStefano Zampini /* TODO: PCBDDCGraphSetAdjacency */ 722674ae819SStefano Zampini /* get CSR into graph structure */ 723da1bb401SStefano Zampini if (copymode == PETSC_COPY_VALUES) { 724854ce69bSBarry Smith ierr = PetscMalloc1(nvtxs+1,&mat_graph->xadj);CHKERRQ(ierr); 725785e854fSJed Brown ierr = PetscMalloc1(xadj[nvtxs],&mat_graph->adjncy);CHKERRQ(ierr); 726674ae819SStefano Zampini ierr = PetscMemcpy(mat_graph->xadj,xadj,(nvtxs+1)*sizeof(PetscInt));CHKERRQ(ierr); 727674ae819SStefano Zampini ierr = PetscMemcpy(mat_graph->adjncy,adjncy,xadj[nvtxs]*sizeof(PetscInt));CHKERRQ(ierr); 728da1bb401SStefano Zampini } else if (copymode == PETSC_OWN_POINTER) { 7291a83f524SJed Brown mat_graph->xadj = (PetscInt*)xadj; 7301a83f524SJed Brown mat_graph->adjncy = (PetscInt*)adjncy; 731674ae819SStefano Zampini } 732575ad6abSStefano Zampini mat_graph->nvtxs_csr = nvtxs; 73336e030ebSStefano Zampini PetscFunctionReturn(0); 73436e030ebSStefano Zampini } 7351e6b0712SBarry Smith 73636e030ebSStefano Zampini #undef __FUNCT__ 737da1bb401SStefano Zampini #define __FUNCT__ "PCBDDCSetLocalAdjacencyGraph" 73836e030ebSStefano Zampini /*@ 7390f202f7eSStefano Zampini PCBDDCSetLocalAdjacencyGraph - Set adjacency structure (CSR graph) of the local matrix 74036e030ebSStefano Zampini 74136e030ebSStefano Zampini Not collective 74236e030ebSStefano Zampini 74336e030ebSStefano Zampini Input Parameters: 74436e030ebSStefano Zampini + pc - the preconditioning context 7450f202f7eSStefano Zampini . nvtxs - number of local vertices of the graph (i.e., the size of the local problem) 74628509bceSStefano Zampini . xadj, adjncy - the CSR graph 74728509bceSStefano Zampini - copymode - either PETSC_COPY_VALUES or PETSC_OWN_POINTER. 74836e030ebSStefano Zampini 74936e030ebSStefano Zampini Level: intermediate 75036e030ebSStefano Zampini 75136e030ebSStefano Zampini Notes: 75236e030ebSStefano Zampini 75328509bceSStefano Zampini .seealso: PCBDDC,PetscCopyMode 75436e030ebSStefano Zampini @*/ 7551a83f524SJed Brown PetscErrorCode PCBDDCSetLocalAdjacencyGraph(PC pc,PetscInt nvtxs,const PetscInt xadj[],const PetscInt adjncy[], PetscCopyMode copymode) 75636e030ebSStefano Zampini { 757575ad6abSStefano Zampini void (*f)(void) = 0; 75836e030ebSStefano Zampini PetscErrorCode ierr; 75936e030ebSStefano Zampini 76036e030ebSStefano Zampini PetscFunctionBegin; 76136e030ebSStefano Zampini PetscValidHeaderSpecific(pc,PC_CLASSID,1); 762674ae819SStefano Zampini PetscValidIntPointer(xadj,3); 763d7de1dd9SStefano Zampini PetscValidIntPointer(adjncy,4); 764*6c4ed002SBarry Smith if (copymode != PETSC_COPY_VALUES && copymode != PETSC_OWN_POINTER) SETERRQ1(PETSC_COMM_SELF,PETSC_ERR_SUP,"Unsupported copy mode %d",copymode); 7651a83f524SJed Brown ierr = PetscTryMethod(pc,"PCBDDCSetLocalAdjacencyGraph_C",(PC,PetscInt,const PetscInt[],const PetscInt[],PetscCopyMode),(pc,nvtxs,xadj,adjncy,copymode));CHKERRQ(ierr); 766575ad6abSStefano Zampini /* free arrays if PCBDDC is not the PC type */ 767575ad6abSStefano Zampini ierr = PetscObjectQueryFunction((PetscObject)pc,"PCBDDCSetLocalAdjacencyGraph_C",&f);CHKERRQ(ierr); 768575ad6abSStefano Zampini if (!f && copymode == PETSC_OWN_POINTER) { 769575ad6abSStefano Zampini ierr = PetscFree(xadj);CHKERRQ(ierr); 770575ad6abSStefano Zampini ierr = PetscFree(adjncy);CHKERRQ(ierr); 771da1bb401SStefano Zampini } 77236e030ebSStefano Zampini PetscFunctionReturn(0); 77336e030ebSStefano Zampini } 7749c0446d6SStefano Zampini /* -------------------------------------------------------------------------- */ 7751e6b0712SBarry Smith 7769c0446d6SStefano Zampini #undef __FUNCT__ 77763602bcaSStefano Zampini #define __FUNCT__ "PCBDDCSetDofsSplittingLocal_BDDC" 77863602bcaSStefano Zampini static PetscErrorCode PCBDDCSetDofsSplittingLocal_BDDC(PC pc,PetscInt n_is, IS ISForDofs[]) 77963602bcaSStefano Zampini { 78063602bcaSStefano Zampini PC_BDDC *pcbddc = (PC_BDDC*)pc->data; 78163602bcaSStefano Zampini PetscInt i; 78263602bcaSStefano Zampini PetscErrorCode ierr; 78363602bcaSStefano Zampini 78463602bcaSStefano Zampini PetscFunctionBegin; 78563602bcaSStefano Zampini /* Destroy ISes if they were already set */ 78663602bcaSStefano Zampini for (i=0;i<pcbddc->n_ISForDofsLocal;i++) { 78763602bcaSStefano Zampini ierr = ISDestroy(&pcbddc->ISForDofsLocal[i]);CHKERRQ(ierr); 78863602bcaSStefano Zampini } 78963602bcaSStefano Zampini ierr = PetscFree(pcbddc->ISForDofsLocal);CHKERRQ(ierr); 79063602bcaSStefano Zampini /* last user setting takes precendence -> destroy any other customization */ 79163602bcaSStefano Zampini for (i=0;i<pcbddc->n_ISForDofs;i++) { 79263602bcaSStefano Zampini ierr = ISDestroy(&pcbddc->ISForDofs[i]);CHKERRQ(ierr); 79363602bcaSStefano Zampini } 79463602bcaSStefano Zampini ierr = PetscFree(pcbddc->ISForDofs);CHKERRQ(ierr); 79563602bcaSStefano Zampini pcbddc->n_ISForDofs = 0; 79663602bcaSStefano Zampini /* allocate space then set */ 797d02579f5SStefano Zampini if (n_is) { 798d02579f5SStefano Zampini ierr = PetscMalloc1(n_is,&pcbddc->ISForDofsLocal);CHKERRQ(ierr); 799d02579f5SStefano Zampini } 80063602bcaSStefano Zampini for (i=0;i<n_is;i++) { 80163602bcaSStefano Zampini ierr = PetscObjectReference((PetscObject)ISForDofs[i]);CHKERRQ(ierr); 80263602bcaSStefano Zampini pcbddc->ISForDofsLocal[i]=ISForDofs[i]; 80363602bcaSStefano Zampini } 80463602bcaSStefano Zampini pcbddc->n_ISForDofsLocal=n_is; 80563602bcaSStefano Zampini if (n_is) pcbddc->user_provided_isfordofs = PETSC_TRUE; 8061a8a16d1SStefano Zampini pcbddc->recompute_topography = PETSC_TRUE; 80763602bcaSStefano Zampini PetscFunctionReturn(0); 80863602bcaSStefano Zampini } 80963602bcaSStefano Zampini 81063602bcaSStefano Zampini #undef __FUNCT__ 81163602bcaSStefano Zampini #define __FUNCT__ "PCBDDCSetDofsSplittingLocal" 81263602bcaSStefano Zampini /*@ 81363602bcaSStefano Zampini PCBDDCSetDofsSplittingLocal - Set index sets defining fields of the local subdomain matrix 81463602bcaSStefano Zampini 81563602bcaSStefano Zampini Collective 81663602bcaSStefano Zampini 81763602bcaSStefano Zampini Input Parameters: 81863602bcaSStefano Zampini + pc - the preconditioning context 8190f202f7eSStefano Zampini . n_is - number of index sets defining the fields 8200f202f7eSStefano Zampini - ISForDofs - array of IS describing the fields in local ordering 82163602bcaSStefano Zampini 82263602bcaSStefano Zampini Level: intermediate 82363602bcaSStefano Zampini 8240f202f7eSStefano Zampini Notes: 8250f202f7eSStefano Zampini n_is should be the same among processes. Not all nodes need to be listed: unlisted nodes will belong to the complement field. 82663602bcaSStefano Zampini 82763602bcaSStefano Zampini .seealso: PCBDDC 82863602bcaSStefano Zampini @*/ 82963602bcaSStefano Zampini PetscErrorCode PCBDDCSetDofsSplittingLocal(PC pc,PetscInt n_is, IS ISForDofs[]) 83063602bcaSStefano Zampini { 83163602bcaSStefano Zampini PetscInt i; 83263602bcaSStefano Zampini PetscErrorCode ierr; 83363602bcaSStefano Zampini 83463602bcaSStefano Zampini PetscFunctionBegin; 83563602bcaSStefano Zampini PetscValidHeaderSpecific(pc,PC_CLASSID,1); 83663602bcaSStefano Zampini PetscValidLogicalCollectiveInt(pc,n_is,2); 83763602bcaSStefano Zampini for (i=0;i<n_is;i++) { 83863602bcaSStefano Zampini PetscCheckSameComm(pc,1,ISForDofs[i],3); 83963602bcaSStefano Zampini PetscValidHeaderSpecific(ISForDofs[i],IS_CLASSID,3); 84063602bcaSStefano Zampini } 841e71e7a71SStefano Zampini ierr = PetscTryMethod(pc,"PCBDDCSetDofsSplittingLocal_C",(PC,PetscInt,IS[]),(pc,n_is,ISForDofs));CHKERRQ(ierr); 84263602bcaSStefano Zampini PetscFunctionReturn(0); 84363602bcaSStefano Zampini } 84463602bcaSStefano Zampini /* -------------------------------------------------------------------------- */ 84563602bcaSStefano Zampini 84663602bcaSStefano Zampini #undef __FUNCT__ 8479c0446d6SStefano Zampini #define __FUNCT__ "PCBDDCSetDofsSplitting_BDDC" 8489c0446d6SStefano Zampini static PetscErrorCode PCBDDCSetDofsSplitting_BDDC(PC pc,PetscInt n_is, IS ISForDofs[]) 8499c0446d6SStefano Zampini { 8509c0446d6SStefano Zampini PC_BDDC *pcbddc = (PC_BDDC*)pc->data; 8519c0446d6SStefano Zampini PetscInt i; 8529c0446d6SStefano Zampini PetscErrorCode ierr; 8539c0446d6SStefano Zampini 8549c0446d6SStefano Zampini PetscFunctionBegin; 855da1bb401SStefano Zampini /* Destroy ISes if they were already set */ 8569c0446d6SStefano Zampini for (i=0;i<pcbddc->n_ISForDofs;i++) { 8579c0446d6SStefano Zampini ierr = ISDestroy(&pcbddc->ISForDofs[i]);CHKERRQ(ierr); 8589c0446d6SStefano Zampini } 859d11ae9bbSstefano_zampini ierr = PetscFree(pcbddc->ISForDofs);CHKERRQ(ierr); 86063602bcaSStefano Zampini /* last user setting takes precendence -> destroy any other customization */ 86163602bcaSStefano Zampini for (i=0;i<pcbddc->n_ISForDofsLocal;i++) { 86263602bcaSStefano Zampini ierr = ISDestroy(&pcbddc->ISForDofsLocal[i]);CHKERRQ(ierr); 86363602bcaSStefano Zampini } 86463602bcaSStefano Zampini ierr = PetscFree(pcbddc->ISForDofsLocal);CHKERRQ(ierr); 86563602bcaSStefano Zampini pcbddc->n_ISForDofsLocal = 0; 866da1bb401SStefano Zampini /* allocate space then set */ 867d02579f5SStefano Zampini if (n_is) { 868785e854fSJed Brown ierr = PetscMalloc1(n_is,&pcbddc->ISForDofs);CHKERRQ(ierr); 869d02579f5SStefano Zampini } 8709c0446d6SStefano Zampini for (i=0;i<n_is;i++) { 871da1bb401SStefano Zampini ierr = PetscObjectReference((PetscObject)ISForDofs[i]);CHKERRQ(ierr); 872da1bb401SStefano Zampini pcbddc->ISForDofs[i]=ISForDofs[i]; 8739c0446d6SStefano Zampini } 8749c0446d6SStefano Zampini pcbddc->n_ISForDofs=n_is; 87563602bcaSStefano Zampini if (n_is) pcbddc->user_provided_isfordofs = PETSC_TRUE; 8761a8a16d1SStefano Zampini pcbddc->recompute_topography = PETSC_TRUE; 8779c0446d6SStefano Zampini PetscFunctionReturn(0); 8789c0446d6SStefano Zampini } 8791e6b0712SBarry Smith 8809c0446d6SStefano Zampini #undef __FUNCT__ 8819c0446d6SStefano Zampini #define __FUNCT__ "PCBDDCSetDofsSplitting" 8829c0446d6SStefano Zampini /*@ 88363602bcaSStefano Zampini PCBDDCSetDofsSplitting - Set index sets defining fields of the global matrix 8849c0446d6SStefano Zampini 88563602bcaSStefano Zampini Collective 8869c0446d6SStefano Zampini 8879c0446d6SStefano Zampini Input Parameters: 8889c0446d6SStefano Zampini + pc - the preconditioning context 8890f202f7eSStefano Zampini . n_is - number of index sets defining the fields 8900f202f7eSStefano Zampini - ISForDofs - array of IS describing the fields in global ordering 8919c0446d6SStefano Zampini 8929c0446d6SStefano Zampini Level: intermediate 8939c0446d6SStefano Zampini 8940f202f7eSStefano Zampini Notes: 8950f202f7eSStefano Zampini Any process can list any global node. Not all nodes need to be listed: unlisted nodes will belong to the complement field. 8969c0446d6SStefano Zampini 8979c0446d6SStefano Zampini .seealso: PCBDDC 8989c0446d6SStefano Zampini @*/ 8999c0446d6SStefano Zampini PetscErrorCode PCBDDCSetDofsSplitting(PC pc,PetscInt n_is, IS ISForDofs[]) 9009c0446d6SStefano Zampini { 9012b510759SStefano Zampini PetscInt i; 9029c0446d6SStefano Zampini PetscErrorCode ierr; 9039c0446d6SStefano Zampini 9049c0446d6SStefano Zampini PetscFunctionBegin; 9059c0446d6SStefano Zampini PetscValidHeaderSpecific(pc,PC_CLASSID,1); 90663602bcaSStefano Zampini PetscValidLogicalCollectiveInt(pc,n_is,2); 9072b510759SStefano Zampini for (i=0;i<n_is;i++) { 90863602bcaSStefano Zampini PetscCheckSameComm(pc,1,ISForDofs[i],3); 90963602bcaSStefano Zampini PetscValidHeaderSpecific(ISForDofs[i],IS_CLASSID,3); 9102b510759SStefano Zampini } 9119c0446d6SStefano Zampini ierr = PetscTryMethod(pc,"PCBDDCSetDofsSplitting_C",(PC,PetscInt,IS[]),(pc,n_is,ISForDofs));CHKERRQ(ierr); 9129c0446d6SStefano Zampini PetscFunctionReturn(0); 9139c0446d6SStefano Zampini } 914906d46d4SStefano Zampini 915da1bb401SStefano Zampini /* -------------------------------------------------------------------------- */ 916534831adSStefano Zampini #undef __FUNCT__ 917534831adSStefano Zampini #define __FUNCT__ "PCPreSolve_BDDC" 918534831adSStefano Zampini /* -------------------------------------------------------------------------- */ 919534831adSStefano Zampini /* 920534831adSStefano Zampini PCPreSolve_BDDC - Changes the right hand side and (if necessary) the initial 921534831adSStefano Zampini guess if a transformation of basis approach has been selected. 9229c0446d6SStefano Zampini 923534831adSStefano Zampini Input Parameter: 924534831adSStefano Zampini + pc - the preconditioner contex 925534831adSStefano Zampini 926534831adSStefano Zampini Application Interface Routine: PCPreSolve() 927534831adSStefano Zampini 928534831adSStefano Zampini Notes: 929534831adSStefano Zampini The interface routine PCPreSolve() is not usually called directly by 930534831adSStefano Zampini the user, but instead is called by KSPSolve(). 931534831adSStefano Zampini */ 932534831adSStefano Zampini static PetscErrorCode PCPreSolve_BDDC(PC pc, KSP ksp, Vec rhs, Vec x) 933534831adSStefano Zampini { 934534831adSStefano Zampini PetscErrorCode ierr; 935534831adSStefano Zampini PC_BDDC *pcbddc = (PC_BDDC*)pc->data; 936534831adSStefano Zampini PC_IS *pcis = (PC_IS*)(pc->data); 9373972b0daSStefano Zampini Vec used_vec; 9388d00608fSStefano Zampini PetscBool copy_rhs = PETSC_TRUE; 939534831adSStefano Zampini 940534831adSStefano Zampini PetscFunctionBegin; 94185c4d303SStefano Zampini /* if we are working with cg, one dirichlet solve can be avoided during Krylov iterations */ 94285c4d303SStefano Zampini if (ksp) { 94385c4d303SStefano Zampini PetscBool iscg; 94485c4d303SStefano Zampini ierr = PetscObjectTypeCompare((PetscObject)ksp,KSPCG,&iscg);CHKERRQ(ierr); 94585c4d303SStefano Zampini if (!iscg) { 94685c4d303SStefano Zampini ierr = PCBDDCSetUseExactDirichlet(pc,PETSC_FALSE);CHKERRQ(ierr); 94785c4d303SStefano Zampini } 94885c4d303SStefano Zampini } 94985c4d303SStefano Zampini /* Creates parallel work vectors used in presolve */ 95062a6ff1dSStefano Zampini if (!pcbddc->original_rhs) { 95162a6ff1dSStefano Zampini ierr = VecDuplicate(pcis->vec1_global,&pcbddc->original_rhs);CHKERRQ(ierr); 95262a6ff1dSStefano Zampini } 95362a6ff1dSStefano Zampini if (!pcbddc->temp_solution) { 95462a6ff1dSStefano Zampini ierr = VecDuplicate(pcis->vec1_global,&pcbddc->temp_solution);CHKERRQ(ierr); 95562a6ff1dSStefano Zampini } 9568d00608fSStefano Zampini 9573972b0daSStefano Zampini if (x) { 9583972b0daSStefano Zampini ierr = PetscObjectReference((PetscObject)x);CHKERRQ(ierr); 9593972b0daSStefano Zampini used_vec = x; 9608d00608fSStefano Zampini } else { /* it can only happen when calling PCBDDCMatFETIDPGetRHS */ 9613972b0daSStefano Zampini ierr = PetscObjectReference((PetscObject)pcbddc->temp_solution);CHKERRQ(ierr); 9623972b0daSStefano Zampini used_vec = pcbddc->temp_solution; 9633972b0daSStefano Zampini ierr = VecSet(used_vec,0.0);CHKERRQ(ierr); 9643972b0daSStefano Zampini } 9658efcfb23SStefano Zampini 9668efcfb23SStefano Zampini /* hack into ksp data structure since PCPreSolve comes earlier than setting to zero the guess in src/ksp/ksp/interface/itfunc.c */ 9673972b0daSStefano Zampini if (ksp) { 968a0cb1b98SStefano Zampini /* store the flag for the initial guess since it will be restored back during PCPostSolve_BDDC */ 9698efcfb23SStefano Zampini ierr = KSPGetInitialGuessNonzero(ksp,&pcbddc->ksp_guess_nonzero);CHKERRQ(ierr); 9708efcfb23SStefano Zampini if (!pcbddc->ksp_guess_nonzero) { 9713972b0daSStefano Zampini ierr = VecSet(used_vec,0.0);CHKERRQ(ierr); 9723972b0daSStefano Zampini } 9733972b0daSStefano Zampini } 9743308cffdSStefano Zampini 9758d00608fSStefano Zampini pcbddc->rhs_change = PETSC_FALSE; 9763972b0daSStefano Zampini 9773972b0daSStefano Zampini /* Take into account zeroed rows -> change rhs and store solution removed */ 978a07ea27aSStefano Zampini if (rhs) { 9793975b054SStefano Zampini IS dirIS = NULL; 9803975b054SStefano Zampini 981a07ea27aSStefano Zampini /* DirichletBoundariesLocal may not be consistent among neighbours; gets a dirichlet dofs IS from graph (may be cached) */ 9823975b054SStefano Zampini ierr = PCBDDCGraphGetDirichletDofs(pcbddc->mat_graph,&dirIS);CHKERRQ(ierr); 9833975b054SStefano Zampini if (dirIS) { 984906d46d4SStefano Zampini Mat_IS *matis = (Mat_IS*)pc->pmat->data; 985785d1243SStefano Zampini PetscInt dirsize,i,*is_indices; 9862b095fd8SStefano Zampini PetscScalar *array_x; 9872b095fd8SStefano Zampini const PetscScalar *array_diagonal; 988785d1243SStefano Zampini 9893972b0daSStefano Zampini ierr = MatGetDiagonal(pc->pmat,pcis->vec1_global);CHKERRQ(ierr); 9903972b0daSStefano Zampini ierr = VecPointwiseDivide(pcis->vec1_global,rhs,pcis->vec1_global);CHKERRQ(ierr); 991e176bc59SStefano Zampini ierr = VecScatterBegin(matis->rctx,pcis->vec1_global,pcis->vec2_N,INSERT_VALUES,SCATTER_FORWARD);CHKERRQ(ierr); 992e176bc59SStefano Zampini ierr = VecScatterEnd(matis->rctx,pcis->vec1_global,pcis->vec2_N,INSERT_VALUES,SCATTER_FORWARD);CHKERRQ(ierr); 993e176bc59SStefano Zampini ierr = VecScatterBegin(matis->rctx,used_vec,pcis->vec1_N,INSERT_VALUES,SCATTER_FORWARD);CHKERRQ(ierr); 994e176bc59SStefano Zampini ierr = VecScatterEnd(matis->rctx,used_vec,pcis->vec1_N,INSERT_VALUES,SCATTER_FORWARD);CHKERRQ(ierr); 99582ba6b80SStefano Zampini ierr = ISGetLocalSize(dirIS,&dirsize);CHKERRQ(ierr); 9963972b0daSStefano Zampini ierr = VecGetArray(pcis->vec1_N,&array_x);CHKERRQ(ierr); 9972b095fd8SStefano Zampini ierr = VecGetArrayRead(pcis->vec2_N,&array_diagonal);CHKERRQ(ierr); 9983972b0daSStefano Zampini ierr = ISGetIndices(dirIS,(const PetscInt**)&is_indices);CHKERRQ(ierr); 9992fa5cd67SKarl Rupp for (i=0; i<dirsize; i++) array_x[is_indices[i]] = array_diagonal[is_indices[i]]; 10003972b0daSStefano Zampini ierr = ISRestoreIndices(dirIS,(const PetscInt**)&is_indices);CHKERRQ(ierr); 10012b095fd8SStefano Zampini ierr = VecRestoreArrayRead(pcis->vec2_N,&array_diagonal);CHKERRQ(ierr); 10023972b0daSStefano Zampini ierr = VecRestoreArray(pcis->vec1_N,&array_x);CHKERRQ(ierr); 1003e176bc59SStefano Zampini ierr = VecScatterBegin(matis->rctx,pcis->vec1_N,used_vec,INSERT_VALUES,SCATTER_REVERSE);CHKERRQ(ierr); 1004e176bc59SStefano Zampini ierr = VecScatterEnd(matis->rctx,pcis->vec1_N,used_vec,INSERT_VALUES,SCATTER_REVERSE);CHKERRQ(ierr); 10058d00608fSStefano Zampini pcbddc->rhs_change = PETSC_TRUE; 10061b968477SStefano Zampini ierr = ISDestroy(&dirIS);CHKERRQ(ierr); 10078efcfb23SStefano Zampini } 1008a07ea27aSStefano Zampini } 1009b76ba322SStefano Zampini 10108efcfb23SStefano Zampini /* remove the computed solution or the initial guess from the rhs */ 10118d00608fSStefano Zampini if (pcbddc->rhs_change || (ksp && pcbddc->ksp_guess_nonzero) ) { 10128d00608fSStefano Zampini /* store the original rhs */ 10138d00608fSStefano Zampini if (copy_rhs) { 10148d00608fSStefano Zampini ierr = VecCopy(rhs,pcbddc->original_rhs);CHKERRQ(ierr); 10158d00608fSStefano Zampini copy_rhs = PETSC_FALSE; 10168d00608fSStefano Zampini } 10178d00608fSStefano Zampini pcbddc->rhs_change = PETSC_TRUE; 10183972b0daSStefano Zampini ierr = VecScale(used_vec,-1.0);CHKERRQ(ierr); 10193972b0daSStefano Zampini ierr = MatMultAdd(pc->pmat,used_vec,rhs,rhs);CHKERRQ(ierr); 10203972b0daSStefano Zampini ierr = VecScale(used_vec,-1.0);CHKERRQ(ierr); 10218efcfb23SStefano Zampini ierr = VecCopy(used_vec,pcbddc->temp_solution);CHKERRQ(ierr); 10227acc28cbSStefano Zampini if (ksp) { 10237acc28cbSStefano Zampini ierr = KSPSetInitialGuessNonzero(ksp,PETSC_FALSE);CHKERRQ(ierr); 10247acc28cbSStefano Zampini } 10253308cffdSStefano Zampini } 10268efcfb23SStefano Zampini ierr = VecDestroy(&used_vec);CHKERRQ(ierr); 1027b76ba322SStefano Zampini 1028b76ba322SStefano Zampini /* store partially computed solution and set initial guess */ 1029b097fa66SStefano Zampini if (x && pcbddc->use_exact_dirichlet_trick) { 10308efcfb23SStefano Zampini ierr = VecSet(x,0.0);CHKERRQ(ierr); 1031b76ba322SStefano Zampini ierr = VecScatterBegin(pcis->global_to_D,rhs,pcis->vec1_D,INSERT_VALUES,SCATTER_FORWARD);CHKERRQ(ierr); 1032b76ba322SStefano Zampini ierr = VecScatterEnd(pcis->global_to_D,rhs,pcis->vec1_D,INSERT_VALUES,SCATTER_FORWARD);CHKERRQ(ierr); 1033b76ba322SStefano Zampini ierr = KSPSolve(pcbddc->ksp_D,pcis->vec1_D,pcis->vec2_D);CHKERRQ(ierr); 10348efcfb23SStefano Zampini ierr = VecScatterBegin(pcis->global_to_D,pcis->vec2_D,x,INSERT_VALUES,SCATTER_REVERSE);CHKERRQ(ierr); 10358efcfb23SStefano Zampini ierr = VecScatterEnd(pcis->global_to_D,pcis->vec2_D,x,INSERT_VALUES,SCATTER_REVERSE);CHKERRQ(ierr); 1036a0cb1b98SStefano Zampini if (ksp) { 1037b76ba322SStefano Zampini ierr = KSPSetInitialGuessNonzero(ksp,PETSC_TRUE);CHKERRQ(ierr); 1038b76ba322SStefano Zampini } 1039b76ba322SStefano Zampini } 1040b76ba322SStefano Zampini 1041b9b85e73SStefano Zampini if (pcbddc->ChangeOfBasisMatrix) { 1042906d46d4SStefano Zampini PCBDDCChange_ctx change_ctx; 1043906d46d4SStefano Zampini 1044906d46d4SStefano Zampini /* get change ctx */ 1045906d46d4SStefano Zampini ierr = MatShellGetContext(pcbddc->new_global_mat,&change_ctx);CHKERRQ(ierr); 1046906d46d4SStefano Zampini 1047906d46d4SStefano Zampini /* set current iteration matrix inside change context (change of basis has been already set into the ctx during PCSetUp) */ 1048906d46d4SStefano Zampini ierr = MatDestroy(&change_ctx->original_mat);CHKERRQ(ierr); 1049906d46d4SStefano Zampini ierr = PetscObjectReference((PetscObject)pc->mat);CHKERRQ(ierr); 1050906d46d4SStefano Zampini change_ctx->original_mat = pc->mat; 1051906d46d4SStefano Zampini 1052906d46d4SStefano Zampini /* change iteration matrix */ 1053906d46d4SStefano Zampini ierr = MatDestroy(&pc->mat);CHKERRQ(ierr); 1054906d46d4SStefano Zampini ierr = PetscObjectReference((PetscObject)pcbddc->new_global_mat);CHKERRQ(ierr); 1055906d46d4SStefano Zampini pc->mat = pcbddc->new_global_mat; 1056906d46d4SStefano Zampini 10578d00608fSStefano Zampini /* store the original rhs */ 10588d00608fSStefano Zampini if (copy_rhs) { 10598d00608fSStefano Zampini ierr = VecCopy(rhs,pcbddc->original_rhs);CHKERRQ(ierr); 10608d00608fSStefano Zampini copy_rhs = PETSC_FALSE; 10618d00608fSStefano Zampini } 10628d00608fSStefano Zampini 1063906d46d4SStefano Zampini /* change rhs */ 1064906d46d4SStefano Zampini ierr = MatMultTranspose(change_ctx->global_change,rhs,pcis->vec1_global);CHKERRQ(ierr); 1065906d46d4SStefano Zampini ierr = VecCopy(pcis->vec1_global,rhs);CHKERRQ(ierr); 10668d00608fSStefano Zampini pcbddc->rhs_change = PETSC_TRUE; 106792e3dcfbSStefano Zampini } 106892e3dcfbSStefano Zampini 106992e3dcfbSStefano Zampini /* remove nullspace if present */ 10708efcfb23SStefano Zampini if (ksp && x && pcbddc->NullSpace) { 10718efcfb23SStefano Zampini ierr = MatNullSpaceRemove(pcbddc->NullSpace,x);CHKERRQ(ierr); 10728d00608fSStefano Zampini /* store the original rhs */ 10738d00608fSStefano Zampini if (copy_rhs) { 10748d00608fSStefano Zampini ierr = VecCopy(rhs,pcbddc->original_rhs);CHKERRQ(ierr); 10758d00608fSStefano Zampini copy_rhs = PETSC_FALSE; 10768d00608fSStefano Zampini } 10778d00608fSStefano Zampini pcbddc->rhs_change = PETSC_TRUE; 1078d0195637SJed Brown ierr = MatNullSpaceRemove(pcbddc->NullSpace,rhs);CHKERRQ(ierr); 1079b76ba322SStefano Zampini } 1080534831adSStefano Zampini PetscFunctionReturn(0); 1081534831adSStefano Zampini } 1082906d46d4SStefano Zampini 1083534831adSStefano Zampini /* -------------------------------------------------------------------------- */ 1084534831adSStefano Zampini #undef __FUNCT__ 1085534831adSStefano Zampini #define __FUNCT__ "PCPostSolve_BDDC" 1086534831adSStefano Zampini /* -------------------------------------------------------------------------- */ 1087534831adSStefano Zampini /* 1088534831adSStefano Zampini PCPostSolve_BDDC - Changes the computed solution if a transformation of basis 1089534831adSStefano Zampini approach has been selected. Also, restores rhs to its original state. 1090534831adSStefano Zampini 1091534831adSStefano Zampini Input Parameter: 1092534831adSStefano Zampini + pc - the preconditioner contex 1093534831adSStefano Zampini 1094534831adSStefano Zampini Application Interface Routine: PCPostSolve() 1095534831adSStefano Zampini 1096534831adSStefano Zampini Notes: 1097534831adSStefano Zampini The interface routine PCPostSolve() is not usually called directly by 1098534831adSStefano Zampini the user, but instead is called by KSPSolve(). 1099534831adSStefano Zampini */ 1100534831adSStefano Zampini static PetscErrorCode PCPostSolve_BDDC(PC pc, KSP ksp, Vec rhs, Vec x) 1101534831adSStefano Zampini { 1102534831adSStefano Zampini PetscErrorCode ierr; 1103534831adSStefano Zampini PC_BDDC *pcbddc = (PC_BDDC*)pc->data; 1104534831adSStefano Zampini 1105534831adSStefano Zampini PetscFunctionBegin; 1106b9b85e73SStefano Zampini if (pcbddc->ChangeOfBasisMatrix) { 1107906d46d4SStefano Zampini PCBDDCChange_ctx change_ctx; 1108906d46d4SStefano Zampini 1109906d46d4SStefano Zampini /* get change ctx */ 1110906d46d4SStefano Zampini ierr = MatShellGetContext(pcbddc->new_global_mat,&change_ctx);CHKERRQ(ierr); 1111906d46d4SStefano Zampini 1112906d46d4SStefano Zampini /* restore iteration matrix */ 1113906d46d4SStefano Zampini ierr = MatDestroy(&pc->mat);CHKERRQ(ierr); 1114906d46d4SStefano Zampini ierr = PetscObjectReference((PetscObject)change_ctx->original_mat);CHKERRQ(ierr); 1115906d46d4SStefano Zampini pc->mat = change_ctx->original_mat; 1116906d46d4SStefano Zampini 1117906d46d4SStefano Zampini /* get solution in original basis */ 1118906d46d4SStefano Zampini if (x) { 1119906d46d4SStefano Zampini PC_IS *pcis = (PC_IS*)(pc->data); 1120906d46d4SStefano Zampini ierr = MatMult(change_ctx->global_change,x,pcis->vec1_global);CHKERRQ(ierr); 1121906d46d4SStefano Zampini ierr = VecCopy(pcis->vec1_global,x);CHKERRQ(ierr); 11223425bc38SStefano Zampini } 1123534831adSStefano Zampini } 1124906d46d4SStefano Zampini 11253972b0daSStefano Zampini /* add solution removed in presolve */ 11266bcfc461SStefano Zampini if (x && pcbddc->rhs_change) { 11273425bc38SStefano Zampini ierr = VecAXPY(x,1.0,pcbddc->temp_solution);CHKERRQ(ierr); 11283425bc38SStefano Zampini } 1129906d46d4SStefano Zampini 1130fb223d50SStefano Zampini /* restore rhs to its original state */ 11318d00608fSStefano Zampini if (rhs && pcbddc->rhs_change) { 1132fb223d50SStefano Zampini ierr = VecCopy(pcbddc->original_rhs,rhs);CHKERRQ(ierr); 1133fb223d50SStefano Zampini } 11348d00608fSStefano Zampini pcbddc->rhs_change = PETSC_FALSE; 11358efcfb23SStefano Zampini 11368efcfb23SStefano Zampini /* restore ksp guess state */ 11378efcfb23SStefano Zampini if (ksp) { 11388efcfb23SStefano Zampini ierr = KSPSetInitialGuessNonzero(ksp,pcbddc->ksp_guess_nonzero);CHKERRQ(ierr); 11398efcfb23SStefano Zampini } 1140534831adSStefano Zampini PetscFunctionReturn(0); 1141534831adSStefano Zampini } 1142534831adSStefano Zampini /* -------------------------------------------------------------------------- */ 114353cdbc3dSStefano Zampini #undef __FUNCT__ 114453cdbc3dSStefano Zampini #define __FUNCT__ "PCSetUp_BDDC" 11450c7d97c5SJed Brown /* -------------------------------------------------------------------------- */ 11460c7d97c5SJed Brown /* 11470c7d97c5SJed Brown PCSetUp_BDDC - Prepares for the use of the BDDC preconditioner 11480c7d97c5SJed Brown by setting data structures and options. 11490c7d97c5SJed Brown 11500c7d97c5SJed Brown Input Parameter: 115153cdbc3dSStefano Zampini + pc - the preconditioner context 11520c7d97c5SJed Brown 11530c7d97c5SJed Brown Application Interface Routine: PCSetUp() 11540c7d97c5SJed Brown 11550c7d97c5SJed Brown Notes: 11560c7d97c5SJed Brown The interface routine PCSetUp() is not usually called directly by 11570c7d97c5SJed Brown the user, but instead is called by PCApply() if necessary. 11580c7d97c5SJed Brown */ 115953cdbc3dSStefano Zampini PetscErrorCode PCSetUp_BDDC(PC pc) 11600c7d97c5SJed Brown { 11610c7d97c5SJed Brown PetscErrorCode ierr; 11620c7d97c5SJed Brown PC_BDDC* pcbddc = (PC_BDDC*)pc->data; 11635e8657edSStefano Zampini Mat_IS* matis; 116408122e43SStefano Zampini MatNullSpace nearnullspace; 116591e8d312SStefano Zampini PetscInt nrows,ncols; 116608122e43SStefano Zampini PetscBool computetopography,computesolvers,computesubschurs; 11678de1fae6SStefano Zampini PetscBool computeconstraintsmatrix; 11685e8657edSStefano Zampini PetscBool new_nearnullspace_provided,ismatis; 11690c7d97c5SJed Brown 11700c7d97c5SJed Brown PetscFunctionBegin; 11715e8657edSStefano Zampini ierr = PetscObjectTypeCompare((PetscObject)pc->pmat,MATIS,&ismatis);CHKERRQ(ierr); 1172*6c4ed002SBarry Smith if (!ismatis) SETERRQ(PetscObjectComm((PetscObject)pc),PETSC_ERR_ARG_WRONG,"PCBDDC preconditioner requires matrix of type MATIS"); 117391e8d312SStefano Zampini ierr = MatGetSize(pc->pmat,&nrows,&ncols);CHKERRQ(ierr); 1174*6c4ed002SBarry Smith if (nrows != ncols) SETERRQ(PetscObjectComm((PetscObject)pc),PETSC_ERR_SUP,"PCBDDC preconditioner requires a square preconditioning matrix"); 11755e8657edSStefano Zampini matis = (Mat_IS*)pc->pmat->data; 1176f4ddd8eeSStefano Zampini /* the following lines of code should be replaced by a better logic between PCIS, PCNN, PCBDDC and other future nonoverlapping preconditioners */ 11773b03a366Sstefano_zampini /* For BDDC we need to define a local "Neumann" problem different to that defined in PCISSetup 1178fcd409f5SStefano Zampini Also, BDDC directly build the Dirichlet problem */ 1179f4ddd8eeSStefano Zampini /* split work */ 1180674ae819SStefano Zampini if (pc->setupcalled) { 1181d1e9a80fSBarry Smith if (pc->flag == SAME_NONZERO_PATTERN) { 1182674ae819SStefano Zampini computetopography = PETSC_FALSE; 1183674ae819SStefano Zampini computesolvers = PETSC_TRUE; 1184674ae819SStefano Zampini } else { /* DIFFERENT_NONZERO_PATTERN */ 1185674ae819SStefano Zampini computetopography = PETSC_TRUE; 1186674ae819SStefano Zampini computesolvers = PETSC_TRUE; 1187674ae819SStefano Zampini } 1188674ae819SStefano Zampini } else { 1189674ae819SStefano Zampini computetopography = PETSC_TRUE; 1190674ae819SStefano Zampini computesolvers = PETSC_TRUE; 1191674ae819SStefano Zampini } 1192fb180af4SStefano Zampini if (pcbddc->recompute_topography) { 1193fb180af4SStefano Zampini computetopography = PETSC_TRUE; 1194fb180af4SStefano Zampini } 11958de1fae6SStefano Zampini computeconstraintsmatrix = PETSC_FALSE; 1196*6c4ed002SBarry Smith if (pcbddc->adaptive_threshold > 0.0 && !pcbddc->use_deluxe_scaling) SETERRQ(PetscObjectComm((PetscObject)pc),PETSC_ERR_SUP,"Cannot compute adaptive constraints without deluxe scaling. Rerun with -pc_bddc_use_deluxe_scaling"); 11975a95e1ceSStefano Zampini pcbddc->adaptive_selection = (PetscBool)(pcbddc->adaptive_threshold > 0.0 && pcbddc->use_deluxe_scaling); 1198862806e4SStefano Zampini if (pcbddc->adaptive_selection) pcbddc->use_faces = PETSC_TRUE; 1199862806e4SStefano Zampini 12005a95e1ceSStefano Zampini computesubschurs = (PetscBool)(pcbddc->adaptive_selection || pcbddc->use_deluxe_scaling); 1201*6c4ed002SBarry Smith if (pcbddc->faster_deluxe && pcbddc->adaptive_selection && pcbddc->use_change_of_basis) SETERRQ(PetscObjectComm((PetscObject)pc),PETSC_ERR_SUP,"Cannot compute faster deluxe if adaptivity and change of basis are both requested. Rerun with -pc_bddc_deluxe_faster false"); 1202*6c4ed002SBarry Smith 1203f4ddd8eeSStefano Zampini /* Get stdout for dbg */ 120470cf5478SStefano Zampini if (pcbddc->dbg_flag) { 120570cf5478SStefano Zampini if (!pcbddc->dbg_viewer) { 120658a03d70SStefano Zampini pcbddc->dbg_viewer = PETSC_VIEWER_STDOUT_(PetscObjectComm((PetscObject)pc)); 12071575c14dSBarry Smith ierr = PetscViewerASCIIPushSynchronized(pcbddc->dbg_viewer);CHKERRQ(ierr); 1208f4ddd8eeSStefano Zampini } 120958a03d70SStefano Zampini ierr = PetscViewerASCIIAddTab(pcbddc->dbg_viewer,2*pcbddc->current_level);CHKERRQ(ierr); 1210f4ddd8eeSStefano Zampini } 1211f4ddd8eeSStefano Zampini 12125e8657edSStefano Zampini if (pcbddc->user_ChangeOfBasisMatrix) { 12135e8657edSStefano Zampini /* use_change_of_basis flag is used to automatically compute a change of basis from constraints */ 12145e8657edSStefano Zampini pcbddc->use_change_of_basis = PETSC_FALSE; 12155e8657edSStefano Zampini ierr = PCBDDCComputeLocalMatrix(pc,pcbddc->user_ChangeOfBasisMatrix);CHKERRQ(ierr); 12165e8657edSStefano Zampini } else { 1217b96c3477SStefano Zampini ierr = MatDestroy(&pcbddc->local_mat);CHKERRQ(ierr); 12185e8657edSStefano Zampini ierr = PetscObjectReference((PetscObject)matis->A);CHKERRQ(ierr); 12195e8657edSStefano Zampini pcbddc->local_mat = matis->A; 1220674ae819SStefano Zampini } 1221f4ddd8eeSStefano Zampini 1222e496cd5dSStefano Zampini /* workaround for reals */ 1223e496cd5dSStefano Zampini #if !defined(PETSC_USE_COMPLEX) 12243301b35fSStefano Zampini if (matis->A->symmetric_set) { 12253301b35fSStefano Zampini ierr = MatSetOption(pcbddc->local_mat,MAT_HERMITIAN,matis->A->symmetric);CHKERRQ(ierr); 12263301b35fSStefano Zampini } 1227e496cd5dSStefano Zampini #endif 1228e496cd5dSStefano Zampini 12295e8657edSStefano Zampini /* Set up all the "iterative substructuring" common block without computing solvers */ 12305e8657edSStefano Zampini { 12315e8657edSStefano Zampini Mat temp_mat; 12325e8657edSStefano Zampini 12335e8657edSStefano Zampini temp_mat = matis->A; 12345e8657edSStefano Zampini matis->A = pcbddc->local_mat; 12355e8657edSStefano Zampini ierr = PCISSetUp(pc,PETSC_FALSE);CHKERRQ(ierr); 12365e8657edSStefano Zampini pcbddc->local_mat = matis->A; 12375e8657edSStefano Zampini matis->A = temp_mat; 12385e8657edSStefano Zampini } 1239684f6988SStefano Zampini 1240b96c3477SStefano Zampini /* Analyze interface and setup sub_schurs data */ 1241674ae819SStefano Zampini if (computetopography) { 1242674ae819SStefano Zampini ierr = PCBDDCAnalyzeInterface(pc);CHKERRQ(ierr); 12438de1fae6SStefano Zampini computeconstraintsmatrix = PETSC_TRUE; 1244674ae819SStefano Zampini } 1245fb8d54d4SStefano Zampini 1246b96c3477SStefano Zampini /* Setup local dirichlet solver ksp_D and sub_schurs solvers */ 1247684f6988SStefano Zampini if (computesolvers) { 1248d5574798SStefano Zampini PCBDDCSubSchurs sub_schurs=pcbddc->sub_schurs; 1249d5574798SStefano Zampini 1250d5574798SStefano Zampini if (computesubschurs && computetopography) { 125108122e43SStefano Zampini ierr = PCBDDCInitSubSchurs(pc);CHKERRQ(ierr); 1252b1b3d7a2SStefano Zampini } 1253d5574798SStefano Zampini if (sub_schurs->use_mumps) { 12542070dbb6SStefano Zampini if (computesubschurs) { 125508122e43SStefano Zampini ierr = PCBDDCSetUpSubSchurs(pc);CHKERRQ(ierr); 12562070dbb6SStefano Zampini } 1257d5574798SStefano Zampini ierr = PCBDDCSetUpLocalSolvers(pc,PETSC_TRUE,PETSC_FALSE);CHKERRQ(ierr); 1258d5574798SStefano Zampini } else { 1259d5574798SStefano Zampini ierr = PCBDDCSetUpLocalSolvers(pc,PETSC_TRUE,PETSC_FALSE);CHKERRQ(ierr); 12602070dbb6SStefano Zampini if (computesubschurs) { 1261d5574798SStefano Zampini ierr = PCBDDCSetUpSubSchurs(pc);CHKERRQ(ierr); 1262d5574798SStefano Zampini } 12632070dbb6SStefano Zampini } 126408122e43SStefano Zampini if (pcbddc->adaptive_selection) { 126508122e43SStefano Zampini ierr = PCBDDCAdaptiveSelection(pc);CHKERRQ(ierr); 12668de1fae6SStefano Zampini computeconstraintsmatrix = PETSC_TRUE; 1267b7eb3628SStefano Zampini } 1268b1b3d7a2SStefano Zampini } 1269684f6988SStefano Zampini 1270f4ddd8eeSStefano Zampini /* infer if NullSpace object attached to Mat via MatSetNearNullSpace has changed */ 1271fb8d54d4SStefano Zampini new_nearnullspace_provided = PETSC_FALSE; 1272f4ddd8eeSStefano Zampini ierr = MatGetNearNullSpace(pc->pmat,&nearnullspace);CHKERRQ(ierr); 1273f4ddd8eeSStefano Zampini if (pcbddc->onearnullspace) { /* already used nearnullspace */ 1274f4ddd8eeSStefano Zampini if (!nearnullspace) { /* near null space attached to mat has been destroyed */ 1275f4ddd8eeSStefano Zampini new_nearnullspace_provided = PETSC_TRUE; 1276f4ddd8eeSStefano Zampini } else { 1277f4ddd8eeSStefano Zampini /* determine if the two nullspaces are different (should be lightweight) */ 1278f4ddd8eeSStefano Zampini if (nearnullspace != pcbddc->onearnullspace) { 1279f4ddd8eeSStefano Zampini new_nearnullspace_provided = PETSC_TRUE; 1280165b64e2SStefano Zampini } else { /* maybe the user has changed the content of the nearnullspace so check vectors ObjectStateId */ 1281f4ddd8eeSStefano Zampini PetscInt i; 1282165b64e2SStefano Zampini const Vec *nearnullvecs; 1283165b64e2SStefano Zampini PetscObjectState state; 1284165b64e2SStefano Zampini PetscInt nnsp_size; 1285165b64e2SStefano Zampini ierr = MatNullSpaceGetVecs(nearnullspace,NULL,&nnsp_size,&nearnullvecs);CHKERRQ(ierr); 1286f4ddd8eeSStefano Zampini for (i=0;i<nnsp_size;i++) { 1287f4ddd8eeSStefano Zampini ierr = PetscObjectStateGet((PetscObject)nearnullvecs[i],&state);CHKERRQ(ierr); 1288165b64e2SStefano Zampini if (pcbddc->onearnullvecs_state[i] != state) { 1289f4ddd8eeSStefano Zampini new_nearnullspace_provided = PETSC_TRUE; 1290f4ddd8eeSStefano Zampini break; 1291f4ddd8eeSStefano Zampini } 1292f4ddd8eeSStefano Zampini } 1293f4ddd8eeSStefano Zampini } 1294f4ddd8eeSStefano Zampini } 1295f4ddd8eeSStefano Zampini } else { 1296f4ddd8eeSStefano Zampini if (!nearnullspace) { /* both nearnullspaces are null */ 1297f4ddd8eeSStefano Zampini new_nearnullspace_provided = PETSC_FALSE; 1298f4ddd8eeSStefano Zampini } else { /* nearnullspace attached later */ 1299f4ddd8eeSStefano Zampini new_nearnullspace_provided = PETSC_TRUE; 1300f4ddd8eeSStefano Zampini } 1301f4ddd8eeSStefano Zampini } 1302f4ddd8eeSStefano Zampini 1303f4ddd8eeSStefano Zampini /* Setup constraints and related work vectors */ 1304727cdba6SStefano Zampini /* reset primal space flags */ 1305f4ddd8eeSStefano Zampini pcbddc->new_primal_space = PETSC_FALSE; 1306727cdba6SStefano Zampini pcbddc->new_primal_space_local = PETSC_FALSE; 13078de1fae6SStefano Zampini if (computeconstraintsmatrix || new_nearnullspace_provided) { 1308727cdba6SStefano Zampini /* It also sets the primal space flags */ 1309674ae819SStefano Zampini ierr = PCBDDCConstraintsSetUp(pc);CHKERRQ(ierr); 1310e7b262bdSStefano Zampini /* Allocate needed local vectors (which depends on quantities defined during ConstraintsSetUp) */ 1311f4ddd8eeSStefano Zampini ierr = PCBDDCSetUpLocalWorkVectors(pc);CHKERRQ(ierr); 13123975b054SStefano Zampini } 13135e8657edSStefano Zampini 13143975b054SStefano Zampini if (computesolvers || pcbddc->new_primal_space) { 13155e8657edSStefano Zampini if (pcbddc->use_change_of_basis) { 13165e8657edSStefano Zampini PC_IS *pcis = (PC_IS*)(pc->data); 13175e8657edSStefano Zampini 13185e8657edSStefano Zampini ierr = PCBDDCComputeLocalMatrix(pc,pcbddc->ChangeOfBasisMatrix);CHKERRQ(ierr); 13195e8657edSStefano Zampini /* get submatrices */ 13205e8657edSStefano Zampini ierr = MatDestroy(&pcis->A_IB);CHKERRQ(ierr); 13215e8657edSStefano Zampini ierr = MatDestroy(&pcis->A_BI);CHKERRQ(ierr); 13225e8657edSStefano Zampini ierr = MatDestroy(&pcis->A_BB);CHKERRQ(ierr); 13235e8657edSStefano Zampini ierr = MatGetSubMatrix(pcbddc->local_mat,pcis->is_B_local,pcis->is_B_local,MAT_INITIAL_MATRIX,&pcis->A_BB);CHKERRQ(ierr); 13245e8657edSStefano Zampini ierr = MatGetSubMatrix(pcbddc->local_mat,pcis->is_I_local,pcis->is_B_local,MAT_INITIAL_MATRIX,&pcis->A_IB);CHKERRQ(ierr); 13255e8657edSStefano Zampini ierr = MatGetSubMatrix(pcbddc->local_mat,pcis->is_B_local,pcis->is_I_local,MAT_INITIAL_MATRIX,&pcis->A_BI);CHKERRQ(ierr); 13263975b054SStefano Zampini /* set flag in pcis to not reuse submatrices during PCISCreate */ 13273975b054SStefano Zampini pcis->reusesubmatrices = PETSC_FALSE; 13285e8657edSStefano Zampini } else if (!pcbddc->user_ChangeOfBasisMatrix) { 1329b96c3477SStefano Zampini ierr = MatDestroy(&pcbddc->local_mat);CHKERRQ(ierr); 13305e8657edSStefano Zampini ierr = PetscObjectReference((PetscObject)matis->A);CHKERRQ(ierr); 13315e8657edSStefano Zampini pcbddc->local_mat = matis->A; 13325e8657edSStefano Zampini } 1333b96c3477SStefano Zampini /* SetUp coarse and local Neumann solvers */ 133499cc7994SStefano Zampini ierr = PCBDDCSetUpSolvers(pc);CHKERRQ(ierr); 1335b96c3477SStefano Zampini /* SetUp Scaling operator */ 1336674ae819SStefano Zampini ierr = PCBDDCScalingSetUp(pc);CHKERRQ(ierr); 13370c7d97c5SJed Brown } 133870cf5478SStefano Zampini 133958a03d70SStefano Zampini if (pcbddc->dbg_flag) { 134058a03d70SStefano Zampini ierr = PetscViewerASCIISubtractTab(pcbddc->dbg_viewer,2*pcbddc->current_level);CHKERRQ(ierr); 13412b510759SStefano Zampini } 13420c7d97c5SJed Brown PetscFunctionReturn(0); 13430c7d97c5SJed Brown } 13440c7d97c5SJed Brown 13450c7d97c5SJed Brown /* -------------------------------------------------------------------------- */ 13460c7d97c5SJed Brown /* 134750efa1b5SStefano Zampini PCApply_BDDC - Applies the BDDC operator to a vector. 13480c7d97c5SJed Brown 13490c7d97c5SJed Brown Input Parameters: 13500f202f7eSStefano Zampini + pc - the preconditioner context 13510f202f7eSStefano Zampini - r - input vector (global) 13520c7d97c5SJed Brown 13530c7d97c5SJed Brown Output Parameter: 13540c7d97c5SJed Brown . z - output vector (global) 13550c7d97c5SJed Brown 13560c7d97c5SJed Brown Application Interface Routine: PCApply() 13570c7d97c5SJed Brown */ 13580c7d97c5SJed Brown #undef __FUNCT__ 13590c7d97c5SJed Brown #define __FUNCT__ "PCApply_BDDC" 136053cdbc3dSStefano Zampini PetscErrorCode PCApply_BDDC(PC pc,Vec r,Vec z) 13610c7d97c5SJed Brown { 13620c7d97c5SJed Brown PC_IS *pcis = (PC_IS*)(pc->data); 13630c7d97c5SJed Brown PC_BDDC *pcbddc = (PC_BDDC*)(pc->data); 1364b097fa66SStefano Zampini PetscInt n_B = pcis->n_B, n_D = pcis->n - n_B; 13650c7d97c5SJed Brown PetscErrorCode ierr; 13663b03a366Sstefano_zampini const PetscScalar one = 1.0; 13673b03a366Sstefano_zampini const PetscScalar m_one = -1.0; 13682617d88aSStefano Zampini const PetscScalar zero = 0.0; 13690c7d97c5SJed Brown 13700c7d97c5SJed Brown /* This code is similar to that provided in nn.c for PCNN 13710c7d97c5SJed Brown NN interface preconditioner changed to BDDC 1372b097fa66SStefano Zampini Added support for M_3 preconditioner in the reference article (code is active if pcbddc->switch_static == PETSC_TRUE) */ 13730c7d97c5SJed Brown 13740c7d97c5SJed Brown PetscFunctionBegin; 137585c4d303SStefano Zampini if (!pcbddc->use_exact_dirichlet_trick) { 1376b097fa66SStefano Zampini ierr = VecCopy(r,z);CHKERRQ(ierr); 13770c7d97c5SJed Brown /* First Dirichlet solve */ 13780c7d97c5SJed Brown ierr = VecScatterBegin(pcis->global_to_D,r,pcis->vec1_D,INSERT_VALUES,SCATTER_FORWARD);CHKERRQ(ierr); 13790c7d97c5SJed Brown ierr = VecScatterEnd(pcis->global_to_D,r,pcis->vec1_D,INSERT_VALUES,SCATTER_FORWARD);CHKERRQ(ierr); 13800c7d97c5SJed Brown /* 13810c7d97c5SJed Brown Assembling right hand side for BDDC operator 1382b097fa66SStefano Zampini - pcis->vec1_D for the Dirichlet part (if needed, i.e. pcbddc->switch_static == PETSC_TRUE) 1383674ae819SStefano Zampini - pcis->vec1_B the interface part of the global vector z 13840c7d97c5SJed Brown */ 1385b097fa66SStefano Zampini if (n_D) { 1386b097fa66SStefano Zampini ierr = KSPSolve(pcbddc->ksp_D,pcis->vec1_D,pcis->vec2_D);CHKERRQ(ierr); 13870c7d97c5SJed Brown ierr = VecScale(pcis->vec2_D,m_one);CHKERRQ(ierr); 13888eeda7d8SStefano Zampini if (pcbddc->switch_static) { ierr = MatMultAdd(pcis->A_II,pcis->vec2_D,pcis->vec1_D,pcis->vec1_D);CHKERRQ(ierr); } 1389b097fa66SStefano Zampini ierr = MatMult(pcis->A_BI,pcis->vec2_D,pcis->vec1_B);CHKERRQ(ierr); 1390b097fa66SStefano Zampini } else { 1391b097fa66SStefano Zampini ierr = VecSet(pcis->vec1_B,zero);CHKERRQ(ierr); 1392b097fa66SStefano Zampini } 13930c7d97c5SJed Brown ierr = VecScatterBegin(pcis->global_to_B,pcis->vec1_B,z,ADD_VALUES,SCATTER_REVERSE);CHKERRQ(ierr); 13940c7d97c5SJed Brown ierr = VecScatterEnd(pcis->global_to_B,pcis->vec1_B,z,ADD_VALUES,SCATTER_REVERSE);CHKERRQ(ierr); 1395674ae819SStefano Zampini ierr = PCBDDCScalingRestriction(pc,z,pcis->vec1_B);CHKERRQ(ierr); 1396b76ba322SStefano Zampini } else { 1397b097fa66SStefano Zampini if (pcbddc->switch_static) { 13980bdf917eSStefano Zampini ierr = VecSet(pcis->vec1_D,zero);CHKERRQ(ierr); 1399b097fa66SStefano Zampini } 1400674ae819SStefano Zampini ierr = PCBDDCScalingRestriction(pc,r,pcis->vec1_B);CHKERRQ(ierr); 1401b76ba322SStefano Zampini } 1402b76ba322SStefano Zampini 14032617d88aSStefano Zampini /* Apply interface preconditioner 14042617d88aSStefano Zampini input/output vecs: pcis->vec1_B and pcis->vec1_D */ 1405dc359a40SStefano Zampini ierr = PCBDDCApplyInterfacePreconditioner(pc,PETSC_FALSE);CHKERRQ(ierr); 14062617d88aSStefano Zampini 1407674ae819SStefano Zampini /* Apply transpose of partition of unity operator */ 1408674ae819SStefano Zampini ierr = PCBDDCScalingExtension(pc,pcis->vec1_B,z);CHKERRQ(ierr); 14090c7d97c5SJed Brown 14103b03a366Sstefano_zampini /* Second Dirichlet solve and assembling of output */ 14110c7d97c5SJed Brown ierr = VecScatterBegin(pcis->global_to_B,z,pcis->vec1_B,INSERT_VALUES,SCATTER_FORWARD);CHKERRQ(ierr); 14120c7d97c5SJed Brown ierr = VecScatterEnd(pcis->global_to_B,z,pcis->vec1_B,INSERT_VALUES,SCATTER_FORWARD);CHKERRQ(ierr); 1413b097fa66SStefano Zampini if (n_B) { 14140c7d97c5SJed Brown ierr = MatMult(pcis->A_IB,pcis->vec1_B,pcis->vec3_D);CHKERRQ(ierr); 14158eeda7d8SStefano Zampini if (pcbddc->switch_static) { ierr = MatMultAdd(pcis->A_II,pcis->vec1_D,pcis->vec3_D,pcis->vec3_D);CHKERRQ(ierr); } 1416b097fa66SStefano Zampini } else if (pcbddc->switch_static) { 1417b097fa66SStefano Zampini ierr = MatMult(pcis->A_II,pcis->vec1_D,pcis->vec3_D);CHKERRQ(ierr); 1418b097fa66SStefano Zampini } 1419df187020SStefano Zampini ierr = KSPSolve(pcbddc->ksp_D,pcis->vec3_D,pcis->vec4_D);CHKERRQ(ierr); 1420b097fa66SStefano Zampini if (!pcbddc->use_exact_dirichlet_trick) { 1421b097fa66SStefano Zampini if (pcbddc->switch_static) { 1422b097fa66SStefano Zampini ierr = VecAXPBYPCZ(pcis->vec2_D,m_one,one,m_one,pcis->vec4_D,pcis->vec1_D);CHKERRQ(ierr); 1423b097fa66SStefano Zampini } else { 1424b097fa66SStefano Zampini ierr = VecAXPBY(pcis->vec2_D,m_one,m_one,pcis->vec4_D);CHKERRQ(ierr); 1425b097fa66SStefano Zampini } 14260c7d97c5SJed Brown ierr = VecScatterBegin(pcis->global_to_D,pcis->vec2_D,z,INSERT_VALUES,SCATTER_REVERSE);CHKERRQ(ierr); 14270c7d97c5SJed Brown ierr = VecScatterEnd(pcis->global_to_D,pcis->vec2_D,z,INSERT_VALUES,SCATTER_REVERSE);CHKERRQ(ierr); 1428b097fa66SStefano Zampini } else { 1429b097fa66SStefano Zampini if (pcbddc->switch_static) { 1430b097fa66SStefano Zampini ierr = VecAXPBY(pcis->vec4_D,one,m_one,pcis->vec1_D);CHKERRQ(ierr); 1431b097fa66SStefano Zampini } else { 1432b097fa66SStefano Zampini ierr = VecScale(pcis->vec4_D,m_one);CHKERRQ(ierr); 1433b097fa66SStefano Zampini } 1434b097fa66SStefano Zampini ierr = VecScatterBegin(pcis->global_to_D,pcis->vec4_D,z,INSERT_VALUES,SCATTER_REVERSE);CHKERRQ(ierr); 1435b097fa66SStefano Zampini ierr = VecScatterEnd(pcis->global_to_D,pcis->vec4_D,z,INSERT_VALUES,SCATTER_REVERSE);CHKERRQ(ierr); 1436b097fa66SStefano Zampini } 14370c7d97c5SJed Brown PetscFunctionReturn(0); 14380c7d97c5SJed Brown } 143950efa1b5SStefano Zampini 144050efa1b5SStefano Zampini /* -------------------------------------------------------------------------- */ 144150efa1b5SStefano Zampini /* 144250efa1b5SStefano Zampini PCApplyTranspose_BDDC - Applies the transpose of the BDDC operator to a vector. 144350efa1b5SStefano Zampini 144450efa1b5SStefano Zampini Input Parameters: 14450f202f7eSStefano Zampini + pc - the preconditioner context 14460f202f7eSStefano Zampini - r - input vector (global) 144750efa1b5SStefano Zampini 144850efa1b5SStefano Zampini Output Parameter: 144950efa1b5SStefano Zampini . z - output vector (global) 145050efa1b5SStefano Zampini 145150efa1b5SStefano Zampini Application Interface Routine: PCApplyTranspose() 145250efa1b5SStefano Zampini */ 145350efa1b5SStefano Zampini #undef __FUNCT__ 145450efa1b5SStefano Zampini #define __FUNCT__ "PCApplyTranspose_BDDC" 145550efa1b5SStefano Zampini PetscErrorCode PCApplyTranspose_BDDC(PC pc,Vec r,Vec z) 145650efa1b5SStefano Zampini { 145750efa1b5SStefano Zampini PC_IS *pcis = (PC_IS*)(pc->data); 145850efa1b5SStefano Zampini PC_BDDC *pcbddc = (PC_BDDC*)(pc->data); 1459b097fa66SStefano Zampini PetscInt n_B = pcis->n_B, n_D = pcis->n - n_B; 146050efa1b5SStefano Zampini PetscErrorCode ierr; 146150efa1b5SStefano Zampini const PetscScalar one = 1.0; 146250efa1b5SStefano Zampini const PetscScalar m_one = -1.0; 146350efa1b5SStefano Zampini const PetscScalar zero = 0.0; 146450efa1b5SStefano Zampini 146550efa1b5SStefano Zampini PetscFunctionBegin; 146650efa1b5SStefano Zampini if (!pcbddc->use_exact_dirichlet_trick) { 1467b097fa66SStefano Zampini ierr = VecCopy(r,z);CHKERRQ(ierr); 146850efa1b5SStefano Zampini /* First Dirichlet solve */ 146950efa1b5SStefano Zampini ierr = VecScatterBegin(pcis->global_to_D,r,pcis->vec1_D,INSERT_VALUES,SCATTER_FORWARD);CHKERRQ(ierr); 147050efa1b5SStefano Zampini ierr = VecScatterEnd(pcis->global_to_D,r,pcis->vec1_D,INSERT_VALUES,SCATTER_FORWARD);CHKERRQ(ierr); 147150efa1b5SStefano Zampini /* 147250efa1b5SStefano Zampini Assembling right hand side for BDDC operator 1473b097fa66SStefano Zampini - pcis->vec1_D for the Dirichlet part (if needed, i.e. pcbddc->switch_static == PETSC_TRUE) 147450efa1b5SStefano Zampini - pcis->vec1_B the interface part of the global vector z 147550efa1b5SStefano Zampini */ 1476b097fa66SStefano Zampini if (n_D) { 1477b097fa66SStefano Zampini ierr = KSPSolveTranspose(pcbddc->ksp_D,pcis->vec1_D,pcis->vec2_D);CHKERRQ(ierr); 147850efa1b5SStefano Zampini ierr = VecScale(pcis->vec2_D,m_one);CHKERRQ(ierr); 147950efa1b5SStefano Zampini if (pcbddc->switch_static) { ierr = MatMultTransposeAdd(pcis->A_II,pcis->vec2_D,pcis->vec1_D,pcis->vec1_D);CHKERRQ(ierr); } 1480b097fa66SStefano Zampini ierr = MatMultTranspose(pcis->A_IB,pcis->vec2_D,pcis->vec1_B);CHKERRQ(ierr); 1481b097fa66SStefano Zampini } else { 1482b097fa66SStefano Zampini ierr = VecSet(pcis->vec1_B,zero);CHKERRQ(ierr); 1483b097fa66SStefano Zampini } 148450efa1b5SStefano Zampini ierr = VecScatterBegin(pcis->global_to_B,pcis->vec1_B,z,ADD_VALUES,SCATTER_REVERSE);CHKERRQ(ierr); 148550efa1b5SStefano Zampini ierr = VecScatterEnd(pcis->global_to_B,pcis->vec1_B,z,ADD_VALUES,SCATTER_REVERSE);CHKERRQ(ierr); 148650efa1b5SStefano Zampini ierr = PCBDDCScalingRestriction(pc,z,pcis->vec1_B);CHKERRQ(ierr); 148750efa1b5SStefano Zampini } else { 1488b097fa66SStefano Zampini if (pcbddc->switch_static) { 148950efa1b5SStefano Zampini ierr = VecSet(pcis->vec1_D,zero);CHKERRQ(ierr); 1490b097fa66SStefano Zampini } 149150efa1b5SStefano Zampini ierr = PCBDDCScalingRestriction(pc,r,pcis->vec1_B);CHKERRQ(ierr); 149250efa1b5SStefano Zampini } 149350efa1b5SStefano Zampini 149450efa1b5SStefano Zampini /* Apply interface preconditioner 149550efa1b5SStefano Zampini input/output vecs: pcis->vec1_B and pcis->vec1_D */ 1496dc359a40SStefano Zampini ierr = PCBDDCApplyInterfacePreconditioner(pc,PETSC_TRUE);CHKERRQ(ierr); 149750efa1b5SStefano Zampini 149850efa1b5SStefano Zampini /* Apply transpose of partition of unity operator */ 149950efa1b5SStefano Zampini ierr = PCBDDCScalingExtension(pc,pcis->vec1_B,z);CHKERRQ(ierr); 150050efa1b5SStefano Zampini 150150efa1b5SStefano Zampini /* Second Dirichlet solve and assembling of output */ 150250efa1b5SStefano Zampini ierr = VecScatterBegin(pcis->global_to_B,z,pcis->vec1_B,INSERT_VALUES,SCATTER_FORWARD);CHKERRQ(ierr); 150350efa1b5SStefano Zampini ierr = VecScatterEnd(pcis->global_to_B,z,pcis->vec1_B,INSERT_VALUES,SCATTER_FORWARD);CHKERRQ(ierr); 1504b097fa66SStefano Zampini if (n_B) { 150550efa1b5SStefano Zampini ierr = MatMultTranspose(pcis->A_BI,pcis->vec1_B,pcis->vec3_D);CHKERRQ(ierr); 150650efa1b5SStefano Zampini if (pcbddc->switch_static) { ierr = MatMultTransposeAdd(pcis->A_II,pcis->vec1_D,pcis->vec3_D,pcis->vec3_D);CHKERRQ(ierr); } 1507b097fa66SStefano Zampini } else if (pcbddc->switch_static) { 1508b097fa66SStefano Zampini ierr = MatMultTranspose(pcis->A_II,pcis->vec1_D,pcis->vec3_D);CHKERRQ(ierr); 1509b097fa66SStefano Zampini } 1510b0147a47SStefano Zampini ierr = KSPSolveTranspose(pcbddc->ksp_D,pcis->vec3_D,pcis->vec4_D);CHKERRQ(ierr); 1511b097fa66SStefano Zampini if (!pcbddc->use_exact_dirichlet_trick) { 1512b097fa66SStefano Zampini if (pcbddc->switch_static) { 1513b097fa66SStefano Zampini ierr = VecAXPBYPCZ(pcis->vec2_D,m_one,one,m_one,pcis->vec4_D,pcis->vec1_D);CHKERRQ(ierr); 1514b097fa66SStefano Zampini } else { 1515b097fa66SStefano Zampini ierr = VecAXPBY(pcis->vec2_D,m_one,m_one,pcis->vec4_D);CHKERRQ(ierr); 1516b097fa66SStefano Zampini } 151750efa1b5SStefano Zampini ierr = VecScatterBegin(pcis->global_to_D,pcis->vec2_D,z,INSERT_VALUES,SCATTER_REVERSE);CHKERRQ(ierr); 151850efa1b5SStefano Zampini ierr = VecScatterEnd(pcis->global_to_D,pcis->vec2_D,z,INSERT_VALUES,SCATTER_REVERSE);CHKERRQ(ierr); 1519b097fa66SStefano Zampini } else { 1520b097fa66SStefano Zampini if (pcbddc->switch_static) { 1521b097fa66SStefano Zampini ierr = VecAXPBY(pcis->vec4_D,one,m_one,pcis->vec1_D);CHKERRQ(ierr); 1522b097fa66SStefano Zampini } else { 1523b097fa66SStefano Zampini ierr = VecScale(pcis->vec4_D,m_one);CHKERRQ(ierr); 1524b097fa66SStefano Zampini } 1525b097fa66SStefano Zampini ierr = VecScatterBegin(pcis->global_to_D,pcis->vec4_D,z,INSERT_VALUES,SCATTER_REVERSE);CHKERRQ(ierr); 1526b097fa66SStefano Zampini ierr = VecScatterEnd(pcis->global_to_D,pcis->vec4_D,z,INSERT_VALUES,SCATTER_REVERSE);CHKERRQ(ierr); 1527b097fa66SStefano Zampini } 152850efa1b5SStefano Zampini PetscFunctionReturn(0); 152950efa1b5SStefano Zampini } 1530da1bb401SStefano Zampini /* -------------------------------------------------------------------------- */ 1531674ae819SStefano Zampini 1532da1bb401SStefano Zampini #undef __FUNCT__ 1533da1bb401SStefano Zampini #define __FUNCT__ "PCDestroy_BDDC" 1534da1bb401SStefano Zampini PetscErrorCode PCDestroy_BDDC(PC pc) 1535da1bb401SStefano Zampini { 1536da1bb401SStefano Zampini PC_BDDC *pcbddc = (PC_BDDC*)pc->data; 1537da1bb401SStefano Zampini PetscErrorCode ierr; 1538da1bb401SStefano Zampini 1539da1bb401SStefano Zampini PetscFunctionBegin; 1540da1bb401SStefano Zampini /* free data created by PCIS */ 1541da1bb401SStefano Zampini ierr = PCISDestroy(pc);CHKERRQ(ierr); 1542674ae819SStefano Zampini /* free BDDC custom data */ 1543674ae819SStefano Zampini ierr = PCBDDCResetCustomization(pc);CHKERRQ(ierr); 1544674ae819SStefano Zampini /* destroy objects related to topography */ 1545674ae819SStefano Zampini ierr = PCBDDCResetTopography(pc);CHKERRQ(ierr); 1546674ae819SStefano Zampini /* free allocated graph structure */ 1547da1bb401SStefano Zampini ierr = PetscFree(pcbddc->mat_graph);CHKERRQ(ierr); 1548b96c3477SStefano Zampini /* free allocated sub schurs structure */ 1549b96c3477SStefano Zampini ierr = PetscFree(pcbddc->sub_schurs);CHKERRQ(ierr); 155034a97f8cSStefano Zampini /* destroy objects for scaling operator */ 1551674ae819SStefano Zampini ierr = PCBDDCScalingDestroy(pc);CHKERRQ(ierr); 155234a97f8cSStefano Zampini ierr = PetscFree(pcbddc->deluxe_ctx);CHKERRQ(ierr); 1553674ae819SStefano Zampini /* free solvers stuff */ 1554674ae819SStefano Zampini ierr = PCBDDCResetSolvers(pc);CHKERRQ(ierr); 155562a6ff1dSStefano Zampini /* free global vectors needed in presolve */ 155662a6ff1dSStefano Zampini ierr = VecDestroy(&pcbddc->temp_solution);CHKERRQ(ierr); 155762a6ff1dSStefano Zampini ierr = VecDestroy(&pcbddc->original_rhs);CHKERRQ(ierr); 1558906d46d4SStefano Zampini /* free stuff for change of basis hooks */ 1559906d46d4SStefano Zampini if (pcbddc->new_global_mat) { 1560906d46d4SStefano Zampini PCBDDCChange_ctx change_ctx; 1561906d46d4SStefano Zampini ierr = MatShellGetContext(pcbddc->new_global_mat,&change_ctx);CHKERRQ(ierr); 1562906d46d4SStefano Zampini ierr = MatDestroy(&change_ctx->original_mat);CHKERRQ(ierr); 1563906d46d4SStefano Zampini ierr = MatDestroy(&change_ctx->global_change);CHKERRQ(ierr); 1564906d46d4SStefano Zampini ierr = VecDestroyVecs(2,&change_ctx->work);CHKERRQ(ierr); 1565906d46d4SStefano Zampini ierr = PetscFree(change_ctx);CHKERRQ(ierr); 1566906d46d4SStefano Zampini } 1567906d46d4SStefano Zampini ierr = MatDestroy(&pcbddc->new_global_mat);CHKERRQ(ierr); 15683425bc38SStefano Zampini /* remove functions */ 1569906d46d4SStefano Zampini ierr = PetscObjectComposeFunction((PetscObject)pc,"PCBDDCSetChangeOfBasisMat_C",NULL);CHKERRQ(ierr); 1570674ae819SStefano Zampini ierr = PetscObjectComposeFunction((PetscObject)pc,"PCBDDCSetPrimalVerticesLocalIS_C",NULL);CHKERRQ(ierr); 1571bdf89e91SBarry Smith ierr = PetscObjectComposeFunction((PetscObject)pc,"PCBDDCSetCoarseningRatio_C",NULL);CHKERRQ(ierr); 15722b510759SStefano Zampini ierr = PetscObjectComposeFunction((PetscObject)pc,"PCBDDCSetLevel_C",NULL);CHKERRQ(ierr); 1573b8ffe317SStefano Zampini ierr = PetscObjectComposeFunction((PetscObject)pc,"PCBDDCSetUseExactDirichlet_C",NULL);CHKERRQ(ierr); 15742b510759SStefano Zampini ierr = PetscObjectComposeFunction((PetscObject)pc,"PCBDDCSetLevels_C",NULL);CHKERRQ(ierr); 1575bdf89e91SBarry Smith ierr = PetscObjectComposeFunction((PetscObject)pc,"PCBDDCSetNullSpace_C",NULL);CHKERRQ(ierr); 1576bdf89e91SBarry Smith ierr = PetscObjectComposeFunction((PetscObject)pc,"PCBDDCSetDirichletBoundaries_C",NULL);CHKERRQ(ierr); 157782ba6b80SStefano Zampini ierr = PetscObjectComposeFunction((PetscObject)pc,"PCBDDCSetDirichletBoundariesLocal_C",NULL);CHKERRQ(ierr); 1578bdf89e91SBarry Smith ierr = PetscObjectComposeFunction((PetscObject)pc,"PCBDDCSetNeumannBoundaries_C",NULL);CHKERRQ(ierr); 157982ba6b80SStefano Zampini ierr = PetscObjectComposeFunction((PetscObject)pc,"PCBDDCSetNeumannBoundariesLocal_C",NULL);CHKERRQ(ierr); 1580bdf89e91SBarry Smith ierr = PetscObjectComposeFunction((PetscObject)pc,"PCBDDCGetDirichletBoundaries_C",NULL);CHKERRQ(ierr); 158182ba6b80SStefano Zampini ierr = PetscObjectComposeFunction((PetscObject)pc,"PCBDDCGetDirichletBoundariesLocal_C",NULL);CHKERRQ(ierr); 1582bdf89e91SBarry Smith ierr = PetscObjectComposeFunction((PetscObject)pc,"PCBDDCGetNeumannBoundaries_C",NULL);CHKERRQ(ierr); 1583785d1243SStefano Zampini ierr = PetscObjectComposeFunction((PetscObject)pc,"PCBDDCGetNeumannBoundariesLocal_C",NULL);CHKERRQ(ierr); 1584bdf89e91SBarry Smith ierr = PetscObjectComposeFunction((PetscObject)pc,"PCBDDCSetDofsSplitting_C",NULL);CHKERRQ(ierr); 158563602bcaSStefano Zampini ierr = PetscObjectComposeFunction((PetscObject)pc,"PCBDDCSetDofsSplittingLocal_C",NULL);CHKERRQ(ierr); 1586bdf89e91SBarry Smith ierr = PetscObjectComposeFunction((PetscObject)pc,"PCBDDCSetLocalAdjacencyGraph_C",NULL);CHKERRQ(ierr); 1587bdf89e91SBarry Smith ierr = PetscObjectComposeFunction((PetscObject)pc,"PCBDDCCreateFETIDPOperators_C",NULL);CHKERRQ(ierr); 1588bdf89e91SBarry Smith ierr = PetscObjectComposeFunction((PetscObject)pc,"PCBDDCMatFETIDPGetRHS_C",NULL);CHKERRQ(ierr); 1589bdf89e91SBarry Smith ierr = PetscObjectComposeFunction((PetscObject)pc,"PCBDDCMatFETIDPGetSolution_C",NULL);CHKERRQ(ierr); 1590674ae819SStefano Zampini /* Free the private data structure */ 1591674ae819SStefano Zampini ierr = PetscFree(pc->data);CHKERRQ(ierr); 1592da1bb401SStefano Zampini PetscFunctionReturn(0); 1593da1bb401SStefano Zampini } 15943425bc38SStefano Zampini /* -------------------------------------------------------------------------- */ 15951e6b0712SBarry Smith 15963425bc38SStefano Zampini #undef __FUNCT__ 15973425bc38SStefano Zampini #define __FUNCT__ "PCBDDCMatFETIDPGetRHS_BDDC" 15983425bc38SStefano Zampini static PetscErrorCode PCBDDCMatFETIDPGetRHS_BDDC(Mat fetidp_mat, Vec standard_rhs, Vec fetidp_flux_rhs) 15993425bc38SStefano Zampini { 1600674ae819SStefano Zampini FETIDPMat_ctx mat_ctx; 1601c08af4c6SStefano Zampini Vec copy_standard_rhs; 16023425bc38SStefano Zampini PC_IS* pcis; 16033425bc38SStefano Zampini PC_BDDC* pcbddc; 16043425bc38SStefano Zampini PetscErrorCode ierr; 16050c7d97c5SJed Brown 16063425bc38SStefano Zampini PetscFunctionBegin; 16073425bc38SStefano Zampini ierr = MatShellGetContext(fetidp_mat,&mat_ctx);CHKERRQ(ierr); 16083425bc38SStefano Zampini pcis = (PC_IS*)mat_ctx->pc->data; 16093425bc38SStefano Zampini pcbddc = (PC_BDDC*)mat_ctx->pc->data; 16103425bc38SStefano Zampini 1611c08af4c6SStefano Zampini /* 1612c08af4c6SStefano Zampini change of basis for physical rhs if needed 1613c08af4c6SStefano Zampini It also changes the rhs in case of dirichlet boundaries 1614c08af4c6SStefano Zampini TODO: better management when FETIDP will have its own class 1615c08af4c6SStefano Zampini */ 1616c08af4c6SStefano Zampini ierr = VecDuplicate(standard_rhs,©_standard_rhs);CHKERRQ(ierr); 1617c08af4c6SStefano Zampini ierr = VecCopy(standard_rhs,copy_standard_rhs);CHKERRQ(ierr); 1618c08af4c6SStefano Zampini ierr = PCPreSolve_BDDC(mat_ctx->pc,NULL,copy_standard_rhs,NULL);CHKERRQ(ierr); 16193425bc38SStefano Zampini /* store vectors for computation of fetidp final solution */ 1620c08af4c6SStefano Zampini ierr = VecScatterBegin(pcis->global_to_D,copy_standard_rhs,mat_ctx->temp_solution_D,INSERT_VALUES,SCATTER_FORWARD);CHKERRQ(ierr); 1621c08af4c6SStefano Zampini ierr = VecScatterEnd(pcis->global_to_D,copy_standard_rhs,mat_ctx->temp_solution_D,INSERT_VALUES,SCATTER_FORWARD);CHKERRQ(ierr); 1622fb223d50SStefano Zampini /* scale rhs since it should be unassembled */ 1623fb223d50SStefano Zampini /* TODO use counter scaling? (also below) */ 1624c08af4c6SStefano Zampini ierr = VecScatterBegin(pcis->global_to_B,copy_standard_rhs,mat_ctx->temp_solution_B,INSERT_VALUES,SCATTER_FORWARD);CHKERRQ(ierr); 1625c08af4c6SStefano Zampini ierr = VecScatterEnd(pcis->global_to_B,copy_standard_rhs,mat_ctx->temp_solution_B,INSERT_VALUES,SCATTER_FORWARD);CHKERRQ(ierr); 1626674ae819SStefano Zampini /* Apply partition of unity */ 16273425bc38SStefano Zampini ierr = VecPointwiseMult(mat_ctx->temp_solution_B,pcis->D,mat_ctx->temp_solution_B);CHKERRQ(ierr); 1628c08af4c6SStefano Zampini /* ierr = PCBDDCScalingRestriction(mat_ctx->pc,copy_standard_rhs,mat_ctx->temp_solution_B);CHKERRQ(ierr); */ 16298eeda7d8SStefano Zampini if (!pcbddc->switch_static) { 16303425bc38SStefano Zampini /* compute partially subassembled Schur complement right-hand side */ 16313425bc38SStefano Zampini ierr = KSPSolve(pcbddc->ksp_D,mat_ctx->temp_solution_D,pcis->vec1_D);CHKERRQ(ierr); 16323425bc38SStefano Zampini ierr = MatMult(pcis->A_BI,pcis->vec1_D,pcis->vec1_B);CHKERRQ(ierr); 16333425bc38SStefano Zampini ierr = VecAXPY(mat_ctx->temp_solution_B,-1.0,pcis->vec1_B);CHKERRQ(ierr); 1634c08af4c6SStefano Zampini ierr = VecSet(copy_standard_rhs,0.0);CHKERRQ(ierr); 1635c08af4c6SStefano Zampini ierr = VecScatterBegin(pcis->global_to_B,mat_ctx->temp_solution_B,copy_standard_rhs,ADD_VALUES,SCATTER_REVERSE);CHKERRQ(ierr); 1636c08af4c6SStefano Zampini ierr = VecScatterEnd(pcis->global_to_B,mat_ctx->temp_solution_B,copy_standard_rhs,ADD_VALUES,SCATTER_REVERSE);CHKERRQ(ierr); 1637c08af4c6SStefano Zampini /* ierr = PCBDDCScalingRestriction(mat_ctx->pc,copy_standard_rhs,mat_ctx->temp_solution_B);CHKERRQ(ierr); */ 1638c08af4c6SStefano Zampini ierr = VecScatterBegin(pcis->global_to_B,copy_standard_rhs,mat_ctx->temp_solution_B,INSERT_VALUES,SCATTER_FORWARD);CHKERRQ(ierr); 1639c08af4c6SStefano Zampini ierr = VecScatterEnd(pcis->global_to_B,copy_standard_rhs,mat_ctx->temp_solution_B,INSERT_VALUES,SCATTER_FORWARD);CHKERRQ(ierr); 16403425bc38SStefano Zampini ierr = VecPointwiseMult(mat_ctx->temp_solution_B,pcis->D,mat_ctx->temp_solution_B);CHKERRQ(ierr); 16413425bc38SStefano Zampini } 1642c08af4c6SStefano Zampini ierr = VecDestroy(©_standard_rhs);CHKERRQ(ierr); 16433425bc38SStefano Zampini /* BDDC rhs */ 16443425bc38SStefano Zampini ierr = VecCopy(mat_ctx->temp_solution_B,pcis->vec1_B);CHKERRQ(ierr); 16458eeda7d8SStefano Zampini if (pcbddc->switch_static) { 16463425bc38SStefano Zampini ierr = VecCopy(mat_ctx->temp_solution_D,pcis->vec1_D);CHKERRQ(ierr); 16473425bc38SStefano Zampini } 16483425bc38SStefano Zampini /* apply BDDC */ 1649dc359a40SStefano Zampini ierr = PCBDDCApplyInterfacePreconditioner(mat_ctx->pc,PETSC_FALSE);CHKERRQ(ierr); 16503425bc38SStefano Zampini /* Application of B_delta and assembling of rhs for fetidp fluxes */ 16513425bc38SStefano Zampini ierr = VecSet(fetidp_flux_rhs,0.0);CHKERRQ(ierr); 16523425bc38SStefano Zampini ierr = MatMult(mat_ctx->B_delta,pcis->vec1_B,mat_ctx->lambda_local);CHKERRQ(ierr); 16533425bc38SStefano Zampini ierr = VecScatterBegin(mat_ctx->l2g_lambda,mat_ctx->lambda_local,fetidp_flux_rhs,ADD_VALUES,SCATTER_FORWARD);CHKERRQ(ierr); 16543425bc38SStefano Zampini ierr = VecScatterEnd(mat_ctx->l2g_lambda,mat_ctx->lambda_local,fetidp_flux_rhs,ADD_VALUES,SCATTER_FORWARD);CHKERRQ(ierr); 16553425bc38SStefano Zampini PetscFunctionReturn(0); 16563425bc38SStefano Zampini } 16571e6b0712SBarry Smith 16583425bc38SStefano Zampini #undef __FUNCT__ 16593425bc38SStefano Zampini #define __FUNCT__ "PCBDDCMatFETIDPGetRHS" 16603425bc38SStefano Zampini /*@ 16610f202f7eSStefano Zampini PCBDDCMatFETIDPGetRHS - Compute the right-hand side for FETI-DP linear system using the physical right-hand side 16623425bc38SStefano Zampini 16633425bc38SStefano Zampini Collective 16643425bc38SStefano Zampini 16653425bc38SStefano Zampini Input Parameters: 16660f202f7eSStefano Zampini + fetidp_mat - the FETI-DP matrix object obtained by a call to PCBDDCCreateFETIDPOperators 16670f202f7eSStefano Zampini - standard_rhs - the right-hand side of the original linear system 16683425bc38SStefano Zampini 16693425bc38SStefano Zampini Output Parameters: 16700f202f7eSStefano Zampini . fetidp_flux_rhs - the right-hand side for the FETI-DP linear system 16713425bc38SStefano Zampini 16723425bc38SStefano Zampini Level: developer 16733425bc38SStefano Zampini 16743425bc38SStefano Zampini Notes: 16753425bc38SStefano Zampini 16760f202f7eSStefano Zampini .seealso: PCBDDC, PCBDDCCreateFETIDPOperators, PCBDDCMatFETIDPGetSolution 16773425bc38SStefano Zampini @*/ 16783425bc38SStefano Zampini PetscErrorCode PCBDDCMatFETIDPGetRHS(Mat fetidp_mat, Vec standard_rhs, Vec fetidp_flux_rhs) 16793425bc38SStefano Zampini { 1680674ae819SStefano Zampini FETIDPMat_ctx mat_ctx; 16813425bc38SStefano Zampini PetscErrorCode ierr; 16823425bc38SStefano Zampini 16833425bc38SStefano Zampini PetscFunctionBegin; 16843425bc38SStefano Zampini ierr = MatShellGetContext(fetidp_mat,&mat_ctx);CHKERRQ(ierr); 16853425bc38SStefano Zampini ierr = PetscTryMethod(mat_ctx->pc,"PCBDDCMatFETIDPGetRHS_C",(Mat,Vec,Vec),(fetidp_mat,standard_rhs,fetidp_flux_rhs));CHKERRQ(ierr); 16863425bc38SStefano Zampini PetscFunctionReturn(0); 16873425bc38SStefano Zampini } 16883425bc38SStefano Zampini /* -------------------------------------------------------------------------- */ 16891e6b0712SBarry Smith 16903425bc38SStefano Zampini #undef __FUNCT__ 16913425bc38SStefano Zampini #define __FUNCT__ "PCBDDCMatFETIDPGetSolution_BDDC" 16923425bc38SStefano Zampini static PetscErrorCode PCBDDCMatFETIDPGetSolution_BDDC(Mat fetidp_mat, Vec fetidp_flux_sol, Vec standard_sol) 16933425bc38SStefano Zampini { 1694674ae819SStefano Zampini FETIDPMat_ctx mat_ctx; 16953425bc38SStefano Zampini PC_IS* pcis; 16963425bc38SStefano Zampini PC_BDDC* pcbddc; 16973425bc38SStefano Zampini PetscErrorCode ierr; 16983425bc38SStefano Zampini 16993425bc38SStefano Zampini PetscFunctionBegin; 17003425bc38SStefano Zampini ierr = MatShellGetContext(fetidp_mat,&mat_ctx);CHKERRQ(ierr); 17013425bc38SStefano Zampini pcis = (PC_IS*)mat_ctx->pc->data; 17023425bc38SStefano Zampini pcbddc = (PC_BDDC*)mat_ctx->pc->data; 17033425bc38SStefano Zampini 17043425bc38SStefano Zampini /* apply B_delta^T */ 17053425bc38SStefano Zampini ierr = VecScatterBegin(mat_ctx->l2g_lambda,fetidp_flux_sol,mat_ctx->lambda_local,INSERT_VALUES,SCATTER_REVERSE);CHKERRQ(ierr); 17063425bc38SStefano Zampini ierr = VecScatterEnd (mat_ctx->l2g_lambda,fetidp_flux_sol,mat_ctx->lambda_local,INSERT_VALUES,SCATTER_REVERSE);CHKERRQ(ierr); 17073425bc38SStefano Zampini ierr = MatMultTranspose(mat_ctx->B_delta,mat_ctx->lambda_local,pcis->vec1_B);CHKERRQ(ierr); 17083425bc38SStefano Zampini /* compute rhs for BDDC application */ 17093425bc38SStefano Zampini ierr = VecAYPX(pcis->vec1_B,-1.0,mat_ctx->temp_solution_B);CHKERRQ(ierr); 17108eeda7d8SStefano Zampini if (pcbddc->switch_static) { 17113425bc38SStefano Zampini ierr = VecCopy(mat_ctx->temp_solution_D,pcis->vec1_D);CHKERRQ(ierr); 17123425bc38SStefano Zampini } 17133425bc38SStefano Zampini /* apply BDDC */ 1714dc359a40SStefano Zampini ierr = PCBDDCApplyInterfacePreconditioner(mat_ctx->pc,PETSC_FALSE);CHKERRQ(ierr); 17153425bc38SStefano Zampini /* put values into standard global vector */ 17163425bc38SStefano Zampini ierr = VecScatterBegin(pcis->global_to_B,pcis->vec1_B,standard_sol,INSERT_VALUES,SCATTER_REVERSE);CHKERRQ(ierr); 17173425bc38SStefano Zampini ierr = VecScatterEnd (pcis->global_to_B,pcis->vec1_B,standard_sol,INSERT_VALUES,SCATTER_REVERSE);CHKERRQ(ierr); 17188eeda7d8SStefano Zampini if (!pcbddc->switch_static) { 17193425bc38SStefano Zampini /* compute values into the interior if solved for the partially subassembled Schur complement */ 17203425bc38SStefano Zampini ierr = MatMult(pcis->A_IB,pcis->vec1_B,pcis->vec1_D);CHKERRQ(ierr); 17213425bc38SStefano Zampini ierr = VecAXPY(mat_ctx->temp_solution_D,-1.0,pcis->vec1_D);CHKERRQ(ierr); 17223425bc38SStefano Zampini ierr = KSPSolve(pcbddc->ksp_D,mat_ctx->temp_solution_D,pcis->vec1_D);CHKERRQ(ierr); 17233425bc38SStefano Zampini } 17243425bc38SStefano Zampini ierr = VecScatterBegin(pcis->global_to_D,pcis->vec1_D,standard_sol,INSERT_VALUES,SCATTER_REVERSE);CHKERRQ(ierr); 17253425bc38SStefano Zampini ierr = VecScatterEnd (pcis->global_to_D,pcis->vec1_D,standard_sol,INSERT_VALUES,SCATTER_REVERSE);CHKERRQ(ierr); 17263425bc38SStefano Zampini /* final change of basis if needed 17273425bc38SStefano Zampini Is also sums the dirichlet part removed during RHS assembling */ 17283308cffdSStefano Zampini ierr = PCPostSolve_BDDC(mat_ctx->pc,NULL,NULL,standard_sol);CHKERRQ(ierr); 17293425bc38SStefano Zampini PetscFunctionReturn(0); 17303425bc38SStefano Zampini } 17311e6b0712SBarry Smith 17323425bc38SStefano Zampini #undef __FUNCT__ 17333425bc38SStefano Zampini #define __FUNCT__ "PCBDDCMatFETIDPGetSolution" 17343425bc38SStefano Zampini /*@ 17350f202f7eSStefano Zampini PCBDDCMatFETIDPGetSolution - Compute the physical solution using the solution of the FETI-DP linear system 17363425bc38SStefano Zampini 17373425bc38SStefano Zampini Collective 17383425bc38SStefano Zampini 17393425bc38SStefano Zampini Input Parameters: 17400f202f7eSStefano Zampini + fetidp_mat - the FETI-DP matrix obtained by a call to PCBDDCCreateFETIDPOperators 17410f202f7eSStefano Zampini - fetidp_flux_sol - the solution of the FETI-DP linear system 17423425bc38SStefano Zampini 17433425bc38SStefano Zampini Output Parameters: 17440f202f7eSStefano Zampini . standard_sol - the solution defined on the physical domain 17453425bc38SStefano Zampini 17463425bc38SStefano Zampini Level: developer 17473425bc38SStefano Zampini 17483425bc38SStefano Zampini Notes: 17493425bc38SStefano Zampini 17500f202f7eSStefano Zampini .seealso: PCBDDC, PCBDDCCreateFETIDPOperators, PCBDDCMatFETIDPGetRHS 17513425bc38SStefano Zampini @*/ 17523425bc38SStefano Zampini PetscErrorCode PCBDDCMatFETIDPGetSolution(Mat fetidp_mat, Vec fetidp_flux_sol, Vec standard_sol) 17533425bc38SStefano Zampini { 1754674ae819SStefano Zampini FETIDPMat_ctx mat_ctx; 17553425bc38SStefano Zampini PetscErrorCode ierr; 17563425bc38SStefano Zampini 17573425bc38SStefano Zampini PetscFunctionBegin; 17583425bc38SStefano Zampini ierr = MatShellGetContext(fetidp_mat,&mat_ctx);CHKERRQ(ierr); 17593425bc38SStefano Zampini ierr = PetscTryMethod(mat_ctx->pc,"PCBDDCMatFETIDPGetSolution_C",(Mat,Vec,Vec),(fetidp_mat,fetidp_flux_sol,standard_sol));CHKERRQ(ierr); 17603425bc38SStefano Zampini PetscFunctionReturn(0); 17613425bc38SStefano Zampini } 17623425bc38SStefano Zampini /* -------------------------------------------------------------------------- */ 17631e6b0712SBarry Smith 1764f23aa3ddSBarry Smith extern PetscErrorCode FETIDPMatMult(Mat,Vec,Vec); 1765edf7251bSStefano Zampini extern PetscErrorCode FETIDPMatMultTranspose(Mat,Vec,Vec); 1766f23aa3ddSBarry Smith extern PetscErrorCode PCBDDCDestroyFETIDPMat(Mat); 1767f23aa3ddSBarry Smith extern PetscErrorCode FETIDPPCApply(PC,Vec,Vec); 1768edf7251bSStefano Zampini extern PetscErrorCode FETIDPPCApplyTranspose(PC,Vec,Vec); 1769f23aa3ddSBarry Smith extern PetscErrorCode PCBDDCDestroyFETIDPPC(PC); 1770674ae819SStefano Zampini 17713425bc38SStefano Zampini #undef __FUNCT__ 17723425bc38SStefano Zampini #define __FUNCT__ "PCBDDCCreateFETIDPOperators_BDDC" 17733425bc38SStefano Zampini static PetscErrorCode PCBDDCCreateFETIDPOperators_BDDC(PC pc, Mat *fetidp_mat, PC *fetidp_pc) 17743425bc38SStefano Zampini { 1775674ae819SStefano Zampini 1776674ae819SStefano Zampini FETIDPMat_ctx fetidpmat_ctx; 17773425bc38SStefano Zampini Mat newmat; 1778674ae819SStefano Zampini FETIDPPC_ctx fetidppc_ctx; 17793425bc38SStefano Zampini PC newpc; 1780ce94432eSBarry Smith MPI_Comm comm; 17813425bc38SStefano Zampini PetscErrorCode ierr; 17823425bc38SStefano Zampini 17833425bc38SStefano Zampini PetscFunctionBegin; 1784ce94432eSBarry Smith ierr = PetscObjectGetComm((PetscObject)pc,&comm);CHKERRQ(ierr); 17853425bc38SStefano Zampini /* FETIDP linear matrix */ 17863425bc38SStefano Zampini ierr = PCBDDCCreateFETIDPMatContext(pc,&fetidpmat_ctx);CHKERRQ(ierr); 17873425bc38SStefano Zampini ierr = PCBDDCSetupFETIDPMatContext(fetidpmat_ctx);CHKERRQ(ierr); 17883425bc38SStefano Zampini ierr = MatCreateShell(comm,PETSC_DECIDE,PETSC_DECIDE,fetidpmat_ctx->n_lambda,fetidpmat_ctx->n_lambda,fetidpmat_ctx,&newmat);CHKERRQ(ierr); 17893425bc38SStefano Zampini ierr = MatShellSetOperation(newmat,MATOP_MULT,(void (*)(void))FETIDPMatMult);CHKERRQ(ierr); 1790edf7251bSStefano Zampini ierr = MatShellSetOperation(newmat,MATOP_MULT_TRANSPOSE,(void (*)(void))FETIDPMatMultTranspose);CHKERRQ(ierr); 17913425bc38SStefano Zampini ierr = MatShellSetOperation(newmat,MATOP_DESTROY,(void (*)(void))PCBDDCDestroyFETIDPMat);CHKERRQ(ierr); 17923425bc38SStefano Zampini ierr = MatSetUp(newmat);CHKERRQ(ierr); 17933425bc38SStefano Zampini /* FETIDP preconditioner */ 17943425bc38SStefano Zampini ierr = PCBDDCCreateFETIDPPCContext(pc,&fetidppc_ctx);CHKERRQ(ierr); 17953425bc38SStefano Zampini ierr = PCBDDCSetupFETIDPPCContext(newmat,fetidppc_ctx);CHKERRQ(ierr); 17963425bc38SStefano Zampini ierr = PCCreate(comm,&newpc);CHKERRQ(ierr); 17973425bc38SStefano Zampini ierr = PCSetType(newpc,PCSHELL);CHKERRQ(ierr); 17983425bc38SStefano Zampini ierr = PCShellSetContext(newpc,fetidppc_ctx);CHKERRQ(ierr); 17993425bc38SStefano Zampini ierr = PCShellSetApply(newpc,FETIDPPCApply);CHKERRQ(ierr); 1800edf7251bSStefano Zampini ierr = PCShellSetApplyTranspose(newpc,FETIDPPCApplyTranspose);CHKERRQ(ierr); 18013425bc38SStefano Zampini ierr = PCShellSetDestroy(newpc,PCBDDCDestroyFETIDPPC);CHKERRQ(ierr); 180223ee1639SBarry Smith ierr = PCSetOperators(newpc,newmat,newmat);CHKERRQ(ierr); 18033425bc38SStefano Zampini ierr = PCSetUp(newpc);CHKERRQ(ierr); 18043425bc38SStefano Zampini /* return pointers for objects created */ 18053425bc38SStefano Zampini *fetidp_mat=newmat; 18063425bc38SStefano Zampini *fetidp_pc=newpc; 18073425bc38SStefano Zampini PetscFunctionReturn(0); 18083425bc38SStefano Zampini } 18091e6b0712SBarry Smith 18103425bc38SStefano Zampini #undef __FUNCT__ 18113425bc38SStefano Zampini #define __FUNCT__ "PCBDDCCreateFETIDPOperators" 18123425bc38SStefano Zampini /*@ 18130f202f7eSStefano Zampini PCBDDCCreateFETIDPOperators - Create FETI-DP operators 18143425bc38SStefano Zampini 18153425bc38SStefano Zampini Collective 18163425bc38SStefano Zampini 18173425bc38SStefano Zampini Input Parameters: 18180f202f7eSStefano Zampini . pc - the BDDC preconditioning context (setup should have been called before) 181928509bceSStefano Zampini 182028509bceSStefano Zampini Output Parameters: 18210f202f7eSStefano Zampini + fetidp_mat - shell FETI-DP matrix object 18220f202f7eSStefano Zampini - fetidp_pc - shell Dirichlet preconditioner for FETI-DP matrix 182328509bceSStefano Zampini 182428509bceSStefano Zampini Options Database Keys: 18250f202f7eSStefano Zampini . -fetidp_fullyredundant <false> - use or not a fully redundant set of Lagrange multipliers 18263425bc38SStefano Zampini 18273425bc38SStefano Zampini Level: developer 18283425bc38SStefano Zampini 18293425bc38SStefano Zampini Notes: 18300f202f7eSStefano Zampini Currently the only operations provided for FETI-DP matrix are MatMult and MatMultTranspose 18313425bc38SStefano Zampini 18320f202f7eSStefano Zampini .seealso: PCBDDC, PCBDDCMatFETIDPGetRHS, PCBDDCMatFETIDPGetSolution 18333425bc38SStefano Zampini @*/ 18343425bc38SStefano Zampini PetscErrorCode PCBDDCCreateFETIDPOperators(PC pc, Mat *fetidp_mat, PC *fetidp_pc) 18353425bc38SStefano Zampini { 18363425bc38SStefano Zampini PetscErrorCode ierr; 18373425bc38SStefano Zampini 18383425bc38SStefano Zampini PetscFunctionBegin; 18393425bc38SStefano Zampini PetscValidHeaderSpecific(pc,PC_CLASSID,1); 18403425bc38SStefano Zampini if (pc->setupcalled) { 1841516d51a7SStefano Zampini ierr = PetscUseMethod(pc,"PCBDDCCreateFETIDPOperators_C",(PC,Mat*,PC*),(pc,fetidp_mat,fetidp_pc));CHKERRQ(ierr); 1842f23aa3ddSBarry Smith } else SETERRQ(PETSC_COMM_SELF,PETSC_ERR_SUP,"You must call PCSetup_BDDC() first \n"); 18433425bc38SStefano Zampini PetscFunctionReturn(0); 18443425bc38SStefano Zampini } 18450c7d97c5SJed Brown /* -------------------------------------------------------------------------- */ 1846da1bb401SStefano Zampini /*MC 1847da1bb401SStefano Zampini PCBDDC - Balancing Domain Decomposition by Constraints. 18480c7d97c5SJed Brown 184928509bceSStefano Zampini An implementation of the BDDC preconditioner based on 185028509bceSStefano Zampini 185128509bceSStefano Zampini .vb 185228509bceSStefano Zampini [1] C. R. Dohrmann. "An approximate BDDC preconditioner", Numerical Linear Algebra with Applications Volume 14, Issue 2, pages 149-168, March 2007 185328509bceSStefano 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 185428509bceSStefano Zampini [3] J. Mandel, B. Sousedik, C. R. Dohrmann. "Multispace and Multilevel BDDC", http://arxiv.org/abs/0712.3977 18550f202f7eSStefano 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 185628509bceSStefano Zampini .ve 185728509bceSStefano Zampini 185828509bceSStefano Zampini The matrix to be preconditioned (Pmat) must be of type MATIS. 185928509bceSStefano Zampini 18600f202f7eSStefano Zampini Currently works with MATIS matrices with local matrices of type MATSEQAIJ, MATSEQBAIJ or MATSEQSBAIJ, either with real or complex numbers. 186128509bceSStefano Zampini 186228509bceSStefano Zampini It also works with unsymmetric and indefinite problems. 186328509bceSStefano Zampini 1864b6fdb6dfSStefano 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. 1865b6fdb6dfSStefano Zampini 18660f202f7eSStefano Zampini Approximate local solvers are automatically adapted for singular linear problems (see [1]) if the user has provided the nullspace using PCBDDCSetNullSpace() 186728509bceSStefano Zampini 18680f202f7eSStefano 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() 18690f202f7eSStefano Zampini Additional information on dofs can be provided by using PCBDDCSetDofsSplitting(), PCBDDCSetDirichletBoundaries(), PCBDDCSetNeumannBoundaries(), and PCBDDCSetPrimalVerticesLocalIS() 187028509bceSStefano Zampini 18710f202f7eSStefano Zampini Constraints can be customized by attaching a MatNullSpace object to the MATIS matrix via MatSetNearNullSpace(). Non-singular modes are retained via SVD. 187228509bceSStefano Zampini 18730f202f7eSStefano 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. 18740f202f7eSStefano Zampini User defined change of basis can be passed to PCBDDC by using PCBDDCSetChangeOfBasisMat() 187528509bceSStefano Zampini 18760f202f7eSStefano Zampini The PETSc implementation also supports multilevel BDDC [3]. Coarse grids are partitioned using a MatPartitioning object. 187728509bceSStefano Zampini 18780f202f7eSStefano Zampini Adaptive selection of primal constraints [4] is supported for SPD systems with high-contrast in the coefficients if MUMPS is present. Future versions of the code will also consider using MKL_PARDISO or PASTIX. 187928509bceSStefano Zampini 18800f202f7eSStefano 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. 18810f202f7eSStefano Zampini Deluxe scaling is not supported yet for FETI-DP. 18820f202f7eSStefano Zampini 18830f202f7eSStefano Zampini Options Database Keys (some of them, run with -h for a complete list): 18840f202f7eSStefano Zampini 18850f202f7eSStefano Zampini . -pc_bddc_use_vertices <true> - use or not vertices in primal space 18860f202f7eSStefano Zampini . -pc_bddc_use_edges <true> - use or not edges in primal space 18870f202f7eSStefano Zampini . -pc_bddc_use_faces <false> - use or not faces in primal space 18880f202f7eSStefano Zampini . -pc_bddc_symmetric <true> - symmetric computation of primal basis functions. Specify false for unsymmetric problems 18890f202f7eSStefano Zampini . -pc_bddc_use_change_of_basis <false> - use change of basis approach (on edges only) 18900f202f7eSStefano Zampini . -pc_bddc_use_change_on_faces <false> - use change of basis approach on faces if change of basis has been requested 18910f202f7eSStefano Zampini . -pc_bddc_switch_static <false> - switches from M_2 (default) to M_3 operator (see reference article [1]) 189228509bceSStefano Zampini . -pc_bddc_levels <0> - maximum number of levels for multilevel 18930f202f7eSStefano 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) 18940f202f7eSStefano 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) 18950f202f7eSStefano Zampini . -pc_bddc_use_deluxe_scaling <false> - use deluxe scaling 18960f202f7eSStefano 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) 18970f202f7eSStefano Zampini . -pc_bddc_adaptive_threshold <0.0> - when a value greater than one is specified, adaptive selection of constraints is performed on edges and faces (requires deluxe scaling and MUMPS installed) 189828509bceSStefano Zampini - -pc_bddc_check_level <0> - set verbosity level of debugging output 189928509bceSStefano Zampini 190028509bceSStefano Zampini Options for Dirichlet, Neumann or coarse solver can be set with 190128509bceSStefano Zampini .vb 190228509bceSStefano Zampini -pc_bddc_dirichlet_ 190328509bceSStefano Zampini -pc_bddc_neumann_ 190428509bceSStefano Zampini -pc_bddc_coarse_ 190528509bceSStefano Zampini .ve 19060f202f7eSStefano Zampini e.g -pc_bddc_dirichlet_ksp_type richardson -pc_bddc_dirichlet_pc_type gamg. PCBDDC uses by default KPSPREONLY and PCLU. 190728509bceSStefano Zampini 19080f202f7eSStefano Zampini When using a multilevel approach, solvers' options at the N-th level (N > 1) can be specified as 190928509bceSStefano Zampini .vb 1910312be037SStefano Zampini -pc_bddc_dirichlet_lN_ 1911312be037SStefano Zampini -pc_bddc_neumann_lN_ 1912312be037SStefano Zampini -pc_bddc_coarse_lN_ 191328509bceSStefano Zampini .ve 19140f202f7eSStefano Zampini Note that level number ranges from the finest (0) to the coarsest (N). 19150f202f7eSStefano 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. 19160f202f7eSStefano Zampini .vb 19170f202f7eSStefano Zampini -pc_bddc_coarse_pc_bddc_adaptive_threshold 5 -pc_bddc_coarse_l1_pc_bddc_redistribute 3 19180f202f7eSStefano Zampini .ve 19190f202f7eSStefano 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 1920da1bb401SStefano Zampini 1921da1bb401SStefano Zampini Level: intermediate 1922da1bb401SStefano Zampini 1923b6fdb6dfSStefano Zampini Developer notes: 1924da1bb401SStefano Zampini 1925da1bb401SStefano Zampini Contributed by Stefano Zampini 1926da1bb401SStefano Zampini 1927da1bb401SStefano Zampini .seealso: PCCreate(), PCSetType(), PCType (for list of available types), PC, MATIS 1928da1bb401SStefano Zampini M*/ 1929b2573a8aSBarry Smith 1930da1bb401SStefano Zampini #undef __FUNCT__ 1931da1bb401SStefano Zampini #define __FUNCT__ "PCCreate_BDDC" 19328cc058d9SJed Brown PETSC_EXTERN PetscErrorCode PCCreate_BDDC(PC pc) 1933da1bb401SStefano Zampini { 1934da1bb401SStefano Zampini PetscErrorCode ierr; 1935da1bb401SStefano Zampini PC_BDDC *pcbddc; 1936da1bb401SStefano Zampini 1937da1bb401SStefano Zampini PetscFunctionBegin; 1938da1bb401SStefano Zampini /* Creates the private data structure for this preconditioner and attach it to the PC object. */ 1939b00a9115SJed Brown ierr = PetscNewLog(pc,&pcbddc);CHKERRQ(ierr); 1940da1bb401SStefano Zampini pc->data = (void*)pcbddc; 1941da1bb401SStefano Zampini 1942da1bb401SStefano Zampini /* create PCIS data structure */ 1943da1bb401SStefano Zampini ierr = PCISCreate(pc);CHKERRQ(ierr); 1944da1bb401SStefano Zampini 194547d04d0dSStefano Zampini /* BDDC customization */ 194608a5cf49SStefano Zampini pcbddc->use_local_adj = PETSC_TRUE; 194747d04d0dSStefano Zampini pcbddc->use_vertices = PETSC_TRUE; 194847d04d0dSStefano Zampini pcbddc->use_edges = PETSC_TRUE; 194947d04d0dSStefano Zampini pcbddc->use_faces = PETSC_FALSE; 195047d04d0dSStefano Zampini pcbddc->use_change_of_basis = PETSC_FALSE; 195147d04d0dSStefano Zampini pcbddc->use_change_on_faces = PETSC_FALSE; 195247d04d0dSStefano Zampini pcbddc->switch_static = PETSC_FALSE; 1953fa434743SStefano Zampini pcbddc->use_nnsp_true = PETSC_FALSE; 1954fa434743SStefano Zampini pcbddc->use_qr_single = PETSC_FALSE; 19553301b35fSStefano Zampini pcbddc->symmetric_primal = PETSC_TRUE; 195647d04d0dSStefano Zampini pcbddc->dbg_flag = 0; 1957b9d89cd5SStefano Zampini /* private */ 1958727cdba6SStefano Zampini pcbddc->local_primal_size = 0; 19590e6343abSStefano Zampini pcbddc->local_primal_size_cc = 0; 19600e6343abSStefano Zampini pcbddc->local_primal_ref_node = 0; 19610e6343abSStefano Zampini pcbddc->local_primal_ref_mult = 0; 1962e9189074SStefano Zampini pcbddc->n_vertices = 0; 1963727cdba6SStefano Zampini pcbddc->primal_indices_local_idxs = 0; 1964fb180af4SStefano Zampini pcbddc->recompute_topography = PETSC_FALSE; 196568457ee5SStefano Zampini pcbddc->coarse_size = -1; 1966f4ddd8eeSStefano Zampini pcbddc->new_primal_space = PETSC_FALSE; 1967727cdba6SStefano Zampini pcbddc->new_primal_space_local = PETSC_FALSE; 1968f4ddd8eeSStefano Zampini pcbddc->global_primal_indices = 0; 1969f4ddd8eeSStefano Zampini pcbddc->onearnullspace = 0; 1970f4ddd8eeSStefano Zampini pcbddc->onearnullvecs_state = 0; 1971674ae819SStefano Zampini pcbddc->user_primal_vertices = 0; 19720bdf917eSStefano Zampini pcbddc->NullSpace = 0; 19733972b0daSStefano Zampini pcbddc->temp_solution = 0; 1974534831adSStefano Zampini pcbddc->original_rhs = 0; 1975534831adSStefano Zampini pcbddc->local_mat = 0; 1976534831adSStefano Zampini pcbddc->ChangeOfBasisMatrix = 0; 1977b9b85e73SStefano Zampini pcbddc->user_ChangeOfBasisMatrix = 0; 1978906d46d4SStefano Zampini pcbddc->new_global_mat = 0; 1979da1bb401SStefano Zampini pcbddc->coarse_vec = 0; 1980da1bb401SStefano Zampini pcbddc->coarse_ksp = 0; 1981da1bb401SStefano Zampini pcbddc->coarse_phi_B = 0; 1982da1bb401SStefano Zampini pcbddc->coarse_phi_D = 0; 198315aaf578SStefano Zampini pcbddc->coarse_psi_B = 0; 198415aaf578SStefano Zampini pcbddc->coarse_psi_D = 0; 1985da1bb401SStefano Zampini pcbddc->vec1_P = 0; 1986da1bb401SStefano Zampini pcbddc->vec1_R = 0; 1987da1bb401SStefano Zampini pcbddc->vec2_R = 0; 1988da1bb401SStefano Zampini pcbddc->local_auxmat1 = 0; 1989da1bb401SStefano Zampini pcbddc->local_auxmat2 = 0; 1990da1bb401SStefano Zampini pcbddc->R_to_B = 0; 1991da1bb401SStefano Zampini pcbddc->R_to_D = 0; 1992da1bb401SStefano Zampini pcbddc->ksp_D = 0; 1993da1bb401SStefano Zampini pcbddc->ksp_R = 0; 1994da1bb401SStefano Zampini pcbddc->NeumannBoundaries = 0; 1995785d1243SStefano Zampini pcbddc->NeumannBoundariesLocal = 0; 1996785d1243SStefano Zampini pcbddc->DirichletBoundaries = 0; 1997785d1243SStefano Zampini pcbddc->DirichletBoundariesLocal = 0; 199860d44989SStefano Zampini pcbddc->user_provided_isfordofs = PETSC_FALSE; 199960d44989SStefano Zampini pcbddc->n_ISForDofs = 0; 200063602bcaSStefano Zampini pcbddc->n_ISForDofsLocal = 0; 2001da1bb401SStefano Zampini pcbddc->ISForDofs = 0; 200263602bcaSStefano Zampini pcbddc->ISForDofsLocal = 0; 2003da1bb401SStefano Zampini pcbddc->ConstraintMatrix = 0; 200485c4d303SStefano Zampini pcbddc->use_exact_dirichlet_trick = PETSC_TRUE; 200547d04d0dSStefano Zampini pcbddc->coarse_loc_to_glob = 0; 200647d04d0dSStefano Zampini pcbddc->coarsening_ratio = 8; 2007b0c7d250SStefano Zampini pcbddc->coarse_adj_red = 0; 20084fad6a16SStefano Zampini pcbddc->current_level = 0; 20092b510759SStefano Zampini pcbddc->max_levels = 0; 2010323d395dSStefano Zampini pcbddc->use_coarse_estimates = PETSC_FALSE; 201174e2c79eSStefano Zampini pcbddc->redistribute_coarse = 0; 2012f3bde8b3SStefano Zampini pcbddc->coarse_subassembling = 0; 2013323d395dSStefano Zampini pcbddc->coarse_subassembling_init = 0; 2014da1bb401SStefano Zampini 2015674ae819SStefano Zampini /* create local graph structure */ 2016674ae819SStefano Zampini ierr = PCBDDCGraphCreate(&pcbddc->mat_graph);CHKERRQ(ierr); 2017674ae819SStefano Zampini 2018674ae819SStefano Zampini /* scaling */ 2019674ae819SStefano Zampini pcbddc->work_scaling = 0; 202034a97f8cSStefano Zampini pcbddc->use_deluxe_scaling = PETSC_FALSE; 2021ac632422SStefano Zampini pcbddc->faster_deluxe = PETSC_FALSE; 2022b96c3477SStefano Zampini 2023b96c3477SStefano Zampini /* create sub schurs structure */ 2024b96c3477SStefano Zampini ierr = PCBDDCSubSchursCreate(&pcbddc->sub_schurs);CHKERRQ(ierr); 2025b96c3477SStefano Zampini pcbddc->sub_schurs_rebuild = PETSC_FALSE; 2026b96c3477SStefano Zampini pcbddc->sub_schurs_layers = -1; 2027b96c3477SStefano Zampini pcbddc->sub_schurs_use_useradj = PETSC_FALSE; 2028b96c3477SStefano Zampini 2029b96c3477SStefano Zampini pcbddc->computed_rowadj = PETSC_FALSE; 2030da1bb401SStefano Zampini 2031b7eb3628SStefano Zampini /* adaptivity */ 2032f6f667cfSStefano Zampini pcbddc->adaptive_threshold = 0.0; 203308122e43SStefano Zampini pcbddc->adaptive_nmax = 0; 2034f6f667cfSStefano Zampini pcbddc->adaptive_nmin = 0; 2035b7eb3628SStefano Zampini 2036da1bb401SStefano Zampini /* function pointers */ 2037da1bb401SStefano Zampini pc->ops->apply = PCApply_BDDC; 203893bd9ae7SStefano Zampini pc->ops->applytranspose = PCApplyTranspose_BDDC; 2039da1bb401SStefano Zampini pc->ops->setup = PCSetUp_BDDC; 2040da1bb401SStefano Zampini pc->ops->destroy = PCDestroy_BDDC; 2041da1bb401SStefano Zampini pc->ops->setfromoptions = PCSetFromOptions_BDDC; 2042da1bb401SStefano Zampini pc->ops->view = 0; 2043da1bb401SStefano Zampini pc->ops->applyrichardson = 0; 2044da1bb401SStefano Zampini pc->ops->applysymmetricleft = 0; 2045da1bb401SStefano Zampini pc->ops->applysymmetricright = 0; 2046534831adSStefano Zampini pc->ops->presolve = PCPreSolve_BDDC; 2047534831adSStefano Zampini pc->ops->postsolve = PCPostSolve_BDDC; 2048da1bb401SStefano Zampini 2049da1bb401SStefano Zampini /* composing function */ 2050906d46d4SStefano Zampini ierr = PetscObjectComposeFunction((PetscObject)pc,"PCBDDCSetChangeOfBasisMat_C",PCBDDCSetChangeOfBasisMat_BDDC);CHKERRQ(ierr); 2051674ae819SStefano Zampini ierr = PetscObjectComposeFunction((PetscObject)pc,"PCBDDCSetPrimalVerticesLocalIS_C",PCBDDCSetPrimalVerticesLocalIS_BDDC);CHKERRQ(ierr); 2052bdf89e91SBarry Smith ierr = PetscObjectComposeFunction((PetscObject)pc,"PCBDDCSetCoarseningRatio_C",PCBDDCSetCoarseningRatio_BDDC);CHKERRQ(ierr); 20532b510759SStefano Zampini ierr = PetscObjectComposeFunction((PetscObject)pc,"PCBDDCSetLevel_C",PCBDDCSetLevel_BDDC);CHKERRQ(ierr); 2054b8ffe317SStefano Zampini ierr = PetscObjectComposeFunction((PetscObject)pc,"PCBDDCSetUseExactDirichlet_C",PCBDDCSetUseExactDirichlet_BDDC);CHKERRQ(ierr); 20552b510759SStefano Zampini ierr = PetscObjectComposeFunction((PetscObject)pc,"PCBDDCSetLevels_C",PCBDDCSetLevels_BDDC);CHKERRQ(ierr); 2056bdf89e91SBarry Smith ierr = PetscObjectComposeFunction((PetscObject)pc,"PCBDDCSetNullSpace_C",PCBDDCSetNullSpace_BDDC);CHKERRQ(ierr); 2057bdf89e91SBarry Smith ierr = PetscObjectComposeFunction((PetscObject)pc,"PCBDDCSetDirichletBoundaries_C",PCBDDCSetDirichletBoundaries_BDDC);CHKERRQ(ierr); 205882ba6b80SStefano Zampini ierr = PetscObjectComposeFunction((PetscObject)pc,"PCBDDCSetDirichletBoundariesLocal_C",PCBDDCSetDirichletBoundariesLocal_BDDC);CHKERRQ(ierr); 2059bdf89e91SBarry Smith ierr = PetscObjectComposeFunction((PetscObject)pc,"PCBDDCSetNeumannBoundaries_C",PCBDDCSetNeumannBoundaries_BDDC);CHKERRQ(ierr); 206082ba6b80SStefano Zampini ierr = PetscObjectComposeFunction((PetscObject)pc,"PCBDDCSetNeumannBoundariesLocal_C",PCBDDCSetNeumannBoundariesLocal_BDDC);CHKERRQ(ierr); 2061bdf89e91SBarry Smith ierr = PetscObjectComposeFunction((PetscObject)pc,"PCBDDCGetDirichletBoundaries_C",PCBDDCGetDirichletBoundaries_BDDC);CHKERRQ(ierr); 206282ba6b80SStefano Zampini ierr = PetscObjectComposeFunction((PetscObject)pc,"PCBDDCGetDirichletBoundariesLocal_C",PCBDDCGetDirichletBoundariesLocal_BDDC);CHKERRQ(ierr); 2063bdf89e91SBarry Smith ierr = PetscObjectComposeFunction((PetscObject)pc,"PCBDDCGetNeumannBoundaries_C",PCBDDCGetNeumannBoundaries_BDDC);CHKERRQ(ierr); 206482ba6b80SStefano Zampini ierr = PetscObjectComposeFunction((PetscObject)pc,"PCBDDCGetNeumannBoundariesLocal_C",PCBDDCGetNeumannBoundariesLocal_BDDC);CHKERRQ(ierr); 2065bdf89e91SBarry Smith ierr = PetscObjectComposeFunction((PetscObject)pc,"PCBDDCSetDofsSplitting_C",PCBDDCSetDofsSplitting_BDDC);CHKERRQ(ierr); 206663602bcaSStefano Zampini ierr = PetscObjectComposeFunction((PetscObject)pc,"PCBDDCSetDofsSplittingLocal_C",PCBDDCSetDofsSplittingLocal_BDDC);CHKERRQ(ierr); 2067bdf89e91SBarry Smith ierr = PetscObjectComposeFunction((PetscObject)pc,"PCBDDCSetLocalAdjacencyGraph_C",PCBDDCSetLocalAdjacencyGraph_BDDC);CHKERRQ(ierr); 2068bdf89e91SBarry Smith ierr = PetscObjectComposeFunction((PetscObject)pc,"PCBDDCCreateFETIDPOperators_C",PCBDDCCreateFETIDPOperators_BDDC);CHKERRQ(ierr); 2069bdf89e91SBarry Smith ierr = PetscObjectComposeFunction((PetscObject)pc,"PCBDDCMatFETIDPGetRHS_C",PCBDDCMatFETIDPGetRHS_BDDC);CHKERRQ(ierr); 2070bdf89e91SBarry Smith ierr = PetscObjectComposeFunction((PetscObject)pc,"PCBDDCMatFETIDPGetSolution_C",PCBDDCMatFETIDPGetSolution_BDDC);CHKERRQ(ierr); 2071da1bb401SStefano Zampini PetscFunctionReturn(0); 2072da1bb401SStefano Zampini } 20733425bc38SStefano Zampini 2074