spmd_exch_nodareai.F File Reference

#include "implicit_f.inc"
#include "spmd.inc"
#include "com01_c.inc"
#include "task_c.inc"

Go to the source code of this file.

Functions/Subroutines
subroutine	spmd_exch_nodareai (nodareai, iad_elem, fr_elem, lenr, weight)

Function/Subroutine Documentation

spmd_exch_nodareai()

subroutine spmd_exch_nodareai	(	integer, dimension(*)	nodareai,
		integer, dimension(2,*)	iad_elem,
		integer, dimension(*)	fr_elem,
		integer	lenr,
		integer, dimension(*)	weight )

Definition at line 36 of file spmd_exch_nodareai.F.

C-----------------------------------------------
C   I m p l i c i t   T y p e s
C-----------------------------------------------
        USE spmd_comm_world_mod, ONLY : spmd_comm_world
#include      "implicit_f.inc"
C-----------------------------------------------------------------
C   M e s s a g e   P a s s i n g
C-----------------------------------------------
#include "spmd.inc"
C-----------------------------------------------
C   C o m m o n   B l o c k s
C-----------------------------------------------
#include      "com01_c.inc"
#include      "task_c.inc"
C-----------------------------------------------
C   D u m m y   A r g u m e n t s
C-----------------------------------------------
      INTEGER  IAD_ELEM(2,*),FR_ELEM(*), SIZE, LENR,WEIGHT(*)
      INTEGER  NODAREAI(*)
C-----------------------------------------------
C   L o c a l  V a r i a b l e s
C-----------------------------------------------
#ifdef MPI
      INTEGER MSGTYP,I,NOD,LOC_PROC,IERROR,
     .        SIZ,J,K,L,NB_NOD,
     .        STATUS(MPI_STATUS_SIZE),
     .        IAD_SEND(NSPMD+1),IAD_RECV(NSPMD+1),
     .        REQ_R(NSPMD),REQ_S(NSPMD),MSGOFF
 
      INTEGER, DIMENSION(:),ALLOCATABLE :: SBUF,RBUF
      DATA msgoff/7035/
C-----------------------------------------------
C   S o u r c e  L i n e s
C-----------------------------------------------
      ALLOCATE(sbuf(lenr))
      ALLOCATE(rbuf(lenr))
C-----------------------------------------------
      loc_proc = ispmd+1
      l = 1
      iad_recv(1)=1
 
C   Mise en place des buffers de reception
      DO i=1,nspmd
         siz = iad_elem(1,i+1)-iad_elem(1,i)
         IF(siz/=0) THEN
           msgtyp = msgoff
           CALL mpi_irecv(
     s      rbuf(l),siz,mpi_integer,it_spmd(i),msgtyp,
     g      spmd_comm_world,req_r(i),ierror)
           l = l+siz
         END IF
         iad_recv(i+1) = l
      END DO
 
! Remplissage des buffers d'envoi et envoi des donnees
 
      l=1
      iad_send(1)=1
      DO i=1,nspmd
         DO j=iad_elem(1,i),iad_elem(1,i+1)-1
            nod = fr_elem(j)
            sbuf(l) = nodareai(nod)
            l = l+1
         END DO
        iad_send(i+1) = l
      END DO
 
      DO i=1,nspmd
 
       IF(iad_elem(1,i+1)-iad_elem(1,i)>0)THEN
 
          
          msgtyp = msgoff
          siz = iad_send(i+1)-iad_send(i)
          l = iad_send(i)
 
          CALL mpi_isend(
     s      sbuf(l),siz,mpi_integer,it_spmd(i),msgtyp,
     g      spmd_comm_world,req_s(i),ierror)
       ENDIF                
      END DO
! Reception et traitement
      DO i = 1, nspmd
C
        nb_nod = iad_elem(1,i+1)-iad_elem(1,i)
        IF(nb_nod>0)THEN
          CALL mpi_wait(req_r(i),status,ierror)
          l = iad_recv(i)
 
          DO j=iad_elem(1,i),iad_elem(1,i+1)-1
 
             nod = fr_elem(j)
             nodareai(nod) = nodareai(nod) + rbuf(l)
             l = l+1
          ENDDO
        ENDIF
      ENDDO
 
      DO i = 1, nspmd
        IF(iad_elem(1,i+1)-iad_elem(1,i)>0)THEN
          CALL mpi_wait(req_s(i),status,ierror)
        ENDIF
      ENDDO
C-----------------------------------------------
      DEALLOCATE(sbuf)
      DEALLOCATE(rbuf)
C-----------------------------------------------
#endif
      RETURN