1 |
jmc |
1.2 |
C $Header: $ |
2 |
|
|
C $Name: $ |
3 |
|
|
|
4 |
jmc |
1.1 |
!======================================================================= |
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 |
|
|
!======================================================================= |