scale_atmos_dyn_tinteg_tracer_rk3 Module Reference

module Atmosphere / Dyn Tinteg More...

Functions/Subroutines
subroutine, public	atmos_dyn_tinteg_tracer_rk3_setup (tinteg_type)
	Setup. More...
subroutine, public	atmos_dyn_tinteg_tracer_rk3_finalize
	finalize More...
subroutine, public	atmos_dyn_tinteg_tracer_rk3 (QTRC, qflx, QTRC0, RHOQ_t, DENS0, DENS, mflx_hi, num_diff, GSQRT, MAPF, CDZ, RCDZ, RCDX, RCDY, BND_W, BND_E, BND_S, BND_N, TwoD, dtl, FLAG_FCT_TRACER, FLAG_FCT_ALONG_STREAM)
	RK3. More...

Detailed Description

module Atmosphere / Dyn Tinteg

Description

Temporal integration for tracer advection for Atmospheric process three stage Runge-Kutta scheme

Author

Team SCALE

Function/Subroutine Documentation

atmos_dyn_tinteg_tracer_rk3_setup()

subroutine, public scale_atmos_dyn_tinteg_tracer_rk3::atmos_dyn_tinteg_tracer_rk3_setup

(

character(len=*)

tinteg_type

)

Setup.

Definition at line 66 of file scale_atmos_dyn_tinteg_tracer_rk3.F90.

    use scale_const, only: &
       undef => const_undef
    use scale_prc, only: &
       prc_abort
    use scale_comm_cartesc, only: &
       comm_vars8_init
    implicit none
 
    character(len=*) :: tinteg_type
 
    integer :: iv
    !---------------------------------------------------------------------------
 
    if ( tinteg_type /= 'RK3WS2002' ) then
       log_error("ATMOS_DYN_Tinteg_tracer_rk3_setup",*) 'TINTEG_TRACER_TYPE is not RK3WS2002. Check!'
       call prc_abort
    end if
 
    allocate( qtrc_rk1(ka,ia,ja) )
    allocate( qtrc_rk2(ka,ia,ja) )
    qtrc_rk1(:,:,:) = undef
    qtrc_rk2(:,:,:) = undef
    !$acc enter data create(QTRC_RK1, QTRC_RK2)
 
    i_comm_rk1 = 1
    call comm_vars8_init( 'QTRC_RK1', qtrc_rk1, i_comm_rk1 )
 
    i_comm_rk2 = 1
    call comm_vars8_init( 'QTRC_RK2', qtrc_rk2, i_comm_rk2 )
 
    return

References scale_comm_cartesc::comm_vars8_init(), scale_const::const_undef, scale_atmos_grid_cartesc_index::ia, scale_atmos_grid_cartesc_index::ja, scale_atmos_grid_cartesc_index::ka, and scale_prc::prc_abort().

Referenced by scale_atmos_dyn_tinteg_tracer::atmos_dyn_tinteg_tracer_setup().

Here is the call graph for this function:

Here is the caller graph for this function:

atmos_dyn_tinteg_tracer_rk3_finalize()

subroutine, public scale_atmos_dyn_tinteg_tracer_rk3::atmos_dyn_tinteg_tracer_rk3_finalize

finalize

Definition at line 102 of file scale_atmos_dyn_tinteg_tracer_rk3.F90.

 
    !$acc exit data delete(QTRC_RK1, QTRC_RK2)
    deallocate( qtrc_rk1 )
    deallocate( qtrc_rk2 )
 
    return

Referenced by scale_atmos_dyn_tinteg_tracer::atmos_dyn_tinteg_tracer_setup().

Here is the caller graph for this function:

atmos_dyn_tinteg_tracer_rk3()

subroutine, public scale_atmos_dyn_tinteg_tracer_rk3::atmos_dyn_tinteg_tracer_rk3	(	real(rp), dimension (ka,ia,ja), intent(inout)	QTRC,
		real(rp), dimension (ka,ia,ja,3), intent(out)	qflx,
		real(rp), dimension (ka,ia,ja), intent(in)	QTRC0,
		real(rp), dimension (ka,ia,ja), intent(in)	RHOQ_t,
		real(rp), dimension (ka,ia,ja), intent(in)	DENS0,
		real(rp), dimension (ka,ia,ja), intent(in)	DENS,
		real(rp), dimension (ka,ia,ja,3), intent(in)	mflx_hi,
		real(rp), dimension(ka,ia,ja,3), intent(in)	num_diff,
		real(rp), dimension (ka,ia,ja,7), intent(in)	GSQRT,
		real(rp), dimension (ia,ja), intent(in)	MAPF,
		real(rp), dimension(ka), intent(in)	CDZ,
		real(rp), dimension(ka), intent(in)	RCDZ,
		real(rp), dimension(ia), intent(in)	RCDX,
		real(rp), dimension(ja), intent(in)	RCDY,
		logical, intent(in)	BND_W,
		logical, intent(in)	BND_E,
		logical, intent(in)	BND_S,
		logical, intent(in)	BND_N,
		logical, intent(in)	TwoD,
		real(rp), intent(in)	dtl,
		logical, intent(in)	FLAG_FCT_TRACER,
		logical, intent(in)	FLAG_FCT_ALONG_STREAM
	)

RK3.

Definition at line 125 of file scale_atmos_dyn_tinteg_tracer_rk3.F90.

    use scale_comm_cartesc, only: &
       comm_vars8, &
       comm_wait
    use scale_atmos_dyn_tstep_tracer, only: &
       atmos_dyn_tstep_tracer
    use scale_atmos_dyn_common, only: &
       atmos_dyn_copy_boundary_tracer
    implicit none
    real(RP), intent(inout) :: QTRC    (KA,IA,JA)
    real(RP), intent(out)   :: qflx    (KA,IA,JA,3)
    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)
    logical,  intent(in)    :: BND_W
    logical,  intent(in)    :: BND_E
    logical,  intent(in)    :: BND_S
    logical,  intent(in)    :: BND_N
    logical,  intent(in)    :: TwoD
    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
 
    !$acc data copy(QTRC), copyout(qflx) &
    !$acc      copyin(QTRC0, RHOQ_t, DENS0, DENS, mflx_hi, num_diff, &
    !$acc             GSQRT, MAPF, CDZ, RCDZ, RCDX, RCDY) &
    !$acc      create(DENS_RK)
    !------------------------------------------------------------------------
    ! Start RK
    !------------------------------------------------------------------------
 
    !$acc kernels
    do j = js-1, je+1
    do i = max(is-1,1), min(ie+1,ia)
    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
    !$acc end kernels
 
    dtrk = dtl / 3.0_rp
    call atmos_dyn_tstep_tracer( &
         qtrc_rk1, & ! (out)
         qflx, & ! (out)
         qtrc, qtrc0, rhoq_t, &! (in)
         dens0, dens_rk, & ! (in)
         mflx_hi, num_diff, & ! (in)
         gsqrt, mapf, & ! (in)
         cdz, rcdz, rcdx, rcdy, & ! (in)
         twod, dtrk, & ! (in)
         .false., flag_fct_along_stream ) ! (in)
 
    call atmos_dyn_copy_boundary_tracer( qtrc_rk1,                   & ! [INOUT]
                                         qtrc0,                      & ! [IN]
                                         bnd_w, bnd_e, bnd_s, bnd_n, & ! [IN]
                                         twod                        ) ! [IN]
 
    call comm_vars8( qtrc_rk1(:,:,:), i_comm_rk1 )
    call comm_wait ( qtrc_rk1(:,:,:), i_comm_rk1, .false. )
 
 
    !$acc kernels
    do j = js-1, je+1
    do i = max(is-1,1), min(ie+1,ia)
    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
    !$acc end kernels
 
    dtrk = dtl / 2.0_rp
    call atmos_dyn_tstep_tracer( &
         qtrc_rk2, & ! (out)
         qflx, & ! (out)
         qtrc_rk1, qtrc0, rhoq_t, &! (in)
         dens0, dens_rk, & ! (in)
         mflx_hi, num_diff, & ! (in)
         gsqrt, mapf, & ! (in)
         cdz, rcdz, rcdx, rcdy, & ! (in)
         twod, dtrk, & ! (in)
         .false., flag_fct_along_stream ) ! (in)
 
    call atmos_dyn_copy_boundary_tracer( qtrc_rk2,                   & ! [INOUT]
                                         qtrc0,                      & ! [IN]
                                         bnd_w, bnd_e, bnd_s, bnd_n, & ! [IN]
                                         twod                        ) ! [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)
         qflx, & ! (out)
         qtrc_rk2, qtrc0, rhoq_t, &! (in)
         dens0, dens, & ! (in)
         mflx_hi, num_diff, & ! (in)
         gsqrt, mapf, & ! (in)
         cdz, rcdz, rcdx, rcdy, & ! (in)
         twod, & ! (in)
         dtrk, & ! (in)
         flag_fct_tracer, flag_fct_along_stream ) ! (in)
 
    !$acc end data
 
    return

References scale_atmos_dyn_common::atmos_dyn_copy_boundary_tracer(), scale_atmos_dyn_tstep_tracer::atmos_dyn_tstep_tracer, scale_atmos_grid_cartesc_index::ia, scale_atmos_grid_cartesc_index::ie, scale_atmos_grid_cartesc_index::is, scale_atmos_grid_cartesc_index::je, scale_atmos_grid_cartesc_index::js, scale_tracer::k, scale_atmos_grid_cartesc_index::ke, and scale_atmos_grid_cartesc_index::ks.

Referenced by scale_atmos_dyn_tinteg_tracer::atmos_dyn_tinteg_tracer_setup().

Here is the call graph for this function:

Here is the caller graph for this function: