mpiswitch_corn.f90

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

        subroutine mpiswitch_corn(j3,n1dfft,Jp2stb,J2stb,lot,n1,md2,nd3,nproc,zmpi1,zw)
        implicit real*8 (a-h,o-z)
        dimension zmpi1(2,n1/2,md2/nproc,nd3/nproc,nproc),zw(2,lot,n1)
! WARNING: Assuming that high frequencies are in the corners
!          and that n1 is multiple of 4

        mfft=0
        do 300,Jp2=Jp2stb,nproc
        do 200,J2=J2stb,md2/nproc
        mfft=mfft+1
        if (mfft.gt.n1dfft) then
        Jp2stb=Jp2
        J2stb=J2
        return
        endif
        do 90,I1=1,n1/4
        zw(1,mfft,I1)=0.d0
        zw(2,mfft,I1)=0.d0
90      continue
        do 100,I1=n1/4+1,3*n1/4
        zw(1,mfft,I1)=zmpi1(1,I1-n1/4,J2,j3,Jp2)
        zw(2,mfft,I1)=zmpi1(2,I1-n1/4,J2,j3,Jp2)
100     continue
        do 110,I1=3*n1/4+1,n1
        zw(1,mfft,I1)=0.d0
        zw(2,mfft,I1)=0.d0
110     continue
200     continue
        J2stb=1
300     continue
        end