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

fennecs/src/resume.f90
35553d3dbae3geom_testpardiso

Last Change
Raw File

resume.f90
View Options

SUBROUTINE resume
!
! Resume from previous run
!
Use beam
Use basic
Use fields
Use sort
Use maxwellsource
IMPLICIT NONE
!
! Local vars and arrays
INTEGER, DIMENSION(:), ALLOCATABLE:: displs,sendcounts
INTEGER:: i
!________________________________________________________________________________
WRITE(*,'(a)') ' Resume from previous run'
!________________________________________________________________________________
!
! Open and read initial conditions from restart file
CALL chkrst(0)
IF (newres) THEN
cstep=0
time=0
END IF
IF(mpisize .gt. 1) THEN
ALLOCATE(sendcounts(0:mpisize-1), displs(0:mpisize-1))
IF(mpirank.eq.0) THEN
CALL bound
CALL localisation
CALL quicksortparts(1,parts%Nptot)
CALL distribpartsonprocs
CALL MPI_Bcast(size(parts%Z),1,MPI_INTEGER,0,MPI_COMM_WORLD,ierr)
WRITE(*,*) "nplocs_all: ", Nplocs_all
CALL MPI_Scatter(Nplocs_all,1,MPI_INTEGER,MPI_IN_PLACE, 1,MPI_INTEGER,0,MPI_COMM_WORLD,ierr)
DO i=0,mpisize-1
displs(i)=i
sendcounts(i)=2
END DO
CALL MPI_Scatterv(Zbounds,sendcounts, displs,MPI_INTEGER,MPI_IN_PLACE,2,MPI_INTEGER,0,MPI_COMM_WORLD,ierr)
displs(0)=0
sendcounts(0)=Nplocs_all(0)
DO i=1,mpisize-1
displs(i)=displs(i-1)+Nplocs_all(i-1)
sendcounts(i)=Nplocs_all(i)
END DO
CALL MPI_BARRIER(MPI_COMM_WORLD, ierr)
CALL MPI_Scatterv(parts%R,sendcounts, displs,MPI_DOUBLE_PRECISION,MPI_IN_PLACE,Nplocs_all(mpirank),MPI_DOUBLE_PRECISION,0,MPI_COMM_WORLD,ierr)
CALL MPI_Scatterv(parts%Z,sendcounts, displs,MPI_DOUBLE_PRECISION,MPI_IN_PLACE,Nplocs_all(mpirank),MPI_DOUBLE_PRECISION,0,MPI_COMM_WORLD,ierr)
CALL MPI_Scatterv(parts%THET,sendcounts, displs,MPI_DOUBLE_PRECISION,MPI_IN_PLACE,Nplocs_all(mpirank),MPI_DOUBLE_PRECISION,0,MPI_COMM_WORLD,ierr)
CALL MPI_Scatterv(parts%UZ,sendcounts, displs,MPI_DOUBLE_PRECISION,MPI_IN_PLACE,Nplocs_all(mpirank),MPI_DOUBLE_PRECISION,0,MPI_COMM_WORLD,ierr)
CALL MPI_Scatterv(parts%UR,sendcounts, displs,MPI_DOUBLE_PRECISION,MPI_IN_PLACE,Nplocs_all(mpirank),MPI_DOUBLE_PRECISION,0,MPI_COMM_WORLD,ierr)
CALL MPI_Scatterv(parts%UTHET,sendcounts, displs,MPI_DOUBLE_PRECISION,MPI_IN_PLACE,Nplocs_all(mpirank),MPI_DOUBLE_PRECISION,0,MPI_COMM_WORLD,ierr)
CALL MPI_Scatterv(parts%Rindex,sendcounts, displs,MPI_INTEGER,MPI_IN_PLACE,Nplocs_all(mpirank),MPI_INTEGER,0,MPI_COMM_WORLD,ierr)
CALL MPI_Scatterv(parts%Zindex,sendcounts, displs,MPI_INTEGER,MPI_IN_PLACE,Nplocs_all(mpirank),MPI_INTEGER,0,MPI_COMM_WORLD,ierr)
CALL MPI_Scatterv(parts%partindex,sendcounts, displs,MPI_INTEGER,MPI_IN_PLACE,Nplocs_all(mpirank),MPI_INTEGER,0,MPI_COMM_WORLD,ierr)
ELSE
CALL MPI_Bcast(nplasma,1,MPI_INTEGER,0,MPI_COMM_WORLD,ierr) ! nplasma used temporarily to store required size for parts
CALL creat_parts(parts,nplasma)
CALL MPI_Scatter(Nplocs_all,1,MPI_INTEGER, Nplocs_all(mpirank), 1,MPI_INTEGER,0,MPI_COMM_WORLD,ierr)
sendcounts=2
CALL MPI_Scatterv(Zbounds,sendcounts,displs,MPI_INTEGER,Zbounds(mpirank),2,MPI_INTEGER,0,MPI_COMM_WORLD,ierr)
parts%Nptot=Nplocs_all(mpirank)
WRITE(*,*) mpirank, "received Nplocs_all : ", Nplocs_all
WRITE(*,*) mpirank, "received Zbounds : ", Zbounds
CALL MPI_BARRIER(MPI_COMM_WORLD, ierr)
CALL MPI_Scatterv(parts%R,sendcounts, displs,MPI_DOUBLE_PRECISION,parts%R,Nplocs_all(mpirank),MPI_DOUBLE_PRECISION,0,MPI_COMM_WORLD,ierr)
CALL MPI_Scatterv(parts%Z,sendcounts, displs,MPI_DOUBLE_PRECISION,parts%Z,Nplocs_all(mpirank),MPI_DOUBLE_PRECISION,0,MPI_COMM_WORLD,ierr)
CALL MPI_Scatterv(parts%THET,sendcounts, displs,MPI_DOUBLE_PRECISION,parts%THET,Nplocs_all(mpirank),MPI_DOUBLE_PRECISION,0,MPI_COMM_WORLD,ierr)
CALL MPI_Scatterv(parts%UZ,sendcounts, displs,MPI_DOUBLE_PRECISION,parts%UZ,Nplocs_all(mpirank),MPI_DOUBLE_PRECISION,0,MPI_COMM_WORLD,ierr)
CALL MPI_Scatterv(parts%UR,sendcounts, displs,MPI_DOUBLE_PRECISION,parts%UR,Nplocs_all(mpirank),MPI_DOUBLE_PRECISION,0,MPI_COMM_WORLD,ierr)
CALL MPI_Scatterv(parts%UTHET,sendcounts, displs,MPI_DOUBLE_PRECISION,parts%UTHET,Nplocs_all(mpirank),MPI_DOUBLE_PRECISION,0,MPI_COMM_WORLD,ierr)
CALL MPI_Scatterv(parts%Rindex,sendcounts, displs,MPI_INTEGER,parts%Rindex,Nplocs_all(mpirank),MPI_INTEGER,0,MPI_COMM_WORLD,ierr)
CALL MPI_Scatterv(parts%Zindex,sendcounts, displs,MPI_INTEGER,parts%Zindex,Nplocs_all(mpirank),MPI_INTEGER,0,MPI_COMM_WORLD,ierr)
CALL MPI_Scatterv(parts%partindex,sendcounts, displs,MPI_INTEGER,parts%partindex,Nplocs_all(mpirank),MPI_INTEGER,0,MPI_COMM_WORLD,ierr)
END IF
CALL MPI_Bcast(nplasma,1,MPI_DOUBLE_PRECISION,0,MPI_COMM_WORLD,ierr)
CALL MPI_Bcast(V,1,MPI_DOUBLE_PRECISION,0,MPI_COMM_WORLD,ierr)
CALL MPI_Bcast(qsim,1,MPI_DOUBLE_PRECISION,0,MPI_COMM_WORLD,ierr)
CALL MPI_Bcast(msim,1,MPI_DOUBLE_PRECISION,0,MPI_COMM_WORLD,ierr)
END IF
! Compute Self consistent electric field
CALL bound
CALL localisation
! Init Electric and Magnetic Fields
CALL fields_init
IF (nlmaxwellsource) CALL maxwellsource_init(lu_in, time)
CALL rhscon
CALL poisson
IF (newres) THEN
CALL diagnostics
END IF
!
END SUBROUTINE resume
c4science · Help