1 #define PETSCTS_DLL 2 3 /* 4 Code for Timestepping with implicit backwards Euler. 5 */ 6 #include "include/private/tsimpl.h" /*I "petscts.h" I*/ 7 8 typedef struct { 9 Vec update; /* work vector where new solution is formed */ 10 Vec func; /* work vector where F(t[i],u[i]) is stored */ 11 Vec rhs; /* work vector for RHS; vec_sol/dt */ 12 13 /* information used for Pseudo-timestepping */ 14 15 PetscErrorCode (*dt)(TS,PetscReal*,void*); /* compute next timestep, and related context */ 16 void *dtctx; 17 PetscErrorCode (*verify)(TS,Vec,void*,PetscReal*,PetscTruth*); /* verify previous timestep and related context */ 18 void *verifyctx; 19 20 PetscReal initial_fnorm,fnorm; /* original and current norm of F(u) */ 21 PetscReal fnorm_previous; 22 23 PetscReal dt_increment; /* scaling that dt is incremented each time-step */ 24 PetscTruth increment_dt_from_initial_dt; 25 } TS_Pseudo; 26 27 /* ------------------------------------------------------------------------------*/ 28 29 #undef __FUNCT__ 30 #define __FUNCT__ "TSPseudoComputeTimeStep" 31 /*@ 32 TSPseudoComputeTimeStep - Computes the next timestep for a currently running 33 pseudo-timestepping process. 34 35 Collective on TS 36 37 Input Parameter: 38 . ts - timestep context 39 40 Output Parameter: 41 . dt - newly computed timestep 42 43 Level: advanced 44 45 Notes: 46 The routine to be called here to compute the timestep should be 47 set by calling TSPseudoSetTimeStep(). 48 49 .keywords: timestep, pseudo, compute 50 51 .seealso: TSPseudoDefaultTimeStep(), TSPseudoSetTimeStep() 52 @*/ 53 PetscErrorCode PETSCTS_DLLEXPORT TSPseudoComputeTimeStep(TS ts,PetscReal *dt) 54 { 55 TS_Pseudo *pseudo = (TS_Pseudo*)ts->data; 56 PetscErrorCode ierr; 57 58 PetscFunctionBegin; 59 ierr = PetscLogEventBegin(TS_PseudoComputeTimeStep,ts,0,0,0);CHKERRQ(ierr); 60 ierr = (*pseudo->dt)(ts,dt,pseudo->dtctx);CHKERRQ(ierr); 61 ierr = PetscLogEventEnd(TS_PseudoComputeTimeStep,ts,0,0,0);CHKERRQ(ierr); 62 PetscFunctionReturn(0); 63 } 64 65 66 /* ------------------------------------------------------------------------------*/ 67 #undef __FUNCT__ 68 #define __FUNCT__ "TSPseudoDefaultVerifyTimeStep" 69 /*@C 70 TSPseudoDefaultVerifyTimeStep - Default code to verify the quality of the last timestep. 71 72 Collective on TS 73 74 Input Parameters: 75 + ts - the timestep context 76 . dtctx - unused timestep context 77 - update - latest solution vector 78 79 Output Parameters: 80 + newdt - the timestep to use for the next step 81 - flag - flag indicating whether the last time step was acceptable 82 83 Level: advanced 84 85 Note: 86 This routine always returns a flag of 1, indicating an acceptable 87 timestep. 88 89 .keywords: timestep, pseudo, default, verify 90 91 .seealso: TSPseudoSetVerifyTimeStep(), TSPseudoVerifyTimeStep() 92 @*/ 93 PetscErrorCode PETSCTS_DLLEXPORT TSPseudoDefaultVerifyTimeStep(TS ts,Vec update,void *dtctx,PetscReal *newdt,PetscTruth *flag) 94 { 95 PetscFunctionBegin; 96 *flag = PETSC_TRUE; 97 PetscFunctionReturn(0); 98 } 99 100 101 #undef __FUNCT__ 102 #define __FUNCT__ "TSPseudoVerifyTimeStep" 103 /*@ 104 TSPseudoVerifyTimeStep - Verifies whether the last timestep was acceptable. 105 106 Collective on TS 107 108 Input Parameters: 109 + ts - timestep context 110 - update - latest solution vector 111 112 Output Parameters: 113 + dt - newly computed timestep (if it had to shrink) 114 - flag - indicates if current timestep was ok 115 116 Level: advanced 117 118 Notes: 119 The routine to be called here to compute the timestep should be 120 set by calling TSPseudoSetVerifyTimeStep(). 121 122 .keywords: timestep, pseudo, verify 123 124 .seealso: TSPseudoSetVerifyTimeStep(), TSPseudoDefaultVerifyTimeStep() 125 @*/ 126 PetscErrorCode PETSCTS_DLLEXPORT TSPseudoVerifyTimeStep(TS ts,Vec update,PetscReal *dt,PetscTruth *flag) 127 { 128 TS_Pseudo *pseudo = (TS_Pseudo*)ts->data; 129 PetscErrorCode ierr; 130 131 PetscFunctionBegin; 132 if (!pseudo->verify) {*flag = PETSC_TRUE; PetscFunctionReturn(0);} 133 134 ierr = (*pseudo->verify)(ts,update,pseudo->verifyctx,dt,flag);CHKERRQ(ierr); 135 136 PetscFunctionReturn(0); 137 } 138 139 /* --------------------------------------------------------------------------------*/ 140 141 #undef __FUNCT__ 142 #define __FUNCT__ "TSStep_Pseudo" 143 static PetscErrorCode TSStep_Pseudo(TS ts,PetscInt *steps,PetscReal *ptime) 144 { 145 Vec sol = ts->vec_sol; 146 PetscErrorCode ierr; 147 PetscInt i,max_steps = ts->max_steps,its,lits; 148 PetscTruth ok; 149 TS_Pseudo *pseudo = (TS_Pseudo*)ts->data; 150 PetscReal current_time_step; 151 152 PetscFunctionBegin; 153 *steps = -ts->steps; 154 155 ierr = VecCopy(sol,pseudo->update);CHKERRQ(ierr); 156 for (i=0; i<max_steps && ts->ptime < ts->max_time; i++) { 157 ierr = TSPseudoComputeTimeStep(ts,&ts->time_step);CHKERRQ(ierr); 158 ierr = TSMonitor(ts,ts->steps,ts->ptime,sol);CHKERRQ(ierr); 159 current_time_step = ts->time_step; 160 while (PETSC_TRUE) { 161 ts->ptime += current_time_step; 162 ierr = SNESSolve(ts->snes,PETSC_NULL,pseudo->update);CHKERRQ(ierr); 163 ierr = SNESGetNumberLinearIterations(ts->snes,&lits);CHKERRQ(ierr); 164 ierr = SNESGetIterationNumber(ts->snes,&its);CHKERRQ(ierr); 165 ts->nonlinear_its += its; ts->linear_its += lits; 166 ierr = TSPseudoVerifyTimeStep(ts,pseudo->update,&ts->time_step,&ok);CHKERRQ(ierr); 167 if (ok) break; 168 ts->ptime -= current_time_step; 169 current_time_step = ts->time_step; 170 } 171 ierr = VecCopy(pseudo->update,sol);CHKERRQ(ierr); 172 ts->steps++; 173 } 174 ierr = TSComputeRHSFunction(ts,ts->ptime,ts->vec_sol,pseudo->func);CHKERRQ(ierr); 175 ierr = VecNorm(pseudo->func,NORM_2,&pseudo->fnorm);CHKERRQ(ierr); 176 ierr = TSMonitor(ts,ts->steps,ts->ptime,sol);CHKERRQ(ierr); 177 178 *steps += ts->steps; 179 *ptime = ts->ptime; 180 PetscFunctionReturn(0); 181 } 182 183 /*------------------------------------------------------------*/ 184 #undef __FUNCT__ 185 #define __FUNCT__ "TSDestroy_Pseudo" 186 static PetscErrorCode TSDestroy_Pseudo(TS ts) 187 { 188 TS_Pseudo *pseudo = (TS_Pseudo*)ts->data; 189 PetscErrorCode ierr; 190 191 PetscFunctionBegin; 192 if (pseudo->update) {ierr = VecDestroy(pseudo->update);CHKERRQ(ierr);} 193 if (pseudo->func) {ierr = VecDestroy(pseudo->func);CHKERRQ(ierr);} 194 if (pseudo->rhs) {ierr = VecDestroy(pseudo->rhs);CHKERRQ(ierr);} 195 ierr = PetscFree(pseudo);CHKERRQ(ierr); 196 PetscFunctionReturn(0); 197 } 198 199 200 /*------------------------------------------------------------*/ 201 202 /* 203 This defines the nonlinear equation that is to be solved with SNES 204 205 (U^{n+1} - U^{n})/dt - F(U^{n+1}) 206 */ 207 #undef __FUNCT__ 208 #define __FUNCT__ "TSPseudoFunction" 209 PetscErrorCode TSPseudoFunction(SNES snes,Vec x,Vec y,void *ctx) 210 { 211 TS ts = (TS) ctx; 212 PetscScalar mdt = 1.0/ts->time_step,*unp1,*un,*Funp1; 213 PetscErrorCode ierr; 214 PetscInt i,n; 215 216 PetscFunctionBegin; 217 /* apply user provided function */ 218 ierr = TSComputeRHSFunction(ts,ts->ptime,x,y);CHKERRQ(ierr); 219 /* compute (u^{n+1) - u^{n})/dt - F(u^{n+1}) */ 220 ierr = VecGetArray(ts->vec_sol,&un);CHKERRQ(ierr); 221 ierr = VecGetArray(x,&unp1);CHKERRQ(ierr); 222 ierr = VecGetArray(y,&Funp1);CHKERRQ(ierr); 223 ierr = VecGetLocalSize(x,&n);CHKERRQ(ierr); 224 for (i=0; i<n; i++) { 225 Funp1[i] = mdt*(unp1[i] - un[i]) - Funp1[i]; 226 } 227 ierr = VecRestoreArray(ts->vec_sol,&un);CHKERRQ(ierr); 228 ierr = VecRestoreArray(x,&unp1);CHKERRQ(ierr); 229 ierr = VecRestoreArray(y,&Funp1);CHKERRQ(ierr); 230 PetscFunctionReturn(0); 231 } 232 233 /* 234 This constructs the Jacobian needed for SNES 235 236 J = I/dt - J_{F} where J_{F} is the given Jacobian of F. 237 */ 238 #undef __FUNCT__ 239 #define __FUNCT__ "TSPseudoJacobian" 240 PetscErrorCode TSPseudoJacobian(SNES snes,Vec x,Mat *AA,Mat *BB,MatStructure *str,void *ctx) 241 { 242 TS ts = (TS) ctx; 243 PetscErrorCode ierr; 244 245 PetscFunctionBegin; 246 /* construct users Jacobian */ 247 ierr = TSComputeRHSJacobian(ts,ts->ptime,x,AA,BB,str);CHKERRQ(ierr); 248 249 /* shift and scale Jacobian */ 250 ierr = TSScaleShiftMatrices(ts,*AA,*BB,*str);CHKERRQ(ierr); 251 PetscFunctionReturn(0); 252 } 253 254 255 #undef __FUNCT__ 256 #define __FUNCT__ "TSSetUp_Pseudo" 257 static PetscErrorCode TSSetUp_Pseudo(TS ts) 258 { 259 TS_Pseudo *pseudo = (TS_Pseudo*)ts->data; 260 PetscErrorCode ierr; 261 262 PetscFunctionBegin; 263 /* ierr = SNESSetFromOptions(ts->snes);CHKERRQ(ierr); */ 264 ierr = VecDuplicate(ts->vec_sol,&pseudo->update);CHKERRQ(ierr); 265 ierr = VecDuplicate(ts->vec_sol,&pseudo->func);CHKERRQ(ierr); 266 ierr = SNESSetFunction(ts->snes,pseudo->func,TSPseudoFunction,ts);CHKERRQ(ierr); 267 ierr = SNESSetJacobian(ts->snes,ts->Arhs,ts->B,TSPseudoJacobian,ts);CHKERRQ(ierr); 268 PetscFunctionReturn(0); 269 } 270 /*------------------------------------------------------------*/ 271 272 #undef __FUNCT__ 273 #define __FUNCT__ "TSPseudoMonitorDefault" 274 PetscErrorCode TSPseudoMonitorDefault(TS ts,PetscInt step,PetscReal ptime,Vec v,void *ctx) 275 { 276 TS_Pseudo *pseudo = (TS_Pseudo*)ts->data; 277 PetscErrorCode ierr; 278 PetscViewerASCIIMonitor viewer = (PetscViewerASCIIMonitor)ctx; 279 280 PetscFunctionBegin; 281 if (!ctx) { 282 ierr = PetscViewerASCIIMonitorCreate(ts->comm,"stdout",0,&viewer);CHKERRQ(ierr); 283 } 284 ierr = PetscViewerASCIIMonitorPrintf(viewer,"TS %D dt %G time %G fnorm %G\n",step,ts->time_step,ptime,pseudo->fnorm);CHKERRQ(ierr); 285 if (!ctx) { 286 ierr = PetscViewerASCIIMonitorDestroy(viewer);CHKERRQ(ierr); 287 } 288 PetscFunctionReturn(0); 289 } 290 291 #undef __FUNCT__ 292 #define __FUNCT__ "TSSetFromOptions_Pseudo" 293 static PetscErrorCode TSSetFromOptions_Pseudo(TS ts) 294 { 295 TS_Pseudo *pseudo = (TS_Pseudo*)ts->data; 296 PetscErrorCode ierr; 297 PetscTruth flg; 298 PetscViewerASCIIMonitor viewer; 299 300 PetscFunctionBegin; 301 ierr = PetscOptionsHead("Pseudo-timestepping options");CHKERRQ(ierr); 302 ierr = PetscOptionsName("-ts_monitor","Monitor convergence","TSPseudoMonitorDefault",&flg);CHKERRQ(ierr); 303 if (flg) { 304 ierr = PetscViewerASCIIMonitorCreate(ts->comm,"stdout",0,&viewer);CHKERRQ(ierr); 305 ierr = TSMonitorSet(ts,TSPseudoMonitorDefault,viewer,(PetscErrorCode (*)(void*))PetscViewerASCIIMonitorDestroy);CHKERRQ(ierr); 306 } 307 ierr = PetscOptionsName("-ts_pseudo_increment_dt_from_initial_dt","Increase dt as a ratio from original dt","TSPseudoIncrementDtFromInitialDt",&flg);CHKERRQ(ierr); 308 if (flg) { 309 ierr = TSPseudoIncrementDtFromInitialDt(ts);CHKERRQ(ierr); 310 } 311 ierr = PetscOptionsReal("-ts_pseudo_increment","Ratio to increase dt","TSPseudoSetTimeStepIncrement",pseudo->dt_increment,&pseudo->dt_increment,0);CHKERRQ(ierr); 312 ierr = PetscOptionsTail();CHKERRQ(ierr); 313 PetscFunctionReturn(0); 314 } 315 316 #undef __FUNCT__ 317 #define __FUNCT__ "TSView_Pseudo" 318 static PetscErrorCode TSView_Pseudo(TS ts,PetscViewer viewer) 319 { 320 PetscFunctionBegin; 321 PetscFunctionReturn(0); 322 } 323 324 /* ----------------------------------------------------------------------------- */ 325 #undef __FUNCT__ 326 #define __FUNCT__ "TSPseudoSetVerifyTimeStep" 327 /*@C 328 TSPseudoSetVerifyTimeStep - Sets a user-defined routine to verify the quality of the 329 last timestep. 330 331 Collective on TS 332 333 Input Parameters: 334 + ts - timestep context 335 . dt - user-defined function to verify timestep 336 - ctx - [optional] user-defined context for private data 337 for the timestep verification routine (may be PETSC_NULL) 338 339 Level: advanced 340 341 Calling sequence of func: 342 . func (TS ts,Vec update,void *ctx,PetscReal *newdt,PetscTruth *flag); 343 344 . update - latest solution vector 345 . ctx - [optional] timestep context 346 . newdt - the timestep to use for the next step 347 . flag - flag indicating whether the last time step was acceptable 348 349 Notes: 350 The routine set here will be called by TSPseudoVerifyTimeStep() 351 during the timestepping process. 352 353 .keywords: timestep, pseudo, set, verify 354 355 .seealso: TSPseudoDefaultVerifyTimeStep(), TSPseudoVerifyTimeStep() 356 @*/ 357 PetscErrorCode PETSCTS_DLLEXPORT TSPseudoSetVerifyTimeStep(TS ts,PetscErrorCode (*dt)(TS,Vec,void*,PetscReal*,PetscTruth*),void* ctx) 358 { 359 PetscErrorCode ierr,(*f)(TS,PetscErrorCode (*)(TS,Vec,void*,PetscReal *,PetscTruth *),void *); 360 361 PetscFunctionBegin; 362 PetscValidHeaderSpecific(ts,TS_COOKIE,1); 363 364 ierr = PetscObjectQueryFunction((PetscObject)ts,"TSPseudoSetVerifyTimeStep_C",(void (**)(void))&f);CHKERRQ(ierr); 365 if (f) { 366 ierr = (*f)(ts,dt,ctx);CHKERRQ(ierr); 367 } 368 PetscFunctionReturn(0); 369 } 370 371 #undef __FUNCT__ 372 #define __FUNCT__ "TSPseudoSetTimeStepIncrement" 373 /*@ 374 TSPseudoSetTimeStepIncrement - Sets the scaling increment applied to 375 dt when using the TSPseudoDefaultTimeStep() routine. 376 377 Collective on TS 378 379 Input Parameters: 380 + ts - the timestep context 381 - inc - the scaling factor >= 1.0 382 383 Options Database Key: 384 $ -ts_pseudo_increment <increment> 385 386 Level: advanced 387 388 .keywords: timestep, pseudo, set, increment 389 390 .seealso: TSPseudoSetTimeStep(), TSPseudoDefaultTimeStep() 391 @*/ 392 PetscErrorCode PETSCTS_DLLEXPORT TSPseudoSetTimeStepIncrement(TS ts,PetscReal inc) 393 { 394 PetscErrorCode ierr,(*f)(TS,PetscReal); 395 396 PetscFunctionBegin; 397 PetscValidHeaderSpecific(ts,TS_COOKIE,1); 398 399 ierr = PetscObjectQueryFunction((PetscObject)ts,"TSPseudoSetTimeStepIncrement_C",(void (**)(void))&f);CHKERRQ(ierr); 400 if (f) { 401 ierr = (*f)(ts,inc);CHKERRQ(ierr); 402 } 403 PetscFunctionReturn(0); 404 } 405 406 #undef __FUNCT__ 407 #define __FUNCT__ "TSPseudoIncrementDtFromInitialDt" 408 /*@ 409 TSPseudoIncrementDtFromInitialDt - Indicates that a new timestep 410 is computed via the formula 411 $ dt = initial_dt*initial_fnorm/current_fnorm 412 rather than the default update, 413 $ dt = current_dt*previous_fnorm/current_fnorm. 414 415 Collective on TS 416 417 Input Parameter: 418 . ts - the timestep context 419 420 Options Database Key: 421 $ -ts_pseudo_increment_dt_from_initial_dt 422 423 Level: advanced 424 425 .keywords: timestep, pseudo, set, increment 426 427 .seealso: TSPseudoSetTimeStep(), TSPseudoDefaultTimeStep() 428 @*/ 429 PetscErrorCode PETSCTS_DLLEXPORT TSPseudoIncrementDtFromInitialDt(TS ts) 430 { 431 PetscErrorCode ierr,(*f)(TS); 432 433 PetscFunctionBegin; 434 PetscValidHeaderSpecific(ts,TS_COOKIE,1); 435 436 ierr = PetscObjectQueryFunction((PetscObject)ts,"TSPseudoIncrementDtFromInitialDt_C",(void (**)(void))&f);CHKERRQ(ierr); 437 if (f) { 438 ierr = (*f)(ts);CHKERRQ(ierr); 439 } 440 PetscFunctionReturn(0); 441 } 442 443 444 #undef __FUNCT__ 445 #define __FUNCT__ "TSPseudoSetTimeStep" 446 /*@C 447 TSPseudoSetTimeStep - Sets the user-defined routine to be 448 called at each pseudo-timestep to update the timestep. 449 450 Collective on TS 451 452 Input Parameters: 453 + ts - timestep context 454 . dt - function to compute timestep 455 - ctx - [optional] user-defined context for private data 456 required by the function (may be PETSC_NULL) 457 458 Level: intermediate 459 460 Calling sequence of func: 461 . func (TS ts,PetscReal *newdt,void *ctx); 462 463 . newdt - the newly computed timestep 464 . ctx - [optional] timestep context 465 466 Notes: 467 The routine set here will be called by TSPseudoComputeTimeStep() 468 during the timestepping process. 469 470 .keywords: timestep, pseudo, set 471 472 .seealso: TSPseudoDefaultTimeStep(), TSPseudoComputeTimeStep() 473 @*/ 474 PetscErrorCode PETSCTS_DLLEXPORT TSPseudoSetTimeStep(TS ts,PetscErrorCode (*dt)(TS,PetscReal*,void*),void* ctx) 475 { 476 PetscErrorCode ierr,(*f)(TS,PetscErrorCode (*)(TS,PetscReal *,void *),void *); 477 478 PetscFunctionBegin; 479 PetscValidHeaderSpecific(ts,TS_COOKIE,1); 480 481 ierr = PetscObjectQueryFunction((PetscObject)ts,"TSPseudoSetTimeStep_C",(void (**)(void))&f);CHKERRQ(ierr); 482 if (f) { 483 ierr = (*f)(ts,dt,ctx);CHKERRQ(ierr); 484 } 485 PetscFunctionReturn(0); 486 } 487 488 /* ----------------------------------------------------------------------------- */ 489 490 typedef PetscErrorCode (*FCN1)(TS,Vec,void*,PetscReal*,PetscTruth*); /* force argument to next function to not be extern C*/ 491 EXTERN_C_BEGIN 492 #undef __FUNCT__ 493 #define __FUNCT__ "TSPseudoSetVerifyTimeStep_Pseudo" 494 PetscErrorCode PETSCTS_DLLEXPORT TSPseudoSetVerifyTimeStep_Pseudo(TS ts,FCN1 dt,void* ctx) 495 { 496 TS_Pseudo *pseudo; 497 498 PetscFunctionBegin; 499 pseudo = (TS_Pseudo*)ts->data; 500 pseudo->verify = dt; 501 pseudo->verifyctx = ctx; 502 PetscFunctionReturn(0); 503 } 504 EXTERN_C_END 505 506 EXTERN_C_BEGIN 507 #undef __FUNCT__ 508 #define __FUNCT__ "TSPseudoSetTimeStepIncrement_Pseudo" 509 PetscErrorCode PETSCTS_DLLEXPORT TSPseudoSetTimeStepIncrement_Pseudo(TS ts,PetscReal inc) 510 { 511 TS_Pseudo *pseudo = (TS_Pseudo*)ts->data; 512 513 PetscFunctionBegin; 514 pseudo->dt_increment = inc; 515 PetscFunctionReturn(0); 516 } 517 EXTERN_C_END 518 519 EXTERN_C_BEGIN 520 #undef __FUNCT__ 521 #define __FUNCT__ "TSPseudoIncrementDtFromInitialDt_Pseudo" 522 PetscErrorCode PETSCTS_DLLEXPORT TSPseudoIncrementDtFromInitialDt_Pseudo(TS ts) 523 { 524 TS_Pseudo *pseudo = (TS_Pseudo*)ts->data; 525 526 PetscFunctionBegin; 527 pseudo->increment_dt_from_initial_dt = PETSC_TRUE; 528 PetscFunctionReturn(0); 529 } 530 EXTERN_C_END 531 532 typedef PetscErrorCode (*FCN2)(TS,PetscReal*,void*); /* force argument to next function to not be extern C*/ 533 EXTERN_C_BEGIN 534 #undef __FUNCT__ 535 #define __FUNCT__ "TSPseudoSetTimeStep_Pseudo" 536 PetscErrorCode PETSCTS_DLLEXPORT TSPseudoSetTimeStep_Pseudo(TS ts,FCN2 dt,void* ctx) 537 { 538 TS_Pseudo *pseudo = (TS_Pseudo*)ts->data; 539 540 PetscFunctionBegin; 541 pseudo->dt = dt; 542 pseudo->dtctx = ctx; 543 PetscFunctionReturn(0); 544 } 545 EXTERN_C_END 546 547 /* ----------------------------------------------------------------------------- */ 548 549 EXTERN_C_BEGIN 550 #undef __FUNCT__ 551 #define __FUNCT__ "TSCreate_Pseudo" 552 PetscErrorCode PETSCTS_DLLEXPORT TSCreate_Pseudo(TS ts) 553 { 554 TS_Pseudo *pseudo; 555 PetscErrorCode ierr; 556 557 PetscFunctionBegin; 558 ts->ops->destroy = TSDestroy_Pseudo; 559 ts->ops->view = TSView_Pseudo; 560 561 if (ts->problem_type == TS_LINEAR) { 562 SETERRQ(PETSC_ERR_ARG_WRONG,"Only for nonlinear problems"); 563 } 564 if (!ts->Arhs) { 565 SETERRQ(PETSC_ERR_ARG_WRONGSTATE,"Must set Jacobian"); 566 } 567 568 ts->ops->setup = TSSetUp_Pseudo; 569 ts->ops->step = TSStep_Pseudo; 570 ts->ops->setfromoptions = TSSetFromOptions_Pseudo; 571 572 /* create the required nonlinear solver context */ 573 ierr = SNESCreate(ts->comm,&ts->snes);CHKERRQ(ierr); 574 575 ierr = PetscNew(TS_Pseudo,&pseudo);CHKERRQ(ierr); 576 ierr = PetscLogObjectMemory(ts,sizeof(TS_Pseudo));CHKERRQ(ierr); 577 578 ierr = PetscMemzero(pseudo,sizeof(TS_Pseudo));CHKERRQ(ierr); 579 ts->data = (void*)pseudo; 580 581 pseudo->dt_increment = 1.1; 582 pseudo->increment_dt_from_initial_dt = PETSC_FALSE; 583 pseudo->dt = TSPseudoDefaultTimeStep; 584 585 ierr = PetscObjectComposeFunctionDynamic((PetscObject)ts,"TSPseudoSetVerifyTimeStep_C", 586 "TSPseudoSetVerifyTimeStep_Pseudo", 587 TSPseudoSetVerifyTimeStep_Pseudo);CHKERRQ(ierr); 588 ierr = PetscObjectComposeFunctionDynamic((PetscObject)ts,"TSPseudoSetTimeStepIncrement_C", 589 "TSPseudoSetTimeStepIncrement_Pseudo", 590 TSPseudoSetTimeStepIncrement_Pseudo);CHKERRQ(ierr); 591 ierr = PetscObjectComposeFunctionDynamic((PetscObject)ts,"TSPseudoIncrementDtFromInitialDt_C", 592 "TSPseudoIncrementDtFromInitialDt_Pseudo", 593 TSPseudoIncrementDtFromInitialDt_Pseudo);CHKERRQ(ierr); 594 ierr = PetscObjectComposeFunctionDynamic((PetscObject)ts,"TSPseudoSetTimeStep_C", 595 "TSPseudoSetTimeStep_Pseudo", 596 TSPseudoSetTimeStep_Pseudo);CHKERRQ(ierr); 597 PetscFunctionReturn(0); 598 } 599 EXTERN_C_END 600 601 #undef __FUNCT__ 602 #define __FUNCT__ "TSPseudoDefaultTimeStep" 603 /*@C 604 TSPseudoDefaultTimeStep - Default code to compute pseudo-timestepping. 605 Use with TSPseudoSetTimeStep(). 606 607 Collective on TS 608 609 Input Parameters: 610 . ts - the timestep context 611 . dtctx - unused timestep context 612 613 Output Parameter: 614 . newdt - the timestep to use for the next step 615 616 Level: advanced 617 618 .keywords: timestep, pseudo, default 619 620 .seealso: TSPseudoSetTimeStep(), TSPseudoComputeTimeStep() 621 @*/ 622 PetscErrorCode PETSCTS_DLLEXPORT TSPseudoDefaultTimeStep(TS ts,PetscReal* newdt,void* dtctx) 623 { 624 TS_Pseudo *pseudo = (TS_Pseudo*)ts->data; 625 PetscReal inc = pseudo->dt_increment,fnorm_previous = pseudo->fnorm_previous; 626 PetscErrorCode ierr; 627 628 PetscFunctionBegin; 629 ierr = TSComputeRHSFunction(ts,ts->ptime,ts->vec_sol,pseudo->func);CHKERRQ(ierr); 630 ierr = VecNorm(pseudo->func,NORM_2,&pseudo->fnorm);CHKERRQ(ierr); 631 if (pseudo->initial_fnorm == 0.0) { 632 /* first time through so compute initial function norm */ 633 pseudo->initial_fnorm = pseudo->fnorm; 634 fnorm_previous = pseudo->fnorm; 635 } 636 if (pseudo->fnorm == 0.0) { 637 *newdt = 1.e12*inc*ts->time_step; 638 } else if (pseudo->increment_dt_from_initial_dt) { 639 *newdt = inc*ts->initial_time_step*pseudo->initial_fnorm/pseudo->fnorm; 640 } else { 641 *newdt = inc*ts->time_step*fnorm_previous/pseudo->fnorm; 642 } 643 pseudo->fnorm_previous = pseudo->fnorm; 644 PetscFunctionReturn(0); 645 } 646 647 648 649 650 651 652 653 654 655 656 657 658 659 660 661 662 663