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