1*a7a1495cSBarry Smith /* $Id: ts.h,v 1.39 2000/02/02 21:21:25 bsmith Exp bsmith $ */ 2818ad0c1SBarry Smith /* 3f64a0f93SLois Curfman McInnes User interface for the timestepping package. This is package 4f64a0f93SLois Curfman McInnes is for use in solving time-dependent PDEs. 5818ad0c1SBarry Smith */ 688d459dfSBarry Smith #if !defined(__TS_H) 788d459dfSBarry Smith #define __TS_H 86d84be18SBarry Smith #include "snes.h" 9818ad0c1SBarry Smith 10f09e8eb9SSatish Balay typedef struct _p_TS* TS; 116d84be18SBarry Smith #define TS_COOKIE PETSC_COOKIE+18 12818ad0c1SBarry Smith 1382bf6240SBarry Smith #define TS_EULER "euler" 1482bf6240SBarry Smith #define TS_BEULER "beuler" 1582bf6240SBarry Smith #define TS_PSEUDO "pseudo" 1667d5d79cSLois Curfman McInnes #define TS_CRANK_NICHOLSON "crank-nicholson" 1782bf6240SBarry Smith #define TS_PVODE "pvode" 1882bf6240SBarry Smith 1982bf6240SBarry Smith typedef char *TSType; 2082bf6240SBarry Smith 2119bcc07fSBarry Smith typedef enum {TS_LINEAR,TS_NONLINEAR} TSProblemType; 22818ad0c1SBarry Smith 236d84be18SBarry Smith extern int TSCreate(MPI_Comm,TSProblemType,TS*); 246d84be18SBarry Smith extern int TSSetType(TS,TSType); 25d4bb536fSBarry Smith extern int TSGetProblemType(TS,TSProblemType*); 266d84be18SBarry Smith extern int TSDestroy(TS); 27818ad0c1SBarry Smith 28329f5518SBarry Smith extern int TSSetMonitor(TS,int(*)(TS,int,double,Vec,void*),void *,int (*)(void*)); 295cd90555SBarry Smith extern int TSClearMonitor(TS); 3082bf6240SBarry Smith extern int TSGetType(TS,TSType*); 31818ad0c1SBarry Smith 32d4bb536fSBarry Smith extern int TSSetOptionsPrefix(TS,char *); 33d4bb536fSBarry Smith extern int TSAppendOptionsPrefix(TS,char *); 34d4bb536fSBarry Smith extern int TSGetOptionsPrefix(TS,char **); 356d84be18SBarry Smith extern int TSSetFromOptions(TS); 3615091d37SBarry Smith extern int TSSetTypeFromOptions(TS); 376d84be18SBarry Smith extern int TSSetUp(TS); 38818ad0c1SBarry Smith 396d84be18SBarry Smith extern int TSSetSolution(TS,Vec); 406d84be18SBarry Smith extern int TSGetSolution(TS,Vec*); 41818ad0c1SBarry Smith 4221c89e3eSBarry Smith extern int TSSetDuration(TS,int,double); 436d84be18SBarry Smith extern int TSPrintHelp(TS); 44818ad0c1SBarry Smith 4521c89e3eSBarry Smith extern int TSDefaultMonitor(TS,int,double,Vec,void*); 4621c89e3eSBarry Smith extern int TSStep(TS,int *,double*); 47818ad0c1SBarry Smith 48c3e30b67SBarry Smith extern int TSSetInitialTimeStep(TS,double,double); 49c3e30b67SBarry Smith extern int TSGetTimeStep(TS,double*); 50d4bb536fSBarry Smith extern int TSGetTime(TS,double*); 5121c89e3eSBarry Smith extern int TSGetTimeStepNumber(TS,int*); 52c3e30b67SBarry Smith extern int TSSetTimeStep(TS,double); 53818ad0c1SBarry Smith 5421c89e3eSBarry Smith extern int TSSetRHSFunction(TS,int (*)(TS,double,Vec,Vec,void*),void*); 55c3e30b67SBarry Smith extern int TSSetRHSMatrix(TS,Mat,Mat,int (*)(TS,double,Mat*,Mat*,MatStructure*,void*),void*); 56c3e30b67SBarry Smith extern int TSSetRHSJacobian(TS,Mat,Mat,int(*)(TS,double,Vec,Mat*,Mat*,MatStructure*,void*),void*); 5767d5d79cSLois Curfman McInnes extern int TSSetRHSBoundaryConditions(TS,int (*)(TS,double,Vec,void*),void*); 58818ad0c1SBarry Smith 592d0c0e3bSBarry Smith extern int TSDefaultComputeJacobianColor(TS,double,Vec,Mat*,Mat*,MatStructure*,void*); 602d0c0e3bSBarry Smith extern int TSDefaultComputeJacobian(TS,double,Vec,Mat*,Mat*,MatStructure*,void*); 61e34be4c2SBarry Smith 62d4bb536fSBarry Smith extern int TSGetRHSMatrix(TS,Mat*,Mat*,void**); 63d4bb536fSBarry Smith extern int TSGetRHSJacobian(TS,Mat*,Mat*,void**); 64d4bb536fSBarry Smith 65d8345c25SBarry Smith extern int TSPseudoSetTimeStep(TS,int(*)(TS,double*,void*),void*); 66d8345c25SBarry Smith extern int TSPseudoDefaultTimeStep(TS,double*,void*); 67d8345c25SBarry Smith extern int TSPseudoComputeTimeStep(TS,double *); 68d8345c25SBarry Smith 69d8345c25SBarry Smith extern int TSPseudoSetVerifyTimeStep(TS,int(*)(TS,Vec,void*,double*,int*),void*); 70d8345c25SBarry Smith extern int TSPseudoDefaultVerifyTimeStep(TS,Vec,void*,double*,int*); 71d8345c25SBarry Smith extern int TSPseudoVerifyTimeStep(TS,Vec,double*,int*); 72bc223fc0SBarry Smith extern int TSPseudoSetTimeStepIncrement(TS,double); 730513a670SBarry Smith extern int TSPseudoIncrementDtFromInitialDt(TS); 7421c89e3eSBarry Smith 75c3e30b67SBarry Smith extern int TSComputeRHSFunction(TS,double,Vec,Vec); 76154e5d31SSatish Balay extern int TSComputeRHSBoundaryConditions(TS,double,Vec); 77*a7a1495cSBarry Smith extern int TSComputeRHSJacobian(TS,double,Vec,Mat*,Mat*,MatStructure*); 78818ad0c1SBarry Smith 79488ecbafSBarry Smith extern FList TSList; 8082bf6240SBarry Smith extern int TSRegisterAll(char*); 81cf256101SBarry Smith extern int TSRegisterDestroy(void); 822bad1931SBarry Smith extern PetscTruth TSRegisterAllCalled; 83818ad0c1SBarry Smith 84f1af5d2fSBarry Smith extern int TSRegister(char*,char*,char*,int(*)(TS)); 85aa482453SBarry Smith #if defined(PETSC_USE_DYNAMIC_LIBRARIES) 86f1af5d2fSBarry Smith #define TSRegisterDynamic(a,b,c,d) TSRegister(a,b,c,0) 876df38c32SLois Curfman McInnes #else 88f1af5d2fSBarry Smith #define TSRegisterDynamic(a,b,c,d) TSRegister(a,b,c,d) 896df38c32SLois Curfman McInnes #endif 906df38c32SLois Curfman McInnes 916d84be18SBarry Smith extern int TSGetSNES(TS,SNES*); 926d84be18SBarry Smith extern int TSGetSLES(TS,SLES*); 93818ad0c1SBarry Smith 94c3e30b67SBarry Smith extern int TSView(TS,Viewer); 9521c89e3eSBarry Smith 9621c89e3eSBarry Smith extern int TSSetApplicationContext(TS,void *); 9721c89e3eSBarry Smith extern int TSGetApplicationContext(TS,void **); 9821c89e3eSBarry Smith 993914022bSBarry Smith extern int TSLGMonitorCreate(char *,char *,int,int,int,int,DrawLG *); 1003914022bSBarry Smith extern int TSLGMonitor(TS,int,double,Vec,void *); 1013914022bSBarry Smith extern int TSLGMonitorDestroy(DrawLG); 1023914022bSBarry Smith 10383e2fdc7SBarry Smith /* 10483e2fdc7SBarry Smith PETSc interface to PVode 10583e2fdc7SBarry Smith */ 106330e16abSSatish Balay #define PVODE_UNMODIFIED_GS PVODE_CLASSICAL_GS 107453b9e1bSBarry Smith typedef enum { PVODE_ADAMS,PVODE_BDF } TSPVodeType; 108330e16abSSatish Balay typedef enum { PVODE_MODIFIED_GS = 0,PVODE_CLASSICAL_GS = 1 } TSPVodeGramSchmidtType; 109453b9e1bSBarry Smith extern int TSPVodeSetType(TS,TSPVodeType); 11083e2fdc7SBarry Smith extern int TSPVodeGetPC(TS,PC*); 11190e6f1a8SBarry Smith extern int TSPVodeSetTolerance(TS,double,double); 112330e16abSSatish Balay extern int TSPVodeGetIterations(TS,int *,int *); 11376be9ce4SBarry Smith extern int TSPVodeSetGramSchmidtType(TS,TSPVodeGramSchmidtType); 11476be9ce4SBarry Smith extern int TSPVodeSetGMRESRestart(TS,int); 115df8cb225SBarry Smith extern int TSPVodeSetLinearTolerance(TS,double); 116e1311b90SBarry Smith extern int TSPVodeSetExactFinalTime(TS,PetscTruth); 11783e2fdc7SBarry Smith 118818ad0c1SBarry Smith #endif 119818ad0c1SBarry Smith 12021c89e3eSBarry Smith 12121c89e3eSBarry Smith 12221c89e3eSBarry Smith 12321c89e3eSBarry Smith 124