xref: /petsc/include/petscksp.h (revision 30de9b251c71d073cc9d88242eefe8db4dcc3e99)
173f4d377SMatthew Knepley /* $Id: petscksp.h,v 1.107 2001/08/06 21:16:38 bsmith Exp $ */
2f26ada1bSBarry Smith /*
3f26ada1bSBarry Smith    Defines the interface functions for the Krylov subspace accelerators.
4f26ada1bSBarry Smith */
50a835dfdSSatish Balay #ifndef __PETSCKSP_H
60a835dfdSSatish Balay #define __PETSCKSP_H
70a835dfdSSatish Balay #include "petscpc.h"
82eac72dbSBarry Smith 
928ce4d24SBarry Smith /*S
1028ce4d24SBarry Smith      KSP - Abstract PETSc object that manages all Krylov methods
1128ce4d24SBarry Smith 
1228ce4d24SBarry Smith    Level: beginner
1328ce4d24SBarry Smith 
1428ce4d24SBarry Smith   Concepts: Krylov methods
1528ce4d24SBarry Smith 
16435da068SBarry Smith .seealso:  KSPCreate(), KSPSetType(), KSPType, SNES, TS, PC, SLES
1728ce4d24SBarry Smith S*/
18e2a1c21fSSatish Balay typedef struct _p_KSP*     KSP;
192eac72dbSBarry Smith 
2028ce4d24SBarry Smith /*E
2128ce4d24SBarry Smith     KSPType - String with the name of a PETSc Krylov method or the creation function
2228ce4d24SBarry Smith        with an optional dynamic library name, for example
2328ce4d24SBarry Smith        http://www.mcs.anl.gov/petsc/lib.a:mykspcreate()
2428ce4d24SBarry Smith 
2528ce4d24SBarry Smith    Level: beginner
2628ce4d24SBarry Smith 
2728ce4d24SBarry Smith .seealso: KSPSetType(), KSP
2828ce4d24SBarry Smith E*/
2982bf6240SBarry Smith #define KSPRICHARDSON "richardson"
3082bf6240SBarry Smith #define KSPCHEBYCHEV  "chebychev"
3182bf6240SBarry Smith #define KSPCG         "cg"
3282bf6240SBarry Smith #define KSPGMRES      "gmres"
3382bf6240SBarry Smith #define KSPTCQMR      "tcqmr"
3482bf6240SBarry Smith #define KSPBCGS       "bcgs"
3582bf6240SBarry Smith #define KSPCGS        "cgs"
3682bf6240SBarry Smith #define KSPTFQMR      "tfqmr"
3782bf6240SBarry Smith #define KSPCR         "cr"
3882bf6240SBarry Smith #define KSPLSQR       "lsqr"
3982bf6240SBarry Smith #define KSPPREONLY    "preonly"
4082bf6240SBarry Smith #define KSPQCG        "qcg"
41c9cf9b11SBarry Smith #define KSPBICG       "bicg"
42c38d4ed2SBarry Smith #define KSPFGMRES     "fgmres"
43b4ac9ba4SBarry Smith #define KSPMINRES     "minres"
4401c5daebSSatish Balay #define KSPSYMMLQ     "symmlq"
45bae98175SBarry Smith #define KSPLGMRES     "lgmres"
4682bf6240SBarry Smith typedef char * KSPType;
472eac72dbSBarry Smith 
488ba1e511SMatthew Knepley /* Logging support */
498ba1e511SMatthew Knepley extern int KSP_COOKIE;
50d5ba7fb7SMatthew Knepley extern int KSP_GMRESOrthogonalization;
518ba1e511SMatthew Knepley 
52ca44d042SBarry Smith EXTERN int KSPCreate(MPI_Comm,KSP *);
53ca44d042SBarry Smith EXTERN int KSPSetType(KSP,KSPType);
54ca44d042SBarry Smith EXTERN int KSPSetUp(KSP);
55ca44d042SBarry Smith EXTERN int KSPSolve(KSP,int *);
56ca44d042SBarry Smith EXTERN int KSPSolveTranspose(KSP,int *);
57ca44d042SBarry Smith EXTERN int KSPDestroy(KSP);
582eac72dbSBarry Smith 
59b0a32e0cSBarry Smith extern PetscFList KSPList;
60ca44d042SBarry Smith EXTERN int KSPRegisterAll(char *);
61ca44d042SBarry Smith EXTERN int KSPRegisterDestroy(void);
622eac72dbSBarry Smith 
63ca44d042SBarry Smith EXTERN int KSPRegister(char*,char*,char*,int(*)(KSP));
64*30de9b25SBarry Smith 
65*30de9b25SBarry Smith /*MC
66*30de9b25SBarry Smith    KSPRegisterDynamic - Adds a method to the Krylov subspace solver package.
67*30de9b25SBarry Smith 
68*30de9b25SBarry Smith    Synopsis:
69*30de9b25SBarry Smith    int KSPRegisterDynamic(char *name_solver,char *path,char *name_create,int (*routine_create)(KSP))
70*30de9b25SBarry Smith 
71*30de9b25SBarry Smith    Not Collective
72*30de9b25SBarry Smith 
73*30de9b25SBarry Smith    Input Parameters:
74*30de9b25SBarry Smith +  name_solver - name of a new user-defined solver
75*30de9b25SBarry Smith .  path - path (either absolute or relative) the library containing this solver
76*30de9b25SBarry Smith .  name_create - name of routine to create method context
77*30de9b25SBarry Smith -  routine_create - routine to create method context
78*30de9b25SBarry Smith 
79*30de9b25SBarry Smith    Notes:
80*30de9b25SBarry Smith    KSPRegisterDynamic() may be called multiple times to add several user-defined solvers.
81*30de9b25SBarry Smith 
82*30de9b25SBarry Smith    If dynamic libraries are used, then the fourth input argument (routine_create)
83*30de9b25SBarry Smith    is ignored.
84*30de9b25SBarry Smith 
85*30de9b25SBarry Smith    Sample usage:
86*30de9b25SBarry Smith .vb
87*30de9b25SBarry Smith    KSPRegisterDynamic("my_solver",/home/username/my_lib/lib/libO/solaris/mylib.a,
88*30de9b25SBarry Smith                "MySolverCreate",MySolverCreate);
89*30de9b25SBarry Smith .ve
90*30de9b25SBarry Smith 
91*30de9b25SBarry Smith    Then, your solver can be chosen with the procedural interface via
92*30de9b25SBarry Smith $     KSPSetType(ksp,"my_solver")
93*30de9b25SBarry Smith    or at runtime via the option
94*30de9b25SBarry Smith $     -ksp_type my_solver
95*30de9b25SBarry Smith 
96*30de9b25SBarry Smith    Level: advanced
97*30de9b25SBarry Smith 
98*30de9b25SBarry Smith    Notes: Environmental variables such as ${PETSC_ARCH}, ${PETSC_DIR}, ${PETSC_LIB_DIR}, ${BOPT},
99*30de9b25SBarry Smith           and others of the form ${any_environmental_variable} occuring in pathname will be
100*30de9b25SBarry Smith           replaced with appropriate values.
101*30de9b25SBarry Smith          If your function is not being put into a shared library then use KSPRegister() instead
102*30de9b25SBarry Smith 
103*30de9b25SBarry Smith .keywords: KSP, register
104*30de9b25SBarry Smith 
105*30de9b25SBarry Smith .seealso: KSPRegisterAll(), KSPRegisterDestroy()
106*30de9b25SBarry Smith 
107*30de9b25SBarry Smith M*/
108aa482453SBarry Smith #if defined(PETSC_USE_DYNAMIC_LIBRARIES)
109f1af5d2fSBarry Smith #define KSPRegisterDynamic(a,b,c,d) KSPRegister(a,b,c,0)
1106df38c32SLois Curfman McInnes #else
111f1af5d2fSBarry Smith #define KSPRegisterDynamic(a,b,c,d) KSPRegister(a,b,c,d)
1126df38c32SLois Curfman McInnes #endif
11382bf6240SBarry Smith 
114ca44d042SBarry Smith EXTERN int KSPGetType(KSP,KSPType *);
115ca44d042SBarry Smith EXTERN int KSPSetPreconditionerSide(KSP,PCSide);
116ca44d042SBarry Smith EXTERN int KSPGetPreconditionerSide(KSP,PCSide*);
11787828ca2SBarry Smith EXTERN int KSPGetTolerances(KSP,PetscReal*,PetscReal*,PetscReal*,int*);
11887828ca2SBarry Smith EXTERN int KSPSetTolerances(KSP,PetscReal,PetscReal,PetscReal,int);
1193a7fca6bSBarry Smith EXTERN int KSPSetInitialGuessNonzero(KSP,PetscTruth);
120ca44d042SBarry Smith EXTERN int KSPGetInitialGuessNonzero(KSP,PetscTruth *);
1216ab422ffSBarry Smith EXTERN int KSPSetInitialGuessKnoll(KSP,PetscTruth);
1226ab422ffSBarry Smith EXTERN int KSPGetInitialGuessKnoll(KSP,PetscTruth*);
1233a7fca6bSBarry Smith EXTERN int KSPSetComputeEigenvalues(KSP,PetscTruth);
1243a7fca6bSBarry Smith EXTERN int KSPSetComputeSingularValues(KSP,PetscTruth);
125ca44d042SBarry Smith EXTERN int KSPSetRhs(KSP,Vec);
126ca44d042SBarry Smith EXTERN int KSPGetRhs(KSP,Vec *);
127ca44d042SBarry Smith EXTERN int KSPSetSolution(KSP,Vec);
128ca44d042SBarry Smith EXTERN int KSPGetSolution(KSP,Vec *);
12987828ca2SBarry Smith EXTERN int KSPGetResidualNorm(KSP,PetscReal*);
130ca44d042SBarry Smith EXTERN int KSPGetIterationNumber(KSP,int*);
1312eac72dbSBarry Smith 
132ca44d042SBarry Smith EXTERN int KSPSetPC(KSP,PC);
133ca44d042SBarry Smith EXTERN int KSPGetPC(KSP,PC*);
134aabeff55SBarry Smith 
13587828ca2SBarry Smith EXTERN int KSPSetMonitor(KSP,int (*)(KSP,int,PetscReal,void*),void *,int (*)(void*));
136ca44d042SBarry Smith EXTERN int KSPClearMonitor(KSP);
137ca44d042SBarry Smith EXTERN int KSPGetMonitorContext(KSP,void **);
13887828ca2SBarry Smith EXTERN int KSPGetResidualHistory(KSP,PetscReal **,int *);
13987828ca2SBarry Smith EXTERN int KSPSetResidualHistory(KSP,PetscReal *,int,PetscTruth);
1404b0e389bSBarry Smith 
1412eac72dbSBarry Smith 
142ca44d042SBarry Smith EXTERN int KSPBuildSolution(KSP,Vec,Vec *);
143ca44d042SBarry Smith EXTERN int KSPBuildResidual(KSP,Vec,Vec,Vec *);
1442eac72dbSBarry Smith 
14587828ca2SBarry Smith EXTERN int KSPRichardsonSetScale(KSP,PetscReal);
14687828ca2SBarry Smith EXTERN int KSPChebychevSetEigenvalues(KSP,PetscReal,PetscReal);
14787828ca2SBarry Smith EXTERN int KSPComputeExtremeSingularValues(KSP,PetscReal*,PetscReal*);
14887828ca2SBarry Smith EXTERN int KSPComputeEigenvalues(KSP,int,PetscReal*,PetscReal*,int *);
14987828ca2SBarry Smith EXTERN int KSPComputeEigenvaluesExplicitly(KSP,int,PetscReal*,PetscReal*);
1504b0e389bSBarry Smith 
1519f236934SBarry Smith #define KSPGMRESSetRestart(ksp,r) PetscTryMethod((ksp),KSPGMRESSetRestart_C,(KSP,int),((ksp),(r)))
1529f236934SBarry Smith #define KSPGMRESSetHapTol(ksp,tol) PetscTryMethod((ksp),KSPGMRESSetHapTol_C,(KSP,PetscReal),((ksp),(tol)))
1539f236934SBarry Smith 
154ca44d042SBarry Smith EXTERN int KSPGMRESSetPreAllocateVectors(KSP);
155ca44d042SBarry Smith EXTERN int KSPGMRESSetOrthogonalization(KSP,int (*)(KSP,int));
156ca44d042SBarry Smith EXTERN int KSPGMRESUnmodifiedGramSchmidtOrthogonalization(KSP,int);
157ca44d042SBarry Smith EXTERN int KSPGMRESModifiedGramSchmidtOrthogonalization(KSP,int);
158ca44d042SBarry Smith EXTERN int KSPGMRESIROrthogonalization(KSP,int);
15908480c60SBarry Smith 
16087828ca2SBarry Smith EXTERN int KSPFGMRESModifyPCNoChange(KSP,int,int,PetscReal,void*);
16187828ca2SBarry Smith EXTERN int KSPFGMRESModifyPCSLES(KSP,int,int,PetscReal,void*);
16287828ca2SBarry Smith EXTERN int KSPFGMRESSetModifyPC(KSP,int (*)(KSP,int,int,PetscReal,void*),void*,int(*)(void*));
163c38d4ed2SBarry Smith 
16487828ca2SBarry Smith EXTERN int KSPQCGSetTrustRegionRadius(KSP,PetscReal);
16587828ca2SBarry Smith EXTERN int KSPQCGGetQuadratic(KSP,PetscReal*);
16687828ca2SBarry Smith EXTERN int KSPQCGGetTrialStepNorm(KSP,PetscReal*);
167121fd945SKris Buschelman 
168ca44d042SBarry Smith EXTERN int KSPSetFromOptions(KSP);
169ca44d042SBarry Smith EXTERN int KSPAddOptionsChecker(int (*)(KSP));
1702eac72dbSBarry Smith 
17187828ca2SBarry Smith EXTERN int KSPSingularValueMonitor(KSP,int,PetscReal,void *);
17287828ca2SBarry Smith EXTERN int KSPDefaultMonitor(KSP,int,PetscReal,void *);
17387828ca2SBarry Smith EXTERN int KSPTrueMonitor(KSP,int,PetscReal,void *);
17487828ca2SBarry Smith EXTERN int KSPDefaultSMonitor(KSP,int,PetscReal,void *);
17587828ca2SBarry Smith EXTERN int KSPVecViewMonitor(KSP,int,PetscReal,void *);
17687828ca2SBarry Smith EXTERN int KSPGMRESKrylovMonitor(KSP,int,PetscReal,void *);
17784cb2905SBarry Smith 
178ca44d042SBarry Smith EXTERN int KSPUnwindPreconditioner(KSP,Vec,Vec);
179ca44d042SBarry Smith EXTERN int KSPDefaultBuildSolution(KSP,Vec,Vec*);
180ca44d042SBarry Smith EXTERN int KSPDefaultBuildResidual(KSP,Vec,Vec,Vec *);
181c01c455dSBarry Smith 
182ca44d042SBarry Smith EXTERN int KSPSetOptionsPrefix(KSP,char*);
183ca44d042SBarry Smith EXTERN int KSPAppendOptionsPrefix(KSP,char*);
184ca44d042SBarry Smith EXTERN int KSPGetOptionsPrefix(KSP,char**);
1851eb62cbbSBarry Smith 
186b0a32e0cSBarry Smith EXTERN int KSPView(KSP,PetscViewer);
1871eb62cbbSBarry Smith 
18828ce4d24SBarry Smith /*E
1898a4b9c5cSBarry Smith     KSPNormType - Norm that is passed in the Krylov convergence
1908a4b9c5cSBarry Smith        test routines.
1918a4b9c5cSBarry Smith 
1928a4b9c5cSBarry Smith    Level: advanced
1938a4b9c5cSBarry Smith 
1948a4b9c5cSBarry Smith    Notes: this must match finclude/petscksp.h
1958a4b9c5cSBarry Smith 
1968a4b9c5cSBarry Smith .seealso: SLESSolve(), KSPSolve(), KSPGetConvergedReason(), KSPSetNormType(),
1978a4b9c5cSBarry Smith           KSPSetConvergenceTest()
1988a4b9c5cSBarry Smith E*/
1998a4b9c5cSBarry Smith typedef enum {KSP_NO_NORM               = 0,
2008a4b9c5cSBarry Smith               KSP_PRECONDITIONED_NORM   = 1,
2018a4b9c5cSBarry Smith               KSP_UNPRECONDITIONED_NORM = 2,
2028a4b9c5cSBarry Smith               KSP_NATURAL_NORM          = 3} KSPNormType;
2038a4b9c5cSBarry Smith EXTERN int KSPSetNormType(KSP,KSPNormType);
2048a4b9c5cSBarry Smith 
2058a4b9c5cSBarry Smith /*E
20628ce4d24SBarry Smith     KSPConvergedReason - reason a Krylov method was said to
20728ce4d24SBarry Smith          have converged or diverged
20828ce4d24SBarry Smith 
20928ce4d24SBarry Smith    Level: beginner
21028ce4d24SBarry Smith 
21128ce4d24SBarry Smith    Notes: this must match finclude/petscksp.h
21228ce4d24SBarry Smith 
21328ce4d24SBarry Smith .seealso: SLESSolve(), KSPSolve(), KSPGetConvergedReason()
21428ce4d24SBarry Smith E*/
215d15094e1SBarry Smith typedef enum {/* converged */
216d15094e1SBarry Smith               KSP_CONVERGED_RTOL               =  2,
217d15094e1SBarry Smith               KSP_CONVERGED_ATOL               =  3,
218b335793eSSatish Balay               KSP_CONVERGED_ITS                =  4,
2198d9717b1SSatish Balay               KSP_CONVERGED_QCG_NEG_CURVE      =  5,
220329f5518SBarry Smith               KSP_CONVERGED_QCG_CONSTRAINED    =  6,
221329f5518SBarry Smith               KSP_CONVERGED_STEP_LENGTH        =  7,
222d15094e1SBarry Smith               /* diverged */
223d15094e1SBarry Smith               KSP_DIVERGED_ITS                 = -3,
224d15094e1SBarry Smith               KSP_DIVERGED_DTOL                = -4,
225d15094e1SBarry Smith               KSP_DIVERGED_BREAKDOWN           = -5,
226b4ac9ba4SBarry Smith               KSP_DIVERGED_BREAKDOWN_BICG      = -6,
227b4ac9ba4SBarry Smith               KSP_DIVERGED_NONSYMMETRIC        = -7,
228b4ac9ba4SBarry Smith               KSP_DIVERGED_INDEFINITE_PC       = -8,
229d15094e1SBarry Smith 
230d15094e1SBarry Smith               KSP_CONVERGED_ITERATING          =  0} KSPConvergedReason;
231d15094e1SBarry Smith 
23287828ca2SBarry Smith EXTERN int KSPSetConvergenceTest(KSP,int (*)(KSP,int,PetscReal,KSPConvergedReason*,void*),void *);
233ca44d042SBarry Smith EXTERN int KSPGetConvergenceContext(KSP,void **);
23487828ca2SBarry Smith EXTERN int KSPDefaultConverged(KSP,int,PetscReal,KSPConvergedReason*,void *);
23587828ca2SBarry Smith EXTERN int KSPSkipConverged(KSP,int,PetscReal,KSPConvergedReason*,void *);
236ca44d042SBarry Smith EXTERN int KSPGetConvergedReason(KSP,KSPConvergedReason *);
237abef13c0SSatish Balay 
238ca44d042SBarry Smith EXTERN int KSPComputeExplicitOperator(KSP,Mat *);
239d4fbbf0eSBarry Smith 
24028ce4d24SBarry Smith /*E
24128ce4d24SBarry Smith     KSPCGType - Determines what type of CG to use
24228ce4d24SBarry Smith 
24328ce4d24SBarry Smith    Level: beginner
24428ce4d24SBarry Smith 
24528ce4d24SBarry Smith .seealso: KSPCGSetType()
24628ce4d24SBarry Smith E*/
2476d4a8577SBarry Smith typedef enum {KSP_CG_SYMMETRIC=1,KSP_CG_HERMITIAN=2} KSPCGType;
24828ce4d24SBarry Smith 
249ca44d042SBarry Smith EXTERN int KSPCGSetType(KSP,KSPCGType);
250e559a7feSLois Curfman McInnes 
251ca44d042SBarry Smith EXTERN int PCPreSolve(PC,KSP);
252ca44d042SBarry Smith EXTERN int PCPostSolve(PC,KSP);
2533369ce9aSBarry Smith 
254b0a32e0cSBarry Smith EXTERN int KSPLGMonitorCreate(char*,char*,int,int,int,int,PetscDrawLG*);
25587828ca2SBarry Smith EXTERN int KSPLGMonitor(KSP,int,PetscReal,void*);
256b0a32e0cSBarry Smith EXTERN int KSPLGMonitorDestroy(PetscDrawLG);
257b0a32e0cSBarry Smith EXTERN int KSPLGTrueMonitorCreate(MPI_Comm,char*,char*,int,int,int,int,PetscDrawLG*);
25887828ca2SBarry Smith EXTERN int KSPLGTrueMonitor(KSP,int,PetscReal,void*);
259b0a32e0cSBarry Smith EXTERN int KSPLGTrueMonitorDestroy(PetscDrawLG);
2601eb62cbbSBarry Smith 
2612eac72dbSBarry Smith #endif
2622eac72dbSBarry Smith 
2632eac72dbSBarry Smith 
264