1 /* 2 User interface for the timestepping package. This is package 3 is for use in solving time-dependent PDEs. 4 */ 5 #if !defined(__PETSCTS_H) 6 #define __PETSCTS_H 7 #include "petscsnes.h" 8 PETSC_EXTERN_CXX_BEGIN 9 10 /*S 11 TS - Abstract PETSc object that manages all time-steppers (ODE integrators) 12 13 Level: beginner 14 15 Concepts: ODE solvers 16 17 .seealso: TSCreate(), TSSetType(), TSType, SNES, KSP, PC 18 S*/ 19 typedef struct _p_TS* TS; 20 21 /*E 22 TSType - String with the name of a PETSc TS method or the creation function 23 with an optional dynamic library name, for example 24 http://www.mcs.anl.gov/petsc/lib.a:mytscreate() 25 26 Level: beginner 27 28 .seealso: TSSetType(), TS 29 E*/ 30 #define TS_EULER "euler" 31 #define TS_BEULER "beuler" 32 #define TS_PSEUDO "pseudo" 33 #define TS_CRANK_NICHOLSON "crank-nicholson" 34 #define TS_PVODE "pvode" 35 #define TS_RUNGE_KUTTA "runge-kutta" 36 #define TSType char* 37 38 /*E 39 TSProblemType - Determines the type of problem this TS object is to be used to solve 40 41 Level: beginner 42 43 .seealso: TSCreate() 44 E*/ 45 typedef enum {TS_LINEAR,TS_NONLINEAR} TSProblemType; 46 47 /* Logging support */ 48 extern PetscCookie TS_COOKIE; 49 extern PetscEvent TS_Step, TS_PseudoComputeTimeStep, TS_FunctionEval, TS_JacobianEval; 50 51 EXTERN PetscErrorCode TSInitializePackage(const char[]); 52 53 EXTERN PetscErrorCode TSCreate(MPI_Comm,TS*); 54 EXTERN PetscErrorCode TSDestroy(TS); 55 56 EXTERN PetscErrorCode TSSetProblemType(TS,TSProblemType); 57 EXTERN PetscErrorCode TSGetProblemType(TS,TSProblemType*); 58 EXTERN PetscErrorCode TSSetMonitor(TS,PetscErrorCode(*)(TS,PetscInt,PetscReal,Vec,void*),void *,PetscErrorCode (*)(void*)); 59 EXTERN PetscErrorCode TSClearMonitor(TS); 60 61 EXTERN PetscErrorCode TSSetOptionsPrefix(TS,const char[]); 62 EXTERN PetscErrorCode TSAppendOptionsPrefix(TS,const char[]); 63 EXTERN PetscErrorCode TSGetOptionsPrefix(TS,char *[]); 64 EXTERN PetscErrorCode TSSetFromOptions(TS); 65 EXTERN PetscErrorCode TSSetUp(TS); 66 67 EXTERN PetscErrorCode TSSetSolution(TS,Vec); 68 EXTERN PetscErrorCode TSGetSolution(TS,Vec*); 69 70 EXTERN PetscErrorCode TSSetDuration(TS,PetscInt,PetscReal); 71 EXTERN PetscErrorCode TSGetDuration(TS,PetscInt*,PetscReal*); 72 73 EXTERN PetscErrorCode TSDefaultMonitor(TS,PetscInt,PetscReal,Vec,void*); 74 EXTERN PetscErrorCode TSVecViewMonitor(TS,PetscInt,PetscReal,Vec,void*); 75 EXTERN PetscErrorCode TSStep(TS,PetscInt *,PetscReal*); 76 77 EXTERN PetscErrorCode TSSetInitialTimeStep(TS,PetscReal,PetscReal); 78 EXTERN PetscErrorCode TSGetTimeStep(TS,PetscReal*); 79 EXTERN PetscErrorCode TSGetTime(TS,PetscReal*); 80 EXTERN PetscErrorCode TSGetTimeStepNumber(TS,PetscInt*); 81 EXTERN PetscErrorCode TSSetTimeStep(TS,PetscReal); 82 83 EXTERN PetscErrorCode TSSetRHSFunction(TS,PetscErrorCode (*)(TS,PetscReal,Vec,Vec,void*),void*); 84 EXTERN PetscErrorCode TSSetRHSMatrix(TS,Mat,Mat,PetscErrorCode (*)(TS,PetscReal,Mat*,Mat*,MatStructure*,void*),void*); 85 EXTERN PetscErrorCode TSSetRHSJacobian(TS,Mat,Mat,PetscErrorCode (*)(TS,PetscReal,Vec,Mat*,Mat*,MatStructure*,void*),void*); 86 EXTERN PetscErrorCode TSSetRHSBoundaryConditions(TS,PetscErrorCode (*)(TS,PetscReal,Vec,void*),void*); 87 88 EXTERN PetscErrorCode TSDefaultComputeJacobianColor(TS,PetscReal,Vec,Mat*,Mat*,MatStructure*,void*); 89 EXTERN PetscErrorCode TSDefaultComputeJacobian(TS,PetscReal,Vec,Mat*,Mat*,MatStructure*,void*); 90 91 EXTERN PetscErrorCode TSGetRHSMatrix(TS,Mat*,Mat*,void**); 92 EXTERN PetscErrorCode TSGetRHSJacobian(TS,Mat*,Mat*,void**); 93 94 EXTERN PetscErrorCode TSSetSystemMatrixBC(TS, PetscErrorCode (*)(TS, Mat, Mat, void *)); 95 EXTERN PetscErrorCode TSSetSolutionBC(TS, PetscErrorCode (*)(TS, Vec, void *)); 96 EXTERN PetscErrorCode TSSetPreStep(TS, PetscErrorCode (*)(TS)); 97 EXTERN PetscErrorCode TSSetUpdate(TS, PetscErrorCode (*)(TS, PetscReal, PetscReal *)); 98 EXTERN PetscErrorCode TSSetPostStep(TS, PetscErrorCode (*)(TS)); 99 EXTERN PetscErrorCode TSDefaultRhsBC(TS, Vec, void *); 100 EXTERN PetscErrorCode TSDefaultSystemMatrixBC(TS, Mat, Mat, void *); 101 EXTERN PetscErrorCode TSDefaultSolutionBC(TS, Vec, void *); 102 EXTERN PetscErrorCode TSDefaultPreStep(TS); 103 EXTERN PetscErrorCode TSDefaultUpdate(TS, PetscReal, PetscReal *); 104 EXTERN PetscErrorCode TSDefaultPostStep(TS); 105 EXTERN PetscErrorCode TSSetIdentity(TS, PetscErrorCode (*)(TS, double, Mat *, void *)); 106 107 EXTERN PetscErrorCode TSPseudoSetTimeStep(TS,PetscErrorCode(*)(TS,PetscReal*,void*),void*); 108 EXTERN PetscErrorCode TSPseudoDefaultTimeStep(TS,PetscReal*,void*); 109 EXTERN PetscErrorCode TSPseudoComputeTimeStep(TS,PetscReal *); 110 111 EXTERN PetscErrorCode TSPseudoSetVerifyTimeStep(TS,PetscErrorCode(*)(TS,Vec,void*,PetscReal*,PetscTruth*),void*); 112 EXTERN PetscErrorCode TSPseudoDefaultVerifyTimeStep(TS,Vec,void*,PetscReal*,PetscTruth*); 113 EXTERN PetscErrorCode TSPseudoVerifyTimeStep(TS,Vec,PetscReal*,PetscTruth*); 114 EXTERN PetscErrorCode TSPseudoSetTimeStepIncrement(TS,PetscReal); 115 EXTERN PetscErrorCode TSPseudoIncrementDtFromInitialDt(TS); 116 117 EXTERN PetscErrorCode TSComputeRHSFunction(TS,PetscReal,Vec,Vec); 118 EXTERN PetscErrorCode TSComputeRHSBoundaryConditions(TS,PetscReal,Vec); 119 EXTERN PetscErrorCode TSComputeRHSJacobian(TS,PetscReal,Vec,Mat*,Mat*,MatStructure*); 120 121 /* Dynamic creation and loading functions */ 122 extern PetscFList TSList; 123 extern PetscTruth TSRegisterAllCalled; 124 EXTERN PetscErrorCode TSGetType(TS,TSType*); 125 EXTERN PetscErrorCode TSSetType(TS,const TSType); 126 EXTERN PetscErrorCode TSRegister(const char[], const char[], const char[], PetscErrorCode (*)(TS)); 127 EXTERN PetscErrorCode TSRegisterAll(const char[]); 128 EXTERN PetscErrorCode TSRegisterDestroy(void); 129 130 /*MC 131 TSRegisterDynamic - Adds a creation method to the TS package. 132 133 Synopsis: 134 PetscErrorCode TSRegisterDynamic(char *name, char *path, char *func_name, PetscErrorCode (*create_func)(TS)) 135 136 Not Collective 137 138 Input Parameters: 139 + name - The name of a new user-defined creation routine 140 . path - The path (either absolute or relative) of the library containing this routine 141 . func_name - The name of the creation routine 142 - create_func - The creation routine itself 143 144 Notes: 145 TSRegisterDynamic() may be called multiple times to add several user-defined tses. 146 147 If dynamic libraries are used, then the fourth input argument (create_func) is ignored. 148 149 Sample usage: 150 .vb 151 TSRegisterDynamic("my_ts", "/home/username/my_lib/lib/libO/solaris/libmy.a", "MyTSCreate", MyTSCreate); 152 .ve 153 154 Then, your ts type can be chosen with the procedural interface via 155 .vb 156 TSCreate(MPI_Comm, TS *); 157 TSSetType(vec, "my_ts") 158 .ve 159 or at runtime via the option 160 .vb 161 -ts_type my_ts 162 .ve 163 164 Notes: $PETSC_ARCH occuring in pathname will be replaced with appropriate values. 165 If your function is not being put into a shared library then use TSRegister() instead 166 167 Level: advanced 168 169 .keywords: TS, register 170 .seealso: TSRegisterAll(), TSRegisterDestroy() 171 M*/ 172 #if defined(PETSC_USE_DYNAMIC_LIBRARIES) 173 #define TSRegisterDynamic(a,b,c,d) TSRegister(a,b,c,0) 174 #else 175 #define TSRegisterDynamic(a,b,c,d) TSRegister(a,b,c,d) 176 #endif 177 178 EXTERN PetscErrorCode TSGetSNES(TS,SNES*); 179 EXTERN PetscErrorCode TSGetKSP(TS,KSP*); 180 181 EXTERN PetscErrorCode TSView(TS,PetscViewer); 182 EXTERN PetscErrorCode TSViewFromOptions(TS,const char[]); 183 184 EXTERN PetscErrorCode TSSetApplicationContext(TS,void *); 185 EXTERN PetscErrorCode TSGetApplicationContext(TS,void **); 186 187 EXTERN PetscErrorCode TSLGMonitorCreate(const char[],const char[],int,int,int,int,PetscDrawLG *); 188 EXTERN PetscErrorCode TSLGMonitor(TS,PetscInt,PetscReal,Vec,void *); 189 EXTERN PetscErrorCode TSLGMonitorDestroy(PetscDrawLG); 190 191 /* 192 PETSc interface to PVode 193 */ 194 #define PVODE_UNMODIFIED_GS PVODE_CLASSICAL_GS 195 typedef enum { PVODE_ADAMS,PVODE_BDF } TSPVodeType; 196 typedef enum { PVODE_MODIFIED_GS = 0,PVODE_CLASSICAL_GS = 1 } TSPVodeGramSchmidtType; 197 EXTERN PetscErrorCode TSPVodeSetType(TS,TSPVodeType); 198 EXTERN PetscErrorCode TSPVodeGetPC(TS,PC*); 199 EXTERN PetscErrorCode TSPVodeSetTolerance(TS,PetscReal,PetscReal); 200 EXTERN PetscErrorCode TSPVodeGetIterations(TS,PetscInt *,PetscInt *); 201 EXTERN PetscErrorCode TSPVodeSetGramSchmidtType(TS,TSPVodeGramSchmidtType); 202 EXTERN PetscErrorCode TSPVodeSetGMRESRestart(TS,PetscInt); 203 EXTERN PetscErrorCode TSPVodeSetLinearTolerance(TS,PetscReal); 204 EXTERN PetscErrorCode TSPVodeSetExactFinalTime(TS,PetscTruth); 205 EXTERN PetscErrorCode TSPVodeGetParameters(TS,PetscInt *,long int*[],double*[]); 206 207 EXTERN PetscErrorCode TSRKSetTolerance(TS,PetscReal); 208 209 PETSC_EXTERN_CXX_END 210 #endif 211