SLiMv2.1/main.f90

Go to the documentation of this file.

program MPI_wrapper
  implicit none
  include 'param.f90'
  integer i,j,k,narr(nz_ghostglobal),kk_send(nz_max) 
  
  call mpi_init(ierr)

! Give each node its number and the total number of nodes.
! Node 0 will handle all io. 
! Also mark the top and bottom boxes. 
  !write(*,*) "Main"
  call mpi_comm_size (MPI_COMM_WORLD, nprocs, ierr)
  call mpi_comm_rank (MPI_COMM_WORLD, my_rank, ierr)
  istop=.false.
  isbottom=.false.  
  !write(*,*) my_rank, ": OK ",nprocs
  if(my_rank .eq.0) isbottom=.true.
  if(my_rank .eq. nprocs-1) istop=.true.

  do k=1,nz_ghostglobal
        narr(k)=k
  enddo   

! determine how to split data

  dn=nz_global/(nprocs)
!excludes ghosts
  do i=1,nprocs
     start(i)=(i-1)*dn+1+2
     ends(i)=(i)*dn+2
  enddo
  start(1)=3
  ends(nprocs)=nz_global+2 
        
 
!the individual nodes need only to know their size and locations (kk)
  if(my_rank .eq. 0) then
     nz=(ends(1)-start(1)+1)+4
        kk(1:ends(1)-start(1)+5)=narr(start(1)-2:ends(1)+2)
     if(nprocs .gt. 1) then 
     do i=1,nprocs-1
        itemp=(ends(i+1)-start(i+1)+1)+4
        call mpi_send(itemp,1,MPI_INTEGER, i,1,MPI_COMM_WORLD, ierr)
        kk_send=0
!converts local index (k) to global index (kk(k))
        kk_send(1:ends(i+1)-start(i+1)+5)=narr(start(i+1)-2:ends(i+1)+2)
        call mpi_send(kk_send,nz_max,MPI_INTEGER, i,1,MPI_COMM_WORLD, ierr)
     enddo
     endif
  endif
  if(my_rank .ne. 0) then
     call mpi_recv(nz,1,MPI_INTEGER, 0,1,MPI_COMM_WORLD,istatus,ierr)
     call mpi_recv(kk,nz_max,MPI_INTEGER, 0,1,MPI_COMM_WORLD,istatus,ierr)
  endif
 if (my_rank .eq. nprocs-1) nz=nz    !-1
 do i=1,nx
    do j=1,ny
       zz(i,j,:)=6.959e10-(1000.-kk)*dz
    enddo
 enddo   
!!!!!!!!!!!!!!!!!!!!!!now run the program!!!!!!!!!!!!!!!!!!!!!!!!!!!!
  call wave()
  call mpi_finalize(ierr)  
endprogram MPI_wrapper
!!!!!!!!!!!!!!!!!!!!!!!!!!!!!!!!!!!!!!!!!!!!!!!!!!!!!!!!!!!!!!!!!!!!!!!!!!!!!!!

---