scale_atmos_dyn_tinteg_rkcommon Module Reference

module scale_atmos_dyn_tinteg_rkcommon More...

Data Types
type	rkinfo

Functions/Subroutines
subroutine, public	atmos_dyn_tinteg_rkcommon_setup (this, rk_stage_num, rk_register_num, rkcoef_a, rkcoef_b, varname_list, is_type_flux, alloc_rkwork_flag, comm_id_offset)
subroutine, public	atmos_dyn_tinteg_rkcommon_rkwork_alloc (this)
subroutine, public	atmos_dyn_tinteg_rkcommon_rkwork_dealloc (this)
subroutine, public	atmos_dyn_tinteg_rkcommon_comm (this)
subroutine, public	atmos_dyn_tinteg_rkcommon_comm_wait (this)
subroutine, public	atmos_dyn_tinteg_rkcommon_nextstage (this, nowstage, io, jo, ko, dt)
subroutine, public	atmos_dyn_tinteg_rkcommon_updatevar (this, io, jo, ko, vs, ve, dt, var)
subroutine, public	atmos_dyn_tinteg_rkcommon_updateflux (this, nowstage, io, jo, ko, va_, flux)

Detailed Description

module scale_atmos_dyn_tinteg_rkcommon

Description

A modoule providing utility for some runge-kutta schemes

Author

Team SCALE

Function/Subroutine Documentation

atmos_dyn_tinteg_rkcommon_setup()

subroutine, public scale_atmos_dyn_tinteg_rkcommon::atmos_dyn_tinteg_rkcommon_setup	(	type(rkinfo), intent(inout)	this,
		integer, intent(in)	rk_stage_num,
		integer, intent(in)	rk_register_num,
		real(rp), dimension(rk_stage_num,rk_stage_num), intent(in)	rkcoef_a,
		real(rp), dimension(rk_stage_num), intent(in)	rkcoef_b,
		character(*), dimension(:), intent(in)	varname_list,
		logical, intent(in), optional	is_type_flux,
		logical, intent(in), optional	alloc_rkwork_flag,
		integer, intent(in), optional	comm_id_offset
	)

Definition at line 81 of file scale_atmos_dyn_tinteg_rkcommon.F90.

 
    use scale_comm_cartesc, only: comm_vars8_init      
    implicit none
    
    type(RKInfo), intent(inout) :: this
    integer, intent(in) :: rk_stage_num
    integer, intent(in) :: rk_register_num
    real(RP), intent(in) :: rkcoef_a(rk_stage_num,rk_stage_num)
    real(RP), intent(in) :: rkcoef_b(rk_stage_num)
    character(*), intent(in) :: varname_list(:)
    logical, intent(in), optional :: is_type_flux
    logical, intent(in), optional :: alloc_rkwork_flag
    integer, intent(in), optional :: comm_id_offset
 
    integer :: reg_id
    integer :: var_id
    character(H_MID) :: rktag
    !------------------------------------------
 
    this%var_num = size(varname_list)
    this%stage_num = rk_stage_num
    this%register_num = rk_register_num
    if (present(is_type_flux)) then
      this%flux_flag = is_type_flux      
    else
      this%flux_flag = .false.
    end if
 
 
    allocate( this%rkcoef_a(this%stage_num,this%stage_num) )
    allocate( this%rkcoef_b(this%stage_num) )
    this%rkcoef_a(:,:) = rkcoef_a(:,:)
    this%rkcoef_b(:) = rkcoef_b(:)
 
    ! If necessary, allocate working arrays to register the tendencies at some RK stagaes. 
 
    if ( present(alloc_rkwork_flag) ) then
      if (alloc_rkwork_flag) call atmos_dyn_tinteg_rkcommon_rkwork_alloc( this )
    end if
 
    ! Allocate an array to store variables at next RK stage
 
    allocate( this%buf(ka,ia,ja,this%var_num) )
    
    !$omp parallel workshare
    this%buf(:,:,:,:)   = undef
    !$omp end parallel workshare
 
    ! Prepair some information for communication with MPI
 
    if ( .not. this%flux_flag ) then
      allocate( this%comm_ind(this%var_num) )
      do var_id = 1, this%var_num
        this%comm_ind(var_id) = var_id
      end do
      if (present(comm_id_offset)) &
        this%comm_ind(:) = comm_id_offset + this%comm_ind(:) 
 
      do var_id = 1, this%var_num
        write(rktag,'(a,a)') trim(varname_list(var_id)), 'RK'
        call comm_vars8_init( trim(rktag), this%buf(:,:,:,var_id), this%comm_ind(var_id) )
      end do
    end if
 
    return

References atmos_dyn_tinteg_rkcommon_rkwork_alloc(), scale_comm_cartesc::comm_vars8_init(), scale_atmos_grid_cartesc_index::ia, scale_atmos_grid_cartesc_index::ja, and scale_atmos_grid_cartesc_index::ka.

Here is the call graph for this function:

atmos_dyn_tinteg_rkcommon_rkwork_alloc()

subroutine, public scale_atmos_dyn_tinteg_rkcommon::atmos_dyn_tinteg_rkcommon_rkwork_alloc

(

type(rkinfo), intent(inout)

this

)

Definition at line 149 of file scale_atmos_dyn_tinteg_rkcommon.F90.

    implicit none
 
    type(RKInfo), intent(inout) :: this
    !------------------------------------------
 
    if (this%register_num > 0 .and. (.not. this%flux_flag) ) then
 
      allocate( this%work0(ka,ia,ja,this%var_num) )
      allocate( this%work(ka,ia,ja,this%var_num,this%register_num) )
 
#ifdef DEBUG      
      !$omp parallel workshare
      this%work0 (:,:,:,:)   = undef
      this%work  (:,:,:,:,:) = undef
      !$omp end parallel workshare
#endif
    end if    
 
    return

References scale_atmos_grid_cartesc_index::ia, scale_atmos_grid_cartesc_index::ja, and scale_atmos_grid_cartesc_index::ka.

Referenced by atmos_dyn_tinteg_rkcommon_setup().

Here is the caller graph for this function:

atmos_dyn_tinteg_rkcommon_rkwork_dealloc()

subroutine, public scale_atmos_dyn_tinteg_rkcommon::atmos_dyn_tinteg_rkcommon_rkwork_dealloc

(

type(rkinfo), intent(inout)

this

)

Definition at line 171 of file scale_atmos_dyn_tinteg_rkcommon.F90.

    implicit none
 
    type(RKInfo), intent(inout) :: this
    !------------------------------------------
 
    if ( allocated(this%work) ) then
      deallocate( this%work0 )
      deallocate( this%work )
    end if    
 
    return

atmos_dyn_tinteg_rkcommon_comm()

subroutine, public scale_atmos_dyn_tinteg_rkcommon::atmos_dyn_tinteg_rkcommon_comm

(

type(rkinfo), intent(inout)

this

)

Definition at line 185 of file scale_atmos_dyn_tinteg_rkcommon.F90.

    use scale_comm_cartesc, only: comm_vars8  
    implicit none
 
    type(RKInfo), intent(inout) :: this
 
    integer :: var_id
    !------------------------------------------
 
    do var_id = 1, this%var_num
      call comm_vars8( this%buf(:,:,:,var_id), this%comm_ind(var_id) )
    end do 
 
    return

atmos_dyn_tinteg_rkcommon_comm_wait()

subroutine, public scale_atmos_dyn_tinteg_rkcommon::atmos_dyn_tinteg_rkcommon_comm_wait

(

type(rkinfo), intent(inout)

this

)

Definition at line 201 of file scale_atmos_dyn_tinteg_rkcommon.F90.

    use scale_comm_cartesc, only: comm_wait
    implicit none
 
    type(RKInfo), intent(inout) :: this
 
    integer :: var_id
    !------------------------------------------
 
    do var_id = 1, this%var_num
      call comm_wait( this%buf(:,:,:,var_id), this%comm_ind(var_id), .false. )
    end do 
    
    return

atmos_dyn_tinteg_rkcommon_nextstage()

subroutine, public scale_atmos_dyn_tinteg_rkcommon::atmos_dyn_tinteg_rkcommon_nextstage	(	type(rkinfo), intent(inout)	this,
		integer, intent(in)	nowstage,
		integer, dimension(this%var_num), intent(in)	io,
		integer, dimension(this%var_num), intent(in)	jo,
		integer, dimension(this%var_num), intent(in)	ko,
		real(rp), intent(in)	dt
	)

Definition at line 217 of file scale_atmos_dyn_tinteg_rkcommon.F90.

    implicit none
    type(RKInfo), intent(inout) :: this
    integer, intent(in) :: nowstage
    integer, intent(in) :: io(this%var_num)
    integer, intent(in) :: jo(this%var_num)
    integer, intent(in) :: ko(this%var_num)   
    real(RP), intent(in) :: dt
 
    integer :: i, j, k, iv, rks
    real(RP) :: var0
 
    real(RP) :: a_(this%stage_num)
    !--------------------------------------
 
    a_(:) = dt * this%RKCoef_a(nowstage+1,:)
 
    !$omp parallel private(j,i,k,var0,iv,rks)
 
    do iv=1, this%var_num
      !$omp do collapse(2)
      do j=js-jo(iv), je
      do i=is-io(iv), ie
      do k=ks-ko(iv), ke
          var0 = this%work0(k,i,j,iv)
          this%work(k,i,j,iv,nowstage) =  this%work(k,i,j,iv,nowstage) - var0
          this%buf(k,i,j,iv) = var0 + a_(nowstage) * this%work(k,i,j,iv,nowstage)
      end do
      end do
      end do
      !$omp end do
    end do
 
    do rks=1, nowstage-1
      if ( abs(this%RKCoef_a(nowstage+1,rks)) < eps ) cycle
      do iv=1, this%var_num
        !$omp do collapse(2)
        do j=js-jo(iv), je
        do i=is-io(iv), ie
        do k=ks-ko(iv), ke
            this%buf(k,i,j,iv) = this%buf(k,i,j,iv) + a_(rks) * this%work(k,i,j,iv,rks)
        end do
        end do
        end do
        !$omp end do
      end do        
    end do
 
    !$omp end parallel
 
    return

References 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_atmos_grid_cartesc_index::ke, and scale_atmos_grid_cartesc_index::ks.

atmos_dyn_tinteg_rkcommon_updatevar()

subroutine, public scale_atmos_dyn_tinteg_rkcommon::atmos_dyn_tinteg_rkcommon_updatevar	(	type(rkinfo), intent(inout)	this,
		integer, dimension(this%var_num), intent(in)	io,
		integer, dimension(this%var_num), intent(in)	jo,
		integer, dimension(this%var_num), intent(in)	ko,
		integer, intent(in)	vs,
		integer, intent(in)	ve,
		real(rp), intent(in)	dt,
		real(rp), dimension(ka,ia,ja,vs:ve), intent(inout)	var
	)

Definition at line 270 of file scale_atmos_dyn_tinteg_rkcommon.F90.

    implicit none
 
    type(RKInfo), intent(inout) :: this
    integer, intent(in) :: io(this%var_num)
    integer, intent(in) :: jo(this%var_num)
    integer, intent(in) :: ko(this%var_num)
    real(RP), intent(in) :: dt
    integer, intent(in) :: vs, ve
    real(RP), intent(inout) :: var(KA,IA,JA,vs:ve)
 
    integer :: i, j, k, iv, rks
    real(RP) :: var0
 
    real(RP) :: b_(this%stage_num)
    !--------------------------------------
 
    b_(:) = dt * this%rkcoef_b(:)
 
    !$omp parallel private(k,i,j,var0)
 
    rks = this%stage_num
    do iv=vs, ve
      !$omp do private(k,i,j,var0) collapse(2)
      do j=js-jo(iv), je
      do i=is-io(iv), ie
      do k=ks-ko(iv), ke      
          var0 = this%work0(k,i,j,iv)
          var(k,i,j,iv) = var0  + b_(rks) * ( this%work(k,i,j,iv,rks) - var0 )
      end do
      end do
      end do
    end do
    do rks=1, this%stage_num-1
      do iv=vs, ve
        !$omp do private(k,i,j,var0) collapse(2)
        do j=js-jo(iv), je
        do i=is-io(iv), ie
        do k=ks-ko(iv), ke      
          var(k,i,j,iv) = var(k,i,j,iv) + b_(rks) * this%work(k,i,j,iv,rks)
        end do
        end do
        end do
      end do
    end do
    !$omp end parallel
    
    return

References 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_atmos_grid_cartesc_index::ke, and scale_atmos_grid_cartesc_index::ks.

atmos_dyn_tinteg_rkcommon_updateflux()

subroutine, public scale_atmos_dyn_tinteg_rkcommon::atmos_dyn_tinteg_rkcommon_updateflux	(	type(rkinfo), intent(inout)	this,
		integer, intent(in)	nowstage,
		integer, intent(in)	io,
		integer, intent(in)	jo,
		integer, intent(in)	ko,
		integer, intent(in)	va_,
		real(rp), dimension(ka,ia,ja,va_), intent(inout)	flux
	)

Definition at line 320 of file scale_atmos_dyn_tinteg_rkcommon.F90.

    implicit none
 
    type(RKInfo), intent(inout) :: this
    integer, intent(in) :: nowstage
    integer, intent(in) :: va_
    integer, intent(in) :: io, jo, ko
    real(RP), intent(inout) :: flux(KA,IA,JA,va_)
 
    integer :: i, j, k, iv
    !--------------------------------------
 
    if ( nowstage == 1) then
      !$omp parallel do private(iv,k,j,i) collapse(3)
      do iv=1, va_
      do j=js-jo, je
      do i=is-io, ie
      do k=ks-ko, ke
          flux(k,i,j,iv) = this%rkcoef_b(nowstage) * this%buf(k,i,j,iv)
      end do
      end do
      end do
      end do  
    else
      !$omp parallel do private(iv,k,j,i) collapse(3)
      do iv=1, va_
      do j=js-jo, je
      do i=is-io, ie
      do k=ks-ko, ke
        flux(k,i,j,iv) = flux(k,i,j,iv) + this%rkcoef_b(nowstage) * this%buf(k,i,j,iv)
      end do
      end do
      end do
      end do
    end if
 
    return

References 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_atmos_grid_cartesc_index::ke, and scale_atmos_grid_cartesc_index::ks.