mod_initialize_amr.t

Go to the documentation of this file.

module mod_initialize_amr
 
  implicit none
  private
 
  public :: initlevelone
  public :: modify_ic
  public :: initial_condition
  public :: improve_initial_condition
 
 
contains
 
 
  !> Generate and initialize all grids at the coarsest level (level one)
  subroutine initlevelone
    use mod_global_parameters
    use mod_ghostcells_update
    use mod_functions_connectivity, only: build_connectivity, getigrids 
    use mod_functions_forest, only: init_forest_root
    use mod_amr_solution_node, only: alloc_node
    use mod_eos, only: prepare_eos_w_fields
 
    integer :: iigrid, igrid{#ifdef evolvingboundary , morton_no}
  
    levmin=1
    levmax=1
  
    call init_forest_root
  
    call getigrids
    call build_connectivity
  
    ! fill solution space of all root grids
    do iigrid=1,igridstail; igrid=igrids(iigrid);
       call alloc_node(igrid)
       ! in case gradient routine used in initial condition, ensure geometry known
       call initial_condition(igrid)
    end do
    {#IFDEF EVOLVINGBOUNDARY
    ! mark physical-boundary blocks on space-filling curve
    do morton_no=morton_start(mype),morton_stop(mype)
       igrid=sfc_to_igrid(morton_no)
       if (phyboundblock(igrid)) sfc_phybound(morton_no)=1
    end do
    call mpi_allreduce(mpi_in_place,sfc_phybound,nleafs,mpi_integer,&
                       mpi_sum,icomm,ierrmpi)
    }
  
    ! update ghost cells
    call getbc(global_time,0.d0,ps,iwstart,nwgc)
    call prepare_eos_w_fields(ps)
  
  end subroutine initlevelone
  
  !> fill in initial condition
  subroutine initial_condition(igrid)
    ! Need only to set the mesh values (can leave ghost cells untouched)
    use mod_usr_methods, only: usr_init_one_grid
    use mod_global_parameters
    use mod_comm_lib, only: mpistop
  
    integer, intent(in) :: igrid
  
    ! in case gradient routine used in initial condition, ensure geometry known
    block=>ps(igrid)
    ^d&dxlevel(^d)=rnode(rpdx^d_,igrid);
  
    if (.not. associated(usr_init_one_grid)) then
       call mpistop("usr_init_one_grid not defined")
    else
       call usr_init_one_grid(ixg^ll,ixm^ll,ps(igrid)%w,ps(igrid)%x)
    end if
  
  end subroutine initial_condition
  
  !> modify initial condition
  subroutine modify_ic
    use mod_usr_methods, only: usr_init_one_grid
    use mod_global_parameters
    use mod_comm_lib, only: mpistop
  
    integer :: iigrid, igrid
  
    do iigrid=1,igridstail; igrid=igrids(iigrid);
       block=>ps(igrid)
       ^d&dxlevel(^d)=rnode(rpdx^d_,igrid);
  
       if (.not. associated(usr_init_one_grid)) then
          call mpistop("usr_init_one_grid not defined")
       else
          call usr_init_one_grid(ixg^ll,ixm^ll,ps(igrid)%w,ps(igrid)%x)
       end if
    end do
  
  end subroutine modify_ic
  
  !> improve initial condition after initialization
  subroutine improve_initial_condition()
    use mod_global_parameters
    use mod_usr_methods
    use mod_constrained_transport
    use mod_multigrid_coupling
    use mod_physics
    use mod_ghostcells_update
  
    logical :: active
  
    if(associated(usr_improve_initial_condition)) then
      call usr_improve_initial_condition
    else if(stagger_grid) then
      if(associated(usr_init_vector_potential)) then
        ! re-calculate magnetic field from the vector potential in a 
        ! completely divergency free way for AMR mesh in 3D
        if(levmax>levmin.and.ndim==3) call recalculateb
      end if
      if(slab_uniform.and.associated(phys_clean_divb)) then
        ! Project out the divB using multigrid poisson solver 
        ! if not initialised from vector potential
        if(.not.use_multigrid) call mg_setup_multigrid()
        call phys_clean_divb(global_time,0.d0,active)
        call getbc(global_time,0.d0,ps,iwstart,nwgc)
      end if
    end if
  
  end subroutine improve_initial_condition
 
end module mod_initialize_amr