1 |
jmc |
1.3 |
C $Header: /u/gcmpack/MITgcm/pkg/compon_communic/mitcoupler_tile_register.F,v 1.2 2007/10/08 23:58:20 jmc Exp $ |
2 |
jmc |
1.2 |
C $Name: $ |
3 |
|
|
|
4 |
jmc |
1.1 |
!======================================================================= |
5 |
|
|
subroutine MITCOUPLER_tile_register( compName, nnx, nny ) |
6 |
|
|
implicit none |
7 |
|
|
|
8 |
|
|
! Arguments |
9 |
|
|
character*(*) compName |
10 |
|
|
integer nnx, nny |
11 |
|
|
|
12 |
|
|
! MPI variables |
13 |
|
|
#include "mpif.h" |
14 |
|
|
|
15 |
|
|
! Predefined constants/arrays |
16 |
|
|
#include "CPLR_SIG.h" |
17 |
|
|
|
18 |
|
|
! Functions |
19 |
|
|
integer mitcplr_match_comp |
20 |
|
|
integer generate_tag |
21 |
jmc |
1.3 |
external mitcplr_match_comp |
22 |
|
|
external generate_tag |
23 |
jmc |
1.1 |
|
24 |
|
|
! Local |
25 |
|
|
integer n,numprocs |
26 |
|
|
integer comm |
27 |
|
|
integer compind,count,dtype,tag,rank |
28 |
jmc |
1.3 |
integer ierr |
29 |
jmc |
1.1 |
integer stat(MPI_STATUS_SIZE) |
30 |
jmc |
1.3 |
integer j, numtiles |
31 |
|
|
integer nx, ny, i0, j0 |
32 |
|
|
integer ibuf(MAX_IBUF) |
33 |
jmc |
1.1 |
|
34 |
|
|
! ------------------------------------------------------------------ |
35 |
|
|
|
36 |
jmc |
1.3 |
write(LogUnit,'(3A)') |
37 |
|
|
& 'MITCOUPLER_tile_register: do "', compName, '" :' |
38 |
|
|
|
39 |
jmc |
1.1 |
! Establish who I am communicating with |
40 |
|
|
compind=mitcplr_match_comp( compName ) |
41 |
jmc |
1.3 |
if (compind.le.0) STOP 'MITCOUPLER_tile_register: Bad component' |
42 |
jmc |
1.1 |
comm=MPI_COMM_compcplr( compind ) |
43 |
|
|
numprocs=num_component_procs(compind) |
44 |
jmc |
1.3 |
|
45 |
|
|
write(LogUnit,'(2(A,I6))') |
46 |
|
|
& ' compind=', compind, ' ; numprocs=', numprocs |
47 |
jmc |
1.1 |
if (numprocs.lt.1) then |
48 |
jmc |
1.3 |
STOP 'MITCOUPLER_tile_register: numprocs < 1' |
49 |
jmc |
1.1 |
endif |
50 |
|
|
|
51 |
|
|
! Foreach component process |
52 |
|
|
do n=1,numprocs |
53 |
|
|
|
54 |
|
|
! Receive message |
55 |
|
|
count=MAX_IBUF |
56 |
|
|
dtype=MPI_INTEGER |
57 |
|
|
tag=generate_tag(112,n,'Register Tiles') |
58 |
|
|
rank=rank_component_procs(n,compind) |
59 |
|
|
|
60 |
|
|
call MPI_Recv(ibuf, count, dtype, rank, tag, comm, stat, ierr) |
61 |
jmc |
1.2 |
|
62 |
jmc |
1.1 |
if (ierr.ne.0) then |
63 |
|
|
write(LogUnit,*) 'MITCOUPLER_tile_register: rank(W,G)=', |
64 |
|
|
& my_rank_in_world,my_rank_in_global, |
65 |
|
|
& ' ierr=',ierr |
66 |
jmc |
1.3 |
STOP 'MITCOUPLER_tile_register: MPI_Recv failed' |
67 |
jmc |
1.1 |
endif |
68 |
|
|
|
69 |
|
|
numtiles=ibuf(1) |
70 |
|
|
if (numtiles.lt.1 .or. numtiles.gt.MAX_TILES) then |
71 |
|
|
write(LogUnit,*) 'MITCOUPLER_tile_register: #tiles = ',numtiles |
72 |
jmc |
1.3 |
STOP 'MITCOUPLER_tile_register: invalid value for numtiles' |
73 |
jmc |
1.1 |
endif |
74 |
|
|
component_num_tiles(n,compind)=numtiles |
75 |
|
|
|
76 |
jmc |
1.3 |
write(LogUnit,'(3(A,I6),A)') '- proc # =', n, |
77 |
|
|
& ' ; rank=', rank, ' ; numtiles=', numtiles, ' -------' |
78 |
jmc |
1.1 |
do j=1,numtiles |
79 |
|
|
|
80 |
|
|
! Receive message |
81 |
|
|
count=MAX_IBUF |
82 |
|
|
dtype=MPI_INTEGER |
83 |
|
|
tag=generate_tag(113,j,'Register each tile') |
84 |
|
|
rank=rank_component_procs(n,compind) |
85 |
|
|
|
86 |
|
|
call MPI_Recv(ibuf, count, dtype, rank, tag, comm, stat, ierr) |
87 |
|
|
|
88 |
|
|
if (ierr.ne.0) then |
89 |
|
|
write(LogUnit,*) 'MITCOUPLER_tile_register: rank(W,G)=', |
90 |
|
|
& my_rank_in_world,my_rank_in_global, |
91 |
|
|
& ' ierr=',ierr |
92 |
jmc |
1.3 |
STOP 'MITCOUPLER_tile_register: MPI_Recv failed' |
93 |
jmc |
1.1 |
endif |
94 |
|
|
|
95 |
jmc |
1.3 |
! Extract data and store |
96 |
|
|
nx = ibuf(1) |
97 |
|
|
ny = ibuf(2) |
98 |
|
|
i0 = ibuf(3) |
99 |
|
|
j0 = ibuf(4) |
100 |
|
|
component_tile_nx(j,n,compind) = nx |
101 |
|
|
component_tile_ny(j,n,compind) = ny |
102 |
|
|
component_tile_i0(j,n,compind) = i0 |
103 |
|
|
component_tile_j0(j,n,compind) = j0 |
104 |
|
|
|
105 |
|
|
! Print and check |
106 |
|
|
write(LogUnit,'(A,I5,A,2I5,A,2I8)') ' tile #:', j, |
107 |
|
|
& ' ; Ni,Nj=', nx, ny, ' ; Io,Jo=', i0, j0 |
108 |
|
|
|
109 |
|
|
if (nx.lt.1) then |
110 |
|
|
STOP 'MITCOUPLER_tile_register: invalid value for nx' |
111 |
|
|
endif |
112 |
|
|
if (ny.lt.1) then |
113 |
|
|
STOP 'MITCOUPLER_tile_register: invalid value for ny' |
114 |
|
|
endif |
115 |
|
|
if (i0.lt.1) then |
116 |
|
|
STOP 'MITCOUPLER_tile_register: invalid value for i0' |
117 |
|
|
endif |
118 |
|
|
if (j0.lt.1) then |
119 |
|
|
STOP 'MITCOUPLER_tile_register: invalid value for j0' |
120 |
|
|
endif |
121 |
|
|
if (i0+nx-1.gt.nnx) then |
122 |
|
|
STOP 'MITCOUPLER_tile_register: i0 + nx -1 > nnx' |
123 |
|
|
endif |
124 |
|
|
if (j0+ny-1.gt.nny) then |
125 |
|
|
STOP 'MITCOUPLER_tile_register: j0 + ny -1 > nny' |
126 |
jmc |
1.1 |
endif |
127 |
|
|
|
128 |
|
|
enddo ! j |
129 |
jmc |
1.3 |
write(LogUnit,'(A,2I8,2(A,I8))') |
130 |
|
|
& ' rank(W,G)=', my_rank_in_world, my_rank_in_global, |
131 |
|
|
& ' , rank = ',rank, ' , num_tiles = ', numtiles |
132 |
jmc |
1.1 |
|
133 |
|
|
enddo ! n |
134 |
|
|
|
135 |
jmc |
1.3 |
write(LogUnit,'(3A)') 'MITCOUPLER_tile_register: comp. "', |
136 |
|
|
& compName, '" done' |
137 |
|
|
|
138 |
jmc |
1.1 |
! ------------------------------------------------------------------ |
139 |
|
|
call flush(LogUnit) |
140 |
|
|
return |
141 |
|
|
end |
142 |
|
|
!======================================================================= |