scale_atmos_dyn_tinteg_tracer_rk3.F90

Go to the documentation of this file.

!-------------------------------------------------------------------------------
!-------------------------------------------------------------------------------
#include "inc_openmp.h"
module scale_atmos_dyn_tinteg_tracer_rk3
  !-----------------------------------------------------------------------------
  !
  !++ used modules
  !
  use scale_precision
  use scale_stdio
  use scale_prof
  use scale_grid_index
  use scale_index
  use scale_tracer

#ifdef DEBUG
  use scale_debug, only: &
     check
  use scale_const, only: &
     undef  => const_undef, &
     iundef => const_undef2
#endif
  !-----------------------------------------------------------------------------
  implicit none
  private
  !-----------------------------------------------------------------------------
  !
  !++ Public procedure
  !
  public :: atmos_dyn_tinteg_tracer_rk3_setup
  public :: atmos_dyn_tinteg_tracer_rk3

  !-----------------------------------------------------------------------------
  !
  !++ Public parameters & variables
  !
  !-----------------------------------------------------------------------------
  !
  !++ Private procedure
  !
  !-----------------------------------------------------------------------------
  !
  !++ Private parameters & variables
  !
  real(RP), allocatable :: qtrc_rk1(:,:,:)
  real(RP), allocatable :: qtrc_rk2(:,:,:)
  integer :: i_comm_rk1 = 1
  integer :: i_comm_rk2 = 1
  !-----------------------------------------------------------------------------
contains

  !-----------------------------------------------------------------------------
  subroutine atmos_dyn_tinteg_tracer_rk3_setup( &
       tinteg_type )
    use scale_process, only: &
       prc_mpistop
    use scale_const, only: &
       undef => const_undef
    use scale_comm, only: &
       comm_vars8_init
    implicit none

    character(len=*) :: tinteg_type

    integer :: iv
    !---------------------------------------------------------------------------

    if ( tinteg_type /= 'RK3WS2002' ) then
       write(*,*) 'xxx TINTEG_LARGE_TYPE is not RK3WS2002. Check!'
       call prc_mpistop
    end if

    allocate( qtrc_rk1(ka,ia,ja) )
    allocate( qtrc_rk2(ka,ia,ja) )

    call comm_vars8_init( qtrc_rk1, i_comm_rk1 )
    call comm_vars8_init( qtrc_rk2, i_comm_rk2 )

    return
  end subroutine atmos_dyn_tinteg_tracer_rk3_setup

  !-----------------------------------------------------------------------------
  subroutine atmos_dyn_tinteg_tracer_rk3( &
       QTRC, & ! (out)
       qtrc0, rhoq_t, &! (in)
       dens0, dens, & ! (in)
       mflx_hi, num_diff, & ! (in)
       gsqrt, mapf, & ! (in)
       cdz, rcdz, rcdx, rcdy, & ! (in)
       dtl, & ! (in)
       flag_fct_tracer, & ! (in)
       flag_fct_along_stream ) ! (in)
    use scale_comm, only: &
       comm_vars8, &
       comm_wait
    use scale_atmos_dyn_tstep_tracer, only: &
       atmos_dyn_tstep_tracer
    implicit none
    real(RP), intent(inout) :: QTRC    (ka,ia,ja)
    real(RP), intent(in)    :: QTRC0   (ka,ia,ja)
    real(RP), intent(in)    :: RHOQ_t  (ka,ia,ja)
    real(RP), intent(in)    :: DENS0   (ka,ia,ja)
    real(RP), intent(in)    :: DENS    (ka,ia,ja)
    real(RP), intent(in)    :: mflx_hi (ka,ia,ja,3)
    real(RP), intent(in)    :: num_diff(ka,ia,ja,3)
    real(RP), intent(in)    :: GSQRT   (ka,ia,ja,7)
    real(RP), intent(in)    :: MAPF    (ia,ja)
    real(RP), intent(in)    :: CDZ(ka)
    real(RP), intent(in)    :: RCDZ(ka)
    real(RP), intent(in)    :: RCDX(ia)
    real(RP), intent(in)    :: RCDY(ja)
    real(RP), intent(in)    :: dtl
    logical,  intent(in)    :: FLAG_FCT_TRACER
    logical,  intent(in)    :: FLAG_FCT_ALONG_STREAM

    real(RP) :: DENS_RK(ka,ia,ja)
    real(RP) :: dtrk
    integer :: k, i, j

    !------------------------------------------------------------------------
    ! Start RK
    !------------------------------------------------------------------------

    do j = js-1, je+1
    do i = is-1, ie+1
    do k = ks, ke
       dens_rk(k,i,j) = dens0(k,i,j) &
                      + ( dens(k,i,j) - dens0(k,i,j) ) / 3.0_rp
    end do
    end do
    end do

    dtrk = dtl / 3.0_rp
    call atmos_dyn_tstep_tracer( &
         qtrc_rk1, & ! (out)
         qtrc, qtrc0, rhoq_t, &! (in)
         dens0, dens_rk, & ! (in)
         mflx_hi, num_diff, & ! (in)
         gsqrt, mapf, & ! (in)
         cdz, rcdz, rcdx, rcdy, & ! (in)
         dtrk, & ! (in)
         .false., flag_fct_along_stream ) ! (in)

    call comm_vars8( qtrc_rk1(:,:,:), i_comm_rk1 )
    call comm_wait ( qtrc_rk1(:,:,:), i_comm_rk1, .false. )
    

    do j = js-1, je+1
    do i = is-1, ie+1
    do k = ks, ke
       dens_rk(k,i,j) = dens0(k,i,j) &
                      + ( dens(k,i,j) - dens0(k,i,j) ) * 0.5_rp
    end do
    end do
    end do

    dtrk = dtl / 2.0_rp
    call atmos_dyn_tstep_tracer( &
         qtrc_rk2, & ! (out)
         qtrc_rk1, qtrc0, rhoq_t, &! (in)
         dens0, dens_rk, & ! (in)
         mflx_hi, num_diff, & ! (in)
         gsqrt, mapf, & ! (in)
         cdz, rcdz, rcdx, rcdy, & ! (in)
         dtrk, & ! (in)
         .false., flag_fct_along_stream ) ! (in)

    call comm_vars8( qtrc_rk2(:,:,:), i_comm_rk2 )
    call comm_wait ( qtrc_rk2(:,:,:), i_comm_rk2, .false. )


    dtrk = dtl
    call atmos_dyn_tstep_tracer( &
         qtrc, & ! (out)
         qtrc_rk2, qtrc0, rhoq_t, &! (in)
         dens0, dens, & ! (in)
         mflx_hi, num_diff, & ! (in)
         gsqrt, mapf, & ! (in)
         cdz, rcdz, rcdx, rcdy, & ! (in)
         dtrk, & ! (in)
         flag_fct_tracer, flag_fct_along_stream ) ! (in)

    return
  end subroutine atmos_dyn_tinteg_tracer_rk3

end module scale_atmos_dyn_tinteg_tracer_rk3