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

fennecs/src/chkrst.f90
b32797dc7b88geom_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
! 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', msim)
CALL getatt(fidrst, '/Basic', 'qsim', qsim)
CALL getatt(fidrst, '/Basic', 'V', V)
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)
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 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( .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', '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)
CALL putarr(fidrst, '/Parts/Z', parts%Z)
CALL putarr(fidrst, '/Parts/R', parts%R)
CALL putarr(fidrst, '/Parts/UZ', parts%UZ)
CALL putarr(fidrst, '/Parts/UR', parts%UR)
CALL putarr(fidrst, '/Parts/UTHET', parts%UTHET)
CALL putarr(fidrst, '/Parts/Zindex', parts%Zindex)
CALL putarr(fidrst, '/Parts/Rindex', parts%Rindex)
!
! Fields
!
CALL closef(fidrst)
WRITE(*,'(3x,a)') &
& "Writing to restart file "//TRIM(rstfile)//" completed!"
!
END SELECT
!
END SUBROUTINE chkrst
c4science · Help