module Atmosphere / Dyn Tinteg More...

Functions/Subroutines
subroutine, public	atmos_dyn_tinteg_tracer_linrk_setup (tinteg_type)
	Setup. More...

subroutine, public	atmos_dyn_tinteg_tracer_linrk (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)
	linear case RK More...

Detailed Description

module Atmosphere / Dyn Tinteg

Description: Temporal integration for tracer advection for Atmospheric process 'Linear case' Runge-Kutta scheme with N stage

For a linear, homogeneous and time-indepent ODE system defined by equation (9) in Baldauf (2008, JCP), the 'Linear case' Runge-Kutta scheme with N stage has Nth order accuracy.

The 'Linear case' Runge-Kutta scheme with 3 stage corresponds to the 3-stage Runge-Kutta scheme in Wicker and Skamarock (2002).

Author: Team SCALE

Function/Subroutine Documentation

◆ atmos_dyn_tinteg_tracer_linrk_setup()

subroutine, public scale_atmos_dyn_tinteg_tracer_linrk::atmos_dyn_tinteg_tracer_linrk_setup ( character(len=*) tinteg_type )

Setup.

Definition at line 69 of file scale_atmos_dyn_tinteg_tracer_linrk.F90.

     use scale_prc, only: &
        prc_abort
     use scale_comm_cartesc, only: &
        comm_vars8_init
     implicit none
  
     character(len=*) :: tinteg_type
  
     !---------------------------------------------------------------------------
  
     if ( tinteg_type(1:5) /= 'LINRK' .or. tinteg_type(7:7) /= 's' ) then
        log_error("ATMOS_DYN_Tinteg_tracer_linrkNs_setup",*) 'TINTEG_TRACER_TYPE is invalid. Check!'
        call prc_abort
     end if
     read(tinteg_type(6:6),*) nstage
  
     allocate( qtrc_rk_list(ka,ia,ja,2) )
     i_comm_rk_list(:) = (/ 1, 2 /)
     call comm_vars8_init( 'QTRC_RK1', qtrc_rk_list(:,:,:,1), i_comm_rk_list(1) )
     call comm_vars8_init( 'QTRC_RK2', qtrc_rk_list(:,:,:,2), i_comm_rk_list(2) )
  
     return

References scale_comm_cartesc::comm_vars8_init(), 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_linrk()

subroutine, public scale_atmos_dyn_tinteg_tracer_linrk::atmos_dyn_tinteg_tracer_linrk	(	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
	)

linear case RK

Definition at line 105 of file scale_atmos_dyn_tinteg_tracer_linrk.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)
     integer :: k, i, j
     integer :: nowstage
     real(RP) :: linrk_coef
     integer :: i_in, i_out, i_tmp
  
     !------------------------------------------------------------------------
     ! Start RK
     !------------------------------------------------------------------------
  
     i_in = 1; i_out = 2
  
     !$omp parallel do collapse(2) private(k)
     do j = js, je
     do i = is, ie
     do k = ks, ke
       qtrc_rk_list(k,i,j,i_in) = qtrc0(k,i,j)
     end do
     end do
     end do    
  
     do nowstage=1, nstage
       linrk_coef = 1.0_rp/real(nstage - nowstage + 1, kind=rp)
     
       call atmos_dyn_copy_boundary_tracer( qtrc_rk_list(:,:,:,i_in),  & ! [INOUT]
          qtrc0,                                                       & ! [IN]
          bnd_w, bnd_e, bnd_s, bnd_n,                                  & ! [IN]
          twod                                                         ) ! [IN]
  
       call comm_vars8( qtrc_rk_list(:,:,:,i_in), i_comm_rk_list(i_in) )
       call comm_wait ( qtrc_rk_list(:,:,:,i_in), i_comm_rk_list(i_in), .false. )
             
       if (nowstage < nstage) then
          !$omp parallel do collapse(2) private(k)
          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) ) * linrk_coef
          end do
          end do
          end do
  
          call atmos_dyn_tstep_tracer( &
                qtrc_rk_list(:,:,:,i_out), qflx,          & ! (out)
                qtrc_rk_list(:,:,:,i_in), qtrc0, rhoq_t,  & ! (in)
                dens0, dens_rk,                           & ! (in)
                mflx_hi, num_diff,                        & ! (in)
                gsqrt, mapf,                              & ! (in)
                cdz, rcdz, rcdx, rcdy,                    & ! (in)
                twod, dtl*linrk_coef,                     & ! (in)
                .false., flag_fct_along_stream            ) ! (in)
                            
       else ! final stage
          call atmos_dyn_tstep_tracer( &
                qtrc, qflx,                              & ! (out)
                qtrc_rk_list(:,:,:,i_in), qtrc0, rhoq_t, & ! (in)
                dens0, dens,                             & ! (in)
                mflx_hi, num_diff,                       & ! (in)
                gsqrt, mapf,                             & ! (in)
                cdz, rcdz, rcdx, rcdy,                   & ! (in)
                twod, dtl,                               & ! (in)
                flag_fct_tracer, flag_fct_along_stream   ) ! (in)    
            
       end if
  
       i_tmp = i_out
       i_out = i_in; i_in = i_tmp
     end do
     
     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.