| 1 |
jmc |
1.1 |
!======================================================================= |
| 2 |
|
|
subroutine couprecv_r8( component, dataname, Nx, Ny, arr ) |
| 3 |
|
|
implicit none |
| 4 |
|
|
! Arguments |
| 5 |
|
|
character*(*) component |
| 6 |
|
|
character*(*) dataname |
| 7 |
|
|
integer Nx,Ny |
| 8 |
|
|
real*8 arr(Nx,Ny) |
| 9 |
|
|
! Predefined constants/arrays |
| 10 |
|
|
#include "CPLR_SIG.h" |
| 11 |
|
|
! MPI variables |
| 12 |
|
|
#include "mpif.h" |
| 13 |
|
|
integer count,dtype,rank,tag,comm,ierr |
| 14 |
|
|
integer stat(MPI_STATUS_SIZE) |
| 15 |
|
|
! Functions |
| 16 |
|
|
integer mitcplr_match_comp |
| 17 |
|
|
integer generate_tag |
| 18 |
|
|
! Local |
| 19 |
|
|
integer compind,numprocs |
| 20 |
|
|
integer i,j,ij,n |
| 21 |
|
|
integer Ni,Io,Nj,Jo |
| 22 |
|
|
character*(MAXLEN_COMP_NAME) recvdname |
| 23 |
|
|
! ------------------------------------------------------------------ |
| 24 |
|
|
|
| 25 |
|
|
! Establish who I am communicating with |
| 26 |
|
|
compind=mitcplr_match_comp( component ) |
| 27 |
|
|
if (compind.le.0) stop 'couprecv_r8: Bad component id' |
| 28 |
|
|
comm=MPI_COMM_compcplr( compind ) |
| 29 |
|
|
numprocs=num_component_procs(compind) |
| 30 |
|
|
if (numprocs.lt.1) then |
| 31 |
|
|
write(LogUnit,*) 'couprecv_r8: compind = ',compind |
| 32 |
|
|
stop 'couprecv_r8: numprocs < 1' |
| 33 |
|
|
endif |
| 34 |
|
|
if (VERB) |
| 35 |
|
|
& write(LogUnit,*) 'couprecv_r8: ',component_Name(compind) |
| 36 |
|
|
if (VERB) |
| 37 |
|
|
& write(LogUnit,*) 'couprecv_r8: dataname=',dataname |
| 38 |
|
|
|
| 39 |
|
|
! Foreach component process |
| 40 |
|
|
do n=1,numprocs |
| 41 |
|
|
|
| 42 |
|
|
! Receive message |
| 43 |
|
|
count=HEADER_SIZE+MAX_R8_BUFLEN |
| 44 |
|
|
dtype=MPI_DOUBLE_PRECISION |
| 45 |
|
|
tag=generate_tag(102,n,dataname) |
| 46 |
|
|
rank=rank_component_procs(n,compind) |
| 47 |
|
|
if (VERB) then |
| 48 |
|
|
write(LogUnit,*) 'couprecv_r8: calling MPI_Recv rank=',rank, |
| 49 |
|
|
& ' proc=',n,'/',numprocs |
| 50 |
|
|
call flush(LogUnit) |
| 51 |
|
|
endif |
| 52 |
|
|
call MPI_Recv(r8buf, count, dtype, rank, tag, comm, stat, ierr) |
| 53 |
|
|
if (VERB) then |
| 54 |
|
|
write(LogUnit,*) 'couprecv_r8: returned ierr=',ierr |
| 55 |
|
|
call flush(LogUnit) |
| 56 |
|
|
endif |
| 57 |
|
|
|
| 58 |
|
|
if (ierr.ne.0) then |
| 59 |
|
|
write(LogUnit,*) 'couprecv_r8tiles: rank(W,G)=', |
| 60 |
|
|
& my_rank_in_world,my_rank_in_global, |
| 61 |
|
|
& ' ierr=',ierr |
| 62 |
|
|
stop 'couprecv_r8: MPI_Recv failed' |
| 63 |
|
|
endif |
| 64 |
|
|
|
| 65 |
|
|
! Extract header |
| 66 |
|
|
Io=int(0.5+r8buf(1)) |
| 67 |
|
|
Ni=int(0.5+r8buf(2)) |
| 68 |
|
|
Jo=int(0.5+r8buf(3)) |
| 69 |
|
|
Nj=int(0.5+r8buf(4)) |
| 70 |
|
|
|
| 71 |
|
|
if (Io+Ni-1.gt.Nx .or. Io.lt.1) then |
| 72 |
|
|
write(LogUnit,*) 'couprecv_r8tiles: Io,Ni=',Io,Ni |
| 73 |
|
|
stop 'couprecv_r8: Incompatible header/target array' |
| 74 |
|
|
endif |
| 75 |
|
|
if (Jo+Nj-1.gt.Ny .or. Jo.lt.1) then |
| 76 |
|
|
write(LogUnit,*) 'couprecv_r8tiles: Jo,Nj=',Jo,Nj |
| 77 |
|
|
stop 'couprecv_r8: Incompatible header/target array' |
| 78 |
|
|
endif |
| 79 |
|
|
|
| 80 |
|
|
call mitcplr_real2char( r8buf(9), recvdname ) |
| 81 |
|
|
if (recvdname .ne. dataname) then |
| 82 |
|
|
write(LogUnit,*) 'couprecv_r8: recvdname = ',recvdname |
| 83 |
|
|
write(LogUnit,*) 'couprecv_r8: dataname = ',dataname |
| 84 |
|
|
stop 'couprecv_r8: recvdname != dataname' |
| 85 |
|
|
endif |
| 86 |
|
|
|
| 87 |
|
|
! Extract data |
| 88 |
|
|
do j=1,Nj |
| 89 |
|
|
do i=1,Ni |
| 90 |
|
|
ij=HEADER_SIZE+i+Ni*(j-1) |
| 91 |
|
|
arr(Io+i-1,Jo+j-1)=r8buf(ij) |
| 92 |
|
|
enddo |
| 93 |
|
|
enddo |
| 94 |
|
|
|
| 95 |
|
|
enddo |
| 96 |
|
|
|
| 97 |
|
|
! ------------------------------------------------------------------ |
| 98 |
|
|
return |
| 99 |
|
|
end |
| 100 |
|
|
!======================================================================= |
Parent Directory
| 
Revision Log
| 
Revision Graph