1 |
!======================================================================= |
2 |
subroutine couprecv_r4( component, dataname, Nx, Ny, arr ) |
3 |
implicit none |
4 |
! Arguments |
5 |
character*(*) component |
6 |
character*(*) dataname |
7 |
integer Nx,Ny |
8 |
real*4 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_r4: 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_r4: compind = ',compind |
32 |
stop 'couprecv_r4: numprocs < 1' |
33 |
endif |
34 |
if (VERB) |
35 |
& write(LogUnit,*) 'couprecv_r4: ',component_Name(compind) |
36 |
if (VERB) |
37 |
& write(LogUnit,*) 'couprecv_r4: dataname=',dataname |
38 |
|
39 |
! Foreach component process |
40 |
do n=1,numprocs |
41 |
|
42 |
! Receive message |
43 |
count=HEADER_SIZE+MAX_R4_BUFLEN |
44 |
dtype=MPI_REAL |
45 |
tag=generate_tag(102,n,dataname) |
46 |
rank=rank_component_procs(n,compind) |
47 |
if (VERB) then |
48 |
write(LogUnit,*) 'couprecv_r4: calling MPI_Recv rank=',rank, |
49 |
& ' proc=',n,'/',numprocs |
50 |
call flush(LogUnit) |
51 |
endif |
52 |
call MPI_Recv(r4buf, count, dtype, rank, tag, comm, stat, ierr) |
53 |
if (VERB) then |
54 |
write(LogUnit,*) 'couprecv_r4: returned ierr=',ierr |
55 |
call flush(LogUnit) |
56 |
endif |
57 |
|
58 |
if (ierr.ne.0) then |
59 |
write(LogUnit,*) 'couprecv_r4tiles: rank(W,G)=', |
60 |
& my_rank_in_world,my_rank_in_global, |
61 |
& ' ierr=',ierr |
62 |
stop 'couprecv_r4: MPI_Recv failed' |
63 |
endif |
64 |
|
65 |
! Extract header |
66 |
Io=int(0.5+r4buf(1)) |
67 |
Ni=int(0.5+r4buf(2)) |
68 |
Jo=int(0.5+r4buf(3)) |
69 |
Nj=int(0.5+r4buf(4)) |
70 |
|
71 |
if (Io+Ni-1.gt.Nx .or. Io.lt.1) then |
72 |
write(LogUnit,*) 'couprecv_r4tiles: Io,Ni=',Io,Ni |
73 |
stop 'couprecv_r4: Incompatible header/target array' |
74 |
endif |
75 |
if (Jo+Nj-1.gt.Ny .or. Jo.lt.1) then |
76 |
write(LogUnit,*) 'couprecv_r4tiles: Jo,Nj=',Jo,Nj |
77 |
stop 'couprecv_r4: Incompatible header/target array' |
78 |
endif |
79 |
|
80 |
call mitcplr_real2char( r4buf(9), recvdname ) |
81 |
if (recvdname .ne. dataname) then |
82 |
write(LogUnit,*) 'couprecv_r4: recvdname = ',recvdname |
83 |
write(LogUnit,*) 'couprecv_r4: dataname = ',dataname |
84 |
stop 'couprecv_r4: 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)=r4buf(ij) |
92 |
enddo |
93 |
enddo |
94 |
|
95 |
enddo |
96 |
|
97 |
! ------------------------------------------------------------------ |
98 |
return |
99 |
end |
100 |
!======================================================================= |