Homec4science
Diffusion FENNECS lite (master) Browse

fennecslite/src/chkrst.f90
2e0aa43721aamaster

Last Change
Raw File

chkrst.f90
View Options

SUBROUTINE chkrst(flag)
!
! Process checkpoint/restart file
!
USE basic
USE futils
USE beam
USE fields
USE constants, ONLY: elchar, me
Use psupply
use filemanip
USE mumps_bsplines
IMPLICIT NONE
INTEGER, INTENT(in) :: flag
INTEGER :: remainingparts
REAL(kind=db):: old_msim, old_qsim, old_n0
INTEGER:: partsrank, partsdim(1), i, err
REAL(kind=db), ALLOCATABLE:: charges(:), weights(:), masses(:)
CHARACTER(len=64):: group
CHARACTER(len=2):: specieindex
real(kind=db):: old_rnorm, old_tnorm, qmratioscale
real(kind=db):: prev_bias
INTEGER:: logical_val, id
! Only process 0 should save on file
SELECT CASE(flag)
!________________________________________________________________________________
! 1. Open and read restart file
CASE(0)
CALL openf(rstfile, fidrst,'r',real_prec='d')
CALL getatt(fidrst, '/Basic', 'cstep', cstep)
CALL getatt(fidrst, '/Basic', 'time', time)
CALL getatt(fidrst, '/Basic', 'n0', old_n0)
IF(isgroup(fidrst,'/Basic/norm')) THEN
CALL getatt(fidrst, '/Basic/norm', 'rnorm', old_rnorm)
CALL getatt(fidrst, '/Basic/norm', 'tnorm', old_tnorm)
ELSE
old_rnorm=rnorm
old_tnorm=tnorm
END IF
IF(isdataset(fidrst,'/Parts/charges')) THEN
! If we have multiple saved species we need to load all of them
CALL getatt(fidrst,'/Parts','nbspecies',nbspecies)
ALLOCATE(charges(nbspecies),masses(nbspecies),weights(nbspecies))
ALLOCATE(partslist(nbspecies+nbaddtestspecies))
ALLOCATE(partslist_towrite(nbspecies+nbaddtestspecies))
CALL getarr(fidrst, '/Parts/charges', charges)
CALL getarr(fidrst, '/Parts/masses', masses)
CALL getarr(fidrst, '/Parts/weights', weights)
weights(1)=weights(1)/old_n0*n0
ELSE
! If we have an old restart file, we load only the electrons
CALL getatt(fidrst, '/Basic', 'msim', old_msim)
CALL getatt(fidrst, '/Basic', 'qsim', old_qsim)
qsim=old_qsim/old_n0*n0
msim=old_msim/old_n0*n0
nbspecies=1
ALLOCATE(charges(nbspecies),masses(nbspecies),weights(nbspecies))
ALLOCATE(partslist(nbspecies))
ALLOCATE(partslist_towrite(nbspecies))
charges(1)=sign(elchar,qsim)
weights(1)=msim/me
masses(1)=me
END IF
IF(newres) THEN
weights=weights*weights_scale
END IF
IF(n0.ne.old_n0) cstep=0
CALL getatt(fidrst, '/Parts', 'nplasma', nplasma)
CALL getatt(fidrst, '/Parts', 'remainingparts', remainingparts)
IF(remainingparts .gt. 0) THEN
CALL getdims(fidrst, '/Parts/Z', partsrank, partsdim)
ELSE
partsdim=0
END IF
IF(samplefactor .gt. 1 ) THEN ! We increase the number of macro particles
CALL creat_parts(partslist(1),max(remainingparts*samplefactor,partsdim(1)))
ELSE
CALL creat_parts(partslist(1),partsdim(1))
CALL creat_parts(partslist_towrite(1),partsdim(1))
END IF
partslist(1)%q=charges(1)
partslist(1)%m=masses(1)
partslist(1)%weight=weights(1)
partslist(1)%qmRatio=charges(1)/masses(1)
err=0
CALL getatt(fidrst, 'Parts', 'qmratioscale', qmratioscale, err)
IF(err .ge.0) partslist(1)%qmRatio=partslist(1)%qmRatio*qmratioscale
IF(remainingparts .gt. 0) THEN
CALL getarr(fidrst, '/Parts/Z', partslist(1)%pos(3,:))
CALL getarr(fidrst, '/Parts/R', partslist(1)%pos(1,:))
! Renormalize R and Z
partslist(1)%pos(1,:)=partslist(1)%pos(1,:)*old_rnorm/rnorm
partslist(1)%pos(3,:)=partslist(1)%pos(3,:)*old_rnorm/rnorm
CALL getarr(fidrst, '/Parts/THET', partslist(1)%pos(2,:))
CALL getarr(fidrst, '/Parts/Upar', partslist(1)%Upar(:))
CALL getarr(fidrst, '/Parts/UZ', partslist(1)%U(3,:))
CALL getarr(fidrst, '/Parts/UR', partslist(1)%U(1,:))
CALL getarr(fidrst, '/Parts/UTHET', partslist(1)%U(2,:))
CALL getarr(fidrst, '/Parts/UZ', partslist(1)%Uold(3,:))
CALL getarr(fidrst, '/Parts/UR', partslist(1)%Uold(1,:))
CALL getarr(fidrst, '/Parts/UTHET', partslist(1)%Uold(2,:))
CALL getarr(fidrst, '/Parts/GAMMA', partslist(1)%Gamma)
CALL getarr(fidrst, '/Parts/Zindex', partslist(1)%cellindex(3,:))
CALL getarr(fidrst, '/Parts/Rindex', partslist(1)%cellindex(1,:))
CALL getarr(fidrst, '/Parts/partindex', partslist(1)%partindex)
END IF
partslist(1)%Nploc=remainingparts
partslist(1)%Nptot=partslist(1)%Nploc
partslist(1)%Newindex=maxval(partslist(1)%partindex)
CALL getatt(fidrst, trim('/Parts'), 'zero_vel', logical_val,err)
IF(err .ge. 0) THEN
IF(logical_val.gt.0) THEN
partslist(1)%zero_vel =.true.
ELSE
partslist(1)%zero_vel =.false.
END IF
END IF
CALL getatt(fidrst, trim('/Parts'), 'see_id',id, err)
IF(err .ge. 0) THEN
partslist(1)%see_id = id
END IF
CALL getatt(fidrst, trim('/Parts'), 'iiee_id',id, err)
IF(err .ge. 0) THEN
partslist(1)%iiee_id = id
END IF
CALL getatt(fidrst, trim('/Parts'), 'neuttype_id',id, err)
IF(err .ge. 0) THEN
partslist(1)%neuttype_id = id
END IF
CALL getatt(fidrst, trim('/Parts'), 'material_id',id, err)
IF(err .ge. 0) THEN
partslist(1)%material_id = id
END IF
WRITE(*,*) "Read ", partslist(1)%Nploc, " particles out of ", remainingparts
IF(nbspecies .gt. 1) THEN
DO i=2,nbspecies
WRITE(group,'(a,i2)')'/Parts/',i
WRITE(specieindex,'(i2)') i
partsdim=0
CALL getatt(fidrst, trim(group), 'remainingparts', remainingparts)
IF(remainingparts .gt. 0) THEN
CALL getdims(fidrst, trim(group) // '/Z', partsrank, partsdim)
ELSE
partsdim=0
END IF
IF(partsdim(1) .gt. remainingparts) THEN
CALL creat_parts(partslist(i),partsdim(1))
CALL creat_parts(partslist_towrite(i),partsdim(1))
partslist(i)%Nploc=remainingparts
ELSE
CALL creat_parts(partslist(i),max(500000,remainingparts))
CALL creat_parts(partslist_towrite(i),max(500000,remainingparts))
ENDIF
partslist(i)%q=charges(i)
partslist(i)%m=masses(i)
partslist(i)%weight=weights(i)
partslist(i)%qmRatio=charges(i)/masses(i)
err=0
CALL getatt(fidrst, trim(group), 'qmratioscale', qmratioscale, err)
IF(err .ge.0) partslist(i)%qmRatio=partslist(i)%qmRatio*qmratioscale
partslist(i)%Nptot=remainingparts
partslist(i)%Nploc=remainingparts
partslist(i)%is_test =.false.
partslist(i)%is_field =.false.
partslist(i)%calc_moments =.false.
err=0
CALL getatt(fidrst, trim(group), 'is_test', logical_val,err)
IF(err .ge. 0) THEN
IF(logical_val.gt.0) partslist(i)%is_test =.true.
END IF
err=0
CALL getatt(fidrst, trim(group), 'is_field', logical_val,err)
IF(err .ge. 0) THEN
IF(logical_val .gt. 0) partslist(i)%is_field =.true.
END IF
err=0
CALL getatt(fidrst, trim(group), 'calc_moments', logical_val,err)
IF(err .ge. 0) THEN
IF(logical_val .gt. 0) partslist(i)%calc_moments =.true.
END IF
CALL getatt(fidrst, trim(group), 'zero_vel', logical_val,err)
IF(err .ge. 0) THEN
IF(logical_val.gt.0) THEN
partslist(i)%zero_vel =.true.
ELSE
partslist(i)%zero_vel =.false.
END IF
END IF
CALL getatt(fidrst, trim(group), 'see_id',id, err)
IF(err .ge. 0) THEN
partslist(i)%see_id = id
END IF
CALL getatt(fidrst, trim(group), 'iiee_id',id, err)
IF(err .ge. 0) THEN
partslist(i)%iiee_id = id
END IF
CALL getatt(fidrst, trim(group), 'neuttype_id',id, err)
IF(err .ge. 0) THEN
partslist(i)%neuttype_id = id
END IF
CALL getatt(fidrst, trim(group), 'material_id',id, err)
IF(err .ge. 0) THEN
partslist(i)%material_id = id
END IF
IF(partslist(i)%Nptot .gt. 0) THEN
CALL getarr(fidrst, trim(group) // '/Z', partslist(i)%pos(3,:))
CALL getarr(fidrst, trim(group) // '/R', partslist(i)%pos(1,:))
CALL getarr(fidrst, trim(group) // '/THET', partslist(i)%pos(2,:))
CALL getarr(fidrst, trim(group) // '/UZ', partslist(i)%U(3,:))
CALL getarr(fidrst, trim(group) // '/UR', partslist(i)%U(1,:))
CALL getarr(fidrst, trim(group) // '/UTHET', partslist(i)%U(2,:))
CALL getarr(fidrst, trim(group) // '/UZ', partslist(i)%Uold(3,:))
CALL getarr(fidrst, trim(group) // '/UR', partslist(i)%Uold(1,:))
CALL getarr(fidrst, trim(group) // '/UTHET', partslist(i)%Uold(2,:))
CALL getarr(fidrst, trim(group) // '/GAMMA', partslist(i)%Gamma)
CALL getarr(fidrst, trim(group) // '/Zindex', partslist(i)%cellindex(3,:))
CALL getarr(fidrst, trim(group) // '/Rindex', partslist(i)%cellindex(1,:))
CALL getarr(fidrst, trim(group) // '/partindex', partslist(i)%partindex)
partslist(i)%pos(1,:)=partslist(i)%pos(1,:)*old_rnorm/rnorm
partslist(i)%pos(3,:)=partslist(i)%pos(3,:)*old_rnorm/rnorm
partslist(i)%Newindex=maxval(partslist(i)%partindex)
END IF
END DO
END IF
IF(isgroup(fidrst,'/psupply')) THEN
CALL getatt(fidrst,'/psupply', 'active', logical_val)
IF(logical_val .gt. 0) THEN
CALL getatt(fidrst,'/psupply', 'bias', prev_bias)
the_ps%active=.true.
the_ps%bias=prev_bias/phinorm
ELSE
the_ps%active=.false.
END IF
END IF
CALL closef(fidrst)
IF(samplefactor .gt. 1) THEN ! We increase the number of macro particles
CALL upsample(partslist(1), samplefactor)
END IF
IF(mpirank.eq. 0) CALL mv2bk(rstfile, '_start')
WRITE(*,'(3x,a)') "Reading from restart file "//TRIM(rstfile)//" completed!"
!________________________________________________________________________________
! 2. Create and write to restart file (DP reals)
CASE(1)
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', partslist(1)%q*partslist(1)%weight)
CALL attach(fidrst, '/Basic', 'msim', partslist(1)%m*partslist(1)%weight)
CALL attach(fidrst, '/Basic', 'n0', n0)
CALL creatg(fidrst, '/Basic/norm', 'Normalisation quantities')
CALL attach(fidrst, '/Basic/norm', 'rnorm', 1.0)
CALL attach(fidrst, '/Basic/norm', 'bnorm', bnorm)
CALL attach(fidrst, '/Basic/norm', 'enorm', enorm)
CALL attach(fidrst, '/Basic/norm', 'tnorm', tnorm)
CALL attach(fidrst, '/Basic/norm', 'phinorm', phinorm)
! 0D variables
CALL creatg(fidrst, '/var0d', '0D variables')
! Parts
CALL creatg(fidrst, '/Parts', 'Particles data')
CALL attach(fidrst, '/Parts', 'nplasma', nplasma)
nbspecies=size(partslist,1)
CALL attach(fidrst, '/Parts', 'nbspecies', nbspecies)
ALLOCATE(charges(nbspecies),masses(nbspecies),weights(nbspecies))
DO i=1,nbspecies
charges(i) = partslist(i)%q
masses(i) = partslist(i)%m
weights(i) = partslist(i)%weight
END DO
CALL putarr(fidrst, '/Parts/charges', charges)
CALL putarr(fidrst, '/Parts/masses', masses )
CALL putarr(fidrst, '/Parts/weights', weights)
IF(mpisize .gt. 1) THEN
remainingparts=sum(Nplocs_all)
ELSE
remainingparts=partslist(1)%Nploc
END IF
CALL attach(fidrst, '/Parts', 'remainingparts', remainingparts)
CALL attach(fidrst, '/Parts', 'qmratioscale',partslist(1)%qmRatio/(partslist(1)%q/partslist(1)%m))
CALL putarr(fidrst, '/Parts/Z', partslist_towrite(1)%pos(3,:)*rnorm)
CALL putarr(fidrst, '/Parts/R', partslist_towrite(1)%pos(1,:)*rnorm)
CALL putarr(fidrst, '/Parts/THET', partslist_towrite(1)%pos(2,:))
CALL putarr(fidrst, '/Parts/Upar', partslist_towrite(1)%Upar(:))
CALL putarr(fidrst, '/Parts/UZ', partslist_towrite(1)%U(3,:))
CALL putarr(fidrst, '/Parts/UR', partslist_towrite(1)%U(1,:))
CALL putarr(fidrst, '/Parts/UTHET', partslist_towrite(1)%U(2,:))
CALL putarr(fidrst, '/Parts/GAMMA', partslist_towrite(1)%Gamma)
CALL putarr(fidrst, '/Parts/Zindex', partslist_towrite(1)%cellindex(3,:))
CALL putarr(fidrst, '/Parts/Rindex', partslist_towrite(1)%cellindex(3,:))
CALL putarr(fidrst, '/Parts/partindex', partslist_towrite(1)%partindex)
partslist(1)%is_field=.true.
partslist(1)%is_test=.false.
partslist(1)%calc_moments=.true.
CALL attach(fidrst, '/Parts', 'is_field', 1)
CALL attach(fidrst, '/Parts', 'calc_moments', 1)
CALL attach(fidrst, '/Parts', 'is_test', 0)
IF(partslist(1)%zero_vel) THEN
CALL attach(fidrst, trim('/Parts'), 'zero_vel', 1)
ELSE
CALL attach(fidrst, trim('/Parts'), 'zero_vel', 0)
END IF
CALL attach(fidrst, trim('/Parts'), 'see_id', partslist(1)%see_id)
CALL attach(fidrst, trim('/Parts'), 'iiee_id', partslist(1)%iiee_id)
CALL attach(fidrst, trim('/Parts'), 'neuttype_id', partslist(1)%neuttype_id)
CALL attach(fidrst, trim('/Parts'), 'material_id', partslist(1)%material_id)
IF(nbspecies .gt. 1) THEN
DO i=2,nbspecies
WRITE(group,'(a,i2)')'/Parts/',i
WRITE(specieindex,'(i2)') i
CALL creatg(fidrst, trim(group), 'Particles ' // specieindex// ' data')
CALL attach(fidrst, trim(group), 'qmratioscale', partslist(i)%qmRatio/(partslist(i)%q/partslist(i)%m))
CALL attach(fidrst, trim(group), 'remainingparts', partslist(i)%Nptot)
IF(partslist(i)%is_test) THEN
CALL attach(fidrst, trim(group), 'is_test', 1)
ELSE
CALL attach(fidrst, trim(group), 'is_test', 0)
END IF
IF(partslist(i)%is_field) THEN
CALL attach(fidrst, trim(group), 'is_field', 1)
ELSE
CALL attach(fidrst, trim(group), 'is_field', 0)
END IF
IF(partslist(i)%calc_moments) THEN
CALL attach(fidrst, trim(group), 'calc_moments', 1)
ELSE
CALL attach(fidrst, trim(group), 'calc_moments', 0)
END IF
IF(partslist(i)%zero_vel) THEN
CALL attach(fidrst, trim(group), 'zero_vel', 1)
ELSE
CALL attach(fidrst, trim(group), 'zero_vel', 0)
END IF
CALL attach(fidrst, trim(group), 'see_id', partslist(i)%see_id)
CALL attach(fidrst, trim(group), 'iiee_id', partslist(i)%iiee_id)
CALL attach(fidrst, trim(group), 'neuttype_id', partslist(i)%neuttype_id)
CALL attach(fidrst, trim(group), 'material_id', partslist(i)%material_id)
IF(partslist(i)%Nptot .gt. 0) THEN
CALL putarr(fidrst, trim(group) // '/Z', partslist_towrite(i)%pos(3,1:partslist(i)%Nptot)*rnorm)
CALL putarr(fidrst, trim(group) // '/R', partslist_towrite(i)%pos(1,1:partslist(i)%Nptot)*rnorm)
CALL putarr(fidrst, trim(group) // '/THET', partslist_towrite(i)%pos(2,1:partslist(i)%Nptot))
CALL putarr(fidrst, trim(group) // '/UZ', partslist_towrite(i)%U(3,1:partslist(i)%Nptot))
CALL putarr(fidrst, trim(group) // '/UR', partslist_towrite(i)%U(1,1:partslist(i)%Nptot))
CALL putarr(fidrst, trim(group) // '/UTHET', partslist_towrite(i)%U(2,1:partslist(i)%Nptot))
CALL putarr(fidrst, trim(group) // '/GAMMA', partslist_towrite(i)%Gamma(1:partslist(i)%Nptot))
CALL putarr(fidrst, trim(group) // '/Zindex', partslist_towrite(i)%cellindex(3,1:partslist(i)%Nptot))
CALL putarr(fidrst, trim(group) // '/Rindex', partslist_towrite(i)%cellindex(1,1:partslist(i)%Nptot))
CALL putarr(fidrst, trim(group) // '/partindex', partslist_towrite(i)%partindex(1:partslist(i)%Nptot))
END IF
END DO
END IF
! Power supply status
CALL creatg(fidrst, '/psupply', 'Power supply status and values')
IF(the_ps%active) THEN
CALL attach(fidrst, '/psupply','active', 1)
ELSE
CALL attach(fidrst, '/psupply','active', 0)
END IF
CALL attach(fidrst, '/psupply','bias', the_ps%bias*phinorm)
CALL closef(fidrst)
WRITE(*,'(3x,a)') "Writing to restart file "//TRIM(rstfile)//" completed!"
END SELECT
END SUBROUTINE chkrst
c4science · Help