static PetscErrorCode TSSetUp_Theta(TS ts)
{
TS_Theta *th = (TS_Theta*)ts->data;
PetscErrorCode ierr;
SNES snes;
TSAdapt adapt;
DM dm;
PetscFunctionBegin;
ierr = VecDuplicate(ts->vec_sol,&th->X);CHKERRQ(ierr);
ierr = VecDuplicate(ts->vec_sol,&th->Xdot);CHKERRQ(ierr);
ierr = VecDuplicate(ts->vec_sol,&th->X0);CHKERRQ(ierr);
ierr = TSGetSNES(ts,&snes);CHKERRQ(ierr);
ierr = TSGetDM(ts,&dm);CHKERRQ(ierr);
if (dm) {
ierr = DMCoarsenHookAdd(dm,DMCoarsenHook_TSTheta,DMRestrictHook_TSTheta,ts);CHKERRQ(ierr);
ierr = DMSubDomainHookAdd(dm,DMSubDomainHook_TSTheta,DMSubDomainRestrictHook_TSTheta,ts);CHKERRQ(ierr);
}
if (th->Theta == 0.5 && th->endpoint) th->order = 2;
else th->order = 1;
ierr = TSGetAdapt(ts,&adapt);CHKERRQ(ierr);
if (!th->adapt) {
ierr = TSAdaptSetType(adapt,TSADAPTNONE);CHKERRQ(ierr);
}
PetscFunctionReturn(0);
}
/*@C
DMCopyDMTS - copies a DM context to a new DM
Logically Collective
Input Arguments:
+ dmsrc - DM to obtain context from
- dmdest - DM to add context to
Level: developer
Note:
The context is copied by reference. This function does not ensure that a context exists.
.seealso: DMGetDMTS(), TSSetDM()
@*/
PetscErrorCode DMCopyDMTS(DM dmsrc,DM dmdest)
{
PetscErrorCode ierr;
PetscFunctionBegin;
PetscValidHeaderSpecific(dmsrc,DM_CLASSID,1);
PetscValidHeaderSpecific(dmdest,DM_CLASSID,2);
ierr = DMTSDestroy((DMTS*)&dmdest->dmts);CHKERRQ(ierr);
dmdest->dmts = dmsrc->dmts;
ierr = PetscObjectReference(dmdest->dmts);CHKERRQ(ierr);
ierr = DMCoarsenHookAdd(dmdest,DMCoarsenHook_DMTS,DMRestrictHook_DMTS,NULL);CHKERRQ(ierr);
ierr = DMSubDomainHookAdd(dmdest,DMSubDomainHook_DMTS,DMSubDomainRestrictHook_DMTS,NULL);CHKERRQ(ierr);
PetscFunctionReturn(0);
}
/*@C
DMGetDMTS - get read-only private DMTS context from a DM
Not Collective
Input Argument:
. dm - DM to be used with TS
Output Argument:
. tsdm - private DMTS context
Level: developer
Notes:
Use DMGetDMTSWrite() if write access is needed. The DMTSSetXXX API should be used wherever possible.
.seealso: DMGetDMTSWrite()
@*/
PetscErrorCode DMGetDMTS(DM dm,DMTS *tsdm)
{
PetscErrorCode ierr;
PetscFunctionBegin;
PetscValidHeaderSpecific(dm,DM_CLASSID,1);
*tsdm = (DMTS) dm->dmts;
if (!*tsdm) {
ierr = PetscInfo(dm,"Creating new DMTS\n");CHKERRQ(ierr);
ierr = DMTSCreate(PetscObjectComm((PetscObject)dm),tsdm);CHKERRQ(ierr);
dm->dmts = (PetscObject) *tsdm;
ierr = DMCoarsenHookAdd(dm,DMCoarsenHook_DMTS,DMRestrictHook_DMTS,NULL);CHKERRQ(ierr);
ierr = DMSubDomainHookAdd(dm,DMSubDomainHook_DMTS,DMSubDomainRestrictHook_DMTS,NULL);CHKERRQ(ierr);
}
PetscFunctionReturn(0);
}
static PetscErrorCode TSSetUp_RK(TS ts)
{
TS_RK *rk = (TS_RK*)ts->data;
PetscErrorCode ierr;
DM dm;
PetscFunctionBegin;
ierr = TSRKTableauSetUp(ts);CHKERRQ(ierr);
if (!rk->VecCostIntegral0 && ts->vec_costintegral && ts->costintegralfwd) { /* back up cost integral */
ierr = VecDuplicate(ts->vec_costintegral,&rk->VecCostIntegral0);CHKERRQ(ierr);
}
ierr = TSGetDM(ts,&dm);CHKERRQ(ierr);
if (dm) {
ierr = DMCoarsenHookAdd(dm,DMCoarsenHook_TSRK,DMRestrictHook_TSRK,ts);CHKERRQ(ierr);
ierr = DMSubDomainHookAdd(dm,DMSubDomainHook_TSRK,DMSubDomainRestrictHook_TSRK,ts);CHKERRQ(ierr);
}
PetscFunctionReturn(0);
}
int main(int argc,char **argv)
{
TS ts;
Vec x,c;
PetscErrorCode ierr;
DM da;
PetscInitialize(&argc,&argv,(char *)0,help);
ierr = TSCreate(PETSC_COMM_WORLD, &ts);CHKERRQ(ierr);
ierr = TSSetType(ts,TSCN);CHKERRQ(ierr);
ierr = TSSetProblemType(ts,TS_NONLINEAR);CHKERRQ(ierr);
ierr = DMDACreate2d(PETSC_COMM_WORLD, DMDA_BOUNDARY_NONE, DMDA_BOUNDARY_NONE,DMDA_STENCIL_STAR,-4,-4,PETSC_DECIDE,PETSC_DECIDE,1,1,PETSC_NULL,PETSC_NULL,&da);CHKERRQ(ierr);
ierr = DMDASetUniformCoordinates(da, 0.0, 1.0, 0.0, 1.0, 0.0, 1.0);CHKERRQ(ierr);
ierr = DMDASetFieldName(da,0,"Heat");CHKERRQ(ierr);
ierr = DMCreateGlobalVector(da,&x);CHKERRQ(ierr);
ierr = FormInitialGuess(da,PETSC_NULL,x);CHKERRQ(ierr);
ierr = DMDATSSetIFunctionLocal(da,INSERT_VALUES,(PetscErrorCode (*)(DMDALocalInfo*,PetscReal,void*,void*,void*,void*))FormIFunctionLocal,PETSC_NULL);CHKERRQ(ierr);
/* set up the coefficient */
ierr = DMGetNamedGlobalVector(da,"coefficient",&c);CHKERRQ(ierr);
ierr = FormDiffusionCoefficient(da,PETSC_NULL,c);CHKERRQ(ierr);
ierr = DMRestoreNamedGlobalVector(da,"coefficient",&c);CHKERRQ(ierr);
ierr = DMCoarsenHookAdd(da,PETSC_NULL,CoefficientRestrictHook,ts);CHKERRQ(ierr);
ierr = DMSubDomainHookAdd(da,PETSC_NULL,CoefficientSubDomainRestrictHook,ts);CHKERRQ(ierr);
ierr = TSSetDM(ts, da);CHKERRQ(ierr);
ierr = TSSetDuration(ts,10000,1000.0);CHKERRQ(ierr);
ierr = TSSetInitialTimeStep(ts,0.0,0.05);CHKERRQ(ierr);
ierr = TSSetSolution(ts,x);CHKERRQ(ierr);
ierr = TSSetFromOptions(ts);CHKERRQ(ierr);
ierr = TSSolve(ts,x);CHKERRQ(ierr);
ierr = VecDestroy(&x);CHKERRQ(ierr);
ierr = TSDestroy(&ts);CHKERRQ(ierr);
ierr = DMDestroy(&da);CHKERRQ(ierr);
ierr = PetscFinalize();
PetscFunctionReturn(0);
}
static PetscErrorCode TSSetUp_Theta(TS ts)
{
TS_Theta *th = (TS_Theta*)ts->data;
PetscErrorCode ierr;
PetscFunctionBegin;
if (!th->VecCostIntegral0 && ts->vec_costintegral && ts->costintegralfwd) { /* back up cost integral */
ierr = VecDuplicate(ts->vec_costintegral,&th->VecCostIntegral0);CHKERRQ(ierr);
}
if (!th->X) {
ierr = VecDuplicate(ts->vec_sol,&th->X);CHKERRQ(ierr);
}
if (!th->Xdot) {
ierr = VecDuplicate(ts->vec_sol,&th->Xdot);CHKERRQ(ierr);
}
if (!th->X0) {
ierr = VecDuplicate(ts->vec_sol,&th->X0);CHKERRQ(ierr);
}
if (th->endpoint) {
ierr = VecDuplicate(ts->vec_sol,&th->affine);CHKERRQ(ierr);
}
th->order = (th->Theta == 0.5) ? 2 : 1;
ierr = TSGetDM(ts,&ts->dm);CHKERRQ(ierr);
ierr = DMCoarsenHookAdd(ts->dm,DMCoarsenHook_TSTheta,DMRestrictHook_TSTheta,ts);CHKERRQ(ierr);
ierr = DMSubDomainHookAdd(ts->dm,DMSubDomainHook_TSTheta,DMSubDomainRestrictHook_TSTheta,ts);CHKERRQ(ierr);
ierr = TSGetAdapt(ts,&ts->adapt);CHKERRQ(ierr);
ierr = TSAdaptCandidatesClear(ts->adapt);CHKERRQ(ierr);
if (!th->adapt) {
ierr = TSAdaptSetType(ts->adapt,TSADAPTNONE);CHKERRQ(ierr);
} else {
ierr = VecDuplicate(ts->vec_sol,&th->vec_sol_prev);CHKERRQ(ierr);
ierr = VecDuplicate(ts->vec_sol,&th->vec_lte_work);CHKERRQ(ierr);
if (ts->exact_final_time == TS_EXACTFINALTIME_UNSPECIFIED)
ts->exact_final_time = TS_EXACTFINALTIME_MATCHSTEP;
}
ierr = TSGetSNES(ts,&ts->snes);CHKERRQ(ierr);
PetscFunctionReturn(0);
}
static PetscErrorCode TSSetUp_RK(TS ts)
{
TS_RK *rk = (TS_RK*)ts->data;
RKTableau tab;
PetscInt s;
PetscErrorCode ierr;
DM dm;
PetscFunctionBegin;
if (!rk->tableau) {
ierr = TSRKSetType(ts,TSRKDefault);CHKERRQ(ierr);
}
tab = rk->tableau;
s = tab->s;
ierr = VecDuplicateVecs(ts->vec_sol,s,&rk->Y);CHKERRQ(ierr);
ierr = VecDuplicateVecs(ts->vec_sol,s,&rk->YdotRHS);CHKERRQ(ierr);
ierr = PetscMalloc1(s,&rk->work);CHKERRQ(ierr);
ierr = TSGetDM(ts,&dm);CHKERRQ(ierr);
if (dm) {
ierr = DMCoarsenHookAdd(dm,DMCoarsenHook_TSRK,DMRestrictHook_TSRK,ts);CHKERRQ(ierr);
ierr = DMSubDomainHookAdd(dm,DMSubDomainHook_TSRK,DMSubDomainRestrictHook_TSRK,ts);CHKERRQ(ierr);
}
PetscFunctionReturn(0);
}
