Homec4science
Diffusion COSOlver (master) Browse

cosolver/src/grid_mod.f90
5b66a09e0f20master

Last Change
Raw File

grid_mod.f90
View Options

module grid
! grid module. Contained Hermite-Laguerre indices
use prec_const
use basic
use basic_mpi
use model
implicit none
public
! local dimensions
! Electron local indices
INTEGER :: lbare_s_l, lbare_e_l
! Ion local indices
INTEGER :: lbari_s_l, lbari_e_l
! Electron-Electron indices
INTEGER :: lbaree_s_l,lbaree_e_l
! ion- ion local indices
INTEGER :: lbarii_s_l,lbarii_e_l
INTEGER :: lbaremax,lbarimax
! ion-ion indicies 1D list
INTEGER, DIMENSION(:), ALLOCATABLE :: lbari_list ! local list (list of row) to be computed on each MPI task
INTEGER, DIMENSION(:), ALLOCATABLE :: rows_i ! Global array of ion rows
INTEGER, DIMENSION(:), ALLOCATABLE :: rows_e ! Global array of electrons rows
! ion indices for GK Coulomb
INTEGER, DIMENSION(:,:), ALLOCATABLE :: rows_i_4dim !
INTEGER, DIMENSION(:,:), ALLOCATABLE :: rows_e_4dim !
contains
subroutine set_grid
! construct grid (rows of collisional matrices) to be computed per species
implicit none
IF( eecolls .or. eicolls) THEN
IF(me .eq. 0) WRITE(*,*) 'Set electron grid ... '
IF(IFGKCOULOMB) THEN
call set_rows_e_4dim
ELSE
call set_rows_e_1dim
ENDIF
IF(me .eq. 0) WRITE(*,*) ' ... done '
ENDIF
IF(iicolls .or. iecolls) THEN
IF(me .eq. 0) WRITE(*,*)'Set ion grid ... '
IF( IFGKCOULOMB) THEN
call set_rows_i_4dim
ELSE
call set_rows_i_1dim
ENDIF
IF(me .eq. 0) WRITE(*,*) '... done'
ENDIF
!
end subroutine set_grid
!
!---------------------------------------------------
!
subroutine set_rows_i_1dim
! construct srumble rows for ion
implicit none
!
! locs. vars.
integer :: id_,ENDOF,irow
logical :: rows_s_file_exist
integer :: l,k
!
! Allocate scamble rows
call allocate_array(rows_i,1,numbi(Pmaxi,Jmaxi))
!
DO irow=1,numbi(Pmaxi,Jmaxi)
rows_i(irow) = irow
ENDDO
!
end subroutine set_rows_i_1dim
!
!---------------------------------------------------
!
subroutine set_rows_i_4dim
! construct srumble rows for ion for GK Coulomb
implicit none
!
! locs. vars.
integer :: id_,ENDOF,irow
integer :: nb_of_rows
logical :: rows_s_file_exist
integer :: l,k,ilbar,ij,in,inp
!
! compute the total number of rows to be distributed among the MPI tasks
IF( .not. GKC_USE_OPT) THEN
! Allocate scamble rows
nb_of_rows = numbi(Pmaxi,Jmaxi)*(npiFLR - niFLRs +1)*(niFLR - niFLRs + 1)*(JEmaxx+1)
call allocate_array(rows_i_4dim,1,nb_of_rows,1,4)
irow = 0
DO in=niFLRs,niFLR
DO inp=niFLRs,npiFLR
DO ilbar=1,numbi(Pmaxi,Jmaxi)
DO ij=0,JEmaxx
irow = irow +1
rows_i_4dim(irow,1) = ilbar
rows_i_4dim(irow,2) = ij
rows_i_4dim(irow,3 ) = in
rows_i_4dim(irow,4) = inp
ENDDO
ENDDO
ENDDO
ENDDO
ENDIF
IF( GKC_USE_OPT) THEN
nb_of_rows = numbi(Pmaxi,Jmaxi)*(npiFLR - niFLRs +1)*(JEmaxx+1)
call allocate_array(rows_i_4dim,1,nb_of_rows,1,4)
irow = 0
DO in=0, 0
DO inp=niFLRs,npiFLR
DO ilbar=1,numbi(Pmaxi,Jmaxi)
DO ij=0,JEmaxx
irow = irow +1
rows_i_4dim(irow,1) = ilbar
rows_i_4dim(irow,2) = ij
rows_i_4dim(irow,3 ) = in
rows_i_4dim(irow,4) = inp
ENDDO
ENDDO
ENDDO
ENDDO
ENDIF
!
IF(me .eq. 0) WRITE(*,*) ' Rows to compute = ',nb_of_rows
!
end subroutine set_rows_i_4dim
!
!---------------------------------------------------
!
subroutine set_rows_e_1dim
! construct srumble rows for electrons
implicit none
!
! locs. vars.
integer :: irow
!
! Allocate scamble rows
call allocate_array(rows_e,1,numbe(Pmaxe,Jmaxe))
!
DO irow=1,numbe(Pmaxe,Jmaxe)
rows_e(irow) = irow
ENDDO
!
!
end subroutine set_rows_e_1dim
!
!---------------------------------------------------
!
subroutine set_rows_e_4dim
! construct srumble rows for electrons for GK Coulomb
implicit none
!
! locs. vars.
integer :: id_,ENDOF,irow
integer :: nb_of_rows
logical :: rows_s_file_exist
integer :: l,k,ilbar,ij,in,inp
!
! compute the total number of rows to be distributed among the MPI tasks
IF( .not. GKC_USE_OPT) THEN
nb_of_rows = numbe(Pmaxe,Jmaxe)*(npeFLR - neFLRs +1)*(neFLR - neFLRs + 1)*(JEmaxx+1)
! Allocate scamble rows
call allocate_array(rows_e_4dim,1,nb_of_rows,1,4)
irow = 0
DO in=neFLRs,neFLR
DO inp=neFLRs, npeFLR
DO ilbar=1,numbi(Pmaxe,Jmaxe)
DO ij=0, JEmaxx
irow = irow +1
rows_e_4dim(irow,1) = ilbar
rows_e_4dim(irow,2) = ij
rows_e_4dim(irow,3 ) = in
rows_e_4dim(irow,4) = inp
ENDDO
ENDDO
ENDDO
ENDDO
ENDIF
IF( GKC_USE_OPT) THEN
nb_of_rows = numbe(Pmaxe,Jmaxe)*(npeFLR - neFLRs +1) *(JEmaxx+1)
call allocate_array(rows_e_4dim,1,nb_of_rows,1,4)
irow = 0
DO in=0, 0
DO inp=neFLRs, npeFLR
DO ilbar=1,numbi(Pmaxe,Jmaxe)
DO ij=0, JEmaxx
irow = irow +1
rows_e_4dim(irow,1) = ilbar
rows_e_4dim(irow,2) = ij
rows_e_4dim(irow,3 ) = in
rows_e_4dim(irow,4) = inp
ENDDO
ENDDO
ENDDO
ENDDO
ENDIF
IF(me .eq. 0) WRITE(*,*) ' Rows to compute = ',nb_of_rows
!
end subroutine set_rows_e_4dim
!
end module grid
c4science · Help