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

fennecs/src/chkrst.f90
6bc6183feba0geom_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 :: remainingparts
REAL(kind=db):: old_msim, old_qsim, old_n0
INTEGER:: partsrank, partsdim(1)
! 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,'r','d')
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', loc_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 getatt(fidrst, '/Parts', 'remainingparts', remainingparts)
IF (samplefactor .gt. 1 ) THEN ! We increase the number of macro particles
CALL creat_parts(parts,remainingparts*samplefactor)
ELSE
CALL getdims(fidrst, '/Parts/Z', partsrank, partsdim)
CALL creat_parts(parts,partsdim(1))
END IF
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)
WRITE(*,*) "Read ", parts%Nptot, " particles out of ", nplasma
IF(isdataset(fidrst,'/Parts/fluidur')) THEN
CALL getarr(fidrst, '/Parts/GAMMA', parts%Gamma)
!CALL getarr(fidrst, '/Parts/fluidur', moments(2,:))
!CALL getarr(fidrst, '/Parts/fluiduthet', moments(3,:))
!CALL getarr(fidrst, '/Parts/fluiduz', moments(4,:))
IF(temprescale .gt. 0) THEN
CALL Temp_rescale
END IF
END IF
CALL closef(fidrst)
IF(samplefactor .gt. 1) THEN ! We increase the number of macro particles
CALL upsample(samplefactor)
END IF
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/GAMMA', parts%Gamma(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))
CALL putarr(fidrst, '/Parts/fluidur', moments(2,:))
CALL putarr(fidrst, '/Parts/fluiduthet', moments(3,:))
CALL putarr(fidrst, '/Parts/fluiduz', moments(4,:))
!
! Fields
!
CALL closef(fidrst)
WRITE(*,'(3x,a)') &
& "Writing to restart file "//TRIM(rstfile)//" completed!"
!
END SELECT
!
END SUBROUTINE chkrst
c4science · Help