@@ -357,10 +357,10 @@ contains
357
357
358
358
end subroutine s_initialize_time_steppers_module
359
359
360
- subroutine s_evolve_q_pb_mv (index , scaler1 , scaler2 , scaler3 ) !! TODO :: Get a better name for this
360
+ subroutine s_evolve_q_pb_mv (index , scaler1 , scaler2 , scaler3 , scaler4 ) !! TODO :: Get a better name for this
361
361
362
362
integer , intent (in ) :: index !! TODO :: Rename this
363
- real (wp), intent (in ) :: scaler1, scaler2, scaler3 !! TODO :: Rename these too
363
+ real (wp), intent (in ) :: scaler1, scaler2, scaler3, scale r4 !! TODO :: Rename these too
364
364
integer :: i, j, k, l, q
365
365
366
366
!$acc parallel loop collapse(4 ) gang vector default(present)
@@ -371,7 +371,7 @@ contains
371
371
q_cons_ts(index)%vf(i)%sf(j, k, l) = &
372
372
(scaler1* q_cons_ts(1 )%vf(i)%sf(j, k, l) &
373
373
+ scaler2* q_cons_ts(2 )%vf(i)%sf(j, k, l) &
374
- + scaler3* dt* rhs_vf(i)%sf(j, k, l))/ (scaler1 + scaler2) !! TODO :: scaler1 + scaler2 should be called a normalization constant
374
+ + scaler3* dt* rhs_vf(i)%sf(j, k, l))/ scale r4 !! TODO :: scaler1 + scaler2 should be called a normalization constant
375
375
end do
376
376
end do
377
377
end do
@@ -388,7 +388,7 @@ contains
388
388
pb_ts(index)%sf(j, k, l, q, i) = &
389
389
(scaler1* pb_ts(1 )%sf(j, k, l, q, i) &
390
390
+ scaler2* pb_ts(2 )%sf(j, k, l, q, i) &
391
- + scaler3* dt* rhs_pb(j, k, l, q, i))/ (scaler1 + scaler2)
391
+ + scaler3* dt* rhs_pb(j, k, l, q, i))/ scale r4
392
392
end do
393
393
end do
394
394
end do
@@ -404,7 +404,7 @@ contains
404
404
mv_ts(index)%sf(j, k, l, q, i) = &
405
405
(scaler1* mv_ts(1 )%sf(j, k, l, q, i) &
406
406
+ scaler2* mv_ts(2 )%sf(j, k, l, q, i) &
407
- + scaler3* dt* rhs_mv(j, k, l, q, i))/ (scaler1 + scaler2)
407
+ + scaler3* dt* rhs_mv(j, k, l, q, i))/ scale r4
408
408
end do
409
409
end do
410
410
end do
@@ -479,6 +479,7 @@ contains
479
479
pb_ts(1 )%sf(j, k, l, q, i) = &
480
480
pb_ts(1 )%sf(j, k, l, q, i) &
481
481
+ dt* rhs_pb(j, k, l, q, i)
482
+
482
483
mv_ts(1 )%sf(j, k, l, q, i) = &
483
484
mv_ts(1 )%sf(j, k, l, q, i) &
484
485
+ dt* rhs_mv(j, k, l, q, i)
@@ -546,7 +547,7 @@ contains
546
547
call s_update_lagrange_tdv_rk(stage= 1 )
547
548
end if
548
549
549
- call s_evolve_q_pb_mv(2 , 1.0_wp , 0.0_wp , 1.0_wp )
550
+ call s_evolve_q_pb_mv(2 , 1.0_wp , 0.0_wp , 1.0_wp , 1.0_wp )
550
551
551
552
if (bodyForces) call s_apply_bodyforces(q_cons_ts(2 )%vf, q_prim_vf, rhs_vf, dt)
552
553
@@ -575,7 +576,7 @@ contains
575
576
call s_update_lagrange_tdv_rk(stage= 2 )
576
577
end if
577
578
578
- call s_evolve_q_pb_mv(1 , 1.0_wp , 1.0_wp , 1.0_wp )
579
+ call s_evolve_q_pb_mv(1 , 1.0_wp , 1.0_wp , 1.0_wp , 2.0_wp )
579
580
580
581
if (bodyForces) call s_apply_bodyforces(q_cons_ts(1 )%vf, q_prim_vf, rhs_vf, 2._wp * dt/ 3._wp )
581
582
@@ -640,7 +641,7 @@ contains
640
641
call s_update_lagrange_tdv_rk(stage= 1 )
641
642
end if
642
643
643
- call s_evolve_q_pb_mv(2 , 1.0_wp , 0.0_wp , 1.0_wp )
644
+ call s_evolve_q_pb_mv(2 , 1.0_wp , 0.0_wp , 1.0_wp , 1.0_wp )
644
645
645
646
if (bodyForces) call s_apply_bodyforces(q_cons_ts(2 )%vf, q_prim_vf, rhs_vf, dt)
646
647
@@ -669,7 +670,7 @@ contains
669
670
call s_update_lagrange_tdv_rk(stage= 2 )
670
671
end if
671
672
672
- call s_evolve_q_pb_mv(2 , 3.0_wp , 1.0_wp , 1.0_wp )
673
+ call s_evolve_q_pb_mv(2 , 3.0_wp , 1.0_wp , 1.0_wp , 4.0_wp )
673
674
674
675
if (bodyForces) call s_apply_bodyforces(q_cons_ts(2 )%vf, q_prim_vf, rhs_vf, dt/ 4._wp )
675
676
@@ -697,7 +698,7 @@ contains
697
698
call s_update_lagrange_tdv_rk(stage= 3 )
698
699
end if
699
700
700
- call s_evolve_q_pb_mv(1 , 1.0_wp , 2.0_wp , 2.0_wp )
701
+ call s_evolve_q_pb_mv(1 , 1.0_wp , 2.0_wp , 2.0_wp , 3.0_wp )
701
702
702
703
if (bodyForces) call s_apply_bodyforces(q_cons_ts(1 )%vf, q_prim_vf, rhs_vf, 2._wp * dt/ 3._wp )
703
704
0 commit comments