Homec4science
Diffusion Introduction to CUDA (solutions) Browse

introCUDA/exercises/gradients/space_grid_mod.F90
e768432c3298solutions

Last Change
Raw File

space_grid_mod.F90
View Options

MODULE space_grid
USE prec_const
IMPLICIT NONE
PRIVATE
!
! Grid module for spatial discretization
!
! GRID Namelist
!
INTEGER, PUBLIC, PROTECTED :: nx=1 ! Number of total internal grip points in x
INTEGER, PUBLIC, PROTECTED :: ny=1 ! Number of total internal grip points in y
INTEGER, PUBLIC, PROTECTED :: nz=1 ! Number of total internal grip points in z
REAL(dp), PUBLIC, PROTECTED :: xmin=0._dp ! x coordinate for left boundary
REAL(dp), PUBLIC, PROTECTED :: xmax=10._dp ! x coordinate for right boundary
REAL(dp), PUBLIC, PROTECTED :: ymin=0._dp ! y coordinate for left boundary
REAL(dp), PUBLIC, PROTECTED :: ymax=10._dp ! y coordinate for right boundary
REAL(dp), PUBLIC, PROTECTED :: zmin=0._dp ! z coordinate for left boundary
REAL(dp), PUBLIC, PROTECTED :: zmax=1._dp ! z coordinate for right boundary
REAL(dp), PUBLIC, PROTECTED :: x_periodic_end = -10000_dp !Field line poloidal periodicity
REAL(dp), PUBLIC, PROTECTED :: Lx = 10._dp
!
! Logical switch to apply boundary conditions at the corners
! with nl_corners_top -> .true. the corner bc's are evaluated
! as vertical/poloidal bc's.
!
LOGICAL, PUBLIC, PROTECTED :: nl_corners_top = .FALSE.
!
!
!
! Number of points in each grid
!
INTEGER, PUBLIC, PROTECTED :: nxp, nyp, nzp
INTEGER, PUBLIC :: izsp, izep, nzpp
!
! Indices to start and end of local and global grids
! s -> start, e-> end (local)
! l -> left , r-> right (global)
!
INTEGER, PUBLIC :: ixs, ixe, ixl, ixr
INTEGER, PUBLIC :: iys, iye, iyl, iyr, iyr_ngrid
INTEGER, PUBLIC :: izs, ize, izl, izr
!
! Number of ghost cells
!
INTEGER, PUBLIC, PROTECTED :: nzgc=1
INTEGER, PUBLIC, PROTECTED :: nygc=1
INTEGER, PUBLIC, PROTECTED :: nxgc=1
!
! Start and end ghost cells in local and global grids
!
INTEGER, PUBLIC :: ixsg, ixeg, ixlg, ixrg
INTEGER, PUBLIC :: iysg, iyeg, iylg, iyrg
INTEGER, PUBLIC :: izsg, izeg, izlg, izrg
!Array indices in x such that each allgatherv results in the full radial length
INTEGER, PUBLIC, PROTECTED :: ix1, ix2
INTEGER, PUBLIC, PROTECTED :: ixl1,ixr2
INTEGER, PUBLIC :: nx2d, nx1d
!Array indices in x like ix1 ix2, but including all ghost cells
INTEGER, PUBLIC, PROTECTED :: ix1g, ix2g
!Array indices in y such that each allgatherv results in the full radial length
INTEGER, PUBLIC, PROTECTED :: iy1, iy2
INTEGER, PUBLIC, PROTECTED :: iyl1,iyr2
INTEGER, PUBLIC :: ny2d, ny1d
!Array indices in y like iy1 iy2, but including all ghost cells
INTEGER, PUBLIC, PROTECTED :: iy1g, iy2g
!Field line poloidal periodicity
INTEGER, PUBLIC, PROTECTED :: ix_min !min index in the open (limited) field line region for each processor
!Step size in each direction and inverses for finite differences
!Radial direction
REAL(dp), PUBLIC, PROTECTED :: deltax
REAL(dp), PUBLIC :: deltaxi
REAL(dp), PUBLIC, PROTECTED :: deltaxih
REAL(dp), PUBLIC, PROTECTED :: deltaxisq
!Poloidal/vertical direction
REAL(dp), PUBLIC, PROTECTED :: deltay
REAL(dp), PUBLIC :: deltayi
REAL(dp), PUBLIC, PROTECTED :: deltayih
REAL(dp), PUBLIC, PROTECTED :: deltayisq
real(dp), PUBLIC, PROTECTED :: shiftgr_v = 0.5_dp
!Toroidal direction
REAL(dp), PUBLIC, PROTECTED :: deltaz
real(dp), PUBLIC :: deltazi
real(dp), PUBLIC, PROTECTED :: deltazisq
!Parallel direction (should be about the same as deltaz)
real(dp), PUBLIC, PROTECTED :: deltaai
real(dp), PUBLIC, PROTECTED :: deltaaih
real(dp), PUBLIC, PROTECTED :: deltaaisq
!x,y,z grids
REAL(dp), DIMENSION(:), ALLOCATABLE, PUBLIC, PROTECTED :: xarray, yarray, zarray
!Poloidal angle theta, its sine and cosine... warning, sin and cosine are defined with theta=0 at high-field-side
real(dp), DIMENSION(:), ALLOCATABLE, PUBLIC :: cos_tht
real(dp), DIMENSION(:), ALLOCATABLE, PUBLIC :: sin_tht
!Variables related to field-aligned parallel gradient operator
!Should generalize to have iyshift=1
INTEGER, PUBLIC, PROTECTED :: iyshift
INTEGER, PUBLIC, PROTECTED :: iyshift_d
INTEGER, PUBLIC, PROTECTED :: iyshift_u
INTEGER, PUBLIC, PROTECTED :: ny_top,ny_bot !Upwinding in n2n parallel gradients
INTEGER, PUBLIC, PROTECTED :: ny_topu,ny_botu !Upwinding in n2v parallel gradient with arbitrary iyshift
INTEGER, PUBLIC, PROTECTED :: ny_topd,ny_botd !Upwinding in v2n parallel gradient with arbitrary iyshift
INTEGER, PUBLIC, PROTECTED :: ny_zg
INTEGER, PUBLIC :: kl,ku,neq
END MODULE space_grid
c4science · Help