mpiswitch_cent.f90

来自「基于linux操作下的fortran快速付立变换的程序」· F90 代码 · 共 40 行

        subroutine mpiswitch_cent(j3,n1dfft,Jp2stb,J2stb,lot,max1,md1,m1,n1,md2proc,nd3proc,nproc,zmpi1,zw)
        implicit real*8 (a-h,o-z)
        dimension zmpi1(2,md1,md2proc,nd3proc,nproc),zw(2,lot,n1)

        mfft=0
        do 300,Jp2=Jp2stb,nproc
        do 200,J2=J2stb,md2proc
        mfft=mfft+1
        if (mfft.gt.n1dfft) then
        Jp2stb=Jp2
        J2stb=J2
        return
        endif

!       Here, zero and positive frequencies
!       In zmpi1, they are stored from 1 to max1+1
        do 90,I1=1,max1+1
        zw(1,mfft,I1)=zmpi1(1,I1,J2,j3,Jp2)
        zw(2,mfft,I1)=zmpi1(2,I1,J2,j3,Jp2)
90      continue

!       Fill the center region with zeros
        do 100,I1=max1+2,n1-m1+max1+1
        zw(1,mfft,I1)=0.d0
        zw(2,mfft,I1)=0.d0
100     continue

!       Here, negative frequencies
!       In zmpi1, they are stored from 1 to m1half
        do 110,I1=max1+2,m1
        zw(1,mfft,I1+n1-m1)=zmpi1(1,I1,J2,j3,Jp2)
        zw(2,mfft,I1+n1-m1)=zmpi1(2,I1,J2,j3,Jp2)
110     continue

200     continue
        J2stb=1
300     continue
        end