Homec4science
Diffusion miniapp_gbs_gradients (master) Browse

R11986/space_grid_mod.F90
787d0516abf7master

Last Change
Raw File

space_grid_mod.F90
View Options

!> @brief Grid module for spatial discretization
module space_grid
use prec_const
implicit none
! Start and end ghost cells in local and global grids
!$omp declare target(ixsg, ixeg, iysg, iyeg, izsg, izeg, deltazi, ixs, iys, izs, ixe, iye, ize)
integer, public :: izs=3,iys=3,ixs=3 ! start index
integer, public :: ixe,iye,ize ! end index
integer, public :: nxGhost=2,nyGhost=2,nzGhost=2 ! size of ghost boundary
integer, public :: ixsg, ixeg ! start/end indexes for x
integer, public :: iysg, iyeg ! start/end indexes for y
integer, public :: izsg, izeg ! start/end indexes for z
real(dp), public :: deltaxi=0.005,deltayi=0.003,deltazi=0.03 ! space step
#ifdef _CUDA
attributes(device) :: deltazi
#endif
interface gbs_allocate
module procedure gbs_allocate_dp1,gbs_allocate_dp2,gbs_allocate_dp3,gbs_allocate_dp4
end interface gbs_allocate
contains
subroutine compute_mesh(nx,ny,nz)
integer, intent(in) :: nx,ny,nz
ixsg=ixs-nxGhost
iysg=iys-nyGhost
izsg=izs-nzGhost
!
ixe=ixs+nx-1
iye=iys+ny-1
ize=izs+nz-1
!
ixeg=ixe+nxGhost
iyeg=iye+nyGhost
izeg=ize+nzGhost
end subroutine compute_mesh
subroutine init_array(a)
real(dp), dimension(iysg:iyeg,ixsg:ixeg,izsg:izeg), intent(inout) :: a
integer :: i,j,k
!$omp parallel do default(shared) private(i, j, k)
do k=izsg,izeg
do i=ixsg,ixeg
do j=iysg,iyeg
a(j, i, k) = (dble(i)*3.2+dble(j)*2.1+dble(k)*45.12)
end do
end do
end do
end subroutine init_array
subroutine init_array2d(a)
real(dp), dimension(iysg:iyeg,ixsg:ixeg), intent(inout) :: a
integer :: i,j
do i=ixsg,ixeg
do j=iysg,iyeg
a(j, i) = (dble(i)*3.1+dble(j)*2.1)
end do
end do
end subroutine init_array2d
!> @brief Wrapper routine to allocate and initialize 1D double array
subroutine gbs_allocate_dp1(a,is1,ie1)
real(dp), dimension(:), allocatable, intent(inout) :: a !< Input array
integer, intent(in) :: is1,ie1 !< Starting and ending indices
allocate(a(is1:ie1))
a = 0.0_dp
end subroutine gbs_allocate_dp1
!> @brief Wrapper routine to allocate and initialize 2D double array
subroutine gbs_allocate_dp2(a,is1,ie1,is2,ie2)
real(dp), dimension(:,:), allocatable, intent(inout) :: a !< Input array
integer, intent(in) :: is1,ie1,is2,ie2 !< Starting and ending indices
allocate(a(is1:ie1,is2:ie2))
a = 0.0_dp
end subroutine gbs_allocate_dp2
!> @brief Wrapper routine to allocate and initialize 3D double array
subroutine gbs_allocate_dp3(a,is1,ie1,is2,ie2,is3,ie3)
real(dp), dimension(:,:,:), allocatable, intent(inout) :: a !< Input array
integer, intent(in) :: is1,ie1,is2,ie2,is3,ie3 !< Starting and ending indices
allocate(a(is1:ie1,is2:ie2,is3:ie3))
a = 0.0_dp
end subroutine gbs_allocate_dp3
!> @brief Wrapper routine to allocate and initialize 4D double array
subroutine gbs_allocate_dp4(a,is1,ie1,is2,ie2,is3,ie3,is4,ie4)
real(dp), dimension(:,:,:,:), allocatable, intent(inout) :: a !< Input array
integer, intent(in) :: is1,ie1,is2,ie2,is3,ie3,is4,ie4 !< Starting and ending indices
allocate(a(is1:ie1,is2:ie2,is3:ie3,is4:ie4))
a = 0.0_dp
end subroutine gbs_allocate_dp4
end module space_grid
c4science · Help