@@ -442,36 +442,36 @@ contains
442442 end do
443443
444444 if (any (time_stepper == (/ 1 , 2 , 3 / ))) then
445- ! Time stepper index
446- if (time_stepper == 1 ) then
447- stor = 1
448- else
449- stor = 2
450- end if
451-
452- @:ALLOCATE (rk_coef(time_stepper, 3 ))
453- if (time_stepper == 1 ) then
454- rk_coef(1 , 1 ) = 1._wp
455- rk_coef(1 , 2 ) = 0._wp
456- rk_coef(1 , 3 ) = 1._wp
457- else if (time_stepper == 2 ) then
458- rk_coef(1 , 1 ) = 1._wp
459- rk_coef(1 , 2 ) = 0._wp
460- rk_coef(1 , 3 ) = 1._wp
461- rk_coef(2 , 1 ) = 0.5_wp
462- rk_coef(2 , 2 ) = 0.5_wp
463- rk_coef(2 , 3 ) = 0.5_wp
464- else if (time_stepper == 3 ) then
465- rk_coef(1 , 1 ) = 1._wp
466- rk_coef(1 , 2 ) = 0._wp
467- rk_coef(1 , 3 ) = 1._wp
468- rk_coef(2 , 1 ) = 1._wp / 4._wp
469- rk_coef(2 , 2 ) = 3._wp / 4._wp
470- rk_coef(2 , 3 ) = 1._wp / 4._wp
471- rk_coef(3 , 1 ) = 2._wp / 3._wp
472- rk_coef(3 , 2 ) = 1._wp / 3._wp
473- rk_coef(3 , 3 ) = 2._wp / 3._wp
474- end if
445+ ! Time stepper index
446+ if (time_stepper == 1 ) then
447+ stor = 1
448+ else
449+ stor = 2
450+ end if
451+
452+ @:ALLOCATE (rk_coef(time_stepper, 3 ))
453+ if (time_stepper == 1 ) then
454+ rk_coef(1 , 1 ) = 1._wp
455+ rk_coef(1 , 2 ) = 0._wp
456+ rk_coef(1 , 3 ) = 1._wp
457+ else if (time_stepper == 2 ) then
458+ rk_coef(1 , 1 ) = 1._wp
459+ rk_coef(1 , 2 ) = 0._wp
460+ rk_coef(1 , 3 ) = 1._wp
461+ rk_coef(2 , 1 ) = 0.5_wp
462+ rk_coef(2 , 2 ) = 0.5_wp
463+ rk_coef(2 , 3 ) = 0.5_wp
464+ else if (time_stepper == 3 ) then
465+ rk_coef(1 , 1 ) = 1._wp
466+ rk_coef(1 , 2 ) = 0._wp
467+ rk_coef(1 , 3 ) = 1._wp
468+ rk_coef(2 , 1 ) = 1._wp / 4._wp
469+ rk_coef(2 , 2 ) = 3._wp / 4._wp
470+ rk_coef(2 , 3 ) = 1._wp / 4._wp
471+ rk_coef(3 , 1 ) = 2._wp / 3._wp
472+ rk_coef(3 , 2 ) = 1._wp / 3._wp
473+ rk_coef(3 , 3 ) = 2._wp / 3._wp
474+ end if
475475 end if
476476
477477 end subroutine s_initialize_time_steppers_module
@@ -491,7 +491,7 @@ contains
491491 call cpu_time(start)
492492 call nvtxStartRange(" TIMESTEP" )
493493
494- ! Adaptive dt: initial stage
494+ ! Adaptive dt: initial stage
495495 if (adap_dt) call s_adaptive_dt_bubble(1 )
496496
497497 do s = 1 , nstage
@@ -523,13 +523,13 @@ contains
523523 do k = 0 , n
524524 do j = 0 , m
525525 if (s == 1 .and. time_stepper /= 1 ) then
526- q_cons_ts(stor)%vf(i)%sf(j, k, l) = &
527- q_cons_ts(1 )%vf(i)%sf(j, k, l)
526+ q_cons_ts(stor)%vf(i)%sf(j, k, l) = &
527+ q_cons_ts(1 )%vf(i)%sf(j, k, l)
528528 end if
529529 q_cons_ts(1 )%vf(i)%sf(j, k, l) = &
530530 rk_coef(s, 1 )* q_cons_ts(1 )%vf(i)%sf(j, k, l) &
531- + rk_coef(s, 2 )* q_cons_ts(stor)%vf(i)%sf(j, k, l) &
532- + rk_coef(s, 3 )* dt* rhs_vf(i)%sf(j, k, l)
531+ + rk_coef(s, 2 )* q_cons_ts(stor)%vf(i)%sf(j, k, l) &
532+ + rk_coef(s, 3 )* dt* rhs_vf(i)%sf(j, k, l)
533533 end do
534534 end do
535535 end do
@@ -544,17 +544,17 @@ contains
544544 do j = 0 , m
545545 do q = 1 , nnode
546546 if (s == 1 .and. time_stepper /= 1 ) then
547- pb_ts(stor)%sf(j, k, l, q, i) = &
548- pb_ts(1 )%sf(j, k, l, q, i)
549- mv_ts(stor)%sf(j, k, l, q, i) = &
550- mv_ts(1 )%sf(j, k, l, q, i)
547+ pb_ts(stor)%sf(j, k, l, q, i) = &
548+ pb_ts(1 )%sf(j, k, l, q, i)
549+ mv_ts(stor)%sf(j, k, l, q, i) = &
550+ mv_ts(1 )%sf(j, k, l, q, i)
551551 end if
552552 pb_ts(1 )%sf(j, k, l, q, i) = &
553- rk_coef(s, 1 )* pb_ts(1 )%sf(j, k, l, q, i) &
553+ rk_coef(s, 1 )* pb_ts(1 )%sf(j, k, l, q, i) &
554554 + rk_coef(s, 2 )* pb_ts(stor)%sf(j, k, l, q, i) &
555555 + rk_coef(s, 3 )* dt* rhs_pb(j, k, l, q, i)
556556 mv_ts(1 )%sf(j, k, l, q, i) = &
557- rk_coef(s, 1 )* mv_ts(1 )%sf(j, k, l, q, i) &
557+ rk_coef(s, 1 )* mv_ts(1 )%sf(j, k, l, q, i) &
558558 + rk_coef(s, 2 )* mv_ts(stor)%sf(j, k, l, q, i) &
559559 + rk_coef(s, 3 )* dt* rhs_mv(j, k, l, q, i)
560560 end do
@@ -583,7 +583,7 @@ contains
583583 end if
584584 end do
585585
586- ! Adaptive dt: final stage
586+ ! Adaptive dt: final stage
587587 if (adap_dt) call s_adaptive_dt_bubble(3 )
588588
589589 call nvtxEndRange
0 commit comments