scale_atmos_dyn_tinteg_tracer_linrk Module Reference

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_finalize
	finalize 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 70 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_finalize()

subroutine, public scale_atmos_dyn_tinteg_tracer_linrk::atmos_dyn_tinteg_tracer_linrk_finalize

finalize

Definition at line 97 of file scale_atmos_dyn_tinteg_tracer_linrk.F90.

 
    deallocate( qtrc_rk_list )
 
    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_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 115 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.

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: