Homec4science
Diffusion COSOlver (improved_sugama) Browse

cosolver/src/ei_colls_mod.f90
cd1286e2d410improved_sugama

Last Change
Raw File

ei_colls_mod.f90
View Options

MODULE ei_colls
! electron-ion collisions module
USE PREC_CONST
USE MODEL
USE BASIC
USE BASIC_MPI
use grid
USE ARRAY, only : CeiTdp,CeiFdp,rCeiT,rCeiF
USE OPERATOR_MODEL
USE OUTPUT
USE restart
IMPLICIT NONE
PROCEDURE(eval_CeiT), POINTER :: evaluate_CeiT_collisional_matrix => null()
PROCEDURE(eval_CeiF), POINTER :: evaluate_CeiF_collisional_matrix => null()
ABSTRACT INTERFACE
SUBROUTINE eval_CeiT
! Empty
END SUBROUTINE eval_CeiT
END INTERFACE
ABSTRACT INTERFACE
SUBROUTINE eval_CeiF
! Empty
END SUBROUTINE eval_CeiF
END INTERFACE
CONTAINS
!
!------------------------------------------------
!
SUBROUTINE set_ei_colls
! Set electron-ions basic/model
IMPLICIT NONE
!
CALL set_model('ei')
CALL set_basic('ei')
!
END SUBROUTINE set_ei_colls
!
!------------------------------------------------
!
! TEST COMPONENT
!
SUBROUTINE set_CeiT_collisional_matrix
! Evaluate the electron-ion test collisional matrix
IMPLICIT NONE
!
!
IF( ETEST .ne. 0 ) THEN
IF(IFCOULOMB .and. GKE) THEN
! To complete
evaluate_CeiT_collisional_matrix => CeiT_gk_coulomb
!
call load_gk_coulomb_coeffs('eiT')
!
ELSE
!
evaluate_CeiT_collisional_matrix => CeiT_dflt
!
ENDIF
ELSE
evaluate_CeiT_collisional_matrix => CeiT_zeros_dflt
ENDIF
!
END SUBROUTINE set_CeiT_collisional_matrix
!
!----------------------------------------------------
!
SUBROUTINE CeiT_dflt
! Evaluate the electron-ion collisional matrix
! Computes the test collisional matrix.
!
!
IMPLICIT NONE
!
! Local variables
INTEGER :: irow,irow_tmp
INTEGER, DIMENSION(2) :: lk
INTEGER :: istep, lbar_start
! Allocate restart coll mat and restart indices
IF(restart_from) CALL get_restart(CeiTdp)
lbar_start = lbare_s_l
IF(restart_from) lbar_start = row_rst_l +1
istep=1
!
! Test-part
IF (ETEST .ne. 0) THEN
DO irow=lbar_start,lbare_e_l
! Get Hermite-Laguerre indices
irow_tmp = rows_e(irow)
lk = invnumbe(irow_tmp)
rCeiT(:)= TO_FM(0)
write(*,'(a,1x,i3,a,i3,a,i3,a,i3)') 'CeiT: ',lk(1),'/',Pmaxe,' ,',lk(2),'/',Jmaxe
CALL COLLISIONTE(lk(1),lk(2),rCeiT)
CeiTdp(irow,:) = TO_DP(rCeiT(:))
! save current state
istep = istep +1
IF(ifrestart .and. istep .gt. nsave_ei) THEN
call restart_out(CeiTdp,(/irow/))
istep = 1
ENDIF
ENDDO
ENDIF
!
IF(ifrestart) call restart_out(CeiTdp,(/irow/))
!
END SUBROUTINE CeiT_dflt
!
!--------------------------------------------------------------
!
SUBROUTINE CeiT_gk_coulomb
! Evaluate electron-ion test GK Coulomb
use restart
use gkcoulomb
implicit none
INTEGER :: l,k,irow
INTEGER :: j,n,np,istep
INTEGER :: lbar_start
INTEGER, DIMENSION(4) :: lbarjnnp
INTEGER, DIMENSION(2) :: lk
!
!
! restart
IF(restart_from) CALL get_restart( CeiTdp )
!
! initialize indices
lbar_start = lbare_s_l
IF(restart_from) lbar_start = row_rst_l +1
istep=1
!
DO irow=lbar_start,lbare_e_l
lbarjnnp = rows_e_4dim(irow,1:4)
! Hermite-Laguerre indices
lk = invnumbe(lbarjnnp(1))
j = lbarjnnp(2)
n = lbarjnnp(3)
np = lbarjnnp(4)
write(*,*) 'CeiT:', irow,'/',lbare_e_l
rCeiT(:) = ZEROFM
CALL coulombTGK(lk(1),lk(2),j,n,np,rCeiT)
CeiTdp(irow,:) = TO_DP(rCeiT(:))
! save current state
istep = istep +1
IF(ifrestart .and. istep .gt. nsave_ei) THEN
! flush current state every nsave_ei.
call restart_out(CeiTdp,(/irow/))
istep = 1
ENDIF
ENDDO
! flush out the final state
IF(ifrestart) call restart_out(CeiTdp,(/irow/))
!
END SUBROUTINE CeiT_gk_Coulomb
!
!------------------------------------------------
!
SUBROUTINE CeiT_zeros_dflt
! Evaluate the electron-ion collisional matrix
! Computes the test collisional matrix.
!
!
IMPLICIT NONE
!
! Local variables
INTEGER :: irow,irow_tmp
INTEGER, DIMENSION(2) :: lk
!
! Test-part
IF (ETEST .ne. 0) THEN
DO irow=lbare_s_l,lbare_e_l
! Get Hermite-Laguerre indices
irow_tmp = rows_e(irow)
lk = invnumbe(irow_tmp)
rCeiT(:)= TO_FM(0)
write(*,*) 'CeiT: ',lk(1),'/',Pmaxe,' ,',lk(2),'/',Jmaxe
CeiTdp(irow,:) = TO_DP(rCeiT(:))
ENDDO
ENDIF
!
!
END SUBROUTINE CeiT_zeros_dflt
!
!--------------------------------------------------------------
!
! FIELD COMPONENT
!
SUBROUTINE set_CeiF_collisional_matrix
! Evaluate the electron-ion collisional matrix
IMPLICIT NONE
!
!
IF (EBACK .ne. 0 ) THEN
IF(IFCOULOMB .and. GKE) THEN
evaluate_CeiF_collisional_matrix => CeiF_gk_coulomb
!
call load_gk_coulomb_coeffs('eiF')
!
ELSE
evaluate_CeiF_collisional_matrix =>CeiF_dflt
!
ENDIF
ELSE
evaluate_CeiF_collisional_matrix => CeiF_zeros_dflt
ENDIF
!
END SUBROUTINE set_CeiF_collisional_matrix
!
!----------------------------------------------------
SUBROUTINE CeiF_dflt
! Evaluate the electron-ion collisional matrix
! Computes the back-reaction collisional matrix.
!
!
IMPLICIT NONE
!
INTEGER :: irow,irow_tmp, lbar_start, istep
INTEGER, DIMENSION(2) :: lk
! Allocate restart coll mat and restart indices
IF(restart_from) CALL get_restart(CeiFdp)
lbar_start = lbare_s_l
IF(restart_from) lbar_start = row_rst_l +1
istep=1
! Back-Reaction
IF (EBACK .ne. 0) THEN
DO irow=lbar_start,lbare_e_l
! Get Hermite-Laguerre indices
irow_tmp = rows_e(irow)
lk = invnumbe(irow_tmp)
write(*,'(a,1x,i3,a,i3,a,i3,a,i3)') 'CeiF: ',lk(1),'/',Pmaxe,' ,',lk(2),'/',Jmaxe
rCeiF(:) = TO_FM(0)
CALL COLLISIONFE(lk(1),lk(2),rCeiF)
CeiFdp(irow,:) = TO_DP(rCeiF(:))
! save current state
istep = istep +1
IF(ifrestart .and. istep .gt. nsave_ei) THEN
call restart_out(CeiFdp,(/irow/))
istep = 1
ENDIF
ENDDO
ENDIF
!
!
IF(ifrestart) call restart_out(CeiFdp,(/irow/))
!
END SUBROUTINE CeiF_dflt
!
!--------------------------------------------
!
SUBROUTINE CeiF_gk_coulomb
! Evaluate electron-ion field GK Coulomb
use restart
use gkcoulomb
implicit none
INTEGER :: l,k,irow
INTEGER :: j,n,np,istep
INTEGER :: lbar_start
INTEGER, DIMENSION(4) :: lbarjnnp
INTEGER, DIMENSION(2) :: lk
!
!
! restart
IF(restart_from) CALL get_restart(CeiFdp)
!
! initialize indices
lbar_start = lbare_s_l
IF(restart_from) lbar_start = row_rst_l +1
istep=1
!
DO irow=lbar_start,lbare_e_l
lbarjnnp = rows_e_4dim(irow,1:4)
! Hermite-Laguerre indices
lk = invnumbe(lbarjnnp(1))
j = lbarjnnp(2)
n = lbarjnnp(3)
np = lbarjnnp(4)
write(*,*) 'CeiF:', irow,'/',lbare_e_l
rCeiF(:) = ZEROFM
CALL coulombFGK(lk(1),lk(2),j,n,np,rCeiF)
CeiFdp(irow,:) = TO_DP(rCeiF(:))
! save current state
istep = istep +1
IF(ifrestart .and. istep .gt. nsave_ei) THEN
! flush current state every nsave_ei.
call restart_out(CeiFdp,(/irow/))
istep = 1
ENDIF
ENDDO
! flush out the final state
IF(ifrestart) call restart_out(CeiFdp,(/irow/))
!
END SUBROUTINE CeiF_gk_Coulomb
!
!--------------------------------------------------
!
SUBROUTINE CeiF_zeros_dflt
! Evaluate the electron-ion collisional matrix
! Computes the back-reaction collisional matrix.
!
!
IMPLICIT NONE
!
INTEGER :: irow,irow_tmp
INTEGER, DIMENSION(2) :: lk
! Back-Reaction
IF (EBACK .ne. 0) THEN
DO irow=lbare_s_l,lbare_e_l
! Get Hermite-Laguerre indices
irow_tmp = rows_e(irow)
lk = invnumbe(irow_tmp)
write(*,*) 'CeiF: ',lk(1),'/',Pmaxe,' ,',lk(2),'/',Jmaxe
rCeiF(:) = TO_FM(0)
CeiFdp(irow,:) = TO_DP(rCeiF(:))
ENDDO
ENDIF
!
!
END SUBROUTINE CeiF_zeros_dflt
!
END MODULE ei_colls
c4science · Help