mpiswitch.f90

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

        subroutine mpiswitch(j3,n1dfft,Jp2st,J2st,lot,n1,nd2proc,nd3proc,nproc,ioption,zmpi1,zw)
        implicit real*8 (a-h,o-z)
        dimension zmpi1(2,n1,nd2proc,nd3proc,nproc),zw(2,lot,n1)

        mfft=0
	if (ioption /= 1) then
        do 300,Jp2=Jp2st,nproc
        do 200,J2=J2st,nd2proc
        mfft=mfft+1
        if (mfft.gt.n1dfft) then
        Jp2st=Jp2
        J2st=J2
        return
        endif
        do 100,I1=1,n1
        zw(1,mfft,I1)=zmpi1(1,I1,J2,j3,Jp2)
        zw(2,mfft,I1)=zmpi1(2,I1,J2,j3,Jp2)
100     continue
200     continue
        J2st=1
300     continue
	else
        do 600,Jp2=Jp2st,nproc
        do 500,J2=J2st,nd2proc
        mfft=mfft+1
        if (mfft.gt.n1dfft) then
        Jp2st=Jp2
        J2st=J2
        return
        endif
	ind=(Jp2-1) * nd2proc + J2
	jj2=(ind-1)/nproc +1
	jjp2=modulo(ind,nproc) +1
        do 400,I1=1,n1
!        zw(1,mfft,I1)=zmpi1(1,I1,J2,j3,Jp2)
!        zw(2,mfft,I1)=zmpi1(2,I1,J2,j3,Jp2)
         zw(1,mfft,I1)=zmpi1(1,I1,jj2,j3,jjp2)
         zw(2,mfft,I1)=zmpi1(2,I1,jj2,j3,jjp2)	
400     continue
500     continue
        J2st=1
600     continue
	endif
        end