mod_mkinit.F90

Go to the documentation of this file.

!-------------------------------------------------------------------------------
!> module INITIAL
!!
!! @par Description
!!          subroutines for preparing initial data
!!
!! @author Yuta Kawai, Team SCALE
!!
!<
!-------------------------------------------------------------------------------
#include "scaleFElib.h"
module mod_mkinit
  !-----------------------------------------------------------------------------
  !
  !++ used modules
  !
  use scale_precision
  use scale_io
  use scale_prof
  use scale_prc
 
  use scale_const, only: &
    pi => const_pi,          &
    grav => const_grav,      &
    rplanet => const_radius, &
    ohm => const_ohm,        &
    rdry => const_rdry,      &
    cpdry => const_cpdry,    &
    cvdry => const_cvdry,    &
    pres00 => const_pre00,   &
    pstd   => const_pstd  
 
 
  use scale_element_base, only: elementbase3d
  use scale_element_hexahedral, only: hexahedralelement
  use scale_localmesh_3d, only: localmesh3d
  use scale_mesh_base3d, only: meshbase3d
  use scale_localmeshfield_base, only: localmeshfieldbase
  use scale_atm_dyn_dgm_hydrostatic, only: &
    hydrostatic_build_rho_xyz
 
  use mod_atmos_component, only: &
    atmoscomponent
      
  !-----------------------------------------------------------------------------
  implicit none
  private
  !-----------------------------------------------------------------------------
  !
  !++ Public procedure
  !
  public :: mkinit_setup
  public :: mkinit
 
  !-----------------------------------------------------------------------------
  !
  !++ Public parameters & variables
  !
  
  integer, public :: mkinit_type   = -1
  integer, public :: i_ignore      = 0
 
  !-----------------------------------------------------------------------------
  !
  !++ Private procedure
  !
 
  !-----------------------------------------------------------------------------  
contains
  !-----------------------------------------------------------------------------
  !> Setup
  subroutine mkinit_setup
    implicit none
 
    character(len=H_SHORT) :: initname = 'NONE'
 
    namelist / param_mkinit / &
      initname
 
    integer :: ierr
    !---------------------------------------------------------------------------
 
    log_newline
    log_info("MKINIT_setup",*) 'Setup'
 
    !--- read namelist
    rewind(io_fid_conf)
    read(io_fid_conf,nml=param_mkinit,iostat=ierr)
    if( ierr < 0 ) then !--- missing
      log_info("MKINIT_setup",*) 'Not found namelist. Default used.'
    elseif( ierr > 0 ) then !--- fatal error
      log_error("MKINIT_setup",*) 'Not appropriate names in namelist PARAM_MKINIT. Check!'
      call prc_abort
    endif
    log_nml(param_mkinit)
 
    select case(trim(initname))
    case('NONE')
      mkinit_type = i_ignore
    end select
    
    return
  end subroutine mkinit_setup
 
  !-----------------------------------------------------------------------------
  !> Driver
  subroutine mkinit( output,                                      & ! (out)
    model_mesh,                                                   & ! (in)
    atm_prgvars_manager, atm_auxvars_manager, atm_trcvars_manager )
  
    use scale_model_var_manager, only: modelvarmanager
    use mod_atmos_mesh, only: atmosmesh
    use mod_atmos_vars, only: &
      atmosvars_getlocalmeshprgvars
 
    implicit none
 
    logical, intent(out) :: output
    class(atmosmesh), target, intent(in) :: model_mesh
    class(modelvarmanager), intent(inout) :: atm_prgvars_manager
    class(modelvarmanager), intent(inout) :: atm_auxvars_manager
    class(modelvarmanager), intent(inout) :: atm_trcvars_manager
 
    class(localmeshfieldbase), pointer :: ddens, momx, momy, momz, drhot
    class(localmeshfieldbase), pointer :: dens_hyd, pres_hyd
    class(localmeshfieldbase), pointer :: rtot, cptot, cvtot
 
    integer :: n
    integer :: ke, p
    class(localmesh3d), pointer :: lcmesh3d
    class(meshbase3d), pointer :: mesh
 
    integer :: np
    !---------------------------------------------------------------------------
 
    mesh => model_mesh%ptr_mesh
 
    if ( mkinit_type == i_ignore ) then
      log_newline
      log_progress(*) 'skip  making initial data'
      output = .false.
    else
      log_newline
      log_progress(*) 'start making initial data'
 
      ! call PROF_rapstart('_MkInit_main',3)   
      
      do n=1, mesh%LOCAL_MESH_NUM
        call atmosvars_getlocalmeshprgvars( n, &
          mesh, atm_prgvars_manager, atm_auxvars_manager, &
          ddens, momx, momy, momz, drhot,                 &
          dens_hyd, pres_hyd, rtot, cvtot, cptot,         &
          lcmesh3d                                        )
        !$acc enter data attach(DDENS%val, MOMX%val, MOMY%val, MOMZ%val, DRHOT%val, Rtot%val, CVtot%val, CPtot%val, lcmesh3D)
 
        np = lcmesh3d%refElem%Np
        !$omp parallel do
        !$acc parallel loop gang present(DDENS%val, MOMX%val, MOMY%val, MOMZ%val, DRHOT%val, DENS_hyd%val, Rtot%val, CVtot%val, CPtot%val, lcmesh3D)
        do ke=lcmesh3d%NeS, lcmesh3d%NeE
          !$acc loop vector
          do p=1, np
            momx %val(p,ke) = 0.0_rp
            momy %val(p,ke) = 0.0_rp
            momz %val(p,ke) = 0.0_rp
            ddens%val(p,ke) = 0.0_rp
            drhot%val(p,ke) = 0.0_rp
 
            cptot%val(p,ke) = cpdry
            cvtot%val(p,ke) = cvdry
            rtot%val(p,ke) = rdry
          end do
        end do
      end do
      ! call PROF_rapend  ('_MkInit_main',3)
        
      ! LOG_PROGRESS(*) 'end   making initial data'
      output = .true.
    end if
 
    return
  end subroutine mkinit
 
  !-- private---------------------------------------------------------------------
end module mod_mkinit