1*f1af5d2fSBarry Smith /* $Id: ts.h,v 1.36 1999/05/12 03:35:01 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 2821c89e3eSBarry Smith extern int TSSetMonitor(TS,int(*)(TS,int,double,Vec,void*),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); 77818ad0c1SBarry Smith 78488ecbafSBarry Smith extern FList TSList; 7982bf6240SBarry Smith extern int TSRegisterAll(char*); 80cf256101SBarry Smith extern int TSRegisterDestroy(void); 8184cb2905SBarry Smith extern int TSRegisterAllCalled; 82818ad0c1SBarry Smith 83*f1af5d2fSBarry Smith extern int TSRegister(char*,char*,char*,int(*)(TS)); 84aa482453SBarry Smith #if defined(PETSC_USE_DYNAMIC_LIBRARIES) 85*f1af5d2fSBarry Smith #define TSRegisterDynamic(a,b,c,d) TSRegister(a,b,c,0) 866df38c32SLois Curfman McInnes #else 87*f1af5d2fSBarry Smith #define TSRegisterDynamic(a,b,c,d) TSRegister(a,b,c,d) 886df38c32SLois Curfman McInnes #endif 896df38c32SLois Curfman McInnes 906d84be18SBarry Smith extern int TSGetSNES(TS,SNES*); 916d84be18SBarry Smith extern int TSGetSLES(TS,SLES*); 92818ad0c1SBarry Smith 93c3e30b67SBarry Smith extern int TSView(TS,Viewer); 9421c89e3eSBarry Smith 9521c89e3eSBarry Smith extern int TSSetApplicationContext(TS,void *); 9621c89e3eSBarry Smith extern int TSGetApplicationContext(TS,void **); 9721c89e3eSBarry Smith 983914022bSBarry Smith extern int TSLGMonitorCreate(char *,char *,int,int,int,int, DrawLG *); 993914022bSBarry Smith extern int TSLGMonitor(TS,int,double,Vec,void *); 1003914022bSBarry Smith extern int TSLGMonitorDestroy(DrawLG); 1013914022bSBarry Smith 10283e2fdc7SBarry Smith /* 10383e2fdc7SBarry Smith PETSc interface to PVode 10483e2fdc7SBarry Smith */ 105330e16abSSatish Balay #define PVODE_UNMODIFIED_GS PVODE_CLASSICAL_GS 106453b9e1bSBarry Smith typedef enum { PVODE_ADAMS, PVODE_BDF } TSPVodeType; 107330e16abSSatish Balay typedef enum { PVODE_MODIFIED_GS = 0, PVODE_CLASSICAL_GS = 1 } TSPVodeGramSchmidtType; 108453b9e1bSBarry Smith extern int TSPVodeSetType(TS,TSPVodeType); 10983e2fdc7SBarry Smith extern int TSPVodeGetPC(TS,PC*); 11090e6f1a8SBarry Smith extern int TSPVodeSetTolerance(TS,double,double); 111330e16abSSatish Balay extern int TSPVodeGetIterations(TS,int *,int *); 11276be9ce4SBarry Smith extern int TSPVodeSetGramSchmidtType(TS,TSPVodeGramSchmidtType); 11376be9ce4SBarry Smith extern int TSPVodeSetGMRESRestart(TS,int); 114df8cb225SBarry Smith extern int TSPVodeSetLinearTolerance(TS,double); 115e1311b90SBarry Smith extern int TSPVodeSetExactFinalTime(TS, PetscTruth); 11683e2fdc7SBarry Smith 117818ad0c1SBarry Smith #endif 118818ad0c1SBarry Smith 11921c89e3eSBarry Smith 12021c89e3eSBarry Smith 12121c89e3eSBarry Smith 12221c89e3eSBarry Smith 123