Homec4science
Diffusion FENNECS: Finite Element Non-Neutral Electron Cloud Simulator (geom_testpardiso) Browse

fennecs/src/chkrst.f90
5e6068e34819geom_testpardiso

Last Change
Raw File

chkrst.f90
View Options

SUBROUTINE chkrst(flag)
!
! Process checkpoint/restart file
!
USE basic
USE futils
USE beam
USE fields
IMPLICIT NONE
INTEGER, INTENT(in) :: flag
INTEGER, DIMENSION(:), ALLOCATABLE:: displs
INTEGER :: i, remainingparts
DOUBLE PRECISION:: old_msim, old_qsim, old_n0
! Only process 0 should save on file
!
! Local vars and arrays
!________________________________________________________________________________
!
SELECT CASE(flag)
!________________________________________________________________________________
! 1. Open and read restart file
!
CASE(0)
IF(mpirank .ne. 0) RETURN
CALL openf(rstfile, fidrst)
CALL getatt(fidrst, '/Basic', 'cstep', cstep)
CALL getatt(fidrst, '/Basic', 'time', time)
CALL getatt(fidrst, '/Basic', 'msim', old_msim)
CALL getatt(fidrst, '/Basic', 'qsim', old_qsim)
CALL getatt(fidrst, '/Basic', 'n0', old_n0)
qsim=old_qsim/old_n0*n0
msim=old_msim/old_n0*n0
CALL getatt(fidrst, '/Basic', 'V', V)
! Renormalize V
V=V*sqrt(n0/old_n0)**3
CALL getatt(fidrst, '/var0d', 'etot0', etot0)
CALL getatt(fidrst, '/var0d', 'epot', epot)
CALL getatt(fidrst, '/var0d', 'ekin', ekin)
CALL getatt(fidrst, '/var0d', 'etot', etot)
CALL getatt(fidrst, '/Parts', 'nplasma', nplasma)
CALL creat_parts(parts,nplasma)
CALL getarr(fidrst, '/Parts/Z', parts%Z)
CALL getarr(fidrst, '/Parts/R', parts%R)
! Renormalize R and Z
parts%R=parts%R*sqrt(n0/old_n0)
parts%Z=parts%Z*sqrt(n0/old_n0)
CALL getarr(fidrst, '/Parts/THET', parts%THET)
CALL getarr(fidrst, '/Parts/UZ', parts%UZ)
CALL getarr(fidrst, '/Parts/UR', parts%UR)
CALL getarr(fidrst, '/Parts/UTHET', parts%UTHET)
CALL getarr(fidrst, '/Parts/Zindex', parts%Zindex)
CALL getarr(fidrst, '/Parts/Rindex', parts%Rindex)
CALL getarr(fidrst, '/Parts/partindex', parts%partindex)
CALL getatt(fidrst, '/Parts', 'remainingparts', parts%Nptot)
CALL closef(fidrst)
WRITE(*,'(3x,a)') &
& "Reading from restart file "//TRIM(rstfile)//" completed!"
!________________________________________________________________________________
! 2. Create and write to restart file (DP reals)
!
CASE(1)
IF(mpisize .gt. 1) THEN
call collectparts
END IF
IF(mpirank .ne. 0) RETURN
IF( .NOT. nlsave ) RETURN
CALL mv2bk(rstfile)
CALL creatf(rstfile, fidrst, real_prec='d', desc='Restart file')
CALL creatg(fidrst, '/Basic', 'Basic data')
CALL attach(fidrst, '/Basic', 'cstep', cstep)
CALL attach(fidrst, '/Basic', 'time', time)
CALL attach(fidrst, '/Basic', 'jobnum', jobnum)
CALL attach(fidrst, '/Basic', 'qsim', qsim)
CALL attach(fidrst, '/Basic', 'msim', msim)
CALL attach(fidrst, '/Basic', 'n0', n0)
CALL attach(fidrst, '/Basic', 'V', V)
!
! 0D variables
!
CALL creatg(fidrst, '/var0d', '0D variables')
CALL attach(fidrst, '/var0d','etot0', etot0)
CALL attach(fidrst, '/var0d','epot', epot)
CALL attach(fidrst, '/var0d','ekin', ekin)
CALL attach(fidrst, '/var0d','etot', etot)
!
! Parts
!
CALL creatg(fidrst, '/Parts', 'Particles data')
CALL attach(fidrst, '/Parts', 'nplasma', nplasma)
IF(mpisize .gt. 1) THEN
remainingparts=sum(Nplocs_all)
ELSE
remainingparts=parts%Nptot
END IF
CALL attach(fidrst, '/Parts', 'remainingparts', remainingparts)
CALL putarr(fidrst, '/Parts/Z', parts%Z(1:remainingparts))
CALL putarr(fidrst, '/Parts/R', parts%R(1:remainingparts))
CALL putarr(fidrst, '/Parts/THET', parts%THET(1:remainingparts))
CALL putarr(fidrst, '/Parts/UZ', parts%UZ(1:remainingparts))
CALL putarr(fidrst, '/Parts/UR', parts%UR(1:remainingparts))
CALL putarr(fidrst, '/Parts/UTHET', parts%UTHET(1:remainingparts))
CALL putarr(fidrst, '/Parts/Zindex', parts%Zindex(1:remainingparts))
CALL putarr(fidrst, '/Parts/Rindex', parts%Rindex(1:remainingparts))
CALL putarr(fidrst, '/Parts/partindex', parts%partindex(1:remainingparts))
!
! Fields
!
CALL closef(fidrst)
WRITE(*,'(3x,a)') &
& "Writing to restart file "//TRIM(rstfile)//" completed!"
!
END SELECT
!
END SUBROUTINE chkrst
c4science · Help