xref: /petsc/src/ksp/pc/impls/factor/cholesky/cholesky.c (revision 35bd34fa3344fb53ae05f23bf99eddd2f9bb15e6)
1dba47a55SKris Buschelman #define PETSCKSP_DLL
29b54502bSHong Zhang 
39b54502bSHong Zhang /*
49b54502bSHong Zhang    Defines a direct factorization preconditioner for any Mat implementation
59b54502bSHong Zhang    Note: this need not be consided a preconditioner since it supplies
69b54502bSHong Zhang          a direct solver.
79b54502bSHong Zhang */
86356e834SBarry Smith #include "private/pcimpl.h"                /*I "petscpc.h" I*/
99b54502bSHong Zhang 
109b54502bSHong Zhang typedef struct {
119b54502bSHong Zhang   Mat              fact;             /* factored matrix */
129b54502bSHong Zhang   PetscReal        actualfill;       /* actual fill in factor */
139b54502bSHong Zhang   PetscTruth       inplace;          /* flag indicating in-place factorization */
149b54502bSHong Zhang   IS               row,col;          /* index sets used for reordering */
159b54502bSHong Zhang   MatOrderingType  ordering;         /* matrix ordering */
169b54502bSHong Zhang   PetscTruth       reuseordering;    /* reuses previous reordering computed */
179b54502bSHong Zhang   PetscTruth       reusefill;        /* reuse fill from previous Cholesky */
189b54502bSHong Zhang   MatFactorInfo    info;
19c7393fdbSBarry Smith   MatSolverPackage solvertype;
209b54502bSHong Zhang } PC_Cholesky;
219b54502bSHong Zhang 
229b54502bSHong Zhang EXTERN_C_BEGIN
239b54502bSHong Zhang #undef __FUNCT__
24a541d17aSBarry Smith #define __FUNCT__ "PCFactorSetMatSolverPackage_Cholesky"
25c7393fdbSBarry Smith PetscErrorCode PETSCKSP_DLLEXPORT PCFactorSetMatSolverPackage_Cholesky(PC pc,const MatSolverPackage stype)
26c62fe420SBarry Smith {
27c62fe420SBarry Smith   PetscErrorCode ierr;
28*35bd34faSBarry Smith   PC_Cholesky    *cholesky = (PC_Cholesky*)pc->data;
29c62fe420SBarry Smith 
30c62fe420SBarry Smith   PetscFunctionBegin;
31*35bd34faSBarry Smith   if (cholesky->fact) {
32*35bd34faSBarry Smith     const MatSolverPackage ltype;
33*35bd34faSBarry Smith     PetscTruth             flg;
34*35bd34faSBarry Smith     ierr = MatFactorGetSolverPackage(cholesky->fact,&ltype);CHKERRQ(ierr);
35*35bd34faSBarry Smith     ierr = PetscStrcmp(stype,ltype,&flg);CHKERRQ(ierr);
36*35bd34faSBarry Smith     if (!flg) {
37*35bd34faSBarry Smith       SETERRQ(PETSC_ERR_ARG_WRONGSTATE,"Cannot change solver matrix package after PC has been setup or used");
38*35bd34faSBarry Smith     }
39*35bd34faSBarry Smith   }
40*35bd34faSBarry Smith   ierr = PetscStrfree(cholesky->solvertype);CHKERRQ(ierr);
41*35bd34faSBarry Smith   ierr = PetscStrallocpy(stype,&cholesky->solvertype);CHKERRQ(ierr);
42c62fe420SBarry Smith   PetscFunctionReturn(0);
43c62fe420SBarry Smith }
44c62fe420SBarry Smith EXTERN_C_END
45c62fe420SBarry Smith 
46c62fe420SBarry Smith EXTERN_C_BEGIN
47c62fe420SBarry Smith #undef __FUNCT__
48afaefe49SHong Zhang #define __FUNCT__ "PCFactorSetZeroPivot_Cholesky"
49dba47a55SKris Buschelman PetscErrorCode PETSCKSP_DLLEXPORT PCFactorSetZeroPivot_Cholesky(PC pc,PetscReal z)
50afaefe49SHong Zhang {
51afaefe49SHong Zhang   PC_Cholesky *ch;
52afaefe49SHong Zhang 
53afaefe49SHong Zhang   PetscFunctionBegin;
54afaefe49SHong Zhang   ch                 = (PC_Cholesky*)pc->data;
55afaefe49SHong Zhang   ch->info.zeropivot = z;
56afaefe49SHong Zhang   PetscFunctionReturn(0);
57afaefe49SHong Zhang }
58afaefe49SHong Zhang EXTERN_C_END
59afaefe49SHong Zhang 
60afaefe49SHong Zhang EXTERN_C_BEGIN
61afaefe49SHong Zhang #undef __FUNCT__
62afaefe49SHong Zhang #define __FUNCT__ "PCFactorSetShiftNonzero_Cholesky"
63dba47a55SKris Buschelman PetscErrorCode PETSCKSP_DLLEXPORT PCFactorSetShiftNonzero_Cholesky(PC pc,PetscReal shift)
64afaefe49SHong Zhang {
65afaefe49SHong Zhang   PC_Cholesky *dir;
66afaefe49SHong Zhang 
67afaefe49SHong Zhang   PetscFunctionBegin;
68afaefe49SHong Zhang   dir = (PC_Cholesky*)pc->data;
69afaefe49SHong Zhang   if (shift == (PetscReal) PETSC_DECIDE) {
70afaefe49SHong Zhang     dir->info.shiftnz = 1.e-12;
71afaefe49SHong Zhang   } else {
72afaefe49SHong Zhang     dir->info.shiftnz = shift;
73afaefe49SHong Zhang   }
74afaefe49SHong Zhang   PetscFunctionReturn(0);
75afaefe49SHong Zhang }
76afaefe49SHong Zhang EXTERN_C_END
77afaefe49SHong Zhang 
78afaefe49SHong Zhang EXTERN_C_BEGIN
79afaefe49SHong Zhang #undef __FUNCT__
80afaefe49SHong Zhang #define __FUNCT__ "PCFactorSetShiftPd_Cholesky"
81dba47a55SKris Buschelman PetscErrorCode PETSCKSP_DLLEXPORT PCFactorSetShiftPd_Cholesky(PC pc,PetscTruth shift)
82afaefe49SHong Zhang {
83afaefe49SHong Zhang   PC_Cholesky *dir;
84afaefe49SHong Zhang 
85afaefe49SHong Zhang   PetscFunctionBegin;
86afaefe49SHong Zhang   dir = (PC_Cholesky*)pc->data;
87fbf22428SSatish Balay   if (shift) {
88fbf22428SSatish Balay     dir->info.shiftpd = 1.0;
89fbf22428SSatish Balay   } else {
90fbf22428SSatish Balay     dir->info.shiftpd = 0.0;
91fbf22428SSatish Balay   }
92afaefe49SHong Zhang   PetscFunctionReturn(0);
93afaefe49SHong Zhang }
94afaefe49SHong Zhang EXTERN_C_END
95afaefe49SHong Zhang 
96afaefe49SHong Zhang EXTERN_C_BEGIN
97afaefe49SHong Zhang #undef __FUNCT__
982401956bSBarry Smith #define __FUNCT__ "PCFactorSetReuseOrdering_Cholesky"
992401956bSBarry Smith PetscErrorCode PETSCKSP_DLLEXPORT PCFactorSetReuseOrdering_Cholesky(PC pc,PetscTruth flag)
1009b54502bSHong Zhang {
1019b54502bSHong Zhang   PC_Cholesky *lu;
1029b54502bSHong Zhang 
1039b54502bSHong Zhang   PetscFunctionBegin;
1049b54502bSHong Zhang   lu               = (PC_Cholesky*)pc->data;
1059b54502bSHong Zhang   lu->reuseordering = flag;
1069b54502bSHong Zhang   PetscFunctionReturn(0);
1079b54502bSHong Zhang }
1089b54502bSHong Zhang EXTERN_C_END
1099b54502bSHong Zhang 
1109b54502bSHong Zhang EXTERN_C_BEGIN
1119b54502bSHong Zhang #undef __FUNCT__
1122401956bSBarry Smith #define __FUNCT__ "PCFactorSetReuseFill_Cholesky"
1132401956bSBarry Smith PetscErrorCode PETSCKSP_DLLEXPORT PCFactorSetReuseFill_Cholesky(PC pc,PetscTruth flag)
1149b54502bSHong Zhang {
1159b54502bSHong Zhang   PC_Cholesky *lu;
1169b54502bSHong Zhang 
1179b54502bSHong Zhang   PetscFunctionBegin;
1189b54502bSHong Zhang   lu = (PC_Cholesky*)pc->data;
1199b54502bSHong Zhang   lu->reusefill = flag;
1209b54502bSHong Zhang   PetscFunctionReturn(0);
1219b54502bSHong Zhang }
1229b54502bSHong Zhang EXTERN_C_END
1239b54502bSHong Zhang 
1249b54502bSHong Zhang #undef __FUNCT__
1259b54502bSHong Zhang #define __FUNCT__ "PCSetFromOptions_Cholesky"
1269b54502bSHong Zhang static PetscErrorCode PCSetFromOptions_Cholesky(PC pc)
1279b54502bSHong Zhang {
1289b54502bSHong Zhang   PC_Cholesky    *lu = (PC_Cholesky*)pc->data;
1299b54502bSHong Zhang   PetscErrorCode ierr;
1309b54502bSHong Zhang   PetscTruth     flg;
1315c9eb25fSBarry Smith   char           tname[256], solvertype[64];
1329b54502bSHong Zhang   PetscFList     ordlist;
1339b54502bSHong Zhang 
1349b54502bSHong Zhang   PetscFunctionBegin;
1359b54502bSHong Zhang   ierr = MatOrderingRegisterAll(PETSC_NULL);CHKERRQ(ierr);
1369b54502bSHong Zhang   ierr = PetscOptionsHead("Cholesky options");CHKERRQ(ierr);
1372401956bSBarry Smith     ierr = PetscOptionsName("-pc_factor_in_place","Form Cholesky in the same memory as the matrix","PCFactorSetUseInPlace",&flg);CHKERRQ(ierr);
1389b54502bSHong Zhang     if (flg) {
1392401956bSBarry Smith       ierr = PCFactorSetUseInPlace(pc);CHKERRQ(ierr);
1409b54502bSHong Zhang     }
14155ba2a51SBarry Smith     ierr = PetscOptionsReal("-pc_factor_fill","Expected non-zeros in Cholesky/non-zeros in matrix","PCFactorSetFill",lu->info.fill,&lu->info.fill,0);CHKERRQ(ierr);
1429b54502bSHong Zhang 
1432401956bSBarry Smith     ierr = PetscOptionsName("-pc_factor_reuse_fill","Use fill from previous factorization","PCFactorSetReuseFill",&flg);CHKERRQ(ierr);
1449b54502bSHong Zhang     if (flg) {
1452401956bSBarry Smith       ierr = PCFactorSetReuseFill(pc,PETSC_TRUE);CHKERRQ(ierr);
1469b54502bSHong Zhang     }
1472401956bSBarry Smith     ierr = PetscOptionsName("-pc_factor_reuse_ordering","Reuse ordering from previous factorization","PCFactorSetReuseOrdering",&flg);CHKERRQ(ierr);
1489b54502bSHong Zhang     if (flg) {
1492401956bSBarry Smith       ierr = PCFactorSetReuseOrdering(pc,PETSC_TRUE);CHKERRQ(ierr);
1509b54502bSHong Zhang     }
1519b54502bSHong Zhang 
1529b54502bSHong Zhang     ierr = MatGetOrderingList(&ordlist);CHKERRQ(ierr);
153e5a9bf91SBarry Smith     ierr = PetscOptionsList("-pc_factor_mat_ordering_type","Reordering to reduce nonzeros in Cholesky","PCFactorSetMatOrderingType",ordlist,lu->ordering,tname,256,&flg);CHKERRQ(ierr);
1549b54502bSHong Zhang     if (flg) {
155e5a9bf91SBarry Smith       ierr = PCFactorSetMatOrderingType(pc,tname);CHKERRQ(ierr);
1569b54502bSHong Zhang     }
1575c9eb25fSBarry Smith 
1585c9eb25fSBarry Smith     /* maybe should have MatGetSolverTypes(Mat,&list) like the ordering list */
159c7393fdbSBarry Smith     ierr = PetscOptionsString("-pc_factor_mat_solver_package","Specific Cholesky solver to use","MatGetFactor",lu->solvertype,solvertype,64,&flg);CHKERRQ(ierr);
1605c9eb25fSBarry Smith     if (flg) {
161c7393fdbSBarry Smith       ierr = PCFactorSetMatSolverPackage(pc,solvertype);CHKERRQ(ierr);
1625c9eb25fSBarry Smith     }
1635c9eb25fSBarry Smith 
1649f95998fSHong Zhang     ierr = PetscOptionsName("-pc_factor_shift_nonzero","Shift added to diagonal","PCFactorSetShiftNonzero",&flg);CHKERRQ(ierr);
1659b54502bSHong Zhang     if (flg) {
166afaefe49SHong Zhang       ierr = PCFactorSetShiftNonzero(pc,(PetscReal) PETSC_DECIDE);CHKERRQ(ierr);
1679b54502bSHong Zhang     }
1689f95998fSHong Zhang     ierr = PetscOptionsReal("-pc_factor_shift_nonzero","Shift added to diagonal","PCFactorSetShiftNonzero",lu->info.shiftnz,&lu->info.shiftnz,0);CHKERRQ(ierr);
1699f95998fSHong Zhang     ierr = PetscOptionsName("-pc_factor_shift_positive_definite","Manteuffel shift applied to diagonal","PCFactorSetShiftPd",&flg);CHKERRQ(ierr);
1709b54502bSHong Zhang     if (flg) {
171afaefe49SHong Zhang       ierr = PCFactorSetShiftPd(pc,PETSC_TRUE);CHKERRQ(ierr);
1729b54502bSHong Zhang     }
173ee45ca4aSHong Zhang     ierr = PetscOptionsReal("-pc_factor_zeropivot","Pivot is considered zero if less than","PCFactorSetZeroPivot",lu->info.zeropivot,&lu->info.zeropivot,0);CHKERRQ(ierr);
1749b54502bSHong Zhang 
1759b54502bSHong Zhang   ierr = PetscOptionsTail();CHKERRQ(ierr);
1769b54502bSHong Zhang   PetscFunctionReturn(0);
1779b54502bSHong Zhang }
1789b54502bSHong Zhang 
1799b54502bSHong Zhang #undef __FUNCT__
1809b54502bSHong Zhang #define __FUNCT__ "PCView_Cholesky"
1819b54502bSHong Zhang static PetscErrorCode PCView_Cholesky(PC pc,PetscViewer viewer)
1829b54502bSHong Zhang {
1839b54502bSHong Zhang   PC_Cholesky    *lu = (PC_Cholesky*)pc->data;
1849b54502bSHong Zhang   PetscErrorCode ierr;
1859b54502bSHong Zhang   PetscTruth     iascii,isstring;
1869b54502bSHong Zhang 
1879b54502bSHong Zhang   PetscFunctionBegin;
1889b54502bSHong Zhang   ierr = PetscTypeCompare((PetscObject)viewer,PETSC_VIEWER_ASCII,&iascii);CHKERRQ(ierr);
1899b54502bSHong Zhang   ierr = PetscTypeCompare((PetscObject)viewer,PETSC_VIEWER_STRING,&isstring);CHKERRQ(ierr);
1909b54502bSHong Zhang   if (iascii) {
1919b54502bSHong Zhang 
1929b54502bSHong Zhang     if (lu->inplace) {ierr = PetscViewerASCIIPrintf(viewer,"  Cholesky: in-place factorization\n");CHKERRQ(ierr);}
1939b54502bSHong Zhang     else             {ierr = PetscViewerASCIIPrintf(viewer,"  Cholesky: out-of-place factorization\n");CHKERRQ(ierr);}
1949b54502bSHong Zhang     ierr = PetscViewerASCIIPrintf(viewer,"    matrix ordering: %s\n",lu->ordering);CHKERRQ(ierr);
1959b54502bSHong Zhang     if (lu->fact) {
196f3a39becSBarry Smith       ierr = PetscViewerASCIIPrintf(viewer,"  Cholesky: factor fill ratio needed %G\n",lu->actualfill);CHKERRQ(ierr);
197f3a39becSBarry Smith       ierr = PetscViewerASCIIPrintf(viewer,"       Factored matrix follows\n");CHKERRQ(ierr);
198f3a39becSBarry Smith       ierr = PetscViewerASCIIPushTab(viewer);CHKERRQ(ierr);
199f3a39becSBarry Smith       ierr = PetscViewerASCIIPushTab(viewer);CHKERRQ(ierr);
200f3a39becSBarry Smith       ierr = PetscViewerASCIIPushTab(viewer);CHKERRQ(ierr);
201f3a39becSBarry Smith       ierr = PetscViewerPushFormat(viewer,PETSC_VIEWER_ASCII_INFO);CHKERRQ(ierr);
2029b54502bSHong Zhang       ierr = MatView(lu->fact,viewer);CHKERRQ(ierr);
2039b54502bSHong Zhang       ierr = PetscViewerPopFormat(viewer);CHKERRQ(ierr);
204f3a39becSBarry Smith       ierr = PetscViewerASCIIPopTab(viewer);CHKERRQ(ierr);
205f3a39becSBarry Smith       ierr = PetscViewerASCIIPopTab(viewer);CHKERRQ(ierr);
206f3a39becSBarry Smith       ierr = PetscViewerASCIIPopTab(viewer);CHKERRQ(ierr);
2079b54502bSHong Zhang     }
2089b54502bSHong Zhang     if (lu->reusefill)    {ierr = PetscViewerASCIIPrintf(viewer,"       Reusing fill from past factorization\n");CHKERRQ(ierr);}
2099b54502bSHong Zhang     if (lu->reuseordering) {ierr = PetscViewerASCIIPrintf(viewer,"       Reusing reordering from past factorization\n");CHKERRQ(ierr);}
2109b54502bSHong Zhang   } else if (isstring) {
2119b54502bSHong Zhang     ierr = PetscViewerStringSPrintf(viewer," order=%s",lu->ordering);CHKERRQ(ierr);CHKERRQ(ierr);
2129b54502bSHong Zhang   } else {
2132401956bSBarry Smith     SETERRQ1(PETSC_ERR_SUP,"Viewer type %s not supported for PCCHOLESKY",((PetscObject)viewer)->type_name);
2149b54502bSHong Zhang   }
2159b54502bSHong Zhang   PetscFunctionReturn(0);
2169b54502bSHong Zhang }
2179b54502bSHong Zhang 
2189b54502bSHong Zhang #undef __FUNCT__
219a4fd02acSBarry Smith #define __FUNCT__ "PCFactorGetMatrix_Cholesky"
220a4fd02acSBarry Smith static PetscErrorCode PCFactorGetMatrix_Cholesky(PC pc,Mat *mat)
2219b54502bSHong Zhang {
2229b54502bSHong Zhang   PC_Cholesky *dir = (PC_Cholesky*)pc->data;
2239b54502bSHong Zhang 
2249b54502bSHong Zhang   PetscFunctionBegin;
2259b54502bSHong Zhang   if (!dir->fact) SETERRQ(PETSC_ERR_ORDER,"Matrix not yet factored; call after KSPSetUp() or PCSetUp()");
2269b54502bSHong Zhang   *mat = dir->fact;
2279b54502bSHong Zhang   PetscFunctionReturn(0);
2289b54502bSHong Zhang }
2299b54502bSHong Zhang 
2309b54502bSHong Zhang #undef __FUNCT__
2319b54502bSHong Zhang #define __FUNCT__ "PCSetUp_Cholesky"
2329b54502bSHong Zhang static PetscErrorCode PCSetUp_Cholesky(PC pc)
2339b54502bSHong Zhang {
2349b54502bSHong Zhang   PetscErrorCode ierr;
2359b54502bSHong Zhang   PetscTruth     flg;
2369b54502bSHong Zhang   PC_Cholesky    *dir = (PC_Cholesky*)pc->data;
2379b54502bSHong Zhang 
2389b54502bSHong Zhang   PetscFunctionBegin;
2399b54502bSHong Zhang   if (dir->reusefill && pc->setupcalled) dir->info.fill = dir->actualfill;
2409b54502bSHong Zhang 
2419b54502bSHong Zhang   if (dir->inplace) {
2429b54502bSHong Zhang     if (dir->row && dir->col && (dir->row != dir->col)) {
2439b54502bSHong Zhang       ierr = ISDestroy(dir->row);CHKERRQ(ierr);
2449b54502bSHong Zhang       dir->row = 0;
2459b54502bSHong Zhang     }
2469b54502bSHong Zhang     if (dir->col) {
2479b54502bSHong Zhang       ierr = ISDestroy(dir->col);CHKERRQ(ierr);
2489b54502bSHong Zhang       dir->col = 0;
2499b54502bSHong Zhang     }
2509b54502bSHong Zhang     ierr = MatGetOrdering(pc->pmat,dir->ordering,&dir->row,&dir->col);CHKERRQ(ierr);
2519b54502bSHong Zhang     if (dir->col && (dir->row != dir->col)) {  /* only use row ordering for SBAIJ */
2529b54502bSHong Zhang       ierr = ISDestroy(dir->col);CHKERRQ(ierr);
2539b54502bSHong Zhang       dir->col=0;
2549b54502bSHong Zhang     }
255efee365bSSatish Balay     if (dir->row) {ierr = PetscLogObjectParent(pc,dir->row);CHKERRQ(ierr);}
2569b54502bSHong Zhang     ierr = MatCholeskyFactor(pc->pmat,dir->row,&dir->info);CHKERRQ(ierr);
2579b54502bSHong Zhang     dir->fact = pc->pmat;
2589b54502bSHong Zhang   } else {
2599b54502bSHong Zhang     MatInfo info;
2609b54502bSHong Zhang     if (!pc->setupcalled) {
2619b54502bSHong Zhang       ierr = MatGetOrdering(pc->pmat,dir->ordering,&dir->row,&dir->col);CHKERRQ(ierr);
2629bfd6278SHong Zhang       /* check if dir->row == dir->col */
2639bfd6278SHong Zhang       ierr = ISEqual(dir->row,dir->col,&flg);CHKERRQ(ierr);
2649bfd6278SHong Zhang       if (!flg) SETERRQ(PETSC_ERR_ARG_INCOMP,"row and column permutations must equal");
2659bfd6278SHong Zhang       ierr = ISDestroy(dir->col);CHKERRQ(ierr); /* only pass one ordering into CholeskyFactor */
2669b54502bSHong Zhang       dir->col=0;
2679bfd6278SHong Zhang 
2687adad957SLisandro Dalcin       ierr = PetscOptionsHasName(((PetscObject)pc)->prefix,"-pc_factor_nonzeros_along_diagonal",&flg);CHKERRQ(ierr);
2699b54502bSHong Zhang       if (flg) {
2709b54502bSHong Zhang         PetscReal tol = 1.e-10;
2717adad957SLisandro Dalcin         ierr = PetscOptionsGetReal(((PetscObject)pc)->prefix,"-pc_factor_nonzeros_along_diagonal",&tol,PETSC_NULL);CHKERRQ(ierr);
2729b54502bSHong Zhang         ierr = MatReorderForNonzeroDiagonal(pc->pmat,tol,dir->row,dir->row);CHKERRQ(ierr);
2739b54502bSHong Zhang       }
274efee365bSSatish Balay       if (dir->row) {ierr = PetscLogObjectParent(pc,dir->row);CHKERRQ(ierr);}
2755c9eb25fSBarry Smith       ierr = MatGetFactor(pc->pmat,dir->solvertype,MAT_FACTOR_CHOLESKY,&dir->fact);CHKERRQ(ierr);
276719d5645SBarry Smith       ierr = MatCholeskyFactorSymbolic(dir->fact,pc->pmat,dir->row,&dir->info);CHKERRQ(ierr);
2779b54502bSHong Zhang       ierr = MatGetInfo(dir->fact,MAT_LOCAL,&info);CHKERRQ(ierr);
2789b54502bSHong Zhang       dir->actualfill = info.fill_ratio_needed;
27952e6d16bSBarry Smith       ierr = PetscLogObjectParent(pc,dir->fact);CHKERRQ(ierr);
2809b54502bSHong Zhang     } else if (pc->flag != SAME_NONZERO_PATTERN) {
2819b54502bSHong Zhang       if (!dir->reuseordering) {
2829b54502bSHong Zhang         if (dir->row && dir->col && (dir->row != dir->col)) {
2839b54502bSHong Zhang           ierr = ISDestroy(dir->row);CHKERRQ(ierr);
2849b54502bSHong Zhang           dir->row = 0;
2859b54502bSHong Zhang         }
2869b54502bSHong Zhang         if (dir->col) {
2879b54502bSHong Zhang           ierr = ISDestroy(dir->col);CHKERRQ(ierr);
2889b54502bSHong Zhang           dir->col =0;
2899b54502bSHong Zhang         }
2909b54502bSHong Zhang         ierr = MatGetOrdering(pc->pmat,dir->ordering,&dir->row,&dir->col);CHKERRQ(ierr);
2919b54502bSHong Zhang         if (dir->col && (dir->row != dir->col)) {  /* only use row ordering for SBAIJ */
2929b54502bSHong Zhang           ierr = ISDestroy(dir->col);CHKERRQ(ierr);
2939b54502bSHong Zhang           dir->col=0;
2949b54502bSHong Zhang         }
2957adad957SLisandro Dalcin         ierr = PetscOptionsHasName(((PetscObject)pc)->prefix,"-pc_factor_nonzeros_along_diagonal",&flg);CHKERRQ(ierr);
2969b54502bSHong Zhang         if (flg) {
2979b54502bSHong Zhang           PetscReal tol = 1.e-10;
2987adad957SLisandro Dalcin           ierr = PetscOptionsGetReal(((PetscObject)pc)->prefix,"-pc_factor_nonzeros_along_diagonal",&tol,PETSC_NULL);CHKERRQ(ierr);
2999b54502bSHong Zhang           ierr = MatReorderForNonzeroDiagonal(pc->pmat,tol,dir->row,dir->row);CHKERRQ(ierr);
3009b54502bSHong Zhang         }
301efee365bSSatish Balay         if (dir->row) {ierr = PetscLogObjectParent(pc,dir->row);CHKERRQ(ierr);}
3029b54502bSHong Zhang       }
3039b54502bSHong Zhang       ierr = MatDestroy(dir->fact);CHKERRQ(ierr);
3045c9eb25fSBarry Smith       ierr = MatGetFactor(pc->pmat,dir->solvertype,MAT_FACTOR_CHOLESKY,&dir->fact);CHKERRQ(ierr);
305719d5645SBarry Smith       ierr = MatCholeskyFactorSymbolic(dir->fact,pc->pmat,dir->row,&dir->info);CHKERRQ(ierr);
3069b54502bSHong Zhang       ierr = MatGetInfo(dir->fact,MAT_LOCAL,&info);CHKERRQ(ierr);
3079b54502bSHong Zhang       dir->actualfill = info.fill_ratio_needed;
30852e6d16bSBarry Smith       ierr = PetscLogObjectParent(pc,dir->fact);CHKERRQ(ierr);
3099b54502bSHong Zhang     }
310719d5645SBarry Smith     ierr = MatCholeskyFactorNumeric(dir->fact,pc->pmat,&dir->info);CHKERRQ(ierr);
3119b54502bSHong Zhang   }
3129b54502bSHong Zhang   PetscFunctionReturn(0);
3139b54502bSHong Zhang }
3149b54502bSHong Zhang 
3159b54502bSHong Zhang #undef __FUNCT__
3169b54502bSHong Zhang #define __FUNCT__ "PCDestroy_Cholesky"
3179b54502bSHong Zhang static PetscErrorCode PCDestroy_Cholesky(PC pc)
3189b54502bSHong Zhang {
3199b54502bSHong Zhang   PC_Cholesky    *dir = (PC_Cholesky*)pc->data;
3209b54502bSHong Zhang   PetscErrorCode ierr;
3219b54502bSHong Zhang 
3229b54502bSHong Zhang   PetscFunctionBegin;
3239b54502bSHong Zhang   if (!dir->inplace && dir->fact) {ierr = MatDestroy(dir->fact);CHKERRQ(ierr);}
3249b54502bSHong Zhang   if (dir->row) {ierr = ISDestroy(dir->row);CHKERRQ(ierr);}
3259b54502bSHong Zhang   if (dir->col) {ierr = ISDestroy(dir->col);CHKERRQ(ierr);}
3269b54502bSHong Zhang   ierr = PetscStrfree(dir->ordering);CHKERRQ(ierr);
3275c9eb25fSBarry Smith   ierr = PetscStrfree(dir->solvertype);CHKERRQ(ierr);
3289b54502bSHong Zhang   ierr = PetscFree(dir);CHKERRQ(ierr);
3299b54502bSHong Zhang   PetscFunctionReturn(0);
3309b54502bSHong Zhang }
3319b54502bSHong Zhang 
3329b54502bSHong Zhang #undef __FUNCT__
3339b54502bSHong Zhang #define __FUNCT__ "PCApply_Cholesky"
3349b54502bSHong Zhang static PetscErrorCode PCApply_Cholesky(PC pc,Vec x,Vec y)
3359b54502bSHong Zhang {
3369b54502bSHong Zhang   PC_Cholesky    *dir = (PC_Cholesky*)pc->data;
3379b54502bSHong Zhang   PetscErrorCode ierr;
3389b54502bSHong Zhang 
3399b54502bSHong Zhang   PetscFunctionBegin;
3409b54502bSHong Zhang   if (dir->inplace) {ierr = MatSolve(pc->pmat,x,y);CHKERRQ(ierr);}
3419b54502bSHong Zhang   else              {ierr = MatSolve(dir->fact,x,y);CHKERRQ(ierr);}
3429b54502bSHong Zhang   PetscFunctionReturn(0);
3439b54502bSHong Zhang }
3449b54502bSHong Zhang 
3459b54502bSHong Zhang #undef __FUNCT__
3469b54502bSHong Zhang #define __FUNCT__ "PCApplyTranspose_Cholesky"
3479b54502bSHong Zhang static PetscErrorCode PCApplyTranspose_Cholesky(PC pc,Vec x,Vec y)
3489b54502bSHong Zhang {
3499b54502bSHong Zhang   PC_Cholesky    *dir = (PC_Cholesky*)pc->data;
3509b54502bSHong Zhang   PetscErrorCode ierr;
3519b54502bSHong Zhang 
3529b54502bSHong Zhang   PetscFunctionBegin;
3539b54502bSHong Zhang   if (dir->inplace) {ierr = MatSolveTranspose(pc->pmat,x,y);CHKERRQ(ierr);}
3549b54502bSHong Zhang   else              {ierr = MatSolveTranspose(dir->fact,x,y);CHKERRQ(ierr);}
3559b54502bSHong Zhang   PetscFunctionReturn(0);
3569b54502bSHong Zhang }
3579b54502bSHong Zhang 
3589b54502bSHong Zhang /* -----------------------------------------------------------------------------------*/
3599b54502bSHong Zhang 
3609b54502bSHong Zhang EXTERN_C_BEGIN
3619b54502bSHong Zhang #undef __FUNCT__
36255ba2a51SBarry Smith #define __FUNCT__ "PCFactorSetFill_Cholesky"
36355ba2a51SBarry Smith PetscErrorCode PETSCKSP_DLLEXPORT PCFactorSetFill_Cholesky(PC pc,PetscReal fill)
3649b54502bSHong Zhang {
3659b54502bSHong Zhang   PC_Cholesky *dir;
3669b54502bSHong Zhang 
3679b54502bSHong Zhang   PetscFunctionBegin;
3689b54502bSHong Zhang   dir = (PC_Cholesky*)pc->data;
3699b54502bSHong Zhang   dir->info.fill = fill;
3709b54502bSHong Zhang   PetscFunctionReturn(0);
3719b54502bSHong Zhang }
3729b54502bSHong Zhang EXTERN_C_END
3739b54502bSHong Zhang 
3749b54502bSHong Zhang EXTERN_C_BEGIN
3759b54502bSHong Zhang #undef __FUNCT__
3762401956bSBarry Smith #define __FUNCT__ "PCFactorSetUseInPlace_Cholesky"
3772401956bSBarry Smith PetscErrorCode PETSCKSP_DLLEXPORT PCFactorSetUseInPlace_Cholesky(PC pc)
3789b54502bSHong Zhang {
3799b54502bSHong Zhang   PC_Cholesky *dir;
3809b54502bSHong Zhang 
3819b54502bSHong Zhang   PetscFunctionBegin;
3829b54502bSHong Zhang   dir = (PC_Cholesky*)pc->data;
3839b54502bSHong Zhang   dir->inplace = PETSC_TRUE;
3849b54502bSHong Zhang   PetscFunctionReturn(0);
3859b54502bSHong Zhang }
3869b54502bSHong Zhang EXTERN_C_END
3879b54502bSHong Zhang 
3889b54502bSHong Zhang EXTERN_C_BEGIN
3899b54502bSHong Zhang #undef __FUNCT__
390e5a9bf91SBarry Smith #define __FUNCT__ "PCFactorSetMatOrderingType_Cholesky"
391e36ff184SSatish Balay PetscErrorCode PETSCKSP_DLLEXPORT PCFactorSetMatOrderingType_Cholesky(PC pc,const MatOrderingType ordering)
3929b54502bSHong Zhang {
3939b54502bSHong Zhang   PC_Cholesky    *dir = (PC_Cholesky*)pc->data;
3949b54502bSHong Zhang   PetscErrorCode ierr;
3959b54502bSHong Zhang 
3969b54502bSHong Zhang   PetscFunctionBegin;
3979b54502bSHong Zhang   ierr = PetscStrfree(dir->ordering);CHKERRQ(ierr);
3989b54502bSHong Zhang   ierr = PetscStrallocpy(ordering,&dir->ordering);CHKERRQ(ierr);
3999b54502bSHong Zhang   PetscFunctionReturn(0);
4009b54502bSHong Zhang }
4019b54502bSHong Zhang EXTERN_C_END
4029b54502bSHong Zhang 
4039b54502bSHong Zhang /* -----------------------------------------------------------------------------------*/
4049b54502bSHong Zhang 
4059b54502bSHong Zhang #undef __FUNCT__
4062401956bSBarry Smith #define __FUNCT__ "PCFactorSetReuseOrdering"
4079b54502bSHong Zhang /*@
4082401956bSBarry Smith    PCFactorSetReuseOrdering - When similar matrices are factored, this
4099b54502bSHong Zhang    causes the ordering computed in the first factor to be used for all
4109b54502bSHong Zhang    following factors.
4119b54502bSHong Zhang 
4129b54502bSHong Zhang    Collective on PC
4139b54502bSHong Zhang 
4149b54502bSHong Zhang    Input Parameters:
4159b54502bSHong Zhang +  pc - the preconditioner context
4169b54502bSHong Zhang -  flag - PETSC_TRUE to reuse else PETSC_FALSE
4179b54502bSHong Zhang 
4189b54502bSHong Zhang    Options Database Key:
4192401956bSBarry Smith .  -pc_factor_reuse_ordering - Activate PCFactorSetReuseOrdering()
4209b54502bSHong Zhang 
4219b54502bSHong Zhang    Level: intermediate
4229b54502bSHong Zhang 
4239b54502bSHong Zhang .keywords: PC, levels, reordering, factorization, incomplete, LU
4249b54502bSHong Zhang 
4252401956bSBarry Smith .seealso: PCFactorSetReuseFill()
4269b54502bSHong Zhang @*/
4272401956bSBarry Smith PetscErrorCode PETSCKSP_DLLEXPORT PCFactorSetReuseOrdering(PC pc,PetscTruth flag)
4289b54502bSHong Zhang {
4299b54502bSHong Zhang   PetscErrorCode ierr,(*f)(PC,PetscTruth);
4309b54502bSHong Zhang 
4319b54502bSHong Zhang   PetscFunctionBegin;
4329b54502bSHong Zhang   PetscValidHeaderSpecific(pc,PC_COOKIE,1);
4332401956bSBarry Smith   ierr = PetscObjectQueryFunction((PetscObject)pc,"PCFactorSetReuseOrdering_C",(void (**)(void))&f);CHKERRQ(ierr);
4349b54502bSHong Zhang   if (f) {
4359b54502bSHong Zhang     ierr = (*f)(pc,flag);CHKERRQ(ierr);
4369b54502bSHong Zhang   }
4379b54502bSHong Zhang   PetscFunctionReturn(0);
4389b54502bSHong Zhang }
4399b54502bSHong Zhang 
4409b54502bSHong Zhang #undef __FUNCT__
4412401956bSBarry Smith #define __FUNCT__ "PCFactorSetReuseFill"
4429b54502bSHong Zhang /*@
4432401956bSBarry Smith    PCFactorSetReuseFill - When matrices with same different nonzero structure are factored,
4442401956bSBarry Smith    this causes later ones to use the fill ratio computed in the initial factorization.
4459b54502bSHong Zhang 
4469b54502bSHong Zhang    Collective on PC
4479b54502bSHong Zhang 
4489b54502bSHong Zhang    Input Parameters:
4499b54502bSHong Zhang +  pc - the preconditioner context
4509b54502bSHong Zhang -  flag - PETSC_TRUE to reuse else PETSC_FALSE
4519b54502bSHong Zhang 
4529b54502bSHong Zhang    Options Database Key:
4532401956bSBarry Smith .  -pc_factor_reuse_fill - Activates PCFactorSetReuseFill()
4549b54502bSHong Zhang 
4559b54502bSHong Zhang    Level: intermediate
4569b54502bSHong Zhang 
4579b54502bSHong Zhang .keywords: PC, levels, reordering, factorization, incomplete, Cholesky
4589b54502bSHong Zhang 
4592401956bSBarry Smith .seealso: PCFactorSetReuseOrdering()
4609b54502bSHong Zhang @*/
4612401956bSBarry Smith PetscErrorCode PETSCKSP_DLLEXPORT PCFactorSetReuseFill(PC pc,PetscTruth flag)
4629b54502bSHong Zhang {
4639b54502bSHong Zhang   PetscErrorCode ierr,(*f)(PC,PetscTruth);
4649b54502bSHong Zhang 
4659b54502bSHong Zhang   PetscFunctionBegin;
4669b54502bSHong Zhang   PetscValidHeaderSpecific(pc,PC_COOKIE,2);
4672401956bSBarry Smith   ierr = PetscObjectQueryFunction((PetscObject)pc,"PCFactorSetReuseFill_C",(void (**)(void))&f);CHKERRQ(ierr);
4689b54502bSHong Zhang   if (f) {
4699b54502bSHong Zhang     ierr = (*f)(pc,flag);CHKERRQ(ierr);
4709b54502bSHong Zhang   }
4719b54502bSHong Zhang   PetscFunctionReturn(0);
4729b54502bSHong Zhang }
4739b54502bSHong Zhang 
4749b54502bSHong Zhang /*MC
47596fc60bcSBarry Smith    PCCHOLESKY - Uses a direct solver, based on Cholesky factorization, as a preconditioner
4769b54502bSHong Zhang 
4779b54502bSHong Zhang    Options Database Keys:
4782401956bSBarry Smith +  -pc_factor_reuse_ordering - Activate PCFactorSetReuseOrdering()
479c7393fdbSBarry Smith .  -pc_factor_mat_solver_package - Actives PCFactorSetMatSolverPackage() to choose the direct solver, like spooles
4802401956bSBarry Smith .  -pc_factor_reuse_fill - Activates PCFactorSetReuseFill()
48155ba2a51SBarry Smith .  -pc_factor_fill <fill> - Sets fill amount
4822401956bSBarry Smith .  -pc_factor_in_place - Activates in-place factorization
4832401956bSBarry Smith .  -pc_factor_mat_ordering_type <nd,rcm,...> - Sets ordering routine
484f251bdbdSHong Zhang .  -pc_factor_shift_nonzero <shift> - Sets shift amount or PETSC_DECIDE for the default
485f251bdbdSHong Zhang -  -pc_factor_shift_positive_definite [PETSC_TRUE/PETSC_FALSE] - Activate/Deactivate PCFactorSetShiftPd(); the value
486f251bdbdSHong Zhang    is optional with PETSC_TRUE being the default
4879b54502bSHong Zhang 
4889b54502bSHong Zhang    Notes: Not all options work for all matrix formats
4899b54502bSHong Zhang 
4909b54502bSHong Zhang    Level: beginner
4919b54502bSHong Zhang 
4929b54502bSHong Zhang    Concepts: Cholesky factorization, direct solver
4939b54502bSHong Zhang 
4949b54502bSHong Zhang    Notes: Usually this will compute an "exact" solution in one iteration and does
4959b54502bSHong Zhang           not need a Krylov method (i.e. you can use -ksp_type preonly, or
4969b54502bSHong Zhang           KSPSetType(ksp,KSPPREONLY) for the Krylov method
4979b54502bSHong Zhang 
4989b54502bSHong Zhang .seealso:  PCCreate(), PCSetType(), PCType (for list of available types), PC,
499a4fd02acSBarry Smith            PCILU, PCLU, PCICC, PCFactorSetReuseOrdering(), PCFactorSetReuseFill(), PCFactorGetMatrix(),
50055ba2a51SBarry Smith            PCFactorSetFill(), PCFactorSetShiftNonzero(), PCFactorSetShiftPd(),
501e5a9bf91SBarry Smith 	   PCFactorSetUseInPlace(), PCFactorSetMatOrderingType(),PCFactorSetShiftNonzero(),PCFactorSetShiftPd()
5029b54502bSHong Zhang 
5039b54502bSHong Zhang M*/
5049b54502bSHong Zhang 
5059b54502bSHong Zhang EXTERN_C_BEGIN
5069b54502bSHong Zhang #undef __FUNCT__
5079b54502bSHong Zhang #define __FUNCT__ "PCCreate_Cholesky"
508dba47a55SKris Buschelman PetscErrorCode PETSCKSP_DLLEXPORT PCCreate_Cholesky(PC pc)
5099b54502bSHong Zhang {
5109b54502bSHong Zhang   PetscErrorCode ierr;
5119b54502bSHong Zhang   PC_Cholesky    *dir;
5129b54502bSHong Zhang 
5139b54502bSHong Zhang   PetscFunctionBegin;
51438f2d2fdSLisandro Dalcin   ierr = PetscNewLog(pc,PC_Cholesky,&dir);CHKERRQ(ierr);
5159b54502bSHong Zhang 
5169b54502bSHong Zhang   dir->fact                   = 0;
5179b54502bSHong Zhang   dir->inplace                = PETSC_FALSE;
5189b54502bSHong Zhang   ierr = MatFactorInfoInitialize(&dir->info);CHKERRQ(ierr);
5199b54502bSHong Zhang   dir->info.fill              = 5.0;
5200a29876aSHong Zhang   dir->info.shiftnz           = 0.0;
521fbf22428SSatish Balay   dir->info.shiftpd           = 0.0; /* false */
5229b54502bSHong Zhang   dir->info.pivotinblocks     = 1.0;
5239b54502bSHong Zhang   dir->col                    = 0;
5249b54502bSHong Zhang   dir->row                    = 0;
5259b54502bSHong Zhang   ierr = PetscStrallocpy(MATORDERING_NATURAL,&dir->ordering);CHKERRQ(ierr);
52643244d56SBarry Smith   ierr = PetscStrallocpy(MAT_SOLVER_PETSC,&dir->solvertype);CHKERRQ(ierr);
5279b54502bSHong Zhang   dir->reusefill        = PETSC_FALSE;
5289b54502bSHong Zhang   dir->reuseordering    = PETSC_FALSE;
5299b54502bSHong Zhang   pc->data              = (void*)dir;
5309b54502bSHong Zhang 
5319b54502bSHong Zhang   pc->ops->destroy           = PCDestroy_Cholesky;
5329b54502bSHong Zhang   pc->ops->apply             = PCApply_Cholesky;
5339b54502bSHong Zhang   pc->ops->applytranspose    = PCApplyTranspose_Cholesky;
5349b54502bSHong Zhang   pc->ops->setup             = PCSetUp_Cholesky;
5359b54502bSHong Zhang   pc->ops->setfromoptions    = PCSetFromOptions_Cholesky;
5369b54502bSHong Zhang   pc->ops->view              = PCView_Cholesky;
5379b54502bSHong Zhang   pc->ops->applyrichardson   = 0;
538a4fd02acSBarry Smith   pc->ops->getfactoredmatrix = PCFactorGetMatrix_Cholesky;
5399b54502bSHong Zhang 
540c7393fdbSBarry Smith   ierr = PetscObjectComposeFunctionDynamic((PetscObject)pc,"PCFactorSetMatSolverPackage_C","PCFactorSetMatSolverPackage_Cholesky",
541c7393fdbSBarry Smith                     PCFactorSetMatSolverPackage_Cholesky);CHKERRQ(ierr);
542afaefe49SHong Zhang   ierr = PetscObjectComposeFunctionDynamic((PetscObject)pc,"PCFactorSetZeroPivot_C","PCFactorSetZeroPivot_Cholesky",
543afaefe49SHong Zhang                     PCFactorSetZeroPivot_Cholesky);CHKERRQ(ierr);
544afaefe49SHong Zhang   ierr = PetscObjectComposeFunctionDynamic((PetscObject)pc,"PCFactorSetShiftNonzero_C","PCFactorSetShiftNonzero_Cholesky",
545afaefe49SHong Zhang                     PCFactorSetShiftNonzero_Cholesky);CHKERRQ(ierr);
546afaefe49SHong Zhang   ierr = PetscObjectComposeFunctionDynamic((PetscObject)pc,"PCFactorSetShiftPd_C","PCFactorSetShiftPd_Cholesky",
547afaefe49SHong Zhang                     PCFactorSetShiftPd_Cholesky);CHKERRQ(ierr);
548afaefe49SHong Zhang 
54955ba2a51SBarry Smith   ierr = PetscObjectComposeFunctionDynamic((PetscObject)pc,"PCFactorSetFill_C","PCFactorSetFill_Cholesky",
55055ba2a51SBarry Smith                     PCFactorSetFill_Cholesky);CHKERRQ(ierr);
5512401956bSBarry Smith   ierr = PetscObjectComposeFunctionDynamic((PetscObject)pc,"PCFactorSetUseInPlace_C","PCFactorSetUseInPlace_Cholesky",
5522401956bSBarry Smith                     PCFactorSetUseInPlace_Cholesky);CHKERRQ(ierr);
553e5a9bf91SBarry Smith   ierr = PetscObjectComposeFunctionDynamic((PetscObject)pc,"PCFactorSetMatOrderingType_C","PCFactorSetMatOrderingType_Cholesky",
554e5a9bf91SBarry Smith                     PCFactorSetMatOrderingType_Cholesky);CHKERRQ(ierr);
5552401956bSBarry Smith   ierr = PetscObjectComposeFunctionDynamic((PetscObject)pc,"PCFactorSetReuseOrdering_C","PCFactorSetReuseOrdering_Cholesky",
5562401956bSBarry Smith                     PCFactorSetReuseOrdering_Cholesky);CHKERRQ(ierr);
5572401956bSBarry Smith   ierr = PetscObjectComposeFunctionDynamic((PetscObject)pc,"PCFactorSetReuseFill_C","PCFactorSetReuseFill_Cholesky",
5582401956bSBarry Smith                     PCFactorSetReuseFill_Cholesky);CHKERRQ(ierr);
5599b54502bSHong Zhang   PetscFunctionReturn(0);
5609b54502bSHong Zhang }
5619b54502bSHong Zhang EXTERN_C_END
562