1*e34be4c2SBarry Smith /* $Id: ts.h,v 1.17 1997/08/22 15:20:23 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 */ 66d84be18SBarry Smith #if !defined(__TS_PACKAGE) 76d84be18SBarry Smith #define __TS_PACKAGE 86d84be18SBarry Smith #include "snes.h" 9818ad0c1SBarry Smith 10f09e8eb9SSatish Balay typedef struct _p_TS* TS; 116d84be18SBarry Smith #define TS_COOKIE PETSC_COOKIE+18 12818ad0c1SBarry Smith 1383e2fdc7SBarry Smith typedef enum { TS_EULER, TS_BEULER, TS_PSEUDO, TS_PVODE, TS_NEW} TSType; 1419bcc07fSBarry Smith typedef enum { TS_LINEAR, TS_NONLINEAR} TSProblemType; 15818ad0c1SBarry Smith 166d84be18SBarry Smith extern int TSCreate(MPI_Comm,TSProblemType,TS*); 176d84be18SBarry Smith extern int TSSetType(TS,TSType); 18d4bb536fSBarry Smith extern int TSGetProblemType(TS,TSProblemType*); 196d84be18SBarry Smith extern int TSDestroy(TS); 20818ad0c1SBarry Smith 2121c89e3eSBarry Smith extern int TSSetMonitor(TS,int(*)(TS,int,double,Vec,void*),void *); 226d84be18SBarry Smith extern int TSGetType(TS,TSType*,char**); 23818ad0c1SBarry Smith 24d4bb536fSBarry Smith extern int TSSetOptionsPrefix(TS, char *); 25d4bb536fSBarry Smith extern int TSAppendOptionsPrefix(TS, char *); 26d4bb536fSBarry Smith extern int TSGetOptionsPrefix(TS, char **); 276d84be18SBarry Smith extern int TSSetFromOptions(TS); 286d84be18SBarry Smith extern int TSSetUp(TS); 29818ad0c1SBarry Smith 306d84be18SBarry Smith extern int TSSetSolution(TS,Vec); 316d84be18SBarry Smith extern int TSGetSolution(TS,Vec*); 32818ad0c1SBarry Smith 3321c89e3eSBarry Smith extern int TSSetDuration(TS,int,double); 346d84be18SBarry Smith extern int TSPrintHelp(TS); 35818ad0c1SBarry Smith 3621c89e3eSBarry Smith extern int TSDefaultMonitor(TS,int,double,Vec,void*); 3721c89e3eSBarry Smith extern int TSStep(TS,int *,double*); 38818ad0c1SBarry Smith 39c3e30b67SBarry Smith extern int TSSetInitialTimeStep(TS,double,double); 40c3e30b67SBarry Smith extern int TSGetTimeStep(TS,double*); 41d4bb536fSBarry Smith extern int TSGetTime(TS,double*); 4221c89e3eSBarry Smith extern int TSGetTimeStepNumber(TS,int*); 43c3e30b67SBarry Smith extern int TSSetTimeStep(TS,double); 44818ad0c1SBarry Smith 4521c89e3eSBarry Smith extern int TSSetRHSFunction(TS,int (*)(TS,double,Vec,Vec,void*),void*); 46c3e30b67SBarry Smith extern int TSSetRHSMatrix(TS,Mat,Mat,int (*)(TS,double,Mat*,Mat*,MatStructure*,void*),void*); 47c3e30b67SBarry Smith extern int TSSetRHSJacobian(TS,Mat,Mat,int(*)(TS,double,Vec,Mat*,Mat*,MatStructure*,void*),void*); 48818ad0c1SBarry Smith 49*e34be4c2SBarry Smith extern int TSDefaultComputeJacobianWithColoring(TS,double,Vec,Mat*,Mat*,MatStructure*,void*); 50*e34be4c2SBarry Smith 51d4bb536fSBarry Smith extern int TSGetRHSMatrix(TS,Mat*,Mat*,void**); 52d4bb536fSBarry Smith extern int TSGetRHSJacobian(TS,Mat*,Mat*,void**); 53d4bb536fSBarry Smith 54d8345c25SBarry Smith extern int TSPseudoSetTimeStep(TS,int(*)(TS,double*,void*),void*); 55d8345c25SBarry Smith extern int TSPseudoDefaultTimeStep(TS,double*,void* ); 56d8345c25SBarry Smith extern int TSPseudoComputeTimeStep(TS,double *); 57d8345c25SBarry Smith 58d8345c25SBarry Smith extern int TSPseudoSetVerifyTimeStep(TS,int(*)(TS,Vec,void*,double*,int*),void*); 59d8345c25SBarry Smith extern int TSPseudoDefaultVerifyTimeStep(TS,Vec,void*,double*,int*); 60d8345c25SBarry Smith extern int TSPseudoVerifyTimeStep(TS,Vec,double*,int*); 61bc223fc0SBarry Smith extern int TSPseudoSetTimeStepIncrement(TS,double); 620513a670SBarry Smith extern int TSPseudoIncrementDtFromInitialDt(TS); 6321c89e3eSBarry Smith 64c3e30b67SBarry Smith extern int TSComputeRHSFunction(TS,double,Vec,Vec); 65818ad0c1SBarry Smith 6684cb2905SBarry Smith extern int TSRegister(TSType,TSType*,char*,int (*)(TS)); 676d84be18SBarry Smith extern int TSRegisterAll(); 6884cb2905SBarry Smith extern int TSRegisterDestroy(); 6984cb2905SBarry Smith extern int TSRegisterAllCalled; 70818ad0c1SBarry Smith 716d84be18SBarry Smith extern int TSGetSNES(TS,SNES*); 726d84be18SBarry Smith extern int TSGetSLES(TS,SLES*); 73818ad0c1SBarry Smith 74c3e30b67SBarry Smith extern int TSView(TS,Viewer); 7521c89e3eSBarry Smith 7621c89e3eSBarry Smith extern int TSSetApplicationContext(TS,void *); 7721c89e3eSBarry Smith extern int TSGetApplicationContext(TS,void **); 7821c89e3eSBarry Smith 793914022bSBarry Smith extern int TSLGMonitorCreate(char *,char *,int,int,int,int, DrawLG *); 803914022bSBarry Smith extern int TSLGMonitor(TS,int,double,Vec,void *); 813914022bSBarry Smith extern int TSLGMonitorDestroy(DrawLG); 823914022bSBarry Smith 8383e2fdc7SBarry Smith /* 8483e2fdc7SBarry Smith PETSc interface to PVode 8583e2fdc7SBarry Smith */ 8683e2fdc7SBarry Smith typedef enum { PVODE_ADAMS, PVODE_BDF } TSPVodeMethod; 8783e2fdc7SBarry Smith 8883e2fdc7SBarry Smith extern int TSPVodeSetPCType(TS,PCType); 8983e2fdc7SBarry Smith extern int TSPVodeGetPC(TS,PC*); 9083e2fdc7SBarry Smith 9183e2fdc7SBarry Smith extern int TSPVodeKSPSetType(TS,KSPType); 9283e2fdc7SBarry Smith extern int TSPVodeSetMethod(TS,TSPVodeMethod); 9383e2fdc7SBarry Smith extern int TSPVodeSetRHSFunction(TS,double,Vec,Vec,void *); 9483e2fdc7SBarry Smith extern int TSPVodeSetRHSJacobian(TS,double,Vec,Mat*,Mat*,MatStructure*,void *); 9583e2fdc7SBarry Smith extern int TSComputeRHSJacobianForPVODE(TS,double,Vec,Mat*, 9683e2fdc7SBarry Smith Mat*,MatStructure*,void*); 9783e2fdc7SBarry Smith extern int TSComputeRHSFunctionForPVODE(TS,double,Vec,Vec,void*); 9883e2fdc7SBarry Smith extern int TSPVodeGetUserData(TS, void*); 9983e2fdc7SBarry Smith 100818ad0c1SBarry Smith #endif 101818ad0c1SBarry Smith 10221c89e3eSBarry Smith 10321c89e3eSBarry Smith 10421c89e3eSBarry Smith 10521c89e3eSBarry Smith 106